feat: Improve polygon drawing UX. #1597

h2oai · Nov 3, 2022 · 040f324 · 040f324
1 parent 85c8436
commit 040f324
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/ui/src/image_annotator_polygon.ts b/ui/src/image_annotator_polygon.ts
@@ -106,13 +106,15 @@ export class PolygonAnnotator {
     this.ctx.beginPath()
     this.ctx.moveTo(points[0].x, points[0].y)
 
-    points.forEach(({ x, y }) => this.drawLine(x, y))
+    const _points = this.draggedPoint ? points.filter(p => !p.isAux) : points
+
+    _points.forEach(({ x, y }) => this.drawLine(x, y))
     if (joinLastPoint) this.drawLine(points[0].x, points[0].y)
 
     if (isFocused) {
       this.ctx.fillStyle = color.substring(0, color.length - 2) + '0.2)'
       this.ctx.fill()
-      points.forEach(({ x, y, isAux }) => this.drawPoint(x, y, isAux))
+      _points.forEach(({ x, y, isAux }) => this.drawPoint(x, y, isAux))
     }
   }