study

Signed-off-by: Felix Schlepper <[email protected]>

Author: f3sch

Detectors/Upgrades/ITS3/study/macros/PlotDCA.C

@@ -88,8 +88,9 @@ void PlotDCA(const char* dets = "all", const char* fName = "its3TrackStudy.root"
 
     for (int cis = 0; cis < GTrackID::NSources; ++cis) {
       const auto cdm = GTrackID::getSourceDetectorsMask(cis);
-      if (!cdm[GTrackID::ITS])
+      if (!cdm[GTrackID::ITS]) {
         continue; // keep same logic as original
+      }
 
       const std::string srcRaw = GTrackID::getSourceName(cis);
       const std::string src = SanitizeSourceName(srcRaw);
@@ -117,6 +118,9 @@ void PlotDCA(const char* dets = "all", const char* fName = "its3TrackStudy.root"
 
   for (int iEntry = 0; tree->LoadTree(iEntry) >= 0; ++iEntry) {
     tree->GetEntry(iEntry);
+    if (!mcTrk->isPrimary()) {
+      continue;
+    }
     auto pdg = std::abs(mcTrk->GetPdgCode());
     if (pdg != 11 && pdg != 211 && pdg != 321 && pdg != 2212) {
       continue;
@@ -129,9 +133,9 @@ void PlotDCA(const char* dets = "all", const char* fName = "its3TrackStudy.root"
     auto phiReco = trk->getPhi();
 
     for (int spe : {-1, pdg}) {
+      hMapDeltaPtVsPtAllLayers[spe][src]->Fill(ptGen, deltaPt);
       hMapDCAxyVsPtAllLayers[spe][src]->Fill(ptGen, dcaXY);
       hMapDCAzVsPtAllLayers[spe][src]->Fill(ptGen, dcaZ);
-      hMapDeltaPtVsPtAllLayers[spe][src]->Fill(ptGen, deltaPt);
     }
   }
 

Detectors/Upgrades/ITS3/study/src/TrackingStudy.cxx

@@ -310,11 +310,15 @@ void TrackingStudySpec::doDCAStudy(o2::globaltracking::RecoContainer& recoData)
       int idMin = vtref.getFirstEntryOfSource(is), idMax = idMin + vtref.getEntriesOfSource(is);
       for (int i = idMin; i < idMax; i++) {
         const auto vid = trackIndex[i];
+        if (!vid.isPVContributor()) {
+          continue;
+        }
         if (!selectITSTrack(vid, recoData)) {
           continue;
         }
 
-        // we fit each different sub-track type, that include ITS
+        // we fit each different sub-track type, that include ITS, e.g.
+        // ITS,ITS-TPC,ITS-TPC-TRD,ITS-TPC-TOF,ITS-TPC-TRD-TOF
         auto contributorsGID = recoData.getSingleDetectorRefs(vid);
         for (int cis = 0; cis < GTrackID::NSources && cis <= is; cis++) {
           const auto cdm = GTrackID::getSourceDetectorsMask(cis);
@@ -347,12 +351,13 @@ void TrackingStudySpec::doDCAStudy(o2::globaltracking::RecoContainer& recoData)
               ++nDCAFitsFail;
               continue;
             }
-            const auto& mcTrk = mcReader.getTrack(recoData.getTrackMCLabel(contributorsGID[cis]));
+            const auto& mcTrk = mcReader.getTrack(lbl);
             if (mcTrk == nullptr) {
               LOGP(fatal, "mcTrk is null did selection fail?");
             }
             stream << "mcTrk=" << *mcTrk
-                   << "dca2MC=" << dcaInfoMC;
+                   << "dca2MC=" << dcaInfoMC
+                   << "lbl=" << lbl;
           }
           stream << "\n";