dynamically allocating manymeans in jackknife methods to allow for large number of...

[phys/latlib.git] / obstat.hpp

diff --git a/obstat.hpp b/obstat.hpp
index 2996043c7324ca2cee0e20b3cc55e46b6748cd3d..57a3431e6faa36078ecad0935256a6b8bc3e45d7 100644 (file)
--- a/obstat.hpp
+++ b/obstat.hpp
@@ -17,7 +17,9 @@ public:
 
   int computeEasy(const int& ival=0) { return mean(&measurements, ival); }
 
-  int computeJack(restype (*func)(vector< vector <meastype> > *vals, void *para)=NULL, void *para=NULL);
+  int computeJack(restype (*func)(vector< vector <meastype> > *vals, void *para), void *para=NULL);
+  int computeJack(void (*preMeasFunc)(vector< vector <meastype> > *allVals, vector <meastype> *preCalculated, void *para), 
+                 restype (*measFunc)(vector <meastype> *preCalculated, vector <meastype> *excludedmeas, int nmeas, void *para), void *para=NULL);
 
   restype getMean(int compid) { return computations[compid].val; }
   restype getErr(int compid) { return computations[compid].err; }
@@ -85,7 +87,7 @@ template <typename meastype, typename restype>
 int obstat<meastype,restype>::computeJack(restype (*func)(vector< vector<meastype> > *vals, void *para), void *para)
 {
   int nmeas=measurements.size();
-  restype manymeans[nmeas];
+  restype *manymeans = new restype[nmeas];
   result jackres;
 
   jackres.val = 0;
@@ -114,6 +116,41 @@ int obstat<meastype,restype>::computeJack(restype (*func)(vector< vector<meastyp
 
   computations.push_back(jackres);
 
+  delete [] manymeans;
+
+  return computations.size()-1;
+}
+
+template <typename meastype, typename restype>
+int obstat<meastype,restype>::computeJack(void (*preMeasFunc)(vector< vector <meastype> > *allVals, vector<meastype> *preCalculated, void *para), 
+                                         restype (*measFunc)(vector<meastype> *preCalculated, vector<meastype> *excludedmeas, int nmeas, void *para),
+                                         void *para) {
+  int nmeas=measurements.size();
+  restype *manymeans = new restype[nmeas];
+  result jackres;
+  vector<meastype> preCalculated;
+
+  jackres.val = 0;
+  jackres.err = 0;
+
+  preMeasFunc(&measurements, &preCalculated, para);
+
+  int imeas=0;
+  for(typename vector< vector<meastype> >::iterator removedIt = measurements.begin(); removedIt != measurements.end(); ++removedIt, imeas++) {
+    manymeans[imeas] = measFunc(&preCalculated, &(*removedIt), measurements.size(), para);
+    jackres.val += manymeans[imeas]; 
+  }
+  jackres.val /= nmeas;
+  
+  for(int imean=0; imean<nmeas; imean++)
+    jackres.err += pow( manymeans[imean] - jackres.val, 2 );
+  jackres.err *= (double)(nmeas-1)/nmeas;
+  jackres.err = sqrt(jackres.err);
+
+  computations.push_back(jackres);
+
+  delete [] manymeans;
+
   return computations.size()-1;
 }