feat(pointcloud): Switch PointCloud needs to change side&back view

open-mmlab · Aug 30, 2022 · 4dbfff5 · 4dbfff5
1 parent 3ef28b7
commit 4dbfff5
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/packages/lb-annotation/src/core/toolOperation/pointCloud2dOperation.ts b/packages/lb-annotation/src/core/toolOperation/pointCloud2dOperation.ts
@@ -224,6 +224,8 @@ class PointCloud2dOperation extends PolygonOperation {
       this.setSelectedIDs([nextSelectedResult.id]);
     }
     this.render();
+
+    return [nextSelectedResult.id];
   }
 
   /**

diff --git a/packages/lb-components/src/components/pointCloudView/hooks/useSingleBox.tsx b/packages/lb-components/src/components/pointCloudView/hooks/useSingleBox.tsx
@@ -15,6 +15,7 @@ export const useSingleBox = () => {
     selectedIDs,
     selectedID,
     mainViewInstance,
+    setSelectedIDs,
   } = useContext(PointCloudContext);
 
   /** Returns { info: selected box, index: selected box index } */
@@ -52,7 +53,8 @@ export const useSingleBox = () => {
 
       const { pointCloud2dOperation } = topViewInstance;
 
-      pointCloud2dOperation.switchToNextPolygon(sort);
+      const newSelectedIDs = pointCloud2dOperation.switchToNextPolygon(sort);
+      setSelectedIDs(newSelectedIDs);
     },
     [topViewInstance],
   );