TRD noise calibration can run as dummy without CCDB upload

martenole · martenole · commit ebffba935b5d · 2023-06-28T13:20:12.000+02:00
diff --git a/Detectors/TRD/workflow/include/TRDWorkflow/NoiseCalibSpec.h b/Detectors/TRD/workflow/include/TRDWorkflow/NoiseCalibSpec.h
@@ -33,7 +33,7 @@ namespace trd
 class TRDNoiseCalibSpec : public o2::framework::Task
 {
  public:
-  TRDNoiseCalibSpec() = default;
+  TRDNoiseCalibSpec(bool dummy) : mIsDummy(dummy) {}
   void init(o2::framework::InitContext& ic) final
   {
     // Do we need to initialize something?
@@ -44,6 +44,10 @@ class TRDNoiseCalibSpec : public o2::framework::Task
     auto digits = pc.inputs().get<gsl::span<Digit>>("trddigits");
     auto trigRecs = pc.inputs().get<gsl::span<TriggerRecord>>("trdtriggerrec");
 
+    if (mIsDummy) {
+      return;
+    }
+
     // Obtain rough time from the data header (done only once)
     if (mStartTime == 0) {
       o2::dataformats::TFIDInfo ti;
@@ -63,7 +67,7 @@ class TRDNoiseCalibSpec : public o2::framework::Task
         sendOutput(pc.outputs());
         mHaveSentOutput = true;
       } else {
-        if ((mNTFsProcessed % 50) == 0) {
+        if ((mNTFsProcessed % 200) == 0) {
           LOGP(important, "Not processing anymore. Seen {} TFs in total. Run can be stopped", mNTFsProcessed);
         }
       }
@@ -96,18 +100,21 @@ class TRDNoiseCalibSpec : public o2::framework::Task
     if (mHaveSentOutput) {
       LOGP(important, "Received EoS after sending calibration object. All OK");
     } else {
-      LOGP(alarm, "Received EoS before sending calibration object. Not enough digits received");
+      if (!mIsDummy) {
+        LOGP(alarm, "Received EoS before sending calibration object. Not enough digits received");
+      }
     }
   }
 
  private:
   CalibratorNoise mCalibrator{};
   size_t mNTFsProcessed{0};
   bool mHaveSentOutput{false};
+  bool mIsDummy{false};
   uint64_t mStartTime{0};
 };
 
-o2::framework::DataProcessorSpec getTRDNoiseCalibSpec()
+o2::framework::DataProcessorSpec getTRDNoiseCalibSpec(bool dummy)
 {
   std::vector<InputSpec> inputs;
   inputs.emplace_back("trddigits", o2::header::gDataOriginTRD, "DIGITS", 0, Lifetime::Timeframe);
@@ -121,7 +128,7 @@ o2::framework::DataProcessorSpec getTRDNoiseCalibSpec()
     "trd-noise-calib",
     inputs,
     outputs,
-    AlgorithmSpec{adaptFromTask<TRDNoiseCalibSpec>()},
+    AlgorithmSpec{adaptFromTask<TRDNoiseCalibSpec>(dummy)},
     Options{}};
 }
 
diff --git a/Detectors/TRD/workflow/src/trd-calib-workflow.cxx b/Detectors/TRD/workflow/src/trd-calib-workflow.cxx
@@ -28,6 +28,7 @@ void customize(std::vector<o2::framework::ConfigParamSpec>& workflowOptions)
     {"vDriftAndExB", o2::framework::VariantType::Bool, false, {"enable vDrift and ExB calibration"}},
     {"noise", o2::framework::VariantType::Bool, false, {"enable noise and pad status calibration"}},
     {"gain", o2::framework::VariantType::Bool, false, {"enable gain calibration"}},
+    {"calib-dds-collection-index", VariantType::Int, -1, {"allow only single collection to produce calibration objects (use -1 for no limit)"}},
     {"configKeyValues", VariantType::String, "", {"Semicolon separated key=value strings"}}};
 
   std::swap(workflowOptions, options);
@@ -55,7 +56,21 @@ WorkflowSpec defineDataProcessing(ConfigContext const& configcontext)
     if (enableRootInp) {
       specs.emplace_back(o2::trd::getTRDDigitReaderSpec(false));
     }
-    specs.emplace_back(o2::trd::getTRDNoiseCalibSpec());
+    int ddsCollectionIdx = configcontext.options().get<int>("calib-dds-collection-index");
+    bool noiseCalibIsDummy = true;
+    if (ddsCollectionIdx != -1) {
+      char* colIdx = getenv("DDS_COLLECTION_INDEX");
+      int myIdx = colIdx ? atoi(colIdx) : -1;
+      if (myIdx == ddsCollectionIdx) {
+        LOG(info) << "TRD noise calib is enabled for this collection, my index " << myIdx;
+        noiseCalibIsDummy = false;
+      } else {
+        LOG(info) << "TRD noise calib is disabled for this collection, my index " << myIdx;
+      }
+    } else {
+      noiseCalibIsDummy = false;
+    }
+    specs.emplace_back(o2::trd::getTRDNoiseCalibSpec(noiseCalibIsDummy));
   }
 
   if (configcontext.options().get<bool>("gain")) {
