ITK
6.0.0
Insight Toolkit
|
#include <itkGPUDataManager.h>
GPU memory manager implemented using OpenCL. Required by GPUImage class.
This class serves as a base class for GPU data container for GPUImage class, which is similar to ImageBase class for Image class. However, all the image-related meta data will be already stored in image class (parent of GPUImage), therefore we did not name it GPUImageBase. Rather, this class is a GPU-specific data manager that provides functionalities for CPU-GPU data synchronization and grafting GPU data.
Definition at line 42 of file itkGPUDataManager.h.
Public Types | |
using | ConstPointer = SmartPointer< const Self > |
using | MutexHolderType = std::lock_guard< std::mutex > |
using | Pointer = SmartPointer< Self > |
using | Self = GPUDataManager |
using | Superclass = Object |
Public Types inherited from itk::Object | |
using | ConstPointer = SmartPointer< const Self > |
using | Pointer = SmartPointer< Self > |
using | Self = Object |
using | Superclass = LightObject |
Public Types inherited from itk::LightObject | |
using | ConstPointer = SmartPointer< const Self > |
using | Pointer = SmartPointer< Self > |
using | Self = LightObject |
Public Member Functions | |
void | Allocate () |
unsigned int | GetBufferSize () const |
void * | GetCPUBufferPointer () |
int | GetCurrentCommandQueueID () const |
cl_mem * | GetGPUBufferPointer () |
const char * | GetNameOfClass () const override |
virtual void | Graft (const GPUDataManager *data) |
virtual void | Initialize () |
bool | IsCPUBufferDirty () const |
bool | IsGPUBufferDirty () const |
void | SetBufferFlag (cl_mem_flags flags) |
void | SetBufferSize (unsigned int num) |
void | SetCPUBufferDirty () |
void | SetCPUBufferPointer (void *ptr) |
void | SetCPUDirtyFlag (bool isDirty) |
void | SetCurrentCommandQueue (int queueid) |
void | SetGPUBufferDirty () |
void | SetGPUDirtyFlag (bool isDirty) |
bool | Update () |
virtual void | UpdateCPUBuffer () |
virtual void | UpdateGPUBuffer () |
Public Member Functions inherited from itk::Object | |
unsigned long | AddObserver (const EventObject &event, Command *cmd) const |
unsigned long | AddObserver (const EventObject &event, std::function< void(const EventObject &)> function) const |
LightObject::Pointer | CreateAnother () const override |
virtual void | DebugOff () const |
virtual void | DebugOn () const |
Command * | GetCommand (unsigned long tag) |
bool | GetDebug () const |
MetaDataDictionary & | GetMetaDataDictionary () |
const MetaDataDictionary & | GetMetaDataDictionary () const |
virtual ModifiedTimeType | GetMTime () const |
const char * | GetNameOfClass () const override |
virtual const TimeStamp & | GetTimeStamp () const |
bool | HasObserver (const EventObject &event) const |
void | InvokeEvent (const EventObject &) |
void | InvokeEvent (const EventObject &) const |
virtual void | Modified () const |
void | Register () const override |
void | RemoveAllObservers () |
void | RemoveObserver (unsigned long tag) const |
void | SetDebug (bool debugFlag) const |
void | SetReferenceCount (int) override |
void | UnRegister () const noexcept override |
void | SetMetaDataDictionary (const MetaDataDictionary &rhs) |
void | SetMetaDataDictionary (MetaDataDictionary &&rrhs) |
virtual void | SetObjectName (std::string _arg) |
virtual const std::string & | GetObjectName () const |
Public Member Functions inherited from itk::LightObject | |
Pointer | Clone () const |
virtual Pointer | CreateAnother () const |
virtual void | Delete () |
virtual const char * | GetNameOfClass () const |
virtual int | GetReferenceCount () const |
void | Print (std::ostream &os, Indent indent=0) const |
virtual void | Register () const |
virtual void | SetReferenceCount (int) |
virtual void | UnRegister () const noexcept |
Static Public Member Functions | |
static Pointer | New () |
Static Public Member Functions inherited from itk::Object | |
static bool | GetGlobalWarningDisplay () |
static void | GlobalWarningDisplayOff () |
static void | GlobalWarningDisplayOn () |
static Pointer | New () |
static void | SetGlobalWarningDisplay (bool val) |
Static Public Member Functions inherited from itk::LightObject | |
static void | BreakOnError () |
static Pointer | New () |
Protected Member Functions | |
GPUDataManager () | |
void | PrintSelf (std::ostream &os, Indent indent) const override |
~GPUDataManager () override | |
Protected Member Functions inherited from itk::Object | |
Object () | |
bool | PrintObservers (std::ostream &os, Indent indent) const |
void | PrintSelf (std::ostream &os, Indent indent) const override |
virtual void | SetTimeStamp (const TimeStamp &timeStamp) |
~Object () override | |
Protected Member Functions inherited from itk::LightObject | |
virtual LightObject::Pointer | InternalClone () const |
LightObject () | |
virtual void | PrintHeader (std::ostream &os, Indent indent) const |
virtual void | PrintSelf (std::ostream &os, Indent indent) const |
virtual void | PrintTrailer (std::ostream &os, Indent indent) const |
virtual | ~LightObject () |
Protected Attributes | |
unsigned int | m_BufferSize {} |
int | m_CommandQueueId {} |
GPUContextManager * | m_ContextManager {} |
void * | m_CPUBuffer {} |
cl_mem | m_GPUBuffer {} |
bool | m_IsCPUBufferDirty {} |
bool | m_IsGPUBufferDirty {} |
cl_mem_flags | m_MemFlags {} |
std::mutex | m_Mutex {} |
Protected Attributes inherited from itk::LightObject | |
std::atomic< int > | m_ReferenceCount {} |
Friends | |
class | GPUKernelManager |
using itk::GPUDataManager::ConstPointer = SmartPointer<const Self> |
Definition at line 53 of file itkGPUDataManager.h.
using itk::GPUDataManager::MutexHolderType = std::lock_guard<std::mutex> |
Definition at line 58 of file itkGPUDataManager.h.
using itk::GPUDataManager::Pointer = SmartPointer<Self> |
Definition at line 52 of file itkGPUDataManager.h.
Definition at line 50 of file itkGPUDataManager.h.
Definition at line 51 of file itkGPUDataManager.h.
|
protected |
|
overrideprotected |
void itk::GPUDataManager::Allocate | ( | ) |
|
inline |
Definition at line 65 of file itkGPUDataManager.h.
References m_BufferSize.
void * itk::GPUDataManager::GetCPUBufferPointer | ( | ) |
Get GPU buffer pointer
int itk::GPUDataManager::GetCurrentCommandQueueID | ( | ) | const |
cl_mem * itk::GPUDataManager::GetGPUBufferPointer | ( | ) |
Get GPU buffer pointer
|
overridevirtual |
Reimplemented from itk::Object.
Reimplemented in itk::GPUImageDataManager< ImageType >, and itk::GPUImageDataManager< itk::GPUImage >.
|
virtual |
Method for grafting the content of one GPUDataManager into another one
|
virtual |
Initialize GPUDataManager
|
inline |
Definition at line 93 of file itkGPUDataManager.h.
References m_IsCPUBufferDirty.
|
inline |
Definition at line 99 of file itkGPUDataManager.h.
References m_IsGPUBufferDirty.
|
static |
|
overrideprotectedvirtual |
Methods invoked by Print() to print information about the object including superclasses. Typically not called by the user (use Print() instead) but used in the hierarchical print process to combine the output of several classes.
Reimplemented from itk::Object.
void itk::GPUDataManager::SetBufferFlag | ( | cl_mem_flags | flags | ) |
void itk::GPUDataManager::SetBufferSize | ( | unsigned int | num | ) |
total buffer size in bytes
void itk::GPUDataManager::SetCPUBufferDirty | ( | ) |
Make GPU up-to-date and mark CPU as dirty. Call this function when you want to modify CPU data
void itk::GPUDataManager::SetCPUBufferPointer | ( | void * | ptr | ) |
void itk::GPUDataManager::SetCPUDirtyFlag | ( | bool | isDirty | ) |
void itk::GPUDataManager::SetCurrentCommandQueue | ( | int | queueid | ) |
void itk::GPUDataManager::SetGPUBufferDirty | ( | ) |
Make CPU up-to-date and mark GPU as dirty. Call this function when you want to modify GPU data
void itk::GPUDataManager::SetGPUDirtyFlag | ( | bool | isDirty | ) |
bool itk::GPUDataManager::Update | ( | ) |
Synchronize CPU and GPU buffers (using dirty flags)
|
virtual |
actual GPU->CPU memory copy takes place here
|
virtual |
actual CPU->GPU memory copy takes place here
|
friend |
allow GPUKernelManager to access GPU buffer pointer
Definition at line 45 of file itkGPUDataManager.h.
|
protected |
Definition at line 150 of file itkGPUDataManager.h.
Referenced by GetBufferSize().
|
protected |
Definition at line 154 of file itkGPUDataManager.h.
|
protected |
Definition at line 152 of file itkGPUDataManager.h.
|
protected |
Definition at line 161 of file itkGPUDataManager.h.
|
protected |
buffer pointers
Definition at line 160 of file itkGPUDataManager.h.
|
protected |
Definition at line 165 of file itkGPUDataManager.h.
Referenced by IsCPUBufferDirty().
|
protected |
checks if buffer needs to be updated
Definition at line 164 of file itkGPUDataManager.h.
Referenced by IsGPUBufferDirty().
|
protected |
buffer type
Definition at line 157 of file itkGPUDataManager.h.
|
protected |
Mutex lock to prevent r/w hazard for multithreaded code
Definition at line 168 of file itkGPUDataManager.h.