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 "TVectorD.h"
#include "TMatrixD.h"
#include "TMatrixDSym.h"
#include "VMatrix.h"


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