@antv/l7-layers 2.25.7 → 2.25.9

package/es/wind/models/wind.js

@@ -0,0 +1,240 @@
+import _asyncToGenerator from "@babel/runtime/helpers/esm/asyncToGenerator";
+import _defineProperty from "@babel/runtime/helpers/esm/defineProperty";
+import { AttributeType, gl } from '@antv/l7-core';
+import { FrequencyController } from '@antv/l7-utils';
+import BaseModel from "../../core/BaseModel";
+import { RasterImageTriangulation } from "../../core/triangulation";
+/* babel-plugin-inline-import '../shaders/wind_frag.glsl' */
+const WindFrag = "precision mediump float;\nuniform float u_opacity: 1.0;\nuniform sampler2D u_texture;\nvarying vec2 v_texCoord;\nvoid main() {\n  vec4 color = texture2D(u_texture,vec2(v_texCoord.x,v_texCoord.y));\n  gl_FragColor = color;\n  gl_FragColor.a *= u_opacity;\n}\n";
+/* babel-plugin-inline-import '../shaders/wind_vert.glsl' */
+const WindVert = "precision highp float;\nuniform mat4 u_ModelMatrix;\n\nlayout(location = ATTRIBUTE_LOCATION_POSITION) in vec3 a_Position;\nlayout(location = ATTRIBUTE_LOCATION_UV) in vec2 a_Uv;\n\nvarying vec2 v_texCoord;\n#pragma include \"projection\"\nvoid main() {\n  v_texCoord = a_Uv;\n  vec4 project_pos = project_position(vec4(a_Position, 1.0));\n\n  gl_Position = project_common_position_to_clipspace(vec4(project_pos.xy, 0.0, 1.0));\n}\n";
+import { Wind } from "./windRender";
+const defaultRampColors = {
+  0.0: '#3288bd',
+  0.1: '#66c2a5',
+  0.2: '#abdda4',
+  0.3: '#e6f598',
+  0.4: '#fee08b',
+  0.5: '#fdae61',
+  0.6: '#f46d43',
+  1.0: '#d53e4f'
+};
+export default class WindModel extends BaseModel {
+  constructor(...args) {
+    super(...args);
+    _defineProperty(this, "texture", void 0);
+    _defineProperty(this, "colorModel", void 0);
+    _defineProperty(this, "wind", void 0);
+    _defineProperty(this, "imageCoords", void 0);
+    _defineProperty(this, "sizeScale", 0.5);
+    // https://mapbox.github.io/webgl-wind/demo/
+    // source: 'http://nomads.ncep.noaa.gov',
+    _defineProperty(this, "frequency", new FrequencyController(7.2));
+    _defineProperty(this, "cacheZoom", void 0);
+  }
+  get attributeLocation() {
+    return Object.assign(super.attributeLocation, {
+      MAX: super.attributeLocation.MAX,
+      UV: 9
+    });
+  }
+  render(options) {
+    this.drawColorMode(options);
+    // Tip: 控制风场的平均更新频率
+    this.frequency.run(() => {
+      this.drawWind();
+    });
+  }
+  getUninforms() {
+    throw new Error('Method not implemented.');
+  }
+  initModels() {
+    var _this = this;
+    return _asyncToGenerator(function* () {
+      var _source$data, _source$data2;
+      const {
+        uMin = -21.32,
+        uMax = 26.8,
+        vMin = -21.57,
+        vMax = 21.42,
+        fadeOpacity = 0.996,
+        speedFactor = 0.25,
+        dropRate = 0.003,
+        dropRateBump = 0.01,
+        rampColors = defaultRampColors,
+        sizeScale = 0.5
+        // mask
+      } = _this.layer.getLayerConfig();
+      const {
+        createTexture2D
+      } = _this.rendererService;
+      const source = _this.layer.getSource();
+      _this.texture = createTexture2D({
+        height: 0,
+        width: 0
+      });
+      _this.cacheZoom = Math.floor(_this.mapService.getZoom());
+      const glContext = _this.rendererService.getGLContext();
+      _this.imageCoords = (_source$data = source.data) === null || _source$data === void 0 ? void 0 : _source$data.dataArray[0].coordinates;
+      (_source$data2 = source.data) === null || _source$data2 === void 0 || (_source$data2 = _source$data2.images) === null || _source$data2 === void 0 || _source$data2.then(imageData => {
+        var _this$texture;
+        _this.sizeScale = sizeScale * _this.getZoomScale();
+        const {
+          imageWidth,
+          imageHeight
+        } = _this.getWindSize();
+        const options = {
+          glContext,
+          imageWidth,
+          imageHeight,
+          fadeOpacity,
+          speedFactor,
+          dropRate,
+          dropRateBump,
+          rampColors
+        };
+        _this.wind = new Wind(options);
+
+        // imageData[0] 风场图
+        _this.wind.setWind({
+          uMin,
+          uMax,
+          vMin,
+          vMax,
+          image: imageData[0]
+        });
+        (_this$texture = _this.texture) === null || _this$texture === void 0 || _this$texture.destroy();
+        _this.texture = createTexture2D({
+          width: imageWidth,
+          height: imageHeight
+        });
+        _this.layerService.reRender();
+      });
+      const model = yield _this.layer.buildLayerModel({
+        moduleName: 'wind',
+        vertexShader: WindVert,
+        fragmentShader: WindFrag,
+        defines: _this.getDefines(),
+        triangulation: RasterImageTriangulation,
+        primitive: gl.TRIANGLES,
+        depth: {
+          enable: false
+        }
+      });
+      _this.colorModel = model;
+      return [model];
+    })();
+  }
+  getWindSize() {
+    const p1 = this.mapService.lngLatToPixel(this.imageCoords[0]);
+    const p2 = this.mapService.lngLatToPixel(this.imageCoords[1]);
+    const imageWidth = Math.min(Math.floor((p2.x - p1.x) * this.sizeScale), 2048);
+    const imageHeight = Math.min(Math.floor((p1.y - p2.y) * this.sizeScale), 2048);
+    return {
+      imageWidth,
+      imageHeight
+    };
+  }
+  buildModels() {
+    var _this2 = this;
+    return _asyncToGenerator(function* () {
+      return _this2.initModels();
+    })();
+  }
+  clearModels() {
+    var _this$texture2, _this$wind;
+    (_this$texture2 = this.texture) === null || _this$texture2 === void 0 || _this$texture2.destroy();
+    (_this$wind = this.wind) === null || _this$wind === void 0 || _this$wind.destroy();
+  }
+  registerBuiltinAttributes() {
+    // point layer size;
+    this.styleAttributeService.registerStyleAttribute({
+      name: 'uv',
+      type: AttributeType.Attribute,
+      descriptor: {
+        name: 'a_Uv',
+        shaderLocation: this.attributeLocation.UV,
+        buffer: {
+          // give the WebGL driver a hint that this buffer may change
+          usage: gl.DYNAMIC_DRAW,
+          data: [],
+          type: gl.FLOAT
+        },
+        size: 2,
+        update: (feature, featureIdx, vertex) => {
+          return [vertex[3], vertex[4]];
+        }
+      }
+    });
+  }
+  getZoomScale() {
+    return Math.min((this.cacheZoom + 4) / 30 * 2, 2);
+  }
+  drawWind() {
+    if (this.wind) {
+      const {
+        uMin = -21.32,
+        uMax = 26.8,
+        vMin = -21.57,
+        vMax = 21.42,
+        numParticles = 65535,
+        fadeOpacity = 0.996,
+        speedFactor = 0.25,
+        dropRate = 0.003,
+        dropRateBump = 0.01,
+        rampColors = defaultRampColors,
+        sizeScale = 0.5
+      } = this.layer.getLayerConfig();
+      let newNumParticles = numParticles;
+      const currentZoom = Math.floor(this.mapService.getZoom());
+      if (typeof sizeScale === 'number' && sizeScale !== this.sizeScale || currentZoom !== this.cacheZoom) {
+        const zoomScale = this.getZoomScale();
+        this.sizeScale = sizeScale;
+        newNumParticles *= zoomScale;
+        const {
+          imageWidth,
+          imageHeight
+        } = this.getWindSize();
+        this.wind.reSize(imageWidth, imageHeight);
+        this.cacheZoom = currentZoom;
+      }
+      this.wind.updateWindDir(uMin, uMax, vMin, vMax);
+      this.wind.updateParticelNum(newNumParticles);
+      this.wind.updateColorRampTexture(rampColors);
+      this.wind.fadeOpacity = fadeOpacity;
+      this.wind.speedFactor = speedFactor;
+      this.wind.dropRate = dropRate;
+      this.wind.dropRateBump = dropRateBump;
+      const {
+        d,
+        w,
+        h
+      } = this.wind.draw();
+      // 恢复 L7 渲染流程中 gl 状态
+      this.rendererService.setBaseState();
+      this.texture.update({
+        data: d,
+        width: w,
+        height: h
+      });
+    }
+  }
+  drawColorMode(options = {}) {
+    var _this$colorModel;
+    const {
+      opacity
+    } = this.layer.getLayerConfig();
+    this.layerService.beforeRenderData(this.layer);
+    this.layer.hooks.beforeRender.call();
+    this.layerService.renderMask(this.layer.masks);
+    (_this$colorModel = this.colorModel) === null || _this$colorModel === void 0 || _this$colorModel.draw({
+      uniforms: {
+        u_opacity: opacity || 1.0,
+        u_texture: this.texture
+      },
+      blend: this.getBlend(),
+      stencil: this.getStencil(options)
+    });
+    this.layer.hooks.afterRender.call();
+  }
+}

package/es/wind/models/windRender.d.ts

@@ -0,0 +1,104 @@
+export interface IWindData {
+    uMin: number;
+    uMax: number;
+    vMin: number;
+    vMax: number;
+    image: HTMLImageElement;
+}
+export interface IWind {
+    width: number;
+    height: number;
+    fadeOpacity: number;
+    speedFactor: number;
+    dropRate: number;
+    dropRateBump: number;
+    setWind: (windData: IWindData) => void;
+    draw: () => {
+        d: Uint8Array;
+        w: number;
+        h: number;
+    };
+    updateParticelNum: (num: number) => void;
+    updateWindDir: (uMin: number, uMax: number, vMin: number, vMax: number) => void;
+    updateColorRampTexture: (rampColors: {
+        [key: number]: string;
+    }) => void;
+    reSize: (width: number, height: number) => void;
+    destroy: () => void;
+}
+export interface IWindProps {
+    glContext: WebGLRenderingContext;
+    imageWidth: number;
+    imageHeight: number;
+    fadeOpacity: number;
+    speedFactor: number;
+    dropRate: number;
+    dropRateBump: number;
+    rampColors: {
+        [key: number]: string;
+    };
+}
+export declare class Wind {
+    width: number;
+    height: number;
+    pixels: Uint8Array;
+    fadeOpacity: number;
+    speedFactor: number;
+    dropRate: number;
+    dropRateBump: number;
+    private gl;
+    private drawProgram;
+    private fullScreenProgram;
+    private updateProgram;
+    private rampColors;
+    private numParticles;
+    private numParticlesSize;
+    private particleStateResolution;
+    private quadBuffer;
+    private particleIndexBuffer;
+    private framebuffer;
+    private colorRampTexture;
+    private backgroundTexture;
+    private screenTexture;
+    private particleStateTexture0;
+    private particleStateTexture1;
+    private windTexture;
+    private windData;
+    constructor(options: IWindProps);
+    init(): void;
+    setWind(windData: IWindData): void;
+    /**
+     * 更新风场粒子数量
+     * @param num
+     */
+    updateParticelNum(num: number): void;
+    /**
+     * 更新风场风向风速
+     * @param uMin
+     * @param uMax
+     * @param vMin
+     * @param vMax
+     */
+    updateWindDir(uMin: number, uMax: number, vMin: number, vMax: number): void;
+    /**
+     * update rampColors
+     * @param rampColors
+     */
+    updateColorRampTexture(rampColors: {
+        [key: number]: string;
+    }): void;
+    isColorChanged(rampColors: {
+        [key: number]: string;
+    }): boolean;
+    reSize(width: number, height: number): void;
+    draw(): {
+        d: Uint8Array;
+        w: number;
+        h: number;
+    };
+    drawScreen(): void;
+    drawFullTexture(texture: any, opacity: number): void;
+    drawParticles(): void;
+    updateParticles(): void;
+    destroy(): void;
+}

package/es/wind/models/windRender.js

@@ -0,0 +1,333 @@
+import _defineProperty from "@babel/runtime/helpers/esm/defineProperty";
+import * as glUtils from "./utils";
+import { drawFrag, drawVert, fullScreenFrag, fullScreenVert, updateFrag, updateVert } from "./windShader";
+function getColorRamp(colors) {
+  let canvas = document.createElement('canvas');
+  const ctx = canvas.getContext('2d');
+  canvas.width = 256;
+  canvas.height = 1;
+  const gradient = ctx.createLinearGradient(0, 0, 256, 0);
+  for (const stop of Object.keys(colors)) {
+    gradient.addColorStop(+stop, colors[+stop]);
+  }
+  ctx.fillStyle = gradient;
+  ctx.fillRect(0, 0, 256, 1);
+
+  // @ts-ignore dispose canvas element
+  canvas = null;
+  return new Uint8Array(ctx.getImageData(0, 0, 256, 1).data);
+}
+function bindAttribute(gl, buffer, attribute, numComponents) {
+  gl.bindBuffer(gl.ARRAY_BUFFER, buffer);
+  gl.enableVertexAttribArray(attribute);
+  gl.vertexAttribPointer(attribute, numComponents, gl.FLOAT, false, 0, 0);
+}
+function bindFramebuffer(gl, framebuffer, texture) {
+  gl.bindFramebuffer(gl.FRAMEBUFFER, framebuffer);
+  if (texture) {
+    gl.framebufferTexture2D(gl.FRAMEBUFFER, gl.COLOR_ATTACHMENT0, gl.TEXTURE_2D, texture, 0);
+  }
+}
+export class Wind {
+  constructor(options) {
+    _defineProperty(this, "width", 512);
+    _defineProperty(this, "height", 512);
+    _defineProperty(this, "pixels", void 0);
+    _defineProperty(this, "fadeOpacity", void 0);
+    _defineProperty(this, "speedFactor", void 0);
+    _defineProperty(this, "dropRate", void 0);
+    _defineProperty(this, "dropRateBump", void 0);
+    _defineProperty(this, "gl", void 0);
+    _defineProperty(this, "drawProgram", void 0);
+    _defineProperty(this, "fullScreenProgram", void 0);
+    _defineProperty(this, "updateProgram", void 0);
+    _defineProperty(this, "rampColors", void 0);
+    _defineProperty(this, "numParticles", 65536);
+    _defineProperty(this, "numParticlesSize", void 0);
+    _defineProperty(this, "particleStateResolution", void 0);
+    _defineProperty(this, "quadBuffer", void 0);
+    _defineProperty(this, "particleIndexBuffer", void 0);
+    _defineProperty(this, "framebuffer", void 0);
+    _defineProperty(this, "colorRampTexture", void 0);
+    _defineProperty(this, "backgroundTexture", void 0);
+    _defineProperty(this, "screenTexture", void 0);
+    _defineProperty(this, "particleStateTexture0", void 0);
+    _defineProperty(this, "particleStateTexture1", void 0);
+    _defineProperty(this, "windTexture", void 0);
+    _defineProperty(this, "windData", void 0);
+    this.gl = options.glContext;
+    this.width = options.imageWidth;
+    this.height = options.imageHeight;
+    this.fadeOpacity = options.fadeOpacity;
+    this.speedFactor = options.speedFactor;
+    this.dropRate = options.dropRate;
+    this.dropRateBump = options.dropRateBump;
+    this.rampColors = options.rampColors;
+    this.init();
+  }
+  init() {
+    const gl = this.gl;
+    this.fadeOpacity = 0.996; // how fast the particle trails fade on each frame
+    this.speedFactor = 0.25; // how fast the particles move
+    this.dropRate = 0.003; // how often the particles move to a random place
+    this.dropRateBump = 0.01; // drop rate increase relative to individual particle speed
+
+    this.drawProgram = glUtils.createProgram(gl, drawVert, drawFrag);
+    this.fullScreenProgram = glUtils.createProgram(gl, fullScreenVert, fullScreenFrag);
+    this.updateProgram = glUtils.createProgram(gl, updateVert, updateFrag);
+    this.quadBuffer = glUtils.createBuffer(gl, new Float32Array([0, 0, 1, 0, 0, 1, 0, 1, 1, 0, 1, 1]));
+    this.framebuffer = gl.createFramebuffer();
+    this.colorRampTexture = glUtils.createTexture(this.gl, this.gl.LINEAR, getColorRamp(this.rampColors), 16, 16);
+    const emptyPixels = new Uint8Array(this.width * this.height * 4);
+
+    // screen textures to hold the drawn screen for the previous and the current frame
+
+    this.backgroundTexture = glUtils.createTexture(gl, gl.NEAREST, emptyPixels, this.width, this.height);
+    this.screenTexture = glUtils.createTexture(gl, gl.NEAREST, emptyPixels, this.width, this.height);
+
+    // we create a square texture where each pixel will hold a particle position encoded as RGBA
+    const particleRes = this.particleStateResolution = Math.ceil(Math.sqrt(this.numParticles));
+    // particleRes size
+    this.numParticlesSize = particleRes * particleRes;
+    const particleState = new Uint8Array(this.numParticlesSize * 4);
+    for (let i = 0; i < particleState.length; i++) {
+      particleState[i] = Math.floor(Math.random() * 256); // randomize the initial particle positions
+    }
+    // textures to hold the particle state for the current and the next frame
+    this.particleStateTexture0 = glUtils.createTexture(gl, gl.NEAREST, particleState, particleRes, particleRes);
+    this.particleStateTexture1 = glUtils.createTexture(gl, gl.NEAREST, particleState, particleRes, particleRes);
+    const particleIndices = new Float32Array(this.numParticlesSize);
+    for (let i$1 = 0; i$1 < this.numParticlesSize; i$1++) {
+      particleIndices[i$1] = i$1;
+    }
+    this.particleIndexBuffer = glUtils.createBuffer(gl, particleIndices);
+  }
+  setWind(windData) {
+    this.windData = windData;
+    this.windTexture = glUtils.createDataTexture(this.gl, this.gl.LINEAR, windData.image);
+  }
+
+  /**
+   * 更新风场粒子数量
+   * @param num
+   */
+  updateParticelNum(num) {
+    const gl = this.gl;
+    if (num !== this.numParticles) {
+      this.numParticles = num; // params number
+
+      // we create a square texture where each pixel will hold a particle position encoded as RGBA
+      const particleRes = this.particleStateResolution = Math.ceil(Math.sqrt(this.numParticles));
+      this.numParticlesSize = particleRes * particleRes;
+      const particleState = new Uint8Array(this.numParticlesSize * 4);
+      for (let i = 0; i < particleState.length; i++) {
+        particleState[i] = Math.floor(Math.random() * 256); // randomize the initial particle positions
+      }
+      // textures to hold the particle state for the current and the next frame
+      this.particleStateTexture0 = glUtils.createTexture(gl, gl.NEAREST, particleState, particleRes, particleRes);
+      this.particleStateTexture1 = glUtils.createTexture(gl, gl.NEAREST, particleState, particleRes, particleRes);
+      const particleIndices = new Float32Array(this.numParticlesSize);
+      for (let i$1 = 0; i$1 < this.numParticlesSize; i$1++) {
+        particleIndices[i$1] = i$1;
+      }
+      this.particleIndexBuffer = glUtils.createBuffer(gl, particleIndices);
+    }
+  }
+
+  /**
+   * 更新风场风向风速
+   * @param uMin
+   * @param uMax
+   * @param vMin
+   * @param vMax
+   */
+  updateWindDir(uMin, uMax, vMin, vMax) {
+    this.windData.uMin = uMin;
+    this.windData.uMax = uMax;
+    this.windData.vMin = vMin;
+    this.windData.vMax = vMax;
+  }
+
+  /**
+   * update rampColors
+   * @param rampColors
+   */
+  updateColorRampTexture(rampColors) {
+    if (this.isColorChanged(rampColors)) {
+      this.rampColors = rampColors;
+      const gl = this.gl;
+      gl.deleteTexture(this.colorRampTexture);
+      this.colorRampTexture = glUtils.createTexture(gl, gl.LINEAR, getColorRamp(rampColors), 16, 16);
+    }
+  }
+  isColorChanged(rampColors) {
+    const keys = Object.keys(rampColors);
+    for (const item of keys) {
+      const key = Number(item);
+      // exist new key -> color need update
+      if (!this.rampColors[key]) {
+        return true;
+      }
+      // value changed -> color need update
+      if (this.rampColors[key] && this.rampColors[key] !== rampColors[key]) {
+        return true;
+      }
+    }
+    return false;
+  }
+  reSize(width, height) {
+    if (width !== this.width || height !== this.height) {
+      const gl = this.gl;
+      gl.deleteTexture(this.backgroundTexture);
+      gl.deleteTexture(this.screenTexture);
+      this.width = width;
+      this.height = height;
+      const emptyPixels = new Uint8Array(width * height * 4);
+      // screen textures to hold the drawn screen for the previous and the current frame
+      this.backgroundTexture = glUtils.createTexture(gl, gl.NEAREST, emptyPixels, width, height);
+      this.screenTexture = glUtils.createTexture(gl, gl.NEAREST, emptyPixels, width, height);
+    }
+  }
+  draw() {
+    var _this$windData;
+    if ((_this$windData = this.windData) !== null && _this$windData !== void 0 && _this$windData.image) {
+      const gl = this.gl;
+      glUtils.bindTexture(gl, this.windTexture, 0);
+      glUtils.bindTexture(gl, this.particleStateTexture0, 1);
+      this.drawScreen(); // draw Particles into framebuffer
+      this.updateParticles();
+      return {
+        d: this.pixels,
+        w: this.width,
+        h: this.height
+      };
+    } else {
+      return {
+        d: new Uint8Array([0, 0, 0, 0]),
+        w: 1,
+        h: 1
+      };
+    }
+  }
+  drawScreen() {
+    const gl = this.gl;
+
+    // draw the screen into a temporary framebuffer to retain it as the background on the next frame
+    bindFramebuffer(gl, this.framebuffer, this.screenTexture);
+    gl.viewport(0, 0, this.width, this.height);
+    gl.disable(gl.BLEND);
+    this.drawFullTexture(this.backgroundTexture, this.fadeOpacity);
+    this.drawParticles();
+    this.pixels = new Uint8Array(4 * this.width * this.height);
+    gl.readPixels(0, 0, this.width, this.height, gl.RGBA, gl.UNSIGNED_BYTE, this.pixels);
+    bindFramebuffer(gl, null, null);
+    gl.viewport(0, 0, this.gl.canvas.width, this.gl.canvas.height);
+
+    //   save the current screen as the background for the next frame
+    const temp = this.backgroundTexture;
+    this.backgroundTexture = this.screenTexture;
+    this.screenTexture = temp;
+  }
+  drawFullTexture(texture, opacity) {
+    const gl = this.gl;
+    const program = this.fullScreenProgram;
+    gl.useProgram(program);
+
+    // bindAttribute(gl, this.quadBuffer, program.a_pos, 2);
+
+    gl.bindBuffer(gl.ARRAY_BUFFER, this.quadBuffer);
+    gl.vertexAttribPointer(program.a_pos, 2, gl.FLOAT, false, 0, 0);
+    gl.enableVertexAttribArray(program.a_pos);
+    gl.bindBuffer(gl.ARRAY_BUFFER, null);
+    glUtils.bindTexture(gl, texture, 2);
+    gl.uniform1i(program.u_screen, 2);
+    gl.uniform1f(program.u_opacity, opacity);
+    gl.drawArrays(gl.TRIANGLES, 0, 6);
+    // gl.drawArrays(gl.POINTS, 0, 6);
+  }
+  drawParticles() {
+    const gl = this.gl;
+    const program = this.drawProgram;
+    gl.useProgram(program);
+    bindAttribute(gl, this.particleIndexBuffer, program.a_index, 1);
+    glUtils.bindTexture(gl, this.colorRampTexture, 2);
+    gl.uniform1i(program.u_wind, 0);
+    gl.uniform1i(program.u_particles, 1);
+    gl.uniform1i(program.u_color_ramp, 2);
+    gl.uniform1f(program.u_particles_res, this.particleStateResolution);
+    gl.uniform2f(program.u_wind_min, this.windData.uMin, this.windData.vMin);
+    gl.uniform2f(program.u_wind_max, this.windData.uMax, this.windData.vMax);
+    gl.drawArrays(gl.POINTS, 0, this.numParticlesSize);
+  }
+  updateParticles() {
+    const gl = this.gl;
+    bindFramebuffer(gl, this.framebuffer, this.particleStateTexture1);
+    gl.viewport(0, 0, this.particleStateResolution, this.particleStateResolution);
+    const program = this.updateProgram;
+    gl.useProgram(program);
+    bindAttribute(gl, this.quadBuffer, program.a_pos, 2);
+    gl.uniform1i(program.u_wind, 0);
+    gl.uniform1i(program.u_particles, 1);
+    gl.uniform1f(program.u_rand_seed, Math.random());
+    gl.uniform2f(program.u_wind_res, this.windData.image.width * 2, this.windData.image.height * 2);
+    gl.uniform2f(program.u_wind_min, this.windData.uMin, this.windData.vMin);
+    gl.uniform2f(program.u_wind_max, this.windData.uMax, this.windData.vMax);
+    gl.uniform1f(program.u_speed_factor, this.speedFactor);
+    gl.uniform1f(program.u_drop_rate, this.dropRate);
+    gl.uniform1f(program.u_drop_rate_bump, this.dropRateBump);
+    gl.drawArrays(gl.TRIANGLES, 0, 6);
+
+    // swap the particle state textures so the new one becomes the current one
+    const temp = this.particleStateTexture0;
+    this.particleStateTexture0 = this.particleStateTexture1;
+    this.particleStateTexture1 = temp;
+    bindFramebuffer(gl, null, null);
+
+    // gl.viewport(0, 0, this.gl.canvas.width, this.gl.canvas.height);
+  }
+  destroy() {
+    // private drawProgram: WebGLProgram;
+    // private fullScreenProgram: WebGLProgram;
+    // private updateProgram: WebGLProgram;
+
+    // private quadBuffer: WebGLBuffer | null;
+    // private particleIndexBuffer: WebGLBuffer | null;
+
+    // private framebuffer: WebGLFramebuffer | null;
+
+    // private colorRampTexture: WebGLTexture | null;
+    // private backgroundTexture: WebGLTexture | null;
+    // private screenTexture: WebGLTexture | null;
+    // private particleStateTexture0: WebGLTexture | null;
+    // private particleStateTexture1: WebGLTexture | null;
+    // private windTexture: WebGLTexture | null;
+
+    this.gl.deleteBuffer(this.quadBuffer);
+    this.gl.deleteBuffer(this.particleIndexBuffer);
+    this.gl.deleteFramebuffer(this.framebuffer);
+
+    // @ts-ignore
+    this.gl.deleteShader(this.drawProgram.vertexShader);
+    // @ts-ignore
+    this.gl.deleteShader(this.drawProgram.fragmentShader);
+    this.gl.deleteProgram(this.drawProgram);
+
+    // @ts-ignore
+    this.gl.deleteShader(this.fullScreenProgram.vertexShader);
+    // @ts-ignore
+    this.gl.deleteShader(this.fullScreenProgram.fragmentShader);
+    this.gl.deleteProgram(this.fullScreenProgram);
+
+    // @ts-ignore
+    this.gl.deleteShader(this.updateProgram.vertexShader);
+    // @ts-ignore
+    this.gl.deleteShader(this.updateProgram.fragmentShader);
+    this.gl.deleteProgram(this.updateProgram);
+    this.gl.deleteTexture(this.colorRampTexture);
+    this.gl.deleteTexture(this.backgroundTexture);
+    this.gl.deleteTexture(this.screenTexture);
+    this.gl.deleteTexture(this.particleStateTexture0);
+    this.gl.deleteTexture(this.particleStateTexture1);
+    this.gl.deleteTexture(this.windTexture);
+  }
+}

package/es/wind/models/windShader.d.ts

@@ -0,0 +1,12 @@
+/**
+ * drawProgram          drawVert drawFrag
+ * screenProgram        screenVert screenFrag
+ * updateProgram        updateVert updateFrag
+ * fullScreenProgram    fullScreenVert fullScreenFrag
+ */
+export declare const drawVert = "\n precision mediump float;\n\n attribute float a_index;\n\n uniform sampler2D u_particles;\n uniform float u_particles_res;\n\n varying vec2 v_particle_pos;\n\n void main() {\n     vec4 color = texture2D(u_particles, vec2(\n         fract(a_index / u_particles_res),\n         floor(a_index / u_particles_res) / u_particles_res)\n     );\n\n     // decode current particle position from the pixel's RGBA value\n     v_particle_pos = vec2( color.r / 255.0 + color.b, color.g / 255.0 + color.a);\n\n     gl_PointSize = 1.0;\n     gl_Position = vec4(2.0 * v_particle_pos.x - 1.0, 1.0 - 2.0 * v_particle_pos.y, 0, 1);\n }";
+export declare const drawFrag = "\n precision mediump float;\n\n uniform sampler2D u_wind;\n uniform vec2 u_wind_min;\n uniform vec2 u_wind_max;\n uniform sampler2D u_color_ramp;\n\n varying vec2 v_particle_pos;\n\n void main() {\n     vec2 velocity = mix(u_wind_min, u_wind_max, texture2D(u_wind, v_particle_pos).rg);\n     float speed_t = length(velocity) / length(u_wind_max);\n\n     // color ramp is encoded in a 16x16 texture\n     vec2 ramp_pos = vec2( fract(16.0 * speed_t), floor(16.0 * speed_t) / 16.0);\n\n     gl_FragColor = texture2D(u_color_ramp, ramp_pos);\n }";
+export declare const updateVert = "\n precision mediump float;\n\n attribute vec2 a_pos;\n\n varying vec2 v_tex_pos;\n\n void main() {\n     v_tex_pos = a_pos;\n     gl_Position = vec4(1.0 - 2.0 * a_pos, 0, 1);\n     // framebuffer \u59CB\u7EC8\u7528\u94FA\u6EE1\u5C4F\u5E55\u7684 texture\n }";
+export declare const updateFrag = "\n precision highp float;\n\n uniform sampler2D u_particles;\n uniform sampler2D u_wind;\n uniform vec2 u_wind_res;\n uniform vec2 u_wind_min;\n uniform vec2 u_wind_max;\n uniform float u_rand_seed;\n uniform float u_speed_factor;\n uniform float u_drop_rate;\n uniform float u_drop_rate_bump;\n\n varying vec2 v_tex_pos;\n\n // pseudo-random generator\n const vec3 rand_constants = vec3(12.9898, 78.233, 4375.85453);\n float rand(const vec2 co) {\n float t = dot(rand_constants.xy, co);\n     return fract(sin(t) * (rand_constants.z + t));\n }\n\n // wind speed lookup; use manual bilinear filtering based on 4 adjacent pixels for smooth interpolation\n vec2 lookup_wind(const vec2 uv) {\n     // return texture2D(u_wind, uv).rg; // lower-res hardware filtering\n     vec2 px = 1.0 / u_wind_res;\n     vec2 vc = (floor(uv * u_wind_res)) * px;\n     vec2 f = fract(uv * u_wind_res);\n     vec2 tl = texture2D(u_wind, vc).rg;\n     vec2 tr = texture2D(u_wind, vc + vec2(px.x, 0)).rg;\n     vec2 bl = texture2D(u_wind, vc + vec2(0, px.y)).rg;\n     vec2 br = texture2D(u_wind, vc + px).rg;\n     return mix(mix(tl, tr, f.x), mix(bl, br, f.x), f.y);\n }\n\n void main() {\n     vec4 color = texture2D(u_particles, v_tex_pos);\n     vec2 pos = vec2(\n         color.r / 255.0 + color.b,\n         color.g / 255.0 + color.a); // decode particle position from pixel RGBA\n     vec2 velocity = mix(u_wind_min, u_wind_max, lookup_wind(pos));\n     float speed_t = length(velocity) / length(u_wind_max);\n\n     // take EPSG:4236 distortion into account for calculating where the particle moved\n     float distortion = cos(radians(pos.y * 180.0 - 90.0));\n     vec2 offset = vec2(velocity.x / distortion, -velocity.y) * 0.0001 * u_speed_factor;\n\n     // update particle position, wrapping around the date line\n     pos = fract(1.0 + pos + offset);\n\n     // a random seed to use for the particle drop\n     vec2 seed = (pos + v_tex_pos) * u_rand_seed;\n\n     // drop rate is a chance a particle will restart at random position, to avoid degeneration\n     float drop_rate = u_drop_rate + speed_t * u_drop_rate_bump;\n     float drop = step(1.0 - drop_rate, rand(seed));\n\n     vec2 random_pos = vec2(\n         rand(seed + 1.3),\n         rand(seed + 2.1));\n         pos = mix(pos, random_pos, drop);\n\n     // encode the new particle position back into RGBA\n     gl_FragColor = vec4(\n         fract(pos * 255.0),\n         floor(pos * 255.0) / 255.0);\n }";
+export declare const fullScreenVert = "\n     precision mediump float;\n\n     attribute vec2 a_pos;\n\n     varying vec2 v_tex_pos;\n\n     void main() {\n         v_tex_pos = a_pos;\n         gl_Position = vec4(1.0 - 2.0 * a_pos, 0.0, 1.0);\n         gl_PointSize = 100.0;\n }";
+export declare const fullScreenFrag = "\n precision mediump float;\n\n uniform sampler2D u_screen;\n uniform float u_opacity;\n varying vec2 v_tex_pos;\n\n void main() {\n     vec4 color = texture2D(u_screen, 1.0 - v_tex_pos);\n\n     // a hack to guarantee opacity fade out even with a value close to 1.0\n     gl_FragColor = vec4(floor(255.0 * color * u_opacity) / 255.0);\n }";