00001
00002
00003
00004
00005
00006
00007
00008
00009
00010
00011
00012
00013
00014
00015
00016
00017 #ifndef __itkCompositeValleyFunction_h
00018 #define __itkCompositeValleyFunction_h
00019
00020 #include "itkArray.h"
00021 #include "itkCacheableScalarFunction.h"
00022 #include <vector>
00023
00024 namespace itk {
00025
00071 class TargetClass
00072 {
00073 public:
00075 TargetClass(double mean, double sigma)
00076 {
00077 m_Mean = mean ;
00078 m_Sigma = sigma ;
00079 }
00080
00082 void SetMean(double mean) { m_Mean = mean ; }
00083 double GetMean() { return m_Mean ; }
00084
00086 void SetSigma(double sigma) { m_Sigma = sigma ; }
00087 double GetSigma() { return m_Sigma ; }
00088
00089 private:
00090 double m_Mean ;
00091 double m_Sigma ;
00092 } ;
00093
00094 class CompositeValleyFunction : public CacheableScalarFunction
00095 {
00096 public:
00097
00099 typedef CacheableScalarFunction Superclass;
00100
00102 typedef Superclass::MeasureType MeasureType;
00103 typedef Superclass::MeasureArrayType MeasureArrayType;
00104
00106 CompositeValleyFunction( const MeasureArrayType & classMeans,
00107 const MeasureArrayType & classSigmas );
00108
00110 virtual ~CompositeValleyFunction() {}
00111
00113 double GetUpperBound() { return m_UpperBound ; }
00114
00116 double GetLowerBound() { return m_LowerBound ; }
00117
00120 MeasureType operator() (MeasureType x)
00121 {
00122 if (x > m_UpperBound || x < m_LowerBound) { return 1; }
00123
00124 if (!this->IsCacheAvailable())
00125 { return this->Evaluate(x); }
00126 else
00127 { return GetCachedValue(x); }
00128 }
00129
00131 inline MeasureType Evaluate(MeasureType x)
00132 {
00133 MeasureType res = 1;
00134
00135 for (unsigned int k = 0 ; k < m_Targets.size() ; k++)
00136 {
00137 res *= valley( ( x - m_Targets[k].GetMean() ) /
00138 m_Targets[k].GetSigma() );
00139 }
00140
00141 return res;
00142 }
00143
00145 inline MeasureType valley(MeasureType d)
00146 { return 1 - 1 / (1+d*d/3); }
00147
00148 protected:
00149 void AddNewClass(double mean, double sigma)
00150 {
00151 TargetClass aClass(mean, sigma) ;
00152 m_Targets.push_back(aClass) ;
00153 }
00154
00156 void Initialize() ;
00157
00158 private:
00160 std::vector<TargetClass> m_Targets;
00161
00164 double m_UpperBound ;
00165
00168 double m_LowerBound ;
00169
00170 } ;
00171
00172 }
00173 #endif