EG9X105XObjectUpdateModifier Class Reference

Inheritance diagram for EG9X105XObjectUpdateModifier:

Classes
class	TokenHandlePair

Public Member Functions
	EG9X105XObjectUpdateModifier (const edm::ParameterSet &conf, edm::ConsumesCollector &cc)
void	modifyObject (reco::GsfElectron &ele) const final
void	modifyObject (reco::Photon &pho) const final
void	modifyObject (pat::Electron &ele) const final
void	modifyObject (pat::Photon &pho) const final
void	setEvent (const edm::Event &) final
void	setEventContent (const edm::EventSetup &) final
	~EG9X105XObjectUpdateModifier () override
Public Member Functions inherited from ModifyObjectValueBase
virtual void	modifyObject (reco::Muon &) const
virtual void	modifyObject (reco::BaseTau &) const
virtual void	modifyObject (reco::Jet &) const
virtual void	modifyObject (pat::Muon &) const
virtual void	modifyObject (pat::Tau &) const
virtual void	modifyObject (pat::Jet &) const
	ModifyObjectValueBase (const edm::ParameterSet &conf)
const std::string &	name () const
virtual	~ModifyObjectValueBase ()

Static Private Member Functions
template<typename ObjType >
static edm::Ptr< ObjType >	getPtrForValueMap (const ObjType &obj, const edm::Handle< edm::View< ObjType > > &objsVMIsKeyedTo)

Private Attributes
bool	allowGsfTrkMatchForConvs_
TokenHandlePair< reco::BeamSpot >	beamspot_
TokenHandlePair< reco::ConversionCollection >	conversions_
TokenHandlePair< EcalRecHitCollection >	ecalRecHitsEB_
TokenHandlePair< EcalRecHitCollection >	ecalRecHitsEE_
TokenHandlePair< edm::View< reco::GsfElectron > >	eleCollVMsAreKeyedTo_
TokenHandlePair< edm::ValueMap< float > >	eleTrkIso04_
TokenHandlePair< edm::ValueMap< float > >	eleTrkIso_
TokenHandlePair< edm::ValueMap< float > >	phoChargedHadIso_
TokenHandlePair< edm::ValueMap< float > >	phoChargedHadPFPVIso_
TokenHandlePair< edm::ValueMap< float > >	phoChargedHadWorstVtxConeVetoIso_
TokenHandlePair< edm::ValueMap< float > >	phoChargedHadWorstVtxIso_
TokenHandlePair< edm::View< reco::Photon > >	phoCollVMsAreKeyedTo_
TokenHandlePair< edm::ValueMap< float > >	phoNeutralHadIso_
TokenHandlePair< edm::ValueMap< float > >	phoPhotonIso_
bool	updateChargedHadPFPVIso_

Detailed Description

Definition at line 24 of file EG9X105XObjectUpdateModifier.cc.

Constructor & Destructor Documentation

EG9X105XObjectUpdateModifier::EG9X105XObjectUpdateModifier	(	const edm::ParameterSet &	conf,
		edm::ConsumesCollector &	cc
	)

Definition at line 89 of file EG9X105XObjectUpdateModifier.cc.

                                                                                                               :
  ModifyObjectValueBase(conf),
  eleCollVMsAreKeyedTo_(conf,"eleCollVMsAreKeyedTo",cc),
  phoCollVMsAreKeyedTo_(conf,"phoCollVMsAreKeyedTo",cc),
  conversions_(conf,"conversions",cc),
  beamspot_(conf,"beamspot",cc),
  ecalRecHitsEB_(conf,"ecalRecHitsEB",cc),
  ecalRecHitsEE_(conf,"ecalRecHitsEE",cc),
  eleTrkIso_(conf,"eleTrkIso",cc),
  eleTrkIso04_(conf,"eleTrkIso04",cc),
  phoPhotonIso_(conf,"phoPhotonIso",cc),
  phoNeutralHadIso_(conf,"phoNeutralHadIso",cc),
  phoChargedHadIso_(conf,"phoChargedHadIso",cc),
  phoChargedHadWorstVtxIso_(conf,"phoChargedHadWorstVtxIso",cc),
  phoChargedHadWorstVtxConeVetoIso_(conf,"phoChargedHadWorstVtxConeVetoIso",cc),
  phoChargedHadPFPVIso_(conf,"phoChargedHadPFPVIso",cc),
  allowGsfTrkMatchForConvs_(conf.getParameter<bool>("allowGsfTrackForConvs")),
  updateChargedHadPFPVIso_(conf.getParameter<bool>("updateChargedHadPFPVIso"))
{
  

}

EG9X105XObjectUpdateModifier::~EG9X105XObjectUpdateModifier ( )

inlineoverride

Definition at line 41 of file EG9X105XObjectUpdateModifier.cc.

References modifyObject(), setEvent(), and setEventContent().

{}

Member Function Documentation

template<typename ObjType >

edm::Ptr< ObjType > EG9X105XObjectUpdateModifier::getPtrForValueMap ( const ObjType &  obj, const edm::Handle< edm::View< ObjType > > &  objsVMIsKeyedTo  )

staticprivate

Definition at line 196 of file EG9X105XObjectUpdateModifier.cc.

References DEFINE_EDM_PLUGIN.

Referenced by modifyObject().

{
  for(auto& objVMPtr : objsVMIsKeyedTo->ptrs()){
    if(obj.superCluster()->seed()->seed()==objVMPtr->superCluster()->seed()->seed()) return objVMPtr;
  }
  return edm::Ptr<ObjType>(objsVMIsKeyedTo.id()); //return null ptr if not found
}

void EG9X105XObjectUpdateModifier::modifyObject ( reco::GsfElectron &  ele ) const

finalvirtual

Reimplemented from ModifyObjectValueBase.

Definition at line 135 of file EG9X105XObjectUpdateModifier.cc.

References allowGsfTrkMatchForConvs_, beamspot_, reco::GsfElectron::conversionRejectionVariables(), conversions_, reco::GsfElectron::core(), reco::GsfElectron::dr03IsolationVariables(), reco::GsfElectron::dr04IsolationVariables(), MillePedeFileConverter_cfg::e, eleCollVMsAreKeyedTo_, eleTrkIso04_, eleTrkIso_, reco::LeafCandidate::et(), reco::LeafCandidate::eta(), Exception, getPtrForValueMap(), ConversionTools::getVtxFitProb(), EG9X105XObjectUpdateModifier::TokenHandlePair< T >::handle(), edm::Ptr< T >::isNull(), ConversionTools::matchedConversion(), reco::BeamSpot::position(), reco::GsfElectron::setConversionRejectionVariables(), reco::GsfElectron::setDr03Isolation(), reco::GsfElectron::setDr04Isolation(), reco::GsfElectron::superCluster(), reco::GsfElectron::IsolationVariables::tkSumPtHEEP, and reco::GsfElectron::ConversionRejection::vtxFitProb.

Referenced by ~EG9X105XObjectUpdateModifier().

{
  edm::Ptr<reco::GsfElectron> ptrForVM = getPtrForValueMap(ele,eleCollVMsAreKeyedTo_.handle());
  if(ptrForVM.isNull()){
    throw cms::Exception("LogicError") <<" in EG9X105ObjectUpdateModifier, line "<<__LINE__<<" electron "<<ele.et()<<" "<<ele.eta()<<" "<<ele.superCluster()->seed()->seed().rawId()<<" failed to match to the electrons the key map was keyed to, check the map collection is correct";
  }
  reco::GsfElectron::ConversionRejection convRejVars = ele.conversionRejectionVariables();
  if(allowGsfTrkMatchForConvs_){
    convRejVars.vtxFitProb = ConversionTools::getVtxFitProb(ConversionTools::matchedConversion(ele.core(),*conversions_.handle(),beamspot_.handle()->position()));
  }else{
    //its rather important to use the core function here to get the org trk ref
    convRejVars.vtxFitProb = ConversionTools::getVtxFitProb(ConversionTools::matchedConversion(ele.core()->ctfTrack(),*conversions_.handle(),beamspot_.handle()->position(),2.0,1e-6,0));
  }
  ele.setConversionRejectionVariables(convRejVars);
  
  reco::GsfElectron::IsolationVariables isolVars03 = ele.dr03IsolationVariables();
  isolVars03.tkSumPtHEEP = (*eleTrkIso_.handle())[ptrForVM];
  ele.setDr03Isolation(isolVars03);
  reco::GsfElectron::IsolationVariables isolVars04 = ele.dr04IsolationVariables();
  isolVars04.tkSumPtHEEP = (*eleTrkIso04_.handle())[ptrForVM];
  ele.setDr04Isolation(isolVars04);
}

void EG9X105XObjectUpdateModifier::modifyObject ( reco::Photon &  pho ) const

finalvirtual

Reimplemented from ModifyObjectValueBase.

Definition at line 158 of file EG9X105XObjectUpdateModifier.cc.

References Cluster2ndMoments::alpha, reco::Photon::PflowIsolationVariables::chargedHadronIso, reco::Photon::PflowIsolationVariables::chargedHadronPFPVIso, reco::Photon::PflowIsolationVariables::chargedHadronWorstVtxGeomVetoIso, reco::Photon::PflowIsolationVariables::chargedHadronWorstVtxIso, EcalClusterToolsT< noZS >::cluster2ndMoments(), EcalBarrel, ecalRecHitsEB_, ecalRecHitsEE_, reco::LeafCandidate::et(), reco::LeafCandidate::eta(), Exception, reco::Photon::full5x5_setShowerShapeVariables(), reco::Photon::full5x5_showerShapeVariables(), reco::Photon::getPflowIsolationVariables(), getPtrForValueMap(), EG9X105XObjectUpdateModifier::TokenHandlePair< T >::handle(), edm::Ptr< T >::isNull(), reco::Photon::PflowIsolationVariables::neutralHadronIso, phoChargedHadIso_, phoChargedHadPFPVIso_, phoChargedHadWorstVtxConeVetoIso_, phoChargedHadWorstVtxIso_, phoCollVMsAreKeyedTo_, phoNeutralHadIso_, phoPhotonIso_, reco::Photon::PflowIsolationVariables::photonIso, reco::Photon::setPflowIsolationVariables(), reco::Photon::setShowerShapeVariables(), reco::Photon::showerShapeVariables(), Cluster2ndMoments::sMaj, reco::Photon::ShowerShape::smAlpha, Cluster2ndMoments::sMin, reco::Photon::ShowerShape::smMajor, reco::Photon::ShowerShape::smMinor, reco::Photon::superCluster(), and updateChargedHadPFPVIso_.

{
  edm::Ptr<reco::Photon> ptrForVM = getPtrForValueMap(pho,phoCollVMsAreKeyedTo_.handle());
  if(ptrForVM.isNull()){
    throw cms::Exception("LogicError") <<" in EG9X105ObjectUpdateModifier, line "<<__LINE__<<" photon "<<pho.et()<<" "<<pho.eta()<<" "<<pho.superCluster()->seed()->seed().rawId()<<" failed to match to the photons the key map was keyed to, check the map collection is correct";
  }
  
  reco::Photon::PflowIsolationVariables pfIso = pho.getPflowIsolationVariables();
  pfIso.photonIso = (*phoPhotonIso_.handle())[ptrForVM];
  pfIso.neutralHadronIso = (*phoNeutralHadIso_.handle())[ptrForVM];
  pfIso.chargedHadronIso = (*phoChargedHadIso_.handle())[ptrForVM];
  pfIso.chargedHadronWorstVtxIso = (*phoChargedHadWorstVtxIso_.handle())[ptrForVM];
  pfIso.chargedHadronWorstVtxGeomVetoIso = (*phoChargedHadWorstVtxConeVetoIso_.handle())[ptrForVM];
  if(updateChargedHadPFPVIso_){
    pfIso.chargedHadronPFPVIso = (*phoChargedHadPFPVIso_.handle())[ptrForVM];
  }
  pho.setPflowIsolationVariables(pfIso);

  reco::Photon::ShowerShape fracSS = pho.showerShapeVariables();
  reco::Photon::ShowerShape fullSS = pho.full5x5_showerShapeVariables();

  const reco::CaloClusterPtr seedClus = pho.superCluster()->seed();
  const bool isEB = seedClus->seed().subdetId()==EcalBarrel;
  const auto& recHits = isEB ? *ecalRecHitsEB_.handle() : *ecalRecHitsEE_.handle();
  Cluster2ndMoments clus2ndMomFrac = EcalClusterTools::cluster2ndMoments(*seedClus,recHits);
  Cluster2ndMoments clus2ndMomFull = noZS::EcalClusterTools::cluster2ndMoments(*seedClus,recHits);
  fracSS.smMajor = clus2ndMomFrac.sMaj;
  fracSS.smMinor = clus2ndMomFrac.sMin;
  fracSS.smAlpha = clus2ndMomFrac.alpha;
  fullSS.smMajor = clus2ndMomFull.sMaj;
  fullSS.smMinor = clus2ndMomFull.sMin;
  fullSS.smAlpha = clus2ndMomFull.alpha;
  pho.setShowerShapeVariables(fracSS);
  pho.full5x5_setShowerShapeVariables(fullSS);

}

void EG9X105XObjectUpdateModifier::modifyObject ( pat::Electron &  ele ) const

inlinefinalvirtual

Reimplemented from ModifyObjectValueBase.

Definition at line 49 of file EG9X105XObjectUpdateModifier.cc.

References modifyObject().

Referenced by modifyObject().

{return modifyObject(static_cast<reco::GsfElectron&>(ele));}

void EG9X105XObjectUpdateModifier::modifyObject ( pat::Photon &  pho ) const

inlinefinalvirtual

Reimplemented from ModifyObjectValueBase.

Definition at line 50 of file EG9X105XObjectUpdateModifier.cc.

References getPtrForValueMap(), modifyObject(), and hgcalPlots::obj.

Referenced by modifyObject().

{return modifyObject(static_cast<reco::Photon&>(pho));}

void EG9X105XObjectUpdateModifier::setEvent ( const edm::Event &  iEvent )

finalvirtual

Reimplemented from ModifyObjectValueBase.

Definition at line 112 of file EG9X105XObjectUpdateModifier.cc.

References beamspot_, conversions_, ecalRecHitsEB_, ecalRecHitsEE_, eleCollVMsAreKeyedTo_, eleTrkIso04_, eleTrkIso_, phoChargedHadIso_, phoChargedHadPFPVIso_, phoChargedHadWorstVtxConeVetoIso_, phoChargedHadWorstVtxIso_, phoCollVMsAreKeyedTo_, phoNeutralHadIso_, phoPhotonIso_, EG9X105XObjectUpdateModifier::TokenHandlePair< T >::setHandle(), and updateChargedHadPFPVIso_.

Referenced by ~EG9X105XObjectUpdateModifier().

{
  eleCollVMsAreKeyedTo_.setHandle(iEvent);
  phoCollVMsAreKeyedTo_.setHandle(iEvent);
  conversions_.setHandle(iEvent);
  beamspot_.setHandle(iEvent);
  ecalRecHitsEB_.setHandle(iEvent);
  ecalRecHitsEE_.setHandle(iEvent);
  eleTrkIso_.setHandle(iEvent);
  eleTrkIso04_.setHandle(iEvent);
  phoPhotonIso_.setHandle(iEvent);
  phoNeutralHadIso_.setHandle(iEvent);
  phoChargedHadIso_.setHandle(iEvent);
  phoChargedHadWorstVtxIso_.setHandle(iEvent);
  phoChargedHadWorstVtxConeVetoIso_.setHandle(iEvent);
  if(updateChargedHadPFPVIso_) phoChargedHadPFPVIso_.setHandle(iEvent);
}

void EG9X105XObjectUpdateModifier::setEventContent ( const edm::EventSetup &  iSetup )

finalvirtual

Reimplemented from ModifyObjectValueBase.

Definition at line 130 of file EG9X105XObjectUpdateModifier.cc.

Referenced by ~EG9X105XObjectUpdateModifier().

{

}

Member Data Documentation

bool EG9X105XObjectUpdateModifier::allowGsfTrkMatchForConvs_

private

Definition at line 79 of file EG9X105XObjectUpdateModifier.cc.

Referenced by modifyObject().

TokenHandlePair<reco::BeamSpot> EG9X105XObjectUpdateModifier::beamspot_

private

Definition at line 61 of file EG9X105XObjectUpdateModifier.cc.

Referenced by modifyObject(), and setEvent().

TokenHandlePair<reco::ConversionCollection> EG9X105XObjectUpdateModifier::conversions_

private

Definition at line 60 of file EG9X105XObjectUpdateModifier.cc.

Referenced by modifyObject(), and setEvent().

TokenHandlePair<EcalRecHitCollection> EG9X105XObjectUpdateModifier::ecalRecHitsEB_

private

Definition at line 62 of file EG9X105XObjectUpdateModifier.cc.

Referenced by modifyObject(), and setEvent().

TokenHandlePair<EcalRecHitCollection> EG9X105XObjectUpdateModifier::ecalRecHitsEE_

private

Definition at line 63 of file EG9X105XObjectUpdateModifier.cc.

Referenced by modifyObject(), and setEvent().

TokenHandlePair<edm::View<reco::GsfElectron> > EG9X105XObjectUpdateModifier::eleCollVMsAreKeyedTo_

private

Definition at line 57 of file EG9X105XObjectUpdateModifier.cc.

Referenced by modifyObject(), and setEvent().

TokenHandlePair<edm::ValueMap<float> > EG9X105XObjectUpdateModifier::eleTrkIso04_

private

Definition at line 66 of file EG9X105XObjectUpdateModifier.cc.

Referenced by modifyObject(), and setEvent().

TokenHandlePair<edm::ValueMap<float> > EG9X105XObjectUpdateModifier::eleTrkIso_

private

Definition at line 65 of file EG9X105XObjectUpdateModifier.cc.

Referenced by modifyObject(), and setEvent().

TokenHandlePair<edm::ValueMap<float> > EG9X105XObjectUpdateModifier::phoChargedHadIso_

private

Definition at line 69 of file EG9X105XObjectUpdateModifier.cc.

Referenced by modifyObject(), and setEvent().

TokenHandlePair<edm::ValueMap<float> > EG9X105XObjectUpdateModifier::phoChargedHadPFPVIso_

private

Definition at line 72 of file EG9X105XObjectUpdateModifier.cc.

Referenced by modifyObject(), and setEvent().

TokenHandlePair<edm::ValueMap<float> > EG9X105XObjectUpdateModifier::phoChargedHadWorstVtxConeVetoIso_

private

Definition at line 71 of file EG9X105XObjectUpdateModifier.cc.

Referenced by modifyObject(), and setEvent().

TokenHandlePair<edm::ValueMap<float> > EG9X105XObjectUpdateModifier::phoChargedHadWorstVtxIso_

private

Definition at line 70 of file EG9X105XObjectUpdateModifier.cc.

Referenced by modifyObject(), and setEvent().

TokenHandlePair<edm::View<reco::Photon> > EG9X105XObjectUpdateModifier::phoCollVMsAreKeyedTo_

private

Definition at line 58 of file EG9X105XObjectUpdateModifier.cc.

Referenced by modifyObject(), and setEvent().

TokenHandlePair<edm::ValueMap<float> > EG9X105XObjectUpdateModifier::phoNeutralHadIso_

private

Definition at line 68 of file EG9X105XObjectUpdateModifier.cc.

Referenced by modifyObject(), and setEvent().

TokenHandlePair<edm::ValueMap<float> > EG9X105XObjectUpdateModifier::phoPhotonIso_

private

Definition at line 67 of file EG9X105XObjectUpdateModifier.cc.

Referenced by modifyObject(), and setEvent().

bool EG9X105XObjectUpdateModifier::updateChargedHadPFPVIso_

private

Definition at line 85 of file EG9X105XObjectUpdateModifier.cc.

Referenced by modifyObject(), and setEvent().