animaMCMPairingMeanSquaresImageToImageMetric.h

Go to the documentation of this file.

#pragma once

#include <animaBaseOrientedModelImageToImageMetric.h>
#include <animaMultiCompartmentModel.h>
#include <animaMCMImage.h>
#include <animaBaseTensorTools.h>

namespace anima
{

template < class TFixedImagePixelType, class TMovingImagePixelType, unsigned int ImageDimension >
class MCMPairingMeanSquaresImageToImageMetric :
public BaseOrientedModelImageToImageMetric< anima::MCMImage < TFixedImagePixelType, ImageDimension >, anima::MCMImage < TMovingImagePixelType, ImageDimension > >
{
public:
    typedef anima::MCMImage < TFixedImagePixelType, ImageDimension > TFixedImage;
    typedef anima::MCMImage < TMovingImagePixelType, ImageDimension > TMovingImage;

    typedef MCMPairingMeanSquaresImageToImageMetric Self;
    typedef BaseOrientedModelImageToImageMetric<TFixedImage, TMovingImage > Superclass;
    typedef itk::SmartPointer<Self> Pointer;
    typedef itk::SmartPointer<const Self> ConstPointer;

    typedef anima::MultiCompartmentModel MCModelType;
    typedef typename MCModelType::Pointer MCModelPointer;
    typedef typename MCModelType::Vector3DType GradientType;

    itkNewMacro(Self)

    
    itkTypeMacro(MCMPairingMeanSquaresImageToImageMetric, BaseOrientedModelImageToImageMetric)

    
    typedef typename TFixedImage::PixelType               PixelType;

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

    typedef typename Superclass::CoordinateRepresentationType CoordinateRepresentationType;

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

    using LECalculatorType = anima::LogEuclideanTensorCalculator <double>;
    using LECalculatorPointer = typename LECalculatorType::Pointer;

    MeasureType GetValue( const TransformParametersType & parameters ) const ITK_OVERRIDE;

    itkSetMacro(OneToOneMapping, bool)

    void PreComputeFixedValues();

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

    bool CheckTensorCompatibility() const;
    double ComputeTensorBasedMetricPart(unsigned int index, const MCModelPointer &movingValue) const;
    double ComputeNonTensorBasedMetricPart(unsigned int index, const MCModelPointer &movingValue) const;

    bool isZero(PixelType &vector) const;

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

    MCModelPointer m_ZeroDiffusionModel;
    PixelType m_ZeroDiffusionVector;

    std::vector <InputPointType> m_FixedImagePoints;
    std::vector <MCModelPointer> m_FixedImageValues;

    bool m_OneToOneMapping;

    LECalculatorPointer m_leCalculator;
};

} // end namespace anima

#include "animaMCMPairingMeanSquaresImageToImageMetric.hxx"