Loading [MathJax]/extensions/tex2jax.js
ATLAS Offline Software
•All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Properties Friends Macros Modules Pages
Namespaces | Classes | Functions
TauGNNUtils Namespace Reference

Namespaces

 Variables
 

Classes

class  GNNVarCalc
 Tool to calculate input variables for the GNN-based tau identification. More...
 

Functions

std::unique_ptr< GNNVarCalc > get_calculator (const std::vector< std::string > &scalar_vars, const std::vector< std::string > &track_vars, const std::vector< std::string > &cluster_vars)
 

Function Documentation

◆ get_calculator()

std::unique_ptr< GNNVarCalc > TauGNNUtils::get_calculator ( const std::vector< std::string > &  scalar_vars,
const std::vector< std::string > &  track_vars,
const std::vector< std::string > &  cluster_vars 
)

Definition at line 113 of file TauGNNUtils.cxx.

115  {
116  auto calc = std::make_unique<GNNVarCalc>();
117 
118  // Scalar variable calculator functions
119  //calc->insert("absEta", Variables::absEta, scalar_vars);
120  calc->insert("isolFrac", Variables::isolFrac, scalar_vars);
121  calc->insert("centFrac", Variables::centFrac, scalar_vars);
122  calc->insert("etOverPtLeadTrk", Variables::etOverPtLeadTrk, scalar_vars);
123  calc->insert("innerTrkAvgDist", Variables::innerTrkAvgDist, scalar_vars);
124  calc->insert("absipSigLeadTrk", Variables::absipSigLeadTrk, scalar_vars);
125  calc->insert("SumPtTrkFrac", Variables::SumPtTrkFrac, scalar_vars);
126  calc->insert("sumEMCellEtOverLeadTrkPt", Variables::sumEMCellEtOverLeadTrkPt, scalar_vars);
127  calc->insert("EMPOverTrkSysP", Variables::EMPOverTrkSysP, scalar_vars);
128  calc->insert("ptRatioEflowApprox", Variables::ptRatioEflowApprox, scalar_vars);
129  calc->insert("mEflowApprox", Variables::mEflowApprox, scalar_vars);
130  calc->insert("dRmax", Variables::dRmax, scalar_vars);
131  calc->insert("trFlightPathSig", Variables::trFlightPathSig, scalar_vars);
132  calc->insert("massTrkSys", Variables::massTrkSys, scalar_vars);
133  calc->insert("pt", Variables::pt, scalar_vars);
134  //calc->insert("pt_tau_log", Variables::pt_tau_log, scalar_vars);
135  //calc->insert("ptDetectorAxis", Variables::ptDetectorAxis, scalar_vars);
136  //calc->insert("ptIntermediateAxis", Variables::ptIntermediateAxis, scalar_vars);
137  //---added for the eVeto
138  /*calc->insert("ptJetSeed_log", Variables::ptJetSeed_log, scalar_vars);
139  calc->insert("absleadTrackEta", Variables::absleadTrackEta, scalar_vars);
140  calc->insert("leadTrackDeltaEta", Variables::leadTrackDeltaEta, scalar_vars);
141  calc->insert("leadTrackDeltaPhi", Variables::leadTrackDeltaPhi, scalar_vars);
142  calc->insert("leadTrackProbNNorHT", Variables::leadTrackProbNNorHT, scalar_vars);
143  calc->insert("EMFracFixed", Variables::EMFracFixed, scalar_vars);
144  calc->insert("etHotShotWinOverPtLeadTrk", Variables::etHotShotWinOverPtLeadTrk, scalar_vars);
145  calc->insert("hadLeakFracFixed", Variables::hadLeakFracFixed, scalar_vars);
146  calc->insert("PSFrac", Variables::PSFrac, scalar_vars);
147  calc->insert("ClustersMeanCenterLambda", Variables::ClustersMeanCenterLambda, scalar_vars);
148  calc->insert("ClustersMeanFirstEngDens", Variables::ClustersMeanFirstEngDens, scalar_vars);
149  calc->insert("ClustersMeanPresamplerFrac", Variables::ClustersMeanPresamplerFrac, scalar_vars);
150  */
151 
152  // Track variable calculator functions
153  calc->insert("pt_log", Variables::Track::pt_log, track_vars);
154  calc->insert("trackPt", Variables::Track::trackPt, track_vars);
155  calc->insert("trackEta", Variables::Track::trackEta, track_vars);
156  calc->insert("trackPhi", Variables::Track::trackPhi, track_vars);
157  //calc->insert("pt_tau_log", Variables::Track::pt_tau_log, track_vars);
158  //calc->insert("pt_jetseed_log", Variables::Track::pt_jetseed_log, track_vars);
159  calc->insert("d0_abs_log", Variables::Track::d0_abs_log, track_vars);
160  calc->insert("z0sinThetaTJVA_abs_log", Variables::Track::z0sinThetaTJVA_abs_log, track_vars);
161  calc->insert("z0sinthetaTJVA", Variables::Track::z0sinthetaTJVA, track_vars);
162  calc->insert("z0sinthetaSigTJVA", Variables::Track::z0sinthetaSigTJVA, track_vars);
163  calc->insert("d0TJVA", Variables::Track::d0TJVA, track_vars);
164  calc->insert("d0SigTJVA", Variables::Track::d0SigTJVA, track_vars);
165  calc->insert("dEta", Variables::Track::dEta, track_vars);
166  calc->insert("dEtaJetSeedAxis", Variables::Track::dEtaJetSeedAxis, track_vars);
167  calc->insert("dPhi", Variables::Track::dPhi, track_vars);
168  calc->insert("dPhiJetSeedAxis", Variables::Track::dPhiJetSeedAxis, track_vars);
169  calc->insert("nInnermostPixelHits", Variables::Track::nInnermostPixelHits, track_vars);
170  calc->insert("nPixelHits", Variables::Track::nPixelHits, track_vars);
171  //calc->insert("nSCTHits", Variables::Track::nSCTHits, track_vars);
172  //calc->insert("nIBLHitsAndExp", Variables::Track::nIBLHitsAndExp, track_vars);
173  calc->insert("nPixelHitsPlusDeadSensors", Variables::Track::nPixelHitsPlusDeadSensors, track_vars);
174  calc->insert("nSCTHitsPlusDeadSensors", Variables::Track::nSCTHitsPlusDeadSensors, track_vars);
175  /* Development variables
176  //calc->insert("eProbabilityHT", Variables::Track::eProbabilityHT, track_vars);
177  //calc->insert("eProbabilityNN", Variables::Track::eProbabilityNN, track_vars);
178  //calc->insert("eProbabilityNNorHT", Variables::Track::eProbabilityNNorHT, track_vars);
179  //calc->insert("chargedScoreRNN", Variables::Track::chargedScoreRNN, track_vars);
180  //calc->insert("isolationScoreRNN", Variables::Track::isolationScoreRNN, track_vars);
181  //calc->insert("conversionScoreRNN", Variables::Track::conversionScoreRNN, track_vars);
182  //calc->insert("fakeScoreRNN", Variables::Track::fakeScoreRNN, track_vars);
183  */
184 
185  //Extension - variables for GNTau
186  calc->insert("numberOfInnermostPixelLayerHits", Variables::Track::numberOfInnermostPixelLayerHits, track_vars);
187  /* Development variables
188  calc->insert("numberOfPixelHits", Variables::Track::numberOfPixelHits, track_vars);
189  calc->insert("numberOfPixelSharedHits", Variables::Track::numberOfPixelSharedHits, track_vars);
190  calc->insert("numberOfPixelDeadSensors", Variables::Track::numberOfPixelDeadSensors, track_vars);
191  calc->insert("numberOfSCTHits", Variables::Track::numberOfSCTHits, track_vars);
192  calc->insert("numberOfSCTSharedHits", Variables::Track::numberOfSCTSharedHits, track_vars);
193  calc->insert("numberOfSCTDeadSensors", Variables::Track::numberOfSCTDeadSensors, track_vars);
194  calc->insert("numberOfTRTHighThresholdHits", Variables::Track::numberOfTRTHighThresholdHits, track_vars);
195  calc->insert("numberOfTRTHits", Variables::Track::numberOfTRTHits, track_vars);
196  calc->insert("nSiHits", Variables::Track::nSiHits, track_vars);
197  calc->insert("expectInnermostPixelLayerHit", Variables::Track::expectInnermostPixelLayerHit, track_vars);
198  calc->insert("expectNextToInnermostPixelLayerHit", Variables::Track::expectNextToInnermostPixelLayerHit, track_vars);
199  calc->insert("numberOfContribPixelLayers", Variables::Track::numberOfContribPixelLayers, track_vars);
200  calc->insert("numberOfPixelHoles", Variables::Track::numberOfPixelHoles, track_vars);
201  calc->insert("d0_old", Variables::Track::d0_old, track_vars);
202  calc->insert("qOverP", Variables::Track::qOverP, track_vars);
203  calc->insert("theta", Variables::Track::theta, track_vars);
204  calc->insert("z0TJVA", Variables::Track::z0TJVA, track_vars);
205  calc->insert("charge", Variables::Track::charge, track_vars);
206  calc->insert("dz0_TV_PV0", Variables::Track::dz0_TV_PV0, track_vars);
207  calc->insert("log_sumpt_TV", Variables::Track::log_sumpt_TV, track_vars);
208  calc->insert("log_sumpt2_TV", Variables::Track::log_sumpt2_TV, track_vars);
209  calc->insert("log_sumpt_PV0", Variables::Track::log_sumpt_PV0, track_vars);
210  calc->insert("log_sumpt2_PV0", Variables::Track::log_sumpt2_PV0, track_vars);
211  */
212 
213  // Cluster variable calculator functions
214  //calc->insert("et_log", Variables::Cluster::et_log, cluster_vars);
215  //calc->insert("pt_tau_log", Variables::Cluster::pt_tau_log, cluster_vars);
216  //calc->insert("pt_jetseed_log", Variables::Cluster::pt_jetseed_log, cluster_vars);
217  calc->insert("dEta", Variables::Cluster::dEta, cluster_vars);
218  calc->insert("dPhi", Variables::Cluster::dPhi, cluster_vars);
219  calc->insert("SECOND_R", Variables::Cluster::SECOND_R, cluster_vars);
220  calc->insert("SECOND_LAMBDA", Variables::Cluster::SECOND_LAMBDA, cluster_vars);
221  calc->insert("CENTER_LAMBDA", Variables::Cluster::CENTER_LAMBDA, cluster_vars);
222  /*---added for the eVeto
223  calc->insert("SECOND_LAMBDAOverClustersMeanSecondLambda", Variables::Cluster::SECOND_LAMBDAOverClustersMeanSecondLambda, cluster_vars);
224  calc->insert("CENTER_LAMBDAOverClustersMeanCenterLambda", Variables::Cluster::CENTER_LAMBDAOverClustersMeanCenterLambda, cluster_vars);
225  calc->insert("FirstEngDensOverClustersMeanFirstEngDens" , Variables::Cluster::FirstEngDensOverClustersMeanFirstEngDens, cluster_vars);
226  */
227 
228  //Extension - Variables for GNTau
229  //calc->insert("e", Variables::Cluster::e, cluster_vars);
230  calc->insert("et", Variables::Cluster::et, cluster_vars);
231 
232  /*Development variables
233  calc->insert("FIRST_ENG_DENS", Variables::Cluster::FIRST_ENG_DENS, cluster_vars);
234  calc->insert("EM_PROBABILITY", Variables::Cluster::EM_PROBABILITY, cluster_vars);
235  calc->insert("CENTER_MAG", Variables::Cluster::CENTER_MAG, cluster_vars);
236  */
237  return calc;
238 }
TauGNNUtils::Variables::Cluster::dPhi
bool dPhi(const xAOD::TauJet &tau, const xAOD::CaloVertexedTopoCluster &cluster, double &out)
Definition: TauGNNUtils.cxx:835
TauGNNUtils::Variables::Cluster::CENTER_LAMBDA
bool CENTER_LAMBDA(const xAOD::TauJet &, const xAOD::CaloVertexedTopoCluster &cluster, double &out)
Definition: TauGNNUtils.cxx:850
TauGNNUtils::Variables::centFrac
bool centFrac(const xAOD::TauJet &tau, double &out)
Definition: TauGNNUtils.cxx:249
TauGNNUtils::Variables::ptRatioEflowApprox
bool ptRatioEflowApprox(const xAOD::TauJet &tau, double &out)
Definition: TauGNNUtils.cxx:307
TauGNNUtils::Variables::absipSigLeadTrk
bool absipSigLeadTrk(const xAOD::TauJet &tau, double &out)
Definition: TauGNNUtils.cxx:279
TauGNNUtils::Variables::Track::nPixelHits
bool nPixelHits(const xAOD::TauJet &, const xAOD::TauTrack &track, double &out)
Definition: TauGNNUtils.cxx:566
TauGNNUtils::Variables::Track::dEtaJetSeedAxis
bool dEtaJetSeedAxis(const xAOD::TauJet &tau, const xAOD::TauTrack &track, double &out)
Definition: TauGNNUtils.cxx:542
TauGNNUtils::Variables::innerTrkAvgDist
bool innerTrkAvgDist(const xAOD::TauJet &tau, double &out)
Definition: TauGNNUtils.cxx:272
TauGNNUtils::Variables::isolFrac
bool isolFrac(const xAOD::TauJet &tau, double &out)
Definition: TauGNNUtils.cxx:257
TauGNNUtils::Variables::Cluster::et
bool et(const xAOD::TauJet &, const xAOD::CaloVertexedTopoCluster &cluster, double &out)
Definition: TauGNNUtils.cxx:907
TauGNNUtils::Variables::massTrkSys
bool massTrkSys(const xAOD::TauJet &tau, double &out)
Definition: TauGNNUtils.cxx:336
TauGNNUtils::Variables::Track::d0_abs_log
bool d0_abs_log(const xAOD::TauJet &, const xAOD::TauTrack &track, double &out)
Definition: TauGNNUtils.cxx:507
TauGNNUtils::Variables::trFlightPathSig
bool trFlightPathSig(const xAOD::TauJet &tau, double &out)
Definition: TauGNNUtils.cxx:329
TauGNNUtils::Variables::Track::z0sinthetaTJVA
bool z0sinthetaTJVA(const xAOD::TauJet &, const xAOD::TauTrack &track, double &out)
Definition: TauGNNUtils.cxx:517
TauGNNUtils::Variables::Track::z0sinthetaSigTJVA
bool z0sinthetaSigTJVA(const xAOD::TauJet &, const xAOD::TauTrack &track, double &out)
Definition: TauGNNUtils.cxx:522
TauGNNUtils::Variables::Track::nInnermostPixelHits
bool nInnermostPixelHits(const xAOD::TauJet &, const xAOD::TauTrack &track, double &out)
Definition: TauGNNUtils.cxx:559
TauGNNUtils::Variables::Cluster::SECOND_LAMBDA
bool SECOND_LAMBDA(const xAOD::TauJet &, const xAOD::CaloVertexedTopoCluster &cluster, double &out)
Definition: TauGNNUtils.cxx:845
TauGNNUtils::Variables::mEflowApprox
bool mEflowApprox(const xAOD::TauJet &tau, double &out)
Definition: TauGNNUtils.cxx:315
TauGNNUtils::Variables::Track::trackPt
bool trackPt(const xAOD::TauJet &, const xAOD::TauTrack &track, double &out)
Definition: TauGNNUtils.cxx:482
TauGNNUtils::Variables::Track::dPhiJetSeedAxis
bool dPhiJetSeedAxis(const xAOD::TauJet &tau, const xAOD::TauTrack &track, double &out)
Definition: TauGNNUtils.cxx:553
TauGNNUtils::Variables::SumPtTrkFrac
bool SumPtTrkFrac(const xAOD::TauJet &tau, double &out)
Definition: TauGNNUtils.cxx:293
TauGNNUtils::Variables::Cluster::SECOND_R
bool SECOND_R(const xAOD::TauJet &, const xAOD::CaloVertexedTopoCluster &cluster, double &out)
Definition: TauGNNUtils.cxx:840
TauGNNUtils::Variables::sumEMCellEtOverLeadTrkPt
bool sumEMCellEtOverLeadTrkPt(const xAOD::TauJet &tau, double &out)
Definition: TauGNNUtils.cxx:286
TauGNNUtils::Variables::Track::nPixelHitsPlusDeadSensors
bool nPixelHitsPlusDeadSensors(const xAOD::TauJet &, const xAOD::TauTrack &track, double &out)
Definition: TauGNNUtils.cxx:589
TauGNNUtils::Variables::dRmax
bool dRmax(const xAOD::TauJet &tau, double &out)
Definition: TauGNNUtils.cxx:322
TauGNNUtils::Variables::EMPOverTrkSysP
bool EMPOverTrkSysP(const xAOD::TauJet &tau, double &out)
Definition: TauGNNUtils.cxx:300
TauGNNUtils::Variables::etOverPtLeadTrk
bool etOverPtLeadTrk(const xAOD::TauJet &tau, double &out)
Definition: TauGNNUtils.cxx:265
TauGNNUtils::Variables::Track::pt_log
bool pt_log(const xAOD::TauJet &, const xAOD::TauTrack &track, double &out)
Definition: TauGNNUtils.cxx:477
TauGNNUtils::Variables::Track::nSCTHitsPlusDeadSensors
bool nSCTHitsPlusDeadSensors(const xAOD::TauJet &, const xAOD::TauTrack &track, double &out)
Definition: TauGNNUtils.cxx:597
TauGNNUtils::Variables::pt
bool pt(const xAOD::TauJet &tau, double &out)
Definition: TauGNNUtils.cxx:343
TauGNNUtils::Variables::Track::trackPhi
bool trackPhi(const xAOD::TauJet &, const xAOD::TauTrack &track, double &out)
Definition: TauGNNUtils.cxx:492
TauGNNUtils::Variables::Track::d0TJVA
bool d0TJVA(const xAOD::TauJet &, const xAOD::TauTrack &track, double &out)
Definition: TauGNNUtils.cxx:527
TauGNNUtils::Variables::Track::z0sinThetaTJVA_abs_log
bool z0sinThetaTJVA_abs_log(const xAOD::TauJet &, const xAOD::TauTrack &track, double &out)
Definition: TauGNNUtils.cxx:512
beamspotnt.calc
calc
Definition: bin/beamspotnt.py:1252
TauGNNUtils::Variables::Track::numberOfInnermostPixelLayerHits
bool numberOfInnermostPixelLayerHits(const xAOD::TauJet &, const xAOD::TauTrack &track, double &out)
Definition: TauGNNUtils.cxx:652
TauGNNUtils::Variables::Cluster::dEta
bool dEta(const xAOD::TauJet &tau, const xAOD::CaloVertexedTopoCluster &cluster, double &out)
Definition: TauGNNUtils.cxx:830
TauGNNUtils::Variables::Track::d0SigTJVA
bool d0SigTJVA(const xAOD::TauJet &, const xAOD::TauTrack &track, double &out)
Definition: TauGNNUtils.cxx:532
TauGNNUtils::Variables::Track::trackEta
bool trackEta(const xAOD::TauJet &, const xAOD::TauTrack &track, double &out)
Definition: TauGNNUtils.cxx:487
Generated on Fri Apr 18 2025 22:13:31 for ATLAS Offline Software by   doxygen 1.8.18