SCorrelatorUtilities/src/GEvtTools.h

0001 /// ---------------------------------------------------------------------------
0002 /*! \file   GEvtTools.h
0003  *  \author Derek Anderson
0004  *  \date   03.06.2024
0005  *
0006  *  Collection of frequent event-level generator methods utilized
0007  *  in the sPHENIX Cold QCD Energy-Energy Correlator analysis.
0008  */
0009 /// ---------------------------------------------------------------------------
0010
0011 #ifndef SCORRELATORUTILITIES_GEVTTOOLS_H
0012 #define SCORRELATORUTILITIES_GEVTTOOLS_H
0013
0014 #pragma GCC diagnostic push
0015 #pragma GCC diagnostic ignored "-Wdeprecated-declarations"
0016
0017 // c++ utilities
0018 #include <limits>
0019 #include <string>
0020 #include <vector>
0021 #include <optional>
0022 // root libraries
0023 #include <Math/Vector3D.h>
0024 // phool libraries
0025 #include <phool/PHCompositeNode.h>
0026 // hepmc includes
0027 #include <HepMC/GenEvent.h>
0028 #include <HepMC/GenVertex.h>
0029 #include <HepMC/GenParticle.h>
0030 #include <phhepmc/PHHepMCGenEvent.h>
0031 #include <phhepmc/PHHepMCGenEventMap.h>
0032 // analysis utilities
0033 #include "ParInfo.h"
0034 #include "ParTools.h"
0035 #include "Constants.h"
0036 #include "Interfaces.h"
0037
0038 #pragma GCC diagnostic pop
0039
0040 // make common namespaces implicit
0041 using namespace std;
0042
0043
0044
0045 namespace SColdQcdCorrelatorAnalysis {
0046   namespace Tools {
0047
0048     // event-level generator tools --------------------------------------------
0049
0050     int64_t        GetNumFinalStatePars(PHCompositeNode* topNode, const vector<int> evtsToGrab, const Const::Subset subset, optional<float> chargeToGrab = nullopt);
0051     double         GetSumFinalStateParEne(PHCompositeNode* topNode, const vector<int> evtsToGrab, const Const::Subset subset, optional<float> chargeToGrab = nullopt);
0052     Types::ParInfo GetPartonInfo(PHCompositeNode* topNode, const int event, const int status);
0053
0054   }  // end Tools namespace
0055 }  // end SColdQcdCorrealtorAnalysis namespace
0056
0057 #endif
0058
0059 // end ------------------------------------------------------------------------