GloHelper: Allow also enable-root-input in InputHelper

Author: f3sch

Detectors/GlobalTrackingWorkflow/helpers/include/GlobalTrackingWorkflowHelpers/InputHelper.h

@@ -41,6 +41,19 @@ class InputHelper
   static int addInputSpecsStrangeTrack(const o2::framework::ConfigContext& configcontext, o2::framework::WorkflowSpec& specs, bool mc);
   static int addInputSpecsCosmics(const o2::framework::ConfigContext& configcontext, o2::framework::WorkflowSpec& specs, bool mc);
   static int addInputSpecsIRFramesITS(const o2::framework::ConfigContext& configcontext, o2::framework::WorkflowSpec& specs);
+
+ private:
+  static bool isROOTInputDisabled(const o2::framework::ConfigContext& ctx)
+  {
+    const auto& opts = ctx.options();
+    if (opts.hasOption("disable-root-input")) {
+      return opts.get<bool>("disable-root-input");
+    } else if (opts.hasOption("enable-root-input")) {
+      return !opts.get<bool>("enable-root-input");
+    } else {
+      throw "ConfigContext does not contain option to disable/enable root input for readers!";
+    }
+  }
 };
 
 } // namespace globaltracking

Detectors/GlobalTrackingWorkflow/helpers/src/InputHelper.cxx

@@ -58,7 +58,7 @@ int InputHelper::addInputSpecs(const ConfigContext& configcontext, WorkflowSpec&
                                bool useMC, GID::mask_t maskClustersMC, GID::mask_t maskTracksMC,
                                bool subSpecStrict)
 {
-  if (configcontext.options().get<bool>("disable-root-input")) {
+  if (isROOTInputDisabled(configcontext)) {
     return 0;
   }
   if (useMC && configcontext.options().get<bool>("disable-mc")) {
@@ -185,7 +185,7 @@ int InputHelper::addInputSpecs(const ConfigContext& configcontext, WorkflowSpec&
 // attach primary vertex reader
 int InputHelper::addInputSpecsPVertex(const o2::framework::ConfigContext& configcontext, o2::framework::WorkflowSpec& specs, bool mc)
 {
-  if (configcontext.options().get<bool>("disable-root-input")) {
+  if (isROOTInputDisabled(configcontext)) {
     return 0;
   }
   specs.emplace_back(o2::vertexing::getPrimaryVertexReaderSpec(mc));
@@ -195,7 +195,7 @@ int InputHelper::addInputSpecsPVertex(const o2::framework::ConfigContext& config
 // attach secondary vertex reader
 int InputHelper::addInputSpecsSVertex(const o2::framework::ConfigContext& configcontext, o2::framework::WorkflowSpec& specs)
 {
-  if (configcontext.options().get<bool>("disable-root-input")) {
+  if (isROOTInputDisabled(configcontext)) {
     return 0;
   }
   specs.emplace_back(o2::vertexing::getSecondaryVertexReaderSpec());
@@ -205,7 +205,7 @@ int InputHelper::addInputSpecsSVertex(const o2::framework::ConfigContext& config
 // attach strangeness tracking reader
 int InputHelper::addInputSpecsStrangeTrack(const o2::framework::ConfigContext& configcontext, o2::framework::WorkflowSpec& specs, bool mc)
 {
-  if (configcontext.options().get<bool>("disable-root-input")) {
+  if (isROOTInputDisabled(configcontext)) {
     return 0;
   }
   specs.emplace_back(o2::strangeness_tracking::getStrangenessTrackingReaderSpec(mc));
@@ -215,7 +215,7 @@ int InputHelper::addInputSpecsStrangeTrack(const o2::framework::ConfigContext& c
 // attach cosmic tracks reader
 int InputHelper::addInputSpecsCosmics(const o2::framework::ConfigContext& configcontext, o2::framework::WorkflowSpec& specs, bool mc)
 {
-  if (configcontext.options().get<bool>("disable-root-input")) {
+  if (isROOTInputDisabled(configcontext)) {
     return 0;
   }
   specs.emplace_back(o2::globaltracking::getTrackCosmicsReaderSpec(mc));
@@ -225,7 +225,7 @@ int InputHelper::addInputSpecsCosmics(const o2::framework::ConfigContext& config
 // attach vector of ITS reconstructed IRFrames
 int InputHelper::addInputSpecsIRFramesITS(const o2::framework::ConfigContext& configcontext, o2::framework::WorkflowSpec& specs)
 {
-  if (configcontext.options().get<bool>("disable-root-input")) {
+  if (isROOTInputDisabled(configcontext)) {
     return 0;
   }
   specs.emplace_back(o2::globaltracking::getIRFrameReaderSpec("ITS", 0, "its-irframe-reader", "o2_its_irframe.root"));