Collaboration diagram for MoFEM::PCMGSubMatrixCtx:

[legend]

Public Member Functions
	PCMGSubMatrixCtx (Mat a, IS is)

virtual	~PCMGSubMatrixCtx ()=default

MoFEMErrorCode	initData (Vec x)

Public Attributes
SmartPetscObj< Mat >	A

SmartPetscObj< IS >	iS

SmartPetscObj< VecScatter >	sCat

SmartPetscObj< Vec >	X

SmartPetscObj< Vec >	F

PetscLogEvent	MOFEM_EVENT_mult

PetscLogEvent	MOFEM_EVENT_sor

Friends
template<InsertMode MODE>
MoFEMErrorCode	sub_mat_mult_generic (Mat a, Vec x, Vec f)

MoFEMErrorCode	sub_mat_sor (Mat mat, Vec b, PetscReal omega, MatSORType flag, PetscReal shift, PetscInt its, PetscInt lits, Vec x)

Detailed Description

Definition at line 29 of file PCMGSetUpViaApproxOrders.cpp.

Constructor & Destructor Documentation

◆ PCMGSubMatrixCtx()

MoFEM::PCMGSubMatrixCtx::PCMGSubMatrixCtx	(	Mat	a,
		IS	is
	)

Definition at line 52 of file PCMGSetUpViaApproxOrders.cpp.

                                                : A(a, true), iS(is, true) {
   PetscLogEventRegister("PCMGSubMatrixCtx_mult", 0, &MOFEM_EVENT_mult);
   PetscLogEventRegister("PCMGSubMatrixCtx_sor", 0, &MOFEM_EVENT_sor);
 }

◆ ~PCMGSubMatrixCtx()

virtual MoFEM::PCMGSubMatrixCtx::~PCMGSubMatrixCtx ( )

virtualdefault

Member Function Documentation

◆ initData()

MoFEMErrorCode MoFEM::PCMGSubMatrixCtx::initData ( Vec x )

Definition at line 57 of file PCMGSetUpViaApproxOrders.cpp.

                                                {
   MoFEMFunctionBegin;
   if (sCat.use_count() == 0) {
     auto [x_tmp, f_tmp] = matCreateVecs(A);
     X = x_tmp;
     F = f_tmp;
     sCat = createVecScatter(X, iS, x, PETSC_NULL);
   }
   MoFEMFunctionReturn(0);
 }

Friends And Related Function Documentation

◆ sub_mat_mult_generic

template<InsertMode MODE>

MoFEMErrorCode sub_mat_mult_generic	(	Mat	a,
		Vec	x,
		Vec	f
	)

friend

Definition at line 144 of file PCMGSetUpViaApproxOrders.cpp.

                                                          {
   void *void_ctx;
   MoFEMFunctionBegin;
   CHKERR MatShellGetContext(a, &void_ctx);
   PCMGSubMatrixCtx *ctx = (PCMGSubMatrixCtx *)void_ctx;
   PetscLogEventBegin(ctx->MOFEM_EVENT_mult, 0, 0, 0, 0);
   CHKERR ctx->initData(x);
   CHKERR VecScatterBegin(ctx->sCat, x, ctx->X, INSERT_VALUES, SCATTER_REVERSE);
   CHKERR VecScatterEnd(ctx->sCat, x, ctx->X, INSERT_VALUES, SCATTER_REVERSE);
   CHKERR MatMult(ctx->A, ctx->X, ctx->F);
   CHKERR VecScatterBegin(ctx->sCat, ctx->F, f, MODE, SCATTER_FORWARD);
   CHKERR VecScatterEnd(ctx->sCat, ctx->F, f, MODE, SCATTER_FORWARD);
   PetscLogEventEnd(ctx->MOFEM_EVENT_mult, 0, 0, 0, 0);
   MoFEMFunctionReturn(0);
 }

◆ sub_mat_sor

MoFEMErrorCode sub_mat_sor	(	Mat	mat,
		Vec	b,
		PetscReal	omega,
		MatSORType	flag,
		PetscReal	shift,
		PetscInt	its,
		PetscInt	lits,
		Vec	x
	)

friend

Definition at line 168 of file PCMGSetUpViaApproxOrders.cpp.

                                   {
  
   //FIXME: that is crap implementation of SOR
  
   void *void_ctx;
   MoFEMFunctionBegin;
   CHKERR MatShellGetContext(mat, &void_ctx);
   PCMGSubMatrixCtx *ctx = (PCMGSubMatrixCtx *)void_ctx;
   PetscLogEventBegin(ctx->MOFEM_EVENT_sor, 0, 0, 0, 0);
   CHKERR ctx->initData(x);
   CHKERR VecScatterBegin(ctx->sCat, b, ctx->X, INSERT_VALUES, SCATTER_REVERSE);
   CHKERR VecScatterEnd(ctx->sCat, b, ctx->X, INSERT_VALUES, SCATTER_REVERSE);
   CHKERR MatSOR(ctx->A, ctx->X, omega, flag, shift, its, lits, ctx->F);
   CHKERR VecScatterBegin(ctx->sCat, ctx->F, x, INSERT_VALUES, SCATTER_FORWARD);
   CHKERR VecScatterEnd(ctx->sCat, ctx->F, x, INSERT_VALUES, SCATTER_FORWARD);
   PetscLogEventEnd(ctx->MOFEM_EVENT_sor, 0, 0, 0, 0);
   MoFEMFunctionReturn(0);
 }