Matriplex::MatriplexSym< T, D, N > Class Template Reference

#include <MatriplexSym.h>

Public Types
typedef T	value_type

Public Member Functions
T fArray [kTotSize]	__attribute__ ((aligned(64)))
void	add (const MatriplexSym &v)
void	addNoiseIntoUpperLeft3x3 (T noise)
T &	At (idx_t n, idx_t i, idx_t j)
const T &	constAt (idx_t n, idx_t i, idx_t j) const
void	copy (idx_t n, idx_t in)
void	copyIn (idx_t n, const T *arr)
void	copyIn (idx_t n, const MatriplexSym &m, idx_t in)
void	copyOut (idx_t n, T *arr) const
void	copySlot (idx_t n, const MatriplexSym &m)
void	invertUpperLeft3x3 ()
	MatriplexSym ()
	MatriplexSym (T v)
idx_t	off (idx_t i) const
const idx_t *	offsets () const
T &	operator() (idx_t n, idx_t i, idx_t j)
const T &	operator() (idx_t n, idx_t i, idx_t j) const
MatriplexSym &	operator= (const MatriplexSym &m)
T	operator[] (idx_t xx) const
T &	operator[] (idx_t xx)
idx_t	plexSize () const
void	scale (T scale)
void	setDiagonal3x3 (idx_t n, T d)
void	setVal (T v)
void	slurpIn (const T *arr, int vi[N], const int N_proc=N)
MatriplexSym &	subtract (const MatriplexSym &a, const MatriplexSym &b)

Static Public Attributes
static constexpr int	kCols = D
	no. of matrix columns More...
static constexpr int	kRows = D
	no. of matrix rows More...
static constexpr int	kSize = (D + 1) * D / 2
	no of elements: lower triangle More...
static constexpr int	kTotSize = N * kSize
	size of the whole matriplex More...

Detailed Description

template<typename T, idx_t D, idx_t N>
class Matriplex::MatriplexSym< T, D, N >

Definition at line 25 of file MatriplexSym.h.

Member Typedef Documentation

value_type

template<typename T, idx_t D, idx_t N>

typedef T Matriplex::MatriplexSym< T, D, N >::value_type

Definition at line 27 of file MatriplexSym.h.

Constructor & Destructor Documentation

MatriplexSym() [1/2]

template<typename T, idx_t D, idx_t N>

Matriplex::MatriplexSym< T, D, N >::MatriplexSym ( )

inline

Definition at line 40 of file MatriplexSym.h.

{}

MatriplexSym() [2/2]

template<typename T, idx_t D, idx_t N>

Matriplex::MatriplexSym< T, D, N >::MatriplexSym ( T  v )

inline

Definition at line 41 of file MatriplexSym.h.

{ setVal(v); }

Member Function Documentation

__attribute__()

template<typename T, idx_t D, idx_t N>

T fArray [kTotSize] Matriplex::MatriplexSym< T, D, N >::__attribute__

(

(aligned(64))

)

add()

template<typename T, idx_t D, idx_t N>

void Matriplex::MatriplexSym< T, D, N >::add ( const MatriplexSym< T, D, N > &  v )

inline

Definition at line 51 of file MatriplexSym.h.

Referenced by counter.Counter::register().

                                    {
      for (idx_t i = 0; i < kTotSize; ++i) {
        fArray[i] += v.fArray[i];
      }
    }

addNoiseIntoUpperLeft3x3()

template<typename T, idx_t D, idx_t N>

void Matriplex::MatriplexSym< T, D, N >::addNoiseIntoUpperLeft3x3 ( T  noise )

inline

Definition at line 227 of file MatriplexSym.h.

                                           {
      T* p = fArray;
      ASSUME_ALIGNED(p, 64);

#pragma omp simd
      for (idx_t n = 0; n < N; ++n) {
        p[0 * N + n] += noise;
        p[2 * N + n] += noise;
        p[5 * N + n] += noise;
      }
    }

At()

template<typename T, idx_t D, idx_t N>

T& Matriplex::MatriplexSym< T, D, N >::At ( idx_t  n, idx_t  i, idx_t  j  )

inline

Definition at line 71 of file MatriplexSym.h.

Referenced by mkfit::applyMaterialEffects(), mkfit::MkFinder::bkFitFitTracksBH(), mkfit::conformalFitMPlex(), mkfit::kalmanOperation(), mkfit::kalmanOperationEndcap(), Matriplex::MatriplexSym< float, LL, NN >::operator()(), mkfit::propagateHelixToRMPlex(), and mkfit::MkFinder::selectHitIndices().

{ return fArray[off(i * D + j) * N + n]; }

constAt()

template<typename T, idx_t D, idx_t N>

const T& Matriplex::MatriplexSym< T, D, N >::constAt ( idx_t  n, idx_t  i, idx_t  j  ) const

inline

Definition at line 69 of file MatriplexSym.h.

Referenced by mkfit::conformalFitMPlex(), mkfit::isStripQCompatible(), mkfit::kalmanOperation(), mkfit::kalmanOperationEndcap(), Matriplex::MatriplexSym< float, LL, NN >::operator()(), mkfit::passStripChargePCMfromTrack(), mkfit::propagateHelixToZMPlex(), and mkfit::MkFinder::selectHitIndices().

{ return fArray[off(i * D + j) * N + n]; }

copy()

template<typename T, idx_t D, idx_t N>

void Matriplex::MatriplexSym< T, D, N >::copy ( idx_t  n, idx_t  in  )

inline

Definition at line 99 of file MatriplexSym.h.

                                 {
      for (idx_t i = n; i < kTotSize; i += N, in += N) {
        fArray[i] = fArray[in];
      }
    }

copyIn() [1/2]

template<typename T, idx_t D, idx_t N>

void Matriplex::MatriplexSym< T, D, N >::copyIn ( idx_t  n, const T *  arr  )

inline

Definition at line 87 of file MatriplexSym.h.

Referenced by mkfit::MkFinder::addBestHit(), mkfit::MkFinder::bkFitFitTracks(), mkfit::MkFinder::bkFitFitTracksBH(), mkfit::MkFinder::copy_in(), mkfit::MkFitter::inputSeedsTracksAndHits(), mkfit::MkFitter::inputTracksAndHitIdx(), mkfit::MkFitter::inputTracksAndHits(), mkfit::MkFinder::inputTracksAndHits(), and mkfit::MkFinder::selectHitIndices().

                                       {
      for (idx_t i = n; i < kTotSize; i += N) {
        fArray[i] = *(arr++);
      }
    }

copyIn() [2/2]

template<typename T, idx_t D, idx_t N>

void Matriplex::MatriplexSym< T, D, N >::copyIn ( idx_t  n, const MatriplexSym< T, D, N > &  m, idx_t  in  )

inline

Definition at line 93 of file MatriplexSym.h.

                                                          {
      for (idx_t i = n; i < kTotSize; i += N, in += N) {
        fArray[i] = m[in];
      }
    }

copyOut()

template<typename T, idx_t D, idx_t N>

void Matriplex::MatriplexSym< T, D, N >::copyOut ( idx_t  n, T *  arr  ) const

inline

Definition at line 195 of file MatriplexSym.h.

Referenced by mkfit::MkFinder::bkFitOutputTracks(), mkfit::MkFinder::copy_out(), mkfit::MkFinder::copyOutParErr(), mkfit::MkFitter::outputFittedTracksAndHitIdx(), and mkfit::MkFitter::outputTracks().

                                        {
      for (idx_t i = n; i < kTotSize; i += N) {
        *(arr++) = fArray[i];
      }
    }

copySlot()

template<typename T, idx_t D, idx_t N>

void Matriplex::MatriplexSym< T, D, N >::copySlot ( idx_t  n, const MatriplexSym< T, D, N > &  m  )

inline

Definition at line 81 of file MatriplexSym.h.

Referenced by mkfit::propagateHelixToRMPlex().

                                                  {
      for (idx_t i = n; i < kTotSize; i += N) {
        fArray[i] = m.fArray[i];
      }
    }

invertUpperLeft3x3()

template<typename T, idx_t D, idx_t N>

void Matriplex::MatriplexSym< T, D, N >::invertUpperLeft3x3 ( )

inline

Definition at line 239 of file MatriplexSym.h.

                              {
      typedef T TT;

      T* a = fArray;
      ASSUME_ALIGNED(a, 64);

#pragma omp simd
      for (idx_t n = 0; n < N; ++n) {
        const TT c00 = a[2 * N + n] * a[5 * N + n] - a[4 * N + n] * a[4 * N + n];
        const TT c01 = a[4 * N + n] * a[3 * N + n] - a[1 * N + n] * a[5 * N + n];
        const TT c02 = a[1 * N + n] * a[4 * N + n] - a[2 * N + n] * a[3 * N + n];
        const TT c11 = a[5 * N + n] * a[0 * N + n] - a[3 * N + n] * a[3 * N + n];
        const TT c12 = a[3 * N + n] * a[1 * N + n] - a[4 * N + n] * a[0 * N + n];
        const TT c22 = a[0 * N + n] * a[2 * N + n] - a[1 * N + n] * a[1 * N + n];

        // Force determinant calculation in double precision.
        const double det = (double)a[0 * N + n] * c00 + (double)a[1 * N + n] * c01 + (double)a[3 * N + n] * c02;
        const TT s = TT(1) / det;

        a[0 * N + n] = s * c00;
        a[1 * N + n] = s * c01;
        a[2 * N + n] = s * c11;
        a[3 * N + n] = s * c02;
        a[4 * N + n] = s * c12;
        a[5 * N + n] = s * c22;
      }
    }

off()

template<typename T, idx_t D, idx_t N>

idx_t Matriplex::MatriplexSym< T, D, N >::off ( idx_t  i ) const

inline

Definition at line 67 of file MatriplexSym.h.

Referenced by Matriplex::MatriplexSym< float, LL, NN >::At(), and Matriplex::MatriplexSym< float, LL, NN >::constAt().

{ return gSymOffsets[D][i]; }

offsets()

template<typename T, idx_t D, idx_t N>

const idx_t* Matriplex::MatriplexSym< T, D, N >::offsets ( ) const

inline

Definition at line 66 of file MatriplexSym.h.

{ return gSymOffsets[D]; }

operator()() [1/2]

template<typename T, idx_t D, idx_t N>

T& Matriplex::MatriplexSym< T, D, N >::operator() ( idx_t  n, idx_t  i, idx_t  j  )

inline

Definition at line 73 of file MatriplexSym.h.

{ return At(n, i, j); }

operator()() [2/2]

template<typename T, idx_t D, idx_t N>

const T& Matriplex::MatriplexSym< T, D, N >::operator() ( idx_t  n, idx_t  i, idx_t  j  ) const

inline

Definition at line 74 of file MatriplexSym.h.

{ return constAt(n, i, j); }

operator=()

template<typename T, idx_t D, idx_t N>

MatriplexSym& Matriplex::MatriplexSym< T, D, N >::operator= ( const MatriplexSym< T, D, N > &  m )

inline

Definition at line 76 of file MatriplexSym.h.

                                                   {
      memcpy(fArray, m.fArray, sizeof(T) * kTotSize);
      return *this;
    }

operator[]() [1/2]

template<typename T, idx_t D, idx_t N>

T Matriplex::MatriplexSym< T, D, N >::operator[] ( idx_t  xx ) const

inline

Definition at line 63 of file MatriplexSym.h.

{ return fArray[xx]; }

operator[]() [2/2]

template<typename T, idx_t D, idx_t N>

T& Matriplex::MatriplexSym< T, D, N >::operator[] ( idx_t  xx )

inline

Definition at line 64 of file MatriplexSym.h.

{ return fArray[xx]; }

plexSize()

template<typename T, idx_t D, idx_t N>

idx_t Matriplex::MatriplexSym< T, D, N >::plexSize ( ) const

inline

Definition at line 43 of file MatriplexSym.h.

{ return N; }

scale()

template<typename T, idx_t D, idx_t N>

void Matriplex::MatriplexSym< T, D, N >::scale ( T  scale )

inline

Definition at line 57 of file MatriplexSym.h.

Referenced by mkfit::MkFinder::bkFitInputTracks(), and Matriplex::MatriplexSym< float, LL, NN >::scale().

                        {
      for (idx_t i = 0; i < kTotSize; ++i) {
        fArray[i] *= scale;
      }
    }

setDiagonal3x3()

template<typename T, idx_t D, idx_t N>

void Matriplex::MatriplexSym< T, D, N >::setDiagonal3x3 ( idx_t  n, T  d  )

inline

Definition at line 201 of file MatriplexSym.h.

Referenced by mkfit::MkFinder::addBestHit().

                                      {
      T* p = fArray + n;

      p[0 * N] = d;
      p[1 * N] = 0;
      p[2 * N] = d;
      p[3 * N] = 0;
      p[4 * N] = 0;
      p[5 * N] = d;
    }

setVal()

template<typename T, idx_t D, idx_t N>

void Matriplex::MatriplexSym< T, D, N >::setVal ( T  v )

inline

Definition at line 45 of file MatriplexSym.h.

Referenced by Matriplex::MatriplexSym< float, LL, NN >::MatriplexSym().

                     {
      for (idx_t i = 0; i < kTotSize; ++i) {
        fArray[i] = v;
      }
    }

slurpIn()

template<typename T, idx_t D, idx_t N>

void Matriplex::MatriplexSym< T, D, N >::slurpIn ( const T *  arr, int  vi[N], const int  N_proc = N  )

inline

Definition at line 176 of file MatriplexSym.h.

                                                                {
      // Separate N_proc == N case (gains about 7% in fit test).
      if (N_proc == N) {
        for (int i = 0; i < kSize; ++i) {
          for (int j = 0; j < N; ++j) {
            fArray[i * N + j] = *(arr + i + vi[j]);
          }
        }
      } else {
        for (int i = 0; i < kSize; ++i) {
          for (int j = 0; j < N_proc; ++j) {
            fArray[i * N + j] = *(arr + i + vi[j]);
          }
        }
      }
    }

subtract()

template<typename T, idx_t D, idx_t N>

MatriplexSym& Matriplex::MatriplexSym< T, D, N >::subtract ( const MatriplexSym< T, D, N > &  a, const MatriplexSym< T, D, N > &  b  )

inline

Definition at line 212 of file MatriplexSym.h.

Referenced by mkfit::kalmanOperation(), and mkfit::kalmanOperationEndcap().

                                                                         {
      // Does *this = a - b;

#pragma omp simd
      for (idx_t i = 0; i < kTotSize; ++i) {
        fArray[i] = a.fArray[i] - b.fArray[i];
      }

      return *this;
    }

Member Data Documentation

kCols

template<typename T, idx_t D, idx_t N>

constexpr int Matriplex::MatriplexSym< T, D, N >::kCols = D

static

no. of matrix columns

Definition at line 32 of file MatriplexSym.h.

kRows

template<typename T, idx_t D, idx_t N>

constexpr int Matriplex::MatriplexSym< T, D, N >::kRows = D

static

no. of matrix rows

Definition at line 30 of file MatriplexSym.h.

kSize

template<typename T, idx_t D, idx_t N>

constexpr int Matriplex::MatriplexSym< T, D, N >::kSize = (D + 1) * D / 2

static

no of elements: lower triangle

Definition at line 34 of file MatriplexSym.h.

Referenced by Matriplex::MatriplexSym< float, LL, NN >::slurpIn().

kTotSize

template<typename T, idx_t D, idx_t N>

constexpr int Matriplex::MatriplexSym< T, D, N >::kTotSize = N * kSize

static

size of the whole matriplex

Definition at line 36 of file MatriplexSym.h.

Referenced by Matriplex::MatriplexSym< float, LL, NN >::add(), Matriplex::MatriplexSym< float, LL, NN >::copy(), Matriplex::MatriplexSym< float, LL, NN >::copyIn(), Matriplex::MatriplexSym< float, LL, NN >::copyOut(), Matriplex::MatriplexSym< float, LL, NN >::copySlot(), Matriplex::MatriplexSym< float, LL, NN >::operator=(), Matriplex::MatriplexSym< float, LL, NN >::scale(), Matriplex::MatriplexSym< float, LL, NN >::setVal(), and Matriplex::MatriplexSym< float, LL, NN >::subtract().