--- /dev/null
+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