@@ -249,6 +249,8 @@ struct nucleiSpectra {
249249 Configurable<bool > cfgCutOnReconstructedRapidity{" cfgCutOnReconstructedRapidity" , false , " Cut on reconstructed rapidity" };
250250 Configurable<float > cfgCutNclusITS{" cfgCutNclusITS" , 5 , " Minimum number of ITS clusters" };
251251 Configurable<float > cfgCutNclusTPC{" cfgCutNclusTPC" , 70 , " Minimum number of TPC clusters" };
252+ Configurable<float > cfgCutPtMinTree{" cfgCutPtMinTree" , 0 .2f , " Minimum track transverse momentum for tree saving" };
253+ Configurable<float > cfgCutPtMaxTree{" cfgCutPtMaxTree" , 15 .0f , " Maximum track transverse momentum for tree saving" };
252254
253255 Configurable<LabeledArray<double >> cfgMomentumScalingBetheBloch{" cfgMomentumScalingBetheBloch" , {nuclei::bbMomScalingDefault[0 ], 5 , 2 , nuclei::names, nuclei::chargeLabelNames}, " TPC Bethe-Bloch momentum scaling for light nuclei" };
254256 Configurable<LabeledArray<double >> cfgBetheBlochParams{" cfgBetheBlochParams" , {nuclei::betheBlochDefault[0 ], 5 , 6 , nuclei::names, nuclei::betheBlochParNames}, " TPC Bethe-Bloch parameterisation for light nuclei" };
@@ -525,7 +527,6 @@ struct nucleiSpectra {
525527 track.itsChi2NCl () > 36 .f ) {
526528 continue ;
527529 }
528-
529530 // temporary fix: tpcInnerParam() returns the momentum in all the software tags before
530531 bool heliumPID = track.pidForTracking () == o2::track::PID::Helium3 || track.pidForTracking () == o2::track::PID::Alpha;
531532 float correctedTpcInnerParam = (heliumPID && cfgCompensatePIDinTracking) ? track.tpcInnerParam () / 2 : track.tpcInnerParam ();
@@ -693,6 +694,10 @@ struct nucleiSpectra {
693694 computeEventPlane (collision.qvecBPosIm (), collision.qvecBPosRe ()),
694695 collision.multTPC ()});
695696 }
697+ if (fillTree) {
698+ if (track.pt () < cfgCutPtMinTree || track.pt () > cfgCutPtMaxTree)
699+ continue ;
700+ }
696701 nuclei::candidates.emplace_back (NucleusCandidate{
697702 static_cast <int >(track.globalIndex ()), static_cast <int >(track.collisionId ()), (1 - 2 * iC) * mTrackParCov .getPt (), mTrackParCov .getEta (), mTrackParCov .getPhi (),
698703 correctedTpcInnerParam, beta, collision.posZ (), dcaInfo[0 ], dcaInfo[1 ], track.tpcSignal (), track.itsChi2NCl (), track.tpcChi2NCl (),
0 commit comments