Fixed edge arrows to always be rendered above the edge

[simantics/district.git] / org.simantics.district.network / src / org / simantics / district / network / profile / MidBranchEdgeSetRequest.java

diff --git a/org.simantics.district.network/src/org/simantics/district/network/profile/MidBranchEdgeSetRequest.java b/org.simantics.district.network/src/org/simantics/district/network/profile/MidBranchEdgeSetRequest.java
index c1a9394fb5e043d00936635fb66f755de636ef4d..ac66b88ffaf89c0c5dbcbf109afa24ebb5e2c447 100644 (file)
--- a/org.simantics.district.network/src/org/simantics/district/network/profile/MidBranchEdgeSetRequest.java
+++ b/org.simantics.district.network/src/org/simantics/district/network/profile/MidBranchEdgeSetRequest.java
@@ -1,6 +1,5 @@
 package org.simantics.district.network.profile;
 
-import java.util.HashSet;
 import java.util.List;
 import java.util.Set;
 
@@ -9,6 +8,7 @@ import org.simantics.db.ReadGraph;
 import org.simantics.db.Resource;
 import org.simantics.db.common.request.ResourceRead;
 import org.simantics.db.exception.DatabaseException;
+import org.simantics.db.service.CollectionSupport;
 
 public class MidBranchEdgeSetRequest extends ResourceRead<Set<Resource>> {
        public MidBranchEdgeSetRequest(Resource resource) {
@@ -18,6 +18,7 @@ public class MidBranchEdgeSetRequest extends ResourceRead<Set<Resource>> {
        @Override
        public Set<Resource> perform(ReadGraph graph) throws DatabaseException {
                List<Resource> edges = Simantics.applySCL("Simantics/District/Algorithm", "midBranchEdges", graph, resource);
-               return new HashSet<>(edges);
+               CollectionSupport cs = graph.getService(CollectionSupport.class);
+               return cs.getResourceSet(graph, edges);
        }
 }