@pirireis/webglobeplugins 0.0.1

package/waveparticles/plugin.js

@@ -0,0 +1,281 @@
+import { GlobeShellWiggle } from "../programs";
+import { defaultblendfunction } from "../util";
+import { vectorfield } from "../programs";
+import { FadeAway } from "../programs";
+const { PixelBasedMove, DrawRectangleParticles, PingPongBufferManager, WaveParticalUboManager } = vectorfield;
+const MoveParticle = PixelBasedMove;
+const DrawParticle = DrawRectangleParticles;
+
+/**
+ * STEPS:
+ * 1. move particle | buffers: read b1 write b2 | swap buffers b1 <-> b2
+ * 2. draw particles | read b1 to drawTexture1 |
+ * 3. globe shell drawTexture1
+ * 4. fade drawTexture1 to drawTexture2 | swap textures
+ * 
+ */
+
+const MAX_PIXELS_ON_DIMENSION = 2200;
+
+
+
+export default class Plugin {
+    constructor(id, { fadeOpacity = 0.83, opacity = 0.75, minLon = -180, minLat = -90, maxLon = 180, maxLat = 90, patricleCount = 8000, dataWidth, dataHeight, drawTextureMaxPixelOnDimension = MAX_PIXELS_ON_DIMENSION } = {}) {
+
+        this.id = id;
+        this.globe = null;
+        this.gl = null;
+        this.moveParticle = null;
+        this.drawParticle = null;
+        this.bufferManager = null;
+        this._rgVectorFieldTexture = null;
+        this.globeShellWiggle = null;
+        this.waveUbo = null;
+        this._drawTextureResolution = {};
+        this._frameBuffer = null;
+        this._fadeOpacity = fadeOpacity;
+        this._opacity = opacity;
+        this._particleCount = patricleCount;
+        this._dataWidth = dataWidth;
+        this._dataHeight = dataHeight;
+        this._drawTextureMaxPixelOnDimension = drawTextureMaxPixelOnDimension;
+        this._globeshellparameters = {
+            minLon,
+            maxLon,
+            minLat,
+            maxLat,
+        }
+        this._stepIndex = 0;
+        this._fullCycleStepCount = 3;
+    }
+
+    init(globe, gl) {
+        this.globe = globe;
+        this.gl = gl;
+        this.moveParticle = new MoveParticle(gl);
+        this.drawParticle = new DrawParticle(gl);
+        // this.moveParticle = glProgramCache.getProgram(gl, MoveParticle); // TODO: create with glProgramCache and free later on.
+        // this.drawParticle = glProgramCache.getProgram(gl, DrawParticle);
+
+
+        this.fadeAway = new FadeAway(gl);
+
+        this.globeShellWiggle = new GlobeShellWiggle(gl, globe, this._globeshellparameters);
+        const inPositionLocation = this.moveParticle.getInPositionLocation();
+        this.bufferManager = new PingPongBufferManager(gl, this._particleCount, inPositionLocation);
+
+        this._rgVectorFieldTexture = this._createRGTexture();
+
+        this.waveUbo = new WaveParticalUboManager(gl, 0);
+
+        this._drawTextures = [this._createDrawTexture(), this._createDrawTexture()];
+        this._frameBuffer = gl.createFramebuffer();
+        this.setBBox(this._globeshellparameters);
+        this.setOpacity(this._opacity);
+        this.____drawIndex = 0;
+
+    }
+
+    _createDrawTexture() {
+        const gl = this.gl
+        const texture = gl.createTexture();
+        gl.bindTexture(gl.TEXTURE_2D, texture);
+        // gl.texImage2D(gl.TEXTURE_2D, 0, gl.RGBA, 2200, 2200, 0, gl.RGBA, gl.UNSIGNED_BYTE, null);
+        gl.texImage2D(gl.TEXTURE_2D, 0, gl.RGBA, this._drawTextureResolution.width, this._drawTextureResolution.height, 0, gl.RGBA, gl.UNSIGNED_BYTE, null);
+        gl.texParameteri(gl.TEXTURE_2D, gl.TEXTURE_MIN_FILTER, gl.LINEAR);
+        gl.texParameteri(gl.TEXTURE_2D, gl.TEXTURE_MAG_FILTER, gl.LINEAR);
+        gl.texParameteri(gl.TEXTURE_2D, gl.TEXTURE_WRAP_S, gl.CLAMP_TO_EDGE);
+        gl.texParameteri(gl.TEXTURE_2D, gl.TEXTURE_WRAP_T, gl.CLAMP_TO_EDGE);
+        gl.bindTexture(gl.TEXTURE_2D, null);
+        return texture;
+    }
+
+
+    _step() {
+        this._stepIndex = (this._stepIndex + 1) % this._fullCycleStepCount;
+    }
+
+    draw3D() {
+        const { gl, moveParticle, drawParticle, bufferManager, globeShellWiggle, waveUbo, fadeAway, _rgVectorFieldTexture } = this;
+
+        gl.disable(gl.DEPTH_TEST);
+        if (this._stepIndex === 0) {
+            waveUbo.update({ random_seed: Math.random() });
+            moveParticle.move(bufferManager, _rgVectorFieldTexture, waveUbo);
+            // 1s swap buffer b1 <-> b2
+            bufferManager.swap();
+        } else if (this._stepIndex === 1) {
+            this.____drawIndex = 1 - this.____drawIndex;
+            const canvasWidth = gl.canvas.width;
+            const canvasHeight = gl.canvas.height;
+            // gl.viewport(0, 0, 2200, 2200);
+            gl.viewport(0, 0, this._drawTextureResolution.width, this._drawTextureResolution.height);
+            gl.bindFramebuffer(gl.FRAMEBUFFER, this._frameBuffer);
+            gl.framebufferTexture2D(gl.FRAMEBUFFER, gl.COLOR_ATTACHMENT0, gl.TEXTURE_2D, this._drawTextures[this.____drawIndex], 0);
+            drawParticle.draw(bufferManager, _rgVectorFieldTexture, waveUbo);
+            // 2s bound frame buffer to null
+            gl.bindFramebuffer(gl.FRAMEBUFFER, null);
+            gl.viewport(0, 0, canvasWidth, canvasHeight);
+        } else {
+            const canvasWidth = gl.canvas.width;
+            const canvasHeight = gl.canvas.height;
+            // gl.viewport(0, 0, 2200, 2200);
+            gl.viewport(0, 0, this._drawTextureResolution.width, this._drawTextureResolution.height);
+            gl.bindFramebuffer(gl.FRAMEBUFFER, this._frameBuffer);
+            gl.enable(gl.BLEND);
+            // gl.blendFunc(gl.SRC_ALPHA, gl.ONE_MINUS_SRC_ALPHA);
+            gl.blendFunc(gl.ONE, gl.ZERO)
+            gl.framebufferTexture2D(gl.FRAMEBUFFER, gl.COLOR_ATTACHMENT0, gl.TEXTURE_2D, this._drawTextures[1 - this.____drawIndex], 0);
+            fadeAway.draw(this._drawTextures[this.____drawIndex], this._fadeOpacity);
+            gl.bindFramebuffer(gl.FRAMEBUFFER, null);
+            defaultblendfunction(gl);
+            gl.viewport(0, 0, canvasWidth, canvasHeight);
+        }
+
+        globeShellWiggle.setTexture(this._drawTextures[this.____drawIndex]);
+        globeShellWiggle.draw();
+        this._step();
+        gl.enable(gl.DEPTH_TEST);
+    }
+
+
+    setCycleStepCount(value) {
+        if (value < 3) {
+            this._fullCycleStepCount = 3;
+        }
+        this._fullCycleStepCount = value;
+    }
+
+    _createRGTexture() {
+        const { gl, _dataWidth, _dataHeight } = this;
+        const texture = gl.createTexture();
+        // R32F
+        gl.bindTexture(gl.TEXTURE_2D, texture);
+        gl.texImage2D(gl.TEXTURE_2D, 0, gl.RG32F, _dataWidth, _dataHeight, 0, gl.RG, gl.FLOAT, null);
+        gl.texParameteri(gl.TEXTURE_2D, gl.TEXTURE_MIN_FILTER, gl.NEAREST);
+        gl.texParameteri(gl.TEXTURE_2D, gl.TEXTURE_MAG_FILTER, gl.NEAREST);
+        gl.texParameteri(gl.TEXTURE_2D, gl.TEXTURE_WRAP_S, gl.CLAMP_TO_EDGE);
+        gl.texParameteri(gl.TEXTURE_2D, gl.TEXTURE_WRAP_T, gl.CLAMP_TO_EDGE);
+        gl.pixelStorei(gl.UNPACK_FLIP_Y_WEBGL, false);
+
+        gl.bindTexture(gl.TEXTURE_2D, null);
+        return texture;
+    }
+
+    // TODO: free all resources
+    free() {
+        const { gl, fadeAway, globeShellWiggle, _rgVectorFieldTexture, bufferManager, waveUbo, _drawTextures, _frameBuffer } = this;
+        fadeAway.free();
+        globeShellWiggle.free();
+        gl.deleteTexture(_rgVectorFieldTexture);
+        // glProgramCache.releaseProgram(gl, MoveParticle);
+        this.moveParticle.free();
+        this.moveParticle = null
+        // glProgramCache.releaseProgram(gl, DrawParticle);
+        this.drawParticle.free();
+        this.drawParticle = null
+        bufferManager.free();
+        waveUbo.free();
+        _drawTextures.forEach(texture => gl.deleteTexture(texture));
+        gl.deleteFramebuffer(_frameBuffer);
+    }
+
+
+    // setters 
+    setParticleCount(value) {
+        this._particleCount = value;
+        this.bufferManager.setParticleCount(value);
+    }
+
+
+    setVectorFieldData(data, dataWidth = null, dataHeight = null) {
+        if (dataWidth !== null && dataHeight !== null) {
+            this._dataWidth = dataWidth;
+            this._dataHeight = dataHeight;
+        }
+        const { gl, _dataWidth, _dataHeight } = this;
+        gl.bindTexture(gl.TEXTURE_2D, this._rgVectorFieldTexture);
+        gl.texImage2D(gl.TEXTURE_2D, 0, gl.RG32F, _dataWidth, _dataHeight, 0, gl.RG, gl.FLOAT, data);
+        gl.bindTexture(gl.TEXTURE_2D, null);
+    }
+
+
+    setHeight(value) {
+        this.globeShellWiggle.setHeight(value);
+    }
+
+    setDropRate(value) {
+        this.waveUbo.update({ drop_rate: value });
+    }
+
+
+
+    _drawTextureSizeFromBbox({ minLon, minLat, maxLon, maxLat }) {
+        let horizon;
+        if (minLon > 0 && maxLon < 0) {
+            horizon = 360 - minLon + maxLon;
+        } else {
+            horizon = maxLon - minLon;
+        }
+        const vertical = maxLat - minLat;
+        let width, height;
+        if (horizon > vertical) {
+            width = this._drawTextureMaxPixelOnDimension;
+            height = Math.floor(this._drawTextureMaxPixelOnDimension * vertical / horizon);
+        } else {
+            height = this._drawTextureMaxPixelOnDimension;
+            width = Math.floor(this._drawTextureMaxPixelOnDimension * horizon / vertical);
+        }
+        return {
+            width,
+            height
+        }
+
+    }
+
+
+    setDrawTextureMaxPixelOnDimension(value) {
+        this._drawTextureMaxPixelOnDimension = value;
+        this._drawTextureResolution = this._drawTextureSizeFromBbox(this._globeshellparameters);
+        this.waveUbo.update({ draw_texture_size: [this._drawTextureResolution.width, this._drawTextureResolution.height] });
+        this._drawTextures = [this._createDrawTexture(), this._createDrawTexture()];
+    }
+
+    setParticleSpeed(value) {
+        this.waveUbo.update({ range: value });
+    }
+
+    setFadeOpacity(value) {
+        this._fadeOpacity = value;
+    }
+
+    setOpacity(value) {
+        this.globeShellWiggle.setOpacity(value);
+    }
+
+    setParticleDimensions(tail, wing) {
+        if (0 < tail || 0 < wing) {
+            this.waveUbo.update({ base_limp: [tail, wing] });
+        } else {
+            console.error("tail and wing must be greater than 0");
+        }
+    }
+
+    setParticleColor(color) {
+        if (color.length !== 3) {
+            console.error("color must be an array of rgb elements");
+            return;
+        }
+        this.waveUbo.update({ color });
+    }
+
+
+    setBBox({ minLon, minLat, maxLon, maxLat }) {
+        this._globeshellparameters = { minLon, minLat, maxLon, maxLat };
+        this.globeShellWiggle.setBBox({ minLon, minLat, maxLon, maxLat });
+        this._drawTextureResolution = this._drawTextureSizeFromBbox({ minLon, minLat, maxLon, maxLat });
+        this.waveUbo.update({ draw_texture_size: [this._drawTextureResolution.width, this._drawTextureResolution.height] });
+        this._drawTextures = [this._createDrawTexture(), this._createDrawTexture()];
+    }
+}
+

package/wind/imagetovectorfieldandmagnitude.js

@@ -0,0 +1,39 @@
+/**
+ * @param {ImageData} imageData
+ * image
+ * uMax 
+ * vMax
+ * height
+ * width
+ * @returns {Float32Array}
+*/
+export default function imageToMagnitude(imageData) {
+    const { image, uMax, vMax, uMin, vMin, height, width } = imageData;
+
+
+
+    const canvas = document.createElement('canvas');
+    const ctx = canvas.getContext('2d');
+    canvas.width = width;
+    canvas.height = height;
+    ctx.drawImage(image, 0, 0, width, height);
+    const img = ctx.getImageData(0, 0, width, height);
+    const data = img.data;
+
+    const magnitudeArray = new Float32Array(height * width);
+
+    const uDiff = uMax - uMin;
+    const vDiff = vMax - vMin;
+
+    for (let i = 0; i < data.length; i += 4) {
+        const r = data[i];
+        const g = data[i + 1];
+        const u = uMin + (uDiff * r) / 255;
+        const v = vMin + (vDiff * g) / 255;
+        const magnitude = Math.sqrt(u * u + v * v);
+        const index = i / 4;
+        magnitudeArray[index] = magnitude;
+    }
+
+    return magnitudeArray;
+}

package/wind/index.js

@@ -0,0 +1,7 @@
+import WindPlugin from "./plugin";
+import createVectorFieldImage from "./vectorfieldimage";
+import imageToMagnitude from "./imagetovectorfieldandmagnitude";
+
+export { createVectorFieldImage, imageToMagnitude, WindPlugin };
+
+// TODO: image resmindeki degerlerden ruzgar siddetinin magnitudunun hesaplanip float32array donderen bir fonksiyonun yazilmasi gerekiyor