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: 2567b8a455SSatish Balay . -snes_error_if_not_converged <true,false> - cause an immediate error condition and stop the program if the solver does not converge 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 33db781477SPatrick Sanan .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 57db781477SPatrick Sanan .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 79db781477SPatrick Sanan .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 102db781477SPatrick Sanan .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 114f0b84518SBarry Smith in the functions domain. For example, a step with 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 123f0b84518SBarry Smith Note: 124f0b84518SBarry Smith You can direct `SNES` to avoid certain steps by using `SNESVISetVariableBounds()`, `SNESVISetComputeVariableBounds()` or 125f0b84518SBarry Smith `SNESLineSearchSetPreCheck()`, `SNESLineSearchSetPostCheck()` 126f0b84518SBarry Smith 127f0b84518SBarry Smith .seealso: `SNESCreate()`, `SNESSetFunction()`, `SNESFunction`, `SNESSetJacobianDomainError()`, `SNESVISetVariableBounds()`, 128f0b84518SBarry Smith `SNESVISetComputeVariableBounds()`, `SNESLineSearchSetPreCheck()`, `SNESLineSearchSetPostCheck()` 1294936397dSBarry Smith @*/ 1307087cfbeSBarry Smith PetscErrorCode SNESSetFunctionDomainError(SNES snes) 1314936397dSBarry Smith { 1324936397dSBarry Smith PetscFunctionBegin; 1330700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 1345f80ce2aSJacob Faibussowitsch PetscCheck(!snes->errorifnotconverged,PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"User code indicates input vector is not in the function domain"); 1354936397dSBarry Smith snes->domainerror = PETSC_TRUE; 1364936397dSBarry Smith PetscFunctionReturn(0); 1374936397dSBarry Smith } 1384936397dSBarry Smith 1396a388c36SPeter Brune /*@ 140f0b84518SBarry Smith SNESSetJacobianDomainError - tells SNES that computeJacobian does not make sense at the proposed step. For example there is a negative element transformation. 14107b62357SFande Kong 14207b62357SFande Kong Logically Collective on SNES 14307b62357SFande Kong 14407b62357SFande Kong Input Parameters: 14507b62357SFande Kong . snes - the SNES context 14607b62357SFande Kong 14707b62357SFande Kong Level: advanced 14807b62357SFande Kong 149f0b84518SBarry Smith Note: 150f0b84518SBarry Smith You can direct `SNES` to avoid certain steps by using `SNESVISetVariableBounds()`, `SNESVISetComputeVariableBounds()` or 151f0b84518SBarry Smith `SNESLineSearchSetPreCheck()`, `SNESLineSearchSetPostCheck()` 152f0b84518SBarry Smith 153f0b84518SBarry Smith .seealso: `SNESCreate()`, `SNESSetFunction()`, `SNESFunction()`, `SNESSetFunctionDomainError()`, `SNESVISetVariableBounds()`, 154f0b84518SBarry Smith `SNESVISetComputeVariableBounds()`, `SNESLineSearchSetPreCheck()`, `SNESLineSearchSetPostCheck()` 15507b62357SFande Kong @*/ 15607b62357SFande Kong PetscErrorCode SNESSetJacobianDomainError(SNES snes) 15707b62357SFande Kong { 15807b62357SFande Kong PetscFunctionBegin; 15907b62357SFande Kong PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 1605f80ce2aSJacob Faibussowitsch PetscCheck(!snes->errorifnotconverged,PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"User code indicates computeJacobian does not make sense"); 16107b62357SFande Kong snes->jacobiandomainerror = PETSC_TRUE; 16207b62357SFande Kong PetscFunctionReturn(0); 16307b62357SFande Kong } 16407b62357SFande Kong 16507b62357SFande Kong /*@ 166b351a90bSFande Kong SNESSetCheckJacobianDomainError - if or not to check jacobian domain error after each Jacobian evaluation. By default, we check Jacobian domain error 167b351a90bSFande Kong in the debug mode, and do not check it in the optimized mode. 168b351a90bSFande Kong 169b351a90bSFande Kong Logically Collective on SNES 170b351a90bSFande Kong 171b351a90bSFande Kong Input Parameters: 172a2b725a8SWilliam Gropp + snes - the SNES context 173a2b725a8SWilliam Gropp - flg - indicates if or not to check jacobian domain error after each Jacobian evaluation 174b351a90bSFande Kong 175b351a90bSFande Kong Level: advanced 176b351a90bSFande Kong 177db781477SPatrick Sanan .seealso: `SNESCreate()`, `SNESSetFunction()`, `SNESFunction()`, `SNESSetFunctionDomainError()`, `SNESGetCheckJacobianDomainError()` 178b351a90bSFande Kong @*/ 179b351a90bSFande Kong PetscErrorCode SNESSetCheckJacobianDomainError(SNES snes, PetscBool flg) 180b351a90bSFande Kong { 181b351a90bSFande Kong PetscFunctionBegin; 182b351a90bSFande Kong PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 183b351a90bSFande Kong snes->checkjacdomainerror = flg; 184b351a90bSFande Kong PetscFunctionReturn(0); 185b351a90bSFande Kong } 186b351a90bSFande Kong 187b351a90bSFande Kong /*@ 1888383d7d7SFande Kong SNESGetCheckJacobianDomainError - Get an indicator whether or not we are checking Jacobian domain errors after each Jacobian evaluation. 1898383d7d7SFande Kong 1908383d7d7SFande Kong Logically Collective on SNES 1918383d7d7SFande Kong 1928383d7d7SFande Kong Input Parameters: 1938383d7d7SFande Kong . snes - the SNES context 1948383d7d7SFande Kong 1958383d7d7SFande Kong Output Parameters: 1968383d7d7SFande Kong . flg - PETSC_FALSE indicates that we don't check jacobian domain errors after each Jacobian evaluation 1978383d7d7SFande Kong 1988383d7d7SFande Kong Level: advanced 1998383d7d7SFande Kong 200db781477SPatrick Sanan .seealso: `SNESCreate()`, `SNESSetFunction()`, `SNESFunction()`, `SNESSetFunctionDomainError()`, `SNESSetCheckJacobianDomainError()` 2018383d7d7SFande Kong @*/ 2028383d7d7SFande Kong PetscErrorCode SNESGetCheckJacobianDomainError(SNES snes, PetscBool *flg) 2038383d7d7SFande Kong { 2048383d7d7SFande Kong PetscFunctionBegin; 2058383d7d7SFande Kong PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 206534a8f05SLisandro Dalcin PetscValidBoolPointer(flg,2); 2078383d7d7SFande Kong *flg = snes->checkjacdomainerror; 2088383d7d7SFande Kong PetscFunctionReturn(0); 2098383d7d7SFande Kong } 2108383d7d7SFande Kong 2118383d7d7SFande Kong /*@ 212c77b2880SPeter Brune SNESGetFunctionDomainError - Gets the status of the domain error after a call to SNESComputeFunction; 2136a388c36SPeter Brune 2146a388c36SPeter Brune Logically Collective on SNES 2156a388c36SPeter Brune 2166a388c36SPeter Brune Input Parameters: 2176a388c36SPeter Brune . snes - the SNES context 2186a388c36SPeter Brune 2196a388c36SPeter Brune Output Parameters: 220bf388a1fSBarry Smith . domainerror - Set to PETSC_TRUE if there's a domain error; PETSC_FALSE otherwise. 2216a388c36SPeter Brune 2226a388c36SPeter Brune Level: advanced 2236a388c36SPeter Brune 224db781477SPatrick Sanan .seealso: `SNESSetFunctionDomainError()`, `SNESComputeFunction()` 2256a388c36SPeter Brune @*/ 2266a388c36SPeter Brune PetscErrorCode SNESGetFunctionDomainError(SNES snes, PetscBool *domainerror) 2276a388c36SPeter Brune { 2286a388c36SPeter Brune PetscFunctionBegin; 2296a388c36SPeter Brune PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 230534a8f05SLisandro Dalcin PetscValidBoolPointer(domainerror,2); 2316a388c36SPeter Brune *domainerror = snes->domainerror; 2326a388c36SPeter Brune PetscFunctionReturn(0); 2336a388c36SPeter Brune } 2346a388c36SPeter Brune 23507b62357SFande Kong /*@ 23607b62357SFande Kong SNESGetJacobianDomainError - Gets the status of the Jacobian domain error after a call to SNESComputeJacobian; 23707b62357SFande Kong 23807b62357SFande Kong Logically Collective on SNES 23907b62357SFande Kong 24007b62357SFande Kong Input Parameters: 24107b62357SFande Kong . snes - the SNES context 24207b62357SFande Kong 24307b62357SFande Kong Output Parameters: 24407b62357SFande Kong . domainerror - Set to PETSC_TRUE if there's a jacobian domain error; PETSC_FALSE otherwise. 24507b62357SFande Kong 24607b62357SFande Kong Level: advanced 24707b62357SFande Kong 248c2e3fba1SPatrick Sanan .seealso: `SNESSetFunctionDomainError()`, `SNESComputeFunction()`, `SNESGetFunctionDomainError()` 24907b62357SFande Kong @*/ 25007b62357SFande Kong PetscErrorCode SNESGetJacobianDomainError(SNES snes, PetscBool *domainerror) 25107b62357SFande Kong { 25207b62357SFande Kong PetscFunctionBegin; 25307b62357SFande Kong PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 254534a8f05SLisandro Dalcin PetscValidBoolPointer(domainerror,2); 25507b62357SFande Kong *domainerror = snes->jacobiandomainerror; 25607b62357SFande Kong PetscFunctionReturn(0); 25707b62357SFande Kong } 25807b62357SFande Kong 25955849f57SBarry Smith /*@C 26055849f57SBarry Smith SNESLoad - Loads a SNES that has been stored in binary with SNESView(). 26155849f57SBarry Smith 26255849f57SBarry Smith Collective on PetscViewer 26355849f57SBarry Smith 26455849f57SBarry Smith Input Parameters: 26555849f57SBarry Smith + newdm - the newly loaded SNES, this needs to have been created with SNESCreate() or 26655849f57SBarry Smith some related function before a call to SNESLoad(). 26755849f57SBarry Smith - viewer - binary file viewer, obtained from PetscViewerBinaryOpen() 26855849f57SBarry Smith 26955849f57SBarry Smith Level: intermediate 27055849f57SBarry Smith 27155849f57SBarry Smith Notes: 27255849f57SBarry Smith The type is determined by the data in the file, any type set into the SNES before this call is ignored. 27355849f57SBarry Smith 27455849f57SBarry Smith Notes for advanced users: 27555849f57SBarry Smith Most users should not need to know the details of the binary storage 27655849f57SBarry Smith format, since SNESLoad() and TSView() completely hide these details. 27755849f57SBarry Smith But for anyone who's interested, the standard binary matrix storage 27855849f57SBarry Smith format is 27955849f57SBarry Smith .vb 28055849f57SBarry Smith has not yet been determined 28155849f57SBarry Smith .ve 28255849f57SBarry Smith 283db781477SPatrick Sanan .seealso: `PetscViewerBinaryOpen()`, `SNESView()`, `MatLoad()`, `VecLoad()` 28455849f57SBarry Smith @*/ 2852d53ad75SBarry Smith PetscErrorCode SNESLoad(SNES snes, PetscViewer viewer) 28655849f57SBarry Smith { 28755849f57SBarry Smith PetscBool isbinary; 288060da220SMatthew G. Knepley PetscInt classid; 28955849f57SBarry Smith char type[256]; 29055849f57SBarry Smith KSP ksp; 2912d53ad75SBarry Smith DM dm; 2922d53ad75SBarry Smith DMSNES dmsnes; 29355849f57SBarry Smith 29455849f57SBarry Smith PetscFunctionBegin; 2952d53ad75SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 29655849f57SBarry Smith PetscValidHeaderSpecific(viewer,PETSC_VIEWER_CLASSID,2); 2979566063dSJacob Faibussowitsch PetscCall(PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERBINARY,&isbinary)); 2985f80ce2aSJacob Faibussowitsch PetscCheck(isbinary,PETSC_COMM_SELF,PETSC_ERR_ARG_WRONG,"Invalid viewer; open viewer with PetscViewerBinaryOpen()"); 29955849f57SBarry Smith 3009566063dSJacob Faibussowitsch PetscCall(PetscViewerBinaryRead(viewer,&classid,1,NULL,PETSC_INT)); 3015f80ce2aSJacob Faibussowitsch PetscCheck(classid == SNES_FILE_CLASSID,PetscObjectComm((PetscObject)snes),PETSC_ERR_ARG_WRONG,"Not SNES next in file"); 3029566063dSJacob Faibussowitsch PetscCall(PetscViewerBinaryRead(viewer,type,256,NULL,PETSC_CHAR)); 3039566063dSJacob Faibussowitsch PetscCall(SNESSetType(snes, type)); 304*dbbe0bcdSBarry Smith PetscTryTypeMethod(snes,load,viewer); 3059566063dSJacob Faibussowitsch PetscCall(SNESGetDM(snes,&dm)); 3069566063dSJacob Faibussowitsch PetscCall(DMGetDMSNES(dm,&dmsnes)); 3079566063dSJacob Faibussowitsch PetscCall(DMSNESLoad(dmsnes,viewer)); 3089566063dSJacob Faibussowitsch PetscCall(SNESGetKSP(snes,&ksp)); 3099566063dSJacob Faibussowitsch PetscCall(KSPLoad(ksp,viewer)); 31055849f57SBarry Smith PetscFunctionReturn(0); 31155849f57SBarry Smith } 3126a388c36SPeter Brune 3139804daf3SBarry Smith #include <petscdraw.h> 314e04113cfSBarry Smith #if defined(PETSC_HAVE_SAWS) 315e04113cfSBarry Smith #include <petscviewersaws.h> 316bfb97211SBarry Smith #endif 3178404b7f3SBarry Smith 318fe2efc57SMark /*@C 319fe2efc57SMark SNESViewFromOptions - View from Options 320fe2efc57SMark 321fe2efc57SMark Collective on SNES 322fe2efc57SMark 323fe2efc57SMark Input Parameters: 324fe2efc57SMark + A - the application ordering context 325736c3998SJose E. Roman . obj - Optional object 326736c3998SJose E. Roman - name - command line option 327fe2efc57SMark 328fe2efc57SMark Level: intermediate 329db781477SPatrick Sanan .seealso: `SNES`, `SNESView`, `PetscObjectViewFromOptions()`, `SNESCreate()` 330fe2efc57SMark @*/ 331fe2efc57SMark PetscErrorCode SNESViewFromOptions(SNES A,PetscObject obj,const char name[]) 332fe2efc57SMark { 333fe2efc57SMark PetscFunctionBegin; 334fe2efc57SMark PetscValidHeaderSpecific(A,SNES_CLASSID,1); 3359566063dSJacob Faibussowitsch PetscCall(PetscObjectViewFromOptions((PetscObject)A,obj,name)); 336fe2efc57SMark PetscFunctionReturn(0); 337fe2efc57SMark } 338fe2efc57SMark 339789d8953SBarry Smith PETSC_EXTERN PetscErrorCode SNESComputeJacobian_DMDA(SNES,Vec,Mat,Mat,void*); 340789d8953SBarry Smith 3417e2c5f70SBarry Smith /*@C 3429b94acceSBarry Smith SNESView - Prints the SNES data structure. 3439b94acceSBarry Smith 3444c49b128SBarry Smith Collective on SNES 345fee21e36SBarry Smith 346c7afd0dbSLois Curfman McInnes Input Parameters: 347c7afd0dbSLois Curfman McInnes + SNES - the SNES context 348c7afd0dbSLois Curfman McInnes - viewer - visualization context 349c7afd0dbSLois Curfman McInnes 3509b94acceSBarry Smith Options Database Key: 351c8a8ba5cSLois Curfman McInnes . -snes_view - Calls SNESView() at end of SNESSolve() 3529b94acceSBarry Smith 3539b94acceSBarry Smith Notes: 3549b94acceSBarry Smith The available visualization contexts include 355b0a32e0cSBarry Smith + PETSC_VIEWER_STDOUT_SELF - standard output (default) 356b0a32e0cSBarry Smith - PETSC_VIEWER_STDOUT_WORLD - synchronized standard 357c8a8ba5cSLois Curfman McInnes output where only the first processor opens 358c8a8ba5cSLois Curfman McInnes the file. All other processors send their 359c8a8ba5cSLois Curfman McInnes data to the first processor to print. 3609b94acceSBarry Smith 361052bf0daSPierre Jolivet The available formats include 362052bf0daSPierre Jolivet + PETSC_VIEWER_DEFAULT - standard output (default) 363052bf0daSPierre Jolivet - PETSC_VIEWER_ASCII_INFO_DETAIL - more verbose output for SNESNASM 364052bf0daSPierre Jolivet 3653e081fefSLois Curfman McInnes The user can open an alternative visualization context with 366b0a32e0cSBarry Smith PetscViewerASCIIOpen() - output to a specified file. 3679b94acceSBarry Smith 368595c91d4SBarry Smith In the debugger you can do "call SNESView(snes,0)" to display the SNES solver. (The same holds for any PETSc object viewer). 369595c91d4SBarry Smith 37036851e7fSLois Curfman McInnes Level: beginner 37136851e7fSLois Curfman McInnes 372db781477SPatrick Sanan .seealso: `PetscViewerASCIIOpen()` 3739b94acceSBarry Smith @*/ 3747087cfbeSBarry Smith PetscErrorCode SNESView(SNES snes,PetscViewer viewer) 3759b94acceSBarry Smith { 376fa9f3622SBarry Smith SNESKSPEW *kctx; 37794b7f48cSBarry Smith KSP ksp; 3787f1410a3SPeter Brune SNESLineSearch linesearch; 37972a02f06SBarry Smith PetscBool iascii,isstring,isbinary,isdraw; 3802d53ad75SBarry Smith DMSNES dmsnes; 381e04113cfSBarry Smith #if defined(PETSC_HAVE_SAWS) 382536b137fSBarry Smith PetscBool issaws; 383bfb97211SBarry Smith #endif 3849b94acceSBarry Smith 3853a40ed3dSBarry Smith PetscFunctionBegin; 3860700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 3873050cee2SBarry Smith if (!viewer) { 3889566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIGetStdout(PetscObjectComm((PetscObject)snes),&viewer)); 3893050cee2SBarry Smith } 3900700a824SBarry Smith PetscValidHeaderSpecific(viewer,PETSC_VIEWER_CLASSID,2); 391c9780b6fSBarry Smith PetscCheckSameComm(snes,1,viewer,2); 39274679c65SBarry Smith 3939566063dSJacob Faibussowitsch PetscCall(PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERASCII,&iascii)); 3949566063dSJacob Faibussowitsch PetscCall(PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERSTRING,&isstring)); 3959566063dSJacob Faibussowitsch PetscCall(PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERBINARY,&isbinary)); 3969566063dSJacob Faibussowitsch PetscCall(PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERDRAW,&isdraw)); 397e04113cfSBarry Smith #if defined(PETSC_HAVE_SAWS) 3989566063dSJacob Faibussowitsch PetscCall(PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERSAWS,&issaws)); 399bfb97211SBarry Smith #endif 40032077d6dSBarry Smith if (iascii) { 401dc0571f2SMatthew G. Knepley SNESNormSchedule normschedule; 4028404b7f3SBarry Smith DM dm; 4038404b7f3SBarry Smith PetscErrorCode (*cJ)(SNES,Vec,Mat,Mat,void*); 4048404b7f3SBarry Smith void *ctx; 405789d8953SBarry Smith const char *pre = ""; 406dc0571f2SMatthew G. Knepley 4079566063dSJacob Faibussowitsch PetscCall(PetscObjectPrintClassNamePrefixType((PetscObject)snes,viewer)); 408fce1e034SJed Brown if (!snes->setupcalled) { 4099566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer," SNES has not been set up so information may be incomplete\n")); 410fce1e034SJed Brown } 411e7788613SBarry Smith if (snes->ops->view) { 4129566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPushTab(viewer)); 413*dbbe0bcdSBarry Smith PetscUseTypeMethod(snes,view ,viewer); 4149566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPopTab(viewer)); 4150ef38995SBarry Smith } 41663a3b9bcSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer," maximum iterations=%" PetscInt_FMT ", maximum function evaluations=%" PetscInt_FMT "\n",snes->max_its,snes->max_funcs)); 4179566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer," tolerances: relative=%g, absolute=%g, solution=%g\n",(double)snes->rtol,(double)snes->abstol,(double)snes->stol)); 418efd4aadfSBarry Smith if (snes->usesksp) { 41963a3b9bcSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer," total number of linear solver iterations=%" PetscInt_FMT "\n",snes->linear_its)); 420efd4aadfSBarry Smith } 42163a3b9bcSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer," total number of function evaluations=%" PetscInt_FMT "\n",snes->nfuncs)); 4229566063dSJacob Faibussowitsch PetscCall(SNESGetNormSchedule(snes, &normschedule)); 4239566063dSJacob Faibussowitsch if (normschedule > 0) PetscCall(PetscViewerASCIIPrintf(viewer," norm schedule %s\n",SNESNormSchedules[normschedule])); 42417fe4bdfSPeter Brune if (snes->gridsequence) { 42563a3b9bcSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer," total number of grid sequence refinements=%" PetscInt_FMT "\n",snes->gridsequence)); 42617fe4bdfSPeter Brune } 4279b94acceSBarry Smith if (snes->ksp_ewconv) { 428fa9f3622SBarry Smith kctx = (SNESKSPEW*)snes->kspconvctx; 4299b94acceSBarry Smith if (kctx) { 43063a3b9bcSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer," Eisenstat-Walker computation of KSP relative tolerance (version %" PetscInt_FMT ")\n",kctx->version)); 4319566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer," rtol_0=%g, rtol_max=%g, threshold=%g\n",(double)kctx->rtol_0,(double)kctx->rtol_max,(double)kctx->threshold)); 4329566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer," gamma=%g, alpha=%g, alpha2=%g\n",(double)kctx->gamma,(double)kctx->alpha,(double)kctx->alpha2)); 4339b94acceSBarry Smith } 4349b94acceSBarry Smith } 435eb1f6c34SBarry Smith if (snes->lagpreconditioner == -1) { 4369566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer," Preconditioned is never rebuilt\n")); 437eb1f6c34SBarry Smith } else if (snes->lagpreconditioner > 1) { 43863a3b9bcSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer," Preconditioned is rebuilt every %" PetscInt_FMT " new Jacobians\n",snes->lagpreconditioner)); 439eb1f6c34SBarry Smith } 440eb1f6c34SBarry Smith if (snes->lagjacobian == -1) { 4419566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer," Jacobian is never rebuilt\n")); 442eb1f6c34SBarry Smith } else if (snes->lagjacobian > 1) { 44363a3b9bcSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer," Jacobian is rebuilt every %" PetscInt_FMT " SNES iterations\n",snes->lagjacobian)); 444eb1f6c34SBarry Smith } 4459566063dSJacob Faibussowitsch PetscCall(SNESGetDM(snes,&dm)); 4469566063dSJacob Faibussowitsch PetscCall(DMSNESGetJacobian(dm,&cJ,&ctx)); 447789d8953SBarry Smith if (snes->mf_operator) { 4489566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer," Jacobian is applied matrix-free with differencing\n")); 449789d8953SBarry Smith pre = "Preconditioning "; 450789d8953SBarry Smith } 4518404b7f3SBarry Smith if (cJ == SNESComputeJacobianDefault) { 4529566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer," %sJacobian is built using finite differences one column at a time\n",pre)); 4538404b7f3SBarry Smith } else if (cJ == SNESComputeJacobianDefaultColor) { 4549566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer," %sJacobian is built using finite differences with coloring\n",pre)); 455789d8953SBarry Smith /* it slightly breaks data encapsulation for access the DMDA information directly */ 456789d8953SBarry Smith } else if (cJ == SNESComputeJacobian_DMDA) { 457789d8953SBarry Smith MatFDColoring fdcoloring; 4589566063dSJacob Faibussowitsch PetscCall(PetscObjectQuery((PetscObject)dm,"DMDASNES_FDCOLORING",(PetscObject*)&fdcoloring)); 459789d8953SBarry Smith if (fdcoloring) { 4609566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer," %sJacobian is built using colored finite differences on a DMDA\n",pre)); 461789d8953SBarry Smith } else { 4629566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer," %sJacobian is built using a DMDA local Jacobian\n",pre)); 463789d8953SBarry Smith } 464789d8953SBarry Smith } else if (snes->mf) { 4659566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer," Jacobian is applied matrix-free with differencing, no explicit Jacobian\n")); 4668404b7f3SBarry Smith } 4670f5bd95cSBarry Smith } else if (isstring) { 468317d6ea6SBarry Smith const char *type; 4699566063dSJacob Faibussowitsch PetscCall(SNESGetType(snes,&type)); 4709566063dSJacob Faibussowitsch PetscCall(PetscViewerStringSPrintf(viewer," SNESType: %-7.7s",type)); 471*dbbe0bcdSBarry Smith PetscTryTypeMethod(snes,view,viewer); 47255849f57SBarry Smith } else if (isbinary) { 47355849f57SBarry Smith PetscInt classid = SNES_FILE_CLASSID; 47455849f57SBarry Smith MPI_Comm comm; 47555849f57SBarry Smith PetscMPIInt rank; 47655849f57SBarry Smith char type[256]; 47755849f57SBarry Smith 4789566063dSJacob Faibussowitsch PetscCall(PetscObjectGetComm((PetscObject)snes,&comm)); 4799566063dSJacob Faibussowitsch PetscCallMPI(MPI_Comm_rank(comm,&rank)); 480dd400576SPatrick Sanan if (rank == 0) { 4819566063dSJacob Faibussowitsch PetscCall(PetscViewerBinaryWrite(viewer,&classid,1,PETSC_INT)); 4829566063dSJacob Faibussowitsch PetscCall(PetscStrncpy(type,((PetscObject)snes)->type_name,sizeof(type))); 4839566063dSJacob Faibussowitsch PetscCall(PetscViewerBinaryWrite(viewer,type,sizeof(type),PETSC_CHAR)); 48455849f57SBarry Smith } 485*dbbe0bcdSBarry Smith PetscTryTypeMethod(snes,view,viewer); 48672a02f06SBarry Smith } else if (isdraw) { 48772a02f06SBarry Smith PetscDraw draw; 48872a02f06SBarry Smith char str[36]; 48989fd9fafSBarry Smith PetscReal x,y,bottom,h; 49072a02f06SBarry Smith 4919566063dSJacob Faibussowitsch PetscCall(PetscViewerDrawGetDraw(viewer,0,&draw)); 4929566063dSJacob Faibussowitsch PetscCall(PetscDrawGetCurrentPoint(draw,&x,&y)); 4939566063dSJacob Faibussowitsch PetscCall(PetscStrncpy(str,"SNES: ",sizeof(str))); 4949566063dSJacob Faibussowitsch PetscCall(PetscStrlcat(str,((PetscObject)snes)->type_name,sizeof(str))); 4959566063dSJacob Faibussowitsch PetscCall(PetscDrawStringBoxed(draw,x,y,PETSC_DRAW_BLUE,PETSC_DRAW_BLACK,str,NULL,&h)); 49689fd9fafSBarry Smith bottom = y - h; 4979566063dSJacob Faibussowitsch PetscCall(PetscDrawPushCurrentPoint(draw,x,bottom)); 498*dbbe0bcdSBarry Smith PetscTryTypeMethod(snes,view,viewer); 499e04113cfSBarry Smith #if defined(PETSC_HAVE_SAWS) 500536b137fSBarry Smith } else if (issaws) { 501d45a07a7SBarry Smith PetscMPIInt rank; 5022657e9d9SBarry Smith const char *name; 503d45a07a7SBarry Smith 5049566063dSJacob Faibussowitsch PetscCall(PetscObjectGetName((PetscObject)snes,&name)); 5059566063dSJacob Faibussowitsch PetscCallMPI(MPI_Comm_rank(PETSC_COMM_WORLD,&rank)); 506dd400576SPatrick Sanan if (!((PetscObject)snes)->amsmem && rank == 0) { 507d45a07a7SBarry Smith char dir[1024]; 508d45a07a7SBarry Smith 5099566063dSJacob Faibussowitsch PetscCall(PetscObjectViewSAWs((PetscObject)snes,viewer)); 5109566063dSJacob Faibussowitsch PetscCall(PetscSNPrintf(dir,1024,"/PETSc/Objects/%s/its",name)); 511792fecdfSBarry Smith PetscCallSAWs(SAWs_Register,(dir,&snes->iter,1,SAWs_READ,SAWs_INT)); 512bfb97211SBarry Smith if (!snes->conv_hist) { 5139566063dSJacob Faibussowitsch PetscCall(SNESSetConvergenceHistory(snes,NULL,NULL,PETSC_DECIDE,PETSC_TRUE)); 514bfb97211SBarry Smith } 5159566063dSJacob Faibussowitsch PetscCall(PetscSNPrintf(dir,1024,"/PETSc/Objects/%s/conv_hist",name)); 516792fecdfSBarry Smith PetscCallSAWs(SAWs_Register,(dir,snes->conv_hist,10,SAWs_READ,SAWs_DOUBLE)); 517f05ece33SBarry Smith } 518bfb97211SBarry Smith #endif 51972a02f06SBarry Smith } 52072a02f06SBarry Smith if (snes->linesearch) { 5219566063dSJacob Faibussowitsch PetscCall(SNESGetLineSearch(snes, &linesearch)); 5229566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPushTab(viewer)); 5239566063dSJacob Faibussowitsch PetscCall(SNESLineSearchView(linesearch, viewer)); 5249566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPopTab(viewer)); 52519bcc07fSBarry Smith } 526efd4aadfSBarry Smith if (snes->npc && snes->usesnpc) { 5279566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPushTab(viewer)); 5289566063dSJacob Faibussowitsch PetscCall(SNESView(snes->npc, viewer)); 5299566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPopTab(viewer)); 5304a0c5b0cSMatthew G Knepley } 5319566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPushTab(viewer)); 5329566063dSJacob Faibussowitsch PetscCall(DMGetDMSNES(snes->dm,&dmsnes)); 5339566063dSJacob Faibussowitsch PetscCall(DMSNESView(dmsnes, viewer)); 5349566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPopTab(viewer)); 5352c155ee1SBarry Smith if (snes->usesksp) { 5369566063dSJacob Faibussowitsch PetscCall(SNESGetKSP(snes,&ksp)); 5379566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPushTab(viewer)); 5389566063dSJacob Faibussowitsch PetscCall(KSPView(ksp,viewer)); 5399566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPopTab(viewer)); 5402c155ee1SBarry Smith } 54172a02f06SBarry Smith if (isdraw) { 54272a02f06SBarry Smith PetscDraw draw; 5439566063dSJacob Faibussowitsch PetscCall(PetscViewerDrawGetDraw(viewer,0,&draw)); 5449566063dSJacob Faibussowitsch PetscCall(PetscDrawPopCurrentPoint(draw)); 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 567db781477SPatrick Sanan .seealso: `SNESSetFromOptions()` 56876b2cf59SMatthew Knepley @*/ 5697087cfbeSBarry Smith PetscErrorCode SNESAddOptionsChecker(PetscErrorCode (*snescheck)(SNES)) 57076b2cf59SMatthew Knepley { 57176b2cf59SMatthew Knepley PetscFunctionBegin; 57263a3b9bcSJacob Faibussowitsch PetscCheck(numberofsetfromoptions < MAXSETFROMOPTIONS,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; 582895c21f2SBarry Smith MatNullSpace nullsp; 583aa3661deSLisandro Dalcin 584aa3661deSLisandro Dalcin PetscFunctionBegin; 5850700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 586aa3661deSLisandro Dalcin 58798613b67SLisandro Dalcin if (!snes->vec_func && (snes->jacobian || snes->jacobian_pre)) { 58898613b67SLisandro Dalcin Mat A = snes->jacobian, B = snes->jacobian_pre; 5899566063dSJacob Faibussowitsch PetscCall(MatCreateVecs(A ? A : B, NULL,&snes->vec_func)); 59098613b67SLisandro Dalcin } 59198613b67SLisandro Dalcin 592aa3661deSLisandro Dalcin if (version == 1) { 5939566063dSJacob Faibussowitsch PetscCall(MatCreateSNESMF(snes,&J)); 5949566063dSJacob Faibussowitsch PetscCall(MatMFFDSetOptionsPrefix(J,((PetscObject)snes)->prefix)); 5959566063dSJacob Faibussowitsch PetscCall(MatSetFromOptions(J)); 5961e2ae407SBarry Smith /* TODO: the version 2 code should be merged into the MatCreateSNESMF() and MatCreateMFFD() infrastructure and then removed */ 597aa3661deSLisandro Dalcin } else if (version == 2) { 5985f80ce2aSJacob Faibussowitsch PetscCheck(snes->vec_func,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) 6009566063dSJacob Faibussowitsch PetscCall(SNESDefaultMatrixFreeCreate2(snes,snes->vec_func,&J)); 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) { 6089566063dSJacob Faibussowitsch PetscCall(MatGetNullSpace(snes->jacobian,&nullsp)); 6091baa6e33SBarry Smith if (nullsp) PetscCall(MatSetNullSpace(J,nullsp)); 610895c21f2SBarry Smith } 611895c21f2SBarry Smith 61263a3b9bcSJacob Faibussowitsch PetscCall(PetscInfo(snes,"Setting default matrix-free operator routines (version %" PetscInt_FMT ")\n", version)); 613d3462f78SMatthew Knepley if (hasOperator) { 6143232da50SPeter Brune 615aa3661deSLisandro Dalcin /* This version replaces the user provided Jacobian matrix with a 616aa3661deSLisandro Dalcin matrix-free version but still employs the user-provided preconditioner matrix. */ 6179566063dSJacob Faibussowitsch PetscCall(SNESSetJacobian(snes,J,NULL,NULL,NULL)); 618aa3661deSLisandro Dalcin } else { 619aa3661deSLisandro Dalcin /* This version replaces both the user-provided Jacobian and the user- 6203232da50SPeter Brune provided preconditioner Jacobian with the default matrix free version. */ 621b552625fSStefano Zampini if (snes->npcside == PC_LEFT && snes->npc) { 6229566063dSJacob Faibussowitsch if (!snes->jacobian) PetscCall(SNESSetJacobian(snes,J,NULL,NULL,NULL)); 623172a4300SPeter Brune } else { 624789d8953SBarry Smith KSP ksp; 625789d8953SBarry Smith PC pc; 626789d8953SBarry Smith PetscBool match; 627789d8953SBarry Smith 6289566063dSJacob Faibussowitsch PetscCall(SNESSetJacobian(snes,J,J,MatMFFDComputeJacobian,NULL)); 629aa3661deSLisandro Dalcin /* Force no preconditioner */ 6309566063dSJacob Faibussowitsch PetscCall(SNESGetKSP(snes,&ksp)); 6319566063dSJacob Faibussowitsch PetscCall(KSPGetPC(ksp,&pc)); 6322698c518SLisandro Dalcin PetscCall(PetscObjectTypeCompareAny((PetscObject)pc,&match,PCSHELL,PCH2OPUS,"")); 633aa3661deSLisandro Dalcin if (!match) { 6349566063dSJacob Faibussowitsch PetscCall(PetscInfo(snes,"Setting default matrix-free preconditioner routines\nThat is no preconditioner is being used\n")); 6359566063dSJacob Faibussowitsch PetscCall(PCSetType(pc,PCNONE)); 636aa3661deSLisandro Dalcin } 637aa3661deSLisandro Dalcin } 638789d8953SBarry Smith } 6399566063dSJacob Faibussowitsch PetscCall(MatDestroy(&J)); 640aa3661deSLisandro Dalcin PetscFunctionReturn(0); 641aa3661deSLisandro Dalcin } 642aa3661deSLisandro Dalcin 643dfe15315SJed Brown static PetscErrorCode DMRestrictHook_SNESVecSol(DM dmfine,Mat Restrict,Vec Rscale,Mat Inject,DM dmcoarse,void *ctx) 644dfe15315SJed Brown { 645dfe15315SJed Brown SNES snes = (SNES)ctx; 6460298fd71SBarry Smith Vec Xfine,Xfine_named = NULL,Xcoarse; 647dfe15315SJed Brown 648dfe15315SJed Brown PetscFunctionBegin; 64916ebb321SJed Brown if (PetscLogPrintInfo) { 65016ebb321SJed Brown PetscInt finelevel,coarselevel,fineclevel,coarseclevel; 6519566063dSJacob Faibussowitsch PetscCall(DMGetRefineLevel(dmfine,&finelevel)); 6529566063dSJacob Faibussowitsch PetscCall(DMGetCoarsenLevel(dmfine,&fineclevel)); 6539566063dSJacob Faibussowitsch PetscCall(DMGetRefineLevel(dmcoarse,&coarselevel)); 6549566063dSJacob Faibussowitsch PetscCall(DMGetCoarsenLevel(dmcoarse,&coarseclevel)); 65563a3b9bcSJacob Faibussowitsch PetscCall(PetscInfo(dmfine,"Restricting SNES solution vector from level %" PetscInt_FMT "-%" PetscInt_FMT " to level %" PetscInt_FMT "-%" PetscInt_FMT "\n",finelevel,fineclevel,coarselevel,coarseclevel)); 65616ebb321SJed Brown } 657dfe15315SJed Brown if (dmfine == snes->dm) Xfine = snes->vec_sol; 658dfe15315SJed Brown else { 6599566063dSJacob Faibussowitsch PetscCall(DMGetNamedGlobalVector(dmfine,"SNESVecSol",&Xfine_named)); 660dfe15315SJed Brown Xfine = Xfine_named; 661dfe15315SJed Brown } 6629566063dSJacob Faibussowitsch PetscCall(DMGetNamedGlobalVector(dmcoarse,"SNESVecSol",&Xcoarse)); 663907f5c5aSLawrence Mitchell if (Inject) { 6649566063dSJacob Faibussowitsch PetscCall(MatRestrict(Inject,Xfine,Xcoarse)); 665907f5c5aSLawrence Mitchell } else { 6669566063dSJacob Faibussowitsch PetscCall(MatRestrict(Restrict,Xfine,Xcoarse)); 6679566063dSJacob Faibussowitsch PetscCall(VecPointwiseMult(Xcoarse,Xcoarse,Rscale)); 668907f5c5aSLawrence Mitchell } 6699566063dSJacob Faibussowitsch PetscCall(DMRestoreNamedGlobalVector(dmcoarse,"SNESVecSol",&Xcoarse)); 6709566063dSJacob Faibussowitsch if (Xfine_named) PetscCall(DMRestoreNamedGlobalVector(dmfine,"SNESVecSol",&Xfine_named)); 671dfe15315SJed Brown PetscFunctionReturn(0); 672dfe15315SJed Brown } 673dfe15315SJed Brown 67416ebb321SJed Brown static PetscErrorCode DMCoarsenHook_SNESVecSol(DM dm,DM dmc,void *ctx) 67516ebb321SJed Brown { 67616ebb321SJed Brown PetscFunctionBegin; 6779566063dSJacob Faibussowitsch PetscCall(DMCoarsenHookAdd(dmc,DMCoarsenHook_SNESVecSol,DMRestrictHook_SNESVecSol,ctx)); 67816ebb321SJed Brown PetscFunctionReturn(0); 67916ebb321SJed Brown } 68016ebb321SJed Brown 681a6950cb2SJed Brown /* This may be called to rediscretize the operator on levels of linear multigrid. The DM shuffle is so the user can 682a6950cb2SJed Brown * safely call SNESGetDM() in their residual evaluation routine. */ 68323ee1639SBarry Smith static PetscErrorCode KSPComputeOperators_SNES(KSP ksp,Mat A,Mat B,void *ctx) 684caa4e7f2SJed Brown { 685caa4e7f2SJed Brown SNES snes = (SNES)ctx; 6860298fd71SBarry Smith Vec X,Xnamed = NULL; 687dfe15315SJed Brown DM dmsave; 6884e269d77SPeter Brune void *ctxsave; 68925ce1634SJed Brown PetscErrorCode (*jac)(SNES,Vec,Mat,Mat,void*) = NULL; 690caa4e7f2SJed Brown 691caa4e7f2SJed Brown PetscFunctionBegin; 692dfe15315SJed Brown dmsave = snes->dm; 6939566063dSJacob Faibussowitsch PetscCall(KSPGetDM(ksp,&snes->dm)); 694dfe15315SJed Brown if (dmsave == snes->dm) X = snes->vec_sol; /* We are on the finest level */ 695dfe15315SJed Brown else { /* We are on a coarser level, this vec was initialized using a DM restrict hook */ 6969566063dSJacob Faibussowitsch PetscCall(DMGetNamedGlobalVector(snes->dm,"SNESVecSol",&Xnamed)); 697dfe15315SJed Brown X = Xnamed; 6989566063dSJacob Faibussowitsch PetscCall(SNESGetJacobian(snes,NULL,NULL,&jac,&ctxsave)); 6994e269d77SPeter Brune /* If the DM's don't match up, the MatFDColoring context needed for the jacobian won't match up either -- fixit. */ 7008d359177SBarry Smith if (jac == SNESComputeJacobianDefaultColor) { 7019566063dSJacob Faibussowitsch PetscCall(SNESSetJacobian(snes,NULL,NULL,SNESComputeJacobianDefaultColor,NULL)); 702dfe15315SJed Brown } 7034e269d77SPeter Brune } 7044dde8bb0SMatthew G. Knepley /* Make sure KSP DM has the Jacobian computation routine */ 7054dde8bb0SMatthew G. Knepley { 7064dde8bb0SMatthew G. Knepley DMSNES sdm; 7074e269d77SPeter Brune 7089566063dSJacob Faibussowitsch PetscCall(DMGetDMSNES(snes->dm, &sdm)); 7094dde8bb0SMatthew G. Knepley if (!sdm->ops->computejacobian) { 7109566063dSJacob Faibussowitsch PetscCall(DMCopyDMSNES(dmsave, snes->dm)); 7114dde8bb0SMatthew G. Knepley } 7124dde8bb0SMatthew G. Knepley } 7132b93b426SMatthew G. Knepley /* Compute the operators */ 7149566063dSJacob Faibussowitsch PetscCall(SNESComputeJacobian(snes,X,A,B)); 7152b93b426SMatthew G. Knepley /* Put the previous context back */ 7168d359177SBarry Smith if (snes->dm != dmsave && jac == SNESComputeJacobianDefaultColor) { 7179566063dSJacob Faibussowitsch PetscCall(SNESSetJacobian(snes,NULL,NULL,jac,ctxsave)); 7184e269d77SPeter Brune } 7194e269d77SPeter Brune 7209566063dSJacob Faibussowitsch if (Xnamed) PetscCall(DMRestoreNamedGlobalVector(snes->dm,"SNESVecSol",&Xnamed)); 721dfe15315SJed Brown snes->dm = dmsave; 722caa4e7f2SJed Brown PetscFunctionReturn(0); 723caa4e7f2SJed Brown } 724caa4e7f2SJed Brown 7256cab3a1bSJed Brown /*@ 7266cab3a1bSJed Brown SNESSetUpMatrices - ensures that matrices are available for SNES, to be called by SNESSetUp_XXX() 7276cab3a1bSJed Brown 7286cab3a1bSJed Brown Collective 7296cab3a1bSJed Brown 7304165533cSJose E. Roman Input Parameter: 7316cab3a1bSJed Brown . snes - snes to configure 7326cab3a1bSJed Brown 7336cab3a1bSJed Brown Level: developer 7346cab3a1bSJed Brown 735db781477SPatrick Sanan .seealso: `SNESSetUp()` 7366cab3a1bSJed Brown @*/ 7376cab3a1bSJed Brown PetscErrorCode SNESSetUpMatrices(SNES snes) 7386cab3a1bSJed Brown { 7396cab3a1bSJed Brown DM dm; 740942e3340SBarry Smith DMSNES sdm; 7416cab3a1bSJed Brown 7426cab3a1bSJed Brown PetscFunctionBegin; 7439566063dSJacob Faibussowitsch PetscCall(SNESGetDM(snes,&dm)); 7449566063dSJacob Faibussowitsch PetscCall(DMGetDMSNES(dm,&sdm)); 74558b371f3SBarry Smith if (!snes->jacobian && snes->mf) { 7466cab3a1bSJed Brown Mat J; 7476cab3a1bSJed Brown void *functx; 7489566063dSJacob Faibussowitsch PetscCall(MatCreateSNESMF(snes,&J)); 7499566063dSJacob Faibussowitsch PetscCall(MatMFFDSetOptionsPrefix(J,((PetscObject)snes)->prefix)); 7509566063dSJacob Faibussowitsch PetscCall(MatSetFromOptions(J)); 7519566063dSJacob Faibussowitsch PetscCall(SNESGetFunction(snes,NULL,NULL,&functx)); 7529566063dSJacob Faibussowitsch PetscCall(SNESSetJacobian(snes,J,J,NULL,NULL)); 7539566063dSJacob Faibussowitsch PetscCall(MatDestroy(&J)); 754caa4e7f2SJed Brown } else if (snes->mf_operator && !snes->jacobian_pre && !snes->jacobian) { 7556cab3a1bSJed Brown Mat J,B; 7569566063dSJacob Faibussowitsch PetscCall(MatCreateSNESMF(snes,&J)); 7579566063dSJacob Faibussowitsch PetscCall(MatMFFDSetOptionsPrefix(J,((PetscObject)snes)->prefix)); 7589566063dSJacob Faibussowitsch PetscCall(MatSetFromOptions(J)); 7599566063dSJacob Faibussowitsch PetscCall(DMCreateMatrix(snes->dm,&B)); 76006f20277SJed Brown /* sdm->computejacobian was already set to reach here */ 7619566063dSJacob Faibussowitsch PetscCall(SNESSetJacobian(snes,J,B,NULL,NULL)); 7629566063dSJacob Faibussowitsch PetscCall(MatDestroy(&J)); 7639566063dSJacob Faibussowitsch PetscCall(MatDestroy(&B)); 764caa4e7f2SJed Brown } else if (!snes->jacobian_pre) { 7651ba9b98eSMatthew G. Knepley PetscDS prob; 7666cab3a1bSJed Brown Mat J, B; 7671ba9b98eSMatthew G. Knepley PetscBool hasPrec = PETSC_FALSE; 7681ba9b98eSMatthew G. Knepley 7696cab3a1bSJed Brown J = snes->jacobian; 7709566063dSJacob Faibussowitsch PetscCall(DMGetDS(dm, &prob)); 7719566063dSJacob Faibussowitsch if (prob) PetscCall(PetscDSHasJacobianPreconditioner(prob, &hasPrec)); 7729566063dSJacob Faibussowitsch if (J) PetscCall(PetscObjectReference((PetscObject) J)); 7739566063dSJacob Faibussowitsch else if (hasPrec) PetscCall(DMCreateMatrix(snes->dm, &J)); 7749566063dSJacob Faibussowitsch PetscCall(DMCreateMatrix(snes->dm, &B)); 7759566063dSJacob Faibussowitsch PetscCall(SNESSetJacobian(snes, J ? J : B, B, NULL, NULL)); 7769566063dSJacob Faibussowitsch PetscCall(MatDestroy(&J)); 7779566063dSJacob Faibussowitsch PetscCall(MatDestroy(&B)); 7786cab3a1bSJed Brown } 779caa4e7f2SJed Brown { 780caa4e7f2SJed Brown KSP ksp; 7819566063dSJacob Faibussowitsch PetscCall(SNESGetKSP(snes,&ksp)); 7829566063dSJacob Faibussowitsch PetscCall(KSPSetComputeOperators(ksp,KSPComputeOperators_SNES,snes)); 7839566063dSJacob Faibussowitsch PetscCall(DMCoarsenHookAdd(snes->dm,DMCoarsenHook_SNESVecSol,DMRestrictHook_SNESVecSol,snes)); 784caa4e7f2SJed Brown } 7856cab3a1bSJed Brown PetscFunctionReturn(0); 7866cab3a1bSJed Brown } 7876cab3a1bSJed Brown 7885e7c47f3SMatthew G. Knepley static PetscErrorCode SNESMonitorPauseFinal_Internal(SNES snes) 7895e7c47f3SMatthew G. Knepley { 7905e7c47f3SMatthew G. Knepley PetscInt i; 7915e7c47f3SMatthew G. Knepley 7925e7c47f3SMatthew G. Knepley PetscFunctionBegin; 7935e7c47f3SMatthew G. Knepley if (!snes->pauseFinal) PetscFunctionReturn(0); 7945e7c47f3SMatthew G. Knepley for (i = 0; i < snes->numbermonitors; ++i) { 7955e7c47f3SMatthew G. Knepley PetscViewerAndFormat *vf = (PetscViewerAndFormat *) snes->monitorcontext[i]; 7965e7c47f3SMatthew G. Knepley PetscDraw draw; 7975e7c47f3SMatthew G. Knepley PetscReal lpause; 7985e7c47f3SMatthew G. Knepley 7995e7c47f3SMatthew G. Knepley if (!vf) continue; 8005e7c47f3SMatthew G. Knepley if (vf->lg) { 8015e7c47f3SMatthew G. Knepley if (!PetscCheckPointer(vf->lg, PETSC_OBJECT)) continue; 8025e7c47f3SMatthew G. Knepley if (((PetscObject) vf->lg)->classid != PETSC_DRAWLG_CLASSID) continue; 8039566063dSJacob Faibussowitsch PetscCall(PetscDrawLGGetDraw(vf->lg, &draw)); 8049566063dSJacob Faibussowitsch PetscCall(PetscDrawGetPause(draw, &lpause)); 8059566063dSJacob Faibussowitsch PetscCall(PetscDrawSetPause(draw, -1.0)); 8069566063dSJacob Faibussowitsch PetscCall(PetscDrawPause(draw)); 8079566063dSJacob Faibussowitsch PetscCall(PetscDrawSetPause(draw, lpause)); 8085e7c47f3SMatthew G. Knepley } else { 8095e7c47f3SMatthew G. Knepley PetscBool isdraw; 8105e7c47f3SMatthew G. Knepley 8115e7c47f3SMatthew G. Knepley if (!PetscCheckPointer(vf->viewer, PETSC_OBJECT)) continue; 8125e7c47f3SMatthew G. Knepley if (((PetscObject) vf->viewer)->classid != PETSC_VIEWER_CLASSID) continue; 8139566063dSJacob Faibussowitsch PetscCall(PetscObjectTypeCompare((PetscObject) vf->viewer, PETSCVIEWERDRAW, &isdraw)); 8145e7c47f3SMatthew G. Knepley if (!isdraw) continue; 8159566063dSJacob Faibussowitsch PetscCall(PetscViewerDrawGetDraw(vf->viewer, 0, &draw)); 8169566063dSJacob Faibussowitsch PetscCall(PetscDrawGetPause(draw, &lpause)); 8179566063dSJacob Faibussowitsch PetscCall(PetscDrawSetPause(draw, -1.0)); 8189566063dSJacob Faibussowitsch PetscCall(PetscDrawPause(draw)); 8199566063dSJacob Faibussowitsch PetscCall(PetscDrawSetPause(draw, lpause)); 8205e7c47f3SMatthew G. Knepley } 8215e7c47f3SMatthew G. Knepley } 8225e7c47f3SMatthew G. Knepley PetscFunctionReturn(0); 8235e7c47f3SMatthew G. Knepley } 8245e7c47f3SMatthew G. Knepley 825fde5950dSBarry Smith /*@C 826fde5950dSBarry Smith SNESMonitorSetFromOptions - Sets a monitor function and viewer appropriate for the type indicated by the user 827fde5950dSBarry Smith 828fde5950dSBarry Smith Collective on SNES 829fde5950dSBarry Smith 830fde5950dSBarry Smith Input Parameters: 831fde5950dSBarry Smith + snes - SNES object you wish to monitor 832fde5950dSBarry Smith . name - the monitor type one is seeking 833fde5950dSBarry Smith . help - message indicating what monitoring is done 834fde5950dSBarry Smith . manual - manual page for the monitor 835fde5950dSBarry Smith . monitor - the monitor function 836fde5950dSBarry 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 837fde5950dSBarry Smith 838fde5950dSBarry Smith Level: developer 839fde5950dSBarry Smith 840db781477SPatrick Sanan .seealso: `PetscOptionsGetViewer()`, `PetscOptionsGetReal()`, `PetscOptionsHasName()`, `PetscOptionsGetString()`, 841db781477SPatrick Sanan `PetscOptionsGetIntArray()`, `PetscOptionsGetRealArray()`, `PetscOptionsBool()` 842db781477SPatrick Sanan `PetscOptionsInt()`, `PetscOptionsString()`, `PetscOptionsReal()`, `PetscOptionsBool()`, 843db781477SPatrick Sanan `PetscOptionsName()`, `PetscOptionsBegin()`, `PetscOptionsEnd()`, `PetscOptionsHeadBegin()`, 844c2e3fba1SPatrick Sanan `PetscOptionsStringArray()`, `PetscOptionsRealArray()`, `PetscOptionsScalar()`, 845db781477SPatrick Sanan `PetscOptionsBoolGroupBegin()`, `PetscOptionsBoolGroup()`, `PetscOptionsBoolGroupEnd()`, 846db781477SPatrick Sanan `PetscOptionsFList()`, `PetscOptionsEList()` 847fde5950dSBarry Smith @*/ 848d43b4f6eSBarry Smith PetscErrorCode SNESMonitorSetFromOptions(SNES snes,const char name[],const char help[], const char manual[],PetscErrorCode (*monitor)(SNES,PetscInt,PetscReal,PetscViewerAndFormat*),PetscErrorCode (*monitorsetup)(SNES,PetscViewerAndFormat*)) 849fde5950dSBarry Smith { 850fde5950dSBarry Smith PetscViewer viewer; 851fde5950dSBarry Smith PetscViewerFormat format; 852fde5950dSBarry Smith PetscBool flg; 853fde5950dSBarry Smith 854fde5950dSBarry Smith PetscFunctionBegin; 8559566063dSJacob Faibussowitsch PetscCall(PetscOptionsGetViewer(PetscObjectComm((PetscObject)snes),((PetscObject) snes)->options,((PetscObject)snes)->prefix,name,&viewer,&format,&flg)); 856fde5950dSBarry Smith if (flg) { 857d43b4f6eSBarry Smith PetscViewerAndFormat *vf; 8589566063dSJacob Faibussowitsch PetscCall(PetscViewerAndFormatCreate(viewer,format,&vf)); 8599566063dSJacob Faibussowitsch PetscCall(PetscObjectDereference((PetscObject)viewer)); 8601baa6e33SBarry Smith if (monitorsetup) PetscCall((*monitorsetup)(snes,vf)); 8619566063dSJacob Faibussowitsch PetscCall(SNESMonitorSet(snes,(PetscErrorCode (*)(SNES,PetscInt,PetscReal,void*))monitor,vf,(PetscErrorCode (*)(void**))PetscViewerAndFormatDestroy)); 862fde5950dSBarry Smith } 863fde5950dSBarry Smith PetscFunctionReturn(0); 864fde5950dSBarry Smith } 865fde5950dSBarry Smith 8660f0abf79SStefano Zampini PetscErrorCode SNESEWSetFromOptions_Private(SNESKSPEW* kctx, MPI_Comm comm, const char* prefix) 8670f0abf79SStefano Zampini { 8680f0abf79SStefano Zampini PetscFunctionBegin; 8690f0abf79SStefano Zampini PetscOptionsBegin(comm,prefix,"Eisenstat and Walker type forcing options","KSP"); 8700f0abf79SStefano Zampini PetscCall(PetscOptionsInt("-ksp_ew_version","Version 1, 2 or 3",NULL,kctx->version,&kctx->version,NULL)); 8710f0abf79SStefano Zampini PetscCall(PetscOptionsReal("-ksp_ew_rtol0","0 <= rtol0 < 1",NULL,kctx->rtol_0,&kctx->rtol_0,NULL)); 8720f0abf79SStefano Zampini PetscCall(PetscOptionsReal("-ksp_ew_rtolmax","0 <= rtolmax < 1",NULL,kctx->rtol_max,&kctx->rtol_max,NULL)); 8730f0abf79SStefano Zampini PetscCall(PetscOptionsReal("-ksp_ew_gamma","0 <= gamma <= 1",NULL,kctx->gamma,&kctx->gamma,NULL)); 8740f0abf79SStefano Zampini PetscCall(PetscOptionsReal("-ksp_ew_alpha","1 < alpha <= 2",NULL,kctx->alpha,&kctx->alpha,NULL)); 8750f0abf79SStefano Zampini PetscCall(PetscOptionsReal("-ksp_ew_alpha2","alpha2",NULL,kctx->alpha2,&kctx->alpha2,NULL)); 8760f0abf79SStefano Zampini PetscCall(PetscOptionsReal("-ksp_ew_threshold","0 < threshold < 1",NULL,kctx->threshold,&kctx->threshold,NULL)); 8770f0abf79SStefano Zampini PetscCall(PetscOptionsReal("-ksp_ew_v4_p1","p1",NULL,kctx->v4_p1,&kctx->v4_p1,NULL)); 8780f0abf79SStefano Zampini PetscCall(PetscOptionsReal("-ksp_ew_v4_p2","p2",NULL,kctx->v4_p2,&kctx->v4_p2,NULL)); 8790f0abf79SStefano Zampini PetscCall(PetscOptionsReal("-ksp_ew_v4_p3","p3",NULL,kctx->v4_p3,&kctx->v4_p3,NULL)); 8800f0abf79SStefano Zampini PetscCall(PetscOptionsReal("-ksp_ew_v4_m1","Scaling when rk-1 in [p2,p3)",NULL,kctx->v4_m1,&kctx->v4_m1,NULL)); 8810f0abf79SStefano Zampini PetscCall(PetscOptionsReal("-ksp_ew_v4_m2","Scaling when rk-1 in [p3,+infty)",NULL,kctx->v4_m2,&kctx->v4_m2,NULL)); 8820f0abf79SStefano Zampini PetscCall(PetscOptionsReal("-ksp_ew_v4_m3","Threshold for successive rtol (0.1 in Eq.7)",NULL,kctx->v4_m3,&kctx->v4_m3,NULL)); 8830f0abf79SStefano Zampini PetscCall(PetscOptionsReal("-ksp_ew_v4_m4","Adaptation scaling (0.5 in Eq.7)",NULL,kctx->v4_m4,&kctx->v4_m4,NULL)); 8840f0abf79SStefano Zampini PetscOptionsEnd(); 8850f0abf79SStefano Zampini PetscFunctionReturn(0); 8860f0abf79SStefano Zampini } 8870f0abf79SStefano Zampini 8889b94acceSBarry Smith /*@ 88994b7f48cSBarry Smith SNESSetFromOptions - Sets various SNES and KSP parameters from user options. 8909b94acceSBarry Smith 891c7afd0dbSLois Curfman McInnes Collective on SNES 892c7afd0dbSLois Curfman McInnes 8939b94acceSBarry Smith Input Parameter: 8949b94acceSBarry Smith . snes - the SNES context 8959b94acceSBarry Smith 89636851e7fSLois Curfman McInnes Options Database Keys: 897722329fbSBarry Smith + -snes_type <type> - newtonls, newtontr, ngmres, ncg, nrichardson, qn, vi, fas, SNESType for complete list 89882738288SBarry Smith . -snes_stol - convergence tolerance in terms of the norm 89982738288SBarry Smith of the change in the solution between steps 90070441072SBarry Smith . -snes_atol <abstol> - absolute tolerance of residual norm 901b39c3a46SLois Curfman McInnes . -snes_rtol <rtol> - relative decrease in tolerance norm from initial 902e4d06f11SPatrick Farrell . -snes_divergence_tolerance <divtol> - if the residual goes above divtol*rnorm0, exit with divergence 903be5caee7SBarry Smith . -snes_force_iteration <force> - force SNESSolve() to take at least one iteration 904b39c3a46SLois Curfman McInnes . -snes_max_it <max_it> - maximum number of iterations 905b39c3a46SLois Curfman McInnes . -snes_max_funcs <max_funcs> - maximum number of function evaluations 9064839bfe8SBarry Smith . -snes_max_fail <max_fail> - maximum number of line search failures allowed before stopping, default is none 907ddf469c8SBarry Smith . -snes_max_linear_solve_fail - number of linear solver failures before SNESSolve() stops 908a8054027SBarry Smith . -snes_lag_preconditioner <lag> - how often preconditioner is rebuilt (use -1 to never rebuild) 9093d5a8a6aSBarry Smith . -snes_lag_preconditioner_persists <true,false> - retains the -snes_lag_preconditioner information across multiple SNESSolve() 910e35cf81dSBarry Smith . -snes_lag_jacobian <lag> - how often Jacobian is rebuilt (use -1 to never rebuild) 9113d5a8a6aSBarry Smith . -snes_lag_jacobian_persists <true,false> - retains the -snes_lag_jacobian information across multiple SNESSolve() 912b39c3a46SLois Curfman McInnes . -snes_trtol <trtol> - trust region tolerance 913f362779dSJed Brown . -snes_convergence_test - <default,skip,correct_pressure> convergence test in nonlinear solver. 914f362779dSJed Brown default SNESConvergedDefault(). skip SNESConvergedSkip() means continue iterating until max_it or some other criterion is reached, saving expense 915f362779dSJed Brown of convergence test. correct_pressure SNESConvergedCorrectPressure() has special handling of a pressure null space. 916fde5950dSBarry Smith . -snes_monitor [ascii][:filename][:viewer format] - prints residual norm at each iteration. if no filename given prints to stdout 917fde5950dSBarry Smith . -snes_monitor_solution [ascii binary draw][:filename][:viewer format] - plots solution at each iteration 918fde5950dSBarry Smith . -snes_monitor_residual [ascii binary draw][:filename][:viewer format] - plots residual (not its norm) at each iteration 919fde5950dSBarry Smith . -snes_monitor_solution_update [ascii binary draw][:filename][:viewer format] - plots update to solution at each iteration 9204619e776SBarry Smith . -snes_monitor_lg_residualnorm - plots residual norm at each iteration 921459f5d12SBarry Smith . -snes_monitor_lg_range - plots residual norm at each iteration 9225e7c47f3SMatthew G. Knepley . -snes_monitor_pause_final - Pauses all monitor drawing after the solver ends 923e24b481bSBarry Smith . -snes_fd - use finite differences to compute Jacobian; very slow, only for testing 924e2e60de9SPeter Brune . -snes_fd_color - use finite differences with coloring to compute Jacobian 9255968eb51SBarry Smith . -snes_mf_ksp_monitor - if using matrix-free multiply then print h at each KSP iteration 926b5badacbSBarry Smith . -snes_converged_reason - print the reason for convergence/divergence after each solve 927e62ac41dSBarry Smith . -npc_snes_type <type> - the SNES type to use as a nonlinear preconditioner 928e62ac41dSBarry 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. 929e62ac41dSBarry 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. 93082738288SBarry Smith 93182738288SBarry Smith Options Database for Eisenstat-Walker method: 932fa9f3622SBarry Smith + -snes_ksp_ew - use Eisenstat-Walker method for determining linear system convergence 9334b27c08aSLois Curfman McInnes . -snes_ksp_ew_version ver - version of Eisenstat-Walker method 93436851e7fSLois Curfman McInnes . -snes_ksp_ew_rtol0 <rtol0> - Sets rtol0 93536851e7fSLois Curfman McInnes . -snes_ksp_ew_rtolmax <rtolmax> - Sets rtolmax 93636851e7fSLois Curfman McInnes . -snes_ksp_ew_gamma <gamma> - Sets gamma 93736851e7fSLois Curfman McInnes . -snes_ksp_ew_alpha <alpha> - Sets alpha 93836851e7fSLois Curfman McInnes . -snes_ksp_ew_alpha2 <alpha2> - Sets alpha2 93936851e7fSLois Curfman McInnes - -snes_ksp_ew_threshold <threshold> - Sets threshold 94082738288SBarry Smith 94111ca99fdSLois Curfman McInnes Notes: 942ec5066bdSBarry Smith To see all options, run your program with the -help option or consult the users manual 943ec5066bdSBarry Smith 944ec5066bdSBarry Smith Notes: 945a5b23f4aSJose E. Roman SNES supports three approaches for computing (approximate) Jacobians: user provided via SNESSetJacobian(), matrix free, and computing explicitly with 946ec5066bdSBarry Smith finite differences and coloring using MatFDColoring. It is also possible to use automatic differentiation and the MatFDColoring object. 94783e2fdc7SBarry Smith 94836851e7fSLois Curfman McInnes Level: beginner 94936851e7fSLois Curfman McInnes 950db781477SPatrick Sanan .seealso: `SNESSetOptionsPrefix()`, `SNESResetFromOptions()`, `SNES`, `SNESCreate()` 9519b94acceSBarry Smith @*/ 9527087cfbeSBarry Smith PetscErrorCode SNESSetFromOptions(SNES snes) 9539b94acceSBarry Smith { 9548afaa268SBarry Smith PetscBool flg,pcset,persist,set; 955d8f46077SPeter Brune PetscInt i,indx,lag,grids; 95604d7464bSBarry Smith const char *deft = SNESNEWTONLS; 957649ef022SMatthew Knepley const char *convtests[] = {"default","skip","correct_pressure"}; 95885385478SLisandro Dalcin SNESKSPEW *kctx = NULL; 9590f0abf79SStefano Zampini char type[256], monfilename[PETSC_MAX_PATH_LEN], ewprefix[256]; 960c40d0f55SPeter Brune PCSide pcside; 961a64e098fSPeter Brune const char *optionsprefix; 9629b94acceSBarry Smith 9633a40ed3dSBarry Smith PetscFunctionBegin; 9640700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 9659566063dSJacob Faibussowitsch PetscCall(SNESRegisterAll()); 966d0609cedSBarry Smith PetscObjectOptionsBegin((PetscObject)snes); 967639ff905SBarry Smith if (((PetscObject)snes)->type_name) deft = ((PetscObject)snes)->type_name; 9689566063dSJacob Faibussowitsch PetscCall(PetscOptionsFList("-snes_type","Nonlinear solver method","SNESSetType",SNESList,deft,type,256,&flg)); 969d64ed03dSBarry Smith if (flg) { 9709566063dSJacob Faibussowitsch PetscCall(SNESSetType(snes,type)); 9717adad957SLisandro Dalcin } else if (!((PetscObject)snes)->type_name) { 9729566063dSJacob Faibussowitsch PetscCall(SNESSetType(snes,deft)); 973d64ed03dSBarry Smith } 9749566063dSJacob Faibussowitsch PetscCall(PetscOptionsReal("-snes_stol","Stop if step length less than","SNESSetTolerances",snes->stol,&snes->stol,NULL)); 9759566063dSJacob Faibussowitsch PetscCall(PetscOptionsReal("-snes_atol","Stop if function norm less than","SNESSetTolerances",snes->abstol,&snes->abstol,NULL)); 976186905e3SBarry Smith 9779566063dSJacob Faibussowitsch PetscCall(PetscOptionsReal("-snes_rtol","Stop if decrease in function norm less than","SNESSetTolerances",snes->rtol,&snes->rtol,NULL)); 9789566063dSJacob Faibussowitsch PetscCall(PetscOptionsReal("-snes_divergence_tolerance","Stop if residual norm increases by this factor","SNESSetDivergenceTolerance",snes->divtol,&snes->divtol,NULL)); 9799566063dSJacob Faibussowitsch PetscCall(PetscOptionsInt("-snes_max_it","Maximum iterations","SNESSetTolerances",snes->max_its,&snes->max_its,NULL)); 9809566063dSJacob Faibussowitsch PetscCall(PetscOptionsInt("-snes_max_funcs","Maximum function evaluations","SNESSetTolerances",snes->max_funcs,&snes->max_funcs,NULL)); 9819566063dSJacob Faibussowitsch PetscCall(PetscOptionsInt("-snes_max_fail","Maximum nonlinear step failures","SNESSetMaxNonlinearStepFailures",snes->maxFailures,&snes->maxFailures,NULL)); 9829566063dSJacob Faibussowitsch PetscCall(PetscOptionsInt("-snes_max_linear_solve_fail","Maximum failures in linear solves allowed","SNESSetMaxLinearSolveFailures",snes->maxLinearSolveFailures,&snes->maxLinearSolveFailures,NULL)); 9839566063dSJacob Faibussowitsch PetscCall(PetscOptionsBool("-snes_error_if_not_converged","Generate error if solver does not converge","SNESSetErrorIfNotConverged",snes->errorifnotconverged,&snes->errorifnotconverged,NULL)); 9849566063dSJacob Faibussowitsch PetscCall(PetscOptionsBool("-snes_force_iteration","Force SNESSolve() to take at least one iteration","SNESSetForceIteration",snes->forceiteration,&snes->forceiteration,NULL)); 9859566063dSJacob Faibussowitsch PetscCall(PetscOptionsBool("-snes_check_jacobian_domain_error","Check Jacobian domain error after Jacobian evaluation","SNESCheckJacobianDomainError",snes->checkjacdomainerror,&snes->checkjacdomainerror,NULL)); 98685385478SLisandro Dalcin 9879566063dSJacob Faibussowitsch PetscCall(PetscOptionsInt("-snes_lag_preconditioner","How often to rebuild preconditioner","SNESSetLagPreconditioner",snes->lagpreconditioner,&lag,&flg)); 988a8054027SBarry Smith if (flg) { 9895f80ce2aSJacob Faibussowitsch PetscCheck(lag != -1,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"); 9909566063dSJacob Faibussowitsch PetscCall(SNESSetLagPreconditioner(snes,lag)); 991a8054027SBarry Smith } 9929566063dSJacob Faibussowitsch PetscCall(PetscOptionsBool("-snes_lag_preconditioner_persists","Preconditioner lagging through multiple SNES solves","SNESSetLagPreconditionerPersists",snes->lagjac_persist,&persist,&flg)); 9931baa6e33SBarry Smith if (flg) PetscCall(SNESSetLagPreconditionerPersists(snes,persist)); 9949566063dSJacob Faibussowitsch PetscCall(PetscOptionsInt("-snes_lag_jacobian","How often to rebuild Jacobian","SNESSetLagJacobian",snes->lagjacobian,&lag,&flg)); 995e35cf81dSBarry Smith if (flg) { 9965f80ce2aSJacob Faibussowitsch PetscCheck(lag != -1,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"); 9979566063dSJacob Faibussowitsch PetscCall(SNESSetLagJacobian(snes,lag)); 998e35cf81dSBarry Smith } 9999566063dSJacob Faibussowitsch PetscCall(PetscOptionsBool("-snes_lag_jacobian_persists","Jacobian lagging through multiple SNES solves","SNESSetLagJacobianPersists",snes->lagjac_persist,&persist,&flg)); 10001baa6e33SBarry Smith if (flg) PetscCall(SNESSetLagJacobianPersists(snes,persist)); 100137ec4e1aSPeter Brune 10029566063dSJacob Faibussowitsch PetscCall(PetscOptionsInt("-snes_grid_sequence","Use grid sequencing to generate initial guess","SNESSetGridSequence",snes->gridsequence,&grids,&flg)); 10031baa6e33SBarry Smith if (flg) PetscCall(SNESSetGridSequence(snes,grids)); 1004a8054027SBarry Smith 10059566063dSJacob Faibussowitsch PetscCall(PetscOptionsEList("-snes_convergence_test","Convergence test","SNESSetConvergenceTest",convtests,sizeof(convtests)/sizeof(char*),"default",&indx,&flg)); 100685385478SLisandro Dalcin if (flg) { 100785385478SLisandro Dalcin switch (indx) { 10089566063dSJacob Faibussowitsch case 0: PetscCall(SNESSetConvergenceTest(snes,SNESConvergedDefault,NULL,NULL)); break; 10099566063dSJacob Faibussowitsch case 1: PetscCall(SNESSetConvergenceTest(snes,SNESConvergedSkip,NULL,NULL)); break; 10109566063dSJacob Faibussowitsch case 2: PetscCall(SNESSetConvergenceTest(snes,SNESConvergedCorrectPressure,NULL,NULL)); break; 101185385478SLisandro Dalcin } 101285385478SLisandro Dalcin } 101385385478SLisandro Dalcin 10149566063dSJacob Faibussowitsch PetscCall(PetscOptionsEList("-snes_norm_schedule","SNES Norm schedule","SNESSetNormSchedule",SNESNormSchedules,5,"function",&indx,&flg)); 10159566063dSJacob Faibussowitsch if (flg) PetscCall(SNESSetNormSchedule(snes,(SNESNormSchedule)indx)); 1016fdacfa88SPeter Brune 10179566063dSJacob Faibussowitsch PetscCall(PetscOptionsEList("-snes_function_type","SNES Norm schedule","SNESSetFunctionType",SNESFunctionTypes,2,"unpreconditioned",&indx,&flg)); 10189566063dSJacob Faibussowitsch if (flg) PetscCall(SNESSetFunctionType(snes,(SNESFunctionType)indx)); 1019186905e3SBarry Smith 102085385478SLisandro Dalcin kctx = (SNESKSPEW*)snes->kspconvctx; 102185385478SLisandro Dalcin 10229566063dSJacob Faibussowitsch PetscCall(PetscOptionsBool("-snes_ksp_ew","Use Eisentat-Walker linear system convergence test","SNESKSPSetUseEW",snes->ksp_ewconv,&snes->ksp_ewconv,NULL)); 1023186905e3SBarry Smith 10240f0abf79SStefano Zampini PetscCall(SNESGetOptionsPrefix(snes,&optionsprefix)); 10250f0abf79SStefano Zampini PetscCall(PetscSNPrintf(ewprefix,sizeof(ewprefix),"%s%s",optionsprefix ? optionsprefix : "","snes_")); 10260f0abf79SStefano Zampini PetscCall(SNESEWSetFromOptions_Private(kctx,PetscObjectComm((PetscObject)snes),ewprefix)); 10270f0abf79SStefano Zampini 10289566063dSJacob Faibussowitsch PetscCall(PetscOptionsInt("-snes_ksp_ew_version","Version 1, 2 or 3","SNESKSPSetParametersEW",kctx->version,&kctx->version,NULL)); 10299566063dSJacob Faibussowitsch PetscCall(PetscOptionsReal("-snes_ksp_ew_rtol0","0 <= rtol0 < 1","SNESKSPSetParametersEW",kctx->rtol_0,&kctx->rtol_0,NULL)); 10309566063dSJacob Faibussowitsch PetscCall(PetscOptionsReal("-snes_ksp_ew_rtolmax","0 <= rtolmax < 1","SNESKSPSetParametersEW",kctx->rtol_max,&kctx->rtol_max,NULL)); 10319566063dSJacob Faibussowitsch PetscCall(PetscOptionsReal("-snes_ksp_ew_gamma","0 <= gamma <= 1","SNESKSPSetParametersEW",kctx->gamma,&kctx->gamma,NULL)); 10329566063dSJacob Faibussowitsch PetscCall(PetscOptionsReal("-snes_ksp_ew_alpha","1 < alpha <= 2","SNESKSPSetParametersEW",kctx->alpha,&kctx->alpha,NULL)); 10339566063dSJacob Faibussowitsch PetscCall(PetscOptionsReal("-snes_ksp_ew_alpha2","alpha2","SNESKSPSetParametersEW",kctx->alpha2,&kctx->alpha2,NULL)); 10349566063dSJacob Faibussowitsch PetscCall(PetscOptionsReal("-snes_ksp_ew_threshold","0 < threshold < 1","SNESKSPSetParametersEW",kctx->threshold,&kctx->threshold,NULL)); 1035186905e3SBarry Smith 103690d69ab7SBarry Smith flg = PETSC_FALSE; 10379566063dSJacob Faibussowitsch PetscCall(PetscOptionsBool("-snes_monitor_cancel","Remove all monitors","SNESMonitorCancel",flg,&flg,&set)); 10389566063dSJacob Faibussowitsch if (set && flg) PetscCall(SNESMonitorCancel(snes)); 1039eabae89aSBarry Smith 10409566063dSJacob Faibussowitsch PetscCall(SNESMonitorSetFromOptions(snes,"-snes_monitor","Monitor norm of function","SNESMonitorDefault",SNESMonitorDefault,SNESMonitorDefaultSetUp)); 10419566063dSJacob Faibussowitsch PetscCall(SNESMonitorSetFromOptions(snes,"-snes_monitor_short","Monitor norm of function with fewer digits","SNESMonitorDefaultShort",SNESMonitorDefaultShort,NULL)); 10429566063dSJacob Faibussowitsch PetscCall(SNESMonitorSetFromOptions(snes,"-snes_monitor_range","Monitor range of elements of function","SNESMonitorRange",SNESMonitorRange,NULL)); 1043eabae89aSBarry Smith 10449566063dSJacob Faibussowitsch PetscCall(SNESMonitorSetFromOptions(snes,"-snes_monitor_ratio","Monitor ratios of the norm of function for consecutive steps","SNESMonitorRatio",SNESMonitorRatio,SNESMonitorRatioSetUp)); 10459566063dSJacob Faibussowitsch PetscCall(SNESMonitorSetFromOptions(snes,"-snes_monitor_field","Monitor norm of function (split into fields)","SNESMonitorDefaultField",SNESMonitorDefaultField,NULL)); 10469566063dSJacob Faibussowitsch PetscCall(SNESMonitorSetFromOptions(snes,"-snes_monitor_solution","View solution at each iteration","SNESMonitorSolution",SNESMonitorSolution,NULL)); 10479566063dSJacob Faibussowitsch PetscCall(SNESMonitorSetFromOptions(snes,"-snes_monitor_solution_update","View correction at each iteration","SNESMonitorSolutionUpdate",SNESMonitorSolutionUpdate,NULL)); 10489566063dSJacob Faibussowitsch PetscCall(SNESMonitorSetFromOptions(snes,"-snes_monitor_residual","View residual at each iteration","SNESMonitorResidual",SNESMonitorResidual,NULL)); 10499566063dSJacob Faibussowitsch PetscCall(SNESMonitorSetFromOptions(snes,"-snes_monitor_jacupdate_spectrum","Print the change in the spectrum of the Jacobian","SNESMonitorJacUpdateSpectrum",SNESMonitorJacUpdateSpectrum,NULL)); 10509566063dSJacob Faibussowitsch PetscCall(SNESMonitorSetFromOptions(snes,"-snes_monitor_fields","Monitor norm of function per field","SNESMonitorSet",SNESMonitorFields,NULL)); 10519566063dSJacob Faibussowitsch PetscCall(PetscOptionsBool("-snes_monitor_pause_final", "Pauses all draw monitors at the final iterate", "SNESMonitorPauseFinal_Internal", PETSC_FALSE, &snes->pauseFinal, NULL)); 10522db13446SMatthew G. Knepley 10539566063dSJacob Faibussowitsch PetscCall(PetscOptionsString("-snes_monitor_python","Use Python function","SNESMonitorSet",NULL,monfilename,sizeof(monfilename),&flg)); 10549566063dSJacob Faibussowitsch if (flg) PetscCall(PetscPythonMonitorSet((PetscObject)snes,monfilename)); 10555180491cSLisandro Dalcin 105690d69ab7SBarry Smith flg = PETSC_FALSE; 10579566063dSJacob Faibussowitsch PetscCall(PetscOptionsBool("-snes_monitor_lg_range","Plot function range at each iteration","SNESMonitorLGRange",flg,&flg,NULL)); 1058459f5d12SBarry Smith if (flg) { 1059459f5d12SBarry Smith PetscViewer ctx; 1060e24b481bSBarry Smith 10619566063dSJacob Faibussowitsch PetscCall(PetscViewerDrawOpen(PetscObjectComm((PetscObject)snes),NULL,NULL,PETSC_DECIDE,PETSC_DECIDE,400,300,&ctx)); 10629566063dSJacob Faibussowitsch PetscCall(SNESMonitorSet(snes,SNESMonitorLGRange,ctx,(PetscErrorCode (*)(void**))PetscViewerDestroy)); 1063459f5d12SBarry Smith } 10642e7541e6SPeter Brune 106590d69ab7SBarry Smith flg = PETSC_FALSE; 10669566063dSJacob Faibussowitsch PetscCall(PetscOptionsBool("-snes_converged_reason_view_cancel","Remove all converged reason viewers","SNESConvergedReasonViewCancel",flg,&flg,&set)); 10679566063dSJacob Faibussowitsch if (set && flg) PetscCall(SNESConvergedReasonViewCancel(snes)); 1068c4421ceaSFande Kong 1069c4421ceaSFande Kong flg = PETSC_FALSE; 10709566063dSJacob Faibussowitsch PetscCall(PetscOptionsBool("-snes_fd","Use finite differences (slow) to compute Jacobian","SNESComputeJacobianDefault",flg,&flg,NULL)); 10714b27c08aSLois Curfman McInnes if (flg) { 10726cab3a1bSJed Brown void *functx; 1073b1f624c7SBarry Smith DM dm; 10749566063dSJacob Faibussowitsch PetscCall(SNESGetDM(snes,&dm)); 1075800f99ffSJeremy L Thompson PetscCall(DMSNESUnsetJacobianContext_Internal(dm)); 10769566063dSJacob Faibussowitsch PetscCall(SNESGetFunction(snes,NULL,NULL,&functx)); 10779566063dSJacob Faibussowitsch PetscCall(SNESSetJacobian(snes,snes->jacobian,snes->jacobian_pre,SNESComputeJacobianDefault,functx)); 10789566063dSJacob Faibussowitsch PetscCall(PetscInfo(snes,"Setting default finite difference Jacobian matrix\n")); 10799b94acceSBarry Smith } 1080639f9d9dSBarry Smith 108144848bc4SPeter Brune flg = PETSC_FALSE; 10829566063dSJacob Faibussowitsch PetscCall(PetscOptionsBool("-snes_fd_function","Use finite differences (slow) to compute function from user objective","SNESObjectiveComputeFunctionDefaultFD",flg,&flg,NULL)); 10831baa6e33SBarry Smith if (flg) PetscCall(SNESSetFunction(snes,NULL,SNESObjectiveComputeFunctionDefaultFD,NULL)); 108497584545SPeter Brune 108597584545SPeter Brune flg = PETSC_FALSE; 10869566063dSJacob Faibussowitsch PetscCall(PetscOptionsBool("-snes_fd_color","Use finite differences with coloring to compute Jacobian","SNESComputeJacobianDefaultColor",flg,&flg,NULL)); 108744848bc4SPeter Brune if (flg) { 1088c52e227fSPeter Brune DM dm; 10899566063dSJacob Faibussowitsch PetscCall(SNESGetDM(snes,&dm)); 1090800f99ffSJeremy L Thompson PetscCall(DMSNESUnsetJacobianContext_Internal(dm)); 10919566063dSJacob Faibussowitsch PetscCall(SNESSetJacobian(snes,snes->jacobian,snes->jacobian_pre,SNESComputeJacobianDefaultColor,NULL)); 10929566063dSJacob Faibussowitsch PetscCall(PetscInfo(snes,"Setting default finite difference coloring Jacobian matrix\n")); 109344848bc4SPeter Brune } 109444848bc4SPeter Brune 1095aa3661deSLisandro Dalcin flg = PETSC_FALSE; 10969566063dSJacob Faibussowitsch PetscCall(PetscOptionsBool("-snes_mf_operator","Use a Matrix-Free Jacobian with user-provided preconditioner matrix","SNESSetUseMatrixFree",PETSC_FALSE,&snes->mf_operator,&flg)); 1097d8f46077SPeter Brune if (flg && snes->mf_operator) { 1098a8248277SBarry Smith snes->mf_operator = PETSC_TRUE; 1099d8f46077SPeter Brune snes->mf = PETSC_TRUE; 1100a8248277SBarry Smith } 1101aa3661deSLisandro Dalcin flg = PETSC_FALSE; 11029566063dSJacob Faibussowitsch PetscCall(PetscOptionsBool("-snes_mf","Use a Matrix-Free Jacobian with no preconditioner matrix","SNESSetUseMatrixFree",PETSC_FALSE,&snes->mf,&flg)); 1103d8f46077SPeter Brune if (!flg && snes->mf_operator) snes->mf = PETSC_TRUE; 11049566063dSJacob Faibussowitsch PetscCall(PetscOptionsInt("-snes_mf_version","Matrix-Free routines version 1 or 2","None",snes->mf_version,&snes->mf_version,NULL)); 1105d28543b3SPeter Brune 1106c40d0f55SPeter Brune flg = PETSC_FALSE; 11079566063dSJacob Faibussowitsch PetscCall(SNESGetNPCSide(snes,&pcside)); 11089566063dSJacob Faibussowitsch PetscCall(PetscOptionsEnum("-snes_npc_side","SNES nonlinear preconditioner side","SNESSetNPCSide",PCSides,(PetscEnum)pcside,(PetscEnum*)&pcside,&flg)); 11099566063dSJacob Faibussowitsch if (flg) PetscCall(SNESSetNPCSide(snes,pcside)); 1110c40d0f55SPeter Brune 1111e04113cfSBarry Smith #if defined(PETSC_HAVE_SAWS) 11128a70d858SHong Zhang /* 11138a70d858SHong Zhang Publish convergence information using SAWs 11148a70d858SHong Zhang */ 11158a70d858SHong Zhang flg = PETSC_FALSE; 11169566063dSJacob Faibussowitsch PetscCall(PetscOptionsBool("-snes_monitor_saws","Publish SNES progress using SAWs","SNESMonitorSet",flg,&flg,NULL)); 11178a70d858SHong Zhang if (flg) { 11188a70d858SHong Zhang void *ctx; 11199566063dSJacob Faibussowitsch PetscCall(SNESMonitorSAWsCreate(snes,&ctx)); 11209566063dSJacob Faibussowitsch PetscCall(SNESMonitorSet(snes,SNESMonitorSAWs,ctx,SNESMonitorSAWsDestroy)); 11218a70d858SHong Zhang } 11228a70d858SHong Zhang #endif 11238a70d858SHong Zhang #if defined(PETSC_HAVE_SAWS) 1124b90c6cbeSBarry Smith { 1125b90c6cbeSBarry Smith PetscBool set; 1126b90c6cbeSBarry Smith flg = PETSC_FALSE; 11279566063dSJacob Faibussowitsch PetscCall(PetscOptionsBool("-snes_saws_block","Block for SAWs at end of SNESSolve","PetscObjectSAWsBlock",((PetscObject)snes)->amspublishblock,&flg,&set)); 11281baa6e33SBarry Smith if (set) PetscCall(PetscObjectSAWsSetBlock((PetscObject)snes,flg)); 1129b90c6cbeSBarry Smith } 1130b90c6cbeSBarry Smith #endif 1131b90c6cbeSBarry Smith 113276b2cf59SMatthew Knepley for (i = 0; i < numberofsetfromoptions; i++) { 11339566063dSJacob Faibussowitsch PetscCall((*othersetfromoptions[i])(snes)); 113476b2cf59SMatthew Knepley } 113576b2cf59SMatthew Knepley 1136*dbbe0bcdSBarry Smith PetscTryTypeMethod(snes,setfromoptions,PetscOptionsObject); 11375d973c19SBarry Smith 11385d973c19SBarry Smith /* process any options handlers added with PetscObjectAddOptionsHandler() */ 1139*dbbe0bcdSBarry Smith PetscCall(PetscObjectProcessOptionsHandlers((PetscObject)snes,PetscOptionsObject)); 1140d0609cedSBarry Smith PetscOptionsEnd(); 11414bbc92c1SBarry Smith 1142d8d34be6SBarry Smith if (snes->linesearch) { 11439566063dSJacob Faibussowitsch PetscCall(SNESGetLineSearch(snes, &snes->linesearch)); 11449566063dSJacob Faibussowitsch PetscCall(SNESLineSearchSetFromOptions(snes->linesearch)); 1145d8d34be6SBarry Smith } 11469e764e56SPeter Brune 11476aa5e7e9SBarry Smith if (snes->usesksp) { 11489566063dSJacob Faibussowitsch if (!snes->ksp) PetscCall(SNESGetKSP(snes,&snes->ksp)); 11499566063dSJacob Faibussowitsch PetscCall(KSPSetOperators(snes->ksp,snes->jacobian,snes->jacobian_pre)); 11509566063dSJacob Faibussowitsch PetscCall(KSPSetFromOptions(snes->ksp)); 11516aa5e7e9SBarry Smith } 11526991f827SBarry Smith 1153b5badacbSBarry Smith /* if user has set the SNES NPC type via options database, create it. */ 11549566063dSJacob Faibussowitsch PetscCall(SNESGetOptionsPrefix(snes, &optionsprefix)); 11559566063dSJacob Faibussowitsch PetscCall(PetscOptionsHasName(((PetscObject)snes)->options,optionsprefix, "-npc_snes_type", &pcset)); 1156efd4aadfSBarry Smith if (pcset && (!snes->npc)) { 11579566063dSJacob Faibussowitsch PetscCall(SNESGetNPC(snes, &snes->npc)); 115851e86f29SPeter Brune } 11591baa6e33SBarry Smith if (snes->npc) PetscCall(SNESSetFromOptions(snes->npc)); 1160b3cd9a81SMatthew G. Knepley snes->setfromoptionscalled++; 1161b3cd9a81SMatthew G. Knepley PetscFunctionReturn(0); 1162b3cd9a81SMatthew G. Knepley } 1163b3cd9a81SMatthew G. Knepley 1164b3cd9a81SMatthew G. Knepley /*@ 1165b3cd9a81SMatthew G. Knepley SNESResetFromOptions - Sets various SNES and KSP parameters from user options ONLY if the SNES was previously set from options 1166b3cd9a81SMatthew G. Knepley 1167b3cd9a81SMatthew G. Knepley Collective on SNES 1168b3cd9a81SMatthew G. Knepley 1169b3cd9a81SMatthew G. Knepley Input Parameter: 1170b3cd9a81SMatthew G. Knepley . snes - the SNES context 1171b3cd9a81SMatthew G. Knepley 1172b3cd9a81SMatthew G. Knepley Level: beginner 1173b3cd9a81SMatthew G. Knepley 1174db781477SPatrick Sanan .seealso: `SNESSetFromOptions()`, `SNESSetOptionsPrefix()` 1175b3cd9a81SMatthew G. Knepley @*/ 1176b3cd9a81SMatthew G. Knepley PetscErrorCode SNESResetFromOptions(SNES snes) 1177b3cd9a81SMatthew G. Knepley { 1178b3cd9a81SMatthew G. Knepley PetscFunctionBegin; 11799566063dSJacob Faibussowitsch if (snes->setfromoptionscalled) PetscCall(SNESSetFromOptions(snes)); 11803a40ed3dSBarry Smith PetscFunctionReturn(0); 11819b94acceSBarry Smith } 11829b94acceSBarry Smith 1183bb9467b5SJed Brown /*@C 1184d25893d9SBarry Smith SNESSetComputeApplicationContext - Sets an optional function to compute a user-defined context for 1185d25893d9SBarry Smith the nonlinear solvers. 1186d25893d9SBarry Smith 1187d25893d9SBarry Smith Logically Collective on SNES 1188d25893d9SBarry Smith 1189d25893d9SBarry Smith Input Parameters: 1190d25893d9SBarry Smith + snes - the SNES context 1191d25893d9SBarry Smith . compute - function to compute the context 1192d25893d9SBarry Smith - destroy - function to destroy the context 1193d25893d9SBarry Smith 1194d25893d9SBarry Smith Level: intermediate 1195d25893d9SBarry Smith 1196bb9467b5SJed Brown Notes: 1197bb9467b5SJed Brown This function is currently not available from Fortran. 1198bb9467b5SJed Brown 1199db781477SPatrick Sanan .seealso: `SNESGetApplicationContext()`, `SNESSetComputeApplicationContext()`, `SNESGetApplicationContext()` 1200d25893d9SBarry Smith @*/ 1201d25893d9SBarry Smith PetscErrorCode SNESSetComputeApplicationContext(SNES snes,PetscErrorCode (*compute)(SNES,void**),PetscErrorCode (*destroy)(void**)) 1202d25893d9SBarry Smith { 1203d25893d9SBarry Smith PetscFunctionBegin; 1204d25893d9SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 1205d25893d9SBarry Smith snes->ops->usercompute = compute; 1206d25893d9SBarry Smith snes->ops->userdestroy = destroy; 1207d25893d9SBarry Smith PetscFunctionReturn(0); 1208d25893d9SBarry Smith } 1209a847f771SSatish Balay 1210b07ff414SBarry Smith /*@ 12119b94acceSBarry Smith SNESSetApplicationContext - Sets the optional user-defined context for 12129b94acceSBarry Smith the nonlinear solvers. 12139b94acceSBarry Smith 12143f9fe445SBarry Smith Logically Collective on SNES 1215fee21e36SBarry Smith 1216c7afd0dbSLois Curfman McInnes Input Parameters: 1217c7afd0dbSLois Curfman McInnes + snes - the SNES context 1218c7afd0dbSLois Curfman McInnes - usrP - optional user context 1219c7afd0dbSLois Curfman McInnes 122036851e7fSLois Curfman McInnes Level: intermediate 122136851e7fSLois Curfman McInnes 122295452b02SPatrick Sanan Fortran Notes: 122395452b02SPatrick Sanan To use this from Fortran you must write a Fortran interface definition for this 1224daf670e6SBarry Smith function that tells Fortran the Fortran derived data type that you are passing in as the ctx argument. 1225daf670e6SBarry Smith 1226db781477SPatrick Sanan .seealso: `SNESGetApplicationContext()` 12279b94acceSBarry Smith @*/ 12287087cfbeSBarry Smith PetscErrorCode SNESSetApplicationContext(SNES snes,void *usrP) 12299b94acceSBarry Smith { 1230b07ff414SBarry Smith KSP ksp; 12311b2093e4SBarry Smith 12323a40ed3dSBarry Smith PetscFunctionBegin; 12330700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 12349566063dSJacob Faibussowitsch PetscCall(SNESGetKSP(snes,&ksp)); 12359566063dSJacob Faibussowitsch PetscCall(KSPSetApplicationContext(ksp,usrP)); 12369b94acceSBarry Smith snes->user = usrP; 12373a40ed3dSBarry Smith PetscFunctionReturn(0); 12389b94acceSBarry Smith } 123974679c65SBarry Smith 1240b07ff414SBarry Smith /*@ 12419b94acceSBarry Smith SNESGetApplicationContext - Gets the user-defined context for the 12429b94acceSBarry Smith nonlinear solvers. 12439b94acceSBarry Smith 1244c7afd0dbSLois Curfman McInnes Not Collective 1245c7afd0dbSLois Curfman McInnes 12469b94acceSBarry Smith Input Parameter: 12479b94acceSBarry Smith . snes - SNES context 12489b94acceSBarry Smith 12499b94acceSBarry Smith Output Parameter: 12509b94acceSBarry Smith . usrP - user context 12519b94acceSBarry Smith 125295452b02SPatrick Sanan Fortran Notes: 125395452b02SPatrick Sanan To use this from Fortran you must write a Fortran interface definition for this 1254daf670e6SBarry Smith function that tells Fortran the Fortran derived data type that you are passing in as the ctx argument. 1255daf670e6SBarry Smith 125636851e7fSLois Curfman McInnes Level: intermediate 125736851e7fSLois Curfman McInnes 1258db781477SPatrick Sanan .seealso: `SNESSetApplicationContext()` 12599b94acceSBarry Smith @*/ 1260e71120c6SJed Brown PetscErrorCode SNESGetApplicationContext(SNES snes,void *usrP) 12619b94acceSBarry Smith { 12623a40ed3dSBarry Smith PetscFunctionBegin; 12630700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 1264e71120c6SJed Brown *(void**)usrP = snes->user; 12653a40ed3dSBarry Smith PetscFunctionReturn(0); 12669b94acceSBarry Smith } 126774679c65SBarry Smith 12689b94acceSBarry Smith /*@ 1269ec5066bdSBarry Smith SNESSetUseMatrixFree - indicates that SNES should use matrix free finite difference matrix vector products internally to apply the Jacobian. 12703565c898SBarry Smith 12713565c898SBarry Smith Collective on SNES 12723565c898SBarry Smith 12733565c898SBarry Smith Input Parameters: 12743565c898SBarry Smith + snes - SNES context 12754ddffce6SLisandro Dalcin . mf_operator - use matrix-free only for the Amat used by SNESSetJacobian(), this means the user provided Pmat will continue to be used 12764ddffce6SLisandro 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 12773565c898SBarry Smith 12783565c898SBarry Smith Options Database: 12793565c898SBarry Smith + -snes_mf - use matrix free for both the mat and pmat operator 1280ec5066bdSBarry Smith . -snes_mf_operator - use matrix free only for the mat operator 1281ec5066bdSBarry Smith . -snes_fd_color - compute the Jacobian via coloring and finite differences. 1282ec5066bdSBarry Smith - -snes_fd - compute the Jacobian via finite differences (slow) 12833565c898SBarry Smith 12843565c898SBarry Smith Level: intermediate 12853565c898SBarry Smith 1286ec5066bdSBarry Smith Notes: 1287a5b23f4aSJose E. Roman SNES supports three approaches for computing (approximate) Jacobians: user provided via SNESSetJacobian(), matrix free, and computing explicitly with 1288ec5066bdSBarry Smith finite differences and coloring using MatFDColoring. It is also possible to use automatic differentiation and the MatFDColoring object. 1289ec5066bdSBarry Smith 1290db781477SPatrick Sanan .seealso: `SNESGetUseMatrixFree()`, `MatCreateSNESMF()`, `SNESComputeJacobianDefaultColor()` 12913565c898SBarry Smith @*/ 12923565c898SBarry Smith PetscErrorCode SNESSetUseMatrixFree(SNES snes,PetscBool mf_operator,PetscBool mf) 12933565c898SBarry Smith { 12943565c898SBarry Smith PetscFunctionBegin; 12953565c898SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 129688b4c220SStefano Zampini PetscValidLogicalCollectiveBool(snes,mf_operator,2); 129788b4c220SStefano Zampini PetscValidLogicalCollectiveBool(snes,mf,3); 12984ddffce6SLisandro Dalcin snes->mf = mf_operator ? PETSC_TRUE : mf; 12993565c898SBarry Smith snes->mf_operator = mf_operator; 13003565c898SBarry Smith PetscFunctionReturn(0); 13013565c898SBarry Smith } 13023565c898SBarry Smith 13033565c898SBarry Smith /*@ 1304ec5066bdSBarry Smith SNESGetUseMatrixFree - indicates if the SNES uses matrix free finite difference matrix vector products to apply the Jacobian. 13053565c898SBarry Smith 13063565c898SBarry Smith Collective on SNES 13073565c898SBarry Smith 13083565c898SBarry Smith Input Parameter: 13093565c898SBarry Smith . snes - SNES context 13103565c898SBarry Smith 13113565c898SBarry Smith Output Parameters: 13124ddffce6SLisandro Dalcin + mf_operator - use matrix-free only for the Amat used by SNESSetJacobian(), this means the user provided Pmat will continue to be used 13134ddffce6SLisandro 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 13143565c898SBarry Smith 13153565c898SBarry Smith Options Database: 13163565c898SBarry Smith + -snes_mf - use matrix free for both the mat and pmat operator 13173565c898SBarry Smith - -snes_mf_operator - use matrix free only for the mat operator 13183565c898SBarry Smith 13193565c898SBarry Smith Level: intermediate 13203565c898SBarry Smith 1321db781477SPatrick Sanan .seealso: `SNESSetUseMatrixFree()`, `MatCreateSNESMF()` 13223565c898SBarry Smith @*/ 13233565c898SBarry Smith PetscErrorCode SNESGetUseMatrixFree(SNES snes,PetscBool *mf_operator,PetscBool *mf) 13243565c898SBarry Smith { 13253565c898SBarry Smith PetscFunctionBegin; 13263565c898SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 13273565c898SBarry Smith if (mf) *mf = snes->mf; 13283565c898SBarry Smith if (mf_operator) *mf_operator = snes->mf_operator; 13293565c898SBarry Smith PetscFunctionReturn(0); 13303565c898SBarry Smith } 13313565c898SBarry Smith 13323565c898SBarry Smith /*@ 1333c8228a4eSBarry Smith SNESGetIterationNumber - Gets the number of nonlinear iterations completed 1334c8228a4eSBarry Smith at this time. 13359b94acceSBarry Smith 1336c7afd0dbSLois Curfman McInnes Not Collective 1337c7afd0dbSLois Curfman McInnes 13389b94acceSBarry Smith Input Parameter: 13399b94acceSBarry Smith . snes - SNES context 13409b94acceSBarry Smith 13419b94acceSBarry Smith Output Parameter: 13429b94acceSBarry Smith . iter - iteration number 13439b94acceSBarry Smith 1344c8228a4eSBarry Smith Notes: 1345c8228a4eSBarry Smith For example, during the computation of iteration 2 this would return 1. 1346c8228a4eSBarry Smith 1347c8228a4eSBarry Smith This is useful for using lagged Jacobians (where one does not recompute the 134808405cd6SLois Curfman McInnes Jacobian at each SNES iteration). For example, the code 134908405cd6SLois Curfman McInnes .vb 135008405cd6SLois Curfman McInnes ierr = SNESGetIterationNumber(snes,&it); 135108405cd6SLois Curfman McInnes if (!(it % 2)) { 135208405cd6SLois Curfman McInnes [compute Jacobian here] 135308405cd6SLois Curfman McInnes } 135408405cd6SLois Curfman McInnes .ve 1355c8228a4eSBarry Smith can be used in your ComputeJacobian() function to cause the Jacobian to be 135608405cd6SLois Curfman McInnes recomputed every second SNES iteration. 1357c8228a4eSBarry Smith 1358c04deec6SBarry Smith After the SNES solve is complete this will return the number of nonlinear iterations used. 1359c04deec6SBarry Smith 136036851e7fSLois Curfman McInnes Level: intermediate 136136851e7fSLois Curfman McInnes 1362db781477SPatrick Sanan .seealso: `SNESGetLinearSolveIterations()` 13639b94acceSBarry Smith @*/ 13647087cfbeSBarry Smith PetscErrorCode SNESGetIterationNumber(SNES snes,PetscInt *iter) 13659b94acceSBarry Smith { 13663a40ed3dSBarry Smith PetscFunctionBegin; 13670700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 13684482741eSBarry Smith PetscValidIntPointer(iter,2); 13699b94acceSBarry Smith *iter = snes->iter; 13703a40ed3dSBarry Smith PetscFunctionReturn(0); 13719b94acceSBarry Smith } 137274679c65SBarry Smith 1373360c497dSPeter Brune /*@ 1374360c497dSPeter Brune SNESSetIterationNumber - Sets the current iteration number. 1375360c497dSPeter Brune 1376360c497dSPeter Brune Not Collective 1377360c497dSPeter Brune 1378d8d19677SJose E. Roman Input Parameters: 1379a2b725a8SWilliam Gropp + snes - SNES context 1380a2b725a8SWilliam Gropp - iter - iteration number 1381360c497dSPeter Brune 1382360c497dSPeter Brune Level: developer 1383360c497dSPeter Brune 1384db781477SPatrick Sanan .seealso: `SNESGetLinearSolveIterations()` 1385360c497dSPeter Brune @*/ 1386360c497dSPeter Brune PetscErrorCode SNESSetIterationNumber(SNES snes,PetscInt iter) 1387360c497dSPeter Brune { 1388360c497dSPeter Brune PetscFunctionBegin; 1389360c497dSPeter Brune PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 13909566063dSJacob Faibussowitsch PetscCall(PetscObjectSAWsTakeAccess((PetscObject)snes)); 1391360c497dSPeter Brune snes->iter = iter; 13929566063dSJacob Faibussowitsch PetscCall(PetscObjectSAWsGrantAccess((PetscObject)snes)); 1393360c497dSPeter Brune PetscFunctionReturn(0); 1394360c497dSPeter Brune } 1395360c497dSPeter Brune 13969b94acceSBarry Smith /*@ 1397b850b91aSLisandro Dalcin SNESGetNonlinearStepFailures - Gets the number of unsuccessful steps 13989b94acceSBarry Smith attempted by the nonlinear solver. 13999b94acceSBarry Smith 1400c7afd0dbSLois Curfman McInnes Not Collective 1401c7afd0dbSLois Curfman McInnes 14029b94acceSBarry Smith Input Parameter: 14039b94acceSBarry Smith . snes - SNES context 14049b94acceSBarry Smith 14059b94acceSBarry Smith Output Parameter: 14069b94acceSBarry Smith . nfails - number of unsuccessful steps attempted 14079b94acceSBarry Smith 1408c96a6f78SLois Curfman McInnes Notes: 1409c96a6f78SLois Curfman McInnes This counter is reset to zero for each successive call to SNESSolve(). 1410c96a6f78SLois Curfman McInnes 141136851e7fSLois Curfman McInnes Level: intermediate 141236851e7fSLois Curfman McInnes 1413db781477SPatrick Sanan .seealso: `SNESGetMaxLinearSolveFailures()`, `SNESGetLinearSolveIterations()`, `SNESSetMaxLinearSolveFailures()`, `SNESGetLinearSolveFailures()`, 1414db781477SPatrick Sanan `SNESSetMaxNonlinearStepFailures()`, `SNESGetMaxNonlinearStepFailures()` 14159b94acceSBarry Smith @*/ 14167087cfbeSBarry Smith PetscErrorCode SNESGetNonlinearStepFailures(SNES snes,PetscInt *nfails) 14179b94acceSBarry Smith { 14183a40ed3dSBarry Smith PetscFunctionBegin; 14190700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 14204482741eSBarry Smith PetscValidIntPointer(nfails,2); 142150ffb88aSMatthew Knepley *nfails = snes->numFailures; 142250ffb88aSMatthew Knepley PetscFunctionReturn(0); 142350ffb88aSMatthew Knepley } 142450ffb88aSMatthew Knepley 142550ffb88aSMatthew Knepley /*@ 1426b850b91aSLisandro Dalcin SNESSetMaxNonlinearStepFailures - Sets the maximum number of unsuccessful steps 142750ffb88aSMatthew Knepley attempted by the nonlinear solver before it gives up. 142850ffb88aSMatthew Knepley 142950ffb88aSMatthew Knepley Not Collective 143050ffb88aSMatthew Knepley 143150ffb88aSMatthew Knepley Input Parameters: 143250ffb88aSMatthew Knepley + snes - SNES context 143350ffb88aSMatthew Knepley - maxFails - maximum of unsuccessful steps 143450ffb88aSMatthew Knepley 143550ffb88aSMatthew Knepley Level: intermediate 143650ffb88aSMatthew Knepley 1437db781477SPatrick Sanan .seealso: `SNESGetMaxLinearSolveFailures()`, `SNESGetLinearSolveIterations()`, `SNESSetMaxLinearSolveFailures()`, `SNESGetLinearSolveFailures()`, 1438db781477SPatrick Sanan `SNESGetMaxNonlinearStepFailures()`, `SNESGetNonlinearStepFailures()` 143950ffb88aSMatthew Knepley @*/ 14407087cfbeSBarry Smith PetscErrorCode SNESSetMaxNonlinearStepFailures(SNES snes, PetscInt maxFails) 144150ffb88aSMatthew Knepley { 144250ffb88aSMatthew Knepley PetscFunctionBegin; 14430700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 144450ffb88aSMatthew Knepley snes->maxFailures = maxFails; 144550ffb88aSMatthew Knepley PetscFunctionReturn(0); 144650ffb88aSMatthew Knepley } 144750ffb88aSMatthew Knepley 144850ffb88aSMatthew Knepley /*@ 1449b850b91aSLisandro Dalcin SNESGetMaxNonlinearStepFailures - Gets the maximum number of unsuccessful steps 145050ffb88aSMatthew Knepley attempted by the nonlinear solver before it gives up. 145150ffb88aSMatthew Knepley 145250ffb88aSMatthew Knepley Not Collective 145350ffb88aSMatthew Knepley 145450ffb88aSMatthew Knepley Input Parameter: 145550ffb88aSMatthew Knepley . snes - SNES context 145650ffb88aSMatthew Knepley 145750ffb88aSMatthew Knepley Output Parameter: 145850ffb88aSMatthew Knepley . maxFails - maximum of unsuccessful steps 145950ffb88aSMatthew Knepley 146050ffb88aSMatthew Knepley Level: intermediate 146150ffb88aSMatthew Knepley 1462db781477SPatrick Sanan .seealso: `SNESGetMaxLinearSolveFailures()`, `SNESGetLinearSolveIterations()`, `SNESSetMaxLinearSolveFailures()`, `SNESGetLinearSolveFailures()`, 1463db781477SPatrick Sanan `SNESSetMaxNonlinearStepFailures()`, `SNESGetNonlinearStepFailures()` 146458ebbce7SBarry Smith 146550ffb88aSMatthew Knepley @*/ 14667087cfbeSBarry Smith PetscErrorCode SNESGetMaxNonlinearStepFailures(SNES snes, PetscInt *maxFails) 146750ffb88aSMatthew Knepley { 146850ffb88aSMatthew Knepley PetscFunctionBegin; 14690700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 14704482741eSBarry Smith PetscValidIntPointer(maxFails,2); 147150ffb88aSMatthew Knepley *maxFails = snes->maxFailures; 14723a40ed3dSBarry Smith PetscFunctionReturn(0); 14739b94acceSBarry Smith } 1474a847f771SSatish Balay 14752541af92SBarry Smith /*@ 14762541af92SBarry Smith SNESGetNumberFunctionEvals - Gets the number of user provided function evaluations 14772541af92SBarry Smith done by SNES. 14782541af92SBarry Smith 14792541af92SBarry Smith Not Collective 14802541af92SBarry Smith 14812541af92SBarry Smith Input Parameter: 14822541af92SBarry Smith . snes - SNES context 14832541af92SBarry Smith 14842541af92SBarry Smith Output Parameter: 14852541af92SBarry Smith . nfuncs - number of evaluations 14862541af92SBarry Smith 14872541af92SBarry Smith Level: intermediate 14882541af92SBarry Smith 148995452b02SPatrick Sanan Notes: 149095452b02SPatrick Sanan Reset every time SNESSolve is called unless SNESSetCountersReset() is used. 1491971e163fSPeter Brune 1492db781477SPatrick Sanan .seealso: `SNESGetMaxLinearSolveFailures()`, `SNESGetLinearSolveIterations()`, `SNESSetMaxLinearSolveFailures()`, `SNESGetLinearSolveFailures()`, `SNESSetCountersReset()` 14932541af92SBarry Smith @*/ 14947087cfbeSBarry Smith PetscErrorCode SNESGetNumberFunctionEvals(SNES snes, PetscInt *nfuncs) 14952541af92SBarry Smith { 14962541af92SBarry Smith PetscFunctionBegin; 14970700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 14982541af92SBarry Smith PetscValidIntPointer(nfuncs,2); 14992541af92SBarry Smith *nfuncs = snes->nfuncs; 15002541af92SBarry Smith PetscFunctionReturn(0); 15012541af92SBarry Smith } 15022541af92SBarry Smith 15033d4c4710SBarry Smith /*@ 15043d4c4710SBarry Smith SNESGetLinearSolveFailures - Gets the number of failed (non-converged) 15053d4c4710SBarry Smith linear solvers. 15063d4c4710SBarry Smith 15073d4c4710SBarry Smith Not Collective 15083d4c4710SBarry Smith 15093d4c4710SBarry Smith Input Parameter: 15103d4c4710SBarry Smith . snes - SNES context 15113d4c4710SBarry Smith 15123d4c4710SBarry Smith Output Parameter: 15133d4c4710SBarry Smith . nfails - number of failed solves 15143d4c4710SBarry Smith 15159d85da0cSMatthew G. Knepley Level: intermediate 15169d85da0cSMatthew G. Knepley 15179d85da0cSMatthew G. Knepley Options Database Keys: 15189d85da0cSMatthew G. Knepley . -snes_max_linear_solve_fail <num> - The number of failures before the solve is terminated 15199d85da0cSMatthew G. Knepley 15203d4c4710SBarry Smith Notes: 15213d4c4710SBarry Smith This counter is reset to zero for each successive call to SNESSolve(). 15223d4c4710SBarry Smith 1523db781477SPatrick Sanan .seealso: `SNESGetMaxLinearSolveFailures()`, `SNESGetLinearSolveIterations()`, `SNESSetMaxLinearSolveFailures()` 15243d4c4710SBarry Smith @*/ 15257087cfbeSBarry Smith PetscErrorCode SNESGetLinearSolveFailures(SNES snes,PetscInt *nfails) 15263d4c4710SBarry Smith { 15273d4c4710SBarry Smith PetscFunctionBegin; 15280700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 15293d4c4710SBarry Smith PetscValidIntPointer(nfails,2); 15303d4c4710SBarry Smith *nfails = snes->numLinearSolveFailures; 15313d4c4710SBarry Smith PetscFunctionReturn(0); 15323d4c4710SBarry Smith } 15333d4c4710SBarry Smith 15343d4c4710SBarry Smith /*@ 15353d4c4710SBarry Smith SNESSetMaxLinearSolveFailures - the number of failed linear solve attempts 15363d4c4710SBarry Smith allowed before SNES returns with a diverged reason of SNES_DIVERGED_LINEAR_SOLVE 15373d4c4710SBarry Smith 15383f9fe445SBarry Smith Logically Collective on SNES 15393d4c4710SBarry Smith 15403d4c4710SBarry Smith Input Parameters: 15413d4c4710SBarry Smith + snes - SNES context 15423d4c4710SBarry Smith - maxFails - maximum allowed linear solve failures 15433d4c4710SBarry Smith 15443d4c4710SBarry Smith Level: intermediate 15453d4c4710SBarry Smith 15469d85da0cSMatthew G. Knepley Options Database Keys: 15479d85da0cSMatthew G. Knepley . -snes_max_linear_solve_fail <num> - The number of failures before the solve is terminated 15489d85da0cSMatthew G. Knepley 154995452b02SPatrick Sanan Notes: 155095452b02SPatrick Sanan By default this is 0; that is SNES returns on the first failed linear solve 15513d4c4710SBarry Smith 1552db781477SPatrick Sanan .seealso: `SNESGetLinearSolveFailures()`, `SNESGetMaxLinearSolveFailures()`, `SNESGetLinearSolveIterations()` 15533d4c4710SBarry Smith @*/ 15547087cfbeSBarry Smith PetscErrorCode SNESSetMaxLinearSolveFailures(SNES snes, PetscInt maxFails) 15553d4c4710SBarry Smith { 15563d4c4710SBarry Smith PetscFunctionBegin; 15570700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 1558c5eb9154SBarry Smith PetscValidLogicalCollectiveInt(snes,maxFails,2); 15593d4c4710SBarry Smith snes->maxLinearSolveFailures = maxFails; 15603d4c4710SBarry Smith PetscFunctionReturn(0); 15613d4c4710SBarry Smith } 15623d4c4710SBarry Smith 15633d4c4710SBarry Smith /*@ 15643d4c4710SBarry Smith SNESGetMaxLinearSolveFailures - gets the maximum number of linear solve failures that 15653d4c4710SBarry Smith are allowed before SNES terminates 15663d4c4710SBarry Smith 15673d4c4710SBarry Smith Not Collective 15683d4c4710SBarry Smith 15693d4c4710SBarry Smith Input Parameter: 15703d4c4710SBarry Smith . snes - SNES context 15713d4c4710SBarry Smith 15723d4c4710SBarry Smith Output Parameter: 15733d4c4710SBarry Smith . maxFails - maximum of unsuccessful solves allowed 15743d4c4710SBarry Smith 15753d4c4710SBarry Smith Level: intermediate 15763d4c4710SBarry Smith 157795452b02SPatrick Sanan Notes: 157895452b02SPatrick Sanan By default this is 1; that is SNES returns on the first failed linear solve 15793d4c4710SBarry Smith 1580db781477SPatrick Sanan .seealso: `SNESGetLinearSolveFailures()`, `SNESGetLinearSolveIterations()`, `SNESSetMaxLinearSolveFailures()`, 15813d4c4710SBarry Smith @*/ 15827087cfbeSBarry Smith PetscErrorCode SNESGetMaxLinearSolveFailures(SNES snes, PetscInt *maxFails) 15833d4c4710SBarry Smith { 15843d4c4710SBarry Smith PetscFunctionBegin; 15850700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 15863d4c4710SBarry Smith PetscValidIntPointer(maxFails,2); 15873d4c4710SBarry Smith *maxFails = snes->maxLinearSolveFailures; 15883d4c4710SBarry Smith PetscFunctionReturn(0); 15893d4c4710SBarry Smith } 15903d4c4710SBarry Smith 1591c96a6f78SLois Curfman McInnes /*@ 1592b850b91aSLisandro Dalcin SNESGetLinearSolveIterations - Gets the total number of linear iterations 1593c96a6f78SLois Curfman McInnes used by the nonlinear solver. 1594c96a6f78SLois Curfman McInnes 1595c7afd0dbSLois Curfman McInnes Not Collective 1596c7afd0dbSLois Curfman McInnes 1597c96a6f78SLois Curfman McInnes Input Parameter: 1598c96a6f78SLois Curfman McInnes . snes - SNES context 1599c96a6f78SLois Curfman McInnes 1600c96a6f78SLois Curfman McInnes Output Parameter: 1601c96a6f78SLois Curfman McInnes . lits - number of linear iterations 1602c96a6f78SLois Curfman McInnes 1603c96a6f78SLois Curfman McInnes Notes: 1604971e163fSPeter Brune This counter is reset to zero for each successive call to SNESSolve() unless SNESSetCountersReset() is used. 1605c96a6f78SLois Curfman McInnes 1606010be392SBarry 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 1607010be392SBarry Smith then call KSPGetIterationNumber() after the failed solve. 1608010be392SBarry Smith 160936851e7fSLois Curfman McInnes Level: intermediate 161036851e7fSLois Curfman McInnes 1611db781477SPatrick Sanan .seealso: `SNESGetIterationNumber()`, `SNESGetLinearSolveFailures()`, `SNESGetMaxLinearSolveFailures()`, `SNESSetCountersReset()` 1612c96a6f78SLois Curfman McInnes @*/ 16137087cfbeSBarry Smith PetscErrorCode SNESGetLinearSolveIterations(SNES snes,PetscInt *lits) 1614c96a6f78SLois Curfman McInnes { 16153a40ed3dSBarry Smith PetscFunctionBegin; 16160700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 16174482741eSBarry Smith PetscValidIntPointer(lits,2); 1618c96a6f78SLois Curfman McInnes *lits = snes->linear_its; 16193a40ed3dSBarry Smith PetscFunctionReturn(0); 1620c96a6f78SLois Curfman McInnes } 1621c96a6f78SLois Curfman McInnes 1622971e163fSPeter Brune /*@ 1623971e163fSPeter Brune SNESSetCountersReset - Sets whether or not the counters for linear iterations and function evaluations 1624971e163fSPeter Brune are reset every time SNESSolve() is called. 1625971e163fSPeter Brune 1626971e163fSPeter Brune Logically Collective on SNES 1627971e163fSPeter Brune 1628d8d19677SJose E. Roman Input Parameters: 1629971e163fSPeter Brune + snes - SNES context 1630971e163fSPeter Brune - reset - whether to reset the counters or not 1631971e163fSPeter Brune 1632971e163fSPeter Brune Notes: 1633fa19ca70SBarry Smith This defaults to PETSC_TRUE 1634971e163fSPeter Brune 1635971e163fSPeter Brune Level: developer 1636971e163fSPeter Brune 1637db781477SPatrick Sanan .seealso: `SNESGetNumberFunctionEvals()`, `SNESGetLinearSolveIterations()`, `SNESGetNPC()` 1638971e163fSPeter Brune @*/ 1639971e163fSPeter Brune PetscErrorCode SNESSetCountersReset(SNES snes,PetscBool reset) 1640971e163fSPeter Brune { 1641971e163fSPeter Brune PetscFunctionBegin; 1642971e163fSPeter Brune PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 1643971e163fSPeter Brune PetscValidLogicalCollectiveBool(snes,reset,2); 1644971e163fSPeter Brune snes->counters_reset = reset; 1645971e163fSPeter Brune PetscFunctionReturn(0); 1646971e163fSPeter Brune } 1647971e163fSPeter Brune 16482999313aSBarry Smith /*@ 16492999313aSBarry Smith SNESSetKSP - Sets a KSP context for the SNES object to use 16502999313aSBarry Smith 16512999313aSBarry Smith Not Collective, but the SNES and KSP objects must live on the same MPI_Comm 16522999313aSBarry Smith 16532999313aSBarry Smith Input Parameters: 16542999313aSBarry Smith + snes - the SNES context 16552999313aSBarry Smith - ksp - the KSP context 16562999313aSBarry Smith 16572999313aSBarry Smith Notes: 16582999313aSBarry Smith The SNES object already has its KSP object, you can obtain with SNESGetKSP() 16592999313aSBarry Smith so this routine is rarely needed. 16602999313aSBarry Smith 16612999313aSBarry Smith The KSP object that is already in the SNES object has its reference count 16622999313aSBarry Smith decreased by one. 16632999313aSBarry Smith 16642999313aSBarry Smith Level: developer 16652999313aSBarry Smith 1666db781477SPatrick Sanan .seealso: `KSPGetPC()`, `SNESCreate()`, `KSPCreate()`, `SNESSetKSP()` 16672999313aSBarry Smith @*/ 16687087cfbeSBarry Smith PetscErrorCode SNESSetKSP(SNES snes,KSP ksp) 16692999313aSBarry Smith { 16702999313aSBarry Smith PetscFunctionBegin; 16710700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 16720700a824SBarry Smith PetscValidHeaderSpecific(ksp,KSP_CLASSID,2); 16732999313aSBarry Smith PetscCheckSameComm(snes,1,ksp,2); 16749566063dSJacob Faibussowitsch PetscCall(PetscObjectReference((PetscObject)ksp)); 16759566063dSJacob Faibussowitsch if (snes->ksp) PetscCall(PetscObjectDereference((PetscObject)snes->ksp)); 16762999313aSBarry Smith snes->ksp = ksp; 16772999313aSBarry Smith PetscFunctionReturn(0); 16782999313aSBarry Smith } 16792999313aSBarry Smith 16809b94acceSBarry Smith /* -----------------------------------------------------------*/ 168152baeb72SSatish Balay /*@ 16829b94acceSBarry Smith SNESCreate - Creates a nonlinear solver context. 16839b94acceSBarry Smith 1684d083f849SBarry Smith Collective 1685c7afd0dbSLois Curfman McInnes 1686c7afd0dbSLois Curfman McInnes Input Parameters: 1687906ed7ccSBarry Smith . comm - MPI communicator 16889b94acceSBarry Smith 16899b94acceSBarry Smith Output Parameter: 16909b94acceSBarry Smith . outsnes - the new SNES context 16919b94acceSBarry Smith 1692c7afd0dbSLois Curfman McInnes Options Database Keys: 1693c7afd0dbSLois Curfman McInnes + -snes_mf - Activates default matrix-free Jacobian-vector products, 1694c7afd0dbSLois Curfman McInnes and no preconditioning matrix 1695c7afd0dbSLois Curfman McInnes . -snes_mf_operator - Activates default matrix-free Jacobian-vector 1696c7afd0dbSLois Curfman McInnes products, and a user-provided preconditioning matrix 1697c7afd0dbSLois Curfman McInnes as set by SNESSetJacobian() 1698c7afd0dbSLois Curfman McInnes - -snes_fd - Uses (slow!) finite differences to compute Jacobian 1699c1f60f51SBarry Smith 170036851e7fSLois Curfman McInnes Level: beginner 170136851e7fSLois Curfman McInnes 170295452b02SPatrick Sanan Developer Notes: 170395452b02SPatrick Sanan SNES always creates a KSP object even though many SNES methods do not use it. This is 1704efd4aadfSBarry Smith unfortunate and should be fixed at some point. The flag snes->usesksp indicates if the 1705efd4aadfSBarry Smith particular method does use KSP and regulates if the information about the KSP is printed 1706efd4aadfSBarry Smith in SNESView(). TSSetFromOptions() does call SNESSetFromOptions() which can lead to users being confused 1707efd4aadfSBarry Smith by help messages about meaningless SNES options. 1708efd4aadfSBarry Smith 1709efd4aadfSBarry Smith SNES always creates the snes->kspconvctx even though it is used by only one type. This should 1710efd4aadfSBarry Smith be fixed. 1711efd4aadfSBarry Smith 1712db781477SPatrick Sanan .seealso: `SNESSolve()`, `SNESDestroy()`, `SNES`, `SNESSetLagPreconditioner()`, `SNESSetLagJacobian()` 1713a8054027SBarry Smith 17149b94acceSBarry Smith @*/ 17157087cfbeSBarry Smith PetscErrorCode SNESCreate(MPI_Comm comm,SNES *outsnes) 17169b94acceSBarry Smith { 17179b94acceSBarry Smith SNES snes; 1718fa9f3622SBarry Smith SNESKSPEW *kctx; 171937fcc0dbSBarry Smith 17203a40ed3dSBarry Smith PetscFunctionBegin; 1721ed1caa07SMatthew Knepley PetscValidPointer(outsnes,2); 17220298fd71SBarry Smith *outsnes = NULL; 17239566063dSJacob Faibussowitsch PetscCall(SNESInitializePackage()); 17248ba1e511SMatthew Knepley 17259566063dSJacob Faibussowitsch PetscCall(PetscHeaderCreate(snes,SNES_CLASSID,"SNES","Nonlinear solver","SNES",comm,SNESDestroy,SNESView)); 17267adad957SLisandro Dalcin 17278d359177SBarry Smith snes->ops->converged = SNESConvergedDefault; 17282c155ee1SBarry Smith snes->usesksp = PETSC_TRUE; 172988976e71SPeter Brune snes->tolerancesset = PETSC_FALSE; 17309b94acceSBarry Smith snes->max_its = 50; 17319750a799SBarry Smith snes->max_funcs = 10000; 17329b94acceSBarry Smith snes->norm = 0.0; 1733c1e67a49SFande Kong snes->xnorm = 0.0; 1734c1e67a49SFande Kong snes->ynorm = 0.0; 1735365a6726SPeter Brune snes->normschedule = SNES_NORM_ALWAYS; 17366c67d002SPeter Brune snes->functype = SNES_FUNCTION_DEFAULT; 17373a2046daSBarry Smith #if defined(PETSC_USE_REAL_SINGLE) 17383a2046daSBarry Smith snes->rtol = 1.e-5; 17393a2046daSBarry Smith #else 1740b4874afaSBarry Smith snes->rtol = 1.e-8; 17413a2046daSBarry Smith #endif 1742b4874afaSBarry Smith snes->ttol = 0.0; 17433a2046daSBarry Smith #if defined(PETSC_USE_REAL_SINGLE) 17443a2046daSBarry Smith snes->abstol = 1.e-25; 17453a2046daSBarry Smith #else 174670441072SBarry Smith snes->abstol = 1.e-50; 17473a2046daSBarry Smith #endif 17487cd0ae37SLisandro Dalcin #if defined(PETSC_USE_REAL_SINGLE) 17497cd0ae37SLisandro Dalcin snes->stol = 1.e-5; 17507cd0ae37SLisandro Dalcin #else 1751c60f73f4SPeter Brune snes->stol = 1.e-8; 17527cd0ae37SLisandro Dalcin #endif 17533a2046daSBarry Smith #if defined(PETSC_USE_REAL_SINGLE) 17543a2046daSBarry Smith snes->deltatol = 1.e-6; 17553a2046daSBarry Smith #else 17564b27c08aSLois Curfman McInnes snes->deltatol = 1.e-12; 17573a2046daSBarry Smith #endif 1758e37c518bSBarry Smith snes->divtol = 1.e4; 1759e37c518bSBarry Smith snes->rnorm0 = 0; 17609b94acceSBarry Smith snes->nfuncs = 0; 176150ffb88aSMatthew Knepley snes->numFailures = 0; 176250ffb88aSMatthew Knepley snes->maxFailures = 1; 17637a00f4a9SLois Curfman McInnes snes->linear_its = 0; 1764e35cf81dSBarry Smith snes->lagjacobian = 1; 176537ec4e1aSPeter Brune snes->jac_iter = 0; 176637ec4e1aSPeter Brune snes->lagjac_persist = PETSC_FALSE; 1767a8054027SBarry Smith snes->lagpreconditioner = 1; 176837ec4e1aSPeter Brune snes->pre_iter = 0; 176937ec4e1aSPeter Brune snes->lagpre_persist = PETSC_FALSE; 1770639f9d9dSBarry Smith snes->numbermonitors = 0; 1771c4421ceaSFande Kong snes->numberreasonviews = 0; 17729e5d0892SLisandro Dalcin snes->data = NULL; 17734dc4c822SBarry Smith snes->setupcalled = PETSC_FALSE; 1774186905e3SBarry Smith snes->ksp_ewconv = PETSC_FALSE; 17756f24a144SLois Curfman McInnes snes->nwork = 0; 17769e5d0892SLisandro Dalcin snes->work = NULL; 177758c9b817SLisandro Dalcin snes->nvwork = 0; 17789e5d0892SLisandro Dalcin snes->vwork = NULL; 1779758f92a0SBarry Smith snes->conv_hist_len = 0; 1780758f92a0SBarry Smith snes->conv_hist_max = 0; 17810298fd71SBarry Smith snes->conv_hist = NULL; 17820298fd71SBarry Smith snes->conv_hist_its = NULL; 1783758f92a0SBarry Smith snes->conv_hist_reset = PETSC_TRUE; 1784971e163fSPeter Brune snes->counters_reset = PETSC_TRUE; 1785e4ed7901SPeter Brune snes->vec_func_init_set = PETSC_FALSE; 1786184914b5SBarry Smith snes->reason = SNES_CONVERGED_ITERATING; 1787efd4aadfSBarry Smith snes->npcside = PC_RIGHT; 1788b3cd9a81SMatthew G. Knepley snes->setfromoptionscalled = 0; 1789c40d0f55SPeter Brune 1790d8f46077SPeter Brune snes->mf = PETSC_FALSE; 1791d8f46077SPeter Brune snes->mf_operator = PETSC_FALSE; 1792d8f46077SPeter Brune snes->mf_version = 1; 1793d8f46077SPeter Brune 17943d4c4710SBarry Smith snes->numLinearSolveFailures = 0; 17953d4c4710SBarry Smith snes->maxLinearSolveFailures = 1; 17963d4c4710SBarry Smith 1797349187a7SBarry Smith snes->vizerotolerance = 1.e-8; 179876bd3646SJed Brown snes->checkjacdomainerror = PetscDefined(USE_DEBUG) ? PETSC_TRUE : PETSC_FALSE; 1799349187a7SBarry Smith 18004fc747eaSLawrence Mitchell /* Set this to true if the implementation of SNESSolve_XXX does compute the residual at the final solution. */ 18014fc747eaSLawrence Mitchell snes->alwayscomputesfinalresidual = PETSC_FALSE; 18024fc747eaSLawrence Mitchell 18039b94acceSBarry Smith /* Create context to compute Eisenstat-Walker relative tolerance for KSP */ 18049566063dSJacob Faibussowitsch PetscCall(PetscNewLog(snes,&kctx)); 1805f5af7f23SKarl Rupp 18069b94acceSBarry Smith snes->kspconvctx = (void*)kctx; 18079b94acceSBarry Smith kctx->version = 2; 18080f0abf79SStefano Zampini kctx->rtol_0 = 0.3; /* Eisenstat and Walker suggest rtol_0=.5, but 18099b94acceSBarry Smith this was too large for some test cases */ 181075567043SBarry Smith kctx->rtol_last = 0.0; 18110f0abf79SStefano Zampini kctx->rtol_max = 0.9; 18129b94acceSBarry Smith kctx->gamma = 1.0; 18130f0abf79SStefano Zampini kctx->alpha = 0.5*(1.0 + PetscSqrtReal(5.0)); 181471f87433Sdalcinl kctx->alpha2 = kctx->alpha; 18150f0abf79SStefano Zampini kctx->threshold = 0.1; 181675567043SBarry Smith kctx->lresid_last = 0.0; 181775567043SBarry Smith kctx->norm_last = 0.0; 18189b94acceSBarry Smith 18190f0abf79SStefano Zampini kctx->rk_last = 0.0; 18200f0abf79SStefano Zampini kctx->rk_last_2 = 0.0; 18210f0abf79SStefano Zampini kctx->rtol_last_2 = 0.0; 18220f0abf79SStefano Zampini kctx->v4_p1 = 0.1; 18230f0abf79SStefano Zampini kctx->v4_p2 = 0.4; 18240f0abf79SStefano Zampini kctx->v4_p3 = 0.7; 18250f0abf79SStefano Zampini kctx->v4_m1 = 0.8; 18260f0abf79SStefano Zampini kctx->v4_m2 = 0.5; 18270f0abf79SStefano Zampini kctx->v4_m3 = 0.1; 18280f0abf79SStefano Zampini kctx->v4_m4 = 0.5; 18290f0abf79SStefano Zampini 18309b94acceSBarry Smith *outsnes = snes; 18313a40ed3dSBarry Smith PetscFunctionReturn(0); 18329b94acceSBarry Smith } 18339b94acceSBarry Smith 183488f0584fSBarry Smith /*MC 1835411c0326SBarry Smith SNESFunction - Functional form used to convey the nonlinear function to be solved by SNES 183688f0584fSBarry Smith 183788f0584fSBarry Smith Synopsis: 1838411c0326SBarry Smith #include "petscsnes.h" 1839411c0326SBarry Smith PetscErrorCode SNESFunction(SNES snes,Vec x,Vec f,void *ctx); 184088f0584fSBarry Smith 18411843f636SBarry Smith Collective on snes 18421843f636SBarry Smith 184388f0584fSBarry Smith Input Parameters: 184488f0584fSBarry Smith + snes - the SNES context 184588f0584fSBarry Smith . x - state at which to evaluate residual 184688f0584fSBarry Smith - ctx - optional user-defined function context, passed in with SNESSetFunction() 184788f0584fSBarry Smith 184888f0584fSBarry Smith Output Parameter: 184988f0584fSBarry Smith . f - vector to put residual (function value) 185088f0584fSBarry Smith 1851878cb397SSatish Balay Level: intermediate 1852878cb397SSatish Balay 1853db781477SPatrick Sanan .seealso: `SNESSetFunction()`, `SNESGetFunction()` 185488f0584fSBarry Smith M*/ 185588f0584fSBarry Smith 18569b94acceSBarry Smith /*@C 18579b94acceSBarry Smith SNESSetFunction - Sets the function evaluation routine and function 18589b94acceSBarry Smith vector for use by the SNES routines in solving systems of nonlinear 18599b94acceSBarry Smith equations. 18609b94acceSBarry Smith 18613f9fe445SBarry Smith Logically Collective on SNES 1862fee21e36SBarry Smith 1863c7afd0dbSLois Curfman McInnes Input Parameters: 1864c7afd0dbSLois Curfman McInnes + snes - the SNES context 18656b7fb656SBarry Smith . r - vector to store function values, may be NULL 1866f8b49ee9SBarry Smith . f - function evaluation routine; see SNESFunction for calling sequence details 1867c7afd0dbSLois Curfman McInnes - ctx - [optional] user-defined context for private data for the 18680298fd71SBarry Smith function evaluation routine (may be NULL) 18699b94acceSBarry Smith 18709b94acceSBarry Smith Notes: 18719b94acceSBarry Smith The Newton-like methods typically solve linear systems of the form 18729b94acceSBarry Smith $ f'(x) x = -f(x), 1873c7afd0dbSLois Curfman McInnes where f'(x) denotes the Jacobian matrix and f(x) is the function. 18749b94acceSBarry Smith 187536851e7fSLois Curfman McInnes Level: beginner 187636851e7fSLois Curfman McInnes 1877db781477SPatrick Sanan .seealso: `SNESGetFunction()`, `SNESComputeFunction()`, `SNESSetJacobian()`, `SNESSetPicard()`, `SNESFunction` 18789b94acceSBarry Smith @*/ 1879f8b49ee9SBarry Smith PetscErrorCode SNESSetFunction(SNES snes,Vec r,PetscErrorCode (*f)(SNES,Vec,Vec,void*),void *ctx) 18809b94acceSBarry Smith { 18816cab3a1bSJed Brown DM dm; 18826cab3a1bSJed Brown 18833a40ed3dSBarry Smith PetscFunctionBegin; 18840700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 1885d2a683ecSLisandro Dalcin if (r) { 1886d2a683ecSLisandro Dalcin PetscValidHeaderSpecific(r,VEC_CLASSID,2); 1887d2a683ecSLisandro Dalcin PetscCheckSameComm(snes,1,r,2); 18889566063dSJacob Faibussowitsch PetscCall(PetscObjectReference((PetscObject)r)); 18899566063dSJacob Faibussowitsch PetscCall(VecDestroy(&snes->vec_func)); 189085385478SLisandro Dalcin snes->vec_func = r; 1891d2a683ecSLisandro Dalcin } 18929566063dSJacob Faibussowitsch PetscCall(SNESGetDM(snes,&dm)); 18939566063dSJacob Faibussowitsch PetscCall(DMSNESSetFunction(dm,f,ctx)); 1894bbc1464cSBarry Smith if (f == SNESPicardComputeFunction) { 18959566063dSJacob Faibussowitsch PetscCall(DMSNESSetMFFunction(dm,SNESPicardComputeMFFunction,ctx)); 1896bbc1464cSBarry Smith } 18973a40ed3dSBarry Smith PetscFunctionReturn(0); 18989b94acceSBarry Smith } 18999b94acceSBarry Smith 1900e4ed7901SPeter Brune /*@C 1901e4ed7901SPeter Brune SNESSetInitialFunction - Sets the function vector to be used as the 1902e4ed7901SPeter Brune function norm at the initialization of the method. In some 1903e4ed7901SPeter Brune instances, the user has precomputed the function before calling 1904e4ed7901SPeter Brune SNESSolve. This function allows one to avoid a redundant call 1905e4ed7901SPeter Brune to SNESComputeFunction in that case. 1906e4ed7901SPeter Brune 1907e4ed7901SPeter Brune Logically Collective on SNES 1908e4ed7901SPeter Brune 1909e4ed7901SPeter Brune Input Parameters: 1910e4ed7901SPeter Brune + snes - the SNES context 1911e4ed7901SPeter Brune - f - vector to store function value 1912e4ed7901SPeter Brune 1913e4ed7901SPeter Brune Notes: 1914e4ed7901SPeter Brune This should not be modified during the solution procedure. 1915e4ed7901SPeter Brune 1916e4ed7901SPeter Brune This is used extensively in the SNESFAS hierarchy and in nonlinear preconditioning. 1917e4ed7901SPeter Brune 1918e4ed7901SPeter Brune Level: developer 1919e4ed7901SPeter Brune 1920db781477SPatrick Sanan .seealso: `SNESSetFunction()`, `SNESComputeFunction()`, `SNESSetInitialFunctionNorm()` 1921e4ed7901SPeter Brune @*/ 1922e4ed7901SPeter Brune PetscErrorCode SNESSetInitialFunction(SNES snes, Vec f) 1923e4ed7901SPeter Brune { 1924e4ed7901SPeter Brune Vec vec_func; 1925e4ed7901SPeter Brune 1926e4ed7901SPeter Brune PetscFunctionBegin; 1927e4ed7901SPeter Brune PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 1928e4ed7901SPeter Brune PetscValidHeaderSpecific(f,VEC_CLASSID,2); 1929e4ed7901SPeter Brune PetscCheckSameComm(snes,1,f,2); 1930efd4aadfSBarry Smith if (snes->npcside == PC_LEFT && snes->functype == SNES_FUNCTION_PRECONDITIONED) { 1931902f982fSPeter Brune snes->vec_func_init_set = PETSC_FALSE; 1932902f982fSPeter Brune PetscFunctionReturn(0); 1933902f982fSPeter Brune } 19349566063dSJacob Faibussowitsch PetscCall(SNESGetFunction(snes,&vec_func,NULL,NULL)); 19359566063dSJacob Faibussowitsch PetscCall(VecCopy(f,vec_func)); 1936f5af7f23SKarl Rupp 1937217b9c2eSPeter Brune snes->vec_func_init_set = PETSC_TRUE; 1938e4ed7901SPeter Brune PetscFunctionReturn(0); 1939e4ed7901SPeter Brune } 1940e4ed7901SPeter Brune 1941534ebe21SPeter Brune /*@ 1942a5b23f4aSJose E. Roman SNESSetNormSchedule - Sets the SNESNormSchedule used in convergence and monitoring 1943534ebe21SPeter Brune of the SNES method. 1944534ebe21SPeter Brune 1945534ebe21SPeter Brune Logically Collective on SNES 1946534ebe21SPeter Brune 1947534ebe21SPeter Brune Input Parameters: 1948534ebe21SPeter Brune + snes - the SNES context 1949365a6726SPeter Brune - normschedule - the frequency of norm computation 1950534ebe21SPeter Brune 1951517f1916SMatthew G. Knepley Options Database Key: 195267b8a455SSatish Balay . -snes_norm_schedule <none, always, initialonly, finalonly, initialfinalonly> - set the schedule 1953517f1916SMatthew G. Knepley 1954534ebe21SPeter Brune Notes: 1955365a6726SPeter Brune Only certain SNES methods support certain SNESNormSchedules. Most require evaluation 1956534ebe21SPeter Brune of the nonlinear function and the taking of its norm at every iteration to 1957534ebe21SPeter Brune even ensure convergence at all. However, methods such as custom Gauss-Seidel methods 1958a5b23f4aSJose E. Roman (SNESNGS) and the like do not require the norm of the function to be computed, and therefore 1959534ebe21SPeter Brune may either be monitored for convergence or not. As these are often used as nonlinear 1960534ebe21SPeter Brune preconditioners, monitoring the norm of their error is not a useful enterprise within 1961534ebe21SPeter Brune their solution. 1962534ebe21SPeter Brune 1963534ebe21SPeter Brune Level: developer 1964534ebe21SPeter Brune 1965db781477SPatrick Sanan .seealso: `SNESGetNormSchedule()`, `SNESComputeFunction()`, `VecNorm()`, `SNESSetFunction()`, `SNESSetInitialFunction()`, `SNESNormSchedule` 1966534ebe21SPeter Brune @*/ 1967365a6726SPeter Brune PetscErrorCode SNESSetNormSchedule(SNES snes, SNESNormSchedule normschedule) 1968534ebe21SPeter Brune { 1969534ebe21SPeter Brune PetscFunctionBegin; 1970534ebe21SPeter Brune PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 1971365a6726SPeter Brune snes->normschedule = normschedule; 1972534ebe21SPeter Brune PetscFunctionReturn(0); 1973534ebe21SPeter Brune } 1974534ebe21SPeter Brune 1975534ebe21SPeter Brune /*@ 1976a5b23f4aSJose E. Roman SNESGetNormSchedule - Gets the SNESNormSchedule used in convergence and monitoring 1977534ebe21SPeter Brune of the SNES method. 1978534ebe21SPeter Brune 1979534ebe21SPeter Brune Logically Collective on SNES 1980534ebe21SPeter Brune 1981534ebe21SPeter Brune Input Parameters: 1982534ebe21SPeter Brune + snes - the SNES context 1983365a6726SPeter Brune - normschedule - the type of the norm used 1984534ebe21SPeter Brune 1985534ebe21SPeter Brune Level: advanced 1986534ebe21SPeter Brune 1987db781477SPatrick Sanan .seealso: `SNESSetNormSchedule()`, `SNESComputeFunction()`, `VecNorm()`, `SNESSetFunction()`, `SNESSetInitialFunction()`, `SNESNormSchedule` 1988534ebe21SPeter Brune @*/ 1989365a6726SPeter Brune PetscErrorCode SNESGetNormSchedule(SNES snes, SNESNormSchedule *normschedule) 1990534ebe21SPeter Brune { 1991534ebe21SPeter Brune PetscFunctionBegin; 1992534ebe21SPeter Brune PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 1993365a6726SPeter Brune *normschedule = snes->normschedule; 1994534ebe21SPeter Brune PetscFunctionReturn(0); 1995534ebe21SPeter Brune } 1996534ebe21SPeter Brune 1997c5ce4427SMatthew G. Knepley /*@ 1998c5ce4427SMatthew G. Knepley SNESSetFunctionNorm - Sets the last computed residual norm. 1999c5ce4427SMatthew G. Knepley 2000c5ce4427SMatthew G. Knepley Logically Collective on SNES 2001c5ce4427SMatthew G. Knepley 2002c5ce4427SMatthew G. Knepley Input Parameters: 2003c5ce4427SMatthew G. Knepley + snes - the SNES context 2004c5ce4427SMatthew G. Knepley 2005c5ce4427SMatthew G. Knepley - normschedule - the frequency of norm computation 2006c5ce4427SMatthew G. Knepley 2007c5ce4427SMatthew G. Knepley Level: developer 2008c5ce4427SMatthew G. Knepley 2009db781477SPatrick Sanan .seealso: `SNESGetNormSchedule()`, `SNESComputeFunction()`, `VecNorm()`, `SNESSetFunction()`, `SNESSetInitialFunction()`, `SNESNormSchedule` 2010c5ce4427SMatthew G. Knepley @*/ 2011c5ce4427SMatthew G. Knepley PetscErrorCode SNESSetFunctionNorm(SNES snes, PetscReal norm) 2012c5ce4427SMatthew G. Knepley { 2013c5ce4427SMatthew G. Knepley PetscFunctionBegin; 2014c5ce4427SMatthew G. Knepley PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 2015c5ce4427SMatthew G. Knepley snes->norm = norm; 2016c5ce4427SMatthew G. Knepley PetscFunctionReturn(0); 2017c5ce4427SMatthew G. Knepley } 2018c5ce4427SMatthew G. Knepley 2019c5ce4427SMatthew G. Knepley /*@ 2020c5ce4427SMatthew G. Knepley SNESGetFunctionNorm - Gets the last computed norm of the residual 2021c5ce4427SMatthew G. Knepley 2022c5ce4427SMatthew G. Knepley Not Collective 2023c5ce4427SMatthew G. Knepley 2024c5ce4427SMatthew G. Knepley Input Parameter: 2025c5ce4427SMatthew G. Knepley . snes - the SNES context 2026c5ce4427SMatthew G. Knepley 2027c5ce4427SMatthew G. Knepley Output Parameter: 2028c5ce4427SMatthew G. Knepley . norm - the last computed residual norm 2029c5ce4427SMatthew G. Knepley 2030c5ce4427SMatthew G. Knepley Level: developer 2031c5ce4427SMatthew G. Knepley 2032db781477SPatrick Sanan .seealso: `SNESSetNormSchedule()`, `SNESComputeFunction()`, `VecNorm()`, `SNESSetFunction()`, `SNESSetInitialFunction()`, `SNESNormSchedule` 2033c5ce4427SMatthew G. Knepley @*/ 2034c5ce4427SMatthew G. Knepley PetscErrorCode SNESGetFunctionNorm(SNES snes, PetscReal *norm) 2035c5ce4427SMatthew G. Knepley { 2036c5ce4427SMatthew G. Knepley PetscFunctionBegin; 2037c5ce4427SMatthew G. Knepley PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 2038dadcf809SJacob Faibussowitsch PetscValidRealPointer(norm, 2); 2039c5ce4427SMatthew G. Knepley *norm = snes->norm; 2040c5ce4427SMatthew G. Knepley PetscFunctionReturn(0); 2041c5ce4427SMatthew G. Knepley } 2042c5ce4427SMatthew G. Knepley 2043c1e67a49SFande Kong /*@ 2044c1e67a49SFande Kong SNESGetUpdateNorm - Gets the last computed norm of the Newton update 2045c1e67a49SFande Kong 2046c1e67a49SFande Kong Not Collective 2047c1e67a49SFande Kong 2048c1e67a49SFande Kong Input Parameter: 2049c1e67a49SFande Kong . snes - the SNES context 2050c1e67a49SFande Kong 2051c1e67a49SFande Kong Output Parameter: 2052c1e67a49SFande Kong . ynorm - the last computed update norm 2053c1e67a49SFande Kong 2054c1e67a49SFande Kong Level: developer 2055c1e67a49SFande Kong 2056db781477SPatrick Sanan .seealso: `SNESSetNormSchedule()`, `SNESComputeFunction()`, `SNESGetFunctionNorm()` 2057c1e67a49SFande Kong @*/ 2058c1e67a49SFande Kong PetscErrorCode SNESGetUpdateNorm(SNES snes, PetscReal *ynorm) 2059c1e67a49SFande Kong { 2060c1e67a49SFande Kong PetscFunctionBegin; 2061c1e67a49SFande Kong PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 2062dadcf809SJacob Faibussowitsch PetscValidRealPointer(ynorm, 2); 2063c1e67a49SFande Kong *ynorm = snes->ynorm; 2064c1e67a49SFande Kong PetscFunctionReturn(0); 2065c1e67a49SFande Kong } 2066c1e67a49SFande Kong 2067c1e67a49SFande Kong /*@ 20684591eaf2SFande Kong SNESGetSolutionNorm - Gets the last computed norm of the solution 2069c1e67a49SFande Kong 2070c1e67a49SFande Kong Not Collective 2071c1e67a49SFande Kong 2072c1e67a49SFande Kong Input Parameter: 2073c1e67a49SFande Kong . snes - the SNES context 2074c1e67a49SFande Kong 2075c1e67a49SFande Kong Output Parameter: 2076c1e67a49SFande Kong . xnorm - the last computed solution norm 2077c1e67a49SFande Kong 2078c1e67a49SFande Kong Level: developer 2079c1e67a49SFande Kong 2080db781477SPatrick Sanan .seealso: `SNESSetNormSchedule()`, `SNESComputeFunction()`, `SNESGetFunctionNorm()`, `SNESGetUpdateNorm()` 2081c1e67a49SFande Kong @*/ 2082c1e67a49SFande Kong PetscErrorCode SNESGetSolutionNorm(SNES snes, PetscReal *xnorm) 2083c1e67a49SFande Kong { 2084c1e67a49SFande Kong PetscFunctionBegin; 2085c1e67a49SFande Kong PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 2086dadcf809SJacob Faibussowitsch PetscValidRealPointer(xnorm, 2); 2087c1e67a49SFande Kong *xnorm = snes->xnorm; 2088c1e67a49SFande Kong PetscFunctionReturn(0); 2089c1e67a49SFande Kong } 2090c1e67a49SFande Kong 209147073ea2SPeter Brune /*@C 2092a5b23f4aSJose E. Roman SNESSetFunctionType - Sets the SNESNormSchedule used in convergence and monitoring 209347073ea2SPeter Brune of the SNES method. 209447073ea2SPeter Brune 209547073ea2SPeter Brune Logically Collective on SNES 209647073ea2SPeter Brune 209747073ea2SPeter Brune Input Parameters: 209847073ea2SPeter Brune + snes - the SNES context 209947073ea2SPeter Brune - normschedule - the frequency of norm computation 210047073ea2SPeter Brune 210147073ea2SPeter Brune Notes: 210247073ea2SPeter Brune Only certain SNES methods support certain SNESNormSchedules. Most require evaluation 210347073ea2SPeter Brune of the nonlinear function and the taking of its norm at every iteration to 210447073ea2SPeter Brune even ensure convergence at all. However, methods such as custom Gauss-Seidel methods 2105a5b23f4aSJose E. Roman (SNESNGS) and the like do not require the norm of the function to be computed, and therefore 210647073ea2SPeter Brune may either be monitored for convergence or not. As these are often used as nonlinear 210747073ea2SPeter Brune preconditioners, monitoring the norm of their error is not a useful enterprise within 210847073ea2SPeter Brune their solution. 210947073ea2SPeter Brune 211047073ea2SPeter Brune Level: developer 211147073ea2SPeter Brune 2112db781477SPatrick Sanan .seealso: `SNESGetNormSchedule()`, `SNESComputeFunction()`, `VecNorm()`, `SNESSetFunction()`, `SNESSetInitialFunction()`, `SNESNormSchedule` 211347073ea2SPeter Brune @*/ 211447073ea2SPeter Brune PetscErrorCode SNESSetFunctionType(SNES snes, SNESFunctionType type) 211547073ea2SPeter Brune { 211647073ea2SPeter Brune PetscFunctionBegin; 211747073ea2SPeter Brune PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 211847073ea2SPeter Brune snes->functype = type; 211947073ea2SPeter Brune PetscFunctionReturn(0); 212047073ea2SPeter Brune } 212147073ea2SPeter Brune 212247073ea2SPeter Brune /*@C 2123a5b23f4aSJose E. Roman SNESGetFunctionType - Gets the SNESNormSchedule used in convergence and monitoring 212447073ea2SPeter Brune of the SNES method. 212547073ea2SPeter Brune 212647073ea2SPeter Brune Logically Collective on SNES 212747073ea2SPeter Brune 212847073ea2SPeter Brune Input Parameters: 212947073ea2SPeter Brune + snes - the SNES context 213047073ea2SPeter Brune - normschedule - the type of the norm used 213147073ea2SPeter Brune 213247073ea2SPeter Brune Level: advanced 213347073ea2SPeter Brune 2134db781477SPatrick Sanan .seealso: `SNESSetNormSchedule()`, `SNESComputeFunction()`, `VecNorm()`, `SNESSetFunction()`, `SNESSetInitialFunction()`, `SNESNormSchedule` 213547073ea2SPeter Brune @*/ 213647073ea2SPeter Brune PetscErrorCode SNESGetFunctionType(SNES snes, SNESFunctionType *type) 213747073ea2SPeter Brune { 213847073ea2SPeter Brune PetscFunctionBegin; 213947073ea2SPeter Brune PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 214047073ea2SPeter Brune *type = snes->functype; 2141534ebe21SPeter Brune PetscFunctionReturn(0); 2142534ebe21SPeter Brune } 2143534ebe21SPeter Brune 2144bf388a1fSBarry Smith /*MC 2145be95d8f1SBarry Smith SNESNGSFunction - function used to convey a Gauss-Seidel sweep on the nonlinear function 2146bf388a1fSBarry Smith 2147bf388a1fSBarry Smith Synopsis: 2148aaa7dc30SBarry Smith #include <petscsnes.h> 2149be95d8f1SBarry Smith $ SNESNGSFunction(SNES snes,Vec x,Vec b,void *ctx); 2150bf388a1fSBarry Smith 21511843f636SBarry Smith Collective on snes 21521843f636SBarry Smith 21531843f636SBarry Smith Input Parameters: 2154bf388a1fSBarry Smith + X - solution vector 2155bf388a1fSBarry Smith . B - RHS vector 2156bf388a1fSBarry Smith - ctx - optional user-defined Gauss-Seidel context 2157bf388a1fSBarry Smith 21581843f636SBarry Smith Output Parameter: 21591843f636SBarry Smith . X - solution vector 21601843f636SBarry Smith 2161878cb397SSatish Balay Level: intermediate 2162878cb397SSatish Balay 2163db781477SPatrick Sanan .seealso: `SNESSetNGS()`, `SNESGetNGS()` 2164bf388a1fSBarry Smith M*/ 2165bf388a1fSBarry Smith 2166c79ef259SPeter Brune /*@C 2167be95d8f1SBarry Smith SNESSetNGS - Sets the user nonlinear Gauss-Seidel routine for 2168c79ef259SPeter Brune use with composed nonlinear solvers. 2169c79ef259SPeter Brune 2170c79ef259SPeter Brune Input Parameters: 2171c79ef259SPeter Brune + snes - the SNES context 2172be95d8f1SBarry Smith . f - function evaluation routine to apply Gauss-Seidel see SNESNGSFunction 2173c79ef259SPeter Brune - ctx - [optional] user-defined context for private data for the 21740298fd71SBarry Smith smoother evaluation routine (may be NULL) 2175c79ef259SPeter Brune 2176c79ef259SPeter Brune Notes: 2177be95d8f1SBarry Smith The NGS routines are used by the composed nonlinear solver to generate 2178c79ef259SPeter Brune a problem appropriate update to the solution, particularly FAS. 2179c79ef259SPeter Brune 2180d28543b3SPeter Brune Level: intermediate 2181c79ef259SPeter Brune 2182db781477SPatrick Sanan .seealso: `SNESGetFunction()`, `SNESComputeNGS()` 2183c79ef259SPeter Brune @*/ 2184be95d8f1SBarry Smith PetscErrorCode SNESSetNGS(SNES snes,PetscErrorCode (*f)(SNES,Vec,Vec,void*),void *ctx) 21856cab3a1bSJed Brown { 21866cab3a1bSJed Brown DM dm; 21876cab3a1bSJed Brown 2188646217ecSPeter Brune PetscFunctionBegin; 21896cab3a1bSJed Brown PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 21909566063dSJacob Faibussowitsch PetscCall(SNESGetDM(snes,&dm)); 21919566063dSJacob Faibussowitsch PetscCall(DMSNESSetNGS(dm,f,ctx)); 2192646217ecSPeter Brune PetscFunctionReturn(0); 2193646217ecSPeter Brune } 2194646217ecSPeter Brune 2195bbc1464cSBarry Smith /* 2196bbc1464cSBarry Smith This is used for -snes_mf_operator; it uses a duplicate of snes->jacobian_pre because snes->jacobian_pre cannot be 2197bbc1464cSBarry Smith changed during the KSPSolve() 2198bbc1464cSBarry Smith */ 2199bbc1464cSBarry Smith PetscErrorCode SNESPicardComputeMFFunction(SNES snes,Vec x,Vec f,void *ctx) 2200bbc1464cSBarry Smith { 2201bbc1464cSBarry Smith DM dm; 2202bbc1464cSBarry Smith DMSNES sdm; 2203bbc1464cSBarry Smith 2204bbc1464cSBarry Smith PetscFunctionBegin; 22059566063dSJacob Faibussowitsch PetscCall(SNESGetDM(snes,&dm)); 22069566063dSJacob Faibussowitsch PetscCall(DMGetDMSNES(dm,&sdm)); 2207bbc1464cSBarry Smith /* A(x)*x - b(x) */ 2208bbc1464cSBarry Smith if (sdm->ops->computepfunction) { 2209792fecdfSBarry Smith PetscCallBack("SNES Picard callback function",(*sdm->ops->computepfunction)(snes,x,f,sdm->pctx)); 22109566063dSJacob Faibussowitsch PetscCall(VecScale(f,-1.0)); 22110df40c35SBarry Smith /* Cannot share nonzero pattern because of the possible use of SNESComputeJacobianDefault() */ 2212ef1023bdSBarry Smith if (!snes->picard) PetscCall(MatDuplicate(snes->jacobian_pre,MAT_DO_NOT_COPY_VALUES,&snes->picard)); 2213792fecdfSBarry Smith PetscCallBack("SNES Picard callback Jacobian",(*sdm->ops->computepjacobian)(snes,x,snes->picard,snes->picard,sdm->pctx)); 22149566063dSJacob Faibussowitsch PetscCall(MatMultAdd(snes->picard,x,f,f)); 2215bbc1464cSBarry Smith } else { 2216792fecdfSBarry Smith PetscCallBack("SNES Picard callback Jacobian",(*sdm->ops->computepjacobian)(snes,x,snes->picard,snes->picard,sdm->pctx)); 22179566063dSJacob Faibussowitsch PetscCall(MatMult(snes->picard,x,f)); 2218bbc1464cSBarry Smith } 2219bbc1464cSBarry Smith PetscFunctionReturn(0); 2220bbc1464cSBarry Smith } 2221bbc1464cSBarry Smith 222225acbd8eSLisandro Dalcin PetscErrorCode SNESPicardComputeFunction(SNES snes,Vec x,Vec f,void *ctx) 22238b0a5094SBarry Smith { 2224e03ab78fSPeter Brune DM dm; 2225942e3340SBarry Smith DMSNES sdm; 22266cab3a1bSJed Brown 22278b0a5094SBarry Smith PetscFunctionBegin; 22289566063dSJacob Faibussowitsch PetscCall(SNESGetDM(snes,&dm)); 22299566063dSJacob Faibussowitsch PetscCall(DMGetDMSNES(dm,&sdm)); 22308b0a5094SBarry Smith /* A(x)*x - b(x) */ 2231bbc1464cSBarry Smith if (sdm->ops->computepfunction) { 2232792fecdfSBarry Smith PetscCallBack("SNES Picard callback function",(*sdm->ops->computepfunction)(snes,x,f,sdm->pctx)); 22339566063dSJacob Faibussowitsch PetscCall(VecScale(f,-1.0)); 2234792fecdfSBarry Smith PetscCallBack("SNES Picard callback Jacobian",(*sdm->ops->computepjacobian)(snes,x,snes->jacobian,snes->jacobian_pre,sdm->pctx)); 22359566063dSJacob Faibussowitsch PetscCall(MatMultAdd(snes->jacobian_pre,x,f,f)); 2236bbc1464cSBarry Smith } else { 2237792fecdfSBarry Smith PetscCallBack("SNES Picard callback Jacobian",(*sdm->ops->computepjacobian)(snes,x,snes->jacobian,snes->jacobian_pre,sdm->pctx)); 22389566063dSJacob Faibussowitsch PetscCall(MatMult(snes->jacobian_pre,x,f)); 2239bbc1464cSBarry Smith } 22408b0a5094SBarry Smith PetscFunctionReturn(0); 22418b0a5094SBarry Smith } 22428b0a5094SBarry Smith 224325acbd8eSLisandro Dalcin PetscErrorCode SNESPicardComputeJacobian(SNES snes,Vec x1,Mat J,Mat B,void *ctx) 22448b0a5094SBarry Smith { 22458b0a5094SBarry Smith PetscFunctionBegin; 2246e03ab78fSPeter Brune /* the jacobian matrix should be pre-filled in SNESPicardComputeFunction */ 2247bbc1464cSBarry Smith /* must assembly if matrix-free to get the last SNES solution */ 22489566063dSJacob Faibussowitsch PetscCall(MatAssemblyBegin(J,MAT_FINAL_ASSEMBLY)); 22499566063dSJacob Faibussowitsch PetscCall(MatAssemblyEnd(J,MAT_FINAL_ASSEMBLY)); 22508b0a5094SBarry Smith PetscFunctionReturn(0); 22518b0a5094SBarry Smith } 22528b0a5094SBarry Smith 22538b0a5094SBarry Smith /*@C 2254bbc1464cSBarry Smith SNESSetPicard - Use SNES to solve the system A(x) x = bp(x) + b via a Picard type iteration (Picard linearization) 22558b0a5094SBarry Smith 22568b0a5094SBarry Smith Logically Collective on SNES 22578b0a5094SBarry Smith 22588b0a5094SBarry Smith Input Parameters: 22598b0a5094SBarry Smith + snes - the SNES context 22606b7fb656SBarry Smith . r - vector to store function values, may be NULL 22616b7fb656SBarry Smith . bp - function evaluation routine, may be NULL 22626b7fb656SBarry Smith . Amat - matrix with which A(x) x - bp(x) - b is to be computed 2263e5d3d808SBarry Smith . Pmat - matrix from which preconditioner is computed (usually the same as Amat) 22646b7fb656SBarry Smith . J - function to compute matrix values, see SNESJacobianFunction() for details on its calling sequence 22656b7fb656SBarry Smith - ctx - [optional] user-defined context for private data for the function evaluation routine (may be NULL) 22668b0a5094SBarry Smith 22678b0a5094SBarry Smith Notes: 22686b7fb656SBarry Smith It is often better to provide the nonlinear function F() and some approximation to its Jacobian directly and use 2269f450aa47SBarry 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. 2270f450aa47SBarry Smith 22718b0a5094SBarry Smith One can call SNESSetPicard() or SNESSetFunction() (and possibly SNESSetJacobian()) but cannot call both 22728b0a5094SBarry Smith 22736b7fb656SBarry Smith $ Solves the equation A(x) x = bp(x) - b via the defect correction algorithm A(x^{n}) (x^{n+1} - x^{n}) = bp(x^{n}) + b - A(x^{n})x^{n} 22746b7fb656SBarry Smith $ Note that when an exact solver is used this corresponds to the "classic" Picard A(x^{n}) x^{n+1} = bp(x^{n}) + b iteration. 22758b0a5094SBarry Smith 22768b0a5094SBarry Smith Run with -snes_mf_operator to solve the system with Newton's method using A(x^{n}) to construct the preconditioner. 22778b0a5094SBarry Smith 22780d04baf8SBarry Smith We implement the defect correction form of the Picard iteration because it converges much more generally when inexact linear solvers are used then 22796b7fb656SBarry Smith the direct Picard iteration A(x^n) x^{n+1} = bp(x^n) + b 22808b0a5094SBarry Smith 22818b0a5094SBarry 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 22828b0a5094SBarry 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 22838b0a5094SBarry Smith different please contact us at petsc-dev@mcs.anl.gov and we'll have an entirely new argument :-). 22848b0a5094SBarry Smith 22856b7fb656SBarry Smith When used with -snes_mf_operator this will run matrix-free Newton's method where the matrix-vector product is of the true Jacobian of A(x)x - bp(x) -b. 22866b7fb656SBarry Smith 22876b7fb656SBarry Smith When used with -snes_fd this will compute the true Jacobian (very slowly one column at at time) and thus represent Newton's method. 22886b7fb656SBarry Smith 22896b7fb656SBarry Smith When used with -snes_fd_coloring this will compute the Jacobian via coloring and thus represent a faster implementation of Newton's method. But the 22906b7fb656SBarry Smith the nonzero structure of the Jacobian is, in general larger than that of the Picard matrix A so you must provide in A the needed nonzero structure for the correct 22916b7fb656SBarry Smith coloring. When using DMDA this may mean creating the matrix A with DMCreateMatrix() using a wider stencil than strictly needed for A or with a DMDA_STENCIL_BOX. 22926b7fb656SBarry Smith See the commment in src/snes/tutorials/ex15.c. 2293bbc1464cSBarry Smith 2294f450aa47SBarry Smith Level: intermediate 22958b0a5094SBarry Smith 2296db781477SPatrick Sanan .seealso: `SNESGetFunction()`, `SNESSetFunction()`, `SNESComputeFunction()`, `SNESSetJacobian()`, `SNESGetPicard()`, `SNESLineSearchPreCheckPicard()`, `SNESJacobianFunction` 22978b0a5094SBarry Smith @*/ 2298bbc1464cSBarry Smith PetscErrorCode SNESSetPicard(SNES snes,Vec r,PetscErrorCode (*bp)(SNES,Vec,Vec,void*),Mat Amat, Mat Pmat, PetscErrorCode (*J)(SNES,Vec,Mat,Mat,void*),void *ctx) 22998b0a5094SBarry Smith { 2300e03ab78fSPeter Brune DM dm; 2301e03ab78fSPeter Brune 23028b0a5094SBarry Smith PetscFunctionBegin; 23038b0a5094SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 23049566063dSJacob Faibussowitsch PetscCall(SNESGetDM(snes, &dm)); 23059566063dSJacob Faibussowitsch PetscCall(DMSNESSetPicard(dm,bp,J,ctx)); 23069566063dSJacob Faibussowitsch PetscCall(DMSNESSetMFFunction(dm,SNESPicardComputeMFFunction,ctx)); 23079566063dSJacob Faibussowitsch PetscCall(SNESSetFunction(snes,r,SNESPicardComputeFunction,ctx)); 23089566063dSJacob Faibussowitsch PetscCall(SNESSetJacobian(snes,Amat,Pmat,SNESPicardComputeJacobian,ctx)); 23098b0a5094SBarry Smith PetscFunctionReturn(0); 23108b0a5094SBarry Smith } 23118b0a5094SBarry Smith 23127971a8bfSPeter Brune /*@C 23137971a8bfSPeter Brune SNESGetPicard - Returns the context for the Picard iteration 23147971a8bfSPeter Brune 23157971a8bfSPeter Brune Not Collective, but Vec is parallel if SNES is parallel. Collective if Vec is requested, but has not been created yet. 23167971a8bfSPeter Brune 23177971a8bfSPeter Brune Input Parameter: 23187971a8bfSPeter Brune . snes - the SNES context 23197971a8bfSPeter Brune 2320d8d19677SJose E. Roman Output Parameters: 23210298fd71SBarry Smith + r - the function (or NULL) 2322f8b49ee9SBarry Smith . f - the function (or NULL); see SNESFunction for calling sequence details 2323e4357dc4SBarry Smith . Amat - the matrix used to defined the operation A(x) x - b(x) (or NULL) 2324e4357dc4SBarry Smith . Pmat - the matrix from which the preconditioner will be constructed (or NULL) 2325f8b49ee9SBarry Smith . J - the function for matrix evaluation (or NULL); see SNESJacobianFunction for calling sequence details 23260298fd71SBarry Smith - ctx - the function context (or NULL) 23277971a8bfSPeter Brune 23287971a8bfSPeter Brune Level: advanced 23297971a8bfSPeter Brune 2330db781477SPatrick Sanan .seealso: `SNESSetPicard()`, `SNESGetFunction()`, `SNESGetJacobian()`, `SNESGetDM()`, `SNESFunction`, `SNESJacobianFunction` 23317971a8bfSPeter Brune @*/ 2332d1e9a80fSBarry 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) 23337971a8bfSPeter Brune { 23347971a8bfSPeter Brune DM dm; 23357971a8bfSPeter Brune 23367971a8bfSPeter Brune PetscFunctionBegin; 23377971a8bfSPeter Brune PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 23389566063dSJacob Faibussowitsch PetscCall(SNESGetFunction(snes,r,NULL,NULL)); 23399566063dSJacob Faibussowitsch PetscCall(SNESGetJacobian(snes,Amat,Pmat,NULL,NULL)); 23409566063dSJacob Faibussowitsch PetscCall(SNESGetDM(snes,&dm)); 23419566063dSJacob Faibussowitsch PetscCall(DMSNESGetPicard(dm,f,J,ctx)); 23427971a8bfSPeter Brune PetscFunctionReturn(0); 23437971a8bfSPeter Brune } 23447971a8bfSPeter Brune 2345d25893d9SBarry Smith /*@C 2346d25893d9SBarry Smith SNESSetComputeInitialGuess - Sets a routine used to compute an initial guess for the problem 2347d25893d9SBarry Smith 2348d25893d9SBarry Smith Logically Collective on SNES 2349d25893d9SBarry Smith 2350d25893d9SBarry Smith Input Parameters: 2351d25893d9SBarry Smith + snes - the SNES context 2352d25893d9SBarry Smith . func - function evaluation routine 2353d25893d9SBarry Smith - ctx - [optional] user-defined context for private data for the 23540298fd71SBarry Smith function evaluation routine (may be NULL) 2355d25893d9SBarry Smith 2356d25893d9SBarry Smith Calling sequence of func: 2357d25893d9SBarry Smith $ func (SNES snes,Vec x,void *ctx); 2358d25893d9SBarry Smith 2359d25893d9SBarry Smith . f - function vector 2360d25893d9SBarry Smith - ctx - optional user-defined function context 2361d25893d9SBarry Smith 2362d25893d9SBarry Smith Level: intermediate 2363d25893d9SBarry Smith 2364db781477SPatrick Sanan .seealso: `SNESGetFunction()`, `SNESComputeFunction()`, `SNESSetJacobian()` 2365d25893d9SBarry Smith @*/ 2366d25893d9SBarry Smith PetscErrorCode SNESSetComputeInitialGuess(SNES snes,PetscErrorCode (*func)(SNES,Vec,void*),void *ctx) 2367d25893d9SBarry Smith { 2368d25893d9SBarry Smith PetscFunctionBegin; 2369d25893d9SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 2370d25893d9SBarry Smith if (func) snes->ops->computeinitialguess = func; 2371d25893d9SBarry Smith if (ctx) snes->initialguessP = ctx; 2372d25893d9SBarry Smith PetscFunctionReturn(0); 2373d25893d9SBarry Smith } 2374d25893d9SBarry Smith 23753ab0aad5SBarry Smith /* --------------------------------------------------------------- */ 23761096aae1SMatthew Knepley /*@C 23771096aae1SMatthew Knepley SNESGetRhs - Gets the vector for solving F(x) = rhs. If rhs is not set 23781096aae1SMatthew Knepley it assumes a zero right hand side. 23791096aae1SMatthew Knepley 23803f9fe445SBarry Smith Logically Collective on SNES 23811096aae1SMatthew Knepley 23821096aae1SMatthew Knepley Input Parameter: 23831096aae1SMatthew Knepley . snes - the SNES context 23841096aae1SMatthew Knepley 23851096aae1SMatthew Knepley Output Parameter: 23860298fd71SBarry Smith . rhs - the right hand side vector or NULL if the right hand side vector is null 23871096aae1SMatthew Knepley 23881096aae1SMatthew Knepley Level: intermediate 23891096aae1SMatthew Knepley 2390db781477SPatrick Sanan .seealso: `SNESGetSolution()`, `SNESGetFunction()`, `SNESComputeFunction()`, `SNESSetJacobian()`, `SNESSetFunction()` 23911096aae1SMatthew Knepley @*/ 23927087cfbeSBarry Smith PetscErrorCode SNESGetRhs(SNES snes,Vec *rhs) 23931096aae1SMatthew Knepley { 23941096aae1SMatthew Knepley PetscFunctionBegin; 23950700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 23961096aae1SMatthew Knepley PetscValidPointer(rhs,2); 239785385478SLisandro Dalcin *rhs = snes->vec_rhs; 23981096aae1SMatthew Knepley PetscFunctionReturn(0); 23991096aae1SMatthew Knepley } 24001096aae1SMatthew Knepley 24019b94acceSBarry Smith /*@ 2402bf388a1fSBarry Smith SNESComputeFunction - Calls the function that has been set with SNESSetFunction(). 24039b94acceSBarry Smith 2404c7afd0dbSLois Curfman McInnes Collective on SNES 2405c7afd0dbSLois Curfman McInnes 24069b94acceSBarry Smith Input Parameters: 2407c7afd0dbSLois Curfman McInnes + snes - the SNES context 2408c7afd0dbSLois Curfman McInnes - x - input vector 24099b94acceSBarry Smith 24109b94acceSBarry Smith Output Parameter: 24113638b69dSLois Curfman McInnes . y - function vector, as set by SNESSetFunction() 24129b94acceSBarry Smith 24131bffabb2SLois Curfman McInnes Notes: 241436851e7fSLois Curfman McInnes SNESComputeFunction() is typically used within nonlinear solvers 2415bbc1464cSBarry Smith implementations, so users would not generally call this routine themselves. 241636851e7fSLois Curfman McInnes 241736851e7fSLois Curfman McInnes Level: developer 241836851e7fSLois Curfman McInnes 2419db781477SPatrick Sanan .seealso: `SNESSetFunction()`, `SNESGetFunction()`, `SNESComputeMFFunction()` 24209b94acceSBarry Smith @*/ 24217087cfbeSBarry Smith PetscErrorCode SNESComputeFunction(SNES snes,Vec x,Vec y) 24229b94acceSBarry Smith { 24236cab3a1bSJed Brown DM dm; 2424942e3340SBarry Smith DMSNES sdm; 24259b94acceSBarry Smith 24263a40ed3dSBarry Smith PetscFunctionBegin; 24270700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 24280700a824SBarry Smith PetscValidHeaderSpecific(x,VEC_CLASSID,2); 24290700a824SBarry Smith PetscValidHeaderSpecific(y,VEC_CLASSID,3); 2430c9780b6fSBarry Smith PetscCheckSameComm(snes,1,x,2); 2431c9780b6fSBarry Smith PetscCheckSameComm(snes,1,y,3); 24329566063dSJacob Faibussowitsch PetscCall(VecValidValues(x,2,PETSC_TRUE)); 2433184914b5SBarry Smith 24349566063dSJacob Faibussowitsch PetscCall(SNESGetDM(snes,&dm)); 24359566063dSJacob Faibussowitsch PetscCall(DMGetDMSNES(dm,&sdm)); 243632f3f7c2SPeter Brune if (sdm->ops->computefunction) { 243794db00ebSBarry Smith if (sdm->ops->computefunction != SNESObjectiveComputeFunctionDefaultFD) { 24389566063dSJacob Faibussowitsch PetscCall(PetscLogEventBegin(SNES_FunctionEval,snes,x,y,0)); 243994db00ebSBarry Smith } 24409566063dSJacob Faibussowitsch PetscCall(VecLockReadPush(x)); 24418ddeebeaSSteve Benbow /* ensure domainerror is false prior to computefunction evaluation (may not have been reset) */ 24428ddeebeaSSteve Benbow snes->domainerror = PETSC_FALSE; 2443800f99ffSJeremy L Thompson { 2444800f99ffSJeremy L Thompson void *ctx; 2445800f99ffSJeremy L Thompson PetscErrorCode (*computefunction)(SNES,Vec,Vec,void*); 2446800f99ffSJeremy L Thompson PetscCall(DMSNESGetFunction(dm,&computefunction,&ctx)); 2447800f99ffSJeremy L Thompson PetscCallBack("SNES callback function",(*computefunction)(snes,x,y,ctx)); 2448800f99ffSJeremy L Thompson } 24499566063dSJacob Faibussowitsch PetscCall(VecLockReadPop(x)); 245094db00ebSBarry Smith if (sdm->ops->computefunction != SNESObjectiveComputeFunctionDefaultFD) { 24519566063dSJacob Faibussowitsch PetscCall(PetscLogEventEnd(SNES_FunctionEval,snes,x,y,0)); 245294db00ebSBarry Smith } 2453c90fad12SPeter Brune } else if (snes->vec_rhs) { 24549566063dSJacob Faibussowitsch PetscCall(MatMult(snes->jacobian, x, y)); 2455644e2e5bSBarry Smith } else SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE, "Must call SNESSetFunction() or SNESSetDM() before SNESComputeFunction(), likely called from SNESSolve()."); 24561baa6e33SBarry Smith if (snes->vec_rhs) PetscCall(VecAXPY(y,-1.0,snes->vec_rhs)); 2457ae3c334cSLois Curfman McInnes snes->nfuncs++; 2458422a814eSBarry Smith /* 2459422a814eSBarry Smith domainerror might not be set on all processes; so we tag vector locally with Inf and the next inner product or norm will 2460422a814eSBarry Smith propagate the value to all processes 2461422a814eSBarry Smith */ 24621baa6e33SBarry Smith if (snes->domainerror) PetscCall(VecSetInf(y)); 24633a40ed3dSBarry Smith PetscFunctionReturn(0); 24649b94acceSBarry Smith } 24659b94acceSBarry Smith 2466c79ef259SPeter Brune /*@ 2467bbc1464cSBarry Smith SNESComputeMFFunction - Calls the function that has been set with SNESSetMFFunction(). 2468bbc1464cSBarry Smith 2469bbc1464cSBarry Smith Collective on SNES 2470bbc1464cSBarry Smith 2471bbc1464cSBarry Smith Input Parameters: 2472bbc1464cSBarry Smith + snes - the SNES context 2473bbc1464cSBarry Smith - x - input vector 2474bbc1464cSBarry Smith 2475bbc1464cSBarry Smith Output Parameter: 2476bbc1464cSBarry Smith . y - function vector, as set by SNESSetMFFunction() 2477bbc1464cSBarry Smith 2478bbc1464cSBarry Smith Notes: 2479bbc1464cSBarry Smith SNESComputeMFFunction() is used within the matrix vector products called by the matrix created with MatCreateSNESMF() 2480bbc1464cSBarry Smith so users would not generally call this routine themselves. 2481bbc1464cSBarry Smith 2482bbc1464cSBarry Smith Since this function is intended for use with finite differencing it does not subtract the right hand side vector provided with SNESSolve() 2483bbc1464cSBarry Smith while SNESComputeFunction() does. As such, this routine cannot be used with MatMFFDSetBase() with a provided F function value even if it applies the 2484bbc1464cSBarry Smith same function as SNESComputeFunction() if a SNESSolve() right hand side vector is use because the two functions difference would include this right hand side function. 2485bbc1464cSBarry Smith 2486bbc1464cSBarry Smith Level: developer 2487bbc1464cSBarry Smith 2488db781477SPatrick Sanan .seealso: `SNESSetFunction()`, `SNESGetFunction()`, `SNESComputeFunction()`, `MatCreateSNESMF` 2489bbc1464cSBarry Smith @*/ 2490bbc1464cSBarry Smith PetscErrorCode SNESComputeMFFunction(SNES snes,Vec x,Vec y) 2491bbc1464cSBarry Smith { 2492bbc1464cSBarry Smith DM dm; 2493bbc1464cSBarry Smith DMSNES sdm; 2494bbc1464cSBarry Smith 2495bbc1464cSBarry Smith PetscFunctionBegin; 2496bbc1464cSBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 2497bbc1464cSBarry Smith PetscValidHeaderSpecific(x,VEC_CLASSID,2); 2498bbc1464cSBarry Smith PetscValidHeaderSpecific(y,VEC_CLASSID,3); 2499bbc1464cSBarry Smith PetscCheckSameComm(snes,1,x,2); 2500bbc1464cSBarry Smith PetscCheckSameComm(snes,1,y,3); 25019566063dSJacob Faibussowitsch PetscCall(VecValidValues(x,2,PETSC_TRUE)); 2502bbc1464cSBarry Smith 25039566063dSJacob Faibussowitsch PetscCall(SNESGetDM(snes,&dm)); 25049566063dSJacob Faibussowitsch PetscCall(DMGetDMSNES(dm,&sdm)); 25059566063dSJacob Faibussowitsch PetscCall(PetscLogEventBegin(SNES_FunctionEval,snes,x,y,0)); 25069566063dSJacob Faibussowitsch PetscCall(VecLockReadPush(x)); 2507bbc1464cSBarry Smith /* ensure domainerror is false prior to computefunction evaluation (may not have been reset) */ 2508bbc1464cSBarry Smith snes->domainerror = PETSC_FALSE; 2509792fecdfSBarry Smith PetscCallBack("SNES callback function",(*sdm->ops->computemffunction)(snes,x,y,sdm->mffunctionctx)); 25109566063dSJacob Faibussowitsch PetscCall(VecLockReadPop(x)); 25119566063dSJacob Faibussowitsch PetscCall(PetscLogEventEnd(SNES_FunctionEval,snes,x,y,0)); 2512bbc1464cSBarry Smith snes->nfuncs++; 2513bbc1464cSBarry Smith /* 2514bbc1464cSBarry Smith domainerror might not be set on all processes; so we tag vector locally with Inf and the next inner product or norm will 2515bbc1464cSBarry Smith propagate the value to all processes 2516bbc1464cSBarry Smith */ 25171baa6e33SBarry Smith if (snes->domainerror) PetscCall(VecSetInf(y)); 2518bbc1464cSBarry Smith PetscFunctionReturn(0); 2519bbc1464cSBarry Smith } 2520bbc1464cSBarry Smith 2521bbc1464cSBarry Smith /*@ 2522be95d8f1SBarry Smith SNESComputeNGS - Calls the Gauss-Seidel function that has been set with SNESSetNGS(). 2523c79ef259SPeter Brune 2524c79ef259SPeter Brune Collective on SNES 2525c79ef259SPeter Brune 2526c79ef259SPeter Brune Input Parameters: 2527c79ef259SPeter Brune + snes - the SNES context 2528c79ef259SPeter Brune . x - input vector 2529c79ef259SPeter Brune - b - rhs vector 2530c79ef259SPeter Brune 2531c79ef259SPeter Brune Output Parameter: 2532c79ef259SPeter Brune . x - new solution vector 2533c79ef259SPeter Brune 2534c79ef259SPeter Brune Notes: 2535be95d8f1SBarry Smith SNESComputeNGS() is typically used within composed nonlinear solver 2536c79ef259SPeter Brune implementations, so most users would not generally call this routine 2537c79ef259SPeter Brune themselves. 2538c79ef259SPeter Brune 2539c79ef259SPeter Brune Level: developer 2540c79ef259SPeter Brune 2541db781477SPatrick Sanan .seealso: `SNESSetNGS()`, `SNESComputeFunction()` 2542c79ef259SPeter Brune @*/ 2543be95d8f1SBarry Smith PetscErrorCode SNESComputeNGS(SNES snes,Vec b,Vec x) 2544646217ecSPeter Brune { 25456cab3a1bSJed Brown DM dm; 2546942e3340SBarry Smith DMSNES sdm; 2547646217ecSPeter Brune 2548646217ecSPeter Brune PetscFunctionBegin; 2549646217ecSPeter Brune PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 2550064a246eSJacob Faibussowitsch PetscValidHeaderSpecific(x,VEC_CLASSID,3); 2551064a246eSJacob Faibussowitsch if (b) PetscValidHeaderSpecific(b,VEC_CLASSID,2); 2552064a246eSJacob Faibussowitsch PetscCheckSameComm(snes,1,x,3); 2553064a246eSJacob Faibussowitsch if (b) PetscCheckSameComm(snes,1,b,2); 25549566063dSJacob Faibussowitsch if (b) PetscCall(VecValidValues(b,2,PETSC_TRUE)); 25559566063dSJacob Faibussowitsch PetscCall(PetscLogEventBegin(SNES_NGSEval,snes,x,b,0)); 25569566063dSJacob Faibussowitsch PetscCall(SNESGetDM(snes,&dm)); 25579566063dSJacob Faibussowitsch PetscCall(DMGetDMSNES(dm,&sdm)); 255822c6f798SBarry Smith if (sdm->ops->computegs) { 25599566063dSJacob Faibussowitsch if (b) PetscCall(VecLockReadPush(b)); 2560792fecdfSBarry Smith PetscCallBack("SNES callback NGS",(*sdm->ops->computegs)(snes,x,b,sdm->gsctx)); 25619566063dSJacob Faibussowitsch if (b) PetscCall(VecLockReadPop(b)); 2562be95d8f1SBarry Smith } else SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE, "Must call SNESSetNGS() before SNESComputeNGS(), likely called from SNESSolve()."); 25639566063dSJacob Faibussowitsch PetscCall(PetscLogEventEnd(SNES_NGSEval,snes,x,b,0)); 2564646217ecSPeter Brune PetscFunctionReturn(0); 2565646217ecSPeter Brune } 2566646217ecSPeter Brune 2567e885f1abSBarry Smith PetscErrorCode SNESTestJacobian(SNES snes) 2568e885f1abSBarry Smith { 256912837594SBarry Smith Mat A,B,C,D,jacobian; 2570e885f1abSBarry Smith Vec x = snes->vec_sol,f = snes->vec_func; 2571e885f1abSBarry Smith PetscReal nrm,gnorm; 257281e7118cSBarry Smith PetscReal threshold = 1.e-5; 25730e276705SLisandro Dalcin MatType mattype; 2574e885f1abSBarry Smith PetscInt m,n,M,N; 2575e885f1abSBarry Smith void *functx; 25762cd624f9SStefano Zampini PetscBool complete_print = PETSC_FALSE,threshold_print = PETSC_FALSE,test = PETSC_FALSE,flg,istranspose; 25773325ff46SBarry Smith PetscViewer viewer,mviewer; 2578e885f1abSBarry Smith MPI_Comm comm; 2579e885f1abSBarry Smith PetscInt tabs; 258012837594SBarry Smith static PetscBool directionsprinted = PETSC_FALSE; 25813325ff46SBarry Smith PetscViewerFormat format; 2582e885f1abSBarry Smith 2583e885f1abSBarry Smith PetscFunctionBegin; 2584d0609cedSBarry Smith PetscObjectOptionsBegin((PetscObject)snes); 25859566063dSJacob Faibussowitsch PetscCall(PetscOptionsName("-snes_test_jacobian","Compare hand-coded and finite difference Jacobians","None",&test)); 25869566063dSJacob Faibussowitsch PetscCall(PetscOptionsReal("-snes_test_jacobian", "Threshold for element difference between hand-coded and finite difference being meaningful", "None", threshold, &threshold,NULL)); 25879566063dSJacob Faibussowitsch PetscCall(PetscOptionsViewer("-snes_test_jacobian_view","View difference between hand-coded and finite difference Jacobians element entries","None",&mviewer,&format,&complete_print)); 258818d89885SKarl Rupp if (!complete_print) { 25899566063dSJacob Faibussowitsch PetscCall(PetscOptionsDeprecated("-snes_test_jacobian_display","-snes_test_jacobian_view","3.13",NULL)); 25909566063dSJacob Faibussowitsch PetscCall(PetscOptionsViewer("-snes_test_jacobian_display","Display difference between hand-coded and finite difference Jacobians","None",&mviewer,&format,&complete_print)); 259118d89885SKarl Rupp } 259218d89885SKarl Rupp /* for compatibility with PETSc 3.9 and older. */ 25939566063dSJacob Faibussowitsch PetscCall(PetscOptionsDeprecated("-snes_test_jacobian_display_threshold","-snes_test_jacobian","3.13","-snes_test_jacobian accepts an optional threshold (since v3.10)")); 25949566063dSJacob Faibussowitsch PetscCall(PetscOptionsReal("-snes_test_jacobian_display_threshold", "Display difference between hand-coded and finite difference Jacobians which exceed input threshold", "None", threshold, &threshold, &threshold_print)); 2595d0609cedSBarry Smith PetscOptionsEnd(); 2596e885f1abSBarry Smith if (!test) PetscFunctionReturn(0); 2597e885f1abSBarry Smith 25989566063dSJacob Faibussowitsch PetscCall(PetscObjectGetComm((PetscObject)snes,&comm)); 25999566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIGetStdout(comm,&viewer)); 26009566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIGetTab(viewer, &tabs)); 26019566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIISetTab(viewer, ((PetscObject)snes)->tablevel)); 26029566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer," ---------- Testing Jacobian -------------\n")); 260312837594SBarry Smith if (!complete_print && !directionsprinted) { 26049566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer," Run with -snes_test_jacobian_view and optionally -snes_test_jacobian <threshold> to show difference\n")); 26059566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer," of hand-coded and finite difference Jacobian entries greater than <threshold>.\n")); 260612837594SBarry Smith } 260712837594SBarry Smith if (!directionsprinted) { 26089566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer," Testing hand-coded Jacobian, if (for double precision runs) ||J - Jfd||_F/||J||_F is\n")); 26099566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer," O(1.e-8), the hand-coded Jacobian is probably correct.\n")); 261012837594SBarry Smith directionsprinted = PETSC_TRUE; 2611e885f1abSBarry Smith } 26121baa6e33SBarry Smith if (complete_print) PetscCall(PetscViewerPushFormat(mviewer,format)); 2613e885f1abSBarry Smith 26149566063dSJacob Faibussowitsch PetscCall(PetscObjectTypeCompare((PetscObject)snes->jacobian,MATMFFD,&flg)); 261512837594SBarry Smith if (!flg) jacobian = snes->jacobian; 261612837594SBarry Smith else jacobian = snes->jacobian_pre; 261712837594SBarry Smith 2618a82339d0SMatthew G. Knepley if (!x) { 26199566063dSJacob Faibussowitsch PetscCall(MatCreateVecs(jacobian, &x, NULL)); 2620a82339d0SMatthew G. Knepley } else { 26219566063dSJacob Faibussowitsch PetscCall(PetscObjectReference((PetscObject) x)); 2622a82339d0SMatthew G. Knepley } 2623a82339d0SMatthew G. Knepley if (!f) { 26249566063dSJacob Faibussowitsch PetscCall(VecDuplicate(x, &f)); 2625a82339d0SMatthew G. Knepley } else { 26269566063dSJacob Faibussowitsch PetscCall(PetscObjectReference((PetscObject) f)); 2627a82339d0SMatthew G. Knepley } 2628a82339d0SMatthew G. Knepley /* evaluate the function at this point because SNESComputeJacobianDefault() assumes that the function has been evaluated and put into snes->vec_func */ 26299566063dSJacob Faibussowitsch PetscCall(SNESComputeFunction(snes,x,f)); 26309566063dSJacob Faibussowitsch PetscCall(VecDestroy(&f)); 26319566063dSJacob Faibussowitsch PetscCall(PetscObjectTypeCompare((PetscObject)snes,SNESKSPTRANSPOSEONLY,&istranspose)); 263212837594SBarry Smith while (jacobian) { 26332cd624f9SStefano Zampini Mat JT = NULL, Jsave = NULL; 26342cd624f9SStefano Zampini 26352cd624f9SStefano Zampini if (istranspose) { 26369566063dSJacob Faibussowitsch PetscCall(MatCreateTranspose(jacobian,&JT)); 26372cd624f9SStefano Zampini Jsave = jacobian; 26382cd624f9SStefano Zampini jacobian = JT; 26392cd624f9SStefano Zampini } 26409566063dSJacob Faibussowitsch PetscCall(PetscObjectBaseTypeCompareAny((PetscObject)jacobian,&flg,MATSEQAIJ,MATMPIAIJ,MATSEQDENSE,MATMPIDENSE,MATSEQBAIJ,MATMPIBAIJ,MATSEQSBAIJ,MATMPISBAIJ,"")); 264112837594SBarry Smith if (flg) { 264212837594SBarry Smith A = jacobian; 26439566063dSJacob Faibussowitsch PetscCall(PetscObjectReference((PetscObject)A)); 264412837594SBarry Smith } else { 26459566063dSJacob Faibussowitsch PetscCall(MatComputeOperator(jacobian,MATAIJ,&A)); 264612837594SBarry Smith } 2647e885f1abSBarry Smith 26489566063dSJacob Faibussowitsch PetscCall(MatGetType(A,&mattype)); 26499566063dSJacob Faibussowitsch PetscCall(MatGetSize(A,&M,&N)); 26509566063dSJacob Faibussowitsch PetscCall(MatGetLocalSize(A,&m,&n)); 26519566063dSJacob Faibussowitsch PetscCall(MatCreate(PetscObjectComm((PetscObject)A),&B)); 26529566063dSJacob Faibussowitsch PetscCall(MatSetType(B,mattype)); 26539566063dSJacob Faibussowitsch PetscCall(MatSetSizes(B,m,n,M,N)); 26549566063dSJacob Faibussowitsch PetscCall(MatSetBlockSizesFromMats(B,A,A)); 26559566063dSJacob Faibussowitsch PetscCall(MatSetUp(B)); 26569566063dSJacob Faibussowitsch PetscCall(MatSetOption(B,MAT_NEW_NONZERO_ALLOCATION_ERR,PETSC_FALSE)); 2657e885f1abSBarry Smith 26589566063dSJacob Faibussowitsch PetscCall(SNESGetFunction(snes,NULL,NULL,&functx)); 26599566063dSJacob Faibussowitsch PetscCall(SNESComputeJacobianDefault(snes,x,B,B,functx)); 266012837594SBarry Smith 26619566063dSJacob Faibussowitsch PetscCall(MatDuplicate(B,MAT_COPY_VALUES,&D)); 26629566063dSJacob Faibussowitsch PetscCall(MatAYPX(D,-1.0,A,DIFFERENT_NONZERO_PATTERN)); 26639566063dSJacob Faibussowitsch PetscCall(MatNorm(D,NORM_FROBENIUS,&nrm)); 26649566063dSJacob Faibussowitsch PetscCall(MatNorm(A,NORM_FROBENIUS,&gnorm)); 26659566063dSJacob Faibussowitsch PetscCall(MatDestroy(&D)); 266612837594SBarry Smith if (!gnorm) gnorm = 1; /* just in case */ 26679566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer," ||J - Jfd||_F/||J||_F = %g, ||J - Jfd||_F = %g\n",(double)(nrm/gnorm),(double)nrm)); 266812837594SBarry Smith 2669e885f1abSBarry Smith if (complete_print) { 26709566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer," Hand-coded Jacobian ----------\n")); 26719566063dSJacob Faibussowitsch PetscCall(MatView(A,mviewer)); 26729566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer," Finite difference Jacobian ----------\n")); 26739566063dSJacob Faibussowitsch PetscCall(MatView(B,mviewer)); 2674e885f1abSBarry Smith } 2675e885f1abSBarry Smith 2676df10fb39SFande Kong if (threshold_print || complete_print) { 2677e885f1abSBarry Smith PetscInt Istart, Iend, *ccols, bncols, cncols, j, row; 2678e885f1abSBarry Smith PetscScalar *cvals; 2679e885f1abSBarry Smith const PetscInt *bcols; 2680e885f1abSBarry Smith const PetscScalar *bvals; 2681e885f1abSBarry Smith 26829566063dSJacob Faibussowitsch PetscCall(MatCreate(PetscObjectComm((PetscObject)A),&C)); 26839566063dSJacob Faibussowitsch PetscCall(MatSetType(C,mattype)); 26849566063dSJacob Faibussowitsch PetscCall(MatSetSizes(C,m,n,M,N)); 26859566063dSJacob Faibussowitsch PetscCall(MatSetBlockSizesFromMats(C,A,A)); 26869566063dSJacob Faibussowitsch PetscCall(MatSetUp(C)); 26879566063dSJacob Faibussowitsch PetscCall(MatSetOption(C,MAT_NEW_NONZERO_ALLOCATION_ERR,PETSC_FALSE)); 26880e276705SLisandro Dalcin 26899566063dSJacob Faibussowitsch PetscCall(MatAYPX(B,-1.0,A,DIFFERENT_NONZERO_PATTERN)); 26909566063dSJacob Faibussowitsch PetscCall(MatGetOwnershipRange(B,&Istart,&Iend)); 2691e885f1abSBarry Smith 2692e885f1abSBarry Smith for (row = Istart; row < Iend; row++) { 26939566063dSJacob Faibussowitsch PetscCall(MatGetRow(B,row,&bncols,&bcols,&bvals)); 26949566063dSJacob Faibussowitsch PetscCall(PetscMalloc2(bncols,&ccols,bncols,&cvals)); 2695e885f1abSBarry Smith for (j = 0, cncols = 0; j < bncols; j++) { 269623a52b1dSBarry Smith if (PetscAbsScalar(bvals[j]) > threshold) { 2697e885f1abSBarry Smith ccols[cncols] = bcols[j]; 2698e885f1abSBarry Smith cvals[cncols] = bvals[j]; 2699e885f1abSBarry Smith cncols += 1; 2700e885f1abSBarry Smith } 2701e885f1abSBarry Smith } 2702e885f1abSBarry Smith if (cncols) { 27039566063dSJacob Faibussowitsch PetscCall(MatSetValues(C,1,&row,cncols,ccols,cvals,INSERT_VALUES)); 2704e885f1abSBarry Smith } 27059566063dSJacob Faibussowitsch PetscCall(MatRestoreRow(B,row,&bncols,&bcols,&bvals)); 27069566063dSJacob Faibussowitsch PetscCall(PetscFree2(ccols,cvals)); 2707e885f1abSBarry Smith } 27089566063dSJacob Faibussowitsch PetscCall(MatAssemblyBegin(C,MAT_FINAL_ASSEMBLY)); 27099566063dSJacob Faibussowitsch PetscCall(MatAssemblyEnd(C,MAT_FINAL_ASSEMBLY)); 27109566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer," Hand-coded minus finite-difference Jacobian with tolerance %g ----------\n",(double)threshold)); 27119566063dSJacob Faibussowitsch PetscCall(MatView(C,complete_print ? mviewer : viewer)); 27129566063dSJacob Faibussowitsch PetscCall(MatDestroy(&C)); 2713e885f1abSBarry Smith } 27149566063dSJacob Faibussowitsch PetscCall(MatDestroy(&A)); 27159566063dSJacob Faibussowitsch PetscCall(MatDestroy(&B)); 27169566063dSJacob Faibussowitsch PetscCall(MatDestroy(&JT)); 27172cd624f9SStefano Zampini if (Jsave) jacobian = Jsave; 271812837594SBarry Smith if (jacobian != snes->jacobian_pre) { 271912837594SBarry Smith jacobian = snes->jacobian_pre; 27209566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer," ---------- Testing Jacobian for preconditioner -------------\n")); 272112837594SBarry Smith } 272212837594SBarry Smith else jacobian = NULL; 272312837594SBarry Smith } 27249566063dSJacob Faibussowitsch PetscCall(VecDestroy(&x)); 27251baa6e33SBarry Smith if (complete_print) PetscCall(PetscViewerPopFormat(mviewer)); 27269566063dSJacob Faibussowitsch if (mviewer) PetscCall(PetscViewerDestroy(&mviewer)); 27279566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIISetTab(viewer,tabs)); 2728e885f1abSBarry Smith PetscFunctionReturn(0); 2729e885f1abSBarry Smith } 2730e885f1abSBarry Smith 273162fef451SLois Curfman McInnes /*@ 2732bf388a1fSBarry Smith SNESComputeJacobian - Computes the Jacobian matrix that has been set with SNESSetJacobian(). 273362fef451SLois Curfman McInnes 2734d083f849SBarry Smith Collective on SNES 2735c7afd0dbSLois Curfman McInnes 273662fef451SLois Curfman McInnes Input Parameters: 2737c7afd0dbSLois Curfman McInnes + snes - the SNES context 2738c7afd0dbSLois Curfman McInnes - x - input vector 273962fef451SLois Curfman McInnes 274062fef451SLois Curfman McInnes Output Parameters: 2741c7afd0dbSLois Curfman McInnes + A - Jacobian matrix 2742d1e9a80fSBarry Smith - B - optional preconditioning matrix 2743fee21e36SBarry Smith 2744e35cf81dSBarry Smith Options Database Keys: 274567b8a455SSatish Balay + -snes_lag_preconditioner <lag> - how often to rebuild preconditioner 274667b8a455SSatish Balay . -snes_lag_jacobian <lag> - how often to rebuild Jacobian 2747455a5933SJed 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. 2748455a5933SJed 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 2749693365a8SJed Brown . -snes_compare_explicit - Compare the computed Jacobian to the finite difference Jacobian and output the differences 2750693365a8SJed Brown . -snes_compare_explicit_draw - Compare the computed Jacobian to the finite difference Jacobian and draw the result 2751693365a8SJed Brown . -snes_compare_explicit_contour - Compare the computed Jacobian to the finite difference Jacobian and draw a contour plot with the result 27524c30e9fbSJed Brown . -snes_compare_operator - Make the comparison options above use the operator instead of the preconditioning matrix 275394d6a431SBarry Smith . -snes_compare_coloring - Compute the finite difference Jacobian using coloring and display norms of difference 2754a5b23f4aSJose E. Roman . -snes_compare_coloring_display - Compute the finite difference Jacobian using coloring and display verbose differences 2755c01495d3SJed Brown . -snes_compare_coloring_threshold - Display only those matrix entries that differ by more than a given threshold 2756c01495d3SJed Brown . -snes_compare_coloring_threshold_atol - Absolute tolerance for difference in matrix entries to be displayed by -snes_compare_coloring_threshold 2757c01495d3SJed Brown . -snes_compare_coloring_threshold_rtol - Relative tolerance for difference in matrix entries to be displayed by -snes_compare_coloring_threshold 2758a5b23f4aSJose E. Roman . -snes_compare_coloring_draw - Compute the finite difference Jacobian using coloring and draw differences 2759a5b23f4aSJose E. Roman - -snes_compare_coloring_draw_contour - Compute the finite difference Jacobian using coloring and show contours of matrices and differences 2760c01495d3SJed Brown 276162fef451SLois Curfman McInnes Notes: 276262fef451SLois Curfman McInnes Most users should not need to explicitly call this routine, as it 276362fef451SLois Curfman McInnes is used internally within the nonlinear solvers. 276462fef451SLois Curfman McInnes 276595452b02SPatrick Sanan Developer Notes: 276695452b02SPatrick 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 2767e885f1abSBarry Smith for with the SNESType of test that has been removed. 2768e885f1abSBarry Smith 276936851e7fSLois Curfman McInnes Level: developer 277036851e7fSLois Curfman McInnes 2771db781477SPatrick Sanan .seealso: `SNESSetJacobian()`, `KSPSetOperators()`, `MatStructure`, `SNESSetLagPreconditioner()`, `SNESSetLagJacobian()` 277262fef451SLois Curfman McInnes @*/ 2773d1e9a80fSBarry Smith PetscErrorCode SNESComputeJacobian(SNES snes,Vec X,Mat A,Mat B) 27749b94acceSBarry Smith { 2775ace3abfcSBarry Smith PetscBool flag; 27766cab3a1bSJed Brown DM dm; 2777942e3340SBarry Smith DMSNES sdm; 2778e0e3a89bSBarry Smith KSP ksp; 27793a40ed3dSBarry Smith 27803a40ed3dSBarry Smith PetscFunctionBegin; 27810700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 27820700a824SBarry Smith PetscValidHeaderSpecific(X,VEC_CLASSID,2); 2783c9780b6fSBarry Smith PetscCheckSameComm(snes,1,X,2); 27849566063dSJacob Faibussowitsch PetscCall(VecValidValues(X,2,PETSC_TRUE)); 27859566063dSJacob Faibussowitsch PetscCall(SNESGetDM(snes,&dm)); 27869566063dSJacob Faibussowitsch PetscCall(DMGetDMSNES(dm,&sdm)); 27873232da50SPeter Brune 2788ebd3b9afSBarry Smith /* make sure that MatAssemblyBegin/End() is called on A matrix if it is matrix free */ 2789fe3ffe1eSBarry Smith if (snes->lagjacobian == -2) { 2790fe3ffe1eSBarry Smith snes->lagjacobian = -1; 2791f5af7f23SKarl Rupp 27929566063dSJacob Faibussowitsch PetscCall(PetscInfo(snes,"Recomputing Jacobian/preconditioner because lag is -2 (means compute Jacobian, but then never again) \n")); 2793fe3ffe1eSBarry Smith } else if (snes->lagjacobian == -1) { 27949566063dSJacob Faibussowitsch PetscCall(PetscInfo(snes,"Reusing Jacobian/preconditioner because lag is -1\n")); 27959566063dSJacob Faibussowitsch PetscCall(PetscObjectTypeCompare((PetscObject)A,MATMFFD,&flag)); 2796ebd3b9afSBarry Smith if (flag) { 27979566063dSJacob Faibussowitsch PetscCall(MatAssemblyBegin(A,MAT_FINAL_ASSEMBLY)); 27989566063dSJacob Faibussowitsch PetscCall(MatAssemblyEnd(A,MAT_FINAL_ASSEMBLY)); 2799ebd3b9afSBarry Smith } 2800e35cf81dSBarry Smith PetscFunctionReturn(0); 280137ec4e1aSPeter Brune } else if (snes->lagjacobian > 1 && (snes->iter + snes->jac_iter) % snes->lagjacobian) { 280263a3b9bcSJacob Faibussowitsch PetscCall(PetscInfo(snes,"Reusing Jacobian/preconditioner because lag is %" PetscInt_FMT " and SNES iteration is %" PetscInt_FMT "\n",snes->lagjacobian,snes->iter)); 28039566063dSJacob Faibussowitsch PetscCall(PetscObjectTypeCompare((PetscObject)A,MATMFFD,&flag)); 2804ebd3b9afSBarry Smith if (flag) { 28059566063dSJacob Faibussowitsch PetscCall(MatAssemblyBegin(A,MAT_FINAL_ASSEMBLY)); 28069566063dSJacob Faibussowitsch PetscCall(MatAssemblyEnd(A,MAT_FINAL_ASSEMBLY)); 2807ebd3b9afSBarry Smith } 2808e35cf81dSBarry Smith PetscFunctionReturn(0); 2809e35cf81dSBarry Smith } 2810efd4aadfSBarry Smith if (snes->npc && snes->npcside == PC_LEFT) { 28119566063dSJacob Faibussowitsch PetscCall(MatAssemblyBegin(A,MAT_FINAL_ASSEMBLY)); 28129566063dSJacob Faibussowitsch PetscCall(MatAssemblyEnd(A,MAT_FINAL_ASSEMBLY)); 2813d728fb7dSPeter Brune PetscFunctionReturn(0); 2814d728fb7dSPeter Brune } 2815e35cf81dSBarry Smith 28169566063dSJacob Faibussowitsch PetscCall(PetscLogEventBegin(SNES_JacobianEval,snes,X,A,B)); 28179566063dSJacob Faibussowitsch PetscCall(VecLockReadPush(X)); 2818800f99ffSJeremy L Thompson { 2819800f99ffSJeremy L Thompson void *ctx; 2820800f99ffSJeremy L Thompson PetscErrorCode (*J)(SNES,Vec,Mat,Mat,void*); 2821800f99ffSJeremy L Thompson PetscCall(DMSNESGetJacobian(dm,&J,&ctx)); 2822800f99ffSJeremy L Thompson PetscCallBack("SNES callback Jacobian",(*J)(snes,X,A,B,ctx)); 2823800f99ffSJeremy L Thompson } 28249566063dSJacob Faibussowitsch PetscCall(VecLockReadPop(X)); 28259566063dSJacob Faibussowitsch PetscCall(PetscLogEventEnd(SNES_JacobianEval,snes,X,A,B)); 282628d58a37SPierre Jolivet 282728d58a37SPierre Jolivet /* attach latest linearization point to the preconditioning matrix */ 28289566063dSJacob Faibussowitsch PetscCall(PetscObjectCompose((PetscObject)B,"__SNES_latest_X",(PetscObject)X)); 2829a8054027SBarry Smith 2830e0e3a89bSBarry Smith /* the next line ensures that snes->ksp exists */ 28319566063dSJacob Faibussowitsch PetscCall(SNESGetKSP(snes,&ksp)); 28323b4f5425SBarry Smith if (snes->lagpreconditioner == -2) { 28339566063dSJacob Faibussowitsch PetscCall(PetscInfo(snes,"Rebuilding preconditioner exactly once since lag is -2\n")); 28349566063dSJacob Faibussowitsch PetscCall(KSPSetReusePreconditioner(snes->ksp,PETSC_FALSE)); 28353b4f5425SBarry Smith snes->lagpreconditioner = -1; 28363b4f5425SBarry Smith } else if (snes->lagpreconditioner == -1) { 28379566063dSJacob Faibussowitsch PetscCall(PetscInfo(snes,"Reusing preconditioner because lag is -1\n")); 28389566063dSJacob Faibussowitsch PetscCall(KSPSetReusePreconditioner(snes->ksp,PETSC_TRUE)); 283937ec4e1aSPeter Brune } else if (snes->lagpreconditioner > 1 && (snes->iter + snes->pre_iter) % snes->lagpreconditioner) { 284063a3b9bcSJacob Faibussowitsch PetscCall(PetscInfo(snes,"Reusing preconditioner because lag is %" PetscInt_FMT " and SNES iteration is %" PetscInt_FMT "\n",snes->lagpreconditioner,snes->iter)); 28419566063dSJacob Faibussowitsch PetscCall(KSPSetReusePreconditioner(snes->ksp,PETSC_TRUE)); 2842d1e9a80fSBarry Smith } else { 28439566063dSJacob Faibussowitsch PetscCall(PetscInfo(snes,"Rebuilding preconditioner\n")); 28449566063dSJacob Faibussowitsch PetscCall(KSPSetReusePreconditioner(snes->ksp,PETSC_FALSE)); 2845a8054027SBarry Smith } 2846a8054027SBarry Smith 28479566063dSJacob Faibussowitsch PetscCall(SNESTestJacobian(snes)); 28486d84be18SBarry Smith /* make sure user returned a correct Jacobian and preconditioner */ 284994ab13aaSBarry Smith /* PetscValidHeaderSpecific(A,MAT_CLASSID,3); 285094ab13aaSBarry Smith PetscValidHeaderSpecific(B,MAT_CLASSID,4); */ 2851693365a8SJed Brown { 2852693365a8SJed Brown PetscBool flag = PETSC_FALSE,flag_draw = PETSC_FALSE,flag_contour = PETSC_FALSE,flag_operator = PETSC_FALSE; 28539566063dSJacob Faibussowitsch PetscCall(PetscOptionsGetViewer(PetscObjectComm((PetscObject)snes),((PetscObject) snes)->options,((PetscObject)snes)->prefix,"-snes_compare_explicit",NULL,NULL,&flag)); 28549566063dSJacob Faibussowitsch PetscCall(PetscOptionsGetViewer(PetscObjectComm((PetscObject)snes),((PetscObject) snes)->options,((PetscObject)snes)->prefix,"-snes_compare_explicit_draw",NULL,NULL,&flag_draw)); 28559566063dSJacob Faibussowitsch PetscCall(PetscOptionsGetViewer(PetscObjectComm((PetscObject)snes),((PetscObject) snes)->options,((PetscObject)snes)->prefix,"-snes_compare_explicit_draw_contour",NULL,NULL,&flag_contour)); 28569566063dSJacob Faibussowitsch PetscCall(PetscOptionsGetViewer(PetscObjectComm((PetscObject)snes),((PetscObject) snes)->options,((PetscObject)snes)->prefix,"-snes_compare_operator",NULL,NULL,&flag_operator)); 2857693365a8SJed Brown if (flag || flag_draw || flag_contour) { 28580298fd71SBarry Smith Mat Bexp_mine = NULL,Bexp,FDexp; 2859693365a8SJed Brown PetscViewer vdraw,vstdout; 28606b3a5b13SJed Brown PetscBool flg; 2861693365a8SJed Brown if (flag_operator) { 28629566063dSJacob Faibussowitsch PetscCall(MatComputeOperator(A,MATAIJ,&Bexp_mine)); 2863693365a8SJed Brown Bexp = Bexp_mine; 2864693365a8SJed Brown } else { 2865693365a8SJed Brown /* See if the preconditioning matrix can be viewed and added directly */ 28669566063dSJacob Faibussowitsch PetscCall(PetscObjectBaseTypeCompareAny((PetscObject)B,&flg,MATSEQAIJ,MATMPIAIJ,MATSEQDENSE,MATMPIDENSE,MATSEQBAIJ,MATMPIBAIJ,MATSEQSBAIJ,MATMPIBAIJ,"")); 286794ab13aaSBarry Smith if (flg) Bexp = B; 2868693365a8SJed Brown else { 2869693365a8SJed Brown /* If the "preconditioning" matrix is itself MATSHELL or some other type without direct support */ 28709566063dSJacob Faibussowitsch PetscCall(MatComputeOperator(B,MATAIJ,&Bexp_mine)); 2871693365a8SJed Brown Bexp = Bexp_mine; 2872693365a8SJed Brown } 2873693365a8SJed Brown } 28749566063dSJacob Faibussowitsch PetscCall(MatConvert(Bexp,MATSAME,MAT_INITIAL_MATRIX,&FDexp)); 28759566063dSJacob Faibussowitsch PetscCall(SNESComputeJacobianDefault(snes,X,FDexp,FDexp,NULL)); 28769566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIGetStdout(PetscObjectComm((PetscObject)snes),&vstdout)); 2877693365a8SJed Brown if (flag_draw || flag_contour) { 28789566063dSJacob Faibussowitsch PetscCall(PetscViewerDrawOpen(PetscObjectComm((PetscObject)snes),NULL,"Explicit Jacobians",PETSC_DECIDE,PETSC_DECIDE,300,300,&vdraw)); 28799566063dSJacob Faibussowitsch if (flag_contour) PetscCall(PetscViewerPushFormat(vdraw,PETSC_VIEWER_DRAW_CONTOUR)); 28800298fd71SBarry Smith } else vdraw = NULL; 28819566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(vstdout,"Explicit %s\n",flag_operator ? "Jacobian" : "preconditioning Jacobian")); 28829566063dSJacob Faibussowitsch if (flag) PetscCall(MatView(Bexp,vstdout)); 28839566063dSJacob Faibussowitsch if (vdraw) PetscCall(MatView(Bexp,vdraw)); 28849566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(vstdout,"Finite difference Jacobian\n")); 28859566063dSJacob Faibussowitsch if (flag) PetscCall(MatView(FDexp,vstdout)); 28869566063dSJacob Faibussowitsch if (vdraw) PetscCall(MatView(FDexp,vdraw)); 28879566063dSJacob Faibussowitsch PetscCall(MatAYPX(FDexp,-1.0,Bexp,SAME_NONZERO_PATTERN)); 28889566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(vstdout,"User-provided matrix minus finite difference Jacobian\n")); 28899566063dSJacob Faibussowitsch if (flag) PetscCall(MatView(FDexp,vstdout)); 2890693365a8SJed Brown if (vdraw) { /* Always use contour for the difference */ 28919566063dSJacob Faibussowitsch PetscCall(PetscViewerPushFormat(vdraw,PETSC_VIEWER_DRAW_CONTOUR)); 28929566063dSJacob Faibussowitsch PetscCall(MatView(FDexp,vdraw)); 28939566063dSJacob Faibussowitsch PetscCall(PetscViewerPopFormat(vdraw)); 2894693365a8SJed Brown } 28959566063dSJacob Faibussowitsch if (flag_contour) PetscCall(PetscViewerPopFormat(vdraw)); 28969566063dSJacob Faibussowitsch PetscCall(PetscViewerDestroy(&vdraw)); 28979566063dSJacob Faibussowitsch PetscCall(MatDestroy(&Bexp_mine)); 28989566063dSJacob Faibussowitsch PetscCall(MatDestroy(&FDexp)); 2899693365a8SJed Brown } 2900693365a8SJed Brown } 29014c30e9fbSJed Brown { 29026719d8e4SJed Brown PetscBool flag = PETSC_FALSE,flag_display = PETSC_FALSE,flag_draw = PETSC_FALSE,flag_contour = PETSC_FALSE,flag_threshold = PETSC_FALSE; 29036719d8e4SJed Brown PetscReal threshold_atol = PETSC_SQRT_MACHINE_EPSILON,threshold_rtol = 10*PETSC_SQRT_MACHINE_EPSILON; 29049566063dSJacob Faibussowitsch PetscCall(PetscOptionsGetViewer(PetscObjectComm((PetscObject)snes),((PetscObject)snes)->options,((PetscObject)snes)->prefix,"-snes_compare_coloring",NULL,NULL,&flag)); 29059566063dSJacob Faibussowitsch PetscCall(PetscOptionsGetViewer(PetscObjectComm((PetscObject)snes),((PetscObject)snes)->options,((PetscObject)snes)->prefix,"-snes_compare_coloring_display",NULL,NULL,&flag_display)); 29069566063dSJacob Faibussowitsch PetscCall(PetscOptionsGetViewer(PetscObjectComm((PetscObject)snes),((PetscObject)snes)->options,((PetscObject)snes)->prefix,"-snes_compare_coloring_draw",NULL,NULL,&flag_draw)); 29079566063dSJacob Faibussowitsch PetscCall(PetscOptionsGetViewer(PetscObjectComm((PetscObject)snes),((PetscObject)snes)->options,((PetscObject)snes)->prefix,"-snes_compare_coloring_draw_contour",NULL,NULL,&flag_contour)); 29089566063dSJacob Faibussowitsch PetscCall(PetscOptionsGetViewer(PetscObjectComm((PetscObject)snes),((PetscObject)snes)->options,((PetscObject)snes)->prefix,"-snes_compare_coloring_threshold",NULL,NULL,&flag_threshold)); 290927b0f280SBarry Smith if (flag_threshold) { 29109566063dSJacob Faibussowitsch PetscCall(PetscOptionsGetReal(((PetscObject)snes)->options,((PetscObject)snes)->prefix,"-snes_compare_coloring_threshold_rtol",&threshold_rtol,NULL)); 29119566063dSJacob Faibussowitsch PetscCall(PetscOptionsGetReal(((PetscObject)snes)->options,((PetscObject)snes)->prefix,"-snes_compare_coloring_threshold_atol",&threshold_atol,NULL)); 291227b0f280SBarry Smith } 29136719d8e4SJed Brown if (flag || flag_display || flag_draw || flag_contour || flag_threshold) { 29144c30e9fbSJed Brown Mat Bfd; 29154c30e9fbSJed Brown PetscViewer vdraw,vstdout; 2916335efc43SPeter Brune MatColoring coloring; 29174c30e9fbSJed Brown ISColoring iscoloring; 29184c30e9fbSJed Brown MatFDColoring matfdcoloring; 29194c30e9fbSJed Brown PetscErrorCode (*func)(SNES,Vec,Vec,void*); 29204c30e9fbSJed Brown void *funcctx; 29216719d8e4SJed Brown PetscReal norm1,norm2,normmax; 29224c30e9fbSJed Brown 29239566063dSJacob Faibussowitsch PetscCall(MatDuplicate(B,MAT_DO_NOT_COPY_VALUES,&Bfd)); 29249566063dSJacob Faibussowitsch PetscCall(MatColoringCreate(Bfd,&coloring)); 29259566063dSJacob Faibussowitsch PetscCall(MatColoringSetType(coloring,MATCOLORINGSL)); 29269566063dSJacob Faibussowitsch PetscCall(MatColoringSetFromOptions(coloring)); 29279566063dSJacob Faibussowitsch PetscCall(MatColoringApply(coloring,&iscoloring)); 29289566063dSJacob Faibussowitsch PetscCall(MatColoringDestroy(&coloring)); 29299566063dSJacob Faibussowitsch PetscCall(MatFDColoringCreate(Bfd,iscoloring,&matfdcoloring)); 29309566063dSJacob Faibussowitsch PetscCall(MatFDColoringSetFromOptions(matfdcoloring)); 29319566063dSJacob Faibussowitsch PetscCall(MatFDColoringSetUp(Bfd,iscoloring,matfdcoloring)); 29329566063dSJacob Faibussowitsch PetscCall(ISColoringDestroy(&iscoloring)); 29334c30e9fbSJed Brown 29344c30e9fbSJed Brown /* This method of getting the function is currently unreliable since it doesn't work for DM local functions. */ 29359566063dSJacob Faibussowitsch PetscCall(SNESGetFunction(snes,NULL,&func,&funcctx)); 29369566063dSJacob Faibussowitsch PetscCall(MatFDColoringSetFunction(matfdcoloring,(PetscErrorCode (*)(void))func,funcctx)); 29379566063dSJacob Faibussowitsch PetscCall(PetscObjectSetOptionsPrefix((PetscObject)matfdcoloring,((PetscObject)snes)->prefix)); 29389566063dSJacob Faibussowitsch PetscCall(PetscObjectAppendOptionsPrefix((PetscObject)matfdcoloring,"coloring_")); 29399566063dSJacob Faibussowitsch PetscCall(MatFDColoringSetFromOptions(matfdcoloring)); 29409566063dSJacob Faibussowitsch PetscCall(MatFDColoringApply(Bfd,matfdcoloring,X,snes)); 29419566063dSJacob Faibussowitsch PetscCall(MatFDColoringDestroy(&matfdcoloring)); 29424c30e9fbSJed Brown 29439566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIGetStdout(PetscObjectComm((PetscObject)snes),&vstdout)); 29444c30e9fbSJed Brown if (flag_draw || flag_contour) { 29459566063dSJacob Faibussowitsch PetscCall(PetscViewerDrawOpen(PetscObjectComm((PetscObject)snes),NULL,"Colored Jacobians",PETSC_DECIDE,PETSC_DECIDE,300,300,&vdraw)); 29469566063dSJacob Faibussowitsch if (flag_contour) PetscCall(PetscViewerPushFormat(vdraw,PETSC_VIEWER_DRAW_CONTOUR)); 29470298fd71SBarry Smith } else vdraw = NULL; 29489566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(vstdout,"Explicit preconditioning Jacobian\n")); 29499566063dSJacob Faibussowitsch if (flag_display) PetscCall(MatView(B,vstdout)); 29509566063dSJacob Faibussowitsch if (vdraw) PetscCall(MatView(B,vdraw)); 29519566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(vstdout,"Colored Finite difference Jacobian\n")); 29529566063dSJacob Faibussowitsch if (flag_display) PetscCall(MatView(Bfd,vstdout)); 29539566063dSJacob Faibussowitsch if (vdraw) PetscCall(MatView(Bfd,vdraw)); 29549566063dSJacob Faibussowitsch PetscCall(MatAYPX(Bfd,-1.0,B,SAME_NONZERO_PATTERN)); 29559566063dSJacob Faibussowitsch PetscCall(MatNorm(Bfd,NORM_1,&norm1)); 29569566063dSJacob Faibussowitsch PetscCall(MatNorm(Bfd,NORM_FROBENIUS,&norm2)); 29579566063dSJacob Faibussowitsch PetscCall(MatNorm(Bfd,NORM_MAX,&normmax)); 29589566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(vstdout,"User-provided matrix minus finite difference Jacobian, norm1=%g normFrob=%g normmax=%g\n",(double)norm1,(double)norm2,(double)normmax)); 29599566063dSJacob Faibussowitsch if (flag_display) PetscCall(MatView(Bfd,vstdout)); 29604c30e9fbSJed Brown if (vdraw) { /* Always use contour for the difference */ 29619566063dSJacob Faibussowitsch PetscCall(PetscViewerPushFormat(vdraw,PETSC_VIEWER_DRAW_CONTOUR)); 29629566063dSJacob Faibussowitsch PetscCall(MatView(Bfd,vdraw)); 29639566063dSJacob Faibussowitsch PetscCall(PetscViewerPopFormat(vdraw)); 29644c30e9fbSJed Brown } 29659566063dSJacob Faibussowitsch if (flag_contour) PetscCall(PetscViewerPopFormat(vdraw)); 29666719d8e4SJed Brown 29676719d8e4SJed Brown if (flag_threshold) { 29686719d8e4SJed Brown PetscInt bs,rstart,rend,i; 29699566063dSJacob Faibussowitsch PetscCall(MatGetBlockSize(B,&bs)); 29709566063dSJacob Faibussowitsch PetscCall(MatGetOwnershipRange(B,&rstart,&rend)); 29716719d8e4SJed Brown for (i=rstart; i<rend; i++) { 29726719d8e4SJed Brown const PetscScalar *ba,*ca; 29736719d8e4SJed Brown const PetscInt *bj,*cj; 29746719d8e4SJed Brown PetscInt bn,cn,j,maxentrycol = -1,maxdiffcol = -1,maxrdiffcol = -1; 29756719d8e4SJed Brown PetscReal maxentry = 0,maxdiff = 0,maxrdiff = 0; 29769566063dSJacob Faibussowitsch PetscCall(MatGetRow(B,i,&bn,&bj,&ba)); 29779566063dSJacob Faibussowitsch PetscCall(MatGetRow(Bfd,i,&cn,&cj,&ca)); 29785f80ce2aSJacob Faibussowitsch PetscCheck(bn == cn,((PetscObject)A)->comm,PETSC_ERR_PLIB,"Unexpected different nonzero pattern in -snes_compare_coloring_threshold"); 29796719d8e4SJed Brown for (j=0; j<bn; j++) { 29806719d8e4SJed Brown PetscReal rdiff = PetscAbsScalar(ca[j]) / (threshold_atol + threshold_rtol*PetscAbsScalar(ba[j])); 29816719d8e4SJed Brown if (PetscAbsScalar(ba[j]) > PetscAbs(maxentry)) { 29826719d8e4SJed Brown maxentrycol = bj[j]; 29836719d8e4SJed Brown maxentry = PetscRealPart(ba[j]); 29846719d8e4SJed Brown } 29856719d8e4SJed Brown if (PetscAbsScalar(ca[j]) > PetscAbs(maxdiff)) { 29866719d8e4SJed Brown maxdiffcol = bj[j]; 29876719d8e4SJed Brown maxdiff = PetscRealPart(ca[j]); 29886719d8e4SJed Brown } 29896719d8e4SJed Brown if (rdiff > maxrdiff) { 29906719d8e4SJed Brown maxrdiffcol = bj[j]; 29916719d8e4SJed Brown maxrdiff = rdiff; 29926719d8e4SJed Brown } 29936719d8e4SJed Brown } 29946719d8e4SJed Brown if (maxrdiff > 1) { 299563a3b9bcSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(vstdout,"row %" PetscInt_FMT " (maxentry=%g at %" PetscInt_FMT ", maxdiff=%g at %" PetscInt_FMT ", maxrdiff=%g at %" PetscInt_FMT "):",i,(double)maxentry,maxentrycol,(double)maxdiff,maxdiffcol,(double)maxrdiff,maxrdiffcol)); 29966719d8e4SJed Brown for (j=0; j<bn; j++) { 29976719d8e4SJed Brown PetscReal rdiff; 29986719d8e4SJed Brown rdiff = PetscAbsScalar(ca[j]) / (threshold_atol + threshold_rtol*PetscAbsScalar(ba[j])); 29996719d8e4SJed Brown if (rdiff > 1) { 300063a3b9bcSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(vstdout," (%" PetscInt_FMT ",%g:%g)",bj[j],(double)PetscRealPart(ba[j]),(double)PetscRealPart(ca[j]))); 30016719d8e4SJed Brown } 30026719d8e4SJed Brown } 300363a3b9bcSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(vstdout,"\n")); 30046719d8e4SJed Brown } 30059566063dSJacob Faibussowitsch PetscCall(MatRestoreRow(B,i,&bn,&bj,&ba)); 30069566063dSJacob Faibussowitsch PetscCall(MatRestoreRow(Bfd,i,&cn,&cj,&ca)); 30076719d8e4SJed Brown } 30086719d8e4SJed Brown } 30099566063dSJacob Faibussowitsch PetscCall(PetscViewerDestroy(&vdraw)); 30109566063dSJacob Faibussowitsch PetscCall(MatDestroy(&Bfd)); 30114c30e9fbSJed Brown } 30124c30e9fbSJed Brown } 30133a40ed3dSBarry Smith PetscFunctionReturn(0); 30149b94acceSBarry Smith } 30159b94acceSBarry Smith 3016bf388a1fSBarry Smith /*MC 3017411c0326SBarry Smith SNESJacobianFunction - Function used to convey the nonlinear Jacobian of the function to be solved by SNES 3018bf388a1fSBarry Smith 3019bf388a1fSBarry Smith Synopsis: 3020411c0326SBarry Smith #include "petscsnes.h" 3021411c0326SBarry Smith PetscErrorCode SNESJacobianFunction(SNES snes,Vec x,Mat Amat,Mat Pmat,void *ctx); 3022bf388a1fSBarry Smith 30231843f636SBarry Smith Collective on snes 30241843f636SBarry Smith 30251843f636SBarry Smith Input Parameters: 30261843f636SBarry Smith + x - input vector, the Jacobian is to be computed at this value 3027bf388a1fSBarry Smith - ctx - [optional] user-defined Jacobian context 3028bf388a1fSBarry Smith 30291843f636SBarry Smith Output Parameters: 30301843f636SBarry Smith + Amat - the matrix that defines the (approximate) Jacobian 30311843f636SBarry Smith - Pmat - the matrix to be used in constructing the preconditioner, usually the same as Amat. 30321843f636SBarry Smith 3033878cb397SSatish Balay Level: intermediate 3034878cb397SSatish Balay 3035db781477SPatrick Sanan .seealso: `SNESSetFunction()`, `SNESGetFunction()`, `SNESSetJacobian()`, `SNESGetJacobian()` 3036bf388a1fSBarry Smith M*/ 3037bf388a1fSBarry Smith 30389b94acceSBarry Smith /*@C 30399b94acceSBarry Smith SNESSetJacobian - Sets the function to compute Jacobian as well as the 3040044dda88SLois Curfman McInnes location to store the matrix. 30419b94acceSBarry Smith 3042d083f849SBarry Smith Logically Collective on SNES 3043c7afd0dbSLois Curfman McInnes 30449b94acceSBarry Smith Input Parameters: 3045c7afd0dbSLois Curfman McInnes + snes - the SNES context 3046e5d3d808SBarry Smith . Amat - the matrix that defines the (approximate) Jacobian 3047e5d3d808SBarry Smith . Pmat - the matrix to be used in constructing the preconditioner, usually the same as Amat. 3048411c0326SBarry Smith . J - Jacobian evaluation routine (if NULL then SNES retains any previously set value), see SNESJacobianFunction for details 3049c7afd0dbSLois Curfman McInnes - ctx - [optional] user-defined context for private data for the 30500298fd71SBarry Smith Jacobian evaluation routine (may be NULL) (if NULL then SNES retains any previously set value) 30519b94acceSBarry Smith 30529b94acceSBarry Smith Notes: 3053e5d3d808SBarry Smith If the Amat matrix and Pmat matrix are different you must call MatAssemblyBegin/End() on 305416913363SBarry Smith each matrix. 305516913363SBarry Smith 3056895c21f2SBarry Smith If you know the operator Amat has a null space you can use MatSetNullSpace() and MatSetTransposeNullSpace() to supply the null 3057895c21f2SBarry Smith space to Amat and the KSP solvers will automatically use that null space as needed during the solution process. 3058895c21f2SBarry Smith 30598d359177SBarry Smith If using SNESComputeJacobianDefaultColor() to assemble a Jacobian, the ctx argument 3060a8a26c1eSJed Brown must be a MatFDColoring. 3061a8a26c1eSJed Brown 3062c3cc8fd1SJed Brown Other defect-correction schemes can be used by computing a different matrix in place of the Jacobian. One common 3063c3cc8fd1SJed Brown example is to use the "Picard linearization" which only differentiates through the highest order parts of each term. 3064c3cc8fd1SJed Brown 306536851e7fSLois Curfman McInnes Level: beginner 306636851e7fSLois Curfman McInnes 3067db781477SPatrick Sanan .seealso: `KSPSetOperators()`, `SNESSetFunction()`, `MatMFFDComputeJacobian()`, `SNESComputeJacobianDefaultColor()`, `MatStructure`, `J`, 3068db781477SPatrick Sanan `SNESSetPicard()`, `SNESJacobianFunction` 30699b94acceSBarry Smith @*/ 3070d1e9a80fSBarry Smith PetscErrorCode SNESSetJacobian(SNES snes,Mat Amat,Mat Pmat,PetscErrorCode (*J)(SNES,Vec,Mat,Mat,void*),void *ctx) 30719b94acceSBarry Smith { 30726cab3a1bSJed Brown DM dm; 30733a7fca6bSBarry Smith 30743a40ed3dSBarry Smith PetscFunctionBegin; 30750700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 3076e5d3d808SBarry Smith if (Amat) PetscValidHeaderSpecific(Amat,MAT_CLASSID,2); 3077e5d3d808SBarry Smith if (Pmat) PetscValidHeaderSpecific(Pmat,MAT_CLASSID,3); 3078e5d3d808SBarry Smith if (Amat) PetscCheckSameComm(snes,1,Amat,2); 3079e5d3d808SBarry Smith if (Pmat) PetscCheckSameComm(snes,1,Pmat,3); 30809566063dSJacob Faibussowitsch PetscCall(SNESGetDM(snes,&dm)); 30819566063dSJacob Faibussowitsch PetscCall(DMSNESSetJacobian(dm,J,ctx)); 3082e5d3d808SBarry Smith if (Amat) { 30839566063dSJacob Faibussowitsch PetscCall(PetscObjectReference((PetscObject)Amat)); 30849566063dSJacob Faibussowitsch PetscCall(MatDestroy(&snes->jacobian)); 3085f5af7f23SKarl Rupp 3086e5d3d808SBarry Smith snes->jacobian = Amat; 30873a7fca6bSBarry Smith } 3088e5d3d808SBarry Smith if (Pmat) { 30899566063dSJacob Faibussowitsch PetscCall(PetscObjectReference((PetscObject)Pmat)); 30909566063dSJacob Faibussowitsch PetscCall(MatDestroy(&snes->jacobian_pre)); 3091f5af7f23SKarl Rupp 3092e5d3d808SBarry Smith snes->jacobian_pre = Pmat; 30933a7fca6bSBarry Smith } 30943a40ed3dSBarry Smith PetscFunctionReturn(0); 30959b94acceSBarry Smith } 309662fef451SLois Curfman McInnes 3097c2aafc4cSSatish Balay /*@C 3098b4fd4287SBarry Smith SNESGetJacobian - Returns the Jacobian matrix and optionally the user 3099b4fd4287SBarry Smith provided context for evaluating the Jacobian. 3100b4fd4287SBarry Smith 3101c7afd0dbSLois Curfman McInnes Not Collective, but Mat object will be parallel if SNES object is 3102c7afd0dbSLois Curfman McInnes 3103b4fd4287SBarry Smith Input Parameter: 3104b4fd4287SBarry Smith . snes - the nonlinear solver context 3105b4fd4287SBarry Smith 3106b4fd4287SBarry Smith Output Parameters: 3107e5d3d808SBarry Smith + Amat - location to stash (approximate) Jacobian matrix (or NULL) 3108e5d3d808SBarry Smith . Pmat - location to stash matrix used to compute the preconditioner (or NULL) 3109411c0326SBarry Smith . J - location to put Jacobian function (or NULL), see SNESJacobianFunction for details on its calling sequence 31100298fd71SBarry Smith - ctx - location to stash Jacobian ctx (or NULL) 3111fee21e36SBarry Smith 311236851e7fSLois Curfman McInnes Level: advanced 311336851e7fSLois Curfman McInnes 3114db781477SPatrick Sanan .seealso: `SNESSetJacobian()`, `SNESComputeJacobian()`, `SNESJacobianFunction`, `SNESGetFunction()` 3115b4fd4287SBarry Smith @*/ 3116d1e9a80fSBarry Smith PetscErrorCode SNESGetJacobian(SNES snes,Mat *Amat,Mat *Pmat,PetscErrorCode (**J)(SNES,Vec,Mat,Mat,void*),void **ctx) 3117b4fd4287SBarry Smith { 31186cab3a1bSJed Brown DM dm; 31196cab3a1bSJed Brown 31203a40ed3dSBarry Smith PetscFunctionBegin; 31210700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 3122e5d3d808SBarry Smith if (Amat) *Amat = snes->jacobian; 3123e5d3d808SBarry Smith if (Pmat) *Pmat = snes->jacobian_pre; 31249566063dSJacob Faibussowitsch PetscCall(SNESGetDM(snes,&dm)); 3125800f99ffSJeremy L Thompson PetscCall(DMSNESGetJacobian(dm,J,ctx)); 31263a40ed3dSBarry Smith PetscFunctionReturn(0); 3127b4fd4287SBarry Smith } 3128b4fd4287SBarry Smith 312958b371f3SBarry Smith static PetscErrorCode SNESSetDefaultComputeJacobian(SNES snes) 313058b371f3SBarry Smith { 313158b371f3SBarry Smith DM dm; 313258b371f3SBarry Smith DMSNES sdm; 313358b371f3SBarry Smith 313458b371f3SBarry Smith PetscFunctionBegin; 31359566063dSJacob Faibussowitsch PetscCall(SNESGetDM(snes,&dm)); 31369566063dSJacob Faibussowitsch PetscCall(DMGetDMSNES(dm,&sdm)); 313758b371f3SBarry Smith if (!sdm->ops->computejacobian && snes->jacobian_pre) { 313858b371f3SBarry Smith DM dm; 313958b371f3SBarry Smith PetscBool isdense,ismf; 314058b371f3SBarry Smith 31419566063dSJacob Faibussowitsch PetscCall(SNESGetDM(snes,&dm)); 31429566063dSJacob Faibussowitsch PetscCall(PetscObjectTypeCompareAny((PetscObject)snes->jacobian_pre,&isdense,MATSEQDENSE,MATMPIDENSE,MATDENSE,NULL)); 31439566063dSJacob Faibussowitsch PetscCall(PetscObjectTypeCompareAny((PetscObject)snes->jacobian_pre,&ismf,MATMFFD,MATSHELL,NULL)); 314458b371f3SBarry Smith if (isdense) { 31459566063dSJacob Faibussowitsch PetscCall(DMSNESSetJacobian(dm,SNESComputeJacobianDefault,NULL)); 314658b371f3SBarry Smith } else if (!ismf) { 31479566063dSJacob Faibussowitsch PetscCall(DMSNESSetJacobian(dm,SNESComputeJacobianDefaultColor,NULL)); 314858b371f3SBarry Smith } 314958b371f3SBarry Smith } 315058b371f3SBarry Smith PetscFunctionReturn(0); 315158b371f3SBarry Smith } 315258b371f3SBarry Smith 31539b94acceSBarry Smith /*@ 31549b94acceSBarry Smith SNESSetUp - Sets up the internal data structures for the later use 3155272ac6f2SLois Curfman McInnes of a nonlinear solver. 31569b94acceSBarry Smith 3157fee21e36SBarry Smith Collective on SNES 3158fee21e36SBarry Smith 3159c7afd0dbSLois Curfman McInnes Input Parameters: 316070e92668SMatthew Knepley . snes - the SNES context 3161c7afd0dbSLois Curfman McInnes 3162272ac6f2SLois Curfman McInnes Notes: 3163272ac6f2SLois Curfman McInnes For basic use of the SNES solvers the user need not explicitly call 3164272ac6f2SLois Curfman McInnes SNESSetUp(), since these actions will automatically occur during 3165272ac6f2SLois Curfman McInnes the call to SNESSolve(). However, if one wishes to control this 3166272ac6f2SLois Curfman McInnes phase separately, SNESSetUp() should be called after SNESCreate() 3167272ac6f2SLois Curfman McInnes and optional routines of the form SNESSetXXX(), but before SNESSolve(). 3168272ac6f2SLois Curfman McInnes 316936851e7fSLois Curfman McInnes Level: advanced 317036851e7fSLois Curfman McInnes 3171db781477SPatrick Sanan .seealso: `SNESCreate()`, `SNESSolve()`, `SNESDestroy()` 31729b94acceSBarry Smith @*/ 31737087cfbeSBarry Smith PetscErrorCode SNESSetUp(SNES snes) 31749b94acceSBarry Smith { 31756cab3a1bSJed Brown DM dm; 3176942e3340SBarry Smith DMSNES sdm; 3177c35f09e5SBarry Smith SNESLineSearch linesearch, pclinesearch; 31786e2a1849SPeter Brune void *lsprectx,*lspostctx; 31796b2b7091SBarry Smith PetscErrorCode (*precheck)(SNESLineSearch,Vec,Vec,PetscBool*,void*); 31806b2b7091SBarry Smith PetscErrorCode (*postcheck)(SNESLineSearch,Vec,Vec,Vec,PetscBool*,PetscBool*,void*); 31816e2a1849SPeter Brune PetscErrorCode (*func)(SNES,Vec,Vec,void*); 31826e2a1849SPeter Brune Vec f,fpc; 31836e2a1849SPeter Brune void *funcctx; 3184d1e9a80fSBarry Smith PetscErrorCode (*jac)(SNES,Vec,Mat,Mat,void*); 31851eb13d49SPeter Brune void *jacctx,*appctx; 318632b97717SPeter Brune Mat j,jpre; 31873a40ed3dSBarry Smith 31883a40ed3dSBarry Smith PetscFunctionBegin; 31890700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 31904dc4c822SBarry Smith if (snes->setupcalled) PetscFunctionReturn(0); 31919566063dSJacob Faibussowitsch PetscCall(PetscLogEventBegin(SNES_Setup,snes,0,0,0)); 31929b94acceSBarry Smith 31937adad957SLisandro Dalcin if (!((PetscObject)snes)->type_name) { 31949566063dSJacob Faibussowitsch PetscCall(SNESSetType(snes,SNESNEWTONLS)); 319585385478SLisandro Dalcin } 319685385478SLisandro Dalcin 31979566063dSJacob Faibussowitsch PetscCall(SNESGetFunction(snes,&snes->vec_func,NULL,NULL)); 319858c9b817SLisandro Dalcin 31999566063dSJacob Faibussowitsch PetscCall(SNESGetDM(snes,&dm)); 32009566063dSJacob Faibussowitsch PetscCall(DMGetDMSNES(dm,&sdm)); 32019566063dSJacob Faibussowitsch PetscCall(SNESSetDefaultComputeJacobian(snes)); 320258b371f3SBarry Smith 32036cab3a1bSJed Brown if (!snes->vec_func) { 32049566063dSJacob Faibussowitsch PetscCall(DMCreateGlobalVector(dm,&snes->vec_func)); 3205214df951SJed Brown } 3206efd51863SBarry Smith 320722d28d08SBarry Smith if (!snes->ksp) { 32089566063dSJacob Faibussowitsch PetscCall(SNESGetKSP(snes, &snes->ksp)); 320922d28d08SBarry Smith } 3210b710008aSBarry Smith 3211d8d34be6SBarry Smith if (snes->linesearch) { 32129566063dSJacob Faibussowitsch PetscCall(SNESGetLineSearch(snes, &snes->linesearch)); 32139566063dSJacob Faibussowitsch PetscCall(SNESLineSearchSetFunction(snes->linesearch,SNESComputeFunction)); 3214d8d34be6SBarry Smith } 32159e764e56SPeter Brune 3216b552625fSStefano Zampini if (snes->npc && snes->npcside == PC_LEFT) { 3217172a4300SPeter Brune snes->mf = PETSC_TRUE; 3218172a4300SPeter Brune snes->mf_operator = PETSC_FALSE; 3219172a4300SPeter Brune } 3220d8f46077SPeter Brune 3221efd4aadfSBarry Smith if (snes->npc) { 32226e2a1849SPeter Brune /* copy the DM over */ 32239566063dSJacob Faibussowitsch PetscCall(SNESGetDM(snes,&dm)); 32249566063dSJacob Faibussowitsch PetscCall(SNESSetDM(snes->npc,dm)); 32256e2a1849SPeter Brune 32269566063dSJacob Faibussowitsch PetscCall(SNESGetFunction(snes,&f,&func,&funcctx)); 32279566063dSJacob Faibussowitsch PetscCall(VecDuplicate(f,&fpc)); 32289566063dSJacob Faibussowitsch PetscCall(SNESSetFunction(snes->npc,fpc,func,funcctx)); 32299566063dSJacob Faibussowitsch PetscCall(SNESGetJacobian(snes,&j,&jpre,&jac,&jacctx)); 32309566063dSJacob Faibussowitsch PetscCall(SNESSetJacobian(snes->npc,j,jpre,jac,jacctx)); 32319566063dSJacob Faibussowitsch PetscCall(SNESGetApplicationContext(snes,&appctx)); 32329566063dSJacob Faibussowitsch PetscCall(SNESSetApplicationContext(snes->npc,appctx)); 32339566063dSJacob Faibussowitsch PetscCall(VecDestroy(&fpc)); 32346e2a1849SPeter Brune 32356e2a1849SPeter Brune /* copy the function pointers over */ 32369566063dSJacob Faibussowitsch PetscCall(PetscObjectCopyFortranFunctionPointers((PetscObject)snes,(PetscObject)snes->npc)); 32376e2a1849SPeter Brune 32386e2a1849SPeter Brune /* default to 1 iteration */ 32399566063dSJacob Faibussowitsch PetscCall(SNESSetTolerances(snes->npc,0.0,0.0,0.0,1,snes->npc->max_funcs)); 3240efd4aadfSBarry Smith if (snes->npcside == PC_RIGHT) { 32419566063dSJacob Faibussowitsch PetscCall(SNESSetNormSchedule(snes->npc,SNES_NORM_FINAL_ONLY)); 3242a9936a0cSPeter Brune } else { 32439566063dSJacob Faibussowitsch PetscCall(SNESSetNormSchedule(snes->npc,SNES_NORM_NONE)); 3244a9936a0cSPeter Brune } 32459566063dSJacob Faibussowitsch PetscCall(SNESSetFromOptions(snes->npc)); 32466e2a1849SPeter Brune 32476e2a1849SPeter Brune /* copy the line search context over */ 3248d8d34be6SBarry Smith if (snes->linesearch && snes->npc->linesearch) { 32499566063dSJacob Faibussowitsch PetscCall(SNESGetLineSearch(snes,&linesearch)); 32509566063dSJacob Faibussowitsch PetscCall(SNESGetLineSearch(snes->npc,&pclinesearch)); 32519566063dSJacob Faibussowitsch PetscCall(SNESLineSearchGetPreCheck(linesearch,&precheck,&lsprectx)); 32529566063dSJacob Faibussowitsch PetscCall(SNESLineSearchGetPostCheck(linesearch,&postcheck,&lspostctx)); 32539566063dSJacob Faibussowitsch PetscCall(SNESLineSearchSetPreCheck(pclinesearch,precheck,lsprectx)); 32549566063dSJacob Faibussowitsch PetscCall(SNESLineSearchSetPostCheck(pclinesearch,postcheck,lspostctx)); 32559566063dSJacob Faibussowitsch PetscCall(PetscObjectCopyFortranFunctionPointers((PetscObject)linesearch, (PetscObject)pclinesearch)); 32566e2a1849SPeter Brune } 3257d8d34be6SBarry Smith } 32581baa6e33SBarry Smith if (snes->mf) PetscCall(SNESSetUpMatrixFree_Private(snes, snes->mf_operator, snes->mf_version)); 325932b97717SPeter Brune if (snes->ops->usercompute && !snes->user) { 32609566063dSJacob Faibussowitsch PetscCall((*snes->ops->usercompute)(snes,(void**)&snes->user)); 326132b97717SPeter Brune } 32626e2a1849SPeter Brune 326337ec4e1aSPeter Brune snes->jac_iter = 0; 326437ec4e1aSPeter Brune snes->pre_iter = 0; 326537ec4e1aSPeter Brune 3266*dbbe0bcdSBarry Smith PetscTryTypeMethod(snes,setup); 326758c9b817SLisandro Dalcin 32689566063dSJacob Faibussowitsch PetscCall(SNESSetDefaultComputeJacobian(snes)); 326958b371f3SBarry Smith 3270b552625fSStefano Zampini if (snes->npc && snes->npcside == PC_LEFT) { 32716c67d002SPeter Brune if (snes->functype == SNES_FUNCTION_PRECONDITIONED) { 3272d8d34be6SBarry Smith if (snes->linesearch) { 32739566063dSJacob Faibussowitsch PetscCall(SNESGetLineSearch(snes,&linesearch)); 32749566063dSJacob Faibussowitsch PetscCall(SNESLineSearchSetFunction(linesearch,SNESComputeFunctionDefaultNPC)); 32756c67d002SPeter Brune } 32766c67d002SPeter Brune } 3277d8d34be6SBarry Smith } 32789566063dSJacob Faibussowitsch PetscCall(PetscLogEventEnd(SNES_Setup,snes,0,0,0)); 32797aaed0d8SBarry Smith snes->setupcalled = PETSC_TRUE; 32803a40ed3dSBarry Smith PetscFunctionReturn(0); 32819b94acceSBarry Smith } 32829b94acceSBarry Smith 328337596af1SLisandro Dalcin /*@ 328437596af1SLisandro Dalcin SNESReset - Resets a SNES context to the snessetupcalled = 0 state and removes any allocated Vecs and Mats 328537596af1SLisandro Dalcin 328637596af1SLisandro Dalcin Collective on SNES 328737596af1SLisandro Dalcin 328837596af1SLisandro Dalcin Input Parameter: 328937596af1SLisandro Dalcin . snes - iterative context obtained from SNESCreate() 329037596af1SLisandro Dalcin 3291d25893d9SBarry Smith Level: intermediate 3292d25893d9SBarry Smith 329395452b02SPatrick Sanan Notes: 329495452b02SPatrick Sanan Also calls the application context destroy routine set with SNESSetComputeApplicationContext() 329537596af1SLisandro Dalcin 3296db781477SPatrick Sanan .seealso: `SNESCreate()`, `SNESSetUp()`, `SNESSolve()` 329737596af1SLisandro Dalcin @*/ 329837596af1SLisandro Dalcin PetscErrorCode SNESReset(SNES snes) 329937596af1SLisandro Dalcin { 330037596af1SLisandro Dalcin PetscFunctionBegin; 330137596af1SLisandro Dalcin PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 3302d25893d9SBarry Smith if (snes->ops->userdestroy && snes->user) { 33039566063dSJacob Faibussowitsch PetscCall((*snes->ops->userdestroy)((void**)&snes->user)); 33040298fd71SBarry Smith snes->user = NULL; 3305d25893d9SBarry Smith } 33061baa6e33SBarry Smith if (snes->npc) PetscCall(SNESReset(snes->npc)); 33078a23116dSBarry Smith 3308*dbbe0bcdSBarry Smith PetscTryTypeMethod(snes,reset); 33091baa6e33SBarry Smith if (snes->ksp) PetscCall(KSPReset(snes->ksp)); 33109e764e56SPeter Brune 33111baa6e33SBarry Smith if (snes->linesearch) PetscCall(SNESLineSearchReset(snes->linesearch)); 33129e764e56SPeter Brune 33139566063dSJacob Faibussowitsch PetscCall(VecDestroy(&snes->vec_rhs)); 33149566063dSJacob Faibussowitsch PetscCall(VecDestroy(&snes->vec_sol)); 33159566063dSJacob Faibussowitsch PetscCall(VecDestroy(&snes->vec_sol_update)); 33169566063dSJacob Faibussowitsch PetscCall(VecDestroy(&snes->vec_func)); 33179566063dSJacob Faibussowitsch PetscCall(MatDestroy(&snes->jacobian)); 33189566063dSJacob Faibussowitsch PetscCall(MatDestroy(&snes->jacobian_pre)); 33199566063dSJacob Faibussowitsch PetscCall(MatDestroy(&snes->picard)); 33209566063dSJacob Faibussowitsch PetscCall(VecDestroyVecs(snes->nwork,&snes->work)); 33219566063dSJacob Faibussowitsch PetscCall(VecDestroyVecs(snes->nvwork,&snes->vwork)); 3322f5af7f23SKarl Rupp 332340fdac6aSLawrence Mitchell snes->alwayscomputesfinalresidual = PETSC_FALSE; 332440fdac6aSLawrence Mitchell 332537596af1SLisandro Dalcin snes->nwork = snes->nvwork = 0; 332637596af1SLisandro Dalcin snes->setupcalled = PETSC_FALSE; 332737596af1SLisandro Dalcin PetscFunctionReturn(0); 332837596af1SLisandro Dalcin } 332937596af1SLisandro Dalcin 333052baeb72SSatish Balay /*@ 3331c4421ceaSFande Kong SNESConvergedReasonViewCancel - Clears all the reasonview functions for a SNES object. 3332c4421ceaSFande Kong 3333c4421ceaSFande Kong Collective on SNES 3334c4421ceaSFande Kong 3335c4421ceaSFande Kong Input Parameter: 3336c4421ceaSFande Kong . snes - iterative context obtained from SNESCreate() 3337c4421ceaSFande Kong 3338c4421ceaSFande Kong Level: intermediate 3339c4421ceaSFande Kong 3340db781477SPatrick Sanan .seealso: `SNESCreate()`, `SNESDestroy()`, `SNESReset()` 3341c4421ceaSFande Kong @*/ 3342c4421ceaSFande Kong PetscErrorCode SNESConvergedReasonViewCancel(SNES snes) 3343c4421ceaSFande Kong { 3344c4421ceaSFande Kong PetscInt i; 3345c4421ceaSFande Kong 3346c4421ceaSFande Kong PetscFunctionBegin; 3347c4421ceaSFande Kong PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 3348c4421ceaSFande Kong for (i=0; i<snes->numberreasonviews; i++) { 3349c4421ceaSFande Kong if (snes->reasonviewdestroy[i]) { 33509566063dSJacob Faibussowitsch PetscCall((*snes->reasonviewdestroy[i])(&snes->reasonviewcontext[i])); 3351c4421ceaSFande Kong } 3352c4421ceaSFande Kong } 3353c4421ceaSFande Kong snes->numberreasonviews = 0; 3354c4421ceaSFande Kong PetscFunctionReturn(0); 3355c4421ceaSFande Kong } 3356c4421ceaSFande Kong 33571fb7b255SJunchao Zhang /*@C 33589b94acceSBarry Smith SNESDestroy - Destroys the nonlinear solver context that was created 33599b94acceSBarry Smith with SNESCreate(). 33609b94acceSBarry Smith 3361c7afd0dbSLois Curfman McInnes Collective on SNES 3362c7afd0dbSLois Curfman McInnes 33639b94acceSBarry Smith Input Parameter: 33649b94acceSBarry Smith . snes - the SNES context 33659b94acceSBarry Smith 336636851e7fSLois Curfman McInnes Level: beginner 336736851e7fSLois Curfman McInnes 3368db781477SPatrick Sanan .seealso: `SNESCreate()`, `SNESSolve()` 33699b94acceSBarry Smith @*/ 33706bf464f9SBarry Smith PetscErrorCode SNESDestroy(SNES *snes) 33719b94acceSBarry Smith { 33723a40ed3dSBarry Smith PetscFunctionBegin; 33736bf464f9SBarry Smith if (!*snes) PetscFunctionReturn(0); 33746bf464f9SBarry Smith PetscValidHeaderSpecific((*snes),SNES_CLASSID,1); 33759e5d0892SLisandro Dalcin if (--((PetscObject)(*snes))->refct > 0) {*snes = NULL; PetscFunctionReturn(0);} 3376d4bb536fSBarry Smith 33779566063dSJacob Faibussowitsch PetscCall(SNESReset((*snes))); 33789566063dSJacob Faibussowitsch PetscCall(SNESDestroy(&(*snes)->npc)); 33796b8b9a38SLisandro Dalcin 3380e04113cfSBarry Smith /* if memory was published with SAWs then destroy it */ 33819566063dSJacob Faibussowitsch PetscCall(PetscObjectSAWsViewOff((PetscObject)*snes)); 3382*dbbe0bcdSBarry Smith PetscTryTypeMethod((*snes),destroy); 33836d4c513bSLisandro Dalcin 33849566063dSJacob Faibussowitsch if ((*snes)->dm) PetscCall(DMCoarsenHookRemove((*snes)->dm,DMCoarsenHook_SNESVecSol,DMRestrictHook_SNESVecSol,*snes)); 33859566063dSJacob Faibussowitsch PetscCall(DMDestroy(&(*snes)->dm)); 33869566063dSJacob Faibussowitsch PetscCall(KSPDestroy(&(*snes)->ksp)); 33879566063dSJacob Faibussowitsch PetscCall(SNESLineSearchDestroy(&(*snes)->linesearch)); 33886b8b9a38SLisandro Dalcin 33899566063dSJacob Faibussowitsch PetscCall(PetscFree((*snes)->kspconvctx)); 33906bf464f9SBarry Smith if ((*snes)->ops->convergeddestroy) { 33919566063dSJacob Faibussowitsch PetscCall((*(*snes)->ops->convergeddestroy)((*snes)->cnvP)); 33926b8b9a38SLisandro Dalcin } 3393071fcb05SBarry Smith if ((*snes)->conv_hist_alloc) { 33949566063dSJacob Faibussowitsch PetscCall(PetscFree2((*snes)->conv_hist,(*snes)->conv_hist_its)); 339558c9b817SLisandro Dalcin } 33969566063dSJacob Faibussowitsch PetscCall(SNESMonitorCancel((*snes))); 33979566063dSJacob Faibussowitsch PetscCall(SNESConvergedReasonViewCancel((*snes))); 33989566063dSJacob Faibussowitsch PetscCall(PetscHeaderDestroy(snes)); 33993a40ed3dSBarry Smith PetscFunctionReturn(0); 34009b94acceSBarry Smith } 34019b94acceSBarry Smith 34029b94acceSBarry Smith /* ----------- Routines to set solver parameters ---------- */ 34039b94acceSBarry Smith 3404a8054027SBarry Smith /*@ 3405a8054027SBarry Smith SNESSetLagPreconditioner - Determines when the preconditioner is rebuilt in the nonlinear solve. 3406a8054027SBarry Smith 34073f9fe445SBarry Smith Logically Collective on SNES 3408a8054027SBarry Smith 3409a8054027SBarry Smith Input Parameters: 3410a8054027SBarry Smith + snes - the SNES context 3411d8e291bfSBarry Smith - lag - 1 means rebuild every time the Jacobian is computed within a single nonlinear solve, 2 means every second time 34123b4f5425SBarry Smith the Jacobian is built etc. -2 indicates rebuild preconditioner at next chance but then never rebuild after that 3413a8054027SBarry Smith 3414a8054027SBarry Smith Options Database Keys: 34153d5a8a6aSBarry Smith + -snes_lag_jacobian_persists <true,false> - sets the persistence 34163d5a8a6aSBarry Smith . -snes_lag_jacobian <-2,1,2,...> - sets the lag 34173d5a8a6aSBarry Smith . -snes_lag_preconditioner_persists <true,false> - sets the persistence 34183d5a8a6aSBarry Smith - -snes_lag_preconditioner <-2,1,2,...> - sets the lag 3419a8054027SBarry Smith 3420a8054027SBarry Smith Notes: 3421a8054027SBarry Smith The default is 1 34223d5a8a6aSBarry Smith The preconditioner is ALWAYS built in the first iteration of a nonlinear solve unless lag is -1 or SNESSetLagPreconditionerPersists() was called 3423d8e291bfSBarry Smith 3424d8e291bfSBarry Smith SNESSetLagPreconditionerPersists() allows using the same uniform lagging (for example every second solve) across multiple solves. 3425a8054027SBarry Smith 3426a8054027SBarry Smith Level: intermediate 3427a8054027SBarry Smith 3428db781477SPatrick Sanan .seealso: `SNESSetTrustRegionTolerance()`, `SNESGetLagPreconditioner()`, `SNESSetLagJacobian()`, `SNESGetLagJacobian()`, `SNESSetLagPreconditionerPersists()`, 3429db781477SPatrick Sanan `SNESSetLagJacobianPersists()` 3430a8054027SBarry Smith 3431a8054027SBarry Smith @*/ 34327087cfbeSBarry Smith PetscErrorCode SNESSetLagPreconditioner(SNES snes,PetscInt lag) 3433a8054027SBarry Smith { 3434a8054027SBarry Smith PetscFunctionBegin; 34350700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 34365f80ce2aSJacob Faibussowitsch PetscCheck(lag >= -2,PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Lag must be -2, -1, 1 or greater"); 34375f80ce2aSJacob Faibussowitsch PetscCheck(lag,PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Lag cannot be 0"); 3438c5eb9154SBarry Smith PetscValidLogicalCollectiveInt(snes,lag,2); 3439a8054027SBarry Smith snes->lagpreconditioner = lag; 3440a8054027SBarry Smith PetscFunctionReturn(0); 3441a8054027SBarry Smith } 3442a8054027SBarry Smith 3443efd51863SBarry Smith /*@ 3444efd51863SBarry Smith SNESSetGridSequence - sets the number of steps of grid sequencing that SNES does 3445efd51863SBarry Smith 3446efd51863SBarry Smith Logically Collective on SNES 3447efd51863SBarry Smith 3448efd51863SBarry Smith Input Parameters: 3449efd51863SBarry Smith + snes - the SNES context 3450efd51863SBarry Smith - steps - the number of refinements to do, defaults to 0 3451efd51863SBarry Smith 3452efd51863SBarry Smith Options Database Keys: 345367b8a455SSatish Balay . -snes_grid_sequence <steps> - Use grid sequencing to generate initial guess 3454efd51863SBarry Smith 3455efd51863SBarry Smith Level: intermediate 3456efd51863SBarry Smith 3457c0df2a02SJed Brown Notes: 3458c0df2a02SJed Brown Use SNESGetSolution() to extract the fine grid solution after grid sequencing. 3459c0df2a02SJed Brown 3460db781477SPatrick Sanan .seealso: `SNESSetTrustRegionTolerance()`, `SNESGetLagPreconditioner()`, `SNESSetLagJacobian()`, `SNESGetLagJacobian()`, `SNESGetGridSequence()` 3461efd51863SBarry Smith 3462efd51863SBarry Smith @*/ 3463efd51863SBarry Smith PetscErrorCode SNESSetGridSequence(SNES snes,PetscInt steps) 3464efd51863SBarry Smith { 3465efd51863SBarry Smith PetscFunctionBegin; 3466efd51863SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 3467efd51863SBarry Smith PetscValidLogicalCollectiveInt(snes,steps,2); 3468efd51863SBarry Smith snes->gridsequence = steps; 3469efd51863SBarry Smith PetscFunctionReturn(0); 3470efd51863SBarry Smith } 3471efd51863SBarry Smith 3472fa19ca70SBarry Smith /*@ 3473fa19ca70SBarry Smith SNESGetGridSequence - gets the number of steps of grid sequencing that SNES does 3474fa19ca70SBarry Smith 3475fa19ca70SBarry Smith Logically Collective on SNES 3476fa19ca70SBarry Smith 3477fa19ca70SBarry Smith Input Parameter: 3478fa19ca70SBarry Smith . snes - the SNES context 3479fa19ca70SBarry Smith 3480fa19ca70SBarry Smith Output Parameter: 3481fa19ca70SBarry Smith . steps - the number of refinements to do, defaults to 0 3482fa19ca70SBarry Smith 3483fa19ca70SBarry Smith Options Database Keys: 348467b8a455SSatish Balay . -snes_grid_sequence <steps> - set number of refinements 3485fa19ca70SBarry Smith 3486fa19ca70SBarry Smith Level: intermediate 3487fa19ca70SBarry Smith 3488fa19ca70SBarry Smith Notes: 3489fa19ca70SBarry Smith Use SNESGetSolution() to extract the fine grid solution after grid sequencing. 3490fa19ca70SBarry Smith 3491db781477SPatrick Sanan .seealso: `SNESSetTrustRegionTolerance()`, `SNESGetLagPreconditioner()`, `SNESSetLagJacobian()`, `SNESGetLagJacobian()`, `SNESSetGridSequence()` 3492fa19ca70SBarry Smith 3493fa19ca70SBarry Smith @*/ 3494fa19ca70SBarry Smith PetscErrorCode SNESGetGridSequence(SNES snes,PetscInt *steps) 3495fa19ca70SBarry Smith { 3496fa19ca70SBarry Smith PetscFunctionBegin; 3497fa19ca70SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 3498fa19ca70SBarry Smith *steps = snes->gridsequence; 3499fa19ca70SBarry Smith PetscFunctionReturn(0); 3500fa19ca70SBarry Smith } 3501fa19ca70SBarry Smith 3502a8054027SBarry Smith /*@ 3503a8054027SBarry Smith SNESGetLagPreconditioner - Indicates how often the preconditioner is rebuilt 3504a8054027SBarry Smith 35053f9fe445SBarry Smith Not Collective 3506a8054027SBarry Smith 3507a8054027SBarry Smith Input Parameter: 3508a8054027SBarry Smith . snes - the SNES context 3509a8054027SBarry Smith 3510a8054027SBarry Smith Output Parameter: 3511a8054027SBarry 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 35123b4f5425SBarry Smith the Jacobian is built etc. -2 indicates rebuild preconditioner at next chance but then never rebuild after that 3513a8054027SBarry Smith 3514a8054027SBarry Smith Options Database Keys: 35153d5a8a6aSBarry Smith + -snes_lag_jacobian_persists <true,false> - sets the persistence 35163d5a8a6aSBarry Smith . -snes_lag_jacobian <-2,1,2,...> - sets the lag 35173d5a8a6aSBarry Smith . -snes_lag_preconditioner_persists <true,false> - sets the persistence 35183d5a8a6aSBarry Smith - -snes_lag_preconditioner <-2,1,2,...> - sets the lag 3519a8054027SBarry Smith 3520a8054027SBarry Smith Notes: 3521a8054027SBarry Smith The default is 1 3522a8054027SBarry Smith The preconditioner is ALWAYS built in the first iteration of a nonlinear solve unless lag is -1 3523a8054027SBarry Smith 3524a8054027SBarry Smith Level: intermediate 3525a8054027SBarry Smith 3526db781477SPatrick Sanan .seealso: `SNESSetTrustRegionTolerance()`, `SNESSetLagPreconditioner()`, `SNESSetLagJacobianPersists()`, `SNESSetLagPreconditionerPersists()` 3527a8054027SBarry Smith 3528a8054027SBarry Smith @*/ 35297087cfbeSBarry Smith PetscErrorCode SNESGetLagPreconditioner(SNES snes,PetscInt *lag) 3530a8054027SBarry Smith { 3531a8054027SBarry Smith PetscFunctionBegin; 35320700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 3533a8054027SBarry Smith *lag = snes->lagpreconditioner; 3534a8054027SBarry Smith PetscFunctionReturn(0); 3535a8054027SBarry Smith } 3536a8054027SBarry Smith 3537e35cf81dSBarry Smith /*@ 3538e35cf81dSBarry Smith SNESSetLagJacobian - Determines when the Jacobian is rebuilt in the nonlinear solve. See SNESSetLagPreconditioner() for determining how 3539e35cf81dSBarry Smith often the preconditioner is rebuilt. 3540e35cf81dSBarry Smith 35413f9fe445SBarry Smith Logically Collective on SNES 3542e35cf81dSBarry Smith 3543e35cf81dSBarry Smith Input Parameters: 3544e35cf81dSBarry Smith + snes - the SNES context 3545e35cf81dSBarry 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 3546fe3ffe1eSBarry Smith the Jacobian is built etc. -2 means rebuild at next chance but then never again 3547e35cf81dSBarry Smith 3548e35cf81dSBarry Smith Options Database Keys: 35493d5a8a6aSBarry Smith + -snes_lag_jacobian_persists <true,false> - sets the persistence 35503d5a8a6aSBarry Smith . -snes_lag_jacobian <-2,1,2,...> - sets the lag 35513d5a8a6aSBarry Smith . -snes_lag_preconditioner_persists <true,false> - sets the persistence 35523d5a8a6aSBarry Smith - -snes_lag_preconditioner <-2,1,2,...> - sets the lag. 3553e35cf81dSBarry Smith 3554e35cf81dSBarry Smith Notes: 3555e35cf81dSBarry Smith The default is 1 3556e35cf81dSBarry Smith The Jacobian is ALWAYS built in the first iteration of a nonlinear solve unless lag is -1 3557fe3ffe1eSBarry 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 3558fe3ffe1eSBarry Smith at the next Newton step but never again (unless it is reset to another value) 3559e35cf81dSBarry Smith 3560e35cf81dSBarry Smith Level: intermediate 3561e35cf81dSBarry Smith 3562db781477SPatrick Sanan .seealso: `SNESSetTrustRegionTolerance()`, `SNESGetLagPreconditioner()`, `SNESSetLagPreconditioner()`, `SNESGetLagJacobianPersists()`, `SNESSetLagPreconditionerPersists()` 3563e35cf81dSBarry Smith 3564e35cf81dSBarry Smith @*/ 35657087cfbeSBarry Smith PetscErrorCode SNESSetLagJacobian(SNES snes,PetscInt lag) 3566e35cf81dSBarry Smith { 3567e35cf81dSBarry Smith PetscFunctionBegin; 35680700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 35695f80ce2aSJacob Faibussowitsch PetscCheck(lag >= -2,PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Lag must be -2, -1, 1 or greater"); 35705f80ce2aSJacob Faibussowitsch PetscCheck(lag,PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Lag cannot be 0"); 3571c5eb9154SBarry Smith PetscValidLogicalCollectiveInt(snes,lag,2); 3572e35cf81dSBarry Smith snes->lagjacobian = lag; 3573e35cf81dSBarry Smith PetscFunctionReturn(0); 3574e35cf81dSBarry Smith } 3575e35cf81dSBarry Smith 3576e35cf81dSBarry Smith /*@ 3577e35cf81dSBarry Smith SNESGetLagJacobian - Indicates how often the Jacobian is rebuilt. See SNESGetLagPreconditioner() to determine when the preconditioner is rebuilt 3578e35cf81dSBarry Smith 35793f9fe445SBarry Smith Not Collective 3580e35cf81dSBarry Smith 3581e35cf81dSBarry Smith Input Parameter: 3582e35cf81dSBarry Smith . snes - the SNES context 3583e35cf81dSBarry Smith 3584e35cf81dSBarry Smith Output Parameter: 3585e35cf81dSBarry 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 3586e35cf81dSBarry Smith the Jacobian is built etc. 3587e35cf81dSBarry Smith 3588e35cf81dSBarry Smith Notes: 3589e35cf81dSBarry Smith The default is 1 35903d5a8a6aSBarry Smith The jacobian is ALWAYS built in the first iteration of a nonlinear solve unless lag is -1 or SNESSetLagJacobianPersists() was called. 3591e35cf81dSBarry Smith 3592e35cf81dSBarry Smith Level: intermediate 3593e35cf81dSBarry Smith 3594db781477SPatrick Sanan .seealso: `SNESSetTrustRegionTolerance()`, `SNESSetLagJacobian()`, `SNESSetLagPreconditioner()`, `SNESGetLagPreconditioner()`, `SNESSetLagJacobianPersists()`, `SNESSetLagPreconditionerPersists()` 3595e35cf81dSBarry Smith 3596e35cf81dSBarry Smith @*/ 35977087cfbeSBarry Smith PetscErrorCode SNESGetLagJacobian(SNES snes,PetscInt *lag) 3598e35cf81dSBarry Smith { 3599e35cf81dSBarry Smith PetscFunctionBegin; 36000700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 3601e35cf81dSBarry Smith *lag = snes->lagjacobian; 3602e35cf81dSBarry Smith PetscFunctionReturn(0); 3603e35cf81dSBarry Smith } 3604e35cf81dSBarry Smith 360537ec4e1aSPeter Brune /*@ 360637ec4e1aSPeter Brune SNESSetLagJacobianPersists - Set whether or not the Jacobian lagging persists through multiple solves 360737ec4e1aSPeter Brune 360837ec4e1aSPeter Brune Logically collective on SNES 360937ec4e1aSPeter Brune 3610d8d19677SJose E. Roman Input Parameters: 361137ec4e1aSPeter Brune + snes - the SNES context 36129d7e2deaSPeter Brune - flg - jacobian lagging persists if true 361337ec4e1aSPeter Brune 361437ec4e1aSPeter Brune Options Database Keys: 36153d5a8a6aSBarry Smith + -snes_lag_jacobian_persists <true,false> - sets the persistence 36163d5a8a6aSBarry Smith . -snes_lag_jacobian <-2,1,2,...> - sets the lag 36173d5a8a6aSBarry Smith . -snes_lag_preconditioner_persists <true,false> - sets the persistence 36183d5a8a6aSBarry Smith - -snes_lag_preconditioner <-2,1,2,...> - sets the lag 36193d5a8a6aSBarry Smith 362095452b02SPatrick Sanan Notes: 362195452b02SPatrick Sanan This is useful both for nonlinear preconditioning, where it's appropriate to have the Jacobian be stale by 362237ec4e1aSPeter Brune several solves, and for implicit time-stepping, where Jacobian lagging in the inner nonlinear solve over several 362337ec4e1aSPeter Brune timesteps may present huge efficiency gains. 362437ec4e1aSPeter Brune 362537ec4e1aSPeter Brune Level: developer 362637ec4e1aSPeter Brune 3627db781477SPatrick Sanan .seealso: `SNESSetLagPreconditionerPersists()`, `SNESSetLagJacobian()`, `SNESGetLagJacobian()`, `SNESGetNPC()`, `SNESSetLagJacobianPersists()` 362837ec4e1aSPeter Brune 362937ec4e1aSPeter Brune @*/ 363037ec4e1aSPeter Brune PetscErrorCode SNESSetLagJacobianPersists(SNES snes,PetscBool flg) 363137ec4e1aSPeter Brune { 363237ec4e1aSPeter Brune PetscFunctionBegin; 363337ec4e1aSPeter Brune PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 363437ec4e1aSPeter Brune PetscValidLogicalCollectiveBool(snes,flg,2); 363537ec4e1aSPeter Brune snes->lagjac_persist = flg; 363637ec4e1aSPeter Brune PetscFunctionReturn(0); 363737ec4e1aSPeter Brune } 363837ec4e1aSPeter Brune 363937ec4e1aSPeter Brune /*@ 3640d8e291bfSBarry Smith SNESSetLagPreconditionerPersists - Set whether or not the preconditioner lagging persists through multiple nonlinear solves 364137ec4e1aSPeter Brune 364237ec4e1aSPeter Brune Logically Collective on SNES 364337ec4e1aSPeter Brune 3644d8d19677SJose E. Roman Input Parameters: 364537ec4e1aSPeter Brune + snes - the SNES context 36469d7e2deaSPeter Brune - flg - preconditioner lagging persists if true 364737ec4e1aSPeter Brune 364837ec4e1aSPeter Brune Options Database Keys: 36493d5a8a6aSBarry Smith + -snes_lag_jacobian_persists <true,false> - sets the persistence 36503d5a8a6aSBarry Smith . -snes_lag_jacobian <-2,1,2,...> - sets the lag 36513d5a8a6aSBarry Smith . -snes_lag_preconditioner_persists <true,false> - sets the persistence 36523d5a8a6aSBarry Smith - -snes_lag_preconditioner <-2,1,2,...> - sets the lag 365337ec4e1aSPeter Brune 365495452b02SPatrick Sanan Notes: 365595452b02SPatrick Sanan This is useful both for nonlinear preconditioning, where it's appropriate to have the preconditioner be stale 365637ec4e1aSPeter Brune by several solves, and for implicit time-stepping, where preconditioner lagging in the inner nonlinear solve over 365737ec4e1aSPeter Brune several timesteps may present huge efficiency gains. 365837ec4e1aSPeter Brune 365937ec4e1aSPeter Brune Level: developer 366037ec4e1aSPeter Brune 3661db781477SPatrick Sanan .seealso: `SNESSetLagJacobianPersists()`, `SNESSetLagJacobian()`, `SNESGetLagJacobian()`, `SNESGetNPC()`, `SNESSetLagPreconditioner()` 366237ec4e1aSPeter Brune 366337ec4e1aSPeter Brune @*/ 366437ec4e1aSPeter Brune PetscErrorCode SNESSetLagPreconditionerPersists(SNES snes,PetscBool flg) 366537ec4e1aSPeter Brune { 366637ec4e1aSPeter Brune PetscFunctionBegin; 366737ec4e1aSPeter Brune PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 366837ec4e1aSPeter Brune PetscValidLogicalCollectiveBool(snes,flg,2); 366937ec4e1aSPeter Brune snes->lagpre_persist = flg; 367037ec4e1aSPeter Brune PetscFunctionReturn(0); 367137ec4e1aSPeter Brune } 367237ec4e1aSPeter Brune 36739b94acceSBarry Smith /*@ 3674be5caee7SBarry Smith SNESSetForceIteration - force SNESSolve() to take at least one iteration regardless of the initial residual norm 3675be5caee7SBarry Smith 3676be5caee7SBarry Smith Logically Collective on SNES 3677be5caee7SBarry Smith 3678be5caee7SBarry Smith Input Parameters: 3679be5caee7SBarry Smith + snes - the SNES context 3680be5caee7SBarry Smith - force - PETSC_TRUE require at least one iteration 3681be5caee7SBarry Smith 3682be5caee7SBarry Smith Options Database Keys: 3683be5caee7SBarry Smith . -snes_force_iteration <force> - Sets forcing an iteration 3684be5caee7SBarry Smith 3685be5caee7SBarry Smith Notes: 3686be5caee7SBarry Smith This is used sometimes with TS to prevent TS from detecting a false steady state solution 3687be5caee7SBarry Smith 3688be5caee7SBarry Smith Level: intermediate 3689be5caee7SBarry Smith 3690db781477SPatrick Sanan .seealso: `SNESSetTrustRegionTolerance()`, `SNESSetDivergenceTolerance()` 3691be5caee7SBarry Smith @*/ 3692be5caee7SBarry Smith PetscErrorCode SNESSetForceIteration(SNES snes,PetscBool force) 3693be5caee7SBarry Smith { 3694be5caee7SBarry Smith PetscFunctionBegin; 3695be5caee7SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 3696be5caee7SBarry Smith snes->forceiteration = force; 3697be5caee7SBarry Smith PetscFunctionReturn(0); 3698be5caee7SBarry Smith } 3699be5caee7SBarry Smith 370085216dc7SFande Kong /*@ 370185216dc7SFande Kong SNESGetForceIteration - Whether or not to force SNESSolve() take at least one iteration regardless of the initial residual norm 370285216dc7SFande Kong 370385216dc7SFande Kong Logically Collective on SNES 370485216dc7SFande Kong 370585216dc7SFande Kong Input Parameters: 370685216dc7SFande Kong . snes - the SNES context 370785216dc7SFande Kong 370885216dc7SFande Kong Output Parameter: 370985216dc7SFande Kong . force - PETSC_TRUE requires at least one iteration. 371085216dc7SFande Kong 371106dd6b0eSSatish Balay Level: intermediate 371206dd6b0eSSatish Balay 3713db781477SPatrick Sanan .seealso: `SNESSetForceIteration()`, `SNESSetTrustRegionTolerance()`, `SNESSetDivergenceTolerance()` 371485216dc7SFande Kong @*/ 371585216dc7SFande Kong PetscErrorCode SNESGetForceIteration(SNES snes,PetscBool *force) 371685216dc7SFande Kong { 371785216dc7SFande Kong PetscFunctionBegin; 371885216dc7SFande Kong PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 371985216dc7SFande Kong *force = snes->forceiteration; 372085216dc7SFande Kong PetscFunctionReturn(0); 372185216dc7SFande Kong } 3722be5caee7SBarry Smith 3723be5caee7SBarry Smith /*@ 3724d7a720efSLois Curfman McInnes SNESSetTolerances - Sets various parameters used in convergence tests. 37259b94acceSBarry Smith 37263f9fe445SBarry Smith Logically Collective on SNES 3727c7afd0dbSLois Curfman McInnes 37289b94acceSBarry Smith Input Parameters: 3729c7afd0dbSLois Curfman McInnes + snes - the SNES context 373070441072SBarry Smith . abstol - absolute convergence tolerance 373133174efeSLois Curfman McInnes . rtol - relative convergence tolerance 37325358d0d4SBarry Smith . stol - convergence tolerance in terms of the norm of the change in the solution between steps, || delta x || < stol*|| x || 373333174efeSLois Curfman McInnes . maxit - maximum number of iterations 3734e71169deSBarry Smith - maxf - maximum number of function evaluations (-1 indicates no limit) 3735fee21e36SBarry Smith 373633174efeSLois Curfman McInnes Options Database Keys: 373770441072SBarry Smith + -snes_atol <abstol> - Sets abstol 3738c7afd0dbSLois Curfman McInnes . -snes_rtol <rtol> - Sets rtol 3739c7afd0dbSLois Curfman McInnes . -snes_stol <stol> - Sets stol 3740c7afd0dbSLois Curfman McInnes . -snes_max_it <maxit> - Sets maxit 3741c7afd0dbSLois Curfman McInnes - -snes_max_funcs <maxf> - Sets maxf 37429b94acceSBarry Smith 3743d7a720efSLois Curfman McInnes Notes: 37449b94acceSBarry Smith The default maximum number of iterations is 50. 37459b94acceSBarry Smith The default maximum number of function evaluations is 1000. 37469b94acceSBarry Smith 374736851e7fSLois Curfman McInnes Level: intermediate 374836851e7fSLois Curfman McInnes 3749db781477SPatrick Sanan .seealso: `SNESSetTrustRegionTolerance()`, `SNESSetDivergenceTolerance()`, `SNESSetForceIteration()` 37509b94acceSBarry Smith @*/ 37517087cfbeSBarry Smith PetscErrorCode SNESSetTolerances(SNES snes,PetscReal abstol,PetscReal rtol,PetscReal stol,PetscInt maxit,PetscInt maxf) 37529b94acceSBarry Smith { 37533a40ed3dSBarry Smith PetscFunctionBegin; 37540700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 3755c5eb9154SBarry Smith PetscValidLogicalCollectiveReal(snes,abstol,2); 3756c5eb9154SBarry Smith PetscValidLogicalCollectiveReal(snes,rtol,3); 3757c5eb9154SBarry Smith PetscValidLogicalCollectiveReal(snes,stol,4); 3758c5eb9154SBarry Smith PetscValidLogicalCollectiveInt(snes,maxit,5); 3759c5eb9154SBarry Smith PetscValidLogicalCollectiveInt(snes,maxf,6); 3760c5eb9154SBarry Smith 3761ab54825eSJed Brown if (abstol != PETSC_DEFAULT) { 37625f80ce2aSJacob Faibussowitsch PetscCheck(abstol >= 0.0,PetscObjectComm((PetscObject)snes),PETSC_ERR_ARG_OUTOFRANGE,"Absolute tolerance %g must be non-negative",(double)abstol); 3763ab54825eSJed Brown snes->abstol = abstol; 3764ab54825eSJed Brown } 3765ab54825eSJed Brown if (rtol != PETSC_DEFAULT) { 37665f80ce2aSJacob Faibussowitsch PetscCheck(rtol >= 0.0 && 1.0 > rtol,PetscObjectComm((PetscObject)snes),PETSC_ERR_ARG_OUTOFRANGE,"Relative tolerance %g must be non-negative and less than 1.0",(double)rtol); 3767ab54825eSJed Brown snes->rtol = rtol; 3768ab54825eSJed Brown } 3769ab54825eSJed Brown if (stol != PETSC_DEFAULT) { 37705f80ce2aSJacob Faibussowitsch PetscCheck(stol >= 0.0,PetscObjectComm((PetscObject)snes),PETSC_ERR_ARG_OUTOFRANGE,"Step tolerance %g must be non-negative",(double)stol); 3771c60f73f4SPeter Brune snes->stol = stol; 3772ab54825eSJed Brown } 3773ab54825eSJed Brown if (maxit != PETSC_DEFAULT) { 377463a3b9bcSJacob Faibussowitsch PetscCheck(maxit >= 0,PetscObjectComm((PetscObject)snes),PETSC_ERR_ARG_OUTOFRANGE,"Maximum number of iterations %" PetscInt_FMT " must be non-negative",maxit); 3775ab54825eSJed Brown snes->max_its = maxit; 3776ab54825eSJed Brown } 3777ab54825eSJed Brown if (maxf != PETSC_DEFAULT) { 377863a3b9bcSJacob Faibussowitsch PetscCheck(maxf >= -1,PetscObjectComm((PetscObject)snes),PETSC_ERR_ARG_OUTOFRANGE,"Maximum number of function evaluations %" PetscInt_FMT " must be -1 or nonnegative",maxf); 3779ab54825eSJed Brown snes->max_funcs = maxf; 3780ab54825eSJed Brown } 378188976e71SPeter Brune snes->tolerancesset = PETSC_TRUE; 37823a40ed3dSBarry Smith PetscFunctionReturn(0); 37839b94acceSBarry Smith } 37849b94acceSBarry Smith 3785e4d06f11SPatrick Farrell /*@ 3786e4d06f11SPatrick Farrell SNESSetDivergenceTolerance - Sets the divergence tolerance used for the SNES divergence test. 3787e4d06f11SPatrick Farrell 3788e4d06f11SPatrick Farrell Logically Collective on SNES 3789e4d06f11SPatrick Farrell 3790e4d06f11SPatrick Farrell Input Parameters: 3791e4d06f11SPatrick Farrell + snes - the SNES context 3792e4d06f11SPatrick Farrell - divtol - the divergence tolerance. Use -1 to deactivate the test. 3793e4d06f11SPatrick Farrell 3794e4d06f11SPatrick Farrell Options Database Keys: 3795a2b725a8SWilliam Gropp . -snes_divergence_tolerance <divtol> - Sets divtol 3796e4d06f11SPatrick Farrell 3797e4d06f11SPatrick Farrell Notes: 3798e4d06f11SPatrick Farrell The default divergence tolerance is 1e4. 3799e4d06f11SPatrick Farrell 3800e4d06f11SPatrick Farrell Level: intermediate 3801e4d06f11SPatrick Farrell 3802db781477SPatrick Sanan .seealso: `SNESSetTolerances()`, `SNESGetDivergenceTolerance` 3803e4d06f11SPatrick Farrell @*/ 3804e4d06f11SPatrick Farrell PetscErrorCode SNESSetDivergenceTolerance(SNES snes,PetscReal divtol) 3805e4d06f11SPatrick Farrell { 3806e4d06f11SPatrick Farrell PetscFunctionBegin; 3807e4d06f11SPatrick Farrell PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 3808e4d06f11SPatrick Farrell PetscValidLogicalCollectiveReal(snes,divtol,2); 3809e4d06f11SPatrick Farrell 3810e4d06f11SPatrick Farrell if (divtol != PETSC_DEFAULT) { 3811e4d06f11SPatrick Farrell snes->divtol = divtol; 3812e4d06f11SPatrick Farrell } 3813e4d06f11SPatrick Farrell else { 3814e4d06f11SPatrick Farrell snes->divtol = 1.0e4; 3815e4d06f11SPatrick Farrell } 3816e4d06f11SPatrick Farrell PetscFunctionReturn(0); 3817e4d06f11SPatrick Farrell } 3818e4d06f11SPatrick Farrell 38199b94acceSBarry Smith /*@ 382033174efeSLois Curfman McInnes SNESGetTolerances - Gets various parameters used in convergence tests. 382133174efeSLois Curfman McInnes 3822c7afd0dbSLois Curfman McInnes Not Collective 3823c7afd0dbSLois Curfman McInnes 382433174efeSLois Curfman McInnes Input Parameters: 3825c7afd0dbSLois Curfman McInnes + snes - the SNES context 382685385478SLisandro Dalcin . atol - absolute convergence tolerance 382733174efeSLois Curfman McInnes . rtol - relative convergence tolerance 382833174efeSLois Curfman McInnes . stol - convergence tolerance in terms of the norm 382933174efeSLois Curfman McInnes of the change in the solution between steps 383033174efeSLois Curfman McInnes . maxit - maximum number of iterations 3831c7afd0dbSLois Curfman McInnes - maxf - maximum number of function evaluations 3832fee21e36SBarry Smith 383333174efeSLois Curfman McInnes Notes: 38340298fd71SBarry Smith The user can specify NULL for any parameter that is not needed. 383533174efeSLois Curfman McInnes 383636851e7fSLois Curfman McInnes Level: intermediate 383736851e7fSLois Curfman McInnes 3838db781477SPatrick Sanan .seealso: `SNESSetTolerances()` 383933174efeSLois Curfman McInnes @*/ 38407087cfbeSBarry Smith PetscErrorCode SNESGetTolerances(SNES snes,PetscReal *atol,PetscReal *rtol,PetscReal *stol,PetscInt *maxit,PetscInt *maxf) 384133174efeSLois Curfman McInnes { 38423a40ed3dSBarry Smith PetscFunctionBegin; 38430700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 384485385478SLisandro Dalcin if (atol) *atol = snes->abstol; 384533174efeSLois Curfman McInnes if (rtol) *rtol = snes->rtol; 3846c60f73f4SPeter Brune if (stol) *stol = snes->stol; 384733174efeSLois Curfman McInnes if (maxit) *maxit = snes->max_its; 384833174efeSLois Curfman McInnes if (maxf) *maxf = snes->max_funcs; 38493a40ed3dSBarry Smith PetscFunctionReturn(0); 385033174efeSLois Curfman McInnes } 385133174efeSLois Curfman McInnes 3852e4d06f11SPatrick Farrell /*@ 3853e4d06f11SPatrick Farrell SNESGetDivergenceTolerance - Gets divergence tolerance used in divergence test. 3854e4d06f11SPatrick Farrell 3855e4d06f11SPatrick Farrell Not Collective 3856e4d06f11SPatrick Farrell 3857e4d06f11SPatrick Farrell Input Parameters: 3858e4d06f11SPatrick Farrell + snes - the SNES context 3859e4d06f11SPatrick Farrell - divtol - divergence tolerance 3860e4d06f11SPatrick Farrell 3861e4d06f11SPatrick Farrell Level: intermediate 3862e4d06f11SPatrick Farrell 3863db781477SPatrick Sanan .seealso: `SNESSetDivergenceTolerance()` 3864e4d06f11SPatrick Farrell @*/ 3865e4d06f11SPatrick Farrell PetscErrorCode SNESGetDivergenceTolerance(SNES snes,PetscReal *divtol) 3866e4d06f11SPatrick Farrell { 3867e4d06f11SPatrick Farrell PetscFunctionBegin; 3868e4d06f11SPatrick Farrell PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 3869e4d06f11SPatrick Farrell if (divtol) *divtol = snes->divtol; 3870e4d06f11SPatrick Farrell PetscFunctionReturn(0); 3871e4d06f11SPatrick Farrell } 3872e4d06f11SPatrick Farrell 387333174efeSLois Curfman McInnes /*@ 38749b94acceSBarry Smith SNESSetTrustRegionTolerance - Sets the trust region parameter tolerance. 38759b94acceSBarry Smith 38763f9fe445SBarry Smith Logically Collective on SNES 3877fee21e36SBarry Smith 3878c7afd0dbSLois Curfman McInnes Input Parameters: 3879c7afd0dbSLois Curfman McInnes + snes - the SNES context 3880c7afd0dbSLois Curfman McInnes - tol - tolerance 3881c7afd0dbSLois Curfman McInnes 38829b94acceSBarry Smith Options Database Key: 3883c7afd0dbSLois Curfman McInnes . -snes_trtol <tol> - Sets tol 38849b94acceSBarry Smith 388536851e7fSLois Curfman McInnes Level: intermediate 388636851e7fSLois Curfman McInnes 3887db781477SPatrick Sanan .seealso: `SNESSetTolerances()` 38889b94acceSBarry Smith @*/ 38897087cfbeSBarry Smith PetscErrorCode SNESSetTrustRegionTolerance(SNES snes,PetscReal tol) 38909b94acceSBarry Smith { 38913a40ed3dSBarry Smith PetscFunctionBegin; 38920700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 3893c5eb9154SBarry Smith PetscValidLogicalCollectiveReal(snes,tol,2); 38949b94acceSBarry Smith snes->deltatol = tol; 38953a40ed3dSBarry Smith PetscFunctionReturn(0); 38969b94acceSBarry Smith } 38979b94acceSBarry Smith 38986ba87a44SLisandro Dalcin PETSC_INTERN PetscErrorCode SNESMonitorRange_Private(SNES,PetscInt,PetscReal*); 38996ba87a44SLisandro Dalcin 39007087cfbeSBarry Smith PetscErrorCode SNESMonitorLGRange(SNES snes,PetscInt n,PetscReal rnorm,void *monctx) 3901b271bb04SBarry Smith { 3902b271bb04SBarry Smith PetscDrawLG lg; 3903b271bb04SBarry Smith PetscReal x,y,per; 3904b271bb04SBarry Smith PetscViewer v = (PetscViewer)monctx; 3905b271bb04SBarry Smith static PetscReal prev; /* should be in the context */ 3906b271bb04SBarry Smith PetscDraw draw; 3907b271bb04SBarry Smith 3908459f5d12SBarry Smith PetscFunctionBegin; 39094d4332d5SBarry Smith PetscValidHeaderSpecific(v,PETSC_VIEWER_CLASSID,4); 39109566063dSJacob Faibussowitsch PetscCall(PetscViewerDrawGetDrawLG(v,0,&lg)); 39119566063dSJacob Faibussowitsch if (!n) PetscCall(PetscDrawLGReset(lg)); 39129566063dSJacob Faibussowitsch PetscCall(PetscDrawLGGetDraw(lg,&draw)); 39139566063dSJacob Faibussowitsch PetscCall(PetscDrawSetTitle(draw,"Residual norm")); 3914b271bb04SBarry Smith x = (PetscReal)n; 391577b4d14cSPeter Brune if (rnorm > 0.0) y = PetscLog10Real(rnorm); 391694c9c6d3SKarl Rupp else y = -15.0; 39179566063dSJacob Faibussowitsch PetscCall(PetscDrawLGAddPoint(lg,&x,&y)); 39186934998bSLisandro Dalcin if (n < 20 || !(n % 5) || snes->reason) { 39199566063dSJacob Faibussowitsch PetscCall(PetscDrawLGDraw(lg)); 39209566063dSJacob Faibussowitsch PetscCall(PetscDrawLGSave(lg)); 3921b271bb04SBarry Smith } 3922b271bb04SBarry Smith 39239566063dSJacob Faibussowitsch PetscCall(PetscViewerDrawGetDrawLG(v,1,&lg)); 39249566063dSJacob Faibussowitsch if (!n) PetscCall(PetscDrawLGReset(lg)); 39259566063dSJacob Faibussowitsch PetscCall(PetscDrawLGGetDraw(lg,&draw)); 39269566063dSJacob Faibussowitsch PetscCall(PetscDrawSetTitle(draw,"% elemts > .2*max elemt")); 39279566063dSJacob Faibussowitsch PetscCall(SNESMonitorRange_Private(snes,n,&per)); 3928b271bb04SBarry Smith x = (PetscReal)n; 3929b271bb04SBarry Smith y = 100.0*per; 39309566063dSJacob Faibussowitsch PetscCall(PetscDrawLGAddPoint(lg,&x,&y)); 39316934998bSLisandro Dalcin if (n < 20 || !(n % 5) || snes->reason) { 39329566063dSJacob Faibussowitsch PetscCall(PetscDrawLGDraw(lg)); 39339566063dSJacob Faibussowitsch PetscCall(PetscDrawLGSave(lg)); 3934b271bb04SBarry Smith } 3935b271bb04SBarry Smith 39369566063dSJacob Faibussowitsch PetscCall(PetscViewerDrawGetDrawLG(v,2,&lg)); 39379566063dSJacob Faibussowitsch if (!n) {prev = rnorm;PetscCall(PetscDrawLGReset(lg));} 39389566063dSJacob Faibussowitsch PetscCall(PetscDrawLGGetDraw(lg,&draw)); 39399566063dSJacob Faibussowitsch PetscCall(PetscDrawSetTitle(draw,"(norm -oldnorm)/oldnorm")); 3940b271bb04SBarry Smith x = (PetscReal)n; 3941b271bb04SBarry Smith y = (prev - rnorm)/prev; 39429566063dSJacob Faibussowitsch PetscCall(PetscDrawLGAddPoint(lg,&x,&y)); 39436934998bSLisandro Dalcin if (n < 20 || !(n % 5) || snes->reason) { 39449566063dSJacob Faibussowitsch PetscCall(PetscDrawLGDraw(lg)); 39459566063dSJacob Faibussowitsch PetscCall(PetscDrawLGSave(lg)); 3946b271bb04SBarry Smith } 3947b271bb04SBarry Smith 39489566063dSJacob Faibussowitsch PetscCall(PetscViewerDrawGetDrawLG(v,3,&lg)); 39499566063dSJacob Faibussowitsch if (!n) PetscCall(PetscDrawLGReset(lg)); 39509566063dSJacob Faibussowitsch PetscCall(PetscDrawLGGetDraw(lg,&draw)); 39519566063dSJacob Faibussowitsch PetscCall(PetscDrawSetTitle(draw,"(norm -oldnorm)/oldnorm*(% > .2 max)")); 3952b271bb04SBarry Smith x = (PetscReal)n; 3953b271bb04SBarry Smith y = (prev - rnorm)/(prev*per); 3954b271bb04SBarry Smith if (n > 2) { /*skip initial crazy value */ 39559566063dSJacob Faibussowitsch PetscCall(PetscDrawLGAddPoint(lg,&x,&y)); 3956b271bb04SBarry Smith } 39576934998bSLisandro Dalcin if (n < 20 || !(n % 5) || snes->reason) { 39589566063dSJacob Faibussowitsch PetscCall(PetscDrawLGDraw(lg)); 39599566063dSJacob Faibussowitsch PetscCall(PetscDrawLGSave(lg)); 3960b271bb04SBarry Smith } 3961b271bb04SBarry Smith prev = rnorm; 3962b271bb04SBarry Smith PetscFunctionReturn(0); 3963b271bb04SBarry Smith } 3964b271bb04SBarry Smith 3965228d79bcSJed Brown /*@ 3966228d79bcSJed Brown SNESMonitor - runs the user provided monitor routines, if they exist 3967228d79bcSJed Brown 3968228d79bcSJed Brown Collective on SNES 3969228d79bcSJed Brown 3970228d79bcSJed Brown Input Parameters: 3971228d79bcSJed Brown + snes - nonlinear solver context obtained from SNESCreate() 3972228d79bcSJed Brown . iter - iteration number 3973228d79bcSJed Brown - rnorm - relative norm of the residual 3974228d79bcSJed Brown 3975228d79bcSJed Brown Notes: 3976228d79bcSJed Brown This routine is called by the SNES implementations. 3977228d79bcSJed Brown It does not typically need to be called by the user. 3978228d79bcSJed Brown 3979228d79bcSJed Brown Level: developer 3980228d79bcSJed Brown 3981db781477SPatrick Sanan .seealso: `SNESMonitorSet()` 3982228d79bcSJed Brown @*/ 39837a03ce2fSLisandro Dalcin PetscErrorCode SNESMonitor(SNES snes,PetscInt iter,PetscReal rnorm) 39847a03ce2fSLisandro Dalcin { 39857a03ce2fSLisandro Dalcin PetscInt i,n = snes->numbermonitors; 39867a03ce2fSLisandro Dalcin 39877a03ce2fSLisandro Dalcin PetscFunctionBegin; 39889566063dSJacob Faibussowitsch PetscCall(VecLockReadPush(snes->vec_sol)); 39897a03ce2fSLisandro Dalcin for (i=0; i<n; i++) { 39909566063dSJacob Faibussowitsch PetscCall((*snes->monitor[i])(snes,iter,rnorm,snes->monitorcontext[i])); 39917a03ce2fSLisandro Dalcin } 39929566063dSJacob Faibussowitsch PetscCall(VecLockReadPop(snes->vec_sol)); 39937a03ce2fSLisandro Dalcin PetscFunctionReturn(0); 39947a03ce2fSLisandro Dalcin } 39957a03ce2fSLisandro Dalcin 39969b94acceSBarry Smith /* ------------ Routines to set performance monitoring options ----------- */ 39979b94acceSBarry Smith 3998bf388a1fSBarry Smith /*MC 3999bf388a1fSBarry Smith SNESMonitorFunction - functional form passed to SNESMonitorSet() to monitor convergence of nonlinear solver 4000bf388a1fSBarry Smith 4001bf388a1fSBarry Smith Synopsis: 4002aaa7dc30SBarry Smith #include <petscsnes.h> 4003bf388a1fSBarry Smith $ PetscErrorCode SNESMonitorFunction(SNES snes,PetscInt its, PetscReal norm,void *mctx) 4004bf388a1fSBarry Smith 40051843f636SBarry Smith Collective on snes 40061843f636SBarry Smith 40071843f636SBarry Smith Input Parameters: 4008bf388a1fSBarry Smith + snes - the SNES context 4009bf388a1fSBarry Smith . its - iteration number 4010bf388a1fSBarry Smith . norm - 2-norm function value (may be estimated) 4011bf388a1fSBarry Smith - mctx - [optional] monitoring context 4012bf388a1fSBarry Smith 4013878cb397SSatish Balay Level: advanced 4014878cb397SSatish Balay 4015db781477SPatrick Sanan .seealso: `SNESMonitorSet()`, `SNESMonitorGet()` 4016bf388a1fSBarry Smith M*/ 4017bf388a1fSBarry Smith 40189b94acceSBarry Smith /*@C 4019a6570f20SBarry Smith SNESMonitorSet - Sets an ADDITIONAL function that is to be used at every 40209b94acceSBarry Smith iteration of the nonlinear solver to display the iteration's 40219b94acceSBarry Smith progress. 40229b94acceSBarry Smith 40233f9fe445SBarry Smith Logically Collective on SNES 4024fee21e36SBarry Smith 4025c7afd0dbSLois Curfman McInnes Input Parameters: 4026c7afd0dbSLois Curfman McInnes + snes - the SNES context 40276e4dcb14SBarry Smith . f - the monitor function, see SNESMonitorFunction for the calling sequence 4028b8a78c4aSBarry Smith . mctx - [optional] user-defined context for private data for the 40290298fd71SBarry Smith monitor routine (use NULL if no context is desired) 4030b3006f0bSLois Curfman McInnes - monitordestroy - [optional] routine that frees monitor context 40310298fd71SBarry Smith (may be NULL) 40329b94acceSBarry Smith 40339665c990SLois Curfman McInnes Options Database Keys: 4034a6570f20SBarry Smith + -snes_monitor - sets SNESMonitorDefault() 4035798534f6SMatthew G. Knepley . -snes_monitor draw::draw_lg - sets line graph monitor, 4036cca6129bSJed Brown - -snes_monitor_cancel - cancels all monitors that have 4037c7afd0dbSLois Curfman McInnes been hardwired into a code by 4038a6570f20SBarry Smith calls to SNESMonitorSet(), but 4039c7afd0dbSLois Curfman McInnes does not cancel those set via 4040c7afd0dbSLois Curfman McInnes the options database. 40419665c990SLois Curfman McInnes 4042639f9d9dSBarry Smith Notes: 40436bc08f3fSLois Curfman McInnes Several different monitoring routines may be set by calling 4044a6570f20SBarry Smith SNESMonitorSet() multiple times; all will be called in the 40456bc08f3fSLois Curfman McInnes order in which they were set. 4046639f9d9dSBarry Smith 404795452b02SPatrick Sanan Fortran Notes: 404895452b02SPatrick Sanan Only a single monitor function can be set for each SNES object 4049025f1a04SBarry Smith 405036851e7fSLois Curfman McInnes Level: intermediate 405136851e7fSLois Curfman McInnes 4052db781477SPatrick Sanan .seealso: `SNESMonitorDefault()`, `SNESMonitorCancel()`, `SNESMonitorFunction` 40539b94acceSBarry Smith @*/ 40546e4dcb14SBarry Smith PetscErrorCode SNESMonitorSet(SNES snes,PetscErrorCode (*f)(SNES,PetscInt,PetscReal,void*),void *mctx,PetscErrorCode (*monitordestroy)(void**)) 40559b94acceSBarry Smith { 4056b90d0a6eSBarry Smith PetscInt i; 405778064530SBarry Smith PetscBool identical; 4058b90d0a6eSBarry Smith 40593a40ed3dSBarry Smith PetscFunctionBegin; 40600700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 4061b90d0a6eSBarry Smith for (i=0; i<snes->numbermonitors;i++) { 40629566063dSJacob Faibussowitsch PetscCall(PetscMonitorCompare((PetscErrorCode (*)(void))f,mctx,monitordestroy,(PetscErrorCode (*)(void))snes->monitor[i],snes->monitorcontext[i],snes->monitordestroy[i],&identical)); 406378064530SBarry Smith if (identical) PetscFunctionReturn(0); 4064649052a6SBarry Smith } 40655f80ce2aSJacob Faibussowitsch PetscCheck(snes->numbermonitors < MAXSNESMONITORS,PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Too many monitors set"); 40666e4dcb14SBarry Smith snes->monitor[snes->numbermonitors] = f; 4067b8a78c4aSBarry Smith snes->monitordestroy[snes->numbermonitors] = monitordestroy; 4068639f9d9dSBarry Smith snes->monitorcontext[snes->numbermonitors++] = (void*)mctx; 40693a40ed3dSBarry Smith PetscFunctionReturn(0); 40709b94acceSBarry Smith } 40719b94acceSBarry Smith 4072a278d85bSSatish Balay /*@ 4073a6570f20SBarry Smith SNESMonitorCancel - Clears all the monitor functions for a SNES object. 40745cd90555SBarry Smith 40753f9fe445SBarry Smith Logically Collective on SNES 4076c7afd0dbSLois Curfman McInnes 40775cd90555SBarry Smith Input Parameters: 40785cd90555SBarry Smith . snes - the SNES context 40795cd90555SBarry Smith 40801a480d89SAdministrator Options Database Key: 4081a6570f20SBarry Smith . -snes_monitor_cancel - cancels all monitors that have been hardwired 4082a6570f20SBarry Smith into a code by calls to SNESMonitorSet(), but does not cancel those 4083c7afd0dbSLois Curfman McInnes set via the options database 40845cd90555SBarry Smith 40855cd90555SBarry Smith Notes: 40865cd90555SBarry Smith There is no way to clear one specific monitor from a SNES object. 40875cd90555SBarry Smith 408836851e7fSLois Curfman McInnes Level: intermediate 408936851e7fSLois Curfman McInnes 4090db781477SPatrick Sanan .seealso: `SNESMonitorDefault()`, `SNESMonitorSet()` 40915cd90555SBarry Smith @*/ 40927087cfbeSBarry Smith PetscErrorCode SNESMonitorCancel(SNES snes) 40935cd90555SBarry Smith { 4094d952e501SBarry Smith PetscInt i; 4095d952e501SBarry Smith 40965cd90555SBarry Smith PetscFunctionBegin; 40970700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 4098d952e501SBarry Smith for (i=0; i<snes->numbermonitors; i++) { 4099d952e501SBarry Smith if (snes->monitordestroy[i]) { 41009566063dSJacob Faibussowitsch PetscCall((*snes->monitordestroy[i])(&snes->monitorcontext[i])); 4101d952e501SBarry Smith } 4102d952e501SBarry Smith } 41035cd90555SBarry Smith snes->numbermonitors = 0; 41045cd90555SBarry Smith PetscFunctionReturn(0); 41055cd90555SBarry Smith } 41065cd90555SBarry Smith 4107bf388a1fSBarry Smith /*MC 4108bf388a1fSBarry Smith SNESConvergenceTestFunction - functional form used for testing of convergence of nonlinear solver 4109bf388a1fSBarry Smith 4110bf388a1fSBarry Smith Synopsis: 4111aaa7dc30SBarry Smith #include <petscsnes.h> 4112bf388a1fSBarry Smith $ PetscErrorCode SNESConvergenceTest(SNES snes,PetscInt it,PetscReal xnorm,PetscReal gnorm,PetscReal f,SNESConvergedReason *reason,void *cctx) 4113bf388a1fSBarry Smith 41141843f636SBarry Smith Collective on snes 41151843f636SBarry Smith 41161843f636SBarry Smith Input Parameters: 4117bf388a1fSBarry Smith + snes - the SNES context 4118bf388a1fSBarry Smith . it - current iteration (0 is the first and is before any Newton step) 4119bf388a1fSBarry Smith . xnorm - 2-norm of current iterate 4120bf388a1fSBarry Smith . gnorm - 2-norm of current step 41211843f636SBarry Smith . f - 2-norm of function 41221843f636SBarry Smith - cctx - [optional] convergence context 41231843f636SBarry Smith 41241843f636SBarry Smith Output Parameter: 41251843f636SBarry Smith . reason - reason for convergence/divergence, only needs to be set when convergence or divergence is detected 4126bf388a1fSBarry Smith 4127878cb397SSatish Balay Level: intermediate 4128bf388a1fSBarry Smith 4129db781477SPatrick Sanan .seealso: `SNESSetConvergenceTest()`, `SNESGetConvergenceTest()` 4130bf388a1fSBarry Smith M*/ 4131bf388a1fSBarry Smith 41329b94acceSBarry Smith /*@C 41339b94acceSBarry Smith SNESSetConvergenceTest - Sets the function that is to be used 41349b94acceSBarry Smith to test for convergence of the nonlinear iterative solution. 41359b94acceSBarry Smith 41363f9fe445SBarry Smith Logically Collective on SNES 4137fee21e36SBarry Smith 4138c7afd0dbSLois Curfman McInnes Input Parameters: 4139c7afd0dbSLois Curfman McInnes + snes - the SNES context 4140bf388a1fSBarry Smith . SNESConvergenceTestFunction - routine to test for convergence 41410298fd71SBarry Smith . cctx - [optional] context for private data for the convergence routine (may be NULL) 4142cf90aa19SBarry Smith - destroy - [optional] destructor for the context (may be NULL; PETSC_NULL_FUNCTION in Fortran) 41439b94acceSBarry Smith 414436851e7fSLois Curfman McInnes Level: advanced 414536851e7fSLois Curfman McInnes 4146db781477SPatrick Sanan .seealso: `SNESConvergedDefault()`, `SNESConvergedSkip()`, `SNESConvergenceTestFunction` 41479b94acceSBarry Smith @*/ 4148bf388a1fSBarry Smith PetscErrorCode SNESSetConvergenceTest(SNES snes,PetscErrorCode (*SNESConvergenceTestFunction)(SNES,PetscInt,PetscReal,PetscReal,PetscReal,SNESConvergedReason*,void*),void *cctx,PetscErrorCode (*destroy)(void*)) 41499b94acceSBarry Smith { 41503a40ed3dSBarry Smith PetscFunctionBegin; 41510700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 4152e2a6519dSDmitry Karpeev if (!SNESConvergenceTestFunction) SNESConvergenceTestFunction = SNESConvergedSkip; 41531baa6e33SBarry Smith if (snes->ops->convergeddestroy) PetscCall((*snes->ops->convergeddestroy)(snes->cnvP)); 4154bf388a1fSBarry Smith snes->ops->converged = SNESConvergenceTestFunction; 41557f7931b9SBarry Smith snes->ops->convergeddestroy = destroy; 415685385478SLisandro Dalcin snes->cnvP = cctx; 41573a40ed3dSBarry Smith PetscFunctionReturn(0); 41589b94acceSBarry Smith } 41599b94acceSBarry Smith 416052baeb72SSatish Balay /*@ 4161184914b5SBarry Smith SNESGetConvergedReason - Gets the reason the SNES iteration was stopped. 4162184914b5SBarry Smith 4163184914b5SBarry Smith Not Collective 4164184914b5SBarry Smith 4165184914b5SBarry Smith Input Parameter: 4166184914b5SBarry Smith . snes - the SNES context 4167184914b5SBarry Smith 4168184914b5SBarry Smith Output Parameter: 41694d0a8057SBarry Smith . reason - negative value indicates diverged, positive value converged, see SNESConvergedReason or the 4170184914b5SBarry Smith manual pages for the individual convergence tests for complete lists 4171184914b5SBarry Smith 41726a4d7782SBarry Smith Options Database: 41736a4d7782SBarry Smith . -snes_converged_reason - prints the reason to standard out 41746a4d7782SBarry Smith 4175184914b5SBarry Smith Level: intermediate 4176184914b5SBarry Smith 417795452b02SPatrick Sanan Notes: 417895452b02SPatrick Sanan Should only be called after the call the SNESSolve() is complete, if it is called earlier it returns the value SNES__CONVERGED_ITERATING. 4179184914b5SBarry Smith 4180db781477SPatrick Sanan .seealso: `SNESSetConvergenceTest()`, `SNESSetConvergedReason()`, `SNESConvergedReason` 4181184914b5SBarry Smith @*/ 41827087cfbeSBarry Smith PetscErrorCode SNESGetConvergedReason(SNES snes,SNESConvergedReason *reason) 4183184914b5SBarry Smith { 4184184914b5SBarry Smith PetscFunctionBegin; 41850700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 41864482741eSBarry Smith PetscValidPointer(reason,2); 4187184914b5SBarry Smith *reason = snes->reason; 4188184914b5SBarry Smith PetscFunctionReturn(0); 4189184914b5SBarry Smith } 4190184914b5SBarry Smith 4191c4421ceaSFande Kong /*@C 4192c4421ceaSFande Kong SNESGetConvergedReasonString - Return a human readable string for snes converged reason 4193c4421ceaSFande Kong 4194c4421ceaSFande Kong Not Collective 4195c4421ceaSFande Kong 4196c4421ceaSFande Kong Input Parameter: 4197c4421ceaSFande Kong . snes - the SNES context 4198c4421ceaSFande Kong 4199c4421ceaSFande Kong Output Parameter: 4200c4421ceaSFande Kong . strreason - a human readable string that describes SNES converged reason 4201c4421ceaSFande Kong 420299c90e12SSatish Balay Level: beginner 4203c4421ceaSFande Kong 4204db781477SPatrick Sanan .seealso: `SNESGetConvergedReason()` 4205c4421ceaSFande Kong @*/ 4206c4421ceaSFande Kong PetscErrorCode SNESGetConvergedReasonString(SNES snes, const char** strreason) 4207c4421ceaSFande Kong { 4208c4421ceaSFande Kong PetscFunctionBegin; 4209c4421ceaSFande Kong PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 4210dadcf809SJacob Faibussowitsch PetscValidPointer(strreason,2); 4211c4421ceaSFande Kong *strreason = SNESConvergedReasons[snes->reason]; 4212c4421ceaSFande Kong PetscFunctionReturn(0); 4213c4421ceaSFande Kong } 4214c4421ceaSFande Kong 421533866048SMatthew G. Knepley /*@ 421633866048SMatthew G. Knepley SNESSetConvergedReason - Sets the reason the SNES iteration was stopped. 421733866048SMatthew G. Knepley 421833866048SMatthew G. Knepley Not Collective 421933866048SMatthew G. Knepley 422033866048SMatthew G. Knepley Input Parameters: 422133866048SMatthew G. Knepley + snes - the SNES context 422233866048SMatthew G. Knepley - reason - negative value indicates diverged, positive value converged, see SNESConvergedReason or the 422333866048SMatthew G. Knepley manual pages for the individual convergence tests for complete lists 422433866048SMatthew G. Knepley 422533866048SMatthew G. Knepley Level: intermediate 422633866048SMatthew G. Knepley 4227db781477SPatrick Sanan .seealso: `SNESGetConvergedReason()`, `SNESSetConvergenceTest()`, `SNESConvergedReason` 422833866048SMatthew G. Knepley @*/ 422933866048SMatthew G. Knepley PetscErrorCode SNESSetConvergedReason(SNES snes,SNESConvergedReason reason) 423033866048SMatthew G. Knepley { 423133866048SMatthew G. Knepley PetscFunctionBegin; 423233866048SMatthew G. Knepley PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 423333866048SMatthew G. Knepley snes->reason = reason; 423433866048SMatthew G. Knepley PetscFunctionReturn(0); 423533866048SMatthew G. Knepley } 423633866048SMatthew G. Knepley 4237c9005455SLois Curfman McInnes /*@ 4238c9005455SLois Curfman McInnes SNESSetConvergenceHistory - Sets the array used to hold the convergence history. 4239c9005455SLois Curfman McInnes 42403f9fe445SBarry Smith Logically Collective on SNES 4241fee21e36SBarry Smith 4242c7afd0dbSLois Curfman McInnes Input Parameters: 4243c7afd0dbSLois Curfman McInnes + snes - iterative context obtained from SNESCreate() 42448c7482ecSBarry Smith . a - array to hold history, this array will contain the function norms computed at each step 4245cd5578b5SBarry Smith . its - integer array holds the number of linear iterations for each solve. 4246758f92a0SBarry Smith . na - size of a and its 424764731454SLois Curfman McInnes - reset - PETSC_TRUE indicates each new nonlinear solve resets the history counter to zero, 4248758f92a0SBarry Smith else it continues storing new values for new nonlinear solves after the old ones 4249c7afd0dbSLois Curfman McInnes 4250308dcc3eSBarry Smith Notes: 42510298fd71SBarry Smith If 'a' and 'its' are NULL then space is allocated for the history. If 'na' PETSC_DECIDE or PETSC_DEFAULT then a 4252308dcc3eSBarry Smith default array of length 10000 is allocated. 4253308dcc3eSBarry Smith 4254c9005455SLois Curfman McInnes This routine is useful, e.g., when running a code for purposes 4255c9005455SLois Curfman McInnes of accurate performance monitoring, when no I/O should be done 4256c9005455SLois Curfman McInnes during the section of code that is being timed. 4257c9005455SLois Curfman McInnes 425836851e7fSLois Curfman McInnes Level: intermediate 425936851e7fSLois Curfman McInnes 4260db781477SPatrick Sanan .seealso: `SNESGetConvergenceHistory()` 4261758f92a0SBarry Smith 4262c9005455SLois Curfman McInnes @*/ 42637087cfbeSBarry Smith PetscErrorCode SNESSetConvergenceHistory(SNES snes,PetscReal a[],PetscInt its[],PetscInt na,PetscBool reset) 4264c9005455SLois Curfman McInnes { 42653a40ed3dSBarry Smith PetscFunctionBegin; 42660700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 4267064a246eSJacob Faibussowitsch if (a) PetscValidRealPointer(a,2); 4268a562a398SLisandro Dalcin if (its) PetscValidIntPointer(its,3); 42697a1ec6d4SBarry Smith if (!a) { 4270308dcc3eSBarry Smith if (na == PETSC_DECIDE || na == PETSC_DEFAULT) na = 1000; 42719566063dSJacob Faibussowitsch PetscCall(PetscCalloc2(na,&a,na,&its)); 4272071fcb05SBarry Smith snes->conv_hist_alloc = PETSC_TRUE; 4273308dcc3eSBarry Smith } 4274c9005455SLois Curfman McInnes snes->conv_hist = a; 4275758f92a0SBarry Smith snes->conv_hist_its = its; 4276115dd874SBarry Smith snes->conv_hist_max = (size_t)na; 4277a12bc48fSLisandro Dalcin snes->conv_hist_len = 0; 4278758f92a0SBarry Smith snes->conv_hist_reset = reset; 4279758f92a0SBarry Smith PetscFunctionReturn(0); 4280758f92a0SBarry Smith } 4281758f92a0SBarry Smith 4282308dcc3eSBarry Smith #if defined(PETSC_HAVE_MATLAB_ENGINE) 4283c6db04a5SJed Brown #include <engine.h> /* MATLAB include file */ 4284c6db04a5SJed Brown #include <mex.h> /* MATLAB include file */ 428599e0435eSBarry Smith 42868cc058d9SJed Brown PETSC_EXTERN mxArray *SNESGetConvergenceHistoryMatlab(SNES snes) 4287308dcc3eSBarry Smith { 4288308dcc3eSBarry Smith mxArray *mat; 4289308dcc3eSBarry Smith PetscInt i; 4290308dcc3eSBarry Smith PetscReal *ar; 4291308dcc3eSBarry Smith 4292308dcc3eSBarry Smith PetscFunctionBegin; 4293308dcc3eSBarry Smith mat = mxCreateDoubleMatrix(snes->conv_hist_len,1,mxREAL); 4294308dcc3eSBarry Smith ar = (PetscReal*) mxGetData(mat); 4295f5af7f23SKarl Rupp for (i=0; i<snes->conv_hist_len; i++) ar[i] = snes->conv_hist[i]; 4296308dcc3eSBarry Smith PetscFunctionReturn(mat); 4297308dcc3eSBarry Smith } 4298308dcc3eSBarry Smith #endif 4299308dcc3eSBarry Smith 43000c4c9dddSBarry Smith /*@C 4301758f92a0SBarry Smith SNESGetConvergenceHistory - Gets the array used to hold the convergence history. 4302758f92a0SBarry Smith 43033f9fe445SBarry Smith Not Collective 4304758f92a0SBarry Smith 4305758f92a0SBarry Smith Input Parameter: 4306758f92a0SBarry Smith . snes - iterative context obtained from SNESCreate() 4307758f92a0SBarry Smith 4308758f92a0SBarry Smith Output Parameters: 4309a2b725a8SWilliam Gropp + a - array to hold history 4310758f92a0SBarry Smith . its - integer array holds the number of linear iterations (or 4311758f92a0SBarry Smith negative if not converged) for each solve. 4312758f92a0SBarry Smith - na - size of a and its 4313758f92a0SBarry Smith 4314758f92a0SBarry Smith Notes: 4315758f92a0SBarry Smith The calling sequence for this routine in Fortran is 4316758f92a0SBarry Smith $ call SNESGetConvergenceHistory(SNES snes, integer na, integer ierr) 4317758f92a0SBarry Smith 4318758f92a0SBarry Smith This routine is useful, e.g., when running a code for purposes 4319758f92a0SBarry Smith of accurate performance monitoring, when no I/O should be done 4320758f92a0SBarry Smith during the section of code that is being timed. 4321758f92a0SBarry Smith 4322758f92a0SBarry Smith Level: intermediate 4323758f92a0SBarry Smith 4324db781477SPatrick Sanan .seealso: `SNESSetConvergenceHistory()` 4325758f92a0SBarry Smith 4326758f92a0SBarry Smith @*/ 43277087cfbeSBarry Smith PetscErrorCode SNESGetConvergenceHistory(SNES snes,PetscReal *a[],PetscInt *its[],PetscInt *na) 4328758f92a0SBarry Smith { 4329758f92a0SBarry Smith PetscFunctionBegin; 43300700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 4331758f92a0SBarry Smith if (a) *a = snes->conv_hist; 4332758f92a0SBarry Smith if (its) *its = snes->conv_hist_its; 4333115dd874SBarry Smith if (na) *na = (PetscInt) snes->conv_hist_len; 43343a40ed3dSBarry Smith PetscFunctionReturn(0); 4335c9005455SLois Curfman McInnes } 4336c9005455SLois Curfman McInnes 4337ac226902SBarry Smith /*@C 433876b2cf59SMatthew Knepley SNESSetUpdate - Sets the general-purpose update function called 4339eec8f646SJed Brown at the beginning of every iteration of the nonlinear solve. Specifically 43407e4bb74cSBarry Smith it is called just before the Jacobian is "evaluated". 434176b2cf59SMatthew Knepley 43423f9fe445SBarry Smith Logically Collective on SNES 434376b2cf59SMatthew Knepley 434476b2cf59SMatthew Knepley Input Parameters: 4345a2b725a8SWilliam Gropp + snes - The nonlinear solver context 4346a2b725a8SWilliam Gropp - func - The function 434776b2cf59SMatthew Knepley 434876b2cf59SMatthew Knepley Calling sequence of func: 4349a2b725a8SWilliam Gropp $ func (SNES snes, PetscInt step); 435076b2cf59SMatthew Knepley 435176b2cf59SMatthew Knepley . step - The current step of the iteration 435276b2cf59SMatthew Knepley 4353fe97e370SBarry Smith Level: advanced 4354fe97e370SBarry Smith 43556b7fb656SBarry Smith Note: 43566b7fb656SBarry Smith 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() 4357fe97e370SBarry Smith This is not used by most users. 435876b2cf59SMatthew Knepley 43596b7fb656SBarry Smith There are a varity of function hooks one many set that are called at different stages of the nonlinear solution process, see the functions listed below. 43606b7fb656SBarry Smith 4361db781477SPatrick Sanan .seealso `SNESSetJacobian()`, `SNESSolve()`, `SNESLineSearchSetPreCheck()`, `SNESLineSearchSetPostCheck()`, `SNESNewtonTRSetPreCheck()`, `SNESNewtonTRSetPostCheck()`, 4362db781477SPatrick Sanan `SNESMonitorSet()`, `SNESSetDivergenceTest()` 436376b2cf59SMatthew Knepley @*/ 43647087cfbeSBarry Smith PetscErrorCode SNESSetUpdate(SNES snes, PetscErrorCode (*func)(SNES, PetscInt)) 436576b2cf59SMatthew Knepley { 436676b2cf59SMatthew Knepley PetscFunctionBegin; 43670700a824SBarry Smith PetscValidHeaderSpecific(snes, SNES_CLASSID,1); 4368e7788613SBarry Smith snes->ops->update = func; 436976b2cf59SMatthew Knepley PetscFunctionReturn(0); 437076b2cf59SMatthew Knepley } 437176b2cf59SMatthew Knepley 43729b94acceSBarry Smith /* 43739b94acceSBarry Smith SNESScaleStep_Private - Scales a step so that its length is less than the 43749b94acceSBarry Smith positive parameter delta. 43759b94acceSBarry Smith 43769b94acceSBarry Smith Input Parameters: 4377c7afd0dbSLois Curfman McInnes + snes - the SNES context 43789b94acceSBarry Smith . y - approximate solution of linear system 43799b94acceSBarry Smith . fnorm - 2-norm of current function 4380c7afd0dbSLois Curfman McInnes - delta - trust region size 43819b94acceSBarry Smith 43829b94acceSBarry Smith Output Parameters: 4383c7afd0dbSLois Curfman McInnes + gpnorm - predicted function norm at the new point, assuming local 43849b94acceSBarry Smith linearization. The value is zero if the step lies within the trust 43859b94acceSBarry Smith region, and exceeds zero otherwise. 4386c7afd0dbSLois Curfman McInnes - ynorm - 2-norm of the step 43879b94acceSBarry Smith 43889b94acceSBarry Smith Note: 438904d7464bSBarry Smith For non-trust region methods such as SNESNEWTONLS, the parameter delta 43909b94acceSBarry Smith is set to be the maximum allowable step size. 43919b94acceSBarry Smith 43929b94acceSBarry Smith */ 4393dfbe8321SBarry Smith PetscErrorCode SNESScaleStep_Private(SNES snes,Vec y,PetscReal *fnorm,PetscReal *delta,PetscReal *gpnorm,PetscReal *ynorm) 43949b94acceSBarry Smith { 4395064f8208SBarry Smith PetscReal nrm; 4396ea709b57SSatish Balay PetscScalar cnorm; 43973a40ed3dSBarry Smith 43983a40ed3dSBarry Smith PetscFunctionBegin; 43990700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 44000700a824SBarry Smith PetscValidHeaderSpecific(y,VEC_CLASSID,2); 4401c9780b6fSBarry Smith PetscCheckSameComm(snes,1,y,2); 4402184914b5SBarry Smith 44039566063dSJacob Faibussowitsch PetscCall(VecNorm(y,NORM_2,&nrm)); 4404064f8208SBarry Smith if (nrm > *delta) { 4405064f8208SBarry Smith nrm = *delta/nrm; 4406064f8208SBarry Smith *gpnorm = (1.0 - nrm)*(*fnorm); 4407064f8208SBarry Smith cnorm = nrm; 44089566063dSJacob Faibussowitsch PetscCall(VecScale(y,cnorm)); 44099b94acceSBarry Smith *ynorm = *delta; 44109b94acceSBarry Smith } else { 44119b94acceSBarry Smith *gpnorm = 0.0; 4412064f8208SBarry Smith *ynorm = nrm; 44139b94acceSBarry Smith } 44143a40ed3dSBarry Smith PetscFunctionReturn(0); 44159b94acceSBarry Smith } 44169b94acceSBarry Smith 441791f3e32bSBarry Smith /*@C 441819a666eeSBarry Smith SNESConvergedReasonView - Displays the reason a SNES solve converged or diverged to a viewer 44192a359c20SBarry Smith 44202a359c20SBarry Smith Collective on SNES 44212a359c20SBarry Smith 44222a359c20SBarry Smith Parameter: 44232a359c20SBarry Smith + snes - iterative context obtained from SNESCreate() 44242a359c20SBarry Smith - viewer - the viewer to display the reason 44252a359c20SBarry Smith 44262a359c20SBarry Smith Options Database Keys: 4427ee300463SSatish Balay + -snes_converged_reason - print reason for converged or diverged, also prints number of iterations 4428ee300463SSatish Balay - -snes_converged_reason ::failed - only print reason and number of iterations when diverged 4429eafd5ff0SAlex Lindsay 443019a666eeSBarry Smith Notes: 443119a666eeSBarry Smith To change the format of the output call PetscViewerPushFormat(viewer,format) before this call. Use PETSC_VIEWER_DEFAULT for the default, 443219a666eeSBarry Smith use PETSC_VIEWER_FAILED to only display a reason if it fails. 44332a359c20SBarry Smith 44342a359c20SBarry Smith Level: beginner 44352a359c20SBarry Smith 4436db781477SPatrick Sanan .seealso: `SNESCreate()`, `SNESSetUp()`, `SNESDestroy()`, `SNESSetTolerances()`, `SNESConvergedDefault()`, `SNESGetConvergedReason()`, `SNESConvergedReasonViewFromOptions()`, 4437db781477SPatrick Sanan `PetscViewerPushFormat()`, `PetscViewerPopFormat()` 44382a359c20SBarry Smith 44392a359c20SBarry Smith @*/ 444019a666eeSBarry Smith PetscErrorCode SNESConvergedReasonView(SNES snes,PetscViewer viewer) 44412a359c20SBarry Smith { 444275cca76cSMatthew G. Knepley PetscViewerFormat format; 44432a359c20SBarry Smith PetscBool isAscii; 44442a359c20SBarry Smith 44452a359c20SBarry Smith PetscFunctionBegin; 444619a666eeSBarry Smith if (!viewer) viewer = PETSC_VIEWER_STDOUT_(PetscObjectComm((PetscObject)snes)); 44479566063dSJacob Faibussowitsch PetscCall(PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERASCII,&isAscii)); 44482a359c20SBarry Smith if (isAscii) { 44499566063dSJacob Faibussowitsch PetscCall(PetscViewerGetFormat(viewer, &format)); 44509566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIAddTab(viewer,((PetscObject)snes)->tablevel)); 445175cca76cSMatthew G. Knepley if (format == PETSC_VIEWER_ASCII_INFO_DETAIL) { 445275cca76cSMatthew G. Knepley DM dm; 445375cca76cSMatthew G. Knepley Vec u; 445475cca76cSMatthew G. Knepley PetscDS prob; 445575cca76cSMatthew G. Knepley PetscInt Nf, f; 445695cbbfd3SMatthew G. Knepley PetscErrorCode (**exactSol)(PetscInt, PetscReal, const PetscReal[], PetscInt, PetscScalar[], void *); 445795cbbfd3SMatthew G. Knepley void **exactCtx; 445875cca76cSMatthew G. Knepley PetscReal error; 445975cca76cSMatthew G. Knepley 44609566063dSJacob Faibussowitsch PetscCall(SNESGetDM(snes, &dm)); 44619566063dSJacob Faibussowitsch PetscCall(SNESGetSolution(snes, &u)); 44629566063dSJacob Faibussowitsch PetscCall(DMGetDS(dm, &prob)); 44639566063dSJacob Faibussowitsch PetscCall(PetscDSGetNumFields(prob, &Nf)); 44649566063dSJacob Faibussowitsch PetscCall(PetscMalloc2(Nf, &exactSol, Nf, &exactCtx)); 44659566063dSJacob Faibussowitsch for (f = 0; f < Nf; ++f) PetscCall(PetscDSGetExactSolution(prob, f, &exactSol[f], &exactCtx[f])); 44669566063dSJacob Faibussowitsch PetscCall(DMComputeL2Diff(dm, 0.0, exactSol, exactCtx, u, &error)); 44679566063dSJacob Faibussowitsch PetscCall(PetscFree2(exactSol, exactCtx)); 44689566063dSJacob Faibussowitsch if (error < 1.0e-11) PetscCall(PetscViewerASCIIPrintf(viewer, "L_2 Error: < 1.0e-11\n")); 446963a3b9bcSJacob Faibussowitsch else PetscCall(PetscViewerASCIIPrintf(viewer, "L_2 Error: %g\n", (double)error)); 447075cca76cSMatthew G. Knepley } 4471eafd5ff0SAlex Lindsay if (snes->reason > 0 && format != PETSC_VIEWER_FAILED) { 44722a359c20SBarry Smith if (((PetscObject) snes)->prefix) { 447363a3b9bcSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer,"Nonlinear %s solve converged due to %s iterations %" PetscInt_FMT "\n",((PetscObject) snes)->prefix,SNESConvergedReasons[snes->reason],snes->iter)); 44742a359c20SBarry Smith } else { 447563a3b9bcSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer,"Nonlinear solve converged due to %s iterations %" PetscInt_FMT "\n",SNESConvergedReasons[snes->reason],snes->iter)); 44762a359c20SBarry Smith } 4477eafd5ff0SAlex Lindsay } else if (snes->reason <= 0) { 44782a359c20SBarry Smith if (((PetscObject) snes)->prefix) { 447963a3b9bcSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer,"Nonlinear %s solve did not converge due to %s iterations %" PetscInt_FMT "\n",((PetscObject) snes)->prefix,SNESConvergedReasons[snes->reason],snes->iter)); 44802a359c20SBarry Smith } else { 448163a3b9bcSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer,"Nonlinear solve did not converge due to %s iterations %" PetscInt_FMT "\n",SNESConvergedReasons[snes->reason],snes->iter)); 44822a359c20SBarry Smith } 44832a359c20SBarry Smith } 44849566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIISubtractTab(viewer,((PetscObject)snes)->tablevel)); 44852a359c20SBarry Smith } 44862a359c20SBarry Smith PetscFunctionReturn(0); 44872a359c20SBarry Smith } 44882a359c20SBarry Smith 4489c4421ceaSFande Kong /*@C 4490c4421ceaSFande Kong SNESConvergedReasonViewSet - Sets an ADDITIONAL function that is to be used at the 4491c4421ceaSFande Kong end of the nonlinear solver to display the conver reason of the nonlinear solver. 4492c4421ceaSFande Kong 4493c4421ceaSFande Kong Logically Collective on SNES 4494c4421ceaSFande Kong 4495c4421ceaSFande Kong Input Parameters: 4496c4421ceaSFande Kong + snes - the SNES context 4497c4421ceaSFande Kong . f - the snes converged reason view function 4498c4421ceaSFande Kong . vctx - [optional] user-defined context for private data for the 4499c4421ceaSFande Kong snes converged reason view routine (use NULL if no context is desired) 4500c4421ceaSFande Kong - reasonviewdestroy - [optional] routine that frees reasonview context 4501c4421ceaSFande Kong (may be NULL) 4502c4421ceaSFande Kong 4503c4421ceaSFande Kong Options Database Keys: 4504c4421ceaSFande Kong + -snes_converged_reason - sets a default SNESConvergedReasonView() 4505c4421ceaSFande Kong - -snes_converged_reason_view_cancel - cancels all converged reason viewers that have 4506c4421ceaSFande Kong been hardwired into a code by 4507c4421ceaSFande Kong calls to SNESConvergedReasonViewSet(), but 4508c4421ceaSFande Kong does not cancel those set via 4509c4421ceaSFande Kong the options database. 4510c4421ceaSFande Kong 4511c4421ceaSFande Kong Notes: 4512c4421ceaSFande Kong Several different converged reason view routines may be set by calling 4513c4421ceaSFande Kong SNESConvergedReasonViewSet() multiple times; all will be called in the 4514c4421ceaSFande Kong order in which they were set. 4515c4421ceaSFande Kong 4516c4421ceaSFande Kong Level: intermediate 4517c4421ceaSFande Kong 4518db781477SPatrick Sanan .seealso: `SNESConvergedReasonView()`, `SNESConvergedReasonViewCancel()` 4519c4421ceaSFande Kong @*/ 4520c4421ceaSFande Kong PetscErrorCode SNESConvergedReasonViewSet(SNES snes,PetscErrorCode (*f)(SNES,void*),void *vctx,PetscErrorCode (*reasonviewdestroy)(void**)) 4521c4421ceaSFande Kong { 4522c4421ceaSFande Kong PetscInt i; 4523c4421ceaSFande Kong PetscBool identical; 4524c4421ceaSFande Kong 4525c4421ceaSFande Kong PetscFunctionBegin; 4526c4421ceaSFande Kong PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 4527c4421ceaSFande Kong for (i=0; i<snes->numberreasonviews;i++) { 45289566063dSJacob Faibussowitsch PetscCall(PetscMonitorCompare((PetscErrorCode (*)(void))f,vctx,reasonviewdestroy,(PetscErrorCode (*)(void))snes->reasonview[i],snes->reasonviewcontext[i],snes->reasonviewdestroy[i],&identical)); 4529c4421ceaSFande Kong if (identical) PetscFunctionReturn(0); 4530c4421ceaSFande Kong } 45315f80ce2aSJacob Faibussowitsch PetscCheck(snes->numberreasonviews < MAXSNESREASONVIEWS,PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Too many SNES reasonview set"); 4532c4421ceaSFande Kong snes->reasonview[snes->numberreasonviews] = f; 4533c4421ceaSFande Kong snes->reasonviewdestroy[snes->numberreasonviews] = reasonviewdestroy; 4534c4421ceaSFande Kong snes->reasonviewcontext[snes->numberreasonviews++] = (void*)vctx; 4535c4421ceaSFande Kong PetscFunctionReturn(0); 4536c4421ceaSFande Kong } 4537c4421ceaSFande Kong 453891f3e32bSBarry Smith /*@ 453919a666eeSBarry Smith SNESConvergedReasonViewFromOptions - Processes command line options to determine if/how a SNESReason is to be viewed. 4540c4421ceaSFande Kong All the user-provided convergedReasonView routines will be involved as well, if they exist. 45412a359c20SBarry Smith 45422a359c20SBarry Smith Collective on SNES 45432a359c20SBarry Smith 45442a359c20SBarry Smith Input Parameters: 45452a359c20SBarry Smith . snes - the SNES object 45462a359c20SBarry Smith 45472a359c20SBarry Smith Level: intermediate 45482a359c20SBarry Smith 4549db781477SPatrick Sanan .seealso: `SNESCreate()`, `SNESSetUp()`, `SNESDestroy()`, `SNESSetTolerances()`, `SNESConvergedDefault()`, `SNESGetConvergedReason()`, `SNESConvergedReasonView()` 455019a666eeSBarry Smith 45512a359c20SBarry Smith @*/ 455219a666eeSBarry Smith PetscErrorCode SNESConvergedReasonViewFromOptions(SNES snes) 45532a359c20SBarry Smith { 45542a359c20SBarry Smith PetscViewer viewer; 45552a359c20SBarry Smith PetscBool flg; 45562a359c20SBarry Smith static PetscBool incall = PETSC_FALSE; 45572a359c20SBarry Smith PetscViewerFormat format; 4558c4421ceaSFande Kong PetscInt i; 45592a359c20SBarry Smith 45602a359c20SBarry Smith PetscFunctionBegin; 45612a359c20SBarry Smith if (incall) PetscFunctionReturn(0); 45622a359c20SBarry Smith incall = PETSC_TRUE; 4563c4421ceaSFande Kong 4564c4421ceaSFande Kong /* All user-provided viewers are called first, if they exist. */ 4565c4421ceaSFande Kong for (i=0; i<snes->numberreasonviews; i++) { 45669566063dSJacob Faibussowitsch PetscCall((*snes->reasonview[i])(snes,snes->reasonviewcontext[i])); 4567c4421ceaSFande Kong } 4568c4421ceaSFande Kong 4569c4421ceaSFande Kong /* Call PETSc default routine if users ask for it */ 45709566063dSJacob Faibussowitsch PetscCall(PetscOptionsGetViewer(PetscObjectComm((PetscObject)snes),((PetscObject)snes)->options,((PetscObject)snes)->prefix,"-snes_converged_reason",&viewer,&format,&flg)); 45712a359c20SBarry Smith if (flg) { 45729566063dSJacob Faibussowitsch PetscCall(PetscViewerPushFormat(viewer,format)); 45739566063dSJacob Faibussowitsch PetscCall(SNESConvergedReasonView(snes,viewer)); 45749566063dSJacob Faibussowitsch PetscCall(PetscViewerPopFormat(viewer)); 45759566063dSJacob Faibussowitsch PetscCall(PetscViewerDestroy(&viewer)); 45762a359c20SBarry Smith } 45772a359c20SBarry Smith incall = PETSC_FALSE; 45782a359c20SBarry Smith PetscFunctionReturn(0); 45792a359c20SBarry Smith } 45802a359c20SBarry Smith 4581487a658cSBarry Smith /*@ 4582f69a0ea3SMatthew Knepley SNESSolve - Solves a nonlinear system F(x) = b. 4583f69a0ea3SMatthew Knepley Call SNESSolve() after calling SNESCreate() and optional routines of the form SNESSetXXX(). 45849b94acceSBarry Smith 4585c7afd0dbSLois Curfman McInnes Collective on SNES 4586c7afd0dbSLois Curfman McInnes 4587b2002411SLois Curfman McInnes Input Parameters: 4588c7afd0dbSLois Curfman McInnes + snes - the SNES context 45890298fd71SBarry Smith . b - the constant part of the equation F(x) = b, or NULL to use zero. 459085385478SLisandro Dalcin - x - the solution vector. 45919b94acceSBarry Smith 4592b2002411SLois Curfman McInnes Notes: 45938ddd3da0SLois Curfman McInnes The user should initialize the vector,x, with the initial guess 45946b7fb656SBarry Smith for the nonlinear solve prior to calling SNESSolve(). In particular, 45958ddd3da0SLois Curfman McInnes to employ an initial guess of zero, the user should explicitly set 45968ddd3da0SLois Curfman McInnes this vector to zero by calling VecSet(). 45978ddd3da0SLois Curfman McInnes 459836851e7fSLois Curfman McInnes Level: beginner 459936851e7fSLois Curfman McInnes 4600db781477SPatrick Sanan .seealso: `SNESCreate()`, `SNESDestroy()`, `SNESSetFunction()`, `SNESSetJacobian()`, `SNESSetGridSequence()`, `SNESGetSolution()`, 4601db781477SPatrick Sanan `SNESNewtonTRSetPreCheck()`, `SNESNewtonTRGetPreCheck()`, `SNESNewtonTRSetPostCheck()`, `SNESNewtonTRGetPostCheck()`, 4602db781477SPatrick Sanan `SNESLineSearchSetPostCheck()`, `SNESLineSearchGetPostCheck()`, `SNESLineSearchSetPreCheck()`, `SNESLineSearchGetPreCheck()` 46039b94acceSBarry Smith @*/ 46047087cfbeSBarry Smith PetscErrorCode SNESSolve(SNES snes,Vec b,Vec x) 46059b94acceSBarry Smith { 4606ace3abfcSBarry Smith PetscBool flg; 4607efd51863SBarry Smith PetscInt grid; 46080298fd71SBarry Smith Vec xcreated = NULL; 4609caa4e7f2SJed Brown DM dm; 4610052efed2SBarry Smith 46113a40ed3dSBarry Smith PetscFunctionBegin; 46120700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 4613a69afd8bSBarry Smith if (x) PetscValidHeaderSpecific(x,VEC_CLASSID,3); 4614a69afd8bSBarry Smith if (x) PetscCheckSameComm(snes,1,x,3); 46150700a824SBarry Smith if (b) PetscValidHeaderSpecific(b,VEC_CLASSID,2); 461685385478SLisandro Dalcin if (b) PetscCheckSameComm(snes,1,b,2); 461785385478SLisandro Dalcin 461834b4d3a8SMatthew G. Knepley /* High level operations using the nonlinear solver */ 461906fc46c8SMatthew G. Knepley { 462006fc46c8SMatthew G. Knepley PetscViewer viewer; 462106fc46c8SMatthew G. Knepley PetscViewerFormat format; 46227c88af5aSMatthew G. Knepley PetscInt num; 462306fc46c8SMatthew G. Knepley PetscBool flg; 462406fc46c8SMatthew G. Knepley static PetscBool incall = PETSC_FALSE; 462506fc46c8SMatthew G. Knepley 462606fc46c8SMatthew G. Knepley if (!incall) { 462734b4d3a8SMatthew G. Knepley /* Estimate the convergence rate of the discretization */ 46289566063dSJacob Faibussowitsch PetscCall(PetscOptionsGetViewer(PetscObjectComm((PetscObject) snes),((PetscObject)snes)->options, ((PetscObject) snes)->prefix, "-snes_convergence_estimate", &viewer, &format, &flg)); 462906fc46c8SMatthew G. Knepley if (flg) { 463006fc46c8SMatthew G. Knepley PetscConvEst conv; 463146079b62SMatthew G. Knepley DM dm; 463246079b62SMatthew G. Knepley PetscReal *alpha; /* Convergence rate of the solution error for each field in the L_2 norm */ 463346079b62SMatthew G. Knepley PetscInt Nf; 463406fc46c8SMatthew G. Knepley 463506fc46c8SMatthew G. Knepley incall = PETSC_TRUE; 46369566063dSJacob Faibussowitsch PetscCall(SNESGetDM(snes, &dm)); 46379566063dSJacob Faibussowitsch PetscCall(DMGetNumFields(dm, &Nf)); 46389566063dSJacob Faibussowitsch PetscCall(PetscCalloc1(Nf, &alpha)); 46399566063dSJacob Faibussowitsch PetscCall(PetscConvEstCreate(PetscObjectComm((PetscObject) snes), &conv)); 46409566063dSJacob Faibussowitsch PetscCall(PetscConvEstSetSolver(conv, (PetscObject) snes)); 46419566063dSJacob Faibussowitsch PetscCall(PetscConvEstSetFromOptions(conv)); 46429566063dSJacob Faibussowitsch PetscCall(PetscConvEstSetUp(conv)); 46439566063dSJacob Faibussowitsch PetscCall(PetscConvEstGetConvRate(conv, alpha)); 46449566063dSJacob Faibussowitsch PetscCall(PetscViewerPushFormat(viewer, format)); 46459566063dSJacob Faibussowitsch PetscCall(PetscConvEstRateView(conv, alpha, viewer)); 46469566063dSJacob Faibussowitsch PetscCall(PetscViewerPopFormat(viewer)); 46479566063dSJacob Faibussowitsch PetscCall(PetscViewerDestroy(&viewer)); 46489566063dSJacob Faibussowitsch PetscCall(PetscConvEstDestroy(&conv)); 46499566063dSJacob Faibussowitsch PetscCall(PetscFree(alpha)); 465006fc46c8SMatthew G. Knepley incall = PETSC_FALSE; 465106fc46c8SMatthew G. Knepley } 465234b4d3a8SMatthew G. Knepley /* Adaptively refine the initial grid */ 4653b2588ea6SMatthew G. Knepley num = 1; 46549566063dSJacob Faibussowitsch PetscCall(PetscOptionsGetInt(NULL, ((PetscObject) snes)->prefix, "-snes_adapt_initial", &num, &flg)); 465534b4d3a8SMatthew G. Knepley if (flg) { 465634b4d3a8SMatthew G. Knepley DMAdaptor adaptor; 465734b4d3a8SMatthew G. Knepley 465834b4d3a8SMatthew G. Knepley incall = PETSC_TRUE; 46599566063dSJacob Faibussowitsch PetscCall(DMAdaptorCreate(PetscObjectComm((PetscObject)snes), &adaptor)); 46609566063dSJacob Faibussowitsch PetscCall(DMAdaptorSetSolver(adaptor, snes)); 46619566063dSJacob Faibussowitsch PetscCall(DMAdaptorSetSequenceLength(adaptor, num)); 46629566063dSJacob Faibussowitsch PetscCall(DMAdaptorSetFromOptions(adaptor)); 46639566063dSJacob Faibussowitsch PetscCall(DMAdaptorSetUp(adaptor)); 46649566063dSJacob Faibussowitsch PetscCall(DMAdaptorAdapt(adaptor, x, DM_ADAPTATION_INITIAL, &dm, &x)); 46659566063dSJacob Faibussowitsch PetscCall(DMAdaptorDestroy(&adaptor)); 466634b4d3a8SMatthew G. Knepley incall = PETSC_FALSE; 466734b4d3a8SMatthew G. Knepley } 46687c88af5aSMatthew G. Knepley /* Use grid sequencing to adapt */ 46697c88af5aSMatthew G. Knepley num = 0; 46709566063dSJacob Faibussowitsch PetscCall(PetscOptionsGetInt(NULL, ((PetscObject) snes)->prefix, "-snes_adapt_sequence", &num, NULL)); 46717c88af5aSMatthew G. Knepley if (num) { 46727c88af5aSMatthew G. Knepley DMAdaptor adaptor; 46737c88af5aSMatthew G. Knepley 46747c88af5aSMatthew G. Knepley incall = PETSC_TRUE; 46759566063dSJacob Faibussowitsch PetscCall(DMAdaptorCreate(PetscObjectComm((PetscObject)snes), &adaptor)); 46769566063dSJacob Faibussowitsch PetscCall(DMAdaptorSetSolver(adaptor, snes)); 46779566063dSJacob Faibussowitsch PetscCall(DMAdaptorSetSequenceLength(adaptor, num)); 46789566063dSJacob Faibussowitsch PetscCall(DMAdaptorSetFromOptions(adaptor)); 46799566063dSJacob Faibussowitsch PetscCall(DMAdaptorSetUp(adaptor)); 46809566063dSJacob Faibussowitsch PetscCall(DMAdaptorAdapt(adaptor, x, DM_ADAPTATION_SEQUENTIAL, &dm, &x)); 46819566063dSJacob Faibussowitsch PetscCall(DMAdaptorDestroy(&adaptor)); 46827c88af5aSMatthew G. Knepley incall = PETSC_FALSE; 46837c88af5aSMatthew G. Knepley } 468406fc46c8SMatthew G. Knepley } 468506fc46c8SMatthew G. Knepley } 468641e867a5SStefano Zampini if (!x) { x = snes->vec_sol; } 4687caa4e7f2SJed Brown if (!x) { 46889566063dSJacob Faibussowitsch PetscCall(SNESGetDM(snes,&dm)); 46899566063dSJacob Faibussowitsch PetscCall(DMCreateGlobalVector(dm,&xcreated)); 4690a69afd8bSBarry Smith x = xcreated; 4691a69afd8bSBarry Smith } 46929566063dSJacob Faibussowitsch PetscCall(SNESViewFromOptions(snes,NULL,"-snes_view_pre")); 4693f05ece33SBarry Smith 46949566063dSJacob Faibussowitsch for (grid=0; grid<snes->gridsequence; grid++) PetscCall(PetscViewerASCIIPushTab(PETSC_VIEWER_STDOUT_(PetscObjectComm((PetscObject)snes)))); 4695efd51863SBarry Smith for (grid=0; grid<snes->gridsequence+1; grid++) { 4696efd51863SBarry Smith 469785385478SLisandro Dalcin /* set solution vector */ 46989566063dSJacob Faibussowitsch if (!grid) PetscCall(PetscObjectReference((PetscObject)x)); 46999566063dSJacob Faibussowitsch PetscCall(VecDestroy(&snes->vec_sol)); 470085385478SLisandro Dalcin snes->vec_sol = x; 47019566063dSJacob Faibussowitsch PetscCall(SNESGetDM(snes,&dm)); 4702caa4e7f2SJed Brown 4703caa4e7f2SJed Brown /* set affine vector if provided */ 47049566063dSJacob Faibussowitsch if (b) PetscCall(PetscObjectReference((PetscObject)b)); 47059566063dSJacob Faibussowitsch PetscCall(VecDestroy(&snes->vec_rhs)); 470685385478SLisandro Dalcin snes->vec_rhs = b; 470785385478SLisandro Dalcin 47085f80ce2aSJacob Faibussowitsch if (snes->vec_rhs) PetscCheck(snes->vec_func != snes->vec_rhs,PETSC_COMM_SELF,PETSC_ERR_ARG_IDN,"Right hand side vector cannot be function vector"); 47095f80ce2aSJacob Faibussowitsch PetscCheck(snes->vec_func != snes->vec_sol,PETSC_COMM_SELF,PETSC_ERR_ARG_IDN,"Solution vector cannot be function vector"); 47105f80ce2aSJacob Faibussowitsch PetscCheck(snes->vec_rhs != snes->vec_sol,PETSC_COMM_SELF,PETSC_ERR_ARG_IDN,"Solution vector cannot be right hand side vector"); 4711154060b5SMatthew G. Knepley if (!snes->vec_sol_update /* && snes->vec_sol */) { 47129566063dSJacob Faibussowitsch PetscCall(VecDuplicate(snes->vec_sol,&snes->vec_sol_update)); 47139566063dSJacob Faibussowitsch PetscCall(PetscLogObjectParent((PetscObject)snes,(PetscObject)snes->vec_sol_update)); 4714154060b5SMatthew G. Knepley } 47159566063dSJacob Faibussowitsch PetscCall(DMShellSetGlobalVector(dm,snes->vec_sol)); 47169566063dSJacob Faibussowitsch PetscCall(SNESSetUp(snes)); 47173f149594SLisandro Dalcin 47187eee914bSBarry Smith if (!grid) { 471925e27a38SBarry Smith if (snes->ops->computeinitialguess) PetscCallBack("SNES callback initial guess",(*snes->ops->computeinitialguess)(snes,snes->vec_sol,snes->initialguessP)); 4720dd568438SSatish Balay } 4721d25893d9SBarry Smith 4722abc0a331SBarry Smith if (snes->conv_hist_reset) snes->conv_hist_len = 0; 4723971e163fSPeter Brune if (snes->counters_reset) {snes->nfuncs = 0; snes->linear_its = 0; snes->numFailures = 0;} 4724d5e45103SBarry Smith 47259566063dSJacob Faibussowitsch PetscCall(PetscLogEventBegin(SNES_Solve,snes,0,0,0)); 4726*dbbe0bcdSBarry Smith PetscUseTypeMethod(snes,solve); 47279566063dSJacob Faibussowitsch PetscCall(PetscLogEventEnd(SNES_Solve,snes,0,0,0)); 47285f80ce2aSJacob Faibussowitsch PetscCheck(snes->reason,PETSC_COMM_SELF,PETSC_ERR_PLIB,"Internal error, solver returned without setting converged reason"); 4729422a814eSBarry Smith snes->domainerror = PETSC_FALSE; /* clear the flag if it has been set */ 47303f149594SLisandro Dalcin 473137ec4e1aSPeter Brune if (snes->lagjac_persist) snes->jac_iter += snes->iter; 473237ec4e1aSPeter Brune if (snes->lagpre_persist) snes->pre_iter += snes->iter; 473337ec4e1aSPeter Brune 47349566063dSJacob Faibussowitsch PetscCall(PetscOptionsGetViewer(PetscObjectComm((PetscObject)snes),((PetscObject)snes)->options,((PetscObject)snes)->prefix,"-snes_test_local_min",NULL,NULL,&flg)); 47359566063dSJacob Faibussowitsch if (flg && !PetscPreLoadingOn) PetscCall(SNESTestLocalMin(snes)); 4736c4421ceaSFande Kong /* Call converged reason views. This may involve user-provided viewers as well */ 47379566063dSJacob Faibussowitsch PetscCall(SNESConvergedReasonViewFromOptions(snes)); 47385968eb51SBarry Smith 47395f80ce2aSJacob Faibussowitsch if (snes->errorifnotconverged) PetscCheck(snes->reason >= 0,PetscObjectComm((PetscObject)snes),PETSC_ERR_NOT_CONVERGED,"SNESSolve has not converged"); 47409c8e83a9SBarry Smith if (snes->reason < 0) break; 4741efd51863SBarry Smith if (grid < snes->gridsequence) { 4742efd51863SBarry Smith DM fine; 4743efd51863SBarry Smith Vec xnew; 4744efd51863SBarry Smith Mat interp; 4745efd51863SBarry Smith 47469566063dSJacob Faibussowitsch PetscCall(DMRefine(snes->dm,PetscObjectComm((PetscObject)snes),&fine)); 47475f80ce2aSJacob Faibussowitsch PetscCheck(fine,PetscObjectComm((PetscObject)snes),PETSC_ERR_ARG_INCOMP,"DMRefine() did not perform any refinement, cannot continue grid sequencing"); 47489566063dSJacob Faibussowitsch PetscCall(DMCreateInterpolation(snes->dm,fine,&interp,NULL)); 47499566063dSJacob Faibussowitsch PetscCall(DMCreateGlobalVector(fine,&xnew)); 47509566063dSJacob Faibussowitsch PetscCall(MatInterpolate(interp,x,xnew)); 47519566063dSJacob Faibussowitsch PetscCall(DMInterpolate(snes->dm,interp,fine)); 47529566063dSJacob Faibussowitsch PetscCall(MatDestroy(&interp)); 4753efd51863SBarry Smith x = xnew; 4754efd51863SBarry Smith 47559566063dSJacob Faibussowitsch PetscCall(SNESReset(snes)); 47569566063dSJacob Faibussowitsch PetscCall(SNESSetDM(snes,fine)); 47579566063dSJacob Faibussowitsch PetscCall(SNESResetFromOptions(snes)); 47589566063dSJacob Faibussowitsch PetscCall(DMDestroy(&fine)); 47599566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPopTab(PETSC_VIEWER_STDOUT_(PetscObjectComm((PetscObject)snes)))); 4760efd51863SBarry Smith } 4761efd51863SBarry Smith } 47629566063dSJacob Faibussowitsch PetscCall(SNESViewFromOptions(snes,NULL,"-snes_view")); 47639566063dSJacob Faibussowitsch PetscCall(VecViewFromOptions(snes->vec_sol,(PetscObject)snes,"-snes_view_solution")); 47649566063dSJacob Faibussowitsch PetscCall(DMMonitor(snes->dm)); 47659566063dSJacob Faibussowitsch PetscCall(SNESMonitorPauseFinal_Internal(snes)); 47663f7e2da0SPeter Brune 47679566063dSJacob Faibussowitsch PetscCall(VecDestroy(&xcreated)); 47689566063dSJacob Faibussowitsch PetscCall(PetscObjectSAWsBlock((PetscObject)snes)); 47693a40ed3dSBarry Smith PetscFunctionReturn(0); 47709b94acceSBarry Smith } 47719b94acceSBarry Smith 47729b94acceSBarry Smith /* --------- Internal routines for SNES Package --------- */ 47739b94acceSBarry Smith 477482bf6240SBarry Smith /*@C 47754b0e389bSBarry Smith SNESSetType - Sets the method for the nonlinear solver. 47769b94acceSBarry Smith 4777fee21e36SBarry Smith Collective on SNES 4778fee21e36SBarry Smith 4779c7afd0dbSLois Curfman McInnes Input Parameters: 4780c7afd0dbSLois Curfman McInnes + snes - the SNES context 4781454a90a3SBarry Smith - type - a known method 4782c7afd0dbSLois Curfman McInnes 4783c7afd0dbSLois Curfman McInnes Options Database Key: 4784454a90a3SBarry Smith . -snes_type <type> - Sets the method; use -help for a list 478504d7464bSBarry Smith of available methods (for instance, newtonls or newtontr) 4786ae12b187SLois Curfman McInnes 47879b94acceSBarry Smith Notes: 4788e090d566SSatish Balay See "petsc/include/petscsnes.h" for available methods (for instance) 478904d7464bSBarry Smith + SNESNEWTONLS - Newton's method with line search 4790c7afd0dbSLois Curfman McInnes (systems of nonlinear equations) 4791a2b725a8SWilliam Gropp - SNESNEWTONTR - Newton's method with trust region 4792c7afd0dbSLois Curfman McInnes (systems of nonlinear equations) 47939b94acceSBarry Smith 4794ae12b187SLois Curfman McInnes Normally, it is best to use the SNESSetFromOptions() command and then 4795ae12b187SLois Curfman McInnes set the SNES solver type from the options database rather than by using 4796ae12b187SLois Curfman McInnes this routine. Using the options database provides the user with 4797ae12b187SLois Curfman McInnes maximum flexibility in evaluating the many nonlinear solvers. 4798ae12b187SLois Curfman McInnes The SNESSetType() routine is provided for those situations where it 4799ae12b187SLois Curfman McInnes is necessary to set the nonlinear solver independently of the command 4800ae12b187SLois Curfman McInnes line or options database. This might be the case, for example, when 4801ae12b187SLois Curfman McInnes the choice of solver changes during the execution of the program, 4802ae12b187SLois Curfman McInnes and the user's application is taking responsibility for choosing the 4803b0a32e0cSBarry Smith appropriate method. 480436851e7fSLois Curfman McInnes 480595452b02SPatrick Sanan Developer Notes: 480695452b02SPatrick Sanan SNESRegister() adds a constructor for a new SNESType to SNESList, SNESSetType() locates 48078f6c3df8SBarry Smith the constructor in that list and calls it to create the spexific object. 48088f6c3df8SBarry Smith 480936851e7fSLois Curfman McInnes Level: intermediate 4810a703fe33SLois Curfman McInnes 4811db781477SPatrick Sanan .seealso: `SNESType`, `SNESCreate()`, `SNESDestroy()`, `SNESGetType()`, `SNESSetFromOptions()` 4812435da068SBarry Smith 48139b94acceSBarry Smith @*/ 481419fd82e9SBarry Smith PetscErrorCode SNESSetType(SNES snes,SNESType type) 48159b94acceSBarry Smith { 4816ace3abfcSBarry Smith PetscBool match; 48175f80ce2aSJacob Faibussowitsch PetscErrorCode (*r)(SNES); 48183a40ed3dSBarry Smith 48193a40ed3dSBarry Smith PetscFunctionBegin; 48200700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 48214482741eSBarry Smith PetscValidCharPointer(type,2); 482282bf6240SBarry Smith 48239566063dSJacob Faibussowitsch PetscCall(PetscObjectTypeCompare((PetscObject)snes,type,&match)); 48240f5bd95cSBarry Smith if (match) PetscFunctionReturn(0); 482592ff6ae8SBarry Smith 48269566063dSJacob Faibussowitsch PetscCall(PetscFunctionListFind(SNESList,type,&r)); 48275f80ce2aSJacob Faibussowitsch PetscCheck(r,PETSC_COMM_SELF,PETSC_ERR_ARG_UNKNOWN_TYPE,"Unable to find requested SNES type %s",type); 482875396ef9SLisandro Dalcin /* Destroy the previous private SNES context */ 4829*dbbe0bcdSBarry Smith PetscTryTypeMethod(snes,destroy); 483075396ef9SLisandro Dalcin /* Reinitialize function pointers in SNESOps structure */ 48319e5d0892SLisandro Dalcin snes->ops->setup = NULL; 48329e5d0892SLisandro Dalcin snes->ops->solve = NULL; 48339e5d0892SLisandro Dalcin snes->ops->view = NULL; 48349e5d0892SLisandro Dalcin snes->ops->setfromoptions = NULL; 48359e5d0892SLisandro Dalcin snes->ops->destroy = NULL; 48367fe760d5SStefano Zampini 48377fe760d5SStefano Zampini /* It may happen the user has customized the line search before calling SNESSetType */ 48389566063dSJacob Faibussowitsch if (((PetscObject)snes)->type_name) PetscCall(SNESLineSearchDestroy(&snes->linesearch)); 48397fe760d5SStefano Zampini 484075396ef9SLisandro Dalcin /* Call the SNESCreate_XXX routine for this particular Nonlinear solver */ 484175396ef9SLisandro Dalcin snes->setupcalled = PETSC_FALSE; 4842f5af7f23SKarl Rupp 48439566063dSJacob Faibussowitsch PetscCall(PetscObjectChangeTypeName((PetscObject)snes,type)); 48449566063dSJacob Faibussowitsch PetscCall((*r)(snes)); 48453a40ed3dSBarry Smith PetscFunctionReturn(0); 48469b94acceSBarry Smith } 48479b94acceSBarry Smith 48489b94acceSBarry Smith /*@C 48499a28b0a6SLois Curfman McInnes SNESGetType - Gets the SNES method type and name (as a string). 48509b94acceSBarry Smith 4851c7afd0dbSLois Curfman McInnes Not Collective 4852c7afd0dbSLois Curfman McInnes 48539b94acceSBarry Smith Input Parameter: 48544b0e389bSBarry Smith . snes - nonlinear solver context 48559b94acceSBarry Smith 48569b94acceSBarry Smith Output Parameter: 48573a7fca6bSBarry Smith . type - SNES method (a character string) 48589b94acceSBarry Smith 485936851e7fSLois Curfman McInnes Level: intermediate 486036851e7fSLois Curfman McInnes 48619b94acceSBarry Smith @*/ 486219fd82e9SBarry Smith PetscErrorCode SNESGetType(SNES snes,SNESType *type) 48639b94acceSBarry Smith { 48643a40ed3dSBarry Smith PetscFunctionBegin; 48650700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 48664482741eSBarry Smith PetscValidPointer(type,2); 48677adad957SLisandro Dalcin *type = ((PetscObject)snes)->type_name; 48683a40ed3dSBarry Smith PetscFunctionReturn(0); 48699b94acceSBarry Smith } 48709b94acceSBarry Smith 48713cd8a7caSMatthew G. Knepley /*@ 48723cd8a7caSMatthew G. Knepley SNESSetSolution - Sets the solution vector for use by the SNES routines. 48733cd8a7caSMatthew G. Knepley 4874d083f849SBarry Smith Logically Collective on SNES 48753cd8a7caSMatthew G. Knepley 48763cd8a7caSMatthew G. Knepley Input Parameters: 48773cd8a7caSMatthew G. Knepley + snes - the SNES context obtained from SNESCreate() 48783cd8a7caSMatthew G. Knepley - u - the solution vector 48793cd8a7caSMatthew G. Knepley 48803cd8a7caSMatthew G. Knepley Level: beginner 48813cd8a7caSMatthew G. Knepley 48823cd8a7caSMatthew G. Knepley @*/ 48833cd8a7caSMatthew G. Knepley PetscErrorCode SNESSetSolution(SNES snes, Vec u) 48843cd8a7caSMatthew G. Knepley { 48853cd8a7caSMatthew G. Knepley DM dm; 48863cd8a7caSMatthew G. Knepley 48873cd8a7caSMatthew G. Knepley PetscFunctionBegin; 48883cd8a7caSMatthew G. Knepley PetscValidHeaderSpecific(snes, SNES_CLASSID, 1); 48893cd8a7caSMatthew G. Knepley PetscValidHeaderSpecific(u, VEC_CLASSID, 2); 48909566063dSJacob Faibussowitsch PetscCall(PetscObjectReference((PetscObject) u)); 48919566063dSJacob Faibussowitsch PetscCall(VecDestroy(&snes->vec_sol)); 48923cd8a7caSMatthew G. Knepley 48933cd8a7caSMatthew G. Knepley snes->vec_sol = u; 48943cd8a7caSMatthew G. Knepley 48959566063dSJacob Faibussowitsch PetscCall(SNESGetDM(snes, &dm)); 48969566063dSJacob Faibussowitsch PetscCall(DMShellSetGlobalVector(dm, u)); 48973cd8a7caSMatthew G. Knepley PetscFunctionReturn(0); 48983cd8a7caSMatthew G. Knepley } 48993cd8a7caSMatthew G. Knepley 490052baeb72SSatish Balay /*@ 49019b94acceSBarry Smith SNESGetSolution - Returns the vector where the approximate solution is 4902c0df2a02SJed Brown stored. This is the fine grid solution when using SNESSetGridSequence(). 49039b94acceSBarry Smith 4904c7afd0dbSLois Curfman McInnes Not Collective, but Vec is parallel if SNES is parallel 4905c7afd0dbSLois Curfman McInnes 49069b94acceSBarry Smith Input Parameter: 49079b94acceSBarry Smith . snes - the SNES context 49089b94acceSBarry Smith 49099b94acceSBarry Smith Output Parameter: 49109b94acceSBarry Smith . x - the solution 49119b94acceSBarry Smith 491270e92668SMatthew Knepley Level: intermediate 491336851e7fSLois Curfman McInnes 4914db781477SPatrick Sanan .seealso: `SNESGetSolutionUpdate()`, `SNESGetFunction()` 49159b94acceSBarry Smith @*/ 49167087cfbeSBarry Smith PetscErrorCode SNESGetSolution(SNES snes,Vec *x) 49179b94acceSBarry Smith { 49183a40ed3dSBarry Smith PetscFunctionBegin; 49190700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 49204482741eSBarry Smith PetscValidPointer(x,2); 492185385478SLisandro Dalcin *x = snes->vec_sol; 492270e92668SMatthew Knepley PetscFunctionReturn(0); 492370e92668SMatthew Knepley } 492470e92668SMatthew Knepley 492552baeb72SSatish Balay /*@ 49269b94acceSBarry Smith SNESGetSolutionUpdate - Returns the vector where the solution update is 49279b94acceSBarry Smith stored. 49289b94acceSBarry Smith 4929c7afd0dbSLois Curfman McInnes Not Collective, but Vec is parallel if SNES is parallel 4930c7afd0dbSLois Curfman McInnes 49319b94acceSBarry Smith Input Parameter: 49329b94acceSBarry Smith . snes - the SNES context 49339b94acceSBarry Smith 49349b94acceSBarry Smith Output Parameter: 49359b94acceSBarry Smith . x - the solution update 49369b94acceSBarry Smith 493736851e7fSLois Curfman McInnes Level: advanced 493836851e7fSLois Curfman McInnes 4939db781477SPatrick Sanan .seealso: `SNESGetSolution()`, `SNESGetFunction()` 49409b94acceSBarry Smith @*/ 49417087cfbeSBarry Smith PetscErrorCode SNESGetSolutionUpdate(SNES snes,Vec *x) 49429b94acceSBarry Smith { 49433a40ed3dSBarry Smith PetscFunctionBegin; 49440700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 49454482741eSBarry Smith PetscValidPointer(x,2); 494685385478SLisandro Dalcin *x = snes->vec_sol_update; 49473a40ed3dSBarry Smith PetscFunctionReturn(0); 49489b94acceSBarry Smith } 49499b94acceSBarry Smith 49509b94acceSBarry Smith /*@C 49513638b69dSLois Curfman McInnes SNESGetFunction - Returns the vector where the function is stored. 49529b94acceSBarry Smith 4953a63bb30eSJed Brown Not Collective, but Vec is parallel if SNES is parallel. Collective if Vec is requested, but has not been created yet. 4954c7afd0dbSLois Curfman McInnes 49559b94acceSBarry Smith Input Parameter: 49569b94acceSBarry Smith . snes - the SNES context 49579b94acceSBarry Smith 4958d8d19677SJose E. Roman Output Parameters: 49590298fd71SBarry Smith + r - the vector that is used to store residuals (or NULL if you don't want it) 4960f8b49ee9SBarry Smith . f - the function (or NULL if you don't want it); see SNESFunction for calling sequence details 49610298fd71SBarry Smith - ctx - the function context (or NULL if you don't want it) 49629b94acceSBarry Smith 496336851e7fSLois Curfman McInnes Level: advanced 496436851e7fSLois Curfman McInnes 496504edfde5SBarry Smith Notes: The vector r DOES NOT, in general contain the current value of the SNES nonlinear function 496604edfde5SBarry Smith 4967db781477SPatrick Sanan .seealso: `SNESSetFunction()`, `SNESGetSolution()`, `SNESFunction` 49689b94acceSBarry Smith @*/ 4969f8b49ee9SBarry Smith PetscErrorCode SNESGetFunction(SNES snes,Vec *r,PetscErrorCode (**f)(SNES,Vec,Vec,void*),void **ctx) 49709b94acceSBarry Smith { 49716cab3a1bSJed Brown DM dm; 4972a63bb30eSJed Brown 49733a40ed3dSBarry Smith PetscFunctionBegin; 49740700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 4975a63bb30eSJed Brown if (r) { 4976a63bb30eSJed Brown if (!snes->vec_func) { 4977a63bb30eSJed Brown if (snes->vec_rhs) { 49789566063dSJacob Faibussowitsch PetscCall(VecDuplicate(snes->vec_rhs,&snes->vec_func)); 4979a63bb30eSJed Brown } else if (snes->vec_sol) { 49809566063dSJacob Faibussowitsch PetscCall(VecDuplicate(snes->vec_sol,&snes->vec_func)); 4981a63bb30eSJed Brown } else if (snes->dm) { 49829566063dSJacob Faibussowitsch PetscCall(DMCreateGlobalVector(snes->dm,&snes->vec_func)); 4983a63bb30eSJed Brown } 4984a63bb30eSJed Brown } 4985a63bb30eSJed Brown *r = snes->vec_func; 4986a63bb30eSJed Brown } 49879566063dSJacob Faibussowitsch PetscCall(SNESGetDM(snes,&dm)); 49889566063dSJacob Faibussowitsch PetscCall(DMSNESGetFunction(dm,f,ctx)); 49893a40ed3dSBarry Smith PetscFunctionReturn(0); 49909b94acceSBarry Smith } 49919b94acceSBarry Smith 4992c79ef259SPeter Brune /*@C 4993be95d8f1SBarry Smith SNESGetNGS - Returns the NGS function and context. 4994c79ef259SPeter Brune 4995c79ef259SPeter Brune Input Parameter: 4996c79ef259SPeter Brune . snes - the SNES context 4997c79ef259SPeter Brune 4998d8d19677SJose E. Roman Output Parameters: 4999be95d8f1SBarry Smith + f - the function (or NULL) see SNESNGSFunction for details 50000298fd71SBarry Smith - ctx - the function context (or NULL) 5001c79ef259SPeter Brune 5002c79ef259SPeter Brune Level: advanced 5003c79ef259SPeter Brune 5004db781477SPatrick Sanan .seealso: `SNESSetNGS()`, `SNESGetFunction()` 5005c79ef259SPeter Brune @*/ 5006c79ef259SPeter Brune 5007be95d8f1SBarry Smith PetscErrorCode SNESGetNGS (SNES snes, PetscErrorCode (**f)(SNES, Vec, Vec, void*), void ** ctx) 5008646217ecSPeter Brune { 50096cab3a1bSJed Brown DM dm; 50106cab3a1bSJed Brown 5011646217ecSPeter Brune PetscFunctionBegin; 5012646217ecSPeter Brune PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 50139566063dSJacob Faibussowitsch PetscCall(SNESGetDM(snes,&dm)); 50149566063dSJacob Faibussowitsch PetscCall(DMSNESGetNGS(dm,f,ctx)); 5015646217ecSPeter Brune PetscFunctionReturn(0); 5016646217ecSPeter Brune } 5017646217ecSPeter Brune 50183c7409f5SSatish Balay /*@C 50193c7409f5SSatish Balay SNESSetOptionsPrefix - Sets the prefix used for searching for all 5020d850072dSLois Curfman McInnes SNES options in the database. 50213c7409f5SSatish Balay 50223f9fe445SBarry Smith Logically Collective on SNES 5023fee21e36SBarry Smith 5024d8d19677SJose E. Roman Input Parameters: 5025c7afd0dbSLois Curfman McInnes + snes - the SNES context 5026c7afd0dbSLois Curfman McInnes - prefix - the prefix to prepend to all option names 5027c7afd0dbSLois Curfman McInnes 5028d850072dSLois Curfman McInnes Notes: 5029a83b1b31SSatish Balay A hyphen (-) must NOT be given at the beginning of the prefix name. 5030c7afd0dbSLois Curfman McInnes The first character of all runtime options is AUTOMATICALLY the hyphen. 5031d850072dSLois Curfman McInnes 503236851e7fSLois Curfman McInnes Level: advanced 503336851e7fSLois Curfman McInnes 5034db781477SPatrick Sanan .seealso: `SNESSetFromOptions()` 50353c7409f5SSatish Balay @*/ 50367087cfbeSBarry Smith PetscErrorCode SNESSetOptionsPrefix(SNES snes,const char prefix[]) 50373c7409f5SSatish Balay { 50383a40ed3dSBarry Smith PetscFunctionBegin; 50390700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 50409566063dSJacob Faibussowitsch PetscCall(PetscObjectSetOptionsPrefix((PetscObject)snes,prefix)); 50419566063dSJacob Faibussowitsch if (!snes->ksp) PetscCall(SNESGetKSP(snes,&snes->ksp)); 504235f5d045SPeter Brune if (snes->linesearch) { 50439566063dSJacob Faibussowitsch PetscCall(SNESGetLineSearch(snes,&snes->linesearch)); 50449566063dSJacob Faibussowitsch PetscCall(PetscObjectSetOptionsPrefix((PetscObject)snes->linesearch,prefix)); 504535f5d045SPeter Brune } 50469566063dSJacob Faibussowitsch PetscCall(KSPSetOptionsPrefix(snes->ksp,prefix)); 50473a40ed3dSBarry Smith PetscFunctionReturn(0); 50483c7409f5SSatish Balay } 50493c7409f5SSatish Balay 50503c7409f5SSatish Balay /*@C 5051f525115eSLois Curfman McInnes SNESAppendOptionsPrefix - Appends to the prefix used for searching for all 5052d850072dSLois Curfman McInnes SNES options in the database. 50533c7409f5SSatish Balay 50543f9fe445SBarry Smith Logically Collective on SNES 5055fee21e36SBarry Smith 5056c7afd0dbSLois Curfman McInnes Input Parameters: 5057c7afd0dbSLois Curfman McInnes + snes - the SNES context 5058c7afd0dbSLois Curfman McInnes - prefix - the prefix to prepend to all option names 5059c7afd0dbSLois Curfman McInnes 5060d850072dSLois Curfman McInnes Notes: 5061a83b1b31SSatish Balay A hyphen (-) must NOT be given at the beginning of the prefix name. 5062c7afd0dbSLois Curfman McInnes The first character of all runtime options is AUTOMATICALLY the hyphen. 5063d850072dSLois Curfman McInnes 506436851e7fSLois Curfman McInnes Level: advanced 506536851e7fSLois Curfman McInnes 5066db781477SPatrick Sanan .seealso: `SNESGetOptionsPrefix()` 50673c7409f5SSatish Balay @*/ 50687087cfbeSBarry Smith PetscErrorCode SNESAppendOptionsPrefix(SNES snes,const char prefix[]) 50693c7409f5SSatish Balay { 50703a40ed3dSBarry Smith PetscFunctionBegin; 50710700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 50729566063dSJacob Faibussowitsch PetscCall(PetscObjectAppendOptionsPrefix((PetscObject)snes,prefix)); 50739566063dSJacob Faibussowitsch if (!snes->ksp) PetscCall(SNESGetKSP(snes,&snes->ksp)); 507435f5d045SPeter Brune if (snes->linesearch) { 50759566063dSJacob Faibussowitsch PetscCall(SNESGetLineSearch(snes,&snes->linesearch)); 50769566063dSJacob Faibussowitsch PetscCall(PetscObjectAppendOptionsPrefix((PetscObject)snes->linesearch,prefix)); 507735f5d045SPeter Brune } 50789566063dSJacob Faibussowitsch PetscCall(KSPAppendOptionsPrefix(snes->ksp,prefix)); 50793a40ed3dSBarry Smith PetscFunctionReturn(0); 50803c7409f5SSatish Balay } 50813c7409f5SSatish Balay 50829ab63eb5SSatish Balay /*@C 50833c7409f5SSatish Balay SNESGetOptionsPrefix - Sets the prefix used for searching for all 50843c7409f5SSatish Balay SNES options in the database. 50853c7409f5SSatish Balay 5086c7afd0dbSLois Curfman McInnes Not Collective 5087c7afd0dbSLois Curfman McInnes 50883c7409f5SSatish Balay Input Parameter: 50893c7409f5SSatish Balay . snes - the SNES context 50903c7409f5SSatish Balay 50913c7409f5SSatish Balay Output Parameter: 50923c7409f5SSatish Balay . prefix - pointer to the prefix string used 50933c7409f5SSatish Balay 509495452b02SPatrick Sanan Notes: 509595452b02SPatrick Sanan On the fortran side, the user should pass in a string 'prefix' of 50969ab63eb5SSatish Balay sufficient length to hold the prefix. 50979ab63eb5SSatish Balay 509836851e7fSLois Curfman McInnes Level: advanced 509936851e7fSLois Curfman McInnes 5100db781477SPatrick Sanan .seealso: `SNESAppendOptionsPrefix()` 51013c7409f5SSatish Balay @*/ 51027087cfbeSBarry Smith PetscErrorCode SNESGetOptionsPrefix(SNES snes,const char *prefix[]) 51033c7409f5SSatish Balay { 51043a40ed3dSBarry Smith PetscFunctionBegin; 51050700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 51069566063dSJacob Faibussowitsch PetscCall(PetscObjectGetOptionsPrefix((PetscObject)snes,prefix)); 51073a40ed3dSBarry Smith PetscFunctionReturn(0); 51083c7409f5SSatish Balay } 51093c7409f5SSatish Balay 51103cea93caSBarry Smith /*@C 51111c84c290SBarry Smith SNESRegister - Adds a method to the nonlinear solver package. 51121c84c290SBarry Smith 51131c84c290SBarry Smith Not collective 51141c84c290SBarry Smith 51151c84c290SBarry Smith Input Parameters: 51161c84c290SBarry Smith + name_solver - name of a new user-defined solver 51171c84c290SBarry Smith - routine_create - routine to create method context 51181c84c290SBarry Smith 51191c84c290SBarry Smith Notes: 51201c84c290SBarry Smith SNESRegister() may be called multiple times to add several user-defined solvers. 51211c84c290SBarry Smith 51221c84c290SBarry Smith Sample usage: 51231c84c290SBarry Smith .vb 5124bdf89e91SBarry Smith SNESRegister("my_solver",MySolverCreate); 51251c84c290SBarry Smith .ve 51261c84c290SBarry Smith 51271c84c290SBarry Smith Then, your solver can be chosen with the procedural interface via 51281c84c290SBarry Smith $ SNESSetType(snes,"my_solver") 51291c84c290SBarry Smith or at runtime via the option 51301c84c290SBarry Smith $ -snes_type my_solver 51311c84c290SBarry Smith 51321c84c290SBarry Smith Level: advanced 51331c84c290SBarry Smith 51341c84c290SBarry Smith Note: If your function is not being put into a shared library then use SNESRegister() instead 51351c84c290SBarry Smith 5136db781477SPatrick Sanan .seealso: `SNESRegisterAll()`, `SNESRegisterDestroy()` 51373cea93caSBarry Smith 51387f6c08e0SMatthew Knepley Level: advanced 51393cea93caSBarry Smith @*/ 5140bdf89e91SBarry Smith PetscErrorCode SNESRegister(const char sname[],PetscErrorCode (*function)(SNES)) 5141b2002411SLois Curfman McInnes { 5142b2002411SLois Curfman McInnes PetscFunctionBegin; 51439566063dSJacob Faibussowitsch PetscCall(SNESInitializePackage()); 51449566063dSJacob Faibussowitsch PetscCall(PetscFunctionListAdd(&SNESList,sname,function)); 5145b2002411SLois Curfman McInnes PetscFunctionReturn(0); 5146b2002411SLois Curfman McInnes } 5147da9b6338SBarry Smith 51487087cfbeSBarry Smith PetscErrorCode SNESTestLocalMin(SNES snes) 5149da9b6338SBarry Smith { 515077431f27SBarry Smith PetscInt N,i,j; 5151da9b6338SBarry Smith Vec u,uh,fh; 5152da9b6338SBarry Smith PetscScalar value; 5153da9b6338SBarry Smith PetscReal norm; 5154da9b6338SBarry Smith 5155da9b6338SBarry Smith PetscFunctionBegin; 51569566063dSJacob Faibussowitsch PetscCall(SNESGetSolution(snes,&u)); 51579566063dSJacob Faibussowitsch PetscCall(VecDuplicate(u,&uh)); 51589566063dSJacob Faibussowitsch PetscCall(VecDuplicate(u,&fh)); 5159da9b6338SBarry Smith 5160da9b6338SBarry Smith /* currently only works for sequential */ 51619566063dSJacob Faibussowitsch PetscCall(PetscPrintf(PetscObjectComm((PetscObject)snes),"Testing FormFunction() for local min\n")); 51629566063dSJacob Faibussowitsch PetscCall(VecGetSize(u,&N)); 5163da9b6338SBarry Smith for (i=0; i<N; i++) { 51649566063dSJacob Faibussowitsch PetscCall(VecCopy(u,uh)); 516563a3b9bcSJacob Faibussowitsch PetscCall(PetscPrintf(PetscObjectComm((PetscObject)snes),"i = %" PetscInt_FMT "\n",i)); 5166da9b6338SBarry Smith for (j=-10; j<11; j++) { 51678b49ba18SBarry Smith value = PetscSign(j)*PetscExpReal(PetscAbs(j)-10.0); 51689566063dSJacob Faibussowitsch PetscCall(VecSetValue(uh,i,value,ADD_VALUES)); 51699566063dSJacob Faibussowitsch PetscCall(SNESComputeFunction(snes,uh,fh)); 51709566063dSJacob Faibussowitsch PetscCall(VecNorm(fh,NORM_2,&norm)); 517163a3b9bcSJacob Faibussowitsch PetscCall(PetscPrintf(PetscObjectComm((PetscObject)snes)," j norm %" PetscInt_FMT " %18.16e\n",j,(double)norm)); 5172da9b6338SBarry Smith value = -value; 51739566063dSJacob Faibussowitsch PetscCall(VecSetValue(uh,i,value,ADD_VALUES)); 5174da9b6338SBarry Smith } 5175da9b6338SBarry Smith } 51769566063dSJacob Faibussowitsch PetscCall(VecDestroy(&uh)); 51779566063dSJacob Faibussowitsch PetscCall(VecDestroy(&fh)); 5178da9b6338SBarry Smith PetscFunctionReturn(0); 5179da9b6338SBarry Smith } 518071f87433Sdalcinl 518171f87433Sdalcinl /*@ 5182fa9f3622SBarry Smith SNESKSPSetUseEW - Sets SNES use Eisenstat-Walker method for 518371f87433Sdalcinl computing relative tolerance for linear solvers within an inexact 518471f87433Sdalcinl Newton method. 518571f87433Sdalcinl 51863f9fe445SBarry Smith Logically Collective on SNES 518771f87433Sdalcinl 518871f87433Sdalcinl Input Parameters: 518971f87433Sdalcinl + snes - SNES context 519071f87433Sdalcinl - flag - PETSC_TRUE or PETSC_FALSE 519171f87433Sdalcinl 519264ba62caSBarry Smith Options Database: 519364ba62caSBarry Smith + -snes_ksp_ew - use Eisenstat-Walker method for determining linear system convergence 519464ba62caSBarry Smith . -snes_ksp_ew_version ver - version of Eisenstat-Walker method 519564ba62caSBarry Smith . -snes_ksp_ew_rtol0 <rtol0> - Sets rtol0 519664ba62caSBarry Smith . -snes_ksp_ew_rtolmax <rtolmax> - Sets rtolmax 519764ba62caSBarry Smith . -snes_ksp_ew_gamma <gamma> - Sets gamma 519864ba62caSBarry Smith . -snes_ksp_ew_alpha <alpha> - Sets alpha 519964ba62caSBarry Smith . -snes_ksp_ew_alpha2 <alpha2> - Sets alpha2 520064ba62caSBarry Smith - -snes_ksp_ew_threshold <threshold> - Sets threshold 520164ba62caSBarry Smith 520271f87433Sdalcinl Notes: 520371f87433Sdalcinl Currently, the default is to use a constant relative tolerance for 520471f87433Sdalcinl the inner linear solvers. Alternatively, one can use the 520571f87433Sdalcinl Eisenstat-Walker method, where the relative convergence tolerance 520671f87433Sdalcinl is reset at each Newton iteration according progress of the nonlinear 520771f87433Sdalcinl solver. 520871f87433Sdalcinl 520971f87433Sdalcinl Level: advanced 521071f87433Sdalcinl 521171f87433Sdalcinl Reference: 521271f87433Sdalcinl S. C. Eisenstat and H. F. Walker, "Choosing the forcing terms in an 521371f87433Sdalcinl inexact Newton method", SISC 17 (1), pp.16-32, 1996. 521471f87433Sdalcinl 5215db781477SPatrick Sanan .seealso: `SNESKSPGetUseEW()`, `SNESKSPGetParametersEW()`, `SNESKSPSetParametersEW()` 521671f87433Sdalcinl @*/ 52177087cfbeSBarry Smith PetscErrorCode SNESKSPSetUseEW(SNES snes,PetscBool flag) 521871f87433Sdalcinl { 521971f87433Sdalcinl PetscFunctionBegin; 52200700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 5221acfcf0e5SJed Brown PetscValidLogicalCollectiveBool(snes,flag,2); 522271f87433Sdalcinl snes->ksp_ewconv = flag; 522371f87433Sdalcinl PetscFunctionReturn(0); 522471f87433Sdalcinl } 522571f87433Sdalcinl 522671f87433Sdalcinl /*@ 5227fa9f3622SBarry Smith SNESKSPGetUseEW - Gets if SNES is using Eisenstat-Walker method 522871f87433Sdalcinl for computing relative tolerance for linear solvers within an 522971f87433Sdalcinl inexact Newton method. 523071f87433Sdalcinl 523171f87433Sdalcinl Not Collective 523271f87433Sdalcinl 523371f87433Sdalcinl Input Parameter: 523471f87433Sdalcinl . snes - SNES context 523571f87433Sdalcinl 523671f87433Sdalcinl Output Parameter: 523771f87433Sdalcinl . flag - PETSC_TRUE or PETSC_FALSE 523871f87433Sdalcinl 523971f87433Sdalcinl Notes: 524071f87433Sdalcinl Currently, the default is to use a constant relative tolerance for 524171f87433Sdalcinl the inner linear solvers. Alternatively, one can use the 524271f87433Sdalcinl Eisenstat-Walker method, where the relative convergence tolerance 524371f87433Sdalcinl is reset at each Newton iteration according progress of the nonlinear 524471f87433Sdalcinl solver. 524571f87433Sdalcinl 524671f87433Sdalcinl Level: advanced 524771f87433Sdalcinl 524871f87433Sdalcinl Reference: 524971f87433Sdalcinl S. C. Eisenstat and H. F. Walker, "Choosing the forcing terms in an 525071f87433Sdalcinl inexact Newton method", SISC 17 (1), pp.16-32, 1996. 525171f87433Sdalcinl 5252db781477SPatrick Sanan .seealso: `SNESKSPSetUseEW()`, `SNESKSPGetParametersEW()`, `SNESKSPSetParametersEW()` 525371f87433Sdalcinl @*/ 52547087cfbeSBarry Smith PetscErrorCode SNESKSPGetUseEW(SNES snes, PetscBool *flag) 525571f87433Sdalcinl { 525671f87433Sdalcinl PetscFunctionBegin; 52570700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 5258534a8f05SLisandro Dalcin PetscValidBoolPointer(flag,2); 525971f87433Sdalcinl *flag = snes->ksp_ewconv; 526071f87433Sdalcinl PetscFunctionReturn(0); 526171f87433Sdalcinl } 526271f87433Sdalcinl 526371f87433Sdalcinl /*@ 5264fa9f3622SBarry Smith SNESKSPSetParametersEW - Sets parameters for Eisenstat-Walker 526571f87433Sdalcinl convergence criteria for the linear solvers within an inexact 526671f87433Sdalcinl Newton method. 526771f87433Sdalcinl 52683f9fe445SBarry Smith Logically Collective on SNES 526971f87433Sdalcinl 527071f87433Sdalcinl Input Parameters: 527171f87433Sdalcinl + snes - SNES context 52720f0abf79SStefano Zampini . version - version 1, 2 (default is 2), 3 or 4 527371f87433Sdalcinl . rtol_0 - initial relative tolerance (0 <= rtol_0 < 1) 527471f87433Sdalcinl . rtol_max - maximum relative tolerance (0 <= rtol_max < 1) 527571f87433Sdalcinl . gamma - multiplicative factor for version 2 rtol computation 527671f87433Sdalcinl (0 <= gamma2 <= 1) 527771f87433Sdalcinl . alpha - power for version 2 rtol computation (1 < alpha <= 2) 527871f87433Sdalcinl . alpha2 - power for safeguard 527971f87433Sdalcinl - threshold - threshold for imposing safeguard (0 < threshold < 1) 528071f87433Sdalcinl 528171f87433Sdalcinl Note: 528271f87433Sdalcinl Version 3 was contributed by Luis Chacon, June 2006. 528371f87433Sdalcinl 528471f87433Sdalcinl Use PETSC_DEFAULT to retain the default for any of the parameters. 528571f87433Sdalcinl 528671f87433Sdalcinl Level: advanced 528771f87433Sdalcinl 528871f87433Sdalcinl Reference: 528971f87433Sdalcinl S. C. Eisenstat and H. F. Walker, "Choosing the forcing terms in an 529071f87433Sdalcinl inexact Newton method", Utah State University Math. Stat. Dept. Res. 529171f87433Sdalcinl Report 6/94/75, June, 1994, to appear in SIAM J. Sci. Comput. 529271f87433Sdalcinl 5293db781477SPatrick Sanan .seealso: `SNESKSPSetUseEW()`, `SNESKSPGetUseEW()`, `SNESKSPGetParametersEW()` 529471f87433Sdalcinl @*/ 5295f5af7f23SKarl Rupp PetscErrorCode SNESKSPSetParametersEW(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; 53025f80ce2aSJacob Faibussowitsch PetscCheck(kctx,PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"No Eisenstat-Walker context existing"); 5303c5eb9154SBarry Smith PetscValidLogicalCollectiveInt(snes,version,2); 5304c5eb9154SBarry Smith PetscValidLogicalCollectiveReal(snes,rtol_0,3); 5305c5eb9154SBarry Smith PetscValidLogicalCollectiveReal(snes,rtol_max,4); 5306c5eb9154SBarry Smith PetscValidLogicalCollectiveReal(snes,gamma,5); 5307c5eb9154SBarry Smith PetscValidLogicalCollectiveReal(snes,alpha,6); 5308c5eb9154SBarry Smith PetscValidLogicalCollectiveReal(snes,alpha2,7); 5309c5eb9154SBarry Smith PetscValidLogicalCollectiveReal(snes,threshold,8); 531071f87433Sdalcinl 531171f87433Sdalcinl if (version != PETSC_DEFAULT) kctx->version = version; 531271f87433Sdalcinl if (rtol_0 != PETSC_DEFAULT) kctx->rtol_0 = rtol_0; 531371f87433Sdalcinl if (rtol_max != PETSC_DEFAULT) kctx->rtol_max = rtol_max; 531471f87433Sdalcinl if (gamma != PETSC_DEFAULT) kctx->gamma = gamma; 531571f87433Sdalcinl if (alpha != PETSC_DEFAULT) kctx->alpha = alpha; 531671f87433Sdalcinl if (alpha2 != PETSC_DEFAULT) kctx->alpha2 = alpha2; 531771f87433Sdalcinl if (threshold != PETSC_DEFAULT) kctx->threshold = threshold; 531871f87433Sdalcinl 53190f0abf79SStefano Zampini PetscCheck(kctx->version >= 1 && kctx->version <= 4,PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Only versions 1 to 4 are supported: %" PetscInt_FMT,kctx->version); 53200b121fc5SBarry Smith PetscCheck(kctx->rtol_0 >= 0.0 && kctx->rtol_0 < 1.0,PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"0.0 <= rtol_0 < 1.0: %g",(double)kctx->rtol_0); 53210b121fc5SBarry Smith PetscCheck(kctx->rtol_max >= 0.0 && kctx->rtol_max < 1.0,PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"0.0 <= rtol_max (%g) < 1.0",(double)kctx->rtol_max); 53220b121fc5SBarry Smith PetscCheck(kctx->gamma >= 0.0 && kctx->gamma <= 1.0,PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"0.0 <= gamma (%g) <= 1.0",(double)kctx->gamma); 53230b121fc5SBarry Smith PetscCheck(kctx->alpha > 1.0 && kctx->alpha <= 2.0,PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"1.0 < alpha (%g) <= 2.0",(double)kctx->alpha); 53240b121fc5SBarry Smith PetscCheck(kctx->threshold > 0.0 && kctx->threshold < 1.0,PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"0.0 < threshold (%g) < 1.0",(double)kctx->threshold); 532571f87433Sdalcinl PetscFunctionReturn(0); 532671f87433Sdalcinl } 532771f87433Sdalcinl 532871f87433Sdalcinl /*@ 5329fa9f3622SBarry Smith SNESKSPGetParametersEW - Gets parameters for Eisenstat-Walker 533071f87433Sdalcinl convergence criteria for the linear solvers within an inexact 533171f87433Sdalcinl Newton method. 533271f87433Sdalcinl 533371f87433Sdalcinl Not Collective 533471f87433Sdalcinl 533597bb3fdcSJose E. Roman Input Parameter: 53366b867d5aSJose E. Roman . snes - SNES context 533771f87433Sdalcinl 533871f87433Sdalcinl Output Parameters: 53390f0abf79SStefano Zampini + version - version 1, 2 (default is 2), 3 or 4 534071f87433Sdalcinl . rtol_0 - initial relative tolerance (0 <= rtol_0 < 1) 534171f87433Sdalcinl . rtol_max - maximum relative tolerance (0 <= rtol_max < 1) 5342bf388a1fSBarry Smith . gamma - multiplicative factor for version 2 rtol computation (0 <= gamma2 <= 1) 534371f87433Sdalcinl . alpha - power for version 2 rtol computation (1 < alpha <= 2) 534471f87433Sdalcinl . alpha2 - power for safeguard 534571f87433Sdalcinl - threshold - threshold for imposing safeguard (0 < threshold < 1) 534671f87433Sdalcinl 534771f87433Sdalcinl Level: advanced 534871f87433Sdalcinl 5349db781477SPatrick Sanan .seealso: `SNESKSPSetUseEW()`, `SNESKSPGetUseEW()`, `SNESKSPSetParametersEW()` 535071f87433Sdalcinl @*/ 5351bf388a1fSBarry Smith PetscErrorCode SNESKSPGetParametersEW(SNES snes,PetscInt *version,PetscReal *rtol_0,PetscReal *rtol_max,PetscReal *gamma,PetscReal *alpha,PetscReal *alpha2,PetscReal *threshold) 535271f87433Sdalcinl { 5353fa9f3622SBarry Smith SNESKSPEW *kctx; 53545fd66863SKarl Rupp 535571f87433Sdalcinl PetscFunctionBegin; 53560700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 5357fa9f3622SBarry Smith kctx = (SNESKSPEW*)snes->kspconvctx; 53585f80ce2aSJacob Faibussowitsch PetscCheck(kctx,PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"No Eisenstat-Walker context existing"); 535971f87433Sdalcinl if (version) *version = kctx->version; 536071f87433Sdalcinl if (rtol_0) *rtol_0 = kctx->rtol_0; 536171f87433Sdalcinl if (rtol_max) *rtol_max = kctx->rtol_max; 536271f87433Sdalcinl if (gamma) *gamma = kctx->gamma; 536371f87433Sdalcinl if (alpha) *alpha = kctx->alpha; 536471f87433Sdalcinl if (alpha2) *alpha2 = kctx->alpha2; 536571f87433Sdalcinl if (threshold) *threshold = kctx->threshold; 536671f87433Sdalcinl PetscFunctionReturn(0); 536771f87433Sdalcinl } 536871f87433Sdalcinl 5369d5378b5fSDmitry Karpeev PetscErrorCode KSPPreSolve_SNESEW(KSP ksp, Vec b, Vec x, SNES snes) 537071f87433Sdalcinl { 5371fa9f3622SBarry Smith SNESKSPEW *kctx = (SNESKSPEW*)snes->kspconvctx; 537271f87433Sdalcinl PetscReal rtol = PETSC_DEFAULT,stol; 537371f87433Sdalcinl 537471f87433Sdalcinl PetscFunctionBegin; 5375d4211eb9SBarry Smith if (!snes->ksp_ewconv) PetscFunctionReturn(0); 537630058271SDmitry Karpeev if (!snes->iter) { 537730058271SDmitry Karpeev rtol = kctx->rtol_0; /* first time in, so use the original user rtol */ 53789566063dSJacob Faibussowitsch PetscCall(VecNorm(snes->vec_func,NORM_2,&kctx->norm_first)); 53790f0abf79SStefano Zampini } else { 538071f87433Sdalcinl if (kctx->version == 1) { 53810f0abf79SStefano Zampini rtol = PetscAbsReal(snes->norm - kctx->lresid_last)/kctx->norm_last; 538285ec1a3cSBarry Smith stol = PetscPowReal(kctx->rtol_last,kctx->alpha2); 538371f87433Sdalcinl if (stol > kctx->threshold) rtol = PetscMax(rtol,stol); 538471f87433Sdalcinl } else if (kctx->version == 2) { 538585ec1a3cSBarry Smith rtol = kctx->gamma * PetscPowReal(snes->norm/kctx->norm_last,kctx->alpha); 538685ec1a3cSBarry Smith stol = kctx->gamma * PetscPowReal(kctx->rtol_last,kctx->alpha); 538771f87433Sdalcinl if (stol > kctx->threshold) rtol = PetscMax(rtol,stol); 538871f87433Sdalcinl } else if (kctx->version == 3) { /* contributed by Luis Chacon, June 2006. */ 538985ec1a3cSBarry Smith rtol = kctx->gamma * PetscPowReal(snes->norm/kctx->norm_last,kctx->alpha); 539071f87433Sdalcinl /* safeguard: avoid sharp decrease of rtol */ 539185ec1a3cSBarry Smith stol = kctx->gamma*PetscPowReal(kctx->rtol_last,kctx->alpha); 539271f87433Sdalcinl stol = PetscMax(rtol,stol); 539371f87433Sdalcinl rtol = PetscMin(kctx->rtol_0,stol); 539471f87433Sdalcinl /* safeguard: avoid oversolving */ 539530058271SDmitry Karpeev stol = kctx->gamma*(kctx->norm_first*snes->rtol)/snes->norm; 539671f87433Sdalcinl stol = PetscMax(rtol,stol); 539771f87433Sdalcinl rtol = PetscMin(kctx->rtol_0,stol); 53980f0abf79SStefano Zampini } else if (kctx->version == 4) { /* H.-B. An et al. Journal of Computational and Applied Mathematics 200 (2007) 47-60 */ 53990f0abf79SStefano Zampini PetscReal ared = PetscAbsReal(kctx->norm_last - snes->norm); 54000f0abf79SStefano Zampini PetscReal pred = PetscAbsReal(kctx->norm_last - kctx->lresid_last); 54010f0abf79SStefano Zampini PetscReal rk = ared / pred; 54020f0abf79SStefano Zampini if (rk < kctx->v4_p1) rtol = 1. - 2.*kctx->v4_p1; 54030f0abf79SStefano Zampini else if (rk < kctx->v4_p2) rtol = kctx->rtol_last; 54040f0abf79SStefano Zampini else if (rk < kctx->v4_p3) rtol = kctx->v4_m1 * kctx->rtol_last; 54050f0abf79SStefano Zampini else rtol = kctx->v4_m2 * kctx->rtol_last; 54060f0abf79SStefano Zampini 54070f0abf79SStefano Zampini if (kctx->rtol_last_2 > kctx->v4_m3 && kctx->rtol_last > kctx->v4_m3 && 54080f0abf79SStefano Zampini kctx->rk_last_2 < kctx->v4_p1 && kctx->rk_last < kctx->v4_p1) { 54090f0abf79SStefano Zampini rtol = kctx->v4_m4 * kctx->rtol_last; 54100f0abf79SStefano Zampini //printf("iter %" PetscInt_FMT ", Eisenstat-Walker (version %" PetscInt_FMT ") KSP rtol=%g (rk %g ps %g %g %g) (AD)\n",snes->iter,kctx->version,(double)rtol,rk,kctx->v4_p1,kctx->v4_p2,kctx->v4_p3); 54110f0abf79SStefano Zampini } else { 54120f0abf79SStefano Zampini //printf("iter %" PetscInt_FMT ", Eisenstat-Walker (version %" PetscInt_FMT ") KSP rtol=%g (rk %g ps %g %g %g)\n",snes->iter,kctx->version,(double)rtol,rk,kctx->v4_p1,kctx->v4_p2,kctx->v4_p3); 541371f87433Sdalcinl } 54140f0abf79SStefano Zampini kctx->rtol_last_2 = kctx->rtol_last; 54150f0abf79SStefano Zampini kctx->rk_last_2 = kctx->rk_last; 54160f0abf79SStefano Zampini kctx->rk_last = rk; 54170f0abf79SStefano Zampini } else SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Only versions 1-4 are supported: %" PetscInt_FMT,kctx->version); 54180f0abf79SStefano Zampini } 54190f0abf79SStefano Zampini /* safeguard: avoid rtol greater than rtol_max */ 542071f87433Sdalcinl rtol = PetscMin(rtol,kctx->rtol_max); 54219566063dSJacob Faibussowitsch PetscCall(KSPSetTolerances(ksp,rtol,PETSC_DEFAULT,PETSC_DEFAULT,PETSC_DEFAULT)); 542263a3b9bcSJacob Faibussowitsch PetscCall(PetscInfo(snes,"iter %" PetscInt_FMT ", Eisenstat-Walker (version %" PetscInt_FMT ") KSP rtol=%g\n",snes->iter,kctx->version,(double)rtol)); 542371f87433Sdalcinl PetscFunctionReturn(0); 542471f87433Sdalcinl } 542571f87433Sdalcinl 5426d5378b5fSDmitry Karpeev PetscErrorCode KSPPostSolve_SNESEW(KSP ksp, Vec b, Vec x, SNES snes) 542771f87433Sdalcinl { 5428fa9f3622SBarry Smith SNESKSPEW *kctx = (SNESKSPEW*)snes->kspconvctx; 542971f87433Sdalcinl PCSide pcside; 543071f87433Sdalcinl Vec lres; 543171f87433Sdalcinl 543271f87433Sdalcinl PetscFunctionBegin; 5433d4211eb9SBarry Smith if (!snes->ksp_ewconv) PetscFunctionReturn(0); 54349566063dSJacob Faibussowitsch PetscCall(KSPGetTolerances(ksp,&kctx->rtol_last,NULL,NULL,NULL)); 543571dbe336SPeter Brune kctx->norm_last = snes->norm; 54360f0abf79SStefano Zampini if (kctx->version == 1 || kctx->version == 4) { 54374f00ce20SMatthew G. Knepley PC pc; 54380f0abf79SStefano Zampini PetscBool getRes; 54394f00ce20SMatthew G. Knepley 54409566063dSJacob Faibussowitsch PetscCall(KSPGetPC(ksp,&pc)); 54410f0abf79SStefano Zampini PetscCall(PetscObjectTypeCompare((PetscObject)pc,PCNONE,&getRes)); 54420f0abf79SStefano Zampini if (!getRes) { 54430f0abf79SStefano Zampini KSPNormType normtype; 54440f0abf79SStefano Zampini 54450f0abf79SStefano Zampini PetscCall(KSPGetNormType(ksp,&normtype)); 54460f0abf79SStefano Zampini getRes = (PetscBool)(normtype == KSP_NORM_UNPRECONDITIONED); 54470f0abf79SStefano Zampini } 54489566063dSJacob Faibussowitsch PetscCall(KSPGetPCSide(ksp,&pcside)); 54490f0abf79SStefano Zampini if (pcside == PC_RIGHT || getRes) { /* KSP residual is true linear residual */ 54509566063dSJacob Faibussowitsch PetscCall(KSPGetResidualNorm(ksp,&kctx->lresid_last)); 545171f87433Sdalcinl } else { 545271f87433Sdalcinl /* KSP residual is preconditioned residual */ 545371f87433Sdalcinl /* compute true linear residual norm */ 54540f0abf79SStefano Zampini Mat J; 54550f0abf79SStefano Zampini PetscCall(KSPGetOperators(ksp,&J,NULL)); 54569566063dSJacob Faibussowitsch PetscCall(VecDuplicate(b,&lres)); 54570f0abf79SStefano Zampini PetscCall(MatMult(J,x,lres)); 54589566063dSJacob Faibussowitsch PetscCall(VecAYPX(lres,-1.0,b)); 54599566063dSJacob Faibussowitsch PetscCall(VecNorm(lres,NORM_2,&kctx->lresid_last)); 54609566063dSJacob Faibussowitsch PetscCall(VecDestroy(&lres)); 546171f87433Sdalcinl } 546271f87433Sdalcinl } 546371f87433Sdalcinl PetscFunctionReturn(0); 546471f87433Sdalcinl } 546571f87433Sdalcinl 5466d4211eb9SBarry Smith /*@ 5467d4211eb9SBarry Smith SNESGetKSP - Returns the KSP context for a SNES solver. 5468d4211eb9SBarry Smith 5469d4211eb9SBarry Smith Not Collective, but if SNES object is parallel, then KSP object is parallel 5470d4211eb9SBarry Smith 5471d4211eb9SBarry Smith Input Parameter: 5472d4211eb9SBarry Smith . snes - the SNES context 5473d4211eb9SBarry Smith 5474d4211eb9SBarry Smith Output Parameter: 5475d4211eb9SBarry Smith . ksp - the KSP context 5476d4211eb9SBarry Smith 5477d4211eb9SBarry Smith Notes: 5478d4211eb9SBarry Smith The user can then directly manipulate the KSP context to set various 5479d4211eb9SBarry Smith options, etc. Likewise, the user can then extract and manipulate the 5480d4211eb9SBarry Smith PC contexts as well. 5481d4211eb9SBarry Smith 5482d4211eb9SBarry Smith Level: beginner 5483d4211eb9SBarry Smith 5484db781477SPatrick Sanan .seealso: `KSPGetPC()`, `SNESCreate()`, `KSPCreate()`, `SNESSetKSP()` 5485d4211eb9SBarry Smith @*/ 5486d4211eb9SBarry Smith PetscErrorCode SNESGetKSP(SNES snes,KSP *ksp) 548771f87433Sdalcinl { 548871f87433Sdalcinl PetscFunctionBegin; 5489d4211eb9SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 5490d4211eb9SBarry Smith PetscValidPointer(ksp,2); 5491d4211eb9SBarry Smith 5492d4211eb9SBarry Smith if (!snes->ksp) { 54939566063dSJacob Faibussowitsch PetscCall(KSPCreate(PetscObjectComm((PetscObject)snes),&snes->ksp)); 54949566063dSJacob Faibussowitsch PetscCall(PetscObjectIncrementTabLevel((PetscObject)snes->ksp,(PetscObject)snes,1)); 54959566063dSJacob Faibussowitsch PetscCall(PetscLogObjectParent((PetscObject)snes,(PetscObject)snes->ksp)); 5496d4211eb9SBarry Smith 54979566063dSJacob Faibussowitsch PetscCall(KSPSetPreSolve(snes->ksp,(PetscErrorCode (*)(KSP,Vec,Vec,void*))KSPPreSolve_SNESEW,snes)); 54989566063dSJacob Faibussowitsch PetscCall(KSPSetPostSolve(snes->ksp,(PetscErrorCode (*)(KSP,Vec,Vec,void*))KSPPostSolve_SNESEW,snes)); 5499a5c2985bSBarry Smith 55009566063dSJacob Faibussowitsch PetscCall(KSPMonitorSetFromOptions(snes->ksp, "-snes_monitor_ksp", "snes_preconditioned_residual", snes)); 55019566063dSJacob Faibussowitsch PetscCall(PetscObjectSetOptions((PetscObject)snes->ksp,((PetscObject)snes)->options)); 5502d4211eb9SBarry Smith } 5503d4211eb9SBarry Smith *ksp = snes->ksp; 550471f87433Sdalcinl PetscFunctionReturn(0); 550571f87433Sdalcinl } 55066c699258SBarry Smith 5507af0996ceSBarry Smith #include <petsc/private/dmimpl.h> 55086c699258SBarry Smith /*@ 55092a808120SBarry Smith SNESSetDM - Sets the DM that may be used by some nonlinear solvers or their underlying preconditioners 55106c699258SBarry Smith 55113f9fe445SBarry Smith Logically Collective on SNES 55126c699258SBarry Smith 55136c699258SBarry Smith Input Parameters: 55142a808120SBarry Smith + snes - the nonlinear solver context 55152a808120SBarry Smith - dm - the dm, cannot be NULL 55166c699258SBarry Smith 5517e03a659cSJed Brown Notes: 5518e03a659cSJed Brown A DM can only be used for solving one problem at a time because information about the problem is stored on the DM, 5519e03a659cSJed Brown even when not using interfaces like DMSNESSetFunction(). Use DMClone() to get a distinct DM when solving different 5520e03a659cSJed Brown problems using the same function space. 5521e03a659cSJed Brown 55226c699258SBarry Smith Level: intermediate 55236c699258SBarry Smith 5524db781477SPatrick Sanan .seealso: `SNESGetDM()`, `KSPSetDM()`, `KSPGetDM()` 55256c699258SBarry Smith @*/ 55267087cfbeSBarry Smith PetscErrorCode SNESSetDM(SNES snes,DM dm) 55276c699258SBarry Smith { 5528345fed2cSBarry Smith KSP ksp; 5529942e3340SBarry Smith DMSNES sdm; 55306c699258SBarry Smith 55316c699258SBarry Smith PetscFunctionBegin; 55320700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 55332a808120SBarry Smith PetscValidHeaderSpecific(dm,DM_CLASSID,2); 55349566063dSJacob Faibussowitsch PetscCall(PetscObjectReference((PetscObject)dm)); 5535942e3340SBarry Smith if (snes->dm) { /* Move the DMSNES context over to the new DM unless the new DM already has one */ 553651f4b3c7SToby Isaac if (snes->dm->dmsnes && !dm->dmsnes) { 55379566063dSJacob Faibussowitsch PetscCall(DMCopyDMSNES(snes->dm,dm)); 55389566063dSJacob Faibussowitsch PetscCall(DMGetDMSNES(snes->dm,&sdm)); 5539f5af7f23SKarl Rupp if (sdm->originaldm == snes->dm) sdm->originaldm = dm; /* Grant write privileges to the replacement DM */ 55406cab3a1bSJed Brown } 55419566063dSJacob Faibussowitsch PetscCall(DMCoarsenHookRemove(snes->dm,DMCoarsenHook_SNESVecSol,DMRestrictHook_SNESVecSol,snes)); 55429566063dSJacob Faibussowitsch PetscCall(DMDestroy(&snes->dm)); 55436cab3a1bSJed Brown } 55446c699258SBarry Smith snes->dm = dm; 5545116d1032SJed Brown snes->dmAuto = PETSC_FALSE; 5546f5af7f23SKarl Rupp 55479566063dSJacob Faibussowitsch PetscCall(SNESGetKSP(snes,&ksp)); 55489566063dSJacob Faibussowitsch PetscCall(KSPSetDM(ksp,dm)); 55499566063dSJacob Faibussowitsch PetscCall(KSPSetDMActive(ksp,PETSC_FALSE)); 5550efd4aadfSBarry Smith if (snes->npc) { 55519566063dSJacob Faibussowitsch PetscCall(SNESSetDM(snes->npc,snes->dm)); 55529566063dSJacob Faibussowitsch PetscCall(SNESSetNPCSide(snes,snes->npcside)); 55532c155ee1SBarry Smith } 55546c699258SBarry Smith PetscFunctionReturn(0); 55556c699258SBarry Smith } 55566c699258SBarry Smith 55576c699258SBarry Smith /*@ 55586c699258SBarry Smith SNESGetDM - Gets the DM that may be used by some preconditioners 55596c699258SBarry Smith 55603f9fe445SBarry Smith Not Collective but DM obtained is parallel on SNES 55616c699258SBarry Smith 55626c699258SBarry Smith Input Parameter: 55636c699258SBarry Smith . snes - the preconditioner context 55646c699258SBarry Smith 55656c699258SBarry Smith Output Parameter: 55666c699258SBarry Smith . dm - the dm 55676c699258SBarry Smith 55686c699258SBarry Smith Level: intermediate 55696c699258SBarry Smith 5570db781477SPatrick Sanan .seealso: `SNESSetDM()`, `KSPSetDM()`, `KSPGetDM()` 55716c699258SBarry Smith @*/ 55727087cfbeSBarry Smith PetscErrorCode SNESGetDM(SNES snes,DM *dm) 55736c699258SBarry Smith { 55746c699258SBarry Smith PetscFunctionBegin; 55750700a824SBarry Smith PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 55766cab3a1bSJed Brown if (!snes->dm) { 55779566063dSJacob Faibussowitsch PetscCall(DMShellCreate(PetscObjectComm((PetscObject)snes),&snes->dm)); 5578116d1032SJed Brown snes->dmAuto = PETSC_TRUE; 55796cab3a1bSJed Brown } 55806c699258SBarry Smith *dm = snes->dm; 55816c699258SBarry Smith PetscFunctionReturn(0); 55826c699258SBarry Smith } 55830807856dSBarry Smith 558431823bd8SMatthew G Knepley /*@ 5585be95d8f1SBarry Smith SNESSetNPC - Sets the nonlinear preconditioner to be used. 558631823bd8SMatthew G Knepley 558731823bd8SMatthew G Knepley Collective on SNES 558831823bd8SMatthew G Knepley 558931823bd8SMatthew G Knepley Input Parameters: 559031823bd8SMatthew G Knepley + snes - iterative context obtained from SNESCreate() 559131823bd8SMatthew G Knepley - pc - the preconditioner object 559231823bd8SMatthew G Knepley 559331823bd8SMatthew G Knepley Notes: 5594be95d8f1SBarry Smith Use SNESGetNPC() to retrieve the preconditioner context (for example, 559531823bd8SMatthew G Knepley to configure it using the API). 559631823bd8SMatthew G Knepley 559731823bd8SMatthew G Knepley Level: developer 559831823bd8SMatthew G Knepley 5599db781477SPatrick Sanan .seealso: `SNESGetNPC()`, `SNESHasNPC()` 560031823bd8SMatthew G Knepley @*/ 5601be95d8f1SBarry Smith PetscErrorCode SNESSetNPC(SNES snes, SNES pc) 560231823bd8SMatthew G Knepley { 560331823bd8SMatthew G Knepley PetscFunctionBegin; 560431823bd8SMatthew G Knepley PetscValidHeaderSpecific(snes, SNES_CLASSID, 1); 560531823bd8SMatthew G Knepley PetscValidHeaderSpecific(pc, SNES_CLASSID, 2); 560631823bd8SMatthew G Knepley PetscCheckSameComm(snes, 1, pc, 2); 56079566063dSJacob Faibussowitsch PetscCall(PetscObjectReference((PetscObject) pc)); 56089566063dSJacob Faibussowitsch PetscCall(SNESDestroy(&snes->npc)); 5609efd4aadfSBarry Smith snes->npc = pc; 56109566063dSJacob Faibussowitsch PetscCall(PetscLogObjectParent((PetscObject)snes, (PetscObject)snes->npc)); 561131823bd8SMatthew G Knepley PetscFunctionReturn(0); 561231823bd8SMatthew G Knepley } 561331823bd8SMatthew G Knepley 561431823bd8SMatthew G Knepley /*@ 5615be95d8f1SBarry Smith SNESGetNPC - Creates a nonlinear preconditioning solver (SNES) to be used to precondition the nonlinear solver. 561631823bd8SMatthew G Knepley 5617951fe5abSBarry Smith Not Collective; but any changes to the obtained SNES object must be applied collectively 561831823bd8SMatthew G Knepley 561931823bd8SMatthew G Knepley Input Parameter: 562031823bd8SMatthew G Knepley . snes - iterative context obtained from SNESCreate() 562131823bd8SMatthew G Knepley 562231823bd8SMatthew G Knepley Output Parameter: 562331823bd8SMatthew G Knepley . pc - preconditioner context 562431823bd8SMatthew G Knepley 5625b5badacbSBarry Smith Options Database: 5626b5badacbSBarry Smith . -npc_snes_type <type> - set the type of the SNES to use as the nonlinear preconditioner 5627b5badacbSBarry Smith 562895452b02SPatrick Sanan Notes: 5629b5badacbSBarry Smith If a SNES was previously set with SNESSetNPC() then that SNES is returned, otherwise a new SNES object is created. 5630be95d8f1SBarry Smith 5631951fe5abSBarry Smith The (preconditioner) SNES returned automatically inherits the same nonlinear function and Jacobian supplied to the original 5632951fe5abSBarry Smith SNES during SNESSetUp() 5633951fe5abSBarry Smith 563431823bd8SMatthew G Knepley Level: developer 563531823bd8SMatthew G Knepley 5636db781477SPatrick Sanan .seealso: `SNESSetNPC()`, `SNESHasNPC()`, `SNES`, `SNESCreate()` 563731823bd8SMatthew G Knepley @*/ 5638be95d8f1SBarry Smith PetscErrorCode SNESGetNPC(SNES snes, SNES *pc) 563931823bd8SMatthew G Knepley { 5640a64e098fSPeter Brune const char *optionsprefix; 564131823bd8SMatthew G Knepley 564231823bd8SMatthew G Knepley PetscFunctionBegin; 564331823bd8SMatthew G Knepley PetscValidHeaderSpecific(snes, SNES_CLASSID, 1); 564431823bd8SMatthew G Knepley PetscValidPointer(pc, 2); 5645efd4aadfSBarry Smith if (!snes->npc) { 56469566063dSJacob Faibussowitsch PetscCall(SNESCreate(PetscObjectComm((PetscObject)snes),&snes->npc)); 56479566063dSJacob Faibussowitsch PetscCall(PetscObjectIncrementTabLevel((PetscObject)snes->npc,(PetscObject)snes,1)); 56489566063dSJacob Faibussowitsch PetscCall(PetscLogObjectParent((PetscObject)snes,(PetscObject)snes->npc)); 56499566063dSJacob Faibussowitsch PetscCall(SNESGetOptionsPrefix(snes,&optionsprefix)); 56509566063dSJacob Faibussowitsch PetscCall(SNESSetOptionsPrefix(snes->npc,optionsprefix)); 56519566063dSJacob Faibussowitsch PetscCall(SNESAppendOptionsPrefix(snes->npc,"npc_")); 56529566063dSJacob Faibussowitsch PetscCall(SNESSetCountersReset(snes->npc,PETSC_FALSE)); 565331823bd8SMatthew G Knepley } 5654efd4aadfSBarry Smith *pc = snes->npc; 565531823bd8SMatthew G Knepley PetscFunctionReturn(0); 565631823bd8SMatthew G Knepley } 565731823bd8SMatthew G Knepley 56583ad1a0b9SPatrick Farrell /*@ 56593ad1a0b9SPatrick Farrell SNESHasNPC - Returns whether a nonlinear preconditioner exists 56603ad1a0b9SPatrick Farrell 56613ad1a0b9SPatrick Farrell Not Collective 56623ad1a0b9SPatrick Farrell 56633ad1a0b9SPatrick Farrell Input Parameter: 56643ad1a0b9SPatrick Farrell . snes - iterative context obtained from SNESCreate() 56653ad1a0b9SPatrick Farrell 56663ad1a0b9SPatrick Farrell Output Parameter: 56673ad1a0b9SPatrick Farrell . has_npc - whether the SNES has an NPC or not 56683ad1a0b9SPatrick Farrell 56693ad1a0b9SPatrick Farrell Level: developer 56703ad1a0b9SPatrick Farrell 5671db781477SPatrick Sanan .seealso: `SNESSetNPC()`, `SNESGetNPC()` 56723ad1a0b9SPatrick Farrell @*/ 56733ad1a0b9SPatrick Farrell PetscErrorCode SNESHasNPC(SNES snes, PetscBool *has_npc) 56743ad1a0b9SPatrick Farrell { 56753ad1a0b9SPatrick Farrell PetscFunctionBegin; 56763ad1a0b9SPatrick Farrell PetscValidHeaderSpecific(snes, SNES_CLASSID, 1); 5677efd4aadfSBarry Smith *has_npc = (PetscBool) (snes->npc ? PETSC_TRUE : PETSC_FALSE); 56783ad1a0b9SPatrick Farrell PetscFunctionReturn(0); 56793ad1a0b9SPatrick Farrell } 56803ad1a0b9SPatrick Farrell 5681c40d0f55SPeter Brune /*@ 5682be95d8f1SBarry Smith SNESSetNPCSide - Sets the preconditioning side. 5683c40d0f55SPeter Brune 5684c40d0f55SPeter Brune Logically Collective on SNES 5685c40d0f55SPeter Brune 5686c40d0f55SPeter Brune Input Parameter: 5687c40d0f55SPeter Brune . snes - iterative context obtained from SNESCreate() 5688c40d0f55SPeter Brune 5689c40d0f55SPeter Brune Output Parameter: 5690c40d0f55SPeter Brune . side - the preconditioning side, where side is one of 5691c40d0f55SPeter Brune .vb 56922d547940SBarry Smith PC_LEFT - left preconditioning 56932d547940SBarry Smith PC_RIGHT - right preconditioning (default for most nonlinear solvers) 5694c40d0f55SPeter Brune .ve 5695c40d0f55SPeter Brune 5696c40d0f55SPeter Brune Options Database Keys: 569767b8a455SSatish Balay . -snes_npc_side <right,left> - nonlinear preconditioner side 5698c40d0f55SPeter Brune 569995452b02SPatrick Sanan Notes: 570095452b02SPatrick Sanan SNESNRICHARDSON and SNESNCG only support left preconditioning. 57012d547940SBarry Smith 5702c40d0f55SPeter Brune Level: intermediate 5703c40d0f55SPeter Brune 5704db781477SPatrick Sanan .seealso: `SNESGetNPCSide()`, `KSPSetPCSide()` 5705c40d0f55SPeter Brune @*/ 5706be95d8f1SBarry Smith PetscErrorCode SNESSetNPCSide(SNES snes,PCSide side) 5707c40d0f55SPeter Brune { 5708c40d0f55SPeter Brune PetscFunctionBegin; 5709c40d0f55SPeter Brune PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 5710c40d0f55SPeter Brune PetscValidLogicalCollectiveEnum(snes,side,2); 5711b552625fSStefano Zampini if (side == PC_SIDE_DEFAULT) side = PC_RIGHT; 571254c59aa7SJacob Faibussowitsch PetscCheck((side == PC_LEFT) || (side == PC_RIGHT),PetscObjectComm((PetscObject)snes),PETSC_ERR_ARG_WRONG,"Only PC_LEFT and PC_RIGHT are supported"); 5713efd4aadfSBarry Smith snes->npcside = side; 5714c40d0f55SPeter Brune PetscFunctionReturn(0); 5715c40d0f55SPeter Brune } 5716c40d0f55SPeter Brune 5717c40d0f55SPeter Brune /*@ 5718be95d8f1SBarry Smith SNESGetNPCSide - Gets the preconditioning side. 5719c40d0f55SPeter Brune 5720c40d0f55SPeter Brune Not Collective 5721c40d0f55SPeter Brune 5722c40d0f55SPeter Brune Input Parameter: 5723c40d0f55SPeter Brune . snes - iterative context obtained from SNESCreate() 5724c40d0f55SPeter Brune 5725c40d0f55SPeter Brune Output Parameter: 5726c40d0f55SPeter Brune . side - the preconditioning side, where side is one of 5727c40d0f55SPeter Brune .vb 57282d547940SBarry Smith PC_LEFT - left preconditioning 57292d547940SBarry Smith PC_RIGHT - right preconditioning (default for most nonlinear solvers) 5730c40d0f55SPeter Brune .ve 5731c40d0f55SPeter Brune 5732c40d0f55SPeter Brune Level: intermediate 5733c40d0f55SPeter Brune 5734db781477SPatrick Sanan .seealso: `SNESSetNPCSide()`, `KSPGetPCSide()` 5735c40d0f55SPeter Brune @*/ 5736be95d8f1SBarry Smith PetscErrorCode SNESGetNPCSide(SNES snes,PCSide *side) 5737c40d0f55SPeter Brune { 5738c40d0f55SPeter Brune PetscFunctionBegin; 5739c40d0f55SPeter Brune PetscValidHeaderSpecific(snes,SNES_CLASSID,1); 5740c40d0f55SPeter Brune PetscValidPointer(side,2); 5741efd4aadfSBarry Smith *side = snes->npcside; 5742c40d0f55SPeter Brune PetscFunctionReturn(0); 5743c40d0f55SPeter Brune } 5744c40d0f55SPeter Brune 57459e764e56SPeter Brune /*@ 57467601faf0SJed Brown SNESSetLineSearch - Sets the linesearch on the SNES instance. 57479e764e56SPeter Brune 57489e764e56SPeter Brune Collective on SNES 57499e764e56SPeter Brune 57509e764e56SPeter Brune Input Parameters: 57519e764e56SPeter Brune + snes - iterative context obtained from SNESCreate() 57529e764e56SPeter Brune - linesearch - the linesearch object 57539e764e56SPeter Brune 57549e764e56SPeter Brune Notes: 57557601faf0SJed Brown Use SNESGetLineSearch() to retrieve the preconditioner context (for example, 57569e764e56SPeter Brune to configure it using the API). 57579e764e56SPeter Brune 57589e764e56SPeter Brune Level: developer 57599e764e56SPeter Brune 5760db781477SPatrick Sanan .seealso: `SNESGetLineSearch()` 57619e764e56SPeter Brune @*/ 57627601faf0SJed Brown PetscErrorCode SNESSetLineSearch(SNES snes, SNESLineSearch linesearch) 57639e764e56SPeter Brune { 57649e764e56SPeter Brune PetscFunctionBegin; 57659e764e56SPeter Brune PetscValidHeaderSpecific(snes, SNES_CLASSID, 1); 5766f1c6b773SPeter Brune PetscValidHeaderSpecific(linesearch, SNESLINESEARCH_CLASSID, 2); 57679e764e56SPeter Brune PetscCheckSameComm(snes, 1, linesearch, 2); 57689566063dSJacob Faibussowitsch PetscCall(PetscObjectReference((PetscObject) linesearch)); 57699566063dSJacob Faibussowitsch PetscCall(SNESLineSearchDestroy(&snes->linesearch)); 5770f5af7f23SKarl Rupp 57719e764e56SPeter Brune snes->linesearch = linesearch; 5772f5af7f23SKarl Rupp 57739566063dSJacob Faibussowitsch PetscCall(PetscLogObjectParent((PetscObject)snes, (PetscObject)snes->linesearch)); 57749e764e56SPeter Brune PetscFunctionReturn(0); 57759e764e56SPeter Brune } 57769e764e56SPeter Brune 5777a34ceb2aSJed Brown /*@ 57787601faf0SJed Brown SNESGetLineSearch - Returns a pointer to the line search context set with SNESSetLineSearch() 57798141a3b9SPeter Brune or creates a default line search instance associated with the SNES and returns it. 57809e764e56SPeter Brune 57819e764e56SPeter Brune Not Collective 57829e764e56SPeter Brune 57839e764e56SPeter Brune Input Parameter: 57849e764e56SPeter Brune . snes - iterative context obtained from SNESCreate() 57859e764e56SPeter Brune 57869e764e56SPeter Brune Output Parameter: 57879e764e56SPeter Brune . linesearch - linesearch context 57889e764e56SPeter Brune 5789162e0bf5SPeter Brune Level: beginner 57909e764e56SPeter Brune 5791db781477SPatrick Sanan .seealso: `SNESSetLineSearch()`, `SNESLineSearchCreate()` 57929e764e56SPeter Brune @*/ 57937601faf0SJed Brown PetscErrorCode SNESGetLineSearch(SNES snes, SNESLineSearch *linesearch) 57949e764e56SPeter Brune { 57959e764e56SPeter Brune const char *optionsprefix; 57969e764e56SPeter Brune 57979e764e56SPeter Brune PetscFunctionBegin; 57989e764e56SPeter Brune PetscValidHeaderSpecific(snes, SNES_CLASSID, 1); 57999e764e56SPeter Brune PetscValidPointer(linesearch, 2); 58009e764e56SPeter Brune if (!snes->linesearch) { 58019566063dSJacob Faibussowitsch PetscCall(SNESGetOptionsPrefix(snes, &optionsprefix)); 58029566063dSJacob Faibussowitsch PetscCall(SNESLineSearchCreate(PetscObjectComm((PetscObject)snes), &snes->linesearch)); 58039566063dSJacob Faibussowitsch PetscCall(SNESLineSearchSetSNES(snes->linesearch, snes)); 58049566063dSJacob Faibussowitsch PetscCall(SNESLineSearchAppendOptionsPrefix(snes->linesearch, optionsprefix)); 58059566063dSJacob Faibussowitsch PetscCall(PetscObjectIncrementTabLevel((PetscObject) snes->linesearch, (PetscObject) snes, 1)); 58069566063dSJacob Faibussowitsch PetscCall(PetscLogObjectParent((PetscObject)snes, (PetscObject)snes->linesearch)); 58079e764e56SPeter Brune } 58089e764e56SPeter Brune *linesearch = snes->linesearch; 58099e764e56SPeter Brune PetscFunctionReturn(0); 58109e764e56SPeter Brune } 5811