RegressionEnergyPatElectronProducer.h

Go to the documentation of this file.

#ifndef RegressionEnergyPatElectronProducer_h
#define RegressionEnergyPatElectronProducer_h

#include "DataFormats/Common/interface/Handle.h"
#include "FWCore/Framework/interface/EDProducer.h"
#include "FWCore/Framework/interface/Event.h"
#include "FWCore/Framework/interface/EventSetup.h"

#include "Geometry/CaloTopology/interface/CaloSubdetectorTopology.h"
#include "Geometry/CaloGeometry/interface/CaloGeometry.h"

#include "EgammaAnalysis/ElectronTools/interface/ElectronEnergyRegressionEvaluate.h"
//#include "EGamma/EGammaAnalysisTools/interface/ElectronEnergyRegressionEvaluate.h"

#include "DataFormats/PatCandidates/interface/Electron.h"
#include "DataFormats/VertexReco/interface/VertexFwd.h"
#include "DataFormats/VertexReco/interface/Vertex.h"

class RegressionEnergyPatElectronProducer: public edm::EDProducer
{
 public:

  explicit RegressionEnergyPatElectronProducer( const edm::ParameterSet & ) ;
  ~RegressionEnergyPatElectronProducer() override;
  void produce( edm::Event &, const edm::EventSetup & ) override ;

 private:

  // input collections
  edm::EDGetTokenT<reco::GsfElectronCollection> inputGsfElectronsToken_ ;
  edm::EDGetTokenT<pat::ElectronCollection> inputPatElectronsToken_ ;
  edm::EDGetTokenT<double> rhoInputToken_ ;
  edm::EDGetTokenT<reco::VertexCollection> verticesInputToken_ ;
  edm::EDGetTokenT<EcalRecHitCollection> recHitCollectionEBToken_;
  edm::EDGetTokenT<EcalRecHitCollection> recHitCollectionEEToken_;

  //
  bool useReducedRecHits_;

  //output collection
  //  std::string outputCollection_;
  std::string nameEnergyReg_;
  std::string nameEnergyErrorReg_;

  uint32_t energyRegressionType_ ;
  uint32_t inputCollectionType_ ;
  std::string regressionInputFile_;
  bool debug_ ;
  ElectronEnergyRegressionEvaluate *regressionEvaluator_;
  bool geomInitialized_;
  bool producePatElectrons_;
  bool produceValueMaps_;


  const CaloTopology * ecalTopology_;
  const CaloGeometry * caloGeometry_;
  unsigned nElectrons_;
} ;

#endif