itkVectorPhiPlusDeltaTTimesVFilter.h

Go to the documentation of this file.

/*=============================================================================

  NifTK: A software platform for medical image computing.

  Copyright (c) University College London (UCL). 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 in the top level directory for details.

=============================================================================*/

#ifndef itkVectorPhiPlusDeltaTTimesVFilter_h
#define itkVectorPhiPlusDeltaTTimesVFilter_h

#include <itkVector.h>
#include <itkImage.h>
#include <itkImageToImageFilter.h>
#include <itkVectorLinearInterpolateImageFunction.h>

namespace itk {
template < typename TScalarType, unsigned int NDimensions = 3>
class ITK_EXPORT VectorPhiPlusDeltaTTimesVFilter : 
public ImageToImageFilter<
                           Image< Vector<TScalarType, NDimensions>,  NDimensions>, // Input images
                           Image< Vector<TScalarType, NDimensions>,  NDimensions>  // Output image
                         >
{
  public:

    typedef VectorPhiPlusDeltaTTimesVFilter                                             Self;
    typedef ImageToImageFilter<Image< Vector<TScalarType, NDimensions>,  NDimensions>,
                               Image< Vector<TScalarType, NDimensions>,  NDimensions>
                              >                                                         Superclass;
    typedef SmartPointer<Self>                                                          Pointer;
    typedef SmartPointer<const Self>                                                    ConstPointer;

    typedef Vector< TScalarType, NDimensions >                                          InputPixelType;
    typedef Image< InputPixelType, NDimensions >                                        InputImageType;
    typedef Image< TScalarType, NDimensions >                                           InputScalarImageType;
    typedef typename InputImageType::IndexType                                          InputImageIndexType;
    typedef typename InputImageType::RegionType                                         InputImageRegionType;
    typedef InputPixelType                                                              OutputPixelType;
    typedef InputImageType                                                              OutputImageType;
    typedef VectorLinearInterpolateImageFunction< InputImageType, 
                                                  TScalarType >                         VectorLinearInterpolatorType;

    itkNewMacro(Self);

    itkTypeMacro(VectorPhiPlusDeltaTTimesVFilter, ImageToImageFilter);

    itkStaticConstMacro(Dimension, unsigned int, NDimensions);

    itkSetMacro(SubtractSteps, bool);
    itkGetMacro(SubtractSteps, bool);

    itkSetMacro(DeltaT, double);
    itkGetMacro(DeltaT, double);
    
    itkSetMacro(NumberOfSteps, unsigned int);
    itkGetMacro(NumberOfSteps, unsigned int);
    
    void SetTimeZeroTransformation(InputImageType* image) { m_PhiZeroTransformation = image; }
    
    void SetThicknessPrior(InputScalarImageType* image) { m_ThicknessPriorImage = image; }
    
    InputScalarImageType* GetThicknessImage() { return m_ThicknessImage.GetPointer(); }
    
  protected:
    VectorPhiPlusDeltaTTimesVFilter();
    ~VectorPhiPlusDeltaTTimesVFilter() {};
    void PrintSelf(std::ostream& os, Indent indent) const;
    
    // Check before we start.
    virtual void BeforeThreadedGenerateData();
    
    // The main method to implement in derived classes, note, its threaded.
    virtual void ThreadedGenerateData( const InputImageRegionType &outputRegionForThread, ThreadIdType threadId);
    
  private:
    
    VectorPhiPlusDeltaTTimesVFilter(const Self&); 
    void operator=(const Self&); 
    
    bool m_SubtractSteps;
    
    double m_DeltaT;
    
    unsigned int m_NumberOfSteps;
    
    typename VectorLinearInterpolatorType::Pointer m_Interpolator;
    
    typename InputImageType::Pointer m_PhiZeroTransformation;
    
    typename InputScalarImageType::Pointer m_ThicknessPriorImage;
    
    typename InputScalarImageType::Pointer m_ThicknessImage;
    
}; // end class

} // end namespace

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

#endif