pzblockdiag.h

Go to the documentation of this file.

#ifndef _TBLOCKDIAG_
#define _TBLOCKDIAG_


#include <iostream>
#include "pzmatrix.h"
#include "pzvec.h"

template <class TVar>
class TPZBlockDiagonal : public TPZMatrix<TVar>
{
                
public:
                TPZBlockDiagonal ();
                TPZBlockDiagonal (const TPZVec<int> &blocksizes, const TPZFMatrix<TVar> &glob );
                TPZBlockDiagonal (const TPZVec<int> &blocksizes);
                TPZBlockDiagonal (const TPZBlockDiagonal & );
                
                CLONEDEF(TPZBlockDiagonal)
                ~TPZBlockDiagonal();
                
                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;
                
                TVar &operator()(const int64_t row, const int64_t col);
                virtual TVar &s(const int64_t row, const int64_t col) override;

                int    PutVal(const int64_t row,const int64_t col,const TVar& value ) override;
                const  TVar &GetVal(const int64_t row,const int64_t col ) const override;
                
                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;
                
                int64_t Dim() const  override { return this->Rows(); }
                
                int Zero() override;
                
                int GetSizeofBlock(int64_t blockid) {return fBlockSize[blockid];}
                
                void Transpose(TPZMatrix<TVar> *const T) const override;
                virtual int Decompose_LU() override;
                virtual int Decompose_LU(std::list<int64_t> &singular) override;
                
                virtual int Substitution( TPZFMatrix<TVar> * B ) const override;
                
                virtual void UpdateFrom(TPZAutoPointer<TPZMatrix<TVar> > mat) override;
                
                static int main();
    
    void AutoFill(int64_t dim, int64_t dimj, int symmetric);
                
private:
                
                int Clear() override;

public:
                void Initialize(const TPZVec<int> &blocksize);
                void AddBlock(int64_t i, TPZFMatrix<TVar> &block);
                void SetBlock(int64_t i, TPZFMatrix<TVar> &block);
                
                void GetBlock(int64_t i, TPZFMatrix<TVar> &block);
                
                void BuildFromMatrix(TPZMatrix<TVar> &matrix);
                virtual void Print(const char *message, std::ostream &out = std::cout, const MatrixOutputFormat format =EFormatted) const override;
                
                int64_t NumberofBlocks() {return fBlockSize.NElements();}
    public:
int ClassId() const override;

protected:
                TPZVec<TVar> fStorage;
                TPZVec<int64_t> fBlockPos;
                TPZVec<int> fBlockSize;
};

template<class TVar>
inline TVar &TPZBlockDiagonal<TVar>::s(const int64_t row, const int64_t col) {
                // verificando se o elemento a inserir esta dentro da matriz
                return this->operator()(row,col);
}

#endif