PFTrackCaloExtensionTool.h

Go to the documentation of this file.

/* Copyright (C) 2002-2026 CERN for the benefit of the ATLAS collaboration */
 
#ifndef EFLOWREC_PFTRACKCALOEXTENSIONTOOL_H
#define EFLOWREC_PFTRACKCALOEXTENSIONTOOL_H
 
#include "eflowTrackExtrapolatorBaseAlgTool.h"
 
#include "ActsEvent/ContextUtility.h"
#include "Acts/Geometry/GeometryIdentifier.hpp"
#include "ActsGeometryInterfaces/ITrackingGeometrySvc.h"
#include "AthenaBaseComps/AthAlgTool.h"
#include "CaloGeoHelpers/CaloSampling.h"
 
#include "xAODTracking/TrackParticleContainer.h"

class PFTrackCaloExtensionTool: virtual public eflowTrackExtrapolatorBaseAlgTool, public AthAlgTool {
 
public:
 
  using AthAlgTool::AthAlgTool;
  ~PFTrackCaloExtensionTool() {};
 
  virtual StatusCode initialize() override;
  virtual std::unique_ptr<eflowTrackCaloPoints> execute(const EventContext& ctx, const xAOD::TrackParticle* track) const override;
  virtual StatusCode finalize() override;
 
private:

     ActsTrk::ContextUtility m_ctxProvider{this};
 
    ServiceHandle<ActsTrk::ITrackingGeometrySvc> m_trackingGeometrySvc{this, "TrackingGeometrySvc", "ActsTrackingGeometrySvc"};
 
    SG::ReadDecorHandleKey<xAOD::TrackParticleContainer> m_extensionReadDecorKey{this, "ExtensionDecoration", "InDetTrackParticles.caloExtensionLink"};
 
    //Map geometry identifier to calo sampling layer enum
    std::map<Acts::GeometryIdentifier, CaloSampling::CaloSample> m_geoIdCaloSampleMap;
 
};
#endif