1 files changed, 51 insertions, 0 deletions
diff --git a/Solvers/VIATRA-Solver/hu.bme.mit.inf.dslreasoner.viatrasolver.reasoner/src/hu/bme/mit/inf/dslreasoner/viatrasolver/reasoner/dse/BalancedActivationSelector.xtend b/Solvers/VIATRA-Solver/hu.bme.mit.inf.dslreasoner.viatrasolver.reasoner/src/hu/bme/mit/inf/dslreasoner/viatrasolver/reasoner/dse/BalancedActivationSelector.xtend
new file mode 100644
index 00000000..2df9957b
--- /dev/null
+++ b/Solvers/VIATRA-Solver/hu.bme.mit.inf.dslreasoner.viatrasolver.reasoner/src/hu/bme/mit/inf/dslreasoner/viatrasolver/reasoner/dse/BalancedActivationSelector.xtend
@@ -0,0 +1,51 @@
+package hu.bme.mit.inf.dslreasoner.viatrasolver.reasoner.dse
+import java.util.Collection
+import java.util.HashMap
+import java.util.Map
+import java.util.List
+import java.util.Random
+import java.util.ArrayList
+import java.util.LinkedList
+import java.util.Collections
+class BalancedActivationSelector extends ActivationSelector{
+        val Random r = new Random
+        
+        new(Random r) {
+                super(r)
+        }
+        
+        override protected internalRandomizationOfActivationIDs(Collection<Object> activationIDs) {
+                val Map<String,List<Object>> urns = new HashMap
+                val res = new ArrayList(activationIDs.size)
+                for(activationID : activationIDs) {
+                        val pair = activationID as Pair<String,? extends Object>
+                        val name = pair.key
+                        val selectedUrn = urns.get(name)
+                        if(selectedUrn!==null) {
+                                selectedUrn.add(activationID)
+                        } else {
+                                val collection = new LinkedList
+                                collection.add(activationID)
+                                urns.put(name,collection)
+                        }
+                }
+                
+                for(list:urns.values) {
+                        Collections.shuffle(list,r)
+                }
+                
+                while(!urns.empty) {
+                        val randomEntry = urns.entrySet.get(r.nextInt(urns.size))
+                        val list = randomEntry.value
+                        val removedLast = list.remove(list.size-1)
+                        res.add(removedLast)
+                        if(list.empty) {
+                                urns.remove(randomEntry.key)
+                        }
+                }
+                return res
+        }
+        
+}
+\ No newline at end of file

diff --git a/Solvers/VIATRA-Solver/hu.bme.mit.inf.dslreasoner.viatrasolver.reasoner/src/hu/bme/mit/inf/dslreasoner/viatrasolver/reasoner/dse/BalancedActivationSelector.xtend b/Solvers/VIATRA-Solver/hu.bme.mit.inf.dslreasoner.viatrasolver.reasoner/src/hu/bme/mit/inf/dslreasoner/viatrasolver/reasoner/dse/BalancedActivationSelector.xtend new file mode 100644 index 00000000..2df9957b --- /dev/null +++ b/Solvers/VIATRA-Solver/hu.bme.mit.inf.dslreasoner.viatrasolver.reasoner/src/hu/bme/mit/inf/dslreasoner/viatrasolver/reasoner/dse/BalancedActivationSelector.xtend
@@ -0,0 +1,51 @@
	1	package hu.bme.mit.inf.dslreasoner.viatrasolver.reasoner.dse
	2
	3	import java.util.Collection
	4	import java.util.HashMap
	5	import java.util.Map
	6	import java.util.List
	7	import java.util.Random
	8	import java.util.ArrayList
	9	import java.util.LinkedList
	10	import java.util.Collections
	11
	12	class BalancedActivationSelector extends ActivationSelector{
	13	val Random r = new Random
	14
	15	new(Random r) {
	16	super(r)
	17	}
	18
	19	override protected internalRandomizationOfActivationIDs(Collection<Object> activationIDs) {
	20	val Map<String,List<Object>> urns = new HashMap
	21	val res = new ArrayList(activationIDs.size)
	22	for(activationID : activationIDs) {
	23	val pair = activationID as Pair<String,? extends Object>
	24	val name = pair.key
	25	val selectedUrn = urns.get(name)
	26	if(selectedUrn!==null) {
	27	selectedUrn.add(activationID)
	28	} else {
	29	val collection = new LinkedList
	30	collection.add(activationID)
	31	urns.put(name,collection)
	32	}
	33	}
	34
	35	for(list:urns.values) {
	36	Collections.shuffle(list,r)
	37	}
	38
	39	while(!urns.empty) {
	40	val randomEntry = urns.entrySet.get(r.nextInt(urns.size))
	41	val list = randomEntry.value
	42	val removedLast = list.remove(list.size-1)
	43	res.add(removedLast)
	44	if(list.empty) {
	45	urns.remove(randomEntry.key)
	46	}
	47	}
	48	return res
	49	}
	50
	51	} \ No newline at end of file