pzsysmp.h

Go to the documentation of this file.

#ifndef SYSMPMATH
#define SYSMPMATH

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

#ifdef USING_MKL

#include "TPZPardisoControl.h"
#endif

template<class TVar>
class TPZSYsmpMatrix : public TPZMatrix<TVar>{
                
#ifdef USING_MKL
    friend class TPZPardisoControl<TVar>;
#endif
    
public :
    TPZSYsmpMatrix();
    TPZSYsmpMatrix(const int64_t rows, const int64_t cols );
    TPZSYsmpMatrix(const TPZSYsmpMatrix<TVar> &cp) : 
    TPZRegisterClassId(&TPZSYsmpMatrix::ClassId),
    TPZMatrix<TVar>(cp), fIA(cp.fIA), fJA(cp.fJA), fA(cp.fA), fDiag(cp.fDiag)
#ifdef USING_MKL
    , fPardisoControl(cp.fPardisoControl)
#endif
    {
#ifdef USING_MKL
        fPardisoControl.SetMatrix(this);
#endif
    }
    
    TPZSYsmpMatrix &operator=(const TPZSYsmpMatrix<TVar> &copy);
    
    CLONEDEF(TPZSYsmpMatrix)
                virtual ~TPZSYsmpMatrix();
    
    virtual int IsSimetric() const  override { return 1; }
    inline int IsSquare() const { return 1;}
    
    virtual int Zero() override {
        fA.Fill(0.);
        fDiag.Fill(0.);
#ifndef USING_MKL
        TPZMatrix<TVar>::fDecomposed = ENoDecompose;
#endif
        return 0;
        
    }

    
    void AutoFill(int64_t nrow, int64_t ncol, int symmetric);
                
                virtual const TVar &GetVal(const int64_t row, const int64_t col ) const override;
    
    virtual int PutVal(const int64_t /*row*/,const int64_t /*col*/,const TVar & val ) override;

                
                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 override;
                
                virtual void SetData(const TPZVec<int64_t> &IA,const TPZVec<int64_t> &JA, const TPZVec<TVar> &A );
                
    TPZVec<TVar> &A()
    {
        return fA;
    }
    
    TPZVec<int64_t> &IA()
    {
        return fIA;
    }
    
    TPZVec<int64_t> &JA()
    {
        return fJA;
    }

                virtual void Print(const char *title, std::ostream &out = std::cout ,const MatrixOutputFormat = EFormatted ) const override;
    
#ifdef USING_MKL

    virtual int Decompose_LDLt(std::list<int64_t> &singular) override;
    virtual int Decompose_LDLt() override;

    virtual int Decompose_Cholesky() override;
    virtual int Decompose_Cholesky(std::list<int64_t> &singular)  override;
    

    virtual int Subst_LForward( TPZFMatrix<TVar>* b ) const override;
    
    virtual int Subst_LBackward( TPZFMatrix<TVar>* b ) const override;
    
    virtual int Subst_Diag( TPZFMatrix<TVar>* b ) const override;

    virtual int Subst_Forward( TPZFMatrix<TVar>* b ) const override;
    
    virtual int Subst_Backward( TPZFMatrix<TVar>* b ) const override;
    

#endif
    public:
int ClassId() const override;

    void ComputeDiagonal();

private:
                
                
                TPZVec<int64_t>  fIA;
                TPZVec<int64_t>  fJA;
                TPZVec<TVar> fA;
                
#ifdef USING_MKL
    TPZPardisoControl<TVar> fPardisoControl;
#endif
                
                TPZVec<TVar> fDiag;
};

template<class TVar>
inline void TPZSYsmpMatrix<TVar>::SetData(const TPZVec<int64_t> &IA,const TPZVec<int64_t> &JA, const TPZVec<TVar> &A )
{
                // Pass the data to the class.
                fIA = IA;
                fJA = JA;
                fA  =  A;
                ComputeDiagonal();
}

#endif