diff options
Diffstat (limited to 'tests/manual/kinectsurface/QtKinectWrapper/OpenNI/Include/XnBitSet.h')
-rw-r--r-- | tests/manual/kinectsurface/QtKinectWrapper/OpenNI/Include/XnBitSet.h | 135 |
1 files changed, 135 insertions, 0 deletions
diff --git a/tests/manual/kinectsurface/QtKinectWrapper/OpenNI/Include/XnBitSet.h b/tests/manual/kinectsurface/QtKinectWrapper/OpenNI/Include/XnBitSet.h new file mode 100644 index 00000000..89ad6450 --- /dev/null +++ b/tests/manual/kinectsurface/QtKinectWrapper/OpenNI/Include/XnBitSet.h @@ -0,0 +1,135 @@ +/**************************************************************************** +* * +* OpenNI 1.x Alpha * +* Copyright (C) 2011 PrimeSense Ltd. * +* * +* This file is part of OpenNI. * +* * +* OpenNI is free software: you can redistribute it and/or modify * +* it under the terms of the GNU Lesser General Public License as published * +* by the Free Software Foundation, either version 3 of the License, or * +* (at your option) any later version. * +* * +* OpenNI is distributed in the hope that it will be useful, * +* but WITHOUT ANY WARRANTY; without even the implied warranty of * +* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the * +* GNU Lesser General Public License for more details. * +* * +* You should have received a copy of the GNU Lesser General Public License * +* along with OpenNI. If not, see <http://www.gnu.org/licenses/>. * +* * +****************************************************************************/ +#ifndef __XNBITSET_H__ +#define __XNBITSET_H__ + +#include <XnArray.h> + +class XnBitSet +{ +public: + XnBitSet() : m_nSize(0) {} + + /** Reserves space in this bitset for the specified number of bits. + This saves you re-allocations and data copies if you know the size in advance. **/ + XnStatus Reserve(XnUInt32 nBits) + { + return m_array.Reserve((nBits >> 5) + 1); + } + + /** Sets the size of the bitset to the specified number of bits and sets them all to 0. **/ + XnStatus SetSize(XnUInt32 nBits) + { + return m_array.SetSize((nBits >> 5) + 1, 0); + } + + /** Sets the bit at nIndex to bValue. **/ + XnStatus Set(XnUInt32 nIndex, XnBool bValue) + { + XnUInt32 nArrayIndex = (nIndex >> 5); + XnUInt32 nMask = (1 << ((~nIndex) & 0x1F)); + XnUInt32 nOldVal = nArrayIndex < m_array.GetSize() ? m_array[nArrayIndex] : 0; + XnUInt32 nNewVal = bValue ? (nOldVal | nMask) : (nOldVal & (~nMask)); + XnStatus nRetVal = m_array.Set(nArrayIndex, nNewVal, 0); + XN_IS_STATUS_OK(nRetVal); + m_nSize = XN_MAX(m_nSize, nIndex + 1); + return XN_STATUS_OK; + } + + /** @returns the value of the bit specified by nIndex. **/ + XnBool IsSet(XnUInt32 nIndex) const + { + XnUInt32 nArrayIndex = (nIndex >> 5); + if (nArrayIndex >= m_array.GetSize()) + { + return FALSE; + } + return (m_array[nArrayIndex] & (1 << ((~nIndex) & 0x1F))) ? TRUE : FALSE; + } + + /** Copies raw data from a buffer of dwords to this bitset. **/ + XnStatus SetData(const XnUInt32* pData, XnUInt32 nSizeInDwords) + { + XnStatus nRetVal = m_array.SetData(pData, nSizeInDwords); + XN_IS_STATUS_OK(nRetVal); + m_nSize = (nSizeInDwords << 5); + return XN_STATUS_OK; + } + + /** Copies raw data from a buffer of bytes to this bitset. **/ + XnStatus SetDataBytes(const XnUInt8* pData, XnUInt32 nSizeInBytes) + { + //XnStatus nRetVal = m_array.SetData(reinterpret_cast<const XnUInt32*>(pData), XN_MAX(1, nSizeInBytes >> 2)); + XnUInt32 nSizeInDwords = XN_MAX(1, nSizeInBytes >> 2); + XnStatus nRetVal = m_array.SetSize(nSizeInDwords); + XN_IS_STATUS_OK(nRetVal); + for (XnUInt32 nDwordIdx = 0, nByteIdx = 0; nDwordIdx < nSizeInDwords; nDwordIdx++, nByteIdx += 4) + { + m_array[nDwordIdx] = ((pData[nByteIdx] << 24) | (pData[nByteIdx + 1] << 16) | (pData[nByteIdx + 2] << 8) | pData[nByteIdx + 3] ); + } + m_nSize = (nSizeInBytes << 3); + return XN_STATUS_OK; + } + + /** @returns The raw data of this bitset as a buffer of dwords. **/ + const XnUInt32* GetData() const + { + return m_array.GetData(); + } + + /** @returns The raw data of this bitset as a buffer of dwords. Allows modification of underlying data. **/ + XnUInt32* GetData() + { + return m_array.GetData(); + } + + /** @returns size in bytes of this bitset. **/ + XnUInt32 GetDataSize() const + { + return m_array.GetSize(); + } + + /** @returns size in bits of this bitset. **/ + XnUInt32 GetSize() const + { + return m_nSize; + } + + /** Clears data in this bitset and sets size to 0. **/ + void Clear() + { + m_array.Clear(); + m_nSize = 0; + } + + /** @returns TRUE if this bitset is empty, FALSE otherwise. **/ + XnBool IsEmpty() const + { + return m_array.IsEmpty(); + } + +private: + XnArray<XnUInt32> m_array; + XnUInt32 m_nSize; +}; + +#endif // __XNBITSET_H__
\ No newline at end of file |