mitkOclBaseData.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 mitkOclBaseData_h
#define mitkOclBaseData_h
 
#include "MitkOpenCLExports.h"
#include "mitkOclUtils.h"
 
#include "mitkOpenCL.h"
 
#include <mitkCommon.h>
#include <itkObjectFactory.h>
#include <itkObject.h>
 
namespace mitk
{
 
  class MITKOPENCL_EXPORT OclBaseData : public itk::Object
  {
  public:
    mitkClassMacroItkParent(OclBaseData, itk::Object);
 
    virtual int TransferDataToGPU(cl_command_queue) = 0;
 
    virtual void* TransferDataToCPU(cl_command_queue) = 0;
 
    void GPUModified();
 
    void CPUModified();
 
 
    virtual cl_mem GetGPUBuffer();
 
  protected:
    OclBaseData();
 
    virtual ~OclBaseData();
 
  private:
 
    size_t m_BufferSize;
 
    cl_mem m_GPUBuffer;
 
    void* m_CPUBuffer;
  };
 
}
 
#endif