RegressionHelper.h

Go to the documentation of this file.

#ifndef __EgammaElectronAlgos_RegressionHelper_h__
#define __EgammaElectronAlgos_RegressionHelper_h__

#include "FWCore/Framework/interface/Event.h"
#include "FWCore/Framework/interface/EventSetup.h"
#include "FWCore/Framework/interface/ESHandle.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "FWCore/Framework/interface/ConsumesCollector.h"

#include "CondFormats/EgammaObjects/interface/GBRForest.h"
#include "Geometry/CaloTopology/interface/CaloTopology.h"
#include "Geometry/CaloGeometry/interface/CaloGeometry.h"

#include "DataFormats/Common/interface/Handle.h"
#include "DataFormats/EgammaReco/interface/SuperCluster.h"
#include "DataFormats/VertexReco/interface/VertexFwd.h"
#include "DataFormats/VertexReco/interface/Vertex.h"
#include "DataFormats/EcalRecHit/interface/EcalRecHit.h"
#include "DataFormats/EcalRecHit/interface/EcalRecHitCollections.h"
#include "DataFormats/EgammaCandidates/interface/GsfElectron.h"


class RegressionHelper {
  public:
  struct Configuration
  {
      // weight files for the regression
    std::vector<std::string> ecalRegressionWeightLabels;
    bool ecalWeightsFromDB ;
    std::vector<std::string> ecalRegressionWeightFiles;
    std::vector<std::string> combinationRegressionWeightLabels;
    bool combinationWeightsFromDB;
    std::vector<std::string> combinationRegressionWeightFiles;
  };
  
  RegressionHelper( const Configuration & ) ;
  void checkSetup( const edm::EventSetup & ) ;
  void readEvent( const edm::Event & ) ;
  void applyEcalRegression(reco::GsfElectron & electron,
               const edm::Handle<reco::VertexCollection>& vertices,
               const edm::Handle<EcalRecHitCollection>& rechitsEB,
               const edm::Handle<EcalRecHitCollection>& rechitsEE) const;

  void applyCombinationRegression(reco::GsfElectron & ele) const;
  ~RegressionHelper() ;

 private:
  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:
  
  const Configuration cfg_ ;
  const CaloTopology * caloTopology_;
  const CaloGeometry * caloGeometry_;
  bool ecalRegressionInitialized_;
  bool combinationRegressionInitialized_;

  unsigned long long caloTopologyCacheId_;
  unsigned long long caloGeometryCacheId_;
  unsigned long long regressionCacheId_;

  //  edm::ESHandle<GBRForest> ecalRegBarrel_ ;
  const GBRForest * ecalRegBarrel_ ;
  const GBRForest * ecalRegEndcap_ ;
  const GBRForest * ecalRegErrorBarrel_ ;  
  const GBRForest * ecalRegErrorEndcap_ ;  
  const GBRForest * combinationReg_ ;
   
};

#endif