itkRegularizedIVIMReconstructionSingleIteration.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 __itkRegularizedIVIMReconstructionSingleIteration_h
#define __itkRegularizedIVIMReconstructionSingleIteration_h

#include "itkImageToImageFilter.h"
#include "itkImage.h"
#include "itkVectorImage.h"

namespace itk
{
template <class TInputPixel, class TOutputPixel, class TRefPixelType>
class RegularizedIVIMReconstructionSingleIteration :
    public ImageToImageFilter< itk::Image<itk::Vector<TInputPixel,3>, 3>, itk::Image<itk::Vector<TOutputPixel,3>, 3> >
{
public:
  typedef itk::Image<itk::Vector<TInputPixel,3>, 3> InputImageType;
  typedef itk::Image<itk::Vector<TOutputPixel,3>, 3> OutputImageType;
  typedef itk::VectorImage<TRefPixelType,3> RefImageType;

  itkStaticConstMacro(InputImageDimension, unsigned int,
                      InputImageType::ImageDimension);
  itkStaticConstMacro(OutputImageDimension, unsigned int,
                      OutputImageType::ImageDimension);

  typedef itk::Image<float,InputImageDimension> LocalVariationImageType;

  typedef RegularizedIVIMReconstructionSingleIteration Self;
  typedef ImageToImageFilter< InputImageType, OutputImageType> Superclass;
  typedef SmartPointer<Self> Pointer;
  typedef SmartPointer<const Self>  ConstPointer;

  itkFactorylessNewMacro(Self)
  itkCloneMacro(Self)

  itkTypeMacro(RegularizedIVIMReconstructionSingleIteration, ImageToImageFilter);

  typedef typename InputImageType::PixelType InputPixelType;
  typedef typename OutputImageType::PixelType OutputPixelType;

  typedef typename InputImageType::RegionType InputImageRegionType;
  typedef typename OutputImageType::RegionType OutputImageRegionType;

  typedef typename InputImageType::SizeType InputSizeType;

  virtual void GenerateInputRequestedRegion()
    throw(InvalidRequestedRegionError);

  itkSetMacro(Lambda, double);
  itkGetMacro(Lambda, double);

  void SetBValues(vnl_vector<double> bvals)
  { this->m_BValues = bvals; }
  vnl_vector<double> GetBValues()
  { return this->m_BValues; }

  void SetOriginalImage(RefImageType* in)
  { this->m_OriginalImage = in; }
  typename RefImageType::Pointer GetOriginialImage()
  { return this->m_OriginalImage; }

protected:
  RegularizedIVIMReconstructionSingleIteration();
  virtual ~RegularizedIVIMReconstructionSingleIteration() {}
  void PrintSelf(std::ostream& os, Indent indent) const;

  void ThreadedGenerateData(const OutputImageRegionType& outputRegionForThread,
                            ThreadIdType threadId );

  void BeforeThreadedGenerateData();

  typename LocalVariationImageType::Pointer m_LocalVariation;

  typename RefImageType::Pointer m_OriginalImage;

  double m_Lambda;

  vnl_vector<double> m_BValues;

private:

  RegularizedIVIMReconstructionSingleIteration(const Self&);

  void operator=(const Self&);

};

} // end namespace itk

#ifndef ITK_MANUAL_INSTANTIATION
#include "itkRegularizedIVIMReconstructionSingleIteration.txx"
#endif

#endif //__itkRegularizedIVIMReconstructionSingleIteration__