animaSVFExponentialImageFilter.h

Go to the documentation of this file.

#pragma once

#include <itkImageToImageFilter.h>

namespace anima
{

template <typename TPixelType, unsigned int Dimension>
class SVFExponentialImageFilter :
public itk::ImageToImageFilter< itk::Image <itk::Vector <TPixelType, Dimension>, Dimension> ,
        itk::Image <itk::Vector <TPixelType, Dimension>, Dimension> >
{
public:
    typedef SVFExponentialImageFilter Self;
    typedef typename itk::Image <itk::Vector <TPixelType, Dimension>, Dimension> InputImageType;
    typedef typename itk::Image <itk::Vector <TPixelType, Dimension * Dimension>, Dimension> JacobianImageType;
    typedef typename itk::Image <itk::Vector <TPixelType, Dimension>, Dimension> OutputImageType;
    typedef itk::ImageToImageFilter <InputImageType, OutputImageType> Superclass;
    typedef itk::SmartPointer<Self> Pointer;
    typedef itk::SmartPointer<const Self> ConstPointer;

    itkNewMacro(Self)

    itkTypeMacro(SVFExponentialImageFilter, itk::ImageToImageFilter)

    typedef typename InputImageType::PixelType InputPixelType;
    typedef typename OutputImageType::PixelType OutputPixelType;
    typedef typename JacobianImageType::Pointer JacobianImagePointer;
    typedef typename JacobianImageType::PixelType JacobianPixelType;

    typedef typename Superclass::OutputImageRegionType OutputImageRegionType;

    itkSetMacro(ExponentiationOrder, unsigned int)
    itkSetMacro(MaximalDisplacementAmplitude, double)

protected:
    SVFExponentialImageFilter()
    {
        m_ExponentiationOrder = 0;
        m_MaximalDisplacementAmplitude = 0.25;
        m_FieldJacobian = 0;
    }

    virtual ~SVFExponentialImageFilter() {}

    void BeforeThreadedGenerateData() ITK_OVERRIDE;
    void DynamicThreadedGenerateData(const OutputImageRegionType &outputRegionForThread) ITK_OVERRIDE;
    void AfterThreadedGenerateData() ITK_OVERRIDE;

private:
    ITK_DISALLOW_COPY_AND_ASSIGN(SVFExponentialImageFilter);

    double m_MaximalDisplacementAmplitude;

    double m_ExponentiationOrder;

    JacobianImagePointer m_FieldJacobian;

    unsigned int m_NumberOfSquarings;
};

} // end namespace anima

#include "animaSVFExponentialImageFilter.hxx"