diff options
Diffstat (limited to 'vmodels/lin1d.hpp')
-rw-r--r-- | vmodels/lin1d.hpp | 58 |
1 files changed, 58 insertions, 0 deletions
diff --git a/vmodels/lin1d.hpp b/vmodels/lin1d.hpp new file mode 100644 index 0000000..8ca8074 --- /dev/null +++ b/vmodels/lin1d.hpp @@ -0,0 +1,58 @@ +#ifndef VLINEAR_MODEL_H_ +#define VLINEAR_MODEL_H_ +#define OPT_HEADER +#include <core/fitter.hpp> +#include <misc/optvec.hpp> +#include <cmath> + +namespace opt_utilities +{ + template <typename T> + class lin1d + :public model<optvec<T>,optvec<T>,optvec<T>,std::string> + { + typedef optvec<T> Tv; + private: + lin1d<T>* do_clone()const + { + return new lin1d<T>(*this); + } + + const char* do_get_type_name()const + { + return "1d linear model"; + } + public: + lin1d() + { + this->push_param_info(param_info<Tv>("k",1)); + this->push_param_info(param_info<Tv>("b",0)); + } + + public: + Tv do_eval(const Tv& x,const Tv& param) + { + Tv result(x.size()); + + //return x*get_element(param,0)+get_element(param,1); + for(size_t i=0;i!=x.size();++i) + { + result[i]=param[0]*x[i]+param[1]; + } + return result; + } + + private: + std::string do_get_information()const + { + return "<math><mrow> <mtext>f(x;k,b)=k x+b</mtext> \ + </mrow> \ +</math>"; + } + }; +} + + + +#endif +//EOF |