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