HLTElectronOneOEMinusOneOPFilterRegional.cc

Go to the documentation of this file.

#include "HLTElectronOneOEMinusOneOPFilterRegional.h"

#include "DataFormats/Common/interface/Handle.h"

#include "DataFormats/RecoCandidate/interface/RecoEcalCandidate.h"
#include "DataFormats/RecoCandidate/interface/RecoEcalCandidateFwd.h"

#include "FWCore/ParameterSet/interface/ConfigurationDescriptions.h"
#include "FWCore/ParameterSet/interface/ParameterSetDescription.h"
#include "FWCore/MessageLogger/interface/MessageLogger.h"

#include "DataFormats/TrackReco/interface/Track.h"
#include "DataFormats/EgammaReco/interface/SuperCluster.h"

//
// constructors and destructor
//
HLTElectronOneOEMinusOneOPFilterRegional::HLTElectronOneOEMinusOneOPFilterRegional(const edm::ParameterSet& iConfig) : HLTFilter(iConfig)
{
  candTag_ = iConfig.getParameter< edm::InputTag > ("candTag");
  electronIsolatedProducer_ = iConfig.getParameter< edm::InputTag > ("electronIsolatedProducer");
  electronNonIsolatedProducer_ = iConfig.getParameter< edm::InputTag > ("electronNonIsolatedProducer");
  barrelcut_  = iConfig.getParameter<double> ("barrelcut");
  endcapcut_  = iConfig.getParameter<double> ("endcapcut");
  ncandcut_  = iConfig.getParameter<int> ("ncandcut");
  doIsolated_  = iConfig.getParameter<bool> ("doIsolated");
  candToken_ =  consumes<trigger::TriggerFilterObjectWithRefs>(candTag_);
  electronIsolatedToken_ = consumes<reco::ElectronCollection>(electronIsolatedProducer_);
  if(!doIsolated_) electronNonIsolatedToken_ = consumes<reco::ElectronCollection>(electronNonIsolatedProducer_);
}

void
HLTElectronOneOEMinusOneOPFilterRegional::fillDescriptions(edm::ConfigurationDescriptions& descriptions) {
  edm::ParameterSetDescription desc;
  makeHLTFilterDescription(desc);
  desc.add<edm::InputTag>("candTag",edm::InputTag("hltL1NonIsoHLTNonIsoSingleElectronEt15LTIPixelMatchFilter"));
  desc.add<edm::InputTag>("electronIsolatedProducer",edm::InputTag("hltPixelMatchElectronsL1Iso"));
  desc.add<edm::InputTag>("electronNonIsolatedProducer",edm::InputTag("hltPixelMatchElectronsL1NonIso"));
  desc.add<double>("barrelcut",999.9);
  desc.add<double>("endcapcut",999.9);
  desc.add<int>("ncandcut",1);
  desc.add<bool>("doIsolated",false);
  descriptions.add("hltElectronOneOEMinusOneOPFilterRegional",desc);
}

HLTElectronOneOEMinusOneOPFilterRegional::~HLTElectronOneOEMinusOneOPFilterRegional()= default;


// ------------ method called to produce the data  ------------
bool
HLTElectronOneOEMinusOneOPFilterRegional::hltFilter(edm::Event& iEvent, const edm::EventSetup& iSetup, trigger::TriggerFilterObjectWithRefs & filterproduct) const
{
  using namespace trigger;

  // The filter object
  if (saveTags()) {
    filterproduct.addCollectionTag(electronIsolatedProducer_);
    if (not doIsolated_) filterproduct.addCollectionTag(electronNonIsolatedProducer_);
  }
  //will be a collection of Ref<reco::ElectronCollection> ref;

  edm::Handle<trigger::TriggerFilterObjectWithRefs> PrevFilterOutput;
  iEvent.getByToken (candToken_,PrevFilterOutput);

  std::vector<edm::Ref<reco::RecoEcalCandidateCollection> > recoecalcands;
  PrevFilterOutput->getObjects(TriggerCluster, recoecalcands);
  if(recoecalcands.empty()) PrevFilterOutput->getObjects(TriggerPhoton,recoecalcands);

  edm::Handle<reco::ElectronCollection> electronIsolatedHandle;
  iEvent.getByToken(electronIsolatedToken_,electronIsolatedHandle);

  edm::Handle<reco::ElectronCollection> electronNonIsolatedHandle;
  if(!doIsolated_) {
    iEvent.getByToken(electronNonIsolatedToken_,electronNonIsolatedHandle);
  }

 // look at all candidates,  check cuts and add to filter object
  int n(0);

    //loop over all the RecoCandidates from the previous filter,
    // associate them with the corresponding Electron object
    //(the matching is done checking the super clusters)
    // and put into the event a Ref to the Electron objects that passes the
    // selections

  edm::RefToBase<reco::Candidate> candref;
  for (auto & recoecalcand : recoecalcands) {

    reco::SuperClusterRef recr2 = recoecalcand->superCluster();
    //loop over the electrons to find the matching one
    for(auto iElectron = electronIsolatedHandle->begin(); iElectron != electronIsolatedHandle->end(); iElectron++){
      // ElectronRef is a Ref<reco::RecoEcalCandidateCollection>
      reco::ElectronRef electronref(reco::ElectronRef(electronIsolatedHandle,iElectron - electronIsolatedHandle->begin()));
      const reco::SuperClusterRef theClus = electronref->superCluster();
      if(&(*recr2) ==  &(*theClus)) {
    
    float elecEoverp = 0;
    const math::XYZVector trackMom =  electronref->track()->momentum();
    if( trackMom.R() != 0) elecEoverp =
                 fabs((1/electronref->superCluster()->energy()) - (1/trackMom.R()));

    if( fabs(electronref->eta()) < 1.5 ){
      if ( elecEoverp < barrelcut_) {
        n++;
        filterproduct.addObject(TriggerElectron, electronref);
      }
    }
    if( fabs(electronref->eta()) > 1.5 ) {
      if ( elecEoverp < endcapcut_) {
        n++;
        filterproduct.addObject(TriggerElectron, electronref);
      }
    }
      }//end of the if checking the matching of the SC from RecoCandidate and the one from Electrons
    }//end of loop over electrons

    if(!doIsolated_) {
    //loop over the electrons to find the matching one
    for(auto iElectron = electronNonIsolatedHandle->begin(); iElectron != electronNonIsolatedHandle->end(); iElectron++){

      reco::ElectronRef electronref(reco::ElectronRef(electronNonIsolatedHandle,iElectron - electronNonIsolatedHandle->begin()));
      const reco::SuperClusterRef theClus = electronref->superCluster();

      if(&(*recr2) ==  &(*theClus)) {

    float elecEoverp = 0;
    const math::XYZVector trackMom =  electronref->track()->momentum();
    if( trackMom.R() != 0) elecEoverp =
                fabs((1/electronref->superCluster()->energy()) - (1/trackMom.R()));

    if( fabs(electronref->eta()) < 1.5 ){
      if ( elecEoverp < barrelcut_) {
        n++;
        filterproduct.addObject(TriggerElectron, electronref);
      }
    }
    if( fabs(electronref->eta()) > 1.5 ){
      if ( elecEoverp < endcapcut_) {
        n++;
        filterproduct.addObject(TriggerElectron, electronref);
      }
    }
      }//end of the if checking the matching of the SC from RecoCandidate and the one from Electrons
    }//end of loop over electrons
    }
  }//end of loop ober candidates

  // filter decision
  bool accept(n>=ncandcut_);

  return accept;
}

// declare this class as a framework plugin
#include "FWCore/Framework/interface/MakerMacros.h"
DEFINE_FWK_MODULE(HLTElectronOneOEMinusOneOPFilterRegional);