bundles/org.simantics.db.impl/src/org/simantics/db/impl/query/PossibleSuperRelation.java

   1 /*******************************************************************************\r
   2  * Copyright (c) 2007, 2010 Association for Decentralized Information Management\r
   3  * in Industry THTH ry.\r
   4  * All rights reserved. This program and the accompanying materials\r
   5  * are made available under the terms of the Eclipse Public License v1.0\r
   6  * which accompanies this distribution, and is available at\r
   7  * http://www.eclipse.org/legal/epl-v10.html\r
   8  *\r
   9  * Contributors:\r
  10  *     VTT Technical Research Centre of Finland - initial API and implementation\r
  11  *******************************************************************************/\r
  12 package org.simantics.db.impl.query;\r
  13 \r
  14 import gnu.trove.procedure.TIntProcedure;\r
  15 import gnu.trove.set.hash.TIntHashSet;\r
  16 \r
  17 import org.simantics.db.Resource;\r
  18 import org.simantics.db.impl.graph.ReadGraphImpl;\r
  19 import org.simantics.db.procedure.AsyncProcedure;\r
  20 import org.simantics.db.procedure.ListenerBase;\r
  21 \r
  22 final public class PossibleSuperRelation extends UnaryQuery<IntProcedure> {\r
  23 \r
  24         private PossibleSuperRelation(final int resource) {\r
  25                 super(resource);\r
  26         }\r
  27 \r
  28         final public static void queryEach(ReadGraphImpl graph, final int r, final QueryProcessor provider, final CacheEntry parent, final ListenerBase listener, final AsyncProcedure<Resource> procedure) {\r
  29 \r
  30                 int result = provider.querySupport.getSingleSuperrelation(r);\r
  31                 if(result != 0) {\r
  32                         procedure.execute(graph, provider.querySupport.getResource(result));\r
  33                 } else {\r
  34                         procedure.execute(graph, null);\r
  35                 }\r
  36 \r
  37         }\r
  38 \r
  39         @Override\r
  40         public UnaryQuery<IntProcedure> getEntry(QueryProcessor provider) {\r
  41                 return null;\r
  42         }\r
  43 \r
  44         @Override\r
  45         public void putEntry(QueryProcessor provider) {\r
  46         }\r
  47 \r
  48         @Override\r
  49         final public void removeEntry(QueryProcessor provider) {\r
  50         }\r
  51 \r
  52         class Koss {\r
  53 \r
  54                 private TIntHashSet set = null;\r
  55                 public int single = 0;\r
  56 \r
  57                 public boolean add(int val) {\r
  58                         if(single == val) return false;\r
  59                         if(single == 0) {\r
  60                                 single = val;\r
  61                                 return true;\r
  62                         }\r
  63                         if(set == null) set = new TIntHashSet(4);\r
  64                         return set.add(val);\r
  65                 }\r
  66 \r
  67                 public int size() {\r
  68 \r
  69                         if(single == 0) return 0;\r
  70                         if(set == null) return 1;\r
  71                         return set.size() + 1;\r
  72 \r
  73                 }\r
  74 \r
  75                 //        public int[] toArray() {\r
  76                         //            \r
  77                         //            int[] result = Arrays.copyOf(set.toArray(), set.size() + 1);\r
  78                         //            result[set.size()] = single;\r
  79                         //            return result;\r
  80                         //            \r
  81                         //        }\r
  82                 //        \r
  83 \r
  84                 public void forEach(TIntProcedure proc) {\r
  85                         if(single > 0) proc.execute(single);\r
  86                         if(set != null) set.forEach(proc);\r
  87                 }\r
  88 \r
  89         }\r
  90 \r
  91         @Override\r
  92         public Object computeForEach(final ReadGraphImpl graph, final QueryProcessor provider, final IntProcedure procedure, final boolean store) {\r
  93 \r
  94                 provider.querySupport.ensureLoaded(graph, id);\r
  95                 int single = provider.querySupport.getSingleSuperrelation(id);\r
  96                 if(single > 0) {\r
  97                         procedure.execute(graph, single);\r
  98                         procedure.finished(graph);\r
  99                         return single;\r
 100                 }\r
 101 \r
 102                 final int subrelationOf = provider.getSubrelationOf();\r
 103 \r
 104                 final IntSet result = new IntSet(provider.querySupport);\r
 105 \r
 106                 final class DirectProcedure extends Koss implements IntProcedure, TIntProcedure {\r
 107                         @Override\r
 108                         final public boolean execute(int r) {\r
 109                                 result.add(r);\r
 110                                 return true;\r
 111                         }\r
 112                         @Override\r
 113                         final public void execute(ReadGraphImpl graph, int r) {\r
 114                                 if(single == 0) {\r
 115                                         single = r;\r
 116                                         return;\r
 117                                 }\r
 118                                 add(r);\r
 119                         }\r
 120                         @Override\r
 121                         public void finished(ReadGraphImpl graph) {\r
 122                         }\r
 123                         @Override\r
 124                         public void exception(ReadGraphImpl graph, Throwable t) {\r
 125                                 throw new Error("Errors are not supported.", t);\r
 126                         }\r
 127 \r
 128                 }\r
 129 \r
 130                 final DirectProcedure directProc = new DirectProcedure();\r
 131 \r
 132                 provider.querySupport.getObjects(graph, id, subrelationOf, directProc);\r
 133 \r
 134                 int size = directProc.size();\r
 135 \r
 136                 if(size == 0) {\r
 137 \r
 138                         procedure.finished(graph);\r
 139 \r
 140                 } else if (size == 1) {\r
 141 \r
 142                         procedure.execute(graph, directProc.single);\r
 143                         procedure.finished(graph);\r
 144 \r
 145                 } else {\r
 146 \r
 147                         directProc.forEach(new TIntProcedure() {\r
 148 \r
 149                                 @Override\r
 150                                 public boolean execute(int arg0) {\r
 151 \r
 152                                         procedure.execute(graph, arg0);\r
 153                                         return true;\r
 154 \r
 155                                 }\r
 156 \r
 157                         });\r
 158 \r
 159                         procedure.finished(graph);\r
 160 \r
 161                 }\r
 162                 \r
 163                 return result;\r
 164 \r
 165         }\r
 166 \r
 167         @Override\r
 168         public String toString() {\r
 169                 return "SuperRelations2[" + id + "]";\r
 170         }\r
 171 \r
 172         @Override\r
 173         public Object performFromCache(ReadGraphImpl graph, QueryProcessor provider, IntProcedure procedure) {\r
 174                 throw new UnsupportedOperationException();\r
 175         }\r
 176 \r
 177         @Override\r
 178         public void recompute(ReadGraphImpl graph, QueryProcessor provider) {\r
 179 \r
 180         }\r
 181 \r
 182 }\r