MuonTrackSummaryHelperTool.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2023 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef MUONTRACKSUMMARYHELPERTOOL_H
#define MUONTRACKSUMMARYHELPERTOOL_H
 
#include <bitset>
#include <string>
#include <vector>
 
#include "AthenaBaseComps/AthAlgTool.h"
#include "GaudiKernel/ServiceHandle.h"
#include "GaudiKernel/ToolHandle.h"
#include "MuonIdHelpers/IMuonIdHelperSvc.h"
#include "MuonPrepRawData/MuonPrepDataContainer.h"
#include "MuonRecHelperTools/IMuonEDMHelperSvc.h"
#include "StoreGate/ReadHandleKey.h"
#include "TrkEventPrimitives/ParticleHypothesis.h"
#include "TrkExInterfaces/IExtrapolator.h"
#include "TrkGeometry/TrackingGeometry.h"
#include "TrkToolInterfaces/IExtendedTrackSummaryHelperTool.h"
#include "TrkToolInterfaces/ITrackHoleSearchTool.h"
#include "TrkTrackSummary/MuonTrackSummary.h"
#include "TrkTrackSummary/TrackSummary.h"
 
namespace Trk {
    class RIO_OnTrack;
    class TrackStateOnSurface;
    class CompetingRIOsOnTrack;
}  // namespace Trk
 
namespace MuonGM {
    class MuonDetectorManager;
}
 
namespace Muon {
 
    class MuonTrackSummaryHelperTool : public extends<AthAlgTool, Trk::IExtendedTrackSummaryHelperTool> {
    public:
        MuonTrackSummaryHelperTool(const std::string&, const std::string&, const IInterface*);
 
        virtual ~MuonTrackSummaryHelperTool() = default;
 
        virtual StatusCode initialize() override;
 
        using IExtendedTrackSummaryHelperTool::addDetailedTrackSummary;
        using IExtendedTrackSummaryHelperTool::analyse;
        virtual void analyse(const Trk::Track& trk, const Trk::RIO_OnTrack* rot, const Trk::TrackStateOnSurface* tsos,
                             std::vector<int>& information, std::bitset<Trk::numberOfDetectorTypes>& hitPattern) const override final;
 
        virtual void analyse(const Trk::Track& trk, const Trk::CompetingRIOsOnTrack* crot, const Trk::TrackStateOnSurface* tsos,
                             std::vector<int>& information, std::bitset<Trk::numberOfDetectorTypes>& hitPattern) const override final;
 
        virtual void searchForHoles(const Trk::Track& track, std::vector<int>& information,
                                    Trk::ParticleHypothesis hyp) const override final;
 
        virtual void addDetailedTrackSummary(const Trk::Track& track, Trk::TrackSummary& summary) const override final;
 
    private:
        const MdtPrepDataCollection* findMdtPrdCollection(const Identifier& chId) const;
        void calculateRoadHits(Trk::MuonTrackSummary::ChamberHitSummary& chamberHitSummary, const Trk::TrackParameters& pars) const;
        bool isFirstProjection(const Identifier& id) const;
        void updateHoleContent(Trk::MuonTrackSummary::ChamberHitSummary& chamberHitSummary) const;
 
        static void increment(int& type) ;
 
        ServiceHandle<Muon::IMuonIdHelperSvc> m_idHelperSvc{this, "MuonIdHelperSvc", "Muon::MuonIdHelperSvc/MuonIdHelperSvc"};
 
        /* used to do hits-in-road search for straight tracks */
        ToolHandle<Trk::IExtrapolator> m_slExtrapolator{this, "StaightLineExtrapolator", "Trk::Extrapolator/MuonStraightLineExtrapolator"};
 
        /* used to do hits-in-road search */
        ToolHandle<Trk::IExtrapolator> m_extrapolator{this, "Extrapolator", "Trk::Extrapolator/AtlasExtrapolator"};
 
        Gaudi::Property<bool> m_calculateCloseHits{this, "CalculateCloseHits", false};
 
        Gaudi::Property<double> m_roadWidth{this, "RoadWidth", 135., "width used to calculate hits within the road (mm)"};
 
        SG::ReadHandleKey<Muon::MdtPrepDataContainer> m_mdtKey{this, "MdtPrepDataContainer", "MDT_DriftCircles", "MDT PRDs"};
 
        SG::ReadCondHandleKey<MuonGM::MuonDetectorManager> m_DetectorManagerKey{this, "DetectorManagerKey", "MuonDetectorManager",
                                                                                "Key of input MuonDetectorManager condition data"};
    };
}  // namespace Muon
#endif