DriftCircle.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2024 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef DCMATH_DRIFTCIRCLE_H
#define DCMATH_DRIFTCIRCLE_H
 
#include <ostream>
#include <vector>
 
#include "GaudiKernel/MsgStream.h"
#include "TrkDriftCircleMath/LocVec2D.h"
#include "TrkDriftCircleMath/MdtId.h"
 
namespace Muon {
    class MdtDriftCircleOnTrack;
}
 
namespace TrkDriftCircleMath {
 
    class DriftCircle {
    public:
        enum DriftState {
            EarlyHit = -1, 
            InTime = 0,    
            LateHit = 1,   
            EmptyTube = 2, 
            NotCrossed = 3 
        };
 
    public:
 
        DriftCircle(const LocVec2D& pos, 
                    double r, 
                    double dr, 
                    DriftState state, 
                    MdtId id,
                    const Muon::MdtDriftCircleOnTrack* rot,
                    unsigned int index = 0) :
            m_pos(pos), m_r(r), m_dr(dr), m_drPrecise(dr), m_state(state), m_id(id), m_rot(rot), m_idx{index} {}
 
        DriftCircle(const LocVec2D& pos, 
                    double r, 
                    double dr, 
                    double drPrecise, 
                    DriftState state, 
                    MdtId id,
                    const Muon::MdtDriftCircleOnTrack* rot,
                    unsigned int index  = 0) :
            m_pos(pos), m_r(r), m_dr(dr), m_drPrecise(drPrecise), m_state(state), m_id(id), m_rot(rot),
            m_idx{index} {}
 
        virtual ~DriftCircle() = default;
 
        const LocVec2D& position() const { return m_pos; }
 
        const MdtId& id() const { return m_id; }
 
        double x() const { return m_pos.x(); }
 
        double y() const { return m_pos.y(); }
 
        double r() const { return m_r; }
 
        double dr() const { return m_dr; }
 
        double drPrecise() const { return m_drPrecise; }
 
        const DriftState& driftState() const { return m_state; }
        const Muon::MdtDriftCircleOnTrack* rot() const { return m_rot; }
 
        unsigned int index() const { return m_idx; }
        void updateRadius(double r, double dr) {
            m_r = r;
            m_dr = dr;
        }
 
    private:
        LocVec2D m_pos{0., 0.};
        double m_r{0.};
        double m_dr{0.};
        double m_drPrecise{0.};
        DriftState m_state;  // Masked by DCOnTrack::m_state which is a DCOnTrackState
        MdtId m_id{};
        const Muon::MdtDriftCircleOnTrack* m_rot{nullptr};
        unsigned int m_idx{0};
    };
 
    typedef std::vector<DriftCircle> DCVec;
    typedef DCVec::iterator DCIt;
    typedef DCVec::const_iterator DCCit;
}  // namespace TrkDriftCircleMath
 
std::ostream& operator<<(std::ostream& os, const TrkDriftCircleMath::DriftCircle& dc);
 
MsgStream& operator<<(MsgStream& os, const TrkDriftCircleMath::DriftCircle& dc);
 
#endif