...

[phys/latlib.git] / obstat.hpp

diff --git a/obstat.hpp b/obstat.hpp
new file mode 100644 (file)
index 0000000..d66467b
--- /dev/null
+++ b/obstat.hpp
@@ -0,0 +1,128 @@
+#ifndef OBSTAT_HPP
+#define OBSTAT_HPP
+
+#include <vector>
+#include <string>
+#include <map>
+#include <math.h>
+
+using namespace std;
+
+template <typename obstype>
+class obstat
+{ 
+public:
+  void addMeas(const obstype& val);
+  void addMeas(obstype val[], int valsize);
+
+  void computeEasy(const string& compid="one", const int& ival=0) { mean(compid, &measurements, ival); }
+
+  void computeJack(const string& compid, double (*func)(vector< vector <obstype> > *vals, void *para)=NULL, void *para=NULL);
+  void computeJack(double (*func)(vector< vector <obstype> > *vals, void *para)=NULL, void *para=NULL) { computeJack("one",func,para); }
+
+  double getMean(const string& compid="one") { return computations[compid].val; }
+  double getErr(const string& compid="one") { return computations[compid].err; }
+
+  void reset();
+
+private:
+  struct result{
+    double val;
+    double err;
+  };
+
+  vector< vector<obstype> > measurements;
+  map<string,result> computations;
+  
+  void mean(const string& compid, vector< vector <double> > *meas, const int& ival);
+  void mean(const string& compid, vector< vector <int> > *meas, const int& ival);
+};
+
+
+template <typename obstype> 
+void obstat<obstype>::reset()
+{
+  computations.clear();
+  measurements.clear();
+}
+
+template <typename obstype> 
+void obstat<obstype>::addMeas(const obstype& val)
+{
+  measurements.push_back( vector<obstype>(1,val) );
+}
+
+template <typename obstype>
+void obstat<obstype>::addMeas(obstype val[], int valsize)
+{
+  vector<obstype> tmpvec;
+  for(int i=0; i<valsize; i++) tmpvec.push_back(val[i]);
+  measurements.push_back(tmpvec);
+}
+
+template <typename obstype>
+void obstat<obstype>::mean(const string& compid, vector< vector<double> > *meas, const int& ival)
+{
+  computations[compid].val = 0;
+  computations[compid].err = 0;
+  
+  int nmeas = meas->size();
+
+  for(vector< vector<double> >::iterator measIt = meas->begin(); measIt != meas->end(); ++measIt)
+    computations[compid].val += (*measIt)[ival];
+  computations[compid].val /= nmeas;
+  
+  for(vector< vector<double> >::iterator measIt = meas->begin(); measIt != meas->end(); ++measIt)
+    computations[compid].err += pow( (*measIt)[ival] - computations[compid].val, 2 );
+  computations[compid].err = sqrt( computations[compid].err ) / nmeas;  
+}
+
+template <typename obstype>
+void obstat<obstype>::mean(const string& compid, vector< vector<int> > *meas, const int& ival)
+{
+  computations[compid].val = 0;
+  computations[compid].err = 0;
+  
+  int nmeas = meas->size();
+
+  for(vector< vector<int> >::iterator measIt = meas->begin(); measIt != meas->end(); ++measIt)
+    computations[compid].val += (*measIt)[ival];
+  computations[compid].val /= nmeas;
+  
+  for(vector< vector<int> >::iterator measIt = meas->begin(); measIt != meas->end(); ++measIt)
+    computations[compid].err += pow( (*measIt)[ival] - computations[compid].val, 2 );
+  computations[compid].err = sqrt( computations[compid].err ) / nmeas;
+}                               
+    
+template <typename obstype>
+void obstat<obstype>::computeJack(const string& compid, double (*func)(vector< vector<obstype> > *vals, void *para), void *para)
+{
+  int nmeas=measurements.size();
+  double manymeans[nmeas];
+
+  computations[compid].val = 0;
+  computations[compid].err = 0;
+
+  int imeas=0;
+  for(typename vector< vector<obstype> >::iterator removedIt = measurements.begin(); removedIt != measurements.end(); ++removedIt, imeas++)
+    {
+      vector<obstype> removed = *removedIt;
+      
+      *removedIt = measurements.back();
+      measurements.pop_back();
+
+      manymeans[imeas] = func(&measurements, para);
+      computations[compid].val += manymeans[imeas];
+      
+      measurements.push_back( *removedIt );
+      *removedIt = removed;
+    }
+  computations[compid].val /= nmeas;
+
+  for(int imean=0; imean<nmeas; imean++)
+    computations[compid].err += pow( manymeans[imean] - computations[compid].val, 2 );
+  computations[compid].err *= (double)(nmeas-1)/nmeas;
+  computations[compid].err = sqrt(computations[compid].err);
+}
+
+#endif