#ifndef FastBitVector_h
#define FastBitVector_h
#include <string.h>
#include <wtf/FastMalloc.h>
#include <wtf/StdLibExtras.h>
namespace WTF {
class PrintStream;
class FastBitVector {
public:
FastBitVector()
: m_array(0)
, m_numBits(0)
{
}
FastBitVector(const FastBitVector& other)
: m_array(0)
, m_numBits(0)
{
*this = other;
}
~FastBitVector()
{
if (m_array)
fastFree(m_array);
}
FastBitVector& operator=(const FastBitVector& other)
{
size_t length = other.arrayLength();
uint32_t* newArray = static_cast<uint32_t*>(fastCalloc(length, 4));
memcpy(newArray, other.m_array, length * 4);
if (m_array)
fastFree(m_array);
m_array = newArray;
m_numBits = other.m_numBits;
return *this;
}
size_t numBits() const { return m_numBits; }
void resize(size_t numBits)
{
size_t newLength = arrayLength(numBits);
uint32_t* newArray = static_cast<uint32_t*>(fastCalloc(newLength, 4));
memcpy(newArray, m_array, arrayLength() * 4);
if (m_array)
fastFree(m_array);
m_array = newArray;
m_numBits = numBits;
}
void setAll()
{
memset(m_array, 255, arrayLength() * 4);
}
void clearAll()
{
memset(m_array, 0, arrayLength() * 4);
}
void set(const FastBitVector& other)
{
ASSERT(m_numBits == other.m_numBits);
memcpy(m_array, other.m_array, arrayLength() * 4);
}
bool setAndCheck(const FastBitVector& other)
{
bool changed = false;
ASSERT(m_numBits == other.m_numBits);
for (unsigned i = arrayLength(); i--;) {
changed |= m_array[i] != other.m_array[i];
m_array[i] = other.m_array[i];
}
return changed;
}
bool equals(const FastBitVector& other) const
{
ASSERT(m_numBits == other.m_numBits);
for (unsigned i = arrayLength(); i--;) {
if (m_array[i] != other.m_array[i])
return false;
}
return true;
}
void merge(const FastBitVector& other)
{
ASSERT(m_numBits == other.m_numBits);
for (unsigned i = arrayLength(); i--;)
m_array[i] |= other.m_array[i];
}
void filter(const FastBitVector& other)
{
ASSERT(m_numBits == other.m_numBits);
for (unsigned i = arrayLength(); i--;)
m_array[i] &= other.m_array[i];
}
void exclude(const FastBitVector& other)
{
ASSERT(m_numBits == other.m_numBits);
for (unsigned i = arrayLength(); i--;)
m_array[i] &= ~other.m_array[i];
}
void set(size_t i)
{
ASSERT_WITH_SECURITY_IMPLICATION(i < m_numBits);
m_array[i >> 5] |= (1 << (i & 31));
}
void clear(size_t i)
{
ASSERT_WITH_SECURITY_IMPLICATION(i < m_numBits);
m_array[i >> 5] &= ~(1 << (i & 31));
}
void set(size_t i, bool value)
{
if (value)
set(i);
else
clear(i);
}
bool get(size_t i) const
{
ASSERT_WITH_SECURITY_IMPLICATION(i < m_numBits);
return !!(m_array[i >> 5] & (1 << (i & 31)));
}
size_t bitCount() const
{
size_t result = 0;
for (unsigned i = arrayLength(); i--;)
result += WTF::bitCount(m_array[i]);
return result;
}
WTF_EXPORT_PRIVATE void dump(PrintStream&) const;
private:
static size_t arrayLength(size_t numBits) { return (numBits + 31) >> 5; }
size_t arrayLength() const { return arrayLength(m_numBits); }
uint32_t* m_array; size_t m_numBits;
};
}
using WTF::FastBitVector;
#endif // FastBitVector_h