CDCTrajectorySZ.h

/**************************************************************************
 * basf2 (Belle II Analysis Software Framework)                           *
 * Author: The Belle II Collaboration                                     *
 *                                                                        *
 * See git log for contributors and copyright holders.                    *
 * This file is licensed under LGPL-3.0, see LICENSE.md.                  *
 **************************************************************************/
#pragma once
 
#include <tracking/trackingUtilities/geometry/UncertainSZLine.h>
 
#include <Math/Vector2D.h>
 
#include <TMath.h>
 
namespace Belle2 {
  namespace TrackingUtilities {

    class CDCTrajectorySZ  {
 
    public:
      static SZCovariance getBasicSZCovariance();
 
    public:
      CDCTrajectorySZ() {}

      explicit CDCTrajectorySZ(const UncertainSZLine& szLine)
        : m_szLine(szLine)
      {}

      CDCTrajectorySZ(const double tanLambda, const double z0)
        : m_szLine(tanLambda, z0, getBasicSZCovariance())
      {}

      static CDCTrajectorySZ basicAssumption();

      double mapSToZ(const double s = 0) const
      { return getSZLine()->map(s); }

      double mapPtToPz(const double pt) const
      { return getSZLine()->slope() * pt; }

      double getZDist(const double s, const double z) const
      { return mapSToZ(s) - z; }

      double getSZDist(const double s, const double z) const
      { return getSZLine()->distance(ROOT::Math::XYVector(s, z)) ; }

      void passiveMoveS(const double deltaS)
      { m_szLine.passiveMoveBy(ROOT::Math::XYVector(deltaS, 0)); }

      bool isFitted() const
      { return not m_szLine->isInvalid(); }

      void clear()
      { m_szLine.invalidate(); }

      double getTanLambda() const
      { return getSZLine()->slope(); }

      double getZ0() const
      { return getSZLine()->intercept(); }

      double getPValue() const
      { return TMath::Prob(getChi2(), getNDF()); }

      double getChi2() const
      { return getSZLine().chi2(); }

      void setChi2(const double chi2)
      { return m_szLine.setChi2(chi2); }

      size_t getNDF() const
      { return getSZLine().ndf(); }

      void setNDF(std::size_t ndf)
      { return m_szLine.setNDF(ndf); }

      const UncertainSZLine& getSZLine() const
      { return m_szLine; }

      void setSZLine(const UncertainSZLine& szLine)
      { m_szLine = szLine; }
 
    private:
      UncertainSZLine m_szLine;
    };
  }
}