CDCTrajectorySZ.h

/**************************************************************************
 * BASF2 (Belle Analysis Framework 2)                                     *
 * Copyright(C) 2012 - 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/UncertainSZLine.h>
#include <tracking/trackFindingCDC/geometry/Vector2D.h>
 
#include <TMath.h>
 
namespace Belle2 {
  namespace TrackFindingCDC {
 
    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(Vector2D(s, z)) ; }
 
      void passiveMoveS(const double deltaS)
      { m_szLine.passiveMoveBy(Vector2D(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;
    };
  }
}