Study: add v0 mc

Signed-off-by: Felix Schlepper <felix.schlepper@cern.ch>

Author: f3sch

Detectors/GlobalTrackingWorkflow/study/CMakeLists.txt

@@ -16,6 +16,7 @@ o2_add_library(GlobalTrackingStudy
                SOURCES src/TPCTrackStudy.cxx
                        src/TrackingStudy.cxx
                        src/SVStudy.cxx
+                       src/SVMCStudy.cxx
                        src/TrackMCStudy.cxx
                        src/TPCDataFilter.cxx
                        src/ITSOffsStudy.cxx
@@ -50,6 +51,11 @@ o2_add_executable(study-workflow
                   SOURCES src/sv-study-workflow.cxx
                   PUBLIC_LINK_LIBRARIES O2::GlobalTrackingStudy)
 
+o2_add_executable(study-workflow
+                  COMPONENT_NAME sv-mc
+                  SOURCES src/sv-mc-study-workflow.cxx
+                  PUBLIC_LINK_LIBRARIES O2::GlobalTrackingStudy)
+
 o2_add_executable(study-workflow
                   COMPONENT_NAME tpc-track
                   SOURCES src/tpc-track-study-workflow.cxx

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

@@ -0,0 +1,26 @@
+// 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_SV_GAMMA_STUDY_H
+#define O2_SV_GAMMA_STUDY_H
+
+#include "ReconstructionDataFormats/GlobalTrackID.h"
+#include "Framework/DataProcessorSpec.h"
+#include "TPCCalibration/CorrectionMapsLoader.h"
+
+namespace o2::svstudy
+{
+
+o2::framework::DataProcessorSpec getSVMCStudySpec(o2::dataformats::GlobalTrackID::mask_t srcTracks, o2::dataformats::GlobalTrackID::mask_t srcCls, const o2::tpc::CorrectionMapsLoaderGloOpts& sclOpts);
+
+} // namespace o2::svstudy
+
+#endif

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

@@ -15,7 +15,9 @@
 #define ALICEO2_V0EXT_H
 
 #include "ReconstructionDataFormats/V0.h"
+#include "ReconstructionDataFormats/PrimaryVertex.h"
 #include "SimulationDataFormat/MCCompLabel.h"
+#include "SimulationDataFormat/MCTrack.h"
 
 namespace o2::dataformats
 {
@@ -44,6 +46,69 @@ struct V0Ext {
   ClassDefNV(V0Ext, 2);
 };
 
+// Ordered enum: higher value = got further in the selection.
+// Use max-assign semantics so the deepest-reached stage is always recorded.
+enum PoolInfo : uint8_t {
+  Unset = 0,       // not yet processed
+  Unassigned,      // seen only in unassigned-track bucket (no collision match)
+  Considered,      // seen in vertex-associated loop
+  RejTPCMaxX,      // processTPCTrack: X position cut
+  RejTPCCorr,      // processTPCTrack: drift correction failed
+  RejTPCPhotonCls, // processTPCTrack: photon-tune min TPC clusters
+  RejTPCPhotonDPV, // processTPCTrack: photon-tune z-to-beam cut
+  RejTPCPhotonRD2, // processTPCTrack: photon-tune transverse radius cut
+  RejPVContrib,    // acceptTrack: PV contributor cut
+  RejDCAToPV,      // acceptTrack: DCA to PV too small
+  RejShortITS,     // buildT2V: short ITS-only track
+  Accepted,        // entered the track pool
+};
+struct ProngMCInfoExt {
+  detectors::DetID::mask_t recoMask; // detector mask
+  o2::track::TrackPar trk;           // most global reco'd track
+  MCTrack mcTrk;
+  PoolInfo poolStatus = PoolInfo::Unset;
+  ClassDefNV(ProngMCInfoExt, 1);
+};
+
+struct V0MCExt {
+  V0 v0; // reconstructed v0
+  MCTrack mcTrk;
+  std::array<ProngMCInfoExt, 2> prInfo{};
+  const ProngMCInfoExt& getPrInfo(int i) const { return prInfo[i]; }
+  bool hasBothProngs() const noexcept { return prInfo[0].recoMask.any() && prInfo[1].recoMask.any(); }
+  enum PairStatus : uint8_t {
+    Unpaired = 0, // never reached pair loop
+    NoOverlap,    // bracket overlap check failed
+    MaxPV,        // PV contributor cut
+    // checkV0 rejections, ordered by depth in the function:
+    RejTgl,         // |tgl_pos - tgl_neg| too large (TPC photon tune)
+    RejD2R,         // circle center distance vs radii sum (TPC photon tune)
+    RejDR,          // estimated conversion radius too large (TPC photon tune)
+    RejDCAFitter,   // DCAFitter found no candidate
+    RejMinR,        // V0 radius too close to beam line
+    RejCausality,   // causality check (V0 radius vs track inner radius)
+    RejPropagation, // propagation to PCA failed
+    RejPt,          // V0 pT too low
+    RejTgLambda,    // V0 tgLambda too large
+    RejHypothesis,  // no mass hypothesis matched
+    RejDCAToMV,     // DCA or cosPAXY to mean vertex failed
+    RejCosPA,       // no PV passed cosPA cut
+    Found,          // reconstructed V0 added to output
+  };
+  PairStatus pairStatus = Unpaired;
+
+  ClassDefNV(V0MCExt, 1);
+};
+
+struct V0MCFull {
+  V0 v0;
+  std::array<bool, 8> hypStatus{};
+  int pdg = -1;
+  o2::MCCompLabel label;
+
+  ClassDefNV(V0MCFull, 1);
+};
+
 } // namespace o2::dataformats
 
 #endif