AOD: switch to flag class for streamer steering

Author: f3sch

Detectors/AOD/include/AODProducerWorkflow/AODProducerWorkflowSpec.h

@@ -30,7 +30,7 @@
 #include "ZDCBase/Constants.h"
 #include "GlobalTracking/MatchGlobalFwd.h"
 #include "CommonUtils/TreeStreamRedirector.h"
-#include "CommonUtils/EnumBitOperators.h"
+#include "CommonUtils/Flags.h"
 
 #include <cstdint>
 #include <limits>
@@ -209,11 +209,8 @@ class BunchCrossings
 
 // Steering bits for additional output during AOD production
 enum struct AODProducerStreamerMask : uint8_t {
-  None = 0,
-  TrackQA = O2_ENUM_SET_BIT(0),
-  All = std::numeric_limits<std::underlying_type_t<AODProducerStreamerMask>>::max(),
+  TrackQA,
 };
-O2_DEFINE_ENUM_BIT_OPERATORS(AODProducerStreamerMask)
 
 class AODProducerWorkflowDPL : public Task
 {
@@ -251,7 +248,7 @@ class AODProducerWorkflowDPL : public Task
   std::unordered_set<GIndex> mGIDUsedBySVtx;
   std::unordered_set<GIndex> mGIDUsedByStr;
 
-  AODProducerStreamerMask mStreamerMask{0};
+  o2::flags::Flags<AODProducerStreamerMask> mStreamerFlags;
   std::shared_ptr<o2::utils::TreeStreamRedirector> mStreamer;
 
   int mNThreads = 1;

Detectors/AOD/src/AODProducerWorkflowSpec.cxx

@@ -1673,9 +1673,9 @@ void AODProducerWorkflowDPL::init(InitContext& ic)
   mPropTracks = ic.options().get<bool>("propagate-tracks");
   mPropMuons = ic.options().get<bool>("propagate-muons");
   if (auto s = ic.options().get<std::string>("with-streamers"); !s.empty()) {
-    mStreamerMask = static_cast<AODProducerStreamerMask>(std::stoul(s, nullptr, 2));
-    if (O2_ENUM_ANY_BIT(mStreamerMask)) {
-      LOGP(info, "Writing streamer data with mask {:0{}b}", static_cast<std::underlying_type_t<AODProducerStreamerMask>>(mStreamerMask), std::numeric_limits<std::underlying_type_t<AODProducerStreamerMask>>::digits);
+    mStreamerFlags.set(s);
+    if (mStreamerFlags) {
+      LOGP(info, "Writing streamer data with mask {}", mStreamerFlags.string());
     } else {
       LOGP(warn, "Specified non-default empty streamer mask!");
     }
@@ -1766,7 +1766,7 @@ void AODProducerWorkflowDPL::init(InitContext& ic)
 
   mTimer.Reset();
 
-  if (O2_ENUM_ANY_BIT(mStreamerMask)) {
+  if (mStreamerFlags) {
     mStreamer = std::make_unique<o2::utils::TreeStreamRedirector>("AO2DStreamer.root", "RECREATE");
   }
 }
@@ -2642,7 +2642,7 @@ AODProducerWorkflowDPL::TrackQA AODProducerWorkflowDPL::processBarrelTrackQA(int
         trackQAHolder.dRefGloTgl = safeInt8Clamp(((itsCopy.getTgl() + tpcCopy.getTgl()) * 0.5f - gloCopy.getTgl()) * scaleGlo(3));
         trackQAHolder.dRefGloQ2Pt = safeInt8Clamp(((itsCopy.getQ2Pt() + tpcCopy.getQ2Pt()) * 0.5f - gloCopy.getQ2Pt()) * scaleGlo(4));
 
-        if (O2_ENUM_TEST_BIT(mStreamerMask, AODProducerStreamerMask::TrackQA)) {
+        if (mStreamerFlags.test(AODProducerStreamerMask::TrackQA)) {
           (*mStreamer) << "trackQA"
                        << "trackITSOrig=" << itsOrig
                        << "trackTPCOrig=" << tpcOrig