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

1 files changed, 52 insertions, 0 deletions

diff --git a/pre_estimaters/vgauss1d_estimater.hpp b/pre_estimaters/vgauss1d_estimater.hpp
new file mode 100644
index 0000000..e3d1526
--- /dev/null
+++ b/pre_estimaters/vgauss1d_estimater.hpp
@@ -0,0 +1,52 @@
+#ifndef VGAUSS1D_ESTIMATER
+#define VGAUSS1D_ESTIMATER
+#include <core/pre_estimater.hpp>
+#include <misc/optvec.hpp>
+#include <vmodels/gauss1d.hpp>
+#include <vector>
+
+namespace opt_utilities
+{
+  template <typename T>
+  class gauss1d_estimater
+    :public pre_estimater<optvec<T>,optvec<T>,optvec<T>,std::string>
+  {
+  public:
+    gauss1d_estimater()
+    {
+      this->set_model_id("1d gaussian");
+    }
+
+    gauss1d_estimater<T>* do_clone()const
+    {
+      return new gauss1d_estimater<T>(*this);
+    }
+
+    void do_estimate(const data_set<optvec<T>,optvec<T> >& d,model<optvec<T>,optvec<T>,optvec<T>,std::string>& m)const
+    {
+      int n=d.size();
+      T xmean=0;
+      T x2mean=0;
+      T wgt=0;
+      T wgt2=0;
+      for(int i=0;i<n;++i)
+	{
+	  T x=d.get_data(i).get_x()[0];
+	  T y=d.get_data(i).get_y()[0];
+	  xmean+=x*y;
+	  x2mean+=x*x*y;
+	  wgt+=y;
+	}
+      xmean/=wgt;
+      x2mean/=wgt;
+      T sigma=std::sqrt(x2mean-xmean*xmean);
+      m.set_param_value("x0",xmean);
+      m.set_param_value("sigma",sigma);
+      
+    }
+  };
+}
+
+
+
+#endif