GenVertex.h

// -*- C++ -*-
//
// This file is part of HepMC
// Copyright (C) 2014-2023 The HepMC collaboration (see AUTHORS for details)
//
//
#ifndef HEPMC3_GENVERTEX_H
#define HEPMC3_GENVERTEX_H
#include <string>
#include "HepMC3/GenParticle_fwd.h"
#include "HepMC3/GenVertex_fwd.h"
#include "HepMC3/Data/GenVertexData.h"
#include "HepMC3/FourVector.h"
 
#ifdef HEPMC3_PROTOBUFIO
namespace HepMC3_pb {
class GenEventData_GenVertexData;
}
#endif
 
namespace HepMC3 {
 
class Attribute;
class GenEvent;

class GenVertex : public std::enable_shared_from_this<GenVertex> {
 
    friend class GenEvent;
 
public:


    GenVertex( const FourVector& position = FourVector::ZERO_VECTOR() );

    GenVertex( const GenVertexData& data );
 
#ifdef HEPMC3_PROTOBUFIO
    GenVertex( HepMC3_pb::GenEventData_GenVertexData const &data );
#endif

 


    GenEvent* parent_event() { return m_event; }

    const GenEvent* parent_event() const { return m_event; }

    bool in_event() const { return parent_event() != nullptr; }

    int id() const { return m_id; }

    int status() const { return m_data.status; }
    void set_status(int stat) { m_data.status = stat; }

    const GenVertexData& data() const { return m_data; }

    void add_particle_in ( GenParticlePtr p);
    void add_particle_out( GenParticlePtr p);
    void remove_particle_in ( GenParticlePtr p);
    void remove_particle_out( GenParticlePtr p);

    inline int particles_in_size() const { return m_particles_in.size(); }
    inline int particles_out_size() const { return m_particles_out.size(); }
 

    const std::vector<GenParticlePtr>& particles_in() { return m_particles_in; }
    const std::vector<ConstGenParticlePtr>& particles_in() const;
    const std::vector<GenParticlePtr>& particles_out() { return m_particles_out; }
    const std::vector<ConstGenParticlePtr>& particles_out() const;

    const FourVector& position() const;
    bool has_set_position() const { return !(m_data.position.is_zero()); }

    void set_position(const FourVector& new_pos); 

    bool add_attribute(const std::string& name, std::shared_ptr<Attribute> att);

    std::vector<std::string> attribute_names() const;

    void remove_attribute(const std::string& name);

    template<class T>
    std::shared_ptr<T> attribute(const std::string& name) const;

    std::string attribute_as_string(const std::string& name) const;
 
 
private:

    GenEvent       *m_event;  
    int             m_id;     
    GenVertexData   m_data;   
 
    std::vector<GenParticlePtr>  m_particles_in;  
 
    std::vector<GenParticlePtr>  m_particles_out; 
 
};
 
 
} // namespace HepMC3
 
#include "HepMC3/GenEvent.h"
namespace HepMC3 {
template<class T> std::shared_ptr<T> GenVertex::attribute(const std::string& name) const {
    return parent_event()?
           parent_event()->attribute<T>(name, id()): std::shared_ptr<T>();
}
}
 
#endif