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_ */