mitkPhotoacousticOCLUsedLinesCalculation.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 _MITKPHOTOACOUSTICSOCLUSEDLINESCALCULATION_H_
#define _MITKPHOTOACOUSTICSOCLUSEDLINESCALCULATION_H_

#if defined(PHOTOACOUSTICS_USE_GPU) || DOXYGEN

#include "mitkOclDataSetToDataSetFilter.h"
#include <itkObject.h>
#include "mitkBeamformingSettings.h"

namespace mitk
{
  class OCLUsedLinesCalculation : public OclDataSetToDataSetFilter, public itk::Object
  {
  public:
    mitkClassMacroItkParent(OCLUsedLinesCalculation, itk::Object);
    mitkNewMacro1Param(Self, mitk::BeamformingSettings::Pointer);

    void Update();

    void SetElementHeightsBuffer(cl_mem elementHeightsBuffer);
    void SetElementPositionsBuffer(cl_mem elementPositionsBuffer);

  protected:

    OCLUsedLinesCalculation(mitk::BeamformingSettings::Pointer settings);

    virtual ~OCLUsedLinesCalculation();

    bool Initialize();

    void Execute();

    mitk::PixelType GetOutputType()
    {
      return mitk::MakeScalarPixelType<unsigned short>();
    }

    int GetBytesPerElem()
    {
      return sizeof(unsigned short);
    }

    virtual us::Module* GetModule();

    int m_sizeThis;

  private:
    cl_kernel m_PixelCalculation;

    BeamformingSettings::Pointer m_Conf;
    float m_part;
    size_t m_ChunkSize[3];
    cl_mem m_ElementHeightsBuffer;
    cl_mem m_ElementPositionsBuffer;
  };
}
#endif
#endif