Optionally extract TPC clusters MC truth resolution

Author: shahor02

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

@@ -15,12 +15,13 @@
 #include "Framework/DataProcessorSpec.h"
 #include "Framework/Task.h"
 #include "ReconstructionDataFormats/GlobalTrackID.h"
+#include "TPCCalibration/CorrectionMapsLoader.h"
 
 namespace o2::trackstudy
 {
 
 /// create a processor spec
-o2::framework::DataProcessorSpec getTrackMCStudySpec(o2::dataformats::GlobalTrackID::mask_t srcTracks, o2::dataformats::GlobalTrackID::mask_t srcClus);
+o2::framework::DataProcessorSpec getTrackMCStudySpec(o2::dataformats::GlobalTrackID::mask_t srcTracks, o2::dataformats::GlobalTrackID::mask_t srcClus, const o2::tpc::CorrectionMapsLoaderGloOpts& sclOpts);
 
 } // namespace o2::trackstudy
 

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

@@ -26,6 +26,11 @@ struct TrackMCStudyConfig : o2::conf::ConfigurableParamHelper<TrackMCStudyConfig
   float maxPVZOffset = 15.;
   float decayMotherMaxT = 1.0f; // max TOF in ns for mother particles to study
   bool requireITSorTPCTrackRefs = true;
+  bool requireTopBottomRefs = false;
+  int minTPCRefsToExtractClRes = 4;
+  float rejectClustersResStat = 0.6;
+  float maxTPCRefExtrap = 2; // max dX to extrapolate the track ref when extrapolating track true posions
+  float maxTRefExtrapErr = 0.005;
   int decayPDG[5] = {310, 3122, 411, 421, -1}; // decays to study, must end by -1
   O2ParamDef(TrackMCStudyConfig, "trmcconf");
 };

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

@@ -43,9 +43,12 @@ struct MCTrackInfo {
   int16_t nTPCClShared = 0;
   uint8_t minTPCRow = -1;
   uint8_t maxTPCRow = 0;
+  uint8_t maxTPCRowInner = 0; // highest row in the sector containing the lowest one
+  uint8_t minTPCRowSect = -1;
+  uint8_t maxTPCRowSect = -1;
   int8_t nITSCl = 0;
   int8_t pattITSCl = 0;
-  ClassDefNV(MCTrackInfo, 1);
+  ClassDefNV(MCTrackInfo, 2);
 };
 
 struct RecTrack {
@@ -109,6 +112,70 @@ struct TrackFamily { // set of tracks related to the same MC label
   ClassDefNV(TrackFamily, 1);
 };
 
+struct ClResTPC {
+  uint8_t sect = 0;
+  uint8_t row = 0;
+  uint8_t ncont = 0;
+  uint8_t flags = 0;
+  float snp = 0;
+  float tgl = 0;
+  float qmax = 0;
+  float qtot = 0;
+  float occ = 0;
+  std::array<float, 3> clPos{};
+  std::array<float, 3> below{};
+  std::array<float, 3> above{};
+
+  int getNExt() const { return below[0] > 1. + above[0] > 1.; }
+
+  float getDY() const { return clPos[1] - getYRef(); }
+  float getDZ() const { return clPos[2] - getZRef(); }
+
+  float getYRef() const
+  {
+    float y = 0;
+    int n = 0;
+    if (below[0] > 1.) {
+      y += below[1];
+      n++;
+    }
+    if (above[0] > 1.) {
+      y += above[1];
+      n++;
+    }
+    return n == 1 ? y : 0.5 * y;
+  }
+
+  float getZRef() const
+  {
+    float z = 0;
+    int n = 0;
+    if (below[0] > 1.) {
+      z += below[2];
+      n++;
+    }
+    if (above[0] > 1.) {
+      z += above[2];
+      n++;
+    }
+    return n == 1 ? z : 0.5 * z;
+  }
+
+  float getDXMin() const
+  {
+    float adxA = 1e9, adxB = 1e9;
+    if (above[0] > 1.) {
+      adxA = clPos[0] - above[0];
+    }
+    if (below[0] > 1.) {
+      adxB = clPos[0] - below[0];
+    }
+    return std::abs(adxA) < std::abs(adxB) ? adxA : adxB;
+  }
+
+  ClassDefNV(ClResTPC, 1);
+};
+
 struct RecPV {
   o2::dataformats::PrimaryVertex pv{};
   o2::MCEventLabel mcEvLbl{};

Detectors/GlobalTrackingWorkflow/study/src/GlobalTrackingStudyLinkDef.h

@@ -33,5 +33,6 @@
 #pragma link C++ class std::vector < o2::trackstudy::RecPV> + ;
 #pragma link C++ class o2::trackstudy::MCVertex + ;
 #pragma link C++ class std::vector < o2::trackstudy::MCVertex> + ;
+#pragma link C++ class o2::trackstudy::ClResTPC + ;
 
 #endif