Changeset 467 in svn for trunk/Resolutions_ATLAS.cpp

Timestamp:

Jul 10, 2009, 5:57:55 PM (15 years ago)

Author:

Xavier Rouby

Message:

cleaning, comments, progressbar, cout

File:

• trunk/Resolutions_ATLAS.cpp (modified) (13 diffs)

trunk/Resolutions_ATLAS.cpp

@@ -38 +38 @@
 #include "TApplication.h"
 #include "TFile.h"
+#include "TStopwatch.h"
 
 #include "ExRootTreeReader.h"
 #include "ExRootTreeWriter.h"
 #include "ExRootTreeBranch.h"
+#include "ExRootProgressBar.h"
 #include "TreeClasses.h"
 
@@ -189 +191 @@
 {
   int appargc = 2;
-  char *appName = "Resolution";
-  char *appargv[] = {appName, "-b"};
+  char *appName= new char[20];
+  char *appOpt= new char[20];
+  sprintf(appName,"Resolution_ATLAS");
+  sprintf(appOpt,"-b");
+  char *appargv[] = {appName,appOpt};
   TApplication app(appName, &appargc, appargv);
-  
+  delete [] appName;
+  delete [] appOpt;
+
   if(argc != 3) {
       cout << " Usage: " << argv[0] << " input_file" << " output_file" << endl;
@@ -200 +207 @@
   }
 
-  srand (time (NULL));         /* Initialisation du générateur */
+
+  // 1. ********** initialisation ***********
+
+  srand (time (NULL));         
+  TStopwatch globalwatch;
+  globalwatch.Start();
   
-  //read the input TROOT file
   string inputfilename(argv[1]), outputfilename(argv[2]);
 
+  // 1.1 checks the input file
+  if(inputfilename.find(".root") > inputfilename.length()   ) {
+    cout << "input_file should be a .root file from Delphes!\n";
+    return -1;
+  }
+  TFile f(inputfilename.c_str());
+  if (!f.FindKey("GEN") || !f.FindKey("Analysis")           ) {
+    cout << "input_file should be a .root file from Delphes!\n";
+    cout << "it should contain both \"GEN\" and \"Analysis\" trees at least.\n";
+    return -1;
+  }
+  f.Close();
+
+  // 1.2 checks the output file
   if(outputfilename.find(".root") > outputfilename.length() ) {
     cout << "output_file should be a .root file!\n";
     return -1;
   }
-
-
-
   TFile *outputFile = TFile::Open(outputfilename.c_str(), "RECREATE");// Creates the file, but should be closed just after
   outputFile->Close();
-  
+ 
+  // 1.3 Reads the trees in  input file 
   TChain chainGEN("GEN");
   chainGEN.Add(inputfilename.c_str());
@@ -228 +251 @@
   const TClonesArray *branchGen = treeReaderGEN->UseBranch("Particle");
   TIter itGen((TCollection*)branchGen);
-  
-  //write the output root file
+ 
+ 
+  // 1.4 Prepares the output root file
   ExRootTreeWriter *treeWriter = new ExRootTreeWriter(outputfilename, "Analysis");
   ExRootTreeBranch *branchjet = treeWriter->NewBranch("JetPTResol", RESOLJET::Class());
@@ -236 +260 @@
   ExRootTreeBranch *branchtaujet = treeWriter->NewBranch("TauJetPTResol", TAUHAD::Class());
   ExRootTreeBranch *branchetmis = treeWriter->NewBranch("ETmisResol",ETMIS::Class());
-  
+ 
+  // 1.5 other initialisations 
   TRootGenParticle *particle;
-  
   RESOLELEC * elementElec;
   RESOLMUON *elementMuon;
@@ -245 +269 @@
   ETMIS *elementEtmis;
   
-  int numTau=0;
-  int numTauRec=0;
+  int numTau=0, numTauRec=0;
   
   RESOLution *DET = new RESOLution();
@@ -259 +282 @@
   DET->ReadDataCard(detectorcard);
 
-
-  //Jet information
   JetsUtil *JETRUN = new JetsUtil(detectorcard);
 
-  TLorentzVector genMomentum(0,0,0,0);//TLorentzVector containing generator level information
-  TLorentzVector recoMomentum(0,0,0,0);//TLorentzVector containing generator level information
+  TLorentzVector genMomentum(0,0,0,0); // generator level information
+  TLorentzVector recoMomentum(0,0,0,0);// reconstruction level information
   LorentzVector jetMomentum;
   
@@ -272 +293 @@
   vector<TLorentzVector> towers;
  
-  // Loop over all events
+  // 2. Loop over all events
   Long64_t entry, allEntries = treeReader->GetEntries();
-  cout << "** Chain contains " << allEntries << " events" << endl;
+  cout << endl << endl;
+  cout <<"*********************************************************************"<< endl;
+  cout <<"**                                                                 **"<< endl;
+  cout <<"**                                                                 **"<< endl;
+  cout <<"**                                                                 **"<< endl;
+  cout <<"**        ####### Start resolution processing    ########          **"<< endl;
+  cout << left  << setw(52) <<"**              Total number of events to run: "<<""
+       << left  << setw(15) << allEntries <<""
+       << right << setw(2) <<"**"<<endl;
+
+  ExRootProgressBar *Progress = new ExRootProgressBar(allEntries);
+
   for(entry = 0; entry < allEntries; ++entry)
     {
+      Progress->Update(entry);
       TLorentzVector PTmisReco(0,0,0,0);
       TLorentzVector PTmisGEN(0,0,0,0);
@@ -282 +315 @@
       treeReaderGEN->ReadEntry(entry);
       treeWriter->Clear();
-      if((entry % 100) == 0 && entry > 0 )  cout << "** Processing element # " << entry << endl;
       
       TSimpleArray<TRootGenParticle> bGen;
@@ -372 +404 @@
         TLorentzVector JETgen(0,0,0,0);
         JETgen.SetPxPyPzE(sorted_jetsGEN[i].px(),sorted_jetsGEN[i].py(),sorted_jetsGEN[i].pz(),sorted_jetsGEN[i].E());
-        PairingJet(JETreco,JETgen,branchJet);
+        PairingJet(JETreco,JETgen,branchJet,dR_jetpairing);
         if(JETreco.Pt()>jet_pt_cut)
           {
@@ -383 +415 @@
           }
       }
-      numTau = numTau+TausHadr.GetEntries();
+      numTau += TausHadr.GetEntries();
 
       TIter itJet((TCollection*)branchJet);
@@ -413 +445 @@
     } // Loop over all events
   treeWriter->Write();
-
-  cout << detectorcard << " has been used.\n";
-  cout << "** Exiting..." << endl;
+  globalwatch.Stop();
+
+
+  // Screen output
+  cout <<"**                                                                 **"<< endl;
+  cout <<"**                                                                 **"<< endl;
+  cout <<"**        ################## Time report #################         **"<< endl;
+  cout << left  << setw(32) <<"**              Time report for "<<""
+       << left  << setw(15) << allEntries <<""
+       << right << setw(22) <<"events         **"<<endl;
+  cout <<"**                                                                 **"<< endl;
+  cout << left  << setw(10) <<"**"<<""
+       << left  << setw(15) <<"Time (s):"<<""
+       << right << setw(15) <<"CPU"<<""
+       << right << setw(15) <<"Real"<<""
+       << right << setw(14)  <<"**"<<endl;
+  cout << left  << setw(10) <<"**"<<""
+       << left  << setw(15) <<" +  Global:"<<""
+       << right << setw(15) <<globalwatch.CpuTime()<<""
+       << right << setw(15) <<globalwatch.RealTime()<<""
+       << right << setw(14)  <<"**"<<endl;
+
+
+  string tempstring = detectorcard + " has been used.";
+  cout <<"**        ################## Configuration ###############         **"<< endl;
+  cout << left << setw(16) << "**        " << ""
+       << left << setw(51) <<  tempstring << "**" << endl;
+
+  tempstring =  outputfilename + " has been created.";
+  cout << left << setw(16) << "**        " << ""
+       << left << setw(51) <<  tempstring << "**" << endl;
+
+  cout << left << setw(16) << "**        " << ""
+       << left << setw(51) <<  get_time_date() << "**" << endl;
+
+  cout <<"**                                                                 **"<< endl;
+  cout <<"**                        Exiting ...                              **"<< endl;
+  cout <<"**                                                                 **"<< endl;
+  cout <<"**                                                                 **"<< endl;
+  cout <<"*********************************************************************"<< endl;
+
   
   delete treeWriter;
@@ -422 +492 @@
 }
 
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+