InDetGNNHardScatterSelection::TracksLoader Class Reference

#include <TracksLoader.h>

Inheritance diagram for InDetGNNHardScatterSelection::TracksLoader:

Collaboration diagram for InDetGNNHardScatterSelection::TracksLoader:

Public Member Functions
	TracksLoader (const ConstituentsInputConfig &cfg)
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::TrackParticle * >	TrackParticles
typedef std::vector< const xAOD::IParticle * >	Particles
typedef std::function< double(const xAOD::TrackParticle *, const Vertex &)>	TrackParticleSortVar
using	AE = SG::AuxElement
using	IPC = xAOD::TrackParticleContainer
using	PartLinks = std::vector< ElementLink< IPC > >
using	IPV = std::vector< const xAOD::TrackParticle * >

Protected Member Functions
std::vector< const xAOD::TrackParticle * >	getTrackParticlesFromVertex (const xAOD::Vertex &vertex) const

Static Protected Member Functions
static TrackParticleSortVar	iparticleSortVar (ConstituentsSortOrder)

Protected Attributes
TrackParticleSortVar	m_iparticleSortVar
getter_utils::CustomSequenceGetter< xAOD::TrackParticle >	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/InDetGNNHardScatterSelection/InDetGNNHardScatterSelection/TracksLoader.h.

Member Typedef Documentation

AE

using InDetGNNHardScatterSelection::TracksLoader::AE = SG::AuxElement

protected

Definition at line 51 of file InnerDetector/InDetRecTools/InDetGNNHardScatterSelection/InDetGNNHardScatterSelection/TracksLoader.h.

IPC

using InDetGNNHardScatterSelection::TracksLoader::IPC = xAOD::TrackParticleContainer

protected

Definition at line 52 of file InnerDetector/InDetRecTools/InDetGNNHardScatterSelection/InDetGNNHardScatterSelection/TracksLoader.h.

IPV

using InDetGNNHardScatterSelection::TracksLoader::IPV = std::vector<const xAOD::TrackParticle*>

protected

Definition at line 54 of file InnerDetector/InDetRecTools/InDetGNNHardScatterSelection/InDetGNNHardScatterSelection/TracksLoader.h.

NamedSeq

typedef std::pair<std::string, std::vector<double> > InDetGNNHardScatterSelection::TracksLoader::NamedSeq

protected

Definition at line 43 of file InnerDetector/InDetRecTools/InDetGNNHardScatterSelection/InDetGNNHardScatterSelection/TracksLoader.h.

NamedVar

typedef std::pair<std::string, double> InDetGNNHardScatterSelection::TracksLoader::NamedVar

protected

Definition at line 42 of file InnerDetector/InDetRecTools/InDetGNNHardScatterSelection/InDetGNNHardScatterSelection/TracksLoader.h.

Particles

typedef std::vector<const xAOD::IParticle*> InDetGNNHardScatterSelection::TracksLoader::Particles

protected

Definition at line 46 of file InnerDetector/InDetRecTools/InDetGNNHardScatterSelection/InDetGNNHardScatterSelection/TracksLoader.h.

PartLinks

using InDetGNNHardScatterSelection::TracksLoader::PartLinks = std::vector<ElementLink<IPC> >

protected

Definition at line 53 of file InnerDetector/InDetRecTools/InDetGNNHardScatterSelection/InDetGNNHardScatterSelection/TracksLoader.h.

TrackParticles

typedef std::vector<const xAOD::TrackParticle*> InDetGNNHardScatterSelection::TracksLoader::TrackParticles

protected

Definition at line 45 of file InnerDetector/InDetRecTools/InDetGNNHardScatterSelection/InDetGNNHardScatterSelection/TracksLoader.h.

TrackParticleSortVar

typedef std::function<double(const xAOD::TrackParticle*, const Vertex&)> InDetGNNHardScatterSelection::TracksLoader::TrackParticleSortVar

protected

Definition at line 48 of file InnerDetector/InDetRecTools/InDetGNNHardScatterSelection/InDetGNNHardScatterSelection/TracksLoader.h.

Vertex

typedef xAOD::Vertex InDetGNNHardScatterSelection::TracksLoader::Vertex

protected

Definition at line 41 of file InnerDetector/InDetRecTools/InDetGNNHardScatterSelection/InDetGNNHardScatterSelection/TracksLoader.h.

Constructor & Destructor Documentation

TracksLoader()

InDetGNNHardScatterSelection::TracksLoader::TracksLoader

(

const ConstituentsInputConfig &

cfg

)

Definition at line 28 of file InnerDetector/InDetRecTools/InDetGNNHardScatterSelection/Root/TracksLoader.cxx.

     :
        IConstituentsLoader(cfg),
        m_iparticleSortVar(TracksLoader::iparticleSortVar(cfg.order)),
        m_customSequenceGetter(getter_utils::CustomSequenceGetter<xAOD::TrackParticle>(
          cfg.inputs))
    {
        const SG::AuxElement::ConstAccessor<PartLinks> acc("trackParticleLinks");
        m_associator = [acc](const xAOD::Vertex& vertex) -> IPV {
          IPV particles;
          for (const ElementLink<IPC>& link : acc(vertex)){
            if (!link.isValid()) {
              throw std::logic_error("invalid particle link");
            }
            particles.push_back(*link);
          }
          return particles;
        };
        m_name = cfg.name;
    }

Member Function Documentation

getData()

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

overridevirtual

Implements InDetGNNHardScatterSelection::IConstituentsLoader.

Definition at line 67 of file InnerDetector/InDetRecTools/InDetGNNHardScatterSelection/Root/TracksLoader.cxx.

                                      {
        TrackParticles sorted_particles = getTrackParticlesFromVertex(vertex);
        std::vector<const xAOD::IParticle*> sorted_particles_ip;
        for (const auto& p: sorted_particles) {
            sorted_particles_ip.push_back(p);
        }
        return std::make_tuple(m_config.output_name, m_customSequenceGetter.getFeats(vertex, sorted_particles), sorted_particles_ip);
    }

getName()

std::string InDetGNNHardScatterSelection::TracksLoader::getName ( ) const

overridevirtual

Implements InDetGNNHardScatterSelection::IConstituentsLoader.

Definition at line 77 of file InnerDetector/InDetRecTools/InDetGNNHardScatterSelection/Root/TracksLoader.cxx.

                                          {
        return m_name;
    }

getTrackParticlesFromVertex()

std::vector< const xAOD::TrackParticle * > InDetGNNHardScatterSelection::TracksLoader::getTrackParticlesFromVertex ( const xAOD::Vertex &  vertex ) const

protected

Definition at line 50 of file InnerDetector/InDetRecTools/InDetGNNHardScatterSelection/Root/TracksLoader.cxx.

    {
        std::vector<std::pair<double, const xAOD::TrackParticle*>> particles;
        for (const xAOD::TrackParticle *tp : m_associator(vertex)) {
          particles.emplace_back(m_iparticleSortVar(tp, vertex), tp);
        }
        std::sort(particles.begin(), particles.end(), std::greater<>());
        std::vector<const xAOD::TrackParticle*> only_particles;
        only_particles.reserve(particles.size());
for (const auto& particle: particles) {
          only_particles.push_back(particle.second);
        }
        return only_particles;
    }

getType()

ConstituentsType InDetGNNHardScatterSelection::TracksLoader::getType ( ) const

overridevirtual

Implements InDetGNNHardScatterSelection::IConstituentsLoader.

Definition at line 80 of file InnerDetector/InDetRecTools/InDetGNNHardScatterSelection/Root/TracksLoader.cxx.

                                                 {
        return m_config.type;
    }

iparticleSortVar()

TracksLoader::TrackParticleSortVar InDetGNNHardScatterSelection::TracksLoader::iparticleSortVar ( ConstituentsSortOrder  config )

staticprotected

Definition at line 14 of file InnerDetector/InDetRecTools/InDetGNNHardScatterSelection/Root/TracksLoader.cxx.

    {
      using Ip = xAOD::TrackParticle;
      using Vertex = xAOD::Vertex;
      switch(config) {
        case ConstituentsSortOrder::PT_DESCENDING:
          return [](const Ip* tp, const Vertex&) {return tp->pt();};
        default: {
          throw std::logic_error("Unknown sort function");
        }
      }
    } // end of iparticle sort getter

Member Data Documentation

m_associator

std::function<IPV(const Vertex&)> InDetGNNHardScatterSelection::TracksLoader::m_associator

protected

Definition at line 62 of file InnerDetector/InDetRecTools/InDetGNNHardScatterSelection/InDetGNNHardScatterSelection/TracksLoader.h.

m_config

ConstituentsInputConfig InDetGNNHardScatterSelection::IConstituentsLoader::m_config

protectedinherited

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

m_customSequenceGetter

getter_utils::CustomSequenceGetter<xAOD::TrackParticle> InDetGNNHardScatterSelection::TracksLoader::m_customSequenceGetter

protected

Definition at line 61 of file InnerDetector/InDetRecTools/InDetGNNHardScatterSelection/InDetGNNHardScatterSelection/TracksLoader.h.

m_iparticleSortVar

TrackParticleSortVar InDetGNNHardScatterSelection::TracksLoader::m_iparticleSortVar

protected

Definition at line 60 of file InnerDetector/InDetRecTools/InDetGNNHardScatterSelection/InDetGNNHardScatterSelection/TracksLoader.h.

m_name

std::string InDetGNNHardScatterSelection::IConstituentsLoader::m_name

protectedinherited

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

---

The documentation for this class was generated from the following files:

• InnerDetector/InDetRecTools/InDetGNNHardScatterSelection/InDetGNNHardScatterSelection/TracksLoader.h
• InnerDetector/InDetRecTools/InDetGNNHardScatterSelection/Root/TracksLoader.cxx