MuonCalibExtendedTrack.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2022 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef MUONCALIBEXTENDEDTRACK_H
#define MUONCALIBEXTENDEDTRACK_H
 
#include <algorithm>
#include <string>
#include <vector>
 
#include "AthenaKernel/getMessageSvc.h"
#include "CxxUtils/checker_macros.h"
#include "GaudiKernel/MsgStream.h"
#include "GeoPrimitives/GeoPrimitives.h"
#include "MuonCalibExtraTreeEvent/MuonCalibExtendedTrackOverlap.h"
#include "MuonCalibExtraTreeEvent/MuonCalibHit_E.h"
#include "MuonCalibExtraTreeEvent/MuonCalibLayerMapping.h"
#include "MuonCalibExtraTreeEvent/MuonCalibRawHitAssociationMap.h"
#include "MuonCalibExtraTreeEvent/MuonCalibTrackSummary.h"
#include "MuonCalibExtraTreeEvent/MuonCalibTrack_E.h"
 
namespace MuonCalib {
 
    class MuonCalibExtendedSegment;
 
    class MuonCalibExtendedTrack : public MuonCalibTrack_E {
    public:
        MuonCalibExtendedTrack(const MuonCalibTrack_E& track, int pdgCode = 0, int barCode = 0) ATLAS_CTORDTOR_NOT_THREAD_SAFE;
        virtual ~MuonCalibExtendedTrack();
 
        // float d0ip() const;
 
        // float z0ip() const;
 
        // float pip() const;
 
        int pdgCode() const;
 
        int barCode() const;
 
        const MuonCalibTrackSummary& summary() const { return m_summary; }
 
        const StationIndexHitsMap& mdtStationIndexHitsMap() const { return m_mdtHitsPerStationIndex; }
 
        const StationIndexHitsMap& rpcStationIndexHitsMap() const { return m_rpcHitsPerStationIndex; }
 
        const StationIndexHitsMap& tgcStationIndexHitsMap() const { return m_tgcHitsPerStationIndex; }
 
        const StationIndexHitsMap& cscStationIndexHitsMap() const { return m_cscHitsPerStationIndex; }
 
        std::string dump() const;
 
        std::string dumpPars() const;
 
        std::string dumpSummary() const;
 
        std::string dumpIntersects() const;
 
        const StationIntersectedLayerMap& intersectedStationLayer() const;
 
        const StationIndexSet& intersectedLayersWithoutHits() const { return m_intersectedLayersWithoutHits; }
 
        const StationIndexSet& intersectedRpcLayersWithoutHits() const { return m_intersectedRpcLayerWithoutHits; }
 
        const StationIndexSet& intersectedTgcLayersWithoutHits() const { return m_intersectedTgcLayerWithoutHits; }
 
        const IdHitMap& idHitMap() const { return m_hitIdMap; }
 
        const IdHitsMap& hitsPerChamberMap() const { return m_hitsPerChamber; }
 
        const std::vector<std::shared_ptr<MuonCalibExtendedSegment>>& associatedSegments() const;
        // const std::vector<std::shared_ptr<MuonCalibExtendedTrack>>& associatedTracks() const { return m_associatedTracks; }
 
        void addSegment(MuonCalibExtendedSegment* seg);
        // void addAssociatedTrack(MuonCalibExtendedTrack* track) { m_associatedTracks.emplace_back(track); }
 
        bool isAssociated(const MuonCalibExtendedSegment* segment) const;
 
        // bool isAssociated(const MuonCalibExtendedTrack* track) const;
 
        const MuonCalibRawHitAssociationMap& rawHitAssociationMap() const { return m_rawHitAssociationMap; }
        MuonCalibRawHitAssociationMap& rawHitAssociationMap() { return m_rawHitAssociationMap; }
 
        MuonCalibExtendedTrackOverlap calculateHitOverlap(const MuonCalibExtendedTrack& track) const;
 
        // std::shared_ptr<MuonCalibExtendedTrack> getAssociatedTrack(int author) const;
 
        bool isIDConfirmed() const;
 
    private:
        int m_pdgCode{0};
        int m_barCode{0};
        Amg::Vector3D m_pos{0., 0., 0.};
 
        Amg::Vector3D m_dir{0., 0., 0.};
 
        MuonCalibTrackSummary m_summary{};
 
        StationIndexHitsMap m_mdtHitsPerStationIndex{};
        StationIndexHitsMap m_rpcHitsPerStationIndex{};
        StationIndexHitsMap m_tgcHitsPerStationIndex{};
        StationIndexHitsMap m_cscHitsPerStationIndex{};
 
        IdHitsMap m_hitsPerChamber{};
 
        IdHitMap m_hitIdMap{};
 
        StationIntersectedLayerMap m_intersectedLayers{};
 
        StationIndexSet m_intersectedLayersWithoutHits{};
 
        StationIndexSet m_intersectedRpcLayerWithoutHits{};
 
        StationIndexSet m_intersectedTgcLayerWithoutHits{};
 
        std::vector<std::shared_ptr<MuonCalibExtendedSegment>> m_associatedSegments;
 
        std::vector<std::shared_ptr<MuonCalibExtendedTrack>> m_associatedTracks;
 
        MuonCalibRawHitAssociationMap m_rawHitAssociationMap;
    };
}  // namespace MuonCalib
#endif