IDTrackCaloDepositsDecoratorAlg.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2026 CERN for the benefit of the ATLAS collaboration
*/
#ifndef IDTRACKCALODEPOSITSDECORATORALG_H_
#define IDTRACKCALODEPOSITSDECORATORALG_H_
 
#include <AthenaBaseComps/AthReentrantAlgorithm.h>
#include "ICaloTrkMuIdTools/ITrackDepositInCaloTool.h"
#include "StoreGate/ReadHandleKey.h"
#include "StoreGate/WriteDecorHandleKey.h"
#include "StoreGate/ReadDecorHandleKeyArray.h"
#include "xAODMuon/MuonContainer.h"
 
namespace DerivationFramework {
 
class IDTrackCaloDepositsDecoratorAlg : public AthReentrantAlgorithm {
public:
    using AthReentrantAlgorithm::AthReentrantAlgorithm;
 
    virtual ~IDTrackCaloDepositsDecoratorAlg() = default;
    StatusCode initialize() override;
 
    StatusCode execute(const EventContext& ctx) const override;
 
private:
 
    ToolHandle<ITrackDepositInCaloTool> m_trkDepositInCalo{this, "TrackDepositInCaloTool", "TrackDepositInCaloTool/TrackDepositInCaloTool"};

    SG::ReadHandleKey<xAOD::IParticleContainer> m_partKey{this, "ContainerKey", "RandomParticle"};
 
    Gaudi::Property<float> m_ptMin{this, "PtMin", 2.5 * Gaudi::Units::GeV, 
                                    "Minimal track pt required to decorate the ID track"};

    Gaudi::Property<std::vector<std::string>> m_trkSelDecors{this, "TrackSelections", {},
                                                  "List of decorator names of which one needs to be true to run the isolation" };
    SG::ReadDecorHandleKeyArray<xAOD::IParticleContainer> m_trkSelKeys{this, "SelectionKeys", {},
                                                                       "Will be overwritten in initialize"};
 
    SG::WriteDecorHandleKey<xAOD::IParticleContainer> m_depositKey{this, "Calo_Deposit", m_partKey, "CaloDeposits", 
                                                "Name of the decorator to store all the energy deposits in the calorimeter"};
    SG::WriteDecorHandleKey<xAOD::IParticleContainer> m_elossKey{this, "Calo_EnergyLoss", m_partKey, "CaloElosses",
                                                                  "Name of the decorator to store the energy loss from EMB1."};
    SG::WriteDecorHandleKey<xAOD::IParticleContainer> m_typeKey{this, "Calo_Type", m_partKey, "CaloDepType",
                                                                 "Name of the decorator to store the energy deposit from EMB2."};
};
 
}
 
#endif