3D framework (Simca 2012)

[simantics/3d.git] / org.simantics.objmap2 / src / org / simantics / objmap / internal / MonotoneForwardMapping.java

diff --git a/org.simantics.objmap2/src/org/simantics/objmap/internal/MonotoneForwardMapping.java b/org.simantics.objmap2/src/org/simantics/objmap/internal/MonotoneForwardMapping.java
new file mode 100644 (file)
index 0000000..51065ff
--- /dev/null
+++ b/org.simantics.objmap2/src/org/simantics/objmap/internal/MonotoneForwardMapping.java
@@ -0,0 +1,47 @@
+package org.simantics.objmap.internal;\r
+\r
+import gnu.trove.map.hash.THashMap;\r
+\r
+import java.util.Collections;\r
+import java.util.Set;\r
+\r
+import org.simantics.db.ReadGraph;\r
+import org.simantics.objmap.exceptions.MappingException;\r
+import org.simantics.objmap.forward.IForwardLinkType;\r
+import org.simantics.objmap.forward.IForwardMapping;\r
+import org.simantics.objmap.forward.IForwardMappingSchema;\r
+\r
+public class MonotoneForwardMapping<Domain, Range> implements IForwardMapping<Domain, Range> {\r
+\r
+    IForwardMappingSchema<Domain, Range> schema;\r
+    THashMap<Domain, Range> map = new THashMap<Domain, Range>();\r
+    \r
+    public MonotoneForwardMapping(IForwardMappingSchema<Domain, Range> schema) {\r
+        this.schema = schema;\r
+    }\r
+\r
+    @Override\r
+    public Set<Domain> getDomain() {\r
+        return Collections.unmodifiableSet(map.keySet());\r
+    }\r
+\r
+    @Override\r
+    public Range get(Domain domainElement) {\r
+        return map.get(domainElement);\r
+    }\r
+\r
+    @Override\r
+    public Range map(ReadGraph graph, Domain domainElement) throws MappingException {\r
+        Range result = get(domainElement);\r
+        if(result == null) {\r
+            IForwardLinkType<Domain, Range> linkType = \r
+                    schema.linkTypeOfDomainElement(graph, domainElement);\r
+            // Two phase creation makes cyclic references possible\r
+            Range rangeElement = linkType.createRangeElement(graph, domainElement);\r
+            map.put(domainElement, rangeElement);\r
+            linkType.createRange(graph, this, domainElement, rangeElement);\r
+        }\r
+        return result;\r
+    }\r
+\r
+}\r