TracksUtilities< TrackerTraits > Struct Template Reference

#include <PixelTrackUtilities.h>

Public Types
using	hindex_type = typename reco::TrackSoA< TrackerTraits >::hindex_type
using	hindex_type = typename TrackSoA< TrackerTraits >::hindex_type
using	TrackSoAConstView = typename reco::TrackSoA< TrackerTraits >::template Layout<>::ConstView
using	TrackSoAConstView = typename TrackSoA< TrackerTraits >::template TrackSoALayout<>::ConstView
using	TrackSoAView = typename reco::TrackSoA< TrackerTraits >::template Layout<>::View
using	TrackSoAView = typename TrackSoA< TrackerTraits >::template TrackSoALayout<>::View

Static Public Member Functions
static __host__ __device__ float	charge (const TrackSoAConstView &tracks, int32_t i)
ALPAKA_FN_HOST_ACC static ALPAKA_FN_INLINE constexpr int	computeNumberOfLayers (const TrackSoAConstView &tracks, int32_t i)
static constexpr __host__ __device__ int	computeNumberOfLayers (const TrackSoAConstView &tracks, int32_t i)
template<typename V3 , typename M3 , typename V2 , typename M2 >
ALPAKA_FN_HOST_ACC static ALPAKA_FN_INLINE constexpr void	copyFromCircle (TrackSoAView &tracks, V3 const &cp, M3 const &ccov, V2 const &lp, M2 const &lcov, float b, int32_t i)
template<typename V3 , typename M3 , typename V2 , typename M2 >
static constexpr __host__ __device__ void	copyFromCircle (TrackSoAView &tracks, V3 const &cp, M3 const &ccov, V2 const &lp, M2 const &lcov, float b, int32_t i)
template<typename V5 , typename M5 >
ALPAKA_FN_HOST_ACC static ALPAKA_FN_INLINE constexpr void	copyFromDense (TrackSoAView &tracks, V5 const &v, M5 const &cov, int32_t i)
template<typename V5 , typename M5 >
static constexpr __host__ __device__ void	copyFromDense (TrackSoAView &tracks, V5 const &v, M5 const &cov, int32_t i)
template<typename V5 , typename M5 >
ALPAKA_FN_HOST_ACC static ALPAKA_FN_INLINE constexpr void	copyToDense (const TrackSoAConstView &tracks, V5 &v, M5 &cov, int32_t i)
template<typename V5 , typename M5 >
static constexpr __host__ __device__ void	copyToDense (const TrackSoAConstView &tracks, V5 &v, M5 &cov, int32_t i)
static constexpr __host__ __device__ bool	isTriplet (const TrackSoAConstView &tracks, int i)
ALPAKA_FN_HOST_ACC static ALPAKA_FN_INLINE constexpr int	nHits (const TrackSoAConstView &tracks, int i)
static constexpr __host__ __device__ int	nHits (const TrackSoAConstView &tracks, int i)
static constexpr __host__ __device__ float	phi (const TrackSoAConstView &tracks, int32_t i)
static constexpr __host__ __device__ float	tip (const TrackSoAConstView &tracks, int32_t i)
static constexpr __host__ __device__ float	zip (const TrackSoAConstView &tracks, int32_t i)

Detailed Description

template<typename TrackerTraits>
struct TracksUtilities< TrackerTraits >

Definition at line 51 of file PixelTrackUtilities.h.

Member Typedef Documentation

hindex_type [1/2]

template<typename TrackerTraits>

using TracksUtilities< TrackerTraits >::hindex_type = typename reco::TrackSoA<TrackerTraits>::hindex_type

Definition at line 13 of file TrackUtilities.h.

hindex_type [2/2]

template<typename TrackerTraits>

using TracksUtilities< TrackerTraits >::hindex_type = typename TrackSoA<TrackerTraits>::hindex_type

Definition at line 54 of file PixelTrackUtilities.h.

TrackSoAConstView [1/2]

template<typename TrackerTraits>

using TracksUtilities< TrackerTraits >::TrackSoAConstView = typename reco::TrackSoA<TrackerTraits>::template Layout<>::ConstView

Definition at line 12 of file TrackUtilities.h.

TrackSoAConstView [2/2]

template<typename TrackerTraits>

using TracksUtilities< TrackerTraits >::TrackSoAConstView = typename TrackSoA<TrackerTraits>::template TrackSoALayout<>::ConstView

Definition at line 53 of file PixelTrackUtilities.h.

TrackSoAView [1/2]

template<typename TrackerTraits>

using TracksUtilities< TrackerTraits >::TrackSoAView = typename reco::TrackSoA<TrackerTraits>::template Layout<>::View

Definition at line 11 of file TrackUtilities.h.

TrackSoAView [2/2]

template<typename TrackerTraits>

using TracksUtilities< TrackerTraits >::TrackSoAView = typename TrackSoA<TrackerTraits>::template TrackSoALayout<>::View

Definition at line 52 of file PixelTrackUtilities.h.

Member Function Documentation

charge()

template<typename TrackerTraits>

static __host__ __device__ float TracksUtilities< TrackerTraits >::charge ( const TrackSoAConstView &  tracks, int32_t  i  )

inlinestatic

Definition at line 58 of file PixelTrackUtilities.h.

References f, mps_fire::i, and DiMuonV_cfg::tracks.

                                                                                             {
    return std::copysign(1.f, tracks[i].state()(2));

computeNumberOfLayers() [1/2]

template<typename TrackerTraits>

ALPAKA_FN_HOST_ACC static ALPAKA_FN_INLINE constexpr int TracksUtilities< TrackerTraits >::computeNumberOfLayers ( const TrackSoAConstView &  tracks, int32_t  i  )

inlinestatic

Definition at line 62 of file TrackUtilities.h.

References mps_fire::i, and DiMuonV_cfg::tracks.

                                                                                            {
    auto pdet = tracks.detIndices().begin(i);
    int nl = 1;
    auto ol = pixelTopology::getLayer<TrackerTraits>(*pdet);
    for (; pdet < tracks.detIndices().end(i); ++pdet) {
      auto il = pixelTopology::getLayer<TrackerTraits>(*pdet);
      if (il != ol)
        ++nl;
      ol = il;
    }
    return nl;
  }

computeNumberOfLayers() [2/2]

template<typename TrackerTraits>

static constexpr __host__ __device__ int TracksUtilities< TrackerTraits >::computeNumberOfLayers ( const TrackSoAConstView &  tracks, int32_t  i  )

inlinestatic

Definition at line 123 of file PixelTrackUtilities.h.

References mps_fire::i, and DiMuonV_cfg::tracks.

Referenced by ALPAKA_ACCELERATOR_NAMESPACE::caHitNtupletGeneratorKernels::Kernel_fillNLayers< TrackerTraits >::operator()().

                                                                                                                    {
    auto pdet = tracks.detIndices().begin(i);
    int nl = 1;
    auto ol = pixelTopology::getLayer<TrackerTraits>(*pdet);
    for (; pdet < tracks.detIndices().end(i); ++pdet) {
      auto il = pixelTopology::getLayer<TrackerTraits>(*pdet);
      if (il != ol)
        ++nl;
      ol = il;
    }
    return nl;

copyFromCircle() [1/2]

template<typename TrackerTraits>

template<typename V3 , typename M3 , typename V2 , typename M2 >

ALPAKA_FN_HOST_ACC static ALPAKA_FN_INLINE constexpr void TracksUtilities< TrackerTraits >::copyFromCircle ( TrackSoAView &  tracks, V3 const &  cp, M3 const &  ccov, V2 const &  lp, M2 const &  lcov, float  b, int32_t  i  )

inlinestatic

Definition at line 18 of file TrackUtilities.h.

References b, nano_mu_digi_cff::float, mps_fire::i, and DiMuonV_cfg::tracks.

                                                                                                            {
    tracks[i].state() << cp.template cast<float>(), lp.template cast<float>();

    tracks[i].state()(2) = tracks[i].state()(2) * b;
    auto cov = tracks[i].covariance();
    cov(0) = ccov(0, 0);
    cov(1) = ccov(0, 1);
    cov(2) = b * float(ccov(0, 2));
    cov(4) = cov(3) = 0;
    cov(5) = ccov(1, 1);
    cov(6) = b * float(ccov(1, 2));
    cov(8) = cov(7) = 0;
    cov(9) = b * b * float(ccov(2, 2));
    cov(11) = cov(10) = 0;
    cov(12) = lcov(0, 0);
    cov(13) = lcov(0, 1);
    cov(14) = lcov(1, 1);
  }

copyFromCircle() [2/2]

template<typename TrackerTraits>

template<typename V3 , typename M3 , typename V2 , typename M2 >

static constexpr __host__ __device__ void TracksUtilities< TrackerTraits >::copyFromCircle ( TrackSoAView &  tracks, V3 const &  cp, M3 const &  ccov, V2 const &  lp, M2 const &  lcov, float  b, int32_t  i  )

inlinestatic

Definition at line 79 of file PixelTrackUtilities.h.

References b, nano_mu_digi_cff::float, mps_fire::i, and DiMuonV_cfg::tracks.

Referenced by ALPAKA_ACCELERATOR_NAMESPACE::Kernel_LineFit< N, TrackerTraits >::operator()(), and ALPAKA_ACCELERATOR_NAMESPACE::Kernel_BLFit< N, TrackerTraits >::operator()().

                                                                                                            {
    tracks[i].state() << cp.template cast<float>(), lp.template cast<float>();

    tracks[i].state()(2) = tracks[i].state()(2) * b;
    auto cov = tracks[i].covariance();
    cov(0) = ccov(0, 0);
    cov(1) = ccov(0, 1);
    cov(2) = b * float(ccov(0, 2));
    cov(4) = cov(3) = 0;
    cov(5) = ccov(1, 1);
    cov(6) = b * float(ccov(1, 2));
    cov(8) = cov(7) = 0;
    cov(9) = b * b * float(ccov(2, 2));
    cov(11) = cov(10) = 0;
    cov(12) = lcov(0, 0);
    cov(13) = lcov(0, 1);
    cov(14) = lcov(1, 1);

copyFromDense() [1/2]

template<typename TrackerTraits>

template<typename V5 , typename M5 >

ALPAKA_FN_HOST_ACC static ALPAKA_FN_INLINE constexpr void TracksUtilities< TrackerTraits >::copyFromDense ( TrackSoAView &  tracks, V5 const &  v, M5 const &  cov, int32_t  i  )

inlinestatic

Definition at line 39 of file TrackUtilities.h.

References mps_fire::i, dqmiolumiharvest::j, dqmdumpme::k, DiMuonV_cfg::tracks, and findQualityFiles::v.

                                                                                     {
    tracks[i].state() = v.template cast<float>();
    for (int j = 0, ind = 0; j < 5; ++j)
      for (auto k = j; k < 5; ++k)
        tracks[i].covariance()(ind++) = cov(j, k);
  }

copyFromDense() [2/2]

template<typename TrackerTraits>

template<typename V5 , typename M5 >

static constexpr __host__ __device__ void TracksUtilities< TrackerTraits >::copyFromDense ( TrackSoAView &  tracks, V5 const &  v, M5 const &  cov, int32_t  i  )

inlinestatic

Definition at line 100 of file PixelTrackUtilities.h.

References mps_fire::i, dqmiolumiharvest::j, dqmdumpme::k, DiMuonV_cfg::tracks, and findQualityFiles::v.

                                                                            {
    tracks[i].state() = v.template cast<float>();
    for (int j = 0, ind = 0; j < 5; ++j)
      for (auto k = j; k < 5; ++k)
        tracks[i].covariance()(ind++) = cov(j, k);

copyToDense() [1/2]

template<typename TrackerTraits>

template<typename V5 , typename M5 >

ALPAKA_FN_HOST_ACC static ALPAKA_FN_INLINE constexpr void TracksUtilities< TrackerTraits >::copyToDense ( const TrackSoAConstView &  tracks, V5 &  v, M5 &  cov, int32_t  i  )

inlinestatic

Definition at line 50 of file TrackUtilities.h.

References mps_fire::i, dqmiolumiharvest::j, dqmdumpme::k, DiMuonV_cfg::tracks, and findQualityFiles::v.

                                                                                   {
    v = tracks[i].state().template cast<typename V5::Scalar>();
    for (int j = 0, ind = 0; j < 5; ++j) {
      cov(j, j) = tracks[i].covariance()(ind++);
      for (auto k = j + 1; k < 5; ++k)
        cov(k, j) = cov(j, k) = tracks[i].covariance()(ind++);
    }
  }

copyToDense() [2/2]

template<typename TrackerTraits>

template<typename V5 , typename M5 >

static constexpr __host__ __device__ void TracksUtilities< TrackerTraits >::copyToDense ( const TrackSoAConstView &  tracks, V5 &  v, M5 &  cov, int32_t  i  )

inlinestatic

Definition at line 111 of file PixelTrackUtilities.h.

References mps_fire::i, dqmiolumiharvest::j, dqmdumpme::k, DiMuonV_cfg::tracks, and findQualityFiles::v.

Referenced by L2TauNNProducerAlpaka::impactParameter(), and L2TauNNProducer::impactParameter().

                                                                          {
    v = tracks[i].state().template cast<typename V5::Scalar>();
    for (int j = 0, ind = 0; j < 5; ++j) {
      cov(j, j) = tracks[i].covariance()(ind++);
      for (auto k = j + 1; k < 5; ++k)
        cov(k, j) = cov(j, k) = tracks[i].covariance()(ind++);
    }

isTriplet()

template<typename TrackerTraits>

static constexpr __host__ __device__ bool TracksUtilities< TrackerTraits >::isTriplet ( const TrackSoAConstView &  tracks, int  i  )

inlinestatic

Definition at line 74 of file PixelTrackUtilities.h.

References mps_fire::i, and DiMuonV_cfg::tracks.

Referenced by gpuVertexFinder::for().

                                                                                                     {
    return tracks[i].nLayers() == 3;

nHits() [1/2]

template<typename TrackerTraits>

ALPAKA_FN_HOST_ACC static ALPAKA_FN_INLINE constexpr int TracksUtilities< TrackerTraits >::nHits ( const TrackSoAConstView &  tracks, int  i  )

inlinestatic

Definition at line 76 of file TrackUtilities.h.

References mps_fire::i, and DiMuonV_cfg::tracks.

                                                                                                         {
    return tracks.detIndices().size(i);
  }

nHits() [2/2]

template<typename TrackerTraits>

static constexpr __host__ __device__ int TracksUtilities< TrackerTraits >::nHits ( const TrackSoAConstView &  tracks, int  i  )

inlinestatic

Definition at line 136 of file PixelTrackUtilities.h.

References mps_fire::i, and DiMuonV_cfg::tracks.

Referenced by gpuVertexFinder::for(), PixelTrackSoAFromCUDAT< TrackerTraits >::produce(), and PixelVertexProducerCUDAT< TrackerTraits >::produceOnCPU().

                                                                                                {
    return tracks.detIndices().size(i);

phi()

template<typename TrackerTraits>

static constexpr __host__ __device__ float TracksUtilities< TrackerTraits >::phi ( const TrackSoAConstView &  tracks, int32_t  i  )

inlinestatic

Definition at line 62 of file PixelTrackUtilities.h.

References mps_fire::i, and DiMuonV_cfg::tracks.

Referenced by Particle.Particle::__str__(), datamodel.Object::DeltaR(), datamodel.Object::p4(), and ntupleDataFormat.Track::phiPull().

                                                                                                    {
    return tracks[i].state()(0);

tip()

template<typename TrackerTraits>

static constexpr __host__ __device__ float TracksUtilities< TrackerTraits >::tip ( const TrackSoAConstView &  tracks, int32_t  i  )

inlinestatic

Definition at line 66 of file PixelTrackUtilities.h.

References mps_fire::i, and DiMuonV_cfg::tracks.

                                                                                                    {
    return tracks[i].state()(1);

zip()

template<typename TrackerTraits>

static constexpr __host__ __device__ float TracksUtilities< TrackerTraits >::zip ( const TrackSoAConstView &  tracks, int32_t  i  )

inlinestatic

Definition at line 70 of file PixelTrackUtilities.h.

References mps_fire::i, and DiMuonV_cfg::tracks.

Referenced by gpuVertexFinder::for().

                                                                                                    {
    return tracks[i].state()(4);