Add Non-HFE–enhanced dataset configuration (#2189)

* Create the Non-HFE–enhanced dataset.

We will produce a Non-HFE–enhanced dataset by generating Monte Carlo samples with increased π⁰ and η production, to improve the statistical precision of the non-HFE efficiency measurement.

Author: rashigupt

MC/config/PWGHF/hybrid/GeneratorHF_Non_Hfe.json

@@ -0,0 +1,27 @@
+{
+  "generators": [
+    {
+      "name": "pythia8",
+      "config": {
+        "config": "${O2DPG_MC_CONFIG_ROOT}/MC/config/PWGHF/pythia8/generator/pythia8_NonHfe.cfg",
+        "hooksFileName": "",
+        "hooksFuncName": "",
+        "includePartonEvent": true,
+        "particleFilter": "",
+        "verbose": 0
+      },
+      "triggers": {
+        "mode": "or",
+        "specs": [
+          {
+            "macro": "${O2DPG_MC_CONFIG_ROOT}/MC/config/PWGHF/trigger/selectNonHfe.C",
+            "function": "selectPionEtaWithinAcc(\"111;221\",-1.5,1.5,1)"
+          }
+        ]
+      }
+    }
+  ],
+  "fractions": [
+    1
+  ]
+}

MC/config/PWGHF/ini/GeneratorHF_Non_Hfe.ini

@@ -0,0 +1,4 @@
+#### This configuration uses the Hybrid external generator to trigger π⁰/η production within the specified rapidity window
+
+[GeneratorHybrid]
+configFile = ${O2DPG_MC_CONFIG_ROOT}/MC/config/PWGHF/hybrid/GeneratorHF_Non_Hfe.json

MC/config/PWGHF/ini/tests/GeneratorHF_Non_Hfe.C

@@ -0,0 +1,61 @@
+int Hybrid() {
+    std::string path{"o2sim_Kine.root"};
+
+    const int pdgPi0 = 111;
+    const int pdgEta = 221;
+    const double yMin = -1.5;
+    const double yMax = 1.5;
+    const int minNb = 1;
+
+    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\n";
+        return 1;
+    }
+
+    std::vector<o2::MCTrack>* tracks{};
+    tree->SetBranchAddress("MCTrack", &tracks);
+
+    int nEvents = tree->GetEntries();
+    int nAccepted = 0;
+    int totalPi0 = 0, totalEta = 0;
+
+    for (int i = 0; i < nEvents; ++i) {
+        tree->GetEntry(i);
+
+        int count = 0;
+        for (auto& track : *tracks) {
+            int pdg = std::abs(track.GetPdgCode());
+            double y = track.GetRapidity();
+
+            if ((pdg == pdgPi0 || pdg == pdgEta) && y >= yMin && y <= yMax) {
+                count++;
+                if (pdg == pdgPi0) totalPi0++;
+                if (pdg == pdgEta) totalEta++;
+            }
+        }
+
+        if (count < minNb) {
+            std::cerr << " Trigger violation in event " << i
+                      << " (found " << count << " π0/η in rapidity window)\n";
+            return 1;
+        }
+
+        nAccepted++;
+    }
+
+    std::cout << "--------------------------------------\n";
+    std::cout << "Trigger test: π0/η within rapidity window\n";
+    std::cout << "Events tested: " << nEvents << "\n";
+    std::cout << "Events accepted: " << nAccepted << "\n";
+    std::cout << "# π0: " << totalPi0 << ", # η: " << totalEta << "\n";
+    std::cout << "Trigger test PASSED\n";
+
+    return 0;
+}

MC/config/PWGHF/pythia8/generator/pythia8_NonHfe.cfg

@@ -0,0 +1,41 @@
+### authors: Rashi Gupta (rashi.gupta@cern.ch)
+### authors: Ravindra Singh (ravindra.singh@cern.ch)
+### electrons from pi0 and eta enhancement
+
+### beams
+Beams:idA 2212			# proton
+Beams:idB 2212 			# proton
+Beams:eCM 13600. 		# GeV
+
+### processes
+SoftQCD:inelastic on		# all inelastic processes
+
+### decays
+ParticleDecays:limitTau0 on	
+ParticleDecays:tau0Max 10.
+
+### switching on Pythia Mode2
+ColourReconnection:mode 1
+ColourReconnection:allowDoubleJunRem off
+ColourReconnection:m0 0.3
+ColourReconnection:allowJunctions on
+ColourReconnection:junctionCorrection 1.20
+ColourReconnection:timeDilationMode 2
+ColourReconnection:timeDilationPar 0.18
+StringPT:sigma 0.335
+StringZ:aLund 0.36
+StringZ:bLund 0.56
+StringFlav:probQQtoQ 0.078
+StringFlav:ProbStoUD 0.2
+StringFlav:probQQ1toQQ0join 0.0275,0.0275,0.0275,0.0275
+MultiPartonInteractions:pT0Ref 2.15
+BeamRemnants:remnantMode 1
+BeamRemnants:saturation 5
+
+
+### switch off all decay channels
+111:onMode = off
+221:onMode = off
+
+111:onIfAny = 11      # pi0 -> e+ e- gamma
+221:onIfAny = 11      # eta -> e+ e- gamma

MC/config/PWGHF/trigger/selectNonHfe.C

@@ -0,0 +1,41 @@
+#include <TParticle.h>
+#include "Generators/Trigger.h"
+#include <vector>
+#include <TMath.h>
+
+///============================================================================
+
+/// Select π⁰ and η within a given rapidity window for enhancement
+/// pdgPartForAccCut: PDG of the particle to select (111=π⁰, 221=η)
+/// minNb: minimum number of such particles per event for enhancement
+
+////  authors: Rashi Gupta (rashi.gupta@cern.ch)
+///  authors: Ravindra Singh (ravindra.singh@cern.ch)
+/// ============================================================================
+o2::eventgen::Trigger selectPionEtaWithinAcc(TString pdgPartForAccCut = "111;221", double rapidityMin = -1.5, double rapidityMax = 1.5, int minNb = 1)
+{
+    return [pdgPartForAccCut, rapidityMin, rapidityMax, minNb](const std::vector<TParticle>& particles) -> bool {
+    TObjArray* obj = pdgPartForAccCut.Tokenize(";");
+    int count = 0;
+    for (const auto& particle : particles) {
+      int pdg = TMath::Abs(particle.GetPdgCode());
+      double y = particle.Y();
+
+      if (y < rapidityMin || y > rapidityMax) continue;
+
+      for (int i = 0; i < obj->GetEntriesFast(); ++i) {
+        int pdgCode = std::stoi(obj->At(i)->GetName());
+
+        if (pdg == pdgCode) {
+          count++;
+          break;
+        }
+      }
+    }
+        // Only accept events with at least minNb π⁰/η
+        if (count >= minNb)
+            return kTRUE;
+        else
+            return kFALSE;
+    };
+}