xref: /petsc/src/ksp/pc/impls/mg/smg.c (revision efb308899f58eff35e31f4136d1431f0c81bf4bc)
1dba47a55SKris Buschelman #define PETSCKSP_DLL
2dba47a55SKris Buschelman 
34b9ad928SBarry Smith /*
44b9ad928SBarry Smith      Additive Multigrid V Cycle routine
54b9ad928SBarry Smith */
64b9ad928SBarry Smith #include "src/ksp/pc/impls/mg/mgimpl.h"
74b9ad928SBarry Smith 
84b9ad928SBarry Smith #undef __FUNCT__
99dcbbd2bSBarry Smith #define __FUNCT__ "PCMGACycle_Private"
109dcbbd2bSBarry Smith PetscErrorCode PCMGACycle_Private(PC_MG **mg)
114b9ad928SBarry Smith {
126849ba73SBarry Smith   PetscErrorCode ierr;
1379416396SBarry Smith   PetscInt       i,l = mg[0]->levels;
144b9ad928SBarry Smith 
154b9ad928SBarry Smith   PetscFunctionBegin;
164b9ad928SBarry Smith   /* compute RHS on each level */
174b9ad928SBarry Smith   for (i=l-1; i>0; i--) {
184b9ad928SBarry Smith     ierr = MatRestrict(mg[i]->restrct,mg[i]->b,mg[i-1]->b);CHKERRQ(ierr);
194b9ad928SBarry Smith   }
204b9ad928SBarry Smith   /* solve seperately on each level */
214b9ad928SBarry Smith   for (i=0; i<l; i++) {
22*efb30889SBarry Smith     ierr = VecSet(mg[i]->x,0.0);CHKERRQ(ierr);
234b9ad928SBarry Smith     if (mg[i]->eventsolve) {ierr = PetscLogEventBegin(mg[i]->eventsolve,0,0,0,0);CHKERRQ(ierr);}
2423ce1328SBarry Smith     ierr = KSPSolve(mg[i]->smoothd,mg[i]->b,mg[i]->x);CHKERRQ(ierr);
254b9ad928SBarry Smith     if (mg[i]->eventsolve) {ierr = PetscLogEventEnd(mg[i]->eventsolve,0,0,0,0);CHKERRQ(ierr);}
264b9ad928SBarry Smith   }
274b9ad928SBarry Smith   for (i=1; i<l; i++) {
284b9ad928SBarry Smith     ierr = MatInterpolateAdd(mg[i]->interpolate,mg[i-1]->x,mg[i]->x,mg[i]->x);CHKERRQ(ierr);
294b9ad928SBarry Smith   }
304b9ad928SBarry Smith   PetscFunctionReturn(0);
314b9ad928SBarry Smith }
32