pzblock.h

Go to the documentation of this file.

#ifndef _TBLOCKHH_
#define _TBLOCKHH_

#include "pzmatrix.h"
#include "pzmanvector.h"
#include "pzreal.h"
#include "TPZSavable.h"


template<class TVar>
class TPZBlock : public TPZSavable
{
public:
    TPZBlock() : TPZRegisterClassId(&TPZBlock::ClassId), fBlock(), fpMatrix(0)
    {
        
    }
                TPZBlock(TPZMatrix<TVar> *const matrix_to_represent,const int num_of_blocks = 0,
                                                 const int initial_blocks_size = 1 );
                
                TPZBlock(const TPZBlock<TVar> &bl);
                
                virtual ~TPZBlock();
                
                void SetMatrix(TPZMatrix<TVar> *const other)
    {
        fpMatrix = other;
    }
                
                TPZMatrix<TVar> *Matrix(){ return fpMatrix;}
                
                int SetNBlocks(const int num_of_blocks );
                
                int Set(const int index,const int dim,const int pos = -1 );
                
                int SetAll( TPZVec<int> & dimensions );

                int Resequence(const int start=0);
                
                int Remove(const int index );
                
                int Verify() const;
                
                TVar & operator()(const int block_row,const int block_col,const int r,const int c ) const;
                
                const TVar & Get(const int block_row,const int block_col,const int r,const int c ) const;
                int Put(const int block_row,const int block_col,const int r,const int c,const TVar& value );
                
                const TVar & Get(const int block_row,const int r,const int c ) const;
                int Put(const int block_row,const int r,const int c,const TVar& value );
                
                const TVar & GetVal(const int bRow,const int bCol,const int r,const int c ) const;
                int PutVal(const int bRow,const int bCol,const int r,const int c,const TVar& value );
                
                int PutBlock(const int block_row,const int block_col,const TPZFMatrix<TVar> & block );
                int GetBlock(const int block_row,const int block_col, TPZFMatrix<TVar> & block ) const;
                int AddBlock(const int block_row,const int block_col, const TPZFMatrix<TVar> & block );
                                
                int InsertBlock(const int block_row,const int block_col,
                                                                                const int row,const int col, TPZMatrix<TVar> &target) const;
                
                TPZBlock<TVar>&operator=(const TPZBlock<TVar>& ); 
                 
                int  PrintBlock(const int block_row,const int block_col,const char *title = "",TPZostream &out = std::cout ) const;
                
                void Print(const char *title = "",TPZostream &out = std::cout,TPZMatrix<TVar> *mat=NULL);
                
                void PrintSolution(const char *title, TPZostream &out);
                
                int MaxBlockSize() const {return fBlock.NElements();}
                int NBlocks() const {return fBlock.NElements();}
                
                int Size(const int block_diagonal) const { return fBlock[block_diagonal].dim; }
                
                int Position(const int block_diagonal) const { return fBlock[block_diagonal].pos;}
                
                int Dim() const {return fBlock.NElements() ? fBlock[fBlock.NElements()-1].pos+fBlock[fBlock.NElements()-1].dim : 0; }
                
        int ClassId() const override;
                
                void Write(TPZStream &buf, int withclassid) const override;
                
                void Read(TPZStream &buf, void *context) override;

private:
                
                class TNode : public TPZSavable {
            public :
                
                                int pos; 
                                int dim; 
                                TNode() {
                                                pos=0;
                                                dim=0;
                                }
                
        int ClassId() const override{
            return Hash("TNode") ^ ClassIdOrHash<TPZBlock<TVar>>() << 1;
        }
                
                                void Read(TPZStream &buf, void *context) override;
                                void Write(TPZStream &buf, int withclassid) const override;
                };
                
                TPZManVector<TNode>    fBlock;
                TPZMatrix<TVar> *fpMatrix;
                static REAL gZero;//zero
                
};

template<class TVar>
int TPZBlock<TVar>::ClassId() const {
    return Hash("TPZBlock") ^ ClassIdOrHash<TVar>() << 1;
}

#endif