Matrix.cxx

Go to the documentation of this file.

#include "AtlfastUtils/Matrix.h"
#include <cstddef>
#include <assert.h>
namespace Atlfast{
  Matrix::~Matrix(){}
  Matrix::Matrix(const Matrix& m): m_v(m.m_v), m_d1(m.m_d1), m_d2(m.m_d2){}
  Matrix::Matrix(size_t x, size_t y){
    m_d1=x;
    m_d2=y;
    m_v=new valarray<double>(x*y);
  }

  Matrix::Matrix(size_t x, size_t y, double init){
    m_d1=x;
    m_d2=y;
    m_v=new valarray<double>(x*y);
    for(size_t i=0; i<m_d1*m_d2; ++i)  m_v[i]=init;
  }

  double& Matrix::operator()(size_t x, size_t y){
    return column(x)[y];
  }

  double mul(const Cslice_iter<double>& v1, 
             const valarray<double>& v2){
    double res=0;
    for( size_t i=0; i<v2.size(); i++) res+=v1[i]*v2[i];
    return res;
  }

  valarray<double> operator*(const Matrix&m, const valarray<double>&v){
    valarray<double> res(m.dim2());
    for(size_t i=0; i<m.dim2(); i++) res[i]=mul(m.row(i), v);
    return res;
  }

  Matrix& Matrix::operator*=(double d){
    (*m_v) *= d;
    return *this;
  }
  Matrix Matrix::sub(size_t minRow, size_t maxRow, 
                     size_t minCol, size_t maxCol) const{
    size_t endRow=maxRow+1;
    size_t endCol=maxCol+1;

    assert(endRow<=m_d1);
    assert(endCol<=m_d2);
    assert(endRow>minRow);
    assert(endCol>minCol);

    size_t nRow=endRow-minRow;
    
    size_t nCol=endCol-minCol;

    Matrix m(nRow, nCol);
    size_t ii=0;
    size_t jj;
    for(size_t i = minRow; i<endRow; ++i){
      jj=0;
      for(size_t j = minCol; j<endCol; ++j) m[ii][jj++]=(*this)[i][j];
      ++ii;
    }
    assert(ii == endRow);
    assert(jj == endCol);
    return m;
  }
  void Matrix::sub(size_t minRow, size_t minCol, Matrix& m) const{
    size_t endRow=minRow+m.m_d1;
    size_t endCol=minCol+m.m_d2;

    assert(endRow<=m_d1);
    assert(endCol<=m_d2);
    assert(endRow>minRow);
    assert(endCol>minCol);

    size_t ii=0;
    size_t jj;
    for(size_t i = minRow; i<endRow; ++i){
      jj=0;
      for(size_t j = minCol; j<endCol; ++j) m[ii][jj++]=(*this)[i][j];
      ++ii;
    }
    assert(ii == endRow);
    assert(jj == endCol);
    return;
  }
}//namespace

---