diff --git a/Common/Tools/TrackPropagationModule.h b/Common/Tools/TrackPropagationModule.h index c49441fe311..ececd68b5ae 100644 --- a/Common/Tools/TrackPropagationModule.h +++ b/Common/Tools/TrackPropagationModule.h @@ -80,6 +80,8 @@ struct TrackPropagationConfigurables : o2::framework::ConfigurableGroup { o2::framework::Configurable trackTunerConfigSource{"trackTunerConfigSource", aod::track_tuner::InputString, "1: input string; 2: TrackTuner Configurables"}; o2::framework::Configurable 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: =|=)"}; 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 useManualPIDForTracking{"useManualPIDForTracking", false, "use manual pid in tracking. (e.g. assume all tracks to be electron)"}; + o2::framework::Configurable 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);