[PWGLF] Add single-channel K1(1270) and Omega(2012) injection configs for ML signal samples

One stratum (charge x decay channel) per job: one injected parent per pp 13.6 TeV inelastic event,
no gap, |y| < 0.7, flat pT (K1: 0-8 GeV/c, Omega(2012): 0-15 GeV/c), finite-width sampling with
generator_pythia8_LF_rapidity_width.C.

- K1(1270)+-: K rho0 (rho0 -> pi+ pi-) and K*(892) pi (K* -> K pi)
- Omega(2012)-+: Xi K0S and Xi(1530)0 K (Xi(1530)0 -> Xi pi)

The forced-decay tables are new files; resonances_width.cfg and the existing recipes are unchanged.
Branching fractions are forced to unity and must be weighted in the analysis. Xi, Lambda and K0S are
decayed by the transport code with their natural branching fractions.

Each INI redirects to a per-channel test that requires exactly one injected parent per event and the
forced chain for both charges; parents produced in the underlying event are counted but not checked.

Author: Bong-Hwi Lim

MC/config/PWGLF/ini/GeneratorLF_ResonancesML_K1Minus_KstarPi_pp1360.ini

@@ -0,0 +1,13 @@
+#---> GeneratorLF_ResonancesML_K1_KstarPi_pp1360
+# Single-channel ML signal sample: one K1(1270)- per pp 13.6 TeV inelastic event (no gap), flat pT 0-8 GeV/c,
+# |y| < 0.7, forced decay chain from resonance_ml_k1_kstarpi.cfg. One stratum (charge x channel) per job.
+[GeneratorExternal]
+fileName=${O2DPG_MC_CONFIG_ROOT}/MC/config/PWGLF/pythia8/generator_pythia8_LF_rapidity_width.C
+funcName=generateLFRapidity("${O2DPG_MC_CONFIG_ROOT}/MC/config/PWGLF/pythia8/generator/resonancegun_ml_k1minus_kstarpi.json", true, 0, false, true, "${O2DPG_MC_CONFIG_ROOT}/MC/config/PWGLF/pythia8/generator/pythia8_inel_136tev.cfg", "")
+
+[GeneratorPythia8] # if triggered then this will be used as the background event
+config=${O2DPG_MC_CONFIG_ROOT}/MC/config/PWGLF/pythia8/generator/pythia8_inel_136tev.cfg
+
+[DecayerPythia8] # after for transport code!
+config[0]=${O2DPG_MC_CONFIG_ROOT}/MC/config/common/pythia8/decayer/base.cfg
+config[1]=${O2DPG_MC_CONFIG_ROOT}/MC/config/PWGLF/pythia8/generator/resonance_ml_k1_kstarpi.cfg

MC/config/PWGLF/ini/GeneratorLF_ResonancesML_K1Minus_RhoK_pp1360.ini

@@ -0,0 +1,13 @@
+#---> GeneratorLF_ResonancesML_K1_RhoK_pp1360
+# Single-channel ML signal sample: one K1(1270)- per pp 13.6 TeV inelastic event (no gap), flat pT 0-8 GeV/c,
+# |y| < 0.7, forced decay chain from resonance_ml_k1_rhok.cfg. One stratum (charge x channel) per job.
+[GeneratorExternal]
+fileName=${O2DPG_MC_CONFIG_ROOT}/MC/config/PWGLF/pythia8/generator_pythia8_LF_rapidity_width.C
+funcName=generateLFRapidity("${O2DPG_MC_CONFIG_ROOT}/MC/config/PWGLF/pythia8/generator/resonancegun_ml_k1minus_rhok.json", true, 0, false, true, "${O2DPG_MC_CONFIG_ROOT}/MC/config/PWGLF/pythia8/generator/pythia8_inel_136tev.cfg", "")
+
+[GeneratorPythia8] # if triggered then this will be used as the background event
+config=${O2DPG_MC_CONFIG_ROOT}/MC/config/PWGLF/pythia8/generator/pythia8_inel_136tev.cfg
+
+[DecayerPythia8] # after for transport code!
+config[0]=${O2DPG_MC_CONFIG_ROOT}/MC/config/common/pythia8/decayer/base.cfg
+config[1]=${O2DPG_MC_CONFIG_ROOT}/MC/config/PWGLF/pythia8/generator/resonance_ml_k1_rhok.cfg

MC/config/PWGLF/ini/GeneratorLF_ResonancesML_K1Plus_KstarPi_pp1360.ini

@@ -0,0 +1,13 @@
+#---> GeneratorLF_ResonancesML_K1_KstarPi_pp1360
+# Single-channel ML signal sample: one K1(1270)+ per pp 13.6 TeV inelastic event (no gap), flat pT 0-8 GeV/c,
+# |y| < 0.7, forced decay chain from resonance_ml_k1_kstarpi.cfg. One stratum (charge x channel) per job.
+[GeneratorExternal]
+fileName=${O2DPG_MC_CONFIG_ROOT}/MC/config/PWGLF/pythia8/generator_pythia8_LF_rapidity_width.C
+funcName=generateLFRapidity("${O2DPG_MC_CONFIG_ROOT}/MC/config/PWGLF/pythia8/generator/resonancegun_ml_k1plus_kstarpi.json", true, 0, false, true, "${O2DPG_MC_CONFIG_ROOT}/MC/config/PWGLF/pythia8/generator/pythia8_inel_136tev.cfg", "")
+
+[GeneratorPythia8] # if triggered then this will be used as the background event
+config=${O2DPG_MC_CONFIG_ROOT}/MC/config/PWGLF/pythia8/generator/pythia8_inel_136tev.cfg
+
+[DecayerPythia8] # after for transport code!
+config[0]=${O2DPG_MC_CONFIG_ROOT}/MC/config/common/pythia8/decayer/base.cfg
+config[1]=${O2DPG_MC_CONFIG_ROOT}/MC/config/PWGLF/pythia8/generator/resonance_ml_k1_kstarpi.cfg

MC/config/PWGLF/ini/GeneratorLF_ResonancesML_K1Plus_RhoK_pp1360.ini

@@ -0,0 +1,13 @@
+#---> GeneratorLF_ResonancesML_K1_RhoK_pp1360
+# Single-channel ML signal sample: one K1(1270)+ per pp 13.6 TeV inelastic event (no gap), flat pT 0-8 GeV/c,
+# |y| < 0.7, forced decay chain from resonance_ml_k1_rhok.cfg. One stratum (charge x channel) per job.
+[GeneratorExternal]
+fileName=${O2DPG_MC_CONFIG_ROOT}/MC/config/PWGLF/pythia8/generator_pythia8_LF_rapidity_width.C
+funcName=generateLFRapidity("${O2DPG_MC_CONFIG_ROOT}/MC/config/PWGLF/pythia8/generator/resonancegun_ml_k1plus_rhok.json", true, 0, false, true, "${O2DPG_MC_CONFIG_ROOT}/MC/config/PWGLF/pythia8/generator/pythia8_inel_136tev.cfg", "")
+
+[GeneratorPythia8] # if triggered then this will be used as the background event
+config=${O2DPG_MC_CONFIG_ROOT}/MC/config/PWGLF/pythia8/generator/pythia8_inel_136tev.cfg
+
+[DecayerPythia8] # after for transport code!
+config[0]=${O2DPG_MC_CONFIG_ROOT}/MC/config/common/pythia8/decayer/base.cfg
+config[1]=${O2DPG_MC_CONFIG_ROOT}/MC/config/PWGLF/pythia8/generator/resonance_ml_k1_rhok.cfg

MC/config/PWGLF/ini/GeneratorLF_ResonancesML_Omega2012Minus_Xi1530K_pp1360.ini

@@ -0,0 +1,13 @@
+#---> GeneratorLF_ResonancesML_Omega2012_Xi1530K_pp1360
+# Single-channel ML signal sample: one Omega(2012)- per pp 13.6 TeV inelastic event (no gap), flat pT 0-15 GeV/c,
+# |y| < 0.7, forced decay chain from resonance_ml_omega2012_xi1530k.cfg. One stratum (charge x channel) per job.
+[GeneratorExternal]
+fileName=${O2DPG_MC_CONFIG_ROOT}/MC/config/PWGLF/pythia8/generator_pythia8_LF_rapidity_width.C
+funcName=generateLFRapidity("${O2DPG_MC_CONFIG_ROOT}/MC/config/PWGLF/pythia8/generator/resonancegun_ml_omega2012minus_xi1530k.json", true, 0, false, true, "${O2DPG_MC_CONFIG_ROOT}/MC/config/PWGLF/pythia8/generator/pythia8_inel_136tev.cfg", "")
+
+[GeneratorPythia8] # if triggered then this will be used as the background event
+config=${O2DPG_MC_CONFIG_ROOT}/MC/config/PWGLF/pythia8/generator/pythia8_inel_136tev.cfg
+
+[DecayerPythia8] # after for transport code!
+config[0]=${O2DPG_MC_CONFIG_ROOT}/MC/config/common/pythia8/decayer/base.cfg
+config[1]=${O2DPG_MC_CONFIG_ROOT}/MC/config/PWGLF/pythia8/generator/resonance_ml_omega2012_xi1530k.cfg

MC/config/PWGLF/ini/GeneratorLF_ResonancesML_Omega2012Minus_XiK0s_pp1360.ini

@@ -0,0 +1,13 @@
+#---> GeneratorLF_ResonancesML_Omega2012_XiK0s_pp1360
+# Single-channel ML signal sample: one Omega(2012)- per pp 13.6 TeV inelastic event (no gap), flat pT 0-15 GeV/c,
+# |y| < 0.7, forced decay chain from resonance_ml_omega2012_xik0s.cfg. One stratum (charge x channel) per job.
+[GeneratorExternal]
+fileName=${O2DPG_MC_CONFIG_ROOT}/MC/config/PWGLF/pythia8/generator_pythia8_LF_rapidity_width.C
+funcName=generateLFRapidity("${O2DPG_MC_CONFIG_ROOT}/MC/config/PWGLF/pythia8/generator/resonancegun_ml_omega2012minus_xik0s.json", true, 0, false, true, "${O2DPG_MC_CONFIG_ROOT}/MC/config/PWGLF/pythia8/generator/pythia8_inel_136tev.cfg", "")
+
+[GeneratorPythia8] # if triggered then this will be used as the background event
+config=${O2DPG_MC_CONFIG_ROOT}/MC/config/PWGLF/pythia8/generator/pythia8_inel_136tev.cfg
+
+[DecayerPythia8] # after for transport code!
+config[0]=${O2DPG_MC_CONFIG_ROOT}/MC/config/common/pythia8/decayer/base.cfg
+config[1]=${O2DPG_MC_CONFIG_ROOT}/MC/config/PWGLF/pythia8/generator/resonance_ml_omega2012_xik0s.cfg

MC/config/PWGLF/ini/GeneratorLF_ResonancesML_Omega2012Plus_Xi1530K_pp1360.ini

@@ -0,0 +1,13 @@
+#---> GeneratorLF_ResonancesML_Omega2012_Xi1530K_pp1360
+# Single-channel ML signal sample: one Omega(2012)+ per pp 13.6 TeV inelastic event (no gap), flat pT 0-15 GeV/c,
+# |y| < 0.7, forced decay chain from resonance_ml_omega2012_xi1530k.cfg. One stratum (charge x channel) per job.
+[GeneratorExternal]
+fileName=${O2DPG_MC_CONFIG_ROOT}/MC/config/PWGLF/pythia8/generator_pythia8_LF_rapidity_width.C
+funcName=generateLFRapidity("${O2DPG_MC_CONFIG_ROOT}/MC/config/PWGLF/pythia8/generator/resonancegun_ml_omega2012plus_xi1530k.json", true, 0, false, true, "${O2DPG_MC_CONFIG_ROOT}/MC/config/PWGLF/pythia8/generator/pythia8_inel_136tev.cfg", "")
+
+[GeneratorPythia8] # if triggered then this will be used as the background event
+config=${O2DPG_MC_CONFIG_ROOT}/MC/config/PWGLF/pythia8/generator/pythia8_inel_136tev.cfg
+
+[DecayerPythia8] # after for transport code!
+config[0]=${O2DPG_MC_CONFIG_ROOT}/MC/config/common/pythia8/decayer/base.cfg
+config[1]=${O2DPG_MC_CONFIG_ROOT}/MC/config/PWGLF/pythia8/generator/resonance_ml_omega2012_xi1530k.cfg

MC/config/PWGLF/ini/GeneratorLF_ResonancesML_Omega2012Plus_XiK0s_pp1360.ini

@@ -0,0 +1,13 @@
+#---> GeneratorLF_ResonancesML_Omega2012_XiK0s_pp1360
+# Single-channel ML signal sample: one Omega(2012)+ per pp 13.6 TeV inelastic event (no gap), flat pT 0-15 GeV/c,
+# |y| < 0.7, forced decay chain from resonance_ml_omega2012_xik0s.cfg. One stratum (charge x channel) per job.
+[GeneratorExternal]
+fileName=${O2DPG_MC_CONFIG_ROOT}/MC/config/PWGLF/pythia8/generator_pythia8_LF_rapidity_width.C
+funcName=generateLFRapidity("${O2DPG_MC_CONFIG_ROOT}/MC/config/PWGLF/pythia8/generator/resonancegun_ml_omega2012plus_xik0s.json", true, 0, false, true, "${O2DPG_MC_CONFIG_ROOT}/MC/config/PWGLF/pythia8/generator/pythia8_inel_136tev.cfg", "")
+
+[GeneratorPythia8] # if triggered then this will be used as the background event
+config=${O2DPG_MC_CONFIG_ROOT}/MC/config/PWGLF/pythia8/generator/pythia8_inel_136tev.cfg
+
+[DecayerPythia8] # after for transport code!
+config[0]=${O2DPG_MC_CONFIG_ROOT}/MC/config/common/pythia8/decayer/base.cfg
+config[1]=${O2DPG_MC_CONFIG_ROOT}/MC/config/PWGLF/pythia8/generator/resonance_ml_omega2012_xik0s.cfg

MC/config/PWGLF/ini/tests/GeneratorLF_ResonancesML_K1_KstarPi_pp1360.C

@@ -0,0 +1,96 @@
+// Test of the single-channel ML signal samples GeneratorLF_ResonancesML_*_pp1360.ini that redirect here.
+// Every event carries exactly one injected parent (no gap); the parent and the intermediate resonances decayed by the
+// generator must follow the forced chain (charge conjugated for the negative-PDG parent). Parents with a mother come
+// from the underlying Pythia event, decay with the default table, and are only counted.
+int External()
+{
+  const std::string path{"o2sim_Kine.root"};
+  const int parentPDG{10323};
+  const std::vector<int> daughtersOfPositive{313, 211};
+  const std::map<int, std::vector<int>> intermediateDecaysOfPositive{{313, {321, -211}}};
+  const std::set<int> selfConjugate{111, 113, 310};
+  auto conjugate = [&](int pdg, int sign) { return (sign > 0 || selfConjugate.count(pdg)) ? pdg : -pdg; };
+  auto sorted = [](std::vector<int> v) { std::sort(v.begin(), v.end()); return v; };
+
+  TFile file(path.c_str(), "READ");
+  if (file.IsZombie()) {
+    std::cerr << "Cannot open ROOT file " << path << "\n";
+    return 1;
+  }
+  auto tree = (TTree*)file.Get("o2sim");
+  if (!tree) {
+    std::cerr << "Cannot find tree o2sim in file " << path << "\n";
+    return 1;
+  }
+  std::vector<o2::MCTrack>* tracks{};
+  tree->SetBranchAddress("MCTrack", &tracks);
+
+  int nEventsWrongParentCount{0}, nFromUnderlyingEvent{0}, nParents{0}, nNotDecayed{0}, nWrongDecay{0}, nIntermediates{0}, nWrongIntermediate{0};
+  std::map<int, int> nParentsBySign;
+  for (Long64_t i = 0; i < tree->GetEntries(); ++i) {
+    tree->GetEntry(i);
+    int nParentsInEvent{0};
+    for (const auto& track : *tracks) {
+      const int pdg = track.GetPdgCode();
+      if (std::abs(pdg) != parentPDG) {
+        continue;
+      }
+      if (track.getMotherTrackId() >= 0) {
+        ++nFromUnderlyingEvent;
+        continue;
+      }
+      ++nParentsInEvent;
+      ++nParents;
+      const int sign = pdg > 0 ? 1 : -1;
+      ++nParentsBySign[sign];
+      if (track.getFirstDaughterTrackId() < 0) {
+        ++nNotDecayed;
+        continue;
+      }
+      std::vector<int> expected, found;
+      for (int d : daughtersOfPositive) {
+        expected.push_back(conjugate(d, sign));
+      }
+      for (int j = track.getFirstDaughterTrackId(); j <= track.getLastDaughterTrackId(); ++j) {
+        const auto& daughter = tracks->at(j);
+        found.push_back(daughter.GetPdgCode());
+        const auto it = intermediateDecaysOfPositive.find(sign > 0 ? daughter.GetPdgCode() : conjugate(daughter.GetPdgCode(), -1));
+        if (it == intermediateDecaysOfPositive.end()) {
+          continue;
+        }
+        ++nIntermediates;
+        std::vector<int> expectedGrand, foundGrand;
+        for (int g : it->second) {
+          expectedGrand.push_back(conjugate(g, sign));
+        }
+        for (int k = daughter.getFirstDaughterTrackId(); k >= 0 && k <= daughter.getLastDaughterTrackId(); ++k) {
+          foundGrand.push_back(tracks->at(k).GetPdgCode());
+        }
+        if (sorted(expectedGrand) != sorted(foundGrand)) {
+          ++nWrongIntermediate;
+        }
+      }
+      if (sorted(expected) != sorted(found)) {
+        ++nWrongDecay;
+      }
+    }
+    if (nParentsInEvent != 1) {
+      ++nEventsWrongParentCount;
+    }
+  }
+
+  std::cout << "--------------------------------\n";
+  std::cout << "# Events: " << tree->GetEntries() << "\n";
+  std::cout << "# Parents " << parentPDG << ": " << nParentsBySign[1] << ", anti: " << nParentsBySign[-1] << "\n";
+  std::cout << "# Parents from the underlying event (not checked): " << nFromUnderlyingEvent << "\n";
+  std::cout << "# Events without exactly one parent: " << nEventsWrongParentCount << "\n";
+  std::cout << "# Parents not decayed: " << nNotDecayed << ", with wrong daughters: " << nWrongDecay << "\n";
+  std::cout << "# Intermediate resonances: " << nIntermediates << ", with wrong daughters: " << nWrongIntermediate << "\n";
+  std::cout << "--------------------------------\n";
+  if (tree->GetEntries() == 0 || nEventsWrongParentCount || nNotDecayed || nWrongDecay || nWrongIntermediate ||
+      (!intermediateDecaysOfPositive.empty() && nIntermediates != nParents)) {
+    std::cerr << "Forced decay chain not reproduced\n";
+    return 1;
+  }
+  return 0;
+}

MC/config/PWGLF/ini/tests/GeneratorLF_ResonancesML_K1_RhoK_pp1360.C

@@ -0,0 +1,96 @@
+// Test of the single-channel ML signal samples GeneratorLF_ResonancesML_*_pp1360.ini that redirect here.
+// Every event carries exactly one injected parent (no gap); the parent and the intermediate resonances decayed by the
+// generator must follow the forced chain (charge conjugated for the negative-PDG parent). Parents with a mother come
+// from the underlying Pythia event, decay with the default table, and are only counted.
+int External()
+{
+  const std::string path{"o2sim_Kine.root"};
+  const int parentPDG{10323};
+  const std::vector<int> daughtersOfPositive{321, 113};
+  const std::map<int, std::vector<int>> intermediateDecaysOfPositive{{113, {211, -211}}};
+  const std::set<int> selfConjugate{111, 113, 310};
+  auto conjugate = [&](int pdg, int sign) { return (sign > 0 || selfConjugate.count(pdg)) ? pdg : -pdg; };
+  auto sorted = [](std::vector<int> v) { std::sort(v.begin(), v.end()); return v; };
+
+  TFile file(path.c_str(), "READ");
+  if (file.IsZombie()) {
+    std::cerr << "Cannot open ROOT file " << path << "\n";
+    return 1;
+  }
+  auto tree = (TTree*)file.Get("o2sim");
+  if (!tree) {
+    std::cerr << "Cannot find tree o2sim in file " << path << "\n";
+    return 1;
+  }
+  std::vector<o2::MCTrack>* tracks{};
+  tree->SetBranchAddress("MCTrack", &tracks);
+
+  int nEventsWrongParentCount{0}, nFromUnderlyingEvent{0}, nParents{0}, nNotDecayed{0}, nWrongDecay{0}, nIntermediates{0}, nWrongIntermediate{0};
+  std::map<int, int> nParentsBySign;
+  for (Long64_t i = 0; i < tree->GetEntries(); ++i) {
+    tree->GetEntry(i);
+    int nParentsInEvent{0};
+    for (const auto& track : *tracks) {
+      const int pdg = track.GetPdgCode();
+      if (std::abs(pdg) != parentPDG) {
+        continue;
+      }
+      if (track.getMotherTrackId() >= 0) {
+        ++nFromUnderlyingEvent;
+        continue;
+      }
+      ++nParentsInEvent;
+      ++nParents;
+      const int sign = pdg > 0 ? 1 : -1;
+      ++nParentsBySign[sign];
+      if (track.getFirstDaughterTrackId() < 0) {
+        ++nNotDecayed;
+        continue;
+      }
+      std::vector<int> expected, found;
+      for (int d : daughtersOfPositive) {
+        expected.push_back(conjugate(d, sign));
+      }
+      for (int j = track.getFirstDaughterTrackId(); j <= track.getLastDaughterTrackId(); ++j) {
+        const auto& daughter = tracks->at(j);
+        found.push_back(daughter.GetPdgCode());
+        const auto it = intermediateDecaysOfPositive.find(sign > 0 ? daughter.GetPdgCode() : conjugate(daughter.GetPdgCode(), -1));
+        if (it == intermediateDecaysOfPositive.end()) {
+          continue;
+        }
+        ++nIntermediates;
+        std::vector<int> expectedGrand, foundGrand;
+        for (int g : it->second) {
+          expectedGrand.push_back(conjugate(g, sign));
+        }
+        for (int k = daughter.getFirstDaughterTrackId(); k >= 0 && k <= daughter.getLastDaughterTrackId(); ++k) {
+          foundGrand.push_back(tracks->at(k).GetPdgCode());
+        }
+        if (sorted(expectedGrand) != sorted(foundGrand)) {
+          ++nWrongIntermediate;
+        }
+      }
+      if (sorted(expected) != sorted(found)) {
+        ++nWrongDecay;
+      }
+    }
+    if (nParentsInEvent != 1) {
+      ++nEventsWrongParentCount;
+    }
+  }
+
+  std::cout << "--------------------------------\n";
+  std::cout << "# Events: " << tree->GetEntries() << "\n";
+  std::cout << "# Parents " << parentPDG << ": " << nParentsBySign[1] << ", anti: " << nParentsBySign[-1] << "\n";
+  std::cout << "# Parents from the underlying event (not checked): " << nFromUnderlyingEvent << "\n";
+  std::cout << "# Events without exactly one parent: " << nEventsWrongParentCount << "\n";
+  std::cout << "# Parents not decayed: " << nNotDecayed << ", with wrong daughters: " << nWrongDecay << "\n";
+  std::cout << "# Intermediate resonances: " << nIntermediates << ", with wrong daughters: " << nWrongIntermediate << "\n";
+  std::cout << "--------------------------------\n";
+  if (tree->GetEntries() == 0 || nEventsWrongParentCount || nNotDecayed || nWrongDecay || nWrongIntermediate ||
+      (!intermediateDecaysOfPositive.empty() && nIntermediates != nParents)) {
+    std::cerr << "Forced decay chain not reproduced\n";
+    return 1;
+  }
+  return 0;
+}