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_NULLPTR);
  }
  MoFEMFunctionReturn(0);
}

Friends And Related Symbol 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);
}