HepMC::GenEventBarcodes Class Reference

#include <GenEvent.h>

Inheritance diagram for HepMC::GenEventBarcodes:

Collaboration diagram for HepMC::GenEventBarcodes:

Public Member Functions
virtual bool	from_string (const std::string &) override
virtual bool	to_string (std::string &att) const override
ConstGenVertexPtr	barcode_to_vertex (int id) const
GenVertexPtr	barcode_to_vertex (int id)
ConstGenParticlePtr	barcode_to_particle (int id) const
GenParticlePtr	barcode_to_particle (int id)
void	add (GenVertexPtr p)
void	remove (GenVertexPtr p)
void	add (GenParticlePtr p)
void	remove (GenParticlePtr p)
std::map< int, ConstGenVertexPtr >	barcode_to_vertex_map () const
std::map< int, ConstGenParticlePtr >	barcode_to_particle_map () const
std::map< int, int >	id_to_barcode_map () const
void	fillAttribute (GenEvent *e)

Private Attributes
std::unordered_map< int, GenVertexPtr >	m_vertexBC
std::unordered_map< int, GenParticlePtr >	m_particleBC

Detailed Description

Definition at line 174 of file GenEvent.h.

Member Function Documentation

add() [1/2]

void HepMC::GenEventBarcodes::add ( GenParticlePtr p )

inline

Definition at line 223 of file GenEvent.h.

                              {
    if (!p) return;
    auto barcode = p->attribute<HepMC3::IntAttribute>(HepMCStr::barcode);
    if (barcode) {
      m_particleBC[barcode->value()] = std::move(p);
    }
  }

add() [2/2]

void HepMC::GenEventBarcodes::add ( GenVertexPtr p )

inline

Definition at line 204 of file GenEvent.h.

                            {
    if (!p) return;
    auto barcode = p->attribute<HepMC3::IntAttribute>(HepMCStr::barcode);
    if (barcode) {
      m_vertexBC[barcode->value()] = std::move(p);
    }
  }

barcode_to_particle() [1/2]

GenParticlePtr HepMC::GenEventBarcodes::barcode_to_particle ( int id )

inline

Definition at line 198 of file GenEvent.h.

                                              {
    const auto it = m_particleBC.find (id);
    if (it != m_particleBC.end()) return it->second;
    return nullptr;
  }

barcode_to_particle() [2/2]

ConstGenParticlePtr HepMC::GenEventBarcodes::barcode_to_particle ( int id ) const

inline

Definition at line 193 of file GenEvent.h.

                                                         {
    const auto it = m_particleBC.find (id);
    if (it != m_particleBC.end()) return it->second;
    return nullptr;
  }

barcode_to_particle_map()

std::map< int, ConstGenParticlePtr > HepMC::GenEventBarcodes::barcode_to_particle_map ( ) const

inline

Definition at line 248 of file GenEvent.h.

                                                                   {
    std::map<int, ConstGenParticlePtr> ret;
    for (const auto &bcpartpair: m_particleBC)
      ret.insert({bcpartpair.first, std::const_pointer_cast<const HepMC3::GenParticle>(bcpartpair.second)});
    return ret;
   }

barcode_to_vertex() [1/2]

GenVertexPtr HepMC::GenEventBarcodes::barcode_to_vertex ( int id )

inline

Definition at line 188 of file GenEvent.h.

                                          {
    const auto it = m_vertexBC.find (id);
    if (it != m_vertexBC.end()) return it->second;
    return nullptr;
  }

barcode_to_vertex() [2/2]

ConstGenVertexPtr HepMC::GenEventBarcodes::barcode_to_vertex ( int id ) const

inline

Definition at line 183 of file GenEvent.h.

                                                     {
    const auto it = m_vertexBC.find (id);
    if (it != m_vertexBC.end()) return it->second;
    return nullptr;
  }

barcode_to_vertex_map()

std::map< int, ConstGenVertexPtr > HepMC::GenEventBarcodes::barcode_to_vertex_map ( ) const

inline

Definition at line 242 of file GenEvent.h.

                                                               {
    std::map<int, ConstGenVertexPtr> ret;
    for (const auto &bcvertpair: m_vertexBC)
      ret.insert({bcvertpair.first, std::const_pointer_cast<const HepMC3::GenVertex>(bcvertpair.second)});
    return ret;
   }

fillAttribute()

void HepMC::GenEventBarcodes::fillAttribute ( GenEvent * e )

inline

Definition at line 262 of file GenEvent.h.

                                  {
    const auto eventAttributes = e->attributes(); // this makes a copy
    const auto barcodeAttributeIt = eventAttributes.find(HepMCStr::barcode);
    const bool hasBarcodeAttribute = barcodeAttributeIt != eventAttributes.end();
 
    const auto &particles = e->particles();
    for (size_t i = 1; i <= particles.size(); i++) {
      if (hasBarcodeAttribute && barcodeAttributeIt->second.count(i) > 0) {
        const auto &ptr = barcodeAttributeIt->second.at(i);
        if (ptr->is_parsed()) {
          m_particleBC[static_cast<HepMC3::IntAttribute*>(ptr.get())->value()] = ptr->particle();
        }
        else {
          m_particleBC[std::atoi(ptr->unparsed_string().c_str())] = ptr->particle();
        }
      } else {
        m_particleBC[i] = particles[i-1];
      }
    }
    const auto &vertices = e->vertices();
    for (size_t i = 1; i <= vertices.size(); i++) {
      if (hasBarcodeAttribute && barcodeAttributeIt->second.count(-i) > 0) {
        const auto &ptr = barcodeAttributeIt->second.at(-i);
        if (ptr->is_parsed()) {
          m_vertexBC[static_cast<HepMC3::IntAttribute*>(ptr.get())->value()] = ptr->vertex();
        }
        else {
          m_vertexBC[std::atoi(ptr->unparsed_string().c_str())] = ptr->vertex();
        }
      } else {
        m_vertexBC[-i] = vertices[i-1];
      }
    }
    set_is_parsed(true);
  }

from_string()

virtual bool HepMC::GenEventBarcodes::from_string ( const std::string & )

inlineoverridevirtual

Definition at line 177 of file GenEvent.h.

{ return false; }

id_to_barcode_map()

std::map< int, int > HepMC::GenEventBarcodes::id_to_barcode_map ( ) const

inline

Definition at line 254 of file GenEvent.h.

                                            {
    std::map<int, int> ret;
    for (const auto &bcvertpair: m_vertexBC) ret.insert({bcvertpair.second->id(), bcvertpair.first});
    for (const auto &bcpartpair: m_particleBC) ret.insert({bcpartpair.second->id(), bcpartpair.first});
    return ret;
   }

remove() [1/2]

void HepMC::GenEventBarcodes::remove ( GenParticlePtr p )

inline

Definition at line 231 of file GenEvent.h.

                                 {
    if (!p) return;
    auto barcode = p->attribute<HepMC3::IntAttribute>(HepMCStr::barcode);
    if (barcode) {
      const auto it = m_particleBC.find (barcode->value());
      if (it != m_particleBC.end()) {
        m_particleBC.erase (it);
      }
    }
  }

remove() [2/2]

void HepMC::GenEventBarcodes::remove ( GenVertexPtr p )

inline

Definition at line 212 of file GenEvent.h.

                               {
    if (!p) return;
    auto barcode = p->attribute<HepMC3::IntAttribute>(HepMCStr::barcode);
    if (barcode) {
      const auto it = m_vertexBC.find (barcode->value());
      if (it != m_vertexBC.end()) {
        m_vertexBC.erase (it);
      }
    }
  }

to_string()

virtual bool HepMC::GenEventBarcodes::to_string ( std::string & att ) const

inlineoverridevirtual

Definition at line 178 of file GenEvent.h.

  {
    att =  "GenEventBarcodes";
    return true;
  }

Member Data Documentation

m_particleBC

std::unordered_map<int, GenParticlePtr> HepMC::GenEventBarcodes::m_particleBC

private

Definition at line 300 of file GenEvent.h.

m_vertexBC

std::unordered_map<int, GenVertexPtr> HepMC::GenEventBarcodes::m_vertexBC

private

Definition at line 299 of file GenEvent.h.

---

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

• GenEvent.h