CompositeRefCandidate.h

Go to the documentation of this file.

#ifndef Candidate_CompositeRefCandidate_h
#define Candidate_CompositeRefCandidate_h
#include "DataFormats/Candidate/interface/LeafCandidate.h"
#include "DataFormats/Candidate/interface/iterator_imp_specific.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 ) { }
    virtual ~CompositeRefCandidate();
    virtual CompositeRefCandidate * clone() const;
    virtual const_iterator begin() const;
    virtual const_iterator end() const;
    virtual iterator begin();
    virtual iterator end();
    virtual size_t numberOfDaughters() const;
    virtual const Candidate * daughter( size_type ) const;
    virtual Candidate * daughter( size_type );
    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 ); }
    virtual size_t numberOfMothers() const;
    virtual const Candidate * mother( size_t i = 0 ) const;

  private:
    typedef candidate::const_iterator_imp_specific<daughters> const_iterator_imp_specific;
    typedef candidate::iterator_imp_specific_dummy<daughters> iterator_imp_specific;
    daughters dau;
    daughters mom;
    virtual bool overlap( const Candidate & ) const;
  };

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

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

#endif