writeTree.cc

Go to the documentation of this file.

#include <unistd.h> // Check the file
 
void writeTree(){
  int rBin,phiBin,zBin;
  const int nR = 43;
  const int nPhi = 6;
  const int nZ = 6;
  float efficiency0,x[400],y[400];
  TTree *t = new TTree("effTree", "colEff");
 
  float t1;
  TH1F* h1 = new TH1F("h1","time distr.(ns)",400,0,20);
  char file[200];
 
  t->Branch("rBin", &rBin, "rBin/I");
  t->Branch("phiBin", &phiBin, "phiBin/I");
  t->Branch("zBin", &zBin, "zBin/I");
  t->Branch("efficiency0", &efficiency0, "efficiency0/F");
  t->Branch("x", x, "x[400]/F");
  t->Branch("y", y, "y[400]/F");
 
  for (Int_t i = 0; i < nR; i++){
    rBin = i;
    for(Int_t j = 0; j<nPhi;j++){
      phiBin = j;
      for(Int_t k=0;k<nZ;k++){
        int rPos = 400+i*10;
        sprintf(file,"/ihepbatch/besd12/liuy/EndcapTof/OpInETof2/results/%d/timeSta_%d_%d_%d.root",rPos,rPos,j,k);
        //if(access(file,F_OK) < 0) // Check if the file exists or not. Thx to zoujh.
        //{
        //  cout << "The file does NOT exist!" << endl;
        //  continue;
        //}
        //else
        //{
          cout << "The file:" << file << endl;
          TFile* f1 = TFile::Open(file);
          TNtuple* n1 = (TNtuple*)f1 -> Get("timeTuple");
          n1->SetBranchAddress("inTimePre",&t1);
          int nEntries = (Int_t)n1->GetEntries();
          cout << "nEntries:" << nEntries << "\n" << endl;
 
          for(int m=0;m<nEntries;m++)
          {
            n1->GetEntry(m);
            h1->Fill(t1);
          }
          double norm = h1->Integral();
          efficiency0 = norm/2000000.; // 200 events per run ; 10k photons per event
          //cout << "norm:" << norm << "; efficiency0: " << efficiency0 << "\n" << endl;
          if(norm!=0)
            h1->Scale(1./norm);
          else
          {
            cout << "The eff is null!" << endl;
            continue;
          }
          TAxis* xaxis = h1->GetXaxis();
          for(int m=0;m<400;m++)
          {
            x[m] = xaxis->GetBinCenter(m);
            y[m] = h1->GetBinContent(m);
          }
          f1->Close();
 
          zBin = k;
          t->Fill();
        //}
      }
    }
 
  }
  TFile *f = new TFile("effTree.root", "recreate");
  cout << "TTree write" <<endl;
  t->Write();
  f->Close();
}