BitArray.h

Go to the documentation of this file.

//! @file BitArray.h
//! CBitArray class for handling array of bits
//
//////////////////////////////////////////////////////////////////////
 
#ifndef CBITARRAY_CLASS_H
#define CBITARRAY_CLASS_H
 
#ifdef _MSC_VER
#pragma once
#endif // _MSC_VER
 
BEGIN_MOOTOOLS_NAMESPACE
 
//! @class CBitArray
//! @brief CBitArray class for handling array of bits
class DLL_TOOLSFUNCTION CBitArray
{
    protected:
        unsigned int size;
        unsigned int *buffer;
 
        unsigned int GetBufferSize() const;
        unsigned int GetIndex(unsigned int index, unsigned int& shift) const;
 
    public:
        CBitArray();
        virtual ~CBitArray();
 
        void RemoveAll();
        void SetSize(unsigned int size, bool value); // Set bitarray size. This currently remove all entries, if buffer size is modified
        unsigned int GetSize() const;
 
        bool operator[](unsigned int index) const;
        bool GetAt(unsigned int index) const;
        void SetAt(unsigned int index, bool value);
 
        unsigned int GetCount(bool value) const; // Return number of element set to value
        bool IsOneSet(bool value) const;
 
        unsigned int *DetachBuffer(unsigned int& buffersize);
        unsigned int *GetBuffer(unsigned int& buffersize);
        bool SetBuffer(unsigned int *buffer, unsigned int size, unsigned int buffersize);
 
        void EraseByte(unsigned int index, bool value); // Fast clear of the byte containing the index for fast processing
};
 
inline unsigned int CBitArray::GetIndex(unsigned int index, unsigned int& shift) const
{
    shift = (index % 32);
    shift = 1U << shift;
 
    return (index / 32);
}
 
inline bool CBitArray::operator[](unsigned int index) const
{
    XASSERT(index < size);
    if (index >= size)
        return false;
 
    unsigned int shift;
    index = GetIndex(index, shift);
 
    return !!(buffer[index] & shift);
}
 
inline bool CBitArray::GetAt(unsigned int index) const
{
    return (*this)[index];
}
 
inline void CBitArray::SetAt(unsigned int index, bool value)
{
    XASSERT(index < size);
    if (index >= size)
        return;
 
    unsigned int shift;
    index = GetIndex(index, shift);
    
    if (value)
        buffer[index] |= shift;
    else
        buffer[index] &= ~shift;
}
 
inline void CBitArray::EraseByte(unsigned int index, bool value)
{
    XASSERT(index < size);
    if (index >= size)
        return;
 
    if (value)
        buffer[index / 32] = 0xFFFFFFFF;
    else
        buffer[index / 32] = 0;
}
 
END_MOOTOOLS_NAMESPACE
 
#endif // !CBITARRAY_CLASS_H