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,7 @@
 #define ALICEO2_V0EXT_H
 
 #include "ReconstructionDataFormats/V0.h"
-#include "SimulationDataFormat/MCCompLabel.h"
+#include "SimulationDataFormat/MCTrack.h"
 
 namespace o2::dataformats
 {
@@ -44,6 +44,30 @@ struct V0Ext {
   ClassDefNV(V0Ext, 2);
 };
 
+enum PoolInfo : uint8_t {
+  Unset = (1 << 0),
+  Considered = (1 << 1), // considered (should be all not unassigned?)
+  Accepted = (1 << 2),   // accepted into pool
+  Unassigned = (1 << 3), // tracks that were not assigned to a collision
+};
+struct ProngMCInfoExt {
+  detectors::DetID::mask_t recoMask; // detector mask
+  o2::track::TrackPar trk;           // most global reco'd track
+  MCTrack mcTrk;
+  uint8_t 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(); }
+
+  ClassDefNV(V0MCExt, 1);
+};
+
 } // namespace o2::dataformats
 
 #endif

Detectors/GlobalTrackingWorkflow/study/src/GlobalTrackingStudyLinkDef.h

@@ -16,7 +16,9 @@
 #pragma link off all functions;
 
 #pragma link C++ class o2::dataformats::ProngInfoExt + ;
+#pragma link C++ class o2::dataformats::ProngMCInfoExt + ;
 #pragma link C++ class o2::dataformats::V0Ext + ;
+#pragma link C++ class o2::dataformats::V0MCExt + ;
 #pragma link C++ class o2::dataformats::TrackInfoExt + ;
 #pragma link C++ class std::vector < o2::dataformats::TrackInfoExt> + ;
 #pragma link C++ class std::vector < o2::dataformats::ProngInfoExt> + ;