SCEnergyCorrectorSemiParm.h

Go to the documentation of this file.

//--------------------------------------------------------------------------------------------------
// $Id $
//
// SCEnergyCorrectorSemiParm
//
// Helper Class for applying regression-based energy corrections with optimized BDT implementation
//
// Authors: J.Bendavid
//--------------------------------------------------------------------------------------------------

#ifndef EGAMMATOOLS_SCEnergyCorrectorSemiParm_H
#define EGAMMATOOLS_SCEnergyCorrectorSemiParm_H
    
#include "FWCore/Framework/interface/EventSetup.h"

#include "DataFormats/VertexReco/interface/VertexFwd.h"
#include "CondFormats/EgammaObjects/interface/GBRForestD.h"
#include "DataFormats/EcalRecHit/interface/EcalRecHit.h"
#include "DataFormats/EcalRecHit/interface/EcalRecHitCollections.h"
#include "Geometry/CaloEventSetup/interface/CaloTopologyRecord.h"
#include "Geometry/CaloTopology/interface/CaloTopology.h" 
#include "Geometry/Records/interface/CaloGeometryRecord.h"
#include "Geometry/CaloGeometry/interface/CaloGeometry.h"
#include "FWCore/Framework/interface/ConsumesCollector.h"
#include "DataFormats/EgammaReco/interface/SuperCluster.h"
#include "FWCore/Framework/interface/Event.h"
#include "FWCore/Framework/interface/ESHandle.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"

class SCEnergyCorrectorSemiParm {
 public:
  SCEnergyCorrectorSemiParm();
  ~SCEnergyCorrectorSemiParm(); 
  
  void setTokens(const edm::ParameterSet &iConfig, edm::ConsumesCollector &cc);

  std::pair<double,double> getCorrections(const reco::SuperCluster &sc) const;
  void modifyObject(reco::SuperCluster &sc);
  
  void setEventSetup(const edm::EventSetup &es);
  void setEvent(const edm::Event &e);
  
 protected:    
  const GBRForestD *foresteb_;
  const GBRForestD *forestee_;
  const GBRForestD *forestsigmaeb_;
  const GBRForestD *forestsigmaee_;
  
  edm::ESHandle<CaloTopology> calotopo_;
  edm::ESHandle<CaloGeometry> calogeom_;
  
  edm::EDGetTokenT<EcalRecHitCollection> tokenEBRecHits_;
  edm::EDGetTokenT<EcalRecHitCollection> tokenEERecHits_;
  edm::EDGetTokenT<reco::VertexCollection> tokenVertices_;    

  edm::Handle<reco::VertexCollection> vertices_;
  edm::Handle<EcalRecHitCollection>  rechitsEB_;
  edm::Handle<EcalRecHitCollection>  rechitsEE_;

  edm::InputTag ecalHitsEBInputTag_;
  edm::InputTag ecalHitsEEInputTag_;
  edm::InputTag vertexInputTag_;

  std::string regressionKeyEB_;
  std::string uncertaintyKeyEB_;
  std::string regressionKeyEE_;
  std::string uncertaintyKeyEE_;

 private:
  bool isHLT_;
  bool applySigmaIetaIphiBug_; //there was a bug in sigmaIetaIphi for the 74X application
  int nHitsAboveThreshold_;
  float eThreshold_;
};
#endif