Skip to content

Commit 0a40d94

Browse files
authored
Fix bug on V0s (#1994)
1 parent 3f2ae54 commit 0a40d94

1 file changed

Lines changed: 5 additions & 2 deletions

File tree

EventFiltering/PWGHF/HFFilter.cxx

Lines changed: 5 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -241,6 +241,7 @@ struct HfFilter { // Main struct for HF triggers
241241
using BigTracksPID = soa::Filtered<soa::Join<aod::Tracks, aod::TracksExtra, aod::TracksDCA, aod::TrackSelection, aod::pidTPCFullPi, aod::pidTOFFullPi, aod::pidTPCFullKa, aod::pidTOFFullKa, aod::pidTPCFullPr, aod::pidTOFFullPr>>;
242242

243243
Preslice<aod::HfTrackAssoc> trackIndicesPerCollision = aod::hf_track_association::collisionId;
244+
Preslice<aod::V0Datas> v0sPerCollision = aod::v0data::collisionId;
244245
Preslice<aod::Hf2Prongs> hf2ProngPerCollision = aod::hf_track_association::collisionId;
245246
Preslice<aod::Hf3Prongs> hf3ProngPerCollision = aod::hf_track_association::collisionId;
246247

@@ -483,7 +484,8 @@ struct HfFilter { // Main struct for HF triggers
483484
} // end loop over tracks
484485

485486
// 2-prong with Gamma (conversion photon)
486-
for (auto& gamma : theV0s) {
487+
auto v0sThisCollision = theV0s.sliceBy(v0sPerCollision, thisCollId);
488+
for (auto& gamma : v0sThisCollision) {
487489
if (!keepEvent[kGammaCharm2P] && (isCharmTagged || isBeautyTagged) && (TESTBIT(selD0, 0) || (TESTBIT(selD0, 1)))) {
488490
float V0CosinePA = gamma.v0cosPA(collision.posX(), collision.posY(), collision.posZ());
489491
bool isGamma = isSelectedGamma(gamma, V0CosinePA, activateQA, hGammaSelected, hGammaEtaBefore, hGammaEtaAfter, hGammaArmPodBefore, hGammaArmPodAfter);
@@ -717,7 +719,8 @@ struct HfFilter { // Main struct for HF triggers
717719
} // end loop over tracks
718720

719721
// 3-prong with Gamma (conversion photon)
720-
for (auto& gamma : theV0s) {
722+
auto v0sThisCollision = theV0s.sliceBy(v0sPerCollision, thisCollId);
723+
for (auto& gamma : v0sThisCollision) {
721724
if (!keepEvent[kGammaCharm3P] && (isCharmTagged[kDs - 1] || isBeautyTagged[kDs - 1]) && (TESTBIT(is3ProngInMass[kDs - 1], 0) || TESTBIT(is3ProngInMass[kDs - 1], 1))) {
722725
float V0CosinePA = gamma.v0cosPA(collision.posX(), collision.posY(), collision.posZ());
723726
bool isGamma = isSelectedGamma(gamma, V0CosinePA, activateQA, hGammaSelected, hGammaEtaBefore, hGammaEtaAfter, hGammaArmPodBefore, hGammaArmPodAfter);

0 commit comments

Comments
 (0)