animaFastMeanSquaresImageToImageMetric.h

Go to the documentation of this file.

#pragma once

#include "itkImageToImageMetric.h"
#include "itkCovariantVector.h"
#include "itkPoint.h"

namespace anima
{
template < class TFixedImage, class TMovingImage >
class FastMeanSquaresImageToImageMetric :
        public itk::ImageToImageMetric< TFixedImage, TMovingImage>
{
public:

    typedef FastMeanSquaresImageToImageMetric Self;
    typedef itk::ImageToImageMetric<TFixedImage, TMovingImage > Superclass;
    typedef itk::SmartPointer<Self> Pointer;
    typedef itk::SmartPointer<const Self> ConstPointer;

    itkNewMacro(Self);

    itkTypeMacro(FastMeanSquaresImageToImageMetric, itk::ImageToImageMetric);


    typedef typename Superclass::RealType                 RealType;
    typedef typename Superclass::TransformType            TransformType;
    typedef typename Superclass::TransformPointer         TransformPointer;
    typedef typename Superclass::TransformParametersType  TransformParametersType;
    typedef typename Superclass::TransformJacobianType    TransformJacobianType;
    typedef typename Superclass::GradientPixelType        GradientPixelType;
    typedef typename Superclass::OutputPointType          OutputPointType;
    typedef typename Superclass::InputPointType           InputPointType;
    typedef typename itk::ContinuousIndex <double,TFixedImage::ImageDimension> ContinuousIndexType;

    typedef typename Superclass::MeasureType              MeasureType;
    typedef typename Superclass::DerivativeType           DerivativeType;
    typedef typename Superclass::FixedImageType           FixedImageType;
    typedef typename Superclass::MovingImageType          MovingImageType;
    typedef typename Superclass::FixedImageConstPointer   FixedImageConstPointer;
    typedef typename Superclass::MovingImageConstPointer  MovingImageConstPointer;


    void GetDerivative(const TransformParametersType & parameters,
                       DerivativeType & Derivative) const ITK_OVERRIDE;

    MeasureType GetValue(const TransformParametersType & parameters) const ITK_OVERRIDE;

    void GetValueAndDerivative(const TransformParametersType & parameters,
                               MeasureType& Value, DerivativeType& Derivative) const ITK_OVERRIDE;

    itkSetMacro(ScaleIntensities, bool)
    itkSetMacro(DefaultBackgroundValue, double)

    void PreComputeFixedValues();

protected:
    FastMeanSquaresImageToImageMetric();
    virtual ~FastMeanSquaresImageToImageMetric() {}

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

    bool m_ScaleIntensities;
    double m_DefaultBackgroundValue;

    std::vector <InputPointType> m_FixedImagePoints;
    std::vector <RealType> m_FixedImageValues;
};

} // end namespace anima

#include "animaFastMeanSquaresImageToImageMetric.hxx"