HelixUtils.h

/**************************************************************************
 * basf2 (Belle II Analysis Software Framework)                           *
 * Author: The Belle II Collaboration                                     *
 * External Contributor: Wouter Hulsbergen                                *
 *                                                                        *
 * See git log for contributors and copyright holders.                    *
 * This file is licensed under LGPL-3.0, see LICENSE.md.                  *
 **************************************************************************/
#pragma once
 
#include <string>
 
#include <framework/dataobjects/Helix.h>
#include <framework/geometry/B2Vector3.h>
#include <Eigen/Core>
namespace TreeFitter {
 
  class HelixUtils {
 
  public:
 
    enum VertexCoor {iX = 0, iY, iZ, iPx, iPy, iPz} ;
 
    enum HelixCoor  {iD0 = 0, iPhi0, iOmega, iZ0, iTanLambda, iArcLength2D} ;
 
    static void helixFromVertexNumerical(const Eigen::Matrix<double, 3, 1>& position,
                                         const Eigen::Matrix<double, 3, 1>& momentum,
                                         int charge, double Bz,
                                         Belle2::Helix& helix,
                                         double& flt,
                                         Eigen::Matrix<double, 5, 6>& jacobian);
 
    static void getJacobianToCartesianFrameworkHelix(Eigen::Matrix<double, 5, 6>& jacobian,
                                                     const double x,
                                                     const double y,
                                                     const double z,
                                                     const double px,
                                                     const double py,
                                                     const double pz,
                                                     const double bfield,
                                                     const double charge
                                                    );
 
    static void getHelixAndJacobianFromVertexNumerical(const Eigen::Matrix<double, 1, 6>& positionAndMom,
                                                       int charge, double Bz,
                                                       Belle2::Helix& helix,
                                                       Eigen::Matrix<double, 5, 6>& jacobian);
 
    static void getJacobianFromVertexNumerical(const Eigen::Matrix<double, 1, 6>& positionAndMom,
                                               int charge, double Bz,
                                               const Belle2::Helix& helix,
                                               Eigen::Matrix<double, 5, 6>& jacobian,
                                               double delta = 1e-5
                                              );
 
    static void helixFromVertex(const Eigen::Matrix<double, 1, 6>& positionAndMomentum,
                                int charge, double Bz,
                                Belle2::Helix& helix,
                                double& L,
                                Eigen::Matrix<double, 5, 6>& jacobian);
 
 
    static void vertexFromHelix(const Belle2::Helix& helix,
                                double L, double Bz,
                                ROOT::Math::XYZVector& position,
                                ROOT::Math::XYZVector& momentum, int& charge);
 
    static std::string helixParName(int i) ;
 
    static std::string vertexParName(int i) ;
 
    static void printVertexPar(const Belle2::B2Vector3D& position, const Belle2::B2Vector3D& momentum, int charge) ;
 
    static double helixPoca(const Belle2::Helix& helix1,
                            const Belle2::Helix& helix2,
                            double& flt1, double& flt2,
                            Belle2::B2Vector3D& vertex, bool parallel = false) ;
 
    static double helixPoca(const Belle2::Helix& helix, const Belle2::B2Vector3D& point,
                            double& flt) ;
 
    static double phidomain(const double phi) ;
 
  } ;
 
}