fix(web): selection without groundPrimitives (#938)

reearth · Mar 25, 2024 · 2f42938 · 2f42938
1 parent 459c94f
commit 2f42938
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/web/src/beta/lib/core/engines/Cesium/Feature/HeatMap/HeatmapMesh.tsx b/web/src/beta/lib/core/engines/Cesium/Feature/HeatMap/HeatmapMesh.tsx
@@ -62,7 +62,7 @@ export const HeatmapMesh = memo(
       ref,
     ) => {
       const { scene } = useCesium();
-      const groundPrimitives = scene?.primitives;
+      const groundPrimitives = scene?.groundPrimitives;
       const primitiveRef = useRef<GroundPrimitive>();
 
       const material = useConstant(() =>

diff --git a/web/src/beta/lib/core/engines/Cesium/pickMany.ts b/web/src/beta/lib/core/engines/Cesium/pickMany.ts
@@ -221,6 +221,8 @@ function pickManyFromViewport(
   windowWidth: number,
   windowHeight: number,
 ): object[] {
+  const showGroundPrimitives = scene.groundPrimitives.show;
+  scene.groundPrimitives.show = false;
   invariant(windowWidth > 0 && windowHeight > 0);
   assertType<PrivateScene>(scene);
   const context = scene.context;
@@ -272,6 +274,8 @@ function pickManyFromViewport(
   const objects = pickFramebufferEnd.apply(view.pickFramebuffer, [rectangleScratch]);
   context.endFrame();
 
+  scene.groundPrimitives.show = showGroundPrimitives;
+
   return objects;
 }