From 3c4da09b2fa589e12f8589ba7d850bfac7e7259f Mon Sep 17 00:00:00 2001
From: Sasha Bylinkin <37345380+abylinkin@users.noreply.github.com>
Date: Mon, 14 Oct 2024 12:46:59 +0200
Subject: [PATCH] Update DGBCCandProducer.cxx

---
 PWGUD/TableProducer/DGBCCandProducer.cxx | 10 ++++------
 1 file changed, 4 insertions(+), 6 deletions(-)

diff --git a/PWGUD/TableProducer/DGBCCandProducer.cxx b/PWGUD/TableProducer/DGBCCandProducer.cxx
index 36d9c4449fe..5ff4f9c4ec7 100644
--- a/PWGUD/TableProducer/DGBCCandProducer.cxx
+++ b/PWGUD/TableProducer/DGBCCandProducer.cxx
@@ -17,6 +17,7 @@
 #include <algorithm>
 #include "Framework/runDataProcessing.h"
 #include "Framework/AnalysisTask.h"
+#include "ReconstructionDataFormats/Vertex.h"
 #include "PWGUD/DataModel/UDTables.h"
 #include "PWGUD/Core/UDHelpers.h"
 #include "PWGUD/Core/UPCHelpers.h"
@@ -417,8 +418,7 @@ struct DGBCCandProducer {
           nCharge = udhelpers::netCharge<true>(colTracks);
           int upc_flag = 0;
           ushort flags = col.flags();
-          const ushort UPCModeMask = 0x1 << 1;
-          if (flags & UPCModeMask)
+          if (flags & Vertex::UPCMode)
             upc_flag = 1;
           updateUDTables(false, col.globalIndex(), bc.globalBC(), bc.runNumber(), col.posX(), col.posY(), col.posZ(), upc_flag,
                          col.numContrib(), nCharge, rtrwTOF, colTracks, fitInfo);
@@ -618,8 +618,7 @@ struct DGBCCandProducer {
             udhelpers::getFITinfo(fitInfo, bc, bcs, ft0s, fv0as, fdds);
             int upc_flag = 0;
             ushort flags = col.flags();
-            const ushort UPCModeMask = 0x1 << 1;
-            if (flags & UPCModeMask)
+            if (flags & Vertex::UPCMode)
               upc_flag = 1;
             updateUDTables(false, col.globalIndex(), bcnum, bc.runNumber(), col.posX(), col.posY(), col.posZ(), upc_flag,
                            col.numContrib(), nCharge, rtrwTOF, colTracks, fitInfo);
@@ -692,8 +691,7 @@ struct DGBCCandProducer {
             int64_t colID = withCollision ? col.globalIndex() : -1;
             int upc_flag = 0;
             ushort flags = col.flags();
-            const ushort UPCModeMask = 0x1 << 1;
-            if (flags & UPCModeMask)
+            if (flags & Vertex::UPCMode)
               upc_flag = 1;
             updateUDTables(false, colID, bcnum, tibc.runNumber(), vpos[0], vpos[1], vpos[2], upc_flag,
                            tracksArray.size(), nCharge, rtrwTOF, tracksArray, fitInfo);