mitkPhotoacousticOCLBeamformingFilter.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 _MITKPHOTOACOUSTICSOCLBEAMFORMER_H_
#define _MITKPHOTOACOUSTICSOCLBEAMFORMER_H_

#include <itkObject.h>

#if defined(PHOTOACOUSTICS_USE_GPU) || DOXYGEN

#include "mitkOclDataSetToDataSetFilter.h"
#include "mitkBeamformingSettings.h"
#include "mitkPhotoacousticOCLDelayCalculation.h"
#include "mitkPhotoacousticOCLUsedLinesCalculation.h"

#include <chrono>

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

    void SetInput(Image::Pointer image);
    void SetInput(void* data, unsigned int* dimensions, unsigned int BpE);
    void* GetOutput();

    mitk::Image::Pointer GetOutputAsImage();

    void Update();

    void SetApodisation(const float* apodisation, unsigned short apodArraySize)
    {
      m_ApodArraySize = apodArraySize;
      m_Apodisation = apodisation;
    }

  protected:

    PhotoacousticOCLBeamformingFilter(BeamformingSettings::Pointer settings);
    virtual ~PhotoacousticOCLBeamformingFilter();

    bool Initialize();

    void UpdateDataBuffers();

    void Execute();

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

    int GetBytesPerElem()
    {
      return sizeof(float);
    }

    virtual us::Module* GetModule();

  private:
    cl_kernel m_PixelCalculation;

    unsigned int m_OutputDim[3];

    const float* m_Apodisation;
    unsigned short m_ApodArraySize;
    unsigned int m_inputSlices;

    unsigned short m_PAImage;

    BeamformingSettings::Pointer m_Conf;

    mitk::Image::Pointer m_InputImage;

    size_t m_ChunkSize[3];

    mitk::OCLUsedLinesCalculation::Pointer m_UsedLinesCalculation;
    mitk::OCLDelayCalculation::Pointer m_DelayCalculation;

    cl_mem m_ApodizationBuffer;
    cl_mem m_DelaysBuffer;
    cl_mem m_UsedLinesBuffer;
    cl_mem m_ElementHeightsBuffer;
    cl_mem m_ElementPositionsBuffer;

  };
}
#else
namespace mitk
{
  class PhotoacousticOCLBeamformingFilter : public itk::Object
  {
  public:
    mitkClassMacroItkParent(mitk::PhotoacousticOCLBeamformingFilter, itk::Object);
    itkNewMacro(Self);

  protected:
    PhotoacousticOCLBeamformingFilter() {}

    ~PhotoacousticOCLBeamformingFilter() override {}
  };
}
#endif
#endif