Add more info on MC vertex (#15740)

Author: shahor02

Detectors/GlobalTrackingWorkflow/study/include/GlobalTrackingStudy/TrackMCStudyConfig.h

@@ -17,12 +17,12 @@
 namespace o2::trackstudy
 {
 struct TrackMCStudyConfig : o2::conf::ConfigurableParamHelper<TrackMCStudyConfig> {
-  float minPt = 0.05;
-  float maxTgl = 1.5;
-  float minPtMC = 0.05;
-  float maxTglMC = 1.5;
+  float minPt = 0.07;
+  float maxTgl = 2.3;
+  float minPtMC = 0.07;
+  float maxTglMC = 2.2;
   float maxRMC = 33.;
-  float maxPosTglMC = 2.;
+  float maxPosTglMC = 2.2;
   float maxPVZOffset = 15.;
   float decayMotherMaxT = 1.0f; // max TOF in ns for mother particles to study
   bool requireITSorTPCTrackRefs = true;
@@ -32,6 +32,7 @@ struct TrackMCStudyConfig : o2::conf::ConfigurableParamHelper<TrackMCStudyConfig
   int minTPCRefsToExtractClRes = 2;
   int nOccBinsDrift = 10; // number of bins for TPC max drift time, where we integrate the occupancies
   int nTBPerOccBin = 48;  // number of TB per occ bin
+  int nMinTPCRowSpan = 50; // number of covered TPC rows to consider it good for TPC tracking
   float rejectClustersResStat = 0.1;
   float maxTPCRefExtrap = 2;                   // max dX to extrapolate the track ref when extrapolating track true posions
   int minITSClForITSoutput = 7;                // create special ITS otput only for long enough tracks

Detectors/GlobalTrackingWorkflow/study/include/GlobalTrackingStudy/TrackMCStudyTypes.h

@@ -313,11 +313,18 @@ struct MCVertex {
 
   std::array<float, 3> pos{0., 0., -1999.f};
   float ts = 0;
-  int nTrackSel = 0; // number of selected MC charged tracks
+  int nTrackSel = 0;       // number of selected MC charged tracks with nominal cuts
+  int nTrackSel100 = 0;    // number of selected MC charged tracks with nominal cuts + pt>100 MeV
+  int nTrackSelRCBL0 = 0;  // number of tracks with at least 4 innermost ITS layers
+  int nTrackSelRCBL1 = 0;  // number of tracks with at least 4 innermost ITS layers and enough TPC rows + primary
+  int nTrackSelRCBL0P = 0; // number of tracks with at least 4 innermost ITS layers
+  int nTrackSelRCBL1P = 0; // number of tracks with at least 4 innermost ITS layers and enough TPC rows + primary
+  int nTrackRecRCBL0 = 0;  // number of reconstructed with RCBL0 condition
+  int nTrackRecRCBL1 = 0;  // number of reconstructed with RCBL1 condition
   int ID = -1;
   std::vector<RecPV> recVtx{};
   std::vector<float> occTPCV{};
-  ClassDefNV(MCVertex, 2);
+  ClassDefNV(MCVertex, 3);
 };
 
 } // namespace o2::trackstudy

Detectors/GlobalTrackingWorkflow/study/src/TrackMCStudy.cxx

@@ -953,6 +953,31 @@ void TrackMCStudy::fillMCClusterInfo(const o2::globaltracking::RecoContainer& re
       mctr.pattITSCl |= 0x1 << o2::itsmft::ChipMappingITS::getLayer(ITSClusters[icl].getChipID());
     }
   }
+
+  for (auto& entry : mSelMCTracks) { // count ITS reconstructable tracks
+    const auto& trackFam = entry.second;
+    const auto& mctr = trackFam.mcTrackInfo;
+    if (mctr.getLowestITSLayer() == 0 && mctr.getNITSClusCont() > 3) { // has 4 innermost layers
+      auto& mcev = mMCVtVec[mctr.label.getEventID()];
+      mcev.nTrackSelRCBL0++;
+      if (mctr.isPrimary()) {
+        mcev.nTrackSelRCBL0P++;
+      }
+      if (trackFam.entITSFound >= 0) {
+        mcev.nTrackRecRCBL0++;
+      }
+
+      if (mctr.maxTPCRow - mctr.minTPCRow >= params.nMinTPCRowSpan) {
+        mcev.nTrackSelRCBL1++;
+        if (mctr.isPrimary()) {
+          mcev.nTrackSelRCBL1P++;
+        }
+        if (trackFam.entITSTPC >= 0) {
+          mcev.nTrackRecRCBL1++;
+        }
+      }
+    }
+  }
 }
 
 bool TrackMCStudy::propagateToRefX(o2::track::TrackParCov& trcTPC, o2::track::TrackParCov& trcITS)
@@ -1183,6 +1208,9 @@ bool TrackMCStudy::addMCParticle(const MCTrack& mcPart, const o2::MCCompLabel& l
   }
   if (mcPart.isPrimary() && mcReader.getNEvents(lb.getSourceID()) == mMCVtVec.size()) {
     mMCVtVec[lb.getEventID()].nTrackSel++;
+    if (mcPart.GetPt() > 0.1) {
+      mMCVtVec[lb.getEventID()].nTrackSel100++;
+    }
   }
   if (mVerbose > 1) {
     LOGP(info, "Adding charged MC pdg={} {} ", mcPart.GetPdgCode(), lb.asString());