Skip to content

Commit

Permalink
[PWGDQ] TableMakerMC CheckSignal set checkSources true (#7621)
Browse files Browse the repository at this point in the history
* [PWGDQ] TableMakerMC CheckSignal set checkSources true

* [PWGDQ] dqEfficiency_withAssoc CheckSignal set checkSources true
  • Loading branch information
feisenhu authored Sep 9, 2024
1 parent 548ecc7 commit 40f8692
Show file tree
Hide file tree
Showing 2 changed files with 12 additions and 12 deletions.
4 changes: 2 additions & 2 deletions PWGDQ/TableProducer/tableMakerMC_withAssoc.cxx
Original file line number Diff line number Diff line change
Expand Up @@ -356,9 +356,9 @@ struct TableMakerMC {
bool checked = false;
if constexpr (soa::is_soa_filtered_v<aod::McParticles>) {
auto mctrack_raw = mcTracks.rawIteratorAt(mctrack.globalIndex());
checked = sig.CheckSignal(false, mctrack_raw);
checked = sig.CheckSignal(true, mctrack_raw);
} else {
checked = sig.CheckSignal(false, mctrack);
checked = sig.CheckSignal(true, mctrack);
}
if (checked) {
mcflags |= (uint16_t(1) << i);
Expand Down
20 changes: 10 additions & 10 deletions PWGDQ/Tasks/dqEfficiency_withAssoc.cxx
Original file line number Diff line number Diff line change
Expand Up @@ -506,7 +506,7 @@ struct AnalysisTrackSelection {
if (filterMap > 0) {
for (auto sig = fMCSignals.begin(); sig != fMCSignals.end(); sig++, isig++) {
if (track.has_reducedMCTrack()) {
if ((*sig).CheckSignal(false, track.reducedMCTrack())) {
if ((*sig).CheckSignal(true, track.reducedMCTrack())) {
mcDecision |= (uint32_t(1) << isig);
}
}
Expand Down Expand Up @@ -774,7 +774,7 @@ struct AnalysisMuonSelection {
for (auto sig = fMCSignals.begin(); sig != fMCSignals.end(); sig++, isig++) {
if constexpr ((TMuonFillMap & VarManager::ObjTypes::ReducedMuon) > 0) {
if (track.has_reducedMCTrack()) {
if ((*sig).CheckSignal(false, track.reducedMCTrack())) {
if ((*sig).CheckSignal(true, track.reducedMCTrack())) {
mcDecision |= (uint32_t(1) << isig);
}
}
Expand Down Expand Up @@ -1439,7 +1439,7 @@ struct AnalysisSameEventPairing {
mcDecision = 0;
for (auto sig = fRecMCSignals.begin(); sig != fRecMCSignals.end(); sig++, isig++) {
if (t1.has_reducedMCTrack() && t2.has_reducedMCTrack()) {
if ((*sig).CheckSignal(false, t1.reducedMCTrack(), t2.reducedMCTrack())) {
if ((*sig).CheckSignal(true, t1.reducedMCTrack(), t2.reducedMCTrack())) {
mcDecision |= (uint32_t(1) << isig);
}
}
Expand Down Expand Up @@ -1489,7 +1489,7 @@ struct AnalysisSameEventPairing {
mcDecision = 0;
for (auto sig = fRecMCSignals.begin(); sig != fRecMCSignals.end(); sig++, isig++) {
if (t1.has_reducedMCTrack() && t2.has_reducedMCTrack()) {
if ((*sig).CheckSignal(false, t1.reducedMCTrack(), t2.reducedMCTrack())) {
if ((*sig).CheckSignal(true, t1.reducedMCTrack(), t2.reducedMCTrack())) {
mcDecision |= (uint32_t(1) << isig);
}
}
Expand Down Expand Up @@ -1651,9 +1651,9 @@ struct AnalysisSameEventPairing {
bool checked = false;
/*if constexpr (soa::is_soa_filtered_v<TTracksMC>) {
auto mctrack_raw = groupedMCTracks.rawIteratorAt(mctrack.globalIndex());
checked = sig.CheckSignal(false, mctrack_raw);
checked = sig.CheckSignal(true, mctrack_raw);
} else {*/
checked = sig.CheckSignal(false, mctrack);
checked = sig.CheckSignal(true, mctrack);
//}
if (checked) {
fHistMan->FillHistClass(Form("MCTruthGen_%s", sig.GetName()), VarManager::fgValues);
Expand All @@ -1672,7 +1672,7 @@ struct AnalysisSameEventPairing {
if (sig.GetNProngs() != 2) { // NOTE: 2-prong signals required here
continue;
}
if (sig.CheckSignal(false, t1_raw, t2_raw)) {
if (sig.CheckSignal(true, t1_raw, t2_raw)) {
VarManager::FillPairMC(t1, t2);
fHistMan->FillHistClass(Form("MCTruthGenPair_%s", sig.GetName()), VarManager::fgValues);
}
Expand Down Expand Up @@ -1961,7 +1961,7 @@ struct AnalysisDileptonTrack {
mcDecision = 0;
isig = 0;
for (auto sig = fRecMCSignals.begin(); sig != fRecMCSignals.end(); sig++, isig++) {
if ((*sig).CheckSignal(false, lepton1MC, lepton2MC, trackMC)) {
if ((*sig).CheckSignal(true, lepton1MC, lepton2MC, trackMC)) {
mcDecision |= (uint32_t(1) << isig);
}
}
Expand All @@ -1982,7 +1982,7 @@ struct AnalysisDileptonTrack {
mcDecision = 0;
isig = 0;
for (auto sig = fRecMCSignals.begin(); sig != fRecMCSignals.end(); sig++, isig++) {
if ((*sig).CheckSignal(false, lepton1MC, lepton2MC, trackMC)) {
if ((*sig).CheckSignal(true, lepton1MC, lepton2MC, trackMC)) {
mcDecision |= (uint32_t(1) << isig);
}
}
Expand Down Expand Up @@ -2070,7 +2070,7 @@ struct AnalysisDileptonTrack {
// TODO: Use the mcReducedFlags to select signals
isig = 0;
for (auto& sig : fGenMCSignals) {
if (sig.CheckSignal(false, track)) {
if (sig.CheckSignal(true, track)) {
fHistMan->FillHistClass(fHistNamesMCgen[isig++], VarManager::fgValues);
}
}
Expand Down

0 comments on commit 40f8692

Please sign in to comment.