bundles/org.simantics.db.layer0/src/org/simantics/db/layer0/util/TGConfigurer.java

   1 package org.simantics.db.layer0.util;
   2
   3 import java.util.ArrayList;
   4 import java.util.Collection;
   5 import java.util.HashMap;
   6 import java.util.Map;
   7
   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;
  18
  19 public class TGConfigurer {
  20
  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;
  26
  27         public TGConfigurer(ReadGraph graph, boolean ignoreVirtualResources, boolean validate) {
  28                 this.graph = graph;
  29                 this.ignoreVirtualResources = ignoreVirtualResources;
  30                 this.validate = validate;
  31         }
  32
  33         public TGConfigurer(ReadGraph graph, boolean ignoreVirtualResources) {
  34                 this(graph, ignoreVirtualResources, true);
  35         }
  36
  37         public TGConfigurer exclusions(Collection<Resource> exclusions) {
  38                 for(Resource r : exclusions) {
  39                         preStatus.put(r, ExtentStatus.EXCLUDED);
  40                 }
  41                 return this;
  42         }
  43
  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);
  52                 }
  53                 return this;
  54         }
  55
  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);
  65                         }
  66                         roots.add(new RootSpec(root, name, true));
  67                         preStatus.put(root, ExtentStatus.INTERNAL);
  68                 }
  69                 return this;
  70         }
  71
  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);
  80                 }
  81                 return this;
  82         }
  83
  84         public TransferableGraphConfiguration2 create() throws DatabaseException {
  85
  86                 // Final preparation
  87
  88                 // -Exclude root library if not explicitly included
  89                 if (!roots.contains(graph.getRootLibrary()))
  90                         preStatus.put(graph.getRootLibrary(), ExtentStatus.EXCLUDED);
  91
  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) {
  96                         if(root.internal) {
  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);
 102                                 }
 103                         } else {
 104                                 preStatus.put(root.resource, ExtentStatus.EXTERNAL);
 105                         }
 106                 }
 107
 108                 Resource owner = CommonDBUtils.getNearestOwner(graph, rootList);
 109                 if(owner != null) {
 110                         Resource indexRoot = graph.syncRequest(new PossibleIndexRoot(owner));
 111                         return new TransferableGraphConfiguration2(indexRoot, roots, preStatus, ignoreVirtualResources, validate);
 112                 } else {
 113                         return new TransferableGraphConfiguration2(null, roots, preStatus, ignoreVirtualResources, validate);
 114                 }
 115
 116         }
 117
 118 }