X-Git-Url: https://gerrit.simantics.org/r/gitweb?p=simantics%2Fplatform.git;a=blobdiff_plain;f=bundles%2Forg.simantics.db.procore%2Fsrc%2Forg%2Fsimantics%2Fdb%2Fprocore%2Fcluster%2FCompleteTableSmall.java;fp=bundles%2Forg.simantics.db.procore%2Fsrc%2Forg%2Fsimantics%2Fdb%2Fprocore%2Fcluster%2FCompleteTableSmall.java;h=c44558cbbf986911d5ee8b640249548e86a939ad;hp=0000000000000000000000000000000000000000;hb=969bd23cab98a79ca9101af33334000879fb60c5;hpb=866dba5cd5a3929bbeae85991796acb212338a08

diff --git a/bundles/org.simantics.db.procore/src/org/simantics/db/procore/cluster/CompleteTableSmall.java b/bundles/org.simantics.db.procore/src/org/simantics/db/procore/cluster/CompleteTableSmall.java
new file mode 100644
index 000000000..c44558cbb
--- /dev/null
+++ b/bundles/org.simantics.db.procore/src/org/simantics/db/procore/cluster/CompleteTableSmall.java
@@ -0,0 +1,93 @@
+package org.simantics.db.procore.cluster;
+
+import gnu.trove.set.hash.TIntHashSet;
+
+import org.simantics.db.exception.DatabaseException;
+import org.simantics.db.impl.ClusterI;
+import org.simantics.db.impl.ClusterSupport;
+import org.simantics.db.impl.ClusterTraitsBase;
+import org.simantics.db.impl.Modifier;
+import org.simantics.db.impl.TableSizeListener;
+
+public class CompleteTableSmall extends CompleteTable {
+    public CompleteTableSmall(TableSizeListener sizeListener, int[] header, int headerBase) {
+        super(sizeListener, header, headerBase);
+    }
+    public CompleteTableSmall(TableSizeListener sizeListener, int[] header, int headerBase, int[] ints) {
+        super(sizeListener, header, headerBase, ints);
+    }
+
+    @Override
+    public <Context> boolean foreachPredicate(int setIndex,
+            ClusterI.PredicateProcedure<Context> procedure,
+            Context context, ClusterSupport support, Modifier modifier)
+    throws DatabaseException {
+        ForeachPredicate<Context> t = new ForeachPredicate<Context>(procedure, support, modifier);
+        return foreachComplete(setIndex, t, context, null, null);
+    }
+
+    @Override
+    public <Context> boolean foreachObject(int setIndex,
+            ClusterI.ObjectProcedure<Context> procedure,
+            Context context, ClusterSupport support, Modifier modifier,
+            ClusterI.CompleteTypeEnum completeType)
+    throws DatabaseException {
+        ForeachObject<Context> t = new ForeachObject<Context>
+        (procedure, support, modifier, completeType);
+        return foreachComplete(setIndex, t, context, null, null);
+    }
+    
+    private static final class ForeachPredicate<Context>
+    implements ClusterI.ObjectProcedure<Context> {
+        private TIntHashSet completeTypes = new TIntHashSet();
+        private ClusterI.PredicateProcedure<Context> procedure; 
+        public ForeachPredicate(ClusterI.PredicateProcedure<Context>
+            procedure, ClusterSupport support, Modifier modifier) {
+            this.procedure = procedure;
+        }
+        @Override
+        public boolean execute(Context context, int completeRef) {
+        	ClusterI.CompleteTypeEnum completeType = ClusterTraitsSmall.completeRefAndTypeGetType(completeRef);
+            if (!completeTypes.contains(completeType.getValue())) {
+                completeTypes.add(completeType.getValue());
+                try {
+                    int pKey = ClusterTraitsBase.getCompleteTypeResourceKeyFromEnum(completeType);
+                    if (procedure.execute(context, pKey, 0))
+                        return true; // loop broken by procedure
+                } catch (DatabaseException e) {
+                    e.printStackTrace();
+                    return false;
+                }
+            }
+            return false; // Continue looping.
+        }
+        
+    }
+    private static class ForeachObject<Context>
+    implements ClusterI.ObjectProcedure<Context> {
+        private ClusterI.ObjectProcedure<Context> procedure; 
+        private Modifier modifier;
+        private ClusterI.CompleteTypeEnum completeType;
+        public ForeachObject(ClusterI.ObjectProcedure<Context>
+            procedure, ClusterSupport support, Modifier modifier, ClusterI.CompleteTypeEnum completeType) {
+            this.procedure = procedure;
+            this.modifier = modifier;
+            this.completeType = completeType;
+        }
+        @Override
+        public boolean execute(Context context, int completeRef) throws DatabaseException {
+        	ClusterI.CompleteTypeEnum completeType2 = ClusterTraitsSmall.completeRefAndTypeGetType(completeRef);
+            if (completeType == completeType2) { // same predicate
+                int objectRef = ClusterTraitsSmall.completeRefAndTypeGetRef(completeRef);
+                int externalRef;
+                if (null == modifier)
+                    externalRef = objectRef;
+                else
+                    externalRef = modifier.execute(objectRef);
+                return procedure.execute(context, externalRef);
+            }
+            return false; // Continue looping.
+        }
+        
+    }
+}