bundles/org.simantics.db.common/src/org/simantics/db/common/TransferableGraphRequests.java

   1 package org.simantics.db.common;\r
   2 \r
   3 import java.util.HashSet;\r
   4 import java.util.Set;\r
   5 \r
   6 import org.simantics.databoard.adapter.AdaptException;\r
   7 import org.simantics.databoard.binding.Binding;\r
   8 import org.simantics.db.ReadGraph;\r
   9 import org.simantics.db.Resource;\r
  10 import org.simantics.db.exception.DatabaseException;\r
  11 import org.simantics.graph.representation.TransferableGraph1;\r
  12 import org.simantics.graph.representation.TransferableGraphUtils;\r
  13 import org.simantics.graph.representation.Value;\r
  14 import org.simantics.layer0.Layer0;\r
  15 \r
  16 public class TransferableGraphRequests {\r
  17         \r
  18         private TransferableGraph1 tg;\r
  19         private ReadGraph graph;\r
  20         \r
  21         public TransferableGraphRequests(TransferableGraph1 tg, ReadGraph graph) {\r
  22                 this.tg = tg;\r
  23                 this.graph = graph;\r
  24         }\r
  25 \r
  26         public Resource getExternal(int id) throws DatabaseException {\r
  27                 String uri = TransferableGraphUtils.getURI(tg, id);\r
  28                 return graph.getPossibleResource(uri);\r
  29         }\r
  30         \r
  31         public Set<Integer> getObjects(int subject, Resource relation) throws DatabaseException {\r
  32 \r
  33                 Set<Integer> result = new HashSet<Integer>();\r
  34                 for(int i=0;i<tg.statements.length;i+=4) {\r
  35                         if(tg.statements[i] == subject) {\r
  36                                 Resource predicate = getExternal(tg.statements[i+1]);\r
  37                                 if(predicate != null)\r
  38                                         if(graph.isSubrelationOf(predicate, relation)) result.add(tg.statements[i+3]); \r
  39                         }\r
  40                 }\r
  41                 return result;\r
  42                 \r
  43         }\r
  44 \r
  45         public Set<Integer> getSubjects(int object, Resource inverseRelation) throws DatabaseException {\r
  46 \r
  47                 Set<Integer> result = new HashSet<Integer>();\r
  48                 for(int i=0;i<tg.statements.length;i+=4) {\r
  49                         if(tg.statements[i+3] == object) {\r
  50                                 Resource predicate = getExternal(tg.statements[i+2]);\r
  51                                 if(predicate != null)\r
  52                                         if(graph.isSubrelationOf(predicate, inverseRelation)) result.add(tg.statements[i]); \r
  53                         }\r
  54                 }\r
  55                 return result;\r
  56                 \r
  57         }\r
  58 \r
  59         public int getPossibleObject(int subject, Resource relation) throws DatabaseException {\r
  60                 Set<Integer> objects = getObjects(subject, relation);\r
  61                 if(objects.size() == 1) return objects.iterator().next();\r
  62                 return -1;\r
  63         }\r
  64 \r
  65         public int getSingleObject(int subject, Resource relation) throws DatabaseException {\r
  66 \r
  67                 Set<Integer> objects = getObjects(subject, relation);\r
  68                 if(objects.size() == 1) return objects.iterator().next();\r
  69                 else if (objects.size() == 0) throw new DatabaseException("No objects for subject " + subject + " and relation " + graph.getURI(relation));\r
  70                 else throw new DatabaseException("Multiple (" + objects.size() + ") objects for subject " + subject + " and relation " + graph.getURI(relation));\r
  71                 \r
  72         }\r
  73 \r
  74         public int getPossibleSubject(int object, Resource inverseRelation) throws DatabaseException {\r
  75                 Set<Integer> subjects = getSubjects(object, inverseRelation);\r
  76                 if(subjects.size() == 1) return subjects.iterator().next();\r
  77                 return -1;\r
  78         }\r
  79 \r
  80         public int getSingleSubject(int object, Resource inverseRelation) throws DatabaseException {\r
  81 \r
  82                 Set<Integer> subjects = getObjects(object, inverseRelation);\r
  83                 if(subjects.size() == 1) return subjects.iterator().next();\r
  84                 else if (subjects.size() == 0) throw new DatabaseException("No subjects for object " + object + " and relation " + graph.getURI(inverseRelation));\r
  85                 else throw new DatabaseException("Multiple (" + subjects.size() + ") subjects for object " + object + " and relation " + graph.getURI(inverseRelation));\r
  86                 \r
  87         }\r
  88 \r
  89         public <T> T getRelatedValue(int subject, Resource relation, Binding binding) throws DatabaseException {\r
  90                 \r
  91                 int object = getSingleObject(subject, relation);\r
  92                 return getValue(object, binding);\r
  93                 \r
  94         }\r
  95 \r
  96         public <T> T getPossibleRelatedValue(int subject, Resource relation, Binding binding) throws DatabaseException {\r
  97                 int object = getPossibleObject(subject, relation);\r
  98                 if (object == -1)\r
  99                         return null;\r
 100                 return getPossibleValue(object, binding);\r
 101         }\r
 102 \r
 103         @SuppressWarnings("unchecked")\r
 104         public <T> T getValue(int subject, Binding binding) throws DatabaseException {\r
 105 \r
 106                 Value value = TransferableGraphUtils.findValue(tg, subject);\r
 107                 if(value == null) throw new DatabaseException("No value for subject " + subject);\r
 108                 try {\r
 109                         return (T)value.value.getValue(binding);\r
 110                 } catch (AdaptException e) {\r
 111                         throw new DatabaseException(e);\r
 112                 }\r
 113                 \r
 114         }\r
 115 \r
 116         @SuppressWarnings("unchecked")\r
 117         public <T> T getPossibleValue(int subject, Binding binding) {\r
 118                 Value value = TransferableGraphUtils.findValue(tg, subject);\r
 119                 if(value == null) return null;\r
 120                 try {\r
 121                         return (T)value.value.getValue(binding);\r
 122                 } catch (AdaptException e) {\r
 123                         return null;\r
 124                 }\r
 125         }\r
 126 \r
 127         public boolean isInstanceOf(int subject, Resource type) throws DatabaseException {\r
 128 \r
 129                 Layer0 L0 = Layer0.getInstance(graph);\r
 130                 for(int t : getObjects(subject, L0.InstanceOf)) {\r
 131                         Resource tr = getExternal(t);\r
 132                         if(tr != null)\r
 133                                 if(graph.isInheritedFrom(tr, type)) return true;\r
 134                 }\r
 135                 \r
 136                 return false;\r
 137                 \r
 138         }\r
 139         \r
 140 \r
 141         public String formatResource(int r) throws DatabaseException {\r
 142                 \r
 143                 String result = "" + r;\r
 144                 String uri = TransferableGraphUtils.getURI(tg, r);\r
 145                 if(!uri.isEmpty()) result = uri;\r
 146                 \r
 147                 Value value = TransferableGraphUtils.findValue(tg, r);\r
 148                 if(value != null) result += " (value " + value.value + ")";\r
 149                 \r
 150                 return result;\r
 151                 \r
 152         }\r
 153         \r
 154         public String listStatements() throws DatabaseException {\r
 155                 \r
 156                 StringBuilder b = new StringBuilder();\r
 157         for(int i=0;i<tg.statements.length;i+=4) {\r
 158                 int subject = tg.statements[i];\r
 159                 int predicate = tg.statements[i+1];\r
 160                 int object = tg.statements[i+3];\r
 161                 b.append(formatResource(subject) + " " + formatResource(predicate) + " " + formatResource(object) + "\n");\r
 162         }\r
 163         return b.toString();\r
 164                 \r
 165         }\r
 166         \r
 167         public ReadGraph getGraph() {\r
 168                 return graph;\r
 169         }\r
 170         \r
 171 }\r