mitkOclFilter.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 __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 ~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 Initialize();

  void CompileSource();

  void SetSourcePreambel(const char* preambel);

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

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