MuonHoughHit.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2020 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef MUONHOUGHPATTERNEVENT_MUONHOUGHHIT_H
#define MUONHOUGHPATTERNEVENT_MUONHOUGHHIT_H
 
#include <GeoPrimitives/GeoPrimitives.h>
 
#include <cmath>
#include <iostream>
#include <string>
#include <vector>
 
namespace MuonHough {
    enum DetectorTechnology { MDT, CSC, RPC, TGC };
    constexpr int phisectors = 16;
    constexpr double half_phisector = M_PI / phisectors;
}  // namespace MuonHough
 
namespace Trk {
    class PrepRawData;
}
 
class MuonHoughHit {
public:
    //  MuonHoughHit();
    MuonHoughHit(const Trk::PrepRawData* prd);
    MuonHoughHit(const Amg::Vector3D& pos_vec, bool measures_phi, MuonHough::DetectorTechnology detector_id, double prob = 1,
                 double weight = 1., const Trk::PrepRawData* prd = nullptr, int id = -1);
    virtual ~MuonHoughHit() = default;

    MuonHough::DetectorTechnology getDetectorId() const;
    std::string getWhichDetector() const;

    const Amg::Vector3D& getPosition() const;
    int getId() const;
    double getHitx() const;
    double getHity() const;
    double getHitz() const;
    double getWeight() const;
    double getOrigWeight() const;
    double getProbability() const;

    double getRadius() const;
    double getAbs() const;
    double getTheta() const;
    double getPhi() const;

    bool isBarrel() const;
    int phiSector() const;
    double getMagneticTrackRatio() const;

    bool getMeasuresPhi() const;

    bool getAssociated() const;

    void setId(int id);

    void setWeight(double weight);

    void setProbability(double prob);

    void setAssociated(bool associated);

    const Trk::PrepRawData* getPrd() const;
 
protected:
    const Trk::PrepRawData* m_prd{nullptr};
 
    Amg::Vector3D m_pos{Amg::Vector3D::Zero()};
    int m_id{0};

    double m_radius{0.};
    double m_abs{0.};
    double m_phi{0.};
    double m_theta{0.};
    bool m_barrel{true};
    int m_phi_sector{0};
    double m_magnetic_trackratio{0.};
    double m_weight{1.};
    const double m_orig_weight{1.};
    double m_probability{1.};

    MuonHough::DetectorTechnology m_detector_id{MuonHough::MDT};

    bool m_measures_phi{false};

    bool m_associated{false};
 
private:
    int calcPhiSector() const;
    double calcMagneticTrackRatio() const;
};
 
inline bool operator==(const MuonHoughHit& hit1, const MuonHoughHit& hit2) {
    bool equal_to = 0;
 
    if (hit1.getHitx() == hit2.getHitx() && hit1.getHity() == hit2.getHity() && hit1.getHitz() == hit2.getHitz()) { equal_to = 1; }
 
    return equal_to;
}
 
inline MuonHough::DetectorTechnology MuonHoughHit::getDetectorId() const { return m_detector_id; }
 
inline int MuonHoughHit::getId() const { return m_id; }
inline const Amg::Vector3D& MuonHoughHit::getPosition() const { return m_pos; }
 
inline double MuonHoughHit::getHitx() const { return m_pos[Amg::x]; }
inline double MuonHoughHit::getHity() const { return m_pos[Amg::y]; }
inline double MuonHoughHit::getHitz() const { return m_pos[Amg::z]; }
inline double MuonHoughHit::getWeight() const { return m_weight; }
inline double MuonHoughHit::getOrigWeight() const { return m_orig_weight; }
inline double MuonHoughHit::getProbability() const { return m_probability; }
 
inline double MuonHoughHit::getRadius() const { return m_radius; }
inline double MuonHoughHit::getAbs() const { return m_abs; }
inline double MuonHoughHit::getTheta() const { return m_theta; }
inline double MuonHoughHit::getPhi() const { return m_phi; }
 
inline bool MuonHoughHit::isBarrel() const { return m_barrel; }
inline int MuonHoughHit::phiSector() const { return m_phi_sector; }
inline double MuonHoughHit::getMagneticTrackRatio() const { return m_magnetic_trackratio; }
 
inline bool MuonHoughHit::getMeasuresPhi() const { return m_measures_phi; }
inline bool MuonHoughHit::getAssociated() const { return m_associated; }
 
inline void MuonHoughHit::setId(int id) { m_id = id; }
inline void MuonHoughHit::setWeight(double weight) { m_weight = weight; }
inline void MuonHoughHit::setProbability(double prob) { m_probability = prob; }
inline void MuonHoughHit::setAssociated(bool associated) { m_associated = associated; }
 
inline const Trk::PrepRawData* MuonHoughHit::getPrd() const { return m_prd; }
 
#endif  // MUONHOUGHPATTERNEVENT_MUONHOUGHHIT_H