ATLAS Offline Software
Loading...
Searching...
No Matches
Public Member Functions | Protected Types | Protected Member Functions | Static Protected Member Functions | Protected Attributes | List of all members
InDetPVNNSelection::ElectronsLoader Class Reference

#include <ElectronsLoader.h>

Inheritance diagram for InDetPVNNSelection::ElectronsLoader:
Collaboration diagram for InDetPVNNSelection::ElectronsLoader:

Public Member Functions

 ElectronsLoader (const ConstituentsInputConfig &)
std::tuple< std::string, FlavorTagInference::Inputs, std::vector< const xAOD::IParticle * > > getData (const xAOD::Vertex &vertex) const override
std::string getName () const override
ConstituentsType getType () const override

Protected Types

typedef xAOD::Vertex Vertex
typedef std::pair< std::string, double > NamedVar
typedef std::pair< std::string, std::vector< double > > NamedSeq
typedef std::vector< const xAOD::Electron * > Electrons
typedef std::vector< const xAOD::IParticle * > Particles
typedef std::function< double(const xAOD::Electron *, const Vertex &)> ElectronSortVar
typedef std::function< NamedSeq(const Vertex &, const Electrons &)> SeqFromElectrons
using AE = SG::AuxElement
using IPC = xAOD::ElectronContainer
using PartLinks = std::vector<ElementLink<IPC>>
using IPV = std::vector<const xAOD::Electron*>

Protected Member Functions

std::vector< const xAOD::Electron * > getElectronsFromVertex (const xAOD::Vertex &vertex) const

Static Protected Member Functions

static ElectronSortVar iparticleSortVar (ConstituentsSortOrder)

Protected Attributes

ElectronSortVar m_iparticleSortVar
getter_utils::CustomSequenceGetter< xAOD::Electron > m_customSequenceGetter
std::function< IPV(const Vertex &)> m_associator
ConstituentsInputConfig m_config
std::string m_name

Detailed Description

Definition at line 32 of file InnerDetector/InDetRecTools/InDetPVNNSelection/InDetPVNNSelection/ElectronsLoader.h.

Member Typedef Documentation

◆ AE

using InDetPVNNSelection::ElectronsLoader::AE = SG::AuxElement
protected

Definition at line 54 of file InnerDetector/InDetRecTools/InDetPVNNSelection/InDetPVNNSelection/ElectronsLoader.h.

◆ Electrons

typedef std::vector<const xAOD::Electron*> InDetPVNNSelection::ElectronsLoader::Electrons
protected

Definition at line 45 of file InnerDetector/InDetRecTools/InDetPVNNSelection/InDetPVNNSelection/ElectronsLoader.h.

◆ ElectronSortVar

typedef std::function<double(const xAOD::Electron*, const Vertex&)> InDetPVNNSelection::ElectronsLoader::ElectronSortVar
protected

Definition at line 48 of file InnerDetector/InDetRecTools/InDetPVNNSelection/InDetPVNNSelection/ElectronsLoader.h.

◆ IPC

using InDetPVNNSelection::ElectronsLoader::IPC = xAOD::ElectronContainer
protected

Definition at line 55 of file InnerDetector/InDetRecTools/InDetPVNNSelection/InDetPVNNSelection/ElectronsLoader.h.

◆ IPV

using InDetPVNNSelection::ElectronsLoader::IPV = std::vector<const xAOD::Electron*>
protected

Definition at line 57 of file InnerDetector/InDetRecTools/InDetPVNNSelection/InDetPVNNSelection/ElectronsLoader.h.

◆ NamedSeq

typedef std::pair<std::string, std::vector<double> > InDetPVNNSelection::ElectronsLoader::NamedSeq
protected

Definition at line 43 of file InnerDetector/InDetRecTools/InDetPVNNSelection/InDetPVNNSelection/ElectronsLoader.h.

◆ NamedVar

typedef std::pair<std::string, double> InDetPVNNSelection::ElectronsLoader::NamedVar
protected

Definition at line 42 of file InnerDetector/InDetRecTools/InDetPVNNSelection/InDetPVNNSelection/ElectronsLoader.h.

◆ Particles

typedef std::vector<const xAOD::IParticle*> InDetPVNNSelection::ElectronsLoader::Particles
protected

Definition at line 46 of file InnerDetector/InDetRecTools/InDetPVNNSelection/InDetPVNNSelection/ElectronsLoader.h.

◆ PartLinks

using InDetPVNNSelection::ElectronsLoader::PartLinks = std::vector<ElementLink<IPC>>
protected

Definition at line 56 of file InnerDetector/InDetRecTools/InDetPVNNSelection/InDetPVNNSelection/ElectronsLoader.h.

◆ SeqFromElectrons

typedef std::function<NamedSeq(const Vertex&, const Electrons&)> InDetPVNNSelection::ElectronsLoader::SeqFromElectrons
protected

Definition at line 51 of file InnerDetector/InDetRecTools/InDetPVNNSelection/InDetPVNNSelection/ElectronsLoader.h.

◆ Vertex

typedef xAOD::Vertex InDetPVNNSelection::ElectronsLoader::Vertex
protected

Definition at line 41 of file InnerDetector/InDetRecTools/InDetPVNNSelection/InDetPVNNSelection/ElectronsLoader.h.

Constructor & Destructor Documentation

◆ ElectronsLoader()

InDetPVNNSelection::ElectronsLoader::ElectronsLoader ( const ConstituentsInputConfig & cfg)

Definition at line 27 of file InnerDetector/InDetRecTools/InDetPVNNSelection/Root/ElectronsLoader.cxx.

29 :
30 IConstituentsLoader(cfg),
31 m_iparticleSortVar(ElectronsLoader::iparticleSortVar(cfg.order)),
32 m_customSequenceGetter(getter_utils::CustomSequenceGetter<xAOD::Electron>(
33 cfg.inputs))
34 {
35 const SG::ConstAccessor<PartLinks> acc("electronLinks");
36 m_associator = [acc](const xAOD::Vertex& vertex) -> IPV {
37 IPV particles;
38 for (const ElementLink<IPC>& link : acc(vertex)){
39 if (!link.isValid()) {
40 throw std::logic_error("invalid particle link");
41 }
42 particles.push_back(*link);
43 }
44 return particles;
45 };
46 m_name = cfg.name;
47 }
SG::ConstAccessor
SG::ConstAccessor< T, ALLOC > ConstAccessor
Definition AuxElement.h:570
SG::link
pointer & link(pointer p) const
Return a reference to the link for an element.
xAOD::Vertex
Vertex_v1 Vertex
Define the latest version of the vertex class.
Definition Event/xAOD/xAODTracking/xAODTracking/Vertex.h:16

Member Function Documentation

◆ getData()

std::tuple< std::string, FlavorTagInference::Inputs, std::vector< const xAOD::IParticle * > > InDetPVNNSelection::ElectronsLoader::getData ( const xAOD::Vertex & vertex) const
overridevirtual

Implements InDetPVNNSelection::IConstituentsLoader.

Definition at line 66 of file InnerDetector/InDetRecTools/InDetPVNNSelection/Root/ElectronsLoader.cxx.

67 {
68 Electrons sorted_particles = getElectronsFromVertex(vertex);
69 std::vector<const xAOD::IParticle*> sorted_particles_ip;
70 for (const auto& p: sorted_particles) {
71 sorted_particles_ip.push_back(p);
72 }
73 return std::make_tuple(m_config.output_name, m_customSequenceGetter.getFeats(vertex, sorted_particles), sorted_particles_ip);
74 }
InDetPVNNSelection::ElectronsLoader::getElectronsFromVertex
std::vector< const xAOD::Electron * > getElectronsFromVertex(const xAOD::Vertex &vertex) const
Definition InnerDetector/InDetRecTools/InDetPVNNSelection/Root/ElectronsLoader.cxx:49

◆ getElectronsFromVertex()

std::vector< const xAOD::Electron * > InDetPVNNSelection::ElectronsLoader::getElectronsFromVertex ( const xAOD::Vertex & vertex) const
protected

Definition at line 49 of file InnerDetector/InDetRecTools/InDetPVNNSelection/Root/ElectronsLoader.cxx.

52 {
53 std::vector<std::pair<double, const xAOD::Electron*>> particles;
54 for (const xAOD::Electron *tp : m_associator(vertex)) {
55 particles.emplace_back(m_iparticleSortVar(tp, vertex), tp);
56 }
57 std::sort(particles.begin(), particles.end(), std::greater<>());
58 std::vector<const xAOD::Electron*> only_particles;
59 only_particles.reserve(particles.size());
60for (const auto& particle: particles) {
61 only_particles.push_back(particle.second);
62 }
63 return only_particles;
64 }
Trk::ParticleSwitcher::particle
constexpr ParticleHypothesis particle[PARTICLEHYPOTHESES]
the array of masses
Definition ParticleHypothesis.h:79
std::sort
void sort(typename DataModel_detail::iterator< DVL > beg, typename DataModel_detail::iterator< DVL > end)
Specialization of sort for DataVector/List.
Definition DVL_algorithms.h:554
xAOD::Electron
Electron_v1 Electron
Definition of the current "egamma version".
Definition Electron.h:17

◆ getName()

std::string InDetPVNNSelection::ElectronsLoader::getName ( ) const
overridevirtual

Implements InDetPVNNSelection::IConstituentsLoader.

Definition at line 76 of file InnerDetector/InDetRecTools/InDetPVNNSelection/Root/ElectronsLoader.cxx.

76 {
77 return m_name;
78 }

◆ getType()

ConstituentsType InDetPVNNSelection::ElectronsLoader::getType ( ) const
overridevirtual

Implements InDetPVNNSelection::IConstituentsLoader.

Definition at line 79 of file InnerDetector/InDetRecTools/InDetPVNNSelection/Root/ElectronsLoader.cxx.

79 {
80 return m_config.type;
81 }

◆ iparticleSortVar()

ElectronsLoader::ElectronSortVar InDetPVNNSelection::ElectronsLoader::iparticleSortVar ( ConstituentsSortOrder config)
staticprotected

Definition at line 13 of file InnerDetector/InDetRecTools/InDetPVNNSelection/Root/ElectronsLoader.cxx.

15 {
16 using Ip = xAOD::Electron;
17 using Vertex = xAOD::Vertex;
18 switch(config) {
19 case ConstituentsSortOrder::PT_DESCENDING:
20 return [](const Ip* tp, const Vertex&) {return tp->pt();};
21 default: {
22 throw std::logic_error("Unknown sort function");
23 }
24 }
25 } // end of iparticle sort getter

Member Data Documentation

◆ m_associator

std::function<IPV(const Vertex&)> InDetPVNNSelection::ElectronsLoader::m_associator
protected

Definition at line 65 of file InnerDetector/InDetRecTools/InDetPVNNSelection/InDetPVNNSelection/ElectronsLoader.h.

◆ m_config

ConstituentsInputConfig InDetPVNNSelection::IConstituentsLoader::m_config
protectedinherited

Definition at line 71 of file InnerDetector/InDetRecTools/InDetPVNNSelection/InDetPVNNSelection/ConstituentsLoader.h.

◆ m_customSequenceGetter

getter_utils::CustomSequenceGetter<xAOD::Electron> InDetPVNNSelection::ElectronsLoader::m_customSequenceGetter
protected

Definition at line 64 of file InnerDetector/InDetRecTools/InDetPVNNSelection/InDetPVNNSelection/ElectronsLoader.h.

◆ m_iparticleSortVar

ElectronSortVar InDetPVNNSelection::ElectronsLoader::m_iparticleSortVar
protected

Definition at line 63 of file InnerDetector/InDetRecTools/InDetPVNNSelection/InDetPVNNSelection/ElectronsLoader.h.

◆ m_name

std::string InDetPVNNSelection::IConstituentsLoader::m_name
protectedinherited

Definition at line 72 of file InnerDetector/InDetRecTools/InDetPVNNSelection/InDetPVNNSelection/ConstituentsLoader.h.

The documentation for this class was generated from the following files:
Generated on for ATLAS Offline Software by doxygen 1.17.0