EG8XObjectUpdateModifier.cc

Go to the documentation of this file.

#include "CommonTools/CandAlgos/interface/ModifyObjectValueBase.h"
#include "FWCore/Utilities/interface/InputTag.h"
#include "FWCore/Utilities/interface/EDGetToken.h"
#include "FWCore/Framework/interface/ESHandle.h" 
#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "DataFormats/Common/interface/Handle.h"
#include "Geometry/CaloTopology/interface/CaloTopology.h"
#include "Geometry/Records/interface/CaloGeometryRecord.h"
#include "Geometry/CaloEventSetup/interface/CaloTopologyRecord.h"
#include "RecoEcal/EgammaCoreTools/interface/EcalClusterTools.h"
#include "DataFormats/EcalRecHit/interface/EcalRecHitCollections.h"
#include "DataFormats/EcalRecHit/interface/EcalRecHit.h"
#include "DataFormats/EcalDetId/interface/EcalSubdetector.h"

#include <vdt/vdtMath.h>

//this modifier fills variables where not present in CMSSW_8X 
//use case is when reading older 80X samples in newer releases, aka legacy

class EG8XObjectUpdateModifier : public ModifyObjectValueBase {
public:
  EG8XObjectUpdateModifier(const edm::ParameterSet& conf);
  ~EG8XObjectUpdateModifier() override{}
  
  void setEvent(const edm::Event&) final;
  void setEventContent(const edm::EventSetup&) final;
  void setConsumes(edm::ConsumesCollector&) final;
  
  void modifyObject(reco::GsfElectron& ele) const final;
  void modifyObject(reco::Photon& pho) const final;
  
  void modifyObject(pat::Electron& ele) const final{return modifyObject(static_cast<reco::GsfElectron&>(ele));}
  void modifyObject(pat::Photon& pho) const final{return modifyObject(static_cast<reco::Photon&>(pho));}
  
private:
  std::pair<int,bool> getSaturationInfo(const reco::SuperCluster& superClus)const;
  
  edm::ESHandle<CaloTopology> caloTopoHandle_;
  edm::Handle<EcalRecHitCollection> ecalRecHitsEBHandle_;
  edm::Handle<EcalRecHitCollection> ecalRecHitsEEHandle_;
  edm::EDGetTokenT<EcalRecHitCollection> ecalRecHitsEBToken_;
  edm::EDGetTokenT<EcalRecHitCollection> ecalRecHitsEEToken_;
  edm::InputTag ecalRecHitsEBTag_;
  edm::InputTag ecalRecHitsEETag_;
  
};

EG8XObjectUpdateModifier::EG8XObjectUpdateModifier(const edm::ParameterSet& conf):
  ModifyObjectValueBase(conf),
  ecalRecHitsEBTag_(conf.getParameter<edm::InputTag>("ecalRecHitsEB")),
  ecalRecHitsEETag_(conf.getParameter<edm::InputTag>("ecalRecHitsEE"))
{
  

}

void EG8XObjectUpdateModifier::setConsumes(edm::ConsumesCollector& cc)
{
  ecalRecHitsEBToken_ = cc.consumes<EcalRecHitCollection>(ecalRecHitsEBTag_);
  ecalRecHitsEEToken_ = cc.consumes<EcalRecHitCollection>(ecalRecHitsEETag_);
}

void EG8XObjectUpdateModifier::setEvent(const edm::Event& iEvent)
{
  iEvent.getByToken(ecalRecHitsEBToken_,ecalRecHitsEBHandle_);
  iEvent.getByToken(ecalRecHitsEEToken_,ecalRecHitsEEHandle_);
}

void EG8XObjectUpdateModifier::setEventContent(const edm::EventSetup& iSetup)
{
  iSetup.get<CaloTopologyRecord>().get(caloTopoHandle_);
}


void EG8XObjectUpdateModifier::modifyObject(reco::GsfElectron& ele)const
{

  const reco::CaloCluster& seedClus = *(ele.superCluster()->seed());
  const EcalRecHitCollection* ecalRecHits = ele.isEB() ? &*ecalRecHitsEBHandle_ : &*ecalRecHitsEEHandle_;
  const auto* caloTopo = caloTopoHandle_.product();

  auto full5x5ShowerShapes = ele.full5x5_showerShape();  
  full5x5ShowerShapes.e2x5Left   = noZS::EcalClusterTools::e2x5Left  (seedClus,ecalRecHits,caloTopo);
  full5x5ShowerShapes.e2x5Right  = noZS::EcalClusterTools::e2x5Right (seedClus,ecalRecHits,caloTopo);
  full5x5ShowerShapes.e2x5Top    = noZS::EcalClusterTools::e2x5Top   (seedClus,ecalRecHits,caloTopo);
  full5x5ShowerShapes.e2x5Bottom = noZS::EcalClusterTools::e2x5Bottom(seedClus,ecalRecHits,caloTopo);
  ele.full5x5_setShowerShape(full5x5ShowerShapes);
  
  auto showerShapes = ele.showerShape();  
  showerShapes.e2x5Left   = EcalClusterTools::e2x5Left  (seedClus,ecalRecHits,caloTopo);
  showerShapes.e2x5Right  = EcalClusterTools::e2x5Right (seedClus,ecalRecHits,caloTopo);
  showerShapes.e2x5Top    = EcalClusterTools::e2x5Top   (seedClus,ecalRecHits,caloTopo);
  showerShapes.e2x5Bottom = EcalClusterTools::e2x5Bottom(seedClus,ecalRecHits,caloTopo);
  ele.setShowerShape(showerShapes);

  reco::GsfElectron::SaturationInfo eleSatInfo;
  auto satInfo = getSaturationInfo(*ele.superCluster());
  eleSatInfo.nSaturatedXtals = satInfo.first;
  eleSatInfo.isSeedSaturated = satInfo.second;
  ele.setSaturationInfo(eleSatInfo);
}

void EG8XObjectUpdateModifier::modifyObject(reco::Photon& pho)const
{
  reco::Photon::SaturationInfo phoSatInfo;
  auto satInfo = getSaturationInfo(*pho.superCluster());
  phoSatInfo.nSaturatedXtals = satInfo.first;
  phoSatInfo.isSeedSaturated = satInfo.second;
  pho.setSaturationInfo(phoSatInfo);
}


std::pair<int,bool> EG8XObjectUpdateModifier::getSaturationInfo(const reco::SuperCluster& superClus)const
{
  bool isEB = superClus.seed()->seed().subdetId()==EcalBarrel;
  const auto& ecalRecHits = isEB ? *ecalRecHitsEBHandle_ : *ecalRecHitsEEHandle_;

  int nrSatCrys = 0;
  bool seedSaturated = false;
  const auto& hitsAndFractions = superClus.seed()->hitsAndFractions();
  for(const auto& hitFractionPair : hitsAndFractions) {    
    auto ecalRecHitIt = ecalRecHits.find(hitFractionPair.first);
    if(ecalRecHitIt != ecalRecHits.end() && 
       ecalRecHitIt->checkFlag(EcalRecHit::Flags::kSaturated)){
      nrSatCrys++;
      if(hitFractionPair.first == superClus.seed()->seed()) seedSaturated = true;
    }
  }
  return {nrSatCrys,seedSaturated};
}
  
DEFINE_EDM_PLUGIN(ModifyObjectValueFactory,
          EG8XObjectUpdateModifier,
          "EG8XObjectUpdateModifier");