X-Git-Url: https://gerrit.simantics.org/r/gitweb?p=simantics%2Fplatform.git;a=blobdiff_plain;f=bundles%2Forg.simantics.graph%2Fsrc%2Forg%2Fsimantics%2Fgraph%2Fstore%2FStatementStore.java;h=02dfddde81085d06727089863185c4e10345d54f;hp=96b6b1c725ee61a073ac2b7cebf949ff94def4bc;hb=c125a1755cc7c4a6241c3c5bf841c3db0ff2d658;hpb=969bd23cab98a79ca9101af33334000879fb60c5

diff --git a/bundles/org.simantics.graph/src/org/simantics/graph/store/StatementStore.java b/bundles/org.simantics.graph/src/org/simantics/graph/store/StatementStore.java
index 96b6b1c72..02dfddde8 100644
--- a/bundles/org.simantics.graph/src/org/simantics/graph/store/StatementStore.java
+++ b/bundles/org.simantics.graph/src/org/simantics/graph/store/StatementStore.java
@@ -1,5 +1,7 @@
 package org.simantics.graph.store;
 
+import java.util.ArrayList;
+
 import gnu.trove.list.array.TIntArrayList;
 import gnu.trove.map.hash.TIntIntHashMap;
 import gnu.trove.map.hash.TIntObjectHashMap;
@@ -10,7 +12,7 @@ import gnu.trove.set.hash.TIntHashSet;
 
 /**
  * Statement store indexes a set of statements. 
- * @author Hannu Niemistö
+ * @author Hannu Niemist�
  */
 public class StatementStore implements IStore {
 	
@@ -287,5 +289,64 @@ public class StatementStore implements IStore {
 			}
 		});
 		return statements.toArray();
+	}
+
+	private static class CollisionSubjectProcedure implements TIntObjectProcedure<TIntObjectHashMap<TIntArrayList>> {
+	    CollisionPredicateProcedure predicateProcedure;
+
+	    public CollisionSubjectProcedure(CollisionPredicateProcedure predicateProcedure) {
+	        this.predicateProcedure = predicateProcedure;
+	    }
+
+	    @Override
+	    public boolean execute(int subject, TIntObjectHashMap<TIntArrayList> predicateObjectMap) {
+	        predicateProcedure.subject = subject;
+	        predicateObjectMap.forEachEntry(predicateProcedure);
+	        return true;
+	    }
+
+	}
+
+	private static class CollisionPredicateProcedure implements TIntObjectProcedure<TIntArrayList> {
+        ArrayList<StatementCollision> collisions;
+	    int subject;
+	    
+	    public CollisionPredicateProcedure(ArrayList<StatementCollision> collisions) {
+            this.collisions = collisions;
+        }
+	    
+        @Override
+	    public boolean execute(int predicate, TIntArrayList objects) {
+	        if(objects.size() > 1) {
+	            objects.sort();
+	            int oldObject = objects.get(0);
+	            int collisionCount = 1;
+	            for(int i=1;i<objects.size();++i) {
+	                int curObject = objects.get(i);
+	                if(curObject == oldObject) {
+	                    ++collisionCount;
+	                }
+	                else {
+	                    if(collisionCount > 1) {
+	                        collisions.add(new StatementCollision(subject, predicate, oldObject, collisionCount));
+	                        collisionCount = 1;
+	                    }
+	                    oldObject = curObject;
+	                }
+	            }
+                if(collisionCount > 1)
+                    collisions.add(new StatementCollision(subject, predicate, oldObject, collisionCount));
+	        }
+	        return true;
+	    }
+
+	}
+
+	public ArrayList<StatementCollision> getCollisions() {
+	    ArrayList<StatementCollision> collisions = new ArrayList<StatementCollision>();
+	    CollisionPredicateProcedure predicateProcedure = new CollisionPredicateProcedure(collisions);
+	    CollisionSubjectProcedure subjectProcedure = new CollisionSubjectProcedure(predicateProcedure);
+	    statements.forEachEntry(subjectProcedure);
+	    return collisions;
 	}	
 }