itkDiffusionModelEstimationImageFilter.h

Go to the documentation of this file.

#ifndef __itkDiffusionModelEstimationImageFilter_h
#define __itkDiffusionModelEstimationImageFilter_h


#include "itkImage.h"
#include "itkVectorImage.h"
#include "itkMaskedImageToImageFilter.h"

#include "utlITK.h"
#include "utlNDArray.h"

#include "itkSamplingSchemeQSpace.h"

namespace itk
{

template < class TInputImage, class TOutputImage >
class ITK_EXPORT DiffusionModelEstimationImageFilter
  : public MaskedImageToImageFilter<TInputImage, TOutputImage>
{
public:
  typedef DiffusionModelEstimationImageFilter         Self;
  typedef MaskedImageToImageFilter<TInputImage,TOutputImage>  Superclass;
  typedef SmartPointer<Self>                            Pointer;
  typedef SmartPointer<const Self>                      ConstPointer;

  itkNewMacro(Self);

  itkTypeMacro( DiffusionModelEstimationImageFilter, MaskedImageToImageFilter );
  
  itkTypedefMaskedImageToImageMacro(Superclass);

  typedef utl::NDArray<double,2>                  MatrixType;
  typedef utl::NDArray<double,1>                  VectorType;
  typedef utl_shared_ptr<MatrixType>              MatrixPointer;
  typedef utl_shared_ptr<VectorType>              VectorPointer;
  typedef std::vector<double>                     STDVectorType;
  typedef utl_shared_ptr<STDVectorType >          STDVectorPointer;
  
  typedef SamplingSchemeQSpace<double>                       SamplingSchemeQSpaceType;
  typedef typename SamplingSchemeQSpaceType::Pointer         SamplingSchemeQSpacePointer;
  
  typedef SamplingScheme3D<double>                           SamplingSchemeRSpaceType;
  typedef typename SamplingSchemeRSpaceType::Pointer         SamplingSchemeRSpacePointer;
  
  itkSetObjectMacro(SamplingSchemeQSpace, SamplingSchemeQSpaceType);
  itkGetObjectMacro(SamplingSchemeQSpace, SamplingSchemeQSpaceType);

  // itkSetMacro(BasisMatrix, MatrixPointer);
  itkGetMacro(BasisMatrix, MatrixPointer);
  
  itkSetMacro(MD0, double);
  itkGetMacro(MD0, double);

  virtual void ComputeBasisMatrix ()
    {}
  
  virtual void ComputeRegularizationWeight ( )
    {}

protected:
  DiffusionModelEstimationImageFilter(); 

  virtual ~DiffusionModelEstimationImageFilter() {};
  
  virtual void VerifyInputParameters() const;

  void PrintSelf(std::ostream& os, Indent indent) const ITK_OVERRIDE;
  typename LightObject::Pointer InternalClone() const ITK_OVERRIDE;
  
  SamplingSchemeQSpacePointer m_SamplingSchemeQSpace;
  
  double m_MD0;

  MatrixPointer  m_BasisMatrix;

  VectorPointer  m_RegularizationWeight;

private:
  DiffusionModelEstimationImageFilter(const Self&);  //purposely not implemented
  void operator=(const Self&);  //purposely not implemented

  
};

}

#if ITK_TEMPLATE_EXPLICIT
# include "Templates/itkDiffusionModelEstimationImageFilter+-.h"
#endif

#if !defined(ITK_MANUAL_INSTANTIATION) && !defined(__itkDiffusionModelEstimationImageFilter_hxx)
#include "itkDiffusionModelEstimationImageFilter.hxx"
#endif


#endif