WSelector.h

Go to the documentation of this file.

#ifndef PhysicsTools_FWLite_WSelector_h
#define PhysicsTools_FWLite_WSelector_h
#include "DataFormats/PatCandidates/interface/MET.h"
#include "DataFormats/PatCandidates/interface/Muon.h"
#include "PhysicsTools/SelectorUtils/interface/EventSelector.h"

class WSelector : public EventSelector {

public:
  WSelector(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_ = nullptr; met_ = nullptr;
    if ( params.exists("cutsToIgnore") ){
      setIgnoredCuts( params.getParameter<std::vector<std::string> >("cutsToIgnore") );
    }
    retInternal_ = getBitTemplate();
  }
  ~WSelector() override {}
  pat::Muon const& wMuon() const { return *wMuon_;}
  pat::MET  const& met()   const { return *met_;  }

  bool operator()( edm::EventBase const & event, pat::strbitset & ret) override{
    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("MET", double()) || ignoreCut("MET") ) 
    passCut(ret, "MET");
    }
    // get the highest pt muon, require to have pt > minimum
    if( gotMuons ){
      if( !ignoreCut("Muon Pt") ){
    if( !muons->empty() ){
      wMuon_ = &muons->at(0);
      if( wMuon_->pt() > cut("Muon Pt", double()) || ignoreCut("Muon Pt") ) 
        passCut(ret, "Muon Pt");
    }
      } 
      else{
    passCut( ret, "Muon Pt");
      }
    }
    setIgnored(ret);
    return (bool)ret;
  }

protected:
  edm::InputTag muonSrc_;
  edm::InputTag metSrc_;
  pat::Muon const* wMuon_;
  pat::MET const* met_;
};
#endif // PhysicsTools_FWLite_WSelector_h