CompositePtrCandidate.h

Go to the documentation of this file.

#ifndef Candidate_CompositePtrCandidate_h
#define Candidate_CompositePtrCandidate_h
#include "DataFormats/Candidate/interface/LeafCandidate.h"
#include "DataFormats/Candidate/interface/iterator_imp_specific.h"

namespace reco {

  class CompositePtrCandidate : public LeafCandidate {
  public:
    typedef std::vector<CandidatePtr> daughters;
    typedef std::vector<CandidatePtr> mothers;
    CompositePtrCandidate() : LeafCandidate() { }
    CompositePtrCandidate( 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 ) { }
    CompositePtrCandidate( 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 CompositePtrCandidate( const Candidate & p ) : LeafCandidate( p ) { }
    virtual ~CompositePtrCandidate();
    virtual CompositePtrCandidate * clone() const;
    virtual const_iterator begin() const;
    virtual const_iterator end() const;
    virtual iterator begin();
    virtual iterator end();
    virtual size_t numberOfDaughters() const;
    virtual size_t numberOfMothers() const;
    virtual const Candidate * daughter( size_type ) const;
    using reco::LeafCandidate::daughter; // avoid hiding the base
    virtual Candidate * daughter( size_type );
    void addDaughter( const CandidatePtr & );    
    void clearDaughters() { dau.clear(); }
    CandidatePtr daughterPtr( size_type i ) const { return dau[ i ]; }
    const daughters & daughterPtrVector() const { return dau; }
    virtual const Candidate * mother( size_t i = 0 ) const;
    virtual size_type numberOfSourceCandidatePtrs() const ;
    virtual CandidatePtr sourceCandidatePtr( size_type i ) 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;
    virtual bool overlap( const Candidate & ) const;
  };

  inline void CompositePtrCandidate::addDaughter( const CandidatePtr & cand ) { 
    dau.push_back( cand ); 
  }

}

#endif