CompositeRefCandidate.h

Go to the documentation of this file.

#ifndef Candidate_CompositeRefCandidate_h
#define Candidate_CompositeRefCandidate_h
#include "DataFormats/Candidate/interface/LeafCandidate.h"
namespace reco {

  class CompositeRefCandidate : public LeafCandidate {
  public:
    typedef CandidateRefVector daughters;
    typedef CandidateRefVector mothers;
    CompositeRefCandidate() : LeafCandidate() {}
    CompositeRefCandidate(Charge q,
                          const LorentzVector& p4,
                          const Point& vtx = Point(0, 0, 0),
                          int pdgId = 0,
                          int status = 0,
                          bool integerCharge = true)
        : LeafCandidate(q, p4, vtx, pdgId, status, integerCharge) {}
    CompositeRefCandidate(Charge q,
                          const PolarLorentzVector& p4,
                          const Point& vtx = Point(0, 0, 0),
                          int pdgId = 0,
                          int status = 0,
                          bool integerCharge = true)
        : LeafCandidate(q, p4, vtx, pdgId, status, integerCharge) {}
    explicit CompositeRefCandidate(const Candidate& p) : LeafCandidate(p) {}
    ~CompositeRefCandidate() override;
    CompositeRefCandidate* clone() const override;
    size_t numberOfDaughters() const override;
    const Candidate* daughter(size_type) const override;
    Candidate* daughter(size_type) override;
    using reco::LeafCandidate::daughter;  // avoid hiding the base
    void addDaughter(const CandidateRef&);
    void addMother(const CandidateRef&);
    void clearDaughters() { dau.clear(); }
    CandidateRef daughterRef(size_type i) const { return dau[i]; }
    const daughters& daughterRefVector() const { return dau; }
    CandidateRef motherRef(size_type i = 0) const { return mom[i]; }
    const mothers& motherRefVector() const { return mom; }
    void resetDaughters(const edm::ProductID& id) { dau = daughters(id); }
    size_t numberOfMothers() const override;
    const Candidate* mother(size_t i = 0) const override;

  private:
    daughters dau;
    daughters mom;
    bool overlap(const Candidate&) const override;
  };

  inline void CompositeRefCandidate::addDaughter(const CandidateRef& cand) { dau.push_back(cand); }

  inline void CompositeRefCandidate::addMother(const CandidateRef& cand) { mom.push_back(cand); }
}  // namespace reco

#endif