xref: /petsc/src/ksp/pc/impls/mg/smg.c (revision af0996ce37bc06907c37d8d91773840993d61e62)
1dba47a55SKris Buschelman 
24b9ad928SBarry Smith /*
34b9ad928SBarry Smith      Additive Multigrid V Cycle routine
44b9ad928SBarry Smith */
5*af0996ceSBarry Smith #include <petsc/private/pcmgimpl.h>
64b9ad928SBarry Smith 
74b9ad928SBarry Smith #undef __FUNCT__
89dcbbd2bSBarry Smith #define __FUNCT__ "PCMGACycle_Private"
931567311SBarry Smith PetscErrorCode PCMGACycle_Private(PC pc,PC_MG_Levels **mglevels)
104b9ad928SBarry Smith {
116849ba73SBarry Smith   PetscErrorCode ierr;
12f3fbd535SBarry Smith   PetscInt       i,l = mglevels[0]->levels;
134b9ad928SBarry Smith 
144b9ad928SBarry Smith   PetscFunctionBegin;
154b9ad928SBarry Smith   /* compute RHS on each level */
164b9ad928SBarry Smith   for (i=l-1; i>0; i--) {
1763e6d426SJed Brown     if (mglevels[i]->eventinterprestrict) {ierr = PetscLogEventBegin(mglevels[i]->eventinterprestrict,0,0,0,0);CHKERRQ(ierr);}
18f3fbd535SBarry Smith     ierr = MatRestrict(mglevels[i]->restrct,mglevels[i]->b,mglevels[i-1]->b);CHKERRQ(ierr);
1963e6d426SJed Brown     if (mglevels[i]->eventinterprestrict) {ierr = PetscLogEventEnd(mglevels[i]->eventinterprestrict,0,0,0,0);CHKERRQ(ierr);}
204b9ad928SBarry Smith   }
21a8c7a070SBarry Smith   /* solve separately on each level */
224b9ad928SBarry Smith   for (i=0; i<l; i++) {
23f3fbd535SBarry Smith     ierr = VecSet(mglevels[i]->x,0.0);CHKERRQ(ierr);
2463e6d426SJed Brown     if (mglevels[i]->eventsmoothsolve) {ierr = PetscLogEventBegin(mglevels[i]->eventsmoothsolve,0,0,0,0);CHKERRQ(ierr);}
25f3fbd535SBarry Smith     ierr = KSPSolve(mglevels[i]->smoothd,mglevels[i]->b,mglevels[i]->x);CHKERRQ(ierr);
2663e6d426SJed Brown     if (mglevels[i]->eventsmoothsolve) {ierr = PetscLogEventEnd(mglevels[i]->eventsmoothsolve,0,0,0,0);CHKERRQ(ierr);}
274b9ad928SBarry Smith   }
284b9ad928SBarry Smith   for (i=1; i<l; i++) {
2963e6d426SJed Brown     if (mglevels[i]->eventinterprestrict) {ierr = PetscLogEventBegin(mglevels[i]->eventinterprestrict,0,0,0,0);CHKERRQ(ierr);}
30f3fbd535SBarry Smith     ierr = MatInterpolateAdd(mglevels[i]->interpolate,mglevels[i-1]->x,mglevels[i]->x,mglevels[i]->x);CHKERRQ(ierr);
3163e6d426SJed Brown     if (mglevels[i]->eventinterprestrict) {ierr = PetscLogEventEnd(mglevels[i]->eventinterprestrict,0,0,0,0);CHKERRQ(ierr);}
324b9ad928SBarry Smith   }
334b9ad928SBarry Smith   PetscFunctionReturn(0);
344b9ad928SBarry Smith }
35