epic-lfhcal-tbana/NewStructure/DataAna.cc

0001 #include <iostream>
0002 #include <fstream>
0003 #include <vector>
0004 #include <map>
0005 #include <utility>
0006 #include <unistd.h> // Add for use on Mac OS -> Same goes for Analyses.cc
0007 #include "TString.h"
0008 #include "TFile.h"
0009 #include "TTree.h"
0010 #include "TCanvas.h"
0011 #include "TF1.h"
0012 #include "TH1D.h"
0013 #include "TObjArray.h"
0014 #include "TObjString.h"
0015
0016 #include "Setup.h"
0017 #include "Calib.h"
0018 #include "Event.h"
0019 #include "Tile.h"
0020 #include "HGCROC.h"
0021 #include "Caen.h"
0022 #include "DataAnalysis.h"
0023
0024 void PrintHelp(char* exe){
0025   std::cout<<"Usage:"<<std::endl;
0026   std::cout<<exe<<" [-option (arguments)]"<<std::endl;
0027   std::cout<<"Options:"<<std::endl;
0028   std::cout<<"-a       printing calib object to file (using name of output root or calib root file ending in txt)"<<std::endl;
0029   std::cout<<"-d [0-n] switch on debug info with debug level 0 to n"<<std::endl;
0030   std::cout<<"-f       Force to write output if already exist"<<std::endl;
0031   std::cout<<"-F fff   set plot extension explicitly, default is pdf "<<std::endl;
0032   std::cout<<"-i uuu   Input file in root format"<<std::endl;
0033   std::cout<<"-n nnn Input the maximum number of events you would like to run"<<std::endl;
0034   std::cout<<"-o vvv   Output file name "<<std::endl;
0035   std::cout<<"-O kkk   Output Histos saved in"<<std::endl;
0036   std::cout<<"-P zzz   Plots directory path"<<std::endl;
0037   std::cout<<"-Q       Run basic QA routine"<<std::endl;
0038   std::cout<<"-r rrr   Name of run list file  2024 PS TB [../configs/DataTakingDB_202409_CAEN.csv] "<<std::endl;
0039   std::cout<<"-y yyyy  setting year externally to narrow parameters"<<std::endl;
0040   std::cout<<"Examples:"<<std::endl;
0041   std::cout<<exe<<" -Q (-f) -P plotDir/ -O OutputHist.root -i input.root (-f to overwrite existing output)"<<std::endl;
0042 }
0043
0044
0045 int main(int argc, char* argv[]){
0046   if(argc<4) {
0047     PrintHelp(argv[0]);
0048     return 0;
0049   }
0050   DataAnalysis AnAnalysis;
0051   int c;
0052   while((c=getopt(argc,argv,"ad:e:fF:hi:n:o:O:P:qQr:y:"))!=-1){
0053     switch(c){
0054     case 'a':
0055       std::cout<<"printing calib object to file"<<std::endl;
0056       AnAnalysis.IsCalibSaveToFile(true);
0057       break;
0058     case 'd':
0059       std::cout<<"enable debug " << optarg <<std::endl;
0060       AnAnalysis.EnableDebug(atoi(optarg));
0061       break;
0062     case 'e':
0063       std::cout<<"enabling extended plotting: "<< optarg<<std::endl;
0064       AnAnalysis.SetExtPlotting(atoi(optarg));
0065       break;
0066     case 'f':
0067       std::cout<<"If output already exists it will be overwritten"<<std::endl;
0068       AnAnalysis.CanOverWrite(true);
0069       break;
0070     case 'F':
0071       std::cout<<"DataPrep: Set Plot extension to: "<< optarg<<std::endl;
0072       AnAnalysis.SetPlotExtension(optarg);
0073       break;
0074     case 'i':
0075       std::cout<<"Root input file is: "<<optarg<<std::endl;
0076       AnAnalysis.SetRootInput(Form("%s",optarg));
0077       break;
0078     case 'n':
0079       std::cout<<"Maximum number of events is changed to "<<optarg<<std::endl;
0080       AnAnalysis.SetMaximumEvents(atoi(optarg));
0081       break;
0082     case 'o':
0083       std::cout<<"Output to be saved in: "<<optarg<<std::endl;
0084       AnAnalysis.SetRootOutput(Form("%s",optarg));
0085       break;
0086     case 'O':
0087       std::cout<<"Output  Histsto be saved in: "<<optarg<<std::endl;
0088       AnAnalysis.SetRootOutputHists(Form("%s",optarg));
0089       break;
0090     case 'P':
0091       std::cout<<"Outputdir plots to be saved in: "<<optarg<<std::endl;
0092       AnAnalysis.SetPlotOutputDir(Form("%s",optarg));
0093       break;
0094     case 'q':
0095       std::cout<<"running very simple QA"<<std::endl;
0096       AnAnalysis.IsToSimpleRunQA(true);
0097       break;
0098     case 'Q':
0099       std::cout<<"running QA"<<std::endl;
0100       AnAnalysis.IsToRunQA(true);
0101       break;
0102     case 'r':
0103       std::cout<<"run list file from: "<<optarg<<std::endl;
0104       AnAnalysis.SetRunListInput(Form("%s",optarg));
0105       break;
0106     case 'y':
0107       std::cout<<"Setting year externally: "<<optarg<<std::endl;
0108       AnAnalysis.SetYear(atoi(optarg));
0109       break;
0110     case '?':
0111       std::cout<<"Option "<<optarg <<" not supported, will be ignored "<<std::endl;
0112       break;
0113     case 'h':
0114       PrintHelp(argv[0]);
0115       return 0;
0116     }
0117   }
0118   if(!AnAnalysis.CheckAndOpenIO()){
0119     std::cout<<"Check input and configurations, inconsistency or error with I/O detected"<<std::endl;
0120     PrintHelp(argv[0]);
0121     return -1;
0122   }
0123
0124   AnAnalysis.Process();
0125   std::cout<<"Exiting"<<std::endl;
0126   return 0;
0127 }