TauWPDecorator.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2025 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 "AsgTools/PropertyWrapper.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;
 
    // properties 
    Gaudi::Property<bool> m_useAbsEta{this, "UseAbsEta", false};
    Gaudi::Property<bool> m_defineWPs{this, "DefineWPs", false};
    Gaudi::Property<std::string> m_scoreName{this, "ScoreName", ""};
    Gaudi::Property<std::string> m_scoreNameTrans{this, "NewScoreName", ""}; 
    Gaudi::Property<std::string> m_file0p{this, "flatteningFile0Prong", ""};
    Gaudi::Property<std::string> m_file1p{this, "flatteningFile1Prong", ""};
    Gaudi::Property<std::string> m_file2p{this, "flatteningFile2Prong", ""};
    Gaudi::Property<std::string> m_file3p{this, "flatteningFile3Prong", ""}; 
    Gaudi::Property<std::vector<int>> m_EDMWPs{this, "CutEnumVals", {}};
    Gaudi::Property<std::vector<float>> m_EDMWPEffs0p{this, "SigEff0P", {}};
    Gaudi::Property<std::vector<float>> m_EDMWPEffs1p{this, "SigEff1P", {}};
    Gaudi::Property<std::vector<float>> m_EDMWPEffs2p{this, "SigEff2P", {}};
    Gaudi::Property<std::vector<float>> m_EDMWPEffs3p{this, "SigEff3P", {}};
    Gaudi::Property<std::vector<std::string>> m_decorWPs{this, "DecorWPNames", {}};
    Gaudi::Property<std::vector<float>> m_decorWPEffs0p{this, "DecorWPCutEffs0P", {}};
    Gaudi::Property<std::vector<float>> m_decorWPEffs1p{this, "DecorWPCutEffs1P", {}};
    Gaudi::Property<std::vector<float>> m_decorWPEffs2p{this, "DecorWPCutEffs2P", {}};
    Gaudi::Property<std::vector<float>> m_decorWPEffs3p{this, "DecorWPCutEffs3P", {}};
 
    std::vector<SG::Accessor<char>> m_charDecors; 
    
    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