feat: Split the function of PolygonOpeartion

open-mmlab · Jul 18, 2022 · fb7631b · fb7631b
1 parent e3dd7ac
commit fb7631b
Showing 1 changed file with 20 additions and 9 deletions.
diff --git a/packages/lb-annotation/src/core/toolOperation/polygonOperation.ts b/packages/lb-annotation/src/core/toolOperation/polygonOperation.ts
@@ -112,6 +112,10 @@ class PolygonOperation extends BasicToolOperation {
     return PolygonUtils.getPolygonByID(this.polygonList, this.selectedID);
   }
 
+  public get hoverPolygon() {
+    return this.polygonList.find((v) => v.id === this.hoverID && v.id !== this.selectedID);
+  }
+
   public get polygonListUnderZoom() {
     return this.polygonList.map((polygon) => ({
       ...polygon,
@@ -140,9 +144,8 @@ class PolygonOperation extends BasicToolOperation {
     return this.polygonList;
   }
 
-  // 更改当前标注模式
-  public setPattern(pattern: EPolygonPattern) {
-    if (this.drawingPointList?.length > 0) {
+  public setPattern(pattern: EPolygonPattern, isForce = false) {
+    if (this.drawingPointList?.length > 0 && isForce === true) {
       // 编辑中不允许直接跳出
       return;
     }
@@ -364,6 +367,10 @@ class PolygonOperation extends BasicToolOperation {
     this.setSelectedID(undefined);
   }
 
+  public clearDrawingStatus() {
+    this.drawingPointList = [];
+  }
+
   // SET DATA
   public setPolygonList(polygonList: IPolygonData[]) {
     const oldLen = this.polygonList.length;
@@ -1531,13 +1538,9 @@ class PolygonOperation extends BasicToolOperation {
     }
   }
 
-  public renderPolygon() {
-    // 1. 静态多边形
-    this.renderStaticPolygon();
-
-    // 2. hover 多边形
+  public renderHoverPolygon() {
     if (this.hoverID && this.hoverID !== this.editPolygonID) {
-      const hoverPolygon = this.polygonList.find((v) => v.id === this.hoverID && v.id !== this.selectedID);
+      const { hoverPolygon } = this;
       if (hoverPolygon) {
         let color = '';
         const toolColor = this.getColor(hoverPolygon.attribute);
@@ -1557,6 +1560,14 @@ class PolygonOperation extends BasicToolOperation {
         );
       }
     }
+  }
+
+  public renderPolygon() {
+    // 1. 静态多边形
+    this.renderStaticPolygon();
+
+    // 2. hover 多边形
+    this.renderHoverPolygon();
 
     // 3. 选中多边形的渲染
     this.renderSelectedPolygon();