Change listening for district elements on network-diagram

[simantics/district.git] / org.simantics.district.network.ui / src / org / simantics / district / network / ui / DNElementRemover.java

diff --git a/org.simantics.district.network.ui/src/org/simantics/district/network/ui/DNElementRemover.java b/org.simantics.district.network.ui/src/org/simantics/district/network/ui/DNElementRemover.java
index 2cbfe98e8b291c58bc2d5de04bef0d8c1943f50d..06b31031f2efc794eca78c498aaaf5e0a1287e29 100644 (file)
--- a/org.simantics.district.network.ui/src/org/simantics/district/network/ui/DNElementRemover.java
+++ b/org.simantics.district.network.ui/src/org/simantics/district/network/ui/DNElementRemover.java
@@ -1,61 +1,81 @@
-package org.simantics.district.network.ui;\r
-\r
-import org.simantics.db.Resource;\r
-import org.simantics.db.WriteGraph;\r
-import org.simantics.db.common.utils.OrderedSetUtils;\r
-import org.simantics.db.exception.DatabaseException;\r
-import org.simantics.db.layer0.adapter.impl.EntityRemover;\r
-import org.simantics.district.network.ontology.DistrictNetworkResource;\r
-import org.simantics.layer0.utils.binaryPredicates.OrderedSetElementsPredicate;\r
-import org.simantics.modeling.adapters.ElementRemover;\r
-\r
-public class DNElementRemover extends ElementRemover {\r
-\r
-    public DNElementRemover(Resource element) {\r
-        super(element);\r
-    }\r
-    \r
-    @Override\r
-    public void removeConnection(WriteGraph graph) throws DatabaseException {\r
-        throw new UnsupportedOperationException("Distrct network should not have STR.Connection resources! " + resource);\r
-    }\r
-    \r
-    @Override\r
-    public void removeElement(WriteGraph graph) throws DatabaseException {\r
-        \r
-        DistrictNetworkResource DN = DistrictNetworkResource.getInstance(graph);\r
-        \r
-        if (graph.isInstanceOf(resource, DN.Vertex)) {\r
-            graph.getObjects(resource, DN.HasEndVertex_Inverse).forEach(res -> {\r
-                try {\r
-                    doRemove(graph, res);\r
-                } catch (DatabaseException e) {\r
-                    e.printStackTrace();\r
-                }\r
-            });\r
-            graph.getObjects(resource, DN.HasStartVertex_Inverse).forEach(res -> {\r
-                try {\r
-                    doRemove(graph, res);\r
-                } catch (DatabaseException e) {\r
-                    e.printStackTrace();\r
-                }\r
-            });\r
-        }\r
-        doRemove(graph, resource);\r
-    }\r
-    \r
-    private static void doRemove(WriteGraph graph, Resource resource) throws DatabaseException {\r
-        // 1. Disconnect element from diagrams\r
-        for (Resource diagram : OrderedSetElementsPredicate.INSTANCE.getSubjects(graph, resource)) {\r
-            OrderedSetUtils.remove(graph, diagram, resource);\r
-        }\r
-\r
-        // 2. Delete element itself\r
-        EntityRemover.remove(graph, resource);\r
-\r
-        // 3. Recursively remove all related degenerate connections\r
-        // i.e. connections that only have one connector after remove the\r
-        // one connected to this removed element.\r
-    }\r
-\r
-}\r
+package org.simantics.district.network.ui;
+
+import org.simantics.db.Resource;
+import org.simantics.db.WriteGraph;
+import org.simantics.db.common.utils.OrderedSetUtils;
+import org.simantics.db.exception.DatabaseException;
+import org.simantics.db.layer0.adapter.impl.EntityRemover;
+import org.simantics.db.layer0.util.RemoverUtil;
+import org.simantics.district.network.DistrictNetworkUtil;
+import org.simantics.district.network.ontology.DistrictNetworkResource;
+import org.simantics.layer0.Layer0;
+import org.simantics.layer0.utils.binaryPredicates.OrderedSetElementsPredicate;
+import org.simantics.modeling.adapters.ElementRemover;
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
+
+public class DNElementRemover extends ElementRemover {
+
+    private static final Logger LOGGER = LoggerFactory.getLogger(DNElementRemover.class);
+
+    public DNElementRemover(Resource element) {
+        super(element);
+    }
+    
+    @Override
+    public void removeConnection(WriteGraph graph) throws DatabaseException {
+        throw new UnsupportedOperationException("Distrct network should not have STR.Connection resources! " + resource);
+    }
+    
+    @Override
+    public void removeElement(WriteGraph graph) throws DatabaseException {
+        
+        DistrictNetworkResource DN = DistrictNetworkResource.getInstance(graph);
+        
+        if (graph.isInstanceOf(resource, DN.Vertex)) {
+            graph.getObjects(resource, DN.HasEndVertex_Inverse).forEach(res -> {
+                try {
+                    doRemove(graph, res);
+                } catch (DatabaseException e) {
+                    LOGGER.error("Could not remove endvertex inverse for {}", resource, e);
+                }
+            });
+            graph.getObjects(resource, DN.HasStartVertex_Inverse).forEach(res -> {
+                try {
+                    doRemove(graph, res);
+                } catch (DatabaseException e) {
+                    LOGGER.error("Could not remove startvertex inverse for {}", resource, e);
+                }
+            });
+        }
+        doRemove(graph, resource);
+    }
+    
+    private static void doRemove(WriteGraph graph, Resource resource) throws DatabaseException {
+        // 1. Disconnect element from diagrams
+        for (Resource diagram : OrderedSetElementsPredicate.INSTANCE.getSubjects(graph, resource)) {
+            OrderedSetUtils.remove(graph, diagram, resource);
+        }
+
+        removePossibleMappedComponents(graph, resource);
+        
+        // 2. Delete element itself
+        EntityRemover.remove(graph, resource);
+
+        // 3. Recursively remove all related degenerate connections
+        // i.e. connections that only have one connector after remove the
+        // one connected to this removed element.
+    }
+
+    private static void removePossibleMappedComponents(WriteGraph graph, Resource resource) throws DatabaseException {
+        Boolean trackChangesEnabled = DistrictNetworkUtil.trackChangesEnabled(graph, graph.getPossibleObject(resource, Layer0.getInstance(graph).PartOf));
+        if (trackChangesEnabled) {
+            DistrictNetworkResource DN = DistrictNetworkResource.getInstance(graph);
+            Resource mappedComponent = graph.getPossibleObject(resource, DN.MappedComponent);
+            if (mappedComponent != null) {
+                RemoverUtil.remove(graph, mappedComponent);
+            }
+        }
+    }
+
+}