pzstencil.h

Go to the documentation of this file.

#ifndef STENMATH
#define STENMATH

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


template<class TVar>
class TPZStencilMatrix : public TPZMatrix<TVar> {
                
                public :
                
    TPZStencilMatrix( int rows, int cols );
                
                virtual ~TPZStencilMatrix();
                
                virtual int Rows() const;
                virtual int Cols() const;
                
                virtual const TVar &GetVal(const int row,const int col ) const;
                
                virtual void MultAdd(const TPZFMatrix<TVar> &x,const TPZFMatrix<TVar> &y, TPZFMatrix<TVar> &z,
                                                                                                 const TVar alpha=1., const TVar beta = 0., const int opt = 0) const;
                
                virtual void Print(const char *title, std::ostream &out = std::cout ,const MatrixOutputFormat form = EFormatted) const;
                
                void SolveSOR( int &numiterations,const TPZFMatrix<TVar> &rhs, TPZFMatrix<TVar> &x,
                                                                  TPZFMatrix<TVar> *residual, TPZFMatrix<TVar> &scratch,
                                                                  const TVar overrelax, TVar &tol,
                                                                  const int FromCurrent = 0,const int direction = 1 );
                
                void SetStencil( int stencilnumber, int inc, int *IA, TVar *A );
                
                void SetNodeStencils( int *stencilnumber );
                
private:
                
                void IncreaseStencilPointers( int stencilnumber );
                
                void ComputeDiagonal();
                
                struct MPStencil {
                                int fNumberOfItems;
                                int fInc;
                                int *fIA;
                                TVar *fA;
                                
                                MPStencil( int inc, int *IA, TVar *A );
                                ~MPStencil();
                                
                } **fMystencils;
                
                int   fNumberOfStencilPointers;
                int   fRows, fCols;
                int  *fStencilNumbers;
                
                TVar *fDiag;
                int   fSolver;
                int   fMaxIterations;
                
                TVar fSORRelaxation;
};

template<class TVar>
inline int TPZStencilMatrix<TVar>::Rows() const{
                return fRows;
}

template<class TVar>
inline int TPZStencilMatrix<TVar>::Cols() const{
                return fCols;
}

#endif