package org.graalvm.compiler.core.common.util;
import java.util.BitSet;
public final class BitMap2D {
private BitSet map;
private final int bitsPerSlot;
private int bitIndex(int slotIndex, int bitWithinSlotIndex) {
return slotIndex * bitsPerSlot + bitWithinSlotIndex;
}
private boolean verifyBitWithinSlotIndex(int index) {
assert index < bitsPerSlot : "index " + index + " is out of bounds " + bitsPerSlot;
return true;
}
public BitMap2D(int sizeInSlots, int bitsPerSlot) {
map = new BitSet(sizeInSlots * bitsPerSlot);
this.bitsPerSlot = bitsPerSlot;
}
public int sizeInBits() {
return map.size();
}
public int sizeInSlots() {
return map.size() / bitsPerSlot;
}
public boolean isValidIndex(int slotIndex, int bitWithinSlotIndex) {
assert verifyBitWithinSlotIndex(bitWithinSlotIndex);
return (bitIndex(slotIndex, bitWithinSlotIndex) < sizeInBits());
}
public boolean at(int slotIndex, int bitWithinSlotIndex) {
assert verifyBitWithinSlotIndex(bitWithinSlotIndex);
return map.get(bitIndex(slotIndex, bitWithinSlotIndex));
}
public void setBit(int slotIndex, int bitWithinSlotIndex) {
assert verifyBitWithinSlotIndex(bitWithinSlotIndex);
map.set(bitIndex(slotIndex, bitWithinSlotIndex));
}
public void clearBit(int slotIndex, int bitWithinSlotIndex) {
assert verifyBitWithinSlotIndex(bitWithinSlotIndex);
map.clear(bitIndex(slotIndex, bitWithinSlotIndex));
}
public void atPutGrow(int slotIndex, int bitWithinSlotIndex, boolean value) {
int size = sizeInSlots();
if (size <= slotIndex) {
while (size <= slotIndex) {
size *= 2;
}
BitSet newBitMap = new BitSet(size * bitsPerSlot);
newBitMap.or(map);
map = newBitMap;
}
if (value) {
setBit(slotIndex, bitWithinSlotIndex);
} else {
clearBit(slotIndex, bitWithinSlotIndex);
}
}
public void clear() {
map.clear();
}
}