Adding eta cuts to the trigger (AliceO2Group#1700)

* PWGUD Starlight interface * Generator tests * Adjusting links * Eta cuts in trigger * Skip photons
fcatalan92 · Aug 8, 2024 · 99e5525 · 99e5525
1 parent 4c65dcd
commit 99e5525
Show file tree

Hide file tree

Showing 4 changed files with 32 additions and 9 deletions.
diff --git a/MC/config/PWGUD/ini/GenStarlight_kCohJpsiToMu_PbPb_5360_Muon.ini b/MC/config/PWGUD/ini/GenStarlight_kCohJpsiToMu_PbPb_5360_Muon.ini
@@ -3,4 +3,4 @@ fileName = ${O2DPG_ROOT}/MC/config/PWGUD/external/generator/GeneratorStarlight.C
 funcName = GeneratorStarlight("kCohJpsiToMu", 5360.000000, 82, 208, 82, 208)  
 [TriggerExternal] 
 fileName = ${O2DPG_ROOT}/MC/config/PWGUD/trigger/selectParticlesInAcceptance.C 
-funcName = selectMotherPartInAcc(-4.0,-2.5) 
+funcName = selectDaughterPartInAcc(-4.0,-2.5) 
diff --git a/MC/config/PWGUD/ini/makeStarlightConfig.py b/MC/config/PWGUD/ini/makeStarlightConfig.py
@@ -14,7 +14,7 @@
 parser.add_argument('--eCM', type=float, default='5360',
                     help='Centre-of-mass energy')
 
-parser.add_argument('--rapidity', default='cent', choices=['cent', 'muon'],
+parser.add_argument('--rapidity', default='cent', choices=['cent_rap', 'muon_rap', 'cent_eta', 'muon_eta'],
                     help='Rapidity to select')
 
 parser.add_argument('--process',default=None, choices=['kTwoGammaToMuLow', 'kTwoGammaToElLow', 'kTwoGammaToMuMedium', 'kTwoGammaToElMedium', 'kTwoGammaToMuHigh', 'kTwoGammaToElHigh', 'kTwoGammaToRhoRho', 'kTwoGammaToF2', 'kCohRhoToPi', 'kCohRhoToElEl', 'kCohRhoToMuMu', 'kCohRhoToPiWithCont', 'kCohRhoToPiFlat', 'kCohPhiToKa', 'kDirectPhiToKaKa','kCohOmegaTo2Pi', 'kCohOmegaTo3Pi', 'kCohOmegaToPiPiPi', 'kCohJpsiToMu', 'kCohJpsiToEl', 'kCohJpsiToElRad', 'kCohJpsiToProton', 'kCohPsi2sToMu','kCohPsi2sToEl', 'kCohPsi2sToMuPi', 'kCohPsi2sToElPi', 'kCohUpsilonToMu', 'kCohUpsilonToEl', 'kIncohRhoToPi', 'kIncohRhoToElEl', 'kIncohRhoToMuMu', 'kIncohRhoToPiWithCont', 'kIncohRhoToPiFlat', 'kIncohPhiToKa', 'kIncohOmegaTo2Pi', 'kIncohOmegaTo3Pi', 'kIncohOmegaToPiPiPi', 'kIncohJpsiToMu', 'kIncohJpsiToEl', 'kIncohJpsiToElRad', 'kIncohJpsiToProton', 'kIncohJpsiToLLbar', 'kIncohPsi2sToMu', 'kIncohPsi2sToEl', 'kIncohPsi2sToMuPi', 'kIncohPsi2sToElPi', 'kIncohUpsilonToMu', 'kIncohUpsilonToEl'],
@@ -78,10 +78,20 @@
 ###Trigger
 fout.write('[TriggerExternal] \n')
 fout.write('fileName = ${O2DPG_ROOT}/MC/config/PWGUD/trigger/selectParticlesInAcceptance.C \n')
-if args.rapidity == 'cent':
-    fout.write('funcName = selectMotherPartInAcc(-0.9,0.9) \n')
-if args.rapidity == 'muon':
-    fout.write('funcName = selectMotherPartInAcc(-4.0,-2.5) \n')
+if 'kTwoGamma' in args.process:
+    if args.rapidity == 'cent_eta':
+        fout.write('funcName = selectDirectPartInAcc(-0.9,0.9) \n')
+    if args.rapidity == 'muon_eta':
+        fout.write('funcName = selectDirectPartInAcc(-4.0,-2.5) \n')
+else:
+    if args.rapidity == 'cent_rap':
+        fout.write('funcName = selectMotherPartInAcc(-0.9,0.9) \n')
+    if args.rapidity == 'muon_rap':
+        fout.write('funcName = selectMotherPartInAcc(-4.0,-2.5) \n')
+    if args.rapidity == 'cent_eta':
+        fout.write('funcName = selectDaughterPartInAcc(-0.9,0.9) \n')
+    if args.rapidity == 'muon_eta':
+        fout.write('funcName = selectDaughterPartInAcc(-4.0,-2.5) \n')
 
 ### close outout file
 fout.close()
diff --git a/MC/config/PWGUD/trigger/selectParticlesInAcceptance.C b/MC/config/PWGUD/trigger/selectParticlesInAcceptance.C
@@ -22,7 +22,20 @@ o2::eventgen::Trigger selectDaughterPartInAcc(double etaMin = -1., double etaMax
 {
   return [etaMin, etaMax](const std::vector<TParticle>& particles) -> bool {
     for (const auto& particle : particles) {
-      if (particle.GetFirstMother() != -1)
+      if (particle.GetFirstMother() == -1)
+        if ((particle.Y() < etaMin) || (particle.Y() > etaMax)) return kFALSE;
+	  if (particle.GetFirstMother() != -1 && particle.GetFirstDaughter() == -1 && particle.GetPdgCode() != 22)
+		  if ((particle.Eta() < etaMin) || (particle.Eta() > etaMax)) return kFALSE; 
+    }
+    return kTRUE;  
+  };
+}
+
+o2::eventgen::Trigger selectDirectPartInAcc(double etaMin = -1., double etaMax = -1.)
+{
+  return [etaMin, etaMax](const std::vector<TParticle>& particles) -> bool {
+    for (const auto& particle : particles) {
+      if (particle.GetFirstMother() == -1)
         if ((particle.Eta() < etaMin) || (particle.Eta() > etaMax))
 	  return kFALSE;
     }

diff --git a/MC/run/PWGUD/runSTARlightANCHOR.sh b/MC/run/PWGUD/runSTARlightANCHOR.sh
@@ -21,7 +21,7 @@ export PRODSPLIT=100
 export CYCLE=0
 export ALIEN_PROC_ID=2963436952
 
-export ALIEN_JDL_ANCHOR_SIM_OPTIONS="-gen external -ini ${PWD}/GenStarlight.ini -nb ${NBKGEVENTS} -colBkg PbPb -genBkg pythia8 -procBkg heavy_ion"
+export ALIEN_JDL_ANCHOR_SIM_OPTIONS="-gen external -ini ${PWD}/GenStarlight.ini --embedding -nb ${NBKGEVENTS} -colBkg PbPb -genBkg pythia8 -procBkg heavy_ion"
 
-${O2DPG_ROOT}/MC/config/PWGUD/ini/makeStarlightConfig.py --process kCohPsi2sToMuPi --collType PbPb --eCM 5360 --rapidity cent
+${O2DPG_ROOT}/MC/config/PWGUD/ini/makeStarlightConfig.py --process kCohPsi2sToMuPi --collType PbPb --eCM 5360 --rapidity cent_eta
 ${O2DPG_ROOT}/MC/run/ANCHOR/anchorMC.sh