ElectronRecalibSuperClusterAssociator.cc

Go to the documentation of this file.

// -*- C++ -*-

// user include files
#include "FWCore/Framework/interface/MakerMacros.h"
#include "Calibration/EcalCalibAlgos/interface/ElectronRecalibSuperClusterAssociator.h"

#include "DataFormats/GsfTrackReco/interface/GsfTrack.h"
#include <iostream>
#include "DataFormats/Math/interface/LorentzVector.h"
#include "DataFormats/Math/interface/deltaR.h"

using namespace reco;
using namespace edm;

ElectronRecalibSuperClusterAssociator::ElectronRecalibSuperClusterAssociator(const edm::ParameterSet& iConfig) {
#ifdef DEBUG
  std::cout << "ElectronRecalibSuperClusterAssociator::ElectronRecalibSuperClusterAssociator" << std::endl;
#endif

  //register your products
  produces<GsfElectronCollection>();
  produces<GsfElectronCoreCollection>();

  superClusterCollectionEB_ = iConfig.getParameter<edm::InputTag>("superClusterCollectionEB");
  superClusterCollectionEE_ = iConfig.getParameter<edm::InputTag>("superClusterCollectionEE");

  outputLabel_ = iConfig.getParameter<std::string>("outputLabel");
  electronSrc_ = iConfig.getParameter<edm::InputTag>("electronSrc");

  electronToken_ = consumes<reco::GsfElectronCollection>(electronSrc_);
  ebScToken_ = consumes<reco::SuperClusterCollection>(superClusterCollectionEB_);
  eeScToken_ = consumes<reco::SuperClusterCollection>(superClusterCollectionEE_);

#ifdef DEBUG
  std::cout << "ElectronRecalibSuperClusterAssociator::ElectronRecalibSuperClusterAssociator::end" << std::endl;
#endif
}

ElectronRecalibSuperClusterAssociator::~ElectronRecalibSuperClusterAssociator() {}

// ------------ method called to produce the data  ------------
void ElectronRecalibSuperClusterAssociator::produce(edm::Event& e, const edm::EventSetup& iSetup) {
#ifdef DEBUG
  std::cout << "GEDElectronRecalibSuperClusterAssociator::produce" << std::endl;
#endif

  // Create the output collections
  auto pOutEle = std::make_unique<GsfElectronCollection>();
  auto pOutEleCore = std::make_unique<GsfElectronCoreCollection>();

  // Get SuperClusters in EB
  Handle<reco::SuperClusterCollection> superClusterEBHandle;
  e.getByToken(ebScToken_, superClusterEBHandle);

#ifdef DEBUG
  std::cout << "EB scCollection->size()" << superClusterEBHandle->size() << std::endl;
#endif

  // Get SuperClusters in EE
  Handle<reco::SuperClusterCollection> superClusterEEHandle;
  e.getByToken(eeScToken_, superClusterEEHandle);

#ifdef DEBUG
  std::cout << "EE scCollection->size()" << superClusterEEHandle->size() << std::endl;
#endif

  // Get Electrons
  edm::Handle<reco::GsfElectronCollection> eleHandle;
  e.getByToken(electronToken_, eleHandle);

  GsfElectronCoreRefProd rEleCore = e.getRefBeforePut<GsfElectronCoreCollection>();
  edm::Ref<GsfElectronCoreCollection>::key_type idxEleCore = 0;

  for (reco::GsfElectronCollection::const_iterator eleIt = eleHandle->begin(); eleIt != eleHandle->end(); eleIt++) {
    float DeltaRMineleSCbarrel(0.15);  //initial minDeltaR
    float DeltaRMineleSCendcap(0.15);
    const reco::SuperCluster* nearestSCbarrel = nullptr;
    const reco::SuperCluster* nearestSCendcap = nullptr;
    int iscRef = -1, iscRefendcap = -1;
    int iSC = 0;

    if (eleIt->trackerDrivenSeed()) {
      edm::LogError("trackerDriven") << "skipping trackerDriven electrons";
      continue;
    }
    // first loop is on EB superClusters
    iSC = 0;
    for (reco::SuperClusterCollection::const_iterator scIt = superClusterEBHandle->begin();
         scIt != superClusterEBHandle->end();
         scIt++, iSC++) {
      double DeltaReleSC = sqrt(reco::deltaR2(eleIt->eta(), eleIt->phi(), scIt->eta(), scIt->phi()));

      if (DeltaReleSC < DeltaRMineleSCbarrel)  //save the nearest SC
      {
        DeltaRMineleSCbarrel = DeltaReleSC;
        nearestSCbarrel = &*scIt;
        iscRef = iSC;
      }
#ifdef DEBUG
      std::cout << "EB: " << scIt - superClusterEBHandle->begin() << " " << iSC << " " << iscRef << "\t"
                << std::setprecision(4) << scIt->energy() << " " << scIt->eta() << " " << scIt->phi() << "\t--\t"
                << eleIt->energy() << " " << eleIt->eta() << " " << eleIt->phi() << "\t" << DeltaRMineleSCbarrel
                << std::endl;
#endif
    }

    // second loop is on EE superClusters
    iSC = 0;
    for (reco::SuperClusterCollection::const_iterator scIt = superClusterEEHandle->begin();
         scIt != superClusterEEHandle->end();
         scIt++, iSC++) {
#ifdef DEBUG
      std::cout << "EE: " << scIt - superClusterEEHandle->begin() << " " << iSC << " " << iscRef << "\t"
                << std::setprecision(4) << scIt->energy() << " " << scIt->eta() << " " << scIt->phi() << "\t--\t "
                << eleIt->energy() << " " << eleIt->eta() << " " << eleIt->phi() << "\t" << DeltaRMineleSCendcap
                << std::endl;
#endif

      double DeltaReleSC = sqrt(reco::deltaR2(eleIt->eta(), eleIt->phi(), scIt->eta(), scIt->phi()));

      if (DeltaReleSC < DeltaRMineleSCendcap) {
        DeltaRMineleSCendcap = DeltaReleSC;
        nearestSCendcap = &*scIt;
        iscRefendcap = iSC;
      }
    }
    if (eleIt->isEB() && DeltaRMineleSCbarrel > DeltaRMineleSCendcap) {
      edm::LogError("ElectronRecalibAssociator") << "EB electron, but nearest SC is in EE";
      ;
      continue;
    }

    if (eleIt->isEB() && nearestSCbarrel) {
      pOutEleCore->push_back(*eleIt->core());  // clone the old core and add to the collection of new cores
      reco::GsfElectronCoreRef newEleCoreRef(rEleCore,
                                             idxEleCore++);  // reference to the new electron core in the new collection
      reco::GsfElectronCore& newEleCore = pOutEleCore->back();                           // pick the clone
      reco::SuperClusterRef scRef(reco::SuperClusterRef(superClusterEBHandle, iscRef));  // Reference to the new SC
#ifndef CMSSW_5_3_X
      newEleCore.setParentSuperCluster(scRef);  // mustache
#endif
      newEleCore.setSuperCluster(scRef);  // let's check this! if it is possible to recreate the pfSC

      pOutEle->push_back(reco::GsfElectron(*eleIt, newEleCoreRef));
      reco::GsfElectron& newEle = pOutEle->back();

      //-- first possibility: set the new p4SC using refined SC
      newEle.setP4(reco::GsfElectron::P4_FROM_SUPER_CLUSTER,
                   eleIt->p4(reco::GsfElectron::P4_FROM_SUPER_CLUSTER),
                   eleIt->p4Error(reco::GsfElectron::P4_FROM_SUPER_CLUSTER),
                   false);  //*newEle.superCluster()->energy()/eleIt->superCluster()->energy());

      //-- second possibility: set the new p4SC using mustache SC
      //newEle.setP4(reco::GsfElectron::P4_FROM_SUPER_CLUSTER, eleIt->p4(reco::GsfElectron::P4_FROM_SUPER_CLUSTER)*newEle.parentSuperCluster()->energy()/eleIt->parentSuperCluster()->energy(), eleIt->p4Error(reco::GsfElectron::P4_FROM_SUPER_CLUSTER), false);

      //-- update the correctedEcalEnergy
      newEle.setCorrectedEcalEnergy(eleIt->ecalEnergy() *
                                    (scRef->energy() / eleIt->p4(reco::GsfElectron::P4_FROM_SUPER_CLUSTER).energy()));
      newEle.setCorrectedEcalEnergyError(eleIt->ecalEnergyError() * (scRef->energy() / eleIt->ecalEnergy()));

    } else if (!(eleIt->isEB()) && nearestSCendcap) {
      pOutEleCore->push_back(*eleIt->core());  // clone the old core and add to the collection of new cores
      reco::GsfElectronCoreRef newEleCoreRef(rEleCore,
                                             idxEleCore++);  // reference to the new electron core in the new collection
      reco::GsfElectronCore& newEleCore = pOutEleCore->back();  // pick the clone
      reco::SuperClusterRef scRef(
          reco::SuperClusterRef(superClusterEEHandle, iscRefendcap));  // Reference to the new SC
#ifndef CMSSW_5_3_X
      newEleCore.setParentSuperCluster(scRef);  // mustache
#endif
      newEleCore.setSuperCluster(scRef);  // let's check this! if it is possible to recreate the pfSC

      pOutEle->push_back(reco::GsfElectron(*eleIt, newEleCoreRef));
      reco::GsfElectron& newEle = pOutEle->back();

      //-- first possibility: set the new p4SC using refined SC
      newEle.setP4(reco::GsfElectron::P4_FROM_SUPER_CLUSTER,
                   eleIt->p4(reco::GsfElectron::P4_FROM_SUPER_CLUSTER),
                   eleIt->p4Error(reco::GsfElectron::P4_FROM_SUPER_CLUSTER),
                   false);  //*newEle.superCluster()->energy()/eleIt->superCluster()->energy());

      //-- second possibility: set the new p4SC using mustache SC
      //newEle.setP4(reco::GsfElectron::P4_FROM_SUPER_CLUSTER, eleIt->p4(reco::GsfElectron::P4_FROM_SUPER_CLUSTER)*newEle.parentSuperCluster()->energy()/eleIt->parentSuperCluster()->energy(), eleIt->p4Error(reco::GsfElectron::P4_FROM_SUPER_CLUSTER), false);

      //-- update the correctedEcalEnergy
      newEle.setCorrectedEcalEnergy(eleIt->ecalEnergy() *
                                    (scRef->energy() / eleIt->p4(reco::GsfElectron::P4_FROM_SUPER_CLUSTER).energy()));
      newEle.setCorrectedEcalEnergyError(eleIt->ecalEnergyError() * (scRef->energy() / eleIt->ecalEnergy()));
    } else {
      edm::LogError("Failed SC association") << "No SC to be associated to the electron";
    }
  }

#ifdef DEBUG
  std::cout << "Filled new electrons  " << pOutEle->size() << std::endl;
  std::cout << "Filled new electronsCore  " << pOutEleCore->size() << std::endl;
#endif

  // put result into the Event

  e.put(std::move(pOutEle));
  e.put(std::move(pOutEleCore));
}

DEFINE_FWK_MODULE(ElectronRecalibSuperClusterAssociator);