moench Tree Maker added

git-svn-id: file:///afs/psi.ch/project/sls_det_software/svn/slsDetectorCalibration@4 113b152e-814d-439b-b186-022a431db7b5

slsDetectorCalibration/moenchMakeTree.C

@@ -0,0 +1,147 @@
+#include <TH1D.h>
+#include <TH2D.h>
+#include <TPad.h>
+#include <TDirectory.h>
+#include <TEntryList.h>
+#include <TFile.h>
+#include <TMath.h>
+#include <TTree.h>
+#include <TChain.h>
+#include <THStack.h>
+#include <TCanvas.h>
+#include <stdio.h>
+#include <deque>
+#include <list>
+#include <queue>
+#include <fstream>
+#include "RunningStat.h"
+#include "MovingStat.h"
+#include "moench02ModuleData.h"
+
+using namespace std;
+
+
+void moenchMakeTree(char *fformat, char *tname, int runmin, int runmax, int sign=1) {
+
+  moench02ModuleData *decoder=new moench02ModuleData();
+  char *buff;
+  char fname[10000];
+  
+  int nf=0;
+  
+  int dum, nPhotons;
+  double me, sig, tot, maxNei, val, valNei;
+
+  MovingStat stat[160][160];
+  MovingStat nPhotonsStat;
+
+  ifstream filebin;
+
+  int nbg=1000;
+
+  int ix, iy, ir, ic;
+  Double_t data[3][3];
+
+  TTree* tall=new TTree(tname,tname);
+  tall->Branch("iFrame",&nf,"iframe/I");
+  tall->Branch("x",&ix,"x/I");
+  tall->Branch("y",&iy,"y/I");
+  tall->Branch("data",data,"data[3][3]/D");
+  tall->Branch("pedestal",&me,"pedestal/D");
+  tall->Branch("rms",&sig,"rms/D");
+
+  for (ir=0; ir<160; ir++) {
+    for (ic=0; ic<160; ic++) {
+      stat[ir][ic].Clear();
+      stat[ir][ic].SetN(nbg);
+    }
+  }
+
+  for (int irun=runmin; irun<runmax; irun++) {
+    sprintf(fname,fformat,irun);
+    cout << "process file " << fname;
+    filebin.open((const char *)(fname), ios::in | ios::binary);
+   
+    while ((buff=decoder->readNextFrame(filebin))) {
+      nPhotons=0;
+      for (ix=0; ix<160; ix++){
+	  for (iy=0; iy<160; iy++) {
+
+	    dum=0; //no hit
+	    tot=0;
+
+	    if (nf>1000) {
+	      me=stat[iy][ix].Mean();
+	      sig=stat[iy][ix].StandardDeviation();
+	      val=sign*decoder->getChannelShort(buff,ix,iy)-me;
+
+	      dum=0; //no hit
+	      tot=0;
+	      maxNei;
+
+	      for (ir=-1; ir<2; ir++){
+		for (ic=-1; ic<2; ic++){
+		  if ((ix+ic)>=0 && (ix+ic)<160 && (iy+ir)>=0 && (iy+ir)<160) {
+		    valNei = decoder->getChannelShort(buff,ix+ic,iy+ir)-stat[iy+ir][ix+ic].Mean();
+		    if (decoder->getChannelShort(buff,ix+ic,iy+ir)>(stat[iy+ir][ix+ic].Mean()+3.*stat[iy+ir][ix+ic].StandardDeviation())) dum=1; //is a hit or neighbour is a hit!
+		    tot+=valNei;
+		    data[ir][ic];
+		    maxNei = max(maxNei,valNei);		    
+		  }
+		}
+	      }
+
+	      if (val<(me-3.*sig)) dum=2; //discard negative events!
+
+	      if(maxNei == val && dum == 1){ // this is an event and we are in the center
+		tall->Fill(); 
+		nPhotons++;
+	      }
+
+	      if (tot>3.*sig) dum=3; //discard events (for pedestal) where sum of the neighbours is too large.	      
+	    }
+	    if (nf<1000 || dum==0) 
+	      stat[iy][ix].Calc(sign*decoder->getChannelShort(buff,ix,iy));
+	   
+	  }
+      }
+      delete [] buff;
+      //cout << "=" ;
+      nPhotonsStat.Calc(nPhotons);
+      nf++;
+    }
+    cout << endl;
+    cout << " done. Avg. Photons/Frame: " << nPhotonsStat.Mean() << " sig: " <<  nPhotonsStat.StandardDeviation() << endl;
+    if (filebin.is_open())
+      filebin.close();	  
+    else
+      cout << "could not open file " << fname << endl;
+  }
+
+  delete decoder;
+  cout << "Read " << nf << " frames" << endl;
+  tall->Write();
+  tall->GetCurrentFile()->Close();
+ 
+}
+
+
+//to compile: g++ -DMYROOT -g `root-config --cflags --glibs` -o moenchMakeTree moenchMakeTree.C
+int main(int argc, char **argv){
+  if(argc != 6) cout << "Usage: inFile outdir tname fileNrStart fileNrEnd" << endl; exit(-1);
+
+  char *inFile = argv[1];
+  char *outDir = argv[2];
+  char *tName = argv[3];
+  int start = atoi(argv[4]);
+  int end = atoi(argv[5]);
+
+  TFile *f;
+  char outfname[1000];
+
+  sprintf(outfname,"%s/%s.root",outDir,tName);
+  f=new TFile(outfname,"RECREATE");
+
+  moenchMakeTree(inFile,tName,start,end);
+}
+