Acts/ActsEvent/ActsEvent/CaloExtension.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2026 CERN for the benefit of the ATLAS collaboration
*/
#ifndef ACTSEVENT_CALOEXTENSION_H
#define ACTSEVENT_CALOEXTENSION_H
 
#include "GeoPrimitives/GeoPrimitives.h"
#include "ActsGeometryInterfaces/GeometryDefs.h"
#include "xAODTracking/TrackParticle.h"
#include "xAODCaloEvent/CaloCluster.h"
 
#include "Acts/EventData/BoundTrackParameters.hpp"
#include "Acts/Utilities/TransformRange.hpp"
 
#include "AthContainers/DataVector.h"
#include "AthenaKernel/CLASS_DEF.h"
 
#include <span>
 
 
namespace ActsTrk{
    class CaloExtension;
    const CaloExtension* getCaloExtension(const xAOD::TrackParticle& track);
 

    class CaloExtension {
        public:
            CaloExtension(const xAOD::TrackParticle* track);
            /*** @brief Returns the associated ID track particle */
            const xAOD::TrackParticle* track() const;
            std::size_t size() const;
            bool empty() const;
            const Acts::BoundTrackParameters& parameters(const std::size_t parIdx) const;
            auto caloParameters(const CaloSampling::CaloSample s) const {
                using namespace ActsTrk::detail::GeoVolIds;
                constexpr std::array caloIds{s_caloEnvelopeID, s_caloBarrelId,
                                             s_caloEndcapAId, s_caloEndcapCId};
                return std::ranges::filter_view(m_parameters, 
                  [&](const Acts::BoundTrackParameters& pars){
                    const Acts::GeometryIdentifier id = pars.referenceSurface().geometryId();
                    return id.layer() == static_cast<std::size_t>(s) && 
                            Acts::rangeContainsValue(caloIds, id.volume());
                });
            }

            using ClusterVec_t = std::vector<const xAOD::CaloCluster*>;
            using ParamVec_t = std::vector<Acts::BoundTrackParameters>;
            void appendParameters(Acts::BoundTrackParameters&& pars);
            void associateCluster(const xAOD::CaloCluster* clust);
            std::optional<Acts::BoundTrackParameters> lastParameters() const;
            const ParamVec_t& parameters() const;
            const ClusterVec_t& associatedClusters() const;
            std::optional<Acts::BoundTrackParameters> lastTrackParameters() const;
        private:
            const xAOD::TrackParticle* m_idTrack{nullptr};
            ParamVec_t m_parameters{};
            ClusterVec_t m_clusters{};
 
    };
    using CaloExtensionContainer = DataVector<CaloExtension>;
 
}
CLASS_DEF( ActsTrk::CaloExtensionContainer , 1237047548 , 1 )
 
#endif