edges and connections get copied exactly when their source and destin…

…ation is selected
DecwLK · Mar 10, 2024 · b9f5e9c · b9f5e9c
1 parent 879b1da
commit b9f5e9c
Showing 1 changed file with 13 additions and 0 deletions.
diff --git a/src/main/java/org/gecko/actions/CopyPositionableViewModelElementAction.java b/src/main/java/org/gecko/actions/CopyPositionableViewModelElementAction.java
@@ -18,6 +18,19 @@ boolean run() throws GeckoException {
         CopyPositionableViewModelElementVisitor visitor = new CopyPositionableViewModelElementVisitor(geckoViewModel);
 
         Set<PositionableViewModelElement<?>> copyQueue = geckoViewModel.getCurrentEditor().getSelectionManager().getCurrentSelection();
+        Set<Element> elementToCopy = copyQueue.stream().map(PositionableViewModelElement::getTarget).collect(Collectors.toSet());
+        for (Edge edge : geckoViewModel.getCurrentEditor().getCurrentSystem().getTarget().getAutomaton().getEdges()) {
+            if (elementToCopy.contains(edge.getSource()) && elementToCopy.contains(edge.getDestination())) {
+                copyQueue.add(geckoViewModel.getViewModelElement(edge));
+            }
+        }
+        for (SystemConnection connection : geckoViewModel.getCurrentEditor().getCurrentSystem().getTarget().getConnections()) {
+            System sourceSystem = geckoViewModel.getGeckoModel().getSystemWithVariable(connection.getSource());
+            System destinationSystem = geckoViewModel.getGeckoModel().getSystemWithVariable(connection.getDestination());
+            if (elementToCopy.contains(sourceSystem) && elementToCopy.contains(destinationSystem)) {
+                copyQueue.add(geckoViewModel.getViewModelElement(connection));
+            }
+        }
         do {
             System.out.println("Trying to copy: " + copyQueue);
             visitor.getFailedCopies().clear();