mitkOclBaseData.h

Go to the documentation of this file.

/*===================================================================

The Medical Imaging Interaction Toolkit (MITK)

Copyright (c) German Cancer Research Center,
Division of Medical and Biological Informatics.
All rights reserved.

This software is distributed WITHOUT ANY WARRANTY; without
even the implied warranty of MERCHANTABILITY or FITNESS FOR
A PARTICULAR PURPOSE.

See LICENSE.txt or http://www.mitk.org for details.

===================================================================*/

#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