CompositeCandidate.h

Go to the documentation of this file.

#ifndef Candidate_CompositeCandidate_h
#define Candidate_CompositeCandidate_h
#include "DataFormats/Candidate/interface/LeafCandidate.h"
#include <memory>
#include "DataFormats/Candidate/interface/iterator_imp_specific.h"
#include "DataFormats/Candidate/interface/CompositeCandidateFwd.h"
#include <string>
#include <vector> 

namespace reco {

  class CompositeCandidate : public LeafCandidate {
  public:
    typedef CandidateCollection daughters;
    typedef std::vector<std::string> role_collection; 
    CompositeCandidate(std::string name="") : LeafCandidate(), name_(name) { }
    CompositeCandidate( Charge q, const LorentzVector & p4, const Point & vtx = Point( 0, 0, 0 ),
            int pdgId = 0, int status = 0, bool integerCharge = true,
            std::string name="") :
      LeafCandidate( q, p4, vtx, pdgId, status, integerCharge ), name_(name) { }
    CompositeCandidate( Charge q, const PolarLorentzVector & p4, const Point & vtx = Point( 0, 0, 0 ),
            int pdgId = 0, int status = 0, bool integerCharge = true,
            std::string name="" ) :
      LeafCandidate( q, p4, vtx, pdgId, status, integerCharge ), name_(name) { }
    explicit CompositeCandidate( const Candidate & p, const std::string& name="" );
    explicit CompositeCandidate( const Candidate & p, const std::string& name, role_collection const & roles );
    virtual ~CompositeCandidate();
    std::string name() const { return name_;}
    void        setName(std::string name) { name_ = name;}
    role_collection const & roles() const { return roles_; }
    void                    setRoles( const role_collection & roles ) { roles_.clear(); roles_ = roles; }
    virtual CompositeCandidate * clone() const;
    virtual const_iterator begin() const;
    virtual const_iterator end() const;
    virtual iterator begin();
    virtual iterator end();
    virtual size_type numberOfDaughters() const;
    virtual const Candidate * daughter( size_type ) const;
    virtual Candidate * daughter( size_type );
    // Get candidate based on role
    virtual Candidate *       daughter(const std::string& s );
    virtual const Candidate * daughter(const std::string& s ) const;
    void addDaughter( const Candidate &, const std::string& s="" );
    void addDaughter( std::auto_ptr<Candidate>, const std::string& s="" );
    void clearDaughters() { dau.clear(); }
    // clear roles
    void clearRoles() { roles_.clear(); }
    // Apply the roles to the objects
    void applyRoles();
    virtual size_type numberOfMothers() const;
    virtual const Candidate * mother( size_type i = 0 ) const;

  private:
    // const iterator implementation
    typedef candidate::const_iterator_imp_specific<daughters> const_iterator_imp_specific;
    // iterator implementation
    typedef candidate::iterator_imp_specific<daughters> iterator_imp_specific;
    daughters dau;
    virtual bool overlap( const Candidate & ) const;
    std::string name_;
    role_collection roles_;
  };

}

#endif