HiggsToZZ4LeptonsSkim.cc

Go to the documentation of this file.

 
/* \class HiggsTo4LeptonsSkim
 *
 * Consult header file for description
 *
 * author:  Dominique Fortin - UC Riverside
 *
 */
 
// system include files
#include <HiggsAnalysis/Skimming/interface/HiggsToZZ4LeptonsSkim.h>
 
// User include files
#include <FWCore/ParameterSet/interface/ParameterSet.h>
 
// Muons:
#include <DataFormats/TrackReco/interface/Track.h>
 
// Electrons
#include "DataFormats/EgammaCandidates/interface/GsfElectron.h"
 
// C++
#include <iostream>
#include <vector>
 
using namespace std;
using namespace edm;
using namespace reco;
 
// Constructor
HiggsToZZ4LeptonsSkim::HiggsToZZ4LeptonsSkim(const edm::ParameterSet& pset) {
  // Local Debug flag
  debug = pset.getParameter<bool>("DebugHiggsToZZ4LeptonsSkim");
 
  // Reconstructed objects
  theGLBMuonToken = consumes<reco::TrackCollection>(pset.getParameter<edm::InputTag>("GlobalMuonCollectionLabel"));
  theGsfEToken = consumes<reco::GsfElectronCollection>(pset.getParameter<edm::InputTag>("ElectronCollectionLabel"));
 
  // Minimum Pt for leptons for skimming
  stiffMinPt = pset.getParameter<double>("stiffMinimumPt");
  softMinPt = pset.getParameter<double>("softMinimumPt");
  nStiffLeptonMin = pset.getParameter<int>("nStiffLeptonMinimum");
  nLeptonMin = pset.getParameter<int>("nLeptonMinimum");
 
  nEvents = 0;
  nSelectedEvents = 0;
}
 
// Destructor
HiggsToZZ4LeptonsSkim::~HiggsToZZ4LeptonsSkim() {
  edm::LogVerbatim("HiggsToZZ4LeptonsSkim")
      << " Number_events_read " << nEvents << " Number_events_kept " << nSelectedEvents << " Efficiency         "
      << ((double)nSelectedEvents) / ((double)nEvents + 0.01) << std::endl;
}
 
// Filter event
bool HiggsToZZ4LeptonsSkim::filter(edm::Event& event, const edm::EventSetup& setup) {
  nEvents++;
 
  using reco::TrackCollection;
 
  bool keepEvent = false;
  int nStiffLeptons = 0;
  int nLeptons = 0;
 
  // First look at muons:
 
  // Get the muon track collection from the event
  edm::Handle<reco::TrackCollection> muTracks;
  event.getByToken(theGLBMuonToken, muTracks);
 
  if (muTracks.isValid()) {
    reco::TrackCollection::const_iterator muons;
 
    // Loop over muon collections and count how many muons there are,
    // and how many are above threshold
    for (muons = muTracks->begin(); muons != muTracks->end(); ++muons) {
      if (muons->pt() > stiffMinPt)
        nStiffLeptons++;
      if (muons->pt() > softMinPt)
        nLeptons++;
    }
  }
 
  // Now look at electrons:
 
  // Get the electron track collection from the event
  edm::Handle<reco::GsfElectronCollection> pTracks;
  event.getByToken(theGsfEToken, pTracks);
 
  if (pTracks.isValid()) {
    const reco::GsfElectronCollection* eTracks = pTracks.product();
 
    reco::GsfElectronCollection::const_iterator electrons;
 
    // Loop over electron collections and count how many muons there are,
    // and how many are above threshold
    for (electrons = eTracks->begin(); electrons != eTracks->end(); ++electrons) {
      float pt_e = electrons->pt();
      if (pt_e > stiffMinPt)
        nStiffLeptons++;
      if (pt_e > softMinPt)
        nLeptons++;
    }
  }
 
  // Make decision:
  if (nStiffLeptons >= nStiffLeptonMin && nLeptons >= nLeptonMin)
    keepEvent = true;
 
  if (keepEvent)
    nSelectedEvents++;
 
  return keepEvent;
}