NamedCandCombiner.h

Go to the documentation of this file.

#ifndef CandUtils_NamedCandCombiner_h
#define CandUtils_NamedCandCombiner_h

#include "CommonTools/CandUtils/interface/NamedCandCombinerBase.h"
#include "CommonTools/CandUtils/interface/AddFourMomenta.h"
#include "CommonTools/CandUtils/interface/CandSelector.h"
#include "CommonTools/UtilAlgos/interface/AnyPairSelector.h"
#include "DataFormats/Candidate/interface/NamedCompositeCandidate.h"
#include "DataFormats/Candidate/interface/ShallowClonePtrCandidate.h"
#include <string>

namespace combiner {
  namespace helpers {
    struct NormalClone {
      template<typename Ptr, typename CMP>
    static void addDaughter(CMP & cmp, const Ptr & c, std::string name) {
    cmp.addDaughter(*c, name);
      }
    };
    
    struct ShallowClone {
      template<typename CMP>
      static void addDaughter(CMP & cmp, const reco::CandidatePtr & c, std::string name) {
    if(c->numberOfDaughters()==0)
      cmp.addDaughter(reco::ShallowClonePtrCandidate(c), name);
    else
      cmp.addDaughter(*c, name);
      }
    };
  }
}

template<typename Selector, 
     typename PairSelector = AnyPairSelector,
     typename Cloner = combiner::helpers::NormalClone, 
     typename Setup = AddFourMomenta>
class NamedCandCombiner : public NamedCandCombinerBase {
public:
  NamedCandCombiner(std::string name) :
    NamedCandCombinerBase(name), 
    select_(), selectPair_(), setup_(), name_(name) { }
  NamedCandCombiner(std::string name, int q1, int q2) :
    NamedCandCombinerBase(name, q1, q2), 
    select_(), selectPair_(), setup_(), name_(name) { }
  NamedCandCombiner(std::string name,  int q1, int q2, int q3 ) :
    NamedCandCombinerBase(name, q1, q2, q3), 
    select_(), selectPair_(), setup_(), name_(name) { }
  NamedCandCombiner(std::string name, int q1, int q2, int q3, int q4) :
    NamedCandCombinerBase(name, q1, q2, q3, q4), 
    select_(), selectPair_(), setup_(), name_(name) { }
  NamedCandCombiner(std::string name, const Selector & select) :
    NamedCandCombinerBase(name), 
    select_(select), selectPair_(), setup_(), name_(name) { }
  NamedCandCombiner(std::string name, const Selector & select, int q1, int q2 ) :
    NamedCandCombinerBase(name, q1, q2), 
    select_(select), selectPair_(), setup_(), name_(name) { }
  NamedCandCombiner(std::string name,  const Selector & select, int q1, int q2, int q3 ) :
    NamedCandCombinerBase(name, q1, q2, q3), 
    select_(select), selectPair_(), setup_(), name_(name) { }
  NamedCandCombiner(std::string name,  const Selector & select, int q1, int q2, int q3, int q4 ) :
    NamedCandCombinerBase(name, q1, q2, q3, q4), 
    select_(select), selectPair_(), setup_(), name_(name) { }
  NamedCandCombiner(std::string name, const Selector & select, const PairSelector & selectPair) :
    NamedCandCombinerBase(name), 
    select_(select), selectPair_(selectPair), setup_(), name_(name) { }
  NamedCandCombiner(std::string name, const Selector & select, const PairSelector & selectPair, int q1, int q2) :
    NamedCandCombinerBase(name, q1, q2), 
    select_(select), selectPair_(selectPair), setup_(), name_(name) { }
  NamedCandCombiner(std::string name, const Selector & select, const PairSelector & selectPair, int q1, int q2, int q3) :
    NamedCandCombinerBase(name, q1, q2, q3), 
    select_(select), selectPair_(selectPair), setup_(), name_(name) { }
  NamedCandCombiner(std::string name, const Selector & select, const PairSelector & selectPair, int q1, int q2, int q3, int q4) :
   NamedCandCombinerBase(name, q1, q2, q3, q4), 
    select_(select), selectPair_(selectPair), setup_(), name_(name) { }
  NamedCandCombiner(std::string name, const Selector & select, const PairSelector & selectPair, const Setup & setup) :
    NamedCandCombinerBase(name), 
    select_(select), selectPair_(selectPair), setup_(setup), name_(name) { }
  NamedCandCombiner(std::string name, const Selector & select, const PairSelector & selectPair, const Setup & setup, int q1, int q2) :
    NamedCandCombinerBase(name, q1, q2), 
    select_(select), selectPair_(selectPair), setup_(setup), name_(name) { }
  NamedCandCombiner(std::string name, const Selector & select, const PairSelector & selectPair, const Setup & setup, int q1, int q2, int q3) :
    NamedCandCombinerBase(name, q1, q2, q3), 
    select_(select), selectPair_(selectPair), setup_(setup), name_(name) { }
  NamedCandCombiner(std::string name, const Selector & select, const PairSelector & selectPair, const Setup & setup, int q1, int q2, int q3, int q4) :
    NamedCandCombinerBase(name, q1, q2, q3, q4), 
    select_(select), selectPair_(selectPair), setup_(setup), name_(name) { }
  NamedCandCombiner(std::string name, const Selector & select, const PairSelector & selectPair, const Setup & setup,const std::vector <int> & dauCharge) : 
    NamedCandCombinerBase(name, true, dauCharge), select_(select), selectPair_(selectPair), setup_(setup), name_(name) { }
  NamedCandCombiner( std::string name, const Selector & select, const PairSelector & selectPair, const std::vector <int> & dauCharge ) : 
    NamedCandCombinerBase(name, true, dauCharge), select_(select), selectPair_(selectPair), setup_(), name_(name) { }
  NamedCandCombiner(std::string name, const std::vector <int> & dauCharge) : 
    NamedCandCombinerBase(name, true, dauCharge), select_(), selectPair_(), setup_(), name_(name) { }
  NamedCandCombiner(std::string name, const Selector & select, const PairSelector & selectPair, const Setup & setup,
           bool checkCharge, bool checkOverlap, const std::vector <int> & dauCharge) : 
    NamedCandCombinerBase(name, checkCharge, checkOverlap, dauCharge), 
    select_(select), selectPair_(selectPair), setup_(setup), name_(name) { }
  Setup & setup() { return setup_; }

private:
  virtual bool select(const reco::Candidate & c) const {
    return select_(c);
  } 
  virtual bool selectPair(const reco::Candidate & c1, const reco::Candidate & c2) const {
    return selectPair_(c1, c2);
  } 
  virtual void setup(reco::NamedCompositeCandidate & c) const {
    setup_.set(c);
  }
  virtual void addDaughter(reco::NamedCompositeCandidate & cmp, const reco::CandidatePtr & c, std::string n) const {
    Cloner::addDaughter(cmp, c, n);
  }
  Selector select_; 
  PairSelector selectPair_; 
  Setup setup_;
  std::string name_;
};

#endif