diff options
Diffstat (limited to 'models/bl1d.hpp')
-rw-r--r-- | models/bl1d.hpp | 56 |
1 files changed, 56 insertions, 0 deletions
diff --git a/models/bl1d.hpp b/models/bl1d.hpp new file mode 100644 index 0000000..bd0ce7d --- /dev/null +++ b/models/bl1d.hpp @@ -0,0 +1,56 @@ +#ifndef BROKEN_LINE_MODEL_H_ +#define BROKEN_LINE_MODEL_H_ +#include <core/fitter.hpp> +#include <cmath> + +namespace opt_utilities +{ + template <typename T> + class bl1d + :public model<T,T,std::vector<T>,std::string> + { + private: + model<T,T,std::vector<T> >* do_clone()const + { + return new bl1d<T>(*this); + } + public: + bl1d() + { + this->push_param_info(param_info<std::vector<T> >("break point y value",1)); + this->push_param_info(param_info<std::vector<T> >("break point x value",1)); + this->push_param_info(param_info<std::vector<T> >("slop 1",1)); + this->push_param_info(param_info<std::vector<T> >("slop 2",1)); + } + + public: + T do_eval(const T& x,const std::vector<T>& param) + { + T x_b=get_element(param,0); + T f_b=get_element(param,1); + T k1=get_element(param,2); + T k2=get_element(param,3); + if(x<x_b) + { + return k1*(x-x_b)+f_b; + } + else + { + return k2*(x-x_b)+f_b; + } + } + + private: + std::string do_to_string()const + { + return "broken linear model\n" + "y=k1*(x-x_b)+y_b for x<x_b\n" + "y=k2*(x-x_b)+y_b otherwise\n"; + } + }; +}; + + + +#endif +//EOF |