00001
00002
00003
00004
00005
00006
00007
00008
00009
00010
00011
00012
00013
00014
00015
00016
00017 #ifndef __itkLevenbergMarquardtOptimizer_h
00018 #define __itkLevenbergMarquardtOptimizer_h
00019
00020 #include "itkMultipleValuedNonLinearVnlOptimizer.h"
00021 #include "vnl/algo/vnl_levenberg_marquardt.h"
00022
00023 namespace itk
00024 {
00025
00031 class ITK_EXPORT LevenbergMarquardtOptimizer :
00032 public MultipleValuedNonLinearVnlOptimizer
00033 {
00034 public:
00036 typedef LevenbergMarquardtOptimizer Self;
00037 typedef MultipleValuedNonLinearVnlOptimizer Superclass;
00038 typedef SmartPointer<Self> Pointer;
00039 typedef SmartPointer<const Self> ConstPointer;
00040
00042 itkNewMacro(Self);
00043
00045 itkTypeMacro( LevenbergMarquardtOptimizer, MultipleValuedNonLinearVnlOptimizer );
00046
00048 typedef vnl_vector<double> InternalParametersType;
00049
00051 typedef vnl_levenberg_marquardt InternalOptimizerType;
00052
00054 vnl_levenberg_marquardt * GetOptimizer(void);
00055
00057 void StartOptimization( void );
00058
00060 virtual void SetCostFunction( MultipleValuedCostFunction * costFunction );
00061
00062 void SetNumberOfIterations(unsigned int iterations);
00063 void SetValueTolerance(double tol);
00064 void SetGradientTolerance(double tol);
00065 void SetEpsilonFunction(double epsilon);
00066
00067 protected:
00068 LevenbergMarquardtOptimizer();
00069 virtual ~LevenbergMarquardtOptimizer();
00070
00071 typedef Superclass::CostFunctionAdaptorType CostFunctionAdaptorType;
00072
00073 private:
00074 LevenbergMarquardtOptimizer(const Self&);
00075 void operator=(const Self&);
00076
00077 bool m_OptimizerInitialized;
00078 InternalOptimizerType * m_VnlOptimizer;
00079 unsigned int m_NumberOfIterations;
00080 double m_ValueTolerance;
00081 double m_GradientTolerance;
00082 double m_EpsilonFunction;
00083
00084 };
00085
00086 }
00087
00088
00089 #endif
00090
00091
00092