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