BorderedBandMatrix.h

Go to the documentation of this file.

/*
 * BorderedBandMatrix.h
 *
 *  Created on: Aug 14, 2011
 *      Author: kleinwrt
 */

#ifndef BORDEREDBANDMATRIX_H_
#define BORDEREDBANDMATRIX_H_

#include<iostream>
#include<vector>
#include<math.h>
#include<cstdlib>
#include "TVectorD.h"
#include "TMatrixD.h"
#include "TMatrixDSym.h"
#include "VMatrix.h"

namespace gbl {


class BorderedBandMatrix {
public:
    BorderedBandMatrix();
    virtual ~BorderedBandMatrix();
    void resize(unsigned int nSize, unsigned int nBorder = 1,
            unsigned int nBand = 5);
    void solveAndInvertBorderedBand(const VVector &aRightHandSide,
            VVector &aSolution);
    void addBlockMatrix(double aWeight,
            const std::vector<unsigned int>* anIndex,
            const std::vector<double>* aVector);
    TMatrixDSym getBlockMatrix(const std::vector<unsigned int> anIndex) const;
    void printMatrix() const;

private:
    unsigned int numSize; 
    unsigned int numBorder; 
    unsigned int numBand; 
    unsigned int numCol; 
    VSymMatrix theBorder; 
    VMatrix theMixed; 
    VMatrix theBand; 

    void decomposeBand();
    VVector solveBand(const VVector &aRightHandSide) const;
    VMatrix solveBand(const VMatrix &aRightHandSide) const;
    VMatrix invertBand();
    VMatrix bandOfAVAT(const VMatrix &anArray,
            const VSymMatrix &aSymArray) const;
};
}
#endif /* BORDEREDBANDMATRIX_H_ */