SZLine.h

/**************************************************************************
 * BASF2 (Belle Analysis Framework 2)                                     *
 * Copyright(C) 2016 - Belle II Collaboration                             *
 *                                                                        *
 * Author: The Belle II Collaboration                                     *
 * Contributors: Oliver Frost                                             *
 *                                                                        *
 * This software is provided "as is" without any warranty.                *
 **************************************************************************/
#pragma once
 
#include <tracking/trackFindingCDC/geometry/SZParameters.h>
#include <tracking/trackFindingCDC/geometry/Vector2D.h>
 
namespace Belle2 {
  namespace TrackFindingCDC {
 
    class SZLine {
 
    public:
      SZLine()
        : m_tanLambda(NAN)
        , m_z0(NAN)
      {
      }
 
      SZLine(double tanLambda, double z0)
        : m_tanLambda(tanLambda)
        , m_z0(z0)
      {
      }
 
      explicit SZLine(const SZParameters& szParameters)
        : m_tanLambda(szParameters(ESZParameter::c_TanL))
        , m_z0(szParameters(ESZParameter::c_Z0))
      {
      }
 
    public:
      double tanLambda() const
      {
        return m_tanLambda;
      }
 
      double slope() const
      {
        return m_tanLambda;
      }
 
      double inverseSlope() const
      {
        return 1 / tanLambda();
      }
 
      double theta() const
      {
        return m_tanLambda;
      }
 
      void setTanLambda(double tanLambda)
      {
        m_tanLambda = tanLambda;
      }
 
      double z0() const
      {
        return m_z0;
      }
 
      double intercept() const
      {
        return m_z0;
      }
 
      void setZ0(double z0)
      {
        m_z0 = z0;
      }
 
      SZParameters szParameters() const
      {
        using namespace NSZParameterIndices;
        SZParameters result;
        result(c_TanL) = tanLambda();
        result(c_Z0) = z0();
        return result;
      }
 
      double zero() const
      {
        return -intercept() / slope();
      }
 
      double map(double s) const
      {
        return tanLambda() * s + z0();
      }
 
      double operator()(double s) const
      {
        return map(s);
      }
 
      double inverseMap(double z) const
      {
        return (z - z0()) / tanLambda();
      }
 
    public:
      void invalidate()
      {
        setZ0(NAN);
        setTanLambda(NAN);
      }
 
      bool isInvalid() const
      {
        return not std::isfinite(tanLambda()) or not std::isfinite(z0());
      }
 
      void reverse()
      {
        m_tanLambda = -m_tanLambda;
      }
 
      SZLine reversed() const
      {
        return SZLine(-tanLambda(), z0());
      }
 
    public:
      double distance(const Vector2D& szPoint) const
      {
        return distance(szPoint.first(), szPoint.second());
      }
 
      double distance(double s, double z) const
      {
        return (map(s) - z) / hypot2(1, tanLambda());
      }
 
      Vector2D intersection(const SZLine& szLine) const;
 
      void passiveMoveBy(const Vector2D& bySZ)
      {
        passiveMoveBy(bySZ.first(), bySZ.second());
      }
 
      void passiveMoveBy(double s, double z)
      {
        m_z0 = map(s) - z;
      }
 
      SZLine passiveMovedBy(const Vector2D& bySZ) const
      {
        return passiveMovedBy(bySZ.first(), bySZ.second());
      }
 
      SZLine passiveMovedBy(double s, double z) const
      {
        return SZLine(tanLambda(), map(s) - z);
      }
 
      SZJacobian passiveMoveByJacobian(const Vector2D& bySZ) const
      {
        using namespace NSZParameterIndices;
        SZJacobian result = SZUtil::identity();
        result(c_Z0, c_TanL) = bySZ.first();
        return result;
      }
 
    private:
      double m_tanLambda;
 
      double m_z0;
 
    };
  }
}