13 static constexpr
double const MeVtoGeV = 1. / 1000.;
16 std::vector<int> initializeChamberIdxOrder() {
23 const std::vector<ChIdx> orderedChIndices{
32 std::vector<int> chamberIndexOrder(orderedChIndices.size());
34 for (
unsigned int i = 0;
i < orderedChIndices.size();
i++) {
35 chamberIndexOrder[
toInt(orderedChIndices[
i])] =
i;
37 return chamberIndexOrder;
42 static const std::vector<int> chamberIndexOrder = initializeChamberIdxOrder();
43 return (chamberIndexOrder[
toInt(
first->chamberIndex())] <
69 m_geoOnTheFly ?
ATH_MSG_INFO(
"Is Run-3 geometry: On-the-fly determination. THIS OPTION IS DEPRECATED AND WILL BE REMOVED SOON. Use IsRun3Geo property instead.")
81 "Not using allAuthors variable as currently missing in many derivations; LowPtEfficiency working point will always return "
82 "false, but this is expected at the moment. Have a look here: "
83 "https://twiki.cern.ch/twiki/bin/view/Atlas/MuonSelectionToolR21#New_LowPtEfficiency_working_poin");
87 ATH_MSG_INFO(
"You have opted to select only 3-station muons in the high-pT selection! "
88 <<
"Please feed 'HighPt3Layers' to the 'WorkingPoint' property to retrieve the appropriate scale-factors");
92 ATH_MSG_WARNING(
"No cut-based selection is defined for segment-tagged muons in the Low-pT working point. "
93 <<
"Please set UseMVALowPt=true if you want to try the UseSegmentTaggedLowPt=true option.");
97 ATH_MSG_INFO(
"MuonSelectionTool will assume both Standard and LRT Muons are being used, and that the necessary information is available to identify the type (standard or LRT).");
98 if (
m_quality!=1)
ATH_MSG_WARNING(
"Currently, only Medium quality is supported for LRT muons. Your chosen WP will be applied (w/o ID cuts), but no recommendations are available for this quality.");
103 m_acceptInfo.
addCut(
"IDHits",
"Selection of muons according to whether they passed the MCP ID Hit cuts");
104 m_acceptInfo.
addCut(
"Preselection",
"Selection of muons according to their type/author");
109 "Invalid quality (i.e. selection WP) set: "
111 <<
" - it must be an integer between 0 and 5! (0=Tight, 1=Medium, 2=Loose, 3=Veryloose, 4=HighPt, 5=LowPtEfficiency)");
112 return StatusCode::FAILURE;
115 ATH_MSG_ERROR(
"Cannot use lowPt working point if allAuthors is not available!");
116 return StatusCode::FAILURE;
119 if(m_caloScoreWP<1 || m_caloScoreWP>4){
120 ATH_MSG_FATAL(
"CaloScoreWP property must be set to 1, 2, 3 or 4");
121 return StatusCode::FAILURE;
125 ATH_MSG_INFO(
"Initialising tight working point histograms...");
126 std::string tightWP_rootFile_fullPath;
131 Form(
"MuonSelectorTools/%s/muonSelection_tightWPHisto.root",
m_calibration_version.value().c_str()));
134 ATH_MSG_INFO(
"Reading muon tight working point histograms from " << tightWP_rootFile_fullPath);
136 std::unique_ptr<TFile>
file(TFile::Open(tightWP_rootFile_fullPath.c_str(),
"READ"));
138 if (!
file->IsOpen()) {
139 ATH_MSG_ERROR(
"Cannot read tight working point file from " << tightWP_rootFile_fullPath);
140 return StatusCode::FAILURE;
154 ATH_MSG_INFO(
"Reading bad muon veto cut functions from " << BMVcutFile_fullPath);
156 std::unique_ptr<TFile> BMVfile(TFile::Open(BMVcutFile_fullPath.c_str(),
"READ"));
158 if (!BMVfile->IsOpen()) {
159 ATH_MSG_ERROR(
"Cannot read bad muon veto cut function file from " << BMVcutFile_fullPath);
160 return StatusCode::FAILURE;
170 return StatusCode::FAILURE;
181 auto make_mva_reader = [](TString
file_path) {
182 std::vector<std::string> mva_var_names{
"momentumBalanceSignificance",
183 "scatteringCurvatureSignificance",
184 "scatteringNeighbourSignificance",
186 "middleLargeHoles+middleSmallHoles",
187 "muonSegmentDeltaEta",
188 "muonSeg1ChamberIdx",
189 "muonSeg2ChamberIdx"};
190 std::unique_ptr<TMVA::Reader>
reader = std::make_unique<TMVA::Reader>(mva_var_names);
207 auto make_mva_reader_MuTagIMO = [](TString
file_path,
bool useSeg2ChamberIndex) {
208 std::vector<std::string> mva_var_names;
209 if (useSeg2ChamberIndex) mva_var_names.push_back(
"muonSeg2ChamberIndex");
210 mva_var_names.push_back(
"muonSeg1ChamberIndex");
211 mva_var_names.push_back(
"muonSeg1NPrecisionHits");
212 mva_var_names.push_back(
"muonSegmentDeltaEta");
213 mva_var_names.push_back(
"muonSeg1GlobalR");
214 mva_var_names.push_back(
"muonSeg1Chi2OverDoF");
215 mva_var_names.push_back(
"muonSCS");
217 std::unique_ptr<TMVA::Reader>
reader = std::make_unique<TMVA::Reader>(mva_var_names);
232 ATH_MSG_ERROR(
"Cannot calculate TightNNScore: ONNX tool not configured! "
233 "Please set the ORTInferenceTool property to a valid AthOnnx::OnnxRuntimeInferenceTool instance.");
234 return StatusCode::FAILURE;
239 }
else ATH_MSG_INFO(
"ONNX tool not configured — skipping retrieval.");
245 return StatusCode::SUCCESS;
251 ATH_MSG_ERROR(
" getHist(...) TFile is nullptr! Check that the Tight cut map is loaded correctly");
252 return StatusCode::FAILURE;
254 TH1* h_ptr =
nullptr;
260 return StatusCode::FAILURE;
262 hist = std::unique_ptr<TH1>{h_ptr};
263 hist->SetDirectory(
nullptr);
264 ATH_MSG_INFO(
"Successfully read tight working point histogram: " <<
hist->GetName());
266 return StatusCode::SUCCESS;
281 ATH_MSG_FATAL(
"accept(...) Failed to cast particle to muon");
292 static std::atomic<bool> checkDone{
false};
300 <<
" while on-the fly check for runNumber "<<
getRunNumber(
true)<<
" indicates isRun3Geo="<<
isRun3(
true));
306 ATH_MSG_WARNING(
"MuonSelectionTool currently supports Loose, Medium, Tight, HighPt, and LowPtEfficiency WPs for Run3; all other WPs can only be used in ExpertDevelopMode mode");
310 ATH_MSG_WARNING(
"For Run3, Tight WP is supported only when ExcludeNSWFromPrecisionLayers=False and RecalcPrecisionLayerswNSW=True");
325 else if (
mu.muonType() == xAOD::Muon::MuonStandAlone)
327 else if (
mu.muonType() == xAOD::Muon::SegmentTagged)
329 else if (
mu.muonType() == xAOD::Muon::CaloTagged)
331 else if (
mu.muonType() == xAOD::Muon::SiliconAssociatedForwardMuon)
358 if (!passIDCuts || !passMuonCuts) {
return acceptData; }
362 bool thisMu_highpt =
false;
364 bool thisMu_lowptE =
false;
367 ATH_MSG_VERBOSE(
"Muon quality: " << thisMu_quality <<
" passes HighPt: " << thisMu_highpt
368 <<
" passes LowPtEfficiency: " << thisMu_lowptE);
369 if (m_quality < 4 && thisMu_quality >
m_quality) {
return acceptData; }
370 if (
m_quality == 4 && !thisMu_highpt) {
return acceptData; }
371 if (
m_quality == 5 && !thisMu_lowptE) {
return acceptData; }
383 const xAOD::TrackParticle* metrack =
mu.trackParticle(xAOD::Muon::ExtrapolatedMuonSpectrometerTrackParticle);
384 if (!idtrack || !metrack) idPt = mePt = -1.;
386 mePt = metrack->
pt();
387 idPt = idtrack->
pt();
389 if (!mePt_acc.isAvailable(
mu) || !idPt_acc.isAvailable(
mu)) {
391 <<
" q:" <<
mu.charge() <<
", author:" <<
mu.author()
392 <<
" is not decorated with calibrated momenta. Please fix");
393 throw std::runtime_error(
"MuonSelectionTool() - qOverP significance calculation failed");
406 <<
" Momentum dependent cuts are disabled. Return 0.");
410 const xAOD::TrackParticle* metrack =
muon.trackParticle(xAOD::Muon::ExtrapolatedMuonSpectrometerTrackParticle);
411 if (!idtrack || !metrack) {
412 ATH_MSG_VERBOSE(
"No ID / MS track. Return dummy large value of 1 mio");
415 float mePt{-1.}, idPt{-1.};
422 return std::abs((metrack->
charge() / meP) - (idtrack->
charge() / idP)) / qOverPsigma;
427 <<
"Momentum dependent cuts are disabled. Return 0.");
430 float mePt{-1.}, idPt{-1.};
432 return std::abs(idPt - mePt) /
muon.pt();
443 if (
mu.muonType() == xAOD::Muon::SegmentTagged) {
446 if (std::abs(
mu.eta()) < 0.1) {
450 ATH_MSG_VERBOSE(
"Do not allow segment-tagged muon at |eta| > 0.1 - return VeryLoose");
456 if (
mu.muonType() == xAOD::Muon::CaloTagged) {
481 ATH_MSG_VERBOSE(
"Muon has out-of-bounds precision hits - return VeryLoose");
487 const xAOD::TrackParticle* metrack =
mu.trackParticle(xAOD::Muon::ExtrapolatedMuonSpectrometerTrackParticle);
491 const float reducedChi2 =
mu.primaryTrackParticle()->chiSquared() /
mu.primaryTrackParticle()->numberDoF();
499 if (
summary.nprecisionLayers > 1 && reducedChi2 < 8 && std::abs(qOverPsignif) < 7) {
506 ATH_MSG_VERBOSE(
"Muon did not pass requirements for tight combined muon");
510 (
summary.nprecisionLayers > 1 ||(
summary.nprecisionLayers == 1 &&
summary.nprecisionHoleLayers < 2 && std::abs(
mu.eta()) < 0.1))
517 ATH_MSG_VERBOSE(
"Muon did not pass requirements for medium combined muon");
523 if ((
summary.nprecisionLayers > 1 ||
524 (
summary.nprecisionLayers == 1 &&
summary.nprecisionHoleLayers < 2 && std::abs(
mu.eta()) < 0.1))) {
540 ATH_MSG_VERBOSE(
"Muon passed selection for loose working point at low pT");
545 ATH_MSG_VERBOSE(
"Did not pass selections for combined muon - returning VeryLoose");
554 ATH_MSG_VERBOSE(
"Standalone muons currently only used when in expert mode for run3");
558 if (std::abs(
mu.eta()) > 2.5) {
569 ATH_MSG_VERBOSE(
"Muon did not pass selection for medium stand-alone muon - return VeryLoose");
574 if (
mu.muonType() == xAOD::Muon::SiliconAssociatedForwardMuon) {
578 ATH_MSG_VERBOSE(
"Silicon-associated forward muon muons currently only used when in expert mode for run3");
583 const xAOD::TrackParticle* metrack =
mu.trackParticle(xAOD::Muon::ExtrapolatedMuonSpectrometerTrackParticle);
585 if (cbtrack && metrack) {
586 if (std::abs(cbtrack->
eta()) > 2.5) {
593 "SiliconForwardAssociated muon has ID track as primary track particle. "
594 <<
"This is a bug fixed starting with xAODMuon-00-17-07, which should be present in this release. "
595 <<
"Please report this to the Muon CP group!");
604 ATH_MSG_VERBOSE(
"Muon did not pass selection for medium silicon-associated forward muon - return VeryLoose");
608 ATH_MSG_VERBOSE(
"Muon did not pass selection for loose/medium/tight for any muon type - return VeryLoose");
618 if (isLRTmuon(
mu))
return true;
625 ATH_MSG_FATAL(
"No information available to tell if the muon is LRT or standard. Either run MuonLRTMergingAlg to decorate with `isLRT` flag, or supply the patternRecoInfo for the original ID track.");
626 throw std::runtime_error(
"MuonSelectionTool() - isLRT decor and patternRecoInfo both unavailable for a muon.");
628 std::bitset<xAOD::NumberOfTrackRecoInfo> patternBitSet(patternAcc(*idtrack));
636 }
else if (
mu.muonType() == xAOD::Muon::SiliconAssociatedForwardMuon) {
638 if (cbtrack && std::abs(cbtrack->
eta()) > 2.5) {
return true; }
641 if (
mu.trackParticle(xAOD::Muon::InnerDetectorTrackParticle))
642 return passedIDCuts(*
mu.trackParticle(xAOD::Muon::InnerDetectorTrackParticle));
643 else if (
mu.primaryTrackParticle())
653 const xAOD::TrackParticle* metrack =
mu.trackParticle(xAOD::Muon::ExtrapolatedMuonSpectrometerTrackParticle);
659 bool IsBadMuon =
false;
660 if (idtrack && metrack && cbtrack) {
662 double qOverP_ID = idtrack->
qOverP();
664 double qOverP_ME = metrack->
qOverP();
666 double qOverP_CB = cbtrack->
qOverP();
679 double IdCbRatio = std::abs((qOverPerr_ID / qOverP_ID) / (qOverPerr_CB / qOverP_CB));
680 double MeCbRatio = std::abs((qOverPerr_ME / qOverP_ME) / (qOverPerr_CB / qOverP_CB));
681 IsBadMuon = (IdCbRatio < 0.8 || MeCbRatio < 0.8 || IsBadMuon);
685 double IdCbRatio = std::abs((qOverPerr_ID / qOverP_ID) / (qOverPerr_CB / qOverP_CB));
686 double MeCbRatio = std::abs((qOverPerr_ME / qOverP_ME) / (qOverPerr_CB / qOverP_CB));
687 IsBadMuon = (IdCbRatio < 0.8 || MeCbRatio < 0.8);
705 ATH_MSG_VERBOSE(
"LowPt WP currently not supported for run3 if not in expert mode");
721 ATH_MSG_VERBOSE(
"Muon is not combined or segment-tagged - fail low-pT");
734 ATH_MSG_VERBOSE(
"Muon is neither MuGirl / MuidCo / MuTagIMO - fail low-pT");
753 ATH_MSG_VERBOSE(
"Not using MVA selection, failing low-pT selection due to medium requirement in forward region");
761 ATH_MSG_VERBOSE(
"Muon has out-of-bounds precision hits - fail low-pT");
769 uint nStationsCut = (std::abs(
mu.eta()) > 1.3 && std::abs(
mu.eta()) < 1.55) ? 2 : 1;
770 if (
summary.nprecisionLayers < nStationsCut) {
771 ATH_MSG_VERBOSE(
"number of precision layers = " << (
int)
summary.nprecisionLayers <<
" is lower than cut value " << nStationsCut
772 <<
" - fail low-pT");
780 ATH_MSG_VERBOSE(
"MuGirl muon is not confirmed by MuTagIMO - fail low-pT");
816 std::vector<const xAOD::MuonSegment*> segments_sorted;
817 segments_sorted.reserve(
mu.nMuonSegments());
819 for (
unsigned int i = 0;
i <
mu.nMuonSegments();
i++) {
820 if (!
mu.muonSegment(
i))
821 ATH_MSG_WARNING(
"The muon reports more segments than are available. Please report this to the muon software community!");
823 segments_sorted.push_back(
mu.muonSegment(
i));
826 std::sort(segments_sorted.begin(), segments_sorted.end(), chamberIndexCompare);
828 return segments_sorted;
834 ATH_MSG_VERBOSE(
"LowPt WP currently not supported for run3 if not in expert mode");
843 float momentumBalanceSig{-1}, CurvatureSig{-1}, energyLoss{-1}, muonSegmentDeltaEta{-1}, scatteringNeigbour{-1};
854 float seg1ChamberIdx{-1.}, seg2ChamberIdx{-1.}, middleHoles{-1.}, seg1NPrecisionHits{-1.}, seg1GlobalR{-1.}, seg1Chi2OverDoF{-1.};
859 ATH_MSG_WARNING(
"passedLowPtEfficiencyMVACut - found segment-tagged muon with no segments!");
862 seg1ChamberIdx = (!muonSegments.empty()) ?
toInt(muonSegments[0]->chamberIndex()) : -9;
863 seg2ChamberIdx = (muonSegments.size() > 1) ?
toInt(muonSegments[1]->chamberIndex()) : -9;
867 seg1NPrecisionHits = (!muonSegments.empty()) ? muonSegments[0]->nPrecisionHits() : -1;
868 seg1GlobalR = (!muonSegments.empty())
869 ? std::hypot(muonSegments[0]->
x(), muonSegments[0]->y(), muonSegments[0]->z())
871 seg1Chi2OverDoF = (!muonSegments.empty()) ? muonSegments[0]->
chiSquared() / muonSegments[0]->numberDoF() : -1;
880 std::vector<float> var_vector;
882 var_vector = {momentumBalanceSig, CurvatureSig, scatteringNeigbour, energyLoss,
883 middleHoles, muonSegmentDeltaEta, seg1ChamberIdx, seg2ChamberIdx};
885 if (std::abs(
mu.eta()) >= 1.3)
886 var_vector = {seg2ChamberIdx, seg1ChamberIdx, seg1NPrecisionHits, muonSegmentDeltaEta,
887 seg1GlobalR, seg1Chi2OverDoF, std::abs(CurvatureSig)};
889 var_vector = {seg1ChamberIdx, seg1NPrecisionHits, muonSegmentDeltaEta,
890 seg1GlobalR, seg1Chi2OverDoF, std::abs(CurvatureSig)};
905 if (std::abs(
mu.eta()) < 0.7)
907 else if (std::abs(
mu.eta()) < 1.3)
913 float BDTdiscriminant;
916 BDTdiscriminant = reader_MUID->EvaluateMVA(var_vector,
"BDTG");
918 BDTdiscriminant = reader_MUGIRL->EvaluateMVA(var_vector,
"BDTG");
920 BDTdiscriminant = reader_MUTAGIMO->EvaluateMVA(var_vector,
"BDT");
922 ATH_MSG_WARNING(
"Invalid author for low-pT MVA, failing selection...");
929 if (BDTdiscriminant > BDTcut) {
955 ATH_MSG_VERBOSE(
"Muon has out-of-bounds precision hits - fail high-pT");
975 ATH_MSG_VERBOSE(
"passedHighPtCuts - No MS track available for muon. Using combined track.");
976 MS_track =
mu.trackParticle(xAOD::Muon::CombinedTrackParticle);
979 if (MS_track && CB_track) {
981 float phiMS = MS_track->
phi();
982 float etaCB = CB_track->
eta();
985 if (!
isRun3() && (std::abs(
etaMS) > 2.0 || std::abs(etaCB) > 2.0)) {
986 if (
summary.cscUnspoiledEtaHits == 0) {
993 if (!
isRun3() &&
mu.eta() < -1.899 && std::abs(
mu.phi()) < 0.211) {
994 ATH_MSG_VERBOSE(
"Muon is in eta/phi region vetoed due to disabled chambers in MC - fail high-pT");
999 if ((1.01 < std::abs(
etaMS) && std::abs(
etaMS) < 1.1) || (1.01 < std::abs(etaCB) && std::abs(etaCB) < 1.1)) {
1000 ATH_MSG_VERBOSE(
"Muon is in barrel/endcap overlap region - fail high-pT");
1030 if (
summary.nprecisionLayers < 4) {
1031 ATH_MSG_VERBOSE(
"Muon is in BEE eta/phi region and does not have 4 precision layers - fail high-pT");
1035 if (std::abs(etaCB) > 1.4) {
1039 if (
summary.nprecisionLayers < 4 && (
summary.extendedSmallHits > 0 ||
summary.extendedSmallHoles > 0)) {
1040 ATH_MSG_VERBOSE(
"Muon is in BEE eta/phi region and does not have 4 precision layers - fail high-pT");
1045 ATH_MSG_WARNING(
"passedHighPtCuts - MS or CB track missing in muon! Failing High-pT selection...");
1051 const xAOD::TrackParticle* metrack =
mu.trackParticle(xAOD::Muon::ExtrapolatedMuonSpectrometerTrackParticle);
1056 if (std::abs(qOverPsignif) > 7) {
1068 if (std::abs(
mu.eta()) > 1.2 &&
summary.extendedSmallHits < 3 &&
summary.extendedLargeHits < 3) {
1092 if (
summary.nprecisionLayers < 3) {
1093 ATH_MSG_VERBOSE(
"Muon has less than 3 precision layers - fail high-pT");
1098 if (
summary.isSmallGoodSectors) {
1099 if (!(
summary.innerSmallHits > 2 &&
summary.middleSmallHits > 2 &&
1100 (
summary.outerSmallHits > 2 ||
summary.extendedSmallHits > 2))) {
1101 ATH_MSG_VERBOSE(
"Muon has small/large sectors overlap - fail high-pT");
1105 if (!(
summary.innerLargeHits > 2 &&
summary.middleLargeHits > 2 &&
1106 (
summary.outerLargeHits > 2 ||
summary.extendedLargeHits > 2))) {
1107 ATH_MSG_VERBOSE(
"Muon has small/large sectors overlap - fail high-pT");
1120 double start_cut = 3.0;
1121 double end_cut = 1.6;
1122 double abs_eta = std::abs(
mu.eta());
1125 double p0(8.0),
p1(0.),
p2(0.);
1132 else if (abs_eta > 1.05 && abs_eta <= 1.3) {
1135 }
else if (abs_eta > 1.3 && abs_eta <= 1.7) {
1138 }
else if (abs_eta > 1.7 && abs_eta <= 2.0) {
1141 }
else if (abs_eta > 2.0) {
1152 else if (abs_eta > 1.05 && abs_eta <= 1.3) {
1155 }
else if (abs_eta > 1.3 && abs_eta <= 1.7) {
1158 }
else if (abs_eta > 1.7 && abs_eta <= 2.0) {
1161 }
else if (abs_eta > 2.0) {
1176 if (abs_eta > 1.05 && abs_eta < 1.3) {
1179 }
else if (abs_eta >= 1.3 && abs_eta < 1.7) {
1182 }
else if (abs_eta >= 1.7 && abs_eta < 2.0) {
1185 }
else if (abs_eta >= 2.0) {
1191 bool passErrorCutCB =
false;
1196 double qOverP_CB = cbtrack->
qOverP();
1202 double a = (end_cut - start_cut) / 4000.0;
1203 double b = end_cut -
a * 5000.0;
1204 double coefficient = (pt_CB > 1000.) ? (
a * pt_CB +
b) : start_cut;
1205 if (std::abs(qOverPerr_CB / qOverP_CB) < coefficient *
sigma) { passErrorCutCB =
true; }
1212 ATH_MSG_DEBUG(
"The current event is a MC event. Use bad muon veto mimic.");
1218 return passErrorCutCB;
1224 if (std::abs(
mu.eta()) < 1.05) {
1234 double qOpRelResolution = std::hypot(
p1,
p2 *
mu.primaryTrackParticle()->pt() *
MeVtoGeV);
1236 double qOverPabs_unsmeared = std::abs(
mu.primaryTrackParticle()->definingParameters()[4]);
1237 double qOverPabs_smeared = 1.0 / (
mu.pt() * std::cosh(
mu.eta()));
1239 if ((qOverPabs_smeared - qOverPabs_unsmeared) / (qOpRelResolution * qOverPabs_unsmeared) <
1240 cutFunction->Eval(
mu.primaryTrackParticle()->pt() *
MeVtoGeV))
1250 if (
mu.muonType() == xAOD::Muon::CaloTagged && std::abs(
mu.eta()) < 0.105)
1257 if (
mu.muonType() == xAOD::Muon::SiliconAssociatedForwardMuon) {
1259 return (cbtrack && std::abs(cbtrack->
eta()) > 2.4);
1270 " !! Tool configured with some of the ID hits requirements changed... FOR DEVELOPMENT ONLY: muon efficiency SF won't be "
1286 const float abseta = std::abs(
track.eta());
1290 if (!((0.1 < abseta && abseta <= 1.9 && totTRThits > 5 &&
value2 < (0.9 * totTRThits)) || (abseta <= 0.1 || abseta > 1.9)))
1305 int CaloMuonIDTag = -20;
1308 bool readID =
mu.parameter(CaloMuonIDTag, xAOD::Muon::CaloMuonIDTag);
1310 ATH_MSG_WARNING(
"Unable to read CaloMuonIDTag Quality information! Rejecting the CALO muon!");
1315 return (CaloMuonIDTag > 10);
1324 float CaloMuonScore{-999.0};
1335 return (CaloMuonScore >= 0.77);
1351 ATH_MSG_VERBOSE(
"for run3, Tight WP is only supported when ExcludeNSWFromPrecisionLayers=False and RecalcPrecisionLayerswNSW=True");
1354 float symmetric_eta = std::abs(
mu.eta());
1358 if (pt < 4.0 || symmetric_eta >= 2.5)
return false;
1359 ATH_MSG_VERBOSE(
"Muon is passing tight WP kinematic cuts with pT,eta " <<
mu.pt() <<
" , " <<
mu.eta());
1366 ATH_MSG_VERBOSE(
"Applying tight WP cuts to a low pt muon with (pt,eta) ( " <<
pt <<
" , " <<
mu.eta() <<
" ) ");
1368 ATH_MSG_VERBOSE(
"Momentum significance value " << oneOverPSig <<
", required to be less than " << qOverPCut);
1370 if (
rho > rhoCut)
return false;
1373 if (oneOverPSig > qOverPCut)
return false;
1374 ATH_MSG_VERBOSE(
"Muon passed tight WP, low pT momentum significance cut");
1382 else if (
pt < 100.0) {
1385 ATH_MSG_VERBOSE(
"Applying tight WP cuts to a medium pt muon with (pt,eta) (" <<
pt <<
"," <<
mu.eta() <<
")");
1389 if (
rho > rhoCut)
return false;
1397 else if (
pt < 500.0) {
1399 ATH_MSG_VERBOSE(
"Applying tight WP cuts to a high pt muon with (pt,eta) (" <<
pt <<
"," <<
mu.eta() <<
")");
1402 ATH_MSG_VERBOSE(
"Rho value " <<
rho <<
", required to be less than " << rhoCut <<
" unless -1, in which no cut is applied");
1404 if (rhoCut < 0.0)
return true;
1405 if (
rho > rhoCut)
return false;
1412 ATH_MSG_VERBOSE(
"Not applying any tight WP cuts to a very high pt muon with (pt,eta) (" <<
pt <<
"," <<
mu.eta() <<
")");
1444 if (std::abs(
muon.eta()) > 2.0) {
1445 ATH_MSG_VERBOSE(
"Recalculating number of precision layers for combined muon");
1454 + (
summary.extendedSmallHits > 2 ||
summary.extendedLargeHits > 2);
1458 if (!eta1stgchits_acc.isAvailable(
muon) || !eta2stgchits_acc.isAvailable(
muon) || !mmhits_acc.isAvailable(
muon)) {
1459 ATH_MSG_FATAL(__FILE__ <<
":" << __LINE__ <<
" Failed to retrieve NSW hits!"
1460 <<
" (Please use DxAODs with p-tags >= p5834 OR set ExcludeNSWFromPrecisionLayers to True (tests only)");
1461 throw std::runtime_error(
"Failed to retrieve NSW hits");
1475 ATH_MSG_FATAL(__FILE__ <<
":" << __LINE__ <<
" Failed to retrieve parameter " << param
1476 <<
" for muon with pT:" <<
muon.pt() *
MeVtoGeV <<
", eta:" <<
muon.eta() <<
", phi: " <<
muon.phi()
1477 <<
", q:" <<
muon.charge() <<
", author: " <<
muon.author());
1478 throw std::runtime_error(
"Failed to retrieve Parameter");
1493 if (
summary.nprecisionLayers == 2)
1494 return ResolutionCategory::highPt2station;
1502 ATH_MSG_VERBOSE(
"getResolutionCategory - No MS track available for muon. Using combined track.");
1503 MS_track =
mu.trackParticle(xAOD::Muon::CombinedTrackParticle);
1507 const float etaMS = MS_track->
eta();
1508 const float etaCB = CB_track->
eta();
1509 const float phiMS = MS_track->
phi();
1514 category = ResolutionCategory::missingInner;
1517 category = ResolutionCategory::missingMiddle;
1521 category = ResolutionCategory::missingOuter;
1523 if (!
isRun3() && (std::abs(
etaMS) > 2.0 || std::abs(etaCB) > 2.0) &&
summary.cscUnspoiledEtaHits == 0)
1524 category = ResolutionCategory::spoiledCSC;
1526 if ((1.01 < std::abs(
etaMS) && std::abs(
etaMS) < 1.1) || (1.01 < std::abs(etaCB) && std::abs(etaCB) < 1.1))
1527 category = ResolutionCategory::BEoverlap;
1532 if (
isBEE(
etaMS, phiMS) || (std::abs(etaCB) > 1.4 && (
summary.extendedSmallHits > 0 ||
summary.extendedSmallHoles > 0))) {
1534 category = ResolutionCategory::missingBEE;
1536 if (
summary.extendedSmallHits >= 3 &&
summary.outerSmallHits < 3)
category = ResolutionCategory::missingOuter;
1538 if (!
summary.isSmallGoodSectors)
1542 if (
summary.nprecisionLayers == 1)
category = ResolutionCategory::oneStation;
1557 ATH_MSG_DEBUG(
"The current event is a data event. Return runNumber.");
1563 static std::atomic<bool> issuedWarningPRW{
false};
1565 unsigned int rn = acc_rnd(*eventInfo);
1566 if (
rn != 0)
return acc_rnd(*eventInfo);
1568 if (!issuedWarningPRW) {
1570 issuedWarningPRW =
true;
1575 if (needOnlyCorrectYear) {
1577 ATH_MSG_DEBUG(
"Random run number not available and this is mc16a or mc20a, returning dummy 2016 run number.");
1580 }
else if (eventInfo->
runNumber() < 310000) {
1581 ATH_MSG_DEBUG(
"Random run number not available and this is mc16d or mc20d, returning dummy 2017 run number.");
1584 }
else if (eventInfo->
runNumber() < 320000) {
1585 ATH_MSG_DEBUG(
"Random run number not available and this is mc16e or mc20e, returning dummy 2018 run number.");
1588 }
else if (eventInfo->
runNumber() < 600000) {
1589 ATH_MSG_DEBUG(
"Random run number not available and this is mc21/mc23, for the time being we're returing a dummy run number.");
1596 ATH_MSG_FATAL(
"Random run number not available, fallback option of using runNumber failed since "<<eventInfo->
runNumber()<<
" cannot be recognised");
1597 throw std::runtime_error(
"MuonSelectionTool() - need RandomRunNumber decoration by the PileupReweightingTool");
1600 ATH_MSG_FATAL(
"Failed to find the RandomRunNumber decoration by the PileupReweightingTool");
1601 throw std::runtime_error(
"MuonSelectionTool() - need RandomRunNumber decoration from PileupReweightingTool");
1607 static constexpr std::array<float, 2> BIS78_eta{1.05, 1.3};
1608 static constexpr std::array<float, 8> BIS78_phi{0.21, 0.57, 1.00, 1.33, 1.78, 2.14, 2.57, 2.93};
1610 float abs_eta = std::abs(eta);
1611 float abs_phi = std::abs(
phi);
1613 if (abs_eta >= BIS78_eta[0] && abs_eta <= BIS78_eta[1]) {
1614 if ((abs_phi >= BIS78_phi[0] && abs_phi <= BIS78_phi[1]) || (abs_phi >= BIS78_phi[2] && abs_phi <= BIS78_phi[3]) ||
1615 (abs_phi >= BIS78_phi[4] && abs_phi <= BIS78_phi[5]) || (abs_phi >= BIS78_phi[6] && abs_phi <= BIS78_phi[7])) {
1625 static constexpr std::array<float, 2> BEE_eta{1.440, 1.692};
1626 static constexpr std::array<float, 8> BEE_phi{0.301, 0.478, 1.086, 1.263, 1.872, 2.049, 2.657, 2.834};
1628 float abs_eta = std::abs(eta);
1629 float abs_phi = std::abs(
phi);
1631 if (abs_eta >= BEE_eta[0] && abs_eta <= BEE_eta[1]) {
1632 if ((abs_phi >= BEE_phi[0] && abs_phi <= BEE_phi[1]) || (abs_phi >= BEE_phi[2] && abs_phi <= BEE_phi[3]) ||
1633 (abs_phi >= BEE_phi[4] && abs_phi <= BEE_phi[5]) || (abs_phi >= BEE_phi[6] && abs_phi <= BEE_phi[7])) {
1643 static constexpr std::array<float, 6> BMG_eta{0.35, 0.47, 0.68, 0.80, 0.925, 1.04};
1644 static constexpr std::array<float, 4> BMG_phi{-1.93, -1.765, -1.38, -1.21};
1646 float abs_eta = std::abs(eta);
1648 if ((abs_eta >= BMG_eta[0] && abs_eta <= BMG_eta[1]) || (abs_eta >= BMG_eta[2] && abs_eta <= BMG_eta[3]) ||
1649 (abs_eta >= BMG_eta[4] && abs_eta <= BMG_eta[5])) {
1650 if ((
phi >= BMG_phi[0] &&
phi <= BMG_phi[1]) || (
phi >= BMG_phi[2] &&
phi <= BMG_phi[3])) {
return true; }
1659 ATH_MSG_ERROR(
"TightNNScore calculation is disabled. Please set the property CalculateTightNNScore to true.");
1660 throw std::runtime_error(
"cannot calculate TightNNScore");
1665 const xAOD::TrackParticle* metrack =
mu.trackParticle(xAOD::Muon::ExtrapolatedMuonSpectrometerTrackParticle);
1666 if(!idtrack || !metrack)
return -999;
1670 if (std::abs(
mu.eta())>2.5)
return -999;
1671 if(
mu.pt()<4000.)
return -999;
1673 std::vector<float> input_features;
1675 int mu_author=
mu.author();
1677 float mu_scatteringCurvatureSignificance=0.;
1679 float mu_scatteringNeighbourSignificance=0.;
1681 float mu_momentumBalanceSignificance=0.;
1684 float mu_reducedChi2=
mu.primaryTrackParticle()->chiSquared() /
mu.primaryTrackParticle()->numberDoF();
1687 float mu_spectrometerFieldIntegral=0.;
1689 float mu_segmentDeltaEta=0;
1691 uint8_t mu_numberOfPixelHits=0;
1693 uint8_t mu_numberOfPixelDeadSensors=0;
1710 input_features = {(
float)mu_author,
1712 mu_scatteringCurvatureSignificance,
1713 mu_scatteringNeighbourSignificance,
1714 mu_momentumBalanceSignificance,
1715 mu_qOverPSignificance,
1719 mu_spectrometerFieldIntegral,
1721 (
float)mu_numberOfPixelHits,
1722 (
float)mu_numberOfPixelDeadSensors,
1723 (
float)mu_innerLargeHits,
1724 (
float)mu_innerSmallHits,
1725 (
float)mu_middleLargeHits,
1726 (
float)mu_middleSmallHits,
1727 (
float)mu_outerLargeHits,
1728 (
float)mu_outerSmallHits};
1732 uint8_t mu_phiLayer1STGCHits=0;
1734 uint8_t mu_phiLayer2STGCHits=0;
1736 uint8_t mu_etaLayer1STGCHits=0;
1738 uint8_t mu_etaLayer2STGCHits=0;
1742 input_features = {(
float)mu_author,
1744 mu_scatteringCurvatureSignificance,
1745 mu_scatteringNeighbourSignificance,
1746 mu_momentumBalanceSignificance,
1747 mu_qOverPSignificance,
1751 mu_spectrometerFieldIntegral,
1753 (
float)mu_numberOfPixelHits,
1754 (
float)mu_numberOfPixelDeadSensors,
1755 (
float)mu_innerLargeHits,
1756 (
float)mu_innerSmallHits,
1757 (
float)mu_middleLargeHits,
1758 (
float)mu_middleSmallHits,
1759 (
float)mu_outerLargeHits,
1760 (
float)mu_outerSmallHits,
1761 (
float)mu_phiLayer1STGCHits,
1762 (
float)mu_phiLayer2STGCHits,
1763 (
float)mu_etaLayer1STGCHits,
1764 (
float)mu_etaLayer2STGCHits,
1769 std::vector<int64_t> inputShape = {1,
static_cast<int64_t
>(input_features.size())};
1772 inputData[
"flatten_input"] = std::make_pair(
1773 inputShape, std::move(input_features)
1777 outputData[
"TightNNScore"] = std::make_pair(
1778 std::vector<int64_t>{1, 1}, std::vector<float>{}
1781 if (!
m_onnxTool->inference(inputData, outputData).isSuccess()) {
1785 const auto& variant = outputData[
"TightNNScore"].second;
1786 if (std::holds_alternative<std::vector<float>>(variant)) {
1787 const auto&
vec = std::get<std::vector<float>>(variant);
1798 ATH_MSG_DEBUG(
"TightNNScore for muon with pT " <<
mu.pt() <<
" GeV, eta " <<
mu.eta() <<
" is " <<
score);