73{
74 const EventContext& ctx = Gaudi::Hive::currentContext();
75
76 SG::ThinningHandle<xAOD::TrackParticleContainer> TrackParticles(
m_inDetSGKey, ctx);
77 if (!TrackParticles.isValid()) {
79 return StatusCode::FAILURE;
80 }
81 SG::ThinningHandle<xAOD::TauTrackContainer> TauTracks(
m_tauTracksSGKey, ctx);
82 if (!TauTracks.isValid()) {
84 return StatusCode::FAILURE;
85 }
86 SG::ThinningHandle<xAOD::TauJetContainer> LepRMTaus(
m_LepRMTauKey,ctx);
87 if (!LepRMTaus.isValid()) {
89 return StatusCode::FAILURE;
90 }
92 if (!OriTaus.isValid()) {
94 return StatusCode::FAILURE;
95 }
96
97
98 std::vector<int> tau_kine_mask = m_parser->evaluateAsVector();
99 std::vector<bool> tau_lep_remove_mask(OriTaus->size(), false);
100 static const SG::AuxElement::ConstAccessor<char> acc_modified("ModifiedInAOD");
101 std::transform(LepRMTaus->cbegin(), LepRMTaus->cend(), tau_lep_remove_mask.begin(),
102 [&](auto lep_remove_tau) -> bool {
103 return static_cast<bool>(acc_modified(*lep_remove_tau));
104 }
105 );
106 std::vector<bool> tau_mask(OriTaus->size(),false);
107 std::transform(tau_lep_remove_mask.cbegin(), tau_lep_remove_mask.cend(), tau_kine_mask.cbegin(), tau_mask.begin(),
108 [](auto tau_lep_remove_pass, auto tau_kine_pass) -> bool
109 {
110 return tau_lep_remove_pass == 1 && tau_kine_pass == 1;
111 }
112 );
113
114 std::vector<const xAOD::TauJet*> LepRMTausAfterThinning;
115 std::transform(tau_mask.cbegin(), tau_mask.cend(), LepRMTaus->cbegin(), std::back_inserter(LepRMTausAfterThinning),
117 {
118 return mask ? tau : nullptr;
119 }
120 );
121 LepRMTausAfterThinning.erase(
std::remove(LepRMTausAfterThinning.begin(), LepRMTausAfterThinning.end(),
nullptr), LepRMTausAfterThinning.end());
122
123
124 std::vector<bool> tau_track_mask(TauTracks->size(), false);
125 std::vector<bool> id_track_mask(TrackParticles->size(), false);
126 std::for_each(LepRMTausAfterThinning.cbegin(), LepRMTausAfterThinning.cend(),
127 [&](auto tau)
128 {
129 auto tau_track_links = tau->tauTrackLinks();
130 std::for_each(tau_track_links.cbegin(), tau_track_links.cend(),
131 [&](auto link)
132 {
133 if (link.isValid())
134 {
135 tau_track_mask[link.index()] = 1;
136 if (auto id_track_link = (*link)->trackLinks().at(0); id_track_link.isValid())
137 {
138 id_track_mask[id_track_link.index()] = 1;
139 }
140 }
141 }
142 );
143 }
144 );
145
146 LepRMTaus.keep(tau_mask);
147 TauTracks.keep(tau_track_mask);
148 TrackParticles.keep(id_track_mask);
149
153 m_npass_taus += std::accumulate(tau_mask.begin(), tau_mask.end(),0);
154 m_npass_trks += std::accumulate(tau_track_mask.begin(), tau_track_mask.end(),0);
155 m_npass_ID_trks += std::accumulate(id_track_mask.begin(), id_track_mask.end(),0);
156 ATH_MSG_DEBUG(std::accumulate(tau_mask.begin(), tau_mask.end(),0) <<
" / " << OriTaus->size() <<
" taus were retained");
157
158 return StatusCode::SUCCESS;
159
160}
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.14.0