Update DGBCCandProducer.cxx

AliceO2Group · Oct 14, 2024 · 3c4da09 · 3c4da09
1 parent e9caf35
commit 3c4da09
Showing 1 changed file with 4 additions and 6 deletions.
diff --git 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);