ElectronSelector.h

Go to the documentation of this file.

#include "DataFormats/EgammaCandidates/interface/GsfElectronFwd.h"
#include "DataFormats/Common/interface/Handle.h"
#include "DataFormats/HLTReco/interface/TriggerEvent.h"
#include "DataFormats/Math/interface/deltaR.h"
#include "DataFormats/Math/interface/LorentzVector.h"
#include <vector>
 
std::vector<reco::GsfElectronRef> electronSelector(const std::vector<reco::GsfElectronRef>& electrons,
                                                   const edm::Handle<trigger::TriggerEvent>& pHLT,
                                                   const int filterId,
                                                   const std::vector<double>& Cuts) {
  std::vector<reco::GsfElectronRef> ChosenOnes;
  const trigger::Keys& ring = pHLT->filterKeys(filterId);
  const trigger::TriggerObjectCollection& HltObjColl = pHLT->getObjects();
  //const edm::ValueMap<double>& eIsoMapTrk = *eIsoMap[0];
  //const edm::ValueMap<double>& eIsoMapEcal = *eIsoMap[1];
  //const edm::ValueMap<double>& eIsoMapHcal = *eIsoMap[2];
  edm::LogDebug_("electronSelector", "", 16) << "Number of electrons to select from = " << electrons.size();
  for (std::vector<reco::GsfElectronRef>::const_iterator Relec = electrons.begin(); Relec != electrons.end(); ++Relec) {
    reco::GsfElectronRef elec = *Relec;
    edm::LogDebug_("electronSelector", "", 17) << "Analysing elec, id = " << elec.id() << "\tkey = " << elec.key();
    double scEta = elec->superCluster()->eta();
    if (fabs(scEta) < 1.4442 || fabs(scEta) > 1.56) {
      bool HLTMatch = false;
      for (unsigned int k = 0; k < ring.size(); ++k) {
        const trigger::TriggerObject& HltObj = HltObjColl[ring[k]];
        if (reco::deltaR(*elec, HltObj) < 0.1)
          HLTMatch = true;
      }
      edm::LogDebug_("electronSelector", "", 16) << "HLT Match = " << HLTMatch;
      std::cout << "HLT Match = " << HLTMatch << std::endl;
      //            if(HLTMatch) ChosenOnes.push_back(elec);
      if (HLTMatch) {
        if (fabs(scEta) < 1.479) {
          float sIhIh = elec->scSigmaIEtaIEta();
          edm::LogDebug_("electronSelector", "", 32) << "SigIetaIeta = " << sIhIh << "\tCut Value = " << Cuts[1];
          if (sIhIh < Cuts[1]) {
            edm::LogDebug_("elecSel", "", 39)
                << "dEta = " << elec->deltaEtaSuperClusterTrackAtVtx() << "\tCut Value = " << Cuts[2];
            if (fabs(elec->deltaEtaSuperClusterTrackAtVtx()) < Cuts[2]) {
              edm::LogDebug_("elecSel", "", 39)
                  << "dPhi = " << elec->deltaPhiSuperClusterTrackAtVtx() << "\tCut Value = " << Cuts[3];
              if (fabs(elec->deltaPhiSuperClusterTrackAtVtx()) < Cuts[3]) {
                float trckiso = elec->isolationVariables03().tkSumPt;
                edm::LogDebug_("", "", 29) << "Track isolation = " << trckiso << "\tCut Value = " << Cuts[4];
                if (trckiso < Cuts[4]) {
                  float ecaliso = elec->isolationVariables04().ecalRecHitSumEt;
                  edm::LogDebug_("", "", 29) << "ECAL isolation = " << ecaliso << "\tCut Value = " << Cuts[5];
                  if (ecaliso < Cuts[5]) {
                    float hcaliso = elec->isolationVariables04().hcalDepth1TowerSumEt +
                                    elec->isolationVariables04().hcalDepth2TowerSumEt;
                    edm::LogDebug_("", "", 29) << "HCAL isolation = " << hcaliso << "\tCut Value = " << Cuts[6];
                    if (hcaliso < Cuts[6])
                      ChosenOnes.push_back(elec);
                  }
                }
              }
            }
          }
        } else {
          float sIhIh = elec->scSigmaIEtaIEta();
          edm::LogDebug_("electronSelector", "", 32) << "SigIetaIeta = " << sIhIh << "\tCut Value = " << Cuts[7];
          if (sIhIh < Cuts[7]) {
            edm::LogDebug_("elecSel", "", 39)
                << "dEta = " << elec->deltaEtaSuperClusterTrackAtVtx() << "\tCut Value = " << Cuts[8];
            if (fabs(elec->deltaEtaSuperClusterTrackAtVtx()) < Cuts[8]) {
              edm::LogDebug_("elecSel", "", 39)
                  << "dPhi = " << elec->deltaPhiSuperClusterTrackAtVtx() << "\tCut Value = " << Cuts[9];
              if (fabs(elec->deltaPhiSuperClusterTrackAtVtx()) < Cuts[9]) {
                float trckiso = elec->isolationVariables03().tkSumPt;
                edm::LogDebug_("", "", 29) << "Track isolation = " << trckiso << "\tCut Value = " << Cuts[10];
                if (trckiso < Cuts[10]) {
                  float ecaliso = elec->isolationVariables04().ecalRecHitSumEt;
                  edm::LogDebug_("", "", 29) << "ECAL isolation = " << ecaliso << "\tCut Value = " << Cuts[11];
                  if (ecaliso < Cuts[11]) {
                    float hcaliso = elec->isolationVariables04().hcalDepth1TowerSumEt +
                                    elec->isolationVariables04().hcalDepth2TowerSumEt;
                    edm::LogDebug_("", "", 29) << "HCAL isolation = " << hcaliso << "\tCut Value = " << Cuts[12];
                    if (hcaliso < Cuts[12])
                      ChosenOnes.push_back(elec);
                  }
                }
              }
            }
          }
        }
        //              ChosenOnes.push_back(elec);
      }
    }
  }
  return ChosenOnes;
}