19b94acceSBarry Smith 2af0996ceSBarry Smith #include <petsc/private/snesimpl.h> /*I "petscsnes.h" I*/ 307475bc1SBarry Smith #include <petscdmshell.h> 4d96771aaSLisandro Dalcin #include <petscdraw.h> 5a01aa210SMatthew G. Knepley #include <petscds.h> 634b4d3a8SMatthew G. Knepley #include <petscdmadaptor.h> 706fc46c8SMatthew G. Knepley #include <petscconvest.h> 89b94acceSBarry Smith 9ace3abfcSBarry Smith PetscBool SNESRegisterAllCalled = PETSC_FALSE; 100298fd71SBarry Smith PetscFunctionList SNESList = NULL; 118ba1e511SMatthew Knepley 128ba1e511SMatthew Knepley /* Logging support */ 1322c6f798SBarry Smith PetscClassId SNES_CLASSID, DMSNES_CLASSID; 1494db00ebSBarry Smith PetscLogEvent SNES_Solve, SNES_FunctionEval, SNES_JacobianEval, SNES_NGSEval, SNES_NGSFuncEval, SNES_NPCSolve, SNES_ObjectiveEval; 15a09944afSBarry Smith 16e113a28aSBarry Smith /*@ 17e113a28aSBarry Smith SNESSetErrorIfNotConverged - Causes SNESSolve() to generate an error if the solver has not converged. 18e113a28aSBarry Smith 193f9fe445SBarry Smith Logically Collective on SNES 20e113a28aSBarry Smith 21e113a28aSBarry Smith Input Parameters: 22e113a28aSBarry Smith + snes - iterative context obtained from SNESCreate() 23e113a28aSBarry Smith - flg - PETSC_TRUE indicates you want the error generated 24e113a28aSBarry Smith 25e113a28aSBarry Smith Options database keys: 26e113a28aSBarry Smith . -snes_error_if_not_converged : this takes an optional truth value (0/1/no/yes/true/false) 27e113a28aSBarry Smith 28e113a28aSBarry Smith Level: intermediate 29e113a28aSBarry Smith 30e113a28aSBarry Smith Notes: 31e113a28aSBarry Smith Normally PETSc continues if a linear solver fails to converge, you can call SNESGetConvergedReason() after a SNESSolve() 32e113a28aSBarry Smith to determine if it has converged. 33e113a28aSBarry Smith 34e113a28aSBarry Smith .keywords: SNES, set, initial guess, nonzero 35e113a28aSBarry Smith 36e113a28aSBarry Smith .seealso: SNESGetErrorIfNotConverged(), KSPGetErrorIfNotConverged(), KSPSetErrorIFNotConverged() 37e113a28aSBarry Smith @*/ 387087cfbeSBarry Smith PetscErrorCode SNESSetErrorIfNotConverged(SNES snes,PetscBool flg) 39e113a28aSBarry Smith { 40e113a28aSBarry Smith PetscFunctionBegin; 41e113a28aSBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 42acfcf0e5SJed Brown PetscValidLogicalCollectiveBool(snes,flg,2); 43e113a28aSBarry Smith snes->errorifnotconverged = flg; 44e113a28aSBarry Smith PetscFunctionReturn(0); 45e113a28aSBarry Smith } 46e113a28aSBarry Smith 47e113a28aSBarry Smith /*@ 48e113a28aSBarry Smith SNESGetErrorIfNotConverged - Will SNESSolve() generate an error if the solver does not converge? 49e113a28aSBarry Smith 50e113a28aSBarry Smith Not Collective 51e113a28aSBarry Smith 52e113a28aSBarry Smith Input Parameter: 53e113a28aSBarry Smith . snes - iterative context obtained from SNESCreate() 54e113a28aSBarry Smith 55e113a28aSBarry Smith Output Parameter: 56e113a28aSBarry Smith . flag - PETSC_TRUE if it will generate an error, else PETSC_FALSE 57e113a28aSBarry Smith 58e113a28aSBarry Smith Level: intermediate 59e113a28aSBarry Smith 60e113a28aSBarry Smith .keywords: SNES, set, initial guess, nonzero 61e113a28aSBarry Smith 62e113a28aSBarry Smith .seealso: SNESSetErrorIfNotConverged(), KSPGetErrorIfNotConverged(), KSPSetErrorIFNotConverged() 63e113a28aSBarry Smith @*/ 647087cfbeSBarry Smith PetscErrorCode SNESGetErrorIfNotConverged(SNES snes,PetscBool *flag) 65e113a28aSBarry Smith { 66e113a28aSBarry Smith PetscFunctionBegin; 67e113a28aSBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 68e113a28aSBarry Smith PetscValidPointer(flag,2); 69e113a28aSBarry Smith *flag = snes->errorifnotconverged; 70e113a28aSBarry Smith PetscFunctionReturn(0); 71e113a28aSBarry Smith } 72e113a28aSBarry Smith 734fc747eaSLawrence Mitchell /*@ 744fc747eaSLawrence Mitchell SNESSetAlwaysComputesFinalResidual - does the SNES always compute the residual at the final solution? 754fc747eaSLawrence Mitchell 764fc747eaSLawrence Mitchell Logically Collective on SNES 774fc747eaSLawrence Mitchell 784fc747eaSLawrence Mitchell Input Parameters: 794fc747eaSLawrence Mitchell + snes - the shell SNES 804fc747eaSLawrence Mitchell - flg - is the residual computed? 814fc747eaSLawrence Mitchell 824fc747eaSLawrence Mitchell Level: advanced 834fc747eaSLawrence Mitchell 844fc747eaSLawrence Mitchell .seealso: SNESGetAlwaysComputesFinalResidual() 854fc747eaSLawrence Mitchell @*/ 864fc747eaSLawrence Mitchell PetscErrorCode SNESSetAlwaysComputesFinalResidual(SNES snes, PetscBool flg) 874fc747eaSLawrence Mitchell { 884fc747eaSLawrence Mitchell PetscFunctionBegin; 894fc747eaSLawrence Mitchell PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 904fc747eaSLawrence Mitchell snes->alwayscomputesfinalresidual = flg; 914fc747eaSLawrence Mitchell PetscFunctionReturn(0); 924fc747eaSLawrence Mitchell } 934fc747eaSLawrence Mitchell 944fc747eaSLawrence Mitchell /*@ 954fc747eaSLawrence Mitchell SNESGetAlwaysComputesFinalResidual - does the SNES always compute the residual at the final solution? 964fc747eaSLawrence Mitchell 974fc747eaSLawrence Mitchell Logically Collective on SNES 984fc747eaSLawrence Mitchell 994fc747eaSLawrence Mitchell Input Parameter: 1004fc747eaSLawrence Mitchell . snes - the shell SNES 1014fc747eaSLawrence Mitchell 1024fc747eaSLawrence Mitchell Output Parameter: 1034fc747eaSLawrence Mitchell . flg - is the residual computed? 1044fc747eaSLawrence Mitchell 1054fc747eaSLawrence Mitchell Level: advanced 1064fc747eaSLawrence Mitchell 1074fc747eaSLawrence Mitchell .seealso: SNESSetAlwaysComputesFinalResidual() 1084fc747eaSLawrence Mitchell @*/ 1094fc747eaSLawrence Mitchell PetscErrorCode SNESGetAlwaysComputesFinalResidual(SNES snes, PetscBool *flg) 1104fc747eaSLawrence Mitchell { 1114fc747eaSLawrence Mitchell PetscFunctionBegin; 1124fc747eaSLawrence Mitchell PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 1134fc747eaSLawrence Mitchell *flg = snes->alwayscomputesfinalresidual; 1144fc747eaSLawrence Mitchell PetscFunctionReturn(0); 1154fc747eaSLawrence Mitchell } 1164fc747eaSLawrence Mitchell 117e725d27bSBarry Smith /*@ 118bf388a1fSBarry Smith SNESSetFunctionDomainError - tells SNES that the input vector to your SNESFunction is not 1194936397dSBarry Smith in the functions domain. For example, negative pressure. 1204936397dSBarry Smith 1213f9fe445SBarry Smith Logically Collective on SNES 1224936397dSBarry Smith 1234936397dSBarry Smith Input Parameters: 1246a388c36SPeter Brune . snes - the SNES context 1254936397dSBarry Smith 12628529972SSatish Balay Level: advanced 1274936397dSBarry Smith 1284936397dSBarry Smith .keywords: SNES, view 1294936397dSBarry Smith 130bf388a1fSBarry Smith .seealso: SNESCreate(), SNESSetFunction(), SNESFunction 1314936397dSBarry Smith @*/ 1327087cfbeSBarry Smith PetscErrorCode SNESSetFunctionDomainError(SNES snes) 1334936397dSBarry Smith { 1344936397dSBarry Smith PetscFunctionBegin; 1350700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 136422a814eSBarry Smith if (snes->errorifnotconverged) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"User code indicates input vector is not in the function domain"); 1374936397dSBarry Smith snes->domainerror = PETSC_TRUE; 1384936397dSBarry Smith PetscFunctionReturn(0); 1394936397dSBarry Smith } 1404936397dSBarry Smith 1416a388c36SPeter Brune /*@ 142c77b2880SPeter Brune SNESGetFunctionDomainError - Gets the status of the domain error after a call to SNESComputeFunction; 1436a388c36SPeter Brune 1446a388c36SPeter Brune Logically Collective on SNES 1456a388c36SPeter Brune 1466a388c36SPeter Brune Input Parameters: 1476a388c36SPeter Brune . snes - the SNES context 1486a388c36SPeter Brune 1496a388c36SPeter Brune Output Parameters: 150bf388a1fSBarry Smith . domainerror - Set to PETSC_TRUE if there's a domain error; PETSC_FALSE otherwise. 1516a388c36SPeter Brune 1526a388c36SPeter Brune Level: advanced 1536a388c36SPeter Brune 1546a388c36SPeter Brune .keywords: SNES, view 1556a388c36SPeter Brune 156bf388a1fSBarry Smith .seealso: SNESSetFunctionDomainError(), SNESComputeFunction() 1576a388c36SPeter Brune @*/ 1586a388c36SPeter Brune PetscErrorCode SNESGetFunctionDomainError(SNES snes, PetscBool *domainerror) 1596a388c36SPeter Brune { 1606a388c36SPeter Brune PetscFunctionBegin; 1616a388c36SPeter Brune PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 1626a388c36SPeter Brune PetscValidPointer(domainerror, 2); 1636a388c36SPeter Brune *domainerror = snes->domainerror; 1646a388c36SPeter Brune PetscFunctionReturn(0); 1656a388c36SPeter Brune } 1666a388c36SPeter Brune 16755849f57SBarry Smith /*@C 16855849f57SBarry Smith SNESLoad - Loads a SNES that has been stored in binary with SNESView(). 16955849f57SBarry Smith 17055849f57SBarry Smith Collective on PetscViewer 17155849f57SBarry Smith 17255849f57SBarry Smith Input Parameters: 17355849f57SBarry Smith + newdm - the newly loaded SNES, this needs to have been created with SNESCreate() or 17455849f57SBarry Smith some related function before a call to SNESLoad(). 17555849f57SBarry Smith - viewer - binary file viewer, obtained from PetscViewerBinaryOpen() 17655849f57SBarry Smith 17755849f57SBarry Smith Level: intermediate 17855849f57SBarry Smith 17955849f57SBarry Smith Notes: 18055849f57SBarry Smith The type is determined by the data in the file, any type set into the SNES before this call is ignored. 18155849f57SBarry Smith 18255849f57SBarry Smith Notes for advanced users: 18355849f57SBarry Smith Most users should not need to know the details of the binary storage 18455849f57SBarry Smith format, since SNESLoad() and TSView() completely hide these details. 18555849f57SBarry Smith But for anyone who's interested, the standard binary matrix storage 18655849f57SBarry Smith format is 18755849f57SBarry Smith .vb 18855849f57SBarry Smith has not yet been determined 18955849f57SBarry Smith .ve 19055849f57SBarry Smith 19155849f57SBarry Smith .seealso: PetscViewerBinaryOpen(), SNESView(), MatLoad(), VecLoad() 19255849f57SBarry Smith @*/ 1932d53ad75SBarry Smith PetscErrorCode SNESLoad(SNES snes, PetscViewer viewer) 19455849f57SBarry Smith { 19555849f57SBarry Smith PetscErrorCode ierr; 19655849f57SBarry Smith PetscBool isbinary; 197060da220SMatthew G. Knepley PetscInt classid; 19855849f57SBarry Smith char type[256]; 19955849f57SBarry Smith KSP ksp; 2002d53ad75SBarry Smith DM dm; 2012d53ad75SBarry Smith DMSNES dmsnes; 20255849f57SBarry Smith 20355849f57SBarry Smith PetscFunctionBegin; 2042d53ad75SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 20555849f57SBarry Smith PetscValidHeaderSpecific(viewer,PETSC_VIEWER_CLASSID,2); 20655849f57SBarry Smith ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERBINARY,&isbinary);CHKERRQ(ierr); 20755849f57SBarry Smith if (!isbinary) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONG,"Invalid viewer; open viewer with PetscViewerBinaryOpen()"); 20855849f57SBarry Smith 209060da220SMatthew G. Knepley ierr = PetscViewerBinaryRead(viewer,&classid,1,NULL,PETSC_INT);CHKERRQ(ierr); 210ce94432eSBarry Smith if (classid != SNES_FILE_CLASSID) SETERRQ(PetscObjectComm((PetscObject)snes),PETSC_ERR_ARG_WRONG,"Not SNES next in file"); 211060da220SMatthew G. Knepley ierr = PetscViewerBinaryRead(viewer,type,256,NULL,PETSC_CHAR);CHKERRQ(ierr); 2122d53ad75SBarry Smith ierr = SNESSetType(snes, type);CHKERRQ(ierr); 2132d53ad75SBarry Smith if (snes->ops->load) { 2142d53ad75SBarry Smith ierr = (*snes->ops->load)(snes,viewer);CHKERRQ(ierr); 215f2c2a1b9SBarry Smith } 2162d53ad75SBarry Smith ierr = SNESGetDM(snes,&dm);CHKERRQ(ierr); 2172d53ad75SBarry Smith ierr = DMGetDMSNES(dm,&dmsnes);CHKERRQ(ierr); 2182d53ad75SBarry Smith ierr = DMSNESLoad(dmsnes,viewer);CHKERRQ(ierr); 2192d53ad75SBarry Smith ierr = SNESGetKSP(snes,&ksp);CHKERRQ(ierr); 22055849f57SBarry Smith ierr = KSPLoad(ksp,viewer);CHKERRQ(ierr); 22155849f57SBarry Smith PetscFunctionReturn(0); 22255849f57SBarry Smith } 2236a388c36SPeter Brune 2249804daf3SBarry Smith #include <petscdraw.h> 225e04113cfSBarry Smith #if defined(PETSC_HAVE_SAWS) 226e04113cfSBarry Smith #include <petscviewersaws.h> 227bfb97211SBarry Smith #endif 2288404b7f3SBarry Smith 2297e2c5f70SBarry Smith /*@C 2309b94acceSBarry Smith SNESView - Prints the SNES data structure. 2319b94acceSBarry Smith 2324c49b128SBarry Smith Collective on SNES 233fee21e36SBarry Smith 234c7afd0dbSLois Curfman McInnes Input Parameters: 235c7afd0dbSLois Curfman McInnes + SNES - the SNES context 236c7afd0dbSLois Curfman McInnes - viewer - visualization context 237c7afd0dbSLois Curfman McInnes 2389b94acceSBarry Smith Options Database Key: 239c8a8ba5cSLois Curfman McInnes . -snes_view - Calls SNESView() at end of SNESSolve() 2409b94acceSBarry Smith 2419b94acceSBarry Smith Notes: 2429b94acceSBarry Smith The available visualization contexts include 243b0a32e0cSBarry Smith + PETSC_VIEWER_STDOUT_SELF - standard output (default) 244b0a32e0cSBarry Smith - PETSC_VIEWER_STDOUT_WORLD - synchronized standard 245c8a8ba5cSLois Curfman McInnes output where only the first processor opens 246c8a8ba5cSLois Curfman McInnes the file. All other processors send their 247c8a8ba5cSLois Curfman McInnes data to the first processor to print. 2489b94acceSBarry Smith 2493e081fefSLois Curfman McInnes The user can open an alternative visualization context with 250b0a32e0cSBarry Smith PetscViewerASCIIOpen() - output to a specified file. 2519b94acceSBarry Smith 25236851e7fSLois Curfman McInnes Level: beginner 25336851e7fSLois Curfman McInnes 2549b94acceSBarry Smith .keywords: SNES, view 2559b94acceSBarry Smith 256b0a32e0cSBarry Smith .seealso: PetscViewerASCIIOpen() 2579b94acceSBarry Smith @*/ 2587087cfbeSBarry Smith PetscErrorCode SNESView(SNES snes,PetscViewer viewer) 2599b94acceSBarry Smith { 260fa9f3622SBarry Smith SNESKSPEW *kctx; 261dfbe8321SBarry Smith PetscErrorCode ierr; 26294b7f48cSBarry Smith KSP ksp; 2637f1410a3SPeter Brune SNESLineSearch linesearch; 26472a02f06SBarry Smith PetscBool iascii,isstring,isbinary,isdraw; 2652d53ad75SBarry Smith DMSNES dmsnes; 266e04113cfSBarry Smith #if defined(PETSC_HAVE_SAWS) 267536b137fSBarry Smith PetscBool issaws; 268bfb97211SBarry Smith #endif 2699b94acceSBarry Smith 2703a40ed3dSBarry Smith PetscFunctionBegin; 2710700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 2723050cee2SBarry Smith if (!viewer) { 273ce94432eSBarry Smith ierr = PetscViewerASCIIGetStdout(PetscObjectComm((PetscObject)snes),&viewer);CHKERRQ(ierr); 2743050cee2SBarry Smith } 2750700a824SBarry Smith PetscValidHeaderSpecific(viewer,PETSC_VIEWER_CLASSID,2); 276c9780b6fSBarry Smith PetscCheckSameComm(snes,1,viewer,2); 27774679c65SBarry Smith 278251f4c67SDmitry Karpeev ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERASCII,&iascii);CHKERRQ(ierr); 279251f4c67SDmitry Karpeev ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERSTRING,&isstring);CHKERRQ(ierr); 28055849f57SBarry Smith ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERBINARY,&isbinary);CHKERRQ(ierr); 28172a02f06SBarry Smith ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERDRAW,&isdraw);CHKERRQ(ierr); 282e04113cfSBarry Smith #if defined(PETSC_HAVE_SAWS) 283536b137fSBarry Smith ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERSAWS,&issaws);CHKERRQ(ierr); 284bfb97211SBarry Smith #endif 28532077d6dSBarry Smith if (iascii) { 286dc0571f2SMatthew G. Knepley SNESNormSchedule normschedule; 2878404b7f3SBarry Smith DM dm; 2888404b7f3SBarry Smith PetscErrorCode (*cJ)(SNES,Vec,Mat,Mat,void*); 2898404b7f3SBarry Smith void *ctx; 290dc0571f2SMatthew G. Knepley 291dae58748SBarry Smith ierr = PetscObjectPrintClassNamePrefixType((PetscObject)snes,viewer);CHKERRQ(ierr); 292fce1e034SJed Brown if (!snes->setupcalled) { 293fce1e034SJed Brown ierr = PetscViewerASCIIPrintf(viewer," SNES has not been set up so information may be incomplete\n");CHKERRQ(ierr); 294fce1e034SJed Brown } 295e7788613SBarry Smith if (snes->ops->view) { 296b0a32e0cSBarry Smith ierr = PetscViewerASCIIPushTab(viewer);CHKERRQ(ierr); 297e7788613SBarry Smith ierr = (*snes->ops->view)(snes,viewer);CHKERRQ(ierr); 298b0a32e0cSBarry Smith ierr = PetscViewerASCIIPopTab(viewer);CHKERRQ(ierr); 2990ef38995SBarry Smith } 30077431f27SBarry Smith ierr = PetscViewerASCIIPrintf(viewer," maximum iterations=%D, maximum function evaluations=%D\n",snes->max_its,snes->max_funcs);CHKERRQ(ierr); 30157622a8eSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," tolerances: relative=%g, absolute=%g, solution=%g\n",(double)snes->rtol,(double)snes->abstol,(double)snes->stol);CHKERRQ(ierr); 302efd4aadfSBarry Smith if (snes->usesksp) { 30377431f27SBarry Smith ierr = PetscViewerASCIIPrintf(viewer," total number of linear solver iterations=%D\n",snes->linear_its);CHKERRQ(ierr); 304efd4aadfSBarry Smith } 30577431f27SBarry Smith ierr = PetscViewerASCIIPrintf(viewer," total number of function evaluations=%D\n",snes->nfuncs);CHKERRQ(ierr); 306dc0571f2SMatthew G. Knepley ierr = SNESGetNormSchedule(snes, &normschedule);CHKERRQ(ierr); 307dc0571f2SMatthew G. Knepley if (normschedule > 0) {ierr = PetscViewerASCIIPrintf(viewer," norm schedule %s\n",SNESNormSchedules[normschedule]);CHKERRQ(ierr);} 30817fe4bdfSPeter Brune if (snes->gridsequence) { 30917fe4bdfSPeter Brune ierr = PetscViewerASCIIPrintf(viewer," total number of grid sequence refinements=%D\n",snes->gridsequence);CHKERRQ(ierr); 31017fe4bdfSPeter Brune } 3119b94acceSBarry Smith if (snes->ksp_ewconv) { 312fa9f3622SBarry Smith kctx = (SNESKSPEW*)snes->kspconvctx; 3139b94acceSBarry Smith if (kctx) { 31477431f27SBarry Smith ierr = PetscViewerASCIIPrintf(viewer," Eisenstat-Walker computation of KSP relative tolerance (version %D)\n",kctx->version);CHKERRQ(ierr); 31557622a8eSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," rtol_0=%g, rtol_max=%g, threshold=%g\n",(double)kctx->rtol_0,(double)kctx->rtol_max,(double)kctx->threshold);CHKERRQ(ierr); 31657622a8eSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," gamma=%g, alpha=%g, alpha2=%g\n",(double)kctx->gamma,(double)kctx->alpha,(double)kctx->alpha2);CHKERRQ(ierr); 3179b94acceSBarry Smith } 3189b94acceSBarry Smith } 319eb1f6c34SBarry Smith if (snes->lagpreconditioner == -1) { 320eb1f6c34SBarry Smith ierr = PetscViewerASCIIPrintf(viewer," Preconditioned is never rebuilt\n");CHKERRQ(ierr); 321eb1f6c34SBarry Smith } else if (snes->lagpreconditioner > 1) { 322eb1f6c34SBarry Smith ierr = PetscViewerASCIIPrintf(viewer," Preconditioned is rebuilt every %D new Jacobians\n",snes->lagpreconditioner);CHKERRQ(ierr); 323eb1f6c34SBarry Smith } 324eb1f6c34SBarry Smith if (snes->lagjacobian == -1) { 325eb1f6c34SBarry Smith ierr = PetscViewerASCIIPrintf(viewer," Jacobian is never rebuilt\n");CHKERRQ(ierr); 326eb1f6c34SBarry Smith } else if (snes->lagjacobian > 1) { 32742f4f86dSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," Jacobian is rebuilt every %D SNES iterations\n",snes->lagjacobian);CHKERRQ(ierr); 328eb1f6c34SBarry Smith } 3298404b7f3SBarry Smith ierr = SNESGetDM(snes,&dm);CHKERRQ(ierr); 3308404b7f3SBarry Smith ierr = DMSNESGetJacobian(dm,&cJ,&ctx);CHKERRQ(ierr); 3318404b7f3SBarry Smith if (cJ == SNESComputeJacobianDefault) { 3328404b7f3SBarry Smith ierr = PetscViewerASCIIPrintf(viewer," Jacobian is built using finite differences one column at a time\n");CHKERRQ(ierr); 3338404b7f3SBarry Smith } else if (cJ == SNESComputeJacobianDefaultColor) { 3348404b7f3SBarry Smith ierr = PetscViewerASCIIPrintf(viewer," Jacobian is built using finite differences with coloring\n");CHKERRQ(ierr); 3358404b7f3SBarry Smith } 3360f5bd95cSBarry Smith } else if (isstring) { 337317d6ea6SBarry Smith const char *type; 338454a90a3SBarry Smith ierr = SNESGetType(snes,&type);CHKERRQ(ierr); 339b0a32e0cSBarry Smith ierr = PetscViewerStringSPrintf(viewer," %-3.3s",type);CHKERRQ(ierr); 34055849f57SBarry Smith } else if (isbinary) { 34155849f57SBarry Smith PetscInt classid = SNES_FILE_CLASSID; 34255849f57SBarry Smith MPI_Comm comm; 34355849f57SBarry Smith PetscMPIInt rank; 34455849f57SBarry Smith char type[256]; 34555849f57SBarry Smith 34655849f57SBarry Smith ierr = PetscObjectGetComm((PetscObject)snes,&comm);CHKERRQ(ierr); 34755849f57SBarry Smith ierr = MPI_Comm_rank(comm,&rank);CHKERRQ(ierr); 34855849f57SBarry Smith if (!rank) { 34955849f57SBarry Smith ierr = PetscViewerBinaryWrite(viewer,&classid,1,PETSC_INT,PETSC_FALSE);CHKERRQ(ierr); 35089d949e2SBarry Smith ierr = PetscStrncpy(type,((PetscObject)snes)->type_name,sizeof(type));CHKERRQ(ierr); 35189d949e2SBarry Smith ierr = PetscViewerBinaryWrite(viewer,type,sizeof(type),PETSC_CHAR,PETSC_FALSE);CHKERRQ(ierr); 35255849f57SBarry Smith } 35355849f57SBarry Smith if (snes->ops->view) { 35455849f57SBarry Smith ierr = (*snes->ops->view)(snes,viewer);CHKERRQ(ierr); 35555849f57SBarry Smith } 35672a02f06SBarry Smith } else if (isdraw) { 35772a02f06SBarry Smith PetscDraw draw; 35872a02f06SBarry Smith char str[36]; 35989fd9fafSBarry Smith PetscReal x,y,bottom,h; 36072a02f06SBarry Smith 36172a02f06SBarry Smith ierr = PetscViewerDrawGetDraw(viewer,0,&draw);CHKERRQ(ierr); 36272a02f06SBarry Smith ierr = PetscDrawGetCurrentPoint(draw,&x,&y);CHKERRQ(ierr); 36372a02f06SBarry Smith ierr = PetscStrcpy(str,"SNES: ");CHKERRQ(ierr); 36472a02f06SBarry Smith ierr = PetscStrcat(str,((PetscObject)snes)->type_name);CHKERRQ(ierr); 36551fa3d41SBarry Smith ierr = PetscDrawStringBoxed(draw,x,y,PETSC_DRAW_BLUE,PETSC_DRAW_BLACK,str,NULL,&h);CHKERRQ(ierr); 36689fd9fafSBarry Smith bottom = y - h; 36772a02f06SBarry Smith ierr = PetscDrawPushCurrentPoint(draw,x,bottom);CHKERRQ(ierr); 368c4646bacSPeter Brune if (snes->ops->view) { 369c4646bacSPeter Brune ierr = (*snes->ops->view)(snes,viewer);CHKERRQ(ierr); 370c4646bacSPeter Brune } 371e04113cfSBarry Smith #if defined(PETSC_HAVE_SAWS) 372536b137fSBarry Smith } else if (issaws) { 373d45a07a7SBarry Smith PetscMPIInt rank; 3742657e9d9SBarry Smith const char *name; 375d45a07a7SBarry Smith 3762657e9d9SBarry Smith ierr = PetscObjectGetName((PetscObject)snes,&name);CHKERRQ(ierr); 377d45a07a7SBarry Smith ierr = MPI_Comm_rank(PETSC_COMM_WORLD,&rank);CHKERRQ(ierr); 378d45a07a7SBarry Smith if (!((PetscObject)snes)->amsmem && !rank) { 379d45a07a7SBarry Smith char dir[1024]; 380d45a07a7SBarry Smith 381e04113cfSBarry Smith ierr = PetscObjectViewSAWs((PetscObject)snes,viewer);CHKERRQ(ierr); 3822657e9d9SBarry Smith ierr = PetscSNPrintf(dir,1024,"/PETSc/Objects/%s/its",name);CHKERRQ(ierr); 3832657e9d9SBarry Smith PetscStackCallSAWs(SAWs_Register,(dir,&snes->iter,1,SAWs_READ,SAWs_INT)); 384bfb97211SBarry Smith if (!snes->conv_hist) { 385a0931e03SBarry Smith ierr = SNESSetConvergenceHistory(snes,NULL,NULL,PETSC_DECIDE,PETSC_TRUE);CHKERRQ(ierr); 386bfb97211SBarry Smith } 3872657e9d9SBarry Smith ierr = PetscSNPrintf(dir,1024,"/PETSc/Objects/%s/conv_hist",name);CHKERRQ(ierr); 3882657e9d9SBarry Smith PetscStackCallSAWs(SAWs_Register,(dir,snes->conv_hist,10,SAWs_READ,SAWs_DOUBLE)); 389f05ece33SBarry Smith } 390bfb97211SBarry Smith #endif 39172a02f06SBarry Smith } 39272a02f06SBarry Smith if (snes->linesearch) { 39372a02f06SBarry Smith ierr = PetscViewerASCIIPushTab(viewer);CHKERRQ(ierr); 3947601faf0SJed Brown ierr = SNESGetLineSearch(snes, &linesearch);CHKERRQ(ierr); 39572a02f06SBarry Smith ierr = SNESLineSearchView(linesearch, viewer);CHKERRQ(ierr); 39672a02f06SBarry Smith ierr = PetscViewerASCIIPopTab(viewer);CHKERRQ(ierr); 39719bcc07fSBarry Smith } 398efd4aadfSBarry Smith if (snes->npc && snes->usesnpc) { 3994a0c5b0cSMatthew G Knepley ierr = PetscViewerASCIIPushTab(viewer);CHKERRQ(ierr); 400efd4aadfSBarry Smith ierr = SNESView(snes->npc, viewer);CHKERRQ(ierr); 4014a0c5b0cSMatthew G Knepley ierr = PetscViewerASCIIPopTab(viewer);CHKERRQ(ierr); 4024a0c5b0cSMatthew G Knepley } 4032d53ad75SBarry Smith ierr = PetscViewerASCIIPushTab(viewer);CHKERRQ(ierr); 4042d53ad75SBarry Smith ierr = DMGetDMSNES(snes->dm,&dmsnes);CHKERRQ(ierr); 4052d53ad75SBarry Smith ierr = DMSNESView(dmsnes, viewer);CHKERRQ(ierr); 4062d53ad75SBarry Smith ierr = PetscViewerASCIIPopTab(viewer);CHKERRQ(ierr); 4072c155ee1SBarry Smith if (snes->usesksp) { 4082c155ee1SBarry Smith ierr = SNESGetKSP(snes,&ksp);CHKERRQ(ierr); 409b0a32e0cSBarry Smith ierr = PetscViewerASCIIPushTab(viewer);CHKERRQ(ierr); 41094b7f48cSBarry Smith ierr = KSPView(ksp,viewer);CHKERRQ(ierr); 411b0a32e0cSBarry Smith ierr = PetscViewerASCIIPopTab(viewer);CHKERRQ(ierr); 4122c155ee1SBarry Smith } 41372a02f06SBarry Smith if (isdraw) { 41472a02f06SBarry Smith PetscDraw draw; 41572a02f06SBarry Smith ierr = PetscViewerDrawGetDraw(viewer,0,&draw);CHKERRQ(ierr); 41672a02f06SBarry Smith ierr = PetscDrawPopCurrentPoint(draw);CHKERRQ(ierr); 4177f1410a3SPeter Brune } 4183a40ed3dSBarry Smith PetscFunctionReturn(0); 4199b94acceSBarry Smith } 4209b94acceSBarry Smith 42176b2cf59SMatthew Knepley /* 42276b2cf59SMatthew Knepley We retain a list of functions that also take SNES command 42376b2cf59SMatthew Knepley line options. These are called at the end SNESSetFromOptions() 42476b2cf59SMatthew Knepley */ 42576b2cf59SMatthew Knepley #define MAXSETFROMOPTIONS 5 426a7cc72afSBarry Smith static PetscInt numberofsetfromoptions; 4276849ba73SBarry Smith static PetscErrorCode (*othersetfromoptions[MAXSETFROMOPTIONS])(SNES); 42876b2cf59SMatthew Knepley 429ac226902SBarry Smith /*@C 43076b2cf59SMatthew Knepley SNESAddOptionsChecker - Adds an additional function to check for SNES options. 43176b2cf59SMatthew Knepley 43276b2cf59SMatthew Knepley Not Collective 43376b2cf59SMatthew Knepley 43476b2cf59SMatthew Knepley Input Parameter: 43576b2cf59SMatthew Knepley . snescheck - function that checks for options 43676b2cf59SMatthew Knepley 43776b2cf59SMatthew Knepley Level: developer 43876b2cf59SMatthew Knepley 43976b2cf59SMatthew Knepley .seealso: SNESSetFromOptions() 44076b2cf59SMatthew Knepley @*/ 4417087cfbeSBarry Smith PetscErrorCode SNESAddOptionsChecker(PetscErrorCode (*snescheck)(SNES)) 44276b2cf59SMatthew Knepley { 44376b2cf59SMatthew Knepley PetscFunctionBegin; 444f23aa3ddSBarry Smith if (numberofsetfromoptions >= MAXSETFROMOPTIONS) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE, "Too many options checkers, only %D allowed", MAXSETFROMOPTIONS); 44576b2cf59SMatthew Knepley othersetfromoptions[numberofsetfromoptions++] = snescheck; 44676b2cf59SMatthew Knepley PetscFunctionReturn(0); 44776b2cf59SMatthew Knepley } 44876b2cf59SMatthew Knepley 4497087cfbeSBarry Smith extern PetscErrorCode SNESDefaultMatrixFreeCreate2(SNES,Vec,Mat*); 450aa3661deSLisandro Dalcin 451ace3abfcSBarry Smith static PetscErrorCode SNESSetUpMatrixFree_Private(SNES snes, PetscBool hasOperator, PetscInt version) 452aa3661deSLisandro Dalcin { 453aa3661deSLisandro Dalcin Mat J; 454aa3661deSLisandro Dalcin KSP ksp; 455aa3661deSLisandro Dalcin PC pc; 456ace3abfcSBarry Smith PetscBool match; 457aa3661deSLisandro Dalcin PetscErrorCode ierr; 458895c21f2SBarry Smith MatNullSpace nullsp; 459aa3661deSLisandro Dalcin 460aa3661deSLisandro Dalcin PetscFunctionBegin; 4610700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 462aa3661deSLisandro Dalcin 46398613b67SLisandro Dalcin if (!snes->vec_func && (snes->jacobian || snes->jacobian_pre)) { 46498613b67SLisandro Dalcin Mat A = snes->jacobian, B = snes->jacobian_pre; 4652a7a6963SBarry Smith ierr = MatCreateVecs(A ? A : B, NULL,&snes->vec_func);CHKERRQ(ierr); 46698613b67SLisandro Dalcin } 46798613b67SLisandro Dalcin 468aa3661deSLisandro Dalcin if (version == 1) { 469aa3661deSLisandro Dalcin ierr = MatCreateSNESMF(snes,&J);CHKERRQ(ierr); 47098613b67SLisandro Dalcin ierr = MatMFFDSetOptionsPrefix(J,((PetscObject)snes)->prefix);CHKERRQ(ierr); 4719c6ac3b3SBarry Smith ierr = MatSetFromOptions(J);CHKERRQ(ierr); 472aa3661deSLisandro Dalcin } else if (version == 2) { 473e32f2f54SBarry Smith if (!snes->vec_func) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"SNESSetFunction() must be called first"); 474570b7f6dSBarry Smith #if !defined(PETSC_USE_COMPLEX) && !defined(PETSC_USE_REAL_SINGLE) && !defined(PETSC_USE_REAL___FLOAT128) && !defined(PETSC_USE_REAL___FP16) 475aa3661deSLisandro Dalcin ierr = SNESDefaultMatrixFreeCreate2(snes,snes->vec_func,&J);CHKERRQ(ierr); 476aa3661deSLisandro Dalcin #else 477e32f2f54SBarry Smith SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP, "matrix-free operator rutines (version 2)"); 478aa3661deSLisandro Dalcin #endif 479a8248277SBarry Smith } else SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE, "matrix-free operator rutines, only version 1 and 2"); 480aa3661deSLisandro Dalcin 481895c21f2SBarry Smith /* attach any user provided null space that was on Amat to the newly created matrix free matrix */ 482895c21f2SBarry Smith if (snes->jacobian) { 483895c21f2SBarry Smith ierr = MatGetNullSpace(snes->jacobian,&nullsp);CHKERRQ(ierr); 484895c21f2SBarry Smith if (nullsp) { 485895c21f2SBarry Smith ierr = MatSetNullSpace(J,nullsp);CHKERRQ(ierr); 486895c21f2SBarry Smith } 487895c21f2SBarry Smith } 488895c21f2SBarry Smith 489aa3661deSLisandro Dalcin ierr = PetscInfo1(snes,"Setting default matrix-free operator routines (version %D)\n", version);CHKERRQ(ierr); 490d3462f78SMatthew Knepley if (hasOperator) { 4913232da50SPeter Brune 492aa3661deSLisandro Dalcin /* This version replaces the user provided Jacobian matrix with a 493aa3661deSLisandro Dalcin matrix-free version but still employs the user-provided preconditioner matrix. */ 494aa3661deSLisandro Dalcin ierr = SNESSetJacobian(snes,J,0,0,0);CHKERRQ(ierr); 495aa3661deSLisandro Dalcin } else { 496aa3661deSLisandro Dalcin /* This version replaces both the user-provided Jacobian and the user- 4973232da50SPeter Brune provided preconditioner Jacobian with the default matrix free version. */ 498efd4aadfSBarry Smith if ((snes->npcside== PC_LEFT) && snes->npc) { 499d728fb7dSPeter Brune if (!snes->jacobian){ierr = SNESSetJacobian(snes,J,0,0,0);CHKERRQ(ierr);} 500172a4300SPeter Brune } else { 50128a52e04SBarry Smith ierr = SNESSetJacobian(snes,J,J,MatMFFDComputeJacobian,0);CHKERRQ(ierr); 502172a4300SPeter Brune } 503aa3661deSLisandro Dalcin /* Force no preconditioner */ 504aa3661deSLisandro Dalcin ierr = SNESGetKSP(snes,&ksp);CHKERRQ(ierr); 505aa3661deSLisandro Dalcin ierr = KSPGetPC(ksp,&pc);CHKERRQ(ierr); 506251f4c67SDmitry Karpeev ierr = PetscObjectTypeCompare((PetscObject)pc,PCSHELL,&match);CHKERRQ(ierr); 507aa3661deSLisandro Dalcin if (!match) { 508aa3661deSLisandro Dalcin ierr = PetscInfo(snes,"Setting default matrix-free preconditioner routines\nThat is no preconditioner is being used\n");CHKERRQ(ierr); 509aa3661deSLisandro Dalcin ierr = PCSetType(pc,PCNONE);CHKERRQ(ierr); 510aa3661deSLisandro Dalcin } 511aa3661deSLisandro Dalcin } 5126bf464f9SBarry Smith ierr = MatDestroy(&J);CHKERRQ(ierr); 513aa3661deSLisandro Dalcin PetscFunctionReturn(0); 514aa3661deSLisandro Dalcin } 515aa3661deSLisandro Dalcin 516dfe15315SJed Brown static PetscErrorCode DMRestrictHook_SNESVecSol(DM dmfine,Mat Restrict,Vec Rscale,Mat Inject,DM dmcoarse,void *ctx) 517dfe15315SJed Brown { 518dfe15315SJed Brown SNES snes = (SNES)ctx; 519dfe15315SJed Brown PetscErrorCode ierr; 5200298fd71SBarry Smith Vec Xfine,Xfine_named = NULL,Xcoarse; 521dfe15315SJed Brown 522dfe15315SJed Brown PetscFunctionBegin; 52316ebb321SJed Brown if (PetscLogPrintInfo) { 52416ebb321SJed Brown PetscInt finelevel,coarselevel,fineclevel,coarseclevel; 52516ebb321SJed Brown ierr = DMGetRefineLevel(dmfine,&finelevel);CHKERRQ(ierr); 52616ebb321SJed Brown ierr = DMGetCoarsenLevel(dmfine,&fineclevel);CHKERRQ(ierr); 52716ebb321SJed Brown ierr = DMGetRefineLevel(dmcoarse,&coarselevel);CHKERRQ(ierr); 52816ebb321SJed Brown ierr = DMGetCoarsenLevel(dmcoarse,&coarseclevel);CHKERRQ(ierr); 52916ebb321SJed Brown ierr = PetscInfo4(dmfine,"Restricting SNES solution vector from level %D-%D to level %D-%D\n",finelevel,fineclevel,coarselevel,coarseclevel);CHKERRQ(ierr); 53016ebb321SJed Brown } 531dfe15315SJed Brown if (dmfine == snes->dm) Xfine = snes->vec_sol; 532dfe15315SJed Brown else { 533dfe15315SJed Brown ierr = DMGetNamedGlobalVector(dmfine,"SNESVecSol",&Xfine_named);CHKERRQ(ierr); 534dfe15315SJed Brown Xfine = Xfine_named; 535dfe15315SJed Brown } 536dfe15315SJed Brown ierr = DMGetNamedGlobalVector(dmcoarse,"SNESVecSol",&Xcoarse);CHKERRQ(ierr); 537907f5c5aSLawrence Mitchell if (Inject) { 538907f5c5aSLawrence Mitchell ierr = MatRestrict(Inject,Xfine,Xcoarse);CHKERRQ(ierr); 539907f5c5aSLawrence Mitchell } else { 540dfe15315SJed Brown ierr = MatRestrict(Restrict,Xfine,Xcoarse);CHKERRQ(ierr); 541dfe15315SJed Brown ierr = VecPointwiseMult(Xcoarse,Xcoarse,Rscale);CHKERRQ(ierr); 542907f5c5aSLawrence Mitchell } 543dfe15315SJed Brown ierr = DMRestoreNamedGlobalVector(dmcoarse,"SNESVecSol",&Xcoarse);CHKERRQ(ierr); 544dfe15315SJed Brown if (Xfine_named) {ierr = DMRestoreNamedGlobalVector(dmfine,"SNESVecSol",&Xfine_named);CHKERRQ(ierr);} 545dfe15315SJed Brown PetscFunctionReturn(0); 546dfe15315SJed Brown } 547dfe15315SJed Brown 54816ebb321SJed Brown static PetscErrorCode DMCoarsenHook_SNESVecSol(DM dm,DM dmc,void *ctx) 54916ebb321SJed Brown { 55016ebb321SJed Brown PetscErrorCode ierr; 55116ebb321SJed Brown 55216ebb321SJed Brown PetscFunctionBegin; 55316ebb321SJed Brown ierr = DMCoarsenHookAdd(dmc,DMCoarsenHook_SNESVecSol,DMRestrictHook_SNESVecSol,ctx);CHKERRQ(ierr); 55416ebb321SJed Brown PetscFunctionReturn(0); 55516ebb321SJed Brown } 55616ebb321SJed Brown 557a6950cb2SJed Brown /* This may be called to rediscretize the operator on levels of linear multigrid. The DM shuffle is so the user can 558a6950cb2SJed Brown * safely call SNESGetDM() in their residual evaluation routine. */ 55923ee1639SBarry Smith static PetscErrorCode KSPComputeOperators_SNES(KSP ksp,Mat A,Mat B,void *ctx) 560caa4e7f2SJed Brown { 561caa4e7f2SJed Brown SNES snes = (SNES)ctx; 562caa4e7f2SJed Brown PetscErrorCode ierr; 563caa4e7f2SJed Brown Mat Asave = A,Bsave = B; 5640298fd71SBarry Smith Vec X,Xnamed = NULL; 565dfe15315SJed Brown DM dmsave; 5664e269d77SPeter Brune void *ctxsave; 567d1e9a80fSBarry Smith PetscErrorCode (*jac)(SNES,Vec,Mat,Mat,void*); 568caa4e7f2SJed Brown 569caa4e7f2SJed Brown PetscFunctionBegin; 570dfe15315SJed Brown dmsave = snes->dm; 571dfe15315SJed Brown ierr = KSPGetDM(ksp,&snes->dm);CHKERRQ(ierr); 572dfe15315SJed Brown if (dmsave == snes->dm) X = snes->vec_sol; /* We are on the finest level */ 573dfe15315SJed Brown else { /* We are on a coarser level, this vec was initialized using a DM restrict hook */ 574dfe15315SJed Brown ierr = DMGetNamedGlobalVector(snes->dm,"SNESVecSol",&Xnamed);CHKERRQ(ierr); 575dfe15315SJed Brown X = Xnamed; 5760298fd71SBarry Smith ierr = SNESGetJacobian(snes,NULL,NULL,&jac,&ctxsave);CHKERRQ(ierr); 5774e269d77SPeter Brune /* If the DM's don't match up, the MatFDColoring context needed for the jacobian won't match up either -- fixit. */ 5788d359177SBarry Smith if (jac == SNESComputeJacobianDefaultColor) { 5798d359177SBarry Smith ierr = SNESSetJacobian(snes,NULL,NULL,SNESComputeJacobianDefaultColor,0);CHKERRQ(ierr); 580dfe15315SJed Brown } 5814e269d77SPeter Brune } 5824e269d77SPeter Brune /* put the previous context back */ 5834e269d77SPeter Brune 584d1e9a80fSBarry Smith ierr = SNESComputeJacobian(snes,X,A,B);CHKERRQ(ierr); 5858d359177SBarry Smith if (snes->dm != dmsave && jac == SNESComputeJacobianDefaultColor) { 5860298fd71SBarry Smith ierr = SNESSetJacobian(snes,NULL,NULL,jac,ctxsave);CHKERRQ(ierr); 5874e269d77SPeter Brune } 5884e269d77SPeter Brune 589ce94432eSBarry Smith if (A != Asave || B != Bsave) SETERRQ(PetscObjectComm((PetscObject)snes),PETSC_ERR_SUP,"No support for changing matrices at this time"); 590dfe15315SJed Brown if (Xnamed) { 591dfe15315SJed Brown ierr = DMRestoreNamedGlobalVector(snes->dm,"SNESVecSol",&Xnamed);CHKERRQ(ierr); 592dfe15315SJed Brown } 593dfe15315SJed Brown snes->dm = dmsave; 594caa4e7f2SJed Brown PetscFunctionReturn(0); 595caa4e7f2SJed Brown } 596caa4e7f2SJed Brown 5976cab3a1bSJed Brown /*@ 5986cab3a1bSJed Brown SNESSetUpMatrices - ensures that matrices are available for SNES, to be called by SNESSetUp_XXX() 5996cab3a1bSJed Brown 6006cab3a1bSJed Brown Collective 6016cab3a1bSJed Brown 6026cab3a1bSJed Brown Input Arguments: 6036cab3a1bSJed Brown . snes - snes to configure 6046cab3a1bSJed Brown 6056cab3a1bSJed Brown Level: developer 6066cab3a1bSJed Brown 6076cab3a1bSJed Brown .seealso: SNESSetUp() 6086cab3a1bSJed Brown @*/ 6096cab3a1bSJed Brown PetscErrorCode SNESSetUpMatrices(SNES snes) 6106cab3a1bSJed Brown { 6116cab3a1bSJed Brown PetscErrorCode ierr; 6126cab3a1bSJed Brown DM dm; 613942e3340SBarry Smith DMSNES sdm; 6146cab3a1bSJed Brown 6156cab3a1bSJed Brown PetscFunctionBegin; 6166cab3a1bSJed Brown ierr = SNESGetDM(snes,&dm);CHKERRQ(ierr); 617942e3340SBarry Smith ierr = DMGetDMSNES(dm,&sdm);CHKERRQ(ierr); 618ce94432eSBarry Smith if (!sdm->ops->computejacobian) SETERRQ(PetscObjectComm((PetscObject)snes),PETSC_ERR_PLIB,"DMSNES not properly configured"); 619f5af7f23SKarl Rupp else if (!snes->jacobian && snes->mf) { 6206cab3a1bSJed Brown Mat J; 6216cab3a1bSJed Brown void *functx; 6226cab3a1bSJed Brown ierr = MatCreateSNESMF(snes,&J);CHKERRQ(ierr); 6236cab3a1bSJed Brown ierr = MatMFFDSetOptionsPrefix(J,((PetscObject)snes)->prefix);CHKERRQ(ierr); 6246cab3a1bSJed Brown ierr = MatSetFromOptions(J);CHKERRQ(ierr); 6250298fd71SBarry Smith ierr = SNESGetFunction(snes,NULL,NULL,&functx);CHKERRQ(ierr); 6263232da50SPeter Brune ierr = SNESSetJacobian(snes,J,J,0,0);CHKERRQ(ierr); 6276cab3a1bSJed Brown ierr = MatDestroy(&J);CHKERRQ(ierr); 628caa4e7f2SJed Brown } else if (snes->mf_operator && !snes->jacobian_pre && !snes->jacobian) { 6296cab3a1bSJed Brown Mat J,B; 6306cab3a1bSJed Brown ierr = MatCreateSNESMF(snes,&J);CHKERRQ(ierr); 6316cab3a1bSJed Brown ierr = MatMFFDSetOptionsPrefix(J,((PetscObject)snes)->prefix);CHKERRQ(ierr); 6326cab3a1bSJed Brown ierr = MatSetFromOptions(J);CHKERRQ(ierr); 633b412c318SBarry Smith ierr = DMCreateMatrix(snes->dm,&B);CHKERRQ(ierr); 63406f20277SJed Brown /* sdm->computejacobian was already set to reach here */ 6350298fd71SBarry Smith ierr = SNESSetJacobian(snes,J,B,NULL,NULL);CHKERRQ(ierr); 6366cab3a1bSJed Brown ierr = MatDestroy(&J);CHKERRQ(ierr); 6376cab3a1bSJed Brown ierr = MatDestroy(&B);CHKERRQ(ierr); 638caa4e7f2SJed Brown } else if (!snes->jacobian_pre) { 6391ba9b98eSMatthew G. Knepley PetscDS prob; 6406cab3a1bSJed Brown Mat J, B; 6411ba9b98eSMatthew G. Knepley PetscBool hasPrec = PETSC_FALSE; 6421ba9b98eSMatthew G. Knepley 6436cab3a1bSJed Brown J = snes->jacobian; 6441ba9b98eSMatthew G. Knepley ierr = DMGetDS(dm, &prob);CHKERRQ(ierr); 6451ba9b98eSMatthew G. Knepley if (prob) {ierr = PetscDSHasJacobianPreconditioner(prob, &hasPrec);CHKERRQ(ierr);} 646ec9a985fSMatthew G. Knepley if (J) {ierr = PetscObjectReference((PetscObject) J);CHKERRQ(ierr);} 647ec9a985fSMatthew G. Knepley else if (hasPrec) {ierr = DMCreateMatrix(snes->dm, &J);CHKERRQ(ierr);} 648b412c318SBarry Smith ierr = DMCreateMatrix(snes->dm, &B);CHKERRQ(ierr); 6490298fd71SBarry Smith ierr = SNESSetJacobian(snes, J ? J : B, B, NULL, NULL);CHKERRQ(ierr); 6501ba9b98eSMatthew G. Knepley ierr = MatDestroy(&J);CHKERRQ(ierr); 6516cab3a1bSJed Brown ierr = MatDestroy(&B);CHKERRQ(ierr); 6526cab3a1bSJed Brown } 653caa4e7f2SJed Brown { 654caa4e7f2SJed Brown KSP ksp; 655caa4e7f2SJed Brown ierr = SNESGetKSP(snes,&ksp);CHKERRQ(ierr); 656caa4e7f2SJed Brown ierr = KSPSetComputeOperators(ksp,KSPComputeOperators_SNES,snes);CHKERRQ(ierr); 65716ebb321SJed Brown ierr = DMCoarsenHookAdd(snes->dm,DMCoarsenHook_SNESVecSol,DMRestrictHook_SNESVecSol,snes);CHKERRQ(ierr); 658caa4e7f2SJed Brown } 6596cab3a1bSJed Brown PetscFunctionReturn(0); 6606cab3a1bSJed Brown } 6616cab3a1bSJed Brown 662fde5950dSBarry Smith /*@C 663fde5950dSBarry Smith SNESMonitorSetFromOptions - Sets a monitor function and viewer appropriate for the type indicated by the user 664fde5950dSBarry Smith 665fde5950dSBarry Smith Collective on SNES 666fde5950dSBarry Smith 667fde5950dSBarry Smith Input Parameters: 668fde5950dSBarry Smith + snes - SNES object you wish to monitor 669fde5950dSBarry Smith . name - the monitor type one is seeking 670fde5950dSBarry Smith . help - message indicating what monitoring is done 671fde5950dSBarry Smith . manual - manual page for the monitor 672fde5950dSBarry Smith . monitor - the monitor function 673fde5950dSBarry Smith - monitorsetup - a function that is called once ONLY if the user selected this monitor that may set additional features of the SNES or PetscViewer objects 674fde5950dSBarry Smith 675fde5950dSBarry Smith Level: developer 676fde5950dSBarry Smith 677fde5950dSBarry Smith .seealso: PetscOptionsGetViewer(), PetscOptionsGetReal(), PetscOptionsHasName(), PetscOptionsGetString(), 678fde5950dSBarry Smith PetscOptionsGetIntArray(), PetscOptionsGetRealArray(), PetscOptionsBool() 679fde5950dSBarry Smith PetscOptionsInt(), PetscOptionsString(), PetscOptionsReal(), PetscOptionsBool(), 680fde5950dSBarry Smith PetscOptionsName(), PetscOptionsBegin(), PetscOptionsEnd(), PetscOptionsHead(), 681fde5950dSBarry Smith PetscOptionsStringArray(),PetscOptionsRealArray(), PetscOptionsScalar(), 682fde5950dSBarry Smith PetscOptionsBoolGroupBegin(), PetscOptionsBoolGroup(), PetscOptionsBoolGroupEnd(), 683fde5950dSBarry Smith PetscOptionsFList(), PetscOptionsEList() 684fde5950dSBarry Smith @*/ 685d43b4f6eSBarry Smith PetscErrorCode SNESMonitorSetFromOptions(SNES snes,const char name[],const char help[], const char manual[],PetscErrorCode (*monitor)(SNES,PetscInt,PetscReal,PetscViewerAndFormat*),PetscErrorCode (*monitorsetup)(SNES,PetscViewerAndFormat*)) 686fde5950dSBarry Smith { 687fde5950dSBarry Smith PetscErrorCode ierr; 688fde5950dSBarry Smith PetscViewer viewer; 689fde5950dSBarry Smith PetscViewerFormat format; 690fde5950dSBarry Smith PetscBool flg; 691fde5950dSBarry Smith 692fde5950dSBarry Smith PetscFunctionBegin; 693fde5950dSBarry Smith ierr = PetscOptionsGetViewer(PetscObjectComm((PetscObject)snes),((PetscObject)snes)->prefix,name,&viewer,&format,&flg);CHKERRQ(ierr); 694fde5950dSBarry Smith if (flg) { 695d43b4f6eSBarry Smith PetscViewerAndFormat *vf; 696d43b4f6eSBarry Smith ierr = PetscViewerAndFormatCreate(viewer,format,&vf);CHKERRQ(ierr); 697d43b4f6eSBarry Smith ierr = PetscObjectDereference((PetscObject)viewer);CHKERRQ(ierr); 698fde5950dSBarry Smith if (monitorsetup) { 699d43b4f6eSBarry Smith ierr = (*monitorsetup)(snes,vf);CHKERRQ(ierr); 700fde5950dSBarry Smith } 701d43b4f6eSBarry Smith ierr = SNESMonitorSet(snes,(PetscErrorCode (*)(SNES,PetscInt,PetscReal,void*))monitor,vf,(PetscErrorCode (*)(void**))PetscViewerAndFormatDestroy);CHKERRQ(ierr); 702fde5950dSBarry Smith } 703fde5950dSBarry Smith PetscFunctionReturn(0); 704fde5950dSBarry Smith } 705fde5950dSBarry Smith 7069b94acceSBarry Smith /*@ 70794b7f48cSBarry Smith SNESSetFromOptions - Sets various SNES and KSP parameters from user options. 7089b94acceSBarry Smith 709c7afd0dbSLois Curfman McInnes Collective on SNES 710c7afd0dbSLois Curfman McInnes 7119b94acceSBarry Smith Input Parameter: 7129b94acceSBarry Smith . snes - the SNES context 7139b94acceSBarry Smith 71436851e7fSLois Curfman McInnes Options Database Keys: 715722329fbSBarry Smith + -snes_type <type> - newtonls, newtontr, ngmres, ncg, nrichardson, qn, vi, fas, SNESType for complete list 71682738288SBarry Smith . -snes_stol - convergence tolerance in terms of the norm 71782738288SBarry Smith of the change in the solution between steps 71870441072SBarry Smith . -snes_atol <abstol> - absolute tolerance of residual norm 719b39c3a46SLois Curfman McInnes . -snes_rtol <rtol> - relative decrease in tolerance norm from initial 720e4d06f11SPatrick Farrell . -snes_divergence_tolerance <divtol> - if the residual goes above divtol*rnorm0, exit with divergence 721be5caee7SBarry Smith . -snes_force_iteration <force> - force SNESSolve() to take at least one iteration 722b39c3a46SLois Curfman McInnes . -snes_max_it <max_it> - maximum number of iterations 723b39c3a46SLois Curfman McInnes . -snes_max_funcs <max_funcs> - maximum number of function evaluations 7244839bfe8SBarry Smith . -snes_max_fail <max_fail> - maximum number of line search failures allowed before stopping, default is none 725ddf469c8SBarry Smith . -snes_max_linear_solve_fail - number of linear solver failures before SNESSolve() stops 726a8054027SBarry Smith . -snes_lag_preconditioner <lag> - how often preconditioner is rebuilt (use -1 to never rebuild) 727e35cf81dSBarry Smith . -snes_lag_jacobian <lag> - how often Jacobian is rebuilt (use -1 to never rebuild) 728b39c3a46SLois Curfman McInnes . -snes_trtol <trtol> - trust region tolerance 7292492ecdbSBarry Smith . -snes_no_convergence_test - skip convergence test in nonlinear 73082738288SBarry Smith solver; hence iterations will continue until max_it 7311fbbfb26SLois Curfman McInnes or some other criterion is reached. Saves expense 73282738288SBarry Smith of convergence test 733fde5950dSBarry Smith . -snes_monitor [ascii][:filename][:viewer format] - prints residual norm at each iteration. if no filename given prints to stdout 734fde5950dSBarry Smith . -snes_monitor_solution [ascii binary draw][:filename][:viewer format] - plots solution at each iteration 735fde5950dSBarry Smith . -snes_monitor_residual [ascii binary draw][:filename][:viewer format] - plots residual (not its norm) at each iteration 736fde5950dSBarry Smith . -snes_monitor_solution_update [ascii binary draw][:filename][:viewer format] - plots update to solution at each iteration 7374619e776SBarry Smith . -snes_monitor_lg_residualnorm - plots residual norm at each iteration 738459f5d12SBarry Smith . -snes_monitor_lg_range - plots residual norm at each iteration 739e24b481bSBarry Smith . -snes_fd - use finite differences to compute Jacobian; very slow, only for testing 740e2e60de9SPeter Brune . -snes_fd_color - use finite differences with coloring to compute Jacobian 7415968eb51SBarry Smith . -snes_mf_ksp_monitor - if using matrix-free multiply then print h at each KSP iteration 742fee2055bSBarry Smith - -snes_converged_reason - print the reason for convergence/divergence after each solve 74382738288SBarry Smith 74482738288SBarry Smith Options Database for Eisenstat-Walker method: 745fa9f3622SBarry Smith + -snes_ksp_ew - use Eisenstat-Walker method for determining linear system convergence 7464b27c08aSLois Curfman McInnes . -snes_ksp_ew_version ver - version of Eisenstat-Walker method 74736851e7fSLois Curfman McInnes . -snes_ksp_ew_rtol0 <rtol0> - Sets rtol0 74836851e7fSLois Curfman McInnes . -snes_ksp_ew_rtolmax <rtolmax> - Sets rtolmax 74936851e7fSLois Curfman McInnes . -snes_ksp_ew_gamma <gamma> - Sets gamma 75036851e7fSLois Curfman McInnes . -snes_ksp_ew_alpha <alpha> - Sets alpha 75136851e7fSLois Curfman McInnes . -snes_ksp_ew_alpha2 <alpha2> - Sets alpha2 75236851e7fSLois Curfman McInnes - -snes_ksp_ew_threshold <threshold> - Sets threshold 75382738288SBarry Smith 75411ca99fdSLois Curfman McInnes Notes: 75511ca99fdSLois Curfman McInnes To see all options, run your program with the -help option or consult 756a7f22e61SSatish Balay Users-Manual: ch_snes 75783e2fdc7SBarry Smith 75836851e7fSLois Curfman McInnes Level: beginner 75936851e7fSLois Curfman McInnes 7609b94acceSBarry Smith .keywords: SNES, nonlinear, set, options, database 7619b94acceSBarry Smith 76269ed319cSSatish Balay .seealso: SNESSetOptionsPrefix() 7639b94acceSBarry Smith @*/ 7647087cfbeSBarry Smith PetscErrorCode SNESSetFromOptions(SNES snes) 7659b94acceSBarry Smith { 7668afaa268SBarry Smith PetscBool flg,pcset,persist,set; 767d8f46077SPeter Brune PetscInt i,indx,lag,grids; 76804d7464bSBarry Smith const char *deft = SNESNEWTONLS; 76985385478SLisandro Dalcin const char *convtests[] = {"default","skip"}; 77085385478SLisandro Dalcin SNESKSPEW *kctx = NULL; 771e8105e01SRichard Katz char type[256], monfilename[PETSC_MAX_PATH_LEN]; 77285385478SLisandro Dalcin PetscErrorCode ierr; 773c40d0f55SPeter Brune PCSide pcside; 774a64e098fSPeter Brune const char *optionsprefix; 7759b94acceSBarry Smith 7763a40ed3dSBarry Smith PetscFunctionBegin; 7770700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 7780f51fdf8SToby Isaac ierr = SNESRegisterAll();CHKERRQ(ierr); 7793194b578SJed Brown ierr = PetscObjectOptionsBegin((PetscObject)snes);CHKERRQ(ierr); 780639ff905SBarry Smith if (((PetscObject)snes)->type_name) deft = ((PetscObject)snes)->type_name; 781a264d7a6SBarry Smith ierr = PetscOptionsFList("-snes_type","Nonlinear solver method","SNESSetType",SNESList,deft,type,256,&flg);CHKERRQ(ierr); 782d64ed03dSBarry Smith if (flg) { 783186905e3SBarry Smith ierr = SNESSetType(snes,type);CHKERRQ(ierr); 7847adad957SLisandro Dalcin } else if (!((PetscObject)snes)->type_name) { 785186905e3SBarry Smith ierr = SNESSetType(snes,deft);CHKERRQ(ierr); 786d64ed03dSBarry Smith } 78794ae4db5SBarry Smith ierr = PetscOptionsReal("-snes_stol","Stop if step length less than","SNESSetTolerances",snes->stol,&snes->stol,NULL);CHKERRQ(ierr); 78894ae4db5SBarry Smith ierr = PetscOptionsReal("-snes_atol","Stop if function norm less than","SNESSetTolerances",snes->abstol,&snes->abstol,NULL);CHKERRQ(ierr); 789186905e3SBarry Smith 79094ae4db5SBarry Smith ierr = PetscOptionsReal("-snes_rtol","Stop if decrease in function norm less than","SNESSetTolerances",snes->rtol,&snes->rtol,NULL);CHKERRQ(ierr); 791e4d06f11SPatrick Farrell ierr = PetscOptionsReal("-snes_divergence_tolerance","Stop if residual norm increases by this factor","SNESSetDivergenceTolerance",snes->divtol,&snes->divtol,NULL);CHKERRQ(ierr); 7920298fd71SBarry Smith ierr = PetscOptionsInt("-snes_max_it","Maximum iterations","SNESSetTolerances",snes->max_its,&snes->max_its,NULL);CHKERRQ(ierr); 7930298fd71SBarry Smith ierr = PetscOptionsInt("-snes_max_funcs","Maximum function evaluations","SNESSetTolerances",snes->max_funcs,&snes->max_funcs,NULL);CHKERRQ(ierr); 7940298fd71SBarry Smith ierr = PetscOptionsInt("-snes_max_fail","Maximum nonlinear step failures","SNESSetMaxNonlinearStepFailures",snes->maxFailures,&snes->maxFailures,NULL);CHKERRQ(ierr); 7950298fd71SBarry Smith ierr = PetscOptionsInt("-snes_max_linear_solve_fail","Maximum failures in linear solves allowed","SNESSetMaxLinearSolveFailures",snes->maxLinearSolveFailures,&snes->maxLinearSolveFailures,NULL);CHKERRQ(ierr); 7960298fd71SBarry Smith ierr = PetscOptionsBool("-snes_error_if_not_converged","Generate error if solver does not converge","SNESSetErrorIfNotConverged",snes->errorifnotconverged,&snes->errorifnotconverged,NULL);CHKERRQ(ierr); 797be5caee7SBarry Smith ierr = PetscOptionsBool("-snes_force_iteration","Force SNESSolve() to take at least one iteration","SNESForceIteration",snes->forceiteration,&snes->forceiteration,NULL);CHKERRQ(ierr); 79885385478SLisandro Dalcin 799a8054027SBarry Smith ierr = PetscOptionsInt("-snes_lag_preconditioner","How often to rebuild preconditioner","SNESSetLagPreconditioner",snes->lagpreconditioner,&lag,&flg);CHKERRQ(ierr); 800a8054027SBarry Smith if (flg) { 801a8054027SBarry Smith ierr = SNESSetLagPreconditioner(snes,lag);CHKERRQ(ierr); 802a8054027SBarry Smith } 80337ec4e1aSPeter Brune ierr = PetscOptionsBool("-snes_lag_preconditioner_persists","Preconditioner lagging through multiple solves","SNESSetLagPreconditionerPersists",snes->lagjac_persist,&persist,&flg);CHKERRQ(ierr); 80437ec4e1aSPeter Brune if (flg) { 80537ec4e1aSPeter Brune ierr = SNESSetLagPreconditionerPersists(snes,persist);CHKERRQ(ierr); 80637ec4e1aSPeter Brune } 807e35cf81dSBarry Smith ierr = PetscOptionsInt("-snes_lag_jacobian","How often to rebuild Jacobian","SNESSetLagJacobian",snes->lagjacobian,&lag,&flg);CHKERRQ(ierr); 808e35cf81dSBarry Smith if (flg) { 809e35cf81dSBarry Smith ierr = SNESSetLagJacobian(snes,lag);CHKERRQ(ierr); 810e35cf81dSBarry Smith } 81137ec4e1aSPeter Brune ierr = PetscOptionsBool("-snes_lag_jacobian_persists","Jacobian lagging through multiple solves","SNESSetLagJacobianPersists",snes->lagjac_persist,&persist,&flg);CHKERRQ(ierr); 81237ec4e1aSPeter Brune if (flg) { 81337ec4e1aSPeter Brune ierr = SNESSetLagJacobianPersists(snes,persist);CHKERRQ(ierr); 81437ec4e1aSPeter Brune } 81537ec4e1aSPeter Brune 816efd51863SBarry Smith ierr = PetscOptionsInt("-snes_grid_sequence","Use grid sequencing to generate initial guess","SNESSetGridSequence",snes->gridsequence,&grids,&flg);CHKERRQ(ierr); 817efd51863SBarry Smith if (flg) { 818efd51863SBarry Smith ierr = SNESSetGridSequence(snes,grids);CHKERRQ(ierr); 819efd51863SBarry Smith } 820a8054027SBarry Smith 82185385478SLisandro Dalcin ierr = PetscOptionsEList("-snes_convergence_test","Convergence test","SNESSetConvergenceTest",convtests,2,"default",&indx,&flg);CHKERRQ(ierr); 82285385478SLisandro Dalcin if (flg) { 82385385478SLisandro Dalcin switch (indx) { 8248d359177SBarry Smith case 0: ierr = SNESSetConvergenceTest(snes,SNESConvergedDefault,NULL,NULL);CHKERRQ(ierr); break; 825e2a6519dSDmitry Karpeev case 1: ierr = SNESSetConvergenceTest(snes,SNESConvergedSkip,NULL,NULL);CHKERRQ(ierr); break; 82685385478SLisandro Dalcin } 82785385478SLisandro Dalcin } 82885385478SLisandro Dalcin 829365a6726SPeter Brune ierr = PetscOptionsEList("-snes_norm_schedule","SNES Norm schedule","SNESSetNormSchedule",SNESNormSchedules,5,"function",&indx,&flg);CHKERRQ(ierr); 830365a6726SPeter Brune if (flg) { ierr = SNESSetNormSchedule(snes,(SNESNormSchedule)indx);CHKERRQ(ierr); } 831fdacfa88SPeter Brune 83247073ea2SPeter Brune ierr = PetscOptionsEList("-snes_function_type","SNES Norm schedule","SNESSetFunctionType",SNESFunctionTypes,2,"unpreconditioned",&indx,&flg);CHKERRQ(ierr); 83347073ea2SPeter Brune if (flg) { ierr = SNESSetFunctionType(snes,(SNESFunctionType)indx);CHKERRQ(ierr); } 834186905e3SBarry Smith 83585385478SLisandro Dalcin kctx = (SNESKSPEW*)snes->kspconvctx; 83685385478SLisandro Dalcin 8370298fd71SBarry Smith ierr = PetscOptionsBool("-snes_ksp_ew","Use Eisentat-Walker linear system convergence test","SNESKSPSetUseEW",snes->ksp_ewconv,&snes->ksp_ewconv,NULL);CHKERRQ(ierr); 838186905e3SBarry Smith 83994ae4db5SBarry Smith ierr = PetscOptionsInt("-snes_ksp_ew_version","Version 1, 2 or 3","SNESKSPSetParametersEW",kctx->version,&kctx->version,NULL);CHKERRQ(ierr); 84094ae4db5SBarry Smith ierr = PetscOptionsReal("-snes_ksp_ew_rtol0","0 <= rtol0 < 1","SNESKSPSetParametersEW",kctx->rtol_0,&kctx->rtol_0,NULL);CHKERRQ(ierr); 84194ae4db5SBarry Smith ierr = PetscOptionsReal("-snes_ksp_ew_rtolmax","0 <= rtolmax < 1","SNESKSPSetParametersEW",kctx->rtol_max,&kctx->rtol_max,NULL);CHKERRQ(ierr); 84294ae4db5SBarry Smith ierr = PetscOptionsReal("-snes_ksp_ew_gamma","0 <= gamma <= 1","SNESKSPSetParametersEW",kctx->gamma,&kctx->gamma,NULL);CHKERRQ(ierr); 84394ae4db5SBarry Smith ierr = PetscOptionsReal("-snes_ksp_ew_alpha","1 < alpha <= 2","SNESKSPSetParametersEW",kctx->alpha,&kctx->alpha,NULL);CHKERRQ(ierr); 84494ae4db5SBarry Smith ierr = PetscOptionsReal("-snes_ksp_ew_alpha2","alpha2","SNESKSPSetParametersEW",kctx->alpha2,&kctx->alpha2,NULL);CHKERRQ(ierr); 84594ae4db5SBarry Smith ierr = PetscOptionsReal("-snes_ksp_ew_threshold","0 < threshold < 1","SNESKSPSetParametersEW",kctx->threshold,&kctx->threshold,NULL);CHKERRQ(ierr); 846186905e3SBarry Smith 84790d69ab7SBarry Smith flg = PETSC_FALSE; 8488afaa268SBarry Smith ierr = PetscOptionsBool("-snes_check_jacobian","Check each Jacobian with a differenced one","SNESUpdateCheckJacobian",flg,&flg,&set);CHKERRQ(ierr); 8498afaa268SBarry Smith if (set && flg) { 8505341784dSBarry Smith ierr = SNESSetUpdate(snes,SNESUpdateCheckJacobian);CHKERRQ(ierr); 8515341784dSBarry Smith } 8525341784dSBarry Smith 8535341784dSBarry Smith flg = PETSC_FALSE; 8548afaa268SBarry Smith ierr = PetscOptionsBool("-snes_monitor_cancel","Remove all monitors","SNESMonitorCancel",flg,&flg,&set);CHKERRQ(ierr); 8558afaa268SBarry Smith if (set && flg) {ierr = SNESMonitorCancel(snes);CHKERRQ(ierr);} 856eabae89aSBarry Smith 857fde5950dSBarry Smith ierr = SNESMonitorSetFromOptions(snes,"-snes_monitor","Monitor norm of function","SNESMonitorDefault",SNESMonitorDefault,NULL);CHKERRQ(ierr); 858fde5950dSBarry Smith ierr = SNESMonitorSetFromOptions(snes,"-snes_monitor_short","Monitor norm of function with fewer digits","SNESMonitorDefaultShort",SNESMonitorDefaultShort,NULL);CHKERRQ(ierr); 859fde5950dSBarry Smith ierr = SNESMonitorSetFromOptions(snes,"-snes_monitor_range","Monitor range of elements of function","SNESMonitorRange",SNESMonitorRange,NULL);CHKERRQ(ierr); 860eabae89aSBarry Smith 861fde5950dSBarry Smith ierr = SNESMonitorSetFromOptions(snes,"-snes_monitor_ratio","Monitor ratios of the norm of function for consecutive steps","SNESMonitorRatio",SNESMonitorRatio,SNESMonitorRatioSetUp);CHKERRQ(ierr); 862fde5950dSBarry Smith ierr = SNESMonitorSetFromOptions(snes,"-snes_monitor_field","Monitor norm of function (split into fields)","SNESMonitorDefaultField",SNESMonitorDefaultField,NULL);CHKERRQ(ierr); 863fde5950dSBarry Smith ierr = SNESMonitorSetFromOptions(snes,"-snes_monitor_solution","View solution at each iteration","SNESMonitorSolution",SNESMonitorSolution,NULL);CHKERRQ(ierr); 864fde5950dSBarry Smith ierr = SNESMonitorSetFromOptions(snes,"-snes_monitor_solution_update","View correction at each iteration","SNESMonitorSolutionUpdate",SNESMonitorSolutionUpdate,NULL);CHKERRQ(ierr); 865fde5950dSBarry Smith ierr = SNESMonitorSetFromOptions(snes,"-snes_monitor_residual","View residual at each iteration","SNESMonitorResidual",SNESMonitorResidual,NULL);CHKERRQ(ierr); 866fde5950dSBarry Smith ierr = SNESMonitorSetFromOptions(snes,"-snes_monitor_jacupdate_spectrum","Print the change in the spectrum of the Jacobian","SNESMonitorJacUpdateSpectrum",SNESMonitorJacUpdateSpectrum,NULL);CHKERRQ(ierr); 867fde5950dSBarry Smith ierr = SNESMonitorSetFromOptions(snes,"-snes_monitor_fields","Monitor norm of function per field","SNESMonitorSet",SNESMonitorFields,NULL);CHKERRQ(ierr); 8682db13446SMatthew G. Knepley 8695180491cSLisandro Dalcin ierr = PetscOptionsString("-snes_monitor_python","Use Python function","SNESMonitorSet",0,monfilename,PETSC_MAX_PATH_LEN,&flg);CHKERRQ(ierr); 8705180491cSLisandro Dalcin if (flg) {ierr = PetscPythonMonitorSet((PetscObject)snes,monfilename);CHKERRQ(ierr);} 8715180491cSLisandro Dalcin 872fde5950dSBarry Smith 87390d69ab7SBarry Smith flg = PETSC_FALSE; 8740298fd71SBarry Smith ierr = PetscOptionsBool("-snes_monitor_lg_residualnorm","Plot function norm at each iteration","SNESMonitorLGResidualNorm",flg,&flg,NULL);CHKERRQ(ierr); 875459f5d12SBarry Smith if (flg) { 876d96771aaSLisandro Dalcin PetscDrawLG ctx; 877459f5d12SBarry Smith 8786ba87a44SLisandro Dalcin ierr = SNESMonitorLGCreate(PetscObjectComm((PetscObject)snes),NULL,NULL,PETSC_DECIDE,PETSC_DECIDE,400,300,&ctx);CHKERRQ(ierr); 879d96771aaSLisandro Dalcin ierr = SNESMonitorSet(snes,SNESMonitorLGResidualNorm,ctx,(PetscErrorCode (*)(void**))PetscDrawLGDestroy);CHKERRQ(ierr); 880459f5d12SBarry Smith } 88190d69ab7SBarry Smith flg = PETSC_FALSE; 8820298fd71SBarry Smith ierr = PetscOptionsBool("-snes_monitor_lg_range","Plot function range at each iteration","SNESMonitorLGRange",flg,&flg,NULL);CHKERRQ(ierr); 883459f5d12SBarry Smith if (flg) { 884459f5d12SBarry Smith PetscViewer ctx; 885e24b481bSBarry Smith 8866ba87a44SLisandro Dalcin ierr = PetscViewerDrawOpen(PetscObjectComm((PetscObject)snes),NULL,NULL,PETSC_DECIDE,PETSC_DECIDE,400,300,&ctx);CHKERRQ(ierr); 887459f5d12SBarry Smith ierr = SNESMonitorSet(snes,SNESMonitorLGRange,ctx,(PetscErrorCode (*)(void**))PetscViewerDestroy);CHKERRQ(ierr); 888459f5d12SBarry Smith } 8892e7541e6SPeter Brune 8902e7541e6SPeter Brune 891cc0c4584SMatthew G. Knepley 89290d69ab7SBarry Smith flg = PETSC_FALSE; 8938d359177SBarry Smith ierr = PetscOptionsBool("-snes_fd","Use finite differences (slow) to compute Jacobian","SNESComputeJacobianDefault",flg,&flg,NULL);CHKERRQ(ierr); 8944b27c08aSLois Curfman McInnes if (flg) { 8956cab3a1bSJed Brown void *functx; 896b1f624c7SBarry Smith DM dm; 897b1f624c7SBarry Smith DMSNES sdm; 898b1f624c7SBarry Smith ierr = SNESGetDM(snes,&dm);CHKERRQ(ierr); 899b1f624c7SBarry Smith ierr = DMGetDMSNES(dm,&sdm);CHKERRQ(ierr); 900b1f624c7SBarry Smith sdm->jacobianctx = NULL; 9010298fd71SBarry Smith ierr = SNESGetFunction(snes,NULL,NULL,&functx);CHKERRQ(ierr); 9028d359177SBarry Smith ierr = SNESSetJacobian(snes,snes->jacobian,snes->jacobian_pre,SNESComputeJacobianDefault,functx);CHKERRQ(ierr); 903ae15b995SBarry Smith ierr = PetscInfo(snes,"Setting default finite difference Jacobian matrix\n");CHKERRQ(ierr); 9049b94acceSBarry Smith } 905639f9d9dSBarry Smith 90644848bc4SPeter Brune flg = PETSC_FALSE; 9078d359177SBarry Smith ierr = PetscOptionsBool("-snes_fd_function","Use finite differences (slow) to compute function from user objective","SNESObjectiveComputeFunctionDefaultFD",flg,&flg,NULL);CHKERRQ(ierr); 90897584545SPeter Brune if (flg) { 9098d359177SBarry Smith ierr = SNESSetFunction(snes,NULL,SNESObjectiveComputeFunctionDefaultFD,NULL);CHKERRQ(ierr); 91097584545SPeter Brune } 91197584545SPeter Brune 91297584545SPeter Brune flg = PETSC_FALSE; 9138d359177SBarry Smith ierr = PetscOptionsBool("-snes_fd_color","Use finite differences with coloring to compute Jacobian","SNESComputeJacobianDefaultColor",flg,&flg,NULL);CHKERRQ(ierr); 91444848bc4SPeter Brune if (flg) { 915c52e227fSPeter Brune DM dm; 916c52e227fSPeter Brune DMSNES sdm; 917c52e227fSPeter Brune ierr = SNESGetDM(snes,&dm);CHKERRQ(ierr); 918aace71b7SPeter Brune ierr = DMGetDMSNES(dm,&sdm);CHKERRQ(ierr); 919aace71b7SPeter Brune sdm->jacobianctx = NULL; 9208d359177SBarry Smith ierr = SNESSetJacobian(snes,snes->jacobian,snes->jacobian_pre,SNESComputeJacobianDefaultColor,0);CHKERRQ(ierr); 92144848bc4SPeter Brune ierr = PetscInfo(snes,"Setting default finite difference coloring Jacobian matrix\n");CHKERRQ(ierr); 92244848bc4SPeter Brune } 92344848bc4SPeter Brune 924aa3661deSLisandro Dalcin flg = PETSC_FALSE; 925f871313dSBarry Smith ierr = PetscOptionsBool("-snes_mf_operator","Use a Matrix-Free Jacobian with user-provided preconditioner matrix","SNESSetUseMatrixFree",PETSC_FALSE,&snes->mf_operator,&flg);CHKERRQ(ierr); 926d8f46077SPeter Brune if (flg && snes->mf_operator) { 927a8248277SBarry Smith snes->mf_operator = PETSC_TRUE; 928d8f46077SPeter Brune snes->mf = PETSC_TRUE; 929a8248277SBarry Smith } 930aa3661deSLisandro Dalcin flg = PETSC_FALSE; 931f871313dSBarry Smith ierr = PetscOptionsBool("-snes_mf","Use a Matrix-Free Jacobian with no preconditioner matrix","SNESSetUseMatrixFree",PETSC_FALSE,&snes->mf,&flg);CHKERRQ(ierr); 932d8f46077SPeter Brune if (!flg && snes->mf_operator) snes->mf = PETSC_TRUE; 933d8f46077SPeter Brune ierr = PetscOptionsInt("-snes_mf_version","Matrix-Free routines version 1 or 2","None",snes->mf_version,&snes->mf_version,0);CHKERRQ(ierr); 934d28543b3SPeter Brune 935c40d0f55SPeter Brune flg = PETSC_FALSE; 936be95d8f1SBarry Smith ierr = SNESGetNPCSide(snes,&pcside);CHKERRQ(ierr); 937be95d8f1SBarry Smith ierr = PetscOptionsEnum("-snes_npc_side","SNES nonlinear preconditioner side","SNESSetNPCSide",PCSides,(PetscEnum)pcside,(PetscEnum*)&pcside,&flg);CHKERRQ(ierr); 938be95d8f1SBarry Smith if (flg) {ierr = SNESSetNPCSide(snes,pcside);CHKERRQ(ierr);} 939c40d0f55SPeter Brune 940e04113cfSBarry Smith #if defined(PETSC_HAVE_SAWS) 9418a70d858SHong Zhang /* 9428a70d858SHong Zhang Publish convergence information using SAWs 9438a70d858SHong Zhang */ 9448a70d858SHong Zhang flg = PETSC_FALSE; 9458a70d858SHong Zhang ierr = PetscOptionsBool("-snes_monitor_saws","Publish SNES progress using SAWs","SNESMonitorSet",flg,&flg,NULL);CHKERRQ(ierr); 9468a70d858SHong Zhang if (flg) { 9478a70d858SHong Zhang void *ctx; 9488a70d858SHong Zhang ierr = SNESMonitorSAWsCreate(snes,&ctx);CHKERRQ(ierr); 9498a70d858SHong Zhang ierr = SNESMonitorSet(snes,SNESMonitorSAWs,ctx,SNESMonitorSAWsDestroy);CHKERRQ(ierr); 9508a70d858SHong Zhang } 9518a70d858SHong Zhang #endif 9528a70d858SHong Zhang #if defined(PETSC_HAVE_SAWS) 953b90c6cbeSBarry Smith { 954b90c6cbeSBarry Smith PetscBool set; 955b90c6cbeSBarry Smith flg = PETSC_FALSE; 9568a70d858SHong Zhang ierr = PetscOptionsBool("-snes_saws_block","Block for SAWs at end of SNESSolve","PetscObjectSAWsBlock",((PetscObject)snes)->amspublishblock,&flg,&set);CHKERRQ(ierr); 957b90c6cbeSBarry Smith if (set) { 958e04113cfSBarry Smith ierr = PetscObjectSAWsSetBlock((PetscObject)snes,flg);CHKERRQ(ierr); 959b90c6cbeSBarry Smith } 960b90c6cbeSBarry Smith } 961b90c6cbeSBarry Smith #endif 962b90c6cbeSBarry Smith 96376b2cf59SMatthew Knepley for (i = 0; i < numberofsetfromoptions; i++) { 96476b2cf59SMatthew Knepley ierr = (*othersetfromoptions[i])(snes);CHKERRQ(ierr); 96576b2cf59SMatthew Knepley } 96676b2cf59SMatthew Knepley 967e7788613SBarry Smith if (snes->ops->setfromoptions) { 968e55864a3SBarry Smith ierr = (*snes->ops->setfromoptions)(PetscOptionsObject,snes);CHKERRQ(ierr); 969639f9d9dSBarry Smith } 9705d973c19SBarry Smith 9715d973c19SBarry Smith /* process any options handlers added with PetscObjectAddOptionsHandler() */ 9720633abcbSJed Brown ierr = PetscObjectProcessOptionsHandlers(PetscOptionsObject,(PetscObject)snes);CHKERRQ(ierr); 973b0a32e0cSBarry Smith ierr = PetscOptionsEnd();CHKERRQ(ierr); 9744bbc92c1SBarry Smith 9759e764e56SPeter Brune if (!snes->linesearch) { 9767601faf0SJed Brown ierr = SNESGetLineSearch(snes, &snes->linesearch);CHKERRQ(ierr); 9779e764e56SPeter Brune } 978f1c6b773SPeter Brune ierr = SNESLineSearchSetFromOptions(snes->linesearch);CHKERRQ(ierr); 9799e764e56SPeter Brune 9806991f827SBarry Smith if (!snes->ksp) {ierr = SNESGetKSP(snes,&snes->ksp);CHKERRQ(ierr);} 9816991f827SBarry Smith ierr = KSPSetOperators(snes->ksp,snes->jacobian,snes->jacobian_pre);CHKERRQ(ierr); 9826991f827SBarry Smith ierr = KSPSetFromOptions(snes->ksp);CHKERRQ(ierr); 9836991f827SBarry Smith 984be95d8f1SBarry Smith /* if someone has set the SNES NPC type, create it. */ 98551e86f29SPeter Brune ierr = SNESGetOptionsPrefix(snes, &optionsprefix);CHKERRQ(ierr); 986c5929fdfSBarry Smith ierr = PetscOptionsHasName(((PetscObject)snes)->options,optionsprefix, "-npc_snes_type", &pcset);CHKERRQ(ierr); 987efd4aadfSBarry Smith if (pcset && (!snes->npc)) { 988efd4aadfSBarry Smith ierr = SNESGetNPC(snes, &snes->npc);CHKERRQ(ierr); 98951e86f29SPeter Brune } 9903a40ed3dSBarry Smith PetscFunctionReturn(0); 9919b94acceSBarry Smith } 9929b94acceSBarry Smith 993bb9467b5SJed Brown /*@C 994d25893d9SBarry Smith SNESSetComputeApplicationContext - Sets an optional function to compute a user-defined context for 995d25893d9SBarry Smith the nonlinear solvers. 996d25893d9SBarry Smith 997d25893d9SBarry Smith Logically Collective on SNES 998d25893d9SBarry Smith 999d25893d9SBarry Smith Input Parameters: 1000d25893d9SBarry Smith + snes - the SNES context 1001d25893d9SBarry Smith . compute - function to compute the context 1002d25893d9SBarry Smith - destroy - function to destroy the context 1003d25893d9SBarry Smith 1004d25893d9SBarry Smith Level: intermediate 1005d25893d9SBarry Smith 1006bb9467b5SJed Brown Notes: 1007bb9467b5SJed Brown This function is currently not available from Fortran. 1008bb9467b5SJed Brown 1009d25893d9SBarry Smith .keywords: SNES, nonlinear, set, application, context 1010d25893d9SBarry Smith 1011d25893d9SBarry Smith .seealso: SNESGetApplicationContext(), SNESSetComputeApplicationContext(), SNESGetApplicationContext() 1012d25893d9SBarry Smith @*/ 1013d25893d9SBarry Smith PetscErrorCode SNESSetComputeApplicationContext(SNES snes,PetscErrorCode (*compute)(SNES,void**),PetscErrorCode (*destroy)(void**)) 1014d25893d9SBarry Smith { 1015d25893d9SBarry Smith PetscFunctionBegin; 1016d25893d9SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 1017d25893d9SBarry Smith snes->ops->usercompute = compute; 1018d25893d9SBarry Smith snes->ops->userdestroy = destroy; 1019d25893d9SBarry Smith PetscFunctionReturn(0); 1020d25893d9SBarry Smith } 1021a847f771SSatish Balay 1022b07ff414SBarry Smith /*@ 10239b94acceSBarry Smith SNESSetApplicationContext - Sets the optional user-defined context for 10249b94acceSBarry Smith the nonlinear solvers. 10259b94acceSBarry Smith 10263f9fe445SBarry Smith Logically Collective on SNES 1027fee21e36SBarry Smith 1028c7afd0dbSLois Curfman McInnes Input Parameters: 1029c7afd0dbSLois Curfman McInnes + snes - the SNES context 1030c7afd0dbSLois Curfman McInnes - usrP - optional user context 1031c7afd0dbSLois Curfman McInnes 103236851e7fSLois Curfman McInnes Level: intermediate 103336851e7fSLois Curfman McInnes 1034daf670e6SBarry Smith Fortran Notes: To use this from Fortran you must write a Fortran interface definition for this 1035daf670e6SBarry Smith function that tells Fortran the Fortran derived data type that you are passing in as the ctx argument. 1036daf670e6SBarry Smith 10379b94acceSBarry Smith .keywords: SNES, nonlinear, set, application, context 10389b94acceSBarry Smith 1039ecaffddaSVictor Eijkhout .seealso: SNESGetApplicationContext() 10409b94acceSBarry Smith @*/ 10417087cfbeSBarry Smith PetscErrorCode SNESSetApplicationContext(SNES snes,void *usrP) 10429b94acceSBarry Smith { 10431b2093e4SBarry Smith PetscErrorCode ierr; 1044b07ff414SBarry Smith KSP ksp; 10451b2093e4SBarry Smith 10463a40ed3dSBarry Smith PetscFunctionBegin; 10470700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 1048b07ff414SBarry Smith ierr = SNESGetKSP(snes,&ksp);CHKERRQ(ierr); 1049b07ff414SBarry Smith ierr = KSPSetApplicationContext(ksp,usrP);CHKERRQ(ierr); 10509b94acceSBarry Smith snes->user = usrP; 10513a40ed3dSBarry Smith PetscFunctionReturn(0); 10529b94acceSBarry Smith } 105374679c65SBarry Smith 1054b07ff414SBarry Smith /*@ 10559b94acceSBarry Smith SNESGetApplicationContext - Gets the user-defined context for the 10569b94acceSBarry Smith nonlinear solvers. 10579b94acceSBarry Smith 1058c7afd0dbSLois Curfman McInnes Not Collective 1059c7afd0dbSLois Curfman McInnes 10609b94acceSBarry Smith Input Parameter: 10619b94acceSBarry Smith . snes - SNES context 10629b94acceSBarry Smith 10639b94acceSBarry Smith Output Parameter: 10649b94acceSBarry Smith . usrP - user context 10659b94acceSBarry Smith 1066daf670e6SBarry Smith Fortran Notes: To use this from Fortran you must write a Fortran interface definition for this 1067daf670e6SBarry Smith function that tells Fortran the Fortran derived data type that you are passing in as the ctx argument. 1068daf670e6SBarry Smith 106936851e7fSLois Curfman McInnes Level: intermediate 107036851e7fSLois Curfman McInnes 10719b94acceSBarry Smith .keywords: SNES, nonlinear, get, application, context 10729b94acceSBarry Smith 10739b94acceSBarry Smith .seealso: SNESSetApplicationContext() 10749b94acceSBarry Smith @*/ 1075e71120c6SJed Brown PetscErrorCode SNESGetApplicationContext(SNES snes,void *usrP) 10769b94acceSBarry Smith { 10773a40ed3dSBarry Smith PetscFunctionBegin; 10780700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 1079e71120c6SJed Brown *(void**)usrP = snes->user; 10803a40ed3dSBarry Smith PetscFunctionReturn(0); 10819b94acceSBarry Smith } 108274679c65SBarry Smith 10839b94acceSBarry Smith /*@ 10843565c898SBarry Smith SNESSetUseMatrixFree - indicates that SNES should use matrix free finite difference matrix vector products internally to apply 10853565c898SBarry Smith the Jacobian. 10863565c898SBarry Smith 10873565c898SBarry Smith Collective on SNES 10883565c898SBarry Smith 10893565c898SBarry Smith Input Parameters: 10903565c898SBarry Smith + snes - SNES context 10913565c898SBarry Smith . mf - use matrix-free for both the Amat and Pmat used by SNESSetJacobian(), both the Amat and Pmat set in SNESSetJacobian() will be ignored 10923565c898SBarry Smith - mf_operator - use matrix-free only for the Amat used by SNESSetJacobian(), this means the user provided Pmat will continue to be used 10933565c898SBarry Smith 10943565c898SBarry Smith Options Database: 10953565c898SBarry Smith + -snes_mf - use matrix free for both the mat and pmat operator 10963565c898SBarry Smith - -snes_mf_operator - use matrix free only for the mat operator 10973565c898SBarry Smith 10983565c898SBarry Smith Level: intermediate 10993565c898SBarry Smith 11003565c898SBarry Smith .keywords: SNES, nonlinear, get, iteration, number, 11013565c898SBarry Smith 11023565c898SBarry Smith .seealso: SNESGetUseMatrixFree(), MatCreateSNESMF() 11033565c898SBarry Smith @*/ 11043565c898SBarry Smith PetscErrorCode SNESSetUseMatrixFree(SNES snes,PetscBool mf_operator,PetscBool mf) 11053565c898SBarry Smith { 11063565c898SBarry Smith PetscFunctionBegin; 11073565c898SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 110888b4c220SStefano Zampini PetscValidLogicalCollectiveBool(snes,mf_operator,2); 110988b4c220SStefano Zampini PetscValidLogicalCollectiveBool(snes,mf,3); 11103565c898SBarry Smith if (mf && !mf_operator) SETERRQ(PetscObjectComm((PetscObject)snes),PETSC_ERR_ARG_INCOMP,"If using mf must also use mf_operator"); 11113565c898SBarry Smith snes->mf = mf; 11123565c898SBarry Smith snes->mf_operator = mf_operator; 11133565c898SBarry Smith PetscFunctionReturn(0); 11143565c898SBarry Smith } 11153565c898SBarry Smith 11163565c898SBarry Smith /*@ 11173565c898SBarry Smith SNESGetUseMatrixFree - indicates if the SNES uses matrix free finite difference matrix vector products to apply 11183565c898SBarry Smith the Jacobian. 11193565c898SBarry Smith 11203565c898SBarry Smith Collective on SNES 11213565c898SBarry Smith 11223565c898SBarry Smith Input Parameter: 11233565c898SBarry Smith . snes - SNES context 11243565c898SBarry Smith 11253565c898SBarry Smith Output Parameters: 11263565c898SBarry Smith + mf - use matrix-free for both the Amat and Pmat used by SNESSetJacobian(), both the Amat and Pmat set in SNESSetJacobian() will be ignored 11273565c898SBarry Smith - mf_operator - use matrix-free only for the Amat used by SNESSetJacobian(), this means the user provided Pmat will continue to be used 11283565c898SBarry Smith 11293565c898SBarry Smith Options Database: 11303565c898SBarry Smith + -snes_mf - use matrix free for both the mat and pmat operator 11313565c898SBarry Smith - -snes_mf_operator - use matrix free only for the mat operator 11323565c898SBarry Smith 11333565c898SBarry Smith Level: intermediate 11343565c898SBarry Smith 11353565c898SBarry Smith .keywords: SNES, nonlinear, get, iteration, number, 11363565c898SBarry Smith 11373565c898SBarry Smith .seealso: SNESSetUseMatrixFree(), MatCreateSNESMF() 11383565c898SBarry Smith @*/ 11393565c898SBarry Smith PetscErrorCode SNESGetUseMatrixFree(SNES snes,PetscBool *mf_operator,PetscBool *mf) 11403565c898SBarry Smith { 11413565c898SBarry Smith PetscFunctionBegin; 11423565c898SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 11433565c898SBarry Smith if (mf) *mf = snes->mf; 11443565c898SBarry Smith if (mf_operator) *mf_operator = snes->mf_operator; 11453565c898SBarry Smith PetscFunctionReturn(0); 11463565c898SBarry Smith } 11473565c898SBarry Smith 11483565c898SBarry Smith /*@ 1149c8228a4eSBarry Smith SNESGetIterationNumber - Gets the number of nonlinear iterations completed 1150c8228a4eSBarry Smith at this time. 11519b94acceSBarry Smith 1152c7afd0dbSLois Curfman McInnes Not Collective 1153c7afd0dbSLois Curfman McInnes 11549b94acceSBarry Smith Input Parameter: 11559b94acceSBarry Smith . snes - SNES context 11569b94acceSBarry Smith 11579b94acceSBarry Smith Output Parameter: 11589b94acceSBarry Smith . iter - iteration number 11599b94acceSBarry Smith 1160c8228a4eSBarry Smith Notes: 1161c8228a4eSBarry Smith For example, during the computation of iteration 2 this would return 1. 1162c8228a4eSBarry Smith 1163c8228a4eSBarry Smith This is useful for using lagged Jacobians (where one does not recompute the 116408405cd6SLois Curfman McInnes Jacobian at each SNES iteration). For example, the code 116508405cd6SLois Curfman McInnes .vb 116608405cd6SLois Curfman McInnes ierr = SNESGetIterationNumber(snes,&it); 116708405cd6SLois Curfman McInnes if (!(it % 2)) { 116808405cd6SLois Curfman McInnes [compute Jacobian here] 116908405cd6SLois Curfman McInnes } 117008405cd6SLois Curfman McInnes .ve 1171c8228a4eSBarry Smith can be used in your ComputeJacobian() function to cause the Jacobian to be 117208405cd6SLois Curfman McInnes recomputed every second SNES iteration. 1173c8228a4eSBarry Smith 1174c04deec6SBarry Smith After the SNES solve is complete this will return the number of nonlinear iterations used. 1175c04deec6SBarry Smith 117636851e7fSLois Curfman McInnes Level: intermediate 117736851e7fSLois Curfman McInnes 11782b668275SBarry Smith .keywords: SNES, nonlinear, get, iteration, number, 11792b668275SBarry Smith 118071dbe336SPeter Brune .seealso: SNESGetLinearSolveIterations() 11819b94acceSBarry Smith @*/ 11827087cfbeSBarry Smith PetscErrorCode SNESGetIterationNumber(SNES snes,PetscInt *iter) 11839b94acceSBarry Smith { 11843a40ed3dSBarry Smith PetscFunctionBegin; 11850700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 11864482741eSBarry Smith PetscValidIntPointer(iter,2); 11879b94acceSBarry Smith *iter = snes->iter; 11883a40ed3dSBarry Smith PetscFunctionReturn(0); 11899b94acceSBarry Smith } 119074679c65SBarry Smith 1191360c497dSPeter Brune /*@ 1192360c497dSPeter Brune SNESSetIterationNumber - Sets the current iteration number. 1193360c497dSPeter Brune 1194360c497dSPeter Brune Not Collective 1195360c497dSPeter Brune 1196360c497dSPeter Brune Input Parameter: 1197360c497dSPeter Brune . snes - SNES context 1198360c497dSPeter Brune . iter - iteration number 1199360c497dSPeter Brune 1200360c497dSPeter Brune Level: developer 1201360c497dSPeter Brune 1202360c497dSPeter Brune .keywords: SNES, nonlinear, set, iteration, number, 1203360c497dSPeter Brune 120471dbe336SPeter Brune .seealso: SNESGetLinearSolveIterations() 1205360c497dSPeter Brune @*/ 1206360c497dSPeter Brune PetscErrorCode SNESSetIterationNumber(SNES snes,PetscInt iter) 1207360c497dSPeter Brune { 1208360c497dSPeter Brune PetscErrorCode ierr; 1209360c497dSPeter Brune 1210360c497dSPeter Brune PetscFunctionBegin; 1211360c497dSPeter Brune PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 1212e04113cfSBarry Smith ierr = PetscObjectSAWsTakeAccess((PetscObject)snes);CHKERRQ(ierr); 1213360c497dSPeter Brune snes->iter = iter; 1214e04113cfSBarry Smith ierr = PetscObjectSAWsGrantAccess((PetscObject)snes);CHKERRQ(ierr); 1215360c497dSPeter Brune PetscFunctionReturn(0); 1216360c497dSPeter Brune } 1217360c497dSPeter Brune 12189b94acceSBarry Smith /*@ 1219b850b91aSLisandro Dalcin SNESGetNonlinearStepFailures - Gets the number of unsuccessful steps 12209b94acceSBarry Smith attempted by the nonlinear solver. 12219b94acceSBarry Smith 1222c7afd0dbSLois Curfman McInnes Not Collective 1223c7afd0dbSLois Curfman McInnes 12249b94acceSBarry Smith Input Parameter: 12259b94acceSBarry Smith . snes - SNES context 12269b94acceSBarry Smith 12279b94acceSBarry Smith Output Parameter: 12289b94acceSBarry Smith . nfails - number of unsuccessful steps attempted 12299b94acceSBarry Smith 1230c96a6f78SLois Curfman McInnes Notes: 1231c96a6f78SLois Curfman McInnes This counter is reset to zero for each successive call to SNESSolve(). 1232c96a6f78SLois Curfman McInnes 123336851e7fSLois Curfman McInnes Level: intermediate 123436851e7fSLois Curfman McInnes 12359b94acceSBarry Smith .keywords: SNES, nonlinear, get, number, unsuccessful, steps 123658ebbce7SBarry Smith 1237e1c61ce8SBarry Smith .seealso: SNESGetMaxLinearSolveFailures(), SNESGetLinearSolveIterations(), SNESSetMaxLinearSolveFailures(), SNESGetLinearSolveFailures(), 123858ebbce7SBarry Smith SNESSetMaxNonlinearStepFailures(), SNESGetMaxNonlinearStepFailures() 12399b94acceSBarry Smith @*/ 12407087cfbeSBarry Smith PetscErrorCode SNESGetNonlinearStepFailures(SNES snes,PetscInt *nfails) 12419b94acceSBarry Smith { 12423a40ed3dSBarry Smith PetscFunctionBegin; 12430700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 12444482741eSBarry Smith PetscValidIntPointer(nfails,2); 124550ffb88aSMatthew Knepley *nfails = snes->numFailures; 124650ffb88aSMatthew Knepley PetscFunctionReturn(0); 124750ffb88aSMatthew Knepley } 124850ffb88aSMatthew Knepley 124950ffb88aSMatthew Knepley /*@ 1250b850b91aSLisandro Dalcin SNESSetMaxNonlinearStepFailures - Sets the maximum number of unsuccessful steps 125150ffb88aSMatthew Knepley attempted by the nonlinear solver before it gives up. 125250ffb88aSMatthew Knepley 125350ffb88aSMatthew Knepley Not Collective 125450ffb88aSMatthew Knepley 125550ffb88aSMatthew Knepley Input Parameters: 125650ffb88aSMatthew Knepley + snes - SNES context 125750ffb88aSMatthew Knepley - maxFails - maximum of unsuccessful steps 125850ffb88aSMatthew Knepley 125950ffb88aSMatthew Knepley Level: intermediate 126050ffb88aSMatthew Knepley 126150ffb88aSMatthew Knepley .keywords: SNES, nonlinear, set, maximum, unsuccessful, steps 126258ebbce7SBarry Smith 1263e1c61ce8SBarry Smith .seealso: SNESGetMaxLinearSolveFailures(), SNESGetLinearSolveIterations(), SNESSetMaxLinearSolveFailures(), SNESGetLinearSolveFailures(), 126458ebbce7SBarry Smith SNESGetMaxNonlinearStepFailures(), SNESGetNonlinearStepFailures() 126550ffb88aSMatthew Knepley @*/ 12667087cfbeSBarry Smith PetscErrorCode SNESSetMaxNonlinearStepFailures(SNES snes, PetscInt maxFails) 126750ffb88aSMatthew Knepley { 126850ffb88aSMatthew Knepley PetscFunctionBegin; 12690700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 127050ffb88aSMatthew Knepley snes->maxFailures = maxFails; 127150ffb88aSMatthew Knepley PetscFunctionReturn(0); 127250ffb88aSMatthew Knepley } 127350ffb88aSMatthew Knepley 127450ffb88aSMatthew Knepley /*@ 1275b850b91aSLisandro Dalcin SNESGetMaxNonlinearStepFailures - Gets the maximum number of unsuccessful steps 127650ffb88aSMatthew Knepley attempted by the nonlinear solver before it gives up. 127750ffb88aSMatthew Knepley 127850ffb88aSMatthew Knepley Not Collective 127950ffb88aSMatthew Knepley 128050ffb88aSMatthew Knepley Input Parameter: 128150ffb88aSMatthew Knepley . snes - SNES context 128250ffb88aSMatthew Knepley 128350ffb88aSMatthew Knepley Output Parameter: 128450ffb88aSMatthew Knepley . maxFails - maximum of unsuccessful steps 128550ffb88aSMatthew Knepley 128650ffb88aSMatthew Knepley Level: intermediate 128750ffb88aSMatthew Knepley 128850ffb88aSMatthew Knepley .keywords: SNES, nonlinear, get, maximum, unsuccessful, steps 128958ebbce7SBarry Smith 1290e1c61ce8SBarry Smith .seealso: SNESGetMaxLinearSolveFailures(), SNESGetLinearSolveIterations(), SNESSetMaxLinearSolveFailures(), SNESGetLinearSolveFailures(), 129158ebbce7SBarry Smith SNESSetMaxNonlinearStepFailures(), SNESGetNonlinearStepFailures() 129258ebbce7SBarry Smith 129350ffb88aSMatthew Knepley @*/ 12947087cfbeSBarry Smith PetscErrorCode SNESGetMaxNonlinearStepFailures(SNES snes, PetscInt *maxFails) 129550ffb88aSMatthew Knepley { 129650ffb88aSMatthew Knepley PetscFunctionBegin; 12970700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 12984482741eSBarry Smith PetscValidIntPointer(maxFails,2); 129950ffb88aSMatthew Knepley *maxFails = snes->maxFailures; 13003a40ed3dSBarry Smith PetscFunctionReturn(0); 13019b94acceSBarry Smith } 1302a847f771SSatish Balay 13032541af92SBarry Smith /*@ 13042541af92SBarry Smith SNESGetNumberFunctionEvals - Gets the number of user provided function evaluations 13052541af92SBarry Smith done by SNES. 13062541af92SBarry Smith 13072541af92SBarry Smith Not Collective 13082541af92SBarry Smith 13092541af92SBarry Smith Input Parameter: 13102541af92SBarry Smith . snes - SNES context 13112541af92SBarry Smith 13122541af92SBarry Smith Output Parameter: 13132541af92SBarry Smith . nfuncs - number of evaluations 13142541af92SBarry Smith 13152541af92SBarry Smith Level: intermediate 13162541af92SBarry Smith 1317971e163fSPeter Brune Notes: Reset every time SNESSolve is called unless SNESSetCountersReset() is used. 1318971e163fSPeter Brune 13192541af92SBarry Smith .keywords: SNES, nonlinear, get, maximum, unsuccessful, steps 132058ebbce7SBarry Smith 1321971e163fSPeter Brune .seealso: SNESGetMaxLinearSolveFailures(), SNESGetLinearSolveIterations(), SNESSetMaxLinearSolveFailures(), SNESGetLinearSolveFailures(), SNESSetCountersReset() 13222541af92SBarry Smith @*/ 13237087cfbeSBarry Smith PetscErrorCode SNESGetNumberFunctionEvals(SNES snes, PetscInt *nfuncs) 13242541af92SBarry Smith { 13252541af92SBarry Smith PetscFunctionBegin; 13260700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 13272541af92SBarry Smith PetscValidIntPointer(nfuncs,2); 13282541af92SBarry Smith *nfuncs = snes->nfuncs; 13292541af92SBarry Smith PetscFunctionReturn(0); 13302541af92SBarry Smith } 13312541af92SBarry Smith 13323d4c4710SBarry Smith /*@ 13333d4c4710SBarry Smith SNESGetLinearSolveFailures - Gets the number of failed (non-converged) 13343d4c4710SBarry Smith linear solvers. 13353d4c4710SBarry Smith 13363d4c4710SBarry Smith Not Collective 13373d4c4710SBarry Smith 13383d4c4710SBarry Smith Input Parameter: 13393d4c4710SBarry Smith . snes - SNES context 13403d4c4710SBarry Smith 13413d4c4710SBarry Smith Output Parameter: 13423d4c4710SBarry Smith . nfails - number of failed solves 13433d4c4710SBarry Smith 13449d85da0cSMatthew G. Knepley Level: intermediate 13459d85da0cSMatthew G. Knepley 13469d85da0cSMatthew G. Knepley Options Database Keys: 13479d85da0cSMatthew G. Knepley . -snes_max_linear_solve_fail <num> - The number of failures before the solve is terminated 13489d85da0cSMatthew G. Knepley 13493d4c4710SBarry Smith Notes: 13503d4c4710SBarry Smith This counter is reset to zero for each successive call to SNESSolve(). 13513d4c4710SBarry Smith 13523d4c4710SBarry Smith .keywords: SNES, nonlinear, get, number, unsuccessful, steps 135358ebbce7SBarry Smith 1354e1c61ce8SBarry Smith .seealso: SNESGetMaxLinearSolveFailures(), SNESGetLinearSolveIterations(), SNESSetMaxLinearSolveFailures() 13553d4c4710SBarry Smith @*/ 13567087cfbeSBarry Smith PetscErrorCode SNESGetLinearSolveFailures(SNES snes,PetscInt *nfails) 13573d4c4710SBarry Smith { 13583d4c4710SBarry Smith PetscFunctionBegin; 13590700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 13603d4c4710SBarry Smith PetscValidIntPointer(nfails,2); 13613d4c4710SBarry Smith *nfails = snes->numLinearSolveFailures; 13623d4c4710SBarry Smith PetscFunctionReturn(0); 13633d4c4710SBarry Smith } 13643d4c4710SBarry Smith 13653d4c4710SBarry Smith /*@ 13663d4c4710SBarry Smith SNESSetMaxLinearSolveFailures - the number of failed linear solve attempts 13673d4c4710SBarry Smith allowed before SNES returns with a diverged reason of SNES_DIVERGED_LINEAR_SOLVE 13683d4c4710SBarry Smith 13693f9fe445SBarry Smith Logically Collective on SNES 13703d4c4710SBarry Smith 13713d4c4710SBarry Smith Input Parameters: 13723d4c4710SBarry Smith + snes - SNES context 13733d4c4710SBarry Smith - maxFails - maximum allowed linear solve failures 13743d4c4710SBarry Smith 13753d4c4710SBarry Smith Level: intermediate 13763d4c4710SBarry Smith 13779d85da0cSMatthew G. Knepley Options Database Keys: 13789d85da0cSMatthew G. Knepley . -snes_max_linear_solve_fail <num> - The number of failures before the solve is terminated 13799d85da0cSMatthew G. Knepley 1380a6796414SBarry Smith Notes: By default this is 0; that is SNES returns on the first failed linear solve 13813d4c4710SBarry Smith 13823d4c4710SBarry Smith .keywords: SNES, nonlinear, set, maximum, unsuccessful, steps 13833d4c4710SBarry Smith 138458ebbce7SBarry Smith .seealso: SNESGetLinearSolveFailures(), SNESGetMaxLinearSolveFailures(), SNESGetLinearSolveIterations() 13853d4c4710SBarry Smith @*/ 13867087cfbeSBarry Smith PetscErrorCode SNESSetMaxLinearSolveFailures(SNES snes, PetscInt maxFails) 13873d4c4710SBarry Smith { 13883d4c4710SBarry Smith PetscFunctionBegin; 13890700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 1390c5eb9154SBarry Smith PetscValidLogicalCollectiveInt(snes,maxFails,2); 13913d4c4710SBarry Smith snes->maxLinearSolveFailures = maxFails; 13923d4c4710SBarry Smith PetscFunctionReturn(0); 13933d4c4710SBarry Smith } 13943d4c4710SBarry Smith 13953d4c4710SBarry Smith /*@ 13963d4c4710SBarry Smith SNESGetMaxLinearSolveFailures - gets the maximum number of linear solve failures that 13973d4c4710SBarry Smith are allowed before SNES terminates 13983d4c4710SBarry Smith 13993d4c4710SBarry Smith Not Collective 14003d4c4710SBarry Smith 14013d4c4710SBarry Smith Input Parameter: 14023d4c4710SBarry Smith . snes - SNES context 14033d4c4710SBarry Smith 14043d4c4710SBarry Smith Output Parameter: 14053d4c4710SBarry Smith . maxFails - maximum of unsuccessful solves allowed 14063d4c4710SBarry Smith 14073d4c4710SBarry Smith Level: intermediate 14083d4c4710SBarry Smith 14093d4c4710SBarry Smith Notes: By default this is 1; that is SNES returns on the first failed linear solve 14103d4c4710SBarry Smith 14113d4c4710SBarry Smith .keywords: SNES, nonlinear, get, maximum, unsuccessful, steps 14123d4c4710SBarry Smith 1413e1c61ce8SBarry Smith .seealso: SNESGetLinearSolveFailures(), SNESGetLinearSolveIterations(), SNESSetMaxLinearSolveFailures(), 14143d4c4710SBarry Smith @*/ 14157087cfbeSBarry Smith PetscErrorCode SNESGetMaxLinearSolveFailures(SNES snes, PetscInt *maxFails) 14163d4c4710SBarry Smith { 14173d4c4710SBarry Smith PetscFunctionBegin; 14180700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 14193d4c4710SBarry Smith PetscValidIntPointer(maxFails,2); 14203d4c4710SBarry Smith *maxFails = snes->maxLinearSolveFailures; 14213d4c4710SBarry Smith PetscFunctionReturn(0); 14223d4c4710SBarry Smith } 14233d4c4710SBarry Smith 1424c96a6f78SLois Curfman McInnes /*@ 1425b850b91aSLisandro Dalcin SNESGetLinearSolveIterations - Gets the total number of linear iterations 1426c96a6f78SLois Curfman McInnes used by the nonlinear solver. 1427c96a6f78SLois Curfman McInnes 1428c7afd0dbSLois Curfman McInnes Not Collective 1429c7afd0dbSLois Curfman McInnes 1430c96a6f78SLois Curfman McInnes Input Parameter: 1431c96a6f78SLois Curfman McInnes . snes - SNES context 1432c96a6f78SLois Curfman McInnes 1433c96a6f78SLois Curfman McInnes Output Parameter: 1434c96a6f78SLois Curfman McInnes . lits - number of linear iterations 1435c96a6f78SLois Curfman McInnes 1436c96a6f78SLois Curfman McInnes Notes: 1437971e163fSPeter Brune This counter is reset to zero for each successive call to SNESSolve() unless SNESSetCountersReset() is used. 1438c96a6f78SLois Curfman McInnes 1439010be392SBarry Smith If the linear solver fails inside the SNESSolve() the iterations for that call to the linear solver are not included. If you wish to count them 1440010be392SBarry Smith then call KSPGetIterationNumber() after the failed solve. 1441010be392SBarry Smith 144236851e7fSLois Curfman McInnes Level: intermediate 144336851e7fSLois Curfman McInnes 1444c96a6f78SLois Curfman McInnes .keywords: SNES, nonlinear, get, number, linear, iterations 14452b668275SBarry Smith 144671dbe336SPeter Brune .seealso: SNESGetIterationNumber(), SNESGetLinearSolveFailures(), SNESGetMaxLinearSolveFailures(), SNESSetCountersReset() 1447c96a6f78SLois Curfman McInnes @*/ 14487087cfbeSBarry Smith PetscErrorCode SNESGetLinearSolveIterations(SNES snes,PetscInt *lits) 1449c96a6f78SLois Curfman McInnes { 14503a40ed3dSBarry Smith PetscFunctionBegin; 14510700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 14524482741eSBarry Smith PetscValidIntPointer(lits,2); 1453c96a6f78SLois Curfman McInnes *lits = snes->linear_its; 14543a40ed3dSBarry Smith PetscFunctionReturn(0); 1455c96a6f78SLois Curfman McInnes } 1456c96a6f78SLois Curfman McInnes 1457971e163fSPeter Brune /*@ 1458971e163fSPeter Brune SNESSetCountersReset - Sets whether or not the counters for linear iterations and function evaluations 1459971e163fSPeter Brune are reset every time SNESSolve() is called. 1460971e163fSPeter Brune 1461971e163fSPeter Brune Logically Collective on SNES 1462971e163fSPeter Brune 1463971e163fSPeter Brune Input Parameter: 1464971e163fSPeter Brune + snes - SNES context 1465971e163fSPeter Brune - reset - whether to reset the counters or not 1466971e163fSPeter Brune 1467971e163fSPeter Brune Notes: 1468fa19ca70SBarry Smith This defaults to PETSC_TRUE 1469971e163fSPeter Brune 1470971e163fSPeter Brune Level: developer 1471971e163fSPeter Brune 1472971e163fSPeter Brune .keywords: SNES, nonlinear, set, reset, number, linear, iterations 1473971e163fSPeter Brune 1474734794cfSBarry Smith .seealso: SNESGetNumberFunctionEvals(), SNESGetLinearSolveIterations(), SNESGetNPC() 1475971e163fSPeter Brune @*/ 1476971e163fSPeter Brune PetscErrorCode SNESSetCountersReset(SNES snes,PetscBool reset) 1477971e163fSPeter Brune { 1478971e163fSPeter Brune PetscFunctionBegin; 1479971e163fSPeter Brune PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 1480971e163fSPeter Brune PetscValidLogicalCollectiveBool(snes,reset,2); 1481971e163fSPeter Brune snes->counters_reset = reset; 1482971e163fSPeter Brune PetscFunctionReturn(0); 1483971e163fSPeter Brune } 1484971e163fSPeter Brune 148582bf6240SBarry Smith 14862999313aSBarry Smith /*@ 14872999313aSBarry Smith SNESSetKSP - Sets a KSP context for the SNES object to use 14882999313aSBarry Smith 14892999313aSBarry Smith Not Collective, but the SNES and KSP objects must live on the same MPI_Comm 14902999313aSBarry Smith 14912999313aSBarry Smith Input Parameters: 14922999313aSBarry Smith + snes - the SNES context 14932999313aSBarry Smith - ksp - the KSP context 14942999313aSBarry Smith 14952999313aSBarry Smith Notes: 14962999313aSBarry Smith The SNES object already has its KSP object, you can obtain with SNESGetKSP() 14972999313aSBarry Smith so this routine is rarely needed. 14982999313aSBarry Smith 14992999313aSBarry Smith The KSP object that is already in the SNES object has its reference count 15002999313aSBarry Smith decreased by one. 15012999313aSBarry Smith 15022999313aSBarry Smith Level: developer 15032999313aSBarry Smith 15042999313aSBarry Smith .keywords: SNES, nonlinear, get, KSP, context 15052999313aSBarry Smith 15062999313aSBarry Smith .seealso: KSPGetPC(), SNESCreate(), KSPCreate(), SNESSetKSP() 15072999313aSBarry Smith @*/ 15087087cfbeSBarry Smith PetscErrorCode SNESSetKSP(SNES snes,KSP ksp) 15092999313aSBarry Smith { 15102999313aSBarry Smith PetscErrorCode ierr; 15112999313aSBarry Smith 15122999313aSBarry Smith PetscFunctionBegin; 15130700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 15140700a824SBarry Smith PetscValidHeaderSpecific(ksp,KSP_CLASSID,2); 15152999313aSBarry Smith PetscCheckSameComm(snes,1,ksp,2); 15167dcf0eaaSdalcinl ierr = PetscObjectReference((PetscObject)ksp);CHKERRQ(ierr); 1517906ed7ccSBarry Smith if (snes->ksp) {ierr = PetscObjectDereference((PetscObject)snes->ksp);CHKERRQ(ierr);} 15182999313aSBarry Smith snes->ksp = ksp; 15192999313aSBarry Smith PetscFunctionReturn(0); 15202999313aSBarry Smith } 15212999313aSBarry Smith 15229b94acceSBarry Smith /* -----------------------------------------------------------*/ 152352baeb72SSatish Balay /*@ 15249b94acceSBarry Smith SNESCreate - Creates a nonlinear solver context. 15259b94acceSBarry Smith 1526c7afd0dbSLois Curfman McInnes Collective on MPI_Comm 1527c7afd0dbSLois Curfman McInnes 1528c7afd0dbSLois Curfman McInnes Input Parameters: 1529906ed7ccSBarry Smith . comm - MPI communicator 15309b94acceSBarry Smith 15319b94acceSBarry Smith Output Parameter: 15329b94acceSBarry Smith . outsnes - the new SNES context 15339b94acceSBarry Smith 1534c7afd0dbSLois Curfman McInnes Options Database Keys: 1535c7afd0dbSLois Curfman McInnes + -snes_mf - Activates default matrix-free Jacobian-vector products, 1536c7afd0dbSLois Curfman McInnes and no preconditioning matrix 1537c7afd0dbSLois Curfman McInnes . -snes_mf_operator - Activates default matrix-free Jacobian-vector 1538c7afd0dbSLois Curfman McInnes products, and a user-provided preconditioning matrix 1539c7afd0dbSLois Curfman McInnes as set by SNESSetJacobian() 1540c7afd0dbSLois Curfman McInnes - -snes_fd - Uses (slow!) finite differences to compute Jacobian 1541c1f60f51SBarry Smith 154236851e7fSLois Curfman McInnes Level: beginner 154336851e7fSLois Curfman McInnes 1544efd4aadfSBarry Smith Developer Notes: SNES always creates a KSP object even though many SNES methods do not use it. This is 1545efd4aadfSBarry Smith unfortunate and should be fixed at some point. The flag snes->usesksp indicates if the 1546efd4aadfSBarry Smith particular method does use KSP and regulates if the information about the KSP is printed 1547efd4aadfSBarry Smith in SNESView(). TSSetFromOptions() does call SNESSetFromOptions() which can lead to users being confused 1548efd4aadfSBarry Smith by help messages about meaningless SNES options. 1549efd4aadfSBarry Smith 1550efd4aadfSBarry Smith SNES always creates the snes->kspconvctx even though it is used by only one type. This should 1551efd4aadfSBarry Smith be fixed. 1552efd4aadfSBarry Smith 15539b94acceSBarry Smith .keywords: SNES, nonlinear, create, context 15549b94acceSBarry Smith 1555a8054027SBarry Smith .seealso: SNESSolve(), SNESDestroy(), SNES, SNESSetLagPreconditioner() 1556a8054027SBarry Smith 15579b94acceSBarry Smith @*/ 15587087cfbeSBarry Smith PetscErrorCode SNESCreate(MPI_Comm comm,SNES *outsnes) 15599b94acceSBarry Smith { 1560dfbe8321SBarry Smith PetscErrorCode ierr; 15619b94acceSBarry Smith SNES snes; 1562fa9f3622SBarry Smith SNESKSPEW *kctx; 156337fcc0dbSBarry Smith 15643a40ed3dSBarry Smith PetscFunctionBegin; 1565ed1caa07SMatthew Knepley PetscValidPointer(outsnes,2); 15660298fd71SBarry Smith *outsnes = NULL; 1567607a6623SBarry Smith ierr = SNESInitializePackage();CHKERRQ(ierr); 15688ba1e511SMatthew Knepley 156973107ff1SLisandro Dalcin ierr = PetscHeaderCreate(snes,SNES_CLASSID,"SNES","Nonlinear solver","SNES",comm,SNESDestroy,SNESView);CHKERRQ(ierr); 15707adad957SLisandro Dalcin 15718d359177SBarry Smith snes->ops->converged = SNESConvergedDefault; 15722c155ee1SBarry Smith snes->usesksp = PETSC_TRUE; 157388976e71SPeter Brune snes->tolerancesset = PETSC_FALSE; 15749b94acceSBarry Smith snes->max_its = 50; 15759750a799SBarry Smith snes->max_funcs = 10000; 15769b94acceSBarry Smith snes->norm = 0.0; 1577365a6726SPeter Brune snes->normschedule = SNES_NORM_ALWAYS; 15786c67d002SPeter Brune snes->functype = SNES_FUNCTION_DEFAULT; 15793a2046daSBarry Smith #if defined(PETSC_USE_REAL_SINGLE) 15803a2046daSBarry Smith snes->rtol = 1.e-5; 15813a2046daSBarry Smith #else 1582b4874afaSBarry Smith snes->rtol = 1.e-8; 15833a2046daSBarry Smith #endif 1584b4874afaSBarry Smith snes->ttol = 0.0; 15853a2046daSBarry Smith #if defined(PETSC_USE_REAL_SINGLE) 15863a2046daSBarry Smith snes->abstol = 1.e-25; 15873a2046daSBarry Smith #else 158870441072SBarry Smith snes->abstol = 1.e-50; 15893a2046daSBarry Smith #endif 15907cd0ae37SLisandro Dalcin #if defined(PETSC_USE_REAL_SINGLE) 15917cd0ae37SLisandro Dalcin snes->stol = 1.e-5; 15927cd0ae37SLisandro Dalcin #else 1593c60f73f4SPeter Brune snes->stol = 1.e-8; 15947cd0ae37SLisandro Dalcin #endif 15953a2046daSBarry Smith #if defined(PETSC_USE_REAL_SINGLE) 15963a2046daSBarry Smith snes->deltatol = 1.e-6; 15973a2046daSBarry Smith #else 15984b27c08aSLois Curfman McInnes snes->deltatol = 1.e-12; 15993a2046daSBarry Smith #endif 1600e37c518bSBarry Smith snes->divtol = 1.e4; 1601e37c518bSBarry Smith snes->rnorm0 = 0; 16029b94acceSBarry Smith snes->nfuncs = 0; 160350ffb88aSMatthew Knepley snes->numFailures = 0; 160450ffb88aSMatthew Knepley snes->maxFailures = 1; 16057a00f4a9SLois Curfman McInnes snes->linear_its = 0; 1606e35cf81dSBarry Smith snes->lagjacobian = 1; 160737ec4e1aSPeter Brune snes->jac_iter = 0; 160837ec4e1aSPeter Brune snes->lagjac_persist = PETSC_FALSE; 1609a8054027SBarry Smith snes->lagpreconditioner = 1; 161037ec4e1aSPeter Brune snes->pre_iter = 0; 161137ec4e1aSPeter Brune snes->lagpre_persist = PETSC_FALSE; 1612639f9d9dSBarry Smith snes->numbermonitors = 0; 16139b94acceSBarry Smith snes->data = 0; 16144dc4c822SBarry Smith snes->setupcalled = PETSC_FALSE; 1615186905e3SBarry Smith snes->ksp_ewconv = PETSC_FALSE; 16166f24a144SLois Curfman McInnes snes->nwork = 0; 161758c9b817SLisandro Dalcin snes->work = 0; 161858c9b817SLisandro Dalcin snes->nvwork = 0; 161958c9b817SLisandro Dalcin snes->vwork = 0; 1620758f92a0SBarry Smith snes->conv_hist_len = 0; 1621758f92a0SBarry Smith snes->conv_hist_max = 0; 16220298fd71SBarry Smith snes->conv_hist = NULL; 16230298fd71SBarry Smith snes->conv_hist_its = NULL; 1624758f92a0SBarry Smith snes->conv_hist_reset = PETSC_TRUE; 1625971e163fSPeter Brune snes->counters_reset = PETSC_TRUE; 1626e4ed7901SPeter Brune snes->vec_func_init_set = PETSC_FALSE; 1627184914b5SBarry Smith snes->reason = SNES_CONVERGED_ITERATING; 1628efd4aadfSBarry Smith snes->npcside = PC_RIGHT; 1629c40d0f55SPeter Brune 1630d8f46077SPeter Brune snes->mf = PETSC_FALSE; 1631d8f46077SPeter Brune snes->mf_operator = PETSC_FALSE; 1632d8f46077SPeter Brune snes->mf_version = 1; 1633d8f46077SPeter Brune 16343d4c4710SBarry Smith snes->numLinearSolveFailures = 0; 16353d4c4710SBarry Smith snes->maxLinearSolveFailures = 1; 16363d4c4710SBarry Smith 1637349187a7SBarry Smith snes->vizerotolerance = 1.e-8; 1638349187a7SBarry Smith 16394fc747eaSLawrence Mitchell /* Set this to true if the implementation of SNESSolve_XXX does compute the residual at the final solution. */ 16404fc747eaSLawrence Mitchell snes->alwayscomputesfinalresidual = PETSC_FALSE; 16414fc747eaSLawrence Mitchell 16429b94acceSBarry Smith /* Create context to compute Eisenstat-Walker relative tolerance for KSP */ 1643b00a9115SJed Brown ierr = PetscNewLog(snes,&kctx);CHKERRQ(ierr); 1644f5af7f23SKarl Rupp 16459b94acceSBarry Smith snes->kspconvctx = (void*)kctx; 16469b94acceSBarry Smith kctx->version = 2; 16479b94acceSBarry Smith kctx->rtol_0 = .3; /* Eisenstat and Walker suggest rtol_0=.5, but 16489b94acceSBarry Smith this was too large for some test cases */ 164975567043SBarry Smith kctx->rtol_last = 0.0; 16509b94acceSBarry Smith kctx->rtol_max = .9; 16519b94acceSBarry Smith kctx->gamma = 1.0; 165262d1f40fSBarry Smith kctx->alpha = .5*(1.0 + PetscSqrtReal(5.0)); 165371f87433Sdalcinl kctx->alpha2 = kctx->alpha; 16549b94acceSBarry Smith kctx->threshold = .1; 165575567043SBarry Smith kctx->lresid_last = 0.0; 165675567043SBarry Smith kctx->norm_last = 0.0; 16579b94acceSBarry Smith 16589b94acceSBarry Smith *outsnes = snes; 16593a40ed3dSBarry Smith PetscFunctionReturn(0); 16609b94acceSBarry Smith } 16619b94acceSBarry Smith 166288f0584fSBarry Smith /*MC 1663411c0326SBarry Smith SNESFunction - Functional form used to convey the nonlinear function to be solved by SNES 166488f0584fSBarry Smith 166588f0584fSBarry Smith Synopsis: 1666411c0326SBarry Smith #include "petscsnes.h" 1667411c0326SBarry Smith PetscErrorCode SNESFunction(SNES snes,Vec x,Vec f,void *ctx); 166888f0584fSBarry Smith 166988f0584fSBarry Smith Input Parameters: 167088f0584fSBarry Smith + snes - the SNES context 167188f0584fSBarry Smith . x - state at which to evaluate residual 167288f0584fSBarry Smith - ctx - optional user-defined function context, passed in with SNESSetFunction() 167388f0584fSBarry Smith 167488f0584fSBarry Smith Output Parameter: 167588f0584fSBarry Smith . f - vector to put residual (function value) 167688f0584fSBarry Smith 1677878cb397SSatish Balay Level: intermediate 1678878cb397SSatish Balay 167988f0584fSBarry Smith .seealso: SNESSetFunction(), SNESGetFunction() 168088f0584fSBarry Smith M*/ 168188f0584fSBarry Smith 16829b94acceSBarry Smith /*@C 16839b94acceSBarry Smith SNESSetFunction - Sets the function evaluation routine and function 16849b94acceSBarry Smith vector for use by the SNES routines in solving systems of nonlinear 16859b94acceSBarry Smith equations. 16869b94acceSBarry Smith 16873f9fe445SBarry Smith Logically Collective on SNES 1688fee21e36SBarry Smith 1689c7afd0dbSLois Curfman McInnes Input Parameters: 1690c7afd0dbSLois Curfman McInnes + snes - the SNES context 1691c7afd0dbSLois Curfman McInnes . r - vector to store function value 1692f8b49ee9SBarry Smith . f - function evaluation routine; see SNESFunction for calling sequence details 1693c7afd0dbSLois Curfman McInnes - ctx - [optional] user-defined context for private data for the 16940298fd71SBarry Smith function evaluation routine (may be NULL) 16959b94acceSBarry Smith 16969b94acceSBarry Smith Notes: 16979b94acceSBarry Smith The Newton-like methods typically solve linear systems of the form 16989b94acceSBarry Smith $ f'(x) x = -f(x), 1699c7afd0dbSLois Curfman McInnes where f'(x) denotes the Jacobian matrix and f(x) is the function. 17009b94acceSBarry Smith 170136851e7fSLois Curfman McInnes Level: beginner 170236851e7fSLois Curfman McInnes 17039b94acceSBarry Smith .keywords: SNES, nonlinear, set, function 17049b94acceSBarry Smith 1705bf388a1fSBarry Smith .seealso: SNESGetFunction(), SNESComputeFunction(), SNESSetJacobian(), SNESSetPicard(), SNESFunction 17069b94acceSBarry Smith @*/ 1707f8b49ee9SBarry Smith PetscErrorCode SNESSetFunction(SNES snes,Vec r,PetscErrorCode (*f)(SNES,Vec,Vec,void*),void *ctx) 17089b94acceSBarry Smith { 170985385478SLisandro Dalcin PetscErrorCode ierr; 17106cab3a1bSJed Brown DM dm; 17116cab3a1bSJed Brown 17123a40ed3dSBarry Smith PetscFunctionBegin; 17130700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 1714d2a683ecSLisandro Dalcin if (r) { 1715d2a683ecSLisandro Dalcin PetscValidHeaderSpecific(r,VEC_CLASSID,2); 1716d2a683ecSLisandro Dalcin PetscCheckSameComm(snes,1,r,2); 171785385478SLisandro Dalcin ierr = PetscObjectReference((PetscObject)r);CHKERRQ(ierr); 17186bf464f9SBarry Smith ierr = VecDestroy(&snes->vec_func);CHKERRQ(ierr); 1719f5af7f23SKarl Rupp 172085385478SLisandro Dalcin snes->vec_func = r; 1721d2a683ecSLisandro Dalcin } 17226cab3a1bSJed Brown ierr = SNESGetDM(snes,&dm);CHKERRQ(ierr); 1723f8b49ee9SBarry Smith ierr = DMSNESSetFunction(dm,f,ctx);CHKERRQ(ierr); 17243a40ed3dSBarry Smith PetscFunctionReturn(0); 17259b94acceSBarry Smith } 17269b94acceSBarry Smith 1727646217ecSPeter Brune 1728e4ed7901SPeter Brune /*@C 1729e4ed7901SPeter Brune SNESSetInitialFunction - Sets the function vector to be used as the 1730e4ed7901SPeter Brune function norm at the initialization of the method. In some 1731e4ed7901SPeter Brune instances, the user has precomputed the function before calling 1732e4ed7901SPeter Brune SNESSolve. This function allows one to avoid a redundant call 1733e4ed7901SPeter Brune to SNESComputeFunction in that case. 1734e4ed7901SPeter Brune 1735e4ed7901SPeter Brune Logically Collective on SNES 1736e4ed7901SPeter Brune 1737e4ed7901SPeter Brune Input Parameters: 1738e4ed7901SPeter Brune + snes - the SNES context 1739e4ed7901SPeter Brune - f - vector to store function value 1740e4ed7901SPeter Brune 1741e4ed7901SPeter Brune Notes: 1742e4ed7901SPeter Brune This should not be modified during the solution procedure. 1743e4ed7901SPeter Brune 1744e4ed7901SPeter Brune This is used extensively in the SNESFAS hierarchy and in nonlinear preconditioning. 1745e4ed7901SPeter Brune 1746e4ed7901SPeter Brune Level: developer 1747e4ed7901SPeter Brune 1748e4ed7901SPeter Brune .keywords: SNES, nonlinear, set, function 1749e4ed7901SPeter Brune 1750e4ed7901SPeter Brune .seealso: SNESSetFunction(), SNESComputeFunction(), SNESSetInitialFunctionNorm() 1751e4ed7901SPeter Brune @*/ 1752e4ed7901SPeter Brune PetscErrorCode SNESSetInitialFunction(SNES snes, Vec f) 1753e4ed7901SPeter Brune { 1754e4ed7901SPeter Brune PetscErrorCode ierr; 1755e4ed7901SPeter Brune Vec vec_func; 1756e4ed7901SPeter Brune 1757e4ed7901SPeter Brune PetscFunctionBegin; 1758e4ed7901SPeter Brune PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 1759e4ed7901SPeter Brune PetscValidHeaderSpecific(f,VEC_CLASSID,2); 1760e4ed7901SPeter Brune PetscCheckSameComm(snes,1,f,2); 1761efd4aadfSBarry Smith if (snes->npcside== PC_LEFT && snes->functype == SNES_FUNCTION_PRECONDITIONED) { 1762902f982fSPeter Brune snes->vec_func_init_set = PETSC_FALSE; 1763902f982fSPeter Brune PetscFunctionReturn(0); 1764902f982fSPeter Brune } 17650298fd71SBarry Smith ierr = SNESGetFunction(snes,&vec_func,NULL,NULL);CHKERRQ(ierr); 1766e4ed7901SPeter Brune ierr = VecCopy(f, vec_func);CHKERRQ(ierr); 1767f5af7f23SKarl Rupp 1768217b9c2eSPeter Brune snes->vec_func_init_set = PETSC_TRUE; 1769e4ed7901SPeter Brune PetscFunctionReturn(0); 1770e4ed7901SPeter Brune } 1771e4ed7901SPeter Brune 1772534ebe21SPeter Brune /*@ 1773365a6726SPeter Brune SNESSetNormSchedule - Sets the SNESNormSchedule used in covergence and monitoring 1774534ebe21SPeter Brune of the SNES method. 1775534ebe21SPeter Brune 1776534ebe21SPeter Brune Logically Collective on SNES 1777534ebe21SPeter Brune 1778534ebe21SPeter Brune Input Parameters: 1779534ebe21SPeter Brune + snes - the SNES context 1780365a6726SPeter Brune - normschedule - the frequency of norm computation 1781534ebe21SPeter Brune 1782517f1916SMatthew G. Knepley Options Database Key: 1783517f1916SMatthew G. Knepley . -snes_norm_schedule <none, always, initialonly, finalonly, initalfinalonly> 1784517f1916SMatthew G. Knepley 1785534ebe21SPeter Brune Notes: 1786365a6726SPeter Brune Only certain SNES methods support certain SNESNormSchedules. Most require evaluation 1787534ebe21SPeter Brune of the nonlinear function and the taking of its norm at every iteration to 1788534ebe21SPeter Brune even ensure convergence at all. However, methods such as custom Gauss-Seidel methods 1789be95d8f1SBarry Smith (SNESNGS) and the like do not require the norm of the function to be computed, and therfore 1790534ebe21SPeter Brune may either be monitored for convergence or not. As these are often used as nonlinear 1791534ebe21SPeter Brune preconditioners, monitoring the norm of their error is not a useful enterprise within 1792534ebe21SPeter Brune their solution. 1793534ebe21SPeter Brune 1794534ebe21SPeter Brune Level: developer 1795534ebe21SPeter Brune 1796534ebe21SPeter Brune .keywords: SNES, nonlinear, set, function, norm, type 1797534ebe21SPeter Brune 1798365a6726SPeter Brune .seealso: SNESGetNormSchedule(), SNESComputeFunction(), VecNorm(), SNESSetFunction(), SNESSetInitialFunction(), SNESNormSchedule 1799534ebe21SPeter Brune @*/ 1800365a6726SPeter Brune PetscErrorCode SNESSetNormSchedule(SNES snes, SNESNormSchedule normschedule) 1801534ebe21SPeter Brune { 1802534ebe21SPeter Brune PetscFunctionBegin; 1803534ebe21SPeter Brune PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 1804365a6726SPeter Brune snes->normschedule = normschedule; 1805534ebe21SPeter Brune PetscFunctionReturn(0); 1806534ebe21SPeter Brune } 1807534ebe21SPeter Brune 1808534ebe21SPeter Brune 1809534ebe21SPeter Brune /*@ 1810365a6726SPeter Brune SNESGetNormSchedule - Gets the SNESNormSchedule used in covergence and monitoring 1811534ebe21SPeter Brune of the SNES method. 1812534ebe21SPeter Brune 1813534ebe21SPeter Brune Logically Collective on SNES 1814534ebe21SPeter Brune 1815534ebe21SPeter Brune Input Parameters: 1816534ebe21SPeter Brune + snes - the SNES context 1817365a6726SPeter Brune - normschedule - the type of the norm used 1818534ebe21SPeter Brune 1819534ebe21SPeter Brune Level: advanced 1820534ebe21SPeter Brune 1821534ebe21SPeter Brune .keywords: SNES, nonlinear, set, function, norm, type 1822534ebe21SPeter Brune 1823365a6726SPeter Brune .seealso: SNESSetNormSchedule(), SNESComputeFunction(), VecNorm(), SNESSetFunction(), SNESSetInitialFunction(), SNESNormSchedule 1824534ebe21SPeter Brune @*/ 1825365a6726SPeter Brune PetscErrorCode SNESGetNormSchedule(SNES snes, SNESNormSchedule *normschedule) 1826534ebe21SPeter Brune { 1827534ebe21SPeter Brune PetscFunctionBegin; 1828534ebe21SPeter Brune PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 1829365a6726SPeter Brune *normschedule = snes->normschedule; 1830534ebe21SPeter Brune PetscFunctionReturn(0); 1831534ebe21SPeter Brune } 1832534ebe21SPeter Brune 183347073ea2SPeter Brune 1834c5ce4427SMatthew G. Knepley /*@ 1835c5ce4427SMatthew G. Knepley SNESSetFunctionNorm - Sets the last computed residual norm. 1836c5ce4427SMatthew G. Knepley 1837c5ce4427SMatthew G. Knepley Logically Collective on SNES 1838c5ce4427SMatthew G. Knepley 1839c5ce4427SMatthew G. Knepley Input Parameters: 1840c5ce4427SMatthew G. Knepley + snes - the SNES context 1841c5ce4427SMatthew G. Knepley 1842c5ce4427SMatthew G. Knepley - normschedule - the frequency of norm computation 1843c5ce4427SMatthew G. Knepley 1844c5ce4427SMatthew G. Knepley Level: developer 1845c5ce4427SMatthew G. Knepley 1846c5ce4427SMatthew G. Knepley .keywords: SNES, nonlinear, set, function, norm, type 1847c5ce4427SMatthew G. Knepley .seealso: SNESGetNormSchedule(), SNESComputeFunction(), VecNorm(), SNESSetFunction(), SNESSetInitialFunction(), SNESNormSchedule 1848c5ce4427SMatthew G. Knepley @*/ 1849c5ce4427SMatthew G. Knepley PetscErrorCode SNESSetFunctionNorm(SNES snes, PetscReal norm) 1850c5ce4427SMatthew G. Knepley { 1851c5ce4427SMatthew G. Knepley PetscFunctionBegin; 1852c5ce4427SMatthew G. Knepley PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 1853c5ce4427SMatthew G. Knepley snes->norm = norm; 1854c5ce4427SMatthew G. Knepley PetscFunctionReturn(0); 1855c5ce4427SMatthew G. Knepley } 1856c5ce4427SMatthew G. Knepley 1857c5ce4427SMatthew G. Knepley /*@ 1858c5ce4427SMatthew G. Knepley SNESGetFunctionNorm - Gets the last computed norm of the residual 1859c5ce4427SMatthew G. Knepley 1860c5ce4427SMatthew G. Knepley Not Collective 1861c5ce4427SMatthew G. Knepley 1862c5ce4427SMatthew G. Knepley Input Parameter: 1863c5ce4427SMatthew G. Knepley . snes - the SNES context 1864c5ce4427SMatthew G. Knepley 1865c5ce4427SMatthew G. Knepley Output Parameter: 1866c5ce4427SMatthew G. Knepley . norm - the last computed residual norm 1867c5ce4427SMatthew G. Knepley 1868c5ce4427SMatthew G. Knepley Level: developer 1869c5ce4427SMatthew G. Knepley 1870c5ce4427SMatthew G. Knepley .keywords: SNES, nonlinear, set, function, norm, type 1871c5ce4427SMatthew G. Knepley .seealso: SNESSetNormSchedule(), SNESComputeFunction(), VecNorm(), SNESSetFunction(), SNESSetInitialFunction(), SNESNormSchedule 1872c5ce4427SMatthew G. Knepley @*/ 1873c5ce4427SMatthew G. Knepley PetscErrorCode SNESGetFunctionNorm(SNES snes, PetscReal *norm) 1874c5ce4427SMatthew G. Knepley { 1875c5ce4427SMatthew G. Knepley PetscFunctionBegin; 1876c5ce4427SMatthew G. Knepley PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 1877c5ce4427SMatthew G. Knepley PetscValidPointer(norm, 2); 1878c5ce4427SMatthew G. Knepley *norm = snes->norm; 1879c5ce4427SMatthew G. Knepley PetscFunctionReturn(0); 1880c5ce4427SMatthew G. Knepley } 1881c5ce4427SMatthew G. Knepley 188247073ea2SPeter Brune /*@C 188347073ea2SPeter Brune SNESSetFunctionType - Sets the SNESNormSchedule used in covergence and monitoring 188447073ea2SPeter Brune of the SNES method. 188547073ea2SPeter Brune 188647073ea2SPeter Brune Logically Collective on SNES 188747073ea2SPeter Brune 188847073ea2SPeter Brune Input Parameters: 188947073ea2SPeter Brune + snes - the SNES context 189047073ea2SPeter Brune - normschedule - the frequency of norm computation 189147073ea2SPeter Brune 189247073ea2SPeter Brune Notes: 189347073ea2SPeter Brune Only certain SNES methods support certain SNESNormSchedules. Most require evaluation 189447073ea2SPeter Brune of the nonlinear function and the taking of its norm at every iteration to 189547073ea2SPeter Brune even ensure convergence at all. However, methods such as custom Gauss-Seidel methods 1896be95d8f1SBarry Smith (SNESNGS) and the like do not require the norm of the function to be computed, and therfore 189747073ea2SPeter Brune may either be monitored for convergence or not. As these are often used as nonlinear 189847073ea2SPeter Brune preconditioners, monitoring the norm of their error is not a useful enterprise within 189947073ea2SPeter Brune their solution. 190047073ea2SPeter Brune 190147073ea2SPeter Brune Level: developer 190247073ea2SPeter Brune 190347073ea2SPeter Brune .keywords: SNES, nonlinear, set, function, norm, type 190447073ea2SPeter Brune 190547073ea2SPeter Brune .seealso: SNESGetNormSchedule(), SNESComputeFunction(), VecNorm(), SNESSetFunction(), SNESSetInitialFunction(), SNESNormSchedule 190647073ea2SPeter Brune @*/ 190747073ea2SPeter Brune PetscErrorCode SNESSetFunctionType(SNES snes, SNESFunctionType type) 190847073ea2SPeter Brune { 190947073ea2SPeter Brune PetscFunctionBegin; 191047073ea2SPeter Brune PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 191147073ea2SPeter Brune snes->functype = type; 191247073ea2SPeter Brune PetscFunctionReturn(0); 191347073ea2SPeter Brune } 191447073ea2SPeter Brune 191547073ea2SPeter Brune 191647073ea2SPeter Brune /*@C 191747073ea2SPeter Brune SNESGetFunctionType - Gets the SNESNormSchedule used in covergence and monitoring 191847073ea2SPeter Brune of the SNES method. 191947073ea2SPeter Brune 192047073ea2SPeter Brune Logically Collective on SNES 192147073ea2SPeter Brune 192247073ea2SPeter Brune Input Parameters: 192347073ea2SPeter Brune + snes - the SNES context 192447073ea2SPeter Brune - normschedule - the type of the norm used 192547073ea2SPeter Brune 192647073ea2SPeter Brune Level: advanced 192747073ea2SPeter Brune 192847073ea2SPeter Brune .keywords: SNES, nonlinear, set, function, norm, type 192947073ea2SPeter Brune 193047073ea2SPeter Brune .seealso: SNESSetNormSchedule(), SNESComputeFunction(), VecNorm(), SNESSetFunction(), SNESSetInitialFunction(), SNESNormSchedule 193147073ea2SPeter Brune @*/ 193247073ea2SPeter Brune PetscErrorCode SNESGetFunctionType(SNES snes, SNESFunctionType *type) 193347073ea2SPeter Brune { 193447073ea2SPeter Brune PetscFunctionBegin; 193547073ea2SPeter Brune PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 193647073ea2SPeter Brune *type = snes->functype; 1937534ebe21SPeter Brune PetscFunctionReturn(0); 1938534ebe21SPeter Brune } 1939534ebe21SPeter Brune 1940bf388a1fSBarry Smith /*MC 1941be95d8f1SBarry Smith SNESNGSFunction - function used to convey a Gauss-Seidel sweep on the nonlinear function 1942bf388a1fSBarry Smith 1943bf388a1fSBarry Smith Synopsis: 1944aaa7dc30SBarry Smith #include <petscsnes.h> 1945be95d8f1SBarry Smith $ SNESNGSFunction(SNES snes,Vec x,Vec b,void *ctx); 1946bf388a1fSBarry Smith 1947bf388a1fSBarry Smith + X - solution vector 1948bf388a1fSBarry Smith . B - RHS vector 1949bf388a1fSBarry Smith - ctx - optional user-defined Gauss-Seidel context 1950bf388a1fSBarry Smith 1951878cb397SSatish Balay Level: intermediate 1952878cb397SSatish Balay 1953be95d8f1SBarry Smith .seealso: SNESSetNGS(), SNESGetNGS() 1954bf388a1fSBarry Smith M*/ 1955bf388a1fSBarry Smith 1956c79ef259SPeter Brune /*@C 1957be95d8f1SBarry Smith SNESSetNGS - Sets the user nonlinear Gauss-Seidel routine for 1958c79ef259SPeter Brune use with composed nonlinear solvers. 1959c79ef259SPeter Brune 1960c79ef259SPeter Brune Input Parameters: 1961c79ef259SPeter Brune + snes - the SNES context 1962be95d8f1SBarry Smith . f - function evaluation routine to apply Gauss-Seidel see SNESNGSFunction 1963c79ef259SPeter Brune - ctx - [optional] user-defined context for private data for the 19640298fd71SBarry Smith smoother evaluation routine (may be NULL) 1965c79ef259SPeter Brune 1966c79ef259SPeter Brune Notes: 1967be95d8f1SBarry Smith The NGS routines are used by the composed nonlinear solver to generate 1968c79ef259SPeter Brune a problem appropriate update to the solution, particularly FAS. 1969c79ef259SPeter Brune 1970d28543b3SPeter Brune Level: intermediate 1971c79ef259SPeter Brune 1972d28543b3SPeter Brune .keywords: SNES, nonlinear, set, Gauss-Seidel 1973c79ef259SPeter Brune 1974be95d8f1SBarry Smith .seealso: SNESGetFunction(), SNESComputeNGS() 1975c79ef259SPeter Brune @*/ 1976be95d8f1SBarry Smith PetscErrorCode SNESSetNGS(SNES snes,PetscErrorCode (*f)(SNES,Vec,Vec,void*),void *ctx) 19776cab3a1bSJed Brown { 19786cab3a1bSJed Brown PetscErrorCode ierr; 19796cab3a1bSJed Brown DM dm; 19806cab3a1bSJed Brown 1981646217ecSPeter Brune PetscFunctionBegin; 19826cab3a1bSJed Brown PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 19836cab3a1bSJed Brown ierr = SNESGetDM(snes,&dm);CHKERRQ(ierr); 1984be95d8f1SBarry Smith ierr = DMSNESSetNGS(dm,f,ctx);CHKERRQ(ierr); 1985646217ecSPeter Brune PetscFunctionReturn(0); 1986646217ecSPeter Brune } 1987646217ecSPeter Brune 198882b59a81SJed Brown PETSC_EXTERN PetscErrorCode SNESPicardComputeFunction(SNES snes,Vec x,Vec f,void *ctx) 19898b0a5094SBarry Smith { 19908b0a5094SBarry Smith PetscErrorCode ierr; 1991e03ab78fSPeter Brune DM dm; 1992942e3340SBarry Smith DMSNES sdm; 19936cab3a1bSJed Brown 19948b0a5094SBarry Smith PetscFunctionBegin; 1995e03ab78fSPeter Brune ierr = SNESGetDM(snes,&dm);CHKERRQ(ierr); 1996942e3340SBarry Smith ierr = DMGetDMSNES(dm,&sdm);CHKERRQ(ierr); 19978b0a5094SBarry Smith /* A(x)*x - b(x) */ 199822c6f798SBarry Smith if (sdm->ops->computepfunction) { 199922c6f798SBarry Smith ierr = (*sdm->ops->computepfunction)(snes,x,f,sdm->pctx);CHKERRQ(ierr); 200022c6f798SBarry Smith } else SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE, "Must call SNESSetPicard() to provide Picard function."); 2001e03ab78fSPeter Brune 200222c6f798SBarry Smith if (sdm->ops->computepjacobian) { 2003d1e9a80fSBarry Smith ierr = (*sdm->ops->computepjacobian)(snes,x,snes->jacobian,snes->jacobian_pre,sdm->pctx);CHKERRQ(ierr); 200474e1e8c1SBarry Smith } else SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE, "Must call SNESSetPicard() to provide Picard matrix."); 20058b0a5094SBarry Smith ierr = VecScale(f,-1.0);CHKERRQ(ierr); 200695eabcedSBarry Smith ierr = MatMultAdd(snes->jacobian,x,f,f);CHKERRQ(ierr); 20078b0a5094SBarry Smith PetscFunctionReturn(0); 20088b0a5094SBarry Smith } 20098b0a5094SBarry Smith 2010d1e9a80fSBarry Smith PETSC_EXTERN PetscErrorCode SNESPicardComputeJacobian(SNES snes,Vec x1,Mat J,Mat B,void *ctx) 20118b0a5094SBarry Smith { 20128b0a5094SBarry Smith PetscFunctionBegin; 2013e03ab78fSPeter Brune /* the jacobian matrix should be pre-filled in SNESPicardComputeFunction */ 20148b0a5094SBarry Smith PetscFunctionReturn(0); 20158b0a5094SBarry Smith } 20168b0a5094SBarry Smith 20178b0a5094SBarry Smith /*@C 20180d04baf8SBarry Smith SNESSetPicard - Use SNES to solve the semilinear-system A(x) x = b(x) via a Picard type iteration (Picard linearization) 20198b0a5094SBarry Smith 20208b0a5094SBarry Smith Logically Collective on SNES 20218b0a5094SBarry Smith 20228b0a5094SBarry Smith Input Parameters: 20238b0a5094SBarry Smith + snes - the SNES context 20248b0a5094SBarry Smith . r - vector to store function value 2025f8b49ee9SBarry Smith . b - function evaluation routine 2026e5d3d808SBarry Smith . Amat - matrix with which A(x) x - b(x) is to be computed 2027e5d3d808SBarry Smith . Pmat - matrix from which preconditioner is computed (usually the same as Amat) 2028411c0326SBarry Smith . J - function to compute matrix value, see SNESJacobianFunction for details on its calling sequence 20298b0a5094SBarry Smith - ctx - [optional] user-defined context for private data for the 20300298fd71SBarry Smith function evaluation routine (may be NULL) 20318b0a5094SBarry Smith 20328b0a5094SBarry Smith Notes: 2033f450aa47SBarry Smith We do not recomemend using this routine. It is far better to provide the nonlinear function F() and some approximation to the Jacobian and use 2034f450aa47SBarry Smith an approximate Newton solver. This interface is provided to allow porting/testing a previous Picard based code in PETSc before converting it to approximate Newton. 2035f450aa47SBarry Smith 20368b0a5094SBarry Smith One can call SNESSetPicard() or SNESSetFunction() (and possibly SNESSetJacobian()) but cannot call both 20378b0a5094SBarry Smith 20388b0a5094SBarry Smith $ Solves the equation A(x) x = b(x) via the defect correction algorithm A(x^{n}) (x^{n+1} - x^{n}) = b(x^{n}) - A(x^{n})x^{n} 20398b0a5094SBarry Smith $ Note that when an exact solver is used this corresponds to the "classic" Picard A(x^{n}) x^{n+1} = b(x^{n}) iteration. 20408b0a5094SBarry Smith 20418b0a5094SBarry Smith Run with -snes_mf_operator to solve the system with Newton's method using A(x^{n}) to construct the preconditioner. 20428b0a5094SBarry Smith 20430d04baf8SBarry Smith We implement the defect correction form of the Picard iteration because it converges much more generally when inexact linear solvers are used then 20440d04baf8SBarry Smith the direct Picard iteration A(x^n) x^{n+1} = b(x^n) 20458b0a5094SBarry Smith 20468b0a5094SBarry Smith There is some controversity over the definition of a Picard iteration for nonlinear systems but almost everyone agrees that it involves a linear solve and some 20478b0a5094SBarry Smith believe it is the iteration A(x^{n}) x^{n+1} = b(x^{n}) hence we use the name Picard. If anyone has an authoritative reference that defines the Picard iteration 20488b0a5094SBarry Smith different please contact us at petsc-dev@mcs.anl.gov and we'll have an entirely new argument :-). 20498b0a5094SBarry Smith 2050f450aa47SBarry Smith Level: intermediate 20518b0a5094SBarry Smith 20528b0a5094SBarry Smith .keywords: SNES, nonlinear, set, function 20538b0a5094SBarry Smith 2054411c0326SBarry Smith .seealso: SNESGetFunction(), SNESSetFunction(), SNESComputeFunction(), SNESSetJacobian(), SNESGetPicard(), SNESLineSearchPreCheckPicard(), SNESJacobianFunction 20558b0a5094SBarry Smith @*/ 2056d1e9a80fSBarry Smith PetscErrorCode SNESSetPicard(SNES snes,Vec r,PetscErrorCode (*b)(SNES,Vec,Vec,void*),Mat Amat, Mat Pmat, PetscErrorCode (*J)(SNES,Vec,Mat,Mat,void*),void *ctx) 20578b0a5094SBarry Smith { 20588b0a5094SBarry Smith PetscErrorCode ierr; 2059e03ab78fSPeter Brune DM dm; 2060e03ab78fSPeter Brune 20618b0a5094SBarry Smith PetscFunctionBegin; 20628b0a5094SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 2063e03ab78fSPeter Brune ierr = SNESGetDM(snes, &dm);CHKERRQ(ierr); 2064f8b49ee9SBarry Smith ierr = DMSNESSetPicard(dm,b,J,ctx);CHKERRQ(ierr); 20658b0a5094SBarry Smith ierr = SNESSetFunction(snes,r,SNESPicardComputeFunction,ctx);CHKERRQ(ierr); 2066e5d3d808SBarry Smith ierr = SNESSetJacobian(snes,Amat,Pmat,SNESPicardComputeJacobian,ctx);CHKERRQ(ierr); 20678b0a5094SBarry Smith PetscFunctionReturn(0); 20688b0a5094SBarry Smith } 20698b0a5094SBarry Smith 20707971a8bfSPeter Brune /*@C 20717971a8bfSPeter Brune SNESGetPicard - Returns the context for the Picard iteration 20727971a8bfSPeter Brune 20737971a8bfSPeter Brune Not Collective, but Vec is parallel if SNES is parallel. Collective if Vec is requested, but has not been created yet. 20747971a8bfSPeter Brune 20757971a8bfSPeter Brune Input Parameter: 20767971a8bfSPeter Brune . snes - the SNES context 20777971a8bfSPeter Brune 20787971a8bfSPeter Brune Output Parameter: 20790298fd71SBarry Smith + r - the function (or NULL) 2080f8b49ee9SBarry Smith . f - the function (or NULL); see SNESFunction for calling sequence details 2081e4357dc4SBarry Smith . Amat - the matrix used to defined the operation A(x) x - b(x) (or NULL) 2082e4357dc4SBarry Smith . Pmat - the matrix from which the preconditioner will be constructed (or NULL) 2083f8b49ee9SBarry Smith . J - the function for matrix evaluation (or NULL); see SNESJacobianFunction for calling sequence details 20840298fd71SBarry Smith - ctx - the function context (or NULL) 20857971a8bfSPeter Brune 20867971a8bfSPeter Brune Level: advanced 20877971a8bfSPeter Brune 20887971a8bfSPeter Brune .keywords: SNES, nonlinear, get, function 20897971a8bfSPeter Brune 2090e4357dc4SBarry Smith .seealso: SNESSetPicard(), SNESGetFunction(), SNESGetJacobian(), SNESGetDM(), SNESFunction, SNESJacobianFunction 20917971a8bfSPeter Brune @*/ 2092d1e9a80fSBarry Smith PetscErrorCode SNESGetPicard(SNES snes,Vec *r,PetscErrorCode (**f)(SNES,Vec,Vec,void*),Mat *Amat, Mat *Pmat, PetscErrorCode (**J)(SNES,Vec,Mat,Mat,void*),void **ctx) 20937971a8bfSPeter Brune { 20947971a8bfSPeter Brune PetscErrorCode ierr; 20957971a8bfSPeter Brune DM dm; 20967971a8bfSPeter Brune 20977971a8bfSPeter Brune PetscFunctionBegin; 20987971a8bfSPeter Brune PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 20990298fd71SBarry Smith ierr = SNESGetFunction(snes,r,NULL,NULL);CHKERRQ(ierr); 2100e4357dc4SBarry Smith ierr = SNESGetJacobian(snes,Amat,Pmat,NULL,NULL);CHKERRQ(ierr); 21017971a8bfSPeter Brune ierr = SNESGetDM(snes,&dm);CHKERRQ(ierr); 2102f8b49ee9SBarry Smith ierr = DMSNESGetPicard(dm,f,J,ctx);CHKERRQ(ierr); 21037971a8bfSPeter Brune PetscFunctionReturn(0); 21047971a8bfSPeter Brune } 21057971a8bfSPeter Brune 2106d25893d9SBarry Smith /*@C 2107d25893d9SBarry Smith SNESSetComputeInitialGuess - Sets a routine used to compute an initial guess for the problem 2108d25893d9SBarry Smith 2109d25893d9SBarry Smith Logically Collective on SNES 2110d25893d9SBarry Smith 2111d25893d9SBarry Smith Input Parameters: 2112d25893d9SBarry Smith + snes - the SNES context 2113d25893d9SBarry Smith . func - function evaluation routine 2114d25893d9SBarry Smith - ctx - [optional] user-defined context for private data for the 21150298fd71SBarry Smith function evaluation routine (may be NULL) 2116d25893d9SBarry Smith 2117d25893d9SBarry Smith Calling sequence of func: 2118d25893d9SBarry Smith $ func (SNES snes,Vec x,void *ctx); 2119d25893d9SBarry Smith 2120d25893d9SBarry Smith . f - function vector 2121d25893d9SBarry Smith - ctx - optional user-defined function context 2122d25893d9SBarry Smith 2123d25893d9SBarry Smith Level: intermediate 2124d25893d9SBarry Smith 2125d25893d9SBarry Smith .keywords: SNES, nonlinear, set, function 2126d25893d9SBarry Smith 2127d25893d9SBarry Smith .seealso: SNESGetFunction(), SNESComputeFunction(), SNESSetJacobian() 2128d25893d9SBarry Smith @*/ 2129d25893d9SBarry Smith PetscErrorCode SNESSetComputeInitialGuess(SNES snes,PetscErrorCode (*func)(SNES,Vec,void*),void *ctx) 2130d25893d9SBarry Smith { 2131d25893d9SBarry Smith PetscFunctionBegin; 2132d25893d9SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 2133d25893d9SBarry Smith if (func) snes->ops->computeinitialguess = func; 2134d25893d9SBarry Smith if (ctx) snes->initialguessP = ctx; 2135d25893d9SBarry Smith PetscFunctionReturn(0); 2136d25893d9SBarry Smith } 2137d25893d9SBarry Smith 21383ab0aad5SBarry Smith /* --------------------------------------------------------------- */ 21391096aae1SMatthew Knepley /*@C 21401096aae1SMatthew Knepley SNESGetRhs - Gets the vector for solving F(x) = rhs. If rhs is not set 21411096aae1SMatthew Knepley it assumes a zero right hand side. 21421096aae1SMatthew Knepley 21433f9fe445SBarry Smith Logically Collective on SNES 21441096aae1SMatthew Knepley 21451096aae1SMatthew Knepley Input Parameter: 21461096aae1SMatthew Knepley . snes - the SNES context 21471096aae1SMatthew Knepley 21481096aae1SMatthew Knepley Output Parameter: 21490298fd71SBarry Smith . rhs - the right hand side vector or NULL if the right hand side vector is null 21501096aae1SMatthew Knepley 21511096aae1SMatthew Knepley Level: intermediate 21521096aae1SMatthew Knepley 21531096aae1SMatthew Knepley .keywords: SNES, nonlinear, get, function, right hand side 21541096aae1SMatthew Knepley 215585385478SLisandro Dalcin .seealso: SNESGetSolution(), SNESGetFunction(), SNESComputeFunction(), SNESSetJacobian(), SNESSetFunction() 21561096aae1SMatthew Knepley @*/ 21577087cfbeSBarry Smith PetscErrorCode SNESGetRhs(SNES snes,Vec *rhs) 21581096aae1SMatthew Knepley { 21591096aae1SMatthew Knepley PetscFunctionBegin; 21600700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 21611096aae1SMatthew Knepley PetscValidPointer(rhs,2); 216285385478SLisandro Dalcin *rhs = snes->vec_rhs; 21631096aae1SMatthew Knepley PetscFunctionReturn(0); 21641096aae1SMatthew Knepley } 21651096aae1SMatthew Knepley 21669b94acceSBarry Smith /*@ 2167bf388a1fSBarry Smith SNESComputeFunction - Calls the function that has been set with SNESSetFunction(). 21689b94acceSBarry Smith 2169c7afd0dbSLois Curfman McInnes Collective on SNES 2170c7afd0dbSLois Curfman McInnes 21719b94acceSBarry Smith Input Parameters: 2172c7afd0dbSLois Curfman McInnes + snes - the SNES context 2173c7afd0dbSLois Curfman McInnes - x - input vector 21749b94acceSBarry Smith 21759b94acceSBarry Smith Output Parameter: 21763638b69dSLois Curfman McInnes . y - function vector, as set by SNESSetFunction() 21779b94acceSBarry Smith 21781bffabb2SLois Curfman McInnes Notes: 217936851e7fSLois Curfman McInnes SNESComputeFunction() is typically used within nonlinear solvers 218036851e7fSLois Curfman McInnes implementations, so most users would not generally call this routine 218136851e7fSLois Curfman McInnes themselves. 218236851e7fSLois Curfman McInnes 218336851e7fSLois Curfman McInnes Level: developer 218436851e7fSLois Curfman McInnes 21859b94acceSBarry Smith .keywords: SNES, nonlinear, compute, function 21869b94acceSBarry Smith 2187a86d99e1SLois Curfman McInnes .seealso: SNESSetFunction(), SNESGetFunction() 21889b94acceSBarry Smith @*/ 21897087cfbeSBarry Smith PetscErrorCode SNESComputeFunction(SNES snes,Vec x,Vec y) 21909b94acceSBarry Smith { 2191dfbe8321SBarry Smith PetscErrorCode ierr; 21926cab3a1bSJed Brown DM dm; 2193942e3340SBarry Smith DMSNES sdm; 21949b94acceSBarry Smith 21953a40ed3dSBarry Smith PetscFunctionBegin; 21960700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 21970700a824SBarry Smith PetscValidHeaderSpecific(x,VEC_CLASSID,2); 21980700a824SBarry Smith PetscValidHeaderSpecific(y,VEC_CLASSID,3); 2199c9780b6fSBarry Smith PetscCheckSameComm(snes,1,x,2); 2200c9780b6fSBarry Smith PetscCheckSameComm(snes,1,y,3); 220162796dfbSBarry Smith ierr = VecValidValues(x,2,PETSC_TRUE);CHKERRQ(ierr); 2202184914b5SBarry Smith 22036cab3a1bSJed Brown ierr = SNESGetDM(snes,&dm);CHKERRQ(ierr); 2204942e3340SBarry Smith ierr = DMGetDMSNES(dm,&sdm);CHKERRQ(ierr); 220532f3f7c2SPeter Brune if (sdm->ops->computefunction) { 220694db00ebSBarry Smith if (sdm->ops->computefunction != SNESObjectiveComputeFunctionDefaultFD) { 2207ccf3c845SPeter Brune ierr = PetscLogEventBegin(SNES_FunctionEval,snes,x,y,0);CHKERRQ(ierr); 220894db00ebSBarry Smith } 22095edff71fSBarry Smith ierr = VecLockPush(x);CHKERRQ(ierr); 2210d64ed03dSBarry Smith PetscStackPush("SNES user function"); 221122c6f798SBarry Smith ierr = (*sdm->ops->computefunction)(snes,x,y,sdm->functionctx);CHKERRQ(ierr); 2212d64ed03dSBarry Smith PetscStackPop; 22135edff71fSBarry Smith ierr = VecLockPop(x);CHKERRQ(ierr); 221494db00ebSBarry Smith if (sdm->ops->computefunction != SNESObjectiveComputeFunctionDefaultFD) { 2215ccf3c845SPeter Brune ierr = PetscLogEventEnd(SNES_FunctionEval,snes,x,y,0);CHKERRQ(ierr); 221694db00ebSBarry Smith } 2217c90fad12SPeter Brune } else if (snes->vec_rhs) { 2218c90fad12SPeter Brune ierr = MatMult(snes->jacobian, x, y);CHKERRQ(ierr); 2219644e2e5bSBarry Smith } else SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE, "Must call SNESSetFunction() or SNESSetDM() before SNESComputeFunction(), likely called from SNESSolve()."); 222085385478SLisandro Dalcin if (snes->vec_rhs) { 222185385478SLisandro Dalcin ierr = VecAXPY(y,-1.0,snes->vec_rhs);CHKERRQ(ierr); 22223ab0aad5SBarry Smith } 2223ae3c334cSLois Curfman McInnes snes->nfuncs++; 2224422a814eSBarry Smith /* 2225422a814eSBarry Smith domainerror might not be set on all processes; so we tag vector locally with Inf and the next inner product or norm will 2226422a814eSBarry Smith propagate the value to all processes 2227422a814eSBarry Smith */ 2228422a814eSBarry Smith if (snes->domainerror) { 2229422a814eSBarry Smith ierr = VecSetInf(y);CHKERRQ(ierr); 2230422a814eSBarry Smith } 22313a40ed3dSBarry Smith PetscFunctionReturn(0); 22329b94acceSBarry Smith } 22339b94acceSBarry Smith 2234c79ef259SPeter Brune /*@ 2235be95d8f1SBarry Smith SNESComputeNGS - Calls the Gauss-Seidel function that has been set with SNESSetNGS(). 2236c79ef259SPeter Brune 2237c79ef259SPeter Brune Collective on SNES 2238c79ef259SPeter Brune 2239c79ef259SPeter Brune Input Parameters: 2240c79ef259SPeter Brune + snes - the SNES context 2241c79ef259SPeter Brune . x - input vector 2242c79ef259SPeter Brune - b - rhs vector 2243c79ef259SPeter Brune 2244c79ef259SPeter Brune Output Parameter: 2245c79ef259SPeter Brune . x - new solution vector 2246c79ef259SPeter Brune 2247c79ef259SPeter Brune Notes: 2248be95d8f1SBarry Smith SNESComputeNGS() is typically used within composed nonlinear solver 2249c79ef259SPeter Brune implementations, so most users would not generally call this routine 2250c79ef259SPeter Brune themselves. 2251c79ef259SPeter Brune 2252c79ef259SPeter Brune Level: developer 2253c79ef259SPeter Brune 2254c79ef259SPeter Brune .keywords: SNES, nonlinear, compute, function 2255c79ef259SPeter Brune 2256be95d8f1SBarry Smith .seealso: SNESSetNGS(), SNESComputeFunction() 2257c79ef259SPeter Brune @*/ 2258be95d8f1SBarry Smith PetscErrorCode SNESComputeNGS(SNES snes,Vec b,Vec x) 2259646217ecSPeter Brune { 2260646217ecSPeter Brune PetscErrorCode ierr; 22616cab3a1bSJed Brown DM dm; 2262942e3340SBarry Smith DMSNES sdm; 2263646217ecSPeter Brune 2264646217ecSPeter Brune PetscFunctionBegin; 2265646217ecSPeter Brune PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 2266646217ecSPeter Brune PetscValidHeaderSpecific(x,VEC_CLASSID,2); 2267646217ecSPeter Brune if (b) PetscValidHeaderSpecific(b,VEC_CLASSID,3); 2268646217ecSPeter Brune PetscCheckSameComm(snes,1,x,2); 2269646217ecSPeter Brune if (b) PetscCheckSameComm(snes,1,b,3); 227062796dfbSBarry Smith if (b) {ierr = VecValidValues(b,2,PETSC_TRUE);CHKERRQ(ierr);} 2271be95d8f1SBarry Smith ierr = PetscLogEventBegin(SNES_NGSEval,snes,x,b,0);CHKERRQ(ierr); 22726cab3a1bSJed Brown ierr = SNESGetDM(snes,&dm);CHKERRQ(ierr); 2273942e3340SBarry Smith ierr = DMGetDMSNES(dm,&sdm);CHKERRQ(ierr); 227422c6f798SBarry Smith if (sdm->ops->computegs) { 22755edff71fSBarry Smith if (b) {ierr = VecLockPush(b);CHKERRQ(ierr);} 2276be95d8f1SBarry Smith PetscStackPush("SNES user NGS"); 227722c6f798SBarry Smith ierr = (*sdm->ops->computegs)(snes,x,b,sdm->gsctx);CHKERRQ(ierr); 2278646217ecSPeter Brune PetscStackPop; 22795edff71fSBarry Smith if (b) {ierr = VecLockPop(b);CHKERRQ(ierr);} 2280be95d8f1SBarry Smith } else SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE, "Must call SNESSetNGS() before SNESComputeNGS(), likely called from SNESSolve()."); 2281be95d8f1SBarry Smith ierr = PetscLogEventEnd(SNES_NGSEval,snes,x,b,0);CHKERRQ(ierr); 2282646217ecSPeter Brune PetscFunctionReturn(0); 2283646217ecSPeter Brune } 2284646217ecSPeter Brune 228562fef451SLois Curfman McInnes /*@ 2286bf388a1fSBarry Smith SNESComputeJacobian - Computes the Jacobian matrix that has been set with SNESSetJacobian(). 228762fef451SLois Curfman McInnes 2288c7afd0dbSLois Curfman McInnes Collective on SNES and Mat 2289c7afd0dbSLois Curfman McInnes 229062fef451SLois Curfman McInnes Input Parameters: 2291c7afd0dbSLois Curfman McInnes + snes - the SNES context 2292c7afd0dbSLois Curfman McInnes - x - input vector 229362fef451SLois Curfman McInnes 229462fef451SLois Curfman McInnes Output Parameters: 2295c7afd0dbSLois Curfman McInnes + A - Jacobian matrix 2296d1e9a80fSBarry Smith - B - optional preconditioning matrix 2297fee21e36SBarry Smith 2298e35cf81dSBarry Smith Options Database Keys: 2299e35cf81dSBarry Smith + -snes_lag_preconditioner <lag> 2300693365a8SJed Brown . -snes_lag_jacobian <lag> 2301693365a8SJed Brown . -snes_compare_explicit - Compare the computed Jacobian to the finite difference Jacobian and output the differences 2302693365a8SJed Brown . -snes_compare_explicit_draw - Compare the computed Jacobian to the finite difference Jacobian and draw the result 2303693365a8SJed Brown . -snes_compare_explicit_contour - Compare the computed Jacobian to the finite difference Jacobian and draw a contour plot with the result 23044c30e9fbSJed Brown . -snes_compare_operator - Make the comparison options above use the operator instead of the preconditioning matrix 230594d6a431SBarry Smith . -snes_compare_coloring - Compute the finite difference Jacobian using coloring and display norms of difference 2306c01495d3SJed Brown . -snes_compare_coloring_display - Compute the finite differece Jacobian using coloring and display verbose differences 2307c01495d3SJed Brown . -snes_compare_coloring_threshold - Display only those matrix entries that differ by more than a given threshold 2308c01495d3SJed Brown . -snes_compare_coloring_threshold_atol - Absolute tolerance for difference in matrix entries to be displayed by -snes_compare_coloring_threshold 2309c01495d3SJed Brown . -snes_compare_coloring_threshold_rtol - Relative tolerance for difference in matrix entries to be displayed by -snes_compare_coloring_threshold 23104c30e9fbSJed Brown . -snes_compare_coloring_draw - Compute the finite differece Jacobian using coloring and draw differences 2311c01495d3SJed Brown - -snes_compare_coloring_draw_contour - Compute the finite differece Jacobian using coloring and show contours of matrices and differences 2312c01495d3SJed Brown 2313e35cf81dSBarry Smith 231462fef451SLois Curfman McInnes Notes: 231562fef451SLois Curfman McInnes Most users should not need to explicitly call this routine, as it 231662fef451SLois Curfman McInnes is used internally within the nonlinear solvers. 231762fef451SLois Curfman McInnes 231836851e7fSLois Curfman McInnes Level: developer 231936851e7fSLois Curfman McInnes 232062fef451SLois Curfman McInnes .keywords: SNES, compute, Jacobian, matrix 232162fef451SLois Curfman McInnes 2322e35cf81dSBarry Smith .seealso: SNESSetJacobian(), KSPSetOperators(), MatStructure, SNESSetLagPreconditioner(), SNESSetLagJacobian() 232362fef451SLois Curfman McInnes @*/ 2324d1e9a80fSBarry Smith PetscErrorCode SNESComputeJacobian(SNES snes,Vec X,Mat A,Mat B) 23259b94acceSBarry Smith { 2326dfbe8321SBarry Smith PetscErrorCode ierr; 2327ace3abfcSBarry Smith PetscBool flag; 23286cab3a1bSJed Brown DM dm; 2329942e3340SBarry Smith DMSNES sdm; 2330e0e3a89bSBarry Smith KSP ksp; 23313a40ed3dSBarry Smith 23323a40ed3dSBarry Smith PetscFunctionBegin; 23330700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 23340700a824SBarry Smith PetscValidHeaderSpecific(X,VEC_CLASSID,2); 2335c9780b6fSBarry Smith PetscCheckSameComm(snes,1,X,2); 233662796dfbSBarry Smith ierr = VecValidValues(X,2,PETSC_TRUE);CHKERRQ(ierr); 23376cab3a1bSJed Brown ierr = SNESGetDM(snes,&dm);CHKERRQ(ierr); 2338942e3340SBarry Smith ierr = DMGetDMSNES(dm,&sdm);CHKERRQ(ierr); 23393232da50SPeter Brune 2340ce94432eSBarry Smith if (!sdm->ops->computejacobian) SETERRQ(PetscObjectComm((PetscObject)snes),PETSC_ERR_USER,"Must call SNESSetJacobian(), DMSNESSetJacobian(), DMDASNESSetJacobianLocal(), etc"); 2341ebd3b9afSBarry Smith 2342ebd3b9afSBarry Smith /* make sure that MatAssemblyBegin/End() is called on A matrix if it is matrix free */ 2343ebd3b9afSBarry Smith 2344fe3ffe1eSBarry Smith if (snes->lagjacobian == -2) { 2345fe3ffe1eSBarry Smith snes->lagjacobian = -1; 2346f5af7f23SKarl Rupp 2347fe3ffe1eSBarry Smith ierr = PetscInfo(snes,"Recomputing Jacobian/preconditioner because lag is -2 (means compute Jacobian, but then never again) \n");CHKERRQ(ierr); 2348fe3ffe1eSBarry Smith } else if (snes->lagjacobian == -1) { 2349e35cf81dSBarry Smith ierr = PetscInfo(snes,"Reusing Jacobian/preconditioner because lag is -1\n");CHKERRQ(ierr); 235094ab13aaSBarry Smith ierr = PetscObjectTypeCompare((PetscObject)A,MATMFFD,&flag);CHKERRQ(ierr); 2351ebd3b9afSBarry Smith if (flag) { 235294ab13aaSBarry Smith ierr = MatAssemblyBegin(A,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 235394ab13aaSBarry Smith ierr = MatAssemblyEnd(A,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 2354ebd3b9afSBarry Smith } 2355e35cf81dSBarry Smith PetscFunctionReturn(0); 235637ec4e1aSPeter Brune } else if (snes->lagjacobian > 1 && (snes->iter + snes->jac_iter) % snes->lagjacobian) { 2357e35cf81dSBarry Smith ierr = PetscInfo2(snes,"Reusing Jacobian/preconditioner because lag is %D and SNES iteration is %D\n",snes->lagjacobian,snes->iter);CHKERRQ(ierr); 235894ab13aaSBarry Smith ierr = PetscObjectTypeCompare((PetscObject)A,MATMFFD,&flag);CHKERRQ(ierr); 2359ebd3b9afSBarry Smith if (flag) { 236094ab13aaSBarry Smith ierr = MatAssemblyBegin(A,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 236194ab13aaSBarry Smith ierr = MatAssemblyEnd(A,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 2362ebd3b9afSBarry Smith } 2363e35cf81dSBarry Smith PetscFunctionReturn(0); 2364e35cf81dSBarry Smith } 2365efd4aadfSBarry Smith if (snes->npc && snes->npcside== PC_LEFT) { 236694ab13aaSBarry Smith ierr = MatAssemblyBegin(A,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 236794ab13aaSBarry Smith ierr = MatAssemblyEnd(A,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 2368d728fb7dSPeter Brune PetscFunctionReturn(0); 2369d728fb7dSPeter Brune } 2370e35cf81dSBarry Smith 237194ab13aaSBarry Smith ierr = PetscLogEventBegin(SNES_JacobianEval,snes,X,A,B);CHKERRQ(ierr); 23725edff71fSBarry Smith ierr = VecLockPush(X);CHKERRQ(ierr); 2373d64ed03dSBarry Smith PetscStackPush("SNES user Jacobian function"); 2374d1e9a80fSBarry Smith ierr = (*sdm->ops->computejacobian)(snes,X,A,B,sdm->jacobianctx);CHKERRQ(ierr); 2375d64ed03dSBarry Smith PetscStackPop; 23765edff71fSBarry Smith ierr = VecLockPop(X);CHKERRQ(ierr); 237794ab13aaSBarry Smith ierr = PetscLogEventEnd(SNES_JacobianEval,snes,X,A,B);CHKERRQ(ierr); 2378a8054027SBarry Smith 2379e0e3a89bSBarry Smith /* the next line ensures that snes->ksp exists */ 2380e0e3a89bSBarry Smith ierr = SNESGetKSP(snes,&ksp);CHKERRQ(ierr); 23813b4f5425SBarry Smith if (snes->lagpreconditioner == -2) { 23823b4f5425SBarry Smith ierr = PetscInfo(snes,"Rebuilding preconditioner exactly once since lag is -2\n");CHKERRQ(ierr); 2383d1e9a80fSBarry Smith ierr = KSPSetReusePreconditioner(snes->ksp,PETSC_FALSE);CHKERRQ(ierr); 23843b4f5425SBarry Smith snes->lagpreconditioner = -1; 23853b4f5425SBarry Smith } else if (snes->lagpreconditioner == -1) { 2386a8054027SBarry Smith ierr = PetscInfo(snes,"Reusing preconditioner because lag is -1\n");CHKERRQ(ierr); 2387d1e9a80fSBarry Smith ierr = KSPSetReusePreconditioner(snes->ksp,PETSC_TRUE);CHKERRQ(ierr); 238837ec4e1aSPeter Brune } else if (snes->lagpreconditioner > 1 && (snes->iter + snes->pre_iter) % snes->lagpreconditioner) { 2389a8054027SBarry Smith ierr = PetscInfo2(snes,"Reusing preconditioner because lag is %D and SNES iteration is %D\n",snes->lagpreconditioner,snes->iter);CHKERRQ(ierr); 2390d1e9a80fSBarry Smith ierr = KSPSetReusePreconditioner(snes->ksp,PETSC_TRUE);CHKERRQ(ierr); 2391d1e9a80fSBarry Smith } else { 2392d1e9a80fSBarry Smith ierr = PetscInfo(snes,"Rebuilding preconditioner\n");CHKERRQ(ierr); 2393d1e9a80fSBarry Smith ierr = KSPSetReusePreconditioner(snes->ksp,PETSC_FALSE);CHKERRQ(ierr); 2394a8054027SBarry Smith } 2395a8054027SBarry Smith 23966d84be18SBarry Smith /* make sure user returned a correct Jacobian and preconditioner */ 239794ab13aaSBarry Smith /* PetscValidHeaderSpecific(A,MAT_CLASSID,3); 239894ab13aaSBarry Smith PetscValidHeaderSpecific(B,MAT_CLASSID,4); */ 2399693365a8SJed Brown { 2400693365a8SJed Brown PetscBool flag = PETSC_FALSE,flag_draw = PETSC_FALSE,flag_contour = PETSC_FALSE,flag_operator = PETSC_FALSE; 240127b0f280SBarry Smith ierr = PetscOptionsGetViewer(PetscObjectComm((PetscObject)snes),((PetscObject)snes)->prefix,"-snes_compare_explicit",NULL,NULL,&flag);CHKERRQ(ierr); 240227b0f280SBarry Smith ierr = PetscOptionsGetViewer(PetscObjectComm((PetscObject)snes),((PetscObject)snes)->prefix,"-snes_compare_explicit_draw",NULL,NULL,&flag_draw);CHKERRQ(ierr); 240327b0f280SBarry Smith ierr = PetscOptionsGetViewer(PetscObjectComm((PetscObject)snes),((PetscObject)snes)->prefix,"-snes_compare_explicit_draw_contour",NULL,NULL,&flag_contour);CHKERRQ(ierr); 240427b0f280SBarry Smith ierr = PetscOptionsGetViewer(PetscObjectComm((PetscObject)snes),((PetscObject)snes)->prefix,"-snes_compare_operator",NULL,NULL,&flag_operator);CHKERRQ(ierr); 2405693365a8SJed Brown if (flag || flag_draw || flag_contour) { 24060298fd71SBarry Smith Mat Bexp_mine = NULL,Bexp,FDexp; 2407693365a8SJed Brown PetscViewer vdraw,vstdout; 24086b3a5b13SJed Brown PetscBool flg; 2409693365a8SJed Brown if (flag_operator) { 241094ab13aaSBarry Smith ierr = MatComputeExplicitOperator(A,&Bexp_mine);CHKERRQ(ierr); 2411693365a8SJed Brown Bexp = Bexp_mine; 2412693365a8SJed Brown } else { 2413693365a8SJed Brown /* See if the preconditioning matrix can be viewed and added directly */ 241494ab13aaSBarry Smith ierr = PetscObjectTypeCompareAny((PetscObject)B,&flg,MATSEQAIJ,MATMPIAIJ,MATSEQDENSE,MATMPIDENSE,MATSEQBAIJ,MATMPIBAIJ,MATSEQSBAIJ,MATMPIBAIJ,"");CHKERRQ(ierr); 241594ab13aaSBarry Smith if (flg) Bexp = B; 2416693365a8SJed Brown else { 2417693365a8SJed Brown /* If the "preconditioning" matrix is itself MATSHELL or some other type without direct support */ 241894ab13aaSBarry Smith ierr = MatComputeExplicitOperator(B,&Bexp_mine);CHKERRQ(ierr); 2419693365a8SJed Brown Bexp = Bexp_mine; 2420693365a8SJed Brown } 2421693365a8SJed Brown } 2422693365a8SJed Brown ierr = MatConvert(Bexp,MATSAME,MAT_INITIAL_MATRIX,&FDexp);CHKERRQ(ierr); 2423d1e9a80fSBarry Smith ierr = SNESComputeJacobianDefault(snes,X,FDexp,FDexp,NULL);CHKERRQ(ierr); 2424ce94432eSBarry Smith ierr = PetscViewerASCIIGetStdout(PetscObjectComm((PetscObject)snes),&vstdout);CHKERRQ(ierr); 2425693365a8SJed Brown if (flag_draw || flag_contour) { 2426ce94432eSBarry Smith ierr = PetscViewerDrawOpen(PetscObjectComm((PetscObject)snes),0,"Explicit Jacobians",PETSC_DECIDE,PETSC_DECIDE,300,300,&vdraw);CHKERRQ(ierr); 2427693365a8SJed Brown if (flag_contour) {ierr = PetscViewerPushFormat(vdraw,PETSC_VIEWER_DRAW_CONTOUR);CHKERRQ(ierr);} 24280298fd71SBarry Smith } else vdraw = NULL; 2429693365a8SJed Brown ierr = PetscViewerASCIIPrintf(vstdout,"Explicit %s\n",flag_operator ? "Jacobian" : "preconditioning Jacobian");CHKERRQ(ierr); 2430693365a8SJed Brown if (flag) {ierr = MatView(Bexp,vstdout);CHKERRQ(ierr);} 2431693365a8SJed Brown if (vdraw) {ierr = MatView(Bexp,vdraw);CHKERRQ(ierr);} 2432693365a8SJed Brown ierr = PetscViewerASCIIPrintf(vstdout,"Finite difference Jacobian\n");CHKERRQ(ierr); 2433693365a8SJed Brown if (flag) {ierr = MatView(FDexp,vstdout);CHKERRQ(ierr);} 2434693365a8SJed Brown if (vdraw) {ierr = MatView(FDexp,vdraw);CHKERRQ(ierr);} 2435693365a8SJed Brown ierr = MatAYPX(FDexp,-1.0,Bexp,SAME_NONZERO_PATTERN);CHKERRQ(ierr); 2436693365a8SJed Brown ierr = PetscViewerASCIIPrintf(vstdout,"User-provided matrix minus finite difference Jacobian\n");CHKERRQ(ierr); 2437693365a8SJed Brown if (flag) {ierr = MatView(FDexp,vstdout);CHKERRQ(ierr);} 2438693365a8SJed Brown if (vdraw) { /* Always use contour for the difference */ 2439693365a8SJed Brown ierr = PetscViewerPushFormat(vdraw,PETSC_VIEWER_DRAW_CONTOUR);CHKERRQ(ierr); 2440693365a8SJed Brown ierr = MatView(FDexp,vdraw);CHKERRQ(ierr); 2441693365a8SJed Brown ierr = PetscViewerPopFormat(vdraw);CHKERRQ(ierr); 2442693365a8SJed Brown } 2443693365a8SJed Brown if (flag_contour) {ierr = PetscViewerPopFormat(vdraw);CHKERRQ(ierr);} 2444693365a8SJed Brown ierr = PetscViewerDestroy(&vdraw);CHKERRQ(ierr); 2445693365a8SJed Brown ierr = MatDestroy(&Bexp_mine);CHKERRQ(ierr); 2446693365a8SJed Brown ierr = MatDestroy(&FDexp);CHKERRQ(ierr); 2447693365a8SJed Brown } 2448693365a8SJed Brown } 24494c30e9fbSJed Brown { 24506719d8e4SJed Brown PetscBool flag = PETSC_FALSE,flag_display = PETSC_FALSE,flag_draw = PETSC_FALSE,flag_contour = PETSC_FALSE,flag_threshold = PETSC_FALSE; 24516719d8e4SJed Brown PetscReal threshold_atol = PETSC_SQRT_MACHINE_EPSILON,threshold_rtol = 10*PETSC_SQRT_MACHINE_EPSILON; 245227b0f280SBarry Smith ierr = PetscOptionsGetViewer(PetscObjectComm((PetscObject)snes),((PetscObject)snes)->prefix,"-snes_compare_coloring",NULL,NULL,&flag);CHKERRQ(ierr); 245327b0f280SBarry Smith ierr = PetscOptionsGetViewer(PetscObjectComm((PetscObject)snes),((PetscObject)snes)->prefix,"-snes_compare_coloring_display",NULL,NULL,&flag_display);CHKERRQ(ierr); 245427b0f280SBarry Smith ierr = PetscOptionsGetViewer(PetscObjectComm((PetscObject)snes),((PetscObject)snes)->prefix,"-snes_compare_coloring_draw",NULL,NULL,&flag_draw);CHKERRQ(ierr); 245527b0f280SBarry Smith ierr = PetscOptionsGetViewer(PetscObjectComm((PetscObject)snes),((PetscObject)snes)->prefix,"-snes_compare_coloring_draw_contour",NULL,NULL,&flag_contour);CHKERRQ(ierr); 245627b0f280SBarry Smith ierr = PetscOptionsGetViewer(PetscObjectComm((PetscObject)snes),((PetscObject)snes)->prefix,"-snes_compare_coloring_threshold",NULL,NULL,&flag_threshold);CHKERRQ(ierr); 245727b0f280SBarry Smith if (flag_threshold) { 2458c5929fdfSBarry Smith ierr = PetscOptionsGetReal(((PetscObject)snes)->options,((PetscObject)snes)->prefix,"-snes_compare_coloring_threshold_rtol",&threshold_rtol,NULL);CHKERRQ(ierr); 2459c5929fdfSBarry Smith ierr = PetscOptionsGetReal(((PetscObject)snes)->options,((PetscObject)snes)->prefix,"-snes_compare_coloring_threshold_atol",&threshold_atol,NULL);CHKERRQ(ierr); 246027b0f280SBarry Smith } 24616719d8e4SJed Brown if (flag || flag_display || flag_draw || flag_contour || flag_threshold) { 24624c30e9fbSJed Brown Mat Bfd; 24634c30e9fbSJed Brown PetscViewer vdraw,vstdout; 2464335efc43SPeter Brune MatColoring coloring; 24654c30e9fbSJed Brown ISColoring iscoloring; 24664c30e9fbSJed Brown MatFDColoring matfdcoloring; 24674c30e9fbSJed Brown PetscErrorCode (*func)(SNES,Vec,Vec,void*); 24684c30e9fbSJed Brown void *funcctx; 24696719d8e4SJed Brown PetscReal norm1,norm2,normmax; 24704c30e9fbSJed Brown 247194ab13aaSBarry Smith ierr = MatDuplicate(B,MAT_DO_NOT_COPY_VALUES,&Bfd);CHKERRQ(ierr); 2472335efc43SPeter Brune ierr = MatColoringCreate(Bfd,&coloring);CHKERRQ(ierr); 2473335efc43SPeter Brune ierr = MatColoringSetType(coloring,MATCOLORINGSL);CHKERRQ(ierr); 2474335efc43SPeter Brune ierr = MatColoringSetFromOptions(coloring);CHKERRQ(ierr); 2475335efc43SPeter Brune ierr = MatColoringApply(coloring,&iscoloring);CHKERRQ(ierr); 2476335efc43SPeter Brune ierr = MatColoringDestroy(&coloring);CHKERRQ(ierr); 24774c30e9fbSJed Brown ierr = MatFDColoringCreate(Bfd,iscoloring,&matfdcoloring);CHKERRQ(ierr); 2478f86b9fbaSHong Zhang ierr = MatFDColoringSetFromOptions(matfdcoloring);CHKERRQ(ierr); 2479f86b9fbaSHong Zhang ierr = MatFDColoringSetUp(Bfd,iscoloring,matfdcoloring);CHKERRQ(ierr); 24804c30e9fbSJed Brown ierr = ISColoringDestroy(&iscoloring);CHKERRQ(ierr); 24814c30e9fbSJed Brown 24824c30e9fbSJed Brown /* This method of getting the function is currently unreliable since it doesn't work for DM local functions. */ 24830298fd71SBarry Smith ierr = SNESGetFunction(snes,NULL,&func,&funcctx);CHKERRQ(ierr); 24844c30e9fbSJed Brown ierr = MatFDColoringSetFunction(matfdcoloring,(PetscErrorCode (*)(void))func,funcctx);CHKERRQ(ierr); 24854c30e9fbSJed Brown ierr = PetscObjectSetOptionsPrefix((PetscObject)matfdcoloring,((PetscObject)snes)->prefix);CHKERRQ(ierr); 24864c30e9fbSJed Brown ierr = PetscObjectAppendOptionsPrefix((PetscObject)matfdcoloring,"coloring_");CHKERRQ(ierr); 24874c30e9fbSJed Brown ierr = MatFDColoringSetFromOptions(matfdcoloring);CHKERRQ(ierr); 2488d1e9a80fSBarry Smith ierr = MatFDColoringApply(Bfd,matfdcoloring,X,snes);CHKERRQ(ierr); 24894c30e9fbSJed Brown ierr = MatFDColoringDestroy(&matfdcoloring);CHKERRQ(ierr); 24904c30e9fbSJed Brown 2491ce94432eSBarry Smith ierr = PetscViewerASCIIGetStdout(PetscObjectComm((PetscObject)snes),&vstdout);CHKERRQ(ierr); 24924c30e9fbSJed Brown if (flag_draw || flag_contour) { 2493ce94432eSBarry Smith ierr = PetscViewerDrawOpen(PetscObjectComm((PetscObject)snes),0,"Colored Jacobians",PETSC_DECIDE,PETSC_DECIDE,300,300,&vdraw);CHKERRQ(ierr); 24944c30e9fbSJed Brown if (flag_contour) {ierr = PetscViewerPushFormat(vdraw,PETSC_VIEWER_DRAW_CONTOUR);CHKERRQ(ierr);} 24950298fd71SBarry Smith } else vdraw = NULL; 24964c30e9fbSJed Brown ierr = PetscViewerASCIIPrintf(vstdout,"Explicit preconditioning Jacobian\n");CHKERRQ(ierr); 249794ab13aaSBarry Smith if (flag_display) {ierr = MatView(B,vstdout);CHKERRQ(ierr);} 249894ab13aaSBarry Smith if (vdraw) {ierr = MatView(B,vdraw);CHKERRQ(ierr);} 24994c30e9fbSJed Brown ierr = PetscViewerASCIIPrintf(vstdout,"Colored Finite difference Jacobian\n");CHKERRQ(ierr); 25006719d8e4SJed Brown if (flag_display) {ierr = MatView(Bfd,vstdout);CHKERRQ(ierr);} 25014c30e9fbSJed Brown if (vdraw) {ierr = MatView(Bfd,vdraw);CHKERRQ(ierr);} 250294ab13aaSBarry Smith ierr = MatAYPX(Bfd,-1.0,B,SAME_NONZERO_PATTERN);CHKERRQ(ierr); 25034c30e9fbSJed Brown ierr = MatNorm(Bfd,NORM_1,&norm1);CHKERRQ(ierr); 25046719d8e4SJed Brown ierr = MatNorm(Bfd,NORM_FROBENIUS,&norm2);CHKERRQ(ierr); 25054c30e9fbSJed Brown ierr = MatNorm(Bfd,NORM_MAX,&normmax);CHKERRQ(ierr); 250657622a8eSBarry Smith ierr = PetscViewerASCIIPrintf(vstdout,"User-provided matrix minus finite difference Jacobian, norm1=%g normFrob=%g normmax=%g\n",(double)norm1,(double)norm2,(double)normmax);CHKERRQ(ierr); 25076719d8e4SJed Brown if (flag_display) {ierr = MatView(Bfd,vstdout);CHKERRQ(ierr);} 25084c30e9fbSJed Brown if (vdraw) { /* Always use contour for the difference */ 25094c30e9fbSJed Brown ierr = PetscViewerPushFormat(vdraw,PETSC_VIEWER_DRAW_CONTOUR);CHKERRQ(ierr); 25104c30e9fbSJed Brown ierr = MatView(Bfd,vdraw);CHKERRQ(ierr); 25114c30e9fbSJed Brown ierr = PetscViewerPopFormat(vdraw);CHKERRQ(ierr); 25124c30e9fbSJed Brown } 25134c30e9fbSJed Brown if (flag_contour) {ierr = PetscViewerPopFormat(vdraw);CHKERRQ(ierr);} 25146719d8e4SJed Brown 25156719d8e4SJed Brown if (flag_threshold) { 25166719d8e4SJed Brown PetscInt bs,rstart,rend,i; 251794ab13aaSBarry Smith ierr = MatGetBlockSize(B,&bs);CHKERRQ(ierr); 251894ab13aaSBarry Smith ierr = MatGetOwnershipRange(B,&rstart,&rend);CHKERRQ(ierr); 25196719d8e4SJed Brown for (i=rstart; i<rend; i++) { 25206719d8e4SJed Brown const PetscScalar *ba,*ca; 25216719d8e4SJed Brown const PetscInt *bj,*cj; 25226719d8e4SJed Brown PetscInt bn,cn,j,maxentrycol = -1,maxdiffcol = -1,maxrdiffcol = -1; 25236719d8e4SJed Brown PetscReal maxentry = 0,maxdiff = 0,maxrdiff = 0; 252494ab13aaSBarry Smith ierr = MatGetRow(B,i,&bn,&bj,&ba);CHKERRQ(ierr); 25256719d8e4SJed Brown ierr = MatGetRow(Bfd,i,&cn,&cj,&ca);CHKERRQ(ierr); 252694ab13aaSBarry Smith if (bn != cn) SETERRQ(((PetscObject)A)->comm,PETSC_ERR_PLIB,"Unexpected different nonzero pattern in -snes_compare_coloring_threshold"); 25276719d8e4SJed Brown for (j=0; j<bn; j++) { 25286719d8e4SJed Brown PetscReal rdiff = PetscAbsScalar(ca[j]) / (threshold_atol + threshold_rtol*PetscAbsScalar(ba[j])); 25296719d8e4SJed Brown if (PetscAbsScalar(ba[j]) > PetscAbs(maxentry)) { 25306719d8e4SJed Brown maxentrycol = bj[j]; 25316719d8e4SJed Brown maxentry = PetscRealPart(ba[j]); 25326719d8e4SJed Brown } 25336719d8e4SJed Brown if (PetscAbsScalar(ca[j]) > PetscAbs(maxdiff)) { 25346719d8e4SJed Brown maxdiffcol = bj[j]; 25356719d8e4SJed Brown maxdiff = PetscRealPart(ca[j]); 25366719d8e4SJed Brown } 25376719d8e4SJed Brown if (rdiff > maxrdiff) { 25386719d8e4SJed Brown maxrdiffcol = bj[j]; 25396719d8e4SJed Brown maxrdiff = rdiff; 25406719d8e4SJed Brown } 25416719d8e4SJed Brown } 25426719d8e4SJed Brown if (maxrdiff > 1) { 254357622a8eSBarry Smith ierr = PetscViewerASCIIPrintf(vstdout,"row %D (maxentry=%g at %D, maxdiff=%g at %D, maxrdiff=%g at %D):",i,(double)maxentry,maxentrycol,(double)maxdiff,maxdiffcol,(double)maxrdiff,maxrdiffcol);CHKERRQ(ierr); 25446719d8e4SJed Brown for (j=0; j<bn; j++) { 25456719d8e4SJed Brown PetscReal rdiff; 25466719d8e4SJed Brown rdiff = PetscAbsScalar(ca[j]) / (threshold_atol + threshold_rtol*PetscAbsScalar(ba[j])); 25476719d8e4SJed Brown if (rdiff > 1) { 254857622a8eSBarry Smith ierr = PetscViewerASCIIPrintf(vstdout," (%D,%g:%g)",bj[j],(double)PetscRealPart(ba[j]),(double)PetscRealPart(ca[j]));CHKERRQ(ierr); 25496719d8e4SJed Brown } 25506719d8e4SJed Brown } 25516719d8e4SJed Brown ierr = PetscViewerASCIIPrintf(vstdout,"\n",i,maxentry,maxdiff,maxrdiff);CHKERRQ(ierr); 25526719d8e4SJed Brown } 255394ab13aaSBarry Smith ierr = MatRestoreRow(B,i,&bn,&bj,&ba);CHKERRQ(ierr); 25546719d8e4SJed Brown ierr = MatRestoreRow(Bfd,i,&cn,&cj,&ca);CHKERRQ(ierr); 25556719d8e4SJed Brown } 25566719d8e4SJed Brown } 25574c30e9fbSJed Brown ierr = PetscViewerDestroy(&vdraw);CHKERRQ(ierr); 25584c30e9fbSJed Brown ierr = MatDestroy(&Bfd);CHKERRQ(ierr); 25594c30e9fbSJed Brown } 25604c30e9fbSJed Brown } 25613a40ed3dSBarry Smith PetscFunctionReturn(0); 25629b94acceSBarry Smith } 25639b94acceSBarry Smith 2564bf388a1fSBarry Smith /*MC 2565411c0326SBarry Smith SNESJacobianFunction - Function used to convey the nonlinear Jacobian of the function to be solved by SNES 2566bf388a1fSBarry Smith 2567bf388a1fSBarry Smith Synopsis: 2568411c0326SBarry Smith #include "petscsnes.h" 2569411c0326SBarry Smith PetscErrorCode SNESJacobianFunction(SNES snes,Vec x,Mat Amat,Mat Pmat,void *ctx); 2570bf388a1fSBarry Smith 2571bf388a1fSBarry Smith + x - input vector 2572e5d3d808SBarry Smith . Amat - the matrix that defines the (approximate) Jacobian 2573e5d3d808SBarry Smith . Pmat - the matrix to be used in constructing the preconditioner, usually the same as Amat. 2574bf388a1fSBarry Smith - ctx - [optional] user-defined Jacobian context 2575bf388a1fSBarry Smith 2576878cb397SSatish Balay Level: intermediate 2577878cb397SSatish Balay 2578bf388a1fSBarry Smith .seealso: SNESSetFunction(), SNESGetFunction(), SNESSetJacobian(), SNESGetJacobian() 2579bf388a1fSBarry Smith M*/ 2580bf388a1fSBarry Smith 25819b94acceSBarry Smith /*@C 25829b94acceSBarry Smith SNESSetJacobian - Sets the function to compute Jacobian as well as the 2583044dda88SLois Curfman McInnes location to store the matrix. 25849b94acceSBarry Smith 25853f9fe445SBarry Smith Logically Collective on SNES and Mat 2586c7afd0dbSLois Curfman McInnes 25879b94acceSBarry Smith Input Parameters: 2588c7afd0dbSLois Curfman McInnes + snes - the SNES context 2589e5d3d808SBarry Smith . Amat - the matrix that defines the (approximate) Jacobian 2590e5d3d808SBarry Smith . Pmat - the matrix to be used in constructing the preconditioner, usually the same as Amat. 2591411c0326SBarry Smith . J - Jacobian evaluation routine (if NULL then SNES retains any previously set value), see SNESJacobianFunction for details 2592c7afd0dbSLois Curfman McInnes - ctx - [optional] user-defined context for private data for the 25930298fd71SBarry Smith Jacobian evaluation routine (may be NULL) (if NULL then SNES retains any previously set value) 25949b94acceSBarry Smith 25959b94acceSBarry Smith Notes: 2596e5d3d808SBarry Smith If the Amat matrix and Pmat matrix are different you must call MatAssemblyBegin/End() on 259716913363SBarry Smith each matrix. 259816913363SBarry Smith 2599895c21f2SBarry Smith If you know the operator Amat has a null space you can use MatSetNullSpace() and MatSetTransposeNullSpace() to supply the null 2600895c21f2SBarry Smith space to Amat and the KSP solvers will automatically use that null space as needed during the solution process. 2601895c21f2SBarry Smith 26028d359177SBarry Smith If using SNESComputeJacobianDefaultColor() to assemble a Jacobian, the ctx argument 2603a8a26c1eSJed Brown must be a MatFDColoring. 2604a8a26c1eSJed Brown 2605c3cc8fd1SJed Brown Other defect-correction schemes can be used by computing a different matrix in place of the Jacobian. One common 2606c3cc8fd1SJed Brown example is to use the "Picard linearization" which only differentiates through the highest order parts of each term. 2607c3cc8fd1SJed Brown 260836851e7fSLois Curfman McInnes Level: beginner 260936851e7fSLois Curfman McInnes 26109b94acceSBarry Smith .keywords: SNES, nonlinear, set, Jacobian, matrix 26119b94acceSBarry Smith 2612411c0326SBarry Smith .seealso: KSPSetOperators(), SNESSetFunction(), MatMFFDComputeJacobian(), SNESComputeJacobianDefaultColor(), MatStructure, J, 2613411c0326SBarry Smith SNESSetPicard(), SNESJacobianFunction 26149b94acceSBarry Smith @*/ 2615d1e9a80fSBarry Smith PetscErrorCode SNESSetJacobian(SNES snes,Mat Amat,Mat Pmat,PetscErrorCode (*J)(SNES,Vec,Mat,Mat,void*),void *ctx) 26169b94acceSBarry Smith { 2617dfbe8321SBarry Smith PetscErrorCode ierr; 26186cab3a1bSJed Brown DM dm; 26193a7fca6bSBarry Smith 26203a40ed3dSBarry Smith PetscFunctionBegin; 26210700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 2622e5d3d808SBarry Smith if (Amat) PetscValidHeaderSpecific(Amat,MAT_CLASSID,2); 2623e5d3d808SBarry Smith if (Pmat) PetscValidHeaderSpecific(Pmat,MAT_CLASSID,3); 2624e5d3d808SBarry Smith if (Amat) PetscCheckSameComm(snes,1,Amat,2); 2625e5d3d808SBarry Smith if (Pmat) PetscCheckSameComm(snes,1,Pmat,3); 26266cab3a1bSJed Brown ierr = SNESGetDM(snes,&dm);CHKERRQ(ierr); 2627f8b49ee9SBarry Smith ierr = DMSNESSetJacobian(dm,J,ctx);CHKERRQ(ierr); 2628e5d3d808SBarry Smith if (Amat) { 2629e5d3d808SBarry Smith ierr = PetscObjectReference((PetscObject)Amat);CHKERRQ(ierr); 26306bf464f9SBarry Smith ierr = MatDestroy(&snes->jacobian);CHKERRQ(ierr); 2631f5af7f23SKarl Rupp 2632e5d3d808SBarry Smith snes->jacobian = Amat; 26333a7fca6bSBarry Smith } 2634e5d3d808SBarry Smith if (Pmat) { 2635e5d3d808SBarry Smith ierr = PetscObjectReference((PetscObject)Pmat);CHKERRQ(ierr); 26366bf464f9SBarry Smith ierr = MatDestroy(&snes->jacobian_pre);CHKERRQ(ierr); 2637f5af7f23SKarl Rupp 2638e5d3d808SBarry Smith snes->jacobian_pre = Pmat; 26393a7fca6bSBarry Smith } 26403a40ed3dSBarry Smith PetscFunctionReturn(0); 26419b94acceSBarry Smith } 264262fef451SLois Curfman McInnes 2643c2aafc4cSSatish Balay /*@C 2644b4fd4287SBarry Smith SNESGetJacobian - Returns the Jacobian matrix and optionally the user 2645b4fd4287SBarry Smith provided context for evaluating the Jacobian. 2646b4fd4287SBarry Smith 2647c7afd0dbSLois Curfman McInnes Not Collective, but Mat object will be parallel if SNES object is 2648c7afd0dbSLois Curfman McInnes 2649b4fd4287SBarry Smith Input Parameter: 2650b4fd4287SBarry Smith . snes - the nonlinear solver context 2651b4fd4287SBarry Smith 2652b4fd4287SBarry Smith Output Parameters: 2653e5d3d808SBarry Smith + Amat - location to stash (approximate) Jacobian matrix (or NULL) 2654e5d3d808SBarry Smith . Pmat - location to stash matrix used to compute the preconditioner (or NULL) 2655411c0326SBarry Smith . J - location to put Jacobian function (or NULL), see SNESJacobianFunction for details on its calling sequence 26560298fd71SBarry Smith - ctx - location to stash Jacobian ctx (or NULL) 2657fee21e36SBarry Smith 265836851e7fSLois Curfman McInnes Level: advanced 265936851e7fSLois Curfman McInnes 2660411c0326SBarry Smith .seealso: SNESSetJacobian(), SNESComputeJacobian(), SNESJacobianFunction, SNESGetFunction() 2661b4fd4287SBarry Smith @*/ 2662d1e9a80fSBarry Smith PetscErrorCode SNESGetJacobian(SNES snes,Mat *Amat,Mat *Pmat,PetscErrorCode (**J)(SNES,Vec,Mat,Mat,void*),void **ctx) 2663b4fd4287SBarry Smith { 26646cab3a1bSJed Brown PetscErrorCode ierr; 26656cab3a1bSJed Brown DM dm; 2666942e3340SBarry Smith DMSNES sdm; 26676cab3a1bSJed Brown 26683a40ed3dSBarry Smith PetscFunctionBegin; 26690700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 2670e5d3d808SBarry Smith if (Amat) *Amat = snes->jacobian; 2671e5d3d808SBarry Smith if (Pmat) *Pmat = snes->jacobian_pre; 26726cab3a1bSJed Brown ierr = SNESGetDM(snes,&dm);CHKERRQ(ierr); 2673942e3340SBarry Smith ierr = DMGetDMSNES(dm,&sdm);CHKERRQ(ierr); 2674f8b49ee9SBarry Smith if (J) *J = sdm->ops->computejacobian; 26756cab3a1bSJed Brown if (ctx) *ctx = sdm->jacobianctx; 26763a40ed3dSBarry Smith PetscFunctionReturn(0); 2677b4fd4287SBarry Smith } 2678b4fd4287SBarry Smith 26799b94acceSBarry Smith /*@ 26809b94acceSBarry Smith SNESSetUp - Sets up the internal data structures for the later use 2681272ac6f2SLois Curfman McInnes of a nonlinear solver. 26829b94acceSBarry Smith 2683fee21e36SBarry Smith Collective on SNES 2684fee21e36SBarry Smith 2685c7afd0dbSLois Curfman McInnes Input Parameters: 268670e92668SMatthew Knepley . snes - the SNES context 2687c7afd0dbSLois Curfman McInnes 2688272ac6f2SLois Curfman McInnes Notes: 2689272ac6f2SLois Curfman McInnes For basic use of the SNES solvers the user need not explicitly call 2690272ac6f2SLois Curfman McInnes SNESSetUp(), since these actions will automatically occur during 2691272ac6f2SLois Curfman McInnes the call to SNESSolve(). However, if one wishes to control this 2692272ac6f2SLois Curfman McInnes phase separately, SNESSetUp() should be called after SNESCreate() 2693272ac6f2SLois Curfman McInnes and optional routines of the form SNESSetXXX(), but before SNESSolve(). 2694272ac6f2SLois Curfman McInnes 269536851e7fSLois Curfman McInnes Level: advanced 269636851e7fSLois Curfman McInnes 26979b94acceSBarry Smith .keywords: SNES, nonlinear, setup 26989b94acceSBarry Smith 26999b94acceSBarry Smith .seealso: SNESCreate(), SNESSolve(), SNESDestroy() 27009b94acceSBarry Smith @*/ 27017087cfbeSBarry Smith PetscErrorCode SNESSetUp(SNES snes) 27029b94acceSBarry Smith { 2703dfbe8321SBarry Smith PetscErrorCode ierr; 27046cab3a1bSJed Brown DM dm; 2705942e3340SBarry Smith DMSNES sdm; 2706c35f09e5SBarry Smith SNESLineSearch linesearch, pclinesearch; 27076e2a1849SPeter Brune void *lsprectx,*lspostctx; 27086b2b7091SBarry Smith PetscErrorCode (*precheck)(SNESLineSearch,Vec,Vec,PetscBool*,void*); 27096b2b7091SBarry Smith PetscErrorCode (*postcheck)(SNESLineSearch,Vec,Vec,Vec,PetscBool*,PetscBool*,void*); 27106e2a1849SPeter Brune PetscErrorCode (*func)(SNES,Vec,Vec,void*); 27116e2a1849SPeter Brune Vec f,fpc; 27126e2a1849SPeter Brune void *funcctx; 2713d1e9a80fSBarry Smith PetscErrorCode (*jac)(SNES,Vec,Mat,Mat,void*); 27141eb13d49SPeter Brune void *jacctx,*appctx; 271532b97717SPeter Brune Mat j,jpre; 27163a40ed3dSBarry Smith 27173a40ed3dSBarry Smith PetscFunctionBegin; 27180700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 27194dc4c822SBarry Smith if (snes->setupcalled) PetscFunctionReturn(0); 27209b94acceSBarry Smith 27217adad957SLisandro Dalcin if (!((PetscObject)snes)->type_name) { 272204d7464bSBarry Smith ierr = SNESSetType(snes,SNESNEWTONLS);CHKERRQ(ierr); 272385385478SLisandro Dalcin } 272485385478SLisandro Dalcin 27250298fd71SBarry Smith ierr = SNESGetFunction(snes,&snes->vec_func,NULL,NULL);CHKERRQ(ierr); 272658c9b817SLisandro Dalcin 27276cab3a1bSJed Brown ierr = SNESGetDM(snes,&dm);CHKERRQ(ierr); 2728942e3340SBarry Smith ierr = DMGetDMSNES(dm,&sdm);CHKERRQ(ierr); 2729ce94432eSBarry Smith if (!sdm->ops->computefunction) SETERRQ(PetscObjectComm((PetscObject)dm),PETSC_ERR_ARG_WRONGSTATE,"Function never provided to SNES object"); 273022c6f798SBarry Smith if (!sdm->ops->computejacobian) { 27318d359177SBarry Smith ierr = DMSNESSetJacobian(dm,SNESComputeJacobianDefaultColor,NULL);CHKERRQ(ierr); 273219f7a02aSBarry Smith } 27336cab3a1bSJed Brown if (!snes->vec_func) { 27346cab3a1bSJed Brown ierr = DMCreateGlobalVector(dm,&snes->vec_func);CHKERRQ(ierr); 2735214df951SJed Brown } 2736efd51863SBarry Smith 273722d28d08SBarry Smith if (!snes->ksp) { 273822d28d08SBarry Smith ierr = SNESGetKSP(snes, &snes->ksp);CHKERRQ(ierr); 273922d28d08SBarry Smith } 2740b710008aSBarry Smith 274122d28d08SBarry Smith if (!snes->linesearch) { 27427601faf0SJed Brown ierr = SNESGetLineSearch(snes, &snes->linesearch);CHKERRQ(ierr); 274322d28d08SBarry Smith } 2744ed07d7d7SPeter Brune ierr = SNESLineSearchSetFunction(snes->linesearch,SNESComputeFunction);CHKERRQ(ierr); 27459e764e56SPeter Brune 2746efd4aadfSBarry Smith if (snes->npc && (snes->npcside== PC_LEFT)) { 2747172a4300SPeter Brune snes->mf = PETSC_TRUE; 2748172a4300SPeter Brune snes->mf_operator = PETSC_FALSE; 2749172a4300SPeter Brune } 2750d8f46077SPeter Brune 2751efd4aadfSBarry Smith if (snes->npc) { 27526e2a1849SPeter Brune /* copy the DM over */ 27536e2a1849SPeter Brune ierr = SNESGetDM(snes,&dm);CHKERRQ(ierr); 2754efd4aadfSBarry Smith ierr = SNESSetDM(snes->npc,dm);CHKERRQ(ierr); 27556e2a1849SPeter Brune 27566e2a1849SPeter Brune ierr = SNESGetFunction(snes,&f,&func,&funcctx);CHKERRQ(ierr); 27576e2a1849SPeter Brune ierr = VecDuplicate(f,&fpc);CHKERRQ(ierr); 2758efd4aadfSBarry Smith ierr = SNESSetFunction(snes->npc,fpc,func,funcctx);CHKERRQ(ierr); 275932b97717SPeter Brune ierr = SNESGetJacobian(snes,&j,&jpre,&jac,&jacctx);CHKERRQ(ierr); 2760efd4aadfSBarry Smith ierr = SNESSetJacobian(snes->npc,j,jpre,jac,jacctx);CHKERRQ(ierr); 27611eb13d49SPeter Brune ierr = SNESGetApplicationContext(snes,&appctx);CHKERRQ(ierr); 2762efd4aadfSBarry Smith ierr = SNESSetApplicationContext(snes->npc,appctx);CHKERRQ(ierr); 27636e2a1849SPeter Brune ierr = VecDestroy(&fpc);CHKERRQ(ierr); 27646e2a1849SPeter Brune 27656e2a1849SPeter Brune /* copy the function pointers over */ 2766efd4aadfSBarry Smith ierr = PetscObjectCopyFortranFunctionPointers((PetscObject)snes,(PetscObject)snes->npc);CHKERRQ(ierr); 27676e2a1849SPeter Brune 27686e2a1849SPeter Brune /* default to 1 iteration */ 2769efd4aadfSBarry Smith ierr = SNESSetTolerances(snes->npc,0.0,0.0,0.0,1,snes->npc->max_funcs);CHKERRQ(ierr); 2770efd4aadfSBarry Smith if (snes->npcside==PC_RIGHT) { 2771efd4aadfSBarry Smith ierr = SNESSetNormSchedule(snes->npc,SNES_NORM_FINAL_ONLY);CHKERRQ(ierr); 2772a9936a0cSPeter Brune } else { 2773efd4aadfSBarry Smith ierr = SNESSetNormSchedule(snes->npc,SNES_NORM_NONE);CHKERRQ(ierr); 2774a9936a0cSPeter Brune } 2775efd4aadfSBarry Smith ierr = SNESSetFromOptions(snes->npc);CHKERRQ(ierr); 27766e2a1849SPeter Brune 27776e2a1849SPeter Brune /* copy the line search context over */ 27787601faf0SJed Brown ierr = SNESGetLineSearch(snes,&linesearch);CHKERRQ(ierr); 2779efd4aadfSBarry Smith ierr = SNESGetLineSearch(snes->npc,&pclinesearch);CHKERRQ(ierr); 27806b2b7091SBarry Smith ierr = SNESLineSearchGetPreCheck(linesearch,&precheck,&lsprectx);CHKERRQ(ierr); 27816b2b7091SBarry Smith ierr = SNESLineSearchGetPostCheck(linesearch,&postcheck,&lspostctx);CHKERRQ(ierr); 27826b2b7091SBarry Smith ierr = SNESLineSearchSetPreCheck(pclinesearch,precheck,lsprectx);CHKERRQ(ierr); 27836b2b7091SBarry Smith ierr = SNESLineSearchSetPostCheck(pclinesearch,postcheck,lspostctx);CHKERRQ(ierr); 27846e2a1849SPeter Brune ierr = PetscObjectCopyFortranFunctionPointers((PetscObject)linesearch, (PetscObject)pclinesearch);CHKERRQ(ierr); 27856e2a1849SPeter Brune } 278632b97717SPeter Brune if (snes->mf) { 278732b97717SPeter Brune ierr = SNESSetUpMatrixFree_Private(snes, snes->mf_operator, snes->mf_version);CHKERRQ(ierr); 278832b97717SPeter Brune } 278932b97717SPeter Brune if (snes->ops->usercompute && !snes->user) { 279032b97717SPeter Brune ierr = (*snes->ops->usercompute)(snes,(void**)&snes->user);CHKERRQ(ierr); 279132b97717SPeter Brune } 27926e2a1849SPeter Brune 279337ec4e1aSPeter Brune snes->jac_iter = 0; 279437ec4e1aSPeter Brune snes->pre_iter = 0; 279537ec4e1aSPeter Brune 2796410397dcSLisandro Dalcin if (snes->ops->setup) { 2797410397dcSLisandro Dalcin ierr = (*snes->ops->setup)(snes);CHKERRQ(ierr); 2798410397dcSLisandro Dalcin } 279958c9b817SLisandro Dalcin 2800efd4aadfSBarry Smith if (snes->npc && (snes->npcside== PC_LEFT)) { 28016c67d002SPeter Brune if (snes->functype == SNES_FUNCTION_PRECONDITIONED) { 280255d4788fSPeter Brune ierr = SNESGetLineSearch(snes,&linesearch);CHKERRQ(ierr); 2803be95d8f1SBarry Smith ierr = SNESLineSearchSetFunction(linesearch,SNESComputeFunctionDefaultNPC);CHKERRQ(ierr); 28046c67d002SPeter Brune } 28056c67d002SPeter Brune } 28066c67d002SPeter Brune 28077aaed0d8SBarry Smith snes->setupcalled = PETSC_TRUE; 28083a40ed3dSBarry Smith PetscFunctionReturn(0); 28099b94acceSBarry Smith } 28109b94acceSBarry Smith 281137596af1SLisandro Dalcin /*@ 281237596af1SLisandro Dalcin SNESReset - Resets a SNES context to the snessetupcalled = 0 state and removes any allocated Vecs and Mats 281337596af1SLisandro Dalcin 281437596af1SLisandro Dalcin Collective on SNES 281537596af1SLisandro Dalcin 281637596af1SLisandro Dalcin Input Parameter: 281737596af1SLisandro Dalcin . snes - iterative context obtained from SNESCreate() 281837596af1SLisandro Dalcin 2819d25893d9SBarry Smith Level: intermediate 2820d25893d9SBarry Smith 2821d25893d9SBarry Smith Notes: Also calls the application context destroy routine set with SNESSetComputeApplicationContext() 282237596af1SLisandro Dalcin 282337596af1SLisandro Dalcin .keywords: SNES, destroy 282437596af1SLisandro Dalcin 282537596af1SLisandro Dalcin .seealso: SNESCreate(), SNESSetUp(), SNESSolve() 282637596af1SLisandro Dalcin @*/ 282737596af1SLisandro Dalcin PetscErrorCode SNESReset(SNES snes) 282837596af1SLisandro Dalcin { 282937596af1SLisandro Dalcin PetscErrorCode ierr; 283037596af1SLisandro Dalcin 283137596af1SLisandro Dalcin PetscFunctionBegin; 283237596af1SLisandro Dalcin PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 2833d25893d9SBarry Smith if (snes->ops->userdestroy && snes->user) { 2834d25893d9SBarry Smith ierr = (*snes->ops->userdestroy)((void**)&snes->user);CHKERRQ(ierr); 28350298fd71SBarry Smith snes->user = NULL; 2836d25893d9SBarry Smith } 2837efd4aadfSBarry Smith if (snes->npc) { 2838efd4aadfSBarry Smith ierr = SNESReset(snes->npc);CHKERRQ(ierr); 28398a23116dSBarry Smith } 28408a23116dSBarry Smith 284137596af1SLisandro Dalcin if (snes->ops->reset) { 284237596af1SLisandro Dalcin ierr = (*snes->ops->reset)(snes);CHKERRQ(ierr); 284337596af1SLisandro Dalcin } 28449e764e56SPeter Brune if (snes->ksp) { 28459e764e56SPeter Brune ierr = KSPReset(snes->ksp);CHKERRQ(ierr); 28469e764e56SPeter Brune } 28479e764e56SPeter Brune 28489e764e56SPeter Brune if (snes->linesearch) { 2849f1c6b773SPeter Brune ierr = SNESLineSearchReset(snes->linesearch);CHKERRQ(ierr); 28509e764e56SPeter Brune } 28519e764e56SPeter Brune 28526bf464f9SBarry Smith ierr = VecDestroy(&snes->vec_rhs);CHKERRQ(ierr); 28536bf464f9SBarry Smith ierr = VecDestroy(&snes->vec_sol);CHKERRQ(ierr); 28546bf464f9SBarry Smith ierr = VecDestroy(&snes->vec_sol_update);CHKERRQ(ierr); 28556bf464f9SBarry Smith ierr = VecDestroy(&snes->vec_func);CHKERRQ(ierr); 28566bf464f9SBarry Smith ierr = MatDestroy(&snes->jacobian);CHKERRQ(ierr); 28576bf464f9SBarry Smith ierr = MatDestroy(&snes->jacobian_pre);CHKERRQ(ierr); 2858c41d97abSJed Brown ierr = VecDestroyVecs(snes->nwork,&snes->work);CHKERRQ(ierr); 2859c41d97abSJed Brown ierr = VecDestroyVecs(snes->nvwork,&snes->vwork);CHKERRQ(ierr); 2860f5af7f23SKarl Rupp 286140fdac6aSLawrence Mitchell snes->alwayscomputesfinalresidual = PETSC_FALSE; 286240fdac6aSLawrence Mitchell 286337596af1SLisandro Dalcin snes->nwork = snes->nvwork = 0; 286437596af1SLisandro Dalcin snes->setupcalled = PETSC_FALSE; 286537596af1SLisandro Dalcin PetscFunctionReturn(0); 286637596af1SLisandro Dalcin } 286737596af1SLisandro Dalcin 286852baeb72SSatish Balay /*@ 28699b94acceSBarry Smith SNESDestroy - Destroys the nonlinear solver context that was created 28709b94acceSBarry Smith with SNESCreate(). 28719b94acceSBarry Smith 2872c7afd0dbSLois Curfman McInnes Collective on SNES 2873c7afd0dbSLois Curfman McInnes 28749b94acceSBarry Smith Input Parameter: 28759b94acceSBarry Smith . snes - the SNES context 28769b94acceSBarry Smith 287736851e7fSLois Curfman McInnes Level: beginner 287836851e7fSLois Curfman McInnes 28799b94acceSBarry Smith .keywords: SNES, nonlinear, destroy 28809b94acceSBarry Smith 288163a78c88SLois Curfman McInnes .seealso: SNESCreate(), SNESSolve() 28829b94acceSBarry Smith @*/ 28836bf464f9SBarry Smith PetscErrorCode SNESDestroy(SNES *snes) 28849b94acceSBarry Smith { 28856849ba73SBarry Smith PetscErrorCode ierr; 28863a40ed3dSBarry Smith 28873a40ed3dSBarry Smith PetscFunctionBegin; 28886bf464f9SBarry Smith if (!*snes) PetscFunctionReturn(0); 28896bf464f9SBarry Smith PetscValidHeaderSpecific((*snes),SNES_CLASSID,1); 28906bf464f9SBarry Smith if (--((PetscObject)(*snes))->refct > 0) {*snes = 0; PetscFunctionReturn(0);} 2891d4bb536fSBarry Smith 28926bf464f9SBarry Smith ierr = SNESReset((*snes));CHKERRQ(ierr); 2893efd4aadfSBarry Smith ierr = SNESDestroy(&(*snes)->npc);CHKERRQ(ierr); 28946b8b9a38SLisandro Dalcin 2895e04113cfSBarry Smith /* if memory was published with SAWs then destroy it */ 2896e04113cfSBarry Smith ierr = PetscObjectSAWsViewOff((PetscObject)*snes);CHKERRQ(ierr); 28976bf464f9SBarry Smith if ((*snes)->ops->destroy) {ierr = (*((*snes))->ops->destroy)((*snes));CHKERRQ(ierr);} 28986d4c513bSLisandro Dalcin 289933124788SMatthew G. Knepley if ((*snes)->dm) {ierr = DMCoarsenHookRemove((*snes)->dm,DMCoarsenHook_SNESVecSol,DMRestrictHook_SNESVecSol,*snes);CHKERRQ(ierr);} 29006bf464f9SBarry Smith ierr = DMDestroy(&(*snes)->dm);CHKERRQ(ierr); 29016bf464f9SBarry Smith ierr = KSPDestroy(&(*snes)->ksp);CHKERRQ(ierr); 2902f1c6b773SPeter Brune ierr = SNESLineSearchDestroy(&(*snes)->linesearch);CHKERRQ(ierr); 29036b8b9a38SLisandro Dalcin 29046bf464f9SBarry Smith ierr = PetscFree((*snes)->kspconvctx);CHKERRQ(ierr); 29056bf464f9SBarry Smith if ((*snes)->ops->convergeddestroy) { 29066bf464f9SBarry Smith ierr = (*(*snes)->ops->convergeddestroy)((*snes)->cnvP);CHKERRQ(ierr); 29076b8b9a38SLisandro Dalcin } 29086bf464f9SBarry Smith if ((*snes)->conv_malloc) { 29096bf464f9SBarry Smith ierr = PetscFree((*snes)->conv_hist);CHKERRQ(ierr); 29106bf464f9SBarry Smith ierr = PetscFree((*snes)->conv_hist_its);CHKERRQ(ierr); 291158c9b817SLisandro Dalcin } 29126bf464f9SBarry Smith ierr = SNESMonitorCancel((*snes));CHKERRQ(ierr); 2913a79aaaedSSatish Balay ierr = PetscHeaderDestroy(snes);CHKERRQ(ierr); 29143a40ed3dSBarry Smith PetscFunctionReturn(0); 29159b94acceSBarry Smith } 29169b94acceSBarry Smith 29179b94acceSBarry Smith /* ----------- Routines to set solver parameters ---------- */ 29189b94acceSBarry Smith 2919a8054027SBarry Smith /*@ 2920a8054027SBarry Smith SNESSetLagPreconditioner - Determines when the preconditioner is rebuilt in the nonlinear solve. 2921a8054027SBarry Smith 29223f9fe445SBarry Smith Logically Collective on SNES 2923a8054027SBarry Smith 2924a8054027SBarry Smith Input Parameters: 2925a8054027SBarry Smith + snes - the SNES context 2926a8054027SBarry Smith - lag - -1 indicates NEVER rebuild, 1 means rebuild every time the Jacobian is computed within a single nonlinear solve, 2 means every second time 29273b4f5425SBarry Smith the Jacobian is built etc. -2 indicates rebuild preconditioner at next chance but then never rebuild after that 2928a8054027SBarry Smith 2929a8054027SBarry Smith Options Database Keys: 2930a8054027SBarry Smith . -snes_lag_preconditioner <lag> 2931a8054027SBarry Smith 2932a8054027SBarry Smith Notes: 2933a8054027SBarry Smith The default is 1 2934a8054027SBarry Smith The preconditioner is ALWAYS built in the first iteration of a nonlinear solve unless lag is -1 2935a8054027SBarry Smith If -1 is used before the very first nonlinear solve the preconditioner is still built because there is no previous preconditioner to use 2936a8054027SBarry Smith 2937a8054027SBarry Smith Level: intermediate 2938a8054027SBarry Smith 2939a8054027SBarry Smith .keywords: SNES, nonlinear, set, convergence, tolerances 2940a8054027SBarry Smith 2941e35cf81dSBarry Smith .seealso: SNESSetTrustRegionTolerance(), SNESGetLagPreconditioner(), SNESSetLagJacobian(), SNESGetLagJacobian() 2942a8054027SBarry Smith 2943a8054027SBarry Smith @*/ 29447087cfbeSBarry Smith PetscErrorCode SNESSetLagPreconditioner(SNES snes,PetscInt lag) 2945a8054027SBarry Smith { 2946a8054027SBarry Smith PetscFunctionBegin; 29470700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 2948e32f2f54SBarry Smith if (lag < -2) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Lag must be -2, -1, 1 or greater"); 2949e32f2f54SBarry Smith if (!lag) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Lag cannot be 0"); 2950c5eb9154SBarry Smith PetscValidLogicalCollectiveInt(snes,lag,2); 2951a8054027SBarry Smith snes->lagpreconditioner = lag; 2952a8054027SBarry Smith PetscFunctionReturn(0); 2953a8054027SBarry Smith } 2954a8054027SBarry Smith 2955efd51863SBarry Smith /*@ 2956efd51863SBarry Smith SNESSetGridSequence - sets the number of steps of grid sequencing that SNES does 2957efd51863SBarry Smith 2958efd51863SBarry Smith Logically Collective on SNES 2959efd51863SBarry Smith 2960efd51863SBarry Smith Input Parameters: 2961efd51863SBarry Smith + snes - the SNES context 2962efd51863SBarry Smith - steps - the number of refinements to do, defaults to 0 2963efd51863SBarry Smith 2964efd51863SBarry Smith Options Database Keys: 2965efd51863SBarry Smith . -snes_grid_sequence <steps> 2966efd51863SBarry Smith 2967efd51863SBarry Smith Level: intermediate 2968efd51863SBarry Smith 2969c0df2a02SJed Brown Notes: 2970c0df2a02SJed Brown Use SNESGetSolution() to extract the fine grid solution after grid sequencing. 2971c0df2a02SJed Brown 2972efd51863SBarry Smith .keywords: SNES, nonlinear, set, convergence, tolerances 2973efd51863SBarry Smith 2974fa19ca70SBarry Smith .seealso: SNESSetTrustRegionTolerance(), SNESGetLagPreconditioner(), SNESSetLagJacobian(), SNESGetLagJacobian(), SNESGetGridSequence() 2975efd51863SBarry Smith 2976efd51863SBarry Smith @*/ 2977efd51863SBarry Smith PetscErrorCode SNESSetGridSequence(SNES snes,PetscInt steps) 2978efd51863SBarry Smith { 2979efd51863SBarry Smith PetscFunctionBegin; 2980efd51863SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 2981efd51863SBarry Smith PetscValidLogicalCollectiveInt(snes,steps,2); 2982efd51863SBarry Smith snes->gridsequence = steps; 2983efd51863SBarry Smith PetscFunctionReturn(0); 2984efd51863SBarry Smith } 2985efd51863SBarry Smith 2986fa19ca70SBarry Smith /*@ 2987fa19ca70SBarry Smith SNESGetGridSequence - gets the number of steps of grid sequencing that SNES does 2988fa19ca70SBarry Smith 2989fa19ca70SBarry Smith Logically Collective on SNES 2990fa19ca70SBarry Smith 2991fa19ca70SBarry Smith Input Parameter: 2992fa19ca70SBarry Smith . snes - the SNES context 2993fa19ca70SBarry Smith 2994fa19ca70SBarry Smith Output Parameter: 2995fa19ca70SBarry Smith . steps - the number of refinements to do, defaults to 0 2996fa19ca70SBarry Smith 2997fa19ca70SBarry Smith Options Database Keys: 2998fa19ca70SBarry Smith . -snes_grid_sequence <steps> 2999fa19ca70SBarry Smith 3000fa19ca70SBarry Smith Level: intermediate 3001fa19ca70SBarry Smith 3002fa19ca70SBarry Smith Notes: 3003fa19ca70SBarry Smith Use SNESGetSolution() to extract the fine grid solution after grid sequencing. 3004fa19ca70SBarry Smith 3005fa19ca70SBarry Smith .keywords: SNES, nonlinear, set, convergence, tolerances 3006fa19ca70SBarry Smith 3007fa19ca70SBarry Smith .seealso: SNESSetTrustRegionTolerance(), SNESGetLagPreconditioner(), SNESSetLagJacobian(), SNESGetLagJacobian(), SNESSetGridSequence() 3008fa19ca70SBarry Smith 3009fa19ca70SBarry Smith @*/ 3010fa19ca70SBarry Smith PetscErrorCode SNESGetGridSequence(SNES snes,PetscInt *steps) 3011fa19ca70SBarry Smith { 3012fa19ca70SBarry Smith PetscFunctionBegin; 3013fa19ca70SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 3014fa19ca70SBarry Smith *steps = snes->gridsequence; 3015fa19ca70SBarry Smith PetscFunctionReturn(0); 3016fa19ca70SBarry Smith } 3017fa19ca70SBarry Smith 3018a8054027SBarry Smith /*@ 3019a8054027SBarry Smith SNESGetLagPreconditioner - Indicates how often the preconditioner is rebuilt 3020a8054027SBarry Smith 30213f9fe445SBarry Smith Not Collective 3022a8054027SBarry Smith 3023a8054027SBarry Smith Input Parameter: 3024a8054027SBarry Smith . snes - the SNES context 3025a8054027SBarry Smith 3026a8054027SBarry Smith Output Parameter: 3027a8054027SBarry Smith . lag - -1 indicates NEVER rebuild, 1 means rebuild every time the Jacobian is computed within a single nonlinear solve, 2 means every second time 30283b4f5425SBarry Smith the Jacobian is built etc. -2 indicates rebuild preconditioner at next chance but then never rebuild after that 3029a8054027SBarry Smith 3030a8054027SBarry Smith Options Database Keys: 3031a8054027SBarry Smith . -snes_lag_preconditioner <lag> 3032a8054027SBarry Smith 3033a8054027SBarry Smith Notes: 3034a8054027SBarry Smith The default is 1 3035a8054027SBarry Smith The preconditioner is ALWAYS built in the first iteration of a nonlinear solve unless lag is -1 3036a8054027SBarry Smith 3037a8054027SBarry Smith Level: intermediate 3038a8054027SBarry Smith 3039a8054027SBarry Smith .keywords: SNES, nonlinear, set, convergence, tolerances 3040a8054027SBarry Smith 3041a8054027SBarry Smith .seealso: SNESSetTrustRegionTolerance(), SNESSetLagPreconditioner() 3042a8054027SBarry Smith 3043a8054027SBarry Smith @*/ 30447087cfbeSBarry Smith PetscErrorCode SNESGetLagPreconditioner(SNES snes,PetscInt *lag) 3045a8054027SBarry Smith { 3046a8054027SBarry Smith PetscFunctionBegin; 30470700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 3048a8054027SBarry Smith *lag = snes->lagpreconditioner; 3049a8054027SBarry Smith PetscFunctionReturn(0); 3050a8054027SBarry Smith } 3051a8054027SBarry Smith 3052e35cf81dSBarry Smith /*@ 3053e35cf81dSBarry Smith SNESSetLagJacobian - Determines when the Jacobian is rebuilt in the nonlinear solve. See SNESSetLagPreconditioner() for determining how 3054e35cf81dSBarry Smith often the preconditioner is rebuilt. 3055e35cf81dSBarry Smith 30563f9fe445SBarry Smith Logically Collective on SNES 3057e35cf81dSBarry Smith 3058e35cf81dSBarry Smith Input Parameters: 3059e35cf81dSBarry Smith + snes - the SNES context 3060e35cf81dSBarry Smith - lag - -1 indicates NEVER rebuild, 1 means rebuild every time the Jacobian is computed within a single nonlinear solve, 2 means every second time 3061fe3ffe1eSBarry Smith the Jacobian is built etc. -2 means rebuild at next chance but then never again 3062e35cf81dSBarry Smith 3063e35cf81dSBarry Smith Options Database Keys: 3064e35cf81dSBarry Smith . -snes_lag_jacobian <lag> 3065e35cf81dSBarry Smith 3066e35cf81dSBarry Smith Notes: 3067e35cf81dSBarry Smith The default is 1 3068e35cf81dSBarry Smith The Jacobian is ALWAYS built in the first iteration of a nonlinear solve unless lag is -1 3069fe3ffe1eSBarry Smith If -1 is used before the very first nonlinear solve the CODE WILL FAIL! because no Jacobian is used, use -2 to indicate you want it recomputed 3070fe3ffe1eSBarry Smith at the next Newton step but never again (unless it is reset to another value) 3071e35cf81dSBarry Smith 3072e35cf81dSBarry Smith Level: intermediate 3073e35cf81dSBarry Smith 3074e35cf81dSBarry Smith .keywords: SNES, nonlinear, set, convergence, tolerances 3075e35cf81dSBarry Smith 3076e35cf81dSBarry Smith .seealso: SNESSetTrustRegionTolerance(), SNESGetLagPreconditioner(), SNESSetLagPreconditioner(), SNESGetLagJacobian() 3077e35cf81dSBarry Smith 3078e35cf81dSBarry Smith @*/ 30797087cfbeSBarry Smith PetscErrorCode SNESSetLagJacobian(SNES snes,PetscInt lag) 3080e35cf81dSBarry Smith { 3081e35cf81dSBarry Smith PetscFunctionBegin; 30820700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 3083e32f2f54SBarry Smith if (lag < -2) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Lag must be -2, -1, 1 or greater"); 3084e32f2f54SBarry Smith if (!lag) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Lag cannot be 0"); 3085c5eb9154SBarry Smith PetscValidLogicalCollectiveInt(snes,lag,2); 3086e35cf81dSBarry Smith snes->lagjacobian = lag; 3087e35cf81dSBarry Smith PetscFunctionReturn(0); 3088e35cf81dSBarry Smith } 3089e35cf81dSBarry Smith 3090e35cf81dSBarry Smith /*@ 3091e35cf81dSBarry Smith SNESGetLagJacobian - Indicates how often the Jacobian is rebuilt. See SNESGetLagPreconditioner() to determine when the preconditioner is rebuilt 3092e35cf81dSBarry Smith 30933f9fe445SBarry Smith Not Collective 3094e35cf81dSBarry Smith 3095e35cf81dSBarry Smith Input Parameter: 3096e35cf81dSBarry Smith . snes - the SNES context 3097e35cf81dSBarry Smith 3098e35cf81dSBarry Smith Output Parameter: 3099e35cf81dSBarry Smith . lag - -1 indicates NEVER rebuild, 1 means rebuild every time the Jacobian is computed within a single nonlinear solve, 2 means every second time 3100e35cf81dSBarry Smith the Jacobian is built etc. 3101e35cf81dSBarry Smith 3102e35cf81dSBarry Smith Options Database Keys: 3103e35cf81dSBarry Smith . -snes_lag_jacobian <lag> 3104e35cf81dSBarry Smith 3105e35cf81dSBarry Smith Notes: 3106e35cf81dSBarry Smith The default is 1 3107e35cf81dSBarry Smith The jacobian is ALWAYS built in the first iteration of a nonlinear solve unless lag is -1 3108e35cf81dSBarry Smith 3109e35cf81dSBarry Smith Level: intermediate 3110e35cf81dSBarry Smith 3111e35cf81dSBarry Smith .keywords: SNES, nonlinear, set, convergence, tolerances 3112e35cf81dSBarry Smith 3113e35cf81dSBarry Smith .seealso: SNESSetTrustRegionTolerance(), SNESSetLagJacobian(), SNESSetLagPreconditioner(), SNESGetLagPreconditioner() 3114e35cf81dSBarry Smith 3115e35cf81dSBarry Smith @*/ 31167087cfbeSBarry Smith PetscErrorCode SNESGetLagJacobian(SNES snes,PetscInt *lag) 3117e35cf81dSBarry Smith { 3118e35cf81dSBarry Smith PetscFunctionBegin; 31190700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 3120e35cf81dSBarry Smith *lag = snes->lagjacobian; 3121e35cf81dSBarry Smith PetscFunctionReturn(0); 3122e35cf81dSBarry Smith } 3123e35cf81dSBarry Smith 312437ec4e1aSPeter Brune /*@ 312537ec4e1aSPeter Brune SNESSetLagJacobianPersists - Set whether or not the Jacobian lagging persists through multiple solves 312637ec4e1aSPeter Brune 312737ec4e1aSPeter Brune Logically collective on SNES 312837ec4e1aSPeter Brune 312937ec4e1aSPeter Brune Input Parameter: 313037ec4e1aSPeter Brune + snes - the SNES context 31319d7e2deaSPeter Brune - flg - jacobian lagging persists if true 313237ec4e1aSPeter Brune 313337ec4e1aSPeter Brune Options Database Keys: 313437ec4e1aSPeter Brune . -snes_lag_jacobian_persists <flg> 313537ec4e1aSPeter Brune 313637ec4e1aSPeter Brune Notes: This is useful both for nonlinear preconditioning, where it's appropriate to have the Jacobian be stale by 313737ec4e1aSPeter Brune several solves, and for implicit time-stepping, where Jacobian lagging in the inner nonlinear solve over several 313837ec4e1aSPeter Brune timesteps may present huge efficiency gains. 313937ec4e1aSPeter Brune 314037ec4e1aSPeter Brune Level: developer 314137ec4e1aSPeter Brune 3142be95d8f1SBarry Smith .keywords: SNES, nonlinear, lag 314337ec4e1aSPeter Brune 3144be95d8f1SBarry Smith .seealso: SNESSetLagPreconditionerPersists(), SNESSetLagJacobian(), SNESGetLagJacobian(), SNESGetNPC() 314537ec4e1aSPeter Brune 314637ec4e1aSPeter Brune @*/ 314737ec4e1aSPeter Brune PetscErrorCode SNESSetLagJacobianPersists(SNES snes,PetscBool flg) 314837ec4e1aSPeter Brune { 314937ec4e1aSPeter Brune PetscFunctionBegin; 315037ec4e1aSPeter Brune PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 315137ec4e1aSPeter Brune PetscValidLogicalCollectiveBool(snes,flg,2); 315237ec4e1aSPeter Brune snes->lagjac_persist = flg; 315337ec4e1aSPeter Brune PetscFunctionReturn(0); 315437ec4e1aSPeter Brune } 315537ec4e1aSPeter Brune 315637ec4e1aSPeter Brune /*@ 315737ec4e1aSPeter Brune SNESSetLagPreconditionerPersists - Set whether or not the preconditioner lagging persists through multiple solves 315837ec4e1aSPeter Brune 315937ec4e1aSPeter Brune Logically Collective on SNES 316037ec4e1aSPeter Brune 316137ec4e1aSPeter Brune Input Parameter: 316237ec4e1aSPeter Brune + snes - the SNES context 31639d7e2deaSPeter Brune - flg - preconditioner lagging persists if true 316437ec4e1aSPeter Brune 316537ec4e1aSPeter Brune Options Database Keys: 316637ec4e1aSPeter Brune . -snes_lag_jacobian_persists <flg> 316737ec4e1aSPeter Brune 316837ec4e1aSPeter Brune Notes: This is useful both for nonlinear preconditioning, where it's appropriate to have the preconditioner be stale 316937ec4e1aSPeter Brune by several solves, and for implicit time-stepping, where preconditioner lagging in the inner nonlinear solve over 317037ec4e1aSPeter Brune several timesteps may present huge efficiency gains. 317137ec4e1aSPeter Brune 317237ec4e1aSPeter Brune Level: developer 317337ec4e1aSPeter Brune 3174be95d8f1SBarry Smith .keywords: SNES, nonlinear, lag 317537ec4e1aSPeter Brune 3176be95d8f1SBarry Smith .seealso: SNESSetLagJacobianPersists(), SNESSetLagJacobian(), SNESGetLagJacobian(), SNESGetNPC() 317737ec4e1aSPeter Brune 317837ec4e1aSPeter Brune @*/ 317937ec4e1aSPeter Brune PetscErrorCode SNESSetLagPreconditionerPersists(SNES snes,PetscBool flg) 318037ec4e1aSPeter Brune { 318137ec4e1aSPeter Brune PetscFunctionBegin; 318237ec4e1aSPeter Brune PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 318337ec4e1aSPeter Brune PetscValidLogicalCollectiveBool(snes,flg,2); 318437ec4e1aSPeter Brune snes->lagpre_persist = flg; 318537ec4e1aSPeter Brune PetscFunctionReturn(0); 318637ec4e1aSPeter Brune } 318737ec4e1aSPeter Brune 31889b94acceSBarry Smith /*@ 3189be5caee7SBarry Smith SNESSetForceIteration - force SNESSolve() to take at least one iteration regardless of the initial residual norm 3190be5caee7SBarry Smith 3191be5caee7SBarry Smith Logically Collective on SNES 3192be5caee7SBarry Smith 3193be5caee7SBarry Smith Input Parameters: 3194be5caee7SBarry Smith + snes - the SNES context 3195be5caee7SBarry Smith - force - PETSC_TRUE require at least one iteration 3196be5caee7SBarry Smith 3197be5caee7SBarry Smith Options Database Keys: 3198be5caee7SBarry Smith . -snes_force_iteration <force> - Sets forcing an iteration 3199be5caee7SBarry Smith 3200be5caee7SBarry Smith Notes: 3201be5caee7SBarry Smith This is used sometimes with TS to prevent TS from detecting a false steady state solution 3202be5caee7SBarry Smith 3203be5caee7SBarry Smith Level: intermediate 3204be5caee7SBarry Smith 3205be5caee7SBarry Smith .keywords: SNES, nonlinear, set, convergence, tolerances 3206be5caee7SBarry Smith 3207be5caee7SBarry Smith .seealso: SNESSetTrustRegionTolerance(), SNESSetDivergenceTolerance() 3208be5caee7SBarry Smith @*/ 3209be5caee7SBarry Smith PetscErrorCode SNESSetForceIteration(SNES snes,PetscBool force) 3210be5caee7SBarry Smith { 3211be5caee7SBarry Smith PetscFunctionBegin; 3212be5caee7SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 3213be5caee7SBarry Smith snes->forceiteration = force; 3214be5caee7SBarry Smith PetscFunctionReturn(0); 3215be5caee7SBarry Smith } 3216be5caee7SBarry Smith 3217be5caee7SBarry Smith 3218be5caee7SBarry Smith /*@ 3219d7a720efSLois Curfman McInnes SNESSetTolerances - Sets various parameters used in convergence tests. 32209b94acceSBarry Smith 32213f9fe445SBarry Smith Logically Collective on SNES 3222c7afd0dbSLois Curfman McInnes 32239b94acceSBarry Smith Input Parameters: 3224c7afd0dbSLois Curfman McInnes + snes - the SNES context 322570441072SBarry Smith . abstol - absolute convergence tolerance 322633174efeSLois Curfman McInnes . rtol - relative convergence tolerance 32275358d0d4SBarry Smith . stol - convergence tolerance in terms of the norm of the change in the solution between steps, || delta x || < stol*|| x || 322833174efeSLois Curfman McInnes . maxit - maximum number of iterations 3229c7afd0dbSLois Curfman McInnes - maxf - maximum number of function evaluations 3230fee21e36SBarry Smith 323133174efeSLois Curfman McInnes Options Database Keys: 323270441072SBarry Smith + -snes_atol <abstol> - Sets abstol 3233c7afd0dbSLois Curfman McInnes . -snes_rtol <rtol> - Sets rtol 3234c7afd0dbSLois Curfman McInnes . -snes_stol <stol> - Sets stol 3235c7afd0dbSLois Curfman McInnes . -snes_max_it <maxit> - Sets maxit 3236c7afd0dbSLois Curfman McInnes - -snes_max_funcs <maxf> - Sets maxf 32379b94acceSBarry Smith 3238d7a720efSLois Curfman McInnes Notes: 32399b94acceSBarry Smith The default maximum number of iterations is 50. 32409b94acceSBarry Smith The default maximum number of function evaluations is 1000. 32419b94acceSBarry Smith 324236851e7fSLois Curfman McInnes Level: intermediate 324336851e7fSLois Curfman McInnes 324433174efeSLois Curfman McInnes .keywords: SNES, nonlinear, set, convergence, tolerances 32459b94acceSBarry Smith 3246be5caee7SBarry Smith .seealso: SNESSetTrustRegionTolerance(), SNESSetDivergenceTolerance(), SNESSetForceIteration() 32479b94acceSBarry Smith @*/ 32487087cfbeSBarry Smith PetscErrorCode SNESSetTolerances(SNES snes,PetscReal abstol,PetscReal rtol,PetscReal stol,PetscInt maxit,PetscInt maxf) 32499b94acceSBarry Smith { 32503a40ed3dSBarry Smith PetscFunctionBegin; 32510700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 3252c5eb9154SBarry Smith PetscValidLogicalCollectiveReal(snes,abstol,2); 3253c5eb9154SBarry Smith PetscValidLogicalCollectiveReal(snes,rtol,3); 3254c5eb9154SBarry Smith PetscValidLogicalCollectiveReal(snes,stol,4); 3255c5eb9154SBarry Smith PetscValidLogicalCollectiveInt(snes,maxit,5); 3256c5eb9154SBarry Smith PetscValidLogicalCollectiveInt(snes,maxf,6); 3257c5eb9154SBarry Smith 3258ab54825eSJed Brown if (abstol != PETSC_DEFAULT) { 325957622a8eSBarry Smith if (abstol < 0.0) SETERRQ1(PetscObjectComm((PetscObject)snes),PETSC_ERR_ARG_OUTOFRANGE,"Absolute tolerance %g must be non-negative",(double)abstol); 3260ab54825eSJed Brown snes->abstol = abstol; 3261ab54825eSJed Brown } 3262ab54825eSJed Brown if (rtol != PETSC_DEFAULT) { 326357622a8eSBarry Smith if (rtol < 0.0 || 1.0 <= rtol) SETERRQ1(PetscObjectComm((PetscObject)snes),PETSC_ERR_ARG_OUTOFRANGE,"Relative tolerance %g must be non-negative and less than 1.0",(double)rtol); 3264ab54825eSJed Brown snes->rtol = rtol; 3265ab54825eSJed Brown } 3266ab54825eSJed Brown if (stol != PETSC_DEFAULT) { 326757622a8eSBarry Smith if (stol < 0.0) SETERRQ1(PetscObjectComm((PetscObject)snes),PETSC_ERR_ARG_OUTOFRANGE,"Step tolerance %g must be non-negative",(double)stol); 3268c60f73f4SPeter Brune snes->stol = stol; 3269ab54825eSJed Brown } 3270ab54825eSJed Brown if (maxit != PETSC_DEFAULT) { 3271ce94432eSBarry Smith if (maxit < 0) SETERRQ1(PetscObjectComm((PetscObject)snes),PETSC_ERR_ARG_OUTOFRANGE,"Maximum number of iterations %D must be non-negative",maxit); 3272ab54825eSJed Brown snes->max_its = maxit; 3273ab54825eSJed Brown } 3274ab54825eSJed Brown if (maxf != PETSC_DEFAULT) { 3275ce94432eSBarry Smith if (maxf < 0) SETERRQ1(PetscObjectComm((PetscObject)snes),PETSC_ERR_ARG_OUTOFRANGE,"Maximum number of function evaluations %D must be non-negative",maxf); 3276ab54825eSJed Brown snes->max_funcs = maxf; 3277ab54825eSJed Brown } 327888976e71SPeter Brune snes->tolerancesset = PETSC_TRUE; 32793a40ed3dSBarry Smith PetscFunctionReturn(0); 32809b94acceSBarry Smith } 32819b94acceSBarry Smith 3282e4d06f11SPatrick Farrell /*@ 3283e4d06f11SPatrick Farrell SNESSetDivergenceTolerance - Sets the divergence tolerance used for the SNES divergence test. 3284e4d06f11SPatrick Farrell 3285e4d06f11SPatrick Farrell Logically Collective on SNES 3286e4d06f11SPatrick Farrell 3287e4d06f11SPatrick Farrell Input Parameters: 3288e4d06f11SPatrick Farrell + snes - the SNES context 3289e4d06f11SPatrick Farrell - divtol - the divergence tolerance. Use -1 to deactivate the test. 3290e4d06f11SPatrick Farrell 3291e4d06f11SPatrick Farrell Options Database Keys: 3292e4d06f11SPatrick Farrell + -snes_divergence_tolerance <divtol> - Sets divtol 3293e4d06f11SPatrick Farrell 3294e4d06f11SPatrick Farrell Notes: 3295e4d06f11SPatrick Farrell The default divergence tolerance is 1e4. 3296e4d06f11SPatrick Farrell 3297e4d06f11SPatrick Farrell Level: intermediate 3298e4d06f11SPatrick Farrell 3299e4d06f11SPatrick Farrell .keywords: SNES, nonlinear, set, divergence, tolerance 3300e4d06f11SPatrick Farrell 3301e4d06f11SPatrick Farrell .seealso: SNESSetTolerances(), SNESGetDivergenceTolerance 3302e4d06f11SPatrick Farrell @*/ 3303e4d06f11SPatrick Farrell PetscErrorCode SNESSetDivergenceTolerance(SNES snes,PetscReal divtol) 3304e4d06f11SPatrick Farrell { 3305e4d06f11SPatrick Farrell PetscFunctionBegin; 3306e4d06f11SPatrick Farrell PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 3307e4d06f11SPatrick Farrell PetscValidLogicalCollectiveReal(snes,divtol,2); 3308e4d06f11SPatrick Farrell 3309e4d06f11SPatrick Farrell if (divtol != PETSC_DEFAULT) { 3310e4d06f11SPatrick Farrell snes->divtol = divtol; 3311e4d06f11SPatrick Farrell } 3312e4d06f11SPatrick Farrell else { 3313e4d06f11SPatrick Farrell snes->divtol = 1.0e4; 3314e4d06f11SPatrick Farrell } 3315e4d06f11SPatrick Farrell PetscFunctionReturn(0); 3316e4d06f11SPatrick Farrell } 3317e4d06f11SPatrick Farrell 33189b94acceSBarry Smith /*@ 331933174efeSLois Curfman McInnes SNESGetTolerances - Gets various parameters used in convergence tests. 332033174efeSLois Curfman McInnes 3321c7afd0dbSLois Curfman McInnes Not Collective 3322c7afd0dbSLois Curfman McInnes 332333174efeSLois Curfman McInnes Input Parameters: 3324c7afd0dbSLois Curfman McInnes + snes - the SNES context 332585385478SLisandro Dalcin . atol - absolute convergence tolerance 332633174efeSLois Curfman McInnes . rtol - relative convergence tolerance 332733174efeSLois Curfman McInnes . stol - convergence tolerance in terms of the norm 332833174efeSLois Curfman McInnes of the change in the solution between steps 332933174efeSLois Curfman McInnes . maxit - maximum number of iterations 3330c7afd0dbSLois Curfman McInnes - maxf - maximum number of function evaluations 3331fee21e36SBarry Smith 333233174efeSLois Curfman McInnes Notes: 33330298fd71SBarry Smith The user can specify NULL for any parameter that is not needed. 333433174efeSLois Curfman McInnes 333536851e7fSLois Curfman McInnes Level: intermediate 333636851e7fSLois Curfman McInnes 333733174efeSLois Curfman McInnes .keywords: SNES, nonlinear, get, convergence, tolerances 333833174efeSLois Curfman McInnes 333933174efeSLois Curfman McInnes .seealso: SNESSetTolerances() 334033174efeSLois Curfman McInnes @*/ 33417087cfbeSBarry Smith PetscErrorCode SNESGetTolerances(SNES snes,PetscReal *atol,PetscReal *rtol,PetscReal *stol,PetscInt *maxit,PetscInt *maxf) 334233174efeSLois Curfman McInnes { 33433a40ed3dSBarry Smith PetscFunctionBegin; 33440700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 334585385478SLisandro Dalcin if (atol) *atol = snes->abstol; 334633174efeSLois Curfman McInnes if (rtol) *rtol = snes->rtol; 3347c60f73f4SPeter Brune if (stol) *stol = snes->stol; 334833174efeSLois Curfman McInnes if (maxit) *maxit = snes->max_its; 334933174efeSLois Curfman McInnes if (maxf) *maxf = snes->max_funcs; 33503a40ed3dSBarry Smith PetscFunctionReturn(0); 335133174efeSLois Curfman McInnes } 335233174efeSLois Curfman McInnes 3353e4d06f11SPatrick Farrell /*@ 3354e4d06f11SPatrick Farrell SNESGetDivergenceTolerance - Gets divergence tolerance used in divergence test. 3355e4d06f11SPatrick Farrell 3356e4d06f11SPatrick Farrell Not Collective 3357e4d06f11SPatrick Farrell 3358e4d06f11SPatrick Farrell Input Parameters: 3359e4d06f11SPatrick Farrell + snes - the SNES context 3360e4d06f11SPatrick Farrell - divtol - divergence tolerance 3361e4d06f11SPatrick Farrell 3362e4d06f11SPatrick Farrell Level: intermediate 3363e4d06f11SPatrick Farrell 3364e4d06f11SPatrick Farrell .keywords: SNES, nonlinear, get, divergence, tolerance 3365e4d06f11SPatrick Farrell 3366e4d06f11SPatrick Farrell .seealso: SNESSetDivergenceTolerance() 3367e4d06f11SPatrick Farrell @*/ 3368e4d06f11SPatrick Farrell PetscErrorCode SNESGetDivergenceTolerance(SNES snes,PetscReal *divtol) 3369e4d06f11SPatrick Farrell { 3370e4d06f11SPatrick Farrell PetscFunctionBegin; 3371e4d06f11SPatrick Farrell PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 3372e4d06f11SPatrick Farrell if (divtol) *divtol = snes->divtol; 3373e4d06f11SPatrick Farrell PetscFunctionReturn(0); 3374e4d06f11SPatrick Farrell } 3375e4d06f11SPatrick Farrell 337633174efeSLois Curfman McInnes /*@ 33779b94acceSBarry Smith SNESSetTrustRegionTolerance - Sets the trust region parameter tolerance. 33789b94acceSBarry Smith 33793f9fe445SBarry Smith Logically Collective on SNES 3380fee21e36SBarry Smith 3381c7afd0dbSLois Curfman McInnes Input Parameters: 3382c7afd0dbSLois Curfman McInnes + snes - the SNES context 3383c7afd0dbSLois Curfman McInnes - tol - tolerance 3384c7afd0dbSLois Curfman McInnes 33859b94acceSBarry Smith Options Database Key: 3386c7afd0dbSLois Curfman McInnes . -snes_trtol <tol> - Sets tol 33879b94acceSBarry Smith 338836851e7fSLois Curfman McInnes Level: intermediate 338936851e7fSLois Curfman McInnes 33909b94acceSBarry Smith .keywords: SNES, nonlinear, set, trust region, tolerance 33919b94acceSBarry Smith 33922492ecdbSBarry Smith .seealso: SNESSetTolerances() 33939b94acceSBarry Smith @*/ 33947087cfbeSBarry Smith PetscErrorCode SNESSetTrustRegionTolerance(SNES snes,PetscReal tol) 33959b94acceSBarry Smith { 33963a40ed3dSBarry Smith PetscFunctionBegin; 33970700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 3398c5eb9154SBarry Smith PetscValidLogicalCollectiveReal(snes,tol,2); 33999b94acceSBarry Smith snes->deltatol = tol; 34003a40ed3dSBarry Smith PetscFunctionReturn(0); 34019b94acceSBarry Smith } 34029b94acceSBarry Smith 3403df9fa365SBarry Smith /* 3404df9fa365SBarry Smith Duplicate the lg monitors for SNES from KSP; for some reason with 3405df9fa365SBarry Smith dynamic libraries things don't work under Sun4 if we just use 3406df9fa365SBarry Smith macros instead of functions 3407df9fa365SBarry Smith */ 3408d96771aaSLisandro Dalcin PetscErrorCode SNESMonitorLGResidualNorm(SNES snes,PetscInt it,PetscReal norm,void *ctx) 3409ce1608b8SBarry Smith { 3410dfbe8321SBarry Smith PetscErrorCode ierr; 3411ce1608b8SBarry Smith 3412ce1608b8SBarry Smith PetscFunctionBegin; 34130700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 3414d96771aaSLisandro Dalcin ierr = KSPMonitorLGResidualNorm((KSP)snes,it,norm,ctx);CHKERRQ(ierr); 3415ce1608b8SBarry Smith PetscFunctionReturn(0); 3416ce1608b8SBarry Smith } 3417ce1608b8SBarry Smith 3418d96771aaSLisandro Dalcin PetscErrorCode SNESMonitorLGCreate(MPI_Comm comm,const char host[],const char label[],int x,int y,int m,int n,PetscDrawLG *lgctx) 3419df9fa365SBarry Smith { 3420dfbe8321SBarry Smith PetscErrorCode ierr; 3421df9fa365SBarry Smith 3422df9fa365SBarry Smith PetscFunctionBegin; 3423d96771aaSLisandro Dalcin ierr = KSPMonitorLGResidualNormCreate(comm,host,label,x,y,m,n,lgctx);CHKERRQ(ierr); 3424df9fa365SBarry Smith PetscFunctionReturn(0); 3425df9fa365SBarry Smith } 3426df9fa365SBarry Smith 34276ba87a44SLisandro Dalcin PETSC_INTERN PetscErrorCode SNESMonitorRange_Private(SNES,PetscInt,PetscReal*); 34286ba87a44SLisandro Dalcin 34297087cfbeSBarry Smith PetscErrorCode SNESMonitorLGRange(SNES snes,PetscInt n,PetscReal rnorm,void *monctx) 3430b271bb04SBarry Smith { 3431b271bb04SBarry Smith PetscDrawLG lg; 3432b271bb04SBarry Smith PetscErrorCode ierr; 3433b271bb04SBarry Smith PetscReal x,y,per; 3434b271bb04SBarry Smith PetscViewer v = (PetscViewer)monctx; 3435b271bb04SBarry Smith static PetscReal prev; /* should be in the context */ 3436b271bb04SBarry Smith PetscDraw draw; 3437b271bb04SBarry Smith 3438459f5d12SBarry Smith PetscFunctionBegin; 34394d4332d5SBarry Smith PetscValidHeaderSpecific(v,PETSC_VIEWER_CLASSID,4); 3440b271bb04SBarry Smith ierr = PetscViewerDrawGetDrawLG(v,0,&lg);CHKERRQ(ierr); 3441b271bb04SBarry Smith if (!n) {ierr = PetscDrawLGReset(lg);CHKERRQ(ierr);} 3442b271bb04SBarry Smith ierr = PetscDrawLGGetDraw(lg,&draw);CHKERRQ(ierr); 3443b271bb04SBarry Smith ierr = PetscDrawSetTitle(draw,"Residual norm");CHKERRQ(ierr); 3444b271bb04SBarry Smith x = (PetscReal)n; 344577b4d14cSPeter Brune if (rnorm > 0.0) y = PetscLog10Real(rnorm); 344694c9c6d3SKarl Rupp else y = -15.0; 3447b271bb04SBarry Smith ierr = PetscDrawLGAddPoint(lg,&x,&y);CHKERRQ(ierr); 34486934998bSLisandro Dalcin if (n < 20 || !(n % 5) || snes->reason) { 3449b271bb04SBarry Smith ierr = PetscDrawLGDraw(lg);CHKERRQ(ierr); 34506934998bSLisandro Dalcin ierr = PetscDrawLGSave(lg);CHKERRQ(ierr); 3451b271bb04SBarry Smith } 3452b271bb04SBarry Smith 3453b271bb04SBarry Smith ierr = PetscViewerDrawGetDrawLG(v,1,&lg);CHKERRQ(ierr); 3454b271bb04SBarry Smith if (!n) {ierr = PetscDrawLGReset(lg);CHKERRQ(ierr);} 3455b271bb04SBarry Smith ierr = PetscDrawLGGetDraw(lg,&draw);CHKERRQ(ierr); 3456b271bb04SBarry Smith ierr = PetscDrawSetTitle(draw,"% elemts > .2*max elemt");CHKERRQ(ierr); 3457b271bb04SBarry Smith ierr = SNESMonitorRange_Private(snes,n,&per);CHKERRQ(ierr); 3458b271bb04SBarry Smith x = (PetscReal)n; 3459b271bb04SBarry Smith y = 100.0*per; 3460b271bb04SBarry Smith ierr = PetscDrawLGAddPoint(lg,&x,&y);CHKERRQ(ierr); 34616934998bSLisandro Dalcin if (n < 20 || !(n % 5) || snes->reason) { 3462b271bb04SBarry Smith ierr = PetscDrawLGDraw(lg);CHKERRQ(ierr); 34636934998bSLisandro Dalcin ierr = PetscDrawLGSave(lg);CHKERRQ(ierr); 3464b271bb04SBarry Smith } 3465b271bb04SBarry Smith 3466b271bb04SBarry Smith ierr = PetscViewerDrawGetDrawLG(v,2,&lg);CHKERRQ(ierr); 3467b271bb04SBarry Smith if (!n) {prev = rnorm;ierr = PetscDrawLGReset(lg);CHKERRQ(ierr);} 3468b271bb04SBarry Smith ierr = PetscDrawLGGetDraw(lg,&draw);CHKERRQ(ierr); 3469b271bb04SBarry Smith ierr = PetscDrawSetTitle(draw,"(norm -oldnorm)/oldnorm");CHKERRQ(ierr); 3470b271bb04SBarry Smith x = (PetscReal)n; 3471b271bb04SBarry Smith y = (prev - rnorm)/prev; 3472b271bb04SBarry Smith ierr = PetscDrawLGAddPoint(lg,&x,&y);CHKERRQ(ierr); 34736934998bSLisandro Dalcin if (n < 20 || !(n % 5) || snes->reason) { 3474b271bb04SBarry Smith ierr = PetscDrawLGDraw(lg);CHKERRQ(ierr); 34756934998bSLisandro Dalcin ierr = PetscDrawLGSave(lg);CHKERRQ(ierr); 3476b271bb04SBarry Smith } 3477b271bb04SBarry Smith 3478b271bb04SBarry Smith ierr = PetscViewerDrawGetDrawLG(v,3,&lg);CHKERRQ(ierr); 3479b271bb04SBarry Smith if (!n) {ierr = PetscDrawLGReset(lg);CHKERRQ(ierr);} 3480b271bb04SBarry Smith ierr = PetscDrawLGGetDraw(lg,&draw);CHKERRQ(ierr); 3481b271bb04SBarry Smith ierr = PetscDrawSetTitle(draw,"(norm -oldnorm)/oldnorm*(% > .2 max)");CHKERRQ(ierr); 3482b271bb04SBarry Smith x = (PetscReal)n; 3483b271bb04SBarry Smith y = (prev - rnorm)/(prev*per); 3484b271bb04SBarry Smith if (n > 2) { /*skip initial crazy value */ 3485b271bb04SBarry Smith ierr = PetscDrawLGAddPoint(lg,&x,&y);CHKERRQ(ierr); 3486b271bb04SBarry Smith } 34876934998bSLisandro Dalcin if (n < 20 || !(n % 5) || snes->reason) { 3488b271bb04SBarry Smith ierr = PetscDrawLGDraw(lg);CHKERRQ(ierr); 34896934998bSLisandro Dalcin ierr = PetscDrawLGSave(lg);CHKERRQ(ierr); 3490b271bb04SBarry Smith } 3491b271bb04SBarry Smith prev = rnorm; 3492b271bb04SBarry Smith PetscFunctionReturn(0); 3493b271bb04SBarry Smith } 3494b271bb04SBarry Smith 3495228d79bcSJed Brown /*@ 3496228d79bcSJed Brown SNESMonitor - runs the user provided monitor routines, if they exist 3497228d79bcSJed Brown 3498228d79bcSJed Brown Collective on SNES 3499228d79bcSJed Brown 3500228d79bcSJed Brown Input Parameters: 3501228d79bcSJed Brown + snes - nonlinear solver context obtained from SNESCreate() 3502228d79bcSJed Brown . iter - iteration number 3503228d79bcSJed Brown - rnorm - relative norm of the residual 3504228d79bcSJed Brown 3505228d79bcSJed Brown Notes: 3506228d79bcSJed Brown This routine is called by the SNES implementations. 3507228d79bcSJed Brown It does not typically need to be called by the user. 3508228d79bcSJed Brown 3509228d79bcSJed Brown Level: developer 3510228d79bcSJed Brown 3511228d79bcSJed Brown .seealso: SNESMonitorSet() 3512228d79bcSJed Brown @*/ 35137a03ce2fSLisandro Dalcin PetscErrorCode SNESMonitor(SNES snes,PetscInt iter,PetscReal rnorm) 35147a03ce2fSLisandro Dalcin { 35157a03ce2fSLisandro Dalcin PetscErrorCode ierr; 35167a03ce2fSLisandro Dalcin PetscInt i,n = snes->numbermonitors; 35177a03ce2fSLisandro Dalcin 35187a03ce2fSLisandro Dalcin PetscFunctionBegin; 35195edff71fSBarry Smith ierr = VecLockPush(snes->vec_sol);CHKERRQ(ierr); 35207a03ce2fSLisandro Dalcin for (i=0; i<n; i++) { 35217a03ce2fSLisandro Dalcin ierr = (*snes->monitor[i])(snes,iter,rnorm,snes->monitorcontext[i]);CHKERRQ(ierr); 35227a03ce2fSLisandro Dalcin } 35235edff71fSBarry Smith ierr = VecLockPop(snes->vec_sol);CHKERRQ(ierr); 35247a03ce2fSLisandro Dalcin PetscFunctionReturn(0); 35257a03ce2fSLisandro Dalcin } 35267a03ce2fSLisandro Dalcin 35279b94acceSBarry Smith /* ------------ Routines to set performance monitoring options ----------- */ 35289b94acceSBarry Smith 3529bf388a1fSBarry Smith /*MC 3530bf388a1fSBarry Smith SNESMonitorFunction - functional form passed to SNESMonitorSet() to monitor convergence of nonlinear solver 3531bf388a1fSBarry Smith 3532bf388a1fSBarry Smith Synopsis: 3533aaa7dc30SBarry Smith #include <petscsnes.h> 3534bf388a1fSBarry Smith $ PetscErrorCode SNESMonitorFunction(SNES snes,PetscInt its, PetscReal norm,void *mctx) 3535bf388a1fSBarry Smith 3536bf388a1fSBarry Smith + snes - the SNES context 3537bf388a1fSBarry Smith . its - iteration number 3538bf388a1fSBarry Smith . norm - 2-norm function value (may be estimated) 3539bf388a1fSBarry Smith - mctx - [optional] monitoring context 3540bf388a1fSBarry Smith 3541878cb397SSatish Balay Level: advanced 3542878cb397SSatish Balay 3543bf388a1fSBarry Smith .seealso: SNESMonitorSet(), SNESMonitorGet() 3544bf388a1fSBarry Smith M*/ 3545bf388a1fSBarry Smith 35469b94acceSBarry Smith /*@C 3547a6570f20SBarry Smith SNESMonitorSet - Sets an ADDITIONAL function that is to be used at every 35489b94acceSBarry Smith iteration of the nonlinear solver to display the iteration's 35499b94acceSBarry Smith progress. 35509b94acceSBarry Smith 35513f9fe445SBarry Smith Logically Collective on SNES 3552fee21e36SBarry Smith 3553c7afd0dbSLois Curfman McInnes Input Parameters: 3554c7afd0dbSLois Curfman McInnes + snes - the SNES context 35556e4dcb14SBarry Smith . f - the monitor function, see SNESMonitorFunction for the calling sequence 3556b8a78c4aSBarry Smith . mctx - [optional] user-defined context for private data for the 35570298fd71SBarry Smith monitor routine (use NULL if no context is desired) 3558b3006f0bSLois Curfman McInnes - monitordestroy - [optional] routine that frees monitor context 35590298fd71SBarry Smith (may be NULL) 35609b94acceSBarry Smith 35619665c990SLois Curfman McInnes Options Database Keys: 3562a6570f20SBarry Smith + -snes_monitor - sets SNESMonitorDefault() 35634619e776SBarry Smith . -snes_monitor_lg_residualnorm - sets line graph monitor, 3564a6570f20SBarry Smith uses SNESMonitorLGCreate() 3565cca6129bSJed Brown - -snes_monitor_cancel - cancels all monitors that have 3566c7afd0dbSLois Curfman McInnes been hardwired into a code by 3567a6570f20SBarry Smith calls to SNESMonitorSet(), but 3568c7afd0dbSLois Curfman McInnes does not cancel those set via 3569c7afd0dbSLois Curfman McInnes the options database. 35709665c990SLois Curfman McInnes 3571639f9d9dSBarry Smith Notes: 35726bc08f3fSLois Curfman McInnes Several different monitoring routines may be set by calling 3573a6570f20SBarry Smith SNESMonitorSet() multiple times; all will be called in the 35746bc08f3fSLois Curfman McInnes order in which they were set. 3575639f9d9dSBarry Smith 3576025f1a04SBarry Smith Fortran notes: Only a single monitor function can be set for each SNES object 3577025f1a04SBarry Smith 357836851e7fSLois Curfman McInnes Level: intermediate 357936851e7fSLois Curfman McInnes 35809b94acceSBarry Smith .keywords: SNES, nonlinear, set, monitor 35819b94acceSBarry Smith 3582bf388a1fSBarry Smith .seealso: SNESMonitorDefault(), SNESMonitorCancel(), SNESMonitorFunction 35839b94acceSBarry Smith @*/ 35846e4dcb14SBarry Smith PetscErrorCode SNESMonitorSet(SNES snes,PetscErrorCode (*f)(SNES,PetscInt,PetscReal,void*),void *mctx,PetscErrorCode (*monitordestroy)(void**)) 35859b94acceSBarry Smith { 3586b90d0a6eSBarry Smith PetscInt i; 3587649052a6SBarry Smith PetscErrorCode ierr; 358878064530SBarry Smith PetscBool identical; 3589b90d0a6eSBarry Smith 35903a40ed3dSBarry Smith PetscFunctionBegin; 35910700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 3592b90d0a6eSBarry Smith for (i=0; i<snes->numbermonitors;i++) { 359378064530SBarry Smith ierr = PetscMonitorCompare((PetscErrorCode (*)(void))f,mctx,monitordestroy,(PetscErrorCode (*)(void))snes->monitor[i],snes->monitorcontext[i],snes->monitordestroy[i],&identical);CHKERRQ(ierr); 359478064530SBarry Smith if (identical) PetscFunctionReturn(0); 3595649052a6SBarry Smith } 359678064530SBarry Smith if (snes->numbermonitors >= MAXSNESMONITORS) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Too many monitors set"); 35976e4dcb14SBarry Smith snes->monitor[snes->numbermonitors] = f; 3598b8a78c4aSBarry Smith snes->monitordestroy[snes->numbermonitors] = monitordestroy; 3599639f9d9dSBarry Smith snes->monitorcontext[snes->numbermonitors++] = (void*)mctx; 36003a40ed3dSBarry Smith PetscFunctionReturn(0); 36019b94acceSBarry Smith } 36029b94acceSBarry Smith 3603a278d85bSSatish Balay /*@ 3604a6570f20SBarry Smith SNESMonitorCancel - Clears all the monitor functions for a SNES object. 36055cd90555SBarry Smith 36063f9fe445SBarry Smith Logically Collective on SNES 3607c7afd0dbSLois Curfman McInnes 36085cd90555SBarry Smith Input Parameters: 36095cd90555SBarry Smith . snes - the SNES context 36105cd90555SBarry Smith 36111a480d89SAdministrator Options Database Key: 3612a6570f20SBarry Smith . -snes_monitor_cancel - cancels all monitors that have been hardwired 3613a6570f20SBarry Smith into a code by calls to SNESMonitorSet(), but does not cancel those 3614c7afd0dbSLois Curfman McInnes set via the options database 36155cd90555SBarry Smith 36165cd90555SBarry Smith Notes: 36175cd90555SBarry Smith There is no way to clear one specific monitor from a SNES object. 36185cd90555SBarry Smith 361936851e7fSLois Curfman McInnes Level: intermediate 362036851e7fSLois Curfman McInnes 36215cd90555SBarry Smith .keywords: SNES, nonlinear, set, monitor 36225cd90555SBarry Smith 3623a6570f20SBarry Smith .seealso: SNESMonitorDefault(), SNESMonitorSet() 36245cd90555SBarry Smith @*/ 36257087cfbeSBarry Smith PetscErrorCode SNESMonitorCancel(SNES snes) 36265cd90555SBarry Smith { 3627d952e501SBarry Smith PetscErrorCode ierr; 3628d952e501SBarry Smith PetscInt i; 3629d952e501SBarry Smith 36305cd90555SBarry Smith PetscFunctionBegin; 36310700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 3632d952e501SBarry Smith for (i=0; i<snes->numbermonitors; i++) { 3633d952e501SBarry Smith if (snes->monitordestroy[i]) { 36343c4aec1bSBarry Smith ierr = (*snes->monitordestroy[i])(&snes->monitorcontext[i]);CHKERRQ(ierr); 3635d952e501SBarry Smith } 3636d952e501SBarry Smith } 36375cd90555SBarry Smith snes->numbermonitors = 0; 36385cd90555SBarry Smith PetscFunctionReturn(0); 36395cd90555SBarry Smith } 36405cd90555SBarry Smith 3641bf388a1fSBarry Smith /*MC 3642bf388a1fSBarry Smith SNESConvergenceTestFunction - functional form used for testing of convergence of nonlinear solver 3643bf388a1fSBarry Smith 3644bf388a1fSBarry Smith Synopsis: 3645aaa7dc30SBarry Smith #include <petscsnes.h> 3646bf388a1fSBarry Smith $ PetscErrorCode SNESConvergenceTest(SNES snes,PetscInt it,PetscReal xnorm,PetscReal gnorm,PetscReal f,SNESConvergedReason *reason,void *cctx) 3647bf388a1fSBarry Smith 3648bf388a1fSBarry Smith + snes - the SNES context 3649bf388a1fSBarry Smith . it - current iteration (0 is the first and is before any Newton step) 3650bf388a1fSBarry Smith . cctx - [optional] convergence context 3651bf388a1fSBarry Smith . reason - reason for convergence/divergence 3652bf388a1fSBarry Smith . xnorm - 2-norm of current iterate 3653bf388a1fSBarry Smith . gnorm - 2-norm of current step 3654bf388a1fSBarry Smith - f - 2-norm of function 3655bf388a1fSBarry Smith 3656878cb397SSatish Balay Level: intermediate 3657bf388a1fSBarry Smith 3658bf388a1fSBarry Smith .seealso: SNESSetConvergenceTest(), SNESGetConvergenceTest() 3659bf388a1fSBarry Smith M*/ 3660bf388a1fSBarry Smith 36619b94acceSBarry Smith /*@C 36629b94acceSBarry Smith SNESSetConvergenceTest - Sets the function that is to be used 36639b94acceSBarry Smith to test for convergence of the nonlinear iterative solution. 36649b94acceSBarry Smith 36653f9fe445SBarry Smith Logically Collective on SNES 3666fee21e36SBarry Smith 3667c7afd0dbSLois Curfman McInnes Input Parameters: 3668c7afd0dbSLois Curfman McInnes + snes - the SNES context 3669bf388a1fSBarry Smith . SNESConvergenceTestFunction - routine to test for convergence 36700298fd71SBarry Smith . cctx - [optional] context for private data for the convergence routine (may be NULL) 36710298fd71SBarry Smith - destroy - [optional] destructor for the context (may be NULL; NULL_FUNCTION in Fortran) 36729b94acceSBarry Smith 367336851e7fSLois Curfman McInnes Level: advanced 367436851e7fSLois Curfman McInnes 36759b94acceSBarry Smith .keywords: SNES, nonlinear, set, convergence, test 36769b94acceSBarry Smith 3677e2a6519dSDmitry Karpeev .seealso: SNESConvergedDefault(), SNESConvergedSkip(), SNESConvergenceTestFunction 36789b94acceSBarry Smith @*/ 3679bf388a1fSBarry Smith PetscErrorCode SNESSetConvergenceTest(SNES snes,PetscErrorCode (*SNESConvergenceTestFunction)(SNES,PetscInt,PetscReal,PetscReal,PetscReal,SNESConvergedReason*,void*),void *cctx,PetscErrorCode (*destroy)(void*)) 36809b94acceSBarry Smith { 36817f7931b9SBarry Smith PetscErrorCode ierr; 36827f7931b9SBarry Smith 36833a40ed3dSBarry Smith PetscFunctionBegin; 36840700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 3685e2a6519dSDmitry Karpeev if (!SNESConvergenceTestFunction) SNESConvergenceTestFunction = SNESConvergedSkip; 36867f7931b9SBarry Smith if (snes->ops->convergeddestroy) { 36877f7931b9SBarry Smith ierr = (*snes->ops->convergeddestroy)(snes->cnvP);CHKERRQ(ierr); 36887f7931b9SBarry Smith } 3689bf388a1fSBarry Smith snes->ops->converged = SNESConvergenceTestFunction; 36907f7931b9SBarry Smith snes->ops->convergeddestroy = destroy; 369185385478SLisandro Dalcin snes->cnvP = cctx; 36923a40ed3dSBarry Smith PetscFunctionReturn(0); 36939b94acceSBarry Smith } 36949b94acceSBarry Smith 369552baeb72SSatish Balay /*@ 3696184914b5SBarry Smith SNESGetConvergedReason - Gets the reason the SNES iteration was stopped. 3697184914b5SBarry Smith 3698184914b5SBarry Smith Not Collective 3699184914b5SBarry Smith 3700184914b5SBarry Smith Input Parameter: 3701184914b5SBarry Smith . snes - the SNES context 3702184914b5SBarry Smith 3703184914b5SBarry Smith Output Parameter: 37044d0a8057SBarry Smith . reason - negative value indicates diverged, positive value converged, see SNESConvergedReason or the 3705184914b5SBarry Smith manual pages for the individual convergence tests for complete lists 3706184914b5SBarry Smith 37076a4d7782SBarry Smith Options Database: 37086a4d7782SBarry Smith . -snes_converged_reason - prints the reason to standard out 37096a4d7782SBarry Smith 3710184914b5SBarry Smith Level: intermediate 3711184914b5SBarry Smith 37126a4d7782SBarry Smith Notes: Should only be called after the call the SNESSolve() is complete, if it is called earlier it returns the value SNES__CONVERGED_ITERATING. 3713184914b5SBarry Smith 3714184914b5SBarry Smith .keywords: SNES, nonlinear, set, convergence, test 3715184914b5SBarry Smith 371633866048SMatthew G. Knepley .seealso: SNESSetConvergenceTest(), SNESSetConvergedReason(), SNESConvergedReason 3717184914b5SBarry Smith @*/ 37187087cfbeSBarry Smith PetscErrorCode SNESGetConvergedReason(SNES snes,SNESConvergedReason *reason) 3719184914b5SBarry Smith { 3720184914b5SBarry Smith PetscFunctionBegin; 37210700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 37224482741eSBarry Smith PetscValidPointer(reason,2); 3723184914b5SBarry Smith *reason = snes->reason; 3724184914b5SBarry Smith PetscFunctionReturn(0); 3725184914b5SBarry Smith } 3726184914b5SBarry Smith 372733866048SMatthew G. Knepley /*@ 372833866048SMatthew G. Knepley SNESSetConvergedReason - Sets the reason the SNES iteration was stopped. 372933866048SMatthew G. Knepley 373033866048SMatthew G. Knepley Not Collective 373133866048SMatthew G. Knepley 373233866048SMatthew G. Knepley Input Parameters: 373333866048SMatthew G. Knepley + snes - the SNES context 373433866048SMatthew G. Knepley - reason - negative value indicates diverged, positive value converged, see SNESConvergedReason or the 373533866048SMatthew G. Knepley manual pages for the individual convergence tests for complete lists 373633866048SMatthew G. Knepley 373733866048SMatthew G. Knepley Level: intermediate 373833866048SMatthew G. Knepley 373933866048SMatthew G. Knepley .keywords: SNES, nonlinear, set, convergence, test 374033866048SMatthew G. Knepley .seealso: SNESGetConvergedReason(), SNESSetConvergenceTest(), SNESConvergedReason 374133866048SMatthew G. Knepley @*/ 374233866048SMatthew G. Knepley PetscErrorCode SNESSetConvergedReason(SNES snes,SNESConvergedReason reason) 374333866048SMatthew G. Knepley { 374433866048SMatthew G. Knepley PetscFunctionBegin; 374533866048SMatthew G. Knepley PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 374633866048SMatthew G. Knepley snes->reason = reason; 374733866048SMatthew G. Knepley PetscFunctionReturn(0); 374833866048SMatthew G. Knepley } 374933866048SMatthew G. Knepley 3750c9005455SLois Curfman McInnes /*@ 3751c9005455SLois Curfman McInnes SNESSetConvergenceHistory - Sets the array used to hold the convergence history. 3752c9005455SLois Curfman McInnes 37533f9fe445SBarry Smith Logically Collective on SNES 3754fee21e36SBarry Smith 3755c7afd0dbSLois Curfman McInnes Input Parameters: 3756c7afd0dbSLois Curfman McInnes + snes - iterative context obtained from SNESCreate() 37578c7482ecSBarry Smith . a - array to hold history, this array will contain the function norms computed at each step 3758cd5578b5SBarry Smith . its - integer array holds the number of linear iterations for each solve. 3759758f92a0SBarry Smith . na - size of a and its 376064731454SLois Curfman McInnes - reset - PETSC_TRUE indicates each new nonlinear solve resets the history counter to zero, 3761758f92a0SBarry Smith else it continues storing new values for new nonlinear solves after the old ones 3762c7afd0dbSLois Curfman McInnes 3763308dcc3eSBarry Smith Notes: 37640298fd71SBarry Smith If 'a' and 'its' are NULL then space is allocated for the history. If 'na' PETSC_DECIDE or PETSC_DEFAULT then a 3765308dcc3eSBarry Smith default array of length 10000 is allocated. 3766308dcc3eSBarry Smith 3767c9005455SLois Curfman McInnes This routine is useful, e.g., when running a code for purposes 3768c9005455SLois Curfman McInnes of accurate performance monitoring, when no I/O should be done 3769c9005455SLois Curfman McInnes during the section of code that is being timed. 3770c9005455SLois Curfman McInnes 377136851e7fSLois Curfman McInnes Level: intermediate 377236851e7fSLois Curfman McInnes 3773c9005455SLois Curfman McInnes .keywords: SNES, set, convergence, history 3774758f92a0SBarry Smith 377508405cd6SLois Curfman McInnes .seealso: SNESGetConvergenceHistory() 3776758f92a0SBarry Smith 3777c9005455SLois Curfman McInnes @*/ 37787087cfbeSBarry Smith PetscErrorCode SNESSetConvergenceHistory(SNES snes,PetscReal a[],PetscInt its[],PetscInt na,PetscBool reset) 3779c9005455SLois Curfman McInnes { 3780308dcc3eSBarry Smith PetscErrorCode ierr; 3781308dcc3eSBarry Smith 37823a40ed3dSBarry Smith PetscFunctionBegin; 37830700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 37847a1ec6d4SBarry Smith if (a) PetscValidScalarPointer(a,2); 3785a562a398SLisandro Dalcin if (its) PetscValidIntPointer(its,3); 37867a1ec6d4SBarry Smith if (!a) { 3787308dcc3eSBarry Smith if (na == PETSC_DECIDE || na == PETSC_DEFAULT) na = 1000; 37887fdeb8b9SBarry Smith ierr = PetscCalloc1(na,&a);CHKERRQ(ierr); 37897fdeb8b9SBarry Smith ierr = PetscCalloc1(na,&its);CHKERRQ(ierr); 3790f5af7f23SKarl Rupp 3791308dcc3eSBarry Smith snes->conv_malloc = PETSC_TRUE; 3792308dcc3eSBarry Smith } 3793c9005455SLois Curfman McInnes snes->conv_hist = a; 3794758f92a0SBarry Smith snes->conv_hist_its = its; 3795758f92a0SBarry Smith snes->conv_hist_max = na; 3796a12bc48fSLisandro Dalcin snes->conv_hist_len = 0; 3797758f92a0SBarry Smith snes->conv_hist_reset = reset; 3798758f92a0SBarry Smith PetscFunctionReturn(0); 3799758f92a0SBarry Smith } 3800758f92a0SBarry Smith 3801308dcc3eSBarry Smith #if defined(PETSC_HAVE_MATLAB_ENGINE) 3802c6db04a5SJed Brown #include <engine.h> /* MATLAB include file */ 3803c6db04a5SJed Brown #include <mex.h> /* MATLAB include file */ 380499e0435eSBarry Smith 38058cc058d9SJed Brown PETSC_EXTERN mxArray *SNESGetConvergenceHistoryMatlab(SNES snes) 3806308dcc3eSBarry Smith { 3807308dcc3eSBarry Smith mxArray *mat; 3808308dcc3eSBarry Smith PetscInt i; 3809308dcc3eSBarry Smith PetscReal *ar; 3810308dcc3eSBarry Smith 3811308dcc3eSBarry Smith PetscFunctionBegin; 3812308dcc3eSBarry Smith mat = mxCreateDoubleMatrix(snes->conv_hist_len,1,mxREAL); 3813308dcc3eSBarry Smith ar = (PetscReal*) mxGetData(mat); 3814f5af7f23SKarl Rupp for (i=0; i<snes->conv_hist_len; i++) ar[i] = snes->conv_hist[i]; 3815308dcc3eSBarry Smith PetscFunctionReturn(mat); 3816308dcc3eSBarry Smith } 3817308dcc3eSBarry Smith #endif 3818308dcc3eSBarry Smith 38190c4c9dddSBarry Smith /*@C 3820758f92a0SBarry Smith SNESGetConvergenceHistory - Gets the array used to hold the convergence history. 3821758f92a0SBarry Smith 38223f9fe445SBarry Smith Not Collective 3823758f92a0SBarry Smith 3824758f92a0SBarry Smith Input Parameter: 3825758f92a0SBarry Smith . snes - iterative context obtained from SNESCreate() 3826758f92a0SBarry Smith 3827758f92a0SBarry Smith Output Parameters: 3828758f92a0SBarry Smith . a - array to hold history 3829758f92a0SBarry Smith . its - integer array holds the number of linear iterations (or 3830758f92a0SBarry Smith negative if not converged) for each solve. 3831758f92a0SBarry Smith - na - size of a and its 3832758f92a0SBarry Smith 3833758f92a0SBarry Smith Notes: 3834758f92a0SBarry Smith The calling sequence for this routine in Fortran is 3835758f92a0SBarry Smith $ call SNESGetConvergenceHistory(SNES snes, integer na, integer ierr) 3836758f92a0SBarry Smith 3837758f92a0SBarry Smith This routine is useful, e.g., when running a code for purposes 3838758f92a0SBarry Smith of accurate performance monitoring, when no I/O should be done 3839758f92a0SBarry Smith during the section of code that is being timed. 3840758f92a0SBarry Smith 3841758f92a0SBarry Smith Level: intermediate 3842758f92a0SBarry Smith 3843758f92a0SBarry Smith .keywords: SNES, get, convergence, history 3844758f92a0SBarry Smith 3845758f92a0SBarry Smith .seealso: SNESSetConvergencHistory() 3846758f92a0SBarry Smith 3847758f92a0SBarry Smith @*/ 38487087cfbeSBarry Smith PetscErrorCode SNESGetConvergenceHistory(SNES snes,PetscReal *a[],PetscInt *its[],PetscInt *na) 3849758f92a0SBarry Smith { 3850758f92a0SBarry Smith PetscFunctionBegin; 38510700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 3852758f92a0SBarry Smith if (a) *a = snes->conv_hist; 3853758f92a0SBarry Smith if (its) *its = snes->conv_hist_its; 3854758f92a0SBarry Smith if (na) *na = snes->conv_hist_len; 38553a40ed3dSBarry Smith PetscFunctionReturn(0); 3856c9005455SLois Curfman McInnes } 3857c9005455SLois Curfman McInnes 3858ac226902SBarry Smith /*@C 385976b2cf59SMatthew Knepley SNESSetUpdate - Sets the general-purpose update function called 3860eec8f646SJed Brown at the beginning of every iteration of the nonlinear solve. Specifically 38617e4bb74cSBarry Smith it is called just before the Jacobian is "evaluated". 386276b2cf59SMatthew Knepley 38633f9fe445SBarry Smith Logically Collective on SNES 386476b2cf59SMatthew Knepley 386576b2cf59SMatthew Knepley Input Parameters: 386676b2cf59SMatthew Knepley . snes - The nonlinear solver context 386776b2cf59SMatthew Knepley . func - The function 386876b2cf59SMatthew Knepley 386976b2cf59SMatthew Knepley Calling sequence of func: 3870b5d30489SBarry Smith . func (SNES snes, PetscInt step); 387176b2cf59SMatthew Knepley 387276b2cf59SMatthew Knepley . step - The current step of the iteration 387376b2cf59SMatthew Knepley 3874fe97e370SBarry Smith Level: advanced 3875fe97e370SBarry Smith 3876fe97e370SBarry Smith Note: This is NOT what one uses to update the ghost points before a function evaluation, that should be done at the beginning of your FormFunction() 3877fe97e370SBarry Smith This is not used by most users. 387876b2cf59SMatthew Knepley 387976b2cf59SMatthew Knepley .keywords: SNES, update 3880b5d30489SBarry Smith 38818d359177SBarry Smith .seealso SNESSetJacobian(), SNESSolve() 388276b2cf59SMatthew Knepley @*/ 38837087cfbeSBarry Smith PetscErrorCode SNESSetUpdate(SNES snes, PetscErrorCode (*func)(SNES, PetscInt)) 388476b2cf59SMatthew Knepley { 388576b2cf59SMatthew Knepley PetscFunctionBegin; 38860700a824SBarry Smith PetscValidHeaderSpecific(snes, SNES_CLASSID,1); 3887e7788613SBarry Smith snes->ops->update = func; 388876b2cf59SMatthew Knepley PetscFunctionReturn(0); 388976b2cf59SMatthew Knepley } 389076b2cf59SMatthew Knepley 38919b94acceSBarry Smith /* 38929b94acceSBarry Smith SNESScaleStep_Private - Scales a step so that its length is less than the 38939b94acceSBarry Smith positive parameter delta. 38949b94acceSBarry Smith 38959b94acceSBarry Smith Input Parameters: 3896c7afd0dbSLois Curfman McInnes + snes - the SNES context 38979b94acceSBarry Smith . y - approximate solution of linear system 38989b94acceSBarry Smith . fnorm - 2-norm of current function 3899c7afd0dbSLois Curfman McInnes - delta - trust region size 39009b94acceSBarry Smith 39019b94acceSBarry Smith Output Parameters: 3902c7afd0dbSLois Curfman McInnes + gpnorm - predicted function norm at the new point, assuming local 39039b94acceSBarry Smith linearization. The value is zero if the step lies within the trust 39049b94acceSBarry Smith region, and exceeds zero otherwise. 3905c7afd0dbSLois Curfman McInnes - ynorm - 2-norm of the step 39069b94acceSBarry Smith 39079b94acceSBarry Smith Note: 390804d7464bSBarry Smith For non-trust region methods such as SNESNEWTONLS, the parameter delta 39099b94acceSBarry Smith is set to be the maximum allowable step size. 39109b94acceSBarry Smith 39119b94acceSBarry Smith .keywords: SNES, nonlinear, scale, step 39129b94acceSBarry Smith */ 3913dfbe8321SBarry Smith PetscErrorCode SNESScaleStep_Private(SNES snes,Vec y,PetscReal *fnorm,PetscReal *delta,PetscReal *gpnorm,PetscReal *ynorm) 39149b94acceSBarry Smith { 3915064f8208SBarry Smith PetscReal nrm; 3916ea709b57SSatish Balay PetscScalar cnorm; 3917dfbe8321SBarry Smith PetscErrorCode ierr; 39183a40ed3dSBarry Smith 39193a40ed3dSBarry Smith PetscFunctionBegin; 39200700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 39210700a824SBarry Smith PetscValidHeaderSpecific(y,VEC_CLASSID,2); 3922c9780b6fSBarry Smith PetscCheckSameComm(snes,1,y,2); 3923184914b5SBarry Smith 3924064f8208SBarry Smith ierr = VecNorm(y,NORM_2,&nrm);CHKERRQ(ierr); 3925064f8208SBarry Smith if (nrm > *delta) { 3926064f8208SBarry Smith nrm = *delta/nrm; 3927064f8208SBarry Smith *gpnorm = (1.0 - nrm)*(*fnorm); 3928064f8208SBarry Smith cnorm = nrm; 39292dcb1b2aSMatthew Knepley ierr = VecScale(y,cnorm);CHKERRQ(ierr); 39309b94acceSBarry Smith *ynorm = *delta; 39319b94acceSBarry Smith } else { 39329b94acceSBarry Smith *gpnorm = 0.0; 3933064f8208SBarry Smith *ynorm = nrm; 39349b94acceSBarry Smith } 39353a40ed3dSBarry Smith PetscFunctionReturn(0); 39369b94acceSBarry Smith } 39379b94acceSBarry Smith 39382a359c20SBarry Smith /*@ 39392a359c20SBarry Smith SNESReasonView - Displays the reason a SNES solve converged or diverged to a viewer 39402a359c20SBarry Smith 39412a359c20SBarry Smith Collective on SNES 39422a359c20SBarry Smith 39432a359c20SBarry Smith Parameter: 39442a359c20SBarry Smith + snes - iterative context obtained from SNESCreate() 39452a359c20SBarry Smith - viewer - the viewer to display the reason 39462a359c20SBarry Smith 39472a359c20SBarry Smith 39482a359c20SBarry Smith Options Database Keys: 39492a359c20SBarry Smith . -snes_converged_reason - print reason for converged or diverged, also prints number of iterations 39502a359c20SBarry Smith 39512a359c20SBarry Smith Level: beginner 39522a359c20SBarry Smith 39532a359c20SBarry Smith .keywords: SNES, solve, linear system 39542a359c20SBarry Smith 39552a359c20SBarry Smith .seealso: SNESCreate(), SNESSetUp(), SNESDestroy(), SNESSetTolerances(), SNESConvergedDefault() 39562a359c20SBarry Smith 39572a359c20SBarry Smith @*/ 39582a359c20SBarry Smith PetscErrorCode SNESReasonView(SNES snes,PetscViewer viewer) 39592a359c20SBarry Smith { 396075cca76cSMatthew G. Knepley PetscViewerFormat format; 39612a359c20SBarry Smith PetscBool isAscii; 396275cca76cSMatthew G. Knepley PetscErrorCode ierr; 39632a359c20SBarry Smith 39642a359c20SBarry Smith PetscFunctionBegin; 39652a359c20SBarry Smith ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERASCII,&isAscii);CHKERRQ(ierr); 39662a359c20SBarry Smith if (isAscii) { 396775cca76cSMatthew G. Knepley ierr = PetscViewerGetFormat(viewer, &format);CHKERRQ(ierr); 39682a359c20SBarry Smith ierr = PetscViewerASCIIAddTab(viewer,((PetscObject)snes)->tablevel);CHKERRQ(ierr); 396975cca76cSMatthew G. Knepley if (format == PETSC_VIEWER_ASCII_INFO_DETAIL) { 397075cca76cSMatthew G. Knepley DM dm; 397175cca76cSMatthew G. Knepley Vec u; 397275cca76cSMatthew G. Knepley PetscDS prob; 397375cca76cSMatthew G. Knepley PetscInt Nf, f; 397475cca76cSMatthew G. Knepley PetscErrorCode (**exactFuncs)(PetscInt, PetscReal, const PetscReal[], PetscInt, PetscScalar[], void *); 397575cca76cSMatthew G. Knepley PetscReal error; 397675cca76cSMatthew G. Knepley 397775cca76cSMatthew G. Knepley ierr = SNESGetDM(snes, &dm);CHKERRQ(ierr); 397875cca76cSMatthew G. Knepley ierr = SNESGetSolution(snes, &u);CHKERRQ(ierr); 397975cca76cSMatthew G. Knepley ierr = DMGetDS(dm, &prob);CHKERRQ(ierr); 398075cca76cSMatthew G. Knepley ierr = PetscDSGetNumFields(prob, &Nf);CHKERRQ(ierr); 398175cca76cSMatthew G. Knepley ierr = PetscMalloc1(Nf, &exactFuncs);CHKERRQ(ierr); 398275cca76cSMatthew G. Knepley for (f = 0; f < Nf; ++f) {ierr = PetscDSGetExactSolution(prob, f, &exactFuncs[f]);CHKERRQ(ierr);} 398375cca76cSMatthew G. Knepley ierr = DMComputeL2Diff(dm, 0.0, exactFuncs, NULL, u, &error);CHKERRQ(ierr); 398475cca76cSMatthew G. Knepley ierr = PetscFree(exactFuncs);CHKERRQ(ierr); 398575cca76cSMatthew G. Knepley if (error < 1.0e-11) {ierr = PetscViewerASCIIPrintf(viewer, "L_2 Error: < 1.0e-11\n");CHKERRQ(ierr);} 398675cca76cSMatthew G. Knepley else {ierr = PetscViewerASCIIPrintf(viewer, "L_2 Error: %g\n", error);CHKERRQ(ierr);} 398775cca76cSMatthew G. Knepley } 39882a359c20SBarry Smith if (snes->reason > 0) { 39892a359c20SBarry Smith if (((PetscObject) snes)->prefix) { 39902a359c20SBarry Smith ierr = PetscViewerASCIIPrintf(viewer,"Nonlinear %s solve converged due to %s iterations %D\n",((PetscObject) snes)->prefix,SNESConvergedReasons[snes->reason],snes->iter);CHKERRQ(ierr); 39912a359c20SBarry Smith } else { 39922a359c20SBarry Smith ierr = PetscViewerASCIIPrintf(viewer,"Nonlinear solve converged due to %s iterations %D\n",SNESConvergedReasons[snes->reason],snes->iter);CHKERRQ(ierr); 39932a359c20SBarry Smith } 39942a359c20SBarry Smith } else { 39952a359c20SBarry Smith if (((PetscObject) snes)->prefix) { 39962a359c20SBarry Smith ierr = PetscViewerASCIIPrintf(viewer,"Nonlinear %s solve did not converge due to %s iterations %D\n",((PetscObject) snes)->prefix,SNESConvergedReasons[snes->reason],snes->iter);CHKERRQ(ierr); 39972a359c20SBarry Smith } else { 39982a359c20SBarry Smith ierr = PetscViewerASCIIPrintf(viewer,"Nonlinear solve did not converge due to %s iterations %D\n",SNESConvergedReasons[snes->reason],snes->iter);CHKERRQ(ierr); 39992a359c20SBarry Smith } 40002a359c20SBarry Smith } 40012a359c20SBarry Smith ierr = PetscViewerASCIISubtractTab(viewer,((PetscObject)snes)->tablevel);CHKERRQ(ierr); 40022a359c20SBarry Smith } 40032a359c20SBarry Smith PetscFunctionReturn(0); 40042a359c20SBarry Smith } 40052a359c20SBarry Smith 40062a359c20SBarry Smith /*@C 40072a359c20SBarry Smith SNESReasonViewFromOptions - Processes command line options to determine if/how a SNESReason is to be viewed. 40082a359c20SBarry Smith 40092a359c20SBarry Smith Collective on SNES 40102a359c20SBarry Smith 40112a359c20SBarry Smith Input Parameters: 40122a359c20SBarry Smith . snes - the SNES object 40132a359c20SBarry Smith 40142a359c20SBarry Smith Level: intermediate 40152a359c20SBarry Smith 40162a359c20SBarry Smith @*/ 40172a359c20SBarry Smith PetscErrorCode SNESReasonViewFromOptions(SNES snes) 40182a359c20SBarry Smith { 40192a359c20SBarry Smith PetscErrorCode ierr; 40202a359c20SBarry Smith PetscViewer viewer; 40212a359c20SBarry Smith PetscBool flg; 40222a359c20SBarry Smith static PetscBool incall = PETSC_FALSE; 40232a359c20SBarry Smith PetscViewerFormat format; 40242a359c20SBarry Smith 40252a359c20SBarry Smith PetscFunctionBegin; 40262a359c20SBarry Smith if (incall) PetscFunctionReturn(0); 40272a359c20SBarry Smith incall = PETSC_TRUE; 40282a359c20SBarry Smith ierr = PetscOptionsGetViewer(PetscObjectComm((PetscObject)snes),((PetscObject)snes)->prefix,"-snes_converged_reason",&viewer,&format,&flg);CHKERRQ(ierr); 40292a359c20SBarry Smith if (flg) { 40302a359c20SBarry Smith ierr = PetscViewerPushFormat(viewer,format);CHKERRQ(ierr); 40312a359c20SBarry Smith ierr = SNESReasonView(snes,viewer);CHKERRQ(ierr); 40322a359c20SBarry Smith ierr = PetscViewerPopFormat(viewer);CHKERRQ(ierr); 40332a359c20SBarry Smith ierr = PetscViewerDestroy(&viewer);CHKERRQ(ierr); 40342a359c20SBarry Smith } 40352a359c20SBarry Smith incall = PETSC_FALSE; 40362a359c20SBarry Smith PetscFunctionReturn(0); 40372a359c20SBarry Smith } 40382a359c20SBarry Smith 40396ce558aeSBarry Smith /*@C 4040f69a0ea3SMatthew Knepley SNESSolve - Solves a nonlinear system F(x) = b. 4041f69a0ea3SMatthew Knepley Call SNESSolve() after calling SNESCreate() and optional routines of the form SNESSetXXX(). 40429b94acceSBarry Smith 4043c7afd0dbSLois Curfman McInnes Collective on SNES 4044c7afd0dbSLois Curfman McInnes 4045b2002411SLois Curfman McInnes Input Parameters: 4046c7afd0dbSLois Curfman McInnes + snes - the SNES context 40470298fd71SBarry Smith . b - the constant part of the equation F(x) = b, or NULL to use zero. 404885385478SLisandro Dalcin - x - the solution vector. 40499b94acceSBarry Smith 4050b2002411SLois Curfman McInnes Notes: 40518ddd3da0SLois Curfman McInnes The user should initialize the vector,x, with the initial guess 40528ddd3da0SLois Curfman McInnes for the nonlinear solve prior to calling SNESSolve. In particular, 40538ddd3da0SLois Curfman McInnes to employ an initial guess of zero, the user should explicitly set 40548ddd3da0SLois Curfman McInnes this vector to zero by calling VecSet(). 40558ddd3da0SLois Curfman McInnes 405636851e7fSLois Curfman McInnes Level: beginner 405736851e7fSLois Curfman McInnes 40589b94acceSBarry Smith .keywords: SNES, nonlinear, solve 40599b94acceSBarry Smith 4060c0df2a02SJed Brown .seealso: SNESCreate(), SNESDestroy(), SNESSetFunction(), SNESSetJacobian(), SNESSetGridSequence(), SNESGetSolution() 40619b94acceSBarry Smith @*/ 40627087cfbeSBarry Smith PetscErrorCode SNESSolve(SNES snes,Vec b,Vec x) 40639b94acceSBarry Smith { 4064dfbe8321SBarry Smith PetscErrorCode ierr; 4065ace3abfcSBarry Smith PetscBool flg; 4066efd51863SBarry Smith PetscInt grid; 40670298fd71SBarry Smith Vec xcreated = NULL; 4068caa4e7f2SJed Brown DM dm; 4069052efed2SBarry Smith 40703a40ed3dSBarry Smith PetscFunctionBegin; 40710700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 4072a69afd8bSBarry Smith if (x) PetscValidHeaderSpecific(x,VEC_CLASSID,3); 4073a69afd8bSBarry Smith if (x) PetscCheckSameComm(snes,1,x,3); 40740700a824SBarry Smith if (b) PetscValidHeaderSpecific(b,VEC_CLASSID,2); 407585385478SLisandro Dalcin if (b) PetscCheckSameComm(snes,1,b,2); 407685385478SLisandro Dalcin 407734b4d3a8SMatthew G. Knepley /* High level operations using the nonlinear solver */ 407806fc46c8SMatthew G. Knepley { 407906fc46c8SMatthew G. Knepley PetscViewer viewer; 408006fc46c8SMatthew G. Knepley PetscViewerFormat format; 40817c88af5aSMatthew G. Knepley PetscInt num; 408206fc46c8SMatthew G. Knepley PetscBool flg; 408306fc46c8SMatthew G. Knepley static PetscBool incall = PETSC_FALSE; 408406fc46c8SMatthew G. Knepley 408506fc46c8SMatthew G. Knepley if (!incall) { 408634b4d3a8SMatthew G. Knepley /* Estimate the convergence rate of the discretization */ 408706fc46c8SMatthew G. Knepley ierr = PetscOptionsGetViewer(PetscObjectComm((PetscObject) snes), ((PetscObject) snes)->prefix, "-snes_convergence_estimate", &viewer, &format, &flg);CHKERRQ(ierr); 408806fc46c8SMatthew G. Knepley if (flg) { 408906fc46c8SMatthew G. Knepley PetscConvEst conv; 409006fc46c8SMatthew G. Knepley PetscReal alpha; /* Convergence rate of the solution error in the L_2 norm */ 409106fc46c8SMatthew G. Knepley 409206fc46c8SMatthew G. Knepley incall = PETSC_TRUE; 409306fc46c8SMatthew G. Knepley ierr = PetscConvEstCreate(PetscObjectComm((PetscObject) snes), &conv);CHKERRQ(ierr); 409406fc46c8SMatthew G. Knepley ierr = PetscConvEstSetSolver(conv, snes);CHKERRQ(ierr); 409506fc46c8SMatthew G. Knepley ierr = PetscConvEstSetFromOptions(conv);CHKERRQ(ierr); 40960955ed61SMatthew G. Knepley ierr = PetscConvEstSetUp(conv);CHKERRQ(ierr); 409706fc46c8SMatthew G. Knepley ierr = PetscConvEstGetConvRate(conv, &alpha);CHKERRQ(ierr); 409806fc46c8SMatthew G. Knepley ierr = PetscViewerPushFormat(viewer, format);CHKERRQ(ierr); 409906fc46c8SMatthew G. Knepley ierr = PetscConvEstRateView(conv, alpha, viewer);CHKERRQ(ierr); 410006fc46c8SMatthew G. Knepley ierr = PetscViewerPopFormat(viewer);CHKERRQ(ierr); 410106fc46c8SMatthew G. Knepley ierr = PetscViewerDestroy(&viewer);CHKERRQ(ierr); 410206fc46c8SMatthew G. Knepley ierr = PetscConvEstDestroy(&conv);CHKERRQ(ierr); 410306fc46c8SMatthew G. Knepley incall = PETSC_FALSE; 410406fc46c8SMatthew G. Knepley } 410534b4d3a8SMatthew G. Knepley /* Adaptively refine the initial grid */ 4106*b2588ea6SMatthew G. Knepley num = 1; 4107*b2588ea6SMatthew G. Knepley ierr = PetscOptionsGetInt(NULL, ((PetscObject) snes)->prefix, "-snes_adapt_initial", &num, &flg);CHKERRQ(ierr); 410834b4d3a8SMatthew G. Knepley if (flg) { 410934b4d3a8SMatthew G. Knepley DMAdaptor adaptor; 411034b4d3a8SMatthew G. Knepley 411134b4d3a8SMatthew G. Knepley incall = PETSC_TRUE; 411234b4d3a8SMatthew G. Knepley ierr = DMAdaptorCreate(PETSC_COMM_WORLD, &adaptor);CHKERRQ(ierr); 411334b4d3a8SMatthew G. Knepley ierr = DMAdaptorSetSolver(adaptor, snes);CHKERRQ(ierr); 4114*b2588ea6SMatthew G. Knepley ierr = DMAdaptorSetSequenceLength(adaptor, num);CHKERRQ(ierr); 411534b4d3a8SMatthew G. Knepley ierr = DMAdaptorSetFromOptions(adaptor);CHKERRQ(ierr); 411634b4d3a8SMatthew G. Knepley ierr = DMAdaptorSetUp(adaptor);CHKERRQ(ierr); 411734b4d3a8SMatthew G. Knepley ierr = DMAdaptorAdapt(adaptor, x, DM_ADAPTATION_INITIAL, &dm, &x);CHKERRQ(ierr); 411834b4d3a8SMatthew G. Knepley ierr = DMAdaptorDestroy(&adaptor);CHKERRQ(ierr); 411934b4d3a8SMatthew G. Knepley incall = PETSC_FALSE; 412034b4d3a8SMatthew G. Knepley } 41217c88af5aSMatthew G. Knepley /* Use grid sequencing to adapt */ 41227c88af5aSMatthew G. Knepley num = 0; 41237c88af5aSMatthew G. Knepley ierr = PetscOptionsGetInt(NULL, ((PetscObject) snes)->prefix, "-snes_adapt_sequence", &num, NULL);CHKERRQ(ierr); 41247c88af5aSMatthew G. Knepley if (num) { 41257c88af5aSMatthew G. Knepley DMAdaptor adaptor; 41267c88af5aSMatthew G. Knepley 41277c88af5aSMatthew G. Knepley incall = PETSC_TRUE; 41287c88af5aSMatthew G. Knepley ierr = DMAdaptorCreate(PETSC_COMM_WORLD, &adaptor);CHKERRQ(ierr); 41297c88af5aSMatthew G. Knepley ierr = DMAdaptorSetSolver(adaptor, snes);CHKERRQ(ierr); 41307c88af5aSMatthew G. Knepley ierr = DMAdaptorSetSequenceLength(adaptor, num);CHKERRQ(ierr); 41317c88af5aSMatthew G. Knepley ierr = DMAdaptorSetFromOptions(adaptor);CHKERRQ(ierr); 41327c88af5aSMatthew G. Knepley ierr = DMAdaptorSetUp(adaptor);CHKERRQ(ierr); 41337c88af5aSMatthew G. Knepley ierr = DMAdaptorAdapt(adaptor, x, DM_ADAPTATION_SEQUENTIAL, &dm, &x);CHKERRQ(ierr); 41347c88af5aSMatthew G. Knepley ierr = DMAdaptorDestroy(&adaptor);CHKERRQ(ierr); 41357c88af5aSMatthew G. Knepley incall = PETSC_FALSE; 41367c88af5aSMatthew G. Knepley } 413706fc46c8SMatthew G. Knepley } 413806fc46c8SMatthew G. Knepley } 4139caa4e7f2SJed Brown if (!x) { 4140caa4e7f2SJed Brown ierr = SNESGetDM(snes,&dm);CHKERRQ(ierr); 4141caa4e7f2SJed Brown ierr = DMCreateGlobalVector(dm,&xcreated);CHKERRQ(ierr); 4142a69afd8bSBarry Smith x = xcreated; 4143a69afd8bSBarry Smith } 4144ce1779c8SBarry Smith ierr = SNESViewFromOptions(snes,NULL,"-snes_view_pre");CHKERRQ(ierr); 4145f05ece33SBarry Smith 4146ce94432eSBarry Smith for (grid=0; grid<snes->gridsequence; grid++) {ierr = PetscViewerASCIIPushTab(PETSC_VIEWER_STDOUT_(PetscObjectComm((PetscObject)snes)));CHKERRQ(ierr);} 4147efd51863SBarry Smith for (grid=0; grid<snes->gridsequence+1; grid++) { 4148efd51863SBarry Smith 414985385478SLisandro Dalcin /* set solution vector */ 4150efd51863SBarry Smith if (!grid) {ierr = PetscObjectReference((PetscObject)x);CHKERRQ(ierr);} 41516bf464f9SBarry Smith ierr = VecDestroy(&snes->vec_sol);CHKERRQ(ierr); 415285385478SLisandro Dalcin snes->vec_sol = x; 4153caa4e7f2SJed Brown ierr = SNESGetDM(snes,&dm);CHKERRQ(ierr); 4154caa4e7f2SJed Brown 4155caa4e7f2SJed Brown /* set affine vector if provided */ 415685385478SLisandro Dalcin if (b) { ierr = PetscObjectReference((PetscObject)b);CHKERRQ(ierr); } 41576bf464f9SBarry Smith ierr = VecDestroy(&snes->vec_rhs);CHKERRQ(ierr); 415885385478SLisandro Dalcin snes->vec_rhs = b; 415985385478SLisandro Dalcin 4160154060b5SMatthew G. Knepley if (snes->vec_func == snes->vec_sol) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_IDN,"Solution vector cannot be function vector"); 4161154060b5SMatthew G. Knepley if (snes->vec_rhs == snes->vec_sol) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_IDN,"Solution vector cannot be right hand side vector"); 4162154060b5SMatthew G. Knepley if (!snes->vec_sol_update /* && snes->vec_sol */) { 4163154060b5SMatthew G. Knepley ierr = VecDuplicate(snes->vec_sol,&snes->vec_sol_update);CHKERRQ(ierr); 4164154060b5SMatthew G. Knepley ierr = PetscLogObjectParent((PetscObject)snes,(PetscObject)snes->vec_sol_update);CHKERRQ(ierr); 4165154060b5SMatthew G. Knepley } 4166154060b5SMatthew G. Knepley ierr = DMShellSetGlobalVector(dm,snes->vec_sol);CHKERRQ(ierr); 416770e92668SMatthew Knepley ierr = SNESSetUp(snes);CHKERRQ(ierr); 41683f149594SLisandro Dalcin 41697eee914bSBarry Smith if (!grid) { 41707eee914bSBarry Smith if (snes->ops->computeinitialguess) { 4171d25893d9SBarry Smith ierr = (*snes->ops->computeinitialguess)(snes,snes->vec_sol,snes->initialguessP);CHKERRQ(ierr); 4172d25893d9SBarry Smith } 4173dd568438SSatish Balay } 4174d25893d9SBarry Smith 4175abc0a331SBarry Smith if (snes->conv_hist_reset) snes->conv_hist_len = 0; 4176971e163fSPeter Brune if (snes->counters_reset) {snes->nfuncs = 0; snes->linear_its = 0; snes->numFailures = 0;} 4177d5e45103SBarry Smith 41783f149594SLisandro Dalcin ierr = PetscLogEventBegin(SNES_Solve,snes,0,0,0);CHKERRQ(ierr); 41794936397dSBarry Smith ierr = (*snes->ops->solve)(snes);CHKERRQ(ierr); 418085385478SLisandro Dalcin ierr = PetscLogEventEnd(SNES_Solve,snes,0,0,0);CHKERRQ(ierr); 418117186662SBarry Smith if (!snes->reason) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_PLIB,"Internal error, solver returned without setting converged reason"); 4182422a814eSBarry Smith snes->domainerror = PETSC_FALSE; /* clear the flag if it has been set */ 41833f149594SLisandro Dalcin 418437ec4e1aSPeter Brune if (snes->lagjac_persist) snes->jac_iter += snes->iter; 418537ec4e1aSPeter Brune if (snes->lagpre_persist) snes->pre_iter += snes->iter; 418637ec4e1aSPeter Brune 418727b0f280SBarry Smith ierr = PetscOptionsGetViewer(PetscObjectComm((PetscObject)snes),((PetscObject)snes)->prefix,"-snes_test_local_min",NULL,NULL,&flg);CHKERRQ(ierr); 4188da9b6338SBarry Smith if (flg && !PetscPreLoadingOn) { ierr = SNESTestLocalMin(snes);CHKERRQ(ierr); } 41892a359c20SBarry Smith ierr = SNESReasonViewFromOptions(snes);CHKERRQ(ierr); 41905968eb51SBarry Smith 4191ce94432eSBarry Smith if (snes->errorifnotconverged && snes->reason < 0) SETERRQ(PetscObjectComm((PetscObject)snes),PETSC_ERR_NOT_CONVERGED,"SNESSolve has not converged"); 41929c8e83a9SBarry Smith if (snes->reason < 0) break; 4193efd51863SBarry Smith if (grid < snes->gridsequence) { 4194efd51863SBarry Smith DM fine; 4195efd51863SBarry Smith Vec xnew; 4196efd51863SBarry Smith Mat interp; 4197efd51863SBarry Smith 4198ce94432eSBarry Smith ierr = DMRefine(snes->dm,PetscObjectComm((PetscObject)snes),&fine);CHKERRQ(ierr); 4199ce94432eSBarry Smith if (!fine) SETERRQ(PetscObjectComm((PetscObject)snes),PETSC_ERR_ARG_INCOMP,"DMRefine() did not perform any refinement, cannot continue grid sequencing"); 42000298fd71SBarry Smith ierr = DMCreateInterpolation(snes->dm,fine,&interp,NULL);CHKERRQ(ierr); 4201efd51863SBarry Smith ierr = DMCreateGlobalVector(fine,&xnew);CHKERRQ(ierr); 4202efd51863SBarry Smith ierr = MatInterpolate(interp,x,xnew);CHKERRQ(ierr); 4203c833c3b5SJed Brown ierr = DMInterpolate(snes->dm,interp,fine);CHKERRQ(ierr); 4204efd51863SBarry Smith ierr = MatDestroy(&interp);CHKERRQ(ierr); 4205efd51863SBarry Smith x = xnew; 4206efd51863SBarry Smith 4207efd51863SBarry Smith ierr = SNESReset(snes);CHKERRQ(ierr); 4208efd51863SBarry Smith ierr = SNESSetDM(snes,fine);CHKERRQ(ierr); 4209efd51863SBarry Smith ierr = DMDestroy(&fine);CHKERRQ(ierr); 4210ce94432eSBarry Smith ierr = PetscViewerASCIIPopTab(PETSC_VIEWER_STDOUT_(PetscObjectComm((PetscObject)snes)));CHKERRQ(ierr); 4211efd51863SBarry Smith } 4212efd51863SBarry Smith } 4213ce1779c8SBarry Smith ierr = SNESViewFromOptions(snes,NULL,"-snes_view");CHKERRQ(ierr); 4214685405a1SBarry Smith ierr = VecViewFromOptions(snes->vec_sol,(PetscObject)snes,"-snes_view_solution");CHKERRQ(ierr); 42153f7e2da0SPeter Brune 4216a69afd8bSBarry Smith ierr = VecDestroy(&xcreated);CHKERRQ(ierr); 4217e04113cfSBarry Smith ierr = PetscObjectSAWsBlock((PetscObject)snes);CHKERRQ(ierr); 42183a40ed3dSBarry Smith PetscFunctionReturn(0); 42199b94acceSBarry Smith } 42209b94acceSBarry Smith 42219b94acceSBarry Smith /* --------- Internal routines for SNES Package --------- */ 42229b94acceSBarry Smith 422382bf6240SBarry Smith /*@C 42244b0e389bSBarry Smith SNESSetType - Sets the method for the nonlinear solver. 42259b94acceSBarry Smith 4226fee21e36SBarry Smith Collective on SNES 4227fee21e36SBarry Smith 4228c7afd0dbSLois Curfman McInnes Input Parameters: 4229c7afd0dbSLois Curfman McInnes + snes - the SNES context 4230454a90a3SBarry Smith - type - a known method 4231c7afd0dbSLois Curfman McInnes 4232c7afd0dbSLois Curfman McInnes Options Database Key: 4233454a90a3SBarry Smith . -snes_type <type> - Sets the method; use -help for a list 423404d7464bSBarry Smith of available methods (for instance, newtonls or newtontr) 4235ae12b187SLois Curfman McInnes 42369b94acceSBarry Smith Notes: 4237e090d566SSatish Balay See "petsc/include/petscsnes.h" for available methods (for instance) 423804d7464bSBarry Smith + SNESNEWTONLS - Newton's method with line search 4239c7afd0dbSLois Curfman McInnes (systems of nonlinear equations) 424004d7464bSBarry Smith . SNESNEWTONTR - Newton's method with trust region 4241c7afd0dbSLois Curfman McInnes (systems of nonlinear equations) 42429b94acceSBarry Smith 4243ae12b187SLois Curfman McInnes Normally, it is best to use the SNESSetFromOptions() command and then 4244ae12b187SLois Curfman McInnes set the SNES solver type from the options database rather than by using 4245ae12b187SLois Curfman McInnes this routine. Using the options database provides the user with 4246ae12b187SLois Curfman McInnes maximum flexibility in evaluating the many nonlinear solvers. 4247ae12b187SLois Curfman McInnes The SNESSetType() routine is provided for those situations where it 4248ae12b187SLois Curfman McInnes is necessary to set the nonlinear solver independently of the command 4249ae12b187SLois Curfman McInnes line or options database. This might be the case, for example, when 4250ae12b187SLois Curfman McInnes the choice of solver changes during the execution of the program, 4251ae12b187SLois Curfman McInnes and the user's application is taking responsibility for choosing the 4252b0a32e0cSBarry Smith appropriate method. 425336851e7fSLois Curfman McInnes 42548f6c3df8SBarry Smith Developer Notes: SNESRegister() adds a constructor for a new SNESType to SNESList, SNESSetType() locates 42558f6c3df8SBarry Smith the constructor in that list and calls it to create the spexific object. 42568f6c3df8SBarry Smith 425736851e7fSLois Curfman McInnes Level: intermediate 4258a703fe33SLois Curfman McInnes 4259454a90a3SBarry Smith .keywords: SNES, set, type 4260435da068SBarry Smith 42618f6c3df8SBarry Smith .seealso: SNESType, SNESCreate(), SNESDestroy(), SNESGetType(), SNESSetFromOptions() 4262435da068SBarry Smith 42639b94acceSBarry Smith @*/ 426419fd82e9SBarry Smith PetscErrorCode SNESSetType(SNES snes,SNESType type) 42659b94acceSBarry Smith { 4266dfbe8321SBarry Smith PetscErrorCode ierr,(*r)(SNES); 4267ace3abfcSBarry Smith PetscBool match; 42683a40ed3dSBarry Smith 42693a40ed3dSBarry Smith PetscFunctionBegin; 42700700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 42714482741eSBarry Smith PetscValidCharPointer(type,2); 427282bf6240SBarry Smith 4273251f4c67SDmitry Karpeev ierr = PetscObjectTypeCompare((PetscObject)snes,type,&match);CHKERRQ(ierr); 42740f5bd95cSBarry Smith if (match) PetscFunctionReturn(0); 427592ff6ae8SBarry Smith 42761c9cd337SJed Brown ierr = PetscFunctionListFind(SNESList,type,&r);CHKERRQ(ierr); 4277e32f2f54SBarry Smith if (!r) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_UNKNOWN_TYPE,"Unable to find requested SNES type %s",type); 427875396ef9SLisandro Dalcin /* Destroy the previous private SNES context */ 4279b5c23020SJed Brown if (snes->ops->destroy) { 4280b5c23020SJed Brown ierr = (*(snes)->ops->destroy)(snes);CHKERRQ(ierr); 42810298fd71SBarry Smith snes->ops->destroy = NULL; 4282b5c23020SJed Brown } 428375396ef9SLisandro Dalcin /* Reinitialize function pointers in SNESOps structure */ 428475396ef9SLisandro Dalcin snes->ops->setup = 0; 428575396ef9SLisandro Dalcin snes->ops->solve = 0; 428675396ef9SLisandro Dalcin snes->ops->view = 0; 428775396ef9SLisandro Dalcin snes->ops->setfromoptions = 0; 428875396ef9SLisandro Dalcin snes->ops->destroy = 0; 428975396ef9SLisandro Dalcin /* Call the SNESCreate_XXX routine for this particular Nonlinear solver */ 429075396ef9SLisandro Dalcin snes->setupcalled = PETSC_FALSE; 4291f5af7f23SKarl Rupp 4292454a90a3SBarry Smith ierr = PetscObjectChangeTypeName((PetscObject)snes,type);CHKERRQ(ierr); 429303bfa161SLisandro Dalcin ierr = (*r)(snes);CHKERRQ(ierr); 42943a40ed3dSBarry Smith PetscFunctionReturn(0); 42959b94acceSBarry Smith } 42969b94acceSBarry Smith 42979b94acceSBarry Smith /*@C 42989a28b0a6SLois Curfman McInnes SNESGetType - Gets the SNES method type and name (as a string). 42999b94acceSBarry Smith 4300c7afd0dbSLois Curfman McInnes Not Collective 4301c7afd0dbSLois Curfman McInnes 43029b94acceSBarry Smith Input Parameter: 43034b0e389bSBarry Smith . snes - nonlinear solver context 43049b94acceSBarry Smith 43059b94acceSBarry Smith Output Parameter: 43063a7fca6bSBarry Smith . type - SNES method (a character string) 43079b94acceSBarry Smith 430836851e7fSLois Curfman McInnes Level: intermediate 430936851e7fSLois Curfman McInnes 4310454a90a3SBarry Smith .keywords: SNES, nonlinear, get, type, name 43119b94acceSBarry Smith @*/ 431219fd82e9SBarry Smith PetscErrorCode SNESGetType(SNES snes,SNESType *type) 43139b94acceSBarry Smith { 43143a40ed3dSBarry Smith PetscFunctionBegin; 43150700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 43164482741eSBarry Smith PetscValidPointer(type,2); 43177adad957SLisandro Dalcin *type = ((PetscObject)snes)->type_name; 43183a40ed3dSBarry Smith PetscFunctionReturn(0); 43199b94acceSBarry Smith } 43209b94acceSBarry Smith 43213cd8a7caSMatthew G. Knepley /*@ 43223cd8a7caSMatthew G. Knepley SNESSetSolution - Sets the solution vector for use by the SNES routines. 43233cd8a7caSMatthew G. Knepley 43243cd8a7caSMatthew G. Knepley Logically Collective on SNES and Vec 43253cd8a7caSMatthew G. Knepley 43263cd8a7caSMatthew G. Knepley Input Parameters: 43273cd8a7caSMatthew G. Knepley + snes - the SNES context obtained from SNESCreate() 43283cd8a7caSMatthew G. Knepley - u - the solution vector 43293cd8a7caSMatthew G. Knepley 43303cd8a7caSMatthew G. Knepley Level: beginner 43313cd8a7caSMatthew G. Knepley 43323cd8a7caSMatthew G. Knepley .keywords: SNES, set, solution 43333cd8a7caSMatthew G. Knepley @*/ 43343cd8a7caSMatthew G. Knepley PetscErrorCode SNESSetSolution(SNES snes, Vec u) 43353cd8a7caSMatthew G. Knepley { 43363cd8a7caSMatthew G. Knepley DM dm; 43373cd8a7caSMatthew G. Knepley PetscErrorCode ierr; 43383cd8a7caSMatthew G. Knepley 43393cd8a7caSMatthew G. Knepley PetscFunctionBegin; 43403cd8a7caSMatthew G. Knepley PetscValidHeaderSpecific(snes, SNES_CLASSID, 1); 43413cd8a7caSMatthew G. Knepley PetscValidHeaderSpecific(u, VEC_CLASSID, 2); 43423cd8a7caSMatthew G. Knepley ierr = PetscObjectReference((PetscObject) u);CHKERRQ(ierr); 43433cd8a7caSMatthew G. Knepley ierr = VecDestroy(&snes->vec_sol);CHKERRQ(ierr); 43443cd8a7caSMatthew G. Knepley 43453cd8a7caSMatthew G. Knepley snes->vec_sol = u; 43463cd8a7caSMatthew G. Knepley 43473cd8a7caSMatthew G. Knepley ierr = SNESGetDM(snes, &dm);CHKERRQ(ierr); 43483cd8a7caSMatthew G. Knepley ierr = DMShellSetGlobalVector(dm, u);CHKERRQ(ierr); 43493cd8a7caSMatthew G. Knepley PetscFunctionReturn(0); 43503cd8a7caSMatthew G. Knepley } 43513cd8a7caSMatthew G. Knepley 435252baeb72SSatish Balay /*@ 43539b94acceSBarry Smith SNESGetSolution - Returns the vector where the approximate solution is 4354c0df2a02SJed Brown stored. This is the fine grid solution when using SNESSetGridSequence(). 43559b94acceSBarry Smith 4356c7afd0dbSLois Curfman McInnes Not Collective, but Vec is parallel if SNES is parallel 4357c7afd0dbSLois Curfman McInnes 43589b94acceSBarry Smith Input Parameter: 43599b94acceSBarry Smith . snes - the SNES context 43609b94acceSBarry Smith 43619b94acceSBarry Smith Output Parameter: 43629b94acceSBarry Smith . x - the solution 43639b94acceSBarry Smith 436470e92668SMatthew Knepley Level: intermediate 436536851e7fSLois Curfman McInnes 43669b94acceSBarry Smith .keywords: SNES, nonlinear, get, solution 43679b94acceSBarry Smith 436885385478SLisandro Dalcin .seealso: SNESGetSolutionUpdate(), SNESGetFunction() 43699b94acceSBarry Smith @*/ 43707087cfbeSBarry Smith PetscErrorCode SNESGetSolution(SNES snes,Vec *x) 43719b94acceSBarry Smith { 43723a40ed3dSBarry Smith PetscFunctionBegin; 43730700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 43744482741eSBarry Smith PetscValidPointer(x,2); 437585385478SLisandro Dalcin *x = snes->vec_sol; 437670e92668SMatthew Knepley PetscFunctionReturn(0); 437770e92668SMatthew Knepley } 437870e92668SMatthew Knepley 437952baeb72SSatish Balay /*@ 43809b94acceSBarry Smith SNESGetSolutionUpdate - Returns the vector where the solution update is 43819b94acceSBarry Smith stored. 43829b94acceSBarry Smith 4383c7afd0dbSLois Curfman McInnes Not Collective, but Vec is parallel if SNES is parallel 4384c7afd0dbSLois Curfman McInnes 43859b94acceSBarry Smith Input Parameter: 43869b94acceSBarry Smith . snes - the SNES context 43879b94acceSBarry Smith 43889b94acceSBarry Smith Output Parameter: 43899b94acceSBarry Smith . x - the solution update 43909b94acceSBarry Smith 439136851e7fSLois Curfman McInnes Level: advanced 439236851e7fSLois Curfman McInnes 43939b94acceSBarry Smith .keywords: SNES, nonlinear, get, solution, update 43949b94acceSBarry Smith 439585385478SLisandro Dalcin .seealso: SNESGetSolution(), SNESGetFunction() 43969b94acceSBarry Smith @*/ 43977087cfbeSBarry Smith PetscErrorCode SNESGetSolutionUpdate(SNES snes,Vec *x) 43989b94acceSBarry Smith { 43993a40ed3dSBarry Smith PetscFunctionBegin; 44000700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 44014482741eSBarry Smith PetscValidPointer(x,2); 440285385478SLisandro Dalcin *x = snes->vec_sol_update; 44033a40ed3dSBarry Smith PetscFunctionReturn(0); 44049b94acceSBarry Smith } 44059b94acceSBarry Smith 44069b94acceSBarry Smith /*@C 44073638b69dSLois Curfman McInnes SNESGetFunction - Returns the vector where the function is stored. 44089b94acceSBarry Smith 4409a63bb30eSJed Brown Not Collective, but Vec is parallel if SNES is parallel. Collective if Vec is requested, but has not been created yet. 4410c7afd0dbSLois Curfman McInnes 44119b94acceSBarry Smith Input Parameter: 44129b94acceSBarry Smith . snes - the SNES context 44139b94acceSBarry Smith 44149b94acceSBarry Smith Output Parameter: 44150298fd71SBarry Smith + r - the vector that is used to store residuals (or NULL if you don't want it) 4416f8b49ee9SBarry Smith . f - the function (or NULL if you don't want it); see SNESFunction for calling sequence details 44170298fd71SBarry Smith - ctx - the function context (or NULL if you don't want it) 44189b94acceSBarry Smith 441936851e7fSLois Curfman McInnes Level: advanced 442036851e7fSLois Curfman McInnes 4421a86d99e1SLois Curfman McInnes .keywords: SNES, nonlinear, get, function 44229b94acceSBarry Smith 4423bf388a1fSBarry Smith .seealso: SNESSetFunction(), SNESGetSolution(), SNESFunction 44249b94acceSBarry Smith @*/ 4425f8b49ee9SBarry Smith PetscErrorCode SNESGetFunction(SNES snes,Vec *r,PetscErrorCode (**f)(SNES,Vec,Vec,void*),void **ctx) 44269b94acceSBarry Smith { 4427a63bb30eSJed Brown PetscErrorCode ierr; 44286cab3a1bSJed Brown DM dm; 4429a63bb30eSJed Brown 44303a40ed3dSBarry Smith PetscFunctionBegin; 44310700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 4432a63bb30eSJed Brown if (r) { 4433a63bb30eSJed Brown if (!snes->vec_func) { 4434a63bb30eSJed Brown if (snes->vec_rhs) { 4435a63bb30eSJed Brown ierr = VecDuplicate(snes->vec_rhs,&snes->vec_func);CHKERRQ(ierr); 4436a63bb30eSJed Brown } else if (snes->vec_sol) { 4437a63bb30eSJed Brown ierr = VecDuplicate(snes->vec_sol,&snes->vec_func);CHKERRQ(ierr); 4438a63bb30eSJed Brown } else if (snes->dm) { 4439a63bb30eSJed Brown ierr = DMCreateGlobalVector(snes->dm,&snes->vec_func);CHKERRQ(ierr); 4440a63bb30eSJed Brown } 4441a63bb30eSJed Brown } 4442a63bb30eSJed Brown *r = snes->vec_func; 4443a63bb30eSJed Brown } 44446cab3a1bSJed Brown ierr = SNESGetDM(snes,&dm);CHKERRQ(ierr); 4445f8b49ee9SBarry Smith ierr = DMSNESGetFunction(dm,f,ctx);CHKERRQ(ierr); 44463a40ed3dSBarry Smith PetscFunctionReturn(0); 44479b94acceSBarry Smith } 44489b94acceSBarry Smith 4449c79ef259SPeter Brune /*@C 4450be95d8f1SBarry Smith SNESGetNGS - Returns the NGS function and context. 4451c79ef259SPeter Brune 4452c79ef259SPeter Brune Input Parameter: 4453c79ef259SPeter Brune . snes - the SNES context 4454c79ef259SPeter Brune 4455c79ef259SPeter Brune Output Parameter: 4456be95d8f1SBarry Smith + f - the function (or NULL) see SNESNGSFunction for details 44570298fd71SBarry Smith - ctx - the function context (or NULL) 4458c79ef259SPeter Brune 4459c79ef259SPeter Brune Level: advanced 4460c79ef259SPeter Brune 4461c79ef259SPeter Brune .keywords: SNES, nonlinear, get, function 4462c79ef259SPeter Brune 4463be95d8f1SBarry Smith .seealso: SNESSetNGS(), SNESGetFunction() 4464c79ef259SPeter Brune @*/ 4465c79ef259SPeter Brune 4466be95d8f1SBarry Smith PetscErrorCode SNESGetNGS (SNES snes, PetscErrorCode (**f)(SNES, Vec, Vec, void*), void ** ctx) 4467646217ecSPeter Brune { 44686cab3a1bSJed Brown PetscErrorCode ierr; 44696cab3a1bSJed Brown DM dm; 44706cab3a1bSJed Brown 4471646217ecSPeter Brune PetscFunctionBegin; 4472646217ecSPeter Brune PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 44736cab3a1bSJed Brown ierr = SNESGetDM(snes,&dm);CHKERRQ(ierr); 4474be95d8f1SBarry Smith ierr = DMSNESGetNGS(dm,f,ctx);CHKERRQ(ierr); 4475646217ecSPeter Brune PetscFunctionReturn(0); 4476646217ecSPeter Brune } 4477646217ecSPeter Brune 44783c7409f5SSatish Balay /*@C 44793c7409f5SSatish Balay SNESSetOptionsPrefix - Sets the prefix used for searching for all 4480d850072dSLois Curfman McInnes SNES options in the database. 44813c7409f5SSatish Balay 44823f9fe445SBarry Smith Logically Collective on SNES 4483fee21e36SBarry Smith 4484c7afd0dbSLois Curfman McInnes Input Parameter: 4485c7afd0dbSLois Curfman McInnes + snes - the SNES context 4486c7afd0dbSLois Curfman McInnes - prefix - the prefix to prepend to all option names 4487c7afd0dbSLois Curfman McInnes 4488d850072dSLois Curfman McInnes Notes: 4489a83b1b31SSatish Balay A hyphen (-) must NOT be given at the beginning of the prefix name. 4490c7afd0dbSLois Curfman McInnes The first character of all runtime options is AUTOMATICALLY the hyphen. 4491d850072dSLois Curfman McInnes 449236851e7fSLois Curfman McInnes Level: advanced 449336851e7fSLois Curfman McInnes 44943c7409f5SSatish Balay .keywords: SNES, set, options, prefix, database 4495a86d99e1SLois Curfman McInnes 4496a86d99e1SLois Curfman McInnes .seealso: SNESSetFromOptions() 44973c7409f5SSatish Balay @*/ 44987087cfbeSBarry Smith PetscErrorCode SNESSetOptionsPrefix(SNES snes,const char prefix[]) 44993c7409f5SSatish Balay { 4500dfbe8321SBarry Smith PetscErrorCode ierr; 45013c7409f5SSatish Balay 45023a40ed3dSBarry Smith PetscFunctionBegin; 45030700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 4504639f9d9dSBarry Smith ierr = PetscObjectSetOptionsPrefix((PetscObject)snes,prefix);CHKERRQ(ierr); 45051cee3971SBarry Smith if (!snes->ksp) {ierr = SNESGetKSP(snes,&snes->ksp);CHKERRQ(ierr);} 450635f5d045SPeter Brune if (snes->linesearch) { 45077601faf0SJed Brown ierr = SNESGetLineSearch(snes,&snes->linesearch);CHKERRQ(ierr); 450808b6c495SPeter Brune ierr = PetscObjectSetOptionsPrefix((PetscObject)snes->linesearch,prefix);CHKERRQ(ierr); 450935f5d045SPeter Brune } 451035f5d045SPeter Brune ierr = KSPSetOptionsPrefix(snes->ksp,prefix);CHKERRQ(ierr); 45113a40ed3dSBarry Smith PetscFunctionReturn(0); 45123c7409f5SSatish Balay } 45133c7409f5SSatish Balay 45143c7409f5SSatish Balay /*@C 4515f525115eSLois Curfman McInnes SNESAppendOptionsPrefix - Appends to the prefix used for searching for all 4516d850072dSLois Curfman McInnes SNES options in the database. 45173c7409f5SSatish Balay 45183f9fe445SBarry Smith Logically Collective on SNES 4519fee21e36SBarry Smith 4520c7afd0dbSLois Curfman McInnes Input Parameters: 4521c7afd0dbSLois Curfman McInnes + snes - the SNES context 4522c7afd0dbSLois Curfman McInnes - prefix - the prefix to prepend to all option names 4523c7afd0dbSLois Curfman McInnes 4524d850072dSLois Curfman McInnes Notes: 4525a83b1b31SSatish Balay A hyphen (-) must NOT be given at the beginning of the prefix name. 4526c7afd0dbSLois Curfman McInnes The first character of all runtime options is AUTOMATICALLY the hyphen. 4527d850072dSLois Curfman McInnes 452836851e7fSLois Curfman McInnes Level: advanced 452936851e7fSLois Curfman McInnes 45303c7409f5SSatish Balay .keywords: SNES, append, options, prefix, database 4531a86d99e1SLois Curfman McInnes 4532a86d99e1SLois Curfman McInnes .seealso: SNESGetOptionsPrefix() 45333c7409f5SSatish Balay @*/ 45347087cfbeSBarry Smith PetscErrorCode SNESAppendOptionsPrefix(SNES snes,const char prefix[]) 45353c7409f5SSatish Balay { 4536dfbe8321SBarry Smith PetscErrorCode ierr; 45373c7409f5SSatish Balay 45383a40ed3dSBarry Smith PetscFunctionBegin; 45390700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 4540639f9d9dSBarry Smith ierr = PetscObjectAppendOptionsPrefix((PetscObject)snes,prefix);CHKERRQ(ierr); 45411cee3971SBarry Smith if (!snes->ksp) {ierr = SNESGetKSP(snes,&snes->ksp);CHKERRQ(ierr);} 454235f5d045SPeter Brune if (snes->linesearch) { 45437601faf0SJed Brown ierr = SNESGetLineSearch(snes,&snes->linesearch);CHKERRQ(ierr); 454408b6c495SPeter Brune ierr = PetscObjectAppendOptionsPrefix((PetscObject)snes->linesearch,prefix);CHKERRQ(ierr); 454535f5d045SPeter Brune } 454635f5d045SPeter Brune ierr = KSPAppendOptionsPrefix(snes->ksp,prefix);CHKERRQ(ierr); 45473a40ed3dSBarry Smith PetscFunctionReturn(0); 45483c7409f5SSatish Balay } 45493c7409f5SSatish Balay 45509ab63eb5SSatish Balay /*@C 45513c7409f5SSatish Balay SNESGetOptionsPrefix - Sets the prefix used for searching for all 45523c7409f5SSatish Balay SNES options in the database. 45533c7409f5SSatish Balay 4554c7afd0dbSLois Curfman McInnes Not Collective 4555c7afd0dbSLois Curfman McInnes 45563c7409f5SSatish Balay Input Parameter: 45573c7409f5SSatish Balay . snes - the SNES context 45583c7409f5SSatish Balay 45593c7409f5SSatish Balay Output Parameter: 45603c7409f5SSatish Balay . prefix - pointer to the prefix string used 45613c7409f5SSatish Balay 45624ef407dbSRichard Tran Mills Notes: On the fortran side, the user should pass in a string 'prefix' of 45639ab63eb5SSatish Balay sufficient length to hold the prefix. 45649ab63eb5SSatish Balay 456536851e7fSLois Curfman McInnes Level: advanced 456636851e7fSLois Curfman McInnes 45673c7409f5SSatish Balay .keywords: SNES, get, options, prefix, database 4568a86d99e1SLois Curfman McInnes 4569a86d99e1SLois Curfman McInnes .seealso: SNESAppendOptionsPrefix() 45703c7409f5SSatish Balay @*/ 45717087cfbeSBarry Smith PetscErrorCode SNESGetOptionsPrefix(SNES snes,const char *prefix[]) 45723c7409f5SSatish Balay { 4573dfbe8321SBarry Smith PetscErrorCode ierr; 45743c7409f5SSatish Balay 45753a40ed3dSBarry Smith PetscFunctionBegin; 45760700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 4577639f9d9dSBarry Smith ierr = PetscObjectGetOptionsPrefix((PetscObject)snes,prefix);CHKERRQ(ierr); 45783a40ed3dSBarry Smith PetscFunctionReturn(0); 45793c7409f5SSatish Balay } 45803c7409f5SSatish Balay 4581b2002411SLois Curfman McInnes 45823cea93caSBarry Smith /*@C 45831c84c290SBarry Smith SNESRegister - Adds a method to the nonlinear solver package. 45841c84c290SBarry Smith 45851c84c290SBarry Smith Not collective 45861c84c290SBarry Smith 45871c84c290SBarry Smith Input Parameters: 45881c84c290SBarry Smith + name_solver - name of a new user-defined solver 45891c84c290SBarry Smith - routine_create - routine to create method context 45901c84c290SBarry Smith 45911c84c290SBarry Smith Notes: 45921c84c290SBarry Smith SNESRegister() may be called multiple times to add several user-defined solvers. 45931c84c290SBarry Smith 45941c84c290SBarry Smith Sample usage: 45951c84c290SBarry Smith .vb 4596bdf89e91SBarry Smith SNESRegister("my_solver",MySolverCreate); 45971c84c290SBarry Smith .ve 45981c84c290SBarry Smith 45991c84c290SBarry Smith Then, your solver can be chosen with the procedural interface via 46001c84c290SBarry Smith $ SNESSetType(snes,"my_solver") 46011c84c290SBarry Smith or at runtime via the option 46021c84c290SBarry Smith $ -snes_type my_solver 46031c84c290SBarry Smith 46041c84c290SBarry Smith Level: advanced 46051c84c290SBarry Smith 46061c84c290SBarry Smith Note: If your function is not being put into a shared library then use SNESRegister() instead 46071c84c290SBarry Smith 46081c84c290SBarry Smith .keywords: SNES, nonlinear, register 46091c84c290SBarry Smith 46101c84c290SBarry Smith .seealso: SNESRegisterAll(), SNESRegisterDestroy() 46113cea93caSBarry Smith 46127f6c08e0SMatthew Knepley Level: advanced 46133cea93caSBarry Smith @*/ 4614bdf89e91SBarry Smith PetscErrorCode SNESRegister(const char sname[],PetscErrorCode (*function)(SNES)) 4615b2002411SLois Curfman McInnes { 4616dfbe8321SBarry Smith PetscErrorCode ierr; 4617b2002411SLois Curfman McInnes 4618b2002411SLois Curfman McInnes PetscFunctionBegin; 4619a240a19fSJed Brown ierr = PetscFunctionListAdd(&SNESList,sname,function);CHKERRQ(ierr); 4620b2002411SLois Curfman McInnes PetscFunctionReturn(0); 4621b2002411SLois Curfman McInnes } 4622da9b6338SBarry Smith 46237087cfbeSBarry Smith PetscErrorCode SNESTestLocalMin(SNES snes) 4624da9b6338SBarry Smith { 4625dfbe8321SBarry Smith PetscErrorCode ierr; 462677431f27SBarry Smith PetscInt N,i,j; 4627da9b6338SBarry Smith Vec u,uh,fh; 4628da9b6338SBarry Smith PetscScalar value; 4629da9b6338SBarry Smith PetscReal norm; 4630da9b6338SBarry Smith 4631da9b6338SBarry Smith PetscFunctionBegin; 4632da9b6338SBarry Smith ierr = SNESGetSolution(snes,&u);CHKERRQ(ierr); 4633da9b6338SBarry Smith ierr = VecDuplicate(u,&uh);CHKERRQ(ierr); 4634da9b6338SBarry Smith ierr = VecDuplicate(u,&fh);CHKERRQ(ierr); 4635da9b6338SBarry Smith 4636da9b6338SBarry Smith /* currently only works for sequential */ 463722d28d08SBarry Smith ierr = PetscPrintf(PETSC_COMM_WORLD,"Testing FormFunction() for local min\n");CHKERRQ(ierr); 4638da9b6338SBarry Smith ierr = VecGetSize(u,&N);CHKERRQ(ierr); 4639da9b6338SBarry Smith for (i=0; i<N; i++) { 4640da9b6338SBarry Smith ierr = VecCopy(u,uh);CHKERRQ(ierr); 464177431f27SBarry Smith ierr = PetscPrintf(PETSC_COMM_WORLD,"i = %D\n",i);CHKERRQ(ierr); 4642da9b6338SBarry Smith for (j=-10; j<11; j++) { 46438b49ba18SBarry Smith value = PetscSign(j)*PetscExpReal(PetscAbs(j)-10.0); 4644da9b6338SBarry Smith ierr = VecSetValue(uh,i,value,ADD_VALUES);CHKERRQ(ierr); 46453ab0aad5SBarry Smith ierr = SNESComputeFunction(snes,uh,fh);CHKERRQ(ierr); 4646da9b6338SBarry Smith ierr = VecNorm(fh,NORM_2,&norm);CHKERRQ(ierr); 464777431f27SBarry Smith ierr = PetscPrintf(PETSC_COMM_WORLD," j norm %D %18.16e\n",j,norm);CHKERRQ(ierr); 4648da9b6338SBarry Smith value = -value; 4649da9b6338SBarry Smith ierr = VecSetValue(uh,i,value,ADD_VALUES);CHKERRQ(ierr); 4650da9b6338SBarry Smith } 4651da9b6338SBarry Smith } 46526bf464f9SBarry Smith ierr = VecDestroy(&uh);CHKERRQ(ierr); 46536bf464f9SBarry Smith ierr = VecDestroy(&fh);CHKERRQ(ierr); 4654da9b6338SBarry Smith PetscFunctionReturn(0); 4655da9b6338SBarry Smith } 465671f87433Sdalcinl 465771f87433Sdalcinl /*@ 4658fa9f3622SBarry Smith SNESKSPSetUseEW - Sets SNES use Eisenstat-Walker method for 465971f87433Sdalcinl computing relative tolerance for linear solvers within an inexact 466071f87433Sdalcinl Newton method. 466171f87433Sdalcinl 46623f9fe445SBarry Smith Logically Collective on SNES 466371f87433Sdalcinl 466471f87433Sdalcinl Input Parameters: 466571f87433Sdalcinl + snes - SNES context 466671f87433Sdalcinl - flag - PETSC_TRUE or PETSC_FALSE 466771f87433Sdalcinl 466864ba62caSBarry Smith Options Database: 466964ba62caSBarry Smith + -snes_ksp_ew - use Eisenstat-Walker method for determining linear system convergence 467064ba62caSBarry Smith . -snes_ksp_ew_version ver - version of Eisenstat-Walker method 467164ba62caSBarry Smith . -snes_ksp_ew_rtol0 <rtol0> - Sets rtol0 467264ba62caSBarry Smith . -snes_ksp_ew_rtolmax <rtolmax> - Sets rtolmax 467364ba62caSBarry Smith . -snes_ksp_ew_gamma <gamma> - Sets gamma 467464ba62caSBarry Smith . -snes_ksp_ew_alpha <alpha> - Sets alpha 467564ba62caSBarry Smith . -snes_ksp_ew_alpha2 <alpha2> - Sets alpha2 467664ba62caSBarry Smith - -snes_ksp_ew_threshold <threshold> - Sets threshold 467764ba62caSBarry Smith 467871f87433Sdalcinl Notes: 467971f87433Sdalcinl Currently, the default is to use a constant relative tolerance for 468071f87433Sdalcinl the inner linear solvers. Alternatively, one can use the 468171f87433Sdalcinl Eisenstat-Walker method, where the relative convergence tolerance 468271f87433Sdalcinl is reset at each Newton iteration according progress of the nonlinear 468371f87433Sdalcinl solver. 468471f87433Sdalcinl 468571f87433Sdalcinl Level: advanced 468671f87433Sdalcinl 468771f87433Sdalcinl Reference: 468871f87433Sdalcinl S. C. Eisenstat and H. F. Walker, "Choosing the forcing terms in an 468971f87433Sdalcinl inexact Newton method", SISC 17 (1), pp.16-32, 1996. 469071f87433Sdalcinl 469171f87433Sdalcinl .keywords: SNES, KSP, Eisenstat, Walker, convergence, test, inexact, Newton 469271f87433Sdalcinl 4693fa9f3622SBarry Smith .seealso: SNESKSPGetUseEW(), SNESKSPGetParametersEW(), SNESKSPSetParametersEW() 469471f87433Sdalcinl @*/ 46957087cfbeSBarry Smith PetscErrorCode SNESKSPSetUseEW(SNES snes,PetscBool flag) 469671f87433Sdalcinl { 469771f87433Sdalcinl PetscFunctionBegin; 46980700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 4699acfcf0e5SJed Brown PetscValidLogicalCollectiveBool(snes,flag,2); 470071f87433Sdalcinl snes->ksp_ewconv = flag; 470171f87433Sdalcinl PetscFunctionReturn(0); 470271f87433Sdalcinl } 470371f87433Sdalcinl 470471f87433Sdalcinl /*@ 4705fa9f3622SBarry Smith SNESKSPGetUseEW - Gets if SNES is using Eisenstat-Walker method 470671f87433Sdalcinl for computing relative tolerance for linear solvers within an 470771f87433Sdalcinl inexact Newton method. 470871f87433Sdalcinl 470971f87433Sdalcinl Not Collective 471071f87433Sdalcinl 471171f87433Sdalcinl Input Parameter: 471271f87433Sdalcinl . snes - SNES context 471371f87433Sdalcinl 471471f87433Sdalcinl Output Parameter: 471571f87433Sdalcinl . flag - PETSC_TRUE or PETSC_FALSE 471671f87433Sdalcinl 471771f87433Sdalcinl Notes: 471871f87433Sdalcinl Currently, the default is to use a constant relative tolerance for 471971f87433Sdalcinl the inner linear solvers. Alternatively, one can use the 472071f87433Sdalcinl Eisenstat-Walker method, where the relative convergence tolerance 472171f87433Sdalcinl is reset at each Newton iteration according progress of the nonlinear 472271f87433Sdalcinl solver. 472371f87433Sdalcinl 472471f87433Sdalcinl Level: advanced 472571f87433Sdalcinl 472671f87433Sdalcinl Reference: 472771f87433Sdalcinl S. C. Eisenstat and H. F. Walker, "Choosing the forcing terms in an 472871f87433Sdalcinl inexact Newton method", SISC 17 (1), pp.16-32, 1996. 472971f87433Sdalcinl 473071f87433Sdalcinl .keywords: SNES, KSP, Eisenstat, Walker, convergence, test, inexact, Newton 473171f87433Sdalcinl 4732fa9f3622SBarry Smith .seealso: SNESKSPSetUseEW(), SNESKSPGetParametersEW(), SNESKSPSetParametersEW() 473371f87433Sdalcinl @*/ 47347087cfbeSBarry Smith PetscErrorCode SNESKSPGetUseEW(SNES snes, PetscBool *flag) 473571f87433Sdalcinl { 473671f87433Sdalcinl PetscFunctionBegin; 47370700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 473871f87433Sdalcinl PetscValidPointer(flag,2); 473971f87433Sdalcinl *flag = snes->ksp_ewconv; 474071f87433Sdalcinl PetscFunctionReturn(0); 474171f87433Sdalcinl } 474271f87433Sdalcinl 474371f87433Sdalcinl /*@ 4744fa9f3622SBarry Smith SNESKSPSetParametersEW - Sets parameters for Eisenstat-Walker 474571f87433Sdalcinl convergence criteria for the linear solvers within an inexact 474671f87433Sdalcinl Newton method. 474771f87433Sdalcinl 47483f9fe445SBarry Smith Logically Collective on SNES 474971f87433Sdalcinl 475071f87433Sdalcinl Input Parameters: 475171f87433Sdalcinl + snes - SNES context 475271f87433Sdalcinl . version - version 1, 2 (default is 2) or 3 475371f87433Sdalcinl . rtol_0 - initial relative tolerance (0 <= rtol_0 < 1) 475471f87433Sdalcinl . rtol_max - maximum relative tolerance (0 <= rtol_max < 1) 475571f87433Sdalcinl . gamma - multiplicative factor for version 2 rtol computation 475671f87433Sdalcinl (0 <= gamma2 <= 1) 475771f87433Sdalcinl . alpha - power for version 2 rtol computation (1 < alpha <= 2) 475871f87433Sdalcinl . alpha2 - power for safeguard 475971f87433Sdalcinl - threshold - threshold for imposing safeguard (0 < threshold < 1) 476071f87433Sdalcinl 476171f87433Sdalcinl Note: 476271f87433Sdalcinl Version 3 was contributed by Luis Chacon, June 2006. 476371f87433Sdalcinl 476471f87433Sdalcinl Use PETSC_DEFAULT to retain the default for any of the parameters. 476571f87433Sdalcinl 476671f87433Sdalcinl Level: advanced 476771f87433Sdalcinl 476871f87433Sdalcinl Reference: 476971f87433Sdalcinl S. C. Eisenstat and H. F. Walker, "Choosing the forcing terms in an 477071f87433Sdalcinl inexact Newton method", Utah State University Math. Stat. Dept. Res. 477171f87433Sdalcinl Report 6/94/75, June, 1994, to appear in SIAM J. Sci. Comput. 477271f87433Sdalcinl 477371f87433Sdalcinl .keywords: SNES, KSP, Eisenstat, Walker, set, parameters 477471f87433Sdalcinl 4775fa9f3622SBarry Smith .seealso: SNESKSPSetUseEW(), SNESKSPGetUseEW(), SNESKSPGetParametersEW() 477671f87433Sdalcinl @*/ 4777f5af7f23SKarl Rupp PetscErrorCode SNESKSPSetParametersEW(SNES snes,PetscInt version,PetscReal rtol_0,PetscReal rtol_max,PetscReal gamma,PetscReal alpha,PetscReal alpha2,PetscReal threshold) 477871f87433Sdalcinl { 4779fa9f3622SBarry Smith SNESKSPEW *kctx; 47805fd66863SKarl Rupp 478171f87433Sdalcinl PetscFunctionBegin; 47820700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 4783fa9f3622SBarry Smith kctx = (SNESKSPEW*)snes->kspconvctx; 4784e32f2f54SBarry Smith if (!kctx) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"No Eisenstat-Walker context existing"); 4785c5eb9154SBarry Smith PetscValidLogicalCollectiveInt(snes,version,2); 4786c5eb9154SBarry Smith PetscValidLogicalCollectiveReal(snes,rtol_0,3); 4787c5eb9154SBarry Smith PetscValidLogicalCollectiveReal(snes,rtol_max,4); 4788c5eb9154SBarry Smith PetscValidLogicalCollectiveReal(snes,gamma,5); 4789c5eb9154SBarry Smith PetscValidLogicalCollectiveReal(snes,alpha,6); 4790c5eb9154SBarry Smith PetscValidLogicalCollectiveReal(snes,alpha2,7); 4791c5eb9154SBarry Smith PetscValidLogicalCollectiveReal(snes,threshold,8); 479271f87433Sdalcinl 479371f87433Sdalcinl if (version != PETSC_DEFAULT) kctx->version = version; 479471f87433Sdalcinl if (rtol_0 != PETSC_DEFAULT) kctx->rtol_0 = rtol_0; 479571f87433Sdalcinl if (rtol_max != PETSC_DEFAULT) kctx->rtol_max = rtol_max; 479671f87433Sdalcinl if (gamma != PETSC_DEFAULT) kctx->gamma = gamma; 479771f87433Sdalcinl if (alpha != PETSC_DEFAULT) kctx->alpha = alpha; 479871f87433Sdalcinl if (alpha2 != PETSC_DEFAULT) kctx->alpha2 = alpha2; 479971f87433Sdalcinl if (threshold != PETSC_DEFAULT) kctx->threshold = threshold; 480071f87433Sdalcinl 4801f23aa3ddSBarry Smith if (kctx->version < 1 || kctx->version > 3) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Only versions 1, 2 and 3 are supported: %D",kctx->version); 480257622a8eSBarry Smith if (kctx->rtol_0 < 0.0 || kctx->rtol_0 >= 1.0) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"0.0 <= rtol_0 < 1.0: %g",(double)kctx->rtol_0); 480357622a8eSBarry Smith if (kctx->rtol_max < 0.0 || kctx->rtol_max >= 1.0) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"0.0 <= rtol_max (%g) < 1.0\n",(double)kctx->rtol_max); 480457622a8eSBarry Smith if (kctx->gamma < 0.0 || kctx->gamma > 1.0) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"0.0 <= gamma (%g) <= 1.0\n",(double)kctx->gamma); 480557622a8eSBarry Smith if (kctx->alpha <= 1.0 || kctx->alpha > 2.0) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"1.0 < alpha (%g) <= 2.0\n",(double)kctx->alpha); 480657622a8eSBarry Smith if (kctx->threshold <= 0.0 || kctx->threshold >= 1.0) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"0.0 < threshold (%g) < 1.0\n",(double)kctx->threshold); 480771f87433Sdalcinl PetscFunctionReturn(0); 480871f87433Sdalcinl } 480971f87433Sdalcinl 481071f87433Sdalcinl /*@ 4811fa9f3622SBarry Smith SNESKSPGetParametersEW - Gets parameters for Eisenstat-Walker 481271f87433Sdalcinl convergence criteria for the linear solvers within an inexact 481371f87433Sdalcinl Newton method. 481471f87433Sdalcinl 481571f87433Sdalcinl Not Collective 481671f87433Sdalcinl 481771f87433Sdalcinl Input Parameters: 481871f87433Sdalcinl snes - SNES context 481971f87433Sdalcinl 482071f87433Sdalcinl Output Parameters: 482171f87433Sdalcinl + version - version 1, 2 (default is 2) or 3 482271f87433Sdalcinl . rtol_0 - initial relative tolerance (0 <= rtol_0 < 1) 482371f87433Sdalcinl . rtol_max - maximum relative tolerance (0 <= rtol_max < 1) 4824bf388a1fSBarry Smith . gamma - multiplicative factor for version 2 rtol computation (0 <= gamma2 <= 1) 482571f87433Sdalcinl . alpha - power for version 2 rtol computation (1 < alpha <= 2) 482671f87433Sdalcinl . alpha2 - power for safeguard 482771f87433Sdalcinl - threshold - threshold for imposing safeguard (0 < threshold < 1) 482871f87433Sdalcinl 482971f87433Sdalcinl Level: advanced 483071f87433Sdalcinl 483171f87433Sdalcinl .keywords: SNES, KSP, Eisenstat, Walker, get, parameters 483271f87433Sdalcinl 4833fa9f3622SBarry Smith .seealso: SNESKSPSetUseEW(), SNESKSPGetUseEW(), SNESKSPSetParametersEW() 483471f87433Sdalcinl @*/ 4835bf388a1fSBarry Smith PetscErrorCode SNESKSPGetParametersEW(SNES snes,PetscInt *version,PetscReal *rtol_0,PetscReal *rtol_max,PetscReal *gamma,PetscReal *alpha,PetscReal *alpha2,PetscReal *threshold) 483671f87433Sdalcinl { 4837fa9f3622SBarry Smith SNESKSPEW *kctx; 48385fd66863SKarl Rupp 483971f87433Sdalcinl PetscFunctionBegin; 48400700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 4841fa9f3622SBarry Smith kctx = (SNESKSPEW*)snes->kspconvctx; 4842e32f2f54SBarry Smith if (!kctx) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"No Eisenstat-Walker context existing"); 484371f87433Sdalcinl if (version) *version = kctx->version; 484471f87433Sdalcinl if (rtol_0) *rtol_0 = kctx->rtol_0; 484571f87433Sdalcinl if (rtol_max) *rtol_max = kctx->rtol_max; 484671f87433Sdalcinl if (gamma) *gamma = kctx->gamma; 484771f87433Sdalcinl if (alpha) *alpha = kctx->alpha; 484871f87433Sdalcinl if (alpha2) *alpha2 = kctx->alpha2; 484971f87433Sdalcinl if (threshold) *threshold = kctx->threshold; 485071f87433Sdalcinl PetscFunctionReturn(0); 485171f87433Sdalcinl } 485271f87433Sdalcinl 4853d5378b5fSDmitry Karpeev PetscErrorCode KSPPreSolve_SNESEW(KSP ksp, Vec b, Vec x, SNES snes) 485471f87433Sdalcinl { 485571f87433Sdalcinl PetscErrorCode ierr; 4856fa9f3622SBarry Smith SNESKSPEW *kctx = (SNESKSPEW*)snes->kspconvctx; 485771f87433Sdalcinl PetscReal rtol = PETSC_DEFAULT,stol; 485871f87433Sdalcinl 485971f87433Sdalcinl PetscFunctionBegin; 4860d4211eb9SBarry Smith if (!snes->ksp_ewconv) PetscFunctionReturn(0); 486130058271SDmitry Karpeev if (!snes->iter) { 486230058271SDmitry Karpeev rtol = kctx->rtol_0; /* first time in, so use the original user rtol */ 486330058271SDmitry Karpeev ierr = VecNorm(snes->vec_func,NORM_2,&kctx->norm_first);CHKERRQ(ierr); 486430058271SDmitry Karpeev } 4865f5af7f23SKarl Rupp else { 486671f87433Sdalcinl if (kctx->version == 1) { 486771f87433Sdalcinl rtol = (snes->norm - kctx->lresid_last)/kctx->norm_last; 486871f87433Sdalcinl if (rtol < 0.0) rtol = -rtol; 486985ec1a3cSBarry Smith stol = PetscPowReal(kctx->rtol_last,kctx->alpha2); 487071f87433Sdalcinl if (stol > kctx->threshold) rtol = PetscMax(rtol,stol); 487171f87433Sdalcinl } else if (kctx->version == 2) { 487285ec1a3cSBarry Smith rtol = kctx->gamma * PetscPowReal(snes->norm/kctx->norm_last,kctx->alpha); 487385ec1a3cSBarry Smith stol = kctx->gamma * PetscPowReal(kctx->rtol_last,kctx->alpha); 487471f87433Sdalcinl if (stol > kctx->threshold) rtol = PetscMax(rtol,stol); 487571f87433Sdalcinl } else if (kctx->version == 3) { /* contributed by Luis Chacon, June 2006. */ 487685ec1a3cSBarry Smith rtol = kctx->gamma * PetscPowReal(snes->norm/kctx->norm_last,kctx->alpha); 487771f87433Sdalcinl /* safeguard: avoid sharp decrease of rtol */ 487885ec1a3cSBarry Smith stol = kctx->gamma*PetscPowReal(kctx->rtol_last,kctx->alpha); 487971f87433Sdalcinl stol = PetscMax(rtol,stol); 488071f87433Sdalcinl rtol = PetscMin(kctx->rtol_0,stol); 488171f87433Sdalcinl /* safeguard: avoid oversolving */ 488230058271SDmitry Karpeev stol = kctx->gamma*(kctx->norm_first*snes->rtol)/snes->norm; 488371f87433Sdalcinl stol = PetscMax(rtol,stol); 488471f87433Sdalcinl rtol = PetscMin(kctx->rtol_0,stol); 4885e32f2f54SBarry Smith } else SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Only versions 1, 2 or 3 are supported: %D",kctx->version); 488671f87433Sdalcinl } 488771f87433Sdalcinl /* safeguard: avoid rtol greater than one */ 488871f87433Sdalcinl rtol = PetscMin(rtol,kctx->rtol_max); 488971f87433Sdalcinl ierr = KSPSetTolerances(ksp,rtol,PETSC_DEFAULT,PETSC_DEFAULT,PETSC_DEFAULT);CHKERRQ(ierr); 489057622a8eSBarry Smith ierr = PetscInfo3(snes,"iter %D, Eisenstat-Walker (version %D) KSP rtol=%g\n",snes->iter,kctx->version,(double)rtol);CHKERRQ(ierr); 489171f87433Sdalcinl PetscFunctionReturn(0); 489271f87433Sdalcinl } 489371f87433Sdalcinl 4894d5378b5fSDmitry Karpeev PetscErrorCode KSPPostSolve_SNESEW(KSP ksp, Vec b, Vec x, SNES snes) 489571f87433Sdalcinl { 489671f87433Sdalcinl PetscErrorCode ierr; 4897fa9f3622SBarry Smith SNESKSPEW *kctx = (SNESKSPEW*)snes->kspconvctx; 489871f87433Sdalcinl PCSide pcside; 489971f87433Sdalcinl Vec lres; 490071f87433Sdalcinl 490171f87433Sdalcinl PetscFunctionBegin; 4902d4211eb9SBarry Smith if (!snes->ksp_ewconv) PetscFunctionReturn(0); 490371f87433Sdalcinl ierr = KSPGetTolerances(ksp,&kctx->rtol_last,0,0,0);CHKERRQ(ierr); 490471dbe336SPeter Brune kctx->norm_last = snes->norm; 490571f87433Sdalcinl if (kctx->version == 1) { 49064f00ce20SMatthew G. Knepley PC pc; 49074f00ce20SMatthew G. Knepley PetscBool isNone; 49084f00ce20SMatthew G. Knepley 49094f00ce20SMatthew G. Knepley ierr = KSPGetPC(ksp, &pc);CHKERRQ(ierr); 49104f00ce20SMatthew G. Knepley ierr = PetscObjectTypeCompare((PetscObject) pc, PCNONE, &isNone);CHKERRQ(ierr); 4911b037da10SBarry Smith ierr = KSPGetPCSide(ksp,&pcside);CHKERRQ(ierr); 49124f00ce20SMatthew G. Knepley if (pcside == PC_RIGHT || isNone) { /* XXX Should we also test KSP_UNPRECONDITIONED_NORM ? */ 491371f87433Sdalcinl /* KSP residual is true linear residual */ 491471f87433Sdalcinl ierr = KSPGetResidualNorm(ksp,&kctx->lresid_last);CHKERRQ(ierr); 491571f87433Sdalcinl } else { 491671f87433Sdalcinl /* KSP residual is preconditioned residual */ 491771f87433Sdalcinl /* compute true linear residual norm */ 491871f87433Sdalcinl ierr = VecDuplicate(b,&lres);CHKERRQ(ierr); 491971f87433Sdalcinl ierr = MatMult(snes->jacobian,x,lres);CHKERRQ(ierr); 492071f87433Sdalcinl ierr = VecAYPX(lres,-1.0,b);CHKERRQ(ierr); 492171f87433Sdalcinl ierr = VecNorm(lres,NORM_2,&kctx->lresid_last);CHKERRQ(ierr); 49226bf464f9SBarry Smith ierr = VecDestroy(&lres);CHKERRQ(ierr); 492371f87433Sdalcinl } 492471f87433Sdalcinl } 492571f87433Sdalcinl PetscFunctionReturn(0); 492671f87433Sdalcinl } 492771f87433Sdalcinl 4928d4211eb9SBarry Smith /*@ 4929d4211eb9SBarry Smith SNESGetKSP - Returns the KSP context for a SNES solver. 4930d4211eb9SBarry Smith 4931d4211eb9SBarry Smith Not Collective, but if SNES object is parallel, then KSP object is parallel 4932d4211eb9SBarry Smith 4933d4211eb9SBarry Smith Input Parameter: 4934d4211eb9SBarry Smith . snes - the SNES context 4935d4211eb9SBarry Smith 4936d4211eb9SBarry Smith Output Parameter: 4937d4211eb9SBarry Smith . ksp - the KSP context 4938d4211eb9SBarry Smith 4939d4211eb9SBarry Smith Notes: 4940d4211eb9SBarry Smith The user can then directly manipulate the KSP context to set various 4941d4211eb9SBarry Smith options, etc. Likewise, the user can then extract and manipulate the 4942d4211eb9SBarry Smith PC contexts as well. 4943d4211eb9SBarry Smith 4944d4211eb9SBarry Smith Level: beginner 4945d4211eb9SBarry Smith 4946d4211eb9SBarry Smith .keywords: SNES, nonlinear, get, KSP, context 4947d4211eb9SBarry Smith 4948d4211eb9SBarry Smith .seealso: KSPGetPC(), SNESCreate(), KSPCreate(), SNESSetKSP() 4949d4211eb9SBarry Smith @*/ 4950d4211eb9SBarry Smith PetscErrorCode SNESGetKSP(SNES snes,KSP *ksp) 495171f87433Sdalcinl { 495271f87433Sdalcinl PetscErrorCode ierr; 495371f87433Sdalcinl 495471f87433Sdalcinl PetscFunctionBegin; 4955d4211eb9SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 4956d4211eb9SBarry Smith PetscValidPointer(ksp,2); 4957d4211eb9SBarry Smith 4958d4211eb9SBarry Smith if (!snes->ksp) { 4959a5c2985bSBarry Smith PetscBool monitor = PETSC_FALSE; 4960a5c2985bSBarry Smith 4961d4211eb9SBarry Smith ierr = KSPCreate(PetscObjectComm((PetscObject)snes),&snes->ksp);CHKERRQ(ierr); 4962d4211eb9SBarry Smith ierr = PetscObjectIncrementTabLevel((PetscObject)snes->ksp,(PetscObject)snes,1);CHKERRQ(ierr); 49633bb1ff40SBarry Smith ierr = PetscLogObjectParent((PetscObject)snes,(PetscObject)snes->ksp);CHKERRQ(ierr); 4964d4211eb9SBarry Smith 4965d5378b5fSDmitry Karpeev ierr = KSPSetPreSolve(snes->ksp,(PetscErrorCode (*)(KSP,Vec,Vec,void*))KSPPreSolve_SNESEW,snes);CHKERRQ(ierr); 4966d5378b5fSDmitry Karpeev ierr = KSPSetPostSolve(snes->ksp,(PetscErrorCode (*)(KSP,Vec,Vec,void*))KSPPostSolve_SNESEW,snes);CHKERRQ(ierr); 4967a5c2985bSBarry Smith 4968c5929fdfSBarry Smith ierr = PetscOptionsGetBool(((PetscObject)snes)->options,((PetscObject)snes)->prefix,"-ksp_monitor_snes",&monitor,NULL);CHKERRQ(ierr); 4969a5c2985bSBarry Smith if (monitor) { 4970a5c2985bSBarry Smith ierr = KSPMonitorSet(snes->ksp,KSPMonitorSNES,snes,NULL);CHKERRQ(ierr); 4971a5c2985bSBarry Smith } 4972e5f7ee39SBarry Smith monitor = PETSC_FALSE; 4973c5929fdfSBarry Smith ierr = PetscOptionsGetBool(((PetscObject)snes)->options,((PetscObject)snes)->prefix,"-ksp_monitor_snes_lg",&monitor,NULL);CHKERRQ(ierr); 4974e5f7ee39SBarry Smith if (monitor) { 4975e5f7ee39SBarry Smith PetscObject *objs; 49768b0b5a47SLisandro Dalcin ierr = KSPMonitorSNESLGResidualNormCreate(PetscObjectComm((PetscObject)snes),NULL,NULL,PETSC_DECIDE,PETSC_DECIDE,600,600,&objs);CHKERRQ(ierr); 4977e5f7ee39SBarry Smith objs[0] = (PetscObject) snes; 4978e5f7ee39SBarry Smith ierr = KSPMonitorSet(snes->ksp,(PetscErrorCode (*)(KSP,PetscInt,PetscReal,void*))KSPMonitorSNESLGResidualNorm,objs,(PetscErrorCode (*)(void**))KSPMonitorSNESLGResidualNormDestroy);CHKERRQ(ierr); 4979e5f7ee39SBarry Smith } 4980d4211eb9SBarry Smith } 4981d4211eb9SBarry Smith *ksp = snes->ksp; 498271f87433Sdalcinl PetscFunctionReturn(0); 498371f87433Sdalcinl } 49846c699258SBarry Smith 4985d4211eb9SBarry Smith 4986af0996ceSBarry Smith #include <petsc/private/dmimpl.h> 49876c699258SBarry Smith /*@ 49882a808120SBarry Smith SNESSetDM - Sets the DM that may be used by some nonlinear solvers or their underlying preconditioners 49896c699258SBarry Smith 49903f9fe445SBarry Smith Logically Collective on SNES 49916c699258SBarry Smith 49926c699258SBarry Smith Input Parameters: 49932a808120SBarry Smith + snes - the nonlinear solver context 49942a808120SBarry Smith - dm - the dm, cannot be NULL 49956c699258SBarry Smith 49966c699258SBarry Smith Level: intermediate 49976c699258SBarry Smith 49984c2026ceSFande Kong .seealso: SNESGetDM(), KSPSetDM(), KSPGetDM() 49996c699258SBarry Smith @*/ 50007087cfbeSBarry Smith PetscErrorCode SNESSetDM(SNES snes,DM dm) 50016c699258SBarry Smith { 50026c699258SBarry Smith PetscErrorCode ierr; 5003345fed2cSBarry Smith KSP ksp; 5004942e3340SBarry Smith DMSNES sdm; 50056c699258SBarry Smith 50066c699258SBarry Smith PetscFunctionBegin; 50070700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 50082a808120SBarry Smith PetscValidHeaderSpecific(dm,DM_CLASSID,2); 50092a808120SBarry Smith ierr = PetscObjectReference((PetscObject)dm);CHKERRQ(ierr); 5010942e3340SBarry Smith if (snes->dm) { /* Move the DMSNES context over to the new DM unless the new DM already has one */ 501151f4b3c7SToby Isaac if (snes->dm->dmsnes && !dm->dmsnes) { 5012942e3340SBarry Smith ierr = DMCopyDMSNES(snes->dm,dm);CHKERRQ(ierr); 5013942e3340SBarry Smith ierr = DMGetDMSNES(snes->dm,&sdm);CHKERRQ(ierr); 5014f5af7f23SKarl Rupp if (sdm->originaldm == snes->dm) sdm->originaldm = dm; /* Grant write privileges to the replacement DM */ 50156cab3a1bSJed Brown } 5016dc822a44SJed Brown ierr = DMCoarsenHookRemove(snes->dm,DMCoarsenHook_SNESVecSol,DMRestrictHook_SNESVecSol,snes);CHKERRQ(ierr); 50176bf464f9SBarry Smith ierr = DMDestroy(&snes->dm);CHKERRQ(ierr); 50186cab3a1bSJed Brown } 50196c699258SBarry Smith snes->dm = dm; 5020116d1032SJed Brown snes->dmAuto = PETSC_FALSE; 5021f5af7f23SKarl Rupp 5022345fed2cSBarry Smith ierr = SNESGetKSP(snes,&ksp);CHKERRQ(ierr); 5023345fed2cSBarry Smith ierr = KSPSetDM(ksp,dm);CHKERRQ(ierr); 5024f22f69f0SBarry Smith ierr = KSPSetDMActive(ksp,PETSC_FALSE);CHKERRQ(ierr); 5025efd4aadfSBarry Smith if (snes->npc) { 5026efd4aadfSBarry Smith ierr = SNESSetDM(snes->npc, snes->dm);CHKERRQ(ierr); 5027efd4aadfSBarry Smith ierr = SNESSetNPCSide(snes,snes->npcside);CHKERRQ(ierr); 50282c155ee1SBarry Smith } 50296c699258SBarry Smith PetscFunctionReturn(0); 50306c699258SBarry Smith } 50316c699258SBarry Smith 50326c699258SBarry Smith /*@ 50336c699258SBarry Smith SNESGetDM - Gets the DM that may be used by some preconditioners 50346c699258SBarry Smith 50353f9fe445SBarry Smith Not Collective but DM obtained is parallel on SNES 50366c699258SBarry Smith 50376c699258SBarry Smith Input Parameter: 50386c699258SBarry Smith . snes - the preconditioner context 50396c699258SBarry Smith 50406c699258SBarry Smith Output Parameter: 50416c699258SBarry Smith . dm - the dm 50426c699258SBarry Smith 50436c699258SBarry Smith Level: intermediate 50446c699258SBarry Smith 50454c2026ceSFande Kong .seealso: SNESSetDM(), KSPSetDM(), KSPGetDM() 50466c699258SBarry Smith @*/ 50477087cfbeSBarry Smith PetscErrorCode SNESGetDM(SNES snes,DM *dm) 50486c699258SBarry Smith { 50496cab3a1bSJed Brown PetscErrorCode ierr; 50506cab3a1bSJed Brown 50516c699258SBarry Smith PetscFunctionBegin; 50520700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 50536cab3a1bSJed Brown if (!snes->dm) { 5054ce94432eSBarry Smith ierr = DMShellCreate(PetscObjectComm((PetscObject)snes),&snes->dm);CHKERRQ(ierr); 5055116d1032SJed Brown snes->dmAuto = PETSC_TRUE; 50566cab3a1bSJed Brown } 50576c699258SBarry Smith *dm = snes->dm; 50586c699258SBarry Smith PetscFunctionReturn(0); 50596c699258SBarry Smith } 50600807856dSBarry Smith 506131823bd8SMatthew G Knepley /*@ 5062be95d8f1SBarry Smith SNESSetNPC - Sets the nonlinear preconditioner to be used. 506331823bd8SMatthew G Knepley 506431823bd8SMatthew G Knepley Collective on SNES 506531823bd8SMatthew G Knepley 506631823bd8SMatthew G Knepley Input Parameters: 506731823bd8SMatthew G Knepley + snes - iterative context obtained from SNESCreate() 506831823bd8SMatthew G Knepley - pc - the preconditioner object 506931823bd8SMatthew G Knepley 507031823bd8SMatthew G Knepley Notes: 5071be95d8f1SBarry Smith Use SNESGetNPC() to retrieve the preconditioner context (for example, 507231823bd8SMatthew G Knepley to configure it using the API). 507331823bd8SMatthew G Knepley 507431823bd8SMatthew G Knepley Level: developer 507531823bd8SMatthew G Knepley 507631823bd8SMatthew G Knepley .keywords: SNES, set, precondition 50773ad1a0b9SPatrick Farrell .seealso: SNESGetNPC(), SNESHasNPC() 507831823bd8SMatthew G Knepley @*/ 5079be95d8f1SBarry Smith PetscErrorCode SNESSetNPC(SNES snes, SNES pc) 508031823bd8SMatthew G Knepley { 508131823bd8SMatthew G Knepley PetscErrorCode ierr; 508231823bd8SMatthew G Knepley 508331823bd8SMatthew G Knepley PetscFunctionBegin; 508431823bd8SMatthew G Knepley PetscValidHeaderSpecific(snes, SNES_CLASSID, 1); 508531823bd8SMatthew G Knepley PetscValidHeaderSpecific(pc, SNES_CLASSID, 2); 508631823bd8SMatthew G Knepley PetscCheckSameComm(snes, 1, pc, 2); 508731823bd8SMatthew G Knepley ierr = PetscObjectReference((PetscObject) pc);CHKERRQ(ierr); 5088efd4aadfSBarry Smith ierr = SNESDestroy(&snes->npc);CHKERRQ(ierr); 5089efd4aadfSBarry Smith snes->npc = pc; 5090efd4aadfSBarry Smith ierr = PetscLogObjectParent((PetscObject)snes, (PetscObject)snes->npc);CHKERRQ(ierr); 509131823bd8SMatthew G Knepley PetscFunctionReturn(0); 509231823bd8SMatthew G Knepley } 509331823bd8SMatthew G Knepley 509431823bd8SMatthew G Knepley /*@ 5095be95d8f1SBarry Smith SNESGetNPC - Creates a nonlinear preconditioning solver (SNES) to be used to precondition the nonlinear solver. 509631823bd8SMatthew G Knepley 509731823bd8SMatthew G Knepley Not Collective 509831823bd8SMatthew G Knepley 509931823bd8SMatthew G Knepley Input Parameter: 510031823bd8SMatthew G Knepley . snes - iterative context obtained from SNESCreate() 510131823bd8SMatthew G Knepley 510231823bd8SMatthew G Knepley Output Parameter: 510331823bd8SMatthew G Knepley . pc - preconditioner context 510431823bd8SMatthew G Knepley 5105be95d8f1SBarry Smith Notes: If a SNES was previously set with SNESSetNPC() then that SNES is returned. 5106be95d8f1SBarry Smith 510731823bd8SMatthew G Knepley Level: developer 510831823bd8SMatthew G Knepley 510931823bd8SMatthew G Knepley .keywords: SNES, get, preconditioner 51103ad1a0b9SPatrick Farrell .seealso: SNESSetNPC(), SNESHasNPC() 511131823bd8SMatthew G Knepley @*/ 5112be95d8f1SBarry Smith PetscErrorCode SNESGetNPC(SNES snes, SNES *pc) 511331823bd8SMatthew G Knepley { 511431823bd8SMatthew G Knepley PetscErrorCode ierr; 5115a64e098fSPeter Brune const char *optionsprefix; 511631823bd8SMatthew G Knepley 511731823bd8SMatthew G Knepley PetscFunctionBegin; 511831823bd8SMatthew G Knepley PetscValidHeaderSpecific(snes, SNES_CLASSID, 1); 511931823bd8SMatthew G Knepley PetscValidPointer(pc, 2); 5120efd4aadfSBarry Smith if (!snes->npc) { 5121efd4aadfSBarry Smith ierr = SNESCreate(PetscObjectComm((PetscObject)snes),&snes->npc);CHKERRQ(ierr); 5122efd4aadfSBarry Smith ierr = PetscObjectIncrementTabLevel((PetscObject)snes->npc,(PetscObject)snes,1);CHKERRQ(ierr); 5123efd4aadfSBarry Smith ierr = PetscLogObjectParent((PetscObject)snes,(PetscObject)snes->npc);CHKERRQ(ierr); 5124a64e098fSPeter Brune ierr = SNESGetOptionsPrefix(snes,&optionsprefix);CHKERRQ(ierr); 5125efd4aadfSBarry Smith ierr = SNESSetOptionsPrefix(snes->npc,optionsprefix);CHKERRQ(ierr); 5126efd4aadfSBarry Smith ierr = SNESAppendOptionsPrefix(snes->npc,"npc_");CHKERRQ(ierr); 5127efd4aadfSBarry Smith ierr = SNESSetCountersReset(snes->npc,PETSC_FALSE);CHKERRQ(ierr); 512831823bd8SMatthew G Knepley } 5129efd4aadfSBarry Smith *pc = snes->npc; 513031823bd8SMatthew G Knepley PetscFunctionReturn(0); 513131823bd8SMatthew G Knepley } 513231823bd8SMatthew G Knepley 51333ad1a0b9SPatrick Farrell /*@ 51343ad1a0b9SPatrick Farrell SNESHasNPC - Returns whether a nonlinear preconditioner exists 51353ad1a0b9SPatrick Farrell 51363ad1a0b9SPatrick Farrell Not Collective 51373ad1a0b9SPatrick Farrell 51383ad1a0b9SPatrick Farrell Input Parameter: 51393ad1a0b9SPatrick Farrell . snes - iterative context obtained from SNESCreate() 51403ad1a0b9SPatrick Farrell 51413ad1a0b9SPatrick Farrell Output Parameter: 51423ad1a0b9SPatrick Farrell . has_npc - whether the SNES has an NPC or not 51433ad1a0b9SPatrick Farrell 51443ad1a0b9SPatrick Farrell Level: developer 51453ad1a0b9SPatrick Farrell 51463ad1a0b9SPatrick Farrell .keywords: SNES, has, preconditioner 51473ad1a0b9SPatrick Farrell .seealso: SNESSetNPC(), SNESGetNPC() 51483ad1a0b9SPatrick Farrell @*/ 51493ad1a0b9SPatrick Farrell PetscErrorCode SNESHasNPC(SNES snes, PetscBool *has_npc) 51503ad1a0b9SPatrick Farrell { 51513ad1a0b9SPatrick Farrell PetscFunctionBegin; 51523ad1a0b9SPatrick Farrell PetscValidHeaderSpecific(snes, SNES_CLASSID, 1); 5153efd4aadfSBarry Smith *has_npc = (PetscBool) (snes->npc ? PETSC_TRUE : PETSC_FALSE); 51543ad1a0b9SPatrick Farrell PetscFunctionReturn(0); 51553ad1a0b9SPatrick Farrell } 51563ad1a0b9SPatrick Farrell 5157c40d0f55SPeter Brune /*@ 5158be95d8f1SBarry Smith SNESSetNPCSide - Sets the preconditioning side. 5159c40d0f55SPeter Brune 5160c40d0f55SPeter Brune Logically Collective on SNES 5161c40d0f55SPeter Brune 5162c40d0f55SPeter Brune Input Parameter: 5163c40d0f55SPeter Brune . snes - iterative context obtained from SNESCreate() 5164c40d0f55SPeter Brune 5165c40d0f55SPeter Brune Output Parameter: 5166c40d0f55SPeter Brune . side - the preconditioning side, where side is one of 5167c40d0f55SPeter Brune .vb 51682d547940SBarry Smith PC_LEFT - left preconditioning 51692d547940SBarry Smith PC_RIGHT - right preconditioning (default for most nonlinear solvers) 5170c40d0f55SPeter Brune .ve 5171c40d0f55SPeter Brune 5172c40d0f55SPeter Brune Options Database Keys: 5173c40d0f55SPeter Brune . -snes_pc_side <right,left> 5174c40d0f55SPeter Brune 51752d547940SBarry Smith Notes: SNESNRICHARDSON and SNESNCG only support left preconditioning. 51762d547940SBarry Smith 5177c40d0f55SPeter Brune Level: intermediate 5178c40d0f55SPeter Brune 5179c40d0f55SPeter Brune .keywords: SNES, set, right, left, side, preconditioner, flag 5180c40d0f55SPeter Brune 5181be95d8f1SBarry Smith .seealso: SNESGetNPCSide(), KSPSetPCSide() 5182c40d0f55SPeter Brune @*/ 5183be95d8f1SBarry Smith PetscErrorCode SNESSetNPCSide(SNES snes,PCSide side) 5184c40d0f55SPeter Brune { 5185c40d0f55SPeter Brune PetscFunctionBegin; 5186c40d0f55SPeter Brune PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 5187c40d0f55SPeter Brune PetscValidLogicalCollectiveEnum(snes,side,2); 5188efd4aadfSBarry Smith snes->npcside= side; 5189c40d0f55SPeter Brune PetscFunctionReturn(0); 5190c40d0f55SPeter Brune } 5191c40d0f55SPeter Brune 5192c40d0f55SPeter Brune /*@ 5193be95d8f1SBarry Smith SNESGetNPCSide - Gets the preconditioning side. 5194c40d0f55SPeter Brune 5195c40d0f55SPeter Brune Not Collective 5196c40d0f55SPeter Brune 5197c40d0f55SPeter Brune Input Parameter: 5198c40d0f55SPeter Brune . snes - iterative context obtained from SNESCreate() 5199c40d0f55SPeter Brune 5200c40d0f55SPeter Brune Output Parameter: 5201c40d0f55SPeter Brune . side - the preconditioning side, where side is one of 5202c40d0f55SPeter Brune .vb 52032d547940SBarry Smith PC_LEFT - left preconditioning 52042d547940SBarry Smith PC_RIGHT - right preconditioning (default for most nonlinear solvers) 5205c40d0f55SPeter Brune .ve 5206c40d0f55SPeter Brune 5207c40d0f55SPeter Brune Level: intermediate 5208c40d0f55SPeter Brune 5209c40d0f55SPeter Brune .keywords: SNES, get, right, left, side, preconditioner, flag 5210c40d0f55SPeter Brune 5211be95d8f1SBarry Smith .seealso: SNESSetNPCSide(), KSPGetPCSide() 5212c40d0f55SPeter Brune @*/ 5213be95d8f1SBarry Smith PetscErrorCode SNESGetNPCSide(SNES snes,PCSide *side) 5214c40d0f55SPeter Brune { 5215c40d0f55SPeter Brune PetscFunctionBegin; 5216c40d0f55SPeter Brune PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 5217c40d0f55SPeter Brune PetscValidPointer(side,2); 5218efd4aadfSBarry Smith *side = snes->npcside; 5219c40d0f55SPeter Brune PetscFunctionReturn(0); 5220c40d0f55SPeter Brune } 5221c40d0f55SPeter Brune 52229e764e56SPeter Brune /*@ 52237601faf0SJed Brown SNESSetLineSearch - Sets the linesearch on the SNES instance. 52249e764e56SPeter Brune 52259e764e56SPeter Brune Collective on SNES 52269e764e56SPeter Brune 52279e764e56SPeter Brune Input Parameters: 52289e764e56SPeter Brune + snes - iterative context obtained from SNESCreate() 52299e764e56SPeter Brune - linesearch - the linesearch object 52309e764e56SPeter Brune 52319e764e56SPeter Brune Notes: 52327601faf0SJed Brown Use SNESGetLineSearch() to retrieve the preconditioner context (for example, 52339e764e56SPeter Brune to configure it using the API). 52349e764e56SPeter Brune 52359e764e56SPeter Brune Level: developer 52369e764e56SPeter Brune 52379e764e56SPeter Brune .keywords: SNES, set, linesearch 52387601faf0SJed Brown .seealso: SNESGetLineSearch() 52399e764e56SPeter Brune @*/ 52407601faf0SJed Brown PetscErrorCode SNESSetLineSearch(SNES snes, SNESLineSearch linesearch) 52419e764e56SPeter Brune { 52429e764e56SPeter Brune PetscErrorCode ierr; 52439e764e56SPeter Brune 52449e764e56SPeter Brune PetscFunctionBegin; 52459e764e56SPeter Brune PetscValidHeaderSpecific(snes, SNES_CLASSID, 1); 5246f1c6b773SPeter Brune PetscValidHeaderSpecific(linesearch, SNESLINESEARCH_CLASSID, 2); 52479e764e56SPeter Brune PetscCheckSameComm(snes, 1, linesearch, 2); 52489e764e56SPeter Brune ierr = PetscObjectReference((PetscObject) linesearch);CHKERRQ(ierr); 5249f1c6b773SPeter Brune ierr = SNESLineSearchDestroy(&snes->linesearch);CHKERRQ(ierr); 5250f5af7f23SKarl Rupp 52519e764e56SPeter Brune snes->linesearch = linesearch; 5252f5af7f23SKarl Rupp 52533bb1ff40SBarry Smith ierr = PetscLogObjectParent((PetscObject)snes, (PetscObject)snes->linesearch);CHKERRQ(ierr); 52549e764e56SPeter Brune PetscFunctionReturn(0); 52559e764e56SPeter Brune } 52569e764e56SPeter Brune 5257a34ceb2aSJed Brown /*@ 52587601faf0SJed Brown SNESGetLineSearch - Returns a pointer to the line search context set with SNESSetLineSearch() 52598141a3b9SPeter Brune or creates a default line search instance associated with the SNES and returns it. 52609e764e56SPeter Brune 52619e764e56SPeter Brune Not Collective 52629e764e56SPeter Brune 52639e764e56SPeter Brune Input Parameter: 52649e764e56SPeter Brune . snes - iterative context obtained from SNESCreate() 52659e764e56SPeter Brune 52669e764e56SPeter Brune Output Parameter: 52679e764e56SPeter Brune . linesearch - linesearch context 52689e764e56SPeter Brune 5269162e0bf5SPeter Brune Level: beginner 52709e764e56SPeter Brune 52719e764e56SPeter Brune .keywords: SNES, get, linesearch 5272162e0bf5SPeter Brune .seealso: SNESSetLineSearch(), SNESLineSearchCreate() 52739e764e56SPeter Brune @*/ 52747601faf0SJed Brown PetscErrorCode SNESGetLineSearch(SNES snes, SNESLineSearch *linesearch) 52759e764e56SPeter Brune { 52769e764e56SPeter Brune PetscErrorCode ierr; 52779e764e56SPeter Brune const char *optionsprefix; 52789e764e56SPeter Brune 52799e764e56SPeter Brune PetscFunctionBegin; 52809e764e56SPeter Brune PetscValidHeaderSpecific(snes, SNES_CLASSID, 1); 52819e764e56SPeter Brune PetscValidPointer(linesearch, 2); 52829e764e56SPeter Brune if (!snes->linesearch) { 52839e764e56SPeter Brune ierr = SNESGetOptionsPrefix(snes, &optionsprefix);CHKERRQ(ierr); 528482f516ccSBarry Smith ierr = SNESLineSearchCreate(PetscObjectComm((PetscObject)snes), &snes->linesearch);CHKERRQ(ierr); 5285f1c6b773SPeter Brune ierr = SNESLineSearchSetSNES(snes->linesearch, snes);CHKERRQ(ierr); 5286b1dca40bSPeter Brune ierr = SNESLineSearchAppendOptionsPrefix(snes->linesearch, optionsprefix);CHKERRQ(ierr); 52879e764e56SPeter Brune ierr = PetscObjectIncrementTabLevel((PetscObject) snes->linesearch, (PetscObject) snes, 1);CHKERRQ(ierr); 52883bb1ff40SBarry Smith ierr = PetscLogObjectParent((PetscObject)snes, (PetscObject)snes->linesearch);CHKERRQ(ierr); 52899e764e56SPeter Brune } 52909e764e56SPeter Brune *linesearch = snes->linesearch; 52919e764e56SPeter Brune PetscFunctionReturn(0); 52929e764e56SPeter Brune } 52939e764e56SPeter Brune 529469b4f73cSBarry Smith #if defined(PETSC_HAVE_MATLAB_ENGINE) 5295c6db04a5SJed Brown #include <mex.h> 529669b4f73cSBarry Smith 52978f6e6473SBarry Smith typedef struct {char *funcname; mxArray *ctx;} SNESMatlabContext; 52988f6e6473SBarry Smith 52990807856dSBarry Smith /* 5300bf388a1fSBarry Smith SNESComputeFunction_Matlab - Calls the function that has been set with SNESSetFunctionMatlab(). 53010807856dSBarry Smith 53020807856dSBarry Smith Collective on SNES 53030807856dSBarry Smith 53040807856dSBarry Smith Input Parameters: 53050807856dSBarry Smith + snes - the SNES context 53060807856dSBarry Smith - x - input vector 53070807856dSBarry Smith 53080807856dSBarry Smith Output Parameter: 53090807856dSBarry Smith . y - function vector, as set by SNESSetFunction() 53100807856dSBarry Smith 53110807856dSBarry Smith Notes: 53120807856dSBarry Smith SNESComputeFunction() is typically used within nonlinear solvers 53130807856dSBarry Smith implementations, so most users would not generally call this routine 53140807856dSBarry Smith themselves. 53150807856dSBarry Smith 53160807856dSBarry Smith Level: developer 53170807856dSBarry Smith 53180807856dSBarry Smith .keywords: SNES, nonlinear, compute, function 53190807856dSBarry Smith 53200807856dSBarry Smith .seealso: SNESSetFunction(), SNESGetFunction() 532161b2408cSBarry Smith */ 53227087cfbeSBarry Smith PetscErrorCode SNESComputeFunction_Matlab(SNES snes,Vec x,Vec y, void *ctx) 53230807856dSBarry Smith { 5324e650e774SBarry Smith PetscErrorCode ierr; 53258f6e6473SBarry Smith SNESMatlabContext *sctx = (SNESMatlabContext*)ctx; 53268f6e6473SBarry Smith int nlhs = 1,nrhs = 5; 53278f6e6473SBarry Smith mxArray *plhs[1],*prhs[5]; 532891621f2eSBarry Smith long long int lx = 0,ly = 0,ls = 0; 5329e650e774SBarry Smith 53300807856dSBarry Smith PetscFunctionBegin; 53310807856dSBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 53320807856dSBarry Smith PetscValidHeaderSpecific(x,VEC_CLASSID,2); 53330807856dSBarry Smith PetscValidHeaderSpecific(y,VEC_CLASSID,3); 53340807856dSBarry Smith PetscCheckSameComm(snes,1,x,2); 53350807856dSBarry Smith PetscCheckSameComm(snes,1,y,3); 53360807856dSBarry Smith 53370807856dSBarry Smith /* call Matlab function in ctx with arguments x and y */ 5338e650e774SBarry Smith 533991621f2eSBarry Smith ierr = PetscMemcpy(&ls,&snes,sizeof(snes));CHKERRQ(ierr); 5340e650e774SBarry Smith ierr = PetscMemcpy(&lx,&x,sizeof(x));CHKERRQ(ierr); 5341e650e774SBarry Smith ierr = PetscMemcpy(&ly,&y,sizeof(x));CHKERRQ(ierr); 534291621f2eSBarry Smith prhs[0] = mxCreateDoubleScalar((double)ls); 534391621f2eSBarry Smith prhs[1] = mxCreateDoubleScalar((double)lx); 534491621f2eSBarry Smith prhs[2] = mxCreateDoubleScalar((double)ly); 53458f6e6473SBarry Smith prhs[3] = mxCreateString(sctx->funcname); 53468f6e6473SBarry Smith prhs[4] = sctx->ctx; 5347b807a863SBarry Smith ierr = mexCallMATLAB(nlhs,plhs,nrhs,prhs,"PetscSNESComputeFunctionInternal");CHKERRQ(ierr); 5348e650e774SBarry Smith ierr = mxGetScalar(plhs[0]);CHKERRQ(ierr); 5349e650e774SBarry Smith mxDestroyArray(prhs[0]); 5350e650e774SBarry Smith mxDestroyArray(prhs[1]); 5351e650e774SBarry Smith mxDestroyArray(prhs[2]); 53528f6e6473SBarry Smith mxDestroyArray(prhs[3]); 5353e650e774SBarry Smith mxDestroyArray(plhs[0]); 53540807856dSBarry Smith PetscFunctionReturn(0); 53550807856dSBarry Smith } 53560807856dSBarry Smith 535761b2408cSBarry Smith /* 53580807856dSBarry Smith SNESSetFunctionMatlab - Sets the function evaluation routine and function 53590807856dSBarry Smith vector for use by the SNES routines in solving systems of nonlinear 5360e3c5b3baSBarry Smith equations from MATLAB. Here the function is a string containing the name of a MATLAB function 53610807856dSBarry Smith 53620807856dSBarry Smith Logically Collective on SNES 53630807856dSBarry Smith 53640807856dSBarry Smith Input Parameters: 53650807856dSBarry Smith + snes - the SNES context 53660807856dSBarry Smith . r - vector to store function value 5367f8b49ee9SBarry Smith - f - function evaluation routine 53680807856dSBarry Smith 53690807856dSBarry Smith Notes: 53700807856dSBarry Smith The Newton-like methods typically solve linear systems of the form 53710807856dSBarry Smith $ f'(x) x = -f(x), 53720807856dSBarry Smith where f'(x) denotes the Jacobian matrix and f(x) is the function. 53730807856dSBarry Smith 53740807856dSBarry Smith Level: beginner 53750807856dSBarry Smith 5376c5b75c40SBarry Smith Developer Note: This bleeds the allocated memory SNESMatlabContext *sctx; 5377c5b75c40SBarry Smith 53780807856dSBarry Smith .keywords: SNES, nonlinear, set, function 53790807856dSBarry Smith 53800807856dSBarry Smith .seealso: SNESGetFunction(), SNESComputeFunction(), SNESSetJacobian(), SNESSetFunction() 538161b2408cSBarry Smith */ 5382f8b49ee9SBarry Smith PetscErrorCode SNESSetFunctionMatlab(SNES snes,Vec r,const char *f,mxArray *ctx) 53830807856dSBarry Smith { 53840807856dSBarry Smith PetscErrorCode ierr; 53858f6e6473SBarry Smith SNESMatlabContext *sctx; 53860807856dSBarry Smith 53870807856dSBarry Smith PetscFunctionBegin; 53888f6e6473SBarry Smith /* currently sctx is memory bleed */ 5389854ce69bSBarry Smith ierr = PetscNew(&sctx);CHKERRQ(ierr); 5390f8b49ee9SBarry Smith ierr = PetscStrallocpy(f,&sctx->funcname);CHKERRQ(ierr); 53918f6e6473SBarry Smith /* 53928f6e6473SBarry Smith This should work, but it doesn't 53938f6e6473SBarry Smith sctx->ctx = ctx; 53948f6e6473SBarry Smith mexMakeArrayPersistent(sctx->ctx); 53958f6e6473SBarry Smith */ 53968f6e6473SBarry Smith sctx->ctx = mxDuplicateArray(ctx); 53978f6e6473SBarry Smith ierr = SNESSetFunction(snes,r,SNESComputeFunction_Matlab,sctx);CHKERRQ(ierr); 53980807856dSBarry Smith PetscFunctionReturn(0); 53990807856dSBarry Smith } 540069b4f73cSBarry Smith 540161b2408cSBarry Smith /* 5402bf388a1fSBarry Smith SNESComputeJacobian_Matlab - Calls the function that has been set with SNESSetJacobianMatlab(). 540361b2408cSBarry Smith 540461b2408cSBarry Smith Collective on SNES 540561b2408cSBarry Smith 540661b2408cSBarry Smith Input Parameters: 540761b2408cSBarry Smith + snes - the SNES context 540861b2408cSBarry Smith . x - input vector 540961b2408cSBarry Smith . A, B - the matrices 541061b2408cSBarry Smith - ctx - user context 541161b2408cSBarry Smith 541261b2408cSBarry Smith Level: developer 541361b2408cSBarry Smith 541461b2408cSBarry Smith .keywords: SNES, nonlinear, compute, function 541561b2408cSBarry Smith 541661b2408cSBarry Smith .seealso: SNESSetFunction(), SNESGetFunction() 541761b2408cSBarry Smith @*/ 5418f3229a78SSatish Balay PetscErrorCode SNESComputeJacobian_Matlab(SNES snes,Vec x,Mat A,Mat B,void *ctx) 541961b2408cSBarry Smith { 542061b2408cSBarry Smith PetscErrorCode ierr; 542161b2408cSBarry Smith SNESMatlabContext *sctx = (SNESMatlabContext*)ctx; 542261b2408cSBarry Smith int nlhs = 2,nrhs = 6; 542361b2408cSBarry Smith mxArray *plhs[2],*prhs[6]; 542461b2408cSBarry Smith long long int lx = 0,lA = 0,ls = 0, lB = 0; 542561b2408cSBarry Smith 542661b2408cSBarry Smith PetscFunctionBegin; 542761b2408cSBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 542861b2408cSBarry Smith PetscValidHeaderSpecific(x,VEC_CLASSID,2); 542961b2408cSBarry Smith 543061b2408cSBarry Smith /* call Matlab function in ctx with arguments x and y */ 543161b2408cSBarry Smith 543261b2408cSBarry Smith ierr = PetscMemcpy(&ls,&snes,sizeof(snes));CHKERRQ(ierr); 543361b2408cSBarry Smith ierr = PetscMemcpy(&lx,&x,sizeof(x));CHKERRQ(ierr); 543461b2408cSBarry Smith ierr = PetscMemcpy(&lA,A,sizeof(x));CHKERRQ(ierr); 543561b2408cSBarry Smith ierr = PetscMemcpy(&lB,B,sizeof(x));CHKERRQ(ierr); 543661b2408cSBarry Smith prhs[0] = mxCreateDoubleScalar((double)ls); 543761b2408cSBarry Smith prhs[1] = mxCreateDoubleScalar((double)lx); 543861b2408cSBarry Smith prhs[2] = mxCreateDoubleScalar((double)lA); 543961b2408cSBarry Smith prhs[3] = mxCreateDoubleScalar((double)lB); 544061b2408cSBarry Smith prhs[4] = mxCreateString(sctx->funcname); 544161b2408cSBarry Smith prhs[5] = sctx->ctx; 5442b807a863SBarry Smith ierr = mexCallMATLAB(nlhs,plhs,nrhs,prhs,"PetscSNESComputeJacobianInternal");CHKERRQ(ierr); 544361b2408cSBarry Smith ierr = mxGetScalar(plhs[0]);CHKERRQ(ierr); 544461b2408cSBarry Smith mxDestroyArray(prhs[0]); 544561b2408cSBarry Smith mxDestroyArray(prhs[1]); 544661b2408cSBarry Smith mxDestroyArray(prhs[2]); 544761b2408cSBarry Smith mxDestroyArray(prhs[3]); 544861b2408cSBarry Smith mxDestroyArray(prhs[4]); 544961b2408cSBarry Smith mxDestroyArray(plhs[0]); 545061b2408cSBarry Smith mxDestroyArray(plhs[1]); 545161b2408cSBarry Smith PetscFunctionReturn(0); 545261b2408cSBarry Smith } 545361b2408cSBarry Smith 545461b2408cSBarry Smith /* 545561b2408cSBarry Smith SNESSetJacobianMatlab - Sets the Jacobian function evaluation routine and two empty Jacobian matrices 545661b2408cSBarry Smith vector for use by the SNES routines in solving systems of nonlinear 5457e3c5b3baSBarry Smith equations from MATLAB. Here the function is a string containing the name of a MATLAB function 545861b2408cSBarry Smith 545961b2408cSBarry Smith Logically Collective on SNES 546061b2408cSBarry Smith 546161b2408cSBarry Smith Input Parameters: 546261b2408cSBarry Smith + snes - the SNES context 546361b2408cSBarry Smith . A,B - Jacobian matrices 5464f8b49ee9SBarry Smith . J - function evaluation routine 546561b2408cSBarry Smith - ctx - user context 546661b2408cSBarry Smith 546761b2408cSBarry Smith Level: developer 546861b2408cSBarry Smith 5469c5b75c40SBarry Smith Developer Note: This bleeds the allocated memory SNESMatlabContext *sctx; 5470c5b75c40SBarry Smith 547161b2408cSBarry Smith .keywords: SNES, nonlinear, set, function 547261b2408cSBarry Smith 5473f8b49ee9SBarry Smith .seealso: SNESGetFunction(), SNESComputeFunction(), SNESSetJacobian(), SNESSetFunction(), J 547461b2408cSBarry Smith */ 5475f8b49ee9SBarry Smith PetscErrorCode SNESSetJacobianMatlab(SNES snes,Mat A,Mat B,const char *J,mxArray *ctx) 547661b2408cSBarry Smith { 547761b2408cSBarry Smith PetscErrorCode ierr; 547861b2408cSBarry Smith SNESMatlabContext *sctx; 547961b2408cSBarry Smith 548061b2408cSBarry Smith PetscFunctionBegin; 548161b2408cSBarry Smith /* currently sctx is memory bleed */ 5482854ce69bSBarry Smith ierr = PetscNew(&sctx);CHKERRQ(ierr); 5483f8b49ee9SBarry Smith ierr = PetscStrallocpy(J,&sctx->funcname);CHKERRQ(ierr); 548461b2408cSBarry Smith /* 548561b2408cSBarry Smith This should work, but it doesn't 548661b2408cSBarry Smith sctx->ctx = ctx; 548761b2408cSBarry Smith mexMakeArrayPersistent(sctx->ctx); 548861b2408cSBarry Smith */ 548961b2408cSBarry Smith sctx->ctx = mxDuplicateArray(ctx); 549061b2408cSBarry Smith ierr = SNESSetJacobian(snes,A,B,SNESComputeJacobian_Matlab,sctx);CHKERRQ(ierr); 549161b2408cSBarry Smith PetscFunctionReturn(0); 549261b2408cSBarry Smith } 549369b4f73cSBarry Smith 5494f9eb7ae2SShri Abhyankar /* 5495f9eb7ae2SShri Abhyankar SNESMonitor_Matlab - Calls the function that has been set with SNESMonitorSetMatlab(). 5496f9eb7ae2SShri Abhyankar 5497f9eb7ae2SShri Abhyankar Collective on SNES 5498f9eb7ae2SShri Abhyankar 5499f9eb7ae2SShri Abhyankar .seealso: SNESSetFunction(), SNESGetFunction() 5500f9eb7ae2SShri Abhyankar @*/ 55017087cfbeSBarry Smith PetscErrorCode SNESMonitor_Matlab(SNES snes,PetscInt it, PetscReal fnorm, void *ctx) 5502f9eb7ae2SShri Abhyankar { 5503f9eb7ae2SShri Abhyankar PetscErrorCode ierr; 550448f37e70SShri Abhyankar SNESMatlabContext *sctx = (SNESMatlabContext*)ctx; 5505f9eb7ae2SShri Abhyankar int nlhs = 1,nrhs = 6; 5506f9eb7ae2SShri Abhyankar mxArray *plhs[1],*prhs[6]; 5507f9eb7ae2SShri Abhyankar long long int lx = 0,ls = 0; 5508f9eb7ae2SShri Abhyankar Vec x = snes->vec_sol; 5509f9eb7ae2SShri Abhyankar 5510f9eb7ae2SShri Abhyankar PetscFunctionBegin; 5511f9eb7ae2SShri Abhyankar PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 5512f9eb7ae2SShri Abhyankar 5513f9eb7ae2SShri Abhyankar ierr = PetscMemcpy(&ls,&snes,sizeof(snes));CHKERRQ(ierr); 5514f9eb7ae2SShri Abhyankar ierr = PetscMemcpy(&lx,&x,sizeof(x));CHKERRQ(ierr); 5515f9eb7ae2SShri Abhyankar prhs[0] = mxCreateDoubleScalar((double)ls); 5516f9eb7ae2SShri Abhyankar prhs[1] = mxCreateDoubleScalar((double)it); 5517f9eb7ae2SShri Abhyankar prhs[2] = mxCreateDoubleScalar((double)fnorm); 5518f9eb7ae2SShri Abhyankar prhs[3] = mxCreateDoubleScalar((double)lx); 5519f9eb7ae2SShri Abhyankar prhs[4] = mxCreateString(sctx->funcname); 5520f9eb7ae2SShri Abhyankar prhs[5] = sctx->ctx; 5521f9eb7ae2SShri Abhyankar ierr = mexCallMATLAB(nlhs,plhs,nrhs,prhs,"PetscSNESMonitorInternal");CHKERRQ(ierr); 5522f9eb7ae2SShri Abhyankar ierr = mxGetScalar(plhs[0]);CHKERRQ(ierr); 5523f9eb7ae2SShri Abhyankar mxDestroyArray(prhs[0]); 5524f9eb7ae2SShri Abhyankar mxDestroyArray(prhs[1]); 5525f9eb7ae2SShri Abhyankar mxDestroyArray(prhs[2]); 5526f9eb7ae2SShri Abhyankar mxDestroyArray(prhs[3]); 5527f9eb7ae2SShri Abhyankar mxDestroyArray(prhs[4]); 5528f9eb7ae2SShri Abhyankar mxDestroyArray(plhs[0]); 5529f9eb7ae2SShri Abhyankar PetscFunctionReturn(0); 5530f9eb7ae2SShri Abhyankar } 5531f9eb7ae2SShri Abhyankar 5532f9eb7ae2SShri Abhyankar /* 5533e3c5b3baSBarry Smith SNESMonitorSetMatlab - Sets the monitor function from MATLAB 5534f9eb7ae2SShri Abhyankar 5535f9eb7ae2SShri Abhyankar Level: developer 5536f9eb7ae2SShri Abhyankar 5537c5b75c40SBarry Smith Developer Note: This bleeds the allocated memory SNESMatlabContext *sctx; 5538c5b75c40SBarry Smith 5539f9eb7ae2SShri Abhyankar .keywords: SNES, nonlinear, set, function 5540f9eb7ae2SShri Abhyankar 5541f9eb7ae2SShri Abhyankar .seealso: SNESGetFunction(), SNESComputeFunction(), SNESSetJacobian(), SNESSetFunction() 5542f9eb7ae2SShri Abhyankar */ 55436e4dcb14SBarry Smith PetscErrorCode SNESMonitorSetMatlab(SNES snes,const char *f,mxArray *ctx) 5544f9eb7ae2SShri Abhyankar { 5545f9eb7ae2SShri Abhyankar PetscErrorCode ierr; 5546f9eb7ae2SShri Abhyankar SNESMatlabContext *sctx; 5547f9eb7ae2SShri Abhyankar 5548f9eb7ae2SShri Abhyankar PetscFunctionBegin; 5549f9eb7ae2SShri Abhyankar /* currently sctx is memory bleed */ 5550854ce69bSBarry Smith ierr = PetscNew(&sctx);CHKERRQ(ierr); 55516e4dcb14SBarry Smith ierr = PetscStrallocpy(f,&sctx->funcname);CHKERRQ(ierr); 5552f9eb7ae2SShri Abhyankar /* 5553f9eb7ae2SShri Abhyankar This should work, but it doesn't 5554f9eb7ae2SShri Abhyankar sctx->ctx = ctx; 5555f9eb7ae2SShri Abhyankar mexMakeArrayPersistent(sctx->ctx); 5556f9eb7ae2SShri Abhyankar */ 5557f9eb7ae2SShri Abhyankar sctx->ctx = mxDuplicateArray(ctx); 55580298fd71SBarry Smith ierr = SNESMonitorSet(snes,SNESMonitor_Matlab,sctx,NULL);CHKERRQ(ierr); 5559f9eb7ae2SShri Abhyankar PetscFunctionReturn(0); 5560f9eb7ae2SShri Abhyankar } 5561f9eb7ae2SShri Abhyankar 556269b4f73cSBarry Smith #endif 5563