InDetCandidate.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2021 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef MUONCOMBINEDEVENT_INDETCANDIDATE_H
#define MUONCOMBINEDEVENT_INDETCANDIDATE_H
 
#include <vector>
 
#include "AthLinks/ElementLink.h"
#include "MuonLayerEvent/MuonSystemExtension.h"
#include "TrkCaloExtension/CaloExtension.h"
#include "xAODTracking/TrackParticle.h"
#include "xAODTracking/TrackParticleContainer.h"
 
namespace MuonCombined {
 
    class InDetCandidate {
    public:
        InDetCandidate(const xAOD::TrackParticle& idTrack);

        InDetCandidate(const ElementLink<xAOD::TrackParticleContainer>& idTrackLink);
        bool operator==(const InDetCandidate& other) const;

        ~InDetCandidate();

        const xAOD::TrackParticle& indetTrackParticle() const;

        const ElementLink<xAOD::TrackParticleContainer>& indetTrackParticleLink() const;
 
        // access MuonSystemExtension
        const Muon::MuonSystemExtension* getExtension() const;
        // access the CaloExtension
        const Trk::CaloExtension* getCaloExtension() const;
 
        // set MuonSystemExtension, taking ownership
        void setExtension(std::unique_ptr<Muon::MuonSystemExtension> extension);
        void setExtension(std::unique_ptr<Trk::CaloExtension>& extension);
        void setExtension(const Trk::CaloExtension* extension);

        bool isSiliconAssociated() const;

        void setSiliconAssociated(bool);

        std::string toString() const;
 
    private:
        ElementLink<xAOD::TrackParticleContainer> m_idTrackParticleLink;

        const xAOD::TrackParticle* m_idTrackParticle{nullptr};

        bool m_siAssociated{false};
 
        std::unique_ptr<Muon::MuonSystemExtension> m_extension{nullptr};
        std::unique_ptr<Trk::CaloExtension> m_calo_extension{nullptr};
        const Trk::CaloExtension* m_calo_extension_ptr{nullptr};
    };
 
}  // namespace MuonCombined
 
#endif