LeptonRecoSkim.cc

Go to the documentation of this file.

// -*- C++ -*-
//
// Package:    LeptonRecoSkim
// Class:      LeptonRecoSkim
//
//
// Original Author:  Massimiliano Chiorboli,40 4-A01,+41227671535,
//         Created:  Wed Mar 31 21:49:08 CEST 2010
// $Id: LeptonRecoSkim.cc,v 1.1 2010/11/05 18:37:51 torimoto Exp $
//
//

#include "Configuration/Skimming/interface/LeptonRecoSkim.h"

using namespace edm;
using namespace reco;
using namespace std;

//
// constructors and destructor
//
LeptonRecoSkim::LeptonRecoSkim(const edm::ParameterSet& iConfig)
    : hltLabel(iConfig.getParameter<edm::InputTag>("HltLabel")),
      filterName(iConfig.getParameter<std::string>("@module_label")),
      m_electronSrc(iConfig.getParameter<edm::InputTag>("electronCollection")),
      m_pfelectronSrc(iConfig.getParameter<edm::InputTag>("pfElectronCollection")),
      m_muonSrc(iConfig.getParameter<edm::InputTag>("muonCollection")),
      m_jetsSrc(iConfig.getParameter<edm::InputTag>("caloJetCollection")),
      m_pfjetsSrc(iConfig.getParameter<edm::InputTag>("PFJetCollection")),
      m_ebRecHitsSrc(iConfig.getParameter<edm::InputTag>("ecalBarrelRecHitsCollection")),
      m_eeRecHitsSrc(iConfig.getParameter<edm::InputTag>("ecalEndcapRecHitsCollection")),
      useElectronSelection(iConfig.getParameter<bool>("UseElectronSelection")),
      usePfElectronSelection(iConfig.getParameter<bool>("UsePfElectronSelection")),
      useMuonSelection(iConfig.getParameter<bool>("UseMuonSelection")),
      useHtSelection(iConfig.getParameter<bool>("UseHtSelection")),
      usePFHtSelection(iConfig.getParameter<bool>("UsePFHtSelection")),
      ptElecMin(iConfig.getParameter<double>("electronPtMin")),
      ptPfElecMin(iConfig.getParameter<double>("pfElectronPtMin")),
      nSelectedElectrons(iConfig.getParameter<int>("electronN")),
      nSelectedPfElectrons(iConfig.getParameter<int>("pfElectronN")),
      ptGlobalMuonMin(iConfig.getParameter<double>("globalMuonPtMin")),
      ptTrackerMuonMin(iConfig.getParameter<double>("trackerMuonPtMin")),
      nSelectedMuons(iConfig.getParameter<int>("muonN")),
      htMin(iConfig.getParameter<double>("HtMin")),
      pfHtMin(iConfig.getParameter<double>("PFHtMin")),
      htJetThreshold(iConfig.getParameter<double>("HtJetThreshold")),
      pfHtJetThreshold(iConfig.getParameter<double>("PFHtJetThreshold")) {
  NeventsTotal = 0;
  NeventsFiltered = 0;
  NHltMu9 = 0;
  NHltDiMu3 = 0;

  NtotalElectrons = 0;
  NmvaElectrons = 0;
}

LeptonRecoSkim::~LeptonRecoSkim() {
  // do anything here that needs to be done at desctruction time
  // (e.g. close files, deallocate resources etc.)
}

//
// member functions
//

// ------------ method called on each new Event  ------------
bool LeptonRecoSkim::filter(edm::Event& iEvent, const edm::EventSetup& iSetup) {
  bool accept = false;

  NeventsTotal++;

  ElectronCutPassed = false;
  PfElectronCutPassed = false;
  MuonCutPassed = false;
  HtCutPassed = false;
  PFHtCutPassed = false;

  //  edm::Handle<TriggerResults> trhv;
  //  iEvent.getByLabel(hltLabel,trhv);

  //  const edm::TriggerNames& triggerNames_ = iEvent.triggerNames(*trhv);

  //  if(trhv->at(triggerNames_.triggerIndex("HLT_Mu9")).accept()) NHltMu9++;
  //  if(trhv->at(triggerNames_.triggerIndex("HLT_DoubleMu3")).accept()) NHltDiMu3++;

  this->handleObjects(iEvent, iSetup);

  if (useElectronSelection) {
    int nElecPassingCut = 0;
    for (unsigned int i = 0; i < theElectronCollection->size(); i++) {
      GsfElectron electron = (*theElectronCollection)[i];
      //      if (electron.ecalDrivenSeed()) {
      float elpt = electron.pt();
      //        if (electron.sigmaIetaIeta() < 0.002) continue;
      if (elpt > ptElecMin) {
        NtotalElectrons++;
        nElecPassingCut++;
        if (electron.mva_e_pi() > -0.1)
          NmvaElectrons++;
      }
      LogDebug("LeptonRecoSkim") << "elpt = " << elpt << endl;
      // } // closes if (electron.ecalDrivenSeed()) {
    }
    if (nElecPassingCut >= nSelectedElectrons)
      ElectronCutPassed = true;
  } else
    ElectronCutPassed = true;

  if (usePfElectronSelection) {
    int nPfElecPassingCut = 0;
    for (unsigned int i = 0; i < thePfCandidateCollection->size(); i++) {
      const reco::PFCandidate& thePfCandidate = (*thePfCandidateCollection)[i];
      if (thePfCandidate.particleId() != reco::PFCandidate::e)
        continue;
      if (thePfCandidate.gsfTrackRef().isNull())
        continue;
      float pfelpt = thePfCandidate.pt();
      //        if (electron.sigmaIetaIeta() < 0.002) continue;
      if (pfelpt > ptPfElecMin)
        nPfElecPassingCut++;
      LogDebug("LeptonRecoSkim") << "pfelpt = " << pfelpt << endl;
    }
    if (nPfElecPassingCut >= nSelectedPfElectrons)
      PfElectronCutPassed = true;
  } else
    PfElectronCutPassed = true;

  if (useMuonSelection) {
    int nMuonPassingCut = 0;
    for (unsigned int i = 0; i < theMuonCollection->size(); i++) {
      Muon muon = (*theMuonCollection)[i];
      if (!(muon.isGlobalMuon() || muon.isTrackerMuon()))
        continue;
      const TrackRef siTrack = muon.innerTrack();
      const TrackRef globalTrack = muon.globalTrack();
      float muonpt;
      float ptMuonMin;
      //      if (siTrack.isNonnull() && globalTrack.isNonnull()) {
      if (muon.isGlobalMuon() && muon.isTrackerMuon()) {
        muonpt = max(siTrack->pt(), globalTrack->pt());
        ptMuonMin = ptGlobalMuonMin;
      } else if (muon.isGlobalMuon() && !(muon.isTrackerMuon())) {
        muonpt = globalTrack->pt();
        ptMuonMin = ptGlobalMuonMin;
      } else if (muon.isTrackerMuon() && !(muon.isGlobalMuon())) {
        muonpt = siTrack->pt();
        ptMuonMin = ptTrackerMuonMin;
      } else {
        muonpt = 0;  // if we get here this is a STA only muon
        ptMuonMin = 999999.;
      }
      if (muonpt > ptMuonMin)
        nMuonPassingCut++;
      LogDebug("RecoSelectorCuts") << "muonpt = " << muonpt << endl;
    }
    if (nMuonPassingCut >= nSelectedMuons)
      MuonCutPassed = true;
  } else
    MuonCutPassed = true;

  if (useHtSelection) {
    double Ht = 0;
    for (unsigned int i = 0; i < theCaloJetCollection->size(); i++) {
      //      if((*theCaloJetCollection)[i].eta()<2.6 && (*theCaloJetCollection)[i].emEnergyFraction() <= 0.01) continue;
      if ((*theCaloJetCollection)[i].pt() > htJetThreshold)
        Ht += (*theCaloJetCollection)[i].pt();
    }
    if (Ht > htMin)
      HtCutPassed = true;
  } else
    HtCutPassed = true;

  if (usePFHtSelection) {
    double PFHt = 0;
    for (unsigned int i = 0; i < thePFJetCollection->size(); i++) {
      if ((*thePFJetCollection)[i].pt() > pfHtJetThreshold)
        PFHt += (*thePFJetCollection)[i].pt();
    }
    if (PFHt > pfHtMin)
      PFHtCutPassed = true;
  } else
    PFHtCutPassed = true;

  if (PfElectronCutPassed && ElectronCutPassed && MuonCutPassed && HtCutPassed && PFHtCutPassed)
    accept = true;

  if (accept)
    NeventsFiltered++;

  firstEvent = false;
  return accept;
}

// ------------ method called once each job just before starting event loop  ------------
void LeptonRecoSkim::beginJob() { firstEvent = true; }

// ------------ method called once each job just after ending the event loop  ------------
void LeptonRecoSkim::endJob() {
  cout << "Filter Name            = " << filterName << endl;
  cout << "Total number of events = " << NeventsTotal << endl;
  cout << "Total HLT_Mu9          = " << NHltMu9 << endl;
  cout << "Total HLT_DoubleMu3    = " << NHltDiMu3 << endl;
  cout << "Filtered events        = " << NeventsFiltered << endl;
  cout << "Filter Efficiency      = " << (float)NeventsFiltered / (float)NeventsTotal << endl;
  cout << endl;
  cout << "N total electrons      = " << NtotalElectrons << endl;
  cout << "N mva>-0.1 electrons   = " << NmvaElectrons << endl;
  cout << endl;
  cout << endl;
}

void LeptonRecoSkim::handleObjects(const edm::Event& iEvent, const edm::EventSetup& iSetup) {
  //Get the electrons
  Handle<GsfElectronCollection> theElectronCollectionHandle;
  iEvent.getByLabel(m_electronSrc, theElectronCollectionHandle);
  theElectronCollection = theElectronCollectionHandle.product();

  //Get the pf electrons
  Handle<reco::PFCandidateCollection> thePfCandidateHandle;
  iEvent.getByLabel(m_pfelectronSrc, thePfCandidateHandle);
  thePfCandidateCollection = thePfCandidateHandle.product();

  //Get the Muons
  Handle<MuonCollection> theMuonCollectionHandle;
  iEvent.getByLabel(m_muonSrc, theMuonCollectionHandle);
  theMuonCollection = theMuonCollectionHandle.product();

  //Get the CaloJets
  Handle<CaloJetCollection> theCaloJetCollectionHandle;
  iEvent.getByLabel(m_jetsSrc, theCaloJetCollectionHandle);
  theCaloJetCollection = theCaloJetCollectionHandle.product();

  //Get the PfJets
  Handle<PFJetCollection> thePFJetCollectionHandle;
  iEvent.getByLabel(m_pfjetsSrc, thePFJetCollectionHandle);
  thePFJetCollection = thePFJetCollectionHandle.product();

  //Get the ECAL rechhits to clean the spikes
  // Get EB RecHits
  edm::Handle<EcalRecHitCollection> ebRecHitsHandle;
  iEvent.getByLabel(m_ebRecHitsSrc, ebRecHitsHandle);
  theEcalBarrelCollection = ebRecHitsHandle.product();
  // Get EE RecHits
  edm::Handle<EcalRecHitCollection> eeRecHitsHandle;
  iEvent.getByLabel(m_eeRecHitsSrc, eeRecHitsHandle);
  theEcalEndcapCollection = eeRecHitsHandle.product();

  // Get topology for spike cleaning
  edm::ESHandle<CaloGeometry> geometryHandle;
  iSetup.get<CaloGeometryRecord>().get(geometryHandle);
  theCaloGeometry = geometryHandle.product();
  //   theCaloBarrelSubdetTopology = new EcalBarrelTopology(geometryHandle);
  //   theCaloEndcapSubdetTopology = new EcalEndcapTopology(geometryHandle);

  edm::ESHandle<CaloTopology> pTopology;
  iSetup.get<CaloTopologyRecord>().get(pTopology);
  theCaloTopology = pTopology.product();
}

//define this as a plug-in
DEFINE_FWK_MODULE(LeptonRecoSkim);