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 "Alignment/ReferenceTrajectories/interface/VMatrix.h"
#include "Eigen/Core"

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);
    void addBlockMatrix(double aWeight, unsigned int nSimple,
                        unsigned int* anIndex, double* aVector);
    Eigen::MatrixXd getBlockMatrix(const std::vector<unsigned int> anIndex) const;
    Eigen::MatrixXd getBlockMatrix(unsigned int aSize, 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_ */