animaModelInitializer.h

Go to the documentation of this file.

#pragma once

#include <itkProcessObject.h>
#include <itkGaussianMembershipFunction.h>

namespace anima
{

class ModelInitializer: public itk::ProcessObject
{
public:

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

    itkNewMacro(Self);

    itkTypeMacro(ModelInitializer, itk::ProcessObject);

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

    std::vector<GaussianFunctionType::Pointer> GetInitialization(){return this->m_GaussianModel;}
    std::vector<double> GetAlphas(){return this->m_Alphas;}

    itkSetMacro(Verbose, bool)
    itkGetMacro(Verbose, bool)

protected:

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

    ModelInitializer()
    {
        m_Verbose = false;
    }
    virtual ~ModelInitializer(){}

    std::vector<double> m_Alphas;

    std::vector<GaussianFunctionType::Pointer> m_GaussianModel;

    bool m_Verbose;
};

}