Adding script for Nuclei at Fwd & rapidity range in the .gun file

Author: lucamicheletti

MC/config/PWGLF/ini/GeneratorLFNucleiFwdppGap.ini

@@ -0,0 +1,6 @@
+[GeneratorExternal]
+fileName=${O2DPG_ROOT}/MC/config/PWGLF/pythia8/generator_pythia8_longlived_gaptriggered.C
+funcName=generateLongLivedGapTriggered("${O2DPG_ROOT}/MC/config/PWGLF/pythia8/generator/nuclei_fwd.gun", 5)
+
+[GeneratorPythia8]
+config=${O2_ROOT}/share/Generators/egconfig/pythia8_inel.cfg

MC/config/PWGLF/pythia8/generator/hypernuclei.gun

@@ -1,6 +1,6 @@
-# PDG           N      ptMin   ptMax
-1000010020      1      0.2     6
-1000010030      1      0.2     6
-1000020030      1      0.2     6
-1000020040      1      0.2     6
-1010010030      1      0.2     6
+# PDG           N      ptMin   ptMax   yMin   yMax
+1000010020      1      0.2     6       -1     1
+1000010030      1      0.2     6       -1     1
+1000020030      1      0.2     6       -1     1
+1000020040      1      0.2     6       -1     1
+1010010030      1      0.2     6       -1     1

MC/config/PWGLF/pythia8/generator/nuclei_fwd.gun

@@ -0,0 +1,2 @@
+# PDG           N      ptMin   ptMax   yMin   yMax
+1000020030      10     0.      10.     -4.    -2.5

MC/config/PWGLF/pythia8/generator_pythia8_longlived_gaptriggered.C

@@ -21,12 +21,13 @@ class GeneratorPythia8LongLivedGapTriggered : public o2::eventgen::GeneratorPyth
 {
 public:
   /// Constructor
-  GeneratorPythia8LongLivedGapTriggered(std::vector<int> input_pdg, int input_trigger_ratio = 1, int n_injected = 1, float pt_min = 1, float pt_max = 10)
+  GeneratorPythia8LongLivedGapTriggered(std::vector<int> input_pdg, int input_trigger_ratio = 1, int n_injected = 1, float pt_min = 1, float pt_max = 10, float y_min = -1, float y_max = 1)
   {
     mPdg = input_pdg;
     setNinjected(n_injected);
     mInverseTriggerRatio = input_trigger_ratio;
     setPt(pt_min, pt_max);
+    setY(y_min, y_max);
     mMass = getMass(input_pdg);
     mGeneratedEvents = 0;
     mAlternatingPDGsign = true;
@@ -42,18 +43,22 @@ public:
     int pdg = 0;
     unsigned long n_inj = 0;
     float pt_min = 0.;
-    float pt_max = 0;
+    float pt_max = 0.;
+    float y_min = 0.;
+    float y_max = 0.;
     if (!config_file.is_open())
     {
       LOGF(fatal, "File %s cannot be opened.", expanded_file_name);
     }
     std::getline(config_file, header); // skip first line
-    while (config_file >> pdg >> n_inj >> pt_min >> pt_max)
+    while (config_file >> pdg >> n_inj >> pt_min >> pt_max >> y_min >> y_max)
     {
       mPdg.push_back(pdg);
       mNinjected.push_back(n_inj);
       mPtMin.push_back(pt_min);
       mPtMax.push_back(pt_max);
+      mYmin.push_back(y_min);
+      mYmax.push_back(y_max);
     }
     config_file.close();
     mInverseTriggerRatio = input_trigger_ratio;
@@ -81,8 +86,11 @@ public:
   /// Set rapidity
   void setY(float y_min, float y_max)
   {
-    mYmin = y_min;
-    mYmax = y_max;
+    for (auto part : mPdg)
+    {
+      mYmin.push_back(y_min);
+      mYmax.push_back(y_max);
+    }
   }
 
   /// Set pseudorapidity
@@ -130,7 +138,7 @@ public:
       for (int i = 0; i < mNinjected[injectionIndex]; ++i)
       {
         const double pt = gRandom->Uniform(mPtMin[injectionIndex], mPtMax[injectionIndex]);
-        const double rapidity = gRandom->Uniform(mYmin, mYmax);
+        const double rapidity = gRandom->Uniform(mYmin[injectionIndex], mYmax[injectionIndex]);
         const double phi = gRandom->Uniform(0, TMath::TwoPi());
         const double px{pt * std::cos(phi)};
         const double py{pt * std::sin(phi)};
@@ -157,8 +165,8 @@ private:
 
   std::vector<double> mPtMin; /// minimum transverse momentum for generated particles
   std::vector<double> mPtMax; /// maximum transverse momentum for generated particles
-  double mYmin = -1.;         /// minimum rapidity for generated particles
-  double mYmax = +1.;         /// maximum rapidity for generated particles
+  std::vector<double> mYmin; /// minimum rapidity for generated particles
+  std::vector<double> mYmax; /// maximum rapidity for generated particles
 
   bool mAlternatingPDGsign = true; /// bool to randomize the PDG code of the core particle
 
@@ -170,9 +178,9 @@ private:
 };
 
 ///___________________________________________________________
-FairGenerator *generateLongLivedGapTriggered(std::vector<int> mPdg, int input_trigger_ratio, int n_injected = 1, float pt_min = 1, float pt_max = 10, bool alternate_sign = true)
+FairGenerator *generateLongLivedGapTriggered(std::vector<int> mPdg, int input_trigger_ratio, int n_injected = 1, float pt_min = 1, float pt_max = 10, float y_min = -1, float y_max = 1, bool alternate_sign = true)
 {
-  auto myGen = new GeneratorPythia8LongLivedGapTriggered(mPdg, input_trigger_ratio, n_injected, pt_min, pt_max);
+  auto myGen = new GeneratorPythia8LongLivedGapTriggered(mPdg, input_trigger_ratio, n_injected, pt_min, pt_max, y_min, y_max);
   myGen->setAlternatingPDGsign(alternate_sign);
   auto seed = (gRandom->TRandom::GetSeed() % 900000000);
   myGen->readString("Random:setSeed on");

MC/run/PWGLF/run_NucleiFwdInjectedGap.sh

@@ -0,0 +1,33 @@
+#!/bin/bash
+
+#
+# A example workflow MC->RECO->AOD for a simple pp min bias
+# production, targetting test beam conditions.
+
+# make sure O2DPG + O2 is loaded
+[ ! "${O2DPG_ROOT}" ] && echo "Error: This needs O2DPG loaded" && exit 1
+[ ! "${O2_ROOT}" ] && echo "Error: This needs O2 loaded" && exit 1
+
+# ----------- LOAD UTILITY FUNCTIONS --------------------------
+. ${O2_ROOT}/share/scripts/jobutils.sh
+
+# ----------- START ACTUAL JOB  -----------------------------
+
+NWORKERS=${NWORKERS:-8}
+MODULES="--skipModules ZDC"
+SIMENGINE=${SIMENGINE:-TGeant3}
+NSIGEVENTS=${NSIGEVENTS:-1}
+NBKGEVENTS=${NBKGEVENTS:-1}
+NTIMEFRAMES=${NTIMEFRAMES:-1}
+INTRATE=${INTRATE:-500000}
+SYSTEM=${SYSTEM:-pp}
+ENERGY=${ENERGY:-13600}
+[[ ${SPLITID} != "" ]] && SEED="-seed ${SPLITID}" || SEED=""
+
+# create workflow
+${O2DPG_ROOT}/MC/bin/o2dpg_sim_workflow.py -eCM ${ENERGY} -col ${SYSTEM} -gen external -j ${NWORKERS} -ns ${NSIGEVENTS} -tf ${NTIMEFRAMES} -interactionRate ${INTRATE} -confKey "Diamond.width[0]=0.1;Diamond.width[1]=0.1;Diamond.width[2]=6." -e ${SIMENGINE} ${SEED} -mod "--skipModules ZDC" \
+        -ini ${O2DPG_ROOT}/MC/config/PWGLF/ini/GeneratorLFNucleiFwd${SYSTEM}Gap.ini
+
+# run workflow
+# allow increased timeframe parallelism with --cpu-limit 32
+${O2DPG_ROOT}/MC/bin/o2_dpg_workflow_runner.py -f workflow.json -tt aod --cpu-limit 32