[Common] add an option to use manual pidForTracking in TrackPropagationModule.h

Author: dsekihat

Common/Tools/TrackPropagationModule.h

@@ -80,6 +80,8 @@ struct TrackPropagationConfigurables : o2::framework::ConfigurableGroup {
   o2::framework::Configurable<int> trackTunerConfigSource{"trackTunerConfigSource", aod::track_tuner::InputString, "1: input string; 2: TrackTuner Configurables"};
   o2::framework::Configurable<std::string> trackTunerParams{"trackTunerParams", "debugInfo=0|updateTrackDCAs=1|updateTrackCovMat=1|updateCurvature=0|updateCurvatureIU=0|updatePulls=0|isInputFileFromCCDB=1|pathInputFile=Users/m/mfaggin/test/inputsTrackTuner/PbPb2022|nameInputFile=trackTuner_DataLHC22sPass5_McLHC22l1b2_run529397.root|pathFileQoverPt=Users/h/hsharma/qOverPtGraphs|nameFileQoverPt=D0sigma_Data_removal_itstps_MC_LHC22b1b.root|usePvRefitCorrections=0|qOverPtMC=-1.|qOverPtData=-1.", "TrackTuner parameter initialization (format: <name>=<value>|<name>=<value>)"};
   o2::framework::ConfigurableAxis axisPtQA{"axisPtQA", {o2::framework::VARIABLE_WIDTH, 0.0f, 0.1f, 0.2f, 0.3f, 0.4f, 0.5f, 0.6f, 0.7f, 0.8f, 0.9f, 1.0f, 1.1f, 1.2f, 1.3f, 1.4f, 1.5f, 1.6f, 1.7f, 1.8f, 1.9f, 2.0f, 2.2f, 2.4f, 2.6f, 2.8f, 3.0f, 3.2f, 3.4f, 3.6f, 3.8f, 4.0f, 4.4f, 4.8f, 5.2f, 5.6f, 6.0f, 6.5f, 7.0f, 7.5f, 8.0f, 9.0f, 10.0f, 11.0f, 12.0f, 13.0f, 14.0f, 15.0f, 17.0f, 19.0f, 21.0f, 23.0f, 25.0f, 30.0f, 35.0f, 40.0f, 50.0f}, "pt axis for QA histograms"};
+  o2::framework::Configurable<bool> useManualPIDForTracking{"useManualPIDForTracking", false, "use manual pid in tracking. (e.g. assume all tracks to be electron)"};
+  o2::framework::Configurable<uint32_t> manualPIDForTracking{"manualPIDForTracking", 0, "manual pidForTracking. 0: electron, 1: muon, 2: pion, 3: kaon, 4: proton, 5: deuteron, 6: triton, 7: helium3, 8: alpha"};
 };
 
 class TrackPropagationModule
@@ -261,7 +263,7 @@ class TrackPropagationModule
         }
         setTrackParCov(track, mTrackParCov);
         if (cGroup.useTrkPid.value) {
-          mTrackParCov.setPID(track.pidForTracking());
+          mTrackParCov.setPID(cGroup.useManualPIDForTracking.value ? cGroup.manualPIDForTracking.value : track.pidForTracking());
         }
       } else {
         if (fillTracksDCA) {
@@ -270,7 +272,7 @@ class TrackPropagationModule
         }
         setTrackPar(track, mTrackPar);
         if (cGroup.useTrkPid.value) {
-          mTrackPar.setPID(track.pidForTracking());
+          mTrackPar.setPID(cGroup.useManualPIDForTracking.value ? cGroup.manualPIDForTracking.value : track.pidForTracking());
         }
       }
       // auto trackParCov = getTrackParCov(track);