xAODEgammaBuilder.h

Go to the documentation of this file.

/*
   Copyright (C) 2002-2024 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef EGAMMAALGS_XAODEGAMMABUILDER_H
#define EGAMMAALGS_XAODEGAMMABUILDER_H
 
#include "AthenaBaseComps/AthReentrantAlgorithm.h"
#include "GaudiKernel/ToolHandle.h"
 
#include "StoreGate/ReadHandleKey.h"
#include "StoreGate/WriteHandleKey.h"
#include "StoreGate/ReadCondHandleKey.h"
 
#include "CaloDetDescr/CaloDetDescrManager.h"
 
#include "xAODEgamma/ElectronFwd.h"
#include "xAODEgamma/PhotonFwd.h"
#include "xAODEgamma/ElectronContainer.h"
#include "xAODEgamma/PhotonContainer.h"
#include "xAODCaloEvent/CaloClusterContainer.h"
 
#include "EgammaAnalysisInterfaces/IEGammaAmbiguityTool.h"
#include "egammaInterfaces/IEMClusterTool.h"
#include "egammaInterfaces/IEMShowerBuilder.h"
#include "egammaInterfaces/IegammaBaseTool.h"
#include "egammaInterfaces/IegammaOQFlagsBuilder.h"
#include "egammaRecEvent/egammaRecContainer.h"
#include "egammaUtils/electronPearShapeAlignmentCorrection.h"
 
#include <memory>
 
class egammaRec;
 
class xAODEgammaBuilder : public AthReentrantAlgorithm
{
public:
 
    xAODEgammaBuilder(const std::string& name, ISvcLocator* pSvcLocator);
 
    StatusCode initialize() override final;
    StatusCode finalize() override final;
    StatusCode execute(const EventContext& ctx) const override final;
 
private:
    bool getElectron(const egammaRec* egRec, xAOD::ElectronContainer *electronContainer,
            const unsigned int author, const uint8_t type) const;

    bool getPhoton(const egammaRec* egRec, xAOD::PhotonContainer *photonContainer,
            const unsigned int author, uint8_t type) const;

    template <typename T>
    StatusCode CallTool(const EventContext& ctx,
                        const ToolHandle<IegammaBaseTool>& tool,
                        DataVector<T> *container) const;

    ToolHandleArray<IegammaBaseTool> m_electronTools {this,
        "ElectronTools", {}, "Tools for dressing ONLY electrons"};

    ToolHandleArray<IegammaBaseTool> m_photonTools {this,
        "PhotonTools", {}, "Tools for dressing ONLY photons"};

    ToolHandle<IEMClusterTool> m_clusterTool {this,
        "EMClusterTool", "egammaTools/EMClusterTool",
        "Tool that does electron/photon final cluster building"};

    ToolHandle<IEMShowerBuilder> m_ShowerTool {this,
        "EMShowerTool", "egammaTools/EMShowerBuilder",
        "Tool that does electron/photon shower shape building"};

    ToolHandle<IEGammaAmbiguityTool> m_ambiguityTool {this,
        "AmbiguityTool", "EGammaAmbiguityTool/egammaambiguitytool",
        "Tool that does electron/photon ambiguity resolution"};

    ToolHandle<IegammaOQFlagsBuilder> m_egammaOQTool {this,
      "ObjectQualityTool", {},
      "Tool that adds electron/photon Object Quality info"};
 
    // Read/Write handlers

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

    SG::WriteHandleKey<xAOD::PhotonContainer> m_photonOutputKey {this,
        "PhotonOutputName", "PhotonContainer",
        "Name of Photon Container to be created"};

    SG::ReadHandleKey<EgammaRecContainer> m_electronClusterRecContainerKey {this,
        "InputElectronRecCollectionName",
        "ElectronSuperRecCollection",
        "Input container for electron  Super Cluster  egammaRec objects"};

    SG::ReadHandleKey<EgammaRecContainer> m_photonClusterRecContainerKey {this,
        "InputPhotonRecCollectionName",
        "PhotonSuperRecCollection",
        "Input container for electron  Super Cluster  egammaRec objects"};
 
    SG::ReadCondHandleKey<CaloDetDescrManager> m_caloDetDescrMgrKey {
        this,
        "CaloDetDescrManager",
        "CaloDetDescrManager",
        "SG Key for CaloDetDescrManager in the Condition Store"
    };

    SG::WriteHandleKey<xAOD::ElectronContainer> m_dummyElectronOutputKey {this,
        "DummyElectronOutputName", "",
        "Name of Dummy Electron Container to be created"};
 
    //
    // Other properties.
    Gaudi::Property<bool> m_isTruth {this, "isTruth", false, "is truth"};
    Gaudi::Property<bool> m_doPhotons {this, "doPhotons", true, "Run the Photon reconstruction"};
    Gaudi::Property<bool> m_doElectrons {this, "doElectrons", true, "Run the Electron reconstruction"};
    std::unique_ptr<electronPearShapeAlignmentCorrection> m_deltaEta1Pear;
    bool m_doAmbiguity{};
    bool m_doOQ{};
    bool m_doDummyElectrons = false;
};
 
#endif