feat: Add delete connections

shahrukhqasim · Oct 13, 2017 · 26e5fa5 · 26e5fa5
1 parent 2ffeee2
commit 26e5fa5
Showing 1 changed file with 26 additions and 1 deletion.
diff --git a/java/label-tool/src/io/github/shahrukhqasim/ties/label/InteractionManager.java b/java/label-tool/src/io/github/shahrukhqasim/ties/label/InteractionManager.java
@@ -58,22 +58,38 @@ public void drag(Point2D startPoint, Point2D endPoint, double scale) {
     }
 
     void select(boolean select, Vector<Box> boxes, double scale, Rectangle2D rectangle) {
-        for(Box box : boxes) {
+        for(Selectable box : boxes) {
+            box.select(rectangle, scale);
+        }
+    }
+
+    void selectConnections(boolean select, Vector<Connection> boxes, double scale, Rectangle2D rectangle) {
+        for(Selectable box : boxes) {
             box.select(rectangle, scale);
         }
     }
 
+
     private void dragReleasedWithoutConnection(Rectangle2D rectangle, double scale, MouseButton button) {
         selectionBox.setBoundingBox(new Rectangle2D(0,0,0,0));
         if (button == MouseButton.PRIMARY) {
             // For OCR
             select(true, this.boxesOcr.getBoxes(), scale, rectangle);
             select(true, boxesCell.getBoxes(), scale, new Rectangle2D(0,0,0,0));
+            selectConnections(true, connections.getConnections(), scale, new Rectangle2D(0,0,0,0));
         }
         else if (button == MouseButton.SECONDARY) {
             // For Cell boxes
             select(true, boxesCell.getBoxes(), scale, rectangle);
             select(true, boxesOcr.getBoxes(), scale, new Rectangle2D(0,0,0,0));
+            selectConnections(true, connections.getConnections(), scale, new Rectangle2D(0,0,0,0));
+        }
+        else if (button == MouseButton.MIDDLE) {
+            // For connections
+            selectConnections(true, connections.getConnections(), scale, rectangle);
+            select(true, boxesOcr.getBoxes(), scale, new Rectangle2D(0,0,0,0));
+            select(true, boxesCell.getBoxes(), scale, new Rectangle2D(0,0,0,0));
+
         }
     }
 
@@ -155,6 +171,15 @@ public void keyPressed(double scale, KeyCode keyCode) {
             for (Box b:toBeDeleted) {
                 boxes.remove(b);
             }
+            Vector<Connection> connections = this.connections.getConnections();
+            Vector<Connection> toBeDeletedConnections = new Vector<>();
+            for (Connection c : connections) {
+                if (c.isSelected())
+                    toBeDeletedConnections.add(c);
+            }
+            for (Connection c : toBeDeletedConnections) {
+                connections.remove(c);
+            }
         }
 
         if (keyCode == KeyCode.C) {