Static information label profile definition.

[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
new file mode 100644 (file)
index 0000000..c1a9394
--- /dev/null
+++ b/org.simantics.district.network/src/org/simantics/district/network/profile/MidBranchEdgeSetRequest.java
@@ -0,0 +1,23 @@
+package org.simantics.district.network.profile;
+
+import java.util.HashSet;
+import java.util.List;
+import java.util.Set;
+
+import org.simantics.Simantics;
+import org.simantics.db.ReadGraph;
+import org.simantics.db.Resource;
+import org.simantics.db.common.request.ResourceRead;
+import org.simantics.db.exception.DatabaseException;
+
+public class MidBranchEdgeSetRequest extends ResourceRead<Set<Resource>> {
+       public MidBranchEdgeSetRequest(Resource resource) {
+               super(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);
+       }
+}