1af0996ceSBarry Smith #include <petsc/private/snesimpl.h> /*I "petscsnes.h" I*/ 207475bc1SBarry Smith #include <petscdmshell.h> 3d96771aaSLisandro Dalcin #include <petscdraw.h> 4a01aa210SMatthew G. Knepley #include <petscds.h> 534b4d3a8SMatthew G. Knepley #include <petscdmadaptor.h> 606fc46c8SMatthew G. Knepley #include <petscconvest.h> 79b94acceSBarry Smith 8ace3abfcSBarry Smith PetscBool SNESRegisterAllCalled = PETSC_FALSE; 90298fd71SBarry Smith PetscFunctionList SNESList = NULL; 108ba1e511SMatthew Knepley 118ba1e511SMatthew Knepley /* Logging support */ 1222c6f798SBarry Smith PetscClassId SNES_CLASSID, DMSNES_CLASSID; 13e3ed9ee7SBarry Smith PetscLogEvent SNES_Solve, SNES_Setup, SNES_FunctionEval, SNES_JacobianEval, SNES_NGSEval, SNES_NGSFuncEval, SNES_NPCSolve, SNES_ObjectiveEval; 14a09944afSBarry Smith 15e113a28aSBarry Smith /*@ 16e113a28aSBarry Smith SNESSetErrorIfNotConverged - Causes SNESSolve() to generate an error if the solver has not converged. 17e113a28aSBarry Smith 183f9fe445SBarry Smith Logically Collective on SNES 19e113a28aSBarry Smith 20e113a28aSBarry Smith Input Parameters: 21e113a28aSBarry Smith + snes - iterative context obtained from SNESCreate() 22e113a28aSBarry Smith - flg - PETSC_TRUE indicates you want the error generated 23e113a28aSBarry Smith 24e113a28aSBarry Smith Options database keys: 25e113a28aSBarry Smith . -snes_error_if_not_converged : this takes an optional truth value (0/1/no/yes/true/false) 26e113a28aSBarry Smith 27e113a28aSBarry Smith Level: intermediate 28e113a28aSBarry Smith 29e113a28aSBarry Smith Notes: 30e113a28aSBarry Smith Normally PETSc continues if a linear solver fails to converge, you can call SNESGetConvergedReason() after a SNESSolve() 31e113a28aSBarry Smith to determine if it has converged. 32e113a28aSBarry Smith 33e113a28aSBarry Smith .seealso: SNESGetErrorIfNotConverged(), KSPGetErrorIfNotConverged(), KSPSetErrorIFNotConverged() 34e113a28aSBarry Smith @*/ 357087cfbeSBarry Smith PetscErrorCode SNESSetErrorIfNotConverged(SNES snes,PetscBool flg) 36e113a28aSBarry Smith { 37e113a28aSBarry Smith PetscFunctionBegin; 38e113a28aSBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 39acfcf0e5SJed Brown PetscValidLogicalCollectiveBool(snes,flg,2); 40e113a28aSBarry Smith snes->errorifnotconverged = flg; 41e113a28aSBarry Smith PetscFunctionReturn(0); 42e113a28aSBarry Smith } 43e113a28aSBarry Smith 44e113a28aSBarry Smith /*@ 45e113a28aSBarry Smith SNESGetErrorIfNotConverged - Will SNESSolve() generate an error if the solver does not converge? 46e113a28aSBarry Smith 47e113a28aSBarry Smith Not Collective 48e113a28aSBarry Smith 49e113a28aSBarry Smith Input Parameter: 50e113a28aSBarry Smith . snes - iterative context obtained from SNESCreate() 51e113a28aSBarry Smith 52e113a28aSBarry Smith Output Parameter: 53e113a28aSBarry Smith . flag - PETSC_TRUE if it will generate an error, else PETSC_FALSE 54e113a28aSBarry Smith 55e113a28aSBarry Smith Level: intermediate 56e113a28aSBarry Smith 57e113a28aSBarry Smith .seealso: SNESSetErrorIfNotConverged(), KSPGetErrorIfNotConverged(), KSPSetErrorIFNotConverged() 58e113a28aSBarry Smith @*/ 597087cfbeSBarry Smith PetscErrorCode SNESGetErrorIfNotConverged(SNES snes,PetscBool *flag) 60e113a28aSBarry Smith { 61e113a28aSBarry Smith PetscFunctionBegin; 62e113a28aSBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 63534a8f05SLisandro Dalcin PetscValidBoolPointer(flag,2); 64e113a28aSBarry Smith *flag = snes->errorifnotconverged; 65e113a28aSBarry Smith PetscFunctionReturn(0); 66e113a28aSBarry Smith } 67e113a28aSBarry Smith 684fc747eaSLawrence Mitchell /*@ 694fc747eaSLawrence Mitchell SNESSetAlwaysComputesFinalResidual - does the SNES always compute the residual at the final solution? 704fc747eaSLawrence Mitchell 714fc747eaSLawrence Mitchell Logically Collective on SNES 724fc747eaSLawrence Mitchell 734fc747eaSLawrence Mitchell Input Parameters: 744fc747eaSLawrence Mitchell + snes - the shell SNES 754fc747eaSLawrence Mitchell - flg - is the residual computed? 764fc747eaSLawrence Mitchell 774fc747eaSLawrence Mitchell Level: advanced 784fc747eaSLawrence Mitchell 794fc747eaSLawrence Mitchell .seealso: SNESGetAlwaysComputesFinalResidual() 804fc747eaSLawrence Mitchell @*/ 814fc747eaSLawrence Mitchell PetscErrorCode SNESSetAlwaysComputesFinalResidual(SNES snes, PetscBool flg) 824fc747eaSLawrence Mitchell { 834fc747eaSLawrence Mitchell PetscFunctionBegin; 844fc747eaSLawrence Mitchell PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 854fc747eaSLawrence Mitchell snes->alwayscomputesfinalresidual = flg; 864fc747eaSLawrence Mitchell PetscFunctionReturn(0); 874fc747eaSLawrence Mitchell } 884fc747eaSLawrence Mitchell 894fc747eaSLawrence Mitchell /*@ 904fc747eaSLawrence Mitchell SNESGetAlwaysComputesFinalResidual - does the SNES always compute the residual at the final solution? 914fc747eaSLawrence Mitchell 924fc747eaSLawrence Mitchell Logically Collective on SNES 934fc747eaSLawrence Mitchell 944fc747eaSLawrence Mitchell Input Parameter: 954fc747eaSLawrence Mitchell . snes - the shell SNES 964fc747eaSLawrence Mitchell 974fc747eaSLawrence Mitchell Output Parameter: 984fc747eaSLawrence Mitchell . flg - is the residual computed? 994fc747eaSLawrence Mitchell 1004fc747eaSLawrence Mitchell Level: advanced 1014fc747eaSLawrence Mitchell 1024fc747eaSLawrence Mitchell .seealso: SNESSetAlwaysComputesFinalResidual() 1034fc747eaSLawrence Mitchell @*/ 1044fc747eaSLawrence Mitchell PetscErrorCode SNESGetAlwaysComputesFinalResidual(SNES snes, PetscBool *flg) 1054fc747eaSLawrence Mitchell { 1064fc747eaSLawrence Mitchell PetscFunctionBegin; 1074fc747eaSLawrence Mitchell PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 1084fc747eaSLawrence Mitchell *flg = snes->alwayscomputesfinalresidual; 1094fc747eaSLawrence Mitchell PetscFunctionReturn(0); 1104fc747eaSLawrence Mitchell } 1114fc747eaSLawrence Mitchell 112e725d27bSBarry Smith /*@ 113bf388a1fSBarry Smith SNESSetFunctionDomainError - tells SNES that the input vector to your SNESFunction is not 1144936397dSBarry Smith in the functions domain. For example, negative pressure. 1154936397dSBarry Smith 1163f9fe445SBarry Smith Logically Collective on SNES 1174936397dSBarry Smith 1184936397dSBarry Smith Input Parameters: 1196a388c36SPeter Brune . snes - the SNES context 1204936397dSBarry Smith 12128529972SSatish Balay Level: advanced 1224936397dSBarry Smith 123bf388a1fSBarry Smith .seealso: SNESCreate(), SNESSetFunction(), SNESFunction 1244936397dSBarry Smith @*/ 1257087cfbeSBarry Smith PetscErrorCode SNESSetFunctionDomainError(SNES snes) 1264936397dSBarry Smith { 1274936397dSBarry Smith PetscFunctionBegin; 1280700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 129422a814eSBarry Smith if (snes->errorifnotconverged) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"User code indicates input vector is not in the function domain"); 1304936397dSBarry Smith snes->domainerror = PETSC_TRUE; 1314936397dSBarry Smith PetscFunctionReturn(0); 1324936397dSBarry Smith } 1334936397dSBarry Smith 1346a388c36SPeter Brune /*@ 13507b62357SFande Kong SNESSetJacobianDomainError - tells SNES that computeJacobian does not make sense any more. For example there is a negative element transformation. 13607b62357SFande Kong 13707b62357SFande Kong Logically Collective on SNES 13807b62357SFande Kong 13907b62357SFande Kong Input Parameters: 14007b62357SFande Kong . snes - the SNES context 14107b62357SFande Kong 14207b62357SFande Kong Level: advanced 14307b62357SFande Kong 14407b62357SFande Kong .seealso: SNESCreate(), SNESSetFunction(), SNESFunction(), SNESSetFunctionDomainError() 14507b62357SFande Kong @*/ 14607b62357SFande Kong PetscErrorCode SNESSetJacobianDomainError(SNES snes) 14707b62357SFande Kong { 14807b62357SFande Kong PetscFunctionBegin; 14907b62357SFande Kong PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 15007b62357SFande Kong if (snes->errorifnotconverged) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"User code indicates computeJacobian does not make sense"); 15107b62357SFande Kong snes->jacobiandomainerror = PETSC_TRUE; 15207b62357SFande Kong PetscFunctionReturn(0); 15307b62357SFande Kong } 15407b62357SFande Kong 15507b62357SFande Kong /*@ 156b351a90bSFande Kong SNESSetCheckJacobianDomainError - if or not to check jacobian domain error after each Jacobian evaluation. By default, we check Jacobian domain error 157b351a90bSFande Kong in the debug mode, and do not check it in the optimized mode. 158b351a90bSFande Kong 159b351a90bSFande Kong Logically Collective on SNES 160b351a90bSFande Kong 161b351a90bSFande Kong Input Parameters: 162a2b725a8SWilliam Gropp + snes - the SNES context 163a2b725a8SWilliam Gropp - flg - indicates if or not to check jacobian domain error after each Jacobian evaluation 164b351a90bSFande Kong 165b351a90bSFande Kong Level: advanced 166b351a90bSFande Kong 1678383d7d7SFande Kong .seealso: SNESCreate(), SNESSetFunction(), SNESFunction(), SNESSetFunctionDomainError(), SNESGetCheckJacobianDomainError() 168b351a90bSFande Kong @*/ 169b351a90bSFande Kong PetscErrorCode SNESSetCheckJacobianDomainError(SNES snes, PetscBool flg) 170b351a90bSFande Kong { 171b351a90bSFande Kong PetscFunctionBegin; 172b351a90bSFande Kong PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 173b351a90bSFande Kong snes->checkjacdomainerror = flg; 174b351a90bSFande Kong PetscFunctionReturn(0); 175b351a90bSFande Kong } 176b351a90bSFande Kong 177b351a90bSFande Kong /*@ 1788383d7d7SFande Kong SNESGetCheckJacobianDomainError - Get an indicator whether or not we are checking Jacobian domain errors after each Jacobian evaluation. 1798383d7d7SFande Kong 1808383d7d7SFande Kong Logically Collective on SNES 1818383d7d7SFande Kong 1828383d7d7SFande Kong Input Parameters: 1838383d7d7SFande Kong . snes - the SNES context 1848383d7d7SFande Kong 1858383d7d7SFande Kong Output Parameters: 1868383d7d7SFande Kong . flg - PETSC_FALSE indicates that we don't check jacobian domain errors after each Jacobian evaluation 1878383d7d7SFande Kong 1888383d7d7SFande Kong Level: advanced 1898383d7d7SFande Kong 1908383d7d7SFande Kong .seealso: SNESCreate(), SNESSetFunction(), SNESFunction(), SNESSetFunctionDomainError(), SNESSetCheckJacobianDomainError() 1918383d7d7SFande Kong @*/ 1928383d7d7SFande Kong PetscErrorCode SNESGetCheckJacobianDomainError(SNES snes, PetscBool *flg) 1938383d7d7SFande Kong { 1948383d7d7SFande Kong PetscFunctionBegin; 1958383d7d7SFande Kong PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 196534a8f05SLisandro Dalcin PetscValidBoolPointer(flg,2); 1978383d7d7SFande Kong *flg = snes->checkjacdomainerror; 1988383d7d7SFande Kong PetscFunctionReturn(0); 1998383d7d7SFande Kong } 2008383d7d7SFande Kong 2018383d7d7SFande Kong /*@ 202c77b2880SPeter Brune SNESGetFunctionDomainError - Gets the status of the domain error after a call to SNESComputeFunction; 2036a388c36SPeter Brune 2046a388c36SPeter Brune Logically Collective on SNES 2056a388c36SPeter Brune 2066a388c36SPeter Brune Input Parameters: 2076a388c36SPeter Brune . snes - the SNES context 2086a388c36SPeter Brune 2096a388c36SPeter Brune Output Parameters: 210bf388a1fSBarry Smith . domainerror - Set to PETSC_TRUE if there's a domain error; PETSC_FALSE otherwise. 2116a388c36SPeter Brune 2126a388c36SPeter Brune Level: advanced 2136a388c36SPeter Brune 214bf388a1fSBarry Smith .seealso: SNESSetFunctionDomainError(), SNESComputeFunction() 2156a388c36SPeter Brune @*/ 2166a388c36SPeter Brune PetscErrorCode SNESGetFunctionDomainError(SNES snes, PetscBool *domainerror) 2176a388c36SPeter Brune { 2186a388c36SPeter Brune PetscFunctionBegin; 2196a388c36SPeter Brune PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 220534a8f05SLisandro Dalcin PetscValidBoolPointer(domainerror,2); 2216a388c36SPeter Brune *domainerror = snes->domainerror; 2226a388c36SPeter Brune PetscFunctionReturn(0); 2236a388c36SPeter Brune } 2246a388c36SPeter Brune 22507b62357SFande Kong /*@ 22607b62357SFande Kong SNESGetJacobianDomainError - Gets the status of the Jacobian domain error after a call to SNESComputeJacobian; 22707b62357SFande Kong 22807b62357SFande Kong Logically Collective on SNES 22907b62357SFande Kong 23007b62357SFande Kong Input Parameters: 23107b62357SFande Kong . snes - the SNES context 23207b62357SFande Kong 23307b62357SFande Kong Output Parameters: 23407b62357SFande Kong . domainerror - Set to PETSC_TRUE if there's a jacobian domain error; PETSC_FALSE otherwise. 23507b62357SFande Kong 23607b62357SFande Kong Level: advanced 23707b62357SFande Kong 23807b62357SFande Kong .seealso: SNESSetFunctionDomainError(), SNESComputeFunction(),SNESGetFunctionDomainError() 23907b62357SFande Kong @*/ 24007b62357SFande Kong PetscErrorCode SNESGetJacobianDomainError(SNES snes, PetscBool *domainerror) 24107b62357SFande Kong { 24207b62357SFande Kong PetscFunctionBegin; 24307b62357SFande Kong PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 244534a8f05SLisandro Dalcin PetscValidBoolPointer(domainerror,2); 24507b62357SFande Kong *domainerror = snes->jacobiandomainerror; 24607b62357SFande Kong PetscFunctionReturn(0); 24707b62357SFande Kong } 24807b62357SFande Kong 24955849f57SBarry Smith /*@C 25055849f57SBarry Smith SNESLoad - Loads a SNES that has been stored in binary with SNESView(). 25155849f57SBarry Smith 25255849f57SBarry Smith Collective on PetscViewer 25355849f57SBarry Smith 25455849f57SBarry Smith Input Parameters: 25555849f57SBarry Smith + newdm - the newly loaded SNES, this needs to have been created with SNESCreate() or 25655849f57SBarry Smith some related function before a call to SNESLoad(). 25755849f57SBarry Smith - viewer - binary file viewer, obtained from PetscViewerBinaryOpen() 25855849f57SBarry Smith 25955849f57SBarry Smith Level: intermediate 26055849f57SBarry Smith 26155849f57SBarry Smith Notes: 26255849f57SBarry Smith The type is determined by the data in the file, any type set into the SNES before this call is ignored. 26355849f57SBarry Smith 26455849f57SBarry Smith Notes for advanced users: 26555849f57SBarry Smith Most users should not need to know the details of the binary storage 26655849f57SBarry Smith format, since SNESLoad() and TSView() completely hide these details. 26755849f57SBarry Smith But for anyone who's interested, the standard binary matrix storage 26855849f57SBarry Smith format is 26955849f57SBarry Smith .vb 27055849f57SBarry Smith has not yet been determined 27155849f57SBarry Smith .ve 27255849f57SBarry Smith 27355849f57SBarry Smith .seealso: PetscViewerBinaryOpen(), SNESView(), MatLoad(), VecLoad() 27455849f57SBarry Smith @*/ 2752d53ad75SBarry Smith PetscErrorCode SNESLoad(SNES snes, PetscViewer viewer) 27655849f57SBarry Smith { 27755849f57SBarry Smith PetscErrorCode ierr; 27855849f57SBarry Smith PetscBool isbinary; 279060da220SMatthew G. Knepley PetscInt classid; 28055849f57SBarry Smith char type[256]; 28155849f57SBarry Smith KSP ksp; 2822d53ad75SBarry Smith DM dm; 2832d53ad75SBarry Smith DMSNES dmsnes; 28455849f57SBarry Smith 28555849f57SBarry Smith PetscFunctionBegin; 2862d53ad75SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 28755849f57SBarry Smith PetscValidHeaderSpecific(viewer,PETSC_VIEWER_CLASSID,2); 28855849f57SBarry Smith ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERBINARY,&isbinary);CHKERRQ(ierr); 28955849f57SBarry Smith if (!isbinary) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONG,"Invalid viewer; open viewer with PetscViewerBinaryOpen()"); 29055849f57SBarry Smith 291060da220SMatthew G. Knepley ierr = PetscViewerBinaryRead(viewer,&classid,1,NULL,PETSC_INT);CHKERRQ(ierr); 292ce94432eSBarry Smith if (classid != SNES_FILE_CLASSID) SETERRQ(PetscObjectComm((PetscObject)snes),PETSC_ERR_ARG_WRONG,"Not SNES next in file"); 293060da220SMatthew G. Knepley ierr = PetscViewerBinaryRead(viewer,type,256,NULL,PETSC_CHAR);CHKERRQ(ierr); 2942d53ad75SBarry Smith ierr = SNESSetType(snes, type);CHKERRQ(ierr); 2952d53ad75SBarry Smith if (snes->ops->load) { 2962d53ad75SBarry Smith ierr = (*snes->ops->load)(snes,viewer);CHKERRQ(ierr); 297f2c2a1b9SBarry Smith } 2982d53ad75SBarry Smith ierr = SNESGetDM(snes,&dm);CHKERRQ(ierr); 2992d53ad75SBarry Smith ierr = DMGetDMSNES(dm,&dmsnes);CHKERRQ(ierr); 3002d53ad75SBarry Smith ierr = DMSNESLoad(dmsnes,viewer);CHKERRQ(ierr); 3012d53ad75SBarry Smith ierr = SNESGetKSP(snes,&ksp);CHKERRQ(ierr); 30255849f57SBarry Smith ierr = KSPLoad(ksp,viewer);CHKERRQ(ierr); 30355849f57SBarry Smith PetscFunctionReturn(0); 30455849f57SBarry Smith } 3056a388c36SPeter Brune 3069804daf3SBarry Smith #include <petscdraw.h> 307e04113cfSBarry Smith #if defined(PETSC_HAVE_SAWS) 308e04113cfSBarry Smith #include <petscviewersaws.h> 309bfb97211SBarry Smith #endif 3108404b7f3SBarry Smith 311fe2efc57SMark /*@C 312fe2efc57SMark SNESViewFromOptions - View from Options 313fe2efc57SMark 314fe2efc57SMark Collective on SNES 315fe2efc57SMark 316fe2efc57SMark Input Parameters: 317fe2efc57SMark + A - the application ordering context 318736c3998SJose E. Roman . obj - Optional object 319736c3998SJose E. Roman - name - command line option 320fe2efc57SMark 321fe2efc57SMark Level: intermediate 322fe2efc57SMark .seealso: SNES, SNESView, PetscObjectViewFromOptions(), SNESCreate() 323fe2efc57SMark @*/ 324fe2efc57SMark PetscErrorCode SNESViewFromOptions(SNES A,PetscObject obj,const char name[]) 325fe2efc57SMark { 326fe2efc57SMark PetscErrorCode ierr; 327fe2efc57SMark 328fe2efc57SMark PetscFunctionBegin; 329fe2efc57SMark PetscValidHeaderSpecific(A,SNES_CLASSID,1); 330fe2efc57SMark ierr = PetscObjectViewFromOptions((PetscObject)A,obj,name);CHKERRQ(ierr); 331fe2efc57SMark PetscFunctionReturn(0); 332fe2efc57SMark } 333fe2efc57SMark 334789d8953SBarry Smith PETSC_EXTERN PetscErrorCode SNESComputeJacobian_DMDA(SNES,Vec,Mat,Mat,void*); 335789d8953SBarry Smith 3367e2c5f70SBarry Smith /*@C 3379b94acceSBarry Smith SNESView - Prints the SNES data structure. 3389b94acceSBarry Smith 3394c49b128SBarry Smith Collective on SNES 340fee21e36SBarry Smith 341c7afd0dbSLois Curfman McInnes Input Parameters: 342c7afd0dbSLois Curfman McInnes + SNES - the SNES context 343c7afd0dbSLois Curfman McInnes - viewer - visualization context 344c7afd0dbSLois Curfman McInnes 3459b94acceSBarry Smith Options Database Key: 346c8a8ba5cSLois Curfman McInnes . -snes_view - Calls SNESView() at end of SNESSolve() 3479b94acceSBarry Smith 3489b94acceSBarry Smith Notes: 3499b94acceSBarry Smith The available visualization contexts include 350b0a32e0cSBarry Smith + PETSC_VIEWER_STDOUT_SELF - standard output (default) 351b0a32e0cSBarry Smith - PETSC_VIEWER_STDOUT_WORLD - synchronized standard 352c8a8ba5cSLois Curfman McInnes output where only the first processor opens 353c8a8ba5cSLois Curfman McInnes the file. All other processors send their 354c8a8ba5cSLois Curfman McInnes data to the first processor to print. 3559b94acceSBarry Smith 356*052bf0daSPierre Jolivet The available formats include 357*052bf0daSPierre Jolivet + PETSC_VIEWER_DEFAULT - standard output (default) 358*052bf0daSPierre Jolivet - PETSC_VIEWER_ASCII_INFO_DETAIL - more verbose output for SNESNASM 359*052bf0daSPierre Jolivet 3603e081fefSLois Curfman McInnes The user can open an alternative visualization context with 361b0a32e0cSBarry Smith PetscViewerASCIIOpen() - output to a specified file. 3629b94acceSBarry Smith 363595c91d4SBarry Smith In the debugger you can do "call SNESView(snes,0)" to display the SNES solver. (The same holds for any PETSc object viewer). 364595c91d4SBarry Smith 36536851e7fSLois Curfman McInnes Level: beginner 36636851e7fSLois Curfman McInnes 367b0a32e0cSBarry Smith .seealso: PetscViewerASCIIOpen() 3689b94acceSBarry Smith @*/ 3697087cfbeSBarry Smith PetscErrorCode SNESView(SNES snes,PetscViewer viewer) 3709b94acceSBarry Smith { 371fa9f3622SBarry Smith SNESKSPEW *kctx; 372dfbe8321SBarry Smith PetscErrorCode ierr; 37394b7f48cSBarry Smith KSP ksp; 3747f1410a3SPeter Brune SNESLineSearch linesearch; 37572a02f06SBarry Smith PetscBool iascii,isstring,isbinary,isdraw; 3762d53ad75SBarry Smith DMSNES dmsnes; 377e04113cfSBarry Smith #if defined(PETSC_HAVE_SAWS) 378536b137fSBarry Smith PetscBool issaws; 379bfb97211SBarry Smith #endif 3809b94acceSBarry Smith 3813a40ed3dSBarry Smith PetscFunctionBegin; 3820700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 3833050cee2SBarry Smith if (!viewer) { 384ce94432eSBarry Smith ierr = PetscViewerASCIIGetStdout(PetscObjectComm((PetscObject)snes),&viewer);CHKERRQ(ierr); 3853050cee2SBarry Smith } 3860700a824SBarry Smith PetscValidHeaderSpecific(viewer,PETSC_VIEWER_CLASSID,2); 387c9780b6fSBarry Smith PetscCheckSameComm(snes,1,viewer,2); 38874679c65SBarry Smith 389251f4c67SDmitry Karpeev ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERASCII,&iascii);CHKERRQ(ierr); 390251f4c67SDmitry Karpeev ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERSTRING,&isstring);CHKERRQ(ierr); 39155849f57SBarry Smith ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERBINARY,&isbinary);CHKERRQ(ierr); 39272a02f06SBarry Smith ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERDRAW,&isdraw);CHKERRQ(ierr); 393e04113cfSBarry Smith #if defined(PETSC_HAVE_SAWS) 394536b137fSBarry Smith ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERSAWS,&issaws);CHKERRQ(ierr); 395bfb97211SBarry Smith #endif 39632077d6dSBarry Smith if (iascii) { 397dc0571f2SMatthew G. Knepley SNESNormSchedule normschedule; 3988404b7f3SBarry Smith DM dm; 3998404b7f3SBarry Smith PetscErrorCode (*cJ)(SNES,Vec,Mat,Mat,void*); 4008404b7f3SBarry Smith void *ctx; 401789d8953SBarry Smith const char *pre = ""; 402dc0571f2SMatthew G. Knepley 403dae58748SBarry Smith ierr = PetscObjectPrintClassNamePrefixType((PetscObject)snes,viewer);CHKERRQ(ierr); 404fce1e034SJed Brown if (!snes->setupcalled) { 405fce1e034SJed Brown ierr = PetscViewerASCIIPrintf(viewer," SNES has not been set up so information may be incomplete\n");CHKERRQ(ierr); 406fce1e034SJed Brown } 407e7788613SBarry Smith if (snes->ops->view) { 408b0a32e0cSBarry Smith ierr = PetscViewerASCIIPushTab(viewer);CHKERRQ(ierr); 409e7788613SBarry Smith ierr = (*snes->ops->view)(snes,viewer);CHKERRQ(ierr); 410b0a32e0cSBarry Smith ierr = PetscViewerASCIIPopTab(viewer);CHKERRQ(ierr); 4110ef38995SBarry Smith } 41277431f27SBarry Smith ierr = PetscViewerASCIIPrintf(viewer," maximum iterations=%D, maximum function evaluations=%D\n",snes->max_its,snes->max_funcs);CHKERRQ(ierr); 41357622a8eSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," tolerances: relative=%g, absolute=%g, solution=%g\n",(double)snes->rtol,(double)snes->abstol,(double)snes->stol);CHKERRQ(ierr); 414efd4aadfSBarry Smith if (snes->usesksp) { 41577431f27SBarry Smith ierr = PetscViewerASCIIPrintf(viewer," total number of linear solver iterations=%D\n",snes->linear_its);CHKERRQ(ierr); 416efd4aadfSBarry Smith } 41777431f27SBarry Smith ierr = PetscViewerASCIIPrintf(viewer," total number of function evaluations=%D\n",snes->nfuncs);CHKERRQ(ierr); 418dc0571f2SMatthew G. Knepley ierr = SNESGetNormSchedule(snes, &normschedule);CHKERRQ(ierr); 419dc0571f2SMatthew G. Knepley if (normschedule > 0) {ierr = PetscViewerASCIIPrintf(viewer," norm schedule %s\n",SNESNormSchedules[normschedule]);CHKERRQ(ierr);} 42017fe4bdfSPeter Brune if (snes->gridsequence) { 42117fe4bdfSPeter Brune ierr = PetscViewerASCIIPrintf(viewer," total number of grid sequence refinements=%D\n",snes->gridsequence);CHKERRQ(ierr); 42217fe4bdfSPeter Brune } 4239b94acceSBarry Smith if (snes->ksp_ewconv) { 424fa9f3622SBarry Smith kctx = (SNESKSPEW*)snes->kspconvctx; 4259b94acceSBarry Smith if (kctx) { 42677431f27SBarry Smith ierr = PetscViewerASCIIPrintf(viewer," Eisenstat-Walker computation of KSP relative tolerance (version %D)\n",kctx->version);CHKERRQ(ierr); 42757622a8eSBarry 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); 42857622a8eSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," gamma=%g, alpha=%g, alpha2=%g\n",(double)kctx->gamma,(double)kctx->alpha,(double)kctx->alpha2);CHKERRQ(ierr); 4299b94acceSBarry Smith } 4309b94acceSBarry Smith } 431eb1f6c34SBarry Smith if (snes->lagpreconditioner == -1) { 432eb1f6c34SBarry Smith ierr = PetscViewerASCIIPrintf(viewer," Preconditioned is never rebuilt\n");CHKERRQ(ierr); 433eb1f6c34SBarry Smith } else if (snes->lagpreconditioner > 1) { 434eb1f6c34SBarry Smith ierr = PetscViewerASCIIPrintf(viewer," Preconditioned is rebuilt every %D new Jacobians\n",snes->lagpreconditioner);CHKERRQ(ierr); 435eb1f6c34SBarry Smith } 436eb1f6c34SBarry Smith if (snes->lagjacobian == -1) { 437eb1f6c34SBarry Smith ierr = PetscViewerASCIIPrintf(viewer," Jacobian is never rebuilt\n");CHKERRQ(ierr); 438eb1f6c34SBarry Smith } else if (snes->lagjacobian > 1) { 43942f4f86dSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," Jacobian is rebuilt every %D SNES iterations\n",snes->lagjacobian);CHKERRQ(ierr); 440eb1f6c34SBarry Smith } 4418404b7f3SBarry Smith ierr = SNESGetDM(snes,&dm);CHKERRQ(ierr); 4428404b7f3SBarry Smith ierr = DMSNESGetJacobian(dm,&cJ,&ctx);CHKERRQ(ierr); 443789d8953SBarry Smith if (snes->mf_operator) { 444789d8953SBarry Smith ierr = PetscViewerASCIIPrintf(viewer," Jacobian is applied matrix-free with differencing\n");CHKERRQ(ierr); 445789d8953SBarry Smith pre = "Preconditioning "; 446789d8953SBarry Smith } 4478404b7f3SBarry Smith if (cJ == SNESComputeJacobianDefault) { 448789d8953SBarry Smith ierr = PetscViewerASCIIPrintf(viewer," %sJacobian is built using finite differences one column at a time\n",pre);CHKERRQ(ierr); 4498404b7f3SBarry Smith } else if (cJ == SNESComputeJacobianDefaultColor) { 450789d8953SBarry Smith ierr = PetscViewerASCIIPrintf(viewer," %sJacobian is built using finite differences with coloring\n",pre);CHKERRQ(ierr); 451789d8953SBarry Smith /* it slightly breaks data encapsulation for access the DMDA information directly */ 452789d8953SBarry Smith } else if (cJ == SNESComputeJacobian_DMDA) { 453789d8953SBarry Smith MatFDColoring fdcoloring; 454789d8953SBarry Smith ierr = PetscObjectQuery((PetscObject)dm,"DMDASNES_FDCOLORING",(PetscObject*)&fdcoloring);CHKERRQ(ierr); 455789d8953SBarry Smith if (fdcoloring) { 456789d8953SBarry Smith ierr = PetscViewerASCIIPrintf(viewer," %sJacobian is built using colored finite differences on a DMDA\n",pre);CHKERRQ(ierr); 457789d8953SBarry Smith } else { 458789d8953SBarry Smith ierr = PetscViewerASCIIPrintf(viewer," %sJacobian is built using a DMDA local Jacobian\n",pre);CHKERRQ(ierr); 459789d8953SBarry Smith } 460789d8953SBarry Smith } else if (snes->mf) { 461789d8953SBarry Smith ierr = PetscViewerASCIIPrintf(viewer," Jacobian is applied matrix-free with differencing, no explict Jacobian\n");CHKERRQ(ierr); 4628404b7f3SBarry Smith } 4630f5bd95cSBarry Smith } else if (isstring) { 464317d6ea6SBarry Smith const char *type; 465454a90a3SBarry Smith ierr = SNESGetType(snes,&type);CHKERRQ(ierr); 46636a9e3b9SBarry Smith ierr = PetscViewerStringSPrintf(viewer," SNESType: %-7.7s",type);CHKERRQ(ierr); 46736a9e3b9SBarry Smith if (snes->ops->view) {ierr = (*snes->ops->view)(snes,viewer);CHKERRQ(ierr);} 46855849f57SBarry Smith } else if (isbinary) { 46955849f57SBarry Smith PetscInt classid = SNES_FILE_CLASSID; 47055849f57SBarry Smith MPI_Comm comm; 47155849f57SBarry Smith PetscMPIInt rank; 47255849f57SBarry Smith char type[256]; 47355849f57SBarry Smith 47455849f57SBarry Smith ierr = PetscObjectGetComm((PetscObject)snes,&comm);CHKERRQ(ierr); 475ffc4695bSBarry Smith ierr = MPI_Comm_rank(comm,&rank);CHKERRMPI(ierr); 47655849f57SBarry Smith if (!rank) { 477f253e43cSLisandro Dalcin ierr = PetscViewerBinaryWrite(viewer,&classid,1,PETSC_INT);CHKERRQ(ierr); 47889d949e2SBarry Smith ierr = PetscStrncpy(type,((PetscObject)snes)->type_name,sizeof(type));CHKERRQ(ierr); 479f253e43cSLisandro Dalcin ierr = PetscViewerBinaryWrite(viewer,type,sizeof(type),PETSC_CHAR);CHKERRQ(ierr); 48055849f57SBarry Smith } 48155849f57SBarry Smith if (snes->ops->view) { 48255849f57SBarry Smith ierr = (*snes->ops->view)(snes,viewer);CHKERRQ(ierr); 48355849f57SBarry Smith } 48472a02f06SBarry Smith } else if (isdraw) { 48572a02f06SBarry Smith PetscDraw draw; 48672a02f06SBarry Smith char str[36]; 48789fd9fafSBarry Smith PetscReal x,y,bottom,h; 48872a02f06SBarry Smith 48972a02f06SBarry Smith ierr = PetscViewerDrawGetDraw(viewer,0,&draw);CHKERRQ(ierr); 49072a02f06SBarry Smith ierr = PetscDrawGetCurrentPoint(draw,&x,&y);CHKERRQ(ierr); 491a126751eSBarry Smith ierr = PetscStrncpy(str,"SNES: ",sizeof(str));CHKERRQ(ierr); 492a126751eSBarry Smith ierr = PetscStrlcat(str,((PetscObject)snes)->type_name,sizeof(str));CHKERRQ(ierr); 49351fa3d41SBarry Smith ierr = PetscDrawStringBoxed(draw,x,y,PETSC_DRAW_BLUE,PETSC_DRAW_BLACK,str,NULL,&h);CHKERRQ(ierr); 49489fd9fafSBarry Smith bottom = y - h; 49572a02f06SBarry Smith ierr = PetscDrawPushCurrentPoint(draw,x,bottom);CHKERRQ(ierr); 496c4646bacSPeter Brune if (snes->ops->view) { 497c4646bacSPeter Brune ierr = (*snes->ops->view)(snes,viewer);CHKERRQ(ierr); 498c4646bacSPeter Brune } 499e04113cfSBarry Smith #if defined(PETSC_HAVE_SAWS) 500536b137fSBarry Smith } else if (issaws) { 501d45a07a7SBarry Smith PetscMPIInt rank; 5022657e9d9SBarry Smith const char *name; 503d45a07a7SBarry Smith 5042657e9d9SBarry Smith ierr = PetscObjectGetName((PetscObject)snes,&name);CHKERRQ(ierr); 505ffc4695bSBarry Smith ierr = MPI_Comm_rank(PETSC_COMM_WORLD,&rank);CHKERRMPI(ierr); 506d45a07a7SBarry Smith if (!((PetscObject)snes)->amsmem && !rank) { 507d45a07a7SBarry Smith char dir[1024]; 508d45a07a7SBarry Smith 509e04113cfSBarry Smith ierr = PetscObjectViewSAWs((PetscObject)snes,viewer);CHKERRQ(ierr); 5102657e9d9SBarry Smith ierr = PetscSNPrintf(dir,1024,"/PETSc/Objects/%s/its",name);CHKERRQ(ierr); 5112657e9d9SBarry Smith PetscStackCallSAWs(SAWs_Register,(dir,&snes->iter,1,SAWs_READ,SAWs_INT)); 512bfb97211SBarry Smith if (!snes->conv_hist) { 513a0931e03SBarry Smith ierr = SNESSetConvergenceHistory(snes,NULL,NULL,PETSC_DECIDE,PETSC_TRUE);CHKERRQ(ierr); 514bfb97211SBarry Smith } 5152657e9d9SBarry Smith ierr = PetscSNPrintf(dir,1024,"/PETSc/Objects/%s/conv_hist",name);CHKERRQ(ierr); 5162657e9d9SBarry Smith PetscStackCallSAWs(SAWs_Register,(dir,snes->conv_hist,10,SAWs_READ,SAWs_DOUBLE)); 517f05ece33SBarry Smith } 518bfb97211SBarry Smith #endif 51972a02f06SBarry Smith } 52072a02f06SBarry Smith if (snes->linesearch) { 5217601faf0SJed Brown ierr = SNESGetLineSearch(snes, &linesearch);CHKERRQ(ierr); 52285928a98SStefano Zampini ierr = PetscViewerASCIIPushTab(viewer);CHKERRQ(ierr); 52372a02f06SBarry Smith ierr = SNESLineSearchView(linesearch, viewer);CHKERRQ(ierr); 52485928a98SStefano Zampini ierr = PetscViewerASCIIPopTab(viewer);CHKERRQ(ierr); 52519bcc07fSBarry Smith } 526efd4aadfSBarry Smith if (snes->npc && snes->usesnpc) { 52785928a98SStefano Zampini ierr = PetscViewerASCIIPushTab(viewer);CHKERRQ(ierr); 528efd4aadfSBarry Smith ierr = SNESView(snes->npc, viewer);CHKERRQ(ierr); 52985928a98SStefano Zampini ierr = PetscViewerASCIIPopTab(viewer);CHKERRQ(ierr); 5304a0c5b0cSMatthew G Knepley } 5312d53ad75SBarry Smith ierr = PetscViewerASCIIPushTab(viewer);CHKERRQ(ierr); 5322d53ad75SBarry Smith ierr = DMGetDMSNES(snes->dm,&dmsnes);CHKERRQ(ierr); 5332d53ad75SBarry Smith ierr = DMSNESView(dmsnes, viewer);CHKERRQ(ierr); 5342d53ad75SBarry Smith ierr = PetscViewerASCIIPopTab(viewer);CHKERRQ(ierr); 5352c155ee1SBarry Smith if (snes->usesksp) { 5362c155ee1SBarry Smith ierr = SNESGetKSP(snes,&ksp);CHKERRQ(ierr); 53785928a98SStefano Zampini ierr = PetscViewerASCIIPushTab(viewer);CHKERRQ(ierr); 53894b7f48cSBarry Smith ierr = KSPView(ksp,viewer);CHKERRQ(ierr); 53985928a98SStefano Zampini ierr = PetscViewerASCIIPopTab(viewer);CHKERRQ(ierr); 5402c155ee1SBarry Smith } 54172a02f06SBarry Smith if (isdraw) { 54272a02f06SBarry Smith PetscDraw draw; 54372a02f06SBarry Smith ierr = PetscViewerDrawGetDraw(viewer,0,&draw);CHKERRQ(ierr); 54472a02f06SBarry Smith ierr = PetscDrawPopCurrentPoint(draw);CHKERRQ(ierr); 5457f1410a3SPeter Brune } 5463a40ed3dSBarry Smith PetscFunctionReturn(0); 5479b94acceSBarry Smith } 5489b94acceSBarry Smith 54976b2cf59SMatthew Knepley /* 55076b2cf59SMatthew Knepley We retain a list of functions that also take SNES command 55176b2cf59SMatthew Knepley line options. These are called at the end SNESSetFromOptions() 55276b2cf59SMatthew Knepley */ 55376b2cf59SMatthew Knepley #define MAXSETFROMOPTIONS 5 554a7cc72afSBarry Smith static PetscInt numberofsetfromoptions; 5556849ba73SBarry Smith static PetscErrorCode (*othersetfromoptions[MAXSETFROMOPTIONS])(SNES); 55676b2cf59SMatthew Knepley 557ac226902SBarry Smith /*@C 55876b2cf59SMatthew Knepley SNESAddOptionsChecker - Adds an additional function to check for SNES options. 55976b2cf59SMatthew Knepley 56076b2cf59SMatthew Knepley Not Collective 56176b2cf59SMatthew Knepley 56276b2cf59SMatthew Knepley Input Parameter: 56376b2cf59SMatthew Knepley . snescheck - function that checks for options 56476b2cf59SMatthew Knepley 56576b2cf59SMatthew Knepley Level: developer 56676b2cf59SMatthew Knepley 56776b2cf59SMatthew Knepley .seealso: SNESSetFromOptions() 56876b2cf59SMatthew Knepley @*/ 5697087cfbeSBarry Smith PetscErrorCode SNESAddOptionsChecker(PetscErrorCode (*snescheck)(SNES)) 57076b2cf59SMatthew Knepley { 57176b2cf59SMatthew Knepley PetscFunctionBegin; 572f23aa3ddSBarry Smith if (numberofsetfromoptions >= MAXSETFROMOPTIONS) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE, "Too many options checkers, only %D allowed", MAXSETFROMOPTIONS); 57376b2cf59SMatthew Knepley othersetfromoptions[numberofsetfromoptions++] = snescheck; 57476b2cf59SMatthew Knepley PetscFunctionReturn(0); 57576b2cf59SMatthew Knepley } 57676b2cf59SMatthew Knepley 57725acbd8eSLisandro Dalcin PETSC_INTERN PetscErrorCode SNESDefaultMatrixFreeCreate2(SNES,Vec,Mat*); 578aa3661deSLisandro Dalcin 579ace3abfcSBarry Smith static PetscErrorCode SNESSetUpMatrixFree_Private(SNES snes, PetscBool hasOperator, PetscInt version) 580aa3661deSLisandro Dalcin { 581aa3661deSLisandro Dalcin Mat J; 582aa3661deSLisandro Dalcin PetscErrorCode ierr; 583895c21f2SBarry Smith MatNullSpace nullsp; 584aa3661deSLisandro Dalcin 585aa3661deSLisandro Dalcin PetscFunctionBegin; 5860700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 587aa3661deSLisandro Dalcin 58898613b67SLisandro Dalcin if (!snes->vec_func && (snes->jacobian || snes->jacobian_pre)) { 58998613b67SLisandro Dalcin Mat A = snes->jacobian, B = snes->jacobian_pre; 5902a7a6963SBarry Smith ierr = MatCreateVecs(A ? A : B, NULL,&snes->vec_func);CHKERRQ(ierr); 59198613b67SLisandro Dalcin } 59298613b67SLisandro Dalcin 593aa3661deSLisandro Dalcin if (version == 1) { 594aa3661deSLisandro Dalcin ierr = MatCreateSNESMF(snes,&J);CHKERRQ(ierr); 59598613b67SLisandro Dalcin ierr = MatMFFDSetOptionsPrefix(J,((PetscObject)snes)->prefix);CHKERRQ(ierr); 5969c6ac3b3SBarry Smith ierr = MatSetFromOptions(J);CHKERRQ(ierr); 597aa3661deSLisandro Dalcin } else if (version == 2) { 598e32f2f54SBarry Smith if (!snes->vec_func) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"SNESSetFunction() must be called first"); 599570b7f6dSBarry Smith #if !defined(PETSC_USE_COMPLEX) && !defined(PETSC_USE_REAL_SINGLE) && !defined(PETSC_USE_REAL___FLOAT128) && !defined(PETSC_USE_REAL___FP16) 600aa3661deSLisandro Dalcin ierr = SNESDefaultMatrixFreeCreate2(snes,snes->vec_func,&J);CHKERRQ(ierr); 601aa3661deSLisandro Dalcin #else 6022479783cSJose E. Roman SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP, "matrix-free operator routines (version 2)"); 603aa3661deSLisandro Dalcin #endif 6042479783cSJose E. Roman } else SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE, "matrix-free operator routines, only version 1 and 2"); 605aa3661deSLisandro Dalcin 606895c21f2SBarry Smith /* attach any user provided null space that was on Amat to the newly created matrix free matrix */ 607895c21f2SBarry Smith if (snes->jacobian) { 608895c21f2SBarry Smith ierr = MatGetNullSpace(snes->jacobian,&nullsp);CHKERRQ(ierr); 609895c21f2SBarry Smith if (nullsp) { 610895c21f2SBarry Smith ierr = MatSetNullSpace(J,nullsp);CHKERRQ(ierr); 611895c21f2SBarry Smith } 612895c21f2SBarry Smith } 613895c21f2SBarry Smith 614aa3661deSLisandro Dalcin ierr = PetscInfo1(snes,"Setting default matrix-free operator routines (version %D)\n", version);CHKERRQ(ierr); 615d3462f78SMatthew Knepley if (hasOperator) { 6163232da50SPeter Brune 617aa3661deSLisandro Dalcin /* This version replaces the user provided Jacobian matrix with a 618aa3661deSLisandro Dalcin matrix-free version but still employs the user-provided preconditioner matrix. */ 6199e5d0892SLisandro Dalcin ierr = SNESSetJacobian(snes,J,NULL,NULL,NULL);CHKERRQ(ierr); 620aa3661deSLisandro Dalcin } else { 621aa3661deSLisandro Dalcin /* This version replaces both the user-provided Jacobian and the user- 6223232da50SPeter Brune provided preconditioner Jacobian with the default matrix free version. */ 623efd4aadfSBarry Smith if ((snes->npcside== PC_LEFT) && snes->npc) { 6249e5d0892SLisandro Dalcin if (!snes->jacobian){ierr = SNESSetJacobian(snes,J,NULL,NULL,NULL);CHKERRQ(ierr);} 625172a4300SPeter Brune } else { 626789d8953SBarry Smith KSP ksp; 627789d8953SBarry Smith PC pc; 628789d8953SBarry Smith PetscBool match; 629789d8953SBarry Smith 6309e5d0892SLisandro Dalcin ierr = SNESSetJacobian(snes,J,J,MatMFFDComputeJacobian,NULL);CHKERRQ(ierr); 631aa3661deSLisandro Dalcin /* Force no preconditioner */ 632aa3661deSLisandro Dalcin ierr = SNESGetKSP(snes,&ksp);CHKERRQ(ierr); 633aa3661deSLisandro Dalcin ierr = KSPGetPC(ksp,&pc);CHKERRQ(ierr); 634251f4c67SDmitry Karpeev ierr = PetscObjectTypeCompare((PetscObject)pc,PCSHELL,&match);CHKERRQ(ierr); 635aa3661deSLisandro Dalcin if (!match) { 636aa3661deSLisandro Dalcin ierr = PetscInfo(snes,"Setting default matrix-free preconditioner routines\nThat is no preconditioner is being used\n");CHKERRQ(ierr); 637aa3661deSLisandro Dalcin ierr = PCSetType(pc,PCNONE);CHKERRQ(ierr); 638aa3661deSLisandro Dalcin } 639aa3661deSLisandro Dalcin } 640789d8953SBarry Smith } 6416bf464f9SBarry Smith ierr = MatDestroy(&J);CHKERRQ(ierr); 642aa3661deSLisandro Dalcin PetscFunctionReturn(0); 643aa3661deSLisandro Dalcin } 644aa3661deSLisandro Dalcin 645dfe15315SJed Brown static PetscErrorCode DMRestrictHook_SNESVecSol(DM dmfine,Mat Restrict,Vec Rscale,Mat Inject,DM dmcoarse,void *ctx) 646dfe15315SJed Brown { 647dfe15315SJed Brown SNES snes = (SNES)ctx; 648dfe15315SJed Brown PetscErrorCode ierr; 6490298fd71SBarry Smith Vec Xfine,Xfine_named = NULL,Xcoarse; 650dfe15315SJed Brown 651dfe15315SJed Brown PetscFunctionBegin; 65216ebb321SJed Brown if (PetscLogPrintInfo) { 65316ebb321SJed Brown PetscInt finelevel,coarselevel,fineclevel,coarseclevel; 65416ebb321SJed Brown ierr = DMGetRefineLevel(dmfine,&finelevel);CHKERRQ(ierr); 65516ebb321SJed Brown ierr = DMGetCoarsenLevel(dmfine,&fineclevel);CHKERRQ(ierr); 65616ebb321SJed Brown ierr = DMGetRefineLevel(dmcoarse,&coarselevel);CHKERRQ(ierr); 65716ebb321SJed Brown ierr = DMGetCoarsenLevel(dmcoarse,&coarseclevel);CHKERRQ(ierr); 65816ebb321SJed Brown ierr = PetscInfo4(dmfine,"Restricting SNES solution vector from level %D-%D to level %D-%D\n",finelevel,fineclevel,coarselevel,coarseclevel);CHKERRQ(ierr); 65916ebb321SJed Brown } 660dfe15315SJed Brown if (dmfine == snes->dm) Xfine = snes->vec_sol; 661dfe15315SJed Brown else { 662dfe15315SJed Brown ierr = DMGetNamedGlobalVector(dmfine,"SNESVecSol",&Xfine_named);CHKERRQ(ierr); 663dfe15315SJed Brown Xfine = Xfine_named; 664dfe15315SJed Brown } 665dfe15315SJed Brown ierr = DMGetNamedGlobalVector(dmcoarse,"SNESVecSol",&Xcoarse);CHKERRQ(ierr); 666907f5c5aSLawrence Mitchell if (Inject) { 667907f5c5aSLawrence Mitchell ierr = MatRestrict(Inject,Xfine,Xcoarse);CHKERRQ(ierr); 668907f5c5aSLawrence Mitchell } else { 669dfe15315SJed Brown ierr = MatRestrict(Restrict,Xfine,Xcoarse);CHKERRQ(ierr); 670dfe15315SJed Brown ierr = VecPointwiseMult(Xcoarse,Xcoarse,Rscale);CHKERRQ(ierr); 671907f5c5aSLawrence Mitchell } 672dfe15315SJed Brown ierr = DMRestoreNamedGlobalVector(dmcoarse,"SNESVecSol",&Xcoarse);CHKERRQ(ierr); 673dfe15315SJed Brown if (Xfine_named) {ierr = DMRestoreNamedGlobalVector(dmfine,"SNESVecSol",&Xfine_named);CHKERRQ(ierr);} 674dfe15315SJed Brown PetscFunctionReturn(0); 675dfe15315SJed Brown } 676dfe15315SJed Brown 67716ebb321SJed Brown static PetscErrorCode DMCoarsenHook_SNESVecSol(DM dm,DM dmc,void *ctx) 67816ebb321SJed Brown { 67916ebb321SJed Brown PetscErrorCode ierr; 68016ebb321SJed Brown 68116ebb321SJed Brown PetscFunctionBegin; 68216ebb321SJed Brown ierr = DMCoarsenHookAdd(dmc,DMCoarsenHook_SNESVecSol,DMRestrictHook_SNESVecSol,ctx);CHKERRQ(ierr); 68316ebb321SJed Brown PetscFunctionReturn(0); 68416ebb321SJed Brown } 68516ebb321SJed Brown 686a6950cb2SJed Brown /* This may be called to rediscretize the operator on levels of linear multigrid. The DM shuffle is so the user can 687a6950cb2SJed Brown * safely call SNESGetDM() in their residual evaluation routine. */ 68823ee1639SBarry Smith static PetscErrorCode KSPComputeOperators_SNES(KSP ksp,Mat A,Mat B,void *ctx) 689caa4e7f2SJed Brown { 690caa4e7f2SJed Brown SNES snes = (SNES)ctx; 691caa4e7f2SJed Brown PetscErrorCode ierr; 6920298fd71SBarry Smith Vec X,Xnamed = NULL; 693dfe15315SJed Brown DM dmsave; 6944e269d77SPeter Brune void *ctxsave; 69525ce1634SJed Brown PetscErrorCode (*jac)(SNES,Vec,Mat,Mat,void*) = NULL; 696caa4e7f2SJed Brown 697caa4e7f2SJed Brown PetscFunctionBegin; 698dfe15315SJed Brown dmsave = snes->dm; 699dfe15315SJed Brown ierr = KSPGetDM(ksp,&snes->dm);CHKERRQ(ierr); 700dfe15315SJed Brown if (dmsave == snes->dm) X = snes->vec_sol; /* We are on the finest level */ 701dfe15315SJed Brown else { /* We are on a coarser level, this vec was initialized using a DM restrict hook */ 702dfe15315SJed Brown ierr = DMGetNamedGlobalVector(snes->dm,"SNESVecSol",&Xnamed);CHKERRQ(ierr); 703dfe15315SJed Brown X = Xnamed; 7040298fd71SBarry Smith ierr = SNESGetJacobian(snes,NULL,NULL,&jac,&ctxsave);CHKERRQ(ierr); 7054e269d77SPeter Brune /* If the DM's don't match up, the MatFDColoring context needed for the jacobian won't match up either -- fixit. */ 7068d359177SBarry Smith if (jac == SNESComputeJacobianDefaultColor) { 7079e5d0892SLisandro Dalcin ierr = SNESSetJacobian(snes,NULL,NULL,SNESComputeJacobianDefaultColor,NULL);CHKERRQ(ierr); 708dfe15315SJed Brown } 7094e269d77SPeter Brune } 7104dde8bb0SMatthew G. Knepley /* Make sure KSP DM has the Jacobian computation routine */ 7114dde8bb0SMatthew G. Knepley { 7124dde8bb0SMatthew G. Knepley DMSNES sdm; 7134e269d77SPeter Brune 7144dde8bb0SMatthew G. Knepley ierr = DMGetDMSNES(snes->dm, &sdm);CHKERRQ(ierr); 7154dde8bb0SMatthew G. Knepley if (!sdm->ops->computejacobian) { 7164dde8bb0SMatthew G. Knepley ierr = DMCopyDMSNES(dmsave, snes->dm);CHKERRQ(ierr); 7174dde8bb0SMatthew G. Knepley } 7184dde8bb0SMatthew G. Knepley } 7192b93b426SMatthew G. Knepley /* Compute the operators */ 720d1e9a80fSBarry Smith ierr = SNESComputeJacobian(snes,X,A,B);CHKERRQ(ierr); 7212b93b426SMatthew G. Knepley /* Put the previous context back */ 7228d359177SBarry Smith if (snes->dm != dmsave && jac == SNESComputeJacobianDefaultColor) { 7230298fd71SBarry Smith ierr = SNESSetJacobian(snes,NULL,NULL,jac,ctxsave);CHKERRQ(ierr); 7244e269d77SPeter Brune } 7254e269d77SPeter Brune 7262b93b426SMatthew G. Knepley if (Xnamed) {ierr = DMRestoreNamedGlobalVector(snes->dm,"SNESVecSol",&Xnamed);CHKERRQ(ierr);} 727dfe15315SJed Brown snes->dm = dmsave; 728caa4e7f2SJed Brown PetscFunctionReturn(0); 729caa4e7f2SJed Brown } 730caa4e7f2SJed Brown 7316cab3a1bSJed Brown /*@ 7326cab3a1bSJed Brown SNESSetUpMatrices - ensures that matrices are available for SNES, to be called by SNESSetUp_XXX() 7336cab3a1bSJed Brown 7346cab3a1bSJed Brown Collective 7356cab3a1bSJed Brown 7366cab3a1bSJed Brown Input Arguments: 7376cab3a1bSJed Brown . snes - snes to configure 7386cab3a1bSJed Brown 7396cab3a1bSJed Brown Level: developer 7406cab3a1bSJed Brown 7416cab3a1bSJed Brown .seealso: SNESSetUp() 7426cab3a1bSJed Brown @*/ 7436cab3a1bSJed Brown PetscErrorCode SNESSetUpMatrices(SNES snes) 7446cab3a1bSJed Brown { 7456cab3a1bSJed Brown PetscErrorCode ierr; 7466cab3a1bSJed Brown DM dm; 747942e3340SBarry Smith DMSNES sdm; 7486cab3a1bSJed Brown 7496cab3a1bSJed Brown PetscFunctionBegin; 7506cab3a1bSJed Brown ierr = SNESGetDM(snes,&dm);CHKERRQ(ierr); 751942e3340SBarry Smith ierr = DMGetDMSNES(dm,&sdm);CHKERRQ(ierr); 75258b371f3SBarry Smith if (!snes->jacobian && snes->mf) { 7536cab3a1bSJed Brown Mat J; 7546cab3a1bSJed Brown void *functx; 7556cab3a1bSJed Brown ierr = MatCreateSNESMF(snes,&J);CHKERRQ(ierr); 7566cab3a1bSJed Brown ierr = MatMFFDSetOptionsPrefix(J,((PetscObject)snes)->prefix);CHKERRQ(ierr); 7576cab3a1bSJed Brown ierr = MatSetFromOptions(J);CHKERRQ(ierr); 7580298fd71SBarry Smith ierr = SNESGetFunction(snes,NULL,NULL,&functx);CHKERRQ(ierr); 7599e5d0892SLisandro Dalcin ierr = SNESSetJacobian(snes,J,J,NULL,NULL);CHKERRQ(ierr); 7606cab3a1bSJed Brown ierr = MatDestroy(&J);CHKERRQ(ierr); 761caa4e7f2SJed Brown } else if (snes->mf_operator && !snes->jacobian_pre && !snes->jacobian) { 7626cab3a1bSJed Brown Mat J,B; 7636cab3a1bSJed Brown ierr = MatCreateSNESMF(snes,&J);CHKERRQ(ierr); 7646cab3a1bSJed Brown ierr = MatMFFDSetOptionsPrefix(J,((PetscObject)snes)->prefix);CHKERRQ(ierr); 7656cab3a1bSJed Brown ierr = MatSetFromOptions(J);CHKERRQ(ierr); 766b412c318SBarry Smith ierr = DMCreateMatrix(snes->dm,&B);CHKERRQ(ierr); 76706f20277SJed Brown /* sdm->computejacobian was already set to reach here */ 7680298fd71SBarry Smith ierr = SNESSetJacobian(snes,J,B,NULL,NULL);CHKERRQ(ierr); 7696cab3a1bSJed Brown ierr = MatDestroy(&J);CHKERRQ(ierr); 7706cab3a1bSJed Brown ierr = MatDestroy(&B);CHKERRQ(ierr); 771caa4e7f2SJed Brown } else if (!snes->jacobian_pre) { 7721ba9b98eSMatthew G. Knepley PetscDS prob; 7736cab3a1bSJed Brown Mat J, B; 7741ba9b98eSMatthew G. Knepley PetscBool hasPrec = PETSC_FALSE; 7751ba9b98eSMatthew G. Knepley 7766cab3a1bSJed Brown J = snes->jacobian; 7771ba9b98eSMatthew G. Knepley ierr = DMGetDS(dm, &prob);CHKERRQ(ierr); 7781ba9b98eSMatthew G. Knepley if (prob) {ierr = PetscDSHasJacobianPreconditioner(prob, &hasPrec);CHKERRQ(ierr);} 779ec9a985fSMatthew G. Knepley if (J) {ierr = PetscObjectReference((PetscObject) J);CHKERRQ(ierr);} 780ec9a985fSMatthew G. Knepley else if (hasPrec) {ierr = DMCreateMatrix(snes->dm, &J);CHKERRQ(ierr);} 781b412c318SBarry Smith ierr = DMCreateMatrix(snes->dm, &B);CHKERRQ(ierr); 7820298fd71SBarry Smith ierr = SNESSetJacobian(snes, J ? J : B, B, NULL, NULL);CHKERRQ(ierr); 7831ba9b98eSMatthew G. Knepley ierr = MatDestroy(&J);CHKERRQ(ierr); 7846cab3a1bSJed Brown ierr = MatDestroy(&B);CHKERRQ(ierr); 7856cab3a1bSJed Brown } 786caa4e7f2SJed Brown { 787caa4e7f2SJed Brown KSP ksp; 788caa4e7f2SJed Brown ierr = SNESGetKSP(snes,&ksp);CHKERRQ(ierr); 789caa4e7f2SJed Brown ierr = KSPSetComputeOperators(ksp,KSPComputeOperators_SNES,snes);CHKERRQ(ierr); 79016ebb321SJed Brown ierr = DMCoarsenHookAdd(snes->dm,DMCoarsenHook_SNESVecSol,DMRestrictHook_SNESVecSol,snes);CHKERRQ(ierr); 791caa4e7f2SJed Brown } 7926cab3a1bSJed Brown PetscFunctionReturn(0); 7936cab3a1bSJed Brown } 7946cab3a1bSJed Brown 795fde5950dSBarry Smith /*@C 796fde5950dSBarry Smith SNESMonitorSetFromOptions - Sets a monitor function and viewer appropriate for the type indicated by the user 797fde5950dSBarry Smith 798fde5950dSBarry Smith Collective on SNES 799fde5950dSBarry Smith 800fde5950dSBarry Smith Input Parameters: 801fde5950dSBarry Smith + snes - SNES object you wish to monitor 802fde5950dSBarry Smith . name - the monitor type one is seeking 803fde5950dSBarry Smith . help - message indicating what monitoring is done 804fde5950dSBarry Smith . manual - manual page for the monitor 805fde5950dSBarry Smith . monitor - the monitor function 806fde5950dSBarry 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 807fde5950dSBarry Smith 808fde5950dSBarry Smith Level: developer 809fde5950dSBarry Smith 810fde5950dSBarry Smith .seealso: PetscOptionsGetViewer(), PetscOptionsGetReal(), PetscOptionsHasName(), PetscOptionsGetString(), 811fde5950dSBarry Smith PetscOptionsGetIntArray(), PetscOptionsGetRealArray(), PetscOptionsBool() 812fde5950dSBarry Smith PetscOptionsInt(), PetscOptionsString(), PetscOptionsReal(), PetscOptionsBool(), 813fde5950dSBarry Smith PetscOptionsName(), PetscOptionsBegin(), PetscOptionsEnd(), PetscOptionsHead(), 814fde5950dSBarry Smith PetscOptionsStringArray(),PetscOptionsRealArray(), PetscOptionsScalar(), 815fde5950dSBarry Smith PetscOptionsBoolGroupBegin(), PetscOptionsBoolGroup(), PetscOptionsBoolGroupEnd(), 816fde5950dSBarry Smith PetscOptionsFList(), PetscOptionsEList() 817fde5950dSBarry Smith @*/ 818d43b4f6eSBarry Smith PetscErrorCode SNESMonitorSetFromOptions(SNES snes,const char name[],const char help[], const char manual[],PetscErrorCode (*monitor)(SNES,PetscInt,PetscReal,PetscViewerAndFormat*),PetscErrorCode (*monitorsetup)(SNES,PetscViewerAndFormat*)) 819fde5950dSBarry Smith { 820fde5950dSBarry Smith PetscErrorCode ierr; 821fde5950dSBarry Smith PetscViewer viewer; 822fde5950dSBarry Smith PetscViewerFormat format; 823fde5950dSBarry Smith PetscBool flg; 824fde5950dSBarry Smith 825fde5950dSBarry Smith PetscFunctionBegin; 82616413a6aSBarry Smith ierr = PetscOptionsGetViewer(PetscObjectComm((PetscObject)snes),((PetscObject) snes)->options,((PetscObject)snes)->prefix,name,&viewer,&format,&flg);CHKERRQ(ierr); 827fde5950dSBarry Smith if (flg) { 828d43b4f6eSBarry Smith PetscViewerAndFormat *vf; 829d43b4f6eSBarry Smith ierr = PetscViewerAndFormatCreate(viewer,format,&vf);CHKERRQ(ierr); 830d43b4f6eSBarry Smith ierr = PetscObjectDereference((PetscObject)viewer);CHKERRQ(ierr); 831fde5950dSBarry Smith if (monitorsetup) { 832d43b4f6eSBarry Smith ierr = (*monitorsetup)(snes,vf);CHKERRQ(ierr); 833fde5950dSBarry Smith } 834d43b4f6eSBarry Smith ierr = SNESMonitorSet(snes,(PetscErrorCode (*)(SNES,PetscInt,PetscReal,void*))monitor,vf,(PetscErrorCode (*)(void**))PetscViewerAndFormatDestroy);CHKERRQ(ierr); 835fde5950dSBarry Smith } 836fde5950dSBarry Smith PetscFunctionReturn(0); 837fde5950dSBarry Smith } 838fde5950dSBarry Smith 8399b94acceSBarry Smith /*@ 84094b7f48cSBarry Smith SNESSetFromOptions - Sets various SNES and KSP parameters from user options. 8419b94acceSBarry Smith 842c7afd0dbSLois Curfman McInnes Collective on SNES 843c7afd0dbSLois Curfman McInnes 8449b94acceSBarry Smith Input Parameter: 8459b94acceSBarry Smith . snes - the SNES context 8469b94acceSBarry Smith 84736851e7fSLois Curfman McInnes Options Database Keys: 848722329fbSBarry Smith + -snes_type <type> - newtonls, newtontr, ngmres, ncg, nrichardson, qn, vi, fas, SNESType for complete list 84982738288SBarry Smith . -snes_stol - convergence tolerance in terms of the norm 85082738288SBarry Smith of the change in the solution between steps 85170441072SBarry Smith . -snes_atol <abstol> - absolute tolerance of residual norm 852b39c3a46SLois Curfman McInnes . -snes_rtol <rtol> - relative decrease in tolerance norm from initial 853e4d06f11SPatrick Farrell . -snes_divergence_tolerance <divtol> - if the residual goes above divtol*rnorm0, exit with divergence 854be5caee7SBarry Smith . -snes_force_iteration <force> - force SNESSolve() to take at least one iteration 855b39c3a46SLois Curfman McInnes . -snes_max_it <max_it> - maximum number of iterations 856b39c3a46SLois Curfman McInnes . -snes_max_funcs <max_funcs> - maximum number of function evaluations 8574839bfe8SBarry Smith . -snes_max_fail <max_fail> - maximum number of line search failures allowed before stopping, default is none 858ddf469c8SBarry Smith . -snes_max_linear_solve_fail - number of linear solver failures before SNESSolve() stops 859a8054027SBarry Smith . -snes_lag_preconditioner <lag> - how often preconditioner is rebuilt (use -1 to never rebuild) 8603d5a8a6aSBarry Smith . -snes_lag_preconditioner_persists <true,false> - retains the -snes_lag_preconditioner information across multiple SNESSolve() 861e35cf81dSBarry Smith . -snes_lag_jacobian <lag> - how often Jacobian is rebuilt (use -1 to never rebuild) 8623d5a8a6aSBarry Smith . -snes_lag_jacobian_persists <true,false> - retains the -snes_lag_jacobian information across multiple SNESSolve() 863b39c3a46SLois Curfman McInnes . -snes_trtol <trtol> - trust region tolerance 8642492ecdbSBarry Smith . -snes_no_convergence_test - skip convergence test in nonlinear 86582738288SBarry Smith solver; hence iterations will continue until max_it 8661fbbfb26SLois Curfman McInnes or some other criterion is reached. Saves expense 86782738288SBarry Smith of convergence test 868fde5950dSBarry Smith . -snes_monitor [ascii][:filename][:viewer format] - prints residual norm at each iteration. if no filename given prints to stdout 869fde5950dSBarry Smith . -snes_monitor_solution [ascii binary draw][:filename][:viewer format] - plots solution at each iteration 870fde5950dSBarry Smith . -snes_monitor_residual [ascii binary draw][:filename][:viewer format] - plots residual (not its norm) at each iteration 871fde5950dSBarry Smith . -snes_monitor_solution_update [ascii binary draw][:filename][:viewer format] - plots update to solution at each iteration 8724619e776SBarry Smith . -snes_monitor_lg_residualnorm - plots residual norm at each iteration 873459f5d12SBarry Smith . -snes_monitor_lg_range - plots residual norm at each iteration 874e24b481bSBarry Smith . -snes_fd - use finite differences to compute Jacobian; very slow, only for testing 875e2e60de9SPeter Brune . -snes_fd_color - use finite differences with coloring to compute Jacobian 8765968eb51SBarry Smith . -snes_mf_ksp_monitor - if using matrix-free multiply then print h at each KSP iteration 877b5badacbSBarry Smith . -snes_converged_reason - print the reason for convergence/divergence after each solve 878e62ac41dSBarry Smith . -npc_snes_type <type> - the SNES type to use as a nonlinear preconditioner 879e62ac41dSBarry Smith . -snes_test_jacobian <optional threshold> - compare the user provided Jacobian with one computed via finite differences to check for errors. If a threshold is given, display only those entries whose difference is greater than the threshold. 880e62ac41dSBarry Smith - -snes_test_jacobian_view - display the user provided Jacobian, the finite difference Jacobian and the difference between them to help users detect the location of errors in the user provided Jacobian. 88182738288SBarry Smith 88282738288SBarry Smith Options Database for Eisenstat-Walker method: 883fa9f3622SBarry Smith + -snes_ksp_ew - use Eisenstat-Walker method for determining linear system convergence 8844b27c08aSLois Curfman McInnes . -snes_ksp_ew_version ver - version of Eisenstat-Walker method 88536851e7fSLois Curfman McInnes . -snes_ksp_ew_rtol0 <rtol0> - Sets rtol0 88636851e7fSLois Curfman McInnes . -snes_ksp_ew_rtolmax <rtolmax> - Sets rtolmax 88736851e7fSLois Curfman McInnes . -snes_ksp_ew_gamma <gamma> - Sets gamma 88836851e7fSLois Curfman McInnes . -snes_ksp_ew_alpha <alpha> - Sets alpha 88936851e7fSLois Curfman McInnes . -snes_ksp_ew_alpha2 <alpha2> - Sets alpha2 89036851e7fSLois Curfman McInnes - -snes_ksp_ew_threshold <threshold> - Sets threshold 89182738288SBarry Smith 89211ca99fdSLois Curfman McInnes Notes: 893ec5066bdSBarry Smith To see all options, run your program with the -help option or consult the users manual 894ec5066bdSBarry Smith 895ec5066bdSBarry Smith Notes: 896ec5066bdSBarry Smith SNES supports three approaches for computing (approximate) Jacobians: user provided via SNESSetJacobian(), matrix free, and computing explictly with 897ec5066bdSBarry Smith finite differences and coloring using MatFDColoring. It is also possible to use automatic differentiation and the MatFDColoring object. 89883e2fdc7SBarry Smith 89936851e7fSLois Curfman McInnes Level: beginner 90036851e7fSLois Curfman McInnes 901ec5066bdSBarry Smith .seealso: SNESSetOptionsPrefix(), SNESResetFromOptions(), SNES, SNESCreate() 9029b94acceSBarry Smith @*/ 9037087cfbeSBarry Smith PetscErrorCode SNESSetFromOptions(SNES snes) 9049b94acceSBarry Smith { 9058afaa268SBarry Smith PetscBool flg,pcset,persist,set; 906d8f46077SPeter Brune PetscInt i,indx,lag,grids; 90704d7464bSBarry Smith const char *deft = SNESNEWTONLS; 908649ef022SMatthew Knepley const char *convtests[] = {"default","skip","correct_pressure"}; 90985385478SLisandro Dalcin SNESKSPEW *kctx = NULL; 910e8105e01SRichard Katz char type[256], monfilename[PETSC_MAX_PATH_LEN]; 91185385478SLisandro Dalcin PetscErrorCode ierr; 912c40d0f55SPeter Brune PCSide pcside; 913a64e098fSPeter Brune const char *optionsprefix; 9149b94acceSBarry Smith 9153a40ed3dSBarry Smith PetscFunctionBegin; 9160700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 9170f51fdf8SToby Isaac ierr = SNESRegisterAll();CHKERRQ(ierr); 9183194b578SJed Brown ierr = PetscObjectOptionsBegin((PetscObject)snes);CHKERRQ(ierr); 919639ff905SBarry Smith if (((PetscObject)snes)->type_name) deft = ((PetscObject)snes)->type_name; 920a264d7a6SBarry Smith ierr = PetscOptionsFList("-snes_type","Nonlinear solver method","SNESSetType",SNESList,deft,type,256,&flg);CHKERRQ(ierr); 921d64ed03dSBarry Smith if (flg) { 922186905e3SBarry Smith ierr = SNESSetType(snes,type);CHKERRQ(ierr); 9237adad957SLisandro Dalcin } else if (!((PetscObject)snes)->type_name) { 924186905e3SBarry Smith ierr = SNESSetType(snes,deft);CHKERRQ(ierr); 925d64ed03dSBarry Smith } 92694ae4db5SBarry Smith ierr = PetscOptionsReal("-snes_stol","Stop if step length less than","SNESSetTolerances",snes->stol,&snes->stol,NULL);CHKERRQ(ierr); 92794ae4db5SBarry Smith ierr = PetscOptionsReal("-snes_atol","Stop if function norm less than","SNESSetTolerances",snes->abstol,&snes->abstol,NULL);CHKERRQ(ierr); 928186905e3SBarry Smith 92994ae4db5SBarry Smith ierr = PetscOptionsReal("-snes_rtol","Stop if decrease in function norm less than","SNESSetTolerances",snes->rtol,&snes->rtol,NULL);CHKERRQ(ierr); 930e4d06f11SPatrick Farrell ierr = PetscOptionsReal("-snes_divergence_tolerance","Stop if residual norm increases by this factor","SNESSetDivergenceTolerance",snes->divtol,&snes->divtol,NULL);CHKERRQ(ierr); 9310298fd71SBarry Smith ierr = PetscOptionsInt("-snes_max_it","Maximum iterations","SNESSetTolerances",snes->max_its,&snes->max_its,NULL);CHKERRQ(ierr); 9320298fd71SBarry Smith ierr = PetscOptionsInt("-snes_max_funcs","Maximum function evaluations","SNESSetTolerances",snes->max_funcs,&snes->max_funcs,NULL);CHKERRQ(ierr); 9330298fd71SBarry Smith ierr = PetscOptionsInt("-snes_max_fail","Maximum nonlinear step failures","SNESSetMaxNonlinearStepFailures",snes->maxFailures,&snes->maxFailures,NULL);CHKERRQ(ierr); 9340298fd71SBarry Smith ierr = PetscOptionsInt("-snes_max_linear_solve_fail","Maximum failures in linear solves allowed","SNESSetMaxLinearSolveFailures",snes->maxLinearSolveFailures,&snes->maxLinearSolveFailures,NULL);CHKERRQ(ierr); 9350298fd71SBarry Smith ierr = PetscOptionsBool("-snes_error_if_not_converged","Generate error if solver does not converge","SNESSetErrorIfNotConverged",snes->errorifnotconverged,&snes->errorifnotconverged,NULL);CHKERRQ(ierr); 9360d6f27a8SBarry Smith ierr = PetscOptionsBool("-snes_force_iteration","Force SNESSolve() to take at least one iteration","SNESSetForceIteration",snes->forceiteration,&snes->forceiteration,NULL);CHKERRQ(ierr); 937b351a90bSFande Kong ierr = PetscOptionsBool("-snes_check_jacobian_domain_error","Check Jacobian domain error after Jacobian evaluation","SNESCheckJacobianDomainError",snes->checkjacdomainerror,&snes->checkjacdomainerror,NULL);CHKERRQ(ierr); 93885385478SLisandro Dalcin 939a8054027SBarry Smith ierr = PetscOptionsInt("-snes_lag_preconditioner","How often to rebuild preconditioner","SNESSetLagPreconditioner",snes->lagpreconditioner,&lag,&flg);CHKERRQ(ierr); 940a8054027SBarry Smith if (flg) { 9413d5a8a6aSBarry Smith if (lag == -1) SETERRQ(PetscObjectComm((PetscObject)snes),PETSC_ERR_USER,"Cannot set the lag to -1 from the command line since the preconditioner must be built as least once, perhaps you mean -2"); 942a8054027SBarry Smith ierr = SNESSetLagPreconditioner(snes,lag);CHKERRQ(ierr); 943a8054027SBarry Smith } 9449590daa0SBarry Smith ierr = PetscOptionsBool("-snes_lag_preconditioner_persists","Preconditioner lagging through multiple SNES solves","SNESSetLagPreconditionerPersists",snes->lagjac_persist,&persist,&flg);CHKERRQ(ierr); 94537ec4e1aSPeter Brune if (flg) { 94637ec4e1aSPeter Brune ierr = SNESSetLagPreconditionerPersists(snes,persist);CHKERRQ(ierr); 94737ec4e1aSPeter Brune } 948e35cf81dSBarry Smith ierr = PetscOptionsInt("-snes_lag_jacobian","How often to rebuild Jacobian","SNESSetLagJacobian",snes->lagjacobian,&lag,&flg);CHKERRQ(ierr); 949e35cf81dSBarry Smith if (flg) { 9503d5a8a6aSBarry Smith if (lag == -1) SETERRQ(PetscObjectComm((PetscObject)snes),PETSC_ERR_USER,"Cannot set the lag to -1 from the command line since the Jacobian must be built as least once, perhaps you mean -2"); 951e35cf81dSBarry Smith ierr = SNESSetLagJacobian(snes,lag);CHKERRQ(ierr); 952e35cf81dSBarry Smith } 9539590daa0SBarry Smith ierr = PetscOptionsBool("-snes_lag_jacobian_persists","Jacobian lagging through multiple SNES solves","SNESSetLagJacobianPersists",snes->lagjac_persist,&persist,&flg);CHKERRQ(ierr); 95437ec4e1aSPeter Brune if (flg) { 95537ec4e1aSPeter Brune ierr = SNESSetLagJacobianPersists(snes,persist);CHKERRQ(ierr); 95637ec4e1aSPeter Brune } 95737ec4e1aSPeter Brune 958efd51863SBarry Smith ierr = PetscOptionsInt("-snes_grid_sequence","Use grid sequencing to generate initial guess","SNESSetGridSequence",snes->gridsequence,&grids,&flg);CHKERRQ(ierr); 959efd51863SBarry Smith if (flg) { 960efd51863SBarry Smith ierr = SNESSetGridSequence(snes,grids);CHKERRQ(ierr); 961efd51863SBarry Smith } 962a8054027SBarry Smith 963649ef022SMatthew Knepley ierr = PetscOptionsEList("-snes_convergence_test","Convergence test","SNESSetConvergenceTest",convtests,sizeof(convtests)/sizeof(char*),"default",&indx,&flg);CHKERRQ(ierr); 96485385478SLisandro Dalcin if (flg) { 96585385478SLisandro Dalcin switch (indx) { 9668d359177SBarry Smith case 0: ierr = SNESSetConvergenceTest(snes,SNESConvergedDefault,NULL,NULL);CHKERRQ(ierr); break; 967e2a6519dSDmitry Karpeev case 1: ierr = SNESSetConvergenceTest(snes,SNESConvergedSkip,NULL,NULL);CHKERRQ(ierr); break; 968649ef022SMatthew Knepley case 2: ierr = SNESSetConvergenceTest(snes,SNESConvergedCorrectPressure,NULL,NULL);CHKERRQ(ierr); break; 96985385478SLisandro Dalcin } 97085385478SLisandro Dalcin } 97185385478SLisandro Dalcin 972365a6726SPeter Brune ierr = PetscOptionsEList("-snes_norm_schedule","SNES Norm schedule","SNESSetNormSchedule",SNESNormSchedules,5,"function",&indx,&flg);CHKERRQ(ierr); 973365a6726SPeter Brune if (flg) { ierr = SNESSetNormSchedule(snes,(SNESNormSchedule)indx);CHKERRQ(ierr); } 974fdacfa88SPeter Brune 97547073ea2SPeter Brune ierr = PetscOptionsEList("-snes_function_type","SNES Norm schedule","SNESSetFunctionType",SNESFunctionTypes,2,"unpreconditioned",&indx,&flg);CHKERRQ(ierr); 97647073ea2SPeter Brune if (flg) { ierr = SNESSetFunctionType(snes,(SNESFunctionType)indx);CHKERRQ(ierr); } 977186905e3SBarry Smith 97885385478SLisandro Dalcin kctx = (SNESKSPEW*)snes->kspconvctx; 97985385478SLisandro Dalcin 9800298fd71SBarry Smith ierr = PetscOptionsBool("-snes_ksp_ew","Use Eisentat-Walker linear system convergence test","SNESKSPSetUseEW",snes->ksp_ewconv,&snes->ksp_ewconv,NULL);CHKERRQ(ierr); 981186905e3SBarry Smith 98294ae4db5SBarry Smith ierr = PetscOptionsInt("-snes_ksp_ew_version","Version 1, 2 or 3","SNESKSPSetParametersEW",kctx->version,&kctx->version,NULL);CHKERRQ(ierr); 98394ae4db5SBarry Smith ierr = PetscOptionsReal("-snes_ksp_ew_rtol0","0 <= rtol0 < 1","SNESKSPSetParametersEW",kctx->rtol_0,&kctx->rtol_0,NULL);CHKERRQ(ierr); 98494ae4db5SBarry Smith ierr = PetscOptionsReal("-snes_ksp_ew_rtolmax","0 <= rtolmax < 1","SNESKSPSetParametersEW",kctx->rtol_max,&kctx->rtol_max,NULL);CHKERRQ(ierr); 98594ae4db5SBarry Smith ierr = PetscOptionsReal("-snes_ksp_ew_gamma","0 <= gamma <= 1","SNESKSPSetParametersEW",kctx->gamma,&kctx->gamma,NULL);CHKERRQ(ierr); 98694ae4db5SBarry Smith ierr = PetscOptionsReal("-snes_ksp_ew_alpha","1 < alpha <= 2","SNESKSPSetParametersEW",kctx->alpha,&kctx->alpha,NULL);CHKERRQ(ierr); 98794ae4db5SBarry Smith ierr = PetscOptionsReal("-snes_ksp_ew_alpha2","alpha2","SNESKSPSetParametersEW",kctx->alpha2,&kctx->alpha2,NULL);CHKERRQ(ierr); 98894ae4db5SBarry Smith ierr = PetscOptionsReal("-snes_ksp_ew_threshold","0 < threshold < 1","SNESKSPSetParametersEW",kctx->threshold,&kctx->threshold,NULL);CHKERRQ(ierr); 989186905e3SBarry Smith 99090d69ab7SBarry Smith flg = PETSC_FALSE; 9918afaa268SBarry Smith ierr = PetscOptionsBool("-snes_monitor_cancel","Remove all monitors","SNESMonitorCancel",flg,&flg,&set);CHKERRQ(ierr); 9928afaa268SBarry Smith if (set && flg) {ierr = SNESMonitorCancel(snes);CHKERRQ(ierr);} 993eabae89aSBarry Smith 994fde5950dSBarry Smith ierr = SNESMonitorSetFromOptions(snes,"-snes_monitor","Monitor norm of function","SNESMonitorDefault",SNESMonitorDefault,NULL);CHKERRQ(ierr); 995fde5950dSBarry Smith ierr = SNESMonitorSetFromOptions(snes,"-snes_monitor_short","Monitor norm of function with fewer digits","SNESMonitorDefaultShort",SNESMonitorDefaultShort,NULL);CHKERRQ(ierr); 996fde5950dSBarry Smith ierr = SNESMonitorSetFromOptions(snes,"-snes_monitor_range","Monitor range of elements of function","SNESMonitorRange",SNESMonitorRange,NULL);CHKERRQ(ierr); 997eabae89aSBarry Smith 998fde5950dSBarry Smith ierr = SNESMonitorSetFromOptions(snes,"-snes_monitor_ratio","Monitor ratios of the norm of function for consecutive steps","SNESMonitorRatio",SNESMonitorRatio,SNESMonitorRatioSetUp);CHKERRQ(ierr); 999fde5950dSBarry Smith ierr = SNESMonitorSetFromOptions(snes,"-snes_monitor_field","Monitor norm of function (split into fields)","SNESMonitorDefaultField",SNESMonitorDefaultField,NULL);CHKERRQ(ierr); 1000fde5950dSBarry Smith ierr = SNESMonitorSetFromOptions(snes,"-snes_monitor_solution","View solution at each iteration","SNESMonitorSolution",SNESMonitorSolution,NULL);CHKERRQ(ierr); 1001fde5950dSBarry Smith ierr = SNESMonitorSetFromOptions(snes,"-snes_monitor_solution_update","View correction at each iteration","SNESMonitorSolutionUpdate",SNESMonitorSolutionUpdate,NULL);CHKERRQ(ierr); 1002fde5950dSBarry Smith ierr = SNESMonitorSetFromOptions(snes,"-snes_monitor_residual","View residual at each iteration","SNESMonitorResidual",SNESMonitorResidual,NULL);CHKERRQ(ierr); 1003fde5950dSBarry Smith ierr = SNESMonitorSetFromOptions(snes,"-snes_monitor_jacupdate_spectrum","Print the change in the spectrum of the Jacobian","SNESMonitorJacUpdateSpectrum",SNESMonitorJacUpdateSpectrum,NULL);CHKERRQ(ierr); 1004fde5950dSBarry Smith ierr = SNESMonitorSetFromOptions(snes,"-snes_monitor_fields","Monitor norm of function per field","SNESMonitorSet",SNESMonitorFields,NULL);CHKERRQ(ierr); 10052db13446SMatthew G. Knepley 1006589a23caSBarry Smith ierr = PetscOptionsString("-snes_monitor_python","Use Python function","SNESMonitorSet",NULL,monfilename,sizeof(monfilename),&flg);CHKERRQ(ierr); 10075180491cSLisandro Dalcin if (flg) {ierr = PetscPythonMonitorSet((PetscObject)snes,monfilename);CHKERRQ(ierr);} 10085180491cSLisandro Dalcin 100990d69ab7SBarry Smith flg = PETSC_FALSE; 10100298fd71SBarry Smith ierr = PetscOptionsBool("-snes_monitor_lg_range","Plot function range at each iteration","SNESMonitorLGRange",flg,&flg,NULL);CHKERRQ(ierr); 1011459f5d12SBarry Smith if (flg) { 1012459f5d12SBarry Smith PetscViewer ctx; 1013e24b481bSBarry Smith 10146ba87a44SLisandro Dalcin ierr = PetscViewerDrawOpen(PetscObjectComm((PetscObject)snes),NULL,NULL,PETSC_DECIDE,PETSC_DECIDE,400,300,&ctx);CHKERRQ(ierr); 1015459f5d12SBarry Smith ierr = SNESMonitorSet(snes,SNESMonitorLGRange,ctx,(PetscErrorCode (*)(void**))PetscViewerDestroy);CHKERRQ(ierr); 1016459f5d12SBarry Smith } 10172e7541e6SPeter Brune 101890d69ab7SBarry Smith flg = PETSC_FALSE; 1019c4421ceaSFande Kong ierr = PetscOptionsBool("-snes_converged_reason_view_cancel","Remove all converged reason viewers","SNESConvergedReasonViewCancel",flg,&flg,&set);CHKERRQ(ierr); 1020c4421ceaSFande Kong if (set && flg) {ierr = SNESConvergedReasonViewCancel(snes);CHKERRQ(ierr);} 1021c4421ceaSFande Kong 1022c4421ceaSFande Kong flg = PETSC_FALSE; 10238d359177SBarry Smith ierr = PetscOptionsBool("-snes_fd","Use finite differences (slow) to compute Jacobian","SNESComputeJacobianDefault",flg,&flg,NULL);CHKERRQ(ierr); 10244b27c08aSLois Curfman McInnes if (flg) { 10256cab3a1bSJed Brown void *functx; 1026b1f624c7SBarry Smith DM dm; 1027b1f624c7SBarry Smith DMSNES sdm; 1028b1f624c7SBarry Smith ierr = SNESGetDM(snes,&dm);CHKERRQ(ierr); 1029b1f624c7SBarry Smith ierr = DMGetDMSNES(dm,&sdm);CHKERRQ(ierr); 1030b1f624c7SBarry Smith sdm->jacobianctx = NULL; 10310298fd71SBarry Smith ierr = SNESGetFunction(snes,NULL,NULL,&functx);CHKERRQ(ierr); 10328d359177SBarry Smith ierr = SNESSetJacobian(snes,snes->jacobian,snes->jacobian_pre,SNESComputeJacobianDefault,functx);CHKERRQ(ierr); 1033ae15b995SBarry Smith ierr = PetscInfo(snes,"Setting default finite difference Jacobian matrix\n");CHKERRQ(ierr); 10349b94acceSBarry Smith } 1035639f9d9dSBarry Smith 103644848bc4SPeter Brune flg = PETSC_FALSE; 10378d359177SBarry Smith ierr = PetscOptionsBool("-snes_fd_function","Use finite differences (slow) to compute function from user objective","SNESObjectiveComputeFunctionDefaultFD",flg,&flg,NULL);CHKERRQ(ierr); 103897584545SPeter Brune if (flg) { 10398d359177SBarry Smith ierr = SNESSetFunction(snes,NULL,SNESObjectiveComputeFunctionDefaultFD,NULL);CHKERRQ(ierr); 104097584545SPeter Brune } 104197584545SPeter Brune 104297584545SPeter Brune flg = PETSC_FALSE; 10438d359177SBarry Smith ierr = PetscOptionsBool("-snes_fd_color","Use finite differences with coloring to compute Jacobian","SNESComputeJacobianDefaultColor",flg,&flg,NULL);CHKERRQ(ierr); 104444848bc4SPeter Brune if (flg) { 1045c52e227fSPeter Brune DM dm; 1046c52e227fSPeter Brune DMSNES sdm; 1047c52e227fSPeter Brune ierr = SNESGetDM(snes,&dm);CHKERRQ(ierr); 1048aace71b7SPeter Brune ierr = DMGetDMSNES(dm,&sdm);CHKERRQ(ierr); 1049aace71b7SPeter Brune sdm->jacobianctx = NULL; 10509e5d0892SLisandro Dalcin ierr = SNESSetJacobian(snes,snes->jacobian,snes->jacobian_pre,SNESComputeJacobianDefaultColor,NULL);CHKERRQ(ierr); 105144848bc4SPeter Brune ierr = PetscInfo(snes,"Setting default finite difference coloring Jacobian matrix\n");CHKERRQ(ierr); 105244848bc4SPeter Brune } 105344848bc4SPeter Brune 1054aa3661deSLisandro Dalcin flg = PETSC_FALSE; 1055f871313dSBarry 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); 1056d8f46077SPeter Brune if (flg && snes->mf_operator) { 1057a8248277SBarry Smith snes->mf_operator = PETSC_TRUE; 1058d8f46077SPeter Brune snes->mf = PETSC_TRUE; 1059a8248277SBarry Smith } 1060aa3661deSLisandro Dalcin flg = PETSC_FALSE; 1061f871313dSBarry Smith ierr = PetscOptionsBool("-snes_mf","Use a Matrix-Free Jacobian with no preconditioner matrix","SNESSetUseMatrixFree",PETSC_FALSE,&snes->mf,&flg);CHKERRQ(ierr); 1062d8f46077SPeter Brune if (!flg && snes->mf_operator) snes->mf = PETSC_TRUE; 10639e5d0892SLisandro Dalcin ierr = PetscOptionsInt("-snes_mf_version","Matrix-Free routines version 1 or 2","None",snes->mf_version,&snes->mf_version,NULL);CHKERRQ(ierr); 1064d28543b3SPeter Brune 1065c40d0f55SPeter Brune flg = PETSC_FALSE; 1066be95d8f1SBarry Smith ierr = SNESGetNPCSide(snes,&pcside);CHKERRQ(ierr); 1067be95d8f1SBarry Smith ierr = PetscOptionsEnum("-snes_npc_side","SNES nonlinear preconditioner side","SNESSetNPCSide",PCSides,(PetscEnum)pcside,(PetscEnum*)&pcside,&flg);CHKERRQ(ierr); 1068be95d8f1SBarry Smith if (flg) {ierr = SNESSetNPCSide(snes,pcside);CHKERRQ(ierr);} 1069c40d0f55SPeter Brune 1070e04113cfSBarry Smith #if defined(PETSC_HAVE_SAWS) 10718a70d858SHong Zhang /* 10728a70d858SHong Zhang Publish convergence information using SAWs 10738a70d858SHong Zhang */ 10748a70d858SHong Zhang flg = PETSC_FALSE; 10758a70d858SHong Zhang ierr = PetscOptionsBool("-snes_monitor_saws","Publish SNES progress using SAWs","SNESMonitorSet",flg,&flg,NULL);CHKERRQ(ierr); 10768a70d858SHong Zhang if (flg) { 10778a70d858SHong Zhang void *ctx; 10788a70d858SHong Zhang ierr = SNESMonitorSAWsCreate(snes,&ctx);CHKERRQ(ierr); 10798a70d858SHong Zhang ierr = SNESMonitorSet(snes,SNESMonitorSAWs,ctx,SNESMonitorSAWsDestroy);CHKERRQ(ierr); 10808a70d858SHong Zhang } 10818a70d858SHong Zhang #endif 10828a70d858SHong Zhang #if defined(PETSC_HAVE_SAWS) 1083b90c6cbeSBarry Smith { 1084b90c6cbeSBarry Smith PetscBool set; 1085b90c6cbeSBarry Smith flg = PETSC_FALSE; 10868a70d858SHong Zhang ierr = PetscOptionsBool("-snes_saws_block","Block for SAWs at end of SNESSolve","PetscObjectSAWsBlock",((PetscObject)snes)->amspublishblock,&flg,&set);CHKERRQ(ierr); 1087b90c6cbeSBarry Smith if (set) { 1088e04113cfSBarry Smith ierr = PetscObjectSAWsSetBlock((PetscObject)snes,flg);CHKERRQ(ierr); 1089b90c6cbeSBarry Smith } 1090b90c6cbeSBarry Smith } 1091b90c6cbeSBarry Smith #endif 1092b90c6cbeSBarry Smith 109376b2cf59SMatthew Knepley for (i = 0; i < numberofsetfromoptions; i++) { 109476b2cf59SMatthew Knepley ierr = (*othersetfromoptions[i])(snes);CHKERRQ(ierr); 109576b2cf59SMatthew Knepley } 109676b2cf59SMatthew Knepley 1097e7788613SBarry Smith if (snes->ops->setfromoptions) { 1098e55864a3SBarry Smith ierr = (*snes->ops->setfromoptions)(PetscOptionsObject,snes);CHKERRQ(ierr); 1099639f9d9dSBarry Smith } 11005d973c19SBarry Smith 11015d973c19SBarry Smith /* process any options handlers added with PetscObjectAddOptionsHandler() */ 11020633abcbSJed Brown ierr = PetscObjectProcessOptionsHandlers(PetscOptionsObject,(PetscObject)snes);CHKERRQ(ierr); 1103b0a32e0cSBarry Smith ierr = PetscOptionsEnd();CHKERRQ(ierr); 11044bbc92c1SBarry Smith 1105d8d34be6SBarry Smith if (snes->linesearch) { 11067601faf0SJed Brown ierr = SNESGetLineSearch(snes, &snes->linesearch);CHKERRQ(ierr); 1107f1c6b773SPeter Brune ierr = SNESLineSearchSetFromOptions(snes->linesearch);CHKERRQ(ierr); 1108d8d34be6SBarry Smith } 11099e764e56SPeter Brune 11106aa5e7e9SBarry Smith if (snes->usesksp) { 11116991f827SBarry Smith if (!snes->ksp) {ierr = SNESGetKSP(snes,&snes->ksp);CHKERRQ(ierr);} 11126991f827SBarry Smith ierr = KSPSetOperators(snes->ksp,snes->jacobian,snes->jacobian_pre);CHKERRQ(ierr); 11136991f827SBarry Smith ierr = KSPSetFromOptions(snes->ksp);CHKERRQ(ierr); 11146aa5e7e9SBarry Smith } 11156991f827SBarry Smith 1116b5badacbSBarry Smith /* if user has set the SNES NPC type via options database, create it. */ 111751e86f29SPeter Brune ierr = SNESGetOptionsPrefix(snes, &optionsprefix);CHKERRQ(ierr); 1118c5929fdfSBarry Smith ierr = PetscOptionsHasName(((PetscObject)snes)->options,optionsprefix, "-npc_snes_type", &pcset);CHKERRQ(ierr); 1119efd4aadfSBarry Smith if (pcset && (!snes->npc)) { 1120efd4aadfSBarry Smith ierr = SNESGetNPC(snes, &snes->npc);CHKERRQ(ierr); 112151e86f29SPeter Brune } 1122b5badacbSBarry Smith if (snes->npc) { 1123b5badacbSBarry Smith ierr = SNESSetFromOptions(snes->npc);CHKERRQ(ierr); 1124b5badacbSBarry Smith } 1125b3cd9a81SMatthew G. Knepley snes->setfromoptionscalled++; 1126b3cd9a81SMatthew G. Knepley PetscFunctionReturn(0); 1127b3cd9a81SMatthew G. Knepley } 1128b3cd9a81SMatthew G. Knepley 1129b3cd9a81SMatthew G. Knepley /*@ 1130b3cd9a81SMatthew G. Knepley SNESResetFromOptions - Sets various SNES and KSP parameters from user options ONLY if the SNES was previously set from options 1131b3cd9a81SMatthew G. Knepley 1132b3cd9a81SMatthew G. Knepley Collective on SNES 1133b3cd9a81SMatthew G. Knepley 1134b3cd9a81SMatthew G. Knepley Input Parameter: 1135b3cd9a81SMatthew G. Knepley . snes - the SNES context 1136b3cd9a81SMatthew G. Knepley 1137b3cd9a81SMatthew G. Knepley Level: beginner 1138b3cd9a81SMatthew G. Knepley 1139b3cd9a81SMatthew G. Knepley .seealso: SNESSetFromOptions(), SNESSetOptionsPrefix() 1140b3cd9a81SMatthew G. Knepley @*/ 1141b3cd9a81SMatthew G. Knepley PetscErrorCode SNESResetFromOptions(SNES snes) 1142b3cd9a81SMatthew G. Knepley { 1143b3cd9a81SMatthew G. Knepley PetscErrorCode ierr; 1144b3cd9a81SMatthew G. Knepley 1145b3cd9a81SMatthew G. Knepley PetscFunctionBegin; 1146b3cd9a81SMatthew G. Knepley if (snes->setfromoptionscalled) {ierr = SNESSetFromOptions(snes);CHKERRQ(ierr);} 11473a40ed3dSBarry Smith PetscFunctionReturn(0); 11489b94acceSBarry Smith } 11499b94acceSBarry Smith 1150bb9467b5SJed Brown /*@C 1151d25893d9SBarry Smith SNESSetComputeApplicationContext - Sets an optional function to compute a user-defined context for 1152d25893d9SBarry Smith the nonlinear solvers. 1153d25893d9SBarry Smith 1154d25893d9SBarry Smith Logically Collective on SNES 1155d25893d9SBarry Smith 1156d25893d9SBarry Smith Input Parameters: 1157d25893d9SBarry Smith + snes - the SNES context 1158d25893d9SBarry Smith . compute - function to compute the context 1159d25893d9SBarry Smith - destroy - function to destroy the context 1160d25893d9SBarry Smith 1161d25893d9SBarry Smith Level: intermediate 1162d25893d9SBarry Smith 1163bb9467b5SJed Brown Notes: 1164bb9467b5SJed Brown This function is currently not available from Fortran. 1165bb9467b5SJed Brown 1166d25893d9SBarry Smith .seealso: SNESGetApplicationContext(), SNESSetComputeApplicationContext(), SNESGetApplicationContext() 1167d25893d9SBarry Smith @*/ 1168d25893d9SBarry Smith PetscErrorCode SNESSetComputeApplicationContext(SNES snes,PetscErrorCode (*compute)(SNES,void**),PetscErrorCode (*destroy)(void**)) 1169d25893d9SBarry Smith { 1170d25893d9SBarry Smith PetscFunctionBegin; 1171d25893d9SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 1172d25893d9SBarry Smith snes->ops->usercompute = compute; 1173d25893d9SBarry Smith snes->ops->userdestroy = destroy; 1174d25893d9SBarry Smith PetscFunctionReturn(0); 1175d25893d9SBarry Smith } 1176a847f771SSatish Balay 1177b07ff414SBarry Smith /*@ 11789b94acceSBarry Smith SNESSetApplicationContext - Sets the optional user-defined context for 11799b94acceSBarry Smith the nonlinear solvers. 11809b94acceSBarry Smith 11813f9fe445SBarry Smith Logically Collective on SNES 1182fee21e36SBarry Smith 1183c7afd0dbSLois Curfman McInnes Input Parameters: 1184c7afd0dbSLois Curfman McInnes + snes - the SNES context 1185c7afd0dbSLois Curfman McInnes - usrP - optional user context 1186c7afd0dbSLois Curfman McInnes 118736851e7fSLois Curfman McInnes Level: intermediate 118836851e7fSLois Curfman McInnes 118995452b02SPatrick Sanan Fortran Notes: 119095452b02SPatrick Sanan To use this from Fortran you must write a Fortran interface definition for this 1191daf670e6SBarry Smith function that tells Fortran the Fortran derived data type that you are passing in as the ctx argument. 1192daf670e6SBarry Smith 1193ecaffddaSVictor Eijkhout .seealso: SNESGetApplicationContext() 11949b94acceSBarry Smith @*/ 11957087cfbeSBarry Smith PetscErrorCode SNESSetApplicationContext(SNES snes,void *usrP) 11969b94acceSBarry Smith { 11971b2093e4SBarry Smith PetscErrorCode ierr; 1198b07ff414SBarry Smith KSP ksp; 11991b2093e4SBarry Smith 12003a40ed3dSBarry Smith PetscFunctionBegin; 12010700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 1202b07ff414SBarry Smith ierr = SNESGetKSP(snes,&ksp);CHKERRQ(ierr); 1203b07ff414SBarry Smith ierr = KSPSetApplicationContext(ksp,usrP);CHKERRQ(ierr); 12049b94acceSBarry Smith snes->user = usrP; 12053a40ed3dSBarry Smith PetscFunctionReturn(0); 12069b94acceSBarry Smith } 120774679c65SBarry Smith 1208b07ff414SBarry Smith /*@ 12099b94acceSBarry Smith SNESGetApplicationContext - Gets the user-defined context for the 12109b94acceSBarry Smith nonlinear solvers. 12119b94acceSBarry Smith 1212c7afd0dbSLois Curfman McInnes Not Collective 1213c7afd0dbSLois Curfman McInnes 12149b94acceSBarry Smith Input Parameter: 12159b94acceSBarry Smith . snes - SNES context 12169b94acceSBarry Smith 12179b94acceSBarry Smith Output Parameter: 12189b94acceSBarry Smith . usrP - user context 12199b94acceSBarry Smith 122095452b02SPatrick Sanan Fortran Notes: 122195452b02SPatrick Sanan To use this from Fortran you must write a Fortran interface definition for this 1222daf670e6SBarry Smith function that tells Fortran the Fortran derived data type that you are passing in as the ctx argument. 1223daf670e6SBarry Smith 122436851e7fSLois Curfman McInnes Level: intermediate 122536851e7fSLois Curfman McInnes 12269b94acceSBarry Smith .seealso: SNESSetApplicationContext() 12279b94acceSBarry Smith @*/ 1228e71120c6SJed Brown PetscErrorCode SNESGetApplicationContext(SNES snes,void *usrP) 12299b94acceSBarry Smith { 12303a40ed3dSBarry Smith PetscFunctionBegin; 12310700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 1232e71120c6SJed Brown *(void**)usrP = snes->user; 12333a40ed3dSBarry Smith PetscFunctionReturn(0); 12349b94acceSBarry Smith } 123574679c65SBarry Smith 12369b94acceSBarry Smith /*@ 1237ec5066bdSBarry Smith SNESSetUseMatrixFree - indicates that SNES should use matrix free finite difference matrix vector products internally to apply the Jacobian. 12383565c898SBarry Smith 12393565c898SBarry Smith Collective on SNES 12403565c898SBarry Smith 12413565c898SBarry Smith Input Parameters: 12423565c898SBarry Smith + snes - SNES context 12434ddffce6SLisandro Dalcin . mf_operator - use matrix-free only for the Amat used by SNESSetJacobian(), this means the user provided Pmat will continue to be used 12444ddffce6SLisandro Dalcin - mf - use matrix-free for both the Amat and Pmat used by SNESSetJacobian(), both the Amat and Pmat set in SNESSetJacobian() will be ignored 12453565c898SBarry Smith 12463565c898SBarry Smith Options Database: 12473565c898SBarry Smith + -snes_mf - use matrix free for both the mat and pmat operator 1248ec5066bdSBarry Smith . -snes_mf_operator - use matrix free only for the mat operator 1249ec5066bdSBarry Smith . -snes_fd_color - compute the Jacobian via coloring and finite differences. 1250ec5066bdSBarry Smith - -snes_fd - compute the Jacobian via finite differences (slow) 12513565c898SBarry Smith 12523565c898SBarry Smith Level: intermediate 12533565c898SBarry Smith 1254ec5066bdSBarry Smith Notes: 1255ec5066bdSBarry Smith SNES supports three approaches for computing (approximate) Jacobians: user provided via SNESSetJacobian(), matrix free, and computing explictly with 1256ec5066bdSBarry Smith finite differences and coloring using MatFDColoring. It is also possible to use automatic differentiation and the MatFDColoring object. 1257ec5066bdSBarry Smith 1258ec5066bdSBarry Smith .seealso: SNESGetUseMatrixFree(), MatCreateSNESMF(), SNESComputeJacobianDefaultColor() 12593565c898SBarry Smith @*/ 12603565c898SBarry Smith PetscErrorCode SNESSetUseMatrixFree(SNES snes,PetscBool mf_operator,PetscBool mf) 12613565c898SBarry Smith { 12623565c898SBarry Smith PetscFunctionBegin; 12633565c898SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 126488b4c220SStefano Zampini PetscValidLogicalCollectiveBool(snes,mf_operator,2); 126588b4c220SStefano Zampini PetscValidLogicalCollectiveBool(snes,mf,3); 12664ddffce6SLisandro Dalcin snes->mf = mf_operator ? PETSC_TRUE : mf; 12673565c898SBarry Smith snes->mf_operator = mf_operator; 12683565c898SBarry Smith PetscFunctionReturn(0); 12693565c898SBarry Smith } 12703565c898SBarry Smith 12713565c898SBarry Smith /*@ 1272ec5066bdSBarry Smith SNESGetUseMatrixFree - indicates if the SNES uses matrix free finite difference matrix vector products to apply the Jacobian. 12733565c898SBarry Smith 12743565c898SBarry Smith Collective on SNES 12753565c898SBarry Smith 12763565c898SBarry Smith Input Parameter: 12773565c898SBarry Smith . snes - SNES context 12783565c898SBarry Smith 12793565c898SBarry Smith Output Parameters: 12804ddffce6SLisandro Dalcin + mf_operator - use matrix-free only for the Amat used by SNESSetJacobian(), this means the user provided Pmat will continue to be used 12814ddffce6SLisandro Dalcin - mf - use matrix-free for both the Amat and Pmat used by SNESSetJacobian(), both the Amat and Pmat set in SNESSetJacobian() will be ignored 12823565c898SBarry Smith 12833565c898SBarry Smith Options Database: 12843565c898SBarry Smith + -snes_mf - use matrix free for both the mat and pmat operator 12853565c898SBarry Smith - -snes_mf_operator - use matrix free only for the mat operator 12863565c898SBarry Smith 12873565c898SBarry Smith Level: intermediate 12883565c898SBarry Smith 12893565c898SBarry Smith .seealso: SNESSetUseMatrixFree(), MatCreateSNESMF() 12903565c898SBarry Smith @*/ 12913565c898SBarry Smith PetscErrorCode SNESGetUseMatrixFree(SNES snes,PetscBool *mf_operator,PetscBool *mf) 12923565c898SBarry Smith { 12933565c898SBarry Smith PetscFunctionBegin; 12943565c898SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 12953565c898SBarry Smith if (mf) *mf = snes->mf; 12963565c898SBarry Smith if (mf_operator) *mf_operator = snes->mf_operator; 12973565c898SBarry Smith PetscFunctionReturn(0); 12983565c898SBarry Smith } 12993565c898SBarry Smith 13003565c898SBarry Smith /*@ 1301c8228a4eSBarry Smith SNESGetIterationNumber - Gets the number of nonlinear iterations completed 1302c8228a4eSBarry Smith at this time. 13039b94acceSBarry Smith 1304c7afd0dbSLois Curfman McInnes Not Collective 1305c7afd0dbSLois Curfman McInnes 13069b94acceSBarry Smith Input Parameter: 13079b94acceSBarry Smith . snes - SNES context 13089b94acceSBarry Smith 13099b94acceSBarry Smith Output Parameter: 13109b94acceSBarry Smith . iter - iteration number 13119b94acceSBarry Smith 1312c8228a4eSBarry Smith Notes: 1313c8228a4eSBarry Smith For example, during the computation of iteration 2 this would return 1. 1314c8228a4eSBarry Smith 1315c8228a4eSBarry Smith This is useful for using lagged Jacobians (where one does not recompute the 131608405cd6SLois Curfman McInnes Jacobian at each SNES iteration). For example, the code 131708405cd6SLois Curfman McInnes .vb 131808405cd6SLois Curfman McInnes ierr = SNESGetIterationNumber(snes,&it); 131908405cd6SLois Curfman McInnes if (!(it % 2)) { 132008405cd6SLois Curfman McInnes [compute Jacobian here] 132108405cd6SLois Curfman McInnes } 132208405cd6SLois Curfman McInnes .ve 1323c8228a4eSBarry Smith can be used in your ComputeJacobian() function to cause the Jacobian to be 132408405cd6SLois Curfman McInnes recomputed every second SNES iteration. 1325c8228a4eSBarry Smith 1326c04deec6SBarry Smith After the SNES solve is complete this will return the number of nonlinear iterations used. 1327c04deec6SBarry Smith 132836851e7fSLois Curfman McInnes Level: intermediate 132936851e7fSLois Curfman McInnes 133071dbe336SPeter Brune .seealso: SNESGetLinearSolveIterations() 13319b94acceSBarry Smith @*/ 13327087cfbeSBarry Smith PetscErrorCode SNESGetIterationNumber(SNES snes,PetscInt *iter) 13339b94acceSBarry Smith { 13343a40ed3dSBarry Smith PetscFunctionBegin; 13350700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 13364482741eSBarry Smith PetscValidIntPointer(iter,2); 13379b94acceSBarry Smith *iter = snes->iter; 13383a40ed3dSBarry Smith PetscFunctionReturn(0); 13399b94acceSBarry Smith } 134074679c65SBarry Smith 1341360c497dSPeter Brune /*@ 1342360c497dSPeter Brune SNESSetIterationNumber - Sets the current iteration number. 1343360c497dSPeter Brune 1344360c497dSPeter Brune Not Collective 1345360c497dSPeter Brune 1346360c497dSPeter Brune Input Parameter: 1347a2b725a8SWilliam Gropp + snes - SNES context 1348a2b725a8SWilliam Gropp - iter - iteration number 1349360c497dSPeter Brune 1350360c497dSPeter Brune Level: developer 1351360c497dSPeter Brune 135271dbe336SPeter Brune .seealso: SNESGetLinearSolveIterations() 1353360c497dSPeter Brune @*/ 1354360c497dSPeter Brune PetscErrorCode SNESSetIterationNumber(SNES snes,PetscInt iter) 1355360c497dSPeter Brune { 1356360c497dSPeter Brune PetscErrorCode ierr; 1357360c497dSPeter Brune 1358360c497dSPeter Brune PetscFunctionBegin; 1359360c497dSPeter Brune PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 1360e04113cfSBarry Smith ierr = PetscObjectSAWsTakeAccess((PetscObject)snes);CHKERRQ(ierr); 1361360c497dSPeter Brune snes->iter = iter; 1362e04113cfSBarry Smith ierr = PetscObjectSAWsGrantAccess((PetscObject)snes);CHKERRQ(ierr); 1363360c497dSPeter Brune PetscFunctionReturn(0); 1364360c497dSPeter Brune } 1365360c497dSPeter Brune 13669b94acceSBarry Smith /*@ 1367b850b91aSLisandro Dalcin SNESGetNonlinearStepFailures - Gets the number of unsuccessful steps 13689b94acceSBarry Smith attempted by the nonlinear solver. 13699b94acceSBarry Smith 1370c7afd0dbSLois Curfman McInnes Not Collective 1371c7afd0dbSLois Curfman McInnes 13729b94acceSBarry Smith Input Parameter: 13739b94acceSBarry Smith . snes - SNES context 13749b94acceSBarry Smith 13759b94acceSBarry Smith Output Parameter: 13769b94acceSBarry Smith . nfails - number of unsuccessful steps attempted 13779b94acceSBarry Smith 1378c96a6f78SLois Curfman McInnes Notes: 1379c96a6f78SLois Curfman McInnes This counter is reset to zero for each successive call to SNESSolve(). 1380c96a6f78SLois Curfman McInnes 138136851e7fSLois Curfman McInnes Level: intermediate 138236851e7fSLois Curfman McInnes 1383e1c61ce8SBarry Smith .seealso: SNESGetMaxLinearSolveFailures(), SNESGetLinearSolveIterations(), SNESSetMaxLinearSolveFailures(), SNESGetLinearSolveFailures(), 138458ebbce7SBarry Smith SNESSetMaxNonlinearStepFailures(), SNESGetMaxNonlinearStepFailures() 13859b94acceSBarry Smith @*/ 13867087cfbeSBarry Smith PetscErrorCode SNESGetNonlinearStepFailures(SNES snes,PetscInt *nfails) 13879b94acceSBarry Smith { 13883a40ed3dSBarry Smith PetscFunctionBegin; 13890700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 13904482741eSBarry Smith PetscValidIntPointer(nfails,2); 139150ffb88aSMatthew Knepley *nfails = snes->numFailures; 139250ffb88aSMatthew Knepley PetscFunctionReturn(0); 139350ffb88aSMatthew Knepley } 139450ffb88aSMatthew Knepley 139550ffb88aSMatthew Knepley /*@ 1396b850b91aSLisandro Dalcin SNESSetMaxNonlinearStepFailures - Sets the maximum number of unsuccessful steps 139750ffb88aSMatthew Knepley attempted by the nonlinear solver before it gives up. 139850ffb88aSMatthew Knepley 139950ffb88aSMatthew Knepley Not Collective 140050ffb88aSMatthew Knepley 140150ffb88aSMatthew Knepley Input Parameters: 140250ffb88aSMatthew Knepley + snes - SNES context 140350ffb88aSMatthew Knepley - maxFails - maximum of unsuccessful steps 140450ffb88aSMatthew Knepley 140550ffb88aSMatthew Knepley Level: intermediate 140650ffb88aSMatthew Knepley 1407e1c61ce8SBarry Smith .seealso: SNESGetMaxLinearSolveFailures(), SNESGetLinearSolveIterations(), SNESSetMaxLinearSolveFailures(), SNESGetLinearSolveFailures(), 140858ebbce7SBarry Smith SNESGetMaxNonlinearStepFailures(), SNESGetNonlinearStepFailures() 140950ffb88aSMatthew Knepley @*/ 14107087cfbeSBarry Smith PetscErrorCode SNESSetMaxNonlinearStepFailures(SNES snes, PetscInt maxFails) 141150ffb88aSMatthew Knepley { 141250ffb88aSMatthew Knepley PetscFunctionBegin; 14130700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 141450ffb88aSMatthew Knepley snes->maxFailures = maxFails; 141550ffb88aSMatthew Knepley PetscFunctionReturn(0); 141650ffb88aSMatthew Knepley } 141750ffb88aSMatthew Knepley 141850ffb88aSMatthew Knepley /*@ 1419b850b91aSLisandro Dalcin SNESGetMaxNonlinearStepFailures - Gets the maximum number of unsuccessful steps 142050ffb88aSMatthew Knepley attempted by the nonlinear solver before it gives up. 142150ffb88aSMatthew Knepley 142250ffb88aSMatthew Knepley Not Collective 142350ffb88aSMatthew Knepley 142450ffb88aSMatthew Knepley Input Parameter: 142550ffb88aSMatthew Knepley . snes - SNES context 142650ffb88aSMatthew Knepley 142750ffb88aSMatthew Knepley Output Parameter: 142850ffb88aSMatthew Knepley . maxFails - maximum of unsuccessful steps 142950ffb88aSMatthew Knepley 143050ffb88aSMatthew Knepley Level: intermediate 143150ffb88aSMatthew Knepley 1432e1c61ce8SBarry Smith .seealso: SNESGetMaxLinearSolveFailures(), SNESGetLinearSolveIterations(), SNESSetMaxLinearSolveFailures(), SNESGetLinearSolveFailures(), 143358ebbce7SBarry Smith SNESSetMaxNonlinearStepFailures(), SNESGetNonlinearStepFailures() 143458ebbce7SBarry Smith 143550ffb88aSMatthew Knepley @*/ 14367087cfbeSBarry Smith PetscErrorCode SNESGetMaxNonlinearStepFailures(SNES snes, PetscInt *maxFails) 143750ffb88aSMatthew Knepley { 143850ffb88aSMatthew Knepley PetscFunctionBegin; 14390700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 14404482741eSBarry Smith PetscValidIntPointer(maxFails,2); 144150ffb88aSMatthew Knepley *maxFails = snes->maxFailures; 14423a40ed3dSBarry Smith PetscFunctionReturn(0); 14439b94acceSBarry Smith } 1444a847f771SSatish Balay 14452541af92SBarry Smith /*@ 14462541af92SBarry Smith SNESGetNumberFunctionEvals - Gets the number of user provided function evaluations 14472541af92SBarry Smith done by SNES. 14482541af92SBarry Smith 14492541af92SBarry Smith Not Collective 14502541af92SBarry Smith 14512541af92SBarry Smith Input Parameter: 14522541af92SBarry Smith . snes - SNES context 14532541af92SBarry Smith 14542541af92SBarry Smith Output Parameter: 14552541af92SBarry Smith . nfuncs - number of evaluations 14562541af92SBarry Smith 14572541af92SBarry Smith Level: intermediate 14582541af92SBarry Smith 145995452b02SPatrick Sanan Notes: 146095452b02SPatrick Sanan Reset every time SNESSolve is called unless SNESSetCountersReset() is used. 1461971e163fSPeter Brune 1462971e163fSPeter Brune .seealso: SNESGetMaxLinearSolveFailures(), SNESGetLinearSolveIterations(), SNESSetMaxLinearSolveFailures(), SNESGetLinearSolveFailures(), SNESSetCountersReset() 14632541af92SBarry Smith @*/ 14647087cfbeSBarry Smith PetscErrorCode SNESGetNumberFunctionEvals(SNES snes, PetscInt *nfuncs) 14652541af92SBarry Smith { 14662541af92SBarry Smith PetscFunctionBegin; 14670700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 14682541af92SBarry Smith PetscValidIntPointer(nfuncs,2); 14692541af92SBarry Smith *nfuncs = snes->nfuncs; 14702541af92SBarry Smith PetscFunctionReturn(0); 14712541af92SBarry Smith } 14722541af92SBarry Smith 14733d4c4710SBarry Smith /*@ 14743d4c4710SBarry Smith SNESGetLinearSolveFailures - Gets the number of failed (non-converged) 14753d4c4710SBarry Smith linear solvers. 14763d4c4710SBarry Smith 14773d4c4710SBarry Smith Not Collective 14783d4c4710SBarry Smith 14793d4c4710SBarry Smith Input Parameter: 14803d4c4710SBarry Smith . snes - SNES context 14813d4c4710SBarry Smith 14823d4c4710SBarry Smith Output Parameter: 14833d4c4710SBarry Smith . nfails - number of failed solves 14843d4c4710SBarry Smith 14859d85da0cSMatthew G. Knepley Level: intermediate 14869d85da0cSMatthew G. Knepley 14879d85da0cSMatthew G. Knepley Options Database Keys: 14889d85da0cSMatthew G. Knepley . -snes_max_linear_solve_fail <num> - The number of failures before the solve is terminated 14899d85da0cSMatthew G. Knepley 14903d4c4710SBarry Smith Notes: 14913d4c4710SBarry Smith This counter is reset to zero for each successive call to SNESSolve(). 14923d4c4710SBarry Smith 1493e1c61ce8SBarry Smith .seealso: SNESGetMaxLinearSolveFailures(), SNESGetLinearSolveIterations(), SNESSetMaxLinearSolveFailures() 14943d4c4710SBarry Smith @*/ 14957087cfbeSBarry Smith PetscErrorCode SNESGetLinearSolveFailures(SNES snes,PetscInt *nfails) 14963d4c4710SBarry Smith { 14973d4c4710SBarry Smith PetscFunctionBegin; 14980700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 14993d4c4710SBarry Smith PetscValidIntPointer(nfails,2); 15003d4c4710SBarry Smith *nfails = snes->numLinearSolveFailures; 15013d4c4710SBarry Smith PetscFunctionReturn(0); 15023d4c4710SBarry Smith } 15033d4c4710SBarry Smith 15043d4c4710SBarry Smith /*@ 15053d4c4710SBarry Smith SNESSetMaxLinearSolveFailures - the number of failed linear solve attempts 15063d4c4710SBarry Smith allowed before SNES returns with a diverged reason of SNES_DIVERGED_LINEAR_SOLVE 15073d4c4710SBarry Smith 15083f9fe445SBarry Smith Logically Collective on SNES 15093d4c4710SBarry Smith 15103d4c4710SBarry Smith Input Parameters: 15113d4c4710SBarry Smith + snes - SNES context 15123d4c4710SBarry Smith - maxFails - maximum allowed linear solve failures 15133d4c4710SBarry Smith 15143d4c4710SBarry Smith Level: intermediate 15153d4c4710SBarry Smith 15169d85da0cSMatthew G. Knepley Options Database Keys: 15179d85da0cSMatthew G. Knepley . -snes_max_linear_solve_fail <num> - The number of failures before the solve is terminated 15189d85da0cSMatthew G. Knepley 151995452b02SPatrick Sanan Notes: 152095452b02SPatrick Sanan By default this is 0; that is SNES returns on the first failed linear solve 15213d4c4710SBarry Smith 152258ebbce7SBarry Smith .seealso: SNESGetLinearSolveFailures(), SNESGetMaxLinearSolveFailures(), SNESGetLinearSolveIterations() 15233d4c4710SBarry Smith @*/ 15247087cfbeSBarry Smith PetscErrorCode SNESSetMaxLinearSolveFailures(SNES snes, PetscInt maxFails) 15253d4c4710SBarry Smith { 15263d4c4710SBarry Smith PetscFunctionBegin; 15270700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 1528c5eb9154SBarry Smith PetscValidLogicalCollectiveInt(snes,maxFails,2); 15293d4c4710SBarry Smith snes->maxLinearSolveFailures = maxFails; 15303d4c4710SBarry Smith PetscFunctionReturn(0); 15313d4c4710SBarry Smith } 15323d4c4710SBarry Smith 15333d4c4710SBarry Smith /*@ 15343d4c4710SBarry Smith SNESGetMaxLinearSolveFailures - gets the maximum number of linear solve failures that 15353d4c4710SBarry Smith are allowed before SNES terminates 15363d4c4710SBarry Smith 15373d4c4710SBarry Smith Not Collective 15383d4c4710SBarry Smith 15393d4c4710SBarry Smith Input Parameter: 15403d4c4710SBarry Smith . snes - SNES context 15413d4c4710SBarry Smith 15423d4c4710SBarry Smith Output Parameter: 15433d4c4710SBarry Smith . maxFails - maximum of unsuccessful solves allowed 15443d4c4710SBarry Smith 15453d4c4710SBarry Smith Level: intermediate 15463d4c4710SBarry Smith 154795452b02SPatrick Sanan Notes: 154895452b02SPatrick Sanan By default this is 1; that is SNES returns on the first failed linear solve 15493d4c4710SBarry Smith 1550e1c61ce8SBarry Smith .seealso: SNESGetLinearSolveFailures(), SNESGetLinearSolveIterations(), SNESSetMaxLinearSolveFailures(), 15513d4c4710SBarry Smith @*/ 15527087cfbeSBarry Smith PetscErrorCode SNESGetMaxLinearSolveFailures(SNES snes, PetscInt *maxFails) 15533d4c4710SBarry Smith { 15543d4c4710SBarry Smith PetscFunctionBegin; 15550700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 15563d4c4710SBarry Smith PetscValidIntPointer(maxFails,2); 15573d4c4710SBarry Smith *maxFails = snes->maxLinearSolveFailures; 15583d4c4710SBarry Smith PetscFunctionReturn(0); 15593d4c4710SBarry Smith } 15603d4c4710SBarry Smith 1561c96a6f78SLois Curfman McInnes /*@ 1562b850b91aSLisandro Dalcin SNESGetLinearSolveIterations - Gets the total number of linear iterations 1563c96a6f78SLois Curfman McInnes used by the nonlinear solver. 1564c96a6f78SLois Curfman McInnes 1565c7afd0dbSLois Curfman McInnes Not Collective 1566c7afd0dbSLois Curfman McInnes 1567c96a6f78SLois Curfman McInnes Input Parameter: 1568c96a6f78SLois Curfman McInnes . snes - SNES context 1569c96a6f78SLois Curfman McInnes 1570c96a6f78SLois Curfman McInnes Output Parameter: 1571c96a6f78SLois Curfman McInnes . lits - number of linear iterations 1572c96a6f78SLois Curfman McInnes 1573c96a6f78SLois Curfman McInnes Notes: 1574971e163fSPeter Brune This counter is reset to zero for each successive call to SNESSolve() unless SNESSetCountersReset() is used. 1575c96a6f78SLois Curfman McInnes 1576010be392SBarry 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 1577010be392SBarry Smith then call KSPGetIterationNumber() after the failed solve. 1578010be392SBarry Smith 157936851e7fSLois Curfman McInnes Level: intermediate 158036851e7fSLois Curfman McInnes 158171dbe336SPeter Brune .seealso: SNESGetIterationNumber(), SNESGetLinearSolveFailures(), SNESGetMaxLinearSolveFailures(), SNESSetCountersReset() 1582c96a6f78SLois Curfman McInnes @*/ 15837087cfbeSBarry Smith PetscErrorCode SNESGetLinearSolveIterations(SNES snes,PetscInt *lits) 1584c96a6f78SLois Curfman McInnes { 15853a40ed3dSBarry Smith PetscFunctionBegin; 15860700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 15874482741eSBarry Smith PetscValidIntPointer(lits,2); 1588c96a6f78SLois Curfman McInnes *lits = snes->linear_its; 15893a40ed3dSBarry Smith PetscFunctionReturn(0); 1590c96a6f78SLois Curfman McInnes } 1591c96a6f78SLois Curfman McInnes 1592971e163fSPeter Brune /*@ 1593971e163fSPeter Brune SNESSetCountersReset - Sets whether or not the counters for linear iterations and function evaluations 1594971e163fSPeter Brune are reset every time SNESSolve() is called. 1595971e163fSPeter Brune 1596971e163fSPeter Brune Logically Collective on SNES 1597971e163fSPeter Brune 1598971e163fSPeter Brune Input Parameter: 1599971e163fSPeter Brune + snes - SNES context 1600971e163fSPeter Brune - reset - whether to reset the counters or not 1601971e163fSPeter Brune 1602971e163fSPeter Brune Notes: 1603fa19ca70SBarry Smith This defaults to PETSC_TRUE 1604971e163fSPeter Brune 1605971e163fSPeter Brune Level: developer 1606971e163fSPeter Brune 1607734794cfSBarry Smith .seealso: SNESGetNumberFunctionEvals(), SNESGetLinearSolveIterations(), SNESGetNPC() 1608971e163fSPeter Brune @*/ 1609971e163fSPeter Brune PetscErrorCode SNESSetCountersReset(SNES snes,PetscBool reset) 1610971e163fSPeter Brune { 1611971e163fSPeter Brune PetscFunctionBegin; 1612971e163fSPeter Brune PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 1613971e163fSPeter Brune PetscValidLogicalCollectiveBool(snes,reset,2); 1614971e163fSPeter Brune snes->counters_reset = reset; 1615971e163fSPeter Brune PetscFunctionReturn(0); 1616971e163fSPeter Brune } 1617971e163fSPeter Brune 161882bf6240SBarry Smith 16192999313aSBarry Smith /*@ 16202999313aSBarry Smith SNESSetKSP - Sets a KSP context for the SNES object to use 16212999313aSBarry Smith 16222999313aSBarry Smith Not Collective, but the SNES and KSP objects must live on the same MPI_Comm 16232999313aSBarry Smith 16242999313aSBarry Smith Input Parameters: 16252999313aSBarry Smith + snes - the SNES context 16262999313aSBarry Smith - ksp - the KSP context 16272999313aSBarry Smith 16282999313aSBarry Smith Notes: 16292999313aSBarry Smith The SNES object already has its KSP object, you can obtain with SNESGetKSP() 16302999313aSBarry Smith so this routine is rarely needed. 16312999313aSBarry Smith 16322999313aSBarry Smith The KSP object that is already in the SNES object has its reference count 16332999313aSBarry Smith decreased by one. 16342999313aSBarry Smith 16352999313aSBarry Smith Level: developer 16362999313aSBarry Smith 16372999313aSBarry Smith .seealso: KSPGetPC(), SNESCreate(), KSPCreate(), SNESSetKSP() 16382999313aSBarry Smith @*/ 16397087cfbeSBarry Smith PetscErrorCode SNESSetKSP(SNES snes,KSP ksp) 16402999313aSBarry Smith { 16412999313aSBarry Smith PetscErrorCode ierr; 16422999313aSBarry Smith 16432999313aSBarry Smith PetscFunctionBegin; 16440700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 16450700a824SBarry Smith PetscValidHeaderSpecific(ksp,KSP_CLASSID,2); 16462999313aSBarry Smith PetscCheckSameComm(snes,1,ksp,2); 16477dcf0eaaSdalcinl ierr = PetscObjectReference((PetscObject)ksp);CHKERRQ(ierr); 1648906ed7ccSBarry Smith if (snes->ksp) {ierr = PetscObjectDereference((PetscObject)snes->ksp);CHKERRQ(ierr);} 16492999313aSBarry Smith snes->ksp = ksp; 16502999313aSBarry Smith PetscFunctionReturn(0); 16512999313aSBarry Smith } 16522999313aSBarry Smith 16539b94acceSBarry Smith /* -----------------------------------------------------------*/ 165452baeb72SSatish Balay /*@ 16559b94acceSBarry Smith SNESCreate - Creates a nonlinear solver context. 16569b94acceSBarry Smith 1657d083f849SBarry Smith Collective 1658c7afd0dbSLois Curfman McInnes 1659c7afd0dbSLois Curfman McInnes Input Parameters: 1660906ed7ccSBarry Smith . comm - MPI communicator 16619b94acceSBarry Smith 16629b94acceSBarry Smith Output Parameter: 16639b94acceSBarry Smith . outsnes - the new SNES context 16649b94acceSBarry Smith 1665c7afd0dbSLois Curfman McInnes Options Database Keys: 1666c7afd0dbSLois Curfman McInnes + -snes_mf - Activates default matrix-free Jacobian-vector products, 1667c7afd0dbSLois Curfman McInnes and no preconditioning matrix 1668c7afd0dbSLois Curfman McInnes . -snes_mf_operator - Activates default matrix-free Jacobian-vector 1669c7afd0dbSLois Curfman McInnes products, and a user-provided preconditioning matrix 1670c7afd0dbSLois Curfman McInnes as set by SNESSetJacobian() 1671c7afd0dbSLois Curfman McInnes - -snes_fd - Uses (slow!) finite differences to compute Jacobian 1672c1f60f51SBarry Smith 167336851e7fSLois Curfman McInnes Level: beginner 167436851e7fSLois Curfman McInnes 167595452b02SPatrick Sanan Developer Notes: 167695452b02SPatrick Sanan SNES always creates a KSP object even though many SNES methods do not use it. This is 1677efd4aadfSBarry Smith unfortunate and should be fixed at some point. The flag snes->usesksp indicates if the 1678efd4aadfSBarry Smith particular method does use KSP and regulates if the information about the KSP is printed 1679efd4aadfSBarry Smith in SNESView(). TSSetFromOptions() does call SNESSetFromOptions() which can lead to users being confused 1680efd4aadfSBarry Smith by help messages about meaningless SNES options. 1681efd4aadfSBarry Smith 1682efd4aadfSBarry Smith SNES always creates the snes->kspconvctx even though it is used by only one type. This should 1683efd4aadfSBarry Smith be fixed. 1684efd4aadfSBarry Smith 16853d5a8a6aSBarry Smith .seealso: SNESSolve(), SNESDestroy(), SNES, SNESSetLagPreconditioner(), SNESSetLagJacobian() 1686a8054027SBarry Smith 16879b94acceSBarry Smith @*/ 16887087cfbeSBarry Smith PetscErrorCode SNESCreate(MPI_Comm comm,SNES *outsnes) 16899b94acceSBarry Smith { 1690dfbe8321SBarry Smith PetscErrorCode ierr; 16919b94acceSBarry Smith SNES snes; 1692fa9f3622SBarry Smith SNESKSPEW *kctx; 169337fcc0dbSBarry Smith 16943a40ed3dSBarry Smith PetscFunctionBegin; 1695ed1caa07SMatthew Knepley PetscValidPointer(outsnes,2); 16960298fd71SBarry Smith *outsnes = NULL; 1697607a6623SBarry Smith ierr = SNESInitializePackage();CHKERRQ(ierr); 16988ba1e511SMatthew Knepley 169973107ff1SLisandro Dalcin ierr = PetscHeaderCreate(snes,SNES_CLASSID,"SNES","Nonlinear solver","SNES",comm,SNESDestroy,SNESView);CHKERRQ(ierr); 17007adad957SLisandro Dalcin 17018d359177SBarry Smith snes->ops->converged = SNESConvergedDefault; 17022c155ee1SBarry Smith snes->usesksp = PETSC_TRUE; 170388976e71SPeter Brune snes->tolerancesset = PETSC_FALSE; 17049b94acceSBarry Smith snes->max_its = 50; 17059750a799SBarry Smith snes->max_funcs = 10000; 17069b94acceSBarry Smith snes->norm = 0.0; 1707c1e67a49SFande Kong snes->xnorm = 0.0; 1708c1e67a49SFande Kong snes->ynorm = 0.0; 1709365a6726SPeter Brune snes->normschedule = SNES_NORM_ALWAYS; 17106c67d002SPeter Brune snes->functype = SNES_FUNCTION_DEFAULT; 17113a2046daSBarry Smith #if defined(PETSC_USE_REAL_SINGLE) 17123a2046daSBarry Smith snes->rtol = 1.e-5; 17133a2046daSBarry Smith #else 1714b4874afaSBarry Smith snes->rtol = 1.e-8; 17153a2046daSBarry Smith #endif 1716b4874afaSBarry Smith snes->ttol = 0.0; 17173a2046daSBarry Smith #if defined(PETSC_USE_REAL_SINGLE) 17183a2046daSBarry Smith snes->abstol = 1.e-25; 17193a2046daSBarry Smith #else 172070441072SBarry Smith snes->abstol = 1.e-50; 17213a2046daSBarry Smith #endif 17227cd0ae37SLisandro Dalcin #if defined(PETSC_USE_REAL_SINGLE) 17237cd0ae37SLisandro Dalcin snes->stol = 1.e-5; 17247cd0ae37SLisandro Dalcin #else 1725c60f73f4SPeter Brune snes->stol = 1.e-8; 17267cd0ae37SLisandro Dalcin #endif 17273a2046daSBarry Smith #if defined(PETSC_USE_REAL_SINGLE) 17283a2046daSBarry Smith snes->deltatol = 1.e-6; 17293a2046daSBarry Smith #else 17304b27c08aSLois Curfman McInnes snes->deltatol = 1.e-12; 17313a2046daSBarry Smith #endif 1732e37c518bSBarry Smith snes->divtol = 1.e4; 1733e37c518bSBarry Smith snes->rnorm0 = 0; 17349b94acceSBarry Smith snes->nfuncs = 0; 173550ffb88aSMatthew Knepley snes->numFailures = 0; 173650ffb88aSMatthew Knepley snes->maxFailures = 1; 17377a00f4a9SLois Curfman McInnes snes->linear_its = 0; 1738e35cf81dSBarry Smith snes->lagjacobian = 1; 173937ec4e1aSPeter Brune snes->jac_iter = 0; 174037ec4e1aSPeter Brune snes->lagjac_persist = PETSC_FALSE; 1741a8054027SBarry Smith snes->lagpreconditioner = 1; 174237ec4e1aSPeter Brune snes->pre_iter = 0; 174337ec4e1aSPeter Brune snes->lagpre_persist = PETSC_FALSE; 1744639f9d9dSBarry Smith snes->numbermonitors = 0; 1745c4421ceaSFande Kong snes->numberreasonviews = 0; 17469e5d0892SLisandro Dalcin snes->data = NULL; 17474dc4c822SBarry Smith snes->setupcalled = PETSC_FALSE; 1748186905e3SBarry Smith snes->ksp_ewconv = PETSC_FALSE; 17496f24a144SLois Curfman McInnes snes->nwork = 0; 17509e5d0892SLisandro Dalcin snes->work = NULL; 175158c9b817SLisandro Dalcin snes->nvwork = 0; 17529e5d0892SLisandro Dalcin snes->vwork = NULL; 1753758f92a0SBarry Smith snes->conv_hist_len = 0; 1754758f92a0SBarry Smith snes->conv_hist_max = 0; 17550298fd71SBarry Smith snes->conv_hist = NULL; 17560298fd71SBarry Smith snes->conv_hist_its = NULL; 1757758f92a0SBarry Smith snes->conv_hist_reset = PETSC_TRUE; 1758971e163fSPeter Brune snes->counters_reset = PETSC_TRUE; 1759e4ed7901SPeter Brune snes->vec_func_init_set = PETSC_FALSE; 1760184914b5SBarry Smith snes->reason = SNES_CONVERGED_ITERATING; 1761efd4aadfSBarry Smith snes->npcside = PC_RIGHT; 1762b3cd9a81SMatthew G. Knepley snes->setfromoptionscalled = 0; 1763c40d0f55SPeter Brune 1764d8f46077SPeter Brune snes->mf = PETSC_FALSE; 1765d8f46077SPeter Brune snes->mf_operator = PETSC_FALSE; 1766d8f46077SPeter Brune snes->mf_version = 1; 1767d8f46077SPeter Brune 17683d4c4710SBarry Smith snes->numLinearSolveFailures = 0; 17693d4c4710SBarry Smith snes->maxLinearSolveFailures = 1; 17703d4c4710SBarry Smith 1771349187a7SBarry Smith snes->vizerotolerance = 1.e-8; 177276bd3646SJed Brown snes->checkjacdomainerror = PetscDefined(USE_DEBUG) ? PETSC_TRUE : PETSC_FALSE; 1773349187a7SBarry Smith 17744fc747eaSLawrence Mitchell /* Set this to true if the implementation of SNESSolve_XXX does compute the residual at the final solution. */ 17754fc747eaSLawrence Mitchell snes->alwayscomputesfinalresidual = PETSC_FALSE; 17764fc747eaSLawrence Mitchell 17779b94acceSBarry Smith /* Create context to compute Eisenstat-Walker relative tolerance for KSP */ 1778b00a9115SJed Brown ierr = PetscNewLog(snes,&kctx);CHKERRQ(ierr); 1779f5af7f23SKarl Rupp 17809b94acceSBarry Smith snes->kspconvctx = (void*)kctx; 17819b94acceSBarry Smith kctx->version = 2; 17829b94acceSBarry Smith kctx->rtol_0 = .3; /* Eisenstat and Walker suggest rtol_0=.5, but 17839b94acceSBarry Smith this was too large for some test cases */ 178475567043SBarry Smith kctx->rtol_last = 0.0; 17859b94acceSBarry Smith kctx->rtol_max = .9; 17869b94acceSBarry Smith kctx->gamma = 1.0; 178762d1f40fSBarry Smith kctx->alpha = .5*(1.0 + PetscSqrtReal(5.0)); 178871f87433Sdalcinl kctx->alpha2 = kctx->alpha; 17899b94acceSBarry Smith kctx->threshold = .1; 179075567043SBarry Smith kctx->lresid_last = 0.0; 179175567043SBarry Smith kctx->norm_last = 0.0; 17929b94acceSBarry Smith 17939b94acceSBarry Smith *outsnes = snes; 17943a40ed3dSBarry Smith PetscFunctionReturn(0); 17959b94acceSBarry Smith } 17969b94acceSBarry Smith 179788f0584fSBarry Smith /*MC 1798411c0326SBarry Smith SNESFunction - Functional form used to convey the nonlinear function to be solved by SNES 179988f0584fSBarry Smith 180088f0584fSBarry Smith Synopsis: 1801411c0326SBarry Smith #include "petscsnes.h" 1802411c0326SBarry Smith PetscErrorCode SNESFunction(SNES snes,Vec x,Vec f,void *ctx); 180388f0584fSBarry Smith 18041843f636SBarry Smith Collective on snes 18051843f636SBarry Smith 180688f0584fSBarry Smith Input Parameters: 180788f0584fSBarry Smith + snes - the SNES context 180888f0584fSBarry Smith . x - state at which to evaluate residual 180988f0584fSBarry Smith - ctx - optional user-defined function context, passed in with SNESSetFunction() 181088f0584fSBarry Smith 181188f0584fSBarry Smith Output Parameter: 181288f0584fSBarry Smith . f - vector to put residual (function value) 181388f0584fSBarry Smith 1814878cb397SSatish Balay Level: intermediate 1815878cb397SSatish Balay 181688f0584fSBarry Smith .seealso: SNESSetFunction(), SNESGetFunction() 181788f0584fSBarry Smith M*/ 181888f0584fSBarry Smith 18199b94acceSBarry Smith /*@C 18209b94acceSBarry Smith SNESSetFunction - Sets the function evaluation routine and function 18219b94acceSBarry Smith vector for use by the SNES routines in solving systems of nonlinear 18229b94acceSBarry Smith equations. 18239b94acceSBarry Smith 18243f9fe445SBarry Smith Logically Collective on SNES 1825fee21e36SBarry Smith 1826c7afd0dbSLois Curfman McInnes Input Parameters: 1827c7afd0dbSLois Curfman McInnes + snes - the SNES context 1828c7afd0dbSLois Curfman McInnes . r - vector to store function value 1829f8b49ee9SBarry Smith . f - function evaluation routine; see SNESFunction for calling sequence details 1830c7afd0dbSLois Curfman McInnes - ctx - [optional] user-defined context for private data for the 18310298fd71SBarry Smith function evaluation routine (may be NULL) 18329b94acceSBarry Smith 18339b94acceSBarry Smith Notes: 18349b94acceSBarry Smith The Newton-like methods typically solve linear systems of the form 18359b94acceSBarry Smith $ f'(x) x = -f(x), 1836c7afd0dbSLois Curfman McInnes where f'(x) denotes the Jacobian matrix and f(x) is the function. 18379b94acceSBarry Smith 183836851e7fSLois Curfman McInnes Level: beginner 183936851e7fSLois Curfman McInnes 1840bf388a1fSBarry Smith .seealso: SNESGetFunction(), SNESComputeFunction(), SNESSetJacobian(), SNESSetPicard(), SNESFunction 18419b94acceSBarry Smith @*/ 1842f8b49ee9SBarry Smith PetscErrorCode SNESSetFunction(SNES snes,Vec r,PetscErrorCode (*f)(SNES,Vec,Vec,void*),void *ctx) 18439b94acceSBarry Smith { 184485385478SLisandro Dalcin PetscErrorCode ierr; 18456cab3a1bSJed Brown DM dm; 18466cab3a1bSJed Brown 18473a40ed3dSBarry Smith PetscFunctionBegin; 18480700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 1849d2a683ecSLisandro Dalcin if (r) { 1850d2a683ecSLisandro Dalcin PetscValidHeaderSpecific(r,VEC_CLASSID,2); 1851d2a683ecSLisandro Dalcin PetscCheckSameComm(snes,1,r,2); 185285385478SLisandro Dalcin ierr = PetscObjectReference((PetscObject)r);CHKERRQ(ierr); 18536bf464f9SBarry Smith ierr = VecDestroy(&snes->vec_func);CHKERRQ(ierr); 1854f5af7f23SKarl Rupp 185585385478SLisandro Dalcin snes->vec_func = r; 1856d2a683ecSLisandro Dalcin } 18576cab3a1bSJed Brown ierr = SNESGetDM(snes,&dm);CHKERRQ(ierr); 1858f8b49ee9SBarry Smith ierr = DMSNESSetFunction(dm,f,ctx);CHKERRQ(ierr); 18593a40ed3dSBarry Smith PetscFunctionReturn(0); 18609b94acceSBarry Smith } 18619b94acceSBarry Smith 1862646217ecSPeter Brune 1863e4ed7901SPeter Brune /*@C 1864e4ed7901SPeter Brune SNESSetInitialFunction - Sets the function vector to be used as the 1865e4ed7901SPeter Brune function norm at the initialization of the method. In some 1866e4ed7901SPeter Brune instances, the user has precomputed the function before calling 1867e4ed7901SPeter Brune SNESSolve. This function allows one to avoid a redundant call 1868e4ed7901SPeter Brune to SNESComputeFunction in that case. 1869e4ed7901SPeter Brune 1870e4ed7901SPeter Brune Logically Collective on SNES 1871e4ed7901SPeter Brune 1872e4ed7901SPeter Brune Input Parameters: 1873e4ed7901SPeter Brune + snes - the SNES context 1874e4ed7901SPeter Brune - f - vector to store function value 1875e4ed7901SPeter Brune 1876e4ed7901SPeter Brune Notes: 1877e4ed7901SPeter Brune This should not be modified during the solution procedure. 1878e4ed7901SPeter Brune 1879e4ed7901SPeter Brune This is used extensively in the SNESFAS hierarchy and in nonlinear preconditioning. 1880e4ed7901SPeter Brune 1881e4ed7901SPeter Brune Level: developer 1882e4ed7901SPeter Brune 1883e4ed7901SPeter Brune .seealso: SNESSetFunction(), SNESComputeFunction(), SNESSetInitialFunctionNorm() 1884e4ed7901SPeter Brune @*/ 1885e4ed7901SPeter Brune PetscErrorCode SNESSetInitialFunction(SNES snes, Vec f) 1886e4ed7901SPeter Brune { 1887e4ed7901SPeter Brune PetscErrorCode ierr; 1888e4ed7901SPeter Brune Vec vec_func; 1889e4ed7901SPeter Brune 1890e4ed7901SPeter Brune PetscFunctionBegin; 1891e4ed7901SPeter Brune PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 1892e4ed7901SPeter Brune PetscValidHeaderSpecific(f,VEC_CLASSID,2); 1893e4ed7901SPeter Brune PetscCheckSameComm(snes,1,f,2); 1894efd4aadfSBarry Smith if (snes->npcside== PC_LEFT && snes->functype == SNES_FUNCTION_PRECONDITIONED) { 1895902f982fSPeter Brune snes->vec_func_init_set = PETSC_FALSE; 1896902f982fSPeter Brune PetscFunctionReturn(0); 1897902f982fSPeter Brune } 18980298fd71SBarry Smith ierr = SNESGetFunction(snes,&vec_func,NULL,NULL);CHKERRQ(ierr); 1899e4ed7901SPeter Brune ierr = VecCopy(f, vec_func);CHKERRQ(ierr); 1900f5af7f23SKarl Rupp 1901217b9c2eSPeter Brune snes->vec_func_init_set = PETSC_TRUE; 1902e4ed7901SPeter Brune PetscFunctionReturn(0); 1903e4ed7901SPeter Brune } 1904e4ed7901SPeter Brune 1905534ebe21SPeter Brune /*@ 1906365a6726SPeter Brune SNESSetNormSchedule - Sets the SNESNormSchedule used in covergence and monitoring 1907534ebe21SPeter Brune of the SNES method. 1908534ebe21SPeter Brune 1909534ebe21SPeter Brune Logically Collective on SNES 1910534ebe21SPeter Brune 1911534ebe21SPeter Brune Input Parameters: 1912534ebe21SPeter Brune + snes - the SNES context 1913365a6726SPeter Brune - normschedule - the frequency of norm computation 1914534ebe21SPeter Brune 1915517f1916SMatthew G. Knepley Options Database Key: 1916517f1916SMatthew G. Knepley . -snes_norm_schedule <none, always, initialonly, finalonly, initalfinalonly> 1917517f1916SMatthew G. Knepley 1918534ebe21SPeter Brune Notes: 1919365a6726SPeter Brune Only certain SNES methods support certain SNESNormSchedules. Most require evaluation 1920534ebe21SPeter Brune of the nonlinear function and the taking of its norm at every iteration to 1921534ebe21SPeter Brune even ensure convergence at all. However, methods such as custom Gauss-Seidel methods 1922be95d8f1SBarry Smith (SNESNGS) and the like do not require the norm of the function to be computed, and therfore 1923534ebe21SPeter Brune may either be monitored for convergence or not. As these are often used as nonlinear 1924534ebe21SPeter Brune preconditioners, monitoring the norm of their error is not a useful enterprise within 1925534ebe21SPeter Brune their solution. 1926534ebe21SPeter Brune 1927534ebe21SPeter Brune Level: developer 1928534ebe21SPeter Brune 1929365a6726SPeter Brune .seealso: SNESGetNormSchedule(), SNESComputeFunction(), VecNorm(), SNESSetFunction(), SNESSetInitialFunction(), SNESNormSchedule 1930534ebe21SPeter Brune @*/ 1931365a6726SPeter Brune PetscErrorCode SNESSetNormSchedule(SNES snes, SNESNormSchedule normschedule) 1932534ebe21SPeter Brune { 1933534ebe21SPeter Brune PetscFunctionBegin; 1934534ebe21SPeter Brune PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 1935365a6726SPeter Brune snes->normschedule = normschedule; 1936534ebe21SPeter Brune PetscFunctionReturn(0); 1937534ebe21SPeter Brune } 1938534ebe21SPeter Brune 1939534ebe21SPeter Brune 1940534ebe21SPeter Brune /*@ 1941365a6726SPeter Brune SNESGetNormSchedule - Gets the SNESNormSchedule used in covergence and monitoring 1942534ebe21SPeter Brune of the SNES method. 1943534ebe21SPeter Brune 1944534ebe21SPeter Brune Logically Collective on SNES 1945534ebe21SPeter Brune 1946534ebe21SPeter Brune Input Parameters: 1947534ebe21SPeter Brune + snes - the SNES context 1948365a6726SPeter Brune - normschedule - the type of the norm used 1949534ebe21SPeter Brune 1950534ebe21SPeter Brune Level: advanced 1951534ebe21SPeter Brune 1952365a6726SPeter Brune .seealso: SNESSetNormSchedule(), SNESComputeFunction(), VecNorm(), SNESSetFunction(), SNESSetInitialFunction(), SNESNormSchedule 1953534ebe21SPeter Brune @*/ 1954365a6726SPeter Brune PetscErrorCode SNESGetNormSchedule(SNES snes, SNESNormSchedule *normschedule) 1955534ebe21SPeter Brune { 1956534ebe21SPeter Brune PetscFunctionBegin; 1957534ebe21SPeter Brune PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 1958365a6726SPeter Brune *normschedule = snes->normschedule; 1959534ebe21SPeter Brune PetscFunctionReturn(0); 1960534ebe21SPeter Brune } 1961534ebe21SPeter Brune 196247073ea2SPeter Brune 1963c5ce4427SMatthew G. Knepley /*@ 1964c5ce4427SMatthew G. Knepley SNESSetFunctionNorm - Sets the last computed residual norm. 1965c5ce4427SMatthew G. Knepley 1966c5ce4427SMatthew G. Knepley Logically Collective on SNES 1967c5ce4427SMatthew G. Knepley 1968c5ce4427SMatthew G. Knepley Input Parameters: 1969c5ce4427SMatthew G. Knepley + snes - the SNES context 1970c5ce4427SMatthew G. Knepley 1971c5ce4427SMatthew G. Knepley - normschedule - the frequency of norm computation 1972c5ce4427SMatthew G. Knepley 1973c5ce4427SMatthew G. Knepley Level: developer 1974c5ce4427SMatthew G. Knepley 1975c5ce4427SMatthew G. Knepley .seealso: SNESGetNormSchedule(), SNESComputeFunction(), VecNorm(), SNESSetFunction(), SNESSetInitialFunction(), SNESNormSchedule 1976c5ce4427SMatthew G. Knepley @*/ 1977c5ce4427SMatthew G. Knepley PetscErrorCode SNESSetFunctionNorm(SNES snes, PetscReal norm) 1978c5ce4427SMatthew G. Knepley { 1979c5ce4427SMatthew G. Knepley PetscFunctionBegin; 1980c5ce4427SMatthew G. Knepley PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 1981c5ce4427SMatthew G. Knepley snes->norm = norm; 1982c5ce4427SMatthew G. Knepley PetscFunctionReturn(0); 1983c5ce4427SMatthew G. Knepley } 1984c5ce4427SMatthew G. Knepley 1985c5ce4427SMatthew G. Knepley /*@ 1986c5ce4427SMatthew G. Knepley SNESGetFunctionNorm - Gets the last computed norm of the residual 1987c5ce4427SMatthew G. Knepley 1988c5ce4427SMatthew G. Knepley Not Collective 1989c5ce4427SMatthew G. Knepley 1990c5ce4427SMatthew G. Knepley Input Parameter: 1991c5ce4427SMatthew G. Knepley . snes - the SNES context 1992c5ce4427SMatthew G. Knepley 1993c5ce4427SMatthew G. Knepley Output Parameter: 1994c5ce4427SMatthew G. Knepley . norm - the last computed residual norm 1995c5ce4427SMatthew G. Knepley 1996c5ce4427SMatthew G. Knepley Level: developer 1997c5ce4427SMatthew G. Knepley 1998c5ce4427SMatthew G. Knepley .seealso: SNESSetNormSchedule(), SNESComputeFunction(), VecNorm(), SNESSetFunction(), SNESSetInitialFunction(), SNESNormSchedule 1999c5ce4427SMatthew G. Knepley @*/ 2000c5ce4427SMatthew G. Knepley PetscErrorCode SNESGetFunctionNorm(SNES snes, PetscReal *norm) 2001c5ce4427SMatthew G. Knepley { 2002c5ce4427SMatthew G. Knepley PetscFunctionBegin; 2003c5ce4427SMatthew G. Knepley PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 2004c5ce4427SMatthew G. Knepley PetscValidPointer(norm, 2); 2005c5ce4427SMatthew G. Knepley *norm = snes->norm; 2006c5ce4427SMatthew G. Knepley PetscFunctionReturn(0); 2007c5ce4427SMatthew G. Knepley } 2008c5ce4427SMatthew G. Knepley 2009c1e67a49SFande Kong /*@ 2010c1e67a49SFande Kong SNESGetUpdateNorm - Gets the last computed norm of the Newton update 2011c1e67a49SFande Kong 2012c1e67a49SFande Kong Not Collective 2013c1e67a49SFande Kong 2014c1e67a49SFande Kong Input Parameter: 2015c1e67a49SFande Kong . snes - the SNES context 2016c1e67a49SFande Kong 2017c1e67a49SFande Kong Output Parameter: 2018c1e67a49SFande Kong . ynorm - the last computed update norm 2019c1e67a49SFande Kong 2020c1e67a49SFande Kong Level: developer 2021c1e67a49SFande Kong 2022c1e67a49SFande Kong .seealso: SNESSetNormSchedule(), SNESComputeFunction(), SNESGetFunctionNorm() 2023c1e67a49SFande Kong @*/ 2024c1e67a49SFande Kong PetscErrorCode SNESGetUpdateNorm(SNES snes, PetscReal *ynorm) 2025c1e67a49SFande Kong { 2026c1e67a49SFande Kong PetscFunctionBegin; 2027c1e67a49SFande Kong PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 2028c1e67a49SFande Kong PetscValidPointer(ynorm, 2); 2029c1e67a49SFande Kong *ynorm = snes->ynorm; 2030c1e67a49SFande Kong PetscFunctionReturn(0); 2031c1e67a49SFande Kong } 2032c1e67a49SFande Kong 2033c1e67a49SFande Kong /*@ 20344591eaf2SFande Kong SNESGetSolutionNorm - Gets the last computed norm of the solution 2035c1e67a49SFande Kong 2036c1e67a49SFande Kong Not Collective 2037c1e67a49SFande Kong 2038c1e67a49SFande Kong Input Parameter: 2039c1e67a49SFande Kong . snes - the SNES context 2040c1e67a49SFande Kong 2041c1e67a49SFande Kong Output Parameter: 2042c1e67a49SFande Kong . xnorm - the last computed solution norm 2043c1e67a49SFande Kong 2044c1e67a49SFande Kong Level: developer 2045c1e67a49SFande Kong 2046c1e67a49SFande Kong .seealso: SNESSetNormSchedule(), SNESComputeFunction(), SNESGetFunctionNorm(), SNESGetUpdateNorm() 2047c1e67a49SFande Kong @*/ 2048c1e67a49SFande Kong PetscErrorCode SNESGetSolutionNorm(SNES snes, PetscReal *xnorm) 2049c1e67a49SFande Kong { 2050c1e67a49SFande Kong PetscFunctionBegin; 2051c1e67a49SFande Kong PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 2052c1e67a49SFande Kong PetscValidPointer(xnorm, 2); 2053c1e67a49SFande Kong *xnorm = snes->xnorm; 2054c1e67a49SFande Kong PetscFunctionReturn(0); 2055c1e67a49SFande Kong } 2056c1e67a49SFande Kong 205747073ea2SPeter Brune /*@C 205847073ea2SPeter Brune SNESSetFunctionType - Sets the SNESNormSchedule used in covergence and monitoring 205947073ea2SPeter Brune of the SNES method. 206047073ea2SPeter Brune 206147073ea2SPeter Brune Logically Collective on SNES 206247073ea2SPeter Brune 206347073ea2SPeter Brune Input Parameters: 206447073ea2SPeter Brune + snes - the SNES context 206547073ea2SPeter Brune - normschedule - the frequency of norm computation 206647073ea2SPeter Brune 206747073ea2SPeter Brune Notes: 206847073ea2SPeter Brune Only certain SNES methods support certain SNESNormSchedules. Most require evaluation 206947073ea2SPeter Brune of the nonlinear function and the taking of its norm at every iteration to 207047073ea2SPeter Brune even ensure convergence at all. However, methods such as custom Gauss-Seidel methods 2071be95d8f1SBarry Smith (SNESNGS) and the like do not require the norm of the function to be computed, and therfore 207247073ea2SPeter Brune may either be monitored for convergence or not. As these are often used as nonlinear 207347073ea2SPeter Brune preconditioners, monitoring the norm of their error is not a useful enterprise within 207447073ea2SPeter Brune their solution. 207547073ea2SPeter Brune 207647073ea2SPeter Brune Level: developer 207747073ea2SPeter Brune 207847073ea2SPeter Brune .seealso: SNESGetNormSchedule(), SNESComputeFunction(), VecNorm(), SNESSetFunction(), SNESSetInitialFunction(), SNESNormSchedule 207947073ea2SPeter Brune @*/ 208047073ea2SPeter Brune PetscErrorCode SNESSetFunctionType(SNES snes, SNESFunctionType type) 208147073ea2SPeter Brune { 208247073ea2SPeter Brune PetscFunctionBegin; 208347073ea2SPeter Brune PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 208447073ea2SPeter Brune snes->functype = type; 208547073ea2SPeter Brune PetscFunctionReturn(0); 208647073ea2SPeter Brune } 208747073ea2SPeter Brune 208847073ea2SPeter Brune 208947073ea2SPeter Brune /*@C 209047073ea2SPeter Brune SNESGetFunctionType - Gets the SNESNormSchedule used in covergence and monitoring 209147073ea2SPeter Brune of the SNES method. 209247073ea2SPeter Brune 209347073ea2SPeter Brune Logically Collective on SNES 209447073ea2SPeter Brune 209547073ea2SPeter Brune Input Parameters: 209647073ea2SPeter Brune + snes - the SNES context 209747073ea2SPeter Brune - normschedule - the type of the norm used 209847073ea2SPeter Brune 209947073ea2SPeter Brune Level: advanced 210047073ea2SPeter Brune 210147073ea2SPeter Brune .seealso: SNESSetNormSchedule(), SNESComputeFunction(), VecNorm(), SNESSetFunction(), SNESSetInitialFunction(), SNESNormSchedule 210247073ea2SPeter Brune @*/ 210347073ea2SPeter Brune PetscErrorCode SNESGetFunctionType(SNES snes, SNESFunctionType *type) 210447073ea2SPeter Brune { 210547073ea2SPeter Brune PetscFunctionBegin; 210647073ea2SPeter Brune PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 210747073ea2SPeter Brune *type = snes->functype; 2108534ebe21SPeter Brune PetscFunctionReturn(0); 2109534ebe21SPeter Brune } 2110534ebe21SPeter Brune 2111bf388a1fSBarry Smith /*MC 2112be95d8f1SBarry Smith SNESNGSFunction - function used to convey a Gauss-Seidel sweep on the nonlinear function 2113bf388a1fSBarry Smith 2114bf388a1fSBarry Smith Synopsis: 2115aaa7dc30SBarry Smith #include <petscsnes.h> 2116be95d8f1SBarry Smith $ SNESNGSFunction(SNES snes,Vec x,Vec b,void *ctx); 2117bf388a1fSBarry Smith 21181843f636SBarry Smith Collective on snes 21191843f636SBarry Smith 21201843f636SBarry Smith Input Parameters: 2121bf388a1fSBarry Smith + X - solution vector 2122bf388a1fSBarry Smith . B - RHS vector 2123bf388a1fSBarry Smith - ctx - optional user-defined Gauss-Seidel context 2124bf388a1fSBarry Smith 21251843f636SBarry Smith Output Parameter: 21261843f636SBarry Smith . X - solution vector 21271843f636SBarry Smith 2128878cb397SSatish Balay Level: intermediate 2129878cb397SSatish Balay 2130be95d8f1SBarry Smith .seealso: SNESSetNGS(), SNESGetNGS() 2131bf388a1fSBarry Smith M*/ 2132bf388a1fSBarry Smith 2133c79ef259SPeter Brune /*@C 2134be95d8f1SBarry Smith SNESSetNGS - Sets the user nonlinear Gauss-Seidel routine for 2135c79ef259SPeter Brune use with composed nonlinear solvers. 2136c79ef259SPeter Brune 2137c79ef259SPeter Brune Input Parameters: 2138c79ef259SPeter Brune + snes - the SNES context 2139be95d8f1SBarry Smith . f - function evaluation routine to apply Gauss-Seidel see SNESNGSFunction 2140c79ef259SPeter Brune - ctx - [optional] user-defined context for private data for the 21410298fd71SBarry Smith smoother evaluation routine (may be NULL) 2142c79ef259SPeter Brune 2143c79ef259SPeter Brune Notes: 2144be95d8f1SBarry Smith The NGS routines are used by the composed nonlinear solver to generate 2145c79ef259SPeter Brune a problem appropriate update to the solution, particularly FAS. 2146c79ef259SPeter Brune 2147d28543b3SPeter Brune Level: intermediate 2148c79ef259SPeter Brune 2149be95d8f1SBarry Smith .seealso: SNESGetFunction(), SNESComputeNGS() 2150c79ef259SPeter Brune @*/ 2151be95d8f1SBarry Smith PetscErrorCode SNESSetNGS(SNES snes,PetscErrorCode (*f)(SNES,Vec,Vec,void*),void *ctx) 21526cab3a1bSJed Brown { 21536cab3a1bSJed Brown PetscErrorCode ierr; 21546cab3a1bSJed Brown DM dm; 21556cab3a1bSJed Brown 2156646217ecSPeter Brune PetscFunctionBegin; 21576cab3a1bSJed Brown PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 21586cab3a1bSJed Brown ierr = SNESGetDM(snes,&dm);CHKERRQ(ierr); 2159be95d8f1SBarry Smith ierr = DMSNESSetNGS(dm,f,ctx);CHKERRQ(ierr); 2160646217ecSPeter Brune PetscFunctionReturn(0); 2161646217ecSPeter Brune } 2162646217ecSPeter Brune 216325acbd8eSLisandro Dalcin PetscErrorCode SNESPicardComputeFunction(SNES snes,Vec x,Vec f,void *ctx) 21648b0a5094SBarry Smith { 21658b0a5094SBarry Smith PetscErrorCode ierr; 2166e03ab78fSPeter Brune DM dm; 2167942e3340SBarry Smith DMSNES sdm; 21686cab3a1bSJed Brown 21698b0a5094SBarry Smith PetscFunctionBegin; 2170e03ab78fSPeter Brune ierr = SNESGetDM(snes,&dm);CHKERRQ(ierr); 2171942e3340SBarry Smith ierr = DMGetDMSNES(dm,&sdm);CHKERRQ(ierr); 217225acbd8eSLisandro Dalcin if (!sdm->ops->computepfunction) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE, "Must call SNESSetPicard() to provide Picard function."); 217325acbd8eSLisandro Dalcin if (!sdm->ops->computepjacobian) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE, "Must call SNESSetPicard() to provide Picard Jacobian."); 21748b0a5094SBarry Smith /* A(x)*x - b(x) */ 217525acbd8eSLisandro Dalcin PetscStackPush("SNES Picard user function"); 217622c6f798SBarry Smith ierr = (*sdm->ops->computepfunction)(snes,x,f,sdm->pctx);CHKERRQ(ierr); 217725acbd8eSLisandro Dalcin PetscStackPop; 217825acbd8eSLisandro Dalcin PetscStackPush("SNES Picard user Jacobian"); 2179d1e9a80fSBarry Smith ierr = (*sdm->ops->computepjacobian)(snes,x,snes->jacobian,snes->jacobian_pre,sdm->pctx);CHKERRQ(ierr); 218025acbd8eSLisandro Dalcin PetscStackPop; 21818b0a5094SBarry Smith ierr = VecScale(f,-1.0);CHKERRQ(ierr); 218295eabcedSBarry Smith ierr = MatMultAdd(snes->jacobian,x,f,f);CHKERRQ(ierr); 21838b0a5094SBarry Smith PetscFunctionReturn(0); 21848b0a5094SBarry Smith } 21858b0a5094SBarry Smith 218625acbd8eSLisandro Dalcin PetscErrorCode SNESPicardComputeJacobian(SNES snes,Vec x1,Mat J,Mat B,void *ctx) 21878b0a5094SBarry Smith { 21888b0a5094SBarry Smith PetscFunctionBegin; 2189e03ab78fSPeter Brune /* the jacobian matrix should be pre-filled in SNESPicardComputeFunction */ 21908b0a5094SBarry Smith PetscFunctionReturn(0); 21918b0a5094SBarry Smith } 21928b0a5094SBarry Smith 21938b0a5094SBarry Smith /*@C 21940d04baf8SBarry Smith SNESSetPicard - Use SNES to solve the semilinear-system A(x) x = b(x) via a Picard type iteration (Picard linearization) 21958b0a5094SBarry Smith 21968b0a5094SBarry Smith Logically Collective on SNES 21978b0a5094SBarry Smith 21988b0a5094SBarry Smith Input Parameters: 21998b0a5094SBarry Smith + snes - the SNES context 22008b0a5094SBarry Smith . r - vector to store function value 2201f8b49ee9SBarry Smith . b - function evaluation routine 2202e5d3d808SBarry Smith . Amat - matrix with which A(x) x - b(x) is to be computed 2203e5d3d808SBarry Smith . Pmat - matrix from which preconditioner is computed (usually the same as Amat) 2204411c0326SBarry Smith . J - function to compute matrix value, see SNESJacobianFunction for details on its calling sequence 22058b0a5094SBarry Smith - ctx - [optional] user-defined context for private data for the 22060298fd71SBarry Smith function evaluation routine (may be NULL) 22078b0a5094SBarry Smith 22088b0a5094SBarry Smith Notes: 2209f450aa47SBarry 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 2210f450aa47SBarry 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. 2211f450aa47SBarry Smith 22128b0a5094SBarry Smith One can call SNESSetPicard() or SNESSetFunction() (and possibly SNESSetJacobian()) but cannot call both 22138b0a5094SBarry Smith 22148b0a5094SBarry 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} 22158b0a5094SBarry 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. 22168b0a5094SBarry Smith 22178b0a5094SBarry Smith Run with -snes_mf_operator to solve the system with Newton's method using A(x^{n}) to construct the preconditioner. 22188b0a5094SBarry Smith 22190d04baf8SBarry Smith We implement the defect correction form of the Picard iteration because it converges much more generally when inexact linear solvers are used then 22200d04baf8SBarry Smith the direct Picard iteration A(x^n) x^{n+1} = b(x^n) 22218b0a5094SBarry Smith 22228b0a5094SBarry 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 22238b0a5094SBarry 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 22248b0a5094SBarry Smith different please contact us at petsc-dev@mcs.anl.gov and we'll have an entirely new argument :-). 22258b0a5094SBarry Smith 2226f450aa47SBarry Smith Level: intermediate 22278b0a5094SBarry Smith 2228411c0326SBarry Smith .seealso: SNESGetFunction(), SNESSetFunction(), SNESComputeFunction(), SNESSetJacobian(), SNESGetPicard(), SNESLineSearchPreCheckPicard(), SNESJacobianFunction 22298b0a5094SBarry Smith @*/ 2230d1e9a80fSBarry 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) 22318b0a5094SBarry Smith { 22328b0a5094SBarry Smith PetscErrorCode ierr; 2233e03ab78fSPeter Brune DM dm; 2234e03ab78fSPeter Brune 22358b0a5094SBarry Smith PetscFunctionBegin; 22368b0a5094SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 2237e03ab78fSPeter Brune ierr = SNESGetDM(snes, &dm);CHKERRQ(ierr); 2238f8b49ee9SBarry Smith ierr = DMSNESSetPicard(dm,b,J,ctx);CHKERRQ(ierr); 22398b0a5094SBarry Smith ierr = SNESSetFunction(snes,r,SNESPicardComputeFunction,ctx);CHKERRQ(ierr); 2240e5d3d808SBarry Smith ierr = SNESSetJacobian(snes,Amat,Pmat,SNESPicardComputeJacobian,ctx);CHKERRQ(ierr); 22418b0a5094SBarry Smith PetscFunctionReturn(0); 22428b0a5094SBarry Smith } 22438b0a5094SBarry Smith 22447971a8bfSPeter Brune /*@C 22457971a8bfSPeter Brune SNESGetPicard - Returns the context for the Picard iteration 22467971a8bfSPeter Brune 22477971a8bfSPeter Brune Not Collective, but Vec is parallel if SNES is parallel. Collective if Vec is requested, but has not been created yet. 22487971a8bfSPeter Brune 22497971a8bfSPeter Brune Input Parameter: 22507971a8bfSPeter Brune . snes - the SNES context 22517971a8bfSPeter Brune 22527971a8bfSPeter Brune Output Parameter: 22530298fd71SBarry Smith + r - the function (or NULL) 2254f8b49ee9SBarry Smith . f - the function (or NULL); see SNESFunction for calling sequence details 2255e4357dc4SBarry Smith . Amat - the matrix used to defined the operation A(x) x - b(x) (or NULL) 2256e4357dc4SBarry Smith . Pmat - the matrix from which the preconditioner will be constructed (or NULL) 2257f8b49ee9SBarry Smith . J - the function for matrix evaluation (or NULL); see SNESJacobianFunction for calling sequence details 22580298fd71SBarry Smith - ctx - the function context (or NULL) 22597971a8bfSPeter Brune 22607971a8bfSPeter Brune Level: advanced 22617971a8bfSPeter Brune 2262e4357dc4SBarry Smith .seealso: SNESSetPicard(), SNESGetFunction(), SNESGetJacobian(), SNESGetDM(), SNESFunction, SNESJacobianFunction 22637971a8bfSPeter Brune @*/ 2264d1e9a80fSBarry 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) 22657971a8bfSPeter Brune { 22667971a8bfSPeter Brune PetscErrorCode ierr; 22677971a8bfSPeter Brune DM dm; 22687971a8bfSPeter Brune 22697971a8bfSPeter Brune PetscFunctionBegin; 22707971a8bfSPeter Brune PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 22710298fd71SBarry Smith ierr = SNESGetFunction(snes,r,NULL,NULL);CHKERRQ(ierr); 2272e4357dc4SBarry Smith ierr = SNESGetJacobian(snes,Amat,Pmat,NULL,NULL);CHKERRQ(ierr); 22737971a8bfSPeter Brune ierr = SNESGetDM(snes,&dm);CHKERRQ(ierr); 2274f8b49ee9SBarry Smith ierr = DMSNESGetPicard(dm,f,J,ctx);CHKERRQ(ierr); 22757971a8bfSPeter Brune PetscFunctionReturn(0); 22767971a8bfSPeter Brune } 22777971a8bfSPeter Brune 2278d25893d9SBarry Smith /*@C 2279d25893d9SBarry Smith SNESSetComputeInitialGuess - Sets a routine used to compute an initial guess for the problem 2280d25893d9SBarry Smith 2281d25893d9SBarry Smith Logically Collective on SNES 2282d25893d9SBarry Smith 2283d25893d9SBarry Smith Input Parameters: 2284d25893d9SBarry Smith + snes - the SNES context 2285d25893d9SBarry Smith . func - function evaluation routine 2286d25893d9SBarry Smith - ctx - [optional] user-defined context for private data for the 22870298fd71SBarry Smith function evaluation routine (may be NULL) 2288d25893d9SBarry Smith 2289d25893d9SBarry Smith Calling sequence of func: 2290d25893d9SBarry Smith $ func (SNES snes,Vec x,void *ctx); 2291d25893d9SBarry Smith 2292d25893d9SBarry Smith . f - function vector 2293d25893d9SBarry Smith - ctx - optional user-defined function context 2294d25893d9SBarry Smith 2295d25893d9SBarry Smith Level: intermediate 2296d25893d9SBarry Smith 2297d25893d9SBarry Smith .seealso: SNESGetFunction(), SNESComputeFunction(), SNESSetJacobian() 2298d25893d9SBarry Smith @*/ 2299d25893d9SBarry Smith PetscErrorCode SNESSetComputeInitialGuess(SNES snes,PetscErrorCode (*func)(SNES,Vec,void*),void *ctx) 2300d25893d9SBarry Smith { 2301d25893d9SBarry Smith PetscFunctionBegin; 2302d25893d9SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 2303d25893d9SBarry Smith if (func) snes->ops->computeinitialguess = func; 2304d25893d9SBarry Smith if (ctx) snes->initialguessP = ctx; 2305d25893d9SBarry Smith PetscFunctionReturn(0); 2306d25893d9SBarry Smith } 2307d25893d9SBarry Smith 23083ab0aad5SBarry Smith /* --------------------------------------------------------------- */ 23091096aae1SMatthew Knepley /*@C 23101096aae1SMatthew Knepley SNESGetRhs - Gets the vector for solving F(x) = rhs. If rhs is not set 23111096aae1SMatthew Knepley it assumes a zero right hand side. 23121096aae1SMatthew Knepley 23133f9fe445SBarry Smith Logically Collective on SNES 23141096aae1SMatthew Knepley 23151096aae1SMatthew Knepley Input Parameter: 23161096aae1SMatthew Knepley . snes - the SNES context 23171096aae1SMatthew Knepley 23181096aae1SMatthew Knepley Output Parameter: 23190298fd71SBarry Smith . rhs - the right hand side vector or NULL if the right hand side vector is null 23201096aae1SMatthew Knepley 23211096aae1SMatthew Knepley Level: intermediate 23221096aae1SMatthew Knepley 232385385478SLisandro Dalcin .seealso: SNESGetSolution(), SNESGetFunction(), SNESComputeFunction(), SNESSetJacobian(), SNESSetFunction() 23241096aae1SMatthew Knepley @*/ 23257087cfbeSBarry Smith PetscErrorCode SNESGetRhs(SNES snes,Vec *rhs) 23261096aae1SMatthew Knepley { 23271096aae1SMatthew Knepley PetscFunctionBegin; 23280700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 23291096aae1SMatthew Knepley PetscValidPointer(rhs,2); 233085385478SLisandro Dalcin *rhs = snes->vec_rhs; 23311096aae1SMatthew Knepley PetscFunctionReturn(0); 23321096aae1SMatthew Knepley } 23331096aae1SMatthew Knepley 23349b94acceSBarry Smith /*@ 2335bf388a1fSBarry Smith SNESComputeFunction - Calls the function that has been set with SNESSetFunction(). 23369b94acceSBarry Smith 2337c7afd0dbSLois Curfman McInnes Collective on SNES 2338c7afd0dbSLois Curfman McInnes 23399b94acceSBarry Smith Input Parameters: 2340c7afd0dbSLois Curfman McInnes + snes - the SNES context 2341c7afd0dbSLois Curfman McInnes - x - input vector 23429b94acceSBarry Smith 23439b94acceSBarry Smith Output Parameter: 23443638b69dSLois Curfman McInnes . y - function vector, as set by SNESSetFunction() 23459b94acceSBarry Smith 23461bffabb2SLois Curfman McInnes Notes: 234736851e7fSLois Curfman McInnes SNESComputeFunction() is typically used within nonlinear solvers 234836851e7fSLois Curfman McInnes implementations, so most users would not generally call this routine 234936851e7fSLois Curfman McInnes themselves. 235036851e7fSLois Curfman McInnes 235136851e7fSLois Curfman McInnes Level: developer 235236851e7fSLois Curfman McInnes 2353a86d99e1SLois Curfman McInnes .seealso: SNESSetFunction(), SNESGetFunction() 23549b94acceSBarry Smith @*/ 23557087cfbeSBarry Smith PetscErrorCode SNESComputeFunction(SNES snes,Vec x,Vec y) 23569b94acceSBarry Smith { 2357dfbe8321SBarry Smith PetscErrorCode ierr; 23586cab3a1bSJed Brown DM dm; 2359942e3340SBarry Smith DMSNES sdm; 23609b94acceSBarry Smith 23613a40ed3dSBarry Smith PetscFunctionBegin; 23620700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 23630700a824SBarry Smith PetscValidHeaderSpecific(x,VEC_CLASSID,2); 23640700a824SBarry Smith PetscValidHeaderSpecific(y,VEC_CLASSID,3); 2365c9780b6fSBarry Smith PetscCheckSameComm(snes,1,x,2); 2366c9780b6fSBarry Smith PetscCheckSameComm(snes,1,y,3); 236762796dfbSBarry Smith ierr = VecValidValues(x,2,PETSC_TRUE);CHKERRQ(ierr); 2368184914b5SBarry Smith 23696cab3a1bSJed Brown ierr = SNESGetDM(snes,&dm);CHKERRQ(ierr); 2370942e3340SBarry Smith ierr = DMGetDMSNES(dm,&sdm);CHKERRQ(ierr); 237132f3f7c2SPeter Brune if (sdm->ops->computefunction) { 237294db00ebSBarry Smith if (sdm->ops->computefunction != SNESObjectiveComputeFunctionDefaultFD) { 2373ccf3c845SPeter Brune ierr = PetscLogEventBegin(SNES_FunctionEval,snes,x,y,0);CHKERRQ(ierr); 237494db00ebSBarry Smith } 23758860a134SJunchao Zhang ierr = VecLockReadPush(x);CHKERRQ(ierr); 2376d64ed03dSBarry Smith PetscStackPush("SNES user function"); 23778ddeebeaSSteve Benbow /* ensure domainerror is false prior to computefunction evaluation (may not have been reset) */ 23788ddeebeaSSteve Benbow snes->domainerror = PETSC_FALSE; 237922c6f798SBarry Smith ierr = (*sdm->ops->computefunction)(snes,x,y,sdm->functionctx);CHKERRQ(ierr); 2380d64ed03dSBarry Smith PetscStackPop; 23818860a134SJunchao Zhang ierr = VecLockReadPop(x);CHKERRQ(ierr); 238294db00ebSBarry Smith if (sdm->ops->computefunction != SNESObjectiveComputeFunctionDefaultFD) { 2383ccf3c845SPeter Brune ierr = PetscLogEventEnd(SNES_FunctionEval,snes,x,y,0);CHKERRQ(ierr); 238494db00ebSBarry Smith } 2385c90fad12SPeter Brune } else if (snes->vec_rhs) { 2386c90fad12SPeter Brune ierr = MatMult(snes->jacobian, x, y);CHKERRQ(ierr); 2387644e2e5bSBarry Smith } else SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE, "Must call SNESSetFunction() or SNESSetDM() before SNESComputeFunction(), likely called from SNESSolve()."); 238885385478SLisandro Dalcin if (snes->vec_rhs) { 238985385478SLisandro Dalcin ierr = VecAXPY(y,-1.0,snes->vec_rhs);CHKERRQ(ierr); 23903ab0aad5SBarry Smith } 2391ae3c334cSLois Curfman McInnes snes->nfuncs++; 2392422a814eSBarry Smith /* 2393422a814eSBarry Smith domainerror might not be set on all processes; so we tag vector locally with Inf and the next inner product or norm will 2394422a814eSBarry Smith propagate the value to all processes 2395422a814eSBarry Smith */ 2396422a814eSBarry Smith if (snes->domainerror) { 2397422a814eSBarry Smith ierr = VecSetInf(y);CHKERRQ(ierr); 2398422a814eSBarry Smith } 23993a40ed3dSBarry Smith PetscFunctionReturn(0); 24009b94acceSBarry Smith } 24019b94acceSBarry Smith 2402c79ef259SPeter Brune /*@ 2403be95d8f1SBarry Smith SNESComputeNGS - Calls the Gauss-Seidel function that has been set with SNESSetNGS(). 2404c79ef259SPeter Brune 2405c79ef259SPeter Brune Collective on SNES 2406c79ef259SPeter Brune 2407c79ef259SPeter Brune Input Parameters: 2408c79ef259SPeter Brune + snes - the SNES context 2409c79ef259SPeter Brune . x - input vector 2410c79ef259SPeter Brune - b - rhs vector 2411c79ef259SPeter Brune 2412c79ef259SPeter Brune Output Parameter: 2413c79ef259SPeter Brune . x - new solution vector 2414c79ef259SPeter Brune 2415c79ef259SPeter Brune Notes: 2416be95d8f1SBarry Smith SNESComputeNGS() is typically used within composed nonlinear solver 2417c79ef259SPeter Brune implementations, so most users would not generally call this routine 2418c79ef259SPeter Brune themselves. 2419c79ef259SPeter Brune 2420c79ef259SPeter Brune Level: developer 2421c79ef259SPeter Brune 2422be95d8f1SBarry Smith .seealso: SNESSetNGS(), SNESComputeFunction() 2423c79ef259SPeter Brune @*/ 2424be95d8f1SBarry Smith PetscErrorCode SNESComputeNGS(SNES snes,Vec b,Vec x) 2425646217ecSPeter Brune { 2426646217ecSPeter Brune PetscErrorCode ierr; 24276cab3a1bSJed Brown DM dm; 2428942e3340SBarry Smith DMSNES sdm; 2429646217ecSPeter Brune 2430646217ecSPeter Brune PetscFunctionBegin; 2431646217ecSPeter Brune PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 2432646217ecSPeter Brune PetscValidHeaderSpecific(x,VEC_CLASSID,2); 2433646217ecSPeter Brune if (b) PetscValidHeaderSpecific(b,VEC_CLASSID,3); 2434646217ecSPeter Brune PetscCheckSameComm(snes,1,x,2); 2435646217ecSPeter Brune if (b) PetscCheckSameComm(snes,1,b,3); 243662796dfbSBarry Smith if (b) {ierr = VecValidValues(b,2,PETSC_TRUE);CHKERRQ(ierr);} 2437be95d8f1SBarry Smith ierr = PetscLogEventBegin(SNES_NGSEval,snes,x,b,0);CHKERRQ(ierr); 24386cab3a1bSJed Brown ierr = SNESGetDM(snes,&dm);CHKERRQ(ierr); 2439942e3340SBarry Smith ierr = DMGetDMSNES(dm,&sdm);CHKERRQ(ierr); 244022c6f798SBarry Smith if (sdm->ops->computegs) { 24418860a134SJunchao Zhang if (b) {ierr = VecLockReadPush(b);CHKERRQ(ierr);} 2442be95d8f1SBarry Smith PetscStackPush("SNES user NGS"); 244322c6f798SBarry Smith ierr = (*sdm->ops->computegs)(snes,x,b,sdm->gsctx);CHKERRQ(ierr); 2444646217ecSPeter Brune PetscStackPop; 24458860a134SJunchao Zhang if (b) {ierr = VecLockReadPop(b);CHKERRQ(ierr);} 2446be95d8f1SBarry Smith } else SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE, "Must call SNESSetNGS() before SNESComputeNGS(), likely called from SNESSolve()."); 2447be95d8f1SBarry Smith ierr = PetscLogEventEnd(SNES_NGSEval,snes,x,b,0);CHKERRQ(ierr); 2448646217ecSPeter Brune PetscFunctionReturn(0); 2449646217ecSPeter Brune } 2450646217ecSPeter Brune 2451e885f1abSBarry Smith PetscErrorCode SNESTestJacobian(SNES snes) 2452e885f1abSBarry Smith { 245312837594SBarry Smith Mat A,B,C,D,jacobian; 2454e885f1abSBarry Smith Vec x = snes->vec_sol,f = snes->vec_func; 2455e885f1abSBarry Smith PetscErrorCode ierr; 2456e885f1abSBarry Smith PetscReal nrm,gnorm; 245781e7118cSBarry Smith PetscReal threshold = 1.e-5; 24580e276705SLisandro Dalcin MatType mattype; 2459e885f1abSBarry Smith PetscInt m,n,M,N; 2460e885f1abSBarry Smith void *functx; 24612cd624f9SStefano Zampini PetscBool complete_print = PETSC_FALSE,threshold_print = PETSC_FALSE,test = PETSC_FALSE,flg,istranspose; 24623325ff46SBarry Smith PetscViewer viewer,mviewer; 2463e885f1abSBarry Smith MPI_Comm comm; 2464e885f1abSBarry Smith PetscInt tabs; 246512837594SBarry Smith static PetscBool directionsprinted = PETSC_FALSE; 24663325ff46SBarry Smith PetscViewerFormat format; 2467e885f1abSBarry Smith 2468e885f1abSBarry Smith PetscFunctionBegin; 2469fc35ed60SBarry Smith ierr = PetscObjectOptionsBegin((PetscObject)snes);CHKERRQ(ierr); 247012837594SBarry Smith ierr = PetscOptionsName("-snes_test_jacobian","Compare hand-coded and finite difference Jacobians","None",&test);CHKERRQ(ierr); 247112837594SBarry Smith ierr = PetscOptionsReal("-snes_test_jacobian", "Threshold for element difference between hand-coded and finite difference being meaningful", "None", threshold, &threshold,NULL);CHKERRQ(ierr); 24723325ff46SBarry Smith ierr = PetscOptionsViewer("-snes_test_jacobian_view","View difference between hand-coded and finite difference Jacobians element entries","None",&mviewer,&format,&complete_print);CHKERRQ(ierr); 247318d89885SKarl Rupp if (!complete_print) { 2474455a5933SJed Brown ierr = PetscOptionsDeprecated("-snes_test_jacobian_display","-snes_test_jacobian_view","3.13",NULL);CHKERRQ(ierr); 247518d89885SKarl Rupp ierr = PetscOptionsViewer("-snes_test_jacobian_display","Display difference between hand-coded and finite difference Jacobians","None",&mviewer,&format,&complete_print);CHKERRQ(ierr); 247618d89885SKarl Rupp } 247718d89885SKarl Rupp /* for compatibility with PETSc 3.9 and older. */ 2478455a5933SJed Brown ierr = PetscOptionsDeprecated("-snes_test_jacobian_display_threshold","-snes_test_jacobian","3.13","-snes_test_jacobian accepts an optional threshold (since v3.10)");CHKERRQ(ierr); 247918d89885SKarl Rupp ierr = PetscOptionsReal("-snes_test_jacobian_display_threshold", "Display difference between hand-coded and finite difference Jacobians which exceed input threshold", "None", threshold, &threshold, &threshold_print);CHKERRQ(ierr); 2480e885f1abSBarry Smith ierr = PetscOptionsEnd();CHKERRQ(ierr); 2481e885f1abSBarry Smith if (!test) PetscFunctionReturn(0); 2482e885f1abSBarry Smith 2483e885f1abSBarry Smith ierr = PetscObjectGetComm((PetscObject)snes,&comm);CHKERRQ(ierr); 2484e885f1abSBarry Smith ierr = PetscViewerASCIIGetStdout(comm,&viewer);CHKERRQ(ierr); 2485e885f1abSBarry Smith ierr = PetscViewerASCIIGetTab(viewer, &tabs);CHKERRQ(ierr); 2486e885f1abSBarry Smith ierr = PetscViewerASCIISetTab(viewer, ((PetscObject)snes)->tablevel);CHKERRQ(ierr); 248712837594SBarry Smith ierr = PetscViewerASCIIPrintf(viewer," ---------- Testing Jacobian -------------\n");CHKERRQ(ierr); 248812837594SBarry Smith if (!complete_print && !directionsprinted) { 248912837594SBarry Smith ierr = PetscViewerASCIIPrintf(viewer," Run with -snes_test_jacobian_view and optionally -snes_test_jacobian <threshold> to show difference\n");CHKERRQ(ierr); 2490fc35ed60SBarry Smith ierr = PetscViewerASCIIPrintf(viewer," of hand-coded and finite difference Jacobian entries greater than <threshold>.\n");CHKERRQ(ierr); 249112837594SBarry Smith } 249212837594SBarry Smith if (!directionsprinted) { 249312837594SBarry Smith ierr = PetscViewerASCIIPrintf(viewer," Testing hand-coded Jacobian, if (for double precision runs) ||J - Jfd||_F/||J||_F is\n");CHKERRQ(ierr); 2494e885f1abSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," O(1.e-8), the hand-coded Jacobian is probably correct.\n");CHKERRQ(ierr); 249512837594SBarry Smith directionsprinted = PETSC_TRUE; 2496e885f1abSBarry Smith } 24973325ff46SBarry Smith if (complete_print) { 24983325ff46SBarry Smith ierr = PetscViewerPushFormat(mviewer,format);CHKERRQ(ierr); 2499e885f1abSBarry Smith } 2500e885f1abSBarry Smith 250112837594SBarry Smith ierr = PetscObjectTypeCompare((PetscObject)snes->jacobian,MATMFFD,&flg);CHKERRQ(ierr); 250212837594SBarry Smith if (!flg) jacobian = snes->jacobian; 250312837594SBarry Smith else jacobian = snes->jacobian_pre; 250412837594SBarry Smith 2505a82339d0SMatthew G. Knepley if (!x) { 2506a82339d0SMatthew G. Knepley ierr = MatCreateVecs(jacobian, &x, NULL);CHKERRQ(ierr); 2507a82339d0SMatthew G. Knepley } else { 2508a82339d0SMatthew G. Knepley ierr = PetscObjectReference((PetscObject) x);CHKERRQ(ierr); 2509a82339d0SMatthew G. Knepley } 2510a82339d0SMatthew G. Knepley if (!f) { 2511a82339d0SMatthew G. Knepley ierr = VecDuplicate(x, &f);CHKERRQ(ierr); 2512a82339d0SMatthew G. Knepley } else { 2513a82339d0SMatthew G. Knepley ierr = PetscObjectReference((PetscObject) f);CHKERRQ(ierr); 2514a82339d0SMatthew G. Knepley } 2515a82339d0SMatthew G. Knepley /* evaluate the function at this point because SNESComputeJacobianDefault() assumes that the function has been evaluated and put into snes->vec_func */ 2516a82339d0SMatthew G. Knepley ierr = SNESComputeFunction(snes,x,f);CHKERRQ(ierr); 2517a82339d0SMatthew G. Knepley ierr = VecDestroy(&f);CHKERRQ(ierr); 25182cd624f9SStefano Zampini ierr = PetscObjectTypeCompare((PetscObject)snes,SNESKSPTRANSPOSEONLY,&istranspose);CHKERRQ(ierr); 251912837594SBarry Smith while (jacobian) { 25202cd624f9SStefano Zampini Mat JT = NULL, Jsave = NULL; 25212cd624f9SStefano Zampini 25222cd624f9SStefano Zampini if (istranspose) { 25232cd624f9SStefano Zampini ierr = MatCreateTranspose(jacobian,&JT);CHKERRQ(ierr); 25242cd624f9SStefano Zampini Jsave = jacobian; 25252cd624f9SStefano Zampini jacobian = JT; 25262cd624f9SStefano Zampini } 25270e276705SLisandro Dalcin ierr = PetscObjectBaseTypeCompareAny((PetscObject)jacobian,&flg,MATSEQAIJ,MATMPIAIJ,MATSEQDENSE,MATMPIDENSE,MATSEQBAIJ,MATMPIBAIJ,MATSEQSBAIJ,MATMPISBAIJ,"");CHKERRQ(ierr); 252812837594SBarry Smith if (flg) { 252912837594SBarry Smith A = jacobian; 253012837594SBarry Smith ierr = PetscObjectReference((PetscObject)A);CHKERRQ(ierr); 253112837594SBarry Smith } else { 25320bacdadaSStefano Zampini ierr = MatComputeOperator(jacobian,MATAIJ,&A);CHKERRQ(ierr); 253312837594SBarry Smith } 2534e885f1abSBarry Smith 25350e276705SLisandro Dalcin ierr = MatGetType(A,&mattype);CHKERRQ(ierr); 2536e885f1abSBarry Smith ierr = MatGetSize(A,&M,&N);CHKERRQ(ierr); 2537e885f1abSBarry Smith ierr = MatGetLocalSize(A,&m,&n);CHKERRQ(ierr); 25380e276705SLisandro Dalcin ierr = MatCreate(PetscObjectComm((PetscObject)A),&B);CHKERRQ(ierr); 25390e276705SLisandro Dalcin ierr = MatSetType(B,mattype);CHKERRQ(ierr); 2540e885f1abSBarry Smith ierr = MatSetSizes(B,m,n,M,N);CHKERRQ(ierr); 25410e276705SLisandro Dalcin ierr = MatSetBlockSizesFromMats(B,A,A);CHKERRQ(ierr); 2542e885f1abSBarry Smith ierr = MatSetUp(B);CHKERRQ(ierr); 2543e885f1abSBarry Smith ierr = MatSetOption(B,MAT_NEW_NONZERO_ALLOCATION_ERR,PETSC_FALSE);CHKERRQ(ierr); 2544e885f1abSBarry Smith 2545e885f1abSBarry Smith ierr = SNESGetFunction(snes,NULL,NULL,&functx);CHKERRQ(ierr); 2546e885f1abSBarry Smith ierr = SNESComputeJacobianDefault(snes,x,B,B,functx);CHKERRQ(ierr); 254712837594SBarry Smith 254812837594SBarry Smith ierr = MatDuplicate(B,MAT_COPY_VALUES,&D);CHKERRQ(ierr); 254912837594SBarry Smith ierr = MatAYPX(D,-1.0,A,DIFFERENT_NONZERO_PATTERN);CHKERRQ(ierr); 255012837594SBarry Smith ierr = MatNorm(D,NORM_FROBENIUS,&nrm);CHKERRQ(ierr); 2551e885f1abSBarry Smith ierr = MatNorm(A,NORM_FROBENIUS,&gnorm);CHKERRQ(ierr); 255212837594SBarry Smith ierr = MatDestroy(&D);CHKERRQ(ierr); 255312837594SBarry Smith if (!gnorm) gnorm = 1; /* just in case */ 255412837594SBarry Smith ierr = PetscViewerASCIIPrintf(viewer," ||J - Jfd||_F/||J||_F = %g, ||J - Jfd||_F = %g\n",(double)(nrm/gnorm),(double)nrm);CHKERRQ(ierr); 255512837594SBarry Smith 2556e885f1abSBarry Smith if (complete_print) { 255712837594SBarry Smith ierr = PetscViewerASCIIPrintf(viewer," Hand-coded Jacobian ----------\n");CHKERRQ(ierr); 25581878987eSStefano Zampini ierr = MatView(A,mviewer);CHKERRQ(ierr); 255912837594SBarry Smith ierr = PetscViewerASCIIPrintf(viewer," Finite difference Jacobian ----------\n");CHKERRQ(ierr); 25603325ff46SBarry Smith ierr = MatView(B,mviewer);CHKERRQ(ierr); 2561e885f1abSBarry Smith } 2562e885f1abSBarry Smith 2563df10fb39SFande Kong if (threshold_print || complete_print) { 2564e885f1abSBarry Smith PetscInt Istart, Iend, *ccols, bncols, cncols, j, row; 2565e885f1abSBarry Smith PetscScalar *cvals; 2566e885f1abSBarry Smith const PetscInt *bcols; 2567e885f1abSBarry Smith const PetscScalar *bvals; 2568e885f1abSBarry Smith 2569e885f1abSBarry Smith ierr = MatCreate(PetscObjectComm((PetscObject)A),&C);CHKERRQ(ierr); 25700e276705SLisandro Dalcin ierr = MatSetType(C,mattype);CHKERRQ(ierr); 2571e885f1abSBarry Smith ierr = MatSetSizes(C,m,n,M,N);CHKERRQ(ierr); 25720e276705SLisandro Dalcin ierr = MatSetBlockSizesFromMats(C,A,A);CHKERRQ(ierr); 2573e885f1abSBarry Smith ierr = MatSetUp(C);CHKERRQ(ierr); 2574e885f1abSBarry Smith ierr = MatSetOption(C,MAT_NEW_NONZERO_ALLOCATION_ERR,PETSC_FALSE);CHKERRQ(ierr); 25750e276705SLisandro Dalcin 25760e276705SLisandro Dalcin ierr = MatAYPX(B,-1.0,A,DIFFERENT_NONZERO_PATTERN);CHKERRQ(ierr); 2577e885f1abSBarry Smith ierr = MatGetOwnershipRange(B,&Istart,&Iend);CHKERRQ(ierr); 2578e885f1abSBarry Smith 2579e885f1abSBarry Smith for (row = Istart; row < Iend; row++) { 2580e885f1abSBarry Smith ierr = MatGetRow(B,row,&bncols,&bcols,&bvals);CHKERRQ(ierr); 2581e885f1abSBarry Smith ierr = PetscMalloc2(bncols,&ccols,bncols,&cvals);CHKERRQ(ierr); 2582e885f1abSBarry Smith for (j = 0, cncols = 0; j < bncols; j++) { 258323a52b1dSBarry Smith if (PetscAbsScalar(bvals[j]) > threshold) { 2584e885f1abSBarry Smith ccols[cncols] = bcols[j]; 2585e885f1abSBarry Smith cvals[cncols] = bvals[j]; 2586e885f1abSBarry Smith cncols += 1; 2587e885f1abSBarry Smith } 2588e885f1abSBarry Smith } 2589e885f1abSBarry Smith if (cncols) { 2590e885f1abSBarry Smith ierr = MatSetValues(C,1,&row,cncols,ccols,cvals,INSERT_VALUES);CHKERRQ(ierr); 2591e885f1abSBarry Smith } 2592e885f1abSBarry Smith ierr = MatRestoreRow(B,row,&bncols,&bcols,&bvals);CHKERRQ(ierr); 2593e885f1abSBarry Smith ierr = PetscFree2(ccols,cvals);CHKERRQ(ierr); 2594e885f1abSBarry Smith } 2595e885f1abSBarry Smith ierr = MatAssemblyBegin(C,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 2596e885f1abSBarry Smith ierr = MatAssemblyEnd(C,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 259712837594SBarry Smith ierr = PetscViewerASCIIPrintf(viewer," Hand-coded minus finite-difference Jacobian with tolerance %g ----------\n",(double)threshold);CHKERRQ(ierr); 259818d89885SKarl Rupp ierr = MatView(C,complete_print ? mviewer : viewer);CHKERRQ(ierr); 2599e885f1abSBarry Smith ierr = MatDestroy(&C);CHKERRQ(ierr); 2600e885f1abSBarry Smith } 260112837594SBarry Smith ierr = MatDestroy(&A);CHKERRQ(ierr); 2602e885f1abSBarry Smith ierr = MatDestroy(&B);CHKERRQ(ierr); 26032cd624f9SStefano Zampini ierr = MatDestroy(&JT);CHKERRQ(ierr); 26042cd624f9SStefano Zampini if (Jsave) jacobian = Jsave; 260512837594SBarry Smith if (jacobian != snes->jacobian_pre) { 260612837594SBarry Smith jacobian = snes->jacobian_pre; 260712837594SBarry Smith ierr = PetscViewerASCIIPrintf(viewer," ---------- Testing Jacobian for preconditioner -------------\n");CHKERRQ(ierr); 260812837594SBarry Smith } 260912837594SBarry Smith else jacobian = NULL; 261012837594SBarry Smith } 2611a82339d0SMatthew G. Knepley ierr = VecDestroy(&x);CHKERRQ(ierr); 26123325ff46SBarry Smith if (complete_print) { 26133325ff46SBarry Smith ierr = PetscViewerPopFormat(mviewer);CHKERRQ(ierr); 26143325ff46SBarry Smith } 2615a0c90127SFande Kong if (mviewer) { ierr = PetscViewerDestroy(&mviewer);CHKERRQ(ierr); } 2616e885f1abSBarry Smith ierr = PetscViewerASCIISetTab(viewer,tabs);CHKERRQ(ierr); 2617e885f1abSBarry Smith PetscFunctionReturn(0); 2618e885f1abSBarry Smith } 2619e885f1abSBarry Smith 262062fef451SLois Curfman McInnes /*@ 2621bf388a1fSBarry Smith SNESComputeJacobian - Computes the Jacobian matrix that has been set with SNESSetJacobian(). 262262fef451SLois Curfman McInnes 2623d083f849SBarry Smith Collective on SNES 2624c7afd0dbSLois Curfman McInnes 262562fef451SLois Curfman McInnes Input Parameters: 2626c7afd0dbSLois Curfman McInnes + snes - the SNES context 2627c7afd0dbSLois Curfman McInnes - x - input vector 262862fef451SLois Curfman McInnes 262962fef451SLois Curfman McInnes Output Parameters: 2630c7afd0dbSLois Curfman McInnes + A - Jacobian matrix 2631d1e9a80fSBarry Smith - B - optional preconditioning matrix 2632fee21e36SBarry Smith 2633e35cf81dSBarry Smith Options Database Keys: 2634e35cf81dSBarry Smith + -snes_lag_preconditioner <lag> 2635693365a8SJed Brown . -snes_lag_jacobian <lag> 2636455a5933SJed Brown . -snes_test_jacobian <optional threshold> - compare the user provided Jacobian with one compute via finite differences to check for errors. If a threshold is given, display only those entries whose difference is greater than the threshold. 2637455a5933SJed Brown . -snes_test_jacobian_view - display the user provided Jacobian, the finite difference Jacobian and the difference between them to help users detect the location of errors in the user provided Jacobian 2638693365a8SJed Brown . -snes_compare_explicit - Compare the computed Jacobian to the finite difference Jacobian and output the differences 2639693365a8SJed Brown . -snes_compare_explicit_draw - Compare the computed Jacobian to the finite difference Jacobian and draw the result 2640693365a8SJed Brown . -snes_compare_explicit_contour - Compare the computed Jacobian to the finite difference Jacobian and draw a contour plot with the result 26414c30e9fbSJed Brown . -snes_compare_operator - Make the comparison options above use the operator instead of the preconditioning matrix 264294d6a431SBarry Smith . -snes_compare_coloring - Compute the finite difference Jacobian using coloring and display norms of difference 2643c01495d3SJed Brown . -snes_compare_coloring_display - Compute the finite differece Jacobian using coloring and display verbose differences 2644c01495d3SJed Brown . -snes_compare_coloring_threshold - Display only those matrix entries that differ by more than a given threshold 2645c01495d3SJed Brown . -snes_compare_coloring_threshold_atol - Absolute tolerance for difference in matrix entries to be displayed by -snes_compare_coloring_threshold 2646c01495d3SJed Brown . -snes_compare_coloring_threshold_rtol - Relative tolerance for difference in matrix entries to be displayed by -snes_compare_coloring_threshold 26474c30e9fbSJed Brown . -snes_compare_coloring_draw - Compute the finite differece Jacobian using coloring and draw differences 2648c01495d3SJed Brown - -snes_compare_coloring_draw_contour - Compute the finite differece Jacobian using coloring and show contours of matrices and differences 2649c01495d3SJed Brown 2650e35cf81dSBarry Smith 265162fef451SLois Curfman McInnes Notes: 265262fef451SLois Curfman McInnes Most users should not need to explicitly call this routine, as it 265362fef451SLois Curfman McInnes is used internally within the nonlinear solvers. 265462fef451SLois Curfman McInnes 265595452b02SPatrick Sanan Developer Notes: 265695452b02SPatrick Sanan This has duplicative ways of checking the accuracy of the user provided Jacobian (see the options above). This is for historical reasons, the routine SNESTestJacobian() use to used 2657e885f1abSBarry Smith for with the SNESType of test that has been removed. 2658e885f1abSBarry Smith 265936851e7fSLois Curfman McInnes Level: developer 266036851e7fSLois Curfman McInnes 2661e35cf81dSBarry Smith .seealso: SNESSetJacobian(), KSPSetOperators(), MatStructure, SNESSetLagPreconditioner(), SNESSetLagJacobian() 266262fef451SLois Curfman McInnes @*/ 2663d1e9a80fSBarry Smith PetscErrorCode SNESComputeJacobian(SNES snes,Vec X,Mat A,Mat B) 26649b94acceSBarry Smith { 2665dfbe8321SBarry Smith PetscErrorCode ierr; 2666ace3abfcSBarry Smith PetscBool flag; 26676cab3a1bSJed Brown DM dm; 2668942e3340SBarry Smith DMSNES sdm; 2669e0e3a89bSBarry Smith KSP ksp; 26703a40ed3dSBarry Smith 26713a40ed3dSBarry Smith PetscFunctionBegin; 26720700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 26730700a824SBarry Smith PetscValidHeaderSpecific(X,VEC_CLASSID,2); 2674c9780b6fSBarry Smith PetscCheckSameComm(snes,1,X,2); 267562796dfbSBarry Smith ierr = VecValidValues(X,2,PETSC_TRUE);CHKERRQ(ierr); 26766cab3a1bSJed Brown ierr = SNESGetDM(snes,&dm);CHKERRQ(ierr); 2677942e3340SBarry Smith ierr = DMGetDMSNES(dm,&sdm);CHKERRQ(ierr); 26783232da50SPeter Brune 2679ce94432eSBarry Smith if (!sdm->ops->computejacobian) SETERRQ(PetscObjectComm((PetscObject)snes),PETSC_ERR_USER,"Must call SNESSetJacobian(), DMSNESSetJacobian(), DMDASNESSetJacobianLocal(), etc"); 2680ebd3b9afSBarry Smith 2681ebd3b9afSBarry Smith /* make sure that MatAssemblyBegin/End() is called on A matrix if it is matrix free */ 2682ebd3b9afSBarry Smith 2683fe3ffe1eSBarry Smith if (snes->lagjacobian == -2) { 2684fe3ffe1eSBarry Smith snes->lagjacobian = -1; 2685f5af7f23SKarl Rupp 2686fe3ffe1eSBarry Smith ierr = PetscInfo(snes,"Recomputing Jacobian/preconditioner because lag is -2 (means compute Jacobian, but then never again) \n");CHKERRQ(ierr); 2687fe3ffe1eSBarry Smith } else if (snes->lagjacobian == -1) { 2688e35cf81dSBarry Smith ierr = PetscInfo(snes,"Reusing Jacobian/preconditioner because lag is -1\n");CHKERRQ(ierr); 268994ab13aaSBarry Smith ierr = PetscObjectTypeCompare((PetscObject)A,MATMFFD,&flag);CHKERRQ(ierr); 2690ebd3b9afSBarry Smith if (flag) { 269194ab13aaSBarry Smith ierr = MatAssemblyBegin(A,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 269294ab13aaSBarry Smith ierr = MatAssemblyEnd(A,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 2693ebd3b9afSBarry Smith } 2694e35cf81dSBarry Smith PetscFunctionReturn(0); 269537ec4e1aSPeter Brune } else if (snes->lagjacobian > 1 && (snes->iter + snes->jac_iter) % snes->lagjacobian) { 2696e35cf81dSBarry Smith ierr = PetscInfo2(snes,"Reusing Jacobian/preconditioner because lag is %D and SNES iteration is %D\n",snes->lagjacobian,snes->iter);CHKERRQ(ierr); 269794ab13aaSBarry Smith ierr = PetscObjectTypeCompare((PetscObject)A,MATMFFD,&flag);CHKERRQ(ierr); 2698ebd3b9afSBarry Smith if (flag) { 269994ab13aaSBarry Smith ierr = MatAssemblyBegin(A,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 270094ab13aaSBarry Smith ierr = MatAssemblyEnd(A,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 2701ebd3b9afSBarry Smith } 2702e35cf81dSBarry Smith PetscFunctionReturn(0); 2703e35cf81dSBarry Smith } 2704efd4aadfSBarry Smith if (snes->npc && snes->npcside== PC_LEFT) { 270594ab13aaSBarry Smith ierr = MatAssemblyBegin(A,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 270694ab13aaSBarry Smith ierr = MatAssemblyEnd(A,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 2707d728fb7dSPeter Brune PetscFunctionReturn(0); 2708d728fb7dSPeter Brune } 2709e35cf81dSBarry Smith 271094ab13aaSBarry Smith ierr = PetscLogEventBegin(SNES_JacobianEval,snes,X,A,B);CHKERRQ(ierr); 27118860a134SJunchao Zhang ierr = VecLockReadPush(X);CHKERRQ(ierr); 2712d64ed03dSBarry Smith PetscStackPush("SNES user Jacobian function"); 2713d1e9a80fSBarry Smith ierr = (*sdm->ops->computejacobian)(snes,X,A,B,sdm->jacobianctx);CHKERRQ(ierr); 2714d64ed03dSBarry Smith PetscStackPop; 27158860a134SJunchao Zhang ierr = VecLockReadPop(X);CHKERRQ(ierr); 271694ab13aaSBarry Smith ierr = PetscLogEventEnd(SNES_JacobianEval,snes,X,A,B);CHKERRQ(ierr); 271728d58a37SPierre Jolivet 271828d58a37SPierre Jolivet /* attach latest linearization point to the preconditioning matrix */ 271928d58a37SPierre Jolivet ierr = PetscObjectCompose((PetscObject)B,"__SNES_latest_X",(PetscObject)X);CHKERRQ(ierr); 2720a8054027SBarry Smith 2721e0e3a89bSBarry Smith /* the next line ensures that snes->ksp exists */ 2722e0e3a89bSBarry Smith ierr = SNESGetKSP(snes,&ksp);CHKERRQ(ierr); 27233b4f5425SBarry Smith if (snes->lagpreconditioner == -2) { 27243b4f5425SBarry Smith ierr = PetscInfo(snes,"Rebuilding preconditioner exactly once since lag is -2\n");CHKERRQ(ierr); 2725d1e9a80fSBarry Smith ierr = KSPSetReusePreconditioner(snes->ksp,PETSC_FALSE);CHKERRQ(ierr); 27263b4f5425SBarry Smith snes->lagpreconditioner = -1; 27273b4f5425SBarry Smith } else if (snes->lagpreconditioner == -1) { 2728a8054027SBarry Smith ierr = PetscInfo(snes,"Reusing preconditioner because lag is -1\n");CHKERRQ(ierr); 2729d1e9a80fSBarry Smith ierr = KSPSetReusePreconditioner(snes->ksp,PETSC_TRUE);CHKERRQ(ierr); 273037ec4e1aSPeter Brune } else if (snes->lagpreconditioner > 1 && (snes->iter + snes->pre_iter) % snes->lagpreconditioner) { 2731a8054027SBarry Smith ierr = PetscInfo2(snes,"Reusing preconditioner because lag is %D and SNES iteration is %D\n",snes->lagpreconditioner,snes->iter);CHKERRQ(ierr); 2732d1e9a80fSBarry Smith ierr = KSPSetReusePreconditioner(snes->ksp,PETSC_TRUE);CHKERRQ(ierr); 2733d1e9a80fSBarry Smith } else { 2734d1e9a80fSBarry Smith ierr = PetscInfo(snes,"Rebuilding preconditioner\n");CHKERRQ(ierr); 2735d1e9a80fSBarry Smith ierr = KSPSetReusePreconditioner(snes->ksp,PETSC_FALSE);CHKERRQ(ierr); 2736a8054027SBarry Smith } 2737a8054027SBarry Smith 2738e885f1abSBarry Smith ierr = SNESTestJacobian(snes);CHKERRQ(ierr); 27396d84be18SBarry Smith /* make sure user returned a correct Jacobian and preconditioner */ 274094ab13aaSBarry Smith /* PetscValidHeaderSpecific(A,MAT_CLASSID,3); 274194ab13aaSBarry Smith PetscValidHeaderSpecific(B,MAT_CLASSID,4); */ 2742693365a8SJed Brown { 2743693365a8SJed Brown PetscBool flag = PETSC_FALSE,flag_draw = PETSC_FALSE,flag_contour = PETSC_FALSE,flag_operator = PETSC_FALSE; 274416413a6aSBarry Smith ierr = PetscOptionsGetViewer(PetscObjectComm((PetscObject)snes),((PetscObject) snes)->options,((PetscObject)snes)->prefix,"-snes_compare_explicit",NULL,NULL,&flag);CHKERRQ(ierr); 274516413a6aSBarry Smith ierr = PetscOptionsGetViewer(PetscObjectComm((PetscObject)snes),((PetscObject) snes)->options,((PetscObject)snes)->prefix,"-snes_compare_explicit_draw",NULL,NULL,&flag_draw);CHKERRQ(ierr); 274616413a6aSBarry Smith ierr = PetscOptionsGetViewer(PetscObjectComm((PetscObject)snes),((PetscObject) snes)->options,((PetscObject)snes)->prefix,"-snes_compare_explicit_draw_contour",NULL,NULL,&flag_contour);CHKERRQ(ierr); 274716413a6aSBarry Smith ierr = PetscOptionsGetViewer(PetscObjectComm((PetscObject)snes),((PetscObject) snes)->options,((PetscObject)snes)->prefix,"-snes_compare_operator",NULL,NULL,&flag_operator);CHKERRQ(ierr); 2748693365a8SJed Brown if (flag || flag_draw || flag_contour) { 27490298fd71SBarry Smith Mat Bexp_mine = NULL,Bexp,FDexp; 2750693365a8SJed Brown PetscViewer vdraw,vstdout; 27516b3a5b13SJed Brown PetscBool flg; 2752693365a8SJed Brown if (flag_operator) { 27530bacdadaSStefano Zampini ierr = MatComputeOperator(A,MATAIJ,&Bexp_mine);CHKERRQ(ierr); 2754693365a8SJed Brown Bexp = Bexp_mine; 2755693365a8SJed Brown } else { 2756693365a8SJed Brown /* See if the preconditioning matrix can be viewed and added directly */ 2757b9e7e5c1SBarry Smith ierr = PetscObjectBaseTypeCompareAny((PetscObject)B,&flg,MATSEQAIJ,MATMPIAIJ,MATSEQDENSE,MATMPIDENSE,MATSEQBAIJ,MATMPIBAIJ,MATSEQSBAIJ,MATMPIBAIJ,"");CHKERRQ(ierr); 275894ab13aaSBarry Smith if (flg) Bexp = B; 2759693365a8SJed Brown else { 2760693365a8SJed Brown /* If the "preconditioning" matrix is itself MATSHELL or some other type without direct support */ 27610bacdadaSStefano Zampini ierr = MatComputeOperator(B,MATAIJ,&Bexp_mine);CHKERRQ(ierr); 2762693365a8SJed Brown Bexp = Bexp_mine; 2763693365a8SJed Brown } 2764693365a8SJed Brown } 2765693365a8SJed Brown ierr = MatConvert(Bexp,MATSAME,MAT_INITIAL_MATRIX,&FDexp);CHKERRQ(ierr); 2766d1e9a80fSBarry Smith ierr = SNESComputeJacobianDefault(snes,X,FDexp,FDexp,NULL);CHKERRQ(ierr); 2767ce94432eSBarry Smith ierr = PetscViewerASCIIGetStdout(PetscObjectComm((PetscObject)snes),&vstdout);CHKERRQ(ierr); 2768693365a8SJed Brown if (flag_draw || flag_contour) { 27699e5d0892SLisandro Dalcin ierr = PetscViewerDrawOpen(PetscObjectComm((PetscObject)snes),NULL,"Explicit Jacobians",PETSC_DECIDE,PETSC_DECIDE,300,300,&vdraw);CHKERRQ(ierr); 2770693365a8SJed Brown if (flag_contour) {ierr = PetscViewerPushFormat(vdraw,PETSC_VIEWER_DRAW_CONTOUR);CHKERRQ(ierr);} 27710298fd71SBarry Smith } else vdraw = NULL; 2772693365a8SJed Brown ierr = PetscViewerASCIIPrintf(vstdout,"Explicit %s\n",flag_operator ? "Jacobian" : "preconditioning Jacobian");CHKERRQ(ierr); 2773693365a8SJed Brown if (flag) {ierr = MatView(Bexp,vstdout);CHKERRQ(ierr);} 2774693365a8SJed Brown if (vdraw) {ierr = MatView(Bexp,vdraw);CHKERRQ(ierr);} 2775693365a8SJed Brown ierr = PetscViewerASCIIPrintf(vstdout,"Finite difference Jacobian\n");CHKERRQ(ierr); 2776693365a8SJed Brown if (flag) {ierr = MatView(FDexp,vstdout);CHKERRQ(ierr);} 2777693365a8SJed Brown if (vdraw) {ierr = MatView(FDexp,vdraw);CHKERRQ(ierr);} 2778693365a8SJed Brown ierr = MatAYPX(FDexp,-1.0,Bexp,SAME_NONZERO_PATTERN);CHKERRQ(ierr); 2779693365a8SJed Brown ierr = PetscViewerASCIIPrintf(vstdout,"User-provided matrix minus finite difference Jacobian\n");CHKERRQ(ierr); 2780693365a8SJed Brown if (flag) {ierr = MatView(FDexp,vstdout);CHKERRQ(ierr);} 2781693365a8SJed Brown if (vdraw) { /* Always use contour for the difference */ 2782693365a8SJed Brown ierr = PetscViewerPushFormat(vdraw,PETSC_VIEWER_DRAW_CONTOUR);CHKERRQ(ierr); 2783693365a8SJed Brown ierr = MatView(FDexp,vdraw);CHKERRQ(ierr); 2784693365a8SJed Brown ierr = PetscViewerPopFormat(vdraw);CHKERRQ(ierr); 2785693365a8SJed Brown } 2786693365a8SJed Brown if (flag_contour) {ierr = PetscViewerPopFormat(vdraw);CHKERRQ(ierr);} 2787693365a8SJed Brown ierr = PetscViewerDestroy(&vdraw);CHKERRQ(ierr); 2788693365a8SJed Brown ierr = MatDestroy(&Bexp_mine);CHKERRQ(ierr); 2789693365a8SJed Brown ierr = MatDestroy(&FDexp);CHKERRQ(ierr); 2790693365a8SJed Brown } 2791693365a8SJed Brown } 27924c30e9fbSJed Brown { 27936719d8e4SJed Brown PetscBool flag = PETSC_FALSE,flag_display = PETSC_FALSE,flag_draw = PETSC_FALSE,flag_contour = PETSC_FALSE,flag_threshold = PETSC_FALSE; 27946719d8e4SJed Brown PetscReal threshold_atol = PETSC_SQRT_MACHINE_EPSILON,threshold_rtol = 10*PETSC_SQRT_MACHINE_EPSILON; 279516413a6aSBarry Smith ierr = PetscOptionsGetViewer(PetscObjectComm((PetscObject)snes),((PetscObject)snes)->options,((PetscObject)snes)->prefix,"-snes_compare_coloring",NULL,NULL,&flag);CHKERRQ(ierr); 279616413a6aSBarry Smith ierr = PetscOptionsGetViewer(PetscObjectComm((PetscObject)snes),((PetscObject)snes)->options,((PetscObject)snes)->prefix,"-snes_compare_coloring_display",NULL,NULL,&flag_display);CHKERRQ(ierr); 279716413a6aSBarry Smith ierr = PetscOptionsGetViewer(PetscObjectComm((PetscObject)snes),((PetscObject)snes)->options,((PetscObject)snes)->prefix,"-snes_compare_coloring_draw",NULL,NULL,&flag_draw);CHKERRQ(ierr); 279816413a6aSBarry Smith ierr = PetscOptionsGetViewer(PetscObjectComm((PetscObject)snes),((PetscObject)snes)->options,((PetscObject)snes)->prefix,"-snes_compare_coloring_draw_contour",NULL,NULL,&flag_contour);CHKERRQ(ierr); 279916413a6aSBarry Smith ierr = PetscOptionsGetViewer(PetscObjectComm((PetscObject)snes),((PetscObject)snes)->options,((PetscObject)snes)->prefix,"-snes_compare_coloring_threshold",NULL,NULL,&flag_threshold);CHKERRQ(ierr); 280027b0f280SBarry Smith if (flag_threshold) { 2801c5929fdfSBarry Smith ierr = PetscOptionsGetReal(((PetscObject)snes)->options,((PetscObject)snes)->prefix,"-snes_compare_coloring_threshold_rtol",&threshold_rtol,NULL);CHKERRQ(ierr); 2802c5929fdfSBarry Smith ierr = PetscOptionsGetReal(((PetscObject)snes)->options,((PetscObject)snes)->prefix,"-snes_compare_coloring_threshold_atol",&threshold_atol,NULL);CHKERRQ(ierr); 280327b0f280SBarry Smith } 28046719d8e4SJed Brown if (flag || flag_display || flag_draw || flag_contour || flag_threshold) { 28054c30e9fbSJed Brown Mat Bfd; 28064c30e9fbSJed Brown PetscViewer vdraw,vstdout; 2807335efc43SPeter Brune MatColoring coloring; 28084c30e9fbSJed Brown ISColoring iscoloring; 28094c30e9fbSJed Brown MatFDColoring matfdcoloring; 28104c30e9fbSJed Brown PetscErrorCode (*func)(SNES,Vec,Vec,void*); 28114c30e9fbSJed Brown void *funcctx; 28126719d8e4SJed Brown PetscReal norm1,norm2,normmax; 28134c30e9fbSJed Brown 281494ab13aaSBarry Smith ierr = MatDuplicate(B,MAT_DO_NOT_COPY_VALUES,&Bfd);CHKERRQ(ierr); 2815335efc43SPeter Brune ierr = MatColoringCreate(Bfd,&coloring);CHKERRQ(ierr); 2816335efc43SPeter Brune ierr = MatColoringSetType(coloring,MATCOLORINGSL);CHKERRQ(ierr); 2817335efc43SPeter Brune ierr = MatColoringSetFromOptions(coloring);CHKERRQ(ierr); 2818335efc43SPeter Brune ierr = MatColoringApply(coloring,&iscoloring);CHKERRQ(ierr); 2819335efc43SPeter Brune ierr = MatColoringDestroy(&coloring);CHKERRQ(ierr); 28204c30e9fbSJed Brown ierr = MatFDColoringCreate(Bfd,iscoloring,&matfdcoloring);CHKERRQ(ierr); 2821f86b9fbaSHong Zhang ierr = MatFDColoringSetFromOptions(matfdcoloring);CHKERRQ(ierr); 2822f86b9fbaSHong Zhang ierr = MatFDColoringSetUp(Bfd,iscoloring,matfdcoloring);CHKERRQ(ierr); 28234c30e9fbSJed Brown ierr = ISColoringDestroy(&iscoloring);CHKERRQ(ierr); 28244c30e9fbSJed Brown 28254c30e9fbSJed Brown /* This method of getting the function is currently unreliable since it doesn't work for DM local functions. */ 28260298fd71SBarry Smith ierr = SNESGetFunction(snes,NULL,&func,&funcctx);CHKERRQ(ierr); 28274c30e9fbSJed Brown ierr = MatFDColoringSetFunction(matfdcoloring,(PetscErrorCode (*)(void))func,funcctx);CHKERRQ(ierr); 28284c30e9fbSJed Brown ierr = PetscObjectSetOptionsPrefix((PetscObject)matfdcoloring,((PetscObject)snes)->prefix);CHKERRQ(ierr); 28294c30e9fbSJed Brown ierr = PetscObjectAppendOptionsPrefix((PetscObject)matfdcoloring,"coloring_");CHKERRQ(ierr); 28304c30e9fbSJed Brown ierr = MatFDColoringSetFromOptions(matfdcoloring);CHKERRQ(ierr); 2831d1e9a80fSBarry Smith ierr = MatFDColoringApply(Bfd,matfdcoloring,X,snes);CHKERRQ(ierr); 28324c30e9fbSJed Brown ierr = MatFDColoringDestroy(&matfdcoloring);CHKERRQ(ierr); 28334c30e9fbSJed Brown 2834ce94432eSBarry Smith ierr = PetscViewerASCIIGetStdout(PetscObjectComm((PetscObject)snes),&vstdout);CHKERRQ(ierr); 28354c30e9fbSJed Brown if (flag_draw || flag_contour) { 28369e5d0892SLisandro Dalcin ierr = PetscViewerDrawOpen(PetscObjectComm((PetscObject)snes),NULL,"Colored Jacobians",PETSC_DECIDE,PETSC_DECIDE,300,300,&vdraw);CHKERRQ(ierr); 28374c30e9fbSJed Brown if (flag_contour) {ierr = PetscViewerPushFormat(vdraw,PETSC_VIEWER_DRAW_CONTOUR);CHKERRQ(ierr);} 28380298fd71SBarry Smith } else vdraw = NULL; 28394c30e9fbSJed Brown ierr = PetscViewerASCIIPrintf(vstdout,"Explicit preconditioning Jacobian\n");CHKERRQ(ierr); 284094ab13aaSBarry Smith if (flag_display) {ierr = MatView(B,vstdout);CHKERRQ(ierr);} 284194ab13aaSBarry Smith if (vdraw) {ierr = MatView(B,vdraw);CHKERRQ(ierr);} 28424c30e9fbSJed Brown ierr = PetscViewerASCIIPrintf(vstdout,"Colored Finite difference Jacobian\n");CHKERRQ(ierr); 28436719d8e4SJed Brown if (flag_display) {ierr = MatView(Bfd,vstdout);CHKERRQ(ierr);} 28444c30e9fbSJed Brown if (vdraw) {ierr = MatView(Bfd,vdraw);CHKERRQ(ierr);} 284594ab13aaSBarry Smith ierr = MatAYPX(Bfd,-1.0,B,SAME_NONZERO_PATTERN);CHKERRQ(ierr); 28464c30e9fbSJed Brown ierr = MatNorm(Bfd,NORM_1,&norm1);CHKERRQ(ierr); 28476719d8e4SJed Brown ierr = MatNorm(Bfd,NORM_FROBENIUS,&norm2);CHKERRQ(ierr); 28484c30e9fbSJed Brown ierr = MatNorm(Bfd,NORM_MAX,&normmax);CHKERRQ(ierr); 284957622a8eSBarry 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); 28506719d8e4SJed Brown if (flag_display) {ierr = MatView(Bfd,vstdout);CHKERRQ(ierr);} 28514c30e9fbSJed Brown if (vdraw) { /* Always use contour for the difference */ 28524c30e9fbSJed Brown ierr = PetscViewerPushFormat(vdraw,PETSC_VIEWER_DRAW_CONTOUR);CHKERRQ(ierr); 28534c30e9fbSJed Brown ierr = MatView(Bfd,vdraw);CHKERRQ(ierr); 28544c30e9fbSJed Brown ierr = PetscViewerPopFormat(vdraw);CHKERRQ(ierr); 28554c30e9fbSJed Brown } 28564c30e9fbSJed Brown if (flag_contour) {ierr = PetscViewerPopFormat(vdraw);CHKERRQ(ierr);} 28576719d8e4SJed Brown 28586719d8e4SJed Brown if (flag_threshold) { 28596719d8e4SJed Brown PetscInt bs,rstart,rend,i; 286094ab13aaSBarry Smith ierr = MatGetBlockSize(B,&bs);CHKERRQ(ierr); 286194ab13aaSBarry Smith ierr = MatGetOwnershipRange(B,&rstart,&rend);CHKERRQ(ierr); 28626719d8e4SJed Brown for (i=rstart; i<rend; i++) { 28636719d8e4SJed Brown const PetscScalar *ba,*ca; 28646719d8e4SJed Brown const PetscInt *bj,*cj; 28656719d8e4SJed Brown PetscInt bn,cn,j,maxentrycol = -1,maxdiffcol = -1,maxrdiffcol = -1; 28666719d8e4SJed Brown PetscReal maxentry = 0,maxdiff = 0,maxrdiff = 0; 286794ab13aaSBarry Smith ierr = MatGetRow(B,i,&bn,&bj,&ba);CHKERRQ(ierr); 28686719d8e4SJed Brown ierr = MatGetRow(Bfd,i,&cn,&cj,&ca);CHKERRQ(ierr); 286994ab13aaSBarry Smith if (bn != cn) SETERRQ(((PetscObject)A)->comm,PETSC_ERR_PLIB,"Unexpected different nonzero pattern in -snes_compare_coloring_threshold"); 28706719d8e4SJed Brown for (j=0; j<bn; j++) { 28716719d8e4SJed Brown PetscReal rdiff = PetscAbsScalar(ca[j]) / (threshold_atol + threshold_rtol*PetscAbsScalar(ba[j])); 28726719d8e4SJed Brown if (PetscAbsScalar(ba[j]) > PetscAbs(maxentry)) { 28736719d8e4SJed Brown maxentrycol = bj[j]; 28746719d8e4SJed Brown maxentry = PetscRealPart(ba[j]); 28756719d8e4SJed Brown } 28766719d8e4SJed Brown if (PetscAbsScalar(ca[j]) > PetscAbs(maxdiff)) { 28776719d8e4SJed Brown maxdiffcol = bj[j]; 28786719d8e4SJed Brown maxdiff = PetscRealPart(ca[j]); 28796719d8e4SJed Brown } 28806719d8e4SJed Brown if (rdiff > maxrdiff) { 28816719d8e4SJed Brown maxrdiffcol = bj[j]; 28826719d8e4SJed Brown maxrdiff = rdiff; 28836719d8e4SJed Brown } 28846719d8e4SJed Brown } 28856719d8e4SJed Brown if (maxrdiff > 1) { 288657622a8eSBarry 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); 28876719d8e4SJed Brown for (j=0; j<bn; j++) { 28886719d8e4SJed Brown PetscReal rdiff; 28896719d8e4SJed Brown rdiff = PetscAbsScalar(ca[j]) / (threshold_atol + threshold_rtol*PetscAbsScalar(ba[j])); 28906719d8e4SJed Brown if (rdiff > 1) { 289157622a8eSBarry Smith ierr = PetscViewerASCIIPrintf(vstdout," (%D,%g:%g)",bj[j],(double)PetscRealPart(ba[j]),(double)PetscRealPart(ca[j]));CHKERRQ(ierr); 28926719d8e4SJed Brown } 28936719d8e4SJed Brown } 28946719d8e4SJed Brown ierr = PetscViewerASCIIPrintf(vstdout,"\n",i,maxentry,maxdiff,maxrdiff);CHKERRQ(ierr); 28956719d8e4SJed Brown } 289694ab13aaSBarry Smith ierr = MatRestoreRow(B,i,&bn,&bj,&ba);CHKERRQ(ierr); 28976719d8e4SJed Brown ierr = MatRestoreRow(Bfd,i,&cn,&cj,&ca);CHKERRQ(ierr); 28986719d8e4SJed Brown } 28996719d8e4SJed Brown } 29004c30e9fbSJed Brown ierr = PetscViewerDestroy(&vdraw);CHKERRQ(ierr); 29014c30e9fbSJed Brown ierr = MatDestroy(&Bfd);CHKERRQ(ierr); 29024c30e9fbSJed Brown } 29034c30e9fbSJed Brown } 29043a40ed3dSBarry Smith PetscFunctionReturn(0); 29059b94acceSBarry Smith } 29069b94acceSBarry Smith 2907bf388a1fSBarry Smith /*MC 2908411c0326SBarry Smith SNESJacobianFunction - Function used to convey the nonlinear Jacobian of the function to be solved by SNES 2909bf388a1fSBarry Smith 2910bf388a1fSBarry Smith Synopsis: 2911411c0326SBarry Smith #include "petscsnes.h" 2912411c0326SBarry Smith PetscErrorCode SNESJacobianFunction(SNES snes,Vec x,Mat Amat,Mat Pmat,void *ctx); 2913bf388a1fSBarry Smith 29141843f636SBarry Smith Collective on snes 29151843f636SBarry Smith 29161843f636SBarry Smith Input Parameters: 29171843f636SBarry Smith + x - input vector, the Jacobian is to be computed at this value 2918bf388a1fSBarry Smith - ctx - [optional] user-defined Jacobian context 2919bf388a1fSBarry Smith 29201843f636SBarry Smith Output Parameters: 29211843f636SBarry Smith + Amat - the matrix that defines the (approximate) Jacobian 29221843f636SBarry Smith - Pmat - the matrix to be used in constructing the preconditioner, usually the same as Amat. 29231843f636SBarry Smith 2924878cb397SSatish Balay Level: intermediate 2925878cb397SSatish Balay 2926bf388a1fSBarry Smith .seealso: SNESSetFunction(), SNESGetFunction(), SNESSetJacobian(), SNESGetJacobian() 2927bf388a1fSBarry Smith M*/ 2928bf388a1fSBarry Smith 29299b94acceSBarry Smith /*@C 29309b94acceSBarry Smith SNESSetJacobian - Sets the function to compute Jacobian as well as the 2931044dda88SLois Curfman McInnes location to store the matrix. 29329b94acceSBarry Smith 2933d083f849SBarry Smith Logically Collective on SNES 2934c7afd0dbSLois Curfman McInnes 29359b94acceSBarry Smith Input Parameters: 2936c7afd0dbSLois Curfman McInnes + snes - the SNES context 2937e5d3d808SBarry Smith . Amat - the matrix that defines the (approximate) Jacobian 2938e5d3d808SBarry Smith . Pmat - the matrix to be used in constructing the preconditioner, usually the same as Amat. 2939411c0326SBarry Smith . J - Jacobian evaluation routine (if NULL then SNES retains any previously set value), see SNESJacobianFunction for details 2940c7afd0dbSLois Curfman McInnes - ctx - [optional] user-defined context for private data for the 29410298fd71SBarry Smith Jacobian evaluation routine (may be NULL) (if NULL then SNES retains any previously set value) 29429b94acceSBarry Smith 29439b94acceSBarry Smith Notes: 2944e5d3d808SBarry Smith If the Amat matrix and Pmat matrix are different you must call MatAssemblyBegin/End() on 294516913363SBarry Smith each matrix. 294616913363SBarry Smith 2947895c21f2SBarry Smith If you know the operator Amat has a null space you can use MatSetNullSpace() and MatSetTransposeNullSpace() to supply the null 2948895c21f2SBarry Smith space to Amat and the KSP solvers will automatically use that null space as needed during the solution process. 2949895c21f2SBarry Smith 29508d359177SBarry Smith If using SNESComputeJacobianDefaultColor() to assemble a Jacobian, the ctx argument 2951a8a26c1eSJed Brown must be a MatFDColoring. 2952a8a26c1eSJed Brown 2953c3cc8fd1SJed Brown Other defect-correction schemes can be used by computing a different matrix in place of the Jacobian. One common 2954c3cc8fd1SJed Brown example is to use the "Picard linearization" which only differentiates through the highest order parts of each term. 2955c3cc8fd1SJed Brown 295636851e7fSLois Curfman McInnes Level: beginner 295736851e7fSLois Curfman McInnes 2958411c0326SBarry Smith .seealso: KSPSetOperators(), SNESSetFunction(), MatMFFDComputeJacobian(), SNESComputeJacobianDefaultColor(), MatStructure, J, 2959411c0326SBarry Smith SNESSetPicard(), SNESJacobianFunction 29609b94acceSBarry Smith @*/ 2961d1e9a80fSBarry Smith PetscErrorCode SNESSetJacobian(SNES snes,Mat Amat,Mat Pmat,PetscErrorCode (*J)(SNES,Vec,Mat,Mat,void*),void *ctx) 29629b94acceSBarry Smith { 2963dfbe8321SBarry Smith PetscErrorCode ierr; 29646cab3a1bSJed Brown DM dm; 29653a7fca6bSBarry Smith 29663a40ed3dSBarry Smith PetscFunctionBegin; 29670700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 2968e5d3d808SBarry Smith if (Amat) PetscValidHeaderSpecific(Amat,MAT_CLASSID,2); 2969e5d3d808SBarry Smith if (Pmat) PetscValidHeaderSpecific(Pmat,MAT_CLASSID,3); 2970e5d3d808SBarry Smith if (Amat) PetscCheckSameComm(snes,1,Amat,2); 2971e5d3d808SBarry Smith if (Pmat) PetscCheckSameComm(snes,1,Pmat,3); 29726cab3a1bSJed Brown ierr = SNESGetDM(snes,&dm);CHKERRQ(ierr); 2973f8b49ee9SBarry Smith ierr = DMSNESSetJacobian(dm,J,ctx);CHKERRQ(ierr); 2974e5d3d808SBarry Smith if (Amat) { 2975e5d3d808SBarry Smith ierr = PetscObjectReference((PetscObject)Amat);CHKERRQ(ierr); 29766bf464f9SBarry Smith ierr = MatDestroy(&snes->jacobian);CHKERRQ(ierr); 2977f5af7f23SKarl Rupp 2978e5d3d808SBarry Smith snes->jacobian = Amat; 29793a7fca6bSBarry Smith } 2980e5d3d808SBarry Smith if (Pmat) { 2981e5d3d808SBarry Smith ierr = PetscObjectReference((PetscObject)Pmat);CHKERRQ(ierr); 29826bf464f9SBarry Smith ierr = MatDestroy(&snes->jacobian_pre);CHKERRQ(ierr); 2983f5af7f23SKarl Rupp 2984e5d3d808SBarry Smith snes->jacobian_pre = Pmat; 29853a7fca6bSBarry Smith } 29863a40ed3dSBarry Smith PetscFunctionReturn(0); 29879b94acceSBarry Smith } 298862fef451SLois Curfman McInnes 2989c2aafc4cSSatish Balay /*@C 2990b4fd4287SBarry Smith SNESGetJacobian - Returns the Jacobian matrix and optionally the user 2991b4fd4287SBarry Smith provided context for evaluating the Jacobian. 2992b4fd4287SBarry Smith 2993c7afd0dbSLois Curfman McInnes Not Collective, but Mat object will be parallel if SNES object is 2994c7afd0dbSLois Curfman McInnes 2995b4fd4287SBarry Smith Input Parameter: 2996b4fd4287SBarry Smith . snes - the nonlinear solver context 2997b4fd4287SBarry Smith 2998b4fd4287SBarry Smith Output Parameters: 2999e5d3d808SBarry Smith + Amat - location to stash (approximate) Jacobian matrix (or NULL) 3000e5d3d808SBarry Smith . Pmat - location to stash matrix used to compute the preconditioner (or NULL) 3001411c0326SBarry Smith . J - location to put Jacobian function (or NULL), see SNESJacobianFunction for details on its calling sequence 30020298fd71SBarry Smith - ctx - location to stash Jacobian ctx (or NULL) 3003fee21e36SBarry Smith 300436851e7fSLois Curfman McInnes Level: advanced 300536851e7fSLois Curfman McInnes 3006411c0326SBarry Smith .seealso: SNESSetJacobian(), SNESComputeJacobian(), SNESJacobianFunction, SNESGetFunction() 3007b4fd4287SBarry Smith @*/ 3008d1e9a80fSBarry Smith PetscErrorCode SNESGetJacobian(SNES snes,Mat *Amat,Mat *Pmat,PetscErrorCode (**J)(SNES,Vec,Mat,Mat,void*),void **ctx) 3009b4fd4287SBarry Smith { 30106cab3a1bSJed Brown PetscErrorCode ierr; 30116cab3a1bSJed Brown DM dm; 3012942e3340SBarry Smith DMSNES sdm; 30136cab3a1bSJed Brown 30143a40ed3dSBarry Smith PetscFunctionBegin; 30150700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 3016e5d3d808SBarry Smith if (Amat) *Amat = snes->jacobian; 3017e5d3d808SBarry Smith if (Pmat) *Pmat = snes->jacobian_pre; 30186cab3a1bSJed Brown ierr = SNESGetDM(snes,&dm);CHKERRQ(ierr); 3019942e3340SBarry Smith ierr = DMGetDMSNES(dm,&sdm);CHKERRQ(ierr); 3020f8b49ee9SBarry Smith if (J) *J = sdm->ops->computejacobian; 30216cab3a1bSJed Brown if (ctx) *ctx = sdm->jacobianctx; 30223a40ed3dSBarry Smith PetscFunctionReturn(0); 3023b4fd4287SBarry Smith } 3024b4fd4287SBarry Smith 302558b371f3SBarry Smith static PetscErrorCode SNESSetDefaultComputeJacobian(SNES snes) 302658b371f3SBarry Smith { 302758b371f3SBarry Smith PetscErrorCode ierr; 302858b371f3SBarry Smith DM dm; 302958b371f3SBarry Smith DMSNES sdm; 303058b371f3SBarry Smith 303158b371f3SBarry Smith PetscFunctionBegin; 303258b371f3SBarry Smith ierr = SNESGetDM(snes,&dm);CHKERRQ(ierr); 303358b371f3SBarry Smith ierr = DMGetDMSNES(dm,&sdm);CHKERRQ(ierr); 303458b371f3SBarry Smith if (!sdm->ops->computejacobian && snes->jacobian_pre) { 303558b371f3SBarry Smith DM dm; 303658b371f3SBarry Smith PetscBool isdense,ismf; 303758b371f3SBarry Smith 303858b371f3SBarry Smith ierr = SNESGetDM(snes,&dm);CHKERRQ(ierr); 303958b371f3SBarry Smith ierr = PetscObjectTypeCompareAny((PetscObject)snes->jacobian_pre,&isdense,MATSEQDENSE,MATMPIDENSE,MATDENSE,NULL);CHKERRQ(ierr); 304058b371f3SBarry Smith ierr = PetscObjectTypeCompareAny((PetscObject)snes->jacobian_pre,&ismf,MATMFFD,MATSHELL,NULL);CHKERRQ(ierr); 304158b371f3SBarry Smith if (isdense) { 304258b371f3SBarry Smith ierr = DMSNESSetJacobian(dm,SNESComputeJacobianDefault,NULL);CHKERRQ(ierr); 304358b371f3SBarry Smith } else if (!ismf) { 304458b371f3SBarry Smith ierr = DMSNESSetJacobian(dm,SNESComputeJacobianDefaultColor,NULL);CHKERRQ(ierr); 304558b371f3SBarry Smith } 304658b371f3SBarry Smith } 304758b371f3SBarry Smith PetscFunctionReturn(0); 304858b371f3SBarry Smith } 304958b371f3SBarry Smith 30509b94acceSBarry Smith /*@ 30519b94acceSBarry Smith SNESSetUp - Sets up the internal data structures for the later use 3052272ac6f2SLois Curfman McInnes of a nonlinear solver. 30539b94acceSBarry Smith 3054fee21e36SBarry Smith Collective on SNES 3055fee21e36SBarry Smith 3056c7afd0dbSLois Curfman McInnes Input Parameters: 305770e92668SMatthew Knepley . snes - the SNES context 3058c7afd0dbSLois Curfman McInnes 3059272ac6f2SLois Curfman McInnes Notes: 3060272ac6f2SLois Curfman McInnes For basic use of the SNES solvers the user need not explicitly call 3061272ac6f2SLois Curfman McInnes SNESSetUp(), since these actions will automatically occur during 3062272ac6f2SLois Curfman McInnes the call to SNESSolve(). However, if one wishes to control this 3063272ac6f2SLois Curfman McInnes phase separately, SNESSetUp() should be called after SNESCreate() 3064272ac6f2SLois Curfman McInnes and optional routines of the form SNESSetXXX(), but before SNESSolve(). 3065272ac6f2SLois Curfman McInnes 306636851e7fSLois Curfman McInnes Level: advanced 306736851e7fSLois Curfman McInnes 30689b94acceSBarry Smith .seealso: SNESCreate(), SNESSolve(), SNESDestroy() 30699b94acceSBarry Smith @*/ 30707087cfbeSBarry Smith PetscErrorCode SNESSetUp(SNES snes) 30719b94acceSBarry Smith { 3072dfbe8321SBarry Smith PetscErrorCode ierr; 30736cab3a1bSJed Brown DM dm; 3074942e3340SBarry Smith DMSNES sdm; 3075c35f09e5SBarry Smith SNESLineSearch linesearch, pclinesearch; 30766e2a1849SPeter Brune void *lsprectx,*lspostctx; 30776b2b7091SBarry Smith PetscErrorCode (*precheck)(SNESLineSearch,Vec,Vec,PetscBool*,void*); 30786b2b7091SBarry Smith PetscErrorCode (*postcheck)(SNESLineSearch,Vec,Vec,Vec,PetscBool*,PetscBool*,void*); 30796e2a1849SPeter Brune PetscErrorCode (*func)(SNES,Vec,Vec,void*); 30806e2a1849SPeter Brune Vec f,fpc; 30816e2a1849SPeter Brune void *funcctx; 3082d1e9a80fSBarry Smith PetscErrorCode (*jac)(SNES,Vec,Mat,Mat,void*); 30831eb13d49SPeter Brune void *jacctx,*appctx; 308432b97717SPeter Brune Mat j,jpre; 30853a40ed3dSBarry Smith 30863a40ed3dSBarry Smith PetscFunctionBegin; 30870700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 30884dc4c822SBarry Smith if (snes->setupcalled) PetscFunctionReturn(0); 3089e3ed9ee7SBarry Smith ierr = PetscLogEventBegin(SNES_Setup,snes,0,0,0);CHKERRQ(ierr); 30909b94acceSBarry Smith 30917adad957SLisandro Dalcin if (!((PetscObject)snes)->type_name) { 309204d7464bSBarry Smith ierr = SNESSetType(snes,SNESNEWTONLS);CHKERRQ(ierr); 309385385478SLisandro Dalcin } 309485385478SLisandro Dalcin 30950298fd71SBarry Smith ierr = SNESGetFunction(snes,&snes->vec_func,NULL,NULL);CHKERRQ(ierr); 309658c9b817SLisandro Dalcin 30976cab3a1bSJed Brown ierr = SNESGetDM(snes,&dm);CHKERRQ(ierr); 3098942e3340SBarry Smith ierr = DMGetDMSNES(dm,&sdm);CHKERRQ(ierr); 3099ce94432eSBarry Smith if (!sdm->ops->computefunction) SETERRQ(PetscObjectComm((PetscObject)dm),PETSC_ERR_ARG_WRONGSTATE,"Function never provided to SNES object"); 310058b371f3SBarry Smith ierr = SNESSetDefaultComputeJacobian(snes);CHKERRQ(ierr); 310158b371f3SBarry Smith 31026cab3a1bSJed Brown if (!snes->vec_func) { 31036cab3a1bSJed Brown ierr = DMCreateGlobalVector(dm,&snes->vec_func);CHKERRQ(ierr); 3104214df951SJed Brown } 3105efd51863SBarry Smith 310622d28d08SBarry Smith if (!snes->ksp) { 310722d28d08SBarry Smith ierr = SNESGetKSP(snes, &snes->ksp);CHKERRQ(ierr); 310822d28d08SBarry Smith } 3109b710008aSBarry Smith 3110d8d34be6SBarry Smith if (snes->linesearch) { 31117601faf0SJed Brown ierr = SNESGetLineSearch(snes, &snes->linesearch);CHKERRQ(ierr); 3112ed07d7d7SPeter Brune ierr = SNESLineSearchSetFunction(snes->linesearch,SNESComputeFunction);CHKERRQ(ierr); 3113d8d34be6SBarry Smith } 31149e764e56SPeter Brune 3115efd4aadfSBarry Smith if (snes->npc && (snes->npcside== PC_LEFT)) { 3116172a4300SPeter Brune snes->mf = PETSC_TRUE; 3117172a4300SPeter Brune snes->mf_operator = PETSC_FALSE; 3118172a4300SPeter Brune } 3119d8f46077SPeter Brune 3120efd4aadfSBarry Smith if (snes->npc) { 31216e2a1849SPeter Brune /* copy the DM over */ 31226e2a1849SPeter Brune ierr = SNESGetDM(snes,&dm);CHKERRQ(ierr); 3123efd4aadfSBarry Smith ierr = SNESSetDM(snes->npc,dm);CHKERRQ(ierr); 31246e2a1849SPeter Brune 31256e2a1849SPeter Brune ierr = SNESGetFunction(snes,&f,&func,&funcctx);CHKERRQ(ierr); 31266e2a1849SPeter Brune ierr = VecDuplicate(f,&fpc);CHKERRQ(ierr); 3127efd4aadfSBarry Smith ierr = SNESSetFunction(snes->npc,fpc,func,funcctx);CHKERRQ(ierr); 312832b97717SPeter Brune ierr = SNESGetJacobian(snes,&j,&jpre,&jac,&jacctx);CHKERRQ(ierr); 3129efd4aadfSBarry Smith ierr = SNESSetJacobian(snes->npc,j,jpre,jac,jacctx);CHKERRQ(ierr); 31301eb13d49SPeter Brune ierr = SNESGetApplicationContext(snes,&appctx);CHKERRQ(ierr); 3131efd4aadfSBarry Smith ierr = SNESSetApplicationContext(snes->npc,appctx);CHKERRQ(ierr); 31326e2a1849SPeter Brune ierr = VecDestroy(&fpc);CHKERRQ(ierr); 31336e2a1849SPeter Brune 31346e2a1849SPeter Brune /* copy the function pointers over */ 3135efd4aadfSBarry Smith ierr = PetscObjectCopyFortranFunctionPointers((PetscObject)snes,(PetscObject)snes->npc);CHKERRQ(ierr); 31366e2a1849SPeter Brune 31376e2a1849SPeter Brune /* default to 1 iteration */ 3138efd4aadfSBarry Smith ierr = SNESSetTolerances(snes->npc,0.0,0.0,0.0,1,snes->npc->max_funcs);CHKERRQ(ierr); 3139efd4aadfSBarry Smith if (snes->npcside==PC_RIGHT) { 3140efd4aadfSBarry Smith ierr = SNESSetNormSchedule(snes->npc,SNES_NORM_FINAL_ONLY);CHKERRQ(ierr); 3141a9936a0cSPeter Brune } else { 3142efd4aadfSBarry Smith ierr = SNESSetNormSchedule(snes->npc,SNES_NORM_NONE);CHKERRQ(ierr); 3143a9936a0cSPeter Brune } 3144efd4aadfSBarry Smith ierr = SNESSetFromOptions(snes->npc);CHKERRQ(ierr); 31456e2a1849SPeter Brune 31466e2a1849SPeter Brune /* copy the line search context over */ 3147d8d34be6SBarry Smith if (snes->linesearch && snes->npc->linesearch) { 31487601faf0SJed Brown ierr = SNESGetLineSearch(snes,&linesearch);CHKERRQ(ierr); 3149efd4aadfSBarry Smith ierr = SNESGetLineSearch(snes->npc,&pclinesearch);CHKERRQ(ierr); 31506b2b7091SBarry Smith ierr = SNESLineSearchGetPreCheck(linesearch,&precheck,&lsprectx);CHKERRQ(ierr); 31516b2b7091SBarry Smith ierr = SNESLineSearchGetPostCheck(linesearch,&postcheck,&lspostctx);CHKERRQ(ierr); 31526b2b7091SBarry Smith ierr = SNESLineSearchSetPreCheck(pclinesearch,precheck,lsprectx);CHKERRQ(ierr); 31536b2b7091SBarry Smith ierr = SNESLineSearchSetPostCheck(pclinesearch,postcheck,lspostctx);CHKERRQ(ierr); 31546e2a1849SPeter Brune ierr = PetscObjectCopyFortranFunctionPointers((PetscObject)linesearch, (PetscObject)pclinesearch);CHKERRQ(ierr); 31556e2a1849SPeter Brune } 3156d8d34be6SBarry Smith } 315732b97717SPeter Brune if (snes->mf) { 315832b97717SPeter Brune ierr = SNESSetUpMatrixFree_Private(snes, snes->mf_operator, snes->mf_version);CHKERRQ(ierr); 315932b97717SPeter Brune } 316032b97717SPeter Brune if (snes->ops->usercompute && !snes->user) { 316132b97717SPeter Brune ierr = (*snes->ops->usercompute)(snes,(void**)&snes->user);CHKERRQ(ierr); 316232b97717SPeter Brune } 31636e2a1849SPeter Brune 316437ec4e1aSPeter Brune snes->jac_iter = 0; 316537ec4e1aSPeter Brune snes->pre_iter = 0; 316637ec4e1aSPeter Brune 3167410397dcSLisandro Dalcin if (snes->ops->setup) { 3168410397dcSLisandro Dalcin ierr = (*snes->ops->setup)(snes);CHKERRQ(ierr); 3169410397dcSLisandro Dalcin } 317058c9b817SLisandro Dalcin 317158b371f3SBarry Smith ierr = SNESSetDefaultComputeJacobian(snes);CHKERRQ(ierr); 317258b371f3SBarry Smith 3173efd4aadfSBarry Smith if (snes->npc && (snes->npcside== PC_LEFT)) { 31746c67d002SPeter Brune if (snes->functype == SNES_FUNCTION_PRECONDITIONED) { 3175d8d34be6SBarry Smith if (snes->linesearch){ 317655d4788fSPeter Brune ierr = SNESGetLineSearch(snes,&linesearch);CHKERRQ(ierr); 3177be95d8f1SBarry Smith ierr = SNESLineSearchSetFunction(linesearch,SNESComputeFunctionDefaultNPC);CHKERRQ(ierr); 31786c67d002SPeter Brune } 31796c67d002SPeter Brune } 3180d8d34be6SBarry Smith } 3181e3ed9ee7SBarry Smith ierr = PetscLogEventEnd(SNES_Setup,snes,0,0,0);CHKERRQ(ierr); 31827aaed0d8SBarry Smith snes->setupcalled = PETSC_TRUE; 31833a40ed3dSBarry Smith PetscFunctionReturn(0); 31849b94acceSBarry Smith } 31859b94acceSBarry Smith 318637596af1SLisandro Dalcin /*@ 318737596af1SLisandro Dalcin SNESReset - Resets a SNES context to the snessetupcalled = 0 state and removes any allocated Vecs and Mats 318837596af1SLisandro Dalcin 318937596af1SLisandro Dalcin Collective on SNES 319037596af1SLisandro Dalcin 319137596af1SLisandro Dalcin Input Parameter: 319237596af1SLisandro Dalcin . snes - iterative context obtained from SNESCreate() 319337596af1SLisandro Dalcin 3194d25893d9SBarry Smith Level: intermediate 3195d25893d9SBarry Smith 319695452b02SPatrick Sanan Notes: 319795452b02SPatrick Sanan Also calls the application context destroy routine set with SNESSetComputeApplicationContext() 319837596af1SLisandro Dalcin 319937596af1SLisandro Dalcin .seealso: SNESCreate(), SNESSetUp(), SNESSolve() 320037596af1SLisandro Dalcin @*/ 320137596af1SLisandro Dalcin PetscErrorCode SNESReset(SNES snes) 320237596af1SLisandro Dalcin { 320337596af1SLisandro Dalcin PetscErrorCode ierr; 320437596af1SLisandro Dalcin 320537596af1SLisandro Dalcin PetscFunctionBegin; 320637596af1SLisandro Dalcin PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 3207d25893d9SBarry Smith if (snes->ops->userdestroy && snes->user) { 3208d25893d9SBarry Smith ierr = (*snes->ops->userdestroy)((void**)&snes->user);CHKERRQ(ierr); 32090298fd71SBarry Smith snes->user = NULL; 3210d25893d9SBarry Smith } 3211efd4aadfSBarry Smith if (snes->npc) { 3212efd4aadfSBarry Smith ierr = SNESReset(snes->npc);CHKERRQ(ierr); 32138a23116dSBarry Smith } 32148a23116dSBarry Smith 321537596af1SLisandro Dalcin if (snes->ops->reset) { 321637596af1SLisandro Dalcin ierr = (*snes->ops->reset)(snes);CHKERRQ(ierr); 321737596af1SLisandro Dalcin } 32189e764e56SPeter Brune if (snes->ksp) { 32199e764e56SPeter Brune ierr = KSPReset(snes->ksp);CHKERRQ(ierr); 32209e764e56SPeter Brune } 32219e764e56SPeter Brune 32229e764e56SPeter Brune if (snes->linesearch) { 3223f1c6b773SPeter Brune ierr = SNESLineSearchReset(snes->linesearch);CHKERRQ(ierr); 32249e764e56SPeter Brune } 32259e764e56SPeter Brune 32266bf464f9SBarry Smith ierr = VecDestroy(&snes->vec_rhs);CHKERRQ(ierr); 32276bf464f9SBarry Smith ierr = VecDestroy(&snes->vec_sol);CHKERRQ(ierr); 32286bf464f9SBarry Smith ierr = VecDestroy(&snes->vec_sol_update);CHKERRQ(ierr); 32296bf464f9SBarry Smith ierr = VecDestroy(&snes->vec_func);CHKERRQ(ierr); 32306bf464f9SBarry Smith ierr = MatDestroy(&snes->jacobian);CHKERRQ(ierr); 32316bf464f9SBarry Smith ierr = MatDestroy(&snes->jacobian_pre);CHKERRQ(ierr); 3232c41d97abSJed Brown ierr = VecDestroyVecs(snes->nwork,&snes->work);CHKERRQ(ierr); 3233c41d97abSJed Brown ierr = VecDestroyVecs(snes->nvwork,&snes->vwork);CHKERRQ(ierr); 3234f5af7f23SKarl Rupp 323540fdac6aSLawrence Mitchell snes->alwayscomputesfinalresidual = PETSC_FALSE; 323640fdac6aSLawrence Mitchell 323737596af1SLisandro Dalcin snes->nwork = snes->nvwork = 0; 323837596af1SLisandro Dalcin snes->setupcalled = PETSC_FALSE; 323937596af1SLisandro Dalcin PetscFunctionReturn(0); 324037596af1SLisandro Dalcin } 324137596af1SLisandro Dalcin 324252baeb72SSatish Balay /*@ 3243c4421ceaSFande Kong SNESConvergedReasonViewCancel - Clears all the reasonview functions for a SNES object. 3244c4421ceaSFande Kong 3245c4421ceaSFande Kong Collective on SNES 3246c4421ceaSFande Kong 3247c4421ceaSFande Kong Input Parameter: 3248c4421ceaSFande Kong . snes - iterative context obtained from SNESCreate() 3249c4421ceaSFande Kong 3250c4421ceaSFande Kong Level: intermediate 3251c4421ceaSFande Kong 3252c4421ceaSFande Kong .seealso: SNESCreate(), SNESDestroy(), SNESReset() 3253c4421ceaSFande Kong @*/ 3254c4421ceaSFande Kong PetscErrorCode SNESConvergedReasonViewCancel(SNES snes) 3255c4421ceaSFande Kong { 3256c4421ceaSFande Kong PetscErrorCode ierr; 3257c4421ceaSFande Kong PetscInt i; 3258c4421ceaSFande Kong 3259c4421ceaSFande Kong PetscFunctionBegin; 3260c4421ceaSFande Kong PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 3261c4421ceaSFande Kong for (i=0; i<snes->numberreasonviews; i++) { 3262c4421ceaSFande Kong if (snes->reasonviewdestroy[i]) { 3263c4421ceaSFande Kong ierr = (*snes->reasonviewdestroy[i])(&snes->reasonviewcontext[i]);CHKERRQ(ierr); 3264c4421ceaSFande Kong } 3265c4421ceaSFande Kong } 3266c4421ceaSFande Kong snes->numberreasonviews = 0; 3267c4421ceaSFande Kong PetscFunctionReturn(0); 3268c4421ceaSFande Kong } 3269c4421ceaSFande Kong 3270c4421ceaSFande Kong /*@ 32719b94acceSBarry Smith SNESDestroy - Destroys the nonlinear solver context that was created 32729b94acceSBarry Smith with SNESCreate(). 32739b94acceSBarry Smith 3274c7afd0dbSLois Curfman McInnes Collective on SNES 3275c7afd0dbSLois Curfman McInnes 32769b94acceSBarry Smith Input Parameter: 32779b94acceSBarry Smith . snes - the SNES context 32789b94acceSBarry Smith 327936851e7fSLois Curfman McInnes Level: beginner 328036851e7fSLois Curfman McInnes 328163a78c88SLois Curfman McInnes .seealso: SNESCreate(), SNESSolve() 32829b94acceSBarry Smith @*/ 32836bf464f9SBarry Smith PetscErrorCode SNESDestroy(SNES *snes) 32849b94acceSBarry Smith { 32856849ba73SBarry Smith PetscErrorCode ierr; 32863a40ed3dSBarry Smith 32873a40ed3dSBarry Smith PetscFunctionBegin; 32886bf464f9SBarry Smith if (!*snes) PetscFunctionReturn(0); 32896bf464f9SBarry Smith PetscValidHeaderSpecific((*snes),SNES_CLASSID,1); 32909e5d0892SLisandro Dalcin if (--((PetscObject)(*snes))->refct > 0) {*snes = NULL; PetscFunctionReturn(0);} 3291d4bb536fSBarry Smith 32926bf464f9SBarry Smith ierr = SNESReset((*snes));CHKERRQ(ierr); 3293efd4aadfSBarry Smith ierr = SNESDestroy(&(*snes)->npc);CHKERRQ(ierr); 32946b8b9a38SLisandro Dalcin 3295e04113cfSBarry Smith /* if memory was published with SAWs then destroy it */ 3296e04113cfSBarry Smith ierr = PetscObjectSAWsViewOff((PetscObject)*snes);CHKERRQ(ierr); 32976bf464f9SBarry Smith if ((*snes)->ops->destroy) {ierr = (*((*snes))->ops->destroy)((*snes));CHKERRQ(ierr);} 32986d4c513bSLisandro Dalcin 329933124788SMatthew G. Knepley if ((*snes)->dm) {ierr = DMCoarsenHookRemove((*snes)->dm,DMCoarsenHook_SNESVecSol,DMRestrictHook_SNESVecSol,*snes);CHKERRQ(ierr);} 33006bf464f9SBarry Smith ierr = DMDestroy(&(*snes)->dm);CHKERRQ(ierr); 33016bf464f9SBarry Smith ierr = KSPDestroy(&(*snes)->ksp);CHKERRQ(ierr); 3302f1c6b773SPeter Brune ierr = SNESLineSearchDestroy(&(*snes)->linesearch);CHKERRQ(ierr); 33036b8b9a38SLisandro Dalcin 33046bf464f9SBarry Smith ierr = PetscFree((*snes)->kspconvctx);CHKERRQ(ierr); 33056bf464f9SBarry Smith if ((*snes)->ops->convergeddestroy) { 33066bf464f9SBarry Smith ierr = (*(*snes)->ops->convergeddestroy)((*snes)->cnvP);CHKERRQ(ierr); 33076b8b9a38SLisandro Dalcin } 3308071fcb05SBarry Smith if ((*snes)->conv_hist_alloc) { 3309071fcb05SBarry Smith ierr = PetscFree2((*snes)->conv_hist,(*snes)->conv_hist_its);CHKERRQ(ierr); 331058c9b817SLisandro Dalcin } 33116bf464f9SBarry Smith ierr = SNESMonitorCancel((*snes));CHKERRQ(ierr); 3312c4421ceaSFande Kong ierr = SNESConvergedReasonViewCancel((*snes));CHKERRQ(ierr); 3313a79aaaedSSatish Balay ierr = PetscHeaderDestroy(snes);CHKERRQ(ierr); 33143a40ed3dSBarry Smith PetscFunctionReturn(0); 33159b94acceSBarry Smith } 33169b94acceSBarry Smith 33179b94acceSBarry Smith /* ----------- Routines to set solver parameters ---------- */ 33189b94acceSBarry Smith 3319a8054027SBarry Smith /*@ 3320a8054027SBarry Smith SNESSetLagPreconditioner - Determines when the preconditioner is rebuilt in the nonlinear solve. 3321a8054027SBarry Smith 33223f9fe445SBarry Smith Logically Collective on SNES 3323a8054027SBarry Smith 3324a8054027SBarry Smith Input Parameters: 3325a8054027SBarry Smith + snes - the SNES context 3326d8e291bfSBarry Smith - lag - 1 means rebuild every time the Jacobian is computed within a single nonlinear solve, 2 means every second time 33273b4f5425SBarry Smith the Jacobian is built etc. -2 indicates rebuild preconditioner at next chance but then never rebuild after that 3328a8054027SBarry Smith 3329a8054027SBarry Smith Options Database Keys: 33303d5a8a6aSBarry Smith + -snes_lag_jacobian_persists <true,false> - sets the persistence 33313d5a8a6aSBarry Smith . -snes_lag_jacobian <-2,1,2,...> - sets the lag 33323d5a8a6aSBarry Smith . -snes_lag_preconditioner_persists <true,false> - sets the persistence 33333d5a8a6aSBarry Smith - -snes_lag_preconditioner <-2,1,2,...> - sets the lag 3334a8054027SBarry Smith 3335a8054027SBarry Smith Notes: 3336a8054027SBarry Smith The default is 1 33373d5a8a6aSBarry Smith The preconditioner is ALWAYS built in the first iteration of a nonlinear solve unless lag is -1 or SNESSetLagPreconditionerPersists() was called 3338d8e291bfSBarry Smith 3339d8e291bfSBarry Smith SNESSetLagPreconditionerPersists() allows using the same uniform lagging (for example every second solve) across multiple solves. 3340a8054027SBarry Smith 3341a8054027SBarry Smith Level: intermediate 3342a8054027SBarry Smith 33433d5a8a6aSBarry Smith .seealso: SNESSetTrustRegionTolerance(), SNESGetLagPreconditioner(), SNESSetLagJacobian(), SNESGetLagJacobian(), SNESSetLagPreconditionerPersists(), 33443d5a8a6aSBarry Smith SNESSetLagJacobianPersists() 3345a8054027SBarry Smith 3346a8054027SBarry Smith @*/ 33477087cfbeSBarry Smith PetscErrorCode SNESSetLagPreconditioner(SNES snes,PetscInt lag) 3348a8054027SBarry Smith { 3349a8054027SBarry Smith PetscFunctionBegin; 33500700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 3351e32f2f54SBarry Smith if (lag < -2) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Lag must be -2, -1, 1 or greater"); 3352e32f2f54SBarry Smith if (!lag) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Lag cannot be 0"); 3353c5eb9154SBarry Smith PetscValidLogicalCollectiveInt(snes,lag,2); 3354a8054027SBarry Smith snes->lagpreconditioner = lag; 3355a8054027SBarry Smith PetscFunctionReturn(0); 3356a8054027SBarry Smith } 3357a8054027SBarry Smith 3358efd51863SBarry Smith /*@ 3359efd51863SBarry Smith SNESSetGridSequence - sets the number of steps of grid sequencing that SNES does 3360efd51863SBarry Smith 3361efd51863SBarry Smith Logically Collective on SNES 3362efd51863SBarry Smith 3363efd51863SBarry Smith Input Parameters: 3364efd51863SBarry Smith + snes - the SNES context 3365efd51863SBarry Smith - steps - the number of refinements to do, defaults to 0 3366efd51863SBarry Smith 3367efd51863SBarry Smith Options Database Keys: 3368efd51863SBarry Smith . -snes_grid_sequence <steps> 3369efd51863SBarry Smith 3370efd51863SBarry Smith Level: intermediate 3371efd51863SBarry Smith 3372c0df2a02SJed Brown Notes: 3373c0df2a02SJed Brown Use SNESGetSolution() to extract the fine grid solution after grid sequencing. 3374c0df2a02SJed Brown 3375fa19ca70SBarry Smith .seealso: SNESSetTrustRegionTolerance(), SNESGetLagPreconditioner(), SNESSetLagJacobian(), SNESGetLagJacobian(), SNESGetGridSequence() 3376efd51863SBarry Smith 3377efd51863SBarry Smith @*/ 3378efd51863SBarry Smith PetscErrorCode SNESSetGridSequence(SNES snes,PetscInt steps) 3379efd51863SBarry Smith { 3380efd51863SBarry Smith PetscFunctionBegin; 3381efd51863SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 3382efd51863SBarry Smith PetscValidLogicalCollectiveInt(snes,steps,2); 3383efd51863SBarry Smith snes->gridsequence = steps; 3384efd51863SBarry Smith PetscFunctionReturn(0); 3385efd51863SBarry Smith } 3386efd51863SBarry Smith 3387fa19ca70SBarry Smith /*@ 3388fa19ca70SBarry Smith SNESGetGridSequence - gets the number of steps of grid sequencing that SNES does 3389fa19ca70SBarry Smith 3390fa19ca70SBarry Smith Logically Collective on SNES 3391fa19ca70SBarry Smith 3392fa19ca70SBarry Smith Input Parameter: 3393fa19ca70SBarry Smith . snes - the SNES context 3394fa19ca70SBarry Smith 3395fa19ca70SBarry Smith Output Parameter: 3396fa19ca70SBarry Smith . steps - the number of refinements to do, defaults to 0 3397fa19ca70SBarry Smith 3398fa19ca70SBarry Smith Options Database Keys: 3399fa19ca70SBarry Smith . -snes_grid_sequence <steps> 3400fa19ca70SBarry Smith 3401fa19ca70SBarry Smith Level: intermediate 3402fa19ca70SBarry Smith 3403fa19ca70SBarry Smith Notes: 3404fa19ca70SBarry Smith Use SNESGetSolution() to extract the fine grid solution after grid sequencing. 3405fa19ca70SBarry Smith 3406fa19ca70SBarry Smith .seealso: SNESSetTrustRegionTolerance(), SNESGetLagPreconditioner(), SNESSetLagJacobian(), SNESGetLagJacobian(), SNESSetGridSequence() 3407fa19ca70SBarry Smith 3408fa19ca70SBarry Smith @*/ 3409fa19ca70SBarry Smith PetscErrorCode SNESGetGridSequence(SNES snes,PetscInt *steps) 3410fa19ca70SBarry Smith { 3411fa19ca70SBarry Smith PetscFunctionBegin; 3412fa19ca70SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 3413fa19ca70SBarry Smith *steps = snes->gridsequence; 3414fa19ca70SBarry Smith PetscFunctionReturn(0); 3415fa19ca70SBarry Smith } 3416fa19ca70SBarry Smith 3417a8054027SBarry Smith /*@ 3418a8054027SBarry Smith SNESGetLagPreconditioner - Indicates how often the preconditioner is rebuilt 3419a8054027SBarry Smith 34203f9fe445SBarry Smith Not Collective 3421a8054027SBarry Smith 3422a8054027SBarry Smith Input Parameter: 3423a8054027SBarry Smith . snes - the SNES context 3424a8054027SBarry Smith 3425a8054027SBarry Smith Output Parameter: 3426a8054027SBarry 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 34273b4f5425SBarry Smith the Jacobian is built etc. -2 indicates rebuild preconditioner at next chance but then never rebuild after that 3428a8054027SBarry Smith 3429a8054027SBarry Smith Options Database Keys: 34303d5a8a6aSBarry Smith + -snes_lag_jacobian_persists <true,false> - sets the persistence 34313d5a8a6aSBarry Smith . -snes_lag_jacobian <-2,1,2,...> - sets the lag 34323d5a8a6aSBarry Smith . -snes_lag_preconditioner_persists <true,false> - sets the persistence 34333d5a8a6aSBarry Smith - -snes_lag_preconditioner <-2,1,2,...> - sets the lag 3434a8054027SBarry Smith 3435a8054027SBarry Smith Notes: 3436a8054027SBarry Smith The default is 1 3437a8054027SBarry Smith The preconditioner is ALWAYS built in the first iteration of a nonlinear solve unless lag is -1 3438a8054027SBarry Smith 3439a8054027SBarry Smith Level: intermediate 3440a8054027SBarry Smith 34413d5a8a6aSBarry Smith .seealso: SNESSetTrustRegionTolerance(), SNESSetLagPreconditioner(), SNESSetLagJacobianPersists(), SNESSetLagPreconditionerPersists() 3442a8054027SBarry Smith 3443a8054027SBarry Smith @*/ 34447087cfbeSBarry Smith PetscErrorCode SNESGetLagPreconditioner(SNES snes,PetscInt *lag) 3445a8054027SBarry Smith { 3446a8054027SBarry Smith PetscFunctionBegin; 34470700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 3448a8054027SBarry Smith *lag = snes->lagpreconditioner; 3449a8054027SBarry Smith PetscFunctionReturn(0); 3450a8054027SBarry Smith } 3451a8054027SBarry Smith 3452e35cf81dSBarry Smith /*@ 3453e35cf81dSBarry Smith SNESSetLagJacobian - Determines when the Jacobian is rebuilt in the nonlinear solve. See SNESSetLagPreconditioner() for determining how 3454e35cf81dSBarry Smith often the preconditioner is rebuilt. 3455e35cf81dSBarry Smith 34563f9fe445SBarry Smith Logically Collective on SNES 3457e35cf81dSBarry Smith 3458e35cf81dSBarry Smith Input Parameters: 3459e35cf81dSBarry Smith + snes - the SNES context 3460e35cf81dSBarry 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 3461fe3ffe1eSBarry Smith the Jacobian is built etc. -2 means rebuild at next chance but then never again 3462e35cf81dSBarry Smith 3463e35cf81dSBarry Smith Options Database Keys: 34643d5a8a6aSBarry Smith + -snes_lag_jacobian_persists <true,false> - sets the persistence 34653d5a8a6aSBarry Smith . -snes_lag_jacobian <-2,1,2,...> - sets the lag 34663d5a8a6aSBarry Smith . -snes_lag_preconditioner_persists <true,false> - sets the persistence 34673d5a8a6aSBarry Smith - -snes_lag_preconditioner <-2,1,2,...> - sets the lag. 3468e35cf81dSBarry Smith 3469e35cf81dSBarry Smith Notes: 3470e35cf81dSBarry Smith The default is 1 3471e35cf81dSBarry Smith The Jacobian is ALWAYS built in the first iteration of a nonlinear solve unless lag is -1 3472fe3ffe1eSBarry 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 3473fe3ffe1eSBarry Smith at the next Newton step but never again (unless it is reset to another value) 3474e35cf81dSBarry Smith 3475e35cf81dSBarry Smith Level: intermediate 3476e35cf81dSBarry Smith 34773d5a8a6aSBarry Smith .seealso: SNESSetTrustRegionTolerance(), SNESGetLagPreconditioner(), SNESSetLagPreconditioner(), SNESGetLagJacobianPersists(), SNESSetLagPreconditionerPersists() 3478e35cf81dSBarry Smith 3479e35cf81dSBarry Smith @*/ 34807087cfbeSBarry Smith PetscErrorCode SNESSetLagJacobian(SNES snes,PetscInt lag) 3481e35cf81dSBarry Smith { 3482e35cf81dSBarry Smith PetscFunctionBegin; 34830700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 3484e32f2f54SBarry Smith if (lag < -2) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Lag must be -2, -1, 1 or greater"); 3485e32f2f54SBarry Smith if (!lag) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Lag cannot be 0"); 3486c5eb9154SBarry Smith PetscValidLogicalCollectiveInt(snes,lag,2); 3487e35cf81dSBarry Smith snes->lagjacobian = lag; 3488e35cf81dSBarry Smith PetscFunctionReturn(0); 3489e35cf81dSBarry Smith } 3490e35cf81dSBarry Smith 3491e35cf81dSBarry Smith /*@ 3492e35cf81dSBarry Smith SNESGetLagJacobian - Indicates how often the Jacobian is rebuilt. See SNESGetLagPreconditioner() to determine when the preconditioner is rebuilt 3493e35cf81dSBarry Smith 34943f9fe445SBarry Smith Not Collective 3495e35cf81dSBarry Smith 3496e35cf81dSBarry Smith Input Parameter: 3497e35cf81dSBarry Smith . snes - the SNES context 3498e35cf81dSBarry Smith 3499e35cf81dSBarry Smith Output Parameter: 3500e35cf81dSBarry 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 3501e35cf81dSBarry Smith the Jacobian is built etc. 3502e35cf81dSBarry Smith 3503e35cf81dSBarry Smith Notes: 3504e35cf81dSBarry Smith The default is 1 35053d5a8a6aSBarry Smith The jacobian is ALWAYS built in the first iteration of a nonlinear solve unless lag is -1 or SNESSetLagJacobianPersists() was called. 3506e35cf81dSBarry Smith 3507e35cf81dSBarry Smith Level: intermediate 3508e35cf81dSBarry Smith 35093d5a8a6aSBarry Smith .seealso: SNESSetTrustRegionTolerance(), SNESSetLagJacobian(), SNESSetLagPreconditioner(), SNESGetLagPreconditioner(), SNESSetLagJacobianPersists(), SNESSetLagPreconditionerPersists() 3510e35cf81dSBarry Smith 3511e35cf81dSBarry Smith @*/ 35127087cfbeSBarry Smith PetscErrorCode SNESGetLagJacobian(SNES snes,PetscInt *lag) 3513e35cf81dSBarry Smith { 3514e35cf81dSBarry Smith PetscFunctionBegin; 35150700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 3516e35cf81dSBarry Smith *lag = snes->lagjacobian; 3517e35cf81dSBarry Smith PetscFunctionReturn(0); 3518e35cf81dSBarry Smith } 3519e35cf81dSBarry Smith 352037ec4e1aSPeter Brune /*@ 352137ec4e1aSPeter Brune SNESSetLagJacobianPersists - Set whether or not the Jacobian lagging persists through multiple solves 352237ec4e1aSPeter Brune 352337ec4e1aSPeter Brune Logically collective on SNES 352437ec4e1aSPeter Brune 352537ec4e1aSPeter Brune Input Parameter: 352637ec4e1aSPeter Brune + snes - the SNES context 35279d7e2deaSPeter Brune - flg - jacobian lagging persists if true 352837ec4e1aSPeter Brune 352937ec4e1aSPeter Brune Options Database Keys: 35303d5a8a6aSBarry Smith + -snes_lag_jacobian_persists <true,false> - sets the persistence 35313d5a8a6aSBarry Smith . -snes_lag_jacobian <-2,1,2,...> - sets the lag 35323d5a8a6aSBarry Smith . -snes_lag_preconditioner_persists <true,false> - sets the persistence 35333d5a8a6aSBarry Smith - -snes_lag_preconditioner <-2,1,2,...> - sets the lag 35343d5a8a6aSBarry Smith 353537ec4e1aSPeter Brune 353695452b02SPatrick Sanan Notes: 353795452b02SPatrick Sanan This is useful both for nonlinear preconditioning, where it's appropriate to have the Jacobian be stale by 353837ec4e1aSPeter Brune several solves, and for implicit time-stepping, where Jacobian lagging in the inner nonlinear solve over several 353937ec4e1aSPeter Brune timesteps may present huge efficiency gains. 354037ec4e1aSPeter Brune 354137ec4e1aSPeter Brune Level: developer 354237ec4e1aSPeter Brune 35433d5a8a6aSBarry Smith .seealso: SNESSetLagPreconditionerPersists(), SNESSetLagJacobian(), SNESGetLagJacobian(), SNESGetNPC(), SNESSetLagJacobianPersists() 354437ec4e1aSPeter Brune 354537ec4e1aSPeter Brune @*/ 354637ec4e1aSPeter Brune PetscErrorCode SNESSetLagJacobianPersists(SNES snes,PetscBool flg) 354737ec4e1aSPeter Brune { 354837ec4e1aSPeter Brune PetscFunctionBegin; 354937ec4e1aSPeter Brune PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 355037ec4e1aSPeter Brune PetscValidLogicalCollectiveBool(snes,flg,2); 355137ec4e1aSPeter Brune snes->lagjac_persist = flg; 355237ec4e1aSPeter Brune PetscFunctionReturn(0); 355337ec4e1aSPeter Brune } 355437ec4e1aSPeter Brune 355537ec4e1aSPeter Brune /*@ 3556d8e291bfSBarry Smith SNESSetLagPreconditionerPersists - Set whether or not the preconditioner lagging persists through multiple nonlinear solves 355737ec4e1aSPeter Brune 355837ec4e1aSPeter Brune Logically Collective on SNES 355937ec4e1aSPeter Brune 356037ec4e1aSPeter Brune Input Parameter: 356137ec4e1aSPeter Brune + snes - the SNES context 35629d7e2deaSPeter Brune - flg - preconditioner lagging persists if true 356337ec4e1aSPeter Brune 356437ec4e1aSPeter Brune Options Database Keys: 35653d5a8a6aSBarry Smith + -snes_lag_jacobian_persists <true,false> - sets the persistence 35663d5a8a6aSBarry Smith . -snes_lag_jacobian <-2,1,2,...> - sets the lag 35673d5a8a6aSBarry Smith . -snes_lag_preconditioner_persists <true,false> - sets the persistence 35683d5a8a6aSBarry Smith - -snes_lag_preconditioner <-2,1,2,...> - sets the lag 356937ec4e1aSPeter Brune 357095452b02SPatrick Sanan Notes: 357195452b02SPatrick Sanan This is useful both for nonlinear preconditioning, where it's appropriate to have the preconditioner be stale 357237ec4e1aSPeter Brune by several solves, and for implicit time-stepping, where preconditioner lagging in the inner nonlinear solve over 357337ec4e1aSPeter Brune several timesteps may present huge efficiency gains. 357437ec4e1aSPeter Brune 357537ec4e1aSPeter Brune Level: developer 357637ec4e1aSPeter Brune 35773d5a8a6aSBarry Smith .seealso: SNESSetLagJacobianPersists(), SNESSetLagJacobian(), SNESGetLagJacobian(), SNESGetNPC(), SNESSetLagPreconditioner() 357837ec4e1aSPeter Brune 357937ec4e1aSPeter Brune @*/ 358037ec4e1aSPeter Brune PetscErrorCode SNESSetLagPreconditionerPersists(SNES snes,PetscBool flg) 358137ec4e1aSPeter Brune { 358237ec4e1aSPeter Brune PetscFunctionBegin; 358337ec4e1aSPeter Brune PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 358437ec4e1aSPeter Brune PetscValidLogicalCollectiveBool(snes,flg,2); 358537ec4e1aSPeter Brune snes->lagpre_persist = flg; 358637ec4e1aSPeter Brune PetscFunctionReturn(0); 358737ec4e1aSPeter Brune } 358837ec4e1aSPeter Brune 35899b94acceSBarry Smith /*@ 3590be5caee7SBarry Smith SNESSetForceIteration - force SNESSolve() to take at least one iteration regardless of the initial residual norm 3591be5caee7SBarry Smith 3592be5caee7SBarry Smith Logically Collective on SNES 3593be5caee7SBarry Smith 3594be5caee7SBarry Smith Input Parameters: 3595be5caee7SBarry Smith + snes - the SNES context 3596be5caee7SBarry Smith - force - PETSC_TRUE require at least one iteration 3597be5caee7SBarry Smith 3598be5caee7SBarry Smith Options Database Keys: 3599be5caee7SBarry Smith . -snes_force_iteration <force> - Sets forcing an iteration 3600be5caee7SBarry Smith 3601be5caee7SBarry Smith Notes: 3602be5caee7SBarry Smith This is used sometimes with TS to prevent TS from detecting a false steady state solution 3603be5caee7SBarry Smith 3604be5caee7SBarry Smith Level: intermediate 3605be5caee7SBarry Smith 3606be5caee7SBarry Smith .seealso: SNESSetTrustRegionTolerance(), SNESSetDivergenceTolerance() 3607be5caee7SBarry Smith @*/ 3608be5caee7SBarry Smith PetscErrorCode SNESSetForceIteration(SNES snes,PetscBool force) 3609be5caee7SBarry Smith { 3610be5caee7SBarry Smith PetscFunctionBegin; 3611be5caee7SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 3612be5caee7SBarry Smith snes->forceiteration = force; 3613be5caee7SBarry Smith PetscFunctionReturn(0); 3614be5caee7SBarry Smith } 3615be5caee7SBarry Smith 361685216dc7SFande Kong /*@ 361785216dc7SFande Kong SNESGetForceIteration - Whether or not to force SNESSolve() take at least one iteration regardless of the initial residual norm 361885216dc7SFande Kong 361985216dc7SFande Kong Logically Collective on SNES 362085216dc7SFande Kong 362185216dc7SFande Kong Input Parameters: 362285216dc7SFande Kong . snes - the SNES context 362385216dc7SFande Kong 362485216dc7SFande Kong Output Parameter: 362585216dc7SFande Kong . force - PETSC_TRUE requires at least one iteration. 362685216dc7SFande Kong 362706dd6b0eSSatish Balay Level: intermediate 362806dd6b0eSSatish Balay 362985216dc7SFande Kong .seealso: SNESSetForceIteration(), SNESSetTrustRegionTolerance(), SNESSetDivergenceTolerance() 363085216dc7SFande Kong @*/ 363185216dc7SFande Kong PetscErrorCode SNESGetForceIteration(SNES snes,PetscBool *force) 363285216dc7SFande Kong { 363385216dc7SFande Kong PetscFunctionBegin; 363485216dc7SFande Kong PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 363585216dc7SFande Kong *force = snes->forceiteration; 363685216dc7SFande Kong PetscFunctionReturn(0); 363785216dc7SFande Kong } 3638be5caee7SBarry Smith 3639be5caee7SBarry Smith /*@ 3640d7a720efSLois Curfman McInnes SNESSetTolerances - Sets various parameters used in convergence tests. 36419b94acceSBarry Smith 36423f9fe445SBarry Smith Logically Collective on SNES 3643c7afd0dbSLois Curfman McInnes 36449b94acceSBarry Smith Input Parameters: 3645c7afd0dbSLois Curfman McInnes + snes - the SNES context 364670441072SBarry Smith . abstol - absolute convergence tolerance 364733174efeSLois Curfman McInnes . rtol - relative convergence tolerance 36485358d0d4SBarry Smith . stol - convergence tolerance in terms of the norm of the change in the solution between steps, || delta x || < stol*|| x || 364933174efeSLois Curfman McInnes . maxit - maximum number of iterations 3650e71169deSBarry Smith - maxf - maximum number of function evaluations (-1 indicates no limit) 3651fee21e36SBarry Smith 365233174efeSLois Curfman McInnes Options Database Keys: 365370441072SBarry Smith + -snes_atol <abstol> - Sets abstol 3654c7afd0dbSLois Curfman McInnes . -snes_rtol <rtol> - Sets rtol 3655c7afd0dbSLois Curfman McInnes . -snes_stol <stol> - Sets stol 3656c7afd0dbSLois Curfman McInnes . -snes_max_it <maxit> - Sets maxit 3657c7afd0dbSLois Curfman McInnes - -snes_max_funcs <maxf> - Sets maxf 36589b94acceSBarry Smith 3659d7a720efSLois Curfman McInnes Notes: 36609b94acceSBarry Smith The default maximum number of iterations is 50. 36619b94acceSBarry Smith The default maximum number of function evaluations is 1000. 36629b94acceSBarry Smith 366336851e7fSLois Curfman McInnes Level: intermediate 366436851e7fSLois Curfman McInnes 3665be5caee7SBarry Smith .seealso: SNESSetTrustRegionTolerance(), SNESSetDivergenceTolerance(), SNESSetForceIteration() 36669b94acceSBarry Smith @*/ 36677087cfbeSBarry Smith PetscErrorCode SNESSetTolerances(SNES snes,PetscReal abstol,PetscReal rtol,PetscReal stol,PetscInt maxit,PetscInt maxf) 36689b94acceSBarry Smith { 36693a40ed3dSBarry Smith PetscFunctionBegin; 36700700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 3671c5eb9154SBarry Smith PetscValidLogicalCollectiveReal(snes,abstol,2); 3672c5eb9154SBarry Smith PetscValidLogicalCollectiveReal(snes,rtol,3); 3673c5eb9154SBarry Smith PetscValidLogicalCollectiveReal(snes,stol,4); 3674c5eb9154SBarry Smith PetscValidLogicalCollectiveInt(snes,maxit,5); 3675c5eb9154SBarry Smith PetscValidLogicalCollectiveInt(snes,maxf,6); 3676c5eb9154SBarry Smith 3677ab54825eSJed Brown if (abstol != PETSC_DEFAULT) { 367857622a8eSBarry Smith if (abstol < 0.0) SETERRQ1(PetscObjectComm((PetscObject)snes),PETSC_ERR_ARG_OUTOFRANGE,"Absolute tolerance %g must be non-negative",(double)abstol); 3679ab54825eSJed Brown snes->abstol = abstol; 3680ab54825eSJed Brown } 3681ab54825eSJed Brown if (rtol != PETSC_DEFAULT) { 368257622a8eSBarry 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); 3683ab54825eSJed Brown snes->rtol = rtol; 3684ab54825eSJed Brown } 3685ab54825eSJed Brown if (stol != PETSC_DEFAULT) { 368657622a8eSBarry Smith if (stol < 0.0) SETERRQ1(PetscObjectComm((PetscObject)snes),PETSC_ERR_ARG_OUTOFRANGE,"Step tolerance %g must be non-negative",(double)stol); 3687c60f73f4SPeter Brune snes->stol = stol; 3688ab54825eSJed Brown } 3689ab54825eSJed Brown if (maxit != PETSC_DEFAULT) { 3690ce94432eSBarry Smith if (maxit < 0) SETERRQ1(PetscObjectComm((PetscObject)snes),PETSC_ERR_ARG_OUTOFRANGE,"Maximum number of iterations %D must be non-negative",maxit); 3691ab54825eSJed Brown snes->max_its = maxit; 3692ab54825eSJed Brown } 3693ab54825eSJed Brown if (maxf != PETSC_DEFAULT) { 3694e71169deSBarry Smith if (maxf < -1) SETERRQ1(PetscObjectComm((PetscObject)snes),PETSC_ERR_ARG_OUTOFRANGE,"Maximum number of function evaluations %D must be -1 or nonnegative",maxf); 3695ab54825eSJed Brown snes->max_funcs = maxf; 3696ab54825eSJed Brown } 369788976e71SPeter Brune snes->tolerancesset = PETSC_TRUE; 36983a40ed3dSBarry Smith PetscFunctionReturn(0); 36999b94acceSBarry Smith } 37009b94acceSBarry Smith 3701e4d06f11SPatrick Farrell /*@ 3702e4d06f11SPatrick Farrell SNESSetDivergenceTolerance - Sets the divergence tolerance used for the SNES divergence test. 3703e4d06f11SPatrick Farrell 3704e4d06f11SPatrick Farrell Logically Collective on SNES 3705e4d06f11SPatrick Farrell 3706e4d06f11SPatrick Farrell Input Parameters: 3707e4d06f11SPatrick Farrell + snes - the SNES context 3708e4d06f11SPatrick Farrell - divtol - the divergence tolerance. Use -1 to deactivate the test. 3709e4d06f11SPatrick Farrell 3710e4d06f11SPatrick Farrell Options Database Keys: 3711a2b725a8SWilliam Gropp . -snes_divergence_tolerance <divtol> - Sets divtol 3712e4d06f11SPatrick Farrell 3713e4d06f11SPatrick Farrell Notes: 3714e4d06f11SPatrick Farrell The default divergence tolerance is 1e4. 3715e4d06f11SPatrick Farrell 3716e4d06f11SPatrick Farrell Level: intermediate 3717e4d06f11SPatrick Farrell 3718e4d06f11SPatrick Farrell .seealso: SNESSetTolerances(), SNESGetDivergenceTolerance 3719e4d06f11SPatrick Farrell @*/ 3720e4d06f11SPatrick Farrell PetscErrorCode SNESSetDivergenceTolerance(SNES snes,PetscReal divtol) 3721e4d06f11SPatrick Farrell { 3722e4d06f11SPatrick Farrell PetscFunctionBegin; 3723e4d06f11SPatrick Farrell PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 3724e4d06f11SPatrick Farrell PetscValidLogicalCollectiveReal(snes,divtol,2); 3725e4d06f11SPatrick Farrell 3726e4d06f11SPatrick Farrell if (divtol != PETSC_DEFAULT) { 3727e4d06f11SPatrick Farrell snes->divtol = divtol; 3728e4d06f11SPatrick Farrell } 3729e4d06f11SPatrick Farrell else { 3730e4d06f11SPatrick Farrell snes->divtol = 1.0e4; 3731e4d06f11SPatrick Farrell } 3732e4d06f11SPatrick Farrell PetscFunctionReturn(0); 3733e4d06f11SPatrick Farrell } 3734e4d06f11SPatrick Farrell 37359b94acceSBarry Smith /*@ 373633174efeSLois Curfman McInnes SNESGetTolerances - Gets various parameters used in convergence tests. 373733174efeSLois Curfman McInnes 3738c7afd0dbSLois Curfman McInnes Not Collective 3739c7afd0dbSLois Curfman McInnes 374033174efeSLois Curfman McInnes Input Parameters: 3741c7afd0dbSLois Curfman McInnes + snes - the SNES context 374285385478SLisandro Dalcin . atol - absolute convergence tolerance 374333174efeSLois Curfman McInnes . rtol - relative convergence tolerance 374433174efeSLois Curfman McInnes . stol - convergence tolerance in terms of the norm 374533174efeSLois Curfman McInnes of the change in the solution between steps 374633174efeSLois Curfman McInnes . maxit - maximum number of iterations 3747c7afd0dbSLois Curfman McInnes - maxf - maximum number of function evaluations 3748fee21e36SBarry Smith 374933174efeSLois Curfman McInnes Notes: 37500298fd71SBarry Smith The user can specify NULL for any parameter that is not needed. 375133174efeSLois Curfman McInnes 375236851e7fSLois Curfman McInnes Level: intermediate 375336851e7fSLois Curfman McInnes 375433174efeSLois Curfman McInnes .seealso: SNESSetTolerances() 375533174efeSLois Curfman McInnes @*/ 37567087cfbeSBarry Smith PetscErrorCode SNESGetTolerances(SNES snes,PetscReal *atol,PetscReal *rtol,PetscReal *stol,PetscInt *maxit,PetscInt *maxf) 375733174efeSLois Curfman McInnes { 37583a40ed3dSBarry Smith PetscFunctionBegin; 37590700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 376085385478SLisandro Dalcin if (atol) *atol = snes->abstol; 376133174efeSLois Curfman McInnes if (rtol) *rtol = snes->rtol; 3762c60f73f4SPeter Brune if (stol) *stol = snes->stol; 376333174efeSLois Curfman McInnes if (maxit) *maxit = snes->max_its; 376433174efeSLois Curfman McInnes if (maxf) *maxf = snes->max_funcs; 37653a40ed3dSBarry Smith PetscFunctionReturn(0); 376633174efeSLois Curfman McInnes } 376733174efeSLois Curfman McInnes 3768e4d06f11SPatrick Farrell /*@ 3769e4d06f11SPatrick Farrell SNESGetDivergenceTolerance - Gets divergence tolerance used in divergence test. 3770e4d06f11SPatrick Farrell 3771e4d06f11SPatrick Farrell Not Collective 3772e4d06f11SPatrick Farrell 3773e4d06f11SPatrick Farrell Input Parameters: 3774e4d06f11SPatrick Farrell + snes - the SNES context 3775e4d06f11SPatrick Farrell - divtol - divergence tolerance 3776e4d06f11SPatrick Farrell 3777e4d06f11SPatrick Farrell Level: intermediate 3778e4d06f11SPatrick Farrell 3779e4d06f11SPatrick Farrell .seealso: SNESSetDivergenceTolerance() 3780e4d06f11SPatrick Farrell @*/ 3781e4d06f11SPatrick Farrell PetscErrorCode SNESGetDivergenceTolerance(SNES snes,PetscReal *divtol) 3782e4d06f11SPatrick Farrell { 3783e4d06f11SPatrick Farrell PetscFunctionBegin; 3784e4d06f11SPatrick Farrell PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 3785e4d06f11SPatrick Farrell if (divtol) *divtol = snes->divtol; 3786e4d06f11SPatrick Farrell PetscFunctionReturn(0); 3787e4d06f11SPatrick Farrell } 3788e4d06f11SPatrick Farrell 378933174efeSLois Curfman McInnes /*@ 37909b94acceSBarry Smith SNESSetTrustRegionTolerance - Sets the trust region parameter tolerance. 37919b94acceSBarry Smith 37923f9fe445SBarry Smith Logically Collective on SNES 3793fee21e36SBarry Smith 3794c7afd0dbSLois Curfman McInnes Input Parameters: 3795c7afd0dbSLois Curfman McInnes + snes - the SNES context 3796c7afd0dbSLois Curfman McInnes - tol - tolerance 3797c7afd0dbSLois Curfman McInnes 37989b94acceSBarry Smith Options Database Key: 3799c7afd0dbSLois Curfman McInnes . -snes_trtol <tol> - Sets tol 38009b94acceSBarry Smith 380136851e7fSLois Curfman McInnes Level: intermediate 380236851e7fSLois Curfman McInnes 38032492ecdbSBarry Smith .seealso: SNESSetTolerances() 38049b94acceSBarry Smith @*/ 38057087cfbeSBarry Smith PetscErrorCode SNESSetTrustRegionTolerance(SNES snes,PetscReal tol) 38069b94acceSBarry Smith { 38073a40ed3dSBarry Smith PetscFunctionBegin; 38080700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 3809c5eb9154SBarry Smith PetscValidLogicalCollectiveReal(snes,tol,2); 38109b94acceSBarry Smith snes->deltatol = tol; 38113a40ed3dSBarry Smith PetscFunctionReturn(0); 38129b94acceSBarry Smith } 38139b94acceSBarry Smith 38146ba87a44SLisandro Dalcin PETSC_INTERN PetscErrorCode SNESMonitorRange_Private(SNES,PetscInt,PetscReal*); 38156ba87a44SLisandro Dalcin 38167087cfbeSBarry Smith PetscErrorCode SNESMonitorLGRange(SNES snes,PetscInt n,PetscReal rnorm,void *monctx) 3817b271bb04SBarry Smith { 3818b271bb04SBarry Smith PetscDrawLG lg; 3819b271bb04SBarry Smith PetscErrorCode ierr; 3820b271bb04SBarry Smith PetscReal x,y,per; 3821b271bb04SBarry Smith PetscViewer v = (PetscViewer)monctx; 3822b271bb04SBarry Smith static PetscReal prev; /* should be in the context */ 3823b271bb04SBarry Smith PetscDraw draw; 3824b271bb04SBarry Smith 3825459f5d12SBarry Smith PetscFunctionBegin; 38264d4332d5SBarry Smith PetscValidHeaderSpecific(v,PETSC_VIEWER_CLASSID,4); 3827b271bb04SBarry Smith ierr = PetscViewerDrawGetDrawLG(v,0,&lg);CHKERRQ(ierr); 3828b271bb04SBarry Smith if (!n) {ierr = PetscDrawLGReset(lg);CHKERRQ(ierr);} 3829b271bb04SBarry Smith ierr = PetscDrawLGGetDraw(lg,&draw);CHKERRQ(ierr); 3830b271bb04SBarry Smith ierr = PetscDrawSetTitle(draw,"Residual norm");CHKERRQ(ierr); 3831b271bb04SBarry Smith x = (PetscReal)n; 383277b4d14cSPeter Brune if (rnorm > 0.0) y = PetscLog10Real(rnorm); 383394c9c6d3SKarl Rupp else y = -15.0; 3834b271bb04SBarry Smith ierr = PetscDrawLGAddPoint(lg,&x,&y);CHKERRQ(ierr); 38356934998bSLisandro Dalcin if (n < 20 || !(n % 5) || snes->reason) { 3836b271bb04SBarry Smith ierr = PetscDrawLGDraw(lg);CHKERRQ(ierr); 38376934998bSLisandro Dalcin ierr = PetscDrawLGSave(lg);CHKERRQ(ierr); 3838b271bb04SBarry Smith } 3839b271bb04SBarry Smith 3840b271bb04SBarry Smith ierr = PetscViewerDrawGetDrawLG(v,1,&lg);CHKERRQ(ierr); 3841b271bb04SBarry Smith if (!n) {ierr = PetscDrawLGReset(lg);CHKERRQ(ierr);} 3842b271bb04SBarry Smith ierr = PetscDrawLGGetDraw(lg,&draw);CHKERRQ(ierr); 3843b271bb04SBarry Smith ierr = PetscDrawSetTitle(draw,"% elemts > .2*max elemt");CHKERRQ(ierr); 3844b271bb04SBarry Smith ierr = SNESMonitorRange_Private(snes,n,&per);CHKERRQ(ierr); 3845b271bb04SBarry Smith x = (PetscReal)n; 3846b271bb04SBarry Smith y = 100.0*per; 3847b271bb04SBarry Smith ierr = PetscDrawLGAddPoint(lg,&x,&y);CHKERRQ(ierr); 38486934998bSLisandro Dalcin if (n < 20 || !(n % 5) || snes->reason) { 3849b271bb04SBarry Smith ierr = PetscDrawLGDraw(lg);CHKERRQ(ierr); 38506934998bSLisandro Dalcin ierr = PetscDrawLGSave(lg);CHKERRQ(ierr); 3851b271bb04SBarry Smith } 3852b271bb04SBarry Smith 3853b271bb04SBarry Smith ierr = PetscViewerDrawGetDrawLG(v,2,&lg);CHKERRQ(ierr); 3854b271bb04SBarry Smith if (!n) {prev = rnorm;ierr = PetscDrawLGReset(lg);CHKERRQ(ierr);} 3855b271bb04SBarry Smith ierr = PetscDrawLGGetDraw(lg,&draw);CHKERRQ(ierr); 3856b271bb04SBarry Smith ierr = PetscDrawSetTitle(draw,"(norm -oldnorm)/oldnorm");CHKERRQ(ierr); 3857b271bb04SBarry Smith x = (PetscReal)n; 3858b271bb04SBarry Smith y = (prev - rnorm)/prev; 3859b271bb04SBarry Smith ierr = PetscDrawLGAddPoint(lg,&x,&y);CHKERRQ(ierr); 38606934998bSLisandro Dalcin if (n < 20 || !(n % 5) || snes->reason) { 3861b271bb04SBarry Smith ierr = PetscDrawLGDraw(lg);CHKERRQ(ierr); 38626934998bSLisandro Dalcin ierr = PetscDrawLGSave(lg);CHKERRQ(ierr); 3863b271bb04SBarry Smith } 3864b271bb04SBarry Smith 3865b271bb04SBarry Smith ierr = PetscViewerDrawGetDrawLG(v,3,&lg);CHKERRQ(ierr); 3866b271bb04SBarry Smith if (!n) {ierr = PetscDrawLGReset(lg);CHKERRQ(ierr);} 3867b271bb04SBarry Smith ierr = PetscDrawLGGetDraw(lg,&draw);CHKERRQ(ierr); 3868b271bb04SBarry Smith ierr = PetscDrawSetTitle(draw,"(norm -oldnorm)/oldnorm*(% > .2 max)");CHKERRQ(ierr); 3869b271bb04SBarry Smith x = (PetscReal)n; 3870b271bb04SBarry Smith y = (prev - rnorm)/(prev*per); 3871b271bb04SBarry Smith if (n > 2) { /*skip initial crazy value */ 3872b271bb04SBarry Smith ierr = PetscDrawLGAddPoint(lg,&x,&y);CHKERRQ(ierr); 3873b271bb04SBarry Smith } 38746934998bSLisandro Dalcin if (n < 20 || !(n % 5) || snes->reason) { 3875b271bb04SBarry Smith ierr = PetscDrawLGDraw(lg);CHKERRQ(ierr); 38766934998bSLisandro Dalcin ierr = PetscDrawLGSave(lg);CHKERRQ(ierr); 3877b271bb04SBarry Smith } 3878b271bb04SBarry Smith prev = rnorm; 3879b271bb04SBarry Smith PetscFunctionReturn(0); 3880b271bb04SBarry Smith } 3881b271bb04SBarry Smith 3882228d79bcSJed Brown /*@ 3883228d79bcSJed Brown SNESMonitor - runs the user provided monitor routines, if they exist 3884228d79bcSJed Brown 3885228d79bcSJed Brown Collective on SNES 3886228d79bcSJed Brown 3887228d79bcSJed Brown Input Parameters: 3888228d79bcSJed Brown + snes - nonlinear solver context obtained from SNESCreate() 3889228d79bcSJed Brown . iter - iteration number 3890228d79bcSJed Brown - rnorm - relative norm of the residual 3891228d79bcSJed Brown 3892228d79bcSJed Brown Notes: 3893228d79bcSJed Brown This routine is called by the SNES implementations. 3894228d79bcSJed Brown It does not typically need to be called by the user. 3895228d79bcSJed Brown 3896228d79bcSJed Brown Level: developer 3897228d79bcSJed Brown 3898228d79bcSJed Brown .seealso: SNESMonitorSet() 3899228d79bcSJed Brown @*/ 39007a03ce2fSLisandro Dalcin PetscErrorCode SNESMonitor(SNES snes,PetscInt iter,PetscReal rnorm) 39017a03ce2fSLisandro Dalcin { 39027a03ce2fSLisandro Dalcin PetscErrorCode ierr; 39037a03ce2fSLisandro Dalcin PetscInt i,n = snes->numbermonitors; 39047a03ce2fSLisandro Dalcin 39057a03ce2fSLisandro Dalcin PetscFunctionBegin; 39068860a134SJunchao Zhang ierr = VecLockReadPush(snes->vec_sol);CHKERRQ(ierr); 39077a03ce2fSLisandro Dalcin for (i=0; i<n; i++) { 39087a03ce2fSLisandro Dalcin ierr = (*snes->monitor[i])(snes,iter,rnorm,snes->monitorcontext[i]);CHKERRQ(ierr); 39097a03ce2fSLisandro Dalcin } 39108860a134SJunchao Zhang ierr = VecLockReadPop(snes->vec_sol);CHKERRQ(ierr); 39117a03ce2fSLisandro Dalcin PetscFunctionReturn(0); 39127a03ce2fSLisandro Dalcin } 39137a03ce2fSLisandro Dalcin 39149b94acceSBarry Smith /* ------------ Routines to set performance monitoring options ----------- */ 39159b94acceSBarry Smith 3916bf388a1fSBarry Smith /*MC 3917bf388a1fSBarry Smith SNESMonitorFunction - functional form passed to SNESMonitorSet() to monitor convergence of nonlinear solver 3918bf388a1fSBarry Smith 3919bf388a1fSBarry Smith Synopsis: 3920aaa7dc30SBarry Smith #include <petscsnes.h> 3921bf388a1fSBarry Smith $ PetscErrorCode SNESMonitorFunction(SNES snes,PetscInt its, PetscReal norm,void *mctx) 3922bf388a1fSBarry Smith 39231843f636SBarry Smith Collective on snes 39241843f636SBarry Smith 39251843f636SBarry Smith Input Parameters: 3926bf388a1fSBarry Smith + snes - the SNES context 3927bf388a1fSBarry Smith . its - iteration number 3928bf388a1fSBarry Smith . norm - 2-norm function value (may be estimated) 3929bf388a1fSBarry Smith - mctx - [optional] monitoring context 3930bf388a1fSBarry Smith 3931878cb397SSatish Balay Level: advanced 3932878cb397SSatish Balay 3933bf388a1fSBarry Smith .seealso: SNESMonitorSet(), SNESMonitorGet() 3934bf388a1fSBarry Smith M*/ 3935bf388a1fSBarry Smith 39369b94acceSBarry Smith /*@C 3937a6570f20SBarry Smith SNESMonitorSet - Sets an ADDITIONAL function that is to be used at every 39389b94acceSBarry Smith iteration of the nonlinear solver to display the iteration's 39399b94acceSBarry Smith progress. 39409b94acceSBarry Smith 39413f9fe445SBarry Smith Logically Collective on SNES 3942fee21e36SBarry Smith 3943c7afd0dbSLois Curfman McInnes Input Parameters: 3944c7afd0dbSLois Curfman McInnes + snes - the SNES context 39456e4dcb14SBarry Smith . f - the monitor function, see SNESMonitorFunction for the calling sequence 3946b8a78c4aSBarry Smith . mctx - [optional] user-defined context for private data for the 39470298fd71SBarry Smith monitor routine (use NULL if no context is desired) 3948b3006f0bSLois Curfman McInnes - monitordestroy - [optional] routine that frees monitor context 39490298fd71SBarry Smith (may be NULL) 39509b94acceSBarry Smith 39519665c990SLois Curfman McInnes Options Database Keys: 3952a6570f20SBarry Smith + -snes_monitor - sets SNESMonitorDefault() 3953798534f6SMatthew G. Knepley . -snes_monitor draw::draw_lg - sets line graph monitor, 3954cca6129bSJed Brown - -snes_monitor_cancel - cancels all monitors that have 3955c7afd0dbSLois Curfman McInnes been hardwired into a code by 3956a6570f20SBarry Smith calls to SNESMonitorSet(), but 3957c7afd0dbSLois Curfman McInnes does not cancel those set via 3958c7afd0dbSLois Curfman McInnes the options database. 39599665c990SLois Curfman McInnes 3960639f9d9dSBarry Smith Notes: 39616bc08f3fSLois Curfman McInnes Several different monitoring routines may be set by calling 3962a6570f20SBarry Smith SNESMonitorSet() multiple times; all will be called in the 39636bc08f3fSLois Curfman McInnes order in which they were set. 3964639f9d9dSBarry Smith 396595452b02SPatrick Sanan Fortran Notes: 396695452b02SPatrick Sanan Only a single monitor function can be set for each SNES object 3967025f1a04SBarry Smith 396836851e7fSLois Curfman McInnes Level: intermediate 396936851e7fSLois Curfman McInnes 3970bf388a1fSBarry Smith .seealso: SNESMonitorDefault(), SNESMonitorCancel(), SNESMonitorFunction 39719b94acceSBarry Smith @*/ 39726e4dcb14SBarry Smith PetscErrorCode SNESMonitorSet(SNES snes,PetscErrorCode (*f)(SNES,PetscInt,PetscReal,void*),void *mctx,PetscErrorCode (*monitordestroy)(void**)) 39739b94acceSBarry Smith { 3974b90d0a6eSBarry Smith PetscInt i; 3975649052a6SBarry Smith PetscErrorCode ierr; 397678064530SBarry Smith PetscBool identical; 3977b90d0a6eSBarry Smith 39783a40ed3dSBarry Smith PetscFunctionBegin; 39790700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 3980b90d0a6eSBarry Smith for (i=0; i<snes->numbermonitors;i++) { 398178064530SBarry Smith ierr = PetscMonitorCompare((PetscErrorCode (*)(void))f,mctx,monitordestroy,(PetscErrorCode (*)(void))snes->monitor[i],snes->monitorcontext[i],snes->monitordestroy[i],&identical);CHKERRQ(ierr); 398278064530SBarry Smith if (identical) PetscFunctionReturn(0); 3983649052a6SBarry Smith } 398478064530SBarry Smith if (snes->numbermonitors >= MAXSNESMONITORS) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Too many monitors set"); 39856e4dcb14SBarry Smith snes->monitor[snes->numbermonitors] = f; 3986b8a78c4aSBarry Smith snes->monitordestroy[snes->numbermonitors] = monitordestroy; 3987639f9d9dSBarry Smith snes->monitorcontext[snes->numbermonitors++] = (void*)mctx; 39883a40ed3dSBarry Smith PetscFunctionReturn(0); 39899b94acceSBarry Smith } 39909b94acceSBarry Smith 3991a278d85bSSatish Balay /*@ 3992a6570f20SBarry Smith SNESMonitorCancel - Clears all the monitor functions for a SNES object. 39935cd90555SBarry Smith 39943f9fe445SBarry Smith Logically Collective on SNES 3995c7afd0dbSLois Curfman McInnes 39965cd90555SBarry Smith Input Parameters: 39975cd90555SBarry Smith . snes - the SNES context 39985cd90555SBarry Smith 39991a480d89SAdministrator Options Database Key: 4000a6570f20SBarry Smith . -snes_monitor_cancel - cancels all monitors that have been hardwired 4001a6570f20SBarry Smith into a code by calls to SNESMonitorSet(), but does not cancel those 4002c7afd0dbSLois Curfman McInnes set via the options database 40035cd90555SBarry Smith 40045cd90555SBarry Smith Notes: 40055cd90555SBarry Smith There is no way to clear one specific monitor from a SNES object. 40065cd90555SBarry Smith 400736851e7fSLois Curfman McInnes Level: intermediate 400836851e7fSLois Curfman McInnes 4009a6570f20SBarry Smith .seealso: SNESMonitorDefault(), SNESMonitorSet() 40105cd90555SBarry Smith @*/ 40117087cfbeSBarry Smith PetscErrorCode SNESMonitorCancel(SNES snes) 40125cd90555SBarry Smith { 4013d952e501SBarry Smith PetscErrorCode ierr; 4014d952e501SBarry Smith PetscInt i; 4015d952e501SBarry Smith 40165cd90555SBarry Smith PetscFunctionBegin; 40170700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 4018d952e501SBarry Smith for (i=0; i<snes->numbermonitors; i++) { 4019d952e501SBarry Smith if (snes->monitordestroy[i]) { 40203c4aec1bSBarry Smith ierr = (*snes->monitordestroy[i])(&snes->monitorcontext[i]);CHKERRQ(ierr); 4021d952e501SBarry Smith } 4022d952e501SBarry Smith } 40235cd90555SBarry Smith snes->numbermonitors = 0; 40245cd90555SBarry Smith PetscFunctionReturn(0); 40255cd90555SBarry Smith } 40265cd90555SBarry Smith 4027bf388a1fSBarry Smith /*MC 4028bf388a1fSBarry Smith SNESConvergenceTestFunction - functional form used for testing of convergence of nonlinear solver 4029bf388a1fSBarry Smith 4030bf388a1fSBarry Smith Synopsis: 4031aaa7dc30SBarry Smith #include <petscsnes.h> 4032bf388a1fSBarry Smith $ PetscErrorCode SNESConvergenceTest(SNES snes,PetscInt it,PetscReal xnorm,PetscReal gnorm,PetscReal f,SNESConvergedReason *reason,void *cctx) 4033bf388a1fSBarry Smith 40341843f636SBarry Smith Collective on snes 40351843f636SBarry Smith 40361843f636SBarry Smith Input Parameters: 4037bf388a1fSBarry Smith + snes - the SNES context 4038bf388a1fSBarry Smith . it - current iteration (0 is the first and is before any Newton step) 4039bf388a1fSBarry Smith . xnorm - 2-norm of current iterate 4040bf388a1fSBarry Smith . gnorm - 2-norm of current step 40411843f636SBarry Smith . f - 2-norm of function 40421843f636SBarry Smith - cctx - [optional] convergence context 40431843f636SBarry Smith 40441843f636SBarry Smith Output Parameter: 40451843f636SBarry Smith . reason - reason for convergence/divergence, only needs to be set when convergence or divergence is detected 4046bf388a1fSBarry Smith 4047878cb397SSatish Balay Level: intermediate 4048bf388a1fSBarry Smith 4049bf388a1fSBarry Smith .seealso: SNESSetConvergenceTest(), SNESGetConvergenceTest() 4050bf388a1fSBarry Smith M*/ 4051bf388a1fSBarry Smith 40529b94acceSBarry Smith /*@C 40539b94acceSBarry Smith SNESSetConvergenceTest - Sets the function that is to be used 40549b94acceSBarry Smith to test for convergence of the nonlinear iterative solution. 40559b94acceSBarry Smith 40563f9fe445SBarry Smith Logically Collective on SNES 4057fee21e36SBarry Smith 4058c7afd0dbSLois Curfman McInnes Input Parameters: 4059c7afd0dbSLois Curfman McInnes + snes - the SNES context 4060bf388a1fSBarry Smith . SNESConvergenceTestFunction - routine to test for convergence 40610298fd71SBarry Smith . cctx - [optional] context for private data for the convergence routine (may be NULL) 4062cf90aa19SBarry Smith - destroy - [optional] destructor for the context (may be NULL; PETSC_NULL_FUNCTION in Fortran) 40639b94acceSBarry Smith 406436851e7fSLois Curfman McInnes Level: advanced 406536851e7fSLois Curfman McInnes 4066e2a6519dSDmitry Karpeev .seealso: SNESConvergedDefault(), SNESConvergedSkip(), SNESConvergenceTestFunction 40679b94acceSBarry Smith @*/ 4068bf388a1fSBarry Smith PetscErrorCode SNESSetConvergenceTest(SNES snes,PetscErrorCode (*SNESConvergenceTestFunction)(SNES,PetscInt,PetscReal,PetscReal,PetscReal,SNESConvergedReason*,void*),void *cctx,PetscErrorCode (*destroy)(void*)) 40699b94acceSBarry Smith { 40707f7931b9SBarry Smith PetscErrorCode ierr; 40717f7931b9SBarry Smith 40723a40ed3dSBarry Smith PetscFunctionBegin; 40730700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 4074e2a6519dSDmitry Karpeev if (!SNESConvergenceTestFunction) SNESConvergenceTestFunction = SNESConvergedSkip; 40757f7931b9SBarry Smith if (snes->ops->convergeddestroy) { 40767f7931b9SBarry Smith ierr = (*snes->ops->convergeddestroy)(snes->cnvP);CHKERRQ(ierr); 40777f7931b9SBarry Smith } 4078bf388a1fSBarry Smith snes->ops->converged = SNESConvergenceTestFunction; 40797f7931b9SBarry Smith snes->ops->convergeddestroy = destroy; 408085385478SLisandro Dalcin snes->cnvP = cctx; 40813a40ed3dSBarry Smith PetscFunctionReturn(0); 40829b94acceSBarry Smith } 40839b94acceSBarry Smith 408452baeb72SSatish Balay /*@ 4085184914b5SBarry Smith SNESGetConvergedReason - Gets the reason the SNES iteration was stopped. 4086184914b5SBarry Smith 4087184914b5SBarry Smith Not Collective 4088184914b5SBarry Smith 4089184914b5SBarry Smith Input Parameter: 4090184914b5SBarry Smith . snes - the SNES context 4091184914b5SBarry Smith 4092184914b5SBarry Smith Output Parameter: 40934d0a8057SBarry Smith . reason - negative value indicates diverged, positive value converged, see SNESConvergedReason or the 4094184914b5SBarry Smith manual pages for the individual convergence tests for complete lists 4095184914b5SBarry Smith 40966a4d7782SBarry Smith Options Database: 40976a4d7782SBarry Smith . -snes_converged_reason - prints the reason to standard out 40986a4d7782SBarry Smith 4099184914b5SBarry Smith Level: intermediate 4100184914b5SBarry Smith 410195452b02SPatrick Sanan Notes: 410295452b02SPatrick Sanan Should only be called after the call the SNESSolve() is complete, if it is called earlier it returns the value SNES__CONVERGED_ITERATING. 4103184914b5SBarry Smith 410433866048SMatthew G. Knepley .seealso: SNESSetConvergenceTest(), SNESSetConvergedReason(), SNESConvergedReason 4105184914b5SBarry Smith @*/ 41067087cfbeSBarry Smith PetscErrorCode SNESGetConvergedReason(SNES snes,SNESConvergedReason *reason) 4107184914b5SBarry Smith { 4108184914b5SBarry Smith PetscFunctionBegin; 41090700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 41104482741eSBarry Smith PetscValidPointer(reason,2); 4111184914b5SBarry Smith *reason = snes->reason; 4112184914b5SBarry Smith PetscFunctionReturn(0); 4113184914b5SBarry Smith } 4114184914b5SBarry Smith 4115c4421ceaSFande Kong /*@C 4116c4421ceaSFande Kong SNESGetConvergedReasonString - Return a human readable string for snes converged reason 4117c4421ceaSFande Kong 4118c4421ceaSFande Kong Not Collective 4119c4421ceaSFande Kong 4120c4421ceaSFande Kong Input Parameter: 4121c4421ceaSFande Kong . snes - the SNES context 4122c4421ceaSFande Kong 4123c4421ceaSFande Kong Output Parameter: 4124c4421ceaSFande Kong . strreason - a human readable string that describes SNES converged reason 4125c4421ceaSFande Kong 412699c90e12SSatish Balay Level: beginner 4127c4421ceaSFande Kong 4128c4421ceaSFande Kong .seealso: SNESGetConvergedReason() 4129c4421ceaSFande Kong @*/ 4130c4421ceaSFande Kong PetscErrorCode SNESGetConvergedReasonString(SNES snes, const char** strreason) 4131c4421ceaSFande Kong { 4132c4421ceaSFande Kong PetscFunctionBegin; 4133c4421ceaSFande Kong PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 4134c4421ceaSFande Kong PetscValidCharPointer(strreason,2); 4135c4421ceaSFande Kong *strreason = SNESConvergedReasons[snes->reason]; 4136c4421ceaSFande Kong PetscFunctionReturn(0); 4137c4421ceaSFande Kong } 4138c4421ceaSFande Kong 413933866048SMatthew G. Knepley /*@ 414033866048SMatthew G. Knepley SNESSetConvergedReason - Sets the reason the SNES iteration was stopped. 414133866048SMatthew G. Knepley 414233866048SMatthew G. Knepley Not Collective 414333866048SMatthew G. Knepley 414433866048SMatthew G. Knepley Input Parameters: 414533866048SMatthew G. Knepley + snes - the SNES context 414633866048SMatthew G. Knepley - reason - negative value indicates diverged, positive value converged, see SNESConvergedReason or the 414733866048SMatthew G. Knepley manual pages for the individual convergence tests for complete lists 414833866048SMatthew G. Knepley 414933866048SMatthew G. Knepley Level: intermediate 415033866048SMatthew G. Knepley 415133866048SMatthew G. Knepley .seealso: SNESGetConvergedReason(), SNESSetConvergenceTest(), SNESConvergedReason 415233866048SMatthew G. Knepley @*/ 415333866048SMatthew G. Knepley PetscErrorCode SNESSetConvergedReason(SNES snes,SNESConvergedReason reason) 415433866048SMatthew G. Knepley { 415533866048SMatthew G. Knepley PetscFunctionBegin; 415633866048SMatthew G. Knepley PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 415733866048SMatthew G. Knepley snes->reason = reason; 415833866048SMatthew G. Knepley PetscFunctionReturn(0); 415933866048SMatthew G. Knepley } 416033866048SMatthew G. Knepley 4161c9005455SLois Curfman McInnes /*@ 4162c9005455SLois Curfman McInnes SNESSetConvergenceHistory - Sets the array used to hold the convergence history. 4163c9005455SLois Curfman McInnes 41643f9fe445SBarry Smith Logically Collective on SNES 4165fee21e36SBarry Smith 4166c7afd0dbSLois Curfman McInnes Input Parameters: 4167c7afd0dbSLois Curfman McInnes + snes - iterative context obtained from SNESCreate() 41688c7482ecSBarry Smith . a - array to hold history, this array will contain the function norms computed at each step 4169cd5578b5SBarry Smith . its - integer array holds the number of linear iterations for each solve. 4170758f92a0SBarry Smith . na - size of a and its 417164731454SLois Curfman McInnes - reset - PETSC_TRUE indicates each new nonlinear solve resets the history counter to zero, 4172758f92a0SBarry Smith else it continues storing new values for new nonlinear solves after the old ones 4173c7afd0dbSLois Curfman McInnes 4174308dcc3eSBarry Smith Notes: 41750298fd71SBarry Smith If 'a' and 'its' are NULL then space is allocated for the history. If 'na' PETSC_DECIDE or PETSC_DEFAULT then a 4176308dcc3eSBarry Smith default array of length 10000 is allocated. 4177308dcc3eSBarry Smith 4178c9005455SLois Curfman McInnes This routine is useful, e.g., when running a code for purposes 4179c9005455SLois Curfman McInnes of accurate performance monitoring, when no I/O should be done 4180c9005455SLois Curfman McInnes during the section of code that is being timed. 4181c9005455SLois Curfman McInnes 418236851e7fSLois Curfman McInnes Level: intermediate 418336851e7fSLois Curfman McInnes 418408405cd6SLois Curfman McInnes .seealso: SNESGetConvergenceHistory() 4185758f92a0SBarry Smith 4186c9005455SLois Curfman McInnes @*/ 41877087cfbeSBarry Smith PetscErrorCode SNESSetConvergenceHistory(SNES snes,PetscReal a[],PetscInt its[],PetscInt na,PetscBool reset) 4188c9005455SLois Curfman McInnes { 4189308dcc3eSBarry Smith PetscErrorCode ierr; 4190308dcc3eSBarry Smith 41913a40ed3dSBarry Smith PetscFunctionBegin; 41920700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 41937a1ec6d4SBarry Smith if (a) PetscValidScalarPointer(a,2); 4194a562a398SLisandro Dalcin if (its) PetscValidIntPointer(its,3); 41957a1ec6d4SBarry Smith if (!a) { 4196308dcc3eSBarry Smith if (na == PETSC_DECIDE || na == PETSC_DEFAULT) na = 1000; 4197071fcb05SBarry Smith ierr = PetscCalloc2(na,&a,na,&its);CHKERRQ(ierr); 4198071fcb05SBarry Smith snes->conv_hist_alloc = PETSC_TRUE; 4199308dcc3eSBarry Smith } 4200c9005455SLois Curfman McInnes snes->conv_hist = a; 4201758f92a0SBarry Smith snes->conv_hist_its = its; 4202758f92a0SBarry Smith snes->conv_hist_max = na; 4203a12bc48fSLisandro Dalcin snes->conv_hist_len = 0; 4204758f92a0SBarry Smith snes->conv_hist_reset = reset; 4205758f92a0SBarry Smith PetscFunctionReturn(0); 4206758f92a0SBarry Smith } 4207758f92a0SBarry Smith 4208308dcc3eSBarry Smith #if defined(PETSC_HAVE_MATLAB_ENGINE) 4209c6db04a5SJed Brown #include <engine.h> /* MATLAB include file */ 4210c6db04a5SJed Brown #include <mex.h> /* MATLAB include file */ 421199e0435eSBarry Smith 42128cc058d9SJed Brown PETSC_EXTERN mxArray *SNESGetConvergenceHistoryMatlab(SNES snes) 4213308dcc3eSBarry Smith { 4214308dcc3eSBarry Smith mxArray *mat; 4215308dcc3eSBarry Smith PetscInt i; 4216308dcc3eSBarry Smith PetscReal *ar; 4217308dcc3eSBarry Smith 4218308dcc3eSBarry Smith PetscFunctionBegin; 4219308dcc3eSBarry Smith mat = mxCreateDoubleMatrix(snes->conv_hist_len,1,mxREAL); 4220308dcc3eSBarry Smith ar = (PetscReal*) mxGetData(mat); 4221f5af7f23SKarl Rupp for (i=0; i<snes->conv_hist_len; i++) ar[i] = snes->conv_hist[i]; 4222308dcc3eSBarry Smith PetscFunctionReturn(mat); 4223308dcc3eSBarry Smith } 4224308dcc3eSBarry Smith #endif 4225308dcc3eSBarry Smith 42260c4c9dddSBarry Smith /*@C 4227758f92a0SBarry Smith SNESGetConvergenceHistory - Gets the array used to hold the convergence history. 4228758f92a0SBarry Smith 42293f9fe445SBarry Smith Not Collective 4230758f92a0SBarry Smith 4231758f92a0SBarry Smith Input Parameter: 4232758f92a0SBarry Smith . snes - iterative context obtained from SNESCreate() 4233758f92a0SBarry Smith 4234758f92a0SBarry Smith Output Parameters: 4235a2b725a8SWilliam Gropp + a - array to hold history 4236758f92a0SBarry Smith . its - integer array holds the number of linear iterations (or 4237758f92a0SBarry Smith negative if not converged) for each solve. 4238758f92a0SBarry Smith - na - size of a and its 4239758f92a0SBarry Smith 4240758f92a0SBarry Smith Notes: 4241758f92a0SBarry Smith The calling sequence for this routine in Fortran is 4242758f92a0SBarry Smith $ call SNESGetConvergenceHistory(SNES snes, integer na, integer ierr) 4243758f92a0SBarry Smith 4244758f92a0SBarry Smith This routine is useful, e.g., when running a code for purposes 4245758f92a0SBarry Smith of accurate performance monitoring, when no I/O should be done 4246758f92a0SBarry Smith during the section of code that is being timed. 4247758f92a0SBarry Smith 4248758f92a0SBarry Smith Level: intermediate 4249758f92a0SBarry Smith 42505ea7661aSPierre Jolivet .seealso: SNESSetConvergenceHistory() 4251758f92a0SBarry Smith 4252758f92a0SBarry Smith @*/ 42537087cfbeSBarry Smith PetscErrorCode SNESGetConvergenceHistory(SNES snes,PetscReal *a[],PetscInt *its[],PetscInt *na) 4254758f92a0SBarry Smith { 4255758f92a0SBarry Smith PetscFunctionBegin; 42560700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 4257758f92a0SBarry Smith if (a) *a = snes->conv_hist; 4258758f92a0SBarry Smith if (its) *its = snes->conv_hist_its; 4259758f92a0SBarry Smith if (na) *na = snes->conv_hist_len; 42603a40ed3dSBarry Smith PetscFunctionReturn(0); 4261c9005455SLois Curfman McInnes } 4262c9005455SLois Curfman McInnes 4263ac226902SBarry Smith /*@C 426476b2cf59SMatthew Knepley SNESSetUpdate - Sets the general-purpose update function called 4265eec8f646SJed Brown at the beginning of every iteration of the nonlinear solve. Specifically 42667e4bb74cSBarry Smith it is called just before the Jacobian is "evaluated". 426776b2cf59SMatthew Knepley 42683f9fe445SBarry Smith Logically Collective on SNES 426976b2cf59SMatthew Knepley 427076b2cf59SMatthew Knepley Input Parameters: 4271a2b725a8SWilliam Gropp + snes - The nonlinear solver context 4272a2b725a8SWilliam Gropp - func - The function 427376b2cf59SMatthew Knepley 427476b2cf59SMatthew Knepley Calling sequence of func: 4275a2b725a8SWilliam Gropp $ func (SNES snes, PetscInt step); 427676b2cf59SMatthew Knepley 427776b2cf59SMatthew Knepley . step - The current step of the iteration 427876b2cf59SMatthew Knepley 4279fe97e370SBarry Smith Level: advanced 4280fe97e370SBarry Smith 4281fe97e370SBarry 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() 4282fe97e370SBarry Smith This is not used by most users. 428376b2cf59SMatthew Knepley 42848d359177SBarry Smith .seealso SNESSetJacobian(), SNESSolve() 428576b2cf59SMatthew Knepley @*/ 42867087cfbeSBarry Smith PetscErrorCode SNESSetUpdate(SNES snes, PetscErrorCode (*func)(SNES, PetscInt)) 428776b2cf59SMatthew Knepley { 428876b2cf59SMatthew Knepley PetscFunctionBegin; 42890700a824SBarry Smith PetscValidHeaderSpecific(snes, SNES_CLASSID,1); 4290e7788613SBarry Smith snes->ops->update = func; 429176b2cf59SMatthew Knepley PetscFunctionReturn(0); 429276b2cf59SMatthew Knepley } 429376b2cf59SMatthew Knepley 42949b94acceSBarry Smith /* 42959b94acceSBarry Smith SNESScaleStep_Private - Scales a step so that its length is less than the 42969b94acceSBarry Smith positive parameter delta. 42979b94acceSBarry Smith 42989b94acceSBarry Smith Input Parameters: 4299c7afd0dbSLois Curfman McInnes + snes - the SNES context 43009b94acceSBarry Smith . y - approximate solution of linear system 43019b94acceSBarry Smith . fnorm - 2-norm of current function 4302c7afd0dbSLois Curfman McInnes - delta - trust region size 43039b94acceSBarry Smith 43049b94acceSBarry Smith Output Parameters: 4305c7afd0dbSLois Curfman McInnes + gpnorm - predicted function norm at the new point, assuming local 43069b94acceSBarry Smith linearization. The value is zero if the step lies within the trust 43079b94acceSBarry Smith region, and exceeds zero otherwise. 4308c7afd0dbSLois Curfman McInnes - ynorm - 2-norm of the step 43099b94acceSBarry Smith 43109b94acceSBarry Smith Note: 431104d7464bSBarry Smith For non-trust region methods such as SNESNEWTONLS, the parameter delta 43129b94acceSBarry Smith is set to be the maximum allowable step size. 43139b94acceSBarry Smith 43149b94acceSBarry Smith */ 4315dfbe8321SBarry Smith PetscErrorCode SNESScaleStep_Private(SNES snes,Vec y,PetscReal *fnorm,PetscReal *delta,PetscReal *gpnorm,PetscReal *ynorm) 43169b94acceSBarry Smith { 4317064f8208SBarry Smith PetscReal nrm; 4318ea709b57SSatish Balay PetscScalar cnorm; 4319dfbe8321SBarry Smith PetscErrorCode ierr; 43203a40ed3dSBarry Smith 43213a40ed3dSBarry Smith PetscFunctionBegin; 43220700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 43230700a824SBarry Smith PetscValidHeaderSpecific(y,VEC_CLASSID,2); 4324c9780b6fSBarry Smith PetscCheckSameComm(snes,1,y,2); 4325184914b5SBarry Smith 4326064f8208SBarry Smith ierr = VecNorm(y,NORM_2,&nrm);CHKERRQ(ierr); 4327064f8208SBarry Smith if (nrm > *delta) { 4328064f8208SBarry Smith nrm = *delta/nrm; 4329064f8208SBarry Smith *gpnorm = (1.0 - nrm)*(*fnorm); 4330064f8208SBarry Smith cnorm = nrm; 43312dcb1b2aSMatthew Knepley ierr = VecScale(y,cnorm);CHKERRQ(ierr); 43329b94acceSBarry Smith *ynorm = *delta; 43339b94acceSBarry Smith } else { 43349b94acceSBarry Smith *gpnorm = 0.0; 4335064f8208SBarry Smith *ynorm = nrm; 43369b94acceSBarry Smith } 43373a40ed3dSBarry Smith PetscFunctionReturn(0); 43389b94acceSBarry Smith } 43399b94acceSBarry Smith 434091f3e32bSBarry Smith /*@C 434119a666eeSBarry Smith SNESConvergedReasonView - Displays the reason a SNES solve converged or diverged to a viewer 43422a359c20SBarry Smith 43432a359c20SBarry Smith Collective on SNES 43442a359c20SBarry Smith 43452a359c20SBarry Smith Parameter: 43462a359c20SBarry Smith + snes - iterative context obtained from SNESCreate() 43472a359c20SBarry Smith - viewer - the viewer to display the reason 43482a359c20SBarry Smith 43492a359c20SBarry Smith 43502a359c20SBarry Smith Options Database Keys: 4351ee300463SSatish Balay + -snes_converged_reason - print reason for converged or diverged, also prints number of iterations 4352ee300463SSatish Balay - -snes_converged_reason ::failed - only print reason and number of iterations when diverged 4353eafd5ff0SAlex Lindsay 435419a666eeSBarry Smith Notes: 435519a666eeSBarry Smith To change the format of the output call PetscViewerPushFormat(viewer,format) before this call. Use PETSC_VIEWER_DEFAULT for the default, 435619a666eeSBarry Smith use PETSC_VIEWER_FAILED to only display a reason if it fails. 43572a359c20SBarry Smith 43582a359c20SBarry Smith Level: beginner 43592a359c20SBarry Smith 436019a666eeSBarry Smith .seealso: SNESCreate(), SNESSetUp(), SNESDestroy(), SNESSetTolerances(), SNESConvergedDefault(), SNESGetConvergedReason(), SNESConvergedReasonViewFromOptions(), 436119a666eeSBarry Smith PetscViewerPushFormat(), PetscViewerPopFormat() 43622a359c20SBarry Smith 43632a359c20SBarry Smith @*/ 436419a666eeSBarry Smith PetscErrorCode SNESConvergedReasonView(SNES snes,PetscViewer viewer) 43652a359c20SBarry Smith { 436675cca76cSMatthew G. Knepley PetscViewerFormat format; 43672a359c20SBarry Smith PetscBool isAscii; 436875cca76cSMatthew G. Knepley PetscErrorCode ierr; 43692a359c20SBarry Smith 43702a359c20SBarry Smith PetscFunctionBegin; 437119a666eeSBarry Smith if (!viewer) viewer = PETSC_VIEWER_STDOUT_(PetscObjectComm((PetscObject)snes)); 43722a359c20SBarry Smith ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERASCII,&isAscii);CHKERRQ(ierr); 43732a359c20SBarry Smith if (isAscii) { 437475cca76cSMatthew G. Knepley ierr = PetscViewerGetFormat(viewer, &format);CHKERRQ(ierr); 43752a359c20SBarry Smith ierr = PetscViewerASCIIAddTab(viewer,((PetscObject)snes)->tablevel);CHKERRQ(ierr); 437675cca76cSMatthew G. Knepley if (format == PETSC_VIEWER_ASCII_INFO_DETAIL) { 437775cca76cSMatthew G. Knepley DM dm; 437875cca76cSMatthew G. Knepley Vec u; 437975cca76cSMatthew G. Knepley PetscDS prob; 438075cca76cSMatthew G. Knepley PetscInt Nf, f; 438195cbbfd3SMatthew G. Knepley PetscErrorCode (**exactSol)(PetscInt, PetscReal, const PetscReal[], PetscInt, PetscScalar[], void *); 438295cbbfd3SMatthew G. Knepley void **exactCtx; 438375cca76cSMatthew G. Knepley PetscReal error; 438475cca76cSMatthew G. Knepley 438575cca76cSMatthew G. Knepley ierr = SNESGetDM(snes, &dm);CHKERRQ(ierr); 438675cca76cSMatthew G. Knepley ierr = SNESGetSolution(snes, &u);CHKERRQ(ierr); 438775cca76cSMatthew G. Knepley ierr = DMGetDS(dm, &prob);CHKERRQ(ierr); 438875cca76cSMatthew G. Knepley ierr = PetscDSGetNumFields(prob, &Nf);CHKERRQ(ierr); 438995cbbfd3SMatthew G. Knepley ierr = PetscMalloc2(Nf, &exactSol, Nf, &exactCtx);CHKERRQ(ierr); 439095cbbfd3SMatthew G. Knepley for (f = 0; f < Nf; ++f) {ierr = PetscDSGetExactSolution(prob, f, &exactSol[f], &exactCtx[f]);CHKERRQ(ierr);} 439195cbbfd3SMatthew G. Knepley ierr = DMComputeL2Diff(dm, 0.0, exactSol, exactCtx, u, &error);CHKERRQ(ierr); 439295cbbfd3SMatthew G. Knepley ierr = PetscFree2(exactSol, exactCtx);CHKERRQ(ierr); 439375cca76cSMatthew G. Knepley if (error < 1.0e-11) {ierr = PetscViewerASCIIPrintf(viewer, "L_2 Error: < 1.0e-11\n");CHKERRQ(ierr);} 439475cca76cSMatthew G. Knepley else {ierr = PetscViewerASCIIPrintf(viewer, "L_2 Error: %g\n", error);CHKERRQ(ierr);} 439575cca76cSMatthew G. Knepley } 4396eafd5ff0SAlex Lindsay if (snes->reason > 0 && format != PETSC_VIEWER_FAILED) { 43972a359c20SBarry Smith if (((PetscObject) snes)->prefix) { 43982a359c20SBarry Smith ierr = PetscViewerASCIIPrintf(viewer,"Nonlinear %s solve converged due to %s iterations %D\n",((PetscObject) snes)->prefix,SNESConvergedReasons[snes->reason],snes->iter);CHKERRQ(ierr); 43992a359c20SBarry Smith } else { 44002a359c20SBarry Smith ierr = PetscViewerASCIIPrintf(viewer,"Nonlinear solve converged due to %s iterations %D\n",SNESConvergedReasons[snes->reason],snes->iter);CHKERRQ(ierr); 44012a359c20SBarry Smith } 4402eafd5ff0SAlex Lindsay } else if (snes->reason <= 0) { 44032a359c20SBarry Smith if (((PetscObject) snes)->prefix) { 44042a359c20SBarry 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); 44052a359c20SBarry Smith } else { 44062a359c20SBarry Smith ierr = PetscViewerASCIIPrintf(viewer,"Nonlinear solve did not converge due to %s iterations %D\n",SNESConvergedReasons[snes->reason],snes->iter);CHKERRQ(ierr); 44072a359c20SBarry Smith } 44082a359c20SBarry Smith } 44092a359c20SBarry Smith ierr = PetscViewerASCIISubtractTab(viewer,((PetscObject)snes)->tablevel);CHKERRQ(ierr); 44102a359c20SBarry Smith } 44112a359c20SBarry Smith PetscFunctionReturn(0); 44122a359c20SBarry Smith } 44132a359c20SBarry Smith 4414c4421ceaSFande Kong /*@C 4415c4421ceaSFande Kong SNESConvergedReasonViewSet - Sets an ADDITIONAL function that is to be used at the 4416c4421ceaSFande Kong end of the nonlinear solver to display the conver reason of the nonlinear solver. 4417c4421ceaSFande Kong 4418c4421ceaSFande Kong Logically Collective on SNES 4419c4421ceaSFande Kong 4420c4421ceaSFande Kong Input Parameters: 4421c4421ceaSFande Kong + snes - the SNES context 4422c4421ceaSFande Kong . f - the snes converged reason view function 4423c4421ceaSFande Kong . vctx - [optional] user-defined context for private data for the 4424c4421ceaSFande Kong snes converged reason view routine (use NULL if no context is desired) 4425c4421ceaSFande Kong - reasonviewdestroy - [optional] routine that frees reasonview context 4426c4421ceaSFande Kong (may be NULL) 4427c4421ceaSFande Kong 4428c4421ceaSFande Kong Options Database Keys: 4429c4421ceaSFande Kong + -snes_converged_reason - sets a default SNESConvergedReasonView() 4430c4421ceaSFande Kong - -snes_converged_reason_view_cancel - cancels all converged reason viewers that have 4431c4421ceaSFande Kong been hardwired into a code by 4432c4421ceaSFande Kong calls to SNESConvergedReasonViewSet(), but 4433c4421ceaSFande Kong does not cancel those set via 4434c4421ceaSFande Kong the options database. 4435c4421ceaSFande Kong 4436c4421ceaSFande Kong Notes: 4437c4421ceaSFande Kong Several different converged reason view routines may be set by calling 4438c4421ceaSFande Kong SNESConvergedReasonViewSet() multiple times; all will be called in the 4439c4421ceaSFande Kong order in which they were set. 4440c4421ceaSFande Kong 4441c4421ceaSFande Kong Level: intermediate 4442c4421ceaSFande Kong 4443c4421ceaSFande Kong .seealso: SNESConvergedReasonView(), SNESConvergedReasonViewCancel() 4444c4421ceaSFande Kong @*/ 4445c4421ceaSFande Kong PetscErrorCode SNESConvergedReasonViewSet(SNES snes,PetscErrorCode (*f)(SNES,void*),void *vctx,PetscErrorCode (*reasonviewdestroy)(void**)) 4446c4421ceaSFande Kong { 4447c4421ceaSFande Kong PetscInt i; 4448c4421ceaSFande Kong PetscErrorCode ierr; 4449c4421ceaSFande Kong PetscBool identical; 4450c4421ceaSFande Kong 4451c4421ceaSFande Kong PetscFunctionBegin; 4452c4421ceaSFande Kong PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 4453c4421ceaSFande Kong for (i=0; i<snes->numberreasonviews;i++) { 4454c4421ceaSFande Kong ierr = PetscMonitorCompare((PetscErrorCode (*)(void))f,vctx,reasonviewdestroy,(PetscErrorCode (*)(void))snes->reasonview[i],snes->reasonviewcontext[i],snes->reasonviewdestroy[i],&identical);CHKERRQ(ierr); 4455c4421ceaSFande Kong if (identical) PetscFunctionReturn(0); 4456c4421ceaSFande Kong } 4457c4421ceaSFande Kong if (snes->numberreasonviews >= MAXSNESREASONVIEWS) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Too many SNES reasonview set"); 4458c4421ceaSFande Kong snes->reasonview[snes->numberreasonviews] = f; 4459c4421ceaSFande Kong snes->reasonviewdestroy[snes->numberreasonviews] = reasonviewdestroy; 4460c4421ceaSFande Kong snes->reasonviewcontext[snes->numberreasonviews++] = (void*)vctx; 4461c4421ceaSFande Kong PetscFunctionReturn(0); 4462c4421ceaSFande Kong } 4463c4421ceaSFande Kong 446491f3e32bSBarry Smith /*@ 446519a666eeSBarry Smith SNESConvergedReasonViewFromOptions - Processes command line options to determine if/how a SNESReason is to be viewed. 4466c4421ceaSFande Kong All the user-provided convergedReasonView routines will be involved as well, if they exist. 44672a359c20SBarry Smith 44682a359c20SBarry Smith Collective on SNES 44692a359c20SBarry Smith 44702a359c20SBarry Smith Input Parameters: 44712a359c20SBarry Smith . snes - the SNES object 44722a359c20SBarry Smith 44732a359c20SBarry Smith Level: intermediate 44742a359c20SBarry Smith 447519a666eeSBarry Smith .seealso: SNESCreate(), SNESSetUp(), SNESDestroy(), SNESSetTolerances(), SNESConvergedDefault(), SNESGetConvergedReason(), SNESConvergedReasonView() 447619a666eeSBarry Smith 44772a359c20SBarry Smith @*/ 447819a666eeSBarry Smith PetscErrorCode SNESConvergedReasonViewFromOptions(SNES snes) 44792a359c20SBarry Smith { 44802a359c20SBarry Smith PetscErrorCode ierr; 44812a359c20SBarry Smith PetscViewer viewer; 44822a359c20SBarry Smith PetscBool flg; 44832a359c20SBarry Smith static PetscBool incall = PETSC_FALSE; 44842a359c20SBarry Smith PetscViewerFormat format; 4485c4421ceaSFande Kong PetscInt i; 44862a359c20SBarry Smith 44872a359c20SBarry Smith PetscFunctionBegin; 44882a359c20SBarry Smith if (incall) PetscFunctionReturn(0); 44892a359c20SBarry Smith incall = PETSC_TRUE; 4490c4421ceaSFande Kong 4491c4421ceaSFande Kong /* All user-provided viewers are called first, if they exist. */ 4492c4421ceaSFande Kong for (i=0; i<snes->numberreasonviews; i++) { 4493c4421ceaSFande Kong ierr = (*snes->reasonview[i])(snes,snes->reasonviewcontext[i]);CHKERRQ(ierr); 4494c4421ceaSFande Kong } 4495c4421ceaSFande Kong 4496c4421ceaSFande Kong /* Call PETSc default routine if users ask for it */ 449716413a6aSBarry Smith ierr = PetscOptionsGetViewer(PetscObjectComm((PetscObject)snes),((PetscObject)snes)->options,((PetscObject)snes)->prefix,"-snes_converged_reason",&viewer,&format,&flg);CHKERRQ(ierr); 44982a359c20SBarry Smith if (flg) { 44992a359c20SBarry Smith ierr = PetscViewerPushFormat(viewer,format);CHKERRQ(ierr); 450019a666eeSBarry Smith ierr = SNESConvergedReasonView(snes,viewer);CHKERRQ(ierr); 45012a359c20SBarry Smith ierr = PetscViewerPopFormat(viewer);CHKERRQ(ierr); 45022a359c20SBarry Smith ierr = PetscViewerDestroy(&viewer);CHKERRQ(ierr); 45032a359c20SBarry Smith } 45042a359c20SBarry Smith incall = PETSC_FALSE; 45052a359c20SBarry Smith PetscFunctionReturn(0); 45062a359c20SBarry Smith } 45072a359c20SBarry Smith 4508487a658cSBarry Smith /*@ 4509f69a0ea3SMatthew Knepley SNESSolve - Solves a nonlinear system F(x) = b. 4510f69a0ea3SMatthew Knepley Call SNESSolve() after calling SNESCreate() and optional routines of the form SNESSetXXX(). 45119b94acceSBarry Smith 4512c7afd0dbSLois Curfman McInnes Collective on SNES 4513c7afd0dbSLois Curfman McInnes 4514b2002411SLois Curfman McInnes Input Parameters: 4515c7afd0dbSLois Curfman McInnes + snes - the SNES context 45160298fd71SBarry Smith . b - the constant part of the equation F(x) = b, or NULL to use zero. 451785385478SLisandro Dalcin - x - the solution vector. 45189b94acceSBarry Smith 4519b2002411SLois Curfman McInnes Notes: 45208ddd3da0SLois Curfman McInnes The user should initialize the vector,x, with the initial guess 45218ddd3da0SLois Curfman McInnes for the nonlinear solve prior to calling SNESSolve. In particular, 45228ddd3da0SLois Curfman McInnes to employ an initial guess of zero, the user should explicitly set 45238ddd3da0SLois Curfman McInnes this vector to zero by calling VecSet(). 45248ddd3da0SLois Curfman McInnes 452536851e7fSLois Curfman McInnes Level: beginner 452636851e7fSLois Curfman McInnes 4527c0df2a02SJed Brown .seealso: SNESCreate(), SNESDestroy(), SNESSetFunction(), SNESSetJacobian(), SNESSetGridSequence(), SNESGetSolution() 45289b94acceSBarry Smith @*/ 45297087cfbeSBarry Smith PetscErrorCode SNESSolve(SNES snes,Vec b,Vec x) 45309b94acceSBarry Smith { 4531dfbe8321SBarry Smith PetscErrorCode ierr; 4532ace3abfcSBarry Smith PetscBool flg; 4533efd51863SBarry Smith PetscInt grid; 45340298fd71SBarry Smith Vec xcreated = NULL; 4535caa4e7f2SJed Brown DM dm; 4536052efed2SBarry Smith 45373a40ed3dSBarry Smith PetscFunctionBegin; 45380700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 4539a69afd8bSBarry Smith if (x) PetscValidHeaderSpecific(x,VEC_CLASSID,3); 4540a69afd8bSBarry Smith if (x) PetscCheckSameComm(snes,1,x,3); 45410700a824SBarry Smith if (b) PetscValidHeaderSpecific(b,VEC_CLASSID,2); 454285385478SLisandro Dalcin if (b) PetscCheckSameComm(snes,1,b,2); 454385385478SLisandro Dalcin 454434b4d3a8SMatthew G. Knepley /* High level operations using the nonlinear solver */ 454506fc46c8SMatthew G. Knepley { 454606fc46c8SMatthew G. Knepley PetscViewer viewer; 454706fc46c8SMatthew G. Knepley PetscViewerFormat format; 45487c88af5aSMatthew G. Knepley PetscInt num; 454906fc46c8SMatthew G. Knepley PetscBool flg; 455006fc46c8SMatthew G. Knepley static PetscBool incall = PETSC_FALSE; 455106fc46c8SMatthew G. Knepley 455206fc46c8SMatthew G. Knepley if (!incall) { 455334b4d3a8SMatthew G. Knepley /* Estimate the convergence rate of the discretization */ 455416413a6aSBarry Smith ierr = PetscOptionsGetViewer(PetscObjectComm((PetscObject) snes),((PetscObject)snes)->options, ((PetscObject) snes)->prefix, "-snes_convergence_estimate", &viewer, &format, &flg);CHKERRQ(ierr); 455506fc46c8SMatthew G. Knepley if (flg) { 455606fc46c8SMatthew G. Knepley PetscConvEst conv; 455746079b62SMatthew G. Knepley DM dm; 455846079b62SMatthew G. Knepley PetscReal *alpha; /* Convergence rate of the solution error for each field in the L_2 norm */ 455946079b62SMatthew G. Knepley PetscInt Nf; 456006fc46c8SMatthew G. Knepley 456106fc46c8SMatthew G. Knepley incall = PETSC_TRUE; 456246079b62SMatthew G. Knepley ierr = SNESGetDM(snes, &dm);CHKERRQ(ierr); 456346079b62SMatthew G. Knepley ierr = DMGetNumFields(dm, &Nf);CHKERRQ(ierr); 4564ca5a3622SMatthew G. Knepley ierr = PetscCalloc1(Nf, &alpha);CHKERRQ(ierr); 456506fc46c8SMatthew G. Knepley ierr = PetscConvEstCreate(PetscObjectComm((PetscObject) snes), &conv);CHKERRQ(ierr); 4566900f6b5bSMatthew G. Knepley ierr = PetscConvEstSetSolver(conv, (PetscObject) snes);CHKERRQ(ierr); 456706fc46c8SMatthew G. Knepley ierr = PetscConvEstSetFromOptions(conv);CHKERRQ(ierr); 45680955ed61SMatthew G. Knepley ierr = PetscConvEstSetUp(conv);CHKERRQ(ierr); 456946079b62SMatthew G. Knepley ierr = PetscConvEstGetConvRate(conv, alpha);CHKERRQ(ierr); 457006fc46c8SMatthew G. Knepley ierr = PetscViewerPushFormat(viewer, format);CHKERRQ(ierr); 457106fc46c8SMatthew G. Knepley ierr = PetscConvEstRateView(conv, alpha, viewer);CHKERRQ(ierr); 457206fc46c8SMatthew G. Knepley ierr = PetscViewerPopFormat(viewer);CHKERRQ(ierr); 457306fc46c8SMatthew G. Knepley ierr = PetscViewerDestroy(&viewer);CHKERRQ(ierr); 457406fc46c8SMatthew G. Knepley ierr = PetscConvEstDestroy(&conv);CHKERRQ(ierr); 457546079b62SMatthew G. Knepley ierr = PetscFree(alpha);CHKERRQ(ierr); 457606fc46c8SMatthew G. Knepley incall = PETSC_FALSE; 457706fc46c8SMatthew G. Knepley } 457834b4d3a8SMatthew G. Knepley /* Adaptively refine the initial grid */ 4579b2588ea6SMatthew G. Knepley num = 1; 4580b2588ea6SMatthew G. Knepley ierr = PetscOptionsGetInt(NULL, ((PetscObject) snes)->prefix, "-snes_adapt_initial", &num, &flg);CHKERRQ(ierr); 458134b4d3a8SMatthew G. Knepley if (flg) { 458234b4d3a8SMatthew G. Knepley DMAdaptor adaptor; 458334b4d3a8SMatthew G. Knepley 458434b4d3a8SMatthew G. Knepley incall = PETSC_TRUE; 4585ea13f565SStefano Zampini ierr = DMAdaptorCreate(PetscObjectComm((PetscObject)snes), &adaptor);CHKERRQ(ierr); 458634b4d3a8SMatthew G. Knepley ierr = DMAdaptorSetSolver(adaptor, snes);CHKERRQ(ierr); 4587b2588ea6SMatthew G. Knepley ierr = DMAdaptorSetSequenceLength(adaptor, num);CHKERRQ(ierr); 458834b4d3a8SMatthew G. Knepley ierr = DMAdaptorSetFromOptions(adaptor);CHKERRQ(ierr); 458934b4d3a8SMatthew G. Knepley ierr = DMAdaptorSetUp(adaptor);CHKERRQ(ierr); 459034b4d3a8SMatthew G. Knepley ierr = DMAdaptorAdapt(adaptor, x, DM_ADAPTATION_INITIAL, &dm, &x);CHKERRQ(ierr); 459134b4d3a8SMatthew G. Knepley ierr = DMAdaptorDestroy(&adaptor);CHKERRQ(ierr); 459234b4d3a8SMatthew G. Knepley incall = PETSC_FALSE; 459334b4d3a8SMatthew G. Knepley } 45947c88af5aSMatthew G. Knepley /* Use grid sequencing to adapt */ 45957c88af5aSMatthew G. Knepley num = 0; 45967c88af5aSMatthew G. Knepley ierr = PetscOptionsGetInt(NULL, ((PetscObject) snes)->prefix, "-snes_adapt_sequence", &num, NULL);CHKERRQ(ierr); 45977c88af5aSMatthew G. Knepley if (num) { 45987c88af5aSMatthew G. Knepley DMAdaptor adaptor; 45997c88af5aSMatthew G. Knepley 46007c88af5aSMatthew G. Knepley incall = PETSC_TRUE; 4601ea13f565SStefano Zampini ierr = DMAdaptorCreate(PetscObjectComm((PetscObject)snes), &adaptor);CHKERRQ(ierr); 46027c88af5aSMatthew G. Knepley ierr = DMAdaptorSetSolver(adaptor, snes);CHKERRQ(ierr); 46037c88af5aSMatthew G. Knepley ierr = DMAdaptorSetSequenceLength(adaptor, num);CHKERRQ(ierr); 46047c88af5aSMatthew G. Knepley ierr = DMAdaptorSetFromOptions(adaptor);CHKERRQ(ierr); 46057c88af5aSMatthew G. Knepley ierr = DMAdaptorSetUp(adaptor);CHKERRQ(ierr); 46067c88af5aSMatthew G. Knepley ierr = DMAdaptorAdapt(adaptor, x, DM_ADAPTATION_SEQUENTIAL, &dm, &x);CHKERRQ(ierr); 46077c88af5aSMatthew G. Knepley ierr = DMAdaptorDestroy(&adaptor);CHKERRQ(ierr); 46087c88af5aSMatthew G. Knepley incall = PETSC_FALSE; 46097c88af5aSMatthew G. Knepley } 461006fc46c8SMatthew G. Knepley } 461106fc46c8SMatthew G. Knepley } 4612caa4e7f2SJed Brown if (!x) { 4613caa4e7f2SJed Brown ierr = SNESGetDM(snes,&dm);CHKERRQ(ierr); 4614caa4e7f2SJed Brown ierr = DMCreateGlobalVector(dm,&xcreated);CHKERRQ(ierr); 4615a69afd8bSBarry Smith x = xcreated; 4616a69afd8bSBarry Smith } 4617ce1779c8SBarry Smith ierr = SNESViewFromOptions(snes,NULL,"-snes_view_pre");CHKERRQ(ierr); 4618f05ece33SBarry Smith 4619ce94432eSBarry Smith for (grid=0; grid<snes->gridsequence; grid++) {ierr = PetscViewerASCIIPushTab(PETSC_VIEWER_STDOUT_(PetscObjectComm((PetscObject)snes)));CHKERRQ(ierr);} 4620efd51863SBarry Smith for (grid=0; grid<snes->gridsequence+1; grid++) { 4621efd51863SBarry Smith 462285385478SLisandro Dalcin /* set solution vector */ 4623efd51863SBarry Smith if (!grid) {ierr = PetscObjectReference((PetscObject)x);CHKERRQ(ierr);} 46246bf464f9SBarry Smith ierr = VecDestroy(&snes->vec_sol);CHKERRQ(ierr); 462585385478SLisandro Dalcin snes->vec_sol = x; 4626caa4e7f2SJed Brown ierr = SNESGetDM(snes,&dm);CHKERRQ(ierr); 4627caa4e7f2SJed Brown 4628caa4e7f2SJed Brown /* set affine vector if provided */ 462985385478SLisandro Dalcin if (b) { ierr = PetscObjectReference((PetscObject)b);CHKERRQ(ierr); } 46306bf464f9SBarry Smith ierr = VecDestroy(&snes->vec_rhs);CHKERRQ(ierr); 463185385478SLisandro Dalcin snes->vec_rhs = b; 463285385478SLisandro Dalcin 4633bfdd6862SPatrick Farrell if (snes->vec_rhs && (snes->vec_func == snes->vec_rhs)) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_IDN,"Right hand side vector cannot be function vector"); 4634154060b5SMatthew G. Knepley if (snes->vec_func == snes->vec_sol) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_IDN,"Solution vector cannot be function vector"); 4635154060b5SMatthew 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"); 4636154060b5SMatthew G. Knepley if (!snes->vec_sol_update /* && snes->vec_sol */) { 4637154060b5SMatthew G. Knepley ierr = VecDuplicate(snes->vec_sol,&snes->vec_sol_update);CHKERRQ(ierr); 4638154060b5SMatthew G. Knepley ierr = PetscLogObjectParent((PetscObject)snes,(PetscObject)snes->vec_sol_update);CHKERRQ(ierr); 4639154060b5SMatthew G. Knepley } 4640154060b5SMatthew G. Knepley ierr = DMShellSetGlobalVector(dm,snes->vec_sol);CHKERRQ(ierr); 464170e92668SMatthew Knepley ierr = SNESSetUp(snes);CHKERRQ(ierr); 46423f149594SLisandro Dalcin 46437eee914bSBarry Smith if (!grid) { 46447eee914bSBarry Smith if (snes->ops->computeinitialguess) { 4645d25893d9SBarry Smith ierr = (*snes->ops->computeinitialguess)(snes,snes->vec_sol,snes->initialguessP);CHKERRQ(ierr); 4646d25893d9SBarry Smith } 4647dd568438SSatish Balay } 4648d25893d9SBarry Smith 4649abc0a331SBarry Smith if (snes->conv_hist_reset) snes->conv_hist_len = 0; 4650971e163fSPeter Brune if (snes->counters_reset) {snes->nfuncs = 0; snes->linear_its = 0; snes->numFailures = 0;} 4651d5e45103SBarry Smith 46523f149594SLisandro Dalcin ierr = PetscLogEventBegin(SNES_Solve,snes,0,0,0);CHKERRQ(ierr); 46534936397dSBarry Smith ierr = (*snes->ops->solve)(snes);CHKERRQ(ierr); 465485385478SLisandro Dalcin ierr = PetscLogEventEnd(SNES_Solve,snes,0,0,0);CHKERRQ(ierr); 465517186662SBarry Smith if (!snes->reason) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_PLIB,"Internal error, solver returned without setting converged reason"); 4656422a814eSBarry Smith snes->domainerror = PETSC_FALSE; /* clear the flag if it has been set */ 46573f149594SLisandro Dalcin 465837ec4e1aSPeter Brune if (snes->lagjac_persist) snes->jac_iter += snes->iter; 465937ec4e1aSPeter Brune if (snes->lagpre_persist) snes->pre_iter += snes->iter; 466037ec4e1aSPeter Brune 466116413a6aSBarry Smith ierr = PetscOptionsGetViewer(PetscObjectComm((PetscObject)snes),((PetscObject)snes)->options,((PetscObject)snes)->prefix,"-snes_test_local_min",NULL,NULL,&flg);CHKERRQ(ierr); 4662da9b6338SBarry Smith if (flg && !PetscPreLoadingOn) { ierr = SNESTestLocalMin(snes);CHKERRQ(ierr); } 4663c4421ceaSFande Kong /* Call converged reason views. This may involve user-provided viewers as well */ 466419a666eeSBarry Smith ierr = SNESConvergedReasonViewFromOptions(snes);CHKERRQ(ierr); 46655968eb51SBarry Smith 4666ce94432eSBarry Smith if (snes->errorifnotconverged && snes->reason < 0) SETERRQ(PetscObjectComm((PetscObject)snes),PETSC_ERR_NOT_CONVERGED,"SNESSolve has not converged"); 46679c8e83a9SBarry Smith if (snes->reason < 0) break; 4668efd51863SBarry Smith if (grid < snes->gridsequence) { 4669efd51863SBarry Smith DM fine; 4670efd51863SBarry Smith Vec xnew; 4671efd51863SBarry Smith Mat interp; 4672efd51863SBarry Smith 4673ce94432eSBarry Smith ierr = DMRefine(snes->dm,PetscObjectComm((PetscObject)snes),&fine);CHKERRQ(ierr); 4674ce94432eSBarry Smith if (!fine) SETERRQ(PetscObjectComm((PetscObject)snes),PETSC_ERR_ARG_INCOMP,"DMRefine() did not perform any refinement, cannot continue grid sequencing"); 46750298fd71SBarry Smith ierr = DMCreateInterpolation(snes->dm,fine,&interp,NULL);CHKERRQ(ierr); 4676efd51863SBarry Smith ierr = DMCreateGlobalVector(fine,&xnew);CHKERRQ(ierr); 4677efd51863SBarry Smith ierr = MatInterpolate(interp,x,xnew);CHKERRQ(ierr); 4678c833c3b5SJed Brown ierr = DMInterpolate(snes->dm,interp,fine);CHKERRQ(ierr); 4679efd51863SBarry Smith ierr = MatDestroy(&interp);CHKERRQ(ierr); 4680efd51863SBarry Smith x = xnew; 4681efd51863SBarry Smith 4682efd51863SBarry Smith ierr = SNESReset(snes);CHKERRQ(ierr); 4683efd51863SBarry Smith ierr = SNESSetDM(snes,fine);CHKERRQ(ierr); 4684352f405dSMatthew G. Knepley ierr = SNESResetFromOptions(snes);CHKERRQ(ierr); 4685efd51863SBarry Smith ierr = DMDestroy(&fine);CHKERRQ(ierr); 4686ce94432eSBarry Smith ierr = PetscViewerASCIIPopTab(PETSC_VIEWER_STDOUT_(PetscObjectComm((PetscObject)snes)));CHKERRQ(ierr); 4687efd51863SBarry Smith } 4688efd51863SBarry Smith } 4689ce1779c8SBarry Smith ierr = SNESViewFromOptions(snes,NULL,"-snes_view");CHKERRQ(ierr); 4690685405a1SBarry Smith ierr = VecViewFromOptions(snes->vec_sol,(PetscObject)snes,"-snes_view_solution");CHKERRQ(ierr); 4691c0f0dcc3SMatthew G. Knepley ierr = DMMonitor(snes->dm);CHKERRQ(ierr); 46923f7e2da0SPeter Brune 4693a69afd8bSBarry Smith ierr = VecDestroy(&xcreated);CHKERRQ(ierr); 4694e04113cfSBarry Smith ierr = PetscObjectSAWsBlock((PetscObject)snes);CHKERRQ(ierr); 46953a40ed3dSBarry Smith PetscFunctionReturn(0); 46969b94acceSBarry Smith } 46979b94acceSBarry Smith 46989b94acceSBarry Smith /* --------- Internal routines for SNES Package --------- */ 46999b94acceSBarry Smith 470082bf6240SBarry Smith /*@C 47014b0e389bSBarry Smith SNESSetType - Sets the method for the nonlinear solver. 47029b94acceSBarry Smith 4703fee21e36SBarry Smith Collective on SNES 4704fee21e36SBarry Smith 4705c7afd0dbSLois Curfman McInnes Input Parameters: 4706c7afd0dbSLois Curfman McInnes + snes - the SNES context 4707454a90a3SBarry Smith - type - a known method 4708c7afd0dbSLois Curfman McInnes 4709c7afd0dbSLois Curfman McInnes Options Database Key: 4710454a90a3SBarry Smith . -snes_type <type> - Sets the method; use -help for a list 471104d7464bSBarry Smith of available methods (for instance, newtonls or newtontr) 4712ae12b187SLois Curfman McInnes 47139b94acceSBarry Smith Notes: 4714e090d566SSatish Balay See "petsc/include/petscsnes.h" for available methods (for instance) 471504d7464bSBarry Smith + SNESNEWTONLS - Newton's method with line search 4716c7afd0dbSLois Curfman McInnes (systems of nonlinear equations) 4717a2b725a8SWilliam Gropp - SNESNEWTONTR - Newton's method with trust region 4718c7afd0dbSLois Curfman McInnes (systems of nonlinear equations) 47199b94acceSBarry Smith 4720ae12b187SLois Curfman McInnes Normally, it is best to use the SNESSetFromOptions() command and then 4721ae12b187SLois Curfman McInnes set the SNES solver type from the options database rather than by using 4722ae12b187SLois Curfman McInnes this routine. Using the options database provides the user with 4723ae12b187SLois Curfman McInnes maximum flexibility in evaluating the many nonlinear solvers. 4724ae12b187SLois Curfman McInnes The SNESSetType() routine is provided for those situations where it 4725ae12b187SLois Curfman McInnes is necessary to set the nonlinear solver independently of the command 4726ae12b187SLois Curfman McInnes line or options database. This might be the case, for example, when 4727ae12b187SLois Curfman McInnes the choice of solver changes during the execution of the program, 4728ae12b187SLois Curfman McInnes and the user's application is taking responsibility for choosing the 4729b0a32e0cSBarry Smith appropriate method. 473036851e7fSLois Curfman McInnes 473195452b02SPatrick Sanan Developer Notes: 473295452b02SPatrick Sanan SNESRegister() adds a constructor for a new SNESType to SNESList, SNESSetType() locates 47338f6c3df8SBarry Smith the constructor in that list and calls it to create the spexific object. 47348f6c3df8SBarry Smith 473536851e7fSLois Curfman McInnes Level: intermediate 4736a703fe33SLois Curfman McInnes 47378f6c3df8SBarry Smith .seealso: SNESType, SNESCreate(), SNESDestroy(), SNESGetType(), SNESSetFromOptions() 4738435da068SBarry Smith 47399b94acceSBarry Smith @*/ 474019fd82e9SBarry Smith PetscErrorCode SNESSetType(SNES snes,SNESType type) 47419b94acceSBarry Smith { 4742dfbe8321SBarry Smith PetscErrorCode ierr,(*r)(SNES); 4743ace3abfcSBarry Smith PetscBool match; 47443a40ed3dSBarry Smith 47453a40ed3dSBarry Smith PetscFunctionBegin; 47460700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 47474482741eSBarry Smith PetscValidCharPointer(type,2); 474882bf6240SBarry Smith 4749251f4c67SDmitry Karpeev ierr = PetscObjectTypeCompare((PetscObject)snes,type,&match);CHKERRQ(ierr); 47500f5bd95cSBarry Smith if (match) PetscFunctionReturn(0); 475192ff6ae8SBarry Smith 47521c9cd337SJed Brown ierr = PetscFunctionListFind(SNESList,type,&r);CHKERRQ(ierr); 4753e32f2f54SBarry Smith if (!r) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_UNKNOWN_TYPE,"Unable to find requested SNES type %s",type); 475475396ef9SLisandro Dalcin /* Destroy the previous private SNES context */ 4755b5c23020SJed Brown if (snes->ops->destroy) { 4756b5c23020SJed Brown ierr = (*(snes)->ops->destroy)(snes);CHKERRQ(ierr); 47570298fd71SBarry Smith snes->ops->destroy = NULL; 4758b5c23020SJed Brown } 475975396ef9SLisandro Dalcin /* Reinitialize function pointers in SNESOps structure */ 47609e5d0892SLisandro Dalcin snes->ops->setup = NULL; 47619e5d0892SLisandro Dalcin snes->ops->solve = NULL; 47629e5d0892SLisandro Dalcin snes->ops->view = NULL; 47639e5d0892SLisandro Dalcin snes->ops->setfromoptions = NULL; 47649e5d0892SLisandro Dalcin snes->ops->destroy = NULL; 47657fe760d5SStefano Zampini 47667fe760d5SStefano Zampini /* It may happen the user has customized the line search before calling SNESSetType */ 47677fe760d5SStefano Zampini if (((PetscObject)snes)->type_name) { 4768d8d34be6SBarry Smith ierr = SNESLineSearchDestroy(&snes->linesearch);CHKERRQ(ierr); 47697fe760d5SStefano Zampini } 47707fe760d5SStefano Zampini 477175396ef9SLisandro Dalcin /* Call the SNESCreate_XXX routine for this particular Nonlinear solver */ 477275396ef9SLisandro Dalcin snes->setupcalled = PETSC_FALSE; 4773f5af7f23SKarl Rupp 4774454a90a3SBarry Smith ierr = PetscObjectChangeTypeName((PetscObject)snes,type);CHKERRQ(ierr); 477503bfa161SLisandro Dalcin ierr = (*r)(snes);CHKERRQ(ierr); 47763a40ed3dSBarry Smith PetscFunctionReturn(0); 47779b94acceSBarry Smith } 47789b94acceSBarry Smith 47799b94acceSBarry Smith /*@C 47809a28b0a6SLois Curfman McInnes SNESGetType - Gets the SNES method type and name (as a string). 47819b94acceSBarry Smith 4782c7afd0dbSLois Curfman McInnes Not Collective 4783c7afd0dbSLois Curfman McInnes 47849b94acceSBarry Smith Input Parameter: 47854b0e389bSBarry Smith . snes - nonlinear solver context 47869b94acceSBarry Smith 47879b94acceSBarry Smith Output Parameter: 47883a7fca6bSBarry Smith . type - SNES method (a character string) 47899b94acceSBarry Smith 479036851e7fSLois Curfman McInnes Level: intermediate 479136851e7fSLois Curfman McInnes 47929b94acceSBarry Smith @*/ 479319fd82e9SBarry Smith PetscErrorCode SNESGetType(SNES snes,SNESType *type) 47949b94acceSBarry Smith { 47953a40ed3dSBarry Smith PetscFunctionBegin; 47960700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 47974482741eSBarry Smith PetscValidPointer(type,2); 47987adad957SLisandro Dalcin *type = ((PetscObject)snes)->type_name; 47993a40ed3dSBarry Smith PetscFunctionReturn(0); 48009b94acceSBarry Smith } 48019b94acceSBarry Smith 48023cd8a7caSMatthew G. Knepley /*@ 48033cd8a7caSMatthew G. Knepley SNESSetSolution - Sets the solution vector for use by the SNES routines. 48043cd8a7caSMatthew G. Knepley 4805d083f849SBarry Smith Logically Collective on SNES 48063cd8a7caSMatthew G. Knepley 48073cd8a7caSMatthew G. Knepley Input Parameters: 48083cd8a7caSMatthew G. Knepley + snes - the SNES context obtained from SNESCreate() 48093cd8a7caSMatthew G. Knepley - u - the solution vector 48103cd8a7caSMatthew G. Knepley 48113cd8a7caSMatthew G. Knepley Level: beginner 48123cd8a7caSMatthew G. Knepley 48133cd8a7caSMatthew G. Knepley @*/ 48143cd8a7caSMatthew G. Knepley PetscErrorCode SNESSetSolution(SNES snes, Vec u) 48153cd8a7caSMatthew G. Knepley { 48163cd8a7caSMatthew G. Knepley DM dm; 48173cd8a7caSMatthew G. Knepley PetscErrorCode ierr; 48183cd8a7caSMatthew G. Knepley 48193cd8a7caSMatthew G. Knepley PetscFunctionBegin; 48203cd8a7caSMatthew G. Knepley PetscValidHeaderSpecific(snes, SNES_CLASSID, 1); 48213cd8a7caSMatthew G. Knepley PetscValidHeaderSpecific(u, VEC_CLASSID, 2); 48223cd8a7caSMatthew G. Knepley ierr = PetscObjectReference((PetscObject) u);CHKERRQ(ierr); 48233cd8a7caSMatthew G. Knepley ierr = VecDestroy(&snes->vec_sol);CHKERRQ(ierr); 48243cd8a7caSMatthew G. Knepley 48253cd8a7caSMatthew G. Knepley snes->vec_sol = u; 48263cd8a7caSMatthew G. Knepley 48273cd8a7caSMatthew G. Knepley ierr = SNESGetDM(snes, &dm);CHKERRQ(ierr); 48283cd8a7caSMatthew G. Knepley ierr = DMShellSetGlobalVector(dm, u);CHKERRQ(ierr); 48293cd8a7caSMatthew G. Knepley PetscFunctionReturn(0); 48303cd8a7caSMatthew G. Knepley } 48313cd8a7caSMatthew G. Knepley 483252baeb72SSatish Balay /*@ 48339b94acceSBarry Smith SNESGetSolution - Returns the vector where the approximate solution is 4834c0df2a02SJed Brown stored. This is the fine grid solution when using SNESSetGridSequence(). 48359b94acceSBarry Smith 4836c7afd0dbSLois Curfman McInnes Not Collective, but Vec is parallel if SNES is parallel 4837c7afd0dbSLois Curfman McInnes 48389b94acceSBarry Smith Input Parameter: 48399b94acceSBarry Smith . snes - the SNES context 48409b94acceSBarry Smith 48419b94acceSBarry Smith Output Parameter: 48429b94acceSBarry Smith . x - the solution 48439b94acceSBarry Smith 484470e92668SMatthew Knepley Level: intermediate 484536851e7fSLois Curfman McInnes 484685385478SLisandro Dalcin .seealso: SNESGetSolutionUpdate(), SNESGetFunction() 48479b94acceSBarry Smith @*/ 48487087cfbeSBarry Smith PetscErrorCode SNESGetSolution(SNES snes,Vec *x) 48499b94acceSBarry Smith { 48503a40ed3dSBarry Smith PetscFunctionBegin; 48510700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 48524482741eSBarry Smith PetscValidPointer(x,2); 485385385478SLisandro Dalcin *x = snes->vec_sol; 485470e92668SMatthew Knepley PetscFunctionReturn(0); 485570e92668SMatthew Knepley } 485670e92668SMatthew Knepley 485752baeb72SSatish Balay /*@ 48589b94acceSBarry Smith SNESGetSolutionUpdate - Returns the vector where the solution update is 48599b94acceSBarry Smith stored. 48609b94acceSBarry Smith 4861c7afd0dbSLois Curfman McInnes Not Collective, but Vec is parallel if SNES is parallel 4862c7afd0dbSLois Curfman McInnes 48639b94acceSBarry Smith Input Parameter: 48649b94acceSBarry Smith . snes - the SNES context 48659b94acceSBarry Smith 48669b94acceSBarry Smith Output Parameter: 48679b94acceSBarry Smith . x - the solution update 48689b94acceSBarry Smith 486936851e7fSLois Curfman McInnes Level: advanced 487036851e7fSLois Curfman McInnes 487185385478SLisandro Dalcin .seealso: SNESGetSolution(), SNESGetFunction() 48729b94acceSBarry Smith @*/ 48737087cfbeSBarry Smith PetscErrorCode SNESGetSolutionUpdate(SNES snes,Vec *x) 48749b94acceSBarry Smith { 48753a40ed3dSBarry Smith PetscFunctionBegin; 48760700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 48774482741eSBarry Smith PetscValidPointer(x,2); 487885385478SLisandro Dalcin *x = snes->vec_sol_update; 48793a40ed3dSBarry Smith PetscFunctionReturn(0); 48809b94acceSBarry Smith } 48819b94acceSBarry Smith 48829b94acceSBarry Smith /*@C 48833638b69dSLois Curfman McInnes SNESGetFunction - Returns the vector where the function is stored. 48849b94acceSBarry Smith 4885a63bb30eSJed Brown Not Collective, but Vec is parallel if SNES is parallel. Collective if Vec is requested, but has not been created yet. 4886c7afd0dbSLois Curfman McInnes 48879b94acceSBarry Smith Input Parameter: 48889b94acceSBarry Smith . snes - the SNES context 48899b94acceSBarry Smith 48909b94acceSBarry Smith Output Parameter: 48910298fd71SBarry Smith + r - the vector that is used to store residuals (or NULL if you don't want it) 4892f8b49ee9SBarry Smith . f - the function (or NULL if you don't want it); see SNESFunction for calling sequence details 48930298fd71SBarry Smith - ctx - the function context (or NULL if you don't want it) 48949b94acceSBarry Smith 489536851e7fSLois Curfman McInnes Level: advanced 489636851e7fSLois Curfman McInnes 489704edfde5SBarry Smith Notes: The vector r DOES NOT, in general contain the current value of the SNES nonlinear function 489804edfde5SBarry Smith 4899bf388a1fSBarry Smith .seealso: SNESSetFunction(), SNESGetSolution(), SNESFunction 49009b94acceSBarry Smith @*/ 4901f8b49ee9SBarry Smith PetscErrorCode SNESGetFunction(SNES snes,Vec *r,PetscErrorCode (**f)(SNES,Vec,Vec,void*),void **ctx) 49029b94acceSBarry Smith { 4903a63bb30eSJed Brown PetscErrorCode ierr; 49046cab3a1bSJed Brown DM dm; 4905a63bb30eSJed Brown 49063a40ed3dSBarry Smith PetscFunctionBegin; 49070700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 4908a63bb30eSJed Brown if (r) { 4909a63bb30eSJed Brown if (!snes->vec_func) { 4910a63bb30eSJed Brown if (snes->vec_rhs) { 4911a63bb30eSJed Brown ierr = VecDuplicate(snes->vec_rhs,&snes->vec_func);CHKERRQ(ierr); 4912a63bb30eSJed Brown } else if (snes->vec_sol) { 4913a63bb30eSJed Brown ierr = VecDuplicate(snes->vec_sol,&snes->vec_func);CHKERRQ(ierr); 4914a63bb30eSJed Brown } else if (snes->dm) { 4915a63bb30eSJed Brown ierr = DMCreateGlobalVector(snes->dm,&snes->vec_func);CHKERRQ(ierr); 4916a63bb30eSJed Brown } 4917a63bb30eSJed Brown } 4918a63bb30eSJed Brown *r = snes->vec_func; 4919a63bb30eSJed Brown } 49206cab3a1bSJed Brown ierr = SNESGetDM(snes,&dm);CHKERRQ(ierr); 4921f8b49ee9SBarry Smith ierr = DMSNESGetFunction(dm,f,ctx);CHKERRQ(ierr); 49223a40ed3dSBarry Smith PetscFunctionReturn(0); 49239b94acceSBarry Smith } 49249b94acceSBarry Smith 4925c79ef259SPeter Brune /*@C 4926be95d8f1SBarry Smith SNESGetNGS - Returns the NGS function and context. 4927c79ef259SPeter Brune 4928c79ef259SPeter Brune Input Parameter: 4929c79ef259SPeter Brune . snes - the SNES context 4930c79ef259SPeter Brune 4931c79ef259SPeter Brune Output Parameter: 4932be95d8f1SBarry Smith + f - the function (or NULL) see SNESNGSFunction for details 49330298fd71SBarry Smith - ctx - the function context (or NULL) 4934c79ef259SPeter Brune 4935c79ef259SPeter Brune Level: advanced 4936c79ef259SPeter Brune 4937be95d8f1SBarry Smith .seealso: SNESSetNGS(), SNESGetFunction() 4938c79ef259SPeter Brune @*/ 4939c79ef259SPeter Brune 4940be95d8f1SBarry Smith PetscErrorCode SNESGetNGS (SNES snes, PetscErrorCode (**f)(SNES, Vec, Vec, void*), void ** ctx) 4941646217ecSPeter Brune { 49426cab3a1bSJed Brown PetscErrorCode ierr; 49436cab3a1bSJed Brown DM dm; 49446cab3a1bSJed Brown 4945646217ecSPeter Brune PetscFunctionBegin; 4946646217ecSPeter Brune PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 49476cab3a1bSJed Brown ierr = SNESGetDM(snes,&dm);CHKERRQ(ierr); 4948be95d8f1SBarry Smith ierr = DMSNESGetNGS(dm,f,ctx);CHKERRQ(ierr); 4949646217ecSPeter Brune PetscFunctionReturn(0); 4950646217ecSPeter Brune } 4951646217ecSPeter Brune 49523c7409f5SSatish Balay /*@C 49533c7409f5SSatish Balay SNESSetOptionsPrefix - Sets the prefix used for searching for all 4954d850072dSLois Curfman McInnes SNES options in the database. 49553c7409f5SSatish Balay 49563f9fe445SBarry Smith Logically Collective on SNES 4957fee21e36SBarry Smith 4958c7afd0dbSLois Curfman McInnes Input Parameter: 4959c7afd0dbSLois Curfman McInnes + snes - the SNES context 4960c7afd0dbSLois Curfman McInnes - prefix - the prefix to prepend to all option names 4961c7afd0dbSLois Curfman McInnes 4962d850072dSLois Curfman McInnes Notes: 4963a83b1b31SSatish Balay A hyphen (-) must NOT be given at the beginning of the prefix name. 4964c7afd0dbSLois Curfman McInnes The first character of all runtime options is AUTOMATICALLY the hyphen. 4965d850072dSLois Curfman McInnes 496636851e7fSLois Curfman McInnes Level: advanced 496736851e7fSLois Curfman McInnes 4968a86d99e1SLois Curfman McInnes .seealso: SNESSetFromOptions() 49693c7409f5SSatish Balay @*/ 49707087cfbeSBarry Smith PetscErrorCode SNESSetOptionsPrefix(SNES snes,const char prefix[]) 49713c7409f5SSatish Balay { 4972dfbe8321SBarry Smith PetscErrorCode ierr; 49733c7409f5SSatish Balay 49743a40ed3dSBarry Smith PetscFunctionBegin; 49750700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 4976639f9d9dSBarry Smith ierr = PetscObjectSetOptionsPrefix((PetscObject)snes,prefix);CHKERRQ(ierr); 49771cee3971SBarry Smith if (!snes->ksp) {ierr = SNESGetKSP(snes,&snes->ksp);CHKERRQ(ierr);} 497835f5d045SPeter Brune if (snes->linesearch) { 49797601faf0SJed Brown ierr = SNESGetLineSearch(snes,&snes->linesearch);CHKERRQ(ierr); 498008b6c495SPeter Brune ierr = PetscObjectSetOptionsPrefix((PetscObject)snes->linesearch,prefix);CHKERRQ(ierr); 498135f5d045SPeter Brune } 498235f5d045SPeter Brune ierr = KSPSetOptionsPrefix(snes->ksp,prefix);CHKERRQ(ierr); 49833a40ed3dSBarry Smith PetscFunctionReturn(0); 49843c7409f5SSatish Balay } 49853c7409f5SSatish Balay 49863c7409f5SSatish Balay /*@C 4987f525115eSLois Curfman McInnes SNESAppendOptionsPrefix - Appends to the prefix used for searching for all 4988d850072dSLois Curfman McInnes SNES options in the database. 49893c7409f5SSatish Balay 49903f9fe445SBarry Smith Logically Collective on SNES 4991fee21e36SBarry Smith 4992c7afd0dbSLois Curfman McInnes Input Parameters: 4993c7afd0dbSLois Curfman McInnes + snes - the SNES context 4994c7afd0dbSLois Curfman McInnes - prefix - the prefix to prepend to all option names 4995c7afd0dbSLois Curfman McInnes 4996d850072dSLois Curfman McInnes Notes: 4997a83b1b31SSatish Balay A hyphen (-) must NOT be given at the beginning of the prefix name. 4998c7afd0dbSLois Curfman McInnes The first character of all runtime options is AUTOMATICALLY the hyphen. 4999d850072dSLois Curfman McInnes 500036851e7fSLois Curfman McInnes Level: advanced 500136851e7fSLois Curfman McInnes 5002a86d99e1SLois Curfman McInnes .seealso: SNESGetOptionsPrefix() 50033c7409f5SSatish Balay @*/ 50047087cfbeSBarry Smith PetscErrorCode SNESAppendOptionsPrefix(SNES snes,const char prefix[]) 50053c7409f5SSatish Balay { 5006dfbe8321SBarry Smith PetscErrorCode ierr; 50073c7409f5SSatish Balay 50083a40ed3dSBarry Smith PetscFunctionBegin; 50090700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 5010639f9d9dSBarry Smith ierr = PetscObjectAppendOptionsPrefix((PetscObject)snes,prefix);CHKERRQ(ierr); 50111cee3971SBarry Smith if (!snes->ksp) {ierr = SNESGetKSP(snes,&snes->ksp);CHKERRQ(ierr);} 501235f5d045SPeter Brune if (snes->linesearch) { 50137601faf0SJed Brown ierr = SNESGetLineSearch(snes,&snes->linesearch);CHKERRQ(ierr); 501408b6c495SPeter Brune ierr = PetscObjectAppendOptionsPrefix((PetscObject)snes->linesearch,prefix);CHKERRQ(ierr); 501535f5d045SPeter Brune } 501635f5d045SPeter Brune ierr = KSPAppendOptionsPrefix(snes->ksp,prefix);CHKERRQ(ierr); 50173a40ed3dSBarry Smith PetscFunctionReturn(0); 50183c7409f5SSatish Balay } 50193c7409f5SSatish Balay 50209ab63eb5SSatish Balay /*@C 50213c7409f5SSatish Balay SNESGetOptionsPrefix - Sets the prefix used for searching for all 50223c7409f5SSatish Balay SNES options in the database. 50233c7409f5SSatish Balay 5024c7afd0dbSLois Curfman McInnes Not Collective 5025c7afd0dbSLois Curfman McInnes 50263c7409f5SSatish Balay Input Parameter: 50273c7409f5SSatish Balay . snes - the SNES context 50283c7409f5SSatish Balay 50293c7409f5SSatish Balay Output Parameter: 50303c7409f5SSatish Balay . prefix - pointer to the prefix string used 50313c7409f5SSatish Balay 503295452b02SPatrick Sanan Notes: 503395452b02SPatrick Sanan On the fortran side, the user should pass in a string 'prefix' of 50349ab63eb5SSatish Balay sufficient length to hold the prefix. 50359ab63eb5SSatish Balay 503636851e7fSLois Curfman McInnes Level: advanced 503736851e7fSLois Curfman McInnes 5038a86d99e1SLois Curfman McInnes .seealso: SNESAppendOptionsPrefix() 50393c7409f5SSatish Balay @*/ 50407087cfbeSBarry Smith PetscErrorCode SNESGetOptionsPrefix(SNES snes,const char *prefix[]) 50413c7409f5SSatish Balay { 5042dfbe8321SBarry Smith PetscErrorCode ierr; 50433c7409f5SSatish Balay 50443a40ed3dSBarry Smith PetscFunctionBegin; 50450700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 5046639f9d9dSBarry Smith ierr = PetscObjectGetOptionsPrefix((PetscObject)snes,prefix);CHKERRQ(ierr); 50473a40ed3dSBarry Smith PetscFunctionReturn(0); 50483c7409f5SSatish Balay } 50493c7409f5SSatish Balay 5050b2002411SLois Curfman McInnes 50513cea93caSBarry Smith /*@C 50521c84c290SBarry Smith SNESRegister - Adds a method to the nonlinear solver package. 50531c84c290SBarry Smith 50541c84c290SBarry Smith Not collective 50551c84c290SBarry Smith 50561c84c290SBarry Smith Input Parameters: 50571c84c290SBarry Smith + name_solver - name of a new user-defined solver 50581c84c290SBarry Smith - routine_create - routine to create method context 50591c84c290SBarry Smith 50601c84c290SBarry Smith Notes: 50611c84c290SBarry Smith SNESRegister() may be called multiple times to add several user-defined solvers. 50621c84c290SBarry Smith 50631c84c290SBarry Smith Sample usage: 50641c84c290SBarry Smith .vb 5065bdf89e91SBarry Smith SNESRegister("my_solver",MySolverCreate); 50661c84c290SBarry Smith .ve 50671c84c290SBarry Smith 50681c84c290SBarry Smith Then, your solver can be chosen with the procedural interface via 50691c84c290SBarry Smith $ SNESSetType(snes,"my_solver") 50701c84c290SBarry Smith or at runtime via the option 50711c84c290SBarry Smith $ -snes_type my_solver 50721c84c290SBarry Smith 50731c84c290SBarry Smith Level: advanced 50741c84c290SBarry Smith 50751c84c290SBarry Smith Note: If your function is not being put into a shared library then use SNESRegister() instead 50761c84c290SBarry Smith 50771c84c290SBarry Smith .seealso: SNESRegisterAll(), SNESRegisterDestroy() 50783cea93caSBarry Smith 50797f6c08e0SMatthew Knepley Level: advanced 50803cea93caSBarry Smith @*/ 5081bdf89e91SBarry Smith PetscErrorCode SNESRegister(const char sname[],PetscErrorCode (*function)(SNES)) 5082b2002411SLois Curfman McInnes { 5083dfbe8321SBarry Smith PetscErrorCode ierr; 5084b2002411SLois Curfman McInnes 5085b2002411SLois Curfman McInnes PetscFunctionBegin; 50861d36bdfdSBarry Smith ierr = SNESInitializePackage();CHKERRQ(ierr); 5087a240a19fSJed Brown ierr = PetscFunctionListAdd(&SNESList,sname,function);CHKERRQ(ierr); 5088b2002411SLois Curfman McInnes PetscFunctionReturn(0); 5089b2002411SLois Curfman McInnes } 5090da9b6338SBarry Smith 50917087cfbeSBarry Smith PetscErrorCode SNESTestLocalMin(SNES snes) 5092da9b6338SBarry Smith { 5093dfbe8321SBarry Smith PetscErrorCode ierr; 509477431f27SBarry Smith PetscInt N,i,j; 5095da9b6338SBarry Smith Vec u,uh,fh; 5096da9b6338SBarry Smith PetscScalar value; 5097da9b6338SBarry Smith PetscReal norm; 5098da9b6338SBarry Smith 5099da9b6338SBarry Smith PetscFunctionBegin; 5100da9b6338SBarry Smith ierr = SNESGetSolution(snes,&u);CHKERRQ(ierr); 5101da9b6338SBarry Smith ierr = VecDuplicate(u,&uh);CHKERRQ(ierr); 5102da9b6338SBarry Smith ierr = VecDuplicate(u,&fh);CHKERRQ(ierr); 5103da9b6338SBarry Smith 5104da9b6338SBarry Smith /* currently only works for sequential */ 5105ea13f565SStefano Zampini ierr = PetscPrintf(PetscObjectComm((PetscObject)snes),"Testing FormFunction() for local min\n");CHKERRQ(ierr); 5106da9b6338SBarry Smith ierr = VecGetSize(u,&N);CHKERRQ(ierr); 5107da9b6338SBarry Smith for (i=0; i<N; i++) { 5108da9b6338SBarry Smith ierr = VecCopy(u,uh);CHKERRQ(ierr); 5109ea13f565SStefano Zampini ierr = PetscPrintf(PetscObjectComm((PetscObject)snes),"i = %D\n",i);CHKERRQ(ierr); 5110da9b6338SBarry Smith for (j=-10; j<11; j++) { 51118b49ba18SBarry Smith value = PetscSign(j)*PetscExpReal(PetscAbs(j)-10.0); 5112da9b6338SBarry Smith ierr = VecSetValue(uh,i,value,ADD_VALUES);CHKERRQ(ierr); 51133ab0aad5SBarry Smith ierr = SNESComputeFunction(snes,uh,fh);CHKERRQ(ierr); 5114da9b6338SBarry Smith ierr = VecNorm(fh,NORM_2,&norm);CHKERRQ(ierr); 5115ea13f565SStefano Zampini ierr = PetscPrintf(PetscObjectComm((PetscObject)snes)," j norm %D %18.16e\n",j,norm);CHKERRQ(ierr); 5116da9b6338SBarry Smith value = -value; 5117da9b6338SBarry Smith ierr = VecSetValue(uh,i,value,ADD_VALUES);CHKERRQ(ierr); 5118da9b6338SBarry Smith } 5119da9b6338SBarry Smith } 51206bf464f9SBarry Smith ierr = VecDestroy(&uh);CHKERRQ(ierr); 51216bf464f9SBarry Smith ierr = VecDestroy(&fh);CHKERRQ(ierr); 5122da9b6338SBarry Smith PetscFunctionReturn(0); 5123da9b6338SBarry Smith } 512471f87433Sdalcinl 512571f87433Sdalcinl /*@ 5126fa9f3622SBarry Smith SNESKSPSetUseEW - Sets SNES use Eisenstat-Walker method for 512771f87433Sdalcinl computing relative tolerance for linear solvers within an inexact 512871f87433Sdalcinl Newton method. 512971f87433Sdalcinl 51303f9fe445SBarry Smith Logically Collective on SNES 513171f87433Sdalcinl 513271f87433Sdalcinl Input Parameters: 513371f87433Sdalcinl + snes - SNES context 513471f87433Sdalcinl - flag - PETSC_TRUE or PETSC_FALSE 513571f87433Sdalcinl 513664ba62caSBarry Smith Options Database: 513764ba62caSBarry Smith + -snes_ksp_ew - use Eisenstat-Walker method for determining linear system convergence 513864ba62caSBarry Smith . -snes_ksp_ew_version ver - version of Eisenstat-Walker method 513964ba62caSBarry Smith . -snes_ksp_ew_rtol0 <rtol0> - Sets rtol0 514064ba62caSBarry Smith . -snes_ksp_ew_rtolmax <rtolmax> - Sets rtolmax 514164ba62caSBarry Smith . -snes_ksp_ew_gamma <gamma> - Sets gamma 514264ba62caSBarry Smith . -snes_ksp_ew_alpha <alpha> - Sets alpha 514364ba62caSBarry Smith . -snes_ksp_ew_alpha2 <alpha2> - Sets alpha2 514464ba62caSBarry Smith - -snes_ksp_ew_threshold <threshold> - Sets threshold 514564ba62caSBarry Smith 514671f87433Sdalcinl Notes: 514771f87433Sdalcinl Currently, the default is to use a constant relative tolerance for 514871f87433Sdalcinl the inner linear solvers. Alternatively, one can use the 514971f87433Sdalcinl Eisenstat-Walker method, where the relative convergence tolerance 515071f87433Sdalcinl is reset at each Newton iteration according progress of the nonlinear 515171f87433Sdalcinl solver. 515271f87433Sdalcinl 515371f87433Sdalcinl Level: advanced 515471f87433Sdalcinl 515571f87433Sdalcinl Reference: 515671f87433Sdalcinl S. C. Eisenstat and H. F. Walker, "Choosing the forcing terms in an 515771f87433Sdalcinl inexact Newton method", SISC 17 (1), pp.16-32, 1996. 515871f87433Sdalcinl 5159fa9f3622SBarry Smith .seealso: SNESKSPGetUseEW(), SNESKSPGetParametersEW(), SNESKSPSetParametersEW() 516071f87433Sdalcinl @*/ 51617087cfbeSBarry Smith PetscErrorCode SNESKSPSetUseEW(SNES snes,PetscBool flag) 516271f87433Sdalcinl { 516371f87433Sdalcinl PetscFunctionBegin; 51640700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 5165acfcf0e5SJed Brown PetscValidLogicalCollectiveBool(snes,flag,2); 516671f87433Sdalcinl snes->ksp_ewconv = flag; 516771f87433Sdalcinl PetscFunctionReturn(0); 516871f87433Sdalcinl } 516971f87433Sdalcinl 517071f87433Sdalcinl /*@ 5171fa9f3622SBarry Smith SNESKSPGetUseEW - Gets if SNES is using Eisenstat-Walker method 517271f87433Sdalcinl for computing relative tolerance for linear solvers within an 517371f87433Sdalcinl inexact Newton method. 517471f87433Sdalcinl 517571f87433Sdalcinl Not Collective 517671f87433Sdalcinl 517771f87433Sdalcinl Input Parameter: 517871f87433Sdalcinl . snes - SNES context 517971f87433Sdalcinl 518071f87433Sdalcinl Output Parameter: 518171f87433Sdalcinl . flag - PETSC_TRUE or PETSC_FALSE 518271f87433Sdalcinl 518371f87433Sdalcinl Notes: 518471f87433Sdalcinl Currently, the default is to use a constant relative tolerance for 518571f87433Sdalcinl the inner linear solvers. Alternatively, one can use the 518671f87433Sdalcinl Eisenstat-Walker method, where the relative convergence tolerance 518771f87433Sdalcinl is reset at each Newton iteration according progress of the nonlinear 518871f87433Sdalcinl solver. 518971f87433Sdalcinl 519071f87433Sdalcinl Level: advanced 519171f87433Sdalcinl 519271f87433Sdalcinl Reference: 519371f87433Sdalcinl S. C. Eisenstat and H. F. Walker, "Choosing the forcing terms in an 519471f87433Sdalcinl inexact Newton method", SISC 17 (1), pp.16-32, 1996. 519571f87433Sdalcinl 5196fa9f3622SBarry Smith .seealso: SNESKSPSetUseEW(), SNESKSPGetParametersEW(), SNESKSPSetParametersEW() 519771f87433Sdalcinl @*/ 51987087cfbeSBarry Smith PetscErrorCode SNESKSPGetUseEW(SNES snes, PetscBool *flag) 519971f87433Sdalcinl { 520071f87433Sdalcinl PetscFunctionBegin; 52010700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 5202534a8f05SLisandro Dalcin PetscValidBoolPointer(flag,2); 520371f87433Sdalcinl *flag = snes->ksp_ewconv; 520471f87433Sdalcinl PetscFunctionReturn(0); 520571f87433Sdalcinl } 520671f87433Sdalcinl 520771f87433Sdalcinl /*@ 5208fa9f3622SBarry Smith SNESKSPSetParametersEW - Sets parameters for Eisenstat-Walker 520971f87433Sdalcinl convergence criteria for the linear solvers within an inexact 521071f87433Sdalcinl Newton method. 521171f87433Sdalcinl 52123f9fe445SBarry Smith Logically Collective on SNES 521371f87433Sdalcinl 521471f87433Sdalcinl Input Parameters: 521571f87433Sdalcinl + snes - SNES context 521671f87433Sdalcinl . version - version 1, 2 (default is 2) or 3 521771f87433Sdalcinl . rtol_0 - initial relative tolerance (0 <= rtol_0 < 1) 521871f87433Sdalcinl . rtol_max - maximum relative tolerance (0 <= rtol_max < 1) 521971f87433Sdalcinl . gamma - multiplicative factor for version 2 rtol computation 522071f87433Sdalcinl (0 <= gamma2 <= 1) 522171f87433Sdalcinl . alpha - power for version 2 rtol computation (1 < alpha <= 2) 522271f87433Sdalcinl . alpha2 - power for safeguard 522371f87433Sdalcinl - threshold - threshold for imposing safeguard (0 < threshold < 1) 522471f87433Sdalcinl 522571f87433Sdalcinl Note: 522671f87433Sdalcinl Version 3 was contributed by Luis Chacon, June 2006. 522771f87433Sdalcinl 522871f87433Sdalcinl Use PETSC_DEFAULT to retain the default for any of the parameters. 522971f87433Sdalcinl 523071f87433Sdalcinl Level: advanced 523171f87433Sdalcinl 523271f87433Sdalcinl Reference: 523371f87433Sdalcinl S. C. Eisenstat and H. F. Walker, "Choosing the forcing terms in an 523471f87433Sdalcinl inexact Newton method", Utah State University Math. Stat. Dept. Res. 523571f87433Sdalcinl Report 6/94/75, June, 1994, to appear in SIAM J. Sci. Comput. 523671f87433Sdalcinl 5237fa9f3622SBarry Smith .seealso: SNESKSPSetUseEW(), SNESKSPGetUseEW(), SNESKSPGetParametersEW() 523871f87433Sdalcinl @*/ 5239f5af7f23SKarl Rupp PetscErrorCode SNESKSPSetParametersEW(SNES snes,PetscInt version,PetscReal rtol_0,PetscReal rtol_max,PetscReal gamma,PetscReal alpha,PetscReal alpha2,PetscReal threshold) 524071f87433Sdalcinl { 5241fa9f3622SBarry Smith SNESKSPEW *kctx; 52425fd66863SKarl Rupp 524371f87433Sdalcinl PetscFunctionBegin; 52440700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 5245fa9f3622SBarry Smith kctx = (SNESKSPEW*)snes->kspconvctx; 5246e32f2f54SBarry Smith if (!kctx) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"No Eisenstat-Walker context existing"); 5247c5eb9154SBarry Smith PetscValidLogicalCollectiveInt(snes,version,2); 5248c5eb9154SBarry Smith PetscValidLogicalCollectiveReal(snes,rtol_0,3); 5249c5eb9154SBarry Smith PetscValidLogicalCollectiveReal(snes,rtol_max,4); 5250c5eb9154SBarry Smith PetscValidLogicalCollectiveReal(snes,gamma,5); 5251c5eb9154SBarry Smith PetscValidLogicalCollectiveReal(snes,alpha,6); 5252c5eb9154SBarry Smith PetscValidLogicalCollectiveReal(snes,alpha2,7); 5253c5eb9154SBarry Smith PetscValidLogicalCollectiveReal(snes,threshold,8); 525471f87433Sdalcinl 525571f87433Sdalcinl if (version != PETSC_DEFAULT) kctx->version = version; 525671f87433Sdalcinl if (rtol_0 != PETSC_DEFAULT) kctx->rtol_0 = rtol_0; 525771f87433Sdalcinl if (rtol_max != PETSC_DEFAULT) kctx->rtol_max = rtol_max; 525871f87433Sdalcinl if (gamma != PETSC_DEFAULT) kctx->gamma = gamma; 525971f87433Sdalcinl if (alpha != PETSC_DEFAULT) kctx->alpha = alpha; 526071f87433Sdalcinl if (alpha2 != PETSC_DEFAULT) kctx->alpha2 = alpha2; 526171f87433Sdalcinl if (threshold != PETSC_DEFAULT) kctx->threshold = threshold; 526271f87433Sdalcinl 5263f23aa3ddSBarry 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); 526457622a8eSBarry 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); 526557622a8eSBarry 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); 526657622a8eSBarry 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); 526757622a8eSBarry 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); 526857622a8eSBarry 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); 526971f87433Sdalcinl PetscFunctionReturn(0); 527071f87433Sdalcinl } 527171f87433Sdalcinl 527271f87433Sdalcinl /*@ 5273fa9f3622SBarry Smith SNESKSPGetParametersEW - Gets parameters for Eisenstat-Walker 527471f87433Sdalcinl convergence criteria for the linear solvers within an inexact 527571f87433Sdalcinl Newton method. 527671f87433Sdalcinl 527771f87433Sdalcinl Not Collective 527871f87433Sdalcinl 527971f87433Sdalcinl Input Parameters: 528071f87433Sdalcinl snes - SNES context 528171f87433Sdalcinl 528271f87433Sdalcinl Output Parameters: 528371f87433Sdalcinl + version - version 1, 2 (default is 2) or 3 528471f87433Sdalcinl . rtol_0 - initial relative tolerance (0 <= rtol_0 < 1) 528571f87433Sdalcinl . rtol_max - maximum relative tolerance (0 <= rtol_max < 1) 5286bf388a1fSBarry Smith . gamma - multiplicative factor for version 2 rtol computation (0 <= gamma2 <= 1) 528771f87433Sdalcinl . alpha - power for version 2 rtol computation (1 < alpha <= 2) 528871f87433Sdalcinl . alpha2 - power for safeguard 528971f87433Sdalcinl - threshold - threshold for imposing safeguard (0 < threshold < 1) 529071f87433Sdalcinl 529171f87433Sdalcinl Level: advanced 529271f87433Sdalcinl 5293fa9f3622SBarry Smith .seealso: SNESKSPSetUseEW(), SNESKSPGetUseEW(), SNESKSPSetParametersEW() 529471f87433Sdalcinl @*/ 5295bf388a1fSBarry Smith PetscErrorCode SNESKSPGetParametersEW(SNES snes,PetscInt *version,PetscReal *rtol_0,PetscReal *rtol_max,PetscReal *gamma,PetscReal *alpha,PetscReal *alpha2,PetscReal *threshold) 529671f87433Sdalcinl { 5297fa9f3622SBarry Smith SNESKSPEW *kctx; 52985fd66863SKarl Rupp 529971f87433Sdalcinl PetscFunctionBegin; 53000700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 5301fa9f3622SBarry Smith kctx = (SNESKSPEW*)snes->kspconvctx; 5302e32f2f54SBarry Smith if (!kctx) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"No Eisenstat-Walker context existing"); 530371f87433Sdalcinl if (version) *version = kctx->version; 530471f87433Sdalcinl if (rtol_0) *rtol_0 = kctx->rtol_0; 530571f87433Sdalcinl if (rtol_max) *rtol_max = kctx->rtol_max; 530671f87433Sdalcinl if (gamma) *gamma = kctx->gamma; 530771f87433Sdalcinl if (alpha) *alpha = kctx->alpha; 530871f87433Sdalcinl if (alpha2) *alpha2 = kctx->alpha2; 530971f87433Sdalcinl if (threshold) *threshold = kctx->threshold; 531071f87433Sdalcinl PetscFunctionReturn(0); 531171f87433Sdalcinl } 531271f87433Sdalcinl 5313d5378b5fSDmitry Karpeev PetscErrorCode KSPPreSolve_SNESEW(KSP ksp, Vec b, Vec x, SNES snes) 531471f87433Sdalcinl { 531571f87433Sdalcinl PetscErrorCode ierr; 5316fa9f3622SBarry Smith SNESKSPEW *kctx = (SNESKSPEW*)snes->kspconvctx; 531771f87433Sdalcinl PetscReal rtol = PETSC_DEFAULT,stol; 531871f87433Sdalcinl 531971f87433Sdalcinl PetscFunctionBegin; 5320d4211eb9SBarry Smith if (!snes->ksp_ewconv) PetscFunctionReturn(0); 532130058271SDmitry Karpeev if (!snes->iter) { 532230058271SDmitry Karpeev rtol = kctx->rtol_0; /* first time in, so use the original user rtol */ 532330058271SDmitry Karpeev ierr = VecNorm(snes->vec_func,NORM_2,&kctx->norm_first);CHKERRQ(ierr); 532430058271SDmitry Karpeev } 5325f5af7f23SKarl Rupp else { 532671f87433Sdalcinl if (kctx->version == 1) { 532771f87433Sdalcinl rtol = (snes->norm - kctx->lresid_last)/kctx->norm_last; 532871f87433Sdalcinl if (rtol < 0.0) rtol = -rtol; 532985ec1a3cSBarry Smith stol = PetscPowReal(kctx->rtol_last,kctx->alpha2); 533071f87433Sdalcinl if (stol > kctx->threshold) rtol = PetscMax(rtol,stol); 533171f87433Sdalcinl } else if (kctx->version == 2) { 533285ec1a3cSBarry Smith rtol = kctx->gamma * PetscPowReal(snes->norm/kctx->norm_last,kctx->alpha); 533385ec1a3cSBarry Smith stol = kctx->gamma * PetscPowReal(kctx->rtol_last,kctx->alpha); 533471f87433Sdalcinl if (stol > kctx->threshold) rtol = PetscMax(rtol,stol); 533571f87433Sdalcinl } else if (kctx->version == 3) { /* contributed by Luis Chacon, June 2006. */ 533685ec1a3cSBarry Smith rtol = kctx->gamma * PetscPowReal(snes->norm/kctx->norm_last,kctx->alpha); 533771f87433Sdalcinl /* safeguard: avoid sharp decrease of rtol */ 533885ec1a3cSBarry Smith stol = kctx->gamma*PetscPowReal(kctx->rtol_last,kctx->alpha); 533971f87433Sdalcinl stol = PetscMax(rtol,stol); 534071f87433Sdalcinl rtol = PetscMin(kctx->rtol_0,stol); 534171f87433Sdalcinl /* safeguard: avoid oversolving */ 534230058271SDmitry Karpeev stol = kctx->gamma*(kctx->norm_first*snes->rtol)/snes->norm; 534371f87433Sdalcinl stol = PetscMax(rtol,stol); 534471f87433Sdalcinl rtol = PetscMin(kctx->rtol_0,stol); 5345e32f2f54SBarry Smith } else SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Only versions 1, 2 or 3 are supported: %D",kctx->version); 534671f87433Sdalcinl } 534771f87433Sdalcinl /* safeguard: avoid rtol greater than one */ 534871f87433Sdalcinl rtol = PetscMin(rtol,kctx->rtol_max); 534971f87433Sdalcinl ierr = KSPSetTolerances(ksp,rtol,PETSC_DEFAULT,PETSC_DEFAULT,PETSC_DEFAULT);CHKERRQ(ierr); 535057622a8eSBarry Smith ierr = PetscInfo3(snes,"iter %D, Eisenstat-Walker (version %D) KSP rtol=%g\n",snes->iter,kctx->version,(double)rtol);CHKERRQ(ierr); 535171f87433Sdalcinl PetscFunctionReturn(0); 535271f87433Sdalcinl } 535371f87433Sdalcinl 5354d5378b5fSDmitry Karpeev PetscErrorCode KSPPostSolve_SNESEW(KSP ksp, Vec b, Vec x, SNES snes) 535571f87433Sdalcinl { 535671f87433Sdalcinl PetscErrorCode ierr; 5357fa9f3622SBarry Smith SNESKSPEW *kctx = (SNESKSPEW*)snes->kspconvctx; 535871f87433Sdalcinl PCSide pcside; 535971f87433Sdalcinl Vec lres; 536071f87433Sdalcinl 536171f87433Sdalcinl PetscFunctionBegin; 5362d4211eb9SBarry Smith if (!snes->ksp_ewconv) PetscFunctionReturn(0); 53639e5d0892SLisandro Dalcin ierr = KSPGetTolerances(ksp,&kctx->rtol_last,NULL,NULL,NULL);CHKERRQ(ierr); 536471dbe336SPeter Brune kctx->norm_last = snes->norm; 536571f87433Sdalcinl if (kctx->version == 1) { 53664f00ce20SMatthew G. Knepley PC pc; 53674f00ce20SMatthew G. Knepley PetscBool isNone; 53684f00ce20SMatthew G. Knepley 53694f00ce20SMatthew G. Knepley ierr = KSPGetPC(ksp, &pc);CHKERRQ(ierr); 53704f00ce20SMatthew G. Knepley ierr = PetscObjectTypeCompare((PetscObject) pc, PCNONE, &isNone);CHKERRQ(ierr); 5371b037da10SBarry Smith ierr = KSPGetPCSide(ksp,&pcside);CHKERRQ(ierr); 53724f00ce20SMatthew G. Knepley if (pcside == PC_RIGHT || isNone) { /* XXX Should we also test KSP_UNPRECONDITIONED_NORM ? */ 537371f87433Sdalcinl /* KSP residual is true linear residual */ 537471f87433Sdalcinl ierr = KSPGetResidualNorm(ksp,&kctx->lresid_last);CHKERRQ(ierr); 537571f87433Sdalcinl } else { 537671f87433Sdalcinl /* KSP residual is preconditioned residual */ 537771f87433Sdalcinl /* compute true linear residual norm */ 537871f87433Sdalcinl ierr = VecDuplicate(b,&lres);CHKERRQ(ierr); 537971f87433Sdalcinl ierr = MatMult(snes->jacobian,x,lres);CHKERRQ(ierr); 538071f87433Sdalcinl ierr = VecAYPX(lres,-1.0,b);CHKERRQ(ierr); 538171f87433Sdalcinl ierr = VecNorm(lres,NORM_2,&kctx->lresid_last);CHKERRQ(ierr); 53826bf464f9SBarry Smith ierr = VecDestroy(&lres);CHKERRQ(ierr); 538371f87433Sdalcinl } 538471f87433Sdalcinl } 538571f87433Sdalcinl PetscFunctionReturn(0); 538671f87433Sdalcinl } 538771f87433Sdalcinl 5388d4211eb9SBarry Smith /*@ 5389d4211eb9SBarry Smith SNESGetKSP - Returns the KSP context for a SNES solver. 5390d4211eb9SBarry Smith 5391d4211eb9SBarry Smith Not Collective, but if SNES object is parallel, then KSP object is parallel 5392d4211eb9SBarry Smith 5393d4211eb9SBarry Smith Input Parameter: 5394d4211eb9SBarry Smith . snes - the SNES context 5395d4211eb9SBarry Smith 5396d4211eb9SBarry Smith Output Parameter: 5397d4211eb9SBarry Smith . ksp - the KSP context 5398d4211eb9SBarry Smith 5399d4211eb9SBarry Smith Notes: 5400d4211eb9SBarry Smith The user can then directly manipulate the KSP context to set various 5401d4211eb9SBarry Smith options, etc. Likewise, the user can then extract and manipulate the 5402d4211eb9SBarry Smith PC contexts as well. 5403d4211eb9SBarry Smith 5404d4211eb9SBarry Smith Level: beginner 5405d4211eb9SBarry Smith 5406d4211eb9SBarry Smith .seealso: KSPGetPC(), SNESCreate(), KSPCreate(), SNESSetKSP() 5407d4211eb9SBarry Smith @*/ 5408d4211eb9SBarry Smith PetscErrorCode SNESGetKSP(SNES snes,KSP *ksp) 540971f87433Sdalcinl { 541071f87433Sdalcinl PetscErrorCode ierr; 541171f87433Sdalcinl 541271f87433Sdalcinl PetscFunctionBegin; 5413d4211eb9SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 5414d4211eb9SBarry Smith PetscValidPointer(ksp,2); 5415d4211eb9SBarry Smith 5416d4211eb9SBarry Smith if (!snes->ksp) { 5417d4211eb9SBarry Smith ierr = KSPCreate(PetscObjectComm((PetscObject)snes),&snes->ksp);CHKERRQ(ierr); 5418d4211eb9SBarry Smith ierr = PetscObjectIncrementTabLevel((PetscObject)snes->ksp,(PetscObject)snes,1);CHKERRQ(ierr); 54193bb1ff40SBarry Smith ierr = PetscLogObjectParent((PetscObject)snes,(PetscObject)snes->ksp);CHKERRQ(ierr); 5420d4211eb9SBarry Smith 5421d5378b5fSDmitry Karpeev ierr = KSPSetPreSolve(snes->ksp,(PetscErrorCode (*)(KSP,Vec,Vec,void*))KSPPreSolve_SNESEW,snes);CHKERRQ(ierr); 5422d5378b5fSDmitry Karpeev ierr = KSPSetPostSolve(snes->ksp,(PetscErrorCode (*)(KSP,Vec,Vec,void*))KSPPostSolve_SNESEW,snes);CHKERRQ(ierr); 5423a5c2985bSBarry Smith 5424798534f6SMatthew G. Knepley ierr = KSPMonitorSetFromOptions(snes->ksp, "-snes_monitor_ksp", "snes_preconditioned_residual", snes);CHKERRQ(ierr); 542516413a6aSBarry Smith ierr = PetscObjectSetOptions((PetscObject)snes->ksp,((PetscObject)snes)->options);CHKERRQ(ierr); 5426d4211eb9SBarry Smith } 5427d4211eb9SBarry Smith *ksp = snes->ksp; 542871f87433Sdalcinl PetscFunctionReturn(0); 542971f87433Sdalcinl } 54306c699258SBarry Smith 5431d4211eb9SBarry Smith 5432af0996ceSBarry Smith #include <petsc/private/dmimpl.h> 54336c699258SBarry Smith /*@ 54342a808120SBarry Smith SNESSetDM - Sets the DM that may be used by some nonlinear solvers or their underlying preconditioners 54356c699258SBarry Smith 54363f9fe445SBarry Smith Logically Collective on SNES 54376c699258SBarry Smith 54386c699258SBarry Smith Input Parameters: 54392a808120SBarry Smith + snes - the nonlinear solver context 54402a808120SBarry Smith - dm - the dm, cannot be NULL 54416c699258SBarry Smith 5442e03a659cSJed Brown Notes: 5443e03a659cSJed Brown A DM can only be used for solving one problem at a time because information about the problem is stored on the DM, 5444e03a659cSJed Brown even when not using interfaces like DMSNESSetFunction(). Use DMClone() to get a distinct DM when solving different 5445e03a659cSJed Brown problems using the same function space. 5446e03a659cSJed Brown 54476c699258SBarry Smith Level: intermediate 54486c699258SBarry Smith 54494c2026ceSFande Kong .seealso: SNESGetDM(), KSPSetDM(), KSPGetDM() 54506c699258SBarry Smith @*/ 54517087cfbeSBarry Smith PetscErrorCode SNESSetDM(SNES snes,DM dm) 54526c699258SBarry Smith { 54536c699258SBarry Smith PetscErrorCode ierr; 5454345fed2cSBarry Smith KSP ksp; 5455942e3340SBarry Smith DMSNES sdm; 54566c699258SBarry Smith 54576c699258SBarry Smith PetscFunctionBegin; 54580700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 54592a808120SBarry Smith PetscValidHeaderSpecific(dm,DM_CLASSID,2); 54602a808120SBarry Smith ierr = PetscObjectReference((PetscObject)dm);CHKERRQ(ierr); 5461942e3340SBarry Smith if (snes->dm) { /* Move the DMSNES context over to the new DM unless the new DM already has one */ 546251f4b3c7SToby Isaac if (snes->dm->dmsnes && !dm->dmsnes) { 5463942e3340SBarry Smith ierr = DMCopyDMSNES(snes->dm,dm);CHKERRQ(ierr); 5464942e3340SBarry Smith ierr = DMGetDMSNES(snes->dm,&sdm);CHKERRQ(ierr); 5465f5af7f23SKarl Rupp if (sdm->originaldm == snes->dm) sdm->originaldm = dm; /* Grant write privileges to the replacement DM */ 54666cab3a1bSJed Brown } 5467dc822a44SJed Brown ierr = DMCoarsenHookRemove(snes->dm,DMCoarsenHook_SNESVecSol,DMRestrictHook_SNESVecSol,snes);CHKERRQ(ierr); 54686bf464f9SBarry Smith ierr = DMDestroy(&snes->dm);CHKERRQ(ierr); 54696cab3a1bSJed Brown } 54706c699258SBarry Smith snes->dm = dm; 5471116d1032SJed Brown snes->dmAuto = PETSC_FALSE; 5472f5af7f23SKarl Rupp 5473345fed2cSBarry Smith ierr = SNESGetKSP(snes,&ksp);CHKERRQ(ierr); 5474345fed2cSBarry Smith ierr = KSPSetDM(ksp,dm);CHKERRQ(ierr); 5475f22f69f0SBarry Smith ierr = KSPSetDMActive(ksp,PETSC_FALSE);CHKERRQ(ierr); 5476efd4aadfSBarry Smith if (snes->npc) { 5477efd4aadfSBarry Smith ierr = SNESSetDM(snes->npc, snes->dm);CHKERRQ(ierr); 5478efd4aadfSBarry Smith ierr = SNESSetNPCSide(snes,snes->npcside);CHKERRQ(ierr); 54792c155ee1SBarry Smith } 54806c699258SBarry Smith PetscFunctionReturn(0); 54816c699258SBarry Smith } 54826c699258SBarry Smith 54836c699258SBarry Smith /*@ 54846c699258SBarry Smith SNESGetDM - Gets the DM that may be used by some preconditioners 54856c699258SBarry Smith 54863f9fe445SBarry Smith Not Collective but DM obtained is parallel on SNES 54876c699258SBarry Smith 54886c699258SBarry Smith Input Parameter: 54896c699258SBarry Smith . snes - the preconditioner context 54906c699258SBarry Smith 54916c699258SBarry Smith Output Parameter: 54926c699258SBarry Smith . dm - the dm 54936c699258SBarry Smith 54946c699258SBarry Smith Level: intermediate 54956c699258SBarry Smith 54964c2026ceSFande Kong .seealso: SNESSetDM(), KSPSetDM(), KSPGetDM() 54976c699258SBarry Smith @*/ 54987087cfbeSBarry Smith PetscErrorCode SNESGetDM(SNES snes,DM *dm) 54996c699258SBarry Smith { 55006cab3a1bSJed Brown PetscErrorCode ierr; 55016cab3a1bSJed Brown 55026c699258SBarry Smith PetscFunctionBegin; 55030700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 55046cab3a1bSJed Brown if (!snes->dm) { 5505ce94432eSBarry Smith ierr = DMShellCreate(PetscObjectComm((PetscObject)snes),&snes->dm);CHKERRQ(ierr); 5506116d1032SJed Brown snes->dmAuto = PETSC_TRUE; 55076cab3a1bSJed Brown } 55086c699258SBarry Smith *dm = snes->dm; 55096c699258SBarry Smith PetscFunctionReturn(0); 55106c699258SBarry Smith } 55110807856dSBarry Smith 551231823bd8SMatthew G Knepley /*@ 5513be95d8f1SBarry Smith SNESSetNPC - Sets the nonlinear preconditioner to be used. 551431823bd8SMatthew G Knepley 551531823bd8SMatthew G Knepley Collective on SNES 551631823bd8SMatthew G Knepley 551731823bd8SMatthew G Knepley Input Parameters: 551831823bd8SMatthew G Knepley + snes - iterative context obtained from SNESCreate() 551931823bd8SMatthew G Knepley - pc - the preconditioner object 552031823bd8SMatthew G Knepley 552131823bd8SMatthew G Knepley Notes: 5522be95d8f1SBarry Smith Use SNESGetNPC() to retrieve the preconditioner context (for example, 552331823bd8SMatthew G Knepley to configure it using the API). 552431823bd8SMatthew G Knepley 552531823bd8SMatthew G Knepley Level: developer 552631823bd8SMatthew G Knepley 55273ad1a0b9SPatrick Farrell .seealso: SNESGetNPC(), SNESHasNPC() 552831823bd8SMatthew G Knepley @*/ 5529be95d8f1SBarry Smith PetscErrorCode SNESSetNPC(SNES snes, SNES pc) 553031823bd8SMatthew G Knepley { 553131823bd8SMatthew G Knepley PetscErrorCode ierr; 553231823bd8SMatthew G Knepley 553331823bd8SMatthew G Knepley PetscFunctionBegin; 553431823bd8SMatthew G Knepley PetscValidHeaderSpecific(snes, SNES_CLASSID, 1); 553531823bd8SMatthew G Knepley PetscValidHeaderSpecific(pc, SNES_CLASSID, 2); 553631823bd8SMatthew G Knepley PetscCheckSameComm(snes, 1, pc, 2); 553731823bd8SMatthew G Knepley ierr = PetscObjectReference((PetscObject) pc);CHKERRQ(ierr); 5538efd4aadfSBarry Smith ierr = SNESDestroy(&snes->npc);CHKERRQ(ierr); 5539efd4aadfSBarry Smith snes->npc = pc; 5540efd4aadfSBarry Smith ierr = PetscLogObjectParent((PetscObject)snes, (PetscObject)snes->npc);CHKERRQ(ierr); 554131823bd8SMatthew G Knepley PetscFunctionReturn(0); 554231823bd8SMatthew G Knepley } 554331823bd8SMatthew G Knepley 554431823bd8SMatthew G Knepley /*@ 5545be95d8f1SBarry Smith SNESGetNPC - Creates a nonlinear preconditioning solver (SNES) to be used to precondition the nonlinear solver. 554631823bd8SMatthew G Knepley 5547951fe5abSBarry Smith Not Collective; but any changes to the obtained SNES object must be applied collectively 554831823bd8SMatthew G Knepley 554931823bd8SMatthew G Knepley Input Parameter: 555031823bd8SMatthew G Knepley . snes - iterative context obtained from SNESCreate() 555131823bd8SMatthew G Knepley 555231823bd8SMatthew G Knepley Output Parameter: 555331823bd8SMatthew G Knepley . pc - preconditioner context 555431823bd8SMatthew G Knepley 5555b5badacbSBarry Smith Options Database: 5556b5badacbSBarry Smith . -npc_snes_type <type> - set the type of the SNES to use as the nonlinear preconditioner 5557b5badacbSBarry Smith 555895452b02SPatrick Sanan Notes: 5559b5badacbSBarry Smith If a SNES was previously set with SNESSetNPC() then that SNES is returned, otherwise a new SNES object is created. 5560be95d8f1SBarry Smith 5561951fe5abSBarry Smith The (preconditioner) SNES returned automatically inherits the same nonlinear function and Jacobian supplied to the original 5562951fe5abSBarry Smith SNES during SNESSetUp() 5563951fe5abSBarry Smith 556431823bd8SMatthew G Knepley Level: developer 556531823bd8SMatthew G Knepley 5566951fe5abSBarry Smith .seealso: SNESSetNPC(), SNESHasNPC(), SNES, SNESCreate() 556731823bd8SMatthew G Knepley @*/ 5568be95d8f1SBarry Smith PetscErrorCode SNESGetNPC(SNES snes, SNES *pc) 556931823bd8SMatthew G Knepley { 557031823bd8SMatthew G Knepley PetscErrorCode ierr; 5571a64e098fSPeter Brune const char *optionsprefix; 557231823bd8SMatthew G Knepley 557331823bd8SMatthew G Knepley PetscFunctionBegin; 557431823bd8SMatthew G Knepley PetscValidHeaderSpecific(snes, SNES_CLASSID, 1); 557531823bd8SMatthew G Knepley PetscValidPointer(pc, 2); 5576efd4aadfSBarry Smith if (!snes->npc) { 5577efd4aadfSBarry Smith ierr = SNESCreate(PetscObjectComm((PetscObject)snes),&snes->npc);CHKERRQ(ierr); 5578efd4aadfSBarry Smith ierr = PetscObjectIncrementTabLevel((PetscObject)snes->npc,(PetscObject)snes,1);CHKERRQ(ierr); 5579efd4aadfSBarry Smith ierr = PetscLogObjectParent((PetscObject)snes,(PetscObject)snes->npc);CHKERRQ(ierr); 5580a64e098fSPeter Brune ierr = SNESGetOptionsPrefix(snes,&optionsprefix);CHKERRQ(ierr); 5581efd4aadfSBarry Smith ierr = SNESSetOptionsPrefix(snes->npc,optionsprefix);CHKERRQ(ierr); 5582efd4aadfSBarry Smith ierr = SNESAppendOptionsPrefix(snes->npc,"npc_");CHKERRQ(ierr); 5583efd4aadfSBarry Smith ierr = SNESSetCountersReset(snes->npc,PETSC_FALSE);CHKERRQ(ierr); 558431823bd8SMatthew G Knepley } 5585efd4aadfSBarry Smith *pc = snes->npc; 558631823bd8SMatthew G Knepley PetscFunctionReturn(0); 558731823bd8SMatthew G Knepley } 558831823bd8SMatthew G Knepley 55893ad1a0b9SPatrick Farrell /*@ 55903ad1a0b9SPatrick Farrell SNESHasNPC - Returns whether a nonlinear preconditioner exists 55913ad1a0b9SPatrick Farrell 55923ad1a0b9SPatrick Farrell Not Collective 55933ad1a0b9SPatrick Farrell 55943ad1a0b9SPatrick Farrell Input Parameter: 55953ad1a0b9SPatrick Farrell . snes - iterative context obtained from SNESCreate() 55963ad1a0b9SPatrick Farrell 55973ad1a0b9SPatrick Farrell Output Parameter: 55983ad1a0b9SPatrick Farrell . has_npc - whether the SNES has an NPC or not 55993ad1a0b9SPatrick Farrell 56003ad1a0b9SPatrick Farrell Level: developer 56013ad1a0b9SPatrick Farrell 56023ad1a0b9SPatrick Farrell .seealso: SNESSetNPC(), SNESGetNPC() 56033ad1a0b9SPatrick Farrell @*/ 56043ad1a0b9SPatrick Farrell PetscErrorCode SNESHasNPC(SNES snes, PetscBool *has_npc) 56053ad1a0b9SPatrick Farrell { 56063ad1a0b9SPatrick Farrell PetscFunctionBegin; 56073ad1a0b9SPatrick Farrell PetscValidHeaderSpecific(snes, SNES_CLASSID, 1); 5608efd4aadfSBarry Smith *has_npc = (PetscBool) (snes->npc ? PETSC_TRUE : PETSC_FALSE); 56093ad1a0b9SPatrick Farrell PetscFunctionReturn(0); 56103ad1a0b9SPatrick Farrell } 56113ad1a0b9SPatrick Farrell 5612c40d0f55SPeter Brune /*@ 5613be95d8f1SBarry Smith SNESSetNPCSide - Sets the preconditioning side. 5614c40d0f55SPeter Brune 5615c40d0f55SPeter Brune Logically Collective on SNES 5616c40d0f55SPeter Brune 5617c40d0f55SPeter Brune Input Parameter: 5618c40d0f55SPeter Brune . snes - iterative context obtained from SNESCreate() 5619c40d0f55SPeter Brune 5620c40d0f55SPeter Brune Output Parameter: 5621c40d0f55SPeter Brune . side - the preconditioning side, where side is one of 5622c40d0f55SPeter Brune .vb 56232d547940SBarry Smith PC_LEFT - left preconditioning 56242d547940SBarry Smith PC_RIGHT - right preconditioning (default for most nonlinear solvers) 5625c40d0f55SPeter Brune .ve 5626c40d0f55SPeter Brune 5627c40d0f55SPeter Brune Options Database Keys: 5628c40d0f55SPeter Brune . -snes_pc_side <right,left> 5629c40d0f55SPeter Brune 563095452b02SPatrick Sanan Notes: 563195452b02SPatrick Sanan SNESNRICHARDSON and SNESNCG only support left preconditioning. 56322d547940SBarry Smith 5633c40d0f55SPeter Brune Level: intermediate 5634c40d0f55SPeter Brune 5635be95d8f1SBarry Smith .seealso: SNESGetNPCSide(), KSPSetPCSide() 5636c40d0f55SPeter Brune @*/ 5637be95d8f1SBarry Smith PetscErrorCode SNESSetNPCSide(SNES snes,PCSide side) 5638c40d0f55SPeter Brune { 5639c40d0f55SPeter Brune PetscFunctionBegin; 5640c40d0f55SPeter Brune PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 5641c40d0f55SPeter Brune PetscValidLogicalCollectiveEnum(snes,side,2); 5642efd4aadfSBarry Smith snes->npcside= side; 5643c40d0f55SPeter Brune PetscFunctionReturn(0); 5644c40d0f55SPeter Brune } 5645c40d0f55SPeter Brune 5646c40d0f55SPeter Brune /*@ 5647be95d8f1SBarry Smith SNESGetNPCSide - Gets the preconditioning side. 5648c40d0f55SPeter Brune 5649c40d0f55SPeter Brune Not Collective 5650c40d0f55SPeter Brune 5651c40d0f55SPeter Brune Input Parameter: 5652c40d0f55SPeter Brune . snes - iterative context obtained from SNESCreate() 5653c40d0f55SPeter Brune 5654c40d0f55SPeter Brune Output Parameter: 5655c40d0f55SPeter Brune . side - the preconditioning side, where side is one of 5656c40d0f55SPeter Brune .vb 56572d547940SBarry Smith PC_LEFT - left preconditioning 56582d547940SBarry Smith PC_RIGHT - right preconditioning (default for most nonlinear solvers) 5659c40d0f55SPeter Brune .ve 5660c40d0f55SPeter Brune 5661c40d0f55SPeter Brune Level: intermediate 5662c40d0f55SPeter Brune 5663be95d8f1SBarry Smith .seealso: SNESSetNPCSide(), KSPGetPCSide() 5664c40d0f55SPeter Brune @*/ 5665be95d8f1SBarry Smith PetscErrorCode SNESGetNPCSide(SNES snes,PCSide *side) 5666c40d0f55SPeter Brune { 5667c40d0f55SPeter Brune PetscFunctionBegin; 5668c40d0f55SPeter Brune PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 5669c40d0f55SPeter Brune PetscValidPointer(side,2); 5670efd4aadfSBarry Smith *side = snes->npcside; 5671c40d0f55SPeter Brune PetscFunctionReturn(0); 5672c40d0f55SPeter Brune } 5673c40d0f55SPeter Brune 56749e764e56SPeter Brune /*@ 56757601faf0SJed Brown SNESSetLineSearch - Sets the linesearch on the SNES instance. 56769e764e56SPeter Brune 56779e764e56SPeter Brune Collective on SNES 56789e764e56SPeter Brune 56799e764e56SPeter Brune Input Parameters: 56809e764e56SPeter Brune + snes - iterative context obtained from SNESCreate() 56819e764e56SPeter Brune - linesearch - the linesearch object 56829e764e56SPeter Brune 56839e764e56SPeter Brune Notes: 56847601faf0SJed Brown Use SNESGetLineSearch() to retrieve the preconditioner context (for example, 56859e764e56SPeter Brune to configure it using the API). 56869e764e56SPeter Brune 56879e764e56SPeter Brune Level: developer 56889e764e56SPeter Brune 56897601faf0SJed Brown .seealso: SNESGetLineSearch() 56909e764e56SPeter Brune @*/ 56917601faf0SJed Brown PetscErrorCode SNESSetLineSearch(SNES snes, SNESLineSearch linesearch) 56929e764e56SPeter Brune { 56939e764e56SPeter Brune PetscErrorCode ierr; 56949e764e56SPeter Brune 56959e764e56SPeter Brune PetscFunctionBegin; 56969e764e56SPeter Brune PetscValidHeaderSpecific(snes, SNES_CLASSID, 1); 5697f1c6b773SPeter Brune PetscValidHeaderSpecific(linesearch, SNESLINESEARCH_CLASSID, 2); 56989e764e56SPeter Brune PetscCheckSameComm(snes, 1, linesearch, 2); 56999e764e56SPeter Brune ierr = PetscObjectReference((PetscObject) linesearch);CHKERRQ(ierr); 5700f1c6b773SPeter Brune ierr = SNESLineSearchDestroy(&snes->linesearch);CHKERRQ(ierr); 5701f5af7f23SKarl Rupp 57029e764e56SPeter Brune snes->linesearch = linesearch; 5703f5af7f23SKarl Rupp 57043bb1ff40SBarry Smith ierr = PetscLogObjectParent((PetscObject)snes, (PetscObject)snes->linesearch);CHKERRQ(ierr); 57059e764e56SPeter Brune PetscFunctionReturn(0); 57069e764e56SPeter Brune } 57079e764e56SPeter Brune 5708a34ceb2aSJed Brown /*@ 57097601faf0SJed Brown SNESGetLineSearch - Returns a pointer to the line search context set with SNESSetLineSearch() 57108141a3b9SPeter Brune or creates a default line search instance associated with the SNES and returns it. 57119e764e56SPeter Brune 57129e764e56SPeter Brune Not Collective 57139e764e56SPeter Brune 57149e764e56SPeter Brune Input Parameter: 57159e764e56SPeter Brune . snes - iterative context obtained from SNESCreate() 57169e764e56SPeter Brune 57179e764e56SPeter Brune Output Parameter: 57189e764e56SPeter Brune . linesearch - linesearch context 57199e764e56SPeter Brune 5720162e0bf5SPeter Brune Level: beginner 57219e764e56SPeter Brune 5722162e0bf5SPeter Brune .seealso: SNESSetLineSearch(), SNESLineSearchCreate() 57239e764e56SPeter Brune @*/ 57247601faf0SJed Brown PetscErrorCode SNESGetLineSearch(SNES snes, SNESLineSearch *linesearch) 57259e764e56SPeter Brune { 57269e764e56SPeter Brune PetscErrorCode ierr; 57279e764e56SPeter Brune const char *optionsprefix; 57289e764e56SPeter Brune 57299e764e56SPeter Brune PetscFunctionBegin; 57309e764e56SPeter Brune PetscValidHeaderSpecific(snes, SNES_CLASSID, 1); 57319e764e56SPeter Brune PetscValidPointer(linesearch, 2); 57329e764e56SPeter Brune if (!snes->linesearch) { 57339e764e56SPeter Brune ierr = SNESGetOptionsPrefix(snes, &optionsprefix);CHKERRQ(ierr); 573482f516ccSBarry Smith ierr = SNESLineSearchCreate(PetscObjectComm((PetscObject)snes), &snes->linesearch);CHKERRQ(ierr); 5735f1c6b773SPeter Brune ierr = SNESLineSearchSetSNES(snes->linesearch, snes);CHKERRQ(ierr); 5736b1dca40bSPeter Brune ierr = SNESLineSearchAppendOptionsPrefix(snes->linesearch, optionsprefix);CHKERRQ(ierr); 57379e764e56SPeter Brune ierr = PetscObjectIncrementTabLevel((PetscObject) snes->linesearch, (PetscObject) snes, 1);CHKERRQ(ierr); 57383bb1ff40SBarry Smith ierr = PetscLogObjectParent((PetscObject)snes, (PetscObject)snes->linesearch);CHKERRQ(ierr); 57399e764e56SPeter Brune } 57409e764e56SPeter Brune *linesearch = snes->linesearch; 57419e764e56SPeter Brune PetscFunctionReturn(0); 57429e764e56SPeter Brune } 5743