 |
ATLAS Offline Software
|
Go to the documentation of this file.
5 #ifndef PhotonVertexSelection_PhotonVertexSelectionTool_h
6 #define PhotonVertexSelection_PhotonVertexSelectionTool_h
24 #include <onnxruntime_cxx_api.h>
55 this,
"EventInfoContName",
"EventInfo",
"event info key"};
57 this,
"VertexContainer",
"PrimaryVertices",
"Vertex container name"};
67 std::unique_ptr<TMVA::Reader>
m_mva1;
68 std::unique_ptr<TMVA::Reader>
m_mva2;
93 std::tuple<std::shared_ptr<Ort::Session>, Ort::AllocatorWithDefaultOptions>
97 std::tuple<std::vector<int64_t>, std::vector<const char*>>
getInputNodes(
98 const std::shared_ptr<Ort::Session>& sessionHandle,
99 Ort::AllocatorWithDefaultOptions& allocator);
102 std::tuple<std::vector<int64_t>, std::vector<const char*>>
getOutputNodes(
103 const std::shared_ptr<Ort::Session>& sessionHandle,
104 Ort::AllocatorWithDefaultOptions& allocator);
107 float getScore(
int nVars,
const std::vector<std::vector<float>>& input_data,
108 const std::shared_ptr<Ort::Session>& sessionHandle,
109 std::vector<int64_t> input_node_dims,
110 std::vector<const char*> input_node_names,
111 std::vector<const char*> output_node_names)
const;
119 static bool sortMLP(
const std::pair<const xAOD::Vertex*, float> &
a,
const std::pair<const xAOD::Vertex*, float> &
b);
125 std::vector<std::pair<const xAOD::Vertex*, float>>&,
150 std::vector<std::pair<const xAOD::Vertex*, float>>
getVertex(
152 bool ignoreConv =
false,
153 bool noDecorate =
false,
155 FailType* failType =
nullptr)
const;
180 #endif // PhotonVertexSelection_PhotonVertexSelectionTool_h
Property holding a SG store/key/clid/attr name from which a WriteDecorHandle is made.
Select isolated Photons, Electrons and Muons.
::StatusCode StatusCode
StatusCode definition for legacy code.
Class describing a Vertex.