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;
}