lasdlfd

Author: f3sch

Detectors/TRD/calibration/include/TRDCalibration/CalibratorPID.h

@@ -11,8 +11,11 @@
 #ifndef O2_TRD_CALIBRATORPID_H
 #define O2_TRD_CALIBRATORPID_H
 
+#include <memory>
+
 #include "DataFormatsGlobalTracking/RecoContainer.h"
 #include "ReconstructionDataFormats/GlobalTrackID.h"
+#include "CommonUtils/TreeStreamRedirector.h"
 
 #include "Rtypes.h"
 
@@ -22,13 +25,14 @@ namespace o2::trd
 class CalibratorPID
 {
  public:
-  CalibratorPID(o2::dataformats::GlobalTrackID::mask_t trcSrc, bool withSV) : mTracksSrc(trcSrc), mWithSV(withSV) {}
+  CalibratorPID(o2::dataformats::GlobalTrackID::mask_t trcSrc) : mTracksSrc(trcSrc) {}
 
   void run(const o2::globaltracking::RecoContainer& input);
+  void setTree(std::shared_ptr<o2::utils::TreeStreamRedirector> tree) { mTree = tree; }
 
  private:
   o2::dataformats::GlobalTrackID::mask_t mTracksSrc{};
-  bool mWithSV{false};
+  std::shared_ptr<o2::utils::TreeStreamRedirector> mTree;
 
   ClassDefNV(CalibratorPID, 1);
 };

Detectors/TRD/calibration/src/CalibratorPID.cxx

@@ -10,10 +10,46 @@
 // or submit itself to any jurisdiction.
 
 #include "TRDCalibration/CalibratorPID.h"
+#include "ReconstructionDataFormats/DecayNBodyIndex.h"
+#include "DataFormatsTRD/TrackTRD.h"
+#include "Framework/Logger.h"
 
 namespace o2::trd
 {
 
-void CalibratorPID::run(const o2::globaltracking::RecoContainer& input) {}
+void CalibratorPID::run(const o2::globaltracking::RecoContainer& reco)
+{
+  const auto& v0sIdx = reco.getV0sIdx();
+  for (const auto& v0Idx : v0sIdx) {
+    const auto prongs{v0Idx.getProngs()};
+    if (!reco.isTrackSourceLoaded(prongs[0].getSource()) || !reco.isTrackSourceLoaded(prongs[1].getSource())) {
+      continue;
+    }
+    auto seedP = reco.getTrackParam(prongs[0]);
+    auto seedN = reco.getTrackParam(prongs[1]);
+    if (!mFitterV0.process(seedP, seedN) && !mFitterV0.isPropagateTracksToVertexDone() && !mFitterV0.propagateTracksToVertex()) {
+      continue;
+    }
+    const auto& trPProp = mFitterV0.getTrack(0);
+    const auto& trNProp = mFitterV0.getTrack(1);
+    std::array<float, 3> pP{}, pN{};
+    trPProp.getPxPyPzGlo(pP);
+    trNProp.getPxPyPzGlo(pN);
+    std::array<float, 3> pV0a = {pP[0] + pN[0], pP[1] + pN[1], pP[2] + pN[2]};
+    float p2V0 = pV0a[0] * pV0a[0] + pV0a[1] * pV0a[1] + pV0a[2] * pV0a[2], pV0Abs = std::sqrt(p2V0);
+    float p2Pos = pP[0] * pP[0] + pP[1] * pP[1] + pP[2] * pP[2], p2Neg = pN[0] * pN[0] + pN[1] * pN[1] + pN[2] * pN[2];
+    float pPos = std::sqrt(p2Pos), pNeg = std::sqrt(p2Neg);
+    float pL1 = (pV0a[0] * pP[0] + pV0a[1] * pP[1] + pV0a[2] * pP[2]) / pV0Abs;
+    float dp = (pV0a[0] * pN[0] + pV0a[1] * pN[1] + pV0a[2] * pN[2]);
+    float pL2 = dp / pV0Abs;
+    float alpha = (pL1 - pL2) / (pL1 + pL2);
+    float qt = std::sqrt((p2Neg - dp * dp) / p2V0);
+
+    (*mTree) << "ap"
+             << "alpha=" << alpha
+             << "qt=" << qt
+             << "\n";
+  }
+}
 
 } // namespace o2::trd

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

@@ -12,12 +12,15 @@
 #ifndef O2_TRD_PIDCALIBSPEC_H
 #define O2_TRD_PIDCALIBSPEC_H
 
+#include <memory>
+
 #include "Framework/Task.h"
 #include "Framework/DataProcessorSpec.h"
 #include "DetectorsBase/GRPGeomHelper.h"
 #include "ReconstructionDataFormats/GlobalTrackID.h"
 #include "DataFormatsGlobalTracking/RecoContainer.h"
 #include "TRDCalibration/CalibratorPID.h"
+#include "CommonUtils/TreeStreamRedirector.h"
 
 namespace o2
 {
@@ -27,12 +30,16 @@ 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), mCalibrator(src, withSV) {}
+  PIDCalibDevice(std::shared_ptr<o2::globaltracking::DataRequest> dr, std::shared_ptr<o2::base::GRPGeomRequest> gr, o2::dataformats::GlobalTrackID::mask_t src)
+    : mDataRequest(dr), mGGCCDBRequest(gr), mTracksSrc(src), mCalibrator(src) {}
 
   void init(o2::framework::InitContext& ic) final
   {
     o2::base::GRPGeomHelper::instance().setRequest(mGGCCDBRequest);
+
+    mTree = std::make_shared<o2::utils::TreeStreamRedirector>("trd_pid_calib.root");
+    mTree->SetMaxFileSize(ic.options().get<uint64_t>("max-file-size"));
+    mCalibrator.setTree(mTree);
   }
 
   void run(o2::framework::ProcessingContext& pc) final
@@ -43,6 +50,11 @@ class PIDCalibDevice final : public o2::framework::Task
     mCalibrator.run(recoData);
   }
 
+  void endOfStream(o2::framework::EndOfStreamContext& ec) final
+  {
+    mTree->Close();
+  }
+
   void finaliseCCDB(o2::framework::ConcreteDataMatcher& matcher, void* obj) final
   {
     o2::base::GRPGeomHelper::instance().finaliseCCDB(matcher, obj);
@@ -54,6 +66,7 @@ class PIDCalibDevice final : public o2::framework::Task
   o2::dataformats::GlobalTrackID::mask_t mTracksSrc{};
   bool mWithSV{false};
   o2::trd::CalibratorPID mCalibrator;
+  std::shared_ptr<o2::utils::TreeStreamRedirector> mTree;
 
   void updateTimeDependentParams(o2::framework::ProcessingContext& pc)
   {
@@ -66,14 +79,12 @@ class PIDCalibDevice final : public o2::framework::Task
 namespace framework
 {
 
-DataProcessorSpec getTRDPIDCalibSpec(o2::dataformats::GlobalTrackID::mask_t srcTracks, bool withSV, bool useMC)
+DataProcessorSpec getTRDPIDCalibSpec(o2::dataformats::GlobalTrackID::mask_t srcTracks, bool useMC)
 {
   std::vector<OutputSpec> outputs;
   auto dataRequest = std::make_shared<o2::globaltracking::DataRequest>();
   dataRequest->requestTracks(srcTracks, useMC);
-  if (withSV) {
-    dataRequest->requestSecondaryVertices(useMC);
-  }
+  dataRequest->requestSecondaryVertices(useMC);
   dataRequest->requestClusters(o2::dataformats::GlobalTrackID::getSourcesMask("TRD"), useMC);
 
   auto ggRequest = std::make_shared<o2::base::GRPGeomRequest>(false,                          // orbitResetTime
@@ -88,8 +99,10 @@ DataProcessorSpec getTRDPIDCalibSpec(o2::dataformats::GlobalTrackID::mask_t srcT
   return DataProcessorSpec{
     "pid-calibration",
     dataRequest->inputs, outputs,
-    AlgorithmSpec{adaptFromTask<o2::calibration::trd::PIDCalibDevice>(dataRequest, ggRequest, srcTracks, withSV)},
-    Options{}};
+    AlgorithmSpec{adaptFromTask<o2::calibration::trd::PIDCalibDevice>(dataRequest, ggRequest, srcTracks)},
+    Options{
+      {"max-file-size", o2::framework::VariantType::UInt64, 0ULL, {"Set max size of output file"}},
+    }};
 }
 } // namespace framework
 } // namespace o2

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

@@ -41,10 +41,9 @@ void customize(std::vector<o2::framework::ConfigParamSpec>& workflowOptions)
     {"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);
+  o2::raw::HBFUtilsInitializer::addConfigOption(options, "none");
   std::swap(workflowOptions, options);
 }
 
@@ -104,11 +103,8 @@ WorkflowSpec defineDataProcessing(ConfigContext const& configcontext)
     auto srcTrc = allowedTrkSources & GID::getSourcesMask(configcontext.options().get<std::string>("track-sources"));
     auto srcClu = GID::getSourcesMask("TRD");
     o2::globaltracking::InputHelper::addInputSpecs(configcontext, specs, srcClu, srcTrc, srcTrc, useMC);
-    auto withSV = configcontext.options().get<bool>("with-sv");
-    if (withSV) {
-      o2::globaltracking::InputHelper::addInputSpecsSVertex(configcontext, specs);
-    }
-    specs.emplace_back(getTRDPIDCalibSpec(srcTrc, withSV, useMC));
+    o2::globaltracking::InputHelper::addInputSpecsSVertex(configcontext, specs);
+    specs.emplace_back(getTRDPIDCalibSpec(srcTrc, useMC));
     o2::raw::HBFUtilsInitializer hbfIni(configcontext, specs);
   }