X-Git-Url: https://gerrit.simantics.org/r/gitweb?a=blobdiff_plain;f=org.simantics.district.network%2Fsrc%2Forg%2Fsimantics%2Fdistrict%2Fnetwork%2Fprofile%2FMidBranchEdgeSetRequest.java;fp=org.simantics.district.network%2Fsrc%2Forg%2Fsimantics%2Fdistrict%2Fnetwork%2Fprofile%2FMidBranchEdgeSetRequest.java;h=ac66b88ffaf89c0c5dbcbf109afa24ebb5e2c447;hb=9b72e2f7a188fcb04add07251cbb0081edc3bfbe;hp=c1a9394fb5e043d00936635fb66f755de636ef4d;hpb=7ebf5a4d72468b19d55cb79c0494d0538081d2ff;p=simantics%2Fdistrict.git

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 c1a9394f..ac66b88f 100644
--- 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);
 	}
 }