mitkDiffusionImageTransformedCreationFilter.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 DIFFUSIONIMAGETRANSFORMEDCREATIONFILTER_H
#define DIFFUSIONIMAGETRANSFORMEDCREATIONFILTER_H

#include "mitkDiffusionPropertyHelper.h"
#include "mitkDiffusionImageCreationFilter.h"
#include "mitkDiffusionImageCorrectionFilter.h"

#include <MitkDiffusionCoreExports.h>


namespace mitk
{


template <typename TTransformType >
class MITKDIFFUSIONCORE_EXPORT DiffusionImageTransformedCreationFilter
    : public ImageToImageFilter
{
public:
  mitkClassMacro( DiffusionImageTransformedCreationFilter,
                  ImageToImageFilter )

  itkNewMacro(Self)

  enum InterpolatorLevel
  {
    NearestNeighbor = 0,
    Trilinear,
    BSpline_Bicubic,
    WindowedSinc
  };

  typedef std::vector< typename TTransformType::Pointer >           TransformContainerType;
  typedef std::vector< typename TTransformType::ParametersType >    TransformParametersContainerType;

  void SetTransforms( const TransformContainerType& );

  void SetTransformParameters( const TransformParametersContainerType& );

  void SetInterpolationLevel( InterpolatorLevel lvl)
  {
    m_InterpolationLevel = lvl;
  }

  void GenerateData() override;

  void SetResamplingReferenceImage( mitk::Image::Pointer image )
  {
    this->m_ResamplingReferenceImage = image;
  }

  void SetReferenceDiffusionImage( mitk::Image::Pointer dwimage )
  {
    this->m_DiffusionReferenceImage = dwimage;
  }

  void SetOutputPrefix( std::string prefix )
{
  this->m_OutputPrefix = prefix;
}

  virtual void GenerateOutputInformation() override;

protected:
  DiffusionImageTransformedCreationFilter();
  ~DiffusionImageTransformedCreationFilter();

  DiffusionImageHeaderDescriptor GetTransformedHeaderInformation();

  TransformContainerType m_InternalTransforms;

  mitk::Image::Pointer m_ResamplingReferenceImage;

  mitk::Image::Pointer m_DiffusionReferenceImage;

  InterpolatorLevel m_InterpolationLevel;

  typedef DiffusionImageCorrectionFilter::TransformMatrixType MatrixType;
  std::vector< MatrixType > m_RotationMatrices;

  std::string m_OutputPrefix;

  mitk::Image::Pointer m_OutputCache;
};

} // end namespace mitk

#ifndef ITK_MANUAL_INSTANTIATION
  #include "mitkDiffusionImageTransformedCreationFilter.cxx"
#endif

#endif // DIFFUSIONIMAGETRANSFORMEDCREATIONFILTER_H