fix: Add viewOperation render data judgment

open-mmlab · Jan 10, 2022 · b35a123 · b35a123
1 parent 6976a51
commit b35a123
Showing 1 changed file with 13 additions and 1 deletion.
diff --git a/packages/lb-annotation/src/core/toolOperation/ViewOperation.ts b/packages/lb-annotation/src/core/toolOperation/ViewOperation.ts
@@ -282,6 +282,10 @@ export default class ViewOperation extends BasicToolOperation {
         }
         case 'polygon': {
           const polygon = annotation.annotation;
+          if (!(polygon?.pointList?.length >= 3)) {
+            return;
+          }
+
           const { lineType = ELineTypes.Line } = polygon;
           const renderPolygon = AxisUtils.changePointListByZoom(polygon?.pointList ?? [], this.zoom, this.currentPos);
           const style = this.getSpecificStyle(polygon);
@@ -345,9 +349,17 @@ export default class ViewOperation extends BasicToolOperation {
 
         case 'line': {
           const line = annotation.annotation;
+          if (!(line?.pointList?.length >= 2)) {
+            return;
+          }
+
           const { lineType = ELineTypes.Line } = line;
+          const renderLine = AxisUtils.changePointListByZoom(
+            (line?.pointList as IPoint[]) ?? [],
+            this.zoom,
+            this.currentPos,
+          );
 
-          const renderLine = AxisUtils.changePointListByZoom(line.pointList as IPoint[], this.zoom, this.currentPos);
           const style = this.getSpecificStyle(line);
           const newPointList = DrawUtils.drawPolygon(this.canvas, renderLine, {
             ...style,