feat: Improve polygon drawing UX. #1597

h2oai · Oct 3, 2022 · 78e7b9a · 78e7b9a
1 parent 8712e3c
commit 78e7b9a
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))
     }
   }