 |
ATLAS Offline Software
|
#include <DiTauOnnxDiscriminantTool.h>
|
virtual void | print () const =0 |
| Print the state of the tool. More...
|
|
virtual void | print () const |
| Print the state of the tool. More...
|
|
ServiceHandle< StoreGateSvc > & | evtStore () |
| The standard StoreGateSvc (event store) Returns (kind of) a pointer to the StoreGateSvc . More...
|
|
const ServiceHandle< StoreGateSvc > & | evtStore () const |
| The standard StoreGateSvc (event store) Returns (kind of) a pointer to the StoreGateSvc . More...
|
|
const ServiceHandle< StoreGateSvc > & | detStore () const |
| The standard StoreGateSvc/DetectorStore Returns (kind of) a pointer to the StoreGateSvc . More...
|
|
virtual StatusCode | sysInitialize () override |
| Perform system initialization for an algorithm. More...
|
|
virtual StatusCode | sysStart () override |
| Handle START transition. More...
|
|
virtual std::vector< Gaudi::DataHandle * > | inputHandles () const override |
| Return this algorithm's input handles. More...
|
|
virtual std::vector< Gaudi::DataHandle * > | outputHandles () const override |
| Return this algorithm's output handles. More...
|
|
Gaudi::Details::PropertyBase & | declareProperty (Gaudi::Property< T, V, H > &t) |
|
Gaudi::Details::PropertyBase * | declareProperty (const std::string &name, SG::VarHandleKey &hndl, const std::string &doc, const SG::VarHandleKeyType &) |
| Declare a new Gaudi property. More...
|
|
Gaudi::Details::PropertyBase * | declareProperty (const std::string &name, SG::VarHandleBase &hndl, const std::string &doc, const SG::VarHandleType &) |
| Declare a new Gaudi property. More...
|
|
Gaudi::Details::PropertyBase * | declareProperty (const std::string &name, SG::VarHandleKeyArray &hndArr, const std::string &doc, const SG::VarHandleKeyArrayType &) |
|
Gaudi::Details::PropertyBase * | declareProperty (const std::string &name, T &property, const std::string &doc, const SG::NotHandleType &) |
| Declare a new Gaudi property. More...
|
|
Gaudi::Details::PropertyBase * | declareProperty (const std::string &name, T &property, const std::string &doc="none") |
| Declare a new Gaudi property. More...
|
|
void | updateVHKA (Gaudi::Details::PropertyBase &) |
|
MsgStream & | msg () const |
|
MsgStream & | msg (const MSG::Level lvl) const |
|
bool | msgLvl (const MSG::Level lvl) const |
|
|
virtual StatusCode | initialize () override |
| Dummy implementation of the initialisation function. More...
|
|
virtual StatusCode | execute (const xAOD::DiTauJet &xDiTau) const override |
| Declare the interface that the class provides. More...
|
|
Ort::Value | create_tensor (std::vector< float > &data, const std::vector< int64_t > &shape) const |
|
InferenceOutput | run_inference (OnnxInputs &inputs) const |
|
float | nan_to_num (float value, float nan_replacement, float posinf_replacement, float neginf_replacement) const |
|
std::vector< float > | flatten (const std::vector< std::vector< float >> &vec_2d) const |
|
std::vector< float > | extract_points (const std::vector< std::vector< float >> &track_features) const |
|
std::vector< float > | create_mask (const std::vector< std::vector< float >> &track_features) const |
|
float | GetDiTauObjOnnxScore (const xAOD::DiTauJet &ditau) const |
|
Gaudi::Details::PropertyBase & | declareGaudiProperty (Gaudi::Property< T, V, H > &hndl, const SG::VarHandleKeyType &) |
| specialization for handling Gaudi::Property<SG::VarHandleKey> More...
|
|
Gaudi::Details::PropertyBase & | declareGaudiProperty (Gaudi::Property< T, V, H > &hndl, const SG::VarHandleKeyArrayType &) |
| specialization for handling Gaudi::Property<SG::VarHandleKeyArray> More...
|
|
Gaudi::Details::PropertyBase & | declareGaudiProperty (Gaudi::Property< T, V, H > &hndl, const SG::VarHandleType &) |
| specialization for handling Gaudi::Property<SG::VarHandleBase> More...
|
|
Gaudi::Details::PropertyBase & | declareGaudiProperty (Gaudi::Property< T, V, H > &t, const SG::NotHandleType &) |
| specialization for handling everything that's not a Gaudi::Property<SG::VarHandleKey> or a <SG::VarHandleKeyArray> More...
|
|
|
Gaudi::Property< std::string > | m_onnxModelPath {this, "onnxModelPath", "TrigTauRec/00-11-02/dev/boosted_ditau_omni_model.onnx"} |
|
Gaudi::Property< size_t > | m_maxTracks {this, "maxTracks", 10} |
|
std::unique_ptr< Ort::Env > | m_ort_env |
|
std::unique_ptr< Ort::Session > | m_ort_session |
|
const std::vector< std::string > | m_input_node_names = {"input_features", "input_points", "input_mask", "input_jet", "input_time"} |
|
const std::vector< std::string > | m_output_node_names = {"output_1", "output_2"} |
|
SG::ReadHandleKey< xAOD::DiTauJetContainer > | m_ditauContainerKey {this, "DiTauContainerName", "DiTauJets", "DiTau container name"} |
|
SG::ReadDecorHandleKey< xAOD::DiTauJetContainer > | m_ditau_pt_DecorKey {this, "DiTauPtDecorName", "DiTauJets.ditau_pt", "Name of the DiTau Pt decoration"} |
|
SG::ReadDecorHandleKey< xAOD::DiTauJetContainer > | m_f_core_lead_DecorKey {this, "DiTauFCoreLeadName", "DiTauJets.f_core_lead", "Name of the Ditau leading subjet core energy fraction decoration"} |
|
SG::ReadDecorHandleKey< xAOD::DiTauJetContainer > | m_f_core_sublead_DecorKey {this, "DiTauFCoreSubLeadName", "DiTauJets.f_core_subl", "Name of the Ditau subleading subjet core energy fraction decoration"} |
|
SG::ReadDecorHandleKey< xAOD::DiTauJetContainer > | m_f_subjet_subl_DecorKey {this, "DiTauSubjetSublName", "DiTauJets.f_subjet_subl", "Name of the Ditau subleading subjet pt fraction decoration"} |
|
SG::ReadDecorHandleKey< xAOD::DiTauJetContainer > | m_f_subjets_DecorKey {this, "DiTauSubjetsName", "DiTauJets.f_subjets", "Name of the DiTau subjets fraction decoration"} |
|
SG::ReadDecorHandleKey< xAOD::DiTauJetContainer > | m_R_max_lead_DecorKey {this, "DiTauRMaxLeadName", "DiTauJets.R_max_lead", "Name of the Ditau Max dR distance track from leading subjet decoration"} |
|
SG::ReadDecorHandleKey< xAOD::DiTauJetContainer > | m_R_max_sublead_DecorKey {this, "DiTauRMaxSubleadName", "DiTauJets.R_max_subl", "Name of the Ditau Max dR distance track from subleading subjet decoration"} |
|
SG::ReadDecorHandleKey< xAOD::DiTauJetContainer > | m_n_track_DecorKey {this, "DiTauNTrackName", "DiTauJets.n_track", "Name of the Ditau number of tracks decoration"} |
|
SG::ReadDecorHandleKey< xAOD::DiTauJetContainer > | m_R_track_all_DecorKey { this, "DiTauRTrackAllName", "DiTauJets.R_track_all", "Name of the Ditau DeltaR tracks over pt in the large region decoration"} |
|
SG::ReadDecorHandleKey< xAOD::DiTauJetContainer > | m_R_isotrack_DecorKey { this, "DiTauRIsoTrackAllName", "DiTauJets.R_isotrack", "Name of the Ditau DeltaR isolated tracks over pt decoration"} |
|
SG::ReadDecorHandleKey< xAOD::DiTauJetContainer > | m_R_track_sublead_DecorKey { this, "DiTauRTrackSubleadName", "DiTauJets.R_tracks_subl", "Name of the Ditau DeltaR tracks over pt in the large region of the subleading subjet decoration"} |
|
SG::ReadDecorHandleKey< xAOD::DiTauJetContainer > | m_M_core_lead_DecorKey { this, "DiTauMCoreLeadName", "DiTauJets.m_core_lead", "Name of the Ditau mass of tracks in the core region of the leading subjet decoration"} |
|
SG::ReadDecorHandleKey< xAOD::DiTauJetContainer > | m_M_core_sublead_DecorKey { this, "DiTauMCoreSubleadName", "DiTauJets.m_core_subl", "Name of the Ditau mass of tracks in the core region of the leading subjet decoration"} |
|
SG::ReadDecorHandleKey< xAOD::DiTauJetContainer > | m_M_track_lead_DecorKey { this, "DiTauMTrackLeadName", "DiTauJets.m_tracks_lead", "Name of the Ditau mass of tracks in the leading subjet decoration"} |
|
SG::ReadDecorHandleKey< xAOD::DiTauJetContainer > | m_d0_leadtrack_lead_DecorKey { this, "DiTauD0LeadTrackLeadName", "DiTauJets.d0_leadtrack_lead", "Name of the DiTau dR between the leading track within the lead subjet with respect to the lead subjet"} |
|
SG::ReadDecorHandleKey< xAOD::DiTauJetContainer > | m_d0_leadtrack_sublead_DecorKey { this, "DiTauD0SubleadTrackLeadName", "DiTauJets.d0_leadtrack_subl", "Name of the DiTau dR between the leading track within the sublead subjet with respect to the sublead subjet"} |
|
SG::ReadDecorHandleKey< xAOD::DiTauJetContainer > | m_f_isotracks_DecorKey { this, "DiTauFIsotracks", "DiTauJets.f_isotracks", "Name of the DiTau energy fraction carried by isolated tracks"} |
|
StoreGateSvc_t | m_evtStore |
| Pointer to StoreGate (event store by default) More...
|
|
StoreGateSvc_t | m_detStore |
| Pointer to StoreGate (detector store by default) More...
|
|
std::vector< SG::VarHandleKeyArray * > | m_vhka |
|
bool | m_varHandleArraysDeclared |
|
◆ StoreGateSvc_t
◆ create_mask()
std::vector< float > DiTauOnnxDiscriminantTool::create_mask |
( |
const std::vector< std::vector< float >> & |
track_features | ) |
const |
|
private |
◆ create_tensor()
Ort::Value DiTauOnnxDiscriminantTool::create_tensor |
( |
std::vector< float > & |
data, |
|
|
const std::vector< int64_t > & |
shape |
|
) |
| const |
|
private |
Definition at line 112 of file ools/Root/DiTauOnnxDiscriminantTool.cxx.
113 Ort::MemoryInfo memory_info = Ort::MemoryInfo::CreateCpu(OrtArenaAllocator, OrtMemTypeDefault);
114 return Ort::Value::CreateTensor<float>(memory_info,
data.data(),
data.size(),shape.data(), shape.size());
◆ declareGaudiProperty() [1/4]
specialization for handling Gaudi::Property<SG::VarHandleKeyArray>
Definition at line 170 of file AthCommonDataStore.h.
175 hndl.documentation());
◆ declareGaudiProperty() [2/4]
specialization for handling Gaudi::Property<SG::VarHandleKey>
Definition at line 156 of file AthCommonDataStore.h.
161 hndl.documentation());
◆ declareGaudiProperty() [3/4]
specialization for handling Gaudi::Property<SG::VarHandleBase>
Definition at line 184 of file AthCommonDataStore.h.
189 hndl.documentation());
◆ declareGaudiProperty() [4/4]
◆ declareProperty() [1/6]
Declare a new Gaudi property.
- Parameters
-
name | Name of the property. |
hndl | Object holding the property value. |
doc | Documentation string for the property. |
This is the version for types that derive from SG::VarHandleBase
. The property value object is put on the input and output lists as appropriate; then we forward to the base class.
Definition at line 245 of file AthCommonDataStore.h.
250 this->declare(hndl.
vhKey());
251 hndl.
vhKey().setOwner(
this);
253 return PBASE::declareProperty(
name,hndl,
doc);
◆ declareProperty() [2/6]
Declare a new Gaudi property.
- Parameters
-
name | Name of the property. |
hndl | Object holding the property value. |
doc | Documentation string for the property. |
This is the version for types that derive from SG::VarHandleKey
. The property value object is put on the input and output lists as appropriate; then we forward to the base class.
Definition at line 221 of file AthCommonDataStore.h.
229 return PBASE::declareProperty(
name,hndl,
doc);
◆ declareProperty() [3/6]
◆ declareProperty() [4/6]
Declare a new Gaudi property.
- Parameters
-
name | Name of the property. |
property | Object holding the property value. |
doc | Documentation string for the property. |
This is the generic version, for types that do not derive from SG::VarHandleKey
. It just forwards to the base class version of declareProperty
.
Definition at line 333 of file AthCommonDataStore.h.
338 return PBASE::declareProperty(
name, property,
doc);
◆ declareProperty() [5/6]
Declare a new Gaudi property.
- Parameters
-
name | Name of the property. |
property | Object holding the property value. |
doc | Documentation string for the property. |
This dispatches to either the generic declareProperty
or the one for VarHandle/Key/KeyArray.
Definition at line 352 of file AthCommonDataStore.h.
◆ declareProperty() [6/6]
◆ detStore()
◆ evtStore() [1/2]
◆ evtStore() [2/2]
◆ execute()
◆ extract_points()
std::vector< float > DiTauOnnxDiscriminantTool::extract_points |
( |
const std::vector< std::vector< float >> & |
track_features | ) |
const |
|
private |
◆ extraDeps_update_handler()
Add StoreName to extra input/output deps as needed.
use the logic of the VarHandleKey to parse the DataObjID keys supplied via the ExtraInputs and ExtraOuputs Properties to add the StoreName if it's not explicitly given
◆ flatten()
std::vector< float > DiTauOnnxDiscriminantTool::flatten |
( |
const std::vector< std::vector< float >> & |
vec_2d | ) |
const |
|
private |
◆ GetDiTauObjOnnxScore()
Definition at line 146 of file ools/Root/DiTauOnnxDiscriminantTool.cxx.
173 std::vector<float> jet_vars = {
174 R_max_leadDec (ditau),
175 R_max_subleadDec (ditau),
176 R_tracks_sublDec (ditau),
177 R_isotracDec (ditau),
178 d0_leadtrack_leadDec (ditau),
179 d0_leadtrack_sublDec (ditau),
180 f_core_leadDec (ditau),
181 f_core_sublDec (ditau),
182 f_subjet_sublDec (ditau),
184 f_isotracks_Dec (ditau),
185 M_core_leadDec (ditau),
186 M_core_sublDec (ditau),
187 M_tracks_leadDec (ditau),
188 static_cast<float>( n_trackDec (ditau)),
190 std::vector<int64_t> jet_shape = {1,
static_cast<int64_t
>(jet_vars.size())};
193 std::vector<std::vector<float>> track_features(
m_maxTracks, std::vector<float>(11, 0.0
f));
195 float jet_eta = ditau.
eta();
199 for (
size_t i = 0;
i < num_tracks; ++
i) {
201 if (!trackLink.
isValid())
continue;
203 float track_eta = xTrack->
eta();
204 float track_phi = xTrack->
phi();
205 float delta_eta = track_eta - jet_eta;
208 float track_pt =
static_cast<float>(xTrack->
pt());
210 float jet_pt = ditau_ptDec(ditau);
211 float pt_ratio = track_pt / jet_pt;
212 float pt_ratio_log =
std::log(1.0
f - pt_ratio + 1
e-8
f);
213 float track_charge = xTrack->
charge();
214 float pt_ratio_log_nan_less =
nan_to_num(pt_ratio_log, 0.0
f, 0.0
f, 0.0
f);
216 track_features[
i] = {
221 pt_ratio_log_nan_less,
224 static_cast<float>(numberOfInrmstPxlLyrHitsAcc(*xTrack)),
225 static_cast<float>(numberOfPixelHitsAcc(*xTrack)),
226 static_cast<float>(numberOfSCTHitsAcc(*xTrack)),
230 std::vector<int64_t> track_shape = {1,
static_cast<int64_t
>(
m_maxTracks), 11};
237 {1, track_shape[1], 2},
241 std::move(jet_shape),
246 return output.output_1[1];
◆ getKey()
Get the (hashed) key of an object that is in the event store.
This is a bit of a special one. StoreGateSvc
and xAOD::TEvent
both provide ways for getting the SG::sgkey_t
key for an object that is in the store, based on a bare pointer. But they provide different interfaces for doing so.
In order to allow tools to efficiently perform this operation, they can use this helper function.
- See also
- asg::AsgTool::getName
- Parameters
-
ptr | The bare pointer to the object that the event store should know about |
- Returns
- The hashed key of the object in the store. If not found, an invalid (zero) key.
Definition at line 119 of file AsgTool.cxx.
121 #ifdef XAOD_STANDALONE
127 return (
proxy ==
nullptr ? 0 :
proxy->sgkey() );
128 #endif // XAOD_STANDALONE
◆ getName()
const std::string & asg::AsgTool::getName |
( |
const void * |
ptr | ) |
const |
|
inherited |
Get the name of an object that is / should be in the event store.
This is a bit of a special one. StoreGateSvc
and xAOD::TEvent
both provide ways for getting the std::string
name for an object that is in the store, based on a bare pointer. But they provide different interfaces for doing so.
In order to allow tools to efficiently perform this operation, they can use this helper function.
- See also
- asg::AsgTool::getKey
- Parameters
-
ptr | The bare pointer to the object that the event store should know about |
- Returns
- The string name of the object in the store. If not found, an empty string.
Definition at line 106 of file AsgTool.cxx.
108 #ifdef XAOD_STANDALONE
114 static const std::string
dummy =
"";
116 #endif // XAOD_STANDALONE
◆ getProperty()
template<class T >
const T* asg::AsgTool::getProperty |
( |
const std::string & |
name | ) |
const |
|
inherited |
Get one of the tool's properties.
◆ initialize()
StatusCode DiTauOnnxDiscriminantTool::initialize |
( |
| ) |
|
|
overrideprivatevirtual |
Dummy implementation of the initialisation function.
It's here to allow the dual-use tools to skip defining an initialisation function. Since many are doing so...
Reimplemented from asg::AsgTool.
Definition at line 27 of file ools/Root/DiTauOnnxDiscriminantTool.cxx.
29 ATH_MSG_INFO(
"Initializing DiTauOnnxDiscriminantTool" );
32 if (model_path.empty()) {
34 return StatusCode::FAILURE;
36 m_ort_env = std::make_unique<Ort::Env>(ORT_LOGGING_LEVEL_WARNING,
"OnnxUtil");
37 Ort::SessionOptions session_options;
38 session_options.SetIntraOpNumThreads(1);
39 session_options.SetGraphOptimizationLevel(GraphOptimizationLevel::ORT_ENABLE_EXTENDED);
61 return StatusCode::SUCCESS;
◆ inputHandles()
Return this algorithm's input handles.
We override this to include handle instances from key arrays if they have not yet been declared. See comments on updateVHKA.
◆ msg() [1/2]
◆ msg() [2/2]
◆ msg_level_name()
const std::string & asg::AsgTool::msg_level_name |
( |
| ) |
const |
|
inherited |
A deprecated function for getting the message level's name.
Instead of using this, weirdly named function, user code should get the string name of the current minimum message level (in case they really need it...), with:
MSG::name( msg().level() )
This function's name doesn't follow the ATLAS coding rules, and as such will be removed in the not too distant future.
- Returns
- The string name of the current minimum message level that's printed
Definition at line 101 of file AsgTool.cxx.
◆ msgLvl()
◆ nan_to_num()
float DiTauOnnxDiscriminantTool::nan_to_num |
( |
float |
value, |
|
|
float |
nan_replacement = 0.0f , |
|
|
float |
posinf_replacement = 0.0f , |
|
|
float |
neginf_replacement = 0.0f |
|
) |
| const |
|
private |
Definition at line 74 of file ools/Root/DiTauOnnxDiscriminantTool.cxx.
75 if (std::isnan(
value))
76 return nan_replacement;
77 if (
value == std::numeric_limits<float>::infinity())
78 return posinf_replacement;
79 if (
value == -std::numeric_limits<float>::infinity())
80 return neginf_replacement;
◆ outputHandles()
Return this algorithm's output handles.
We override this to include handle instances from key arrays if they have not yet been declared. See comments on updateVHKA.
◆ print() [1/2]
void asg::AsgTool::print |
( |
| ) |
const |
|
virtualinherited |
Print the state of the tool.
Implements asg::IAsgTool.
Reimplemented in JetRecTool, JetFinder, JetModifiedMassDrop, JetFromPseudojet, JetReclusterer, JetReclusteringTool, JetTruthLabelingTool, JetPileupLabelingTool, HI::HIPileupTool, LundVariablesTool, JetDumper, JetBottomUpSoftDrop, JetRecursiveSoftDrop, JetSoftDrop, JetConstituentsRetriever, JetSubStructureMomentToolsBase, JetSplitter, JetToolRunner, JetPruner, JetPseudojetRetriever, JetTrimmer, AsgHelloTool, and KtDeltaRTool.
Definition at line 131 of file AsgTool.cxx.
◆ print() [2/2]
virtual void asg::IAsgTool::print |
( |
| ) |
const |
|
pure virtualinherited |
Print the state of the tool.
Implemented in JetRecTool, JetFinder, JetModifiedMassDrop, JetFromPseudojet, JetReclusterer, JetReclusteringTool, JetTruthLabelingTool, JetPileupLabelingTool, HI::HIPileupTool, LundVariablesTool, asg::AsgTool, JetDumper, JetBottomUpSoftDrop, JetRecursiveSoftDrop, JetSoftDrop, JetConstituentsRetriever, JetSubStructureMomentToolsBase, JetSplitter, JetToolRunner, JetPruner, JetPseudojetRetriever, JetTrimmer, AsgHelloTool, and KtDeltaRTool.
◆ renounce()
◆ renounceArray()
◆ run_inference()
Definition at line 117 of file ools/Root/DiTauOnnxDiscriminantTool.cxx.
118 std::vector<Ort::Value> input_tensors;
126 std::vector<const char *> input_node_names;
130 std::vector<const char *> output_node_names;
134 auto output_tensors =
m_ort_session->Run(Ort::RunOptions{
nullptr}, input_node_names.data(), input_tensors.data(), input_node_names.size(), output_node_names.data(), output_node_names.size());
137 for (
size_t i = 0;
i < output_tensors.size(); ++
i) {
138 const auto &tensor = output_tensors[
i];
139 const size_t length = tensor.GetTensorTypeAndShapeInfo().GetElementCount();
140 const float *
data = tensor.GetTensorData<
float>();
◆ sysInitialize()
◆ sysStart()
Handle START transition.
We override this in order to make sure that conditions handle keys can cache a pointer to the conditions container.
◆ updateVHKA()
◆ m_d0_leadtrack_lead_DecorKey
SG::ReadDecorHandleKey<xAOD::DiTauJetContainer> DiTauRecTools::DiTauOnnxDiscriminantTool::m_d0_leadtrack_lead_DecorKey { this, "DiTauD0LeadTrackLeadName", "DiTauJets.d0_leadtrack_lead", "Name of the DiTau dR between the leading track within the lead subjet with respect to the lead subjet"} |
|
private |
◆ m_d0_leadtrack_sublead_DecorKey
SG::ReadDecorHandleKey<xAOD::DiTauJetContainer> DiTauRecTools::DiTauOnnxDiscriminantTool::m_d0_leadtrack_sublead_DecorKey { this, "DiTauD0SubleadTrackLeadName", "DiTauJets.d0_leadtrack_subl", "Name of the DiTau dR between the leading track within the sublead subjet with respect to the sublead subjet"} |
|
private |
◆ m_detStore
◆ m_ditau_pt_DecorKey
◆ m_ditauContainerKey
◆ m_evtStore
◆ m_f_core_lead_DecorKey
◆ m_f_core_sublead_DecorKey
◆ m_f_isotracks_DecorKey
◆ m_f_subjet_subl_DecorKey
◆ m_f_subjets_DecorKey
◆ m_input_node_names
const std::vector<std::string> DiTauRecTools::DiTauOnnxDiscriminantTool::m_input_node_names = {"input_features", "input_points", "input_mask", "input_jet", "input_time"} |
|
private |
◆ m_M_core_lead_DecorKey
SG::ReadDecorHandleKey<xAOD::DiTauJetContainer> DiTauRecTools::DiTauOnnxDiscriminantTool::m_M_core_lead_DecorKey { this, "DiTauMCoreLeadName", "DiTauJets.m_core_lead", "Name of the Ditau mass of tracks in the core region of the leading subjet decoration"} |
|
private |
◆ m_M_core_sublead_DecorKey
SG::ReadDecorHandleKey<xAOD::DiTauJetContainer> DiTauRecTools::DiTauOnnxDiscriminantTool::m_M_core_sublead_DecorKey { this, "DiTauMCoreSubleadName", "DiTauJets.m_core_subl", "Name of the Ditau mass of tracks in the core region of the leading subjet decoration"} |
|
private |
◆ m_M_track_lead_DecorKey
◆ m_maxTracks
Gaudi::Property<size_t> DiTauRecTools::DiTauOnnxDiscriminantTool::m_maxTracks {this, "maxTracks", 10} |
|
private |
◆ m_n_track_DecorKey
◆ m_onnxModelPath
Gaudi::Property<std::string> DiTauRecTools::DiTauOnnxDiscriminantTool::m_onnxModelPath {this, "onnxModelPath", "TrigTauRec/00-11-02/dev/boosted_ditau_omni_model.onnx"} |
|
private |
◆ m_ort_env
std::unique_ptr<Ort::Env> DiTauRecTools::DiTauOnnxDiscriminantTool::m_ort_env |
|
private |
◆ m_ort_session
std::unique_ptr<Ort::Session> DiTauRecTools::DiTauOnnxDiscriminantTool::m_ort_session |
|
private |
◆ m_output_node_names
const std::vector<std::string> DiTauRecTools::DiTauOnnxDiscriminantTool::m_output_node_names = {"output_1", "output_2"} |
|
private |
◆ m_R_isotrack_DecorKey
◆ m_R_max_lead_DecorKey
◆ m_R_max_sublead_DecorKey
SG::ReadDecorHandleKey<xAOD::DiTauJetContainer> DiTauRecTools::DiTauOnnxDiscriminantTool::m_R_max_sublead_DecorKey {this, "DiTauRMaxSubleadName", "DiTauJets.R_max_subl", "Name of the Ditau Max dR distance track from subleading subjet decoration"} |
|
private |
◆ m_R_track_all_DecorKey
◆ m_R_track_sublead_DecorKey
SG::ReadDecorHandleKey<xAOD::DiTauJetContainer> DiTauRecTools::DiTauOnnxDiscriminantTool::m_R_track_sublead_DecorKey { this, "DiTauRTrackSubleadName", "DiTauJets.R_tracks_subl", "Name of the Ditau DeltaR tracks over pt in the large region of the subleading subjet decoration"} |
|
private |
◆ m_varHandleArraysDeclared
◆ m_vhka
The documentation for this class was generated from the following files:
virtual double pt() const override final
The transverse momentum ( ) of the particle.
char data[hepevt_bytes_allocation_ATLAS]
float charge() const
Returns the charge.
virtual double eta() const override final
The pseudorapidity ( ) of the particle.
def delta_R(eta1, phi1, eta2, phi2)
StoreGateSvc_t m_evtStore
Pointer to StoreGate (event store by default)
std::vector< SG::VarHandleKeyArray * > m_vhka
virtual double eta() const
The pseudorapidity ( ) of the particle.
bool isValid() const
Test to see if the link can be dereferenced.
virtual void setOwner(IDataHandleHolder *o)=0
Gaudi::Details::PropertyBase & declareGaudiProperty(Gaudi::Property< T, V, H > &hndl, const SG::VarHandleKeyType &)
specialization for handling Gaudi::Property<SG::VarHandleKey>
ServiceHandle< StoreGateSvc > & evtStore()
The standard StoreGateSvc (event store) Returns (kind of) a pointer to the StoreGateSvc.
virtual double phi() const
The azimuthal angle ( ) of the particle.
Handle class for reading a decoration on an object.
Gaudi::Details::PropertyBase & declareProperty(Gaudi::Property< T, V, H > &t)
Amg::Vector3D transform(Amg::Vector3D &v, Amg::Transform3D &tr)
Transform a point from a Trasformation3D.
const std::string & name(Level lvl)
Convenience function for translating message levels to strings.
StoreGateSvc_t m_detStore
Pointer to StoreGate (detector store by default)
virtual void renounce()=0
std::conditional< std::is_base_of< SG::VarHandleKeyArray, T >::value, VarHandleKeyArrayType, type2 >::type type
std::vector< std::string > remainder(const std::vector< std::string > &v1, const std::vector< std::string > &v2)
std::string PathResolverFindCalibFile(const std::string &logical_file_name)
bool pt_log(const xAOD::TauJet &, const xAOD::TauTrack &track, double &out)
def delta_phi(phi1, phi2)
SG::VarHandleKey & vhKey()
Return a non-const reference to the HandleKey.
Class describing a TrackParticle.
const TrackParticleLinks_t & trackLinks() const
virtual double phi() const override final
The azimuthal angle ( ) of the particle (has range to .)