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/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) { }
    template<typename P4>
    CompositeCandidate( Charge q, const P4 & 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 );
    ~CompositeCandidate() override;
    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; }
    CompositeCandidate * clone() const override;
    size_type numberOfDaughters() const override;
    const Candidate * daughter( size_type ) const override;
    Candidate * daughter( size_type ) override;
    // Get candidate based on role
    Candidate *       daughter(const std::string& s ) override;
    const Candidate * daughter(const std::string& s ) const override;
    void addDaughter( const Candidate &, const std::string& s="" );
    void addDaughter( std::unique_ptr<Candidate>, const std::string& s="" );
    void clearDaughters() { dau.clear(); }
    // clear roles
    void clearRoles() { roles_.clear(); }
    // Apply the roles to the objects
    void applyRoles();
    size_type numberOfMothers() const override;
    const Candidate * mother( size_type i = 0 ) const override;

  private:
    daughters dau;
    bool overlap( const Candidate & ) const override;
    std::string name_;
    role_collection roles_;
  };

}

#endif