feat: PointCloud keyboard event: switchToNextBox

open-mmlab · Aug 3, 2022 · d00774e · d00774e
1 parent d0fa1c1
commit d00774e
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 2 deletions.
diff --git a/packages/lb-annotation/src/core/toolOperation/pointCloud2dOperation.ts b/packages/lb-annotation/src/core/toolOperation/pointCloud2dOperation.ts
@@ -17,6 +17,11 @@ import StyleUtils from '@/utils/tool/StyleUtils';
 import PolygonOperation from './polygonOperation';
 
 class PointCloud2dOperation extends PolygonOperation {
+  /**
+   * remove default keydown event
+   */
+  public onKeyDown = () => {};
+
   public getLineColor() {
     return 'rgba(0, 255, 255, 0.5)';
   }

diff --git a/packages/lb-components/src/components/pointCloudView/PointCloudListener.tsx b/packages/lb-components/src/components/pointCloudView/PointCloudListener.tsx
@@ -1,13 +1,15 @@
 import { useContext, useEffect } from 'react';
-import { PointCloudContext, useRotate } from './PointCloudContext';
-import { cTool } from '@labelbee/lb-annotation';
+import { PointCloudContext, useNextOne, useRotate } from './PointCloudContext';
+import { cTool, cAnnotation } from '@labelbee/lb-annotation';
 import { message } from 'antd';
+const { ESortDirection } = cAnnotation;
 
 const { EPolygonPattern } = cTool;
 
 const PointCloudListener = () => {
   const ptCtx = useContext(PointCloudContext);
   const { updateRotate } = useRotate();
+  const { switchToNextPolygon } = useNextOne();
 
   useEffect(() => {
     const { topViewInstance, mainViewInstance } = ptCtx;
@@ -72,6 +74,14 @@ const PointCloudListener = () => {
           ptCtx.setPointCloudValid(!ptCtx.valid);
           break;
 
+        case 'z':
+          switchToNextPolygon(ESortDirection.ascend);
+          break;
+
+        case 'c':
+          switchToNextPolygon(ESortDirection.descend);
+          break;
+
         default: {
           return;
         }