src/work/alpar/matrix.cc
author alpar
Sun, 06 Mar 2005 21:13:24 +0000
changeset 1200 ae69f556b429
permissions -rw-r--r--
Doc improvements
alpar@994
     1
#include <lemon/time_measure.h>
alpar@994
     2
#include <vector>
alpar@994
     3
alpar@994
     4
template<class T>
alpar@994
     5
class Matrix 
alpar@994
     6
{
alpar@994
     7
public:
alpar@994
     8
  int m,n;
alpar@994
     9
  std::vector<T> values;
alpar@994
    10
  
alpar@994
    11
  Matrix(int _m, int _n) : m(_m), n(_n), values(n*m) {}
alpar@994
    12
  typename std::vector<T>::reference operator()(int i, int j)
alpar@994
    13
  { return values[i*n+j];}
alpar@994
    14
alpar@994
    15
  class Row 
alpar@994
    16
  {
alpar@994
    17
  public:
alpar@994
    18
    Matrix<T> &mat;
alpar@994
    19
    int row;
alpar@994
    20
    
alpar@994
    21
    Row(Matrix<T> &_mat,int _row) : mat(_mat), row(_row) {}
alpar@994
    22
    typename std::vector<T>::reference operator()(int j)
alpar@994
    23
    { return mat.values[row*mat.n+j];}
alpar@994
    24
  };
alpar@994
    25
  class Column 
alpar@994
    26
  {
alpar@994
    27
  public:
alpar@994
    28
    Matrix<T> &mat;
alpar@994
    29
    int col;
alpar@994
    30
    
alpar@994
    31
    Column(Matrix<T> &_mat,int _col) : mat(_mat), col(_col) {}
alpar@994
    32
    typename std::vector<T>::reference operator[](int i)
alpar@994
    33
    { return mat.values[i*mat.n+col];}
alpar@994
    34
  };
alpar@994
    35
  
alpar@994
    36
  Row operator[](int i) { return Row(*this,i); }
alpar@994
    37
  Column operator()(int j) { return Column(*this,j); }
alpar@994
    38
};
alpar@994
    39
alpar@994
    40
main() 
alpar@994
    41
{
alpar@994
    42
  const int COUNT=1000000;
alpar@994
    43
  const int SIZE=100;
alpar@994
    44
  
alpar@994
    45
  Matrix<int> A(SIZE,SIZE);
alpar@994
    46
  
alpar@994
    47
  for(int i=0;i<SIZE;i++) for(int j=0;j<SIZE;j++) A(i,j)=i+j;
alpar@994
    48
alpar@994
    49
  int s=0;
alpar@994
    50
  lemon::Timer T;
alpar@994
    51
  for(int c=0;c<COUNT;c++) {
alpar@994
    52
    for(int i=0;i<SIZE;i++) for(int j=0;j<SIZE;j++) s+=A[i](j);
alpar@994
    53
  }
alpar@994
    54
  std::cout << "A[i](j): " << T << ' ' << s <<'\n';
alpar@994
    55
  s=0;
alpar@994
    56
  T.reset();
alpar@994
    57
  for(int c=0;c<COUNT;c++) {
alpar@994
    58
    for(int i=0;i<SIZE;i++) for(int j=0;j<SIZE;j++) s+=A(i,j);
alpar@994
    59
  }
alpar@994
    60
  std::cout << "A(i,j): " << T << ' ' << s << '\n';
alpar@994
    61
  s=0;
alpar@994
    62
  T.reset();
alpar@994
    63
  for(int c=0;c<COUNT;c++) {
alpar@994
    64
    for(int i=0;i<SIZE;i++) for(int j=0;j<SIZE;j++) s+=A(j)[i];
alpar@994
    65
  }
alpar@994
    66
  std::cout << "A(j)[i]: " << T << ' ' << s << '\n';
alpar@994
    67
  s=0;
alpar@994
    68
  T.reset();
alpar@994
    69
  for(int c=0;c<COUNT;c++) {
alpar@994
    70
    for(int i=0;i<SIZE;i++) for(int j=0;j<SIZE;j++) s+=A(i,j);
alpar@994
    71
  }
alpar@994
    72
  std::cout << "A(j,i): " << T << ' ' << s << "\n\n";
alpar@994
    73
alpar@994
    74
  return 0;
alpar@994
    75
}