Migrate UPC gap classification to SGSelector with BC range checking

minjungkim12 · claude · minjungkim12 · commit aca23b5d868b · 2025-11-13T18:54:24.000+01:00
This commit updates the UPC (Ultra-Peripheral Collision) gap determination in taskD0 and taskDplus to use SGSelector from PWGUD/Core instead of custom gap classification logic. Key changes: - Migrate from custom GapType enum to SGSelector's TrueGap enum in utilsUpcHf.h - Update gap type classification to support 7 categories: NoGap (-1), SingleGapA (0), SingleGapC (1), DoubleGap (2), NoUpc (3), TrkOutOfRange (4), BadDoubleGap (5) - Add SGSelector instance to taskD0 and taskDplus for gap determination - Add FV0-A amplitude threshold configurable for more comprehensive gap classification - Remove unused energyThresholdZDC parameter as SGCutParHolder only supports FIT amplitudes - Add BC range checking via udhelpers::compatibleBCs for improved gap classification accuracy - Update histogram axes from 3 bins to 7 bins to accommodate new gap types - Add required library dependencies (SGCutParHolder, EventFilteringUtils) to CMakeLists.txt Technical details: - SGSelector uses SGCutParHolder to configure thresholds for FIT detectors (FV0A, FT0A, FT0C) - BC range checking examines multiple BCs around collision using NDtColl=1000ns and MinNBCs=7 - Gap determination now uses SGSelector::IsSelected() with proper BC range instead of simple threshold comparison - This aligns PWGHF UPC analysis with standard gap classification used in PWGUD 🤖 Generated with [Claude Code](https://claude.com/claude-code) Co-Authored-By: Claude <noreply@anthropic.com>
diff --git a/PWGHF/D2H/Tasks/CMakeLists.txt b/PWGHF/D2H/Tasks/CMakeLists.txt
@@ -71,7 +71,7 @@ o2physics_add_dpl_workflow(task-charm-reso-to-d-trk-reduced
 
 o2physics_add_dpl_workflow(task-d0
                     SOURCES taskD0.cxx
-                    PUBLIC_LINK_LIBRARIES O2Physics::AnalysisCore O2Physics::AnalysisCCDB
+                    PUBLIC_LINK_LIBRARIES O2Physics::AnalysisCore O2Physics::AnalysisCCDB O2Physics::SGCutParHolder O2Physics::EventFilteringUtils
                     COMPONENT_NAME Analysis)
 
 o2physics_add_dpl_workflow(task-directed-flow-charm-hadrons
@@ -81,7 +81,7 @@ o2physics_add_dpl_workflow(task-directed-flow-charm-hadrons
 
 o2physics_add_dpl_workflow(task-dplus
                     SOURCES taskDplus.cxx
-                    PUBLIC_LINK_LIBRARIES O2Physics::AnalysisCore
+                    PUBLIC_LINK_LIBRARIES O2Physics::AnalysisCore O2Physics::AnalysisCCDB O2Physics::SGCutParHolder O2Physics::EventFilteringUtils
                     COMPONENT_NAME Analysis)
 
 o2physics_add_dpl_workflow(task-ds
diff --git a/PWGHF/D2H/Tasks/taskD0.cxx b/PWGHF/D2H/Tasks/taskD0.cxx
@@ -97,11 +97,13 @@ struct HfTaskD0 {
   Configurable<std::string> irSource{"irSource", "ZNC hadronic", "Estimator of the interaction rate (Recommended: pp --> T0VTX, Pb-Pb --> ZNC hadronic)"};
 
   // UPC gap determination thresholds
+  Configurable<float> upcFV0AThreshold{"upcFV0AThreshold", 100.0f, "FV0-A amplitude threshold for UPC gap determination (a.u.)"};
   Configurable<float> upcFT0AThreshold{"upcFT0AThreshold", 100.0f, "FT0-A amplitude threshold for UPC gap determination (a.u.)"};
   Configurable<float> upcFT0CThreshold{"upcFT0CThreshold", 50.0f, "FT0-C amplitude threshold for UPC gap determination (a.u.)"};
   Configurable<float> upcZDCThreshold{"upcZDCThreshold", 1.0f, "ZDC energy threshold for UPC gap determination (a.u.)"};
 
   HfEventSelection hfEvSel; // event selection and monitoring
+  SGSelector sgSelector;    // UPC gap selector
 
   ctpRateFetcher mRateFetcher;
 
@@ -158,7 +160,7 @@ struct HfTaskD0 {
   ConfigurableAxis thnConfigAxisMinItsNCls{"thnConfigAxisMinItsNCls", {5, 3, 8}, "axis for minimum ITS NCls of candidate prongs"};
   ConfigurableAxis thnConfigAxisMinTpcNCrossedRows{"thnConfigAxisMinTpcNCrossedRows", {10, 70, 180}, "axis for minimum TPC NCls crossed rows of candidate prongs"};
   ConfigurableAxis thnConfigAxisIR{"thnConfigAxisIR", {5000, 0, 500}, "Interaction rate (kHz)"};
-  ConfigurableAxis thnConfigAxisGapType{"thnConfigAxisGapType", {3, -0.5, 2.5}, "axis for UPC gap type (0=GapA, 1=GapC, 2=DoubleGap)"};
+  ConfigurableAxis thnConfigAxisGapType{"thnConfigAxisGapType", {7, -1.5, 5.5}, "axis for UPC gap type (-1=NoGap, 0=SingleGapA, 1=SingleGapC, 2=DoubleGap, 3=NoUpc, 4=TrkOutOfRange, 5=BadDoubleGap)"};
   ConfigurableAxis thnConfigAxisFT0A{"thnConfigAxisFT0A", {1001, -1.5, 999.5}, "axis for FT0-A amplitude (a.u.)"};
   ConfigurableAxis thnConfigAxisFT0C{"thnConfigAxisFT0C", {1001, -1.5, 999.5}, "axis for FT0-C amplitude (a.u.)"};
 
@@ -371,10 +373,7 @@ struct HfTaskD0 {
 
     registry.add("Data/fitInfo/ampFT0A_vs_ampFT0C", "FT0-A vs FT0-C amplitude;FT0-A amplitude (a.u.);FT0-C amplitude (a.u.)", {HistType::kTH2F, {{2500, 0., 250}, {2500, 0., 250}}});
     registry.add("Data/zdc/energyZNA_vs_energyZNC", "ZNA vs ZNC common energy;E_{ZNA}^{common} (a.u.);E_{ZNC}^{common} (a.u.)", {HistType::kTH2F, {{200, 0., 20}, {200, 0., 20}}});
-    registry.add("Data/hUpcGapAfterSelection", "UPC gap type after selection;Gap side;Counts", {HistType::kTH1F, {{3, -0.5, 2.5}}});
-    registry.get<TH1>(HIST("Data/hUpcGapAfterSelection"))->GetXaxis()->SetBinLabel(static_cast<int>(GapType::GapA) + 1, "A");
-    registry.get<TH1>(HIST("Data/hUpcGapAfterSelection"))->GetXaxis()->SetBinLabel(static_cast<int>(GapType::GapC) + 1, "C");
-    registry.get<TH1>(HIST("Data/hUpcGapAfterSelection"))->GetXaxis()->SetBinLabel(static_cast<int>(GapType::DoubleGap) + 1, "Double");
+    registry.add("Data/hUpcGapAfterSelection", "UPC gap type after selection;Gap type;Counts", {HistType::kTH1F, {{7, -1.5, 5.5}}});
 
     hfEvSel.addHistograms(registry);
 
@@ -581,15 +580,17 @@ struct HfTaskD0 {
       upchelpers::FITInfo fitInfo{};
       udhelpers::getFITinfo(fitInfo, bc, bcs, ft0s, fv0as, fdds);
 
-      GapType gap = GapType::DoubleGap;
+      // Determine gap type using SGSelector with BC range checking
+      int gap = hf_upc::determineGapType(collision, bcs, sgSelector,
+                                         upcFV0AThreshold, upcFT0AThreshold, upcFT0CThreshold);
+
       if (bc.has_zdc()) {
         const auto& zdc = bc.zdc();
         registry.fill(HIST("Data/fitInfo/ampFT0A_vs_ampFT0C"), fitInfo.ampFT0A, fitInfo.ampFT0C);
         registry.fill(HIST("Data/zdc/energyZNA_vs_energyZNC"), zdc.energyCommonZNA(), zdc.energyCommonZNC());
-        gap = hf_upc::determineGapType(fitInfo.ampFT0A, fitInfo.ampFT0C, zdc.energyCommonZNA(), zdc.energyCommonZNC(),
-                                       upcFT0AThreshold, upcFT0CThreshold, upcZDCThreshold);
         registry.fill(HIST("Data/hUpcGapAfterSelection"), hf_upc::gapTypeToInt(gap));
       }
+
       if (hf_upc::isSingleSidedGap(gap)) {
         int const gapTypeInt = hf_upc::gapTypeToInt(gap);
         const auto thisCollId = collision.globalIndex();
@@ -626,11 +627,11 @@ struct HfTaskD0 {
           // Fill THnSparse with structure matching histogram axes: [mass, pt, (mlScores if FillMl), rapidity, d0Type, (cent if storeCentrality), (occ, ir if storeOccupancyAndIR), gapType, FT0A, FT0C]
           auto fillTHnData = [&](float mass, int d0Type) {
             // Pre-calculate vector size to avoid reallocations
-            constexpr int nAxesBase = 7;                        // mass, pt, rapidity, d0Type, gapType, FT0A, FT0C
-            constexpr int nAxesMl = FillMl ? 3 : 0;             // 3 ML scores if FillMl
+            constexpr int NAxesBase = 7;                        // mass, pt, rapidity, d0Type, gapType, FT0A, FT0C
+            constexpr int NAxesMl = FillMl ? 3 : 0;             // 3 ML scores if FillMl
             int const nAxesCent = storeCentrality ? 1 : 0;      // centrality if storeCentrality
             int const nAxesOccIR = storeOccupancyAndIR ? 2 : 0; // occupancy and IR if storeOccupancyAndIR
-            int const nAxesTotal = nAxesBase + nAxesMl + nAxesCent + nAxesOccIR;
+            int const nAxesTotal = NAxesBase + NAxesMl + nAxesCent + nAxesOccIR;
 
             std::vector<double> valuesToFill;
             valuesToFill.reserve(nAxesTotal);
diff --git a/PWGHF/D2H/Tasks/taskDplus.cxx b/PWGHF/D2H/Tasks/taskDplus.cxx
@@ -22,6 +22,7 @@
 #include "PWGHF/Core/DecayChannels.h"
 #include "PWGHF/Core/HfHelper.h"
 #include "PWGHF/Core/SelectorCuts.h"
+#include "PWGHF/DataModel/AliasTables.h"
 #include "PWGHF/DataModel/CandidateReconstructionTables.h"
 #include "PWGHF/DataModel/CandidateSelectionTables.h"
 #include "PWGHF/DataModel/TrackIndexSkimmingTables.h"
@@ -86,11 +87,13 @@ struct HfTaskDplus {
   Configurable<std::string> irSource{"irSource", "ZNC hadronic", "Estimator of the interaction rate (Recommended: pp --> T0VTX, Pb-Pb --> ZNC hadronic)"};
 
   // UPC gap determination thresholds
+  Configurable<float> upcFV0AThreshold{"upcFV0AThreshold", 100.0f, "FV0-A amplitude threshold for UPC gap determination (a.u.)"};
   Configurable<float> upcFT0AThreshold{"upcFT0AThreshold", 100.0f, "FT0-A amplitude threshold for UPC gap determination (a.u.)"};
   Configurable<float> upcFT0CThreshold{"upcFT0CThreshold", 50.0f, "FT0-C amplitude threshold for UPC gap determination (a.u.)"};
   Configurable<float> upcZDCThreshold{"upcZDCThreshold", 1.0f, "ZDC energy threshold for UPC gap determination (a.u.)"};
 
   HfEventSelection hfEvSel;    // event selection and monitoring
+  SGSelector sgSelector;       // UPC gap selector
   ctpRateFetcher mRateFetcher; // interaction rate fetcher
 
   Service<o2::ccdb::BasicCCDBManager> ccdb;
@@ -132,7 +135,7 @@ struct HfTaskDplus {
   ConfigurableAxis thnConfigAxisMlScore0{"thnConfigAxisMlScore0", {100, 0., 1.}, "axis for ML output score 0"};
   ConfigurableAxis thnConfigAxisMlScore1{"thnConfigAxisMlScore1", {100, 0., 1.}, "axis for ML output score 1"};
   ConfigurableAxis thnConfigAxisMlScore2{"thnConfigAxisMlScore2", {100, 0., 1.}, "axis for ML output score 2"};
-  ConfigurableAxis thnConfigAxisGapType{"thnConfigAxisGapType", {3, -0.5, 2.5}, "axis for UPC gap type (0=GapA, 1=GapC, 2=DoubleGap)"};
+  ConfigurableAxis thnConfigAxisGapType{"thnConfigAxisGapType", {7, -1.5, 5.5}, "axis for UPC gap type (-1=NoGap, 0=SingleGapA, 1=SingleGapC, 2=DoubleGap, 3=NoUpc, 4=TrkOutOfRange, 5=BadDoubleGap)"};
   ConfigurableAxis thnConfigAxisFT0A{"thnConfigAxisFT0A", {1001, -1.5, 999.5}, "axis for FT0-A amplitude (a.u.)"};
   ConfigurableAxis thnConfigAxisFT0C{"thnConfigAxisFT0C", {1001, -1.5, 999.5}, "axis for FT0-C amplitude (a.u.)"};
 
@@ -280,10 +283,7 @@ struct HfTaskDplus {
 
     registry.add("Data/fitInfo/ampFT0A_vs_ampFT0C", "FT0-A vs FT0-C amplitude;FT0-A amplitude (a.u.);FT0-C amplitude (a.u.)", {HistType::kTH2F, {{2500, 0., 250}, {2500, 0., 250}}});
     registry.add("Data/zdc/energyZNA_vs_energyZNC", "ZNA vs ZNC common energy;E_{ZNA}^{common} (a.u.);E_{ZNC}^{common} (a.u.)", {HistType::kTH2F, {{200, 0., 20}, {200, 0., 20}}});
-    registry.add("Data/hUpcGapAfterSelection", "UPC gap type after selection;Gap side;Counts", {HistType::kTH1F, {{3, -0.5, 2.5}}});
-    registry.get<TH1>(HIST("Data/hUpcGapAfterSelection"))->GetXaxis()->SetBinLabel(static_cast<int>(GapType::GapA) + 1, "A");
-    registry.get<TH1>(HIST("Data/hUpcGapAfterSelection"))->GetXaxis()->SetBinLabel(static_cast<int>(GapType::GapC) + 1, "C");
-    registry.get<TH1>(HIST("Data/hUpcGapAfterSelection"))->GetXaxis()->SetBinLabel(static_cast<int>(GapType::DoubleGap) + 1, "Double");
+    registry.add("Data/hUpcGapAfterSelection", "UPC gap type after selection;Gap type;Counts", {HistType::kTH1F, {{7, -1.5, 5.5}}});
 
     hfEvSel.addHistograms(registry); // collision monitoring
 
@@ -716,13 +716,14 @@ struct HfTaskDplus {
       upchelpers::FITInfo fitInfo{};
       udhelpers::getFITinfo(fitInfo, bc, bcs, ft0s, fv0as, fdds);
 
-      GapType gap = GapType::DoubleGap;
+      // Determine gap type using SGSelector with BC range checking
+      int gap = hf_upc::determineGapType(collision, bcs, sgSelector,
+                                         upcFV0AThreshold, upcFT0AThreshold, upcFT0CThreshold);
+
       if (bc.has_zdc()) {
         const auto& zdc = bc.zdc();
         registry.fill(HIST("Data/fitInfo/ampFT0A_vs_ampFT0C"), fitInfo.ampFT0A, fitInfo.ampFT0C);
         registry.fill(HIST("Data/zdc/energyZNA_vs_energyZNC"), zdc.energyCommonZNA(), zdc.energyCommonZNC());
-        gap = hf_upc::determineGapType(fitInfo.ampFT0A, fitInfo.ampFT0C, zdc.energyCommonZNA(), zdc.energyCommonZNC(),
-                                       upcFT0AThreshold, upcFT0CThreshold, upcZDCThreshold);
         registry.fill(HIST("Data/hUpcGapAfterSelection"), hf_upc::gapTypeToInt(gap));
       }
       if (hf_upc::isSingleSidedGap(gap)) {
@@ -744,12 +745,12 @@ struct HfTaskDplus {
         // Lambda function to fill THn - handles both ML and non-ML cases
         auto fillTHnData = [&](const auto& candidate) {
           // Pre-calculate vector size to avoid reallocations
-          constexpr int nAxesBase = 5;                   // mass, pt, gapType, FT0A, FT0C
-          constexpr int nAxesMl = FillMl ? 3 : 0;        // 3 ML scores if FillMl
+          constexpr int NAxesBase = 5;                   // mass, pt, gapType, FT0A, FT0C
+          constexpr int NAxesMl = FillMl ? 3 : 0;        // 3 ML scores if FillMl
           int const nAxesCent = storeCentrality ? 1 : 0; // centrality if storeCentrality
           int const nAxesOcc = storeOccupancy ? 1 : 0;   // occupancy if storeOccupancy
           int const nAxesIR = storeIR ? 1 : 0;           // IR if storeIR
-          int const nAxesTotal = nAxesBase + nAxesMl + nAxesCent + nAxesOcc + nAxesIR;
+          int const nAxesTotal = NAxesBase + NAxesMl + nAxesCent + nAxesOcc + nAxesIR;
 
           std::vector<double> valuesToFill;
           valuesToFill.reserve(nAxesTotal);
diff --git a/PWGHF/Utils/utilsUpcHf.h b/PWGHF/Utils/utilsUpcHf.h
@@ -17,87 +17,108 @@
 #ifndef PWGHF_UTILS_UTILSUPCHF_H_
 #define PWGHF_UTILS_UTILSUPCHF_H_
 
-#include <cstdint>
+#include "PWGUD/Core/SGCutParHolder.h"
+#include "PWGUD/Core/SGSelector.h"
+#include "PWGUD/Core/UDHelpers.h"
 
 namespace o2::analysis::hf_upc
 {
 
-/// \brief Gap type classification for UPC events
-enum class GapType : uint8_t {
-  GapA = 0,     ///< Gap on A-side (C-side active)
-  GapC = 1,     ///< Gap on C-side (A-side active)
-  DoubleGap = 2 ///< Double gap (both sides empty)
-};
+/// \brief Use TrueGap enum from SGSelector for gap type classification
+using o2::aod::sgselector::TrueGap;
 
 /// \brief Default thresholds for gap determination
 namespace defaults
 {
+constexpr float AmplitudeThresholdFV0A = 100.0f; ///< Amplitude threshold for FV0-A (a.u.)
 constexpr float AmplitudeThresholdFT0A = 100.0f; ///< Amplitude threshold for FT0-A (a.u.)
 constexpr float AmplitudeThresholdFT0C = 50.0f;  ///< Amplitude threshold for FT0-C (a.u.)
-constexpr float EnergyThresholdZDC = 1.0f;       ///< Energy threshold for ZDC (a.u.)
+constexpr float MaxFITTime = 4.0f;               ///< Maximum FIT time (ns)
+constexpr int NDtColl = 1000;                    ///< Time window for BC range (ns)
+constexpr int MinNBCs = 7;                       ///< Minimum number of BCs to check
+constexpr int MinNTracks = 0;                    ///< Minimum number of tracks
+constexpr int MaxNTracks = 100;                  ///< Maximum number of tracks
 } // namespace defaults
 
-/// \brief Determine gap type based on FIT and ZDC signals
-/// \param ft0A FT0-A amplitude
-/// \param ft0C FT0-C amplitude
-/// \param zdcA ZDC-A (ZNA) common energy
-/// \param zdcC ZDC-C (ZNC) common energy
+/// \brief Determine gap type using SGSelector with BC range checking
+/// \tparam TCollision Collision type
+/// \tparam TBCs BC table type
+/// \param collision Collision object
+/// \param bcs BC table
+/// \param sgSelector SGSelector instance
+/// \param amplitudeThresholdFV0A Threshold for FV0-A (default: 100.0)
 /// \param amplitudeThresholdFT0A Threshold for FT0-A (default: 100.0)
 /// \param amplitudeThresholdFT0C Threshold for FT0-C (default: 50.0)
-/// \param energyThresholdZDC Threshold for ZDC (default: 1.0)
-/// \return Gap type classification
-inline GapType determineGapType(float ft0A, float ft0C, float zdcA, float zdcC,
-                                float amplitudeThresholdFT0A = defaults::AmplitudeThresholdFT0A,
-                                float amplitudeThresholdFT0C = defaults::AmplitudeThresholdFT0C,
-                                float energyThresholdZDC = defaults::EnergyThresholdZDC)
+/// \return TrueGap enum value (-1=NoGap, 0=SingleGapA, 1=SingleGapC, 2=DoubleGap, 3=NoUpc, 4=TrkOutOfRange, 5=BadDoubleGap)
+template <typename TCollision, typename TBCs>
+inline int determineGapType(TCollision const& collision,
+                            TBCs const& bcs,
+                            SGSelector& sgSelector,
+                            float amplitudeThresholdFV0A = defaults::AmplitudeThresholdFV0A,
+                            float amplitudeThresholdFT0A = defaults::AmplitudeThresholdFT0A,
+                            float amplitudeThresholdFT0C = defaults::AmplitudeThresholdFT0C)
 {
-  // Gap on A-side: FT0-A empty, FT0-C active, ZNA empty, ZNC active
-  if (ft0A < amplitudeThresholdFT0A && ft0C > amplitudeThresholdFT0C &&
-      zdcA < energyThresholdZDC && zdcC > energyThresholdZDC) {
-    return GapType::GapA;
-  }
+  // Configure SGSelector thresholds
+  SGCutParHolder sgCuts;
+  sgCuts.SetNDtcoll(defaults::NDtColl);
+  sgCuts.SetMinNBCs(defaults::MinNBCs);
+  sgCuts.SetNTracks(defaults::MinNTracks, defaults::MaxNTracks);
+  sgCuts.SetMaxFITtime(defaults::MaxFITTime);
+  sgCuts.SetFITAmpLimits({amplitudeThresholdFV0A, amplitudeThresholdFT0A, amplitudeThresholdFT0C});
 
-  // Gap on C-side: FT0-A active, FT0-C empty, ZNA active, ZNC empty
-  if (ft0A > amplitudeThresholdFT0A && ft0C < amplitudeThresholdFT0C &&
-      zdcA > energyThresholdZDC && zdcC < energyThresholdZDC) {
-    return GapType::GapC;
+  // Get BC and BC range
+  if (!collision.has_foundBC()) {
+    return TrueGap::NoGap;
   }
 
-  // Default: Double gap (or no clear gap)
-  return GapType::DoubleGap;
+  const auto bc = collision.template foundBC_as<TBCs>();
+  const auto bcRange = udhelpers::compatibleBCs(collision, sgCuts.NDtcoll(), bcs, sgCuts.minNBCs());
+
+  // Use SGSelector to determine gap type with BC range checking
+  const auto sgResult = sgSelector.IsSelected(sgCuts, collision, bcRange, bc);
+
+  return sgResult.value;
 }
 
-/// \brief Check if the gap type is a single-sided gap (GapA or GapC)
-/// \param gap Gap type
+/// \brief Check if the gap type is a single-sided gap (SingleGapA or SingleGapC)
+/// \param gap TrueGap enum value
 /// \return true if single-sided gap, false otherwise
-inline bool isSingleSidedGap(GapType gap)
+inline bool isSingleSidedGap(int gap)
 {
-  return (gap == GapType::GapA || gap == GapType::GapC);
+  return (gap == TrueGap::SingleGapA || gap == TrueGap::SingleGapC);
 }
 
 /// \brief Get gap type name as string
-/// \param gap Gap type
+/// \param gap TrueGap enum value
 /// \return String representation of gap type
-inline const char* getGapTypeName(GapType gap)
+inline const char* getGapTypeName(int gap)
 {
   switch (gap) {
-    case GapType::GapA:
-      return "GapA";
-    case GapType::GapC:
-      return "GapC";
-    case GapType::DoubleGap:
+    case TrueGap::NoGap:
+      return "NoGap";
+    case TrueGap::SingleGapA:
+      return "SingleGapA";
+    case TrueGap::SingleGapC:
+      return "SingleGapC";
+    case TrueGap::DoubleGap:
       return "DoubleGap";
+    case TrueGap::NoUpc:
+      return "NoUpc";
+    case TrueGap::TrkOutOfRange:
+      return "TrkOutOfRange";
+    case TrueGap::BadDoubleGap:
+      return "BadDoubleGap";
     default:
       return "Unknown";
   }
 }
 
 /// \brief Convert gap type to integer for histogram filling
-/// \param gap Gap type
-/// \return Integer representation (0=GapA, 1=GapC, 2=DoubleGap)
-inline int gapTypeToInt(GapType gap)
+/// \param gap TrueGap enum value
+/// \return Integer representation (-1, 0, 1, 2, 3, 4, 5)
+inline int gapTypeToInt(int gap)
 {
-  return static_cast<int>(gap);
+  return gap;
 }
 
 } // namespace o2::analysis::hf_upc
