AOD: Add trd tracklet streamer

Signed-off-by: Felix Schlepper <felix.schlepper@cern.ch>

Author: f3sch

Detectors/AOD/include/AODProducerWorkflow/AODProducerWorkflowSpec.h

@@ -20,6 +20,7 @@
 #include "DataFormatsPHOS/Cell.h"
 #include "DataFormatsTRD/TrackTRD.h"
 #include "TRDBase/PadCalibrationsAliases.h"
+#include "TRDBase/TrackletTransformer.h"
 #include "DataFormatsTRD/NoiseCalibration.h"
 #include "DetectorsBase/GRPGeomHelper.h"
 #include "DetectorsBase/Propagator.h"
@@ -250,6 +251,7 @@ class AODProducerWorkflowDPL : public Task
 
   const o2::trd::LocalGainFactor* mTRDLocalGain; // TRD local gain factors from krypton calibration
   const o2::trd::NoiseStatusMCM* mTRDNoiseMap;   // TRD noise map
+  std::unique_ptr<o2::trd::TrackletTransformer> mTRDTransformer;
 
   std::unordered_set<GIndex> mGIDUsedBySVtx;
   std::unordered_set<GIndex> mGIDUsedByStr;

Detectors/AOD/src/AODProducerWorkflowSpec.cxx

@@ -396,7 +396,9 @@ void AODProducerWorkflowDPL::addToTRDsExtra(const o2::globaltracking::RecoContai
   int q0s[NLAYERS] = {-1}, q1s[NLAYERS] = {-1}, q2s[NLAYERS] = {-1};
   float q0sCor[NLAYERS] = {-1}, q1sCor[NLAYERS] = {-1}, q2sCor[NLAYERS] = {-1};
   float ttgls[NLAYERS] = {-999}, tphis[NLAYERS] = {-999};
-  o2::trd::Tracklet64 trkletsa[NLAYERS];
+  std::vector<o2::trd::Tracklet64> trkletsa(NLAYERS);
+  std::vector<o2::trd::CalibratedTracklet> ctrkletsa(NLAYERS);
+  std::vector<o2::trd::CalibratedTracklet> cloctrkletsa(NLAYERS);
 
   auto contributorsGID = recoData.getSingleDetectorRefs(trkIdx);
   if (!contributorsGID[GIndex::Source::TRD].isIndexSet()) { // should be redunant
@@ -418,6 +420,8 @@ void AODProducerWorkflowDPL::addToTRDsExtra(const o2::globaltracking::RecoContai
       continue;
     }
     trkletsa[iLay] = tracklet;
+    ctrkletsa[iLay] = ctrklets[trkltId];
+    cloctrkletsa[iLay] = mTRDTransformer->transformTracklet(tracklet, false);
     // we need to propagate into TRD local system
     int trkltDet = tracklet.getDetector();
     int trkltSec = trkltDet / 30;
@@ -433,16 +437,14 @@ void AODProducerWorkflowDPL::addToTRDsExtra(const o2::globaltracking::RecoContai
     auto tphi = trkC.getSnp() / std::sqrt((1.f - trkC.getSnp()) * (1.f + trkC.getSnp()));
     auto trackletLength = std::sqrt(1.f + tphi * tphi + trkC.getTgl() * trkC.getTgl());
     float cor = mTRDLocalGain->getValue(tracklet.getHCID() / 2, tracklet.getPadCol(), tracklet.getPadRow()) * trackletLength;
-    if (mEnableTRDextra) {
-      q0s[iLay] = tracklet.getQ0();
-      q1s[iLay] = tracklet.getQ1();
-      q2s[iLay] = tracklet.getQ2();
-      q0sCor[iLay] = (float)tracklet.getQ0() / cor;
-      q1sCor[iLay] = (float)tracklet.getQ1() / cor;
-      q2sCor[iLay] = (float)tracklet.getQ2() / cor;
-      ttgls[iLay] = trkC.getTgl();
-      tphis[iLay] = tphi;
-    }
+    q0s[iLay] = tracklet.getQ0();
+    q1s[iLay] = tracklet.getQ1();
+    q2s[iLay] = tracklet.getQ2();
+    q0sCor[iLay] = (float)tracklet.getQ0() / cor;
+    q1sCor[iLay] = (float)tracklet.getQ1() / cor;
+    q2sCor[iLay] = (float)tracklet.getQ2() / cor;
+    ttgls[iLay] = trkC.getTgl();
+    tphis[iLay] = tphi;
 
     dEdx += (float)tracklet.getQTot() / cor;
   }
@@ -462,49 +464,16 @@ void AODProducerWorkflowDPL::addToTRDsExtra(const o2::globaltracking::RecoContai
     (*mStreamer) << "trdExtra"
                  << "dEdx=" << dEdx
                  << "eProb=" << trk.getSignal()
-                 << "q00=" << q0s[0]
-                 << "q01=" << q0s[1]
-                 << "q02=" << q0s[2]
-                 << "q03=" << q0s[3]
-                 << "q04=" << q0s[4]
-                 << "q05=" << q0s[5]
-                 << "q10=" << q1s[0]
-                 << "q11=" << q1s[1]
-                 << "q12=" << q1s[2]
-                 << "q13=" << q1s[3]
-                 << "q14=" << q1s[4]
-                 << "q15=" << q1s[5]
-                 << "q20=" << q2s[0]
-                 << "q21=" << q2s[1]
-                 << "q22=" << q2s[2]
-                 << "q23=" << q2s[3]
-                 << "q24=" << q2s[4]
-                 << "q25=" << q2s[5]
-                 << "q00Cor=" << q0sCor[0]
-                 << "q01Cor=" << q0sCor[1]
-                 << "q02Cor=" << q0sCor[2]
-                 << "q03Cor=" << q0sCor[3]
-                 << "q04Cor=" << q0sCor[4]
-                 << "q05Cor=" << q0sCor[5]
-                 << "q10Cor=" << q1sCor[0]
-                 << "q11Cor=" << q1sCor[1]
-                 << "q12Cor=" << q1sCor[2]
-                 << "q13Cor=" << q1sCor[3]
-                 << "q14Cor=" << q1sCor[4]
-                 << "q15Cor=" << q1sCor[5]
-                 << "q20Cor=" << q2sCor[0]
-                 << "q21Cor=" << q2sCor[1]
-                 << "q22Cor=" << q2sCor[2]
-                 << "q23Cor=" << q2sCor[3]
-                 << "q24Cor=" << q2sCor[4]
-                 << "q25Cor=" << q2sCor[5]
+                 << "q0s[6]=" << q0s
+                 << "q1s[6]=" << q1s
+                 << "q2s[6]=" << q2s
+                 << "q0sCor[6]=" << q0sCor
+                 << "q1sCor[6]=" << q1sCor
+                 << "q2sCor[6]=" << q2sCor
                  << "pattern=" << getTRDPattern(trk)
-                 << "tracklet0=" << trkletsa[0]
-                 << "tracklet1=" << trkletsa[1]
-                 << "tracklet2=" << trkletsa[2]
-                 << "tracklet3=" << trkletsa[3]
-                 << "tracklet4=" << trkletsa[4]
-                 << "tracklet5=" << trkletsa[5]
+                 << "tracklets=" << trkletsa
+                 << "ctracklets=" << ctrkletsa
+                 << "cloctracklets=" << cloctrkletsa
                  << "tpctrk=" << tpctrk
                  << "trdtrk=" << trk
                  << "globaltrk=" << recoData.getTrack<o2::track::TrackParCov>(trkIdx)
@@ -3036,6 +3005,9 @@ void AODProducerWorkflowDPL::updateTimeDependentParams(ProcessingContext& pc)
     if (mEnableTRDextra) {
       mTRDLocalGain = pc.inputs().get<o2::trd::LocalGainFactor*>("trdlocalgainfactors").get();
       mTRDNoiseMap = pc.inputs().get<o2::trd::NoiseStatusMCM*>("trdnoisemap").get();
+      mTRDTransformer = std::make_unique<o2::trd::TrackletTransformer>();
+      mTRDTransformer->init();
+      pc.inputs().get<o2::trd::CalVdriftExB*>("calvdexb");
     }
   }
   if (mPropTracks) {
@@ -3070,6 +3042,11 @@ void AODProducerWorkflowDPL::finaliseCCDB(ConcreteDataMatcher& matcher, void* ob
     mVtx = *(const o2::dataformats::MeanVertexObject*)obj;
     return;
   }
+  if (matcher == ConcreteDataMatcher("TRD", "CALVDRIFTEXB", 0)) {
+    LOG(info) << "TRD CalVdriftExB object has been updated";
+    mTRDTransformer->setCalVdriftExB((const o2::trd::CalVdriftExB*)obj);
+    return;
+  }
   if (matcher == ConcreteDataMatcher("CTP", "CTPCONFIG", 0)) {
     // construct mask with EMCAL trigger classes for rejection of incomplete triggers
     auto ctpconfig = *(const o2::ctp::CTPConfiguration*)obj;
@@ -3348,6 +3325,7 @@ DataProcessorSpec getAODProducerWorkflowSpec(GID::mask_t src, bool enableSV, boo
     outputs.push_back(OutputForTable<TRDsExtra>::spec());
     dataRequest->inputs.emplace_back("trdlocalgainfactors", "TRD", "LOCALGAINFACTORS", 0, Lifetime::Condition, ccdbParamSpec("TRD/Calib/LocalGainFactor"));
     dataRequest->inputs.emplace_back("trdnoisemap", "TRD", "NOISEMAP", 0, Lifetime::Condition, ccdbParamSpec("TRD/Calib/NoiseMapMCM"));
+    dataRequest->inputs.emplace_back("calvdexb", "TRD", "CALVDRIFTEXB", 0, Lifetime::Condition, ccdbParamSpec("TRD/Calib/CalVdriftExB"));
   }
   if (useMC) {
     outputs.insert(outputs.end(),