MuonTrackQuery.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2021 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef MUIDTRACKBUILDER_MUONTRACKQUERY_H
#define MUIDTRACKBUILDER_MUONTRACKQUERY_H
 
#include "AthenaBaseComps/AthAlgTool.h"
#include "GaudiKernel/ServiceHandle.h"
#include "GaudiKernel/ToolHandle.h"
#include "MuidInterfaces/IMuonTrackQuery.h"
#include "MuonIdHelpers/IMuonIdHelperSvc.h"
#include "MuonRecHelperTools/IMuonEDMHelperSvc.h"
#include "MuonRecToolInterfaces/IMdtDriftCircleOnTrackCreator.h"
#include "StoreGate/ReadCondHandleKey.h"
#include "TrkFitterInterfaces/ITrackFitter.h"
#include "TrkGeometry/TrackingGeometry.h"
namespace Rec {
 
    class MuonTrackQuery : public AthAlgTool, virtual public IMuonTrackQuery {
    public:
        MuonTrackQuery(const std::string& type, const std::string& name, const IInterface* parent);
        ~MuonTrackQuery() = default;
 
        StatusCode initialize() override;

        const CaloEnergy* caloEnergy(const Trk::Track& track) const override;

        double caloEnergyDeposit(const Trk::Track& track, const EventContext& ctx) const override;

        FieldIntegral fieldIntegral(const Trk::Track& track, const EventContext& ctx) const override;

        bool isCaloAssociated(const Trk::Track& track, const EventContext& ctx) const override;

        bool isCombined(const Trk::Track& track, const EventContext& ctx) const override;

        bool isExtrapolated(const Trk::Track& track, const EventContext& ctx) const override;

        bool isLineFit(const Trk::Track& track) const override;

        bool isProjective(const Trk::Track& track) const override;

        bool isSectorOverlap(const Trk::Track& track) const override;

        bool isSlimmed(const Trk::Track& track) const override;

        double momentumBalanceSignificance(const Trk::Track& track, const EventContext& ctx) const override;

        unsigned numberPseudoMeasurements(const Trk::Track& track) const override;

        std::unique_ptr<const Trk::Perigee> outgoingPerigee(const Trk::Track& track) const override;

        ScatteringAngleSignificance scatteringAngleSignificance(const Trk::Track& track, const EventContext& ctx) const override;

        std::unique_ptr<Trk::TrackParameters> spectrometerParameters(const Trk::Track& track, const EventContext& ctx) const override;

        unsigned spectrometerPhiQuality(const Trk::Track& track, const EventContext& ctx) const override;

        std::unique_ptr<const Trk::TrackParameters> triggerStationParameters(const Trk::Track& track,
                                                                             const EventContext& ctx) const override;
 
    private:
        static std::unique_ptr<Trk::TrackParameters> flippedParameters(const Trk::TrackParameters& params);
 
        // tools and services
        ToolHandle<Trk::ITrackFitter> m_fitter{this, "Fitter", "", "Track fitter tool"};
 
        ServiceHandle<Muon::IMuonEDMHelperSvc> m_edmHelperSvc{
            this,
            "edmHelper",
            "Muon::MuonEDMHelperSvc/MuonEDMHelperSvc",
            "Handle to the service providing the IMuonEDMHelperSvc interface",
        };
 
        ServiceHandle<Muon::IMuonIdHelperSvc> m_idHelperSvc{this, "MuonIdHelperSvc", "Muon::MuonIdHelperSvc/MuonIdHelperSvc"};
 
        ToolHandle<Muon::IMdtDriftCircleOnTrackCreator> m_mdtRotCreator{
            this, "MdtRotCreator", "Muon::MdtDriftCircleOnTrackCreator/MdtDriftCircleOnTrackCreator", "MdtDriftCircleOnTrackCreator tool"};
 
        SG::ReadCondHandleKey<Trk::TrackingGeometry> m_trackingGeometryReadKey{this, "TrackingGeometryReadKey", "",
                                                                               "Key of the TrackingGeometry conditions data."};
 
        inline const Trk::TrackingVolume* getVolume(const std::string&& vol_name, const EventContext& ctx) const {
            SG::ReadCondHandle<Trk::TrackingGeometry> handle(m_trackingGeometryReadKey, ctx);
            if (!handle.isValid()) {
                ATH_MSG_WARNING("Could not retrieve a valid tracking geometry");
                return nullptr;
            }
            return handle->trackingVolume(vol_name);
            
        }
 
    };  // end of class MuonTrackQuery
 
}  // end of namespace Rec
 
#endif  // MUIDTRACKBUILDER_MUONTRACKQUERY_H