1 package org.simantics.db.layer0.util;
3 import java.util.ArrayList;
4 import java.util.Collection;
5 import java.util.HashMap;
8 import org.simantics.databoard.Bindings;
9 import org.simantics.db.ReadGraph;
10 import org.simantics.db.Resource;
11 import org.simantics.db.common.NamedResource;
12 import org.simantics.db.common.request.PossibleIndexRoot;
13 import org.simantics.db.common.utils.CommonDBUtils;
14 import org.simantics.db.exception.DatabaseException;
15 import org.simantics.db.layer0.adapter.SubgraphExtent.ExtentStatus;
16 import org.simantics.db.layer0.util.TransferableGraphConfiguration2.RootSpec;
17 import org.simantics.layer0.Layer0;
19 public class TGConfigurer {
21 final private ReadGraph graph;
22 final Collection<RootSpec> roots = new ArrayList<RootSpec>();
23 final Map<Resource, ExtentStatus> preStatus = new HashMap<Resource, ExtentStatus>();
24 final boolean ignoreVirtualResources;
25 final boolean validate;
27 public TGConfigurer(ReadGraph graph, boolean ignoreVirtualResources, boolean validate) {
29 this.ignoreVirtualResources = ignoreVirtualResources;
30 this.validate = validate;
33 public TGConfigurer(ReadGraph graph, boolean ignoreVirtualResources) {
34 this(graph, ignoreVirtualResources, true);
37 public TGConfigurer exclusions(Collection<Resource> exclusions) {
38 for(Resource r : exclusions) {
39 preStatus.put(r, ExtentStatus.EXCLUDED);
44 public TGConfigurer roots(Collection<RootSpec> rs) throws DatabaseException {
45 Layer0 L0 = Layer0.getInstance(graph);
46 for(RootSpec root : rs) {
47 Resource nameResource = graph.getPossibleObject(root.resource, L0.HasName);
48 if(nameResource != null) preStatus.put(nameResource, ExtentStatus.EXCLUDED);
49 // String name = graph.getValue(nameResource, Bindings.STRING);
50 roots.add(new RootSpec(root.resource, root.name, root.internal, root.type));
51 preStatus.put(root.resource, ExtentStatus.INTERNAL);
56 public TGConfigurer roots2(Collection<Resource> rs) throws DatabaseException {
57 Layer0 L0 = Layer0.getInstance(graph);
58 for(Resource root : rs) {
59 Resource nameResource = graph.getPossibleObject(root, L0.HasName);
60 String name = "<no name>";
61 if(nameResource != null) {
62 name = graph.getPossibleValue(nameResource, Bindings.STRING);
63 if(name == null) name = "<no name>";
64 preStatus.put(nameResource, ExtentStatus.EXCLUDED);
66 roots.add(new RootSpec(root, name, true));
67 preStatus.put(root, ExtentStatus.INTERNAL);
72 public TGConfigurer roots3(Collection<NamedResource> rs) throws DatabaseException {
73 Layer0 L0 = Layer0.getInstance(graph);
74 for(NamedResource root : rs) {
75 Resource nameResource = graph.getPossibleObject(root.getResource(), L0.HasName);
76 // String name = graph.getValue(nameResource, Bindings.STRING);
77 roots.add(new RootSpec(root.getResource(), root.getName(), true));
78 preStatus.put(root.getResource(), ExtentStatus.INTERNAL);
79 preStatus.put(nameResource, ExtentStatus.EXCLUDED);
84 public TransferableGraphConfiguration2 create() throws DatabaseException {
88 // -Exclude root library if not explicitly included
89 if (!roots.contains(graph.getRootLibrary()))
90 preStatus.put(graph.getRootLibrary(), ExtentStatus.EXCLUDED);
92 // -Exclude owners if not explicitly defined
93 Layer0 L0 = Layer0.getInstance(graph);
94 ArrayList<Resource> rootList = new ArrayList<Resource>();
95 for (RootSpec root : roots) {
97 rootList.add(root.resource);
98 for (Resource owner : graph.getObjects(root.resource, L0.IsOwnedBy)) {
99 ExtentStatus ownerStatus = preStatus.get(owner);
100 if(ownerStatus == null)
101 preStatus.put(owner, ExtentStatus.EXCLUDED);
104 preStatus.put(root.resource, ExtentStatus.EXTERNAL);
108 Resource owner = CommonDBUtils.getNearestOwner(graph, rootList);
110 Resource indexRoot = graph.syncRequest(new PossibleIndexRoot(owner));
111 return new TransferableGraphConfiguration2(indexRoot, roots, preStatus, ignoreVirtualResources, validate);
113 return new TransferableGraphConfiguration2(null, roots, preStatus, ignoreVirtualResources, validate);