@deck.gl/aggregation-layers 9.0.0-beta.1 → 9.0.0-beta.3

package/dist/esm/utils/gpu-grid-aggregation/gpu-grid-aggregator.js

@@ -1,720 +0,0 @@
-import { Model, TextureTransform } from '@luma.gl/engine';
-import { fp64arithmetic } from '@luma.gl/shadertools';
-import { readPixelsToBuffer, withGLParameters, clear } from '@luma.gl/webgl';
-import { GL } from '@luma.gl/constants';
-import { log, project32, _mergeShaders as mergeShaders, getShaderAssembler } from '@deck.gl/core';
-import { DEFAULT_RUN_PARAMS, MAX_32_BIT_FLOAT, MIN_BLEND_EQUATION, MAX_BLEND_EQUATION, MAX_MIN_BLEND_EQUATION, EQUATION_MAP, DEFAULT_WEIGHT_PARAMS, PIXEL_SIZE } from './gpu-grid-aggregator-constants';
-import { AGGREGATION_OPERATION } from '../aggregation-operation-utils';
-import AGGREGATE_TO_GRID_VS from './aggregate-to-grid-vs.glsl';
-import AGGREGATE_TO_GRID_FS from './aggregate-to-grid-fs.glsl';
-import AGGREGATE_ALL_VS from './aggregate-all-vs.glsl';
-import AGGREGATE_ALL_FS from './aggregate-all-fs.glsl';
-import TRANSFORM_MEAN_VS from './transform-mean-vs.glsl';
-import { getFloatTexture, getFramebuffer } from './../resource-utils';
-const BUFFER_NAMES = ['aggregationBuffer', 'maxMinBuffer', 'minBuffer', 'maxBuffer'];
-const ARRAY_BUFFER_MAP = {
-  maxData: 'maxBuffer',
-  minData: 'minBuffer',
-  maxMinData: 'maxMinBuffer'
-};
-const REQUIRED_FEATURES = ['webgl2', 'blend-minmax-webgl1', 'texture-renderable-float32-webgl', 'texture-blend-float-webgl1', 'texture-formats-float32-webgl1'];
-export default class GPUGridAggregator {
-  static getAggregationData({
-    aggregationData,
-    maxData,
-    minData,
-    maxMinData,
-    pixelIndex
-  }) {
-    const index = pixelIndex * PIXEL_SIZE;
-    const results = {};
-
-    if (aggregationData) {
-      results.cellCount = aggregationData[index + 3];
-      results.cellWeight = aggregationData[index];
-    }
-
-    if (maxMinData) {
-      results.maxCellWieght = maxMinData[0];
-      results.minCellWeight = maxMinData[3];
-    } else {
-      if (maxData) {
-        results.maxCellWieght = maxData[0];
-        results.totalCount = maxData[3];
-      }
-
-      if (minData) {
-        results.minCellWeight = minData[0];
-        results.totalCount = minData[3];
-      }
-    }
-
-    return results;
-  }
-
-  static getCellData({
-    countsData,
-    size = 1
-  }) {
-    const numCells = countsData.length / 4;
-    const cellWeights = new Float32Array(numCells * size);
-    const cellCounts = new Uint32Array(numCells);
-
-    for (let i = 0; i < numCells; i++) {
-      for (let sizeIndex = 0; sizeIndex < size; sizeIndex++) {
-        cellWeights[i * size + sizeIndex] = countsData[i * 4 + sizeIndex];
-      }
-
-      cellCounts[i] = countsData[i * 4 + 3];
-    }
-
-    return {
-      cellCounts,
-      cellWeights
-    };
-  }
-
-  static isSupported(device) {
-    return REQUIRED_FEATURES.every(feature => device.features.has(feature));
-  }
-
-  constructor(device, props = {}) {
-    this.state = {
-      weightAttributes: {},
-      textures: {},
-      meanTextures: {},
-      buffers: {},
-      framebuffers: {},
-      maxMinFramebuffers: {},
-      minFramebuffers: {},
-      maxFramebuffers: {},
-      equations: {},
-      shaderOptions: {},
-      modelDirty: false,
-      resources: {},
-      results: {}
-    };
-    this.id = void 0;
-    this.device = void 0;
-    this._hasGPUSupport = void 0;
-    this.gridAggregationModel = void 0;
-    this.allAggregationModel = void 0;
-    this.meanTransform = void 0;
-    this.id = props.id || 'gpu-grid-aggregator';
-    this.device = device;
-    const REQUIRED_FEATURES = ['blend-minmax-webgl1', 'texture-renderable-float32-webgl', 'texture-formats-float32-webgl1'];
-    this._hasGPUSupport = this.device.info.type === 'webgl2' && REQUIRED_FEATURES.every(feature => device.features.has(feature));
-
-    if (this._hasGPUSupport) {
-      this._setupModels();
-    }
-  }
-
-  delete() {
-    const {
-      gridAggregationModel,
-      allAggregationModel,
-      meanTransform
-    } = this;
-    const {
-      textures,
-      framebuffers,
-      maxMinFramebuffers,
-      minFramebuffers,
-      maxFramebuffers,
-      meanTextures,
-      resources
-    } = this.state;
-    gridAggregationModel === null || gridAggregationModel === void 0 ? void 0 : gridAggregationModel.destroy();
-    allAggregationModel === null || allAggregationModel === void 0 ? void 0 : allAggregationModel.destroy();
-    meanTransform === null || meanTransform === void 0 ? void 0 : meanTransform.destroy();
-    deleteResources([framebuffers, textures, maxMinFramebuffers, minFramebuffers, maxFramebuffers, meanTextures, resources]);
-  }
-
-  run(opts = {}) {
-    this.setState({
-      results: {}
-    });
-
-    const aggregationParams = this._normalizeAggregationParams(opts);
-
-    if (!this._hasGPUSupport) {
-      log.log(1, 'GPUGridAggregator: not supported')();
-    }
-
-    return this._runAggregation(aggregationParams);
-  }
-
-  getData(weightId) {
-    const data = {};
-    const results = this.state.results;
-
-    if (!results[weightId].aggregationData) {
-      results[weightId].aggregationData = results[weightId].aggregationBuffer.getData();
-    }
-
-    data.aggregationData = results[weightId].aggregationData;
-
-    for (const arrayName in ARRAY_BUFFER_MAP) {
-      const bufferName = ARRAY_BUFFER_MAP[arrayName];
-
-      if (results[weightId][arrayName] || results[weightId][bufferName]) {
-        results[weightId][arrayName] = results[weightId][arrayName] || results[weightId][bufferName].getData();
-        data[arrayName] = results[weightId][arrayName];
-      }
-    }
-
-    return data;
-  }
-
-  updateShaders(shaderOptions = {}) {
-    this.setState({
-      shaderOptions,
-      modelDirty: true
-    });
-  }
-
-  _normalizeAggregationParams(opts) {
-    const aggregationParams = { ...DEFAULT_RUN_PARAMS,
-      ...opts
-    };
-    const {
-      weights
-    } = aggregationParams;
-
-    if (weights) {
-      aggregationParams.weights = normalizeWeightParams(weights);
-    }
-
-    return aggregationParams;
-  }
-
-  setState(updateObject) {
-    Object.assign(this.state, updateObject);
-  }
-
-  _getAggregateData(opts) {
-    const results = {};
-    const {
-      textures,
-      framebuffers,
-      maxMinFramebuffers,
-      minFramebuffers,
-      maxFramebuffers,
-      resources
-    } = this.state;
-    const {
-      weights
-    } = opts;
-
-    for (const id in weights) {
-      results[id] = {};
-      const {
-        needMin,
-        needMax,
-        combineMaxMin
-      } = weights[id];
-      results[id].aggregationTexture = textures[id];
-      results[id].aggregationBuffer = readPixelsToBuffer(framebuffers[id], {
-        target: weights[id].aggregationBuffer,
-        sourceType: 5126
-      });
-
-      if (needMin && needMax && combineMaxMin) {
-        results[id].maxMinBuffer = readPixelsToBuffer(maxMinFramebuffers[id], {
-          target: weights[id].maxMinBuffer,
-          sourceType: 5126
-        });
-        results[id].maxMinTexture = resources[`${id}-maxMinTexture`];
-      } else {
-        if (needMin) {
-          results[id].minBuffer = readPixelsToBuffer(minFramebuffers[id], {
-            target: weights[id].minBuffer,
-            sourceType: 5126
-          });
-          results[id].minTexture = resources[`${id}-minTexture`];
-        }
-
-        if (needMax) {
-          results[id].maxBuffer = readPixelsToBuffer(maxFramebuffers[id], {
-            target: weights[id].maxBuffer,
-            sourceType: 5126
-          });
-          results[id].maxTexture = resources[`${id}-maxTexture`];
-        }
-      }
-    }
-
-    this._trackGPUResultBuffers(results, weights);
-
-    return results;
-  }
-
-  _renderAggregateData(opts) {
-    const {
-      cellSize,
-      projectPoints,
-      attributes,
-      moduleSettings,
-      numCol,
-      numRow,
-      weights,
-      translation,
-      scaling
-    } = opts;
-    const {
-      maxMinFramebuffers,
-      minFramebuffers,
-      maxFramebuffers
-    } = this.state;
-    const gridSize = [numCol, numRow];
-    const parameters = {
-      blend: true,
-      depthTest: false,
-      blendFunc: [1, 1]
-    };
-    const uniforms = {
-      cellSize,
-      gridSize,
-      projectPoints,
-      translation,
-      scaling
-    };
-
-    for (const id in weights) {
-      const {
-        needMin,
-        needMax
-      } = weights[id];
-      const combineMaxMin = needMin && needMax && weights[id].combineMaxMin;
-
-      this._renderToWeightsTexture({
-        id,
-        parameters,
-        moduleSettings,
-        uniforms,
-        gridSize,
-        attributes,
-        weights
-      });
-
-      if (combineMaxMin) {
-        this._renderToMaxMinTexture({
-          id,
-          parameters: { ...parameters,
-            blendEquation: MAX_MIN_BLEND_EQUATION
-          },
-          gridSize,
-          minOrMaxFb: maxMinFramebuffers[id],
-          clearParams: {
-            clearColor: [0, 0, 0, MAX_32_BIT_FLOAT]
-          },
-          combineMaxMin
-        });
-      } else {
-        if (needMin) {
-          this._renderToMaxMinTexture({
-            id,
-            parameters: { ...parameters,
-              blendEquation: MIN_BLEND_EQUATION
-            },
-            gridSize,
-            minOrMaxFb: minFramebuffers[id],
-            clearParams: {
-              clearColor: [MAX_32_BIT_FLOAT, MAX_32_BIT_FLOAT, MAX_32_BIT_FLOAT, 0]
-            },
-            combineMaxMin
-          });
-        }
-
-        if (needMax) {
-          this._renderToMaxMinTexture({
-            id,
-            parameters: { ...parameters,
-              blendEquation: MAX_BLEND_EQUATION
-            },
-            gridSize,
-            minOrMaxFb: maxFramebuffers[id],
-            clearParams: {
-              clearColor: [0, 0, 0, 0]
-            },
-            combineMaxMin
-          });
-        }
-      }
-    }
-  }
-
-  _renderToMaxMinTexture(opts) {
-    const {
-      id,
-      parameters,
-      gridSize,
-      minOrMaxFb,
-      combineMaxMin,
-      clearParams = {}
-    } = opts;
-    const {
-      framebuffers
-    } = this.state;
-    const {
-      allAggregationModel
-    } = this;
-    withGLParameters(this.device, { ...clearParams,
-      framebuffer: minOrMaxFb,
-      viewport: [0, 0, gridSize[0], gridSize[1]]
-    }, () => {
-      clear(this.device, {
-        color: true
-      });
-      allAggregationModel.setUniforms({
-        gridSize,
-        combineMaxMin
-      });
-      allAggregationModel.setBindings({
-        uSampler: framebuffers[id].texture
-      });
-      allAggregationModel.draw();
-    });
-  }
-
-  _renderToWeightsTexture(opts) {
-    const {
-      id,
-      parameters,
-      moduleSettings,
-      uniforms,
-      gridSize,
-      weights
-    } = opts;
-    const {
-      framebuffers,
-      equations,
-      weightAttributes
-    } = this.state;
-    const {
-      gridAggregationModel
-    } = this;
-    const {
-      operation
-    } = weights[id];
-    const clearColor = operation === AGGREGATION_OPERATION.MIN ? [MAX_32_BIT_FLOAT, MAX_32_BIT_FLOAT, MAX_32_BIT_FLOAT, 0] : [0, 0, 0, 0];
-    withGLParameters(this.device, {
-      framebuffer: framebuffers[id],
-      viewport: [0, 0, gridSize[0], gridSize[1]],
-      clearColor
-    }, () => {
-      clear(this.device, {
-        color: true
-      });
-      const attributes = {
-        weights: weightAttributes[id]
-      };
-      gridAggregationModel.draw({
-        parameters: { ...parameters,
-          blendEquation: equations[id]
-        },
-        moduleSettings,
-        uniforms,
-        attributes
-      });
-    });
-
-    if (operation === AGGREGATION_OPERATION.MEAN) {
-      const {
-        meanTextures,
-        textures
-      } = this.state;
-      const transformOptions = {
-        _sourceTextures: {
-          aggregationValues: meanTextures[id]
-        },
-        _targetTexture: textures[id],
-        elementCount: textures[id].width * textures[id].height
-      };
-
-      if (this.meanTransform) {
-        this.meanTransform.update(transformOptions);
-      } else {
-        this.meanTransform = getMeanTransform(this.device, transformOptions);
-      }
-
-      this.meanTransform.run({
-        parameters: {
-          blend: false,
-          depthTest: false
-        }
-      });
-      framebuffers[id].attach({
-        [36064]: textures[id]
-      });
-    }
-  }
-
-  _runAggregation(opts) {
-    this._updateModels(opts);
-
-    this._setupFramebuffers(opts);
-
-    this._renderAggregateData(opts);
-
-    const results = this._getAggregateData(opts);
-
-    this.setState({
-      results
-    });
-    return results;
-  }
-
-  _setupFramebuffers(opts) {
-    const {
-      textures,
-      framebuffers,
-      maxMinFramebuffers,
-      minFramebuffers,
-      maxFramebuffers,
-      meanTextures,
-      equations
-    } = this.state;
-    const {
-      weights
-    } = opts;
-    const {
-      numCol,
-      numRow
-    } = opts;
-    const framebufferSize = {
-      width: numCol,
-      height: numRow
-    };
-
-    for (const id in weights) {
-      const {
-        needMin,
-        needMax,
-        combineMaxMin,
-        operation
-      } = weights[id];
-      textures[id] = weights[id].aggregationTexture || textures[id] || getFloatTexture(this.device, {
-        id: `${id}-texture`,
-        width: numCol,
-        height: numRow
-      });
-      textures[id].resize(framebufferSize);
-      let texture = textures[id];
-
-      if (operation === AGGREGATION_OPERATION.MEAN) {
-        meanTextures[id] = meanTextures[id] || getFloatTexture(this.device, {
-          id: `${id}-mean-texture`,
-          width: numCol,
-          height: numRow
-        });
-        meanTextures[id].resize(framebufferSize);
-        texture = meanTextures[id];
-      }
-
-      if (framebuffers[id]) {
-        framebuffers[id].attach({
-          [36064]: texture
-        });
-      } else {
-        framebuffers[id] = getFramebuffer(this.device, {
-          id: `${id}-fb`,
-          width: numCol,
-          height: numRow,
-          texture
-        });
-      }
-
-      framebuffers[id].resize(framebufferSize);
-      equations[id] = EQUATION_MAP[operation] || EQUATION_MAP[AGGREGATION_OPERATION.SUM];
-
-      if (needMin || needMax) {
-        if (needMin && needMax && combineMaxMin) {
-          if (!maxMinFramebuffers[id]) {
-            texture = weights[id].maxMinTexture || this._getMinMaxTexture(`${id}-maxMinTexture`);
-            maxMinFramebuffers[id] = getFramebuffer(this.device, {
-              id: `${id}-maxMinFb`,
-              texture
-            });
-          }
-        } else {
-          if (needMin) {
-            if (!minFramebuffers[id]) {
-              texture = weights[id].minTexture || this._getMinMaxTexture(`${id}-minTexture`);
-              minFramebuffers[id] = getFramebuffer(this.device, {
-                id: `${id}-minFb`,
-                texture
-              });
-            }
-          }
-
-          if (needMax) {
-            if (!maxFramebuffers[id]) {
-              texture = weights[id].maxTexture || this._getMinMaxTexture(`${id}-maxTexture`);
-              maxFramebuffers[id] = getFramebuffer(this.device, {
-                id: `${id}-maxFb`,
-                texture
-              });
-            }
-          }
-        }
-      }
-    }
-  }
-
-  _getMinMaxTexture(name) {
-    const {
-      resources
-    } = this.state;
-
-    if (!resources[name]) {
-      resources[name] = getFloatTexture(this.device, {
-        id: 'resourceName'
-      });
-    }
-
-    return resources[name];
-  }
-
-  _setupModels({
-    numCol = 0,
-    numRow = 0
-  } = {}) {
-    var _this$gridAggregation;
-
-    const {
-      shaderOptions
-    } = this.state;
-    (_this$gridAggregation = this.gridAggregationModel) === null || _this$gridAggregation === void 0 ? void 0 : _this$gridAggregation.destroy();
-    this.gridAggregationModel = getAggregationModel(this.device, shaderOptions);
-
-    if (!this.allAggregationModel) {
-      const instanceCount = numCol * numRow;
-      this.allAggregationModel = getAllAggregationModel(this.device, instanceCount);
-    }
-  }
-
-  _setupWeightAttributes(opts) {
-    const {
-      weightAttributes
-    } = this.state;
-    const {
-      weights
-    } = opts;
-
-    for (const id in weights) {
-      weightAttributes[id] = opts.attributes[id];
-    }
-  }
-
-  _trackGPUResultBuffers(results, weights) {
-    const {
-      resources
-    } = this.state;
-
-    for (const id in results) {
-      if (results[id]) {
-        for (const bufferName of BUFFER_NAMES) {
-          if (results[id][bufferName] && weights[id][bufferName] !== results[id][bufferName]) {
-            const name = `gpu-result-${id}-${bufferName}`;
-
-            if (resources[name]) {
-              resources[name].delete();
-            }
-
-            resources[name] = results[id][bufferName];
-          }
-        }
-      }
-    }
-  }
-
-  _updateModels(opts) {
-    const {
-      vertexCount,
-      attributes,
-      numCol,
-      numRow
-    } = opts;
-    const {
-      modelDirty
-    } = this.state;
-
-    if (modelDirty) {
-      this._setupModels(opts);
-
-      this.setState({
-        modelDirty: false
-      });
-    }
-
-    this._setupWeightAttributes(opts);
-
-    this.gridAggregationModel.setVertexCount(vertexCount);
-    this.gridAggregationModel.setAttributes(attributes);
-    this.allAggregationModel.setInstanceCount(numCol * numRow);
-  }
-
-}
-
-function normalizeWeightParams(weights) {
-  const result = {};
-
-  for (const id in weights) {
-    result[id] = { ...DEFAULT_WEIGHT_PARAMS,
-      ...weights[id]
-    };
-  }
-
-  return result;
-}
-
-function deleteResources(resources) {
-  resources = Array.isArray(resources) ? resources : [resources];
-  resources.forEach(obj => {
-    for (const name in obj) {
-      obj[name].delete();
-    }
-  });
-}
-
-function getAggregationModel(device, shaderOptions) {
-  const shaders = mergeShaders({
-    vs: AGGREGATE_TO_GRID_VS,
-    fs: AGGREGATE_TO_GRID_FS,
-    modules: [fp64arithmetic, project32]
-  }, shaderOptions);
-  return new Model(device, {
-    id: 'Grid-Aggregation-Model',
-    vertexCount: 1,
-    drawMode: 0,
-    shaderAssembler: getShaderAssembler(),
-    ...shaders
-  });
-}
-
-function getAllAggregationModel(device, instanceCount) {
-  return new Model(device, {
-    id: 'All-Aggregation-Model',
-    vs: AGGREGATE_ALL_VS,
-    fs: AGGREGATE_ALL_FS,
-    modules: [fp64arithmetic],
-    vertexCount: 1,
-    topology: 'point-list',
-    isInstanced: true,
-    instanceCount,
-    attributes: {
-      position: [0, 0]
-    }
-  });
-}
-
-function getMeanTransform(device, opts) {
-  return new TextureTransform(device, {
-    vs: TRANSFORM_MEAN_VS,
-    _targetTextureVarying: 'meanValues',
-    ...opts
-  });
-}
-//# sourceMappingURL=gpu-grid-aggregator.js.map