NamedCandCombinerBase.h

Go to the documentation of this file.

#ifndef CommonTools_CandUtils_CandCombinerBase_h
#define CommonTools_CandUtils_CandCombinerBase_h

#include "DataFormats/Candidate/interface/OverlapChecker.h"
#include "DataFormats/Candidate/interface/NamedCompositeCandidate.h"
#include <vector>
#include <string>

class NamedCandCombinerBase {
public:
  typedef std::vector<std::string> string_coll;
  NamedCandCombinerBase(std::string name);
  NamedCandCombinerBase(std::string name, int, int);
  NamedCandCombinerBase(std::string name, int, int, int);
  NamedCandCombinerBase(std::string name, int, int, int, int);
  NamedCandCombinerBase(std::string name, bool checkCharge, bool checkOverlap, const std::vector<int> &);
  virtual ~NamedCandCombinerBase();
  std::unique_ptr<reco::NamedCompositeCandidateCollection> combine(const std::vector<reco::CandidatePtrVector> &,
                                                                   string_coll const &) const;
  std::unique_ptr<reco::NamedCompositeCandidateCollection> combine(const reco::CandidatePtrVector &,
                                                                   string_coll const &) const;
  std::unique_ptr<reco::NamedCompositeCandidateCollection> combine(const reco::CandidatePtrVector &,
                                                                   const reco::CandidatePtrVector &,
                                                                   string_coll const &) const;
  std::unique_ptr<reco::NamedCompositeCandidateCollection> combine(const reco::CandidatePtrVector &,
                                                                   const reco::CandidatePtrVector &,
                                                                   const reco::CandidatePtrVector &,
                                                                   string_coll const &) const;
  std::unique_ptr<reco::NamedCompositeCandidateCollection> combine(const reco::CandidatePtrVector &,
                                                                   const reco::CandidatePtrVector &,
                                                                   const reco::CandidatePtrVector &,
                                                                   const reco::CandidatePtrVector &,
                                                                   string_coll const &) const;

private:
  bool preselect(const reco::Candidate &, const reco::Candidate &) const;
  void combine(reco::NamedCompositeCandidate &,
               const reco::CandidatePtr &,
               const reco::CandidatePtr &,
               std::string,
               std::string) const;
  typedef std::vector<
      std::pair<std::pair<reco::CandidatePtr, size_t>, std::vector<reco::CandidatePtrVector>::const_iterator> >
      CandStack;
  typedef std::vector<int> ChargeStack;
  void combine(size_t collectionIndex,
               CandStack &,
               ChargeStack &,
               string_coll const &names,
               std::vector<reco::CandidatePtrVector>::const_iterator begin,
               std::vector<reco::CandidatePtrVector>::const_iterator end,
               std::unique_ptr<reco::NamedCompositeCandidateCollection> &comps) const;
  virtual bool select(const reco::Candidate &) const = 0;
  virtual bool selectPair(const reco::Candidate &c1, const reco::Candidate &c2) const = 0;
  virtual void setup(reco::NamedCompositeCandidate &) const = 0;
  virtual void addDaughter(reco::NamedCompositeCandidate &cmp, const reco::CandidatePtr &c, std::string name) const = 0;
  bool checkCharge_;
  bool checkOverlap_;
  std::vector<int> dauCharge_;
  OverlapChecker overlap_;
  std::string name_;
};

#endif