WSelectorFast.h

Go to the documentation of this file.

#ifndef PhysicsTools_FWLite_WSelectorFast_h
#define PhysicsTools_FWLite_WSelectorFast_h
#include "DataFormats/PatCandidates/interface/MET.h"
#include "DataFormats/PatCandidates/interface/Muon.h"
#include "PhysicsTools/SelectorUtils/interface/EventSelector.h"
 
class WSelectorFast : public EventSelector {
public:
  WSelectorFast(edm::ParameterSet const& params)
      : muonSrc_(params.getParameter<edm::InputTag>("muonSrc")), metSrc_(params.getParameter<edm::InputTag>("metSrc")) {
    double muonPtMin = params.getParameter<double>("muonPtMin");
    double metMin = params.getParameter<double>("metMin");
    push_back("Muon Pt", muonPtMin);
    push_back("MET", metMin);
    set("Muon Pt");
    set("MET");
    wMuon_ = 0;
    met_ = 0;
    if (params.exists("cutsToIgnore")) {
      setIgnoredCuts(params.getParameter<std::vector<std::string> >("cutsToIgnore"));
    }
    retInternal_ = getBitTemplate();
  }
  virtual ~WSelectorFast() {}
  pat::Muon const& wMuon() const { return *wMuon_; }
  pat::MET const& met() const { return *met_; }
 
  virtual bool operator()(edm::EventBase const& event, pat::strbitset& ret) {
    ret.set(false);
    // Handle to the muon collection
    edm::Handle<std::vector<pat::Muon> > muons;
    // Handle to the MET collection
    edm::Handle<std::vector<pat::MET> > met;
    // get the objects from the event
    bool gotMuons = event.getByLabel(muonSrc_, muons);
    bool gotMET = event.getByLabel(metSrc_, met);
    // get the MET, require to be > minimum
    if (gotMET) {
      met_ = &met->at(0);
      if (met_->pt() > cut(metIndex_, double()) || ignoreCut(metIndex_))
        passCut(ret, metIndex_);
    }
    // get the highest pt muon, require to have pt > minimum
    if (gotMuons) {
      if (!ignoreCut(muonPtIndex_)) {
        if (muons->size() > 0) {
          wMuon_ = &muons->at(0);
          if (wMuon_->pt() > cut(muonPtIndex_, double()) || ignoreCut(muonPtIndex_))
            passCut(ret, muonPtIndex_);
        }
      } else {
        passCut(ret, muonPtIndex_);
      }
    }
    setIgnored(ret);
    return (bool)ret;
  }
 
protected:
  edm::InputTag muonSrc_;
  edm::InputTag metSrc_;
  pat::Muon const* wMuon_;
  pat::MET const* met_;
  index_type muonPtIndex_;
  index_type metIndex_;
};
#endif  // PhysicsTools_FWLite_WSelectorFast_h