Add workflow for ITS PID study (AliceO2Group#12478)

* Add workflow for ITS PID study * Update author * Update author in the header * Add MC matching + fix output name * Add custom BB resolution * Update PIDStudy.cxx Co-authored-by: Matteo Concas <[email protected]> --------- Co-authored-by: Matteo Concas <[email protected]>
mhemmer-cern · Dec 19, 2023 · d3ad989 · d3ad989
1 parent 08e4115
commit d3ad989
Show file tree

Hide file tree

Showing 7 changed files with 408 additions and 1 deletion.
diff --git a/Detectors/ITSMFT/ITS/postprocessing/studies/CMakeLists.txt b/Detectors/ITSMFT/ITS/postprocessing/studies/CMakeLists.txt
@@ -12,6 +12,7 @@
 o2_add_library(ITSPostprocessing
                SOURCES src/ImpactParameter.cxx
                        src/AvgClusSize.cxx
+                       src/PIDStudy.cxx
                        src/ITSStudiesConfigParam.cxx
                        src/TrackCheck.cxx
                        src/Helpers.cxx

diff --git a/Detectors/ITSMFT/ITS/postprocessing/studies/include/ITSStudies/ITSStudiesConfigParam.h b/Detectors/ITSMFT/ITS/postprocessing/studies/include/ITSStudies/ITSStudiesConfigParam.h
@@ -73,6 +73,15 @@ struct ITSAvgClusSizeParamConfig : public o2::conf::ConfigurableParamHelper<ITSA
   O2ParamDef(ITSAvgClusSizeParamConfig, "ITSAvgClusSizeParam");
 };
 
+struct PIDStudyParamConfig : public o2::conf::ConfigurableParamHelper<PIDStudyParamConfig> {
+  std::string outFileName = "its_PIDStudy.root";
+  // default: average 2023 from C. Sonnabend, Nov 2023: ([0.217553   4.02762    0.00850178 2.33324    0.880904  ])
+  // to-do: grab from CCDB when available
+  float mBBpars[5] = {0.217553, 4.02762, 0.00850178, 2.33324, 0.880904};
+  float mBBres = 0.07; // default: 7% resolution
+  O2ParamDef(PIDStudyParamConfig, "PIDStudyParam");
+};
+
 struct ITSImpactParameterParamConfig : public o2::conf::ConfigurableParamHelper<ITSImpactParameterParamConfig> {
   std::string outFileName = "its_ImpParameter.root";
   int minNumberOfContributors = 0;

diff --git a/Detectors/ITSMFT/ITS/postprocessing/studies/include/ITSStudies/PIDStudy.h b/Detectors/ITSMFT/ITS/postprocessing/studies/include/ITSStudies/PIDStudy.h
@@ -0,0 +1,36 @@
+// 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.
+
+/// \file PIDStudy.h
+/// \author Francesco Mazzaschi, [email protected]
+
+#ifndef O2_PID_STUDY_H
+#define O2_PID_STUDY_H
+
+#include "Framework/DataProcessorSpec.h"
+#include "ReconstructionDataFormats/GlobalTrackID.h"
+#include <Steer/MCKinematicsReader.h>
+
+namespace o2
+{
+namespace its
+{
+namespace study
+{
+
+using mask_t = o2::dataformats::GlobalTrackID::mask_t;
+
+o2::framework::DataProcessorSpec getPIDStudy(mask_t srcTracksMask, mask_t srcClustersMask, bool useMC, std::shared_ptr<o2::steer::MCKinematicsReader> kineReader);
+} // namespace study
+} // namespace its
+} // namespace o2
+
+#endif
diff --git a/Detectors/ITSMFT/ITS/postprocessing/studies/src/ITSStudiesConfigParam.cxx b/Detectors/ITSMFT/ITS/postprocessing/studies/src/ITSStudiesConfigParam.cxx
@@ -18,10 +18,12 @@ namespace its
 namespace study
 {
 static auto& sAvgClusSizeParamITS = o2::its::study::ITSAvgClusSizeParamConfig::Instance();
+static auto& sPIDStudyParamITS = o2::its::study::PIDStudyParamConfig::Instance();
 static auto& sCheckTracksParamsITS = o2::its::study::ITSCheckTracksParamConfig::Instance();
 static auto& sImpactParameterParamsITS = o2::its::study::ITSImpactParameterParamConfig::Instance();
 
 O2ParamImpl(o2::its::study::ITSAvgClusSizeParamConfig);
+O2ParamImpl(o2::its::study::PIDStudyParamConfig);
 O2ParamImpl(o2::its::study::ITSCheckTracksParamConfig);
 O2ParamImpl(o2::its::study::ITSImpactParameterParamConfig);
 

diff --git a/Detectors/ITSMFT/ITS/postprocessing/studies/src/ITSStudiesLinkDef.h b/Detectors/ITSMFT/ITS/postprocessing/studies/src/ITSStudiesLinkDef.h
@@ -16,8 +16,10 @@
 #pragma link off all functions;
 
 #pragma link C++ class o2::its::study::ITSAvgClusSizeParamConfig + ;
+#pragma link C++ class o2::its::study::PIDStudyParamConfig + ;
 #pragma link C++ class o2::its::study::ITSImpactParameterParamConfig + ;
 #pragma link C++ class o2::conf::ConfigurableParamHelper < o2::its::study::ITSAvgClusSizeParamConfig> + ;
+#pragma link C++ class o2::conf::ConfigurableParamHelper < o2::its::study::PIDStudyParamConfig> + ;
 #pragma link C++ class o2::conf::ConfigurableParamHelper < o2::its::study::ITSCheckTracksParamConfig> + ;
 #pragma link C++ class o2::conf::ConfigurableParamHelper < o2::its::study::ITSImpactParameterParamConfig> + ;
 #pragma link C++ function o2::its::studies::makeLogBinning + ;