mirror of
https://gitlab.kitware.com/vtk/vtk-m
synced 2024-09-08 21:33:55 +00:00
2f04dc3aef
Previously you had to create a special virtual object to place in the `Buffer`'s metadata. This added a lot of difficulty that is probably unnecessary. Instead, just have `Buffer` hold an arbitrary object and some simple functions to copy and delete it. There may be issues with type safety across translation units, but we'll deal with that when/if we run into it.
142 lines
4.4 KiB
C++
142 lines
4.4 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.
|
|
//============================================================================
|
|
|
|
#include <vtkm/cont/BitField.h>
|
|
#include <vtkm/cont/Logging.h>
|
|
|
|
namespace
|
|
{
|
|
|
|
struct DeviceCheckFunctor
|
|
{
|
|
vtkm::cont::DeviceAdapterId FoundDevice = vtkm::cont::DeviceAdapterTagUndefined{};
|
|
|
|
VTKM_CONT void operator()(vtkm::cont::DeviceAdapterId device,
|
|
const vtkm::cont::internal::Buffer& buffer)
|
|
{
|
|
if (this->FoundDevice == vtkm::cont::DeviceAdapterTagUndefined{})
|
|
{
|
|
if (buffer.IsAllocatedOnDevice(device))
|
|
{
|
|
this->FoundDevice = device;
|
|
}
|
|
}
|
|
}
|
|
};
|
|
|
|
} // anonymous namespace
|
|
|
|
namespace vtkm
|
|
{
|
|
namespace cont
|
|
{
|
|
|
|
namespace detail
|
|
{
|
|
|
|
} // namespace detail
|
|
|
|
BitField::BitField()
|
|
{
|
|
this->Buffer.SetMetaData(internal::BitFieldMetaData{});
|
|
}
|
|
|
|
vtkm::Id BitField::GetNumberOfBits() const
|
|
{
|
|
return this->Buffer.GetMetaData<internal::BitFieldMetaData>().NumberOfBits;
|
|
}
|
|
|
|
void BitField::Allocate(vtkm::Id numberOfBits,
|
|
vtkm::CopyFlag preserve,
|
|
vtkm::cont::Token& token) const
|
|
{
|
|
const vtkm::BufferSizeType bytesNeeded = (numberOfBits + CHAR_BIT - 1) / CHAR_BIT;
|
|
const vtkm::BufferSizeType blocksNeeded = (bytesNeeded + BlockSize - 1) / BlockSize;
|
|
const vtkm::BufferSizeType numBytes = blocksNeeded * BlockSize;
|
|
|
|
VTKM_LOG_F(vtkm::cont::LogLevel::MemCont,
|
|
"BitField Allocation: %llu bits, blocked up to %s bytes.",
|
|
static_cast<unsigned long long>(numberOfBits),
|
|
vtkm::cont::GetSizeString(static_cast<vtkm::UInt64>(numBytes)).c_str());
|
|
|
|
this->Buffer.SetNumberOfBytes(numBytes, preserve, token);
|
|
this->Buffer.GetMetaData<internal::BitFieldMetaData>().NumberOfBits = numberOfBits;
|
|
}
|
|
|
|
void BitField::ReleaseResourcesExecution()
|
|
{
|
|
this->Buffer.ReleaseDeviceResources();
|
|
}
|
|
|
|
void BitField::ReleaseResources()
|
|
{
|
|
vtkm::cont::Token token;
|
|
this->Buffer.SetNumberOfBytes(0, vtkm::CopyFlag::Off, token);
|
|
this->Buffer.GetMetaData<internal::BitFieldMetaData>().NumberOfBits = 0;
|
|
}
|
|
|
|
void BitField::SyncControlArray() const
|
|
{
|
|
vtkm::cont::Token token;
|
|
this->Buffer.ReadPointerHost(token);
|
|
}
|
|
|
|
bool BitField::IsOnDevice(vtkm::cont::DeviceAdapterId device) const
|
|
{
|
|
return this->Buffer.IsAllocatedOnDevice(device);
|
|
}
|
|
|
|
vtkm::cont::DeviceAdapterId BitField::GetDeviceAdapterId() const
|
|
{
|
|
DeviceCheckFunctor functor;
|
|
vtkm::ListForEach(functor, VTKM_DEFAULT_DEVICE_ADAPTER_LIST{}, this->Buffer);
|
|
return functor.FoundDevice;
|
|
}
|
|
|
|
BitField::WritePortalType BitField::WritePortal() const
|
|
{
|
|
vtkm::cont::Token token;
|
|
return WritePortalType(this->Buffer.WritePointerHost(token),
|
|
this->Buffer.GetMetaData<internal::BitFieldMetaData>().NumberOfBits);
|
|
}
|
|
|
|
BitField::ReadPortalType BitField::ReadPortal() const
|
|
{
|
|
vtkm::cont::Token token;
|
|
return ReadPortalType(this->Buffer.ReadPointerHost(token),
|
|
this->Buffer.GetMetaData<internal::BitFieldMetaData>().NumberOfBits);
|
|
}
|
|
|
|
BitField::ReadPortalType BitField::PrepareForInput(vtkm::cont::DeviceAdapterId device,
|
|
vtkm::cont::Token& token) const
|
|
{
|
|
return ReadPortalType(this->Buffer.ReadPointerDevice(device, token),
|
|
this->Buffer.GetMetaData<internal::BitFieldMetaData>().NumberOfBits);
|
|
}
|
|
|
|
BitField::WritePortalType BitField::PrepareForOutput(vtkm::Id numBits,
|
|
vtkm::cont::DeviceAdapterId device,
|
|
vtkm::cont::Token& token) const
|
|
{
|
|
this->Allocate(numBits, vtkm::CopyFlag::Off, token);
|
|
return WritePortalType(this->Buffer.WritePointerDevice(device, token),
|
|
this->Buffer.GetMetaData<internal::BitFieldMetaData>().NumberOfBits);
|
|
}
|
|
|
|
BitField::WritePortalType BitField::PrepareForInPlace(vtkm::cont::DeviceAdapterId device,
|
|
vtkm::cont::Token& token) const
|
|
{
|
|
return WritePortalType(this->Buffer.WritePointerDevice(device, token),
|
|
this->Buffer.GetMetaData<internal::BitFieldMetaData>().NumberOfBits);
|
|
}
|
|
|
|
}
|
|
} // namespace vtkm::cont
|