macros/CaloProduction/Fun4All_New_Year2_Fitting.C

0001 #ifndef FUN4ALL_YEAR2_FITTING_C
0002 #define FUN4ALL_YEAR2_FITTING_C
0003
0004 #include <Calo_Fitting.C>
0005 #include <QA.C>
0006
0007 #include <calotrigger/TriggerRunInfoReco.h>
0008
0009 #include <calovalid/CaloFittingQA.h>
0010
0011 #include <calopacketskimmer/CaloPacketSkimmer.h>
0012
0013 #include <ffamodules/CDBInterface.h>
0014 #include <ffamodules/FlagHandler.h>
0015 #include <ffamodules/HeadReco.h>
0016 #include <ffamodules/SyncReco.h>
0017
0018 #include <fun4all/Fun4AllDstInputManager.h>
0019 #include <fun4all/Fun4AllDstOutputManager.h>
0020 #include <fun4all/Fun4AllInputManager.h>
0021 #include <fun4all/Fun4AllRunNodeInputManager.h>
0022 #include <fun4all/Fun4AllServer.h>
0023 #include <fun4all/Fun4AllUtils.h>
0024 #include <fun4all/SubsysReco.h>
0025
0026 #include <phool/recoConsts.h>
0027
0028 #include <TSystem.h>
0029
0030 #include <format>
0031 #include <fstream>
0032
0033 R__LOAD_LIBRARY(libfun4allraw.so)
0034 R__LOAD_LIBRARY(libcalovalid.so)
0035 R__LOAD_LIBRARY(libcalotrigger.so)
0036 R__LOAD_LIBRARY(libCaloPacketSkimmer.so);
0037
0038 // this pass containis the reco process that's stable wrt time stamps(raw tower building)
0039 void Fun4All_New_Year2_Fitting(int nEvents = 100,
0040                const std::string &inlist = "files.list",
0041                            const std::string &outfile = "DST_CALOFITTING_run2auau_ana487_2024p018_v001",
0042                            const std::string &outfile_hist = "HIST_CALOFITTINGQA_run2auau_ana487_2024p018_v001",
0043                            const std::string &dbtag = "ProdA_2024")
0044 {
0045   gSystem->Load("libg4dst.so");
0046
0047   Fun4AllServer *se = Fun4AllServer::instance();
0048   se->Verbosity(1);
0049   se->VerbosityDownscale(1000);
0050
0051   recoConsts *rc = recoConsts::instance();
0052
0053   // conditions DB global tag
0054   rc->set_StringFlag("CDB_GLOBALTAG", dbtag);
0055   CDBInterface::instance()->Verbosity(1);
0056
0057   FlagHandler *flag = new FlagHandler();
0058   se->registerSubsystem(flag);
0059
0060   // Get info from DB and store in DSTs
0061   TriggerRunInfoReco *triggerinfo = new TriggerRunInfoReco();
0062   se->registerSubsystem(triggerinfo);
0063
0064   CaloPacketSkimmer *calopacket = new CaloPacketSkimmer();
0065   se->registerSubsystem(calopacket);
0066
0067   Process_Calo_Fitting();
0068
0069   ///////////////////////////////////
0070   // Validation
0071   CaloFittingQA *ca = new CaloFittingQA("CaloFittingQA");
0072   se->registerSubsystem(ca);
0073
0074 // loop over all files in file list and create an input manager for each one
0075   Fun4AllInputManager *In = nullptr;
0076   std::ifstream infile;
0077   infile.open(inlist);
0078   int iman = 0;
0079   std::string line;
0080   bool first {true};
0081   int runnumber = 0;
0082   int segment = 99999;
0083   if (infile.is_open())
0084   {
0085     while (std::getline(infile, line))
0086     {
0087       if (line[0] == '#')
0088       {
0089     std::cout << "found commented out line " << line << std::endl;
0090     continue;
0091       }
0092       // extract run number from first not commented out file in list
0093       if (first)
0094       {
0095     std::pair<int, int> runseg = Fun4AllUtils::GetRunSegment(line);
0096     runnumber = runseg.first;
0097     segment = runseg.second;
0098     rc->set_uint64Flag("TIMESTAMP", runnumber);
0099     first = false;
0100       }
0101       std::string magname = "DSTin_" + std::to_string(iman);
0102       In = new Fun4AllDstInputManager(magname);
0103       In->Verbosity(1);
0104       In->AddFile(line);
0105       se->registerInputManager(In);
0106       iman++;
0107     }
0108     infile.close();
0109   }
0110
0111 // this strips all nodes under the Packets PHCompositeNode
0112 // (means removes all offline packets)
0113   std::string dstoutfile = std::format("{}-{:08}-{:05}.root",outfile, runnumber,segment);
0114   Fun4AllDstOutputManager *out = new Fun4AllDstOutputManager("DSTOUT", dstoutfile);
0115   out->StripCompositeNode("Packets");
0116   se->registerOutputManager(out);
0117   // se->Print();
0118   if (nEvents < 0)
0119   {
0120     return;
0121   }
0122   se->run(nEvents);
0123   se->End();
0124
0125   dstoutfile = std::format("{}-{:08}-{:05}.root",outfile_hist, runnumber,segment);
0126   QAHistManagerDef::saveQARootFile(dstoutfile);
0127
0128   CDBInterface::instance()->Print();  // print used DB files
0129   se->PrintTimer();
0130   delete se;
0131   std::cout << "All done!" << std::endl;
0132   gSystem->Exit(0);
0133 }
0134 #endif