pzbdstrmatrix.h

Go to the documentation of this file.

#ifndef TPZBLOCKDIAGONALSTRUCTMATRIX_H
#define TPZBLOCKDIAGONALSTRUCTMATRIX_H

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

#include "pzcmesh.h"
#include "pzvec.h"
#include "pzblockdiag.h"

class TPZBlockDiagonalStructMatrix : public TPZStructMatrix {
public:
    
    int ClassId() const override;

    enum MBlockStructure {ENodeBased, EVertexBased, EElementBased};
    
    TPZBlockDiagonalStructMatrix(TPZCompMesh *);
    
    ~TPZBlockDiagonalStructMatrix();
    
    TPZBlockDiagonalStructMatrix(const TPZBlockDiagonalStructMatrix &copy) : 
    TPZRegisterClassId(&TPZBlockDiagonalStructMatrix::ClassId), TPZStructMatrix(copy),
    fBlockStructure(copy.fBlockStructure),fOverlap(copy.fOverlap)
    {
    }
    
    virtual TPZMatrix<STATE> * Create() override;
    
    virtual TPZMatrix<STATE> * CreateAssemble(TPZFMatrix<STATE> &rhs,TPZAutoPointer<TPZGuiInterface> guiInterface) override;
    
    virtual TPZStructMatrix * Clone() override;
    
public:
    
    void AssembleBlockDiagonal(TPZBlockDiagonal<STATE> & block);
private:
    TPZBlockDiagonalStructMatrix();
    
    void BlockSizes(TPZVec < int > & blocksizes);
    
    friend TPZPersistenceManager;
    
    MBlockStructure fBlockStructure;
    int fOverlap;
    
    
};

#endif //TPZBLOCKDIAGONALSTRUCTMATRIX_H