GlobalNNCalibration.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2024 CERN for the benefit of the ATLAS collaboration
*/
 
 
#ifndef JetCalibTools_GlobalNNCalibration_H
#define JetCalibTools_GlobalNNCalibration_H
 
#include "TH1.h"
#include "TString.h"
#include "lwtnn/LightweightGraph.hh"
 
// Local includes
#include "JetCalibTools/JetCalibrationStep.h"
#include <string>
#include <vector>
#include <map>
#include <memory>
 
class JetEventInfo;
class TEnv;
 
 
class GlobalNNCalibration : virtual public ::JetCalibrationStep {
 
  public:
  // Constructor/destructor/init
  GlobalNNCalibration();      
 
  GlobalNNCalibration(const std::string& name);      
 
  GlobalNNCalibration(const std::string& name, TEnv * config, TString jetAlgo, const TString& calibAreaTag, bool dev);
 
  virtual ~GlobalNNCalibration() = default;
 
  virtual StatusCode initialize() override;
 
 
  protected:
  // @brief          Calibrates the jet, and decorates it with the calibration using the name "JetGNNCScaleMomentum"
  // @param jet_reco The jet
  // @param jetEventInfo A set of information about the event and jet
  virtual StatusCode calibrate(xAOD::Jet& jet, JetEventInfo&) const override;
 
  private:
    
  double getJetChargedFraction(const xAOD::Jet& jet_reco, int PVindex) const;
 
  double getJetDetEta(const xAOD::Jet& jet_reco) const;
 
  int getJetNtrk1000(const xAOD::Jet& jet_reco, int PVindex) const;
 
  double getJetWtrk1000(const xAOD::Jet& jet_reco, int PVindex) const;
 
  double getJESPt(const xAOD::Jet& jet_reco) const;
 
  std::map<std::string,double> getJetFeatures(const xAOD::Jet& jet_reco, JetEventInfo& jetEventInfo) const;
 
  double getSplineSlope(const int ieta, const double minPt) const;
 
  void loadSplineHists(const TString & fileName, const std::string &etajes_name = "etaJes");
 
  double getSplineCorr(const int etaBin, double E) const;
 
 
  int getEtaBin(const xAOD::Jet& jet_reco, const std::vector<double>& etaBins) const;
 
 
  std::vector<std::unique_ptr<lwt::LightweightGraph> > m_lwnns;
  std::vector<std::unique_ptr<TH1> > m_ptCorrFactors;
  std::vector<double> m_nnEtaBins;
  std::vector<double> m_closureEtaBins;
  std::vector<TString> m_NNInputs;
 
  
  TEnv * m_config{};
  TString m_jetAlgo;
  std::string m_calibAreaTag;
  bool m_dev{};
  bool m_doSplineCorr{true};
  bool m_doLogPtScaling{};
  double m_minNNCorrection{};
  double m_maxNNCorrection{};
 
  std::vector<double> m_JPtS_MinPt_Slopes;
  std::vector<double> m_JPtS_MinPt_Pt;
  std::vector<double> m_JPtS_MinPt_R;
 
 
}; // Class GlobalNNCalibration
 
 
#endif