Loading [MathJax]/extensions/tex2jax.js
 |
ATLAS Offline Software
|
Go to the documentation of this file.
38 ATH_MSG_INFO(
"Intialize TauTrackRNNClassifier tool : " << classifier );
44 return StatusCode::SUCCESS;
51 if (!vertexInHandle.
isValid()) {
52 ATH_MSG_ERROR (
"Could not retrieve HiveDataObj with key " << vertexInHandle.
key());
53 return StatusCode::FAILURE;
70 std::vector<xAOD::TauTrack*> vLRTs;
72 while(
it != vTracks.end()) {
75 it = vTracks.erase(
it);
84 ATH_CHECK(classifier->classifyTracks(vLRTs, xTau, vertexContainer,
true));
90 ATH_CHECK(classifier->classifyTracks(vTracks, xTau, vertexContainer));
94 std::sort(tauTrackLinks.begin(), tauTrackLinks.end(),
sortTracks);
97 charge += trk->track()->charge();
120 return StatusCode::SUCCESS;
128 TrackRNN::TrackRNN(
const std::string&
name)
130 , m_inputWeightsPath(
"")
137 TrackRNN::~TrackRNN()
145 ATH_MSG_INFO(
"Using calibration file: " << inputWeightsPath);
147 std::ifstream nn_config_istream(inputWeightsPath);
154 return StatusCode::FAILURE;
157 return StatusCode::SUCCESS;
164 bool skipTracks)
const
166 if(vTracks.empty()) {
167 return StatusCode::SUCCESS;
178 idScoreCharged(*track) = 0.;
179 idScoreConv(*track) = 0.;
180 idScoreIso(*track) = 0.;
181 idScoreFake(*track) = 0.;
183 return StatusCode::SUCCESS;
194 seqInput[
"input_1"] = valueMap;
198 std::vector<double> vClassProb(5);
200 for (
unsigned int i = 0;
i < vTracks.size(); ++
i){
209 vClassProb[0] = mValue[
"type_0"][
i];
210 vClassProb[1] = mValue[
"type_1"][
i];
211 vClassProb[2] = mValue[
"type_2"][
i];
212 vClassProb[3] = mValue[
"type_3"][
i];
216 idScoreCharged(*vTracks[
i]) = vClassProb[0];
217 idScoreConv(*vTracks[
i]) = vClassProb[1];
218 idScoreIso(*vTracks[
i]) = vClassProb[2];
219 idScoreFake(*vTracks[
i]) = vClassProb[3];
222 for (
unsigned int j = 1; j < vClassProb.size(); ++j){
223 if(vClassProb[j] > vClassProb[iMaxIndex]) iMaxIndex = j;
232 }
else if(iMaxIndex == 0){
234 }
else if(iMaxIndex == 1){
236 }
else if(iMaxIndex == 2){
241 return StatusCode::SUCCESS;
253 unsigned int n_timeSteps = vTracks.size();
258 valueMap[
"log(trackPt)"] = std::vector<double>(n_timeSteps);
259 valueMap[
"log(jetSeedPt)"] = std::vector<double>(n_timeSteps);
260 valueMap[
"trackPt/tauPtIntermediateAxis"] = std::vector<double>(n_timeSteps);
261 valueMap[
"trackEta"] = std::vector<double>(n_timeSteps);
262 valueMap[
"z0sinthetaTJVA"] = std::vector<double>(n_timeSteps);
263 valueMap[
"z0sinthetaSigTJVA"] = std::vector<double>(n_timeSteps);
264 valueMap[
"log(rConv)"] = std::vector<double>(n_timeSteps);
265 valueMap[
"tanh(rConvII/500)"] = std::vector<double>(n_timeSteps);
266 valueMap[
"dRJetSeedAxis"] = std::vector<double>(n_timeSteps);
267 valueMap[
"dRIntermediateAxis"] = std::vector<double>(n_timeSteps);
268 valueMap[
"tanh(d0SigTJVA/10)"] = std::vector<double>(n_timeSteps);
269 valueMap[
"tanh(d0TJVA/10)"] = std::vector<double>(n_timeSteps);
270 valueMap[
"qOverP*1000"] = std::vector<double>(n_timeSteps);
271 valueMap[
"numberOfInnermostPixelLayerHits"] = std::vector<double>(n_timeSteps);
272 valueMap[
"numberOfPixelSharedHits"] = std::vector<double>(n_timeSteps);
273 valueMap[
"numberOfSCTSharedHits"] = std::vector<double>(n_timeSteps);
274 valueMap[
"numberOfTRTHits"] = std::vector<double>(n_timeSteps);
275 valueMap[
"eProbabilityHT"] = std::vector<double>(n_timeSteps);
276 valueMap[
"nPixHits"] = std::vector<double>(n_timeSteps);
277 valueMap[
"nSCTHits"] = std::vector<double>(n_timeSteps);
278 valueMap[
"dz0_TV_PV0"] = std::vector<double>(n_timeSteps);
279 valueMap[
"log_sumpt_TV"] = std::vector<double>(n_timeSteps);
280 valueMap[
"log_sumpt2_TV"] = std::vector<double>(n_timeSteps);
281 valueMap[
"log_sumpt_PV0"] = std::vector<double>(n_timeSteps);
282 valueMap[
"log_sumpt2_PV0"] = std::vector<double>(n_timeSteps);
283 valueMap[
"charge"] = std::vector<double>(n_timeSteps);
289 double dz0_TV_PV0 = 0., sumpt_TV = 0., sumpt2_TV = 0., sumpt_PV0 = 0., sumpt2_PV0 = 0.;
290 if(vertexContainer !=
nullptr && !vertexContainer->
empty() && xTau.
vertex()!=
nullptr) {
294 sumpt_PV0 += (*trk)->pt();
295 sumpt2_PV0 +=
pow((*trk)->pt(), 2.);
298 sumpt_TV += (*trk)->pt();
299 sumpt2_TV +=
pow((*trk)->pt(), 2.);
325 valueMap[
"log(trackPt)"][
i] =
std::log( xTrackParticle->
pt() );
326 valueMap[
"log(jetSeedPt)"][
i] = log_ptJetSeed;
328 valueMap[
"trackEta"][
i] = xTrackParticle->
eta();
329 valueMap[
"z0sinthetaTJVA"][
i] = xTrack->z0sinthetaTJVA();
330 valueMap[
"z0sinthetaSigTJVA"][
i] = xTrack->z0sinthetaSigTJVA();
331 valueMap[
"log(rConv)"][
i] =
std::log( xTrack->rConv() );
332 valueMap[
"tanh(rConvII/500)"][
i] = std::tanh( xTrack->rConvII()/500. );
336 valueMap[
"tanh(d0SigTJVA/10)"][
i] = std::tanh( xTrack->d0SigTJVA()/10. );
337 valueMap[
"tanh(d0TJVA/10)"][
i] = std::tanh( xTrack->d0TJVA()/10. );
338 valueMap[
"qOverP*1000"][
i] = xTrackParticle->
qOverP()*1000.;
351 valueMap[
"charge"][
i] = xTrackParticle->
charge();
359 return StatusCode::SUCCESS;
JetConstituentVector::iterator iterator
bool log_sumpt2_TV(const xAOD::TauJet &tau, const xAOD::TauTrack &, double &out)
virtual double pt() const override final
The transverse momentum ( ) of the particle.
@ numberOfSCTSharedHits
number of SCT hits shared by several tracks [unit8_t].
bool log_sumpt_PV0(const xAOD::TauJet &tau, const xAOD::TauTrack &, double &out)
const_pointer_type cptr()
Dereference the pointer.
float charge() const
Returns the charge.
Gaudi::Details::PropertyBase & declareProperty(Gaudi::Property< T > &t)
virtual double eta() const override final
The pseudorapidity ( ) of the particle.
const TrackParticleLinks_t & trackParticleLinks() const
Get all the particles associated with the vertex.
bool dz0_TV_PV0(const xAOD::TauJet &tau, const xAOD::TauTrack &, double &out)
bool summaryValue(uint8_t &value, const SummaryType &information) const
Accessor for TrackSummary values.
size_t nTracks(TauJetParameters::TauTrackFlag flag=TauJetParameters::TauTrackFlag::classifiedCharged) const
@ nModifiedIsolationTracks
@ numberOfPixelHits
these are the pixel hits, including the b-layer [unit8_t].
@ numberOfTRTHits
number of TRT hits [unit8_t].
float nPixelHits(const U &p)
float nSCTSharedHits(const U &p)
std::vector< xAOD::TauTrack * > allTauTracksNonConst(const xAOD::TauJet *tau, xAOD::TauTrackContainer *trackCont)
@ numberOfPixelSharedHits
number of Pixel all-layer hits shared by several tracks [unit8_t].
float nSCTHits(const U &p)
float nTRTHits(const U &p)
::StatusCode StatusCode
StatusCode definition for legacy code.
Class describing a tau jet.
bool log_sumpt_TV(const xAOD::TauJet &tau, const xAOD::TauTrack &, double &out)
float z() const
Returns the z position.
TauTrackLinks_t & allTauTrackLinksNonConst()
In order to sort track links.
virtual bool isValid() override final
Can the handle be successfully dereferenced?
float qOverP() const
Returns the parameter.
double charge(const T &p)
virtual const std::string & key() const override final
Return the StoreGate ID for the referenced object.
GraphConfig parse_json_graph(std::istream &json)
float nPixelSharedHits(const U &p)
const Vertex * vertex() const
@ numberOfSCTDeadSensors
number of dead SCT sensors crossed [unit8_t].
virtual FourMom_t p4() const
The full 4-momentum of the particle.
void setDetail(TauJetParameters::Detail detail, int value)
@ eProbabilityHT
Electron probability from High Threshold (HT) information [float].
@ numberOfSCTHits
number of hits in SCT [unit8_t].
std::map< std::string, OutputNodeConfig > outputs
@ numberOfPixelDeadSensors
number of dead pixel sensors crossed [unit8_t].
Class describing a TrackParticle.
const T * at(size_type n) const
Access an element, as an rvalue.
const Trk::Track * track() const
Returns a pointer (which can be NULL) to the Trk::Track which was used to make this TrackParticle.
double ptIntermediateAxis() const
constexpr int pow(int base, int exp) noexcept
std::vector< const TauTrack * > tracks(TauJetParameters::TauTrackFlag flag=TauJetParameters::TauTrackFlag::classifiedCharged) const
Get the v<const pointer> to a given tauTrack collection associated with this tau.
bool empty() const noexcept
Returns true if the collection is empty.
@ numberOfInnermostPixelLayerHits
these are the hits in the 0th pixel barrel layer
bool log_sumpt2_PV0(const xAOD::TauJet &tau, const xAOD::TauTrack &, double &out)