Skip to content

Commit 7ec2f86

Browse files
authored
[PWGHF] Add eta selection in mlBasedTrackSelector (#18087)
1 parent 5f03bc3 commit 7ec2f86

1 file changed

Lines changed: 2 additions & 1 deletion

File tree

‎PWGHF/D2H/TableProducer/mlBasedTrackSelector.cxx‎

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -312,6 +312,7 @@ struct HfTrackSelectorTagSelTracks {
312312
Configurable<bool> testAcknowledgement{"testAcknowledgement", false, "test acknowledgement"};
313313
Configurable<bool> fillHistograms{"fillHistograms", true, "fill histograms"};
314314
Configurable<float> ptMinTrack{"ptMinTrack", 0.3f, "min. track pT entering the charm combinatorics"};
315+
Configurable<float> etaMaxTrack{"etaMaxTrack", 0.8f, "max. track eta entering the charm combinatorics"};
315316
Configurable<bool> enableTiming{"enableTiming", false, "fill hTiming with the CPU of the feature building and of each model evaluation (adds two clock reads per call)"};
316317
// D+ model
317318
Configurable<bool> applyMlDplus{"applyMlDplus", true, "evaluate the D+ track model"};
@@ -589,7 +590,7 @@ struct HfTrackSelectorTagSelTracks {
589590
registry.fill(HIST("hPtNoCuts"), track.pt());
590591
}
591592

592-
const bool passesQuality = track.pt() >= config.ptMinTrack && track.isGlobalTrackWoDCA();
593+
const bool passesQuality = track.pt() >= config.ptMinTrack && std::abs(track.eta()) <= config.etaMaxTrack && track.isGlobalTrackWoDCA();
593594
if (passesQuality && !scoreCollision && config.fillHistograms) {
594595
registry.fill(HIST("hPtQualityRejColl"), track.pt());
595596
}

0 commit comments

Comments
 (0)