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 "VMatrix.h"
 
#include <iostream>
#include <vector>
#include <math.h>
#include <cstdlib>
#include <TVectorD.h>
#include <TMatrixD.h>
#include <TMatrixDSym.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_ */