pzmgsolver.h

Go to the documentation of this file.

#ifndef TPZMGSOLVER_H
#define TPZMGSOLVER_H
#include "pzsolve.h"
#include "pzstepsolver.h"
#include "pzfmatrix.h"
#include "pztransfer.h"

#define TPZMGSOLVER_ID 28291008

template <class TVar>
class TPZMGSolver: public TPZMatrixSolver<TVar>
{
public:
                TPZMGSolver() : TPZRegisterClassId(&TPZMGSolver::ClassId),TPZMatrixSolver<TVar>() {}
                TPZMGSolver(TPZAutoPointer<TPZMatrix<TVar> > trf, const TPZMatrixSolver<TVar> &sol,
                                                                int nvar, TPZAutoPointer<TPZMatrix<TVar> > refmat);
                TPZMGSolver(TPZAutoPointer<TPZMatrix<TVar> > trf, const TPZMatrixSolver<TVar> &sol,
                                                                int nvar);
                
                TPZMGSolver(const TPZMGSolver<TVar> & copy);
                ~TPZMGSolver();
                
                void SetTransferMatrix(TPZAutoPointer<TPZMatrix<TVar> > Refmat);
                void ResetTransferMatrix();
                
                TPZAutoPointer<TPZMatrix<TVar> > TransferMatrix()
                {
                                return this->fTransfer;
                }
                
                TPZSolver<TVar> * Clone() const override;
                
                void Solve(const TPZFMatrix<TVar> &F, TPZFMatrix<TVar> &result, TPZFMatrix<TVar> *residual = 0) override;
                
                public:
int ClassId() const override;

                void Write(TPZStream &buf, int withclassid) const override;
                void Read(TPZStream &buf, void *context) override;
                
                
private:
                TPZMatrixSolver<TVar> * fCoarse;
                int fNVar;
                TPZAutoPointer<TPZMatrix<TVar> > fTransfer;
                //    TPZMatrixSolver::TPZContainer *fTransfer;
};

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

#endif //TPZMGSOLVER_H