treeCreatorPidTpcQa: mainly working version

Author: lubynets

DPG/Tasks/AOTTrack/PID/TPC/treeCreatorPidTpcQa.cxx

@@ -38,6 +38,7 @@
 #include <ReconstructionDataFormats/PID.h>
 
 #include <cmath>
+#include <cstdint>
 #include <string>
 
 using namespace o2;
@@ -122,7 +123,7 @@ struct treeCreatorPidTpcQa {
     ccdb->setFatalWhenNull(false);
   }
 
-  template <o2::track::PID::ID Id, typename TrackType>
+  template <o2::track::PID::ID Id, bool IsFullTable, bool IsTofTable, typename TrackType>
   void processSingleParticle(CollisionsExtra const& collisions,
                              TrackType const& tracks)
   {
@@ -162,12 +163,30 @@ struct treeCreatorPidTpcQa {
           continue;
         }
 
-        const int nClNormalized = std::sqrt(nClNorm / track.tpcNClsFound());
+        const float nClNormalized = std::sqrt(nClNorm / track.tpcNClsFound());
         const float phi = track.phi();
         const float tgl = track.tgl();
         const float tpcInnerParam = track.tpcInnerParam();
+        const int16_t trackSign = track.sign();
+        const float nSigmaTpc = o2::aod::pidutils::tpcNSigma<Id>(track);
 
-        rowPidTpcQa(Id, ft0Occ, hadronicRate, multTPC, nClNormalized, phi, tgl, tpcInnerParam, rapidity);
+        float dedxDiff{UndefValueFloat};
+        float dedxExpected{UndefValueFloat};
+        float expSigma{UndefValueFloat};
+
+        if constexpr (IsFullTable) {
+          dedxDiff = o2::aod::pidutils::tpcExpSignalDiff<Id>(track);
+          dedxExpected = track.tpcSignal() - dedxDiff;
+          expSigma = o2::aod::pidutils::tpcExpSigma<Id>(track);
+        }
+
+        float nSigmaTof{UndefValueFloat};
+
+        if constexpr (IsTofTable) {
+          nSigmaTof = o2::aod::pidutils::tofNSigma<Id>(track);
+        }
+
+        rowPidTpcQa(Id, ft0Occ, hadronicRate, multTPC, nClNormalized, phi, tgl, tpcInnerParam, rapidity, trackSign, nSigmaTpc, dedxExpected, dedxDiff, expSigma, nSigmaTof);
       } // tracksFromCollision
     } // collisions
   }
@@ -178,7 +197,7 @@ struct treeCreatorPidTpcQa {
                            soa::Join<TrackCandidates, PidTableTPC> const& tracks, \
                            aod::BCsWithTimestamps const&)                         \
   {                                                                               \
-    processSingleParticle<PID::ParticleId>(collisions, tracks);                   \
+    processSingleParticle<PID::ParticleId, false, false>(collisions, tracks);     \
   }                                                                               \
   PROCESS_SWITCH(treeCreatorPidTpcQa, process##ParticleId, Form("Process for the %s hypothesis for TPC NSigma QA", #ParticleId), false);
 
@@ -199,7 +218,7 @@ struct treeCreatorPidTpcQa {
                                soa::Join<TrackCandidates, PidTableTPC> const& tracks, \
                                aod::BCsWithTimestamps const&)                         \
   {                                                                                   \
-    processSingleParticle<PID::ParticleId>(collisions, tracks);                       \
+    processSingleParticle<PID::ParticleId, true, false>(collisions, tracks);          \
   }                                                                                   \
   PROCESS_SWITCH(treeCreatorPidTpcQa, processFull##ParticleId, Form("Process for the %s hypothesis for full TPC PID QA", #ParticleId), false);
 
@@ -220,7 +239,7 @@ struct treeCreatorPidTpcQa {
                                       soa::Join<TrackCandidates, PidTableTPC, PidTableTOF> const& tracks, \
                                       aod::BCsWithTimestamps const&)                                      \
   {                                                                                                       \
-    processSingleParticle<PID::ParticleId>(collisions, tracks);                                           \
+    processSingleParticle<PID::ParticleId, true, true>(collisions, tracks);                               \
   }                                                                                                       \
   PROCESS_SWITCH(treeCreatorPidTpcQa, processFullWithTOF##ParticleId, Form("Process for the %s hypothesis for full TPC PID QA with the TOF info added", #ParticleId), false);
 

DPG/Tasks/AOTTrack/PID/TPC/treeCreatorPidTpcQa.h

@@ -22,8 +22,20 @@
 #include <Framework/ASoA.h>
 #include <Framework/AnalysisDataModel.h>
 
+#include <cstdint>
+
 namespace o2::aod
 {
+namespace dpg_tpcpidqa
+{
+DECLARE_SOA_COLUMN(Sign, sign, int16_t);
+DECLARE_SOA_COLUMN(NSigmaTpc, nSigmaTpc, float);
+DECLARE_SOA_COLUMN(DedxExpected, dedxTpc, float);
+DECLARE_SOA_COLUMN(DedxDiff, dedxDiff, float);
+DECLARE_SOA_COLUMN(ExpSigma, expSigma, float);
+DECLARE_SOA_COLUMN(NSigmaTof, nSigmaTof, float);
+} // namespace dpg_tpcpidqa
+
 DECLARE_SOA_TABLE(QaPidTpc, "AOD", "QAPIDTPC",
                   tpcskims::PidIndex,
                   tpcskims::Ft0Occ,
@@ -33,7 +45,13 @@ DECLARE_SOA_TABLE(QaPidTpc, "AOD", "QAPIDTPC",
                   o2::aod::track::Phi,
                   o2::aod::track::Tgl,
                   o2::aod::track::TPCInnerParam,
-                  o2::aod::track::Y)
+                  o2::aod::track::Y,
+                  dpg_tpcpidqa::Sign,
+                  dpg_tpcpidqa::NSigmaTpc,
+                  dpg_tpcpidqa::DedxExpected,
+                  dpg_tpcpidqa::DedxDiff,
+                  dpg_tpcpidqa::ExpSigma,
+                  dpg_tpcpidqa::NSigmaTof)
 } // namespace o2::aod
 
 #endif // DPG_TASKS_AOTTRACK_PID_TPC_TREECREATORPIDTPCQA_H_