ATLAS Offline Software
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 114 of file TauGNNUtils.cxx.

116  {
117  auto calc = std::make_unique<GNNVarCalc>();
118 
119  // Scalar variable calculator functions
120  //calc->insert("absEta", Variables::absEta, scalar_vars);
121  calc->insert("isolFrac", Variables::isolFrac, scalar_vars);
122  calc->insert("centFrac", Variables::centFrac, scalar_vars);
123  calc->insert("etOverPtLeadTrk", Variables::etOverPtLeadTrk, scalar_vars);
124  calc->insert("innerTrkAvgDist", Variables::innerTrkAvgDist, scalar_vars);
125  calc->insert("absipSigLeadTrk", Variables::absipSigLeadTrk, scalar_vars);
126  calc->insert("SumPtTrkFrac", Variables::SumPtTrkFrac, scalar_vars);
127  calc->insert("sumEMCellEtOverLeadTrkPt", Variables::sumEMCellEtOverLeadTrkPt, scalar_vars);
128  calc->insert("EMPOverTrkSysP", Variables::EMPOverTrkSysP, scalar_vars);
129  calc->insert("ptRatioEflowApprox", Variables::ptRatioEflowApprox, scalar_vars);
130  calc->insert("mEflowApprox", Variables::mEflowApprox, scalar_vars);
131  calc->insert("dRmax", Variables::dRmax, scalar_vars);
132  calc->insert("trFlightPathSig", Variables::trFlightPathSig, scalar_vars);
133  calc->insert("massTrkSys", Variables::massTrkSys, scalar_vars);
134  calc->insert("pt", Variables::pt, scalar_vars);
135  //calc->insert("pt_tau_log", Variables::pt_tau_log, scalar_vars);
136  //calc->insert("ptDetectorAxis", Variables::ptDetectorAxis, scalar_vars);
137  //calc->insert("ptIntermediateAxis", Variables::ptIntermediateAxis, scalar_vars);
138  //---added for the eVeto
139  /*calc->insert("ptJetSeed_log", Variables::ptJetSeed_log, scalar_vars);
140  calc->insert("absleadTrackEta", Variables::absleadTrackEta, scalar_vars);
141  calc->insert("leadTrackDeltaEta", Variables::leadTrackDeltaEta, scalar_vars);
142  calc->insert("leadTrackDeltaPhi", Variables::leadTrackDeltaPhi, scalar_vars);
143  calc->insert("leadTrackProbNNorHT", Variables::leadTrackProbNNorHT, scalar_vars);
144  calc->insert("EMFracFixed", Variables::EMFracFixed, scalar_vars);
145  calc->insert("etHotShotWinOverPtLeadTrk", Variables::etHotShotWinOverPtLeadTrk, scalar_vars);
146  calc->insert("hadLeakFracFixed", Variables::hadLeakFracFixed, scalar_vars);
147  calc->insert("PSFrac", Variables::PSFrac, scalar_vars);
148  calc->insert("ClustersMeanCenterLambda", Variables::ClustersMeanCenterLambda, scalar_vars);
149  calc->insert("ClustersMeanFirstEngDens", Variables::ClustersMeanFirstEngDens, scalar_vars);
150  calc->insert("ClustersMeanPresamplerFrac", Variables::ClustersMeanPresamplerFrac, scalar_vars);
151  */
152 
153  // Track variable calculator functions
154  calc->insert("pt_log", Variables::Track::pt_log, track_vars);
155  calc->insert("trackPt", Variables::Track::trackPt, track_vars);
156  calc->insert("trackEta", Variables::Track::trackEta, track_vars);
157  calc->insert("trackPhi", Variables::Track::trackPhi, track_vars);
158  //calc->insert("pt_tau_log", Variables::Track::pt_tau_log, track_vars);
159  //calc->insert("pt_jetseed_log", Variables::Track::pt_jetseed_log, track_vars);
160  calc->insert("d0_abs_log", Variables::Track::d0_abs_log, track_vars);
161  calc->insert("z0sinThetaTJVA_abs_log", Variables::Track::z0sinThetaTJVA_abs_log, track_vars);
162  calc->insert("z0sinthetaTJVA", Variables::Track::z0sinthetaTJVA, track_vars);
163  calc->insert("z0sinthetaSigTJVA", Variables::Track::z0sinthetaSigTJVA, track_vars);
164  calc->insert("d0TJVA", Variables::Track::d0TJVA, track_vars);
165  calc->insert("d0SigTJVA", Variables::Track::d0SigTJVA, track_vars);
166  calc->insert("dEta", Variables::Track::dEta, track_vars);
167  calc->insert("dEtaJetSeedAxis", Variables::Track::dEtaJetSeedAxis, track_vars);
168  calc->insert("dPhi", Variables::Track::dPhi, track_vars);
169  calc->insert("dPhiJetSeedAxis", Variables::Track::dPhiJetSeedAxis, track_vars);
170  calc->insert("nInnermostPixelHits", Variables::Track::nInnermostPixelHits, track_vars);
171  calc->insert("nPixelHits", Variables::Track::nPixelHits, track_vars);
172  //calc->insert("nSCTHits", Variables::Track::nSCTHits, track_vars);
173  //calc->insert("nIBLHitsAndExp", Variables::Track::nIBLHitsAndExp, track_vars);
174  calc->insert("nPixelHitsPlusDeadSensors", Variables::Track::nPixelHitsPlusDeadSensors, track_vars);
175  calc->insert("nSCTHitsPlusDeadSensors", Variables::Track::nSCTHitsPlusDeadSensors, track_vars);
176  /* Development variables
177  //calc->insert("eProbabilityHT", Variables::Track::eProbabilityHT, track_vars);
178  //calc->insert("eProbabilityNN", Variables::Track::eProbabilityNN, track_vars);
179  //calc->insert("eProbabilityNNorHT", Variables::Track::eProbabilityNNorHT, track_vars);
180  //calc->insert("chargedScoreRNN", Variables::Track::chargedScoreRNN, track_vars);
181  //calc->insert("isolationScoreRNN", Variables::Track::isolationScoreRNN, track_vars);
182  //calc->insert("conversionScoreRNN", Variables::Track::conversionScoreRNN, track_vars);
183  //calc->insert("fakeScoreRNN", Variables::Track::fakeScoreRNN, track_vars);
184  */
185 
186  //Extension - variables for GNTau
187  calc->insert("numberOfInnermostPixelLayerHits", Variables::Track::numberOfInnermostPixelLayerHits, track_vars);
188  /* Development variables
189  calc->insert("numberOfPixelHits", Variables::Track::numberOfPixelHits, track_vars);
190  calc->insert("numberOfPixelSharedHits", Variables::Track::numberOfPixelSharedHits, track_vars);
191  calc->insert("numberOfPixelDeadSensors", Variables::Track::numberOfPixelDeadSensors, track_vars);
192  calc->insert("numberOfSCTHits", Variables::Track::numberOfSCTHits, track_vars);
193  calc->insert("numberOfSCTSharedHits", Variables::Track::numberOfSCTSharedHits, track_vars);
194  calc->insert("numberOfSCTDeadSensors", Variables::Track::numberOfSCTDeadSensors, track_vars);
195  calc->insert("numberOfTRTHighThresholdHits", Variables::Track::numberOfTRTHighThresholdHits, track_vars);
196  calc->insert("numberOfTRTHits", Variables::Track::numberOfTRTHits, track_vars);
197  calc->insert("nSiHits", Variables::Track::nSiHits, track_vars);
198  calc->insert("expectInnermostPixelLayerHit", Variables::Track::expectInnermostPixelLayerHit, track_vars);
199  calc->insert("expectNextToInnermostPixelLayerHit", Variables::Track::expectNextToInnermostPixelLayerHit, track_vars);
200  calc->insert("numberOfContribPixelLayers", Variables::Track::numberOfContribPixelLayers, track_vars);
201  calc->insert("numberOfPixelHoles", Variables::Track::numberOfPixelHoles, track_vars);
202  calc->insert("d0_old", Variables::Track::d0_old, track_vars);
203  calc->insert("qOverP", Variables::Track::qOverP, track_vars);
204  calc->insert("theta", Variables::Track::theta, track_vars);
205  calc->insert("z0TJVA", Variables::Track::z0TJVA, track_vars);
206  calc->insert("charge", Variables::Track::charge, track_vars);
207  calc->insert("dz0_TV_PV0", Variables::Track::dz0_TV_PV0, track_vars);
208  calc->insert("log_sumpt_TV", Variables::Track::log_sumpt_TV, track_vars);
209  calc->insert("log_sumpt2_TV", Variables::Track::log_sumpt2_TV, track_vars);
210  calc->insert("log_sumpt_PV0", Variables::Track::log_sumpt_PV0, track_vars);
211  calc->insert("log_sumpt2_PV0", Variables::Track::log_sumpt2_PV0, track_vars);
212  */
213 
214  // Cluster variable calculator functions
215  //calc->insert("et_log", Variables::Cluster::et_log, cluster_vars);
216  //calc->insert("pt_tau_log", Variables::Cluster::pt_tau_log, cluster_vars);
217  //calc->insert("pt_jetseed_log", Variables::Cluster::pt_jetseed_log, cluster_vars);
218  calc->insert("dEta", Variables::Cluster::dEta, cluster_vars);
219  calc->insert("dPhi", Variables::Cluster::dPhi, cluster_vars);
220  calc->insert("SECOND_R", Variables::Cluster::SECOND_R, cluster_vars);
221  calc->insert("SECOND_LAMBDA", Variables::Cluster::SECOND_LAMBDA, cluster_vars);
222  calc->insert("CENTER_LAMBDA", Variables::Cluster::CENTER_LAMBDA, cluster_vars);
223  /*---added for the eVeto
224  calc->insert("SECOND_LAMBDAOverClustersMeanSecondLambda", Variables::Cluster::SECOND_LAMBDAOverClustersMeanSecondLambda, cluster_vars);
225  calc->insert("CENTER_LAMBDAOverClustersMeanCenterLambda", Variables::Cluster::CENTER_LAMBDAOverClustersMeanCenterLambda, cluster_vars);
226  calc->insert("FirstEngDensOverClustersMeanFirstEngDens" , Variables::Cluster::FirstEngDensOverClustersMeanFirstEngDens, cluster_vars);
227  */
228 
229  //Extension - Variables for GNTau
230  //calc->insert("e", Variables::Cluster::e, cluster_vars);
231  calc->insert("et", Variables::Cluster::et, cluster_vars);
232 
233  /*Development variables
234  calc->insert("FIRST_ENG_DENS", Variables::Cluster::FIRST_ENG_DENS, cluster_vars);
235  calc->insert("EM_PROBABILITY", Variables::Cluster::EM_PROBABILITY, cluster_vars);
236  calc->insert("CENTER_MAG", Variables::Cluster::CENTER_MAG, cluster_vars);
237  */
238  return calc;
239 }
TauGNNUtils::Variables::Cluster::dPhi
bool dPhi(const xAOD::TauJet &tau, const xAOD::CaloVertexedTopoCluster &cluster, double &out)
Definition: TauGNNUtils.cxx:836
TauGNNUtils::Variables::Cluster::CENTER_LAMBDA
bool CENTER_LAMBDA(const xAOD::TauJet &, const xAOD::CaloVertexedTopoCluster &cluster, double &out)
Definition: TauGNNUtils.cxx:851
TauGNNUtils::Variables::centFrac
bool centFrac(const xAOD::TauJet &tau, double &out)
Definition: TauGNNUtils.cxx:250
TauGNNUtils::Variables::ptRatioEflowApprox
bool ptRatioEflowApprox(const xAOD::TauJet &tau, double &out)
Definition: TauGNNUtils.cxx:308
TauGNNUtils::Variables::absipSigLeadTrk
bool absipSigLeadTrk(const xAOD::TauJet &tau, double &out)
Definition: TauGNNUtils.cxx:280
TauGNNUtils::Variables::Track::nPixelHits
bool nPixelHits(const xAOD::TauJet &, const xAOD::TauTrack &track, double &out)
Definition: TauGNNUtils.cxx:567
TauGNNUtils::Variables::Track::dEtaJetSeedAxis
bool dEtaJetSeedAxis(const xAOD::TauJet &tau, const xAOD::TauTrack &track, double &out)
Definition: TauGNNUtils.cxx:543
TauGNNUtils::Variables::innerTrkAvgDist
bool innerTrkAvgDist(const xAOD::TauJet &tau, double &out)
Definition: TauGNNUtils.cxx:273
TauGNNUtils::Variables::isolFrac
bool isolFrac(const xAOD::TauJet &tau, double &out)
Definition: TauGNNUtils.cxx:258
TauGNNUtils::Variables::Cluster::et
bool et(const xAOD::TauJet &, const xAOD::CaloVertexedTopoCluster &cluster, double &out)
Definition: TauGNNUtils.cxx:908
TauGNNUtils::Variables::massTrkSys
bool massTrkSys(const xAOD::TauJet &tau, double &out)
Definition: TauGNNUtils.cxx:337
TauGNNUtils::Variables::Track::d0_abs_log
bool d0_abs_log(const xAOD::TauJet &, const xAOD::TauTrack &track, double &out)
Definition: TauGNNUtils.cxx:508
TauGNNUtils::Variables::trFlightPathSig
bool trFlightPathSig(const xAOD::TauJet &tau, double &out)
Definition: TauGNNUtils.cxx:330
TauGNNUtils::Variables::Track::z0sinthetaTJVA
bool z0sinthetaTJVA(const xAOD::TauJet &, const xAOD::TauTrack &track, double &out)
Definition: TauGNNUtils.cxx:518
TauGNNUtils::Variables::Track::z0sinthetaSigTJVA
bool z0sinthetaSigTJVA(const xAOD::TauJet &, const xAOD::TauTrack &track, double &out)
Definition: TauGNNUtils.cxx:523
TauGNNUtils::Variables::Track::nInnermostPixelHits
bool nInnermostPixelHits(const xAOD::TauJet &, const xAOD::TauTrack &track, double &out)
Definition: TauGNNUtils.cxx:560
TauGNNUtils::Variables::Cluster::SECOND_LAMBDA
bool SECOND_LAMBDA(const xAOD::TauJet &, const xAOD::CaloVertexedTopoCluster &cluster, double &out)
Definition: TauGNNUtils.cxx:846
TauGNNUtils::Variables::mEflowApprox
bool mEflowApprox(const xAOD::TauJet &tau, double &out)
Definition: TauGNNUtils.cxx:316
TauGNNUtils::Variables::Track::trackPt
bool trackPt(const xAOD::TauJet &, const xAOD::TauTrack &track, double &out)
Definition: TauGNNUtils.cxx:483
TauGNNUtils::Variables::Track::dPhiJetSeedAxis
bool dPhiJetSeedAxis(const xAOD::TauJet &tau, const xAOD::TauTrack &track, double &out)
Definition: TauGNNUtils.cxx:554
TauGNNUtils::Variables::SumPtTrkFrac
bool SumPtTrkFrac(const xAOD::TauJet &tau, double &out)
Definition: TauGNNUtils.cxx:294
TauGNNUtils::Variables::Cluster::SECOND_R
bool SECOND_R(const xAOD::TauJet &, const xAOD::CaloVertexedTopoCluster &cluster, double &out)
Definition: TauGNNUtils.cxx:841
TauGNNUtils::Variables::sumEMCellEtOverLeadTrkPt
bool sumEMCellEtOverLeadTrkPt(const xAOD::TauJet &tau, double &out)
Definition: TauGNNUtils.cxx:287
TauGNNUtils::Variables::Track::nPixelHitsPlusDeadSensors
bool nPixelHitsPlusDeadSensors(const xAOD::TauJet &, const xAOD::TauTrack &track, double &out)
Definition: TauGNNUtils.cxx:590
TauGNNUtils::Variables::dRmax
bool dRmax(const xAOD::TauJet &tau, double &out)
Definition: TauGNNUtils.cxx:323
TauGNNUtils::Variables::EMPOverTrkSysP
bool EMPOverTrkSysP(const xAOD::TauJet &tau, double &out)
Definition: TauGNNUtils.cxx:301
TauGNNUtils::Variables::etOverPtLeadTrk
bool etOverPtLeadTrk(const xAOD::TauJet &tau, double &out)
Definition: TauGNNUtils.cxx:266
TauGNNUtils::Variables::Track::pt_log
bool pt_log(const xAOD::TauJet &, const xAOD::TauTrack &track, double &out)
Definition: TauGNNUtils.cxx:478
TauGNNUtils::Variables::Track::nSCTHitsPlusDeadSensors
bool nSCTHitsPlusDeadSensors(const xAOD::TauJet &, const xAOD::TauTrack &track, double &out)
Definition: TauGNNUtils.cxx:598
TauGNNUtils::Variables::pt
bool pt(const xAOD::TauJet &tau, double &out)
Definition: TauGNNUtils.cxx:344
TauGNNUtils::Variables::Track::trackPhi
bool trackPhi(const xAOD::TauJet &, const xAOD::TauTrack &track, double &out)
Definition: TauGNNUtils.cxx:493
TauGNNUtils::Variables::Track::d0TJVA
bool d0TJVA(const xAOD::TauJet &, const xAOD::TauTrack &track, double &out)
Definition: TauGNNUtils.cxx:528
TauGNNUtils::Variables::Track::z0sinThetaTJVA_abs_log
bool z0sinThetaTJVA_abs_log(const xAOD::TauJet &, const xAOD::TauTrack &track, double &out)
Definition: TauGNNUtils.cxx:513
beamspotnt.calc
calc
Definition: bin/beamspotnt.py:1251
TauGNNUtils::Variables::Track::numberOfInnermostPixelLayerHits
bool numberOfInnermostPixelLayerHits(const xAOD::TauJet &, const xAOD::TauTrack &track, double &out)
Definition: TauGNNUtils.cxx:653
TauGNNUtils::Variables::Cluster::dEta
bool dEta(const xAOD::TauJet &tau, const xAOD::CaloVertexedTopoCluster &cluster, double &out)
Definition: TauGNNUtils.cxx:831
TauGNNUtils::Variables::Track::d0SigTJVA
bool d0SigTJVA(const xAOD::TauJet &, const xAOD::TauTrack &track, double &out)
Definition: TauGNNUtils.cxx:533
TauGNNUtils::Variables::Track::trackEta
bool trackEta(const xAOD::TauJet &, const xAOD::TauTrack &track, double &out)
Definition: TauGNNUtils.cxx:488
Generated on Mon Sep 1 2025 22:09:14 for ATLAS Offline Software by   doxygen 1.8.18