TPZNLMultGridAnalysis.h

Go to the documentation of this file.

#ifndef TPZNLMGANALYSIS_H
#define TPZNLMGANALYSIS_H

#include <iosfwd>                 // for string, ostream
#include "pzanalysis.h"           // for TPZAnalysis
#include "pzmatrix.h"             // for TPZFMatrix, TPZMatrix
#include "pzreal.h"               // for STATE, REAL
#include "pzstack.h"              // for TPZStack

class TPZCompMesh;
class TPZMaterial;
template <class TVar> class TPZMatrixSolver;


class TPZNonLinMultGridAnalysis : public TPZAnalysis {
                
                TPZStack < TPZCompMesh * > fMeshes;
                
                TPZStack <TPZFMatrix<STATE> *> fSolutions;
                
                TPZStack <TPZMatrixSolver<STATE> *> fSolvers;
                
                TPZStack <TPZMatrixSolver<STATE> *> fPrecondition;
                
                clock_t fBegin,fInit;
                
public:
                
                ~TPZNonLinMultGridAnalysis();
                
                TPZNonLinMultGridAnalysis(TPZCompMesh *cmesh);
                
                void AppendMesh (TPZCompMesh * mesh);
                
                TPZCompMesh *PopMesh ();
                
                int NMeshes() {return fMeshes.NElements();}
                
                TPZCompMesh *IMesh(int64_t index);
                
                static TPZCompMesh *UniformlyRefineMesh (TPZCompMesh *coarcmesh,int levelnumbertorefine,int setdegree);
                
                static TPZCompMesh *AgglomerateMesh(TPZCompMesh *finemesh,int levelnumbertogroup);
                
                void SmoothingSolution(REAL tol,int numiter,TPZMaterial * mat,TPZAnalysis &an,int marcha = 0 ,
                                                                                                   const std::string &dxout = "plotfile.dx");
                
                void SmoothingSolution(REAL tol,int numiter,TPZMaterial * mat,TPZAnalysis &an,TPZFMatrix<STATE> &rhs);
                
                void SmoothingSolution2(REAL tol,int numiter,TPZMaterial * mat,TPZAnalysis &an,int marcha,
                                                                                                                const std::string &dxout);
                
                void ResetReference(TPZCompMesh *aggcmesh);
                
                void SetReference(TPZCompMesh *aggcmesh);
                
                void SetDeltaTime(TPZCompMesh *CompMesh,TPZMaterial * mat);
                
                void CoutTime(clock_t &start,const char *title);
                
                void OneGridAlgorithm(std::ostream &out,int nummat);
                
                void TwoGridAlgorithm(std::ostream &out,int nummat);
                
                void CalcResidual(TPZMatrix<STATE> &sol,TPZAnalysis &an,const std::string  &decompose,TPZFMatrix<STATE> &res);
                
                void CalcResidual(TPZMatrix<STATE> &sol,TPZFMatrix<STATE> &anres,TPZFMatrix<STATE> &res,TPZAnalysis &an,const std::string &decompose);
                
protected:
                
                void (*fFunction)(TPZMaterial *mat,TPZCompMesh *cmesh);
                
public:
                
                void SetAnalysisFunction(void (*fp)(TPZMaterial *mat,TPZCompMesh *cmesh)){
                                fFunction = fp;
                }
                
};

#endif