EgammaRegressionContainer.h

Go to the documentation of this file.

#ifndef RecoEgamma_EgammaTools_EgammaRegressionContainer_h
#define RecoEgamma_EgammaTools_EgammaRegressionContainer_h

//author: Sam Harper (RAL)
//description:
//  egamma energy regressions are binned in barrel/endcap and pt
//  this simply contains the regression for each of (currently) 4 bins
//  as well as the parameters to convert the raw BDT output back to
//  the physical real value
//  currently e/gamma also can optionally force saturated electrons
//  to always be in the high et training

#include "RecoEgamma/EgammaTools/interface/EgammaBDTOutputTransformer.h"

#include <string>

namespace edm {
  class ParameterSet;
  class ParameterSetDescription;
  class EventSetup;
}  // namespace edm
class GBRForestD;

class EgammaRegressionContainer {
public:
  EgammaRegressionContainer(const edm::ParameterSet& iConfig);
  ~EgammaRegressionContainer() {}

  static edm::ParameterSetDescription makePSetDescription();

  void setEventContent(const edm::EventSetup& iSetup);

  float operator()(const float et, const bool isEB, const bool isSaturated, const float* data) const;

  bool useLowEtBin(const float et, const bool isSaturated) const;

private:
  const EgammaBDTOutputTransformer outputTransformerLowEt_;
  const EgammaBDTOutputTransformer outputTransformerHighEt_;

  bool forceHighEnergyTrainingIfSaturated_;
  const float lowEtHighEtBoundary_;
  const std::string ebLowEtForestName_;
  const std::string ebHighEtForestName_;
  const std::string eeLowEtForestName_;
  const std::string eeHighEtForestName_;

  const GBRForestD* ebLowEtForest_;   //not owned
  const GBRForestD* ebHighEtForest_;  //not owned
  const GBRForestD* eeLowEtForest_;   //not owned
  const GBRForestD* eeHighEtForest_;  //not owned
};

#endif