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 //__mitkOclDataSet_h