CompositeRefCandidateT.h

Go to the documentation of this file.

#ifndef Candidate_CompositeRefCandidateT_h
#define Candidate_CompositeRefCandidateT_h

#include "DataFormats/Candidate/interface/LeafCandidate.h"
#include "DataFormats/Common/interface/CMS_CLASS_VERSION.h"

namespace reco {

  template<typename D>
  class CompositeRefCandidateT : public LeafCandidate {
  public:
    typedef D daughters;
    typedef D mothers;
    CompositeRefCandidateT() : LeafCandidate() { }
    CompositeRefCandidateT( 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 ) { }
    CompositeRefCandidateT( 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 CompositeRefCandidateT( const LeafCandidate& c ) : LeafCandidate( c ) { }
    ~CompositeRefCandidateT() override;
    CompositeRefCandidateT<D> * clone() const override;
    size_t numberOfDaughters() const override;
    size_t numberOfMothers() const override;
    const Candidate * daughter(size_type) const override;
    using ::reco::LeafCandidate::daughter; // avoid hiding the base
    const Candidate * mother(size_type = 0) const override;
    Candidate * daughter(size_type) override;
    void addDaughter( const typename daughters::value_type & );    
    void addMother( const typename mothers::value_type & );    
    void clearDaughters() { dau.clear(); }
    void clearMothers() { mom.clear(); }
    typename daughters::value_type daughterRef( size_type i ) const { return dau[ i ]; }
    const daughters & daughterRefVector() const { return dau; }
    typename daughters::value_type motherRef( size_type i = 0 ) const { return mom[ i ]; }
    const mothers & motherRefVector() const { return mom; }
    void resetDaughters( const edm::ProductID & id ) { dau = daughters( id ); }
    void resetMothers( const edm::ProductID & id ) { mom = mothers( id ); }

    CMS_CLASS_VERSION(13)

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

  template<typename D>
  inline void CompositeRefCandidateT<D>::addDaughter( const typename daughters::value_type & cand ) { 
    dau.push_back( cand ); 
  }

  template<typename D>
  inline void CompositeRefCandidateT<D>::addMother( const typename daughters::value_type & cand ) { 
    mom.push_back( cand ); 
  }

  template<typename D>
  CompositeRefCandidateT<D>::~CompositeRefCandidateT() { 
  }
  
  template<typename D>
  CompositeRefCandidateT<D> * CompositeRefCandidateT<D>::clone() const { 
    return new CompositeRefCandidateT( * this ); 
  }
  
  template<typename D>
  const Candidate * CompositeRefCandidateT<D>::daughter( size_type i ) const { 
    return ( i < numberOfDaughters() ) ? & * dau[ i ] : nullptr;
  }
  
  template<typename D>
  const Candidate * CompositeRefCandidateT<D>::mother( size_type i ) const { 
    return ( i < numberOfMothers() ) ? & * mom[ i ] : nullptr;
  }
  
  template<typename D>
  Candidate * CompositeRefCandidateT<D>::daughter( size_type i ) { 
    return nullptr;
  }
  
  template<typename D>
  size_t CompositeRefCandidateT<D>::numberOfDaughters() const { 
    return dau.size(); 
  }
  
  template<typename D>
  size_t CompositeRefCandidateT<D>::numberOfMothers() const { 
    return mom.size(); 
  }
  
  template<typename D>
  bool CompositeRefCandidateT<D>::overlap( const Candidate & c2 ) const {
    throw cms::Exception( "Error" ) << "can't check overlap internally for CompositeRefCanddate";
  }
}

#endif