فهرست منبع

delete Object Edges Fix

Kevin Trometer 8 سال پیش
والد
کامیت
946e4260a1
1فایلهای تغییر یافته به همراه9 افزوده شده و 1 حذف شده
  1. 9 1
      src/ui/controller/CanvasController.java

+ 9 - 1
src/ui/controller/CanvasController.java

@@ -2,7 +2,9 @@ package ui.controller;
 
 import Interfaces.CategoryListener;
 import Interfaces.ObjectListener;
+import classes.CpsEdge;
 import classes.CpsObject;
+import javafx.util.Pair;
 import ui.model.Model;
 
 public class CanvasController {
@@ -45,8 +47,14 @@ public class CanvasController {
 	 * @param CpsObject
 	 */
 	public void deleteObjectOnCanvas(CpsObject obj) {
+		Pair<CpsObject, CpsEdge> tempPair = null;
 		for (CpsObject cps : MODEL.getObjectsOnCanvas()) {
-			cps.getConnectedTo().remove(obj);
+			for (Pair<CpsObject, CpsEdge> p : cps.getConnections()) {
+				if (p.getKey() == obj) {
+					tempPair = p;
+				}
+			}
+			cps.getConnectedTo().remove(tempPair);
 		}
 		MODEL.getObjectsOnCanvas().remove(obj);
 	}