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