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