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();
}