Migrated source code from Simantics SVN

[simantics/platform.git] / bundles / org.simantics.structural2 / src / org / simantics / structural2 / StructuralRVIResolver.java

diff --git a/bundles/org.simantics.structural2/src/org/simantics/structural2/StructuralRVIResolver.java b/bundles/org.simantics.structural2/src/org/simantics/structural2/StructuralRVIResolver.java
new file mode 100644 (file)
index 0000000..5fdec43
--- /dev/null
+++ b/bundles/org.simantics.structural2/src/org/simantics/structural2/StructuralRVIResolver.java
@@ -0,0 +1,119 @@
+package org.simantics.structural2;\r
+\r
+import java.util.Collection;\r
+import java.util.LinkedList;\r
+\r
+import org.simantics.databoard.Databoard;\r
+import org.simantics.databoard.binding.Binding;\r
+import org.simantics.datatypes.literal.GUID;\r
+import org.simantics.db.ReadGraph;\r
+import org.simantics.db.Resource;\r
+import org.simantics.db.exception.DatabaseException;\r
+import org.simantics.db.layer0.adapter.Instances;\r
+import org.simantics.db.layer0.exception.MissingVariableException;\r
+import org.simantics.db.layer0.variable.RVI;\r
+import org.simantics.db.layer0.variable.RVI.GuidRVIPart;\r
+import org.simantics.db.layer0.variable.RVI.RVIPart;\r
+import org.simantics.db.layer0.variable.RVIBuilder;\r
+import org.simantics.db.layer0.variable.StandardRVIResolver;\r
+import org.simantics.db.layer0.variable.Variable;\r
+import org.simantics.db.layer0.variable.Variables;\r
+import org.simantics.layer0.Layer0;\r
+import org.simantics.simulation.ontology.SimulationResource;\r
+import org.simantics.structural.stubs.StructuralResource2;\r
+\r
+public class StructuralRVIResolver extends StandardRVIResolver {\r
+\r
+    protected boolean isRVIBase(ReadGraph graph, Variable variable) throws DatabaseException {\r
+        if (Variables.isContext(graph, variable)) return true;\r
+        StructuralResource2 STR = StructuralResource2.getInstance(graph);\r
+        Resource represents = variable.getRepresents(graph);\r
+        if (represents == null) return false;\r
+        if (graph.isInstanceOf(represents, STR.Composite)) return false;\r
+        return true;\r
+    }\r
+    \r
+    @Override\r
+    public RVI getRVI(ReadGraph graph, Variable variable) throws DatabaseException {\r
+        if (Variables.isContext(graph, variable)) {\r
+            Databoard databoard = graph.getService( Databoard.class );\r
+            Binding rviBinding = databoard.getBindingUnchecked( RVI.class );\r
+            return RVI.empty(rviBinding);\r
+        }\r
+        Variable base = variable.getParent(graph);\r
+        while(!isRVIBase(graph, base)) base = base.getParent(graph);\r
+        RVIPart part = getRVIPart(graph, variable);\r
+        return new RVIBuilder(base.getRVI(graph)).append(part).toRVI();\r
+    }\r
+\r
+    protected boolean isPartOfComponentType(ReadGraph graph, Resource resource)\r
+            throws DatabaseException {\r
+        Layer0 L0 = Layer0.getInstance(graph);\r
+        StructuralResource2 STR = StructuralResource2.getInstance(graph);\r
+        Resource container = graph.getPossibleObject(resource, L0.PartOf);\r
+        if(container != null && graph.isInstanceOf(container, STR.Composite)) {\r
+            return graph.hasStatement(container, STR.Defines);\r
+        }\r
+        return false;\r
+    }\r
+\r
+    protected Resource getBase(ReadGraph graph, Variable variable, Resource resource) throws DatabaseException {\r
+        Layer0 L0 = Layer0.getInstance(graph);\r
+        SimulationResource SIMU = SimulationResource.getInstance(graph);\r
+        Resource represents = variable.getPossibleRepresents(graph);\r
+        if(represents != null && graph.isInstanceOf(represents, SIMU.Run)) return Variables.getPossibleConfigurationContextResource(graph, represents);\r
+        else if(isPartOfComponentType(graph, resource)) return graph.getPossibleObject(resource, L0.PartOf);\r
+        return represents;\r
+    }\r
+\r
+    protected Collection<Resource> getRVIPath(ReadGraph graph, Variable variable, Resource resource)\r
+            throws DatabaseException {\r
+        Resource base = getBase(graph, variable, resource);\r
+        if (base == null)\r
+            return null;\r
+        LinkedList<Resource> result = new LinkedList<Resource>();\r
+        result.add(resource);\r
+        Layer0 L0 = Layer0.getInstance(graph);\r
+        resource = graph.getPossibleObject(resource, L0.PartOf);\r
+        if(resource == null) return null;\r
+        while(!base.equals(resource)) {\r
+            result.addFirst(resource);\r
+            resource = graph.getPossibleObject(resource, L0.PartOf);\r
+            if(resource == null) return null;\r
+        }\r
+        return result;\r
+    }\r
+    \r
+    protected Collection<Resource> getRVIPath(ReadGraph graph, Variable variable, GuidRVIPart part)\r
+            throws DatabaseException {\r
+\r
+       if(part.resource != null) {\r
+               return getRVIPath(graph, variable, part.resource);\r
+       } else {\r
+               Resource indexRoot = variable.getIndexRoot(graph);\r
+               Instances instances = graph.adapt(Layer0.getInstance(graph).Entity, Instances.class);\r
+               GUID guid = new GUID(part.mostSignificant, part.leastSignificant);\r
+               Collection<Resource> queryResult = instances.find(graph, indexRoot, "GUID:"+guid.indexString());\r
+               if(queryResult.size() != 1) return null;\r
+               return getRVIPath(graph, variable, queryResult.iterator().next());\r
+       }\r
+       \r
+    }\r
+\r
+    @Override\r
+    protected Variable resolveChild(ReadGraph graph, Variable variable, Resource resource) throws DatabaseException {\r
+        Collection<Resource> path = getRVIPath(graph, variable, resource);\r
+        if(path == null) throw new MissingVariableException("Didn't find a variable related to " + resource + ".", resource);\r
+        for(Resource r : path) variable = variable.browse(graph, r);\r
+        return variable;\r
+    }\r
+    \r
+    @Override\r
+    protected Variable resolveChild(ReadGraph graph, Variable variable, GuidRVIPart part) throws DatabaseException {\r
+        Collection<Resource> path = getRVIPath(graph, variable, part);\r
+        if(path == null) throw new MissingVariableException("Didn't find a variable related to " + part + ".");\r
+        for(Resource r : path) variable = variable.browse(graph, r);\r
+        return variable;\r
+    }\r
+\r
+}\r