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 "AsgTools/PropertyWrapper.h"
#include "xAODTau/TauJetContainer.h"
#include "xAODTau/TauDefs.h"
#include "xAODEventInfo/EventInfo.h"
#include "AsgDataHandles/ReadDecorHandleKey.h"
#include "AsgDataHandles/WriteDecorHandleKeyArray.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;
 
    Gaudi::Property<bool> m_useAbsEta{this, "UseAbsEta", false, "Whether we are flatterning electron veto WP"};
    Gaudi::Property<bool> m_defineWPs{this, "DefineWPs", false, "Whether to decorate the WPs"};
    Gaudi::Property<std::string> m_scoreName{this, "ScoreName", "", "Name of the original score"};
    Gaudi::Property<std::string> m_scoreNameTrans{this, "NewScoreName", "", "Name of the transformed score"};  
    Gaudi::Property<std::string> m_file0p{this, "flatteningFile0Prong", "", "Calibration file name of 0-prong taus"};
    Gaudi::Property<std::string> m_file1p{this, "flatteningFile1Prong", "", "Calibration file name of 1-prong taus"};
    Gaudi::Property<std::string> m_file2p{this, "flatteningFile2Prong", "", "Calibration file name of 2-prong taus"};
    Gaudi::Property<std::string> m_file3p{this, "flatteningFile3Prong", "", "Calibration file name of 3-prong taus"}; 
    Gaudi::Property<std::vector<int>> m_EDMWPs{this, "CutEnumVals", {}, "Vector of WPs in the EDM"};
    Gaudi::Property<std::vector<float>> m_EDMWPEffs0p{this, "SigEff0P", {}, "Efficiency of each WP in EDM for 0-prong taus"};
    Gaudi::Property<std::vector<float>> m_EDMWPEffs1p{this, "SigEff1P", {}, "Efficiency of each WP in EDM for 1-prong taus"};
    Gaudi::Property<std::vector<float>> m_EDMWPEffs2p{this, "SigEff2P", {}, "Efficiency of each WP in EDM for 2-prong taus"};
    Gaudi::Property<std::vector<float>> m_EDMWPEffs3p{this, "SigEff3P", {}, "Efficiency of each WP in EDM for 3-prong taus"}; 
    Gaudi::Property<std::vector<std::string>> m_decorWPs{this, "DecorWPNames", {}, "Name of WPs"};
    Gaudi::Property<std::vector<float>> m_decorWPEffs0p{this, "DecorWPCutEffs0P", {}, "Efficiency of each WP to be docorated for 0-prong taus"};
    Gaudi::Property<std::vector<float>> m_decorWPEffs1p{this, "DecorWPCutEffs1P", {}, "Efficiency of each WP to be docorated for 1-prong taus"};
    Gaudi::Property<std::vector<float>> m_decorWPEffs2p{this, "DecorWPCutEffs2P", {}, "Efficiency of each WP to be docorated for 2-prong taus"};
    Gaudi::Property<std::vector<float>> m_decorWPEffs3p{this, "DecorWPCutEffs3P", {}, "Efficiency of each WP to be docorated for 3-prong taus"};        
    // for WPs not implemented in the EDM (i.e. not encoded in IsTauFlag)
    // use Accessors unless necessary
    std::vector<SG::AuxElement::Accessor<char>> m_charDecors;
    // when data handles are required (currently in tau trigger offline monitoring), need to use Write(Read)DecorHandleKeys
    // redundant with above accessors, will be improved in the future but has implications for DAOD workflow
    Gaudi::Property<std::string> m_tauContainerName{this, "TauContainerName", "", "Name of TauJetContainer, must be set when using "};
    SG::WriteDecorHandleKeyArray<xAOD::TauJetContainer> m_decorHandleKeys {this, "DecorHandleKeys",{},"Name of WPs to be decorated"};
   
    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