git-svn-id: file:///home/svn/opt_utilities@1 ed2142bd-67ad-457f-ba7c-d818d4011675

1 files changed, 67 insertions, 0 deletions

diff --git a/models/poly1d.hpp b/models/poly1d.hpp
new file mode 100644
index 0000000..a8c994b
--- /dev/null
+++ b/models/poly1d.hpp
@@ -0,0 +1,67 @@
+#ifndef POLY_MODEL_H_
+#define POLY_MODEL_H_
+#include <core/fitter.hpp>
+#include <cmath>
+#include <sstream>
+#include <cassert>
+
+namespace opt_utilities
+{
+  template <typename T,int n>
+  class poly1d
+    :public model<T,T,std::vector<T>,std::string>
+  {
+  private:
+    model<T,T,std::vector<T> >* do_clone()const
+    {
+      return new poly1d<T,n>(*this);
+    }
+  public:
+    poly1d()
+    {
+      assert(n>=0);
+      for(int i=0;i<=n;++i)
+	{
+	  std::ostringstream ostr;
+	  ostr<<"a"<<i;
+	  this->push_param_info(param_info<std::vector<T> >(ostr.str().c_str(),1));  
+	}
+      
+    }
+
+  public:
+    T do_eval(const T& x,const std::vector<T>& param)
+    {
+      //      return x*get_element(param,0)+get_element(param,1);
+      T result(0);
+      for(int i=0;i<=n;++i)
+	{
+	  T xn(1);
+	  for(int j=0;j<i;++j)
+	    {
+	      xn*=x;
+	    }
+	  result+=get_element(param,i)*xn;
+	}
+      return result;
+    }
+
+  private:
+    std::string do_to_string()const
+    {
+      std::ostringstream ostr;
+      ostr<<n<<"-order polynorminal model\n";
+      for(int i=0;i<n;++i)
+	{
+	  ostr<<"a"<<i<<"*x^"<<i<<"+";
+	}
+      ostr<<"a"<<n<<"*x^"<<n;
+      return ostr.str();
+    }
+  };
+};
+
+
+
+#endif
+//EOF