diff --git a/PWGCF/FemtoDream/FemtoDreamTrackSelection.h b/PWGCF/FemtoDream/FemtoDreamTrackSelection.h index a4f575e976e..34459e08f4b 100644 --- a/PWGCF/FemtoDream/FemtoDreamTrackSelection.h +++ b/PWGCF/FemtoDream/FemtoDreamTrackSelection.h @@ -59,7 +59,8 @@ enum TrackContainerPosition { class FemtoDreamTrackSelection : public FemtoDreamObjectSelection { public: - FemtoDreamTrackSelection() : nPtMinSel(0), + FemtoDreamTrackSelection() : nRejectNotPropagatedTracks(false), + nPtMinSel(0), nPtMaxSel(0), nEtaSel(0), nTPCnMinSel(0), @@ -193,7 +194,13 @@ class FemtoDreamTrackSelection : public FemtoDreamObjectSelection 0 && std::abs(dca) < dcaMin) { return false; } - if (std::abs(dca) > 1e3) { + if (nRejectNotPropagatedTracks && std::abs(dca) > 1e3) { return false; } diff --git a/PWGCF/FemtoDream/FemtoDreamV0Selection.h b/PWGCF/FemtoDream/FemtoDreamV0Selection.h index 59928e3a029..76e2e36f2bb 100644 --- a/PWGCF/FemtoDream/FemtoDreamV0Selection.h +++ b/PWGCF/FemtoDream/FemtoDreamV0Selection.h @@ -165,6 +165,15 @@ class FemtoDreamV0Selection : public FemtoDreamObjectSelection ConfEvtOfflineCheck{"ConfEvtOfflineCheck", false, "Evt sel: check for offline selection"}; Configurable ConfStoreV0{"ConfStoreV0", true, "True: store V0 table"}; - + Configurable ConfRejectNotPropagatedTracks{"ConfRejectNotPropagatedTracks", false, "True: reject not propagated tracks"}; FemtoDreamV0Selection v0Cuts; /// \todo Labeled array (see Track-Track task) @@ -152,6 +152,8 @@ struct femtoDreamProducerTaskV0Only { v0Cuts.setChildPIDSpecies(femtoDreamV0Selection::kNegTrack, ConfV0DaughTPIDspecies); v0Cuts.init(&qaRegistry); v0Cuts.setInvMassLimits(ConfInvMassLowLimit, ConfInvMassUpLimit); + v0Cuts.setChildRejectNotPropagatedTracks(femtoDreamV0Selection::kPosTrack, ConfRejectNotPropagatedTracks); + v0Cuts.setChildRejectNotPropagatedTracks(femtoDreamV0Selection::kNegTrack, ConfRejectNotPropagatedTracks); } }