TauWPDecorator.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2024 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef TAURECTOOLS_TAUWPDECORATOR_H
#define TAURECTOOLS_TAUWPDECORATOR_H
 
#include "tauRecTools/TauRecToolBase.h"
 
#include "xAODTau/TauDefs.h"
#include "xAODEventInfo/EventInfo.h"
#include "AsgDataHandles/ReadDecorHandleKey.h"
 
#include <utility>
#include <memory>
#include <vector>
#include <map>
 
class TH2;
 
class TauWPDecorator : public TauRecToolBase {
  public:
    
    ASG_TOOL_CLASS2(TauWPDecorator, TauRecToolBase, ITauToolBase)
    
    
    TauWPDecorator(const std::string& name="TauWPDecorator");
    
    ~TauWPDecorator();
 
    virtual StatusCode initialize() override;
 
    virtual StatusCode execute(xAOD::TauJet& tau) const override;
    
  private:
 
    StatusCode retrieveHistos(int nProng);
 
    StatusCode storeLimits(int nProng);
 
    double transformScore(double score, double cutLow, double effLow, double cutHigh, double effHigh) const;
 
    bool m_useAbsEta; 
    bool m_defineWPs; 
    
    std::string m_scoreName; 
    std::string m_scoreNameTrans; 
 
    std::string m_file0p; 
    std::string m_file1p; 
    std::string m_file2p; 
    std::string m_file3p; 
 
    std::vector<int> m_EDMWPs; 
    std::vector<float> m_EDMWPEffs0p; 
    std::vector<float> m_EDMWPEffs1p; 
    std::vector<float> m_EDMWPEffs2p; 
    std::vector<float> m_EDMWPEffs3p; 
 
    std::vector<std::string> m_decorWPs; 
    std::vector<SG::Accessor<char>> m_charDecors; 
    std::vector<float> m_decorWPEffs0p; 
    std::vector<float> m_decorWPEffs1p; 
    std::vector<float> m_decorWPEffs2p; 
    std::vector<float> m_decorWPEffs3p; 
    
    SG::ReadDecorHandleKey<xAOD::EventInfo> m_aveIntPerXKey {this, 
        "averageInteractionsPerCrossingKey", 
        "EventInfo.averageInteractionsPerCrossing",
        "Decoration for Average Interaction Per Crossing"};
    
    typedef std::pair<double, std::shared_ptr<TH2> > m_pair_t;
 
    std::shared_ptr<std::vector<m_pair_t>> m_hists0p; 
    std::shared_ptr<std::vector<m_pair_t>> m_hists1p; 
    std::shared_ptr<std::vector<m_pair_t>> m_hists2p; 
    std::shared_ptr<std::vector<m_pair_t>> m_hists3p; 
    
    std::map<int, double> m_xMin; 
    std::map<int, double> m_yMin; 
    std::map<int, double> m_xMax; 
    std::map<int, double> m_yMax; 
};
 
#endif // TAURECTOOLS_TAUWPDECORATOR_H