MCJESInputBase.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2024 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef JETTOOLHELPERS_MCJESINPUTBASE_H
#define JETTOOLHELPERS_MCJESINPUTBASE_H
 
 
#include "TString.h"
#include "TEnv.h"
 
#include "JetAnalysisInterfaces/IVarTool.h"
#include "AsgTools/AsgTool.h"
#include "AsgTools/PropertyWrapper.h"
#include <memory>
#include <string>
class TAxis;
 
namespace JetHelper {
 
 
class MCJESInputBase :public asg::AsgTool, virtual public IVarTool
{
    ASG_TOOL_CLASS(MCJESInputBase, IVarTool)
 
    public:
        MCJESInputBase(const std::string& myname);
        virtual ~MCJESInputBase() {};
        std::string getFileName() const { return m_fileName; };
        Gaudi::Property< std::string > m_corrName { this, "corrName", "JES", "MCJES input" };
   
    private:
        // support functions to extract information from text files
        double getLogPolN(const double *factors, double x) const;
        std::vector<double> VectorizeD(const TString& str, TString sep=" ") const;
        int getEtaBin(double eta_det) const; 
        Gaudi::Property< std::string > m_fileName { this, "inputfile", "/afs/cern.ch/work/s/stapiaar/JetDev4/athena/JetToolHelpers/data/file_JES.config", "File containing histograms" };
        Gaudi::Property< std::string > m_jetAlgo { this, "jetAlgo", "AntiKt4EMPFlow", "jet collection" };
 
    protected:
        std::unique_ptr<TEnv> m_config;
        unsigned int m_nPar{}; // number of parameters in config file
        const static unsigned int s_nEtaBins = 90;
        const static unsigned int s_nParMax = 9; 
        double m_JESFactors[s_nEtaBins][s_nParMax]{};
        double m_etaCorrFactors[s_nEtaBins][s_nParMax]{};
        double m_energyFreezeJES[s_nEtaBins]{};
        TAxis * m_etaBinAxis{};
        
        bool readMCJESFromText();
        double getJES(const double X, const double Y=0, const double Emax=-1) const;
        double getEtaCorr(const double X, const double Y=0) const;
        double getEmaxJES(const double Y) const;
};
} // namespace JetHelper
#endif