ElectronIDSelectorNeuralNet.h

Go to the documentation of this file.

#ifndef ElectronIDSelectorNeuralNet_h
#define ElectronIDSelectorNeuralNet_h

#include "FWCore/Framework/interface/Event.h"
#include "FWCore/Framework/interface/EventSetup.h"
#include "FWCore/Framework/interface/ConsumesCollector.h"
#include "DataFormats/Common/interface/Handle.h"
#include "FWCore/Framework/interface/ESHandle.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "DataFormats/EgammaCandidates/interface/GsfElectron.h"
#include "RecoEgamma/ElectronIdentification/interface/ElectronNeuralNet.h"

class ElectronIDSelectorNeuralNet
{
 public:

  explicit ElectronIDSelectorNeuralNet (const edm::ParameterSet& conf, edm::ConsumesCollector && iC) :
    ElectronIDSelectorNeuralNet(conf, iC) {}
  explicit ElectronIDSelectorNeuralNet (const edm::ParameterSet& conf, edm::ConsumesCollector & iC) ;
  virtual ~ElectronIDSelectorNeuralNet () ;

  void newEvent (const edm::Event&, const edm::EventSetup&) ;
  double operator() (const reco::GsfElectron&, const edm::Event&, const edm::EventSetup&) ;

 private:

  ElectronNeuralNet* neuralNetAlgo_;

  edm::ParameterSet conf_;

  bool doNeuralNet_;

};

#endif