Add Non-HFE–enhanced dataset

MC/config/PWGHF/external/generator/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;
+    };
+}

MC/config/PWGHF/ini/GeneratorHF_Non_Hfe.ini

@@ -0,0 +1,14 @@
+### The external generator trigger for π⁰/η within rapidity window
+[TriggerExternal]
+# Path to your trigger C++ file
+fileName=${O2DPG_MC_CONFIG_ROOT}/MC/config/PWGHF/external/generator/selectNonHfe.C
+
+# Function name and arguments
+# Arguments: pdgPartForAccCut (TString "111;221"), rapidityMin, rapidityMax, minNb
+funcName = selectPionEtaWithinAcc("111;221",-1.5,1.5,1)
+
+### The Pythia8 generator configuration
+[GeneratorPythia8]
+config = ${O2DPG_MC_CONFIG_ROOT}/MC/config/PWGHF/pythia8/generator/pythia8_NonHfe.cfg
+includePartonEvent = true
+

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

@@ -0,0 +1,68 @@
+#include <TFile.h>
+#include <TTree.h>
+#include <vector>
+#include <iostream>
+#include <cmath>
+#include "DataFormats/MCTrack.h"
+
+int External() {
+    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