Inheritance diagram for EGRegressionModifierV2:

Classes
struct	electron_config

struct	photon_config

Public Types
typedef std::pair< edm::InputTag, ValMapFloatToken >	ValMapFloatTagTokenPair

typedef edm::EDGetTokenT< edm::ValueMap< float > >	ValMapFloatToken

typedef std::pair< edm::InputTag, ValMapIntToken >	ValMapIntTagTokenPair

typedef edm::EDGetTokenT< edm::ValueMap< int > >	ValMapIntToken

Public Member Functions
	EGRegressionModifierV2 (const edm::ParameterSet &conf)

void	modifyObject (reco::GsfElectron &) const final

void	modifyObject (reco::Photon &) const final

void	modifyObject (pat::Electron &) const final

void	modifyObject (pat::Photon &) const final

void	setConsumes (edm::ConsumesCollector &) final

void	setEvent (const edm::Event &) final

void	setEventContent (const edm::EventSetup &) final

	~EGRegressionModifierV2 () 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 ()

Private Attributes
electron_config	e_conf

std::vector< const GBRForestD * >	e_forestH_mean_

std::vector< const GBRForestD * >	e_forestH_sigma_

std::unordered_map< unsigned, edm::Handle< edm::ValueMap< int > > >	ele_int_vmaps

std::unordered_map< unsigned, edm::Handle< edm::ValueMap< float > > >	ele_vmaps

std::unordered_map< unsigned, edm::Ptr< reco::GsfElectron > >	eles_by_oop

double	eOverP_ECALTRKThr_

double	epDiffSig_ECALTRKThr_

double	epSig_ECALTRKThr_

bool	forceHighEnergyEcalTrainingIfSaturated_

double	highEnergy_ECALTRKThr_

const edm::EventSetup *	iSetup_

double	lowEnergy_ECALonlyThr_

double	lowEnergy_ECALTRKThr_

photon_config	ph_conf

std::vector< const GBRForestD * >	ph_forestH_mean_

std::vector< const GBRForestD * >	ph_forestH_sigma_

std::unordered_map< unsigned, edm::Handle< edm::ValueMap< int > > >	pho_int_vmaps

std::unordered_map< unsigned, edm::Handle< edm::ValueMap< float > > >	pho_vmaps

std::unordered_map< unsigned, edm::Ptr< reco::Photon > >	phos_by_oop

edm::InputTag	rhoTag_

edm::EDGetTokenT< double >	rhoToken_

float	rhoValue_

Detailed Description

Definition at line 23 of file EGRegressionModifierV2.cc.

Member Typedef Documentation

typedef std::pair<edm::InputTag, ValMapFloatToken> EGRegressionModifierV2::ValMapFloatTagTokenPair

Definition at line 27 of file EGRegressionModifierV2.cc.

typedef edm::EDGetTokenT<edm::ValueMap<float> > EGRegressionModifierV2::ValMapFloatToken

Definition at line 25 of file EGRegressionModifierV2.cc.

typedef std::pair<edm::InputTag, ValMapIntToken> EGRegressionModifierV2::ValMapIntTagTokenPair

Definition at line 28 of file EGRegressionModifierV2.cc.

typedef edm::EDGetTokenT<edm::ValueMap<int> > EGRegressionModifierV2::ValMapIntToken

Definition at line 26 of file EGRegressionModifierV2.cc.

Constructor & Destructor Documentation

EGRegressionModifierV2::EGRegressionModifierV2 ( const edm::ParameterSet & conf )

Definition at line 102 of file EGRegressionModifierV2.cc.

                                                                           :
   ModifyObjectValueBase(conf) {
 
   lowEnergy_ECALonlyThr_ = conf.getParameter<double>("lowEnergy_ECALonlyThr");
   lowEnergy_ECALTRKThr_ = conf.getParameter<double>("lowEnergy_ECALTRKThr");
   highEnergy_ECALTRKThr_ = conf.getParameter<double>("highEnergy_ECALTRKThr");
   eOverP_ECALTRKThr_ = conf.getParameter<double>("eOverP_ECALTRKThr");
   epDiffSig_ECALTRKThr_ = conf.getParameter<double>("epDiffSig_ECALTRKThr");
   epSig_ECALTRKThr_ = conf.getParameter<double>("epSig_ECALTRKThr");
   forceHighEnergyEcalTrainingIfSaturated_ = conf.getParameter<bool>("forceHighEnergyEcalTrainingIfSaturated");
   rhoTag_ = conf.getParameter<edm::InputTag>("rhoCollection");
 
   constexpr char electronSrc[] =  "electronSrc";
   constexpr char photonSrc[] =  "photonSrc";
     
   if(conf.exists("electron_config")) {
     const edm::ParameterSet& electrons = conf.getParameter<edm::ParameterSet>("electron_config");
     if( electrons.exists(electronSrc) ) 
       e_conf.electron_src = electrons.getParameter<edm::InputTag>(electronSrc);
     
     std::vector<std::string> intValueMaps;
     if ( electrons.existsAs<std::vector<std::string> >("intValueMaps")) 
       intValueMaps = electrons.getParameter<std::vector<std::string> >("intValueMaps");
 
     const std::vector<std::string> parameters = electrons.getParameterNames();
     for( const std::string& name : parameters ) {
       if( std::string(electronSrc) == name ) 
     continue;
       if( electrons.existsAs<edm::InputTag>(name)) {
     for (auto vmp : intValueMaps) {
       if (name == vmp) {
         e_conf.tag_int_token_map[name] = ValMapIntTagTokenPair(electrons.getParameter<edm::InputTag>(name), ValMapIntToken());
         break;
       } 
     }
     e_conf.tag_float_token_map[name] = ValMapFloatTagTokenPair(electrons.getParameter<edm::InputTag>(name), ValMapFloatToken());
       }
     }
     
     e_conf.condnames_ecalonly_mean  = electrons.getParameter<std::vector<std::string> >("regressionKey_ecalonly");
     e_conf.condnames_ecalonly_sigma = electrons.getParameter<std::vector<std::string> >("uncertaintyKey_ecalonly");
     e_conf.condnames_ecaltrk_mean   = electrons.getParameter<std::vector<std::string> >("regressionKey_ecaltrk");
     e_conf.condnames_ecaltrk_sigma  = electrons.getParameter<std::vector<std::string> >("uncertaintyKey_ecaltrk");
 
     unsigned int encor = e_conf.condnames_ecalonly_mean.size();
     e_forestH_mean_.reserve(2*encor);
     e_forestH_sigma_.reserve(2*encor);
 
   }
   
   if( conf.exists("photon_config") ) { 
     const edm::ParameterSet& photons = conf.getParameter<edm::ParameterSet>("photon_config");
 
     if( photons.exists(photonSrc) ) 
       ph_conf.photon_src = photons.getParameter<edm::InputTag>(photonSrc);
 
     std::vector<std::string> intValueMaps;
     if ( photons.existsAs<std::vector<std::string> >("intValueMaps")) 
       intValueMaps = photons.getParameter<std::vector<std::string> >("intValueMaps");
 
     const std::vector<std::string> parameters = photons.getParameterNames();
     for( const std::string& name : parameters ) {
       if( std::string(photonSrc) == name ) 
     continue;
       if( photons.existsAs<edm::InputTag>(name)) {
     for (auto vmp : intValueMaps) {
       if (name == vmp) {
         ph_conf.tag_int_token_map[name] = ValMapIntTagTokenPair(photons.getParameter<edm::InputTag>(name), ValMapIntToken());
         break;
       } 
     }
     ph_conf.tag_float_token_map[name] = ValMapFloatTagTokenPair(photons.getParameter<edm::InputTag>(name), ValMapFloatToken());
       }
     }
 
     ph_conf.condnames_ecalonly_mean  = photons.getParameter<std::vector<std::string>>("regressionKey_ecalonly");
     ph_conf.condnames_ecalonly_sigma = photons.getParameter<std::vector<std::string>>("uncertaintyKey_ecalonly");
 
     unsigned int ncor = ph_conf.condnames_ecalonly_mean.size();
     ph_forestH_mean_.reserve(ncor);
     ph_forestH_sigma_.reserve(ncor);
 
   }
 
 
 }

EGRegressionModifierV2::~EGRegressionModifierV2 ( )

override

Definition at line 198 of file EGRegressionModifierV2.cc.

198 {}

Member Function Documentation

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

finalvirtual

Reimplemented from ModifyObjectValueBase.

Definition at line 356 of file EGRegressionModifierV2.cc.

Referenced by modifyObject().

                                                                     {
 
   // regression calculation needs no additional valuemaps
 
   const reco::SuperClusterRef& the_sc = ele.superCluster();
   const edm::Ptr<reco::CaloCluster>& theseed = the_sc->seed();
 
   // skip HGCAL for now
   if( theseed->seed().det() == DetId::Forward ) return;
 
   const int numberOfClusters =  the_sc->clusters().size();
   const bool missing_clusters = !the_sc->clusters()[numberOfClusters-1].isAvailable();
   if( missing_clusters ) return ; // do not apply corrections in case of missing info (slimmed MiniAOD electrons)
   
   //check if fbrem is filled as its needed for E/p combination so abort if its set to the default value 
   //this will be the case for <5 (or current cuts) for miniAOD electrons
   if(ele.fbrem()==reco::GsfElectron::ClassificationVariables().trackFbrem) return;
 
   const bool iseb = ele.isEB();  
 
   std::array<float, 32> eval;  
   const double raw_energy = the_sc->rawEnergy(); 
   const double raw_es_energy = the_sc->preshowerEnergy();
   const auto& full5x5_ess = ele.full5x5_showerShape();
 
   float e5x5Inverse = full5x5_ess.e5x5 != 0. ? vdt::fast_inv(full5x5_ess.e5x5) : 0.;
 
   eval[0]  = raw_energy;
   eval[1]  = the_sc->etaWidth();
   eval[2]  = the_sc->phiWidth(); 
   eval[3]  = the_sc->seed()->energy()/raw_energy;
   eval[4]  = full5x5_ess.e5x5/raw_energy;
   eval[5]  = ele.hcalOverEcalBc();
   eval[6]  = rhoValue_;
   eval[7]  = theseed->eta() - the_sc->position().Eta();
   eval[8]  = reco::deltaPhi( theseed->phi(),the_sc->position().Phi());
   eval[9]  = full5x5_ess.r9;
   eval[10]  = full5x5_ess.sigmaIetaIeta;
   eval[11]  = full5x5_ess.sigmaIetaIphi;
   eval[12]  = full5x5_ess.sigmaIphiIphi;
   eval[13]  = full5x5_ess.eMax*e5x5Inverse;
   eval[14]  = full5x5_ess.e2nd*e5x5Inverse;
   eval[15]  = full5x5_ess.eTop*e5x5Inverse;
   eval[16]  = full5x5_ess.eBottom*e5x5Inverse;
   eval[17]  = full5x5_ess.eLeft*e5x5Inverse;
   eval[18]  = full5x5_ess.eRight*e5x5Inverse;
   eval[19]  = full5x5_ess.e2x5Max*e5x5Inverse;
   eval[20]  = full5x5_ess.e2x5Left*e5x5Inverse;
   eval[21]  = full5x5_ess.e2x5Right*e5x5Inverse;
   eval[22]  = full5x5_ess.e2x5Top*e5x5Inverse;
   eval[23]  = full5x5_ess.e2x5Bottom*e5x5Inverse;
   eval[24]  = ele.nSaturatedXtals();
   eval[25]  = std::max(0,numberOfClusters);
       
   // calculate coordinate variables
   EcalClusterLocal _ecalLocal;
   if (iseb) {
 
     float dummy;
     int ieta;
     int iphi;
     _ecalLocal.localCoordsEB(*theseed, *iSetup_, dummy, dummy, ieta, iphi, dummy, dummy);
     eval[26] = ieta;
     eval[27] = iphi;
     int signieta = ieta > 0 ? +1 : -1;
     eval[28] = (ieta-signieta)%5;
     eval[29] = (iphi-1)%2;
     eval[30] = (abs(ieta)<=25)*((ieta-signieta)) + (abs(ieta)>25)*((ieta-26*signieta)%20);  
     eval[31] = (iphi-1)%20;
 
   } else {
 
     float dummy;
     int ix;
     int iy;
     _ecalLocal.localCoordsEE(*theseed, *iSetup_, dummy, dummy, ix, iy, dummy, dummy);
     eval[26] = ix;
     eval[27] = iy;
     eval[28] = raw_es_energy/raw_energy;
 
   }
 
   //magic numbers for MINUIT-like transformation of BDT output onto limited range
   //(These should be stored inside the conditions object in the future as well)
   constexpr double meanlimlow  = -1.0;
   constexpr double meanlimhigh = 3.0;
   constexpr double meanoffset  = meanlimlow + 0.5*(meanlimhigh-meanlimlow);
   constexpr double meanscale   = 0.5*(meanlimhigh-meanlimlow);
   
   constexpr double sigmalimlow  = 0.0002;
   constexpr double sigmalimhigh = 0.5;
   constexpr double sigmaoffset  = sigmalimlow + 0.5*(sigmalimhigh-sigmalimlow);
   constexpr double sigmascale   = 0.5*(sigmalimhigh-sigmalimlow);  
 
   
   size_t coridx = 0;
   float raw_pt = raw_energy*the_sc->position().rho()/the_sc->position().r();
   bool isSaturated = ele.nSaturatedXtals()!=0;
   
   if(raw_pt >= lowEnergy_ECALonlyThr_ || 
      (isSaturated && forceHighEnergyEcalTrainingIfSaturated_)){
     if(iseb) coridx = 1;
     else coridx = 3;
   }else{
     if(iseb) coridx = 0;
     else coridx = 2;
   }
   
   
   //these are the actual BDT responses
   double rawmean = e_forestH_mean_[coridx]->GetResponse(eval.data());
   double rawsigma = e_forestH_sigma_[coridx]->GetResponse(eval.data());
   
   //apply transformation to limited output range (matching the training)
   double mean = meanoffset + meanscale*vdt::fast_sin(rawmean);
   double sigma = sigmaoffset + sigmascale*vdt::fast_sin(rawsigma);
 
   // Correct the energy. A negative energy means that the correction went
   // outside the boundaries of the training. In this case uses raw.
   // The resolution estimation, on the other hand should be ok.
   if (mean < 0.) mean = 1.0;
 
   const double ecor = mean*(raw_energy + raw_es_energy);
   const double sigmacor = sigma*ecor;
   
   ele.setCorrectedEcalEnergy(ecor);
   ele.setCorrectedEcalEnergyError(sigmacor);
 
   double combinedEnergy = ecor;
   double combinedEnergyError = sigmacor;
 
   auto el_track = ele.gsfTrack();
   const float trkMomentum = el_track->pMode();
   const float trkEta      = el_track->etaMode();
   const float trkPhi      = el_track->phiMode();  
   const float trkMomentumError = std::abs(el_track->qoverpModeError())*trkMomentum*trkMomentum;  
 
   const float eOverP = (raw_energy+raw_es_energy)*mean/trkMomentum;
   const float fbrem = ele.fbrem();
 
   // E-p combination
   if (ecor < highEnergy_ECALTRKThr_ &&
       eOverP > eOverP_ECALTRKThr_ && 
       std::abs(ecor - trkMomentum) < epDiffSig_ECALTRKThr_*std::sqrt(trkMomentumError*trkMomentumError+sigmacor*sigmacor) && 
       trkMomentumError < epSig_ECALTRKThr_*trkMomentum) { 
 
     raw_pt = ecor/cosh(trkEta);
     if (iseb && raw_pt < lowEnergy_ECALTRKThr_)
       coridx = 4;
     else if (iseb && raw_pt >= lowEnergy_ECALTRKThr_)
       coridx = 5;
     else if (!iseb && raw_pt < lowEnergy_ECALTRKThr_)
       coridx = 6;
     else if (!iseb && raw_pt >= lowEnergy_ECALTRKThr_)
       coridx = 7;
   
     eval[0] = ecor;
     eval[1] = sigma/mean;
     eval[2] = trkMomentumError/trkMomentum;
     eval[3] = eOverP;
     eval[4] = ele.ecalDrivenSeed();
     eval[5] = full5x5_ess.r9;
     eval[6] = fbrem;
     eval[7] = trkEta; 
     eval[8] = trkPhi; 
 
     float ecalEnergyVar = (raw_energy + raw_es_energy)*sigma; 
     float rawcombNormalization = (trkMomentumError*trkMomentumError + ecalEnergyVar*ecalEnergyVar);
     float rawcomb = ( ecor*trkMomentumError*trkMomentumError + trkMomentum*ecalEnergyVar*ecalEnergyVar ) / rawcombNormalization;
 
     //these are the actual BDT responses
     double rawmean_trk = e_forestH_mean_[coridx]->GetResponse(eval.data());
     double rawsigma_trk = e_forestH_sigma_[coridx]->GetResponse(eval.data());
     
     //apply transformation to limited output range (matching the training)
     double mean_trk = meanoffset + meanscale*vdt::fast_sin(rawmean_trk);
     double sigma_trk = sigmaoffset + sigmascale*vdt::fast_sin(rawsigma_trk);
     
     // Final correction
     // A negative energy means that the correction went
     // outside the boundaries of the training. In this case uses raw.
     // The resolution estimation, on the other hand should be ok.
     if (mean_trk < 0.) mean_trk = 1.0;
 
     combinedEnergy = mean_trk*rawcomb;
     combinedEnergyError = sigma_trk*rawcomb;
   }
 
   math::XYZTLorentzVector oldFourMomentum = ele.p4();
   math::XYZTLorentzVector newFourMomentum = math::XYZTLorentzVector(oldFourMomentum.x()*combinedEnergy/oldFourMomentum.t(),
                                     oldFourMomentum.y()*combinedEnergy/oldFourMomentum.t(),
                                     oldFourMomentum.z()*combinedEnergy/oldFourMomentum.t(),
                                     combinedEnergy);
 
   ele.correctMomentum(newFourMomentum, ele.trackMomentumError(), combinedEnergyError);
 }

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

finalvirtual

Reimplemented from ModifyObjectValueBase.

Definition at line 557 of file EGRegressionModifierV2.cc.

References funct::abs(), constexpr, reco::deltaPhi(), DetId::det(), reco::Photon::ShowerShape::e5x5, reco::CaloCluster::eta(), forceHighEnergyEcalTrainingIfSaturated_, DetId::Forward, reco::Photon::full5x5_r9(), reco::Photon::full5x5_showerShapeVariables(), reco::Photon::hadronicOverEm(), edm::Ref< C, T, F >::isAvailable(), reco::Photon::isEB(), iSetup_, CastorSimpleRecAlgoImpl::isSaturated(), EcalClusterLocal::localCoordsEB(), EcalClusterLocal::localCoordsEE(), lowEnergy_ECALonlyThr_, SiStripPI::max, SiStripPI::mean, reco::Photon::nSaturatedXtals(), ph_forestH_mean_, ph_forestH_sigma_, reco::CaloCluster::phi(), mathSSE::return(), rhoValue_, reco::CaloCluster::seed(), reco::Photon::setCorrectedEnergy(), and reco::Photon::superCluster().

                                                                {
   // regression calculation needs no additional valuemaps
   
   const reco::SuperClusterRef& the_sc = pho.superCluster();
   const edm::Ptr<reco::CaloCluster>& theseed = the_sc->seed();  
 
   // skip HGCAL for now
   if( theseed->seed().det() == DetId::Forward ) return;
 
   const int numberOfClusters =  the_sc->clusters().size();
   const bool missing_clusters = !the_sc->clusters()[numberOfClusters-1].isAvailable();
   if( missing_clusters ) return ; // do not apply corrections in case of missing info (slimmed MiniAOD electrons)
 
   const bool iseb = pho.isEB();  
   
   std::array<float, 32> eval;  
   const double raw_energy = the_sc->rawEnergy(); 
   const double raw_es_energy = the_sc->preshowerEnergy();
   const auto& full5x5_pss = pho.full5x5_showerShapeVariables();
 
   float e5x5Inverse = full5x5_pss.e5x5 != 0. ? vdt::fast_inv(full5x5_pss.e5x5) : 0.;
 
   eval[0]  = raw_energy;
   eval[1]  = the_sc->etaWidth();
   eval[2]  = the_sc->phiWidth(); 
   eval[3]  = the_sc->seed()->energy()/raw_energy;
   eval[4]  = full5x5_pss.e5x5/raw_energy;
   eval[5]  = pho.hadronicOverEm();
   eval[6]  = rhoValue_;
   eval[7]  = theseed->eta() - the_sc->position().Eta();
   eval[8]  = reco::deltaPhi( theseed->phi(),the_sc->position().Phi());
   eval[9]  = pho.full5x5_r9();
   eval[10]  = full5x5_pss.sigmaIetaIeta;
   eval[11]  = full5x5_pss.sigmaIetaIphi;
   eval[12]  = full5x5_pss.sigmaIphiIphi;
   eval[13]  = full5x5_pss.maxEnergyXtal*e5x5Inverse;
   eval[14]  = full5x5_pss.e2nd*e5x5Inverse;
   eval[15]  = full5x5_pss.eTop*e5x5Inverse;
   eval[16]  = full5x5_pss.eBottom*e5x5Inverse;
   eval[17]  = full5x5_pss.eLeft*e5x5Inverse;
   eval[18]  = full5x5_pss.eRight*e5x5Inverse;
   eval[19]  = full5x5_pss.e2x5Max*e5x5Inverse;
   eval[20]  = full5x5_pss.e2x5Left*e5x5Inverse;
   eval[21]  = full5x5_pss.e2x5Right*e5x5Inverse;
   eval[22]  = full5x5_pss.e2x5Top*e5x5Inverse;
   eval[23]  = full5x5_pss.e2x5Bottom*e5x5Inverse;
   eval[24]  = pho.nSaturatedXtals();
   eval[25]  = std::max(0,numberOfClusters);
       
   // calculate coordinate variables
   EcalClusterLocal _ecalLocal;
 
   if (iseb) {
 
     float dummy;
     int ieta;
     int iphi;
     _ecalLocal.localCoordsEB(*theseed, *iSetup_, dummy, dummy, ieta, iphi, dummy, dummy);
     eval[26] = ieta;
     eval[27] = iphi;
     int signieta = ieta > 0 ? +1 : -1;
     eval[28] = (ieta-signieta)%5;
     eval[29] = (iphi-1)%2;
     eval[30] = (abs(ieta)<=25)*((ieta-signieta)) + (abs(ieta)>25)*((ieta-26*signieta)%20);  
     eval[31] = (iphi-1)%20;
 
   } else {
 
     float dummy;
     int ix;
     int iy;
     _ecalLocal.localCoordsEE(*theseed, *iSetup_, dummy, dummy, ix, iy, dummy, dummy);
     eval[26] = ix;
     eval[27] = iy;
     eval[28] = raw_es_energy/raw_energy;
 
   }
 
   //magic numbers for MINUIT-like transformation of BDT output onto limited range
   //(These should be stored inside the conditions object in the future as well)
   constexpr double meanlimlow  = -1.0;
   constexpr double meanlimhigh = 3.0;
   constexpr double meanoffset  = meanlimlow + 0.5*(meanlimhigh-meanlimlow);
   constexpr double meanscale   = 0.5*(meanlimhigh-meanlimlow);
   
   constexpr double sigmalimlow  = 0.0002;
   constexpr double sigmalimhigh = 0.5;
   constexpr double sigmaoffset  = sigmalimlow + 0.5*(sigmalimhigh-sigmalimlow);
   constexpr double sigmascale   = 0.5*(sigmalimhigh-sigmalimlow);  
   
   size_t coridx = 0;
   float raw_pt = raw_energy*the_sc->position().rho()/the_sc->position().r();
   bool isSaturated = pho.nSaturatedXtals();
   
   if(raw_pt >= lowEnergy_ECALonlyThr_ || 
      (isSaturated && forceHighEnergyEcalTrainingIfSaturated_)){
     if(iseb) coridx = 1;
     else coridx = 3;
   }else{
     if(iseb) coridx = 0;
     else coridx = 2;
   }
   
   //these are the actual BDT responses
   double rawmean = ph_forestH_mean_[coridx]->GetResponse(eval.data());
   double rawsigma = ph_forestH_sigma_[coridx]->GetResponse(eval.data());
   
   //apply transformation to limited output range (matching the training)
   double mean = meanoffset + meanscale*vdt::fast_sin(rawmean);
   double sigma = sigmaoffset + sigmascale*vdt::fast_sin(rawsigma);
   
   // Correct the energy. A negative energy means that the correction went
   // outside the boundaries of the training. In this case uses raw.
   // The resolution estimation, on the other hand should be ok.
   if (mean < 0.) mean = 1.0;
 
   const double ecor = mean*(raw_energy + raw_es_energy);
   const double sigmacor = sigma*ecor;
   
   pho.setCorrectedEnergy(reco::Photon::P4type::regression2, ecor, sigmacor, true);     
 }

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

finalvirtual

Reimplemented from ModifyObjectValueBase.

Definition at line 553 of file EGRegressionModifierV2.cc.

References modifyObject().

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

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

finalvirtual

Reimplemented from ModifyObjectValueBase.

Definition at line 679 of file EGRegressionModifierV2.cc.

References modifyObject().

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

void EGRegressionModifierV2::setConsumes ( edm::ConsumesCollector & sumes )

finalvirtual

Reimplemented from ModifyObjectValueBase.

Definition at line 312 of file EGRegressionModifierV2.cc.

                                                                     {
  
   rhoToken_ = sumes.consumes<double>(rhoTag_);
 
   //setup electrons
   if(!(empty_tag == e_conf.electron_src))
     e_conf.tok_electron_src = sumes.consumes<edm::View<pat::Electron> >(e_conf.electron_src);  
 
   for ( std::unordered_map<std::string, ValMapFloatTagTokenPair>::iterator imap = e_conf.tag_float_token_map.begin(); 
     imap != e_conf.tag_float_token_map.end(); 
     imap++) {
     make_consumes(imap->second.first, imap->second.second, sumes);
   }  
 
   for ( std::unordered_map<std::string, ValMapIntTagTokenPair>::iterator imap = e_conf.tag_int_token_map.begin(); 
     imap != e_conf.tag_int_token_map.end(); 
     imap++) {
     make_int_consumes(imap->second.first, imap->second.second, sumes);
   }  
   
   // setup photons 
   if(!(empty_tag == ph_conf.photon_src)) 
     ph_conf.tok_photon_src = sumes.consumes<edm::View<pat::Photon> >(ph_conf.photon_src);
 
   for ( std::unordered_map<std::string, ValMapFloatTagTokenPair>::iterator imap = ph_conf.tag_float_token_map.begin(); 
     imap != ph_conf.tag_float_token_map.end(); 
     imap++) {
     make_consumes(imap->second.first, imap->second.second, sumes);
   }  
 
   for ( std::unordered_map<std::string, ValMapIntTagTokenPair>::iterator imap = ph_conf.tag_int_token_map.begin(); 
     imap != ph_conf.tag_int_token_map.end(); 
     imap++) {
     make_int_consumes(imap->second.first, imap->second.second, sumes);
   }  
 }

void EGRegressionModifierV2::setEvent ( const edm::Event & evt )

finalvirtual

Reimplemented from ModifyObjectValueBase.

Definition at line 200 of file EGRegressionModifierV2.cc.

References e_conf, ele_int_vmaps, ele_vmaps, eles_by_oop, edm::Event::getByToken(), mps_fire::i, edm::EDGetTokenT< T >::isUninitialized(), edm::Ptr< T >::key(), pat::PATObject< ObjectType >::originalObjectRef(), ph_conf, pho_int_vmaps, pho_vmaps, phos_by_oop, rhoToken_, rhoValue_, EGRegressionModifierV2::electron_config::tag_float_token_map, EGRegressionModifierV2::photon_config::tag_float_token_map, EGRegressionModifierV2::electron_config::tag_int_token_map, EGRegressionModifierV2::photon_config::tag_int_token_map, EGRegressionModifierV2::electron_config::tok_electron_src, and EGRegressionModifierV2::photon_config::tok_photon_src.

                                                          {
 
   eles_by_oop.clear();
   phos_by_oop.clear();  
   ele_vmaps.clear();
   ele_int_vmaps.clear();
   pho_vmaps.clear();
   pho_int_vmaps.clear();
   
   if( !e_conf.tok_electron_src.isUninitialized() ) {
     edm::Handle<edm::View<pat::Electron> > eles;
     evt.getByToken(e_conf.tok_electron_src, eles);
     
     for( unsigned i = 0; i < eles->size(); ++i ) {
       edm::Ptr<pat::Electron> ptr = eles->ptrAt(i);
       eles_by_oop[ptr->originalObjectRef().key()] = ptr;
     }    
   }
 
   for (std::unordered_map<std::string, ValMapFloatTagTokenPair>::iterator imap = e_conf.tag_float_token_map.begin(); 
        imap != e_conf.tag_float_token_map.end(); 
        imap++) {
     get_product(evt, imap->second.second, ele_vmaps);
   }
 
   for (std::unordered_map<std::string, ValMapIntTagTokenPair>::iterator imap = e_conf.tag_int_token_map.begin(); 
        imap != e_conf.tag_int_token_map.end(); 
        imap++) {
     get_product(evt, imap->second.second, ele_int_vmaps);
   }
   
   if( !ph_conf.tok_photon_src.isUninitialized() ) {
     edm::Handle<edm::View<pat::Photon> > phos;
     evt.getByToken(ph_conf.tok_photon_src,phos);
   
     for( unsigned i = 0; i < phos->size(); ++i ) {
       edm::Ptr<pat::Photon> ptr = phos->ptrAt(i);
       phos_by_oop[ptr->originalObjectRef().key()] = ptr;
     }
   }
    
 
   for (std::unordered_map<std::string, ValMapFloatTagTokenPair>::iterator imap = ph_conf.tag_float_token_map.begin(); 
        imap != ph_conf.tag_float_token_map.end(); 
        imap++) {
     get_product(evt, imap->second.second, pho_vmaps);
   }
 
   for (std::unordered_map<std::string, ValMapIntTagTokenPair>::iterator imap = ph_conf.tag_int_token_map.begin(); 
        imap != ph_conf.tag_int_token_map.end(); 
        imap++) {
     get_product(evt, imap->second.second, pho_int_vmaps);
   }
   
   edm::Handle<double> rhoH;
   evt.getByToken(rhoToken_, rhoH);
   rhoValue_ = *rhoH;
    
 }

void EGRegressionModifierV2::setEventContent ( const edm::EventSetup & evs )

finalvirtual

Reimplemented from ModifyObjectValueBase.

Definition at line 260 of file EGRegressionModifierV2.cc.

References EGRegressionModifierV2::electron_config::condnames_ecalonly_mean, EGRegressionModifierV2::photon_config::condnames_ecalonly_mean, EGRegressionModifierV2::electron_config::condnames_ecalonly_sigma, EGRegressionModifierV2::photon_config::condnames_ecalonly_sigma, EGRegressionModifierV2::electron_config::condnames_ecaltrk_mean, EGRegressionModifierV2::electron_config::condnames_ecaltrk_sigma, e_conf, e_forestH_mean_, e_forestH_sigma_, edm::EventSetup::get(), iSetup_, ph_conf, ph_forestH_mean_, ph_forestH_sigma_, edm::ESHandle< T >::product(), GlobalPosition_Frontier_DevDB_cff::tag, and mitigatedMETSequence_cff::U.

                                                                      {
 
   iSetup_ = &evs;
 
   edm::ESHandle<GBRForestD> forestDEH;
   
   const std::vector<std::string> ph_condnames_ecalonly_mean  = ph_conf.condnames_ecalonly_mean;
   const std::vector<std::string> ph_condnames_ecalonly_sigma = ph_conf.condnames_ecalonly_sigma;
 
   unsigned int ncor = ph_condnames_ecalonly_mean.size();
   for (unsigned int icor=0; icor<ncor; ++icor) {
     evs.get<GBRDWrapperRcd>().get(ph_condnames_ecalonly_mean[icor], forestDEH);
     ph_forestH_mean_[icor] = forestDEH.product();
     evs.get<GBRDWrapperRcd>().get(ph_condnames_ecalonly_sigma[icor], forestDEH);
     ph_forestH_sigma_[icor] = forestDEH.product();
   }
 
   const std::vector<std::string> e_condnames_ecalonly_mean  = e_conf.condnames_ecalonly_mean;
   const std::vector<std::string> e_condnames_ecalonly_sigma = e_conf.condnames_ecalonly_sigma;
   const std::vector<std::string> e_condnames_ecaltrk_mean  = e_conf.condnames_ecaltrk_mean;
   const std::vector<std::string> e_condnames_ecaltrk_sigma = e_conf.condnames_ecaltrk_sigma;
 
   unsigned int encor = e_condnames_ecalonly_mean.size();  
   for (unsigned int icor=0; icor<encor; ++icor) {
     evs.get<GBRDWrapperRcd>().get(e_condnames_ecalonly_mean[icor], forestDEH);
     e_forestH_mean_[icor] = forestDEH.product();
     evs.get<GBRDWrapperRcd>().get(e_condnames_ecalonly_sigma[icor], forestDEH);
     e_forestH_sigma_[icor] = forestDEH.product();
   } 
   for (unsigned int icor=0; icor<encor; ++icor) {
     evs.get<GBRDWrapperRcd>().get(e_condnames_ecaltrk_mean[icor], forestDEH);
     e_forestH_mean_[icor+encor] = forestDEH.product();
     evs.get<GBRDWrapperRcd>().get(e_condnames_ecaltrk_sigma[icor], forestDEH);
     e_forestH_sigma_[icor+encor] = forestDEH.product();
   }
     
 }

Member Data Documentation

electron_config EGRegressionModifierV2::e_conf

private

Definition at line 67 of file EGRegressionModifierV2.cc.

Referenced by setConsumes(), setEvent(), and setEventContent().

std::vector<const GBRForestD*> EGRegressionModifierV2::e_forestH_mean_

private

Definition at line 84 of file EGRegressionModifierV2.cc.

Referenced by modifyObject(), and setEventContent().

std::vector<const GBRForestD*> EGRegressionModifierV2::e_forestH_sigma_

private

Definition at line 85 of file EGRegressionModifierV2.cc.

Referenced by modifyObject(), and setEventContent().

std::unordered_map<unsigned,edm::Handle<edm::ValueMap<int> > > EGRegressionModifierV2::ele_int_vmaps

private

Definition at line 71 of file EGRegressionModifierV2.cc.

Referenced by setEvent().

std::unordered_map<unsigned,edm::Handle<edm::ValueMap<float> > > EGRegressionModifierV2::ele_vmaps

private

Definition at line 70 of file EGRegressionModifierV2.cc.

Referenced by setEvent().

std::unordered_map<unsigned,edm::Ptr<reco::GsfElectron> > EGRegressionModifierV2::eles_by_oop

private

Definition at line 69 of file EGRegressionModifierV2.cc.

Referenced by setEvent().

double EGRegressionModifierV2::eOverP_ECALTRKThr_

private

Definition at line 90 of file EGRegressionModifierV2.cc.

Referenced by modifyObject().

double EGRegressionModifierV2::epDiffSig_ECALTRKThr_

private

Definition at line 91 of file EGRegressionModifierV2.cc.

Referenced by modifyObject().

double EGRegressionModifierV2::epSig_ECALTRKThr_

private

Definition at line 92 of file EGRegressionModifierV2.cc.

Referenced by modifyObject().

bool EGRegressionModifierV2::forceHighEnergyEcalTrainingIfSaturated_

private

Definition at line 93 of file EGRegressionModifierV2.cc.

Referenced by modifyObject().

double EGRegressionModifierV2::highEnergy_ECALTRKThr_

private

Definition at line 89 of file EGRegressionModifierV2.cc.

Referenced by modifyObject().

const edm::EventSetup* EGRegressionModifierV2::iSetup_

private

Definition at line 80 of file EGRegressionModifierV2.cc.

Referenced by modifyObject(), and setEventContent().

double EGRegressionModifierV2::lowEnergy_ECALonlyThr_

private

Definition at line 87 of file EGRegressionModifierV2.cc.

Referenced by modifyObject().

double EGRegressionModifierV2::lowEnergy_ECALTRKThr_

private

Definition at line 88 of file EGRegressionModifierV2.cc.

Referenced by modifyObject().

photon_config EGRegressionModifierV2::ph_conf

private

Definition at line 68 of file EGRegressionModifierV2.cc.

Referenced by setConsumes(), setEvent(), and setEventContent().

std::vector<const GBRForestD*> EGRegressionModifierV2::ph_forestH_mean_

private

Definition at line 82 of file EGRegressionModifierV2.cc.

Referenced by modifyObject(), and setEventContent().

std::vector<const GBRForestD*> EGRegressionModifierV2::ph_forestH_sigma_

private

Definition at line 83 of file EGRegressionModifierV2.cc.

Referenced by modifyObject(), and setEventContent().

std::unordered_map<unsigned,edm::Handle<edm::ValueMap<int> > > EGRegressionModifierV2::pho_int_vmaps

private

Definition at line 74 of file EGRegressionModifierV2.cc.

Referenced by setEvent().

std::unordered_map<unsigned,edm::Handle<edm::ValueMap<float> > > EGRegressionModifierV2::pho_vmaps

private

Definition at line 73 of file EGRegressionModifierV2.cc.

Referenced by setEvent().

std::unordered_map<unsigned,edm::Ptr<reco::Photon> > EGRegressionModifierV2::phos_by_oop

private

Definition at line 72 of file EGRegressionModifierV2.cc.

Referenced by setEvent().

edm::InputTag EGRegressionModifierV2::rhoTag_

private

Definition at line 77 of file EGRegressionModifierV2.cc.

Referenced by setConsumes().

edm::EDGetTokenT<double> EGRegressionModifierV2::rhoToken_

private

Definition at line 78 of file EGRegressionModifierV2.cc.

Referenced by setConsumes(), and setEvent().

float EGRegressionModifierV2::rhoValue_

private

Definition at line 76 of file EGRegressionModifierV2.cc.

Referenced by modifyObject(), and setEvent().

Classes

Public Types

Public Member Functions

Private Attributes

Detailed Description

Member Typedef Documentation

Constructor & Destructor Documentation

Member Function Documentation

Member Data Documentation