VMatrix.h

Go to the documentation of this file.

/*
 * VMatrix.h
 *
 *  Created on: Feb 15, 2012
 *      Author: kleinwrt
 */
 
#ifndef VMATRIX_H_
#define VMATRIX_H_
 
#include<iostream>
#include<iomanip>
#include<vector>
#include<cstring>
#include<math.h>
 
namespace gbl {
 
class VVector {
public:
  VVector(const unsigned int nRows = 0);
  VVector(const VVector &aVector);
  virtual ~VVector();
  void resize(const unsigned int nRows);
  VVector getVec(unsigned int len, unsigned int start = 0) const;
  void putVec(const VVector &aVector, unsigned int start = 0);
  inline double &operator()(unsigned int i);
  inline double operator()(unsigned int i) const;
  unsigned int getNumRows() const;
  void print() const;
  VVector operator-(const VVector &aVector) const;
  VVector &operator=(const VVector &aVector);
private:
  unsigned int numRows; 
  std::vector<double> theVec; 
};
 
class VMatrix {
public:
  VMatrix(const unsigned int nRows = 0, const unsigned int nCols = 0);
  VMatrix(const VMatrix &aMatrix);
  virtual ~VMatrix();
  void resize(const unsigned int nRows, const unsigned int nCols);
  VMatrix transpose() const;
  inline double &operator()(unsigned int i, unsigned int j);
  inline double operator()(unsigned int i, unsigned int j) const;
  unsigned int getNumRows() const;
  unsigned int getNumCols() const;
  void print() const;
  VVector operator*(const VVector &aVector) const;
  VMatrix operator*(const VMatrix &aMatrix) const;
  VMatrix operator+(const VMatrix &aMatrix) const;
  VMatrix &operator=(const VMatrix &aMatrix);
private:
  unsigned int numRows; 
  unsigned int numCols; 
  std::vector<double> theVec; 
};
 
class VSymMatrix {
public:
  VSymMatrix(const unsigned int nRows = 0);
  virtual ~VSymMatrix();
  void resize(const unsigned int nRows);
  unsigned int invert();
  inline double &operator()(unsigned int i, unsigned int j);
  inline double operator()(unsigned int i, unsigned int j) const;
  unsigned int getNumRows() const;
  void print() const;
  VSymMatrix operator-(const VMatrix &aMatrix) const;
  VVector operator*(const VVector &aVector) const;
  VMatrix operator*(const VMatrix &aMatrix) const;
private:
  unsigned int numRows; 
  std::vector<double> theVec; 
};
 
inline double &VMatrix::operator()(unsigned int iRow, unsigned int iCol) {
  return theVec[numCols * iRow + iCol];
}
 
inline double VMatrix::operator()(unsigned int iRow, unsigned int iCol) const {
  return theVec[numCols * iRow + iCol];
}
 
inline double &VVector::operator()(unsigned int iRow) {
  return theVec[iRow];
}
 
inline double VVector::operator()(unsigned int iRow) const {
  return theVec[iRow];
}
 
inline double &VSymMatrix::operator()(unsigned int iRow, unsigned int iCol) {
  return theVec[(iRow * iRow + iRow) / 2 + iCol]; // assuming iCol <= iRow
}
 
inline double VSymMatrix::operator()(unsigned int iRow,
    unsigned int iCol) const {
  return theVec[(iRow * iRow + iRow) / 2 + iCol]; // assuming iCol <= iRow
}
}
#endif /* VMATRIX_H_ */