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> 6*34b4d3a8SMatthew G. Knepley #include <petscdmadaptor.h> 706fc46c8SMatthew G. Knepley #include <petscconvest.h> 89b94acceSBarry Smith 9ace3abfcSBarry Smith PetscBool SNESRegisterAllCalled = PETSC_FALSE; 100298fd71SBarry Smith PetscFunctionList SNESList = NULL; 118ba1e511SMatthew Knepley 128ba1e511SMatthew Knepley /* Logging support */ 1322c6f798SBarry Smith PetscClassId SNES_CLASSID, DMSNES_CLASSID; 1494db00ebSBarry Smith PetscLogEvent SNES_Solve, SNES_FunctionEval, SNES_JacobianEval, SNES_NGSEval, SNES_NGSFuncEval, SNES_NPCSolve, SNES_ObjectiveEval; 15a09944afSBarry Smith 16e113a28aSBarry Smith /*@ 17e113a28aSBarry Smith SNESSetErrorIfNotConverged - Causes SNESSolve() to generate an error if the solver has not converged. 18e113a28aSBarry Smith 193f9fe445SBarry Smith Logically Collective on SNES 20e113a28aSBarry Smith 21e113a28aSBarry Smith Input Parameters: 22e113a28aSBarry Smith + snes - iterative context obtained from SNESCreate() 23e113a28aSBarry Smith - flg - PETSC_TRUE indicates you want the error generated 24e113a28aSBarry Smith 25e113a28aSBarry Smith Options database keys: 26e113a28aSBarry Smith . -snes_error_if_not_converged : this takes an optional truth value (0/1/no/yes/true/false) 27e113a28aSBarry Smith 28e113a28aSBarry Smith Level: intermediate 29e113a28aSBarry Smith 30e113a28aSBarry Smith Notes: 31e113a28aSBarry Smith Normally PETSc continues if a linear solver fails to converge, you can call SNESGetConvergedReason() after a SNESSolve() 32e113a28aSBarry Smith to determine if it has converged. 33e113a28aSBarry Smith 34e113a28aSBarry Smith .keywords: SNES, set, initial guess, nonzero 35e113a28aSBarry Smith 36e113a28aSBarry Smith .seealso: SNESGetErrorIfNotConverged(), KSPGetErrorIfNotConverged(), KSPSetErrorIFNotConverged() 37e113a28aSBarry Smith @*/ 387087cfbeSBarry Smith PetscErrorCode SNESSetErrorIfNotConverged(SNES snes,PetscBool flg) 39e113a28aSBarry Smith { 40e113a28aSBarry Smith PetscFunctionBegin; 41e113a28aSBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 42acfcf0e5SJed Brown PetscValidLogicalCollectiveBool(snes,flg,2); 43e113a28aSBarry Smith snes->errorifnotconverged = flg; 44e113a28aSBarry Smith PetscFunctionReturn(0); 45e113a28aSBarry Smith } 46e113a28aSBarry Smith 47e113a28aSBarry Smith /*@ 48e113a28aSBarry Smith SNESGetErrorIfNotConverged - Will SNESSolve() generate an error if the solver does not converge? 49e113a28aSBarry Smith 50e113a28aSBarry Smith Not Collective 51e113a28aSBarry Smith 52e113a28aSBarry Smith Input Parameter: 53e113a28aSBarry Smith . snes - iterative context obtained from SNESCreate() 54e113a28aSBarry Smith 55e113a28aSBarry Smith Output Parameter: 56e113a28aSBarry Smith . flag - PETSC_TRUE if it will generate an error, else PETSC_FALSE 57e113a28aSBarry Smith 58e113a28aSBarry Smith Level: intermediate 59e113a28aSBarry Smith 60e113a28aSBarry Smith .keywords: SNES, set, initial guess, nonzero 61e113a28aSBarry Smith 62e113a28aSBarry Smith .seealso: SNESSetErrorIfNotConverged(), KSPGetErrorIfNotConverged(), KSPSetErrorIFNotConverged() 63e113a28aSBarry Smith @*/ 647087cfbeSBarry Smith PetscErrorCode SNESGetErrorIfNotConverged(SNES snes,PetscBool *flag) 65e113a28aSBarry Smith { 66e113a28aSBarry Smith PetscFunctionBegin; 67e113a28aSBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 68e113a28aSBarry Smith PetscValidPointer(flag,2); 69e113a28aSBarry Smith *flag = snes->errorifnotconverged; 70e113a28aSBarry Smith PetscFunctionReturn(0); 71e113a28aSBarry Smith } 72e113a28aSBarry Smith 734fc747eaSLawrence Mitchell /*@ 744fc747eaSLawrence Mitchell SNESSetAlwaysComputesFinalResidual - does the SNES always compute the residual at the final solution? 754fc747eaSLawrence Mitchell 764fc747eaSLawrence Mitchell Logically Collective on SNES 774fc747eaSLawrence Mitchell 784fc747eaSLawrence Mitchell Input Parameters: 794fc747eaSLawrence Mitchell + snes - the shell SNES 804fc747eaSLawrence Mitchell - flg - is the residual computed? 814fc747eaSLawrence Mitchell 824fc747eaSLawrence Mitchell Level: advanced 834fc747eaSLawrence Mitchell 844fc747eaSLawrence Mitchell .seealso: SNESGetAlwaysComputesFinalResidual() 854fc747eaSLawrence Mitchell @*/ 864fc747eaSLawrence Mitchell PetscErrorCode SNESSetAlwaysComputesFinalResidual(SNES snes, PetscBool flg) 874fc747eaSLawrence Mitchell { 884fc747eaSLawrence Mitchell PetscFunctionBegin; 894fc747eaSLawrence Mitchell PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 904fc747eaSLawrence Mitchell snes->alwayscomputesfinalresidual = flg; 914fc747eaSLawrence Mitchell PetscFunctionReturn(0); 924fc747eaSLawrence Mitchell } 934fc747eaSLawrence Mitchell 944fc747eaSLawrence Mitchell /*@ 954fc747eaSLawrence Mitchell SNESGetAlwaysComputesFinalResidual - does the SNES always compute the residual at the final solution? 964fc747eaSLawrence Mitchell 974fc747eaSLawrence Mitchell Logically Collective on SNES 984fc747eaSLawrence Mitchell 994fc747eaSLawrence Mitchell Input Parameter: 1004fc747eaSLawrence Mitchell . snes - the shell SNES 1014fc747eaSLawrence Mitchell 1024fc747eaSLawrence Mitchell Output Parameter: 1034fc747eaSLawrence Mitchell . flg - is the residual computed? 1044fc747eaSLawrence Mitchell 1054fc747eaSLawrence Mitchell Level: advanced 1064fc747eaSLawrence Mitchell 1074fc747eaSLawrence Mitchell .seealso: SNESSetAlwaysComputesFinalResidual() 1084fc747eaSLawrence Mitchell @*/ 1094fc747eaSLawrence Mitchell PetscErrorCode SNESGetAlwaysComputesFinalResidual(SNES snes, PetscBool *flg) 1104fc747eaSLawrence Mitchell { 1114fc747eaSLawrence Mitchell PetscFunctionBegin; 1124fc747eaSLawrence Mitchell PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 1134fc747eaSLawrence Mitchell *flg = snes->alwayscomputesfinalresidual; 1144fc747eaSLawrence Mitchell PetscFunctionReturn(0); 1154fc747eaSLawrence Mitchell } 1164fc747eaSLawrence Mitchell 117e725d27bSBarry Smith /*@ 118bf388a1fSBarry Smith SNESSetFunctionDomainError - tells SNES that the input vector to your SNESFunction is not 1194936397dSBarry Smith in the functions domain. For example, negative pressure. 1204936397dSBarry Smith 1213f9fe445SBarry Smith Logically Collective on SNES 1224936397dSBarry Smith 1234936397dSBarry Smith Input Parameters: 1246a388c36SPeter Brune . snes - the SNES context 1254936397dSBarry Smith 12628529972SSatish Balay Level: advanced 1274936397dSBarry Smith 1284936397dSBarry Smith .keywords: SNES, view 1294936397dSBarry Smith 130bf388a1fSBarry Smith .seealso: SNESCreate(), SNESSetFunction(), SNESFunction 1314936397dSBarry Smith @*/ 1327087cfbeSBarry Smith PetscErrorCode SNESSetFunctionDomainError(SNES snes) 1334936397dSBarry Smith { 1344936397dSBarry Smith PetscFunctionBegin; 1350700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 136422a814eSBarry Smith if (snes->errorifnotconverged) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"User code indicates input vector is not in the function domain"); 1374936397dSBarry Smith snes->domainerror = PETSC_TRUE; 1384936397dSBarry Smith PetscFunctionReturn(0); 1394936397dSBarry Smith } 1404936397dSBarry Smith 1416a388c36SPeter Brune /*@ 142c77b2880SPeter Brune SNESGetFunctionDomainError - Gets the status of the domain error after a call to SNESComputeFunction; 1436a388c36SPeter Brune 1446a388c36SPeter Brune Logically Collective on SNES 1456a388c36SPeter Brune 1466a388c36SPeter Brune Input Parameters: 1476a388c36SPeter Brune . snes - the SNES context 1486a388c36SPeter Brune 1496a388c36SPeter Brune Output Parameters: 150bf388a1fSBarry Smith . domainerror - Set to PETSC_TRUE if there's a domain error; PETSC_FALSE otherwise. 1516a388c36SPeter Brune 1526a388c36SPeter Brune Level: advanced 1536a388c36SPeter Brune 1546a388c36SPeter Brune .keywords: SNES, view 1556a388c36SPeter Brune 156bf388a1fSBarry Smith .seealso: SNESSetFunctionDomainError(), SNESComputeFunction() 1576a388c36SPeter Brune @*/ 1586a388c36SPeter Brune PetscErrorCode SNESGetFunctionDomainError(SNES snes, PetscBool *domainerror) 1596a388c36SPeter Brune { 1606a388c36SPeter Brune PetscFunctionBegin; 1616a388c36SPeter Brune PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 1626a388c36SPeter Brune PetscValidPointer(domainerror, 2); 1636a388c36SPeter Brune *domainerror = snes->domainerror; 1646a388c36SPeter Brune PetscFunctionReturn(0); 1656a388c36SPeter Brune } 1666a388c36SPeter Brune 16755849f57SBarry Smith /*@C 16855849f57SBarry Smith SNESLoad - Loads a SNES that has been stored in binary with SNESView(). 16955849f57SBarry Smith 17055849f57SBarry Smith Collective on PetscViewer 17155849f57SBarry Smith 17255849f57SBarry Smith Input Parameters: 17355849f57SBarry Smith + newdm - the newly loaded SNES, this needs to have been created with SNESCreate() or 17455849f57SBarry Smith some related function before a call to SNESLoad(). 17555849f57SBarry Smith - viewer - binary file viewer, obtained from PetscViewerBinaryOpen() 17655849f57SBarry Smith 17755849f57SBarry Smith Level: intermediate 17855849f57SBarry Smith 17955849f57SBarry Smith Notes: 18055849f57SBarry Smith The type is determined by the data in the file, any type set into the SNES before this call is ignored. 18155849f57SBarry Smith 18255849f57SBarry Smith Notes for advanced users: 18355849f57SBarry Smith Most users should not need to know the details of the binary storage 18455849f57SBarry Smith format, since SNESLoad() and TSView() completely hide these details. 18555849f57SBarry Smith But for anyone who's interested, the standard binary matrix storage 18655849f57SBarry Smith format is 18755849f57SBarry Smith .vb 18855849f57SBarry Smith has not yet been determined 18955849f57SBarry Smith .ve 19055849f57SBarry Smith 19155849f57SBarry Smith .seealso: PetscViewerBinaryOpen(), SNESView(), MatLoad(), VecLoad() 19255849f57SBarry Smith @*/ 1932d53ad75SBarry Smith PetscErrorCode SNESLoad(SNES snes, PetscViewer viewer) 19455849f57SBarry Smith { 19555849f57SBarry Smith PetscErrorCode ierr; 19655849f57SBarry Smith PetscBool isbinary; 197060da220SMatthew G. Knepley PetscInt classid; 19855849f57SBarry Smith char type[256]; 19955849f57SBarry Smith KSP ksp; 2002d53ad75SBarry Smith DM dm; 2012d53ad75SBarry Smith DMSNES dmsnes; 20255849f57SBarry Smith 20355849f57SBarry Smith PetscFunctionBegin; 2042d53ad75SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 20555849f57SBarry Smith PetscValidHeaderSpecific(viewer,PETSC_VIEWER_CLASSID,2); 20655849f57SBarry Smith ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERBINARY,&isbinary);CHKERRQ(ierr); 20755849f57SBarry Smith if (!isbinary) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONG,"Invalid viewer; open viewer with PetscViewerBinaryOpen()"); 20855849f57SBarry Smith 209060da220SMatthew G. Knepley ierr = PetscViewerBinaryRead(viewer,&classid,1,NULL,PETSC_INT);CHKERRQ(ierr); 210ce94432eSBarry Smith if (classid != SNES_FILE_CLASSID) SETERRQ(PetscObjectComm((PetscObject)snes),PETSC_ERR_ARG_WRONG,"Not SNES next in file"); 211060da220SMatthew G. Knepley ierr = PetscViewerBinaryRead(viewer,type,256,NULL,PETSC_CHAR);CHKERRQ(ierr); 2122d53ad75SBarry Smith ierr = SNESSetType(snes, type);CHKERRQ(ierr); 2132d53ad75SBarry Smith if (snes->ops->load) { 2142d53ad75SBarry Smith ierr = (*snes->ops->load)(snes,viewer);CHKERRQ(ierr); 215f2c2a1b9SBarry Smith } 2162d53ad75SBarry Smith ierr = SNESGetDM(snes,&dm);CHKERRQ(ierr); 2172d53ad75SBarry Smith ierr = DMGetDMSNES(dm,&dmsnes);CHKERRQ(ierr); 2182d53ad75SBarry Smith ierr = DMSNESLoad(dmsnes,viewer);CHKERRQ(ierr); 2192d53ad75SBarry Smith ierr = SNESGetKSP(snes,&ksp);CHKERRQ(ierr); 22055849f57SBarry Smith ierr = KSPLoad(ksp,viewer);CHKERRQ(ierr); 22155849f57SBarry Smith PetscFunctionReturn(0); 22255849f57SBarry Smith } 2236a388c36SPeter Brune 2249804daf3SBarry Smith #include <petscdraw.h> 225e04113cfSBarry Smith #if defined(PETSC_HAVE_SAWS) 226e04113cfSBarry Smith #include <petscviewersaws.h> 227bfb97211SBarry Smith #endif 2287e2c5f70SBarry Smith /*@C 2299b94acceSBarry Smith SNESView - Prints the SNES data structure. 2309b94acceSBarry Smith 2314c49b128SBarry Smith Collective on SNES 232fee21e36SBarry Smith 233c7afd0dbSLois Curfman McInnes Input Parameters: 234c7afd0dbSLois Curfman McInnes + SNES - the SNES context 235c7afd0dbSLois Curfman McInnes - viewer - visualization context 236c7afd0dbSLois Curfman McInnes 2379b94acceSBarry Smith Options Database Key: 238c8a8ba5cSLois Curfman McInnes . -snes_view - Calls SNESView() at end of SNESSolve() 2399b94acceSBarry Smith 2409b94acceSBarry Smith Notes: 2419b94acceSBarry Smith The available visualization contexts include 242b0a32e0cSBarry Smith + PETSC_VIEWER_STDOUT_SELF - standard output (default) 243b0a32e0cSBarry Smith - PETSC_VIEWER_STDOUT_WORLD - synchronized standard 244c8a8ba5cSLois Curfman McInnes output where only the first processor opens 245c8a8ba5cSLois Curfman McInnes the file. All other processors send their 246c8a8ba5cSLois Curfman McInnes data to the first processor to print. 2479b94acceSBarry Smith 2483e081fefSLois Curfman McInnes The user can open an alternative visualization context with 249b0a32e0cSBarry Smith PetscViewerASCIIOpen() - output to a specified file. 2509b94acceSBarry Smith 25136851e7fSLois Curfman McInnes Level: beginner 25236851e7fSLois Curfman McInnes 2539b94acceSBarry Smith .keywords: SNES, view 2549b94acceSBarry Smith 255b0a32e0cSBarry Smith .seealso: PetscViewerASCIIOpen() 2569b94acceSBarry Smith @*/ 2577087cfbeSBarry Smith PetscErrorCode SNESView(SNES snes,PetscViewer viewer) 2589b94acceSBarry Smith { 259fa9f3622SBarry Smith SNESKSPEW *kctx; 260dfbe8321SBarry Smith PetscErrorCode ierr; 26194b7f48cSBarry Smith KSP ksp; 2627f1410a3SPeter Brune SNESLineSearch linesearch; 26372a02f06SBarry Smith PetscBool iascii,isstring,isbinary,isdraw; 2642d53ad75SBarry Smith DMSNES dmsnes; 265e04113cfSBarry Smith #if defined(PETSC_HAVE_SAWS) 266536b137fSBarry Smith PetscBool issaws; 267bfb97211SBarry Smith #endif 2689b94acceSBarry Smith 2693a40ed3dSBarry Smith PetscFunctionBegin; 2700700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 2713050cee2SBarry Smith if (!viewer) { 272ce94432eSBarry Smith ierr = PetscViewerASCIIGetStdout(PetscObjectComm((PetscObject)snes),&viewer);CHKERRQ(ierr); 2733050cee2SBarry Smith } 2740700a824SBarry Smith PetscValidHeaderSpecific(viewer,PETSC_VIEWER_CLASSID,2); 275c9780b6fSBarry Smith PetscCheckSameComm(snes,1,viewer,2); 27674679c65SBarry Smith 277251f4c67SDmitry Karpeev ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERASCII,&iascii);CHKERRQ(ierr); 278251f4c67SDmitry Karpeev ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERSTRING,&isstring);CHKERRQ(ierr); 27955849f57SBarry Smith ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERBINARY,&isbinary);CHKERRQ(ierr); 28072a02f06SBarry Smith ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERDRAW,&isdraw);CHKERRQ(ierr); 281e04113cfSBarry Smith #if defined(PETSC_HAVE_SAWS) 282536b137fSBarry Smith ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERSAWS,&issaws);CHKERRQ(ierr); 283bfb97211SBarry Smith #endif 28432077d6dSBarry Smith if (iascii) { 285dc0571f2SMatthew G. Knepley SNESNormSchedule normschedule; 286dc0571f2SMatthew G. Knepley 287dae58748SBarry Smith ierr = PetscObjectPrintClassNamePrefixType((PetscObject)snes,viewer);CHKERRQ(ierr); 288fce1e034SJed Brown if (!snes->setupcalled) { 289fce1e034SJed Brown ierr = PetscViewerASCIIPrintf(viewer," SNES has not been set up so information may be incomplete\n");CHKERRQ(ierr); 290fce1e034SJed Brown } 291e7788613SBarry Smith if (snes->ops->view) { 292b0a32e0cSBarry Smith ierr = PetscViewerASCIIPushTab(viewer);CHKERRQ(ierr); 293e7788613SBarry Smith ierr = (*snes->ops->view)(snes,viewer);CHKERRQ(ierr); 294b0a32e0cSBarry Smith ierr = PetscViewerASCIIPopTab(viewer);CHKERRQ(ierr); 2950ef38995SBarry Smith } 29677431f27SBarry Smith ierr = PetscViewerASCIIPrintf(viewer," maximum iterations=%D, maximum function evaluations=%D\n",snes->max_its,snes->max_funcs);CHKERRQ(ierr); 29757622a8eSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," tolerances: relative=%g, absolute=%g, solution=%g\n",(double)snes->rtol,(double)snes->abstol,(double)snes->stol);CHKERRQ(ierr); 298efd4aadfSBarry Smith if (snes->usesksp) { 29977431f27SBarry Smith ierr = PetscViewerASCIIPrintf(viewer," total number of linear solver iterations=%D\n",snes->linear_its);CHKERRQ(ierr); 300efd4aadfSBarry Smith } 30177431f27SBarry Smith ierr = PetscViewerASCIIPrintf(viewer," total number of function evaluations=%D\n",snes->nfuncs);CHKERRQ(ierr); 302dc0571f2SMatthew G. Knepley ierr = SNESGetNormSchedule(snes, &normschedule);CHKERRQ(ierr); 303dc0571f2SMatthew G. Knepley if (normschedule > 0) {ierr = PetscViewerASCIIPrintf(viewer," norm schedule %s\n",SNESNormSchedules[normschedule]);CHKERRQ(ierr);} 30417fe4bdfSPeter Brune if (snes->gridsequence) { 30517fe4bdfSPeter Brune ierr = PetscViewerASCIIPrintf(viewer," total number of grid sequence refinements=%D\n",snes->gridsequence);CHKERRQ(ierr); 30617fe4bdfSPeter Brune } 3079b94acceSBarry Smith if (snes->ksp_ewconv) { 308fa9f3622SBarry Smith kctx = (SNESKSPEW*)snes->kspconvctx; 3099b94acceSBarry Smith if (kctx) { 31077431f27SBarry Smith ierr = PetscViewerASCIIPrintf(viewer," Eisenstat-Walker computation of KSP relative tolerance (version %D)\n",kctx->version);CHKERRQ(ierr); 31157622a8eSBarry 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); 31257622a8eSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," gamma=%g, alpha=%g, alpha2=%g\n",(double)kctx->gamma,(double)kctx->alpha,(double)kctx->alpha2);CHKERRQ(ierr); 3139b94acceSBarry Smith } 3149b94acceSBarry Smith } 315eb1f6c34SBarry Smith if (snes->lagpreconditioner == -1) { 316eb1f6c34SBarry Smith ierr = PetscViewerASCIIPrintf(viewer," Preconditioned is never rebuilt\n");CHKERRQ(ierr); 317eb1f6c34SBarry Smith } else if (snes->lagpreconditioner > 1) { 318eb1f6c34SBarry Smith ierr = PetscViewerASCIIPrintf(viewer," Preconditioned is rebuilt every %D new Jacobians\n",snes->lagpreconditioner);CHKERRQ(ierr); 319eb1f6c34SBarry Smith } 320eb1f6c34SBarry Smith if (snes->lagjacobian == -1) { 321eb1f6c34SBarry Smith ierr = PetscViewerASCIIPrintf(viewer," Jacobian is never rebuilt\n");CHKERRQ(ierr); 322eb1f6c34SBarry Smith } else if (snes->lagjacobian > 1) { 32342f4f86dSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," Jacobian is rebuilt every %D SNES iterations\n",snes->lagjacobian);CHKERRQ(ierr); 324eb1f6c34SBarry Smith } 3250f5bd95cSBarry Smith } else if (isstring) { 326317d6ea6SBarry Smith const char *type; 327454a90a3SBarry Smith ierr = SNESGetType(snes,&type);CHKERRQ(ierr); 328b0a32e0cSBarry Smith ierr = PetscViewerStringSPrintf(viewer," %-3.3s",type);CHKERRQ(ierr); 32955849f57SBarry Smith } else if (isbinary) { 33055849f57SBarry Smith PetscInt classid = SNES_FILE_CLASSID; 33155849f57SBarry Smith MPI_Comm comm; 33255849f57SBarry Smith PetscMPIInt rank; 33355849f57SBarry Smith char type[256]; 33455849f57SBarry Smith 33555849f57SBarry Smith ierr = PetscObjectGetComm((PetscObject)snes,&comm);CHKERRQ(ierr); 33655849f57SBarry Smith ierr = MPI_Comm_rank(comm,&rank);CHKERRQ(ierr); 33755849f57SBarry Smith if (!rank) { 33855849f57SBarry Smith ierr = PetscViewerBinaryWrite(viewer,&classid,1,PETSC_INT,PETSC_FALSE);CHKERRQ(ierr); 33989d949e2SBarry Smith ierr = PetscStrncpy(type,((PetscObject)snes)->type_name,sizeof(type));CHKERRQ(ierr); 34089d949e2SBarry Smith ierr = PetscViewerBinaryWrite(viewer,type,sizeof(type),PETSC_CHAR,PETSC_FALSE);CHKERRQ(ierr); 34155849f57SBarry Smith } 34255849f57SBarry Smith if (snes->ops->view) { 34355849f57SBarry Smith ierr = (*snes->ops->view)(snes,viewer);CHKERRQ(ierr); 34455849f57SBarry Smith } 34572a02f06SBarry Smith } else if (isdraw) { 34672a02f06SBarry Smith PetscDraw draw; 34772a02f06SBarry Smith char str[36]; 34889fd9fafSBarry Smith PetscReal x,y,bottom,h; 34972a02f06SBarry Smith 35072a02f06SBarry Smith ierr = PetscViewerDrawGetDraw(viewer,0,&draw);CHKERRQ(ierr); 35172a02f06SBarry Smith ierr = PetscDrawGetCurrentPoint(draw,&x,&y);CHKERRQ(ierr); 35272a02f06SBarry Smith ierr = PetscStrcpy(str,"SNES: ");CHKERRQ(ierr); 35372a02f06SBarry Smith ierr = PetscStrcat(str,((PetscObject)snes)->type_name);CHKERRQ(ierr); 35451fa3d41SBarry Smith ierr = PetscDrawStringBoxed(draw,x,y,PETSC_DRAW_BLUE,PETSC_DRAW_BLACK,str,NULL,&h);CHKERRQ(ierr); 35589fd9fafSBarry Smith bottom = y - h; 35672a02f06SBarry Smith ierr = PetscDrawPushCurrentPoint(draw,x,bottom);CHKERRQ(ierr); 357c4646bacSPeter Brune if (snes->ops->view) { 358c4646bacSPeter Brune ierr = (*snes->ops->view)(snes,viewer);CHKERRQ(ierr); 359c4646bacSPeter Brune } 360e04113cfSBarry Smith #if defined(PETSC_HAVE_SAWS) 361536b137fSBarry Smith } else if (issaws) { 362d45a07a7SBarry Smith PetscMPIInt rank; 3632657e9d9SBarry Smith const char *name; 364d45a07a7SBarry Smith 3652657e9d9SBarry Smith ierr = PetscObjectGetName((PetscObject)snes,&name);CHKERRQ(ierr); 366d45a07a7SBarry Smith ierr = MPI_Comm_rank(PETSC_COMM_WORLD,&rank);CHKERRQ(ierr); 367d45a07a7SBarry Smith if (!((PetscObject)snes)->amsmem && !rank) { 368d45a07a7SBarry Smith char dir[1024]; 369d45a07a7SBarry Smith 370e04113cfSBarry Smith ierr = PetscObjectViewSAWs((PetscObject)snes,viewer);CHKERRQ(ierr); 3712657e9d9SBarry Smith ierr = PetscSNPrintf(dir,1024,"/PETSc/Objects/%s/its",name);CHKERRQ(ierr); 3722657e9d9SBarry Smith PetscStackCallSAWs(SAWs_Register,(dir,&snes->iter,1,SAWs_READ,SAWs_INT)); 373bfb97211SBarry Smith if (!snes->conv_hist) { 374a0931e03SBarry Smith ierr = SNESSetConvergenceHistory(snes,NULL,NULL,PETSC_DECIDE,PETSC_TRUE);CHKERRQ(ierr); 375bfb97211SBarry Smith } 3762657e9d9SBarry Smith ierr = PetscSNPrintf(dir,1024,"/PETSc/Objects/%s/conv_hist",name);CHKERRQ(ierr); 3772657e9d9SBarry Smith PetscStackCallSAWs(SAWs_Register,(dir,snes->conv_hist,10,SAWs_READ,SAWs_DOUBLE)); 378f05ece33SBarry Smith } 379bfb97211SBarry Smith #endif 38072a02f06SBarry Smith } 38172a02f06SBarry Smith if (snes->linesearch) { 38272a02f06SBarry Smith ierr = PetscViewerASCIIPushTab(viewer);CHKERRQ(ierr); 3837601faf0SJed Brown ierr = SNESGetLineSearch(snes, &linesearch);CHKERRQ(ierr); 38472a02f06SBarry Smith ierr = SNESLineSearchView(linesearch, viewer);CHKERRQ(ierr); 38572a02f06SBarry Smith ierr = PetscViewerASCIIPopTab(viewer);CHKERRQ(ierr); 38619bcc07fSBarry Smith } 387efd4aadfSBarry Smith if (snes->npc && snes->usesnpc) { 3884a0c5b0cSMatthew G Knepley ierr = PetscViewerASCIIPushTab(viewer);CHKERRQ(ierr); 389efd4aadfSBarry Smith ierr = SNESView(snes->npc, viewer);CHKERRQ(ierr); 3904a0c5b0cSMatthew G Knepley ierr = PetscViewerASCIIPopTab(viewer);CHKERRQ(ierr); 3914a0c5b0cSMatthew G Knepley } 3922d53ad75SBarry Smith ierr = PetscViewerASCIIPushTab(viewer);CHKERRQ(ierr); 3932d53ad75SBarry Smith ierr = DMGetDMSNES(snes->dm,&dmsnes);CHKERRQ(ierr); 3942d53ad75SBarry Smith ierr = DMSNESView(dmsnes, viewer);CHKERRQ(ierr); 3952d53ad75SBarry Smith ierr = PetscViewerASCIIPopTab(viewer);CHKERRQ(ierr); 3962c155ee1SBarry Smith if (snes->usesksp) { 3972c155ee1SBarry Smith ierr = SNESGetKSP(snes,&ksp);CHKERRQ(ierr); 398b0a32e0cSBarry Smith ierr = PetscViewerASCIIPushTab(viewer);CHKERRQ(ierr); 39994b7f48cSBarry Smith ierr = KSPView(ksp,viewer);CHKERRQ(ierr); 400b0a32e0cSBarry Smith ierr = PetscViewerASCIIPopTab(viewer);CHKERRQ(ierr); 4012c155ee1SBarry Smith } 40272a02f06SBarry Smith if (isdraw) { 40372a02f06SBarry Smith PetscDraw draw; 40472a02f06SBarry Smith ierr = PetscViewerDrawGetDraw(viewer,0,&draw);CHKERRQ(ierr); 40572a02f06SBarry Smith ierr = PetscDrawPopCurrentPoint(draw);CHKERRQ(ierr); 4067f1410a3SPeter Brune } 4073a40ed3dSBarry Smith PetscFunctionReturn(0); 4089b94acceSBarry Smith } 4099b94acceSBarry Smith 41076b2cf59SMatthew Knepley /* 41176b2cf59SMatthew Knepley We retain a list of functions that also take SNES command 41276b2cf59SMatthew Knepley line options. These are called at the end SNESSetFromOptions() 41376b2cf59SMatthew Knepley */ 41476b2cf59SMatthew Knepley #define MAXSETFROMOPTIONS 5 415a7cc72afSBarry Smith static PetscInt numberofsetfromoptions; 4166849ba73SBarry Smith static PetscErrorCode (*othersetfromoptions[MAXSETFROMOPTIONS])(SNES); 41776b2cf59SMatthew Knepley 418ac226902SBarry Smith /*@C 41976b2cf59SMatthew Knepley SNESAddOptionsChecker - Adds an additional function to check for SNES options. 42076b2cf59SMatthew Knepley 42176b2cf59SMatthew Knepley Not Collective 42276b2cf59SMatthew Knepley 42376b2cf59SMatthew Knepley Input Parameter: 42476b2cf59SMatthew Knepley . snescheck - function that checks for options 42576b2cf59SMatthew Knepley 42676b2cf59SMatthew Knepley Level: developer 42776b2cf59SMatthew Knepley 42876b2cf59SMatthew Knepley .seealso: SNESSetFromOptions() 42976b2cf59SMatthew Knepley @*/ 4307087cfbeSBarry Smith PetscErrorCode SNESAddOptionsChecker(PetscErrorCode (*snescheck)(SNES)) 43176b2cf59SMatthew Knepley { 43276b2cf59SMatthew Knepley PetscFunctionBegin; 433f23aa3ddSBarry Smith if (numberofsetfromoptions >= MAXSETFROMOPTIONS) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE, "Too many options checkers, only %D allowed", MAXSETFROMOPTIONS); 43476b2cf59SMatthew Knepley othersetfromoptions[numberofsetfromoptions++] = snescheck; 43576b2cf59SMatthew Knepley PetscFunctionReturn(0); 43676b2cf59SMatthew Knepley } 43776b2cf59SMatthew Knepley 4387087cfbeSBarry Smith extern PetscErrorCode SNESDefaultMatrixFreeCreate2(SNES,Vec,Mat*); 439aa3661deSLisandro Dalcin 440ace3abfcSBarry Smith static PetscErrorCode SNESSetUpMatrixFree_Private(SNES snes, PetscBool hasOperator, PetscInt version) 441aa3661deSLisandro Dalcin { 442aa3661deSLisandro Dalcin Mat J; 443aa3661deSLisandro Dalcin KSP ksp; 444aa3661deSLisandro Dalcin PC pc; 445ace3abfcSBarry Smith PetscBool match; 446aa3661deSLisandro Dalcin PetscErrorCode ierr; 447895c21f2SBarry Smith MatNullSpace nullsp; 448aa3661deSLisandro Dalcin 449aa3661deSLisandro Dalcin PetscFunctionBegin; 4500700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 451aa3661deSLisandro Dalcin 45298613b67SLisandro Dalcin if (!snes->vec_func && (snes->jacobian || snes->jacobian_pre)) { 45398613b67SLisandro Dalcin Mat A = snes->jacobian, B = snes->jacobian_pre; 4542a7a6963SBarry Smith ierr = MatCreateVecs(A ? A : B, NULL,&snes->vec_func);CHKERRQ(ierr); 45598613b67SLisandro Dalcin } 45698613b67SLisandro Dalcin 457aa3661deSLisandro Dalcin if (version == 1) { 458aa3661deSLisandro Dalcin ierr = MatCreateSNESMF(snes,&J);CHKERRQ(ierr); 45998613b67SLisandro Dalcin ierr = MatMFFDSetOptionsPrefix(J,((PetscObject)snes)->prefix);CHKERRQ(ierr); 4609c6ac3b3SBarry Smith ierr = MatSetFromOptions(J);CHKERRQ(ierr); 461aa3661deSLisandro Dalcin } else if (version == 2) { 462e32f2f54SBarry Smith if (!snes->vec_func) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"SNESSetFunction() must be called first"); 463570b7f6dSBarry Smith #if !defined(PETSC_USE_COMPLEX) && !defined(PETSC_USE_REAL_SINGLE) && !defined(PETSC_USE_REAL___FLOAT128) && !defined(PETSC_USE_REAL___FP16) 464aa3661deSLisandro Dalcin ierr = SNESDefaultMatrixFreeCreate2(snes,snes->vec_func,&J);CHKERRQ(ierr); 465aa3661deSLisandro Dalcin #else 466e32f2f54SBarry Smith SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP, "matrix-free operator rutines (version 2)"); 467aa3661deSLisandro Dalcin #endif 468a8248277SBarry Smith } else SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE, "matrix-free operator rutines, only version 1 and 2"); 469aa3661deSLisandro Dalcin 470895c21f2SBarry Smith /* attach any user provided null space that was on Amat to the newly created matrix free matrix */ 471895c21f2SBarry Smith if (snes->jacobian) { 472895c21f2SBarry Smith ierr = MatGetNullSpace(snes->jacobian,&nullsp);CHKERRQ(ierr); 473895c21f2SBarry Smith if (nullsp) { 474895c21f2SBarry Smith ierr = MatSetNullSpace(J,nullsp);CHKERRQ(ierr); 475895c21f2SBarry Smith } 476895c21f2SBarry Smith } 477895c21f2SBarry Smith 478aa3661deSLisandro Dalcin ierr = PetscInfo1(snes,"Setting default matrix-free operator routines (version %D)\n", version);CHKERRQ(ierr); 479d3462f78SMatthew Knepley if (hasOperator) { 4803232da50SPeter Brune 481aa3661deSLisandro Dalcin /* This version replaces the user provided Jacobian matrix with a 482aa3661deSLisandro Dalcin matrix-free version but still employs the user-provided preconditioner matrix. */ 483aa3661deSLisandro Dalcin ierr = SNESSetJacobian(snes,J,0,0,0);CHKERRQ(ierr); 484aa3661deSLisandro Dalcin } else { 485aa3661deSLisandro Dalcin /* This version replaces both the user-provided Jacobian and the user- 4863232da50SPeter Brune provided preconditioner Jacobian with the default matrix free version. */ 487efd4aadfSBarry Smith if ((snes->npcside== PC_LEFT) && snes->npc) { 488d728fb7dSPeter Brune if (!snes->jacobian){ierr = SNESSetJacobian(snes,J,0,0,0);CHKERRQ(ierr);} 489172a4300SPeter Brune } else { 49028a52e04SBarry Smith ierr = SNESSetJacobian(snes,J,J,MatMFFDComputeJacobian,0);CHKERRQ(ierr); 491172a4300SPeter Brune } 492aa3661deSLisandro Dalcin /* Force no preconditioner */ 493aa3661deSLisandro Dalcin ierr = SNESGetKSP(snes,&ksp);CHKERRQ(ierr); 494aa3661deSLisandro Dalcin ierr = KSPGetPC(ksp,&pc);CHKERRQ(ierr); 495251f4c67SDmitry Karpeev ierr = PetscObjectTypeCompare((PetscObject)pc,PCSHELL,&match);CHKERRQ(ierr); 496aa3661deSLisandro Dalcin if (!match) { 497aa3661deSLisandro Dalcin ierr = PetscInfo(snes,"Setting default matrix-free preconditioner routines\nThat is no preconditioner is being used\n");CHKERRQ(ierr); 498aa3661deSLisandro Dalcin ierr = PCSetType(pc,PCNONE);CHKERRQ(ierr); 499aa3661deSLisandro Dalcin } 500aa3661deSLisandro Dalcin } 5016bf464f9SBarry Smith ierr = MatDestroy(&J);CHKERRQ(ierr); 502aa3661deSLisandro Dalcin PetscFunctionReturn(0); 503aa3661deSLisandro Dalcin } 504aa3661deSLisandro Dalcin 505dfe15315SJed Brown static PetscErrorCode DMRestrictHook_SNESVecSol(DM dmfine,Mat Restrict,Vec Rscale,Mat Inject,DM dmcoarse,void *ctx) 506dfe15315SJed Brown { 507dfe15315SJed Brown SNES snes = (SNES)ctx; 508dfe15315SJed Brown PetscErrorCode ierr; 5090298fd71SBarry Smith Vec Xfine,Xfine_named = NULL,Xcoarse; 510dfe15315SJed Brown 511dfe15315SJed Brown PetscFunctionBegin; 51216ebb321SJed Brown if (PetscLogPrintInfo) { 51316ebb321SJed Brown PetscInt finelevel,coarselevel,fineclevel,coarseclevel; 51416ebb321SJed Brown ierr = DMGetRefineLevel(dmfine,&finelevel);CHKERRQ(ierr); 51516ebb321SJed Brown ierr = DMGetCoarsenLevel(dmfine,&fineclevel);CHKERRQ(ierr); 51616ebb321SJed Brown ierr = DMGetRefineLevel(dmcoarse,&coarselevel);CHKERRQ(ierr); 51716ebb321SJed Brown ierr = DMGetCoarsenLevel(dmcoarse,&coarseclevel);CHKERRQ(ierr); 51816ebb321SJed Brown ierr = PetscInfo4(dmfine,"Restricting SNES solution vector from level %D-%D to level %D-%D\n",finelevel,fineclevel,coarselevel,coarseclevel);CHKERRQ(ierr); 51916ebb321SJed Brown } 520dfe15315SJed Brown if (dmfine == snes->dm) Xfine = snes->vec_sol; 521dfe15315SJed Brown else { 522dfe15315SJed Brown ierr = DMGetNamedGlobalVector(dmfine,"SNESVecSol",&Xfine_named);CHKERRQ(ierr); 523dfe15315SJed Brown Xfine = Xfine_named; 524dfe15315SJed Brown } 525dfe15315SJed Brown ierr = DMGetNamedGlobalVector(dmcoarse,"SNESVecSol",&Xcoarse);CHKERRQ(ierr); 526907f5c5aSLawrence Mitchell if (Inject) { 527907f5c5aSLawrence Mitchell ierr = MatRestrict(Inject,Xfine,Xcoarse);CHKERRQ(ierr); 528907f5c5aSLawrence Mitchell } else { 529dfe15315SJed Brown ierr = MatRestrict(Restrict,Xfine,Xcoarse);CHKERRQ(ierr); 530dfe15315SJed Brown ierr = VecPointwiseMult(Xcoarse,Xcoarse,Rscale);CHKERRQ(ierr); 531907f5c5aSLawrence Mitchell } 532dfe15315SJed Brown ierr = DMRestoreNamedGlobalVector(dmcoarse,"SNESVecSol",&Xcoarse);CHKERRQ(ierr); 533dfe15315SJed Brown if (Xfine_named) {ierr = DMRestoreNamedGlobalVector(dmfine,"SNESVecSol",&Xfine_named);CHKERRQ(ierr);} 534dfe15315SJed Brown PetscFunctionReturn(0); 535dfe15315SJed Brown } 536dfe15315SJed Brown 53716ebb321SJed Brown static PetscErrorCode DMCoarsenHook_SNESVecSol(DM dm,DM dmc,void *ctx) 53816ebb321SJed Brown { 53916ebb321SJed Brown PetscErrorCode ierr; 54016ebb321SJed Brown 54116ebb321SJed Brown PetscFunctionBegin; 54216ebb321SJed Brown ierr = DMCoarsenHookAdd(dmc,DMCoarsenHook_SNESVecSol,DMRestrictHook_SNESVecSol,ctx);CHKERRQ(ierr); 54316ebb321SJed Brown PetscFunctionReturn(0); 54416ebb321SJed Brown } 54516ebb321SJed Brown 546a6950cb2SJed Brown /* This may be called to rediscretize the operator on levels of linear multigrid. The DM shuffle is so the user can 547a6950cb2SJed Brown * safely call SNESGetDM() in their residual evaluation routine. */ 54823ee1639SBarry Smith static PetscErrorCode KSPComputeOperators_SNES(KSP ksp,Mat A,Mat B,void *ctx) 549caa4e7f2SJed Brown { 550caa4e7f2SJed Brown SNES snes = (SNES)ctx; 551caa4e7f2SJed Brown PetscErrorCode ierr; 552caa4e7f2SJed Brown Mat Asave = A,Bsave = B; 5530298fd71SBarry Smith Vec X,Xnamed = NULL; 554dfe15315SJed Brown DM dmsave; 5554e269d77SPeter Brune void *ctxsave; 556d1e9a80fSBarry Smith PetscErrorCode (*jac)(SNES,Vec,Mat,Mat,void*); 557caa4e7f2SJed Brown 558caa4e7f2SJed Brown PetscFunctionBegin; 559dfe15315SJed Brown dmsave = snes->dm; 560dfe15315SJed Brown ierr = KSPGetDM(ksp,&snes->dm);CHKERRQ(ierr); 561dfe15315SJed Brown if (dmsave == snes->dm) X = snes->vec_sol; /* We are on the finest level */ 562dfe15315SJed Brown else { /* We are on a coarser level, this vec was initialized using a DM restrict hook */ 563dfe15315SJed Brown ierr = DMGetNamedGlobalVector(snes->dm,"SNESVecSol",&Xnamed);CHKERRQ(ierr); 564dfe15315SJed Brown X = Xnamed; 5650298fd71SBarry Smith ierr = SNESGetJacobian(snes,NULL,NULL,&jac,&ctxsave);CHKERRQ(ierr); 5664e269d77SPeter Brune /* If the DM's don't match up, the MatFDColoring context needed for the jacobian won't match up either -- fixit. */ 5678d359177SBarry Smith if (jac == SNESComputeJacobianDefaultColor) { 5688d359177SBarry Smith ierr = SNESSetJacobian(snes,NULL,NULL,SNESComputeJacobianDefaultColor,0);CHKERRQ(ierr); 569dfe15315SJed Brown } 5704e269d77SPeter Brune } 5714e269d77SPeter Brune /* put the previous context back */ 5724e269d77SPeter Brune 573d1e9a80fSBarry Smith ierr = SNESComputeJacobian(snes,X,A,B);CHKERRQ(ierr); 5748d359177SBarry Smith if (snes->dm != dmsave && jac == SNESComputeJacobianDefaultColor) { 5750298fd71SBarry Smith ierr = SNESSetJacobian(snes,NULL,NULL,jac,ctxsave);CHKERRQ(ierr); 5764e269d77SPeter Brune } 5774e269d77SPeter Brune 578ce94432eSBarry Smith if (A != Asave || B != Bsave) SETERRQ(PetscObjectComm((PetscObject)snes),PETSC_ERR_SUP,"No support for changing matrices at this time"); 579dfe15315SJed Brown if (Xnamed) { 580dfe15315SJed Brown ierr = DMRestoreNamedGlobalVector(snes->dm,"SNESVecSol",&Xnamed);CHKERRQ(ierr); 581dfe15315SJed Brown } 582dfe15315SJed Brown snes->dm = dmsave; 583caa4e7f2SJed Brown PetscFunctionReturn(0); 584caa4e7f2SJed Brown } 585caa4e7f2SJed Brown 5866cab3a1bSJed Brown /*@ 5876cab3a1bSJed Brown SNESSetUpMatrices - ensures that matrices are available for SNES, to be called by SNESSetUp_XXX() 5886cab3a1bSJed Brown 5896cab3a1bSJed Brown Collective 5906cab3a1bSJed Brown 5916cab3a1bSJed Brown Input Arguments: 5926cab3a1bSJed Brown . snes - snes to configure 5936cab3a1bSJed Brown 5946cab3a1bSJed Brown Level: developer 5956cab3a1bSJed Brown 5966cab3a1bSJed Brown .seealso: SNESSetUp() 5976cab3a1bSJed Brown @*/ 5986cab3a1bSJed Brown PetscErrorCode SNESSetUpMatrices(SNES snes) 5996cab3a1bSJed Brown { 6006cab3a1bSJed Brown PetscErrorCode ierr; 6016cab3a1bSJed Brown DM dm; 602942e3340SBarry Smith DMSNES sdm; 6036cab3a1bSJed Brown 6046cab3a1bSJed Brown PetscFunctionBegin; 6056cab3a1bSJed Brown ierr = SNESGetDM(snes,&dm);CHKERRQ(ierr); 606942e3340SBarry Smith ierr = DMGetDMSNES(dm,&sdm);CHKERRQ(ierr); 607ce94432eSBarry Smith if (!sdm->ops->computejacobian) SETERRQ(PetscObjectComm((PetscObject)snes),PETSC_ERR_PLIB,"DMSNES not properly configured"); 608f5af7f23SKarl Rupp else if (!snes->jacobian && snes->mf) { 6096cab3a1bSJed Brown Mat J; 6106cab3a1bSJed Brown void *functx; 6116cab3a1bSJed Brown ierr = MatCreateSNESMF(snes,&J);CHKERRQ(ierr); 6126cab3a1bSJed Brown ierr = MatMFFDSetOptionsPrefix(J,((PetscObject)snes)->prefix);CHKERRQ(ierr); 6136cab3a1bSJed Brown ierr = MatSetFromOptions(J);CHKERRQ(ierr); 6140298fd71SBarry Smith ierr = SNESGetFunction(snes,NULL,NULL,&functx);CHKERRQ(ierr); 6153232da50SPeter Brune ierr = SNESSetJacobian(snes,J,J,0,0);CHKERRQ(ierr); 6166cab3a1bSJed Brown ierr = MatDestroy(&J);CHKERRQ(ierr); 617caa4e7f2SJed Brown } else if (snes->mf_operator && !snes->jacobian_pre && !snes->jacobian) { 6186cab3a1bSJed Brown Mat J,B; 6196cab3a1bSJed Brown ierr = MatCreateSNESMF(snes,&J);CHKERRQ(ierr); 6206cab3a1bSJed Brown ierr = MatMFFDSetOptionsPrefix(J,((PetscObject)snes)->prefix);CHKERRQ(ierr); 6216cab3a1bSJed Brown ierr = MatSetFromOptions(J);CHKERRQ(ierr); 622b412c318SBarry Smith ierr = DMCreateMatrix(snes->dm,&B);CHKERRQ(ierr); 62306f20277SJed Brown /* sdm->computejacobian was already set to reach here */ 6240298fd71SBarry Smith ierr = SNESSetJacobian(snes,J,B,NULL,NULL);CHKERRQ(ierr); 6256cab3a1bSJed Brown ierr = MatDestroy(&J);CHKERRQ(ierr); 6266cab3a1bSJed Brown ierr = MatDestroy(&B);CHKERRQ(ierr); 627caa4e7f2SJed Brown } else if (!snes->jacobian_pre) { 6281ba9b98eSMatthew G. Knepley PetscDS prob; 6296cab3a1bSJed Brown Mat J, B; 6301ba9b98eSMatthew G. Knepley PetscBool hasPrec = PETSC_FALSE; 6311ba9b98eSMatthew G. Knepley 6326cab3a1bSJed Brown J = snes->jacobian; 6331ba9b98eSMatthew G. Knepley ierr = DMGetDS(dm, &prob);CHKERRQ(ierr); 6341ba9b98eSMatthew G. Knepley if (prob) {ierr = PetscDSHasJacobianPreconditioner(prob, &hasPrec);CHKERRQ(ierr);} 635ec9a985fSMatthew G. Knepley if (J) {ierr = PetscObjectReference((PetscObject) J);CHKERRQ(ierr);} 636ec9a985fSMatthew G. Knepley else if (hasPrec) {ierr = DMCreateMatrix(snes->dm, &J);CHKERRQ(ierr);} 637b412c318SBarry Smith ierr = DMCreateMatrix(snes->dm, &B);CHKERRQ(ierr); 6380298fd71SBarry Smith ierr = SNESSetJacobian(snes, J ? J : B, B, NULL, NULL);CHKERRQ(ierr); 6391ba9b98eSMatthew G. Knepley ierr = MatDestroy(&J);CHKERRQ(ierr); 6406cab3a1bSJed Brown ierr = MatDestroy(&B);CHKERRQ(ierr); 6416cab3a1bSJed Brown } 642caa4e7f2SJed Brown { 643caa4e7f2SJed Brown KSP ksp; 644caa4e7f2SJed Brown ierr = SNESGetKSP(snes,&ksp);CHKERRQ(ierr); 645caa4e7f2SJed Brown ierr = KSPSetComputeOperators(ksp,KSPComputeOperators_SNES,snes);CHKERRQ(ierr); 64616ebb321SJed Brown ierr = DMCoarsenHookAdd(snes->dm,DMCoarsenHook_SNESVecSol,DMRestrictHook_SNESVecSol,snes);CHKERRQ(ierr); 647caa4e7f2SJed Brown } 6486cab3a1bSJed Brown PetscFunctionReturn(0); 6496cab3a1bSJed Brown } 6506cab3a1bSJed Brown 651fde5950dSBarry Smith /*@C 652fde5950dSBarry Smith SNESMonitorSetFromOptions - Sets a monitor function and viewer appropriate for the type indicated by the user 653fde5950dSBarry Smith 654fde5950dSBarry Smith Collective on SNES 655fde5950dSBarry Smith 656fde5950dSBarry Smith Input Parameters: 657fde5950dSBarry Smith + snes - SNES object you wish to monitor 658fde5950dSBarry Smith . name - the monitor type one is seeking 659fde5950dSBarry Smith . help - message indicating what monitoring is done 660fde5950dSBarry Smith . manual - manual page for the monitor 661fde5950dSBarry Smith . monitor - the monitor function 662fde5950dSBarry 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 663fde5950dSBarry Smith 664fde5950dSBarry Smith Level: developer 665fde5950dSBarry Smith 666fde5950dSBarry Smith .seealso: PetscOptionsGetViewer(), PetscOptionsGetReal(), PetscOptionsHasName(), PetscOptionsGetString(), 667fde5950dSBarry Smith PetscOptionsGetIntArray(), PetscOptionsGetRealArray(), PetscOptionsBool() 668fde5950dSBarry Smith PetscOptionsInt(), PetscOptionsString(), PetscOptionsReal(), PetscOptionsBool(), 669fde5950dSBarry Smith PetscOptionsName(), PetscOptionsBegin(), PetscOptionsEnd(), PetscOptionsHead(), 670fde5950dSBarry Smith PetscOptionsStringArray(),PetscOptionsRealArray(), PetscOptionsScalar(), 671fde5950dSBarry Smith PetscOptionsBoolGroupBegin(), PetscOptionsBoolGroup(), PetscOptionsBoolGroupEnd(), 672fde5950dSBarry Smith PetscOptionsFList(), PetscOptionsEList() 673fde5950dSBarry Smith @*/ 674d43b4f6eSBarry Smith PetscErrorCode SNESMonitorSetFromOptions(SNES snes,const char name[],const char help[], const char manual[],PetscErrorCode (*monitor)(SNES,PetscInt,PetscReal,PetscViewerAndFormat*),PetscErrorCode (*monitorsetup)(SNES,PetscViewerAndFormat*)) 675fde5950dSBarry Smith { 676fde5950dSBarry Smith PetscErrorCode ierr; 677fde5950dSBarry Smith PetscViewer viewer; 678fde5950dSBarry Smith PetscViewerFormat format; 679fde5950dSBarry Smith PetscBool flg; 680fde5950dSBarry Smith 681fde5950dSBarry Smith PetscFunctionBegin; 682fde5950dSBarry Smith ierr = PetscOptionsGetViewer(PetscObjectComm((PetscObject)snes),((PetscObject)snes)->prefix,name,&viewer,&format,&flg);CHKERRQ(ierr); 683fde5950dSBarry Smith if (flg) { 684d43b4f6eSBarry Smith PetscViewerAndFormat *vf; 685d43b4f6eSBarry Smith ierr = PetscViewerAndFormatCreate(viewer,format,&vf);CHKERRQ(ierr); 686d43b4f6eSBarry Smith ierr = PetscObjectDereference((PetscObject)viewer);CHKERRQ(ierr); 687fde5950dSBarry Smith if (monitorsetup) { 688d43b4f6eSBarry Smith ierr = (*monitorsetup)(snes,vf);CHKERRQ(ierr); 689fde5950dSBarry Smith } 690d43b4f6eSBarry Smith ierr = SNESMonitorSet(snes,(PetscErrorCode (*)(SNES,PetscInt,PetscReal,void*))monitor,vf,(PetscErrorCode (*)(void**))PetscViewerAndFormatDestroy);CHKERRQ(ierr); 691fde5950dSBarry Smith } 692fde5950dSBarry Smith PetscFunctionReturn(0); 693fde5950dSBarry Smith } 694fde5950dSBarry Smith 6959b94acceSBarry Smith /*@ 69694b7f48cSBarry Smith SNESSetFromOptions - Sets various SNES and KSP parameters from user options. 6979b94acceSBarry Smith 698c7afd0dbSLois Curfman McInnes Collective on SNES 699c7afd0dbSLois Curfman McInnes 7009b94acceSBarry Smith Input Parameter: 7019b94acceSBarry Smith . snes - the SNES context 7029b94acceSBarry Smith 70336851e7fSLois Curfman McInnes Options Database Keys: 704722329fbSBarry Smith + -snes_type <type> - newtonls, newtontr, ngmres, ncg, nrichardson, qn, vi, fas, SNESType for complete list 70582738288SBarry Smith . -snes_stol - convergence tolerance in terms of the norm 70682738288SBarry Smith of the change in the solution between steps 70770441072SBarry Smith . -snes_atol <abstol> - absolute tolerance of residual norm 708b39c3a46SLois Curfman McInnes . -snes_rtol <rtol> - relative decrease in tolerance norm from initial 709e4d06f11SPatrick Farrell . -snes_divergence_tolerance <divtol> - if the residual goes above divtol*rnorm0, exit with divergence 710be5caee7SBarry Smith . -snes_force_iteration <force> - force SNESSolve() to take at least one iteration 711b39c3a46SLois Curfman McInnes . -snes_max_it <max_it> - maximum number of iterations 712b39c3a46SLois Curfman McInnes . -snes_max_funcs <max_funcs> - maximum number of function evaluations 7134839bfe8SBarry Smith . -snes_max_fail <max_fail> - maximum number of line search failures allowed before stopping, default is none 714ddf469c8SBarry Smith . -snes_max_linear_solve_fail - number of linear solver failures before SNESSolve() stops 715a8054027SBarry Smith . -snes_lag_preconditioner <lag> - how often preconditioner is rebuilt (use -1 to never rebuild) 716e35cf81dSBarry Smith . -snes_lag_jacobian <lag> - how often Jacobian is rebuilt (use -1 to never rebuild) 717b39c3a46SLois Curfman McInnes . -snes_trtol <trtol> - trust region tolerance 7182492ecdbSBarry Smith . -snes_no_convergence_test - skip convergence test in nonlinear 71982738288SBarry Smith solver; hence iterations will continue until max_it 7201fbbfb26SLois Curfman McInnes or some other criterion is reached. Saves expense 72182738288SBarry Smith of convergence test 722fde5950dSBarry Smith . -snes_monitor [ascii][:filename][:viewer format] - prints residual norm at each iteration. if no filename given prints to stdout 723fde5950dSBarry Smith . -snes_monitor_solution [ascii binary draw][:filename][:viewer format] - plots solution at each iteration 724fde5950dSBarry Smith . -snes_monitor_residual [ascii binary draw][:filename][:viewer format] - plots residual (not its norm) at each iteration 725fde5950dSBarry Smith . -snes_monitor_solution_update [ascii binary draw][:filename][:viewer format] - plots update to solution at each iteration 7264619e776SBarry Smith . -snes_monitor_lg_residualnorm - plots residual norm at each iteration 727459f5d12SBarry Smith . -snes_monitor_lg_range - plots residual norm at each iteration 728e24b481bSBarry Smith . -snes_fd - use finite differences to compute Jacobian; very slow, only for testing 729e2e60de9SPeter Brune . -snes_fd_color - use finite differences with coloring to compute Jacobian 7305968eb51SBarry Smith . -snes_mf_ksp_monitor - if using matrix-free multiply then print h at each KSP iteration 731fee2055bSBarry Smith - -snes_converged_reason - print the reason for convergence/divergence after each solve 73282738288SBarry Smith 73382738288SBarry Smith Options Database for Eisenstat-Walker method: 734fa9f3622SBarry Smith + -snes_ksp_ew - use Eisenstat-Walker method for determining linear system convergence 7354b27c08aSLois Curfman McInnes . -snes_ksp_ew_version ver - version of Eisenstat-Walker method 73636851e7fSLois Curfman McInnes . -snes_ksp_ew_rtol0 <rtol0> - Sets rtol0 73736851e7fSLois Curfman McInnes . -snes_ksp_ew_rtolmax <rtolmax> - Sets rtolmax 73836851e7fSLois Curfman McInnes . -snes_ksp_ew_gamma <gamma> - Sets gamma 73936851e7fSLois Curfman McInnes . -snes_ksp_ew_alpha <alpha> - Sets alpha 74036851e7fSLois Curfman McInnes . -snes_ksp_ew_alpha2 <alpha2> - Sets alpha2 74136851e7fSLois Curfman McInnes - -snes_ksp_ew_threshold <threshold> - Sets threshold 74282738288SBarry Smith 74311ca99fdSLois Curfman McInnes Notes: 74411ca99fdSLois Curfman McInnes To see all options, run your program with the -help option or consult 745a7f22e61SSatish Balay Users-Manual: ch_snes 74683e2fdc7SBarry Smith 74736851e7fSLois Curfman McInnes Level: beginner 74836851e7fSLois Curfman McInnes 7499b94acceSBarry Smith .keywords: SNES, nonlinear, set, options, database 7509b94acceSBarry Smith 75169ed319cSSatish Balay .seealso: SNESSetOptionsPrefix() 7529b94acceSBarry Smith @*/ 7537087cfbeSBarry Smith PetscErrorCode SNESSetFromOptions(SNES snes) 7549b94acceSBarry Smith { 7558afaa268SBarry Smith PetscBool flg,pcset,persist,set; 756d8f46077SPeter Brune PetscInt i,indx,lag,grids; 75704d7464bSBarry Smith const char *deft = SNESNEWTONLS; 75885385478SLisandro Dalcin const char *convtests[] = {"default","skip"}; 75985385478SLisandro Dalcin SNESKSPEW *kctx = NULL; 760e8105e01SRichard Katz char type[256], monfilename[PETSC_MAX_PATH_LEN]; 76185385478SLisandro Dalcin PetscErrorCode ierr; 762c40d0f55SPeter Brune PCSide pcside; 763a64e098fSPeter Brune const char *optionsprefix; 7649b94acceSBarry Smith 7653a40ed3dSBarry Smith PetscFunctionBegin; 7660700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 7670f51fdf8SToby Isaac ierr = SNESRegisterAll();CHKERRQ(ierr); 7683194b578SJed Brown ierr = PetscObjectOptionsBegin((PetscObject)snes);CHKERRQ(ierr); 769639ff905SBarry Smith if (((PetscObject)snes)->type_name) deft = ((PetscObject)snes)->type_name; 770a264d7a6SBarry Smith ierr = PetscOptionsFList("-snes_type","Nonlinear solver method","SNESSetType",SNESList,deft,type,256,&flg);CHKERRQ(ierr); 771d64ed03dSBarry Smith if (flg) { 772186905e3SBarry Smith ierr = SNESSetType(snes,type);CHKERRQ(ierr); 7737adad957SLisandro Dalcin } else if (!((PetscObject)snes)->type_name) { 774186905e3SBarry Smith ierr = SNESSetType(snes,deft);CHKERRQ(ierr); 775d64ed03dSBarry Smith } 77694ae4db5SBarry Smith ierr = PetscOptionsReal("-snes_stol","Stop if step length less than","SNESSetTolerances",snes->stol,&snes->stol,NULL);CHKERRQ(ierr); 77794ae4db5SBarry Smith ierr = PetscOptionsReal("-snes_atol","Stop if function norm less than","SNESSetTolerances",snes->abstol,&snes->abstol,NULL);CHKERRQ(ierr); 778186905e3SBarry Smith 77994ae4db5SBarry Smith ierr = PetscOptionsReal("-snes_rtol","Stop if decrease in function norm less than","SNESSetTolerances",snes->rtol,&snes->rtol,NULL);CHKERRQ(ierr); 780e4d06f11SPatrick Farrell ierr = PetscOptionsReal("-snes_divergence_tolerance","Stop if residual norm increases by this factor","SNESSetDivergenceTolerance",snes->divtol,&snes->divtol,NULL);CHKERRQ(ierr); 7810298fd71SBarry Smith ierr = PetscOptionsInt("-snes_max_it","Maximum iterations","SNESSetTolerances",snes->max_its,&snes->max_its,NULL);CHKERRQ(ierr); 7820298fd71SBarry Smith ierr = PetscOptionsInt("-snes_max_funcs","Maximum function evaluations","SNESSetTolerances",snes->max_funcs,&snes->max_funcs,NULL);CHKERRQ(ierr); 7830298fd71SBarry Smith ierr = PetscOptionsInt("-snes_max_fail","Maximum nonlinear step failures","SNESSetMaxNonlinearStepFailures",snes->maxFailures,&snes->maxFailures,NULL);CHKERRQ(ierr); 7840298fd71SBarry Smith ierr = PetscOptionsInt("-snes_max_linear_solve_fail","Maximum failures in linear solves allowed","SNESSetMaxLinearSolveFailures",snes->maxLinearSolveFailures,&snes->maxLinearSolveFailures,NULL);CHKERRQ(ierr); 7850298fd71SBarry Smith ierr = PetscOptionsBool("-snes_error_if_not_converged","Generate error if solver does not converge","SNESSetErrorIfNotConverged",snes->errorifnotconverged,&snes->errorifnotconverged,NULL);CHKERRQ(ierr); 786be5caee7SBarry Smith ierr = PetscOptionsBool("-snes_force_iteration","Force SNESSolve() to take at least one iteration","SNESForceIteration",snes->forceiteration,&snes->forceiteration,NULL);CHKERRQ(ierr); 78785385478SLisandro Dalcin 788a8054027SBarry Smith ierr = PetscOptionsInt("-snes_lag_preconditioner","How often to rebuild preconditioner","SNESSetLagPreconditioner",snes->lagpreconditioner,&lag,&flg);CHKERRQ(ierr); 789a8054027SBarry Smith if (flg) { 790a8054027SBarry Smith ierr = SNESSetLagPreconditioner(snes,lag);CHKERRQ(ierr); 791a8054027SBarry Smith } 79237ec4e1aSPeter Brune ierr = PetscOptionsBool("-snes_lag_preconditioner_persists","Preconditioner lagging through multiple solves","SNESSetLagPreconditionerPersists",snes->lagjac_persist,&persist,&flg);CHKERRQ(ierr); 79337ec4e1aSPeter Brune if (flg) { 79437ec4e1aSPeter Brune ierr = SNESSetLagPreconditionerPersists(snes,persist);CHKERRQ(ierr); 79537ec4e1aSPeter Brune } 796e35cf81dSBarry Smith ierr = PetscOptionsInt("-snes_lag_jacobian","How often to rebuild Jacobian","SNESSetLagJacobian",snes->lagjacobian,&lag,&flg);CHKERRQ(ierr); 797e35cf81dSBarry Smith if (flg) { 798e35cf81dSBarry Smith ierr = SNESSetLagJacobian(snes,lag);CHKERRQ(ierr); 799e35cf81dSBarry Smith } 80037ec4e1aSPeter Brune ierr = PetscOptionsBool("-snes_lag_jacobian_persists","Jacobian lagging through multiple solves","SNESSetLagJacobianPersists",snes->lagjac_persist,&persist,&flg);CHKERRQ(ierr); 80137ec4e1aSPeter Brune if (flg) { 80237ec4e1aSPeter Brune ierr = SNESSetLagJacobianPersists(snes,persist);CHKERRQ(ierr); 80337ec4e1aSPeter Brune } 80437ec4e1aSPeter Brune 805efd51863SBarry Smith ierr = PetscOptionsInt("-snes_grid_sequence","Use grid sequencing to generate initial guess","SNESSetGridSequence",snes->gridsequence,&grids,&flg);CHKERRQ(ierr); 806efd51863SBarry Smith if (flg) { 807efd51863SBarry Smith ierr = SNESSetGridSequence(snes,grids);CHKERRQ(ierr); 808efd51863SBarry Smith } 809a8054027SBarry Smith 81085385478SLisandro Dalcin ierr = PetscOptionsEList("-snes_convergence_test","Convergence test","SNESSetConvergenceTest",convtests,2,"default",&indx,&flg);CHKERRQ(ierr); 81185385478SLisandro Dalcin if (flg) { 81285385478SLisandro Dalcin switch (indx) { 8138d359177SBarry Smith case 0: ierr = SNESSetConvergenceTest(snes,SNESConvergedDefault,NULL,NULL);CHKERRQ(ierr); break; 814e2a6519dSDmitry Karpeev case 1: ierr = SNESSetConvergenceTest(snes,SNESConvergedSkip,NULL,NULL);CHKERRQ(ierr); break; 81585385478SLisandro Dalcin } 81685385478SLisandro Dalcin } 81785385478SLisandro Dalcin 818365a6726SPeter Brune ierr = PetscOptionsEList("-snes_norm_schedule","SNES Norm schedule","SNESSetNormSchedule",SNESNormSchedules,5,"function",&indx,&flg);CHKERRQ(ierr); 819365a6726SPeter Brune if (flg) { ierr = SNESSetNormSchedule(snes,(SNESNormSchedule)indx);CHKERRQ(ierr); } 820fdacfa88SPeter Brune 82147073ea2SPeter Brune ierr = PetscOptionsEList("-snes_function_type","SNES Norm schedule","SNESSetFunctionType",SNESFunctionTypes,2,"unpreconditioned",&indx,&flg);CHKERRQ(ierr); 82247073ea2SPeter Brune if (flg) { ierr = SNESSetFunctionType(snes,(SNESFunctionType)indx);CHKERRQ(ierr); } 823186905e3SBarry Smith 82485385478SLisandro Dalcin kctx = (SNESKSPEW*)snes->kspconvctx; 82585385478SLisandro Dalcin 8260298fd71SBarry Smith ierr = PetscOptionsBool("-snes_ksp_ew","Use Eisentat-Walker linear system convergence test","SNESKSPSetUseEW",snes->ksp_ewconv,&snes->ksp_ewconv,NULL);CHKERRQ(ierr); 827186905e3SBarry Smith 82894ae4db5SBarry Smith ierr = PetscOptionsInt("-snes_ksp_ew_version","Version 1, 2 or 3","SNESKSPSetParametersEW",kctx->version,&kctx->version,NULL);CHKERRQ(ierr); 82994ae4db5SBarry Smith ierr = PetscOptionsReal("-snes_ksp_ew_rtol0","0 <= rtol0 < 1","SNESKSPSetParametersEW",kctx->rtol_0,&kctx->rtol_0,NULL);CHKERRQ(ierr); 83094ae4db5SBarry Smith ierr = PetscOptionsReal("-snes_ksp_ew_rtolmax","0 <= rtolmax < 1","SNESKSPSetParametersEW",kctx->rtol_max,&kctx->rtol_max,NULL);CHKERRQ(ierr); 83194ae4db5SBarry Smith ierr = PetscOptionsReal("-snes_ksp_ew_gamma","0 <= gamma <= 1","SNESKSPSetParametersEW",kctx->gamma,&kctx->gamma,NULL);CHKERRQ(ierr); 83294ae4db5SBarry Smith ierr = PetscOptionsReal("-snes_ksp_ew_alpha","1 < alpha <= 2","SNESKSPSetParametersEW",kctx->alpha,&kctx->alpha,NULL);CHKERRQ(ierr); 83394ae4db5SBarry Smith ierr = PetscOptionsReal("-snes_ksp_ew_alpha2","alpha2","SNESKSPSetParametersEW",kctx->alpha2,&kctx->alpha2,NULL);CHKERRQ(ierr); 83494ae4db5SBarry Smith ierr = PetscOptionsReal("-snes_ksp_ew_threshold","0 < threshold < 1","SNESKSPSetParametersEW",kctx->threshold,&kctx->threshold,NULL);CHKERRQ(ierr); 835186905e3SBarry Smith 83690d69ab7SBarry Smith flg = PETSC_FALSE; 8378afaa268SBarry Smith ierr = PetscOptionsBool("-snes_check_jacobian","Check each Jacobian with a differenced one","SNESUpdateCheckJacobian",flg,&flg,&set);CHKERRQ(ierr); 8388afaa268SBarry Smith if (set && flg) { 8395341784dSBarry Smith ierr = SNESSetUpdate(snes,SNESUpdateCheckJacobian);CHKERRQ(ierr); 8405341784dSBarry Smith } 8415341784dSBarry Smith 8425341784dSBarry Smith flg = PETSC_FALSE; 8438afaa268SBarry Smith ierr = PetscOptionsBool("-snes_monitor_cancel","Remove all monitors","SNESMonitorCancel",flg,&flg,&set);CHKERRQ(ierr); 8448afaa268SBarry Smith if (set && flg) {ierr = SNESMonitorCancel(snes);CHKERRQ(ierr);} 845eabae89aSBarry Smith 846fde5950dSBarry Smith ierr = SNESMonitorSetFromOptions(snes,"-snes_monitor","Monitor norm of function","SNESMonitorDefault",SNESMonitorDefault,NULL);CHKERRQ(ierr); 847fde5950dSBarry Smith ierr = SNESMonitorSetFromOptions(snes,"-snes_monitor_short","Monitor norm of function with fewer digits","SNESMonitorDefaultShort",SNESMonitorDefaultShort,NULL);CHKERRQ(ierr); 848fde5950dSBarry Smith ierr = SNESMonitorSetFromOptions(snes,"-snes_monitor_range","Monitor range of elements of function","SNESMonitorRange",SNESMonitorRange,NULL);CHKERRQ(ierr); 849eabae89aSBarry Smith 850fde5950dSBarry Smith ierr = SNESMonitorSetFromOptions(snes,"-snes_monitor_ratio","Monitor ratios of the norm of function for consecutive steps","SNESMonitorRatio",SNESMonitorRatio,SNESMonitorRatioSetUp);CHKERRQ(ierr); 851fde5950dSBarry Smith ierr = SNESMonitorSetFromOptions(snes,"-snes_monitor_field","Monitor norm of function (split into fields)","SNESMonitorDefaultField",SNESMonitorDefaultField,NULL);CHKERRQ(ierr); 852fde5950dSBarry Smith ierr = SNESMonitorSetFromOptions(snes,"-snes_monitor_solution","View solution at each iteration","SNESMonitorSolution",SNESMonitorSolution,NULL);CHKERRQ(ierr); 853fde5950dSBarry Smith ierr = SNESMonitorSetFromOptions(snes,"-snes_monitor_solution_update","View correction at each iteration","SNESMonitorSolutionUpdate",SNESMonitorSolutionUpdate,NULL);CHKERRQ(ierr); 854fde5950dSBarry Smith ierr = SNESMonitorSetFromOptions(snes,"-snes_monitor_residual","View residual at each iteration","SNESMonitorResidual",SNESMonitorResidual,NULL);CHKERRQ(ierr); 855fde5950dSBarry Smith ierr = SNESMonitorSetFromOptions(snes,"-snes_monitor_jacupdate_spectrum","Print the change in the spectrum of the Jacobian","SNESMonitorJacUpdateSpectrum",SNESMonitorJacUpdateSpectrum,NULL);CHKERRQ(ierr); 856fde5950dSBarry Smith ierr = SNESMonitorSetFromOptions(snes,"-snes_monitor_fields","Monitor norm of function per field","SNESMonitorSet",SNESMonitorFields,NULL);CHKERRQ(ierr); 8572db13446SMatthew G. Knepley 8585180491cSLisandro Dalcin ierr = PetscOptionsString("-snes_monitor_python","Use Python function","SNESMonitorSet",0,monfilename,PETSC_MAX_PATH_LEN,&flg);CHKERRQ(ierr); 8595180491cSLisandro Dalcin if (flg) {ierr = PetscPythonMonitorSet((PetscObject)snes,monfilename);CHKERRQ(ierr);} 8605180491cSLisandro Dalcin 861fde5950dSBarry Smith 86290d69ab7SBarry Smith flg = PETSC_FALSE; 8630298fd71SBarry Smith ierr = PetscOptionsBool("-snes_monitor_lg_residualnorm","Plot function norm at each iteration","SNESMonitorLGResidualNorm",flg,&flg,NULL);CHKERRQ(ierr); 864459f5d12SBarry Smith if (flg) { 865d96771aaSLisandro Dalcin PetscDrawLG ctx; 866459f5d12SBarry Smith 8676ba87a44SLisandro Dalcin ierr = SNESMonitorLGCreate(PetscObjectComm((PetscObject)snes),NULL,NULL,PETSC_DECIDE,PETSC_DECIDE,400,300,&ctx);CHKERRQ(ierr); 868d96771aaSLisandro Dalcin ierr = SNESMonitorSet(snes,SNESMonitorLGResidualNorm,ctx,(PetscErrorCode (*)(void**))PetscDrawLGDestroy);CHKERRQ(ierr); 869459f5d12SBarry Smith } 87090d69ab7SBarry Smith flg = PETSC_FALSE; 8710298fd71SBarry Smith ierr = PetscOptionsBool("-snes_monitor_lg_range","Plot function range at each iteration","SNESMonitorLGRange",flg,&flg,NULL);CHKERRQ(ierr); 872459f5d12SBarry Smith if (flg) { 873459f5d12SBarry Smith PetscViewer ctx; 874e24b481bSBarry Smith 8756ba87a44SLisandro Dalcin ierr = PetscViewerDrawOpen(PetscObjectComm((PetscObject)snes),NULL,NULL,PETSC_DECIDE,PETSC_DECIDE,400,300,&ctx);CHKERRQ(ierr); 876459f5d12SBarry Smith ierr = SNESMonitorSet(snes,SNESMonitorLGRange,ctx,(PetscErrorCode (*)(void**))PetscViewerDestroy);CHKERRQ(ierr); 877459f5d12SBarry Smith } 8782e7541e6SPeter Brune 8792e7541e6SPeter Brune 880cc0c4584SMatthew G. Knepley 88190d69ab7SBarry Smith flg = PETSC_FALSE; 8828d359177SBarry Smith ierr = PetscOptionsBool("-snes_fd","Use finite differences (slow) to compute Jacobian","SNESComputeJacobianDefault",flg,&flg,NULL);CHKERRQ(ierr); 8834b27c08aSLois Curfman McInnes if (flg) { 8846cab3a1bSJed Brown void *functx; 885b1f624c7SBarry Smith DM dm; 886b1f624c7SBarry Smith DMSNES sdm; 887b1f624c7SBarry Smith ierr = SNESGetDM(snes,&dm);CHKERRQ(ierr); 888b1f624c7SBarry Smith ierr = DMGetDMSNES(dm,&sdm);CHKERRQ(ierr); 889b1f624c7SBarry Smith sdm->jacobianctx = NULL; 8900298fd71SBarry Smith ierr = SNESGetFunction(snes,NULL,NULL,&functx);CHKERRQ(ierr); 8918d359177SBarry Smith ierr = SNESSetJacobian(snes,snes->jacobian,snes->jacobian_pre,SNESComputeJacobianDefault,functx);CHKERRQ(ierr); 892ae15b995SBarry Smith ierr = PetscInfo(snes,"Setting default finite difference Jacobian matrix\n");CHKERRQ(ierr); 8939b94acceSBarry Smith } 894639f9d9dSBarry Smith 89544848bc4SPeter Brune flg = PETSC_FALSE; 8968d359177SBarry Smith ierr = PetscOptionsBool("-snes_fd_function","Use finite differences (slow) to compute function from user objective","SNESObjectiveComputeFunctionDefaultFD",flg,&flg,NULL);CHKERRQ(ierr); 89797584545SPeter Brune if (flg) { 8988d359177SBarry Smith ierr = SNESSetFunction(snes,NULL,SNESObjectiveComputeFunctionDefaultFD,NULL);CHKERRQ(ierr); 89997584545SPeter Brune } 90097584545SPeter Brune 90197584545SPeter Brune flg = PETSC_FALSE; 9028d359177SBarry Smith ierr = PetscOptionsBool("-snes_fd_color","Use finite differences with coloring to compute Jacobian","SNESComputeJacobianDefaultColor",flg,&flg,NULL);CHKERRQ(ierr); 90344848bc4SPeter Brune if (flg) { 904c52e227fSPeter Brune DM dm; 905c52e227fSPeter Brune DMSNES sdm; 906c52e227fSPeter Brune ierr = SNESGetDM(snes,&dm);CHKERRQ(ierr); 907aace71b7SPeter Brune ierr = DMGetDMSNES(dm,&sdm);CHKERRQ(ierr); 908aace71b7SPeter Brune sdm->jacobianctx = NULL; 9098d359177SBarry Smith ierr = SNESSetJacobian(snes,snes->jacobian,snes->jacobian_pre,SNESComputeJacobianDefaultColor,0);CHKERRQ(ierr); 91044848bc4SPeter Brune ierr = PetscInfo(snes,"Setting default finite difference coloring Jacobian matrix\n");CHKERRQ(ierr); 91144848bc4SPeter Brune } 91244848bc4SPeter Brune 913aa3661deSLisandro Dalcin flg = PETSC_FALSE; 914f871313dSBarry 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); 915d8f46077SPeter Brune if (flg && snes->mf_operator) { 916a8248277SBarry Smith snes->mf_operator = PETSC_TRUE; 917d8f46077SPeter Brune snes->mf = PETSC_TRUE; 918a8248277SBarry Smith } 919aa3661deSLisandro Dalcin flg = PETSC_FALSE; 920f871313dSBarry Smith ierr = PetscOptionsBool("-snes_mf","Use a Matrix-Free Jacobian with no preconditioner matrix","SNESSetUseMatrixFree",PETSC_FALSE,&snes->mf,&flg);CHKERRQ(ierr); 921d8f46077SPeter Brune if (!flg && snes->mf_operator) snes->mf = PETSC_TRUE; 922d8f46077SPeter Brune ierr = PetscOptionsInt("-snes_mf_version","Matrix-Free routines version 1 or 2","None",snes->mf_version,&snes->mf_version,0);CHKERRQ(ierr); 923d28543b3SPeter Brune 924c40d0f55SPeter Brune flg = PETSC_FALSE; 925be95d8f1SBarry Smith ierr = SNESGetNPCSide(snes,&pcside);CHKERRQ(ierr); 926be95d8f1SBarry Smith ierr = PetscOptionsEnum("-snes_npc_side","SNES nonlinear preconditioner side","SNESSetNPCSide",PCSides,(PetscEnum)pcside,(PetscEnum*)&pcside,&flg);CHKERRQ(ierr); 927be95d8f1SBarry Smith if (flg) {ierr = SNESSetNPCSide(snes,pcside);CHKERRQ(ierr);} 928c40d0f55SPeter Brune 929e04113cfSBarry Smith #if defined(PETSC_HAVE_SAWS) 9308a70d858SHong Zhang /* 9318a70d858SHong Zhang Publish convergence information using SAWs 9328a70d858SHong Zhang */ 9338a70d858SHong Zhang flg = PETSC_FALSE; 9348a70d858SHong Zhang ierr = PetscOptionsBool("-snes_monitor_saws","Publish SNES progress using SAWs","SNESMonitorSet",flg,&flg,NULL);CHKERRQ(ierr); 9358a70d858SHong Zhang if (flg) { 9368a70d858SHong Zhang void *ctx; 9378a70d858SHong Zhang ierr = SNESMonitorSAWsCreate(snes,&ctx);CHKERRQ(ierr); 9388a70d858SHong Zhang ierr = SNESMonitorSet(snes,SNESMonitorSAWs,ctx,SNESMonitorSAWsDestroy);CHKERRQ(ierr); 9398a70d858SHong Zhang } 9408a70d858SHong Zhang #endif 9418a70d858SHong Zhang #if defined(PETSC_HAVE_SAWS) 942b90c6cbeSBarry Smith { 943b90c6cbeSBarry Smith PetscBool set; 944b90c6cbeSBarry Smith flg = PETSC_FALSE; 9458a70d858SHong Zhang ierr = PetscOptionsBool("-snes_saws_block","Block for SAWs at end of SNESSolve","PetscObjectSAWsBlock",((PetscObject)snes)->amspublishblock,&flg,&set);CHKERRQ(ierr); 946b90c6cbeSBarry Smith if (set) { 947e04113cfSBarry Smith ierr = PetscObjectSAWsSetBlock((PetscObject)snes,flg);CHKERRQ(ierr); 948b90c6cbeSBarry Smith } 949b90c6cbeSBarry Smith } 950b90c6cbeSBarry Smith #endif 951b90c6cbeSBarry Smith 95276b2cf59SMatthew Knepley for (i = 0; i < numberofsetfromoptions; i++) { 95376b2cf59SMatthew Knepley ierr = (*othersetfromoptions[i])(snes);CHKERRQ(ierr); 95476b2cf59SMatthew Knepley } 95576b2cf59SMatthew Knepley 956e7788613SBarry Smith if (snes->ops->setfromoptions) { 957e55864a3SBarry Smith ierr = (*snes->ops->setfromoptions)(PetscOptionsObject,snes);CHKERRQ(ierr); 958639f9d9dSBarry Smith } 9595d973c19SBarry Smith 9605d973c19SBarry Smith /* process any options handlers added with PetscObjectAddOptionsHandler() */ 9610633abcbSJed Brown ierr = PetscObjectProcessOptionsHandlers(PetscOptionsObject,(PetscObject)snes);CHKERRQ(ierr); 962b0a32e0cSBarry Smith ierr = PetscOptionsEnd();CHKERRQ(ierr); 9634bbc92c1SBarry Smith 9649e764e56SPeter Brune if (!snes->linesearch) { 9657601faf0SJed Brown ierr = SNESGetLineSearch(snes, &snes->linesearch);CHKERRQ(ierr); 9669e764e56SPeter Brune } 967f1c6b773SPeter Brune ierr = SNESLineSearchSetFromOptions(snes->linesearch);CHKERRQ(ierr); 9689e764e56SPeter Brune 9696991f827SBarry Smith if (!snes->ksp) {ierr = SNESGetKSP(snes,&snes->ksp);CHKERRQ(ierr);} 9706991f827SBarry Smith ierr = KSPSetOperators(snes->ksp,snes->jacobian,snes->jacobian_pre);CHKERRQ(ierr); 9716991f827SBarry Smith ierr = KSPSetFromOptions(snes->ksp);CHKERRQ(ierr); 9726991f827SBarry Smith 973be95d8f1SBarry Smith /* if someone has set the SNES NPC type, create it. */ 97451e86f29SPeter Brune ierr = SNESGetOptionsPrefix(snes, &optionsprefix);CHKERRQ(ierr); 975c5929fdfSBarry Smith ierr = PetscOptionsHasName(((PetscObject)snes)->options,optionsprefix, "-npc_snes_type", &pcset);CHKERRQ(ierr); 976efd4aadfSBarry Smith if (pcset && (!snes->npc)) { 977efd4aadfSBarry Smith ierr = SNESGetNPC(snes, &snes->npc);CHKERRQ(ierr); 97851e86f29SPeter Brune } 9793a40ed3dSBarry Smith PetscFunctionReturn(0); 9809b94acceSBarry Smith } 9819b94acceSBarry Smith 982bb9467b5SJed Brown /*@C 983d25893d9SBarry Smith SNESSetComputeApplicationContext - Sets an optional function to compute a user-defined context for 984d25893d9SBarry Smith the nonlinear solvers. 985d25893d9SBarry Smith 986d25893d9SBarry Smith Logically Collective on SNES 987d25893d9SBarry Smith 988d25893d9SBarry Smith Input Parameters: 989d25893d9SBarry Smith + snes - the SNES context 990d25893d9SBarry Smith . compute - function to compute the context 991d25893d9SBarry Smith - destroy - function to destroy the context 992d25893d9SBarry Smith 993d25893d9SBarry Smith Level: intermediate 994d25893d9SBarry Smith 995bb9467b5SJed Brown Notes: 996bb9467b5SJed Brown This function is currently not available from Fortran. 997bb9467b5SJed Brown 998d25893d9SBarry Smith .keywords: SNES, nonlinear, set, application, context 999d25893d9SBarry Smith 1000d25893d9SBarry Smith .seealso: SNESGetApplicationContext(), SNESSetComputeApplicationContext(), SNESGetApplicationContext() 1001d25893d9SBarry Smith @*/ 1002d25893d9SBarry Smith PetscErrorCode SNESSetComputeApplicationContext(SNES snes,PetscErrorCode (*compute)(SNES,void**),PetscErrorCode (*destroy)(void**)) 1003d25893d9SBarry Smith { 1004d25893d9SBarry Smith PetscFunctionBegin; 1005d25893d9SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 1006d25893d9SBarry Smith snes->ops->usercompute = compute; 1007d25893d9SBarry Smith snes->ops->userdestroy = destroy; 1008d25893d9SBarry Smith PetscFunctionReturn(0); 1009d25893d9SBarry Smith } 1010a847f771SSatish Balay 1011b07ff414SBarry Smith /*@ 10129b94acceSBarry Smith SNESSetApplicationContext - Sets the optional user-defined context for 10139b94acceSBarry Smith the nonlinear solvers. 10149b94acceSBarry Smith 10153f9fe445SBarry Smith Logically Collective on SNES 1016fee21e36SBarry Smith 1017c7afd0dbSLois Curfman McInnes Input Parameters: 1018c7afd0dbSLois Curfman McInnes + snes - the SNES context 1019c7afd0dbSLois Curfman McInnes - usrP - optional user context 1020c7afd0dbSLois Curfman McInnes 102136851e7fSLois Curfman McInnes Level: intermediate 102236851e7fSLois Curfman McInnes 1023daf670e6SBarry Smith Fortran Notes: To use this from Fortran you must write a Fortran interface definition for this 1024daf670e6SBarry Smith function that tells Fortran the Fortran derived data type that you are passing in as the ctx argument. 1025daf670e6SBarry Smith 10269b94acceSBarry Smith .keywords: SNES, nonlinear, set, application, context 10279b94acceSBarry Smith 1028ecaffddaSVictor Eijkhout .seealso: SNESGetApplicationContext() 10299b94acceSBarry Smith @*/ 10307087cfbeSBarry Smith PetscErrorCode SNESSetApplicationContext(SNES snes,void *usrP) 10319b94acceSBarry Smith { 10321b2093e4SBarry Smith PetscErrorCode ierr; 1033b07ff414SBarry Smith KSP ksp; 10341b2093e4SBarry Smith 10353a40ed3dSBarry Smith PetscFunctionBegin; 10360700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 1037b07ff414SBarry Smith ierr = SNESGetKSP(snes,&ksp);CHKERRQ(ierr); 1038b07ff414SBarry Smith ierr = KSPSetApplicationContext(ksp,usrP);CHKERRQ(ierr); 10399b94acceSBarry Smith snes->user = usrP; 10403a40ed3dSBarry Smith PetscFunctionReturn(0); 10419b94acceSBarry Smith } 104274679c65SBarry Smith 1043b07ff414SBarry Smith /*@ 10449b94acceSBarry Smith SNESGetApplicationContext - Gets the user-defined context for the 10459b94acceSBarry Smith nonlinear solvers. 10469b94acceSBarry Smith 1047c7afd0dbSLois Curfman McInnes Not Collective 1048c7afd0dbSLois Curfman McInnes 10499b94acceSBarry Smith Input Parameter: 10509b94acceSBarry Smith . snes - SNES context 10519b94acceSBarry Smith 10529b94acceSBarry Smith Output Parameter: 10539b94acceSBarry Smith . usrP - user context 10549b94acceSBarry Smith 1055daf670e6SBarry Smith Fortran Notes: To use this from Fortran you must write a Fortran interface definition for this 1056daf670e6SBarry Smith function that tells Fortran the Fortran derived data type that you are passing in as the ctx argument. 1057daf670e6SBarry Smith 105836851e7fSLois Curfman McInnes Level: intermediate 105936851e7fSLois Curfman McInnes 10609b94acceSBarry Smith .keywords: SNES, nonlinear, get, application, context 10619b94acceSBarry Smith 10629b94acceSBarry Smith .seealso: SNESSetApplicationContext() 10639b94acceSBarry Smith @*/ 1064e71120c6SJed Brown PetscErrorCode SNESGetApplicationContext(SNES snes,void *usrP) 10659b94acceSBarry Smith { 10663a40ed3dSBarry Smith PetscFunctionBegin; 10670700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 1068e71120c6SJed Brown *(void**)usrP = snes->user; 10693a40ed3dSBarry Smith PetscFunctionReturn(0); 10709b94acceSBarry Smith } 107174679c65SBarry Smith 10729b94acceSBarry Smith /*@ 10733565c898SBarry Smith SNESSetUseMatrixFree - indicates that SNES should use matrix free finite difference matrix vector products internally to apply 10743565c898SBarry Smith the Jacobian. 10753565c898SBarry Smith 10763565c898SBarry Smith Collective on SNES 10773565c898SBarry Smith 10783565c898SBarry Smith Input Parameters: 10793565c898SBarry Smith + snes - SNES context 10803565c898SBarry 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 10813565c898SBarry Smith - mf_operator - use matrix-free only for the Amat used by SNESSetJacobian(), this means the user provided Pmat will continue to be used 10823565c898SBarry Smith 10833565c898SBarry Smith Options Database: 10843565c898SBarry Smith + -snes_mf - use matrix free for both the mat and pmat operator 10853565c898SBarry Smith - -snes_mf_operator - use matrix free only for the mat operator 10863565c898SBarry Smith 10873565c898SBarry Smith Level: intermediate 10883565c898SBarry Smith 10893565c898SBarry Smith .keywords: SNES, nonlinear, get, iteration, number, 10903565c898SBarry Smith 10913565c898SBarry Smith .seealso: SNESGetUseMatrixFree(), MatCreateSNESMF() 10923565c898SBarry Smith @*/ 10933565c898SBarry Smith PetscErrorCode SNESSetUseMatrixFree(SNES snes,PetscBool mf_operator,PetscBool mf) 10943565c898SBarry Smith { 10953565c898SBarry Smith PetscFunctionBegin; 10963565c898SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 10973565c898SBarry Smith if (mf && !mf_operator) SETERRQ(PetscObjectComm((PetscObject)snes),PETSC_ERR_ARG_INCOMP,"If using mf must also use mf_operator"); 10983565c898SBarry Smith snes->mf = mf; 10993565c898SBarry Smith snes->mf_operator = mf_operator; 11003565c898SBarry Smith PetscFunctionReturn(0); 11013565c898SBarry Smith } 11023565c898SBarry Smith 11033565c898SBarry Smith /*@ 11043565c898SBarry Smith SNESGetUseMatrixFree - indicates if the SNES uses matrix free finite difference matrix vector products to apply 11053565c898SBarry Smith the Jacobian. 11063565c898SBarry Smith 11073565c898SBarry Smith Collective on SNES 11083565c898SBarry Smith 11093565c898SBarry Smith Input Parameter: 11103565c898SBarry Smith . snes - SNES context 11113565c898SBarry Smith 11123565c898SBarry Smith Output Parameters: 11133565c898SBarry 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 11143565c898SBarry Smith - mf_operator - use matrix-free only for the Amat used by SNESSetJacobian(), this means the user provided Pmat will continue to be used 11153565c898SBarry Smith 11163565c898SBarry Smith Options Database: 11173565c898SBarry Smith + -snes_mf - use matrix free for both the mat and pmat operator 11183565c898SBarry Smith - -snes_mf_operator - use matrix free only for the mat operator 11193565c898SBarry Smith 11203565c898SBarry Smith Level: intermediate 11213565c898SBarry Smith 11223565c898SBarry Smith .keywords: SNES, nonlinear, get, iteration, number, 11233565c898SBarry Smith 11243565c898SBarry Smith .seealso: SNESSetUseMatrixFree(), MatCreateSNESMF() 11253565c898SBarry Smith @*/ 11263565c898SBarry Smith PetscErrorCode SNESGetUseMatrixFree(SNES snes,PetscBool *mf_operator,PetscBool *mf) 11273565c898SBarry Smith { 11283565c898SBarry Smith PetscFunctionBegin; 11293565c898SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 11303565c898SBarry Smith if (mf) *mf = snes->mf; 11313565c898SBarry Smith if (mf_operator) *mf_operator = snes->mf_operator; 11323565c898SBarry Smith PetscFunctionReturn(0); 11333565c898SBarry Smith } 11343565c898SBarry Smith 11353565c898SBarry Smith /*@ 1136c8228a4eSBarry Smith SNESGetIterationNumber - Gets the number of nonlinear iterations completed 1137c8228a4eSBarry Smith at this time. 11389b94acceSBarry Smith 1139c7afd0dbSLois Curfman McInnes Not Collective 1140c7afd0dbSLois Curfman McInnes 11419b94acceSBarry Smith Input Parameter: 11429b94acceSBarry Smith . snes - SNES context 11439b94acceSBarry Smith 11449b94acceSBarry Smith Output Parameter: 11459b94acceSBarry Smith . iter - iteration number 11469b94acceSBarry Smith 1147c8228a4eSBarry Smith Notes: 1148c8228a4eSBarry Smith For example, during the computation of iteration 2 this would return 1. 1149c8228a4eSBarry Smith 1150c8228a4eSBarry Smith This is useful for using lagged Jacobians (where one does not recompute the 115108405cd6SLois Curfman McInnes Jacobian at each SNES iteration). For example, the code 115208405cd6SLois Curfman McInnes .vb 115308405cd6SLois Curfman McInnes ierr = SNESGetIterationNumber(snes,&it); 115408405cd6SLois Curfman McInnes if (!(it % 2)) { 115508405cd6SLois Curfman McInnes [compute Jacobian here] 115608405cd6SLois Curfman McInnes } 115708405cd6SLois Curfman McInnes .ve 1158c8228a4eSBarry Smith can be used in your ComputeJacobian() function to cause the Jacobian to be 115908405cd6SLois Curfman McInnes recomputed every second SNES iteration. 1160c8228a4eSBarry Smith 116136851e7fSLois Curfman McInnes Level: intermediate 116236851e7fSLois Curfman McInnes 11632b668275SBarry Smith .keywords: SNES, nonlinear, get, iteration, number, 11642b668275SBarry Smith 116571dbe336SPeter Brune .seealso: SNESGetLinearSolveIterations() 11669b94acceSBarry Smith @*/ 11677087cfbeSBarry Smith PetscErrorCode SNESGetIterationNumber(SNES snes,PetscInt *iter) 11689b94acceSBarry Smith { 11693a40ed3dSBarry Smith PetscFunctionBegin; 11700700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 11714482741eSBarry Smith PetscValidIntPointer(iter,2); 11729b94acceSBarry Smith *iter = snes->iter; 11733a40ed3dSBarry Smith PetscFunctionReturn(0); 11749b94acceSBarry Smith } 117574679c65SBarry Smith 1176360c497dSPeter Brune /*@ 1177360c497dSPeter Brune SNESSetIterationNumber - Sets the current iteration number. 1178360c497dSPeter Brune 1179360c497dSPeter Brune Not Collective 1180360c497dSPeter Brune 1181360c497dSPeter Brune Input Parameter: 1182360c497dSPeter Brune . snes - SNES context 1183360c497dSPeter Brune . iter - iteration number 1184360c497dSPeter Brune 1185360c497dSPeter Brune Level: developer 1186360c497dSPeter Brune 1187360c497dSPeter Brune .keywords: SNES, nonlinear, set, iteration, number, 1188360c497dSPeter Brune 118971dbe336SPeter Brune .seealso: SNESGetLinearSolveIterations() 1190360c497dSPeter Brune @*/ 1191360c497dSPeter Brune PetscErrorCode SNESSetIterationNumber(SNES snes,PetscInt iter) 1192360c497dSPeter Brune { 1193360c497dSPeter Brune PetscErrorCode ierr; 1194360c497dSPeter Brune 1195360c497dSPeter Brune PetscFunctionBegin; 1196360c497dSPeter Brune PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 1197e04113cfSBarry Smith ierr = PetscObjectSAWsTakeAccess((PetscObject)snes);CHKERRQ(ierr); 1198360c497dSPeter Brune snes->iter = iter; 1199e04113cfSBarry Smith ierr = PetscObjectSAWsGrantAccess((PetscObject)snes);CHKERRQ(ierr); 1200360c497dSPeter Brune PetscFunctionReturn(0); 1201360c497dSPeter Brune } 1202360c497dSPeter Brune 12039b94acceSBarry Smith /*@ 1204b850b91aSLisandro Dalcin SNESGetNonlinearStepFailures - Gets the number of unsuccessful steps 12059b94acceSBarry Smith attempted by the nonlinear solver. 12069b94acceSBarry Smith 1207c7afd0dbSLois Curfman McInnes Not Collective 1208c7afd0dbSLois Curfman McInnes 12099b94acceSBarry Smith Input Parameter: 12109b94acceSBarry Smith . snes - SNES context 12119b94acceSBarry Smith 12129b94acceSBarry Smith Output Parameter: 12139b94acceSBarry Smith . nfails - number of unsuccessful steps attempted 12149b94acceSBarry Smith 1215c96a6f78SLois Curfman McInnes Notes: 1216c96a6f78SLois Curfman McInnes This counter is reset to zero for each successive call to SNESSolve(). 1217c96a6f78SLois Curfman McInnes 121836851e7fSLois Curfman McInnes Level: intermediate 121936851e7fSLois Curfman McInnes 12209b94acceSBarry Smith .keywords: SNES, nonlinear, get, number, unsuccessful, steps 122158ebbce7SBarry Smith 1222e1c61ce8SBarry Smith .seealso: SNESGetMaxLinearSolveFailures(), SNESGetLinearSolveIterations(), SNESSetMaxLinearSolveFailures(), SNESGetLinearSolveFailures(), 122358ebbce7SBarry Smith SNESSetMaxNonlinearStepFailures(), SNESGetMaxNonlinearStepFailures() 12249b94acceSBarry Smith @*/ 12257087cfbeSBarry Smith PetscErrorCode SNESGetNonlinearStepFailures(SNES snes,PetscInt *nfails) 12269b94acceSBarry Smith { 12273a40ed3dSBarry Smith PetscFunctionBegin; 12280700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 12294482741eSBarry Smith PetscValidIntPointer(nfails,2); 123050ffb88aSMatthew Knepley *nfails = snes->numFailures; 123150ffb88aSMatthew Knepley PetscFunctionReturn(0); 123250ffb88aSMatthew Knepley } 123350ffb88aSMatthew Knepley 123450ffb88aSMatthew Knepley /*@ 1235b850b91aSLisandro Dalcin SNESSetMaxNonlinearStepFailures - Sets the maximum number of unsuccessful steps 123650ffb88aSMatthew Knepley attempted by the nonlinear solver before it gives up. 123750ffb88aSMatthew Knepley 123850ffb88aSMatthew Knepley Not Collective 123950ffb88aSMatthew Knepley 124050ffb88aSMatthew Knepley Input Parameters: 124150ffb88aSMatthew Knepley + snes - SNES context 124250ffb88aSMatthew Knepley - maxFails - maximum of unsuccessful steps 124350ffb88aSMatthew Knepley 124450ffb88aSMatthew Knepley Level: intermediate 124550ffb88aSMatthew Knepley 124650ffb88aSMatthew Knepley .keywords: SNES, nonlinear, set, maximum, unsuccessful, steps 124758ebbce7SBarry Smith 1248e1c61ce8SBarry Smith .seealso: SNESGetMaxLinearSolveFailures(), SNESGetLinearSolveIterations(), SNESSetMaxLinearSolveFailures(), SNESGetLinearSolveFailures(), 124958ebbce7SBarry Smith SNESGetMaxNonlinearStepFailures(), SNESGetNonlinearStepFailures() 125050ffb88aSMatthew Knepley @*/ 12517087cfbeSBarry Smith PetscErrorCode SNESSetMaxNonlinearStepFailures(SNES snes, PetscInt maxFails) 125250ffb88aSMatthew Knepley { 125350ffb88aSMatthew Knepley PetscFunctionBegin; 12540700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 125550ffb88aSMatthew Knepley snes->maxFailures = maxFails; 125650ffb88aSMatthew Knepley PetscFunctionReturn(0); 125750ffb88aSMatthew Knepley } 125850ffb88aSMatthew Knepley 125950ffb88aSMatthew Knepley /*@ 1260b850b91aSLisandro Dalcin SNESGetMaxNonlinearStepFailures - Gets the maximum number of unsuccessful steps 126150ffb88aSMatthew Knepley attempted by the nonlinear solver before it gives up. 126250ffb88aSMatthew Knepley 126350ffb88aSMatthew Knepley Not Collective 126450ffb88aSMatthew Knepley 126550ffb88aSMatthew Knepley Input Parameter: 126650ffb88aSMatthew Knepley . snes - SNES context 126750ffb88aSMatthew Knepley 126850ffb88aSMatthew Knepley Output Parameter: 126950ffb88aSMatthew Knepley . maxFails - maximum of unsuccessful steps 127050ffb88aSMatthew Knepley 127150ffb88aSMatthew Knepley Level: intermediate 127250ffb88aSMatthew Knepley 127350ffb88aSMatthew Knepley .keywords: SNES, nonlinear, get, maximum, unsuccessful, steps 127458ebbce7SBarry Smith 1275e1c61ce8SBarry Smith .seealso: SNESGetMaxLinearSolveFailures(), SNESGetLinearSolveIterations(), SNESSetMaxLinearSolveFailures(), SNESGetLinearSolveFailures(), 127658ebbce7SBarry Smith SNESSetMaxNonlinearStepFailures(), SNESGetNonlinearStepFailures() 127758ebbce7SBarry Smith 127850ffb88aSMatthew Knepley @*/ 12797087cfbeSBarry Smith PetscErrorCode SNESGetMaxNonlinearStepFailures(SNES snes, PetscInt *maxFails) 128050ffb88aSMatthew Knepley { 128150ffb88aSMatthew Knepley PetscFunctionBegin; 12820700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 12834482741eSBarry Smith PetscValidIntPointer(maxFails,2); 128450ffb88aSMatthew Knepley *maxFails = snes->maxFailures; 12853a40ed3dSBarry Smith PetscFunctionReturn(0); 12869b94acceSBarry Smith } 1287a847f771SSatish Balay 12882541af92SBarry Smith /*@ 12892541af92SBarry Smith SNESGetNumberFunctionEvals - Gets the number of user provided function evaluations 12902541af92SBarry Smith done by SNES. 12912541af92SBarry Smith 12922541af92SBarry Smith Not Collective 12932541af92SBarry Smith 12942541af92SBarry Smith Input Parameter: 12952541af92SBarry Smith . snes - SNES context 12962541af92SBarry Smith 12972541af92SBarry Smith Output Parameter: 12982541af92SBarry Smith . nfuncs - number of evaluations 12992541af92SBarry Smith 13002541af92SBarry Smith Level: intermediate 13012541af92SBarry Smith 1302971e163fSPeter Brune Notes: Reset every time SNESSolve is called unless SNESSetCountersReset() is used. 1303971e163fSPeter Brune 13042541af92SBarry Smith .keywords: SNES, nonlinear, get, maximum, unsuccessful, steps 130558ebbce7SBarry Smith 1306971e163fSPeter Brune .seealso: SNESGetMaxLinearSolveFailures(), SNESGetLinearSolveIterations(), SNESSetMaxLinearSolveFailures(), SNESGetLinearSolveFailures(), SNESSetCountersReset() 13072541af92SBarry Smith @*/ 13087087cfbeSBarry Smith PetscErrorCode SNESGetNumberFunctionEvals(SNES snes, PetscInt *nfuncs) 13092541af92SBarry Smith { 13102541af92SBarry Smith PetscFunctionBegin; 13110700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 13122541af92SBarry Smith PetscValidIntPointer(nfuncs,2); 13132541af92SBarry Smith *nfuncs = snes->nfuncs; 13142541af92SBarry Smith PetscFunctionReturn(0); 13152541af92SBarry Smith } 13162541af92SBarry Smith 13173d4c4710SBarry Smith /*@ 13183d4c4710SBarry Smith SNESGetLinearSolveFailures - Gets the number of failed (non-converged) 13193d4c4710SBarry Smith linear solvers. 13203d4c4710SBarry Smith 13213d4c4710SBarry Smith Not Collective 13223d4c4710SBarry Smith 13233d4c4710SBarry Smith Input Parameter: 13243d4c4710SBarry Smith . snes - SNES context 13253d4c4710SBarry Smith 13263d4c4710SBarry Smith Output Parameter: 13273d4c4710SBarry Smith . nfails - number of failed solves 13283d4c4710SBarry Smith 13299d85da0cSMatthew G. Knepley Level: intermediate 13309d85da0cSMatthew G. Knepley 13319d85da0cSMatthew G. Knepley Options Database Keys: 13329d85da0cSMatthew G. Knepley . -snes_max_linear_solve_fail <num> - The number of failures before the solve is terminated 13339d85da0cSMatthew G. Knepley 13343d4c4710SBarry Smith Notes: 13353d4c4710SBarry Smith This counter is reset to zero for each successive call to SNESSolve(). 13363d4c4710SBarry Smith 13373d4c4710SBarry Smith .keywords: SNES, nonlinear, get, number, unsuccessful, steps 133858ebbce7SBarry Smith 1339e1c61ce8SBarry Smith .seealso: SNESGetMaxLinearSolveFailures(), SNESGetLinearSolveIterations(), SNESSetMaxLinearSolveFailures() 13403d4c4710SBarry Smith @*/ 13417087cfbeSBarry Smith PetscErrorCode SNESGetLinearSolveFailures(SNES snes,PetscInt *nfails) 13423d4c4710SBarry Smith { 13433d4c4710SBarry Smith PetscFunctionBegin; 13440700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 13453d4c4710SBarry Smith PetscValidIntPointer(nfails,2); 13463d4c4710SBarry Smith *nfails = snes->numLinearSolveFailures; 13473d4c4710SBarry Smith PetscFunctionReturn(0); 13483d4c4710SBarry Smith } 13493d4c4710SBarry Smith 13503d4c4710SBarry Smith /*@ 13513d4c4710SBarry Smith SNESSetMaxLinearSolveFailures - the number of failed linear solve attempts 13523d4c4710SBarry Smith allowed before SNES returns with a diverged reason of SNES_DIVERGED_LINEAR_SOLVE 13533d4c4710SBarry Smith 13543f9fe445SBarry Smith Logically Collective on SNES 13553d4c4710SBarry Smith 13563d4c4710SBarry Smith Input Parameters: 13573d4c4710SBarry Smith + snes - SNES context 13583d4c4710SBarry Smith - maxFails - maximum allowed linear solve failures 13593d4c4710SBarry Smith 13603d4c4710SBarry Smith Level: intermediate 13613d4c4710SBarry Smith 13629d85da0cSMatthew G. Knepley Options Database Keys: 13639d85da0cSMatthew G. Knepley . -snes_max_linear_solve_fail <num> - The number of failures before the solve is terminated 13649d85da0cSMatthew G. Knepley 1365a6796414SBarry Smith Notes: By default this is 0; that is SNES returns on the first failed linear solve 13663d4c4710SBarry Smith 13673d4c4710SBarry Smith .keywords: SNES, nonlinear, set, maximum, unsuccessful, steps 13683d4c4710SBarry Smith 136958ebbce7SBarry Smith .seealso: SNESGetLinearSolveFailures(), SNESGetMaxLinearSolveFailures(), SNESGetLinearSolveIterations() 13703d4c4710SBarry Smith @*/ 13717087cfbeSBarry Smith PetscErrorCode SNESSetMaxLinearSolveFailures(SNES snes, PetscInt maxFails) 13723d4c4710SBarry Smith { 13733d4c4710SBarry Smith PetscFunctionBegin; 13740700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 1375c5eb9154SBarry Smith PetscValidLogicalCollectiveInt(snes,maxFails,2); 13763d4c4710SBarry Smith snes->maxLinearSolveFailures = maxFails; 13773d4c4710SBarry Smith PetscFunctionReturn(0); 13783d4c4710SBarry Smith } 13793d4c4710SBarry Smith 13803d4c4710SBarry Smith /*@ 13813d4c4710SBarry Smith SNESGetMaxLinearSolveFailures - gets the maximum number of linear solve failures that 13823d4c4710SBarry Smith are allowed before SNES terminates 13833d4c4710SBarry Smith 13843d4c4710SBarry Smith Not Collective 13853d4c4710SBarry Smith 13863d4c4710SBarry Smith Input Parameter: 13873d4c4710SBarry Smith . snes - SNES context 13883d4c4710SBarry Smith 13893d4c4710SBarry Smith Output Parameter: 13903d4c4710SBarry Smith . maxFails - maximum of unsuccessful solves allowed 13913d4c4710SBarry Smith 13923d4c4710SBarry Smith Level: intermediate 13933d4c4710SBarry Smith 13943d4c4710SBarry Smith Notes: By default this is 1; that is SNES returns on the first failed linear solve 13953d4c4710SBarry Smith 13963d4c4710SBarry Smith .keywords: SNES, nonlinear, get, maximum, unsuccessful, steps 13973d4c4710SBarry Smith 1398e1c61ce8SBarry Smith .seealso: SNESGetLinearSolveFailures(), SNESGetLinearSolveIterations(), SNESSetMaxLinearSolveFailures(), 13993d4c4710SBarry Smith @*/ 14007087cfbeSBarry Smith PetscErrorCode SNESGetMaxLinearSolveFailures(SNES snes, PetscInt *maxFails) 14013d4c4710SBarry Smith { 14023d4c4710SBarry Smith PetscFunctionBegin; 14030700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 14043d4c4710SBarry Smith PetscValidIntPointer(maxFails,2); 14053d4c4710SBarry Smith *maxFails = snes->maxLinearSolveFailures; 14063d4c4710SBarry Smith PetscFunctionReturn(0); 14073d4c4710SBarry Smith } 14083d4c4710SBarry Smith 1409c96a6f78SLois Curfman McInnes /*@ 1410b850b91aSLisandro Dalcin SNESGetLinearSolveIterations - Gets the total number of linear iterations 1411c96a6f78SLois Curfman McInnes used by the nonlinear solver. 1412c96a6f78SLois Curfman McInnes 1413c7afd0dbSLois Curfman McInnes Not Collective 1414c7afd0dbSLois Curfman McInnes 1415c96a6f78SLois Curfman McInnes Input Parameter: 1416c96a6f78SLois Curfman McInnes . snes - SNES context 1417c96a6f78SLois Curfman McInnes 1418c96a6f78SLois Curfman McInnes Output Parameter: 1419c96a6f78SLois Curfman McInnes . lits - number of linear iterations 1420c96a6f78SLois Curfman McInnes 1421c96a6f78SLois Curfman McInnes Notes: 1422971e163fSPeter Brune This counter is reset to zero for each successive call to SNESSolve() unless SNESSetCountersReset() is used. 1423c96a6f78SLois Curfman McInnes 1424010be392SBarry 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 1425010be392SBarry Smith then call KSPGetIterationNumber() after the failed solve. 1426010be392SBarry Smith 142736851e7fSLois Curfman McInnes Level: intermediate 142836851e7fSLois Curfman McInnes 1429c96a6f78SLois Curfman McInnes .keywords: SNES, nonlinear, get, number, linear, iterations 14302b668275SBarry Smith 143171dbe336SPeter Brune .seealso: SNESGetIterationNumber(), SNESGetLinearSolveFailures(), SNESGetMaxLinearSolveFailures(), SNESSetCountersReset() 1432c96a6f78SLois Curfman McInnes @*/ 14337087cfbeSBarry Smith PetscErrorCode SNESGetLinearSolveIterations(SNES snes,PetscInt *lits) 1434c96a6f78SLois Curfman McInnes { 14353a40ed3dSBarry Smith PetscFunctionBegin; 14360700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 14374482741eSBarry Smith PetscValidIntPointer(lits,2); 1438c96a6f78SLois Curfman McInnes *lits = snes->linear_its; 14393a40ed3dSBarry Smith PetscFunctionReturn(0); 1440c96a6f78SLois Curfman McInnes } 1441c96a6f78SLois Curfman McInnes 1442971e163fSPeter Brune /*@ 1443971e163fSPeter Brune SNESSetCountersReset - Sets whether or not the counters for linear iterations and function evaluations 1444971e163fSPeter Brune are reset every time SNESSolve() is called. 1445971e163fSPeter Brune 1446971e163fSPeter Brune Logically Collective on SNES 1447971e163fSPeter Brune 1448971e163fSPeter Brune Input Parameter: 1449971e163fSPeter Brune + snes - SNES context 1450971e163fSPeter Brune - reset - whether to reset the counters or not 1451971e163fSPeter Brune 1452971e163fSPeter Brune Notes: 1453fa19ca70SBarry Smith This defaults to PETSC_TRUE 1454971e163fSPeter Brune 1455971e163fSPeter Brune Level: developer 1456971e163fSPeter Brune 1457971e163fSPeter Brune .keywords: SNES, nonlinear, set, reset, number, linear, iterations 1458971e163fSPeter Brune 1459734794cfSBarry Smith .seealso: SNESGetNumberFunctionEvals(), SNESGetLinearSolveIterations(), SNESGetNPC() 1460971e163fSPeter Brune @*/ 1461971e163fSPeter Brune PetscErrorCode SNESSetCountersReset(SNES snes,PetscBool reset) 1462971e163fSPeter Brune { 1463971e163fSPeter Brune PetscFunctionBegin; 1464971e163fSPeter Brune PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 1465971e163fSPeter Brune PetscValidLogicalCollectiveBool(snes,reset,2); 1466971e163fSPeter Brune snes->counters_reset = reset; 1467971e163fSPeter Brune PetscFunctionReturn(0); 1468971e163fSPeter Brune } 1469971e163fSPeter Brune 147082bf6240SBarry Smith 14712999313aSBarry Smith /*@ 14722999313aSBarry Smith SNESSetKSP - Sets a KSP context for the SNES object to use 14732999313aSBarry Smith 14742999313aSBarry Smith Not Collective, but the SNES and KSP objects must live on the same MPI_Comm 14752999313aSBarry Smith 14762999313aSBarry Smith Input Parameters: 14772999313aSBarry Smith + snes - the SNES context 14782999313aSBarry Smith - ksp - the KSP context 14792999313aSBarry Smith 14802999313aSBarry Smith Notes: 14812999313aSBarry Smith The SNES object already has its KSP object, you can obtain with SNESGetKSP() 14822999313aSBarry Smith so this routine is rarely needed. 14832999313aSBarry Smith 14842999313aSBarry Smith The KSP object that is already in the SNES object has its reference count 14852999313aSBarry Smith decreased by one. 14862999313aSBarry Smith 14872999313aSBarry Smith Level: developer 14882999313aSBarry Smith 14892999313aSBarry Smith .keywords: SNES, nonlinear, get, KSP, context 14902999313aSBarry Smith 14912999313aSBarry Smith .seealso: KSPGetPC(), SNESCreate(), KSPCreate(), SNESSetKSP() 14922999313aSBarry Smith @*/ 14937087cfbeSBarry Smith PetscErrorCode SNESSetKSP(SNES snes,KSP ksp) 14942999313aSBarry Smith { 14952999313aSBarry Smith PetscErrorCode ierr; 14962999313aSBarry Smith 14972999313aSBarry Smith PetscFunctionBegin; 14980700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 14990700a824SBarry Smith PetscValidHeaderSpecific(ksp,KSP_CLASSID,2); 15002999313aSBarry Smith PetscCheckSameComm(snes,1,ksp,2); 15017dcf0eaaSdalcinl ierr = PetscObjectReference((PetscObject)ksp);CHKERRQ(ierr); 1502906ed7ccSBarry Smith if (snes->ksp) {ierr = PetscObjectDereference((PetscObject)snes->ksp);CHKERRQ(ierr);} 15032999313aSBarry Smith snes->ksp = ksp; 15042999313aSBarry Smith PetscFunctionReturn(0); 15052999313aSBarry Smith } 15062999313aSBarry Smith 15079b94acceSBarry Smith /* -----------------------------------------------------------*/ 150852baeb72SSatish Balay /*@ 15099b94acceSBarry Smith SNESCreate - Creates a nonlinear solver context. 15109b94acceSBarry Smith 1511c7afd0dbSLois Curfman McInnes Collective on MPI_Comm 1512c7afd0dbSLois Curfman McInnes 1513c7afd0dbSLois Curfman McInnes Input Parameters: 1514906ed7ccSBarry Smith . comm - MPI communicator 15159b94acceSBarry Smith 15169b94acceSBarry Smith Output Parameter: 15179b94acceSBarry Smith . outsnes - the new SNES context 15189b94acceSBarry Smith 1519c7afd0dbSLois Curfman McInnes Options Database Keys: 1520c7afd0dbSLois Curfman McInnes + -snes_mf - Activates default matrix-free Jacobian-vector products, 1521c7afd0dbSLois Curfman McInnes and no preconditioning matrix 1522c7afd0dbSLois Curfman McInnes . -snes_mf_operator - Activates default matrix-free Jacobian-vector 1523c7afd0dbSLois Curfman McInnes products, and a user-provided preconditioning matrix 1524c7afd0dbSLois Curfman McInnes as set by SNESSetJacobian() 1525c7afd0dbSLois Curfman McInnes - -snes_fd - Uses (slow!) finite differences to compute Jacobian 1526c1f60f51SBarry Smith 152736851e7fSLois Curfman McInnes Level: beginner 152836851e7fSLois Curfman McInnes 1529efd4aadfSBarry Smith Developer Notes: SNES always creates a KSP object even though many SNES methods do not use it. This is 1530efd4aadfSBarry Smith unfortunate and should be fixed at some point. The flag snes->usesksp indicates if the 1531efd4aadfSBarry Smith particular method does use KSP and regulates if the information about the KSP is printed 1532efd4aadfSBarry Smith in SNESView(). TSSetFromOptions() does call SNESSetFromOptions() which can lead to users being confused 1533efd4aadfSBarry Smith by help messages about meaningless SNES options. 1534efd4aadfSBarry Smith 1535efd4aadfSBarry Smith SNES always creates the snes->kspconvctx even though it is used by only one type. This should 1536efd4aadfSBarry Smith be fixed. 1537efd4aadfSBarry Smith 15389b94acceSBarry Smith .keywords: SNES, nonlinear, create, context 15399b94acceSBarry Smith 1540a8054027SBarry Smith .seealso: SNESSolve(), SNESDestroy(), SNES, SNESSetLagPreconditioner() 1541a8054027SBarry Smith 15429b94acceSBarry Smith @*/ 15437087cfbeSBarry Smith PetscErrorCode SNESCreate(MPI_Comm comm,SNES *outsnes) 15449b94acceSBarry Smith { 1545dfbe8321SBarry Smith PetscErrorCode ierr; 15469b94acceSBarry Smith SNES snes; 1547fa9f3622SBarry Smith SNESKSPEW *kctx; 154837fcc0dbSBarry Smith 15493a40ed3dSBarry Smith PetscFunctionBegin; 1550ed1caa07SMatthew Knepley PetscValidPointer(outsnes,2); 15510298fd71SBarry Smith *outsnes = NULL; 1552607a6623SBarry Smith ierr = SNESInitializePackage();CHKERRQ(ierr); 15538ba1e511SMatthew Knepley 155473107ff1SLisandro Dalcin ierr = PetscHeaderCreate(snes,SNES_CLASSID,"SNES","Nonlinear solver","SNES",comm,SNESDestroy,SNESView);CHKERRQ(ierr); 15557adad957SLisandro Dalcin 15568d359177SBarry Smith snes->ops->converged = SNESConvergedDefault; 15572c155ee1SBarry Smith snes->usesksp = PETSC_TRUE; 155888976e71SPeter Brune snes->tolerancesset = PETSC_FALSE; 15599b94acceSBarry Smith snes->max_its = 50; 15609750a799SBarry Smith snes->max_funcs = 10000; 15619b94acceSBarry Smith snes->norm = 0.0; 1562365a6726SPeter Brune snes->normschedule = SNES_NORM_ALWAYS; 15636c67d002SPeter Brune snes->functype = SNES_FUNCTION_DEFAULT; 15643a2046daSBarry Smith #if defined(PETSC_USE_REAL_SINGLE) 15653a2046daSBarry Smith snes->rtol = 1.e-5; 15663a2046daSBarry Smith #else 1567b4874afaSBarry Smith snes->rtol = 1.e-8; 15683a2046daSBarry Smith #endif 1569b4874afaSBarry Smith snes->ttol = 0.0; 15703a2046daSBarry Smith #if defined(PETSC_USE_REAL_SINGLE) 15713a2046daSBarry Smith snes->abstol = 1.e-25; 15723a2046daSBarry Smith #else 157370441072SBarry Smith snes->abstol = 1.e-50; 15743a2046daSBarry Smith #endif 15757cd0ae37SLisandro Dalcin #if defined(PETSC_USE_REAL_SINGLE) 15767cd0ae37SLisandro Dalcin snes->stol = 1.e-5; 15777cd0ae37SLisandro Dalcin #else 1578c60f73f4SPeter Brune snes->stol = 1.e-8; 15797cd0ae37SLisandro Dalcin #endif 15803a2046daSBarry Smith #if defined(PETSC_USE_REAL_SINGLE) 15813a2046daSBarry Smith snes->deltatol = 1.e-6; 15823a2046daSBarry Smith #else 15834b27c08aSLois Curfman McInnes snes->deltatol = 1.e-12; 15843a2046daSBarry Smith #endif 1585e37c518bSBarry Smith snes->divtol = 1.e4; 1586e37c518bSBarry Smith snes->rnorm0 = 0; 15879b94acceSBarry Smith snes->nfuncs = 0; 158850ffb88aSMatthew Knepley snes->numFailures = 0; 158950ffb88aSMatthew Knepley snes->maxFailures = 1; 15907a00f4a9SLois Curfman McInnes snes->linear_its = 0; 1591e35cf81dSBarry Smith snes->lagjacobian = 1; 159237ec4e1aSPeter Brune snes->jac_iter = 0; 159337ec4e1aSPeter Brune snes->lagjac_persist = PETSC_FALSE; 1594a8054027SBarry Smith snes->lagpreconditioner = 1; 159537ec4e1aSPeter Brune snes->pre_iter = 0; 159637ec4e1aSPeter Brune snes->lagpre_persist = PETSC_FALSE; 1597639f9d9dSBarry Smith snes->numbermonitors = 0; 15989b94acceSBarry Smith snes->data = 0; 15994dc4c822SBarry Smith snes->setupcalled = PETSC_FALSE; 1600186905e3SBarry Smith snes->ksp_ewconv = PETSC_FALSE; 16016f24a144SLois Curfman McInnes snes->nwork = 0; 160258c9b817SLisandro Dalcin snes->work = 0; 160358c9b817SLisandro Dalcin snes->nvwork = 0; 160458c9b817SLisandro Dalcin snes->vwork = 0; 1605758f92a0SBarry Smith snes->conv_hist_len = 0; 1606758f92a0SBarry Smith snes->conv_hist_max = 0; 16070298fd71SBarry Smith snes->conv_hist = NULL; 16080298fd71SBarry Smith snes->conv_hist_its = NULL; 1609758f92a0SBarry Smith snes->conv_hist_reset = PETSC_TRUE; 1610971e163fSPeter Brune snes->counters_reset = PETSC_TRUE; 1611e4ed7901SPeter Brune snes->vec_func_init_set = PETSC_FALSE; 1612184914b5SBarry Smith snes->reason = SNES_CONVERGED_ITERATING; 1613efd4aadfSBarry Smith snes->npcside = PC_RIGHT; 1614c40d0f55SPeter Brune 1615d8f46077SPeter Brune snes->mf = PETSC_FALSE; 1616d8f46077SPeter Brune snes->mf_operator = PETSC_FALSE; 1617d8f46077SPeter Brune snes->mf_version = 1; 1618d8f46077SPeter Brune 16193d4c4710SBarry Smith snes->numLinearSolveFailures = 0; 16203d4c4710SBarry Smith snes->maxLinearSolveFailures = 1; 16213d4c4710SBarry Smith 1622349187a7SBarry Smith snes->vizerotolerance = 1.e-8; 1623349187a7SBarry Smith 16244fc747eaSLawrence Mitchell /* Set this to true if the implementation of SNESSolve_XXX does compute the residual at the final solution. */ 16254fc747eaSLawrence Mitchell snes->alwayscomputesfinalresidual = PETSC_FALSE; 16264fc747eaSLawrence Mitchell 16279b94acceSBarry Smith /* Create context to compute Eisenstat-Walker relative tolerance for KSP */ 1628b00a9115SJed Brown ierr = PetscNewLog(snes,&kctx);CHKERRQ(ierr); 1629f5af7f23SKarl Rupp 16309b94acceSBarry Smith snes->kspconvctx = (void*)kctx; 16319b94acceSBarry Smith kctx->version = 2; 16329b94acceSBarry Smith kctx->rtol_0 = .3; /* Eisenstat and Walker suggest rtol_0=.5, but 16339b94acceSBarry Smith this was too large for some test cases */ 163475567043SBarry Smith kctx->rtol_last = 0.0; 16359b94acceSBarry Smith kctx->rtol_max = .9; 16369b94acceSBarry Smith kctx->gamma = 1.0; 163762d1f40fSBarry Smith kctx->alpha = .5*(1.0 + PetscSqrtReal(5.0)); 163871f87433Sdalcinl kctx->alpha2 = kctx->alpha; 16399b94acceSBarry Smith kctx->threshold = .1; 164075567043SBarry Smith kctx->lresid_last = 0.0; 164175567043SBarry Smith kctx->norm_last = 0.0; 16429b94acceSBarry Smith 16439b94acceSBarry Smith *outsnes = snes; 16443a40ed3dSBarry Smith PetscFunctionReturn(0); 16459b94acceSBarry Smith } 16469b94acceSBarry Smith 164788f0584fSBarry Smith /*MC 1648411c0326SBarry Smith SNESFunction - Functional form used to convey the nonlinear function to be solved by SNES 164988f0584fSBarry Smith 165088f0584fSBarry Smith Synopsis: 1651411c0326SBarry Smith #include "petscsnes.h" 1652411c0326SBarry Smith PetscErrorCode SNESFunction(SNES snes,Vec x,Vec f,void *ctx); 165388f0584fSBarry Smith 165488f0584fSBarry Smith Input Parameters: 165588f0584fSBarry Smith + snes - the SNES context 165688f0584fSBarry Smith . x - state at which to evaluate residual 165788f0584fSBarry Smith - ctx - optional user-defined function context, passed in with SNESSetFunction() 165888f0584fSBarry Smith 165988f0584fSBarry Smith Output Parameter: 166088f0584fSBarry Smith . f - vector to put residual (function value) 166188f0584fSBarry Smith 1662878cb397SSatish Balay Level: intermediate 1663878cb397SSatish Balay 166488f0584fSBarry Smith .seealso: SNESSetFunction(), SNESGetFunction() 166588f0584fSBarry Smith M*/ 166688f0584fSBarry Smith 16679b94acceSBarry Smith /*@C 16689b94acceSBarry Smith SNESSetFunction - Sets the function evaluation routine and function 16699b94acceSBarry Smith vector for use by the SNES routines in solving systems of nonlinear 16709b94acceSBarry Smith equations. 16719b94acceSBarry Smith 16723f9fe445SBarry Smith Logically Collective on SNES 1673fee21e36SBarry Smith 1674c7afd0dbSLois Curfman McInnes Input Parameters: 1675c7afd0dbSLois Curfman McInnes + snes - the SNES context 1676c7afd0dbSLois Curfman McInnes . r - vector to store function value 1677f8b49ee9SBarry Smith . f - function evaluation routine; see SNESFunction for calling sequence details 1678c7afd0dbSLois Curfman McInnes - ctx - [optional] user-defined context for private data for the 16790298fd71SBarry Smith function evaluation routine (may be NULL) 16809b94acceSBarry Smith 16819b94acceSBarry Smith Notes: 16829b94acceSBarry Smith The Newton-like methods typically solve linear systems of the form 16839b94acceSBarry Smith $ f'(x) x = -f(x), 1684c7afd0dbSLois Curfman McInnes where f'(x) denotes the Jacobian matrix and f(x) is the function. 16859b94acceSBarry Smith 168636851e7fSLois Curfman McInnes Level: beginner 168736851e7fSLois Curfman McInnes 16889b94acceSBarry Smith .keywords: SNES, nonlinear, set, function 16899b94acceSBarry Smith 1690bf388a1fSBarry Smith .seealso: SNESGetFunction(), SNESComputeFunction(), SNESSetJacobian(), SNESSetPicard(), SNESFunction 16919b94acceSBarry Smith @*/ 1692f8b49ee9SBarry Smith PetscErrorCode SNESSetFunction(SNES snes,Vec r,PetscErrorCode (*f)(SNES,Vec,Vec,void*),void *ctx) 16939b94acceSBarry Smith { 169485385478SLisandro Dalcin PetscErrorCode ierr; 16956cab3a1bSJed Brown DM dm; 16966cab3a1bSJed Brown 16973a40ed3dSBarry Smith PetscFunctionBegin; 16980700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 1699d2a683ecSLisandro Dalcin if (r) { 1700d2a683ecSLisandro Dalcin PetscValidHeaderSpecific(r,VEC_CLASSID,2); 1701d2a683ecSLisandro Dalcin PetscCheckSameComm(snes,1,r,2); 170285385478SLisandro Dalcin ierr = PetscObjectReference((PetscObject)r);CHKERRQ(ierr); 17036bf464f9SBarry Smith ierr = VecDestroy(&snes->vec_func);CHKERRQ(ierr); 1704f5af7f23SKarl Rupp 170585385478SLisandro Dalcin snes->vec_func = r; 1706d2a683ecSLisandro Dalcin } 17076cab3a1bSJed Brown ierr = SNESGetDM(snes,&dm);CHKERRQ(ierr); 1708f8b49ee9SBarry Smith ierr = DMSNESSetFunction(dm,f,ctx);CHKERRQ(ierr); 17093a40ed3dSBarry Smith PetscFunctionReturn(0); 17109b94acceSBarry Smith } 17119b94acceSBarry Smith 1712646217ecSPeter Brune 1713e4ed7901SPeter Brune /*@C 1714e4ed7901SPeter Brune SNESSetInitialFunction - Sets the function vector to be used as the 1715e4ed7901SPeter Brune function norm at the initialization of the method. In some 1716e4ed7901SPeter Brune instances, the user has precomputed the function before calling 1717e4ed7901SPeter Brune SNESSolve. This function allows one to avoid a redundant call 1718e4ed7901SPeter Brune to SNESComputeFunction in that case. 1719e4ed7901SPeter Brune 1720e4ed7901SPeter Brune Logically Collective on SNES 1721e4ed7901SPeter Brune 1722e4ed7901SPeter Brune Input Parameters: 1723e4ed7901SPeter Brune + snes - the SNES context 1724e4ed7901SPeter Brune - f - vector to store function value 1725e4ed7901SPeter Brune 1726e4ed7901SPeter Brune Notes: 1727e4ed7901SPeter Brune This should not be modified during the solution procedure. 1728e4ed7901SPeter Brune 1729e4ed7901SPeter Brune This is used extensively in the SNESFAS hierarchy and in nonlinear preconditioning. 1730e4ed7901SPeter Brune 1731e4ed7901SPeter Brune Level: developer 1732e4ed7901SPeter Brune 1733e4ed7901SPeter Brune .keywords: SNES, nonlinear, set, function 1734e4ed7901SPeter Brune 1735e4ed7901SPeter Brune .seealso: SNESSetFunction(), SNESComputeFunction(), SNESSetInitialFunctionNorm() 1736e4ed7901SPeter Brune @*/ 1737e4ed7901SPeter Brune PetscErrorCode SNESSetInitialFunction(SNES snes, Vec f) 1738e4ed7901SPeter Brune { 1739e4ed7901SPeter Brune PetscErrorCode ierr; 1740e4ed7901SPeter Brune Vec vec_func; 1741e4ed7901SPeter Brune 1742e4ed7901SPeter Brune PetscFunctionBegin; 1743e4ed7901SPeter Brune PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 1744e4ed7901SPeter Brune PetscValidHeaderSpecific(f,VEC_CLASSID,2); 1745e4ed7901SPeter Brune PetscCheckSameComm(snes,1,f,2); 1746efd4aadfSBarry Smith if (snes->npcside== PC_LEFT && snes->functype == SNES_FUNCTION_PRECONDITIONED) { 1747902f982fSPeter Brune snes->vec_func_init_set = PETSC_FALSE; 1748902f982fSPeter Brune PetscFunctionReturn(0); 1749902f982fSPeter Brune } 17500298fd71SBarry Smith ierr = SNESGetFunction(snes,&vec_func,NULL,NULL);CHKERRQ(ierr); 1751e4ed7901SPeter Brune ierr = VecCopy(f, vec_func);CHKERRQ(ierr); 1752f5af7f23SKarl Rupp 1753217b9c2eSPeter Brune snes->vec_func_init_set = PETSC_TRUE; 1754e4ed7901SPeter Brune PetscFunctionReturn(0); 1755e4ed7901SPeter Brune } 1756e4ed7901SPeter Brune 1757534ebe21SPeter Brune /*@ 1758365a6726SPeter Brune SNESSetNormSchedule - Sets the SNESNormSchedule used in covergence and monitoring 1759534ebe21SPeter Brune of the SNES method. 1760534ebe21SPeter Brune 1761534ebe21SPeter Brune Logically Collective on SNES 1762534ebe21SPeter Brune 1763534ebe21SPeter Brune Input Parameters: 1764534ebe21SPeter Brune + snes - the SNES context 1765365a6726SPeter Brune - normschedule - the frequency of norm computation 1766534ebe21SPeter Brune 1767517f1916SMatthew G. Knepley Options Database Key: 1768517f1916SMatthew G. Knepley . -snes_norm_schedule <none, always, initialonly, finalonly, initalfinalonly> 1769517f1916SMatthew G. Knepley 1770534ebe21SPeter Brune Notes: 1771365a6726SPeter Brune Only certain SNES methods support certain SNESNormSchedules. Most require evaluation 1772534ebe21SPeter Brune of the nonlinear function and the taking of its norm at every iteration to 1773534ebe21SPeter Brune even ensure convergence at all. However, methods such as custom Gauss-Seidel methods 1774be95d8f1SBarry Smith (SNESNGS) and the like do not require the norm of the function to be computed, and therfore 1775534ebe21SPeter Brune may either be monitored for convergence or not. As these are often used as nonlinear 1776534ebe21SPeter Brune preconditioners, monitoring the norm of their error is not a useful enterprise within 1777534ebe21SPeter Brune their solution. 1778534ebe21SPeter Brune 1779534ebe21SPeter Brune Level: developer 1780534ebe21SPeter Brune 1781534ebe21SPeter Brune .keywords: SNES, nonlinear, set, function, norm, type 1782534ebe21SPeter Brune 1783365a6726SPeter Brune .seealso: SNESGetNormSchedule(), SNESComputeFunction(), VecNorm(), SNESSetFunction(), SNESSetInitialFunction(), SNESNormSchedule 1784534ebe21SPeter Brune @*/ 1785365a6726SPeter Brune PetscErrorCode SNESSetNormSchedule(SNES snes, SNESNormSchedule normschedule) 1786534ebe21SPeter Brune { 1787534ebe21SPeter Brune PetscFunctionBegin; 1788534ebe21SPeter Brune PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 1789365a6726SPeter Brune snes->normschedule = normschedule; 1790534ebe21SPeter Brune PetscFunctionReturn(0); 1791534ebe21SPeter Brune } 1792534ebe21SPeter Brune 1793534ebe21SPeter Brune 1794534ebe21SPeter Brune /*@ 1795365a6726SPeter Brune SNESGetNormSchedule - Gets the SNESNormSchedule used in covergence and monitoring 1796534ebe21SPeter Brune of the SNES method. 1797534ebe21SPeter Brune 1798534ebe21SPeter Brune Logically Collective on SNES 1799534ebe21SPeter Brune 1800534ebe21SPeter Brune Input Parameters: 1801534ebe21SPeter Brune + snes - the SNES context 1802365a6726SPeter Brune - normschedule - the type of the norm used 1803534ebe21SPeter Brune 1804534ebe21SPeter Brune Level: advanced 1805534ebe21SPeter Brune 1806534ebe21SPeter Brune .keywords: SNES, nonlinear, set, function, norm, type 1807534ebe21SPeter Brune 1808365a6726SPeter Brune .seealso: SNESSetNormSchedule(), SNESComputeFunction(), VecNorm(), SNESSetFunction(), SNESSetInitialFunction(), SNESNormSchedule 1809534ebe21SPeter Brune @*/ 1810365a6726SPeter Brune PetscErrorCode SNESGetNormSchedule(SNES snes, SNESNormSchedule *normschedule) 1811534ebe21SPeter Brune { 1812534ebe21SPeter Brune PetscFunctionBegin; 1813534ebe21SPeter Brune PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 1814365a6726SPeter Brune *normschedule = snes->normschedule; 1815534ebe21SPeter Brune PetscFunctionReturn(0); 1816534ebe21SPeter Brune } 1817534ebe21SPeter Brune 181847073ea2SPeter Brune 1819c5ce4427SMatthew G. Knepley /*@ 1820c5ce4427SMatthew G. Knepley SNESSetFunctionNorm - Sets the last computed residual norm. 1821c5ce4427SMatthew G. Knepley 1822c5ce4427SMatthew G. Knepley Logically Collective on SNES 1823c5ce4427SMatthew G. Knepley 1824c5ce4427SMatthew G. Knepley Input Parameters: 1825c5ce4427SMatthew G. Knepley + snes - the SNES context 1826c5ce4427SMatthew G. Knepley 1827c5ce4427SMatthew G. Knepley - normschedule - the frequency of norm computation 1828c5ce4427SMatthew G. Knepley 1829c5ce4427SMatthew G. Knepley Level: developer 1830c5ce4427SMatthew G. Knepley 1831c5ce4427SMatthew G. Knepley .keywords: SNES, nonlinear, set, function, norm, type 1832c5ce4427SMatthew G. Knepley .seealso: SNESGetNormSchedule(), SNESComputeFunction(), VecNorm(), SNESSetFunction(), SNESSetInitialFunction(), SNESNormSchedule 1833c5ce4427SMatthew G. Knepley @*/ 1834c5ce4427SMatthew G. Knepley PetscErrorCode SNESSetFunctionNorm(SNES snes, PetscReal norm) 1835c5ce4427SMatthew G. Knepley { 1836c5ce4427SMatthew G. Knepley PetscFunctionBegin; 1837c5ce4427SMatthew G. Knepley PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 1838c5ce4427SMatthew G. Knepley snes->norm = norm; 1839c5ce4427SMatthew G. Knepley PetscFunctionReturn(0); 1840c5ce4427SMatthew G. Knepley } 1841c5ce4427SMatthew G. Knepley 1842c5ce4427SMatthew G. Knepley /*@ 1843c5ce4427SMatthew G. Knepley SNESGetFunctionNorm - Gets the last computed norm of the residual 1844c5ce4427SMatthew G. Knepley 1845c5ce4427SMatthew G. Knepley Not Collective 1846c5ce4427SMatthew G. Knepley 1847c5ce4427SMatthew G. Knepley Input Parameter: 1848c5ce4427SMatthew G. Knepley . snes - the SNES context 1849c5ce4427SMatthew G. Knepley 1850c5ce4427SMatthew G. Knepley Output Parameter: 1851c5ce4427SMatthew G. Knepley . norm - the last computed residual norm 1852c5ce4427SMatthew G. Knepley 1853c5ce4427SMatthew G. Knepley Level: developer 1854c5ce4427SMatthew G. Knepley 1855c5ce4427SMatthew G. Knepley .keywords: SNES, nonlinear, set, function, norm, type 1856c5ce4427SMatthew G. Knepley .seealso: SNESSetNormSchedule(), SNESComputeFunction(), VecNorm(), SNESSetFunction(), SNESSetInitialFunction(), SNESNormSchedule 1857c5ce4427SMatthew G. Knepley @*/ 1858c5ce4427SMatthew G. Knepley PetscErrorCode SNESGetFunctionNorm(SNES snes, PetscReal *norm) 1859c5ce4427SMatthew G. Knepley { 1860c5ce4427SMatthew G. Knepley PetscFunctionBegin; 1861c5ce4427SMatthew G. Knepley PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 1862c5ce4427SMatthew G. Knepley PetscValidPointer(norm, 2); 1863c5ce4427SMatthew G. Knepley *norm = snes->norm; 1864c5ce4427SMatthew G. Knepley PetscFunctionReturn(0); 1865c5ce4427SMatthew G. Knepley } 1866c5ce4427SMatthew G. Knepley 186747073ea2SPeter Brune /*@C 186847073ea2SPeter Brune SNESSetFunctionType - Sets the SNESNormSchedule used in covergence and monitoring 186947073ea2SPeter Brune of the SNES method. 187047073ea2SPeter Brune 187147073ea2SPeter Brune Logically Collective on SNES 187247073ea2SPeter Brune 187347073ea2SPeter Brune Input Parameters: 187447073ea2SPeter Brune + snes - the SNES context 187547073ea2SPeter Brune - normschedule - the frequency of norm computation 187647073ea2SPeter Brune 187747073ea2SPeter Brune Notes: 187847073ea2SPeter Brune Only certain SNES methods support certain SNESNormSchedules. Most require evaluation 187947073ea2SPeter Brune of the nonlinear function and the taking of its norm at every iteration to 188047073ea2SPeter Brune even ensure convergence at all. However, methods such as custom Gauss-Seidel methods 1881be95d8f1SBarry Smith (SNESNGS) and the like do not require the norm of the function to be computed, and therfore 188247073ea2SPeter Brune may either be monitored for convergence or not. As these are often used as nonlinear 188347073ea2SPeter Brune preconditioners, monitoring the norm of their error is not a useful enterprise within 188447073ea2SPeter Brune their solution. 188547073ea2SPeter Brune 188647073ea2SPeter Brune Level: developer 188747073ea2SPeter Brune 188847073ea2SPeter Brune .keywords: SNES, nonlinear, set, function, norm, type 188947073ea2SPeter Brune 189047073ea2SPeter Brune .seealso: SNESGetNormSchedule(), SNESComputeFunction(), VecNorm(), SNESSetFunction(), SNESSetInitialFunction(), SNESNormSchedule 189147073ea2SPeter Brune @*/ 189247073ea2SPeter Brune PetscErrorCode SNESSetFunctionType(SNES snes, SNESFunctionType type) 189347073ea2SPeter Brune { 189447073ea2SPeter Brune PetscFunctionBegin; 189547073ea2SPeter Brune PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 189647073ea2SPeter Brune snes->functype = type; 189747073ea2SPeter Brune PetscFunctionReturn(0); 189847073ea2SPeter Brune } 189947073ea2SPeter Brune 190047073ea2SPeter Brune 190147073ea2SPeter Brune /*@C 190247073ea2SPeter Brune SNESGetFunctionType - Gets the SNESNormSchedule used in covergence and monitoring 190347073ea2SPeter Brune of the SNES method. 190447073ea2SPeter Brune 190547073ea2SPeter Brune Logically Collective on SNES 190647073ea2SPeter Brune 190747073ea2SPeter Brune Input Parameters: 190847073ea2SPeter Brune + snes - the SNES context 190947073ea2SPeter Brune - normschedule - the type of the norm used 191047073ea2SPeter Brune 191147073ea2SPeter Brune Level: advanced 191247073ea2SPeter Brune 191347073ea2SPeter Brune .keywords: SNES, nonlinear, set, function, norm, type 191447073ea2SPeter Brune 191547073ea2SPeter Brune .seealso: SNESSetNormSchedule(), SNESComputeFunction(), VecNorm(), SNESSetFunction(), SNESSetInitialFunction(), SNESNormSchedule 191647073ea2SPeter Brune @*/ 191747073ea2SPeter Brune PetscErrorCode SNESGetFunctionType(SNES snes, SNESFunctionType *type) 191847073ea2SPeter Brune { 191947073ea2SPeter Brune PetscFunctionBegin; 192047073ea2SPeter Brune PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 192147073ea2SPeter Brune *type = snes->functype; 1922534ebe21SPeter Brune PetscFunctionReturn(0); 1923534ebe21SPeter Brune } 1924534ebe21SPeter Brune 1925bf388a1fSBarry Smith /*MC 1926be95d8f1SBarry Smith SNESNGSFunction - function used to convey a Gauss-Seidel sweep on the nonlinear function 1927bf388a1fSBarry Smith 1928bf388a1fSBarry Smith Synopsis: 1929aaa7dc30SBarry Smith #include <petscsnes.h> 1930be95d8f1SBarry Smith $ SNESNGSFunction(SNES snes,Vec x,Vec b,void *ctx); 1931bf388a1fSBarry Smith 1932bf388a1fSBarry Smith + X - solution vector 1933bf388a1fSBarry Smith . B - RHS vector 1934bf388a1fSBarry Smith - ctx - optional user-defined Gauss-Seidel context 1935bf388a1fSBarry Smith 1936878cb397SSatish Balay Level: intermediate 1937878cb397SSatish Balay 1938be95d8f1SBarry Smith .seealso: SNESSetNGS(), SNESGetNGS() 1939bf388a1fSBarry Smith M*/ 1940bf388a1fSBarry Smith 1941c79ef259SPeter Brune /*@C 1942be95d8f1SBarry Smith SNESSetNGS - Sets the user nonlinear Gauss-Seidel routine for 1943c79ef259SPeter Brune use with composed nonlinear solvers. 1944c79ef259SPeter Brune 1945c79ef259SPeter Brune Input Parameters: 1946c79ef259SPeter Brune + snes - the SNES context 1947be95d8f1SBarry Smith . f - function evaluation routine to apply Gauss-Seidel see SNESNGSFunction 1948c79ef259SPeter Brune - ctx - [optional] user-defined context for private data for the 19490298fd71SBarry Smith smoother evaluation routine (may be NULL) 1950c79ef259SPeter Brune 1951c79ef259SPeter Brune Notes: 1952be95d8f1SBarry Smith The NGS routines are used by the composed nonlinear solver to generate 1953c79ef259SPeter Brune a problem appropriate update to the solution, particularly FAS. 1954c79ef259SPeter Brune 1955d28543b3SPeter Brune Level: intermediate 1956c79ef259SPeter Brune 1957d28543b3SPeter Brune .keywords: SNES, nonlinear, set, Gauss-Seidel 1958c79ef259SPeter Brune 1959be95d8f1SBarry Smith .seealso: SNESGetFunction(), SNESComputeNGS() 1960c79ef259SPeter Brune @*/ 1961be95d8f1SBarry Smith PetscErrorCode SNESSetNGS(SNES snes,PetscErrorCode (*f)(SNES,Vec,Vec,void*),void *ctx) 19626cab3a1bSJed Brown { 19636cab3a1bSJed Brown PetscErrorCode ierr; 19646cab3a1bSJed Brown DM dm; 19656cab3a1bSJed Brown 1966646217ecSPeter Brune PetscFunctionBegin; 19676cab3a1bSJed Brown PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 19686cab3a1bSJed Brown ierr = SNESGetDM(snes,&dm);CHKERRQ(ierr); 1969be95d8f1SBarry Smith ierr = DMSNESSetNGS(dm,f,ctx);CHKERRQ(ierr); 1970646217ecSPeter Brune PetscFunctionReturn(0); 1971646217ecSPeter Brune } 1972646217ecSPeter Brune 197382b59a81SJed Brown PETSC_EXTERN PetscErrorCode SNESPicardComputeFunction(SNES snes,Vec x,Vec f,void *ctx) 19748b0a5094SBarry Smith { 19758b0a5094SBarry Smith PetscErrorCode ierr; 1976e03ab78fSPeter Brune DM dm; 1977942e3340SBarry Smith DMSNES sdm; 19786cab3a1bSJed Brown 19798b0a5094SBarry Smith PetscFunctionBegin; 1980e03ab78fSPeter Brune ierr = SNESGetDM(snes,&dm);CHKERRQ(ierr); 1981942e3340SBarry Smith ierr = DMGetDMSNES(dm,&sdm);CHKERRQ(ierr); 19828b0a5094SBarry Smith /* A(x)*x - b(x) */ 198322c6f798SBarry Smith if (sdm->ops->computepfunction) { 198422c6f798SBarry Smith ierr = (*sdm->ops->computepfunction)(snes,x,f,sdm->pctx);CHKERRQ(ierr); 198522c6f798SBarry Smith } else SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE, "Must call SNESSetPicard() to provide Picard function."); 1986e03ab78fSPeter Brune 198722c6f798SBarry Smith if (sdm->ops->computepjacobian) { 1988d1e9a80fSBarry Smith ierr = (*sdm->ops->computepjacobian)(snes,x,snes->jacobian,snes->jacobian_pre,sdm->pctx);CHKERRQ(ierr); 198974e1e8c1SBarry Smith } else SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE, "Must call SNESSetPicard() to provide Picard matrix."); 19908b0a5094SBarry Smith ierr = VecScale(f,-1.0);CHKERRQ(ierr); 199195eabcedSBarry Smith ierr = MatMultAdd(snes->jacobian,x,f,f);CHKERRQ(ierr); 19928b0a5094SBarry Smith PetscFunctionReturn(0); 19938b0a5094SBarry Smith } 19948b0a5094SBarry Smith 1995d1e9a80fSBarry Smith PETSC_EXTERN PetscErrorCode SNESPicardComputeJacobian(SNES snes,Vec x1,Mat J,Mat B,void *ctx) 19968b0a5094SBarry Smith { 19978b0a5094SBarry Smith PetscFunctionBegin; 1998e03ab78fSPeter Brune /* the jacobian matrix should be pre-filled in SNESPicardComputeFunction */ 19998b0a5094SBarry Smith PetscFunctionReturn(0); 20008b0a5094SBarry Smith } 20018b0a5094SBarry Smith 20028b0a5094SBarry Smith /*@C 20030d04baf8SBarry Smith SNESSetPicard - Use SNES to solve the semilinear-system A(x) x = b(x) via a Picard type iteration (Picard linearization) 20048b0a5094SBarry Smith 20058b0a5094SBarry Smith Logically Collective on SNES 20068b0a5094SBarry Smith 20078b0a5094SBarry Smith Input Parameters: 20088b0a5094SBarry Smith + snes - the SNES context 20098b0a5094SBarry Smith . r - vector to store function value 2010f8b49ee9SBarry Smith . b - function evaluation routine 2011e5d3d808SBarry Smith . Amat - matrix with which A(x) x - b(x) is to be computed 2012e5d3d808SBarry Smith . Pmat - matrix from which preconditioner is computed (usually the same as Amat) 2013411c0326SBarry Smith . J - function to compute matrix value, see SNESJacobianFunction for details on its calling sequence 20148b0a5094SBarry Smith - ctx - [optional] user-defined context for private data for the 20150298fd71SBarry Smith function evaluation routine (may be NULL) 20168b0a5094SBarry Smith 20178b0a5094SBarry Smith Notes: 2018f450aa47SBarry 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 2019f450aa47SBarry 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. 2020f450aa47SBarry Smith 20218b0a5094SBarry Smith One can call SNESSetPicard() or SNESSetFunction() (and possibly SNESSetJacobian()) but cannot call both 20228b0a5094SBarry Smith 20238b0a5094SBarry 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} 20248b0a5094SBarry 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. 20258b0a5094SBarry Smith 20268b0a5094SBarry Smith Run with -snes_mf_operator to solve the system with Newton's method using A(x^{n}) to construct the preconditioner. 20278b0a5094SBarry Smith 20280d04baf8SBarry Smith We implement the defect correction form of the Picard iteration because it converges much more generally when inexact linear solvers are used then 20290d04baf8SBarry Smith the direct Picard iteration A(x^n) x^{n+1} = b(x^n) 20308b0a5094SBarry Smith 20318b0a5094SBarry 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 20328b0a5094SBarry 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 20338b0a5094SBarry Smith different please contact us at petsc-dev@mcs.anl.gov and we'll have an entirely new argument :-). 20348b0a5094SBarry Smith 2035f450aa47SBarry Smith Level: intermediate 20368b0a5094SBarry Smith 20378b0a5094SBarry Smith .keywords: SNES, nonlinear, set, function 20388b0a5094SBarry Smith 2039411c0326SBarry Smith .seealso: SNESGetFunction(), SNESSetFunction(), SNESComputeFunction(), SNESSetJacobian(), SNESGetPicard(), SNESLineSearchPreCheckPicard(), SNESJacobianFunction 20408b0a5094SBarry Smith @*/ 2041d1e9a80fSBarry 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) 20428b0a5094SBarry Smith { 20438b0a5094SBarry Smith PetscErrorCode ierr; 2044e03ab78fSPeter Brune DM dm; 2045e03ab78fSPeter Brune 20468b0a5094SBarry Smith PetscFunctionBegin; 20478b0a5094SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 2048e03ab78fSPeter Brune ierr = SNESGetDM(snes, &dm);CHKERRQ(ierr); 2049f8b49ee9SBarry Smith ierr = DMSNESSetPicard(dm,b,J,ctx);CHKERRQ(ierr); 20508b0a5094SBarry Smith ierr = SNESSetFunction(snes,r,SNESPicardComputeFunction,ctx);CHKERRQ(ierr); 2051e5d3d808SBarry Smith ierr = SNESSetJacobian(snes,Amat,Pmat,SNESPicardComputeJacobian,ctx);CHKERRQ(ierr); 20528b0a5094SBarry Smith PetscFunctionReturn(0); 20538b0a5094SBarry Smith } 20548b0a5094SBarry Smith 20557971a8bfSPeter Brune /*@C 20567971a8bfSPeter Brune SNESGetPicard - Returns the context for the Picard iteration 20577971a8bfSPeter Brune 20587971a8bfSPeter Brune Not Collective, but Vec is parallel if SNES is parallel. Collective if Vec is requested, but has not been created yet. 20597971a8bfSPeter Brune 20607971a8bfSPeter Brune Input Parameter: 20617971a8bfSPeter Brune . snes - the SNES context 20627971a8bfSPeter Brune 20637971a8bfSPeter Brune Output Parameter: 20640298fd71SBarry Smith + r - the function (or NULL) 2065f8b49ee9SBarry Smith . f - the function (or NULL); see SNESFunction for calling sequence details 2066e4357dc4SBarry Smith . Amat - the matrix used to defined the operation A(x) x - b(x) (or NULL) 2067e4357dc4SBarry Smith . Pmat - the matrix from which the preconditioner will be constructed (or NULL) 2068f8b49ee9SBarry Smith . J - the function for matrix evaluation (or NULL); see SNESJacobianFunction for calling sequence details 20690298fd71SBarry Smith - ctx - the function context (or NULL) 20707971a8bfSPeter Brune 20717971a8bfSPeter Brune Level: advanced 20727971a8bfSPeter Brune 20737971a8bfSPeter Brune .keywords: SNES, nonlinear, get, function 20747971a8bfSPeter Brune 2075e4357dc4SBarry Smith .seealso: SNESSetPicard(), SNESGetFunction(), SNESGetJacobian(), SNESGetDM(), SNESFunction, SNESJacobianFunction 20767971a8bfSPeter Brune @*/ 2077d1e9a80fSBarry 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) 20787971a8bfSPeter Brune { 20797971a8bfSPeter Brune PetscErrorCode ierr; 20807971a8bfSPeter Brune DM dm; 20817971a8bfSPeter Brune 20827971a8bfSPeter Brune PetscFunctionBegin; 20837971a8bfSPeter Brune PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 20840298fd71SBarry Smith ierr = SNESGetFunction(snes,r,NULL,NULL);CHKERRQ(ierr); 2085e4357dc4SBarry Smith ierr = SNESGetJacobian(snes,Amat,Pmat,NULL,NULL);CHKERRQ(ierr); 20867971a8bfSPeter Brune ierr = SNESGetDM(snes,&dm);CHKERRQ(ierr); 2087f8b49ee9SBarry Smith ierr = DMSNESGetPicard(dm,f,J,ctx);CHKERRQ(ierr); 20887971a8bfSPeter Brune PetscFunctionReturn(0); 20897971a8bfSPeter Brune } 20907971a8bfSPeter Brune 2091d25893d9SBarry Smith /*@C 2092d25893d9SBarry Smith SNESSetComputeInitialGuess - Sets a routine used to compute an initial guess for the problem 2093d25893d9SBarry Smith 2094d25893d9SBarry Smith Logically Collective on SNES 2095d25893d9SBarry Smith 2096d25893d9SBarry Smith Input Parameters: 2097d25893d9SBarry Smith + snes - the SNES context 2098d25893d9SBarry Smith . func - function evaluation routine 2099d25893d9SBarry Smith - ctx - [optional] user-defined context for private data for the 21000298fd71SBarry Smith function evaluation routine (may be NULL) 2101d25893d9SBarry Smith 2102d25893d9SBarry Smith Calling sequence of func: 2103d25893d9SBarry Smith $ func (SNES snes,Vec x,void *ctx); 2104d25893d9SBarry Smith 2105d25893d9SBarry Smith . f - function vector 2106d25893d9SBarry Smith - ctx - optional user-defined function context 2107d25893d9SBarry Smith 2108d25893d9SBarry Smith Level: intermediate 2109d25893d9SBarry Smith 2110d25893d9SBarry Smith .keywords: SNES, nonlinear, set, function 2111d25893d9SBarry Smith 2112d25893d9SBarry Smith .seealso: SNESGetFunction(), SNESComputeFunction(), SNESSetJacobian() 2113d25893d9SBarry Smith @*/ 2114d25893d9SBarry Smith PetscErrorCode SNESSetComputeInitialGuess(SNES snes,PetscErrorCode (*func)(SNES,Vec,void*),void *ctx) 2115d25893d9SBarry Smith { 2116d25893d9SBarry Smith PetscFunctionBegin; 2117d25893d9SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 2118d25893d9SBarry Smith if (func) snes->ops->computeinitialguess = func; 2119d25893d9SBarry Smith if (ctx) snes->initialguessP = ctx; 2120d25893d9SBarry Smith PetscFunctionReturn(0); 2121d25893d9SBarry Smith } 2122d25893d9SBarry Smith 21233ab0aad5SBarry Smith /* --------------------------------------------------------------- */ 21241096aae1SMatthew Knepley /*@C 21251096aae1SMatthew Knepley SNESGetRhs - Gets the vector for solving F(x) = rhs. If rhs is not set 21261096aae1SMatthew Knepley it assumes a zero right hand side. 21271096aae1SMatthew Knepley 21283f9fe445SBarry Smith Logically Collective on SNES 21291096aae1SMatthew Knepley 21301096aae1SMatthew Knepley Input Parameter: 21311096aae1SMatthew Knepley . snes - the SNES context 21321096aae1SMatthew Knepley 21331096aae1SMatthew Knepley Output Parameter: 21340298fd71SBarry Smith . rhs - the right hand side vector or NULL if the right hand side vector is null 21351096aae1SMatthew Knepley 21361096aae1SMatthew Knepley Level: intermediate 21371096aae1SMatthew Knepley 21381096aae1SMatthew Knepley .keywords: SNES, nonlinear, get, function, right hand side 21391096aae1SMatthew Knepley 214085385478SLisandro Dalcin .seealso: SNESGetSolution(), SNESGetFunction(), SNESComputeFunction(), SNESSetJacobian(), SNESSetFunction() 21411096aae1SMatthew Knepley @*/ 21427087cfbeSBarry Smith PetscErrorCode SNESGetRhs(SNES snes,Vec *rhs) 21431096aae1SMatthew Knepley { 21441096aae1SMatthew Knepley PetscFunctionBegin; 21450700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 21461096aae1SMatthew Knepley PetscValidPointer(rhs,2); 214785385478SLisandro Dalcin *rhs = snes->vec_rhs; 21481096aae1SMatthew Knepley PetscFunctionReturn(0); 21491096aae1SMatthew Knepley } 21501096aae1SMatthew Knepley 21519b94acceSBarry Smith /*@ 2152bf388a1fSBarry Smith SNESComputeFunction - Calls the function that has been set with SNESSetFunction(). 21539b94acceSBarry Smith 2154c7afd0dbSLois Curfman McInnes Collective on SNES 2155c7afd0dbSLois Curfman McInnes 21569b94acceSBarry Smith Input Parameters: 2157c7afd0dbSLois Curfman McInnes + snes - the SNES context 2158c7afd0dbSLois Curfman McInnes - x - input vector 21599b94acceSBarry Smith 21609b94acceSBarry Smith Output Parameter: 21613638b69dSLois Curfman McInnes . y - function vector, as set by SNESSetFunction() 21629b94acceSBarry Smith 21631bffabb2SLois Curfman McInnes Notes: 216436851e7fSLois Curfman McInnes SNESComputeFunction() is typically used within nonlinear solvers 216536851e7fSLois Curfman McInnes implementations, so most users would not generally call this routine 216636851e7fSLois Curfman McInnes themselves. 216736851e7fSLois Curfman McInnes 216836851e7fSLois Curfman McInnes Level: developer 216936851e7fSLois Curfman McInnes 21709b94acceSBarry Smith .keywords: SNES, nonlinear, compute, function 21719b94acceSBarry Smith 2172a86d99e1SLois Curfman McInnes .seealso: SNESSetFunction(), SNESGetFunction() 21739b94acceSBarry Smith @*/ 21747087cfbeSBarry Smith PetscErrorCode SNESComputeFunction(SNES snes,Vec x,Vec y) 21759b94acceSBarry Smith { 2176dfbe8321SBarry Smith PetscErrorCode ierr; 21776cab3a1bSJed Brown DM dm; 2178942e3340SBarry Smith DMSNES sdm; 21799b94acceSBarry Smith 21803a40ed3dSBarry Smith PetscFunctionBegin; 21810700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 21820700a824SBarry Smith PetscValidHeaderSpecific(x,VEC_CLASSID,2); 21830700a824SBarry Smith PetscValidHeaderSpecific(y,VEC_CLASSID,3); 2184c9780b6fSBarry Smith PetscCheckSameComm(snes,1,x,2); 2185c9780b6fSBarry Smith PetscCheckSameComm(snes,1,y,3); 218662796dfbSBarry Smith ierr = VecValidValues(x,2,PETSC_TRUE);CHKERRQ(ierr); 2187184914b5SBarry Smith 21886cab3a1bSJed Brown ierr = SNESGetDM(snes,&dm);CHKERRQ(ierr); 2189942e3340SBarry Smith ierr = DMGetDMSNES(dm,&sdm);CHKERRQ(ierr); 219032f3f7c2SPeter Brune if (sdm->ops->computefunction) { 219194db00ebSBarry Smith if (sdm->ops->computefunction != SNESObjectiveComputeFunctionDefaultFD) { 2192ccf3c845SPeter Brune ierr = PetscLogEventBegin(SNES_FunctionEval,snes,x,y,0);CHKERRQ(ierr); 219394db00ebSBarry Smith } 21945edff71fSBarry Smith ierr = VecLockPush(x);CHKERRQ(ierr); 2195d64ed03dSBarry Smith PetscStackPush("SNES user function"); 219622c6f798SBarry Smith ierr = (*sdm->ops->computefunction)(snes,x,y,sdm->functionctx);CHKERRQ(ierr); 2197d64ed03dSBarry Smith PetscStackPop; 21985edff71fSBarry Smith ierr = VecLockPop(x);CHKERRQ(ierr); 219994db00ebSBarry Smith if (sdm->ops->computefunction != SNESObjectiveComputeFunctionDefaultFD) { 2200ccf3c845SPeter Brune ierr = PetscLogEventEnd(SNES_FunctionEval,snes,x,y,0);CHKERRQ(ierr); 220194db00ebSBarry Smith } 2202c90fad12SPeter Brune } else if (snes->vec_rhs) { 2203c90fad12SPeter Brune ierr = MatMult(snes->jacobian, x, y);CHKERRQ(ierr); 2204644e2e5bSBarry Smith } else SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE, "Must call SNESSetFunction() or SNESSetDM() before SNESComputeFunction(), likely called from SNESSolve()."); 220585385478SLisandro Dalcin if (snes->vec_rhs) { 220685385478SLisandro Dalcin ierr = VecAXPY(y,-1.0,snes->vec_rhs);CHKERRQ(ierr); 22073ab0aad5SBarry Smith } 2208ae3c334cSLois Curfman McInnes snes->nfuncs++; 2209422a814eSBarry Smith /* 2210422a814eSBarry Smith domainerror might not be set on all processes; so we tag vector locally with Inf and the next inner product or norm will 2211422a814eSBarry Smith propagate the value to all processes 2212422a814eSBarry Smith */ 2213422a814eSBarry Smith if (snes->domainerror) { 2214422a814eSBarry Smith ierr = VecSetInf(y);CHKERRQ(ierr); 2215422a814eSBarry Smith } 22163a40ed3dSBarry Smith PetscFunctionReturn(0); 22179b94acceSBarry Smith } 22189b94acceSBarry Smith 2219c79ef259SPeter Brune /*@ 2220be95d8f1SBarry Smith SNESComputeNGS - Calls the Gauss-Seidel function that has been set with SNESSetNGS(). 2221c79ef259SPeter Brune 2222c79ef259SPeter Brune Collective on SNES 2223c79ef259SPeter Brune 2224c79ef259SPeter Brune Input Parameters: 2225c79ef259SPeter Brune + snes - the SNES context 2226c79ef259SPeter Brune . x - input vector 2227c79ef259SPeter Brune - b - rhs vector 2228c79ef259SPeter Brune 2229c79ef259SPeter Brune Output Parameter: 2230c79ef259SPeter Brune . x - new solution vector 2231c79ef259SPeter Brune 2232c79ef259SPeter Brune Notes: 2233be95d8f1SBarry Smith SNESComputeNGS() is typically used within composed nonlinear solver 2234c79ef259SPeter Brune implementations, so most users would not generally call this routine 2235c79ef259SPeter Brune themselves. 2236c79ef259SPeter Brune 2237c79ef259SPeter Brune Level: developer 2238c79ef259SPeter Brune 2239c79ef259SPeter Brune .keywords: SNES, nonlinear, compute, function 2240c79ef259SPeter Brune 2241be95d8f1SBarry Smith .seealso: SNESSetNGS(), SNESComputeFunction() 2242c79ef259SPeter Brune @*/ 2243be95d8f1SBarry Smith PetscErrorCode SNESComputeNGS(SNES snes,Vec b,Vec x) 2244646217ecSPeter Brune { 2245646217ecSPeter Brune PetscErrorCode ierr; 22466cab3a1bSJed Brown DM dm; 2247942e3340SBarry Smith DMSNES sdm; 2248646217ecSPeter Brune 2249646217ecSPeter Brune PetscFunctionBegin; 2250646217ecSPeter Brune PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 2251646217ecSPeter Brune PetscValidHeaderSpecific(x,VEC_CLASSID,2); 2252646217ecSPeter Brune if (b) PetscValidHeaderSpecific(b,VEC_CLASSID,3); 2253646217ecSPeter Brune PetscCheckSameComm(snes,1,x,2); 2254646217ecSPeter Brune if (b) PetscCheckSameComm(snes,1,b,3); 225562796dfbSBarry Smith if (b) {ierr = VecValidValues(b,2,PETSC_TRUE);CHKERRQ(ierr);} 2256be95d8f1SBarry Smith ierr = PetscLogEventBegin(SNES_NGSEval,snes,x,b,0);CHKERRQ(ierr); 22576cab3a1bSJed Brown ierr = SNESGetDM(snes,&dm);CHKERRQ(ierr); 2258942e3340SBarry Smith ierr = DMGetDMSNES(dm,&sdm);CHKERRQ(ierr); 225922c6f798SBarry Smith if (sdm->ops->computegs) { 22605edff71fSBarry Smith if (b) {ierr = VecLockPush(b);CHKERRQ(ierr);} 2261be95d8f1SBarry Smith PetscStackPush("SNES user NGS"); 226222c6f798SBarry Smith ierr = (*sdm->ops->computegs)(snes,x,b,sdm->gsctx);CHKERRQ(ierr); 2263646217ecSPeter Brune PetscStackPop; 22645edff71fSBarry Smith if (b) {ierr = VecLockPop(b);CHKERRQ(ierr);} 2265be95d8f1SBarry Smith } else SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE, "Must call SNESSetNGS() before SNESComputeNGS(), likely called from SNESSolve()."); 2266be95d8f1SBarry Smith ierr = PetscLogEventEnd(SNES_NGSEval,snes,x,b,0);CHKERRQ(ierr); 2267646217ecSPeter Brune PetscFunctionReturn(0); 2268646217ecSPeter Brune } 2269646217ecSPeter Brune 227062fef451SLois Curfman McInnes /*@ 2271bf388a1fSBarry Smith SNESComputeJacobian - Computes the Jacobian matrix that has been set with SNESSetJacobian(). 227262fef451SLois Curfman McInnes 2273c7afd0dbSLois Curfman McInnes Collective on SNES and Mat 2274c7afd0dbSLois Curfman McInnes 227562fef451SLois Curfman McInnes Input Parameters: 2276c7afd0dbSLois Curfman McInnes + snes - the SNES context 2277c7afd0dbSLois Curfman McInnes - x - input vector 227862fef451SLois Curfman McInnes 227962fef451SLois Curfman McInnes Output Parameters: 2280c7afd0dbSLois Curfman McInnes + A - Jacobian matrix 2281d1e9a80fSBarry Smith - B - optional preconditioning matrix 2282fee21e36SBarry Smith 2283e35cf81dSBarry Smith Options Database Keys: 2284e35cf81dSBarry Smith + -snes_lag_preconditioner <lag> 2285693365a8SJed Brown . -snes_lag_jacobian <lag> 2286693365a8SJed Brown . -snes_compare_explicit - Compare the computed Jacobian to the finite difference Jacobian and output the differences 2287693365a8SJed Brown . -snes_compare_explicit_draw - Compare the computed Jacobian to the finite difference Jacobian and draw the result 2288693365a8SJed Brown . -snes_compare_explicit_contour - Compare the computed Jacobian to the finite difference Jacobian and draw a contour plot with the result 22894c30e9fbSJed Brown . -snes_compare_operator - Make the comparison options above use the operator instead of the preconditioning matrix 229094d6a431SBarry Smith . -snes_compare_coloring - Compute the finite difference Jacobian using coloring and display norms of difference 2291c01495d3SJed Brown . -snes_compare_coloring_display - Compute the finite differece Jacobian using coloring and display verbose differences 2292c01495d3SJed Brown . -snes_compare_coloring_threshold - Display only those matrix entries that differ by more than a given threshold 2293c01495d3SJed Brown . -snes_compare_coloring_threshold_atol - Absolute tolerance for difference in matrix entries to be displayed by -snes_compare_coloring_threshold 2294c01495d3SJed Brown . -snes_compare_coloring_threshold_rtol - Relative tolerance for difference in matrix entries to be displayed by -snes_compare_coloring_threshold 22954c30e9fbSJed Brown . -snes_compare_coloring_draw - Compute the finite differece Jacobian using coloring and draw differences 2296c01495d3SJed Brown - -snes_compare_coloring_draw_contour - Compute the finite differece Jacobian using coloring and show contours of matrices and differences 2297c01495d3SJed Brown 2298e35cf81dSBarry Smith 229962fef451SLois Curfman McInnes Notes: 230062fef451SLois Curfman McInnes Most users should not need to explicitly call this routine, as it 230162fef451SLois Curfman McInnes is used internally within the nonlinear solvers. 230262fef451SLois Curfman McInnes 230336851e7fSLois Curfman McInnes Level: developer 230436851e7fSLois Curfman McInnes 230562fef451SLois Curfman McInnes .keywords: SNES, compute, Jacobian, matrix 230662fef451SLois Curfman McInnes 2307e35cf81dSBarry Smith .seealso: SNESSetJacobian(), KSPSetOperators(), MatStructure, SNESSetLagPreconditioner(), SNESSetLagJacobian() 230862fef451SLois Curfman McInnes @*/ 2309d1e9a80fSBarry Smith PetscErrorCode SNESComputeJacobian(SNES snes,Vec X,Mat A,Mat B) 23109b94acceSBarry Smith { 2311dfbe8321SBarry Smith PetscErrorCode ierr; 2312ace3abfcSBarry Smith PetscBool flag; 23136cab3a1bSJed Brown DM dm; 2314942e3340SBarry Smith DMSNES sdm; 2315e0e3a89bSBarry Smith KSP ksp; 23163a40ed3dSBarry Smith 23173a40ed3dSBarry Smith PetscFunctionBegin; 23180700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 23190700a824SBarry Smith PetscValidHeaderSpecific(X,VEC_CLASSID,2); 2320c9780b6fSBarry Smith PetscCheckSameComm(snes,1,X,2); 232162796dfbSBarry Smith ierr = VecValidValues(X,2,PETSC_TRUE);CHKERRQ(ierr); 23226cab3a1bSJed Brown ierr = SNESGetDM(snes,&dm);CHKERRQ(ierr); 2323942e3340SBarry Smith ierr = DMGetDMSNES(dm,&sdm);CHKERRQ(ierr); 23243232da50SPeter Brune 2325ce94432eSBarry Smith if (!sdm->ops->computejacobian) SETERRQ(PetscObjectComm((PetscObject)snes),PETSC_ERR_USER,"Must call SNESSetJacobian(), DMSNESSetJacobian(), DMDASNESSetJacobianLocal(), etc"); 2326ebd3b9afSBarry Smith 2327ebd3b9afSBarry Smith /* make sure that MatAssemblyBegin/End() is called on A matrix if it is matrix free */ 2328ebd3b9afSBarry Smith 2329fe3ffe1eSBarry Smith if (snes->lagjacobian == -2) { 2330fe3ffe1eSBarry Smith snes->lagjacobian = -1; 2331f5af7f23SKarl Rupp 2332fe3ffe1eSBarry Smith ierr = PetscInfo(snes,"Recomputing Jacobian/preconditioner because lag is -2 (means compute Jacobian, but then never again) \n");CHKERRQ(ierr); 2333fe3ffe1eSBarry Smith } else if (snes->lagjacobian == -1) { 2334e35cf81dSBarry Smith ierr = PetscInfo(snes,"Reusing Jacobian/preconditioner because lag is -1\n");CHKERRQ(ierr); 233594ab13aaSBarry Smith ierr = PetscObjectTypeCompare((PetscObject)A,MATMFFD,&flag);CHKERRQ(ierr); 2336ebd3b9afSBarry Smith if (flag) { 233794ab13aaSBarry Smith ierr = MatAssemblyBegin(A,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 233894ab13aaSBarry Smith ierr = MatAssemblyEnd(A,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 2339ebd3b9afSBarry Smith } 2340e35cf81dSBarry Smith PetscFunctionReturn(0); 234137ec4e1aSPeter Brune } else if (snes->lagjacobian > 1 && (snes->iter + snes->jac_iter) % snes->lagjacobian) { 2342e35cf81dSBarry Smith ierr = PetscInfo2(snes,"Reusing Jacobian/preconditioner because lag is %D and SNES iteration is %D\n",snes->lagjacobian,snes->iter);CHKERRQ(ierr); 234394ab13aaSBarry Smith ierr = PetscObjectTypeCompare((PetscObject)A,MATMFFD,&flag);CHKERRQ(ierr); 2344ebd3b9afSBarry Smith if (flag) { 234594ab13aaSBarry Smith ierr = MatAssemblyBegin(A,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 234694ab13aaSBarry Smith ierr = MatAssemblyEnd(A,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 2347ebd3b9afSBarry Smith } 2348e35cf81dSBarry Smith PetscFunctionReturn(0); 2349e35cf81dSBarry Smith } 2350efd4aadfSBarry Smith if (snes->npc && snes->npcside== PC_LEFT) { 235194ab13aaSBarry Smith ierr = MatAssemblyBegin(A,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 235294ab13aaSBarry Smith ierr = MatAssemblyEnd(A,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 2353d728fb7dSPeter Brune PetscFunctionReturn(0); 2354d728fb7dSPeter Brune } 2355e35cf81dSBarry Smith 235694ab13aaSBarry Smith ierr = PetscLogEventBegin(SNES_JacobianEval,snes,X,A,B);CHKERRQ(ierr); 23575edff71fSBarry Smith ierr = VecLockPush(X);CHKERRQ(ierr); 2358d64ed03dSBarry Smith PetscStackPush("SNES user Jacobian function"); 2359d1e9a80fSBarry Smith ierr = (*sdm->ops->computejacobian)(snes,X,A,B,sdm->jacobianctx);CHKERRQ(ierr); 2360d64ed03dSBarry Smith PetscStackPop; 23615edff71fSBarry Smith ierr = VecLockPop(X);CHKERRQ(ierr); 236294ab13aaSBarry Smith ierr = PetscLogEventEnd(SNES_JacobianEval,snes,X,A,B);CHKERRQ(ierr); 2363a8054027SBarry Smith 2364e0e3a89bSBarry Smith /* the next line ensures that snes->ksp exists */ 2365e0e3a89bSBarry Smith ierr = SNESGetKSP(snes,&ksp);CHKERRQ(ierr); 23663b4f5425SBarry Smith if (snes->lagpreconditioner == -2) { 23673b4f5425SBarry Smith ierr = PetscInfo(snes,"Rebuilding preconditioner exactly once since lag is -2\n");CHKERRQ(ierr); 2368d1e9a80fSBarry Smith ierr = KSPSetReusePreconditioner(snes->ksp,PETSC_FALSE);CHKERRQ(ierr); 23693b4f5425SBarry Smith snes->lagpreconditioner = -1; 23703b4f5425SBarry Smith } else if (snes->lagpreconditioner == -1) { 2371a8054027SBarry Smith ierr = PetscInfo(snes,"Reusing preconditioner because lag is -1\n");CHKERRQ(ierr); 2372d1e9a80fSBarry Smith ierr = KSPSetReusePreconditioner(snes->ksp,PETSC_TRUE);CHKERRQ(ierr); 237337ec4e1aSPeter Brune } else if (snes->lagpreconditioner > 1 && (snes->iter + snes->pre_iter) % snes->lagpreconditioner) { 2374a8054027SBarry Smith ierr = PetscInfo2(snes,"Reusing preconditioner because lag is %D and SNES iteration is %D\n",snes->lagpreconditioner,snes->iter);CHKERRQ(ierr); 2375d1e9a80fSBarry Smith ierr = KSPSetReusePreconditioner(snes->ksp,PETSC_TRUE);CHKERRQ(ierr); 2376d1e9a80fSBarry Smith } else { 2377d1e9a80fSBarry Smith ierr = PetscInfo(snes,"Rebuilding preconditioner\n");CHKERRQ(ierr); 2378d1e9a80fSBarry Smith ierr = KSPSetReusePreconditioner(snes->ksp,PETSC_FALSE);CHKERRQ(ierr); 2379a8054027SBarry Smith } 2380a8054027SBarry Smith 23816d84be18SBarry Smith /* make sure user returned a correct Jacobian and preconditioner */ 238294ab13aaSBarry Smith /* PetscValidHeaderSpecific(A,MAT_CLASSID,3); 238394ab13aaSBarry Smith PetscValidHeaderSpecific(B,MAT_CLASSID,4); */ 2384693365a8SJed Brown { 2385693365a8SJed Brown PetscBool flag = PETSC_FALSE,flag_draw = PETSC_FALSE,flag_contour = PETSC_FALSE,flag_operator = PETSC_FALSE; 238627b0f280SBarry Smith ierr = PetscOptionsGetViewer(PetscObjectComm((PetscObject)snes),((PetscObject)snes)->prefix,"-snes_compare_explicit",NULL,NULL,&flag);CHKERRQ(ierr); 238727b0f280SBarry Smith ierr = PetscOptionsGetViewer(PetscObjectComm((PetscObject)snes),((PetscObject)snes)->prefix,"-snes_compare_explicit_draw",NULL,NULL,&flag_draw);CHKERRQ(ierr); 238827b0f280SBarry Smith ierr = PetscOptionsGetViewer(PetscObjectComm((PetscObject)snes),((PetscObject)snes)->prefix,"-snes_compare_explicit_draw_contour",NULL,NULL,&flag_contour);CHKERRQ(ierr); 238927b0f280SBarry Smith ierr = PetscOptionsGetViewer(PetscObjectComm((PetscObject)snes),((PetscObject)snes)->prefix,"-snes_compare_operator",NULL,NULL,&flag_operator);CHKERRQ(ierr); 2390693365a8SJed Brown if (flag || flag_draw || flag_contour) { 23910298fd71SBarry Smith Mat Bexp_mine = NULL,Bexp,FDexp; 2392693365a8SJed Brown PetscViewer vdraw,vstdout; 23936b3a5b13SJed Brown PetscBool flg; 2394693365a8SJed Brown if (flag_operator) { 239594ab13aaSBarry Smith ierr = MatComputeExplicitOperator(A,&Bexp_mine);CHKERRQ(ierr); 2396693365a8SJed Brown Bexp = Bexp_mine; 2397693365a8SJed Brown } else { 2398693365a8SJed Brown /* See if the preconditioning matrix can be viewed and added directly */ 239994ab13aaSBarry Smith ierr = PetscObjectTypeCompareAny((PetscObject)B,&flg,MATSEQAIJ,MATMPIAIJ,MATSEQDENSE,MATMPIDENSE,MATSEQBAIJ,MATMPIBAIJ,MATSEQSBAIJ,MATMPIBAIJ,"");CHKERRQ(ierr); 240094ab13aaSBarry Smith if (flg) Bexp = B; 2401693365a8SJed Brown else { 2402693365a8SJed Brown /* If the "preconditioning" matrix is itself MATSHELL or some other type without direct support */ 240394ab13aaSBarry Smith ierr = MatComputeExplicitOperator(B,&Bexp_mine);CHKERRQ(ierr); 2404693365a8SJed Brown Bexp = Bexp_mine; 2405693365a8SJed Brown } 2406693365a8SJed Brown } 2407693365a8SJed Brown ierr = MatConvert(Bexp,MATSAME,MAT_INITIAL_MATRIX,&FDexp);CHKERRQ(ierr); 2408d1e9a80fSBarry Smith ierr = SNESComputeJacobianDefault(snes,X,FDexp,FDexp,NULL);CHKERRQ(ierr); 2409ce94432eSBarry Smith ierr = PetscViewerASCIIGetStdout(PetscObjectComm((PetscObject)snes),&vstdout);CHKERRQ(ierr); 2410693365a8SJed Brown if (flag_draw || flag_contour) { 2411ce94432eSBarry Smith ierr = PetscViewerDrawOpen(PetscObjectComm((PetscObject)snes),0,"Explicit Jacobians",PETSC_DECIDE,PETSC_DECIDE,300,300,&vdraw);CHKERRQ(ierr); 2412693365a8SJed Brown if (flag_contour) {ierr = PetscViewerPushFormat(vdraw,PETSC_VIEWER_DRAW_CONTOUR);CHKERRQ(ierr);} 24130298fd71SBarry Smith } else vdraw = NULL; 2414693365a8SJed Brown ierr = PetscViewerASCIIPrintf(vstdout,"Explicit %s\n",flag_operator ? "Jacobian" : "preconditioning Jacobian");CHKERRQ(ierr); 2415693365a8SJed Brown if (flag) {ierr = MatView(Bexp,vstdout);CHKERRQ(ierr);} 2416693365a8SJed Brown if (vdraw) {ierr = MatView(Bexp,vdraw);CHKERRQ(ierr);} 2417693365a8SJed Brown ierr = PetscViewerASCIIPrintf(vstdout,"Finite difference Jacobian\n");CHKERRQ(ierr); 2418693365a8SJed Brown if (flag) {ierr = MatView(FDexp,vstdout);CHKERRQ(ierr);} 2419693365a8SJed Brown if (vdraw) {ierr = MatView(FDexp,vdraw);CHKERRQ(ierr);} 2420693365a8SJed Brown ierr = MatAYPX(FDexp,-1.0,Bexp,SAME_NONZERO_PATTERN);CHKERRQ(ierr); 2421693365a8SJed Brown ierr = PetscViewerASCIIPrintf(vstdout,"User-provided matrix minus finite difference Jacobian\n");CHKERRQ(ierr); 2422693365a8SJed Brown if (flag) {ierr = MatView(FDexp,vstdout);CHKERRQ(ierr);} 2423693365a8SJed Brown if (vdraw) { /* Always use contour for the difference */ 2424693365a8SJed Brown ierr = PetscViewerPushFormat(vdraw,PETSC_VIEWER_DRAW_CONTOUR);CHKERRQ(ierr); 2425693365a8SJed Brown ierr = MatView(FDexp,vdraw);CHKERRQ(ierr); 2426693365a8SJed Brown ierr = PetscViewerPopFormat(vdraw);CHKERRQ(ierr); 2427693365a8SJed Brown } 2428693365a8SJed Brown if (flag_contour) {ierr = PetscViewerPopFormat(vdraw);CHKERRQ(ierr);} 2429693365a8SJed Brown ierr = PetscViewerDestroy(&vdraw);CHKERRQ(ierr); 2430693365a8SJed Brown ierr = MatDestroy(&Bexp_mine);CHKERRQ(ierr); 2431693365a8SJed Brown ierr = MatDestroy(&FDexp);CHKERRQ(ierr); 2432693365a8SJed Brown } 2433693365a8SJed Brown } 24344c30e9fbSJed Brown { 24356719d8e4SJed Brown PetscBool flag = PETSC_FALSE,flag_display = PETSC_FALSE,flag_draw = PETSC_FALSE,flag_contour = PETSC_FALSE,flag_threshold = PETSC_FALSE; 24366719d8e4SJed Brown PetscReal threshold_atol = PETSC_SQRT_MACHINE_EPSILON,threshold_rtol = 10*PETSC_SQRT_MACHINE_EPSILON; 243727b0f280SBarry Smith ierr = PetscOptionsGetViewer(PetscObjectComm((PetscObject)snes),((PetscObject)snes)->prefix,"-snes_compare_coloring",NULL,NULL,&flag);CHKERRQ(ierr); 243827b0f280SBarry Smith ierr = PetscOptionsGetViewer(PetscObjectComm((PetscObject)snes),((PetscObject)snes)->prefix,"-snes_compare_coloring_display",NULL,NULL,&flag_display);CHKERRQ(ierr); 243927b0f280SBarry Smith ierr = PetscOptionsGetViewer(PetscObjectComm((PetscObject)snes),((PetscObject)snes)->prefix,"-snes_compare_coloring_draw",NULL,NULL,&flag_draw);CHKERRQ(ierr); 244027b0f280SBarry Smith ierr = PetscOptionsGetViewer(PetscObjectComm((PetscObject)snes),((PetscObject)snes)->prefix,"-snes_compare_coloring_draw_contour",NULL,NULL,&flag_contour);CHKERRQ(ierr); 244127b0f280SBarry Smith ierr = PetscOptionsGetViewer(PetscObjectComm((PetscObject)snes),((PetscObject)snes)->prefix,"-snes_compare_coloring_threshold",NULL,NULL,&flag_threshold);CHKERRQ(ierr); 244227b0f280SBarry Smith if (flag_threshold) { 2443c5929fdfSBarry Smith ierr = PetscOptionsGetReal(((PetscObject)snes)->options,((PetscObject)snes)->prefix,"-snes_compare_coloring_threshold_rtol",&threshold_rtol,NULL);CHKERRQ(ierr); 2444c5929fdfSBarry Smith ierr = PetscOptionsGetReal(((PetscObject)snes)->options,((PetscObject)snes)->prefix,"-snes_compare_coloring_threshold_atol",&threshold_atol,NULL);CHKERRQ(ierr); 244527b0f280SBarry Smith } 24466719d8e4SJed Brown if (flag || flag_display || flag_draw || flag_contour || flag_threshold) { 24474c30e9fbSJed Brown Mat Bfd; 24484c30e9fbSJed Brown PetscViewer vdraw,vstdout; 2449335efc43SPeter Brune MatColoring coloring; 24504c30e9fbSJed Brown ISColoring iscoloring; 24514c30e9fbSJed Brown MatFDColoring matfdcoloring; 24524c30e9fbSJed Brown PetscErrorCode (*func)(SNES,Vec,Vec,void*); 24534c30e9fbSJed Brown void *funcctx; 24546719d8e4SJed Brown PetscReal norm1,norm2,normmax; 24554c30e9fbSJed Brown 245694ab13aaSBarry Smith ierr = MatDuplicate(B,MAT_DO_NOT_COPY_VALUES,&Bfd);CHKERRQ(ierr); 2457335efc43SPeter Brune ierr = MatColoringCreate(Bfd,&coloring);CHKERRQ(ierr); 2458335efc43SPeter Brune ierr = MatColoringSetType(coloring,MATCOLORINGSL);CHKERRQ(ierr); 2459335efc43SPeter Brune ierr = MatColoringSetFromOptions(coloring);CHKERRQ(ierr); 2460335efc43SPeter Brune ierr = MatColoringApply(coloring,&iscoloring);CHKERRQ(ierr); 2461335efc43SPeter Brune ierr = MatColoringDestroy(&coloring);CHKERRQ(ierr); 24624c30e9fbSJed Brown ierr = MatFDColoringCreate(Bfd,iscoloring,&matfdcoloring);CHKERRQ(ierr); 2463f86b9fbaSHong Zhang ierr = MatFDColoringSetFromOptions(matfdcoloring);CHKERRQ(ierr); 2464f86b9fbaSHong Zhang ierr = MatFDColoringSetUp(Bfd,iscoloring,matfdcoloring);CHKERRQ(ierr); 24654c30e9fbSJed Brown ierr = ISColoringDestroy(&iscoloring);CHKERRQ(ierr); 24664c30e9fbSJed Brown 24674c30e9fbSJed Brown /* This method of getting the function is currently unreliable since it doesn't work for DM local functions. */ 24680298fd71SBarry Smith ierr = SNESGetFunction(snes,NULL,&func,&funcctx);CHKERRQ(ierr); 24694c30e9fbSJed Brown ierr = MatFDColoringSetFunction(matfdcoloring,(PetscErrorCode (*)(void))func,funcctx);CHKERRQ(ierr); 24704c30e9fbSJed Brown ierr = PetscObjectSetOptionsPrefix((PetscObject)matfdcoloring,((PetscObject)snes)->prefix);CHKERRQ(ierr); 24714c30e9fbSJed Brown ierr = PetscObjectAppendOptionsPrefix((PetscObject)matfdcoloring,"coloring_");CHKERRQ(ierr); 24724c30e9fbSJed Brown ierr = MatFDColoringSetFromOptions(matfdcoloring);CHKERRQ(ierr); 2473d1e9a80fSBarry Smith ierr = MatFDColoringApply(Bfd,matfdcoloring,X,snes);CHKERRQ(ierr); 24744c30e9fbSJed Brown ierr = MatFDColoringDestroy(&matfdcoloring);CHKERRQ(ierr); 24754c30e9fbSJed Brown 2476ce94432eSBarry Smith ierr = PetscViewerASCIIGetStdout(PetscObjectComm((PetscObject)snes),&vstdout);CHKERRQ(ierr); 24774c30e9fbSJed Brown if (flag_draw || flag_contour) { 2478ce94432eSBarry Smith ierr = PetscViewerDrawOpen(PetscObjectComm((PetscObject)snes),0,"Colored Jacobians",PETSC_DECIDE,PETSC_DECIDE,300,300,&vdraw);CHKERRQ(ierr); 24794c30e9fbSJed Brown if (flag_contour) {ierr = PetscViewerPushFormat(vdraw,PETSC_VIEWER_DRAW_CONTOUR);CHKERRQ(ierr);} 24800298fd71SBarry Smith } else vdraw = NULL; 24814c30e9fbSJed Brown ierr = PetscViewerASCIIPrintf(vstdout,"Explicit preconditioning Jacobian\n");CHKERRQ(ierr); 248294ab13aaSBarry Smith if (flag_display) {ierr = MatView(B,vstdout);CHKERRQ(ierr);} 248394ab13aaSBarry Smith if (vdraw) {ierr = MatView(B,vdraw);CHKERRQ(ierr);} 24844c30e9fbSJed Brown ierr = PetscViewerASCIIPrintf(vstdout,"Colored Finite difference Jacobian\n");CHKERRQ(ierr); 24856719d8e4SJed Brown if (flag_display) {ierr = MatView(Bfd,vstdout);CHKERRQ(ierr);} 24864c30e9fbSJed Brown if (vdraw) {ierr = MatView(Bfd,vdraw);CHKERRQ(ierr);} 248794ab13aaSBarry Smith ierr = MatAYPX(Bfd,-1.0,B,SAME_NONZERO_PATTERN);CHKERRQ(ierr); 24884c30e9fbSJed Brown ierr = MatNorm(Bfd,NORM_1,&norm1);CHKERRQ(ierr); 24896719d8e4SJed Brown ierr = MatNorm(Bfd,NORM_FROBENIUS,&norm2);CHKERRQ(ierr); 24904c30e9fbSJed Brown ierr = MatNorm(Bfd,NORM_MAX,&normmax);CHKERRQ(ierr); 249157622a8eSBarry 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); 24926719d8e4SJed Brown if (flag_display) {ierr = MatView(Bfd,vstdout);CHKERRQ(ierr);} 24934c30e9fbSJed Brown if (vdraw) { /* Always use contour for the difference */ 24944c30e9fbSJed Brown ierr = PetscViewerPushFormat(vdraw,PETSC_VIEWER_DRAW_CONTOUR);CHKERRQ(ierr); 24954c30e9fbSJed Brown ierr = MatView(Bfd,vdraw);CHKERRQ(ierr); 24964c30e9fbSJed Brown ierr = PetscViewerPopFormat(vdraw);CHKERRQ(ierr); 24974c30e9fbSJed Brown } 24984c30e9fbSJed Brown if (flag_contour) {ierr = PetscViewerPopFormat(vdraw);CHKERRQ(ierr);} 24996719d8e4SJed Brown 25006719d8e4SJed Brown if (flag_threshold) { 25016719d8e4SJed Brown PetscInt bs,rstart,rend,i; 250294ab13aaSBarry Smith ierr = MatGetBlockSize(B,&bs);CHKERRQ(ierr); 250394ab13aaSBarry Smith ierr = MatGetOwnershipRange(B,&rstart,&rend);CHKERRQ(ierr); 25046719d8e4SJed Brown for (i=rstart; i<rend; i++) { 25056719d8e4SJed Brown const PetscScalar *ba,*ca; 25066719d8e4SJed Brown const PetscInt *bj,*cj; 25076719d8e4SJed Brown PetscInt bn,cn,j,maxentrycol = -1,maxdiffcol = -1,maxrdiffcol = -1; 25086719d8e4SJed Brown PetscReal maxentry = 0,maxdiff = 0,maxrdiff = 0; 250994ab13aaSBarry Smith ierr = MatGetRow(B,i,&bn,&bj,&ba);CHKERRQ(ierr); 25106719d8e4SJed Brown ierr = MatGetRow(Bfd,i,&cn,&cj,&ca);CHKERRQ(ierr); 251194ab13aaSBarry Smith if (bn != cn) SETERRQ(((PetscObject)A)->comm,PETSC_ERR_PLIB,"Unexpected different nonzero pattern in -snes_compare_coloring_threshold"); 25126719d8e4SJed Brown for (j=0; j<bn; j++) { 25136719d8e4SJed Brown PetscReal rdiff = PetscAbsScalar(ca[j]) / (threshold_atol + threshold_rtol*PetscAbsScalar(ba[j])); 25146719d8e4SJed Brown if (PetscAbsScalar(ba[j]) > PetscAbs(maxentry)) { 25156719d8e4SJed Brown maxentrycol = bj[j]; 25166719d8e4SJed Brown maxentry = PetscRealPart(ba[j]); 25176719d8e4SJed Brown } 25186719d8e4SJed Brown if (PetscAbsScalar(ca[j]) > PetscAbs(maxdiff)) { 25196719d8e4SJed Brown maxdiffcol = bj[j]; 25206719d8e4SJed Brown maxdiff = PetscRealPart(ca[j]); 25216719d8e4SJed Brown } 25226719d8e4SJed Brown if (rdiff > maxrdiff) { 25236719d8e4SJed Brown maxrdiffcol = bj[j]; 25246719d8e4SJed Brown maxrdiff = rdiff; 25256719d8e4SJed Brown } 25266719d8e4SJed Brown } 25276719d8e4SJed Brown if (maxrdiff > 1) { 252857622a8eSBarry 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); 25296719d8e4SJed Brown for (j=0; j<bn; j++) { 25306719d8e4SJed Brown PetscReal rdiff; 25316719d8e4SJed Brown rdiff = PetscAbsScalar(ca[j]) / (threshold_atol + threshold_rtol*PetscAbsScalar(ba[j])); 25326719d8e4SJed Brown if (rdiff > 1) { 253357622a8eSBarry Smith ierr = PetscViewerASCIIPrintf(vstdout," (%D,%g:%g)",bj[j],(double)PetscRealPart(ba[j]),(double)PetscRealPart(ca[j]));CHKERRQ(ierr); 25346719d8e4SJed Brown } 25356719d8e4SJed Brown } 25366719d8e4SJed Brown ierr = PetscViewerASCIIPrintf(vstdout,"\n",i,maxentry,maxdiff,maxrdiff);CHKERRQ(ierr); 25376719d8e4SJed Brown } 253894ab13aaSBarry Smith ierr = MatRestoreRow(B,i,&bn,&bj,&ba);CHKERRQ(ierr); 25396719d8e4SJed Brown ierr = MatRestoreRow(Bfd,i,&cn,&cj,&ca);CHKERRQ(ierr); 25406719d8e4SJed Brown } 25416719d8e4SJed Brown } 25424c30e9fbSJed Brown ierr = PetscViewerDestroy(&vdraw);CHKERRQ(ierr); 25434c30e9fbSJed Brown ierr = MatDestroy(&Bfd);CHKERRQ(ierr); 25444c30e9fbSJed Brown } 25454c30e9fbSJed Brown } 25463a40ed3dSBarry Smith PetscFunctionReturn(0); 25479b94acceSBarry Smith } 25489b94acceSBarry Smith 2549bf388a1fSBarry Smith /*MC 2550411c0326SBarry Smith SNESJacobianFunction - Function used to convey the nonlinear Jacobian of the function to be solved by SNES 2551bf388a1fSBarry Smith 2552bf388a1fSBarry Smith Synopsis: 2553411c0326SBarry Smith #include "petscsnes.h" 2554411c0326SBarry Smith PetscErrorCode SNESJacobianFunction(SNES snes,Vec x,Mat Amat,Mat Pmat,void *ctx); 2555bf388a1fSBarry Smith 2556bf388a1fSBarry Smith + x - input vector 2557e5d3d808SBarry Smith . Amat - the matrix that defines the (approximate) Jacobian 2558e5d3d808SBarry Smith . Pmat - the matrix to be used in constructing the preconditioner, usually the same as Amat. 2559bf388a1fSBarry Smith - ctx - [optional] user-defined Jacobian context 2560bf388a1fSBarry Smith 2561878cb397SSatish Balay Level: intermediate 2562878cb397SSatish Balay 2563bf388a1fSBarry Smith .seealso: SNESSetFunction(), SNESGetFunction(), SNESSetJacobian(), SNESGetJacobian() 2564bf388a1fSBarry Smith M*/ 2565bf388a1fSBarry Smith 25669b94acceSBarry Smith /*@C 25679b94acceSBarry Smith SNESSetJacobian - Sets the function to compute Jacobian as well as the 2568044dda88SLois Curfman McInnes location to store the matrix. 25699b94acceSBarry Smith 25703f9fe445SBarry Smith Logically Collective on SNES and Mat 2571c7afd0dbSLois Curfman McInnes 25729b94acceSBarry Smith Input Parameters: 2573c7afd0dbSLois Curfman McInnes + snes - the SNES context 2574e5d3d808SBarry Smith . Amat - the matrix that defines the (approximate) Jacobian 2575e5d3d808SBarry Smith . Pmat - the matrix to be used in constructing the preconditioner, usually the same as Amat. 2576411c0326SBarry Smith . J - Jacobian evaluation routine (if NULL then SNES retains any previously set value), see SNESJacobianFunction for details 2577c7afd0dbSLois Curfman McInnes - ctx - [optional] user-defined context for private data for the 25780298fd71SBarry Smith Jacobian evaluation routine (may be NULL) (if NULL then SNES retains any previously set value) 25799b94acceSBarry Smith 25809b94acceSBarry Smith Notes: 2581e5d3d808SBarry Smith If the Amat matrix and Pmat matrix are different you must call MatAssemblyBegin/End() on 258216913363SBarry Smith each matrix. 258316913363SBarry Smith 2584895c21f2SBarry Smith If you know the operator Amat has a null space you can use MatSetNullSpace() and MatSetTransposeNullSpace() to supply the null 2585895c21f2SBarry Smith space to Amat and the KSP solvers will automatically use that null space as needed during the solution process. 2586895c21f2SBarry Smith 25878d359177SBarry Smith If using SNESComputeJacobianDefaultColor() to assemble a Jacobian, the ctx argument 2588a8a26c1eSJed Brown must be a MatFDColoring. 2589a8a26c1eSJed Brown 2590c3cc8fd1SJed Brown Other defect-correction schemes can be used by computing a different matrix in place of the Jacobian. One common 2591c3cc8fd1SJed Brown example is to use the "Picard linearization" which only differentiates through the highest order parts of each term. 2592c3cc8fd1SJed Brown 259336851e7fSLois Curfman McInnes Level: beginner 259436851e7fSLois Curfman McInnes 25959b94acceSBarry Smith .keywords: SNES, nonlinear, set, Jacobian, matrix 25969b94acceSBarry Smith 2597411c0326SBarry Smith .seealso: KSPSetOperators(), SNESSetFunction(), MatMFFDComputeJacobian(), SNESComputeJacobianDefaultColor(), MatStructure, J, 2598411c0326SBarry Smith SNESSetPicard(), SNESJacobianFunction 25999b94acceSBarry Smith @*/ 2600d1e9a80fSBarry Smith PetscErrorCode SNESSetJacobian(SNES snes,Mat Amat,Mat Pmat,PetscErrorCode (*J)(SNES,Vec,Mat,Mat,void*),void *ctx) 26019b94acceSBarry Smith { 2602dfbe8321SBarry Smith PetscErrorCode ierr; 26036cab3a1bSJed Brown DM dm; 26043a7fca6bSBarry Smith 26053a40ed3dSBarry Smith PetscFunctionBegin; 26060700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 2607e5d3d808SBarry Smith if (Amat) PetscValidHeaderSpecific(Amat,MAT_CLASSID,2); 2608e5d3d808SBarry Smith if (Pmat) PetscValidHeaderSpecific(Pmat,MAT_CLASSID,3); 2609e5d3d808SBarry Smith if (Amat) PetscCheckSameComm(snes,1,Amat,2); 2610e5d3d808SBarry Smith if (Pmat) PetscCheckSameComm(snes,1,Pmat,3); 26116cab3a1bSJed Brown ierr = SNESGetDM(snes,&dm);CHKERRQ(ierr); 2612f8b49ee9SBarry Smith ierr = DMSNESSetJacobian(dm,J,ctx);CHKERRQ(ierr); 2613e5d3d808SBarry Smith if (Amat) { 2614e5d3d808SBarry Smith ierr = PetscObjectReference((PetscObject)Amat);CHKERRQ(ierr); 26156bf464f9SBarry Smith ierr = MatDestroy(&snes->jacobian);CHKERRQ(ierr); 2616f5af7f23SKarl Rupp 2617e5d3d808SBarry Smith snes->jacobian = Amat; 26183a7fca6bSBarry Smith } 2619e5d3d808SBarry Smith if (Pmat) { 2620e5d3d808SBarry Smith ierr = PetscObjectReference((PetscObject)Pmat);CHKERRQ(ierr); 26216bf464f9SBarry Smith ierr = MatDestroy(&snes->jacobian_pre);CHKERRQ(ierr); 2622f5af7f23SKarl Rupp 2623e5d3d808SBarry Smith snes->jacobian_pre = Pmat; 26243a7fca6bSBarry Smith } 26253a40ed3dSBarry Smith PetscFunctionReturn(0); 26269b94acceSBarry Smith } 262762fef451SLois Curfman McInnes 2628c2aafc4cSSatish Balay /*@C 2629b4fd4287SBarry Smith SNESGetJacobian - Returns the Jacobian matrix and optionally the user 2630b4fd4287SBarry Smith provided context for evaluating the Jacobian. 2631b4fd4287SBarry Smith 2632c7afd0dbSLois Curfman McInnes Not Collective, but Mat object will be parallel if SNES object is 2633c7afd0dbSLois Curfman McInnes 2634b4fd4287SBarry Smith Input Parameter: 2635b4fd4287SBarry Smith . snes - the nonlinear solver context 2636b4fd4287SBarry Smith 2637b4fd4287SBarry Smith Output Parameters: 2638e5d3d808SBarry Smith + Amat - location to stash (approximate) Jacobian matrix (or NULL) 2639e5d3d808SBarry Smith . Pmat - location to stash matrix used to compute the preconditioner (or NULL) 2640411c0326SBarry Smith . J - location to put Jacobian function (or NULL), see SNESJacobianFunction for details on its calling sequence 26410298fd71SBarry Smith - ctx - location to stash Jacobian ctx (or NULL) 2642fee21e36SBarry Smith 264336851e7fSLois Curfman McInnes Level: advanced 264436851e7fSLois Curfman McInnes 2645411c0326SBarry Smith .seealso: SNESSetJacobian(), SNESComputeJacobian(), SNESJacobianFunction, SNESGetFunction() 2646b4fd4287SBarry Smith @*/ 2647d1e9a80fSBarry Smith PetscErrorCode SNESGetJacobian(SNES snes,Mat *Amat,Mat *Pmat,PetscErrorCode (**J)(SNES,Vec,Mat,Mat,void*),void **ctx) 2648b4fd4287SBarry Smith { 26496cab3a1bSJed Brown PetscErrorCode ierr; 26506cab3a1bSJed Brown DM dm; 2651942e3340SBarry Smith DMSNES sdm; 26526cab3a1bSJed Brown 26533a40ed3dSBarry Smith PetscFunctionBegin; 26540700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 2655e5d3d808SBarry Smith if (Amat) *Amat = snes->jacobian; 2656e5d3d808SBarry Smith if (Pmat) *Pmat = snes->jacobian_pre; 26576cab3a1bSJed Brown ierr = SNESGetDM(snes,&dm);CHKERRQ(ierr); 2658942e3340SBarry Smith ierr = DMGetDMSNES(dm,&sdm);CHKERRQ(ierr); 2659f8b49ee9SBarry Smith if (J) *J = sdm->ops->computejacobian; 26606cab3a1bSJed Brown if (ctx) *ctx = sdm->jacobianctx; 26613a40ed3dSBarry Smith PetscFunctionReturn(0); 2662b4fd4287SBarry Smith } 2663b4fd4287SBarry Smith 26649b94acceSBarry Smith /*@ 26659b94acceSBarry Smith SNESSetUp - Sets up the internal data structures for the later use 2666272ac6f2SLois Curfman McInnes of a nonlinear solver. 26679b94acceSBarry Smith 2668fee21e36SBarry Smith Collective on SNES 2669fee21e36SBarry Smith 2670c7afd0dbSLois Curfman McInnes Input Parameters: 267170e92668SMatthew Knepley . snes - the SNES context 2672c7afd0dbSLois Curfman McInnes 2673272ac6f2SLois Curfman McInnes Notes: 2674272ac6f2SLois Curfman McInnes For basic use of the SNES solvers the user need not explicitly call 2675272ac6f2SLois Curfman McInnes SNESSetUp(), since these actions will automatically occur during 2676272ac6f2SLois Curfman McInnes the call to SNESSolve(). However, if one wishes to control this 2677272ac6f2SLois Curfman McInnes phase separately, SNESSetUp() should be called after SNESCreate() 2678272ac6f2SLois Curfman McInnes and optional routines of the form SNESSetXXX(), but before SNESSolve(). 2679272ac6f2SLois Curfman McInnes 268036851e7fSLois Curfman McInnes Level: advanced 268136851e7fSLois Curfman McInnes 26829b94acceSBarry Smith .keywords: SNES, nonlinear, setup 26839b94acceSBarry Smith 26849b94acceSBarry Smith .seealso: SNESCreate(), SNESSolve(), SNESDestroy() 26859b94acceSBarry Smith @*/ 26867087cfbeSBarry Smith PetscErrorCode SNESSetUp(SNES snes) 26879b94acceSBarry Smith { 2688dfbe8321SBarry Smith PetscErrorCode ierr; 26896cab3a1bSJed Brown DM dm; 2690942e3340SBarry Smith DMSNES sdm; 2691c35f09e5SBarry Smith SNESLineSearch linesearch, pclinesearch; 26926e2a1849SPeter Brune void *lsprectx,*lspostctx; 26936b2b7091SBarry Smith PetscErrorCode (*precheck)(SNESLineSearch,Vec,Vec,PetscBool*,void*); 26946b2b7091SBarry Smith PetscErrorCode (*postcheck)(SNESLineSearch,Vec,Vec,Vec,PetscBool*,PetscBool*,void*); 26956e2a1849SPeter Brune PetscErrorCode (*func)(SNES,Vec,Vec,void*); 26966e2a1849SPeter Brune Vec f,fpc; 26976e2a1849SPeter Brune void *funcctx; 2698d1e9a80fSBarry Smith PetscErrorCode (*jac)(SNES,Vec,Mat,Mat,void*); 26991eb13d49SPeter Brune void *jacctx,*appctx; 270032b97717SPeter Brune Mat j,jpre; 27013a40ed3dSBarry Smith 27023a40ed3dSBarry Smith PetscFunctionBegin; 27030700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 27044dc4c822SBarry Smith if (snes->setupcalled) PetscFunctionReturn(0); 27059b94acceSBarry Smith 27067adad957SLisandro Dalcin if (!((PetscObject)snes)->type_name) { 270704d7464bSBarry Smith ierr = SNESSetType(snes,SNESNEWTONLS);CHKERRQ(ierr); 270885385478SLisandro Dalcin } 270985385478SLisandro Dalcin 27100298fd71SBarry Smith ierr = SNESGetFunction(snes,&snes->vec_func,NULL,NULL);CHKERRQ(ierr); 271158c9b817SLisandro Dalcin 27126cab3a1bSJed Brown ierr = SNESGetDM(snes,&dm);CHKERRQ(ierr); 2713942e3340SBarry Smith ierr = DMGetDMSNES(dm,&sdm);CHKERRQ(ierr); 2714ce94432eSBarry Smith if (!sdm->ops->computefunction) SETERRQ(PetscObjectComm((PetscObject)dm),PETSC_ERR_ARG_WRONGSTATE,"Function never provided to SNES object"); 271522c6f798SBarry Smith if (!sdm->ops->computejacobian) { 27168d359177SBarry Smith ierr = DMSNESSetJacobian(dm,SNESComputeJacobianDefaultColor,NULL);CHKERRQ(ierr); 271719f7a02aSBarry Smith } 27186cab3a1bSJed Brown if (!snes->vec_func) { 27196cab3a1bSJed Brown ierr = DMCreateGlobalVector(dm,&snes->vec_func);CHKERRQ(ierr); 2720214df951SJed Brown } 2721efd51863SBarry Smith 272222d28d08SBarry Smith if (!snes->ksp) { 272322d28d08SBarry Smith ierr = SNESGetKSP(snes, &snes->ksp);CHKERRQ(ierr); 272422d28d08SBarry Smith } 2725b710008aSBarry Smith 272622d28d08SBarry Smith if (!snes->linesearch) { 27277601faf0SJed Brown ierr = SNESGetLineSearch(snes, &snes->linesearch);CHKERRQ(ierr); 272822d28d08SBarry Smith } 2729ed07d7d7SPeter Brune ierr = SNESLineSearchSetFunction(snes->linesearch,SNESComputeFunction);CHKERRQ(ierr); 27309e764e56SPeter Brune 2731efd4aadfSBarry Smith if (snes->npc && (snes->npcside== PC_LEFT)) { 2732172a4300SPeter Brune snes->mf = PETSC_TRUE; 2733172a4300SPeter Brune snes->mf_operator = PETSC_FALSE; 2734172a4300SPeter Brune } 2735d8f46077SPeter Brune 2736efd4aadfSBarry Smith if (snes->npc) { 27376e2a1849SPeter Brune /* copy the DM over */ 27386e2a1849SPeter Brune ierr = SNESGetDM(snes,&dm);CHKERRQ(ierr); 2739efd4aadfSBarry Smith ierr = SNESSetDM(snes->npc,dm);CHKERRQ(ierr); 27406e2a1849SPeter Brune 27416e2a1849SPeter Brune ierr = SNESGetFunction(snes,&f,&func,&funcctx);CHKERRQ(ierr); 27426e2a1849SPeter Brune ierr = VecDuplicate(f,&fpc);CHKERRQ(ierr); 2743efd4aadfSBarry Smith ierr = SNESSetFunction(snes->npc,fpc,func,funcctx);CHKERRQ(ierr); 274432b97717SPeter Brune ierr = SNESGetJacobian(snes,&j,&jpre,&jac,&jacctx);CHKERRQ(ierr); 2745efd4aadfSBarry Smith ierr = SNESSetJacobian(snes->npc,j,jpre,jac,jacctx);CHKERRQ(ierr); 27461eb13d49SPeter Brune ierr = SNESGetApplicationContext(snes,&appctx);CHKERRQ(ierr); 2747efd4aadfSBarry Smith ierr = SNESSetApplicationContext(snes->npc,appctx);CHKERRQ(ierr); 27486e2a1849SPeter Brune ierr = VecDestroy(&fpc);CHKERRQ(ierr); 27496e2a1849SPeter Brune 27506e2a1849SPeter Brune /* copy the function pointers over */ 2751efd4aadfSBarry Smith ierr = PetscObjectCopyFortranFunctionPointers((PetscObject)snes,(PetscObject)snes->npc);CHKERRQ(ierr); 27526e2a1849SPeter Brune 27536e2a1849SPeter Brune /* default to 1 iteration */ 2754efd4aadfSBarry Smith ierr = SNESSetTolerances(snes->npc,0.0,0.0,0.0,1,snes->npc->max_funcs);CHKERRQ(ierr); 2755efd4aadfSBarry Smith if (snes->npcside==PC_RIGHT) { 2756efd4aadfSBarry Smith ierr = SNESSetNormSchedule(snes->npc,SNES_NORM_FINAL_ONLY);CHKERRQ(ierr); 2757a9936a0cSPeter Brune } else { 2758efd4aadfSBarry Smith ierr = SNESSetNormSchedule(snes->npc,SNES_NORM_NONE);CHKERRQ(ierr); 2759a9936a0cSPeter Brune } 2760efd4aadfSBarry Smith ierr = SNESSetFromOptions(snes->npc);CHKERRQ(ierr); 27616e2a1849SPeter Brune 27626e2a1849SPeter Brune /* copy the line search context over */ 27637601faf0SJed Brown ierr = SNESGetLineSearch(snes,&linesearch);CHKERRQ(ierr); 2764efd4aadfSBarry Smith ierr = SNESGetLineSearch(snes->npc,&pclinesearch);CHKERRQ(ierr); 27656b2b7091SBarry Smith ierr = SNESLineSearchGetPreCheck(linesearch,&precheck,&lsprectx);CHKERRQ(ierr); 27666b2b7091SBarry Smith ierr = SNESLineSearchGetPostCheck(linesearch,&postcheck,&lspostctx);CHKERRQ(ierr); 27676b2b7091SBarry Smith ierr = SNESLineSearchSetPreCheck(pclinesearch,precheck,lsprectx);CHKERRQ(ierr); 27686b2b7091SBarry Smith ierr = SNESLineSearchSetPostCheck(pclinesearch,postcheck,lspostctx);CHKERRQ(ierr); 27696e2a1849SPeter Brune ierr = PetscObjectCopyFortranFunctionPointers((PetscObject)linesearch, (PetscObject)pclinesearch);CHKERRQ(ierr); 27706e2a1849SPeter Brune } 277132b97717SPeter Brune if (snes->mf) { 277232b97717SPeter Brune ierr = SNESSetUpMatrixFree_Private(snes, snes->mf_operator, snes->mf_version);CHKERRQ(ierr); 277332b97717SPeter Brune } 277432b97717SPeter Brune if (snes->ops->usercompute && !snes->user) { 277532b97717SPeter Brune ierr = (*snes->ops->usercompute)(snes,(void**)&snes->user);CHKERRQ(ierr); 277632b97717SPeter Brune } 27776e2a1849SPeter Brune 277837ec4e1aSPeter Brune snes->jac_iter = 0; 277937ec4e1aSPeter Brune snes->pre_iter = 0; 278037ec4e1aSPeter Brune 2781410397dcSLisandro Dalcin if (snes->ops->setup) { 2782410397dcSLisandro Dalcin ierr = (*snes->ops->setup)(snes);CHKERRQ(ierr); 2783410397dcSLisandro Dalcin } 278458c9b817SLisandro Dalcin 2785efd4aadfSBarry Smith if (snes->npc && (snes->npcside== PC_LEFT)) { 27866c67d002SPeter Brune if (snes->functype == SNES_FUNCTION_PRECONDITIONED) { 278755d4788fSPeter Brune ierr = SNESGetLineSearch(snes,&linesearch);CHKERRQ(ierr); 2788be95d8f1SBarry Smith ierr = SNESLineSearchSetFunction(linesearch,SNESComputeFunctionDefaultNPC);CHKERRQ(ierr); 27896c67d002SPeter Brune } 27906c67d002SPeter Brune } 27916c67d002SPeter Brune 27927aaed0d8SBarry Smith snes->setupcalled = PETSC_TRUE; 27933a40ed3dSBarry Smith PetscFunctionReturn(0); 27949b94acceSBarry Smith } 27959b94acceSBarry Smith 279637596af1SLisandro Dalcin /*@ 279737596af1SLisandro Dalcin SNESReset - Resets a SNES context to the snessetupcalled = 0 state and removes any allocated Vecs and Mats 279837596af1SLisandro Dalcin 279937596af1SLisandro Dalcin Collective on SNES 280037596af1SLisandro Dalcin 280137596af1SLisandro Dalcin Input Parameter: 280237596af1SLisandro Dalcin . snes - iterative context obtained from SNESCreate() 280337596af1SLisandro Dalcin 2804d25893d9SBarry Smith Level: intermediate 2805d25893d9SBarry Smith 2806d25893d9SBarry Smith Notes: Also calls the application context destroy routine set with SNESSetComputeApplicationContext() 280737596af1SLisandro Dalcin 280837596af1SLisandro Dalcin .keywords: SNES, destroy 280937596af1SLisandro Dalcin 281037596af1SLisandro Dalcin .seealso: SNESCreate(), SNESSetUp(), SNESSolve() 281137596af1SLisandro Dalcin @*/ 281237596af1SLisandro Dalcin PetscErrorCode SNESReset(SNES snes) 281337596af1SLisandro Dalcin { 281437596af1SLisandro Dalcin PetscErrorCode ierr; 281537596af1SLisandro Dalcin 281637596af1SLisandro Dalcin PetscFunctionBegin; 281737596af1SLisandro Dalcin PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 2818d25893d9SBarry Smith if (snes->ops->userdestroy && snes->user) { 2819d25893d9SBarry Smith ierr = (*snes->ops->userdestroy)((void**)&snes->user);CHKERRQ(ierr); 28200298fd71SBarry Smith snes->user = NULL; 2821d25893d9SBarry Smith } 2822efd4aadfSBarry Smith if (snes->npc) { 2823efd4aadfSBarry Smith ierr = SNESReset(snes->npc);CHKERRQ(ierr); 28248a23116dSBarry Smith } 28258a23116dSBarry Smith 282637596af1SLisandro Dalcin if (snes->ops->reset) { 282737596af1SLisandro Dalcin ierr = (*snes->ops->reset)(snes);CHKERRQ(ierr); 282837596af1SLisandro Dalcin } 28299e764e56SPeter Brune if (snes->ksp) { 28309e764e56SPeter Brune ierr = KSPReset(snes->ksp);CHKERRQ(ierr); 28319e764e56SPeter Brune } 28329e764e56SPeter Brune 28339e764e56SPeter Brune if (snes->linesearch) { 2834f1c6b773SPeter Brune ierr = SNESLineSearchReset(snes->linesearch);CHKERRQ(ierr); 28359e764e56SPeter Brune } 28369e764e56SPeter Brune 28376bf464f9SBarry Smith ierr = VecDestroy(&snes->vec_rhs);CHKERRQ(ierr); 28386bf464f9SBarry Smith ierr = VecDestroy(&snes->vec_sol);CHKERRQ(ierr); 28396bf464f9SBarry Smith ierr = VecDestroy(&snes->vec_sol_update);CHKERRQ(ierr); 28406bf464f9SBarry Smith ierr = VecDestroy(&snes->vec_func);CHKERRQ(ierr); 28416bf464f9SBarry Smith ierr = MatDestroy(&snes->jacobian);CHKERRQ(ierr); 28426bf464f9SBarry Smith ierr = MatDestroy(&snes->jacobian_pre);CHKERRQ(ierr); 2843c41d97abSJed Brown ierr = VecDestroyVecs(snes->nwork,&snes->work);CHKERRQ(ierr); 2844c41d97abSJed Brown ierr = VecDestroyVecs(snes->nvwork,&snes->vwork);CHKERRQ(ierr); 2845f5af7f23SKarl Rupp 284640fdac6aSLawrence Mitchell snes->alwayscomputesfinalresidual = PETSC_FALSE; 284740fdac6aSLawrence Mitchell 284837596af1SLisandro Dalcin snes->nwork = snes->nvwork = 0; 284937596af1SLisandro Dalcin snes->setupcalled = PETSC_FALSE; 285037596af1SLisandro Dalcin PetscFunctionReturn(0); 285137596af1SLisandro Dalcin } 285237596af1SLisandro Dalcin 285352baeb72SSatish Balay /*@ 28549b94acceSBarry Smith SNESDestroy - Destroys the nonlinear solver context that was created 28559b94acceSBarry Smith with SNESCreate(). 28569b94acceSBarry Smith 2857c7afd0dbSLois Curfman McInnes Collective on SNES 2858c7afd0dbSLois Curfman McInnes 28599b94acceSBarry Smith Input Parameter: 28609b94acceSBarry Smith . snes - the SNES context 28619b94acceSBarry Smith 286236851e7fSLois Curfman McInnes Level: beginner 286336851e7fSLois Curfman McInnes 28649b94acceSBarry Smith .keywords: SNES, nonlinear, destroy 28659b94acceSBarry Smith 286663a78c88SLois Curfman McInnes .seealso: SNESCreate(), SNESSolve() 28679b94acceSBarry Smith @*/ 28686bf464f9SBarry Smith PetscErrorCode SNESDestroy(SNES *snes) 28699b94acceSBarry Smith { 28706849ba73SBarry Smith PetscErrorCode ierr; 28713a40ed3dSBarry Smith 28723a40ed3dSBarry Smith PetscFunctionBegin; 28736bf464f9SBarry Smith if (!*snes) PetscFunctionReturn(0); 28746bf464f9SBarry Smith PetscValidHeaderSpecific((*snes),SNES_CLASSID,1); 28756bf464f9SBarry Smith if (--((PetscObject)(*snes))->refct > 0) {*snes = 0; PetscFunctionReturn(0);} 2876d4bb536fSBarry Smith 28776bf464f9SBarry Smith ierr = SNESReset((*snes));CHKERRQ(ierr); 2878efd4aadfSBarry Smith ierr = SNESDestroy(&(*snes)->npc);CHKERRQ(ierr); 28796b8b9a38SLisandro Dalcin 2880e04113cfSBarry Smith /* if memory was published with SAWs then destroy it */ 2881e04113cfSBarry Smith ierr = PetscObjectSAWsViewOff((PetscObject)*snes);CHKERRQ(ierr); 28826bf464f9SBarry Smith if ((*snes)->ops->destroy) {ierr = (*((*snes))->ops->destroy)((*snes));CHKERRQ(ierr);} 28836d4c513bSLisandro Dalcin 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 4040*34b4d3a8SMatthew G. Knepley /* High level operations using the nonlinear solver */ 404106fc46c8SMatthew G. Knepley { 404206fc46c8SMatthew G. Knepley PetscViewer viewer; 404306fc46c8SMatthew G. Knepley PetscViewerFormat format; 404406fc46c8SMatthew G. Knepley PetscBool flg; 404506fc46c8SMatthew G. Knepley static PetscBool incall = PETSC_FALSE; 404606fc46c8SMatthew G. Knepley 404706fc46c8SMatthew G. Knepley if (!incall) { 4048*34b4d3a8SMatthew G. Knepley /* Estimate the convergence rate of the discretization */ 404906fc46c8SMatthew G. Knepley ierr = PetscOptionsGetViewer(PetscObjectComm((PetscObject) snes), ((PetscObject) snes)->prefix, "-snes_convergence_estimate", &viewer, &format, &flg);CHKERRQ(ierr); 405006fc46c8SMatthew G. Knepley if (flg) { 405106fc46c8SMatthew G. Knepley PetscConvEst conv; 405206fc46c8SMatthew G. Knepley PetscReal alpha; /* Convergence rate of the solution error in the L_2 norm */ 405306fc46c8SMatthew G. Knepley 405406fc46c8SMatthew G. Knepley incall = PETSC_TRUE; 405506fc46c8SMatthew G. Knepley ierr = PetscConvEstCreate(PetscObjectComm((PetscObject) snes), &conv);CHKERRQ(ierr); 405606fc46c8SMatthew G. Knepley ierr = PetscConvEstSetSolver(conv, snes);CHKERRQ(ierr); 405706fc46c8SMatthew G. Knepley ierr = PetscConvEstSetFromOptions(conv);CHKERRQ(ierr); 40580955ed61SMatthew G. Knepley ierr = PetscConvEstSetUp(conv);CHKERRQ(ierr); 405906fc46c8SMatthew G. Knepley ierr = PetscConvEstGetConvRate(conv, &alpha);CHKERRQ(ierr); 406006fc46c8SMatthew G. Knepley ierr = PetscViewerPushFormat(viewer, format);CHKERRQ(ierr); 406106fc46c8SMatthew G. Knepley ierr = PetscConvEstRateView(conv, alpha, viewer);CHKERRQ(ierr); 406206fc46c8SMatthew G. Knepley ierr = PetscViewerPopFormat(viewer);CHKERRQ(ierr); 406306fc46c8SMatthew G. Knepley ierr = PetscViewerDestroy(&viewer);CHKERRQ(ierr); 406406fc46c8SMatthew G. Knepley ierr = PetscConvEstDestroy(&conv);CHKERRQ(ierr); 406506fc46c8SMatthew G. Knepley incall = PETSC_FALSE; 406606fc46c8SMatthew G. Knepley } 4067*34b4d3a8SMatthew G. Knepley /* Adaptively refine the initial grid */ 4068*34b4d3a8SMatthew G. Knepley flg = PETSC_FALSE; 4069*34b4d3a8SMatthew G. Knepley ierr = PetscOptionsGetBool(NULL, ((PetscObject) snes)->prefix, "-snes_adapt_initial", &flg, NULL);CHKERRQ(ierr); 4070*34b4d3a8SMatthew G. Knepley if (flg) { 4071*34b4d3a8SMatthew G. Knepley DMAdaptor adaptor; 4072*34b4d3a8SMatthew G. Knepley 4073*34b4d3a8SMatthew G. Knepley incall = PETSC_TRUE; 4074*34b4d3a8SMatthew G. Knepley ierr = DMAdaptorCreate(PETSC_COMM_WORLD, &adaptor);CHKERRQ(ierr); 4075*34b4d3a8SMatthew G. Knepley ierr = DMAdaptorSetSolver(adaptor, snes);CHKERRQ(ierr); 4076*34b4d3a8SMatthew G. Knepley ierr = DMAdaptorSetFromOptions(adaptor);CHKERRQ(ierr); 4077*34b4d3a8SMatthew G. Knepley ierr = DMAdaptorSetUp(adaptor);CHKERRQ(ierr); 4078*34b4d3a8SMatthew G. Knepley ierr = DMAdaptorAdapt(adaptor, x, DM_ADAPTATION_INITIAL, &dm, &x);CHKERRQ(ierr); 4079*34b4d3a8SMatthew G. Knepley ierr = DMAdaptorDestroy(&adaptor);CHKERRQ(ierr); 4080*34b4d3a8SMatthew G. Knepley incall = PETSC_FALSE; 4081*34b4d3a8SMatthew G. Knepley } 408206fc46c8SMatthew G. Knepley } 408306fc46c8SMatthew G. Knepley } 4084caa4e7f2SJed Brown if (!x) { 4085caa4e7f2SJed Brown ierr = SNESGetDM(snes,&dm);CHKERRQ(ierr); 4086caa4e7f2SJed Brown ierr = DMCreateGlobalVector(dm,&xcreated);CHKERRQ(ierr); 4087a69afd8bSBarry Smith x = xcreated; 4088a69afd8bSBarry Smith } 4089ce1779c8SBarry Smith ierr = SNESViewFromOptions(snes,NULL,"-snes_view_pre");CHKERRQ(ierr); 4090f05ece33SBarry Smith 4091ce94432eSBarry Smith for (grid=0; grid<snes->gridsequence; grid++) {ierr = PetscViewerASCIIPushTab(PETSC_VIEWER_STDOUT_(PetscObjectComm((PetscObject)snes)));CHKERRQ(ierr);} 4092efd51863SBarry Smith for (grid=0; grid<snes->gridsequence+1; grid++) { 4093efd51863SBarry Smith 409485385478SLisandro Dalcin /* set solution vector */ 4095efd51863SBarry Smith if (!grid) {ierr = PetscObjectReference((PetscObject)x);CHKERRQ(ierr);} 40966bf464f9SBarry Smith ierr = VecDestroy(&snes->vec_sol);CHKERRQ(ierr); 409785385478SLisandro Dalcin snes->vec_sol = x; 4098caa4e7f2SJed Brown ierr = SNESGetDM(snes,&dm);CHKERRQ(ierr); 4099caa4e7f2SJed Brown 4100caa4e7f2SJed Brown /* set affine vector if provided */ 410185385478SLisandro Dalcin if (b) { ierr = PetscObjectReference((PetscObject)b);CHKERRQ(ierr); } 41026bf464f9SBarry Smith ierr = VecDestroy(&snes->vec_rhs);CHKERRQ(ierr); 410385385478SLisandro Dalcin snes->vec_rhs = b; 410485385478SLisandro Dalcin 4105154060b5SMatthew G. Knepley if (snes->vec_func == snes->vec_sol) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_IDN,"Solution vector cannot be function vector"); 4106154060b5SMatthew 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"); 4107154060b5SMatthew G. Knepley if (!snes->vec_sol_update /* && snes->vec_sol */) { 4108154060b5SMatthew G. Knepley ierr = VecDuplicate(snes->vec_sol,&snes->vec_sol_update);CHKERRQ(ierr); 4109154060b5SMatthew G. Knepley ierr = PetscLogObjectParent((PetscObject)snes,(PetscObject)snes->vec_sol_update);CHKERRQ(ierr); 4110154060b5SMatthew G. Knepley } 4111154060b5SMatthew G. Knepley ierr = DMShellSetGlobalVector(dm,snes->vec_sol);CHKERRQ(ierr); 411270e92668SMatthew Knepley ierr = SNESSetUp(snes);CHKERRQ(ierr); 41133f149594SLisandro Dalcin 41147eee914bSBarry Smith if (!grid) { 41157eee914bSBarry Smith if (snes->ops->computeinitialguess) { 4116d25893d9SBarry Smith ierr = (*snes->ops->computeinitialguess)(snes,snes->vec_sol,snes->initialguessP);CHKERRQ(ierr); 4117d25893d9SBarry Smith } 4118dd568438SSatish Balay } 4119d25893d9SBarry Smith 4120abc0a331SBarry Smith if (snes->conv_hist_reset) snes->conv_hist_len = 0; 4121971e163fSPeter Brune if (snes->counters_reset) {snes->nfuncs = 0; snes->linear_its = 0; snes->numFailures = 0;} 4122d5e45103SBarry Smith 41233f149594SLisandro Dalcin ierr = PetscLogEventBegin(SNES_Solve,snes,0,0,0);CHKERRQ(ierr); 41244936397dSBarry Smith ierr = (*snes->ops->solve)(snes);CHKERRQ(ierr); 412585385478SLisandro Dalcin ierr = PetscLogEventEnd(SNES_Solve,snes,0,0,0);CHKERRQ(ierr); 412617186662SBarry Smith if (!snes->reason) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_PLIB,"Internal error, solver returned without setting converged reason"); 4127422a814eSBarry Smith snes->domainerror = PETSC_FALSE; /* clear the flag if it has been set */ 41283f149594SLisandro Dalcin 412937ec4e1aSPeter Brune if (snes->lagjac_persist) snes->jac_iter += snes->iter; 413037ec4e1aSPeter Brune if (snes->lagpre_persist) snes->pre_iter += snes->iter; 413137ec4e1aSPeter Brune 413227b0f280SBarry Smith ierr = PetscOptionsGetViewer(PetscObjectComm((PetscObject)snes),((PetscObject)snes)->prefix,"-snes_test_local_min",NULL,NULL,&flg);CHKERRQ(ierr); 4133da9b6338SBarry Smith if (flg && !PetscPreLoadingOn) { ierr = SNESTestLocalMin(snes);CHKERRQ(ierr); } 41342a359c20SBarry Smith ierr = SNESReasonViewFromOptions(snes);CHKERRQ(ierr); 41355968eb51SBarry Smith 4136ce94432eSBarry Smith if (snes->errorifnotconverged && snes->reason < 0) SETERRQ(PetscObjectComm((PetscObject)snes),PETSC_ERR_NOT_CONVERGED,"SNESSolve has not converged"); 41379c8e83a9SBarry Smith if (snes->reason < 0) break; 4138efd51863SBarry Smith if (grid < snes->gridsequence) { 4139efd51863SBarry Smith DM fine; 4140efd51863SBarry Smith Vec xnew; 4141efd51863SBarry Smith Mat interp; 4142efd51863SBarry Smith 4143ce94432eSBarry Smith ierr = DMRefine(snes->dm,PetscObjectComm((PetscObject)snes),&fine);CHKERRQ(ierr); 4144ce94432eSBarry Smith if (!fine) SETERRQ(PetscObjectComm((PetscObject)snes),PETSC_ERR_ARG_INCOMP,"DMRefine() did not perform any refinement, cannot continue grid sequencing"); 41450298fd71SBarry Smith ierr = DMCreateInterpolation(snes->dm,fine,&interp,NULL);CHKERRQ(ierr); 4146efd51863SBarry Smith ierr = DMCreateGlobalVector(fine,&xnew);CHKERRQ(ierr); 4147efd51863SBarry Smith ierr = MatInterpolate(interp,x,xnew);CHKERRQ(ierr); 4148c833c3b5SJed Brown ierr = DMInterpolate(snes->dm,interp,fine);CHKERRQ(ierr); 4149efd51863SBarry Smith ierr = MatDestroy(&interp);CHKERRQ(ierr); 4150efd51863SBarry Smith x = xnew; 4151efd51863SBarry Smith 4152efd51863SBarry Smith ierr = SNESReset(snes);CHKERRQ(ierr); 4153efd51863SBarry Smith ierr = SNESSetDM(snes,fine);CHKERRQ(ierr); 4154efd51863SBarry Smith ierr = DMDestroy(&fine);CHKERRQ(ierr); 4155ce94432eSBarry Smith ierr = PetscViewerASCIIPopTab(PETSC_VIEWER_STDOUT_(PetscObjectComm((PetscObject)snes)));CHKERRQ(ierr); 4156efd51863SBarry Smith } 4157efd51863SBarry Smith } 4158ce1779c8SBarry Smith ierr = SNESViewFromOptions(snes,NULL,"-snes_view");CHKERRQ(ierr); 4159685405a1SBarry Smith ierr = VecViewFromOptions(snes->vec_sol,(PetscObject)snes,"-snes_view_solution");CHKERRQ(ierr); 41603f7e2da0SPeter Brune 4161a69afd8bSBarry Smith ierr = VecDestroy(&xcreated);CHKERRQ(ierr); 4162e04113cfSBarry Smith ierr = PetscObjectSAWsBlock((PetscObject)snes);CHKERRQ(ierr); 41633a40ed3dSBarry Smith PetscFunctionReturn(0); 41649b94acceSBarry Smith } 41659b94acceSBarry Smith 41669b94acceSBarry Smith /* --------- Internal routines for SNES Package --------- */ 41679b94acceSBarry Smith 416882bf6240SBarry Smith /*@C 41694b0e389bSBarry Smith SNESSetType - Sets the method for the nonlinear solver. 41709b94acceSBarry Smith 4171fee21e36SBarry Smith Collective on SNES 4172fee21e36SBarry Smith 4173c7afd0dbSLois Curfman McInnes Input Parameters: 4174c7afd0dbSLois Curfman McInnes + snes - the SNES context 4175454a90a3SBarry Smith - type - a known method 4176c7afd0dbSLois Curfman McInnes 4177c7afd0dbSLois Curfman McInnes Options Database Key: 4178454a90a3SBarry Smith . -snes_type <type> - Sets the method; use -help for a list 417904d7464bSBarry Smith of available methods (for instance, newtonls or newtontr) 4180ae12b187SLois Curfman McInnes 41819b94acceSBarry Smith Notes: 4182e090d566SSatish Balay See "petsc/include/petscsnes.h" for available methods (for instance) 418304d7464bSBarry Smith + SNESNEWTONLS - Newton's method with line search 4184c7afd0dbSLois Curfman McInnes (systems of nonlinear equations) 418504d7464bSBarry Smith . SNESNEWTONTR - Newton's method with trust region 4186c7afd0dbSLois Curfman McInnes (systems of nonlinear equations) 41879b94acceSBarry Smith 4188ae12b187SLois Curfman McInnes Normally, it is best to use the SNESSetFromOptions() command and then 4189ae12b187SLois Curfman McInnes set the SNES solver type from the options database rather than by using 4190ae12b187SLois Curfman McInnes this routine. Using the options database provides the user with 4191ae12b187SLois Curfman McInnes maximum flexibility in evaluating the many nonlinear solvers. 4192ae12b187SLois Curfman McInnes The SNESSetType() routine is provided for those situations where it 4193ae12b187SLois Curfman McInnes is necessary to set the nonlinear solver independently of the command 4194ae12b187SLois Curfman McInnes line or options database. This might be the case, for example, when 4195ae12b187SLois Curfman McInnes the choice of solver changes during the execution of the program, 4196ae12b187SLois Curfman McInnes and the user's application is taking responsibility for choosing the 4197b0a32e0cSBarry Smith appropriate method. 419836851e7fSLois Curfman McInnes 41998f6c3df8SBarry Smith Developer Notes: SNESRegister() adds a constructor for a new SNESType to SNESList, SNESSetType() locates 42008f6c3df8SBarry Smith the constructor in that list and calls it to create the spexific object. 42018f6c3df8SBarry Smith 420236851e7fSLois Curfman McInnes Level: intermediate 4203a703fe33SLois Curfman McInnes 4204454a90a3SBarry Smith .keywords: SNES, set, type 4205435da068SBarry Smith 42068f6c3df8SBarry Smith .seealso: SNESType, SNESCreate(), SNESDestroy(), SNESGetType(), SNESSetFromOptions() 4207435da068SBarry Smith 42089b94acceSBarry Smith @*/ 420919fd82e9SBarry Smith PetscErrorCode SNESSetType(SNES snes,SNESType type) 42109b94acceSBarry Smith { 4211dfbe8321SBarry Smith PetscErrorCode ierr,(*r)(SNES); 4212ace3abfcSBarry Smith PetscBool match; 42133a40ed3dSBarry Smith 42143a40ed3dSBarry Smith PetscFunctionBegin; 42150700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 42164482741eSBarry Smith PetscValidCharPointer(type,2); 421782bf6240SBarry Smith 4218251f4c67SDmitry Karpeev ierr = PetscObjectTypeCompare((PetscObject)snes,type,&match);CHKERRQ(ierr); 42190f5bd95cSBarry Smith if (match) PetscFunctionReturn(0); 422092ff6ae8SBarry Smith 42211c9cd337SJed Brown ierr = PetscFunctionListFind(SNESList,type,&r);CHKERRQ(ierr); 4222e32f2f54SBarry Smith if (!r) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_UNKNOWN_TYPE,"Unable to find requested SNES type %s",type); 422375396ef9SLisandro Dalcin /* Destroy the previous private SNES context */ 4224b5c23020SJed Brown if (snes->ops->destroy) { 4225b5c23020SJed Brown ierr = (*(snes)->ops->destroy)(snes);CHKERRQ(ierr); 42260298fd71SBarry Smith snes->ops->destroy = NULL; 4227b5c23020SJed Brown } 422875396ef9SLisandro Dalcin /* Reinitialize function pointers in SNESOps structure */ 422975396ef9SLisandro Dalcin snes->ops->setup = 0; 423075396ef9SLisandro Dalcin snes->ops->solve = 0; 423175396ef9SLisandro Dalcin snes->ops->view = 0; 423275396ef9SLisandro Dalcin snes->ops->setfromoptions = 0; 423375396ef9SLisandro Dalcin snes->ops->destroy = 0; 423475396ef9SLisandro Dalcin /* Call the SNESCreate_XXX routine for this particular Nonlinear solver */ 423575396ef9SLisandro Dalcin snes->setupcalled = PETSC_FALSE; 4236f5af7f23SKarl Rupp 4237454a90a3SBarry Smith ierr = PetscObjectChangeTypeName((PetscObject)snes,type);CHKERRQ(ierr); 423803bfa161SLisandro Dalcin ierr = (*r)(snes);CHKERRQ(ierr); 42393a40ed3dSBarry Smith PetscFunctionReturn(0); 42409b94acceSBarry Smith } 42419b94acceSBarry Smith 42429b94acceSBarry Smith /*@C 42439a28b0a6SLois Curfman McInnes SNESGetType - Gets the SNES method type and name (as a string). 42449b94acceSBarry Smith 4245c7afd0dbSLois Curfman McInnes Not Collective 4246c7afd0dbSLois Curfman McInnes 42479b94acceSBarry Smith Input Parameter: 42484b0e389bSBarry Smith . snes - nonlinear solver context 42499b94acceSBarry Smith 42509b94acceSBarry Smith Output Parameter: 42513a7fca6bSBarry Smith . type - SNES method (a character string) 42529b94acceSBarry Smith 425336851e7fSLois Curfman McInnes Level: intermediate 425436851e7fSLois Curfman McInnes 4255454a90a3SBarry Smith .keywords: SNES, nonlinear, get, type, name 42569b94acceSBarry Smith @*/ 425719fd82e9SBarry Smith PetscErrorCode SNESGetType(SNES snes,SNESType *type) 42589b94acceSBarry Smith { 42593a40ed3dSBarry Smith PetscFunctionBegin; 42600700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 42614482741eSBarry Smith PetscValidPointer(type,2); 42627adad957SLisandro Dalcin *type = ((PetscObject)snes)->type_name; 42633a40ed3dSBarry Smith PetscFunctionReturn(0); 42649b94acceSBarry Smith } 42659b94acceSBarry Smith 42663cd8a7caSMatthew G. Knepley /*@ 42673cd8a7caSMatthew G. Knepley SNESSetSolution - Sets the solution vector for use by the SNES routines. 42683cd8a7caSMatthew G. Knepley 42693cd8a7caSMatthew G. Knepley Logically Collective on SNES and Vec 42703cd8a7caSMatthew G. Knepley 42713cd8a7caSMatthew G. Knepley Input Parameters: 42723cd8a7caSMatthew G. Knepley + snes - the SNES context obtained from SNESCreate() 42733cd8a7caSMatthew G. Knepley - u - the solution vector 42743cd8a7caSMatthew G. Knepley 42753cd8a7caSMatthew G. Knepley Level: beginner 42763cd8a7caSMatthew G. Knepley 42773cd8a7caSMatthew G. Knepley .keywords: SNES, set, solution 42783cd8a7caSMatthew G. Knepley @*/ 42793cd8a7caSMatthew G. Knepley PetscErrorCode SNESSetSolution(SNES snes, Vec u) 42803cd8a7caSMatthew G. Knepley { 42813cd8a7caSMatthew G. Knepley DM dm; 42823cd8a7caSMatthew G. Knepley PetscErrorCode ierr; 42833cd8a7caSMatthew G. Knepley 42843cd8a7caSMatthew G. Knepley PetscFunctionBegin; 42853cd8a7caSMatthew G. Knepley PetscValidHeaderSpecific(snes, SNES_CLASSID, 1); 42863cd8a7caSMatthew G. Knepley PetscValidHeaderSpecific(u, VEC_CLASSID, 2); 42873cd8a7caSMatthew G. Knepley ierr = PetscObjectReference((PetscObject) u);CHKERRQ(ierr); 42883cd8a7caSMatthew G. Knepley ierr = VecDestroy(&snes->vec_sol);CHKERRQ(ierr); 42893cd8a7caSMatthew G. Knepley 42903cd8a7caSMatthew G. Knepley snes->vec_sol = u; 42913cd8a7caSMatthew G. Knepley 42923cd8a7caSMatthew G. Knepley ierr = SNESGetDM(snes, &dm);CHKERRQ(ierr); 42933cd8a7caSMatthew G. Knepley ierr = DMShellSetGlobalVector(dm, u);CHKERRQ(ierr); 42943cd8a7caSMatthew G. Knepley PetscFunctionReturn(0); 42953cd8a7caSMatthew G. Knepley } 42963cd8a7caSMatthew G. Knepley 429752baeb72SSatish Balay /*@ 42989b94acceSBarry Smith SNESGetSolution - Returns the vector where the approximate solution is 4299c0df2a02SJed Brown stored. This is the fine grid solution when using SNESSetGridSequence(). 43009b94acceSBarry Smith 4301c7afd0dbSLois Curfman McInnes Not Collective, but Vec is parallel if SNES is parallel 4302c7afd0dbSLois Curfman McInnes 43039b94acceSBarry Smith Input Parameter: 43049b94acceSBarry Smith . snes - the SNES context 43059b94acceSBarry Smith 43069b94acceSBarry Smith Output Parameter: 43079b94acceSBarry Smith . x - the solution 43089b94acceSBarry Smith 430970e92668SMatthew Knepley Level: intermediate 431036851e7fSLois Curfman McInnes 43119b94acceSBarry Smith .keywords: SNES, nonlinear, get, solution 43129b94acceSBarry Smith 431385385478SLisandro Dalcin .seealso: SNESGetSolutionUpdate(), SNESGetFunction() 43149b94acceSBarry Smith @*/ 43157087cfbeSBarry Smith PetscErrorCode SNESGetSolution(SNES snes,Vec *x) 43169b94acceSBarry Smith { 43173a40ed3dSBarry Smith PetscFunctionBegin; 43180700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 43194482741eSBarry Smith PetscValidPointer(x,2); 432085385478SLisandro Dalcin *x = snes->vec_sol; 432170e92668SMatthew Knepley PetscFunctionReturn(0); 432270e92668SMatthew Knepley } 432370e92668SMatthew Knepley 432452baeb72SSatish Balay /*@ 43259b94acceSBarry Smith SNESGetSolutionUpdate - Returns the vector where the solution update is 43269b94acceSBarry Smith stored. 43279b94acceSBarry Smith 4328c7afd0dbSLois Curfman McInnes Not Collective, but Vec is parallel if SNES is parallel 4329c7afd0dbSLois Curfman McInnes 43309b94acceSBarry Smith Input Parameter: 43319b94acceSBarry Smith . snes - the SNES context 43329b94acceSBarry Smith 43339b94acceSBarry Smith Output Parameter: 43349b94acceSBarry Smith . x - the solution update 43359b94acceSBarry Smith 433636851e7fSLois Curfman McInnes Level: advanced 433736851e7fSLois Curfman McInnes 43389b94acceSBarry Smith .keywords: SNES, nonlinear, get, solution, update 43399b94acceSBarry Smith 434085385478SLisandro Dalcin .seealso: SNESGetSolution(), SNESGetFunction() 43419b94acceSBarry Smith @*/ 43427087cfbeSBarry Smith PetscErrorCode SNESGetSolutionUpdate(SNES snes,Vec *x) 43439b94acceSBarry Smith { 43443a40ed3dSBarry Smith PetscFunctionBegin; 43450700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 43464482741eSBarry Smith PetscValidPointer(x,2); 434785385478SLisandro Dalcin *x = snes->vec_sol_update; 43483a40ed3dSBarry Smith PetscFunctionReturn(0); 43499b94acceSBarry Smith } 43509b94acceSBarry Smith 43519b94acceSBarry Smith /*@C 43523638b69dSLois Curfman McInnes SNESGetFunction - Returns the vector where the function is stored. 43539b94acceSBarry Smith 4354a63bb30eSJed Brown Not Collective, but Vec is parallel if SNES is parallel. Collective if Vec is requested, but has not been created yet. 4355c7afd0dbSLois Curfman McInnes 43569b94acceSBarry Smith Input Parameter: 43579b94acceSBarry Smith . snes - the SNES context 43589b94acceSBarry Smith 43599b94acceSBarry Smith Output Parameter: 43600298fd71SBarry Smith + r - the vector that is used to store residuals (or NULL if you don't want it) 4361f8b49ee9SBarry Smith . f - the function (or NULL if you don't want it); see SNESFunction for calling sequence details 43620298fd71SBarry Smith - ctx - the function context (or NULL if you don't want it) 43639b94acceSBarry Smith 436436851e7fSLois Curfman McInnes Level: advanced 436536851e7fSLois Curfman McInnes 4366a86d99e1SLois Curfman McInnes .keywords: SNES, nonlinear, get, function 43679b94acceSBarry Smith 4368bf388a1fSBarry Smith .seealso: SNESSetFunction(), SNESGetSolution(), SNESFunction 43699b94acceSBarry Smith @*/ 4370f8b49ee9SBarry Smith PetscErrorCode SNESGetFunction(SNES snes,Vec *r,PetscErrorCode (**f)(SNES,Vec,Vec,void*),void **ctx) 43719b94acceSBarry Smith { 4372a63bb30eSJed Brown PetscErrorCode ierr; 43736cab3a1bSJed Brown DM dm; 4374a63bb30eSJed Brown 43753a40ed3dSBarry Smith PetscFunctionBegin; 43760700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 4377a63bb30eSJed Brown if (r) { 4378a63bb30eSJed Brown if (!snes->vec_func) { 4379a63bb30eSJed Brown if (snes->vec_rhs) { 4380a63bb30eSJed Brown ierr = VecDuplicate(snes->vec_rhs,&snes->vec_func);CHKERRQ(ierr); 4381a63bb30eSJed Brown } else if (snes->vec_sol) { 4382a63bb30eSJed Brown ierr = VecDuplicate(snes->vec_sol,&snes->vec_func);CHKERRQ(ierr); 4383a63bb30eSJed Brown } else if (snes->dm) { 4384a63bb30eSJed Brown ierr = DMCreateGlobalVector(snes->dm,&snes->vec_func);CHKERRQ(ierr); 4385a63bb30eSJed Brown } 4386a63bb30eSJed Brown } 4387a63bb30eSJed Brown *r = snes->vec_func; 4388a63bb30eSJed Brown } 43896cab3a1bSJed Brown ierr = SNESGetDM(snes,&dm);CHKERRQ(ierr); 4390f8b49ee9SBarry Smith ierr = DMSNESGetFunction(dm,f,ctx);CHKERRQ(ierr); 43913a40ed3dSBarry Smith PetscFunctionReturn(0); 43929b94acceSBarry Smith } 43939b94acceSBarry Smith 4394c79ef259SPeter Brune /*@C 4395be95d8f1SBarry Smith SNESGetNGS - Returns the NGS function and context. 4396c79ef259SPeter Brune 4397c79ef259SPeter Brune Input Parameter: 4398c79ef259SPeter Brune . snes - the SNES context 4399c79ef259SPeter Brune 4400c79ef259SPeter Brune Output Parameter: 4401be95d8f1SBarry Smith + f - the function (or NULL) see SNESNGSFunction for details 44020298fd71SBarry Smith - ctx - the function context (or NULL) 4403c79ef259SPeter Brune 4404c79ef259SPeter Brune Level: advanced 4405c79ef259SPeter Brune 4406c79ef259SPeter Brune .keywords: SNES, nonlinear, get, function 4407c79ef259SPeter Brune 4408be95d8f1SBarry Smith .seealso: SNESSetNGS(), SNESGetFunction() 4409c79ef259SPeter Brune @*/ 4410c79ef259SPeter Brune 4411be95d8f1SBarry Smith PetscErrorCode SNESGetNGS (SNES snes, PetscErrorCode (**f)(SNES, Vec, Vec, void*), void ** ctx) 4412646217ecSPeter Brune { 44136cab3a1bSJed Brown PetscErrorCode ierr; 44146cab3a1bSJed Brown DM dm; 44156cab3a1bSJed Brown 4416646217ecSPeter Brune PetscFunctionBegin; 4417646217ecSPeter Brune PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 44186cab3a1bSJed Brown ierr = SNESGetDM(snes,&dm);CHKERRQ(ierr); 4419be95d8f1SBarry Smith ierr = DMSNESGetNGS(dm,f,ctx);CHKERRQ(ierr); 4420646217ecSPeter Brune PetscFunctionReturn(0); 4421646217ecSPeter Brune } 4422646217ecSPeter Brune 44233c7409f5SSatish Balay /*@C 44243c7409f5SSatish Balay SNESSetOptionsPrefix - Sets the prefix used for searching for all 4425d850072dSLois Curfman McInnes SNES options in the database. 44263c7409f5SSatish Balay 44273f9fe445SBarry Smith Logically Collective on SNES 4428fee21e36SBarry Smith 4429c7afd0dbSLois Curfman McInnes Input Parameter: 4430c7afd0dbSLois Curfman McInnes + snes - the SNES context 4431c7afd0dbSLois Curfman McInnes - prefix - the prefix to prepend to all option names 4432c7afd0dbSLois Curfman McInnes 4433d850072dSLois Curfman McInnes Notes: 4434a83b1b31SSatish Balay A hyphen (-) must NOT be given at the beginning of the prefix name. 4435c7afd0dbSLois Curfman McInnes The first character of all runtime options is AUTOMATICALLY the hyphen. 4436d850072dSLois Curfman McInnes 443736851e7fSLois Curfman McInnes Level: advanced 443836851e7fSLois Curfman McInnes 44393c7409f5SSatish Balay .keywords: SNES, set, options, prefix, database 4440a86d99e1SLois Curfman McInnes 4441a86d99e1SLois Curfman McInnes .seealso: SNESSetFromOptions() 44423c7409f5SSatish Balay @*/ 44437087cfbeSBarry Smith PetscErrorCode SNESSetOptionsPrefix(SNES snes,const char prefix[]) 44443c7409f5SSatish Balay { 4445dfbe8321SBarry Smith PetscErrorCode ierr; 44463c7409f5SSatish Balay 44473a40ed3dSBarry Smith PetscFunctionBegin; 44480700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 4449639f9d9dSBarry Smith ierr = PetscObjectSetOptionsPrefix((PetscObject)snes,prefix);CHKERRQ(ierr); 44501cee3971SBarry Smith if (!snes->ksp) {ierr = SNESGetKSP(snes,&snes->ksp);CHKERRQ(ierr);} 445135f5d045SPeter Brune if (snes->linesearch) { 44527601faf0SJed Brown ierr = SNESGetLineSearch(snes,&snes->linesearch);CHKERRQ(ierr); 445308b6c495SPeter Brune ierr = PetscObjectSetOptionsPrefix((PetscObject)snes->linesearch,prefix);CHKERRQ(ierr); 445435f5d045SPeter Brune } 445535f5d045SPeter Brune ierr = KSPSetOptionsPrefix(snes->ksp,prefix);CHKERRQ(ierr); 44563a40ed3dSBarry Smith PetscFunctionReturn(0); 44573c7409f5SSatish Balay } 44583c7409f5SSatish Balay 44593c7409f5SSatish Balay /*@C 4460f525115eSLois Curfman McInnes SNESAppendOptionsPrefix - Appends to the prefix used for searching for all 4461d850072dSLois Curfman McInnes SNES options in the database. 44623c7409f5SSatish Balay 44633f9fe445SBarry Smith Logically Collective on SNES 4464fee21e36SBarry Smith 4465c7afd0dbSLois Curfman McInnes Input Parameters: 4466c7afd0dbSLois Curfman McInnes + snes - the SNES context 4467c7afd0dbSLois Curfman McInnes - prefix - the prefix to prepend to all option names 4468c7afd0dbSLois Curfman McInnes 4469d850072dSLois Curfman McInnes Notes: 4470a83b1b31SSatish Balay A hyphen (-) must NOT be given at the beginning of the prefix name. 4471c7afd0dbSLois Curfman McInnes The first character of all runtime options is AUTOMATICALLY the hyphen. 4472d850072dSLois Curfman McInnes 447336851e7fSLois Curfman McInnes Level: advanced 447436851e7fSLois Curfman McInnes 44753c7409f5SSatish Balay .keywords: SNES, append, options, prefix, database 4476a86d99e1SLois Curfman McInnes 4477a86d99e1SLois Curfman McInnes .seealso: SNESGetOptionsPrefix() 44783c7409f5SSatish Balay @*/ 44797087cfbeSBarry Smith PetscErrorCode SNESAppendOptionsPrefix(SNES snes,const char prefix[]) 44803c7409f5SSatish Balay { 4481dfbe8321SBarry Smith PetscErrorCode ierr; 44823c7409f5SSatish Balay 44833a40ed3dSBarry Smith PetscFunctionBegin; 44840700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 4485639f9d9dSBarry Smith ierr = PetscObjectAppendOptionsPrefix((PetscObject)snes,prefix);CHKERRQ(ierr); 44861cee3971SBarry Smith if (!snes->ksp) {ierr = SNESGetKSP(snes,&snes->ksp);CHKERRQ(ierr);} 448735f5d045SPeter Brune if (snes->linesearch) { 44887601faf0SJed Brown ierr = SNESGetLineSearch(snes,&snes->linesearch);CHKERRQ(ierr); 448908b6c495SPeter Brune ierr = PetscObjectAppendOptionsPrefix((PetscObject)snes->linesearch,prefix);CHKERRQ(ierr); 449035f5d045SPeter Brune } 449135f5d045SPeter Brune ierr = KSPAppendOptionsPrefix(snes->ksp,prefix);CHKERRQ(ierr); 44923a40ed3dSBarry Smith PetscFunctionReturn(0); 44933c7409f5SSatish Balay } 44943c7409f5SSatish Balay 44959ab63eb5SSatish Balay /*@C 44963c7409f5SSatish Balay SNESGetOptionsPrefix - Sets the prefix used for searching for all 44973c7409f5SSatish Balay SNES options in the database. 44983c7409f5SSatish Balay 4499c7afd0dbSLois Curfman McInnes Not Collective 4500c7afd0dbSLois Curfman McInnes 45013c7409f5SSatish Balay Input Parameter: 45023c7409f5SSatish Balay . snes - the SNES context 45033c7409f5SSatish Balay 45043c7409f5SSatish Balay Output Parameter: 45053c7409f5SSatish Balay . prefix - pointer to the prefix string used 45063c7409f5SSatish Balay 45074ef407dbSRichard Tran Mills Notes: On the fortran side, the user should pass in a string 'prefix' of 45089ab63eb5SSatish Balay sufficient length to hold the prefix. 45099ab63eb5SSatish Balay 451036851e7fSLois Curfman McInnes Level: advanced 451136851e7fSLois Curfman McInnes 45123c7409f5SSatish Balay .keywords: SNES, get, options, prefix, database 4513a86d99e1SLois Curfman McInnes 4514a86d99e1SLois Curfman McInnes .seealso: SNESAppendOptionsPrefix() 45153c7409f5SSatish Balay @*/ 45167087cfbeSBarry Smith PetscErrorCode SNESGetOptionsPrefix(SNES snes,const char *prefix[]) 45173c7409f5SSatish Balay { 4518dfbe8321SBarry Smith PetscErrorCode ierr; 45193c7409f5SSatish Balay 45203a40ed3dSBarry Smith PetscFunctionBegin; 45210700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 4522639f9d9dSBarry Smith ierr = PetscObjectGetOptionsPrefix((PetscObject)snes,prefix);CHKERRQ(ierr); 45233a40ed3dSBarry Smith PetscFunctionReturn(0); 45243c7409f5SSatish Balay } 45253c7409f5SSatish Balay 4526b2002411SLois Curfman McInnes 45273cea93caSBarry Smith /*@C 45281c84c290SBarry Smith SNESRegister - Adds a method to the nonlinear solver package. 45291c84c290SBarry Smith 45301c84c290SBarry Smith Not collective 45311c84c290SBarry Smith 45321c84c290SBarry Smith Input Parameters: 45331c84c290SBarry Smith + name_solver - name of a new user-defined solver 45341c84c290SBarry Smith - routine_create - routine to create method context 45351c84c290SBarry Smith 45361c84c290SBarry Smith Notes: 45371c84c290SBarry Smith SNESRegister() may be called multiple times to add several user-defined solvers. 45381c84c290SBarry Smith 45391c84c290SBarry Smith Sample usage: 45401c84c290SBarry Smith .vb 4541bdf89e91SBarry Smith SNESRegister("my_solver",MySolverCreate); 45421c84c290SBarry Smith .ve 45431c84c290SBarry Smith 45441c84c290SBarry Smith Then, your solver can be chosen with the procedural interface via 45451c84c290SBarry Smith $ SNESSetType(snes,"my_solver") 45461c84c290SBarry Smith or at runtime via the option 45471c84c290SBarry Smith $ -snes_type my_solver 45481c84c290SBarry Smith 45491c84c290SBarry Smith Level: advanced 45501c84c290SBarry Smith 45511c84c290SBarry Smith Note: If your function is not being put into a shared library then use SNESRegister() instead 45521c84c290SBarry Smith 45531c84c290SBarry Smith .keywords: SNES, nonlinear, register 45541c84c290SBarry Smith 45551c84c290SBarry Smith .seealso: SNESRegisterAll(), SNESRegisterDestroy() 45563cea93caSBarry Smith 45577f6c08e0SMatthew Knepley Level: advanced 45583cea93caSBarry Smith @*/ 4559bdf89e91SBarry Smith PetscErrorCode SNESRegister(const char sname[],PetscErrorCode (*function)(SNES)) 4560b2002411SLois Curfman McInnes { 4561dfbe8321SBarry Smith PetscErrorCode ierr; 4562b2002411SLois Curfman McInnes 4563b2002411SLois Curfman McInnes PetscFunctionBegin; 4564a240a19fSJed Brown ierr = PetscFunctionListAdd(&SNESList,sname,function);CHKERRQ(ierr); 4565b2002411SLois Curfman McInnes PetscFunctionReturn(0); 4566b2002411SLois Curfman McInnes } 4567da9b6338SBarry Smith 45687087cfbeSBarry Smith PetscErrorCode SNESTestLocalMin(SNES snes) 4569da9b6338SBarry Smith { 4570dfbe8321SBarry Smith PetscErrorCode ierr; 457177431f27SBarry Smith PetscInt N,i,j; 4572da9b6338SBarry Smith Vec u,uh,fh; 4573da9b6338SBarry Smith PetscScalar value; 4574da9b6338SBarry Smith PetscReal norm; 4575da9b6338SBarry Smith 4576da9b6338SBarry Smith PetscFunctionBegin; 4577da9b6338SBarry Smith ierr = SNESGetSolution(snes,&u);CHKERRQ(ierr); 4578da9b6338SBarry Smith ierr = VecDuplicate(u,&uh);CHKERRQ(ierr); 4579da9b6338SBarry Smith ierr = VecDuplicate(u,&fh);CHKERRQ(ierr); 4580da9b6338SBarry Smith 4581da9b6338SBarry Smith /* currently only works for sequential */ 458222d28d08SBarry Smith ierr = PetscPrintf(PETSC_COMM_WORLD,"Testing FormFunction() for local min\n");CHKERRQ(ierr); 4583da9b6338SBarry Smith ierr = VecGetSize(u,&N);CHKERRQ(ierr); 4584da9b6338SBarry Smith for (i=0; i<N; i++) { 4585da9b6338SBarry Smith ierr = VecCopy(u,uh);CHKERRQ(ierr); 458677431f27SBarry Smith ierr = PetscPrintf(PETSC_COMM_WORLD,"i = %D\n",i);CHKERRQ(ierr); 4587da9b6338SBarry Smith for (j=-10; j<11; j++) { 45888b49ba18SBarry Smith value = PetscSign(j)*PetscExpReal(PetscAbs(j)-10.0); 4589da9b6338SBarry Smith ierr = VecSetValue(uh,i,value,ADD_VALUES);CHKERRQ(ierr); 45903ab0aad5SBarry Smith ierr = SNESComputeFunction(snes,uh,fh);CHKERRQ(ierr); 4591da9b6338SBarry Smith ierr = VecNorm(fh,NORM_2,&norm);CHKERRQ(ierr); 459277431f27SBarry Smith ierr = PetscPrintf(PETSC_COMM_WORLD," j norm %D %18.16e\n",j,norm);CHKERRQ(ierr); 4593da9b6338SBarry Smith value = -value; 4594da9b6338SBarry Smith ierr = VecSetValue(uh,i,value,ADD_VALUES);CHKERRQ(ierr); 4595da9b6338SBarry Smith } 4596da9b6338SBarry Smith } 45976bf464f9SBarry Smith ierr = VecDestroy(&uh);CHKERRQ(ierr); 45986bf464f9SBarry Smith ierr = VecDestroy(&fh);CHKERRQ(ierr); 4599da9b6338SBarry Smith PetscFunctionReturn(0); 4600da9b6338SBarry Smith } 460171f87433Sdalcinl 460271f87433Sdalcinl /*@ 4603fa9f3622SBarry Smith SNESKSPSetUseEW - Sets SNES use Eisenstat-Walker method for 460471f87433Sdalcinl computing relative tolerance for linear solvers within an inexact 460571f87433Sdalcinl Newton method. 460671f87433Sdalcinl 46073f9fe445SBarry Smith Logically Collective on SNES 460871f87433Sdalcinl 460971f87433Sdalcinl Input Parameters: 461071f87433Sdalcinl + snes - SNES context 461171f87433Sdalcinl - flag - PETSC_TRUE or PETSC_FALSE 461271f87433Sdalcinl 461364ba62caSBarry Smith Options Database: 461464ba62caSBarry Smith + -snes_ksp_ew - use Eisenstat-Walker method for determining linear system convergence 461564ba62caSBarry Smith . -snes_ksp_ew_version ver - version of Eisenstat-Walker method 461664ba62caSBarry Smith . -snes_ksp_ew_rtol0 <rtol0> - Sets rtol0 461764ba62caSBarry Smith . -snes_ksp_ew_rtolmax <rtolmax> - Sets rtolmax 461864ba62caSBarry Smith . -snes_ksp_ew_gamma <gamma> - Sets gamma 461964ba62caSBarry Smith . -snes_ksp_ew_alpha <alpha> - Sets alpha 462064ba62caSBarry Smith . -snes_ksp_ew_alpha2 <alpha2> - Sets alpha2 462164ba62caSBarry Smith - -snes_ksp_ew_threshold <threshold> - Sets threshold 462264ba62caSBarry Smith 462371f87433Sdalcinl Notes: 462471f87433Sdalcinl Currently, the default is to use a constant relative tolerance for 462571f87433Sdalcinl the inner linear solvers. Alternatively, one can use the 462671f87433Sdalcinl Eisenstat-Walker method, where the relative convergence tolerance 462771f87433Sdalcinl is reset at each Newton iteration according progress of the nonlinear 462871f87433Sdalcinl solver. 462971f87433Sdalcinl 463071f87433Sdalcinl Level: advanced 463171f87433Sdalcinl 463271f87433Sdalcinl Reference: 463371f87433Sdalcinl S. C. Eisenstat and H. F. Walker, "Choosing the forcing terms in an 463471f87433Sdalcinl inexact Newton method", SISC 17 (1), pp.16-32, 1996. 463571f87433Sdalcinl 463671f87433Sdalcinl .keywords: SNES, KSP, Eisenstat, Walker, convergence, test, inexact, Newton 463771f87433Sdalcinl 4638fa9f3622SBarry Smith .seealso: SNESKSPGetUseEW(), SNESKSPGetParametersEW(), SNESKSPSetParametersEW() 463971f87433Sdalcinl @*/ 46407087cfbeSBarry Smith PetscErrorCode SNESKSPSetUseEW(SNES snes,PetscBool flag) 464171f87433Sdalcinl { 464271f87433Sdalcinl PetscFunctionBegin; 46430700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 4644acfcf0e5SJed Brown PetscValidLogicalCollectiveBool(snes,flag,2); 464571f87433Sdalcinl snes->ksp_ewconv = flag; 464671f87433Sdalcinl PetscFunctionReturn(0); 464771f87433Sdalcinl } 464871f87433Sdalcinl 464971f87433Sdalcinl /*@ 4650fa9f3622SBarry Smith SNESKSPGetUseEW - Gets if SNES is using Eisenstat-Walker method 465171f87433Sdalcinl for computing relative tolerance for linear solvers within an 465271f87433Sdalcinl inexact Newton method. 465371f87433Sdalcinl 465471f87433Sdalcinl Not Collective 465571f87433Sdalcinl 465671f87433Sdalcinl Input Parameter: 465771f87433Sdalcinl . snes - SNES context 465871f87433Sdalcinl 465971f87433Sdalcinl Output Parameter: 466071f87433Sdalcinl . flag - PETSC_TRUE or PETSC_FALSE 466171f87433Sdalcinl 466271f87433Sdalcinl Notes: 466371f87433Sdalcinl Currently, the default is to use a constant relative tolerance for 466471f87433Sdalcinl the inner linear solvers. Alternatively, one can use the 466571f87433Sdalcinl Eisenstat-Walker method, where the relative convergence tolerance 466671f87433Sdalcinl is reset at each Newton iteration according progress of the nonlinear 466771f87433Sdalcinl solver. 466871f87433Sdalcinl 466971f87433Sdalcinl Level: advanced 467071f87433Sdalcinl 467171f87433Sdalcinl Reference: 467271f87433Sdalcinl S. C. Eisenstat and H. F. Walker, "Choosing the forcing terms in an 467371f87433Sdalcinl inexact Newton method", SISC 17 (1), pp.16-32, 1996. 467471f87433Sdalcinl 467571f87433Sdalcinl .keywords: SNES, KSP, Eisenstat, Walker, convergence, test, inexact, Newton 467671f87433Sdalcinl 4677fa9f3622SBarry Smith .seealso: SNESKSPSetUseEW(), SNESKSPGetParametersEW(), SNESKSPSetParametersEW() 467871f87433Sdalcinl @*/ 46797087cfbeSBarry Smith PetscErrorCode SNESKSPGetUseEW(SNES snes, PetscBool *flag) 468071f87433Sdalcinl { 468171f87433Sdalcinl PetscFunctionBegin; 46820700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 468371f87433Sdalcinl PetscValidPointer(flag,2); 468471f87433Sdalcinl *flag = snes->ksp_ewconv; 468571f87433Sdalcinl PetscFunctionReturn(0); 468671f87433Sdalcinl } 468771f87433Sdalcinl 468871f87433Sdalcinl /*@ 4689fa9f3622SBarry Smith SNESKSPSetParametersEW - Sets parameters for Eisenstat-Walker 469071f87433Sdalcinl convergence criteria for the linear solvers within an inexact 469171f87433Sdalcinl Newton method. 469271f87433Sdalcinl 46933f9fe445SBarry Smith Logically Collective on SNES 469471f87433Sdalcinl 469571f87433Sdalcinl Input Parameters: 469671f87433Sdalcinl + snes - SNES context 469771f87433Sdalcinl . version - version 1, 2 (default is 2) or 3 469871f87433Sdalcinl . rtol_0 - initial relative tolerance (0 <= rtol_0 < 1) 469971f87433Sdalcinl . rtol_max - maximum relative tolerance (0 <= rtol_max < 1) 470071f87433Sdalcinl . gamma - multiplicative factor for version 2 rtol computation 470171f87433Sdalcinl (0 <= gamma2 <= 1) 470271f87433Sdalcinl . alpha - power for version 2 rtol computation (1 < alpha <= 2) 470371f87433Sdalcinl . alpha2 - power for safeguard 470471f87433Sdalcinl - threshold - threshold for imposing safeguard (0 < threshold < 1) 470571f87433Sdalcinl 470671f87433Sdalcinl Note: 470771f87433Sdalcinl Version 3 was contributed by Luis Chacon, June 2006. 470871f87433Sdalcinl 470971f87433Sdalcinl Use PETSC_DEFAULT to retain the default for any of the parameters. 471071f87433Sdalcinl 471171f87433Sdalcinl Level: advanced 471271f87433Sdalcinl 471371f87433Sdalcinl Reference: 471471f87433Sdalcinl S. C. Eisenstat and H. F. Walker, "Choosing the forcing terms in an 471571f87433Sdalcinl inexact Newton method", Utah State University Math. Stat. Dept. Res. 471671f87433Sdalcinl Report 6/94/75, June, 1994, to appear in SIAM J. Sci. Comput. 471771f87433Sdalcinl 471871f87433Sdalcinl .keywords: SNES, KSP, Eisenstat, Walker, set, parameters 471971f87433Sdalcinl 4720fa9f3622SBarry Smith .seealso: SNESKSPSetUseEW(), SNESKSPGetUseEW(), SNESKSPGetParametersEW() 472171f87433Sdalcinl @*/ 4722f5af7f23SKarl Rupp PetscErrorCode SNESKSPSetParametersEW(SNES snes,PetscInt version,PetscReal rtol_0,PetscReal rtol_max,PetscReal gamma,PetscReal alpha,PetscReal alpha2,PetscReal threshold) 472371f87433Sdalcinl { 4724fa9f3622SBarry Smith SNESKSPEW *kctx; 47255fd66863SKarl Rupp 472671f87433Sdalcinl PetscFunctionBegin; 47270700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 4728fa9f3622SBarry Smith kctx = (SNESKSPEW*)snes->kspconvctx; 4729e32f2f54SBarry Smith if (!kctx) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"No Eisenstat-Walker context existing"); 4730c5eb9154SBarry Smith PetscValidLogicalCollectiveInt(snes,version,2); 4731c5eb9154SBarry Smith PetscValidLogicalCollectiveReal(snes,rtol_0,3); 4732c5eb9154SBarry Smith PetscValidLogicalCollectiveReal(snes,rtol_max,4); 4733c5eb9154SBarry Smith PetscValidLogicalCollectiveReal(snes,gamma,5); 4734c5eb9154SBarry Smith PetscValidLogicalCollectiveReal(snes,alpha,6); 4735c5eb9154SBarry Smith PetscValidLogicalCollectiveReal(snes,alpha2,7); 4736c5eb9154SBarry Smith PetscValidLogicalCollectiveReal(snes,threshold,8); 473771f87433Sdalcinl 473871f87433Sdalcinl if (version != PETSC_DEFAULT) kctx->version = version; 473971f87433Sdalcinl if (rtol_0 != PETSC_DEFAULT) kctx->rtol_0 = rtol_0; 474071f87433Sdalcinl if (rtol_max != PETSC_DEFAULT) kctx->rtol_max = rtol_max; 474171f87433Sdalcinl if (gamma != PETSC_DEFAULT) kctx->gamma = gamma; 474271f87433Sdalcinl if (alpha != PETSC_DEFAULT) kctx->alpha = alpha; 474371f87433Sdalcinl if (alpha2 != PETSC_DEFAULT) kctx->alpha2 = alpha2; 474471f87433Sdalcinl if (threshold != PETSC_DEFAULT) kctx->threshold = threshold; 474571f87433Sdalcinl 4746f23aa3ddSBarry 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); 474757622a8eSBarry 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); 474857622a8eSBarry 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); 474957622a8eSBarry 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); 475057622a8eSBarry 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); 475157622a8eSBarry 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); 475271f87433Sdalcinl PetscFunctionReturn(0); 475371f87433Sdalcinl } 475471f87433Sdalcinl 475571f87433Sdalcinl /*@ 4756fa9f3622SBarry Smith SNESKSPGetParametersEW - Gets parameters for Eisenstat-Walker 475771f87433Sdalcinl convergence criteria for the linear solvers within an inexact 475871f87433Sdalcinl Newton method. 475971f87433Sdalcinl 476071f87433Sdalcinl Not Collective 476171f87433Sdalcinl 476271f87433Sdalcinl Input Parameters: 476371f87433Sdalcinl snes - SNES context 476471f87433Sdalcinl 476571f87433Sdalcinl Output Parameters: 476671f87433Sdalcinl + version - version 1, 2 (default is 2) or 3 476771f87433Sdalcinl . rtol_0 - initial relative tolerance (0 <= rtol_0 < 1) 476871f87433Sdalcinl . rtol_max - maximum relative tolerance (0 <= rtol_max < 1) 4769bf388a1fSBarry Smith . gamma - multiplicative factor for version 2 rtol computation (0 <= gamma2 <= 1) 477071f87433Sdalcinl . alpha - power for version 2 rtol computation (1 < alpha <= 2) 477171f87433Sdalcinl . alpha2 - power for safeguard 477271f87433Sdalcinl - threshold - threshold for imposing safeguard (0 < threshold < 1) 477371f87433Sdalcinl 477471f87433Sdalcinl Level: advanced 477571f87433Sdalcinl 477671f87433Sdalcinl .keywords: SNES, KSP, Eisenstat, Walker, get, parameters 477771f87433Sdalcinl 4778fa9f3622SBarry Smith .seealso: SNESKSPSetUseEW(), SNESKSPGetUseEW(), SNESKSPSetParametersEW() 477971f87433Sdalcinl @*/ 4780bf388a1fSBarry Smith PetscErrorCode SNESKSPGetParametersEW(SNES snes,PetscInt *version,PetscReal *rtol_0,PetscReal *rtol_max,PetscReal *gamma,PetscReal *alpha,PetscReal *alpha2,PetscReal *threshold) 478171f87433Sdalcinl { 4782fa9f3622SBarry Smith SNESKSPEW *kctx; 47835fd66863SKarl Rupp 478471f87433Sdalcinl PetscFunctionBegin; 47850700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 4786fa9f3622SBarry Smith kctx = (SNESKSPEW*)snes->kspconvctx; 4787e32f2f54SBarry Smith if (!kctx) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"No Eisenstat-Walker context existing"); 478871f87433Sdalcinl if (version) *version = kctx->version; 478971f87433Sdalcinl if (rtol_0) *rtol_0 = kctx->rtol_0; 479071f87433Sdalcinl if (rtol_max) *rtol_max = kctx->rtol_max; 479171f87433Sdalcinl if (gamma) *gamma = kctx->gamma; 479271f87433Sdalcinl if (alpha) *alpha = kctx->alpha; 479371f87433Sdalcinl if (alpha2) *alpha2 = kctx->alpha2; 479471f87433Sdalcinl if (threshold) *threshold = kctx->threshold; 479571f87433Sdalcinl PetscFunctionReturn(0); 479671f87433Sdalcinl } 479771f87433Sdalcinl 4798d5378b5fSDmitry Karpeev PetscErrorCode KSPPreSolve_SNESEW(KSP ksp, Vec b, Vec x, SNES snes) 479971f87433Sdalcinl { 480071f87433Sdalcinl PetscErrorCode ierr; 4801fa9f3622SBarry Smith SNESKSPEW *kctx = (SNESKSPEW*)snes->kspconvctx; 480271f87433Sdalcinl PetscReal rtol = PETSC_DEFAULT,stol; 480371f87433Sdalcinl 480471f87433Sdalcinl PetscFunctionBegin; 4805d4211eb9SBarry Smith if (!snes->ksp_ewconv) PetscFunctionReturn(0); 480630058271SDmitry Karpeev if (!snes->iter) { 480730058271SDmitry Karpeev rtol = kctx->rtol_0; /* first time in, so use the original user rtol */ 480830058271SDmitry Karpeev ierr = VecNorm(snes->vec_func,NORM_2,&kctx->norm_first);CHKERRQ(ierr); 480930058271SDmitry Karpeev } 4810f5af7f23SKarl Rupp else { 481171f87433Sdalcinl if (kctx->version == 1) { 481271f87433Sdalcinl rtol = (snes->norm - kctx->lresid_last)/kctx->norm_last; 481371f87433Sdalcinl if (rtol < 0.0) rtol = -rtol; 481485ec1a3cSBarry Smith stol = PetscPowReal(kctx->rtol_last,kctx->alpha2); 481571f87433Sdalcinl if (stol > kctx->threshold) rtol = PetscMax(rtol,stol); 481671f87433Sdalcinl } else if (kctx->version == 2) { 481785ec1a3cSBarry Smith rtol = kctx->gamma * PetscPowReal(snes->norm/kctx->norm_last,kctx->alpha); 481885ec1a3cSBarry Smith stol = kctx->gamma * PetscPowReal(kctx->rtol_last,kctx->alpha); 481971f87433Sdalcinl if (stol > kctx->threshold) rtol = PetscMax(rtol,stol); 482071f87433Sdalcinl } else if (kctx->version == 3) { /* contributed by Luis Chacon, June 2006. */ 482185ec1a3cSBarry Smith rtol = kctx->gamma * PetscPowReal(snes->norm/kctx->norm_last,kctx->alpha); 482271f87433Sdalcinl /* safeguard: avoid sharp decrease of rtol */ 482385ec1a3cSBarry Smith stol = kctx->gamma*PetscPowReal(kctx->rtol_last,kctx->alpha); 482471f87433Sdalcinl stol = PetscMax(rtol,stol); 482571f87433Sdalcinl rtol = PetscMin(kctx->rtol_0,stol); 482671f87433Sdalcinl /* safeguard: avoid oversolving */ 482730058271SDmitry Karpeev stol = kctx->gamma*(kctx->norm_first*snes->rtol)/snes->norm; 482871f87433Sdalcinl stol = PetscMax(rtol,stol); 482971f87433Sdalcinl rtol = PetscMin(kctx->rtol_0,stol); 4830e32f2f54SBarry Smith } else SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Only versions 1, 2 or 3 are supported: %D",kctx->version); 483171f87433Sdalcinl } 483271f87433Sdalcinl /* safeguard: avoid rtol greater than one */ 483371f87433Sdalcinl rtol = PetscMin(rtol,kctx->rtol_max); 483471f87433Sdalcinl ierr = KSPSetTolerances(ksp,rtol,PETSC_DEFAULT,PETSC_DEFAULT,PETSC_DEFAULT);CHKERRQ(ierr); 483557622a8eSBarry Smith ierr = PetscInfo3(snes,"iter %D, Eisenstat-Walker (version %D) KSP rtol=%g\n",snes->iter,kctx->version,(double)rtol);CHKERRQ(ierr); 483671f87433Sdalcinl PetscFunctionReturn(0); 483771f87433Sdalcinl } 483871f87433Sdalcinl 4839d5378b5fSDmitry Karpeev PetscErrorCode KSPPostSolve_SNESEW(KSP ksp, Vec b, Vec x, SNES snes) 484071f87433Sdalcinl { 484171f87433Sdalcinl PetscErrorCode ierr; 4842fa9f3622SBarry Smith SNESKSPEW *kctx = (SNESKSPEW*)snes->kspconvctx; 484371f87433Sdalcinl PCSide pcside; 484471f87433Sdalcinl Vec lres; 484571f87433Sdalcinl 484671f87433Sdalcinl PetscFunctionBegin; 4847d4211eb9SBarry Smith if (!snes->ksp_ewconv) PetscFunctionReturn(0); 484871f87433Sdalcinl ierr = KSPGetTolerances(ksp,&kctx->rtol_last,0,0,0);CHKERRQ(ierr); 484971dbe336SPeter Brune kctx->norm_last = snes->norm; 485071f87433Sdalcinl if (kctx->version == 1) { 48514f00ce20SMatthew G. Knepley PC pc; 48524f00ce20SMatthew G. Knepley PetscBool isNone; 48534f00ce20SMatthew G. Knepley 48544f00ce20SMatthew G. Knepley ierr = KSPGetPC(ksp, &pc);CHKERRQ(ierr); 48554f00ce20SMatthew G. Knepley ierr = PetscObjectTypeCompare((PetscObject) pc, PCNONE, &isNone);CHKERRQ(ierr); 4856b037da10SBarry Smith ierr = KSPGetPCSide(ksp,&pcside);CHKERRQ(ierr); 48574f00ce20SMatthew G. Knepley if (pcside == PC_RIGHT || isNone) { /* XXX Should we also test KSP_UNPRECONDITIONED_NORM ? */ 485871f87433Sdalcinl /* KSP residual is true linear residual */ 485971f87433Sdalcinl ierr = KSPGetResidualNorm(ksp,&kctx->lresid_last);CHKERRQ(ierr); 486071f87433Sdalcinl } else { 486171f87433Sdalcinl /* KSP residual is preconditioned residual */ 486271f87433Sdalcinl /* compute true linear residual norm */ 486371f87433Sdalcinl ierr = VecDuplicate(b,&lres);CHKERRQ(ierr); 486471f87433Sdalcinl ierr = MatMult(snes->jacobian,x,lres);CHKERRQ(ierr); 486571f87433Sdalcinl ierr = VecAYPX(lres,-1.0,b);CHKERRQ(ierr); 486671f87433Sdalcinl ierr = VecNorm(lres,NORM_2,&kctx->lresid_last);CHKERRQ(ierr); 48676bf464f9SBarry Smith ierr = VecDestroy(&lres);CHKERRQ(ierr); 486871f87433Sdalcinl } 486971f87433Sdalcinl } 487071f87433Sdalcinl PetscFunctionReturn(0); 487171f87433Sdalcinl } 487271f87433Sdalcinl 4873d4211eb9SBarry Smith /*@ 4874d4211eb9SBarry Smith SNESGetKSP - Returns the KSP context for a SNES solver. 4875d4211eb9SBarry Smith 4876d4211eb9SBarry Smith Not Collective, but if SNES object is parallel, then KSP object is parallel 4877d4211eb9SBarry Smith 4878d4211eb9SBarry Smith Input Parameter: 4879d4211eb9SBarry Smith . snes - the SNES context 4880d4211eb9SBarry Smith 4881d4211eb9SBarry Smith Output Parameter: 4882d4211eb9SBarry Smith . ksp - the KSP context 4883d4211eb9SBarry Smith 4884d4211eb9SBarry Smith Notes: 4885d4211eb9SBarry Smith The user can then directly manipulate the KSP context to set various 4886d4211eb9SBarry Smith options, etc. Likewise, the user can then extract and manipulate the 4887d4211eb9SBarry Smith PC contexts as well. 4888d4211eb9SBarry Smith 4889d4211eb9SBarry Smith Level: beginner 4890d4211eb9SBarry Smith 4891d4211eb9SBarry Smith .keywords: SNES, nonlinear, get, KSP, context 4892d4211eb9SBarry Smith 4893d4211eb9SBarry Smith .seealso: KSPGetPC(), SNESCreate(), KSPCreate(), SNESSetKSP() 4894d4211eb9SBarry Smith @*/ 4895d4211eb9SBarry Smith PetscErrorCode SNESGetKSP(SNES snes,KSP *ksp) 489671f87433Sdalcinl { 489771f87433Sdalcinl PetscErrorCode ierr; 489871f87433Sdalcinl 489971f87433Sdalcinl PetscFunctionBegin; 4900d4211eb9SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 4901d4211eb9SBarry Smith PetscValidPointer(ksp,2); 4902d4211eb9SBarry Smith 4903d4211eb9SBarry Smith if (!snes->ksp) { 4904a5c2985bSBarry Smith PetscBool monitor = PETSC_FALSE; 4905a5c2985bSBarry Smith 4906d4211eb9SBarry Smith ierr = KSPCreate(PetscObjectComm((PetscObject)snes),&snes->ksp);CHKERRQ(ierr); 4907d4211eb9SBarry Smith ierr = PetscObjectIncrementTabLevel((PetscObject)snes->ksp,(PetscObject)snes,1);CHKERRQ(ierr); 49083bb1ff40SBarry Smith ierr = PetscLogObjectParent((PetscObject)snes,(PetscObject)snes->ksp);CHKERRQ(ierr); 4909d4211eb9SBarry Smith 4910d5378b5fSDmitry Karpeev ierr = KSPSetPreSolve(snes->ksp,(PetscErrorCode (*)(KSP,Vec,Vec,void*))KSPPreSolve_SNESEW,snes);CHKERRQ(ierr); 4911d5378b5fSDmitry Karpeev ierr = KSPSetPostSolve(snes->ksp,(PetscErrorCode (*)(KSP,Vec,Vec,void*))KSPPostSolve_SNESEW,snes);CHKERRQ(ierr); 4912a5c2985bSBarry Smith 4913c5929fdfSBarry Smith ierr = PetscOptionsGetBool(((PetscObject)snes)->options,((PetscObject)snes)->prefix,"-ksp_monitor_snes",&monitor,NULL);CHKERRQ(ierr); 4914a5c2985bSBarry Smith if (monitor) { 4915a5c2985bSBarry Smith ierr = KSPMonitorSet(snes->ksp,KSPMonitorSNES,snes,NULL);CHKERRQ(ierr); 4916a5c2985bSBarry Smith } 4917e5f7ee39SBarry Smith monitor = PETSC_FALSE; 4918c5929fdfSBarry Smith ierr = PetscOptionsGetBool(((PetscObject)snes)->options,((PetscObject)snes)->prefix,"-ksp_monitor_snes_lg",&monitor,NULL);CHKERRQ(ierr); 4919e5f7ee39SBarry Smith if (monitor) { 4920e5f7ee39SBarry Smith PetscObject *objs; 49218b0b5a47SLisandro Dalcin ierr = KSPMonitorSNESLGResidualNormCreate(PetscObjectComm((PetscObject)snes),NULL,NULL,PETSC_DECIDE,PETSC_DECIDE,600,600,&objs);CHKERRQ(ierr); 4922e5f7ee39SBarry Smith objs[0] = (PetscObject) snes; 4923e5f7ee39SBarry Smith ierr = KSPMonitorSet(snes->ksp,(PetscErrorCode (*)(KSP,PetscInt,PetscReal,void*))KSPMonitorSNESLGResidualNorm,objs,(PetscErrorCode (*)(void**))KSPMonitorSNESLGResidualNormDestroy);CHKERRQ(ierr); 4924e5f7ee39SBarry Smith } 4925d4211eb9SBarry Smith } 4926d4211eb9SBarry Smith *ksp = snes->ksp; 492771f87433Sdalcinl PetscFunctionReturn(0); 492871f87433Sdalcinl } 49296c699258SBarry Smith 4930d4211eb9SBarry Smith 4931af0996ceSBarry Smith #include <petsc/private/dmimpl.h> 49326c699258SBarry Smith /*@ 49332a808120SBarry Smith SNESSetDM - Sets the DM that may be used by some nonlinear solvers or their underlying preconditioners 49346c699258SBarry Smith 49353f9fe445SBarry Smith Logically Collective on SNES 49366c699258SBarry Smith 49376c699258SBarry Smith Input Parameters: 49382a808120SBarry Smith + snes - the nonlinear solver context 49392a808120SBarry Smith - dm - the dm, cannot be NULL 49406c699258SBarry Smith 49416c699258SBarry Smith Level: intermediate 49426c699258SBarry Smith 494390deff25SFande Kong .seealso: SNESGetDM(), SNESHasDM(), KSPSetDM(), KSPGetDM() 49446c699258SBarry Smith @*/ 49457087cfbeSBarry Smith PetscErrorCode SNESSetDM(SNES snes,DM dm) 49466c699258SBarry Smith { 49476c699258SBarry Smith PetscErrorCode ierr; 4948345fed2cSBarry Smith KSP ksp; 4949942e3340SBarry Smith DMSNES sdm; 49506c699258SBarry Smith 49516c699258SBarry Smith PetscFunctionBegin; 49520700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 49532a808120SBarry Smith PetscValidHeaderSpecific(dm,DM_CLASSID,2); 49542a808120SBarry Smith ierr = PetscObjectReference((PetscObject)dm);CHKERRQ(ierr); 4955942e3340SBarry Smith if (snes->dm) { /* Move the DMSNES context over to the new DM unless the new DM already has one */ 495651f4b3c7SToby Isaac if (snes->dm->dmsnes && !dm->dmsnes) { 4957942e3340SBarry Smith ierr = DMCopyDMSNES(snes->dm,dm);CHKERRQ(ierr); 4958942e3340SBarry Smith ierr = DMGetDMSNES(snes->dm,&sdm);CHKERRQ(ierr); 4959f5af7f23SKarl Rupp if (sdm->originaldm == snes->dm) sdm->originaldm = dm; /* Grant write privileges to the replacement DM */ 49606cab3a1bSJed Brown } 49616bf464f9SBarry Smith ierr = DMDestroy(&snes->dm);CHKERRQ(ierr); 49626cab3a1bSJed Brown } 49636c699258SBarry Smith snes->dm = dm; 4964116d1032SJed Brown snes->dmAuto = PETSC_FALSE; 4965f5af7f23SKarl Rupp 4966345fed2cSBarry Smith ierr = SNESGetKSP(snes,&ksp);CHKERRQ(ierr); 4967345fed2cSBarry Smith ierr = KSPSetDM(ksp,dm);CHKERRQ(ierr); 4968f22f69f0SBarry Smith ierr = KSPSetDMActive(ksp,PETSC_FALSE);CHKERRQ(ierr); 4969efd4aadfSBarry Smith if (snes->npc) { 4970efd4aadfSBarry Smith ierr = SNESSetDM(snes->npc, snes->dm);CHKERRQ(ierr); 4971efd4aadfSBarry Smith ierr = SNESSetNPCSide(snes,snes->npcside);CHKERRQ(ierr); 49722c155ee1SBarry Smith } 49736c699258SBarry Smith PetscFunctionReturn(0); 49746c699258SBarry Smith } 49756c699258SBarry Smith 49766c699258SBarry Smith /*@ 49776c699258SBarry Smith SNESGetDM - Gets the DM that may be used by some preconditioners 49786c699258SBarry Smith 49793f9fe445SBarry Smith Not Collective but DM obtained is parallel on SNES 49806c699258SBarry Smith 49816c699258SBarry Smith Input Parameter: 49826c699258SBarry Smith . snes - the preconditioner context 49836c699258SBarry Smith 49846c699258SBarry Smith Output Parameter: 49856c699258SBarry Smith . dm - the dm 49866c699258SBarry Smith 49876c699258SBarry Smith Level: intermediate 49886c699258SBarry Smith 498990deff25SFande Kong .seealso: SNESSetDM(), SNESHasDM(), KSPSetDM(), KSPGetDM() 49906c699258SBarry Smith @*/ 49917087cfbeSBarry Smith PetscErrorCode SNESGetDM(SNES snes,DM *dm) 49926c699258SBarry Smith { 49936cab3a1bSJed Brown PetscErrorCode ierr; 49946cab3a1bSJed Brown 49956c699258SBarry Smith PetscFunctionBegin; 49960700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 49976cab3a1bSJed Brown if (!snes->dm) { 4998ce94432eSBarry Smith ierr = DMShellCreate(PetscObjectComm((PetscObject)snes),&snes->dm);CHKERRQ(ierr); 4999116d1032SJed Brown snes->dmAuto = PETSC_TRUE; 50006cab3a1bSJed Brown } 50016c699258SBarry Smith *dm = snes->dm; 50026c699258SBarry Smith PetscFunctionReturn(0); 50036c699258SBarry Smith } 50040807856dSBarry Smith 500590deff25SFande Kong 500690deff25SFande Kong /*@ 500790deff25SFande Kong SNESHasDM - Whether snes has dm 500890deff25SFande Kong 500990deff25SFande Kong Not collective but all processes must return the same value 501090deff25SFande Kong 501190deff25SFande Kong Input Parameter: 501290deff25SFande Kong . snes - the nonlinear solver object 501390deff25SFande Kong 501490deff25SFande Kong Output Parameter: 501590deff25SFande Kong . hasdm - a flag indicates whether there is dm in snes 501690deff25SFande Kong 501790deff25SFande Kong Level: intermediate 501890deff25SFande Kong 501990deff25SFande Kong .seealso: SNESGetDM(), SNESSetDM(), KSPSetDM(), KSPGetDM() 502090deff25SFande Kong @*/ 502190deff25SFande Kong PetscErrorCode SNESHasDM(SNES snes,PetscBool *hasdm) 502290deff25SFande Kong { 502390deff25SFande Kong PetscFunctionBegin; 502490deff25SFande Kong PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 502590deff25SFande Kong PetscValidPointer(hasdm,2); 502690deff25SFande Kong if (snes->dm) *hasdm = PETSC_TRUE; 502790deff25SFande Kong else *hasdm = PETSC_FALSE; 502890deff25SFande Kong PetscFunctionReturn(0); 502990deff25SFande Kong } 503090deff25SFande Kong 503131823bd8SMatthew G Knepley /*@ 5032be95d8f1SBarry Smith SNESSetNPC - Sets the nonlinear preconditioner to be used. 503331823bd8SMatthew G Knepley 503431823bd8SMatthew G Knepley Collective on SNES 503531823bd8SMatthew G Knepley 503631823bd8SMatthew G Knepley Input Parameters: 503731823bd8SMatthew G Knepley + snes - iterative context obtained from SNESCreate() 503831823bd8SMatthew G Knepley - pc - the preconditioner object 503931823bd8SMatthew G Knepley 504031823bd8SMatthew G Knepley Notes: 5041be95d8f1SBarry Smith Use SNESGetNPC() to retrieve the preconditioner context (for example, 504231823bd8SMatthew G Knepley to configure it using the API). 504331823bd8SMatthew G Knepley 504431823bd8SMatthew G Knepley Level: developer 504531823bd8SMatthew G Knepley 504631823bd8SMatthew G Knepley .keywords: SNES, set, precondition 50473ad1a0b9SPatrick Farrell .seealso: SNESGetNPC(), SNESHasNPC() 504831823bd8SMatthew G Knepley @*/ 5049be95d8f1SBarry Smith PetscErrorCode SNESSetNPC(SNES snes, SNES pc) 505031823bd8SMatthew G Knepley { 505131823bd8SMatthew G Knepley PetscErrorCode ierr; 505231823bd8SMatthew G Knepley 505331823bd8SMatthew G Knepley PetscFunctionBegin; 505431823bd8SMatthew G Knepley PetscValidHeaderSpecific(snes, SNES_CLASSID, 1); 505531823bd8SMatthew G Knepley PetscValidHeaderSpecific(pc, SNES_CLASSID, 2); 505631823bd8SMatthew G Knepley PetscCheckSameComm(snes, 1, pc, 2); 505731823bd8SMatthew G Knepley ierr = PetscObjectReference((PetscObject) pc);CHKERRQ(ierr); 5058efd4aadfSBarry Smith ierr = SNESDestroy(&snes->npc);CHKERRQ(ierr); 5059efd4aadfSBarry Smith snes->npc = pc; 5060efd4aadfSBarry Smith ierr = PetscLogObjectParent((PetscObject)snes, (PetscObject)snes->npc);CHKERRQ(ierr); 506131823bd8SMatthew G Knepley PetscFunctionReturn(0); 506231823bd8SMatthew G Knepley } 506331823bd8SMatthew G Knepley 506431823bd8SMatthew G Knepley /*@ 5065be95d8f1SBarry Smith SNESGetNPC - Creates a nonlinear preconditioning solver (SNES) to be used to precondition the nonlinear solver. 506631823bd8SMatthew G Knepley 506731823bd8SMatthew G Knepley Not Collective 506831823bd8SMatthew G Knepley 506931823bd8SMatthew G Knepley Input Parameter: 507031823bd8SMatthew G Knepley . snes - iterative context obtained from SNESCreate() 507131823bd8SMatthew G Knepley 507231823bd8SMatthew G Knepley Output Parameter: 507331823bd8SMatthew G Knepley . pc - preconditioner context 507431823bd8SMatthew G Knepley 5075be95d8f1SBarry Smith Notes: If a SNES was previously set with SNESSetNPC() then that SNES is returned. 5076be95d8f1SBarry Smith 507731823bd8SMatthew G Knepley Level: developer 507831823bd8SMatthew G Knepley 507931823bd8SMatthew G Knepley .keywords: SNES, get, preconditioner 50803ad1a0b9SPatrick Farrell .seealso: SNESSetNPC(), SNESHasNPC() 508131823bd8SMatthew G Knepley @*/ 5082be95d8f1SBarry Smith PetscErrorCode SNESGetNPC(SNES snes, SNES *pc) 508331823bd8SMatthew G Knepley { 508431823bd8SMatthew G Knepley PetscErrorCode ierr; 5085a64e098fSPeter Brune const char *optionsprefix; 508631823bd8SMatthew G Knepley 508731823bd8SMatthew G Knepley PetscFunctionBegin; 508831823bd8SMatthew G Knepley PetscValidHeaderSpecific(snes, SNES_CLASSID, 1); 508931823bd8SMatthew G Knepley PetscValidPointer(pc, 2); 5090efd4aadfSBarry Smith if (!snes->npc) { 5091efd4aadfSBarry Smith ierr = SNESCreate(PetscObjectComm((PetscObject)snes),&snes->npc);CHKERRQ(ierr); 5092efd4aadfSBarry Smith ierr = PetscObjectIncrementTabLevel((PetscObject)snes->npc,(PetscObject)snes,1);CHKERRQ(ierr); 5093efd4aadfSBarry Smith ierr = PetscLogObjectParent((PetscObject)snes,(PetscObject)snes->npc);CHKERRQ(ierr); 5094a64e098fSPeter Brune ierr = SNESGetOptionsPrefix(snes,&optionsprefix);CHKERRQ(ierr); 5095efd4aadfSBarry Smith ierr = SNESSetOptionsPrefix(snes->npc,optionsprefix);CHKERRQ(ierr); 5096efd4aadfSBarry Smith ierr = SNESAppendOptionsPrefix(snes->npc,"npc_");CHKERRQ(ierr); 5097efd4aadfSBarry Smith ierr = SNESSetCountersReset(snes->npc,PETSC_FALSE);CHKERRQ(ierr); 509831823bd8SMatthew G Knepley } 5099efd4aadfSBarry Smith *pc = snes->npc; 510031823bd8SMatthew G Knepley PetscFunctionReturn(0); 510131823bd8SMatthew G Knepley } 510231823bd8SMatthew G Knepley 51033ad1a0b9SPatrick Farrell /*@ 51043ad1a0b9SPatrick Farrell SNESHasNPC - Returns whether a nonlinear preconditioner exists 51053ad1a0b9SPatrick Farrell 51063ad1a0b9SPatrick Farrell Not Collective 51073ad1a0b9SPatrick Farrell 51083ad1a0b9SPatrick Farrell Input Parameter: 51093ad1a0b9SPatrick Farrell . snes - iterative context obtained from SNESCreate() 51103ad1a0b9SPatrick Farrell 51113ad1a0b9SPatrick Farrell Output Parameter: 51123ad1a0b9SPatrick Farrell . has_npc - whether the SNES has an NPC or not 51133ad1a0b9SPatrick Farrell 51143ad1a0b9SPatrick Farrell Level: developer 51153ad1a0b9SPatrick Farrell 51163ad1a0b9SPatrick Farrell .keywords: SNES, has, preconditioner 51173ad1a0b9SPatrick Farrell .seealso: SNESSetNPC(), SNESGetNPC() 51183ad1a0b9SPatrick Farrell @*/ 51193ad1a0b9SPatrick Farrell PetscErrorCode SNESHasNPC(SNES snes, PetscBool *has_npc) 51203ad1a0b9SPatrick Farrell { 51213ad1a0b9SPatrick Farrell PetscFunctionBegin; 51223ad1a0b9SPatrick Farrell PetscValidHeaderSpecific(snes, SNES_CLASSID, 1); 5123efd4aadfSBarry Smith *has_npc = (PetscBool) (snes->npc ? PETSC_TRUE : PETSC_FALSE); 51243ad1a0b9SPatrick Farrell PetscFunctionReturn(0); 51253ad1a0b9SPatrick Farrell } 51263ad1a0b9SPatrick Farrell 5127c40d0f55SPeter Brune /*@ 5128be95d8f1SBarry Smith SNESSetNPCSide - Sets the preconditioning side. 5129c40d0f55SPeter Brune 5130c40d0f55SPeter Brune Logically Collective on SNES 5131c40d0f55SPeter Brune 5132c40d0f55SPeter Brune Input Parameter: 5133c40d0f55SPeter Brune . snes - iterative context obtained from SNESCreate() 5134c40d0f55SPeter Brune 5135c40d0f55SPeter Brune Output Parameter: 5136c40d0f55SPeter Brune . side - the preconditioning side, where side is one of 5137c40d0f55SPeter Brune .vb 51382d547940SBarry Smith PC_LEFT - left preconditioning 51392d547940SBarry Smith PC_RIGHT - right preconditioning (default for most nonlinear solvers) 5140c40d0f55SPeter Brune .ve 5141c40d0f55SPeter Brune 5142c40d0f55SPeter Brune Options Database Keys: 5143c40d0f55SPeter Brune . -snes_pc_side <right,left> 5144c40d0f55SPeter Brune 51452d547940SBarry Smith Notes: SNESNRICHARDSON and SNESNCG only support left preconditioning. 51462d547940SBarry Smith 5147c40d0f55SPeter Brune Level: intermediate 5148c40d0f55SPeter Brune 5149c40d0f55SPeter Brune .keywords: SNES, set, right, left, side, preconditioner, flag 5150c40d0f55SPeter Brune 5151be95d8f1SBarry Smith .seealso: SNESGetNPCSide(), KSPSetPCSide() 5152c40d0f55SPeter Brune @*/ 5153be95d8f1SBarry Smith PetscErrorCode SNESSetNPCSide(SNES snes,PCSide side) 5154c40d0f55SPeter Brune { 5155c40d0f55SPeter Brune PetscFunctionBegin; 5156c40d0f55SPeter Brune PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 5157c40d0f55SPeter Brune PetscValidLogicalCollectiveEnum(snes,side,2); 5158efd4aadfSBarry Smith snes->npcside= side; 5159c40d0f55SPeter Brune PetscFunctionReturn(0); 5160c40d0f55SPeter Brune } 5161c40d0f55SPeter Brune 5162c40d0f55SPeter Brune /*@ 5163be95d8f1SBarry Smith SNESGetNPCSide - Gets the preconditioning side. 5164c40d0f55SPeter Brune 5165c40d0f55SPeter Brune Not Collective 5166c40d0f55SPeter Brune 5167c40d0f55SPeter Brune Input Parameter: 5168c40d0f55SPeter Brune . snes - iterative context obtained from SNESCreate() 5169c40d0f55SPeter Brune 5170c40d0f55SPeter Brune Output Parameter: 5171c40d0f55SPeter Brune . side - the preconditioning side, where side is one of 5172c40d0f55SPeter Brune .vb 51732d547940SBarry Smith PC_LEFT - left preconditioning 51742d547940SBarry Smith PC_RIGHT - right preconditioning (default for most nonlinear solvers) 5175c40d0f55SPeter Brune .ve 5176c40d0f55SPeter Brune 5177c40d0f55SPeter Brune Level: intermediate 5178c40d0f55SPeter Brune 5179c40d0f55SPeter Brune .keywords: SNES, get, right, left, side, preconditioner, flag 5180c40d0f55SPeter Brune 5181be95d8f1SBarry Smith .seealso: SNESSetNPCSide(), KSPGetPCSide() 5182c40d0f55SPeter Brune @*/ 5183be95d8f1SBarry Smith PetscErrorCode SNESGetNPCSide(SNES snes,PCSide *side) 5184c40d0f55SPeter Brune { 5185c40d0f55SPeter Brune PetscFunctionBegin; 5186c40d0f55SPeter Brune PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 5187c40d0f55SPeter Brune PetscValidPointer(side,2); 5188efd4aadfSBarry Smith *side = snes->npcside; 5189c40d0f55SPeter Brune PetscFunctionReturn(0); 5190c40d0f55SPeter Brune } 5191c40d0f55SPeter Brune 51929e764e56SPeter Brune /*@ 51937601faf0SJed Brown SNESSetLineSearch - Sets the linesearch on the SNES instance. 51949e764e56SPeter Brune 51959e764e56SPeter Brune Collective on SNES 51969e764e56SPeter Brune 51979e764e56SPeter Brune Input Parameters: 51989e764e56SPeter Brune + snes - iterative context obtained from SNESCreate() 51999e764e56SPeter Brune - linesearch - the linesearch object 52009e764e56SPeter Brune 52019e764e56SPeter Brune Notes: 52027601faf0SJed Brown Use SNESGetLineSearch() to retrieve the preconditioner context (for example, 52039e764e56SPeter Brune to configure it using the API). 52049e764e56SPeter Brune 52059e764e56SPeter Brune Level: developer 52069e764e56SPeter Brune 52079e764e56SPeter Brune .keywords: SNES, set, linesearch 52087601faf0SJed Brown .seealso: SNESGetLineSearch() 52099e764e56SPeter Brune @*/ 52107601faf0SJed Brown PetscErrorCode SNESSetLineSearch(SNES snes, SNESLineSearch linesearch) 52119e764e56SPeter Brune { 52129e764e56SPeter Brune PetscErrorCode ierr; 52139e764e56SPeter Brune 52149e764e56SPeter Brune PetscFunctionBegin; 52159e764e56SPeter Brune PetscValidHeaderSpecific(snes, SNES_CLASSID, 1); 5216f1c6b773SPeter Brune PetscValidHeaderSpecific(linesearch, SNESLINESEARCH_CLASSID, 2); 52179e764e56SPeter Brune PetscCheckSameComm(snes, 1, linesearch, 2); 52189e764e56SPeter Brune ierr = PetscObjectReference((PetscObject) linesearch);CHKERRQ(ierr); 5219f1c6b773SPeter Brune ierr = SNESLineSearchDestroy(&snes->linesearch);CHKERRQ(ierr); 5220f5af7f23SKarl Rupp 52219e764e56SPeter Brune snes->linesearch = linesearch; 5222f5af7f23SKarl Rupp 52233bb1ff40SBarry Smith ierr = PetscLogObjectParent((PetscObject)snes, (PetscObject)snes->linesearch);CHKERRQ(ierr); 52249e764e56SPeter Brune PetscFunctionReturn(0); 52259e764e56SPeter Brune } 52269e764e56SPeter Brune 5227a34ceb2aSJed Brown /*@ 52287601faf0SJed Brown SNESGetLineSearch - Returns a pointer to the line search context set with SNESSetLineSearch() 52298141a3b9SPeter Brune or creates a default line search instance associated with the SNES and returns it. 52309e764e56SPeter Brune 52319e764e56SPeter Brune Not Collective 52329e764e56SPeter Brune 52339e764e56SPeter Brune Input Parameter: 52349e764e56SPeter Brune . snes - iterative context obtained from SNESCreate() 52359e764e56SPeter Brune 52369e764e56SPeter Brune Output Parameter: 52379e764e56SPeter Brune . linesearch - linesearch context 52389e764e56SPeter Brune 5239162e0bf5SPeter Brune Level: beginner 52409e764e56SPeter Brune 52419e764e56SPeter Brune .keywords: SNES, get, linesearch 5242162e0bf5SPeter Brune .seealso: SNESSetLineSearch(), SNESLineSearchCreate() 52439e764e56SPeter Brune @*/ 52447601faf0SJed Brown PetscErrorCode SNESGetLineSearch(SNES snes, SNESLineSearch *linesearch) 52459e764e56SPeter Brune { 52469e764e56SPeter Brune PetscErrorCode ierr; 52479e764e56SPeter Brune const char *optionsprefix; 52489e764e56SPeter Brune 52499e764e56SPeter Brune PetscFunctionBegin; 52509e764e56SPeter Brune PetscValidHeaderSpecific(snes, SNES_CLASSID, 1); 52519e764e56SPeter Brune PetscValidPointer(linesearch, 2); 52529e764e56SPeter Brune if (!snes->linesearch) { 52539e764e56SPeter Brune ierr = SNESGetOptionsPrefix(snes, &optionsprefix);CHKERRQ(ierr); 525482f516ccSBarry Smith ierr = SNESLineSearchCreate(PetscObjectComm((PetscObject)snes), &snes->linesearch);CHKERRQ(ierr); 5255f1c6b773SPeter Brune ierr = SNESLineSearchSetSNES(snes->linesearch, snes);CHKERRQ(ierr); 5256b1dca40bSPeter Brune ierr = SNESLineSearchAppendOptionsPrefix(snes->linesearch, optionsprefix);CHKERRQ(ierr); 52579e764e56SPeter Brune ierr = PetscObjectIncrementTabLevel((PetscObject) snes->linesearch, (PetscObject) snes, 1);CHKERRQ(ierr); 52583bb1ff40SBarry Smith ierr = PetscLogObjectParent((PetscObject)snes, (PetscObject)snes->linesearch);CHKERRQ(ierr); 52599e764e56SPeter Brune } 52609e764e56SPeter Brune *linesearch = snes->linesearch; 52619e764e56SPeter Brune PetscFunctionReturn(0); 52629e764e56SPeter Brune } 52639e764e56SPeter Brune 526469b4f73cSBarry Smith #if defined(PETSC_HAVE_MATLAB_ENGINE) 5265c6db04a5SJed Brown #include <mex.h> 526669b4f73cSBarry Smith 52678f6e6473SBarry Smith typedef struct {char *funcname; mxArray *ctx;} SNESMatlabContext; 52688f6e6473SBarry Smith 52690807856dSBarry Smith /* 5270bf388a1fSBarry Smith SNESComputeFunction_Matlab - Calls the function that has been set with SNESSetFunctionMatlab(). 52710807856dSBarry Smith 52720807856dSBarry Smith Collective on SNES 52730807856dSBarry Smith 52740807856dSBarry Smith Input Parameters: 52750807856dSBarry Smith + snes - the SNES context 52760807856dSBarry Smith - x - input vector 52770807856dSBarry Smith 52780807856dSBarry Smith Output Parameter: 52790807856dSBarry Smith . y - function vector, as set by SNESSetFunction() 52800807856dSBarry Smith 52810807856dSBarry Smith Notes: 52820807856dSBarry Smith SNESComputeFunction() is typically used within nonlinear solvers 52830807856dSBarry Smith implementations, so most users would not generally call this routine 52840807856dSBarry Smith themselves. 52850807856dSBarry Smith 52860807856dSBarry Smith Level: developer 52870807856dSBarry Smith 52880807856dSBarry Smith .keywords: SNES, nonlinear, compute, function 52890807856dSBarry Smith 52900807856dSBarry Smith .seealso: SNESSetFunction(), SNESGetFunction() 529161b2408cSBarry Smith */ 52927087cfbeSBarry Smith PetscErrorCode SNESComputeFunction_Matlab(SNES snes,Vec x,Vec y, void *ctx) 52930807856dSBarry Smith { 5294e650e774SBarry Smith PetscErrorCode ierr; 52958f6e6473SBarry Smith SNESMatlabContext *sctx = (SNESMatlabContext*)ctx; 52968f6e6473SBarry Smith int nlhs = 1,nrhs = 5; 52978f6e6473SBarry Smith mxArray *plhs[1],*prhs[5]; 529891621f2eSBarry Smith long long int lx = 0,ly = 0,ls = 0; 5299e650e774SBarry Smith 53000807856dSBarry Smith PetscFunctionBegin; 53010807856dSBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 53020807856dSBarry Smith PetscValidHeaderSpecific(x,VEC_CLASSID,2); 53030807856dSBarry Smith PetscValidHeaderSpecific(y,VEC_CLASSID,3); 53040807856dSBarry Smith PetscCheckSameComm(snes,1,x,2); 53050807856dSBarry Smith PetscCheckSameComm(snes,1,y,3); 53060807856dSBarry Smith 53070807856dSBarry Smith /* call Matlab function in ctx with arguments x and y */ 5308e650e774SBarry Smith 530991621f2eSBarry Smith ierr = PetscMemcpy(&ls,&snes,sizeof(snes));CHKERRQ(ierr); 5310e650e774SBarry Smith ierr = PetscMemcpy(&lx,&x,sizeof(x));CHKERRQ(ierr); 5311e650e774SBarry Smith ierr = PetscMemcpy(&ly,&y,sizeof(x));CHKERRQ(ierr); 531291621f2eSBarry Smith prhs[0] = mxCreateDoubleScalar((double)ls); 531391621f2eSBarry Smith prhs[1] = mxCreateDoubleScalar((double)lx); 531491621f2eSBarry Smith prhs[2] = mxCreateDoubleScalar((double)ly); 53158f6e6473SBarry Smith prhs[3] = mxCreateString(sctx->funcname); 53168f6e6473SBarry Smith prhs[4] = sctx->ctx; 5317b807a863SBarry Smith ierr = mexCallMATLAB(nlhs,plhs,nrhs,prhs,"PetscSNESComputeFunctionInternal");CHKERRQ(ierr); 5318e650e774SBarry Smith ierr = mxGetScalar(plhs[0]);CHKERRQ(ierr); 5319e650e774SBarry Smith mxDestroyArray(prhs[0]); 5320e650e774SBarry Smith mxDestroyArray(prhs[1]); 5321e650e774SBarry Smith mxDestroyArray(prhs[2]); 53228f6e6473SBarry Smith mxDestroyArray(prhs[3]); 5323e650e774SBarry Smith mxDestroyArray(plhs[0]); 53240807856dSBarry Smith PetscFunctionReturn(0); 53250807856dSBarry Smith } 53260807856dSBarry Smith 532761b2408cSBarry Smith /* 53280807856dSBarry Smith SNESSetFunctionMatlab - Sets the function evaluation routine and function 53290807856dSBarry Smith vector for use by the SNES routines in solving systems of nonlinear 5330e3c5b3baSBarry Smith equations from MATLAB. Here the function is a string containing the name of a MATLAB function 53310807856dSBarry Smith 53320807856dSBarry Smith Logically Collective on SNES 53330807856dSBarry Smith 53340807856dSBarry Smith Input Parameters: 53350807856dSBarry Smith + snes - the SNES context 53360807856dSBarry Smith . r - vector to store function value 5337f8b49ee9SBarry Smith - f - function evaluation routine 53380807856dSBarry Smith 53390807856dSBarry Smith Notes: 53400807856dSBarry Smith The Newton-like methods typically solve linear systems of the form 53410807856dSBarry Smith $ f'(x) x = -f(x), 53420807856dSBarry Smith where f'(x) denotes the Jacobian matrix and f(x) is the function. 53430807856dSBarry Smith 53440807856dSBarry Smith Level: beginner 53450807856dSBarry Smith 5346c5b75c40SBarry Smith Developer Note: This bleeds the allocated memory SNESMatlabContext *sctx; 5347c5b75c40SBarry Smith 53480807856dSBarry Smith .keywords: SNES, nonlinear, set, function 53490807856dSBarry Smith 53500807856dSBarry Smith .seealso: SNESGetFunction(), SNESComputeFunction(), SNESSetJacobian(), SNESSetFunction() 535161b2408cSBarry Smith */ 5352f8b49ee9SBarry Smith PetscErrorCode SNESSetFunctionMatlab(SNES snes,Vec r,const char *f,mxArray *ctx) 53530807856dSBarry Smith { 53540807856dSBarry Smith PetscErrorCode ierr; 53558f6e6473SBarry Smith SNESMatlabContext *sctx; 53560807856dSBarry Smith 53570807856dSBarry Smith PetscFunctionBegin; 53588f6e6473SBarry Smith /* currently sctx is memory bleed */ 5359854ce69bSBarry Smith ierr = PetscNew(&sctx);CHKERRQ(ierr); 5360f8b49ee9SBarry Smith ierr = PetscStrallocpy(f,&sctx->funcname);CHKERRQ(ierr); 53618f6e6473SBarry Smith /* 53628f6e6473SBarry Smith This should work, but it doesn't 53638f6e6473SBarry Smith sctx->ctx = ctx; 53648f6e6473SBarry Smith mexMakeArrayPersistent(sctx->ctx); 53658f6e6473SBarry Smith */ 53668f6e6473SBarry Smith sctx->ctx = mxDuplicateArray(ctx); 53678f6e6473SBarry Smith ierr = SNESSetFunction(snes,r,SNESComputeFunction_Matlab,sctx);CHKERRQ(ierr); 53680807856dSBarry Smith PetscFunctionReturn(0); 53690807856dSBarry Smith } 537069b4f73cSBarry Smith 537161b2408cSBarry Smith /* 5372bf388a1fSBarry Smith SNESComputeJacobian_Matlab - Calls the function that has been set with SNESSetJacobianMatlab(). 537361b2408cSBarry Smith 537461b2408cSBarry Smith Collective on SNES 537561b2408cSBarry Smith 537661b2408cSBarry Smith Input Parameters: 537761b2408cSBarry Smith + snes - the SNES context 537861b2408cSBarry Smith . x - input vector 537961b2408cSBarry Smith . A, B - the matrices 538061b2408cSBarry Smith - ctx - user context 538161b2408cSBarry Smith 538261b2408cSBarry Smith Level: developer 538361b2408cSBarry Smith 538461b2408cSBarry Smith .keywords: SNES, nonlinear, compute, function 538561b2408cSBarry Smith 538661b2408cSBarry Smith .seealso: SNESSetFunction(), SNESGetFunction() 538761b2408cSBarry Smith @*/ 5388f3229a78SSatish Balay PetscErrorCode SNESComputeJacobian_Matlab(SNES snes,Vec x,Mat A,Mat B,void *ctx) 538961b2408cSBarry Smith { 539061b2408cSBarry Smith PetscErrorCode ierr; 539161b2408cSBarry Smith SNESMatlabContext *sctx = (SNESMatlabContext*)ctx; 539261b2408cSBarry Smith int nlhs = 2,nrhs = 6; 539361b2408cSBarry Smith mxArray *plhs[2],*prhs[6]; 539461b2408cSBarry Smith long long int lx = 0,lA = 0,ls = 0, lB = 0; 539561b2408cSBarry Smith 539661b2408cSBarry Smith PetscFunctionBegin; 539761b2408cSBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 539861b2408cSBarry Smith PetscValidHeaderSpecific(x,VEC_CLASSID,2); 539961b2408cSBarry Smith 540061b2408cSBarry Smith /* call Matlab function in ctx with arguments x and y */ 540161b2408cSBarry Smith 540261b2408cSBarry Smith ierr = PetscMemcpy(&ls,&snes,sizeof(snes));CHKERRQ(ierr); 540361b2408cSBarry Smith ierr = PetscMemcpy(&lx,&x,sizeof(x));CHKERRQ(ierr); 540461b2408cSBarry Smith ierr = PetscMemcpy(&lA,A,sizeof(x));CHKERRQ(ierr); 540561b2408cSBarry Smith ierr = PetscMemcpy(&lB,B,sizeof(x));CHKERRQ(ierr); 540661b2408cSBarry Smith prhs[0] = mxCreateDoubleScalar((double)ls); 540761b2408cSBarry Smith prhs[1] = mxCreateDoubleScalar((double)lx); 540861b2408cSBarry Smith prhs[2] = mxCreateDoubleScalar((double)lA); 540961b2408cSBarry Smith prhs[3] = mxCreateDoubleScalar((double)lB); 541061b2408cSBarry Smith prhs[4] = mxCreateString(sctx->funcname); 541161b2408cSBarry Smith prhs[5] = sctx->ctx; 5412b807a863SBarry Smith ierr = mexCallMATLAB(nlhs,plhs,nrhs,prhs,"PetscSNESComputeJacobianInternal");CHKERRQ(ierr); 541361b2408cSBarry Smith ierr = mxGetScalar(plhs[0]);CHKERRQ(ierr); 541461b2408cSBarry Smith mxDestroyArray(prhs[0]); 541561b2408cSBarry Smith mxDestroyArray(prhs[1]); 541661b2408cSBarry Smith mxDestroyArray(prhs[2]); 541761b2408cSBarry Smith mxDestroyArray(prhs[3]); 541861b2408cSBarry Smith mxDestroyArray(prhs[4]); 541961b2408cSBarry Smith mxDestroyArray(plhs[0]); 542061b2408cSBarry Smith mxDestroyArray(plhs[1]); 542161b2408cSBarry Smith PetscFunctionReturn(0); 542261b2408cSBarry Smith } 542361b2408cSBarry Smith 542461b2408cSBarry Smith /* 542561b2408cSBarry Smith SNESSetJacobianMatlab - Sets the Jacobian function evaluation routine and two empty Jacobian matrices 542661b2408cSBarry Smith vector for use by the SNES routines in solving systems of nonlinear 5427e3c5b3baSBarry Smith equations from MATLAB. Here the function is a string containing the name of a MATLAB function 542861b2408cSBarry Smith 542961b2408cSBarry Smith Logically Collective on SNES 543061b2408cSBarry Smith 543161b2408cSBarry Smith Input Parameters: 543261b2408cSBarry Smith + snes - the SNES context 543361b2408cSBarry Smith . A,B - Jacobian matrices 5434f8b49ee9SBarry Smith . J - function evaluation routine 543561b2408cSBarry Smith - ctx - user context 543661b2408cSBarry Smith 543761b2408cSBarry Smith Level: developer 543861b2408cSBarry Smith 5439c5b75c40SBarry Smith Developer Note: This bleeds the allocated memory SNESMatlabContext *sctx; 5440c5b75c40SBarry Smith 544161b2408cSBarry Smith .keywords: SNES, nonlinear, set, function 544261b2408cSBarry Smith 5443f8b49ee9SBarry Smith .seealso: SNESGetFunction(), SNESComputeFunction(), SNESSetJacobian(), SNESSetFunction(), J 544461b2408cSBarry Smith */ 5445f8b49ee9SBarry Smith PetscErrorCode SNESSetJacobianMatlab(SNES snes,Mat A,Mat B,const char *J,mxArray *ctx) 544661b2408cSBarry Smith { 544761b2408cSBarry Smith PetscErrorCode ierr; 544861b2408cSBarry Smith SNESMatlabContext *sctx; 544961b2408cSBarry Smith 545061b2408cSBarry Smith PetscFunctionBegin; 545161b2408cSBarry Smith /* currently sctx is memory bleed */ 5452854ce69bSBarry Smith ierr = PetscNew(&sctx);CHKERRQ(ierr); 5453f8b49ee9SBarry Smith ierr = PetscStrallocpy(J,&sctx->funcname);CHKERRQ(ierr); 545461b2408cSBarry Smith /* 545561b2408cSBarry Smith This should work, but it doesn't 545661b2408cSBarry Smith sctx->ctx = ctx; 545761b2408cSBarry Smith mexMakeArrayPersistent(sctx->ctx); 545861b2408cSBarry Smith */ 545961b2408cSBarry Smith sctx->ctx = mxDuplicateArray(ctx); 546061b2408cSBarry Smith ierr = SNESSetJacobian(snes,A,B,SNESComputeJacobian_Matlab,sctx);CHKERRQ(ierr); 546161b2408cSBarry Smith PetscFunctionReturn(0); 546261b2408cSBarry Smith } 546369b4f73cSBarry Smith 5464f9eb7ae2SShri Abhyankar /* 5465f9eb7ae2SShri Abhyankar SNESMonitor_Matlab - Calls the function that has been set with SNESMonitorSetMatlab(). 5466f9eb7ae2SShri Abhyankar 5467f9eb7ae2SShri Abhyankar Collective on SNES 5468f9eb7ae2SShri Abhyankar 5469f9eb7ae2SShri Abhyankar .seealso: SNESSetFunction(), SNESGetFunction() 5470f9eb7ae2SShri Abhyankar @*/ 54717087cfbeSBarry Smith PetscErrorCode SNESMonitor_Matlab(SNES snes,PetscInt it, PetscReal fnorm, void *ctx) 5472f9eb7ae2SShri Abhyankar { 5473f9eb7ae2SShri Abhyankar PetscErrorCode ierr; 547448f37e70SShri Abhyankar SNESMatlabContext *sctx = (SNESMatlabContext*)ctx; 5475f9eb7ae2SShri Abhyankar int nlhs = 1,nrhs = 6; 5476f9eb7ae2SShri Abhyankar mxArray *plhs[1],*prhs[6]; 5477f9eb7ae2SShri Abhyankar long long int lx = 0,ls = 0; 5478f9eb7ae2SShri Abhyankar Vec x = snes->vec_sol; 5479f9eb7ae2SShri Abhyankar 5480f9eb7ae2SShri Abhyankar PetscFunctionBegin; 5481f9eb7ae2SShri Abhyankar PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 5482f9eb7ae2SShri Abhyankar 5483f9eb7ae2SShri Abhyankar ierr = PetscMemcpy(&ls,&snes,sizeof(snes));CHKERRQ(ierr); 5484f9eb7ae2SShri Abhyankar ierr = PetscMemcpy(&lx,&x,sizeof(x));CHKERRQ(ierr); 5485f9eb7ae2SShri Abhyankar prhs[0] = mxCreateDoubleScalar((double)ls); 5486f9eb7ae2SShri Abhyankar prhs[1] = mxCreateDoubleScalar((double)it); 5487f9eb7ae2SShri Abhyankar prhs[2] = mxCreateDoubleScalar((double)fnorm); 5488f9eb7ae2SShri Abhyankar prhs[3] = mxCreateDoubleScalar((double)lx); 5489f9eb7ae2SShri Abhyankar prhs[4] = mxCreateString(sctx->funcname); 5490f9eb7ae2SShri Abhyankar prhs[5] = sctx->ctx; 5491f9eb7ae2SShri Abhyankar ierr = mexCallMATLAB(nlhs,plhs,nrhs,prhs,"PetscSNESMonitorInternal");CHKERRQ(ierr); 5492f9eb7ae2SShri Abhyankar ierr = mxGetScalar(plhs[0]);CHKERRQ(ierr); 5493f9eb7ae2SShri Abhyankar mxDestroyArray(prhs[0]); 5494f9eb7ae2SShri Abhyankar mxDestroyArray(prhs[1]); 5495f9eb7ae2SShri Abhyankar mxDestroyArray(prhs[2]); 5496f9eb7ae2SShri Abhyankar mxDestroyArray(prhs[3]); 5497f9eb7ae2SShri Abhyankar mxDestroyArray(prhs[4]); 5498f9eb7ae2SShri Abhyankar mxDestroyArray(plhs[0]); 5499f9eb7ae2SShri Abhyankar PetscFunctionReturn(0); 5500f9eb7ae2SShri Abhyankar } 5501f9eb7ae2SShri Abhyankar 5502f9eb7ae2SShri Abhyankar /* 5503e3c5b3baSBarry Smith SNESMonitorSetMatlab - Sets the monitor function from MATLAB 5504f9eb7ae2SShri Abhyankar 5505f9eb7ae2SShri Abhyankar Level: developer 5506f9eb7ae2SShri Abhyankar 5507c5b75c40SBarry Smith Developer Note: This bleeds the allocated memory SNESMatlabContext *sctx; 5508c5b75c40SBarry Smith 5509f9eb7ae2SShri Abhyankar .keywords: SNES, nonlinear, set, function 5510f9eb7ae2SShri Abhyankar 5511f9eb7ae2SShri Abhyankar .seealso: SNESGetFunction(), SNESComputeFunction(), SNESSetJacobian(), SNESSetFunction() 5512f9eb7ae2SShri Abhyankar */ 55136e4dcb14SBarry Smith PetscErrorCode SNESMonitorSetMatlab(SNES snes,const char *f,mxArray *ctx) 5514f9eb7ae2SShri Abhyankar { 5515f9eb7ae2SShri Abhyankar PetscErrorCode ierr; 5516f9eb7ae2SShri Abhyankar SNESMatlabContext *sctx; 5517f9eb7ae2SShri Abhyankar 5518f9eb7ae2SShri Abhyankar PetscFunctionBegin; 5519f9eb7ae2SShri Abhyankar /* currently sctx is memory bleed */ 5520854ce69bSBarry Smith ierr = PetscNew(&sctx);CHKERRQ(ierr); 55216e4dcb14SBarry Smith ierr = PetscStrallocpy(f,&sctx->funcname);CHKERRQ(ierr); 5522f9eb7ae2SShri Abhyankar /* 5523f9eb7ae2SShri Abhyankar This should work, but it doesn't 5524f9eb7ae2SShri Abhyankar sctx->ctx = ctx; 5525f9eb7ae2SShri Abhyankar mexMakeArrayPersistent(sctx->ctx); 5526f9eb7ae2SShri Abhyankar */ 5527f9eb7ae2SShri Abhyankar sctx->ctx = mxDuplicateArray(ctx); 55280298fd71SBarry Smith ierr = SNESMonitorSet(snes,SNESMonitor_Matlab,sctx,NULL);CHKERRQ(ierr); 5529f9eb7ae2SShri Abhyankar PetscFunctionReturn(0); 5530f9eb7ae2SShri Abhyankar } 5531f9eb7ae2SShri Abhyankar 553269b4f73cSBarry Smith #endif 5533