mitkOptimizer.h

/*=========================================================================

  Program:   3DMed
  Date:      $Date: 2014-02-25 18:30:00 +0800 $
  Version:   $Version: 4.6.0 $
  Copyright: MIPG, Institute of Automation, Chinese Academy of Sciences

=========================================================================*/


#ifndef __mitkOptimizer_h
#define __mitkOptimizer_h

#include "mitkRegistrationIncludes.h"
#include "mitkProcessObject.h"
#include "mitkMetric.h"


class MITK_REGISTRATION_API mitkOptimizer : public mitkProcessObject
{
public:
    MITK_TYPE(mitkOptimizer, mitkProcessObject) 
    virtual void PrintSelf(ostream &os);

    virtual void SetInitialParameters(VectorParameterType & param ) {m_InitialParameters = param;}

    virtual VectorParameterType& GetInitialParameters();

    void SetScales(VectorParameterType& scales);

    void SetScales(ScalarParameterType* scales, unsigned int length);

    VectorParameterType & GetScales();

    virtual VectorParameterType & GetLastParameters();

    void SetMetric(mitkMetric *metric) {m_Metric = metric;}

    mitkMetric* GetMetric();

    void SetMaxIterations(unsigned int n) {m_MaxIterations = n;}

    unsigned int GetMaxIterationNumber() { return m_MaxIterations; }

    virtual void Update();

    void SetNumberOfParameters(unsigned int num){ m_NumberOfParameters = num; }

    unsigned int GetNumberOfParameters() { return m_NumberOfParameters; }

    void SetSpaceDimension(unsigned int dim) { m_SpaceDimension = dim; }

    unsigned int GetSpaceDimension() { return m_SpaceDimension; }

    void SetCurrentIterationNumber(unsigned int iterations) { m_Iterations = iterations; }

    unsigned int GetCurrentIterationNumber() { return m_Iterations; }

    VectorParameterType &   GetCurrentPosition() { return m_CurrentPosition; }

    void GetCurrentPosition(double* currentPosition);

    void SetMaximizeFlag(bool flag) { m_FlagMaximize = flag; }

    bool GetMaximizeFlag() { return m_FlagMaximize; }

    void SetStopFlag(bool flag) { m_FlagStop = flag; }

    bool GetStopFlag() { return m_FlagStop; }

    enum StopCondition{ Tolerance, StepTooSmall, Terminated, ParameterSizeErr, 
        GetValueErr, MaxIterations, StopByUser, RoutineFail};

    void StopOptimization( StopCondition condition );

protected:
    mitkOptimizer();
    virtual ~mitkOptimizer();
    virtual bool Execute();

    mitkRCPtr<mitkMetric>                       m_Metric;

    VectorParameterType                         m_InitialParameters;
    VectorParameterType                         m_LastParameters;
    VectorParameterType                         m_Scales;
    VectorParameterType                         m_CurrentPosition;

    unsigned int                                m_MaxIterations;
    unsigned int                                m_Iterations;
    unsigned int                                m_NumberOfParameters;
    unsigned int                                m_SpaceDimension;

    bool                                        m_FlagMaximize;
    bool                                        m_FlagStop;
    
private:
    mitkOptimizer(const mitkOptimizer&);
    void operator = (const mitkOptimizer&);
};


//#define DEFINED_mitkOptimizer



#endif

---