mitkPhotoacousticOCLDelayCalculation.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 _MITKPHOTOACOUSTICSDELAYCALC_H_
#define _MITKPHOTOACOUSTICSDELAYCALC_H_

#if defined(PHOTOACOUSTICS_USE_GPU) || DOXYGEN

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

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

    void Update();

    void SetInputs(cl_mem usedLines)
    {
      m_UsedLines = usedLines;
    }

  protected:

    OCLDelayCalculation(mitk::BeamformingSettings::Pointer settings);
    virtual ~OCLDelayCalculation();

    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;
    cl_mem m_UsedLines;
    unsigned int m_BufferSize;
    float m_DelayMultiplicatorRaw;
    char m_IsPAImage;
    size_t m_ChunkSize[3];
  };
}
#endif
#endif