WSelectorFast.h

Go to the documentation of this file.

#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_ = 0; met_ = 0;
    if ( params.exists("cutsToIgnore") ){
      setIgnoredCuts( params.getParameter<std::vector<std::string> >("cutsToIgnore") );
    }
    retInternal_ = getBitTemplate();
  }
  virtual ~WSelector() {}
  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_;
};