--- /dev/null
+package org.simantics.graph.query;\r
+\r
+import gnu.trove.list.array.TIntArrayList;\r
+import gnu.trove.map.hash.THashMap;\r
+import gnu.trove.set.hash.THashSet;\r
+import gnu.trove.set.hash.TIntHashSet;\r
+\r
+import java.util.ArrayList;\r
+import java.util.Collection;\r
+import java.util.Collections;\r
+\r
+import org.simantics.databoard.adapter.AdaptException;\r
+import org.simantics.databoard.adapter.RuntimeAdaptException;\r
+import org.simantics.databoard.binding.Binding;\r
+import org.simantics.databoard.binding.mutable.Variant;\r
+import org.simantics.databoard.type.Datatype;\r
+import org.simantics.graph.store.GraphStore;\r
+import org.simantics.graph.store.IdRes;\r
+import org.simantics.graph.store.IdentityStore;\r
+import org.simantics.graph.store.PathPattern;\r
+import org.simantics.graph.store.StatementStore;\r
+\r
+public class CompositeGraph implements IGraph {\r
+\r
+ ArrayList<GraphStore> fragments = new ArrayList<GraphStore>();\r
+ Paths paths;\r
+ \r
+ public CompositeGraph(Paths paths) {\r
+ this.paths = paths;\r
+ }\r
+\r
+ public void addFragment(GraphStore fragment) {\r
+ fragments.add(fragment);\r
+ }\r
+ \r
+ public void undoAddFragment() {\r
+ fragments.remove(fragments.size()-1);\r
+ }\r
+ \r
+ public void addFragments(Collection<GraphStore> fragments) {\r
+ fragments.addAll(fragments);\r
+ }\r
+ \r
+ protected void rawGetObjects(GraphStore fragment, int subject, Res predicate, Collection<Res> result) {\r
+ int predicateId;\r
+ if(predicate instanceof Path) {\r
+ predicateId = fragment.identities.pathToId((Path)predicate);\r
+ if(predicateId < 0)\r
+ return;\r
+ } \r
+ else {\r
+ IdRes idPredicate = (IdRes)predicate;\r
+ if(idPredicate.fragment != fragment)\r
+ return;\r
+ predicateId = idPredicate.id;\r
+ }\r
+ \r
+ TIntArrayList objects = fragment.statements.getObjects(subject, predicateId);\r
+ fragment.addIdsToResult(objects, result);\r
+ }\r
+ \r
+ private void rawGetObjects(Res subject, Res predicate, Collection<Res> result) {\r
+ if(subject instanceof Path) {\r
+ Path path = (Path)subject;\r
+ for(GraphStore fragment : fragments) {\r
+ int id = fragment.identities.pathToId(path);\r
+ if(id >= 0)\r
+ rawGetObjects(fragment, id, predicate, result);\r
+ }\r
+ }\r
+ else {\r
+ IdRes idRes = (IdRes)subject; \r
+ rawGetObjects(idRes.fragment, idRes.id, predicate, result);\r
+ }\r
+ }\r
+ \r
+ public boolean hasRawObjects(Res subject, Path predicate) {\r
+ ArrayList<Res> result = new ArrayList<Res>();\r
+ rawGetObjects(subject, predicate, result);\r
+ return !result.isEmpty();\r
+ }\r
+ \r
+ @Override\r
+ public Collection<Res> rawGetObjects(Res subject, Res predicate) {\r
+ ArrayList<Res> result = new ArrayList<Res>();\r
+ rawGetObjects(subject, predicate, result);\r
+ return result;\r
+ }\r
+ \r
+ @Override\r
+ public Res singleRawObject(Res subject, Res predicate) throws NoUniqueObjectException {\r
+ ArrayList<Res> result = new ArrayList<Res>(1);\r
+ rawGetObjects(subject, predicate, result);\r
+ if(result.size() != 1)\r
+ throw new NoUniqueObjectException("No unique objects (" + result.size()+ ") for " +\r
+ subject + " -> " + predicate); \r
+ return result.get(0);\r
+ }\r
+ \r
+ @Override\r
+ public Collection<Res> getTypes(Res subject) {\r
+ THashSet<Res> result = new THashSet<Res>(); \r
+ rawGetObjects(subject, paths.InstanceOf, result);\r
+ for(Res type : result.toArray(new Res[result.size()]))\r
+ collectSupertypes(type, result);\r
+ return result;\r
+ }\r
+ \r
+ public Collection<Res> getSupertypes(Res subject) {\r
+ THashSet<Res> result = new THashSet<Res>(); \r
+ result.add(subject); \r
+ collectSupertypes(subject, result);\r
+ return result;\r
+ }\r
+ \r
+ private void collectSupertypes(Res type, THashSet<Res> result) {\r
+ for(Res supertype : rawGetObjects(type, paths.Inherits))\r
+ if(result.add(supertype))\r
+ collectSupertypes(supertype, result); \r
+ }\r
+\r
+ @Override\r
+ public Collection<Res> getObjects(Res subject, Res predicate) {\r
+ ArrayList<Res> result = new ArrayList<Res>();\r
+ rawGetObjects(subject, predicate, result);\r
+ for(Res type : getTypes(subject)) \r
+ for(Res assertion : rawGetObjects(type, paths.Asserts)) {\r
+ Res pred = singleRawObject(assertion, paths.HasPredicate);\r
+ if(equals(pred, predicate))\r
+ result.add(singleRawObject(assertion, paths.HasObject));\r
+ }\r
+ return result;\r
+ }\r
+ \r
+ public Collection<Res> getAssertedObjects(Res subject, Path predicate) {\r
+ ArrayList<Res> result = new ArrayList<Res>();\r
+ for(Res type : getSupertypes(subject)) \r
+ for(Res assertion : rawGetObjects(type, paths.Asserts)) {\r
+ Res pred = singleRawObject(assertion, paths.HasPredicate);\r
+ if(equals(pred, predicate))\r
+ result.add(singleRawObject(assertion, paths.HasObject));\r
+ }\r
+ return result;\r
+ }\r
+ \r
+ private static boolean equals(Res r1, Res r2) {\r
+ return r1.equals(r2);\r
+ }\r
+ \r
+ interface ResourceProcedure {\r
+ public void execute(GraphStore fragment, int id);\r
+ }\r
+ \r
+ interface ResourceFunction<T> {\r
+ public T execute(GraphStore fragment, int id);\r
+ }\r
+ \r
+ public void forEachFragmentContaining(Res resource, ResourceProcedure proc) {\r
+ if(resource instanceof Path) {\r
+ Path path = (Path)resource;\r
+ for(GraphStore fragment : fragments) {\r
+ int id = fragment.identities.pathToId(path);\r
+ if(id >= 0)\r
+ proc.execute(fragment, id);\r
+ }\r
+ }\r
+ else {\r
+ IdRes res = (IdRes)resource;\r
+ proc.execute(res.fragment, res.id);\r
+ }\r
+ }\r
+ \r
+ public <T> T apply(Res resource, ResourceFunction <T> func) {\r
+ if(resource instanceof Path) {\r
+ Path path = (Path)resource;\r
+ for(GraphStore fragment : fragments) {\r
+ int id = fragment.identities.pathToId(path);\r
+ if(id >= 0) {\r
+ T value = func.execute(fragment, id);\r
+ if(value != null)\r
+ return value;\r
+ }\r
+ }\r
+ return null;\r
+ }\r
+ else {\r
+ IdRes res = (IdRes)resource;\r
+ return func.execute(res.fragment, res.id);\r
+ }\r
+ }\r
+ \r
+ private static ResourceFunction<Datatype> getDatatype = \r
+ new ResourceFunction<Datatype>() {\r
+ @Override\r
+ public Datatype execute(GraphStore fragment, int id) {\r
+ return fragment.values.getDatatypeValue(id);\r
+ } \r
+ };\r
+ \r
+ private static ResourceFunction<Variant> getValue = \r
+ new ResourceFunction<Variant>() {\r
+ @Override\r
+ public Variant execute(GraphStore fragment, int id) {\r
+ return fragment.values.getByteValue(id);\r
+ } \r
+ };\r
+ \r
+ THashMap<Res, Datatype> datatypeCache = new THashMap<Res, Datatype>();\r
+ \r
+ @Override\r
+ public Datatype getDatatype(Res resource) {\r
+ for(Res dt : getObjects(resource, paths.HasDatatype)) {\r
+ Datatype type = datatypeCache.get(dt);\r
+ if(type == null) {\r
+ type = apply(dt, getDatatype);\r
+ datatypeCache.put(dt, type);\r
+ }\r
+ return type;\r
+ }\r
+ return null;\r
+ }\r
+ \r
+ @Override\r
+ public Datatype getAssertedDatatype(Res resource) {\r
+ for(Res dt : getAssertedObjects(resource, paths.HasDatatype)) {\r
+ Datatype type = datatypeCache.get(dt);\r
+ if(type == null) {\r
+ type = apply(dt, getDatatype);\r
+ datatypeCache.put(dt, type);\r
+ }\r
+ return type;\r
+ }\r
+ return null;\r
+ }\r
+ \r
+ @Override\r
+ public Variant getValue(Res resource) {\r
+ return apply(resource, getValue);\r
+ }\r
+ \r
+ @Override\r
+ public Object getValue(Res resource, Binding binding) throws NoValueException {\r
+ Variant value = getValue(resource);\r
+ if(value == null)\r
+ throw new NoValueException();\r
+ try {\r
+ return value.getValue(binding);\r
+ } catch (AdaptException e) {\r
+ throw new RuntimeAdaptException(e);\r
+ }\r
+ }\r
+ \r
+ @Override\r
+ public void setValue(Res resource, Object value, Binding binding) {\r
+ final Variant variant = new Variant(binding, value);\r
+ apply(resource, new ResourceFunction<Object>() {\r
+ @Override\r
+ public Object execute(GraphStore fragment, int id) {\r
+ fragment.values.setValue(id, variant);\r
+ return null;\r
+ } \r
+ });\r
+ }\r
+ \r
+ /**\r
+ * Tells in how many fragments the resource occurs.\r
+ */\r
+ public int countOccurences(Res resource) {\r
+ if(resource instanceof IdRes)\r
+ return 1;\r
+ else if(resource instanceof Path) {\r
+ Path path = (Path)resource;\r
+ int count = 0;\r
+ for(GraphStore fragment : fragments)\r
+ if(fragment.identities.contains(path))\r
+ ++count;\r
+ return count;\r
+ }\r
+ else\r
+ return 0;\r
+ }\r
+ \r
+ private void collectSubtypes(THashSet<Res> types, Res type) {\r
+ if(types.add(type)) \r
+ for(Res subtype : rawGetObjects(type, paths.SupertypeOf))\r
+ collectSubtypes(types, subtype);\r
+ }\r
+ \r
+ @Override\r
+ public Collection<Res> getInstances(Res supertype) {\r
+ THashSet<Res> types = new THashSet<Res>(); \r
+ collectSubtypes(types, supertype);\r
+ \r
+ ArrayList<Res> result = new ArrayList<Res>();\r
+ fragmentLoop:\r
+ for(GraphStore fragment : fragments) {\r
+ IdentityStore identities = fragment.identities;\r
+ StatementStore statements = fragment.statements;\r
+ \r
+ TIntHashSet ids = new TIntHashSet(types.size());\r
+ for(Res type : types) {\r
+ if(type instanceof Path) {\r
+ int id = identities.pathToId((Path)type);\r
+ if(id >= 0)\r
+ ids.add(id);\r
+ }\r
+ else {\r
+ IdRes idRes = (IdRes)type;\r
+ if(idRes.fragment == fragment)\r
+ ids.add(idRes.id);\r
+ }\r
+ }\r
+ if(ids.isEmpty())\r
+ continue;\r
+ \r
+ int instanceOfId = identities.pathToId(paths.InstanceOf);\r
+ if(instanceOfId < 0)\r
+ continue;\r
+ \r
+ int resourceCount = identities.getResourceCount();\r
+ nextResource:\r
+ for(int i=0;i<resourceCount;++i)\r
+ for(int typeId : statements.getObjects(i, instanceOfId).toArray())\r
+ if(ids.contains(typeId)) {\r
+ result.add(fragment.idToRes(i));\r
+ continue nextResource;\r
+ }\r
+ }\r
+ \r
+ return result;\r
+ }\r
+\r
+ @Override\r
+ public Collection<Res> getChildren(Res res) {\r
+ if(res instanceof Path) {\r
+ THashSet<Res> result = new THashSet<Res>();\r
+ for(GraphStore store : fragments) {\r
+ IdentityStore ids = store.identities;\r
+ int id = ids.pathToId((Path)res);\r
+ if(id >= 0) {\r
+ for(int child : ids.getChildren(id))\r
+ result.add(store.idToRes(child));\r
+ }\r
+ }\r
+ return result;\r
+ }\r
+ else\r
+ return Collections.<Res>emptyList();\r
+ }\r
+ \r
+ public Collection<Path> searchByPattern(String pattern) {\r
+ THashSet<Path> result = new THashSet<Path>();\r
+ PathPattern pathPattern = PathPattern.compile(pattern);\r
+ for(GraphStore store : fragments)\r
+ pathPattern.search(store.identities, result);\r
+ return result;\r
+ }\r
+ \r
+ @Override\r
+ public Paths getPaths() {\r
+ return paths;\r
+ }\r
+}\r