Back to home page

EIC code displayed by LXR

 
 

    


File indexing completed on 2025-02-22 09:39:21

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 "Analyses.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<<"-A aaa   stripping only calib and setup object to external file"<<std::endl;
0030   std::cout<<"-b       calculate bad channel map internally"<<std::endl;
0031   std::cout<<"-B lll   apply external bad channel map during transfer of calibs"<<std::endl;
0032   std::cout<<"-C yyy   Apply calibrations stored in yyy root file to the input uncalibrated file"<<std::endl;
0033   std::cout<<"-d [0-n] switch on debug info with debug level 0 to n"<<std::endl;
0034   std::cout<<"-e       extended plotting = 1"<<std::endl;
0035   std::cout<<"-E       extended plotting = 2"<<std::endl;
0036   std::cout<<"-f       Force to write output if already exist"<<std::endl;
0037   std::cout<<"-F fff   set explicit plot extension explicitly, default is pdf "<<std::endl;
0038   std::cout<<"-i uuu   Input file in root format"<<std::endl;
0039   std::cout<<"-k kkk   enabling overwriting of calib file using external calib txt file"<<std::endl;
0040   std::cout<<"-L LLL   enable testing with only limited number of events"<<std::endl;
0041   std::cout<<"-m www   Name of mapping file  2024 PS TB [../configs/mappingFile_202409_CAEN.txt] "<<std::endl;
0042   std::cout<<"-M       save mip triggered cells only"<<std::endl;
0043   std::cout<<"-n       extract noise after first mip fits and reevaluate pedestals"<<std::endl;
0044   std::cout<<"-N       save noise triggered cells only"<<std::endl;
0045   std::cout<<"-o vvv   Output file name (mandatory)"<<std::endl;
0046   std::cout<<"-O kkk   Output directory name for plots (mandatory)"<<std::endl;
0047   std::cout<<"-p       Extract pedestal from input file and store it in output file"<<std::endl;
0048   std::cout<<"-P zzz   Correct pedestals stored in zzz root file and applied to input file"<<std::endl;
0049   std::cout<<"-r rrr   Name of run list file  2024 PS TB [../configs/DataTakingDB_202409_CAEN.csv] "<<std::endl;
0050   std::cout<<"-s       extract scaling constant from input data"<<std::endl;
0051   std::cout<<"-S       extract scaling constant from input data in 2nd iteration based on pretriggered data"<<std::endl;
0052   std::cout<<"-t       use local trigger eval from existing input, don't redo in calibrate"<<std::endl;
0053   std::cout<<"-T ttt   evaluate local triggers before calibrating, use external calib file ttt"<<std::endl;
0054   std::cout<<"-y yyyy  setting year externally to narrow parameters"<<std::endl;
0055   std::cout<<"-h       this help"<<std::endl<<std::endl;
0056   std::cout<<"Examples:"<<std::endl;
0057   std::cout<<exe<<" -p (-f) -o OutputWithExtractedPedestal.root -i input.root (-f to overwrite existing output)"<<std::endl;
0058   std::cout<<exe<<" -s (-f) -o OutputWithMIPscaling.root -i input.root (-f to overwrite existing output)"<<std::endl;
0059   std::cout<<exe<<" -S (-f) -o OutputWithMIPscaling.root -i input.root (-f to overwrite existing output)"<<std::endl;
0060   std::cout<<exe<<" -s (-f) -o OutputWithMIPscaling.root -P PedestalInput.root -i RawInput.root (-f to overwrite existing output)"<<std::endl;
0061   std::cout<<exe<<" -C Calibration.root (-f) -o CalibratedOutput.root -i Input.root (-f to overwrite existing output)"<<std::endl;
0062 }
0063   
0064 
0065 int main(int argc, char* argv[]){
0066   if(argc<4) {
0067     PrintHelp(argv[0]);
0068     return 0;
0069   }
0070   Analyses AnAnalysis;
0071   int c;
0072   while((c=getopt(argc,argv,"c:F:pT:sk:P:SnbB:L:NtMC:fo:O:aA:eEm:d:i:y:r:h"))!=-1){
0073     switch(c){
0074     case 'a':
0075       std::cout<<"DataPrep: printing calib object to file"<<std::endl;
0076       AnAnalysis.IsCalibSaveToFile(true);
0077       break;
0078     case 'A':
0079       std::cout<<"DataPrep: stripping calib object to file: " << optarg<<std::endl;
0080       AnAnalysis.IsToSaveCalibOnly(true);
0081       AnAnalysis.SetRootCalibOutput(optarg);
0082       break;
0083     case 'b':
0084       std::cout<<"DataPrep: run Bad channel determination"<<std::endl;
0085       AnAnalysis.SetCalcBadChannel(2);
0086       break;
0087     case 'B':
0088       std::cout<<"DataPrep: read bad channel map from external file: "<<optarg<<std::endl;
0089       AnAnalysis.SetExternalBadChannelMap(Form("%s",optarg));
0090       AnAnalysis.SetCalcBadChannel(1);
0091       break;
0092     case 'C':
0093       std::cout<<"DataPrep: Apply calibration (pedestal correction and scaling factor) from: "<<optarg<<std::endl;
0094       AnAnalysis.SetRootCalibInput(Form("%s",optarg));
0095       AnAnalysis.IsToApplyCalibration(true);
0096       break;
0097     case 'd':
0098       std::cout<<"DataPrep: enable debug " << optarg <<std::endl;
0099       AnAnalysis.EnableDebug(atoi(optarg));
0100       break;
0101     case 'e':
0102       std::cout<<"DataPrep: enabling extended plotting"<<std::endl;
0103       AnAnalysis.SetExtPlotting(1);
0104       break;
0105     case 'E':
0106       std::cout<<"DataPrep: enabling more extended plotting"<<std::endl;
0107       AnAnalysis.SetExtPlotting(2);
0108       break;
0109     case 'f':
0110       std::cout<<"DataPrep: If output already exists it will be overwritten"<<std::endl;
0111       AnAnalysis.CanOverWrite(true);
0112       break;
0113     case 'F':
0114       std::cout<<"DataPrep: Set Plot extension to: "<< optarg<<std::endl;
0115       AnAnalysis.SetPlotExtension(optarg);
0116       break;
0117     case 'i':
0118       std::cout<<"DataPrep: Root input file is: "<<optarg<<std::endl;
0119       AnAnalysis.SetRootInput(Form("%s",optarg));
0120       break;
0121     case 'k':
0122       std::cout<<"DataPrep: enable overwrite from external text file: "<< optarg <<std::endl;
0123       AnAnalysis.SetExternalCalibFile(optarg);
0124       AnAnalysis.SetOverWriteCalib(true);
0125       break;
0126     case 'L':
0127       std::cout<<"DataPrep: SetMaxEvents processed:"<<optarg<<std::endl;
0128       AnAnalysis.SetMaxEvents(atoi(optarg));
0129       break;
0130     case 'm':
0131       std::cout<<"DataPrep: Mapping file from: "<<optarg<<std::endl;
0132       AnAnalysis.SetMapInput(Form("%s",optarg));
0133       break;
0134     case 'M':
0135       std::cout<<"DataPrep: Save mip triggered cells only to create minimal sample"<<std::endl;
0136       AnAnalysis.IsToSaveMipsOnly(true);
0137       break;
0138     case 'n':
0139       std::cout<<"DataPrep: Extract noise after 1st mip fits and reevaluate pedestals"<<std::endl;
0140       AnAnalysis.IsToReextractNoise(true);
0141       break;
0142     case 'N':
0143       std::cout<<"DataPrep: Save noise only cells to create new pedestal"<<std::endl;
0144       AnAnalysis.IsToSaveNoiseOnly(true);
0145       break;
0146     case 'o':
0147       std::cout<<"DataPrep: Output to be saved in: "<<optarg<<std::endl;
0148       AnAnalysis.SetRootOutput(Form("%s",optarg));
0149       break;
0150     case 'O':
0151       std::cout<<"DataPrep: Outputdir plots to be saved in: "<<optarg<<std::endl;
0152       AnAnalysis.SetPlotOutputDir(Form("%s",optarg));
0153       break;
0154     case 'p':
0155       std::cout<<"DataPrep: Extract pedestal from input data"<<std::endl;
0156       AnAnalysis.IsToExtractPedestal(true);
0157       break;
0158     case 'P':
0159       std::cout<<"DataPrep: Apply pedestal correction from: "<<optarg<<std::endl;
0160       AnAnalysis.SetRootPedestalInput(Form("%s",optarg));
0161       AnAnalysis.IsToApplyPedestalCorrection(true);
0162       break;
0163     case 'r':
0164       std::cout<<"DataPrep: run list file from: "<<optarg<<std::endl;
0165       AnAnalysis.SetRunListInput(Form("%s",optarg));
0166       break;
0167     case 's':
0168       std::cout<<"DataPrep: Extract scaling constant from input data"<<std::endl;
0169       AnAnalysis.IsToExtractScaling(true);
0170       break;
0171     case 'S':
0172       std::cout<<"DataPrep: Extract scaling improved from 2nd iteration"<<std::endl;
0173       AnAnalysis.IsToExtractScalingImproved(true);
0174       break;
0175     case 't':
0176       std::cout<<"DataPrep: run without trigger eval, use from file" <<std::endl;
0177       AnAnalysis.UseLocTriggFromFile(true);
0178       break;
0179     case 'T':
0180       std::cout<<"DataPrep: run local trigger, with calib file:" << optarg<<std::endl;
0181       AnAnalysis.IsToEvalLocalTrigg(true);
0182       AnAnalysis.SetRootCalibInput(Form("%s",optarg));
0183       break;
0184     case 'y':
0185       std::cout<<"DataPrep: Setting year externally: "<<optarg<<std::endl;
0186       AnAnalysis.SetYear(atoi(optarg));
0187       break;
0188     case '?':
0189       std::cout<<"DataPrep: Option "<<optarg <<" not supported, will be ignored "<<std::endl;
0190       break;
0191     case 'h':
0192       PrintHelp(argv[0]);
0193       return 0;
0194     }
0195   }
0196   if(!AnAnalysis.CheckAndOpenIO()){
0197     std::cout<<"Check input and configurations, inconsistency or error with I/O detected"<<std::endl;
0198     PrintHelp(argv[0]);
0199     return -1;
0200   }
0201 
0202   AnAnalysis.Process();
0203   std::cout<<"Exiting"<<std::endl;
0204   return 0;
0205 }