Migrated source code from Simantics SVN

[simantics/platform.git] / bundles / org.simantics.db.layer0 / src / org / simantics / db / layer0 / request / ModelInstances.java

diff --git a/bundles/org.simantics.db.layer0/src/org/simantics/db/layer0/request/ModelInstances.java b/bundles/org.simantics.db.layer0/src/org/simantics/db/layer0/request/ModelInstances.java
new file mode 100644 (file)
index 0000000..6b25f54
--- /dev/null
+++ b/bundles/org.simantics.db.layer0/src/org/simantics/db/layer0/request/ModelInstances.java
@@ -0,0 +1,72 @@
+package org.simantics.db.layer0.request;\r
+\r
+import java.util.Collection;\r
+import java.util.Collections;\r
+import java.util.HashMap;\r
+import java.util.Map;\r
+\r
+import org.simantics.databoard.Bindings;\r
+import org.simantics.db.ReadGraph;\r
+import org.simantics.db.Resource;\r
+import org.simantics.db.common.request.PossibleIndexRoot;\r
+import org.simantics.db.common.request.ResourceRead2;\r
+import org.simantics.db.exception.DatabaseException;\r
+import org.simantics.db.layer0.adapter.Instances;\r
+import org.simantics.layer0.Layer0;\r
+import org.simantics.simulation.ontology.SimulationResource;\r
+\r
+/**\r
+ * Returns a Name->Resource map for instances of given type in a given model.\r
+ * \r
+ * Uses Dependencies-index for obtaining instances. Uses L0.HasName for\r
+ * obtaining name.\r
+ * \r
+ * @author Antti Villberg\r
+ */\r
+public class ModelInstances extends ResourceRead2<Map<String, Resource>> {\r
+\r
+    /**\r
+     * @parameter modelPart is a Resource with URI under a model e.g. model can\r
+     *            be found using L0.PartOf\r
+     * @parameter type is the type to search instances for. The query returns\r
+     *            properly all instances of all subtypes.\r
+     */\r
+    public ModelInstances(Resource modelPart, Resource type) {\r
+        super(modelPart, type);\r
+    }\r
+\r
+    private Resource getModel(ReadGraph graph) throws DatabaseException {\r
+        SimulationResource SIMU = SimulationResource.getInstance(graph);\r
+        if (graph.isInstanceOf(resource, SIMU.Model)) {\r
+            return resource;\r
+        } else {\r
+            return graph.sync(new PossibleIndexRoot(resource));\r
+        }\r
+    }\r
+\r
+    @Override\r
+    public Map<String, Resource> perform(ReadGraph graph) throws DatabaseException {\r
+\r
+        Layer0 L0 = Layer0.getInstance(graph);\r
+\r
+        Instances query = graph.adapt(resource2, Instances.class);\r
+        if (query == null)\r
+            return Collections.emptyMap();\r
+\r
+        Resource model = getModel(graph);\r
+        if(model == null)\r
+            return Collections.emptyMap();\r
+        Collection<Resource> queryResult = query.find(graph, model);\r
+        if (queryResult.isEmpty())\r
+            return Collections.emptyMap();\r
+\r
+        Map<String, Resource> result = new HashMap<String, Resource>(queryResult.size());\r
+        for(Resource instance : queryResult) {\r
+            String name = graph.getRelatedValue(instance, L0.HasName, Bindings.STRING);\r
+            result.put(name, instance);\r
+        }\r
+        return result;\r
+\r
+    }\r
+\r
+}\r