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