animaRandomInitializer.h

Go to the documentation of this file.

#pragma once

#include "AnimaGraphCutSegmentationExport.h"
#include "animaModelInitializer.h"
#include "time.h"

namespace anima
{

class ANIMAGRAPHCUTSEGMENTATION_EXPORT RandomInitializer: public ModelInitializer
{
public:

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

    itkNewMacro(Self)

    
    itkTypeMacro(RandomInitializer, itk::ProcessObject)

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

    void Update() ITK_OVERRIDE;

    void SetMinValues(std::vector<double> &min){this->minValues=min;}

    void SetMaxValues(std::vector<double> &max){this->maxValues=max;}

    itkSetMacro(NbGaussian, unsigned int)
    itkGetMacro(NbGaussian, unsigned int)

    itkSetMacro(DimensionGaussian, unsigned int)
    itkGetMacro(DimensionGaussian, unsigned int)

protected:
    RandomInitializer(const Self&); //purposely not implemented
    void operator=(const Self&); //purposely not implemented

    RandomInitializer()
    {
        srand(time(NULL));

        m_NbGaussian = 1;
        m_DimensionGaussian= 1;

    }
    virtual ~RandomInitializer(){}

    itk::Statistics::GaussianMembershipFunction<itk::VariableLengthVector<double> >::Pointer randomDistribution();

    double randUniform(double min,double max);


    std::vector<double> minValues;

    std::vector<double> maxValues;

    unsigned int m_NbGaussian;
    unsigned int m_DimensionGaussian;

};

}