Back to home page

Project CMSSW displayed by LXR

 
 

    


File indexing completed on 2021-02-14 13:34:20

0001 #ifndef RecoEgamma_EgammaTools_EleEnergyRetriever_h
0002 #define RecoEgamma_EgammaTools_EleEnergyRetriever_h
0003 
0004 #include "DataFormats/EgammaCandidates/interface/GsfElectron.h"
0005 #include "FWCore/Utilities/interface/Exception.h"
0006 
0007 class EleEnergyRetriever {
0008 public:
0009   enum class EnergyType { EcalTrk, Ecal, SuperCluster, SuperClusterRaw };
0010 
0011   EleEnergyRetriever(const std::string& typeStr) : type_(convertFromStr(typeStr)) {}
0012 
0013   static EnergyType convertFromStr(const std::string& typeStr) {
0014     if (typeStr == "EcalTrk")
0015       return EnergyType::EcalTrk;
0016     else if (typeStr == "Ecal")
0017       return EnergyType::Ecal;
0018     else if (typeStr == "SuperCluster")
0019       return EnergyType::SuperCluster;
0020     else if (typeStr == "SuperClusterRaw")
0021       return EnergyType::SuperClusterRaw;
0022     else {
0023       throw cms::Exception("ConfigError")
0024           << " type \"" << typeStr << "\" not recognised, must be of type EcalTrk,Ecal,SuperCluster,SuperClusterRaw";
0025     }
0026   }
0027 
0028   float operator()(const reco::GsfElectron& ele) const {
0029     switch (type_) {
0030       case EnergyType::EcalTrk:
0031         return ele.energy();
0032       case EnergyType::Ecal:
0033         return ele.ecalEnergy();
0034       case EnergyType::SuperCluster:
0035         return ele.superCluster()->energy();
0036       case EnergyType::SuperClusterRaw:
0037         return ele.superCluster()->rawEnergy();
0038     }
0039     return 0.;
0040   }
0041 
0042 private:
0043   EnergyType type_;
0044 };
0045 
0046 #endif