egammaAODFixes.h

Go to the documentation of this file.

/*
   Copyright (C) 2002-2025 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef EGAMMAALGS_EGAMMAAODFIXES_H
#define EGAMMAALGS_EGAMMAAODFIXES_H
 
#include "AthenaBaseComps/AthReentrantAlgorithm.h"
 
#include "StoreGate/ReadHandleKey.h"
#include "StoreGate/WriteHandleKey.h"
 
//#include "xAODEgamma/EgammaContainer.h"
#include "xAODEgamma/ElectronContainer.h"
#include "xAODEgamma/PhotonContainer.h"
#include "xAODCaloEvent/CaloClusterContainer.h"
#include "xAODEgamma/EgammaFwd.h"
#include <memory>
 
#include "CaloDetDescr/CaloDetDescrManager.h"
#include "IsolationCorrections/IIsolationCorrectionTool.h"
#include "EgammaAnalysisInterfaces/IegammaMVASvc.h"
#include "egammaInterfaces/IegammaCellRecoveryTool.h"
#include "egammaInterfaces/IegammaSwTool.h"
#include "CaloEvent/CaloCellContainer.h"
#include "CaloEvent/CaloCell.h"

class egammaAODFixes : public AthReentrantAlgorithm
{
public:
 
    egammaAODFixes(const std::string& name, ISvcLocator* pSvcLocator);
 
    StatusCode initialize() override final;
    StatusCode execute(const EventContext& ctx) const override final;
 
private:
 
    // Read/Write handlers

    SG::WriteHandleKey<xAOD::ElectronContainer> m_electronOutputKey {this,
        "ElectronOutputName", "Electrons",
        "Name of electron container to be created"};

    SG::ReadHandleKey<xAOD::ElectronContainer> m_electronInputKey {this,
        "ElectronInputName", "old_Electrons",
        "Name of electron container to be read in"};
    
    SG::WriteHandleKey<xAOD::PhotonContainer> m_photonOutputKey {this,
        "PhotonOutputName", "Photons",
        "Name of photon container to be created"};

    SG::ReadHandleKey<xAOD::PhotonContainer> m_photonInputKey {this,
        "PhotonInputName", "old_Photons",
        "Name of photon container to be read in"};

    SG::ReadHandleKey<CaloCellContainer> m_CaloCellsKey {this,
    "CaloCellsName", "AllCalo", "Name of calo cell container"};

    SG::WriteHandleKey<xAOD::CaloClusterContainer> m_egClusOutputKey {this,
        "EGammaClustersOutputName", "",
        "Name of egamma cluster container to be created"};

    SG::ReadHandleKey<xAOD::CaloClusterContainer> m_egClusInputKey {this,
        "EGammaClustersInputName", "",
        "Name of egamma cluster container to be read in"};

    SG::WriteHandleKey<CaloClusterCellLinkContainer>
     m_egClusCellLinkOutputKey{this,
       "DoNotSet_OutputClusterContainerLinks",
       "",
       "Key of the output cluster container cell links; Do not set! Name taken "
       "from associated container"
     };

    SG::ReadCondHandleKey<CaloDetDescrManager> m_caloDetDescrMgrKey {this,
    "CaloDetDescrManager",
    "",
    "SG Key for CaloDetDescrManager in the Condition Store"
    };

    ToolHandle<IegammaCellRecoveryTool> m_egammaCellRecoveryTool{
      this,
    "egammaCellRecoveryTool",
    "egammaCellRecoveryTool/egammaCellRecoveryTool",
    "Tool that adds cells in L2 or L3 "
    "that could have been rejected by timing cut"
    };

    Gaudi::Property<bool> m_correctCluster {this,
    "CorrectCluster", false,
    "Whether or not to correct the cluster"};

    Gaudi::Property<bool> m_ambiguityFix {this,
    "FixAmbiguityLinks", true,
    "Whether or not to fix the ambiguity links"};

    Gaudi::Property<bool> m_tpetcFix {this,
    "FixEGTopoetcone", true,
    "Whether or not to fix topoetcone variables"};

    ToolHandle<IegammaSwTool> m_clusterCorrectionTool{ this,
    "ClusterCorrectionTool",
    "",
    "tool that applies cluster corrections" };

    ServiceHandle<IegammaMVASvc> m_MVACalibSvc{ this,
    "MVACalibSvc",
    "",
    "calibration service" };

    ToolHandle<CP::IIsolationCorrectionTool> m_IsoLeakCorrectionTool {this,
    "IsoLeakCorrectionTool", "",
    "Handle on the leakage correction tool"};
 
    void getLayerE(std::vector<const CaloCell*>& cells,
           std::vector<float>& lE) const;
    StatusCode getCorrection(const xAOD::Egamma* eg,
                 float &aET,
                 std::vector<float>& lECl) const;
    StatusCode getCorrectionC(const xAOD::Egamma* eg,
                 float &aET,
                 IegammaCellRecoveryTool::Info &info) const;
 
    void rebuildLink(xAOD::EgammaContainer *egC,
             xAOD::CaloClusterContainer *egclC,
             const EventContext& ctx) const;
};
 
#endif