ElectronIDSelectorNeuralNet.cc

Go to the documentation of this file.

#include "RecoEgamma/ElectronIdentification/plugins/ElectronIDSelectorNeuralNet.h"

ElectronIDSelectorNeuralNet::ElectronIDSelectorNeuralNet (const edm::ParameterSet& conf) : conf_ (conf) 
{
  doNeuralNet_ = conf_.getParameter<bool> ("doNeuralNet");
  
  if (doNeuralNet_) 
    neuralNetAlgo_ = new ElectronNeuralNet();
}

ElectronIDSelectorNeuralNet::~ElectronIDSelectorNeuralNet () 
{
  if (doNeuralNet_) 
    delete neuralNetAlgo_ ;
}

void ElectronIDSelectorNeuralNet::newEvent (const edm::Event& e, const edm::EventSetup& es)
{
  if (doNeuralNet_) 
    neuralNetAlgo_->setup (conf_);
}

double ElectronIDSelectorNeuralNet::operator () (const reco::GsfElectron & ele, const edm::Event& e, const edm::EventSetup& es) 
{
  if (doNeuralNet_) 
    return static_cast<double>(neuralNetAlgo_->result (& (ele), e) );
  return 0. ;
}