TRD: PID calibration workflow

Author: f3sch

Detectors/TRD/workflow/CMakeLists.txt

@@ -28,6 +28,7 @@ o2_add_library(TRDWorkflow
                        include/TRDWorkflow/TRDGlobalTrackingQCSpec.h
                        include/TRDWorkflow/NoiseCalibSpec.h
                        include/TRDWorkflow/T0FitSpec.h
+                       include/TRDWorkflow/PIDCalibSpec.h
                PUBLIC_LINK_LIBRARIES O2::Framework O2::DPLUtils
                                      O2::Steer
                                      O2::Algorithm

Detectors/TRD/workflow/include/TRDWorkflow/PIDCalibSpec.h

@@ -0,0 +1,92 @@
+// Copyright 2019-2020 CERN and copyright holders of ALICE O2.
+// See https://alice-o2.web.cern.ch/copyright for details of the copyright holders.
+// All rights not expressly granted are reserved.
+//
+// This software is distributed under the terms of the GNU General Public
+// License v3 (GPL Version 3), copied verbatim in the file "COPYING".
+//
+// In applying this license CERN does not waive the privileges and immunities
+// granted to it by virtue of its status as an Intergovernmental Organization
+// or submit itself to any jurisdiction.
+
+#ifndef O2_TRD_PIDCALIBSPEC_H
+#define O2_TRD_PIDCALIBSPEC_H
+
+#include "Framework/Task.h"
+#include "Framework/DataProcessorSpec.h"
+#include "DetectorsBase/GRPGeomHelper.h"
+#include "ReconstructionDataFormats/GlobalTrackID.h"
+#include "DataFormatsGlobalTracking/RecoContainer.h"
+
+namespace o2
+{
+namespace calibration::trd
+{
+
+class PIDCalibDevice final : public o2::framework::Task
+{
+ public:
+  PIDCalibDevice(std::shared_ptr<o2::globaltracking::DataRequest> dr, std::shared_ptr<o2::base::GRPGeomRequest> gr, o2::dataformats::GlobalTrackID::mask_t src, bool withSV)
+    : mDataRequest(dr), mGGCCDBRequest(gr), mTracksSrc(src), mWithSV(withSV) {}
+
+  void init(o2::framework::InitContext& ic) final
+  {
+    o2::base::GRPGeomHelper::instance().setRequest(mGGCCDBRequest);
+  }
+
+  void run(o2::framework::ProcessingContext& pc) final
+  {
+    o2::globaltracking::RecoContainer recoData;
+    recoData.collectData(pc, *mDataRequest.get()); // select tracks of needed type, with minimal cuts, the real selected will be done in the vertexer
+    updateTimeDependentParams(pc);                 // Make sure this is called after recoData.collectData, which may load some conditions
+  }
+
+  void finaliseCCDB(o2::framework::ConcreteDataMatcher& matcher, void* obj) final
+  {
+    o2::base::GRPGeomHelper::instance().finaliseCCDB(matcher, obj);
+  }
+
+ private:
+  std::shared_ptr<o2::globaltracking::DataRequest> mDataRequest;
+  std::shared_ptr<o2::base::GRPGeomRequest> mGGCCDBRequest;
+  o2::dataformats::GlobalTrackID::mask_t mTracksSrc{};
+  bool mWithSV{false};
+
+  void updateTimeDependentParams(o2::framework::ProcessingContext& pc)
+  {
+    o2::base::GRPGeomHelper::instance().checkUpdates(pc);
+  }
+};
+
+} // namespace calibration::trd
+
+namespace framework
+{
+
+DataProcessorSpec getTRDPIDCalibSpec(o2::dataformats::GlobalTrackID::mask_t srcTracks, bool withSV)
+{
+  std::vector<OutputSpec> outputs;
+  auto dataRequest = std::make_shared<o2::globaltracking::DataRequest>();
+  dataRequest->requestTracks(srcTracks, false);
+  if (withSV) {
+    dataRequest->requestSecondaryVertices(false);
+  }
+
+  auto ggRequest = std::make_shared<o2::base::GRPGeomRequest>(false,                          // orbitResetTime
+                                                              false,                          // GRPECS=true
+                                                              false,                          // GRPLHCIF
+                                                              true,                           // GRPMagField
+                                                              true,                           // askMatLUT
+                                                              o2::base::GRPGeomRequest::None, // geometry
+                                                              dataRequest->inputs,
+                                                              true);
+  return DataProcessorSpec{
+    "pid-calibration",
+    dataRequest->inputs, outputs,
+    AlgorithmSpec{adaptFromTask<o2::calibration::trd::PIDCalibDevice>(dataRequest, ggRequest, srcTracks, withSV)},
+    Options{}};
+}
+} // namespace framework
+} // namespace o2
+
+#endif

Detectors/TRD/workflow/src/trd-calib-workflow.cxx

@@ -17,9 +17,16 @@
 #include "TRDWorkflow/GainCalibSpec.h"
 #include "TRDWorkflow/NoiseCalibSpec.h"
 #include "TRDWorkflow/T0FitSpec.h"
+#include "TRDWorkflow/PIDCalibSpec.h"
 #include "CommonUtils/ConfigurableParam.h"
+#include "ReconstructionDataFormats/GlobalTrackID.h"
+#include "DetectorsCommonDataFormats/DetID.h"
+#include "GlobalTrackingWorkflowHelpers/InputHelper.h"
+#include "DetectorsRaw/HBFUtilsInitializer.h"
 
 using namespace o2::framework;
+using GID = o2::dataformats::GlobalTrackID;
+using DetID = o2::detectors::DetID;
 
 // we need to add workflow options before including Framework/runDataProcessing
 void customize(std::vector<o2::framework::ConfigParamSpec>& workflowOptions)
@@ -31,9 +38,12 @@ void customize(std::vector<o2::framework::ConfigParamSpec>& workflowOptions)
     {"noise", o2::framework::VariantType::Bool, false, {"enable noise and pad status calibration"}},
     {"gain", o2::framework::VariantType::Bool, false, {"enable gain calibration"}},
     {"t0", o2::framework::VariantType::Bool, false, {"enable t0 fit"}},
+    {"pid", o2::framework::VariantType::Bool, false, {"enable pid calibration"}},
+    {"track-sources", VariantType::String, std::string{GID::ALL}, {"comma-separated list of track sources to use"}},
+    {"with-sv", o2::framework::VariantType::Bool, false, {"request secondary-vertices"}},
     {"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"}}};
-
+  o2::raw::HBFUtilsInitializer::addConfigOption(options);
   std::swap(workflowOptions, options);
 }
 
@@ -87,5 +97,17 @@ WorkflowSpec defineDataProcessing(ConfigContext const& configcontext)
     specs.emplace_back(getTRDT0FitSpec());
   }
 
+  if (configcontext.options().get<bool>("pid")) {
+    const auto allowedTrkSources = GID::getSourcesMask("ITS,TPC,TRD,TOF,ITS-TPC-TRD,TPC-TRD,TPC-TRD-TOF,ITS-TPC-TRD-TOF");
+    auto srcTrc = allowedTrkSources & GID::getSourcesMask(configcontext.options().get<std::string>("track-sources"));
+    o2::globaltracking::InputHelper::addInputSpecs(configcontext, specs, {}, srcTrc, srcTrc, false);
+    auto withSV = configcontext.options().get<bool>("with-sv");
+    if (withSV) {
+      o2::globaltracking::InputHelper::addInputSpecsSVertex(configcontext, specs);
+    }
+    specs.emplace_back(getTRDPIDCalibSpec(srcTrc, withSV));
+    o2::raw::HBFUtilsInitializer hbfIni(configcontext, specs);
+  }
+
   return specs;
 }