Migrated source code from Simantics SVN

[simantics/platform.git] / bundles / org.simantics.spreadsheet.graph / src / org / simantics / spreadsheet / graph / SpreadsheetStates.java

diff --git a/bundles/org.simantics.spreadsheet.graph/src/org/simantics/spreadsheet/graph/SpreadsheetStates.java b/bundles/org.simantics.spreadsheet.graph/src/org/simantics/spreadsheet/graph/SpreadsheetStates.java
new file mode 100644 (file)
index 0000000..926d9d2
--- /dev/null
+++ b/bundles/org.simantics.spreadsheet.graph/src/org/simantics/spreadsheet/graph/SpreadsheetStates.java
@@ -0,0 +1,41 @@
+package org.simantics.spreadsheet.graph;\r
+\r
+import java.util.Collection;\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.ObjectsWithType;\r
+import org.simantics.db.exception.DatabaseException;\r
+import org.simantics.db.layer0.request.VariableRead;\r
+import org.simantics.db.layer0.variable.Variable;\r
+import org.simantics.layer0.Layer0;\r
+import org.simantics.spreadsheet.resource.SpreadsheetResource;\r
+\r
+public class SpreadsheetStates extends VariableRead<Map<String, Resource>> {\r
+\r
+    public SpreadsheetStates(Variable variable) {\r
+        super(variable);\r
+    }\r
+\r
+    @Override\r
+    public Map<String, Resource> perform(ReadGraph graph) throws DatabaseException {\r
+        Resource book = variable.getParent(graph).getRepresents(graph);\r
+        Layer0 L0 = Layer0.getInstance(graph);\r
+        Collection<Resource> states = graph.syncRequest(new ObjectsWithType(book, L0.ConsistsOf, SpreadsheetResource.getInstance(graph).InitialCondition));\r
+        Map<String, Resource> result = new HashMap<>(states.size());\r
+        states.forEach(state -> {\r
+            try {\r
+                String name = graph.getPossibleRelatedValue(state, L0.HasName, Bindings.STRING);\r
+                result.put(name, state);\r
+            } catch (Exception e) {\r
+                e.printStackTrace();\r
+            }\r
+        });\r
+        return result;\r
+        \r
+    }\r
+\r
+}\r