252 static size_t hash(
float x,
float alp,
float y,
float z,
float snp,
float tgl,
float q2pt);
263 static constexpr
value_t InvalidX = -99999.
f;
269 uint16_t mUserField = 0;
271 ClassDefNV(TrackParametrization, 3);
275template <
typename value_T>
280 math_utils::detail::bringToPMPi<value_t>(mAlpha);
287template <
typename value_T>
288GPUdi()
void TrackParametrization<value_T>::set(value_t
x, value_t
alpha, const params_t& par,
int charge, const
PID pid)
294template <
typename value_T>
295GPUdi()
void TrackParametrization<value_T>::set(value_t
x, value_t
alpha, const value_t* par,
int charge, const
PID pid)
299 math_utils::detail::bringToPMPi<value_t>(mAlpha);
308template <
typename value_T>
309GPUdi() auto TrackParametrization<value_T>::getParams() const -> const value_t*
315template <
typename value_T>
316GPUdi() auto TrackParametrization<value_T>::getParam(
int i) const -> value_t
322template <
typename value_T>
323GPUdi() auto TrackParametrization<value_T>::getX() const -> value_t
329template <
typename value_T>
330GPUdi() auto TrackParametrization<value_T>::getAlpha() const -> value_t
336template <
typename value_T>
337GPUdi() auto TrackParametrization<value_T>::getY() const -> value_t
343template <
typename value_T>
344GPUdi() auto TrackParametrization<value_T>::getZ() const -> value_t
350template <
typename value_T>
351GPUdi() auto TrackParametrization<value_T>::getSnp() const -> value_t
357template <
typename value_T>
358GPUdi() auto TrackParametrization<value_T>::getTgl() const -> value_t
364template <
typename value_T>
365GPUhdi() auto TrackParametrization<value_T>::getQ2Pt() const -> value_t
371template <
typename value_T>
372GPUdi() auto TrackParametrization<value_T>::getCharge2Pt() const -> value_t
374 return mAbsCharge ? mP[
kQ2Pt] : 0.f;
378template <
typename value_T>
379GPUdi()
int TrackParametrization<value_T>::getAbsCharge()
const
385template <
typename value_T>
386GPUdi()
PID TrackParametrization<value_T>::getPID()
const
392template <
typename value_T>
393GPUdi()
void TrackParametrization<value_T>::setPID(const
PID pid,
bool passCharge)
397 setAbsCharge(
pid.getCharge());
402template <
typename value_T>
403GPUdi() auto TrackParametrization<value_T>::getCsp2() const -> value_t
410template <
typename value_T>
411GPUdi() auto TrackParametrization<value_T>::getCsp() const -> value_t
413 return gpu::CAMath::Sqrt(getCsp2());
417template <
typename value_T>
418GPUdi()
void TrackParametrization<value_T>::setX(value_t
v)
424template <
typename value_T>
425GPUdi()
void TrackParametrization<value_T>::setParam(value_t
v,
int i)
431template <
typename value_T>
432GPUdi()
void TrackParametrization<value_T>::setAlpha(value_t
v)
435 math_utils::detail::bringToPMPi<value_t>(mAlpha);
439template <
typename value_T>
440GPUdi()
void TrackParametrization<value_T>::setY(value_t
v)
446template <
typename value_T>
447GPUdi()
void TrackParametrization<value_T>::setZ(value_t
v)
453template <
typename value_T>
454GPUdi()
void TrackParametrization<value_T>::setSnp(value_t
v)
460template <
typename value_T>
461GPUdi()
void TrackParametrization<value_T>::setTgl(value_t
v)
467template <
typename value_T>
468GPUdi()
void TrackParametrization<value_T>::setQ2Pt(value_t
v)
474template <
typename value_T>
475GPUdi()
void TrackParametrization<value_T>::setAbsCharge(
int q)
477 mAbsCharge = gpu::CAMath::Abs(q);
481template <
typename value_T>
482GPUdi()
void TrackParametrization<value_T>::getCircleParamsLoc(value_t bz,
o2::math_utils::CircleXY<value_t>&
c)
const
485 c.rC = getCurvature(bz);
487 constexpr value_t MinSagitta = 0.01f, TPCMidR = 160.f, MinCurv = 8 * MinSagitta / (TPCMidR * TPCMidR);
488 if (gpu::CAMath::Abs(
c.rC) > MinCurv) {
489 c.rC = 1.f / getCurvature(bz);
490 value_t sn = getSnp(), cs = gpu::CAMath::Sqrt((1.f - sn) * (1.f + sn));
491 c.xC =
getX() - sn *
c.rC;
492 c.yC =
getY() + cs *
c.rC;
493 c.rC = gpu::CAMath::Abs(
c.rC);
502template <
typename value_T>
503GPUdi()
void TrackParametrization<value_T>::getCircleParams(value_t bz,
o2::math_utils::CircleXY<value_t>&
c, value_t& sna, value_t& csa)
const
506 getCircleParamsLoc(bz,
c);
507 o2::math_utils::detail::sincos(getAlpha(), sna, csa);
508 o2::math_utils::detail::rotateZ<value_t>(
c.xC,
c.yC,
c.xC,
c.yC, sna, csa);
512template <
typename value_T>
513GPUdi()
void TrackParametrization<value_T>::getLineParams(
o2::math_utils::IntervalXY<value_t>& ln, value_t& sna, value_t& csa)
const
516 o2::math_utils::detail::sincos(getAlpha(), sna, csa);
517 o2::math_utils::detail::rotateZ<value_t>(
getX(),
getY(), ln.getX0(), ln.getY0(), sna, csa);
518 value_t snp = getSnp(), csp = gpu::CAMath::Sqrt((1.f - snp) * (1.f + snp));
519 ln.setDX(csp * csa - snp * sna);
520 ln.setDY(snp * csa + csp * sna);
524template <
typename value_T>
525GPUdi() auto TrackParametrization<value_T>::getCurvature(value_t
b) const -> value_t
531template <
typename value_T>
532GPUdi()
int TrackParametrization<value_T>::getCharge()
const
534 return getSign() > 0 ? mAbsCharge : -mAbsCharge;
538template <
typename value_T>
539GPUdi()
int TrackParametrization<value_T>::getSign()
const
541 return mAbsCharge ? (mP[
kQ2Pt] > 0.f ? 1 : -1) : 0;
545template <
typename value_T>
546GPUdi() auto TrackParametrization<value_T>::getPhi() const -> value_t
549 value_t phi = gpu::CAMath::ASin(getSnp()) + getAlpha();
550 math_utils::detail::bringTo02Pi<value_t>(phi);
555template <
typename value_T>
556GPUdi() auto TrackParametrization<value_T>::getPhiPos() const -> value_t
560 math_utils::detail::bringTo02Pi<value_t>(phi);
565template <
typename value_T>
566GPUdi() auto TrackParametrization<value_T>::getQ2P2() const -> value_t
573 return q2pt2 / (1.f + getTgl() * getTgl());
577template <
typename value_T>
578GPUdi() auto TrackParametrization<value_T>::getPtInv() const -> value_t
585 return (mAbsCharge > 1) ? ptInv / mAbsCharge : ptInv;
589template <
typename value_T>
590GPUdi() auto TrackParametrization<value_T>::getP2Inv() const -> value_t
594 return p2 *
p2 / (1.f + getTgl() * getTgl());
598template <
typename value_T>
599GPUdi() auto TrackParametrization<value_T>::getP2() const -> value_t
602 return 1.f / getP2Inv();
606template <
typename value_T>
607GPUdi() auto TrackParametrization<value_T>::getPInv() const -> value_t
610 return getPtInv() / gpu::CAMath::Sqrt(1.f + getTgl() * getTgl());
614template <
typename value_T>
615GPUdi() auto TrackParametrization<value_T>::getP() const -> value_t
618 return 1.f / getPInv();
622template <
typename value_T>
623GPUdi() auto TrackParametrization<value_T>::getE2() const -> value_t
626 return getP2() + getPID().getMass2();
630template <
typename value_T>
631GPUdi() auto TrackParametrization<value_T>::getE() const -> value_t
634 return gpu::CAMath::Sqrt(getE2());
638template <
typename value_T>
639GPUdi() auto TrackParametrization<value_T>::getPt() const -> value_t
642 return 1.f / getPtInv();
646template <
typename value_T>
647GPUdi() auto TrackParametrization<value_T>::getTheta() const -> value_t
653template <
typename value_T>
654GPUdi() auto TrackParametrization<value_T>::getEta() const -> value_t
656 return -gpu::CAMath::Log(gpu::CAMath::Tan(0.5f * getTheta()));
660template <
typename value_T>
661GPUdi() auto TrackParametrization<value_T>::getXYZGlo() const -> math_utils::
Point3D<value_t>
663#ifndef GPUCA_ALIGPUCODE
664 return math_utils::Rotation2D<value_t>(getAlpha())(math_utils::Point3D<value_t>(
getX(),
getY(), getZ()));
667 gpu::CAMath::SinCos(getAlpha(), sina, cosa);
668 return math_utils::Point3D<value_t>(cosa *
getX() - sina *
getY(), cosa *
getY() + sina *
getX(), getZ());
673template <
typename value_T>
674GPUdi()
void TrackParametrization<value_T>::getXYZGlo(dim3_t& xyz)
const
680 math_utils::detail::rotateZ<value_t>(xyz, getAlpha());
684template <
typename value_T>
685GPUdi() auto TrackParametrization<value_T>::getXYZGloAt(value_t xk, value_t
b,
bool& ok) const -> math_utils::
Point3D<value_t>
691 ok = getYZAt(xk,
b,
y,
z);
693#ifndef GPUCA_ALIGPUCODE
694 return math_utils::Rotation2D<value_t>(getAlpha())(math_utils::Point3D<value_t>(xk,
y,
z));
697 gpu::CAMath::SinCos(getAlpha(), sina, cosa);
698 return math_utils::Point3D<value_t>(cosa * xk - sina *
y, cosa *
y + sina * xk,
z);
701 return math_utils::Point3D<value_t>();
706template <
typename value_T>
707GPUdi() bool TrackParametrization<value_T>::
isValid()
const
709 return mX != InvalidX;
713template <
typename value_T>
714GPUdi()
void TrackParametrization<value_T>::invalidate()
719template <
typename value_T>
725template <
typename value_T>
726GPUhdi()
void TrackParametrization<value_T>::setUserField(uint16_t
v)
732template <
typename value_T>
733GPUdi()
void TrackParametrization<value_T>::updateParam(value_t delta,
int i)
739template <
typename value_T>
740GPUdi()
void TrackParametrization<value_T>::updateParams(const params_t& delta)
742 updateParams(delta.data());
746template <
typename value_T>
747GPUdi()
void TrackParametrization<value_T>::updateParams(const value_t* delta)
760#ifndef GPUCA_ALIGPUCODE
761template <
typename value_T>
764 size_t h = std::hash<float>{}(o2::math_utils::detail::truncateFloatFraction(
x, 0xFFFFFFF0));
765 h ^= std::hash<float>{}(o2::math_utils::detail::truncateFloatFraction(alp, 0xFFFFFFF0)) << 1;
766 h ^= std::hash<float>{}(o2::math_utils::detail::truncateFloatFraction(
y, 0xFFFFFFF0)) << 1;
767 h ^= std::hash<float>{}(o2::math_utils::detail::truncateFloatFraction(
z, 0xFFFFFFF0)) << 1;
768 h ^= std::hash<float>{}(o2::math_utils::detail::truncateFloatFraction(snp, 0xFFFFFF00)) << 1;
769 h ^= std::hash<float>{}(o2::math_utils::detail::truncateFloatFraction(tgl, 0xFFFFFF00)) << 1;
770 h ^= std::hash<float>{}(o2::math_utils::detail::truncateFloatFraction(q2pt, 0xFFFFFC00)) << 1;