HeatmapLayer fix to work with project UBO

visgl · Jul 23, 2024 · 4239e47 · 4239e47
1 parent 4861a55
commit 4239e47
Showing 1 changed file with 12 additions and 1 deletion.
diff --git a/modules/aggregation-layers/src/heatmap-layer/heatmap-layer.ts b/modules/aggregation-layers/src/heatmap-layer/heatmap-layer.ts
@@ -41,7 +41,8 @@ import {
   log,
   Position,
   UpdateParameters,
-  DefaultProps
+  DefaultProps,
+  project32
 } from '@deck.gl/core';
 import TriangleLayer from './triangle-layer';
 import AggregationLayer, {AggregationLayerProps} from '../aggregation-layer';
@@ -201,6 +202,10 @@ export default class HeatmapLayer<
     viewportCorners: number[][];
   };
 
+  getShaders(shaders: any) {
+    return super.getShaders({...shaders, modules: [project32]});
+  }
+
   initializeState() {
     super.initializeAggregationLayer(DIMENSIONS);
     this.setState({colorDomain: DEFAULT_COLOR_DOMAIN});
@@ -593,6 +598,12 @@ export default class HeatmapLayer<
     weightsTransform.model.setVertexCount(this.getNumInstances());
     weightsTransform.model.setUniforms(uniforms);
     weightsTransform.model.updateModuleSettings(moduleSettings);
+
+    const {viewport, devicePixelRatio, coordinateSystem, coordinateOrigin} = moduleSettings;
+    const {modelMatrix} = this.props;
+    weightsTransform.model.shaderInputs.setProps({
+      project: {viewport, devicePixelRatio, modelMatrix, coordinateSystem, coordinateOrigin}
+    });
     weightsTransform.run({
       parameters: {viewport: [0, 0, textureSize, textureSize]},
       clearColor: [0, 0, 0, 0]