pzespmat.h

Go to the documentation of this file.

#ifndef TESPMATH
#define TESPMATH

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

#ifdef WORKPOOL

#include "pzworkpool.h"

#endif

#ifdef OOPARLIB

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

#endif

template<class TVar>
class TPZSSpMatrix;

template<class TVar>
class TPZSpMatrix : public TPZMatrix<TVar>
{
                friend class TPZSSpMatrix<TVar>;
                
public:
                
                struct TPZNode
                {
                                REAL elem;
                                int64_t    col;
                };
                
public:
                TPZSpMatrix ()
    : TPZRegisterClassId(&TPZSpMatrix::ClassId),
    TPZMatrix<TVar>( 0, 0 )  { fElem = NULL; }
                TPZSpMatrix ( int64_t rows, int64_t cols );
                TPZSpMatrix (const TPZSpMatrix<TVar> &A )
    : TPZRegisterClassId(&TPZSpMatrix::ClassId),
    TPZMatrix<TVar>( A )  { fCopy( &A ); }
                
                CLONEDEF(TPZSpMatrix)
                ~TPZSpMatrix();
                
                int    Put(const int64_t row,const int64_t col,const TVar & value ) override;
                const TVar &Get(const int64_t row,const int64_t col ) const override;
                
                // Nao verifica limites da matriz (e' mais rapido).
                int    PutVal(const int64_t row,const int64_t col,const TVar & element ) override;
                const TVar &GetVal(const int64_t row,const int64_t col ) const override;
                
                TPZSpMatrix &operator= (const TPZSpMatrix<TVar> &A );
                TPZSpMatrix operator+  (const TPZSpMatrix<TVar> &A ) const;
                TPZSpMatrix operator-  (const TPZSpMatrix<TVar> &A ) const;
                TPZSpMatrix &operator+=(const TPZSpMatrix<TVar> &A );
                TPZSpMatrix &operator-=(const TPZSpMatrix<TVar> &A );
                
                //TPZSpMatrix & operator= (const TPZMatrix<TVar> & A );
                
                virtual void MultAdd(const TPZFMatrix<TVar> &x,const TPZFMatrix<TVar> &y, TPZFMatrix<TVar> &z,
                                                                                                 const REAL alpha ,const REAL beta = 0.,const int opt = 0) const override;
                
                
                TPZSpMatrix operator*  (const TVar v ) const;
                TPZSpMatrix &operator*=(const TVar v );
                
                TPZSpMatrix operator-() const { return operator*(-1.0); }
                
                TPZSpMatrix &Reset();
                
                int Resize(const int64_t newRows,const int64_t newCols ) override;
                int Resize(const int64_t newDim )   { return Resize( newDim, newDim ); }
                
                int Redim(const int64_t newRows,const int64_t newCols ) override;
                int Redim(const int64_t newDim )    { return Redim( newDim, newDim ); }
                
                int Zero() override;
                
                /*** Resolucao de sistemas ***/
                
#ifdef OOPARLIB
                
                virtual int Unpack( TReceiveStorage *buf );
                static TSaveable *CreateInstance(TReceiveStorage *buf);
                inline virtual int Pack( TSendStorage *buf ) const;
                virtual std::string ClassName() const   { return( "TPZSpMatrix" ); }
                virtual int DerivedFrom(const int64_t Classid) const;
                virtual int DerivedFrom(const char *classname) const;
                
#endif
                public:
int ClassId() const override;

protected:
                
                int fAdd(const TPZSpMatrix<TVar> *const A );               // operator +=
//              int fAdd(const TPZSpMatrix<TVar> *A, TPZSpMatrix<TVar> *B ); // operator +
                
                int fSub(const TPZSpMatrix<TVar> *const A );               // operator -=
//              int fSub( TPZSpMatrix<TVar> *A, TPZSpMatrix<TVar> *B ); // operator -
                
                int fCopy(const TPZSpMatrix<TVar> *const A );              // operator =
                
                int fMult( TVar val );                // operator *
                
                inline void Swap( int64_t *a, int64_t *b );
                
                int  Clear() override { delete [] fElem; return ( 1 );}
                
                REAL ProdEsc( TPZLink<TPZNode> *row_i, TPZLink<TPZNode> *row_j,
                                                                 int64_t k );
                
                
                TPZLink<TPZNode> *fElem;
                
#ifdef WORKPOOL
                TPZWorkPool fWp;
#endif
};

/*** Swap ***/
template<class TVar>
inline void
TPZSpMatrix<TVar>::Swap( int64_t *a, int64_t *b )
{
                int64_t c = *a;
                *a = *b;
                *b = c;
}

#include "pzsespmat.h"

#endif