mitkOclFilter.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 __mitkOclFilter_h
#define __mitkOclFilter_h

#include "mitkOclUtils.h"
#include "mitkCommon.h"

#include <vector>
#include <usModule.h>

#include <MitkOpenCLExports.h>

namespace mitk
{
class MITKOPENCL_EXPORT OclFilter
{
public:
    void AddSourceFile(const char* filename);

  void SetCompilerFlags(const char* flags);

  virtual bool IsInitialized();

  virtual unsigned long GetDeviceMemory();

  virtual ~OclFilter();

protected:

  typedef std::vector<const char*> CStringList;
  typedef std::vector<size_t> ClSizeList;

  OclFilter();

  OclFilter(const char* filename);

  const char* m_ClCompilerFlags;

  cl_program m_ClProgram;

  cl_command_queue m_CommandQue;

  std::string m_FilterID;

  const char* m_Preambel;

  CStringList m_ClFiles;

  bool m_Initialized;

  size_t m_LocalWorkSize[3];

  size_t m_GlobalWorkSize[3];

  void SetWorkingSize(unsigned int locx, unsigned int dimx,
                      unsigned int locy = 1, unsigned int dimy = 1,
                      unsigned int locz = 1, unsigned int dimz = 1);

  bool ExecuteKernel( cl_kernel kernel, unsigned int workSizeDim );

  bool ExecuteKernelChunks( cl_kernel kernel, unsigned int workSizeDim, size_t* chunksDim );

  bool ExecuteKernelChunksInBatches(cl_kernel kernel, unsigned int workSizeDim, size_t* chunksDim, size_t batchSize, int waitTimems);
  bool Initialize();

  void CompileSource();

  void SetSourcePreambel(const char* preambel);

  virtual us::Module* GetModule() = 0;

  void LoadSourceFiles(CStringList &SourceCodeList, ClSizeList &SourceCodeSizeList);
};
}
#endif // __mitkOclFilter_h