MuonCombinedFitTagTool.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2021 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef MUONCOMBINEDBASETOOLS_MUONCOMBINEDFITTAGTOOL_H
#define MUONCOMBINEDBASETOOLS_MUONCOMBINEDFITTAGTOOL_H
 
#include <string>
#include <vector>
 
#include "AthenaBaseComps/AthAlgTool.h"
#include "GaudiKernel/ServiceHandle.h"
#include "GaudiKernel/ToolHandle.h"
#include "MagFieldConditions/AtlasFieldCacheCondObj.h"
#include "MuidInterfaces/ICombinedMuonTrackBuilder.h"
#include "MuidInterfaces/IMuidMuonRecovery.h"
#include "MuidInterfaces/IMuonMatchQuality.h"
#include "MuidInterfaces/IMuonTrackQuery.h"
#include "MuonCombinedToolInterfaces/IMuonCombinedTagTool.h"
#include "MuonCombinedToolInterfaces/IMuonCombinedTool.h"
#include "MuonCombinedToolInterfaces/IMuonMomentumBalanceSignificance.h"
#include "MuonIdHelpers/IMuonIdHelperSvc.h"
#include "MuonRecHelperTools/MuonEDMPrinterTool.h"
#include "TrkParameters/TrackParameters.h"
#include "TrkSegment/SegmentCollection.h"
#include "TrkToolInterfaces/ITrackScoringTool.h"
#include "TrkTrack/TrackCollection.h"
#include "xAODTracking/VertexContainer.h"
 
namespace MuonCombined {
    class InDetCandidate;
    class MuonCandidate;
    class CombinedFitTag;
    class InDetCandidateToTagMap;
 
    class MuonCombinedFitTagTool : public AthAlgTool, virtual public IMuonCombinedTagTool {
    public:
        MuonCombinedFitTagTool(const std::string& type, const std::string& name, const IInterface* parent);
        virtual ~MuonCombinedFitTagTool() = default;
 
        virtual StatusCode initialize() override;
 
        virtual void combine(const MuonCandidate& muonCandidate, const std::vector<const InDetCandidate*>& indetCandidates,
                             InDetCandidateToTagMap& tagMap, TrackCollection* combTracks, TrackCollection* METracks,
                             const EventContext& ctx) const override;
 
    private:
        std::unique_ptr<Trk::Track> buildCombinedTrack(const EventContext& ctx, const Trk::Track& indetTrack, const Trk::Track& spectrometerTrack,
                                                       const Trk::Track* extrapolatedTrack) const;
 
        bool combinedTrackQualityCheck(const EventContext& ctx, const Trk::Track& combinedTrack, const Trk::Track& indetTrack) const;
 
        std::unique_ptr<Trk::Track> evaluateMatchProperties(const EventContext& ctx, const Trk::Track* combinedTrack, CombinedFitTag& tag,
                                                            const xAOD::TrackParticle& idTrackParticle) const;
 
        bool extrapolatedNeedsRefit(const EventContext& ctx, const Trk::Track& combTrack, const Trk::Track* extrTrack) const;
 
        bool bestMatchChooser(const InDetCandidate& curCandidate, const CombinedFitTag& curTag, const Trk::Track& curTrack,
                              const Trk::Track* curMETrack, const InDetCandidate& bestCandidate, const CombinedFitTag& bestTag,
                              const Trk::Track& bestTrack, const Trk::Track* bestMETrack) const;
 
        void dumpCaloEloss(const EventContext& ctx, const Trk::Track* track, const std::string& txt) const;
 
        ServiceHandle<Muon::IMuonIdHelperSvc> m_idHelperSvc{this, "MuonIdHelperSvc", "Muon::MuonIdHelperSvc/MuonIdHelperSvc"};
 
        ToolHandle<Muon::MuonEDMPrinterTool> m_printer{this, "Printer", "Muon::MuonEDMPrinterTool/MuonEDMPrinterTool"};
        ToolHandle<Rec::ICombinedMuonTrackBuilder> m_trackBuilder{this, "TrackBuilder", ""};
        ToolHandle<Rec::ICombinedMuonTrackBuilder> m_outwardsBuilder{this, "OutwardsTrackBuilder", ""};
        ToolHandle<Rec::IMuonTrackQuery> m_trackQuery{this, "TrackQuery", "Rec::MuonTrackQuery/MuonTrackQuery"};
        ToolHandle<Rec::IMuonMomentumBalanceSignificance> m_momentumBalanceTool{
            this, "MomentumBalanceTool", "Rec::MuonMomentumBalanceSignificanceTool/MuonMomentumBalanceSignifTool"};
        ToolHandle<Rec::IMuidMuonRecovery> m_muonRecovery{this, "MuonRecovery", ""};
        ToolHandle<Rec::IMuonMatchQuality> m_matchQuality{this, "MatchQuality", "Rec::MuonMatchQuality/MuonMatchQuality"};
        ToolHandle<Trk::ITrackScoringTool> m_trackScoringTool{this, "TrackScoringTool", "Muon::MuonTrackScoringTool/MuonTrackScoringTool"};
 
        Gaudi::Property<double> m_badFitChi2{this, "BadFitChi2", 2.5};
        Gaudi::Property<double> m_momentumBalanceCut{this, "MomentumBalanceCut", 6};
        Gaudi::Property<double> m_indetPullCut{this, "IndetPullCut", 6};
        Gaudi::Property<double> m_matchChiSquaredCut{this, "MatchChiSquaredCut", 30};
 
        // Read handle for conditions object to get the field cache
        SG::ReadCondHandleKey<AtlasFieldCacheCondObj> m_fieldCacheCondObjInputKey{this, "AtlasFieldCacheCondObj", "fieldCondObj",
                                                                                  "Name of the Magnetic Field conditions object key"};
 
        SG::ReadHandleKey<xAOD::VertexContainer> m_vertexKey{this, "VertexContainer", "PrimaryVertices", "primary vertex container"};
    };
 
}  // namespace MuonCombined
 
#endif  // MUONCOMBINEDBASETOOLS_MUONCOMBINEDFITTAGTOOL_H