pzsespmat.h

Go to the documentation of this file.

#ifndef TSESPMATH
#define TSESPMATH

#include "pzlink.h"
#include "pzmatrix.h"
#include "pzfmatrix.h"
#include "pzsfulmat.h"
#include "pzespmat.h"

#ifdef OOPARLIB

#include "pzsaveable.h"
#include "pzmatdefs.h"

#endif

template<class TVar>
class TPZSSpMatrix : public TPZMatrix<TVar>
{
public:
                TPZSSpMatrix() : TPZMatrix<TVar>(0,0)  {}
                TPZSSpMatrix(const int64_t dim ) : TPZMatrix<TVar>(dim,dim), fMat(dim, dim) {}
                TPZSSpMatrix(const TPZSSpMatrix<TVar> & );
                
                CLONEDEF(TPZSSpMatrix)
                
                inline int    PutVal(const int64_t row,const int64_t col,const TVar&  element );
                inline const TVar & GetVal(const int64_t row,const int64_t col ) const;
                
                // @{
                TPZSSpMatrix &operator= (const TPZSSpMatrix<TVar> &A );
                TPZSSpMatrix operator+  (const TPZSSpMatrix<TVar> &A ) const;
                TPZSSpMatrix operator-  (const TPZSSpMatrix<TVar> &A ) const;
                TPZSSpMatrix &operator+=(const TPZSSpMatrix<TVar> &A );
                TPZSSpMatrix &operator-=(const TPZSSpMatrix<TVar> &A );
                // @}
                
                // @{
                TPZSpMatrix<TVar> operator+(const TPZSpMatrix<TVar> &A ) const;
                TPZSpMatrix<TVar> operator-(const TPZSpMatrix<TVar> &A ) const;
                // @}
                
                // @{
                TPZSSpMatrix operator*  (const TVar v ) const;
                TPZSSpMatrix &operator*=(const TVar v );
                // @}
                
                int Resize(const int64_t newDim ,const int64_t )
    { this->fRow = this->fCol = newDim; return fMat.Resize( newDim, newDim ); }
                
                int Redim(const int64_t newDim) { return Redim(newDim,newDim);}
                int Redim(const int64_t newDim ,const int64_t )
    { this->fRow = this->fCol = newDim; return fMat.Redim( newDim, newDim ); }
                
                // Zeroes all the elements
                int Zero()
    {return fMat.Zero();}
                
                
                /*** @brief To solve linear systems ***/
                // @{
                int Decompose_Cholesky();  // Faz A = GGt.
                int Decompose_LDLt    ();  // Faz A = LDLt.
                int Decompose_Cholesky(std::list<int64_t> &singular);  // Faz A = GGt.
                int Decompose_LDLt    (std::list<int64_t> &singular);  // Faz A = LDLt.
                
                int Subst_Forward  ( TPZFMatrix<TVar> *b ) const;
                //int Subst_Backward ( TPZMatrix<>*b );
                int Subst_LForward ( TPZFMatrix<TVar> *b ) const;
                //int Subst_LBackward( TPZMatrix<>*b );
                int Subst_Diag     ( TPZFMatrix<TVar> *b ) const;
                // @}
                
#ifdef OOPARLIB
                
                virtual int Unpack( TReceiveStorage *buf );
                static TSaveable *CreateInstance(TReceiveStorage *buf);
                virtual int Pack( TSendStorage *buf ) const;
                virtual std::string ClassName() const   { return( "TPZSSpMatrix"); }
                virtual int DerivedFrom(const int64_t Classid) const;
                virtual int DerivedFrom(const char *classname) const; // a class with name classname
                
#endif
                
private:
                TVar ProdEsc( TPZLink<TPZSpMatrix<REAL>::TPZNode> *row_i,
                                                                 TPZLink<TPZSpMatrix<REAL>::TPZNode> *row_j, int64_t k );
                
                TPZSpMatrix<TVar> fMat;
};


/**************/
/*** PutVal ***/
//
//  Escreve um elemento na matriz, sem verificar fronteiras.
//  O valor da linha (row) deve ser maior ou igual ao da coluna
// (col).
//
template<class TVar>
inline int
TPZSSpMatrix<TVar>::PutVal(const int64_t r,const int64_t c,const TVar&  value )
{
                // Inicializando row e col para trabalhar com a matriz
                //  triangular inferior.
                int64_t row(r),col(c);
                if ( row < col )
                                this->Swap( &row, &col );
                
                this->fDecomposed = 0;
                return( fMat.PutVal( row, col, value ) );
}

/**************/
/*** GetVal ***/
//
//  Le um elemento da matriz, sem verificar fronteiras. O valor
// da linha (row) deve ser maior ou igual ao da coluna (col).
//
template<class TVar>
inline const TVar &
TPZSSpMatrix<TVar>::GetVal(const int64_t r,const int64_t c ) const
{
                // inicializando row e col para trabalhar com a matriz
                // triangular inferior.
                int64_t row(r),col(c);
                if ( row < col )
                                this->Swap( &row, &col );
                
                return( fMat.GetVal( row, col ) );
}

#endif