sPhenix code displayed by LXR master/​analysis/​LightFlavorRatios/​geometric_acceptance/​simulation/​G4_Input.C	Source navigation Diff markup Identifier search General search
	Version: master Revert   Change

File indexing completed on 2026-04-06 08:10:18

 #ifndef MACRO_G4INPUT_C
 #define MACRO_G4INPUT_C
 
 #include <GlobalVariables.C>
 
 #include <G4_TrkrVariables.C>
 
 #include <phpythia8/PHPythia8.h>
 
 #include <g4main/CosmicSpray.h>
 #include <g4main/HepMCNodeReader.h>
 #include <g4main/PHG4IonGun.h>
 #include <g4main/PHG4ParticleGenerator.h>
 #include <g4main/PHG4ParticleGeneratorD0.h>
 #include <g4main/PHG4ParticleGeneratorVectorMeson.h>
 #include <g4main/PHG4ParticleGun.h>
 #include <g4main/PHG4SimpleEventGenerator.h>
 // #include <g4main/ReadEICFiles.h>
 
 #include <fermimotionafterburner/FermimotionAfterburner.h>
 
 #include <hijingflipafterburner/HIJINGFlipAfterburner.h>
 
 #include <reactionplaneafterburner/ReactionPlaneAfterburner.h>
 
 #include <phhepmc/Fun4AllHepMCInputManager.h>
 #include <phhepmc/Fun4AllHepMCPileupInputManager.h>
 #include <phhepmc/HepMCFlowAfterBurner.h>
 #include <phhepmc/PHHepMCGenHelper.h>
 
 #include <fun4all/Fun4AllDstInputManager.h>
 #include <fun4all/Fun4AllDummyInputManager.h>
 #include <fun4all/Fun4AllInputManager.h>
 #include <fun4all/Fun4AllNoSyncDstInputManager.h>
 #include <fun4all/Fun4AllServer.h>
 
 #include <TSystem.h>  // for gSystem
 
 #include <set>
 
 R__LOAD_LIBRARY(libfun4all.so)
 R__LOAD_LIBRARY(libg4testbench.so)
 R__LOAD_LIBRARY(libPHPythia8.so)
 R__LOAD_LIBRARY(libFermimotionAfterburner.so)
 R__LOAD_LIBRARY(libHIJINGFlipAfterburner.so)
 R__LOAD_LIBRARY(libReactionPlaneAfterburner.so)
 
 namespace Input
 {
   // Real Event generators
   bool PYTHIA6 = false;
   int PYTHIA6_EmbedId = 0;
 
   bool PYTHIA8 = false;
   int PYTHIA8_NUMBER = 1;
   int PYTHIA8_VERBOSITY = 0;
   std::set<int> PYTHIA8_EmbedIds;
 
   // Single/multiple particle generators
   bool DZERO = false;
   int DZERO_NUMBER = 1;
   int DZERO_VERBOSITY = 0;
   std::set<int> DZERO_EmbedIds;
 
   bool GUN = false;
   int GUN_NUMBER = 1;
   int GUN_VERBOSITY = 0;
   std::set<int> GUN_EmbedIds;
 
   bool IONGUN = false;
   int IONGUN_NUMBER = 1;
   int IONGUN_VERBOSITY = 0;
   std::set<int> IONGUN_EmbedIds;
 
   bool PGEN = false;
   int PGEN_NUMBER = 1;
   int PGEN_VERBOSITY = 0;
   std::set<int> PGEN_EmbedIds;
 
   bool SIMPLE = false;
   int SIMPLE_NUMBER = 1;
   int SIMPLE_VERBOSITY = 0;
 
   int UPSILON_NUMBER = 1;
   int UPSILON_VERBOSITY = 0;
   // other UPSILON settings which are also used elsewhere are in GlobalVariables.C
 
   double PILEUPRATE = 0.;
   bool READHITS = false;
   int VERBOSITY = 0;
   int EmbedId = 1;
   int VertexEmbedId = 0;
 
   bool COSMIC = false;
   double COSMIC_R = 650.;
 
   double beam_crossing = -1.5;  // -1.5 mRad
   //! apply reference sPHENIX nominal beam parameter with 1.5mrad crossing as used in 2024
   //! \param[in] HepMCGen any HepMC generator, e.g. Fun4AllHepMCInputManager, Fun4AllHepMCPileupInputManager, PHPythia8, PHPythia6, ReadEICFiles
   //! \param[in] collision_type select the beam configuration with Input::BeamConfiguration
   void ApplysPHENIXBeamParameter(PHHepMCGenHelper *HepMCGen, const Input::BeamConfiguration &beam_config)
   {
     if (HepMCGen == nullptr)
     {
       std::cout << "ApplysPHENIXBeamParameter(): Fatal Error - null input pointer HepMCGen" << std::endl;
       exit(1);
     }
     double localbcross = Input::beam_crossing / 2. * 1e-3;
     switch (beam_config)
     {
     case AA_COLLISION:
     case mRad_10:
       // heavy ion mode
       Input::beam_crossing = 1.;  // +1 mRad for late 2024 with triggered readout for mvtx
       //  Xing angle is split among both beams, means set to 0.5 mRad
       localbcross = Input::beam_crossing / 2. * 1e-3;
       HepMCGen->set_beam_direction_theta_phi(localbcross, 0, M_PI - localbcross, 0);
       HepMCGen->set_vertex_distribution_width(
           100e-4,         // approximation from past STAR/Run16 AuAu data
           100e-4,         // approximation from past STAR/Run16 AuAu data
           13.5,           // measured 2024 with 1mRad beam Xing
           20 / 29.9792);  // 20cm collision length / speed of light in cm/ns
 
       HepMCGen->set_vertex_distribution_mean(-0.058,0.133,0,0); // rough position of actual beam spot. This can vary by a few hundred microns
       break;
     case pA_COLLISION:
 
       // pA mode
 
       // 1.5mRad is split among both beams, means set to 0.75 mRad
       HepMCGen->set_beam_direction_theta_phi(localbcross, 0, M_PI - localbcross, 0);
       HepMCGen->set_vertex_distribution_width(
           100e-4,         // set to be similar to AA
           100e-4,         // set to be similar to AA
           8,              // sPH-TRG-2022-001. Fig B.4
           20 / 29.9792);  // 20cm collision length / speed of light in cm/ns
 
       break;
     case pp_COLLISION:
     case mRad_15:
 
       // pp mode
       // 1.5mRad is split among both beams, means set to 0.75 mRad
       HepMCGen->set_beam_direction_theta_phi(localbcross, 0, M_PI - localbcross, 0);
       HepMCGen->set_vertex_distribution_width(
           120e-4,         // approximation from past PHENIX data
           120e-4,         // approximation from past PHENIX data
           16,             // measured in 2024 for 1.5mrad Xing angle
           20 / 29.9792);  // 20cm collision length / speed of light in cm/ns
       HepMCGen->set_vertex_distribution_mean(-0.0716,0.1401,-0.4175,0); // rough position of actual beam spot. This can vary by a few hundred microns
       
       break;
     case pp_ZEROANGLE:
     case mRad_00:
 
       // pp mode
 
       HepMCGen->set_vertex_distribution_width(
           120e-4,         // approximation from past PHENIX data
           120e-4,         // approximation from past PHENIX data
           65,             // measured in 2024 for 0 Xing angle
           20 / 29.9792);  // 20cm collision length / speed of light in cm/ns
 
       break;
 
     case ppg02:
       Input::beam_crossing = 1.;  // +1 mRad for late 2024 with triggered readout for mvtx
       localbcross = Input::beam_crossing / 2. * 1e-3;
       //  Xing angle is split among both beams, means set to 0.5 mRad
       HepMCGen->set_beam_direction_theta_phi(localbcross, 0, M_PI - localbcross, 0);  // 1.5mrad x-ing of sPHENIX
       HepMCGen->set_vertex_distribution_mean(-0.022, 0.223, -4.03, 0.);
       HepMCGen->set_vertex_distribution_width(
           120e-4,         // approximation from past PHENIX data
           120e-4,         // approximation from past PHENIX data
           9.358,          // measured by intt
           20 / 29.9792);  // 20cm collision length / speed of light in cm/ns
       break;
 
     case mRad_05:
       Input::beam_crossing = 0.5;
       //0.5 mRad is split among both beams, means set to 0.25 mRad
       localbcross = Input::beam_crossing / 2. * 1e-3;
 
       HepMCGen->set_beam_direction_theta_phi(localbcross, 0, M_PI - localbcross, 0);
       HepMCGen->set_vertex_distribution_width(
           120e-4,         // approximation from past PHENIX data
           120e-4,         // approximation from past PHENIX data
           24.5,             // estimate from Colorado
           20 / 29.9792);  // 20cm collision length / speed of light in cm/ns
 
       break;
     case mRad_075:
       Input::beam_crossing = 0.75;
       //0.75 mRad is split among both beams, means set to 0.375 mRad
       localbcross = Input::beam_crossing / 2. * 1e-3;
 
       HepMCGen->set_beam_direction_theta_phi(localbcross, 0, M_PI - localbcross, 0);
       HepMCGen->set_vertex_distribution_width(
           120e-4,         // approximation from past PHENIX data
           120e-4,         // approximation from past PHENIX data
           20,             // from online monitoring mbd
           20 / 29.9792);  // 20cm collision length / speed of light in cm/ns
 
       break;
     default:
       std::cout << "ApplysPHENIXBeamParameter: invalid beam_config = " << beam_config << std::endl;
 
       exit(1);
     }
 
     HepMCGen->set_vertex_distribution_function(
         PHHepMCGenHelper::Gaus,
         PHHepMCGenHelper::Gaus,
         PHHepMCGenHelper::Gaus,
         PHHepMCGenHelper::Gaus);
   }
 
   //! apply sPHENIX nominal beam parameter according to the beam collision setting of Input::IS_PP_COLLISION
   //! \param[in] HepMCGen any HepMC generator, e.g. Fun4AllHepMCInputManager, Fun4AllHepMCPileupInputManager, PHPythia8, PHPythia6, ReadEICFiles
   void ApplysPHENIXBeamParameter(PHHepMCGenHelper *HepMCGen)
   {
     ApplysPHENIXBeamParameter(HepMCGen, Input::BEAM_CONFIGURATION);
   }
 
   void ApplysPHENIXBeamParameter(std::vector<PHPythia8 *> &HepMCGenVec)
   {
     PHHepMCGenHelper *gen = dynamic_cast<PHHepMCGenHelper *> (HepMCGenVec[0]);
     if (gen)
     {
       ApplysPHENIXBeamParameter(gen, Input::BEAM_CONFIGURATION);
     }
   }
 
   //! apply EIC beam parameter to any HepMC generator following EIC CDR,
   //! including in-time collision's space time shift, beam crossing angle and angular divergence
   //! \param[in] HepMCGen any HepMC generator, e.g. Fun4AllHepMCInputManager, Fun4AllHepMCPileupInputManager, PHPythia8, PHPythia6, ReadEICFiles
   void ApplyEICBeamParameter(PHHepMCGenHelper *HepMCGen)
   {
     if (HepMCGen == nullptr)
     {
       std::cout << "ApplyEICBeamParameter(): Fatal Error - null input pointer HepMCGen" << std::endl;
       exit(1);
     }
 
     // 25mrad x-ing as in EIC CDR
     const double EIC_hadron_crossing_angle = 25e-3;
 
     HepMCGen->set_beam_direction_theta_phi(
         EIC_hadron_crossing_angle,  // beamA_theta
         0,                          // beamA_phi
         M_PI,                       // beamB_theta
         0                           // beamB_phi
     );
     HepMCGen->set_beam_angular_divergence_hv(
         119e-6, 119e-6,  // proton beam divergence horizontal & vertical, as in EIC CDR Table 1.1
         211e-6, 152e-6   // electron beam divergence horizontal & vertical, as in EIC CDR Table 1.1
     );
 
     // angular kick within a bunch as result of crab cavity
     // using an naive assumption of transfer matrix from the cavity to IP,
     // which is NOT yet validated with accelerator optics simulations!
     const double z_hadron_cavity = 52e2;  // CDR Fig 3.3
     const double z_e_cavity = 38e2;       // CDR Fig 3.2
     HepMCGen->set_beam_angular_z_coefficient_hv(
         -EIC_hadron_crossing_angle / 2. / z_hadron_cavity, 0,
         -EIC_hadron_crossing_angle / 2. / z_e_cavity, 0);
 
     // calculate beam sigma width at IP  as in EIC CDR table 1.1
     const double sigma_p_h = sqrt(80 * 11.3e-7);
     const double sigma_p_v = sqrt(7.2 * 1.0e-7);
     const double sigma_p_l = 6;
     const double sigma_e_h = sqrt(45 * 20.0e-7);
     const double sigma_e_v = sqrt(5.6 * 1.3e-7);
     const double sigma_e_l = 2;
 
     // combine two beam gives the collision sigma in z
     const double collision_sigma_z = sqrt(sigma_p_l * sigma_p_l + sigma_e_l * sigma_e_l) / 2;
     const double collision_sigma_t = collision_sigma_z / 29.9792;  // speed of light in cm/ns
 
     HepMCGen->set_vertex_distribution_width(
         sigma_p_h * sigma_e_h / sqrt(sigma_p_h * sigma_p_h + sigma_e_h * sigma_e_h),  // x
         sigma_p_v * sigma_e_v / sqrt(sigma_p_v * sigma_p_v + sigma_e_v * sigma_e_v),  // y
         collision_sigma_z,                                                            // z
         collision_sigma_t);                                                           // t
     HepMCGen->set_vertex_distribution_function(
         PHHepMCGenHelper::Gaus,   // x
         PHHepMCGenHelper::Gaus,   // y
         PHHepMCGenHelper::Gaus,   // z
         PHHepMCGenHelper::Gaus);  // t
   }
 }  // namespace Input
 
 namespace INPUTHEPMC
 {
   std::string filename;
   std::string listfile;
   int EmbedId = 0;
   bool FLOW = false;
   int FLOW_VERBOSITY = 0;
   bool FLOW_FLUCTUATIONS = false;
   float FLOW_SCALING = 1.0;  // scaling factor for flow
   bool FERMIMOTION = false;
   bool HIJINGFLIP = false;
   bool REACTIONPLANERAND = false;
   float HEPMC_STRANGENESS_FRACTION = -1.;
 
 }  // namespace INPUTHEPMC
 
 namespace INPUTREADEIC
 {
   std::string filename;
 }  // namespace INPUTREADEIC
 
 namespace INPUTREADHITS
 {
   std::map<unsigned int, std::string> filename;
   std::map<unsigned int, std::string> listfile;
 }  // namespace INPUTREADHITS
 
 namespace INPUTEMBED
 {
   std::map<unsigned int, std::string> filename;
   std::map<unsigned int, std::string> listfile;
   bool REPEAT = true;
 }  // namespace INPUTEMBED
 
 namespace PYTHIA6
 {
   std::string config_file = std::string(getenv("CALIBRATIONROOT")) + "/Generators/phpythia6.cfg";
 }
 
 namespace PYTHIA8
 {
   std::map<int, std::string> config_file =
   {
     {0, std::string(getenv("CALIBRATIONROOT")) + "/Generators/phpythia8.cfg"}
   };
 }
 
 namespace PILEUP
 {
   std::string pileupfile = "/sphenix/sim/sim01/sphnxpro/MDC1/sHijing_HepMC/data/sHijing_0_20fm-0000000001-00000.dat";
   double TpcDriftVelocity = G4TPC::tpc_drift_velocity_sim;
 }  // namespace PILEUP
 
 // collection of pointers to particle generators we can grab in the Fun4All macro
 namespace INPUTGENERATOR
 {
   std::vector<PHG4IonGun *> IonGun;
   std::vector<PHG4ParticleGenerator *> ParticleGenerator;
   std::vector<PHG4ParticleGeneratorD0 *> DZeroMesonGenerator;
   std::vector<PHG4ParticleGeneratorVectorMeson *> VectorMesonGenerator;
   std::vector<PHG4SimpleEventGenerator *> SimpleEventGenerator;
   std::vector<PHG4ParticleGun *> Gun;
   PHPythia8 *Pythia6 = nullptr;
   std::vector<PHPythia8 *> Pythia8;
   //  ReadEICFiles *EICFileReader = nullptr;
   CosmicSpray *Cosmic = nullptr;
 }  // namespace INPUTGENERATOR
 
 namespace INPUTMANAGER
 {
   Fun4AllHepMCInputManager *HepMCInputManager = nullptr;
   Fun4AllHepMCPileupInputManager *HepMCPileupInputManager = nullptr;
 }  // namespace INPUTMANAGER
 
 void InputInit()
 {
   // for pileup sims embed id is 1, to distinguish particles
   // which will be embedded (when Input::EMBED = true) into pileup sims
   // we need to start at embedid = 2
   if (Input::EMBED)
   {
     Input::EmbedId = 2;
   }
   // first consistency checks - not all input generators play nice
   // with each other
   if (Input::READHITS && Input::EMBED)
   {
     std::cout << "Reading Hits and Embedding into background at the same time is not supported" << std::endl;
     gSystem->Exit(1);
   }
   if (Input::READHITS && (Input::PYTHIA6 || Input::PYTHIA8 || Input::SIMPLE || Input::GUN || Input::UPSILON || Input::HEPMC))
   {
     std::cout << "Reading Hits and running G4 simultanously is not supported" << std::endl;
     gSystem->Exit(1);
   }
   if (Input::PYTHIA6 && Input::PYTHIA8)
   {
     std::cout << "Pythia6 and Pythia8 cannot be run together - might be possible but needs R&D" << std::endl;
     gSystem->Exit(1);
   }
 
   if (INPUTHEPMC::FLOW && Input::PILEUPRATE > 0)
   {
     std::cout << "Flow Afterburner and Pileup cannot be run simultanously" << std::endl;
     gSystem->Exit(1);
   }
   // done with consistency checks, create generators in no specific order
 
   //  Fun4AllServer *se = Fun4AllServer::instance();
   if (Input::PYTHIA6)
   {
     std::cout << "Pythia6 not implemented" << std::endl;
     gSystem->Exit(1);
     // INPUTGENERATOR::Pythia6 = new PHPythia6();
     // INPUTGENERATOR::Pythia6->set_config_file(PYTHIA6::config_file);
 
     // INPUTGENERATOR::Pythia6->set_embedding_id(Input::EmbedId);
     // Input::PYTHIA6_EmbedId = Input::EmbedId;
     // Input::EmbedId++;
   }
   if (Input::PYTHIA8)
   {
     for (int i = 0; i < Input::PYTHIA8_NUMBER; i++)
     {
       std::string name = "PYTHIA_" + std::to_string(i);
       PHPythia8 *pythia8 = new PHPythia8(name);
       INPUTGENERATOR::Pythia8.push_back(pythia8);
       pythia8->set_embedding_id(Input::EmbedId);
       // see coresoftware/generators/PHPythia8 for example config
       if (PYTHIA8::config_file[i].empty())
       {
     std::cout << "No Pythia8 config file for pythia8 generator no " << i << std::endl;
     gSystem->Exit(1);
       }
       pythia8->set_config_file(PYTHIA8::config_file[i]);
       // luminosity makes no sense when running multiple pythia8 generators
       if (Input::PYTHIA8_NUMBER > 1)
       {
     pythia8->save_integrated_luminosity(false);
       }
 
       Input::PYTHIA8_EmbedIds.insert(Input::EmbedId);
       Input::EmbedId++;
       if (Input::EMBED)
       {
     pythia8->set_reuse_vertex(Input::VertexEmbedId);
       }
     }
   }
   // single particle generators
   if (Input::DZERO)
   {
     for (int i = 0; i < Input::DZERO_NUMBER; ++i)
     {
       std::string name = "DZERO_" + std::to_string(i);
       PHG4ParticleGeneratorD0 *dzero = new PHG4ParticleGeneratorD0(name);
       dzero->Embed(Input::EmbedId);
       Input::DZERO_EmbedIds.insert(Input::EmbedId);
       Input::EmbedId++;
       INPUTGENERATOR::DZeroMesonGenerator.push_back(dzero);
     }
   }
   if (Input::GUN)
   {
     for (int i = 0; i < Input::GUN_NUMBER; ++i)
     {
       std::string name = "GUN_" + std::to_string(i);
       PHG4ParticleGun *gun = new PHG4ParticleGun(name);
       gun->Embed(Input::EmbedId);
       Input::GUN_EmbedIds.insert(Input::EmbedId);
       Input::EmbedId++;
       INPUTGENERATOR::Gun.push_back(gun);
     }
   }
   if (Input::IONGUN)
   {
     for (int i = 0; i < Input::IONGUN_NUMBER; ++i)
     {
       std::string name = "IONGUN_" + std::to_string(i);
       PHG4IonGun *iongun = new PHG4IonGun(name);
       iongun->Embed(Input::EmbedId);
       Input::IONGUN_EmbedIds.insert(Input::EmbedId);
       Input::EmbedId++;
       INPUTGENERATOR::IonGun.push_back(iongun);
     }
   }
   if (Input::PGEN)
   {
     for (int i = 0; i < Input::PGEN_NUMBER; ++i)
     {
       std::string name = "PGEN_" + std::to_string(i);
       PHG4ParticleGenerator *pgen = new PHG4ParticleGenerator(name);
       pgen->Embed(Input::EmbedId);
       Input::PGEN_EmbedIds.insert(Input::EmbedId);
       Input::EmbedId++;
       INPUTGENERATOR::ParticleGenerator.push_back(pgen);
     }
   }
   if (Input::SIMPLE)
   {
     for (int i = 0; i < Input::SIMPLE_NUMBER; ++i)
     {
       std::string name = "EVTGENERATOR_" + std::to_string(i);
       PHG4SimpleEventGenerator *simple = new PHG4SimpleEventGenerator(name);
       simple->Embed(Input::EmbedId);
       Input::PGEN_EmbedIds.insert(Input::EmbedId);
       Input::EmbedId++;
       INPUTGENERATOR::SimpleEventGenerator.push_back(simple);
     }
   }
   if (Input::UPSILON)
   {
     for (int i = 0; i < Input::UPSILON_NUMBER; ++i)
     {
       std::string name = "UPSILON_" + std::to_string(i);
       PHG4ParticleGeneratorVectorMeson *upsilon = new PHG4ParticleGeneratorVectorMeson(name);
       upsilon->Embed(Input::EmbedId);
       Input::UPSILON_EmbedIds.insert(Input::EmbedId);
       Input::EmbedId++;
       INPUTGENERATOR::VectorMesonGenerator.push_back(upsilon);
     }
   }
 
   // input managers for which we might need to set options
   if (Input::HEPMC)
   {
     INPUTMANAGER::HepMCInputManager = new Fun4AllHepMCInputManager("HEPMCin");
     INPUTMANAGER::HepMCInputManager->set_embedding_id(INPUTHEPMC::EmbedId);
   }
   if (Input::PILEUPRATE > 0)
   {
     INPUTMANAGER::HepMCPileupInputManager = new Fun4AllHepMCPileupInputManager("HepMCPileupInput");
   }
 }
 
 void InputRegister()
 {
   Fun4AllServer *se = Fun4AllServer::instance();
   // if (Input::PYTHIA6)
   // {
   //   se->registerSubsystem(INPUTGENERATOR::Pythia6);
   // }
   if (Input::PYTHIA8)
   {
     int verbosity = std::max(Input::PYTHIA8_VERBOSITY, Input::VERBOSITY);
     for (auto &generator : INPUTGENERATOR::Pythia8)
     {
       generator->Verbosity(verbosity);
       se->registerSubsystem(generator);
     }
   }
   if (Input::DZERO)
   {
     int verbosity = std::max(Input::DZERO_VERBOSITY, Input::VERBOSITY);
     for (auto &generator : INPUTGENERATOR::DZeroMesonGenerator)
     {
       generator->Verbosity(verbosity);
       se->registerSubsystem(generator);
     }
   }
   if (Input::GUN)
   {
     int verbosity = std::max(Input::GUN_VERBOSITY, Input::VERBOSITY);
     for (auto &generator : INPUTGENERATOR::Gun)
     {
       generator->Verbosity(verbosity);
       se->registerSubsystem(generator);
     }
   }
   if (Input::IONGUN)
   {
     int verbosity = std::max(Input::IONGUN_VERBOSITY, Input::VERBOSITY);
     for (auto &generator : INPUTGENERATOR::IonGun)
     {
       generator->Verbosity(verbosity);
       se->registerSubsystem(generator);
     }
   }
   if (Input::PGEN)
   {
     int verbosity = std::max(Input::PGEN_VERBOSITY, Input::VERBOSITY);
     for (auto &generator : INPUTGENERATOR::ParticleGenerator)
     {
       generator->Verbosity(verbosity);
       se->registerSubsystem(generator);
     }
   }
   if (Input::SIMPLE)
   {
     int verbosity = std::max(Input::SIMPLE_VERBOSITY, Input::VERBOSITY);
     for (auto &generator : INPUTGENERATOR::SimpleEventGenerator)
     {
       generator->Verbosity(verbosity);
       se->registerSubsystem(generator);
     }
   }
   if (Input::UPSILON)
   {
     for (auto &generator : INPUTGENERATOR::VectorMesonGenerator)
     {
       int verbosity = std::max(Input::UPSILON_VERBOSITY, Input::VERBOSITY);
       if (Input::HEPMC || Input::SIMPLE)
       {
         generator->set_reuse_existing_vertex(true);
       }
       generator->Verbosity(verbosity);
       se->registerSubsystem(generator);
     }
   }
   if (Input::READEIC)
   {
     std::cout << "Eic File Reading disabled" << std::endl;
     gSystem->Exit(1);
     // INPUTGENERATOR::EICFileReader = new ReadEICFiles();
     // INPUTGENERATOR::EICFileReader->OpenInputFile(INPUTREADEIC::filename);
     // INPUTGENERATOR::EICFileReader->Verbosity(Input::VERBOSITY);
     // se->registerSubsystem(INPUTGENERATOR::EICFileReader);
   }
   if (Input::COSMIC)
   {
     INPUTGENERATOR::Cosmic = new CosmicSpray("COSMIC", Input::COSMIC_R);
     se->registerSubsystem(INPUTGENERATOR::Cosmic);
   }
   // here are the various utility modules which read particles and
   // put them onto the G4 particle stack
   if (Input::HEPMC || Input::PYTHIA8 || Input::PYTHIA6 || Input::READEIC)
   {
     if (Input::HEPMC)
     {
       if (INPUTHEPMC::REACTIONPLANERAND)
       {
         ReactionPlaneAfterburner *rp = new ReactionPlaneAfterburner();
         se->registerSubsystem(rp);
       }
 
       if (INPUTHEPMC::HIJINGFLIP)
       {
         HIJINGFlipAfterburner *flip = new HIJINGFlipAfterburner();
         se->registerSubsystem(flip);
       }
       // these need to be applied before the HepMCNodeReader since they
       // work on the hepmc records
       if (INPUTHEPMC::FLOW)
       {
         HepMCFlowAfterBurner *burn = new HepMCFlowAfterBurner();
         burn->Verbosity(INPUTHEPMC::FLOW_VERBOSITY);
         burn->enableFluctuations(INPUTHEPMC::FLOW_FLUCTUATIONS);
         burn->scaleFlow(INPUTHEPMC::FLOW_SCALING);
         se->registerSubsystem(burn);
       }
       if (INPUTHEPMC::FERMIMOTION)
       {
         FermimotionAfterburner *fermi = new FermimotionAfterburner();
         se->registerSubsystem(fermi);
       }
     }
     // copy HepMC records into G4
     HepMCNodeReader *hr = new HepMCNodeReader();
     if (INPUTHEPMC::HEPMC_STRANGENESS_FRACTION >= 0)
     {
       hr->AddStrangeness(INPUTHEPMC::HEPMC_STRANGENESS_FRACTION);
     }
     se->registerSubsystem(hr);
   }
 }
 
 void InputManagers()
 {
   Fun4AllServer *se = Fun4AllServer::instance();
   if (Input::EMBED)
   {
     gSystem->Load("libg4dst.so");
     if (!INPUTEMBED::filename.empty() && !INPUTEMBED::listfile.empty())
     {
       std::cout << "only filenames or filelists are supported, not mixtures" << std::endl;
       gSystem->Exit(1);
     }
     if (INPUTEMBED::filename.empty() && INPUTEMBED::listfile.empty())
     {
       std::cout << "you need to give an input filenames or filelist" << std::endl;
       gSystem->Exit(1);
     }
     for (auto &iter : INPUTEMBED::filename)
     {
       std::string mgrname = "DSTin" + std::to_string(iter.first);
       Fun4AllInputManager *hitsin = new Fun4AllDstInputManager(mgrname);
       hitsin->fileopen(iter.second);
       hitsin->Verbosity(Input::VERBOSITY);
       if (INPUTEMBED::REPEAT)
       {
         hitsin->Repeat();
       }
       se->registerInputManager(hitsin);
     }
     for (auto &iter : INPUTEMBED::listfile)
     {
       std::string mgrname = "DSTin" + std::to_string(iter.first);
       Fun4AllInputManager *hitsin = new Fun4AllDstInputManager(mgrname);
       hitsin->AddListFile(iter.second);
       hitsin->Verbosity(Input::VERBOSITY);
       if (INPUTEMBED::REPEAT)
       {
         hitsin->Repeat();
       }
       se->registerInputManager(hitsin);
     }
   }
   if (Input::HEPMC)
   {
     INPUTMANAGER::HepMCInputManager->Verbosity(Input::VERBOSITY);
     se->registerInputManager(INPUTMANAGER::HepMCInputManager);
     if (!INPUTHEPMC::filename.empty() && INPUTHEPMC::listfile.empty())
     {
       INPUTMANAGER::HepMCInputManager->fileopen(INPUTHEPMC::filename);
     }
     else if (!INPUTHEPMC::listfile.empty())
     {
       INPUTMANAGER::HepMCInputManager->AddListFile(INPUTHEPMC::listfile);
     }
     else
     {
       std::cout << "no filename INPUTHEPMC::filename or listfile INPUTHEPMC::listfile given" << std::endl;
       gSystem->Exit(1);
     }
   }
   else if (Input::READHITS)
   {
     gSystem->Load("libg4dst.so");
     if (!INPUTREADHITS::filename.empty() && !INPUTREADHITS::listfile.empty())
     {
       std::cout << "only filenames or filelists are supported, not mixtures" << std::endl;
       gSystem->Exit(1);
     }
     if (INPUTREADHITS::filename.empty() && INPUTREADHITS::listfile.empty())
     {
       std::cout << "you need to give an input filenames or filelist" << std::endl;
       gSystem->Exit(1);
     }
     for (auto &iter : INPUTREADHITS::filename)
     {
       std::string mgrname = "DSTin" + std::to_string(iter.first);
       Fun4AllInputManager *hitsin = new Fun4AllDstInputManager(mgrname);
       hitsin->fileopen(iter.second);
       hitsin->Verbosity(Input::VERBOSITY);
       se->registerInputManager(hitsin);
     }
     for (auto &iter : INPUTREADHITS::listfile)
     {
       std::string mgrname = "DSTin" + std::to_string(iter.first);
       Fun4AllInputManager *hitsin = new Fun4AllDstInputManager(mgrname);
       hitsin->AddListFile(iter.second);
       hitsin->Verbosity(Input::VERBOSITY);
       se->registerInputManager(hitsin);
     }
   }
   else
   {
     Fun4AllInputManager *in = new Fun4AllDummyInputManager("JADE");
     in->Verbosity(Input::VERBOSITY);
     se->registerInputManager(in);
   }
   if (Input::PILEUPRATE > 0)
   {
     INPUTMANAGER::HepMCPileupInputManager->SignalInputManager(INPUTMANAGER::HepMCInputManager);
     INPUTMANAGER::HepMCPileupInputManager->Verbosity(Input::VERBOSITY);
     INPUTMANAGER::HepMCPileupInputManager->AddFile(PILEUP::pileupfile);
     INPUTMANAGER::HepMCPileupInputManager->set_collision_rate(Input::PILEUPRATE);
     double time_window = G4TPC::maxDriftLength / PILEUP::TpcDriftVelocity;
     double extended_readout_time = 0.0;
     if (TRACKING::pp_mode)
     {
       extended_readout_time = TRACKING::pp_extended_readout_time;
     }
     INPUTMANAGER::HepMCPileupInputManager->set_time_window(-time_window, time_window + extended_readout_time);
     std::cout << "Pileup window is from " << -time_window << " to " << time_window + extended_readout_time << std::endl;
     se->registerInputManager(INPUTMANAGER::HepMCPileupInputManager);
   }
 }
 #endif

This page was automatically generated by the 2.3.7 LXR engine. The LXR team