CombinedFitAlg.h

Go to the documentation of this file.

/*
   Copyright (C) 2002-2026 CERN for the benefit of the ATLAS collaboration
*/
#ifndef MUONCOMBINEDALGSR4_COMBINEDFITALG_H
#define MUONCOMBINEDALGSR4_COMBINEDFITALG_H
 
#include "AthenaBaseComps/AthReentrantAlgorithm.h"
 
 
 
#include "ActsToolInterfaces/IFitterTool.h"
#include "ActsToolInterfaces/ITrackToTrackParticleCnvTool.h"
#include "ActsGeometryInterfaces/IExtrapolationTool.h"
 
#include "ActsEvent/ContextUtility.h"
#include "ActsEvent/TrackContainer.h"
 
#include "StoreGate/ReadHandle.h"
#include "StoreGate/WriteHandle.h"
#include "StoreGate/WriteDecorHandleKey.h"
 
 
#include "MuonTrackEvent/MuonTag.h"
#include "MuonTrackEvent/TrackingHelpers.h"
 
#include "xAODMuonViews/FillContainer.h"
#include "xAODTracking/TrackParticleAuxContainer.h"
 
namespace MuonCombinedR4 {
 
     class CombinedFitAlg : public AthReentrantAlgorithm {
        public:
            using AthReentrantAlgorithm::AthReentrantAlgorithm;
 
            virtual StatusCode initialize() override final;
            virtual StatusCode execute(const EventContext& ctx) const override final;
 
        private:
            struct DataShip{
                std::map<const xAOD::TrackParticle*,
                        std::vector<const MuonR4::MuonTag*>,
                        MuonR4::ParticleSorter> preMatchedTags{};
                xAOD::FillContainer<MuonR4::MuonTagContainer, void*> outputTags{};
                xAOD::FillContainer<xAOD::TrackParticleContainer,
                                    xAOD::TrackParticleAuxContainer> cmbParticles{};
                xAOD::FillContainer<xAOD::TrackParticleContainer,
                                    xAOD::TrackParticleAuxContainer> meParticles{};
                Acts::GeometryContext tgContext{Acts::GeometryContext::dangerouslyDefaultConstruct()};
                Acts::MagneticFieldContext mfContext{};
                Acts::CalibrationContext calContext{};
                ActsTrk::MutableTrackContainer actsCmbTracks{Acts::VectorTrackContainer{}, 
                                                                  Acts::VectorMultiTrajectory{}};

                ActsTrk::MutableTrackContainer actsMeTracks{Acts::VectorTrackContainer{}, 
                                                            Acts::VectorMultiTrajectory{}};
            };

            StatusCode prepareDataShip(const EventContext& ctx, DataShip& ship) const;
            MuonR4::MuonTag* fitMuonTag(const MuonR4::MuonTag& tag,
                                        DataShip& ship) const;
            
            void buildMETrack(MuonR4::MuonTag& combinedTag,
                               DataShip& ship) const;
            StatusCode writeTracksToStoreGate(const EventContext& ctx,
                                              const SG::WriteHandleKey<ActsTrk::TrackContainer>& key,
                                              const SG::WriteDecorHandleKey<xAOD::TrackParticleContainer>& decorKey,
                                              xAOD::TrackParticleContainer& particleContainer,
                                              ActsTrk::MutableTrackContainer&& fittedTracks) const;

            std::vector<Acts::SourceLink> extractMeasurements(const MuonR4::MuonTag& muonTag) const;
            std::vector<Acts::SourceLink> extractMeasurements(const xAOD::TrackParticle& trackPart) const;
            ToolHandle<ActsTrk::ITrackToTrackParticleCnvTool> m_cnvTool{this, "TrackToTrackParticleCnvTool", ""};
            ToolHandle<ActsTrk::IFitterTool> m_trackFitTool{this, "FittingTool", ""};
            ToolHandle<ActsTrk::IExtrapolationTool> m_extrapolationTool{this, "ExtrapolationTool" ,"" };
            ActsTrk::ContextUtility m_ctxProvider{this};
            SG::ReadHandleKey<MuonR4::MuonTagContainer> m_stacoKey{this, "StacoKey", "MuonTagsSTACO"};
            SG::WriteHandleKey<MuonR4::MuonTagContainer> m_outTagKey{this, "MuonTagKey" , "MuonTagsMuidCo"};
            SG::ReadHandleKey<xAOD::UncalibratedMeasurementContainer> m_beamSpotKey{this, "BeamSpotKey", "BeamSpotMeasurements"};
            SG::WriteHandleKey<xAOD::TrackParticleContainer> m_CmbParticleKey{this, "CombinedParticleKey", 
                                                                                 "MuonCmbTrackParticles"};

            SG::WriteHandleKey<xAOD::TrackParticleContainer> m_MeParticleKey{this, "MeParticleKey",
                                                                                "MuonMeTrackParticles"};

            SG::WriteHandleKey<ActsTrk::TrackContainer> m_cmbTrackKey{this, "CombinedTrackKey", "MuonCmbTracks"};
            SG::WriteHandleKey<ActsTrk::TrackContainer> m_meTrackKey{this, "MeTrackKey", "MuonMeTracks"};

            SG::WriteDecorHandleKey<xAOD::TrackParticleContainer> m_cmbTrackLinkKey{this, "CombinedTrackLinkKey", 
                                                                                    m_CmbParticleKey, "actsTrack"};
 
            SG::WriteDecorHandleKey<xAOD::TrackParticleContainer> m_meTrackLinkKey{this, "MeTrackLinKey", 
                                                                                  m_MeParticleKey, "actsTrack"};

            Gaudi::Property<bool> m_fitMEtrack{this, "fitMETracks" , false};
    };
}
 
 
#endif