RungeKuttaUtils.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2022 CERN for the benefit of the ATLAS collaboration
*/

//  Header file for class  RungeKuttaUtils, (c) ATLAS Detector software
//  author Igor Gavrilenko
 
#ifndef RungeKuttaUtils_H
#define RungeKuttaUtils_H
 
#include "CxxUtils/restrict.h"
#include "TrkNeutralParameters/NeutralParameters.h"
#include "TrkParameters/TrackParameters.h"
#include "TrkSurfaces/BoundaryCheck.h"
#include "EventPrimitives/EventPrimitives.h"
 
#include <map>
#include <vector>
#include <utility>
 
namespace Trk {
 
class ConeSurface;
class DiscSurface;
class PlaneSurface;
class CylinderSurface;
class StraightLineSurface;
class PatternTrackParameters;

 
namespace RungeKuttaUtils {

// Step estimators to surface
double
stepEstimator(int kind,
              double* ATH_RESTRICT Su,
              const double* ATH_RESTRICT P,
              bool& Q);
double
stepEstimator(Trk::SurfaceType surfaceType,
              double* ATH_RESTRICT Su,
              const double* ATH_RESTRICT P,
              bool& Q);
// Step estimators to surfaces
std::pair<double, int>
stepEstimator(std::vector<std::pair<const Trk::Surface*, Trk::BoundaryCheck>>&,
              std::multimap<double, int>&,
              const double* ATH_RESTRICT,
              const double* ATH_RESTRICT,
              double,
              double,
              int Nv,
              bool&);
 
int
fillDistancesMap(
  std::vector<std::pair<const Trk::Surface*, Trk::BoundaryCheck>>&,
  std::multimap<double, int>&,
  const double* ATH_RESTRICT,
  double,
  const Trk::Surface*,
  double* ATH_RESTRICT);

// Transformations from local to global system coordinates
// for Trk::TrackParameters and Trk::NeutralParameters
bool
transformLocalToGlobal(bool, const Trk::TrackParameters&, double* ATH_RESTRICT);
bool
transformLocalToGlobal(bool, const Trk::NeutralParameters&, double* ATH_RESTRICT);

// Transformations from local to global system coordinates
bool
transformLocalToGlobal(bool,
                       const Trk::Surface*,
                       const AmgVector(5) & ATH_RESTRICT,
                       double* ATH_RESTRICT);

// Transformations from local to global system coordinates
// for Trk::PatternTrackParameters
bool
transformLocalToGlobal(bool, const Trk::PatternTrackParameters&, double*);

// Transformations from global to local system coordinates
void
transformGlobalToLocal(double* ATH_RESTRICT, double* ATH_RESTRICT);
void
transformGlobalToLocal(const Trk::Surface*,
                       bool,
                       double* ATH_RESTRICT,
                       double* ATH_RESTRICT,
                       double* ATH_RESTRICT);

// Covariance matrix production for Trk::TrackParameters
AmgSymMatrix(5)
newCovarianceMatrix(const double* ATH_RESTRICT, const AmgSymMatrix(5) & ATH_RESTRICT);

// Transformations from curvilinear to global system coordinates
// covariance matrix only
void
transformCurvilinearToGlobal(double* ATH_RESTRICT, double* ATH_RESTRICT);

// Transformations from global to curvilinear system coordinates
// covariance matrix only
void
transformGlobalToCurvilinear(bool,
                             double* ATH_RESTRICT,
                             double* ATH_RESTRICT,
                             double* ATH_RESTRICT);

// Jacobian of transformations from curvilinear to local system coordinates
void
jacobianTransformCurvilinearToLocal(const Trk::TrackParameters&, double* ATH_RESTRICT);
void
jacobianTransformCurvilinearToLocal(const Trk::PatternTrackParameters&,
                                    double* ATH_RESTRICT);
void
jacobianTransformCurvilinearToLocal(double* ATH_RESTRICT,
                                    const Trk::Surface*,
                                    double* ATH_RESTRICT);
 
}
}
 
#endif // RungeKuttaUtils_H