windows-nt/Source/XPSP1/NT/shell/osshell/dskquota/common/bitset.cpp

149 lines
2.3 KiB
C++
Raw Permalink Normal View History

2020-09-26 03:20:57 -05:00
#include "pch.h"
#pragma hdrstop
#include "bitset.h"
BitSet::BitSet(
int cBits
) : m_pBuffer(NULL),
m_cBits(0),
m_cSet(0),
m_cElements(0)
{
Initialize(cBits);
}
BitSet::BitSet(
const BitSet& rhs
) : m_pBuffer(NULL),
m_cBits(0),
m_cSet(0),
m_cElements(0)
{
*this = rhs;
}
BitSet&
BitSet::operator = (
const BitSet& rhs
)
{
if (this != &rhs)
{
Initialize(rhs.m_cBits);
for (int i = 0; i < m_cElements; i++)
{
m_pBuffer[i] = rhs.m_pBuffer[i];
}
}
return *this;
}
BitSet::~BitSet(
void
)
{
delete [] m_pBuffer;
}
void
BitSet::Initialize(
int cBits
)
{
delete [] m_pBuffer;
m_cSet = 0;
m_cBits = cBits;
m_cElements = ElementInBuffer(m_cBits) + 1;
m_pBuffer = new ELEMENT_TYPE[m_cElements];
ClrAll();
}
bool
BitSet::GetBitState(
int iBit
) const
{
ValidateBitNumber(iBit);
int iElement = ElementInBuffer(iBit);
bool bBitState = false;
ELEMENT_TYPE mask = MaskFromBit(iBit);
bBitState = (*(m_pBuffer + iElement) & mask) != 0;
return bBitState;
}
void
BitSet::SetBitState(
int iBit,
bool bSet
)
{
ValidateBitNumber(iBit);
int iElement = ElementInBuffer(iBit);
ELEMENT_TYPE mask = MaskFromBit(iBit);
if (bSet)
{
*(m_pBuffer + iElement) |= mask;
m_cSet++;
}
else
{
*(m_pBuffer + iElement) &= ~mask;
m_cSet--;
}
}
void
BitSet::Complement(
void
)
{
for (int i = 0; i < m_cElements; i++)
{
m_pBuffer[i] = ~m_pBuffer[i];
}
m_cSet = CountClr();
}
#if DBG
void
BitSet::Dump(
void
) const
{
#define BitSetOUT(s) *pszOut++ = TEXT('\r'); \
*pszOut++ = TEXT('\n'); \
*pszOut++ = TEXT('\0'); \
OutputDebugString(s)
TCHAR szOut[80];
LPTSTR pszOut = szOut;
const TCHAR szBits[] = TEXT("01");
for (int i = 0; i < m_cBits; i++)
{
if (i % 8 == 0)
{
if (i % 32 == 0)
{
BitSetOUT(szOut);
pszOut = szOut;
}
else
*pszOut++ = TEXT(' ');
}
*pszOut++ = szBits[int(GetBitState(i))];
}
BitSetOUT(szOut);
}
#endif // DBG