xref: /petsc/src/ksp/pc/impls/mg/mg.c (revision 3821be0afdf874bf2c399643c829ed821dc5dd56)
1dba47a55SKris Buschelman 
24b9ad928SBarry Smith /*
34b9ad928SBarry Smith     Defines the multigrid preconditioner interface.
44b9ad928SBarry Smith */
5af0996ceSBarry Smith #include <petsc/private/pcmgimpl.h> /*I "petscksp.h" I*/
641b6fd38SMatthew G. Knepley #include <petsc/private/kspimpl.h>
71e25c274SJed Brown #include <petscdm.h>
8391689abSStefano Zampini PETSC_INTERN PetscErrorCode PCPreSolveChangeRHS(PC, PetscBool *);
94b9ad928SBarry Smith 
10f3b08a26SMatthew G. Knepley /*
11f3b08a26SMatthew G. Knepley    Contains the list of registered coarse space construction routines
12f3b08a26SMatthew G. Knepley */
13f3b08a26SMatthew G. Knepley PetscFunctionList PCMGCoarseList = NULL;
14f3b08a26SMatthew G. Knepley 
15d71ae5a4SJacob Faibussowitsch PetscErrorCode PCMGMCycle_Private(PC pc, PC_MG_Levels **mglevelsin, PetscBool transpose, PetscBool matapp, PCRichardsonConvergedReason *reason)
16d71ae5a4SJacob Faibussowitsch {
1731567311SBarry Smith   PC_MG        *mg = (PC_MG *)pc->data;
1831567311SBarry Smith   PC_MG_Levels *mgc, *mglevels = *mglevelsin;
1931567311SBarry Smith   PetscInt      cycles = (mglevels->level == 1) ? 1 : (PetscInt)mglevels->cycles;
204b9ad928SBarry Smith 
214b9ad928SBarry Smith   PetscFunctionBegin;
229566063dSJacob Faibussowitsch   if (mglevels->eventsmoothsolve) PetscCall(PetscLogEventBegin(mglevels->eventsmoothsolve, 0, 0, 0, 0));
23fcb023d4SJed Brown   if (!transpose) {
2430b0564aSStefano Zampini     if (matapp) {
259566063dSJacob Faibussowitsch       PetscCall(KSPMatSolve(mglevels->smoothd, mglevels->B, mglevels->X)); /* pre-smooth */
269566063dSJacob Faibussowitsch       PetscCall(KSPCheckSolve(mglevels->smoothd, pc, NULL));
2730b0564aSStefano Zampini     } else {
289566063dSJacob Faibussowitsch       PetscCall(KSPSolve(mglevels->smoothd, mglevels->b, mglevels->x)); /* pre-smooth */
299566063dSJacob Faibussowitsch       PetscCall(KSPCheckSolve(mglevels->smoothd, pc, mglevels->x));
3030b0564aSStefano Zampini     }
31fcb023d4SJed Brown   } else {
3228b400f6SJacob Faibussowitsch     PetscCheck(!matapp, PetscObjectComm((PetscObject)pc), PETSC_ERR_SUP, "Not supported");
339566063dSJacob Faibussowitsch     PetscCall(KSPSolveTranspose(mglevels->smoothu, mglevels->b, mglevels->x)); /* transpose of post-smooth */
349566063dSJacob Faibussowitsch     PetscCall(KSPCheckSolve(mglevels->smoothu, pc, mglevels->x));
35fcb023d4SJed Brown   }
369566063dSJacob Faibussowitsch   if (mglevels->eventsmoothsolve) PetscCall(PetscLogEventEnd(mglevels->eventsmoothsolve, 0, 0, 0, 0));
3731567311SBarry Smith   if (mglevels->level) { /* not the coarsest grid */
389566063dSJacob Faibussowitsch     if (mglevels->eventresidual) PetscCall(PetscLogEventBegin(mglevels->eventresidual, 0, 0, 0, 0));
3948a46eb9SPierre Jolivet     if (matapp && !mglevels->R) PetscCall(MatDuplicate(mglevels->B, MAT_DO_NOT_COPY_VALUES, &mglevels->R));
40fcb023d4SJed Brown     if (!transpose) {
419566063dSJacob Faibussowitsch       if (matapp) PetscCall((*mglevels->matresidual)(mglevels->A, mglevels->B, mglevels->X, mglevels->R));
429566063dSJacob Faibussowitsch       else PetscCall((*mglevels->residual)(mglevels->A, mglevels->b, mglevels->x, mglevels->r));
43fcb023d4SJed Brown     } else {
449566063dSJacob Faibussowitsch       if (matapp) PetscCall((*mglevels->matresidualtranspose)(mglevels->A, mglevels->B, mglevels->X, mglevels->R));
459566063dSJacob Faibussowitsch       else PetscCall((*mglevels->residualtranspose)(mglevels->A, mglevels->b, mglevels->x, mglevels->r));
46fcb023d4SJed Brown     }
479566063dSJacob Faibussowitsch     if (mglevels->eventresidual) PetscCall(PetscLogEventEnd(mglevels->eventresidual, 0, 0, 0, 0));
484b9ad928SBarry Smith 
494b9ad928SBarry Smith     /* if on finest level and have convergence criteria set */
5031567311SBarry Smith     if (mglevels->level == mglevels->levels - 1 && mg->ttol && reason) {
514b9ad928SBarry Smith       PetscReal rnorm;
529566063dSJacob Faibussowitsch       PetscCall(VecNorm(mglevels->r, NORM_2, &rnorm));
534b9ad928SBarry Smith       if (rnorm <= mg->ttol) {
5470441072SBarry Smith         if (rnorm < mg->abstol) {
554d0a8057SBarry Smith           *reason = PCRICHARDSON_CONVERGED_ATOL;
569566063dSJacob Faibussowitsch           PetscCall(PetscInfo(pc, "Linear solver has converged. Residual norm %g is less than absolute tolerance %g\n", (double)rnorm, (double)mg->abstol));
574b9ad928SBarry Smith         } else {
584d0a8057SBarry Smith           *reason = PCRICHARDSON_CONVERGED_RTOL;
599566063dSJacob Faibussowitsch           PetscCall(PetscInfo(pc, "Linear solver has converged. Residual norm %g is less than relative tolerance times initial residual norm %g\n", (double)rnorm, (double)mg->ttol));
604b9ad928SBarry Smith         }
613ba16761SJacob Faibussowitsch         PetscFunctionReturn(PETSC_SUCCESS);
624b9ad928SBarry Smith       }
634b9ad928SBarry Smith     }
644b9ad928SBarry Smith 
6531567311SBarry Smith     mgc = *(mglevelsin - 1);
669566063dSJacob Faibussowitsch     if (mglevels->eventinterprestrict) PetscCall(PetscLogEventBegin(mglevels->eventinterprestrict, 0, 0, 0, 0));
67fcb023d4SJed Brown     if (!transpose) {
689566063dSJacob Faibussowitsch       if (matapp) PetscCall(MatMatRestrict(mglevels->restrct, mglevels->R, &mgc->B));
699566063dSJacob Faibussowitsch       else PetscCall(MatRestrict(mglevels->restrct, mglevels->r, mgc->b));
70fcb023d4SJed Brown     } else {
719566063dSJacob Faibussowitsch       if (matapp) PetscCall(MatMatRestrict(mglevels->interpolate, mglevels->R, &mgc->B));
729566063dSJacob Faibussowitsch       else PetscCall(MatRestrict(mglevels->interpolate, mglevels->r, mgc->b));
73fcb023d4SJed Brown     }
749566063dSJacob Faibussowitsch     if (mglevels->eventinterprestrict) PetscCall(PetscLogEventEnd(mglevels->eventinterprestrict, 0, 0, 0, 0));
7530b0564aSStefano Zampini     if (matapp) {
7630b0564aSStefano Zampini       if (!mgc->X) {
779566063dSJacob Faibussowitsch         PetscCall(MatDuplicate(mgc->B, MAT_DO_NOT_COPY_VALUES, &mgc->X));
7830b0564aSStefano Zampini       } else {
799566063dSJacob Faibussowitsch         PetscCall(MatZeroEntries(mgc->X));
8030b0564aSStefano Zampini       }
8130b0564aSStefano Zampini     } else {
829566063dSJacob Faibussowitsch       PetscCall(VecZeroEntries(mgc->x));
8330b0564aSStefano Zampini     }
8448a46eb9SPierre Jolivet     while (cycles--) PetscCall(PCMGMCycle_Private(pc, mglevelsin - 1, transpose, matapp, reason));
859566063dSJacob Faibussowitsch     if (mglevels->eventinterprestrict) PetscCall(PetscLogEventBegin(mglevels->eventinterprestrict, 0, 0, 0, 0));
86fcb023d4SJed Brown     if (!transpose) {
879566063dSJacob Faibussowitsch       if (matapp) PetscCall(MatMatInterpolateAdd(mglevels->interpolate, mgc->X, mglevels->X, &mglevels->X));
889566063dSJacob Faibussowitsch       else PetscCall(MatInterpolateAdd(mglevels->interpolate, mgc->x, mglevels->x, mglevels->x));
89fcb023d4SJed Brown     } else {
909566063dSJacob Faibussowitsch       PetscCall(MatInterpolateAdd(mglevels->restrct, mgc->x, mglevels->x, mglevels->x));
91fcb023d4SJed Brown     }
929566063dSJacob Faibussowitsch     if (mglevels->eventinterprestrict) PetscCall(PetscLogEventEnd(mglevels->eventinterprestrict, 0, 0, 0, 0));
939566063dSJacob Faibussowitsch     if (mglevels->eventsmoothsolve) PetscCall(PetscLogEventBegin(mglevels->eventsmoothsolve, 0, 0, 0, 0));
94fcb023d4SJed Brown     if (!transpose) {
9530b0564aSStefano Zampini       if (matapp) {
969566063dSJacob Faibussowitsch         PetscCall(KSPMatSolve(mglevels->smoothu, mglevels->B, mglevels->X)); /* post smooth */
979566063dSJacob Faibussowitsch         PetscCall(KSPCheckSolve(mglevels->smoothu, pc, NULL));
9830b0564aSStefano Zampini       } else {
999566063dSJacob Faibussowitsch         PetscCall(KSPSolve(mglevels->smoothu, mglevels->b, mglevels->x)); /* post smooth */
1009566063dSJacob Faibussowitsch         PetscCall(KSPCheckSolve(mglevels->smoothu, pc, mglevels->x));
10130b0564aSStefano Zampini       }
102fcb023d4SJed Brown     } else {
10328b400f6SJacob Faibussowitsch       PetscCheck(!matapp, PetscObjectComm((PetscObject)pc), PETSC_ERR_SUP, "Not supported");
1049566063dSJacob Faibussowitsch       PetscCall(KSPSolveTranspose(mglevels->smoothd, mglevels->b, mglevels->x)); /* post smooth */
1059566063dSJacob Faibussowitsch       PetscCall(KSPCheckSolve(mglevels->smoothd, pc, mglevels->x));
106fcb023d4SJed Brown     }
10741b6fd38SMatthew G. Knepley     if (mglevels->cr) {
10828b400f6SJacob Faibussowitsch       PetscCheck(!matapp, PetscObjectComm((PetscObject)pc), PETSC_ERR_SUP, "Not supported");
10941b6fd38SMatthew G. Knepley       /* TODO Turn on copy and turn off noisy if we have an exact solution
1109566063dSJacob Faibussowitsch       PetscCall(VecCopy(mglevels->x, mglevels->crx));
1119566063dSJacob Faibussowitsch       PetscCall(VecCopy(mglevels->b, mglevels->crb)); */
1129566063dSJacob Faibussowitsch       PetscCall(KSPSetNoisy_Private(mglevels->crx));
1139566063dSJacob Faibussowitsch       PetscCall(KSPSolve(mglevels->cr, mglevels->crb, mglevels->crx)); /* compatible relaxation */
1149566063dSJacob Faibussowitsch       PetscCall(KSPCheckSolve(mglevels->cr, pc, mglevels->crx));
11541b6fd38SMatthew G. Knepley     }
1169566063dSJacob Faibussowitsch     if (mglevels->eventsmoothsolve) PetscCall(PetscLogEventEnd(mglevels->eventsmoothsolve, 0, 0, 0, 0));
1174b9ad928SBarry Smith   }
1183ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
1194b9ad928SBarry Smith }
1204b9ad928SBarry Smith 
121d71ae5a4SJacob Faibussowitsch static PetscErrorCode PCApplyRichardson_MG(PC pc, Vec b, Vec x, Vec w, PetscReal rtol, PetscReal abstol, PetscReal dtol, PetscInt its, PetscBool zeroguess, PetscInt *outits, PCRichardsonConvergedReason *reason)
122d71ae5a4SJacob Faibussowitsch {
123f3fbd535SBarry Smith   PC_MG         *mg       = (PC_MG *)pc->data;
124f3fbd535SBarry Smith   PC_MG_Levels **mglevels = mg->levels;
125391689abSStefano Zampini   PC             tpc;
126391689abSStefano Zampini   PetscBool      changeu, changed;
127f3fbd535SBarry Smith   PetscInt       levels = mglevels[0]->levels, i;
1284b9ad928SBarry Smith 
1294b9ad928SBarry Smith   PetscFunctionBegin;
130a762d673SBarry Smith   /* When the DM is supplying the matrix then it will not exist until here */
131a762d673SBarry Smith   for (i = 0; i < levels; i++) {
132a762d673SBarry Smith     if (!mglevels[i]->A) {
1339566063dSJacob Faibussowitsch       PetscCall(KSPGetOperators(mglevels[i]->smoothu, &mglevels[i]->A, NULL));
1349566063dSJacob Faibussowitsch       PetscCall(PetscObjectReference((PetscObject)mglevels[i]->A));
135a762d673SBarry Smith     }
136a762d673SBarry Smith   }
137391689abSStefano Zampini 
1389566063dSJacob Faibussowitsch   PetscCall(KSPGetPC(mglevels[levels - 1]->smoothd, &tpc));
1399566063dSJacob Faibussowitsch   PetscCall(PCPreSolveChangeRHS(tpc, &changed));
1409566063dSJacob Faibussowitsch   PetscCall(KSPGetPC(mglevels[levels - 1]->smoothu, &tpc));
1419566063dSJacob Faibussowitsch   PetscCall(PCPreSolveChangeRHS(tpc, &changeu));
142391689abSStefano Zampini   if (!changed && !changeu) {
1439566063dSJacob Faibussowitsch     PetscCall(VecDestroy(&mglevels[levels - 1]->b));
144f3fbd535SBarry Smith     mglevels[levels - 1]->b = b;
145391689abSStefano Zampini   } else { /* if the smoother changes the rhs during PreSolve, we cannot use the input vector */
146391689abSStefano Zampini     if (!mglevels[levels - 1]->b) {
147391689abSStefano Zampini       Vec *vec;
148391689abSStefano Zampini 
1499566063dSJacob Faibussowitsch       PetscCall(KSPCreateVecs(mglevels[levels - 1]->smoothd, 1, &vec, 0, NULL));
150391689abSStefano Zampini       mglevels[levels - 1]->b = *vec;
1519566063dSJacob Faibussowitsch       PetscCall(PetscFree(vec));
152391689abSStefano Zampini     }
1539566063dSJacob Faibussowitsch     PetscCall(VecCopy(b, mglevels[levels - 1]->b));
154391689abSStefano Zampini   }
155f3fbd535SBarry Smith   mglevels[levels - 1]->x = x;
1564b9ad928SBarry Smith 
15731567311SBarry Smith   mg->rtol   = rtol;
15831567311SBarry Smith   mg->abstol = abstol;
15931567311SBarry Smith   mg->dtol   = dtol;
1604b9ad928SBarry Smith   if (rtol) {
1614b9ad928SBarry Smith     /* compute initial residual norm for relative convergence test */
1624b9ad928SBarry Smith     PetscReal rnorm;
1637319c654SBarry Smith     if (zeroguess) {
1649566063dSJacob Faibussowitsch       PetscCall(VecNorm(b, NORM_2, &rnorm));
1657319c654SBarry Smith     } else {
1669566063dSJacob Faibussowitsch       PetscCall((*mglevels[levels - 1]->residual)(mglevels[levels - 1]->A, b, x, w));
1679566063dSJacob Faibussowitsch       PetscCall(VecNorm(w, NORM_2, &rnorm));
1687319c654SBarry Smith     }
16931567311SBarry Smith     mg->ttol = PetscMax(rtol * rnorm, abstol);
1702fa5cd67SKarl Rupp   } else if (abstol) mg->ttol = abstol;
1712fa5cd67SKarl Rupp   else mg->ttol = 0.0;
1724b9ad928SBarry Smith 
1734d0a8057SBarry Smith   /* since smoother is applied to full system, not just residual we need to make sure that smoothers don't
174336babb1SJed Brown      stop prematurely due to small residual */
1754d0a8057SBarry Smith   for (i = 1; i < levels; i++) {
1769566063dSJacob Faibussowitsch     PetscCall(KSPSetTolerances(mglevels[i]->smoothu, 0, PETSC_DEFAULT, PETSC_DEFAULT, PETSC_DEFAULT));
177f3fbd535SBarry Smith     if (mglevels[i]->smoothu != mglevels[i]->smoothd) {
17823067569SBarry Smith       /* For Richardson the initial guess is nonzero since it is solving in each cycle the original system not just applying as a preconditioner */
1799566063dSJacob Faibussowitsch       PetscCall(KSPSetInitialGuessNonzero(mglevels[i]->smoothd, PETSC_TRUE));
1809566063dSJacob Faibussowitsch       PetscCall(KSPSetTolerances(mglevels[i]->smoothd, 0, PETSC_DEFAULT, PETSC_DEFAULT, PETSC_DEFAULT));
1814b9ad928SBarry Smith     }
1824d0a8057SBarry Smith   }
1834d0a8057SBarry Smith 
1844d0a8057SBarry Smith   *reason = (PCRichardsonConvergedReason)0;
1854d0a8057SBarry Smith   for (i = 0; i < its; i++) {
1869566063dSJacob Faibussowitsch     PetscCall(PCMGMCycle_Private(pc, mglevels + levels - 1, PETSC_FALSE, PETSC_FALSE, reason));
1874d0a8057SBarry Smith     if (*reason) break;
1884d0a8057SBarry Smith   }
1894d0a8057SBarry Smith   if (!*reason) *reason = PCRICHARDSON_CONVERGED_ITS;
1904d0a8057SBarry Smith   *outits = i;
191391689abSStefano Zampini   if (!changed && !changeu) mglevels[levels - 1]->b = NULL;
1923ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
1934b9ad928SBarry Smith }
1944b9ad928SBarry Smith 
195d71ae5a4SJacob Faibussowitsch PetscErrorCode PCReset_MG(PC pc)
196d71ae5a4SJacob Faibussowitsch {
1973aeaf226SBarry Smith   PC_MG         *mg       = (PC_MG *)pc->data;
1983aeaf226SBarry Smith   PC_MG_Levels **mglevels = mg->levels;
1992b3cbbdaSStefano Zampini   PetscInt       i, n;
2003aeaf226SBarry Smith 
2013aeaf226SBarry Smith   PetscFunctionBegin;
2023aeaf226SBarry Smith   if (mglevels) {
2033aeaf226SBarry Smith     n = mglevels[0]->levels;
2043aeaf226SBarry Smith     for (i = 0; i < n - 1; i++) {
2059566063dSJacob Faibussowitsch       PetscCall(VecDestroy(&mglevels[i + 1]->r));
2069566063dSJacob Faibussowitsch       PetscCall(VecDestroy(&mglevels[i]->b));
2079566063dSJacob Faibussowitsch       PetscCall(VecDestroy(&mglevels[i]->x));
2089566063dSJacob Faibussowitsch       PetscCall(MatDestroy(&mglevels[i + 1]->R));
2099566063dSJacob Faibussowitsch       PetscCall(MatDestroy(&mglevels[i]->B));
2109566063dSJacob Faibussowitsch       PetscCall(MatDestroy(&mglevels[i]->X));
2119566063dSJacob Faibussowitsch       PetscCall(VecDestroy(&mglevels[i]->crx));
2129566063dSJacob Faibussowitsch       PetscCall(VecDestroy(&mglevels[i]->crb));
2139566063dSJacob Faibussowitsch       PetscCall(MatDestroy(&mglevels[i + 1]->restrct));
2149566063dSJacob Faibussowitsch       PetscCall(MatDestroy(&mglevels[i + 1]->interpolate));
2159566063dSJacob Faibussowitsch       PetscCall(MatDestroy(&mglevels[i + 1]->inject));
2169566063dSJacob Faibussowitsch       PetscCall(VecDestroy(&mglevels[i + 1]->rscale));
2173aeaf226SBarry Smith     }
2189566063dSJacob Faibussowitsch     PetscCall(VecDestroy(&mglevels[n - 1]->crx));
2199566063dSJacob Faibussowitsch     PetscCall(VecDestroy(&mglevels[n - 1]->crb));
220391689abSStefano Zampini     /* this is not null only if the smoother on the finest level
221391689abSStefano Zampini        changes the rhs during PreSolve */
2229566063dSJacob Faibussowitsch     PetscCall(VecDestroy(&mglevels[n - 1]->b));
2239566063dSJacob Faibussowitsch     PetscCall(MatDestroy(&mglevels[n - 1]->B));
2243aeaf226SBarry Smith 
2253aeaf226SBarry Smith     for (i = 0; i < n; i++) {
2262b3cbbdaSStefano Zampini       PetscCall(MatDestroy(&mglevels[i]->coarseSpace));
2279566063dSJacob Faibussowitsch       PetscCall(MatDestroy(&mglevels[i]->A));
22848a46eb9SPierre Jolivet       if (mglevels[i]->smoothd != mglevels[i]->smoothu) PetscCall(KSPReset(mglevels[i]->smoothd));
2299566063dSJacob Faibussowitsch       PetscCall(KSPReset(mglevels[i]->smoothu));
2309566063dSJacob Faibussowitsch       if (mglevels[i]->cr) PetscCall(KSPReset(mglevels[i]->cr));
2313aeaf226SBarry Smith     }
232f3b08a26SMatthew G. Knepley     mg->Nc = 0;
2333aeaf226SBarry Smith   }
2343ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
2353aeaf226SBarry Smith }
2363aeaf226SBarry Smith 
23741b6fd38SMatthew G. Knepley /* Implementing CR
23841b6fd38SMatthew G. Knepley 
23941b6fd38SMatthew G. Knepley We only want to make corrections that ``do not change'' the coarse solution. What we mean by not changing is that if I prolong my coarse solution to the fine grid and then inject that fine solution back to the coarse grid, I get the same answer. Injection is what Brannick calls R. We want the complementary projector to Inj, which we will call S, after Brannick, so that Inj S = 0. Now the orthogonal projector onto the range of Inj^T is
24041b6fd38SMatthew G. Knepley 
24141b6fd38SMatthew G. Knepley   Inj^T (Inj Inj^T)^{-1} Inj
24241b6fd38SMatthew G. Knepley 
24341b6fd38SMatthew G. Knepley and if Inj is a VecScatter, as it is now in PETSc, we have
24441b6fd38SMatthew G. Knepley 
24541b6fd38SMatthew G. Knepley   Inj^T Inj
24641b6fd38SMatthew G. Knepley 
24741b6fd38SMatthew G. Knepley and
24841b6fd38SMatthew G. Knepley 
24941b6fd38SMatthew G. Knepley   S = I - Inj^T Inj
25041b6fd38SMatthew G. Knepley 
25141b6fd38SMatthew G. Knepley since
25241b6fd38SMatthew G. Knepley 
25341b6fd38SMatthew G. Knepley   Inj S = Inj - (Inj Inj^T) Inj = 0.
25441b6fd38SMatthew G. Knepley 
25541b6fd38SMatthew G. Knepley Brannick suggests
25641b6fd38SMatthew G. Knepley 
25741b6fd38SMatthew G. Knepley   A \to S^T A S  \qquad\mathrm{and}\qquad M \to S^T M S
25841b6fd38SMatthew G. Knepley 
25941b6fd38SMatthew G. Knepley but I do not think his :math:`S^T S = I` is correct. Our S is an orthogonal projector, so :math:`S^T S = S^2 = S`. We will use
26041b6fd38SMatthew G. Knepley 
26141b6fd38SMatthew G. Knepley   M^{-1} A \to S M^{-1} A S
26241b6fd38SMatthew G. Knepley 
26341b6fd38SMatthew G. Knepley In fact, since it is somewhat hard in PETSc to do the symmetric application, we will just apply S on the left.
26441b6fd38SMatthew G. Knepley 
26541b6fd38SMatthew G. Knepley   Check: || Inj P - I ||_F < tol
26641b6fd38SMatthew G. Knepley   Check: In general, Inj Inj^T = I
26741b6fd38SMatthew G. Knepley */
26841b6fd38SMatthew G. Knepley 
26941b6fd38SMatthew G. Knepley typedef struct {
27041b6fd38SMatthew G. Knepley   PC       mg;  /* The PCMG object */
27141b6fd38SMatthew G. Knepley   PetscInt l;   /* The multigrid level for this solver */
27241b6fd38SMatthew G. Knepley   Mat      Inj; /* The injection matrix */
27341b6fd38SMatthew G. Knepley   Mat      S;   /* I - Inj^T Inj */
27441b6fd38SMatthew G. Knepley } CRContext;
27541b6fd38SMatthew G. Knepley 
276d71ae5a4SJacob Faibussowitsch static PetscErrorCode CRSetup_Private(PC pc)
277d71ae5a4SJacob Faibussowitsch {
27841b6fd38SMatthew G. Knepley   CRContext *ctx;
27941b6fd38SMatthew G. Knepley   Mat        It;
28041b6fd38SMatthew G. Knepley 
28141b6fd38SMatthew G. Knepley   PetscFunctionBeginUser;
2829566063dSJacob Faibussowitsch   PetscCall(PCShellGetContext(pc, &ctx));
2839566063dSJacob Faibussowitsch   PetscCall(PCMGGetInjection(ctx->mg, ctx->l, &It));
28428b400f6SJacob Faibussowitsch   PetscCheck(It, PetscObjectComm((PetscObject)pc), PETSC_ERR_ARG_WRONGSTATE, "CR requires that injection be defined for this PCMG");
2859566063dSJacob Faibussowitsch   PetscCall(MatCreateTranspose(It, &ctx->Inj));
2869566063dSJacob Faibussowitsch   PetscCall(MatCreateNormal(ctx->Inj, &ctx->S));
2879566063dSJacob Faibussowitsch   PetscCall(MatScale(ctx->S, -1.0));
2889566063dSJacob Faibussowitsch   PetscCall(MatShift(ctx->S, 1.0));
2893ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
29041b6fd38SMatthew G. Knepley }
29141b6fd38SMatthew G. Knepley 
292d71ae5a4SJacob Faibussowitsch static PetscErrorCode CRApply_Private(PC pc, Vec x, Vec y)
293d71ae5a4SJacob Faibussowitsch {
29441b6fd38SMatthew G. Knepley   CRContext *ctx;
29541b6fd38SMatthew G. Knepley 
29641b6fd38SMatthew G. Knepley   PetscFunctionBeginUser;
2979566063dSJacob Faibussowitsch   PetscCall(PCShellGetContext(pc, &ctx));
2989566063dSJacob Faibussowitsch   PetscCall(MatMult(ctx->S, x, y));
2993ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
30041b6fd38SMatthew G. Knepley }
30141b6fd38SMatthew G. Knepley 
302d71ae5a4SJacob Faibussowitsch static PetscErrorCode CRDestroy_Private(PC pc)
303d71ae5a4SJacob Faibussowitsch {
30441b6fd38SMatthew G. Knepley   CRContext *ctx;
30541b6fd38SMatthew G. Knepley 
30641b6fd38SMatthew G. Knepley   PetscFunctionBeginUser;
3079566063dSJacob Faibussowitsch   PetscCall(PCShellGetContext(pc, &ctx));
3089566063dSJacob Faibussowitsch   PetscCall(MatDestroy(&ctx->Inj));
3099566063dSJacob Faibussowitsch   PetscCall(MatDestroy(&ctx->S));
3109566063dSJacob Faibussowitsch   PetscCall(PetscFree(ctx));
3119566063dSJacob Faibussowitsch   PetscCall(PCShellSetContext(pc, NULL));
3123ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
31341b6fd38SMatthew G. Knepley }
31441b6fd38SMatthew G. Knepley 
315d71ae5a4SJacob Faibussowitsch static PetscErrorCode CreateCR_Private(PC pc, PetscInt l, PC *cr)
316d71ae5a4SJacob Faibussowitsch {
31741b6fd38SMatthew G. Knepley   CRContext *ctx;
31841b6fd38SMatthew G. Knepley 
31941b6fd38SMatthew G. Knepley   PetscFunctionBeginUser;
3209566063dSJacob Faibussowitsch   PetscCall(PCCreate(PetscObjectComm((PetscObject)pc), cr));
3219566063dSJacob Faibussowitsch   PetscCall(PetscObjectSetName((PetscObject)*cr, "S (complementary projector to injection)"));
3229566063dSJacob Faibussowitsch   PetscCall(PetscCalloc1(1, &ctx));
32341b6fd38SMatthew G. Knepley   ctx->mg = pc;
32441b6fd38SMatthew G. Knepley   ctx->l  = l;
3259566063dSJacob Faibussowitsch   PetscCall(PCSetType(*cr, PCSHELL));
3269566063dSJacob Faibussowitsch   PetscCall(PCShellSetContext(*cr, ctx));
3279566063dSJacob Faibussowitsch   PetscCall(PCShellSetApply(*cr, CRApply_Private));
3289566063dSJacob Faibussowitsch   PetscCall(PCShellSetSetUp(*cr, CRSetup_Private));
3299566063dSJacob Faibussowitsch   PetscCall(PCShellSetDestroy(*cr, CRDestroy_Private));
3303ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
33141b6fd38SMatthew G. Knepley }
33241b6fd38SMatthew G. Knepley 
333d71ae5a4SJacob Faibussowitsch PetscErrorCode PCMGSetLevels_MG(PC pc, PetscInt levels, MPI_Comm *comms)
334d71ae5a4SJacob Faibussowitsch {
335f3fbd535SBarry Smith   PC_MG         *mg = (PC_MG *)pc->data;
336ce94432eSBarry Smith   MPI_Comm       comm;
3373aeaf226SBarry Smith   PC_MG_Levels **mglevels = mg->levels;
33810eca3edSLisandro Dalcin   PCMGType       mgtype   = mg->am;
33910167fecSBarry Smith   PetscInt       mgctype  = (PetscInt)PC_MG_CYCLE_V;
340f3fbd535SBarry Smith   PetscInt       i;
341f3fbd535SBarry Smith   PetscMPIInt    size;
342f3fbd535SBarry Smith   const char    *prefix;
343f3fbd535SBarry Smith   PC             ipc;
3443aeaf226SBarry Smith   PetscInt       n;
3454b9ad928SBarry Smith 
3464b9ad928SBarry Smith   PetscFunctionBegin;
3470700a824SBarry Smith   PetscValidHeaderSpecific(pc, PC_CLASSID, 1);
348548767e0SJed Brown   PetscValidLogicalCollectiveInt(pc, levels, 2);
3493ba16761SJacob Faibussowitsch   if (mg->nlevels == levels) PetscFunctionReturn(PETSC_SUCCESS);
3509566063dSJacob Faibussowitsch   PetscCall(PetscObjectGetComm((PetscObject)pc, &comm));
3513aeaf226SBarry Smith   if (mglevels) {
35210eca3edSLisandro Dalcin     mgctype = mglevels[0]->cycles;
3533aeaf226SBarry Smith     /* changing the number of levels so free up the previous stuff */
3549566063dSJacob Faibussowitsch     PetscCall(PCReset_MG(pc));
3553aeaf226SBarry Smith     n = mglevels[0]->levels;
3563aeaf226SBarry Smith     for (i = 0; i < n; i++) {
35748a46eb9SPierre Jolivet       if (mglevels[i]->smoothd != mglevels[i]->smoothu) PetscCall(KSPDestroy(&mglevels[i]->smoothd));
3589566063dSJacob Faibussowitsch       PetscCall(KSPDestroy(&mglevels[i]->smoothu));
3599566063dSJacob Faibussowitsch       PetscCall(KSPDestroy(&mglevels[i]->cr));
3609566063dSJacob Faibussowitsch       PetscCall(PetscFree(mglevels[i]));
3613aeaf226SBarry Smith     }
3629566063dSJacob Faibussowitsch     PetscCall(PetscFree(mg->levels));
3633aeaf226SBarry Smith   }
364f3fbd535SBarry Smith 
365f3fbd535SBarry Smith   mg->nlevels = levels;
366f3fbd535SBarry Smith 
3679566063dSJacob Faibussowitsch   PetscCall(PetscMalloc1(levels, &mglevels));
368f3fbd535SBarry Smith 
3699566063dSJacob Faibussowitsch   PetscCall(PCGetOptionsPrefix(pc, &prefix));
370f3fbd535SBarry Smith 
371a9db87a2SMatthew G Knepley   mg->stageApply = 0;
372f3fbd535SBarry Smith   for (i = 0; i < levels; i++) {
3734dfa11a4SJacob Faibussowitsch     PetscCall(PetscNew(&mglevels[i]));
3742fa5cd67SKarl Rupp 
375f3fbd535SBarry Smith     mglevels[i]->level               = i;
376f3fbd535SBarry Smith     mglevels[i]->levels              = levels;
37710eca3edSLisandro Dalcin     mglevels[i]->cycles              = mgctype;
378336babb1SJed Brown     mg->default_smoothu              = 2;
379336babb1SJed Brown     mg->default_smoothd              = 2;
38063e6d426SJed Brown     mglevels[i]->eventsmoothsetup    = 0;
38163e6d426SJed Brown     mglevels[i]->eventsmoothsolve    = 0;
38263e6d426SJed Brown     mglevels[i]->eventresidual       = 0;
38363e6d426SJed Brown     mglevels[i]->eventinterprestrict = 0;
384f3fbd535SBarry Smith 
385f3fbd535SBarry Smith     if (comms) comm = comms[i];
386d5a8f7e6SBarry Smith     if (comm != MPI_COMM_NULL) {
3879566063dSJacob Faibussowitsch       PetscCall(KSPCreate(comm, &mglevels[i]->smoothd));
388*3821be0aSBarry Smith       PetscCall(KSPSetNestLevel(mglevels[i]->smoothd, pc->kspnestlevel));
3899566063dSJacob Faibussowitsch       PetscCall(KSPSetErrorIfNotConverged(mglevels[i]->smoothd, pc->erroriffailure));
3909566063dSJacob Faibussowitsch       PetscCall(PetscObjectIncrementTabLevel((PetscObject)mglevels[i]->smoothd, (PetscObject)pc, levels - i));
3919566063dSJacob Faibussowitsch       PetscCall(KSPSetOptionsPrefix(mglevels[i]->smoothd, prefix));
3929566063dSJacob Faibussowitsch       PetscCall(PetscObjectComposedDataSetInt((PetscObject)mglevels[i]->smoothd, PetscMGLevelId, mglevels[i]->level));
3930ee9a668SBarry Smith       if (i || levels == 1) {
3940ee9a668SBarry Smith         char tprefix[128];
3950ee9a668SBarry Smith 
3969566063dSJacob Faibussowitsch         PetscCall(KSPSetType(mglevels[i]->smoothd, KSPCHEBYSHEV));
3979566063dSJacob Faibussowitsch         PetscCall(KSPSetConvergenceTest(mglevels[i]->smoothd, KSPConvergedSkip, NULL, NULL));
3989566063dSJacob Faibussowitsch         PetscCall(KSPSetNormType(mglevels[i]->smoothd, KSP_NORM_NONE));
3999566063dSJacob Faibussowitsch         PetscCall(KSPGetPC(mglevels[i]->smoothd, &ipc));
4009566063dSJacob Faibussowitsch         PetscCall(PCSetType(ipc, PCSOR));
4019566063dSJacob Faibussowitsch         PetscCall(KSPSetTolerances(mglevels[i]->smoothd, PETSC_DEFAULT, PETSC_DEFAULT, PETSC_DEFAULT, mg->default_smoothd));
402f3fbd535SBarry Smith 
4039566063dSJacob Faibussowitsch         PetscCall(PetscSNPrintf(tprefix, 128, "mg_levels_%d_", (int)i));
4049566063dSJacob Faibussowitsch         PetscCall(KSPAppendOptionsPrefix(mglevels[i]->smoothd, tprefix));
4050ee9a668SBarry Smith       } else {
4069566063dSJacob Faibussowitsch         PetscCall(KSPAppendOptionsPrefix(mglevels[0]->smoothd, "mg_coarse_"));
407f3fbd535SBarry Smith 
4089dbfc187SHong Zhang         /* coarse solve is (redundant) LU by default; set shifttype NONZERO to avoid annoying zero-pivot in LU preconditioner */
4099566063dSJacob Faibussowitsch         PetscCall(KSPSetType(mglevels[0]->smoothd, KSPPREONLY));
4109566063dSJacob Faibussowitsch         PetscCall(KSPGetPC(mglevels[0]->smoothd, &ipc));
4119566063dSJacob Faibussowitsch         PetscCallMPI(MPI_Comm_size(comm, &size));
412f3fbd535SBarry Smith         if (size > 1) {
4139566063dSJacob Faibussowitsch           PetscCall(PCSetType(ipc, PCREDUNDANT));
414f3fbd535SBarry Smith         } else {
4159566063dSJacob Faibussowitsch           PetscCall(PCSetType(ipc, PCLU));
416f3fbd535SBarry Smith         }
4179566063dSJacob Faibussowitsch         PetscCall(PCFactorSetShiftType(ipc, MAT_SHIFT_INBLOCKS));
418f3fbd535SBarry Smith       }
419d5a8f7e6SBarry Smith     }
420f3fbd535SBarry Smith     mglevels[i]->smoothu = mglevels[i]->smoothd;
42131567311SBarry Smith     mg->rtol             = 0.0;
42231567311SBarry Smith     mg->abstol           = 0.0;
42331567311SBarry Smith     mg->dtol             = 0.0;
42431567311SBarry Smith     mg->ttol             = 0.0;
42531567311SBarry Smith     mg->cyclesperpcapply = 1;
426f3fbd535SBarry Smith   }
427f3fbd535SBarry Smith   mg->levels = mglevels;
4289566063dSJacob Faibussowitsch   PetscCall(PCMGSetType(pc, mgtype));
4293ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
4304b9ad928SBarry Smith }
4314b9ad928SBarry Smith 
43297d33e41SMatthew G. Knepley /*@C
433f1580f4eSBarry Smith   PCMGSetLevels - Sets the number of levels to use with `PCMG`.
434f1580f4eSBarry Smith   Must be called before any other `PCMG` routine.
43597d33e41SMatthew G. Knepley 
436c3339decSBarry Smith   Logically Collective
43797d33e41SMatthew G. Knepley 
43897d33e41SMatthew G. Knepley   Input Parameters:
43997d33e41SMatthew G. Knepley + pc     - the preconditioner context
44097d33e41SMatthew G. Knepley . levels - the number of levels
44197d33e41SMatthew G. Knepley - comms  - optional communicators for each level; this is to allow solving the coarser problems
442d5a8f7e6SBarry Smith            on smaller sets of processes. For processes that are not included in the computation
44320f4b53cSBarry Smith            you must pass `MPI_COMM_NULL`. Use comms = `NULL` to specify that all processes
44405552d4bSJunchao Zhang            should participate in each level of problem.
44597d33e41SMatthew G. Knepley 
44697d33e41SMatthew G. Knepley   Level: intermediate
44797d33e41SMatthew G. Knepley 
44897d33e41SMatthew G. Knepley   Notes:
44920f4b53cSBarry Smith   If the number of levels is one then the multigrid uses the `-mg_levels` prefix
45020f4b53cSBarry Smith   for setting the level options rather than the `-mg_coarse` prefix.
45197d33e41SMatthew G. Knepley 
452d5a8f7e6SBarry Smith   You can free the information in comms after this routine is called.
453d5a8f7e6SBarry Smith 
454f1580f4eSBarry Smith   The array of MPI communicators must contain `MPI_COMM_NULL` for those ranks that at each level
455d5a8f7e6SBarry Smith   are not participating in the coarser solve. For example, with 2 levels and 1 and 2 ranks on
456d5a8f7e6SBarry Smith   the two levels, rank 0 in the original communicator will pass in an array of 2 communicators
457d5a8f7e6SBarry Smith   of size 2 and 1, while rank 1 in the original communicator will pass in array of 2 communicators
458f1580f4eSBarry Smith   the first of size 2 and the second of value `MPI_COMM_NULL` since the rank 1 does not participate
459d5a8f7e6SBarry Smith   in the coarse grid solve.
460d5a8f7e6SBarry Smith 
461f1580f4eSBarry Smith   Since each coarser level may have a new `MPI_Comm` with fewer ranks than the previous, one
462d5a8f7e6SBarry Smith   must take special care in providing the restriction and interpolation operation. We recommend
463d5a8f7e6SBarry Smith   providing these as two step operations; first perform a standard restriction or interpolation on
464d5a8f7e6SBarry Smith   the full number of ranks for that level and then use an MPI call to copy the resulting vector
46505552d4bSJunchao Zhang   array entries (after calls to VecGetArray()) to the smaller or larger number of ranks, note in both
466d5a8f7e6SBarry Smith   cases the MPI calls must be made on the larger of the two communicators. Traditional MPI send and
46720f4b53cSBarry Smith   receives or `MPI_AlltoAllv()` could be used to do the reshuffling of the vector entries.
468d5a8f7e6SBarry Smith 
469feefa0e1SJacob Faibussowitsch   Fortran Notes:
47020f4b53cSBarry Smith   Use comms = `PETSC_NULL_MPI_COMM` as the equivalent of `NULL` in the C interface. Note `PETSC_NULL_MPI_COMM`
471f1580f4eSBarry Smith   is not `MPI_COMM_NULL`. It is more like `PETSC_NULL_INTEGER`, `PETSC_NULL_REAL` etc.
472d5a8f7e6SBarry Smith 
473db781477SPatrick Sanan .seealso: `PCMGSetType()`, `PCMGGetLevels()`
47497d33e41SMatthew G. Knepley @*/
475d71ae5a4SJacob Faibussowitsch PetscErrorCode PCMGSetLevels(PC pc, PetscInt levels, MPI_Comm *comms)
476d71ae5a4SJacob Faibussowitsch {
47797d33e41SMatthew G. Knepley   PetscFunctionBegin;
47897d33e41SMatthew G. Knepley   PetscValidHeaderSpecific(pc, PC_CLASSID, 1);
4794f572ea9SToby Isaac   if (comms) PetscAssertPointer(comms, 3);
480cac4c232SBarry Smith   PetscTryMethod(pc, "PCMGSetLevels_C", (PC, PetscInt, MPI_Comm *), (pc, levels, comms));
4813ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
48297d33e41SMatthew G. Knepley }
48397d33e41SMatthew G. Knepley 
484d71ae5a4SJacob Faibussowitsch PetscErrorCode PCDestroy_MG(PC pc)
485d71ae5a4SJacob Faibussowitsch {
486a06653b4SBarry Smith   PC_MG         *mg       = (PC_MG *)pc->data;
487a06653b4SBarry Smith   PC_MG_Levels **mglevels = mg->levels;
488a06653b4SBarry Smith   PetscInt       i, n;
489c07bf074SBarry Smith 
490c07bf074SBarry Smith   PetscFunctionBegin;
4919566063dSJacob Faibussowitsch   PetscCall(PCReset_MG(pc));
492a06653b4SBarry Smith   if (mglevels) {
493a06653b4SBarry Smith     n = mglevels[0]->levels;
494a06653b4SBarry Smith     for (i = 0; i < n; i++) {
49548a46eb9SPierre Jolivet       if (mglevels[i]->smoothd != mglevels[i]->smoothu) PetscCall(KSPDestroy(&mglevels[i]->smoothd));
4969566063dSJacob Faibussowitsch       PetscCall(KSPDestroy(&mglevels[i]->smoothu));
4979566063dSJacob Faibussowitsch       PetscCall(KSPDestroy(&mglevels[i]->cr));
4989566063dSJacob Faibussowitsch       PetscCall(PetscFree(mglevels[i]));
499a06653b4SBarry Smith     }
5009566063dSJacob Faibussowitsch     PetscCall(PetscFree(mg->levels));
501a06653b4SBarry Smith   }
5029566063dSJacob Faibussowitsch   PetscCall(PetscFree(pc->data));
5039566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)pc, "PCGetInterpolations_C", NULL));
5049566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)pc, "PCGetCoarseOperators_C", NULL));
5052b3cbbdaSStefano Zampini   PetscCall(PetscObjectComposeFunction((PetscObject)pc, "PCMGSetGalerkin_C", NULL));
5062b3cbbdaSStefano Zampini   PetscCall(PetscObjectComposeFunction((PetscObject)pc, "PCMGGetLevels_C", NULL));
5072b3cbbdaSStefano Zampini   PetscCall(PetscObjectComposeFunction((PetscObject)pc, "PCMGSetLevels_C", NULL));
5082b3cbbdaSStefano Zampini   PetscCall(PetscObjectComposeFunction((PetscObject)pc, "PCGetInterpolations_C", NULL));
5092b3cbbdaSStefano Zampini   PetscCall(PetscObjectComposeFunction((PetscObject)pc, "PCGetCoarseOperators_C", NULL));
5102b3cbbdaSStefano Zampini   PetscCall(PetscObjectComposeFunction((PetscObject)pc, "PCMGSetAdaptInterpolation_C", NULL));
5112b3cbbdaSStefano Zampini   PetscCall(PetscObjectComposeFunction((PetscObject)pc, "PCMGGetAdaptInterpolation_C", NULL));
5122b3cbbdaSStefano Zampini   PetscCall(PetscObjectComposeFunction((PetscObject)pc, "PCMGSetAdaptCR_C", NULL));
5132b3cbbdaSStefano Zampini   PetscCall(PetscObjectComposeFunction((PetscObject)pc, "PCMGGetAdaptCR_C", NULL));
5142b3cbbdaSStefano Zampini   PetscCall(PetscObjectComposeFunction((PetscObject)pc, "PCMGSetAdaptCoarseSpaceType_C", NULL));
5152b3cbbdaSStefano Zampini   PetscCall(PetscObjectComposeFunction((PetscObject)pc, "PCMGGetAdaptCoarseSpaceType_C", NULL));
5163ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
517f3fbd535SBarry Smith }
518f3fbd535SBarry Smith 
519f3fbd535SBarry Smith /*
520f3fbd535SBarry Smith    PCApply_MG - Runs either an additive, multiplicative, Kaskadic
521f3fbd535SBarry Smith              or full cycle of multigrid.
522f3fbd535SBarry Smith 
523f3fbd535SBarry Smith   Note:
524f3fbd535SBarry Smith   A simple wrapper which calls PCMGMCycle(),PCMGACycle(), or PCMGFCycle().
525f3fbd535SBarry Smith */
526d71ae5a4SJacob Faibussowitsch static PetscErrorCode PCApply_MG_Internal(PC pc, Vec b, Vec x, Mat B, Mat X, PetscBool transpose)
527d71ae5a4SJacob Faibussowitsch {
528f3fbd535SBarry Smith   PC_MG         *mg       = (PC_MG *)pc->data;
529f3fbd535SBarry Smith   PC_MG_Levels **mglevels = mg->levels;
530391689abSStefano Zampini   PC             tpc;
531f3fbd535SBarry Smith   PetscInt       levels = mglevels[0]->levels, i;
53230b0564aSStefano Zampini   PetscBool      changeu, changed, matapp;
533f3fbd535SBarry Smith 
534f3fbd535SBarry Smith   PetscFunctionBegin;
53530b0564aSStefano Zampini   matapp = (PetscBool)(B && X);
5369566063dSJacob Faibussowitsch   if (mg->stageApply) PetscCall(PetscLogStagePush(mg->stageApply));
537e1d8e5deSBarry Smith   /* When the DM is supplying the matrix then it will not exist until here */
538298cc208SBarry Smith   for (i = 0; i < levels; i++) {
539e1d8e5deSBarry Smith     if (!mglevels[i]->A) {
5409566063dSJacob Faibussowitsch       PetscCall(KSPGetOperators(mglevels[i]->smoothu, &mglevels[i]->A, NULL));
5419566063dSJacob Faibussowitsch       PetscCall(PetscObjectReference((PetscObject)mglevels[i]->A));
542e1d8e5deSBarry Smith     }
543e1d8e5deSBarry Smith   }
544e1d8e5deSBarry Smith 
5459566063dSJacob Faibussowitsch   PetscCall(KSPGetPC(mglevels[levels - 1]->smoothd, &tpc));
5469566063dSJacob Faibussowitsch   PetscCall(PCPreSolveChangeRHS(tpc, &changed));
5479566063dSJacob Faibussowitsch   PetscCall(KSPGetPC(mglevels[levels - 1]->smoothu, &tpc));
5489566063dSJacob Faibussowitsch   PetscCall(PCPreSolveChangeRHS(tpc, &changeu));
549391689abSStefano Zampini   if (!changeu && !changed) {
55030b0564aSStefano Zampini     if (matapp) {
5519566063dSJacob Faibussowitsch       PetscCall(MatDestroy(&mglevels[levels - 1]->B));
55230b0564aSStefano Zampini       mglevels[levels - 1]->B = B;
55330b0564aSStefano Zampini     } else {
5549566063dSJacob Faibussowitsch       PetscCall(VecDestroy(&mglevels[levels - 1]->b));
555f3fbd535SBarry Smith       mglevels[levels - 1]->b = b;
55630b0564aSStefano Zampini     }
557391689abSStefano Zampini   } else { /* if the smoother changes the rhs during PreSolve, we cannot use the input vector */
55830b0564aSStefano Zampini     if (matapp) {
55930b0564aSStefano Zampini       if (mglevels[levels - 1]->B) {
56030b0564aSStefano Zampini         PetscInt  N1, N2;
56130b0564aSStefano Zampini         PetscBool flg;
56230b0564aSStefano Zampini 
5639566063dSJacob Faibussowitsch         PetscCall(MatGetSize(mglevels[levels - 1]->B, NULL, &N1));
5649566063dSJacob Faibussowitsch         PetscCall(MatGetSize(B, NULL, &N2));
5659566063dSJacob Faibussowitsch         PetscCall(PetscObjectTypeCompare((PetscObject)mglevels[levels - 1]->B, ((PetscObject)B)->type_name, &flg));
56648a46eb9SPierre Jolivet         if (N1 != N2 || !flg) PetscCall(MatDestroy(&mglevels[levels - 1]->B));
56730b0564aSStefano Zampini       }
56830b0564aSStefano Zampini       if (!mglevels[levels - 1]->B) {
5699566063dSJacob Faibussowitsch         PetscCall(MatDuplicate(B, MAT_COPY_VALUES, &mglevels[levels - 1]->B));
57030b0564aSStefano Zampini       } else {
5719566063dSJacob Faibussowitsch         PetscCall(MatCopy(B, mglevels[levels - 1]->B, SAME_NONZERO_PATTERN));
57230b0564aSStefano Zampini       }
57330b0564aSStefano Zampini     } else {
574391689abSStefano Zampini       if (!mglevels[levels - 1]->b) {
575391689abSStefano Zampini         Vec *vec;
576391689abSStefano Zampini 
5779566063dSJacob Faibussowitsch         PetscCall(KSPCreateVecs(mglevels[levels - 1]->smoothd, 1, &vec, 0, NULL));
578391689abSStefano Zampini         mglevels[levels - 1]->b = *vec;
5799566063dSJacob Faibussowitsch         PetscCall(PetscFree(vec));
580391689abSStefano Zampini       }
5819566063dSJacob Faibussowitsch       PetscCall(VecCopy(b, mglevels[levels - 1]->b));
582391689abSStefano Zampini     }
58330b0564aSStefano Zampini   }
5849371c9d4SSatish Balay   if (matapp) {
5859371c9d4SSatish Balay     mglevels[levels - 1]->X = X;
5869371c9d4SSatish Balay   } else {
5879371c9d4SSatish Balay     mglevels[levels - 1]->x = x;
5889371c9d4SSatish Balay   }
58930b0564aSStefano Zampini 
59030b0564aSStefano Zampini   /* If coarser Xs are present, it means we have already block applied the PC at least once
59130b0564aSStefano Zampini      Reset operators if sizes/type do no match */
59230b0564aSStefano Zampini   if (matapp && levels > 1 && mglevels[levels - 2]->X) {
59330b0564aSStefano Zampini     PetscInt  Xc, Bc;
59430b0564aSStefano Zampini     PetscBool flg;
59530b0564aSStefano Zampini 
5969566063dSJacob Faibussowitsch     PetscCall(MatGetSize(mglevels[levels - 2]->X, NULL, &Xc));
5979566063dSJacob Faibussowitsch     PetscCall(MatGetSize(mglevels[levels - 1]->B, NULL, &Bc));
5989566063dSJacob Faibussowitsch     PetscCall(PetscObjectTypeCompare((PetscObject)mglevels[levels - 2]->X, ((PetscObject)mglevels[levels - 1]->X)->type_name, &flg));
59930b0564aSStefano Zampini     if (Xc != Bc || !flg) {
6009566063dSJacob Faibussowitsch       PetscCall(MatDestroy(&mglevels[levels - 1]->R));
60130b0564aSStefano Zampini       for (i = 0; i < levels - 1; i++) {
6029566063dSJacob Faibussowitsch         PetscCall(MatDestroy(&mglevels[i]->R));
6039566063dSJacob Faibussowitsch         PetscCall(MatDestroy(&mglevels[i]->B));
6049566063dSJacob Faibussowitsch         PetscCall(MatDestroy(&mglevels[i]->X));
60530b0564aSStefano Zampini       }
60630b0564aSStefano Zampini     }
60730b0564aSStefano Zampini   }
608391689abSStefano Zampini 
60931567311SBarry Smith   if (mg->am == PC_MG_MULTIPLICATIVE) {
6109566063dSJacob Faibussowitsch     if (matapp) PetscCall(MatZeroEntries(X));
6119566063dSJacob Faibussowitsch     else PetscCall(VecZeroEntries(x));
61248a46eb9SPierre Jolivet     for (i = 0; i < mg->cyclesperpcapply; i++) PetscCall(PCMGMCycle_Private(pc, mglevels + levels - 1, transpose, matapp, NULL));
6132fa5cd67SKarl Rupp   } else if (mg->am == PC_MG_ADDITIVE) {
6149566063dSJacob Faibussowitsch     PetscCall(PCMGACycle_Private(pc, mglevels, transpose, matapp));
6152fa5cd67SKarl Rupp   } else if (mg->am == PC_MG_KASKADE) {
6169566063dSJacob Faibussowitsch     PetscCall(PCMGKCycle_Private(pc, mglevels, transpose, matapp));
6172fa5cd67SKarl Rupp   } else {
6189566063dSJacob Faibussowitsch     PetscCall(PCMGFCycle_Private(pc, mglevels, transpose, matapp));
619f3fbd535SBarry Smith   }
6209566063dSJacob Faibussowitsch   if (mg->stageApply) PetscCall(PetscLogStagePop());
62130b0564aSStefano Zampini   if (!changeu && !changed) {
6229371c9d4SSatish Balay     if (matapp) {
6239371c9d4SSatish Balay       mglevels[levels - 1]->B = NULL;
6249371c9d4SSatish Balay     } else {
6259371c9d4SSatish Balay       mglevels[levels - 1]->b = NULL;
6269371c9d4SSatish Balay     }
62730b0564aSStefano Zampini   }
6283ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
629f3fbd535SBarry Smith }
630f3fbd535SBarry Smith 
631d71ae5a4SJacob Faibussowitsch static PetscErrorCode PCApply_MG(PC pc, Vec b, Vec x)
632d71ae5a4SJacob Faibussowitsch {
633fcb023d4SJed Brown   PetscFunctionBegin;
6349566063dSJacob Faibussowitsch   PetscCall(PCApply_MG_Internal(pc, b, x, NULL, NULL, PETSC_FALSE));
6353ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
636fcb023d4SJed Brown }
637fcb023d4SJed Brown 
638d71ae5a4SJacob Faibussowitsch static PetscErrorCode PCApplyTranspose_MG(PC pc, Vec b, Vec x)
639d71ae5a4SJacob Faibussowitsch {
640fcb023d4SJed Brown   PetscFunctionBegin;
6419566063dSJacob Faibussowitsch   PetscCall(PCApply_MG_Internal(pc, b, x, NULL, NULL, PETSC_TRUE));
6423ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
64330b0564aSStefano Zampini }
64430b0564aSStefano Zampini 
645d71ae5a4SJacob Faibussowitsch static PetscErrorCode PCMatApply_MG(PC pc, Mat b, Mat x)
646d71ae5a4SJacob Faibussowitsch {
64730b0564aSStefano Zampini   PetscFunctionBegin;
6489566063dSJacob Faibussowitsch   PetscCall(PCApply_MG_Internal(pc, NULL, NULL, b, x, PETSC_FALSE));
6493ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
650fcb023d4SJed Brown }
651f3fbd535SBarry Smith 
652d71ae5a4SJacob Faibussowitsch PetscErrorCode PCSetFromOptions_MG(PC pc, PetscOptionItems *PetscOptionsObject)
653d71ae5a4SJacob Faibussowitsch {
654710315b6SLawrence Mitchell   PetscInt            levels, cycles;
655f3b08a26SMatthew G. Knepley   PetscBool           flg, flg2;
656f3fbd535SBarry Smith   PC_MG              *mg = (PC_MG *)pc->data;
6573d3eaba7SBarry Smith   PC_MG_Levels      **mglevels;
658f3fbd535SBarry Smith   PCMGType            mgtype;
659f3fbd535SBarry Smith   PCMGCycleType       mgctype;
6602134b1e4SBarry Smith   PCMGGalerkinType    gtype;
6612b3cbbdaSStefano Zampini   PCMGCoarseSpaceType coarseSpaceType;
662f3fbd535SBarry Smith 
663f3fbd535SBarry Smith   PetscFunctionBegin;
6641d743356SStefano Zampini   levels = PetscMax(mg->nlevels, 1);
665d0609cedSBarry Smith   PetscOptionsHeadBegin(PetscOptionsObject, "Multigrid options");
6669566063dSJacob Faibussowitsch   PetscCall(PetscOptionsInt("-pc_mg_levels", "Number of Levels", "PCMGSetLevels", levels, &levels, &flg));
6671a97d7b6SStefano Zampini   if (!flg && !mg->levels && pc->dm) {
6689566063dSJacob Faibussowitsch     PetscCall(DMGetRefineLevel(pc->dm, &levels));
669eb3f98d2SBarry Smith     levels++;
6703aeaf226SBarry Smith     mg->usedmfornumberoflevels = PETSC_TRUE;
671eb3f98d2SBarry Smith   }
6729566063dSJacob Faibussowitsch   PetscCall(PCMGSetLevels(pc, levels, NULL));
6733aeaf226SBarry Smith   mglevels = mg->levels;
6743aeaf226SBarry Smith 
675f3fbd535SBarry Smith   mgctype = (PCMGCycleType)mglevels[0]->cycles;
6769566063dSJacob Faibussowitsch   PetscCall(PetscOptionsEnum("-pc_mg_cycle_type", "V cycle or for W-cycle", "PCMGSetCycleType", PCMGCycleTypes, (PetscEnum)mgctype, (PetscEnum *)&mgctype, &flg));
6771baa6e33SBarry Smith   if (flg) PetscCall(PCMGSetCycleType(pc, mgctype));
6782134b1e4SBarry Smith   gtype = mg->galerkin;
6799566063dSJacob Faibussowitsch   PetscCall(PetscOptionsEnum("-pc_mg_galerkin", "Use Galerkin process to compute coarser operators", "PCMGSetGalerkin", PCMGGalerkinTypes, (PetscEnum)gtype, (PetscEnum *)&gtype, &flg));
6801baa6e33SBarry Smith   if (flg) PetscCall(PCMGSetGalerkin(pc, gtype));
6812b3cbbdaSStefano Zampini   coarseSpaceType = mg->coarseSpaceType;
6822b3cbbdaSStefano Zampini   PetscCall(PetscOptionsEnum("-pc_mg_adapt_interp_coarse_space", "Type of adaptive coarse space: none, polynomial, harmonic, eigenvector, generalized_eigenvector, gdsw", "PCMGSetAdaptCoarseSpaceType", PCMGCoarseSpaceTypes, (PetscEnum)coarseSpaceType, (PetscEnum *)&coarseSpaceType, &flg));
6832b3cbbdaSStefano Zampini   if (flg) PetscCall(PCMGSetAdaptCoarseSpaceType(pc, coarseSpaceType));
6849566063dSJacob Faibussowitsch   PetscCall(PetscOptionsInt("-pc_mg_adapt_interp_n", "Size of the coarse space for adaptive interpolation", "PCMGSetCoarseSpace", mg->Nc, &mg->Nc, &flg));
6859566063dSJacob Faibussowitsch   PetscCall(PetscOptionsBool("-pc_mg_mesp_monitor", "Monitor the multilevel eigensolver", "PCMGSetAdaptInterpolation", PETSC_FALSE, &mg->mespMonitor, &flg));
68641b6fd38SMatthew G. Knepley   flg2 = PETSC_FALSE;
6879566063dSJacob Faibussowitsch   PetscCall(PetscOptionsBool("-pc_mg_adapt_cr", "Monitor coarse space quality using Compatible Relaxation (CR)", "PCMGSetAdaptCR", PETSC_FALSE, &flg2, &flg));
6889566063dSJacob Faibussowitsch   if (flg) PetscCall(PCMGSetAdaptCR(pc, flg2));
689f56b1265SBarry Smith   flg = PETSC_FALSE;
6909566063dSJacob Faibussowitsch   PetscCall(PetscOptionsBool("-pc_mg_distinct_smoothup", "Create separate smoothup KSP and append the prefix _up", "PCMGSetDistinctSmoothUp", PETSC_FALSE, &flg, NULL));
6911baa6e33SBarry Smith   if (flg) PetscCall(PCMGSetDistinctSmoothUp(pc));
69231567311SBarry Smith   mgtype = mg->am;
6939566063dSJacob Faibussowitsch   PetscCall(PetscOptionsEnum("-pc_mg_type", "Multigrid type", "PCMGSetType", PCMGTypes, (PetscEnum)mgtype, (PetscEnum *)&mgtype, &flg));
6941baa6e33SBarry Smith   if (flg) PetscCall(PCMGSetType(pc, mgtype));
69531567311SBarry Smith   if (mg->am == PC_MG_MULTIPLICATIVE) {
6969566063dSJacob Faibussowitsch     PetscCall(PetscOptionsInt("-pc_mg_multiplicative_cycles", "Number of cycles for each preconditioner step", "PCMGMultiplicativeSetCycles", mg->cyclesperpcapply, &cycles, &flg));
6971baa6e33SBarry Smith     if (flg) PetscCall(PCMGMultiplicativeSetCycles(pc, cycles));
698f3fbd535SBarry Smith   }
699f3fbd535SBarry Smith   flg = PETSC_FALSE;
7009566063dSJacob Faibussowitsch   PetscCall(PetscOptionsBool("-pc_mg_log", "Log times for each multigrid level", "None", flg, &flg, NULL));
701f3fbd535SBarry Smith   if (flg) {
702f3fbd535SBarry Smith     PetscInt i;
703f3fbd535SBarry Smith     char     eventname[128];
7041a97d7b6SStefano Zampini 
705f3fbd535SBarry Smith     levels = mglevels[0]->levels;
706f3fbd535SBarry Smith     for (i = 0; i < levels; i++) {
707a364092eSJacob Faibussowitsch       PetscCall(PetscSNPrintf(eventname, PETSC_STATIC_ARRAY_LENGTH(eventname), "MGSetup Level %d", (int)i));
7089566063dSJacob Faibussowitsch       PetscCall(PetscLogEventRegister(eventname, ((PetscObject)pc)->classid, &mglevels[i]->eventsmoothsetup));
709a364092eSJacob Faibussowitsch       PetscCall(PetscSNPrintf(eventname, PETSC_STATIC_ARRAY_LENGTH(eventname), "MGSmooth Level %d", (int)i));
7109566063dSJacob Faibussowitsch       PetscCall(PetscLogEventRegister(eventname, ((PetscObject)pc)->classid, &mglevels[i]->eventsmoothsolve));
711f3fbd535SBarry Smith       if (i) {
712a364092eSJacob Faibussowitsch         PetscCall(PetscSNPrintf(eventname, PETSC_STATIC_ARRAY_LENGTH(eventname), "MGResid Level %d", (int)i));
7139566063dSJacob Faibussowitsch         PetscCall(PetscLogEventRegister(eventname, ((PetscObject)pc)->classid, &mglevels[i]->eventresidual));
714a364092eSJacob Faibussowitsch         PetscCall(PetscSNPrintf(eventname, PETSC_STATIC_ARRAY_LENGTH(eventname), "MGInterp Level %d", (int)i));
7159566063dSJacob Faibussowitsch         PetscCall(PetscLogEventRegister(eventname, ((PetscObject)pc)->classid, &mglevels[i]->eventinterprestrict));
716f3fbd535SBarry Smith       }
717f3fbd535SBarry Smith     }
718eec35531SMatthew G Knepley 
719ec60ca73SSatish Balay #if defined(PETSC_USE_LOG)
720eec35531SMatthew G Knepley     {
721eec35531SMatthew G Knepley       const char   *sname = "MG Apply";
722eec35531SMatthew G Knepley       PetscStageLog stageLog;
723eec35531SMatthew G Knepley       PetscInt      st;
724eec35531SMatthew G Knepley 
7259566063dSJacob Faibussowitsch       PetscCall(PetscLogGetStageLog(&stageLog));
726eec35531SMatthew G Knepley       for (st = 0; st < stageLog->numStages; ++st) {
727eec35531SMatthew G Knepley         PetscBool same;
728eec35531SMatthew G Knepley 
7299566063dSJacob Faibussowitsch         PetscCall(PetscStrcmp(stageLog->stageInfo[st].name, sname, &same));
7302fa5cd67SKarl Rupp         if (same) mg->stageApply = st;
731eec35531SMatthew G Knepley       }
73248a46eb9SPierre Jolivet       if (!mg->stageApply) PetscCall(PetscLogStageRegister(sname, &mg->stageApply));
733eec35531SMatthew G Knepley     }
734ec60ca73SSatish Balay #endif
735f3fbd535SBarry Smith   }
736d0609cedSBarry Smith   PetscOptionsHeadEnd();
737f3b08a26SMatthew G. Knepley   /* Check option consistency */
7389566063dSJacob Faibussowitsch   PetscCall(PCMGGetGalerkin(pc, &gtype));
7399566063dSJacob Faibussowitsch   PetscCall(PCMGGetAdaptInterpolation(pc, &flg));
74008401ef6SPierre Jolivet   PetscCheck(!flg || !(gtype >= PC_MG_GALERKIN_NONE), PetscObjectComm((PetscObject)pc), PETSC_ERR_ARG_INCOMP, "Must use Galerkin coarse operators when adapting the interpolator");
7413ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
742f3fbd535SBarry Smith }
743f3fbd535SBarry Smith 
7440a545947SLisandro Dalcin const char *const PCMGTypes[]            = {"MULTIPLICATIVE", "ADDITIVE", "FULL", "KASKADE", "PCMGType", "PC_MG", NULL};
7450a545947SLisandro Dalcin const char *const PCMGCycleTypes[]       = {"invalid", "v", "w", "PCMGCycleType", "PC_MG_CYCLE", NULL};
7460a545947SLisandro Dalcin const char *const PCMGGalerkinTypes[]    = {"both", "pmat", "mat", "none", "external", "PCMGGalerkinType", "PC_MG_GALERKIN", NULL};
7472b3cbbdaSStefano Zampini const char *const PCMGCoarseSpaceTypes[] = {"none", "polynomial", "harmonic", "eigenvector", "generalized_eigenvector", "gdsw", "PCMGCoarseSpaceType", "PCMG_ADAPT_NONE", NULL};
748f3fbd535SBarry Smith 
7499804daf3SBarry Smith #include <petscdraw.h>
750d71ae5a4SJacob Faibussowitsch PetscErrorCode PCView_MG(PC pc, PetscViewer viewer)
751d71ae5a4SJacob Faibussowitsch {
752f3fbd535SBarry Smith   PC_MG         *mg       = (PC_MG *)pc->data;
753f3fbd535SBarry Smith   PC_MG_Levels **mglevels = mg->levels;
754e3deeeafSJed Brown   PetscInt       levels   = mglevels ? mglevels[0]->levels : 0, i;
755219b1045SBarry Smith   PetscBool      iascii, isbinary, isdraw;
756f3fbd535SBarry Smith 
757f3fbd535SBarry Smith   PetscFunctionBegin;
7589566063dSJacob Faibussowitsch   PetscCall(PetscObjectTypeCompare((PetscObject)viewer, PETSCVIEWERASCII, &iascii));
7599566063dSJacob Faibussowitsch   PetscCall(PetscObjectTypeCompare((PetscObject)viewer, PETSCVIEWERBINARY, &isbinary));
7609566063dSJacob Faibussowitsch   PetscCall(PetscObjectTypeCompare((PetscObject)viewer, PETSCVIEWERDRAW, &isdraw));
761f3fbd535SBarry Smith   if (iascii) {
762e3deeeafSJed Brown     const char *cyclename = levels ? (mglevels[0]->cycles == PC_MG_CYCLE_V ? "v" : "w") : "unknown";
76363a3b9bcSJacob Faibussowitsch     PetscCall(PetscViewerASCIIPrintf(viewer, "  type is %s, levels=%" PetscInt_FMT " cycles=%s\n", PCMGTypes[mg->am], levels, cyclename));
76448a46eb9SPierre Jolivet     if (mg->am == PC_MG_MULTIPLICATIVE) PetscCall(PetscViewerASCIIPrintf(viewer, "    Cycles per PCApply=%" PetscInt_FMT "\n", mg->cyclesperpcapply));
7652134b1e4SBarry Smith     if (mg->galerkin == PC_MG_GALERKIN_BOTH) {
7669566063dSJacob Faibussowitsch       PetscCall(PetscViewerASCIIPrintf(viewer, "    Using Galerkin computed coarse grid matrices\n"));
7672134b1e4SBarry Smith     } else if (mg->galerkin == PC_MG_GALERKIN_PMAT) {
7689566063dSJacob Faibussowitsch       PetscCall(PetscViewerASCIIPrintf(viewer, "    Using Galerkin computed coarse grid matrices for pmat\n"));
7692134b1e4SBarry Smith     } else if (mg->galerkin == PC_MG_GALERKIN_MAT) {
7709566063dSJacob Faibussowitsch       PetscCall(PetscViewerASCIIPrintf(viewer, "    Using Galerkin computed coarse grid matrices for mat\n"));
7712134b1e4SBarry Smith     } else if (mg->galerkin == PC_MG_GALERKIN_EXTERNAL) {
7729566063dSJacob Faibussowitsch       PetscCall(PetscViewerASCIIPrintf(viewer, "    Using externally compute Galerkin coarse grid matrices\n"));
7734f66f45eSBarry Smith     } else {
7749566063dSJacob Faibussowitsch       PetscCall(PetscViewerASCIIPrintf(viewer, "    Not using Galerkin computed coarse grid matrices\n"));
775f3fbd535SBarry Smith     }
7761baa6e33SBarry Smith     if (mg->view) PetscCall((*mg->view)(pc, viewer));
777f3fbd535SBarry Smith     for (i = 0; i < levels; i++) {
77863a3b9bcSJacob Faibussowitsch       if (i) {
77963a3b9bcSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, "Down solver (pre-smoother) on level %" PetscInt_FMT " -------------------------------\n", i));
780f3fbd535SBarry Smith       } else {
78163a3b9bcSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, "Coarse grid solver -- level %" PetscInt_FMT " -------------------------------\n", i));
782f3fbd535SBarry Smith       }
7839566063dSJacob Faibussowitsch       PetscCall(PetscViewerASCIIPushTab(viewer));
7849566063dSJacob Faibussowitsch       PetscCall(KSPView(mglevels[i]->smoothd, viewer));
7859566063dSJacob Faibussowitsch       PetscCall(PetscViewerASCIIPopTab(viewer));
786f3fbd535SBarry Smith       if (i && mglevels[i]->smoothd == mglevels[i]->smoothu) {
7879566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, "Up solver (post-smoother) same as down solver (pre-smoother)\n"));
788f3fbd535SBarry Smith       } else if (i) {
78963a3b9bcSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, "Up solver (post-smoother) on level %" PetscInt_FMT " -------------------------------\n", i));
7909566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPushTab(viewer));
7919566063dSJacob Faibussowitsch         PetscCall(KSPView(mglevels[i]->smoothu, viewer));
7929566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPopTab(viewer));
793f3fbd535SBarry Smith       }
79441b6fd38SMatthew G. Knepley       if (i && mglevels[i]->cr) {
79563a3b9bcSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, "CR solver on level %" PetscInt_FMT " -------------------------------\n", i));
7969566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPushTab(viewer));
7979566063dSJacob Faibussowitsch         PetscCall(KSPView(mglevels[i]->cr, viewer));
7989566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPopTab(viewer));
79941b6fd38SMatthew G. Knepley       }
800f3fbd535SBarry Smith     }
8015b0b0462SBarry Smith   } else if (isbinary) {
8025b0b0462SBarry Smith     for (i = levels - 1; i >= 0; i--) {
8039566063dSJacob Faibussowitsch       PetscCall(KSPView(mglevels[i]->smoothd, viewer));
80448a46eb9SPierre Jolivet       if (i && mglevels[i]->smoothd != mglevels[i]->smoothu) PetscCall(KSPView(mglevels[i]->smoothu, viewer));
8055b0b0462SBarry Smith     }
806219b1045SBarry Smith   } else if (isdraw) {
807219b1045SBarry Smith     PetscDraw draw;
808b4375e8dSPeter Brune     PetscReal x, w, y, bottom, th;
8099566063dSJacob Faibussowitsch     PetscCall(PetscViewerDrawGetDraw(viewer, 0, &draw));
8109566063dSJacob Faibussowitsch     PetscCall(PetscDrawGetCurrentPoint(draw, &x, &y));
8119566063dSJacob Faibussowitsch     PetscCall(PetscDrawStringGetSize(draw, NULL, &th));
812219b1045SBarry Smith     bottom = y - th;
813219b1045SBarry Smith     for (i = levels - 1; i >= 0; i--) {
814b4375e8dSPeter Brune       if (!mglevels[i]->smoothu || (mglevels[i]->smoothu == mglevels[i]->smoothd)) {
8159566063dSJacob Faibussowitsch         PetscCall(PetscDrawPushCurrentPoint(draw, x, bottom));
8169566063dSJacob Faibussowitsch         PetscCall(KSPView(mglevels[i]->smoothd, viewer));
8179566063dSJacob Faibussowitsch         PetscCall(PetscDrawPopCurrentPoint(draw));
818b4375e8dSPeter Brune       } else {
819b4375e8dSPeter Brune         w = 0.5 * PetscMin(1.0 - x, x);
8209566063dSJacob Faibussowitsch         PetscCall(PetscDrawPushCurrentPoint(draw, x + w, bottom));
8219566063dSJacob Faibussowitsch         PetscCall(KSPView(mglevels[i]->smoothd, viewer));
8229566063dSJacob Faibussowitsch         PetscCall(PetscDrawPopCurrentPoint(draw));
8239566063dSJacob Faibussowitsch         PetscCall(PetscDrawPushCurrentPoint(draw, x - w, bottom));
8249566063dSJacob Faibussowitsch         PetscCall(KSPView(mglevels[i]->smoothu, viewer));
8259566063dSJacob Faibussowitsch         PetscCall(PetscDrawPopCurrentPoint(draw));
826b4375e8dSPeter Brune       }
8279566063dSJacob Faibussowitsch       PetscCall(PetscDrawGetBoundingBox(draw, NULL, &bottom, NULL, NULL));
8281cd381d6SBarry Smith       bottom -= th;
829219b1045SBarry Smith     }
8305b0b0462SBarry Smith   }
8313ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
832f3fbd535SBarry Smith }
833f3fbd535SBarry Smith 
834af0996ceSBarry Smith #include <petsc/private/kspimpl.h>
835cab2e9ccSBarry Smith 
836f3fbd535SBarry Smith /*
837f3fbd535SBarry Smith     Calls setup for the KSP on each level
838f3fbd535SBarry Smith */
839d71ae5a4SJacob Faibussowitsch PetscErrorCode PCSetUp_MG(PC pc)
840d71ae5a4SJacob Faibussowitsch {
841f3fbd535SBarry Smith   PC_MG         *mg       = (PC_MG *)pc->data;
842f3fbd535SBarry Smith   PC_MG_Levels **mglevels = mg->levels;
8431a97d7b6SStefano Zampini   PetscInt       i, n;
84498e04984SBarry Smith   PC             cpc;
8453db492dfSBarry Smith   PetscBool      dump = PETSC_FALSE, opsset, use_amat, missinginterpolate = PETSC_FALSE;
846f3fbd535SBarry Smith   Mat            dA, dB;
847f3fbd535SBarry Smith   Vec            tvec;
848218a07d4SBarry Smith   DM            *dms;
8490a545947SLisandro Dalcin   PetscViewer    viewer = NULL;
85041b6fd38SMatthew G. Knepley   PetscBool      dAeqdB = PETSC_FALSE, needRestricts = PETSC_FALSE, doCR = PETSC_FALSE;
8512b3cbbdaSStefano Zampini   PetscBool      adaptInterpolation = mg->adaptInterpolation;
852f3fbd535SBarry Smith 
853f3fbd535SBarry Smith   PetscFunctionBegin;
85428b400f6SJacob Faibussowitsch   PetscCheck(mglevels, PetscObjectComm((PetscObject)pc), PETSC_ERR_ARG_WRONGSTATE, "Must set MG levels with PCMGSetLevels() before setting up");
8551a97d7b6SStefano Zampini   n = mglevels[0]->levels;
85601bc414fSDmitry Karpeev   /* FIX: Move this to PCSetFromOptions_MG? */
8573aeaf226SBarry Smith   if (mg->usedmfornumberoflevels) {
8583aeaf226SBarry Smith     PetscInt levels;
8599566063dSJacob Faibussowitsch     PetscCall(DMGetRefineLevel(pc->dm, &levels));
8603aeaf226SBarry Smith     levels++;
8613aeaf226SBarry Smith     if (levels > n) { /* the problem is now being solved on a finer grid */
8629566063dSJacob Faibussowitsch       PetscCall(PCMGSetLevels(pc, levels, NULL));
8633aeaf226SBarry Smith       n = levels;
8649566063dSJacob Faibussowitsch       PetscCall(PCSetFromOptions(pc)); /* it is bad to call this here, but otherwise will never be called for the new hierarchy */
8653aeaf226SBarry Smith       mglevels = mg->levels;
8663aeaf226SBarry Smith     }
8673aeaf226SBarry Smith   }
8689566063dSJacob Faibussowitsch   PetscCall(KSPGetPC(mglevels[0]->smoothd, &cpc));
8693aeaf226SBarry Smith 
870f3fbd535SBarry Smith   /* If user did not provide fine grid operators OR operator was not updated since last global KSPSetOperators() */
871f3fbd535SBarry Smith   /* so use those from global PC */
872f3fbd535SBarry Smith   /* Is this what we always want? What if user wants to keep old one? */
8739566063dSJacob Faibussowitsch   PetscCall(KSPGetOperatorsSet(mglevels[n - 1]->smoothd, NULL, &opsset));
87498e04984SBarry Smith   if (opsset) {
87598e04984SBarry Smith     Mat mmat;
8769566063dSJacob Faibussowitsch     PetscCall(KSPGetOperators(mglevels[n - 1]->smoothd, NULL, &mmat));
87798e04984SBarry Smith     if (mmat == pc->pmat) opsset = PETSC_FALSE;
87898e04984SBarry Smith   }
8794a5f07a5SMark F. Adams 
88041b6fd38SMatthew G. Knepley   /* Create CR solvers */
8819566063dSJacob Faibussowitsch   PetscCall(PCMGGetAdaptCR(pc, &doCR));
88241b6fd38SMatthew G. Knepley   if (doCR) {
88341b6fd38SMatthew G. Knepley     const char *prefix;
88441b6fd38SMatthew G. Knepley 
8859566063dSJacob Faibussowitsch     PetscCall(PCGetOptionsPrefix(pc, &prefix));
88641b6fd38SMatthew G. Knepley     for (i = 1; i < n; ++i) {
88741b6fd38SMatthew G. Knepley       PC   ipc, cr;
88841b6fd38SMatthew G. Knepley       char crprefix[128];
88941b6fd38SMatthew G. Knepley 
8909566063dSJacob Faibussowitsch       PetscCall(KSPCreate(PetscObjectComm((PetscObject)pc), &mglevels[i]->cr));
891*3821be0aSBarry Smith       PetscCall(KSPSetNestLevel(mglevels[i]->cr, pc->kspnestlevel));
8929566063dSJacob Faibussowitsch       PetscCall(KSPSetErrorIfNotConverged(mglevels[i]->cr, PETSC_FALSE));
8939566063dSJacob Faibussowitsch       PetscCall(PetscObjectIncrementTabLevel((PetscObject)mglevels[i]->cr, (PetscObject)pc, n - i));
8949566063dSJacob Faibussowitsch       PetscCall(KSPSetOptionsPrefix(mglevels[i]->cr, prefix));
8959566063dSJacob Faibussowitsch       PetscCall(PetscObjectComposedDataSetInt((PetscObject)mglevels[i]->cr, PetscMGLevelId, mglevels[i]->level));
8969566063dSJacob Faibussowitsch       PetscCall(KSPSetType(mglevels[i]->cr, KSPCHEBYSHEV));
8979566063dSJacob Faibussowitsch       PetscCall(KSPSetConvergenceTest(mglevels[i]->cr, KSPConvergedSkip, NULL, NULL));
8989566063dSJacob Faibussowitsch       PetscCall(KSPSetNormType(mglevels[i]->cr, KSP_NORM_PRECONDITIONED));
8999566063dSJacob Faibussowitsch       PetscCall(KSPGetPC(mglevels[i]->cr, &ipc));
90041b6fd38SMatthew G. Knepley 
9019566063dSJacob Faibussowitsch       PetscCall(PCSetType(ipc, PCCOMPOSITE));
9029566063dSJacob Faibussowitsch       PetscCall(PCCompositeSetType(ipc, PC_COMPOSITE_MULTIPLICATIVE));
9039566063dSJacob Faibussowitsch       PetscCall(PCCompositeAddPCType(ipc, PCSOR));
9049566063dSJacob Faibussowitsch       PetscCall(CreateCR_Private(pc, i, &cr));
9059566063dSJacob Faibussowitsch       PetscCall(PCCompositeAddPC(ipc, cr));
9069566063dSJacob Faibussowitsch       PetscCall(PCDestroy(&cr));
90741b6fd38SMatthew G. Knepley 
9089566063dSJacob Faibussowitsch       PetscCall(KSPSetTolerances(mglevels[i]->cr, PETSC_DEFAULT, PETSC_DEFAULT, PETSC_DEFAULT, mg->default_smoothd));
9099566063dSJacob Faibussowitsch       PetscCall(KSPSetInitialGuessNonzero(mglevels[i]->cr, PETSC_TRUE));
9109566063dSJacob Faibussowitsch       PetscCall(PetscSNPrintf(crprefix, 128, "mg_levels_%d_cr_", (int)i));
9119566063dSJacob Faibussowitsch       PetscCall(KSPAppendOptionsPrefix(mglevels[i]->cr, crprefix));
91241b6fd38SMatthew G. Knepley     }
91341b6fd38SMatthew G. Knepley   }
91441b6fd38SMatthew G. Knepley 
9154a5f07a5SMark F. Adams   if (!opsset) {
9169566063dSJacob Faibussowitsch     PetscCall(PCGetUseAmat(pc, &use_amat));
9174a5f07a5SMark F. Adams     if (use_amat) {
9189566063dSJacob Faibussowitsch       PetscCall(PetscInfo(pc, "Using outer operators to define finest grid operator \n  because PCMGGetSmoother(pc,nlevels-1,&ksp);KSPSetOperators(ksp,...); was not called.\n"));
9199566063dSJacob Faibussowitsch       PetscCall(KSPSetOperators(mglevels[n - 1]->smoothd, pc->mat, pc->pmat));
92069858f1bSStefano Zampini     } else {
9219566063dSJacob Faibussowitsch       PetscCall(PetscInfo(pc, "Using matrix (pmat) operators to define finest grid operator \n  because PCMGGetSmoother(pc,nlevels-1,&ksp);KSPSetOperators(ksp,...); was not called.\n"));
9229566063dSJacob Faibussowitsch       PetscCall(KSPSetOperators(mglevels[n - 1]->smoothd, pc->pmat, pc->pmat));
9234a5f07a5SMark F. Adams     }
9244a5f07a5SMark F. Adams   }
925f3fbd535SBarry Smith 
9269c7ffb39SBarry Smith   for (i = n - 1; i > 0; i--) {
9279c7ffb39SBarry Smith     if (!(mglevels[i]->interpolate || mglevels[i]->restrct)) {
9289c7ffb39SBarry Smith       missinginterpolate = PETSC_TRUE;
9292b3cbbdaSStefano Zampini       break;
9309c7ffb39SBarry Smith     }
9319c7ffb39SBarry Smith   }
9322134b1e4SBarry Smith 
9339566063dSJacob Faibussowitsch   PetscCall(KSPGetOperators(mglevels[n - 1]->smoothd, &dA, &dB));
9342134b1e4SBarry Smith   if (dA == dB) dAeqdB = PETSC_TRUE;
9352b3cbbdaSStefano Zampini   if (mg->galerkin == PC_MG_GALERKIN_NONE || ((mg->galerkin == PC_MG_GALERKIN_PMAT || mg->galerkin == PC_MG_GALERKIN_MAT) && !dAeqdB)) {
9362134b1e4SBarry Smith     needRestricts = PETSC_TRUE; /* user must compute either mat, pmat, or both so must restrict x to coarser levels */
9372134b1e4SBarry Smith   }
9382134b1e4SBarry Smith 
9392b3cbbdaSStefano Zampini   if (pc->dm && !pc->setupcalled) {
9402b3cbbdaSStefano Zampini     /* finest smoother also gets DM but it is not active, independent of whether galerkin==PC_MG_GALERKIN_EXTERNAL */
9412b3cbbdaSStefano Zampini     PetscCall(KSPSetDM(mglevels[n - 1]->smoothd, pc->dm));
9422b3cbbdaSStefano Zampini     PetscCall(KSPSetDMActive(mglevels[n - 1]->smoothd, PETSC_FALSE));
9432b3cbbdaSStefano Zampini     if (mglevels[n - 1]->smoothd != mglevels[n - 1]->smoothu) {
9442b3cbbdaSStefano Zampini       PetscCall(KSPSetDM(mglevels[n - 1]->smoothu, pc->dm));
9452b3cbbdaSStefano Zampini       PetscCall(KSPSetDMActive(mglevels[n - 1]->smoothu, PETSC_FALSE));
9462b3cbbdaSStefano Zampini     }
9472b3cbbdaSStefano Zampini     if (mglevels[n - 1]->cr) {
9482b3cbbdaSStefano Zampini       PetscCall(KSPSetDM(mglevels[n - 1]->cr, pc->dm));
9492b3cbbdaSStefano Zampini       PetscCall(KSPSetDMActive(mglevels[n - 1]->cr, PETSC_FALSE));
9502b3cbbdaSStefano Zampini     }
9512b3cbbdaSStefano Zampini   }
9522b3cbbdaSStefano Zampini 
9539c7ffb39SBarry Smith   /*
9549c7ffb39SBarry Smith    Skipping if user has provided all interpolation/restriction needed (since DM might not be able to produce them (when coming from SNES/TS)
9552b3cbbdaSStefano Zampini    Skipping for externally managed hierarchy (such as ML and GAMG). Cleaner logic here would be great. Wrap ML/GAMG as DMs?
9569c7ffb39SBarry Smith   */
95732fe681dSStefano Zampini   if (missinginterpolate && mg->galerkin != PC_MG_GALERKIN_EXTERNAL && !pc->setupcalled) {
95832fe681dSStefano Zampini     /* first see if we can compute a coarse space */
95932fe681dSStefano Zampini     if (mg->coarseSpaceType == PCMG_ADAPT_GDSW) {
96032fe681dSStefano Zampini       for (i = n - 2; i > -1; i--) {
96132fe681dSStefano Zampini         if (!mglevels[i + 1]->restrct && !mglevels[i + 1]->interpolate) {
96232fe681dSStefano Zampini           PetscCall(PCMGComputeCoarseSpace_Internal(pc, i + 1, mg->coarseSpaceType, mg->Nc, NULL, &mglevels[i + 1]->coarseSpace));
96332fe681dSStefano Zampini           PetscCall(PCMGSetInterpolation(pc, i + 1, mglevels[i + 1]->coarseSpace));
96432fe681dSStefano Zampini         }
96532fe681dSStefano Zampini       }
96632fe681dSStefano Zampini     } else { /* construct the interpolation from the DMs */
9672e499ae9SBarry Smith       Mat p;
96873250ac0SBarry Smith       Vec rscale;
9699566063dSJacob Faibussowitsch       PetscCall(PetscMalloc1(n, &dms));
970218a07d4SBarry Smith       dms[n - 1] = pc->dm;
971ef1267afSMatthew G. Knepley       /* Separately create them so we do not get DMKSP interference between levels */
9729566063dSJacob Faibussowitsch       for (i = n - 2; i > -1; i--) PetscCall(DMCoarsen(dms[i + 1], MPI_COMM_NULL, &dms[i]));
973218a07d4SBarry Smith       for (i = n - 2; i > -1; i--) {
974942e3340SBarry Smith         DMKSP     kdm;
975eab52d2bSLawrence Mitchell         PetscBool dmhasrestrict, dmhasinject;
9769566063dSJacob Faibussowitsch         PetscCall(KSPSetDM(mglevels[i]->smoothd, dms[i]));
9779566063dSJacob Faibussowitsch         if (!needRestricts) PetscCall(KSPSetDMActive(mglevels[i]->smoothd, PETSC_FALSE));
978c27ee7a3SPatrick Farrell         if (mglevels[i]->smoothd != mglevels[i]->smoothu) {
9799566063dSJacob Faibussowitsch           PetscCall(KSPSetDM(mglevels[i]->smoothu, dms[i]));
9809566063dSJacob Faibussowitsch           if (!needRestricts) PetscCall(KSPSetDMActive(mglevels[i]->smoothu, PETSC_FALSE));
981c27ee7a3SPatrick Farrell         }
98241b6fd38SMatthew G. Knepley         if (mglevels[i]->cr) {
9839566063dSJacob Faibussowitsch           PetscCall(KSPSetDM(mglevels[i]->cr, dms[i]));
9849566063dSJacob Faibussowitsch           if (!needRestricts) PetscCall(KSPSetDMActive(mglevels[i]->cr, PETSC_FALSE));
98541b6fd38SMatthew G. Knepley         }
9869566063dSJacob Faibussowitsch         PetscCall(DMGetDMKSPWrite(dms[i], &kdm));
987d1a3e677SJed Brown         /* Ugly hack so that the next KSPSetUp() will use the RHS that we set. A better fix is to change dmActive to take
988d1a3e677SJed Brown          * a bitwise OR of computing the matrix, RHS, and initial iterate. */
9890298fd71SBarry Smith         kdm->ops->computerhs = NULL;
9900298fd71SBarry Smith         kdm->rhsctx          = NULL;
99124c3aa18SBarry Smith         if (!mglevels[i + 1]->interpolate) {
9929566063dSJacob Faibussowitsch           PetscCall(DMCreateInterpolation(dms[i], dms[i + 1], &p, &rscale));
9939566063dSJacob Faibussowitsch           PetscCall(PCMGSetInterpolation(pc, i + 1, p));
9949566063dSJacob Faibussowitsch           if (rscale) PetscCall(PCMGSetRScale(pc, i + 1, rscale));
9959566063dSJacob Faibussowitsch           PetscCall(VecDestroy(&rscale));
9969566063dSJacob Faibussowitsch           PetscCall(MatDestroy(&p));
997218a07d4SBarry Smith         }
9989566063dSJacob Faibussowitsch         PetscCall(DMHasCreateRestriction(dms[i], &dmhasrestrict));
9993ad4599aSBarry Smith         if (dmhasrestrict && !mglevels[i + 1]->restrct) {
10009566063dSJacob Faibussowitsch           PetscCall(DMCreateRestriction(dms[i], dms[i + 1], &p));
10019566063dSJacob Faibussowitsch           PetscCall(PCMGSetRestriction(pc, i + 1, p));
10029566063dSJacob Faibussowitsch           PetscCall(MatDestroy(&p));
10033ad4599aSBarry Smith         }
10049566063dSJacob Faibussowitsch         PetscCall(DMHasCreateInjection(dms[i], &dmhasinject));
1005eab52d2bSLawrence Mitchell         if (dmhasinject && !mglevels[i + 1]->inject) {
10069566063dSJacob Faibussowitsch           PetscCall(DMCreateInjection(dms[i], dms[i + 1], &p));
10079566063dSJacob Faibussowitsch           PetscCall(PCMGSetInjection(pc, i + 1, p));
10089566063dSJacob Faibussowitsch           PetscCall(MatDestroy(&p));
1009eab52d2bSLawrence Mitchell         }
101024c3aa18SBarry Smith       }
10112d2b81a6SBarry Smith 
10129566063dSJacob Faibussowitsch       for (i = n - 2; i > -1; i--) PetscCall(DMDestroy(&dms[i]));
10139566063dSJacob Faibussowitsch       PetscCall(PetscFree(dms));
1014ce4cda84SJed Brown     }
101532fe681dSStefano Zampini   }
1016cab2e9ccSBarry Smith 
10172134b1e4SBarry Smith   if (mg->galerkin < PC_MG_GALERKIN_NONE) {
10182134b1e4SBarry Smith     Mat       A, B;
10192134b1e4SBarry Smith     PetscBool doA = PETSC_FALSE, doB = PETSC_FALSE;
10202134b1e4SBarry Smith     MatReuse  reuse = MAT_INITIAL_MATRIX;
10212134b1e4SBarry Smith 
10222b3cbbdaSStefano Zampini     if (mg->galerkin == PC_MG_GALERKIN_PMAT || mg->galerkin == PC_MG_GALERKIN_BOTH) doB = PETSC_TRUE;
10232b3cbbdaSStefano Zampini     if (mg->galerkin == PC_MG_GALERKIN_MAT || (mg->galerkin == PC_MG_GALERKIN_BOTH && dA != dB)) doA = PETSC_TRUE;
10242134b1e4SBarry Smith     if (pc->setupcalled) reuse = MAT_REUSE_MATRIX;
1025f3fbd535SBarry Smith     for (i = n - 2; i > -1; i--) {
10267827d75bSBarry Smith       PetscCheck(mglevels[i + 1]->restrct || mglevels[i + 1]->interpolate, PetscObjectComm((PetscObject)pc), PETSC_ERR_ARG_WRONGSTATE, "Must provide interpolation or restriction for each MG level except level 0");
102748a46eb9SPierre Jolivet       if (!mglevels[i + 1]->interpolate) PetscCall(PCMGSetInterpolation(pc, i + 1, mglevels[i + 1]->restrct));
102848a46eb9SPierre Jolivet       if (!mglevels[i + 1]->restrct) PetscCall(PCMGSetRestriction(pc, i + 1, mglevels[i + 1]->interpolate));
102948a46eb9SPierre Jolivet       if (reuse == MAT_REUSE_MATRIX) PetscCall(KSPGetOperators(mglevels[i]->smoothd, &A, &B));
103048a46eb9SPierre Jolivet       if (doA) PetscCall(MatGalerkin(mglevels[i + 1]->restrct, dA, mglevels[i + 1]->interpolate, reuse, 1.0, &A));
103148a46eb9SPierre Jolivet       if (doB) PetscCall(MatGalerkin(mglevels[i + 1]->restrct, dB, mglevels[i + 1]->interpolate, reuse, 1.0, &B));
10322134b1e4SBarry Smith       /* the management of the PetscObjectReference() and PetscObjecDereference() below is rather delicate */
10332134b1e4SBarry Smith       if (!doA && dAeqdB) {
10349566063dSJacob Faibussowitsch         if (reuse == MAT_INITIAL_MATRIX) PetscCall(PetscObjectReference((PetscObject)B));
10352134b1e4SBarry Smith         A = B;
10362134b1e4SBarry Smith       } else if (!doA && reuse == MAT_INITIAL_MATRIX) {
10379566063dSJacob Faibussowitsch         PetscCall(KSPGetOperators(mglevels[i]->smoothd, &A, NULL));
10389566063dSJacob Faibussowitsch         PetscCall(PetscObjectReference((PetscObject)A));
1039b9367914SBarry Smith       }
10402134b1e4SBarry Smith       if (!doB && dAeqdB) {
10419566063dSJacob Faibussowitsch         if (reuse == MAT_INITIAL_MATRIX) PetscCall(PetscObjectReference((PetscObject)A));
10422134b1e4SBarry Smith         B = A;
10432134b1e4SBarry Smith       } else if (!doB && reuse == MAT_INITIAL_MATRIX) {
10449566063dSJacob Faibussowitsch         PetscCall(KSPGetOperators(mglevels[i]->smoothd, NULL, &B));
10459566063dSJacob Faibussowitsch         PetscCall(PetscObjectReference((PetscObject)B));
10467d537d92SJed Brown       }
10472134b1e4SBarry Smith       if (reuse == MAT_INITIAL_MATRIX) {
10489566063dSJacob Faibussowitsch         PetscCall(KSPSetOperators(mglevels[i]->smoothd, A, B));
10499566063dSJacob Faibussowitsch         PetscCall(PetscObjectDereference((PetscObject)A));
10509566063dSJacob Faibussowitsch         PetscCall(PetscObjectDereference((PetscObject)B));
10512134b1e4SBarry Smith       }
10522134b1e4SBarry Smith       dA = A;
1053f3fbd535SBarry Smith       dB = B;
1054f3fbd535SBarry Smith     }
1055f3fbd535SBarry Smith   }
1056f3b08a26SMatthew G. Knepley 
1057f3b08a26SMatthew G. Knepley   /* Adapt interpolation matrices */
10582b3cbbdaSStefano Zampini   if (adaptInterpolation) {
1059f3b08a26SMatthew G. Knepley     for (i = 0; i < n; ++i) {
106048a46eb9SPierre Jolivet       if (!mglevels[i]->coarseSpace) PetscCall(PCMGComputeCoarseSpace_Internal(pc, i, mg->coarseSpaceType, mg->Nc, !i ? NULL : mglevels[i - 1]->coarseSpace, &mglevels[i]->coarseSpace));
10612b3cbbdaSStefano Zampini       if (i) PetscCall(PCMGAdaptInterpolator_Internal(pc, i, mglevels[i - 1]->smoothu, mglevels[i]->smoothu, mglevels[i - 1]->coarseSpace, mglevels[i]->coarseSpace));
1062f3b08a26SMatthew G. Knepley     }
106348a46eb9SPierre Jolivet     for (i = n - 2; i > -1; --i) PetscCall(PCMGRecomputeLevelOperators_Internal(pc, i));
1064f3b08a26SMatthew G. Knepley   }
1065f3b08a26SMatthew G. Knepley 
10662134b1e4SBarry Smith   if (needRestricts && pc->dm) {
1067caa4e7f2SJed Brown     for (i = n - 2; i >= 0; i--) {
1068ccceb50cSJed Brown       DM  dmfine, dmcoarse;
1069caa4e7f2SJed Brown       Mat Restrict, Inject;
1070caa4e7f2SJed Brown       Vec rscale;
10719566063dSJacob Faibussowitsch       PetscCall(KSPGetDM(mglevels[i + 1]->smoothd, &dmfine));
10729566063dSJacob Faibussowitsch       PetscCall(KSPGetDM(mglevels[i]->smoothd, &dmcoarse));
10739566063dSJacob Faibussowitsch       PetscCall(PCMGGetRestriction(pc, i + 1, &Restrict));
10749566063dSJacob Faibussowitsch       PetscCall(PCMGGetRScale(pc, i + 1, &rscale));
10759566063dSJacob Faibussowitsch       PetscCall(PCMGGetInjection(pc, i + 1, &Inject));
10769566063dSJacob Faibussowitsch       PetscCall(DMRestrict(dmfine, Restrict, rscale, Inject, dmcoarse));
1077caa4e7f2SJed Brown     }
1078caa4e7f2SJed Brown   }
1079f3fbd535SBarry Smith 
1080f3fbd535SBarry Smith   if (!pc->setupcalled) {
108148a46eb9SPierre Jolivet     for (i = 0; i < n; i++) PetscCall(KSPSetFromOptions(mglevels[i]->smoothd));
1082f3fbd535SBarry Smith     for (i = 1; i < n; i++) {
108348a46eb9SPierre Jolivet       if (mglevels[i]->smoothu && (mglevels[i]->smoothu != mglevels[i]->smoothd)) PetscCall(KSPSetFromOptions(mglevels[i]->smoothu));
108448a46eb9SPierre Jolivet       if (mglevels[i]->cr) PetscCall(KSPSetFromOptions(mglevels[i]->cr));
1085f3fbd535SBarry Smith     }
10863ad4599aSBarry Smith     /* insure that if either interpolation or restriction is set the other other one is set */
1087f3fbd535SBarry Smith     for (i = 1; i < n; i++) {
10889566063dSJacob Faibussowitsch       PetscCall(PCMGGetInterpolation(pc, i, NULL));
10899566063dSJacob Faibussowitsch       PetscCall(PCMGGetRestriction(pc, i, NULL));
1090f3fbd535SBarry Smith     }
1091f3fbd535SBarry Smith     for (i = 0; i < n - 1; i++) {
1092f3fbd535SBarry Smith       if (!mglevels[i]->b) {
1093f3fbd535SBarry Smith         Vec *vec;
10949566063dSJacob Faibussowitsch         PetscCall(KSPCreateVecs(mglevels[i]->smoothd, 1, &vec, 0, NULL));
10959566063dSJacob Faibussowitsch         PetscCall(PCMGSetRhs(pc, i, *vec));
10969566063dSJacob Faibussowitsch         PetscCall(VecDestroy(vec));
10979566063dSJacob Faibussowitsch         PetscCall(PetscFree(vec));
1098f3fbd535SBarry Smith       }
1099f3fbd535SBarry Smith       if (!mglevels[i]->r && i) {
11009566063dSJacob Faibussowitsch         PetscCall(VecDuplicate(mglevels[i]->b, &tvec));
11019566063dSJacob Faibussowitsch         PetscCall(PCMGSetR(pc, i, tvec));
11029566063dSJacob Faibussowitsch         PetscCall(VecDestroy(&tvec));
1103f3fbd535SBarry Smith       }
1104f3fbd535SBarry Smith       if (!mglevels[i]->x) {
11059566063dSJacob Faibussowitsch         PetscCall(VecDuplicate(mglevels[i]->b, &tvec));
11069566063dSJacob Faibussowitsch         PetscCall(PCMGSetX(pc, i, tvec));
11079566063dSJacob Faibussowitsch         PetscCall(VecDestroy(&tvec));
1108f3fbd535SBarry Smith       }
110941b6fd38SMatthew G. Knepley       if (doCR) {
11109566063dSJacob Faibussowitsch         PetscCall(VecDuplicate(mglevels[i]->b, &mglevels[i]->crx));
11119566063dSJacob Faibussowitsch         PetscCall(VecDuplicate(mglevels[i]->b, &mglevels[i]->crb));
11129566063dSJacob Faibussowitsch         PetscCall(VecZeroEntries(mglevels[i]->crb));
111341b6fd38SMatthew G. Knepley       }
1114f3fbd535SBarry Smith     }
1115f3fbd535SBarry Smith     if (n != 1 && !mglevels[n - 1]->r) {
1116f3fbd535SBarry Smith       /* PCMGSetR() on the finest level if user did not supply it */
1117f3fbd535SBarry Smith       Vec *vec;
11189566063dSJacob Faibussowitsch       PetscCall(KSPCreateVecs(mglevels[n - 1]->smoothd, 1, &vec, 0, NULL));
11199566063dSJacob Faibussowitsch       PetscCall(PCMGSetR(pc, n - 1, *vec));
11209566063dSJacob Faibussowitsch       PetscCall(VecDestroy(vec));
11219566063dSJacob Faibussowitsch       PetscCall(PetscFree(vec));
1122f3fbd535SBarry Smith     }
112341b6fd38SMatthew G. Knepley     if (doCR) {
11249566063dSJacob Faibussowitsch       PetscCall(VecDuplicate(mglevels[n - 1]->r, &mglevels[n - 1]->crx));
11259566063dSJacob Faibussowitsch       PetscCall(VecDuplicate(mglevels[n - 1]->r, &mglevels[n - 1]->crb));
11269566063dSJacob Faibussowitsch       PetscCall(VecZeroEntries(mglevels[n - 1]->crb));
112741b6fd38SMatthew G. Knepley     }
1128f3fbd535SBarry Smith   }
1129f3fbd535SBarry Smith 
113098e04984SBarry Smith   if (pc->dm) {
113198e04984SBarry Smith     /* need to tell all the coarser levels to rebuild the matrix using the DM for that level */
113298e04984SBarry Smith     for (i = 0; i < n - 1; i++) {
113398e04984SBarry Smith       if (mglevels[i]->smoothd->setupstage != KSP_SETUP_NEW) mglevels[i]->smoothd->setupstage = KSP_SETUP_NEWMATRIX;
113498e04984SBarry Smith     }
113598e04984SBarry Smith   }
113608549ed4SJed Brown   // We got here (PCSetUp_MG) because the matrix has changed, which means the smoother needs to be set up again (e.g.,
113708549ed4SJed Brown   // new diagonal for Jacobi). Setting it here allows it to be logged under PCSetUp rather than deep inside a PCApply.
113808549ed4SJed Brown   if (mglevels[n - 1]->smoothd->setupstage != KSP_SETUP_NEW) mglevels[n - 1]->smoothd->setupstage = KSP_SETUP_NEWMATRIX;
1139f3fbd535SBarry Smith 
1140f3fbd535SBarry Smith   for (i = 1; i < n; i++) {
11412a21e185SBarry Smith     if (mglevels[i]->smoothu == mglevels[i]->smoothd || mg->am == PC_MG_FULL || mg->am == PC_MG_KASKADE || mg->cyclesperpcapply > 1) {
1142f3fbd535SBarry Smith       /* if doing only down then initial guess is zero */
11439566063dSJacob Faibussowitsch       PetscCall(KSPSetInitialGuessNonzero(mglevels[i]->smoothd, PETSC_TRUE));
1144f3fbd535SBarry Smith     }
11459566063dSJacob Faibussowitsch     if (mglevels[i]->cr) PetscCall(KSPSetInitialGuessNonzero(mglevels[i]->cr, PETSC_TRUE));
11469566063dSJacob Faibussowitsch     if (mglevels[i]->eventsmoothsetup) PetscCall(PetscLogEventBegin(mglevels[i]->eventsmoothsetup, 0, 0, 0, 0));
11479566063dSJacob Faibussowitsch     PetscCall(KSPSetUp(mglevels[i]->smoothd));
1148d4645a75SStefano Zampini     if (mglevels[i]->smoothd->reason) pc->failedreason = PC_SUBPC_ERROR;
11499566063dSJacob Faibussowitsch     if (mglevels[i]->eventsmoothsetup) PetscCall(PetscLogEventEnd(mglevels[i]->eventsmoothsetup, 0, 0, 0, 0));
1150d42688cbSBarry Smith     if (!mglevels[i]->residual) {
1151d42688cbSBarry Smith       Mat mat;
11529566063dSJacob Faibussowitsch       PetscCall(KSPGetOperators(mglevels[i]->smoothd, &mat, NULL));
11539566063dSJacob Faibussowitsch       PetscCall(PCMGSetResidual(pc, i, PCMGResidualDefault, mat));
1154d42688cbSBarry Smith     }
1155fcb023d4SJed Brown     if (!mglevels[i]->residualtranspose) {
1156fcb023d4SJed Brown       Mat mat;
11579566063dSJacob Faibussowitsch       PetscCall(KSPGetOperators(mglevels[i]->smoothd, &mat, NULL));
11589566063dSJacob Faibussowitsch       PetscCall(PCMGSetResidualTranspose(pc, i, PCMGResidualTransposeDefault, mat));
1159fcb023d4SJed Brown     }
1160f3fbd535SBarry Smith   }
1161f3fbd535SBarry Smith   for (i = 1; i < n; i++) {
1162f3fbd535SBarry Smith     if (mglevels[i]->smoothu && mglevels[i]->smoothu != mglevels[i]->smoothd) {
1163f3fbd535SBarry Smith       Mat downmat, downpmat;
1164f3fbd535SBarry Smith 
1165f3fbd535SBarry Smith       /* check if operators have been set for up, if not use down operators to set them */
11669566063dSJacob Faibussowitsch       PetscCall(KSPGetOperatorsSet(mglevels[i]->smoothu, &opsset, NULL));
1167f3fbd535SBarry Smith       if (!opsset) {
11689566063dSJacob Faibussowitsch         PetscCall(KSPGetOperators(mglevels[i]->smoothd, &downmat, &downpmat));
11699566063dSJacob Faibussowitsch         PetscCall(KSPSetOperators(mglevels[i]->smoothu, downmat, downpmat));
1170f3fbd535SBarry Smith       }
1171f3fbd535SBarry Smith 
11729566063dSJacob Faibussowitsch       PetscCall(KSPSetInitialGuessNonzero(mglevels[i]->smoothu, PETSC_TRUE));
11739566063dSJacob Faibussowitsch       if (mglevels[i]->eventsmoothsetup) PetscCall(PetscLogEventBegin(mglevels[i]->eventsmoothsetup, 0, 0, 0, 0));
11749566063dSJacob Faibussowitsch       PetscCall(KSPSetUp(mglevels[i]->smoothu));
1175d4645a75SStefano Zampini       if (mglevels[i]->smoothu->reason) pc->failedreason = PC_SUBPC_ERROR;
11769566063dSJacob Faibussowitsch       if (mglevels[i]->eventsmoothsetup) PetscCall(PetscLogEventEnd(mglevels[i]->eventsmoothsetup, 0, 0, 0, 0));
1177f3fbd535SBarry Smith     }
117841b6fd38SMatthew G. Knepley     if (mglevels[i]->cr) {
117941b6fd38SMatthew G. Knepley       Mat downmat, downpmat;
118041b6fd38SMatthew G. Knepley 
118141b6fd38SMatthew G. Knepley       /* check if operators have been set for up, if not use down operators to set them */
11829566063dSJacob Faibussowitsch       PetscCall(KSPGetOperatorsSet(mglevels[i]->cr, &opsset, NULL));
118341b6fd38SMatthew G. Knepley       if (!opsset) {
11849566063dSJacob Faibussowitsch         PetscCall(KSPGetOperators(mglevels[i]->smoothd, &downmat, &downpmat));
11859566063dSJacob Faibussowitsch         PetscCall(KSPSetOperators(mglevels[i]->cr, downmat, downpmat));
118641b6fd38SMatthew G. Knepley       }
118741b6fd38SMatthew G. Knepley 
11889566063dSJacob Faibussowitsch       PetscCall(KSPSetInitialGuessNonzero(mglevels[i]->cr, PETSC_TRUE));
11899566063dSJacob Faibussowitsch       if (mglevels[i]->eventsmoothsetup) PetscCall(PetscLogEventBegin(mglevels[i]->eventsmoothsetup, 0, 0, 0, 0));
11909566063dSJacob Faibussowitsch       PetscCall(KSPSetUp(mglevels[i]->cr));
1191d4645a75SStefano Zampini       if (mglevels[i]->cr->reason) pc->failedreason = PC_SUBPC_ERROR;
11929566063dSJacob Faibussowitsch       if (mglevels[i]->eventsmoothsetup) PetscCall(PetscLogEventEnd(mglevels[i]->eventsmoothsetup, 0, 0, 0, 0));
119341b6fd38SMatthew G. Knepley     }
1194f3fbd535SBarry Smith   }
1195f3fbd535SBarry Smith 
11969566063dSJacob Faibussowitsch   if (mglevels[0]->eventsmoothsetup) PetscCall(PetscLogEventBegin(mglevels[0]->eventsmoothsetup, 0, 0, 0, 0));
11979566063dSJacob Faibussowitsch   PetscCall(KSPSetUp(mglevels[0]->smoothd));
1198d4645a75SStefano Zampini   if (mglevels[0]->smoothd->reason) pc->failedreason = PC_SUBPC_ERROR;
11999566063dSJacob Faibussowitsch   if (mglevels[0]->eventsmoothsetup) PetscCall(PetscLogEventEnd(mglevels[0]->eventsmoothsetup, 0, 0, 0, 0));
1200f3fbd535SBarry Smith 
1201f3fbd535SBarry Smith     /*
1202f3fbd535SBarry Smith      Dump the interpolation/restriction matrices plus the
1203e3c5b3baSBarry Smith    Jacobian/stiffness on each level. This allows MATLAB users to
1204f3fbd535SBarry Smith    easily check if the Galerkin condition A_c = R A_f R^T is satisfied.
1205f3fbd535SBarry Smith 
1206f3fbd535SBarry Smith    Only support one or the other at the same time.
1207f3fbd535SBarry Smith   */
1208f3fbd535SBarry Smith #if defined(PETSC_USE_SOCKET_VIEWER)
12099566063dSJacob Faibussowitsch   PetscCall(PetscOptionsGetBool(((PetscObject)pc)->options, ((PetscObject)pc)->prefix, "-pc_mg_dump_matlab", &dump, NULL));
1210ce94432eSBarry Smith   if (dump) viewer = PETSC_VIEWER_SOCKET_(PetscObjectComm((PetscObject)pc));
1211f3fbd535SBarry Smith   dump = PETSC_FALSE;
1212f3fbd535SBarry Smith #endif
12139566063dSJacob Faibussowitsch   PetscCall(PetscOptionsGetBool(((PetscObject)pc)->options, ((PetscObject)pc)->prefix, "-pc_mg_dump_binary", &dump, NULL));
1214ce94432eSBarry Smith   if (dump) viewer = PETSC_VIEWER_BINARY_(PetscObjectComm((PetscObject)pc));
1215f3fbd535SBarry Smith 
1216f3fbd535SBarry Smith   if (viewer) {
121748a46eb9SPierre Jolivet     for (i = 1; i < n; i++) PetscCall(MatView(mglevels[i]->restrct, viewer));
1218f3fbd535SBarry Smith     for (i = 0; i < n; i++) {
12199566063dSJacob Faibussowitsch       PetscCall(KSPGetPC(mglevels[i]->smoothd, &pc));
12209566063dSJacob Faibussowitsch       PetscCall(MatView(pc->mat, viewer));
1221f3fbd535SBarry Smith     }
1222f3fbd535SBarry Smith   }
12233ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
1224f3fbd535SBarry Smith }
1225f3fbd535SBarry Smith 
1226f3fbd535SBarry Smith /* -------------------------------------------------------------------------------------*/
1227f3fbd535SBarry Smith 
1228d71ae5a4SJacob Faibussowitsch PetscErrorCode PCMGGetLevels_MG(PC pc, PetscInt *levels)
1229d71ae5a4SJacob Faibussowitsch {
123097d33e41SMatthew G. Knepley   PC_MG *mg = (PC_MG *)pc->data;
123197d33e41SMatthew G. Knepley 
123297d33e41SMatthew G. Knepley   PetscFunctionBegin;
123397d33e41SMatthew G. Knepley   *levels = mg->nlevels;
12343ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
123597d33e41SMatthew G. Knepley }
123697d33e41SMatthew G. Knepley 
12374b9ad928SBarry Smith /*@
1238f1580f4eSBarry Smith   PCMGGetLevels - Gets the number of levels to use with `PCMG`.
12394b9ad928SBarry Smith 
12404b9ad928SBarry Smith   Not Collective
12414b9ad928SBarry Smith 
12424b9ad928SBarry Smith   Input Parameter:
12434b9ad928SBarry Smith . pc - the preconditioner context
12444b9ad928SBarry Smith 
1245feefa0e1SJacob Faibussowitsch   Output Parameter:
12464b9ad928SBarry Smith . levels - the number of levels
12474b9ad928SBarry Smith 
12484b9ad928SBarry Smith   Level: advanced
12494b9ad928SBarry Smith 
1250f1580f4eSBarry Smith .seealso: `PCMG`, `PCMGSetLevels()`
12514b9ad928SBarry Smith @*/
1252d71ae5a4SJacob Faibussowitsch PetscErrorCode PCMGGetLevels(PC pc, PetscInt *levels)
1253d71ae5a4SJacob Faibussowitsch {
12544b9ad928SBarry Smith   PetscFunctionBegin;
12550700a824SBarry Smith   PetscValidHeaderSpecific(pc, PC_CLASSID, 1);
12564f572ea9SToby Isaac   PetscAssertPointer(levels, 2);
125797d33e41SMatthew G. Knepley   *levels = 0;
1258cac4c232SBarry Smith   PetscTryMethod(pc, "PCMGGetLevels_C", (PC, PetscInt *), (pc, levels));
12593ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
12604b9ad928SBarry Smith }
12614b9ad928SBarry Smith 
12624b9ad928SBarry Smith /*@
1263f1580f4eSBarry Smith   PCMGGetGridComplexity - compute operator and grid complexity of the `PCMG` hierarchy
1264e7d4b4cbSMark Adams 
1265e7d4b4cbSMark Adams   Input Parameter:
1266e7d4b4cbSMark Adams . pc - the preconditioner context
1267e7d4b4cbSMark Adams 
126890db8557SMark Adams   Output Parameters:
126990db8557SMark Adams + gc - grid complexity = sum_i(n_i) / n_0
127090db8557SMark Adams - oc - operator complexity = sum_i(nnz_i) / nnz_0
1271e7d4b4cbSMark Adams 
1272e7d4b4cbSMark Adams   Level: advanced
1273e7d4b4cbSMark Adams 
1274f1580f4eSBarry Smith   Note:
1275f1580f4eSBarry Smith   This is often call the operator complexity in multigrid literature
1276f1580f4eSBarry Smith 
1277f1580f4eSBarry Smith .seealso: `PCMG`, `PCMGGetLevels()`, `PCMGSetLevels()`
1278e7d4b4cbSMark Adams @*/
1279d71ae5a4SJacob Faibussowitsch PetscErrorCode PCMGGetGridComplexity(PC pc, PetscReal *gc, PetscReal *oc)
1280d71ae5a4SJacob Faibussowitsch {
1281e7d4b4cbSMark Adams   PC_MG         *mg       = (PC_MG *)pc->data;
1282e7d4b4cbSMark Adams   PC_MG_Levels **mglevels = mg->levels;
1283e7d4b4cbSMark Adams   PetscInt       lev, N;
1284e7d4b4cbSMark Adams   PetscLogDouble nnz0 = 0, sgc = 0, soc = 0, n0 = 0;
1285e7d4b4cbSMark Adams   MatInfo        info;
1286e7d4b4cbSMark Adams 
1287e7d4b4cbSMark Adams   PetscFunctionBegin;
128890db8557SMark Adams   PetscValidHeaderSpecific(pc, PC_CLASSID, 1);
12894f572ea9SToby Isaac   if (gc) PetscAssertPointer(gc, 2);
12904f572ea9SToby Isaac   if (oc) PetscAssertPointer(oc, 3);
1291e7d4b4cbSMark Adams   if (!pc->setupcalled) {
129290db8557SMark Adams     if (gc) *gc = 0;
129390db8557SMark Adams     if (oc) *oc = 0;
12943ba16761SJacob Faibussowitsch     PetscFunctionReturn(PETSC_SUCCESS);
1295e7d4b4cbSMark Adams   }
1296e7d4b4cbSMark Adams   PetscCheck(mg->nlevels > 0, PETSC_COMM_SELF, PETSC_ERR_PLIB, "MG has no levels");
1297e7d4b4cbSMark Adams   for (lev = 0; lev < mg->nlevels; lev++) {
1298e7d4b4cbSMark Adams     Mat dB;
12999566063dSJacob Faibussowitsch     PetscCall(KSPGetOperators(mglevels[lev]->smoothd, NULL, &dB));
13009566063dSJacob Faibussowitsch     PetscCall(MatGetInfo(dB, MAT_GLOBAL_SUM, &info)); /* global reduction */
13019566063dSJacob Faibussowitsch     PetscCall(MatGetSize(dB, &N, NULL));
1302e7d4b4cbSMark Adams     sgc += N;
1303e7d4b4cbSMark Adams     soc += info.nz_used;
13049371c9d4SSatish Balay     if (lev == mg->nlevels - 1) {
13059371c9d4SSatish Balay       nnz0 = info.nz_used;
13069371c9d4SSatish Balay       n0   = N;
13079371c9d4SSatish Balay     }
1308e7d4b4cbSMark Adams   }
13090fdf79fbSJacob Faibussowitsch   PetscCheck(n0 > 0 && gc, PETSC_COMM_SELF, PETSC_ERR_PLIB, "Number for grid points on finest level is not available");
13100fdf79fbSJacob Faibussowitsch   *gc = (PetscReal)(sgc / n0);
131190db8557SMark Adams   if (nnz0 > 0 && oc) *oc = (PetscReal)(soc / nnz0);
13123ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
1313e7d4b4cbSMark Adams }
1314e7d4b4cbSMark Adams 
1315e7d4b4cbSMark Adams /*@
131697177400SBarry Smith   PCMGSetType - Determines the form of multigrid to use:
13174b9ad928SBarry Smith   multiplicative, additive, full, or the Kaskade algorithm.
13184b9ad928SBarry Smith 
1319c3339decSBarry Smith   Logically Collective
13204b9ad928SBarry Smith 
13214b9ad928SBarry Smith   Input Parameters:
13224b9ad928SBarry Smith + pc   - the preconditioner context
1323f1580f4eSBarry Smith - form - multigrid form, one of `PC_MG_MULTIPLICATIVE`, `PC_MG_ADDITIVE`, `PC_MG_FULL`, `PC_MG_KASKADE`
13244b9ad928SBarry Smith 
13254b9ad928SBarry Smith   Options Database Key:
132620f4b53cSBarry Smith . -pc_mg_type <form> - Sets <form>, one of multiplicative, additive, full, kaskade
13274b9ad928SBarry Smith 
13284b9ad928SBarry Smith   Level: advanced
13294b9ad928SBarry Smith 
1330f1580f4eSBarry Smith .seealso: `PCMGType`, `PCMG`, `PCMGGetLevels()`, `PCMGSetLevels()`, `PCMGGetType()`, `PCMGCycleType`
13314b9ad928SBarry Smith @*/
1332d71ae5a4SJacob Faibussowitsch PetscErrorCode PCMGSetType(PC pc, PCMGType form)
1333d71ae5a4SJacob Faibussowitsch {
1334f3fbd535SBarry Smith   PC_MG *mg = (PC_MG *)pc->data;
13354b9ad928SBarry Smith 
13364b9ad928SBarry Smith   PetscFunctionBegin;
13370700a824SBarry Smith   PetscValidHeaderSpecific(pc, PC_CLASSID, 1);
1338c5eb9154SBarry Smith   PetscValidLogicalCollectiveEnum(pc, form, 2);
133931567311SBarry Smith   mg->am = form;
13409dcbbd2bSBarry Smith   if (form == PC_MG_MULTIPLICATIVE) pc->ops->applyrichardson = PCApplyRichardson_MG;
1341c60c7ad4SBarry Smith   else pc->ops->applyrichardson = NULL;
13423ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
1343c60c7ad4SBarry Smith }
1344c60c7ad4SBarry Smith 
1345c60c7ad4SBarry Smith /*@
1346f1580f4eSBarry Smith   PCMGGetType - Finds the form of multigrid the `PCMG` is using  multiplicative, additive, full, or the Kaskade algorithm.
1347c60c7ad4SBarry Smith 
1348c3339decSBarry Smith   Logically Collective
1349c60c7ad4SBarry Smith 
1350c60c7ad4SBarry Smith   Input Parameter:
1351c60c7ad4SBarry Smith . pc - the preconditioner context
1352c60c7ad4SBarry Smith 
1353c60c7ad4SBarry Smith   Output Parameter:
1354f1580f4eSBarry Smith . type - one of `PC_MG_MULTIPLICATIVE`, `PC_MG_ADDITIVE`, `PC_MG_FULL`, `PC_MG_KASKADE`, `PCMGCycleType`
1355c60c7ad4SBarry Smith 
1356c60c7ad4SBarry Smith   Level: advanced
1357c60c7ad4SBarry Smith 
1358f1580f4eSBarry Smith .seealso: `PCMGType`, `PCMG`, `PCMGGetLevels()`, `PCMGSetLevels()`, `PCMGSetType()`
1359c60c7ad4SBarry Smith @*/
1360d71ae5a4SJacob Faibussowitsch PetscErrorCode PCMGGetType(PC pc, PCMGType *type)
1361d71ae5a4SJacob Faibussowitsch {
1362c60c7ad4SBarry Smith   PC_MG *mg = (PC_MG *)pc->data;
1363c60c7ad4SBarry Smith 
1364c60c7ad4SBarry Smith   PetscFunctionBegin;
1365c60c7ad4SBarry Smith   PetscValidHeaderSpecific(pc, PC_CLASSID, 1);
1366c60c7ad4SBarry Smith   *type = mg->am;
13673ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
13684b9ad928SBarry Smith }
13694b9ad928SBarry Smith 
13704b9ad928SBarry Smith /*@
1371f1580f4eSBarry Smith   PCMGSetCycleType - Sets the type cycles to use.  Use `PCMGSetCycleTypeOnLevel()` for more
13724b9ad928SBarry Smith   complicated cycling.
13734b9ad928SBarry Smith 
1374c3339decSBarry Smith   Logically Collective
13754b9ad928SBarry Smith 
13764b9ad928SBarry Smith   Input Parameters:
1377c2be2410SBarry Smith + pc - the multigrid context
1378f1580f4eSBarry Smith - n  - either `PC_MG_CYCLE_V` or `PC_MG_CYCLE_W`
13794b9ad928SBarry Smith 
13804b9ad928SBarry Smith   Options Database Key:
1381c1cbb1deSBarry Smith . -pc_mg_cycle_type <v,w> - provide the cycle desired
13824b9ad928SBarry Smith 
13834b9ad928SBarry Smith   Level: advanced
13844b9ad928SBarry Smith 
1385f1580f4eSBarry Smith .seealso: `PCMG`, `PCMGSetCycleTypeOnLevel()`, `PCMGType`, `PCMGCycleType`
13864b9ad928SBarry Smith @*/
1387d71ae5a4SJacob Faibussowitsch PetscErrorCode PCMGSetCycleType(PC pc, PCMGCycleType n)
1388d71ae5a4SJacob Faibussowitsch {
1389f3fbd535SBarry Smith   PC_MG         *mg       = (PC_MG *)pc->data;
1390f3fbd535SBarry Smith   PC_MG_Levels **mglevels = mg->levels;
139179416396SBarry Smith   PetscInt       i, levels;
13924b9ad928SBarry Smith 
13934b9ad928SBarry Smith   PetscFunctionBegin;
13940700a824SBarry Smith   PetscValidHeaderSpecific(pc, PC_CLASSID, 1);
139569fbec6eSBarry Smith   PetscValidLogicalCollectiveEnum(pc, n, 2);
139628b400f6SJacob Faibussowitsch   PetscCheck(mglevels, PetscObjectComm((PetscObject)pc), PETSC_ERR_ORDER, "Must set MG levels with PCMGSetLevels() before calling");
1397f3fbd535SBarry Smith   levels = mglevels[0]->levels;
13982fa5cd67SKarl Rupp   for (i = 0; i < levels; i++) mglevels[i]->cycles = n;
13993ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
14004b9ad928SBarry Smith }
14014b9ad928SBarry Smith 
14028cc2d5dfSBarry Smith /*@
14038cc2d5dfSBarry Smith   PCMGMultiplicativeSetCycles - Sets the number of cycles to use for each preconditioner step
1404f1580f4eSBarry Smith   of multigrid when `PCMGType` is `PC_MG_MULTIPLICATIVE`
14058cc2d5dfSBarry Smith 
1406c3339decSBarry Smith   Logically Collective
14078cc2d5dfSBarry Smith 
14088cc2d5dfSBarry Smith   Input Parameters:
14098cc2d5dfSBarry Smith + pc - the multigrid context
14108cc2d5dfSBarry Smith - n  - number of cycles (default is 1)
14118cc2d5dfSBarry Smith 
14128cc2d5dfSBarry Smith   Options Database Key:
1413147403d9SBarry Smith . -pc_mg_multiplicative_cycles n - set the number of cycles
14148cc2d5dfSBarry Smith 
14158cc2d5dfSBarry Smith   Level: advanced
14168cc2d5dfSBarry Smith 
1417f1580f4eSBarry Smith   Note:
1418f1580f4eSBarry Smith   This is not associated with setting a v or w cycle, that is set with `PCMGSetCycleType()`
14198cc2d5dfSBarry Smith 
1420f1580f4eSBarry Smith .seealso: `PCMGSetCycleTypeOnLevel()`, `PCMGSetCycleType()`, `PCMGCycleType`, `PCMGType`
14218cc2d5dfSBarry Smith @*/
1422d71ae5a4SJacob Faibussowitsch PetscErrorCode PCMGMultiplicativeSetCycles(PC pc, PetscInt n)
1423d71ae5a4SJacob Faibussowitsch {
1424f3fbd535SBarry Smith   PC_MG *mg = (PC_MG *)pc->data;
14258cc2d5dfSBarry Smith 
14268cc2d5dfSBarry Smith   PetscFunctionBegin;
14270700a824SBarry Smith   PetscValidHeaderSpecific(pc, PC_CLASSID, 1);
1428c5eb9154SBarry Smith   PetscValidLogicalCollectiveInt(pc, n, 2);
14292a21e185SBarry Smith   mg->cyclesperpcapply = n;
14303ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
14318cc2d5dfSBarry Smith }
14328cc2d5dfSBarry Smith 
1433d71ae5a4SJacob Faibussowitsch PetscErrorCode PCMGSetGalerkin_MG(PC pc, PCMGGalerkinType use)
1434d71ae5a4SJacob Faibussowitsch {
1435fb15c04eSBarry Smith   PC_MG *mg = (PC_MG *)pc->data;
1436fb15c04eSBarry Smith 
1437fb15c04eSBarry Smith   PetscFunctionBegin;
14382134b1e4SBarry Smith   mg->galerkin = use;
14393ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
1440fb15c04eSBarry Smith }
1441fb15c04eSBarry Smith 
1442c2be2410SBarry Smith /*@
144397177400SBarry Smith   PCMGSetGalerkin - Causes the coarser grid matrices to be computed from the
144482b87d87SMatthew G. Knepley   finest grid via the Galerkin process: A_i-1 = r_i * A_i * p_i
1445c2be2410SBarry Smith 
1446c3339decSBarry Smith   Logically Collective
1447c2be2410SBarry Smith 
1448c2be2410SBarry Smith   Input Parameters:
1449c91913d3SJed Brown + pc  - the multigrid context
1450f1580f4eSBarry Smith - use - one of `PC_MG_GALERKIN_BOTH`, `PC_MG_GALERKIN_PMAT`, `PC_MG_GALERKIN_MAT`, or `PC_MG_GALERKIN_NONE`
1451c2be2410SBarry Smith 
1452c2be2410SBarry Smith   Options Database Key:
1453147403d9SBarry Smith . -pc_mg_galerkin <both,pmat,mat,none> - set the matrices to form via the Galerkin process
1454c2be2410SBarry Smith 
1455c2be2410SBarry Smith   Level: intermediate
1456c2be2410SBarry Smith 
1457f1580f4eSBarry Smith   Note:
1458f1580f4eSBarry Smith   Some codes that use `PCMG` such as `PCGAMG` use Galerkin internally while constructing the hierarchy and thus do not
1459f1580f4eSBarry Smith   use the `PCMG` construction of the coarser grids.
14601c1aac46SBarry Smith 
1461f1580f4eSBarry Smith .seealso: `PCMG`, `PCMGGetGalerkin()`, `PCMGGalerkinType`
1462c2be2410SBarry Smith @*/
1463d71ae5a4SJacob Faibussowitsch PetscErrorCode PCMGSetGalerkin(PC pc, PCMGGalerkinType use)
1464d71ae5a4SJacob Faibussowitsch {
1465c2be2410SBarry Smith   PetscFunctionBegin;
14660700a824SBarry Smith   PetscValidHeaderSpecific(pc, PC_CLASSID, 1);
1467cac4c232SBarry Smith   PetscTryMethod(pc, "PCMGSetGalerkin_C", (PC, PCMGGalerkinType), (pc, use));
14683ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
1469c2be2410SBarry Smith }
1470c2be2410SBarry Smith 
14713fc8bf9cSBarry Smith /*@
1472f1580f4eSBarry Smith   PCMGGetGalerkin - Checks if Galerkin multigrid is being used, i.e. A_i-1 = r_i * A_i * p_i
14733fc8bf9cSBarry Smith 
14743fc8bf9cSBarry Smith   Not Collective
14753fc8bf9cSBarry Smith 
14763fc8bf9cSBarry Smith   Input Parameter:
14773fc8bf9cSBarry Smith . pc - the multigrid context
14783fc8bf9cSBarry Smith 
14793fc8bf9cSBarry Smith   Output Parameter:
1480f1580f4eSBarry Smith . galerkin - one of `PC_MG_GALERKIN_BOTH`,`PC_MG_GALERKIN_PMAT`,`PC_MG_GALERKIN_MAT`, `PC_MG_GALERKIN_NONE`, or `PC_MG_GALERKIN_EXTERNAL`
14813fc8bf9cSBarry Smith 
14823fc8bf9cSBarry Smith   Level: intermediate
14833fc8bf9cSBarry Smith 
1484f1580f4eSBarry Smith .seealso: `PCMG`, `PCMGSetGalerkin()`, `PCMGGalerkinType`
14853fc8bf9cSBarry Smith @*/
1486d71ae5a4SJacob Faibussowitsch PetscErrorCode PCMGGetGalerkin(PC pc, PCMGGalerkinType *galerkin)
1487d71ae5a4SJacob Faibussowitsch {
1488f3fbd535SBarry Smith   PC_MG *mg = (PC_MG *)pc->data;
14893fc8bf9cSBarry Smith 
14903fc8bf9cSBarry Smith   PetscFunctionBegin;
14910700a824SBarry Smith   PetscValidHeaderSpecific(pc, PC_CLASSID, 1);
14922134b1e4SBarry Smith   *galerkin = mg->galerkin;
14933ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
14943fc8bf9cSBarry Smith }
14953fc8bf9cSBarry Smith 
1496d71ae5a4SJacob Faibussowitsch PetscErrorCode PCMGSetAdaptInterpolation_MG(PC pc, PetscBool adapt)
1497d71ae5a4SJacob Faibussowitsch {
1498f3b08a26SMatthew G. Knepley   PC_MG *mg = (PC_MG *)pc->data;
1499f3b08a26SMatthew G. Knepley 
1500f3b08a26SMatthew G. Knepley   PetscFunctionBegin;
1501f3b08a26SMatthew G. Knepley   mg->adaptInterpolation = adapt;
15023ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
1503f3b08a26SMatthew G. Knepley }
1504f3b08a26SMatthew G. Knepley 
1505d71ae5a4SJacob Faibussowitsch PetscErrorCode PCMGGetAdaptInterpolation_MG(PC pc, PetscBool *adapt)
1506d71ae5a4SJacob Faibussowitsch {
1507f3b08a26SMatthew G. Knepley   PC_MG *mg = (PC_MG *)pc->data;
1508f3b08a26SMatthew G. Knepley 
1509f3b08a26SMatthew G. Knepley   PetscFunctionBegin;
1510f3b08a26SMatthew G. Knepley   *adapt = mg->adaptInterpolation;
15113ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
1512f3b08a26SMatthew G. Knepley }
1513f3b08a26SMatthew G. Knepley 
1514d71ae5a4SJacob Faibussowitsch PetscErrorCode PCMGSetAdaptCoarseSpaceType_MG(PC pc, PCMGCoarseSpaceType ctype)
1515d71ae5a4SJacob Faibussowitsch {
15162b3cbbdaSStefano Zampini   PC_MG *mg = (PC_MG *)pc->data;
15172b3cbbdaSStefano Zampini 
15182b3cbbdaSStefano Zampini   PetscFunctionBegin;
15192b3cbbdaSStefano Zampini   mg->adaptInterpolation = ctype != PCMG_ADAPT_NONE ? PETSC_TRUE : PETSC_FALSE;
15202b3cbbdaSStefano Zampini   mg->coarseSpaceType    = ctype;
15213ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
15222b3cbbdaSStefano Zampini }
15232b3cbbdaSStefano Zampini 
1524d71ae5a4SJacob Faibussowitsch PetscErrorCode PCMGGetAdaptCoarseSpaceType_MG(PC pc, PCMGCoarseSpaceType *ctype)
1525d71ae5a4SJacob Faibussowitsch {
15262b3cbbdaSStefano Zampini   PC_MG *mg = (PC_MG *)pc->data;
15272b3cbbdaSStefano Zampini 
15282b3cbbdaSStefano Zampini   PetscFunctionBegin;
15292b3cbbdaSStefano Zampini   *ctype = mg->coarseSpaceType;
15303ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
15312b3cbbdaSStefano Zampini }
15322b3cbbdaSStefano Zampini 
1533d71ae5a4SJacob Faibussowitsch PetscErrorCode PCMGSetAdaptCR_MG(PC pc, PetscBool cr)
1534d71ae5a4SJacob Faibussowitsch {
153541b6fd38SMatthew G. Knepley   PC_MG *mg = (PC_MG *)pc->data;
153641b6fd38SMatthew G. Knepley 
153741b6fd38SMatthew G. Knepley   PetscFunctionBegin;
153841b6fd38SMatthew G. Knepley   mg->compatibleRelaxation = cr;
15393ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
154041b6fd38SMatthew G. Knepley }
154141b6fd38SMatthew G. Knepley 
1542d71ae5a4SJacob Faibussowitsch PetscErrorCode PCMGGetAdaptCR_MG(PC pc, PetscBool *cr)
1543d71ae5a4SJacob Faibussowitsch {
154441b6fd38SMatthew G. Knepley   PC_MG *mg = (PC_MG *)pc->data;
154541b6fd38SMatthew G. Knepley 
154641b6fd38SMatthew G. Knepley   PetscFunctionBegin;
154741b6fd38SMatthew G. Knepley   *cr = mg->compatibleRelaxation;
15483ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
154941b6fd38SMatthew G. Knepley }
155041b6fd38SMatthew G. Knepley 
15512b3cbbdaSStefano Zampini /*@C
15522b3cbbdaSStefano Zampini   PCMGSetAdaptCoarseSpaceType - Set the type of adaptive coarse space.
15532b3cbbdaSStefano Zampini 
15542b3cbbdaSStefano Zampini   Adapts or creates the interpolator based upon a vector space which should be accurately captured by the next coarser mesh, and thus accurately interpolated.
15552b3cbbdaSStefano Zampini 
1556c3339decSBarry Smith   Logically Collective
15572b3cbbdaSStefano Zampini 
15582b3cbbdaSStefano Zampini   Input Parameters:
15592b3cbbdaSStefano Zampini + pc    - the multigrid context
15602b3cbbdaSStefano Zampini - ctype - the type of coarse space
15612b3cbbdaSStefano Zampini 
15622b3cbbdaSStefano Zampini   Options Database Keys:
15632b3cbbdaSStefano Zampini + -pc_mg_adapt_interp_n <int>             - The number of modes to use
15642b3cbbdaSStefano Zampini - -pc_mg_adapt_interp_coarse_space <type> - The type of coarse space: none, polynomial, harmonic, eigenvector, generalized_eigenvector, gdsw
15652b3cbbdaSStefano Zampini 
15662b3cbbdaSStefano Zampini   Level: intermediate
15672b3cbbdaSStefano Zampini 
1568f1580f4eSBarry Smith .seealso: `PCMG`, `PCMGCoarseSpaceType`, `PCMGGetAdaptCoarseSpaceType()`, `PCMGSetGalerkin()`, `PCMGSetAdaptInterpolation()`
15692b3cbbdaSStefano Zampini @*/
1570d71ae5a4SJacob Faibussowitsch PetscErrorCode PCMGSetAdaptCoarseSpaceType(PC pc, PCMGCoarseSpaceType ctype)
1571d71ae5a4SJacob Faibussowitsch {
15722b3cbbdaSStefano Zampini   PetscFunctionBegin;
15732b3cbbdaSStefano Zampini   PetscValidHeaderSpecific(pc, PC_CLASSID, 1);
15742b3cbbdaSStefano Zampini   PetscValidLogicalCollectiveEnum(pc, ctype, 2);
15752b3cbbdaSStefano Zampini   PetscTryMethod(pc, "PCMGSetAdaptCoarseSpaceType_C", (PC, PCMGCoarseSpaceType), (pc, ctype));
15763ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
15772b3cbbdaSStefano Zampini }
15782b3cbbdaSStefano Zampini 
15792b3cbbdaSStefano Zampini /*@C
15802b3cbbdaSStefano Zampini   PCMGGetAdaptCoarseSpaceType - Get the type of adaptive coarse space.
15812b3cbbdaSStefano Zampini 
15822b3cbbdaSStefano Zampini   Not Collective
15832b3cbbdaSStefano Zampini 
15842b3cbbdaSStefano Zampini   Input Parameter:
15852b3cbbdaSStefano Zampini . pc - the multigrid context
15862b3cbbdaSStefano Zampini 
15872b3cbbdaSStefano Zampini   Output Parameter:
15882b3cbbdaSStefano Zampini . ctype - the type of coarse space
15892b3cbbdaSStefano Zampini 
15902b3cbbdaSStefano Zampini   Level: intermediate
15912b3cbbdaSStefano Zampini 
1592f1580f4eSBarry Smith .seealso: `PCMG`, `PCMGCoarseSpaceType`, `PCMGSetAdaptCoarseSpaceType()`, `PCMGSetGalerkin()`, `PCMGSetAdaptInterpolation()`
15932b3cbbdaSStefano Zampini @*/
1594d71ae5a4SJacob Faibussowitsch PetscErrorCode PCMGGetAdaptCoarseSpaceType(PC pc, PCMGCoarseSpaceType *ctype)
1595d71ae5a4SJacob Faibussowitsch {
15962b3cbbdaSStefano Zampini   PetscFunctionBegin;
15972b3cbbdaSStefano Zampini   PetscValidHeaderSpecific(pc, PC_CLASSID, 1);
15984f572ea9SToby Isaac   PetscAssertPointer(ctype, 2);
15992b3cbbdaSStefano Zampini   PetscUseMethod(pc, "PCMGGetAdaptCoarseSpaceType_C", (PC, PCMGCoarseSpaceType *), (pc, ctype));
16003ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
16012b3cbbdaSStefano Zampini }
16022b3cbbdaSStefano Zampini 
16032b3cbbdaSStefano Zampini /* MATT: REMOVE? */
1604f3b08a26SMatthew G. Knepley /*@
1605f3b08a26SMatthew G. Knepley   PCMGSetAdaptInterpolation - Adapt the interpolator based upon a vector space which should be accurately captured by the next coarser mesh, and thus accurately interpolated.
1606f3b08a26SMatthew G. Knepley 
1607c3339decSBarry Smith   Logically Collective
1608f3b08a26SMatthew G. Knepley 
1609f3b08a26SMatthew G. Knepley   Input Parameters:
1610f3b08a26SMatthew G. Knepley + pc    - the multigrid context
1611f3b08a26SMatthew G. Knepley - adapt - flag for adaptation of the interpolator
1612f3b08a26SMatthew G. Knepley 
1613f3b08a26SMatthew G. Knepley   Options Database Keys:
1614f3b08a26SMatthew G. Knepley + -pc_mg_adapt_interp                     - Turn on adaptation
1615f3b08a26SMatthew G. Knepley . -pc_mg_adapt_interp_n <int>             - The number of modes to use, should be divisible by dimension
1616f3b08a26SMatthew G. Knepley - -pc_mg_adapt_interp_coarse_space <type> - The type of coarse space: polynomial, harmonic, eigenvector, generalized_eigenvector
1617f3b08a26SMatthew G. Knepley 
1618f3b08a26SMatthew G. Knepley   Level: intermediate
1619f3b08a26SMatthew G. Knepley 
1620f1580f4eSBarry Smith .seealso: `PCMG`, `PCMGGetAdaptInterpolation()`, `PCMGSetGalerkin()`, `PCMGGetAdaptCoarseSpaceType()`, `PCMGSetAdaptCoarseSpaceType()`
1621f3b08a26SMatthew G. Knepley @*/
1622d71ae5a4SJacob Faibussowitsch PetscErrorCode PCMGSetAdaptInterpolation(PC pc, PetscBool adapt)
1623d71ae5a4SJacob Faibussowitsch {
1624f3b08a26SMatthew G. Knepley   PetscFunctionBegin;
1625f3b08a26SMatthew G. Knepley   PetscValidHeaderSpecific(pc, PC_CLASSID, 1);
1626cac4c232SBarry Smith   PetscTryMethod(pc, "PCMGSetAdaptInterpolation_C", (PC, PetscBool), (pc, adapt));
16273ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
1628f3b08a26SMatthew G. Knepley }
1629f3b08a26SMatthew G. Knepley 
1630f3b08a26SMatthew G. Knepley /*@
1631f1580f4eSBarry Smith   PCMGGetAdaptInterpolation - Get the flag to adapt the interpolator based upon a vector space which should be accurately captured by the next coarser mesh,
1632f1580f4eSBarry Smith   and thus accurately interpolated.
1633f3b08a26SMatthew G. Knepley 
163420f4b53cSBarry Smith   Not Collective
1635f3b08a26SMatthew G. Knepley 
1636f3b08a26SMatthew G. Knepley   Input Parameter:
1637f3b08a26SMatthew G. Knepley . pc - the multigrid context
1638f3b08a26SMatthew G. Knepley 
1639f3b08a26SMatthew G. Knepley   Output Parameter:
1640f3b08a26SMatthew G. Knepley . adapt - flag for adaptation of the interpolator
1641f3b08a26SMatthew G. Knepley 
1642f3b08a26SMatthew G. Knepley   Level: intermediate
1643f3b08a26SMatthew G. Knepley 
1644f1580f4eSBarry Smith .seealso: `PCMG`, `PCMGSetAdaptInterpolation()`, `PCMGSetGalerkin()`, `PCMGGetAdaptCoarseSpaceType()`, `PCMGSetAdaptCoarseSpaceType()`
1645f3b08a26SMatthew G. Knepley @*/
1646d71ae5a4SJacob Faibussowitsch PetscErrorCode PCMGGetAdaptInterpolation(PC pc, PetscBool *adapt)
1647d71ae5a4SJacob Faibussowitsch {
1648f3b08a26SMatthew G. Knepley   PetscFunctionBegin;
1649f3b08a26SMatthew G. Knepley   PetscValidHeaderSpecific(pc, PC_CLASSID, 1);
16504f572ea9SToby Isaac   PetscAssertPointer(adapt, 2);
1651cac4c232SBarry Smith   PetscUseMethod(pc, "PCMGGetAdaptInterpolation_C", (PC, PetscBool *), (pc, adapt));
16523ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
1653f3b08a26SMatthew G. Knepley }
1654f3b08a26SMatthew G. Knepley 
16554b9ad928SBarry Smith /*@
165641b6fd38SMatthew G. Knepley   PCMGSetAdaptCR - Monitor the coarse space quality using an auxiliary solve with compatible relaxation.
165741b6fd38SMatthew G. Knepley 
1658c3339decSBarry Smith   Logically Collective
165941b6fd38SMatthew G. Knepley 
166041b6fd38SMatthew G. Knepley   Input Parameters:
166141b6fd38SMatthew G. Knepley + pc - the multigrid context
166241b6fd38SMatthew G. Knepley - cr - flag for compatible relaxation
166341b6fd38SMatthew G. Knepley 
1664f1580f4eSBarry Smith   Options Database Key:
166541b6fd38SMatthew G. Knepley . -pc_mg_adapt_cr - Turn on compatible relaxation
166641b6fd38SMatthew G. Knepley 
166741b6fd38SMatthew G. Knepley   Level: intermediate
166841b6fd38SMatthew G. Knepley 
1669f1580f4eSBarry Smith .seealso: `PCMG`, `PCMGGetAdaptCR()`, `PCMGSetAdaptInterpolation()`, `PCMGSetGalerkin()`, `PCMGGetAdaptCoarseSpaceType()`, `PCMGSetAdaptCoarseSpaceType()`
167041b6fd38SMatthew G. Knepley @*/
1671d71ae5a4SJacob Faibussowitsch PetscErrorCode PCMGSetAdaptCR(PC pc, PetscBool cr)
1672d71ae5a4SJacob Faibussowitsch {
167341b6fd38SMatthew G. Knepley   PetscFunctionBegin;
167441b6fd38SMatthew G. Knepley   PetscValidHeaderSpecific(pc, PC_CLASSID, 1);
1675cac4c232SBarry Smith   PetscTryMethod(pc, "PCMGSetAdaptCR_C", (PC, PetscBool), (pc, cr));
16763ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
167741b6fd38SMatthew G. Knepley }
167841b6fd38SMatthew G. Knepley 
167941b6fd38SMatthew G. Knepley /*@
168041b6fd38SMatthew G. Knepley   PCMGGetAdaptCR - Get the flag to monitor coarse space quality using an auxiliary solve with compatible relaxation.
168141b6fd38SMatthew G. Knepley 
168220f4b53cSBarry Smith   Not Collective
168341b6fd38SMatthew G. Knepley 
168441b6fd38SMatthew G. Knepley   Input Parameter:
168541b6fd38SMatthew G. Knepley . pc - the multigrid context
168641b6fd38SMatthew G. Knepley 
168741b6fd38SMatthew G. Knepley   Output Parameter:
168841b6fd38SMatthew G. Knepley . cr - flag for compatible relaxaion
168941b6fd38SMatthew G. Knepley 
169041b6fd38SMatthew G. Knepley   Level: intermediate
169141b6fd38SMatthew G. Knepley 
1692f1580f4eSBarry Smith .seealso: `PCMGSetAdaptCR()`, `PCMGGetAdaptInterpolation()`, `PCMGSetGalerkin()`, `PCMGGetAdaptCoarseSpaceType()`, `PCMGSetAdaptCoarseSpaceType()`
169341b6fd38SMatthew G. Knepley @*/
1694d71ae5a4SJacob Faibussowitsch PetscErrorCode PCMGGetAdaptCR(PC pc, PetscBool *cr)
1695d71ae5a4SJacob Faibussowitsch {
169641b6fd38SMatthew G. Knepley   PetscFunctionBegin;
169741b6fd38SMatthew G. Knepley   PetscValidHeaderSpecific(pc, PC_CLASSID, 1);
16984f572ea9SToby Isaac   PetscAssertPointer(cr, 2);
1699cac4c232SBarry Smith   PetscUseMethod(pc, "PCMGGetAdaptCR_C", (PC, PetscBool *), (pc, cr));
17003ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
170141b6fd38SMatthew G. Knepley }
170241b6fd38SMatthew G. Knepley 
170341b6fd38SMatthew G. Knepley /*@
170406792cafSBarry Smith   PCMGSetNumberSmooth - Sets the number of pre and post-smoothing steps to use
1705f1580f4eSBarry Smith   on all levels.  Use `PCMGDistinctSmoothUp()` to create separate up and down smoothers if you want different numbers of
1706710315b6SLawrence Mitchell   pre- and post-smoothing steps.
170706792cafSBarry Smith 
1708c3339decSBarry Smith   Logically Collective
170906792cafSBarry Smith 
171006792cafSBarry Smith   Input Parameters:
1711feefa0e1SJacob Faibussowitsch + pc - the multigrid context
171206792cafSBarry Smith - n  - the number of smoothing steps
171306792cafSBarry Smith 
171406792cafSBarry Smith   Options Database Key:
1715a2b725a8SWilliam Gropp . -mg_levels_ksp_max_it <n> - Sets number of pre and post-smoothing steps
171606792cafSBarry Smith 
171706792cafSBarry Smith   Level: advanced
171806792cafSBarry Smith 
1719f1580f4eSBarry Smith   Note:
1720f1580f4eSBarry Smith   This does not set a value on the coarsest grid, since we assume that there is no separate smooth up on the coarsest grid.
172106792cafSBarry Smith 
1722f1580f4eSBarry Smith .seealso: `PCMG`, `PCMGSetDistinctSmoothUp()`
172306792cafSBarry Smith @*/
1724d71ae5a4SJacob Faibussowitsch PetscErrorCode PCMGSetNumberSmooth(PC pc, PetscInt n)
1725d71ae5a4SJacob Faibussowitsch {
172606792cafSBarry Smith   PC_MG         *mg       = (PC_MG *)pc->data;
172706792cafSBarry Smith   PC_MG_Levels **mglevels = mg->levels;
172806792cafSBarry Smith   PetscInt       i, levels;
172906792cafSBarry Smith 
173006792cafSBarry Smith   PetscFunctionBegin;
173106792cafSBarry Smith   PetscValidHeaderSpecific(pc, PC_CLASSID, 1);
173206792cafSBarry Smith   PetscValidLogicalCollectiveInt(pc, n, 2);
173328b400f6SJacob Faibussowitsch   PetscCheck(mglevels, PetscObjectComm((PetscObject)pc), PETSC_ERR_ORDER, "Must set MG levels with PCMGSetLevels() before calling");
173406792cafSBarry Smith   levels = mglevels[0]->levels;
173506792cafSBarry Smith 
173606792cafSBarry Smith   for (i = 1; i < levels; i++) {
17379566063dSJacob Faibussowitsch     PetscCall(KSPSetTolerances(mglevels[i]->smoothu, PETSC_DEFAULT, PETSC_DEFAULT, PETSC_DEFAULT, n));
17389566063dSJacob Faibussowitsch     PetscCall(KSPSetTolerances(mglevels[i]->smoothd, PETSC_DEFAULT, PETSC_DEFAULT, PETSC_DEFAULT, n));
173906792cafSBarry Smith     mg->default_smoothu = n;
174006792cafSBarry Smith     mg->default_smoothd = n;
174106792cafSBarry Smith   }
17423ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
174306792cafSBarry Smith }
174406792cafSBarry Smith 
1745f442ab6aSBarry Smith /*@
1746f1580f4eSBarry Smith   PCMGSetDistinctSmoothUp - sets the up (post) smoother to be a separate `KSP` from the down (pre) smoother on all levels
1747710315b6SLawrence Mitchell   and adds the suffix _up to the options name
1748f442ab6aSBarry Smith 
1749c3339decSBarry Smith   Logically Collective
1750f442ab6aSBarry Smith 
1751f1580f4eSBarry Smith   Input Parameter:
1752f442ab6aSBarry Smith . pc - the preconditioner context
1753f442ab6aSBarry Smith 
1754f442ab6aSBarry Smith   Options Database Key:
1755147403d9SBarry Smith . -pc_mg_distinct_smoothup <bool> - use distinct smoothing objects
1756f442ab6aSBarry Smith 
1757f442ab6aSBarry Smith   Level: advanced
1758f442ab6aSBarry Smith 
1759f1580f4eSBarry Smith   Note:
1760f1580f4eSBarry Smith   This does not set a value on the coarsest grid, since we assume that there is no separate smooth up on the coarsest grid.
1761f442ab6aSBarry Smith 
1762f1580f4eSBarry Smith .seealso: `PCMG`, `PCMGSetNumberSmooth()`
1763f442ab6aSBarry Smith @*/
1764d71ae5a4SJacob Faibussowitsch PetscErrorCode PCMGSetDistinctSmoothUp(PC pc)
1765d71ae5a4SJacob Faibussowitsch {
1766f442ab6aSBarry Smith   PC_MG         *mg       = (PC_MG *)pc->data;
1767f442ab6aSBarry Smith   PC_MG_Levels **mglevels = mg->levels;
1768f442ab6aSBarry Smith   PetscInt       i, levels;
1769f442ab6aSBarry Smith   KSP            subksp;
1770f442ab6aSBarry Smith 
1771f442ab6aSBarry Smith   PetscFunctionBegin;
1772f442ab6aSBarry Smith   PetscValidHeaderSpecific(pc, PC_CLASSID, 1);
177328b400f6SJacob Faibussowitsch   PetscCheck(mglevels, PetscObjectComm((PetscObject)pc), PETSC_ERR_ORDER, "Must set MG levels with PCMGSetLevels() before calling");
1774f442ab6aSBarry Smith   levels = mglevels[0]->levels;
1775f442ab6aSBarry Smith 
1776f442ab6aSBarry Smith   for (i = 1; i < levels; i++) {
1777710315b6SLawrence Mitchell     const char *prefix = NULL;
1778f442ab6aSBarry Smith     /* make sure smoother up and down are different */
17799566063dSJacob Faibussowitsch     PetscCall(PCMGGetSmootherUp(pc, i, &subksp));
17809566063dSJacob Faibussowitsch     PetscCall(KSPGetOptionsPrefix(mglevels[i]->smoothd, &prefix));
17819566063dSJacob Faibussowitsch     PetscCall(KSPSetOptionsPrefix(subksp, prefix));
17829566063dSJacob Faibussowitsch     PetscCall(KSPAppendOptionsPrefix(subksp, "up_"));
1783f442ab6aSBarry Smith   }
17843ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
1785f442ab6aSBarry Smith }
1786f442ab6aSBarry Smith 
178707a4832bSFande Kong /* No new matrices are created, and the coarse operator matrices are the references to the original ones */
1788d71ae5a4SJacob Faibussowitsch PetscErrorCode PCGetInterpolations_MG(PC pc, PetscInt *num_levels, Mat *interpolations[])
1789d71ae5a4SJacob Faibussowitsch {
179007a4832bSFande Kong   PC_MG         *mg       = (PC_MG *)pc->data;
179107a4832bSFande Kong   PC_MG_Levels **mglevels = mg->levels;
179207a4832bSFande Kong   Mat           *mat;
179307a4832bSFande Kong   PetscInt       l;
179407a4832bSFande Kong 
179507a4832bSFande Kong   PetscFunctionBegin;
179628b400f6SJacob Faibussowitsch   PetscCheck(mglevels, PetscObjectComm((PetscObject)pc), PETSC_ERR_ARG_WRONGSTATE, "Must set MG levels before calling");
17979566063dSJacob Faibussowitsch   PetscCall(PetscMalloc1(mg->nlevels, &mat));
179807a4832bSFande Kong   for (l = 1; l < mg->nlevels; l++) {
179907a4832bSFande Kong     mat[l - 1] = mglevels[l]->interpolate;
18009566063dSJacob Faibussowitsch     PetscCall(PetscObjectReference((PetscObject)mat[l - 1]));
180107a4832bSFande Kong   }
180207a4832bSFande Kong   *num_levels     = mg->nlevels;
180307a4832bSFande Kong   *interpolations = mat;
18043ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
180507a4832bSFande Kong }
180607a4832bSFande Kong 
180707a4832bSFande Kong /* No new matrices are created, and the coarse operator matrices are the references to the original ones */
1808d71ae5a4SJacob Faibussowitsch PetscErrorCode PCGetCoarseOperators_MG(PC pc, PetscInt *num_levels, Mat *coarseOperators[])
1809d71ae5a4SJacob Faibussowitsch {
181007a4832bSFande Kong   PC_MG         *mg       = (PC_MG *)pc->data;
181107a4832bSFande Kong   PC_MG_Levels **mglevels = mg->levels;
181207a4832bSFande Kong   PetscInt       l;
181307a4832bSFande Kong   Mat           *mat;
181407a4832bSFande Kong 
181507a4832bSFande Kong   PetscFunctionBegin;
181628b400f6SJacob Faibussowitsch   PetscCheck(mglevels, PetscObjectComm((PetscObject)pc), PETSC_ERR_ARG_WRONGSTATE, "Must set MG levels before calling");
18179566063dSJacob Faibussowitsch   PetscCall(PetscMalloc1(mg->nlevels, &mat));
181807a4832bSFande Kong   for (l = 0; l < mg->nlevels - 1; l++) {
18199566063dSJacob Faibussowitsch     PetscCall(KSPGetOperators(mglevels[l]->smoothd, NULL, &(mat[l])));
18209566063dSJacob Faibussowitsch     PetscCall(PetscObjectReference((PetscObject)mat[l]));
182107a4832bSFande Kong   }
182207a4832bSFande Kong   *num_levels      = mg->nlevels;
182307a4832bSFande Kong   *coarseOperators = mat;
18243ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
182507a4832bSFande Kong }
182607a4832bSFande Kong 
1827f3b08a26SMatthew G. Knepley /*@C
1828f1580f4eSBarry Smith   PCMGRegisterCoarseSpaceConstructor -  Adds a method to the `PCMG` package for coarse space construction.
1829f3b08a26SMatthew G. Knepley 
183020f4b53cSBarry Smith   Not Collective
1831f3b08a26SMatthew G. Knepley 
1832f3b08a26SMatthew G. Knepley   Input Parameters:
1833f3b08a26SMatthew G. Knepley + name     - name of the constructor
1834f3b08a26SMatthew G. Knepley - function - constructor routine
1835f3b08a26SMatthew G. Knepley 
183620f4b53cSBarry Smith   Calling sequence of `function`:
183720f4b53cSBarry Smith $  PetscErrorCode my_csp(PC pc, PetscInt l, DM dm, KSP smooth, PetscInt Nc, Mat initGuess, Mat *coarseSp)
183820f4b53cSBarry Smith +  pc        - The `PC` object
1839f1580f4eSBarry Smith .  l         - The multigrid level, 0 is the coarse level
184020f4b53cSBarry Smith .  dm        - The `DM` for this level
1841f1580f4eSBarry Smith .  smooth    - The level smoother
1842f1580f4eSBarry Smith .  Nc        - The size of the coarse space
1843f1580f4eSBarry Smith .  initGuess - Basis for an initial guess for the space
1844f1580f4eSBarry Smith -  coarseSp  - A basis for the computed coarse space
1845f3b08a26SMatthew G. Knepley 
1846f3b08a26SMatthew G. Knepley   Level: advanced
1847f3b08a26SMatthew G. Knepley 
1848feefa0e1SJacob Faibussowitsch   Developer Notes:
1849f1580f4eSBarry Smith   How come this is not used by `PCGAMG`?
1850f1580f4eSBarry Smith 
1851f1580f4eSBarry Smith .seealso: `PCMG`, `PCMGGetCoarseSpaceConstructor()`, `PCRegister()`
1852f3b08a26SMatthew G. Knepley @*/
1853d71ae5a4SJacob Faibussowitsch PetscErrorCode PCMGRegisterCoarseSpaceConstructor(const char name[], PetscErrorCode (*function)(PC, PetscInt, DM, KSP, PetscInt, Mat, Mat *))
1854d71ae5a4SJacob Faibussowitsch {
1855f3b08a26SMatthew G. Knepley   PetscFunctionBegin;
18569566063dSJacob Faibussowitsch   PetscCall(PCInitializePackage());
18579566063dSJacob Faibussowitsch   PetscCall(PetscFunctionListAdd(&PCMGCoarseList, name, function));
18583ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
1859f3b08a26SMatthew G. Knepley }
1860f3b08a26SMatthew G. Knepley 
1861f3b08a26SMatthew G. Knepley /*@C
1862f3b08a26SMatthew G. Knepley   PCMGGetCoarseSpaceConstructor -  Returns the given coarse space construction method.
1863f3b08a26SMatthew G. Knepley 
186420f4b53cSBarry Smith   Not Collective
1865f3b08a26SMatthew G. Knepley 
1866f3b08a26SMatthew G. Knepley   Input Parameter:
1867f3b08a26SMatthew G. Knepley . name - name of the constructor
1868f3b08a26SMatthew G. Knepley 
1869f3b08a26SMatthew G. Knepley   Output Parameter:
1870f3b08a26SMatthew G. Knepley . function - constructor routine
1871f3b08a26SMatthew G. Knepley 
1872f3b08a26SMatthew G. Knepley   Level: advanced
1873f3b08a26SMatthew G. Knepley 
1874f1580f4eSBarry Smith .seealso: `PCMG`, `PCMGRegisterCoarseSpaceConstructor()`, `PCRegister()`
1875f3b08a26SMatthew G. Knepley @*/
1876d71ae5a4SJacob Faibussowitsch PetscErrorCode PCMGGetCoarseSpaceConstructor(const char name[], PetscErrorCode (**function)(PC, PetscInt, DM, KSP, PetscInt, Mat, Mat *))
1877d71ae5a4SJacob Faibussowitsch {
1878f3b08a26SMatthew G. Knepley   PetscFunctionBegin;
18799566063dSJacob Faibussowitsch   PetscCall(PetscFunctionListFind(PCMGCoarseList, name, function));
18803ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
1881f3b08a26SMatthew G. Knepley }
1882f3b08a26SMatthew G. Knepley 
18833b09bd56SBarry Smith /*MC
1884ccb205f8SBarry Smith    PCMG - Use multigrid preconditioning. This preconditioner requires you provide additional
18853b09bd56SBarry Smith     information about the coarser grid matrices and restriction/interpolation operators.
18863b09bd56SBarry Smith 
18873b09bd56SBarry Smith    Options Database Keys:
18883b09bd56SBarry Smith +  -pc_mg_levels <nlevels> - number of levels including finest
1889391689abSStefano Zampini .  -pc_mg_cycle_type <v,w> - provide the cycle desired
18908c1c2452SJed Brown .  -pc_mg_type <additive,multiplicative,full,kaskade> - multiplicative is the default
18913b09bd56SBarry Smith .  -pc_mg_log - log information about time spent on each level of the solver
1892710315b6SLawrence Mitchell .  -pc_mg_distinct_smoothup - configure up (after interpolation) and down (before restriction) smoothers separately (with different options prefixes)
18932134b1e4SBarry Smith .  -pc_mg_galerkin <both,pmat,mat,none> - use Galerkin process to compute coarser operators, i.e. Acoarse = R A R'
18948cf6037eSBarry Smith .  -pc_mg_multiplicative_cycles - number of cycles to use as the preconditioner (defaults to 1)
18958cf6037eSBarry Smith .  -pc_mg_dump_matlab - dumps the matrices for each level and the restriction/interpolation matrices
1896e3c5b3baSBarry Smith                         to the Socket viewer for reading from MATLAB.
18978cf6037eSBarry Smith -  -pc_mg_dump_binary - dumps the matrices for each level and the restriction/interpolation matrices
18988cf6037eSBarry Smith                         to the binary output file called binaryoutput
18993b09bd56SBarry Smith 
190020f4b53cSBarry Smith    Level: intermediate
190120f4b53cSBarry Smith 
190295452b02SPatrick Sanan    Notes:
1903f1580f4eSBarry Smith     If one uses a Krylov method such `KSPGMRES` or `KSPCG` as the smoother then one must use `KSPFGMRES`, `KSPGCR`, or `KSPRICHARDSON` as the outer Krylov method
19043b09bd56SBarry Smith 
19058cf6037eSBarry Smith        When run with a single level the smoother options are used on that level NOT the coarse grid solver options
19068cf6037eSBarry Smith 
1907f1580f4eSBarry Smith        When run with `KSPRICHARDSON` the convergence test changes slightly if monitor is turned on. The iteration count may change slightly. This
190823067569SBarry Smith        is because without monitoring the residual norm is computed WITHIN each multigrid cycle on the finest level after the pre-smoothing
190923067569SBarry Smith        (because the residual has just been computed for the multigrid algorithm and is hence available for free) while with monitoring the
191023067569SBarry Smith        residual is computed at the end of each cycle.
191123067569SBarry Smith 
1912db781477SPatrick Sanan .seealso: `PCCreate()`, `PCSetType()`, `PCType`, `PC`, `PCMGType`, `PCEXOTIC`, `PCGAMG`, `PCML`, `PCHYPRE`
1913db781477SPatrick Sanan           `PCMGSetLevels()`, `PCMGGetLevels()`, `PCMGSetType()`, `PCMGSetCycleType()`,
1914db781477SPatrick Sanan           `PCMGSetDistinctSmoothUp()`, `PCMGGetCoarseSolve()`, `PCMGSetResidual()`, `PCMGSetInterpolation()`,
1915db781477SPatrick Sanan           `PCMGSetRestriction()`, `PCMGGetSmoother()`, `PCMGGetSmootherUp()`, `PCMGGetSmootherDown()`,
1916f1580f4eSBarry Smith           `PCMGSetCycleTypeOnLevel()`, `PCMGSetRhs()`, `PCMGSetX()`, `PCMGSetR()`,
1917f1580f4eSBarry Smith           `PCMGSetAdaptCR()`, `PCMGGetAdaptInterpolation()`, `PCMGSetGalerkin()`, `PCMGGetAdaptCoarseSpaceType()`, `PCMGSetAdaptCoarseSpaceType()`
19183b09bd56SBarry Smith M*/
19193b09bd56SBarry Smith 
1920d71ae5a4SJacob Faibussowitsch PETSC_EXTERN PetscErrorCode PCCreate_MG(PC pc)
1921d71ae5a4SJacob Faibussowitsch {
1922f3fbd535SBarry Smith   PC_MG *mg;
1923f3fbd535SBarry Smith 
19244b9ad928SBarry Smith   PetscFunctionBegin;
19254dfa11a4SJacob Faibussowitsch   PetscCall(PetscNew(&mg));
19263ec1f749SStefano Zampini   pc->data               = mg;
1927f3fbd535SBarry Smith   mg->nlevels            = -1;
192810eca3edSLisandro Dalcin   mg->am                 = PC_MG_MULTIPLICATIVE;
19292134b1e4SBarry Smith   mg->galerkin           = PC_MG_GALERKIN_NONE;
1930f3b08a26SMatthew G. Knepley   mg->adaptInterpolation = PETSC_FALSE;
1931f3b08a26SMatthew G. Knepley   mg->Nc                 = -1;
1932f3b08a26SMatthew G. Knepley   mg->eigenvalue         = -1;
1933f3fbd535SBarry Smith 
193437a44384SMark Adams   pc->useAmat = PETSC_TRUE;
193537a44384SMark Adams 
19364b9ad928SBarry Smith   pc->ops->apply          = PCApply_MG;
1937fcb023d4SJed Brown   pc->ops->applytranspose = PCApplyTranspose_MG;
193830b0564aSStefano Zampini   pc->ops->matapply       = PCMatApply_MG;
19394b9ad928SBarry Smith   pc->ops->setup          = PCSetUp_MG;
1940a06653b4SBarry Smith   pc->ops->reset          = PCReset_MG;
19414b9ad928SBarry Smith   pc->ops->destroy        = PCDestroy_MG;
19424b9ad928SBarry Smith   pc->ops->setfromoptions = PCSetFromOptions_MG;
19434b9ad928SBarry Smith   pc->ops->view           = PCView_MG;
1944fb15c04eSBarry Smith 
19459566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposedDataRegister(&mg->eigenvalue));
19469566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)pc, "PCMGSetGalerkin_C", PCMGSetGalerkin_MG));
19479566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)pc, "PCMGGetLevels_C", PCMGGetLevels_MG));
19489566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)pc, "PCMGSetLevels_C", PCMGSetLevels_MG));
19499566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)pc, "PCGetInterpolations_C", PCGetInterpolations_MG));
19509566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)pc, "PCGetCoarseOperators_C", PCGetCoarseOperators_MG));
19519566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)pc, "PCMGSetAdaptInterpolation_C", PCMGSetAdaptInterpolation_MG));
19529566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)pc, "PCMGGetAdaptInterpolation_C", PCMGGetAdaptInterpolation_MG));
19539566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)pc, "PCMGSetAdaptCR_C", PCMGSetAdaptCR_MG));
19549566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)pc, "PCMGGetAdaptCR_C", PCMGGetAdaptCR_MG));
19552b3cbbdaSStefano Zampini   PetscCall(PetscObjectComposeFunction((PetscObject)pc, "PCMGSetAdaptCoarseSpaceType_C", PCMGSetAdaptCoarseSpaceType_MG));
19562b3cbbdaSStefano Zampini   PetscCall(PetscObjectComposeFunction((PetscObject)pc, "PCMGGetAdaptCoarseSpaceType_C", PCMGGetAdaptCoarseSpaceType_MG));
19573ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
19584b9ad928SBarry Smith }
1959