ElectronLikelihood.h

Go to the documentation of this file.

#ifndef ElectronLikelihood_H
#define ElectronLikelihood_H

#include "FWCore/Framework/interface/Event.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "DataFormats/EgammaCandidates/interface/GsfElectron.h"
#include "RecoEgamma/ElectronIdentification/interface/ElectronIDAlgo.h"
#include "RecoEgamma/ElectronIdentification/interface/LikelihoodSwitches.h"
#include "RecoEgamma/ElectronIdentification/interface/LikelihoodPdfProduct.h"
#include "RecoEcal/EgammaCoreTools/interface/EcalClusterLazyTools.h"
#include "CondFormats/DataRecord/interface/ElectronLikelihoodRcd.h"
#include "CondFormats/EgammaObjects/interface/ElectronLikelihoodCalibration.h"
#include <TDirectory.h>
#include <vector>


class ElectronLikelihood {

 public:
  
  ElectronLikelihood () {} ;

  ElectronLikelihood (const ElectronLikelihoodCalibration *calibration,
              LikelihoodSwitches eleIDSwitches,
              std::string signalWeightSplitting,
              std::string backgroundWeightSplitting,
              bool splitSignalPdfs,
              bool splitBackgroundPdfs) ;

  virtual ~ElectronLikelihood () ;

  void setup (const edm::ParameterSet& conf) {} ;

  float result (const reco::GsfElectron &electron, 
                const EcalClusterLazyTools&) const ;
  float resultLog (const reco::GsfElectron &electron, 
                   const EcalClusterLazyTools&) const ;

 private:

  void Setup (const ElectronLikelihoodCalibration *calibration,
          std::string signalWeightSplitting,
          std::string backgroundWeightSplitting,
          bool splitSignalPdfs,
          bool splitBackgroundPdfs) ;


  void getInputVar (const reco::GsfElectron &electron, 
                    std::vector<float> &measuremnts, 
                    const EcalClusterLazyTools&) const ;

  //Class to enforce that we only call const functions on LikelihoodPdfProduct
  // when we are in const functions of ElectronLikelihood
  class LikelihoodPdfProductPtr {
  public:
    LikelihoodPdfProductPtr(): m_ptr(nullptr) {}
    LikelihoodPdfProductPtr( LikelihoodPdfProduct* iPtr): m_ptr(iPtr) {}

    LikelihoodPdfProduct* operator->() { return m_ptr;}
    const LikelihoodPdfProduct* operator->() const { return m_ptr;}

    LikelihoodPdfProduct* get() { return m_ptr;}

  private:
    LikelihoodPdfProduct* m_ptr;
  };

  LikelihoodPdfProductPtr _EB0lt15lh, _EB1lt15lh, _EElt15lh;
  LikelihoodPdfProductPtr _EB0gt15lh, _EB1gt15lh, _EEgt15lh;

  LikelihoodSwitches m_eleIDSwitches ;

  std::string m_signalWeightSplitting;
  std::string m_backgroundWeightSplitting;
  bool m_splitSignalPdfs;
  bool m_splitBackgroundPdfs;

};

#include "FWCore/Framework/interface/data_default_record_trait.h"
EVENTSETUP_DATA_DEFAULT_RECORD (ElectronLikelihood, ElectronLikelihoodRcd)

#endif // ElectronLikelihood_H