mitkDiffusionImageCreationFilter.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 DIFFUSIONIMAGECREATIONFILTER_H
#define DIFFUSIONIMAGECREATIONFILTER_H

#include "mitkImageToImageFilter.h"
#include "mitkDiffusionPropertyHelper.h"
#include <MitkDiffusionCoreExports.h>

namespace mitk
{

struct MITKDIFFUSIONCORE_EXPORT DiffusionImageHeaderDescriptor
{
  DiffusionImageHeaderDescriptor()
    : m_GradientDirections(nullptr)
  {
    m_BValue = -1;
  }

  float                                                           m_BValue;
  mitk::DiffusionPropertyHelper::BValueMapType                    m_BValueMapType;
  mitk::DiffusionPropertyHelper::MeasurementFrameType             m_MeasurementFrame;
  mitk::DiffusionPropertyHelper::GradientDirectionsContainerType::Pointer  m_GradientDirections;
};

class MITKDIFFUSIONCORE_EXPORT DiffusionImageCreationFilter
    : public ImageToImageFilter
{
public:
  mitkClassMacro( DiffusionImageCreationFilter,
                  ImageToImageFilter )

  itkNewMacro(Self)

  typedef short                                                           DiffusionPixelType;
  typedef mitk::DiffusionPropertyHelper                                   DPH;

  typedef mitk::Image                                                     OutputType;
  typedef mitk::DiffusionPropertyHelper::ImageType                        VectorImageType;
  typedef mitk::DiffusionPropertyHelper::GradientDirectionType            GradientDirectionType;
  typedef mitk::DiffusionPropertyHelper::MeasurementFrameType             MeasurementFrameType;
  typedef mitk::DiffusionPropertyHelper::GradientDirectionsContainerType  GradientDirectionContainerType;

  void SetReferenceImage( mitk::Image::Pointer reference_image );

  void SetHeaderDescriptor( DiffusionImageHeaderDescriptor header_descriptor );

  virtual void GenerateData() override;
protected:
  DiffusionImageCreationFilter();
  ~DiffusionImageCreationFilter();

  GradientDirectionContainerType::Pointer InternalGetGradientDirections( );
  MeasurementFrameType InternalGetMeasurementFrame();
  float InternalGetBValue();

  VectorImageType::Pointer RemapIntoVectorImage( mitk::Image::Pointer input);

  mitk::Image::Pointer m_ReferenceImage;

  DiffusionImageHeaderDescriptor m_HeaderDescriptor;
  bool m_HeaderDescriptorSet;
};

} //end namespace mitk

#endif // DIFFUSIONIMAGECREATIONFILTER_H