DPL: add support for lhcPeriod in DataTakingContext

ktf · ktf · commit 435dcbc9f433 · 2022-05-27T08:46:08.000+02:00
diff --git a/Framework/Core/include/Framework/DataTakingContext.h b/Framework/Core/include/Framework/DataTakingContext.h
@@ -34,6 +34,14 @@ struct DataTakingContext {
   uint64_t orbitResetTime = INVALID_RESET_TIME;
   // What currently set the orbitResetTime value.
   OrbitResetTimeSource source = OrbitResetTimeSource::Default;
+  /// The current lhc period
+  std::string lhcPeriod = "unknown";
+  /// The run type of the current run
+  std::string runType = "unknown";
+  /// The environment ID for the deployment
+  std::string envId = "unknown";
+  /// The list of detectors taking part in the run
+  std::string detectors = "unknown";
 };
 
 } // namespace o2::framework
diff --git a/Framework/Core/src/CommonServices.cxx b/Framework/Core/src/CommonServices.cxx
@@ -168,6 +168,29 @@ o2::framework::ServiceSpec CommonServices::datatakingContextSpec()
       if (extRunNumber != "unspecified" || context.runNumber == "0") {
         context.runNumber = extRunNumber;
       }
+      auto extLHCPeriod = services.get<RawDeviceService>().device()->fConfig->GetProperty<std::string>("lhc_period", "unspecified");
+      if (extLHCPeriod != "unspecified") {
+        context.lhcPeriod = extLHCPeriod;
+      } else {
+        static const char* months[12] = {"JAN", "FEB", "MAR", "APR", "MAY", "JUN", "JUL", "AUG", "SEP", "OCT", "NOV", "DEC"};
+        time_t now = time(nullptr);
+        auto ltm = gmtime(&now);
+        context.lhcPeriod = months[ltm->tm_mon];
+        LOG(warning) << "LHCPeriod is not available, using current month " << context.lhcPeriod;
+      }
+
+      auto extRunType = services.get<RawDeviceService>().device()->fConfig->GetProperty<std::string>("run_type", "unspecified");
+      if (extRunType != "unspecified") {
+        context.runType = extRunType;
+      }
+      auto extEnvId = services.get<RawDeviceService>().device()->fConfig->GetProperty<std::string>("environment_id", "unspecified");
+      if (extEnvId != "unspecified") {
+        context.runType = extEnvId;
+      }
+      auto extDetectors = services.get<RawDeviceService>().device()->fConfig->GetProperty<std::string>("detectors", "unspecified");
+      if (extDetectors != "unspecified") {
+        context.detectors = extDetectors;
+      }
       // FIXME: we actually need to get the orbit, not only to know where it is
       std::string orbitResetTimeUrl = services.get<RawDeviceService>().device()->fConfig->GetProperty<std::string>("orbit-reset-time", "ccdb://CTP/Calib/OrbitResetTime");
       auto is_number = [](const std::string& s) -> bool {
