xref: /petsc/src/ksp/pc/interface/pcset.c (revision 7d0a6c19129e7069c8a40e210b34ed62989173db)
1*7d0a6c19SBarry Smith 
24b9ad928SBarry Smith /*
34b9ad928SBarry Smith     Routines to set PC methods and options.
44b9ad928SBarry Smith */
54b9ad928SBarry Smith 
66356e834SBarry Smith #include "private/pcimpl.h"      /*I "petscpc.h" I*/
74b9ad928SBarry Smith 
8ace3abfcSBarry Smith PetscBool  PCRegisterAllCalled = PETSC_FALSE;
94b9ad928SBarry Smith /*
104b9ad928SBarry Smith    Contains the list of registered KSP routines
114b9ad928SBarry Smith */
124b9ad928SBarry Smith PetscFList PCList = 0;
134b9ad928SBarry Smith 
144b9ad928SBarry Smith #undef __FUNCT__
154b9ad928SBarry Smith #define __FUNCT__ "PCSetType"
164b9ad928SBarry Smith /*@C
174b9ad928SBarry Smith    PCSetType - Builds PC for a particular preconditioner.
184b9ad928SBarry Smith 
194b9ad928SBarry Smith    Collective on PC
204b9ad928SBarry Smith 
214b9ad928SBarry Smith    Input Parameter:
224b9ad928SBarry Smith +  pc - the preconditioner context.
234b9ad928SBarry Smith -  type - a known method
244b9ad928SBarry Smith 
254b9ad928SBarry Smith    Options Database Key:
264b9ad928SBarry Smith .  -pc_type <type> - Sets PC type
274b9ad928SBarry Smith 
284b9ad928SBarry Smith    Use -help for a list of available methods (for instance,
294b9ad928SBarry Smith    jacobi or bjacobi)
304b9ad928SBarry Smith 
314b9ad928SBarry Smith   Notes:
324b9ad928SBarry Smith   See "petsc/include/petscpc.h" for available methods (for instance,
334b9ad928SBarry Smith   PCJACOBI, PCILU, or PCBJACOBI).
344b9ad928SBarry Smith 
354b9ad928SBarry Smith   Normally, it is best to use the KSPSetFromOptions() command and
364b9ad928SBarry Smith   then set the PC type from the options database rather than by using
374b9ad928SBarry Smith   this routine.  Using the options database provides the user with
384b9ad928SBarry Smith   maximum flexibility in evaluating the many different preconditioners.
394b9ad928SBarry Smith   The PCSetType() routine is provided for those situations where it
404b9ad928SBarry Smith   is necessary to set the preconditioner independently of the command
414b9ad928SBarry Smith   line or options database.  This might be the case, for example, when
424b9ad928SBarry Smith   the choice of preconditioner changes during the execution of the
434b9ad928SBarry Smith   program, and the user's application is taking responsibility for
444b9ad928SBarry Smith   choosing the appropriate preconditioner.  In other words, this
454b9ad928SBarry Smith   routine is not for beginners.
464b9ad928SBarry Smith 
474b9ad928SBarry Smith   Level: intermediate
484b9ad928SBarry Smith 
494b9ad928SBarry Smith .keywords: PC, set, method, type
504b9ad928SBarry Smith 
514b9ad928SBarry Smith .seealso: KSPSetType(), PCType
524b9ad928SBarry Smith 
534b9ad928SBarry Smith @*/
547087cfbeSBarry Smith PetscErrorCode  PCSetType(PC pc,const PCType type)
554b9ad928SBarry Smith {
56dfbe8321SBarry Smith   PetscErrorCode ierr,(*r)(PC);
57ace3abfcSBarry Smith   PetscBool      match;
584b9ad928SBarry Smith 
594b9ad928SBarry Smith   PetscFunctionBegin;
600700a824SBarry Smith   PetscValidHeaderSpecific(pc,PC_CLASSID,1);
614482741eSBarry Smith   PetscValidCharPointer(type,2);
624b9ad928SBarry Smith 
634b9ad928SBarry Smith   ierr = PetscTypeCompare((PetscObject)pc,type,&match);CHKERRQ(ierr);
644b9ad928SBarry Smith   if (match) PetscFunctionReturn(0);
654b9ad928SBarry Smith 
667adad957SLisandro Dalcin   ierr =  PetscFListFind(PCList,((PetscObject)pc)->comm,type,(void (**)(void)) &r);CHKERRQ(ierr);
6765e19b50SBarry Smith   if (!r) SETERRQ1(((PetscObject)pc)->comm,PETSC_ERR_ARG_UNKNOWN_TYPE,"Unable to find requested PC type %s",type);
68a7d21a52SLisandro Dalcin   /* Destroy the previous private PC context */
6914ca34e6SBarry Smith   if (pc->ops->destroy) { ierr =  (*pc->ops->destroy)(pc);CHKERRQ(ierr); pc->data = 0;}
707adad957SLisandro Dalcin   ierr = PetscFListDestroy(&((PetscObject)pc)->qlist);CHKERRQ(ierr);
71a7d21a52SLisandro Dalcin   /* Reinitialize function pointers in PCOps structure */
72a7d21a52SLisandro Dalcin   ierr = PetscMemzero(pc->ops,sizeof(struct _PCOps));CHKERRQ(ierr);
73a7d21a52SLisandro Dalcin   /* XXX Is this OK?? */
74a7d21a52SLisandro Dalcin   pc->modifysubmatrices        = 0;
75a7d21a52SLisandro Dalcin   pc->modifysubmatricesP       = 0;
76a7d21a52SLisandro Dalcin   /* Call the PCCreate_XXX routine for this particular preconditioner */
77a7d21a52SLisandro Dalcin   pc->setupcalled = 0;
784b9ad928SBarry Smith   ierr = (*r)(pc);CHKERRQ(ierr);
794b9ad928SBarry Smith   ierr = PetscObjectChangeTypeName((PetscObject)pc,type);CHKERRQ(ierr);
809fb22e1aSBarry Smith #if defined(PETSC_HAVE_AMS)
819fb22e1aSBarry Smith   if (PetscAMSPublishAll) {
829fb22e1aSBarry Smith     ierr = PetscObjectAMSPublish((PetscObject)pc);CHKERRQ(ierr);
839fb22e1aSBarry Smith   }
849fb22e1aSBarry Smith #endif
854b9ad928SBarry Smith   PetscFunctionReturn(0);
864b9ad928SBarry Smith }
874b9ad928SBarry Smith 
884b9ad928SBarry Smith #undef __FUNCT__
894b9ad928SBarry Smith #define __FUNCT__ "PCRegisterDestroy"
90f39d8e23SSatish Balay /*@
914b9ad928SBarry Smith    PCRegisterDestroy - Frees the list of preconditioners that were
924b9ad928SBarry Smith    registered by PCRegisterDynamic().
934b9ad928SBarry Smith 
944b9ad928SBarry Smith    Not Collective
954b9ad928SBarry Smith 
964b9ad928SBarry Smith    Level: advanced
974b9ad928SBarry Smith 
984b9ad928SBarry Smith .keywords: PC, register, destroy
994b9ad928SBarry Smith 
1004b9ad928SBarry Smith .seealso: PCRegisterAll(), PCRegisterAll()
1014b9ad928SBarry Smith 
1024b9ad928SBarry Smith @*/
1037087cfbeSBarry Smith PetscErrorCode  PCRegisterDestroy(void)
1044b9ad928SBarry Smith {
105dfbe8321SBarry Smith   PetscErrorCode ierr;
1064b9ad928SBarry Smith 
1074b9ad928SBarry Smith   PetscFunctionBegin;
1081441b1d3SBarry Smith   ierr = PetscFListDestroy(&PCList);CHKERRQ(ierr);
1094b9ad928SBarry Smith   PCRegisterAllCalled = PETSC_FALSE;
1104b9ad928SBarry Smith   PetscFunctionReturn(0);
1114b9ad928SBarry Smith }
1124b9ad928SBarry Smith 
1134b9ad928SBarry Smith #undef __FUNCT__
1144b9ad928SBarry Smith #define __FUNCT__ "PCGetType"
1154b9ad928SBarry Smith /*@C
1164b9ad928SBarry Smith    PCGetType - Gets the PC method type and name (as a string) from the PC
1174b9ad928SBarry Smith    context.
1184b9ad928SBarry Smith 
1194b9ad928SBarry Smith    Not Collective
1204b9ad928SBarry Smith 
1214b9ad928SBarry Smith    Input Parameter:
1224b9ad928SBarry Smith .  pc - the preconditioner context
1234b9ad928SBarry Smith 
1244b9ad928SBarry Smith    Output Parameter:
125c4e43342SLisandro Dalcin .  type - name of preconditioner method
1264b9ad928SBarry Smith 
1274b9ad928SBarry Smith    Level: intermediate
1284b9ad928SBarry Smith 
1294b9ad928SBarry Smith .keywords: PC, get, method, name, type
1304b9ad928SBarry Smith 
1314b9ad928SBarry Smith .seealso: PCSetType()
1324b9ad928SBarry Smith 
1334b9ad928SBarry Smith @*/
1347087cfbeSBarry Smith PetscErrorCode  PCGetType(PC pc,const PCType *type)
1354b9ad928SBarry Smith {
1364b9ad928SBarry Smith   PetscFunctionBegin;
1370700a824SBarry Smith   PetscValidHeaderSpecific(pc,PC_CLASSID,1);
138c4e43342SLisandro Dalcin   PetscValidPointer(type,2);
139c4e43342SLisandro Dalcin   *type = ((PetscObject)pc)->type_name;
1404b9ad928SBarry Smith   PetscFunctionReturn(0);
1414b9ad928SBarry Smith }
1424b9ad928SBarry Smith 
14309573ac7SBarry Smith extern PetscErrorCode PCGetDefaultType_Private(PC,const char*[]);
144566e8bf2SBarry Smith 
1454b9ad928SBarry Smith #undef __FUNCT__
1464b9ad928SBarry Smith #define __FUNCT__ "PCSetFromOptions"
1474b9ad928SBarry Smith /*@
1484b9ad928SBarry Smith    PCSetFromOptions - Sets PC options from the options database.
1494b9ad928SBarry Smith    This routine must be called before PCSetUp() if the user is to be
1504b9ad928SBarry Smith    allowed to set the preconditioner method.
1514b9ad928SBarry Smith 
1524b9ad928SBarry Smith    Collective on PC
1534b9ad928SBarry Smith 
1544b9ad928SBarry Smith    Input Parameter:
1554b9ad928SBarry Smith .  pc - the preconditioner context
1564b9ad928SBarry Smith 
1574b9ad928SBarry Smith    Level: developer
1584b9ad928SBarry Smith 
1594b9ad928SBarry Smith .keywords: PC, set, from, options, database
1604b9ad928SBarry Smith 
1614b9ad928SBarry Smith .seealso:
1624b9ad928SBarry Smith 
1634b9ad928SBarry Smith @*/
1647087cfbeSBarry Smith PetscErrorCode  PCSetFromOptions(PC pc)
1654b9ad928SBarry Smith {
166dfbe8321SBarry Smith   PetscErrorCode ierr;
1672fc52814SBarry Smith   char           type[256];
1682fc52814SBarry Smith   const char     *def;
169ace3abfcSBarry Smith   PetscBool      flg;
1704b9ad928SBarry Smith 
1714b9ad928SBarry Smith   PetscFunctionBegin;
1720700a824SBarry Smith   PetscValidHeaderSpecific(pc,PC_CLASSID,1);
1734b9ad928SBarry Smith 
1744b9ad928SBarry Smith   if (!PCRegisterAllCalled) {ierr = PCRegisterAll(PETSC_NULL);CHKERRQ(ierr);}
1757adad957SLisandro Dalcin   ierr = PetscOptionsBegin(((PetscObject)pc)->comm,((PetscObject)pc)->prefix,"Preconditioner (PC) Options","PC");CHKERRQ(ierr);
1767adad957SLisandro Dalcin     if (!((PetscObject)pc)->type_name) {
177566e8bf2SBarry Smith       ierr = PCGetDefaultType_Private(pc,&def);CHKERRQ(ierr);
1784b9ad928SBarry Smith     } else {
1797adad957SLisandro Dalcin       def = ((PetscObject)pc)->type_name;
1804b9ad928SBarry Smith     }
1814b9ad928SBarry Smith 
1824b9ad928SBarry Smith     ierr = PetscOptionsList("-pc_type","Preconditioner","PCSetType",PCList,def,type,256,&flg);CHKERRQ(ierr);
1834b9ad928SBarry Smith     if (flg) {
1844b9ad928SBarry Smith       ierr = PCSetType(pc,type);CHKERRQ(ierr);
1857adad957SLisandro Dalcin     } else if (!((PetscObject)pc)->type_name){
186566e8bf2SBarry Smith       ierr = PCSetType(pc,def);CHKERRQ(ierr);
1874b9ad928SBarry Smith     }
1884b9ad928SBarry Smith 
1894b9ad928SBarry Smith     if (pc->ops->setfromoptions) {
1904b9ad928SBarry Smith       ierr = (*pc->ops->setfromoptions)(pc);CHKERRQ(ierr);
1914b9ad928SBarry Smith     }
1925d973c19SBarry Smith 
1935d973c19SBarry Smith     /* process any options handlers added with PetscObjectAddOptionsHandler() */
1945d973c19SBarry Smith     ierr = PetscObjectProcessOptionsHandlers((PetscObject)pc);CHKERRQ(ierr);
1954b9ad928SBarry Smith   ierr = PetscOptionsEnd();CHKERRQ(ierr);
1960c24e6a1SHong Zhang   pc->setfromoptionscalled++;
1974b9ad928SBarry Smith   PetscFunctionReturn(0);
1984b9ad928SBarry Smith }
1996c699258SBarry Smith 
2006c699258SBarry Smith #undef __FUNCT__
2016c699258SBarry Smith #define __FUNCT__ "PCSetDM"
2026c699258SBarry Smith /*@
2036c699258SBarry Smith    PCSetDM - Sets the DM that may be used by some preconditioners
2046c699258SBarry Smith 
2053f9fe445SBarry Smith    Logically Collective on PC
2066c699258SBarry Smith 
2076c699258SBarry Smith    Input Parameters:
2086c699258SBarry Smith +  pc - the preconditioner context
2096c699258SBarry Smith -  dm - the dm
2106c699258SBarry Smith 
2116c699258SBarry Smith    Level: intermediate
2126c699258SBarry Smith 
2136c699258SBarry Smith 
2146c699258SBarry Smith .seealso: PCGetDM(), KSPSetDM(), KSPGetDM()
2156c699258SBarry Smith @*/
2167087cfbeSBarry Smith PetscErrorCode  PCSetDM(PC pc,DM dm)
2176c699258SBarry Smith {
2186c699258SBarry Smith   PetscErrorCode ierr;
2196c699258SBarry Smith 
2206c699258SBarry Smith   PetscFunctionBegin;
2210700a824SBarry Smith   PetscValidHeaderSpecific(pc,PC_CLASSID,1);
22270663e4aSLisandro Dalcin   ierr = PetscObjectReference((PetscObject)dm);CHKERRQ(ierr);
2236c699258SBarry Smith   if (pc->dm) {ierr = DMDestroy(pc->dm);CHKERRQ(ierr);}
2246c699258SBarry Smith   pc->dm = dm;
2256c699258SBarry Smith   PetscFunctionReturn(0);
2266c699258SBarry Smith }
2276c699258SBarry Smith 
2286c699258SBarry Smith #undef __FUNCT__
2296c699258SBarry Smith #define __FUNCT__ "PCGetDM"
2306c699258SBarry Smith /*@
2316c699258SBarry Smith    PCGetDM - Gets the DM that may be used by some preconditioners
2326c699258SBarry Smith 
2333f9fe445SBarry Smith    Not Collective
2346c699258SBarry Smith 
2356c699258SBarry Smith    Input Parameter:
2366c699258SBarry Smith . pc - the preconditioner context
2376c699258SBarry Smith 
2386c699258SBarry Smith    Output Parameter:
2396c699258SBarry Smith .  dm - the dm
2406c699258SBarry Smith 
2416c699258SBarry Smith    Level: intermediate
2426c699258SBarry Smith 
2436c699258SBarry Smith 
2446c699258SBarry Smith .seealso: PCSetDM(), KSPSetDM(), KSPGetDM()
2456c699258SBarry Smith @*/
2467087cfbeSBarry Smith PetscErrorCode  PCGetDM(PC pc,DM *dm)
2476c699258SBarry Smith {
2486c699258SBarry Smith   PetscFunctionBegin;
2490700a824SBarry Smith   PetscValidHeaderSpecific(pc,PC_CLASSID,1);
2506c699258SBarry Smith   *dm = pc->dm;
2516c699258SBarry Smith   PetscFunctionReturn(0);
2526c699258SBarry Smith }
2536c699258SBarry Smith 
254