matrix_util.h

Go to the documentation of this file.

#ifndef MATRIX_UTIL_H
#define MATRIX_UTIL_H

// utility functions to fill with random data

#ifndef USE_VC

template<class V>
void fillRandomVec(TRandom & r, V  & v, unsigned int len, unsigned int start = 0, double offset = 1) {
  for(unsigned int i = start; i < len+start; ++i)
      v[i] = r.Rndm() + offset;
}

template<class M>
void fillRandomMat(TRandom & r, M  & m, unsigned int first, unsigned int second, unsigned int start = 0, double offset = 1) {
  for(unsigned int i = start; i < first+start; ++i)
    for(unsigned int j = start; j < second+start; ++j)
      m(i,j) = r.Rndm() + offset;
}

template<class M>
void fillRandomSym(TRandom & r, M  & m, unsigned int first, unsigned int start = 0, double offset = 1) {
  for(unsigned int i = start; i < first+start; ++i) {
    for(unsigned int j = i; j < first+start; ++j) {
      if ( i != j ) {
         m(i,j) = r.Rndm() + offset;
         m(j,i) = m(i,j);
      }
       else // add extra offset to make no singular when inverting
       m(i,i) = r.Rndm() + 3*offset;
    }
  }
}

#else

// case of Vc
template<class V>
void fillRandomVec(TRandom & r, V  & v, unsigned int len, unsigned int start = 0, double offset = 1) {
   for (int k = 0; k < Vc::double_v::Size; ++k) {
      for(unsigned int i = start; i < len+start; ++i) {
         typename V::value_type x = v[i];
         x[k] = r.Rndm() + offset;
         v[i] = x;
      }
   }
}

template<class M>
void fillRandomMat(TRandom & r, M  & m, unsigned int first, unsigned int second, unsigned int start = 0, double offset = 1) {
   for (int k = 0; k < Vc::double_v::Size; ++k) {
      for(unsigned int i = start; i < first+start; ++i) {
         for(unsigned int j = start; j < second+start; ++j) {
            typename M::value_type x = m(i,j);
            x[k] = r.Rndm() + offset;
            m(i,j) = x;
         }
      }
   }
}

template<class M>
void fillRandomSym(TRandom & r, M  & m, unsigned int first, unsigned int start = 0, double offset = 1) {
   for (int k = 0; k < Vc::double_v::Size; ++k) {
      for(unsigned int i = start; i < first+start; ++i) {
         for(unsigned int j = i; j < first+start; ++j) {
            typename M::value_type x = m(i,j);
            if ( i != j ) {
               x[k] = r.Rndm() + offset;
               m(i,j) = x;
               m(j,i) = m(i,j);
            }
            else {// add extra offset to make no singular when inverting
               x[k] = r.Rndm() + 3*offset;
               m(i,i) = x;
            }
         }
      }
   }
}
#endif

#endif