MuonHoughPattern.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2020 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef MUONHOUGHPATTERNEVENT_MUONHOUGHPATTERN_H
#define MUONHOUGHPATTERNEVENT_MUONHOUGHPATTERN_H
 
#include "AthContainers/DataVector.h"
#include "EventPrimitives/EventPrimitives.h"
#include "MuonHoughPatternEvent/MuonHoughHitContainer.h"
#include "MuonHoughPatternEvent/MuonHoughMathUtils.h"
namespace MuonHough {
    enum MuonHoughTransformers { hough_xy, hough_rzcosmics, hough_curved_at_a_cylinder, hough_rz, hough_rz_mdt, hough_rz_rpc, hough_yz };
}  // namespace MuonHough
 
class MuonHoughPattern : public MuonHoughHitContainer {
 
public:
    MuonHoughPattern(int id_number);

    virtual ~MuonHoughPattern() = default;

    void resetTracksegment();

    bool hitInHoughPattern(const std::shared_ptr<MuonHoughHit>& hit) const;

    double calculateEZ() const;
    double patternLength() const;

    void printHoughPattern() const;

    int getIdNumber() const;

    double getEPhi() const;
    double getERPhi() const;
    double getETheta() const;
    double getERTheta() const;
    double getECurvature() const;
    double getMaximumHistogram() const;

    void setEPhi(double ephi);
    void setERPhi(double erphi);
    void setETheta(double etheta);
    void setERTheta(double ertheta);
    void setECurvature(double curvature);
    void setMaximumHistogram(double maximumhistogram);

    double getEAngle() const;
    double getER() const;
    void setEAngle(double eangle);
    void setER(double er);

    void setWhichSegment(bool which_segment);

    Amg::Vector3D getEPos() const;
    Amg::Vector3D getEDir() const;

    void updateParametersRPhi(bool cosmics = false);
 
    using MuonHoughPatternCollection = DataVector<MuonHoughPattern>;
 
private:
    int m_id_number{-1};

    bool m_whichsegment{false};

    double m_ephi{-M_PI_2};
    double m_erphi{0.};
    double m_etheta{M_PI_2};
    double m_ertheta{0.};
    double m_ecurvature{1.};
    double m_maximumhistogram{0.};
};
 
inline int MuonHoughPattern::getIdNumber() const { return m_id_number; }
 
inline double MuonHoughPattern::getEPhi() const { return m_ephi; }
inline double MuonHoughPattern::getERPhi() const { return m_erphi; }
inline double MuonHoughPattern::getETheta() const { return m_etheta; }
inline double MuonHoughPattern::getERTheta() const { return m_ertheta; }
inline double MuonHoughPattern::getECurvature() const { return m_ecurvature; }
inline double MuonHoughPattern::getMaximumHistogram() const { return m_maximumhistogram; }
 
inline void MuonHoughPattern::setEPhi(double ephi) { m_ephi = ephi; }
inline void MuonHoughPattern::setERPhi(double erphi) { m_erphi = erphi; }
inline void MuonHoughPattern::setETheta(double etheta) { m_etheta = etheta; }
inline void MuonHoughPattern::setERTheta(double ertheta) { m_ertheta = ertheta; }
inline void MuonHoughPattern::setECurvature(double ecurvature) { m_ecurvature = ecurvature; }
inline void MuonHoughPattern::setMaximumHistogram(double maximumhistogram) { m_maximumhistogram = maximumhistogram; }
 
inline void MuonHoughPattern::setWhichSegment(bool which_segment) { m_whichsegment = which_segment; }
#endif  // MUONHOUGHPATTERNEVENT_MUONHOUGHPATTERN_H