From 3ebb5756828705cdc47c69c311ed82705841a63a Mon Sep 17 00:00:00 2001
From: Ionut Cristian Arsene <iarsene@cern.ch>
Date: Thu, 13 Feb 2025 13:21:47 +0100
Subject: [PATCH] Adding a generator for combined inclusive jpsi and psi2S via
 hadron triggered pythia

---
 ...enerator_pythia8_HadronTriggered_withGap.C | 37 +++++++++++++++++++
 ...lusiveJpsiPsi2SMidy_Pythia8_TriggerGap.ini |  7 ++++
 .../PWGDQ/pythia8/decayer/switchOffJpsi.cfg   |  1 +
 3 files changed, 45 insertions(+)
 create mode 100644 MC/config/PWGDQ/ini/Generator_InjectedInclusiveJpsiPsi2SMidy_Pythia8_TriggerGap.ini

diff --git a/MC/config/PWGDQ/external/generator/generator_pythia8_HadronTriggered_withGap.C b/MC/config/PWGDQ/external/generator/generator_pythia8_HadronTriggered_withGap.C
index 38bc79ec4..8e8f46938 100644
--- a/MC/config/PWGDQ/external/generator/generator_pythia8_HadronTriggered_withGap.C
+++ b/MC/config/PWGDQ/external/generator/generator_pythia8_HadronTriggered_withGap.C
@@ -173,5 +173,42 @@ FairGenerator*
   // print debug
   // gen->PrintDebug();
 
+  return gen;
+
+}
+
+// Predefined generators:
+FairGenerator*
+  GeneratorInclusiveJpsiPsi2S_EvtGenMidY(int triggerGap, double rapidityMin = -1.5, double rapidityMax = 1.5, bool verbose = false)
+{
+  auto gen = new o2::eventgen::GeneratorEvtGen<o2::eventgen::GeneratorPythia8HadronTriggeredWithGap>();
+  gen->setTriggerGap(triggerGap);
+  gen->setRapidityRange(rapidityMin, rapidityMax);
+  gen->addHadronPDGs(443);
+  gen->addHadronPDGs(100443);
+  gen->setVerbose(verbose);
+
+  TString pathO2table = gSystem->ExpandPathName("${O2DPG_MC_CONFIG_ROOT}/MC/config/PWGDQ/pythia8/decayer/switchOffJpsi.cfg");
+  gen->readFile(pathO2table.Data());
+  gen->setConfigMBdecays(pathO2table);
+  gen->PrintDebug(true);
+
+  gen->SetSizePdg(2);
+  gen->AddPdg(443, 0);
+  gen->AddPdg(100443, 1);
+
+  gen->SetForceDecay(kEvtDiElectron);
+
+  // set random seed
+  gen->readString("Random:setSeed on");
+  uint random_seed;
+  unsigned long long int random_value = 0;
+  ifstream urandom("/dev/urandom", ios::in|ios::binary);
+  urandom.read(reinterpret_cast<char*>(&random_value), sizeof(random_seed));
+  gen->readString(Form("Random:seed = %llu", random_value % 900000001));
+
+  // print debug
+  // gen->PrintDebug();
+
   return gen;
 }
\ No newline at end of file
diff --git a/MC/config/PWGDQ/ini/Generator_InjectedInclusiveJpsiPsi2SMidy_Pythia8_TriggerGap.ini b/MC/config/PWGDQ/ini/Generator_InjectedInclusiveJpsiPsi2SMidy_Pythia8_TriggerGap.ini
new file mode 100644
index 000000000..1c6f929ac
--- /dev/null
+++ b/MC/config/PWGDQ/ini/Generator_InjectedInclusiveJpsiPsi2SMidy_Pythia8_TriggerGap.ini
@@ -0,0 +1,7 @@
+### The external generator derives from GeneratorPythia8.
+[GeneratorExternal]
+fileName=${O2DPG_MC_CONFIG_ROOT}/MC/config/PWGDQ/external/generator/generator_pythia8_HadronTriggered_withGap.C
+funcName=GeneratorInclusiveJpsiPsi2S_EvtGenMidY(5,-1.5,1.5)
+
+[GeneratorPythia8]
+config=${O2DPG_MC_CONFIG_ROOT}/MC/config/PWGDQ/pythia8/generator/pythia8_inel_triggerGap.cfg
\ No newline at end of file
diff --git a/MC/config/PWGDQ/pythia8/decayer/switchOffJpsi.cfg b/MC/config/PWGDQ/pythia8/decayer/switchOffJpsi.cfg
index da279b4aa..2cc29e535 100644
--- a/MC/config/PWGDQ/pythia8/decayer/switchOffJpsi.cfg
+++ b/MC/config/PWGDQ/pythia8/decayer/switchOffJpsi.cfg
@@ -1,3 +1,4 @@
 #switch off jpsi decays
 
 443:mayDecay  off	# J/psi
+100443:mayDecay  off	# psi(2S)