feat(pointcloud2d): Switch polygon when it has next one

open-mmlab · Aug 30, 2022 · 37b964b · 37b964b
1 parent aacbbdd
commit 37b964b
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 4 deletions.
diff --git a/packages/lb-annotation/src/core/toolOperation/pointCloud2dOperation.ts b/packages/lb-annotation/src/core/toolOperation/pointCloud2dOperation.ts
@@ -222,10 +222,9 @@ class PointCloud2dOperation extends PolygonOperation {
     const nextSelectedResult = CommonToolUtils.getNextSelectedRectID(sortList, sort, this.selectedID);
     if (nextSelectedResult) {
       this.setSelectedIDs([nextSelectedResult.id]);
+      this.render();
+      return [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
@@ -54,7 +54,9 @@ export const useSingleBox = () => {
       const { pointCloud2dOperation } = topViewInstance;
 
       const newSelectedIDs = pointCloud2dOperation.switchToNextPolygon(sort);
-      setSelectedIDs(newSelectedIDs);
+      if (newSelectedIDs) {
+        setSelectedIDs(newSelectedIDs);
+      }
     },
     [topViewInstance],
   );