mitkDiffusionImageCorrectionFilter.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 MITKDIFFUSIONIMAGECORRECTIONFILTER_H
#define MITKDIFFUSIONIMAGECORRECTIONFILTER_H

#include "mitkImageSource.h"
#include <MitkDiffusionCoreExports.h>

namespace mitk
{
class MITKDIFFUSIONCORE_EXPORT DiffusionImageCorrectionFilter
        : public ImageSource
{
public:
  mitkClassMacro( DiffusionImageCorrectionFilter,
                  ImageSource )

  itkSimpleNewMacro(Self)

  typedef short DiffusionPixelType;
  typedef vnl_vector_fixed< double, 3 >      GradientDirectionType;
  typedef vnl_matrix_fixed< double, 3, 3 >   TransformMatrixType;
  typedef itk::VectorContainer< unsigned int, GradientDirectionType >
                                             GradientDirectionContainerType;
  typedef GradientDirectionContainerType::Pointer GradientDirectionContainerPointerType;

  typedef std::vector< TransformMatrixType >  TransformsVectorType;

  typedef Superclass::OutputType DiffusionImageType;
  typedef DiffusionImageType::Pointer    DiffusionImageTypePointer;
  typedef itk::VectorImage<DiffusionPixelType,3>     ImageType;

  void SetImage( DiffusionImageTypePointer input )
  {
    m_SourceImage = input;
  }

  void CorrectDirections( const TransformsVectorType& );

  void CorrectDirections( const TransformMatrixType& );

  virtual void GenerateOutputInformation() override {}


protected:
  DiffusionImageCorrectionFilter();
  virtual ~DiffusionImageCorrectionFilter() {}

  TransformMatrixType GetRotationComponent( const TransformMatrixType& );

  DiffusionImageTypePointer m_SourceImage;
};

}

#endif // MITKDIFFUSIONIMAGECORRECTIONFILTER_H