TransformToLine.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2024 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef DCMATH_TRANSFORMTOLINE_H
#define DCMATH_TRANSFORMTOLINE_H
 
#include <cmath>
 
#include "TrkDriftCircleMath/DriftCircle.h"
#include "TrkDriftCircleMath/Line.h"
 
namespace TrkDriftCircleMath {
 
    class TransformToLine {
    public:
        TransformToLine() = default;
        TransformToLine(const Line& line) : m_toLineRot(line.phi()), m_toLocalRot(-line.phi()), m_shift(line.x0(), line.y0()) {}
 
        LocVec2D toLine(const LocVec2D& pos) const { return m_toLineRot * pos - m_shift; }

        double toLineX(const LocVec2D& pos) const { return m_toLineRot.xval(pos) - m_shift.x(); }
        double toLineY(const LocVec2D& pos) const { return m_toLineRot.yval(pos) - m_shift.y(); }
 
        LocVec2D toLocal(const LocVec2D& pos) const { return m_toLocalRot * (pos + m_shift); }
 
        void set(const Line& line) {
            m_toLineRot.set(line.phi());
            m_toLocalRot.set(-line.phi());
            m_shift.setX(line.x0());
            m_shift.setY(line.y0());
        }
 
        DriftCircle toLine(const DriftCircle& dc) const {
            return DriftCircle(toLine(dc.position()), dc.r(), dc.dr(), dc.driftState(), dc.id(), dc.rot(), dc.index());
        }
        DriftCircle toLocal(const DriftCircle& dc) const {
            return DriftCircle(toLocal(dc.position()), dc.r(), dc.dr(), dc.driftState(), dc.id(), dc.rot(), dc.index());
        }

        const RotPhi& toLineRotation() const { return m_toLineRot; }

        const RotPhi& toLocalRotation() const { return m_toLocalRot; }
 
    private:
        RotPhi m_toLineRot{0.};
        RotPhi m_toLocalRot{0.};
        LocVec2D m_shift{0., 0.};
    };
 
}  // namespace TrkDriftCircleMath
 
#endif