GenParticle.hpp

#pragma once

#include <mensura/core/PhysicsObjects.hpp>

#include <list>
#include <initializer_list>


class GenParticle: public Candidate
{
public:
    typedef std::list<GenParticle const *> collection_t;

public:
    GenParticle() noexcept;
    
    GenParticle(TLorentzVector const &p4, int pdgId = 0) noexcept;
    
    GenParticle(GenParticle const &) = default;
    
    GenParticle(GenParticle &&) = default;

public:
    void AddDaughter(GenParticle const *p);
    
    void AddMother(GenParticle const *p);
    
    GenParticle const *FindFirstDaughter(std::initializer_list<int> const &pdgIds) const;
    
     GenParticle const *FindFirstDaughterRecursive(std::initializer_list<int> const &pdgIds)
      const;
    
    collection_t const &GetDaughters() const;
    
    GenParticle const *GetFirstMother() const;
    
    int GetFirstMotherPdgId() const;
    
    collection_t const &GetMothers() const;
    
    int GetPdgId() const;
    
    void SetPdgId(int pdgId_);
    
private:
    int pdgId;
    
    collection_t mothers;
    
    collection_t daughters;
};