xref: /petsc/src/ksp/pc/impls/mg/mg.c (revision ad4df1005a67cf9575a5c4294f11fd96229b2bb4)
1dba47a55SKris Buschelman #define PETSCKSP_DLL
2dba47a55SKris Buschelman 
34b9ad928SBarry Smith /*
44b9ad928SBarry Smith     Defines the multigrid preconditioner interface.
54b9ad928SBarry Smith */
67c4f633dSBarry Smith #include "../src/ksp/pc/impls/mg/mgimpl.h"                    /*I "petscmg.h" I*/
74b9ad928SBarry Smith 
84b9ad928SBarry Smith 
94b9ad928SBarry Smith #undef __FUNCT__
109dcbbd2bSBarry Smith #define __FUNCT__ "PCMGMCycle_Private"
1131567311SBarry Smith PetscErrorCode PCMGMCycle_Private(PC pc,PC_MG_Levels **mglevelsin,PCRichardsonConvergedReason *reason)
124b9ad928SBarry Smith {
1331567311SBarry Smith   PC_MG          *mg = (PC_MG*)pc->data;
1431567311SBarry Smith   PC_MG_Levels   *mgc,*mglevels = *mglevelsin;
156849ba73SBarry Smith   PetscErrorCode ierr;
1631567311SBarry Smith   PetscInt       cycles = (mglevels->level == 1) ? 1 : (PetscInt) mglevels->cycles;
174b9ad928SBarry Smith 
184b9ad928SBarry Smith   PetscFunctionBegin;
194b9ad928SBarry Smith 
2032cf1786SBarry Smith   if (mg->eventsmoothsolve) {ierr = PetscLogEventBegin(mg->eventsmoothsolve,0,0,0,0);CHKERRQ(ierr);}
2131567311SBarry Smith   ierr = KSPSolve(mglevels->smoothd,mglevels->b,mglevels->x);CHKERRQ(ierr);  /* pre-smooth */
2232cf1786SBarry Smith   if (mg->eventsmoothsolve) {ierr = PetscLogEventEnd(mg->eventsmoothsolve,0,0,0,0);CHKERRQ(ierr);}
2331567311SBarry Smith   if (mglevels->level) {  /* not the coarsest grid */
2432cf1786SBarry Smith     if (mg->eventresidual) {ierr = PetscLogEventBegin(mg->eventresidual,0,0,0,0);CHKERRQ(ierr);}
2531567311SBarry Smith     ierr = (*mglevels->residual)(mglevels->A,mglevels->b,mglevels->x,mglevels->r);CHKERRQ(ierr);
2632cf1786SBarry Smith     if (mg->eventresidual) {ierr = PetscLogEventEnd(mg->eventresidual,0,0,0,0);CHKERRQ(ierr);}
274b9ad928SBarry Smith 
284b9ad928SBarry Smith     /* if on finest level and have convergence criteria set */
2931567311SBarry Smith     if (mglevels->level == mglevels->levels-1 && mg->ttol && reason) {
304b9ad928SBarry Smith       PetscReal rnorm;
3131567311SBarry Smith       ierr = VecNorm(mglevels->r,NORM_2,&rnorm);CHKERRQ(ierr);
324b9ad928SBarry Smith       if (rnorm <= mg->ttol) {
3370441072SBarry Smith         if (rnorm < mg->abstol) {
344d0a8057SBarry Smith           *reason = PCRICHARDSON_CONVERGED_ATOL;
351e2582c4SBarry Smith           ierr = PetscInfo2(pc,"Linear solver has converged. Residual norm %G is less than absolute tolerance %G\n",rnorm,mg->abstol);CHKERRQ(ierr);
364b9ad928SBarry Smith         } else {
374d0a8057SBarry Smith           *reason = PCRICHARDSON_CONVERGED_RTOL;
381e2582c4SBarry Smith           ierr = PetscInfo2(pc,"Linear solver has converged. Residual norm %G is less than relative tolerance times initial residual norm %G\n",rnorm,mg->ttol);CHKERRQ(ierr);
394b9ad928SBarry Smith         }
404b9ad928SBarry Smith         PetscFunctionReturn(0);
414b9ad928SBarry Smith       }
424b9ad928SBarry Smith     }
434b9ad928SBarry Smith 
4431567311SBarry Smith     mgc = *(mglevelsin - 1);
4532cf1786SBarry Smith     if (mg->eventinterprestrict) {ierr = PetscLogEventBegin(mg->eventinterprestrict,0,0,0,0);CHKERRQ(ierr);}
4631567311SBarry Smith     ierr = MatRestrict(mglevels->restrct,mglevels->r,mgc->b);CHKERRQ(ierr);
4732cf1786SBarry Smith     if (mg->eventinterprestrict) {ierr = PetscLogEventEnd(mg->eventinterprestrict,0,0,0,0);CHKERRQ(ierr);}
48efb30889SBarry Smith     ierr = VecSet(mgc->x,0.0);CHKERRQ(ierr);
494b9ad928SBarry Smith     while (cycles--) {
5031567311SBarry Smith       ierr = PCMGMCycle_Private(pc,mglevelsin-1,reason);CHKERRQ(ierr);
514b9ad928SBarry Smith     }
5232cf1786SBarry Smith     if (mg->eventinterprestrict) {ierr = PetscLogEventBegin(mg->eventinterprestrict,0,0,0,0);CHKERRQ(ierr);}
5331567311SBarry Smith     ierr = MatInterpolateAdd(mglevels->interpolate,mgc->x,mglevels->x,mglevels->x);CHKERRQ(ierr);
5432cf1786SBarry Smith     if (mg->eventinterprestrict) {ierr = PetscLogEventEnd(mg->eventinterprestrict,0,0,0,0);CHKERRQ(ierr);}
5532cf1786SBarry Smith     if (mg->eventsmoothsolve) {ierr = PetscLogEventBegin(mg->eventsmoothsolve,0,0,0,0);CHKERRQ(ierr);}
5631567311SBarry Smith     ierr = KSPSolve(mglevels->smoothu,mglevels->b,mglevels->x);CHKERRQ(ierr);    /* post smooth */
5732cf1786SBarry Smith     if (mg->eventsmoothsolve) {ierr = PetscLogEventEnd(mg->eventsmoothsolve,0,0,0,0);CHKERRQ(ierr);}
584b9ad928SBarry Smith   }
594b9ad928SBarry Smith   PetscFunctionReturn(0);
604b9ad928SBarry Smith }
614b9ad928SBarry Smith 
624b9ad928SBarry Smith #undef __FUNCT__
634b9ad928SBarry Smith #define __FUNCT__ "PCApplyRichardson_MG"
647319c654SBarry Smith static PetscErrorCode PCApplyRichardson_MG(PC pc,Vec b,Vec x,Vec w,PetscReal rtol,PetscReal abstol, PetscReal dtol,PetscInt its,PetscTruth zeroguess,PetscInt *outits,PCRichardsonConvergedReason *reason)
654b9ad928SBarry Smith {
66f3fbd535SBarry Smith   PC_MG          *mg = (PC_MG*)pc->data;
67f3fbd535SBarry Smith   PC_MG_Levels   **mglevels = mg->levels;
68dfbe8321SBarry Smith   PetscErrorCode ierr;
69f3fbd535SBarry Smith   PetscInt       levels = mglevels[0]->levels,i;
704b9ad928SBarry Smith 
714b9ad928SBarry Smith   PetscFunctionBegin;
72f3fbd535SBarry Smith   mglevels[levels-1]->b    = b;
73f3fbd535SBarry Smith   mglevels[levels-1]->x    = x;
744b9ad928SBarry Smith 
7531567311SBarry Smith   mg->rtol = rtol;
7631567311SBarry Smith   mg->abstol = abstol;
7731567311SBarry Smith   mg->dtol = dtol;
784b9ad928SBarry Smith   if (rtol) {
794b9ad928SBarry Smith     /* compute initial residual norm for relative convergence test */
804b9ad928SBarry Smith     PetscReal rnorm;
817319c654SBarry Smith     if (zeroguess) {
827319c654SBarry Smith       ierr               = VecNorm(b,NORM_2,&rnorm);CHKERRQ(ierr);
837319c654SBarry Smith     } else {
84f3fbd535SBarry Smith       ierr               = (*mglevels[levels-1]->residual)(mglevels[levels-1]->A,b,x,w);CHKERRQ(ierr);
854b9ad928SBarry Smith       ierr               = VecNorm(w,NORM_2,&rnorm);CHKERRQ(ierr);
867319c654SBarry Smith     }
8731567311SBarry Smith     mg->ttol = PetscMax(rtol*rnorm,abstol);
8870441072SBarry Smith   } else if (abstol) {
8931567311SBarry Smith     mg->ttol = abstol;
904b9ad928SBarry Smith   } else {
9131567311SBarry Smith     mg->ttol = 0.0;
924b9ad928SBarry Smith   }
934b9ad928SBarry Smith 
944d0a8057SBarry Smith   /* since smoother is applied to full system, not just residual we need to make sure that smoothers don't
954d0a8057SBarry Smith      stop prematurely do to small residual */
964d0a8057SBarry Smith   for (i=1; i<levels; i++) {
97f3fbd535SBarry Smith     ierr = KSPSetTolerances(mglevels[i]->smoothu,0,PETSC_DEFAULT,PETSC_DEFAULT,PETSC_DEFAULT);CHKERRQ(ierr);
98f3fbd535SBarry Smith     if (mglevels[i]->smoothu != mglevels[i]->smoothd) {
99f3fbd535SBarry Smith       ierr = KSPSetTolerances(mglevels[i]->smoothd,0,PETSC_DEFAULT,PETSC_DEFAULT,PETSC_DEFAULT);CHKERRQ(ierr);
1004b9ad928SBarry Smith     }
1014d0a8057SBarry Smith   }
1024d0a8057SBarry Smith 
1034d0a8057SBarry Smith   *reason = (PCRichardsonConvergedReason)0;
1044d0a8057SBarry Smith   for (i=0; i<its; i++) {
105f3fbd535SBarry Smith     ierr = PCMGMCycle_Private(pc,mglevels+levels-1,reason);CHKERRQ(ierr);
1064d0a8057SBarry Smith     if (*reason) break;
1074d0a8057SBarry Smith   }
1084d0a8057SBarry Smith   if (!*reason) *reason = PCRICHARDSON_CONVERGED_ITS;
1094d0a8057SBarry Smith   *outits = i;
1104b9ad928SBarry Smith   PetscFunctionReturn(0);
1114b9ad928SBarry Smith }
1124b9ad928SBarry Smith 
1134b9ad928SBarry Smith #undef __FUNCT__
1149dcbbd2bSBarry Smith #define __FUNCT__ "PCMGSetLevels"
1154b9ad928SBarry Smith /*@C
11697177400SBarry Smith    PCMGSetLevels - Sets the number of levels to use with MG.
1174b9ad928SBarry Smith    Must be called before any other MG routine.
1184b9ad928SBarry Smith 
119*ad4df100SBarry Smith    Logically Collective on PC
1204b9ad928SBarry Smith 
1214b9ad928SBarry Smith    Input Parameters:
1224b9ad928SBarry Smith +  pc - the preconditioner context
1234b9ad928SBarry Smith .  levels - the number of levels
1244b9ad928SBarry Smith -  comms - optional communicators for each level; this is to allow solving the coarser problems
1254b9ad928SBarry Smith            on smaller sets of processors. Use PETSC_NULL_OBJECT for default in Fortran
1264b9ad928SBarry Smith 
1274b9ad928SBarry Smith    Level: intermediate
1284b9ad928SBarry Smith 
1294b9ad928SBarry Smith    Notes:
1304b9ad928SBarry Smith      If the number of levels is one then the multigrid uses the -mg_levels prefix
1314b9ad928SBarry Smith   for setting the level options rather than the -mg_coarse prefix.
1324b9ad928SBarry Smith 
1334b9ad928SBarry Smith .keywords: MG, set, levels, multigrid
1344b9ad928SBarry Smith 
13597177400SBarry Smith .seealso: PCMGSetType(), PCMGGetLevels()
1364b9ad928SBarry Smith @*/
13797177400SBarry Smith PetscErrorCode PETSCKSP_DLLEXPORT PCMGSetLevels(PC pc,PetscInt levels,MPI_Comm *comms)
1384b9ad928SBarry Smith {
139dfbe8321SBarry Smith   PetscErrorCode ierr;
140f3fbd535SBarry Smith   PC_MG          *mg = (PC_MG*)pc->data;
141f3fbd535SBarry Smith   MPI_Comm       comm = ((PetscObject)pc)->comm;
142f3fbd535SBarry Smith   PC_MG_Levels   **mglevels;
143f3fbd535SBarry Smith   PetscInt       i;
144f3fbd535SBarry Smith   PetscMPIInt    size;
145f3fbd535SBarry Smith   const char     *prefix;
146f3fbd535SBarry Smith   PC             ipc;
1474b9ad928SBarry Smith 
1484b9ad928SBarry Smith   PetscFunctionBegin;
1490700a824SBarry Smith   PetscValidHeaderSpecific(pc,PC_CLASSID,1);
150e7e72b3dSBarry Smith   if (mg->nlevels > -1) SETERRQ(((PetscObject)pc)->comm,PETSC_ERR_ORDER,"Number levels already set for MG\n  make sure that you call PCMGSetLevels() before KSPSetFromOptions()");
151e32f2f54SBarry Smith   if (mg->levels) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_PLIB,"Internal error in PETSc, this array should not yet exist");
152f3fbd535SBarry Smith 
153f3fbd535SBarry Smith   mg->nlevels      = levels;
154218a07d4SBarry Smith   mg->galerkin     = PETSC_FALSE;
155218a07d4SBarry Smith   mg->galerkinused = PETSC_FALSE;
156f3fbd535SBarry Smith 
157f3fbd535SBarry Smith   ierr = PetscMalloc(levels*sizeof(PC_MG*),&mglevels);CHKERRQ(ierr);
158f3fbd535SBarry Smith   ierr = PetscLogObjectMemory(pc,levels*(sizeof(PC_MG*)));CHKERRQ(ierr);
159f3fbd535SBarry Smith 
160f3fbd535SBarry Smith   ierr = PCGetOptionsPrefix(pc,&prefix);CHKERRQ(ierr);
161f3fbd535SBarry Smith 
162f3fbd535SBarry Smith   for (i=0; i<levels; i++) {
163f3fbd535SBarry Smith     ierr = PetscNewLog(pc,PC_MG_Levels,&mglevels[i]);CHKERRQ(ierr);
164f3fbd535SBarry Smith     mglevels[i]->level           = i;
165f3fbd535SBarry Smith     mglevels[i]->levels          = levels;
166f3fbd535SBarry Smith     mglevels[i]->cycles          = PC_MG_CYCLE_V;
16731567311SBarry Smith     mg->default_smoothu = 1;
16831567311SBarry Smith     mg->default_smoothd = 1;
169f3fbd535SBarry Smith 
170f3fbd535SBarry Smith     if (comms) comm = comms[i];
171f3fbd535SBarry Smith     ierr = KSPCreate(comm,&mglevels[i]->smoothd);CHKERRQ(ierr);
172f3fbd535SBarry Smith     ierr = PetscObjectIncrementTabLevel((PetscObject)mglevels[i]->smoothd,(PetscObject)pc,levels-i);CHKERRQ(ierr);
17331567311SBarry Smith     ierr = KSPSetTolerances(mglevels[i]->smoothd,PETSC_DEFAULT,PETSC_DEFAULT,PETSC_DEFAULT, mg->default_smoothd);CHKERRQ(ierr);
174f3fbd535SBarry Smith     ierr = KSPSetOptionsPrefix(mglevels[i]->smoothd,prefix);CHKERRQ(ierr);
175f3fbd535SBarry Smith 
176f3fbd535SBarry Smith     /* do special stuff for coarse grid */
177f3fbd535SBarry Smith     if (!i && levels > 1) {
178f3fbd535SBarry Smith       ierr = KSPAppendOptionsPrefix(mglevels[0]->smoothd,"mg_coarse_");CHKERRQ(ierr);
179f3fbd535SBarry Smith 
180f3fbd535SBarry Smith       /* coarse solve is (redundant) LU by default */
181f3fbd535SBarry Smith       ierr = KSPSetType(mglevels[0]->smoothd,KSPPREONLY);CHKERRQ(ierr);
182f3fbd535SBarry Smith       ierr = KSPGetPC(mglevels[0]->smoothd,&ipc);CHKERRQ(ierr);
183f3fbd535SBarry Smith       ierr = MPI_Comm_size(comm,&size);CHKERRQ(ierr);
184f3fbd535SBarry Smith       if (size > 1) {
185f3fbd535SBarry Smith         ierr = PCSetType(ipc,PCREDUNDANT);CHKERRQ(ierr);
186f3fbd535SBarry Smith       } else {
187f3fbd535SBarry Smith         ierr = PCSetType(ipc,PCLU);CHKERRQ(ierr);
188f3fbd535SBarry Smith       }
189f3fbd535SBarry Smith 
190f3fbd535SBarry Smith     } else {
191f3fbd535SBarry Smith       char tprefix[128];
192f3fbd535SBarry Smith       sprintf(tprefix,"mg_levels_%d_",(int)i);
193f3fbd535SBarry Smith       ierr = KSPAppendOptionsPrefix(mglevels[i]->smoothd,tprefix);CHKERRQ(ierr);
194f3fbd535SBarry Smith     }
195f3fbd535SBarry Smith     ierr = PetscLogObjectParent(pc,mglevels[i]->smoothd);CHKERRQ(ierr);
196f3fbd535SBarry Smith     mglevels[i]->smoothu    = mglevels[i]->smoothd;
19731567311SBarry Smith     mg->rtol                = 0.0;
19831567311SBarry Smith     mg->abstol              = 0.0;
19931567311SBarry Smith     mg->dtol                = 0.0;
20031567311SBarry Smith     mg->ttol                = 0.0;
20131567311SBarry Smith     mg->eventsmoothsetup    = 0;
20231567311SBarry Smith     mg->eventsmoothsolve    = 0;
20331567311SBarry Smith     mg->eventresidual       = 0;
20431567311SBarry Smith     mg->eventinterprestrict = 0;
20531567311SBarry Smith     mg->cyclesperpcapply    = 1;
206f3fbd535SBarry Smith   }
20731567311SBarry Smith   mg->am          = PC_MG_MULTIPLICATIVE;
208f3fbd535SBarry Smith   mg->levels      = mglevels;
2094b9ad928SBarry Smith   pc->ops->applyrichardson = PCApplyRichardson_MG;
2104b9ad928SBarry Smith   PetscFunctionReturn(0);
2114b9ad928SBarry Smith }
2124b9ad928SBarry Smith 
2134b9ad928SBarry Smith #undef __FUNCT__
214c07bf074SBarry Smith #define __FUNCT__ "PCDestroy_MG_Private"
215c07bf074SBarry Smith PetscErrorCode PCDestroy_MG_Private(PC pc)
216f3fbd535SBarry Smith {
217f3fbd535SBarry Smith   PC_MG          *mg = (PC_MG*)pc->data;
218f3fbd535SBarry Smith   PC_MG_Levels   **mglevels = mg->levels;
219f3fbd535SBarry Smith   PetscErrorCode ierr;
220f3fbd535SBarry Smith   PetscInt       i,n;
221f3fbd535SBarry Smith 
222f3fbd535SBarry Smith   PetscFunctionBegin;
223f3fbd535SBarry Smith   if (mglevels) {
224f3fbd535SBarry Smith     n = mglevels[0]->levels;
225f3fbd535SBarry Smith     for (i=0; i<n-1; i++) {
226f3fbd535SBarry Smith       if (mglevels[i+1]->r) {ierr = VecDestroy(mglevels[i+1]->r);CHKERRQ(ierr);}
227f3fbd535SBarry Smith       if (mglevels[i]->b) {ierr = VecDestroy(mglevels[i]->b);CHKERRQ(ierr);}
228f3fbd535SBarry Smith       if (mglevels[i]->x) {ierr = VecDestroy(mglevels[i]->x);CHKERRQ(ierr);}
229f3fbd535SBarry Smith       if (mglevels[i+1]->restrct) {ierr = MatDestroy(mglevels[i+1]->restrct);CHKERRQ(ierr);}
230f3fbd535SBarry Smith       if (mglevels[i+1]->interpolate) {ierr = MatDestroy(mglevels[i+1]->interpolate);CHKERRQ(ierr);}
231f3fbd535SBarry Smith     }
232f3fbd535SBarry Smith 
233f3fbd535SBarry Smith     for (i=0; i<n; i++) {
234f3fbd535SBarry Smith       if (mglevels[i]->smoothd != mglevels[i]->smoothu) {
235f3fbd535SBarry Smith 	ierr = KSPDestroy(mglevels[i]->smoothd);CHKERRQ(ierr);
236f3fbd535SBarry Smith       }
237f3fbd535SBarry Smith       ierr = KSPDestroy(mglevels[i]->smoothu);CHKERRQ(ierr);
238f3fbd535SBarry Smith       ierr = PetscFree(mglevels[i]);CHKERRQ(ierr);
239f3fbd535SBarry Smith     }
240f3fbd535SBarry Smith     ierr = PetscFree(mglevels);CHKERRQ(ierr);
241f3fbd535SBarry Smith   }
242c07bf074SBarry Smith   mg->nlevels = -1;
243c07bf074SBarry Smith   mg->levels  = PETSC_NULL;
244c07bf074SBarry Smith   PetscFunctionReturn(0);
245c07bf074SBarry Smith }
246c07bf074SBarry Smith 
247c07bf074SBarry Smith #undef __FUNCT__
248c07bf074SBarry Smith #define __FUNCT__ "PCDestroy_MG"
249c07bf074SBarry Smith PetscErrorCode PCDestroy_MG(PC pc)
250c07bf074SBarry Smith {
251c07bf074SBarry Smith   PC_MG          *mg = (PC_MG*)pc->data;
252c07bf074SBarry Smith   PetscErrorCode ierr;
253c07bf074SBarry Smith 
254c07bf074SBarry Smith   PetscFunctionBegin;
255c07bf074SBarry Smith   ierr = PCDestroy_MG_Private(pc);CHKERRQ(ierr);
256f3fbd535SBarry Smith   ierr = PetscFree(mg);CHKERRQ(ierr);
257f3fbd535SBarry Smith   PetscFunctionReturn(0);
258f3fbd535SBarry Smith }
259f3fbd535SBarry Smith 
260f3fbd535SBarry Smith 
261f3fbd535SBarry Smith 
26231567311SBarry Smith EXTERN PetscErrorCode PCMGACycle_Private(PC,PC_MG_Levels**);
263f3fbd535SBarry Smith EXTERN PetscErrorCode PCMGFCycle_Private(PC,PC_MG_Levels**);
26431567311SBarry Smith EXTERN PetscErrorCode PCMGKCycle_Private(PC,PC_MG_Levels**);
265f3fbd535SBarry Smith 
266f3fbd535SBarry Smith /*
267f3fbd535SBarry Smith    PCApply_MG - Runs either an additive, multiplicative, Kaskadic
268f3fbd535SBarry Smith              or full cycle of multigrid.
269f3fbd535SBarry Smith 
270f3fbd535SBarry Smith   Note:
271f3fbd535SBarry Smith   A simple wrapper which calls PCMGMCycle(),PCMGACycle(), or PCMGFCycle().
272f3fbd535SBarry Smith */
273f3fbd535SBarry Smith #undef __FUNCT__
274f3fbd535SBarry Smith #define __FUNCT__ "PCApply_MG"
275f3fbd535SBarry Smith static PetscErrorCode PCApply_MG(PC pc,Vec b,Vec x)
276f3fbd535SBarry Smith {
277f3fbd535SBarry Smith   PC_MG          *mg = (PC_MG*)pc->data;
278f3fbd535SBarry Smith   PC_MG_Levels   **mglevels = mg->levels;
279f3fbd535SBarry Smith   PetscErrorCode ierr;
280f3fbd535SBarry Smith   PetscInt       levels = mglevels[0]->levels,i;
281f3fbd535SBarry Smith 
282f3fbd535SBarry Smith   PetscFunctionBegin;
283e1d8e5deSBarry Smith 
284e1d8e5deSBarry Smith   /* When the DM is supplying the matrix then it will not exist until here */
285e1d8e5deSBarry Smith   for (i=0; i<levels-1; i++) {
286e1d8e5deSBarry Smith     if (!mglevels[i]->A) {
287e1d8e5deSBarry Smith       ierr = KSPGetOperators(mglevels[i]->smoothu,&mglevels[i]->A,PETSC_NULL,PETSC_NULL);CHKERRQ(ierr);
288e1d8e5deSBarry Smith     }
289e1d8e5deSBarry Smith   }
290e1d8e5deSBarry Smith 
291f3fbd535SBarry Smith   mglevels[levels-1]->b = b;
292f3fbd535SBarry Smith   mglevels[levels-1]->x = x;
29331567311SBarry Smith   if (mg->am == PC_MG_MULTIPLICATIVE) {
294f3fbd535SBarry Smith     ierr = VecSet(x,0.0);CHKERRQ(ierr);
29531567311SBarry Smith     for (i=0; i<mg->cyclesperpcapply; i++) {
296f3fbd535SBarry Smith       ierr = PCMGMCycle_Private(pc,mglevels+levels-1,PETSC_NULL);CHKERRQ(ierr);
297f3fbd535SBarry Smith     }
298f3fbd535SBarry Smith   }
29931567311SBarry Smith   else if (mg->am == PC_MG_ADDITIVE) {
30031567311SBarry Smith     ierr = PCMGACycle_Private(pc,mglevels);CHKERRQ(ierr);
301f3fbd535SBarry Smith   }
30231567311SBarry Smith   else if (mg->am == PC_MG_KASKADE) {
30331567311SBarry Smith     ierr = PCMGKCycle_Private(pc,mglevels);CHKERRQ(ierr);
304f3fbd535SBarry Smith   }
305f3fbd535SBarry Smith   else {
306f3fbd535SBarry Smith     ierr = PCMGFCycle_Private(pc,mglevels);CHKERRQ(ierr);
307f3fbd535SBarry Smith   }
308f3fbd535SBarry Smith   PetscFunctionReturn(0);
309f3fbd535SBarry Smith }
310f3fbd535SBarry Smith 
311f3fbd535SBarry Smith 
312f3fbd535SBarry Smith #undef __FUNCT__
313f3fbd535SBarry Smith #define __FUNCT__ "PCSetFromOptions_MG"
314f3fbd535SBarry Smith PetscErrorCode PCSetFromOptions_MG(PC pc)
315f3fbd535SBarry Smith {
316f3fbd535SBarry Smith   PetscErrorCode ierr;
317f3fbd535SBarry Smith   PetscInt       m,levels = 1,cycles;
318f3fbd535SBarry Smith   PetscTruth     flg;
319f3fbd535SBarry Smith   PC_MG          *mg = (PC_MG*)pc->data;
320f3fbd535SBarry Smith   PC_MG_Levels   **mglevels = mg->levels;
321f3fbd535SBarry Smith   PCMGType       mgtype;
322f3fbd535SBarry Smith   PCMGCycleType  mgctype;
323f3fbd535SBarry Smith 
324f3fbd535SBarry Smith   PetscFunctionBegin;
325f3fbd535SBarry Smith   ierr = PetscOptionsHead("Multigrid options");CHKERRQ(ierr);
32618aabeadSBarry Smith     if (!mglevels) {
327f3fbd535SBarry Smith       ierr = PetscOptionsInt("-pc_mg_levels","Number of Levels","PCMGSetLevels",levels,&levels,&flg);CHKERRQ(ierr);
328f3fbd535SBarry Smith       ierr = PCMGSetLevels(pc,levels,PETSC_NULL);CHKERRQ(ierr);
329f3fbd535SBarry Smith       mglevels = mg->levels;
330f3fbd535SBarry Smith     }
331f3fbd535SBarry Smith     mgctype = (PCMGCycleType) mglevels[0]->cycles;
332f3fbd535SBarry Smith     ierr = PetscOptionsEnum("-pc_mg_cycle_type","V cycle or for W-cycle","PCMGSetCycleType",PCMGCycleTypes,(PetscEnum)mgctype,(PetscEnum*)&mgctype,&flg);CHKERRQ(ierr);
333f3fbd535SBarry Smith     if (flg) {
334f3fbd535SBarry Smith       ierr = PCMGSetCycleType(pc,mgctype);CHKERRQ(ierr);
335f3fbd535SBarry Smith     };
336f3fbd535SBarry Smith     flg  = PETSC_FALSE;
337f3fbd535SBarry Smith     ierr = PetscOptionsTruth("-pc_mg_galerkin","Use Galerkin process to compute coarser operators","PCMGSetGalerkin",flg,&flg,PETSC_NULL);CHKERRQ(ierr);
338f3fbd535SBarry Smith     if (flg) {
339f3fbd535SBarry Smith       ierr = PCMGSetGalerkin(pc);CHKERRQ(ierr);
340f3fbd535SBarry Smith     }
341f3fbd535SBarry Smith     ierr = PetscOptionsInt("-pc_mg_smoothup","Number of post-smoothing steps","PCMGSetNumberSmoothUp",1,&m,&flg);CHKERRQ(ierr);
342f3fbd535SBarry Smith     if (flg) {
343f3fbd535SBarry Smith       ierr = PCMGSetNumberSmoothUp(pc,m);CHKERRQ(ierr);
344f3fbd535SBarry Smith     }
345f3fbd535SBarry Smith     ierr = PetscOptionsInt("-pc_mg_smoothdown","Number of pre-smoothing steps","PCMGSetNumberSmoothDown",1,&m,&flg);CHKERRQ(ierr);
346f3fbd535SBarry Smith     if (flg) {
347f3fbd535SBarry Smith       ierr = PCMGSetNumberSmoothDown(pc,m);CHKERRQ(ierr);
348f3fbd535SBarry Smith     }
34931567311SBarry Smith     mgtype = mg->am;
350f3fbd535SBarry Smith     ierr = PetscOptionsEnum("-pc_mg_type","Multigrid type","PCMGSetType",PCMGTypes,(PetscEnum)mgtype,(PetscEnum*)&mgtype,&flg);CHKERRQ(ierr);
351f3fbd535SBarry Smith     if (flg) {
352f3fbd535SBarry Smith       ierr = PCMGSetType(pc,mgtype);CHKERRQ(ierr);
353f3fbd535SBarry Smith     }
35431567311SBarry Smith     if (mg->am == PC_MG_MULTIPLICATIVE) {
35531567311SBarry Smith       ierr = PetscOptionsInt("-pc_mg_multiplicative_cycles","Number of cycles for each preconditioner step","PCMGSetLevels",mg->cyclesperpcapply,&cycles,&flg);CHKERRQ(ierr);
356f3fbd535SBarry Smith       if (flg) {
357f3fbd535SBarry Smith 	ierr = PCMGMultiplicativeSetCycles(pc,cycles);CHKERRQ(ierr);
358f3fbd535SBarry Smith       }
359f3fbd535SBarry Smith     }
360f3fbd535SBarry Smith     flg  = PETSC_FALSE;
361f3fbd535SBarry Smith     ierr = PetscOptionsTruth("-pc_mg_log","Log times for each multigrid level","None",flg,&flg,PETSC_NULL);CHKERRQ(ierr);
362f3fbd535SBarry Smith     if (flg) {
363f3fbd535SBarry Smith       PetscInt i;
364f3fbd535SBarry Smith       char     eventname[128];
365e7e72b3dSBarry Smith       if (!mg) SETERRQ(((PetscObject)pc)->comm,PETSC_ERR_ARG_WRONGSTATE,"Must set MG levels before calling");
366f3fbd535SBarry Smith       levels = mglevels[0]->levels;
367f3fbd535SBarry Smith       for (i=0; i<levels; i++) {
368f3fbd535SBarry Smith         sprintf(eventname,"MGSetup Level %d",(int)i);
3690700a824SBarry Smith         ierr = PetscLogEventRegister(eventname,((PetscObject)pc)->classid,&mg->eventsmoothsetup);CHKERRQ(ierr);
370f3fbd535SBarry Smith         sprintf(eventname,"MGSmooth Level %d",(int)i);
3710700a824SBarry Smith         ierr = PetscLogEventRegister(eventname,((PetscObject)pc)->classid,&mg->eventsmoothsolve);CHKERRQ(ierr);
372f3fbd535SBarry Smith         if (i) {
373f3fbd535SBarry Smith           sprintf(eventname,"MGResid Level %d",(int)i);
3740700a824SBarry Smith           ierr = PetscLogEventRegister(eventname,((PetscObject)pc)->classid,&mg->eventresidual);CHKERRQ(ierr);
375f3fbd535SBarry Smith           sprintf(eventname,"MGInterp Level %d",(int)i);
3760700a824SBarry Smith           ierr = PetscLogEventRegister(eventname,((PetscObject)pc)->classid,&mg->eventinterprestrict);CHKERRQ(ierr);
377f3fbd535SBarry Smith         }
378f3fbd535SBarry Smith       }
379f3fbd535SBarry Smith     }
380f3fbd535SBarry Smith   ierr = PetscOptionsTail();CHKERRQ(ierr);
381f3fbd535SBarry Smith   PetscFunctionReturn(0);
382f3fbd535SBarry Smith }
383f3fbd535SBarry Smith 
384f3fbd535SBarry Smith const char *PCMGTypes[] = {"MULTIPLICATIVE","ADDITIVE","FULL","KASKADE","PCMGType","PC_MG",0};
385f3fbd535SBarry Smith const char *PCMGCycleTypes[] = {"invalid","v","w","PCMGCycleType","PC_MG_CYCLE",0};
386f3fbd535SBarry Smith 
387f3fbd535SBarry Smith #undef __FUNCT__
388f3fbd535SBarry Smith #define __FUNCT__ "PCView_MG"
389f3fbd535SBarry Smith PetscErrorCode PCView_MG(PC pc,PetscViewer viewer)
390f3fbd535SBarry Smith {
391f3fbd535SBarry Smith   PC_MG          *mg = (PC_MG*)pc->data;
392f3fbd535SBarry Smith   PC_MG_Levels   **mglevels = mg->levels;
393f3fbd535SBarry Smith   PetscErrorCode ierr;
394f3fbd535SBarry Smith   PetscInt       levels = mglevels[0]->levels,i;
395f3fbd535SBarry Smith   PetscTruth     iascii;
396f3fbd535SBarry Smith 
397f3fbd535SBarry Smith   PetscFunctionBegin;
3982692d6eeSBarry Smith   ierr = PetscTypeCompare((PetscObject)viewer,PETSCVIEWERASCII,&iascii);CHKERRQ(ierr);
399f3fbd535SBarry Smith   if (iascii) {
40031567311SBarry Smith     ierr = PetscViewerASCIIPrintf(viewer,"  MG: type is %s, levels=%D cycles=%s\n", PCMGTypes[mg->am],levels,(mglevels[0]->cycles == PC_MG_CYCLE_V) ? "v" : "w");CHKERRQ(ierr);
40131567311SBarry Smith     if (mg->am == PC_MG_MULTIPLICATIVE) {
40231567311SBarry Smith       ierr = PetscViewerASCIIPrintf(viewer,"    Cycles per PCApply=%d\n",mg->cyclesperpcapply);CHKERRQ(ierr);
403f3fbd535SBarry Smith     }
404218a07d4SBarry Smith     if (mg->galerkin) {
405f3fbd535SBarry Smith       ierr = PetscViewerASCIIPrintf(viewer,"    Using Galerkin computed coarse grid matrices\n");CHKERRQ(ierr);
4064f66f45eSBarry Smith     } else {
4074f66f45eSBarry Smith       ierr = PetscViewerASCIIPrintf(viewer,"    Not using Galerkin computed coarse grid matrices\n");CHKERRQ(ierr);
408f3fbd535SBarry Smith     }
409f3fbd535SBarry Smith     for (i=0; i<levels; i++) {
410f3fbd535SBarry Smith       if (!i) {
41131567311SBarry Smith         ierr = PetscViewerASCIIPrintf(viewer,"Coarse grid solver -- level %D presmooths=%D postsmooths=%D -----\n",i,mg->default_smoothd,mg->default_smoothu);CHKERRQ(ierr);
412f3fbd535SBarry Smith       } else {
41331567311SBarry Smith         ierr = PetscViewerASCIIPrintf(viewer,"Down solver (pre-smoother) on level %D smooths=%D --------------------\n",i,mg->default_smoothd);CHKERRQ(ierr);
414f3fbd535SBarry Smith       }
415f3fbd535SBarry Smith       ierr = PetscViewerASCIIPushTab(viewer);CHKERRQ(ierr);
416f3fbd535SBarry Smith       ierr = KSPView(mglevels[i]->smoothd,viewer);CHKERRQ(ierr);
417f3fbd535SBarry Smith       ierr = PetscViewerASCIIPopTab(viewer);CHKERRQ(ierr);
418f3fbd535SBarry Smith       if (i && mglevels[i]->smoothd == mglevels[i]->smoothu) {
419f3fbd535SBarry Smith         ierr = PetscViewerASCIIPrintf(viewer,"Up solver (post-smoother) same as down solver (pre-smoother)\n");CHKERRQ(ierr);
420f3fbd535SBarry Smith       } else if (i){
42131567311SBarry Smith         ierr = PetscViewerASCIIPrintf(viewer,"Up solver (post-smoother) on level %D smooths=%D --------------------\n",i,mg->default_smoothu);CHKERRQ(ierr);
422f3fbd535SBarry Smith         ierr = PetscViewerASCIIPushTab(viewer);CHKERRQ(ierr);
423f3fbd535SBarry Smith         ierr = KSPView(mglevels[i]->smoothu,viewer);CHKERRQ(ierr);
424f3fbd535SBarry Smith         ierr = PetscViewerASCIIPopTab(viewer);CHKERRQ(ierr);
425f3fbd535SBarry Smith       }
426f3fbd535SBarry Smith     }
427f3fbd535SBarry Smith   } else {
42865e19b50SBarry Smith     SETERRQ1(((PetscObject)pc)->comm,PETSC_ERR_SUP,"Viewer type %s not supported for PCMG",((PetscObject)viewer)->type_name);
429f3fbd535SBarry Smith   }
430f3fbd535SBarry Smith   PetscFunctionReturn(0);
431f3fbd535SBarry Smith }
432f3fbd535SBarry Smith 
433f3fbd535SBarry Smith /*
434f3fbd535SBarry Smith     Calls setup for the KSP on each level
435f3fbd535SBarry Smith */
436f3fbd535SBarry Smith #undef __FUNCT__
437f3fbd535SBarry Smith #define __FUNCT__ "PCSetUp_MG"
438f3fbd535SBarry Smith PetscErrorCode PCSetUp_MG(PC pc)
439f3fbd535SBarry Smith {
440f3fbd535SBarry Smith   PC_MG                   *mg = (PC_MG*)pc->data;
441f3fbd535SBarry Smith   PC_MG_Levels            **mglevels = mg->levels;
442f3fbd535SBarry Smith   PetscErrorCode          ierr;
443f3fbd535SBarry Smith   PetscInt                i,n = mglevels[0]->levels;
444f3fbd535SBarry Smith   PC                      cpc,mpc;
445f3fbd535SBarry Smith   PetscTruth              preonly,lu,redundant,cholesky,monitor = PETSC_FALSE,dump = PETSC_FALSE,opsset;
446f3fbd535SBarry Smith   PetscViewerASCIIMonitor ascii;
447f3fbd535SBarry Smith   PetscViewer             viewer = PETSC_NULL;
448f3fbd535SBarry Smith   MPI_Comm                comm;
449f3fbd535SBarry Smith   Mat                     dA,dB;
450f3fbd535SBarry Smith   MatStructure            uflag;
451f3fbd535SBarry Smith   Vec                     tvec;
452218a07d4SBarry Smith   DM                      *dms;
453f3fbd535SBarry Smith 
454f3fbd535SBarry Smith   PetscFunctionBegin;
455f3fbd535SBarry Smith 
456f3fbd535SBarry Smith   /* If user did not provide fine grid operators OR operator was not updated since last global KSPSetOperators() */
457f3fbd535SBarry Smith   /* so use those from global PC */
458f3fbd535SBarry Smith   /* Is this what we always want? What if user wants to keep old one? */
459f3fbd535SBarry Smith   ierr = KSPGetOperatorsSet(mglevels[n-1]->smoothd,PETSC_NULL,&opsset);CHKERRQ(ierr);
460f3fbd535SBarry Smith   ierr = KSPGetPC(mglevels[0]->smoothd,&cpc);CHKERRQ(ierr);
461f3fbd535SBarry Smith   ierr = KSPGetPC(mglevels[n-1]->smoothd,&mpc);CHKERRQ(ierr);
4621338a6b9SJed Brown   if (!opsset || ((cpc->setupcalled == 1) && (mpc->setupcalled == 2)) || ((mpc == cpc) && (mpc->setupcalled == 2))) {
463f3fbd535SBarry Smith     ierr = PetscInfo(pc,"Using outer operators to define finest grid operator \n  because PCMGGetSmoother(pc,nlevels-1,&ksp);KSPSetOperators(ksp,...); was not called.\n");CHKERRQ(ierr);
464f3fbd535SBarry Smith     ierr = KSPSetOperators(mglevels[n-1]->smoothd,pc->mat,pc->pmat,pc->flag);CHKERRQ(ierr);
465f3fbd535SBarry Smith   }
466f3fbd535SBarry Smith 
4672d2b81a6SBarry Smith   if (pc->dm && !pc->setupcalled) {
4682d2b81a6SBarry Smith     /* construct the interpolation from the DMs */
4692e499ae9SBarry Smith     Mat p;
470218a07d4SBarry Smith     ierr = PetscMalloc(n*sizeof(DM),&dms);CHKERRQ(ierr);
471218a07d4SBarry Smith     dms[n-1] = pc->dm;
472218a07d4SBarry Smith     for (i=n-2; i>-1; i--) {
473218a07d4SBarry Smith       ierr = DMCoarsen(dms[i+1],PETSC_NULL,&dms[i]);CHKERRQ(ierr);
47411629dbeSBarry Smith       ierr = DMSetFunction(dms[i],0);
47511629dbeSBarry Smith       ierr = DMSetInitialGuess(dms[i],0);
47624c3aa18SBarry Smith       if (!mglevels[i+1]->interpolate) {
4772d2b81a6SBarry Smith 	ierr = DMGetInterpolation(dms[i],dms[i+1],&p,PETSC_NULL);CHKERRQ(ierr);
4782d2b81a6SBarry Smith 	ierr = PCMGSetInterpolation(pc,i+1,p);CHKERRQ(ierr);
4792d2b81a6SBarry Smith         ierr = MatDestroy(p);CHKERRQ(ierr);
480218a07d4SBarry Smith       }
48124c3aa18SBarry Smith     }
4822d2b81a6SBarry Smith 
4832d2b81a6SBarry Smith     if (!mg->galerkin) {
4842e499ae9SBarry Smith       /* each coarse level gets its DM */
4852d2b81a6SBarry Smith       for (i=n-2; i>-1; i--) {
4862e499ae9SBarry Smith         ierr = KSPSetDM(mglevels[i]->smoothd,dms[i]);CHKERRQ(ierr);
4872d2b81a6SBarry Smith       }
4882d2b81a6SBarry Smith     }
4892d2b81a6SBarry Smith 
490218a07d4SBarry Smith     for (i=n-2; i>-1; i--) {
491218a07d4SBarry Smith       ierr = DMDestroy(dms[i]);CHKERRQ(ierr);
492218a07d4SBarry Smith     }
4932d2b81a6SBarry Smith     ierr = PetscFree(dms);CHKERRQ(ierr);
494218a07d4SBarry Smith   }
495218a07d4SBarry Smith 
496218a07d4SBarry Smith   if (mg->galerkin) {
497f3fbd535SBarry Smith     Mat B;
498218a07d4SBarry Smith     mg->galerkinused = PETSC_TRUE;
499f3fbd535SBarry Smith     /* currently only handle case where mat and pmat are the same on coarser levels */
500f3fbd535SBarry Smith     ierr = KSPGetOperators(mglevels[n-1]->smoothd,&dA,&dB,&uflag);CHKERRQ(ierr);
501f3fbd535SBarry Smith     if (!pc->setupcalled) {
502f3fbd535SBarry Smith       for (i=n-2; i>-1; i--) {
503f3fbd535SBarry Smith         ierr = MatPtAP(dB,mglevels[i+1]->interpolate,MAT_INITIAL_MATRIX,1.0,&B);CHKERRQ(ierr);
504f3fbd535SBarry Smith         ierr = KSPSetOperators(mglevels[i]->smoothd,B,B,uflag);CHKERRQ(ierr);
505f3fbd535SBarry Smith 	if (i != n-2) {ierr = PetscObjectDereference((PetscObject)dB);CHKERRQ(ierr);}
506f3fbd535SBarry Smith         dB   = B;
507f3fbd535SBarry Smith       }
508cd9507b2SBarry Smith       if (n > 1) {ierr = PetscObjectDereference((PetscObject)dB);CHKERRQ(ierr);}
509f3fbd535SBarry Smith     } else {
510f3fbd535SBarry Smith       for (i=n-2; i>-1; i--) {
511f3fbd535SBarry Smith         ierr = KSPGetOperators(mglevels[i]->smoothd,PETSC_NULL,&B,PETSC_NULL);CHKERRQ(ierr);
512f3fbd535SBarry Smith         ierr = MatPtAP(dB,mglevels[i+1]->interpolate,MAT_REUSE_MATRIX,1.0,&B);CHKERRQ(ierr);
513f3fbd535SBarry Smith         ierr = KSPSetOperators(mglevels[i]->smoothd,B,B,uflag);CHKERRQ(ierr);
514f3fbd535SBarry Smith         dB   = B;
515f3fbd535SBarry Smith       }
516f3fbd535SBarry Smith     }
517f3fbd535SBarry Smith   }
518f3fbd535SBarry Smith 
519f3fbd535SBarry Smith   if (!pc->setupcalled) {
520f3fbd535SBarry Smith     ierr = PetscOptionsGetTruth(0,"-pc_mg_monitor",&monitor,PETSC_NULL);CHKERRQ(ierr);
521f3fbd535SBarry Smith 
522f3fbd535SBarry Smith     for (i=0; i<n; i++) {
523f3fbd535SBarry Smith       if (monitor) {
524f3fbd535SBarry Smith         ierr = PetscObjectGetComm((PetscObject)mglevels[i]->smoothd,&comm);CHKERRQ(ierr);
525f3fbd535SBarry Smith         ierr = PetscViewerASCIIMonitorCreate(comm,"stdout",n-i,&ascii);CHKERRQ(ierr);
526f3fbd535SBarry Smith         ierr = KSPMonitorSet(mglevels[i]->smoothd,KSPMonitorDefault,ascii,(PetscErrorCode(*)(void*))PetscViewerASCIIMonitorDestroy);CHKERRQ(ierr);
527f3fbd535SBarry Smith       }
528f3fbd535SBarry Smith       ierr = KSPSetFromOptions(mglevels[i]->smoothd);CHKERRQ(ierr);
529f3fbd535SBarry Smith     }
530f3fbd535SBarry Smith     for (i=1; i<n; i++) {
531f3fbd535SBarry Smith       if (mglevels[i]->smoothu && (mglevels[i]->smoothu != mglevels[i]->smoothd)) {
532f3fbd535SBarry Smith         if (monitor) {
533f3fbd535SBarry Smith           ierr = PetscObjectGetComm((PetscObject)mglevels[i]->smoothu,&comm);CHKERRQ(ierr);
534f3fbd535SBarry Smith           ierr = PetscViewerASCIIMonitorCreate(comm,"stdout",n-i,&ascii);CHKERRQ(ierr);
535f3fbd535SBarry Smith           ierr = KSPMonitorSet(mglevels[i]->smoothu,KSPMonitorDefault,ascii,(PetscErrorCode(*)(void*))PetscViewerASCIIMonitorDestroy);CHKERRQ(ierr);
536f3fbd535SBarry Smith         }
537f3fbd535SBarry Smith         ierr = KSPSetFromOptions(mglevels[i]->smoothu);CHKERRQ(ierr);
538f3fbd535SBarry Smith       }
539f3fbd535SBarry Smith     }
540f3fbd535SBarry Smith     for (i=1; i<n; i++) {
541f3fbd535SBarry Smith       if (!mglevels[i]->residual) {
542f3fbd535SBarry Smith         Mat mat;
543f3fbd535SBarry Smith         ierr = KSPGetOperators(mglevels[i]->smoothd,PETSC_NULL,&mat,PETSC_NULL);CHKERRQ(ierr);
544f3fbd535SBarry Smith         ierr = PCMGSetResidual(pc,i,PCMGDefaultResidual,mat);CHKERRQ(ierr);
545f3fbd535SBarry Smith       }
546f3fbd535SBarry Smith       if (mglevels[i]->restrct && !mglevels[i]->interpolate) {
547f3fbd535SBarry Smith         ierr = PCMGSetInterpolation(pc,i,mglevels[i]->restrct);CHKERRQ(ierr);
548f3fbd535SBarry Smith       }
549f3fbd535SBarry Smith       if (!mglevels[i]->restrct && mglevels[i]->interpolate) {
550f3fbd535SBarry Smith         ierr = PCMGSetRestriction(pc,i,mglevels[i]->interpolate);CHKERRQ(ierr);
551f3fbd535SBarry Smith       }
552f3fbd535SBarry Smith #if defined(PETSC_USE_DEBUG)
553f3fbd535SBarry Smith       if (!mglevels[i]->restrct || !mglevels[i]->interpolate) {
55465e19b50SBarry Smith         SETERRQ1(((PetscObject)pc)->comm,PETSC_ERR_ARG_WRONGSTATE,"Need to set restriction or interpolation on level %d",(int)i);
555f3fbd535SBarry Smith       }
556f3fbd535SBarry Smith #endif
557f3fbd535SBarry Smith     }
558f3fbd535SBarry Smith     for (i=0; i<n-1; i++) {
559f3fbd535SBarry Smith       if (!mglevels[i]->b) {
560f3fbd535SBarry Smith         Vec *vec;
561f3fbd535SBarry Smith         ierr = KSPGetVecs(mglevels[i]->smoothd,1,&vec,0,PETSC_NULL);CHKERRQ(ierr);
562f3fbd535SBarry Smith         ierr = PCMGSetRhs(pc,i,*vec);CHKERRQ(ierr);
563f3fbd535SBarry Smith         ierr = VecDestroy(*vec);CHKERRQ(ierr);
564f3fbd535SBarry Smith         ierr = PetscFree(vec);CHKERRQ(ierr);
565f3fbd535SBarry Smith       }
566f3fbd535SBarry Smith       if (!mglevels[i]->r && i) {
567f3fbd535SBarry Smith         ierr = VecDuplicate(mglevels[i]->b,&tvec);CHKERRQ(ierr);
568f3fbd535SBarry Smith         ierr = PCMGSetR(pc,i,tvec);CHKERRQ(ierr);
569f3fbd535SBarry Smith         ierr = VecDestroy(tvec);CHKERRQ(ierr);
570f3fbd535SBarry Smith       }
571f3fbd535SBarry Smith       if (!mglevels[i]->x) {
572f3fbd535SBarry Smith         ierr = VecDuplicate(mglevels[i]->b,&tvec);CHKERRQ(ierr);
573f3fbd535SBarry Smith         ierr = PCMGSetX(pc,i,tvec);CHKERRQ(ierr);
574f3fbd535SBarry Smith         ierr = VecDestroy(tvec);CHKERRQ(ierr);
575f3fbd535SBarry Smith       }
576f3fbd535SBarry Smith     }
577f3fbd535SBarry Smith     if (n != 1 && !mglevels[n-1]->r) {
578f3fbd535SBarry Smith       /* PCMGSetR() on the finest level if user did not supply it */
579f3fbd535SBarry Smith       Vec *vec;
580f3fbd535SBarry Smith       ierr = KSPGetVecs(mglevels[n-1]->smoothd,1,&vec,0,PETSC_NULL);CHKERRQ(ierr);
581f3fbd535SBarry Smith       ierr = PCMGSetR(pc,n-1,*vec);CHKERRQ(ierr);
582f3fbd535SBarry Smith       ierr = VecDestroy(*vec);CHKERRQ(ierr);
583f3fbd535SBarry Smith       ierr = PetscFree(vec);CHKERRQ(ierr);
584f3fbd535SBarry Smith     }
585f3fbd535SBarry Smith   }
586f3fbd535SBarry Smith 
587f3fbd535SBarry Smith 
588f3fbd535SBarry Smith   for (i=1; i<n; i++) {
589f3fbd535SBarry Smith     if (mglevels[i]->smoothu == mglevels[i]->smoothd) {
590f3fbd535SBarry Smith       /* if doing only down then initial guess is zero */
591f3fbd535SBarry Smith       ierr = KSPSetInitialGuessNonzero(mglevels[i]->smoothd,PETSC_TRUE);CHKERRQ(ierr);
592f3fbd535SBarry Smith     }
59331567311SBarry Smith     if (mg->eventsmoothsetup) {ierr = PetscLogEventBegin(mg->eventsmoothsetup,0,0,0,0);CHKERRQ(ierr);}
594f3fbd535SBarry Smith     ierr = KSPSetUp(mglevels[i]->smoothd);CHKERRQ(ierr);
59531567311SBarry Smith     if (mg->eventsmoothsetup) {ierr = PetscLogEventEnd(mg->eventsmoothsetup,0,0,0,0);CHKERRQ(ierr);}
596f3fbd535SBarry Smith   }
597f3fbd535SBarry Smith   for (i=1; i<n; i++) {
598f3fbd535SBarry Smith     if (mglevels[i]->smoothu && mglevels[i]->smoothu != mglevels[i]->smoothd) {
599f3fbd535SBarry Smith       Mat          downmat,downpmat;
600f3fbd535SBarry Smith       MatStructure matflag;
601f3fbd535SBarry Smith       PetscTruth   opsset;
602f3fbd535SBarry Smith 
603f3fbd535SBarry Smith       /* check if operators have been set for up, if not use down operators to set them */
604f3fbd535SBarry Smith       ierr = KSPGetOperatorsSet(mglevels[i]->smoothu,&opsset,PETSC_NULL);CHKERRQ(ierr);
605f3fbd535SBarry Smith       if (!opsset) {
606f3fbd535SBarry Smith         ierr = KSPGetOperators(mglevels[i]->smoothd,&downmat,&downpmat,&matflag);CHKERRQ(ierr);
607f3fbd535SBarry Smith         ierr = KSPSetOperators(mglevels[i]->smoothu,downmat,downpmat,matflag);CHKERRQ(ierr);
608f3fbd535SBarry Smith       }
609f3fbd535SBarry Smith 
610f3fbd535SBarry Smith       ierr = KSPSetInitialGuessNonzero(mglevels[i]->smoothu,PETSC_TRUE);CHKERRQ(ierr);
61131567311SBarry Smith       if (mg->eventsmoothsetup) {ierr = PetscLogEventBegin(mg->eventsmoothsetup,0,0,0,0);CHKERRQ(ierr);}
612f3fbd535SBarry Smith       ierr = KSPSetUp(mglevels[i]->smoothu);CHKERRQ(ierr);
61331567311SBarry Smith       if (mg->eventsmoothsetup) {ierr = PetscLogEventEnd(mg->eventsmoothsetup,0,0,0,0);CHKERRQ(ierr);}
614f3fbd535SBarry Smith     }
615f3fbd535SBarry Smith   }
616f3fbd535SBarry Smith 
617f3fbd535SBarry Smith   /*
618f3fbd535SBarry Smith       If coarse solver is not direct method then DO NOT USE preonly
619f3fbd535SBarry Smith   */
620f3fbd535SBarry Smith   ierr = PetscTypeCompare((PetscObject)mglevels[0]->smoothd,KSPPREONLY,&preonly);CHKERRQ(ierr);
621f3fbd535SBarry Smith   if (preonly) {
622f3fbd535SBarry Smith     ierr = PetscTypeCompare((PetscObject)cpc,PCLU,&lu);CHKERRQ(ierr);
623f3fbd535SBarry Smith     ierr = PetscTypeCompare((PetscObject)cpc,PCREDUNDANT,&redundant);CHKERRQ(ierr);
624f3fbd535SBarry Smith     ierr = PetscTypeCompare((PetscObject)cpc,PCCHOLESKY,&cholesky);CHKERRQ(ierr);
625f3fbd535SBarry Smith     if (!lu && !redundant && !cholesky) {
626f3fbd535SBarry Smith       ierr = KSPSetType(mglevels[0]->smoothd,KSPGMRES);CHKERRQ(ierr);
627f3fbd535SBarry Smith     }
628f3fbd535SBarry Smith   }
629f3fbd535SBarry Smith 
630f3fbd535SBarry Smith   if (!pc->setupcalled) {
631f3fbd535SBarry Smith     if (monitor) {
632f3fbd535SBarry Smith       ierr = PetscObjectGetComm((PetscObject)mglevels[0]->smoothd,&comm);CHKERRQ(ierr);
633f3fbd535SBarry Smith       ierr = PetscViewerASCIIMonitorCreate(comm,"stdout",n,&ascii);CHKERRQ(ierr);
634f3fbd535SBarry Smith       ierr = KSPMonitorSet(mglevels[0]->smoothd,KSPMonitorDefault,ascii,(PetscErrorCode(*)(void*))PetscViewerASCIIMonitorDestroy);CHKERRQ(ierr);
635f3fbd535SBarry Smith     }
636f3fbd535SBarry Smith     ierr = KSPSetFromOptions(mglevels[0]->smoothd);CHKERRQ(ierr);
637f3fbd535SBarry Smith   }
638f3fbd535SBarry Smith 
63931567311SBarry Smith   if (mg->eventsmoothsetup) {ierr = PetscLogEventBegin(mg->eventsmoothsetup,0,0,0,0);CHKERRQ(ierr);}
640f3fbd535SBarry Smith   ierr = KSPSetUp(mglevels[0]->smoothd);CHKERRQ(ierr);
64131567311SBarry Smith   if (mg->eventsmoothsetup) {ierr = PetscLogEventEnd(mg->eventsmoothsetup,0,0,0,0);CHKERRQ(ierr);}
642f3fbd535SBarry Smith 
643f3fbd535SBarry Smith   /*
644f3fbd535SBarry Smith      Dump the interpolation/restriction matrices plus the
645f3fbd535SBarry Smith    Jacobian/stiffness on each level. This allows Matlab users to
646f3fbd535SBarry Smith    easily check if the Galerkin condition A_c = R A_f R^T is satisfied.
647f3fbd535SBarry Smith 
648f3fbd535SBarry Smith    Only support one or the other at the same time.
649f3fbd535SBarry Smith   */
650f3fbd535SBarry Smith #if defined(PETSC_USE_SOCKET_VIEWER)
651f3fbd535SBarry Smith   ierr = PetscOptionsGetTruth(((PetscObject)pc)->prefix,"-pc_mg_dump_matlab",&dump,PETSC_NULL);CHKERRQ(ierr);
652f3fbd535SBarry Smith   if (dump) {
653f3fbd535SBarry Smith     viewer = PETSC_VIEWER_SOCKET_(((PetscObject)pc)->comm);
654f3fbd535SBarry Smith   }
655f3fbd535SBarry Smith   dump = PETSC_FALSE;
656f3fbd535SBarry Smith #endif
657f3fbd535SBarry Smith   ierr = PetscOptionsGetTruth(((PetscObject)pc)->prefix,"-pc_mg_dump_binary",&dump,PETSC_NULL);CHKERRQ(ierr);
658f3fbd535SBarry Smith   if (dump) {
659f3fbd535SBarry Smith     viewer = PETSC_VIEWER_BINARY_(((PetscObject)pc)->comm);
660f3fbd535SBarry Smith   }
661f3fbd535SBarry Smith 
662f3fbd535SBarry Smith   if (viewer) {
663f3fbd535SBarry Smith     for (i=1; i<n; i++) {
664f3fbd535SBarry Smith       ierr = MatView(mglevels[i]->restrct,viewer);CHKERRQ(ierr);
665f3fbd535SBarry Smith     }
666f3fbd535SBarry Smith     for (i=0; i<n; i++) {
667f3fbd535SBarry Smith       ierr = KSPGetPC(mglevels[i]->smoothd,&pc);CHKERRQ(ierr);
668f3fbd535SBarry Smith       ierr = MatView(pc->mat,viewer);CHKERRQ(ierr);
669f3fbd535SBarry Smith     }
670f3fbd535SBarry Smith   }
671f3fbd535SBarry Smith   PetscFunctionReturn(0);
672f3fbd535SBarry Smith }
673f3fbd535SBarry Smith 
674f3fbd535SBarry Smith /* -------------------------------------------------------------------------------------*/
675f3fbd535SBarry Smith 
676f3fbd535SBarry Smith #undef __FUNCT__
6779dcbbd2bSBarry Smith #define __FUNCT__ "PCMGGetLevels"
6784b9ad928SBarry Smith /*@
67997177400SBarry Smith    PCMGGetLevels - Gets the number of levels to use with MG.
6804b9ad928SBarry Smith 
6814b9ad928SBarry Smith    Not Collective
6824b9ad928SBarry Smith 
6834b9ad928SBarry Smith    Input Parameter:
6844b9ad928SBarry Smith .  pc - the preconditioner context
6854b9ad928SBarry Smith 
6864b9ad928SBarry Smith    Output parameter:
6874b9ad928SBarry Smith .  levels - the number of levels
6884b9ad928SBarry Smith 
6894b9ad928SBarry Smith    Level: advanced
6904b9ad928SBarry Smith 
6914b9ad928SBarry Smith .keywords: MG, get, levels, multigrid
6924b9ad928SBarry Smith 
69397177400SBarry Smith .seealso: PCMGSetLevels()
6944b9ad928SBarry Smith @*/
69597177400SBarry Smith PetscErrorCode PETSCKSP_DLLEXPORT PCMGGetLevels(PC pc,PetscInt *levels)
6964b9ad928SBarry Smith {
697f3fbd535SBarry Smith   PC_MG *mg = (PC_MG*)pc->data;
6984b9ad928SBarry Smith 
6994b9ad928SBarry Smith   PetscFunctionBegin;
7000700a824SBarry Smith   PetscValidHeaderSpecific(pc,PC_CLASSID,1);
7014482741eSBarry Smith   PetscValidIntPointer(levels,2);
702f3fbd535SBarry Smith   *levels = mg->nlevels;
7034b9ad928SBarry Smith   PetscFunctionReturn(0);
7044b9ad928SBarry Smith }
7054b9ad928SBarry Smith 
7064b9ad928SBarry Smith #undef __FUNCT__
7079dcbbd2bSBarry Smith #define __FUNCT__ "PCMGSetType"
7084b9ad928SBarry Smith /*@
70997177400SBarry Smith    PCMGSetType - Determines the form of multigrid to use:
7104b9ad928SBarry Smith    multiplicative, additive, full, or the Kaskade algorithm.
7114b9ad928SBarry Smith 
712*ad4df100SBarry Smith    Logically Collective on PC
7134b9ad928SBarry Smith 
7144b9ad928SBarry Smith    Input Parameters:
7154b9ad928SBarry Smith +  pc - the preconditioner context
7169dcbbd2bSBarry Smith -  form - multigrid form, one of PC_MG_MULTIPLICATIVE, PC_MG_ADDITIVE,
7179dcbbd2bSBarry Smith    PC_MG_FULL, PC_MG_KASKADE
7184b9ad928SBarry Smith 
7194b9ad928SBarry Smith    Options Database Key:
7204b9ad928SBarry Smith .  -pc_mg_type <form> - Sets <form>, one of multiplicative,
7214b9ad928SBarry Smith    additive, full, kaskade
7224b9ad928SBarry Smith 
7234b9ad928SBarry Smith    Level: advanced
7244b9ad928SBarry Smith 
7254b9ad928SBarry Smith .keywords: MG, set, method, multiplicative, additive, full, Kaskade, multigrid
7264b9ad928SBarry Smith 
72797177400SBarry Smith .seealso: PCMGSetLevels()
7284b9ad928SBarry Smith @*/
7299dcbbd2bSBarry Smith PetscErrorCode PETSCKSP_DLLEXPORT PCMGSetType(PC pc,PCMGType form)
7304b9ad928SBarry Smith {
731f3fbd535SBarry Smith   PC_MG                   *mg = (PC_MG*)pc->data;
7324b9ad928SBarry Smith 
7334b9ad928SBarry Smith   PetscFunctionBegin;
7340700a824SBarry Smith   PetscValidHeaderSpecific(pc,PC_CLASSID,1);
73531567311SBarry Smith   mg->am = form;
7369dcbbd2bSBarry Smith   if (form == PC_MG_MULTIPLICATIVE) pc->ops->applyrichardson = PCApplyRichardson_MG;
7374b9ad928SBarry Smith   else pc->ops->applyrichardson = 0;
7384b9ad928SBarry Smith   PetscFunctionReturn(0);
7394b9ad928SBarry Smith }
7404b9ad928SBarry Smith 
7414b9ad928SBarry Smith #undef __FUNCT__
7420d353602SBarry Smith #define __FUNCT__ "PCMGSetCycleType"
7434b9ad928SBarry Smith /*@
7440d353602SBarry Smith    PCMGSetCycleType - Sets the type cycles to use.  Use PCMGSetCycleTypeOnLevel() for more
7454b9ad928SBarry Smith    complicated cycling.
7464b9ad928SBarry Smith 
747*ad4df100SBarry Smith    Logically Collective on PC
7484b9ad928SBarry Smith 
7494b9ad928SBarry Smith    Input Parameters:
750c2be2410SBarry Smith +  pc - the multigrid context
7510d353602SBarry Smith -  PC_MG_CYCLE_V or PC_MG_CYCLE_W
7524b9ad928SBarry Smith 
7534b9ad928SBarry Smith    Options Database Key:
7540d353602SBarry Smith $  -pc_mg_cycle_type v or w
7554b9ad928SBarry Smith 
7564b9ad928SBarry Smith    Level: advanced
7574b9ad928SBarry Smith 
7584b9ad928SBarry Smith .keywords: MG, set, cycles, V-cycle, W-cycle, multigrid
7594b9ad928SBarry Smith 
7600d353602SBarry Smith .seealso: PCMGSetCycleTypeOnLevel()
7614b9ad928SBarry Smith @*/
7620d353602SBarry Smith PetscErrorCode PETSCKSP_DLLEXPORT PCMGSetCycleType(PC pc,PCMGCycleType n)
7634b9ad928SBarry Smith {
764f3fbd535SBarry Smith   PC_MG        *mg = (PC_MG*)pc->data;
765f3fbd535SBarry Smith   PC_MG_Levels **mglevels = mg->levels;
76679416396SBarry Smith   PetscInt     i,levels;
7674b9ad928SBarry Smith 
7684b9ad928SBarry Smith   PetscFunctionBegin;
7690700a824SBarry Smith   PetscValidHeaderSpecific(pc,PC_CLASSID,1);
770e7e72b3dSBarry Smith   if (!mglevels) SETERRQ(((PetscObject)pc)->comm,PETSC_ERR_ARG_WRONGSTATE,"Must set MG levels before calling");
771f3fbd535SBarry Smith   levels = mglevels[0]->levels;
7724b9ad928SBarry Smith 
7734b9ad928SBarry Smith   for (i=0; i<levels; i++) {
774f3fbd535SBarry Smith     mglevels[i]->cycles  = n;
7754b9ad928SBarry Smith   }
7764b9ad928SBarry Smith   PetscFunctionReturn(0);
7774b9ad928SBarry Smith }
7784b9ad928SBarry Smith 
7794b9ad928SBarry Smith #undef __FUNCT__
7808cc2d5dfSBarry Smith #define __FUNCT__ "PCMGMultiplicativeSetCycles"
7818cc2d5dfSBarry Smith /*@
7828cc2d5dfSBarry Smith    PCMGMultiplicativeSetCycles - Sets the number of cycles to use for each preconditioner step
7838cc2d5dfSBarry Smith          of multigrid when PCMGType of PC_MG_MULTIPLICATIVE is used
7848cc2d5dfSBarry Smith 
785*ad4df100SBarry Smith    Logically Collective on PC
7868cc2d5dfSBarry Smith 
7878cc2d5dfSBarry Smith    Input Parameters:
7888cc2d5dfSBarry Smith +  pc - the multigrid context
7898cc2d5dfSBarry Smith -  n - number of cycles (default is 1)
7908cc2d5dfSBarry Smith 
7918cc2d5dfSBarry Smith    Options Database Key:
7928cc2d5dfSBarry Smith $  -pc_mg_multiplicative_cycles n
7938cc2d5dfSBarry Smith 
7948cc2d5dfSBarry Smith    Level: advanced
7958cc2d5dfSBarry Smith 
7968cc2d5dfSBarry Smith    Notes: This is not associated with setting a v or w cycle, that is set with PCMGSetCycleType()
7978cc2d5dfSBarry Smith 
7988cc2d5dfSBarry Smith .keywords: MG, set, cycles, V-cycle, W-cycle, multigrid
7998cc2d5dfSBarry Smith 
8008cc2d5dfSBarry Smith .seealso: PCMGSetCycleTypeOnLevel(), PCMGSetCycleType()
8018cc2d5dfSBarry Smith @*/
8028cc2d5dfSBarry Smith PetscErrorCode PETSCKSP_DLLEXPORT PCMGMultiplicativeSetCycles(PC pc,PetscInt n)
8038cc2d5dfSBarry Smith {
804f3fbd535SBarry Smith   PC_MG        *mg = (PC_MG*)pc->data;
805f3fbd535SBarry Smith   PC_MG_Levels **mglevels = mg->levels;
8068cc2d5dfSBarry Smith   PetscInt     i,levels;
8078cc2d5dfSBarry Smith 
8088cc2d5dfSBarry Smith   PetscFunctionBegin;
8090700a824SBarry Smith   PetscValidHeaderSpecific(pc,PC_CLASSID,1);
810e7e72b3dSBarry Smith   if (!mglevels) SETERRQ(((PetscObject)pc)->comm,PETSC_ERR_ARG_WRONGSTATE,"Must set MG levels before calling");
811f3fbd535SBarry Smith   levels = mglevels[0]->levels;
8128cc2d5dfSBarry Smith 
8138cc2d5dfSBarry Smith   for (i=0; i<levels; i++) {
81431567311SBarry Smith     mg->cyclesperpcapply  = n;
8158cc2d5dfSBarry Smith   }
8168cc2d5dfSBarry Smith   PetscFunctionReturn(0);
8178cc2d5dfSBarry Smith }
8188cc2d5dfSBarry Smith 
8198cc2d5dfSBarry Smith #undef __FUNCT__
8209dcbbd2bSBarry Smith #define __FUNCT__ "PCMGSetGalerkin"
821c2be2410SBarry Smith /*@
82297177400SBarry Smith    PCMGSetGalerkin - Causes the coarser grid matrices to be computed from the
823c2be2410SBarry Smith       finest grid via the Galerkin process: A_i-1 = r_i * A_i * r_i^t
824c2be2410SBarry Smith 
825*ad4df100SBarry Smith    Logically Collective on PC
826c2be2410SBarry Smith 
827c2be2410SBarry Smith    Input Parameters:
8283fc8bf9cSBarry Smith .  pc - the multigrid context
829c2be2410SBarry Smith 
830c2be2410SBarry Smith    Options Database Key:
831c2be2410SBarry Smith $  -pc_mg_galerkin
832c2be2410SBarry Smith 
833c2be2410SBarry Smith    Level: intermediate
834c2be2410SBarry Smith 
835c2be2410SBarry Smith .keywords: MG, set, Galerkin
836c2be2410SBarry Smith 
8373fc8bf9cSBarry Smith .seealso: PCMGGetGalerkin()
8383fc8bf9cSBarry Smith 
839c2be2410SBarry Smith @*/
84097177400SBarry Smith PetscErrorCode PETSCKSP_DLLEXPORT PCMGSetGalerkin(PC pc)
841c2be2410SBarry Smith {
842f3fbd535SBarry Smith   PC_MG        *mg = (PC_MG*)pc->data;
843c2be2410SBarry Smith 
844c2be2410SBarry Smith   PetscFunctionBegin;
8450700a824SBarry Smith   PetscValidHeaderSpecific(pc,PC_CLASSID,1);
846218a07d4SBarry Smith   mg->galerkin = PETSC_TRUE;
847c2be2410SBarry Smith   PetscFunctionReturn(0);
848c2be2410SBarry Smith }
849c2be2410SBarry Smith 
850c2be2410SBarry Smith #undef __FUNCT__
8513fc8bf9cSBarry Smith #define __FUNCT__ "PCMGGetGalerkin"
8523fc8bf9cSBarry Smith /*@
8533fc8bf9cSBarry Smith    PCMGGetGalerkin - Checks if Galerkin multigrid is being used, i.e.
8543fc8bf9cSBarry Smith       A_i-1 = r_i * A_i * r_i^t
8553fc8bf9cSBarry Smith 
8563fc8bf9cSBarry Smith    Not Collective
8573fc8bf9cSBarry Smith 
8583fc8bf9cSBarry Smith    Input Parameter:
8593fc8bf9cSBarry Smith .  pc - the multigrid context
8603fc8bf9cSBarry Smith 
8613fc8bf9cSBarry Smith    Output Parameter:
8623fc8bf9cSBarry Smith .  gelerkin - PETSC_TRUE or PETSC_FALSE
8633fc8bf9cSBarry Smith 
8643fc8bf9cSBarry Smith    Options Database Key:
8653fc8bf9cSBarry Smith $  -pc_mg_galerkin
8663fc8bf9cSBarry Smith 
8673fc8bf9cSBarry Smith    Level: intermediate
8683fc8bf9cSBarry Smith 
8693fc8bf9cSBarry Smith .keywords: MG, set, Galerkin
8703fc8bf9cSBarry Smith 
8713fc8bf9cSBarry Smith .seealso: PCMGSetGalerkin()
8723fc8bf9cSBarry Smith 
8733fc8bf9cSBarry Smith @*/
8743fc8bf9cSBarry Smith PetscErrorCode PETSCKSP_DLLEXPORT PCMGGetGalerkin(PC pc,PetscTruth *galerkin)
8753fc8bf9cSBarry Smith {
876f3fbd535SBarry Smith   PC_MG        *mg = (PC_MG*)pc->data;
8773fc8bf9cSBarry Smith 
8783fc8bf9cSBarry Smith   PetscFunctionBegin;
8790700a824SBarry Smith   PetscValidHeaderSpecific(pc,PC_CLASSID,1);
880218a07d4SBarry Smith   *galerkin = mg->galerkin;
8813fc8bf9cSBarry Smith   PetscFunctionReturn(0);
8823fc8bf9cSBarry Smith }
8833fc8bf9cSBarry Smith 
8843fc8bf9cSBarry Smith #undef __FUNCT__
8859dcbbd2bSBarry Smith #define __FUNCT__ "PCMGSetNumberSmoothDown"
8864b9ad928SBarry Smith /*@
88797177400SBarry Smith    PCMGSetNumberSmoothDown - Sets the number of pre-smoothing steps to
88897177400SBarry Smith    use on all levels. Use PCMGGetSmootherDown() to set different
8894b9ad928SBarry Smith    pre-smoothing steps on different levels.
8904b9ad928SBarry Smith 
891*ad4df100SBarry Smith    Logically Collective on PC
8924b9ad928SBarry Smith 
8934b9ad928SBarry Smith    Input Parameters:
8944b9ad928SBarry Smith +  mg - the multigrid context
8954b9ad928SBarry Smith -  n - the number of smoothing steps
8964b9ad928SBarry Smith 
8974b9ad928SBarry Smith    Options Database Key:
8984b9ad928SBarry Smith .  -pc_mg_smoothdown <n> - Sets number of pre-smoothing steps
8994b9ad928SBarry Smith 
9004b9ad928SBarry Smith    Level: advanced
9014b9ad928SBarry Smith 
9024b9ad928SBarry Smith .keywords: MG, smooth, down, pre-smoothing, steps, multigrid
9034b9ad928SBarry Smith 
90497177400SBarry Smith .seealso: PCMGSetNumberSmoothUp()
9054b9ad928SBarry Smith @*/
90697177400SBarry Smith PetscErrorCode PETSCKSP_DLLEXPORT PCMGSetNumberSmoothDown(PC pc,PetscInt n)
9074b9ad928SBarry Smith {
908f3fbd535SBarry Smith   PC_MG          *mg = (PC_MG*)pc->data;
909f3fbd535SBarry Smith   PC_MG_Levels   **mglevels = mg->levels;
9106849ba73SBarry Smith   PetscErrorCode ierr;
91179416396SBarry Smith   PetscInt       i,levels;
9124b9ad928SBarry Smith 
9134b9ad928SBarry Smith   PetscFunctionBegin;
9140700a824SBarry Smith   PetscValidHeaderSpecific(pc,PC_CLASSID,1);
915e7e72b3dSBarry Smith   if (!mglevels) SETERRQ(((PetscObject)pc)->comm,PETSC_ERR_ARG_WRONGSTATE,"Must set MG levels before calling");
916f3fbd535SBarry Smith   levels = mglevels[0]->levels;
9174b9ad928SBarry Smith 
918b05257ddSBarry Smith   for (i=1; i<levels; i++) {
9194b9ad928SBarry Smith     /* make sure smoother up and down are different */
92097177400SBarry Smith     ierr = PCMGGetSmootherUp(pc,i,PETSC_NULL);CHKERRQ(ierr);
921f3fbd535SBarry Smith     ierr = KSPSetTolerances(mglevels[i]->smoothd,PETSC_DEFAULT,PETSC_DEFAULT,PETSC_DEFAULT,n);CHKERRQ(ierr);
92231567311SBarry Smith     mg->default_smoothd = n;
9234b9ad928SBarry Smith   }
9244b9ad928SBarry Smith   PetscFunctionReturn(0);
9254b9ad928SBarry Smith }
9264b9ad928SBarry Smith 
9274b9ad928SBarry Smith #undef __FUNCT__
9289dcbbd2bSBarry Smith #define __FUNCT__ "PCMGSetNumberSmoothUp"
9294b9ad928SBarry Smith /*@
93097177400SBarry Smith    PCMGSetNumberSmoothUp - Sets the number of post-smoothing steps to use
93197177400SBarry Smith    on all levels. Use PCMGGetSmootherUp() to set different numbers of
9324b9ad928SBarry Smith    post-smoothing steps on different levels.
9334b9ad928SBarry Smith 
934*ad4df100SBarry Smith    Logically Collective on PC
9354b9ad928SBarry Smith 
9364b9ad928SBarry Smith    Input Parameters:
9374b9ad928SBarry Smith +  mg - the multigrid context
9384b9ad928SBarry Smith -  n - the number of smoothing steps
9394b9ad928SBarry Smith 
9404b9ad928SBarry Smith    Options Database Key:
9414b9ad928SBarry Smith .  -pc_mg_smoothup <n> - Sets number of post-smoothing steps
9424b9ad928SBarry Smith 
9434b9ad928SBarry Smith    Level: advanced
9444b9ad928SBarry Smith 
9454b9ad928SBarry Smith    Note: this does not set a value on the coarsest grid, since we assume that
946a8c7a070SBarry Smith     there is no separate smooth up on the coarsest grid.
9474b9ad928SBarry Smith 
9484b9ad928SBarry Smith .keywords: MG, smooth, up, post-smoothing, steps, multigrid
9494b9ad928SBarry Smith 
95097177400SBarry Smith .seealso: PCMGSetNumberSmoothDown()
9514b9ad928SBarry Smith @*/
95297177400SBarry Smith PetscErrorCode PETSCKSP_DLLEXPORT PCMGSetNumberSmoothUp(PC pc,PetscInt n)
9534b9ad928SBarry Smith {
954f3fbd535SBarry Smith   PC_MG          *mg = (PC_MG*)pc->data;
955f3fbd535SBarry Smith   PC_MG_Levels   **mglevels = mg->levels;
9566849ba73SBarry Smith   PetscErrorCode ierr;
95779416396SBarry Smith   PetscInt       i,levels;
9584b9ad928SBarry Smith 
9594b9ad928SBarry Smith   PetscFunctionBegin;
9600700a824SBarry Smith   PetscValidHeaderSpecific(pc,PC_CLASSID,1);
961e7e72b3dSBarry Smith   if (!mglevels) SETERRQ(((PetscObject)pc)->comm,PETSC_ERR_ARG_WRONGSTATE,"Must set MG levels before calling");
962f3fbd535SBarry Smith   levels = mglevels[0]->levels;
9634b9ad928SBarry Smith 
9644b9ad928SBarry Smith   for (i=1; i<levels; i++) {
9654b9ad928SBarry Smith     /* make sure smoother up and down are different */
96697177400SBarry Smith     ierr = PCMGGetSmootherUp(pc,i,PETSC_NULL);CHKERRQ(ierr);
967f3fbd535SBarry Smith     ierr = KSPSetTolerances(mglevels[i]->smoothu,PETSC_DEFAULT,PETSC_DEFAULT,PETSC_DEFAULT,n);CHKERRQ(ierr);
96831567311SBarry Smith     mg->default_smoothu = n;
9694b9ad928SBarry Smith   }
9704b9ad928SBarry Smith   PetscFunctionReturn(0);
9714b9ad928SBarry Smith }
9724b9ad928SBarry Smith 
9734b9ad928SBarry Smith /* ----------------------------------------------------------------------------------------*/
9744b9ad928SBarry Smith 
9753b09bd56SBarry Smith /*MC
976ccb205f8SBarry Smith    PCMG - Use multigrid preconditioning. This preconditioner requires you provide additional
9773b09bd56SBarry Smith     information about the coarser grid matrices and restriction/interpolation operators.
9783b09bd56SBarry Smith 
9793b09bd56SBarry Smith    Options Database Keys:
9803b09bd56SBarry Smith +  -pc_mg_levels <nlevels> - number of levels including finest
9810d353602SBarry Smith .  -pc_mg_cycles v or w
98279416396SBarry Smith .  -pc_mg_smoothup <n> - number of smoothing steps after interpolation
9833b09bd56SBarry Smith .  -pc_mg_smoothdown <n> - number of smoothing steps before applying restriction operator
9843b09bd56SBarry Smith .  -pc_mg_type <additive,multiplicative,full,cascade> - multiplicative is the default
9853b09bd56SBarry Smith .  -pc_mg_log - log information about time spent on each level of the solver
9863b09bd56SBarry Smith .  -pc_mg_monitor - print information on the multigrid convergence
98768eff7e6SBarry Smith .  -pc_mg_galerkin - use Galerkin process to compute coarser operators
9883b09bd56SBarry Smith -  -pc_mg_dump_matlab - dumps the matrices for each level and the restriction/interpolation matrices
9893b09bd56SBarry Smith                         to the Socket viewer for reading from Matlab.
9903b09bd56SBarry Smith 
99124c3aa18SBarry Smith    Notes: By default this uses GMRES on the fine grid smoother so this should be used with KSPFGMRES or the smoother changed to not use GMRES
9923b09bd56SBarry Smith 
9933b09bd56SBarry Smith    Level: intermediate
9943b09bd56SBarry Smith 
9958f87f92bSBarry Smith    Concepts: multigrid/multilevel
9963b09bd56SBarry Smith 
99724c3aa18SBarry Smith .seealso:  PCCreate(), PCSetType(), PCType (for list of available types), PC, PCMGType, PCEXOTIC
9980d353602SBarry Smith            PCMGSetLevels(), PCMGGetLevels(), PCMGSetType(), PCMGSetCycleType(), PCMGSetNumberSmoothDown(),
99997177400SBarry Smith            PCMGSetNumberSmoothUp(), PCMGGetCoarseSolve(), PCMGSetResidual(), PCMGSetInterpolation(),
100097177400SBarry Smith            PCMGSetRestriction(), PCMGGetSmoother(), PCMGGetSmootherUp(), PCMGGetSmootherDown(),
10010d353602SBarry Smith            PCMGSetCycleTypeOnLevel(), PCMGSetRhs(), PCMGSetX(), PCMGSetR()
10023b09bd56SBarry Smith M*/
10033b09bd56SBarry Smith 
10044b9ad928SBarry Smith EXTERN_C_BEGIN
10054b9ad928SBarry Smith #undef __FUNCT__
10064b9ad928SBarry Smith #define __FUNCT__ "PCCreate_MG"
1007dba47a55SKris Buschelman PetscErrorCode PETSCKSP_DLLEXPORT PCCreate_MG(PC pc)
10084b9ad928SBarry Smith {
1009f3fbd535SBarry Smith   PC_MG          *mg;
1010f3fbd535SBarry Smith   PetscErrorCode ierr;
1011f3fbd535SBarry Smith 
10124b9ad928SBarry Smith   PetscFunctionBegin;
1013f3fbd535SBarry Smith   ierr        = PetscNewLog(pc,PC_MG,&mg);CHKERRQ(ierr);
1014f3fbd535SBarry Smith   pc->data    = (void*)mg;
1015f3fbd535SBarry Smith   mg->nlevels = -1;
1016f3fbd535SBarry Smith 
10174b9ad928SBarry Smith   pc->ops->apply          = PCApply_MG;
10184b9ad928SBarry Smith   pc->ops->setup          = PCSetUp_MG;
10194b9ad928SBarry Smith   pc->ops->destroy        = PCDestroy_MG;
10204b9ad928SBarry Smith   pc->ops->setfromoptions = PCSetFromOptions_MG;
10214b9ad928SBarry Smith   pc->ops->view           = PCView_MG;
10224b9ad928SBarry Smith   PetscFunctionReturn(0);
10234b9ad928SBarry Smith }
10244b9ad928SBarry Smith EXTERN_C_END
1025