Migrated source code from Simantics SVN

[simantics/platform.git] / bundles / org.simantics.objmap2 / src / org / simantics / objmap / internal / BidirectionalMapping.java

diff --git a/bundles/org.simantics.objmap2/src/org/simantics/objmap/internal/BidirectionalMapping.java b/bundles/org.simantics.objmap2/src/org/simantics/objmap/internal/BidirectionalMapping.java
new file mode 100644 (file)
index 0000000..7861503
--- /dev/null
+++ b/bundles/org.simantics.objmap2/src/org/simantics/objmap/internal/BidirectionalMapping.java
@@ -0,0 +1,140 @@
+/*******************************************************************************\r
+ * Copyright (c) 2012, 2013 Association for Decentralized Information Management in\r
+ * Industry THTH ry.\r
+ * All rights reserved. This program and the accompanying materials\r
+ * are made available under the terms of the Eclipse Public License v1.0\r
+ * which accompanies this distribution, and is available at\r
+ * http://www.eclipse.org/legal/epl-v10.html\r
+ *\r
+ * Contributors:\r
+ *     VTT Technical Research Centre of Finland - initial API and implementation\r
+ *******************************************************************************/\r
+package org.simantics.objmap.internal;\r
+\r
+import gnu.trove.map.hash.THashMap;\r
+\r
+import java.util.ArrayList;\r
+import java.util.Collection;\r
+import java.util.Collections;\r
+import java.util.Set;\r
+\r
+import org.simantics.db.ReadGraph;\r
+import org.simantics.db.WriteGraph;\r
+import org.simantics.objmap.bidirectional.IBidirectionalLinkType;\r
+import org.simantics.objmap.bidirectional.IBidirectionalMapping;\r
+import org.simantics.objmap.bidirectional.IBidirectionalMappingSchema;\r
+import org.simantics.objmap.exceptions.MappingException;\r
+\r
+public class BidirectionalMapping<Domain, Range> implements IBidirectionalMapping<Domain, Range> {\r
+\r
+    IBidirectionalMappingSchema<Domain, Range> schema;\r
+    THashMap<Domain, BidirectionalLink<Domain, Range>> forwardMap =\r
+            new THashMap<Domain, BidirectionalLink<Domain, Range>>();\r
+    THashMap<Range, BidirectionalLink<Domain, Range>> backwardMap =\r
+            new THashMap<Range, BidirectionalLink<Domain, Range>>();\r
+    \r
+    ArrayList<BidirectionalLink<Domain, Range>> modifiedDomain = new ArrayList<BidirectionalLink<Domain, Range>>();\r
+    ArrayList<BidirectionalLink<Domain, Range>> modifiedRange = new ArrayList<BidirectionalLink<Domain, Range>>();\r
+    \r
+    private void markDomainModified(BidirectionalLink<Domain, Range> link) {\r
+        if(!link.domainModified) {\r
+            link.domainModified = true;\r
+            modifiedDomain.add(link);\r
+        }\r
+    }\r
+    \r
+    private void markRangeModified(BidirectionalLink<Domain, Range> link) {\r
+        if(!link.rangeModified) {\r
+            link.rangeModified = true;\r
+            modifiedRange.add(link);\r
+        }\r
+    }\r
+    \r
+    public BidirectionalMapping(IBidirectionalMappingSchema<Domain, Range> schema) {\r
+        this.schema = schema;\r
+    }\r
+\r
+    private BidirectionalLink<Domain, Range> addLink(IBidirectionalLinkType<Domain, Range> linkType, Domain domainElement, Range rangeElement) {\r
+        BidirectionalLink<Domain, Range> link = \r
+                new BidirectionalLink<Domain, Range>(linkType, domainElement, rangeElement);\r
+        forwardMap.put(domainElement, link);\r
+        backwardMap.put(rangeElement, link);\r
+        return link;\r
+    }\r
+    \r
+    @Override\r
+    public Set<Domain> getDomain() {\r
+        return Collections.unmodifiableSet(forwardMap.keySet());\r
+    }\r
+\r
+    @Override\r
+    public Range get(Domain domainElement) {\r
+        BidirectionalLink<Domain, Range> link = forwardMap.get(domainElement);\r
+        if(link == null)\r
+            return null;\r
+        return link.rangeElement;\r
+    }\r
+\r
+    @Override\r
+    public Range map(ReadGraph graph, Domain domainElement) throws MappingException {\r
+        Range result = get(domainElement);\r
+        if(result == null) {\r
+            IBidirectionalLinkType<Domain, Range> linkType = \r
+                    schema.linkTypeOfDomainElement(graph, domainElement);\r
+            Range rangeElement = linkType.createRangeElement(graph, domainElement);\r
+            addLink(linkType, domainElement, rangeElement);\r
+            linkType.createRange(graph, this, domainElement, rangeElement);\r
+        }\r
+        return result;\r
+    }\r
+    \r
+    public Collection<Range> updateRange(ReadGraph graph) throws MappingException {\r
+        ArrayList<Range> updated = new ArrayList<Range>(Math.max(10, modifiedDomain.size())); \r
+        for(BidirectionalLink<Domain, Range> link : modifiedDomain) {\r
+            link.domainModified = false;\r
+            if(link.linkType.updateRange(graph, this, link.domainElement, link.rangeElement))\r
+                updated.add(link.rangeElement);\r
+        }\r
+        modifiedDomain.clear();\r
+        return updated;        \r
+    }\r
+\r
+    @Override\r
+    public Set<Range> getRange() {\r
+        return Collections.unmodifiableSet(backwardMap.keySet());\r
+    }\r
+\r
+    @Override\r
+    public Domain inverseGet(Range rangeElement) {\r
+        BidirectionalLink<Domain, Range> link = backwardMap.get(rangeElement);\r
+        if(link == null)\r
+            return null;\r
+        return link.domainElement;\r
+    }\r
+\r
+    @Override\r
+    public Domain inverseMap(WriteGraph graph, Range rangeElement)\r
+            throws MappingException {\r
+        Domain result = inverseGet(rangeElement);\r
+        if(result == null) {\r
+            IBidirectionalLinkType<Domain, Range> linkType = \r
+                    schema.linkTypeOfRangeElement(graph, rangeElement);\r
+            Domain domainElement = linkType.createDomainElement(graph, rangeElement);\r
+            addLink(linkType, domainElement, rangeElement);\r
+            linkType.createDomain(graph, this, domainElement, rangeElement);\r
+        }\r
+        return result;\r
+    }\r
+    \r
+    public Collection<Domain> updateDomain(WriteGraph graph) throws MappingException {\r
+        ArrayList<Domain> updated = new ArrayList<Domain>(Math.max(10, modifiedRange.size())); \r
+        for(BidirectionalLink<Domain, Range> link : modifiedRange) {\r
+            link.rangeModified = false;\r
+            if(link.linkType.updateDomain(graph, this, link.domainElement, link.rangeElement))\r
+                updated.add(link.domainElement);\r
+        }\r
+        modifiedDomain.clear();\r
+        return updated;        \r
+    }\r
+\r
+}\r