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);
        void setZero();
        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);
        void setZero();
        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);
        void setZero();
        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_ */