1a5eb4965SSatish Balay #ifdef PETSC_RCS_HEADER 2*15091d37SBarry Smith static char vcid[] = "$Id: snes.c,v 1.176 1999/03/15 02:45:30 curfman Exp bsmith $"; 39b94acceSBarry Smith #endif 49b94acceSBarry Smith 570f55243SBarry Smith #include "src/snes/snesimpl.h" /*I "snes.h" I*/ 6f5eb4b81SSatish Balay #include "src/sys/nreg.h" 79b94acceSBarry Smith 884cb2905SBarry Smith int SNESRegisterAllCalled = 0; 9488ecbafSBarry Smith FList SNESList = 0; 1082bf6240SBarry Smith 115615d1e5SSatish Balay #undef __FUNC__ 12d4bb536fSBarry Smith #define __FUNC__ "SNESView" 139b94acceSBarry Smith /*@ 149b94acceSBarry Smith SNESView - Prints the SNES data structure. 159b94acceSBarry Smith 16fee21e36SBarry Smith Collective on SNES, unless Viewer is VIEWER_STDOUT_SELF 17fee21e36SBarry Smith 18c7afd0dbSLois Curfman McInnes Input Parameters: 19c7afd0dbSLois Curfman McInnes + SNES - the SNES context 20c7afd0dbSLois Curfman McInnes - viewer - visualization context 21c7afd0dbSLois Curfman McInnes 229b94acceSBarry Smith Options Database Key: 23c8a8ba5cSLois Curfman McInnes . -snes_view - Calls SNESView() at end of SNESSolve() 249b94acceSBarry Smith 259b94acceSBarry Smith Notes: 269b94acceSBarry Smith The available visualization contexts include 27c7afd0dbSLois Curfman McInnes + VIEWER_STDOUT_SELF - standard output (default) 28c7afd0dbSLois Curfman McInnes - VIEWER_STDOUT_WORLD - synchronized standard 29c8a8ba5cSLois Curfman McInnes output where only the first processor opens 30c8a8ba5cSLois Curfman McInnes the file. All other processors send their 31c8a8ba5cSLois Curfman McInnes data to the first processor to print. 329b94acceSBarry Smith 333e081fefSLois Curfman McInnes The user can open an alternative visualization context with 3477ed5343SBarry Smith ViewerASCIIOpen() - output to a specified file. 359b94acceSBarry Smith 3636851e7fSLois Curfman McInnes Level: beginner 3736851e7fSLois Curfman McInnes 389b94acceSBarry Smith .keywords: SNES, view 399b94acceSBarry Smith 4077ed5343SBarry Smith .seealso: ViewerASCIIOpen() 419b94acceSBarry Smith @*/ 429b94acceSBarry Smith int SNESView(SNES snes,Viewer viewer) 439b94acceSBarry Smith { 449b94acceSBarry Smith SNES_KSP_EW_ConvCtx *kctx; 459b94acceSBarry Smith int ierr; 469b94acceSBarry Smith SLES sles; 479b94acceSBarry Smith char *method; 4819bcc07fSBarry Smith ViewerType vtype; 499b94acceSBarry Smith 503a40ed3dSBarry Smith PetscFunctionBegin; 5174679c65SBarry Smith PetscValidHeaderSpecific(snes,SNES_COOKIE); 5274679c65SBarry Smith if (viewer) {PetscValidHeader(viewer);} 5374679c65SBarry Smith else { viewer = VIEWER_STDOUT_SELF; } 5474679c65SBarry Smith 5519bcc07fSBarry Smith ierr = ViewerGetType(viewer,&vtype); CHKERRQ(ierr); 563f1db9ecSBarry Smith if (PetscTypeCompare(vtype,ASCII_VIEWER)) { 570ef38995SBarry Smith ViewerASCIIPrintf(viewer,"SNES Object:\n"); 5882bf6240SBarry Smith SNESGetType(snes,&method); 590ef38995SBarry Smith ViewerASCIIPrintf(viewer," method: %s\n",method); 600ef38995SBarry Smith if (snes->view) { 610ef38995SBarry Smith ierr = ViewerASCIIPushTab(viewer);CHKERRQ(ierr); 620ef38995SBarry Smith ierr = (*snes->view)(snes,viewer);CHKERRQ(ierr); 630ef38995SBarry Smith ierr = ViewerASCIIPopTab(viewer);CHKERRQ(ierr); 640ef38995SBarry Smith } 650ef38995SBarry Smith ViewerASCIIPrintf(viewer," maximum iterations=%d, maximum function evaluations=%d\n",snes->max_its,snes->max_funcs); 660ef38995SBarry Smith ViewerASCIIPrintf(viewer," tolerances: relative=%g, absolute=%g, truncation=%g, solution=%g\n", 679b94acceSBarry Smith snes->rtol, snes->atol, snes->trunctol, snes->xtol); 680ef38995SBarry Smith ViewerASCIIPrintf(viewer," total number of linear solver iterations=%d\n",snes->linear_its); 690ef38995SBarry Smith ViewerASCIIPrintf(viewer," total number of function evaluations=%d\n",snes->nfuncs); 700ef38995SBarry Smith if (snes->method_class == SNES_UNCONSTRAINED_MINIMIZATION) { 710ef38995SBarry Smith ViewerASCIIPrintf(viewer," min function tolerance=%g\n",snes->fmin); 720ef38995SBarry Smith } 739b94acceSBarry Smith if (snes->ksp_ewconv) { 749b94acceSBarry Smith kctx = (SNES_KSP_EW_ConvCtx *)snes->kspconvctx; 759b94acceSBarry Smith if (kctx) { 760ef38995SBarry Smith ViewerASCIIPrintf(viewer," Eisenstat-Walker computation of KSP relative tolerance (version %d)\n",kctx->version); 770ef38995SBarry Smith ViewerASCIIPrintf(viewer," rtol_0=%g, rtol_max=%g, threshold=%g\n",kctx->rtol_0,kctx->rtol_max,kctx->threshold); 780ef38995SBarry Smith ViewerASCIIPrintf(viewer," gamma=%g, alpha=%g, alpha2=%g\n",kctx->gamma,kctx->alpha,kctx->alpha2); 799b94acceSBarry Smith } 809b94acceSBarry Smith } 813f1db9ecSBarry Smith } else if (PetscTypeCompare(vtype,STRING_VIEWER)) { 820ef38995SBarry Smith ierr = SNESGetType(snes,&method);CHKERRQ(ierr); 830ef38995SBarry Smith ierr = ViewerStringSPrintf(viewer," %-3.3s",method);CHKERRQ(ierr); 8419bcc07fSBarry Smith } 8577ed5343SBarry Smith ierr = SNESGetSLES(snes,&sles);CHKERRQ(ierr); 860ef38995SBarry Smith ierr = ViewerASCIIPushTab(viewer);CHKERRQ(ierr); 879b94acceSBarry Smith ierr = SLESView(sles,viewer); CHKERRQ(ierr); 880ef38995SBarry Smith ierr = ViewerASCIIPopTab(viewer);CHKERRQ(ierr); 893a40ed3dSBarry Smith PetscFunctionReturn(0); 909b94acceSBarry Smith } 919b94acceSBarry Smith 92639f9d9dSBarry Smith /* 93639f9d9dSBarry Smith We retain a list of functions that also take SNES command 94639f9d9dSBarry Smith line options. These are called at the end SNESSetFromOptions() 95639f9d9dSBarry Smith */ 96639f9d9dSBarry Smith #define MAXSETFROMOPTIONS 5 97639f9d9dSBarry Smith static int numberofsetfromoptions; 98639f9d9dSBarry Smith static int (*othersetfromoptions[MAXSETFROMOPTIONS])(SNES); 99639f9d9dSBarry Smith 1005615d1e5SSatish Balay #undef __FUNC__ 101d4bb536fSBarry Smith #define __FUNC__ "SNESAddOptionsChecker" 102639f9d9dSBarry Smith /*@ 103639f9d9dSBarry Smith SNESAddOptionsChecker - Adds an additional function to check for SNES options. 104639f9d9dSBarry Smith 105c7afd0dbSLois Curfman McInnes Not Collective 106c7afd0dbSLois Curfman McInnes 107639f9d9dSBarry Smith Input Parameter: 108639f9d9dSBarry Smith . snescheck - function that checks for options 109639f9d9dSBarry Smith 11036851e7fSLois Curfman McInnes Level: developer 11136851e7fSLois Curfman McInnes 112639f9d9dSBarry Smith .seealso: SNESSetFromOptions() 113639f9d9dSBarry Smith @*/ 114639f9d9dSBarry Smith int SNESAddOptionsChecker(int (*snescheck)(SNES) ) 115639f9d9dSBarry Smith { 1163a40ed3dSBarry Smith PetscFunctionBegin; 117639f9d9dSBarry Smith if (numberofsetfromoptions >= MAXSETFROMOPTIONS) { 118a8c6a408SBarry Smith SETERRQ(PETSC_ERR_ARG_OUTOFRANGE,0,"Too many options checkers, only 5 allowed"); 119639f9d9dSBarry Smith } 120639f9d9dSBarry Smith 121639f9d9dSBarry Smith othersetfromoptions[numberofsetfromoptions++] = snescheck; 1223a40ed3dSBarry Smith PetscFunctionReturn(0); 123639f9d9dSBarry Smith } 124639f9d9dSBarry Smith 1255615d1e5SSatish Balay #undef __FUNC__ 126*15091d37SBarry Smith #define __FUNC__ "SNESSetTypeFromOptions" 127*15091d37SBarry Smith /*@ 128*15091d37SBarry Smith SNESSetTypeFromOptions - Sets the SNES solver type from the options database, 129*15091d37SBarry Smith or sets a default if none is give. 130*15091d37SBarry Smith 131*15091d37SBarry Smith Collective on SNES 132*15091d37SBarry Smith 133*15091d37SBarry Smith Input Parameter: 134*15091d37SBarry Smith . snes - the SNES context 135*15091d37SBarry Smith 136*15091d37SBarry Smith Options Database Keys: 137*15091d37SBarry Smith . -snes_type <type> - SNES_EQ_LS, SNES_EQ_TR, SNES_UM_TR, SNES_UM_LS etc 138*15091d37SBarry Smith 139*15091d37SBarry Smith Level: beginner 140*15091d37SBarry Smith 141*15091d37SBarry Smith .keywords: SNES, nonlinear, set, options, database 142*15091d37SBarry Smith 143*15091d37SBarry Smith .seealso: SNESPrintHelp(), SNESSetOptionsPrefix(), SNESSetFromOptions() 144*15091d37SBarry Smith @*/ 145*15091d37SBarry Smith int SNESSetTypeFromOptions(SNES snes) 146*15091d37SBarry Smith { 147*15091d37SBarry Smith char method[256]; 148*15091d37SBarry Smith int ierr, flg; 149*15091d37SBarry Smith 150*15091d37SBarry Smith PetscFunctionBegin; 151*15091d37SBarry Smith PetscValidHeaderSpecific(snes,SNES_COOKIE); 152*15091d37SBarry Smith if (snes->setupcalled) SETERRQ(PETSC_ERR_ARG_WRONGSTATE,0,"Must call prior to SNESSetUp()"); 153*15091d37SBarry Smith ierr = OptionsGetString(snes->prefix,"-snes_type",method,256,&flg); 154*15091d37SBarry Smith if (flg) { 155*15091d37SBarry Smith ierr = SNESSetType(snes,(SNESType) method); CHKERRQ(ierr); 156*15091d37SBarry Smith } 157*15091d37SBarry Smith /* 158*15091d37SBarry Smith If SNES type has not yet been set, set it now 159*15091d37SBarry Smith */ 160*15091d37SBarry Smith if (!snes->type_name) { 161*15091d37SBarry Smith if (snes->method_class == SNES_NONLINEAR_EQUATIONS) { 162*15091d37SBarry Smith ierr = SNESSetType(snes,SNES_EQ_LS); CHKERRQ(ierr); 163*15091d37SBarry Smith } else { 164*15091d37SBarry Smith ierr = SNESSetType(snes,SNES_UM_TR); CHKERRQ(ierr); 165*15091d37SBarry Smith } 166*15091d37SBarry Smith } 167*15091d37SBarry Smith PetscFunctionReturn(0); 168*15091d37SBarry Smith } 169*15091d37SBarry Smith 170*15091d37SBarry Smith #undef __FUNC__ 1715615d1e5SSatish Balay #define __FUNC__ "SNESSetFromOptions" 1729b94acceSBarry Smith /*@ 173682d7d0cSBarry Smith SNESSetFromOptions - Sets various SNES and SLES parameters from user options. 1749b94acceSBarry Smith 175c7afd0dbSLois Curfman McInnes Collective on SNES 176c7afd0dbSLois Curfman McInnes 1779b94acceSBarry Smith Input Parameter: 1789b94acceSBarry Smith . snes - the SNES context 1799b94acceSBarry Smith 18036851e7fSLois Curfman McInnes Options Database Keys: 181b39c3a46SLois Curfman McInnes + -snes_type <type> - SNES_EQ_LS, SNES_EQ_TR, SNES_UM_TR, SNES_UM_LS etc 18282738288SBarry Smith . -snes_stol - convergence tolerance in terms of the norm 18382738288SBarry Smith of the change in the solution between steps 184b39c3a46SLois Curfman McInnes . -snes_atol <atol> - absolute tolerance of residual norm 185b39c3a46SLois Curfman McInnes . -snes_rtol <rtol> - relative decrease in tolerance norm from initial 186b39c3a46SLois Curfman McInnes . -snes_max_it <max_it> - maximum number of iterations 187b39c3a46SLois Curfman McInnes . -snes_max_funcs <max_funcs> - maximum number of function evaluations 188b39c3a46SLois Curfman McInnes . -snes_trtol <trtol> - trust region tolerance 18982738288SBarry Smith . -snes_no_convergence_test - skip convergence test in nonlinear or minimization 19082738288SBarry Smith solver; hence iterations will continue until max_it 1911fbbfb26SLois Curfman McInnes or some other criterion is reached. Saves expense 19282738288SBarry Smith of convergence test 19382738288SBarry Smith . -snes_monitor - prints residual norm at each iteration 19482738288SBarry Smith . -snes_xmonitor - plots residual norm at each iteration 195e24b481bSBarry Smith . -snes_fd - use finite differences to compute Jacobian; very slow, only for testing 19636851e7fSLois Curfman McInnes - -snes_mf_ksp_monitor - if using matrix-free multiply then print h at each KSP iteration 19782738288SBarry Smith 19882738288SBarry Smith Options Database for Eisenstat-Walker method: 19982738288SBarry Smith + -snes_ksp_eq_conv - use Eisenstat-Walker method for determining linear system convergence 20082738288SBarry Smith . -snes_ksp_eq_version ver - version of Eisenstat-Walker method 20136851e7fSLois Curfman McInnes . -snes_ksp_ew_rtol0 <rtol0> - Sets rtol0 20236851e7fSLois Curfman McInnes . -snes_ksp_ew_rtolmax <rtolmax> - Sets rtolmax 20336851e7fSLois Curfman McInnes . -snes_ksp_ew_gamma <gamma> - Sets gamma 20436851e7fSLois Curfman McInnes . -snes_ksp_ew_alpha <alpha> - Sets alpha 20536851e7fSLois Curfman McInnes . -snes_ksp_ew_alpha2 <alpha2> - Sets alpha2 20636851e7fSLois Curfman McInnes - -snes_ksp_ew_threshold <threshold> - Sets threshold 20782738288SBarry Smith 20811ca99fdSLois Curfman McInnes Notes: 20911ca99fdSLois Curfman McInnes To see all options, run your program with the -help option or consult 21011ca99fdSLois Curfman McInnes the users manual. 21183e2fdc7SBarry Smith 21236851e7fSLois Curfman McInnes Level: beginner 21336851e7fSLois Curfman McInnes 2149b94acceSBarry Smith .keywords: SNES, nonlinear, set, options, database 2159b94acceSBarry Smith 216*15091d37SBarry Smith .seealso: SNESPrintHelp(), SNESSetOptionsPrefix(), SNESSetTypeFromOptions() 2179b94acceSBarry Smith @*/ 2189b94acceSBarry Smith int SNESSetFromOptions(SNES snes) 2199b94acceSBarry Smith { 2209b94acceSBarry Smith double tmp; 2219b94acceSBarry Smith SLES sles; 22281f57ec7SBarry Smith int ierr, flg,i,loc[4],nmax = 4; 2233c7409f5SSatish Balay int version = PETSC_DEFAULT; 2249b94acceSBarry Smith double rtol_0 = PETSC_DEFAULT; 2259b94acceSBarry Smith double rtol_max = PETSC_DEFAULT; 2269b94acceSBarry Smith double gamma2 = PETSC_DEFAULT; 2279b94acceSBarry Smith double alpha = PETSC_DEFAULT; 2289b94acceSBarry Smith double alpha2 = PETSC_DEFAULT; 2299b94acceSBarry Smith double threshold = PETSC_DEFAULT; 2309b94acceSBarry Smith 2313a40ed3dSBarry Smith PetscFunctionBegin; 23277c4ece6SBarry Smith PetscValidHeaderSpecific(snes,SNES_COOKIE); 233*15091d37SBarry Smith ierr = SNESSetTypeFromOptions(snes);CHKERRQ(ierr); 234ca161407SBarry Smith 235*15091d37SBarry Smith loc[0] = PETSC_DECIDE; loc[1] = PETSC_DECIDE; loc[2] = 300; loc[3] = 300; 2363c7409f5SSatish Balay ierr = OptionsGetDouble(snes->prefix,"-snes_stol",&tmp, &flg); CHKERRQ(ierr); 237d64ed03dSBarry Smith if (flg) { 238d64ed03dSBarry Smith ierr = SNESSetTolerances(snes,PETSC_DEFAULT,PETSC_DEFAULT,tmp,PETSC_DEFAULT,PETSC_DEFAULT);CHKERRQ(ierr); 239d64ed03dSBarry Smith } 2403c7409f5SSatish Balay ierr = OptionsGetDouble(snes->prefix,"-snes_atol",&tmp, &flg); CHKERRQ(ierr); 241d64ed03dSBarry Smith if (flg) { 242d64ed03dSBarry Smith ierr = SNESSetTolerances(snes,tmp,PETSC_DEFAULT,PETSC_DEFAULT,PETSC_DEFAULT,PETSC_DEFAULT);CHKERRQ(ierr); 243d64ed03dSBarry Smith } 2443c7409f5SSatish Balay ierr = OptionsGetDouble(snes->prefix,"-snes_rtol",&tmp, &flg); CHKERRQ(ierr); 245d64ed03dSBarry Smith if (flg) { 246d64ed03dSBarry Smith ierr = SNESSetTolerances(snes,PETSC_DEFAULT,tmp,PETSC_DEFAULT,PETSC_DEFAULT,PETSC_DEFAULT);CHKERRQ(ierr); 247d64ed03dSBarry Smith } 2483c7409f5SSatish Balay ierr = OptionsGetInt(snes->prefix,"-snes_max_it",&snes->max_its, &flg); CHKERRQ(ierr); 2493c7409f5SSatish Balay ierr = OptionsGetInt(snes->prefix,"-snes_max_funcs",&snes->max_funcs, &flg);CHKERRQ(ierr); 250d7a720efSLois Curfman McInnes ierr = OptionsGetDouble(snes->prefix,"-snes_trtol",&tmp, &flg); CHKERRQ(ierr); 251d64ed03dSBarry Smith if (flg) { ierr = SNESSetTrustRegionTolerance(snes,tmp); CHKERRQ(ierr); } 252d7a720efSLois Curfman McInnes ierr = OptionsGetDouble(snes->prefix,"-snes_fmin",&tmp, &flg); CHKERRQ(ierr); 253d64ed03dSBarry Smith if (flg) { ierr = SNESSetMinimizationFunctionTolerance(snes,tmp); CHKERRQ(ierr);} 2543c7409f5SSatish Balay ierr = OptionsHasName(snes->prefix,"-snes_ksp_ew_conv", &flg); CHKERRQ(ierr); 2553c7409f5SSatish Balay if (flg) { snes->ksp_ewconv = 1; } 256b18e04deSLois Curfman McInnes ierr = OptionsGetInt(snes->prefix,"-snes_ksp_ew_version",&version,&flg); CHKERRQ(ierr); 257b18e04deSLois Curfman McInnes ierr = OptionsGetDouble(snes->prefix,"-snes_ksp_ew_rtol0",&rtol_0,&flg); CHKERRQ(ierr); 258b18e04deSLois Curfman McInnes ierr = OptionsGetDouble(snes->prefix,"-snes_ksp_ew_rtolmax",&rtol_max,&flg);CHKERRQ(ierr); 259b18e04deSLois Curfman McInnes ierr = OptionsGetDouble(snes->prefix,"-snes_ksp_ew_gamma",&gamma2,&flg); CHKERRQ(ierr); 260b18e04deSLois Curfman McInnes ierr = OptionsGetDouble(snes->prefix,"-snes_ksp_ew_alpha",&alpha,&flg); CHKERRQ(ierr); 261b18e04deSLois Curfman McInnes ierr = OptionsGetDouble(snes->prefix,"-snes_ksp_ew_alpha2",&alpha2,&flg); CHKERRQ(ierr); 262b18e04deSLois Curfman McInnes ierr = OptionsGetDouble(snes->prefix,"-snes_ksp_ew_threshold",&threshold,&flg);CHKERRQ(ierr); 26393c39befSBarry Smith 26493c39befSBarry Smith ierr = OptionsHasName(snes->prefix,"-snes_no_convergence_test",&flg); CHKERRQ(ierr); 26593c39befSBarry Smith if (flg) {snes->converged = 0;} 26693c39befSBarry Smith 2679b94acceSBarry Smith ierr = SNES_KSP_SetParametersEW(snes,version,rtol_0,rtol_max,gamma2,alpha, 2689b94acceSBarry Smith alpha2,threshold); CHKERRQ(ierr); 2695bbad29bSBarry Smith ierr = OptionsHasName(snes->prefix,"-snes_cancelmonitors",&flg); CHKERRQ(ierr); 2705cd90555SBarry Smith if (flg) {ierr = SNESClearMonitor(snes);CHKERRQ(ierr);} 2713c7409f5SSatish Balay ierr = OptionsHasName(snes->prefix,"-snes_monitor",&flg); CHKERRQ(ierr); 272639f9d9dSBarry Smith if (flg) {ierr = SNESSetMonitor(snes,SNESDefaultMonitor,0);CHKERRQ(ierr);} 2733c7409f5SSatish Balay ierr = OptionsHasName(snes->prefix,"-snes_smonitor",&flg); CHKERRQ(ierr); 2743f1db9ecSBarry Smith if (flg) {ierr = SNESSetMonitor(snes,SNESDefaultSMonitor,0); CHKERRQ(ierr);} 2753f1db9ecSBarry Smith ierr = OptionsHasName(snes->prefix,"-snes_vecmonitor",&flg); CHKERRQ(ierr); 2763f1db9ecSBarry Smith if (flg) {ierr = SNESSetMonitor(snes,SNESVecViewMonitor,0); CHKERRQ(ierr);} 27781f57ec7SBarry Smith ierr = OptionsGetIntArray(snes->prefix,"-snes_xmonitor",loc,&nmax,&flg);CHKERRQ(ierr); 2783c7409f5SSatish Balay if (flg) { 27917699dbbSLois Curfman McInnes int rank = 0; 280d7e8b826SBarry Smith DrawLG lg; 28117699dbbSLois Curfman McInnes MPI_Initialized(&rank); 28217699dbbSLois Curfman McInnes if (rank) MPI_Comm_rank(snes->comm,&rank); 28317699dbbSLois Curfman McInnes if (!rank) { 28481f57ec7SBarry Smith ierr = SNESLGMonitorCreate(0,0,loc[0],loc[1],loc[2],loc[3],&lg); CHKERRQ(ierr); 2859b94acceSBarry Smith ierr = SNESSetMonitor(snes,SNESLGMonitor,(void *)lg); CHKERRQ(ierr); 286f8c826e1SBarry Smith snes->xmonitor = lg; 2879b94acceSBarry Smith PLogObjectParent(snes,lg); 2889b94acceSBarry Smith } 2899b94acceSBarry Smith } 290e24b481bSBarry Smith 2913c7409f5SSatish Balay ierr = OptionsHasName(snes->prefix,"-snes_fd", &flg); CHKERRQ(ierr); 2923c7409f5SSatish Balay if (flg && snes->method_class == SNES_NONLINEAR_EQUATIONS) { 2939b94acceSBarry Smith ierr = SNESSetJacobian(snes,snes->jacobian,snes->jacobian_pre, 2949b94acceSBarry Smith SNESDefaultComputeJacobian,snes->funP); CHKERRQ(ierr); 295981c4779SBarry Smith PLogInfo(snes,"SNESSetFromOptions: Setting default finite difference Jacobian matrix\n"); 296981c4779SBarry Smith } else if (flg && snes->method_class == SNES_UNCONSTRAINED_MINIMIZATION) { 29731615d04SLois Curfman McInnes ierr = SNESSetHessian(snes,snes->jacobian,snes->jacobian_pre, 29831615d04SLois Curfman McInnes SNESDefaultComputeHessian,snes->funP); CHKERRQ(ierr); 299d64ed03dSBarry Smith PLogInfo(snes,"SNESSetFromOptions: Setting default finite difference Hessian matrix\n"); 3009b94acceSBarry Smith } 301639f9d9dSBarry Smith 302639f9d9dSBarry Smith for ( i=0; i<numberofsetfromoptions; i++ ) { 303639f9d9dSBarry Smith ierr = (*othersetfromoptions[i])(snes); CHKERRQ(ierr); 304639f9d9dSBarry Smith } 305639f9d9dSBarry Smith 3069b94acceSBarry Smith ierr = SNESGetSLES(snes,&sles); CHKERRQ(ierr); 3079b94acceSBarry Smith ierr = SLESSetFromOptions(sles); CHKERRQ(ierr); 30893993e2dSLois Curfman McInnes 309e24b481bSBarry Smith /* set the special KSP monitor for matrix-free application */ 310e24b481bSBarry Smith ierr = OptionsHasName(snes->prefix,"-snes_mf_ksp_monitor",&flg); CHKERRQ(ierr); 311e24b481bSBarry Smith if (flg) { 312e24b481bSBarry Smith KSP ksp; 313e24b481bSBarry Smith ierr = SLESGetKSP(sles,&ksp);CHKERRQ(ierr); 314e24b481bSBarry Smith ierr = KSPSetMonitor(ksp,MatSNESFDMFKSPMonitor,PETSC_NULL);CHKERRQ(ierr); 315e24b481bSBarry Smith } 316e24b481bSBarry Smith 31782bf6240SBarry Smith ierr = OptionsHasName(PETSC_NULL,"-help", &flg); CHKERRQ(ierr); 31882bf6240SBarry Smith if (flg) { ierr = SNESPrintHelp(snes); CHKERRQ(ierr);} 31982bf6240SBarry Smith 3203a40ed3dSBarry Smith if (snes->setfromoptions) { 3213a40ed3dSBarry Smith ierr = (*snes->setfromoptions)(snes);CHKERRQ(ierr); 3223a40ed3dSBarry Smith } 3233a40ed3dSBarry Smith PetscFunctionReturn(0); 3249b94acceSBarry Smith } 3259b94acceSBarry Smith 326a847f771SSatish Balay 3275615d1e5SSatish Balay #undef __FUNC__ 328d4bb536fSBarry Smith #define __FUNC__ "SNESSetApplicationContext" 3299b94acceSBarry Smith /*@ 3309b94acceSBarry Smith SNESSetApplicationContext - Sets the optional user-defined context for 3319b94acceSBarry Smith the nonlinear solvers. 3329b94acceSBarry Smith 333fee21e36SBarry Smith Collective on SNES 334fee21e36SBarry Smith 335c7afd0dbSLois Curfman McInnes Input Parameters: 336c7afd0dbSLois Curfman McInnes + snes - the SNES context 337c7afd0dbSLois Curfman McInnes - usrP - optional user context 338c7afd0dbSLois Curfman McInnes 33936851e7fSLois Curfman McInnes Level: intermediate 34036851e7fSLois Curfman McInnes 3419b94acceSBarry Smith .keywords: SNES, nonlinear, set, application, context 3429b94acceSBarry Smith 3439b94acceSBarry Smith .seealso: SNESGetApplicationContext() 3449b94acceSBarry Smith @*/ 3459b94acceSBarry Smith int SNESSetApplicationContext(SNES snes,void *usrP) 3469b94acceSBarry Smith { 3473a40ed3dSBarry Smith PetscFunctionBegin; 34877c4ece6SBarry Smith PetscValidHeaderSpecific(snes,SNES_COOKIE); 3499b94acceSBarry Smith snes->user = usrP; 3503a40ed3dSBarry Smith PetscFunctionReturn(0); 3519b94acceSBarry Smith } 35274679c65SBarry Smith 3535615d1e5SSatish Balay #undef __FUNC__ 354d4bb536fSBarry Smith #define __FUNC__ "SNESGetApplicationContext" 3559b94acceSBarry Smith /*@C 3569b94acceSBarry Smith SNESGetApplicationContext - Gets the user-defined context for the 3579b94acceSBarry Smith nonlinear solvers. 3589b94acceSBarry Smith 359c7afd0dbSLois Curfman McInnes Not Collective 360c7afd0dbSLois Curfman McInnes 3619b94acceSBarry Smith Input Parameter: 3629b94acceSBarry Smith . snes - SNES context 3639b94acceSBarry Smith 3649b94acceSBarry Smith Output Parameter: 3659b94acceSBarry Smith . usrP - user context 3669b94acceSBarry Smith 36736851e7fSLois Curfman McInnes Level: intermediate 36836851e7fSLois Curfman McInnes 3699b94acceSBarry Smith .keywords: SNES, nonlinear, get, application, context 3709b94acceSBarry Smith 3719b94acceSBarry Smith .seealso: SNESSetApplicationContext() 3729b94acceSBarry Smith @*/ 3739b94acceSBarry Smith int SNESGetApplicationContext( SNES snes, void **usrP ) 3749b94acceSBarry Smith { 3753a40ed3dSBarry Smith PetscFunctionBegin; 37677c4ece6SBarry Smith PetscValidHeaderSpecific(snes,SNES_COOKIE); 3779b94acceSBarry Smith *usrP = snes->user; 3783a40ed3dSBarry Smith PetscFunctionReturn(0); 3799b94acceSBarry Smith } 38074679c65SBarry Smith 3815615d1e5SSatish Balay #undef __FUNC__ 382d4bb536fSBarry Smith #define __FUNC__ "SNESGetIterationNumber" 3839b94acceSBarry Smith /*@ 384c8228a4eSBarry Smith SNESGetIterationNumber - Gets the number of nonlinear iterations completed 385c8228a4eSBarry Smith at this time. 3869b94acceSBarry Smith 387c7afd0dbSLois Curfman McInnes Not Collective 388c7afd0dbSLois Curfman McInnes 3899b94acceSBarry Smith Input Parameter: 3909b94acceSBarry Smith . snes - SNES context 3919b94acceSBarry Smith 3929b94acceSBarry Smith Output Parameter: 3939b94acceSBarry Smith . iter - iteration number 3949b94acceSBarry Smith 395c8228a4eSBarry Smith Notes: 396c8228a4eSBarry Smith For example, during the computation of iteration 2 this would return 1. 397c8228a4eSBarry Smith 398c8228a4eSBarry Smith This is useful for using lagged Jacobians (where one does not recompute the 39908405cd6SLois Curfman McInnes Jacobian at each SNES iteration). For example, the code 40008405cd6SLois Curfman McInnes .vb 40108405cd6SLois Curfman McInnes ierr = SNESGetIterationNumber(snes,&it); 40208405cd6SLois Curfman McInnes if (!(it % 2)) { 40308405cd6SLois Curfman McInnes [compute Jacobian here] 40408405cd6SLois Curfman McInnes } 40508405cd6SLois Curfman McInnes .ve 406c8228a4eSBarry Smith can be used in your ComputeJacobian() function to cause the Jacobian to be 40708405cd6SLois Curfman McInnes recomputed every second SNES iteration. 408c8228a4eSBarry Smith 40936851e7fSLois Curfman McInnes Level: intermediate 41036851e7fSLois Curfman McInnes 4119b94acceSBarry Smith .keywords: SNES, nonlinear, get, iteration, number 4129b94acceSBarry Smith @*/ 4139b94acceSBarry Smith int SNESGetIterationNumber(SNES snes,int* iter) 4149b94acceSBarry Smith { 4153a40ed3dSBarry Smith PetscFunctionBegin; 41677c4ece6SBarry Smith PetscValidHeaderSpecific(snes,SNES_COOKIE); 41774679c65SBarry Smith PetscValidIntPointer(iter); 4189b94acceSBarry Smith *iter = snes->iter; 4193a40ed3dSBarry Smith PetscFunctionReturn(0); 4209b94acceSBarry Smith } 42174679c65SBarry Smith 4225615d1e5SSatish Balay #undef __FUNC__ 4235615d1e5SSatish Balay #define __FUNC__ "SNESGetFunctionNorm" 4249b94acceSBarry Smith /*@ 4259b94acceSBarry Smith SNESGetFunctionNorm - Gets the norm of the current function that was set 4269b94acceSBarry Smith with SNESSSetFunction(). 4279b94acceSBarry Smith 428c7afd0dbSLois Curfman McInnes Collective on SNES 429c7afd0dbSLois Curfman McInnes 4309b94acceSBarry Smith Input Parameter: 4319b94acceSBarry Smith . snes - SNES context 4329b94acceSBarry Smith 4339b94acceSBarry Smith Output Parameter: 4349b94acceSBarry Smith . fnorm - 2-norm of function 4359b94acceSBarry Smith 4369b94acceSBarry Smith Note: 4379b94acceSBarry Smith SNESGetFunctionNorm() is valid for SNES_NONLINEAR_EQUATIONS methods only. 438a86d99e1SLois Curfman McInnes A related routine for SNES_UNCONSTRAINED_MINIMIZATION methods is 439a86d99e1SLois Curfman McInnes SNESGetGradientNorm(). 4409b94acceSBarry Smith 44136851e7fSLois Curfman McInnes Level: intermediate 44236851e7fSLois Curfman McInnes 4439b94acceSBarry Smith .keywords: SNES, nonlinear, get, function, norm 444a86d99e1SLois Curfman McInnes 44508405cd6SLois Curfman McInnes .seealso: SNESGetFunction() 4469b94acceSBarry Smith @*/ 4479b94acceSBarry Smith int SNESGetFunctionNorm(SNES snes,Scalar *fnorm) 4489b94acceSBarry Smith { 4493a40ed3dSBarry Smith PetscFunctionBegin; 45077c4ece6SBarry Smith PetscValidHeaderSpecific(snes,SNES_COOKIE); 45174679c65SBarry Smith PetscValidScalarPointer(fnorm); 45274679c65SBarry Smith if (snes->method_class != SNES_NONLINEAR_EQUATIONS) { 453d252947aSBarry Smith SETERRQ(PETSC_ERR_ARG_WRONGSTATE,0,"For SNES_NONLINEAR_EQUATIONS only"); 45474679c65SBarry Smith } 4559b94acceSBarry Smith *fnorm = snes->norm; 4563a40ed3dSBarry Smith PetscFunctionReturn(0); 4579b94acceSBarry Smith } 45874679c65SBarry Smith 4595615d1e5SSatish Balay #undef __FUNC__ 4605615d1e5SSatish Balay #define __FUNC__ "SNESGetGradientNorm" 4619b94acceSBarry Smith /*@ 4629b94acceSBarry Smith SNESGetGradientNorm - Gets the norm of the current gradient that was set 4639b94acceSBarry Smith with SNESSSetGradient(). 4649b94acceSBarry Smith 465c7afd0dbSLois Curfman McInnes Collective on SNES 466c7afd0dbSLois Curfman McInnes 4679b94acceSBarry Smith Input Parameter: 4689b94acceSBarry Smith . snes - SNES context 4699b94acceSBarry Smith 4709b94acceSBarry Smith Output Parameter: 4719b94acceSBarry Smith . fnorm - 2-norm of gradient 4729b94acceSBarry Smith 4739b94acceSBarry Smith Note: 4749b94acceSBarry Smith SNESGetGradientNorm() is valid for SNES_UNCONSTRAINED_MINIMIZATION 475a86d99e1SLois Curfman McInnes methods only. A related routine for SNES_NONLINEAR_EQUATIONS methods 476a86d99e1SLois Curfman McInnes is SNESGetFunctionNorm(). 4779b94acceSBarry Smith 47836851e7fSLois Curfman McInnes Level: intermediate 47936851e7fSLois Curfman McInnes 4809b94acceSBarry Smith .keywords: SNES, nonlinear, get, gradient, norm 481a86d99e1SLois Curfman McInnes 482a86d99e1SLois Curfman McInnes .seelso: SNESSetGradient() 4839b94acceSBarry Smith @*/ 4849b94acceSBarry Smith int SNESGetGradientNorm(SNES snes,Scalar *gnorm) 4859b94acceSBarry Smith { 4863a40ed3dSBarry Smith PetscFunctionBegin; 48777c4ece6SBarry Smith PetscValidHeaderSpecific(snes,SNES_COOKIE); 48874679c65SBarry Smith PetscValidScalarPointer(gnorm); 48974679c65SBarry Smith if (snes->method_class != SNES_UNCONSTRAINED_MINIMIZATION) { 490d252947aSBarry Smith SETERRQ(PETSC_ERR_ARG_WRONGSTATE,0,"For SNES_UNCONSTRAINED_MINIMIZATION only"); 49174679c65SBarry Smith } 4929b94acceSBarry Smith *gnorm = snes->norm; 4933a40ed3dSBarry Smith PetscFunctionReturn(0); 4949b94acceSBarry Smith } 49574679c65SBarry Smith 4965615d1e5SSatish Balay #undef __FUNC__ 497d4bb536fSBarry Smith #define __FUNC__ "SNESGetNumberUnsuccessfulSteps" 4989b94acceSBarry Smith /*@ 4999b94acceSBarry Smith SNESGetNumberUnsuccessfulSteps - Gets the number of unsuccessful steps 5009b94acceSBarry Smith attempted by the nonlinear solver. 5019b94acceSBarry Smith 502c7afd0dbSLois Curfman McInnes Not Collective 503c7afd0dbSLois Curfman McInnes 5049b94acceSBarry Smith Input Parameter: 5059b94acceSBarry Smith . snes - SNES context 5069b94acceSBarry Smith 5079b94acceSBarry Smith Output Parameter: 5089b94acceSBarry Smith . nfails - number of unsuccessful steps attempted 5099b94acceSBarry Smith 510c96a6f78SLois Curfman McInnes Notes: 511c96a6f78SLois Curfman McInnes This counter is reset to zero for each successive call to SNESSolve(). 512c96a6f78SLois Curfman McInnes 51336851e7fSLois Curfman McInnes Level: intermediate 51436851e7fSLois Curfman McInnes 5159b94acceSBarry Smith .keywords: SNES, nonlinear, get, number, unsuccessful, steps 5169b94acceSBarry Smith @*/ 5179b94acceSBarry Smith int SNESGetNumberUnsuccessfulSteps(SNES snes,int* nfails) 5189b94acceSBarry Smith { 5193a40ed3dSBarry Smith PetscFunctionBegin; 52077c4ece6SBarry Smith PetscValidHeaderSpecific(snes,SNES_COOKIE); 52174679c65SBarry Smith PetscValidIntPointer(nfails); 5229b94acceSBarry Smith *nfails = snes->nfailures; 5233a40ed3dSBarry Smith PetscFunctionReturn(0); 5249b94acceSBarry Smith } 525a847f771SSatish Balay 5265615d1e5SSatish Balay #undef __FUNC__ 527d4bb536fSBarry Smith #define __FUNC__ "SNESGetNumberLinearIterations" 528c96a6f78SLois Curfman McInnes /*@ 529c96a6f78SLois Curfman McInnes SNESGetNumberLinearIterations - Gets the total number of linear iterations 530c96a6f78SLois Curfman McInnes used by the nonlinear solver. 531c96a6f78SLois Curfman McInnes 532c7afd0dbSLois Curfman McInnes Not Collective 533c7afd0dbSLois Curfman McInnes 534c96a6f78SLois Curfman McInnes Input Parameter: 535c96a6f78SLois Curfman McInnes . snes - SNES context 536c96a6f78SLois Curfman McInnes 537c96a6f78SLois Curfman McInnes Output Parameter: 538c96a6f78SLois Curfman McInnes . lits - number of linear iterations 539c96a6f78SLois Curfman McInnes 540c96a6f78SLois Curfman McInnes Notes: 541c96a6f78SLois Curfman McInnes This counter is reset to zero for each successive call to SNESSolve(). 542c96a6f78SLois Curfman McInnes 54336851e7fSLois Curfman McInnes Level: intermediate 54436851e7fSLois Curfman McInnes 545c96a6f78SLois Curfman McInnes .keywords: SNES, nonlinear, get, number, linear, iterations 546c96a6f78SLois Curfman McInnes @*/ 54786bddb7dSBarry Smith int SNESGetNumberLinearIterations(SNES snes,int* lits) 548c96a6f78SLois Curfman McInnes { 5493a40ed3dSBarry Smith PetscFunctionBegin; 550c96a6f78SLois Curfman McInnes PetscValidHeaderSpecific(snes,SNES_COOKIE); 551c96a6f78SLois Curfman McInnes PetscValidIntPointer(lits); 552c96a6f78SLois Curfman McInnes *lits = snes->linear_its; 5533a40ed3dSBarry Smith PetscFunctionReturn(0); 554c96a6f78SLois Curfman McInnes } 555c96a6f78SLois Curfman McInnes 556c96a6f78SLois Curfman McInnes #undef __FUNC__ 557d4bb536fSBarry Smith #define __FUNC__ "SNESGetSLES" 5589b94acceSBarry Smith /*@C 5599b94acceSBarry Smith SNESGetSLES - Returns the SLES context for a SNES solver. 5609b94acceSBarry Smith 561c7afd0dbSLois Curfman McInnes Not Collective, but if SNES object is parallel, then SLES object is parallel 562c7afd0dbSLois Curfman McInnes 5639b94acceSBarry Smith Input Parameter: 5649b94acceSBarry Smith . snes - the SNES context 5659b94acceSBarry Smith 5669b94acceSBarry Smith Output Parameter: 5679b94acceSBarry Smith . sles - the SLES context 5689b94acceSBarry Smith 5699b94acceSBarry Smith Notes: 5709b94acceSBarry Smith The user can then directly manipulate the SLES context to set various 5719b94acceSBarry Smith options, etc. Likewise, the user can then extract and manipulate the 5729b94acceSBarry Smith KSP and PC contexts as well. 5739b94acceSBarry Smith 57436851e7fSLois Curfman McInnes Level: beginner 57536851e7fSLois Curfman McInnes 5769b94acceSBarry Smith .keywords: SNES, nonlinear, get, SLES, context 5779b94acceSBarry Smith 5789b94acceSBarry Smith .seealso: SLESGetPC(), SLESGetKSP() 5799b94acceSBarry Smith @*/ 5809b94acceSBarry Smith int SNESGetSLES(SNES snes,SLES *sles) 5819b94acceSBarry Smith { 5823a40ed3dSBarry Smith PetscFunctionBegin; 58377c4ece6SBarry Smith PetscValidHeaderSpecific(snes,SNES_COOKIE); 5849b94acceSBarry Smith *sles = snes->sles; 5853a40ed3dSBarry Smith PetscFunctionReturn(0); 5869b94acceSBarry Smith } 58782bf6240SBarry Smith 588e24b481bSBarry Smith #undef __FUNC__ 589e24b481bSBarry Smith #define __FUNC__ "SNESPublish_Petsc" 590e24b481bSBarry Smith static int SNESPublish_Petsc(PetscObject object) 591e24b481bSBarry Smith { 592e24b481bSBarry Smith #if defined(HAVE_AMS) 593e24b481bSBarry Smith SNES v = (SNES) object; 594e24b481bSBarry Smith int ierr; 595e24b481bSBarry Smith 596e24b481bSBarry Smith PetscFunctionBegin; 597e24b481bSBarry Smith 598e24b481bSBarry Smith /* if it is already published then return */ 599e24b481bSBarry Smith if (v->amem >=0 ) PetscFunctionReturn(0); 600e24b481bSBarry Smith 6013f1db9ecSBarry Smith ierr = PetscObjectPublishBaseBegin(object);CHKERRQ(ierr); 602e24b481bSBarry Smith ierr = AMS_Memory_add_field((AMS_Memory)v->amem,"Iteration",&v->iter,1,AMS_INT,AMS_READ, 603e24b481bSBarry Smith AMS_COMMON,AMS_REDUCT_UNDEF);CHKERRQ(ierr); 604e24b481bSBarry Smith ierr = AMS_Memory_add_field((AMS_Memory)v->amem,"Residual",&v->norm,1,AMS_DOUBLE,AMS_READ, 605e24b481bSBarry Smith AMS_COMMON,AMS_REDUCT_UNDEF);CHKERRQ(ierr); 606e24b481bSBarry Smith ierr = PetscObjectPublishBaseEnd(object);CHKERRQ(ierr); 607e24b481bSBarry Smith #else 608e24b481bSBarry Smith PetscFunctionBegin; 609e24b481bSBarry Smith #endif 610e24b481bSBarry Smith PetscFunctionReturn(0); 611e24b481bSBarry Smith } 612e24b481bSBarry Smith 6139b94acceSBarry Smith /* -----------------------------------------------------------*/ 6145615d1e5SSatish Balay #undef __FUNC__ 6155615d1e5SSatish Balay #define __FUNC__ "SNESCreate" 6169b94acceSBarry Smith /*@C 6179b94acceSBarry Smith SNESCreate - Creates a nonlinear solver context. 6189b94acceSBarry Smith 619c7afd0dbSLois Curfman McInnes Collective on MPI_Comm 620c7afd0dbSLois Curfman McInnes 621c7afd0dbSLois Curfman McInnes Input Parameters: 622c7afd0dbSLois Curfman McInnes + comm - MPI communicator 623c7afd0dbSLois Curfman McInnes - type - type of method, either 624c7afd0dbSLois Curfman McInnes SNES_NONLINEAR_EQUATIONS (for systems of nonlinear equations) 625c7afd0dbSLois Curfman McInnes or SNES_UNCONSTRAINED_MINIMIZATION (for unconstrained minimization) 6269b94acceSBarry Smith 6279b94acceSBarry Smith Output Parameter: 6289b94acceSBarry Smith . outsnes - the new SNES context 6299b94acceSBarry Smith 630c7afd0dbSLois Curfman McInnes Options Database Keys: 631c7afd0dbSLois Curfman McInnes + -snes_mf - Activates default matrix-free Jacobian-vector products, 632c7afd0dbSLois Curfman McInnes and no preconditioning matrix 633c7afd0dbSLois Curfman McInnes . -snes_mf_operator - Activates default matrix-free Jacobian-vector 634c7afd0dbSLois Curfman McInnes products, and a user-provided preconditioning matrix 635c7afd0dbSLois Curfman McInnes as set by SNESSetJacobian() 636c7afd0dbSLois Curfman McInnes - -snes_fd - Uses (slow!) finite differences to compute Jacobian 637c1f60f51SBarry Smith 63836851e7fSLois Curfman McInnes Level: beginner 63936851e7fSLois Curfman McInnes 6409b94acceSBarry Smith .keywords: SNES, nonlinear, create, context 6419b94acceSBarry Smith 64263a78c88SLois Curfman McInnes .seealso: SNESSolve(), SNESDestroy() 6439b94acceSBarry Smith @*/ 6444b0e389bSBarry Smith int SNESCreate(MPI_Comm comm,SNESProblemType type,SNES *outsnes) 6459b94acceSBarry Smith { 6469b94acceSBarry Smith int ierr; 6479b94acceSBarry Smith SNES snes; 6489b94acceSBarry Smith SNES_KSP_EW_ConvCtx *kctx; 64937fcc0dbSBarry Smith 6503a40ed3dSBarry Smith PetscFunctionBegin; 6519b94acceSBarry Smith *outsnes = 0; 652d64ed03dSBarry Smith if (type != SNES_UNCONSTRAINED_MINIMIZATION && type != SNES_NONLINEAR_EQUATIONS){ 653d252947aSBarry Smith SETERRQ(PETSC_ERR_ARG_OUTOFRANGE,0,"incorrect method type"); 654d64ed03dSBarry Smith } 6553f1db9ecSBarry Smith PetscHeaderCreate(snes,_p_SNES,int,SNES_COOKIE,0,"SNES",comm,SNESDestroy,SNESView); 6569b94acceSBarry Smith PLogObjectCreate(snes); 657e24b481bSBarry Smith snes->bops->publish = SNESPublish_Petsc; 6589b94acceSBarry Smith snes->max_its = 50; 6599750a799SBarry Smith snes->max_funcs = 10000; 6609b94acceSBarry Smith snes->norm = 0.0; 6615a2d0531SBarry Smith if (type == SNES_UNCONSTRAINED_MINIMIZATION) { 662b18e04deSLois Curfman McInnes snes->rtol = 1.e-8; 663b18e04deSLois Curfman McInnes snes->ttol = 0.0; 6649b94acceSBarry Smith snes->atol = 1.e-10; 665d64ed03dSBarry Smith } else { 666b4874afaSBarry Smith snes->rtol = 1.e-8; 667b4874afaSBarry Smith snes->ttol = 0.0; 668b4874afaSBarry Smith snes->atol = 1.e-50; 669b4874afaSBarry Smith } 6709b94acceSBarry Smith snes->xtol = 1.e-8; 671b18e04deSLois Curfman McInnes snes->trunctol = 1.e-12; /* no longer used */ 6729b94acceSBarry Smith snes->nfuncs = 0; 6739b94acceSBarry Smith snes->nfailures = 0; 6747a00f4a9SLois Curfman McInnes snes->linear_its = 0; 675639f9d9dSBarry Smith snes->numbermonitors = 0; 6769b94acceSBarry Smith snes->data = 0; 6779b94acceSBarry Smith snes->view = 0; 6789b94acceSBarry Smith snes->computeumfunction = 0; 6799b94acceSBarry Smith snes->umfunP = 0; 6809b94acceSBarry Smith snes->fc = 0; 6819b94acceSBarry Smith snes->deltatol = 1.e-12; 6829b94acceSBarry Smith snes->fmin = -1.e30; 6839b94acceSBarry Smith snes->method_class = type; 6849b94acceSBarry Smith snes->set_method_called = 0; 68582bf6240SBarry Smith snes->setupcalled = 0; 6869b94acceSBarry Smith snes->ksp_ewconv = 0; 6876f24a144SLois Curfman McInnes snes->xmonitor = 0; 6886f24a144SLois Curfman McInnes snes->vwork = 0; 6896f24a144SLois Curfman McInnes snes->nwork = 0; 690758f92a0SBarry Smith snes->conv_hist_len = 0; 691758f92a0SBarry Smith snes->conv_hist_max = 0; 692758f92a0SBarry Smith snes->conv_hist = PETSC_NULL; 693758f92a0SBarry Smith snes->conv_hist_its = PETSC_NULL; 694758f92a0SBarry Smith snes->conv_hist_reset = PETSC_TRUE; 6959b94acceSBarry Smith 6969b94acceSBarry Smith /* Create context to compute Eisenstat-Walker relative tolerance for KSP */ 6970452661fSBarry Smith kctx = PetscNew(SNES_KSP_EW_ConvCtx); CHKPTRQ(kctx); 698eed86810SBarry Smith PLogObjectMemory(snes,sizeof(SNES_KSP_EW_ConvCtx)); 6999b94acceSBarry Smith snes->kspconvctx = (void*)kctx; 7009b94acceSBarry Smith kctx->version = 2; 7019b94acceSBarry Smith kctx->rtol_0 = .3; /* Eisenstat and Walker suggest rtol_0=.5, but 7029b94acceSBarry Smith this was too large for some test cases */ 7039b94acceSBarry Smith kctx->rtol_last = 0; 7049b94acceSBarry Smith kctx->rtol_max = .9; 7059b94acceSBarry Smith kctx->gamma = 1.0; 7069b94acceSBarry Smith kctx->alpha2 = .5*(1.0 + sqrt(5.0)); 7079b94acceSBarry Smith kctx->alpha = kctx->alpha2; 7089b94acceSBarry Smith kctx->threshold = .1; 7099b94acceSBarry Smith kctx->lresid_last = 0; 7109b94acceSBarry Smith kctx->norm_last = 0; 7119b94acceSBarry Smith 7129b94acceSBarry Smith ierr = SLESCreate(comm,&snes->sles); CHKERRQ(ierr); 7139b94acceSBarry Smith PLogObjectParent(snes,snes->sles) 7145334005bSBarry Smith 7159b94acceSBarry Smith *outsnes = snes; 716e24b481bSBarry Smith PetscPublishAll(snes); 7173a40ed3dSBarry Smith PetscFunctionReturn(0); 7189b94acceSBarry Smith } 7199b94acceSBarry Smith 7209b94acceSBarry Smith /* --------------------------------------------------------------- */ 7215615d1e5SSatish Balay #undef __FUNC__ 722d4bb536fSBarry Smith #define __FUNC__ "SNESSetFunction" 7239b94acceSBarry Smith /*@C 7249b94acceSBarry Smith SNESSetFunction - Sets the function evaluation routine and function 7259b94acceSBarry Smith vector for use by the SNES routines in solving systems of nonlinear 7269b94acceSBarry Smith equations. 7279b94acceSBarry Smith 728fee21e36SBarry Smith Collective on SNES 729fee21e36SBarry Smith 730c7afd0dbSLois Curfman McInnes Input Parameters: 731c7afd0dbSLois Curfman McInnes + snes - the SNES context 732c7afd0dbSLois Curfman McInnes . func - function evaluation routine 733c7afd0dbSLois Curfman McInnes . r - vector to store function value 734c7afd0dbSLois Curfman McInnes - ctx - [optional] user-defined context for private data for the 735c7afd0dbSLois Curfman McInnes function evaluation routine (may be PETSC_NULL) 7369b94acceSBarry Smith 737c7afd0dbSLois Curfman McInnes Calling sequence of func: 7388d76a1e5SLois Curfman McInnes $ func (SNES snes,Vec x,Vec f,void *ctx); 739c7afd0dbSLois Curfman McInnes 740313e4042SLois Curfman McInnes . f - function vector 741c7afd0dbSLois Curfman McInnes - ctx - optional user-defined function context 7429b94acceSBarry Smith 7439b94acceSBarry Smith Notes: 7449b94acceSBarry Smith The Newton-like methods typically solve linear systems of the form 7459b94acceSBarry Smith $ f'(x) x = -f(x), 746c7afd0dbSLois Curfman McInnes where f'(x) denotes the Jacobian matrix and f(x) is the function. 7479b94acceSBarry Smith 7489b94acceSBarry Smith SNESSetFunction() is valid for SNES_NONLINEAR_EQUATIONS methods only. 7499b94acceSBarry Smith Analogous routines for SNES_UNCONSTRAINED_MINIMIZATION methods are 7509b94acceSBarry Smith SNESSetMinimizationFunction() and SNESSetGradient(); 7519b94acceSBarry Smith 75236851e7fSLois Curfman McInnes Level: beginner 75336851e7fSLois Curfman McInnes 7549b94acceSBarry Smith .keywords: SNES, nonlinear, set, function 7559b94acceSBarry Smith 756a86d99e1SLois Curfman McInnes .seealso: SNESGetFunction(), SNESComputeFunction(), SNESSetJacobian() 7579b94acceSBarry Smith @*/ 7585334005bSBarry Smith int SNESSetFunction( SNES snes, Vec r, int (*func)(SNES,Vec,Vec,void*),void *ctx) 7599b94acceSBarry Smith { 7603a40ed3dSBarry Smith PetscFunctionBegin; 76177c4ece6SBarry Smith PetscValidHeaderSpecific(snes,SNES_COOKIE); 762a8c6a408SBarry Smith if (snes->method_class != SNES_NONLINEAR_EQUATIONS) { 763a8c6a408SBarry Smith SETERRQ(PETSC_ERR_ARG_WRONG,0,"For SNES_NONLINEAR_EQUATIONS only"); 764a8c6a408SBarry Smith } 7659b94acceSBarry Smith snes->computefunction = func; 7669b94acceSBarry Smith snes->vec_func = snes->vec_func_always = r; 7679b94acceSBarry Smith snes->funP = ctx; 7683a40ed3dSBarry Smith PetscFunctionReturn(0); 7699b94acceSBarry Smith } 7709b94acceSBarry Smith 7715615d1e5SSatish Balay #undef __FUNC__ 7725615d1e5SSatish Balay #define __FUNC__ "SNESComputeFunction" 7739b94acceSBarry Smith /*@ 77436851e7fSLois Curfman McInnes SNESComputeFunction - Calls the function that has been set with 7759b94acceSBarry Smith SNESSetFunction(). 7769b94acceSBarry Smith 777c7afd0dbSLois Curfman McInnes Collective on SNES 778c7afd0dbSLois Curfman McInnes 7799b94acceSBarry Smith Input Parameters: 780c7afd0dbSLois Curfman McInnes + snes - the SNES context 781c7afd0dbSLois Curfman McInnes - x - input vector 7829b94acceSBarry Smith 7839b94acceSBarry Smith Output Parameter: 7843638b69dSLois Curfman McInnes . y - function vector, as set by SNESSetFunction() 7859b94acceSBarry Smith 7861bffabb2SLois Curfman McInnes Notes: 7879b94acceSBarry Smith SNESComputeFunction() is valid for SNES_NONLINEAR_EQUATIONS methods only. 7889b94acceSBarry Smith Analogous routines for SNES_UNCONSTRAINED_MINIMIZATION methods are 7899b94acceSBarry Smith SNESComputeMinimizationFunction() and SNESComputeGradient(); 7909b94acceSBarry Smith 79136851e7fSLois Curfman McInnes SNESComputeFunction() is typically used within nonlinear solvers 79236851e7fSLois Curfman McInnes implementations, so most users would not generally call this routine 79336851e7fSLois Curfman McInnes themselves. 79436851e7fSLois Curfman McInnes 79536851e7fSLois Curfman McInnes Level: developer 79636851e7fSLois Curfman McInnes 7979b94acceSBarry Smith .keywords: SNES, nonlinear, compute, function 7989b94acceSBarry Smith 799a86d99e1SLois Curfman McInnes .seealso: SNESSetFunction(), SNESGetFunction() 8009b94acceSBarry Smith @*/ 8019b94acceSBarry Smith int SNESComputeFunction(SNES snes,Vec x, Vec y) 8029b94acceSBarry Smith { 8039b94acceSBarry Smith int ierr; 8049b94acceSBarry Smith 8053a40ed3dSBarry Smith PetscFunctionBegin; 806d4bb536fSBarry Smith if (snes->method_class != SNES_NONLINEAR_EQUATIONS) { 807a8c6a408SBarry Smith SETERRQ(PETSC_ERR_ARG_WRONG,0,"For SNES_NONLINEAR_EQUATIONS only"); 808d4bb536fSBarry Smith } 8099b94acceSBarry Smith PLogEventBegin(SNES_FunctionEval,snes,x,y,0); 810d64ed03dSBarry Smith PetscStackPush("SNES user function"); 8119b94acceSBarry Smith ierr = (*snes->computefunction)(snes,x,y,snes->funP); CHKERRQ(ierr); 812d64ed03dSBarry Smith PetscStackPop; 813ae3c334cSLois Curfman McInnes snes->nfuncs++; 8149b94acceSBarry Smith PLogEventEnd(SNES_FunctionEval,snes,x,y,0); 8153a40ed3dSBarry Smith PetscFunctionReturn(0); 8169b94acceSBarry Smith } 8179b94acceSBarry Smith 8185615d1e5SSatish Balay #undef __FUNC__ 819d4bb536fSBarry Smith #define __FUNC__ "SNESSetMinimizationFunction" 8209b94acceSBarry Smith /*@C 8219b94acceSBarry Smith SNESSetMinimizationFunction - Sets the function evaluation routine for 8229b94acceSBarry Smith unconstrained minimization. 8239b94acceSBarry Smith 824fee21e36SBarry Smith Collective on SNES 825fee21e36SBarry Smith 826c7afd0dbSLois Curfman McInnes Input Parameters: 827c7afd0dbSLois Curfman McInnes + snes - the SNES context 828c7afd0dbSLois Curfman McInnes . func - function evaluation routine 829c7afd0dbSLois Curfman McInnes - ctx - [optional] user-defined context for private data for the 830c7afd0dbSLois Curfman McInnes function evaluation routine (may be PETSC_NULL) 8319b94acceSBarry Smith 832c7afd0dbSLois Curfman McInnes Calling sequence of func: 8338d76a1e5SLois Curfman McInnes $ func (SNES snes,Vec x,double *f,void *ctx); 834c7afd0dbSLois Curfman McInnes 835c7afd0dbSLois Curfman McInnes + x - input vector 8369b94acceSBarry Smith . f - function 837c7afd0dbSLois Curfman McInnes - ctx - [optional] user-defined function context 8389b94acceSBarry Smith 83936851e7fSLois Curfman McInnes Level: beginner 84036851e7fSLois Curfman McInnes 8419b94acceSBarry Smith Notes: 8429b94acceSBarry Smith SNESSetMinimizationFunction() is valid for SNES_UNCONSTRAINED_MINIMIZATION 8439b94acceSBarry Smith methods only. An analogous routine for SNES_NONLINEAR_EQUATIONS methods is 8449b94acceSBarry Smith SNESSetFunction(). 8459b94acceSBarry Smith 8469b94acceSBarry Smith .keywords: SNES, nonlinear, set, minimization, function 8479b94acceSBarry Smith 848a86d99e1SLois Curfman McInnes .seealso: SNESGetMinimizationFunction(), SNESComputeMinimizationFunction(), 849a86d99e1SLois Curfman McInnes SNESSetHessian(), SNESSetGradient() 8509b94acceSBarry Smith @*/ 8519b94acceSBarry Smith int SNESSetMinimizationFunction(SNES snes,int (*func)(SNES,Vec,double*,void*), 8529b94acceSBarry Smith void *ctx) 8539b94acceSBarry Smith { 8543a40ed3dSBarry Smith PetscFunctionBegin; 85577c4ece6SBarry Smith PetscValidHeaderSpecific(snes,SNES_COOKIE); 856a8c6a408SBarry Smith if (snes->method_class != SNES_UNCONSTRAINED_MINIMIZATION) { 857a8c6a408SBarry Smith SETERRQ(PETSC_ERR_ARG_WRONG,0,"Only for SNES_UNCONSTRAINED_MINIMIZATION"); 858a8c6a408SBarry Smith } 8599b94acceSBarry Smith snes->computeumfunction = func; 8609b94acceSBarry Smith snes->umfunP = ctx; 8613a40ed3dSBarry Smith PetscFunctionReturn(0); 8629b94acceSBarry Smith } 8639b94acceSBarry Smith 8645615d1e5SSatish Balay #undef __FUNC__ 8655615d1e5SSatish Balay #define __FUNC__ "SNESComputeMinimizationFunction" 8669b94acceSBarry Smith /*@ 8679b94acceSBarry Smith SNESComputeMinimizationFunction - Computes the function that has been 8689b94acceSBarry Smith set with SNESSetMinimizationFunction(). 8699b94acceSBarry Smith 870c7afd0dbSLois Curfman McInnes Collective on SNES 871c7afd0dbSLois Curfman McInnes 8729b94acceSBarry Smith Input Parameters: 873c7afd0dbSLois Curfman McInnes + snes - the SNES context 874c7afd0dbSLois Curfman McInnes - x - input vector 8759b94acceSBarry Smith 8769b94acceSBarry Smith Output Parameter: 8779b94acceSBarry Smith . y - function value 8789b94acceSBarry Smith 8799b94acceSBarry Smith Notes: 8809b94acceSBarry Smith SNESComputeMinimizationFunction() is valid only for 8819b94acceSBarry Smith SNES_UNCONSTRAINED_MINIMIZATION methods. An analogous routine for 8829b94acceSBarry Smith SNES_NONLINEAR_EQUATIONS methods is SNESComputeFunction(). 883a86d99e1SLois Curfman McInnes 88436851e7fSLois Curfman McInnes SNESComputeMinimizationFunction() is typically used within minimization 88536851e7fSLois Curfman McInnes implementations, so most users would not generally call this routine 88636851e7fSLois Curfman McInnes themselves. 88736851e7fSLois Curfman McInnes 88836851e7fSLois Curfman McInnes Level: developer 88936851e7fSLois Curfman McInnes 890a86d99e1SLois Curfman McInnes .keywords: SNES, nonlinear, compute, minimization, function 891a86d99e1SLois Curfman McInnes 892a86d99e1SLois Curfman McInnes .seealso: SNESSetMinimizationFunction(), SNESGetMinimizationFunction(), 893a86d99e1SLois Curfman McInnes SNESComputeGradient(), SNESComputeHessian() 8949b94acceSBarry Smith @*/ 8959b94acceSBarry Smith int SNESComputeMinimizationFunction(SNES snes,Vec x,double *y) 8969b94acceSBarry Smith { 8979b94acceSBarry Smith int ierr; 8983a40ed3dSBarry Smith 8993a40ed3dSBarry Smith PetscFunctionBegin; 900a8c6a408SBarry Smith if (snes->method_class != SNES_UNCONSTRAINED_MINIMIZATION) { 901a8c6a408SBarry Smith SETERRQ(PETSC_ERR_ARG_WRONG,0,"Only for SNES_UNCONSTRAINED_MINIMIZATION"); 902a8c6a408SBarry Smith } 9039b94acceSBarry Smith PLogEventBegin(SNES_MinimizationFunctionEval,snes,x,y,0); 904d64ed03dSBarry Smith PetscStackPush("SNES user minimzation function"); 9059b94acceSBarry Smith ierr = (*snes->computeumfunction)(snes,x,y,snes->umfunP); CHKERRQ(ierr); 906d64ed03dSBarry Smith PetscStackPop; 907ae3c334cSLois Curfman McInnes snes->nfuncs++; 9089b94acceSBarry Smith PLogEventEnd(SNES_MinimizationFunctionEval,snes,x,y,0); 9093a40ed3dSBarry Smith PetscFunctionReturn(0); 9109b94acceSBarry Smith } 9119b94acceSBarry Smith 9125615d1e5SSatish Balay #undef __FUNC__ 913d4bb536fSBarry Smith #define __FUNC__ "SNESSetGradient" 9149b94acceSBarry Smith /*@C 9159b94acceSBarry Smith SNESSetGradient - Sets the gradient evaluation routine and gradient 9169b94acceSBarry Smith vector for use by the SNES routines. 9179b94acceSBarry Smith 918c7afd0dbSLois Curfman McInnes Collective on SNES 919c7afd0dbSLois Curfman McInnes 9209b94acceSBarry Smith Input Parameters: 921c7afd0dbSLois Curfman McInnes + snes - the SNES context 9229b94acceSBarry Smith . func - function evaluation routine 923044dda88SLois Curfman McInnes . ctx - optional user-defined context for private data for the 924044dda88SLois Curfman McInnes gradient evaluation routine (may be PETSC_NULL) 925c7afd0dbSLois Curfman McInnes - r - vector to store gradient value 926fee21e36SBarry Smith 9279b94acceSBarry Smith Calling sequence of func: 9288d76a1e5SLois Curfman McInnes $ func (SNES, Vec x, Vec g, void *ctx); 9299b94acceSBarry Smith 930c7afd0dbSLois Curfman McInnes + x - input vector 9319b94acceSBarry Smith . g - gradient vector 932c7afd0dbSLois Curfman McInnes - ctx - optional user-defined gradient context 9339b94acceSBarry Smith 9349b94acceSBarry Smith Notes: 9359b94acceSBarry Smith SNESSetMinimizationFunction() is valid for SNES_UNCONSTRAINED_MINIMIZATION 9369b94acceSBarry Smith methods only. An analogous routine for SNES_NONLINEAR_EQUATIONS methods is 9379b94acceSBarry Smith SNESSetFunction(). 9389b94acceSBarry Smith 93936851e7fSLois Curfman McInnes Level: beginner 94036851e7fSLois Curfman McInnes 9419b94acceSBarry Smith .keywords: SNES, nonlinear, set, function 9429b94acceSBarry Smith 943a86d99e1SLois Curfman McInnes .seealso: SNESGetGradient(), SNESComputeGradient(), SNESSetHessian(), 944a86d99e1SLois Curfman McInnes SNESSetMinimizationFunction(), 9459b94acceSBarry Smith @*/ 94674679c65SBarry Smith int SNESSetGradient(SNES snes,Vec r,int (*func)(SNES,Vec,Vec,void*),void *ctx) 9479b94acceSBarry Smith { 9483a40ed3dSBarry Smith PetscFunctionBegin; 94977c4ece6SBarry Smith PetscValidHeaderSpecific(snes,SNES_COOKIE); 950a8c6a408SBarry Smith if (snes->method_class != SNES_UNCONSTRAINED_MINIMIZATION) { 951a8c6a408SBarry Smith SETERRQ(PETSC_ERR_ARG_WRONG,0,"For SNES_UNCONSTRAINED_MINIMIZATION only"); 952a8c6a408SBarry Smith } 9539b94acceSBarry Smith snes->computefunction = func; 9549b94acceSBarry Smith snes->vec_func = snes->vec_func_always = r; 9559b94acceSBarry Smith snes->funP = ctx; 9563a40ed3dSBarry Smith PetscFunctionReturn(0); 9579b94acceSBarry Smith } 9589b94acceSBarry Smith 9595615d1e5SSatish Balay #undef __FUNC__ 9605615d1e5SSatish Balay #define __FUNC__ "SNESComputeGradient" 9619b94acceSBarry Smith /*@ 962a86d99e1SLois Curfman McInnes SNESComputeGradient - Computes the gradient that has been set with 963a86d99e1SLois Curfman McInnes SNESSetGradient(). 9649b94acceSBarry Smith 965c7afd0dbSLois Curfman McInnes Collective on SNES 966c7afd0dbSLois Curfman McInnes 9679b94acceSBarry Smith Input Parameters: 968c7afd0dbSLois Curfman McInnes + snes - the SNES context 969c7afd0dbSLois Curfman McInnes - x - input vector 9709b94acceSBarry Smith 9719b94acceSBarry Smith Output Parameter: 9729b94acceSBarry Smith . y - gradient vector 9739b94acceSBarry Smith 9749b94acceSBarry Smith Notes: 9759b94acceSBarry Smith SNESComputeGradient() is valid only for 9769b94acceSBarry Smith SNES_UNCONSTRAINED_MINIMIZATION methods. An analogous routine for 9779b94acceSBarry Smith SNES_NONLINEAR_EQUATIONS methods is SNESComputeFunction(). 978a86d99e1SLois Curfman McInnes 97936851e7fSLois Curfman McInnes SNESComputeGradient() is typically used within minimization 98036851e7fSLois Curfman McInnes implementations, so most users would not generally call this routine 98136851e7fSLois Curfman McInnes themselves. 98236851e7fSLois Curfman McInnes 98336851e7fSLois Curfman McInnes Level: developer 98436851e7fSLois Curfman McInnes 985a86d99e1SLois Curfman McInnes .keywords: SNES, nonlinear, compute, gradient 986a86d99e1SLois Curfman McInnes 987a86d99e1SLois Curfman McInnes .seealso: SNESSetGradient(), SNESGetGradient(), 988a86d99e1SLois Curfman McInnes SNESComputeMinimizationFunction(), SNESComputeHessian() 9899b94acceSBarry Smith @*/ 9909b94acceSBarry Smith int SNESComputeGradient(SNES snes,Vec x, Vec y) 9919b94acceSBarry Smith { 9929b94acceSBarry Smith int ierr; 9933a40ed3dSBarry Smith 9943a40ed3dSBarry Smith PetscFunctionBegin; 9953a40ed3dSBarry Smith if (snes->method_class != SNES_UNCONSTRAINED_MINIMIZATION) { 996a8c6a408SBarry Smith SETERRQ(PETSC_ERR_ARG_WRONG,0,"For SNES_UNCONSTRAINED_MINIMIZATION only"); 9973a40ed3dSBarry Smith } 9989b94acceSBarry Smith PLogEventBegin(SNES_GradientEval,snes,x,y,0); 999d64ed03dSBarry Smith PetscStackPush("SNES user gradient function"); 10009b94acceSBarry Smith ierr = (*snes->computefunction)(snes,x,y,snes->funP); CHKERRQ(ierr); 1001d64ed03dSBarry Smith PetscStackPop; 10029b94acceSBarry Smith PLogEventEnd(SNES_GradientEval,snes,x,y,0); 10033a40ed3dSBarry Smith PetscFunctionReturn(0); 10049b94acceSBarry Smith } 10059b94acceSBarry Smith 10065615d1e5SSatish Balay #undef __FUNC__ 10075615d1e5SSatish Balay #define __FUNC__ "SNESComputeJacobian" 100862fef451SLois Curfman McInnes /*@ 100962fef451SLois Curfman McInnes SNESComputeJacobian - Computes the Jacobian matrix that has been 101062fef451SLois Curfman McInnes set with SNESSetJacobian(). 101162fef451SLois Curfman McInnes 1012c7afd0dbSLois Curfman McInnes Collective on SNES and Mat 1013c7afd0dbSLois Curfman McInnes 101462fef451SLois Curfman McInnes Input Parameters: 1015c7afd0dbSLois Curfman McInnes + snes - the SNES context 1016c7afd0dbSLois Curfman McInnes - x - input vector 101762fef451SLois Curfman McInnes 101862fef451SLois Curfman McInnes Output Parameters: 1019c7afd0dbSLois Curfman McInnes + A - Jacobian matrix 102062fef451SLois Curfman McInnes . B - optional preconditioning matrix 1021c7afd0dbSLois Curfman McInnes - flag - flag indicating matrix structure 1022fee21e36SBarry Smith 102362fef451SLois Curfman McInnes Notes: 102462fef451SLois Curfman McInnes Most users should not need to explicitly call this routine, as it 102562fef451SLois Curfman McInnes is used internally within the nonlinear solvers. 102662fef451SLois Curfman McInnes 1027dc5a77f8SLois Curfman McInnes See SLESSetOperators() for important information about setting the 1028dc5a77f8SLois Curfman McInnes flag parameter. 102962fef451SLois Curfman McInnes 103062fef451SLois Curfman McInnes SNESComputeJacobian() is valid only for SNES_NONLINEAR_EQUATIONS 103162fef451SLois Curfman McInnes methods. An analogous routine for SNES_UNCONSTRAINED_MINIMIZATION 1032005c665bSBarry Smith methods is SNESComputeHessian(). 103362fef451SLois Curfman McInnes 103436851e7fSLois Curfman McInnes SNESComputeJacobian() is typically used within nonlinear solver 103536851e7fSLois Curfman McInnes implementations, so most users would not generally call this routine 103636851e7fSLois Curfman McInnes themselves. 103736851e7fSLois Curfman McInnes 103836851e7fSLois Curfman McInnes Level: developer 103936851e7fSLois Curfman McInnes 104062fef451SLois Curfman McInnes .keywords: SNES, compute, Jacobian, matrix 104162fef451SLois Curfman McInnes 104262fef451SLois Curfman McInnes .seealso: SNESSetJacobian(), SLESSetOperators() 104362fef451SLois Curfman McInnes @*/ 10449b94acceSBarry Smith int SNESComputeJacobian(SNES snes,Vec X,Mat *A,Mat *B,MatStructure *flg) 10459b94acceSBarry Smith { 10469b94acceSBarry Smith int ierr; 10473a40ed3dSBarry Smith 10483a40ed3dSBarry Smith PetscFunctionBegin; 10493a40ed3dSBarry Smith if (snes->method_class != SNES_NONLINEAR_EQUATIONS) { 1050a8c6a408SBarry Smith SETERRQ(PETSC_ERR_ARG_WRONG,0,"For SNES_NONLINEAR_EQUATIONS only"); 10513a40ed3dSBarry Smith } 10523a40ed3dSBarry Smith if (!snes->computejacobian) PetscFunctionReturn(0); 10539b94acceSBarry Smith PLogEventBegin(SNES_JacobianEval,snes,X,*A,*B); 1054c4fc05e7SBarry Smith *flg = DIFFERENT_NONZERO_PATTERN; 1055d64ed03dSBarry Smith PetscStackPush("SNES user Jacobian function"); 10569b94acceSBarry Smith ierr = (*snes->computejacobian)(snes,X,A,B,flg,snes->jacP); CHKERRQ(ierr); 1057d64ed03dSBarry Smith PetscStackPop; 10589b94acceSBarry Smith PLogEventEnd(SNES_JacobianEval,snes,X,*A,*B); 10596d84be18SBarry Smith /* make sure user returned a correct Jacobian and preconditioner */ 106077c4ece6SBarry Smith PetscValidHeaderSpecific(*A,MAT_COOKIE); 106177c4ece6SBarry Smith PetscValidHeaderSpecific(*B,MAT_COOKIE); 10623a40ed3dSBarry Smith PetscFunctionReturn(0); 10639b94acceSBarry Smith } 10649b94acceSBarry Smith 10655615d1e5SSatish Balay #undef __FUNC__ 10665615d1e5SSatish Balay #define __FUNC__ "SNESComputeHessian" 106762fef451SLois Curfman McInnes /*@ 106862fef451SLois Curfman McInnes SNESComputeHessian - Computes the Hessian matrix that has been 106962fef451SLois Curfman McInnes set with SNESSetHessian(). 107062fef451SLois Curfman McInnes 1071c7afd0dbSLois Curfman McInnes Collective on SNES and Mat 1072c7afd0dbSLois Curfman McInnes 107362fef451SLois Curfman McInnes Input Parameters: 1074c7afd0dbSLois Curfman McInnes + snes - the SNES context 1075c7afd0dbSLois Curfman McInnes - x - input vector 107662fef451SLois Curfman McInnes 107762fef451SLois Curfman McInnes Output Parameters: 1078c7afd0dbSLois Curfman McInnes + A - Hessian matrix 107962fef451SLois Curfman McInnes . B - optional preconditioning matrix 1080c7afd0dbSLois Curfman McInnes - flag - flag indicating matrix structure 1081fee21e36SBarry Smith 108262fef451SLois Curfman McInnes Notes: 108362fef451SLois Curfman McInnes Most users should not need to explicitly call this routine, as it 108462fef451SLois Curfman McInnes is used internally within the nonlinear solvers. 108562fef451SLois Curfman McInnes 1086dc5a77f8SLois Curfman McInnes See SLESSetOperators() for important information about setting the 1087dc5a77f8SLois Curfman McInnes flag parameter. 108862fef451SLois Curfman McInnes 108962fef451SLois Curfman McInnes SNESComputeHessian() is valid only for 109062fef451SLois Curfman McInnes SNES_UNCONSTRAINED_MINIMIZATION methods. An analogous routine for 109162fef451SLois Curfman McInnes SNES_NONLINEAR_EQUATIONS methods is SNESComputeJacobian(). 109262fef451SLois Curfman McInnes 109336851e7fSLois Curfman McInnes SNESComputeHessian() is typically used within minimization 109436851e7fSLois Curfman McInnes implementations, so most users would not generally call this routine 109536851e7fSLois Curfman McInnes themselves. 109636851e7fSLois Curfman McInnes 109736851e7fSLois Curfman McInnes Level: developer 109836851e7fSLois Curfman McInnes 109962fef451SLois Curfman McInnes .keywords: SNES, compute, Hessian, matrix 110062fef451SLois Curfman McInnes 1101a86d99e1SLois Curfman McInnes .seealso: SNESSetHessian(), SLESSetOperators(), SNESComputeGradient(), 1102a86d99e1SLois Curfman McInnes SNESComputeMinimizationFunction() 110362fef451SLois Curfman McInnes @*/ 110462fef451SLois Curfman McInnes int SNESComputeHessian(SNES snes,Vec x,Mat *A,Mat *B,MatStructure *flag) 11059b94acceSBarry Smith { 11069b94acceSBarry Smith int ierr; 11073a40ed3dSBarry Smith 11083a40ed3dSBarry Smith PetscFunctionBegin; 11093a40ed3dSBarry Smith if (snes->method_class != SNES_UNCONSTRAINED_MINIMIZATION) { 1110a8c6a408SBarry Smith SETERRQ(PETSC_ERR_ARG_WRONG,0,"For SNES_UNCONSTRAINED_MINIMIZATION only"); 11113a40ed3dSBarry Smith } 11123a40ed3dSBarry Smith if (!snes->computejacobian) PetscFunctionReturn(0); 111362fef451SLois Curfman McInnes PLogEventBegin(SNES_HessianEval,snes,x,*A,*B); 11140f4a323eSLois Curfman McInnes *flag = DIFFERENT_NONZERO_PATTERN; 1115d64ed03dSBarry Smith PetscStackPush("SNES user Hessian function"); 111662fef451SLois Curfman McInnes ierr = (*snes->computejacobian)(snes,x,A,B,flag,snes->jacP); CHKERRQ(ierr); 1117d64ed03dSBarry Smith PetscStackPop; 111862fef451SLois Curfman McInnes PLogEventEnd(SNES_HessianEval,snes,x,*A,*B); 11190f4a323eSLois Curfman McInnes /* make sure user returned a correct Jacobian and preconditioner */ 112077c4ece6SBarry Smith PetscValidHeaderSpecific(*A,MAT_COOKIE); 112177c4ece6SBarry Smith PetscValidHeaderSpecific(*B,MAT_COOKIE); 11223a40ed3dSBarry Smith PetscFunctionReturn(0); 11239b94acceSBarry Smith } 11249b94acceSBarry Smith 11255615d1e5SSatish Balay #undef __FUNC__ 1126d4bb536fSBarry Smith #define __FUNC__ "SNESSetJacobian" 11279b94acceSBarry Smith /*@C 11289b94acceSBarry Smith SNESSetJacobian - Sets the function to compute Jacobian as well as the 1129044dda88SLois Curfman McInnes location to store the matrix. 11309b94acceSBarry Smith 1131c7afd0dbSLois Curfman McInnes Collective on SNES and Mat 1132c7afd0dbSLois Curfman McInnes 11339b94acceSBarry Smith Input Parameters: 1134c7afd0dbSLois Curfman McInnes + snes - the SNES context 11359b94acceSBarry Smith . A - Jacobian matrix 11369b94acceSBarry Smith . B - preconditioner matrix (usually same as the Jacobian) 11379b94acceSBarry Smith . func - Jacobian evaluation routine 1138c7afd0dbSLois Curfman McInnes - ctx - [optional] user-defined context for private data for the 11392cd2dfdcSLois Curfman McInnes Jacobian evaluation routine (may be PETSC_NULL) 11409b94acceSBarry Smith 11419b94acceSBarry Smith Calling sequence of func: 11428d76a1e5SLois Curfman McInnes $ func (SNES snes,Vec x,Mat *A,Mat *B,int *flag,void *ctx); 11439b94acceSBarry Smith 1144c7afd0dbSLois Curfman McInnes + x - input vector 11459b94acceSBarry Smith . A - Jacobian matrix 11469b94acceSBarry Smith . B - preconditioner matrix, usually the same as A 1147ac21db08SLois Curfman McInnes . flag - flag indicating information about the preconditioner matrix 1148ac21db08SLois Curfman McInnes structure (same as flag in SLESSetOperators()) 1149c7afd0dbSLois Curfman McInnes - ctx - [optional] user-defined Jacobian context 11509b94acceSBarry Smith 11519b94acceSBarry Smith Notes: 1152dc5a77f8SLois Curfman McInnes See SLESSetOperators() for important information about setting the flag 11532cd2dfdcSLois Curfman McInnes output parameter in the routine func(). Be sure to read this information! 1154ac21db08SLois Curfman McInnes 1155ac21db08SLois Curfman McInnes The routine func() takes Mat * as the matrix arguments rather than Mat. 11569b94acceSBarry Smith This allows the Jacobian evaluation routine to replace A and/or B with a 11579b94acceSBarry Smith completely new new matrix structure (not just different matrix elements) 11589b94acceSBarry Smith when appropriate, for instance, if the nonzero structure is changing 11599b94acceSBarry Smith throughout the global iterations. 11609b94acceSBarry Smith 116136851e7fSLois Curfman McInnes Level: beginner 116236851e7fSLois Curfman McInnes 11639b94acceSBarry Smith .keywords: SNES, nonlinear, set, Jacobian, matrix 11649b94acceSBarry Smith 1165ac21db08SLois Curfman McInnes .seealso: SLESSetOperators(), SNESSetFunction() 11669b94acceSBarry Smith @*/ 11679b94acceSBarry Smith int SNESSetJacobian(SNES snes,Mat A,Mat B,int (*func)(SNES,Vec,Mat*,Mat*, 11689b94acceSBarry Smith MatStructure*,void*),void *ctx) 11699b94acceSBarry Smith { 11703a40ed3dSBarry Smith PetscFunctionBegin; 117177c4ece6SBarry Smith PetscValidHeaderSpecific(snes,SNES_COOKIE); 1172a8c6a408SBarry Smith if (snes->method_class != SNES_NONLINEAR_EQUATIONS) { 1173a8c6a408SBarry Smith SETERRQ(PETSC_ERR_ARG_WRONG,0,"For SNES_NONLINEAR_EQUATIONS only"); 1174a8c6a408SBarry Smith } 11759b94acceSBarry Smith snes->computejacobian = func; 11769b94acceSBarry Smith snes->jacP = ctx; 11779b94acceSBarry Smith snes->jacobian = A; 11789b94acceSBarry Smith snes->jacobian_pre = B; 11793a40ed3dSBarry Smith PetscFunctionReturn(0); 11809b94acceSBarry Smith } 118162fef451SLois Curfman McInnes 11825615d1e5SSatish Balay #undef __FUNC__ 1183d4bb536fSBarry Smith #define __FUNC__ "SNESGetJacobian" 1184b4fd4287SBarry Smith /*@ 1185b4fd4287SBarry Smith SNESGetJacobian - Returns the Jacobian matrix and optionally the user 1186b4fd4287SBarry Smith provided context for evaluating the Jacobian. 1187b4fd4287SBarry Smith 1188c7afd0dbSLois Curfman McInnes Not Collective, but Mat object will be parallel if SNES object is 1189c7afd0dbSLois Curfman McInnes 1190b4fd4287SBarry Smith Input Parameter: 1191b4fd4287SBarry Smith . snes - the nonlinear solver context 1192b4fd4287SBarry Smith 1193b4fd4287SBarry Smith Output Parameters: 1194c7afd0dbSLois Curfman McInnes + A - location to stash Jacobian matrix (or PETSC_NULL) 1195b4fd4287SBarry Smith . B - location to stash preconditioner matrix (or PETSC_NULL) 1196c7afd0dbSLois Curfman McInnes - ctx - location to stash Jacobian ctx (or PETSC_NULL) 1197fee21e36SBarry Smith 119836851e7fSLois Curfman McInnes Level: advanced 119936851e7fSLois Curfman McInnes 1200b4fd4287SBarry Smith .seealso: SNESSetJacobian(), SNESComputeJacobian() 1201b4fd4287SBarry Smith @*/ 1202b4fd4287SBarry Smith int SNESGetJacobian(SNES snes,Mat *A,Mat *B, void **ctx) 1203b4fd4287SBarry Smith { 12043a40ed3dSBarry Smith PetscFunctionBegin; 12053a40ed3dSBarry Smith if (snes->method_class != SNES_NONLINEAR_EQUATIONS) { 1206a8c6a408SBarry Smith SETERRQ(PETSC_ERR_ARG_WRONG,0,"For SNES_NONLINEAR_EQUATIONS only"); 12073a40ed3dSBarry Smith } 1208b4fd4287SBarry Smith if (A) *A = snes->jacobian; 1209b4fd4287SBarry Smith if (B) *B = snes->jacobian_pre; 1210b4fd4287SBarry Smith if (ctx) *ctx = snes->jacP; 12113a40ed3dSBarry Smith PetscFunctionReturn(0); 1212b4fd4287SBarry Smith } 1213b4fd4287SBarry Smith 12145615d1e5SSatish Balay #undef __FUNC__ 1215d4bb536fSBarry Smith #define __FUNC__ "SNESSetHessian" 12169b94acceSBarry Smith /*@C 12179b94acceSBarry Smith SNESSetHessian - Sets the function to compute Hessian as well as the 1218044dda88SLois Curfman McInnes location to store the matrix. 12199b94acceSBarry Smith 1220c7afd0dbSLois Curfman McInnes Collective on SNES and Mat 1221c7afd0dbSLois Curfman McInnes 12229b94acceSBarry Smith Input Parameters: 1223c7afd0dbSLois Curfman McInnes + snes - the SNES context 12249b94acceSBarry Smith . A - Hessian matrix 12259b94acceSBarry Smith . B - preconditioner matrix (usually same as the Hessian) 12269b94acceSBarry Smith . func - Jacobian evaluation routine 1227c7afd0dbSLois Curfman McInnes - ctx - [optional] user-defined context for private data for the 1228313e4042SLois Curfman McInnes Hessian evaluation routine (may be PETSC_NULL) 12299b94acceSBarry Smith 12309b94acceSBarry Smith Calling sequence of func: 12318d76a1e5SLois Curfman McInnes $ func (SNES snes,Vec x,Mat *A,Mat *B,int *flag,void *ctx); 12329b94acceSBarry Smith 1233c7afd0dbSLois Curfman McInnes + x - input vector 12349b94acceSBarry Smith . A - Hessian matrix 12359b94acceSBarry Smith . B - preconditioner matrix, usually the same as A 1236ac21db08SLois Curfman McInnes . flag - flag indicating information about the preconditioner matrix 1237ac21db08SLois Curfman McInnes structure (same as flag in SLESSetOperators()) 1238c7afd0dbSLois Curfman McInnes - ctx - [optional] user-defined Hessian context 12399b94acceSBarry Smith 12409b94acceSBarry Smith Notes: 1241dc5a77f8SLois Curfman McInnes See SLESSetOperators() for important information about setting the flag 12422cd2dfdcSLois Curfman McInnes output parameter in the routine func(). Be sure to read this information! 1243ac21db08SLois Curfman McInnes 12449b94acceSBarry Smith The function func() takes Mat * as the matrix arguments rather than Mat. 12459b94acceSBarry Smith This allows the Hessian evaluation routine to replace A and/or B with a 12469b94acceSBarry Smith completely new new matrix structure (not just different matrix elements) 12479b94acceSBarry Smith when appropriate, for instance, if the nonzero structure is changing 12489b94acceSBarry Smith throughout the global iterations. 12499b94acceSBarry Smith 125036851e7fSLois Curfman McInnes Level: beginner 125136851e7fSLois Curfman McInnes 12529b94acceSBarry Smith .keywords: SNES, nonlinear, set, Hessian, matrix 12539b94acceSBarry Smith 1254ac21db08SLois Curfman McInnes .seealso: SNESSetMinimizationFunction(), SNESSetGradient(), SLESSetOperators() 12559b94acceSBarry Smith @*/ 12569b94acceSBarry Smith int SNESSetHessian(SNES snes,Mat A,Mat B,int (*func)(SNES,Vec,Mat*,Mat*, 12579b94acceSBarry Smith MatStructure*,void*),void *ctx) 12589b94acceSBarry Smith { 12593a40ed3dSBarry Smith PetscFunctionBegin; 126077c4ece6SBarry Smith PetscValidHeaderSpecific(snes,SNES_COOKIE); 1261d4bb536fSBarry Smith if (snes->method_class != SNES_UNCONSTRAINED_MINIMIZATION) { 1262a8c6a408SBarry Smith SETERRQ(PETSC_ERR_ARG_WRONG,0,"For SNES_UNCONSTRAINED_MINIMIZATION only"); 1263d4bb536fSBarry Smith } 12649b94acceSBarry Smith snes->computejacobian = func; 12659b94acceSBarry Smith snes->jacP = ctx; 12669b94acceSBarry Smith snes->jacobian = A; 12679b94acceSBarry Smith snes->jacobian_pre = B; 12683a40ed3dSBarry Smith PetscFunctionReturn(0); 12699b94acceSBarry Smith } 12709b94acceSBarry Smith 12715615d1e5SSatish Balay #undef __FUNC__ 1272d4bb536fSBarry Smith #define __FUNC__ "SNESGetHessian" 127362fef451SLois Curfman McInnes /*@ 127462fef451SLois Curfman McInnes SNESGetHessian - Returns the Hessian matrix and optionally the user 127562fef451SLois Curfman McInnes provided context for evaluating the Hessian. 127662fef451SLois Curfman McInnes 1277c7afd0dbSLois Curfman McInnes Not Collective, but Mat object is parallel if SNES object is parallel 1278c7afd0dbSLois Curfman McInnes 127962fef451SLois Curfman McInnes Input Parameter: 128062fef451SLois Curfman McInnes . snes - the nonlinear solver context 128162fef451SLois Curfman McInnes 128262fef451SLois Curfman McInnes Output Parameters: 1283c7afd0dbSLois Curfman McInnes + A - location to stash Hessian matrix (or PETSC_NULL) 128462fef451SLois Curfman McInnes . B - location to stash preconditioner matrix (or PETSC_NULL) 1285c7afd0dbSLois Curfman McInnes - ctx - location to stash Hessian ctx (or PETSC_NULL) 1286fee21e36SBarry Smith 128736851e7fSLois Curfman McInnes Level: advanced 128836851e7fSLois Curfman McInnes 128962fef451SLois Curfman McInnes .seealso: SNESSetHessian(), SNESComputeHessian() 1290c7afd0dbSLois Curfman McInnes 1291c7afd0dbSLois Curfman McInnes .keywords: SNES, get, Hessian 129262fef451SLois Curfman McInnes @*/ 129362fef451SLois Curfman McInnes int SNESGetHessian(SNES snes,Mat *A,Mat *B, void **ctx) 129462fef451SLois Curfman McInnes { 12953a40ed3dSBarry Smith PetscFunctionBegin; 12963a40ed3dSBarry Smith if (snes->method_class != SNES_UNCONSTRAINED_MINIMIZATION){ 1297a8c6a408SBarry Smith SETERRQ(PETSC_ERR_ARG_WRONG,0,"For SNES_UNCONSTRAINED_MINIMIZATION only"); 12983a40ed3dSBarry Smith } 129962fef451SLois Curfman McInnes if (A) *A = snes->jacobian; 130062fef451SLois Curfman McInnes if (B) *B = snes->jacobian_pre; 130162fef451SLois Curfman McInnes if (ctx) *ctx = snes->jacP; 13023a40ed3dSBarry Smith PetscFunctionReturn(0); 130362fef451SLois Curfman McInnes } 130462fef451SLois Curfman McInnes 13059b94acceSBarry Smith /* ----- Routines to initialize and destroy a nonlinear solver ---- */ 13069b94acceSBarry Smith 13075615d1e5SSatish Balay #undef __FUNC__ 13085615d1e5SSatish Balay #define __FUNC__ "SNESSetUp" 13099b94acceSBarry Smith /*@ 13109b94acceSBarry Smith SNESSetUp - Sets up the internal data structures for the later use 1311272ac6f2SLois Curfman McInnes of a nonlinear solver. 13129b94acceSBarry Smith 1313fee21e36SBarry Smith Collective on SNES 1314fee21e36SBarry Smith 1315c7afd0dbSLois Curfman McInnes Input Parameters: 1316c7afd0dbSLois Curfman McInnes + snes - the SNES context 1317c7afd0dbSLois Curfman McInnes - x - the solution vector 1318c7afd0dbSLois Curfman McInnes 1319272ac6f2SLois Curfman McInnes Notes: 1320272ac6f2SLois Curfman McInnes For basic use of the SNES solvers the user need not explicitly call 1321272ac6f2SLois Curfman McInnes SNESSetUp(), since these actions will automatically occur during 1322272ac6f2SLois Curfman McInnes the call to SNESSolve(). However, if one wishes to control this 1323272ac6f2SLois Curfman McInnes phase separately, SNESSetUp() should be called after SNESCreate() 1324272ac6f2SLois Curfman McInnes and optional routines of the form SNESSetXXX(), but before SNESSolve(). 1325272ac6f2SLois Curfman McInnes 132636851e7fSLois Curfman McInnes Level: advanced 132736851e7fSLois Curfman McInnes 13289b94acceSBarry Smith .keywords: SNES, nonlinear, setup 13299b94acceSBarry Smith 13309b94acceSBarry Smith .seealso: SNESCreate(), SNESSolve(), SNESDestroy() 13319b94acceSBarry Smith @*/ 13328ddd3da0SLois Curfman McInnes int SNESSetUp(SNES snes,Vec x) 13339b94acceSBarry Smith { 1334272ac6f2SLois Curfman McInnes int ierr, flg; 13353a40ed3dSBarry Smith 13363a40ed3dSBarry Smith PetscFunctionBegin; 133777c4ece6SBarry Smith PetscValidHeaderSpecific(snes,SNES_COOKIE); 133877c4ece6SBarry Smith PetscValidHeaderSpecific(x,VEC_COOKIE); 13398ddd3da0SLois Curfman McInnes snes->vec_sol = snes->vec_sol_always = x; 13409b94acceSBarry Smith 1341c1f60f51SBarry Smith ierr = OptionsHasName(snes->prefix,"-snes_mf_operator", &flg); CHKERRQ(ierr); 1342c1f60f51SBarry Smith /* 1343c1f60f51SBarry Smith This version replaces the user provided Jacobian matrix with a 1344dfa02198SLois Curfman McInnes matrix-free version but still employs the user-provided preconditioner matrix 1345c1f60f51SBarry Smith */ 1346c1f60f51SBarry Smith if (flg) { 1347c1f60f51SBarry Smith Mat J; 1348e24b481bSBarry Smith ierr = MatCreateSNESFDMF(snes,snes->vec_sol,&J);CHKERRQ(ierr); 1349c1f60f51SBarry Smith PLogObjectParent(snes,J); 1350c1f60f51SBarry Smith snes->mfshell = J; 1351c1f60f51SBarry Smith if (snes->method_class == SNES_NONLINEAR_EQUATIONS) { 1352c1f60f51SBarry Smith snes->jacobian = J; 1353c1f60f51SBarry Smith PLogInfo(snes,"SNESSetUp: Setting default matrix-free operator Jacobian routines\n"); 1354d64ed03dSBarry Smith } else if (snes->method_class == SNES_UNCONSTRAINED_MINIMIZATION) { 1355c1f60f51SBarry Smith snes->jacobian = J; 1356c1f60f51SBarry Smith PLogInfo(snes,"SNESSetUp: Setting default matrix-free operator Hessian routines\n"); 1357d4bb536fSBarry Smith } else { 1358a8c6a408SBarry Smith SETERRQ(PETSC_ERR_SUP,0,"Method class doesn't support matrix-free operator option"); 1359d4bb536fSBarry Smith } 1360e24b481bSBarry Smith ierr = MatSNESFDMFSetFromOptions(J);CHKERRQ(ierr); 1361c1f60f51SBarry Smith } 1362272ac6f2SLois Curfman McInnes ierr = OptionsHasName(snes->prefix,"-snes_mf", &flg); CHKERRQ(ierr); 1363c1f60f51SBarry Smith /* 1364dfa02198SLois Curfman McInnes This version replaces both the user-provided Jacobian and the user- 1365c1f60f51SBarry Smith provided preconditioner matrix with the default matrix free version. 1366c1f60f51SBarry Smith */ 136731615d04SLois Curfman McInnes if (flg) { 1368272ac6f2SLois Curfman McInnes Mat J; 1369e24b481bSBarry Smith ierr = MatCreateSNESFDMF(snes,snes->vec_sol,&J);CHKERRQ(ierr); 1370272ac6f2SLois Curfman McInnes PLogObjectParent(snes,J); 1371272ac6f2SLois Curfman McInnes snes->mfshell = J; 137283e56afdSLois Curfman McInnes if (snes->method_class == SNES_NONLINEAR_EQUATIONS) { 137383e56afdSLois Curfman McInnes ierr = SNESSetJacobian(snes,J,J,0,snes->funP); CHKERRQ(ierr); 137494a424c1SBarry Smith PLogInfo(snes,"SNESSetUp: Setting default matrix-free Jacobian routines\n"); 1375d64ed03dSBarry Smith } else if (snes->method_class == SNES_UNCONSTRAINED_MINIMIZATION) { 137683e56afdSLois Curfman McInnes ierr = SNESSetHessian(snes,J,J,0,snes->funP); CHKERRQ(ierr); 137794a424c1SBarry Smith PLogInfo(snes,"SNESSetUp: Setting default matrix-free Hessian routines\n"); 1378d4bb536fSBarry Smith } else { 1379a8c6a408SBarry Smith SETERRQ(PETSC_ERR_SUP,0,"Method class doesn't support matrix-free option"); 1380d4bb536fSBarry Smith } 1381e24b481bSBarry Smith ierr = MatSNESFDMFSetFromOptions(J);CHKERRQ(ierr); 1382272ac6f2SLois Curfman McInnes } 13839b94acceSBarry Smith if ((snes->method_class == SNES_NONLINEAR_EQUATIONS)) { 1384a8c6a408SBarry Smith if (!snes->vec_func) SETERRQ(PETSC_ERR_ARG_WRONGSTATE,0,"Must call SNESSetFunction() first"); 1385a8c6a408SBarry Smith if (!snes->computefunction) SETERRQ(PETSC_ERR_ARG_WRONGSTATE,0,"Must call SNESSetFunction() first"); 1386a8c6a408SBarry Smith if (!snes->jacobian) SETERRQ(PETSC_ERR_ARG_WRONGSTATE,0,"Must call SNESSetJacobian() first"); 1387a8c6a408SBarry Smith if (snes->vec_func == snes->vec_sol) { 1388a8c6a408SBarry Smith SETERRQ(PETSC_ERR_ARG_IDN,0,"Solution vector cannot be function vector"); 1389a8c6a408SBarry Smith } 1390a703fe33SLois Curfman McInnes 1391a703fe33SLois Curfman McInnes /* Set the KSP stopping criterion to use the Eisenstat-Walker method */ 139282bf6240SBarry Smith if (snes->ksp_ewconv && PetscStrcmp(snes->type_name,SNES_EQ_TR)) { 1393a703fe33SLois Curfman McInnes SLES sles; KSP ksp; 1394a703fe33SLois Curfman McInnes ierr = SNESGetSLES(snes,&sles); CHKERRQ(ierr); 1395a703fe33SLois Curfman McInnes ierr = SLESGetKSP(sles,&ksp); CHKERRQ(ierr); 1396a703fe33SLois Curfman McInnes ierr = KSPSetConvergenceTest(ksp,SNES_KSP_EW_Converged_Private, 1397a703fe33SLois Curfman McInnes (void *)snes); CHKERRQ(ierr); 1398a703fe33SLois Curfman McInnes } 13999b94acceSBarry Smith } else if ((snes->method_class == SNES_UNCONSTRAINED_MINIMIZATION)) { 1400a8c6a408SBarry Smith if (!snes->vec_func) SETERRQ(PETSC_ERR_ARG_WRONGSTATE,0,"Must call SNESSetGradient() first"); 1401a8c6a408SBarry Smith if (!snes->computefunction) SETERRQ(PETSC_ERR_ARG_WRONGSTATE,0,"Must call SNESSetGradient() first"); 1402a8c6a408SBarry Smith if (!snes->computeumfunction) { 1403a8c6a408SBarry Smith SETERRQ(PETSC_ERR_ARG_WRONGSTATE,0,"Must call SNESSetMinimizationFunction() first"); 1404a8c6a408SBarry Smith } 1405a8c6a408SBarry Smith if (!snes->jacobian) SETERRQ(PETSC_ERR_ARG_WRONGSTATE,0,"Must call SNESSetHessian() first"); 1406d4bb536fSBarry Smith } else { 1407a8c6a408SBarry Smith SETERRQ(PETSC_ERR_ARG_OUTOFRANGE,0,"Unknown method class"); 1408d4bb536fSBarry Smith } 1409a703fe33SLois Curfman McInnes if (snes->setup) {ierr = (*snes->setup)(snes); CHKERRQ(ierr);} 141082bf6240SBarry Smith snes->setupcalled = 1; 14113a40ed3dSBarry Smith PetscFunctionReturn(0); 14129b94acceSBarry Smith } 14139b94acceSBarry Smith 14145615d1e5SSatish Balay #undef __FUNC__ 1415d4bb536fSBarry Smith #define __FUNC__ "SNESDestroy" 14169b94acceSBarry Smith /*@C 14179b94acceSBarry Smith SNESDestroy - Destroys the nonlinear solver context that was created 14189b94acceSBarry Smith with SNESCreate(). 14199b94acceSBarry Smith 1420c7afd0dbSLois Curfman McInnes Collective on SNES 1421c7afd0dbSLois Curfman McInnes 14229b94acceSBarry Smith Input Parameter: 14239b94acceSBarry Smith . snes - the SNES context 14249b94acceSBarry Smith 142536851e7fSLois Curfman McInnes Level: beginner 142636851e7fSLois Curfman McInnes 14279b94acceSBarry Smith .keywords: SNES, nonlinear, destroy 14289b94acceSBarry Smith 142963a78c88SLois Curfman McInnes .seealso: SNESCreate(), SNESSolve() 14309b94acceSBarry Smith @*/ 14319b94acceSBarry Smith int SNESDestroy(SNES snes) 14329b94acceSBarry Smith { 14339b94acceSBarry Smith int ierr; 14343a40ed3dSBarry Smith 14353a40ed3dSBarry Smith PetscFunctionBegin; 143677c4ece6SBarry Smith PetscValidHeaderSpecific(snes,SNES_COOKIE); 14373a40ed3dSBarry Smith if (--snes->refct > 0) PetscFunctionReturn(0); 1438d4bb536fSBarry Smith 1439e1311b90SBarry Smith if (snes->destroy) {ierr = (*(snes)->destroy)(snes); CHKERRQ(ierr);} 14400452661fSBarry Smith if (snes->kspconvctx) PetscFree(snes->kspconvctx); 1441522c5e43SBarry Smith if (snes->mfshell) {ierr = MatDestroy(snes->mfshell);CHKERRQ(ierr);} 14429b94acceSBarry Smith ierr = SLESDestroy(snes->sles); CHKERRQ(ierr); 1443522c5e43SBarry Smith if (snes->xmonitor) {ierr = SNESLGMonitorDestroy(snes->xmonitor);CHKERRQ(ierr);} 1444522c5e43SBarry Smith if (snes->vwork) {ierr = VecDestroyVecs(snes->vwork,snes->nvwork);CHKERRQ(ierr);} 14459b94acceSBarry Smith PLogObjectDestroy((PetscObject)snes); 14460452661fSBarry Smith PetscHeaderDestroy((PetscObject)snes); 14473a40ed3dSBarry Smith PetscFunctionReturn(0); 14489b94acceSBarry Smith } 14499b94acceSBarry Smith 14509b94acceSBarry Smith /* ----------- Routines to set solver parameters ---------- */ 14519b94acceSBarry Smith 14525615d1e5SSatish Balay #undef __FUNC__ 14535615d1e5SSatish Balay #define __FUNC__ "SNESSetTolerances" 14549b94acceSBarry Smith /*@ 1455d7a720efSLois Curfman McInnes SNESSetTolerances - Sets various parameters used in convergence tests. 14569b94acceSBarry Smith 1457c7afd0dbSLois Curfman McInnes Collective on SNES 1458c7afd0dbSLois Curfman McInnes 14599b94acceSBarry Smith Input Parameters: 1460c7afd0dbSLois Curfman McInnes + snes - the SNES context 146133174efeSLois Curfman McInnes . atol - absolute convergence tolerance 146233174efeSLois Curfman McInnes . rtol - relative convergence tolerance 146333174efeSLois Curfman McInnes . stol - convergence tolerance in terms of the norm 146433174efeSLois Curfman McInnes of the change in the solution between steps 146533174efeSLois Curfman McInnes . maxit - maximum number of iterations 1466c7afd0dbSLois Curfman McInnes - maxf - maximum number of function evaluations 1467fee21e36SBarry Smith 146833174efeSLois Curfman McInnes Options Database Keys: 1469c7afd0dbSLois Curfman McInnes + -snes_atol <atol> - Sets atol 1470c7afd0dbSLois Curfman McInnes . -snes_rtol <rtol> - Sets rtol 1471c7afd0dbSLois Curfman McInnes . -snes_stol <stol> - Sets stol 1472c7afd0dbSLois Curfman McInnes . -snes_max_it <maxit> - Sets maxit 1473c7afd0dbSLois Curfman McInnes - -snes_max_funcs <maxf> - Sets maxf 14749b94acceSBarry Smith 1475d7a720efSLois Curfman McInnes Notes: 14769b94acceSBarry Smith The default maximum number of iterations is 50. 14779b94acceSBarry Smith The default maximum number of function evaluations is 1000. 14789b94acceSBarry Smith 147936851e7fSLois Curfman McInnes Level: intermediate 148036851e7fSLois Curfman McInnes 148133174efeSLois Curfman McInnes .keywords: SNES, nonlinear, set, convergence, tolerances 14829b94acceSBarry Smith 1483d7a720efSLois Curfman McInnes .seealso: SNESSetTrustRegionTolerance(), SNESSetMinimizationFunctionTolerance() 14849b94acceSBarry Smith @*/ 1485d7a720efSLois Curfman McInnes int SNESSetTolerances(SNES snes,double atol,double rtol,double stol,int maxit,int maxf) 14869b94acceSBarry Smith { 14873a40ed3dSBarry Smith PetscFunctionBegin; 148877c4ece6SBarry Smith PetscValidHeaderSpecific(snes,SNES_COOKIE); 1489d7a720efSLois Curfman McInnes if (atol != PETSC_DEFAULT) snes->atol = atol; 1490d7a720efSLois Curfman McInnes if (rtol != PETSC_DEFAULT) snes->rtol = rtol; 1491d7a720efSLois Curfman McInnes if (stol != PETSC_DEFAULT) snes->xtol = stol; 1492d7a720efSLois Curfman McInnes if (maxit != PETSC_DEFAULT) snes->max_its = maxit; 1493d7a720efSLois Curfman McInnes if (maxf != PETSC_DEFAULT) snes->max_funcs = maxf; 14943a40ed3dSBarry Smith PetscFunctionReturn(0); 14959b94acceSBarry Smith } 14969b94acceSBarry Smith 14975615d1e5SSatish Balay #undef __FUNC__ 14985615d1e5SSatish Balay #define __FUNC__ "SNESGetTolerances" 14999b94acceSBarry Smith /*@ 150033174efeSLois Curfman McInnes SNESGetTolerances - Gets various parameters used in convergence tests. 150133174efeSLois Curfman McInnes 1502c7afd0dbSLois Curfman McInnes Not Collective 1503c7afd0dbSLois Curfman McInnes 150433174efeSLois Curfman McInnes Input Parameters: 1505c7afd0dbSLois Curfman McInnes + snes - the SNES context 150633174efeSLois Curfman McInnes . atol - absolute convergence tolerance 150733174efeSLois Curfman McInnes . rtol - relative convergence tolerance 150833174efeSLois Curfman McInnes . stol - convergence tolerance in terms of the norm 150933174efeSLois Curfman McInnes of the change in the solution between steps 151033174efeSLois Curfman McInnes . maxit - maximum number of iterations 1511c7afd0dbSLois Curfman McInnes - maxf - maximum number of function evaluations 1512fee21e36SBarry Smith 151333174efeSLois Curfman McInnes Notes: 151433174efeSLois Curfman McInnes The user can specify PETSC_NULL for any parameter that is not needed. 151533174efeSLois Curfman McInnes 151636851e7fSLois Curfman McInnes Level: intermediate 151736851e7fSLois Curfman McInnes 151833174efeSLois Curfman McInnes .keywords: SNES, nonlinear, get, convergence, tolerances 151933174efeSLois Curfman McInnes 152033174efeSLois Curfman McInnes .seealso: SNESSetTolerances() 152133174efeSLois Curfman McInnes @*/ 152233174efeSLois Curfman McInnes int SNESGetTolerances(SNES snes,double *atol,double *rtol,double *stol,int *maxit,int *maxf) 152333174efeSLois Curfman McInnes { 15243a40ed3dSBarry Smith PetscFunctionBegin; 152533174efeSLois Curfman McInnes PetscValidHeaderSpecific(snes,SNES_COOKIE); 152633174efeSLois Curfman McInnes if (atol) *atol = snes->atol; 152733174efeSLois Curfman McInnes if (rtol) *rtol = snes->rtol; 152833174efeSLois Curfman McInnes if (stol) *stol = snes->xtol; 152933174efeSLois Curfman McInnes if (maxit) *maxit = snes->max_its; 153033174efeSLois Curfman McInnes if (maxf) *maxf = snes->max_funcs; 15313a40ed3dSBarry Smith PetscFunctionReturn(0); 153233174efeSLois Curfman McInnes } 153333174efeSLois Curfman McInnes 15345615d1e5SSatish Balay #undef __FUNC__ 15355615d1e5SSatish Balay #define __FUNC__ "SNESSetTrustRegionTolerance" 153633174efeSLois Curfman McInnes /*@ 15379b94acceSBarry Smith SNESSetTrustRegionTolerance - Sets the trust region parameter tolerance. 15389b94acceSBarry Smith 1539fee21e36SBarry Smith Collective on SNES 1540fee21e36SBarry Smith 1541c7afd0dbSLois Curfman McInnes Input Parameters: 1542c7afd0dbSLois Curfman McInnes + snes - the SNES context 1543c7afd0dbSLois Curfman McInnes - tol - tolerance 1544c7afd0dbSLois Curfman McInnes 15459b94acceSBarry Smith Options Database Key: 1546c7afd0dbSLois Curfman McInnes . -snes_trtol <tol> - Sets tol 15479b94acceSBarry Smith 154836851e7fSLois Curfman McInnes Level: intermediate 154936851e7fSLois Curfman McInnes 15509b94acceSBarry Smith .keywords: SNES, nonlinear, set, trust region, tolerance 15519b94acceSBarry Smith 1552d7a720efSLois Curfman McInnes .seealso: SNESSetTolerances(), SNESSetMinimizationFunctionTolerance() 15539b94acceSBarry Smith @*/ 15549b94acceSBarry Smith int SNESSetTrustRegionTolerance(SNES snes,double tol) 15559b94acceSBarry Smith { 15563a40ed3dSBarry Smith PetscFunctionBegin; 155777c4ece6SBarry Smith PetscValidHeaderSpecific(snes,SNES_COOKIE); 15589b94acceSBarry Smith snes->deltatol = tol; 15593a40ed3dSBarry Smith PetscFunctionReturn(0); 15609b94acceSBarry Smith } 15619b94acceSBarry Smith 15625615d1e5SSatish Balay #undef __FUNC__ 15635615d1e5SSatish Balay #define __FUNC__ "SNESSetMinimizationFunctionTolerance" 15649b94acceSBarry Smith /*@ 156577c4ece6SBarry Smith SNESSetMinimizationFunctionTolerance - Sets the minimum allowable function tolerance 15669b94acceSBarry Smith for unconstrained minimization solvers. 15679b94acceSBarry Smith 1568fee21e36SBarry Smith Collective on SNES 1569fee21e36SBarry Smith 1570c7afd0dbSLois Curfman McInnes Input Parameters: 1571c7afd0dbSLois Curfman McInnes + snes - the SNES context 1572c7afd0dbSLois Curfman McInnes - ftol - minimum function tolerance 1573c7afd0dbSLois Curfman McInnes 15749b94acceSBarry Smith Options Database Key: 1575c7afd0dbSLois Curfman McInnes . -snes_fmin <ftol> - Sets ftol 15769b94acceSBarry Smith 15779b94acceSBarry Smith Note: 157877c4ece6SBarry Smith SNESSetMinimizationFunctionTolerance() is valid for SNES_UNCONSTRAINED_MINIMIZATION 15799b94acceSBarry Smith methods only. 15809b94acceSBarry Smith 158136851e7fSLois Curfman McInnes Level: intermediate 158236851e7fSLois Curfman McInnes 15839b94acceSBarry Smith .keywords: SNES, nonlinear, set, minimum, convergence, function, tolerance 15849b94acceSBarry Smith 1585d7a720efSLois Curfman McInnes .seealso: SNESSetTolerances(), SNESSetTrustRegionTolerance() 15869b94acceSBarry Smith @*/ 158777c4ece6SBarry Smith int SNESSetMinimizationFunctionTolerance(SNES snes,double ftol) 15889b94acceSBarry Smith { 15893a40ed3dSBarry Smith PetscFunctionBegin; 159077c4ece6SBarry Smith PetscValidHeaderSpecific(snes,SNES_COOKIE); 15919b94acceSBarry Smith snes->fmin = ftol; 15923a40ed3dSBarry Smith PetscFunctionReturn(0); 15939b94acceSBarry Smith } 15949b94acceSBarry Smith 1595ce1608b8SBarry Smith #undef __FUNC__ 1596ce1608b8SBarry Smith #define __FUNC__ "SNESLGMonitor" 1597ce1608b8SBarry Smith int SNESLGMonitor(SNES snes,int it,double norm,void *ctx) 1598ce1608b8SBarry Smith { 1599ce1608b8SBarry Smith int ierr; 1600ce1608b8SBarry Smith 1601ce1608b8SBarry Smith PetscFunctionBegin; 1602ce1608b8SBarry Smith ierr = KSPLGMonitor((KSP)snes,it,norm,ctx);CHKERRQ(ierr); 1603ce1608b8SBarry Smith PetscFunctionReturn(0); 1604ce1608b8SBarry Smith } 1605ce1608b8SBarry Smith 16069b94acceSBarry Smith /* ------------ Routines to set performance monitoring options ----------- */ 16079b94acceSBarry Smith 16085615d1e5SSatish Balay #undef __FUNC__ 1609d4bb536fSBarry Smith #define __FUNC__ "SNESSetMonitor" 16109b94acceSBarry Smith /*@C 16115cd90555SBarry Smith SNESSetMonitor - Sets an ADDITIONAL function that is to be used at every 16129b94acceSBarry Smith iteration of the nonlinear solver to display the iteration's 16139b94acceSBarry Smith progress. 16149b94acceSBarry Smith 1615fee21e36SBarry Smith Collective on SNES 1616fee21e36SBarry Smith 1617c7afd0dbSLois Curfman McInnes Input Parameters: 1618c7afd0dbSLois Curfman McInnes + snes - the SNES context 1619c7afd0dbSLois Curfman McInnes . func - monitoring routine 1620c7afd0dbSLois Curfman McInnes - mctx - [optional] user-defined context for private data for the 1621c7afd0dbSLois Curfman McInnes monitor routine (may be PETSC_NULL) 16229b94acceSBarry Smith 1623c7afd0dbSLois Curfman McInnes Calling sequence of func: 162440a0c1c6SLois Curfman McInnes $ int func(SNES snes,int its, double norm,void *mctx) 1625c7afd0dbSLois Curfman McInnes 1626c7afd0dbSLois Curfman McInnes + snes - the SNES context 1627c7afd0dbSLois Curfman McInnes . its - iteration number 1628c7afd0dbSLois Curfman McInnes . norm - 2-norm function value (may be estimated) 1629c7afd0dbSLois Curfman McInnes for SNES_NONLINEAR_EQUATIONS methods 163040a0c1c6SLois Curfman McInnes . norm - 2-norm gradient value (may be estimated) 1631c7afd0dbSLois Curfman McInnes for SNES_UNCONSTRAINED_MINIMIZATION methods 163240a0c1c6SLois Curfman McInnes - mctx - [optional] monitoring context 16339b94acceSBarry Smith 16349665c990SLois Curfman McInnes Options Database Keys: 1635c7afd0dbSLois Curfman McInnes + -snes_monitor - sets SNESDefaultMonitor() 1636c7afd0dbSLois Curfman McInnes . -snes_xmonitor - sets line graph monitor, 1637c7afd0dbSLois Curfman McInnes uses SNESLGMonitorCreate() 1638c7afd0dbSLois Curfman McInnes _ -snes_cancelmonitors - cancels all monitors that have 1639c7afd0dbSLois Curfman McInnes been hardwired into a code by 1640c7afd0dbSLois Curfman McInnes calls to SNESSetMonitor(), but 1641c7afd0dbSLois Curfman McInnes does not cancel those set via 1642c7afd0dbSLois Curfman McInnes the options database. 16439665c990SLois Curfman McInnes 1644639f9d9dSBarry Smith Notes: 16456bc08f3fSLois Curfman McInnes Several different monitoring routines may be set by calling 16466bc08f3fSLois Curfman McInnes SNESSetMonitor() multiple times; all will be called in the 16476bc08f3fSLois Curfman McInnes order in which they were set. 1648639f9d9dSBarry Smith 164936851e7fSLois Curfman McInnes Level: intermediate 165036851e7fSLois Curfman McInnes 16519b94acceSBarry Smith .keywords: SNES, nonlinear, set, monitor 16529b94acceSBarry Smith 16535cd90555SBarry Smith .seealso: SNESDefaultMonitor(), SNESClearMonitor() 16549b94acceSBarry Smith @*/ 165574679c65SBarry Smith int SNESSetMonitor( SNES snes, int (*func)(SNES,int,double,void*),void *mctx ) 16569b94acceSBarry Smith { 16573a40ed3dSBarry Smith PetscFunctionBegin; 1658639f9d9dSBarry Smith if (snes->numbermonitors >= MAXSNESMONITORS) { 1659a8c6a408SBarry Smith SETERRQ(PETSC_ERR_ARG_OUTOFRANGE,0,"Too many monitors set"); 1660639f9d9dSBarry Smith } 1661639f9d9dSBarry Smith 1662639f9d9dSBarry Smith snes->monitor[snes->numbermonitors] = func; 1663639f9d9dSBarry Smith snes->monitorcontext[snes->numbermonitors++] = (void*)mctx; 16643a40ed3dSBarry Smith PetscFunctionReturn(0); 16659b94acceSBarry Smith } 16669b94acceSBarry Smith 16675615d1e5SSatish Balay #undef __FUNC__ 16685cd90555SBarry Smith #define __FUNC__ "SNESClearMonitor" 16695cd90555SBarry Smith /*@C 16705cd90555SBarry Smith SNESClearMonitor - Clears all the monitor functions for a SNES object. 16715cd90555SBarry Smith 1672c7afd0dbSLois Curfman McInnes Collective on SNES 1673c7afd0dbSLois Curfman McInnes 16745cd90555SBarry Smith Input Parameters: 16755cd90555SBarry Smith . snes - the SNES context 16765cd90555SBarry Smith 16775cd90555SBarry Smith Options Database: 1678c7afd0dbSLois Curfman McInnes . -snes_cancelmonitors - cancels all monitors that have been hardwired 1679c7afd0dbSLois Curfman McInnes into a code by calls to SNESSetMonitor(), but does not cancel those 1680c7afd0dbSLois Curfman McInnes set via the options database 16815cd90555SBarry Smith 16825cd90555SBarry Smith Notes: 16835cd90555SBarry Smith There is no way to clear one specific monitor from a SNES object. 16845cd90555SBarry Smith 168536851e7fSLois Curfman McInnes Level: intermediate 168636851e7fSLois Curfman McInnes 16875cd90555SBarry Smith .keywords: SNES, nonlinear, set, monitor 16885cd90555SBarry Smith 16895cd90555SBarry Smith .seealso: SNESDefaultMonitor(), SNESSetMonitor() 16905cd90555SBarry Smith @*/ 16915cd90555SBarry Smith int SNESClearMonitor( SNES snes ) 16925cd90555SBarry Smith { 16935cd90555SBarry Smith PetscFunctionBegin; 16945cd90555SBarry Smith snes->numbermonitors = 0; 16955cd90555SBarry Smith PetscFunctionReturn(0); 16965cd90555SBarry Smith } 16975cd90555SBarry Smith 16985cd90555SBarry Smith #undef __FUNC__ 1699d4bb536fSBarry Smith #define __FUNC__ "SNESSetConvergenceTest" 17009b94acceSBarry Smith /*@C 17019b94acceSBarry Smith SNESSetConvergenceTest - Sets the function that is to be used 17029b94acceSBarry Smith to test for convergence of the nonlinear iterative solution. 17039b94acceSBarry Smith 1704fee21e36SBarry Smith Collective on SNES 1705fee21e36SBarry Smith 1706c7afd0dbSLois Curfman McInnes Input Parameters: 1707c7afd0dbSLois Curfman McInnes + snes - the SNES context 1708c7afd0dbSLois Curfman McInnes . func - routine to test for convergence 1709c7afd0dbSLois Curfman McInnes - cctx - [optional] context for private data for the convergence routine 1710c7afd0dbSLois Curfman McInnes (may be PETSC_NULL) 17119b94acceSBarry Smith 1712c7afd0dbSLois Curfman McInnes Calling sequence of func: 17138d76a1e5SLois Curfman McInnes $ int func (SNES snes,double xnorm,double gnorm,double f,void *cctx) 1714c7afd0dbSLois Curfman McInnes 1715c7afd0dbSLois Curfman McInnes + snes - the SNES context 1716c7afd0dbSLois Curfman McInnes . cctx - [optional] convergence context 1717c7afd0dbSLois Curfman McInnes . xnorm - 2-norm of current iterate 1718c7afd0dbSLois Curfman McInnes . gnorm - 2-norm of current step (SNES_NONLINEAR_EQUATIONS methods) 1719c7afd0dbSLois Curfman McInnes . f - 2-norm of function (SNES_NONLINEAR_EQUATIONS methods) 1720c7afd0dbSLois Curfman McInnes . gnorm - 2-norm of current gradient (SNES_UNCONSTRAINED_MINIMIZATION methods) 1721c7afd0dbSLois Curfman McInnes - f - function value (SNES_UNCONSTRAINED_MINIMIZATION methods) 17229b94acceSBarry Smith 172336851e7fSLois Curfman McInnes Level: advanced 172436851e7fSLois Curfman McInnes 17259b94acceSBarry Smith .keywords: SNES, nonlinear, set, convergence, test 17269b94acceSBarry Smith 172740191667SLois Curfman McInnes .seealso: SNESConverged_EQ_LS(), SNESConverged_EQ_TR(), 172840191667SLois Curfman McInnes SNESConverged_UM_LS(), SNESConverged_UM_TR() 17299b94acceSBarry Smith @*/ 173074679c65SBarry Smith int SNESSetConvergenceTest(SNES snes,int (*func)(SNES,double,double,double,void*),void *cctx) 17319b94acceSBarry Smith { 17323a40ed3dSBarry Smith PetscFunctionBegin; 17339b94acceSBarry Smith (snes)->converged = func; 17349b94acceSBarry Smith (snes)->cnvP = cctx; 17353a40ed3dSBarry Smith PetscFunctionReturn(0); 17369b94acceSBarry Smith } 17379b94acceSBarry Smith 17385615d1e5SSatish Balay #undef __FUNC__ 1739d4bb536fSBarry Smith #define __FUNC__ "SNESSetConvergenceHistory" 1740c9005455SLois Curfman McInnes /*@ 1741c9005455SLois Curfman McInnes SNESSetConvergenceHistory - Sets the array used to hold the convergence history. 1742c9005455SLois Curfman McInnes 1743fee21e36SBarry Smith Collective on SNES 1744fee21e36SBarry Smith 1745c7afd0dbSLois Curfman McInnes Input Parameters: 1746c7afd0dbSLois Curfman McInnes + snes - iterative context obtained from SNESCreate() 1747c7afd0dbSLois Curfman McInnes . a - array to hold history 1748758f92a0SBarry Smith . its - integer array holds the number of linear iterations (or 1749758f92a0SBarry Smith negative if not converged) for each solve. 1750758f92a0SBarry Smith . na - size of a and its 1751758f92a0SBarry Smith - reset - PETSC_TRUTH indicates each new nonlinear solve resets the history counter to zero, 1752758f92a0SBarry Smith else it continues storing new values for new nonlinear solves after the old ones 1753c7afd0dbSLois Curfman McInnes 1754c9005455SLois Curfman McInnes Notes: 1755c9005455SLois Curfman McInnes If set, this array will contain the function norms (for 1756c9005455SLois Curfman McInnes SNES_NONLINEAR_EQUATIONS methods) or gradient norms 1757c9005455SLois Curfman McInnes (for SNES_UNCONSTRAINED_MINIMIZATION methods) computed 1758c9005455SLois Curfman McInnes at each step. 1759c9005455SLois Curfman McInnes 1760c9005455SLois Curfman McInnes This routine is useful, e.g., when running a code for purposes 1761c9005455SLois Curfman McInnes of accurate performance monitoring, when no I/O should be done 1762c9005455SLois Curfman McInnes during the section of code that is being timed. 1763c9005455SLois Curfman McInnes 176436851e7fSLois Curfman McInnes Level: intermediate 176536851e7fSLois Curfman McInnes 1766c9005455SLois Curfman McInnes .keywords: SNES, set, convergence, history 1767758f92a0SBarry Smith 176808405cd6SLois Curfman McInnes .seealso: SNESGetConvergenceHistory() 1769758f92a0SBarry Smith 1770c9005455SLois Curfman McInnes @*/ 1771758f92a0SBarry Smith int SNESSetConvergenceHistory(SNES snes, double *a, int *its,int na,PetscTruth reset) 1772c9005455SLois Curfman McInnes { 17733a40ed3dSBarry Smith PetscFunctionBegin; 1774c9005455SLois Curfman McInnes PetscValidHeaderSpecific(snes,SNES_COOKIE); 1775c9005455SLois Curfman McInnes if (na) PetscValidScalarPointer(a); 1776c9005455SLois Curfman McInnes snes->conv_hist = a; 1777758f92a0SBarry Smith snes->conv_hist_its = its; 1778758f92a0SBarry Smith snes->conv_hist_max = na; 1779758f92a0SBarry Smith snes->conv_hist_reset = reset; 1780758f92a0SBarry Smith PetscFunctionReturn(0); 1781758f92a0SBarry Smith } 1782758f92a0SBarry Smith 1783758f92a0SBarry Smith #undef __FUNC__ 1784758f92a0SBarry Smith #define __FUNC__ "SNESGetConvergenceHistory" 17850c4c9dddSBarry Smith /*@C 1786758f92a0SBarry Smith SNESGetConvergenceHistory - Gets the array used to hold the convergence history. 1787758f92a0SBarry Smith 1788758f92a0SBarry Smith Collective on SNES 1789758f92a0SBarry Smith 1790758f92a0SBarry Smith Input Parameter: 1791758f92a0SBarry Smith . snes - iterative context obtained from SNESCreate() 1792758f92a0SBarry Smith 1793758f92a0SBarry Smith Output Parameters: 1794758f92a0SBarry Smith . a - array to hold history 1795758f92a0SBarry Smith . its - integer array holds the number of linear iterations (or 1796758f92a0SBarry Smith negative if not converged) for each solve. 1797758f92a0SBarry Smith - na - size of a and its 1798758f92a0SBarry Smith 1799758f92a0SBarry Smith Notes: 1800758f92a0SBarry Smith The calling sequence for this routine in Fortran is 1801758f92a0SBarry Smith $ call SNESGetConvergenceHistory(SNES snes, integer na, integer ierr) 1802758f92a0SBarry Smith 1803758f92a0SBarry Smith This routine is useful, e.g., when running a code for purposes 1804758f92a0SBarry Smith of accurate performance monitoring, when no I/O should be done 1805758f92a0SBarry Smith during the section of code that is being timed. 1806758f92a0SBarry Smith 1807758f92a0SBarry Smith Level: intermediate 1808758f92a0SBarry Smith 1809758f92a0SBarry Smith .keywords: SNES, get, convergence, history 1810758f92a0SBarry Smith 1811758f92a0SBarry Smith .seealso: SNESSetConvergencHistory() 1812758f92a0SBarry Smith 1813758f92a0SBarry Smith @*/ 1814758f92a0SBarry Smith int SNESGetConvergenceHistory(SNES snes, double **a, int **its,int *na) 1815758f92a0SBarry Smith { 1816758f92a0SBarry Smith PetscFunctionBegin; 1817758f92a0SBarry Smith PetscValidHeaderSpecific(snes,SNES_COOKIE); 1818758f92a0SBarry Smith if (a) *a = snes->conv_hist; 1819758f92a0SBarry Smith if (its) *its = snes->conv_hist_its; 1820758f92a0SBarry Smith if (na) *na = snes->conv_hist_len; 18213a40ed3dSBarry Smith PetscFunctionReturn(0); 1822c9005455SLois Curfman McInnes } 1823c9005455SLois Curfman McInnes 1824c9005455SLois Curfman McInnes #undef __FUNC__ 18255615d1e5SSatish Balay #define __FUNC__ "SNESScaleStep_Private" 18269b94acceSBarry Smith /* 18279b94acceSBarry Smith SNESScaleStep_Private - Scales a step so that its length is less than the 18289b94acceSBarry Smith positive parameter delta. 18299b94acceSBarry Smith 18309b94acceSBarry Smith Input Parameters: 1831c7afd0dbSLois Curfman McInnes + snes - the SNES context 18329b94acceSBarry Smith . y - approximate solution of linear system 18339b94acceSBarry Smith . fnorm - 2-norm of current function 1834c7afd0dbSLois Curfman McInnes - delta - trust region size 18359b94acceSBarry Smith 18369b94acceSBarry Smith Output Parameters: 1837c7afd0dbSLois Curfman McInnes + gpnorm - predicted function norm at the new point, assuming local 18389b94acceSBarry Smith linearization. The value is zero if the step lies within the trust 18399b94acceSBarry Smith region, and exceeds zero otherwise. 1840c7afd0dbSLois Curfman McInnes - ynorm - 2-norm of the step 18419b94acceSBarry Smith 18429b94acceSBarry Smith Note: 184340191667SLois Curfman McInnes For non-trust region methods such as SNES_EQ_LS, the parameter delta 18449b94acceSBarry Smith is set to be the maximum allowable step size. 18459b94acceSBarry Smith 18469b94acceSBarry Smith .keywords: SNES, nonlinear, scale, step 18479b94acceSBarry Smith */ 18489b94acceSBarry Smith int SNESScaleStep_Private(SNES snes,Vec y,double *fnorm,double *delta, 18499b94acceSBarry Smith double *gpnorm,double *ynorm) 18509b94acceSBarry Smith { 18519b94acceSBarry Smith double norm; 18529b94acceSBarry Smith Scalar cnorm; 18533a40ed3dSBarry Smith int ierr; 18543a40ed3dSBarry Smith 18553a40ed3dSBarry Smith PetscFunctionBegin; 18563a40ed3dSBarry Smith ierr = VecNorm(y,NORM_2, &norm );CHKERRQ(ierr); 18579b94acceSBarry Smith if (norm > *delta) { 18589b94acceSBarry Smith norm = *delta/norm; 18599b94acceSBarry Smith *gpnorm = (1.0 - norm)*(*fnorm); 18609b94acceSBarry Smith cnorm = norm; 18619b94acceSBarry Smith VecScale( &cnorm, y ); 18629b94acceSBarry Smith *ynorm = *delta; 18639b94acceSBarry Smith } else { 18649b94acceSBarry Smith *gpnorm = 0.0; 18659b94acceSBarry Smith *ynorm = norm; 18669b94acceSBarry Smith } 18673a40ed3dSBarry Smith PetscFunctionReturn(0); 18689b94acceSBarry Smith } 18699b94acceSBarry Smith 18705615d1e5SSatish Balay #undef __FUNC__ 18715615d1e5SSatish Balay #define __FUNC__ "SNESSolve" 18729b94acceSBarry Smith /*@ 18739b94acceSBarry Smith SNESSolve - Solves a nonlinear system. Call SNESSolve after calling 187463a78c88SLois Curfman McInnes SNESCreate() and optional routines of the form SNESSetXXX(). 18759b94acceSBarry Smith 1876c7afd0dbSLois Curfman McInnes Collective on SNES 1877c7afd0dbSLois Curfman McInnes 1878b2002411SLois Curfman McInnes Input Parameters: 1879c7afd0dbSLois Curfman McInnes + snes - the SNES context 1880c7afd0dbSLois Curfman McInnes - x - the solution vector 18819b94acceSBarry Smith 18829b94acceSBarry Smith Output Parameter: 1883b2002411SLois Curfman McInnes . its - number of iterations until termination 18849b94acceSBarry Smith 1885b2002411SLois Curfman McInnes Notes: 18868ddd3da0SLois Curfman McInnes The user should initialize the vector, x, with the initial guess 18878ddd3da0SLois Curfman McInnes for the nonlinear solve prior to calling SNESSolve. In particular, 18888ddd3da0SLois Curfman McInnes to employ an initial guess of zero, the user should explicitly set 18898ddd3da0SLois Curfman McInnes this vector to zero by calling VecSet(). 18908ddd3da0SLois Curfman McInnes 189136851e7fSLois Curfman McInnes Level: beginner 189236851e7fSLois Curfman McInnes 18939b94acceSBarry Smith .keywords: SNES, nonlinear, solve 18949b94acceSBarry Smith 189563a78c88SLois Curfman McInnes .seealso: SNESCreate(), SNESDestroy() 18969b94acceSBarry Smith @*/ 18978ddd3da0SLois Curfman McInnes int SNESSolve(SNES snes,Vec x,int *its) 18989b94acceSBarry Smith { 18993c7409f5SSatish Balay int ierr, flg; 1900052efed2SBarry Smith 19013a40ed3dSBarry Smith PetscFunctionBegin; 190277c4ece6SBarry Smith PetscValidHeaderSpecific(snes,SNES_COOKIE); 190374679c65SBarry Smith PetscValidIntPointer(its); 190482bf6240SBarry Smith if (!snes->setupcalled) {ierr = SNESSetUp(snes,x); CHKERRQ(ierr);} 1905c4fc05e7SBarry Smith else {snes->vec_sol = snes->vec_sol_always = x;} 1906758f92a0SBarry Smith if (snes->conv_hist_reset == PETSC_TRUE) snes->conv_hist_len = 0; 19079b94acceSBarry Smith PLogEventBegin(SNES_Solve,snes,0,0,0); 1908c96a6f78SLois Curfman McInnes snes->nfuncs = 0; snes->linear_its = 0; snes->nfailures = 0; 19099b94acceSBarry Smith ierr = (*(snes)->solve)(snes,its); CHKERRQ(ierr); 19109b94acceSBarry Smith PLogEventEnd(SNES_Solve,snes,0,0,0); 19113c7409f5SSatish Balay ierr = OptionsHasName(PETSC_NULL,"-snes_view", &flg); CHKERRQ(ierr); 19126d4a8577SBarry Smith if (flg) { ierr = SNESView(snes,VIEWER_STDOUT_WORLD); CHKERRQ(ierr); } 19133a40ed3dSBarry Smith PetscFunctionReturn(0); 19149b94acceSBarry Smith } 19159b94acceSBarry Smith 19169b94acceSBarry Smith /* --------- Internal routines for SNES Package --------- */ 19179b94acceSBarry Smith 19185615d1e5SSatish Balay #undef __FUNC__ 19195615d1e5SSatish Balay #define __FUNC__ "SNESSetType" 192082bf6240SBarry Smith /*@C 19214b0e389bSBarry Smith SNESSetType - Sets the method for the nonlinear solver. 19229b94acceSBarry Smith 1923fee21e36SBarry Smith Collective on SNES 1924fee21e36SBarry Smith 1925c7afd0dbSLois Curfman McInnes Input Parameters: 1926c7afd0dbSLois Curfman McInnes + snes - the SNES context 1927c7afd0dbSLois Curfman McInnes - method - a known method 1928c7afd0dbSLois Curfman McInnes 1929c7afd0dbSLois Curfman McInnes Options Database Key: 1930c7afd0dbSLois Curfman McInnes . -snes_type <method> - Sets the method; use -help for a list 1931c7afd0dbSLois Curfman McInnes of available methods (for instance, ls or tr) 1932ae12b187SLois Curfman McInnes 19339b94acceSBarry Smith Notes: 19349b94acceSBarry Smith See "petsc/include/snes.h" for available methods (for instance) 193536851e7fSLois Curfman McInnes + SNES_EQ_LS - Newton's method with line search 1936c7afd0dbSLois Curfman McInnes (systems of nonlinear equations) 1937c7afd0dbSLois Curfman McInnes . SNES_EQ_TR - Newton's method with trust region 1938c7afd0dbSLois Curfman McInnes (systems of nonlinear equations) 1939c7afd0dbSLois Curfman McInnes . SNES_UM_TR - Newton's method with trust region 1940c7afd0dbSLois Curfman McInnes (unconstrained minimization) 194136851e7fSLois Curfman McInnes - SNES_UM_LS - Newton's method with line search 1942c7afd0dbSLois Curfman McInnes (unconstrained minimization) 19439b94acceSBarry Smith 1944ae12b187SLois Curfman McInnes Normally, it is best to use the SNESSetFromOptions() command and then 1945ae12b187SLois Curfman McInnes set the SNES solver type from the options database rather than by using 1946ae12b187SLois Curfman McInnes this routine. Using the options database provides the user with 1947ae12b187SLois Curfman McInnes maximum flexibility in evaluating the many nonlinear solvers. 1948ae12b187SLois Curfman McInnes The SNESSetType() routine is provided for those situations where it 1949ae12b187SLois Curfman McInnes is necessary to set the nonlinear solver independently of the command 1950ae12b187SLois Curfman McInnes line or options database. This might be the case, for example, when 1951ae12b187SLois Curfman McInnes the choice of solver changes during the execution of the program, 1952ae12b187SLois Curfman McInnes and the user's application is taking responsibility for choosing the 195336851e7fSLois Curfman McInnes appropriate method. In other words, this routine is not for beginners. 195436851e7fSLois Curfman McInnes 195536851e7fSLois Curfman McInnes Level: intermediate 1956a703fe33SLois Curfman McInnes 1957f536c699SSatish Balay .keywords: SNES, set, method 19589b94acceSBarry Smith @*/ 19594b0e389bSBarry Smith int SNESSetType(SNES snes,SNESType method) 19609b94acceSBarry Smith { 196184cb2905SBarry Smith int ierr; 19629b94acceSBarry Smith int (*r)(SNES); 19633a40ed3dSBarry Smith 19643a40ed3dSBarry Smith PetscFunctionBegin; 196577c4ece6SBarry Smith PetscValidHeaderSpecific(snes,SNES_COOKIE); 196682bf6240SBarry Smith 19673f1db9ecSBarry Smith if (PetscTypeCompare(snes->type_name,method)) PetscFunctionReturn(0); 196882bf6240SBarry Smith 196982bf6240SBarry Smith if (snes->setupcalled) { 1970e1311b90SBarry Smith ierr = (*(snes)->destroy)(snes);CHKERRQ(ierr); 197182bf6240SBarry Smith snes->data = 0; 197282bf6240SBarry Smith } 19737d1a2b2bSBarry Smith 19749b94acceSBarry Smith /* Get the function pointers for the iterative method requested */ 197582bf6240SBarry Smith if (!SNESRegisterAllCalled) {ierr = SNESRegisterAll(PETSC_NULL); CHKERRQ(ierr);} 197682bf6240SBarry Smith 1977488ecbafSBarry Smith ierr = FListFind(snes->comm, SNESList, method,(int (**)(void *)) &r );CHKERRQ(ierr); 197882bf6240SBarry Smith 1979596552b5SBarry Smith if (!r) SETERRQ1(1,1,"Unable to find requested SNES type %s",method); 19801548b14aSBarry Smith 19810452661fSBarry Smith if (snes->data) PetscFree(snes->data); 198282bf6240SBarry Smith snes->data = 0; 19833a40ed3dSBarry Smith ierr = (*r)(snes); CHKERRQ(ierr); 198482bf6240SBarry Smith 198582bf6240SBarry Smith if (snes->type_name) PetscFree(snes->type_name); 198682bf6240SBarry Smith snes->type_name = (char *) PetscMalloc((PetscStrlen(method)+1)*sizeof(char));CHKPTRQ(snes->type_name); 198782bf6240SBarry Smith PetscStrcpy(snes->type_name,method); 198882bf6240SBarry Smith snes->set_method_called = 1; 198982bf6240SBarry Smith 19903a40ed3dSBarry Smith PetscFunctionReturn(0); 19919b94acceSBarry Smith } 19929b94acceSBarry Smith 1993a847f771SSatish Balay 19949b94acceSBarry Smith /* --------------------------------------------------------------------- */ 19955615d1e5SSatish Balay #undef __FUNC__ 1996d4bb536fSBarry Smith #define __FUNC__ "SNESRegisterDestroy" 19979b94acceSBarry Smith /*@C 19989b94acceSBarry Smith SNESRegisterDestroy - Frees the list of nonlinear solvers that were 19999b94acceSBarry Smith registered by SNESRegister(). 20009b94acceSBarry Smith 2001fee21e36SBarry Smith Not Collective 2002fee21e36SBarry Smith 200336851e7fSLois Curfman McInnes Level: advanced 200436851e7fSLois Curfman McInnes 20059b94acceSBarry Smith .keywords: SNES, nonlinear, register, destroy 20069b94acceSBarry Smith 20079b94acceSBarry Smith .seealso: SNESRegisterAll(), SNESRegisterAll() 20089b94acceSBarry Smith @*/ 2009cf256101SBarry Smith int SNESRegisterDestroy(void) 20109b94acceSBarry Smith { 201182bf6240SBarry Smith int ierr; 201282bf6240SBarry Smith 20133a40ed3dSBarry Smith PetscFunctionBegin; 201482bf6240SBarry Smith if (SNESList) { 2015488ecbafSBarry Smith ierr = FListDestroy( SNESList );CHKERRQ(ierr); 201682bf6240SBarry Smith SNESList = 0; 20179b94acceSBarry Smith } 201884cb2905SBarry Smith SNESRegisterAllCalled = 0; 20193a40ed3dSBarry Smith PetscFunctionReturn(0); 20209b94acceSBarry Smith } 20219b94acceSBarry Smith 20225615d1e5SSatish Balay #undef __FUNC__ 2023d4bb536fSBarry Smith #define __FUNC__ "SNESGetType" 20249b94acceSBarry Smith /*@C 20259a28b0a6SLois Curfman McInnes SNESGetType - Gets the SNES method type and name (as a string). 20269b94acceSBarry Smith 2027c7afd0dbSLois Curfman McInnes Not Collective 2028c7afd0dbSLois Curfman McInnes 20299b94acceSBarry Smith Input Parameter: 20304b0e389bSBarry Smith . snes - nonlinear solver context 20319b94acceSBarry Smith 20329b94acceSBarry Smith Output Parameter: 203382bf6240SBarry Smith . method - SNES method (a charactor string) 20349b94acceSBarry Smith 203536851e7fSLois Curfman McInnes Level: intermediate 203636851e7fSLois Curfman McInnes 20379b94acceSBarry Smith .keywords: SNES, nonlinear, get, method, name 20389b94acceSBarry Smith @*/ 203982bf6240SBarry Smith int SNESGetType(SNES snes, SNESType *method) 20409b94acceSBarry Smith { 20413a40ed3dSBarry Smith PetscFunctionBegin; 204282bf6240SBarry Smith *method = snes->type_name; 20433a40ed3dSBarry Smith PetscFunctionReturn(0); 20449b94acceSBarry Smith } 20459b94acceSBarry Smith 20465615d1e5SSatish Balay #undef __FUNC__ 2047d4bb536fSBarry Smith #define __FUNC__ "SNESGetSolution" 20489b94acceSBarry Smith /*@C 20499b94acceSBarry Smith SNESGetSolution - Returns the vector where the approximate solution is 20509b94acceSBarry Smith stored. 20519b94acceSBarry Smith 2052c7afd0dbSLois Curfman McInnes Not Collective, but Vec is parallel if SNES is parallel 2053c7afd0dbSLois Curfman McInnes 20549b94acceSBarry Smith Input Parameter: 20559b94acceSBarry Smith . snes - the SNES context 20569b94acceSBarry Smith 20579b94acceSBarry Smith Output Parameter: 20589b94acceSBarry Smith . x - the solution 20599b94acceSBarry Smith 206036851e7fSLois Curfman McInnes Level: advanced 206136851e7fSLois Curfman McInnes 20629b94acceSBarry Smith .keywords: SNES, nonlinear, get, solution 20639b94acceSBarry Smith 2064a86d99e1SLois Curfman McInnes .seealso: SNESGetFunction(), SNESGetGradient(), SNESGetSolutionUpdate() 20659b94acceSBarry Smith @*/ 20669b94acceSBarry Smith int SNESGetSolution(SNES snes,Vec *x) 20679b94acceSBarry Smith { 20683a40ed3dSBarry Smith PetscFunctionBegin; 206977c4ece6SBarry Smith PetscValidHeaderSpecific(snes,SNES_COOKIE); 20709b94acceSBarry Smith *x = snes->vec_sol_always; 20713a40ed3dSBarry Smith PetscFunctionReturn(0); 20729b94acceSBarry Smith } 20739b94acceSBarry Smith 20745615d1e5SSatish Balay #undef __FUNC__ 2075d4bb536fSBarry Smith #define __FUNC__ "SNESGetSolutionUpdate" 20769b94acceSBarry Smith /*@C 20779b94acceSBarry Smith SNESGetSolutionUpdate - Returns the vector where the solution update is 20789b94acceSBarry Smith stored. 20799b94acceSBarry Smith 2080c7afd0dbSLois Curfman McInnes Not Collective, but Vec is parallel if SNES is parallel 2081c7afd0dbSLois Curfman McInnes 20829b94acceSBarry Smith Input Parameter: 20839b94acceSBarry Smith . snes - the SNES context 20849b94acceSBarry Smith 20859b94acceSBarry Smith Output Parameter: 20869b94acceSBarry Smith . x - the solution update 20879b94acceSBarry Smith 208836851e7fSLois Curfman McInnes Level: advanced 208936851e7fSLois Curfman McInnes 20909b94acceSBarry Smith .keywords: SNES, nonlinear, get, solution, update 20919b94acceSBarry Smith 20929b94acceSBarry Smith .seealso: SNESGetSolution(), SNESGetFunction 20939b94acceSBarry Smith @*/ 20949b94acceSBarry Smith int SNESGetSolutionUpdate(SNES snes,Vec *x) 20959b94acceSBarry Smith { 20963a40ed3dSBarry Smith PetscFunctionBegin; 209777c4ece6SBarry Smith PetscValidHeaderSpecific(snes,SNES_COOKIE); 20989b94acceSBarry Smith *x = snes->vec_sol_update_always; 20993a40ed3dSBarry Smith PetscFunctionReturn(0); 21009b94acceSBarry Smith } 21019b94acceSBarry Smith 21025615d1e5SSatish Balay #undef __FUNC__ 2103d4bb536fSBarry Smith #define __FUNC__ "SNESGetFunction" 21049b94acceSBarry Smith /*@C 21053638b69dSLois Curfman McInnes SNESGetFunction - Returns the vector where the function is stored. 21069b94acceSBarry Smith 2107c7afd0dbSLois Curfman McInnes Not Collective, but Vec is parallel if SNES is parallel 2108c7afd0dbSLois Curfman McInnes 21099b94acceSBarry Smith Input Parameter: 21109b94acceSBarry Smith . snes - the SNES context 21119b94acceSBarry Smith 21129b94acceSBarry Smith Output Parameter: 21133638b69dSLois Curfman McInnes . r - the function 21149b94acceSBarry Smith 21159b94acceSBarry Smith Notes: 21169b94acceSBarry Smith SNESGetFunction() is valid for SNES_NONLINEAR_EQUATIONS methods only 21179b94acceSBarry Smith Analogous routines for SNES_UNCONSTRAINED_MINIMIZATION methods are 21189b94acceSBarry Smith SNESGetMinimizationFunction() and SNESGetGradient(); 21199b94acceSBarry Smith 212036851e7fSLois Curfman McInnes Level: advanced 212136851e7fSLois Curfman McInnes 2122a86d99e1SLois Curfman McInnes .keywords: SNES, nonlinear, get, function 21239b94acceSBarry Smith 212431615d04SLois Curfman McInnes .seealso: SNESSetFunction(), SNESGetSolution(), SNESGetMinimizationFunction(), 212531615d04SLois Curfman McInnes SNESGetGradient() 21269b94acceSBarry Smith @*/ 21279b94acceSBarry Smith int SNESGetFunction(SNES snes,Vec *r) 21289b94acceSBarry Smith { 21293a40ed3dSBarry Smith PetscFunctionBegin; 213077c4ece6SBarry Smith PetscValidHeaderSpecific(snes,SNES_COOKIE); 2131a8c6a408SBarry Smith if (snes->method_class != SNES_NONLINEAR_EQUATIONS) { 2132a8c6a408SBarry Smith SETERRQ(PETSC_ERR_ARG_WRONG,0,"For SNES_NONLINEAR_EQUATIONS only"); 2133a8c6a408SBarry Smith } 21349b94acceSBarry Smith *r = snes->vec_func_always; 21353a40ed3dSBarry Smith PetscFunctionReturn(0); 21369b94acceSBarry Smith } 21379b94acceSBarry Smith 21385615d1e5SSatish Balay #undef __FUNC__ 2139d4bb536fSBarry Smith #define __FUNC__ "SNESGetGradient" 21409b94acceSBarry Smith /*@C 21413638b69dSLois Curfman McInnes SNESGetGradient - Returns the vector where the gradient is stored. 21429b94acceSBarry Smith 2143c7afd0dbSLois Curfman McInnes Not Collective, but Vec is parallel if SNES is parallel 2144c7afd0dbSLois Curfman McInnes 21459b94acceSBarry Smith Input Parameter: 21469b94acceSBarry Smith . snes - the SNES context 21479b94acceSBarry Smith 21489b94acceSBarry Smith Output Parameter: 21499b94acceSBarry Smith . r - the gradient 21509b94acceSBarry Smith 21519b94acceSBarry Smith Notes: 21529b94acceSBarry Smith SNESGetGradient() is valid for SNES_UNCONSTRAINED_MINIMIZATION methods 21539b94acceSBarry Smith only. An analogous routine for SNES_NONLINEAR_EQUATIONS methods is 21549b94acceSBarry Smith SNESGetFunction(). 21559b94acceSBarry Smith 215636851e7fSLois Curfman McInnes Level: advanced 215736851e7fSLois Curfman McInnes 21589b94acceSBarry Smith .keywords: SNES, nonlinear, get, gradient 21599b94acceSBarry Smith 216031615d04SLois Curfman McInnes .seealso: SNESGetMinimizationFunction(), SNESGetSolution(), SNESGetFunction() 21619b94acceSBarry Smith @*/ 21629b94acceSBarry Smith int SNESGetGradient(SNES snes,Vec *r) 21639b94acceSBarry Smith { 21643a40ed3dSBarry Smith PetscFunctionBegin; 216577c4ece6SBarry Smith PetscValidHeaderSpecific(snes,SNES_COOKIE); 21663a40ed3dSBarry Smith if (snes->method_class != SNES_UNCONSTRAINED_MINIMIZATION) { 2167a8c6a408SBarry Smith SETERRQ(PETSC_ERR_ARG_WRONG,0,"For SNES_UNCONSTRAINED_MINIMIZATION only"); 21683a40ed3dSBarry Smith } 21699b94acceSBarry Smith *r = snes->vec_func_always; 21703a40ed3dSBarry Smith PetscFunctionReturn(0); 21719b94acceSBarry Smith } 21729b94acceSBarry Smith 21735615d1e5SSatish Balay #undef __FUNC__ 2174d4bb536fSBarry Smith #define __FUNC__ "SNESGetMinimizationFunction" 21759b94acceSBarry Smith /*@ 21769b94acceSBarry Smith SNESGetMinimizationFunction - Returns the scalar function value for 21779b94acceSBarry Smith unconstrained minimization problems. 21789b94acceSBarry Smith 2179c7afd0dbSLois Curfman McInnes Not Collective 2180c7afd0dbSLois Curfman McInnes 21819b94acceSBarry Smith Input Parameter: 21829b94acceSBarry Smith . snes - the SNES context 21839b94acceSBarry Smith 21849b94acceSBarry Smith Output Parameter: 21859b94acceSBarry Smith . r - the function 21869b94acceSBarry Smith 21879b94acceSBarry Smith Notes: 21889b94acceSBarry Smith SNESGetMinimizationFunction() is valid for SNES_UNCONSTRAINED_MINIMIZATION 21899b94acceSBarry Smith methods only. An analogous routine for SNES_NONLINEAR_EQUATIONS methods is 21909b94acceSBarry Smith SNESGetFunction(). 21919b94acceSBarry Smith 219236851e7fSLois Curfman McInnes Level: advanced 219336851e7fSLois Curfman McInnes 21949b94acceSBarry Smith .keywords: SNES, nonlinear, get, function 21959b94acceSBarry Smith 219631615d04SLois Curfman McInnes .seealso: SNESGetGradient(), SNESGetSolution(), SNESGetFunction() 21979b94acceSBarry Smith @*/ 21989b94acceSBarry Smith int SNESGetMinimizationFunction(SNES snes,double *r) 21999b94acceSBarry Smith { 22003a40ed3dSBarry Smith PetscFunctionBegin; 220177c4ece6SBarry Smith PetscValidHeaderSpecific(snes,SNES_COOKIE); 220274679c65SBarry Smith PetscValidScalarPointer(r); 22033a40ed3dSBarry Smith if (snes->method_class != SNES_UNCONSTRAINED_MINIMIZATION) { 2204a8c6a408SBarry Smith SETERRQ(PETSC_ERR_ARG_WRONG,0,"For SNES_UNCONSTRAINED_MINIMIZATION only"); 22053a40ed3dSBarry Smith } 22069b94acceSBarry Smith *r = snes->fc; 22073a40ed3dSBarry Smith PetscFunctionReturn(0); 22089b94acceSBarry Smith } 22099b94acceSBarry Smith 22105615d1e5SSatish Balay #undef __FUNC__ 2211d4bb536fSBarry Smith #define __FUNC__ "SNESSetOptionsPrefix" 22123c7409f5SSatish Balay /*@C 22133c7409f5SSatish Balay SNESSetOptionsPrefix - Sets the prefix used for searching for all 2214d850072dSLois Curfman McInnes SNES options in the database. 22153c7409f5SSatish Balay 2216fee21e36SBarry Smith Collective on SNES 2217fee21e36SBarry Smith 2218c7afd0dbSLois Curfman McInnes Input Parameter: 2219c7afd0dbSLois Curfman McInnes + snes - the SNES context 2220c7afd0dbSLois Curfman McInnes - prefix - the prefix to prepend to all option names 2221c7afd0dbSLois Curfman McInnes 2222d850072dSLois Curfman McInnes Notes: 2223a83b1b31SSatish Balay A hyphen (-) must NOT be given at the beginning of the prefix name. 2224c7afd0dbSLois Curfman McInnes The first character of all runtime options is AUTOMATICALLY the hyphen. 2225d850072dSLois Curfman McInnes 222636851e7fSLois Curfman McInnes Level: advanced 222736851e7fSLois Curfman McInnes 22283c7409f5SSatish Balay .keywords: SNES, set, options, prefix, database 2229a86d99e1SLois Curfman McInnes 2230a86d99e1SLois Curfman McInnes .seealso: SNESSetFromOptions() 22313c7409f5SSatish Balay @*/ 22323c7409f5SSatish Balay int SNESSetOptionsPrefix(SNES snes,char *prefix) 22333c7409f5SSatish Balay { 22343c7409f5SSatish Balay int ierr; 22353c7409f5SSatish Balay 22363a40ed3dSBarry Smith PetscFunctionBegin; 223777c4ece6SBarry Smith PetscValidHeaderSpecific(snes,SNES_COOKIE); 2238639f9d9dSBarry Smith ierr = PetscObjectSetOptionsPrefix((PetscObject)snes, prefix); CHKERRQ(ierr); 22393c7409f5SSatish Balay ierr = SLESSetOptionsPrefix(snes->sles,prefix);CHKERRQ(ierr); 22403a40ed3dSBarry Smith PetscFunctionReturn(0); 22413c7409f5SSatish Balay } 22423c7409f5SSatish Balay 22435615d1e5SSatish Balay #undef __FUNC__ 2244d4bb536fSBarry Smith #define __FUNC__ "SNESAppendOptionsPrefix" 22453c7409f5SSatish Balay /*@C 2246f525115eSLois Curfman McInnes SNESAppendOptionsPrefix - Appends to the prefix used for searching for all 2247d850072dSLois Curfman McInnes SNES options in the database. 22483c7409f5SSatish Balay 2249fee21e36SBarry Smith Collective on SNES 2250fee21e36SBarry Smith 2251c7afd0dbSLois Curfman McInnes Input Parameters: 2252c7afd0dbSLois Curfman McInnes + snes - the SNES context 2253c7afd0dbSLois Curfman McInnes - prefix - the prefix to prepend to all option names 2254c7afd0dbSLois Curfman McInnes 2255d850072dSLois Curfman McInnes Notes: 2256a83b1b31SSatish Balay A hyphen (-) must NOT be given at the beginning of the prefix name. 2257c7afd0dbSLois Curfman McInnes The first character of all runtime options is AUTOMATICALLY the hyphen. 2258d850072dSLois Curfman McInnes 225936851e7fSLois Curfman McInnes Level: advanced 226036851e7fSLois Curfman McInnes 22613c7409f5SSatish Balay .keywords: SNES, append, options, prefix, database 2262a86d99e1SLois Curfman McInnes 2263a86d99e1SLois Curfman McInnes .seealso: SNESGetOptionsPrefix() 22643c7409f5SSatish Balay @*/ 22653c7409f5SSatish Balay int SNESAppendOptionsPrefix(SNES snes,char *prefix) 22663c7409f5SSatish Balay { 22673c7409f5SSatish Balay int ierr; 22683c7409f5SSatish Balay 22693a40ed3dSBarry Smith PetscFunctionBegin; 227077c4ece6SBarry Smith PetscValidHeaderSpecific(snes,SNES_COOKIE); 2271639f9d9dSBarry Smith ierr = PetscObjectAppendOptionsPrefix((PetscObject)snes, prefix); CHKERRQ(ierr); 22723c7409f5SSatish Balay ierr = SLESAppendOptionsPrefix(snes->sles,prefix);CHKERRQ(ierr); 22733a40ed3dSBarry Smith PetscFunctionReturn(0); 22743c7409f5SSatish Balay } 22753c7409f5SSatish Balay 22765615d1e5SSatish Balay #undef __FUNC__ 2277d4bb536fSBarry Smith #define __FUNC__ "SNESGetOptionsPrefix" 2278c83590e2SSatish Balay /*@ 22793c7409f5SSatish Balay SNESGetOptionsPrefix - Sets the prefix used for searching for all 22803c7409f5SSatish Balay SNES options in the database. 22813c7409f5SSatish Balay 2282c7afd0dbSLois Curfman McInnes Not Collective 2283c7afd0dbSLois Curfman McInnes 22843c7409f5SSatish Balay Input Parameter: 22853c7409f5SSatish Balay . snes - the SNES context 22863c7409f5SSatish Balay 22873c7409f5SSatish Balay Output Parameter: 22883c7409f5SSatish Balay . prefix - pointer to the prefix string used 22893c7409f5SSatish Balay 229036851e7fSLois Curfman McInnes Level: advanced 229136851e7fSLois Curfman McInnes 22923c7409f5SSatish Balay .keywords: SNES, get, options, prefix, database 2293a86d99e1SLois Curfman McInnes 2294a86d99e1SLois Curfman McInnes .seealso: SNESAppendOptionsPrefix() 22953c7409f5SSatish Balay @*/ 22963c7409f5SSatish Balay int SNESGetOptionsPrefix(SNES snes,char **prefix) 22973c7409f5SSatish Balay { 22983c7409f5SSatish Balay int ierr; 22993c7409f5SSatish Balay 23003a40ed3dSBarry Smith PetscFunctionBegin; 230177c4ece6SBarry Smith PetscValidHeaderSpecific(snes,SNES_COOKIE); 2302639f9d9dSBarry Smith ierr = PetscObjectGetOptionsPrefix((PetscObject)snes, prefix); CHKERRQ(ierr); 23033a40ed3dSBarry Smith PetscFunctionReturn(0); 23043c7409f5SSatish Balay } 23053c7409f5SSatish Balay 2306ca161407SBarry Smith #undef __FUNC__ 2307ca161407SBarry Smith #define __FUNC__ "SNESPrintHelp" 2308ca161407SBarry Smith /*@ 2309ca161407SBarry Smith SNESPrintHelp - Prints all options for the SNES component. 2310ca161407SBarry Smith 2311c7afd0dbSLois Curfman McInnes Collective on SNES 2312c7afd0dbSLois Curfman McInnes 2313ca161407SBarry Smith Input Parameter: 2314ca161407SBarry Smith . snes - the SNES context 2315ca161407SBarry Smith 2316ca161407SBarry Smith Options Database Keys: 2317c7afd0dbSLois Curfman McInnes + -help - Prints SNES options 2318c7afd0dbSLois Curfman McInnes - -h - Prints SNES options 2319ca161407SBarry Smith 232036851e7fSLois Curfman McInnes Level: beginner 232136851e7fSLois Curfman McInnes 2322ca161407SBarry Smith .keywords: SNES, nonlinear, help 2323ca161407SBarry Smith 2324ca161407SBarry Smith .seealso: SNESSetFromOptions() 2325ca161407SBarry Smith @*/ 2326ca161407SBarry Smith int SNESPrintHelp(SNES snes) 2327ca161407SBarry Smith { 2328ca161407SBarry Smith char p[64]; 2329ca161407SBarry Smith SNES_KSP_EW_ConvCtx *kctx; 2330ca161407SBarry Smith int ierr; 2331ca161407SBarry Smith 2332ca161407SBarry Smith PetscFunctionBegin; 2333ca161407SBarry Smith PetscValidHeaderSpecific(snes,SNES_COOKIE); 2334ca161407SBarry Smith 2335ca161407SBarry Smith PetscStrcpy(p,"-"); 2336ca161407SBarry Smith if (snes->prefix) PetscStrcat(p, snes->prefix); 2337ca161407SBarry Smith 2338ca161407SBarry Smith kctx = (SNES_KSP_EW_ConvCtx *)snes->kspconvctx; 2339ca161407SBarry Smith 234076be9ce4SBarry Smith (*PetscHelpPrintf)(snes->comm,"SNES options ------------------------------------------------\n"); 2341488ecbafSBarry Smith ierr = FListPrintTypes(snes->comm,stdout,snes->prefix,"snes_type",SNESList);CHKERRQ(ierr); 234276be9ce4SBarry Smith (*PetscHelpPrintf)(snes->comm," %ssnes_view: view SNES info after each nonlinear solve\n",p); 234376be9ce4SBarry Smith (*PetscHelpPrintf)(snes->comm," %ssnes_max_it <its>: max iterations (default %d)\n",p,snes->max_its); 234476be9ce4SBarry Smith (*PetscHelpPrintf)(snes->comm," %ssnes_max_funcs <maxf>: max function evals (default %d)\n",p,snes->max_funcs); 234576be9ce4SBarry Smith (*PetscHelpPrintf)(snes->comm," %ssnes_stol <stol>: successive step tolerance (default %g)\n",p,snes->xtol); 234676be9ce4SBarry Smith (*PetscHelpPrintf)(snes->comm," %ssnes_atol <atol>: absolute tolerance (default %g)\n",p,snes->atol); 234776be9ce4SBarry Smith (*PetscHelpPrintf)(snes->comm," %ssnes_rtol <rtol>: relative tolerance (default %g)\n",p,snes->rtol); 234876be9ce4SBarry Smith (*PetscHelpPrintf)(snes->comm," %ssnes_trtol <trtol>: trust region parameter tolerance (default %g)\n",p,snes->deltatol); 234976be9ce4SBarry Smith (*PetscHelpPrintf)(snes->comm," SNES Monitoring Options: Choose any of the following\n"); 235076be9ce4SBarry Smith (*PetscHelpPrintf)(snes->comm," %ssnes_cancelmonitors: cancels all monitors hardwired in code\n",p); 235176be9ce4SBarry Smith (*PetscHelpPrintf)(snes->comm," %ssnes_monitor: use default SNES convergence monitor, prints\n\ 2352ca161407SBarry Smith residual norm at each iteration.\n",p); 235376be9ce4SBarry Smith (*PetscHelpPrintf)(snes->comm," %ssnes_smonitor: same as the above, but prints fewer digits of the\n\ 2354ca161407SBarry Smith residual norm for small residual norms. This is useful to conceal\n\ 2355ca161407SBarry Smith meaningless digits that may be different on different machines.\n",p); 235676be9ce4SBarry Smith (*PetscHelpPrintf)(snes->comm," %ssnes_xmonitor [x,y,w,h]: use X graphics convergence monitor\n",p); 2357ca161407SBarry Smith if (snes->type == SNES_NONLINEAR_EQUATIONS) { 235876be9ce4SBarry Smith (*PetscHelpPrintf)(snes->comm, 2359ca161407SBarry Smith " Options for solving systems of nonlinear equations only:\n"); 236076be9ce4SBarry Smith (*PetscHelpPrintf)(snes->comm," %ssnes_fd: use finite differences for Jacobian\n",p); 236176be9ce4SBarry Smith (*PetscHelpPrintf)(snes->comm," %ssnes_mf: use matrix-free Jacobian\n",p); 236276be9ce4SBarry Smith (*PetscHelpPrintf)(snes->comm," %ssnes_mf_operator:use matrix-free Jacobian and user-provided preconditioning matrix\n",p); 2363e24b481bSBarry Smith (*PetscHelpPrintf)(snes->comm," %ssnes_mf_ksp_monitor - if using matrix-free multiply then prints h at each KSP iteration\n",p); 236476be9ce4SBarry Smith (*PetscHelpPrintf)(snes->comm," %ssnes_no_convergence_test: Do not test for convergence, always run to SNES max its\n",p); 236576be9ce4SBarry Smith (*PetscHelpPrintf)(snes->comm," %ssnes_ksp_ew_conv: use Eisenstat-Walker computation of KSP rtol. Params are:\n",p); 236676be9ce4SBarry Smith (*PetscHelpPrintf)(snes->comm, 2367ca161407SBarry Smith " %ssnes_ksp_ew_version <version> (1 or 2, default is %d)\n",p,kctx->version); 236876be9ce4SBarry Smith (*PetscHelpPrintf)(snes->comm, 2369ca161407SBarry Smith " %ssnes_ksp_ew_rtol0 <rtol0> (0 <= rtol0 < 1, default %g)\n",p,kctx->rtol_0); 237076be9ce4SBarry Smith (*PetscHelpPrintf)(snes->comm, 2371ca161407SBarry Smith " %ssnes_ksp_ew_rtolmax <rtolmax> (0 <= rtolmax < 1, default %g)\n",p,kctx->rtol_max); 237276be9ce4SBarry Smith (*PetscHelpPrintf)(snes->comm, 2373ca161407SBarry Smith " %ssnes_ksp_ew_gamma <gamma> (0 <= gamma <= 1, default %g)\n",p,kctx->gamma); 237476be9ce4SBarry Smith (*PetscHelpPrintf)(snes->comm, 2375ca161407SBarry Smith " %ssnes_ksp_ew_alpha <alpha> (1 < alpha <= 2, default %g)\n",p,kctx->alpha); 237676be9ce4SBarry Smith (*PetscHelpPrintf)(snes->comm, 2377ca161407SBarry Smith " %ssnes_ksp_ew_alpha2 <alpha2> (default %g)\n",p,kctx->alpha2); 237876be9ce4SBarry Smith (*PetscHelpPrintf)(snes->comm, 2379ca161407SBarry Smith " %ssnes_ksp_ew_threshold <threshold> (0 < threshold < 1, default %g)\n",p,kctx->threshold); 2380ca161407SBarry Smith } else if (snes->type == SNES_UNCONSTRAINED_MINIMIZATION) { 238176be9ce4SBarry Smith (*PetscHelpPrintf)(snes->comm, 2382ca161407SBarry Smith " Options for solving unconstrained minimization problems only:\n"); 238376be9ce4SBarry Smith (*PetscHelpPrintf)(snes->comm," %ssnes_fmin <ftol>: minimum function tolerance (default %g)\n",p,snes->fmin); 238476be9ce4SBarry Smith (*PetscHelpPrintf)(snes->comm," %ssnes_fd: use finite differences for Hessian\n",p); 238576be9ce4SBarry Smith (*PetscHelpPrintf)(snes->comm," %ssnes_mf: use matrix-free Hessian\n",p); 2386ca161407SBarry Smith } 238776be9ce4SBarry Smith (*PetscHelpPrintf)(snes->comm," Run program with -help %ssnes_type <method> for help on ",p); 238876be9ce4SBarry Smith (*PetscHelpPrintf)(snes->comm,"a particular method\n"); 2389ca161407SBarry Smith if (snes->printhelp) { 2390ca161407SBarry Smith ierr = (*snes->printhelp)(snes,p);CHKERRQ(ierr); 2391ca161407SBarry Smith } 2392ca161407SBarry Smith PetscFunctionReturn(0); 2393ca161407SBarry Smith } 23943c7409f5SSatish Balay 2395acb85ae6SSatish Balay /*MC 2396b2002411SLois Curfman McInnes SNESRegister - Adds a method to the nonlinear solver package. 23979b94acceSBarry Smith 2398b2002411SLois Curfman McInnes Synopsis: 2399b2002411SLois Curfman McInnes SNESRegister(char *name_solver,char *path,char *name_create,int (*routine_create)(SNES)) 24009b94acceSBarry Smith 24018d76a1e5SLois Curfman McInnes Not collective 24028d76a1e5SLois Curfman McInnes 2403b2002411SLois Curfman McInnes Input Parameters: 2404c7afd0dbSLois Curfman McInnes + name_solver - name of a new user-defined solver 2405b2002411SLois Curfman McInnes . path - path (either absolute or relative) the library containing this solver 2406b2002411SLois Curfman McInnes . name_create - name of routine to create method context 2407c7afd0dbSLois Curfman McInnes - routine_create - routine to create method context 24089b94acceSBarry Smith 2409b2002411SLois Curfman McInnes Notes: 2410b2002411SLois Curfman McInnes SNESRegister() may be called multiple times to add several user-defined solvers. 24113a40ed3dSBarry Smith 2412b2002411SLois Curfman McInnes If dynamic libraries are used, then the fourth input argument (routine_create) 2413b2002411SLois Curfman McInnes is ignored. 2414b2002411SLois Curfman McInnes 2415b2002411SLois Curfman McInnes Sample usage: 241669225d78SLois Curfman McInnes .vb 2417b2002411SLois Curfman McInnes SNESRegister("my_solver",/home/username/my_lib/lib/libg/solaris/mylib.a, 2418b2002411SLois Curfman McInnes "MySolverCreate",MySolverCreate); 241969225d78SLois Curfman McInnes .ve 2420b2002411SLois Curfman McInnes 2421b2002411SLois Curfman McInnes Then, your solver can be chosen with the procedural interface via 2422b2002411SLois Curfman McInnes $ SNESSetType(snes,"my_solver") 2423b2002411SLois Curfman McInnes or at runtime via the option 2424b2002411SLois Curfman McInnes $ -snes_type my_solver 2425b2002411SLois Curfman McInnes 242636851e7fSLois Curfman McInnes Level: advanced 242736851e7fSLois Curfman McInnes 2428b2002411SLois Curfman McInnes .keywords: SNES, nonlinear, register 2429b2002411SLois Curfman McInnes 2430b2002411SLois Curfman McInnes .seealso: SNESRegisterAll(), SNESRegisterDestroy() 2431b2002411SLois Curfman McInnes M*/ 2432b2002411SLois Curfman McInnes 2433b2002411SLois Curfman McInnes #undef __FUNC__ 2434b2002411SLois Curfman McInnes #define __FUNC__ "SNESRegister_Private" 2435b2002411SLois Curfman McInnes int SNESRegister_Private(char *sname,char *path,char *name,int (*function)(SNES)) 2436b2002411SLois Curfman McInnes { 2437b2002411SLois Curfman McInnes char fullname[256]; 2438b2002411SLois Curfman McInnes int ierr; 2439b2002411SLois Curfman McInnes 2440b2002411SLois Curfman McInnes PetscFunctionBegin; 2441b2002411SLois Curfman McInnes PetscStrcpy(fullname,path); PetscStrcat(fullname,":");PetscStrcat(fullname,name); 2442488ecbafSBarry Smith ierr = FListAdd_Private(&SNESList,sname,fullname, (int (*)(void*))function);CHKERRQ(ierr); 2443b2002411SLois Curfman McInnes PetscFunctionReturn(0); 2444b2002411SLois Curfman McInnes } 2445