pztransfer.h

Go to the documentation of this file.

#ifndef TRANSFERH
#define TRANSFERH

#include "pzblock.h"
#include "pzvec.h"
#include "pzmanvector.h"
#include "pzmatrix.h"

template<class TVar>
class TPZTransfer : public TPZMatrix<TVar> {
                
public :
    int ClassId() const override;

    TPZTransfer();
                
                //TPZTransfer(TPZBlock<REAL> &row, TPZBlock<REAL> &col,int nvar, int nrowblocks, int ncolblocks);
                TPZTransfer(TPZBlock<TVar> &row, TPZBlock<TVar> &col,int nvar, int nrowblocks, int ncolblocks);
    
                TPZTransfer(const TPZTransfer &cp) : TPZRegisterClassId(&TPZTransfer::ClassId),
    TPZMatrix<TVar>(cp),
                fNTVarVar(cp.fNTVarVar), fRowBlock(cp.fRowBlock),
                fColBlock(cp.fColBlock),fColPosition(cp.fColPosition),
                fNumberofColumnBlocks(cp.fNumberofColumnBlocks),
                fColumnBlockNumber(cp.fColumnBlockNumber),
                fColumnBlockLastUsed(cp.fColumnBlockLastUsed),
                fDoubleValues(cp.fDoubleValues),
                fDoubleValLastUsed(cp.fDoubleValLastUsed)
                {
                }
                
                virtual TPZMatrix<TVar> *Clone() const  override { return new TPZTransfer(*this); }
                
                //TPZMatrix<REAL> : EFormatted, EInputFormat, EMathematicaInput
                virtual void Print(const char *name = NULL, std::ostream &out = std::cout , const MatrixOutputFormat form = EFormatted) const override;
                
                void SetNTVarVariables(int TVarvar) { fNTVarVar = TVarvar; }
                
                //void SetBlocks(TPZBlock<REAL> &row, TPZBlock<REAL> &col, int nvar, int nrowblocks, int ncolblocks);
                void SetBlocks(TPZBlock<TVar> &row, TPZBlock<TVar> &col, int nvar, int nrowblocks, int ncolblocks);
                
                int HasRowDefinition(int row);
                
                void AddBlockNumbers(int row, TPZVec<int> &colnumbers);
                
                void SetBlockMatrix(int row, int col, TPZFMatrix<TVar> &mat);
                
                void MultAdd(const TPZFMatrix<TVar> &x,const TPZFMatrix<TVar> &y, TPZFMatrix<TVar> &z,
                                                                 const TVar alpha,const TVar beta, const int opt = 0) const  override;
                
    void MultAddScalar(const TPZFMatrix<TVar> &x,const TPZFMatrix<TVar> &y, TPZFMatrix<TVar> &z,
                 const TVar alpha,const TVar beta, const int opt = 0) const;
                void TransferSolution(const TPZFMatrix<TVar> &coarsesol, TPZFMatrix<TVar> &finesol);
                
                void TransferResidual(const TPZFMatrix<TVar> &fine, TPZFMatrix<TVar> &coarse);
                
//              void Multiply(const TPZFMatrix<TVar> &A, TPZFMatrix<TVar> &B, int opt) const override;
                
    void MultiplyScalar(const TPZFMatrix<TVar> &A, TPZFMatrix<TVar> &B, int opt) const;
    
private:
                
                void ExpandColumnVectorEntries(int numcol);
                
                void ExpandDoubleValueEntries(int numval);
                
                int fNTVarVar;
                //TPZBlock<REAL> fRowBlock;
                TPZBlock<TVar> fRowBlock;
                //TPZBlock<REAL> fColBlock;
                TPZBlock<TVar> fColBlock;
                TPZVec<int> fColPosition;
                TPZVec<int> fNumberofColumnBlocks;
                TPZManVector<int> fColumnBlockPosition;
                TPZManVector<int> fColumnBlockNumber;
                int fColumnBlockLastUsed;
                TPZManVector<TVar> fDoubleValues;
                int fDoubleValLastUsed;
                
};

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

#endif