Skip to content

Commit 0b0313f

Browse files
authored
PWGDQ: quick fix filter bit for e/mu (#4396)
1 parent 60f34a5 commit 0b0313f

File tree

1 file changed

+11
-8
lines changed

1 file changed

+11
-8
lines changed

Analysis/Core/include/Analysis/VarManager.h

Lines changed: 11 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -379,18 +379,21 @@ void VarManager::FillPair(T const& t1, T const& t2, float* values)
379379
if (!values)
380380
values = fgValues;
381381

382-
float mass1;
383-
float mass2;
382+
float mass1 = fgkElectronMass;
383+
float mass2 = fgkElectronMass;
384384

385-
if (t1.filteringFlags() > 0)
386-
mass1 = fgkElectronMass;
387-
else
388-
mass1 = fgkMuonMass;
385+
bool isMuon1 = t1.filteringFlags() & (1 << 0);
386+
bool isMuon2 = t2.filteringFlags() & (1 << 0);
389387

390-
if (t1.filteringFlags() > 0)
391-
mass2 = fgkElectronMass;
388+
if (isMuon1)
389+
mass1 = fgkMuonMass;
392390
else
391+
mass1 = fgkElectronMass;
392+
393+
if (isMuon2)
393394
mass2 = fgkMuonMass;
395+
else
396+
mass2 = fgkElectronMass;
394397

395398
ROOT::Math::PtEtaPhiMVector v1(t1.pt(), t1.eta(), t1.phi(), mass1);
396399
ROOT::Math::PtEtaPhiMVector v2(t2.pt(), t2.eta(), t2.phi(), mass2);

0 commit comments

Comments
 (0)