mitkOclDataSet.h

Go to the documentation of this file.

/*============================================================================
 
The Medical Imaging Interaction Toolkit (MITK)
 
Copyright (c) German Cancer Research Center (DKFZ)
All rights reserved.
 
Use of this source code is governed by a 3-clause BSD license that can be
found in the LICENSE file.
 
============================================================================*/
 
#ifndef mitkOclDataSet_h
#define mitkOclDataSet_h
 
#define GPU_DATA 0
#define CPU_DATA 1
 
#include <mitkImage.h>
#include "MitkOpenCLExports.h"
 
#include "mitkOclBaseData.h"
#include "mitkOpenCLActivator.h"
 
#include <mitkException.h>
 
namespace mitk {
 
class MITKOPENCL_EXPORT OclDataSet : public OclBaseData
{
public:
  mitkClassMacro(OclDataSet, OclBaseData);
  itkFactorylessNewMacro(Self);
  itkCloneMacro(Self);
 
  virtual int TransferDataToGPU(cl_command_queue);
 
  virtual void* TransferDataToCPU(cl_command_queue);
 
  void* GetData()
  {
    return m_Data;
  }
 
  cl_mem GetGPUBuffer();
 
  cl_mem CreateGPUBuffer();
 
  bool IsModified(int _type);
 
  using OclBaseData::Modified;
 
  void Modified(int _type);
 
  void SetData(void* data)
  {
    this->m_cpuModified = true;
    this->m_gpuModified = false;
    m_Data = data;
  }
 
  size_t GetBufferSize() const
  {
    return this->m_bufferSize;
  }
 
  short GetBytesPerElement() const
  {
    return this->m_BpE;
  }
 
  void SetBufferSize(size_t size);
 
  void SetBpE(unsigned short BpE);
 
protected:
  OclDataSet();
 
  virtual ~OclDataSet();
 
  cl_mem m_gpuBuffer;
 
  cl_context m_context;
 
private:
  size_t m_bufferSize;
 
  bool m_gpuModified;
  bool m_cpuModified;
 
  void* m_Data;
 
  unsigned short m_BpE;
};
 
}
#endif