mirror of
https://gitlab.kitware.com/vtk/vtk-m
synced 2024-07-06 07:17:25 +00:00
70 lines
1.9 KiB
C++
70 lines
1.9 KiB
C++
//============================================================================
|
|
// Copyright (c) Kitware, Inc.
|
|
// All rights reserved.
|
|
// See LICENSE.txt for details.
|
|
//
|
|
// This software is distributed WITHOUT ANY WARRANTY; without even
|
|
// the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR
|
|
// PURPOSE. See the above copyright notice for more information.
|
|
//============================================================================
|
|
|
|
#ifndef vtk_m_Bitset_h
|
|
#define vtk_m_Bitset_h
|
|
|
|
#include <cassert>
|
|
#include <limits>
|
|
#include <vtkm/Types.h>
|
|
#include <vtkm/internal/ExportMacros.h>
|
|
|
|
namespace vtkm
|
|
{
|
|
|
|
/// \brief A bitmap to serve different needs.
|
|
/// Ex. Editing particular bits in a byte(s), checkint if particular bit values
|
|
/// are present or not. Once Cuda supports std::bitset, we should use the
|
|
/// standard one if possible. Additional cast in logical operations are required
|
|
/// to avoid compiler warnings when using 16 or 8 bit MaskType.
|
|
template <typename MaskType>
|
|
struct Bitset
|
|
{
|
|
VTKM_EXEC_CONT void set(vtkm::Id bitIndex)
|
|
{
|
|
this->Mask = static_cast<MaskType>(this->Mask | (static_cast<MaskType>(1) << bitIndex));
|
|
}
|
|
|
|
VTKM_EXEC_CONT void set(vtkm::Id bitIndex, bool val)
|
|
{
|
|
if (val)
|
|
this->set(bitIndex);
|
|
else
|
|
this->reset(bitIndex);
|
|
}
|
|
|
|
VTKM_EXEC_CONT void reset(vtkm::Id bitIndex)
|
|
{
|
|
this->Mask = static_cast<MaskType>(this->Mask & ~(static_cast<MaskType>(1) << bitIndex));
|
|
}
|
|
|
|
VTKM_EXEC_CONT void toggle(vtkm::Id bitIndex)
|
|
{
|
|
this->Mask = this->Mask ^ (static_cast<MaskType>(0) << bitIndex);
|
|
}
|
|
|
|
VTKM_EXEC_CONT bool test(vtkm::Id bitIndex) const
|
|
{
|
|
return ((this->Mask & (static_cast<MaskType>(1) << bitIndex)) != 0);
|
|
}
|
|
|
|
VTKM_EXEC_CONT bool operator==(const vtkm::Bitset<MaskType>& otherBitset) const
|
|
{
|
|
return this->Mask == otherBitset.Mask;
|
|
}
|
|
|
|
private:
|
|
MaskType Mask = 0;
|
|
};
|
|
|
|
} // namespace vtkm
|
|
|
|
#endif //vtk_m_Bitset_h
|