Change BitVectorRepository to use LinkedLists

This removes a nasty source of non-determinism when using
MutableSharedBitVectorIntSets.
This commit is contained in:
Manu Sridharan 2013-02-13 16:17:23 -08:00
parent 1cf9a0008b
commit 7579f5644a
1 changed files with 27 additions and 16 deletions

View File

@ -10,12 +10,12 @@
*******************************************************************************/
package com.ibm.wala.util.intset;
import java.lang.ref.WeakReference;
import java.util.Iterator;
import java.util.LinkedList;
import java.util.Map;
import java.util.WeakHashMap;
import com.ibm.wala.util.collections.HashMapFactory;
import com.ibm.wala.util.collections.MapUtil;
/**
* A repository for shared bit vectors as described by Heintze
@ -35,7 +35,7 @@ public class BitVectorRepository {
/**
* A Mapping from Integer -> WeakHashMap
*/
final private static Map<Object, WeakHashMap<BitVectorIntSet,Object>> buckets = HashMapFactory.make();
final private static Map<Integer, LinkedList<WeakReference<BitVectorIntSet>>> buckets = HashMapFactory.make();
/**
* @param value
@ -56,24 +56,35 @@ public class BitVectorRepository {
}
int size = value.size();
for (int i = size; i > size - SUBSET_DELTA; i--) {
WeakHashMap<?, ?> m = buckets.get(Integer.valueOf(i));
LinkedList<WeakReference<BitVectorIntSet>> m = buckets.get(Integer.valueOf(i));
if (m != null) {
for (Iterator<?> it = m.keySet().iterator(); it.hasNext();) {
BitVectorIntSet bv = (BitVectorIntSet) it.next();
if (bv.isSubset(value)) {
// FOUND ONE!
if (STATS) {
hits++;
Iterator<WeakReference<BitVectorIntSet>> it = m.iterator();
while (it.hasNext()) {
WeakReference<BitVectorIntSet> wr = it.next();
BitVectorIntSet bv = wr.get();
if (bv != null) {
if (bv.isSubset(value)) {
// FOUND ONE!
if (STATS) {
hits++;
}
return bv;
}
return bv;
} else {
// might as well clear out the weak ref
it.remove();
}
}
}
}
// didn't find one. create one.
WeakHashMap<BitVectorIntSet, Object> m = MapUtil.findOrCreateWeakHashMap(buckets, new Integer(size));
LinkedList<WeakReference<BitVectorIntSet>> m = buckets.get(size);
if (m == null) {
m = new LinkedList<WeakReference<BitVectorIntSet>>();
buckets.put(size, m);
}
BitVectorIntSet bv = new BitVectorIntSet(value);
m.put(bv, null);
m.add(new WeakReference<BitVectorIntSet>(bv));
return bv;
}
@ -90,9 +101,9 @@ public class BitVectorRepository {
*/
private static int countEntries() {
int result = 0;
for (Iterator<WeakHashMap<BitVectorIntSet,Object>> it = buckets.values().iterator(); it.hasNext();) {
WeakHashMap<?, ?> m = it.next();
result += m.size();
for (LinkedList<WeakReference<BitVectorIntSet>> l : buckets.values()) {
// don't worry about cleared WeakReferences; count will be rough
result += l.size();
}
return result;
}