animaGaussianREMEstimator.h

Go to the documentation of this file.

#pragma once

#include "animaGaussianEMEstimator.h"
#include "itkProcessObject.h"

namespace anima
{

template <typename TInputImage, typename TMaskImage>
class GaussianREMEstimator : public GaussianEMEstimator<TInputImage,TMaskImage>
{
public:

    typedef GaussianREMEstimator  Self;
    typedef itk::ProcessObject Superclass;
    typedef itk::SmartPointer <Self> Pointer;
    typedef itk::SmartPointer <const Self> ConstPointer;

    itkNewMacro(Self);

    itkTypeMacro(GaussianREMEstimator, itk::ProcessObject);

    typedef double                    NumericType;
    typedef itk::VariableSizeMatrix< NumericType >::InternalMatrixType DoubleVariableSizeMatrixVnlType;

    typedef double Ocurrences;
    typedef unsigned short MeasureType;
    typedef std::vector<MeasureType> Intensities;
    typedef std::map< Intensities, std::vector<Ocurrences> > GenericContainer;
    typedef std::map<Intensities,Ocurrences> Histogram;
    typedef std::map<double,std::vector<unsigned short> > ResidualMap;

    typedef itk::VariableLengthVector<double> MeasurementVectorType;
    typedef itk::Statistics::GaussianMembershipFunction< MeasurementVectorType > GaussianFunctionType;

    virtual void Update() ITK_OVERRIDE;

    virtual bool concentration();

    int PrintSolution(std::vector<double> alphas, std::vector<GaussianFunctionType::Pointer> model);


    Histogram GetConcentrationJointHistogram(){return this->m_JointHistogram;}

    itkSetMacro(RejectionRatio, double);
    itkGetMacro(RejectionRatio, double);

    itkSetMacro(MaxIterationsConc, int);
    itkGetMacro(MaxIterationsConc, int);

    itkSetMacro(StremMode, bool);
    itkGetMacro(StremMode, bool);


protected:

    GaussianREMEstimator(const Self&); //purposely not implemented
    void operator=(const Self&); //purposely not implemented

    GaussianREMEstimator(): GaussianEMEstimator<TInputImage,TMaskImage>()
    {
        this->m_ModelMinDistance = 1e-4;
        this->m_MaxIterations = 100;

        this->m_MaxIterationsConc = 1;
        this->m_StremMode=false;
    }

    virtual ~GaussianREMEstimator(){}

    double m_RejectionRatio;

    int m_MaxIterationsConc;

    bool m_StremMode;

    Histogram m_OriginalJointHistogram;

};

}

#include "animaGaussianREMEstimator.hxx"