Follow MC single track and decays reco status

Author: shahor02

Detectors/GlobalTrackingWorkflow/study/CMakeLists.txt

@@ -9,7 +9,7 @@
 # granted to it by virtue of its status as an Intergovernmental Organization
 # or submit itself to any jurisdiction.
 
-# add_compile_options(-O0 -g -fPIC)
+add_compile_options(-O0 -g -fPIC)
 
 o2_add_library(GlobalTrackingStudy
                SOURCES src/TPCTrackStudy.cxx
@@ -21,6 +21,8 @@ o2_add_library(GlobalTrackingStudy
                        src/DumpTracks.cxx
                        src/V0Ext.cxx
                        src/TrackInfoExt.cxx
+                       src/TrackMCStudyConfig.cxx
+                       src/TrackMCStudyTypes.cxx
                PUBLIC_LINK_LIBRARIES O2::GlobalTracking
                                      O2::GlobalTrackingWorkflowReaders
                                      O2::GlobalTrackingWorkflowHelpers
@@ -33,6 +35,8 @@ o2_target_root_dictionary(
   GlobalTrackingStudy
   HEADERS include/GlobalTrackingStudy/V0Ext.h
           include/GlobalTrackingStudy/TrackInfoExt.h
+          include/GlobalTrackingStudy/TrackMCStudyConfig.h
+          include/GlobalTrackingStudy/TrackMCStudyTypes.h
 )
 
 o2_add_executable(study-workflow

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

@@ -12,17 +12,15 @@
 #ifndef O2_TRACKING_STUDY_H
 #define O2_TRACKING_STUDY_H
 
-#include "ReconstructionDataFormats/GlobalTrackID.h"
-#include "Framework/Task.h"
 #include "Framework/DataProcessorSpec.h"
-#include "ReconstructionDataFormats/Track.h"
-#include "MathUtils/detail/Bracket.h"
-#include "DataFormatsTPC/ClusterNative.h"
+#include "Framework/Task.h"
+#include "ReconstructionDataFormats/GlobalTrackID.h"
 
 namespace o2::trackstudy
 {
+
 /// create a processor spec
-o2::framework::DataProcessorSpec getTrackMCStudySpec(o2::dataformats::GlobalTrackID::mask_t srcTracks, o2::dataformats::GlobalTrackID::mask_t srcClus, bool checkMatching);
+o2::framework::DataProcessorSpec getTrackMCStudySpec(o2::dataformats::GlobalTrackID::mask_t srcTracks, o2::dataformats::GlobalTrackID::mask_t srcClus);
 
 } // namespace o2::trackstudy
 

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

@@ -0,0 +1,34 @@
+// Copyright 2019-2020 CERN and copyright holders of ALICE O2.
+// See https://alice-o2.web.cern.ch/copyright for details of the copyright holders.
+// All rights not expressly granted are reserved.
+//
+// This software is distributed under the terms of the GNU General Public
+// License v3 (GPL Version 3), copied verbatim in the file "COPYING".
+//
+// In applying this license CERN does not waive the privileges and immunities
+// granted to it by virtue of its status as an Intergovernmental Organization
+// or submit itself to any jurisdiction.
+
+#ifndef O2_TRACKING_STUDY_CONFIG_H
+#define O2_TRACKING_STUDY_CONFIG_H
+#include "CommonUtils/ConfigurableParam.h"
+#include "CommonUtils/ConfigurableParamHelper.h"
+
+namespace o2::trackstudy
+{
+struct TrackMCStudyConfig : o2::conf::ConfigurableParamHelper<TrackMCStudyConfig> {
+  float minPt = 0.05;
+  float maxTgl = 1.2;
+  float minPtMC = 0.05;
+  float maxTglMC = 1.2;
+  float minRMC = 33.;
+  float maxPosTglMC = 2.;
+  float maxPVZOffset = 15.;
+  float decayMotherMaxT = 0.1; // max TOF in ns for mother particles to study
+  bool requireITSorTPCTrackRefs = true;
+  int decayPDG[5] = {310, 3122, -1, -1, -1}; // decays to study, must end by -1
+  O2ParamDef(TrackMCStudyConfig, "trmcconf");
+};
+} // namespace o2::trackstudy
+
+#endif

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

@@ -0,0 +1,68 @@
+// Copyright 2019-2020 CERN and copyright holders of ALICE O2.
+// See https://alice-o2.web.cern.ch/copyright for details of the copyright holders.
+// All rights not expressly granted are reserved.
+//
+// This software is distributed under the terms of the GNU General Public
+// License v3 (GPL Version 3), copied verbatim in the file "COPYING".
+//
+// In applying this license CERN does not waive the privileges and immunities
+// granted to it by virtue of its status as an Intergovernmental Organization
+// or submit itself to any jurisdiction.
+
+#ifndef O2_TRACKING_STUDY_TYPES_H
+#define O2_TRACKING_STUDY_TYPES_H
+#include "ReconstructionDataFormats/GlobalTrackID.h"
+#include "ReconstructionDataFormats/VtxTrackIndex.h"
+#include "ReconstructionDataFormats/Track.h"
+#include "SimulationDataFormat/MCCompLabel.h"
+#include "Framework/Task.h"
+
+namespace o2::trackstudy
+{
+struct MCTrackInfo {
+  o2::track::TrackPar track{};
+  o2::MCCompLabel label{};
+  float occTPC = -1.f;
+  int bcInTF = -1;
+  int pdg = 0;
+  int pdgParent = 0;
+  int16_t nTPCCl = 0;
+  int16_t nTPCClShared = 0;
+  uint8_t minTPCRow = -1;
+  uint8_t maxTPCRow = 0;
+  int8_t nITSCl = 0;
+  int8_t pattITSCl = 0;
+  ClassDefNV(MCTrackInfo, 1);
+};
+
+struct RecTrack {
+  o2::track::TrackParCov track{};
+  o2::dataformats::VtxTrackIndex gid{};
+  uint8_t nClITS = 0;
+  uint8_t nClTPC = 0;
+  uint8_t pattITS = 0;
+  int8_t lowestPadRow = -1;
+  bool isFake = false;
+  ClassDefNV(RecTrack, 1);
+};
+
+struct TrackFamily { // set of tracks related to the same MC label
+  MCTrackInfo mcTrackInfo{};
+  std::vector<RecTrack> recTracks{};
+  int8_t entITS = -1;
+  int8_t entTPC = -1;
+  int8_t entITSTPC = -1;
+  bool contains(const o2::dataformats::VtxTrackIndex& ref) const
+  {
+    for (const auto& tr : recTracks) {
+      if (ref == tr.gid) {
+        return true;
+      }
+    }
+    return false;
+  }
+
+  ClassDefNV(TrackFamily, 1);
+};
+} // namespace o2::trackstudy
+#endif

Detectors/GlobalTrackingWorkflow/study/src/GlobalTrackingStudyLinkDef.h

@@ -21,5 +21,12 @@
 #pragma link C++ class std::vector < o2::dataformats::TrackInfoExt> + ;
 #pragma link C++ class std::vector < o2::dataformats::ProngInfoExt> + ;
 #pragma link C++ class std::vector < o2::dataformats::V0Ext> + ;
-
+#pragma link C++ class o2::trackstudy::TrackMCStudyConfig + ;
+#pragma link C++ class o2::conf::ConfigurableParamHelper < o2::trackstudy::TrackMCStudyConfig> + ;
+#pragma link C++ class o2::trackstudy::RecTrack + ;
+#pragma link C++ class std::vector < o2::trackstudy::RecTrack> + ;
+#pragma link C++ class o2::trackstudy::TrackFamily + ;
+#pragma link C++ class std::vector < o2::trackstudy::TrackFamily> + ;
+#pragma link C++ class o2::trackstudy::MCTrackInfo + ;
+#pragma link C++ class std::vector < o2::trackstudy::MCTrackInfo> + ;
 #endif