RegressionHelper Class Reference

#include <RegressionHelper.h>

Classes
struct	Configuration

Public Member Functions
void	applyCombinationRegression (reco::GsfElectron &ele) const
void	applyEcalRegression (reco::GsfElectron &electron, const edm::Handle< reco::VertexCollection > &vertices, const edm::Handle< EcalRecHitCollection > &rechitsEB, const edm::Handle< EcalRecHitCollection > &rechitsEE) const
void	checkSetup (const edm::EventSetup &)
void	readEvent (const edm::Event &)
	RegressionHelper (const Configuration &)
	~RegressionHelper ()

Private Member Functions
void	getEcalRegression (const reco::SuperCluster &sc, const edm::Handle< reco::VertexCollection > &vertices, const edm::Handle< EcalRecHitCollection > &rechitsEB, const edm::Handle< EcalRecHitCollection > &rechitsEE, double &energyFactor, double &errorFactor) const

Private Attributes
const CaloGeometry *	caloGeometry_
unsigned long long	caloGeometryCacheId_
const CaloTopology *	caloTopology_
unsigned long long	caloTopologyCacheId_
const Configuration	cfg_
const GBRForest *	combinationReg_
bool	combinationRegressionInitialized_
const GBRForest *	ecalRegBarrel_
const GBRForest *	ecalRegEndcap_
const GBRForest *	ecalRegErrorBarrel_
const GBRForest *	ecalRegErrorEndcap_
bool	ecalRegressionInitialized_
unsigned long long	regressionCacheId_

Detailed Description

Definition at line 24 of file RegressionHelper.h.

Constructor & Destructor Documentation

RegressionHelper::RegressionHelper

(

const Configuration &

config

)

Definition at line 11 of file RegressionHelper.cc.

                                                               :
  cfg_(config),ecalRegressionInitialized_(false),combinationRegressionInitialized_(false),
  caloTopologyCacheId_(0), caloGeometryCacheId_(0),regressionCacheId_(0)
{;}

RegressionHelper::~RegressionHelper

(

)

Definition at line 16 of file RegressionHelper.cc.

{;}

Member Function Documentation

void RegressionHelper::applyCombinationRegression

(

reco::GsfElectron &

ele

)

const

Definition at line 165 of file RegressionHelper.cc.

References reco::GsfElectron::BADTRACK, reco::GsfElectron::BIGBREM, reco::GsfElectron::classification(), combinationReg_, reco::GsfElectron::correctedEcalEnergy(), reco::GsfElectron::correctedEcalEnergyError(), reco::GsfElectron::ecalDriven(), energyError(), reco::GsfElectron::GAP, GBRForest::GetResponse(), reco::GsfElectron::GOLDEN, reco::GsfElectron::isEB(), reco::GsfElectron::p4(), reco::GsfElectron::P4_COMBINATION, reco::GsfElectron::setP4(), reco::GsfElectron::SHOWERING, mathSSE::sqrt(), reco::GsfElectron::trackerDrivenSeed(), reco::GsfElectron::trackMomentumAtVtx(), reco::GsfElectron::trackMomentumError(), and mps_merge::weight.

Referenced by GsfElectronAlgo::createElectron().

{
  float energy = ele.correctedEcalEnergy();
  float energyError = ele.correctedEcalEnergyError();
  float momentum = ele.trackMomentumAtVtx().R();
  float momentumError = ele.trackMomentumError();
  int elClass = -1;
  
  switch (ele.classification()) {
  case reco::GsfElectron::GOLDEN:
    elClass = 0;
    break;
  case reco::GsfElectron::BIGBREM:
    elClass = 1;
    break;
  case reco::GsfElectron::BADTRACK:
    elClass = 2;
    break;
  case  reco::GsfElectron::SHOWERING:
    elClass = 3;
    break;
  case reco::GsfElectron::GAP:
    elClass = 4;
    break;
  default:
    elClass = -1;
  }


  bool isEcalDriven = ele.ecalDriven();
  bool isTrackerDriven =  ele.trackerDrivenSeed();
  bool isEB = ele.isEB();
  
  // compute relative errors and ratio of errors
  float energyRelError = energyError / energy;
  float momentumRelError = momentumError / momentum;
  float errorRatio = energyRelError / momentumRelError;

  // calculate E/p and corresponding error
  float eOverP = energy / momentum;  
  float eOverPerror = eOverP*std::hypot(energyRelError,momentumRelError);
  
  // fill input variables
  std::vector<float> regressionInputs ;
  regressionInputs.resize(11,0.);
  
  regressionInputs[0]  = energy;
  regressionInputs[1]  = energyRelError;
  regressionInputs[2]  = momentum;
  regressionInputs[3]  = momentumRelError;
  regressionInputs[4]  = errorRatio;
  regressionInputs[5]  = eOverP;
  regressionInputs[6]  = eOverPerror;
  regressionInputs[7]  = static_cast<float>(isEcalDriven);
  regressionInputs[8]  = static_cast<float>(isTrackerDriven);
  regressionInputs[9]  = static_cast<float>(elClass);
  regressionInputs[10] = static_cast<float>(isEB);
  
  // retrieve combination weight
  float weight = 0.;
  if(eOverP>0.025 
     &&fabs(momentum-energy)<15.*sqrt(momentumError*momentumError + energyError*energyError)
     ) // protection against crazy track measurement
    {
      weight = combinationReg_->GetResponse(regressionInputs.data());
      if(weight>1.) weight = 1.;
      else if(weight<0.) weight = 0.;
    }
  
  float combinedMomentum = weight*momentum + (1.-weight)*energy;
  float combinedMomentumError = sqrt(weight*weight*momentumError*momentumError + (1.-weight)*(1.-weight)*energyError*energyError);
  
  // FIXME : pure tracker electrons have track momentum error of 999.
  // If the combination try to combine such electrons then the original combined momentum is kept
  if(momentumError!=999. || weight==0.)
    {
      math::XYZTLorentzVector oldMomentum = ele.p4() ;
      math::XYZTLorentzVector newMomentum(oldMomentum.x()*combinedMomentum/oldMomentum.t(),
                      oldMomentum.y()*combinedMomentum/oldMomentum.t(),
                      oldMomentum.z()*combinedMomentum/oldMomentum.t(),
                      combinedMomentum);
      
      ele.setP4(reco::GsfElectron::P4_COMBINATION,newMomentum,combinedMomentumError,true);
    }  
}

void RegressionHelper::applyEcalRegression	(	reco::GsfElectron &	electron,
		const edm::Handle< reco::VertexCollection > &	vertices,
		const edm::Handle< EcalRecHitCollection > &	rechitsEB,
		const edm::Handle< EcalRecHitCollection > &	rechitsEE
	)		const

Definition at line 18 of file RegressionHelper.cc.

References getEcalRegression(), reco::GsfElectron::setCorrectedEcalEnergy(), reco::GsfElectron::setCorrectedEcalEnergyError(), reco::GsfElectron::superCluster(), and electrons_cff::vertices.

Referenced by GsfElectronAlgo::createElectron().

                                                                               {
  double cor, err;
  getEcalRegression(*ele.superCluster(), vertices, rechitsEB, rechitsEE, cor, err);
  ele.setCorrectedEcalEnergy(cor * ele.superCluster()->correctedEnergy());
  ele.setCorrectedEcalEnergyError( err * ele.superCluster()->correctedEnergy());    

}

void RegressionHelper::checkSetup

(

const edm::EventSetup &

es

)

Definition at line 29 of file RegressionHelper.cc.

References caloGeometry_, caloGeometryCacheId_, caloTopology_, caloTopologyCacheId_, cfg_, combinationReg_, combinationRegressionInitialized_, RegressionHelper::Configuration::combinationRegressionWeightFiles, RegressionHelper::Configuration::combinationRegressionWeightLabels, RegressionHelper::Configuration::combinationWeightsFromDB, ecalRegBarrel_, ecalRegEndcap_, ecalRegErrorBarrel_, ecalRegErrorEndcap_, ecalRegressionInitialized_, RegressionHelper::Configuration::ecalRegressionWeightFiles, RegressionHelper::Configuration::ecalRegressionWeightLabels, RegressionHelper::Configuration::ecalWeightsFromDB, edmOneToOneComparison::file1, edmOneToOneComparison::file2, edm::EventSetup::get(), edm::eventsetup::EventSetupRecordImplementation< T >::get(), and regressionCacheId_.

Referenced by GsfElectronAlgo::checkSetup().

                                                          {

  // Topology 
  unsigned long long newCaloTopologyCacheId 
    = es.get<CaloTopologyRecord>().cacheIdentifier() ;
  if (!caloTopologyCacheId_ || ( caloTopologyCacheId_ != newCaloTopologyCacheId ) )
    {
      caloTopologyCacheId_ = newCaloTopologyCacheId;
      edm::ESHandle<CaloTopology> caloTopo;
      es.get<CaloTopologyRecord>().get(caloTopo);
      caloTopology_ = &(*caloTopo);
    }
    
  // Geometry
  unsigned long long newCaloGeometryCacheId
    = es.get<CaloGeometryRecord>().cacheIdentifier() ;
  if (!caloGeometryCacheId_ || ( caloGeometryCacheId_ != newCaloGeometryCacheId ) ) 
    {
      caloGeometryCacheId_ = newCaloGeometryCacheId;
      edm::ESHandle<CaloGeometry> caloGeom;
      es.get<CaloGeometryRecord>().get(caloGeom);
      caloGeometry_ = &(*caloGeom);
    }

  // Ecal regression 
  
  // if at least one of the set of weights come from the DB
  if(cfg_.ecalWeightsFromDB)
    {
      unsigned long long newRegressionCacheId
    = es.get<GBRWrapperRcd>().cacheIdentifier() ;
      if ( !regressionCacheId_ || (newRegressionCacheId != regressionCacheId_ ) ) {
    
    const GBRWrapperRcd& gbrRcd = es.get<GBRWrapperRcd>();

    // ECAL barrel
    edm::ESHandle<GBRForest> ecalRegBarrelH;
    gbrRcd.get(cfg_.ecalRegressionWeightLabels[0].c_str(),ecalRegBarrelH);
    ecalRegBarrel_ = &(*ecalRegBarrelH);
    
    // ECAL endcaps
    edm::ESHandle<GBRForest> ecalRegEndcapH;
    gbrRcd.get(cfg_.ecalRegressionWeightLabels[1].c_str(),ecalRegEndcapH);
    ecalRegEndcap_= &(*ecalRegEndcapH);
    
    // ECAL barrel error
    edm::ESHandle<GBRForest> ecalRegErrorBarrelH;
    gbrRcd.get(cfg_.ecalRegressionWeightLabels[2].c_str(),ecalRegErrorBarrelH);
    ecalRegErrorBarrel_ = &(*ecalRegErrorBarrelH);
    
    // ECAL endcap error
    edm::ESHandle<GBRForest> ecalRegErrorEndcapH;
    gbrRcd.get(cfg_.ecalRegressionWeightLabels[3].c_str(),ecalRegErrorEndcapH);
    ecalRegErrorEndcap_ = &(*ecalRegErrorEndcapH);
    
    ecalRegressionInitialized_ = true;
      }
    }
  if( cfg_.combinationWeightsFromDB) 
    {

    // Combination
    if(cfg_.combinationWeightsFromDB) 
      {
        unsigned long long newRegressionCacheId
          = es.get<GBRWrapperRcd>().cacheIdentifier() ;
        if ( !regressionCacheId_ || (newRegressionCacheId != regressionCacheId_ ) ) {

          const GBRWrapperRcd& gbrRcd = es.get<GBRWrapperRcd>();

          edm::ESHandle<GBRForest> combinationRegH;
          gbrRcd.get(cfg_.combinationRegressionWeightLabels[0].c_str(),combinationRegH);
          combinationReg_ = &(*combinationRegH);
          
          combinationRegressionInitialized_ = true;
        }
      }
    }

  // read weights from file - for debugging. Even if it is one single files, 4 files should b set in the vector
  if(!cfg_.ecalWeightsFromDB && !ecalRegressionInitialized_ &&
     !cfg_.ecalRegressionWeightFiles.empty() ) {
    TFile file0 (edm::FileInPath(cfg_.ecalRegressionWeightFiles[0].c_str()).fullPath().c_str());
    ecalRegBarrel_ = (const GBRForest*)file0.Get(cfg_.ecalRegressionWeightLabels[0].c_str());
    file0.Close();
    TFile file1 (edm::FileInPath(cfg_.ecalRegressionWeightFiles[1].c_str()).fullPath().c_str());
    ecalRegEndcap_ = (const GBRForest*)file1.Get(cfg_.ecalRegressionWeightLabels[1].c_str());
    file1.Close();
    TFile file2 (edm::FileInPath(cfg_.ecalRegressionWeightFiles[2].c_str()).fullPath().c_str());
    ecalRegErrorBarrel_ = (const GBRForest*)file2.Get(cfg_.ecalRegressionWeightLabels[2].c_str());
    file2.Close();
    TFile file3 (edm::FileInPath(cfg_.ecalRegressionWeightFiles[3].c_str()).fullPath().c_str());
    ecalRegErrorEndcap_ = (const GBRForest*)file3.Get(cfg_.ecalRegressionWeightLabels[3].c_str());
    ecalRegressionInitialized_ = true;
    file3.Close();
  }
  
  if(!cfg_.combinationWeightsFromDB && !combinationRegressionInitialized_ && 
     !cfg_.combinationRegressionWeightFiles.empty() ) 
    {
      TFile file0 (edm::FileInPath(cfg_.combinationRegressionWeightFiles[0].c_str()).fullPath().c_str());
      combinationReg_ = (const GBRForest*)file0.Get(cfg_.combinationRegressionWeightLabels[0].c_str());
      combinationRegressionInitialized_ = true;
      file0.Close();
    }
  
}

void RegressionHelper::getEcalRegression ( const reco::SuperCluster &  sc, const edm::Handle< reco::VertexCollection > &  vertices, const edm::Handle< EcalRecHitCollection > &  rechitsEB, const edm::Handle< EcalRecHitCollection > &  rechitsEE, double &  energyFactor, double &  errorFactor  ) const

private

Definition at line 139 of file RegressionHelper.cc.

References caloGeometry_, caloTopology_, EcalBarrel, EcalEndcap, ecalRegBarrel_, ecalRegEndcap_, ecalRegErrorBarrel_, ecalRegErrorEndcap_, Exception, EcalRegressionData::fill(), EcalRegressionData::fillVec(), GBRForest::GetResponse(), edm::Handle< T >::product(), and reco::SuperCluster::seed().

Referenced by applyEcalRegression().

                                                                        {
  
  energyFactor = -999.;
  errorFactor = -999.;

  std::vector<float> rInputs;
  EcalRegressionData regData;
  regData.fill(sc,rechitsEB.product(),rechitsEE.product(),caloGeometry_,caloTopology_,vertices.product());
  regData.fillVec(rInputs);
  if(sc.seed()->hitsAndFractions()[0].first.subdetId()==EcalBarrel){
    energyFactor = ecalRegBarrel_->GetResponse(&rInputs[0]);
    errorFactor = ecalRegErrorBarrel_->GetResponse(&rInputs[0]);
  }else if(sc.seed()->hitsAndFractions()[0].first.subdetId()==EcalEndcap){  
    energyFactor = ecalRegEndcap_->GetResponse(&rInputs[0]);
    errorFactor = ecalRegErrorEndcap_->GetResponse(&rInputs[0]);
  }else{
    throw cms::Exception("RegressionHelper::calculateRegressedEnergy")
      << "Supercluster seed is either EB nor EE!" << std::endl;
  }

}

void RegressionHelper::readEvent

(

const edm::Event &

)

Definition at line 137 of file RegressionHelper.cc.

{;}

Member Data Documentation

const CaloGeometry* RegressionHelper::caloGeometry_

private

Definition at line 60 of file RegressionHelper.h.

Referenced by checkSetup(), and getEcalRegression().

unsigned long long RegressionHelper::caloGeometryCacheId_

private

Definition at line 65 of file RegressionHelper.h.

Referenced by checkSetup().

const CaloTopology* RegressionHelper::caloTopology_

private

Definition at line 59 of file RegressionHelper.h.

Referenced by checkSetup(), and getEcalRegression().

unsigned long long RegressionHelper::caloTopologyCacheId_

private

Definition at line 64 of file RegressionHelper.h.

Referenced by checkSetup().

const Configuration RegressionHelper::cfg_

private

Definition at line 58 of file RegressionHelper.h.

Referenced by checkSetup().

const GBRForest* RegressionHelper::combinationReg_

private

Definition at line 73 of file RegressionHelper.h.

Referenced by applyCombinationRegression(), and checkSetup().

bool RegressionHelper::combinationRegressionInitialized_

private

Definition at line 62 of file RegressionHelper.h.

Referenced by checkSetup().

const GBRForest* RegressionHelper::ecalRegBarrel_

private

Definition at line 69 of file RegressionHelper.h.

Referenced by checkSetup(), and getEcalRegression().

const GBRForest* RegressionHelper::ecalRegEndcap_

private

Definition at line 70 of file RegressionHelper.h.

Referenced by checkSetup(), and getEcalRegression().

const GBRForest* RegressionHelper::ecalRegErrorBarrel_

private

Definition at line 71 of file RegressionHelper.h.

Referenced by checkSetup(), and getEcalRegression().

const GBRForest* RegressionHelper::ecalRegErrorEndcap_

private

Definition at line 72 of file RegressionHelper.h.

Referenced by checkSetup(), and getEcalRegression().

bool RegressionHelper::ecalRegressionInitialized_

private

Definition at line 61 of file RegressionHelper.h.

Referenced by checkSetup().

unsigned long long RegressionHelper::regressionCacheId_

private

Definition at line 66 of file RegressionHelper.h.

Referenced by checkSetup().