FlavorTagDiscriminants::IParticlesLoader Class Reference

#include <IParticlesLoader.h>

Inheritance diagram for FlavorTagDiscriminants::IParticlesLoader:

Collaboration diagram for FlavorTagDiscriminants::IParticlesLoader:

Public Member Functions
	IParticlesLoader (ConstituentsInputConfig, const FTagOptions &options)
std::tuple< std::string, Inputs, std::vector< const xAOD::IParticle * > >	getData (const xAOD::Jet &jet, [[maybe_unused]] const SG::AuxElement &btag) const override
FTagDataDependencyNames	getDependencies () const override
std::set< std::string >	getUsedRemap () const override
std::string	getName () const override
ConstituentsType	getType () const override

Protected Types
typedef xAOD::Jet	Jet
typedef std::pair< std::string, double >	NamedVar
typedef std::pair< std::string, std::vector< double > >	NamedSeq
typedef std::vector< const xAOD::IParticle * >	IParticles
typedef std::function< double(const xAOD::IParticle *, const Jet &)>	IParticleSortVar
typedef std::function< NamedSeq(const Jet &, const IParticles &)>	SeqFromIParticles
using	AE = SG::AuxElement
using	IPC = xAOD::IParticleContainer
using	PartLinks = std::vector< ElementLink< IPC > >
using	IPV = std::vector< const xAOD::IParticle * >

Protected Member Functions
IParticleSortVar	iparticleSortVar (ConstituentsSortOrder)
std::vector< const xAOD::IParticle * >	getIParticlesFromJet (const xAOD::Jet &jet) const

Protected Attributes
IParticleSortVar	m_iparticleSortVar
getter_utils::SeqGetter< xAOD::IParticle >	m_seqGetter
std::function< IPV(const Jet &)>	m_associator
bool	m_isCharged
FTagDataDependencyNames	m_deps
ConstituentsInputConfig	m_config
std::set< std::string >	m_used_remap
std::string	m_name

Detailed Description

Definition at line 34 of file IParticlesLoader.h.

Member Typedef Documentation

AE

using FlavorTagDiscriminants::IParticlesLoader::AE = SG::AuxElement

protected

Definition at line 58 of file IParticlesLoader.h.

IParticles

typedef std::vector<const xAOD::IParticle*> FlavorTagDiscriminants::IParticlesLoader::IParticles

protected

Definition at line 50 of file IParticlesLoader.h.

IParticleSortVar

typedef std::function<double(const xAOD::IParticle*, const Jet&)> FlavorTagDiscriminants::IParticlesLoader::IParticleSortVar

protected

Definition at line 52 of file IParticlesLoader.h.

IPC

using FlavorTagDiscriminants::IParticlesLoader::IPC = xAOD::IParticleContainer

protected

Definition at line 59 of file IParticlesLoader.h.

IPV

using FlavorTagDiscriminants::IParticlesLoader::IPV = std::vector<const xAOD::IParticle*>

protected

Definition at line 61 of file IParticlesLoader.h.

Jet

typedef xAOD::Jet FlavorTagDiscriminants::IParticlesLoader::Jet

protected

Definition at line 46 of file IParticlesLoader.h.

NamedSeq

typedef std::pair<std::string, std::vector<double> > FlavorTagDiscriminants::IParticlesLoader::NamedSeq

protected

Definition at line 48 of file IParticlesLoader.h.

NamedVar

typedef std::pair<std::string, double> FlavorTagDiscriminants::IParticlesLoader::NamedVar

protected

Definition at line 47 of file IParticlesLoader.h.

PartLinks

using FlavorTagDiscriminants::IParticlesLoader::PartLinks = std::vector<ElementLink<IPC> >

protected

Definition at line 60 of file IParticlesLoader.h.

SeqFromIParticles

typedef std::function<NamedSeq(const Jet&, const IParticles&)> FlavorTagDiscriminants::IParticlesLoader::SeqFromIParticles

protected

Definition at line 55 of file IParticlesLoader.h.

Constructor & Destructor Documentation

IParticlesLoader()

FlavorTagDiscriminants::IParticlesLoader::IParticlesLoader	(	ConstituentsInputConfig	cfg,
		const FTagOptions &	options
	)

Definition at line 27 of file IParticlesLoader.cxx.

     :
        IConstituentsLoader(cfg),
        m_iparticleSortVar(IParticlesLoader::iparticleSortVar(cfg.order)),
        m_seqGetter(getter_utils::SeqGetter<xAOD::IParticle>(
          cfg.inputs, options))
    {
        SG::AuxElement::ConstAccessor<PartLinks> acc("constituentLinks");
        m_associator = [acc](const xAOD::Jet& jet) -> IPV {
          IPV particles;
          for (const ElementLink<IPC>& link : acc(jet)){
            if (!link.isValid()) {
              throw std::logic_error("invalid particle link");
            }
            particles.push_back(*link);
          }
          return particles;
        };
 
        if (cfg.name.find("charged") != std::string::npos){
            m_isCharged = true;
        } else {
            m_isCharged = false;
        }
        m_used_remap = m_seqGetter.getUsedRemap();
        m_name = cfg.name;
    }

Member Function Documentation

getData()

std::tuple< std::string, Inputs, std::vector< const xAOD::IParticle * > > FlavorTagDiscriminants::IParticlesLoader::getData ( const xAOD::Jet &  jet, [[maybe_unused] ] const SG::AuxElement &  btag  ) const

overridevirtual

Implements FlavorTagDiscriminants::IConstituentsLoader.

Definition at line 90 of file IParticlesLoader.cxx.

                                                       {
        IParticles sorted_particles = getIParticlesFromJet(jet);
 
        return std::make_tuple(m_config.output_name, m_seqGetter.getFeats(jet, sorted_particles), sorted_particles);
    }

getDependencies()

FTagDataDependencyNames FlavorTagDiscriminants::IParticlesLoader::getDependencies ( ) const

overridevirtual

Implements FlavorTagDiscriminants::IConstituentsLoader.

Definition at line 98 of file IParticlesLoader.cxx.

                                                                    {
        return m_deps;
    }

getIParticlesFromJet()

std::vector< const xAOD::IParticle * > FlavorTagDiscriminants::IParticlesLoader::getIParticlesFromJet ( const xAOD::Jet &  jet ) const

protected

Definition at line 57 of file IParticlesLoader.cxx.

    {
        std::vector<std::pair<double, const xAOD::IParticle*>> particles;
        for (const xAOD::IParticle *tp : m_associator(jet)) {
          particles.push_back({m_iparticleSortVar(tp, jet), tp});
        }
        std::sort(particles.begin(), particles.end(), std::greater<>());
        std::vector<const xAOD::IParticle*> only_particles;
        for (const auto& particle: particles) {
          auto* flow = dynamic_cast<const xAOD::FlowElement*>(particle.second);
          const xAOD::IParticle* obj = nullptr;
          if (!flow){
            throw std::runtime_error("IParticlesLoader: Dynamic cast to FlowElement failed");
          }
          if ((flow->isCharged() != m_isCharged)) continue;
          else {
            if (m_isCharged){
              obj = flow->chargedObject(0);
            }
            else{
              obj = flow;
            }
          }
          if (!obj){
            continue;
          }
          only_particles.push_back(obj);
        }
        return only_particles;
    }

getName()

std::string FlavorTagDiscriminants::IParticlesLoader::getName ( ) const

overridevirtual

Implements FlavorTagDiscriminants::IConstituentsLoader.

Definition at line 104 of file IParticlesLoader.cxx.

                                              {
        return m_name;
    }

getType()

ConstituentsType FlavorTagDiscriminants::IParticlesLoader::getType ( ) const

overridevirtual

Implements FlavorTagDiscriminants::IConstituentsLoader.

Definition at line 107 of file IParticlesLoader.cxx.

                                                     {
        return m_config.type;
    }

getUsedRemap()

std::set< std::string > FlavorTagDiscriminants::IParticlesLoader::getUsedRemap ( ) const

overridevirtual

Implements FlavorTagDiscriminants::IConstituentsLoader.

Definition at line 101 of file IParticlesLoader.cxx.

                                                           {
        return m_used_remap;
    }

iparticleSortVar()

IParticlesLoader::IParticleSortVar FlavorTagDiscriminants::IParticlesLoader::iparticleSortVar ( ConstituentsSortOrder  config )

protected

Definition at line 13 of file IParticlesLoader.cxx.

    {
      typedef xAOD::IParticle Ip;
      typedef xAOD::Jet Jet;
      switch(config) {
        case ConstituentsSortOrder::PT_DESCENDING:
          return [](const Ip* tp, const Jet&) {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 Jet&)> FlavorTagDiscriminants::IParticlesLoader::m_associator

protected

Definition at line 69 of file IParticlesLoader.h.

m_config

ConstituentsInputConfig FlavorTagDiscriminants::IConstituentsLoader::m_config

protectedinherited

Definition at line 91 of file ConstituentsLoader.h.

m_deps

FTagDataDependencyNames FlavorTagDiscriminants::IConstituentsLoader::m_deps

protectedinherited

Definition at line 90 of file ConstituentsLoader.h.

m_iparticleSortVar

IParticleSortVar FlavorTagDiscriminants::IParticlesLoader::m_iparticleSortVar

protected

Definition at line 67 of file IParticlesLoader.h.

m_isCharged

bool FlavorTagDiscriminants::IParticlesLoader::m_isCharged

protected

Definition at line 70 of file IParticlesLoader.h.

m_name

std::string FlavorTagDiscriminants::IConstituentsLoader::m_name

protectedinherited

Definition at line 93 of file ConstituentsLoader.h.

m_seqGetter

getter_utils::SeqGetter<xAOD::IParticle> FlavorTagDiscriminants::IParticlesLoader::m_seqGetter

protected

Definition at line 68 of file IParticlesLoader.h.

m_used_remap

std::set<std::string> FlavorTagDiscriminants::IConstituentsLoader::m_used_remap

protectedinherited

Definition at line 92 of file ConstituentsLoader.h.

---

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

• IParticlesLoader.h
• IParticlesLoader.cxx