72{
73
74 SG::ThinningHandle<xAOD::TrackParticleContainer> TrackParticles(
m_inDetSGKey, ctx);
75 if (!TrackParticles.isValid()) {
77 return StatusCode::FAILURE;
78 }
79 SG::ThinningHandle<xAOD::TauTrackContainer> TauTracks(
m_tauTracksSGKey, ctx);
80 if (!TauTracks.isValid()) {
82 return StatusCode::FAILURE;
83 }
84 SG::ThinningHandle<xAOD::TauJetContainer> LepRMTaus(
m_LepRMTauKey,ctx);
85 if (!LepRMTaus.isValid()) {
87 return StatusCode::FAILURE;
88 }
90 if (!OriTaus.isValid()) {
92 return StatusCode::FAILURE;
93 }
94
95
96 std::vector<int> tau_kine_mask = m_parser->evaluateAsVector();
97 std::vector<bool> tau_lep_remove_mask(OriTaus->size(), false);
98 static const SG::AuxElement::ConstAccessor<char> acc_modified("ModifiedInAOD");
99 std::transform(LepRMTaus->cbegin(), LepRMTaus->cend(), tau_lep_remove_mask.begin(),
100 [&](auto lep_remove_tau) -> bool {
101 return static_cast<bool>(acc_modified(*lep_remove_tau));
102 }
103 );
104 std::vector<bool> tau_mask(OriTaus->size(),false);
105 std::transform(tau_lep_remove_mask.cbegin(), tau_lep_remove_mask.cend(), tau_kine_mask.cbegin(), tau_mask.begin(),
106 [](auto tau_lep_remove_pass, auto tau_kine_pass) -> bool
107 {
108 return tau_lep_remove_pass == 1 && tau_kine_pass == 1;
109 }
110 );
111
112 std::vector<const xAOD::TauJet*> LepRMTausAfterThinning;
113 std::transform(tau_mask.cbegin(), tau_mask.cend(), LepRMTaus->cbegin(), std::back_inserter(LepRMTausAfterThinning),
115 {
116 return mask ? tau : nullptr;
117 }
118 );
119 LepRMTausAfterThinning.erase(
std::remove(LepRMTausAfterThinning.begin(), LepRMTausAfterThinning.end(),
nullptr), LepRMTausAfterThinning.end());
120
121
122 std::vector<bool> tau_track_mask(TauTracks->size(), false);
123 std::vector<bool> id_track_mask(TrackParticles->size(), false);
124 std::for_each(LepRMTausAfterThinning.cbegin(), LepRMTausAfterThinning.cend(),
125 [&](auto tau)
126 {
127 auto tau_track_links = tau->tauTrackLinks();
128 std::for_each(tau_track_links.cbegin(), tau_track_links.cend(),
129 [&](auto link)
130 {
131 if (link.isValid())
132 {
133 tau_track_mask[link.index()] = 1;
134 if (auto id_track_link = (*link)->trackLinks().at(0); id_track_link.isValid())
135 {
136 id_track_mask[id_track_link.index()] = 1;
137 }
138 }
139 }
140 );
141 }
142 );
143
144 LepRMTaus.keep(tau_mask);
145 TauTracks.keep(tau_track_mask);
146 TrackParticles.keep(id_track_mask);
147
151 m_npass_taus += std::accumulate(tau_mask.begin(), tau_mask.end(),0);
152 m_npass_trks += std::accumulate(tau_track_mask.begin(), tau_track_mask.end(),0);
153 m_npass_ID_trks += std::accumulate(id_track_mask.begin(), id_track_mask.end(),0);
154 ATH_MSG_DEBUG(std::accumulate(tau_mask.begin(), tau_mask.end(),0) <<
" / " << OriTaus->size() <<
" taus were retained");
155
156 return StatusCode::SUCCESS;
157
158}
std::atomic< unsigned int > m_npass_ID_trks
std::atomic< unsigned int > m_ntot_ID_trks
SG::ThinningHandleKey< xAOD::TauTrackContainer > m_tauTracksSGKey
SG::ThinningHandleKey< xAOD::TauJetContainer > m_LepRMTauKey
std::atomic< unsigned int > m_npass_trks
SG::ReadHandleKey< xAOD::TauJetContainer > m_originalTauKey
std::atomic< unsigned int > m_npass_taus
std::atomic< unsigned int > m_ntot_taus
std::atomic< unsigned int > m_ntot_trks
SG::ThinningHandleKey< xAOD::TrackParticleContainer > m_inDetSGKey
DataModel_detail::iterator< DVL > remove(typename DataModel_detail::iterator< DVL > beg, typename DataModel_detail::iterator< DVL > end, const T &value)
Specialization of remove for DataVector/List.
TauJet_v3 TauJet
Definition of the current "tau version".
1.16.1