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_;
  };

}  // namespace reco

#endif