GlobalLargeRDNNCalibration.h

Go to the documentation of this file.

/*
   Copyright (C) 2002-2024 CERN for the benefit of the ATLAS collaboration
*/
 
/* ***********************************************************************************\
 *                                                                                   *
 *      Name: GlobalLargeRDNNCalibration                                             *
 *      Purpose: Perform the DNN JES and JMS step of the large-R jets' calibration   *
 *                                                                                   *
 *  #   Date    Comments                   By                                        *
 * -- -------- -------------------------- ------------------------------------------ *
 *  1 31/01/23  First Version              G. Albouy, P.-A. Delsart                  * 
\*************************************************************************************/
 
 
#ifndef JetCalibTools_GlobalLargeRDNNCalibration_H
#define JetCalibTools_GlobalLargeRDNNCalibration_H
 
 
// Other packages includes
#include "AthOnnxInterfaces/IOnnxRuntimeSvc.h" //Ort::Session
 
// Local includes
#include "JetCalibTools/JetCalibrationStep.h"
 
#include <string>
#include <vector>
 
class EventInfo;
class TEnv;
 
class GlobalLargeRDNNCalibration : virtual public ::JetCalibrationStep {
  
public:
  // Constructor/destructor/init
    GlobalLargeRDNNCalibration();      
 
    GlobalLargeRDNNCalibration(const std::string& name);      
 
    GlobalLargeRDNNCalibration(const std::string& name, TEnv * config, const TString& calibArea, bool dev);
 
    virtual ~GlobalLargeRDNNCalibration() ;
 
    virtual StatusCode initialize() override;
  
 
    struct VarRetriever;
 
    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:
 
    std::vector<float> getJetFeatures( xAOD::Jet& jet_reco, JetEventInfo& jetEventInfo) const;
 
    std::vector<TString> m_NNInputs;
    std::vector<double> m_eScales;
    std::vector<double> m_NormOffsets;
    std::vector<double> m_NormScales;
    std::string m_modelFileName;
 
    std::vector<VarRetriever*> m_varretrievers;
  
    std::unique_ptr< Ort::Session > m_session;
    std::vector<int64_t> m_input_node_dims;
    std::vector<const char*> m_input_node_names;
    std::vector<int64_t> m_output_node_dims;
    std::vector<const char*> m_output_node_names;
    
    TEnv * m_config{};
    std::string m_calibArea;
    bool m_devMode{};
 
 
}; // Class GlobalLargeRDNNCalibration
 
 
#endif