TauTrackCaloExtrapolator.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2026 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef XAOD_ANALYSIS
#ifndef TAURECTOOLS_TAUTRACKCALOEXTRAPOLATOR_H
#define TAURECTOOLS_TAUTRACKCALOEXTRAPOLATOR_H
 
#include "tauRecTools/TauRecToolBase.h"
#include "GaudiKernel/ToolHandle.h"
#include "StoreGate/ReadHandleKey.h"
#include "TrkCaloExtension/CaloExtensionCollection.h"
 
#include "RecoToolInterfaces/IParticleCaloExtensionTool.h"
 
#include "ActsGeometryInterfaces/IExtrapolationTool.h"
#include "ActsGeometryInterfaces/ITrackingGeometrySvc.h"
 
#include <vector>


 
class TauTrackCaloExtrapolator : public TauRecToolBase {
public:
    //-------------------------------------------------------------
    //-------------------------------------------------------------
    TauTrackCaloExtrapolator(const std::string& name);
    ASG_TOOL_CLASS2(TauTrackCaloExtrapolator, TauRecToolBase, ITauToolBase);
    ~TauTrackCaloExtrapolator();
 
    //-------------------------------------------------------------
    //-------------------------------------------------------------
    virtual StatusCode initialize() override;
 
    using TauRecToolBase::executeTool;
    virtual StatusCode executeTool(xAOD::TauJet& pTau,
                   const EventContext& ctx,
                   xAOD::TauTrackContainer& tauTrackCon) const override;
    
private:
 
    //-------------------------------------------------------------
    //-------------------------------------------------------------
    ToolHandle<Trk::IParticleCaloExtensionTool> m_caloExtensionTool {this, "ParticleCaloExtensionTool", "Trk::ParticleCaloExtensionTool/ParticleCaloExtensionTool", "Tool for the extrapolation of charged tracks"};
    SG::ReadHandleKey<CaloExtensionCollection> m_ParticleCacheKey {this,"tauParticleCache", "ParticleCaloExtension", "Name of the particle measurement extrapolation cache for TauTrackCaloExtrapolator"};
    Gaudi::Property<bool> m_extrapOnlyChargedTracks{this,"ExtrapOnlyChargedTracks", false};        
 
    std::set<CaloSampling::CaloSample> m_EMSamplings;
    std::set<CaloSampling::CaloSample> m_HadSamplings;
 
    // ACTS Extrapolator
    Gaudi::Property<bool> m_useACTSExtrapolator{this, "useACTSExtrapolator", false};
    ToolHandle<ActsTrk::IExtrapolationTool> m_extrapolationTool {this, "ExtrapolationTool", "ExtrapolationTool", "Tool to run propagation in an ACTS tracking geometry" };
    ServiceHandle<ActsTrk::ITrackingGeometrySvc> m_trackingGeometrySvc{this, "TrackingGeometrySvc", "ActsTrackingGeometrySvc"};
 
    std::map<unsigned int, std::string> m_caloNameGeoIDMap;
 
};
 
#endif // TAURECTOOLS_TAUTRACKCALOEXTRAPOLATOR_H
#endif // XAOD_ANALYSIS