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> 606fc46c8SMatthew G. Knepley #include <petscconvest.h> 79b94acceSBarry Smith 8ace3abfcSBarry Smith PetscBool SNESRegisterAllCalled = PETSC_FALSE; 90298fd71SBarry Smith PetscFunctionList SNESList = NULL; 108ba1e511SMatthew Knepley 118ba1e511SMatthew Knepley /* Logging support */ 1222c6f798SBarry Smith PetscClassId SNES_CLASSID, DMSNES_CLASSID; 1394db00ebSBarry Smith PetscLogEvent SNES_Solve, SNES_FunctionEval, SNES_JacobianEval, SNES_NGSEval, SNES_NGSFuncEval, SNES_NPCSolve, SNES_ObjectiveEval; 14a09944afSBarry Smith 15e113a28aSBarry Smith /*@ 16e113a28aSBarry Smith SNESSetErrorIfNotConverged - Causes SNESSolve() to generate an error if the solver has not converged. 17e113a28aSBarry Smith 183f9fe445SBarry Smith Logically Collective on SNES 19e113a28aSBarry Smith 20e113a28aSBarry Smith Input Parameters: 21e113a28aSBarry Smith + snes - iterative context obtained from SNESCreate() 22e113a28aSBarry Smith - flg - PETSC_TRUE indicates you want the error generated 23e113a28aSBarry Smith 24e113a28aSBarry Smith Options database keys: 25e113a28aSBarry Smith . -snes_error_if_not_converged : this takes an optional truth value (0/1/no/yes/true/false) 26e113a28aSBarry Smith 27e113a28aSBarry Smith Level: intermediate 28e113a28aSBarry Smith 29e113a28aSBarry Smith Notes: 30e113a28aSBarry Smith Normally PETSc continues if a linear solver fails to converge, you can call SNESGetConvergedReason() after a SNESSolve() 31e113a28aSBarry Smith to determine if it has converged. 32e113a28aSBarry Smith 33e113a28aSBarry Smith .keywords: SNES, set, initial guess, nonzero 34e113a28aSBarry Smith 35e113a28aSBarry Smith .seealso: SNESGetErrorIfNotConverged(), KSPGetErrorIfNotConverged(), KSPSetErrorIFNotConverged() 36e113a28aSBarry Smith @*/ 377087cfbeSBarry Smith PetscErrorCode SNESSetErrorIfNotConverged(SNES snes,PetscBool flg) 38e113a28aSBarry Smith { 39e113a28aSBarry Smith PetscFunctionBegin; 40e113a28aSBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 41acfcf0e5SJed Brown PetscValidLogicalCollectiveBool(snes,flg,2); 42e113a28aSBarry Smith snes->errorifnotconverged = flg; 43e113a28aSBarry Smith PetscFunctionReturn(0); 44e113a28aSBarry Smith } 45e113a28aSBarry Smith 46e113a28aSBarry Smith /*@ 47e113a28aSBarry Smith SNESGetErrorIfNotConverged - Will SNESSolve() generate an error if the solver does not converge? 48e113a28aSBarry Smith 49e113a28aSBarry Smith Not Collective 50e113a28aSBarry Smith 51e113a28aSBarry Smith Input Parameter: 52e113a28aSBarry Smith . snes - iterative context obtained from SNESCreate() 53e113a28aSBarry Smith 54e113a28aSBarry Smith Output Parameter: 55e113a28aSBarry Smith . flag - PETSC_TRUE if it will generate an error, else PETSC_FALSE 56e113a28aSBarry Smith 57e113a28aSBarry Smith Level: intermediate 58e113a28aSBarry Smith 59e113a28aSBarry Smith .keywords: SNES, set, initial guess, nonzero 60e113a28aSBarry Smith 61e113a28aSBarry Smith .seealso: SNESSetErrorIfNotConverged(), KSPGetErrorIfNotConverged(), KSPSetErrorIFNotConverged() 62e113a28aSBarry Smith @*/ 637087cfbeSBarry Smith PetscErrorCode SNESGetErrorIfNotConverged(SNES snes,PetscBool *flag) 64e113a28aSBarry Smith { 65e113a28aSBarry Smith PetscFunctionBegin; 66e113a28aSBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 67e113a28aSBarry Smith PetscValidPointer(flag,2); 68e113a28aSBarry Smith *flag = snes->errorifnotconverged; 69e113a28aSBarry Smith PetscFunctionReturn(0); 70e113a28aSBarry Smith } 71e113a28aSBarry Smith 724fc747eaSLawrence Mitchell /*@ 734fc747eaSLawrence Mitchell SNESSetAlwaysComputesFinalResidual - does the SNES always compute the residual at the final solution? 744fc747eaSLawrence Mitchell 754fc747eaSLawrence Mitchell Logically Collective on SNES 764fc747eaSLawrence Mitchell 774fc747eaSLawrence Mitchell Input Parameters: 784fc747eaSLawrence Mitchell + snes - the shell SNES 794fc747eaSLawrence Mitchell - flg - is the residual computed? 804fc747eaSLawrence Mitchell 814fc747eaSLawrence Mitchell Level: advanced 824fc747eaSLawrence Mitchell 834fc747eaSLawrence Mitchell .seealso: SNESGetAlwaysComputesFinalResidual() 844fc747eaSLawrence Mitchell @*/ 854fc747eaSLawrence Mitchell PetscErrorCode SNESSetAlwaysComputesFinalResidual(SNES snes, PetscBool flg) 864fc747eaSLawrence Mitchell { 874fc747eaSLawrence Mitchell PetscFunctionBegin; 884fc747eaSLawrence Mitchell PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 894fc747eaSLawrence Mitchell snes->alwayscomputesfinalresidual = flg; 904fc747eaSLawrence Mitchell PetscFunctionReturn(0); 914fc747eaSLawrence Mitchell } 924fc747eaSLawrence Mitchell 934fc747eaSLawrence Mitchell /*@ 944fc747eaSLawrence Mitchell SNESGetAlwaysComputesFinalResidual - does the SNES always compute the residual at the final solution? 954fc747eaSLawrence Mitchell 964fc747eaSLawrence Mitchell Logically Collective on SNES 974fc747eaSLawrence Mitchell 984fc747eaSLawrence Mitchell Input Parameter: 994fc747eaSLawrence Mitchell . snes - the shell SNES 1004fc747eaSLawrence Mitchell 1014fc747eaSLawrence Mitchell Output Parameter: 1024fc747eaSLawrence Mitchell . flg - is the residual computed? 1034fc747eaSLawrence Mitchell 1044fc747eaSLawrence Mitchell Level: advanced 1054fc747eaSLawrence Mitchell 1064fc747eaSLawrence Mitchell .seealso: SNESSetAlwaysComputesFinalResidual() 1074fc747eaSLawrence Mitchell @*/ 1084fc747eaSLawrence Mitchell PetscErrorCode SNESGetAlwaysComputesFinalResidual(SNES snes, PetscBool *flg) 1094fc747eaSLawrence Mitchell { 1104fc747eaSLawrence Mitchell PetscFunctionBegin; 1114fc747eaSLawrence Mitchell PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 1124fc747eaSLawrence Mitchell *flg = snes->alwayscomputesfinalresidual; 1134fc747eaSLawrence Mitchell PetscFunctionReturn(0); 1144fc747eaSLawrence Mitchell } 1154fc747eaSLawrence Mitchell 116e725d27bSBarry Smith /*@ 117bf388a1fSBarry Smith SNESSetFunctionDomainError - tells SNES that the input vector to your SNESFunction is not 1184936397dSBarry Smith in the functions domain. For example, negative pressure. 1194936397dSBarry Smith 1203f9fe445SBarry Smith Logically Collective on SNES 1214936397dSBarry Smith 1224936397dSBarry Smith Input Parameters: 1236a388c36SPeter Brune . snes - the SNES context 1244936397dSBarry Smith 12528529972SSatish Balay Level: advanced 1264936397dSBarry Smith 1274936397dSBarry Smith .keywords: SNES, view 1284936397dSBarry Smith 129bf388a1fSBarry Smith .seealso: SNESCreate(), SNESSetFunction(), SNESFunction 1304936397dSBarry Smith @*/ 1317087cfbeSBarry Smith PetscErrorCode SNESSetFunctionDomainError(SNES snes) 1324936397dSBarry Smith { 1334936397dSBarry Smith PetscFunctionBegin; 1340700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 135422a814eSBarry Smith if (snes->errorifnotconverged) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"User code indicates input vector is not in the function domain"); 1364936397dSBarry Smith snes->domainerror = PETSC_TRUE; 1374936397dSBarry Smith PetscFunctionReturn(0); 1384936397dSBarry Smith } 1394936397dSBarry Smith 1406a388c36SPeter Brune /*@ 141c77b2880SPeter Brune SNESGetFunctionDomainError - Gets the status of the domain error after a call to SNESComputeFunction; 1426a388c36SPeter Brune 1436a388c36SPeter Brune Logically Collective on SNES 1446a388c36SPeter Brune 1456a388c36SPeter Brune Input Parameters: 1466a388c36SPeter Brune . snes - the SNES context 1476a388c36SPeter Brune 1486a388c36SPeter Brune Output Parameters: 149bf388a1fSBarry Smith . domainerror - Set to PETSC_TRUE if there's a domain error; PETSC_FALSE otherwise. 1506a388c36SPeter Brune 1516a388c36SPeter Brune Level: advanced 1526a388c36SPeter Brune 1536a388c36SPeter Brune .keywords: SNES, view 1546a388c36SPeter Brune 155bf388a1fSBarry Smith .seealso: SNESSetFunctionDomainError(), SNESComputeFunction() 1566a388c36SPeter Brune @*/ 1576a388c36SPeter Brune PetscErrorCode SNESGetFunctionDomainError(SNES snes, PetscBool *domainerror) 1586a388c36SPeter Brune { 1596a388c36SPeter Brune PetscFunctionBegin; 1606a388c36SPeter Brune PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 1616a388c36SPeter Brune PetscValidPointer(domainerror, 2); 1626a388c36SPeter Brune *domainerror = snes->domainerror; 1636a388c36SPeter Brune PetscFunctionReturn(0); 1646a388c36SPeter Brune } 1656a388c36SPeter Brune 16655849f57SBarry Smith /*@C 16755849f57SBarry Smith SNESLoad - Loads a SNES that has been stored in binary with SNESView(). 16855849f57SBarry Smith 16955849f57SBarry Smith Collective on PetscViewer 17055849f57SBarry Smith 17155849f57SBarry Smith Input Parameters: 17255849f57SBarry Smith + newdm - the newly loaded SNES, this needs to have been created with SNESCreate() or 17355849f57SBarry Smith some related function before a call to SNESLoad(). 17455849f57SBarry Smith - viewer - binary file viewer, obtained from PetscViewerBinaryOpen() 17555849f57SBarry Smith 17655849f57SBarry Smith Level: intermediate 17755849f57SBarry Smith 17855849f57SBarry Smith Notes: 17955849f57SBarry Smith The type is determined by the data in the file, any type set into the SNES before this call is ignored. 18055849f57SBarry Smith 18155849f57SBarry Smith Notes for advanced users: 18255849f57SBarry Smith Most users should not need to know the details of the binary storage 18355849f57SBarry Smith format, since SNESLoad() and TSView() completely hide these details. 18455849f57SBarry Smith But for anyone who's interested, the standard binary matrix storage 18555849f57SBarry Smith format is 18655849f57SBarry Smith .vb 18755849f57SBarry Smith has not yet been determined 18855849f57SBarry Smith .ve 18955849f57SBarry Smith 19055849f57SBarry Smith .seealso: PetscViewerBinaryOpen(), SNESView(), MatLoad(), VecLoad() 19155849f57SBarry Smith @*/ 1922d53ad75SBarry Smith PetscErrorCode SNESLoad(SNES snes, PetscViewer viewer) 19355849f57SBarry Smith { 19455849f57SBarry Smith PetscErrorCode ierr; 19555849f57SBarry Smith PetscBool isbinary; 196060da220SMatthew G. Knepley PetscInt classid; 19755849f57SBarry Smith char type[256]; 19855849f57SBarry Smith KSP ksp; 1992d53ad75SBarry Smith DM dm; 2002d53ad75SBarry Smith DMSNES dmsnes; 20155849f57SBarry Smith 20255849f57SBarry Smith PetscFunctionBegin; 2032d53ad75SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 20455849f57SBarry Smith PetscValidHeaderSpecific(viewer,PETSC_VIEWER_CLASSID,2); 20555849f57SBarry Smith ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERBINARY,&isbinary);CHKERRQ(ierr); 20655849f57SBarry Smith if (!isbinary) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONG,"Invalid viewer; open viewer with PetscViewerBinaryOpen()"); 20755849f57SBarry Smith 208060da220SMatthew G. Knepley ierr = PetscViewerBinaryRead(viewer,&classid,1,NULL,PETSC_INT);CHKERRQ(ierr); 209ce94432eSBarry Smith if (classid != SNES_FILE_CLASSID) SETERRQ(PetscObjectComm((PetscObject)snes),PETSC_ERR_ARG_WRONG,"Not SNES next in file"); 210060da220SMatthew G. Knepley ierr = PetscViewerBinaryRead(viewer,type,256,NULL,PETSC_CHAR);CHKERRQ(ierr); 2112d53ad75SBarry Smith ierr = SNESSetType(snes, type);CHKERRQ(ierr); 2122d53ad75SBarry Smith if (snes->ops->load) { 2132d53ad75SBarry Smith ierr = (*snes->ops->load)(snes,viewer);CHKERRQ(ierr); 214f2c2a1b9SBarry Smith } 2152d53ad75SBarry Smith ierr = SNESGetDM(snes,&dm);CHKERRQ(ierr); 2162d53ad75SBarry Smith ierr = DMGetDMSNES(dm,&dmsnes);CHKERRQ(ierr); 2172d53ad75SBarry Smith ierr = DMSNESLoad(dmsnes,viewer);CHKERRQ(ierr); 2182d53ad75SBarry Smith ierr = SNESGetKSP(snes,&ksp);CHKERRQ(ierr); 21955849f57SBarry Smith ierr = KSPLoad(ksp,viewer);CHKERRQ(ierr); 22055849f57SBarry Smith PetscFunctionReturn(0); 22155849f57SBarry Smith } 2226a388c36SPeter Brune 2239804daf3SBarry Smith #include <petscdraw.h> 224e04113cfSBarry Smith #if defined(PETSC_HAVE_SAWS) 225e04113cfSBarry Smith #include <petscviewersaws.h> 226bfb97211SBarry Smith #endif 2277e2c5f70SBarry Smith /*@C 2289b94acceSBarry Smith SNESView - Prints the SNES data structure. 2299b94acceSBarry Smith 2304c49b128SBarry Smith Collective on SNES 231fee21e36SBarry Smith 232c7afd0dbSLois Curfman McInnes Input Parameters: 233c7afd0dbSLois Curfman McInnes + SNES - the SNES context 234c7afd0dbSLois Curfman McInnes - viewer - visualization context 235c7afd0dbSLois Curfman McInnes 2369b94acceSBarry Smith Options Database Key: 237c8a8ba5cSLois Curfman McInnes . -snes_view - Calls SNESView() at end of SNESSolve() 2389b94acceSBarry Smith 2399b94acceSBarry Smith Notes: 2409b94acceSBarry Smith The available visualization contexts include 241b0a32e0cSBarry Smith + PETSC_VIEWER_STDOUT_SELF - standard output (default) 242b0a32e0cSBarry Smith - PETSC_VIEWER_STDOUT_WORLD - synchronized standard 243c8a8ba5cSLois Curfman McInnes output where only the first processor opens 244c8a8ba5cSLois Curfman McInnes the file. All other processors send their 245c8a8ba5cSLois Curfman McInnes data to the first processor to print. 2469b94acceSBarry Smith 2473e081fefSLois Curfman McInnes The user can open an alternative visualization context with 248b0a32e0cSBarry Smith PetscViewerASCIIOpen() - output to a specified file. 2499b94acceSBarry Smith 25036851e7fSLois Curfman McInnes Level: beginner 25136851e7fSLois Curfman McInnes 2529b94acceSBarry Smith .keywords: SNES, view 2539b94acceSBarry Smith 254b0a32e0cSBarry Smith .seealso: PetscViewerASCIIOpen() 2559b94acceSBarry Smith @*/ 2567087cfbeSBarry Smith PetscErrorCode SNESView(SNES snes,PetscViewer viewer) 2579b94acceSBarry Smith { 258fa9f3622SBarry Smith SNESKSPEW *kctx; 259dfbe8321SBarry Smith PetscErrorCode ierr; 26094b7f48cSBarry Smith KSP ksp; 2617f1410a3SPeter Brune SNESLineSearch linesearch; 26272a02f06SBarry Smith PetscBool iascii,isstring,isbinary,isdraw; 2632d53ad75SBarry Smith DMSNES dmsnes; 264e04113cfSBarry Smith #if defined(PETSC_HAVE_SAWS) 265536b137fSBarry Smith PetscBool issaws; 266bfb97211SBarry Smith #endif 2679b94acceSBarry Smith 2683a40ed3dSBarry Smith PetscFunctionBegin; 2690700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 2703050cee2SBarry Smith if (!viewer) { 271ce94432eSBarry Smith ierr = PetscViewerASCIIGetStdout(PetscObjectComm((PetscObject)snes),&viewer);CHKERRQ(ierr); 2723050cee2SBarry Smith } 2730700a824SBarry Smith PetscValidHeaderSpecific(viewer,PETSC_VIEWER_CLASSID,2); 274c9780b6fSBarry Smith PetscCheckSameComm(snes,1,viewer,2); 27574679c65SBarry Smith 276251f4c67SDmitry Karpeev ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERASCII,&iascii);CHKERRQ(ierr); 277251f4c67SDmitry Karpeev ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERSTRING,&isstring);CHKERRQ(ierr); 27855849f57SBarry Smith ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERBINARY,&isbinary);CHKERRQ(ierr); 27972a02f06SBarry Smith ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERDRAW,&isdraw);CHKERRQ(ierr); 280e04113cfSBarry Smith #if defined(PETSC_HAVE_SAWS) 281536b137fSBarry Smith ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERSAWS,&issaws);CHKERRQ(ierr); 282bfb97211SBarry Smith #endif 28332077d6dSBarry Smith if (iascii) { 284dc0571f2SMatthew G. Knepley SNESNormSchedule normschedule; 285dc0571f2SMatthew G. Knepley 286dae58748SBarry Smith ierr = PetscObjectPrintClassNamePrefixType((PetscObject)snes,viewer);CHKERRQ(ierr); 287fce1e034SJed Brown if (!snes->setupcalled) { 288fce1e034SJed Brown ierr = PetscViewerASCIIPrintf(viewer," SNES has not been set up so information may be incomplete\n");CHKERRQ(ierr); 289fce1e034SJed Brown } 290e7788613SBarry Smith if (snes->ops->view) { 291b0a32e0cSBarry Smith ierr = PetscViewerASCIIPushTab(viewer);CHKERRQ(ierr); 292e7788613SBarry Smith ierr = (*snes->ops->view)(snes,viewer);CHKERRQ(ierr); 293b0a32e0cSBarry Smith ierr = PetscViewerASCIIPopTab(viewer);CHKERRQ(ierr); 2940ef38995SBarry Smith } 29577431f27SBarry Smith ierr = PetscViewerASCIIPrintf(viewer," maximum iterations=%D, maximum function evaluations=%D\n",snes->max_its,snes->max_funcs);CHKERRQ(ierr); 29657622a8eSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," tolerances: relative=%g, absolute=%g, solution=%g\n",(double)snes->rtol,(double)snes->abstol,(double)snes->stol);CHKERRQ(ierr); 297efd4aadfSBarry Smith if (snes->usesksp) { 29877431f27SBarry Smith ierr = PetscViewerASCIIPrintf(viewer," total number of linear solver iterations=%D\n",snes->linear_its);CHKERRQ(ierr); 299efd4aadfSBarry Smith } 30077431f27SBarry Smith ierr = PetscViewerASCIIPrintf(viewer," total number of function evaluations=%D\n",snes->nfuncs);CHKERRQ(ierr); 301dc0571f2SMatthew G. Knepley ierr = SNESGetNormSchedule(snes, &normschedule);CHKERRQ(ierr); 302dc0571f2SMatthew G. Knepley if (normschedule > 0) {ierr = PetscViewerASCIIPrintf(viewer," norm schedule %s\n",SNESNormSchedules[normschedule]);CHKERRQ(ierr);} 30317fe4bdfSPeter Brune if (snes->gridsequence) { 30417fe4bdfSPeter Brune ierr = PetscViewerASCIIPrintf(viewer," total number of grid sequence refinements=%D\n",snes->gridsequence);CHKERRQ(ierr); 30517fe4bdfSPeter Brune } 3069b94acceSBarry Smith if (snes->ksp_ewconv) { 307fa9f3622SBarry Smith kctx = (SNESKSPEW*)snes->kspconvctx; 3089b94acceSBarry Smith if (kctx) { 30977431f27SBarry Smith ierr = PetscViewerASCIIPrintf(viewer," Eisenstat-Walker computation of KSP relative tolerance (version %D)\n",kctx->version);CHKERRQ(ierr); 31057622a8eSBarry 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); 31157622a8eSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," gamma=%g, alpha=%g, alpha2=%g\n",(double)kctx->gamma,(double)kctx->alpha,(double)kctx->alpha2);CHKERRQ(ierr); 3129b94acceSBarry Smith } 3139b94acceSBarry Smith } 314eb1f6c34SBarry Smith if (snes->lagpreconditioner == -1) { 315eb1f6c34SBarry Smith ierr = PetscViewerASCIIPrintf(viewer," Preconditioned is never rebuilt\n");CHKERRQ(ierr); 316eb1f6c34SBarry Smith } else if (snes->lagpreconditioner > 1) { 317eb1f6c34SBarry Smith ierr = PetscViewerASCIIPrintf(viewer," Preconditioned is rebuilt every %D new Jacobians\n",snes->lagpreconditioner);CHKERRQ(ierr); 318eb1f6c34SBarry Smith } 319eb1f6c34SBarry Smith if (snes->lagjacobian == -1) { 320eb1f6c34SBarry Smith ierr = PetscViewerASCIIPrintf(viewer," Jacobian is never rebuilt\n");CHKERRQ(ierr); 321eb1f6c34SBarry Smith } else if (snes->lagjacobian > 1) { 32242f4f86dSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," Jacobian is rebuilt every %D SNES iterations\n",snes->lagjacobian);CHKERRQ(ierr); 323eb1f6c34SBarry Smith } 3240f5bd95cSBarry Smith } else if (isstring) { 325317d6ea6SBarry Smith const char *type; 326454a90a3SBarry Smith ierr = SNESGetType(snes,&type);CHKERRQ(ierr); 327b0a32e0cSBarry Smith ierr = PetscViewerStringSPrintf(viewer," %-3.3s",type);CHKERRQ(ierr); 32855849f57SBarry Smith } else if (isbinary) { 32955849f57SBarry Smith PetscInt classid = SNES_FILE_CLASSID; 33055849f57SBarry Smith MPI_Comm comm; 33155849f57SBarry Smith PetscMPIInt rank; 33255849f57SBarry Smith char type[256]; 33355849f57SBarry Smith 33455849f57SBarry Smith ierr = PetscObjectGetComm((PetscObject)snes,&comm);CHKERRQ(ierr); 33555849f57SBarry Smith ierr = MPI_Comm_rank(comm,&rank);CHKERRQ(ierr); 33655849f57SBarry Smith if (!rank) { 33755849f57SBarry Smith ierr = PetscViewerBinaryWrite(viewer,&classid,1,PETSC_INT,PETSC_FALSE);CHKERRQ(ierr); 33889d949e2SBarry Smith ierr = PetscStrncpy(type,((PetscObject)snes)->type_name,sizeof(type));CHKERRQ(ierr); 33989d949e2SBarry Smith ierr = PetscViewerBinaryWrite(viewer,type,sizeof(type),PETSC_CHAR,PETSC_FALSE);CHKERRQ(ierr); 34055849f57SBarry Smith } 34155849f57SBarry Smith if (snes->ops->view) { 34255849f57SBarry Smith ierr = (*snes->ops->view)(snes,viewer);CHKERRQ(ierr); 34355849f57SBarry Smith } 34472a02f06SBarry Smith } else if (isdraw) { 34572a02f06SBarry Smith PetscDraw draw; 34672a02f06SBarry Smith char str[36]; 34789fd9fafSBarry Smith PetscReal x,y,bottom,h; 34872a02f06SBarry Smith 34972a02f06SBarry Smith ierr = PetscViewerDrawGetDraw(viewer,0,&draw);CHKERRQ(ierr); 35072a02f06SBarry Smith ierr = PetscDrawGetCurrentPoint(draw,&x,&y);CHKERRQ(ierr); 35172a02f06SBarry Smith ierr = PetscStrcpy(str,"SNES: ");CHKERRQ(ierr); 35272a02f06SBarry Smith ierr = PetscStrcat(str,((PetscObject)snes)->type_name);CHKERRQ(ierr); 35351fa3d41SBarry Smith ierr = PetscDrawStringBoxed(draw,x,y,PETSC_DRAW_BLUE,PETSC_DRAW_BLACK,str,NULL,&h);CHKERRQ(ierr); 35489fd9fafSBarry Smith bottom = y - h; 35572a02f06SBarry Smith ierr = PetscDrawPushCurrentPoint(draw,x,bottom);CHKERRQ(ierr); 356c4646bacSPeter Brune if (snes->ops->view) { 357c4646bacSPeter Brune ierr = (*snes->ops->view)(snes,viewer);CHKERRQ(ierr); 358c4646bacSPeter Brune } 359e04113cfSBarry Smith #if defined(PETSC_HAVE_SAWS) 360536b137fSBarry Smith } else if (issaws) { 361d45a07a7SBarry Smith PetscMPIInt rank; 3622657e9d9SBarry Smith const char *name; 363d45a07a7SBarry Smith 3642657e9d9SBarry Smith ierr = PetscObjectGetName((PetscObject)snes,&name);CHKERRQ(ierr); 365d45a07a7SBarry Smith ierr = MPI_Comm_rank(PETSC_COMM_WORLD,&rank);CHKERRQ(ierr); 366d45a07a7SBarry Smith if (!((PetscObject)snes)->amsmem && !rank) { 367d45a07a7SBarry Smith char dir[1024]; 368d45a07a7SBarry Smith 369e04113cfSBarry Smith ierr = PetscObjectViewSAWs((PetscObject)snes,viewer);CHKERRQ(ierr); 3702657e9d9SBarry Smith ierr = PetscSNPrintf(dir,1024,"/PETSc/Objects/%s/its",name);CHKERRQ(ierr); 3712657e9d9SBarry Smith PetscStackCallSAWs(SAWs_Register,(dir,&snes->iter,1,SAWs_READ,SAWs_INT)); 372bfb97211SBarry Smith if (!snes->conv_hist) { 373a0931e03SBarry Smith ierr = SNESSetConvergenceHistory(snes,NULL,NULL,PETSC_DECIDE,PETSC_TRUE);CHKERRQ(ierr); 374bfb97211SBarry Smith } 3752657e9d9SBarry Smith ierr = PetscSNPrintf(dir,1024,"/PETSc/Objects/%s/conv_hist",name);CHKERRQ(ierr); 3762657e9d9SBarry Smith PetscStackCallSAWs(SAWs_Register,(dir,snes->conv_hist,10,SAWs_READ,SAWs_DOUBLE)); 377f05ece33SBarry Smith } 378bfb97211SBarry Smith #endif 37972a02f06SBarry Smith } 38072a02f06SBarry Smith if (snes->linesearch) { 38172a02f06SBarry Smith ierr = PetscViewerASCIIPushTab(viewer);CHKERRQ(ierr); 3827601faf0SJed Brown ierr = SNESGetLineSearch(snes, &linesearch);CHKERRQ(ierr); 38372a02f06SBarry Smith ierr = SNESLineSearchView(linesearch, viewer);CHKERRQ(ierr); 38472a02f06SBarry Smith ierr = PetscViewerASCIIPopTab(viewer);CHKERRQ(ierr); 38519bcc07fSBarry Smith } 386efd4aadfSBarry Smith if (snes->npc && snes->usesnpc) { 3874a0c5b0cSMatthew G Knepley ierr = PetscViewerASCIIPushTab(viewer);CHKERRQ(ierr); 388efd4aadfSBarry Smith ierr = SNESView(snes->npc, viewer);CHKERRQ(ierr); 3894a0c5b0cSMatthew G Knepley ierr = PetscViewerASCIIPopTab(viewer);CHKERRQ(ierr); 3904a0c5b0cSMatthew G Knepley } 3912d53ad75SBarry Smith ierr = PetscViewerASCIIPushTab(viewer);CHKERRQ(ierr); 3922d53ad75SBarry Smith ierr = DMGetDMSNES(snes->dm,&dmsnes);CHKERRQ(ierr); 3932d53ad75SBarry Smith ierr = DMSNESView(dmsnes, viewer);CHKERRQ(ierr); 3942d53ad75SBarry Smith ierr = PetscViewerASCIIPopTab(viewer);CHKERRQ(ierr); 3952c155ee1SBarry Smith if (snes->usesksp) { 3962c155ee1SBarry Smith ierr = SNESGetKSP(snes,&ksp);CHKERRQ(ierr); 397b0a32e0cSBarry Smith ierr = PetscViewerASCIIPushTab(viewer);CHKERRQ(ierr); 39894b7f48cSBarry Smith ierr = KSPView(ksp,viewer);CHKERRQ(ierr); 399b0a32e0cSBarry Smith ierr = PetscViewerASCIIPopTab(viewer);CHKERRQ(ierr); 4002c155ee1SBarry Smith } 40172a02f06SBarry Smith if (isdraw) { 40272a02f06SBarry Smith PetscDraw draw; 40372a02f06SBarry Smith ierr = PetscViewerDrawGetDraw(viewer,0,&draw);CHKERRQ(ierr); 40472a02f06SBarry Smith ierr = PetscDrawPopCurrentPoint(draw);CHKERRQ(ierr); 4057f1410a3SPeter Brune } 4063a40ed3dSBarry Smith PetscFunctionReturn(0); 4079b94acceSBarry Smith } 4089b94acceSBarry Smith 40976b2cf59SMatthew Knepley /* 41076b2cf59SMatthew Knepley We retain a list of functions that also take SNES command 41176b2cf59SMatthew Knepley line options. These are called at the end SNESSetFromOptions() 41276b2cf59SMatthew Knepley */ 41376b2cf59SMatthew Knepley #define MAXSETFROMOPTIONS 5 414a7cc72afSBarry Smith static PetscInt numberofsetfromoptions; 4156849ba73SBarry Smith static PetscErrorCode (*othersetfromoptions[MAXSETFROMOPTIONS])(SNES); 41676b2cf59SMatthew Knepley 417ac226902SBarry Smith /*@C 41876b2cf59SMatthew Knepley SNESAddOptionsChecker - Adds an additional function to check for SNES options. 41976b2cf59SMatthew Knepley 42076b2cf59SMatthew Knepley Not Collective 42176b2cf59SMatthew Knepley 42276b2cf59SMatthew Knepley Input Parameter: 42376b2cf59SMatthew Knepley . snescheck - function that checks for options 42476b2cf59SMatthew Knepley 42576b2cf59SMatthew Knepley Level: developer 42676b2cf59SMatthew Knepley 42776b2cf59SMatthew Knepley .seealso: SNESSetFromOptions() 42876b2cf59SMatthew Knepley @*/ 4297087cfbeSBarry Smith PetscErrorCode SNESAddOptionsChecker(PetscErrorCode (*snescheck)(SNES)) 43076b2cf59SMatthew Knepley { 43176b2cf59SMatthew Knepley PetscFunctionBegin; 432f23aa3ddSBarry Smith if (numberofsetfromoptions >= MAXSETFROMOPTIONS) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE, "Too many options checkers, only %D allowed", MAXSETFROMOPTIONS); 43376b2cf59SMatthew Knepley othersetfromoptions[numberofsetfromoptions++] = snescheck; 43476b2cf59SMatthew Knepley PetscFunctionReturn(0); 43576b2cf59SMatthew Knepley } 43676b2cf59SMatthew Knepley 4377087cfbeSBarry Smith extern PetscErrorCode SNESDefaultMatrixFreeCreate2(SNES,Vec,Mat*); 438aa3661deSLisandro Dalcin 439ace3abfcSBarry Smith static PetscErrorCode SNESSetUpMatrixFree_Private(SNES snes, PetscBool hasOperator, PetscInt version) 440aa3661deSLisandro Dalcin { 441aa3661deSLisandro Dalcin Mat J; 442aa3661deSLisandro Dalcin KSP ksp; 443aa3661deSLisandro Dalcin PC pc; 444ace3abfcSBarry Smith PetscBool match; 445aa3661deSLisandro Dalcin PetscErrorCode ierr; 446895c21f2SBarry Smith MatNullSpace nullsp; 447aa3661deSLisandro Dalcin 448aa3661deSLisandro Dalcin PetscFunctionBegin; 4490700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 450aa3661deSLisandro Dalcin 45198613b67SLisandro Dalcin if (!snes->vec_func && (snes->jacobian || snes->jacobian_pre)) { 45298613b67SLisandro Dalcin Mat A = snes->jacobian, B = snes->jacobian_pre; 4532a7a6963SBarry Smith ierr = MatCreateVecs(A ? A : B, NULL,&snes->vec_func);CHKERRQ(ierr); 45498613b67SLisandro Dalcin } 45598613b67SLisandro Dalcin 456aa3661deSLisandro Dalcin if (version == 1) { 457aa3661deSLisandro Dalcin ierr = MatCreateSNESMF(snes,&J);CHKERRQ(ierr); 45898613b67SLisandro Dalcin ierr = MatMFFDSetOptionsPrefix(J,((PetscObject)snes)->prefix);CHKERRQ(ierr); 4599c6ac3b3SBarry Smith ierr = MatSetFromOptions(J);CHKERRQ(ierr); 460aa3661deSLisandro Dalcin } else if (version == 2) { 461e32f2f54SBarry Smith if (!snes->vec_func) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"SNESSetFunction() must be called first"); 462570b7f6dSBarry Smith #if !defined(PETSC_USE_COMPLEX) && !defined(PETSC_USE_REAL_SINGLE) && !defined(PETSC_USE_REAL___FLOAT128) && !defined(PETSC_USE_REAL___FP16) 463aa3661deSLisandro Dalcin ierr = SNESDefaultMatrixFreeCreate2(snes,snes->vec_func,&J);CHKERRQ(ierr); 464aa3661deSLisandro Dalcin #else 465e32f2f54SBarry Smith SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP, "matrix-free operator rutines (version 2)"); 466aa3661deSLisandro Dalcin #endif 467a8248277SBarry Smith } else SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE, "matrix-free operator rutines, only version 1 and 2"); 468aa3661deSLisandro Dalcin 469895c21f2SBarry Smith /* attach any user provided null space that was on Amat to the newly created matrix free matrix */ 470895c21f2SBarry Smith if (snes->jacobian) { 471895c21f2SBarry Smith ierr = MatGetNullSpace(snes->jacobian,&nullsp);CHKERRQ(ierr); 472895c21f2SBarry Smith if (nullsp) { 473895c21f2SBarry Smith ierr = MatSetNullSpace(J,nullsp);CHKERRQ(ierr); 474895c21f2SBarry Smith } 475895c21f2SBarry Smith } 476895c21f2SBarry Smith 477aa3661deSLisandro Dalcin ierr = PetscInfo1(snes,"Setting default matrix-free operator routines (version %D)\n", version);CHKERRQ(ierr); 478d3462f78SMatthew Knepley if (hasOperator) { 4793232da50SPeter Brune 480aa3661deSLisandro Dalcin /* This version replaces the user provided Jacobian matrix with a 481aa3661deSLisandro Dalcin matrix-free version but still employs the user-provided preconditioner matrix. */ 482aa3661deSLisandro Dalcin ierr = SNESSetJacobian(snes,J,0,0,0);CHKERRQ(ierr); 483aa3661deSLisandro Dalcin } else { 484aa3661deSLisandro Dalcin /* This version replaces both the user-provided Jacobian and the user- 4853232da50SPeter Brune provided preconditioner Jacobian with the default matrix free version. */ 486efd4aadfSBarry Smith if ((snes->npcside== PC_LEFT) && snes->npc) { 487d728fb7dSPeter Brune if (!snes->jacobian){ierr = SNESSetJacobian(snes,J,0,0,0);CHKERRQ(ierr);} 488172a4300SPeter Brune } else { 48928a52e04SBarry Smith ierr = SNESSetJacobian(snes,J,J,MatMFFDComputeJacobian,0);CHKERRQ(ierr); 490172a4300SPeter Brune } 491aa3661deSLisandro Dalcin /* Force no preconditioner */ 492aa3661deSLisandro Dalcin ierr = SNESGetKSP(snes,&ksp);CHKERRQ(ierr); 493aa3661deSLisandro Dalcin ierr = KSPGetPC(ksp,&pc);CHKERRQ(ierr); 494251f4c67SDmitry Karpeev ierr = PetscObjectTypeCompare((PetscObject)pc,PCSHELL,&match);CHKERRQ(ierr); 495aa3661deSLisandro Dalcin if (!match) { 496aa3661deSLisandro Dalcin ierr = PetscInfo(snes,"Setting default matrix-free preconditioner routines\nThat is no preconditioner is being used\n");CHKERRQ(ierr); 497aa3661deSLisandro Dalcin ierr = PCSetType(pc,PCNONE);CHKERRQ(ierr); 498aa3661deSLisandro Dalcin } 499aa3661deSLisandro Dalcin } 5006bf464f9SBarry Smith ierr = MatDestroy(&J);CHKERRQ(ierr); 501aa3661deSLisandro Dalcin PetscFunctionReturn(0); 502aa3661deSLisandro Dalcin } 503aa3661deSLisandro Dalcin 504dfe15315SJed Brown static PetscErrorCode DMRestrictHook_SNESVecSol(DM dmfine,Mat Restrict,Vec Rscale,Mat Inject,DM dmcoarse,void *ctx) 505dfe15315SJed Brown { 506dfe15315SJed Brown SNES snes = (SNES)ctx; 507dfe15315SJed Brown PetscErrorCode ierr; 5080298fd71SBarry Smith Vec Xfine,Xfine_named = NULL,Xcoarse; 509dfe15315SJed Brown 510dfe15315SJed Brown PetscFunctionBegin; 51116ebb321SJed Brown if (PetscLogPrintInfo) { 51216ebb321SJed Brown PetscInt finelevel,coarselevel,fineclevel,coarseclevel; 51316ebb321SJed Brown ierr = DMGetRefineLevel(dmfine,&finelevel);CHKERRQ(ierr); 51416ebb321SJed Brown ierr = DMGetCoarsenLevel(dmfine,&fineclevel);CHKERRQ(ierr); 51516ebb321SJed Brown ierr = DMGetRefineLevel(dmcoarse,&coarselevel);CHKERRQ(ierr); 51616ebb321SJed Brown ierr = DMGetCoarsenLevel(dmcoarse,&coarseclevel);CHKERRQ(ierr); 51716ebb321SJed Brown ierr = PetscInfo4(dmfine,"Restricting SNES solution vector from level %D-%D to level %D-%D\n",finelevel,fineclevel,coarselevel,coarseclevel);CHKERRQ(ierr); 51816ebb321SJed Brown } 519dfe15315SJed Brown if (dmfine == snes->dm) Xfine = snes->vec_sol; 520dfe15315SJed Brown else { 521dfe15315SJed Brown ierr = DMGetNamedGlobalVector(dmfine,"SNESVecSol",&Xfine_named);CHKERRQ(ierr); 522dfe15315SJed Brown Xfine = Xfine_named; 523dfe15315SJed Brown } 524dfe15315SJed Brown ierr = DMGetNamedGlobalVector(dmcoarse,"SNESVecSol",&Xcoarse);CHKERRQ(ierr); 525907f5c5aSLawrence Mitchell if (Inject) { 526907f5c5aSLawrence Mitchell ierr = MatRestrict(Inject,Xfine,Xcoarse);CHKERRQ(ierr); 527907f5c5aSLawrence Mitchell } else { 528dfe15315SJed Brown ierr = MatRestrict(Restrict,Xfine,Xcoarse);CHKERRQ(ierr); 529dfe15315SJed Brown ierr = VecPointwiseMult(Xcoarse,Xcoarse,Rscale);CHKERRQ(ierr); 530907f5c5aSLawrence Mitchell } 531dfe15315SJed Brown ierr = DMRestoreNamedGlobalVector(dmcoarse,"SNESVecSol",&Xcoarse);CHKERRQ(ierr); 532dfe15315SJed Brown if (Xfine_named) {ierr = DMRestoreNamedGlobalVector(dmfine,"SNESVecSol",&Xfine_named);CHKERRQ(ierr);} 533dfe15315SJed Brown PetscFunctionReturn(0); 534dfe15315SJed Brown } 535dfe15315SJed Brown 53616ebb321SJed Brown static PetscErrorCode DMCoarsenHook_SNESVecSol(DM dm,DM dmc,void *ctx) 53716ebb321SJed Brown { 53816ebb321SJed Brown PetscErrorCode ierr; 53916ebb321SJed Brown 54016ebb321SJed Brown PetscFunctionBegin; 54116ebb321SJed Brown ierr = DMCoarsenHookAdd(dmc,DMCoarsenHook_SNESVecSol,DMRestrictHook_SNESVecSol,ctx);CHKERRQ(ierr); 54216ebb321SJed Brown PetscFunctionReturn(0); 54316ebb321SJed Brown } 54416ebb321SJed Brown 545a6950cb2SJed Brown /* This may be called to rediscretize the operator on levels of linear multigrid. The DM shuffle is so the user can 546a6950cb2SJed Brown * safely call SNESGetDM() in their residual evaluation routine. */ 54723ee1639SBarry Smith static PetscErrorCode KSPComputeOperators_SNES(KSP ksp,Mat A,Mat B,void *ctx) 548caa4e7f2SJed Brown { 549caa4e7f2SJed Brown SNES snes = (SNES)ctx; 550caa4e7f2SJed Brown PetscErrorCode ierr; 551caa4e7f2SJed Brown Mat Asave = A,Bsave = B; 5520298fd71SBarry Smith Vec X,Xnamed = NULL; 553dfe15315SJed Brown DM dmsave; 5544e269d77SPeter Brune void *ctxsave; 555d1e9a80fSBarry Smith PetscErrorCode (*jac)(SNES,Vec,Mat,Mat,void*); 556caa4e7f2SJed Brown 557caa4e7f2SJed Brown PetscFunctionBegin; 558dfe15315SJed Brown dmsave = snes->dm; 559dfe15315SJed Brown ierr = KSPGetDM(ksp,&snes->dm);CHKERRQ(ierr); 560dfe15315SJed Brown if (dmsave == snes->dm) X = snes->vec_sol; /* We are on the finest level */ 561dfe15315SJed Brown else { /* We are on a coarser level, this vec was initialized using a DM restrict hook */ 562dfe15315SJed Brown ierr = DMGetNamedGlobalVector(snes->dm,"SNESVecSol",&Xnamed);CHKERRQ(ierr); 563dfe15315SJed Brown X = Xnamed; 5640298fd71SBarry Smith ierr = SNESGetJacobian(snes,NULL,NULL,&jac,&ctxsave);CHKERRQ(ierr); 5654e269d77SPeter Brune /* If the DM's don't match up, the MatFDColoring context needed for the jacobian won't match up either -- fixit. */ 5668d359177SBarry Smith if (jac == SNESComputeJacobianDefaultColor) { 5678d359177SBarry Smith ierr = SNESSetJacobian(snes,NULL,NULL,SNESComputeJacobianDefaultColor,0);CHKERRQ(ierr); 568dfe15315SJed Brown } 5694e269d77SPeter Brune } 5704e269d77SPeter Brune /* put the previous context back */ 5714e269d77SPeter Brune 572d1e9a80fSBarry Smith ierr = SNESComputeJacobian(snes,X,A,B);CHKERRQ(ierr); 5738d359177SBarry Smith if (snes->dm != dmsave && jac == SNESComputeJacobianDefaultColor) { 5740298fd71SBarry Smith ierr = SNESSetJacobian(snes,NULL,NULL,jac,ctxsave);CHKERRQ(ierr); 5754e269d77SPeter Brune } 5764e269d77SPeter Brune 577ce94432eSBarry Smith if (A != Asave || B != Bsave) SETERRQ(PetscObjectComm((PetscObject)snes),PETSC_ERR_SUP,"No support for changing matrices at this time"); 578dfe15315SJed Brown if (Xnamed) { 579dfe15315SJed Brown ierr = DMRestoreNamedGlobalVector(snes->dm,"SNESVecSol",&Xnamed);CHKERRQ(ierr); 580dfe15315SJed Brown } 581dfe15315SJed Brown snes->dm = dmsave; 582caa4e7f2SJed Brown PetscFunctionReturn(0); 583caa4e7f2SJed Brown } 584caa4e7f2SJed Brown 5856cab3a1bSJed Brown /*@ 5866cab3a1bSJed Brown SNESSetUpMatrices - ensures that matrices are available for SNES, to be called by SNESSetUp_XXX() 5876cab3a1bSJed Brown 5886cab3a1bSJed Brown Collective 5896cab3a1bSJed Brown 5906cab3a1bSJed Brown Input Arguments: 5916cab3a1bSJed Brown . snes - snes to configure 5926cab3a1bSJed Brown 5936cab3a1bSJed Brown Level: developer 5946cab3a1bSJed Brown 5956cab3a1bSJed Brown .seealso: SNESSetUp() 5966cab3a1bSJed Brown @*/ 5976cab3a1bSJed Brown PetscErrorCode SNESSetUpMatrices(SNES snes) 5986cab3a1bSJed Brown { 5996cab3a1bSJed Brown PetscErrorCode ierr; 6006cab3a1bSJed Brown DM dm; 601942e3340SBarry Smith DMSNES sdm; 6026cab3a1bSJed Brown 6036cab3a1bSJed Brown PetscFunctionBegin; 6046cab3a1bSJed Brown ierr = SNESGetDM(snes,&dm);CHKERRQ(ierr); 605942e3340SBarry Smith ierr = DMGetDMSNES(dm,&sdm);CHKERRQ(ierr); 606ce94432eSBarry Smith if (!sdm->ops->computejacobian) SETERRQ(PetscObjectComm((PetscObject)snes),PETSC_ERR_PLIB,"DMSNES not properly configured"); 607f5af7f23SKarl Rupp else if (!snes->jacobian && snes->mf) { 6086cab3a1bSJed Brown Mat J; 6096cab3a1bSJed Brown void *functx; 6106cab3a1bSJed Brown ierr = MatCreateSNESMF(snes,&J);CHKERRQ(ierr); 6116cab3a1bSJed Brown ierr = MatMFFDSetOptionsPrefix(J,((PetscObject)snes)->prefix);CHKERRQ(ierr); 6126cab3a1bSJed Brown ierr = MatSetFromOptions(J);CHKERRQ(ierr); 6130298fd71SBarry Smith ierr = SNESGetFunction(snes,NULL,NULL,&functx);CHKERRQ(ierr); 6143232da50SPeter Brune ierr = SNESSetJacobian(snes,J,J,0,0);CHKERRQ(ierr); 6156cab3a1bSJed Brown ierr = MatDestroy(&J);CHKERRQ(ierr); 616caa4e7f2SJed Brown } else if (snes->mf_operator && !snes->jacobian_pre && !snes->jacobian) { 6176cab3a1bSJed Brown Mat J,B; 6186cab3a1bSJed Brown ierr = MatCreateSNESMF(snes,&J);CHKERRQ(ierr); 6196cab3a1bSJed Brown ierr = MatMFFDSetOptionsPrefix(J,((PetscObject)snes)->prefix);CHKERRQ(ierr); 6206cab3a1bSJed Brown ierr = MatSetFromOptions(J);CHKERRQ(ierr); 621b412c318SBarry Smith ierr = DMCreateMatrix(snes->dm,&B);CHKERRQ(ierr); 62206f20277SJed Brown /* sdm->computejacobian was already set to reach here */ 6230298fd71SBarry Smith ierr = SNESSetJacobian(snes,J,B,NULL,NULL);CHKERRQ(ierr); 6246cab3a1bSJed Brown ierr = MatDestroy(&J);CHKERRQ(ierr); 6256cab3a1bSJed Brown ierr = MatDestroy(&B);CHKERRQ(ierr); 626caa4e7f2SJed Brown } else if (!snes->jacobian_pre) { 6271ba9b98eSMatthew G. Knepley PetscDS prob; 6286cab3a1bSJed Brown Mat J, B; 6291ba9b98eSMatthew G. Knepley PetscBool hasPrec = PETSC_FALSE; 6301ba9b98eSMatthew G. Knepley 6316cab3a1bSJed Brown J = snes->jacobian; 6321ba9b98eSMatthew G. Knepley ierr = DMGetDS(dm, &prob);CHKERRQ(ierr); 6331ba9b98eSMatthew G. Knepley if (prob) {ierr = PetscDSHasJacobianPreconditioner(prob, &hasPrec);CHKERRQ(ierr);} 634ec9a985fSMatthew G. Knepley if (J) {ierr = PetscObjectReference((PetscObject) J);CHKERRQ(ierr);} 635ec9a985fSMatthew G. Knepley else if (hasPrec) {ierr = DMCreateMatrix(snes->dm, &J);CHKERRQ(ierr);} 636b412c318SBarry Smith ierr = DMCreateMatrix(snes->dm, &B);CHKERRQ(ierr); 6370298fd71SBarry Smith ierr = SNESSetJacobian(snes, J ? J : B, B, NULL, NULL);CHKERRQ(ierr); 6381ba9b98eSMatthew G. Knepley ierr = MatDestroy(&J);CHKERRQ(ierr); 6396cab3a1bSJed Brown ierr = MatDestroy(&B);CHKERRQ(ierr); 6406cab3a1bSJed Brown } 641caa4e7f2SJed Brown { 642caa4e7f2SJed Brown KSP ksp; 643caa4e7f2SJed Brown ierr = SNESGetKSP(snes,&ksp);CHKERRQ(ierr); 644caa4e7f2SJed Brown ierr = KSPSetComputeOperators(ksp,KSPComputeOperators_SNES,snes);CHKERRQ(ierr); 64516ebb321SJed Brown ierr = DMCoarsenHookAdd(snes->dm,DMCoarsenHook_SNESVecSol,DMRestrictHook_SNESVecSol,snes);CHKERRQ(ierr); 646caa4e7f2SJed Brown } 6476cab3a1bSJed Brown PetscFunctionReturn(0); 6486cab3a1bSJed Brown } 6496cab3a1bSJed Brown 650fde5950dSBarry Smith /*@C 651fde5950dSBarry Smith SNESMonitorSetFromOptions - Sets a monitor function and viewer appropriate for the type indicated by the user 652fde5950dSBarry Smith 653fde5950dSBarry Smith Collective on SNES 654fde5950dSBarry Smith 655fde5950dSBarry Smith Input Parameters: 656fde5950dSBarry Smith + snes - SNES object you wish to monitor 657fde5950dSBarry Smith . name - the monitor type one is seeking 658fde5950dSBarry Smith . help - message indicating what monitoring is done 659fde5950dSBarry Smith . manual - manual page for the monitor 660fde5950dSBarry Smith . monitor - the monitor function 661fde5950dSBarry 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 662fde5950dSBarry Smith 663fde5950dSBarry Smith Level: developer 664fde5950dSBarry Smith 665fde5950dSBarry Smith .seealso: PetscOptionsGetViewer(), PetscOptionsGetReal(), PetscOptionsHasName(), PetscOptionsGetString(), 666fde5950dSBarry Smith PetscOptionsGetIntArray(), PetscOptionsGetRealArray(), PetscOptionsBool() 667fde5950dSBarry Smith PetscOptionsInt(), PetscOptionsString(), PetscOptionsReal(), PetscOptionsBool(), 668fde5950dSBarry Smith PetscOptionsName(), PetscOptionsBegin(), PetscOptionsEnd(), PetscOptionsHead(), 669fde5950dSBarry Smith PetscOptionsStringArray(),PetscOptionsRealArray(), PetscOptionsScalar(), 670fde5950dSBarry Smith PetscOptionsBoolGroupBegin(), PetscOptionsBoolGroup(), PetscOptionsBoolGroupEnd(), 671fde5950dSBarry Smith PetscOptionsFList(), PetscOptionsEList() 672fde5950dSBarry Smith @*/ 673d43b4f6eSBarry Smith PetscErrorCode SNESMonitorSetFromOptions(SNES snes,const char name[],const char help[], const char manual[],PetscErrorCode (*monitor)(SNES,PetscInt,PetscReal,PetscViewerAndFormat*),PetscErrorCode (*monitorsetup)(SNES,PetscViewerAndFormat*)) 674fde5950dSBarry Smith { 675fde5950dSBarry Smith PetscErrorCode ierr; 676fde5950dSBarry Smith PetscViewer viewer; 677fde5950dSBarry Smith PetscViewerFormat format; 678fde5950dSBarry Smith PetscBool flg; 679fde5950dSBarry Smith 680fde5950dSBarry Smith PetscFunctionBegin; 681fde5950dSBarry Smith ierr = PetscOptionsGetViewer(PetscObjectComm((PetscObject)snes),((PetscObject)snes)->prefix,name,&viewer,&format,&flg);CHKERRQ(ierr); 682fde5950dSBarry Smith if (flg) { 683d43b4f6eSBarry Smith PetscViewerAndFormat *vf; 684d43b4f6eSBarry Smith ierr = PetscViewerAndFormatCreate(viewer,format,&vf);CHKERRQ(ierr); 685d43b4f6eSBarry Smith ierr = PetscObjectDereference((PetscObject)viewer);CHKERRQ(ierr); 686fde5950dSBarry Smith if (monitorsetup) { 687d43b4f6eSBarry Smith ierr = (*monitorsetup)(snes,vf);CHKERRQ(ierr); 688fde5950dSBarry Smith } 689d43b4f6eSBarry Smith ierr = SNESMonitorSet(snes,(PetscErrorCode (*)(SNES,PetscInt,PetscReal,void*))monitor,vf,(PetscErrorCode (*)(void**))PetscViewerAndFormatDestroy);CHKERRQ(ierr); 690fde5950dSBarry Smith } 691fde5950dSBarry Smith PetscFunctionReturn(0); 692fde5950dSBarry Smith } 693fde5950dSBarry Smith 6949b94acceSBarry Smith /*@ 69594b7f48cSBarry Smith SNESSetFromOptions - Sets various SNES and KSP parameters from user options. 6969b94acceSBarry Smith 697c7afd0dbSLois Curfman McInnes Collective on SNES 698c7afd0dbSLois Curfman McInnes 6999b94acceSBarry Smith Input Parameter: 7009b94acceSBarry Smith . snes - the SNES context 7019b94acceSBarry Smith 70236851e7fSLois Curfman McInnes Options Database Keys: 703722329fbSBarry Smith + -snes_type <type> - newtonls, newtontr, ngmres, ncg, nrichardson, qn, vi, fas, SNESType for complete list 70482738288SBarry Smith . -snes_stol - convergence tolerance in terms of the norm 70582738288SBarry Smith of the change in the solution between steps 70670441072SBarry Smith . -snes_atol <abstol> - absolute tolerance of residual norm 707b39c3a46SLois Curfman McInnes . -snes_rtol <rtol> - relative decrease in tolerance norm from initial 708e4d06f11SPatrick Farrell . -snes_divergence_tolerance <divtol> - if the residual goes above divtol*rnorm0, exit with divergence 709be5caee7SBarry Smith . -snes_force_iteration <force> - force SNESSolve() to take at least one iteration 710b39c3a46SLois Curfman McInnes . -snes_max_it <max_it> - maximum number of iterations 711b39c3a46SLois Curfman McInnes . -snes_max_funcs <max_funcs> - maximum number of function evaluations 7124839bfe8SBarry Smith . -snes_max_fail <max_fail> - maximum number of line search failures allowed before stopping, default is none 713ddf469c8SBarry Smith . -snes_max_linear_solve_fail - number of linear solver failures before SNESSolve() stops 714a8054027SBarry Smith . -snes_lag_preconditioner <lag> - how often preconditioner is rebuilt (use -1 to never rebuild) 715e35cf81dSBarry Smith . -snes_lag_jacobian <lag> - how often Jacobian is rebuilt (use -1 to never rebuild) 716b39c3a46SLois Curfman McInnes . -snes_trtol <trtol> - trust region tolerance 7172492ecdbSBarry Smith . -snes_no_convergence_test - skip convergence test in nonlinear 71882738288SBarry Smith solver; hence iterations will continue until max_it 7191fbbfb26SLois Curfman McInnes or some other criterion is reached. Saves expense 72082738288SBarry Smith of convergence test 721fde5950dSBarry Smith . -snes_monitor [ascii][:filename][:viewer format] - prints residual norm at each iteration. if no filename given prints to stdout 722fde5950dSBarry Smith . -snes_monitor_solution [ascii binary draw][:filename][:viewer format] - plots solution at each iteration 723fde5950dSBarry Smith . -snes_monitor_residual [ascii binary draw][:filename][:viewer format] - plots residual (not its norm) at each iteration 724fde5950dSBarry Smith . -snes_monitor_solution_update [ascii binary draw][:filename][:viewer format] - plots update to solution at each iteration 7254619e776SBarry Smith . -snes_monitor_lg_residualnorm - plots residual norm at each iteration 726459f5d12SBarry Smith . -snes_monitor_lg_range - plots residual norm at each iteration 727e24b481bSBarry Smith . -snes_fd - use finite differences to compute Jacobian; very slow, only for testing 728e2e60de9SPeter Brune . -snes_fd_color - use finite differences with coloring to compute Jacobian 7295968eb51SBarry Smith . -snes_mf_ksp_monitor - if using matrix-free multiply then print h at each KSP iteration 730fee2055bSBarry Smith - -snes_converged_reason - print the reason for convergence/divergence after each solve 73182738288SBarry Smith 73282738288SBarry Smith Options Database for Eisenstat-Walker method: 733fa9f3622SBarry Smith + -snes_ksp_ew - use Eisenstat-Walker method for determining linear system convergence 7344b27c08aSLois Curfman McInnes . -snes_ksp_ew_version ver - version of Eisenstat-Walker method 73536851e7fSLois Curfman McInnes . -snes_ksp_ew_rtol0 <rtol0> - Sets rtol0 73636851e7fSLois Curfman McInnes . -snes_ksp_ew_rtolmax <rtolmax> - Sets rtolmax 73736851e7fSLois Curfman McInnes . -snes_ksp_ew_gamma <gamma> - Sets gamma 73836851e7fSLois Curfman McInnes . -snes_ksp_ew_alpha <alpha> - Sets alpha 73936851e7fSLois Curfman McInnes . -snes_ksp_ew_alpha2 <alpha2> - Sets alpha2 74036851e7fSLois Curfman McInnes - -snes_ksp_ew_threshold <threshold> - Sets threshold 74182738288SBarry Smith 74211ca99fdSLois Curfman McInnes Notes: 74311ca99fdSLois Curfman McInnes To see all options, run your program with the -help option or consult 744a7f22e61SSatish Balay Users-Manual: ch_snes 74583e2fdc7SBarry Smith 74636851e7fSLois Curfman McInnes Level: beginner 74736851e7fSLois Curfman McInnes 7489b94acceSBarry Smith .keywords: SNES, nonlinear, set, options, database 7499b94acceSBarry Smith 75069ed319cSSatish Balay .seealso: SNESSetOptionsPrefix() 7519b94acceSBarry Smith @*/ 7527087cfbeSBarry Smith PetscErrorCode SNESSetFromOptions(SNES snes) 7539b94acceSBarry Smith { 7548afaa268SBarry Smith PetscBool flg,pcset,persist,set; 755d8f46077SPeter Brune PetscInt i,indx,lag,grids; 75604d7464bSBarry Smith const char *deft = SNESNEWTONLS; 75785385478SLisandro Dalcin const char *convtests[] = {"default","skip"}; 75885385478SLisandro Dalcin SNESKSPEW *kctx = NULL; 759e8105e01SRichard Katz char type[256], monfilename[PETSC_MAX_PATH_LEN]; 76085385478SLisandro Dalcin PetscErrorCode ierr; 761c40d0f55SPeter Brune PCSide pcside; 762a64e098fSPeter Brune const char *optionsprefix; 7639b94acceSBarry Smith 7643a40ed3dSBarry Smith PetscFunctionBegin; 7650700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 7660f51fdf8SToby Isaac ierr = SNESRegisterAll();CHKERRQ(ierr); 7673194b578SJed Brown ierr = PetscObjectOptionsBegin((PetscObject)snes);CHKERRQ(ierr); 768639ff905SBarry Smith if (((PetscObject)snes)->type_name) deft = ((PetscObject)snes)->type_name; 769a264d7a6SBarry Smith ierr = PetscOptionsFList("-snes_type","Nonlinear solver method","SNESSetType",SNESList,deft,type,256,&flg);CHKERRQ(ierr); 770d64ed03dSBarry Smith if (flg) { 771186905e3SBarry Smith ierr = SNESSetType(snes,type);CHKERRQ(ierr); 7727adad957SLisandro Dalcin } else if (!((PetscObject)snes)->type_name) { 773186905e3SBarry Smith ierr = SNESSetType(snes,deft);CHKERRQ(ierr); 774d64ed03dSBarry Smith } 77594ae4db5SBarry Smith ierr = PetscOptionsReal("-snes_stol","Stop if step length less than","SNESSetTolerances",snes->stol,&snes->stol,NULL);CHKERRQ(ierr); 77694ae4db5SBarry Smith ierr = PetscOptionsReal("-snes_atol","Stop if function norm less than","SNESSetTolerances",snes->abstol,&snes->abstol,NULL);CHKERRQ(ierr); 777186905e3SBarry Smith 77894ae4db5SBarry Smith ierr = PetscOptionsReal("-snes_rtol","Stop if decrease in function norm less than","SNESSetTolerances",snes->rtol,&snes->rtol,NULL);CHKERRQ(ierr); 779e4d06f11SPatrick Farrell ierr = PetscOptionsReal("-snes_divergence_tolerance","Stop if residual norm increases by this factor","SNESSetDivergenceTolerance",snes->divtol,&snes->divtol,NULL);CHKERRQ(ierr); 7800298fd71SBarry Smith ierr = PetscOptionsInt("-snes_max_it","Maximum iterations","SNESSetTolerances",snes->max_its,&snes->max_its,NULL);CHKERRQ(ierr); 7810298fd71SBarry Smith ierr = PetscOptionsInt("-snes_max_funcs","Maximum function evaluations","SNESSetTolerances",snes->max_funcs,&snes->max_funcs,NULL);CHKERRQ(ierr); 7820298fd71SBarry Smith ierr = PetscOptionsInt("-snes_max_fail","Maximum nonlinear step failures","SNESSetMaxNonlinearStepFailures",snes->maxFailures,&snes->maxFailures,NULL);CHKERRQ(ierr); 7830298fd71SBarry Smith ierr = PetscOptionsInt("-snes_max_linear_solve_fail","Maximum failures in linear solves allowed","SNESSetMaxLinearSolveFailures",snes->maxLinearSolveFailures,&snes->maxLinearSolveFailures,NULL);CHKERRQ(ierr); 7840298fd71SBarry Smith ierr = PetscOptionsBool("-snes_error_if_not_converged","Generate error if solver does not converge","SNESSetErrorIfNotConverged",snes->errorifnotconverged,&snes->errorifnotconverged,NULL);CHKERRQ(ierr); 785be5caee7SBarry Smith ierr = PetscOptionsBool("-snes_force_iteration","Force SNESSolve() to take at least one iteration","SNESForceIteration",snes->forceiteration,&snes->forceiteration,NULL);CHKERRQ(ierr); 78685385478SLisandro Dalcin 787a8054027SBarry Smith ierr = PetscOptionsInt("-snes_lag_preconditioner","How often to rebuild preconditioner","SNESSetLagPreconditioner",snes->lagpreconditioner,&lag,&flg);CHKERRQ(ierr); 788a8054027SBarry Smith if (flg) { 789a8054027SBarry Smith ierr = SNESSetLagPreconditioner(snes,lag);CHKERRQ(ierr); 790a8054027SBarry Smith } 79137ec4e1aSPeter Brune ierr = PetscOptionsBool("-snes_lag_preconditioner_persists","Preconditioner lagging through multiple solves","SNESSetLagPreconditionerPersists",snes->lagjac_persist,&persist,&flg);CHKERRQ(ierr); 79237ec4e1aSPeter Brune if (flg) { 79337ec4e1aSPeter Brune ierr = SNESSetLagPreconditionerPersists(snes,persist);CHKERRQ(ierr); 79437ec4e1aSPeter Brune } 795e35cf81dSBarry Smith ierr = PetscOptionsInt("-snes_lag_jacobian","How often to rebuild Jacobian","SNESSetLagJacobian",snes->lagjacobian,&lag,&flg);CHKERRQ(ierr); 796e35cf81dSBarry Smith if (flg) { 797e35cf81dSBarry Smith ierr = SNESSetLagJacobian(snes,lag);CHKERRQ(ierr); 798e35cf81dSBarry Smith } 79937ec4e1aSPeter Brune ierr = PetscOptionsBool("-snes_lag_jacobian_persists","Jacobian lagging through multiple solves","SNESSetLagJacobianPersists",snes->lagjac_persist,&persist,&flg);CHKERRQ(ierr); 80037ec4e1aSPeter Brune if (flg) { 80137ec4e1aSPeter Brune ierr = SNESSetLagJacobianPersists(snes,persist);CHKERRQ(ierr); 80237ec4e1aSPeter Brune } 80337ec4e1aSPeter Brune 804efd51863SBarry Smith ierr = PetscOptionsInt("-snes_grid_sequence","Use grid sequencing to generate initial guess","SNESSetGridSequence",snes->gridsequence,&grids,&flg);CHKERRQ(ierr); 805efd51863SBarry Smith if (flg) { 806efd51863SBarry Smith ierr = SNESSetGridSequence(snes,grids);CHKERRQ(ierr); 807efd51863SBarry Smith } 808a8054027SBarry Smith 80985385478SLisandro Dalcin ierr = PetscOptionsEList("-snes_convergence_test","Convergence test","SNESSetConvergenceTest",convtests,2,"default",&indx,&flg);CHKERRQ(ierr); 81085385478SLisandro Dalcin if (flg) { 81185385478SLisandro Dalcin switch (indx) { 8128d359177SBarry Smith case 0: ierr = SNESSetConvergenceTest(snes,SNESConvergedDefault,NULL,NULL);CHKERRQ(ierr); break; 813e2a6519dSDmitry Karpeev case 1: ierr = SNESSetConvergenceTest(snes,SNESConvergedSkip,NULL,NULL);CHKERRQ(ierr); break; 81485385478SLisandro Dalcin } 81585385478SLisandro Dalcin } 81685385478SLisandro Dalcin 817365a6726SPeter Brune ierr = PetscOptionsEList("-snes_norm_schedule","SNES Norm schedule","SNESSetNormSchedule",SNESNormSchedules,5,"function",&indx,&flg);CHKERRQ(ierr); 818365a6726SPeter Brune if (flg) { ierr = SNESSetNormSchedule(snes,(SNESNormSchedule)indx);CHKERRQ(ierr); } 819fdacfa88SPeter Brune 82047073ea2SPeter Brune ierr = PetscOptionsEList("-snes_function_type","SNES Norm schedule","SNESSetFunctionType",SNESFunctionTypes,2,"unpreconditioned",&indx,&flg);CHKERRQ(ierr); 82147073ea2SPeter Brune if (flg) { ierr = SNESSetFunctionType(snes,(SNESFunctionType)indx);CHKERRQ(ierr); } 822186905e3SBarry Smith 82385385478SLisandro Dalcin kctx = (SNESKSPEW*)snes->kspconvctx; 82485385478SLisandro Dalcin 8250298fd71SBarry Smith ierr = PetscOptionsBool("-snes_ksp_ew","Use Eisentat-Walker linear system convergence test","SNESKSPSetUseEW",snes->ksp_ewconv,&snes->ksp_ewconv,NULL);CHKERRQ(ierr); 826186905e3SBarry Smith 82794ae4db5SBarry Smith ierr = PetscOptionsInt("-snes_ksp_ew_version","Version 1, 2 or 3","SNESKSPSetParametersEW",kctx->version,&kctx->version,NULL);CHKERRQ(ierr); 82894ae4db5SBarry Smith ierr = PetscOptionsReal("-snes_ksp_ew_rtol0","0 <= rtol0 < 1","SNESKSPSetParametersEW",kctx->rtol_0,&kctx->rtol_0,NULL);CHKERRQ(ierr); 82994ae4db5SBarry Smith ierr = PetscOptionsReal("-snes_ksp_ew_rtolmax","0 <= rtolmax < 1","SNESKSPSetParametersEW",kctx->rtol_max,&kctx->rtol_max,NULL);CHKERRQ(ierr); 83094ae4db5SBarry Smith ierr = PetscOptionsReal("-snes_ksp_ew_gamma","0 <= gamma <= 1","SNESKSPSetParametersEW",kctx->gamma,&kctx->gamma,NULL);CHKERRQ(ierr); 83194ae4db5SBarry Smith ierr = PetscOptionsReal("-snes_ksp_ew_alpha","1 < alpha <= 2","SNESKSPSetParametersEW",kctx->alpha,&kctx->alpha,NULL);CHKERRQ(ierr); 83294ae4db5SBarry Smith ierr = PetscOptionsReal("-snes_ksp_ew_alpha2","alpha2","SNESKSPSetParametersEW",kctx->alpha2,&kctx->alpha2,NULL);CHKERRQ(ierr); 83394ae4db5SBarry Smith ierr = PetscOptionsReal("-snes_ksp_ew_threshold","0 < threshold < 1","SNESKSPSetParametersEW",kctx->threshold,&kctx->threshold,NULL);CHKERRQ(ierr); 834186905e3SBarry Smith 83590d69ab7SBarry Smith flg = PETSC_FALSE; 8368afaa268SBarry Smith ierr = PetscOptionsBool("-snes_check_jacobian","Check each Jacobian with a differenced one","SNESUpdateCheckJacobian",flg,&flg,&set);CHKERRQ(ierr); 8378afaa268SBarry Smith if (set && flg) { 8385341784dSBarry Smith ierr = SNESSetUpdate(snes,SNESUpdateCheckJacobian);CHKERRQ(ierr); 8395341784dSBarry Smith } 8405341784dSBarry Smith 8415341784dSBarry Smith flg = PETSC_FALSE; 8428afaa268SBarry Smith ierr = PetscOptionsBool("-snes_monitor_cancel","Remove all monitors","SNESMonitorCancel",flg,&flg,&set);CHKERRQ(ierr); 8438afaa268SBarry Smith if (set && flg) {ierr = SNESMonitorCancel(snes);CHKERRQ(ierr);} 844eabae89aSBarry Smith 845fde5950dSBarry Smith ierr = SNESMonitorSetFromOptions(snes,"-snes_monitor","Monitor norm of function","SNESMonitorDefault",SNESMonitorDefault,NULL);CHKERRQ(ierr); 846fde5950dSBarry Smith ierr = SNESMonitorSetFromOptions(snes,"-snes_monitor_short","Monitor norm of function with fewer digits","SNESMonitorDefaultShort",SNESMonitorDefaultShort,NULL);CHKERRQ(ierr); 847fde5950dSBarry Smith ierr = SNESMonitorSetFromOptions(snes,"-snes_monitor_range","Monitor range of elements of function","SNESMonitorRange",SNESMonitorRange,NULL);CHKERRQ(ierr); 848eabae89aSBarry Smith 849fde5950dSBarry Smith ierr = SNESMonitorSetFromOptions(snes,"-snes_monitor_ratio","Monitor ratios of the norm of function for consecutive steps","SNESMonitorRatio",SNESMonitorRatio,SNESMonitorRatioSetUp);CHKERRQ(ierr); 850fde5950dSBarry Smith ierr = SNESMonitorSetFromOptions(snes,"-snes_monitor_field","Monitor norm of function (split into fields)","SNESMonitorDefaultField",SNESMonitorDefaultField,NULL);CHKERRQ(ierr); 851fde5950dSBarry Smith ierr = SNESMonitorSetFromOptions(snes,"-snes_monitor_solution","View solution at each iteration","SNESMonitorSolution",SNESMonitorSolution,NULL);CHKERRQ(ierr); 852fde5950dSBarry Smith ierr = SNESMonitorSetFromOptions(snes,"-snes_monitor_solution_update","View correction at each iteration","SNESMonitorSolutionUpdate",SNESMonitorSolutionUpdate,NULL);CHKERRQ(ierr); 853fde5950dSBarry Smith ierr = SNESMonitorSetFromOptions(snes,"-snes_monitor_residual","View residual at each iteration","SNESMonitorResidual",SNESMonitorResidual,NULL);CHKERRQ(ierr); 854fde5950dSBarry Smith ierr = SNESMonitorSetFromOptions(snes,"-snes_monitor_jacupdate_spectrum","Print the change in the spectrum of the Jacobian","SNESMonitorJacUpdateSpectrum",SNESMonitorJacUpdateSpectrum,NULL);CHKERRQ(ierr); 855fde5950dSBarry Smith ierr = SNESMonitorSetFromOptions(snes,"-snes_monitor_fields","Monitor norm of function per field","SNESMonitorSet",SNESMonitorFields,NULL);CHKERRQ(ierr); 8562db13446SMatthew G. Knepley 8575180491cSLisandro Dalcin ierr = PetscOptionsString("-snes_monitor_python","Use Python function","SNESMonitorSet",0,monfilename,PETSC_MAX_PATH_LEN,&flg);CHKERRQ(ierr); 8585180491cSLisandro Dalcin if (flg) {ierr = PetscPythonMonitorSet((PetscObject)snes,monfilename);CHKERRQ(ierr);} 8595180491cSLisandro Dalcin 860fde5950dSBarry Smith 86190d69ab7SBarry Smith flg = PETSC_FALSE; 8620298fd71SBarry Smith ierr = PetscOptionsBool("-snes_monitor_lg_residualnorm","Plot function norm at each iteration","SNESMonitorLGResidualNorm",flg,&flg,NULL);CHKERRQ(ierr); 863459f5d12SBarry Smith if (flg) { 864d96771aaSLisandro Dalcin PetscDrawLG ctx; 865459f5d12SBarry Smith 8666ba87a44SLisandro Dalcin ierr = SNESMonitorLGCreate(PetscObjectComm((PetscObject)snes),NULL,NULL,PETSC_DECIDE,PETSC_DECIDE,400,300,&ctx);CHKERRQ(ierr); 867d96771aaSLisandro Dalcin ierr = SNESMonitorSet(snes,SNESMonitorLGResidualNorm,ctx,(PetscErrorCode (*)(void**))PetscDrawLGDestroy);CHKERRQ(ierr); 868459f5d12SBarry Smith } 86990d69ab7SBarry Smith flg = PETSC_FALSE; 8700298fd71SBarry Smith ierr = PetscOptionsBool("-snes_monitor_lg_range","Plot function range at each iteration","SNESMonitorLGRange",flg,&flg,NULL);CHKERRQ(ierr); 871459f5d12SBarry Smith if (flg) { 872459f5d12SBarry Smith PetscViewer ctx; 873e24b481bSBarry Smith 8746ba87a44SLisandro Dalcin ierr = PetscViewerDrawOpen(PetscObjectComm((PetscObject)snes),NULL,NULL,PETSC_DECIDE,PETSC_DECIDE,400,300,&ctx);CHKERRQ(ierr); 875459f5d12SBarry Smith ierr = SNESMonitorSet(snes,SNESMonitorLGRange,ctx,(PetscErrorCode (*)(void**))PetscViewerDestroy);CHKERRQ(ierr); 876459f5d12SBarry Smith } 8772e7541e6SPeter Brune 8782e7541e6SPeter Brune 879cc0c4584SMatthew G. Knepley 88090d69ab7SBarry Smith flg = PETSC_FALSE; 8818d359177SBarry Smith ierr = PetscOptionsBool("-snes_fd","Use finite differences (slow) to compute Jacobian","SNESComputeJacobianDefault",flg,&flg,NULL);CHKERRQ(ierr); 8824b27c08aSLois Curfman McInnes if (flg) { 8836cab3a1bSJed Brown void *functx; 884b1f624c7SBarry Smith DM dm; 885b1f624c7SBarry Smith DMSNES sdm; 886b1f624c7SBarry Smith ierr = SNESGetDM(snes,&dm);CHKERRQ(ierr); 887b1f624c7SBarry Smith ierr = DMGetDMSNES(dm,&sdm);CHKERRQ(ierr); 888b1f624c7SBarry Smith sdm->jacobianctx = NULL; 8890298fd71SBarry Smith ierr = SNESGetFunction(snes,NULL,NULL,&functx);CHKERRQ(ierr); 8908d359177SBarry Smith ierr = SNESSetJacobian(snes,snes->jacobian,snes->jacobian_pre,SNESComputeJacobianDefault,functx);CHKERRQ(ierr); 891ae15b995SBarry Smith ierr = PetscInfo(snes,"Setting default finite difference Jacobian matrix\n");CHKERRQ(ierr); 8929b94acceSBarry Smith } 893639f9d9dSBarry Smith 89444848bc4SPeter Brune flg = PETSC_FALSE; 8958d359177SBarry Smith ierr = PetscOptionsBool("-snes_fd_function","Use finite differences (slow) to compute function from user objective","SNESObjectiveComputeFunctionDefaultFD",flg,&flg,NULL);CHKERRQ(ierr); 89697584545SPeter Brune if (flg) { 8978d359177SBarry Smith ierr = SNESSetFunction(snes,NULL,SNESObjectiveComputeFunctionDefaultFD,NULL);CHKERRQ(ierr); 89897584545SPeter Brune } 89997584545SPeter Brune 90097584545SPeter Brune flg = PETSC_FALSE; 9018d359177SBarry Smith ierr = PetscOptionsBool("-snes_fd_color","Use finite differences with coloring to compute Jacobian","SNESComputeJacobianDefaultColor",flg,&flg,NULL);CHKERRQ(ierr); 90244848bc4SPeter Brune if (flg) { 903c52e227fSPeter Brune DM dm; 904c52e227fSPeter Brune DMSNES sdm; 905c52e227fSPeter Brune ierr = SNESGetDM(snes,&dm);CHKERRQ(ierr); 906aace71b7SPeter Brune ierr = DMGetDMSNES(dm,&sdm);CHKERRQ(ierr); 907aace71b7SPeter Brune sdm->jacobianctx = NULL; 9088d359177SBarry Smith ierr = SNESSetJacobian(snes,snes->jacobian,snes->jacobian_pre,SNESComputeJacobianDefaultColor,0);CHKERRQ(ierr); 90944848bc4SPeter Brune ierr = PetscInfo(snes,"Setting default finite difference coloring Jacobian matrix\n");CHKERRQ(ierr); 91044848bc4SPeter Brune } 91144848bc4SPeter Brune 912aa3661deSLisandro Dalcin flg = PETSC_FALSE; 913f871313dSBarry 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); 914d8f46077SPeter Brune if (flg && snes->mf_operator) { 915a8248277SBarry Smith snes->mf_operator = PETSC_TRUE; 916d8f46077SPeter Brune snes->mf = PETSC_TRUE; 917a8248277SBarry Smith } 918aa3661deSLisandro Dalcin flg = PETSC_FALSE; 919f871313dSBarry Smith ierr = PetscOptionsBool("-snes_mf","Use a Matrix-Free Jacobian with no preconditioner matrix","SNESSetUseMatrixFree",PETSC_FALSE,&snes->mf,&flg);CHKERRQ(ierr); 920d8f46077SPeter Brune if (!flg && snes->mf_operator) snes->mf = PETSC_TRUE; 921d8f46077SPeter Brune ierr = PetscOptionsInt("-snes_mf_version","Matrix-Free routines version 1 or 2","None",snes->mf_version,&snes->mf_version,0);CHKERRQ(ierr); 922d28543b3SPeter Brune 923c40d0f55SPeter Brune flg = PETSC_FALSE; 924be95d8f1SBarry Smith ierr = SNESGetNPCSide(snes,&pcside);CHKERRQ(ierr); 925be95d8f1SBarry Smith ierr = PetscOptionsEnum("-snes_npc_side","SNES nonlinear preconditioner side","SNESSetNPCSide",PCSides,(PetscEnum)pcside,(PetscEnum*)&pcside,&flg);CHKERRQ(ierr); 926be95d8f1SBarry Smith if (flg) {ierr = SNESSetNPCSide(snes,pcside);CHKERRQ(ierr);} 927c40d0f55SPeter Brune 928e04113cfSBarry Smith #if defined(PETSC_HAVE_SAWS) 9298a70d858SHong Zhang /* 9308a70d858SHong Zhang Publish convergence information using SAWs 9318a70d858SHong Zhang */ 9328a70d858SHong Zhang flg = PETSC_FALSE; 9338a70d858SHong Zhang ierr = PetscOptionsBool("-snes_monitor_saws","Publish SNES progress using SAWs","SNESMonitorSet",flg,&flg,NULL);CHKERRQ(ierr); 9348a70d858SHong Zhang if (flg) { 9358a70d858SHong Zhang void *ctx; 9368a70d858SHong Zhang ierr = SNESMonitorSAWsCreate(snes,&ctx);CHKERRQ(ierr); 9378a70d858SHong Zhang ierr = SNESMonitorSet(snes,SNESMonitorSAWs,ctx,SNESMonitorSAWsDestroy);CHKERRQ(ierr); 9388a70d858SHong Zhang } 9398a70d858SHong Zhang #endif 9408a70d858SHong Zhang #if defined(PETSC_HAVE_SAWS) 941b90c6cbeSBarry Smith { 942b90c6cbeSBarry Smith PetscBool set; 943b90c6cbeSBarry Smith flg = PETSC_FALSE; 9448a70d858SHong Zhang ierr = PetscOptionsBool("-snes_saws_block","Block for SAWs at end of SNESSolve","PetscObjectSAWsBlock",((PetscObject)snes)->amspublishblock,&flg,&set);CHKERRQ(ierr); 945b90c6cbeSBarry Smith if (set) { 946e04113cfSBarry Smith ierr = PetscObjectSAWsSetBlock((PetscObject)snes,flg);CHKERRQ(ierr); 947b90c6cbeSBarry Smith } 948b90c6cbeSBarry Smith } 949b90c6cbeSBarry Smith #endif 950b90c6cbeSBarry Smith 95176b2cf59SMatthew Knepley for (i = 0; i < numberofsetfromoptions; i++) { 95276b2cf59SMatthew Knepley ierr = (*othersetfromoptions[i])(snes);CHKERRQ(ierr); 95376b2cf59SMatthew Knepley } 95476b2cf59SMatthew Knepley 955e7788613SBarry Smith if (snes->ops->setfromoptions) { 956e55864a3SBarry Smith ierr = (*snes->ops->setfromoptions)(PetscOptionsObject,snes);CHKERRQ(ierr); 957639f9d9dSBarry Smith } 9585d973c19SBarry Smith 9595d973c19SBarry Smith /* process any options handlers added with PetscObjectAddOptionsHandler() */ 9600633abcbSJed Brown ierr = PetscObjectProcessOptionsHandlers(PetscOptionsObject,(PetscObject)snes);CHKERRQ(ierr); 961b0a32e0cSBarry Smith ierr = PetscOptionsEnd();CHKERRQ(ierr); 9624bbc92c1SBarry Smith 9639e764e56SPeter Brune if (!snes->linesearch) { 9647601faf0SJed Brown ierr = SNESGetLineSearch(snes, &snes->linesearch);CHKERRQ(ierr); 9659e764e56SPeter Brune } 966f1c6b773SPeter Brune ierr = SNESLineSearchSetFromOptions(snes->linesearch);CHKERRQ(ierr); 9679e764e56SPeter Brune 9686991f827SBarry Smith if (!snes->ksp) {ierr = SNESGetKSP(snes,&snes->ksp);CHKERRQ(ierr);} 9696991f827SBarry Smith ierr = KSPSetOperators(snes->ksp,snes->jacobian,snes->jacobian_pre);CHKERRQ(ierr); 9706991f827SBarry Smith ierr = KSPSetFromOptions(snes->ksp);CHKERRQ(ierr); 9716991f827SBarry Smith 972be95d8f1SBarry Smith /* if someone has set the SNES NPC type, create it. */ 97351e86f29SPeter Brune ierr = SNESGetOptionsPrefix(snes, &optionsprefix);CHKERRQ(ierr); 974c5929fdfSBarry Smith ierr = PetscOptionsHasName(((PetscObject)snes)->options,optionsprefix, "-npc_snes_type", &pcset);CHKERRQ(ierr); 975efd4aadfSBarry Smith if (pcset && (!snes->npc)) { 976efd4aadfSBarry Smith ierr = SNESGetNPC(snes, &snes->npc);CHKERRQ(ierr); 97751e86f29SPeter Brune } 9783a40ed3dSBarry Smith PetscFunctionReturn(0); 9799b94acceSBarry Smith } 9809b94acceSBarry Smith 981bb9467b5SJed Brown /*@C 982d25893d9SBarry Smith SNESSetComputeApplicationContext - Sets an optional function to compute a user-defined context for 983d25893d9SBarry Smith the nonlinear solvers. 984d25893d9SBarry Smith 985d25893d9SBarry Smith Logically Collective on SNES 986d25893d9SBarry Smith 987d25893d9SBarry Smith Input Parameters: 988d25893d9SBarry Smith + snes - the SNES context 989d25893d9SBarry Smith . compute - function to compute the context 990d25893d9SBarry Smith - destroy - function to destroy the context 991d25893d9SBarry Smith 992d25893d9SBarry Smith Level: intermediate 993d25893d9SBarry Smith 994bb9467b5SJed Brown Notes: 995bb9467b5SJed Brown This function is currently not available from Fortran. 996bb9467b5SJed Brown 997d25893d9SBarry Smith .keywords: SNES, nonlinear, set, application, context 998d25893d9SBarry Smith 999d25893d9SBarry Smith .seealso: SNESGetApplicationContext(), SNESSetComputeApplicationContext(), SNESGetApplicationContext() 1000d25893d9SBarry Smith @*/ 1001d25893d9SBarry Smith PetscErrorCode SNESSetComputeApplicationContext(SNES snes,PetscErrorCode (*compute)(SNES,void**),PetscErrorCode (*destroy)(void**)) 1002d25893d9SBarry Smith { 1003d25893d9SBarry Smith PetscFunctionBegin; 1004d25893d9SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 1005d25893d9SBarry Smith snes->ops->usercompute = compute; 1006d25893d9SBarry Smith snes->ops->userdestroy = destroy; 1007d25893d9SBarry Smith PetscFunctionReturn(0); 1008d25893d9SBarry Smith } 1009a847f771SSatish Balay 1010b07ff414SBarry Smith /*@ 10119b94acceSBarry Smith SNESSetApplicationContext - Sets the optional user-defined context for 10129b94acceSBarry Smith the nonlinear solvers. 10139b94acceSBarry Smith 10143f9fe445SBarry Smith Logically Collective on SNES 1015fee21e36SBarry Smith 1016c7afd0dbSLois Curfman McInnes Input Parameters: 1017c7afd0dbSLois Curfman McInnes + snes - the SNES context 1018c7afd0dbSLois Curfman McInnes - usrP - optional user context 1019c7afd0dbSLois Curfman McInnes 102036851e7fSLois Curfman McInnes Level: intermediate 102136851e7fSLois Curfman McInnes 1022daf670e6SBarry Smith Fortran Notes: To use this from Fortran you must write a Fortran interface definition for this 1023daf670e6SBarry Smith function that tells Fortran the Fortran derived data type that you are passing in as the ctx argument. 1024daf670e6SBarry Smith 10259b94acceSBarry Smith .keywords: SNES, nonlinear, set, application, context 10269b94acceSBarry Smith 1027ecaffddaSVictor Eijkhout .seealso: SNESGetApplicationContext() 10289b94acceSBarry Smith @*/ 10297087cfbeSBarry Smith PetscErrorCode SNESSetApplicationContext(SNES snes,void *usrP) 10309b94acceSBarry Smith { 10311b2093e4SBarry Smith PetscErrorCode ierr; 1032b07ff414SBarry Smith KSP ksp; 10331b2093e4SBarry Smith 10343a40ed3dSBarry Smith PetscFunctionBegin; 10350700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 1036b07ff414SBarry Smith ierr = SNESGetKSP(snes,&ksp);CHKERRQ(ierr); 1037b07ff414SBarry Smith ierr = KSPSetApplicationContext(ksp,usrP);CHKERRQ(ierr); 10389b94acceSBarry Smith snes->user = usrP; 10393a40ed3dSBarry Smith PetscFunctionReturn(0); 10409b94acceSBarry Smith } 104174679c65SBarry Smith 1042b07ff414SBarry Smith /*@ 10439b94acceSBarry Smith SNESGetApplicationContext - Gets the user-defined context for the 10449b94acceSBarry Smith nonlinear solvers. 10459b94acceSBarry Smith 1046c7afd0dbSLois Curfman McInnes Not Collective 1047c7afd0dbSLois Curfman McInnes 10489b94acceSBarry Smith Input Parameter: 10499b94acceSBarry Smith . snes - SNES context 10509b94acceSBarry Smith 10519b94acceSBarry Smith Output Parameter: 10529b94acceSBarry Smith . usrP - user context 10539b94acceSBarry Smith 1054daf670e6SBarry Smith Fortran Notes: To use this from Fortran you must write a Fortran interface definition for this 1055daf670e6SBarry Smith function that tells Fortran the Fortran derived data type that you are passing in as the ctx argument. 1056daf670e6SBarry Smith 105736851e7fSLois Curfman McInnes Level: intermediate 105836851e7fSLois Curfman McInnes 10599b94acceSBarry Smith .keywords: SNES, nonlinear, get, application, context 10609b94acceSBarry Smith 10619b94acceSBarry Smith .seealso: SNESSetApplicationContext() 10629b94acceSBarry Smith @*/ 1063e71120c6SJed Brown PetscErrorCode SNESGetApplicationContext(SNES snes,void *usrP) 10649b94acceSBarry Smith { 10653a40ed3dSBarry Smith PetscFunctionBegin; 10660700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 1067e71120c6SJed Brown *(void**)usrP = snes->user; 10683a40ed3dSBarry Smith PetscFunctionReturn(0); 10699b94acceSBarry Smith } 107074679c65SBarry Smith 10719b94acceSBarry Smith /*@ 10723565c898SBarry Smith SNESSetUseMatrixFree - indicates that SNES should use matrix free finite difference matrix vector products internally to apply 10733565c898SBarry Smith the Jacobian. 10743565c898SBarry Smith 10753565c898SBarry Smith Collective on SNES 10763565c898SBarry Smith 10773565c898SBarry Smith Input Parameters: 10783565c898SBarry Smith + snes - SNES context 10793565c898SBarry 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 10803565c898SBarry Smith - mf_operator - use matrix-free only for the Amat used by SNESSetJacobian(), this means the user provided Pmat will continue to be used 10813565c898SBarry Smith 10823565c898SBarry Smith Options Database: 10833565c898SBarry Smith + -snes_mf - use matrix free for both the mat and pmat operator 10843565c898SBarry Smith - -snes_mf_operator - use matrix free only for the mat operator 10853565c898SBarry Smith 10863565c898SBarry Smith Level: intermediate 10873565c898SBarry Smith 10883565c898SBarry Smith .keywords: SNES, nonlinear, get, iteration, number, 10893565c898SBarry Smith 10903565c898SBarry Smith .seealso: SNESGetUseMatrixFree(), MatCreateSNESMF() 10913565c898SBarry Smith @*/ 10923565c898SBarry Smith PetscErrorCode SNESSetUseMatrixFree(SNES snes,PetscBool mf_operator,PetscBool mf) 10933565c898SBarry Smith { 10943565c898SBarry Smith PetscFunctionBegin; 10953565c898SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 10963565c898SBarry Smith if (mf && !mf_operator) SETERRQ(PetscObjectComm((PetscObject)snes),PETSC_ERR_ARG_INCOMP,"If using mf must also use mf_operator"); 10973565c898SBarry Smith snes->mf = mf; 10983565c898SBarry Smith snes->mf_operator = mf_operator; 10993565c898SBarry Smith PetscFunctionReturn(0); 11003565c898SBarry Smith } 11013565c898SBarry Smith 11023565c898SBarry Smith /*@ 11033565c898SBarry Smith SNESGetUseMatrixFree - indicates if the SNES uses matrix free finite difference matrix vector products to apply 11043565c898SBarry Smith the Jacobian. 11053565c898SBarry Smith 11063565c898SBarry Smith Collective on SNES 11073565c898SBarry Smith 11083565c898SBarry Smith Input Parameter: 11093565c898SBarry Smith . snes - SNES context 11103565c898SBarry Smith 11113565c898SBarry Smith Output Parameters: 11123565c898SBarry 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 11133565c898SBarry Smith - mf_operator - use matrix-free only for the Amat used by SNESSetJacobian(), this means the user provided Pmat will continue to be used 11143565c898SBarry Smith 11153565c898SBarry Smith Options Database: 11163565c898SBarry Smith + -snes_mf - use matrix free for both the mat and pmat operator 11173565c898SBarry Smith - -snes_mf_operator - use matrix free only for the mat operator 11183565c898SBarry Smith 11193565c898SBarry Smith Level: intermediate 11203565c898SBarry Smith 11213565c898SBarry Smith .keywords: SNES, nonlinear, get, iteration, number, 11223565c898SBarry Smith 11233565c898SBarry Smith .seealso: SNESSetUseMatrixFree(), MatCreateSNESMF() 11243565c898SBarry Smith @*/ 11253565c898SBarry Smith PetscErrorCode SNESGetUseMatrixFree(SNES snes,PetscBool *mf_operator,PetscBool *mf) 11263565c898SBarry Smith { 11273565c898SBarry Smith PetscFunctionBegin; 11283565c898SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 11293565c898SBarry Smith if (mf) *mf = snes->mf; 11303565c898SBarry Smith if (mf_operator) *mf_operator = snes->mf_operator; 11313565c898SBarry Smith PetscFunctionReturn(0); 11323565c898SBarry Smith } 11333565c898SBarry Smith 11343565c898SBarry Smith /*@ 1135c8228a4eSBarry Smith SNESGetIterationNumber - Gets the number of nonlinear iterations completed 1136c8228a4eSBarry Smith at this time. 11379b94acceSBarry Smith 1138c7afd0dbSLois Curfman McInnes Not Collective 1139c7afd0dbSLois Curfman McInnes 11409b94acceSBarry Smith Input Parameter: 11419b94acceSBarry Smith . snes - SNES context 11429b94acceSBarry Smith 11439b94acceSBarry Smith Output Parameter: 11449b94acceSBarry Smith . iter - iteration number 11459b94acceSBarry Smith 1146c8228a4eSBarry Smith Notes: 1147c8228a4eSBarry Smith For example, during the computation of iteration 2 this would return 1. 1148c8228a4eSBarry Smith 1149c8228a4eSBarry Smith This is useful for using lagged Jacobians (where one does not recompute the 115008405cd6SLois Curfman McInnes Jacobian at each SNES iteration). For example, the code 115108405cd6SLois Curfman McInnes .vb 115208405cd6SLois Curfman McInnes ierr = SNESGetIterationNumber(snes,&it); 115308405cd6SLois Curfman McInnes if (!(it % 2)) { 115408405cd6SLois Curfman McInnes [compute Jacobian here] 115508405cd6SLois Curfman McInnes } 115608405cd6SLois Curfman McInnes .ve 1157c8228a4eSBarry Smith can be used in your ComputeJacobian() function to cause the Jacobian to be 115808405cd6SLois Curfman McInnes recomputed every second SNES iteration. 1159c8228a4eSBarry Smith 116036851e7fSLois Curfman McInnes Level: intermediate 116136851e7fSLois Curfman McInnes 11622b668275SBarry Smith .keywords: SNES, nonlinear, get, iteration, number, 11632b668275SBarry Smith 116471dbe336SPeter Brune .seealso: SNESGetLinearSolveIterations() 11659b94acceSBarry Smith @*/ 11667087cfbeSBarry Smith PetscErrorCode SNESGetIterationNumber(SNES snes,PetscInt *iter) 11679b94acceSBarry Smith { 11683a40ed3dSBarry Smith PetscFunctionBegin; 11690700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 11704482741eSBarry Smith PetscValidIntPointer(iter,2); 11719b94acceSBarry Smith *iter = snes->iter; 11723a40ed3dSBarry Smith PetscFunctionReturn(0); 11739b94acceSBarry Smith } 117474679c65SBarry Smith 1175360c497dSPeter Brune /*@ 1176360c497dSPeter Brune SNESSetIterationNumber - Sets the current iteration number. 1177360c497dSPeter Brune 1178360c497dSPeter Brune Not Collective 1179360c497dSPeter Brune 1180360c497dSPeter Brune Input Parameter: 1181360c497dSPeter Brune . snes - SNES context 1182360c497dSPeter Brune . iter - iteration number 1183360c497dSPeter Brune 1184360c497dSPeter Brune Level: developer 1185360c497dSPeter Brune 1186360c497dSPeter Brune .keywords: SNES, nonlinear, set, iteration, number, 1187360c497dSPeter Brune 118871dbe336SPeter Brune .seealso: SNESGetLinearSolveIterations() 1189360c497dSPeter Brune @*/ 1190360c497dSPeter Brune PetscErrorCode SNESSetIterationNumber(SNES snes,PetscInt iter) 1191360c497dSPeter Brune { 1192360c497dSPeter Brune PetscErrorCode ierr; 1193360c497dSPeter Brune 1194360c497dSPeter Brune PetscFunctionBegin; 1195360c497dSPeter Brune PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 1196e04113cfSBarry Smith ierr = PetscObjectSAWsTakeAccess((PetscObject)snes);CHKERRQ(ierr); 1197360c497dSPeter Brune snes->iter = iter; 1198e04113cfSBarry Smith ierr = PetscObjectSAWsGrantAccess((PetscObject)snes);CHKERRQ(ierr); 1199360c497dSPeter Brune PetscFunctionReturn(0); 1200360c497dSPeter Brune } 1201360c497dSPeter Brune 12029b94acceSBarry Smith /*@ 1203b850b91aSLisandro Dalcin SNESGetNonlinearStepFailures - Gets the number of unsuccessful steps 12049b94acceSBarry Smith attempted by the nonlinear solver. 12059b94acceSBarry Smith 1206c7afd0dbSLois Curfman McInnes Not Collective 1207c7afd0dbSLois Curfman McInnes 12089b94acceSBarry Smith Input Parameter: 12099b94acceSBarry Smith . snes - SNES context 12109b94acceSBarry Smith 12119b94acceSBarry Smith Output Parameter: 12129b94acceSBarry Smith . nfails - number of unsuccessful steps attempted 12139b94acceSBarry Smith 1214c96a6f78SLois Curfman McInnes Notes: 1215c96a6f78SLois Curfman McInnes This counter is reset to zero for each successive call to SNESSolve(). 1216c96a6f78SLois Curfman McInnes 121736851e7fSLois Curfman McInnes Level: intermediate 121836851e7fSLois Curfman McInnes 12199b94acceSBarry Smith .keywords: SNES, nonlinear, get, number, unsuccessful, steps 122058ebbce7SBarry Smith 1221e1c61ce8SBarry Smith .seealso: SNESGetMaxLinearSolveFailures(), SNESGetLinearSolveIterations(), SNESSetMaxLinearSolveFailures(), SNESGetLinearSolveFailures(), 122258ebbce7SBarry Smith SNESSetMaxNonlinearStepFailures(), SNESGetMaxNonlinearStepFailures() 12239b94acceSBarry Smith @*/ 12247087cfbeSBarry Smith PetscErrorCode SNESGetNonlinearStepFailures(SNES snes,PetscInt *nfails) 12259b94acceSBarry Smith { 12263a40ed3dSBarry Smith PetscFunctionBegin; 12270700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 12284482741eSBarry Smith PetscValidIntPointer(nfails,2); 122950ffb88aSMatthew Knepley *nfails = snes->numFailures; 123050ffb88aSMatthew Knepley PetscFunctionReturn(0); 123150ffb88aSMatthew Knepley } 123250ffb88aSMatthew Knepley 123350ffb88aSMatthew Knepley /*@ 1234b850b91aSLisandro Dalcin SNESSetMaxNonlinearStepFailures - Sets the maximum number of unsuccessful steps 123550ffb88aSMatthew Knepley attempted by the nonlinear solver before it gives up. 123650ffb88aSMatthew Knepley 123750ffb88aSMatthew Knepley Not Collective 123850ffb88aSMatthew Knepley 123950ffb88aSMatthew Knepley Input Parameters: 124050ffb88aSMatthew Knepley + snes - SNES context 124150ffb88aSMatthew Knepley - maxFails - maximum of unsuccessful steps 124250ffb88aSMatthew Knepley 124350ffb88aSMatthew Knepley Level: intermediate 124450ffb88aSMatthew Knepley 124550ffb88aSMatthew Knepley .keywords: SNES, nonlinear, set, maximum, unsuccessful, steps 124658ebbce7SBarry Smith 1247e1c61ce8SBarry Smith .seealso: SNESGetMaxLinearSolveFailures(), SNESGetLinearSolveIterations(), SNESSetMaxLinearSolveFailures(), SNESGetLinearSolveFailures(), 124858ebbce7SBarry Smith SNESGetMaxNonlinearStepFailures(), SNESGetNonlinearStepFailures() 124950ffb88aSMatthew Knepley @*/ 12507087cfbeSBarry Smith PetscErrorCode SNESSetMaxNonlinearStepFailures(SNES snes, PetscInt maxFails) 125150ffb88aSMatthew Knepley { 125250ffb88aSMatthew Knepley PetscFunctionBegin; 12530700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 125450ffb88aSMatthew Knepley snes->maxFailures = maxFails; 125550ffb88aSMatthew Knepley PetscFunctionReturn(0); 125650ffb88aSMatthew Knepley } 125750ffb88aSMatthew Knepley 125850ffb88aSMatthew Knepley /*@ 1259b850b91aSLisandro Dalcin SNESGetMaxNonlinearStepFailures - Gets the maximum number of unsuccessful steps 126050ffb88aSMatthew Knepley attempted by the nonlinear solver before it gives up. 126150ffb88aSMatthew Knepley 126250ffb88aSMatthew Knepley Not Collective 126350ffb88aSMatthew Knepley 126450ffb88aSMatthew Knepley Input Parameter: 126550ffb88aSMatthew Knepley . snes - SNES context 126650ffb88aSMatthew Knepley 126750ffb88aSMatthew Knepley Output Parameter: 126850ffb88aSMatthew Knepley . maxFails - maximum of unsuccessful steps 126950ffb88aSMatthew Knepley 127050ffb88aSMatthew Knepley Level: intermediate 127150ffb88aSMatthew Knepley 127250ffb88aSMatthew Knepley .keywords: SNES, nonlinear, get, maximum, unsuccessful, steps 127358ebbce7SBarry Smith 1274e1c61ce8SBarry Smith .seealso: SNESGetMaxLinearSolveFailures(), SNESGetLinearSolveIterations(), SNESSetMaxLinearSolveFailures(), SNESGetLinearSolveFailures(), 127558ebbce7SBarry Smith SNESSetMaxNonlinearStepFailures(), SNESGetNonlinearStepFailures() 127658ebbce7SBarry Smith 127750ffb88aSMatthew Knepley @*/ 12787087cfbeSBarry Smith PetscErrorCode SNESGetMaxNonlinearStepFailures(SNES snes, PetscInt *maxFails) 127950ffb88aSMatthew Knepley { 128050ffb88aSMatthew Knepley PetscFunctionBegin; 12810700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 12824482741eSBarry Smith PetscValidIntPointer(maxFails,2); 128350ffb88aSMatthew Knepley *maxFails = snes->maxFailures; 12843a40ed3dSBarry Smith PetscFunctionReturn(0); 12859b94acceSBarry Smith } 1286a847f771SSatish Balay 12872541af92SBarry Smith /*@ 12882541af92SBarry Smith SNESGetNumberFunctionEvals - Gets the number of user provided function evaluations 12892541af92SBarry Smith done by SNES. 12902541af92SBarry Smith 12912541af92SBarry Smith Not Collective 12922541af92SBarry Smith 12932541af92SBarry Smith Input Parameter: 12942541af92SBarry Smith . snes - SNES context 12952541af92SBarry Smith 12962541af92SBarry Smith Output Parameter: 12972541af92SBarry Smith . nfuncs - number of evaluations 12982541af92SBarry Smith 12992541af92SBarry Smith Level: intermediate 13002541af92SBarry Smith 1301971e163fSPeter Brune Notes: Reset every time SNESSolve is called unless SNESSetCountersReset() is used. 1302971e163fSPeter Brune 13032541af92SBarry Smith .keywords: SNES, nonlinear, get, maximum, unsuccessful, steps 130458ebbce7SBarry Smith 1305971e163fSPeter Brune .seealso: SNESGetMaxLinearSolveFailures(), SNESGetLinearSolveIterations(), SNESSetMaxLinearSolveFailures(), SNESGetLinearSolveFailures(), SNESSetCountersReset() 13062541af92SBarry Smith @*/ 13077087cfbeSBarry Smith PetscErrorCode SNESGetNumberFunctionEvals(SNES snes, PetscInt *nfuncs) 13082541af92SBarry Smith { 13092541af92SBarry Smith PetscFunctionBegin; 13100700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 13112541af92SBarry Smith PetscValidIntPointer(nfuncs,2); 13122541af92SBarry Smith *nfuncs = snes->nfuncs; 13132541af92SBarry Smith PetscFunctionReturn(0); 13142541af92SBarry Smith } 13152541af92SBarry Smith 13163d4c4710SBarry Smith /*@ 13173d4c4710SBarry Smith SNESGetLinearSolveFailures - Gets the number of failed (non-converged) 13183d4c4710SBarry Smith linear solvers. 13193d4c4710SBarry Smith 13203d4c4710SBarry Smith Not Collective 13213d4c4710SBarry Smith 13223d4c4710SBarry Smith Input Parameter: 13233d4c4710SBarry Smith . snes - SNES context 13243d4c4710SBarry Smith 13253d4c4710SBarry Smith Output Parameter: 13263d4c4710SBarry Smith . nfails - number of failed solves 13273d4c4710SBarry Smith 13289d85da0cSMatthew G. Knepley Level: intermediate 13299d85da0cSMatthew G. Knepley 13309d85da0cSMatthew G. Knepley Options Database Keys: 13319d85da0cSMatthew G. Knepley . -snes_max_linear_solve_fail <num> - The number of failures before the solve is terminated 13329d85da0cSMatthew G. Knepley 13333d4c4710SBarry Smith Notes: 13343d4c4710SBarry Smith This counter is reset to zero for each successive call to SNESSolve(). 13353d4c4710SBarry Smith 13363d4c4710SBarry Smith .keywords: SNES, nonlinear, get, number, unsuccessful, steps 133758ebbce7SBarry Smith 1338e1c61ce8SBarry Smith .seealso: SNESGetMaxLinearSolveFailures(), SNESGetLinearSolveIterations(), SNESSetMaxLinearSolveFailures() 13393d4c4710SBarry Smith @*/ 13407087cfbeSBarry Smith PetscErrorCode SNESGetLinearSolveFailures(SNES snes,PetscInt *nfails) 13413d4c4710SBarry Smith { 13423d4c4710SBarry Smith PetscFunctionBegin; 13430700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 13443d4c4710SBarry Smith PetscValidIntPointer(nfails,2); 13453d4c4710SBarry Smith *nfails = snes->numLinearSolveFailures; 13463d4c4710SBarry Smith PetscFunctionReturn(0); 13473d4c4710SBarry Smith } 13483d4c4710SBarry Smith 13493d4c4710SBarry Smith /*@ 13503d4c4710SBarry Smith SNESSetMaxLinearSolveFailures - the number of failed linear solve attempts 13513d4c4710SBarry Smith allowed before SNES returns with a diverged reason of SNES_DIVERGED_LINEAR_SOLVE 13523d4c4710SBarry Smith 13533f9fe445SBarry Smith Logically Collective on SNES 13543d4c4710SBarry Smith 13553d4c4710SBarry Smith Input Parameters: 13563d4c4710SBarry Smith + snes - SNES context 13573d4c4710SBarry Smith - maxFails - maximum allowed linear solve failures 13583d4c4710SBarry Smith 13593d4c4710SBarry Smith Level: intermediate 13603d4c4710SBarry Smith 13619d85da0cSMatthew G. Knepley Options Database Keys: 13629d85da0cSMatthew G. Knepley . -snes_max_linear_solve_fail <num> - The number of failures before the solve is terminated 13639d85da0cSMatthew G. Knepley 1364a6796414SBarry Smith Notes: By default this is 0; that is SNES returns on the first failed linear solve 13653d4c4710SBarry Smith 13663d4c4710SBarry Smith .keywords: SNES, nonlinear, set, maximum, unsuccessful, steps 13673d4c4710SBarry Smith 136858ebbce7SBarry Smith .seealso: SNESGetLinearSolveFailures(), SNESGetMaxLinearSolveFailures(), SNESGetLinearSolveIterations() 13693d4c4710SBarry Smith @*/ 13707087cfbeSBarry Smith PetscErrorCode SNESSetMaxLinearSolveFailures(SNES snes, PetscInt maxFails) 13713d4c4710SBarry Smith { 13723d4c4710SBarry Smith PetscFunctionBegin; 13730700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 1374c5eb9154SBarry Smith PetscValidLogicalCollectiveInt(snes,maxFails,2); 13753d4c4710SBarry Smith snes->maxLinearSolveFailures = maxFails; 13763d4c4710SBarry Smith PetscFunctionReturn(0); 13773d4c4710SBarry Smith } 13783d4c4710SBarry Smith 13793d4c4710SBarry Smith /*@ 13803d4c4710SBarry Smith SNESGetMaxLinearSolveFailures - gets the maximum number of linear solve failures that 13813d4c4710SBarry Smith are allowed before SNES terminates 13823d4c4710SBarry Smith 13833d4c4710SBarry Smith Not Collective 13843d4c4710SBarry Smith 13853d4c4710SBarry Smith Input Parameter: 13863d4c4710SBarry Smith . snes - SNES context 13873d4c4710SBarry Smith 13883d4c4710SBarry Smith Output Parameter: 13893d4c4710SBarry Smith . maxFails - maximum of unsuccessful solves allowed 13903d4c4710SBarry Smith 13913d4c4710SBarry Smith Level: intermediate 13923d4c4710SBarry Smith 13933d4c4710SBarry Smith Notes: By default this is 1; that is SNES returns on the first failed linear solve 13943d4c4710SBarry Smith 13953d4c4710SBarry Smith .keywords: SNES, nonlinear, get, maximum, unsuccessful, steps 13963d4c4710SBarry Smith 1397e1c61ce8SBarry Smith .seealso: SNESGetLinearSolveFailures(), SNESGetLinearSolveIterations(), SNESSetMaxLinearSolveFailures(), 13983d4c4710SBarry Smith @*/ 13997087cfbeSBarry Smith PetscErrorCode SNESGetMaxLinearSolveFailures(SNES snes, PetscInt *maxFails) 14003d4c4710SBarry Smith { 14013d4c4710SBarry Smith PetscFunctionBegin; 14020700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 14033d4c4710SBarry Smith PetscValidIntPointer(maxFails,2); 14043d4c4710SBarry Smith *maxFails = snes->maxLinearSolveFailures; 14053d4c4710SBarry Smith PetscFunctionReturn(0); 14063d4c4710SBarry Smith } 14073d4c4710SBarry Smith 1408c96a6f78SLois Curfman McInnes /*@ 1409b850b91aSLisandro Dalcin SNESGetLinearSolveIterations - Gets the total number of linear iterations 1410c96a6f78SLois Curfman McInnes used by the nonlinear solver. 1411c96a6f78SLois Curfman McInnes 1412c7afd0dbSLois Curfman McInnes Not Collective 1413c7afd0dbSLois Curfman McInnes 1414c96a6f78SLois Curfman McInnes Input Parameter: 1415c96a6f78SLois Curfman McInnes . snes - SNES context 1416c96a6f78SLois Curfman McInnes 1417c96a6f78SLois Curfman McInnes Output Parameter: 1418c96a6f78SLois Curfman McInnes . lits - number of linear iterations 1419c96a6f78SLois Curfman McInnes 1420c96a6f78SLois Curfman McInnes Notes: 1421971e163fSPeter Brune This counter is reset to zero for each successive call to SNESSolve() unless SNESSetCountersReset() is used. 1422c96a6f78SLois Curfman McInnes 1423010be392SBarry 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 1424010be392SBarry Smith then call KSPGetIterationNumber() after the failed solve. 1425010be392SBarry Smith 142636851e7fSLois Curfman McInnes Level: intermediate 142736851e7fSLois Curfman McInnes 1428c96a6f78SLois Curfman McInnes .keywords: SNES, nonlinear, get, number, linear, iterations 14292b668275SBarry Smith 143071dbe336SPeter Brune .seealso: SNESGetIterationNumber(), SNESGetLinearSolveFailures(), SNESGetMaxLinearSolveFailures(), SNESSetCountersReset() 1431c96a6f78SLois Curfman McInnes @*/ 14327087cfbeSBarry Smith PetscErrorCode SNESGetLinearSolveIterations(SNES snes,PetscInt *lits) 1433c96a6f78SLois Curfman McInnes { 14343a40ed3dSBarry Smith PetscFunctionBegin; 14350700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 14364482741eSBarry Smith PetscValidIntPointer(lits,2); 1437c96a6f78SLois Curfman McInnes *lits = snes->linear_its; 14383a40ed3dSBarry Smith PetscFunctionReturn(0); 1439c96a6f78SLois Curfman McInnes } 1440c96a6f78SLois Curfman McInnes 1441971e163fSPeter Brune /*@ 1442971e163fSPeter Brune SNESSetCountersReset - Sets whether or not the counters for linear iterations and function evaluations 1443971e163fSPeter Brune are reset every time SNESSolve() is called. 1444971e163fSPeter Brune 1445971e163fSPeter Brune Logically Collective on SNES 1446971e163fSPeter Brune 1447971e163fSPeter Brune Input Parameter: 1448971e163fSPeter Brune + snes - SNES context 1449971e163fSPeter Brune - reset - whether to reset the counters or not 1450971e163fSPeter Brune 1451971e163fSPeter Brune Notes: 1452fa19ca70SBarry Smith This defaults to PETSC_TRUE 1453971e163fSPeter Brune 1454971e163fSPeter Brune Level: developer 1455971e163fSPeter Brune 1456971e163fSPeter Brune .keywords: SNES, nonlinear, set, reset, number, linear, iterations 1457971e163fSPeter Brune 1458734794cfSBarry Smith .seealso: SNESGetNumberFunctionEvals(), SNESGetLinearSolveIterations(), SNESGetNPC() 1459971e163fSPeter Brune @*/ 1460971e163fSPeter Brune PetscErrorCode SNESSetCountersReset(SNES snes,PetscBool reset) 1461971e163fSPeter Brune { 1462971e163fSPeter Brune PetscFunctionBegin; 1463971e163fSPeter Brune PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 1464971e163fSPeter Brune PetscValidLogicalCollectiveBool(snes,reset,2); 1465971e163fSPeter Brune snes->counters_reset = reset; 1466971e163fSPeter Brune PetscFunctionReturn(0); 1467971e163fSPeter Brune } 1468971e163fSPeter Brune 146982bf6240SBarry Smith 14702999313aSBarry Smith /*@ 14712999313aSBarry Smith SNESSetKSP - Sets a KSP context for the SNES object to use 14722999313aSBarry Smith 14732999313aSBarry Smith Not Collective, but the SNES and KSP objects must live on the same MPI_Comm 14742999313aSBarry Smith 14752999313aSBarry Smith Input Parameters: 14762999313aSBarry Smith + snes - the SNES context 14772999313aSBarry Smith - ksp - the KSP context 14782999313aSBarry Smith 14792999313aSBarry Smith Notes: 14802999313aSBarry Smith The SNES object already has its KSP object, you can obtain with SNESGetKSP() 14812999313aSBarry Smith so this routine is rarely needed. 14822999313aSBarry Smith 14832999313aSBarry Smith The KSP object that is already in the SNES object has its reference count 14842999313aSBarry Smith decreased by one. 14852999313aSBarry Smith 14862999313aSBarry Smith Level: developer 14872999313aSBarry Smith 14882999313aSBarry Smith .keywords: SNES, nonlinear, get, KSP, context 14892999313aSBarry Smith 14902999313aSBarry Smith .seealso: KSPGetPC(), SNESCreate(), KSPCreate(), SNESSetKSP() 14912999313aSBarry Smith @*/ 14927087cfbeSBarry Smith PetscErrorCode SNESSetKSP(SNES snes,KSP ksp) 14932999313aSBarry Smith { 14942999313aSBarry Smith PetscErrorCode ierr; 14952999313aSBarry Smith 14962999313aSBarry Smith PetscFunctionBegin; 14970700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 14980700a824SBarry Smith PetscValidHeaderSpecific(ksp,KSP_CLASSID,2); 14992999313aSBarry Smith PetscCheckSameComm(snes,1,ksp,2); 15007dcf0eaaSdalcinl ierr = PetscObjectReference((PetscObject)ksp);CHKERRQ(ierr); 1501906ed7ccSBarry Smith if (snes->ksp) {ierr = PetscObjectDereference((PetscObject)snes->ksp);CHKERRQ(ierr);} 15022999313aSBarry Smith snes->ksp = ksp; 15032999313aSBarry Smith PetscFunctionReturn(0); 15042999313aSBarry Smith } 15052999313aSBarry Smith 15069b94acceSBarry Smith /* -----------------------------------------------------------*/ 150752baeb72SSatish Balay /*@ 15089b94acceSBarry Smith SNESCreate - Creates a nonlinear solver context. 15099b94acceSBarry Smith 1510c7afd0dbSLois Curfman McInnes Collective on MPI_Comm 1511c7afd0dbSLois Curfman McInnes 1512c7afd0dbSLois Curfman McInnes Input Parameters: 1513906ed7ccSBarry Smith . comm - MPI communicator 15149b94acceSBarry Smith 15159b94acceSBarry Smith Output Parameter: 15169b94acceSBarry Smith . outsnes - the new SNES context 15179b94acceSBarry Smith 1518c7afd0dbSLois Curfman McInnes Options Database Keys: 1519c7afd0dbSLois Curfman McInnes + -snes_mf - Activates default matrix-free Jacobian-vector products, 1520c7afd0dbSLois Curfman McInnes and no preconditioning matrix 1521c7afd0dbSLois Curfman McInnes . -snes_mf_operator - Activates default matrix-free Jacobian-vector 1522c7afd0dbSLois Curfman McInnes products, and a user-provided preconditioning matrix 1523c7afd0dbSLois Curfman McInnes as set by SNESSetJacobian() 1524c7afd0dbSLois Curfman McInnes - -snes_fd - Uses (slow!) finite differences to compute Jacobian 1525c1f60f51SBarry Smith 152636851e7fSLois Curfman McInnes Level: beginner 152736851e7fSLois Curfman McInnes 1528efd4aadfSBarry Smith Developer Notes: SNES always creates a KSP object even though many SNES methods do not use it. This is 1529efd4aadfSBarry Smith unfortunate and should be fixed at some point. The flag snes->usesksp indicates if the 1530efd4aadfSBarry Smith particular method does use KSP and regulates if the information about the KSP is printed 1531efd4aadfSBarry Smith in SNESView(). TSSetFromOptions() does call SNESSetFromOptions() which can lead to users being confused 1532efd4aadfSBarry Smith by help messages about meaningless SNES options. 1533efd4aadfSBarry Smith 1534efd4aadfSBarry Smith SNES always creates the snes->kspconvctx even though it is used by only one type. This should 1535efd4aadfSBarry Smith be fixed. 1536efd4aadfSBarry Smith 15379b94acceSBarry Smith .keywords: SNES, nonlinear, create, context 15389b94acceSBarry Smith 1539a8054027SBarry Smith .seealso: SNESSolve(), SNESDestroy(), SNES, SNESSetLagPreconditioner() 1540a8054027SBarry Smith 15419b94acceSBarry Smith @*/ 15427087cfbeSBarry Smith PetscErrorCode SNESCreate(MPI_Comm comm,SNES *outsnes) 15439b94acceSBarry Smith { 1544dfbe8321SBarry Smith PetscErrorCode ierr; 15459b94acceSBarry Smith SNES snes; 1546fa9f3622SBarry Smith SNESKSPEW *kctx; 154737fcc0dbSBarry Smith 15483a40ed3dSBarry Smith PetscFunctionBegin; 1549ed1caa07SMatthew Knepley PetscValidPointer(outsnes,2); 15500298fd71SBarry Smith *outsnes = NULL; 1551607a6623SBarry Smith ierr = SNESInitializePackage();CHKERRQ(ierr); 15528ba1e511SMatthew Knepley 155373107ff1SLisandro Dalcin ierr = PetscHeaderCreate(snes,SNES_CLASSID,"SNES","Nonlinear solver","SNES",comm,SNESDestroy,SNESView);CHKERRQ(ierr); 15547adad957SLisandro Dalcin 15558d359177SBarry Smith snes->ops->converged = SNESConvergedDefault; 15562c155ee1SBarry Smith snes->usesksp = PETSC_TRUE; 155788976e71SPeter Brune snes->tolerancesset = PETSC_FALSE; 15589b94acceSBarry Smith snes->max_its = 50; 15599750a799SBarry Smith snes->max_funcs = 10000; 15609b94acceSBarry Smith snes->norm = 0.0; 1561365a6726SPeter Brune snes->normschedule = SNES_NORM_ALWAYS; 15626c67d002SPeter Brune snes->functype = SNES_FUNCTION_DEFAULT; 15633a2046daSBarry Smith #if defined(PETSC_USE_REAL_SINGLE) 15643a2046daSBarry Smith snes->rtol = 1.e-5; 15653a2046daSBarry Smith #else 1566b4874afaSBarry Smith snes->rtol = 1.e-8; 15673a2046daSBarry Smith #endif 1568b4874afaSBarry Smith snes->ttol = 0.0; 15693a2046daSBarry Smith #if defined(PETSC_USE_REAL_SINGLE) 15703a2046daSBarry Smith snes->abstol = 1.e-25; 15713a2046daSBarry Smith #else 157270441072SBarry Smith snes->abstol = 1.e-50; 15733a2046daSBarry Smith #endif 15747cd0ae37SLisandro Dalcin #if defined(PETSC_USE_REAL_SINGLE) 15757cd0ae37SLisandro Dalcin snes->stol = 1.e-5; 15767cd0ae37SLisandro Dalcin #else 1577c60f73f4SPeter Brune snes->stol = 1.e-8; 15787cd0ae37SLisandro Dalcin #endif 15793a2046daSBarry Smith #if defined(PETSC_USE_REAL_SINGLE) 15803a2046daSBarry Smith snes->deltatol = 1.e-6; 15813a2046daSBarry Smith #else 15824b27c08aSLois Curfman McInnes snes->deltatol = 1.e-12; 15833a2046daSBarry Smith #endif 1584e37c518bSBarry Smith snes->divtol = 1.e4; 1585e37c518bSBarry Smith snes->rnorm0 = 0; 15869b94acceSBarry Smith snes->nfuncs = 0; 158750ffb88aSMatthew Knepley snes->numFailures = 0; 158850ffb88aSMatthew Knepley snes->maxFailures = 1; 15897a00f4a9SLois Curfman McInnes snes->linear_its = 0; 1590e35cf81dSBarry Smith snes->lagjacobian = 1; 159137ec4e1aSPeter Brune snes->jac_iter = 0; 159237ec4e1aSPeter Brune snes->lagjac_persist = PETSC_FALSE; 1593a8054027SBarry Smith snes->lagpreconditioner = 1; 159437ec4e1aSPeter Brune snes->pre_iter = 0; 159537ec4e1aSPeter Brune snes->lagpre_persist = PETSC_FALSE; 1596639f9d9dSBarry Smith snes->numbermonitors = 0; 15979b94acceSBarry Smith snes->data = 0; 15984dc4c822SBarry Smith snes->setupcalled = PETSC_FALSE; 1599186905e3SBarry Smith snes->ksp_ewconv = PETSC_FALSE; 16006f24a144SLois Curfman McInnes snes->nwork = 0; 160158c9b817SLisandro Dalcin snes->work = 0; 160258c9b817SLisandro Dalcin snes->nvwork = 0; 160358c9b817SLisandro Dalcin snes->vwork = 0; 1604758f92a0SBarry Smith snes->conv_hist_len = 0; 1605758f92a0SBarry Smith snes->conv_hist_max = 0; 16060298fd71SBarry Smith snes->conv_hist = NULL; 16070298fd71SBarry Smith snes->conv_hist_its = NULL; 1608758f92a0SBarry Smith snes->conv_hist_reset = PETSC_TRUE; 1609971e163fSPeter Brune snes->counters_reset = PETSC_TRUE; 1610e4ed7901SPeter Brune snes->vec_func_init_set = PETSC_FALSE; 1611184914b5SBarry Smith snes->reason = SNES_CONVERGED_ITERATING; 1612efd4aadfSBarry Smith snes->npcside = PC_RIGHT; 1613c40d0f55SPeter Brune 1614d8f46077SPeter Brune snes->mf = PETSC_FALSE; 1615d8f46077SPeter Brune snes->mf_operator = PETSC_FALSE; 1616d8f46077SPeter Brune snes->mf_version = 1; 1617d8f46077SPeter Brune 16183d4c4710SBarry Smith snes->numLinearSolveFailures = 0; 16193d4c4710SBarry Smith snes->maxLinearSolveFailures = 1; 16203d4c4710SBarry Smith 1621349187a7SBarry Smith snes->vizerotolerance = 1.e-8; 1622349187a7SBarry Smith 16234fc747eaSLawrence Mitchell /* Set this to true if the implementation of SNESSolve_XXX does compute the residual at the final solution. */ 16244fc747eaSLawrence Mitchell snes->alwayscomputesfinalresidual = PETSC_FALSE; 16254fc747eaSLawrence Mitchell 16269b94acceSBarry Smith /* Create context to compute Eisenstat-Walker relative tolerance for KSP */ 1627b00a9115SJed Brown ierr = PetscNewLog(snes,&kctx);CHKERRQ(ierr); 1628f5af7f23SKarl Rupp 16299b94acceSBarry Smith snes->kspconvctx = (void*)kctx; 16309b94acceSBarry Smith kctx->version = 2; 16319b94acceSBarry Smith kctx->rtol_0 = .3; /* Eisenstat and Walker suggest rtol_0=.5, but 16329b94acceSBarry Smith this was too large for some test cases */ 163375567043SBarry Smith kctx->rtol_last = 0.0; 16349b94acceSBarry Smith kctx->rtol_max = .9; 16359b94acceSBarry Smith kctx->gamma = 1.0; 163662d1f40fSBarry Smith kctx->alpha = .5*(1.0 + PetscSqrtReal(5.0)); 163771f87433Sdalcinl kctx->alpha2 = kctx->alpha; 16389b94acceSBarry Smith kctx->threshold = .1; 163975567043SBarry Smith kctx->lresid_last = 0.0; 164075567043SBarry Smith kctx->norm_last = 0.0; 16419b94acceSBarry Smith 16429b94acceSBarry Smith *outsnes = snes; 16433a40ed3dSBarry Smith PetscFunctionReturn(0); 16449b94acceSBarry Smith } 16459b94acceSBarry Smith 164688f0584fSBarry Smith /*MC 1647411c0326SBarry Smith SNESFunction - Functional form used to convey the nonlinear function to be solved by SNES 164888f0584fSBarry Smith 164988f0584fSBarry Smith Synopsis: 1650411c0326SBarry Smith #include "petscsnes.h" 1651411c0326SBarry Smith PetscErrorCode SNESFunction(SNES snes,Vec x,Vec f,void *ctx); 165288f0584fSBarry Smith 165388f0584fSBarry Smith Input Parameters: 165488f0584fSBarry Smith + snes - the SNES context 165588f0584fSBarry Smith . x - state at which to evaluate residual 165688f0584fSBarry Smith - ctx - optional user-defined function context, passed in with SNESSetFunction() 165788f0584fSBarry Smith 165888f0584fSBarry Smith Output Parameter: 165988f0584fSBarry Smith . f - vector to put residual (function value) 166088f0584fSBarry Smith 1661878cb397SSatish Balay Level: intermediate 1662878cb397SSatish Balay 166388f0584fSBarry Smith .seealso: SNESSetFunction(), SNESGetFunction() 166488f0584fSBarry Smith M*/ 166588f0584fSBarry Smith 16669b94acceSBarry Smith /*@C 16679b94acceSBarry Smith SNESSetFunction - Sets the function evaluation routine and function 16689b94acceSBarry Smith vector for use by the SNES routines in solving systems of nonlinear 16699b94acceSBarry Smith equations. 16709b94acceSBarry Smith 16713f9fe445SBarry Smith Logically Collective on SNES 1672fee21e36SBarry Smith 1673c7afd0dbSLois Curfman McInnes Input Parameters: 1674c7afd0dbSLois Curfman McInnes + snes - the SNES context 1675c7afd0dbSLois Curfman McInnes . r - vector to store function value 1676f8b49ee9SBarry Smith . f - function evaluation routine; see SNESFunction for calling sequence details 1677c7afd0dbSLois Curfman McInnes - ctx - [optional] user-defined context for private data for the 16780298fd71SBarry Smith function evaluation routine (may be NULL) 16799b94acceSBarry Smith 16809b94acceSBarry Smith Notes: 16819b94acceSBarry Smith The Newton-like methods typically solve linear systems of the form 16829b94acceSBarry Smith $ f'(x) x = -f(x), 1683c7afd0dbSLois Curfman McInnes where f'(x) denotes the Jacobian matrix and f(x) is the function. 16849b94acceSBarry Smith 168536851e7fSLois Curfman McInnes Level: beginner 168636851e7fSLois Curfman McInnes 16879b94acceSBarry Smith .keywords: SNES, nonlinear, set, function 16889b94acceSBarry Smith 1689bf388a1fSBarry Smith .seealso: SNESGetFunction(), SNESComputeFunction(), SNESSetJacobian(), SNESSetPicard(), SNESFunction 16909b94acceSBarry Smith @*/ 1691f8b49ee9SBarry Smith PetscErrorCode SNESSetFunction(SNES snes,Vec r,PetscErrorCode (*f)(SNES,Vec,Vec,void*),void *ctx) 16929b94acceSBarry Smith { 169385385478SLisandro Dalcin PetscErrorCode ierr; 16946cab3a1bSJed Brown DM dm; 16956cab3a1bSJed Brown 16963a40ed3dSBarry Smith PetscFunctionBegin; 16970700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 1698d2a683ecSLisandro Dalcin if (r) { 1699d2a683ecSLisandro Dalcin PetscValidHeaderSpecific(r,VEC_CLASSID,2); 1700d2a683ecSLisandro Dalcin PetscCheckSameComm(snes,1,r,2); 170185385478SLisandro Dalcin ierr = PetscObjectReference((PetscObject)r);CHKERRQ(ierr); 17026bf464f9SBarry Smith ierr = VecDestroy(&snes->vec_func);CHKERRQ(ierr); 1703f5af7f23SKarl Rupp 170485385478SLisandro Dalcin snes->vec_func = r; 1705d2a683ecSLisandro Dalcin } 17066cab3a1bSJed Brown ierr = SNESGetDM(snes,&dm);CHKERRQ(ierr); 1707f8b49ee9SBarry Smith ierr = DMSNESSetFunction(dm,f,ctx);CHKERRQ(ierr); 17083a40ed3dSBarry Smith PetscFunctionReturn(0); 17099b94acceSBarry Smith } 17109b94acceSBarry Smith 1711646217ecSPeter Brune 1712e4ed7901SPeter Brune /*@C 1713e4ed7901SPeter Brune SNESSetInitialFunction - Sets the function vector to be used as the 1714e4ed7901SPeter Brune function norm at the initialization of the method. In some 1715e4ed7901SPeter Brune instances, the user has precomputed the function before calling 1716e4ed7901SPeter Brune SNESSolve. This function allows one to avoid a redundant call 1717e4ed7901SPeter Brune to SNESComputeFunction in that case. 1718e4ed7901SPeter Brune 1719e4ed7901SPeter Brune Logically Collective on SNES 1720e4ed7901SPeter Brune 1721e4ed7901SPeter Brune Input Parameters: 1722e4ed7901SPeter Brune + snes - the SNES context 1723e4ed7901SPeter Brune - f - vector to store function value 1724e4ed7901SPeter Brune 1725e4ed7901SPeter Brune Notes: 1726e4ed7901SPeter Brune This should not be modified during the solution procedure. 1727e4ed7901SPeter Brune 1728e4ed7901SPeter Brune This is used extensively in the SNESFAS hierarchy and in nonlinear preconditioning. 1729e4ed7901SPeter Brune 1730e4ed7901SPeter Brune Level: developer 1731e4ed7901SPeter Brune 1732e4ed7901SPeter Brune .keywords: SNES, nonlinear, set, function 1733e4ed7901SPeter Brune 1734e4ed7901SPeter Brune .seealso: SNESSetFunction(), SNESComputeFunction(), SNESSetInitialFunctionNorm() 1735e4ed7901SPeter Brune @*/ 1736e4ed7901SPeter Brune PetscErrorCode SNESSetInitialFunction(SNES snes, Vec f) 1737e4ed7901SPeter Brune { 1738e4ed7901SPeter Brune PetscErrorCode ierr; 1739e4ed7901SPeter Brune Vec vec_func; 1740e4ed7901SPeter Brune 1741e4ed7901SPeter Brune PetscFunctionBegin; 1742e4ed7901SPeter Brune PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 1743e4ed7901SPeter Brune PetscValidHeaderSpecific(f,VEC_CLASSID,2); 1744e4ed7901SPeter Brune PetscCheckSameComm(snes,1,f,2); 1745efd4aadfSBarry Smith if (snes->npcside== PC_LEFT && snes->functype == SNES_FUNCTION_PRECONDITIONED) { 1746902f982fSPeter Brune snes->vec_func_init_set = PETSC_FALSE; 1747902f982fSPeter Brune PetscFunctionReturn(0); 1748902f982fSPeter Brune } 17490298fd71SBarry Smith ierr = SNESGetFunction(snes,&vec_func,NULL,NULL);CHKERRQ(ierr); 1750e4ed7901SPeter Brune ierr = VecCopy(f, vec_func);CHKERRQ(ierr); 1751f5af7f23SKarl Rupp 1752217b9c2eSPeter Brune snes->vec_func_init_set = PETSC_TRUE; 1753e4ed7901SPeter Brune PetscFunctionReturn(0); 1754e4ed7901SPeter Brune } 1755e4ed7901SPeter Brune 1756534ebe21SPeter Brune /*@ 1757365a6726SPeter Brune SNESSetNormSchedule - Sets the SNESNormSchedule used in covergence and monitoring 1758534ebe21SPeter Brune of the SNES method. 1759534ebe21SPeter Brune 1760534ebe21SPeter Brune Logically Collective on SNES 1761534ebe21SPeter Brune 1762534ebe21SPeter Brune Input Parameters: 1763534ebe21SPeter Brune + snes - the SNES context 1764365a6726SPeter Brune - normschedule - the frequency of norm computation 1765534ebe21SPeter Brune 1766517f1916SMatthew G. Knepley Options Database Key: 1767517f1916SMatthew G. Knepley . -snes_norm_schedule <none, always, initialonly, finalonly, initalfinalonly> 1768517f1916SMatthew G. Knepley 1769534ebe21SPeter Brune Notes: 1770365a6726SPeter Brune Only certain SNES methods support certain SNESNormSchedules. Most require evaluation 1771534ebe21SPeter Brune of the nonlinear function and the taking of its norm at every iteration to 1772534ebe21SPeter Brune even ensure convergence at all. However, methods such as custom Gauss-Seidel methods 1773be95d8f1SBarry Smith (SNESNGS) and the like do not require the norm of the function to be computed, and therfore 1774534ebe21SPeter Brune may either be monitored for convergence or not. As these are often used as nonlinear 1775534ebe21SPeter Brune preconditioners, monitoring the norm of their error is not a useful enterprise within 1776534ebe21SPeter Brune their solution. 1777534ebe21SPeter Brune 1778534ebe21SPeter Brune Level: developer 1779534ebe21SPeter Brune 1780534ebe21SPeter Brune .keywords: SNES, nonlinear, set, function, norm, type 1781534ebe21SPeter Brune 1782365a6726SPeter Brune .seealso: SNESGetNormSchedule(), SNESComputeFunction(), VecNorm(), SNESSetFunction(), SNESSetInitialFunction(), SNESNormSchedule 1783534ebe21SPeter Brune @*/ 1784365a6726SPeter Brune PetscErrorCode SNESSetNormSchedule(SNES snes, SNESNormSchedule normschedule) 1785534ebe21SPeter Brune { 1786534ebe21SPeter Brune PetscFunctionBegin; 1787534ebe21SPeter Brune PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 1788365a6726SPeter Brune snes->normschedule = normschedule; 1789534ebe21SPeter Brune PetscFunctionReturn(0); 1790534ebe21SPeter Brune } 1791534ebe21SPeter Brune 1792534ebe21SPeter Brune 1793534ebe21SPeter Brune /*@ 1794365a6726SPeter Brune SNESGetNormSchedule - Gets the SNESNormSchedule used in covergence and monitoring 1795534ebe21SPeter Brune of the SNES method. 1796534ebe21SPeter Brune 1797534ebe21SPeter Brune Logically Collective on SNES 1798534ebe21SPeter Brune 1799534ebe21SPeter Brune Input Parameters: 1800534ebe21SPeter Brune + snes - the SNES context 1801365a6726SPeter Brune - normschedule - the type of the norm used 1802534ebe21SPeter Brune 1803534ebe21SPeter Brune Level: advanced 1804534ebe21SPeter Brune 1805534ebe21SPeter Brune .keywords: SNES, nonlinear, set, function, norm, type 1806534ebe21SPeter Brune 1807365a6726SPeter Brune .seealso: SNESSetNormSchedule(), SNESComputeFunction(), VecNorm(), SNESSetFunction(), SNESSetInitialFunction(), SNESNormSchedule 1808534ebe21SPeter Brune @*/ 1809365a6726SPeter Brune PetscErrorCode SNESGetNormSchedule(SNES snes, SNESNormSchedule *normschedule) 1810534ebe21SPeter Brune { 1811534ebe21SPeter Brune PetscFunctionBegin; 1812534ebe21SPeter Brune PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 1813365a6726SPeter Brune *normschedule = snes->normschedule; 1814534ebe21SPeter Brune PetscFunctionReturn(0); 1815534ebe21SPeter Brune } 1816534ebe21SPeter Brune 181747073ea2SPeter Brune 1818c5ce4427SMatthew G. Knepley /*@ 1819c5ce4427SMatthew G. Knepley SNESSetFunctionNorm - Sets the last computed residual norm. 1820c5ce4427SMatthew G. Knepley 1821c5ce4427SMatthew G. Knepley Logically Collective on SNES 1822c5ce4427SMatthew G. Knepley 1823c5ce4427SMatthew G. Knepley Input Parameters: 1824c5ce4427SMatthew G. Knepley + snes - the SNES context 1825c5ce4427SMatthew G. Knepley 1826c5ce4427SMatthew G. Knepley - normschedule - the frequency of norm computation 1827c5ce4427SMatthew G. Knepley 1828c5ce4427SMatthew G. Knepley Level: developer 1829c5ce4427SMatthew G. Knepley 1830c5ce4427SMatthew G. Knepley .keywords: SNES, nonlinear, set, function, norm, type 1831c5ce4427SMatthew G. Knepley .seealso: SNESGetNormSchedule(), SNESComputeFunction(), VecNorm(), SNESSetFunction(), SNESSetInitialFunction(), SNESNormSchedule 1832c5ce4427SMatthew G. Knepley @*/ 1833c5ce4427SMatthew G. Knepley PetscErrorCode SNESSetFunctionNorm(SNES snes, PetscReal norm) 1834c5ce4427SMatthew G. Knepley { 1835c5ce4427SMatthew G. Knepley PetscFunctionBegin; 1836c5ce4427SMatthew G. Knepley PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 1837c5ce4427SMatthew G. Knepley snes->norm = norm; 1838c5ce4427SMatthew G. Knepley PetscFunctionReturn(0); 1839c5ce4427SMatthew G. Knepley } 1840c5ce4427SMatthew G. Knepley 1841c5ce4427SMatthew G. Knepley /*@ 1842c5ce4427SMatthew G. Knepley SNESGetFunctionNorm - Gets the last computed norm of the residual 1843c5ce4427SMatthew G. Knepley 1844c5ce4427SMatthew G. Knepley Not Collective 1845c5ce4427SMatthew G. Knepley 1846c5ce4427SMatthew G. Knepley Input Parameter: 1847c5ce4427SMatthew G. Knepley . snes - the SNES context 1848c5ce4427SMatthew G. Knepley 1849c5ce4427SMatthew G. Knepley Output Parameter: 1850c5ce4427SMatthew G. Knepley . norm - the last computed residual norm 1851c5ce4427SMatthew G. Knepley 1852c5ce4427SMatthew G. Knepley Level: developer 1853c5ce4427SMatthew G. Knepley 1854c5ce4427SMatthew G. Knepley .keywords: SNES, nonlinear, set, function, norm, type 1855c5ce4427SMatthew G. Knepley .seealso: SNESSetNormSchedule(), SNESComputeFunction(), VecNorm(), SNESSetFunction(), SNESSetInitialFunction(), SNESNormSchedule 1856c5ce4427SMatthew G. Knepley @*/ 1857c5ce4427SMatthew G. Knepley PetscErrorCode SNESGetFunctionNorm(SNES snes, PetscReal *norm) 1858c5ce4427SMatthew G. Knepley { 1859c5ce4427SMatthew G. Knepley PetscFunctionBegin; 1860c5ce4427SMatthew G. Knepley PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 1861c5ce4427SMatthew G. Knepley PetscValidPointer(norm, 2); 1862c5ce4427SMatthew G. Knepley *norm = snes->norm; 1863c5ce4427SMatthew G. Knepley PetscFunctionReturn(0); 1864c5ce4427SMatthew G. Knepley } 1865c5ce4427SMatthew G. Knepley 186647073ea2SPeter Brune /*@C 186747073ea2SPeter Brune SNESSetFunctionType - Sets the SNESNormSchedule used in covergence and monitoring 186847073ea2SPeter Brune of the SNES method. 186947073ea2SPeter Brune 187047073ea2SPeter Brune Logically Collective on SNES 187147073ea2SPeter Brune 187247073ea2SPeter Brune Input Parameters: 187347073ea2SPeter Brune + snes - the SNES context 187447073ea2SPeter Brune - normschedule - the frequency of norm computation 187547073ea2SPeter Brune 187647073ea2SPeter Brune Notes: 187747073ea2SPeter Brune Only certain SNES methods support certain SNESNormSchedules. Most require evaluation 187847073ea2SPeter Brune of the nonlinear function and the taking of its norm at every iteration to 187947073ea2SPeter Brune even ensure convergence at all. However, methods such as custom Gauss-Seidel methods 1880be95d8f1SBarry Smith (SNESNGS) and the like do not require the norm of the function to be computed, and therfore 188147073ea2SPeter Brune may either be monitored for convergence or not. As these are often used as nonlinear 188247073ea2SPeter Brune preconditioners, monitoring the norm of their error is not a useful enterprise within 188347073ea2SPeter Brune their solution. 188447073ea2SPeter Brune 188547073ea2SPeter Brune Level: developer 188647073ea2SPeter Brune 188747073ea2SPeter Brune .keywords: SNES, nonlinear, set, function, norm, type 188847073ea2SPeter Brune 188947073ea2SPeter Brune .seealso: SNESGetNormSchedule(), SNESComputeFunction(), VecNorm(), SNESSetFunction(), SNESSetInitialFunction(), SNESNormSchedule 189047073ea2SPeter Brune @*/ 189147073ea2SPeter Brune PetscErrorCode SNESSetFunctionType(SNES snes, SNESFunctionType type) 189247073ea2SPeter Brune { 189347073ea2SPeter Brune PetscFunctionBegin; 189447073ea2SPeter Brune PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 189547073ea2SPeter Brune snes->functype = type; 189647073ea2SPeter Brune PetscFunctionReturn(0); 189747073ea2SPeter Brune } 189847073ea2SPeter Brune 189947073ea2SPeter Brune 190047073ea2SPeter Brune /*@C 190147073ea2SPeter Brune SNESGetFunctionType - Gets the SNESNormSchedule used in covergence and monitoring 190247073ea2SPeter Brune of the SNES method. 190347073ea2SPeter Brune 190447073ea2SPeter Brune Logically Collective on SNES 190547073ea2SPeter Brune 190647073ea2SPeter Brune Input Parameters: 190747073ea2SPeter Brune + snes - the SNES context 190847073ea2SPeter Brune - normschedule - the type of the norm used 190947073ea2SPeter Brune 191047073ea2SPeter Brune Level: advanced 191147073ea2SPeter Brune 191247073ea2SPeter Brune .keywords: SNES, nonlinear, set, function, norm, type 191347073ea2SPeter Brune 191447073ea2SPeter Brune .seealso: SNESSetNormSchedule(), SNESComputeFunction(), VecNorm(), SNESSetFunction(), SNESSetInitialFunction(), SNESNormSchedule 191547073ea2SPeter Brune @*/ 191647073ea2SPeter Brune PetscErrorCode SNESGetFunctionType(SNES snes, SNESFunctionType *type) 191747073ea2SPeter Brune { 191847073ea2SPeter Brune PetscFunctionBegin; 191947073ea2SPeter Brune PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 192047073ea2SPeter Brune *type = snes->functype; 1921534ebe21SPeter Brune PetscFunctionReturn(0); 1922534ebe21SPeter Brune } 1923534ebe21SPeter Brune 1924bf388a1fSBarry Smith /*MC 1925be95d8f1SBarry Smith SNESNGSFunction - function used to convey a Gauss-Seidel sweep on the nonlinear function 1926bf388a1fSBarry Smith 1927bf388a1fSBarry Smith Synopsis: 1928aaa7dc30SBarry Smith #include <petscsnes.h> 1929be95d8f1SBarry Smith $ SNESNGSFunction(SNES snes,Vec x,Vec b,void *ctx); 1930bf388a1fSBarry Smith 1931bf388a1fSBarry Smith + X - solution vector 1932bf388a1fSBarry Smith . B - RHS vector 1933bf388a1fSBarry Smith - ctx - optional user-defined Gauss-Seidel context 1934bf388a1fSBarry Smith 1935878cb397SSatish Balay Level: intermediate 1936878cb397SSatish Balay 1937be95d8f1SBarry Smith .seealso: SNESSetNGS(), SNESGetNGS() 1938bf388a1fSBarry Smith M*/ 1939bf388a1fSBarry Smith 1940c79ef259SPeter Brune /*@C 1941be95d8f1SBarry Smith SNESSetNGS - Sets the user nonlinear Gauss-Seidel routine for 1942c79ef259SPeter Brune use with composed nonlinear solvers. 1943c79ef259SPeter Brune 1944c79ef259SPeter Brune Input Parameters: 1945c79ef259SPeter Brune + snes - the SNES context 1946be95d8f1SBarry Smith . f - function evaluation routine to apply Gauss-Seidel see SNESNGSFunction 1947c79ef259SPeter Brune - ctx - [optional] user-defined context for private data for the 19480298fd71SBarry Smith smoother evaluation routine (may be NULL) 1949c79ef259SPeter Brune 1950c79ef259SPeter Brune Notes: 1951be95d8f1SBarry Smith The NGS routines are used by the composed nonlinear solver to generate 1952c79ef259SPeter Brune a problem appropriate update to the solution, particularly FAS. 1953c79ef259SPeter Brune 1954d28543b3SPeter Brune Level: intermediate 1955c79ef259SPeter Brune 1956d28543b3SPeter Brune .keywords: SNES, nonlinear, set, Gauss-Seidel 1957c79ef259SPeter Brune 1958be95d8f1SBarry Smith .seealso: SNESGetFunction(), SNESComputeNGS() 1959c79ef259SPeter Brune @*/ 1960be95d8f1SBarry Smith PetscErrorCode SNESSetNGS(SNES snes,PetscErrorCode (*f)(SNES,Vec,Vec,void*),void *ctx) 19616cab3a1bSJed Brown { 19626cab3a1bSJed Brown PetscErrorCode ierr; 19636cab3a1bSJed Brown DM dm; 19646cab3a1bSJed Brown 1965646217ecSPeter Brune PetscFunctionBegin; 19666cab3a1bSJed Brown PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 19676cab3a1bSJed Brown ierr = SNESGetDM(snes,&dm);CHKERRQ(ierr); 1968be95d8f1SBarry Smith ierr = DMSNESSetNGS(dm,f,ctx);CHKERRQ(ierr); 1969646217ecSPeter Brune PetscFunctionReturn(0); 1970646217ecSPeter Brune } 1971646217ecSPeter Brune 197282b59a81SJed Brown PETSC_EXTERN PetscErrorCode SNESPicardComputeFunction(SNES snes,Vec x,Vec f,void *ctx) 19738b0a5094SBarry Smith { 19748b0a5094SBarry Smith PetscErrorCode ierr; 1975e03ab78fSPeter Brune DM dm; 1976942e3340SBarry Smith DMSNES sdm; 19776cab3a1bSJed Brown 19788b0a5094SBarry Smith PetscFunctionBegin; 1979e03ab78fSPeter Brune ierr = SNESGetDM(snes,&dm);CHKERRQ(ierr); 1980942e3340SBarry Smith ierr = DMGetDMSNES(dm,&sdm);CHKERRQ(ierr); 19818b0a5094SBarry Smith /* A(x)*x - b(x) */ 198222c6f798SBarry Smith if (sdm->ops->computepfunction) { 198322c6f798SBarry Smith ierr = (*sdm->ops->computepfunction)(snes,x,f,sdm->pctx);CHKERRQ(ierr); 198422c6f798SBarry Smith } else SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE, "Must call SNESSetPicard() to provide Picard function."); 1985e03ab78fSPeter Brune 198622c6f798SBarry Smith if (sdm->ops->computepjacobian) { 1987d1e9a80fSBarry Smith ierr = (*sdm->ops->computepjacobian)(snes,x,snes->jacobian,snes->jacobian_pre,sdm->pctx);CHKERRQ(ierr); 198874e1e8c1SBarry Smith } else SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE, "Must call SNESSetPicard() to provide Picard matrix."); 19898b0a5094SBarry Smith ierr = VecScale(f,-1.0);CHKERRQ(ierr); 199095eabcedSBarry Smith ierr = MatMultAdd(snes->jacobian,x,f,f);CHKERRQ(ierr); 19918b0a5094SBarry Smith PetscFunctionReturn(0); 19928b0a5094SBarry Smith } 19938b0a5094SBarry Smith 1994d1e9a80fSBarry Smith PETSC_EXTERN PetscErrorCode SNESPicardComputeJacobian(SNES snes,Vec x1,Mat J,Mat B,void *ctx) 19958b0a5094SBarry Smith { 19968b0a5094SBarry Smith PetscFunctionBegin; 1997e03ab78fSPeter Brune /* the jacobian matrix should be pre-filled in SNESPicardComputeFunction */ 19988b0a5094SBarry Smith PetscFunctionReturn(0); 19998b0a5094SBarry Smith } 20008b0a5094SBarry Smith 20018b0a5094SBarry Smith /*@C 20020d04baf8SBarry Smith SNESSetPicard - Use SNES to solve the semilinear-system A(x) x = b(x) via a Picard type iteration (Picard linearization) 20038b0a5094SBarry Smith 20048b0a5094SBarry Smith Logically Collective on SNES 20058b0a5094SBarry Smith 20068b0a5094SBarry Smith Input Parameters: 20078b0a5094SBarry Smith + snes - the SNES context 20088b0a5094SBarry Smith . r - vector to store function value 2009f8b49ee9SBarry Smith . b - function evaluation routine 2010e5d3d808SBarry Smith . Amat - matrix with which A(x) x - b(x) is to be computed 2011e5d3d808SBarry Smith . Pmat - matrix from which preconditioner is computed (usually the same as Amat) 2012411c0326SBarry Smith . J - function to compute matrix value, see SNESJacobianFunction for details on its calling sequence 20138b0a5094SBarry Smith - ctx - [optional] user-defined context for private data for the 20140298fd71SBarry Smith function evaluation routine (may be NULL) 20158b0a5094SBarry Smith 20168b0a5094SBarry Smith Notes: 2017f450aa47SBarry 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 2018f450aa47SBarry 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. 2019f450aa47SBarry Smith 20208b0a5094SBarry Smith One can call SNESSetPicard() or SNESSetFunction() (and possibly SNESSetJacobian()) but cannot call both 20218b0a5094SBarry Smith 20228b0a5094SBarry 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} 20238b0a5094SBarry 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. 20248b0a5094SBarry Smith 20258b0a5094SBarry Smith Run with -snes_mf_operator to solve the system with Newton's method using A(x^{n}) to construct the preconditioner. 20268b0a5094SBarry Smith 20270d04baf8SBarry Smith We implement the defect correction form of the Picard iteration because it converges much more generally when inexact linear solvers are used then 20280d04baf8SBarry Smith the direct Picard iteration A(x^n) x^{n+1} = b(x^n) 20298b0a5094SBarry Smith 20308b0a5094SBarry 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 20318b0a5094SBarry 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 20328b0a5094SBarry Smith different please contact us at petsc-dev@mcs.anl.gov and we'll have an entirely new argument :-). 20338b0a5094SBarry Smith 2034f450aa47SBarry Smith Level: intermediate 20358b0a5094SBarry Smith 20368b0a5094SBarry Smith .keywords: SNES, nonlinear, set, function 20378b0a5094SBarry Smith 2038411c0326SBarry Smith .seealso: SNESGetFunction(), SNESSetFunction(), SNESComputeFunction(), SNESSetJacobian(), SNESGetPicard(), SNESLineSearchPreCheckPicard(), SNESJacobianFunction 20398b0a5094SBarry Smith @*/ 2040d1e9a80fSBarry 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) 20418b0a5094SBarry Smith { 20428b0a5094SBarry Smith PetscErrorCode ierr; 2043e03ab78fSPeter Brune DM dm; 2044e03ab78fSPeter Brune 20458b0a5094SBarry Smith PetscFunctionBegin; 20468b0a5094SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 2047e03ab78fSPeter Brune ierr = SNESGetDM(snes, &dm);CHKERRQ(ierr); 2048f8b49ee9SBarry Smith ierr = DMSNESSetPicard(dm,b,J,ctx);CHKERRQ(ierr); 20498b0a5094SBarry Smith ierr = SNESSetFunction(snes,r,SNESPicardComputeFunction,ctx);CHKERRQ(ierr); 2050e5d3d808SBarry Smith ierr = SNESSetJacobian(snes,Amat,Pmat,SNESPicardComputeJacobian,ctx);CHKERRQ(ierr); 20518b0a5094SBarry Smith PetscFunctionReturn(0); 20528b0a5094SBarry Smith } 20538b0a5094SBarry Smith 20547971a8bfSPeter Brune /*@C 20557971a8bfSPeter Brune SNESGetPicard - Returns the context for the Picard iteration 20567971a8bfSPeter Brune 20577971a8bfSPeter Brune Not Collective, but Vec is parallel if SNES is parallel. Collective if Vec is requested, but has not been created yet. 20587971a8bfSPeter Brune 20597971a8bfSPeter Brune Input Parameter: 20607971a8bfSPeter Brune . snes - the SNES context 20617971a8bfSPeter Brune 20627971a8bfSPeter Brune Output Parameter: 20630298fd71SBarry Smith + r - the function (or NULL) 2064f8b49ee9SBarry Smith . f - the function (or NULL); see SNESFunction for calling sequence details 2065e4357dc4SBarry Smith . Amat - the matrix used to defined the operation A(x) x - b(x) (or NULL) 2066e4357dc4SBarry Smith . Pmat - the matrix from which the preconditioner will be constructed (or NULL) 2067f8b49ee9SBarry Smith . J - the function for matrix evaluation (or NULL); see SNESJacobianFunction for calling sequence details 20680298fd71SBarry Smith - ctx - the function context (or NULL) 20697971a8bfSPeter Brune 20707971a8bfSPeter Brune Level: advanced 20717971a8bfSPeter Brune 20727971a8bfSPeter Brune .keywords: SNES, nonlinear, get, function 20737971a8bfSPeter Brune 2074e4357dc4SBarry Smith .seealso: SNESSetPicard(), SNESGetFunction(), SNESGetJacobian(), SNESGetDM(), SNESFunction, SNESJacobianFunction 20757971a8bfSPeter Brune @*/ 2076d1e9a80fSBarry 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) 20777971a8bfSPeter Brune { 20787971a8bfSPeter Brune PetscErrorCode ierr; 20797971a8bfSPeter Brune DM dm; 20807971a8bfSPeter Brune 20817971a8bfSPeter Brune PetscFunctionBegin; 20827971a8bfSPeter Brune PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 20830298fd71SBarry Smith ierr = SNESGetFunction(snes,r,NULL,NULL);CHKERRQ(ierr); 2084e4357dc4SBarry Smith ierr = SNESGetJacobian(snes,Amat,Pmat,NULL,NULL);CHKERRQ(ierr); 20857971a8bfSPeter Brune ierr = SNESGetDM(snes,&dm);CHKERRQ(ierr); 2086f8b49ee9SBarry Smith ierr = DMSNESGetPicard(dm,f,J,ctx);CHKERRQ(ierr); 20877971a8bfSPeter Brune PetscFunctionReturn(0); 20887971a8bfSPeter Brune } 20897971a8bfSPeter Brune 2090d25893d9SBarry Smith /*@C 2091d25893d9SBarry Smith SNESSetComputeInitialGuess - Sets a routine used to compute an initial guess for the problem 2092d25893d9SBarry Smith 2093d25893d9SBarry Smith Logically Collective on SNES 2094d25893d9SBarry Smith 2095d25893d9SBarry Smith Input Parameters: 2096d25893d9SBarry Smith + snes - the SNES context 2097d25893d9SBarry Smith . func - function evaluation routine 2098d25893d9SBarry Smith - ctx - [optional] user-defined context for private data for the 20990298fd71SBarry Smith function evaluation routine (may be NULL) 2100d25893d9SBarry Smith 2101d25893d9SBarry Smith Calling sequence of func: 2102d25893d9SBarry Smith $ func (SNES snes,Vec x,void *ctx); 2103d25893d9SBarry Smith 2104d25893d9SBarry Smith . f - function vector 2105d25893d9SBarry Smith - ctx - optional user-defined function context 2106d25893d9SBarry Smith 2107d25893d9SBarry Smith Level: intermediate 2108d25893d9SBarry Smith 2109d25893d9SBarry Smith .keywords: SNES, nonlinear, set, function 2110d25893d9SBarry Smith 2111d25893d9SBarry Smith .seealso: SNESGetFunction(), SNESComputeFunction(), SNESSetJacobian() 2112d25893d9SBarry Smith @*/ 2113d25893d9SBarry Smith PetscErrorCode SNESSetComputeInitialGuess(SNES snes,PetscErrorCode (*func)(SNES,Vec,void*),void *ctx) 2114d25893d9SBarry Smith { 2115d25893d9SBarry Smith PetscFunctionBegin; 2116d25893d9SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 2117d25893d9SBarry Smith if (func) snes->ops->computeinitialguess = func; 2118d25893d9SBarry Smith if (ctx) snes->initialguessP = ctx; 2119d25893d9SBarry Smith PetscFunctionReturn(0); 2120d25893d9SBarry Smith } 2121d25893d9SBarry Smith 21223ab0aad5SBarry Smith /* --------------------------------------------------------------- */ 21231096aae1SMatthew Knepley /*@C 21241096aae1SMatthew Knepley SNESGetRhs - Gets the vector for solving F(x) = rhs. If rhs is not set 21251096aae1SMatthew Knepley it assumes a zero right hand side. 21261096aae1SMatthew Knepley 21273f9fe445SBarry Smith Logically Collective on SNES 21281096aae1SMatthew Knepley 21291096aae1SMatthew Knepley Input Parameter: 21301096aae1SMatthew Knepley . snes - the SNES context 21311096aae1SMatthew Knepley 21321096aae1SMatthew Knepley Output Parameter: 21330298fd71SBarry Smith . rhs - the right hand side vector or NULL if the right hand side vector is null 21341096aae1SMatthew Knepley 21351096aae1SMatthew Knepley Level: intermediate 21361096aae1SMatthew Knepley 21371096aae1SMatthew Knepley .keywords: SNES, nonlinear, get, function, right hand side 21381096aae1SMatthew Knepley 213985385478SLisandro Dalcin .seealso: SNESGetSolution(), SNESGetFunction(), SNESComputeFunction(), SNESSetJacobian(), SNESSetFunction() 21401096aae1SMatthew Knepley @*/ 21417087cfbeSBarry Smith PetscErrorCode SNESGetRhs(SNES snes,Vec *rhs) 21421096aae1SMatthew Knepley { 21431096aae1SMatthew Knepley PetscFunctionBegin; 21440700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 21451096aae1SMatthew Knepley PetscValidPointer(rhs,2); 214685385478SLisandro Dalcin *rhs = snes->vec_rhs; 21471096aae1SMatthew Knepley PetscFunctionReturn(0); 21481096aae1SMatthew Knepley } 21491096aae1SMatthew Knepley 21509b94acceSBarry Smith /*@ 2151bf388a1fSBarry Smith SNESComputeFunction - Calls the function that has been set with SNESSetFunction(). 21529b94acceSBarry Smith 2153c7afd0dbSLois Curfman McInnes Collective on SNES 2154c7afd0dbSLois Curfman McInnes 21559b94acceSBarry Smith Input Parameters: 2156c7afd0dbSLois Curfman McInnes + snes - the SNES context 2157c7afd0dbSLois Curfman McInnes - x - input vector 21589b94acceSBarry Smith 21599b94acceSBarry Smith Output Parameter: 21603638b69dSLois Curfman McInnes . y - function vector, as set by SNESSetFunction() 21619b94acceSBarry Smith 21621bffabb2SLois Curfman McInnes Notes: 216336851e7fSLois Curfman McInnes SNESComputeFunction() is typically used within nonlinear solvers 216436851e7fSLois Curfman McInnes implementations, so most users would not generally call this routine 216536851e7fSLois Curfman McInnes themselves. 216636851e7fSLois Curfman McInnes 216736851e7fSLois Curfman McInnes Level: developer 216836851e7fSLois Curfman McInnes 21699b94acceSBarry Smith .keywords: SNES, nonlinear, compute, function 21709b94acceSBarry Smith 2171a86d99e1SLois Curfman McInnes .seealso: SNESSetFunction(), SNESGetFunction() 21729b94acceSBarry Smith @*/ 21737087cfbeSBarry Smith PetscErrorCode SNESComputeFunction(SNES snes,Vec x,Vec y) 21749b94acceSBarry Smith { 2175dfbe8321SBarry Smith PetscErrorCode ierr; 21766cab3a1bSJed Brown DM dm; 2177942e3340SBarry Smith DMSNES sdm; 21789b94acceSBarry Smith 21793a40ed3dSBarry Smith PetscFunctionBegin; 21800700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 21810700a824SBarry Smith PetscValidHeaderSpecific(x,VEC_CLASSID,2); 21820700a824SBarry Smith PetscValidHeaderSpecific(y,VEC_CLASSID,3); 2183c9780b6fSBarry Smith PetscCheckSameComm(snes,1,x,2); 2184c9780b6fSBarry Smith PetscCheckSameComm(snes,1,y,3); 218562796dfbSBarry Smith ierr = VecValidValues(x,2,PETSC_TRUE);CHKERRQ(ierr); 2186184914b5SBarry Smith 21876cab3a1bSJed Brown ierr = SNESGetDM(snes,&dm);CHKERRQ(ierr); 2188942e3340SBarry Smith ierr = DMGetDMSNES(dm,&sdm);CHKERRQ(ierr); 218932f3f7c2SPeter Brune if (sdm->ops->computefunction) { 219094db00ebSBarry Smith if (sdm->ops->computefunction != SNESObjectiveComputeFunctionDefaultFD) { 2191ccf3c845SPeter Brune ierr = PetscLogEventBegin(SNES_FunctionEval,snes,x,y,0);CHKERRQ(ierr); 219294db00ebSBarry Smith } 21935edff71fSBarry Smith ierr = VecLockPush(x);CHKERRQ(ierr); 2194d64ed03dSBarry Smith PetscStackPush("SNES user function"); 219522c6f798SBarry Smith ierr = (*sdm->ops->computefunction)(snes,x,y,sdm->functionctx);CHKERRQ(ierr); 2196d64ed03dSBarry Smith PetscStackPop; 21975edff71fSBarry Smith ierr = VecLockPop(x);CHKERRQ(ierr); 219894db00ebSBarry Smith if (sdm->ops->computefunction != SNESObjectiveComputeFunctionDefaultFD) { 2199ccf3c845SPeter Brune ierr = PetscLogEventEnd(SNES_FunctionEval,snes,x,y,0);CHKERRQ(ierr); 220094db00ebSBarry Smith } 2201c90fad12SPeter Brune } else if (snes->vec_rhs) { 2202c90fad12SPeter Brune ierr = MatMult(snes->jacobian, x, y);CHKERRQ(ierr); 2203644e2e5bSBarry Smith } else SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE, "Must call SNESSetFunction() or SNESSetDM() before SNESComputeFunction(), likely called from SNESSolve()."); 220485385478SLisandro Dalcin if (snes->vec_rhs) { 220585385478SLisandro Dalcin ierr = VecAXPY(y,-1.0,snes->vec_rhs);CHKERRQ(ierr); 22063ab0aad5SBarry Smith } 2207ae3c334cSLois Curfman McInnes snes->nfuncs++; 2208422a814eSBarry Smith /* 2209422a814eSBarry Smith domainerror might not be set on all processes; so we tag vector locally with Inf and the next inner product or norm will 2210422a814eSBarry Smith propagate the value to all processes 2211422a814eSBarry Smith */ 2212422a814eSBarry Smith if (snes->domainerror) { 2213422a814eSBarry Smith ierr = VecSetInf(y);CHKERRQ(ierr); 2214422a814eSBarry Smith } 22153a40ed3dSBarry Smith PetscFunctionReturn(0); 22169b94acceSBarry Smith } 22179b94acceSBarry Smith 2218c79ef259SPeter Brune /*@ 2219be95d8f1SBarry Smith SNESComputeNGS - Calls the Gauss-Seidel function that has been set with SNESSetNGS(). 2220c79ef259SPeter Brune 2221c79ef259SPeter Brune Collective on SNES 2222c79ef259SPeter Brune 2223c79ef259SPeter Brune Input Parameters: 2224c79ef259SPeter Brune + snes - the SNES context 2225c79ef259SPeter Brune . x - input vector 2226c79ef259SPeter Brune - b - rhs vector 2227c79ef259SPeter Brune 2228c79ef259SPeter Brune Output Parameter: 2229c79ef259SPeter Brune . x - new solution vector 2230c79ef259SPeter Brune 2231c79ef259SPeter Brune Notes: 2232be95d8f1SBarry Smith SNESComputeNGS() is typically used within composed nonlinear solver 2233c79ef259SPeter Brune implementations, so most users would not generally call this routine 2234c79ef259SPeter Brune themselves. 2235c79ef259SPeter Brune 2236c79ef259SPeter Brune Level: developer 2237c79ef259SPeter Brune 2238c79ef259SPeter Brune .keywords: SNES, nonlinear, compute, function 2239c79ef259SPeter Brune 2240be95d8f1SBarry Smith .seealso: SNESSetNGS(), SNESComputeFunction() 2241c79ef259SPeter Brune @*/ 2242be95d8f1SBarry Smith PetscErrorCode SNESComputeNGS(SNES snes,Vec b,Vec x) 2243646217ecSPeter Brune { 2244646217ecSPeter Brune PetscErrorCode ierr; 22456cab3a1bSJed Brown DM dm; 2246942e3340SBarry Smith DMSNES sdm; 2247646217ecSPeter Brune 2248646217ecSPeter Brune PetscFunctionBegin; 2249646217ecSPeter Brune PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 2250646217ecSPeter Brune PetscValidHeaderSpecific(x,VEC_CLASSID,2); 2251646217ecSPeter Brune if (b) PetscValidHeaderSpecific(b,VEC_CLASSID,3); 2252646217ecSPeter Brune PetscCheckSameComm(snes,1,x,2); 2253646217ecSPeter Brune if (b) PetscCheckSameComm(snes,1,b,3); 225462796dfbSBarry Smith if (b) {ierr = VecValidValues(b,2,PETSC_TRUE);CHKERRQ(ierr);} 2255be95d8f1SBarry Smith ierr = PetscLogEventBegin(SNES_NGSEval,snes,x,b,0);CHKERRQ(ierr); 22566cab3a1bSJed Brown ierr = SNESGetDM(snes,&dm);CHKERRQ(ierr); 2257942e3340SBarry Smith ierr = DMGetDMSNES(dm,&sdm);CHKERRQ(ierr); 225822c6f798SBarry Smith if (sdm->ops->computegs) { 22595edff71fSBarry Smith if (b) {ierr = VecLockPush(b);CHKERRQ(ierr);} 2260be95d8f1SBarry Smith PetscStackPush("SNES user NGS"); 226122c6f798SBarry Smith ierr = (*sdm->ops->computegs)(snes,x,b,sdm->gsctx);CHKERRQ(ierr); 2262646217ecSPeter Brune PetscStackPop; 22635edff71fSBarry Smith if (b) {ierr = VecLockPop(b);CHKERRQ(ierr);} 2264be95d8f1SBarry Smith } else SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE, "Must call SNESSetNGS() before SNESComputeNGS(), likely called from SNESSolve()."); 2265be95d8f1SBarry Smith ierr = PetscLogEventEnd(SNES_NGSEval,snes,x,b,0);CHKERRQ(ierr); 2266646217ecSPeter Brune PetscFunctionReturn(0); 2267646217ecSPeter Brune } 2268646217ecSPeter Brune 226962fef451SLois Curfman McInnes /*@ 2270bf388a1fSBarry Smith SNESComputeJacobian - Computes the Jacobian matrix that has been set with SNESSetJacobian(). 227162fef451SLois Curfman McInnes 2272c7afd0dbSLois Curfman McInnes Collective on SNES and Mat 2273c7afd0dbSLois Curfman McInnes 227462fef451SLois Curfman McInnes Input Parameters: 2275c7afd0dbSLois Curfman McInnes + snes - the SNES context 2276c7afd0dbSLois Curfman McInnes - x - input vector 227762fef451SLois Curfman McInnes 227862fef451SLois Curfman McInnes Output Parameters: 2279c7afd0dbSLois Curfman McInnes + A - Jacobian matrix 2280d1e9a80fSBarry Smith - B - optional preconditioning matrix 2281fee21e36SBarry Smith 2282e35cf81dSBarry Smith Options Database Keys: 2283e35cf81dSBarry Smith + -snes_lag_preconditioner <lag> 2284693365a8SJed Brown . -snes_lag_jacobian <lag> 2285693365a8SJed Brown . -snes_compare_explicit - Compare the computed Jacobian to the finite difference Jacobian and output the differences 2286693365a8SJed Brown . -snes_compare_explicit_draw - Compare the computed Jacobian to the finite difference Jacobian and draw the result 2287693365a8SJed Brown . -snes_compare_explicit_contour - Compare the computed Jacobian to the finite difference Jacobian and draw a contour plot with the result 22884c30e9fbSJed Brown . -snes_compare_operator - Make the comparison options above use the operator instead of the preconditioning matrix 228994d6a431SBarry Smith . -snes_compare_coloring - Compute the finite difference Jacobian using coloring and display norms of difference 2290c01495d3SJed Brown . -snes_compare_coloring_display - Compute the finite differece Jacobian using coloring and display verbose differences 2291c01495d3SJed Brown . -snes_compare_coloring_threshold - Display only those matrix entries that differ by more than a given threshold 2292c01495d3SJed Brown . -snes_compare_coloring_threshold_atol - Absolute tolerance for difference in matrix entries to be displayed by -snes_compare_coloring_threshold 2293c01495d3SJed Brown . -snes_compare_coloring_threshold_rtol - Relative tolerance for difference in matrix entries to be displayed by -snes_compare_coloring_threshold 22944c30e9fbSJed Brown . -snes_compare_coloring_draw - Compute the finite differece Jacobian using coloring and draw differences 2295c01495d3SJed Brown - -snes_compare_coloring_draw_contour - Compute the finite differece Jacobian using coloring and show contours of matrices and differences 2296c01495d3SJed Brown 2297e35cf81dSBarry Smith 229862fef451SLois Curfman McInnes Notes: 229962fef451SLois Curfman McInnes Most users should not need to explicitly call this routine, as it 230062fef451SLois Curfman McInnes is used internally within the nonlinear solvers. 230162fef451SLois Curfman McInnes 230236851e7fSLois Curfman McInnes Level: developer 230336851e7fSLois Curfman McInnes 230462fef451SLois Curfman McInnes .keywords: SNES, compute, Jacobian, matrix 230562fef451SLois Curfman McInnes 2306e35cf81dSBarry Smith .seealso: SNESSetJacobian(), KSPSetOperators(), MatStructure, SNESSetLagPreconditioner(), SNESSetLagJacobian() 230762fef451SLois Curfman McInnes @*/ 2308d1e9a80fSBarry Smith PetscErrorCode SNESComputeJacobian(SNES snes,Vec X,Mat A,Mat B) 23099b94acceSBarry Smith { 2310dfbe8321SBarry Smith PetscErrorCode ierr; 2311ace3abfcSBarry Smith PetscBool flag; 23126cab3a1bSJed Brown DM dm; 2313942e3340SBarry Smith DMSNES sdm; 2314e0e3a89bSBarry Smith KSP ksp; 23153a40ed3dSBarry Smith 23163a40ed3dSBarry Smith PetscFunctionBegin; 23170700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 23180700a824SBarry Smith PetscValidHeaderSpecific(X,VEC_CLASSID,2); 2319c9780b6fSBarry Smith PetscCheckSameComm(snes,1,X,2); 232062796dfbSBarry Smith ierr = VecValidValues(X,2,PETSC_TRUE);CHKERRQ(ierr); 23216cab3a1bSJed Brown ierr = SNESGetDM(snes,&dm);CHKERRQ(ierr); 2322942e3340SBarry Smith ierr = DMGetDMSNES(dm,&sdm);CHKERRQ(ierr); 23233232da50SPeter Brune 2324ce94432eSBarry Smith if (!sdm->ops->computejacobian) SETERRQ(PetscObjectComm((PetscObject)snes),PETSC_ERR_USER,"Must call SNESSetJacobian(), DMSNESSetJacobian(), DMDASNESSetJacobianLocal(), etc"); 2325ebd3b9afSBarry Smith 2326ebd3b9afSBarry Smith /* make sure that MatAssemblyBegin/End() is called on A matrix if it is matrix free */ 2327ebd3b9afSBarry Smith 2328fe3ffe1eSBarry Smith if (snes->lagjacobian == -2) { 2329fe3ffe1eSBarry Smith snes->lagjacobian = -1; 2330f5af7f23SKarl Rupp 2331fe3ffe1eSBarry Smith ierr = PetscInfo(snes,"Recomputing Jacobian/preconditioner because lag is -2 (means compute Jacobian, but then never again) \n");CHKERRQ(ierr); 2332fe3ffe1eSBarry Smith } else if (snes->lagjacobian == -1) { 2333e35cf81dSBarry Smith ierr = PetscInfo(snes,"Reusing Jacobian/preconditioner because lag is -1\n");CHKERRQ(ierr); 233494ab13aaSBarry Smith ierr = PetscObjectTypeCompare((PetscObject)A,MATMFFD,&flag);CHKERRQ(ierr); 2335ebd3b9afSBarry Smith if (flag) { 233694ab13aaSBarry Smith ierr = MatAssemblyBegin(A,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 233794ab13aaSBarry Smith ierr = MatAssemblyEnd(A,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 2338ebd3b9afSBarry Smith } 2339e35cf81dSBarry Smith PetscFunctionReturn(0); 234037ec4e1aSPeter Brune } else if (snes->lagjacobian > 1 && (snes->iter + snes->jac_iter) % snes->lagjacobian) { 2341e35cf81dSBarry Smith ierr = PetscInfo2(snes,"Reusing Jacobian/preconditioner because lag is %D and SNES iteration is %D\n",snes->lagjacobian,snes->iter);CHKERRQ(ierr); 234294ab13aaSBarry Smith ierr = PetscObjectTypeCompare((PetscObject)A,MATMFFD,&flag);CHKERRQ(ierr); 2343ebd3b9afSBarry Smith if (flag) { 234494ab13aaSBarry Smith ierr = MatAssemblyBegin(A,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 234594ab13aaSBarry Smith ierr = MatAssemblyEnd(A,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 2346ebd3b9afSBarry Smith } 2347e35cf81dSBarry Smith PetscFunctionReturn(0); 2348e35cf81dSBarry Smith } 2349efd4aadfSBarry Smith if (snes->npc && snes->npcside== PC_LEFT) { 235094ab13aaSBarry Smith ierr = MatAssemblyBegin(A,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 235194ab13aaSBarry Smith ierr = MatAssemblyEnd(A,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 2352d728fb7dSPeter Brune PetscFunctionReturn(0); 2353d728fb7dSPeter Brune } 2354e35cf81dSBarry Smith 235594ab13aaSBarry Smith ierr = PetscLogEventBegin(SNES_JacobianEval,snes,X,A,B);CHKERRQ(ierr); 23565edff71fSBarry Smith ierr = VecLockPush(X);CHKERRQ(ierr); 2357d64ed03dSBarry Smith PetscStackPush("SNES user Jacobian function"); 2358d1e9a80fSBarry Smith ierr = (*sdm->ops->computejacobian)(snes,X,A,B,sdm->jacobianctx);CHKERRQ(ierr); 2359d64ed03dSBarry Smith PetscStackPop; 23605edff71fSBarry Smith ierr = VecLockPop(X);CHKERRQ(ierr); 236194ab13aaSBarry Smith ierr = PetscLogEventEnd(SNES_JacobianEval,snes,X,A,B);CHKERRQ(ierr); 2362a8054027SBarry Smith 2363e0e3a89bSBarry Smith /* the next line ensures that snes->ksp exists */ 2364e0e3a89bSBarry Smith ierr = SNESGetKSP(snes,&ksp);CHKERRQ(ierr); 23653b4f5425SBarry Smith if (snes->lagpreconditioner == -2) { 23663b4f5425SBarry Smith ierr = PetscInfo(snes,"Rebuilding preconditioner exactly once since lag is -2\n");CHKERRQ(ierr); 2367d1e9a80fSBarry Smith ierr = KSPSetReusePreconditioner(snes->ksp,PETSC_FALSE);CHKERRQ(ierr); 23683b4f5425SBarry Smith snes->lagpreconditioner = -1; 23693b4f5425SBarry Smith } else if (snes->lagpreconditioner == -1) { 2370a8054027SBarry Smith ierr = PetscInfo(snes,"Reusing preconditioner because lag is -1\n");CHKERRQ(ierr); 2371d1e9a80fSBarry Smith ierr = KSPSetReusePreconditioner(snes->ksp,PETSC_TRUE);CHKERRQ(ierr); 237237ec4e1aSPeter Brune } else if (snes->lagpreconditioner > 1 && (snes->iter + snes->pre_iter) % snes->lagpreconditioner) { 2373a8054027SBarry Smith ierr = PetscInfo2(snes,"Reusing preconditioner because lag is %D and SNES iteration is %D\n",snes->lagpreconditioner,snes->iter);CHKERRQ(ierr); 2374d1e9a80fSBarry Smith ierr = KSPSetReusePreconditioner(snes->ksp,PETSC_TRUE);CHKERRQ(ierr); 2375d1e9a80fSBarry Smith } else { 2376d1e9a80fSBarry Smith ierr = PetscInfo(snes,"Rebuilding preconditioner\n");CHKERRQ(ierr); 2377d1e9a80fSBarry Smith ierr = KSPSetReusePreconditioner(snes->ksp,PETSC_FALSE);CHKERRQ(ierr); 2378a8054027SBarry Smith } 2379a8054027SBarry Smith 23806d84be18SBarry Smith /* make sure user returned a correct Jacobian and preconditioner */ 238194ab13aaSBarry Smith /* PetscValidHeaderSpecific(A,MAT_CLASSID,3); 238294ab13aaSBarry Smith PetscValidHeaderSpecific(B,MAT_CLASSID,4); */ 2383693365a8SJed Brown { 2384693365a8SJed Brown PetscBool flag = PETSC_FALSE,flag_draw = PETSC_FALSE,flag_contour = PETSC_FALSE,flag_operator = PETSC_FALSE; 238527b0f280SBarry Smith ierr = PetscOptionsGetViewer(PetscObjectComm((PetscObject)snes),((PetscObject)snes)->prefix,"-snes_compare_explicit",NULL,NULL,&flag);CHKERRQ(ierr); 238627b0f280SBarry Smith ierr = PetscOptionsGetViewer(PetscObjectComm((PetscObject)snes),((PetscObject)snes)->prefix,"-snes_compare_explicit_draw",NULL,NULL,&flag_draw);CHKERRQ(ierr); 238727b0f280SBarry Smith ierr = PetscOptionsGetViewer(PetscObjectComm((PetscObject)snes),((PetscObject)snes)->prefix,"-snes_compare_explicit_draw_contour",NULL,NULL,&flag_contour);CHKERRQ(ierr); 238827b0f280SBarry Smith ierr = PetscOptionsGetViewer(PetscObjectComm((PetscObject)snes),((PetscObject)snes)->prefix,"-snes_compare_operator",NULL,NULL,&flag_operator);CHKERRQ(ierr); 2389693365a8SJed Brown if (flag || flag_draw || flag_contour) { 23900298fd71SBarry Smith Mat Bexp_mine = NULL,Bexp,FDexp; 2391693365a8SJed Brown PetscViewer vdraw,vstdout; 23926b3a5b13SJed Brown PetscBool flg; 2393693365a8SJed Brown if (flag_operator) { 239494ab13aaSBarry Smith ierr = MatComputeExplicitOperator(A,&Bexp_mine);CHKERRQ(ierr); 2395693365a8SJed Brown Bexp = Bexp_mine; 2396693365a8SJed Brown } else { 2397693365a8SJed Brown /* See if the preconditioning matrix can be viewed and added directly */ 239894ab13aaSBarry Smith ierr = PetscObjectTypeCompareAny((PetscObject)B,&flg,MATSEQAIJ,MATMPIAIJ,MATSEQDENSE,MATMPIDENSE,MATSEQBAIJ,MATMPIBAIJ,MATSEQSBAIJ,MATMPIBAIJ,"");CHKERRQ(ierr); 239994ab13aaSBarry Smith if (flg) Bexp = B; 2400693365a8SJed Brown else { 2401693365a8SJed Brown /* If the "preconditioning" matrix is itself MATSHELL or some other type without direct support */ 240294ab13aaSBarry Smith ierr = MatComputeExplicitOperator(B,&Bexp_mine);CHKERRQ(ierr); 2403693365a8SJed Brown Bexp = Bexp_mine; 2404693365a8SJed Brown } 2405693365a8SJed Brown } 2406693365a8SJed Brown ierr = MatConvert(Bexp,MATSAME,MAT_INITIAL_MATRIX,&FDexp);CHKERRQ(ierr); 2407d1e9a80fSBarry Smith ierr = SNESComputeJacobianDefault(snes,X,FDexp,FDexp,NULL);CHKERRQ(ierr); 2408ce94432eSBarry Smith ierr = PetscViewerASCIIGetStdout(PetscObjectComm((PetscObject)snes),&vstdout);CHKERRQ(ierr); 2409693365a8SJed Brown if (flag_draw || flag_contour) { 2410ce94432eSBarry Smith ierr = PetscViewerDrawOpen(PetscObjectComm((PetscObject)snes),0,"Explicit Jacobians",PETSC_DECIDE,PETSC_DECIDE,300,300,&vdraw);CHKERRQ(ierr); 2411693365a8SJed Brown if (flag_contour) {ierr = PetscViewerPushFormat(vdraw,PETSC_VIEWER_DRAW_CONTOUR);CHKERRQ(ierr);} 24120298fd71SBarry Smith } else vdraw = NULL; 2413693365a8SJed Brown ierr = PetscViewerASCIIPrintf(vstdout,"Explicit %s\n",flag_operator ? "Jacobian" : "preconditioning Jacobian");CHKERRQ(ierr); 2414693365a8SJed Brown if (flag) {ierr = MatView(Bexp,vstdout);CHKERRQ(ierr);} 2415693365a8SJed Brown if (vdraw) {ierr = MatView(Bexp,vdraw);CHKERRQ(ierr);} 2416693365a8SJed Brown ierr = PetscViewerASCIIPrintf(vstdout,"Finite difference Jacobian\n");CHKERRQ(ierr); 2417693365a8SJed Brown if (flag) {ierr = MatView(FDexp,vstdout);CHKERRQ(ierr);} 2418693365a8SJed Brown if (vdraw) {ierr = MatView(FDexp,vdraw);CHKERRQ(ierr);} 2419693365a8SJed Brown ierr = MatAYPX(FDexp,-1.0,Bexp,SAME_NONZERO_PATTERN);CHKERRQ(ierr); 2420693365a8SJed Brown ierr = PetscViewerASCIIPrintf(vstdout,"User-provided matrix minus finite difference Jacobian\n");CHKERRQ(ierr); 2421693365a8SJed Brown if (flag) {ierr = MatView(FDexp,vstdout);CHKERRQ(ierr);} 2422693365a8SJed Brown if (vdraw) { /* Always use contour for the difference */ 2423693365a8SJed Brown ierr = PetscViewerPushFormat(vdraw,PETSC_VIEWER_DRAW_CONTOUR);CHKERRQ(ierr); 2424693365a8SJed Brown ierr = MatView(FDexp,vdraw);CHKERRQ(ierr); 2425693365a8SJed Brown ierr = PetscViewerPopFormat(vdraw);CHKERRQ(ierr); 2426693365a8SJed Brown } 2427693365a8SJed Brown if (flag_contour) {ierr = PetscViewerPopFormat(vdraw);CHKERRQ(ierr);} 2428693365a8SJed Brown ierr = PetscViewerDestroy(&vdraw);CHKERRQ(ierr); 2429693365a8SJed Brown ierr = MatDestroy(&Bexp_mine);CHKERRQ(ierr); 2430693365a8SJed Brown ierr = MatDestroy(&FDexp);CHKERRQ(ierr); 2431693365a8SJed Brown } 2432693365a8SJed Brown } 24334c30e9fbSJed Brown { 24346719d8e4SJed Brown PetscBool flag = PETSC_FALSE,flag_display = PETSC_FALSE,flag_draw = PETSC_FALSE,flag_contour = PETSC_FALSE,flag_threshold = PETSC_FALSE; 24356719d8e4SJed Brown PetscReal threshold_atol = PETSC_SQRT_MACHINE_EPSILON,threshold_rtol = 10*PETSC_SQRT_MACHINE_EPSILON; 243627b0f280SBarry Smith ierr = PetscOptionsGetViewer(PetscObjectComm((PetscObject)snes),((PetscObject)snes)->prefix,"-snes_compare_coloring",NULL,NULL,&flag);CHKERRQ(ierr); 243727b0f280SBarry Smith ierr = PetscOptionsGetViewer(PetscObjectComm((PetscObject)snes),((PetscObject)snes)->prefix,"-snes_compare_coloring_display",NULL,NULL,&flag_display);CHKERRQ(ierr); 243827b0f280SBarry Smith ierr = PetscOptionsGetViewer(PetscObjectComm((PetscObject)snes),((PetscObject)snes)->prefix,"-snes_compare_coloring_draw",NULL,NULL,&flag_draw);CHKERRQ(ierr); 243927b0f280SBarry Smith ierr = PetscOptionsGetViewer(PetscObjectComm((PetscObject)snes),((PetscObject)snes)->prefix,"-snes_compare_coloring_draw_contour",NULL,NULL,&flag_contour);CHKERRQ(ierr); 244027b0f280SBarry Smith ierr = PetscOptionsGetViewer(PetscObjectComm((PetscObject)snes),((PetscObject)snes)->prefix,"-snes_compare_coloring_threshold",NULL,NULL,&flag_threshold);CHKERRQ(ierr); 244127b0f280SBarry Smith if (flag_threshold) { 2442c5929fdfSBarry Smith ierr = PetscOptionsGetReal(((PetscObject)snes)->options,((PetscObject)snes)->prefix,"-snes_compare_coloring_threshold_rtol",&threshold_rtol,NULL);CHKERRQ(ierr); 2443c5929fdfSBarry Smith ierr = PetscOptionsGetReal(((PetscObject)snes)->options,((PetscObject)snes)->prefix,"-snes_compare_coloring_threshold_atol",&threshold_atol,NULL);CHKERRQ(ierr); 244427b0f280SBarry Smith } 24456719d8e4SJed Brown if (flag || flag_display || flag_draw || flag_contour || flag_threshold) { 24464c30e9fbSJed Brown Mat Bfd; 24474c30e9fbSJed Brown PetscViewer vdraw,vstdout; 2448335efc43SPeter Brune MatColoring coloring; 24494c30e9fbSJed Brown ISColoring iscoloring; 24504c30e9fbSJed Brown MatFDColoring matfdcoloring; 24514c30e9fbSJed Brown PetscErrorCode (*func)(SNES,Vec,Vec,void*); 24524c30e9fbSJed Brown void *funcctx; 24536719d8e4SJed Brown PetscReal norm1,norm2,normmax; 24544c30e9fbSJed Brown 245594ab13aaSBarry Smith ierr = MatDuplicate(B,MAT_DO_NOT_COPY_VALUES,&Bfd);CHKERRQ(ierr); 2456335efc43SPeter Brune ierr = MatColoringCreate(Bfd,&coloring);CHKERRQ(ierr); 2457335efc43SPeter Brune ierr = MatColoringSetType(coloring,MATCOLORINGSL);CHKERRQ(ierr); 2458335efc43SPeter Brune ierr = MatColoringSetFromOptions(coloring);CHKERRQ(ierr); 2459335efc43SPeter Brune ierr = MatColoringApply(coloring,&iscoloring);CHKERRQ(ierr); 2460335efc43SPeter Brune ierr = MatColoringDestroy(&coloring);CHKERRQ(ierr); 24614c30e9fbSJed Brown ierr = MatFDColoringCreate(Bfd,iscoloring,&matfdcoloring);CHKERRQ(ierr); 2462f86b9fbaSHong Zhang ierr = MatFDColoringSetFromOptions(matfdcoloring);CHKERRQ(ierr); 2463f86b9fbaSHong Zhang ierr = MatFDColoringSetUp(Bfd,iscoloring,matfdcoloring);CHKERRQ(ierr); 24644c30e9fbSJed Brown ierr = ISColoringDestroy(&iscoloring);CHKERRQ(ierr); 24654c30e9fbSJed Brown 24664c30e9fbSJed Brown /* This method of getting the function is currently unreliable since it doesn't work for DM local functions. */ 24670298fd71SBarry Smith ierr = SNESGetFunction(snes,NULL,&func,&funcctx);CHKERRQ(ierr); 24684c30e9fbSJed Brown ierr = MatFDColoringSetFunction(matfdcoloring,(PetscErrorCode (*)(void))func,funcctx);CHKERRQ(ierr); 24694c30e9fbSJed Brown ierr = PetscObjectSetOptionsPrefix((PetscObject)matfdcoloring,((PetscObject)snes)->prefix);CHKERRQ(ierr); 24704c30e9fbSJed Brown ierr = PetscObjectAppendOptionsPrefix((PetscObject)matfdcoloring,"coloring_");CHKERRQ(ierr); 24714c30e9fbSJed Brown ierr = MatFDColoringSetFromOptions(matfdcoloring);CHKERRQ(ierr); 2472d1e9a80fSBarry Smith ierr = MatFDColoringApply(Bfd,matfdcoloring,X,snes);CHKERRQ(ierr); 24734c30e9fbSJed Brown ierr = MatFDColoringDestroy(&matfdcoloring);CHKERRQ(ierr); 24744c30e9fbSJed Brown 2475ce94432eSBarry Smith ierr = PetscViewerASCIIGetStdout(PetscObjectComm((PetscObject)snes),&vstdout);CHKERRQ(ierr); 24764c30e9fbSJed Brown if (flag_draw || flag_contour) { 2477ce94432eSBarry Smith ierr = PetscViewerDrawOpen(PetscObjectComm((PetscObject)snes),0,"Colored Jacobians",PETSC_DECIDE,PETSC_DECIDE,300,300,&vdraw);CHKERRQ(ierr); 24784c30e9fbSJed Brown if (flag_contour) {ierr = PetscViewerPushFormat(vdraw,PETSC_VIEWER_DRAW_CONTOUR);CHKERRQ(ierr);} 24790298fd71SBarry Smith } else vdraw = NULL; 24804c30e9fbSJed Brown ierr = PetscViewerASCIIPrintf(vstdout,"Explicit preconditioning Jacobian\n");CHKERRQ(ierr); 248194ab13aaSBarry Smith if (flag_display) {ierr = MatView(B,vstdout);CHKERRQ(ierr);} 248294ab13aaSBarry Smith if (vdraw) {ierr = MatView(B,vdraw);CHKERRQ(ierr);} 24834c30e9fbSJed Brown ierr = PetscViewerASCIIPrintf(vstdout,"Colored Finite difference Jacobian\n");CHKERRQ(ierr); 24846719d8e4SJed Brown if (flag_display) {ierr = MatView(Bfd,vstdout);CHKERRQ(ierr);} 24854c30e9fbSJed Brown if (vdraw) {ierr = MatView(Bfd,vdraw);CHKERRQ(ierr);} 248694ab13aaSBarry Smith ierr = MatAYPX(Bfd,-1.0,B,SAME_NONZERO_PATTERN);CHKERRQ(ierr); 24874c30e9fbSJed Brown ierr = MatNorm(Bfd,NORM_1,&norm1);CHKERRQ(ierr); 24886719d8e4SJed Brown ierr = MatNorm(Bfd,NORM_FROBENIUS,&norm2);CHKERRQ(ierr); 24894c30e9fbSJed Brown ierr = MatNorm(Bfd,NORM_MAX,&normmax);CHKERRQ(ierr); 249057622a8eSBarry 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); 24916719d8e4SJed Brown if (flag_display) {ierr = MatView(Bfd,vstdout);CHKERRQ(ierr);} 24924c30e9fbSJed Brown if (vdraw) { /* Always use contour for the difference */ 24934c30e9fbSJed Brown ierr = PetscViewerPushFormat(vdraw,PETSC_VIEWER_DRAW_CONTOUR);CHKERRQ(ierr); 24944c30e9fbSJed Brown ierr = MatView(Bfd,vdraw);CHKERRQ(ierr); 24954c30e9fbSJed Brown ierr = PetscViewerPopFormat(vdraw);CHKERRQ(ierr); 24964c30e9fbSJed Brown } 24974c30e9fbSJed Brown if (flag_contour) {ierr = PetscViewerPopFormat(vdraw);CHKERRQ(ierr);} 24986719d8e4SJed Brown 24996719d8e4SJed Brown if (flag_threshold) { 25006719d8e4SJed Brown PetscInt bs,rstart,rend,i; 250194ab13aaSBarry Smith ierr = MatGetBlockSize(B,&bs);CHKERRQ(ierr); 250294ab13aaSBarry Smith ierr = MatGetOwnershipRange(B,&rstart,&rend);CHKERRQ(ierr); 25036719d8e4SJed Brown for (i=rstart; i<rend; i++) { 25046719d8e4SJed Brown const PetscScalar *ba,*ca; 25056719d8e4SJed Brown const PetscInt *bj,*cj; 25066719d8e4SJed Brown PetscInt bn,cn,j,maxentrycol = -1,maxdiffcol = -1,maxrdiffcol = -1; 25076719d8e4SJed Brown PetscReal maxentry = 0,maxdiff = 0,maxrdiff = 0; 250894ab13aaSBarry Smith ierr = MatGetRow(B,i,&bn,&bj,&ba);CHKERRQ(ierr); 25096719d8e4SJed Brown ierr = MatGetRow(Bfd,i,&cn,&cj,&ca);CHKERRQ(ierr); 251094ab13aaSBarry Smith if (bn != cn) SETERRQ(((PetscObject)A)->comm,PETSC_ERR_PLIB,"Unexpected different nonzero pattern in -snes_compare_coloring_threshold"); 25116719d8e4SJed Brown for (j=0; j<bn; j++) { 25126719d8e4SJed Brown PetscReal rdiff = PetscAbsScalar(ca[j]) / (threshold_atol + threshold_rtol*PetscAbsScalar(ba[j])); 25136719d8e4SJed Brown if (PetscAbsScalar(ba[j]) > PetscAbs(maxentry)) { 25146719d8e4SJed Brown maxentrycol = bj[j]; 25156719d8e4SJed Brown maxentry = PetscRealPart(ba[j]); 25166719d8e4SJed Brown } 25176719d8e4SJed Brown if (PetscAbsScalar(ca[j]) > PetscAbs(maxdiff)) { 25186719d8e4SJed Brown maxdiffcol = bj[j]; 25196719d8e4SJed Brown maxdiff = PetscRealPart(ca[j]); 25206719d8e4SJed Brown } 25216719d8e4SJed Brown if (rdiff > maxrdiff) { 25226719d8e4SJed Brown maxrdiffcol = bj[j]; 25236719d8e4SJed Brown maxrdiff = rdiff; 25246719d8e4SJed Brown } 25256719d8e4SJed Brown } 25266719d8e4SJed Brown if (maxrdiff > 1) { 252757622a8eSBarry 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); 25286719d8e4SJed Brown for (j=0; j<bn; j++) { 25296719d8e4SJed Brown PetscReal rdiff; 25306719d8e4SJed Brown rdiff = PetscAbsScalar(ca[j]) / (threshold_atol + threshold_rtol*PetscAbsScalar(ba[j])); 25316719d8e4SJed Brown if (rdiff > 1) { 253257622a8eSBarry Smith ierr = PetscViewerASCIIPrintf(vstdout," (%D,%g:%g)",bj[j],(double)PetscRealPart(ba[j]),(double)PetscRealPart(ca[j]));CHKERRQ(ierr); 25336719d8e4SJed Brown } 25346719d8e4SJed Brown } 25356719d8e4SJed Brown ierr = PetscViewerASCIIPrintf(vstdout,"\n",i,maxentry,maxdiff,maxrdiff);CHKERRQ(ierr); 25366719d8e4SJed Brown } 253794ab13aaSBarry Smith ierr = MatRestoreRow(B,i,&bn,&bj,&ba);CHKERRQ(ierr); 25386719d8e4SJed Brown ierr = MatRestoreRow(Bfd,i,&cn,&cj,&ca);CHKERRQ(ierr); 25396719d8e4SJed Brown } 25406719d8e4SJed Brown } 25414c30e9fbSJed Brown ierr = PetscViewerDestroy(&vdraw);CHKERRQ(ierr); 25424c30e9fbSJed Brown ierr = MatDestroy(&Bfd);CHKERRQ(ierr); 25434c30e9fbSJed Brown } 25444c30e9fbSJed Brown } 25453a40ed3dSBarry Smith PetscFunctionReturn(0); 25469b94acceSBarry Smith } 25479b94acceSBarry Smith 2548bf388a1fSBarry Smith /*MC 2549411c0326SBarry Smith SNESJacobianFunction - Function used to convey the nonlinear Jacobian of the function to be solved by SNES 2550bf388a1fSBarry Smith 2551bf388a1fSBarry Smith Synopsis: 2552411c0326SBarry Smith #include "petscsnes.h" 2553411c0326SBarry Smith PetscErrorCode SNESJacobianFunction(SNES snes,Vec x,Mat Amat,Mat Pmat,void *ctx); 2554bf388a1fSBarry Smith 2555bf388a1fSBarry Smith + x - input vector 2556e5d3d808SBarry Smith . Amat - the matrix that defines the (approximate) Jacobian 2557e5d3d808SBarry Smith . Pmat - the matrix to be used in constructing the preconditioner, usually the same as Amat. 2558bf388a1fSBarry Smith - ctx - [optional] user-defined Jacobian context 2559bf388a1fSBarry Smith 2560878cb397SSatish Balay Level: intermediate 2561878cb397SSatish Balay 2562bf388a1fSBarry Smith .seealso: SNESSetFunction(), SNESGetFunction(), SNESSetJacobian(), SNESGetJacobian() 2563bf388a1fSBarry Smith M*/ 2564bf388a1fSBarry Smith 25659b94acceSBarry Smith /*@C 25669b94acceSBarry Smith SNESSetJacobian - Sets the function to compute Jacobian as well as the 2567044dda88SLois Curfman McInnes location to store the matrix. 25689b94acceSBarry Smith 25693f9fe445SBarry Smith Logically Collective on SNES and Mat 2570c7afd0dbSLois Curfman McInnes 25719b94acceSBarry Smith Input Parameters: 2572c7afd0dbSLois Curfman McInnes + snes - the SNES context 2573e5d3d808SBarry Smith . Amat - the matrix that defines the (approximate) Jacobian 2574e5d3d808SBarry Smith . Pmat - the matrix to be used in constructing the preconditioner, usually the same as Amat. 2575411c0326SBarry Smith . J - Jacobian evaluation routine (if NULL then SNES retains any previously set value), see SNESJacobianFunction for details 2576c7afd0dbSLois Curfman McInnes - ctx - [optional] user-defined context for private data for the 25770298fd71SBarry Smith Jacobian evaluation routine (may be NULL) (if NULL then SNES retains any previously set value) 25789b94acceSBarry Smith 25799b94acceSBarry Smith Notes: 2580e5d3d808SBarry Smith If the Amat matrix and Pmat matrix are different you must call MatAssemblyBegin/End() on 258116913363SBarry Smith each matrix. 258216913363SBarry Smith 2583895c21f2SBarry Smith If you know the operator Amat has a null space you can use MatSetNullSpace() and MatSetTransposeNullSpace() to supply the null 2584895c21f2SBarry Smith space to Amat and the KSP solvers will automatically use that null space as needed during the solution process. 2585895c21f2SBarry Smith 25868d359177SBarry Smith If using SNESComputeJacobianDefaultColor() to assemble a Jacobian, the ctx argument 2587a8a26c1eSJed Brown must be a MatFDColoring. 2588a8a26c1eSJed Brown 2589c3cc8fd1SJed Brown Other defect-correction schemes can be used by computing a different matrix in place of the Jacobian. One common 2590c3cc8fd1SJed Brown example is to use the "Picard linearization" which only differentiates through the highest order parts of each term. 2591c3cc8fd1SJed Brown 259236851e7fSLois Curfman McInnes Level: beginner 259336851e7fSLois Curfman McInnes 25949b94acceSBarry Smith .keywords: SNES, nonlinear, set, Jacobian, matrix 25959b94acceSBarry Smith 2596411c0326SBarry Smith .seealso: KSPSetOperators(), SNESSetFunction(), MatMFFDComputeJacobian(), SNESComputeJacobianDefaultColor(), MatStructure, J, 2597411c0326SBarry Smith SNESSetPicard(), SNESJacobianFunction 25989b94acceSBarry Smith @*/ 2599d1e9a80fSBarry Smith PetscErrorCode SNESSetJacobian(SNES snes,Mat Amat,Mat Pmat,PetscErrorCode (*J)(SNES,Vec,Mat,Mat,void*),void *ctx) 26009b94acceSBarry Smith { 2601dfbe8321SBarry Smith PetscErrorCode ierr; 26026cab3a1bSJed Brown DM dm; 26033a7fca6bSBarry Smith 26043a40ed3dSBarry Smith PetscFunctionBegin; 26050700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 2606e5d3d808SBarry Smith if (Amat) PetscValidHeaderSpecific(Amat,MAT_CLASSID,2); 2607e5d3d808SBarry Smith if (Pmat) PetscValidHeaderSpecific(Pmat,MAT_CLASSID,3); 2608e5d3d808SBarry Smith if (Amat) PetscCheckSameComm(snes,1,Amat,2); 2609e5d3d808SBarry Smith if (Pmat) PetscCheckSameComm(snes,1,Pmat,3); 26106cab3a1bSJed Brown ierr = SNESGetDM(snes,&dm);CHKERRQ(ierr); 2611f8b49ee9SBarry Smith ierr = DMSNESSetJacobian(dm,J,ctx);CHKERRQ(ierr); 2612e5d3d808SBarry Smith if (Amat) { 2613e5d3d808SBarry Smith ierr = PetscObjectReference((PetscObject)Amat);CHKERRQ(ierr); 26146bf464f9SBarry Smith ierr = MatDestroy(&snes->jacobian);CHKERRQ(ierr); 2615f5af7f23SKarl Rupp 2616e5d3d808SBarry Smith snes->jacobian = Amat; 26173a7fca6bSBarry Smith } 2618e5d3d808SBarry Smith if (Pmat) { 2619e5d3d808SBarry Smith ierr = PetscObjectReference((PetscObject)Pmat);CHKERRQ(ierr); 26206bf464f9SBarry Smith ierr = MatDestroy(&snes->jacobian_pre);CHKERRQ(ierr); 2621f5af7f23SKarl Rupp 2622e5d3d808SBarry Smith snes->jacobian_pre = Pmat; 26233a7fca6bSBarry Smith } 26243a40ed3dSBarry Smith PetscFunctionReturn(0); 26259b94acceSBarry Smith } 262662fef451SLois Curfman McInnes 2627c2aafc4cSSatish Balay /*@C 2628b4fd4287SBarry Smith SNESGetJacobian - Returns the Jacobian matrix and optionally the user 2629b4fd4287SBarry Smith provided context for evaluating the Jacobian. 2630b4fd4287SBarry Smith 2631c7afd0dbSLois Curfman McInnes Not Collective, but Mat object will be parallel if SNES object is 2632c7afd0dbSLois Curfman McInnes 2633b4fd4287SBarry Smith Input Parameter: 2634b4fd4287SBarry Smith . snes - the nonlinear solver context 2635b4fd4287SBarry Smith 2636b4fd4287SBarry Smith Output Parameters: 2637e5d3d808SBarry Smith + Amat - location to stash (approximate) Jacobian matrix (or NULL) 2638e5d3d808SBarry Smith . Pmat - location to stash matrix used to compute the preconditioner (or NULL) 2639411c0326SBarry Smith . J - location to put Jacobian function (or NULL), see SNESJacobianFunction for details on its calling sequence 26400298fd71SBarry Smith - ctx - location to stash Jacobian ctx (or NULL) 2641fee21e36SBarry Smith 264236851e7fSLois Curfman McInnes Level: advanced 264336851e7fSLois Curfman McInnes 2644411c0326SBarry Smith .seealso: SNESSetJacobian(), SNESComputeJacobian(), SNESJacobianFunction, SNESGetFunction() 2645b4fd4287SBarry Smith @*/ 2646d1e9a80fSBarry Smith PetscErrorCode SNESGetJacobian(SNES snes,Mat *Amat,Mat *Pmat,PetscErrorCode (**J)(SNES,Vec,Mat,Mat,void*),void **ctx) 2647b4fd4287SBarry Smith { 26486cab3a1bSJed Brown PetscErrorCode ierr; 26496cab3a1bSJed Brown DM dm; 2650942e3340SBarry Smith DMSNES sdm; 26516cab3a1bSJed Brown 26523a40ed3dSBarry Smith PetscFunctionBegin; 26530700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 2654e5d3d808SBarry Smith if (Amat) *Amat = snes->jacobian; 2655e5d3d808SBarry Smith if (Pmat) *Pmat = snes->jacobian_pre; 26566cab3a1bSJed Brown ierr = SNESGetDM(snes,&dm);CHKERRQ(ierr); 2657942e3340SBarry Smith ierr = DMGetDMSNES(dm,&sdm);CHKERRQ(ierr); 2658f8b49ee9SBarry Smith if (J) *J = sdm->ops->computejacobian; 26596cab3a1bSJed Brown if (ctx) *ctx = sdm->jacobianctx; 26603a40ed3dSBarry Smith PetscFunctionReturn(0); 2661b4fd4287SBarry Smith } 2662b4fd4287SBarry Smith 26639b94acceSBarry Smith /*@ 26649b94acceSBarry Smith SNESSetUp - Sets up the internal data structures for the later use 2665272ac6f2SLois Curfman McInnes of a nonlinear solver. 26669b94acceSBarry Smith 2667fee21e36SBarry Smith Collective on SNES 2668fee21e36SBarry Smith 2669c7afd0dbSLois Curfman McInnes Input Parameters: 267070e92668SMatthew Knepley . snes - the SNES context 2671c7afd0dbSLois Curfman McInnes 2672272ac6f2SLois Curfman McInnes Notes: 2673272ac6f2SLois Curfman McInnes For basic use of the SNES solvers the user need not explicitly call 2674272ac6f2SLois Curfman McInnes SNESSetUp(), since these actions will automatically occur during 2675272ac6f2SLois Curfman McInnes the call to SNESSolve(). However, if one wishes to control this 2676272ac6f2SLois Curfman McInnes phase separately, SNESSetUp() should be called after SNESCreate() 2677272ac6f2SLois Curfman McInnes and optional routines of the form SNESSetXXX(), but before SNESSolve(). 2678272ac6f2SLois Curfman McInnes 267936851e7fSLois Curfman McInnes Level: advanced 268036851e7fSLois Curfman McInnes 26819b94acceSBarry Smith .keywords: SNES, nonlinear, setup 26829b94acceSBarry Smith 26839b94acceSBarry Smith .seealso: SNESCreate(), SNESSolve(), SNESDestroy() 26849b94acceSBarry Smith @*/ 26857087cfbeSBarry Smith PetscErrorCode SNESSetUp(SNES snes) 26869b94acceSBarry Smith { 2687dfbe8321SBarry Smith PetscErrorCode ierr; 26886cab3a1bSJed Brown DM dm; 2689942e3340SBarry Smith DMSNES sdm; 2690c35f09e5SBarry Smith SNESLineSearch linesearch, pclinesearch; 26916e2a1849SPeter Brune void *lsprectx,*lspostctx; 26926b2b7091SBarry Smith PetscErrorCode (*precheck)(SNESLineSearch,Vec,Vec,PetscBool*,void*); 26936b2b7091SBarry Smith PetscErrorCode (*postcheck)(SNESLineSearch,Vec,Vec,Vec,PetscBool*,PetscBool*,void*); 26946e2a1849SPeter Brune PetscErrorCode (*func)(SNES,Vec,Vec,void*); 26956e2a1849SPeter Brune Vec f,fpc; 26966e2a1849SPeter Brune void *funcctx; 2697d1e9a80fSBarry Smith PetscErrorCode (*jac)(SNES,Vec,Mat,Mat,void*); 26981eb13d49SPeter Brune void *jacctx,*appctx; 269932b97717SPeter Brune Mat j,jpre; 27003a40ed3dSBarry Smith 27013a40ed3dSBarry Smith PetscFunctionBegin; 27020700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 27034dc4c822SBarry Smith if (snes->setupcalled) PetscFunctionReturn(0); 27049b94acceSBarry Smith 27057adad957SLisandro Dalcin if (!((PetscObject)snes)->type_name) { 270604d7464bSBarry Smith ierr = SNESSetType(snes,SNESNEWTONLS);CHKERRQ(ierr); 270785385478SLisandro Dalcin } 270885385478SLisandro Dalcin 27090298fd71SBarry Smith ierr = SNESGetFunction(snes,&snes->vec_func,NULL,NULL);CHKERRQ(ierr); 271058c9b817SLisandro Dalcin 27116cab3a1bSJed Brown ierr = SNESGetDM(snes,&dm);CHKERRQ(ierr); 2712942e3340SBarry Smith ierr = DMGetDMSNES(dm,&sdm);CHKERRQ(ierr); 2713ce94432eSBarry Smith if (!sdm->ops->computefunction) SETERRQ(PetscObjectComm((PetscObject)dm),PETSC_ERR_ARG_WRONGSTATE,"Function never provided to SNES object"); 271422c6f798SBarry Smith if (!sdm->ops->computejacobian) { 27158d359177SBarry Smith ierr = DMSNESSetJacobian(dm,SNESComputeJacobianDefaultColor,NULL);CHKERRQ(ierr); 271619f7a02aSBarry Smith } 27176cab3a1bSJed Brown if (!snes->vec_func) { 27186cab3a1bSJed Brown ierr = DMCreateGlobalVector(dm,&snes->vec_func);CHKERRQ(ierr); 2719214df951SJed Brown } 2720efd51863SBarry Smith 272122d28d08SBarry Smith if (!snes->ksp) { 272222d28d08SBarry Smith ierr = SNESGetKSP(snes, &snes->ksp);CHKERRQ(ierr); 272322d28d08SBarry Smith } 2724b710008aSBarry Smith 272522d28d08SBarry Smith if (!snes->linesearch) { 27267601faf0SJed Brown ierr = SNESGetLineSearch(snes, &snes->linesearch);CHKERRQ(ierr); 272722d28d08SBarry Smith } 2728ed07d7d7SPeter Brune ierr = SNESLineSearchSetFunction(snes->linesearch,SNESComputeFunction);CHKERRQ(ierr); 27299e764e56SPeter Brune 2730efd4aadfSBarry Smith if (snes->npc && (snes->npcside== PC_LEFT)) { 2731172a4300SPeter Brune snes->mf = PETSC_TRUE; 2732172a4300SPeter Brune snes->mf_operator = PETSC_FALSE; 2733172a4300SPeter Brune } 2734d8f46077SPeter Brune 2735efd4aadfSBarry Smith if (snes->npc) { 27366e2a1849SPeter Brune /* copy the DM over */ 27376e2a1849SPeter Brune ierr = SNESGetDM(snes,&dm);CHKERRQ(ierr); 2738efd4aadfSBarry Smith ierr = SNESSetDM(snes->npc,dm);CHKERRQ(ierr); 27396e2a1849SPeter Brune 27406e2a1849SPeter Brune ierr = SNESGetFunction(snes,&f,&func,&funcctx);CHKERRQ(ierr); 27416e2a1849SPeter Brune ierr = VecDuplicate(f,&fpc);CHKERRQ(ierr); 2742efd4aadfSBarry Smith ierr = SNESSetFunction(snes->npc,fpc,func,funcctx);CHKERRQ(ierr); 274332b97717SPeter Brune ierr = SNESGetJacobian(snes,&j,&jpre,&jac,&jacctx);CHKERRQ(ierr); 2744efd4aadfSBarry Smith ierr = SNESSetJacobian(snes->npc,j,jpre,jac,jacctx);CHKERRQ(ierr); 27451eb13d49SPeter Brune ierr = SNESGetApplicationContext(snes,&appctx);CHKERRQ(ierr); 2746efd4aadfSBarry Smith ierr = SNESSetApplicationContext(snes->npc,appctx);CHKERRQ(ierr); 27476e2a1849SPeter Brune ierr = VecDestroy(&fpc);CHKERRQ(ierr); 27486e2a1849SPeter Brune 27496e2a1849SPeter Brune /* copy the function pointers over */ 2750efd4aadfSBarry Smith ierr = PetscObjectCopyFortranFunctionPointers((PetscObject)snes,(PetscObject)snes->npc);CHKERRQ(ierr); 27516e2a1849SPeter Brune 27526e2a1849SPeter Brune /* default to 1 iteration */ 2753efd4aadfSBarry Smith ierr = SNESSetTolerances(snes->npc,0.0,0.0,0.0,1,snes->npc->max_funcs);CHKERRQ(ierr); 2754efd4aadfSBarry Smith if (snes->npcside==PC_RIGHT) { 2755efd4aadfSBarry Smith ierr = SNESSetNormSchedule(snes->npc,SNES_NORM_FINAL_ONLY);CHKERRQ(ierr); 2756a9936a0cSPeter Brune } else { 2757efd4aadfSBarry Smith ierr = SNESSetNormSchedule(snes->npc,SNES_NORM_NONE);CHKERRQ(ierr); 2758a9936a0cSPeter Brune } 2759efd4aadfSBarry Smith ierr = SNESSetFromOptions(snes->npc);CHKERRQ(ierr); 27606e2a1849SPeter Brune 27616e2a1849SPeter Brune /* copy the line search context over */ 27627601faf0SJed Brown ierr = SNESGetLineSearch(snes,&linesearch);CHKERRQ(ierr); 2763efd4aadfSBarry Smith ierr = SNESGetLineSearch(snes->npc,&pclinesearch);CHKERRQ(ierr); 27646b2b7091SBarry Smith ierr = SNESLineSearchGetPreCheck(linesearch,&precheck,&lsprectx);CHKERRQ(ierr); 27656b2b7091SBarry Smith ierr = SNESLineSearchGetPostCheck(linesearch,&postcheck,&lspostctx);CHKERRQ(ierr); 27666b2b7091SBarry Smith ierr = SNESLineSearchSetPreCheck(pclinesearch,precheck,lsprectx);CHKERRQ(ierr); 27676b2b7091SBarry Smith ierr = SNESLineSearchSetPostCheck(pclinesearch,postcheck,lspostctx);CHKERRQ(ierr); 27686e2a1849SPeter Brune ierr = PetscObjectCopyFortranFunctionPointers((PetscObject)linesearch, (PetscObject)pclinesearch);CHKERRQ(ierr); 27696e2a1849SPeter Brune } 277032b97717SPeter Brune if (snes->mf) { 277132b97717SPeter Brune ierr = SNESSetUpMatrixFree_Private(snes, snes->mf_operator, snes->mf_version);CHKERRQ(ierr); 277232b97717SPeter Brune } 277332b97717SPeter Brune if (snes->ops->usercompute && !snes->user) { 277432b97717SPeter Brune ierr = (*snes->ops->usercompute)(snes,(void**)&snes->user);CHKERRQ(ierr); 277532b97717SPeter Brune } 27766e2a1849SPeter Brune 277737ec4e1aSPeter Brune snes->jac_iter = 0; 277837ec4e1aSPeter Brune snes->pre_iter = 0; 277937ec4e1aSPeter Brune 2780410397dcSLisandro Dalcin if (snes->ops->setup) { 2781410397dcSLisandro Dalcin ierr = (*snes->ops->setup)(snes);CHKERRQ(ierr); 2782410397dcSLisandro Dalcin } 278358c9b817SLisandro Dalcin 2784efd4aadfSBarry Smith if (snes->npc && (snes->npcside== PC_LEFT)) { 27856c67d002SPeter Brune if (snes->functype == SNES_FUNCTION_PRECONDITIONED) { 278655d4788fSPeter Brune ierr = SNESGetLineSearch(snes,&linesearch);CHKERRQ(ierr); 2787be95d8f1SBarry Smith ierr = SNESLineSearchSetFunction(linesearch,SNESComputeFunctionDefaultNPC);CHKERRQ(ierr); 27886c67d002SPeter Brune } 27896c67d002SPeter Brune } 27906c67d002SPeter Brune 27917aaed0d8SBarry Smith snes->setupcalled = PETSC_TRUE; 27923a40ed3dSBarry Smith PetscFunctionReturn(0); 27939b94acceSBarry Smith } 27949b94acceSBarry Smith 279537596af1SLisandro Dalcin /*@ 279637596af1SLisandro Dalcin SNESReset - Resets a SNES context to the snessetupcalled = 0 state and removes any allocated Vecs and Mats 279737596af1SLisandro Dalcin 279837596af1SLisandro Dalcin Collective on SNES 279937596af1SLisandro Dalcin 280037596af1SLisandro Dalcin Input Parameter: 280137596af1SLisandro Dalcin . snes - iterative context obtained from SNESCreate() 280237596af1SLisandro Dalcin 2803d25893d9SBarry Smith Level: intermediate 2804d25893d9SBarry Smith 2805d25893d9SBarry Smith Notes: Also calls the application context destroy routine set with SNESSetComputeApplicationContext() 280637596af1SLisandro Dalcin 280737596af1SLisandro Dalcin .keywords: SNES, destroy 280837596af1SLisandro Dalcin 280937596af1SLisandro Dalcin .seealso: SNESCreate(), SNESSetUp(), SNESSolve() 281037596af1SLisandro Dalcin @*/ 281137596af1SLisandro Dalcin PetscErrorCode SNESReset(SNES snes) 281237596af1SLisandro Dalcin { 281337596af1SLisandro Dalcin PetscErrorCode ierr; 281437596af1SLisandro Dalcin 281537596af1SLisandro Dalcin PetscFunctionBegin; 281637596af1SLisandro Dalcin PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 2817d25893d9SBarry Smith if (snes->ops->userdestroy && snes->user) { 2818d25893d9SBarry Smith ierr = (*snes->ops->userdestroy)((void**)&snes->user);CHKERRQ(ierr); 28190298fd71SBarry Smith snes->user = NULL; 2820d25893d9SBarry Smith } 2821efd4aadfSBarry Smith if (snes->npc) { 2822efd4aadfSBarry Smith ierr = SNESReset(snes->npc);CHKERRQ(ierr); 28238a23116dSBarry Smith } 28248a23116dSBarry Smith 282537596af1SLisandro Dalcin if (snes->ops->reset) { 282637596af1SLisandro Dalcin ierr = (*snes->ops->reset)(snes);CHKERRQ(ierr); 282737596af1SLisandro Dalcin } 28289e764e56SPeter Brune if (snes->ksp) { 28299e764e56SPeter Brune ierr = KSPReset(snes->ksp);CHKERRQ(ierr); 28309e764e56SPeter Brune } 28319e764e56SPeter Brune 28329e764e56SPeter Brune if (snes->linesearch) { 2833f1c6b773SPeter Brune ierr = SNESLineSearchReset(snes->linesearch);CHKERRQ(ierr); 28349e764e56SPeter Brune } 28359e764e56SPeter Brune 28366bf464f9SBarry Smith ierr = VecDestroy(&snes->vec_rhs);CHKERRQ(ierr); 28376bf464f9SBarry Smith ierr = VecDestroy(&snes->vec_sol);CHKERRQ(ierr); 28386bf464f9SBarry Smith ierr = VecDestroy(&snes->vec_sol_update);CHKERRQ(ierr); 28396bf464f9SBarry Smith ierr = VecDestroy(&snes->vec_func);CHKERRQ(ierr); 28406bf464f9SBarry Smith ierr = MatDestroy(&snes->jacobian);CHKERRQ(ierr); 28416bf464f9SBarry Smith ierr = MatDestroy(&snes->jacobian_pre);CHKERRQ(ierr); 2842c41d97abSJed Brown ierr = VecDestroyVecs(snes->nwork,&snes->work);CHKERRQ(ierr); 2843c41d97abSJed Brown ierr = VecDestroyVecs(snes->nvwork,&snes->vwork);CHKERRQ(ierr); 2844f5af7f23SKarl Rupp 284540fdac6aSLawrence Mitchell snes->alwayscomputesfinalresidual = PETSC_FALSE; 284640fdac6aSLawrence Mitchell 284737596af1SLisandro Dalcin snes->nwork = snes->nvwork = 0; 284837596af1SLisandro Dalcin snes->setupcalled = PETSC_FALSE; 284937596af1SLisandro Dalcin PetscFunctionReturn(0); 285037596af1SLisandro Dalcin } 285137596af1SLisandro Dalcin 285252baeb72SSatish Balay /*@ 28539b94acceSBarry Smith SNESDestroy - Destroys the nonlinear solver context that was created 28549b94acceSBarry Smith with SNESCreate(). 28559b94acceSBarry Smith 2856c7afd0dbSLois Curfman McInnes Collective on SNES 2857c7afd0dbSLois Curfman McInnes 28589b94acceSBarry Smith Input Parameter: 28599b94acceSBarry Smith . snes - the SNES context 28609b94acceSBarry Smith 286136851e7fSLois Curfman McInnes Level: beginner 286236851e7fSLois Curfman McInnes 28639b94acceSBarry Smith .keywords: SNES, nonlinear, destroy 28649b94acceSBarry Smith 286563a78c88SLois Curfman McInnes .seealso: SNESCreate(), SNESSolve() 28669b94acceSBarry Smith @*/ 28676bf464f9SBarry Smith PetscErrorCode SNESDestroy(SNES *snes) 28689b94acceSBarry Smith { 28696849ba73SBarry Smith PetscErrorCode ierr; 28703a40ed3dSBarry Smith 28713a40ed3dSBarry Smith PetscFunctionBegin; 28726bf464f9SBarry Smith if (!*snes) PetscFunctionReturn(0); 28736bf464f9SBarry Smith PetscValidHeaderSpecific((*snes),SNES_CLASSID,1); 28746bf464f9SBarry Smith if (--((PetscObject)(*snes))->refct > 0) {*snes = 0; PetscFunctionReturn(0);} 2875d4bb536fSBarry Smith 28766bf464f9SBarry Smith ierr = SNESReset((*snes));CHKERRQ(ierr); 2877efd4aadfSBarry Smith ierr = SNESDestroy(&(*snes)->npc);CHKERRQ(ierr); 28786b8b9a38SLisandro Dalcin 2879e04113cfSBarry Smith /* if memory was published with SAWs then destroy it */ 2880e04113cfSBarry Smith ierr = PetscObjectSAWsViewOff((PetscObject)*snes);CHKERRQ(ierr); 28816bf464f9SBarry Smith if ((*snes)->ops->destroy) {ierr = (*((*snes))->ops->destroy)((*snes));CHKERRQ(ierr);} 28826d4c513bSLisandro Dalcin 2883*dc822a44SJed Brown ierr = DMCoarsenHookRemove((*snes)->dm,DMCoarsenHook_SNESVecSol,DMRestrictHook_SNESVecSol,*snes);CHKERRQ(ierr); 28846bf464f9SBarry Smith ierr = DMDestroy(&(*snes)->dm);CHKERRQ(ierr); 28856bf464f9SBarry Smith ierr = KSPDestroy(&(*snes)->ksp);CHKERRQ(ierr); 2886f1c6b773SPeter Brune ierr = SNESLineSearchDestroy(&(*snes)->linesearch);CHKERRQ(ierr); 28876b8b9a38SLisandro Dalcin 28886bf464f9SBarry Smith ierr = PetscFree((*snes)->kspconvctx);CHKERRQ(ierr); 28896bf464f9SBarry Smith if ((*snes)->ops->convergeddestroy) { 28906bf464f9SBarry Smith ierr = (*(*snes)->ops->convergeddestroy)((*snes)->cnvP);CHKERRQ(ierr); 28916b8b9a38SLisandro Dalcin } 28926bf464f9SBarry Smith if ((*snes)->conv_malloc) { 28936bf464f9SBarry Smith ierr = PetscFree((*snes)->conv_hist);CHKERRQ(ierr); 28946bf464f9SBarry Smith ierr = PetscFree((*snes)->conv_hist_its);CHKERRQ(ierr); 289558c9b817SLisandro Dalcin } 28966bf464f9SBarry Smith ierr = SNESMonitorCancel((*snes));CHKERRQ(ierr); 2897a79aaaedSSatish Balay ierr = PetscHeaderDestroy(snes);CHKERRQ(ierr); 28983a40ed3dSBarry Smith PetscFunctionReturn(0); 28999b94acceSBarry Smith } 29009b94acceSBarry Smith 29019b94acceSBarry Smith /* ----------- Routines to set solver parameters ---------- */ 29029b94acceSBarry Smith 2903a8054027SBarry Smith /*@ 2904a8054027SBarry Smith SNESSetLagPreconditioner - Determines when the preconditioner is rebuilt in the nonlinear solve. 2905a8054027SBarry Smith 29063f9fe445SBarry Smith Logically Collective on SNES 2907a8054027SBarry Smith 2908a8054027SBarry Smith Input Parameters: 2909a8054027SBarry Smith + snes - the SNES context 2910a8054027SBarry 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 29113b4f5425SBarry Smith the Jacobian is built etc. -2 indicates rebuild preconditioner at next chance but then never rebuild after that 2912a8054027SBarry Smith 2913a8054027SBarry Smith Options Database Keys: 2914a8054027SBarry Smith . -snes_lag_preconditioner <lag> 2915a8054027SBarry Smith 2916a8054027SBarry Smith Notes: 2917a8054027SBarry Smith The default is 1 2918a8054027SBarry Smith The preconditioner is ALWAYS built in the first iteration of a nonlinear solve unless lag is -1 2919a8054027SBarry Smith If -1 is used before the very first nonlinear solve the preconditioner is still built because there is no previous preconditioner to use 2920a8054027SBarry Smith 2921a8054027SBarry Smith Level: intermediate 2922a8054027SBarry Smith 2923a8054027SBarry Smith .keywords: SNES, nonlinear, set, convergence, tolerances 2924a8054027SBarry Smith 2925e35cf81dSBarry Smith .seealso: SNESSetTrustRegionTolerance(), SNESGetLagPreconditioner(), SNESSetLagJacobian(), SNESGetLagJacobian() 2926a8054027SBarry Smith 2927a8054027SBarry Smith @*/ 29287087cfbeSBarry Smith PetscErrorCode SNESSetLagPreconditioner(SNES snes,PetscInt lag) 2929a8054027SBarry Smith { 2930a8054027SBarry Smith PetscFunctionBegin; 29310700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 2932e32f2f54SBarry Smith if (lag < -2) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Lag must be -2, -1, 1 or greater"); 2933e32f2f54SBarry Smith if (!lag) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Lag cannot be 0"); 2934c5eb9154SBarry Smith PetscValidLogicalCollectiveInt(snes,lag,2); 2935a8054027SBarry Smith snes->lagpreconditioner = lag; 2936a8054027SBarry Smith PetscFunctionReturn(0); 2937a8054027SBarry Smith } 2938a8054027SBarry Smith 2939efd51863SBarry Smith /*@ 2940efd51863SBarry Smith SNESSetGridSequence - sets the number of steps of grid sequencing that SNES does 2941efd51863SBarry Smith 2942efd51863SBarry Smith Logically Collective on SNES 2943efd51863SBarry Smith 2944efd51863SBarry Smith Input Parameters: 2945efd51863SBarry Smith + snes - the SNES context 2946efd51863SBarry Smith - steps - the number of refinements to do, defaults to 0 2947efd51863SBarry Smith 2948efd51863SBarry Smith Options Database Keys: 2949efd51863SBarry Smith . -snes_grid_sequence <steps> 2950efd51863SBarry Smith 2951efd51863SBarry Smith Level: intermediate 2952efd51863SBarry Smith 2953c0df2a02SJed Brown Notes: 2954c0df2a02SJed Brown Use SNESGetSolution() to extract the fine grid solution after grid sequencing. 2955c0df2a02SJed Brown 2956efd51863SBarry Smith .keywords: SNES, nonlinear, set, convergence, tolerances 2957efd51863SBarry Smith 2958fa19ca70SBarry Smith .seealso: SNESSetTrustRegionTolerance(), SNESGetLagPreconditioner(), SNESSetLagJacobian(), SNESGetLagJacobian(), SNESGetGridSequence() 2959efd51863SBarry Smith 2960efd51863SBarry Smith @*/ 2961efd51863SBarry Smith PetscErrorCode SNESSetGridSequence(SNES snes,PetscInt steps) 2962efd51863SBarry Smith { 2963efd51863SBarry Smith PetscFunctionBegin; 2964efd51863SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 2965efd51863SBarry Smith PetscValidLogicalCollectiveInt(snes,steps,2); 2966efd51863SBarry Smith snes->gridsequence = steps; 2967efd51863SBarry Smith PetscFunctionReturn(0); 2968efd51863SBarry Smith } 2969efd51863SBarry Smith 2970fa19ca70SBarry Smith /*@ 2971fa19ca70SBarry Smith SNESGetGridSequence - gets the number of steps of grid sequencing that SNES does 2972fa19ca70SBarry Smith 2973fa19ca70SBarry Smith Logically Collective on SNES 2974fa19ca70SBarry Smith 2975fa19ca70SBarry Smith Input Parameter: 2976fa19ca70SBarry Smith . snes - the SNES context 2977fa19ca70SBarry Smith 2978fa19ca70SBarry Smith Output Parameter: 2979fa19ca70SBarry Smith . steps - the number of refinements to do, defaults to 0 2980fa19ca70SBarry Smith 2981fa19ca70SBarry Smith Options Database Keys: 2982fa19ca70SBarry Smith . -snes_grid_sequence <steps> 2983fa19ca70SBarry Smith 2984fa19ca70SBarry Smith Level: intermediate 2985fa19ca70SBarry Smith 2986fa19ca70SBarry Smith Notes: 2987fa19ca70SBarry Smith Use SNESGetSolution() to extract the fine grid solution after grid sequencing. 2988fa19ca70SBarry Smith 2989fa19ca70SBarry Smith .keywords: SNES, nonlinear, set, convergence, tolerances 2990fa19ca70SBarry Smith 2991fa19ca70SBarry Smith .seealso: SNESSetTrustRegionTolerance(), SNESGetLagPreconditioner(), SNESSetLagJacobian(), SNESGetLagJacobian(), SNESSetGridSequence() 2992fa19ca70SBarry Smith 2993fa19ca70SBarry Smith @*/ 2994fa19ca70SBarry Smith PetscErrorCode SNESGetGridSequence(SNES snes,PetscInt *steps) 2995fa19ca70SBarry Smith { 2996fa19ca70SBarry Smith PetscFunctionBegin; 2997fa19ca70SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 2998fa19ca70SBarry Smith *steps = snes->gridsequence; 2999fa19ca70SBarry Smith PetscFunctionReturn(0); 3000fa19ca70SBarry Smith } 3001fa19ca70SBarry Smith 3002a8054027SBarry Smith /*@ 3003a8054027SBarry Smith SNESGetLagPreconditioner - Indicates how often the preconditioner is rebuilt 3004a8054027SBarry Smith 30053f9fe445SBarry Smith Not Collective 3006a8054027SBarry Smith 3007a8054027SBarry Smith Input Parameter: 3008a8054027SBarry Smith . snes - the SNES context 3009a8054027SBarry Smith 3010a8054027SBarry Smith Output Parameter: 3011a8054027SBarry 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 30123b4f5425SBarry Smith the Jacobian is built etc. -2 indicates rebuild preconditioner at next chance but then never rebuild after that 3013a8054027SBarry Smith 3014a8054027SBarry Smith Options Database Keys: 3015a8054027SBarry Smith . -snes_lag_preconditioner <lag> 3016a8054027SBarry Smith 3017a8054027SBarry Smith Notes: 3018a8054027SBarry Smith The default is 1 3019a8054027SBarry Smith The preconditioner is ALWAYS built in the first iteration of a nonlinear solve unless lag is -1 3020a8054027SBarry Smith 3021a8054027SBarry Smith Level: intermediate 3022a8054027SBarry Smith 3023a8054027SBarry Smith .keywords: SNES, nonlinear, set, convergence, tolerances 3024a8054027SBarry Smith 3025a8054027SBarry Smith .seealso: SNESSetTrustRegionTolerance(), SNESSetLagPreconditioner() 3026a8054027SBarry Smith 3027a8054027SBarry Smith @*/ 30287087cfbeSBarry Smith PetscErrorCode SNESGetLagPreconditioner(SNES snes,PetscInt *lag) 3029a8054027SBarry Smith { 3030a8054027SBarry Smith PetscFunctionBegin; 30310700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 3032a8054027SBarry Smith *lag = snes->lagpreconditioner; 3033a8054027SBarry Smith PetscFunctionReturn(0); 3034a8054027SBarry Smith } 3035a8054027SBarry Smith 3036e35cf81dSBarry Smith /*@ 3037e35cf81dSBarry Smith SNESSetLagJacobian - Determines when the Jacobian is rebuilt in the nonlinear solve. See SNESSetLagPreconditioner() for determining how 3038e35cf81dSBarry Smith often the preconditioner is rebuilt. 3039e35cf81dSBarry Smith 30403f9fe445SBarry Smith Logically Collective on SNES 3041e35cf81dSBarry Smith 3042e35cf81dSBarry Smith Input Parameters: 3043e35cf81dSBarry Smith + snes - the SNES context 3044e35cf81dSBarry 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 3045fe3ffe1eSBarry Smith the Jacobian is built etc. -2 means rebuild at next chance but then never again 3046e35cf81dSBarry Smith 3047e35cf81dSBarry Smith Options Database Keys: 3048e35cf81dSBarry Smith . -snes_lag_jacobian <lag> 3049e35cf81dSBarry Smith 3050e35cf81dSBarry Smith Notes: 3051e35cf81dSBarry Smith The default is 1 3052e35cf81dSBarry Smith The Jacobian is ALWAYS built in the first iteration of a nonlinear solve unless lag is -1 3053fe3ffe1eSBarry 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 3054fe3ffe1eSBarry Smith at the next Newton step but never again (unless it is reset to another value) 3055e35cf81dSBarry Smith 3056e35cf81dSBarry Smith Level: intermediate 3057e35cf81dSBarry Smith 3058e35cf81dSBarry Smith .keywords: SNES, nonlinear, set, convergence, tolerances 3059e35cf81dSBarry Smith 3060e35cf81dSBarry Smith .seealso: SNESSetTrustRegionTolerance(), SNESGetLagPreconditioner(), SNESSetLagPreconditioner(), SNESGetLagJacobian() 3061e35cf81dSBarry Smith 3062e35cf81dSBarry Smith @*/ 30637087cfbeSBarry Smith PetscErrorCode SNESSetLagJacobian(SNES snes,PetscInt lag) 3064e35cf81dSBarry Smith { 3065e35cf81dSBarry Smith PetscFunctionBegin; 30660700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 3067e32f2f54SBarry Smith if (lag < -2) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Lag must be -2, -1, 1 or greater"); 3068e32f2f54SBarry Smith if (!lag) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Lag cannot be 0"); 3069c5eb9154SBarry Smith PetscValidLogicalCollectiveInt(snes,lag,2); 3070e35cf81dSBarry Smith snes->lagjacobian = lag; 3071e35cf81dSBarry Smith PetscFunctionReturn(0); 3072e35cf81dSBarry Smith } 3073e35cf81dSBarry Smith 3074e35cf81dSBarry Smith /*@ 3075e35cf81dSBarry Smith SNESGetLagJacobian - Indicates how often the Jacobian is rebuilt. See SNESGetLagPreconditioner() to determine when the preconditioner is rebuilt 3076e35cf81dSBarry Smith 30773f9fe445SBarry Smith Not Collective 3078e35cf81dSBarry Smith 3079e35cf81dSBarry Smith Input Parameter: 3080e35cf81dSBarry Smith . snes - the SNES context 3081e35cf81dSBarry Smith 3082e35cf81dSBarry Smith Output Parameter: 3083e35cf81dSBarry 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 3084e35cf81dSBarry Smith the Jacobian is built etc. 3085e35cf81dSBarry Smith 3086e35cf81dSBarry Smith Options Database Keys: 3087e35cf81dSBarry Smith . -snes_lag_jacobian <lag> 3088e35cf81dSBarry Smith 3089e35cf81dSBarry Smith Notes: 3090e35cf81dSBarry Smith The default is 1 3091e35cf81dSBarry Smith The jacobian is ALWAYS built in the first iteration of a nonlinear solve unless lag is -1 3092e35cf81dSBarry Smith 3093e35cf81dSBarry Smith Level: intermediate 3094e35cf81dSBarry Smith 3095e35cf81dSBarry Smith .keywords: SNES, nonlinear, set, convergence, tolerances 3096e35cf81dSBarry Smith 3097e35cf81dSBarry Smith .seealso: SNESSetTrustRegionTolerance(), SNESSetLagJacobian(), SNESSetLagPreconditioner(), SNESGetLagPreconditioner() 3098e35cf81dSBarry Smith 3099e35cf81dSBarry Smith @*/ 31007087cfbeSBarry Smith PetscErrorCode SNESGetLagJacobian(SNES snes,PetscInt *lag) 3101e35cf81dSBarry Smith { 3102e35cf81dSBarry Smith PetscFunctionBegin; 31030700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 3104e35cf81dSBarry Smith *lag = snes->lagjacobian; 3105e35cf81dSBarry Smith PetscFunctionReturn(0); 3106e35cf81dSBarry Smith } 3107e35cf81dSBarry Smith 310837ec4e1aSPeter Brune /*@ 310937ec4e1aSPeter Brune SNESSetLagJacobianPersists - Set whether or not the Jacobian lagging persists through multiple solves 311037ec4e1aSPeter Brune 311137ec4e1aSPeter Brune Logically collective on SNES 311237ec4e1aSPeter Brune 311337ec4e1aSPeter Brune Input Parameter: 311437ec4e1aSPeter Brune + snes - the SNES context 31159d7e2deaSPeter Brune - flg - jacobian lagging persists if true 311637ec4e1aSPeter Brune 311737ec4e1aSPeter Brune Options Database Keys: 311837ec4e1aSPeter Brune . -snes_lag_jacobian_persists <flg> 311937ec4e1aSPeter Brune 312037ec4e1aSPeter Brune Notes: This is useful both for nonlinear preconditioning, where it's appropriate to have the Jacobian be stale by 312137ec4e1aSPeter Brune several solves, and for implicit time-stepping, where Jacobian lagging in the inner nonlinear solve over several 312237ec4e1aSPeter Brune timesteps may present huge efficiency gains. 312337ec4e1aSPeter Brune 312437ec4e1aSPeter Brune Level: developer 312537ec4e1aSPeter Brune 3126be95d8f1SBarry Smith .keywords: SNES, nonlinear, lag 312737ec4e1aSPeter Brune 3128be95d8f1SBarry Smith .seealso: SNESSetLagPreconditionerPersists(), SNESSetLagJacobian(), SNESGetLagJacobian(), SNESGetNPC() 312937ec4e1aSPeter Brune 313037ec4e1aSPeter Brune @*/ 313137ec4e1aSPeter Brune PetscErrorCode SNESSetLagJacobianPersists(SNES snes,PetscBool flg) 313237ec4e1aSPeter Brune { 313337ec4e1aSPeter Brune PetscFunctionBegin; 313437ec4e1aSPeter Brune PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 313537ec4e1aSPeter Brune PetscValidLogicalCollectiveBool(snes,flg,2); 313637ec4e1aSPeter Brune snes->lagjac_persist = flg; 313737ec4e1aSPeter Brune PetscFunctionReturn(0); 313837ec4e1aSPeter Brune } 313937ec4e1aSPeter Brune 314037ec4e1aSPeter Brune /*@ 314137ec4e1aSPeter Brune SNESSetLagPreconditionerPersists - Set whether or not the preconditioner lagging persists through multiple solves 314237ec4e1aSPeter Brune 314337ec4e1aSPeter Brune Logically Collective on SNES 314437ec4e1aSPeter Brune 314537ec4e1aSPeter Brune Input Parameter: 314637ec4e1aSPeter Brune + snes - the SNES context 31479d7e2deaSPeter Brune - flg - preconditioner lagging persists if true 314837ec4e1aSPeter Brune 314937ec4e1aSPeter Brune Options Database Keys: 315037ec4e1aSPeter Brune . -snes_lag_jacobian_persists <flg> 315137ec4e1aSPeter Brune 315237ec4e1aSPeter Brune Notes: This is useful both for nonlinear preconditioning, where it's appropriate to have the preconditioner be stale 315337ec4e1aSPeter Brune by several solves, and for implicit time-stepping, where preconditioner lagging in the inner nonlinear solve over 315437ec4e1aSPeter Brune several timesteps may present huge efficiency gains. 315537ec4e1aSPeter Brune 315637ec4e1aSPeter Brune Level: developer 315737ec4e1aSPeter Brune 3158be95d8f1SBarry Smith .keywords: SNES, nonlinear, lag 315937ec4e1aSPeter Brune 3160be95d8f1SBarry Smith .seealso: SNESSetLagJacobianPersists(), SNESSetLagJacobian(), SNESGetLagJacobian(), SNESGetNPC() 316137ec4e1aSPeter Brune 316237ec4e1aSPeter Brune @*/ 316337ec4e1aSPeter Brune PetscErrorCode SNESSetLagPreconditionerPersists(SNES snes,PetscBool flg) 316437ec4e1aSPeter Brune { 316537ec4e1aSPeter Brune PetscFunctionBegin; 316637ec4e1aSPeter Brune PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 316737ec4e1aSPeter Brune PetscValidLogicalCollectiveBool(snes,flg,2); 316837ec4e1aSPeter Brune snes->lagpre_persist = flg; 316937ec4e1aSPeter Brune PetscFunctionReturn(0); 317037ec4e1aSPeter Brune } 317137ec4e1aSPeter Brune 31729b94acceSBarry Smith /*@ 3173be5caee7SBarry Smith SNESSetForceIteration - force SNESSolve() to take at least one iteration regardless of the initial residual norm 3174be5caee7SBarry Smith 3175be5caee7SBarry Smith Logically Collective on SNES 3176be5caee7SBarry Smith 3177be5caee7SBarry Smith Input Parameters: 3178be5caee7SBarry Smith + snes - the SNES context 3179be5caee7SBarry Smith - force - PETSC_TRUE require at least one iteration 3180be5caee7SBarry Smith 3181be5caee7SBarry Smith Options Database Keys: 3182be5caee7SBarry Smith . -snes_force_iteration <force> - Sets forcing an iteration 3183be5caee7SBarry Smith 3184be5caee7SBarry Smith Notes: 3185be5caee7SBarry Smith This is used sometimes with TS to prevent TS from detecting a false steady state solution 3186be5caee7SBarry Smith 3187be5caee7SBarry Smith Level: intermediate 3188be5caee7SBarry Smith 3189be5caee7SBarry Smith .keywords: SNES, nonlinear, set, convergence, tolerances 3190be5caee7SBarry Smith 3191be5caee7SBarry Smith .seealso: SNESSetTrustRegionTolerance(), SNESSetDivergenceTolerance() 3192be5caee7SBarry Smith @*/ 3193be5caee7SBarry Smith PetscErrorCode SNESSetForceIteration(SNES snes,PetscBool force) 3194be5caee7SBarry Smith { 3195be5caee7SBarry Smith PetscFunctionBegin; 3196be5caee7SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 3197be5caee7SBarry Smith snes->forceiteration = force; 3198be5caee7SBarry Smith PetscFunctionReturn(0); 3199be5caee7SBarry Smith } 3200be5caee7SBarry Smith 3201be5caee7SBarry Smith 3202be5caee7SBarry Smith /*@ 3203d7a720efSLois Curfman McInnes SNESSetTolerances - Sets various parameters used in convergence tests. 32049b94acceSBarry Smith 32053f9fe445SBarry Smith Logically Collective on SNES 3206c7afd0dbSLois Curfman McInnes 32079b94acceSBarry Smith Input Parameters: 3208c7afd0dbSLois Curfman McInnes + snes - the SNES context 320970441072SBarry Smith . abstol - absolute convergence tolerance 321033174efeSLois Curfman McInnes . rtol - relative convergence tolerance 32115358d0d4SBarry Smith . stol - convergence tolerance in terms of the norm of the change in the solution between steps, || delta x || < stol*|| x || 321233174efeSLois Curfman McInnes . maxit - maximum number of iterations 3213c7afd0dbSLois Curfman McInnes - maxf - maximum number of function evaluations 3214fee21e36SBarry Smith 321533174efeSLois Curfman McInnes Options Database Keys: 321670441072SBarry Smith + -snes_atol <abstol> - Sets abstol 3217c7afd0dbSLois Curfman McInnes . -snes_rtol <rtol> - Sets rtol 3218c7afd0dbSLois Curfman McInnes . -snes_stol <stol> - Sets stol 3219c7afd0dbSLois Curfman McInnes . -snes_max_it <maxit> - Sets maxit 3220c7afd0dbSLois Curfman McInnes - -snes_max_funcs <maxf> - Sets maxf 32219b94acceSBarry Smith 3222d7a720efSLois Curfman McInnes Notes: 32239b94acceSBarry Smith The default maximum number of iterations is 50. 32249b94acceSBarry Smith The default maximum number of function evaluations is 1000. 32259b94acceSBarry Smith 322636851e7fSLois Curfman McInnes Level: intermediate 322736851e7fSLois Curfman McInnes 322833174efeSLois Curfman McInnes .keywords: SNES, nonlinear, set, convergence, tolerances 32299b94acceSBarry Smith 3230be5caee7SBarry Smith .seealso: SNESSetTrustRegionTolerance(), SNESSetDivergenceTolerance(), SNESSetForceIteration() 32319b94acceSBarry Smith @*/ 32327087cfbeSBarry Smith PetscErrorCode SNESSetTolerances(SNES snes,PetscReal abstol,PetscReal rtol,PetscReal stol,PetscInt maxit,PetscInt maxf) 32339b94acceSBarry Smith { 32343a40ed3dSBarry Smith PetscFunctionBegin; 32350700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 3236c5eb9154SBarry Smith PetscValidLogicalCollectiveReal(snes,abstol,2); 3237c5eb9154SBarry Smith PetscValidLogicalCollectiveReal(snes,rtol,3); 3238c5eb9154SBarry Smith PetscValidLogicalCollectiveReal(snes,stol,4); 3239c5eb9154SBarry Smith PetscValidLogicalCollectiveInt(snes,maxit,5); 3240c5eb9154SBarry Smith PetscValidLogicalCollectiveInt(snes,maxf,6); 3241c5eb9154SBarry Smith 3242ab54825eSJed Brown if (abstol != PETSC_DEFAULT) { 324357622a8eSBarry Smith if (abstol < 0.0) SETERRQ1(PetscObjectComm((PetscObject)snes),PETSC_ERR_ARG_OUTOFRANGE,"Absolute tolerance %g must be non-negative",(double)abstol); 3244ab54825eSJed Brown snes->abstol = abstol; 3245ab54825eSJed Brown } 3246ab54825eSJed Brown if (rtol != PETSC_DEFAULT) { 324757622a8eSBarry 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); 3248ab54825eSJed Brown snes->rtol = rtol; 3249ab54825eSJed Brown } 3250ab54825eSJed Brown if (stol != PETSC_DEFAULT) { 325157622a8eSBarry Smith if (stol < 0.0) SETERRQ1(PetscObjectComm((PetscObject)snes),PETSC_ERR_ARG_OUTOFRANGE,"Step tolerance %g must be non-negative",(double)stol); 3252c60f73f4SPeter Brune snes->stol = stol; 3253ab54825eSJed Brown } 3254ab54825eSJed Brown if (maxit != PETSC_DEFAULT) { 3255ce94432eSBarry Smith if (maxit < 0) SETERRQ1(PetscObjectComm((PetscObject)snes),PETSC_ERR_ARG_OUTOFRANGE,"Maximum number of iterations %D must be non-negative",maxit); 3256ab54825eSJed Brown snes->max_its = maxit; 3257ab54825eSJed Brown } 3258ab54825eSJed Brown if (maxf != PETSC_DEFAULT) { 3259ce94432eSBarry Smith if (maxf < 0) SETERRQ1(PetscObjectComm((PetscObject)snes),PETSC_ERR_ARG_OUTOFRANGE,"Maximum number of function evaluations %D must be non-negative",maxf); 3260ab54825eSJed Brown snes->max_funcs = maxf; 3261ab54825eSJed Brown } 326288976e71SPeter Brune snes->tolerancesset = PETSC_TRUE; 32633a40ed3dSBarry Smith PetscFunctionReturn(0); 32649b94acceSBarry Smith } 32659b94acceSBarry Smith 3266e4d06f11SPatrick Farrell /*@ 3267e4d06f11SPatrick Farrell SNESSetDivergenceTolerance - Sets the divergence tolerance used for the SNES divergence test. 3268e4d06f11SPatrick Farrell 3269e4d06f11SPatrick Farrell Logically Collective on SNES 3270e4d06f11SPatrick Farrell 3271e4d06f11SPatrick Farrell Input Parameters: 3272e4d06f11SPatrick Farrell + snes - the SNES context 3273e4d06f11SPatrick Farrell - divtol - the divergence tolerance. Use -1 to deactivate the test. 3274e4d06f11SPatrick Farrell 3275e4d06f11SPatrick Farrell Options Database Keys: 3276e4d06f11SPatrick Farrell + -snes_divergence_tolerance <divtol> - Sets divtol 3277e4d06f11SPatrick Farrell 3278e4d06f11SPatrick Farrell Notes: 3279e4d06f11SPatrick Farrell The default divergence tolerance is 1e4. 3280e4d06f11SPatrick Farrell 3281e4d06f11SPatrick Farrell Level: intermediate 3282e4d06f11SPatrick Farrell 3283e4d06f11SPatrick Farrell .keywords: SNES, nonlinear, set, divergence, tolerance 3284e4d06f11SPatrick Farrell 3285e4d06f11SPatrick Farrell .seealso: SNESSetTolerances(), SNESGetDivergenceTolerance 3286e4d06f11SPatrick Farrell @*/ 3287e4d06f11SPatrick Farrell PetscErrorCode SNESSetDivergenceTolerance(SNES snes,PetscReal divtol) 3288e4d06f11SPatrick Farrell { 3289e4d06f11SPatrick Farrell PetscFunctionBegin; 3290e4d06f11SPatrick Farrell PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 3291e4d06f11SPatrick Farrell PetscValidLogicalCollectiveReal(snes,divtol,2); 3292e4d06f11SPatrick Farrell 3293e4d06f11SPatrick Farrell if (divtol != PETSC_DEFAULT) { 3294e4d06f11SPatrick Farrell snes->divtol = divtol; 3295e4d06f11SPatrick Farrell } 3296e4d06f11SPatrick Farrell else { 3297e4d06f11SPatrick Farrell snes->divtol = 1.0e4; 3298e4d06f11SPatrick Farrell } 3299e4d06f11SPatrick Farrell PetscFunctionReturn(0); 3300e4d06f11SPatrick Farrell } 3301e4d06f11SPatrick Farrell 33029b94acceSBarry Smith /*@ 330333174efeSLois Curfman McInnes SNESGetTolerances - Gets various parameters used in convergence tests. 330433174efeSLois Curfman McInnes 3305c7afd0dbSLois Curfman McInnes Not Collective 3306c7afd0dbSLois Curfman McInnes 330733174efeSLois Curfman McInnes Input Parameters: 3308c7afd0dbSLois Curfman McInnes + snes - the SNES context 330985385478SLisandro Dalcin . atol - absolute convergence tolerance 331033174efeSLois Curfman McInnes . rtol - relative convergence tolerance 331133174efeSLois Curfman McInnes . stol - convergence tolerance in terms of the norm 331233174efeSLois Curfman McInnes of the change in the solution between steps 331333174efeSLois Curfman McInnes . maxit - maximum number of iterations 3314c7afd0dbSLois Curfman McInnes - maxf - maximum number of function evaluations 3315fee21e36SBarry Smith 331633174efeSLois Curfman McInnes Notes: 33170298fd71SBarry Smith The user can specify NULL for any parameter that is not needed. 331833174efeSLois Curfman McInnes 331936851e7fSLois Curfman McInnes Level: intermediate 332036851e7fSLois Curfman McInnes 332133174efeSLois Curfman McInnes .keywords: SNES, nonlinear, get, convergence, tolerances 332233174efeSLois Curfman McInnes 332333174efeSLois Curfman McInnes .seealso: SNESSetTolerances() 332433174efeSLois Curfman McInnes @*/ 33257087cfbeSBarry Smith PetscErrorCode SNESGetTolerances(SNES snes,PetscReal *atol,PetscReal *rtol,PetscReal *stol,PetscInt *maxit,PetscInt *maxf) 332633174efeSLois Curfman McInnes { 33273a40ed3dSBarry Smith PetscFunctionBegin; 33280700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 332985385478SLisandro Dalcin if (atol) *atol = snes->abstol; 333033174efeSLois Curfman McInnes if (rtol) *rtol = snes->rtol; 3331c60f73f4SPeter Brune if (stol) *stol = snes->stol; 333233174efeSLois Curfman McInnes if (maxit) *maxit = snes->max_its; 333333174efeSLois Curfman McInnes if (maxf) *maxf = snes->max_funcs; 33343a40ed3dSBarry Smith PetscFunctionReturn(0); 333533174efeSLois Curfman McInnes } 333633174efeSLois Curfman McInnes 3337e4d06f11SPatrick Farrell /*@ 3338e4d06f11SPatrick Farrell SNESGetDivergenceTolerance - Gets divergence tolerance used in divergence test. 3339e4d06f11SPatrick Farrell 3340e4d06f11SPatrick Farrell Not Collective 3341e4d06f11SPatrick Farrell 3342e4d06f11SPatrick Farrell Input Parameters: 3343e4d06f11SPatrick Farrell + snes - the SNES context 3344e4d06f11SPatrick Farrell - divtol - divergence tolerance 3345e4d06f11SPatrick Farrell 3346e4d06f11SPatrick Farrell Level: intermediate 3347e4d06f11SPatrick Farrell 3348e4d06f11SPatrick Farrell .keywords: SNES, nonlinear, get, divergence, tolerance 3349e4d06f11SPatrick Farrell 3350e4d06f11SPatrick Farrell .seealso: SNESSetDivergenceTolerance() 3351e4d06f11SPatrick Farrell @*/ 3352e4d06f11SPatrick Farrell PetscErrorCode SNESGetDivergenceTolerance(SNES snes,PetscReal *divtol) 3353e4d06f11SPatrick Farrell { 3354e4d06f11SPatrick Farrell PetscFunctionBegin; 3355e4d06f11SPatrick Farrell PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 3356e4d06f11SPatrick Farrell if (divtol) *divtol = snes->divtol; 3357e4d06f11SPatrick Farrell PetscFunctionReturn(0); 3358e4d06f11SPatrick Farrell } 3359e4d06f11SPatrick Farrell 336033174efeSLois Curfman McInnes /*@ 33619b94acceSBarry Smith SNESSetTrustRegionTolerance - Sets the trust region parameter tolerance. 33629b94acceSBarry Smith 33633f9fe445SBarry Smith Logically Collective on SNES 3364fee21e36SBarry Smith 3365c7afd0dbSLois Curfman McInnes Input Parameters: 3366c7afd0dbSLois Curfman McInnes + snes - the SNES context 3367c7afd0dbSLois Curfman McInnes - tol - tolerance 3368c7afd0dbSLois Curfman McInnes 33699b94acceSBarry Smith Options Database Key: 3370c7afd0dbSLois Curfman McInnes . -snes_trtol <tol> - Sets tol 33719b94acceSBarry Smith 337236851e7fSLois Curfman McInnes Level: intermediate 337336851e7fSLois Curfman McInnes 33749b94acceSBarry Smith .keywords: SNES, nonlinear, set, trust region, tolerance 33759b94acceSBarry Smith 33762492ecdbSBarry Smith .seealso: SNESSetTolerances() 33779b94acceSBarry Smith @*/ 33787087cfbeSBarry Smith PetscErrorCode SNESSetTrustRegionTolerance(SNES snes,PetscReal tol) 33799b94acceSBarry Smith { 33803a40ed3dSBarry Smith PetscFunctionBegin; 33810700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 3382c5eb9154SBarry Smith PetscValidLogicalCollectiveReal(snes,tol,2); 33839b94acceSBarry Smith snes->deltatol = tol; 33843a40ed3dSBarry Smith PetscFunctionReturn(0); 33859b94acceSBarry Smith } 33869b94acceSBarry Smith 3387df9fa365SBarry Smith /* 3388df9fa365SBarry Smith Duplicate the lg monitors for SNES from KSP; for some reason with 3389df9fa365SBarry Smith dynamic libraries things don't work under Sun4 if we just use 3390df9fa365SBarry Smith macros instead of functions 3391df9fa365SBarry Smith */ 3392d96771aaSLisandro Dalcin PetscErrorCode SNESMonitorLGResidualNorm(SNES snes,PetscInt it,PetscReal norm,void *ctx) 3393ce1608b8SBarry Smith { 3394dfbe8321SBarry Smith PetscErrorCode ierr; 3395ce1608b8SBarry Smith 3396ce1608b8SBarry Smith PetscFunctionBegin; 33970700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 3398d96771aaSLisandro Dalcin ierr = KSPMonitorLGResidualNorm((KSP)snes,it,norm,ctx);CHKERRQ(ierr); 3399ce1608b8SBarry Smith PetscFunctionReturn(0); 3400ce1608b8SBarry Smith } 3401ce1608b8SBarry Smith 3402d96771aaSLisandro Dalcin PetscErrorCode SNESMonitorLGCreate(MPI_Comm comm,const char host[],const char label[],int x,int y,int m,int n,PetscDrawLG *lgctx) 3403df9fa365SBarry Smith { 3404dfbe8321SBarry Smith PetscErrorCode ierr; 3405df9fa365SBarry Smith 3406df9fa365SBarry Smith PetscFunctionBegin; 3407d96771aaSLisandro Dalcin ierr = KSPMonitorLGResidualNormCreate(comm,host,label,x,y,m,n,lgctx);CHKERRQ(ierr); 3408df9fa365SBarry Smith PetscFunctionReturn(0); 3409df9fa365SBarry Smith } 3410df9fa365SBarry Smith 34116ba87a44SLisandro Dalcin PETSC_INTERN PetscErrorCode SNESMonitorRange_Private(SNES,PetscInt,PetscReal*); 34126ba87a44SLisandro Dalcin 34137087cfbeSBarry Smith PetscErrorCode SNESMonitorLGRange(SNES snes,PetscInt n,PetscReal rnorm,void *monctx) 3414b271bb04SBarry Smith { 3415b271bb04SBarry Smith PetscDrawLG lg; 3416b271bb04SBarry Smith PetscErrorCode ierr; 3417b271bb04SBarry Smith PetscReal x,y,per; 3418b271bb04SBarry Smith PetscViewer v = (PetscViewer)monctx; 3419b271bb04SBarry Smith static PetscReal prev; /* should be in the context */ 3420b271bb04SBarry Smith PetscDraw draw; 3421b271bb04SBarry Smith 3422459f5d12SBarry Smith PetscFunctionBegin; 34234d4332d5SBarry Smith PetscValidHeaderSpecific(v,PETSC_VIEWER_CLASSID,4); 3424b271bb04SBarry Smith ierr = PetscViewerDrawGetDrawLG(v,0,&lg);CHKERRQ(ierr); 3425b271bb04SBarry Smith if (!n) {ierr = PetscDrawLGReset(lg);CHKERRQ(ierr);} 3426b271bb04SBarry Smith ierr = PetscDrawLGGetDraw(lg,&draw);CHKERRQ(ierr); 3427b271bb04SBarry Smith ierr = PetscDrawSetTitle(draw,"Residual norm");CHKERRQ(ierr); 3428b271bb04SBarry Smith x = (PetscReal)n; 342977b4d14cSPeter Brune if (rnorm > 0.0) y = PetscLog10Real(rnorm); 343094c9c6d3SKarl Rupp else y = -15.0; 3431b271bb04SBarry Smith ierr = PetscDrawLGAddPoint(lg,&x,&y);CHKERRQ(ierr); 34326934998bSLisandro Dalcin if (n < 20 || !(n % 5) || snes->reason) { 3433b271bb04SBarry Smith ierr = PetscDrawLGDraw(lg);CHKERRQ(ierr); 34346934998bSLisandro Dalcin ierr = PetscDrawLGSave(lg);CHKERRQ(ierr); 3435b271bb04SBarry Smith } 3436b271bb04SBarry Smith 3437b271bb04SBarry Smith ierr = PetscViewerDrawGetDrawLG(v,1,&lg);CHKERRQ(ierr); 3438b271bb04SBarry Smith if (!n) {ierr = PetscDrawLGReset(lg);CHKERRQ(ierr);} 3439b271bb04SBarry Smith ierr = PetscDrawLGGetDraw(lg,&draw);CHKERRQ(ierr); 3440b271bb04SBarry Smith ierr = PetscDrawSetTitle(draw,"% elemts > .2*max elemt");CHKERRQ(ierr); 3441b271bb04SBarry Smith ierr = SNESMonitorRange_Private(snes,n,&per);CHKERRQ(ierr); 3442b271bb04SBarry Smith x = (PetscReal)n; 3443b271bb04SBarry Smith y = 100.0*per; 3444b271bb04SBarry Smith ierr = PetscDrawLGAddPoint(lg,&x,&y);CHKERRQ(ierr); 34456934998bSLisandro Dalcin if (n < 20 || !(n % 5) || snes->reason) { 3446b271bb04SBarry Smith ierr = PetscDrawLGDraw(lg);CHKERRQ(ierr); 34476934998bSLisandro Dalcin ierr = PetscDrawLGSave(lg);CHKERRQ(ierr); 3448b271bb04SBarry Smith } 3449b271bb04SBarry Smith 3450b271bb04SBarry Smith ierr = PetscViewerDrawGetDrawLG(v,2,&lg);CHKERRQ(ierr); 3451b271bb04SBarry Smith if (!n) {prev = rnorm;ierr = PetscDrawLGReset(lg);CHKERRQ(ierr);} 3452b271bb04SBarry Smith ierr = PetscDrawLGGetDraw(lg,&draw);CHKERRQ(ierr); 3453b271bb04SBarry Smith ierr = PetscDrawSetTitle(draw,"(norm -oldnorm)/oldnorm");CHKERRQ(ierr); 3454b271bb04SBarry Smith x = (PetscReal)n; 3455b271bb04SBarry Smith y = (prev - rnorm)/prev; 3456b271bb04SBarry Smith ierr = PetscDrawLGAddPoint(lg,&x,&y);CHKERRQ(ierr); 34576934998bSLisandro Dalcin if (n < 20 || !(n % 5) || snes->reason) { 3458b271bb04SBarry Smith ierr = PetscDrawLGDraw(lg);CHKERRQ(ierr); 34596934998bSLisandro Dalcin ierr = PetscDrawLGSave(lg);CHKERRQ(ierr); 3460b271bb04SBarry Smith } 3461b271bb04SBarry Smith 3462b271bb04SBarry Smith ierr = PetscViewerDrawGetDrawLG(v,3,&lg);CHKERRQ(ierr); 3463b271bb04SBarry Smith if (!n) {ierr = PetscDrawLGReset(lg);CHKERRQ(ierr);} 3464b271bb04SBarry Smith ierr = PetscDrawLGGetDraw(lg,&draw);CHKERRQ(ierr); 3465b271bb04SBarry Smith ierr = PetscDrawSetTitle(draw,"(norm -oldnorm)/oldnorm*(% > .2 max)");CHKERRQ(ierr); 3466b271bb04SBarry Smith x = (PetscReal)n; 3467b271bb04SBarry Smith y = (prev - rnorm)/(prev*per); 3468b271bb04SBarry Smith if (n > 2) { /*skip initial crazy value */ 3469b271bb04SBarry Smith ierr = PetscDrawLGAddPoint(lg,&x,&y);CHKERRQ(ierr); 3470b271bb04SBarry Smith } 34716934998bSLisandro Dalcin if (n < 20 || !(n % 5) || snes->reason) { 3472b271bb04SBarry Smith ierr = PetscDrawLGDraw(lg);CHKERRQ(ierr); 34736934998bSLisandro Dalcin ierr = PetscDrawLGSave(lg);CHKERRQ(ierr); 3474b271bb04SBarry Smith } 3475b271bb04SBarry Smith prev = rnorm; 3476b271bb04SBarry Smith PetscFunctionReturn(0); 3477b271bb04SBarry Smith } 3478b271bb04SBarry Smith 3479228d79bcSJed Brown /*@ 3480228d79bcSJed Brown SNESMonitor - runs the user provided monitor routines, if they exist 3481228d79bcSJed Brown 3482228d79bcSJed Brown Collective on SNES 3483228d79bcSJed Brown 3484228d79bcSJed Brown Input Parameters: 3485228d79bcSJed Brown + snes - nonlinear solver context obtained from SNESCreate() 3486228d79bcSJed Brown . iter - iteration number 3487228d79bcSJed Brown - rnorm - relative norm of the residual 3488228d79bcSJed Brown 3489228d79bcSJed Brown Notes: 3490228d79bcSJed Brown This routine is called by the SNES implementations. 3491228d79bcSJed Brown It does not typically need to be called by the user. 3492228d79bcSJed Brown 3493228d79bcSJed Brown Level: developer 3494228d79bcSJed Brown 3495228d79bcSJed Brown .seealso: SNESMonitorSet() 3496228d79bcSJed Brown @*/ 34977a03ce2fSLisandro Dalcin PetscErrorCode SNESMonitor(SNES snes,PetscInt iter,PetscReal rnorm) 34987a03ce2fSLisandro Dalcin { 34997a03ce2fSLisandro Dalcin PetscErrorCode ierr; 35007a03ce2fSLisandro Dalcin PetscInt i,n = snes->numbermonitors; 35017a03ce2fSLisandro Dalcin 35027a03ce2fSLisandro Dalcin PetscFunctionBegin; 35035edff71fSBarry Smith ierr = VecLockPush(snes->vec_sol);CHKERRQ(ierr); 35047a03ce2fSLisandro Dalcin for (i=0; i<n; i++) { 35057a03ce2fSLisandro Dalcin ierr = (*snes->monitor[i])(snes,iter,rnorm,snes->monitorcontext[i]);CHKERRQ(ierr); 35067a03ce2fSLisandro Dalcin } 35075edff71fSBarry Smith ierr = VecLockPop(snes->vec_sol);CHKERRQ(ierr); 35087a03ce2fSLisandro Dalcin PetscFunctionReturn(0); 35097a03ce2fSLisandro Dalcin } 35107a03ce2fSLisandro Dalcin 35119b94acceSBarry Smith /* ------------ Routines to set performance monitoring options ----------- */ 35129b94acceSBarry Smith 3513bf388a1fSBarry Smith /*MC 3514bf388a1fSBarry Smith SNESMonitorFunction - functional form passed to SNESMonitorSet() to monitor convergence of nonlinear solver 3515bf388a1fSBarry Smith 3516bf388a1fSBarry Smith Synopsis: 3517aaa7dc30SBarry Smith #include <petscsnes.h> 3518bf388a1fSBarry Smith $ PetscErrorCode SNESMonitorFunction(SNES snes,PetscInt its, PetscReal norm,void *mctx) 3519bf388a1fSBarry Smith 3520bf388a1fSBarry Smith + snes - the SNES context 3521bf388a1fSBarry Smith . its - iteration number 3522bf388a1fSBarry Smith . norm - 2-norm function value (may be estimated) 3523bf388a1fSBarry Smith - mctx - [optional] monitoring context 3524bf388a1fSBarry Smith 3525878cb397SSatish Balay Level: advanced 3526878cb397SSatish Balay 3527bf388a1fSBarry Smith .seealso: SNESMonitorSet(), SNESMonitorGet() 3528bf388a1fSBarry Smith M*/ 3529bf388a1fSBarry Smith 35309b94acceSBarry Smith /*@C 3531a6570f20SBarry Smith SNESMonitorSet - Sets an ADDITIONAL function that is to be used at every 35329b94acceSBarry Smith iteration of the nonlinear solver to display the iteration's 35339b94acceSBarry Smith progress. 35349b94acceSBarry Smith 35353f9fe445SBarry Smith Logically Collective on SNES 3536fee21e36SBarry Smith 3537c7afd0dbSLois Curfman McInnes Input Parameters: 3538c7afd0dbSLois Curfman McInnes + snes - the SNES context 35396e4dcb14SBarry Smith . f - the monitor function, see SNESMonitorFunction for the calling sequence 3540b8a78c4aSBarry Smith . mctx - [optional] user-defined context for private data for the 35410298fd71SBarry Smith monitor routine (use NULL if no context is desired) 3542b3006f0bSLois Curfman McInnes - monitordestroy - [optional] routine that frees monitor context 35430298fd71SBarry Smith (may be NULL) 35449b94acceSBarry Smith 35459665c990SLois Curfman McInnes Options Database Keys: 3546a6570f20SBarry Smith + -snes_monitor - sets SNESMonitorDefault() 35474619e776SBarry Smith . -snes_monitor_lg_residualnorm - sets line graph monitor, 3548a6570f20SBarry Smith uses SNESMonitorLGCreate() 3549cca6129bSJed Brown - -snes_monitor_cancel - cancels all monitors that have 3550c7afd0dbSLois Curfman McInnes been hardwired into a code by 3551a6570f20SBarry Smith calls to SNESMonitorSet(), but 3552c7afd0dbSLois Curfman McInnes does not cancel those set via 3553c7afd0dbSLois Curfman McInnes the options database. 35549665c990SLois Curfman McInnes 3555639f9d9dSBarry Smith Notes: 35566bc08f3fSLois Curfman McInnes Several different monitoring routines may be set by calling 3557a6570f20SBarry Smith SNESMonitorSet() multiple times; all will be called in the 35586bc08f3fSLois Curfman McInnes order in which they were set. 3559639f9d9dSBarry Smith 3560025f1a04SBarry Smith Fortran notes: Only a single monitor function can be set for each SNES object 3561025f1a04SBarry Smith 356236851e7fSLois Curfman McInnes Level: intermediate 356336851e7fSLois Curfman McInnes 35649b94acceSBarry Smith .keywords: SNES, nonlinear, set, monitor 35659b94acceSBarry Smith 3566bf388a1fSBarry Smith .seealso: SNESMonitorDefault(), SNESMonitorCancel(), SNESMonitorFunction 35679b94acceSBarry Smith @*/ 35686e4dcb14SBarry Smith PetscErrorCode SNESMonitorSet(SNES snes,PetscErrorCode (*f)(SNES,PetscInt,PetscReal,void*),void *mctx,PetscErrorCode (*monitordestroy)(void**)) 35699b94acceSBarry Smith { 3570b90d0a6eSBarry Smith PetscInt i; 3571649052a6SBarry Smith PetscErrorCode ierr; 357278064530SBarry Smith PetscBool identical; 3573b90d0a6eSBarry Smith 35743a40ed3dSBarry Smith PetscFunctionBegin; 35750700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 3576b90d0a6eSBarry Smith for (i=0; i<snes->numbermonitors;i++) { 357778064530SBarry Smith ierr = PetscMonitorCompare((PetscErrorCode (*)(void))f,mctx,monitordestroy,(PetscErrorCode (*)(void))snes->monitor[i],snes->monitorcontext[i],snes->monitordestroy[i],&identical);CHKERRQ(ierr); 357878064530SBarry Smith if (identical) PetscFunctionReturn(0); 3579649052a6SBarry Smith } 358078064530SBarry Smith if (snes->numbermonitors >= MAXSNESMONITORS) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Too many monitors set"); 35816e4dcb14SBarry Smith snes->monitor[snes->numbermonitors] = f; 3582b8a78c4aSBarry Smith snes->monitordestroy[snes->numbermonitors] = monitordestroy; 3583639f9d9dSBarry Smith snes->monitorcontext[snes->numbermonitors++] = (void*)mctx; 35843a40ed3dSBarry Smith PetscFunctionReturn(0); 35859b94acceSBarry Smith } 35869b94acceSBarry Smith 3587a278d85bSSatish Balay /*@ 3588a6570f20SBarry Smith SNESMonitorCancel - Clears all the monitor functions for a SNES object. 35895cd90555SBarry Smith 35903f9fe445SBarry Smith Logically Collective on SNES 3591c7afd0dbSLois Curfman McInnes 35925cd90555SBarry Smith Input Parameters: 35935cd90555SBarry Smith . snes - the SNES context 35945cd90555SBarry Smith 35951a480d89SAdministrator Options Database Key: 3596a6570f20SBarry Smith . -snes_monitor_cancel - cancels all monitors that have been hardwired 3597a6570f20SBarry Smith into a code by calls to SNESMonitorSet(), but does not cancel those 3598c7afd0dbSLois Curfman McInnes set via the options database 35995cd90555SBarry Smith 36005cd90555SBarry Smith Notes: 36015cd90555SBarry Smith There is no way to clear one specific monitor from a SNES object. 36025cd90555SBarry Smith 360336851e7fSLois Curfman McInnes Level: intermediate 360436851e7fSLois Curfman McInnes 36055cd90555SBarry Smith .keywords: SNES, nonlinear, set, monitor 36065cd90555SBarry Smith 3607a6570f20SBarry Smith .seealso: SNESMonitorDefault(), SNESMonitorSet() 36085cd90555SBarry Smith @*/ 36097087cfbeSBarry Smith PetscErrorCode SNESMonitorCancel(SNES snes) 36105cd90555SBarry Smith { 3611d952e501SBarry Smith PetscErrorCode ierr; 3612d952e501SBarry Smith PetscInt i; 3613d952e501SBarry Smith 36145cd90555SBarry Smith PetscFunctionBegin; 36150700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 3616d952e501SBarry Smith for (i=0; i<snes->numbermonitors; i++) { 3617d952e501SBarry Smith if (snes->monitordestroy[i]) { 36183c4aec1bSBarry Smith ierr = (*snes->monitordestroy[i])(&snes->monitorcontext[i]);CHKERRQ(ierr); 3619d952e501SBarry Smith } 3620d952e501SBarry Smith } 36215cd90555SBarry Smith snes->numbermonitors = 0; 36225cd90555SBarry Smith PetscFunctionReturn(0); 36235cd90555SBarry Smith } 36245cd90555SBarry Smith 3625bf388a1fSBarry Smith /*MC 3626bf388a1fSBarry Smith SNESConvergenceTestFunction - functional form used for testing of convergence of nonlinear solver 3627bf388a1fSBarry Smith 3628bf388a1fSBarry Smith Synopsis: 3629aaa7dc30SBarry Smith #include <petscsnes.h> 3630bf388a1fSBarry Smith $ PetscErrorCode SNESConvergenceTest(SNES snes,PetscInt it,PetscReal xnorm,PetscReal gnorm,PetscReal f,SNESConvergedReason *reason,void *cctx) 3631bf388a1fSBarry Smith 3632bf388a1fSBarry Smith + snes - the SNES context 3633bf388a1fSBarry Smith . it - current iteration (0 is the first and is before any Newton step) 3634bf388a1fSBarry Smith . cctx - [optional] convergence context 3635bf388a1fSBarry Smith . reason - reason for convergence/divergence 3636bf388a1fSBarry Smith . xnorm - 2-norm of current iterate 3637bf388a1fSBarry Smith . gnorm - 2-norm of current step 3638bf388a1fSBarry Smith - f - 2-norm of function 3639bf388a1fSBarry Smith 3640878cb397SSatish Balay Level: intermediate 3641bf388a1fSBarry Smith 3642bf388a1fSBarry Smith .seealso: SNESSetConvergenceTest(), SNESGetConvergenceTest() 3643bf388a1fSBarry Smith M*/ 3644bf388a1fSBarry Smith 36459b94acceSBarry Smith /*@C 36469b94acceSBarry Smith SNESSetConvergenceTest - Sets the function that is to be used 36479b94acceSBarry Smith to test for convergence of the nonlinear iterative solution. 36489b94acceSBarry Smith 36493f9fe445SBarry Smith Logically Collective on SNES 3650fee21e36SBarry Smith 3651c7afd0dbSLois Curfman McInnes Input Parameters: 3652c7afd0dbSLois Curfman McInnes + snes - the SNES context 3653bf388a1fSBarry Smith . SNESConvergenceTestFunction - routine to test for convergence 36540298fd71SBarry Smith . cctx - [optional] context for private data for the convergence routine (may be NULL) 36550298fd71SBarry Smith - destroy - [optional] destructor for the context (may be NULL; NULL_FUNCTION in Fortran) 36569b94acceSBarry Smith 365736851e7fSLois Curfman McInnes Level: advanced 365836851e7fSLois Curfman McInnes 36599b94acceSBarry Smith .keywords: SNES, nonlinear, set, convergence, test 36609b94acceSBarry Smith 3661e2a6519dSDmitry Karpeev .seealso: SNESConvergedDefault(), SNESConvergedSkip(), SNESConvergenceTestFunction 36629b94acceSBarry Smith @*/ 3663bf388a1fSBarry Smith PetscErrorCode SNESSetConvergenceTest(SNES snes,PetscErrorCode (*SNESConvergenceTestFunction)(SNES,PetscInt,PetscReal,PetscReal,PetscReal,SNESConvergedReason*,void*),void *cctx,PetscErrorCode (*destroy)(void*)) 36649b94acceSBarry Smith { 36657f7931b9SBarry Smith PetscErrorCode ierr; 36667f7931b9SBarry Smith 36673a40ed3dSBarry Smith PetscFunctionBegin; 36680700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 3669e2a6519dSDmitry Karpeev if (!SNESConvergenceTestFunction) SNESConvergenceTestFunction = SNESConvergedSkip; 36707f7931b9SBarry Smith if (snes->ops->convergeddestroy) { 36717f7931b9SBarry Smith ierr = (*snes->ops->convergeddestroy)(snes->cnvP);CHKERRQ(ierr); 36727f7931b9SBarry Smith } 3673bf388a1fSBarry Smith snes->ops->converged = SNESConvergenceTestFunction; 36747f7931b9SBarry Smith snes->ops->convergeddestroy = destroy; 367585385478SLisandro Dalcin snes->cnvP = cctx; 36763a40ed3dSBarry Smith PetscFunctionReturn(0); 36779b94acceSBarry Smith } 36789b94acceSBarry Smith 367952baeb72SSatish Balay /*@ 3680184914b5SBarry Smith SNESGetConvergedReason - Gets the reason the SNES iteration was stopped. 3681184914b5SBarry Smith 3682184914b5SBarry Smith Not Collective 3683184914b5SBarry Smith 3684184914b5SBarry Smith Input Parameter: 3685184914b5SBarry Smith . snes - the SNES context 3686184914b5SBarry Smith 3687184914b5SBarry Smith Output Parameter: 36884d0a8057SBarry Smith . reason - negative value indicates diverged, positive value converged, see SNESConvergedReason or the 3689184914b5SBarry Smith manual pages for the individual convergence tests for complete lists 3690184914b5SBarry Smith 36916a4d7782SBarry Smith Options Database: 36926a4d7782SBarry Smith . -snes_converged_reason - prints the reason to standard out 36936a4d7782SBarry Smith 3694184914b5SBarry Smith Level: intermediate 3695184914b5SBarry Smith 36966a4d7782SBarry 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. 3697184914b5SBarry Smith 3698184914b5SBarry Smith .keywords: SNES, nonlinear, set, convergence, test 3699184914b5SBarry Smith 370033866048SMatthew G. Knepley .seealso: SNESSetConvergenceTest(), SNESSetConvergedReason(), SNESConvergedReason 3701184914b5SBarry Smith @*/ 37027087cfbeSBarry Smith PetscErrorCode SNESGetConvergedReason(SNES snes,SNESConvergedReason *reason) 3703184914b5SBarry Smith { 3704184914b5SBarry Smith PetscFunctionBegin; 37050700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 37064482741eSBarry Smith PetscValidPointer(reason,2); 3707184914b5SBarry Smith *reason = snes->reason; 3708184914b5SBarry Smith PetscFunctionReturn(0); 3709184914b5SBarry Smith } 3710184914b5SBarry Smith 371133866048SMatthew G. Knepley /*@ 371233866048SMatthew G. Knepley SNESSetConvergedReason - Sets the reason the SNES iteration was stopped. 371333866048SMatthew G. Knepley 371433866048SMatthew G. Knepley Not Collective 371533866048SMatthew G. Knepley 371633866048SMatthew G. Knepley Input Parameters: 371733866048SMatthew G. Knepley + snes - the SNES context 371833866048SMatthew G. Knepley - reason - negative value indicates diverged, positive value converged, see SNESConvergedReason or the 371933866048SMatthew G. Knepley manual pages for the individual convergence tests for complete lists 372033866048SMatthew G. Knepley 372133866048SMatthew G. Knepley Level: intermediate 372233866048SMatthew G. Knepley 372333866048SMatthew G. Knepley .keywords: SNES, nonlinear, set, convergence, test 372433866048SMatthew G. Knepley .seealso: SNESGetConvergedReason(), SNESSetConvergenceTest(), SNESConvergedReason 372533866048SMatthew G. Knepley @*/ 372633866048SMatthew G. Knepley PetscErrorCode SNESSetConvergedReason(SNES snes,SNESConvergedReason reason) 372733866048SMatthew G. Knepley { 372833866048SMatthew G. Knepley PetscFunctionBegin; 372933866048SMatthew G. Knepley PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 373033866048SMatthew G. Knepley snes->reason = reason; 373133866048SMatthew G. Knepley PetscFunctionReturn(0); 373233866048SMatthew G. Knepley } 373333866048SMatthew G. Knepley 3734c9005455SLois Curfman McInnes /*@ 3735c9005455SLois Curfman McInnes SNESSetConvergenceHistory - Sets the array used to hold the convergence history. 3736c9005455SLois Curfman McInnes 37373f9fe445SBarry Smith Logically Collective on SNES 3738fee21e36SBarry Smith 3739c7afd0dbSLois Curfman McInnes Input Parameters: 3740c7afd0dbSLois Curfman McInnes + snes - iterative context obtained from SNESCreate() 37418c7482ecSBarry Smith . a - array to hold history, this array will contain the function norms computed at each step 3742cd5578b5SBarry Smith . its - integer array holds the number of linear iterations for each solve. 3743758f92a0SBarry Smith . na - size of a and its 374464731454SLois Curfman McInnes - reset - PETSC_TRUE indicates each new nonlinear solve resets the history counter to zero, 3745758f92a0SBarry Smith else it continues storing new values for new nonlinear solves after the old ones 3746c7afd0dbSLois Curfman McInnes 3747308dcc3eSBarry Smith Notes: 37480298fd71SBarry Smith If 'a' and 'its' are NULL then space is allocated for the history. If 'na' PETSC_DECIDE or PETSC_DEFAULT then a 3749308dcc3eSBarry Smith default array of length 10000 is allocated. 3750308dcc3eSBarry Smith 3751c9005455SLois Curfman McInnes This routine is useful, e.g., when running a code for purposes 3752c9005455SLois Curfman McInnes of accurate performance monitoring, when no I/O should be done 3753c9005455SLois Curfman McInnes during the section of code that is being timed. 3754c9005455SLois Curfman McInnes 375536851e7fSLois Curfman McInnes Level: intermediate 375636851e7fSLois Curfman McInnes 3757c9005455SLois Curfman McInnes .keywords: SNES, set, convergence, history 3758758f92a0SBarry Smith 375908405cd6SLois Curfman McInnes .seealso: SNESGetConvergenceHistory() 3760758f92a0SBarry Smith 3761c9005455SLois Curfman McInnes @*/ 37627087cfbeSBarry Smith PetscErrorCode SNESSetConvergenceHistory(SNES snes,PetscReal a[],PetscInt its[],PetscInt na,PetscBool reset) 3763c9005455SLois Curfman McInnes { 3764308dcc3eSBarry Smith PetscErrorCode ierr; 3765308dcc3eSBarry Smith 37663a40ed3dSBarry Smith PetscFunctionBegin; 37670700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 37687a1ec6d4SBarry Smith if (a) PetscValidScalarPointer(a,2); 3769a562a398SLisandro Dalcin if (its) PetscValidIntPointer(its,3); 37707a1ec6d4SBarry Smith if (!a) { 3771308dcc3eSBarry Smith if (na == PETSC_DECIDE || na == PETSC_DEFAULT) na = 1000; 37727fdeb8b9SBarry Smith ierr = PetscCalloc1(na,&a);CHKERRQ(ierr); 37737fdeb8b9SBarry Smith ierr = PetscCalloc1(na,&its);CHKERRQ(ierr); 3774f5af7f23SKarl Rupp 3775308dcc3eSBarry Smith snes->conv_malloc = PETSC_TRUE; 3776308dcc3eSBarry Smith } 3777c9005455SLois Curfman McInnes snes->conv_hist = a; 3778758f92a0SBarry Smith snes->conv_hist_its = its; 3779758f92a0SBarry Smith snes->conv_hist_max = na; 3780a12bc48fSLisandro Dalcin snes->conv_hist_len = 0; 3781758f92a0SBarry Smith snes->conv_hist_reset = reset; 3782758f92a0SBarry Smith PetscFunctionReturn(0); 3783758f92a0SBarry Smith } 3784758f92a0SBarry Smith 3785308dcc3eSBarry Smith #if defined(PETSC_HAVE_MATLAB_ENGINE) 3786c6db04a5SJed Brown #include <engine.h> /* MATLAB include file */ 3787c6db04a5SJed Brown #include <mex.h> /* MATLAB include file */ 378899e0435eSBarry Smith 37898cc058d9SJed Brown PETSC_EXTERN mxArray *SNESGetConvergenceHistoryMatlab(SNES snes) 3790308dcc3eSBarry Smith { 3791308dcc3eSBarry Smith mxArray *mat; 3792308dcc3eSBarry Smith PetscInt i; 3793308dcc3eSBarry Smith PetscReal *ar; 3794308dcc3eSBarry Smith 3795308dcc3eSBarry Smith PetscFunctionBegin; 3796308dcc3eSBarry Smith mat = mxCreateDoubleMatrix(snes->conv_hist_len,1,mxREAL); 3797308dcc3eSBarry Smith ar = (PetscReal*) mxGetData(mat); 3798f5af7f23SKarl Rupp for (i=0; i<snes->conv_hist_len; i++) ar[i] = snes->conv_hist[i]; 3799308dcc3eSBarry Smith PetscFunctionReturn(mat); 3800308dcc3eSBarry Smith } 3801308dcc3eSBarry Smith #endif 3802308dcc3eSBarry Smith 38030c4c9dddSBarry Smith /*@C 3804758f92a0SBarry Smith SNESGetConvergenceHistory - Gets the array used to hold the convergence history. 3805758f92a0SBarry Smith 38063f9fe445SBarry Smith Not Collective 3807758f92a0SBarry Smith 3808758f92a0SBarry Smith Input Parameter: 3809758f92a0SBarry Smith . snes - iterative context obtained from SNESCreate() 3810758f92a0SBarry Smith 3811758f92a0SBarry Smith Output Parameters: 3812758f92a0SBarry Smith . a - array to hold history 3813758f92a0SBarry Smith . its - integer array holds the number of linear iterations (or 3814758f92a0SBarry Smith negative if not converged) for each solve. 3815758f92a0SBarry Smith - na - size of a and its 3816758f92a0SBarry Smith 3817758f92a0SBarry Smith Notes: 3818758f92a0SBarry Smith The calling sequence for this routine in Fortran is 3819758f92a0SBarry Smith $ call SNESGetConvergenceHistory(SNES snes, integer na, integer ierr) 3820758f92a0SBarry Smith 3821758f92a0SBarry Smith This routine is useful, e.g., when running a code for purposes 3822758f92a0SBarry Smith of accurate performance monitoring, when no I/O should be done 3823758f92a0SBarry Smith during the section of code that is being timed. 3824758f92a0SBarry Smith 3825758f92a0SBarry Smith Level: intermediate 3826758f92a0SBarry Smith 3827758f92a0SBarry Smith .keywords: SNES, get, convergence, history 3828758f92a0SBarry Smith 3829758f92a0SBarry Smith .seealso: SNESSetConvergencHistory() 3830758f92a0SBarry Smith 3831758f92a0SBarry Smith @*/ 38327087cfbeSBarry Smith PetscErrorCode SNESGetConvergenceHistory(SNES snes,PetscReal *a[],PetscInt *its[],PetscInt *na) 3833758f92a0SBarry Smith { 3834758f92a0SBarry Smith PetscFunctionBegin; 38350700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 3836758f92a0SBarry Smith if (a) *a = snes->conv_hist; 3837758f92a0SBarry Smith if (its) *its = snes->conv_hist_its; 3838758f92a0SBarry Smith if (na) *na = snes->conv_hist_len; 38393a40ed3dSBarry Smith PetscFunctionReturn(0); 3840c9005455SLois Curfman McInnes } 3841c9005455SLois Curfman McInnes 3842ac226902SBarry Smith /*@C 384376b2cf59SMatthew Knepley SNESSetUpdate - Sets the general-purpose update function called 3844eec8f646SJed Brown at the beginning of every iteration of the nonlinear solve. Specifically 38457e4bb74cSBarry Smith it is called just before the Jacobian is "evaluated". 384676b2cf59SMatthew Knepley 38473f9fe445SBarry Smith Logically Collective on SNES 384876b2cf59SMatthew Knepley 384976b2cf59SMatthew Knepley Input Parameters: 385076b2cf59SMatthew Knepley . snes - The nonlinear solver context 385176b2cf59SMatthew Knepley . func - The function 385276b2cf59SMatthew Knepley 385376b2cf59SMatthew Knepley Calling sequence of func: 3854b5d30489SBarry Smith . func (SNES snes, PetscInt step); 385576b2cf59SMatthew Knepley 385676b2cf59SMatthew Knepley . step - The current step of the iteration 385776b2cf59SMatthew Knepley 3858fe97e370SBarry Smith Level: advanced 3859fe97e370SBarry Smith 3860fe97e370SBarry 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() 3861fe97e370SBarry Smith This is not used by most users. 386276b2cf59SMatthew Knepley 386376b2cf59SMatthew Knepley .keywords: SNES, update 3864b5d30489SBarry Smith 38658d359177SBarry Smith .seealso SNESSetJacobian(), SNESSolve() 386676b2cf59SMatthew Knepley @*/ 38677087cfbeSBarry Smith PetscErrorCode SNESSetUpdate(SNES snes, PetscErrorCode (*func)(SNES, PetscInt)) 386876b2cf59SMatthew Knepley { 386976b2cf59SMatthew Knepley PetscFunctionBegin; 38700700a824SBarry Smith PetscValidHeaderSpecific(snes, SNES_CLASSID,1); 3871e7788613SBarry Smith snes->ops->update = func; 387276b2cf59SMatthew Knepley PetscFunctionReturn(0); 387376b2cf59SMatthew Knepley } 387476b2cf59SMatthew Knepley 38759b94acceSBarry Smith /* 38769b94acceSBarry Smith SNESScaleStep_Private - Scales a step so that its length is less than the 38779b94acceSBarry Smith positive parameter delta. 38789b94acceSBarry Smith 38799b94acceSBarry Smith Input Parameters: 3880c7afd0dbSLois Curfman McInnes + snes - the SNES context 38819b94acceSBarry Smith . y - approximate solution of linear system 38829b94acceSBarry Smith . fnorm - 2-norm of current function 3883c7afd0dbSLois Curfman McInnes - delta - trust region size 38849b94acceSBarry Smith 38859b94acceSBarry Smith Output Parameters: 3886c7afd0dbSLois Curfman McInnes + gpnorm - predicted function norm at the new point, assuming local 38879b94acceSBarry Smith linearization. The value is zero if the step lies within the trust 38889b94acceSBarry Smith region, and exceeds zero otherwise. 3889c7afd0dbSLois Curfman McInnes - ynorm - 2-norm of the step 38909b94acceSBarry Smith 38919b94acceSBarry Smith Note: 389204d7464bSBarry Smith For non-trust region methods such as SNESNEWTONLS, the parameter delta 38939b94acceSBarry Smith is set to be the maximum allowable step size. 38949b94acceSBarry Smith 38959b94acceSBarry Smith .keywords: SNES, nonlinear, scale, step 38969b94acceSBarry Smith */ 3897dfbe8321SBarry Smith PetscErrorCode SNESScaleStep_Private(SNES snes,Vec y,PetscReal *fnorm,PetscReal *delta,PetscReal *gpnorm,PetscReal *ynorm) 38989b94acceSBarry Smith { 3899064f8208SBarry Smith PetscReal nrm; 3900ea709b57SSatish Balay PetscScalar cnorm; 3901dfbe8321SBarry Smith PetscErrorCode ierr; 39023a40ed3dSBarry Smith 39033a40ed3dSBarry Smith PetscFunctionBegin; 39040700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 39050700a824SBarry Smith PetscValidHeaderSpecific(y,VEC_CLASSID,2); 3906c9780b6fSBarry Smith PetscCheckSameComm(snes,1,y,2); 3907184914b5SBarry Smith 3908064f8208SBarry Smith ierr = VecNorm(y,NORM_2,&nrm);CHKERRQ(ierr); 3909064f8208SBarry Smith if (nrm > *delta) { 3910064f8208SBarry Smith nrm = *delta/nrm; 3911064f8208SBarry Smith *gpnorm = (1.0 - nrm)*(*fnorm); 3912064f8208SBarry Smith cnorm = nrm; 39132dcb1b2aSMatthew Knepley ierr = VecScale(y,cnorm);CHKERRQ(ierr); 39149b94acceSBarry Smith *ynorm = *delta; 39159b94acceSBarry Smith } else { 39169b94acceSBarry Smith *gpnorm = 0.0; 3917064f8208SBarry Smith *ynorm = nrm; 39189b94acceSBarry Smith } 39193a40ed3dSBarry Smith PetscFunctionReturn(0); 39209b94acceSBarry Smith } 39219b94acceSBarry Smith 39222a359c20SBarry Smith /*@ 39232a359c20SBarry Smith SNESReasonView - Displays the reason a SNES solve converged or diverged to a viewer 39242a359c20SBarry Smith 39252a359c20SBarry Smith Collective on SNES 39262a359c20SBarry Smith 39272a359c20SBarry Smith Parameter: 39282a359c20SBarry Smith + snes - iterative context obtained from SNESCreate() 39292a359c20SBarry Smith - viewer - the viewer to display the reason 39302a359c20SBarry Smith 39312a359c20SBarry Smith 39322a359c20SBarry Smith Options Database Keys: 39332a359c20SBarry Smith . -snes_converged_reason - print reason for converged or diverged, also prints number of iterations 39342a359c20SBarry Smith 39352a359c20SBarry Smith Level: beginner 39362a359c20SBarry Smith 39372a359c20SBarry Smith .keywords: SNES, solve, linear system 39382a359c20SBarry Smith 39392a359c20SBarry Smith .seealso: SNESCreate(), SNESSetUp(), SNESDestroy(), SNESSetTolerances(), SNESConvergedDefault() 39402a359c20SBarry Smith 39412a359c20SBarry Smith @*/ 39422a359c20SBarry Smith PetscErrorCode SNESReasonView(SNES snes,PetscViewer viewer) 39432a359c20SBarry Smith { 39442a359c20SBarry Smith PetscErrorCode ierr; 39452a359c20SBarry Smith PetscBool isAscii; 39462a359c20SBarry Smith 39472a359c20SBarry Smith PetscFunctionBegin; 39482a359c20SBarry Smith ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERASCII,&isAscii);CHKERRQ(ierr); 39492a359c20SBarry Smith if (isAscii) { 39502a359c20SBarry Smith ierr = PetscViewerASCIIAddTab(viewer,((PetscObject)snes)->tablevel);CHKERRQ(ierr); 39512a359c20SBarry Smith if (snes->reason > 0) { 39522a359c20SBarry Smith if (((PetscObject) snes)->prefix) { 39532a359c20SBarry Smith ierr = PetscViewerASCIIPrintf(viewer,"Nonlinear %s solve converged due to %s iterations %D\n",((PetscObject) snes)->prefix,SNESConvergedReasons[snes->reason],snes->iter);CHKERRQ(ierr); 39542a359c20SBarry Smith } else { 39552a359c20SBarry Smith ierr = PetscViewerASCIIPrintf(viewer,"Nonlinear solve converged due to %s iterations %D\n",SNESConvergedReasons[snes->reason],snes->iter);CHKERRQ(ierr); 39562a359c20SBarry Smith } 39572a359c20SBarry Smith } else { 39582a359c20SBarry Smith if (((PetscObject) snes)->prefix) { 39592a359c20SBarry 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); 39602a359c20SBarry Smith } else { 39612a359c20SBarry Smith ierr = PetscViewerASCIIPrintf(viewer,"Nonlinear solve did not converge due to %s iterations %D\n",SNESConvergedReasons[snes->reason],snes->iter);CHKERRQ(ierr); 39622a359c20SBarry Smith } 39632a359c20SBarry Smith } 39642a359c20SBarry Smith ierr = PetscViewerASCIISubtractTab(viewer,((PetscObject)snes)->tablevel);CHKERRQ(ierr); 39652a359c20SBarry Smith } 39662a359c20SBarry Smith PetscFunctionReturn(0); 39672a359c20SBarry Smith } 39682a359c20SBarry Smith 39692a359c20SBarry Smith /*@C 39702a359c20SBarry Smith SNESReasonViewFromOptions - Processes command line options to determine if/how a SNESReason is to be viewed. 39712a359c20SBarry Smith 39722a359c20SBarry Smith Collective on SNES 39732a359c20SBarry Smith 39742a359c20SBarry Smith Input Parameters: 39752a359c20SBarry Smith . snes - the SNES object 39762a359c20SBarry Smith 39772a359c20SBarry Smith Level: intermediate 39782a359c20SBarry Smith 39792a359c20SBarry Smith @*/ 39802a359c20SBarry Smith PetscErrorCode SNESReasonViewFromOptions(SNES snes) 39812a359c20SBarry Smith { 39822a359c20SBarry Smith PetscErrorCode ierr; 39832a359c20SBarry Smith PetscViewer viewer; 39842a359c20SBarry Smith PetscBool flg; 39852a359c20SBarry Smith static PetscBool incall = PETSC_FALSE; 39862a359c20SBarry Smith PetscViewerFormat format; 39872a359c20SBarry Smith 39882a359c20SBarry Smith PetscFunctionBegin; 39892a359c20SBarry Smith if (incall) PetscFunctionReturn(0); 39902a359c20SBarry Smith incall = PETSC_TRUE; 39912a359c20SBarry Smith ierr = PetscOptionsGetViewer(PetscObjectComm((PetscObject)snes),((PetscObject)snes)->prefix,"-snes_converged_reason",&viewer,&format,&flg);CHKERRQ(ierr); 39922a359c20SBarry Smith if (flg) { 39932a359c20SBarry Smith ierr = PetscViewerPushFormat(viewer,format);CHKERRQ(ierr); 39942a359c20SBarry Smith ierr = SNESReasonView(snes,viewer);CHKERRQ(ierr); 39952a359c20SBarry Smith ierr = PetscViewerPopFormat(viewer);CHKERRQ(ierr); 39962a359c20SBarry Smith ierr = PetscViewerDestroy(&viewer);CHKERRQ(ierr); 39972a359c20SBarry Smith } 39982a359c20SBarry Smith incall = PETSC_FALSE; 39992a359c20SBarry Smith PetscFunctionReturn(0); 40002a359c20SBarry Smith } 40012a359c20SBarry Smith 40026ce558aeSBarry Smith /*@C 4003f69a0ea3SMatthew Knepley SNESSolve - Solves a nonlinear system F(x) = b. 4004f69a0ea3SMatthew Knepley Call SNESSolve() after calling SNESCreate() and optional routines of the form SNESSetXXX(). 40059b94acceSBarry Smith 4006c7afd0dbSLois Curfman McInnes Collective on SNES 4007c7afd0dbSLois Curfman McInnes 4008b2002411SLois Curfman McInnes Input Parameters: 4009c7afd0dbSLois Curfman McInnes + snes - the SNES context 40100298fd71SBarry Smith . b - the constant part of the equation F(x) = b, or NULL to use zero. 401185385478SLisandro Dalcin - x - the solution vector. 40129b94acceSBarry Smith 4013b2002411SLois Curfman McInnes Notes: 40148ddd3da0SLois Curfman McInnes The user should initialize the vector,x, with the initial guess 40158ddd3da0SLois Curfman McInnes for the nonlinear solve prior to calling SNESSolve. In particular, 40168ddd3da0SLois Curfman McInnes to employ an initial guess of zero, the user should explicitly set 40178ddd3da0SLois Curfman McInnes this vector to zero by calling VecSet(). 40188ddd3da0SLois Curfman McInnes 401936851e7fSLois Curfman McInnes Level: beginner 402036851e7fSLois Curfman McInnes 40219b94acceSBarry Smith .keywords: SNES, nonlinear, solve 40229b94acceSBarry Smith 4023c0df2a02SJed Brown .seealso: SNESCreate(), SNESDestroy(), SNESSetFunction(), SNESSetJacobian(), SNESSetGridSequence(), SNESGetSolution() 40249b94acceSBarry Smith @*/ 40257087cfbeSBarry Smith PetscErrorCode SNESSolve(SNES snes,Vec b,Vec x) 40269b94acceSBarry Smith { 4027dfbe8321SBarry Smith PetscErrorCode ierr; 4028ace3abfcSBarry Smith PetscBool flg; 4029efd51863SBarry Smith PetscInt grid; 40300298fd71SBarry Smith Vec xcreated = NULL; 4031caa4e7f2SJed Brown DM dm; 4032052efed2SBarry Smith 40333a40ed3dSBarry Smith PetscFunctionBegin; 40340700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 4035a69afd8bSBarry Smith if (x) PetscValidHeaderSpecific(x,VEC_CLASSID,3); 4036a69afd8bSBarry Smith if (x) PetscCheckSameComm(snes,1,x,3); 40370700a824SBarry Smith if (b) PetscValidHeaderSpecific(b,VEC_CLASSID,2); 403885385478SLisandro Dalcin if (b) PetscCheckSameComm(snes,1,b,2); 403985385478SLisandro Dalcin 404006fc46c8SMatthew G. Knepley { 404106fc46c8SMatthew G. Knepley PetscViewer viewer; 404206fc46c8SMatthew G. Knepley PetscViewerFormat format; 404306fc46c8SMatthew G. Knepley PetscBool flg; 404406fc46c8SMatthew G. Knepley static PetscBool incall = PETSC_FALSE; 404506fc46c8SMatthew G. Knepley 404606fc46c8SMatthew G. Knepley if (!incall) { 404706fc46c8SMatthew G. Knepley ierr = PetscOptionsGetViewer(PetscObjectComm((PetscObject) snes), ((PetscObject) snes)->prefix, "-snes_convergence_estimate", &viewer, &format, &flg);CHKERRQ(ierr); 404806fc46c8SMatthew G. Knepley if (flg) { 404906fc46c8SMatthew G. Knepley PetscConvEst conv; 405006fc46c8SMatthew G. Knepley PetscReal alpha; /* Convergence rate of the solution error in the L_2 norm */ 405106fc46c8SMatthew G. Knepley 405206fc46c8SMatthew G. Knepley incall = PETSC_TRUE; 405306fc46c8SMatthew G. Knepley ierr = PetscConvEstCreate(PetscObjectComm((PetscObject) snes), &conv);CHKERRQ(ierr); 405406fc46c8SMatthew G. Knepley ierr = PetscConvEstSetSolver(conv, snes);CHKERRQ(ierr); 405506fc46c8SMatthew G. Knepley ierr = PetscConvEstSetFromOptions(conv);CHKERRQ(ierr); 40560955ed61SMatthew G. Knepley ierr = PetscConvEstSetUp(conv);CHKERRQ(ierr); 405706fc46c8SMatthew G. Knepley ierr = PetscConvEstGetConvRate(conv, &alpha);CHKERRQ(ierr); 405806fc46c8SMatthew G. Knepley ierr = PetscViewerPushFormat(viewer, format);CHKERRQ(ierr); 405906fc46c8SMatthew G. Knepley ierr = PetscConvEstRateView(conv, alpha, viewer);CHKERRQ(ierr); 406006fc46c8SMatthew G. Knepley ierr = PetscViewerPopFormat(viewer);CHKERRQ(ierr); 406106fc46c8SMatthew G. Knepley ierr = PetscViewerDestroy(&viewer);CHKERRQ(ierr); 406206fc46c8SMatthew G. Knepley ierr = PetscConvEstDestroy(&conv);CHKERRQ(ierr); 406306fc46c8SMatthew G. Knepley incall = PETSC_FALSE; 406406fc46c8SMatthew G. Knepley } 406506fc46c8SMatthew G. Knepley } 406606fc46c8SMatthew G. Knepley } 4067caa4e7f2SJed Brown if (!x) { 4068caa4e7f2SJed Brown ierr = SNESGetDM(snes,&dm);CHKERRQ(ierr); 4069caa4e7f2SJed Brown ierr = DMCreateGlobalVector(dm,&xcreated);CHKERRQ(ierr); 4070a69afd8bSBarry Smith x = xcreated; 4071a69afd8bSBarry Smith } 4072ce1779c8SBarry Smith ierr = SNESViewFromOptions(snes,NULL,"-snes_view_pre");CHKERRQ(ierr); 4073f05ece33SBarry Smith 4074ce94432eSBarry Smith for (grid=0; grid<snes->gridsequence; grid++) {ierr = PetscViewerASCIIPushTab(PETSC_VIEWER_STDOUT_(PetscObjectComm((PetscObject)snes)));CHKERRQ(ierr);} 4075efd51863SBarry Smith for (grid=0; grid<snes->gridsequence+1; grid++) { 4076efd51863SBarry Smith 407785385478SLisandro Dalcin /* set solution vector */ 4078efd51863SBarry Smith if (!grid) {ierr = PetscObjectReference((PetscObject)x);CHKERRQ(ierr);} 40796bf464f9SBarry Smith ierr = VecDestroy(&snes->vec_sol);CHKERRQ(ierr); 408085385478SLisandro Dalcin snes->vec_sol = x; 4081caa4e7f2SJed Brown ierr = SNESGetDM(snes,&dm);CHKERRQ(ierr); 4082caa4e7f2SJed Brown 4083caa4e7f2SJed Brown /* set affine vector if provided */ 408485385478SLisandro Dalcin if (b) { ierr = PetscObjectReference((PetscObject)b);CHKERRQ(ierr); } 40856bf464f9SBarry Smith ierr = VecDestroy(&snes->vec_rhs);CHKERRQ(ierr); 408685385478SLisandro Dalcin snes->vec_rhs = b; 408785385478SLisandro Dalcin 4088154060b5SMatthew G. Knepley if (snes->vec_func == snes->vec_sol) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_IDN,"Solution vector cannot be function vector"); 4089154060b5SMatthew 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"); 4090154060b5SMatthew G. Knepley if (!snes->vec_sol_update /* && snes->vec_sol */) { 4091154060b5SMatthew G. Knepley ierr = VecDuplicate(snes->vec_sol,&snes->vec_sol_update);CHKERRQ(ierr); 4092154060b5SMatthew G. Knepley ierr = PetscLogObjectParent((PetscObject)snes,(PetscObject)snes->vec_sol_update);CHKERRQ(ierr); 4093154060b5SMatthew G. Knepley } 4094154060b5SMatthew G. Knepley ierr = DMShellSetGlobalVector(dm,snes->vec_sol);CHKERRQ(ierr); 409570e92668SMatthew Knepley ierr = SNESSetUp(snes);CHKERRQ(ierr); 40963f149594SLisandro Dalcin 40977eee914bSBarry Smith if (!grid) { 40987eee914bSBarry Smith if (snes->ops->computeinitialguess) { 4099d25893d9SBarry Smith ierr = (*snes->ops->computeinitialguess)(snes,snes->vec_sol,snes->initialguessP);CHKERRQ(ierr); 4100d25893d9SBarry Smith } 4101dd568438SSatish Balay } 4102d25893d9SBarry Smith 4103abc0a331SBarry Smith if (snes->conv_hist_reset) snes->conv_hist_len = 0; 4104971e163fSPeter Brune if (snes->counters_reset) {snes->nfuncs = 0; snes->linear_its = 0; snes->numFailures = 0;} 4105d5e45103SBarry Smith 41063f149594SLisandro Dalcin ierr = PetscLogEventBegin(SNES_Solve,snes,0,0,0);CHKERRQ(ierr); 41074936397dSBarry Smith ierr = (*snes->ops->solve)(snes);CHKERRQ(ierr); 410885385478SLisandro Dalcin ierr = PetscLogEventEnd(SNES_Solve,snes,0,0,0);CHKERRQ(ierr); 410917186662SBarry Smith if (!snes->reason) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_PLIB,"Internal error, solver returned without setting converged reason"); 4110422a814eSBarry Smith snes->domainerror = PETSC_FALSE; /* clear the flag if it has been set */ 41113f149594SLisandro Dalcin 411237ec4e1aSPeter Brune if (snes->lagjac_persist) snes->jac_iter += snes->iter; 411337ec4e1aSPeter Brune if (snes->lagpre_persist) snes->pre_iter += snes->iter; 411437ec4e1aSPeter Brune 411527b0f280SBarry Smith ierr = PetscOptionsGetViewer(PetscObjectComm((PetscObject)snes),((PetscObject)snes)->prefix,"-snes_test_local_min",NULL,NULL,&flg);CHKERRQ(ierr); 4116da9b6338SBarry Smith if (flg && !PetscPreLoadingOn) { ierr = SNESTestLocalMin(snes);CHKERRQ(ierr); } 41172a359c20SBarry Smith ierr = SNESReasonViewFromOptions(snes);CHKERRQ(ierr); 41185968eb51SBarry Smith 4119ce94432eSBarry Smith if (snes->errorifnotconverged && snes->reason < 0) SETERRQ(PetscObjectComm((PetscObject)snes),PETSC_ERR_NOT_CONVERGED,"SNESSolve has not converged"); 41209c8e83a9SBarry Smith if (snes->reason < 0) break; 4121efd51863SBarry Smith if (grid < snes->gridsequence) { 4122efd51863SBarry Smith DM fine; 4123efd51863SBarry Smith Vec xnew; 4124efd51863SBarry Smith Mat interp; 4125efd51863SBarry Smith 4126ce94432eSBarry Smith ierr = DMRefine(snes->dm,PetscObjectComm((PetscObject)snes),&fine);CHKERRQ(ierr); 4127ce94432eSBarry Smith if (!fine) SETERRQ(PetscObjectComm((PetscObject)snes),PETSC_ERR_ARG_INCOMP,"DMRefine() did not perform any refinement, cannot continue grid sequencing"); 41280298fd71SBarry Smith ierr = DMCreateInterpolation(snes->dm,fine,&interp,NULL);CHKERRQ(ierr); 4129efd51863SBarry Smith ierr = DMCreateGlobalVector(fine,&xnew);CHKERRQ(ierr); 4130efd51863SBarry Smith ierr = MatInterpolate(interp,x,xnew);CHKERRQ(ierr); 4131c833c3b5SJed Brown ierr = DMInterpolate(snes->dm,interp,fine);CHKERRQ(ierr); 4132efd51863SBarry Smith ierr = MatDestroy(&interp);CHKERRQ(ierr); 4133efd51863SBarry Smith x = xnew; 4134efd51863SBarry Smith 4135efd51863SBarry Smith ierr = SNESReset(snes);CHKERRQ(ierr); 4136efd51863SBarry Smith ierr = SNESSetDM(snes,fine);CHKERRQ(ierr); 4137efd51863SBarry Smith ierr = DMDestroy(&fine);CHKERRQ(ierr); 4138ce94432eSBarry Smith ierr = PetscViewerASCIIPopTab(PETSC_VIEWER_STDOUT_(PetscObjectComm((PetscObject)snes)));CHKERRQ(ierr); 4139efd51863SBarry Smith } 4140efd51863SBarry Smith } 4141ce1779c8SBarry Smith ierr = SNESViewFromOptions(snes,NULL,"-snes_view");CHKERRQ(ierr); 4142685405a1SBarry Smith ierr = VecViewFromOptions(snes->vec_sol,(PetscObject)snes,"-snes_view_solution");CHKERRQ(ierr); 41433f7e2da0SPeter Brune 4144a69afd8bSBarry Smith ierr = VecDestroy(&xcreated);CHKERRQ(ierr); 4145e04113cfSBarry Smith ierr = PetscObjectSAWsBlock((PetscObject)snes);CHKERRQ(ierr); 41463a40ed3dSBarry Smith PetscFunctionReturn(0); 41479b94acceSBarry Smith } 41489b94acceSBarry Smith 41499b94acceSBarry Smith /* --------- Internal routines for SNES Package --------- */ 41509b94acceSBarry Smith 415182bf6240SBarry Smith /*@C 41524b0e389bSBarry Smith SNESSetType - Sets the method for the nonlinear solver. 41539b94acceSBarry Smith 4154fee21e36SBarry Smith Collective on SNES 4155fee21e36SBarry Smith 4156c7afd0dbSLois Curfman McInnes Input Parameters: 4157c7afd0dbSLois Curfman McInnes + snes - the SNES context 4158454a90a3SBarry Smith - type - a known method 4159c7afd0dbSLois Curfman McInnes 4160c7afd0dbSLois Curfman McInnes Options Database Key: 4161454a90a3SBarry Smith . -snes_type <type> - Sets the method; use -help for a list 416204d7464bSBarry Smith of available methods (for instance, newtonls or newtontr) 4163ae12b187SLois Curfman McInnes 41649b94acceSBarry Smith Notes: 4165e090d566SSatish Balay See "petsc/include/petscsnes.h" for available methods (for instance) 416604d7464bSBarry Smith + SNESNEWTONLS - Newton's method with line search 4167c7afd0dbSLois Curfman McInnes (systems of nonlinear equations) 416804d7464bSBarry Smith . SNESNEWTONTR - Newton's method with trust region 4169c7afd0dbSLois Curfman McInnes (systems of nonlinear equations) 41709b94acceSBarry Smith 4171ae12b187SLois Curfman McInnes Normally, it is best to use the SNESSetFromOptions() command and then 4172ae12b187SLois Curfman McInnes set the SNES solver type from the options database rather than by using 4173ae12b187SLois Curfman McInnes this routine. Using the options database provides the user with 4174ae12b187SLois Curfman McInnes maximum flexibility in evaluating the many nonlinear solvers. 4175ae12b187SLois Curfman McInnes The SNESSetType() routine is provided for those situations where it 4176ae12b187SLois Curfman McInnes is necessary to set the nonlinear solver independently of the command 4177ae12b187SLois Curfman McInnes line or options database. This might be the case, for example, when 4178ae12b187SLois Curfman McInnes the choice of solver changes during the execution of the program, 4179ae12b187SLois Curfman McInnes and the user's application is taking responsibility for choosing the 4180b0a32e0cSBarry Smith appropriate method. 418136851e7fSLois Curfman McInnes 41828f6c3df8SBarry Smith Developer Notes: SNESRegister() adds a constructor for a new SNESType to SNESList, SNESSetType() locates 41838f6c3df8SBarry Smith the constructor in that list and calls it to create the spexific object. 41848f6c3df8SBarry Smith 418536851e7fSLois Curfman McInnes Level: intermediate 4186a703fe33SLois Curfman McInnes 4187454a90a3SBarry Smith .keywords: SNES, set, type 4188435da068SBarry Smith 41898f6c3df8SBarry Smith .seealso: SNESType, SNESCreate(), SNESDestroy(), SNESGetType(), SNESSetFromOptions() 4190435da068SBarry Smith 41919b94acceSBarry Smith @*/ 419219fd82e9SBarry Smith PetscErrorCode SNESSetType(SNES snes,SNESType type) 41939b94acceSBarry Smith { 4194dfbe8321SBarry Smith PetscErrorCode ierr,(*r)(SNES); 4195ace3abfcSBarry Smith PetscBool match; 41963a40ed3dSBarry Smith 41973a40ed3dSBarry Smith PetscFunctionBegin; 41980700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 41994482741eSBarry Smith PetscValidCharPointer(type,2); 420082bf6240SBarry Smith 4201251f4c67SDmitry Karpeev ierr = PetscObjectTypeCompare((PetscObject)snes,type,&match);CHKERRQ(ierr); 42020f5bd95cSBarry Smith if (match) PetscFunctionReturn(0); 420392ff6ae8SBarry Smith 42041c9cd337SJed Brown ierr = PetscFunctionListFind(SNESList,type,&r);CHKERRQ(ierr); 4205e32f2f54SBarry Smith if (!r) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_UNKNOWN_TYPE,"Unable to find requested SNES type %s",type); 420675396ef9SLisandro Dalcin /* Destroy the previous private SNES context */ 4207b5c23020SJed Brown if (snes->ops->destroy) { 4208b5c23020SJed Brown ierr = (*(snes)->ops->destroy)(snes);CHKERRQ(ierr); 42090298fd71SBarry Smith snes->ops->destroy = NULL; 4210b5c23020SJed Brown } 421175396ef9SLisandro Dalcin /* Reinitialize function pointers in SNESOps structure */ 421275396ef9SLisandro Dalcin snes->ops->setup = 0; 421375396ef9SLisandro Dalcin snes->ops->solve = 0; 421475396ef9SLisandro Dalcin snes->ops->view = 0; 421575396ef9SLisandro Dalcin snes->ops->setfromoptions = 0; 421675396ef9SLisandro Dalcin snes->ops->destroy = 0; 421775396ef9SLisandro Dalcin /* Call the SNESCreate_XXX routine for this particular Nonlinear solver */ 421875396ef9SLisandro Dalcin snes->setupcalled = PETSC_FALSE; 4219f5af7f23SKarl Rupp 4220454a90a3SBarry Smith ierr = PetscObjectChangeTypeName((PetscObject)snes,type);CHKERRQ(ierr); 422103bfa161SLisandro Dalcin ierr = (*r)(snes);CHKERRQ(ierr); 42223a40ed3dSBarry Smith PetscFunctionReturn(0); 42239b94acceSBarry Smith } 42249b94acceSBarry Smith 42259b94acceSBarry Smith /*@C 42269a28b0a6SLois Curfman McInnes SNESGetType - Gets the SNES method type and name (as a string). 42279b94acceSBarry Smith 4228c7afd0dbSLois Curfman McInnes Not Collective 4229c7afd0dbSLois Curfman McInnes 42309b94acceSBarry Smith Input Parameter: 42314b0e389bSBarry Smith . snes - nonlinear solver context 42329b94acceSBarry Smith 42339b94acceSBarry Smith Output Parameter: 42343a7fca6bSBarry Smith . type - SNES method (a character string) 42359b94acceSBarry Smith 423636851e7fSLois Curfman McInnes Level: intermediate 423736851e7fSLois Curfman McInnes 4238454a90a3SBarry Smith .keywords: SNES, nonlinear, get, type, name 42399b94acceSBarry Smith @*/ 424019fd82e9SBarry Smith PetscErrorCode SNESGetType(SNES snes,SNESType *type) 42419b94acceSBarry Smith { 42423a40ed3dSBarry Smith PetscFunctionBegin; 42430700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 42444482741eSBarry Smith PetscValidPointer(type,2); 42457adad957SLisandro Dalcin *type = ((PetscObject)snes)->type_name; 42463a40ed3dSBarry Smith PetscFunctionReturn(0); 42479b94acceSBarry Smith } 42489b94acceSBarry Smith 42493cd8a7caSMatthew G. Knepley /*@ 42503cd8a7caSMatthew G. Knepley SNESSetSolution - Sets the solution vector for use by the SNES routines. 42513cd8a7caSMatthew G. Knepley 42523cd8a7caSMatthew G. Knepley Logically Collective on SNES and Vec 42533cd8a7caSMatthew G. Knepley 42543cd8a7caSMatthew G. Knepley Input Parameters: 42553cd8a7caSMatthew G. Knepley + snes - the SNES context obtained from SNESCreate() 42563cd8a7caSMatthew G. Knepley - u - the solution vector 42573cd8a7caSMatthew G. Knepley 42583cd8a7caSMatthew G. Knepley Level: beginner 42593cd8a7caSMatthew G. Knepley 42603cd8a7caSMatthew G. Knepley .keywords: SNES, set, solution 42613cd8a7caSMatthew G. Knepley @*/ 42623cd8a7caSMatthew G. Knepley PetscErrorCode SNESSetSolution(SNES snes, Vec u) 42633cd8a7caSMatthew G. Knepley { 42643cd8a7caSMatthew G. Knepley DM dm; 42653cd8a7caSMatthew G. Knepley PetscErrorCode ierr; 42663cd8a7caSMatthew G. Knepley 42673cd8a7caSMatthew G. Knepley PetscFunctionBegin; 42683cd8a7caSMatthew G. Knepley PetscValidHeaderSpecific(snes, SNES_CLASSID, 1); 42693cd8a7caSMatthew G. Knepley PetscValidHeaderSpecific(u, VEC_CLASSID, 2); 42703cd8a7caSMatthew G. Knepley ierr = PetscObjectReference((PetscObject) u);CHKERRQ(ierr); 42713cd8a7caSMatthew G. Knepley ierr = VecDestroy(&snes->vec_sol);CHKERRQ(ierr); 42723cd8a7caSMatthew G. Knepley 42733cd8a7caSMatthew G. Knepley snes->vec_sol = u; 42743cd8a7caSMatthew G. Knepley 42753cd8a7caSMatthew G. Knepley ierr = SNESGetDM(snes, &dm);CHKERRQ(ierr); 42763cd8a7caSMatthew G. Knepley ierr = DMShellSetGlobalVector(dm, u);CHKERRQ(ierr); 42773cd8a7caSMatthew G. Knepley PetscFunctionReturn(0); 42783cd8a7caSMatthew G. Knepley } 42793cd8a7caSMatthew G. Knepley 428052baeb72SSatish Balay /*@ 42819b94acceSBarry Smith SNESGetSolution - Returns the vector where the approximate solution is 4282c0df2a02SJed Brown stored. This is the fine grid solution when using SNESSetGridSequence(). 42839b94acceSBarry Smith 4284c7afd0dbSLois Curfman McInnes Not Collective, but Vec is parallel if SNES is parallel 4285c7afd0dbSLois Curfman McInnes 42869b94acceSBarry Smith Input Parameter: 42879b94acceSBarry Smith . snes - the SNES context 42889b94acceSBarry Smith 42899b94acceSBarry Smith Output Parameter: 42909b94acceSBarry Smith . x - the solution 42919b94acceSBarry Smith 429270e92668SMatthew Knepley Level: intermediate 429336851e7fSLois Curfman McInnes 42949b94acceSBarry Smith .keywords: SNES, nonlinear, get, solution 42959b94acceSBarry Smith 429685385478SLisandro Dalcin .seealso: SNESGetSolutionUpdate(), SNESGetFunction() 42979b94acceSBarry Smith @*/ 42987087cfbeSBarry Smith PetscErrorCode SNESGetSolution(SNES snes,Vec *x) 42999b94acceSBarry Smith { 43003a40ed3dSBarry Smith PetscFunctionBegin; 43010700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 43024482741eSBarry Smith PetscValidPointer(x,2); 430385385478SLisandro Dalcin *x = snes->vec_sol; 430470e92668SMatthew Knepley PetscFunctionReturn(0); 430570e92668SMatthew Knepley } 430670e92668SMatthew Knepley 430752baeb72SSatish Balay /*@ 43089b94acceSBarry Smith SNESGetSolutionUpdate - Returns the vector where the solution update is 43099b94acceSBarry Smith stored. 43109b94acceSBarry Smith 4311c7afd0dbSLois Curfman McInnes Not Collective, but Vec is parallel if SNES is parallel 4312c7afd0dbSLois Curfman McInnes 43139b94acceSBarry Smith Input Parameter: 43149b94acceSBarry Smith . snes - the SNES context 43159b94acceSBarry Smith 43169b94acceSBarry Smith Output Parameter: 43179b94acceSBarry Smith . x - the solution update 43189b94acceSBarry Smith 431936851e7fSLois Curfman McInnes Level: advanced 432036851e7fSLois Curfman McInnes 43219b94acceSBarry Smith .keywords: SNES, nonlinear, get, solution, update 43229b94acceSBarry Smith 432385385478SLisandro Dalcin .seealso: SNESGetSolution(), SNESGetFunction() 43249b94acceSBarry Smith @*/ 43257087cfbeSBarry Smith PetscErrorCode SNESGetSolutionUpdate(SNES snes,Vec *x) 43269b94acceSBarry Smith { 43273a40ed3dSBarry Smith PetscFunctionBegin; 43280700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 43294482741eSBarry Smith PetscValidPointer(x,2); 433085385478SLisandro Dalcin *x = snes->vec_sol_update; 43313a40ed3dSBarry Smith PetscFunctionReturn(0); 43329b94acceSBarry Smith } 43339b94acceSBarry Smith 43349b94acceSBarry Smith /*@C 43353638b69dSLois Curfman McInnes SNESGetFunction - Returns the vector where the function is stored. 43369b94acceSBarry Smith 4337a63bb30eSJed Brown Not Collective, but Vec is parallel if SNES is parallel. Collective if Vec is requested, but has not been created yet. 4338c7afd0dbSLois Curfman McInnes 43399b94acceSBarry Smith Input Parameter: 43409b94acceSBarry Smith . snes - the SNES context 43419b94acceSBarry Smith 43429b94acceSBarry Smith Output Parameter: 43430298fd71SBarry Smith + r - the vector that is used to store residuals (or NULL if you don't want it) 4344f8b49ee9SBarry Smith . f - the function (or NULL if you don't want it); see SNESFunction for calling sequence details 43450298fd71SBarry Smith - ctx - the function context (or NULL if you don't want it) 43469b94acceSBarry Smith 434736851e7fSLois Curfman McInnes Level: advanced 434836851e7fSLois Curfman McInnes 4349a86d99e1SLois Curfman McInnes .keywords: SNES, nonlinear, get, function 43509b94acceSBarry Smith 4351bf388a1fSBarry Smith .seealso: SNESSetFunction(), SNESGetSolution(), SNESFunction 43529b94acceSBarry Smith @*/ 4353f8b49ee9SBarry Smith PetscErrorCode SNESGetFunction(SNES snes,Vec *r,PetscErrorCode (**f)(SNES,Vec,Vec,void*),void **ctx) 43549b94acceSBarry Smith { 4355a63bb30eSJed Brown PetscErrorCode ierr; 43566cab3a1bSJed Brown DM dm; 4357a63bb30eSJed Brown 43583a40ed3dSBarry Smith PetscFunctionBegin; 43590700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 4360a63bb30eSJed Brown if (r) { 4361a63bb30eSJed Brown if (!snes->vec_func) { 4362a63bb30eSJed Brown if (snes->vec_rhs) { 4363a63bb30eSJed Brown ierr = VecDuplicate(snes->vec_rhs,&snes->vec_func);CHKERRQ(ierr); 4364a63bb30eSJed Brown } else if (snes->vec_sol) { 4365a63bb30eSJed Brown ierr = VecDuplicate(snes->vec_sol,&snes->vec_func);CHKERRQ(ierr); 4366a63bb30eSJed Brown } else if (snes->dm) { 4367a63bb30eSJed Brown ierr = DMCreateGlobalVector(snes->dm,&snes->vec_func);CHKERRQ(ierr); 4368a63bb30eSJed Brown } 4369a63bb30eSJed Brown } 4370a63bb30eSJed Brown *r = snes->vec_func; 4371a63bb30eSJed Brown } 43726cab3a1bSJed Brown ierr = SNESGetDM(snes,&dm);CHKERRQ(ierr); 4373f8b49ee9SBarry Smith ierr = DMSNESGetFunction(dm,f,ctx);CHKERRQ(ierr); 43743a40ed3dSBarry Smith PetscFunctionReturn(0); 43759b94acceSBarry Smith } 43769b94acceSBarry Smith 4377c79ef259SPeter Brune /*@C 4378be95d8f1SBarry Smith SNESGetNGS - Returns the NGS function and context. 4379c79ef259SPeter Brune 4380c79ef259SPeter Brune Input Parameter: 4381c79ef259SPeter Brune . snes - the SNES context 4382c79ef259SPeter Brune 4383c79ef259SPeter Brune Output Parameter: 4384be95d8f1SBarry Smith + f - the function (or NULL) see SNESNGSFunction for details 43850298fd71SBarry Smith - ctx - the function context (or NULL) 4386c79ef259SPeter Brune 4387c79ef259SPeter Brune Level: advanced 4388c79ef259SPeter Brune 4389c79ef259SPeter Brune .keywords: SNES, nonlinear, get, function 4390c79ef259SPeter Brune 4391be95d8f1SBarry Smith .seealso: SNESSetNGS(), SNESGetFunction() 4392c79ef259SPeter Brune @*/ 4393c79ef259SPeter Brune 4394be95d8f1SBarry Smith PetscErrorCode SNESGetNGS (SNES snes, PetscErrorCode (**f)(SNES, Vec, Vec, void*), void ** ctx) 4395646217ecSPeter Brune { 43966cab3a1bSJed Brown PetscErrorCode ierr; 43976cab3a1bSJed Brown DM dm; 43986cab3a1bSJed Brown 4399646217ecSPeter Brune PetscFunctionBegin; 4400646217ecSPeter Brune PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 44016cab3a1bSJed Brown ierr = SNESGetDM(snes,&dm);CHKERRQ(ierr); 4402be95d8f1SBarry Smith ierr = DMSNESGetNGS(dm,f,ctx);CHKERRQ(ierr); 4403646217ecSPeter Brune PetscFunctionReturn(0); 4404646217ecSPeter Brune } 4405646217ecSPeter Brune 44063c7409f5SSatish Balay /*@C 44073c7409f5SSatish Balay SNESSetOptionsPrefix - Sets the prefix used for searching for all 4408d850072dSLois Curfman McInnes SNES options in the database. 44093c7409f5SSatish Balay 44103f9fe445SBarry Smith Logically Collective on SNES 4411fee21e36SBarry Smith 4412c7afd0dbSLois Curfman McInnes Input Parameter: 4413c7afd0dbSLois Curfman McInnes + snes - the SNES context 4414c7afd0dbSLois Curfman McInnes - prefix - the prefix to prepend to all option names 4415c7afd0dbSLois Curfman McInnes 4416d850072dSLois Curfman McInnes Notes: 4417a83b1b31SSatish Balay A hyphen (-) must NOT be given at the beginning of the prefix name. 4418c7afd0dbSLois Curfman McInnes The first character of all runtime options is AUTOMATICALLY the hyphen. 4419d850072dSLois Curfman McInnes 442036851e7fSLois Curfman McInnes Level: advanced 442136851e7fSLois Curfman McInnes 44223c7409f5SSatish Balay .keywords: SNES, set, options, prefix, database 4423a86d99e1SLois Curfman McInnes 4424a86d99e1SLois Curfman McInnes .seealso: SNESSetFromOptions() 44253c7409f5SSatish Balay @*/ 44267087cfbeSBarry Smith PetscErrorCode SNESSetOptionsPrefix(SNES snes,const char prefix[]) 44273c7409f5SSatish Balay { 4428dfbe8321SBarry Smith PetscErrorCode ierr; 44293c7409f5SSatish Balay 44303a40ed3dSBarry Smith PetscFunctionBegin; 44310700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 4432639f9d9dSBarry Smith ierr = PetscObjectSetOptionsPrefix((PetscObject)snes,prefix);CHKERRQ(ierr); 44331cee3971SBarry Smith if (!snes->ksp) {ierr = SNESGetKSP(snes,&snes->ksp);CHKERRQ(ierr);} 443435f5d045SPeter Brune if (snes->linesearch) { 44357601faf0SJed Brown ierr = SNESGetLineSearch(snes,&snes->linesearch);CHKERRQ(ierr); 443608b6c495SPeter Brune ierr = PetscObjectSetOptionsPrefix((PetscObject)snes->linesearch,prefix);CHKERRQ(ierr); 443735f5d045SPeter Brune } 443835f5d045SPeter Brune ierr = KSPSetOptionsPrefix(snes->ksp,prefix);CHKERRQ(ierr); 44393a40ed3dSBarry Smith PetscFunctionReturn(0); 44403c7409f5SSatish Balay } 44413c7409f5SSatish Balay 44423c7409f5SSatish Balay /*@C 4443f525115eSLois Curfman McInnes SNESAppendOptionsPrefix - Appends to the prefix used for searching for all 4444d850072dSLois Curfman McInnes SNES options in the database. 44453c7409f5SSatish Balay 44463f9fe445SBarry Smith Logically Collective on SNES 4447fee21e36SBarry Smith 4448c7afd0dbSLois Curfman McInnes Input Parameters: 4449c7afd0dbSLois Curfman McInnes + snes - the SNES context 4450c7afd0dbSLois Curfman McInnes - prefix - the prefix to prepend to all option names 4451c7afd0dbSLois Curfman McInnes 4452d850072dSLois Curfman McInnes Notes: 4453a83b1b31SSatish Balay A hyphen (-) must NOT be given at the beginning of the prefix name. 4454c7afd0dbSLois Curfman McInnes The first character of all runtime options is AUTOMATICALLY the hyphen. 4455d850072dSLois Curfman McInnes 445636851e7fSLois Curfman McInnes Level: advanced 445736851e7fSLois Curfman McInnes 44583c7409f5SSatish Balay .keywords: SNES, append, options, prefix, database 4459a86d99e1SLois Curfman McInnes 4460a86d99e1SLois Curfman McInnes .seealso: SNESGetOptionsPrefix() 44613c7409f5SSatish Balay @*/ 44627087cfbeSBarry Smith PetscErrorCode SNESAppendOptionsPrefix(SNES snes,const char prefix[]) 44633c7409f5SSatish Balay { 4464dfbe8321SBarry Smith PetscErrorCode ierr; 44653c7409f5SSatish Balay 44663a40ed3dSBarry Smith PetscFunctionBegin; 44670700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 4468639f9d9dSBarry Smith ierr = PetscObjectAppendOptionsPrefix((PetscObject)snes,prefix);CHKERRQ(ierr); 44691cee3971SBarry Smith if (!snes->ksp) {ierr = SNESGetKSP(snes,&snes->ksp);CHKERRQ(ierr);} 447035f5d045SPeter Brune if (snes->linesearch) { 44717601faf0SJed Brown ierr = SNESGetLineSearch(snes,&snes->linesearch);CHKERRQ(ierr); 447208b6c495SPeter Brune ierr = PetscObjectAppendOptionsPrefix((PetscObject)snes->linesearch,prefix);CHKERRQ(ierr); 447335f5d045SPeter Brune } 447435f5d045SPeter Brune ierr = KSPAppendOptionsPrefix(snes->ksp,prefix);CHKERRQ(ierr); 44753a40ed3dSBarry Smith PetscFunctionReturn(0); 44763c7409f5SSatish Balay } 44773c7409f5SSatish Balay 44789ab63eb5SSatish Balay /*@C 44793c7409f5SSatish Balay SNESGetOptionsPrefix - Sets the prefix used for searching for all 44803c7409f5SSatish Balay SNES options in the database. 44813c7409f5SSatish Balay 4482c7afd0dbSLois Curfman McInnes Not Collective 4483c7afd0dbSLois Curfman McInnes 44843c7409f5SSatish Balay Input Parameter: 44853c7409f5SSatish Balay . snes - the SNES context 44863c7409f5SSatish Balay 44873c7409f5SSatish Balay Output Parameter: 44883c7409f5SSatish Balay . prefix - pointer to the prefix string used 44893c7409f5SSatish Balay 44904ef407dbSRichard Tran Mills Notes: On the fortran side, the user should pass in a string 'prefix' of 44919ab63eb5SSatish Balay sufficient length to hold the prefix. 44929ab63eb5SSatish Balay 449336851e7fSLois Curfman McInnes Level: advanced 449436851e7fSLois Curfman McInnes 44953c7409f5SSatish Balay .keywords: SNES, get, options, prefix, database 4496a86d99e1SLois Curfman McInnes 4497a86d99e1SLois Curfman McInnes .seealso: SNESAppendOptionsPrefix() 44983c7409f5SSatish Balay @*/ 44997087cfbeSBarry Smith PetscErrorCode SNESGetOptionsPrefix(SNES snes,const char *prefix[]) 45003c7409f5SSatish Balay { 4501dfbe8321SBarry Smith PetscErrorCode ierr; 45023c7409f5SSatish Balay 45033a40ed3dSBarry Smith PetscFunctionBegin; 45040700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 4505639f9d9dSBarry Smith ierr = PetscObjectGetOptionsPrefix((PetscObject)snes,prefix);CHKERRQ(ierr); 45063a40ed3dSBarry Smith PetscFunctionReturn(0); 45073c7409f5SSatish Balay } 45083c7409f5SSatish Balay 4509b2002411SLois Curfman McInnes 45103cea93caSBarry Smith /*@C 45111c84c290SBarry Smith SNESRegister - Adds a method to the nonlinear solver package. 45121c84c290SBarry Smith 45131c84c290SBarry Smith Not collective 45141c84c290SBarry Smith 45151c84c290SBarry Smith Input Parameters: 45161c84c290SBarry Smith + name_solver - name of a new user-defined solver 45171c84c290SBarry Smith - routine_create - routine to create method context 45181c84c290SBarry Smith 45191c84c290SBarry Smith Notes: 45201c84c290SBarry Smith SNESRegister() may be called multiple times to add several user-defined solvers. 45211c84c290SBarry Smith 45221c84c290SBarry Smith Sample usage: 45231c84c290SBarry Smith .vb 4524bdf89e91SBarry Smith SNESRegister("my_solver",MySolverCreate); 45251c84c290SBarry Smith .ve 45261c84c290SBarry Smith 45271c84c290SBarry Smith Then, your solver can be chosen with the procedural interface via 45281c84c290SBarry Smith $ SNESSetType(snes,"my_solver") 45291c84c290SBarry Smith or at runtime via the option 45301c84c290SBarry Smith $ -snes_type my_solver 45311c84c290SBarry Smith 45321c84c290SBarry Smith Level: advanced 45331c84c290SBarry Smith 45341c84c290SBarry Smith Note: If your function is not being put into a shared library then use SNESRegister() instead 45351c84c290SBarry Smith 45361c84c290SBarry Smith .keywords: SNES, nonlinear, register 45371c84c290SBarry Smith 45381c84c290SBarry Smith .seealso: SNESRegisterAll(), SNESRegisterDestroy() 45393cea93caSBarry Smith 45407f6c08e0SMatthew Knepley Level: advanced 45413cea93caSBarry Smith @*/ 4542bdf89e91SBarry Smith PetscErrorCode SNESRegister(const char sname[],PetscErrorCode (*function)(SNES)) 4543b2002411SLois Curfman McInnes { 4544dfbe8321SBarry Smith PetscErrorCode ierr; 4545b2002411SLois Curfman McInnes 4546b2002411SLois Curfman McInnes PetscFunctionBegin; 4547a240a19fSJed Brown ierr = PetscFunctionListAdd(&SNESList,sname,function);CHKERRQ(ierr); 4548b2002411SLois Curfman McInnes PetscFunctionReturn(0); 4549b2002411SLois Curfman McInnes } 4550da9b6338SBarry Smith 45517087cfbeSBarry Smith PetscErrorCode SNESTestLocalMin(SNES snes) 4552da9b6338SBarry Smith { 4553dfbe8321SBarry Smith PetscErrorCode ierr; 455477431f27SBarry Smith PetscInt N,i,j; 4555da9b6338SBarry Smith Vec u,uh,fh; 4556da9b6338SBarry Smith PetscScalar value; 4557da9b6338SBarry Smith PetscReal norm; 4558da9b6338SBarry Smith 4559da9b6338SBarry Smith PetscFunctionBegin; 4560da9b6338SBarry Smith ierr = SNESGetSolution(snes,&u);CHKERRQ(ierr); 4561da9b6338SBarry Smith ierr = VecDuplicate(u,&uh);CHKERRQ(ierr); 4562da9b6338SBarry Smith ierr = VecDuplicate(u,&fh);CHKERRQ(ierr); 4563da9b6338SBarry Smith 4564da9b6338SBarry Smith /* currently only works for sequential */ 456522d28d08SBarry Smith ierr = PetscPrintf(PETSC_COMM_WORLD,"Testing FormFunction() for local min\n");CHKERRQ(ierr); 4566da9b6338SBarry Smith ierr = VecGetSize(u,&N);CHKERRQ(ierr); 4567da9b6338SBarry Smith for (i=0; i<N; i++) { 4568da9b6338SBarry Smith ierr = VecCopy(u,uh);CHKERRQ(ierr); 456977431f27SBarry Smith ierr = PetscPrintf(PETSC_COMM_WORLD,"i = %D\n",i);CHKERRQ(ierr); 4570da9b6338SBarry Smith for (j=-10; j<11; j++) { 45718b49ba18SBarry Smith value = PetscSign(j)*PetscExpReal(PetscAbs(j)-10.0); 4572da9b6338SBarry Smith ierr = VecSetValue(uh,i,value,ADD_VALUES);CHKERRQ(ierr); 45733ab0aad5SBarry Smith ierr = SNESComputeFunction(snes,uh,fh);CHKERRQ(ierr); 4574da9b6338SBarry Smith ierr = VecNorm(fh,NORM_2,&norm);CHKERRQ(ierr); 457577431f27SBarry Smith ierr = PetscPrintf(PETSC_COMM_WORLD," j norm %D %18.16e\n",j,norm);CHKERRQ(ierr); 4576da9b6338SBarry Smith value = -value; 4577da9b6338SBarry Smith ierr = VecSetValue(uh,i,value,ADD_VALUES);CHKERRQ(ierr); 4578da9b6338SBarry Smith } 4579da9b6338SBarry Smith } 45806bf464f9SBarry Smith ierr = VecDestroy(&uh);CHKERRQ(ierr); 45816bf464f9SBarry Smith ierr = VecDestroy(&fh);CHKERRQ(ierr); 4582da9b6338SBarry Smith PetscFunctionReturn(0); 4583da9b6338SBarry Smith } 458471f87433Sdalcinl 458571f87433Sdalcinl /*@ 4586fa9f3622SBarry Smith SNESKSPSetUseEW - Sets SNES use Eisenstat-Walker method for 458771f87433Sdalcinl computing relative tolerance for linear solvers within an inexact 458871f87433Sdalcinl Newton method. 458971f87433Sdalcinl 45903f9fe445SBarry Smith Logically Collective on SNES 459171f87433Sdalcinl 459271f87433Sdalcinl Input Parameters: 459371f87433Sdalcinl + snes - SNES context 459471f87433Sdalcinl - flag - PETSC_TRUE or PETSC_FALSE 459571f87433Sdalcinl 459664ba62caSBarry Smith Options Database: 459764ba62caSBarry Smith + -snes_ksp_ew - use Eisenstat-Walker method for determining linear system convergence 459864ba62caSBarry Smith . -snes_ksp_ew_version ver - version of Eisenstat-Walker method 459964ba62caSBarry Smith . -snes_ksp_ew_rtol0 <rtol0> - Sets rtol0 460064ba62caSBarry Smith . -snes_ksp_ew_rtolmax <rtolmax> - Sets rtolmax 460164ba62caSBarry Smith . -snes_ksp_ew_gamma <gamma> - Sets gamma 460264ba62caSBarry Smith . -snes_ksp_ew_alpha <alpha> - Sets alpha 460364ba62caSBarry Smith . -snes_ksp_ew_alpha2 <alpha2> - Sets alpha2 460464ba62caSBarry Smith - -snes_ksp_ew_threshold <threshold> - Sets threshold 460564ba62caSBarry Smith 460671f87433Sdalcinl Notes: 460771f87433Sdalcinl Currently, the default is to use a constant relative tolerance for 460871f87433Sdalcinl the inner linear solvers. Alternatively, one can use the 460971f87433Sdalcinl Eisenstat-Walker method, where the relative convergence tolerance 461071f87433Sdalcinl is reset at each Newton iteration according progress of the nonlinear 461171f87433Sdalcinl solver. 461271f87433Sdalcinl 461371f87433Sdalcinl Level: advanced 461471f87433Sdalcinl 461571f87433Sdalcinl Reference: 461671f87433Sdalcinl S. C. Eisenstat and H. F. Walker, "Choosing the forcing terms in an 461771f87433Sdalcinl inexact Newton method", SISC 17 (1), pp.16-32, 1996. 461871f87433Sdalcinl 461971f87433Sdalcinl .keywords: SNES, KSP, Eisenstat, Walker, convergence, test, inexact, Newton 462071f87433Sdalcinl 4621fa9f3622SBarry Smith .seealso: SNESKSPGetUseEW(), SNESKSPGetParametersEW(), SNESKSPSetParametersEW() 462271f87433Sdalcinl @*/ 46237087cfbeSBarry Smith PetscErrorCode SNESKSPSetUseEW(SNES snes,PetscBool flag) 462471f87433Sdalcinl { 462571f87433Sdalcinl PetscFunctionBegin; 46260700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 4627acfcf0e5SJed Brown PetscValidLogicalCollectiveBool(snes,flag,2); 462871f87433Sdalcinl snes->ksp_ewconv = flag; 462971f87433Sdalcinl PetscFunctionReturn(0); 463071f87433Sdalcinl } 463171f87433Sdalcinl 463271f87433Sdalcinl /*@ 4633fa9f3622SBarry Smith SNESKSPGetUseEW - Gets if SNES is using Eisenstat-Walker method 463471f87433Sdalcinl for computing relative tolerance for linear solvers within an 463571f87433Sdalcinl inexact Newton method. 463671f87433Sdalcinl 463771f87433Sdalcinl Not Collective 463871f87433Sdalcinl 463971f87433Sdalcinl Input Parameter: 464071f87433Sdalcinl . snes - SNES context 464171f87433Sdalcinl 464271f87433Sdalcinl Output Parameter: 464371f87433Sdalcinl . flag - PETSC_TRUE or PETSC_FALSE 464471f87433Sdalcinl 464571f87433Sdalcinl Notes: 464671f87433Sdalcinl Currently, the default is to use a constant relative tolerance for 464771f87433Sdalcinl the inner linear solvers. Alternatively, one can use the 464871f87433Sdalcinl Eisenstat-Walker method, where the relative convergence tolerance 464971f87433Sdalcinl is reset at each Newton iteration according progress of the nonlinear 465071f87433Sdalcinl solver. 465171f87433Sdalcinl 465271f87433Sdalcinl Level: advanced 465371f87433Sdalcinl 465471f87433Sdalcinl Reference: 465571f87433Sdalcinl S. C. Eisenstat and H. F. Walker, "Choosing the forcing terms in an 465671f87433Sdalcinl inexact Newton method", SISC 17 (1), pp.16-32, 1996. 465771f87433Sdalcinl 465871f87433Sdalcinl .keywords: SNES, KSP, Eisenstat, Walker, convergence, test, inexact, Newton 465971f87433Sdalcinl 4660fa9f3622SBarry Smith .seealso: SNESKSPSetUseEW(), SNESKSPGetParametersEW(), SNESKSPSetParametersEW() 466171f87433Sdalcinl @*/ 46627087cfbeSBarry Smith PetscErrorCode SNESKSPGetUseEW(SNES snes, PetscBool *flag) 466371f87433Sdalcinl { 466471f87433Sdalcinl PetscFunctionBegin; 46650700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 466671f87433Sdalcinl PetscValidPointer(flag,2); 466771f87433Sdalcinl *flag = snes->ksp_ewconv; 466871f87433Sdalcinl PetscFunctionReturn(0); 466971f87433Sdalcinl } 467071f87433Sdalcinl 467171f87433Sdalcinl /*@ 4672fa9f3622SBarry Smith SNESKSPSetParametersEW - Sets parameters for Eisenstat-Walker 467371f87433Sdalcinl convergence criteria for the linear solvers within an inexact 467471f87433Sdalcinl Newton method. 467571f87433Sdalcinl 46763f9fe445SBarry Smith Logically Collective on SNES 467771f87433Sdalcinl 467871f87433Sdalcinl Input Parameters: 467971f87433Sdalcinl + snes - SNES context 468071f87433Sdalcinl . version - version 1, 2 (default is 2) or 3 468171f87433Sdalcinl . rtol_0 - initial relative tolerance (0 <= rtol_0 < 1) 468271f87433Sdalcinl . rtol_max - maximum relative tolerance (0 <= rtol_max < 1) 468371f87433Sdalcinl . gamma - multiplicative factor for version 2 rtol computation 468471f87433Sdalcinl (0 <= gamma2 <= 1) 468571f87433Sdalcinl . alpha - power for version 2 rtol computation (1 < alpha <= 2) 468671f87433Sdalcinl . alpha2 - power for safeguard 468771f87433Sdalcinl - threshold - threshold for imposing safeguard (0 < threshold < 1) 468871f87433Sdalcinl 468971f87433Sdalcinl Note: 469071f87433Sdalcinl Version 3 was contributed by Luis Chacon, June 2006. 469171f87433Sdalcinl 469271f87433Sdalcinl Use PETSC_DEFAULT to retain the default for any of the parameters. 469371f87433Sdalcinl 469471f87433Sdalcinl Level: advanced 469571f87433Sdalcinl 469671f87433Sdalcinl Reference: 469771f87433Sdalcinl S. C. Eisenstat and H. F. Walker, "Choosing the forcing terms in an 469871f87433Sdalcinl inexact Newton method", Utah State University Math. Stat. Dept. Res. 469971f87433Sdalcinl Report 6/94/75, June, 1994, to appear in SIAM J. Sci. Comput. 470071f87433Sdalcinl 470171f87433Sdalcinl .keywords: SNES, KSP, Eisenstat, Walker, set, parameters 470271f87433Sdalcinl 4703fa9f3622SBarry Smith .seealso: SNESKSPSetUseEW(), SNESKSPGetUseEW(), SNESKSPGetParametersEW() 470471f87433Sdalcinl @*/ 4705f5af7f23SKarl Rupp PetscErrorCode SNESKSPSetParametersEW(SNES snes,PetscInt version,PetscReal rtol_0,PetscReal rtol_max,PetscReal gamma,PetscReal alpha,PetscReal alpha2,PetscReal threshold) 470671f87433Sdalcinl { 4707fa9f3622SBarry Smith SNESKSPEW *kctx; 47085fd66863SKarl Rupp 470971f87433Sdalcinl PetscFunctionBegin; 47100700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 4711fa9f3622SBarry Smith kctx = (SNESKSPEW*)snes->kspconvctx; 4712e32f2f54SBarry Smith if (!kctx) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"No Eisenstat-Walker context existing"); 4713c5eb9154SBarry Smith PetscValidLogicalCollectiveInt(snes,version,2); 4714c5eb9154SBarry Smith PetscValidLogicalCollectiveReal(snes,rtol_0,3); 4715c5eb9154SBarry Smith PetscValidLogicalCollectiveReal(snes,rtol_max,4); 4716c5eb9154SBarry Smith PetscValidLogicalCollectiveReal(snes,gamma,5); 4717c5eb9154SBarry Smith PetscValidLogicalCollectiveReal(snes,alpha,6); 4718c5eb9154SBarry Smith PetscValidLogicalCollectiveReal(snes,alpha2,7); 4719c5eb9154SBarry Smith PetscValidLogicalCollectiveReal(snes,threshold,8); 472071f87433Sdalcinl 472171f87433Sdalcinl if (version != PETSC_DEFAULT) kctx->version = version; 472271f87433Sdalcinl if (rtol_0 != PETSC_DEFAULT) kctx->rtol_0 = rtol_0; 472371f87433Sdalcinl if (rtol_max != PETSC_DEFAULT) kctx->rtol_max = rtol_max; 472471f87433Sdalcinl if (gamma != PETSC_DEFAULT) kctx->gamma = gamma; 472571f87433Sdalcinl if (alpha != PETSC_DEFAULT) kctx->alpha = alpha; 472671f87433Sdalcinl if (alpha2 != PETSC_DEFAULT) kctx->alpha2 = alpha2; 472771f87433Sdalcinl if (threshold != PETSC_DEFAULT) kctx->threshold = threshold; 472871f87433Sdalcinl 4729f23aa3ddSBarry 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); 473057622a8eSBarry 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); 473157622a8eSBarry 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); 473257622a8eSBarry 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); 473357622a8eSBarry 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); 473457622a8eSBarry 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); 473571f87433Sdalcinl PetscFunctionReturn(0); 473671f87433Sdalcinl } 473771f87433Sdalcinl 473871f87433Sdalcinl /*@ 4739fa9f3622SBarry Smith SNESKSPGetParametersEW - Gets parameters for Eisenstat-Walker 474071f87433Sdalcinl convergence criteria for the linear solvers within an inexact 474171f87433Sdalcinl Newton method. 474271f87433Sdalcinl 474371f87433Sdalcinl Not Collective 474471f87433Sdalcinl 474571f87433Sdalcinl Input Parameters: 474671f87433Sdalcinl snes - SNES context 474771f87433Sdalcinl 474871f87433Sdalcinl Output Parameters: 474971f87433Sdalcinl + version - version 1, 2 (default is 2) or 3 475071f87433Sdalcinl . rtol_0 - initial relative tolerance (0 <= rtol_0 < 1) 475171f87433Sdalcinl . rtol_max - maximum relative tolerance (0 <= rtol_max < 1) 4752bf388a1fSBarry Smith . gamma - multiplicative factor for version 2 rtol computation (0 <= gamma2 <= 1) 475371f87433Sdalcinl . alpha - power for version 2 rtol computation (1 < alpha <= 2) 475471f87433Sdalcinl . alpha2 - power for safeguard 475571f87433Sdalcinl - threshold - threshold for imposing safeguard (0 < threshold < 1) 475671f87433Sdalcinl 475771f87433Sdalcinl Level: advanced 475871f87433Sdalcinl 475971f87433Sdalcinl .keywords: SNES, KSP, Eisenstat, Walker, get, parameters 476071f87433Sdalcinl 4761fa9f3622SBarry Smith .seealso: SNESKSPSetUseEW(), SNESKSPGetUseEW(), SNESKSPSetParametersEW() 476271f87433Sdalcinl @*/ 4763bf388a1fSBarry Smith PetscErrorCode SNESKSPGetParametersEW(SNES snes,PetscInt *version,PetscReal *rtol_0,PetscReal *rtol_max,PetscReal *gamma,PetscReal *alpha,PetscReal *alpha2,PetscReal *threshold) 476471f87433Sdalcinl { 4765fa9f3622SBarry Smith SNESKSPEW *kctx; 47665fd66863SKarl Rupp 476771f87433Sdalcinl PetscFunctionBegin; 47680700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 4769fa9f3622SBarry Smith kctx = (SNESKSPEW*)snes->kspconvctx; 4770e32f2f54SBarry Smith if (!kctx) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"No Eisenstat-Walker context existing"); 477171f87433Sdalcinl if (version) *version = kctx->version; 477271f87433Sdalcinl if (rtol_0) *rtol_0 = kctx->rtol_0; 477371f87433Sdalcinl if (rtol_max) *rtol_max = kctx->rtol_max; 477471f87433Sdalcinl if (gamma) *gamma = kctx->gamma; 477571f87433Sdalcinl if (alpha) *alpha = kctx->alpha; 477671f87433Sdalcinl if (alpha2) *alpha2 = kctx->alpha2; 477771f87433Sdalcinl if (threshold) *threshold = kctx->threshold; 477871f87433Sdalcinl PetscFunctionReturn(0); 477971f87433Sdalcinl } 478071f87433Sdalcinl 4781d5378b5fSDmitry Karpeev PetscErrorCode KSPPreSolve_SNESEW(KSP ksp, Vec b, Vec x, SNES snes) 478271f87433Sdalcinl { 478371f87433Sdalcinl PetscErrorCode ierr; 4784fa9f3622SBarry Smith SNESKSPEW *kctx = (SNESKSPEW*)snes->kspconvctx; 478571f87433Sdalcinl PetscReal rtol = PETSC_DEFAULT,stol; 478671f87433Sdalcinl 478771f87433Sdalcinl PetscFunctionBegin; 4788d4211eb9SBarry Smith if (!snes->ksp_ewconv) PetscFunctionReturn(0); 478930058271SDmitry Karpeev if (!snes->iter) { 479030058271SDmitry Karpeev rtol = kctx->rtol_0; /* first time in, so use the original user rtol */ 479130058271SDmitry Karpeev ierr = VecNorm(snes->vec_func,NORM_2,&kctx->norm_first);CHKERRQ(ierr); 479230058271SDmitry Karpeev } 4793f5af7f23SKarl Rupp else { 479471f87433Sdalcinl if (kctx->version == 1) { 479571f87433Sdalcinl rtol = (snes->norm - kctx->lresid_last)/kctx->norm_last; 479671f87433Sdalcinl if (rtol < 0.0) rtol = -rtol; 479785ec1a3cSBarry Smith stol = PetscPowReal(kctx->rtol_last,kctx->alpha2); 479871f87433Sdalcinl if (stol > kctx->threshold) rtol = PetscMax(rtol,stol); 479971f87433Sdalcinl } else if (kctx->version == 2) { 480085ec1a3cSBarry Smith rtol = kctx->gamma * PetscPowReal(snes->norm/kctx->norm_last,kctx->alpha); 480185ec1a3cSBarry Smith stol = kctx->gamma * PetscPowReal(kctx->rtol_last,kctx->alpha); 480271f87433Sdalcinl if (stol > kctx->threshold) rtol = PetscMax(rtol,stol); 480371f87433Sdalcinl } else if (kctx->version == 3) { /* contributed by Luis Chacon, June 2006. */ 480485ec1a3cSBarry Smith rtol = kctx->gamma * PetscPowReal(snes->norm/kctx->norm_last,kctx->alpha); 480571f87433Sdalcinl /* safeguard: avoid sharp decrease of rtol */ 480685ec1a3cSBarry Smith stol = kctx->gamma*PetscPowReal(kctx->rtol_last,kctx->alpha); 480771f87433Sdalcinl stol = PetscMax(rtol,stol); 480871f87433Sdalcinl rtol = PetscMin(kctx->rtol_0,stol); 480971f87433Sdalcinl /* safeguard: avoid oversolving */ 481030058271SDmitry Karpeev stol = kctx->gamma*(kctx->norm_first*snes->rtol)/snes->norm; 481171f87433Sdalcinl stol = PetscMax(rtol,stol); 481271f87433Sdalcinl rtol = PetscMin(kctx->rtol_0,stol); 4813e32f2f54SBarry Smith } else SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Only versions 1, 2 or 3 are supported: %D",kctx->version); 481471f87433Sdalcinl } 481571f87433Sdalcinl /* safeguard: avoid rtol greater than one */ 481671f87433Sdalcinl rtol = PetscMin(rtol,kctx->rtol_max); 481771f87433Sdalcinl ierr = KSPSetTolerances(ksp,rtol,PETSC_DEFAULT,PETSC_DEFAULT,PETSC_DEFAULT);CHKERRQ(ierr); 481857622a8eSBarry Smith ierr = PetscInfo3(snes,"iter %D, Eisenstat-Walker (version %D) KSP rtol=%g\n",snes->iter,kctx->version,(double)rtol);CHKERRQ(ierr); 481971f87433Sdalcinl PetscFunctionReturn(0); 482071f87433Sdalcinl } 482171f87433Sdalcinl 4822d5378b5fSDmitry Karpeev PetscErrorCode KSPPostSolve_SNESEW(KSP ksp, Vec b, Vec x, SNES snes) 482371f87433Sdalcinl { 482471f87433Sdalcinl PetscErrorCode ierr; 4825fa9f3622SBarry Smith SNESKSPEW *kctx = (SNESKSPEW*)snes->kspconvctx; 482671f87433Sdalcinl PCSide pcside; 482771f87433Sdalcinl Vec lres; 482871f87433Sdalcinl 482971f87433Sdalcinl PetscFunctionBegin; 4830d4211eb9SBarry Smith if (!snes->ksp_ewconv) PetscFunctionReturn(0); 483171f87433Sdalcinl ierr = KSPGetTolerances(ksp,&kctx->rtol_last,0,0,0);CHKERRQ(ierr); 483271dbe336SPeter Brune kctx->norm_last = snes->norm; 483371f87433Sdalcinl if (kctx->version == 1) { 48344f00ce20SMatthew G. Knepley PC pc; 48354f00ce20SMatthew G. Knepley PetscBool isNone; 48364f00ce20SMatthew G. Knepley 48374f00ce20SMatthew G. Knepley ierr = KSPGetPC(ksp, &pc);CHKERRQ(ierr); 48384f00ce20SMatthew G. Knepley ierr = PetscObjectTypeCompare((PetscObject) pc, PCNONE, &isNone);CHKERRQ(ierr); 4839b037da10SBarry Smith ierr = KSPGetPCSide(ksp,&pcside);CHKERRQ(ierr); 48404f00ce20SMatthew G. Knepley if (pcside == PC_RIGHT || isNone) { /* XXX Should we also test KSP_UNPRECONDITIONED_NORM ? */ 484171f87433Sdalcinl /* KSP residual is true linear residual */ 484271f87433Sdalcinl ierr = KSPGetResidualNorm(ksp,&kctx->lresid_last);CHKERRQ(ierr); 484371f87433Sdalcinl } else { 484471f87433Sdalcinl /* KSP residual is preconditioned residual */ 484571f87433Sdalcinl /* compute true linear residual norm */ 484671f87433Sdalcinl ierr = VecDuplicate(b,&lres);CHKERRQ(ierr); 484771f87433Sdalcinl ierr = MatMult(snes->jacobian,x,lres);CHKERRQ(ierr); 484871f87433Sdalcinl ierr = VecAYPX(lres,-1.0,b);CHKERRQ(ierr); 484971f87433Sdalcinl ierr = VecNorm(lres,NORM_2,&kctx->lresid_last);CHKERRQ(ierr); 48506bf464f9SBarry Smith ierr = VecDestroy(&lres);CHKERRQ(ierr); 485171f87433Sdalcinl } 485271f87433Sdalcinl } 485371f87433Sdalcinl PetscFunctionReturn(0); 485471f87433Sdalcinl } 485571f87433Sdalcinl 4856d4211eb9SBarry Smith /*@ 4857d4211eb9SBarry Smith SNESGetKSP - Returns the KSP context for a SNES solver. 4858d4211eb9SBarry Smith 4859d4211eb9SBarry Smith Not Collective, but if SNES object is parallel, then KSP object is parallel 4860d4211eb9SBarry Smith 4861d4211eb9SBarry Smith Input Parameter: 4862d4211eb9SBarry Smith . snes - the SNES context 4863d4211eb9SBarry Smith 4864d4211eb9SBarry Smith Output Parameter: 4865d4211eb9SBarry Smith . ksp - the KSP context 4866d4211eb9SBarry Smith 4867d4211eb9SBarry Smith Notes: 4868d4211eb9SBarry Smith The user can then directly manipulate the KSP context to set various 4869d4211eb9SBarry Smith options, etc. Likewise, the user can then extract and manipulate the 4870d4211eb9SBarry Smith PC contexts as well. 4871d4211eb9SBarry Smith 4872d4211eb9SBarry Smith Level: beginner 4873d4211eb9SBarry Smith 4874d4211eb9SBarry Smith .keywords: SNES, nonlinear, get, KSP, context 4875d4211eb9SBarry Smith 4876d4211eb9SBarry Smith .seealso: KSPGetPC(), SNESCreate(), KSPCreate(), SNESSetKSP() 4877d4211eb9SBarry Smith @*/ 4878d4211eb9SBarry Smith PetscErrorCode SNESGetKSP(SNES snes,KSP *ksp) 487971f87433Sdalcinl { 488071f87433Sdalcinl PetscErrorCode ierr; 488171f87433Sdalcinl 488271f87433Sdalcinl PetscFunctionBegin; 4883d4211eb9SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 4884d4211eb9SBarry Smith PetscValidPointer(ksp,2); 4885d4211eb9SBarry Smith 4886d4211eb9SBarry Smith if (!snes->ksp) { 4887a5c2985bSBarry Smith PetscBool monitor = PETSC_FALSE; 4888a5c2985bSBarry Smith 4889d4211eb9SBarry Smith ierr = KSPCreate(PetscObjectComm((PetscObject)snes),&snes->ksp);CHKERRQ(ierr); 4890d4211eb9SBarry Smith ierr = PetscObjectIncrementTabLevel((PetscObject)snes->ksp,(PetscObject)snes,1);CHKERRQ(ierr); 48913bb1ff40SBarry Smith ierr = PetscLogObjectParent((PetscObject)snes,(PetscObject)snes->ksp);CHKERRQ(ierr); 4892d4211eb9SBarry Smith 4893d5378b5fSDmitry Karpeev ierr = KSPSetPreSolve(snes->ksp,(PetscErrorCode (*)(KSP,Vec,Vec,void*))KSPPreSolve_SNESEW,snes);CHKERRQ(ierr); 4894d5378b5fSDmitry Karpeev ierr = KSPSetPostSolve(snes->ksp,(PetscErrorCode (*)(KSP,Vec,Vec,void*))KSPPostSolve_SNESEW,snes);CHKERRQ(ierr); 4895a5c2985bSBarry Smith 4896c5929fdfSBarry Smith ierr = PetscOptionsGetBool(((PetscObject)snes)->options,((PetscObject)snes)->prefix,"-ksp_monitor_snes",&monitor,NULL);CHKERRQ(ierr); 4897a5c2985bSBarry Smith if (monitor) { 4898a5c2985bSBarry Smith ierr = KSPMonitorSet(snes->ksp,KSPMonitorSNES,snes,NULL);CHKERRQ(ierr); 4899a5c2985bSBarry Smith } 4900e5f7ee39SBarry Smith monitor = PETSC_FALSE; 4901c5929fdfSBarry Smith ierr = PetscOptionsGetBool(((PetscObject)snes)->options,((PetscObject)snes)->prefix,"-ksp_monitor_snes_lg",&monitor,NULL);CHKERRQ(ierr); 4902e5f7ee39SBarry Smith if (monitor) { 4903e5f7ee39SBarry Smith PetscObject *objs; 49048b0b5a47SLisandro Dalcin ierr = KSPMonitorSNESLGResidualNormCreate(PetscObjectComm((PetscObject)snes),NULL,NULL,PETSC_DECIDE,PETSC_DECIDE,600,600,&objs);CHKERRQ(ierr); 4905e5f7ee39SBarry Smith objs[0] = (PetscObject) snes; 4906e5f7ee39SBarry Smith ierr = KSPMonitorSet(snes->ksp,(PetscErrorCode (*)(KSP,PetscInt,PetscReal,void*))KSPMonitorSNESLGResidualNorm,objs,(PetscErrorCode (*)(void**))KSPMonitorSNESLGResidualNormDestroy);CHKERRQ(ierr); 4907e5f7ee39SBarry Smith } 4908d4211eb9SBarry Smith } 4909d4211eb9SBarry Smith *ksp = snes->ksp; 491071f87433Sdalcinl PetscFunctionReturn(0); 491171f87433Sdalcinl } 49126c699258SBarry Smith 4913d4211eb9SBarry Smith 4914af0996ceSBarry Smith #include <petsc/private/dmimpl.h> 49156c699258SBarry Smith /*@ 49162a808120SBarry Smith SNESSetDM - Sets the DM that may be used by some nonlinear solvers or their underlying preconditioners 49176c699258SBarry Smith 49183f9fe445SBarry Smith Logically Collective on SNES 49196c699258SBarry Smith 49206c699258SBarry Smith Input Parameters: 49212a808120SBarry Smith + snes - the nonlinear solver context 49222a808120SBarry Smith - dm - the dm, cannot be NULL 49236c699258SBarry Smith 49246c699258SBarry Smith Level: intermediate 49256c699258SBarry Smith 492690deff25SFande Kong .seealso: SNESGetDM(), SNESHasDM(), KSPSetDM(), KSPGetDM() 49276c699258SBarry Smith @*/ 49287087cfbeSBarry Smith PetscErrorCode SNESSetDM(SNES snes,DM dm) 49296c699258SBarry Smith { 49306c699258SBarry Smith PetscErrorCode ierr; 4931345fed2cSBarry Smith KSP ksp; 4932942e3340SBarry Smith DMSNES sdm; 49336c699258SBarry Smith 49346c699258SBarry Smith PetscFunctionBegin; 49350700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 49362a808120SBarry Smith PetscValidHeaderSpecific(dm,DM_CLASSID,2); 49372a808120SBarry Smith ierr = PetscObjectReference((PetscObject)dm);CHKERRQ(ierr); 4938942e3340SBarry Smith if (snes->dm) { /* Move the DMSNES context over to the new DM unless the new DM already has one */ 493951f4b3c7SToby Isaac if (snes->dm->dmsnes && !dm->dmsnes) { 4940942e3340SBarry Smith ierr = DMCopyDMSNES(snes->dm,dm);CHKERRQ(ierr); 4941942e3340SBarry Smith ierr = DMGetDMSNES(snes->dm,&sdm);CHKERRQ(ierr); 4942f5af7f23SKarl Rupp if (sdm->originaldm == snes->dm) sdm->originaldm = dm; /* Grant write privileges to the replacement DM */ 49436cab3a1bSJed Brown } 4944*dc822a44SJed Brown ierr = DMCoarsenHookRemove(snes->dm,DMCoarsenHook_SNESVecSol,DMRestrictHook_SNESVecSol,snes);CHKERRQ(ierr); 49456bf464f9SBarry Smith ierr = DMDestroy(&snes->dm);CHKERRQ(ierr); 49466cab3a1bSJed Brown } 49476c699258SBarry Smith snes->dm = dm; 4948116d1032SJed Brown snes->dmAuto = PETSC_FALSE; 4949f5af7f23SKarl Rupp 4950345fed2cSBarry Smith ierr = SNESGetKSP(snes,&ksp);CHKERRQ(ierr); 4951345fed2cSBarry Smith ierr = KSPSetDM(ksp,dm);CHKERRQ(ierr); 4952f22f69f0SBarry Smith ierr = KSPSetDMActive(ksp,PETSC_FALSE);CHKERRQ(ierr); 4953efd4aadfSBarry Smith if (snes->npc) { 4954efd4aadfSBarry Smith ierr = SNESSetDM(snes->npc, snes->dm);CHKERRQ(ierr); 4955efd4aadfSBarry Smith ierr = SNESSetNPCSide(snes,snes->npcside);CHKERRQ(ierr); 49562c155ee1SBarry Smith } 49576c699258SBarry Smith PetscFunctionReturn(0); 49586c699258SBarry Smith } 49596c699258SBarry Smith 49606c699258SBarry Smith /*@ 49616c699258SBarry Smith SNESGetDM - Gets the DM that may be used by some preconditioners 49626c699258SBarry Smith 49633f9fe445SBarry Smith Not Collective but DM obtained is parallel on SNES 49646c699258SBarry Smith 49656c699258SBarry Smith Input Parameter: 49666c699258SBarry Smith . snes - the preconditioner context 49676c699258SBarry Smith 49686c699258SBarry Smith Output Parameter: 49696c699258SBarry Smith . dm - the dm 49706c699258SBarry Smith 49716c699258SBarry Smith Level: intermediate 49726c699258SBarry Smith 497390deff25SFande Kong .seealso: SNESSetDM(), SNESHasDM(), KSPSetDM(), KSPGetDM() 49746c699258SBarry Smith @*/ 49757087cfbeSBarry Smith PetscErrorCode SNESGetDM(SNES snes,DM *dm) 49766c699258SBarry Smith { 49776cab3a1bSJed Brown PetscErrorCode ierr; 49786cab3a1bSJed Brown 49796c699258SBarry Smith PetscFunctionBegin; 49800700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 49816cab3a1bSJed Brown if (!snes->dm) { 4982ce94432eSBarry Smith ierr = DMShellCreate(PetscObjectComm((PetscObject)snes),&snes->dm);CHKERRQ(ierr); 4983116d1032SJed Brown snes->dmAuto = PETSC_TRUE; 49846cab3a1bSJed Brown } 49856c699258SBarry Smith *dm = snes->dm; 49866c699258SBarry Smith PetscFunctionReturn(0); 49876c699258SBarry Smith } 49880807856dSBarry Smith 498990deff25SFande Kong 499090deff25SFande Kong /*@ 499190deff25SFande Kong SNESHasDM - Whether snes has dm 499290deff25SFande Kong 499390deff25SFande Kong Not collective but all processes must return the same value 499490deff25SFande Kong 499590deff25SFande Kong Input Parameter: 499690deff25SFande Kong . snes - the nonlinear solver object 499790deff25SFande Kong 499890deff25SFande Kong Output Parameter: 499990deff25SFande Kong . hasdm - a flag indicates whether there is dm in snes 500090deff25SFande Kong 500190deff25SFande Kong Level: intermediate 500290deff25SFande Kong 500390deff25SFande Kong .seealso: SNESGetDM(), SNESSetDM(), KSPSetDM(), KSPGetDM() 500490deff25SFande Kong @*/ 500590deff25SFande Kong PetscErrorCode SNESHasDM(SNES snes,PetscBool *hasdm) 500690deff25SFande Kong { 500790deff25SFande Kong PetscFunctionBegin; 500890deff25SFande Kong PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 500990deff25SFande Kong PetscValidPointer(hasdm,2); 501090deff25SFande Kong if (snes->dm) *hasdm = PETSC_TRUE; 501190deff25SFande Kong else *hasdm = PETSC_FALSE; 501290deff25SFande Kong PetscFunctionReturn(0); 501390deff25SFande Kong } 501490deff25SFande Kong 501531823bd8SMatthew G Knepley /*@ 5016be95d8f1SBarry Smith SNESSetNPC - Sets the nonlinear preconditioner to be used. 501731823bd8SMatthew G Knepley 501831823bd8SMatthew G Knepley Collective on SNES 501931823bd8SMatthew G Knepley 502031823bd8SMatthew G Knepley Input Parameters: 502131823bd8SMatthew G Knepley + snes - iterative context obtained from SNESCreate() 502231823bd8SMatthew G Knepley - pc - the preconditioner object 502331823bd8SMatthew G Knepley 502431823bd8SMatthew G Knepley Notes: 5025be95d8f1SBarry Smith Use SNESGetNPC() to retrieve the preconditioner context (for example, 502631823bd8SMatthew G Knepley to configure it using the API). 502731823bd8SMatthew G Knepley 502831823bd8SMatthew G Knepley Level: developer 502931823bd8SMatthew G Knepley 503031823bd8SMatthew G Knepley .keywords: SNES, set, precondition 50313ad1a0b9SPatrick Farrell .seealso: SNESGetNPC(), SNESHasNPC() 503231823bd8SMatthew G Knepley @*/ 5033be95d8f1SBarry Smith PetscErrorCode SNESSetNPC(SNES snes, SNES pc) 503431823bd8SMatthew G Knepley { 503531823bd8SMatthew G Knepley PetscErrorCode ierr; 503631823bd8SMatthew G Knepley 503731823bd8SMatthew G Knepley PetscFunctionBegin; 503831823bd8SMatthew G Knepley PetscValidHeaderSpecific(snes, SNES_CLASSID, 1); 503931823bd8SMatthew G Knepley PetscValidHeaderSpecific(pc, SNES_CLASSID, 2); 504031823bd8SMatthew G Knepley PetscCheckSameComm(snes, 1, pc, 2); 504131823bd8SMatthew G Knepley ierr = PetscObjectReference((PetscObject) pc);CHKERRQ(ierr); 5042efd4aadfSBarry Smith ierr = SNESDestroy(&snes->npc);CHKERRQ(ierr); 5043efd4aadfSBarry Smith snes->npc = pc; 5044efd4aadfSBarry Smith ierr = PetscLogObjectParent((PetscObject)snes, (PetscObject)snes->npc);CHKERRQ(ierr); 504531823bd8SMatthew G Knepley PetscFunctionReturn(0); 504631823bd8SMatthew G Knepley } 504731823bd8SMatthew G Knepley 504831823bd8SMatthew G Knepley /*@ 5049be95d8f1SBarry Smith SNESGetNPC - Creates a nonlinear preconditioning solver (SNES) to be used to precondition the nonlinear solver. 505031823bd8SMatthew G Knepley 505131823bd8SMatthew G Knepley Not Collective 505231823bd8SMatthew G Knepley 505331823bd8SMatthew G Knepley Input Parameter: 505431823bd8SMatthew G Knepley . snes - iterative context obtained from SNESCreate() 505531823bd8SMatthew G Knepley 505631823bd8SMatthew G Knepley Output Parameter: 505731823bd8SMatthew G Knepley . pc - preconditioner context 505831823bd8SMatthew G Knepley 5059be95d8f1SBarry Smith Notes: If a SNES was previously set with SNESSetNPC() then that SNES is returned. 5060be95d8f1SBarry Smith 506131823bd8SMatthew G Knepley Level: developer 506231823bd8SMatthew G Knepley 506331823bd8SMatthew G Knepley .keywords: SNES, get, preconditioner 50643ad1a0b9SPatrick Farrell .seealso: SNESSetNPC(), SNESHasNPC() 506531823bd8SMatthew G Knepley @*/ 5066be95d8f1SBarry Smith PetscErrorCode SNESGetNPC(SNES snes, SNES *pc) 506731823bd8SMatthew G Knepley { 506831823bd8SMatthew G Knepley PetscErrorCode ierr; 5069a64e098fSPeter Brune const char *optionsprefix; 507031823bd8SMatthew G Knepley 507131823bd8SMatthew G Knepley PetscFunctionBegin; 507231823bd8SMatthew G Knepley PetscValidHeaderSpecific(snes, SNES_CLASSID, 1); 507331823bd8SMatthew G Knepley PetscValidPointer(pc, 2); 5074efd4aadfSBarry Smith if (!snes->npc) { 5075efd4aadfSBarry Smith ierr = SNESCreate(PetscObjectComm((PetscObject)snes),&snes->npc);CHKERRQ(ierr); 5076efd4aadfSBarry Smith ierr = PetscObjectIncrementTabLevel((PetscObject)snes->npc,(PetscObject)snes,1);CHKERRQ(ierr); 5077efd4aadfSBarry Smith ierr = PetscLogObjectParent((PetscObject)snes,(PetscObject)snes->npc);CHKERRQ(ierr); 5078a64e098fSPeter Brune ierr = SNESGetOptionsPrefix(snes,&optionsprefix);CHKERRQ(ierr); 5079efd4aadfSBarry Smith ierr = SNESSetOptionsPrefix(snes->npc,optionsprefix);CHKERRQ(ierr); 5080efd4aadfSBarry Smith ierr = SNESAppendOptionsPrefix(snes->npc,"npc_");CHKERRQ(ierr); 5081efd4aadfSBarry Smith ierr = SNESSetCountersReset(snes->npc,PETSC_FALSE);CHKERRQ(ierr); 508231823bd8SMatthew G Knepley } 5083efd4aadfSBarry Smith *pc = snes->npc; 508431823bd8SMatthew G Knepley PetscFunctionReturn(0); 508531823bd8SMatthew G Knepley } 508631823bd8SMatthew G Knepley 50873ad1a0b9SPatrick Farrell /*@ 50883ad1a0b9SPatrick Farrell SNESHasNPC - Returns whether a nonlinear preconditioner exists 50893ad1a0b9SPatrick Farrell 50903ad1a0b9SPatrick Farrell Not Collective 50913ad1a0b9SPatrick Farrell 50923ad1a0b9SPatrick Farrell Input Parameter: 50933ad1a0b9SPatrick Farrell . snes - iterative context obtained from SNESCreate() 50943ad1a0b9SPatrick Farrell 50953ad1a0b9SPatrick Farrell Output Parameter: 50963ad1a0b9SPatrick Farrell . has_npc - whether the SNES has an NPC or not 50973ad1a0b9SPatrick Farrell 50983ad1a0b9SPatrick Farrell Level: developer 50993ad1a0b9SPatrick Farrell 51003ad1a0b9SPatrick Farrell .keywords: SNES, has, preconditioner 51013ad1a0b9SPatrick Farrell .seealso: SNESSetNPC(), SNESGetNPC() 51023ad1a0b9SPatrick Farrell @*/ 51033ad1a0b9SPatrick Farrell PetscErrorCode SNESHasNPC(SNES snes, PetscBool *has_npc) 51043ad1a0b9SPatrick Farrell { 51053ad1a0b9SPatrick Farrell PetscFunctionBegin; 51063ad1a0b9SPatrick Farrell PetscValidHeaderSpecific(snes, SNES_CLASSID, 1); 5107efd4aadfSBarry Smith *has_npc = (PetscBool) (snes->npc ? PETSC_TRUE : PETSC_FALSE); 51083ad1a0b9SPatrick Farrell PetscFunctionReturn(0); 51093ad1a0b9SPatrick Farrell } 51103ad1a0b9SPatrick Farrell 5111c40d0f55SPeter Brune /*@ 5112be95d8f1SBarry Smith SNESSetNPCSide - Sets the preconditioning side. 5113c40d0f55SPeter Brune 5114c40d0f55SPeter Brune Logically Collective on SNES 5115c40d0f55SPeter Brune 5116c40d0f55SPeter Brune Input Parameter: 5117c40d0f55SPeter Brune . snes - iterative context obtained from SNESCreate() 5118c40d0f55SPeter Brune 5119c40d0f55SPeter Brune Output Parameter: 5120c40d0f55SPeter Brune . side - the preconditioning side, where side is one of 5121c40d0f55SPeter Brune .vb 51222d547940SBarry Smith PC_LEFT - left preconditioning 51232d547940SBarry Smith PC_RIGHT - right preconditioning (default for most nonlinear solvers) 5124c40d0f55SPeter Brune .ve 5125c40d0f55SPeter Brune 5126c40d0f55SPeter Brune Options Database Keys: 5127c40d0f55SPeter Brune . -snes_pc_side <right,left> 5128c40d0f55SPeter Brune 51292d547940SBarry Smith Notes: SNESNRICHARDSON and SNESNCG only support left preconditioning. 51302d547940SBarry Smith 5131c40d0f55SPeter Brune Level: intermediate 5132c40d0f55SPeter Brune 5133c40d0f55SPeter Brune .keywords: SNES, set, right, left, side, preconditioner, flag 5134c40d0f55SPeter Brune 5135be95d8f1SBarry Smith .seealso: SNESGetNPCSide(), KSPSetPCSide() 5136c40d0f55SPeter Brune @*/ 5137be95d8f1SBarry Smith PetscErrorCode SNESSetNPCSide(SNES snes,PCSide side) 5138c40d0f55SPeter Brune { 5139c40d0f55SPeter Brune PetscFunctionBegin; 5140c40d0f55SPeter Brune PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 5141c40d0f55SPeter Brune PetscValidLogicalCollectiveEnum(snes,side,2); 5142efd4aadfSBarry Smith snes->npcside= side; 5143c40d0f55SPeter Brune PetscFunctionReturn(0); 5144c40d0f55SPeter Brune } 5145c40d0f55SPeter Brune 5146c40d0f55SPeter Brune /*@ 5147be95d8f1SBarry Smith SNESGetNPCSide - Gets the preconditioning side. 5148c40d0f55SPeter Brune 5149c40d0f55SPeter Brune Not Collective 5150c40d0f55SPeter Brune 5151c40d0f55SPeter Brune Input Parameter: 5152c40d0f55SPeter Brune . snes - iterative context obtained from SNESCreate() 5153c40d0f55SPeter Brune 5154c40d0f55SPeter Brune Output Parameter: 5155c40d0f55SPeter Brune . side - the preconditioning side, where side is one of 5156c40d0f55SPeter Brune .vb 51572d547940SBarry Smith PC_LEFT - left preconditioning 51582d547940SBarry Smith PC_RIGHT - right preconditioning (default for most nonlinear solvers) 5159c40d0f55SPeter Brune .ve 5160c40d0f55SPeter Brune 5161c40d0f55SPeter Brune Level: intermediate 5162c40d0f55SPeter Brune 5163c40d0f55SPeter Brune .keywords: SNES, get, right, left, side, preconditioner, flag 5164c40d0f55SPeter Brune 5165be95d8f1SBarry Smith .seealso: SNESSetNPCSide(), KSPGetPCSide() 5166c40d0f55SPeter Brune @*/ 5167be95d8f1SBarry Smith PetscErrorCode SNESGetNPCSide(SNES snes,PCSide *side) 5168c40d0f55SPeter Brune { 5169c40d0f55SPeter Brune PetscFunctionBegin; 5170c40d0f55SPeter Brune PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 5171c40d0f55SPeter Brune PetscValidPointer(side,2); 5172efd4aadfSBarry Smith *side = snes->npcside; 5173c40d0f55SPeter Brune PetscFunctionReturn(0); 5174c40d0f55SPeter Brune } 5175c40d0f55SPeter Brune 51769e764e56SPeter Brune /*@ 51777601faf0SJed Brown SNESSetLineSearch - Sets the linesearch on the SNES instance. 51789e764e56SPeter Brune 51799e764e56SPeter Brune Collective on SNES 51809e764e56SPeter Brune 51819e764e56SPeter Brune Input Parameters: 51829e764e56SPeter Brune + snes - iterative context obtained from SNESCreate() 51839e764e56SPeter Brune - linesearch - the linesearch object 51849e764e56SPeter Brune 51859e764e56SPeter Brune Notes: 51867601faf0SJed Brown Use SNESGetLineSearch() to retrieve the preconditioner context (for example, 51879e764e56SPeter Brune to configure it using the API). 51889e764e56SPeter Brune 51899e764e56SPeter Brune Level: developer 51909e764e56SPeter Brune 51919e764e56SPeter Brune .keywords: SNES, set, linesearch 51927601faf0SJed Brown .seealso: SNESGetLineSearch() 51939e764e56SPeter Brune @*/ 51947601faf0SJed Brown PetscErrorCode SNESSetLineSearch(SNES snes, SNESLineSearch linesearch) 51959e764e56SPeter Brune { 51969e764e56SPeter Brune PetscErrorCode ierr; 51979e764e56SPeter Brune 51989e764e56SPeter Brune PetscFunctionBegin; 51999e764e56SPeter Brune PetscValidHeaderSpecific(snes, SNES_CLASSID, 1); 5200f1c6b773SPeter Brune PetscValidHeaderSpecific(linesearch, SNESLINESEARCH_CLASSID, 2); 52019e764e56SPeter Brune PetscCheckSameComm(snes, 1, linesearch, 2); 52029e764e56SPeter Brune ierr = PetscObjectReference((PetscObject) linesearch);CHKERRQ(ierr); 5203f1c6b773SPeter Brune ierr = SNESLineSearchDestroy(&snes->linesearch);CHKERRQ(ierr); 5204f5af7f23SKarl Rupp 52059e764e56SPeter Brune snes->linesearch = linesearch; 5206f5af7f23SKarl Rupp 52073bb1ff40SBarry Smith ierr = PetscLogObjectParent((PetscObject)snes, (PetscObject)snes->linesearch);CHKERRQ(ierr); 52089e764e56SPeter Brune PetscFunctionReturn(0); 52099e764e56SPeter Brune } 52109e764e56SPeter Brune 5211a34ceb2aSJed Brown /*@ 52127601faf0SJed Brown SNESGetLineSearch - Returns a pointer to the line search context set with SNESSetLineSearch() 52138141a3b9SPeter Brune or creates a default line search instance associated with the SNES and returns it. 52149e764e56SPeter Brune 52159e764e56SPeter Brune Not Collective 52169e764e56SPeter Brune 52179e764e56SPeter Brune Input Parameter: 52189e764e56SPeter Brune . snes - iterative context obtained from SNESCreate() 52199e764e56SPeter Brune 52209e764e56SPeter Brune Output Parameter: 52219e764e56SPeter Brune . linesearch - linesearch context 52229e764e56SPeter Brune 5223162e0bf5SPeter Brune Level: beginner 52249e764e56SPeter Brune 52259e764e56SPeter Brune .keywords: SNES, get, linesearch 5226162e0bf5SPeter Brune .seealso: SNESSetLineSearch(), SNESLineSearchCreate() 52279e764e56SPeter Brune @*/ 52287601faf0SJed Brown PetscErrorCode SNESGetLineSearch(SNES snes, SNESLineSearch *linesearch) 52299e764e56SPeter Brune { 52309e764e56SPeter Brune PetscErrorCode ierr; 52319e764e56SPeter Brune const char *optionsprefix; 52329e764e56SPeter Brune 52339e764e56SPeter Brune PetscFunctionBegin; 52349e764e56SPeter Brune PetscValidHeaderSpecific(snes, SNES_CLASSID, 1); 52359e764e56SPeter Brune PetscValidPointer(linesearch, 2); 52369e764e56SPeter Brune if (!snes->linesearch) { 52379e764e56SPeter Brune ierr = SNESGetOptionsPrefix(snes, &optionsprefix);CHKERRQ(ierr); 523882f516ccSBarry Smith ierr = SNESLineSearchCreate(PetscObjectComm((PetscObject)snes), &snes->linesearch);CHKERRQ(ierr); 5239f1c6b773SPeter Brune ierr = SNESLineSearchSetSNES(snes->linesearch, snes);CHKERRQ(ierr); 5240b1dca40bSPeter Brune ierr = SNESLineSearchAppendOptionsPrefix(snes->linesearch, optionsprefix);CHKERRQ(ierr); 52419e764e56SPeter Brune ierr = PetscObjectIncrementTabLevel((PetscObject) snes->linesearch, (PetscObject) snes, 1);CHKERRQ(ierr); 52423bb1ff40SBarry Smith ierr = PetscLogObjectParent((PetscObject)snes, (PetscObject)snes->linesearch);CHKERRQ(ierr); 52439e764e56SPeter Brune } 52449e764e56SPeter Brune *linesearch = snes->linesearch; 52459e764e56SPeter Brune PetscFunctionReturn(0); 52469e764e56SPeter Brune } 52479e764e56SPeter Brune 524869b4f73cSBarry Smith #if defined(PETSC_HAVE_MATLAB_ENGINE) 5249c6db04a5SJed Brown #include <mex.h> 525069b4f73cSBarry Smith 52518f6e6473SBarry Smith typedef struct {char *funcname; mxArray *ctx;} SNESMatlabContext; 52528f6e6473SBarry Smith 52530807856dSBarry Smith /* 5254bf388a1fSBarry Smith SNESComputeFunction_Matlab - Calls the function that has been set with SNESSetFunctionMatlab(). 52550807856dSBarry Smith 52560807856dSBarry Smith Collective on SNES 52570807856dSBarry Smith 52580807856dSBarry Smith Input Parameters: 52590807856dSBarry Smith + snes - the SNES context 52600807856dSBarry Smith - x - input vector 52610807856dSBarry Smith 52620807856dSBarry Smith Output Parameter: 52630807856dSBarry Smith . y - function vector, as set by SNESSetFunction() 52640807856dSBarry Smith 52650807856dSBarry Smith Notes: 52660807856dSBarry Smith SNESComputeFunction() is typically used within nonlinear solvers 52670807856dSBarry Smith implementations, so most users would not generally call this routine 52680807856dSBarry Smith themselves. 52690807856dSBarry Smith 52700807856dSBarry Smith Level: developer 52710807856dSBarry Smith 52720807856dSBarry Smith .keywords: SNES, nonlinear, compute, function 52730807856dSBarry Smith 52740807856dSBarry Smith .seealso: SNESSetFunction(), SNESGetFunction() 527561b2408cSBarry Smith */ 52767087cfbeSBarry Smith PetscErrorCode SNESComputeFunction_Matlab(SNES snes,Vec x,Vec y, void *ctx) 52770807856dSBarry Smith { 5278e650e774SBarry Smith PetscErrorCode ierr; 52798f6e6473SBarry Smith SNESMatlabContext *sctx = (SNESMatlabContext*)ctx; 52808f6e6473SBarry Smith int nlhs = 1,nrhs = 5; 52818f6e6473SBarry Smith mxArray *plhs[1],*prhs[5]; 528291621f2eSBarry Smith long long int lx = 0,ly = 0,ls = 0; 5283e650e774SBarry Smith 52840807856dSBarry Smith PetscFunctionBegin; 52850807856dSBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 52860807856dSBarry Smith PetscValidHeaderSpecific(x,VEC_CLASSID,2); 52870807856dSBarry Smith PetscValidHeaderSpecific(y,VEC_CLASSID,3); 52880807856dSBarry Smith PetscCheckSameComm(snes,1,x,2); 52890807856dSBarry Smith PetscCheckSameComm(snes,1,y,3); 52900807856dSBarry Smith 52910807856dSBarry Smith /* call Matlab function in ctx with arguments x and y */ 5292e650e774SBarry Smith 529391621f2eSBarry Smith ierr = PetscMemcpy(&ls,&snes,sizeof(snes));CHKERRQ(ierr); 5294e650e774SBarry Smith ierr = PetscMemcpy(&lx,&x,sizeof(x));CHKERRQ(ierr); 5295e650e774SBarry Smith ierr = PetscMemcpy(&ly,&y,sizeof(x));CHKERRQ(ierr); 529691621f2eSBarry Smith prhs[0] = mxCreateDoubleScalar((double)ls); 529791621f2eSBarry Smith prhs[1] = mxCreateDoubleScalar((double)lx); 529891621f2eSBarry Smith prhs[2] = mxCreateDoubleScalar((double)ly); 52998f6e6473SBarry Smith prhs[3] = mxCreateString(sctx->funcname); 53008f6e6473SBarry Smith prhs[4] = sctx->ctx; 5301b807a863SBarry Smith ierr = mexCallMATLAB(nlhs,plhs,nrhs,prhs,"PetscSNESComputeFunctionInternal");CHKERRQ(ierr); 5302e650e774SBarry Smith ierr = mxGetScalar(plhs[0]);CHKERRQ(ierr); 5303e650e774SBarry Smith mxDestroyArray(prhs[0]); 5304e650e774SBarry Smith mxDestroyArray(prhs[1]); 5305e650e774SBarry Smith mxDestroyArray(prhs[2]); 53068f6e6473SBarry Smith mxDestroyArray(prhs[3]); 5307e650e774SBarry Smith mxDestroyArray(plhs[0]); 53080807856dSBarry Smith PetscFunctionReturn(0); 53090807856dSBarry Smith } 53100807856dSBarry Smith 531161b2408cSBarry Smith /* 53120807856dSBarry Smith SNESSetFunctionMatlab - Sets the function evaluation routine and function 53130807856dSBarry Smith vector for use by the SNES routines in solving systems of nonlinear 5314e3c5b3baSBarry Smith equations from MATLAB. Here the function is a string containing the name of a MATLAB function 53150807856dSBarry Smith 53160807856dSBarry Smith Logically Collective on SNES 53170807856dSBarry Smith 53180807856dSBarry Smith Input Parameters: 53190807856dSBarry Smith + snes - the SNES context 53200807856dSBarry Smith . r - vector to store function value 5321f8b49ee9SBarry Smith - f - function evaluation routine 53220807856dSBarry Smith 53230807856dSBarry Smith Notes: 53240807856dSBarry Smith The Newton-like methods typically solve linear systems of the form 53250807856dSBarry Smith $ f'(x) x = -f(x), 53260807856dSBarry Smith where f'(x) denotes the Jacobian matrix and f(x) is the function. 53270807856dSBarry Smith 53280807856dSBarry Smith Level: beginner 53290807856dSBarry Smith 5330c5b75c40SBarry Smith Developer Note: This bleeds the allocated memory SNESMatlabContext *sctx; 5331c5b75c40SBarry Smith 53320807856dSBarry Smith .keywords: SNES, nonlinear, set, function 53330807856dSBarry Smith 53340807856dSBarry Smith .seealso: SNESGetFunction(), SNESComputeFunction(), SNESSetJacobian(), SNESSetFunction() 533561b2408cSBarry Smith */ 5336f8b49ee9SBarry Smith PetscErrorCode SNESSetFunctionMatlab(SNES snes,Vec r,const char *f,mxArray *ctx) 53370807856dSBarry Smith { 53380807856dSBarry Smith PetscErrorCode ierr; 53398f6e6473SBarry Smith SNESMatlabContext *sctx; 53400807856dSBarry Smith 53410807856dSBarry Smith PetscFunctionBegin; 53428f6e6473SBarry Smith /* currently sctx is memory bleed */ 5343854ce69bSBarry Smith ierr = PetscNew(&sctx);CHKERRQ(ierr); 5344f8b49ee9SBarry Smith ierr = PetscStrallocpy(f,&sctx->funcname);CHKERRQ(ierr); 53458f6e6473SBarry Smith /* 53468f6e6473SBarry Smith This should work, but it doesn't 53478f6e6473SBarry Smith sctx->ctx = ctx; 53488f6e6473SBarry Smith mexMakeArrayPersistent(sctx->ctx); 53498f6e6473SBarry Smith */ 53508f6e6473SBarry Smith sctx->ctx = mxDuplicateArray(ctx); 53518f6e6473SBarry Smith ierr = SNESSetFunction(snes,r,SNESComputeFunction_Matlab,sctx);CHKERRQ(ierr); 53520807856dSBarry Smith PetscFunctionReturn(0); 53530807856dSBarry Smith } 535469b4f73cSBarry Smith 535561b2408cSBarry Smith /* 5356bf388a1fSBarry Smith SNESComputeJacobian_Matlab - Calls the function that has been set with SNESSetJacobianMatlab(). 535761b2408cSBarry Smith 535861b2408cSBarry Smith Collective on SNES 535961b2408cSBarry Smith 536061b2408cSBarry Smith Input Parameters: 536161b2408cSBarry Smith + snes - the SNES context 536261b2408cSBarry Smith . x - input vector 536361b2408cSBarry Smith . A, B - the matrices 536461b2408cSBarry Smith - ctx - user context 536561b2408cSBarry Smith 536661b2408cSBarry Smith Level: developer 536761b2408cSBarry Smith 536861b2408cSBarry Smith .keywords: SNES, nonlinear, compute, function 536961b2408cSBarry Smith 537061b2408cSBarry Smith .seealso: SNESSetFunction(), SNESGetFunction() 537161b2408cSBarry Smith @*/ 5372f3229a78SSatish Balay PetscErrorCode SNESComputeJacobian_Matlab(SNES snes,Vec x,Mat A,Mat B,void *ctx) 537361b2408cSBarry Smith { 537461b2408cSBarry Smith PetscErrorCode ierr; 537561b2408cSBarry Smith SNESMatlabContext *sctx = (SNESMatlabContext*)ctx; 537661b2408cSBarry Smith int nlhs = 2,nrhs = 6; 537761b2408cSBarry Smith mxArray *plhs[2],*prhs[6]; 537861b2408cSBarry Smith long long int lx = 0,lA = 0,ls = 0, lB = 0; 537961b2408cSBarry Smith 538061b2408cSBarry Smith PetscFunctionBegin; 538161b2408cSBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 538261b2408cSBarry Smith PetscValidHeaderSpecific(x,VEC_CLASSID,2); 538361b2408cSBarry Smith 538461b2408cSBarry Smith /* call Matlab function in ctx with arguments x and y */ 538561b2408cSBarry Smith 538661b2408cSBarry Smith ierr = PetscMemcpy(&ls,&snes,sizeof(snes));CHKERRQ(ierr); 538761b2408cSBarry Smith ierr = PetscMemcpy(&lx,&x,sizeof(x));CHKERRQ(ierr); 538861b2408cSBarry Smith ierr = PetscMemcpy(&lA,A,sizeof(x));CHKERRQ(ierr); 538961b2408cSBarry Smith ierr = PetscMemcpy(&lB,B,sizeof(x));CHKERRQ(ierr); 539061b2408cSBarry Smith prhs[0] = mxCreateDoubleScalar((double)ls); 539161b2408cSBarry Smith prhs[1] = mxCreateDoubleScalar((double)lx); 539261b2408cSBarry Smith prhs[2] = mxCreateDoubleScalar((double)lA); 539361b2408cSBarry Smith prhs[3] = mxCreateDoubleScalar((double)lB); 539461b2408cSBarry Smith prhs[4] = mxCreateString(sctx->funcname); 539561b2408cSBarry Smith prhs[5] = sctx->ctx; 5396b807a863SBarry Smith ierr = mexCallMATLAB(nlhs,plhs,nrhs,prhs,"PetscSNESComputeJacobianInternal");CHKERRQ(ierr); 539761b2408cSBarry Smith ierr = mxGetScalar(plhs[0]);CHKERRQ(ierr); 539861b2408cSBarry Smith mxDestroyArray(prhs[0]); 539961b2408cSBarry Smith mxDestroyArray(prhs[1]); 540061b2408cSBarry Smith mxDestroyArray(prhs[2]); 540161b2408cSBarry Smith mxDestroyArray(prhs[3]); 540261b2408cSBarry Smith mxDestroyArray(prhs[4]); 540361b2408cSBarry Smith mxDestroyArray(plhs[0]); 540461b2408cSBarry Smith mxDestroyArray(plhs[1]); 540561b2408cSBarry Smith PetscFunctionReturn(0); 540661b2408cSBarry Smith } 540761b2408cSBarry Smith 540861b2408cSBarry Smith /* 540961b2408cSBarry Smith SNESSetJacobianMatlab - Sets the Jacobian function evaluation routine and two empty Jacobian matrices 541061b2408cSBarry Smith vector for use by the SNES routines in solving systems of nonlinear 5411e3c5b3baSBarry Smith equations from MATLAB. Here the function is a string containing the name of a MATLAB function 541261b2408cSBarry Smith 541361b2408cSBarry Smith Logically Collective on SNES 541461b2408cSBarry Smith 541561b2408cSBarry Smith Input Parameters: 541661b2408cSBarry Smith + snes - the SNES context 541761b2408cSBarry Smith . A,B - Jacobian matrices 5418f8b49ee9SBarry Smith . J - function evaluation routine 541961b2408cSBarry Smith - ctx - user context 542061b2408cSBarry Smith 542161b2408cSBarry Smith Level: developer 542261b2408cSBarry Smith 5423c5b75c40SBarry Smith Developer Note: This bleeds the allocated memory SNESMatlabContext *sctx; 5424c5b75c40SBarry Smith 542561b2408cSBarry Smith .keywords: SNES, nonlinear, set, function 542661b2408cSBarry Smith 5427f8b49ee9SBarry Smith .seealso: SNESGetFunction(), SNESComputeFunction(), SNESSetJacobian(), SNESSetFunction(), J 542861b2408cSBarry Smith */ 5429f8b49ee9SBarry Smith PetscErrorCode SNESSetJacobianMatlab(SNES snes,Mat A,Mat B,const char *J,mxArray *ctx) 543061b2408cSBarry Smith { 543161b2408cSBarry Smith PetscErrorCode ierr; 543261b2408cSBarry Smith SNESMatlabContext *sctx; 543361b2408cSBarry Smith 543461b2408cSBarry Smith PetscFunctionBegin; 543561b2408cSBarry Smith /* currently sctx is memory bleed */ 5436854ce69bSBarry Smith ierr = PetscNew(&sctx);CHKERRQ(ierr); 5437f8b49ee9SBarry Smith ierr = PetscStrallocpy(J,&sctx->funcname);CHKERRQ(ierr); 543861b2408cSBarry Smith /* 543961b2408cSBarry Smith This should work, but it doesn't 544061b2408cSBarry Smith sctx->ctx = ctx; 544161b2408cSBarry Smith mexMakeArrayPersistent(sctx->ctx); 544261b2408cSBarry Smith */ 544361b2408cSBarry Smith sctx->ctx = mxDuplicateArray(ctx); 544461b2408cSBarry Smith ierr = SNESSetJacobian(snes,A,B,SNESComputeJacobian_Matlab,sctx);CHKERRQ(ierr); 544561b2408cSBarry Smith PetscFunctionReturn(0); 544661b2408cSBarry Smith } 544769b4f73cSBarry Smith 5448f9eb7ae2SShri Abhyankar /* 5449f9eb7ae2SShri Abhyankar SNESMonitor_Matlab - Calls the function that has been set with SNESMonitorSetMatlab(). 5450f9eb7ae2SShri Abhyankar 5451f9eb7ae2SShri Abhyankar Collective on SNES 5452f9eb7ae2SShri Abhyankar 5453f9eb7ae2SShri Abhyankar .seealso: SNESSetFunction(), SNESGetFunction() 5454f9eb7ae2SShri Abhyankar @*/ 54557087cfbeSBarry Smith PetscErrorCode SNESMonitor_Matlab(SNES snes,PetscInt it, PetscReal fnorm, void *ctx) 5456f9eb7ae2SShri Abhyankar { 5457f9eb7ae2SShri Abhyankar PetscErrorCode ierr; 545848f37e70SShri Abhyankar SNESMatlabContext *sctx = (SNESMatlabContext*)ctx; 5459f9eb7ae2SShri Abhyankar int nlhs = 1,nrhs = 6; 5460f9eb7ae2SShri Abhyankar mxArray *plhs[1],*prhs[6]; 5461f9eb7ae2SShri Abhyankar long long int lx = 0,ls = 0; 5462f9eb7ae2SShri Abhyankar Vec x = snes->vec_sol; 5463f9eb7ae2SShri Abhyankar 5464f9eb7ae2SShri Abhyankar PetscFunctionBegin; 5465f9eb7ae2SShri Abhyankar PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 5466f9eb7ae2SShri Abhyankar 5467f9eb7ae2SShri Abhyankar ierr = PetscMemcpy(&ls,&snes,sizeof(snes));CHKERRQ(ierr); 5468f9eb7ae2SShri Abhyankar ierr = PetscMemcpy(&lx,&x,sizeof(x));CHKERRQ(ierr); 5469f9eb7ae2SShri Abhyankar prhs[0] = mxCreateDoubleScalar((double)ls); 5470f9eb7ae2SShri Abhyankar prhs[1] = mxCreateDoubleScalar((double)it); 5471f9eb7ae2SShri Abhyankar prhs[2] = mxCreateDoubleScalar((double)fnorm); 5472f9eb7ae2SShri Abhyankar prhs[3] = mxCreateDoubleScalar((double)lx); 5473f9eb7ae2SShri Abhyankar prhs[4] = mxCreateString(sctx->funcname); 5474f9eb7ae2SShri Abhyankar prhs[5] = sctx->ctx; 5475f9eb7ae2SShri Abhyankar ierr = mexCallMATLAB(nlhs,plhs,nrhs,prhs,"PetscSNESMonitorInternal");CHKERRQ(ierr); 5476f9eb7ae2SShri Abhyankar ierr = mxGetScalar(plhs[0]);CHKERRQ(ierr); 5477f9eb7ae2SShri Abhyankar mxDestroyArray(prhs[0]); 5478f9eb7ae2SShri Abhyankar mxDestroyArray(prhs[1]); 5479f9eb7ae2SShri Abhyankar mxDestroyArray(prhs[2]); 5480f9eb7ae2SShri Abhyankar mxDestroyArray(prhs[3]); 5481f9eb7ae2SShri Abhyankar mxDestroyArray(prhs[4]); 5482f9eb7ae2SShri Abhyankar mxDestroyArray(plhs[0]); 5483f9eb7ae2SShri Abhyankar PetscFunctionReturn(0); 5484f9eb7ae2SShri Abhyankar } 5485f9eb7ae2SShri Abhyankar 5486f9eb7ae2SShri Abhyankar /* 5487e3c5b3baSBarry Smith SNESMonitorSetMatlab - Sets the monitor function from MATLAB 5488f9eb7ae2SShri Abhyankar 5489f9eb7ae2SShri Abhyankar Level: developer 5490f9eb7ae2SShri Abhyankar 5491c5b75c40SBarry Smith Developer Note: This bleeds the allocated memory SNESMatlabContext *sctx; 5492c5b75c40SBarry Smith 5493f9eb7ae2SShri Abhyankar .keywords: SNES, nonlinear, set, function 5494f9eb7ae2SShri Abhyankar 5495f9eb7ae2SShri Abhyankar .seealso: SNESGetFunction(), SNESComputeFunction(), SNESSetJacobian(), SNESSetFunction() 5496f9eb7ae2SShri Abhyankar */ 54976e4dcb14SBarry Smith PetscErrorCode SNESMonitorSetMatlab(SNES snes,const char *f,mxArray *ctx) 5498f9eb7ae2SShri Abhyankar { 5499f9eb7ae2SShri Abhyankar PetscErrorCode ierr; 5500f9eb7ae2SShri Abhyankar SNESMatlabContext *sctx; 5501f9eb7ae2SShri Abhyankar 5502f9eb7ae2SShri Abhyankar PetscFunctionBegin; 5503f9eb7ae2SShri Abhyankar /* currently sctx is memory bleed */ 5504854ce69bSBarry Smith ierr = PetscNew(&sctx);CHKERRQ(ierr); 55056e4dcb14SBarry Smith ierr = PetscStrallocpy(f,&sctx->funcname);CHKERRQ(ierr); 5506f9eb7ae2SShri Abhyankar /* 5507f9eb7ae2SShri Abhyankar This should work, but it doesn't 5508f9eb7ae2SShri Abhyankar sctx->ctx = ctx; 5509f9eb7ae2SShri Abhyankar mexMakeArrayPersistent(sctx->ctx); 5510f9eb7ae2SShri Abhyankar */ 5511f9eb7ae2SShri Abhyankar sctx->ctx = mxDuplicateArray(ctx); 55120298fd71SBarry Smith ierr = SNESMonitorSet(snes,SNESMonitor_Matlab,sctx,NULL);CHKERRQ(ierr); 5513f9eb7ae2SShri Abhyankar PetscFunctionReturn(0); 5514f9eb7ae2SShri Abhyankar } 5515f9eb7ae2SShri Abhyankar 551669b4f73cSBarry Smith #endif 5517