WPlusJetsEventSelector.cc

Go to the documentation of this file.

#include "PhysicsTools/SelectorUtils/interface/WPlusJetsEventSelector.h"
#include "DataFormats/Candidate/interface/ShallowCloneCandidate.h"

#include <iostream>

using namespace std;

WPlusJetsEventSelector::WPlusJetsEventSelector( edm::ParameterSet const & params ) :
  EventSelector(),
  muonTag_         (params.getParameter<edm::InputTag>("muonSrc") ),
  electronTag_     (params.getParameter<edm::InputTag>("electronSrc") ),
  jetTag_          (params.getParameter<edm::InputTag>("jetSrc") ),
  metTag_          (params.getParameter<edm::InputTag>("metSrc") ),
  trigTag_         (params.getParameter<edm::InputTag>("trigSrc") ),
  muTrig_          (params.getParameter<std::string>("muTrig")),
  eleTrig_         (params.getParameter<std::string>("eleTrig")),
  pvSelector_      (params.getParameter<edm::ParameterSet>("pvSelector") ),
  muonIdTight_     (params.getParameter<edm::ParameterSet>("muonIdTight") ),
  electronIdTight_ (params.getParameter<edm::ParameterSet>("electronIdTight") ),
  muonIdLoose_     (params.getParameter<edm::ParameterSet>("muonIdLoose") ),
  electronIdLoose_ (params.getParameter<edm::ParameterSet>("electronIdLoose") ),
  jetIdLoose_      (params.getParameter<edm::ParameterSet>("jetIdLoose") ),
  pfjetIdLoose_    (params.getParameter<edm::ParameterSet>("pfjetIdLoose") ),
  minJets_         (params.getParameter<int> ("minJets") ),
  muJetDR_         (params.getParameter<double>("muJetDR")),
  eleJetDR_        (params.getParameter<double>("eleJetDR")),
  muPlusJets_      (params.getParameter<bool>("muPlusJets") ),
  ePlusJets_       (params.getParameter<bool>("ePlusJets") ),
  muPtMin_         (params.getParameter<double>("muPtMin")),
  muEtaMax_        (params.getParameter<double>("muEtaMax")),
  eleEtMin_        (params.getParameter<double>("eleEtMin")),
  eleEtaMax_       (params.getParameter<double>("eleEtaMax")),
  muPtMinLoose_    (params.getParameter<double>("muPtMinLoose")),
  muEtaMaxLoose_   (params.getParameter<double>("muEtaMaxLoose")),
  eleEtMinLoose_   (params.getParameter<double>("eleEtMinLoose")),
  eleEtaMaxLoose_  (params.getParameter<double>("eleEtaMaxLoose")),
  jetPtMin_        (params.getParameter<double>("jetPtMin")),
  jetEtaMax_       (params.getParameter<double>("jetEtaMax")),
  jetScale_        (params.getParameter<double>("jetScale")),
  metMin_          (params.getParameter<double>("metMin"))
{
  // make the bitset
  push_back( "Inclusive"      );
  push_back( "Trigger"        );
  push_back( "PV"             );
  push_back( ">= 1 Lepton"    );
  push_back( "== 1 Tight Lepton"    );
  push_back( "== 1 Tight Lepton, Mu Veto");
  push_back( "== 1 Lepton"    );
  push_back( "MET Cut"        );
  push_back( "Z Veto"         );
  push_back( "Conversion Veto");
  push_back( "Cosmic Veto"    );
  push_back( ">=1 Jets"       );
  push_back( ">=2 Jets"       );
  push_back( ">=3 Jets"       );
  push_back( ">=4 Jets"       );
  push_back( ">=5 Jets"       );


  // turn (almost) everything on by default
  set( "Inclusive"      );
  set( "Trigger"        );
  set( "PV"             );
  set( ">= 1 Lepton"    );
  set( "== 1 Tight Lepton"    );
  set( "== 1 Tight Lepton, Mu Veto");
  set( "== 1 Lepton"    );
  set( "MET Cut"        );
  set( "Z Veto"         );
  set( "Conversion Veto");
  set( "Cosmic Veto"    );
  set( ">=1 Jets", minJets_ >= 1);
  set( ">=2 Jets", minJets_ >= 2);
  set( ">=3 Jets", minJets_ >= 3);
  set( ">=4 Jets", minJets_ >= 4);
  set( ">=5 Jets", minJets_ >= 5);


  inclusiveIndex_ = index_type(&bits_, std::string("Inclusive"      ));
  triggerIndex_ = index_type(&bits_, std::string("Trigger"        ));
  pvIndex_ = index_type(&bits_, std::string("PV"             ));
  lep1Index_ = index_type(&bits_, std::string(">= 1 Lepton"    ));
  lep2Index_ = index_type(&bits_, std::string("== 1 Tight Lepton"    ));
  lep3Index_ = index_type(&bits_, std::string("== 1 Tight Lepton, Mu Veto"));
  lep4Index_ = index_type(&bits_, std::string("== 1 Lepton"    ));
  metIndex_ = index_type(&bits_, std::string("MET Cut"        ));
  zvetoIndex_ = index_type(&bits_, std::string("Z Veto"         ));
  conversionIndex_ = index_type(&bits_, std::string("Conversion Veto"));
  cosmicIndex_ = index_type(&bits_, std::string("Cosmic Veto"    ));
  jet1Index_ = index_type(&bits_, std::string(">=1 Jets"));
  jet2Index_ = index_type(&bits_, std::string(">=2 Jets"));
  jet3Index_ = index_type(&bits_, std::string(">=3 Jets"));
  jet4Index_ = index_type(&bits_, std::string(">=4 Jets"));
  jet5Index_ = index_type(&bits_, std::string(">=5 Jets"));

  if ( params.exists("cutsToIgnore") )
    setIgnoredCuts( params.getParameter<std::vector<std::string> >("cutsToIgnore") );


  retInternal_ = getBitTemplate();
}

bool WPlusJetsEventSelector::operator() ( edm::EventBase const & event, pat::strbitset & ret)
{

  ret.set(false);

  selectedJets_.clear();
  cleanedJets_.clear();
  selectedMuons_.clear();
  selectedElectrons_.clear();
  looseMuons_.clear();
  looseElectrons_.clear();
  selectedMETs_.clear();


  passCut( ret, inclusiveIndex_);


  bool passTrig = false;
  if (!ignoreCut(triggerIndex_) ) {


    edm::Handle<pat::TriggerEvent> triggerEvent;
    event.getByLabel(trigTag_, triggerEvent);

    pat::TriggerEvent const * trig = &*triggerEvent;

    if ( trig->wasRun() && trig->wasAccept() ) {

      pat::TriggerPath const * muPath = trig->path(muTrig_);

      pat::TriggerPath const * elePath = trig->path(eleTrig_);

      if ( muPlusJets_ && muPath != 0 && muPath->wasAccept() ) {
    passTrig = true;
      }

      if ( ePlusJets_ && elePath != 0 && elePath->wasAccept() ) {
    passTrig = true;
      }
    }
  }




  if ( ignoreCut(triggerIndex_) ||
       passTrig ) {
    passCut(ret, triggerIndex_);


    bool passPV = false;

    passPV = pvSelector_( event );
    if ( ignoreCut(pvIndex_) || passPV ) {
      passCut(ret, pvIndex_);

      edm::Handle< vector< pat::Electron > > electronHandle;
      event.getByLabel (electronTag_, electronHandle);

      edm::Handle< vector< pat::Muon > > muonHandle;
      event.getByLabel (muonTag_, muonHandle);

      edm::Handle< vector< pat::Jet > > jetHandle;

      edm::Handle< edm::OwnVector<reco::Candidate> > jetClonesHandle ;

      edm::Handle< vector< pat::MET > > metHandle;
      event.getByLabel (metTag_, metHandle);

      int nElectrons = 0;
      for ( std::vector<pat::Electron>::const_iterator electronBegin = electronHandle->begin(),
          electronEnd = electronHandle->end(), ielectron = electronBegin;
        ielectron != electronEnd; ++ielectron ) {
    ++nElectrons;
    // Tight cuts
    if ( ielectron->et() > eleEtMin_ && fabs(ielectron->eta()) < eleEtaMax_ &&
         electronIdTight_(*ielectron) &&
         ielectron->electronID( "eidRobustTight" ) > 0  ) {
      selectedElectrons_.push_back( reco::ShallowClonePtrCandidate( edm::Ptr<pat::Electron>( electronHandle, ielectron - electronBegin ) ) );
    } else {
      // Loose cuts
      if ( ielectron->et() > eleEtMinLoose_ && fabs(ielectron->eta()) < eleEtaMaxLoose_ &&
           electronIdLoose_(*ielectron) ) {
        looseElectrons_.push_back( reco::ShallowClonePtrCandidate( edm::Ptr<pat::Electron>( electronHandle, ielectron - electronBegin ) ) );
      }
    }
      }


      for ( std::vector<pat::Muon>::const_iterator muonBegin = muonHandle->begin(),
          muonEnd = muonHandle->end(), imuon = muonBegin;
        imuon != muonEnd; ++imuon ) {
    if ( !imuon->isGlobalMuon() ) continue;

    // Tight cuts
    bool passTight = muonIdTight_(*imuon,event) && imuon->isTrackerMuon() ;
    if (  imuon->pt() > muPtMin_ && fabs(imuon->eta()) < muEtaMax_ &&
         passTight ) {

      selectedMuons_.push_back( reco::ShallowClonePtrCandidate( edm::Ptr<pat::Muon>( muonHandle, imuon - muonBegin ) ) );
    } else {
      // Loose cuts
      if ( imuon->pt() > muPtMinLoose_ && fabs(imuon->eta()) < muEtaMaxLoose_ &&
           muonIdLoose_(*imuon,event) ) {
        looseMuons_.push_back( reco::ShallowClonePtrCandidate( edm::Ptr<pat::Muon>( muonHandle, imuon - muonBegin ) ) );
      }
    }
      }


      met_ = reco::ShallowClonePtrCandidate( edm::Ptr<pat::MET>( metHandle, 0),
                         metHandle->at(0).charge(),
                         metHandle->at(0).p4() );



      event.getByLabel (jetTag_, jetHandle);
      pat::strbitset ret1 = jetIdLoose_.getBitTemplate();
      pat::strbitset ret2 = pfjetIdLoose_.getBitTemplate();
      for ( std::vector<pat::Jet>::const_iterator jetBegin = jetHandle->begin(),
          jetEnd = jetHandle->end(), ijet = jetBegin;
        ijet != jetEnd; ++ijet ) {
    reco::ShallowClonePtrCandidate scaledJet ( reco::ShallowClonePtrCandidate( edm::Ptr<pat::Jet>( jetHandle, ijet - jetBegin ),
                                           ijet->charge(),
                                           ijet->p4() * jetScale_ ) );
    bool passJetID = false;
    if ( ijet->isCaloJet() || ijet->isJPTJet() ) passJetID = jetIdLoose_(*ijet, ret1);
    else passJetID = pfjetIdLoose_(*ijet, ret2);
    if ( scaledJet.pt() > jetPtMin_ && fabs(scaledJet.eta()) < jetEtaMax_ && passJetID ) {
      selectedJets_.push_back( scaledJet );

      if ( muPlusJets_ ) {

        //Remove some jets
        bool indeltaR = false;
        for( std::vector<reco::ShallowClonePtrCandidate>::const_iterator muonBegin = selectedMuons_.begin(),
           muonEnd = selectedMuons_.end(), imuon = muonBegin;
         imuon != muonEnd; ++imuon ) {
          if( reco::deltaR( imuon->eta(), imuon->phi(), scaledJet.eta(), scaledJet.phi() ) < muJetDR_ )
        {  indeltaR = true; }
        }
        if( !indeltaR ) {
          cleanedJets_.push_back( scaledJet );
        }// end if jet is not within dR of a muon
      }// end if mu+jets
      else {
        //Remove some jets
        bool indeltaR = false;
        for( std::vector<reco::ShallowClonePtrCandidate>::const_iterator electronBegin = selectedElectrons_.begin(),
           electronEnd = selectedElectrons_.end(), ielectron = electronBegin;
         ielectron != electronEnd; ++ielectron ) {
          if( reco::deltaR( ielectron->eta(), ielectron->phi(), scaledJet.eta(), scaledJet.phi() ) < eleJetDR_ )
        {  indeltaR = true; }
        }
        if( !indeltaR ) {
          cleanedJets_.push_back( scaledJet );
        }// end if jet is not within dR of an electron
      }// end if e+jets
    }// end if pass id and kin cuts
      }// end loop over jets



      int nleptons = 0;
      if ( muPlusJets_ )
    nleptons += selectedMuons_.size();

      if ( ePlusJets_ )
    nleptons += selectedElectrons_.size();

      if ( ignoreCut(lep1Index_) ||
       ( nleptons > 0 ) ){
    passCut( ret, lep1Index_);

    if ( ignoreCut(lep2Index_) ||
         ( nleptons == 1 ) ){
      passCut( ret, lep2Index_);

      bool oneMuon =
        ( selectedMuons_.size() == 1 &&
          looseMuons_.size() + selectedElectrons_.size() + looseElectrons_.size() == 0
          );
      bool oneElectron =
        ( selectedElectrons_.size() == 1 &&
          selectedMuons_.size() == 0
          );

      bool oneMuonMuVeto =
        ( selectedMuons_.size() == 1 &&
          looseMuons_.size() == 0
          );


      if ( ignoreCut(lep3Index_) ||
           ePlusJets_ ||
           (muPlusJets_ && oneMuonMuVeto)
           ) {
        passCut(ret, lep3Index_);

        if ( ignoreCut(lep4Index_) ||
         ( (muPlusJets_ && oneMuon) ^ (ePlusJets_ && oneElectron )  )
         ) {
          passCut(ret, lep4Index_);

          bool metCut = met_.pt() > metMin_;
          if ( ignoreCut(metIndex_) ||
           metCut ) {
        passCut( ret, metIndex_ );


        bool zVeto = true;
        if ( selectedMuons_.size() == 2 ) {
        }
        if ( selectedElectrons_.size() == 2 ) {
        }
        if ( ignoreCut(zvetoIndex_) ||
             zVeto ){
          passCut(ret, zvetoIndex_);


          bool conversionVeto = true;
          if ( ignoreCut(conversionIndex_) ||
               conversionVeto ) {
            passCut(ret,conversionIndex_);



            bool cosmicVeto = true;
            if ( ignoreCut(cosmicIndex_) ||
             cosmicVeto ) {
              passCut(ret,cosmicIndex_);

              if ( ignoreCut(jet1Index_) ||
               static_cast<int>(cleanedJets_.size()) >=  1 ){
            passCut(ret,jet1Index_);
              } // end if >=1 tight jets

              if ( ignoreCut(jet2Index_) ||
               static_cast<int>(cleanedJets_.size()) >=  2 ){
            passCut(ret,jet2Index_);
              } // end if >=2 tight jets

              if ( ignoreCut(jet3Index_) ||
               static_cast<int>(cleanedJets_.size()) >=  3 ){
            passCut(ret,jet3Index_);
              } // end if >=3 tight jets

              if ( ignoreCut(jet4Index_) ||
               static_cast<int>(cleanedJets_.size()) >=  4 ){
            passCut(ret,jet4Index_);
              } // end if >=4 tight jets

              if ( ignoreCut(jet5Index_) ||
               static_cast<int>(cleanedJets_.size()) >=  5 ){
            passCut(ret,jet5Index_);
              } // end if >=5 tight jets



            } // end if cosmic veto

          } // end if conversion veto

        } // end if z veto

          } // end if met cut

        } // end if == 1 lepton

      } // end if == 1 tight lepton with a muon veto separately

    } // end if == 1 tight lepton

      } // end if >= 1 lepton

    } // end if PV

  } // end if trigger


  setIgnored(ret);

  return (bool)ret;
}