aboutsummaryrefslogtreecommitdiffstats
path: root/models/bl1d.hpp
diff options
context:
space:
mode:
Diffstat (limited to 'models/bl1d.hpp')
-rw-r--r--models/bl1d.hpp56
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