xref: /petsc/src/tao/interface/taosolver.c (revision b2dc45dd0bdc2547aa4dee56e3ff258d1ad19137)
1af0996ceSBarry Smith #include <petsc/private/taoimpl.h> /*I "petsctao.h" I*/
20f0abf79SStefano Zampini #include <petsc/private/snesimpl.h>
3a7e14dcfSSatish Balay 
4441846f8SBarry Smith PetscBool         TaoRegisterAllCalled = PETSC_FALSE;
5441846f8SBarry Smith PetscFunctionList TaoList              = NULL;
6a7e14dcfSSatish Balay 
7441846f8SBarry Smith PetscClassId TAO_CLASSID;
80ebee16dSLisandro Dalcin 
90ebee16dSLisandro Dalcin PetscLogEvent TAO_Solve;
100ebee16dSLisandro Dalcin PetscLogEvent TAO_ObjectiveEval;
110ebee16dSLisandro Dalcin PetscLogEvent TAO_GradientEval;
120ebee16dSLisandro Dalcin PetscLogEvent TAO_ObjGradEval;
130ebee16dSLisandro Dalcin PetscLogEvent TAO_HessianEval;
140ebee16dSLisandro Dalcin PetscLogEvent TAO_JacobianEval;
150ebee16dSLisandro Dalcin PetscLogEvent TAO_ConstraintsEval;
16a7e14dcfSSatish Balay 
1783c8fe1dSLisandro Dalcin const char *TaoSubSetTypes[] = {"subvec", "mask", "matrixfree", "TaoSubSetType", "TAO_SUBSET_", NULL};
18a7e14dcfSSatish Balay 
19e882e171SHong Zhang struct _n_TaoMonitorDrawCtx {
20e882e171SHong Zhang   PetscViewer viewer;
21e882e171SHong Zhang   PetscInt    howoften; /* when > 0 uses iteration % howoften, when negative only final solution plotted */
22e882e171SHong Zhang };
23e882e171SHong Zhang 
24d71ae5a4SJacob Faibussowitsch static PetscErrorCode KSPPreSolve_TAOEW_Private(KSP ksp, Vec b, Vec x, Tao tao)
25d71ae5a4SJacob Faibussowitsch {
260f0abf79SStefano Zampini   SNES snes_ewdummy = tao->snes_ewdummy;
270f0abf79SStefano Zampini 
280f0abf79SStefano Zampini   PetscFunctionBegin;
293ba16761SJacob Faibussowitsch   if (!snes_ewdummy) PetscFunctionReturn(PETSC_SUCCESS);
300f0abf79SStefano Zampini   /* populate snes_ewdummy struct values used in KSPPreSolve_SNESEW */
310f0abf79SStefano Zampini   snes_ewdummy->vec_func = b;
320f0abf79SStefano Zampini   snes_ewdummy->rtol     = tao->gttol;
330f0abf79SStefano Zampini   snes_ewdummy->iter     = tao->niter;
340f0abf79SStefano Zampini   PetscCall(VecNorm(b, NORM_2, &snes_ewdummy->norm));
350f0abf79SStefano Zampini   PetscCall(KSPPreSolve_SNESEW(ksp, b, x, snes_ewdummy));
360f0abf79SStefano Zampini   snes_ewdummy->vec_func = NULL;
373ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
380f0abf79SStefano Zampini }
390f0abf79SStefano Zampini 
40d71ae5a4SJacob Faibussowitsch static PetscErrorCode KSPPostSolve_TAOEW_Private(KSP ksp, Vec b, Vec x, Tao tao)
41d71ae5a4SJacob Faibussowitsch {
420f0abf79SStefano Zampini   SNES snes_ewdummy = tao->snes_ewdummy;
430f0abf79SStefano Zampini 
440f0abf79SStefano Zampini   PetscFunctionBegin;
453ba16761SJacob Faibussowitsch   if (!snes_ewdummy) PetscFunctionReturn(PETSC_SUCCESS);
460f0abf79SStefano Zampini   PetscCall(KSPPostSolve_SNESEW(ksp, b, x, snes_ewdummy));
473ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
480f0abf79SStefano Zampini }
490f0abf79SStefano Zampini 
50d71ae5a4SJacob Faibussowitsch static PetscErrorCode TaoSetUpEW_Private(Tao tao)
51d71ae5a4SJacob Faibussowitsch {
520f0abf79SStefano Zampini   SNESKSPEW  *kctx;
530f0abf79SStefano Zampini   const char *ewprefix;
540f0abf79SStefano Zampini 
550f0abf79SStefano Zampini   PetscFunctionBegin;
563ba16761SJacob Faibussowitsch   if (!tao->ksp) PetscFunctionReturn(PETSC_SUCCESS);
570f0abf79SStefano Zampini   if (tao->ksp_ewconv) {
580f0abf79SStefano Zampini     if (!tao->snes_ewdummy) PetscCall(SNESCreate(PetscObjectComm((PetscObject)tao), &tao->snes_ewdummy));
590f0abf79SStefano Zampini     tao->snes_ewdummy->ksp_ewconv = PETSC_TRUE;
600f0abf79SStefano Zampini     PetscCall(KSPSetPreSolve(tao->ksp, (PetscErrorCode(*)(KSP, Vec, Vec, void *))KSPPreSolve_TAOEW_Private, tao));
610f0abf79SStefano Zampini     PetscCall(KSPSetPostSolve(tao->ksp, (PetscErrorCode(*)(KSP, Vec, Vec, void *))KSPPostSolve_TAOEW_Private, tao));
620f0abf79SStefano Zampini 
630f0abf79SStefano Zampini     PetscCall(KSPGetOptionsPrefix(tao->ksp, &ewprefix));
640f0abf79SStefano Zampini     kctx = (SNESKSPEW *)tao->snes_ewdummy->kspconvctx;
65a4598233SStefano Zampini     PetscCall(SNESEWSetFromOptions_Private(kctx, PETSC_FALSE, PetscObjectComm((PetscObject)tao), ewprefix));
660f0abf79SStefano Zampini   } else PetscCall(SNESDestroy(&tao->snes_ewdummy));
673ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
680f0abf79SStefano Zampini }
690f0abf79SStefano Zampini 
70a7e14dcfSSatish Balay /*@
7165ba42b6SBarry Smith   TaoCreate - Creates a Tao solver
72a7e14dcfSSatish Balay 
73d083f849SBarry Smith   Collective
74a7e14dcfSSatish Balay 
75a7e14dcfSSatish Balay   Input Parameter:
76a7e14dcfSSatish Balay . comm - MPI communicator
77a7e14dcfSSatish Balay 
78a7e14dcfSSatish Balay   Output Parameter:
7947450a7bSBarry Smith . newtao - the new `Tao` context
80a7e14dcfSSatish Balay 
8147450a7bSBarry Smith   Options Database Key:
8265ba42b6SBarry Smith . -tao_type - select which method Tao should use
83a7e14dcfSSatish Balay 
84a7e14dcfSSatish Balay   Level: beginner
85a7e14dcfSSatish Balay 
861cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoSolve()`, `TaoDestroy()`, `TAOSetFromOptions()`, `TAOSetType()`
87a7e14dcfSSatish Balay @*/
88d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoCreate(MPI_Comm comm, Tao *newtao)
89d71ae5a4SJacob Faibussowitsch {
90441846f8SBarry Smith   Tao tao;
91a7e14dcfSSatish Balay 
92a7e14dcfSSatish Balay   PetscFunctionBegin;
934f572ea9SToby Isaac   PetscAssertPointer(newtao, 2);
949566063dSJacob Faibussowitsch   PetscCall(TaoInitializePackage());
959566063dSJacob Faibussowitsch   PetscCall(TaoLineSearchInitializePackage());
969566063dSJacob Faibussowitsch   PetscCall(PetscHeaderCreate(tao, TAO_CLASSID, "Tao", "Optimization solver", "Tao", comm, TaoDestroy, TaoView));
97a7e14dcfSSatish Balay 
9894511df7SStefano Zampini   /* Set non-NULL defaults */
9994511df7SStefano Zampini   tao->ops->convergencetest = TaoDefaultConvergenceTest;
100a7e14dcfSSatish Balay 
101a7e14dcfSSatish Balay   tao->max_it    = 10000;
10294511df7SStefano Zampini   tao->max_funcs = -1;
1036f4723b1SBarry Smith #if defined(PETSC_USE_REAL_SINGLE)
1046f4723b1SBarry Smith   tao->gatol = 1e-5;
1056f4723b1SBarry Smith   tao->grtol = 1e-5;
1066f1f5f59SAlp Dener   tao->crtol = 1e-5;
1076f1f5f59SAlp Dener   tao->catol = 1e-5;
1086f4723b1SBarry Smith #else
109a7e14dcfSSatish Balay   tao->gatol = 1e-8;
110a7e14dcfSSatish Balay   tao->grtol = 1e-8;
1116f1f5f59SAlp Dener   tao->crtol = 1e-8;
1126f1f5f59SAlp Dener   tao->catol = 1e-8;
1136f4723b1SBarry Smith #endif
1146552cf8aSJason Sarich   tao->gttol   = 0.0;
1154f1535bcSTodd Munson   tao->steptol = 0.0;
116e270355aSBarry Smith   tao->trust0  = PETSC_INFINITY;
1176f4723b1SBarry Smith   tao->fmin    = PETSC_NINFINITY;
1189fa2b5dcSStefano Zampini 
119a7e14dcfSSatish Balay   tao->hist_reset = PETSC_TRUE;
120a7e14dcfSSatish Balay 
1219566063dSJacob Faibussowitsch   PetscCall(TaoResetStatistics(tao));
122a7e14dcfSSatish Balay   *newtao = tao;
1233ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
124a7e14dcfSSatish Balay }
125a7e14dcfSSatish Balay 
126a7e14dcfSSatish Balay /*@
127a7e14dcfSSatish Balay   TaoSolve - Solves an optimization problem min F(x) s.t. l <= x <= u
128a7e14dcfSSatish Balay 
129c3339decSBarry Smith   Collective
130a7e14dcfSSatish Balay 
13147450a7bSBarry Smith   Input Parameter:
13247450a7bSBarry Smith . tao - the `Tao` context
133a7e14dcfSSatish Balay 
13467be906fSBarry Smith   Level: beginner
13567be906fSBarry Smith 
136a7e14dcfSSatish Balay   Notes:
13747450a7bSBarry Smith   The user must set up the `Tao` object  with calls to `TaoSetSolution()`, `TaoSetObjective()`, `TaoSetGradient()`, and (if using 2nd order method) `TaoSetHessian()`.
138a7e14dcfSSatish Balay 
13967be906fSBarry Smith   You should call `TaoGetConvergedReason()` or run with `-tao_converged_reason` to determine if the optimization algorithm actually succeeded or
140a35d58b8SBarry Smith   why it failed.
141a35d58b8SBarry Smith 
1421cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoCreate()`, `TaoSetObjective()`, `TaoSetGradient()`, `TaoSetHessian()`, `TaoGetConvergedReason()`, `TaoSetUp()`
143a7e14dcfSSatish Balay  @*/
144d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoSolve(Tao tao)
145d71ae5a4SJacob Faibussowitsch {
146e2379f4fSBarry Smith   static PetscBool set = PETSC_FALSE;
14747a47007SBarry Smith 
148a7e14dcfSSatish Balay   PetscFunctionBegin;
149441846f8SBarry Smith   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
150d0609cedSBarry Smith   PetscCall(PetscCitationsRegister("@TechReport{tao-user-ref,\n"
151e2379f4fSBarry Smith                                    "title   = {Toolkit for Advanced Optimization (TAO) Users Manual},\n"
152e2379f4fSBarry Smith                                    "author  = {Todd Munson and Jason Sarich and Stefan Wild and Steve Benson and Lois Curfman McInnes},\n"
153e2379f4fSBarry Smith                                    "Institution = {Argonne National Laboratory},\n"
154e2379f4fSBarry Smith                                    "Year   = 2014,\n"
155e2379f4fSBarry Smith                                    "Number = {ANL/MCS-TM-322 - Revision 3.5},\n"
1569371c9d4SSatish Balay                                    "url    = {https://www.mcs.anl.gov/research/projects/tao/}\n}\n",
1579371c9d4SSatish Balay                                    &set));
158494bef23SAlp Dener   tao->header_printed = PETSC_FALSE;
1599566063dSJacob Faibussowitsch   PetscCall(TaoSetUp(tao));
1609566063dSJacob Faibussowitsch   PetscCall(TaoResetStatistics(tao));
1611baa6e33SBarry Smith   if (tao->linesearch) PetscCall(TaoLineSearchReset(tao->linesearch));
162a7e14dcfSSatish Balay 
1639566063dSJacob Faibussowitsch   PetscCall(PetscLogEventBegin(TAO_Solve, tao, 0, 0, 0));
164dbbe0bcdSBarry Smith   PetscTryTypeMethod(tao, solve);
1659566063dSJacob Faibussowitsch   PetscCall(PetscLogEventEnd(TAO_Solve, tao, 0, 0, 0));
166a7e14dcfSSatish Balay 
1679566063dSJacob Faibussowitsch   PetscCall(VecViewFromOptions(tao->solution, (PetscObject)tao, "-tao_view_solution"));
1687cf37e64SBarry Smith 
1698931d482SJason Sarich   tao->ntotalits += tao->niter;
170a7e14dcfSSatish Balay 
171a7e14dcfSSatish Balay   if (tao->printreason) {
172f642d338SBarry Smith     PetscViewer viewer = PETSC_VIEWER_STDOUT_(((PetscObject)tao)->comm);
173f642d338SBarry Smith     PetscCall(PetscViewerASCIIAddTab(viewer, ((PetscObject)tao)->tablevel));
174a7e14dcfSSatish Balay     if (tao->reason > 0) {
175f642d338SBarry Smith       PetscCall(PetscViewerASCIIPrintf(viewer, "  TAO %s solve converged due to %s iterations %" PetscInt_FMT "\n", ((PetscObject)tao)->prefix ? ((PetscObject)tao)->prefix : "", TaoConvergedReasons[tao->reason], tao->niter));
176a7e14dcfSSatish Balay     } else {
177f642d338SBarry Smith       PetscCall(PetscViewerASCIIPrintf(viewer, "  TAO %s solve did not converge due to %s iteration %" PetscInt_FMT "\n", ((PetscObject)tao)->prefix ? ((PetscObject)tao)->prefix : "", TaoConvergedReasons[tao->reason], tao->niter));
178a7e14dcfSSatish Balay     }
179f642d338SBarry Smith     PetscCall(PetscViewerASCIISubtractTab(viewer, ((PetscObject)tao)->tablevel));
180a7e14dcfSSatish Balay   }
181f642d338SBarry Smith   PetscCall(TaoViewFromOptions(tao, NULL, "-tao_view"));
1823ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
183a7e14dcfSSatish Balay }
184a7e14dcfSSatish Balay 
185a7e14dcfSSatish Balay /*@
186a7e14dcfSSatish Balay   TaoSetUp - Sets up the internal data structures for the later use
187a7e14dcfSSatish Balay   of a Tao solver
188a7e14dcfSSatish Balay 
189c3339decSBarry Smith   Collective
190a7e14dcfSSatish Balay 
19147450a7bSBarry Smith   Input Parameter:
19247450a7bSBarry Smith . tao - the `Tao` context
193a7e14dcfSSatish Balay 
19467be906fSBarry Smith   Level: advanced
19567be906fSBarry Smith 
19647450a7bSBarry Smith   Note:
19765ba42b6SBarry Smith   The user will not need to explicitly call `TaoSetUp()`, as it will
19865ba42b6SBarry Smith   automatically be called in `TaoSolve()`.  However, if the user
19965ba42b6SBarry Smith   desires to call it explicitly, it should come after `TaoCreate()`
20065ba42b6SBarry Smith   and any TaoSetSomething() routines, but before `TaoSolve()`.
201a7e14dcfSSatish Balay 
2021cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoCreate()`, `TaoSolve()`
203a7e14dcfSSatish Balay @*/
204d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoSetUp(Tao tao)
205d71ae5a4SJacob Faibussowitsch {
206a7e14dcfSSatish Balay   PetscFunctionBegin;
207441846f8SBarry Smith   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
2083ba16761SJacob Faibussowitsch   if (tao->setupcalled) PetscFunctionReturn(PETSC_SUCCESS);
2090f0abf79SStefano Zampini   PetscCall(TaoSetUpEW_Private(tao));
2103c859ba3SBarry Smith   PetscCheck(tao->solution, PetscObjectComm((PetscObject)tao), PETSC_ERR_ARG_WRONGSTATE, "Must call TaoSetSolution");
211dbbe0bcdSBarry Smith   PetscTryTypeMethod(tao, setup);
212a7e14dcfSSatish Balay   tao->setupcalled = PETSC_TRUE;
2133ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
214a7e14dcfSSatish Balay }
215a7e14dcfSSatish Balay 
2161fb7b255SJunchao Zhang /*@C
21747450a7bSBarry Smith   TaoDestroy - Destroys the `Tao` context that was created with `TaoCreate()`
218a7e14dcfSSatish Balay 
219c3339decSBarry Smith   Collective
220a7e14dcfSSatish Balay 
221a7e14dcfSSatish Balay   Input Parameter:
22247450a7bSBarry Smith . tao - the `Tao` context
223a7e14dcfSSatish Balay 
224a7e14dcfSSatish Balay   Level: beginner
225a7e14dcfSSatish Balay 
2261cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoCreate()`, `TaoSolve()`
227a7e14dcfSSatish Balay @*/
228d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoDestroy(Tao *tao)
229d71ae5a4SJacob Faibussowitsch {
230a7e14dcfSSatish Balay   PetscFunctionBegin;
2313ba16761SJacob Faibussowitsch   if (!*tao) PetscFunctionReturn(PETSC_SUCCESS);
232441846f8SBarry Smith   PetscValidHeaderSpecific(*tao, TAO_CLASSID, 1);
2339371c9d4SSatish Balay   if (--((PetscObject)*tao)->refct > 0) {
2349371c9d4SSatish Balay     *tao = NULL;
2353ba16761SJacob Faibussowitsch     PetscFunctionReturn(PETSC_SUCCESS);
236a7e14dcfSSatish Balay   }
2379371c9d4SSatish Balay 
23848a46eb9SPierre Jolivet   if ((*tao)->ops->destroy) PetscCall((*((*tao))->ops->destroy)(*tao));
2399566063dSJacob Faibussowitsch   PetscCall(KSPDestroy(&(*tao)->ksp));
2400f0abf79SStefano Zampini   PetscCall(SNESDestroy(&(*tao)->snes_ewdummy));
2419566063dSJacob Faibussowitsch   PetscCall(TaoLineSearchDestroy(&(*tao)->linesearch));
242a7e14dcfSSatish Balay 
243a7e14dcfSSatish Balay   if ((*tao)->ops->convergencedestroy) {
2449566063dSJacob Faibussowitsch     PetscCall((*(*tao)->ops->convergencedestroy)((*tao)->cnvP));
24548a46eb9SPierre Jolivet     if ((*tao)->jacobian_state_inv) PetscCall(MatDestroy(&(*tao)->jacobian_state_inv));
246a7e14dcfSSatish Balay   }
2479566063dSJacob Faibussowitsch   PetscCall(VecDestroy(&(*tao)->solution));
2489566063dSJacob Faibussowitsch   PetscCall(VecDestroy(&(*tao)->gradient));
2499566063dSJacob Faibussowitsch   PetscCall(VecDestroy(&(*tao)->ls_res));
250a7e14dcfSSatish Balay 
251a9603a14SPatrick Farrell   if ((*tao)->gradient_norm) {
2529566063dSJacob Faibussowitsch     PetscCall(PetscObjectDereference((PetscObject)(*tao)->gradient_norm));
2539566063dSJacob Faibussowitsch     PetscCall(VecDestroy(&(*tao)->gradient_norm_tmp));
254a9603a14SPatrick Farrell   }
255a9603a14SPatrick Farrell 
2569566063dSJacob Faibussowitsch   PetscCall(VecDestroy(&(*tao)->XL));
2579566063dSJacob Faibussowitsch   PetscCall(VecDestroy(&(*tao)->XU));
2589566063dSJacob Faibussowitsch   PetscCall(VecDestroy(&(*tao)->IL));
2599566063dSJacob Faibussowitsch   PetscCall(VecDestroy(&(*tao)->IU));
2609566063dSJacob Faibussowitsch   PetscCall(VecDestroy(&(*tao)->DE));
2619566063dSJacob Faibussowitsch   PetscCall(VecDestroy(&(*tao)->DI));
2629566063dSJacob Faibussowitsch   PetscCall(VecDestroy(&(*tao)->constraints));
2639566063dSJacob Faibussowitsch   PetscCall(VecDestroy(&(*tao)->constraints_equality));
2649566063dSJacob Faibussowitsch   PetscCall(VecDestroy(&(*tao)->constraints_inequality));
2659566063dSJacob Faibussowitsch   PetscCall(VecDestroy(&(*tao)->stepdirection));
2669566063dSJacob Faibussowitsch   PetscCall(MatDestroy(&(*tao)->hessian_pre));
2679566063dSJacob Faibussowitsch   PetscCall(MatDestroy(&(*tao)->hessian));
2689566063dSJacob Faibussowitsch   PetscCall(MatDestroy(&(*tao)->ls_jac));
2699566063dSJacob Faibussowitsch   PetscCall(MatDestroy(&(*tao)->ls_jac_pre));
2709566063dSJacob Faibussowitsch   PetscCall(MatDestroy(&(*tao)->jacobian_pre));
2719566063dSJacob Faibussowitsch   PetscCall(MatDestroy(&(*tao)->jacobian));
2729566063dSJacob Faibussowitsch   PetscCall(MatDestroy(&(*tao)->jacobian_state_pre));
2739566063dSJacob Faibussowitsch   PetscCall(MatDestroy(&(*tao)->jacobian_state));
2749566063dSJacob Faibussowitsch   PetscCall(MatDestroy(&(*tao)->jacobian_state_inv));
2759566063dSJacob Faibussowitsch   PetscCall(MatDestroy(&(*tao)->jacobian_design));
2769566063dSJacob Faibussowitsch   PetscCall(MatDestroy(&(*tao)->jacobian_equality));
2779566063dSJacob Faibussowitsch   PetscCall(MatDestroy(&(*tao)->jacobian_equality_pre));
2789566063dSJacob Faibussowitsch   PetscCall(MatDestroy(&(*tao)->jacobian_inequality));
2799566063dSJacob Faibussowitsch   PetscCall(MatDestroy(&(*tao)->jacobian_inequality_pre));
2809566063dSJacob Faibussowitsch   PetscCall(ISDestroy(&(*tao)->state_is));
2819566063dSJacob Faibussowitsch   PetscCall(ISDestroy(&(*tao)->design_is));
2829566063dSJacob Faibussowitsch   PetscCall(VecDestroy(&(*tao)->res_weights_v));
283*b2dc45ddSPierre Jolivet   PetscCall(TaoMonitorCancel(*tao));
28448a46eb9SPierre Jolivet   if ((*tao)->hist_malloc) PetscCall(PetscFree4((*tao)->hist_obj, (*tao)->hist_resid, (*tao)->hist_cnorm, (*tao)->hist_lits));
285737f463aSAlp Dener   if ((*tao)->res_weights_n) {
2869566063dSJacob Faibussowitsch     PetscCall(PetscFree((*tao)->res_weights_rows));
2879566063dSJacob Faibussowitsch     PetscCall(PetscFree((*tao)->res_weights_cols));
2889566063dSJacob Faibussowitsch     PetscCall(PetscFree((*tao)->res_weights_w));
289125ddc8aSJason Sarich   }
2909566063dSJacob Faibussowitsch   PetscCall(PetscHeaderDestroy(tao));
2913ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
292a7e14dcfSSatish Balay }
293a7e14dcfSSatish Balay 
294a7e14dcfSSatish Balay /*@
2951d27aa22SBarry Smith   TaoKSPSetUseEW - Sets `SNES` to use Eisenstat-Walker method {cite}`ew96`for computing relative tolerance for linear solvers.
2960f0abf79SStefano Zampini 
297c3339decSBarry Smith   Logically Collective
2980f0abf79SStefano Zampini 
2990f0abf79SStefano Zampini   Input Parameters:
3000f0abf79SStefano Zampini + tao  - Tao context
30165ba42b6SBarry Smith - flag - `PETSC_TRUE` or `PETSC_FALSE`
3020f0abf79SStefano Zampini 
30367be906fSBarry Smith   Level: advanced
30467be906fSBarry Smith 
30547450a7bSBarry Smith   Note:
30665ba42b6SBarry Smith   See `SNESKSPSetUseEW()` for customization details.
3070f0abf79SStefano Zampini 
3081cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `SNESKSPSetUseEW()`
3090f0abf79SStefano Zampini @*/
310d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoKSPSetUseEW(Tao tao, PetscBool flag)
311d71ae5a4SJacob Faibussowitsch {
3120f0abf79SStefano Zampini   PetscFunctionBegin;
3130f0abf79SStefano Zampini   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
3140f0abf79SStefano Zampini   PetscValidLogicalCollectiveBool(tao, flag, 2);
3150f0abf79SStefano Zampini   tao->ksp_ewconv = flag;
3163ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
3170f0abf79SStefano Zampini }
3180f0abf79SStefano Zampini 
3190f0abf79SStefano Zampini /*@
32047450a7bSBarry Smith   TaoSetFromOptions - Sets various Tao parameters from the options database
321a7e14dcfSSatish Balay 
322c3339decSBarry Smith   Collective
323a7e14dcfSSatish Balay 
32401d2d390SJose E. Roman   Input Parameter:
32547450a7bSBarry Smith . tao - the `Tao` solver context
326a7e14dcfSSatish Balay 
32747450a7bSBarry Smith   Options Database Keys:
32865ba42b6SBarry Smith + -tao_type <type>        - The algorithm that Tao uses (lmvm, nls, etc.)
329a7e14dcfSSatish Balay . -tao_gatol <gatol>      - absolute error tolerance for ||gradient||
330a7e14dcfSSatish Balay . -tao_grtol <grtol>      - relative error tolerance for ||gradient||
331a7e14dcfSSatish Balay . -tao_gttol <gttol>      - reduction of ||gradient|| relative to initial gradient
332a7e14dcfSSatish Balay . -tao_max_it <max>       - sets maximum number of iterations
333a7e14dcfSSatish Balay . -tao_max_funcs <max>    - sets maximum number of function evaluations
334a7e14dcfSSatish Balay . -tao_fmin <fmin>        - stop if function value reaches fmin
335a7e14dcfSSatish Balay . -tao_steptol <tol>      - stop if trust region radius less than <tol>
336a7e14dcfSSatish Balay . -tao_trust0 <t>         - initial trust region radius
33747450a7bSBarry Smith . -tao_monitor            - prints function value and residual norm at each iteration
338a7e14dcfSSatish Balay . -tao_smonitor           - same as tao_monitor, but truncates very small values
339a7e14dcfSSatish Balay . -tao_cmonitor           - prints function value, residual, and constraint norm at each iteration
340a7e14dcfSSatish Balay . -tao_view_solution      - prints solution vector at each iteration
3414a48860cSAlp Dener . -tao_view_ls_residual   - prints least-squares residual vector at each iteration
342147403d9SBarry Smith . -tao_view_stepdirection - prints step direction vector at each iteration
343a7e14dcfSSatish Balay . -tao_view_gradient      - prints gradient vector at each iteration
344a7e14dcfSSatish Balay . -tao_draw_solution      - graphically view solution vector at each iteration
345a7e14dcfSSatish Balay . -tao_draw_step          - graphically view step vector at each iteration
346a7e14dcfSSatish Balay . -tao_draw_gradient      - graphically view gradient at each iteration
347a7e14dcfSSatish Balay . -tao_fd_gradient        - use gradient computed with finite differences
348f4c1ad5cSStefano Zampini . -tao_fd_hessian         - use hessian computed with finite differences
349f4c1ad5cSStefano Zampini . -tao_mf_hessian         - use matrix-free hessian computed with finite differences
350*b2dc45ddSPierre Jolivet . -tao_monitor_cancel     - cancels all monitors (except those set with command line)
351441846f8SBarry Smith . -tao_view               - prints information about the Tao after solving
35265ba42b6SBarry Smith - -tao_converged_reason   - prints the reason Tao stopped iterating
353a7e14dcfSSatish Balay 
35467be906fSBarry Smith   Level: beginner
35567be906fSBarry Smith 
35667be906fSBarry Smith   Note:
35747450a7bSBarry Smith   To see all options, run your program with the `-help` option or consult the
35865ba42b6SBarry Smith   user's manual. Should be called after `TaoCreate()` but before `TaoSolve()`
359a7e14dcfSSatish Balay 
3601cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoCreate()`, `TaoSolve()`
361a7e14dcfSSatish Balay @*/
362d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoSetFromOptions(Tao tao)
363d71ae5a4SJacob Faibussowitsch {
364b625d6c7SJed Brown   TaoType     default_type = TAOLMVM;
365a7e14dcfSSatish Balay   char        type[256], monfilename[PETSC_MAX_PATH_LEN];
366a7e14dcfSSatish Balay   PetscViewer monviewer;
367a7e14dcfSSatish Balay   PetscBool   flg;
368a7e14dcfSSatish Balay   MPI_Comm    comm;
369a7e14dcfSSatish Balay 
370a7e14dcfSSatish Balay   PetscFunctionBegin;
371441846f8SBarry Smith   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
3729566063dSJacob Faibussowitsch   PetscCall(PetscObjectGetComm((PetscObject)tao, &comm));
373125ddc8aSJason Sarich 
37460b70c5cSStefano Zampini   if (((PetscObject)tao)->type_name) default_type = ((PetscObject)tao)->type_name;
37560b70c5cSStefano Zampini 
376d0609cedSBarry Smith   PetscObjectOptionsBegin((PetscObject)tao);
377a7e14dcfSSatish Balay   /* Check for type from options */
3789566063dSJacob Faibussowitsch   PetscCall(PetscOptionsFList("-tao_type", "Tao Solver type", "TaoSetType", TaoList, default_type, type, 256, &flg));
379a7e14dcfSSatish Balay   if (flg) {
3809566063dSJacob Faibussowitsch     PetscCall(TaoSetType(tao, type));
381a7e14dcfSSatish Balay   } else if (!((PetscObject)tao)->type_name) {
3829566063dSJacob Faibussowitsch     PetscCall(TaoSetType(tao, default_type));
383a7e14dcfSSatish Balay   }
384a7e14dcfSSatish Balay 
38560b70c5cSStefano Zampini   /* Tao solvers do not set the prefix, set it here if not yet done
38660b70c5cSStefano Zampini      We do it after SetType since solver may have been changed */
38760b70c5cSStefano Zampini   if (tao->linesearch) {
38860b70c5cSStefano Zampini     const char *prefix;
38960b70c5cSStefano Zampini     PetscCall(TaoLineSearchGetOptionsPrefix(tao->linesearch, &prefix));
39060b70c5cSStefano Zampini     if (!prefix) PetscCall(TaoLineSearchSetOptionsPrefix(tao->linesearch, ((PetscObject)(tao))->prefix));
39160b70c5cSStefano Zampini   }
39260b70c5cSStefano Zampini 
3939566063dSJacob Faibussowitsch   PetscCall(PetscOptionsReal("-tao_catol", "Stop if constraints violations within", "TaoSetConstraintTolerances", tao->catol, &tao->catol, &flg));
3946552cf8aSJason Sarich   if (flg) tao->catol_changed = PETSC_TRUE;
3956aad120cSJose E. Roman   PetscCall(PetscOptionsReal("-tao_crtol", "Stop if relative constraint violations within", "TaoSetConstraintTolerances", tao->crtol, &tao->crtol, &flg));
3966552cf8aSJason Sarich   if (flg) tao->crtol_changed = PETSC_TRUE;
3979566063dSJacob Faibussowitsch   PetscCall(PetscOptionsReal("-tao_gatol", "Stop if norm of gradient less than", "TaoSetTolerances", tao->gatol, &tao->gatol, &flg));
3986552cf8aSJason Sarich   if (flg) tao->gatol_changed = PETSC_TRUE;
3999566063dSJacob Faibussowitsch   PetscCall(PetscOptionsReal("-tao_grtol", "Stop if norm of gradient divided by the function value is less than", "TaoSetTolerances", tao->grtol, &tao->grtol, &flg));
4006552cf8aSJason Sarich   if (flg) tao->grtol_changed = PETSC_TRUE;
4019566063dSJacob Faibussowitsch   PetscCall(PetscOptionsReal("-tao_gttol", "Stop if the norm of the gradient is less than the norm of the initial gradient times tol", "TaoSetTolerances", tao->gttol, &tao->gttol, &flg));
4026552cf8aSJason Sarich   if (flg) tao->gttol_changed = PETSC_TRUE;
4039566063dSJacob Faibussowitsch   PetscCall(PetscOptionsInt("-tao_max_it", "Stop if iteration number exceeds", "TaoSetMaximumIterations", tao->max_it, &tao->max_it, &flg));
4046552cf8aSJason Sarich   if (flg) tao->max_it_changed = PETSC_TRUE;
4059566063dSJacob Faibussowitsch   PetscCall(PetscOptionsInt("-tao_max_funcs", "Stop if number of function evaluations exceeds", "TaoSetMaximumFunctionEvaluations", tao->max_funcs, &tao->max_funcs, &flg));
4066552cf8aSJason Sarich   if (flg) tao->max_funcs_changed = PETSC_TRUE;
4079566063dSJacob Faibussowitsch   PetscCall(PetscOptionsReal("-tao_fmin", "Stop if function less than", "TaoSetFunctionLowerBound", tao->fmin, &tao->fmin, &flg));
4086552cf8aSJason Sarich   if (flg) tao->fmin_changed = PETSC_TRUE;
4099566063dSJacob Faibussowitsch   PetscCall(PetscOptionsReal("-tao_steptol", "Stop if step size or trust region radius less than", "", tao->steptol, &tao->steptol, &flg));
4106552cf8aSJason Sarich   if (flg) tao->steptol_changed = PETSC_TRUE;
4119566063dSJacob Faibussowitsch   PetscCall(PetscOptionsReal("-tao_trust0", "Initial trust region radius", "TaoSetTrustRegionRadius", tao->trust0, &tao->trust0, &flg));
4126552cf8aSJason Sarich   if (flg) tao->trust0_changed = PETSC_TRUE;
4139566063dSJacob Faibussowitsch   PetscCall(PetscOptionsString("-tao_view_solution", "view solution vector after each evaluation", "TaoSetMonitor", "stdout", monfilename, sizeof(monfilename), &flg));
414a7e14dcfSSatish Balay   if (flg) {
4159566063dSJacob Faibussowitsch     PetscCall(PetscViewerASCIIOpen(comm, monfilename, &monviewer));
4169566063dSJacob Faibussowitsch     PetscCall(TaoSetMonitor(tao, TaoSolutionMonitor, monviewer, (PetscErrorCode(*)(void **))PetscViewerDestroy));
417a7e14dcfSSatish Balay   }
418a7e14dcfSSatish Balay 
41965ba42b6SBarry Smith   PetscCall(PetscOptionsBool("-tao_converged_reason", "Print reason for Tao converged", "TaoSolve", tao->printreason, &tao->printreason, NULL));
4209566063dSJacob Faibussowitsch   PetscCall(PetscOptionsString("-tao_view_gradient", "view gradient vector after each evaluation", "TaoSetMonitor", "stdout", monfilename, sizeof(monfilename), &flg));
421a7e14dcfSSatish Balay   if (flg) {
4229566063dSJacob Faibussowitsch     PetscCall(PetscViewerASCIIOpen(comm, monfilename, &monviewer));
4239566063dSJacob Faibussowitsch     PetscCall(TaoSetMonitor(tao, TaoGradientMonitor, monviewer, (PetscErrorCode(*)(void **))PetscViewerDestroy));
424a7e14dcfSSatish Balay   }
425a7e14dcfSSatish Balay 
4269566063dSJacob Faibussowitsch   PetscCall(PetscOptionsString("-tao_view_stepdirection", "view step direction vector after each iteration", "TaoSetMonitor", "stdout", monfilename, sizeof(monfilename), &flg));
427a7e14dcfSSatish Balay   if (flg) {
4289566063dSJacob Faibussowitsch     PetscCall(PetscViewerASCIIOpen(comm, monfilename, &monviewer));
4299566063dSJacob Faibussowitsch     PetscCall(TaoSetMonitor(tao, TaoStepDirectionMonitor, monviewer, (PetscErrorCode(*)(void **))PetscViewerDestroy));
430a7e14dcfSSatish Balay   }
431a7e14dcfSSatish Balay 
4329566063dSJacob Faibussowitsch   PetscCall(PetscOptionsString("-tao_view_residual", "view least-squares residual vector after each evaluation", "TaoSetMonitor", "stdout", monfilename, sizeof(monfilename), &flg));
433a7e14dcfSSatish Balay   if (flg) {
4349566063dSJacob Faibussowitsch     PetscCall(PetscViewerASCIIOpen(comm, monfilename, &monviewer));
4359566063dSJacob Faibussowitsch     PetscCall(TaoSetMonitor(tao, TaoResidualMonitor, monviewer, (PetscErrorCode(*)(void **))PetscViewerDestroy));
436a7e14dcfSSatish Balay   }
437a7e14dcfSSatish Balay 
4389566063dSJacob Faibussowitsch   PetscCall(PetscOptionsString("-tao_monitor", "Use the default convergence monitor", "TaoSetMonitor", "stdout", monfilename, sizeof(monfilename), &flg));
439a7e14dcfSSatish Balay   if (flg) {
4409566063dSJacob Faibussowitsch     PetscCall(PetscViewerASCIIOpen(comm, monfilename, &monviewer));
4419566063dSJacob Faibussowitsch     PetscCall(TaoSetMonitor(tao, TaoMonitorDefault, monviewer, (PetscErrorCode(*)(void **))PetscViewerDestroy));
442a7e14dcfSSatish Balay   }
443a7e14dcfSSatish Balay 
4449566063dSJacob Faibussowitsch   PetscCall(PetscOptionsString("-tao_gmonitor", "Use the convergence monitor with extra globalization info", "TaoSetMonitor", "stdout", monfilename, sizeof(monfilename), &flg));
4458d5ead36SAlp Dener   if (flg) {
4469566063dSJacob Faibussowitsch     PetscCall(PetscViewerASCIIOpen(comm, monfilename, &monviewer));
4479566063dSJacob Faibussowitsch     PetscCall(TaoSetMonitor(tao, TaoDefaultGMonitor, monviewer, (PetscErrorCode(*)(void **))PetscViewerDestroy));
4488d5ead36SAlp Dener   }
4498d5ead36SAlp Dener 
4509566063dSJacob Faibussowitsch   PetscCall(PetscOptionsString("-tao_smonitor", "Use the short convergence monitor", "TaoSetMonitor", "stdout", monfilename, sizeof(monfilename), &flg));
451a7e14dcfSSatish Balay   if (flg) {
4529566063dSJacob Faibussowitsch     PetscCall(PetscViewerASCIIOpen(comm, monfilename, &monviewer));
4539566063dSJacob Faibussowitsch     PetscCall(TaoSetMonitor(tao, TaoDefaultSMonitor, monviewer, (PetscErrorCode(*)(void **))PetscViewerDestroy));
454a7e14dcfSSatish Balay   }
455a7e14dcfSSatish Balay 
4569566063dSJacob Faibussowitsch   PetscCall(PetscOptionsString("-tao_cmonitor", "Use the default convergence monitor with constraint norm", "TaoSetMonitor", "stdout", monfilename, sizeof(monfilename), &flg));
457a7e14dcfSSatish Balay   if (flg) {
4589566063dSJacob Faibussowitsch     PetscCall(PetscViewerASCIIOpen(comm, monfilename, &monviewer));
4599566063dSJacob Faibussowitsch     PetscCall(TaoSetMonitor(tao, TaoDefaultCMonitor, monviewer, (PetscErrorCode(*)(void **))PetscViewerDestroy));
460a7e14dcfSSatish Balay   }
461a7e14dcfSSatish Balay 
4628afaa268SBarry Smith   flg = PETSC_FALSE;
463*b2dc45ddSPierre Jolivet   PetscCall(PetscOptionsDeprecated("-tao_cancelmonitors", "-tao_monitor_cancel", "3.21", NULL));
464*b2dc45ddSPierre Jolivet   PetscCall(PetscOptionsBool("-tao_monitor_cancel", "cancel all monitors and call any registered destroy routines", "TaoMonitorCancel", flg, &flg, NULL));
465*b2dc45ddSPierre Jolivet   if (flg) PetscCall(TaoMonitorCancel(tao));
466a7e14dcfSSatish Balay 
4678afaa268SBarry Smith   flg = PETSC_FALSE;
4689566063dSJacob Faibussowitsch   PetscCall(PetscOptionsBool("-tao_draw_solution", "Plot solution vector at each iteration", "TaoSetMonitor", flg, &flg, NULL));
469a7e14dcfSSatish Balay   if (flg) {
470e882e171SHong Zhang     TaoMonitorDrawCtx drawctx;
471e882e171SHong Zhang     PetscInt          howoften = 1;
4729566063dSJacob Faibussowitsch     PetscCall(TaoMonitorDrawCtxCreate(PetscObjectComm((PetscObject)tao), NULL, NULL, PETSC_DECIDE, PETSC_DECIDE, 300, 300, howoften, &drawctx));
4739566063dSJacob Faibussowitsch     PetscCall(TaoSetMonitor(tao, TaoDrawSolutionMonitor, drawctx, (PetscErrorCode(*)(void **))TaoMonitorDrawCtxDestroy));
474a7e14dcfSSatish Balay   }
475a7e14dcfSSatish Balay 
4768afaa268SBarry Smith   flg = PETSC_FALSE;
4779566063dSJacob Faibussowitsch   PetscCall(PetscOptionsBool("-tao_draw_step", "plots step direction at each iteration", "TaoSetMonitor", flg, &flg, NULL));
4781baa6e33SBarry Smith   if (flg) PetscCall(TaoSetMonitor(tao, TaoDrawStepMonitor, NULL, NULL));
479a7e14dcfSSatish Balay 
4808afaa268SBarry Smith   flg = PETSC_FALSE;
4819566063dSJacob Faibussowitsch   PetscCall(PetscOptionsBool("-tao_draw_gradient", "plots gradient at each iteration", "TaoSetMonitor", flg, &flg, NULL));
482a7e14dcfSSatish Balay   if (flg) {
483e882e171SHong Zhang     TaoMonitorDrawCtx drawctx;
484e882e171SHong Zhang     PetscInt          howoften = 1;
4859566063dSJacob Faibussowitsch     PetscCall(TaoMonitorDrawCtxCreate(PetscObjectComm((PetscObject)tao), NULL, NULL, PETSC_DECIDE, PETSC_DECIDE, 300, 300, howoften, &drawctx));
4869566063dSJacob Faibussowitsch     PetscCall(TaoSetMonitor(tao, TaoDrawGradientMonitor, drawctx, (PetscErrorCode(*)(void **))TaoMonitorDrawCtxDestroy));
487a7e14dcfSSatish Balay   }
4888afaa268SBarry Smith   flg = PETSC_FALSE;
4899566063dSJacob Faibussowitsch   PetscCall(PetscOptionsBool("-tao_fd_gradient", "compute gradient using finite differences", "TaoDefaultComputeGradient", flg, &flg, NULL));
4901baa6e33SBarry Smith   if (flg) PetscCall(TaoSetGradient(tao, NULL, TaoDefaultComputeGradient, NULL));
491f4c1ad5cSStefano Zampini   flg = PETSC_FALSE;
4929566063dSJacob Faibussowitsch   PetscCall(PetscOptionsBool("-tao_fd_hessian", "compute hessian using finite differences", "TaoDefaultComputeHessian", flg, &flg, NULL));
493f4c1ad5cSStefano Zampini   if (flg) {
494f4c1ad5cSStefano Zampini     Mat H;
495f4c1ad5cSStefano Zampini 
4969566063dSJacob Faibussowitsch     PetscCall(MatCreate(PetscObjectComm((PetscObject)tao), &H));
4979566063dSJacob Faibussowitsch     PetscCall(MatSetType(H, MATAIJ));
4989566063dSJacob Faibussowitsch     PetscCall(TaoSetHessian(tao, H, H, TaoDefaultComputeHessian, NULL));
4999566063dSJacob Faibussowitsch     PetscCall(MatDestroy(&H));
500f4c1ad5cSStefano Zampini   }
501f4c1ad5cSStefano Zampini   flg = PETSC_FALSE;
5029566063dSJacob Faibussowitsch   PetscCall(PetscOptionsBool("-tao_mf_hessian", "compute matrix-free hessian using finite differences", "TaoDefaultComputeHessianMFFD", flg, &flg, NULL));
503f4c1ad5cSStefano Zampini   if (flg) {
504f4c1ad5cSStefano Zampini     Mat H;
505f4c1ad5cSStefano Zampini 
5069566063dSJacob Faibussowitsch     PetscCall(MatCreate(PetscObjectComm((PetscObject)tao), &H));
5079566063dSJacob Faibussowitsch     PetscCall(TaoSetHessian(tao, H, H, TaoDefaultComputeHessianMFFD, NULL));
5089566063dSJacob Faibussowitsch     PetscCall(MatDestroy(&H));
509f4c1ad5cSStefano Zampini   }
510414d97d3SAlp Dener   flg = PETSC_FALSE;
5119566063dSJacob Faibussowitsch   PetscCall(PetscOptionsBool("-tao_recycle_history", "enable recycling/re-using information from the previous TaoSolve() call for some algorithms", "TaoSetRecycleHistory", flg, &flg, NULL));
5121baa6e33SBarry Smith   if (flg) PetscCall(TaoSetRecycleHistory(tao, PETSC_TRUE));
5139566063dSJacob Faibussowitsch   PetscCall(PetscOptionsEnum("-tao_subset_type", "subset type", "", TaoSubSetTypes, (PetscEnum)tao->subset_type, (PetscEnum *)&tao->subset_type, NULL));
514a7e14dcfSSatish Balay 
5150f0abf79SStefano Zampini   if (tao->ksp) {
5160f0abf79SStefano Zampini     PetscCall(PetscOptionsBool("-tao_ksp_ew", "Use Eisentat-Walker linear system convergence test", "TaoKSPSetUseEW", tao->ksp_ewconv, &tao->ksp_ewconv, NULL));
5170f0abf79SStefano Zampini     PetscCall(TaoKSPSetUseEW(tao, tao->ksp_ewconv));
5180f0abf79SStefano Zampini   }
5190f0abf79SStefano Zampini 
520dbbe0bcdSBarry Smith   PetscTryTypeMethod(tao, setfromoptions, PetscOptionsObject);
52160b70c5cSStefano Zampini 
52260b70c5cSStefano Zampini   /* process any options handlers added with PetscObjectAddOptionsHandler() */
52360b70c5cSStefano Zampini   PetscCall(PetscObjectProcessOptionsHandlers((PetscObject)tao, PetscOptionsObject));
524d0609cedSBarry Smith   PetscOptionsEnd();
52560b70c5cSStefano Zampini 
52660b70c5cSStefano Zampini   if (tao->linesearch) PetscCall(TaoLineSearchSetFromOptions(tao->linesearch));
5273ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
528a7e14dcfSSatish Balay }
529a7e14dcfSSatish Balay 
530a7e14dcfSSatish Balay /*@C
53147450a7bSBarry Smith   TaoViewFromOptions - View a `Tao` object based on values in the options database
532fe2efc57SMark 
533c3339decSBarry Smith   Collective
534fe2efc57SMark 
535fe2efc57SMark   Input Parameters:
53647450a7bSBarry Smith + A    - the  `Tao` context
53747450a7bSBarry Smith . obj  - Optional object that provides the prefix for the options database
538736c3998SJose E. Roman - name - command line option
539fe2efc57SMark 
540fe2efc57SMark   Level: intermediate
54167be906fSBarry Smith 
5421cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoView`, `PetscObjectViewFromOptions()`, `TaoCreate()`
543fe2efc57SMark @*/
544d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoViewFromOptions(Tao A, PetscObject obj, const char name[])
545d71ae5a4SJacob Faibussowitsch {
546fe2efc57SMark   PetscFunctionBegin;
547fe2efc57SMark   PetscValidHeaderSpecific(A, TAO_CLASSID, 1);
5489566063dSJacob Faibussowitsch   PetscCall(PetscObjectViewFromOptions((PetscObject)A, obj, name));
5493ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
550fe2efc57SMark }
551fe2efc57SMark 
552fe2efc57SMark /*@C
55347450a7bSBarry Smith   TaoView - Prints information about the `Tao` object
554a7e14dcfSSatish Balay 
555c3339decSBarry Smith   Collective
556a7e14dcfSSatish Balay 
557a7e14dcfSSatish Balay   Input Parameters:
55847450a7bSBarry Smith + tao    - the `Tao` context
559a7e14dcfSSatish Balay - viewer - visualization context
560a7e14dcfSSatish Balay 
561a7e14dcfSSatish Balay   Options Database Key:
56265ba42b6SBarry Smith . -tao_view - Calls `TaoView()` at the end of `TaoSolve()`
563a7e14dcfSSatish Balay 
56467be906fSBarry Smith   Level: beginner
56567be906fSBarry Smith 
566a7e14dcfSSatish Balay   Notes:
567a7e14dcfSSatish Balay   The available visualization contexts include
56865ba42b6SBarry Smith +     `PETSC_VIEWER_STDOUT_SELF` - standard output (default)
56965ba42b6SBarry Smith -     `PETSC_VIEWER_STDOUT_WORLD` - synchronized standard
570a7e14dcfSSatish Balay   output where only the first processor opens
571a7e14dcfSSatish Balay   the file.  All other processors send their
572a7e14dcfSSatish Balay   data to the first processor to print.
573a7e14dcfSSatish Balay 
5741cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `PetscViewerASCIIOpen()`
575a7e14dcfSSatish Balay @*/
576d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoView(Tao tao, PetscViewer viewer)
577d71ae5a4SJacob Faibussowitsch {
578a7e14dcfSSatish Balay   PetscBool isascii, isstring;
579b625d6c7SJed Brown   TaoType   type;
58047a47007SBarry Smith 
581a7e14dcfSSatish Balay   PetscFunctionBegin;
582441846f8SBarry Smith   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
58348a46eb9SPierre Jolivet   if (!viewer) PetscCall(PetscViewerASCIIGetStdout(((PetscObject)tao)->comm, &viewer));
584a7e14dcfSSatish Balay   PetscValidHeaderSpecific(viewer, PETSC_VIEWER_CLASSID, 2);
585a7e14dcfSSatish Balay   PetscCheckSameComm(tao, 1, viewer, 2);
586a7e14dcfSSatish Balay 
5879566063dSJacob Faibussowitsch   PetscCall(PetscObjectTypeCompare((PetscObject)viewer, PETSCVIEWERASCII, &isascii));
5889566063dSJacob Faibussowitsch   PetscCall(PetscObjectTypeCompare((PetscObject)viewer, PETSCVIEWERSTRING, &isstring));
589a7e14dcfSSatish Balay   if (isascii) {
5909566063dSJacob Faibussowitsch     PetscCall(PetscObjectPrintClassNamePrefixType((PetscObject)tao, viewer));
591a7e14dcfSSatish Balay 
592a7e14dcfSSatish Balay     if (tao->ops->view) {
5939566063dSJacob Faibussowitsch       PetscCall(PetscViewerASCIIPushTab(viewer));
594dbbe0bcdSBarry Smith       PetscUseTypeMethod(tao, view, viewer);
5959566063dSJacob Faibussowitsch       PetscCall(PetscViewerASCIIPopTab(viewer));
596a7e14dcfSSatish Balay     }
5978d3f3ddaSAlp Dener     if (tao->linesearch) {
5989566063dSJacob Faibussowitsch       PetscCall(PetscViewerASCIIPushTab(viewer));
5999566063dSJacob Faibussowitsch       PetscCall(TaoLineSearchView(tao->linesearch, viewer));
6009566063dSJacob Faibussowitsch       PetscCall(PetscViewerASCIIPopTab(viewer));
6018d3f3ddaSAlp Dener     }
602a7e14dcfSSatish Balay     if (tao->ksp) {
6039566063dSJacob Faibussowitsch       PetscCall(PetscViewerASCIIPushTab(viewer));
6049566063dSJacob Faibussowitsch       PetscCall(KSPView(tao->ksp, viewer));
60563a3b9bcSJacob Faibussowitsch       PetscCall(PetscViewerASCIIPrintf(viewer, "total KSP iterations: %" PetscInt_FMT "\n", tao->ksp_tot_its));
6069566063dSJacob Faibussowitsch       PetscCall(PetscViewerASCIIPopTab(viewer));
607a7e14dcfSSatish Balay     }
60819b5c101SAlp Dener 
6099566063dSJacob Faibussowitsch     PetscCall(PetscViewerASCIIPushTab(viewer));
61019b5c101SAlp Dener 
61148a46eb9SPierre Jolivet     if (tao->XL || tao->XU) PetscCall(PetscViewerASCIIPrintf(viewer, "Active Set subset type: %s\n", TaoSubSetTypes[tao->subset_type]));
612a7e14dcfSSatish Balay 
6139566063dSJacob Faibussowitsch     PetscCall(PetscViewerASCIIPrintf(viewer, "convergence tolerances: gatol=%g,", (double)tao->gatol));
6149566063dSJacob Faibussowitsch     PetscCall(PetscViewerASCIIPrintf(viewer, " steptol=%g,", (double)tao->steptol));
6159566063dSJacob Faibussowitsch     PetscCall(PetscViewerASCIIPrintf(viewer, " gttol=%g\n", (double)tao->gttol));
6169566063dSJacob Faibussowitsch     PetscCall(PetscViewerASCIIPrintf(viewer, "Residual in Function/Gradient:=%g\n", (double)tao->residual));
617a7e14dcfSSatish Balay 
6186246e8cdSAlp Dener     if (tao->constrained) {
6199566063dSJacob Faibussowitsch       PetscCall(PetscViewerASCIIPrintf(viewer, "convergence tolerances:"));
6209566063dSJacob Faibussowitsch       PetscCall(PetscViewerASCIIPrintf(viewer, " catol=%g,", (double)tao->catol));
6219566063dSJacob Faibussowitsch       PetscCall(PetscViewerASCIIPrintf(viewer, " crtol=%g\n", (double)tao->crtol));
6229566063dSJacob Faibussowitsch       PetscCall(PetscViewerASCIIPrintf(viewer, "Residual in Constraints:=%g\n", (double)tao->cnorm));
623a7e14dcfSSatish Balay     }
624a7e14dcfSSatish Balay 
625a7e14dcfSSatish Balay     if (tao->trust < tao->steptol) {
6269566063dSJacob Faibussowitsch       PetscCall(PetscViewerASCIIPrintf(viewer, "convergence tolerances: steptol=%g\n", (double)tao->steptol));
6279566063dSJacob Faibussowitsch       PetscCall(PetscViewerASCIIPrintf(viewer, "Final trust region radius:=%g\n", (double)tao->trust));
628a7e14dcfSSatish Balay     }
629a7e14dcfSSatish Balay 
63048a46eb9SPierre Jolivet     if (tao->fmin > -1.e25) PetscCall(PetscViewerASCIIPrintf(viewer, "convergence tolerances: function minimum=%g\n", (double)tao->fmin));
6319566063dSJacob Faibussowitsch     PetscCall(PetscViewerASCIIPrintf(viewer, "Objective value=%g\n", (double)tao->fc));
632a7e14dcfSSatish Balay 
63363a3b9bcSJacob Faibussowitsch     PetscCall(PetscViewerASCIIPrintf(viewer, "total number of iterations=%" PetscInt_FMT ",          ", tao->niter));
63463a3b9bcSJacob Faibussowitsch     PetscCall(PetscViewerASCIIPrintf(viewer, "              (max: %" PetscInt_FMT ")\n", tao->max_it));
635a7e14dcfSSatish Balay 
636a7e14dcfSSatish Balay     if (tao->nfuncs > 0) {
63763a3b9bcSJacob Faibussowitsch       PetscCall(PetscViewerASCIIPrintf(viewer, "total number of function evaluations=%" PetscInt_FMT ",", tao->nfuncs));
63863a3b9bcSJacob Faibussowitsch       PetscCall(PetscViewerASCIIPrintf(viewer, "                max: %" PetscInt_FMT "\n", tao->max_funcs));
639a7e14dcfSSatish Balay     }
640a7e14dcfSSatish Balay     if (tao->ngrads > 0) {
64163a3b9bcSJacob Faibussowitsch       PetscCall(PetscViewerASCIIPrintf(viewer, "total number of gradient evaluations=%" PetscInt_FMT ",", tao->ngrads));
64263a3b9bcSJacob Faibussowitsch       PetscCall(PetscViewerASCIIPrintf(viewer, "                max: %" PetscInt_FMT "\n", tao->max_funcs));
643a7e14dcfSSatish Balay     }
644a7e14dcfSSatish Balay     if (tao->nfuncgrads > 0) {
64563a3b9bcSJacob Faibussowitsch       PetscCall(PetscViewerASCIIPrintf(viewer, "total number of function/gradient evaluations=%" PetscInt_FMT ",", tao->nfuncgrads));
64663a3b9bcSJacob Faibussowitsch       PetscCall(PetscViewerASCIIPrintf(viewer, "    (max: %" PetscInt_FMT ")\n", tao->max_funcs));
647a7e14dcfSSatish Balay     }
64848a46eb9SPierre Jolivet     if (tao->nhess > 0) PetscCall(PetscViewerASCIIPrintf(viewer, "total number of Hessian evaluations=%" PetscInt_FMT "\n", tao->nhess));
64948a46eb9SPierre Jolivet     if (tao->nconstraints > 0) PetscCall(PetscViewerASCIIPrintf(viewer, "total number of constraint function evaluations=%" PetscInt_FMT "\n", tao->nconstraints));
65048a46eb9SPierre Jolivet     if (tao->njac > 0) PetscCall(PetscViewerASCIIPrintf(viewer, "total number of Jacobian evaluations=%" PetscInt_FMT "\n", tao->njac));
651a7e14dcfSSatish Balay 
652a7e14dcfSSatish Balay     if (tao->reason > 0) {
6539566063dSJacob Faibussowitsch       PetscCall(PetscViewerASCIIPrintf(viewer, "Solution converged: "));
654a7e14dcfSSatish Balay       switch (tao->reason) {
655d71ae5a4SJacob Faibussowitsch       case TAO_CONVERGED_GATOL:
656d71ae5a4SJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, " ||g(X)|| <= gatol\n"));
657d71ae5a4SJacob Faibussowitsch         break;
658d71ae5a4SJacob Faibussowitsch       case TAO_CONVERGED_GRTOL:
659d71ae5a4SJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, " ||g(X)||/|f(X)| <= grtol\n"));
660d71ae5a4SJacob Faibussowitsch         break;
661d71ae5a4SJacob Faibussowitsch       case TAO_CONVERGED_GTTOL:
662d71ae5a4SJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, " ||g(X)||/||g(X0)|| <= gttol\n"));
663d71ae5a4SJacob Faibussowitsch         break;
664d71ae5a4SJacob Faibussowitsch       case TAO_CONVERGED_STEPTOL:
665d71ae5a4SJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, " Steptol -- step size small\n"));
666d71ae5a4SJacob Faibussowitsch         break;
667d71ae5a4SJacob Faibussowitsch       case TAO_CONVERGED_MINF:
668d71ae5a4SJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, " Minf --  f < fmin\n"));
669d71ae5a4SJacob Faibussowitsch         break;
670d71ae5a4SJacob Faibussowitsch       case TAO_CONVERGED_USER:
671d71ae5a4SJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, " User Terminated\n"));
672d71ae5a4SJacob Faibussowitsch         break;
673d71ae5a4SJacob Faibussowitsch       default:
674d71ae5a4SJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, "\n"));
675d71ae5a4SJacob Faibussowitsch         break;
676a7e14dcfSSatish Balay       }
677a7e14dcfSSatish Balay     } else {
6789566063dSJacob Faibussowitsch       PetscCall(PetscViewerASCIIPrintf(viewer, "Solver terminated: %d", tao->reason));
679a7e14dcfSSatish Balay       switch (tao->reason) {
680d71ae5a4SJacob Faibussowitsch       case TAO_DIVERGED_MAXITS:
681d71ae5a4SJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, " Maximum Iterations\n"));
682d71ae5a4SJacob Faibussowitsch         break;
683d71ae5a4SJacob Faibussowitsch       case TAO_DIVERGED_NAN:
684d71ae5a4SJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, " NAN or Inf encountered\n"));
685d71ae5a4SJacob Faibussowitsch         break;
686d71ae5a4SJacob Faibussowitsch       case TAO_DIVERGED_MAXFCN:
687d71ae5a4SJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, " Maximum Function Evaluations\n"));
688d71ae5a4SJacob Faibussowitsch         break;
689d71ae5a4SJacob Faibussowitsch       case TAO_DIVERGED_LS_FAILURE:
690d71ae5a4SJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, " Line Search Failure\n"));
691d71ae5a4SJacob Faibussowitsch         break;
692d71ae5a4SJacob Faibussowitsch       case TAO_DIVERGED_TR_REDUCTION:
693d71ae5a4SJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, " Trust Region too small\n"));
694d71ae5a4SJacob Faibussowitsch         break;
695d71ae5a4SJacob Faibussowitsch       case TAO_DIVERGED_USER:
696d71ae5a4SJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, " User Terminated\n"));
697d71ae5a4SJacob Faibussowitsch         break;
698d71ae5a4SJacob Faibussowitsch       default:
699d71ae5a4SJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, "\n"));
700d71ae5a4SJacob Faibussowitsch         break;
701a7e14dcfSSatish Balay       }
702a7e14dcfSSatish Balay     }
7039566063dSJacob Faibussowitsch     PetscCall(PetscViewerASCIIPopTab(viewer));
704a7e14dcfSSatish Balay   } else if (isstring) {
7059566063dSJacob Faibussowitsch     PetscCall(TaoGetType(tao, &type));
7069566063dSJacob Faibussowitsch     PetscCall(PetscViewerStringSPrintf(viewer, " %-3.3s", type));
707a7e14dcfSSatish Balay   }
7083ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
709a7e14dcfSSatish Balay }
710a7e14dcfSSatish Balay 
711a7e14dcfSSatish Balay /*@
712414d97d3SAlp Dener   TaoSetRecycleHistory - Sets the boolean flag to enable/disable re-using
71365ba42b6SBarry Smith   iterate information from the previous `TaoSolve()`. This feature is disabled by
714414d97d3SAlp Dener   default.
715414d97d3SAlp Dener 
71667be906fSBarry Smith   Logically Collective
71767be906fSBarry Smith 
71867be906fSBarry Smith   Input Parameters:
71947450a7bSBarry Smith + tao     - the `Tao` context
72067be906fSBarry Smith - recycle - boolean flag
72167be906fSBarry Smith 
72247450a7bSBarry Smith   Options Database Key:
72367be906fSBarry Smith . -tao_recycle_history <true,false> - reuse the history
72467be906fSBarry Smith 
72567be906fSBarry Smith   Level: intermediate
72667be906fSBarry Smith 
72767be906fSBarry Smith   Notes:
72865ba42b6SBarry Smith   For conjugate gradient methods (`TAOBNCG`), this re-uses the latest search direction
72965ba42b6SBarry Smith   from the previous `TaoSolve()` call when computing the first search direction in a
730414d97d3SAlp Dener   new solution. By default, CG methods set the first search direction to the
731414d97d3SAlp Dener   negative gradient.
732414d97d3SAlp Dener 
73365ba42b6SBarry Smith   For quasi-Newton family of methods (`TAOBQNLS`, `TAOBQNKLS`, `TAOBQNKTR`, `TAOBQNKTL`), this re-uses
73465ba42b6SBarry Smith   the accumulated quasi-Newton Hessian approximation from the previous `TaoSolve()`
735414d97d3SAlp Dener   call. By default, QN family of methods reset the initial Hessian approximation to
736414d97d3SAlp Dener   the identity matrix.
737414d97d3SAlp Dener 
738414d97d3SAlp Dener   For any other algorithm, this setting has no effect.
739414d97d3SAlp Dener 
7401cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoGetRecycleHistory()`, `TAOBNCG`, `TAOBQNLS`, `TAOBQNKLS`, `TAOBQNKTR`, `TAOBQNKTL`
741414d97d3SAlp Dener @*/
742d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoSetRecycleHistory(Tao tao, PetscBool recycle)
743d71ae5a4SJacob Faibussowitsch {
744414d97d3SAlp Dener   PetscFunctionBegin;
745414d97d3SAlp Dener   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
74694511df7SStefano Zampini   PetscValidLogicalCollectiveBool(tao, recycle, 2);
747414d97d3SAlp Dener   tao->recycle = recycle;
7483ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
749414d97d3SAlp Dener }
750414d97d3SAlp Dener 
751414d97d3SAlp Dener /*@
752414d97d3SAlp Dener   TaoGetRecycleHistory - Retrieve the boolean flag for re-using iterate information
75365ba42b6SBarry Smith   from the previous `TaoSolve()`. This feature is disabled by default.
754414d97d3SAlp Dener 
75567be906fSBarry Smith   Logically Collective
756414d97d3SAlp Dener 
75747450a7bSBarry Smith   Input Parameter:
75847450a7bSBarry Smith . tao - the `Tao` context
759414d97d3SAlp Dener 
76047450a7bSBarry Smith   Output Parameter:
761147403d9SBarry Smith . recycle - boolean flag
762414d97d3SAlp Dener 
763414d97d3SAlp Dener   Level: intermediate
764414d97d3SAlp Dener 
7651cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoSetRecycleHistory()`, `TAOBNCG`, `TAOBQNLS`, `TAOBQNKLS`, `TAOBQNKTR`, `TAOBQNKTL`
766414d97d3SAlp Dener @*/
767d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoGetRecycleHistory(Tao tao, PetscBool *recycle)
768d71ae5a4SJacob Faibussowitsch {
769414d97d3SAlp Dener   PetscFunctionBegin;
770414d97d3SAlp Dener   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
7714f572ea9SToby Isaac   PetscAssertPointer(recycle, 2);
772414d97d3SAlp Dener   *recycle = tao->recycle;
7733ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
774414d97d3SAlp Dener }
775414d97d3SAlp Dener 
776414d97d3SAlp Dener /*@
77747450a7bSBarry Smith   TaoSetTolerances - Sets parameters used in `TaoSolve()` convergence tests
778a7e14dcfSSatish Balay 
77967be906fSBarry Smith   Logically Collective
780a7e14dcfSSatish Balay 
781a7e14dcfSSatish Balay   Input Parameters:
78247450a7bSBarry Smith + tao   - the `Tao` context
783a7e14dcfSSatish Balay . gatol - stop if norm of gradient is less than this
784a7e14dcfSSatish Balay . grtol - stop if relative norm of gradient is less than this
785a7e14dcfSSatish Balay - gttol - stop if norm of gradient is reduced by this factor
786a7e14dcfSSatish Balay 
787a7e14dcfSSatish Balay   Options Database Keys:
788e52336cbSBarry Smith + -tao_gatol <gatol> - Sets gatol
789a7e14dcfSSatish Balay . -tao_grtol <grtol> - Sets grtol
790a7e14dcfSSatish Balay - -tao_gttol <gttol> - Sets gttol
791a7e14dcfSSatish Balay 
7923b242c63SJacob Faibussowitsch   Stopping Criteria\:
79367be906fSBarry Smith .vb
79467be906fSBarry Smith   ||g(X)||                            <= gatol
79567be906fSBarry Smith   ||g(X)|| / |f(X)|                   <= grtol
79667be906fSBarry Smith   ||g(X)|| / ||g(X0)||                <= gttol
79767be906fSBarry Smith .ve
798a7e14dcfSSatish Balay 
799a7e14dcfSSatish Balay   Level: beginner
800a7e14dcfSSatish Balay 
80167be906fSBarry Smith   Note:
80267be906fSBarry Smith   Use `PETSC_DEFAULT` to leave one or more tolerances unchanged.
803a7e14dcfSSatish Balay 
8041cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoConvergedReason`, `TaoGetTolerances()`
805a7e14dcfSSatish Balay @*/
806d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoSetTolerances(Tao tao, PetscReal gatol, PetscReal grtol, PetscReal gttol)
807d71ae5a4SJacob Faibussowitsch {
808a7e14dcfSSatish Balay   PetscFunctionBegin;
809441846f8SBarry Smith   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
81094511df7SStefano Zampini   PetscValidLogicalCollectiveReal(tao, gatol, 2);
81194511df7SStefano Zampini   PetscValidLogicalCollectiveReal(tao, grtol, 3);
81294511df7SStefano Zampini   PetscValidLogicalCollectiveReal(tao, gttol, 4);
813a7e14dcfSSatish Balay 
81413bcc0bdSJacob Faibussowitsch   if (gatol != (PetscReal)PETSC_DEFAULT) {
815a7e14dcfSSatish Balay     if (gatol < 0) {
8169566063dSJacob Faibussowitsch       PetscCall(PetscInfo(tao, "Tried to set negative gatol -- ignored.\n"));
817a7e14dcfSSatish Balay     } else {
818a7e14dcfSSatish Balay       tao->gatol         = PetscMax(0, gatol);
8196552cf8aSJason Sarich       tao->gatol_changed = PETSC_TRUE;
820a7e14dcfSSatish Balay     }
821a7e14dcfSSatish Balay   }
822a7e14dcfSSatish Balay 
82313bcc0bdSJacob Faibussowitsch   if (grtol != (PetscReal)PETSC_DEFAULT) {
824a7e14dcfSSatish Balay     if (grtol < 0) {
8259566063dSJacob Faibussowitsch       PetscCall(PetscInfo(tao, "Tried to set negative grtol -- ignored.\n"));
826a7e14dcfSSatish Balay     } else {
827a7e14dcfSSatish Balay       tao->grtol         = PetscMax(0, grtol);
8286552cf8aSJason Sarich       tao->grtol_changed = PETSC_TRUE;
829a7e14dcfSSatish Balay     }
830a7e14dcfSSatish Balay   }
831a7e14dcfSSatish Balay 
83213bcc0bdSJacob Faibussowitsch   if (gttol != (PetscReal)PETSC_DEFAULT) {
833a7e14dcfSSatish Balay     if (gttol < 0) {
8349566063dSJacob Faibussowitsch       PetscCall(PetscInfo(tao, "Tried to set negative gttol -- ignored.\n"));
835a7e14dcfSSatish Balay     } else {
836a7e14dcfSSatish Balay       tao->gttol         = PetscMax(0, gttol);
8376552cf8aSJason Sarich       tao->gttol_changed = PETSC_TRUE;
838a7e14dcfSSatish Balay     }
839a7e14dcfSSatish Balay   }
8403ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
841a7e14dcfSSatish Balay }
842a7e14dcfSSatish Balay 
843a7e14dcfSSatish Balay /*@
84447450a7bSBarry Smith   TaoSetConstraintTolerances - Sets constraint tolerance parameters used in `TaoSolve()` convergence tests
845a7e14dcfSSatish Balay 
84667be906fSBarry Smith   Logically Collective
847a7e14dcfSSatish Balay 
848a7e14dcfSSatish Balay   Input Parameters:
84947450a7bSBarry Smith + tao   - the `Tao` context
85067be906fSBarry Smith . catol - absolute constraint tolerance, constraint norm must be less than `catol` for used for gatol convergence criteria
85167be906fSBarry Smith - crtol - relative constraint tolerance, constraint norm must be less than `crtol` for used for gatol, gttol convergence criteria
852a7e14dcfSSatish Balay 
853a7e14dcfSSatish Balay   Options Database Keys:
854a7e14dcfSSatish Balay + -tao_catol <catol> - Sets catol
855a7e14dcfSSatish Balay - -tao_crtol <crtol> - Sets crtol
856a7e14dcfSSatish Balay 
857a7e14dcfSSatish Balay   Level: intermediate
858a7e14dcfSSatish Balay 
85967be906fSBarry Smith   Notes:
86067be906fSBarry Smith   Use `PETSC_DEFAULT` to leave any tolerance unchanged.
861a7e14dcfSSatish Balay 
8621cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoConvergedReason`, `TaoGetTolerances()`, `TaoGetConstraintTolerances()`, `TaoSetTolerances()`
863a7e14dcfSSatish Balay @*/
864d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoSetConstraintTolerances(Tao tao, PetscReal catol, PetscReal crtol)
865d71ae5a4SJacob Faibussowitsch {
866a7e14dcfSSatish Balay   PetscFunctionBegin;
867441846f8SBarry Smith   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
86894511df7SStefano Zampini   PetscValidLogicalCollectiveReal(tao, catol, 2);
86994511df7SStefano Zampini   PetscValidLogicalCollectiveReal(tao, crtol, 3);
870a7e14dcfSSatish Balay 
87113bcc0bdSJacob Faibussowitsch   if (catol != (PetscReal)PETSC_DEFAULT) {
872a7e14dcfSSatish Balay     if (catol < 0) {
8739566063dSJacob Faibussowitsch       PetscCall(PetscInfo(tao, "Tried to set negative catol -- ignored.\n"));
874a7e14dcfSSatish Balay     } else {
875a7e14dcfSSatish Balay       tao->catol         = PetscMax(0, catol);
8766552cf8aSJason Sarich       tao->catol_changed = PETSC_TRUE;
877a7e14dcfSSatish Balay     }
878a7e14dcfSSatish Balay   }
879a7e14dcfSSatish Balay 
88013bcc0bdSJacob Faibussowitsch   if (crtol != (PetscReal)PETSC_DEFAULT) {
881a7e14dcfSSatish Balay     if (crtol < 0) {
8829566063dSJacob Faibussowitsch       PetscCall(PetscInfo(tao, "Tried to set negative crtol -- ignored.\n"));
883a7e14dcfSSatish Balay     } else {
884a7e14dcfSSatish Balay       tao->crtol         = PetscMax(0, crtol);
8856552cf8aSJason Sarich       tao->crtol_changed = PETSC_TRUE;
886a7e14dcfSSatish Balay     }
887a7e14dcfSSatish Balay   }
8883ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
889a7e14dcfSSatish Balay }
890a7e14dcfSSatish Balay 
89158417fe7SBarry Smith /*@
89247450a7bSBarry Smith   TaoGetConstraintTolerances - Gets constraint tolerance parameters used in `TaoSolve()` convergence tests
89358417fe7SBarry Smith 
89467be906fSBarry Smith   Not Collective
89558417fe7SBarry Smith 
89658417fe7SBarry Smith   Input Parameter:
89747450a7bSBarry Smith . tao - the `Tao` context
89858417fe7SBarry Smith 
899d8d19677SJose E. Roman   Output Parameters:
90067be906fSBarry Smith + catol - absolute constraint tolerance, constraint norm must be less than `catol` for used for gatol convergence criteria
90167be906fSBarry Smith - crtol - relative constraint tolerance, constraint norm must be less than `crtol` for used for gatol, gttol convergence criteria
90258417fe7SBarry Smith 
90358417fe7SBarry Smith   Level: intermediate
90458417fe7SBarry Smith 
9051cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoConvergedReasons`,`TaoGetTolerances()`, `TaoSetTolerances()`, `TaoSetConstraintTolerances()`
90658417fe7SBarry Smith @*/
907d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoGetConstraintTolerances(Tao tao, PetscReal *catol, PetscReal *crtol)
908d71ae5a4SJacob Faibussowitsch {
90958417fe7SBarry Smith   PetscFunctionBegin;
91058417fe7SBarry Smith   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
91158417fe7SBarry Smith   if (catol) *catol = tao->catol;
91258417fe7SBarry Smith   if (crtol) *crtol = tao->crtol;
9133ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
91458417fe7SBarry Smith }
91558417fe7SBarry Smith 
916a7e14dcfSSatish Balay /*@
917a7e14dcfSSatish Balay   TaoSetFunctionLowerBound - Sets a bound on the solution objective value.
918a7e14dcfSSatish Balay   When an approximate solution with an objective value below this number
919a7e14dcfSSatish Balay   has been found, the solver will terminate.
920a7e14dcfSSatish Balay 
921c3339decSBarry Smith   Logically Collective
922a7e14dcfSSatish Balay 
923a7e14dcfSSatish Balay   Input Parameters:
924441846f8SBarry Smith + tao  - the Tao solver context
925a7e14dcfSSatish Balay - fmin - the tolerance
926a7e14dcfSSatish Balay 
92747450a7bSBarry Smith   Options Database Key:
928a7e14dcfSSatish Balay . -tao_fmin <fmin> - sets the minimum function value
929a7e14dcfSSatish Balay 
930a7e14dcfSSatish Balay   Level: intermediate
931a7e14dcfSSatish Balay 
9321cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoConvergedReason`, `TaoSetTolerances()`
933a7e14dcfSSatish Balay @*/
934d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoSetFunctionLowerBound(Tao tao, PetscReal fmin)
935d71ae5a4SJacob Faibussowitsch {
936a7e14dcfSSatish Balay   PetscFunctionBegin;
937441846f8SBarry Smith   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
93894511df7SStefano Zampini   PetscValidLogicalCollectiveReal(tao, fmin, 2);
939a7e14dcfSSatish Balay   tao->fmin         = fmin;
9406552cf8aSJason Sarich   tao->fmin_changed = PETSC_TRUE;
9413ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
942a7e14dcfSSatish Balay }
943a7e14dcfSSatish Balay 
944a7e14dcfSSatish Balay /*@
9458e96d397SJason Sarich   TaoGetFunctionLowerBound - Gets the bound on the solution objective value.
946a7e14dcfSSatish Balay   When an approximate solution with an objective value below this number
947a7e14dcfSSatish Balay   has been found, the solver will terminate.
948a7e14dcfSSatish Balay 
94967be906fSBarry Smith   Not Collective
950a7e14dcfSSatish Balay 
95147450a7bSBarry Smith   Input Parameter:
95247450a7bSBarry Smith . tao - the `Tao` solver context
953a7e14dcfSSatish Balay 
95447450a7bSBarry Smith   Output Parameter:
955a7e14dcfSSatish Balay . fmin - the minimum function value
956a7e14dcfSSatish Balay 
957a7e14dcfSSatish Balay   Level: intermediate
958a7e14dcfSSatish Balay 
9591cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoConvergedReason`, `TaoSetFunctionLowerBound()`
960a7e14dcfSSatish Balay @*/
961d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoGetFunctionLowerBound(Tao tao, PetscReal *fmin)
962d71ae5a4SJacob Faibussowitsch {
963a7e14dcfSSatish Balay   PetscFunctionBegin;
964441846f8SBarry Smith   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
9654f572ea9SToby Isaac   PetscAssertPointer(fmin, 2);
966a7e14dcfSSatish Balay   *fmin = tao->fmin;
9673ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
968a7e14dcfSSatish Balay }
969a7e14dcfSSatish Balay 
970a7e14dcfSSatish Balay /*@
97147450a7bSBarry Smith   TaoSetMaximumFunctionEvaluations - Sets a maximum number of function evaluations allowed for a `TaoSolve()`.
972a7e14dcfSSatish Balay 
973c3339decSBarry Smith   Logically Collective
974a7e14dcfSSatish Balay 
975a7e14dcfSSatish Balay   Input Parameters:
97647450a7bSBarry Smith + tao  - the `Tao` solver context
977a7e14dcfSSatish Balay - nfcn - the maximum number of function evaluations (>=0)
978a7e14dcfSSatish Balay 
97947450a7bSBarry Smith   Options Database Key:
980a7e14dcfSSatish Balay . -tao_max_funcs <nfcn> - sets the maximum number of function evaluations
981a7e14dcfSSatish Balay 
982a7e14dcfSSatish Balay   Level: intermediate
983a7e14dcfSSatish Balay 
9841cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoSetTolerances()`, `TaoSetMaximumIterations()`
985a7e14dcfSSatish Balay @*/
986d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoSetMaximumFunctionEvaluations(Tao tao, PetscInt nfcn)
987d71ae5a4SJacob Faibussowitsch {
988a7e14dcfSSatish Balay   PetscFunctionBegin;
989441846f8SBarry Smith   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
99094511df7SStefano Zampini   PetscValidLogicalCollectiveInt(tao, nfcn, 2);
9919371c9d4SSatish Balay   if (nfcn >= 0) {
9929371c9d4SSatish Balay     tao->max_funcs = PetscMax(0, nfcn);
9939371c9d4SSatish Balay   } else {
9949371c9d4SSatish Balay     tao->max_funcs = -1;
9959371c9d4SSatish Balay   }
9966552cf8aSJason Sarich   tao->max_funcs_changed = PETSC_TRUE;
9973ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
998a7e14dcfSSatish Balay }
999a7e14dcfSSatish Balay 
1000a7e14dcfSSatish Balay /*@
100147450a7bSBarry Smith   TaoGetMaximumFunctionEvaluations - Gets a maximum number of function evaluations allowed for a `TaoSolve()`
1002a7e14dcfSSatish Balay 
1003c3339decSBarry Smith   Logically Collective
1004a7e14dcfSSatish Balay 
100547450a7bSBarry Smith   Input Parameter:
100647450a7bSBarry Smith . tao - the `Tao` solver context
1007a7e14dcfSSatish Balay 
100847450a7bSBarry Smith   Output Parameter:
1009a7e14dcfSSatish Balay . nfcn - the maximum number of function evaluations
1010a7e14dcfSSatish Balay 
1011a7e14dcfSSatish Balay   Level: intermediate
1012a7e14dcfSSatish Balay 
10131cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoSetMaximumFunctionEvaluations()`, `TaoGetMaximumIterations()`
1014a7e14dcfSSatish Balay @*/
1015d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoGetMaximumFunctionEvaluations(Tao tao, PetscInt *nfcn)
1016d71ae5a4SJacob Faibussowitsch {
1017a7e14dcfSSatish Balay   PetscFunctionBegin;
1018441846f8SBarry Smith   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
10194f572ea9SToby Isaac   PetscAssertPointer(nfcn, 2);
1020a7e14dcfSSatish Balay   *nfcn = tao->max_funcs;
10213ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
1022a7e14dcfSSatish Balay }
1023a7e14dcfSSatish Balay 
1024770232b9SCe Qin /*@
102547450a7bSBarry Smith   TaoGetCurrentFunctionEvaluations - Get current number of function evaluations used by a `Tao` object
1026770232b9SCe Qin 
1027770232b9SCe Qin   Not Collective
1028770232b9SCe Qin 
102947450a7bSBarry Smith   Input Parameter:
103047450a7bSBarry Smith . tao - the `Tao` solver context
1031770232b9SCe Qin 
103247450a7bSBarry Smith   Output Parameter:
103394511df7SStefano Zampini . nfuncs - the current number of function evaluations (maximum between gradient and function evaluations)
1034770232b9SCe Qin 
1035770232b9SCe Qin   Level: intermediate
1036770232b9SCe Qin 
10371cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoSetMaximumFunctionEvaluations()`, `TaoGetMaximumFunctionEvaluations()`, `TaoGetMaximumIterations()`
1038770232b9SCe Qin @*/
1039d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoGetCurrentFunctionEvaluations(Tao tao, PetscInt *nfuncs)
1040d71ae5a4SJacob Faibussowitsch {
1041770232b9SCe Qin   PetscFunctionBegin;
1042770232b9SCe Qin   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
10434f572ea9SToby Isaac   PetscAssertPointer(nfuncs, 2);
1044770232b9SCe Qin   *nfuncs = PetscMax(tao->nfuncs, tao->nfuncgrads);
10453ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
1046770232b9SCe Qin }
1047770232b9SCe Qin 
1048a7e14dcfSSatish Balay /*@
104947450a7bSBarry Smith   TaoSetMaximumIterations - Sets a maximum number of iterates to be used in `TaoSolve()`
1050a7e14dcfSSatish Balay 
1051c3339decSBarry Smith   Logically Collective
1052a7e14dcfSSatish Balay 
1053a7e14dcfSSatish Balay   Input Parameters:
105447450a7bSBarry Smith + tao    - the `Tao` solver context
1055a7e14dcfSSatish Balay - maxits - the maximum number of iterates (>=0)
1056a7e14dcfSSatish Balay 
105747450a7bSBarry Smith   Options Database Key:
1058a7e14dcfSSatish Balay . -tao_max_it <its> - sets the maximum number of iterations
1059a7e14dcfSSatish Balay 
1060a7e14dcfSSatish Balay   Level: intermediate
1061a7e14dcfSSatish Balay 
10621cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoSetTolerances()`, `TaoSetMaximumFunctionEvaluations()`
1063a7e14dcfSSatish Balay @*/
1064d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoSetMaximumIterations(Tao tao, PetscInt maxits)
1065d71ae5a4SJacob Faibussowitsch {
1066a7e14dcfSSatish Balay   PetscFunctionBegin;
1067441846f8SBarry Smith   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
106894511df7SStefano Zampini   PetscValidLogicalCollectiveInt(tao, maxits, 2);
106947a47007SBarry Smith   tao->max_it         = PetscMax(0, maxits);
10706552cf8aSJason Sarich   tao->max_it_changed = PETSC_TRUE;
10713ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
1072a7e14dcfSSatish Balay }
1073a7e14dcfSSatish Balay 
1074a7e14dcfSSatish Balay /*@
107565ba42b6SBarry Smith   TaoGetMaximumIterations - Gets a maximum number of iterates that will be used
1076a7e14dcfSSatish Balay 
1077a7e14dcfSSatish Balay   Not Collective
1078a7e14dcfSSatish Balay 
107947450a7bSBarry Smith   Input Parameter:
108047450a7bSBarry Smith . tao - the `Tao` solver context
1081a7e14dcfSSatish Balay 
108247450a7bSBarry Smith   Output Parameter:
1083a7e14dcfSSatish Balay . maxits - the maximum number of iterates
1084a7e14dcfSSatish Balay 
1085a7e14dcfSSatish Balay   Level: intermediate
1086a7e14dcfSSatish Balay 
10871cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoSetMaximumIterations()`, `TaoGetMaximumFunctionEvaluations()`
1088a7e14dcfSSatish Balay @*/
1089d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoGetMaximumIterations(Tao tao, PetscInt *maxits)
1090d71ae5a4SJacob Faibussowitsch {
1091a7e14dcfSSatish Balay   PetscFunctionBegin;
1092441846f8SBarry Smith   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
10934f572ea9SToby Isaac   PetscAssertPointer(maxits, 2);
1094a7e14dcfSSatish Balay   *maxits = tao->max_it;
10953ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
1096a7e14dcfSSatish Balay }
1097a7e14dcfSSatish Balay 
1098a7e14dcfSSatish Balay /*@
1099a7e14dcfSSatish Balay   TaoSetInitialTrustRegionRadius - Sets the initial trust region radius.
1100a7e14dcfSSatish Balay 
110167be906fSBarry Smith   Logically Collective
1102a7e14dcfSSatish Balay 
1103d8d19677SJose E. Roman   Input Parameters:
110447450a7bSBarry Smith + tao    - a `Tao` optimization solver
1105a7e14dcfSSatish Balay - radius - the trust region radius
1106a7e14dcfSSatish Balay 
1107a7e14dcfSSatish Balay   Options Database Key:
1108a7e14dcfSSatish Balay . -tao_trust0 <t0> - sets initial trust region radius
1109a7e14dcfSSatish Balay 
111047450a7bSBarry Smith   Level: intermediate
111147450a7bSBarry Smith 
11121cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoGetTrustRegionRadius()`, `TaoSetTrustRegionTolerance()`, `TAONTR`
1113a7e14dcfSSatish Balay @*/
1114d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoSetInitialTrustRegionRadius(Tao tao, PetscReal radius)
1115d71ae5a4SJacob Faibussowitsch {
1116a7e14dcfSSatish Balay   PetscFunctionBegin;
1117441846f8SBarry Smith   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
111894511df7SStefano Zampini   PetscValidLogicalCollectiveReal(tao, radius, 2);
1119a7e14dcfSSatish Balay   tao->trust0         = PetscMax(0.0, radius);
11206552cf8aSJason Sarich   tao->trust0_changed = PETSC_TRUE;
11213ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
1122a7e14dcfSSatish Balay }
1123a7e14dcfSSatish Balay 
1124a7e14dcfSSatish Balay /*@
112565ba42b6SBarry Smith   TaoGetInitialTrustRegionRadius - Gets the initial trust region radius.
1126a7e14dcfSSatish Balay 
1127a7e14dcfSSatish Balay   Not Collective
1128a7e14dcfSSatish Balay 
1129a7e14dcfSSatish Balay   Input Parameter:
113047450a7bSBarry Smith . tao - a `Tao` optimization solver
1131a7e14dcfSSatish Balay 
1132a7e14dcfSSatish Balay   Output Parameter:
1133a7e14dcfSSatish Balay . radius - the trust region radius
1134a7e14dcfSSatish Balay 
1135a7e14dcfSSatish Balay   Level: intermediate
1136a7e14dcfSSatish Balay 
11371cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoSetInitialTrustRegionRadius()`, `TaoGetCurrentTrustRegionRadius()`, `TAONTR`
1138a7e14dcfSSatish Balay @*/
1139d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoGetInitialTrustRegionRadius(Tao tao, PetscReal *radius)
1140d71ae5a4SJacob Faibussowitsch {
1141a7e14dcfSSatish Balay   PetscFunctionBegin;
1142441846f8SBarry Smith   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
11434f572ea9SToby Isaac   PetscAssertPointer(radius, 2);
1144a7e14dcfSSatish Balay   *radius = tao->trust0;
11453ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
1146a7e14dcfSSatish Balay }
1147a7e14dcfSSatish Balay 
1148a7e14dcfSSatish Balay /*@
1149a7e14dcfSSatish Balay   TaoGetCurrentTrustRegionRadius - Gets the current trust region radius.
1150a7e14dcfSSatish Balay 
1151a7e14dcfSSatish Balay   Not Collective
1152a7e14dcfSSatish Balay 
1153a7e14dcfSSatish Balay   Input Parameter:
115447450a7bSBarry Smith . tao - a `Tao` optimization solver
1155a7e14dcfSSatish Balay 
1156a7e14dcfSSatish Balay   Output Parameter:
1157a7e14dcfSSatish Balay . radius - the trust region radius
1158a7e14dcfSSatish Balay 
1159a7e14dcfSSatish Balay   Level: intermediate
1160a7e14dcfSSatish Balay 
11611cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoSetInitialTrustRegionRadius()`, `TaoGetInitialTrustRegionRadius()`, `TAONTR`
1162a7e14dcfSSatish Balay @*/
1163d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoGetCurrentTrustRegionRadius(Tao tao, PetscReal *radius)
1164d71ae5a4SJacob Faibussowitsch {
1165a7e14dcfSSatish Balay   PetscFunctionBegin;
1166441846f8SBarry Smith   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
11674f572ea9SToby Isaac   PetscAssertPointer(radius, 2);
1168a7e14dcfSSatish Balay   *radius = tao->trust;
11693ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
1170a7e14dcfSSatish Balay }
1171a7e14dcfSSatish Balay 
1172a7e14dcfSSatish Balay /*@
117347450a7bSBarry Smith   TaoGetTolerances - gets the current values of some tolerances used for the convergence testing of `TaoSolve()`
1174a7e14dcfSSatish Balay 
1175a7e14dcfSSatish Balay   Not Collective
1176a7e14dcfSSatish Balay 
1177f899ff85SJose E. Roman   Input Parameter:
117847450a7bSBarry Smith . tao - the `Tao` context
1179a7e14dcfSSatish Balay 
1180a7e14dcfSSatish Balay   Output Parameters:
1181e52336cbSBarry Smith + gatol - stop if norm of gradient is less than this
1182a7e14dcfSSatish Balay . grtol - stop if relative norm of gradient is less than this
1183a7e14dcfSSatish Balay - gttol - stop if norm of gradient is reduced by a this factor
1184a7e14dcfSSatish Balay 
1185a7e14dcfSSatish Balay   Level: intermediate
1186a1cb98faSBarry Smith 
1187a1cb98faSBarry Smith   Note:
118847450a7bSBarry Smith   `NULL` can be used as an argument if not all tolerances values are needed
1189a1cb98faSBarry Smith 
11901cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoSetTolerances()`
1191a7e14dcfSSatish Balay @*/
1192d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoGetTolerances(Tao tao, PetscReal *gatol, PetscReal *grtol, PetscReal *gttol)
1193d71ae5a4SJacob Faibussowitsch {
1194a7e14dcfSSatish Balay   PetscFunctionBegin;
1195441846f8SBarry Smith   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
1196a7e14dcfSSatish Balay   if (gatol) *gatol = tao->gatol;
1197a7e14dcfSSatish Balay   if (grtol) *grtol = tao->grtol;
1198a7e14dcfSSatish Balay   if (gttol) *gttol = tao->gttol;
11993ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
1200a7e14dcfSSatish Balay }
1201a7e14dcfSSatish Balay 
1202a7e14dcfSSatish Balay /*@
1203a7e14dcfSSatish Balay   TaoGetKSP - Gets the linear solver used by the optimization solver.
1204a7e14dcfSSatish Balay 
1205a7e14dcfSSatish Balay   Not Collective
1206a7e14dcfSSatish Balay 
120747450a7bSBarry Smith   Input Parameter:
120847450a7bSBarry Smith . tao - the `Tao` solver
1209a7e14dcfSSatish Balay 
121047450a7bSBarry Smith   Output Parameter:
121147450a7bSBarry Smith . ksp - the `KSP` linear solver used in the optimization solver
1212a7e14dcfSSatish Balay 
1213a7e14dcfSSatish Balay   Level: intermediate
1214a7e14dcfSSatish Balay 
12151cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `KSP`
1216a7e14dcfSSatish Balay @*/
1217d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoGetKSP(Tao tao, KSP *ksp)
1218d71ae5a4SJacob Faibussowitsch {
1219a7e14dcfSSatish Balay   PetscFunctionBegin;
122094511df7SStefano Zampini   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
12214f572ea9SToby Isaac   PetscAssertPointer(ksp, 2);
1222a7e14dcfSSatish Balay   *ksp = tao->ksp;
12233ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
1224a7e14dcfSSatish Balay }
1225a7e14dcfSSatish Balay 
1226025e9500SJason Sarich /*@
1227025e9500SJason Sarich   TaoGetLinearSolveIterations - Gets the total number of linear iterations
122847450a7bSBarry Smith   used by the `Tao` solver
1229025e9500SJason Sarich 
1230025e9500SJason Sarich   Not Collective
1231025e9500SJason Sarich 
1232025e9500SJason Sarich   Input Parameter:
123347450a7bSBarry Smith . tao - the `Tao` context
1234025e9500SJason Sarich 
1235025e9500SJason Sarich   Output Parameter:
1236025e9500SJason Sarich . lits - number of linear iterations
1237025e9500SJason Sarich 
1238025e9500SJason Sarich   Level: intermediate
1239025e9500SJason Sarich 
124047450a7bSBarry Smith   Note:
124147450a7bSBarry Smith   This counter is reset to zero for each successive call to `TaoSolve()`
124247450a7bSBarry Smith 
12431cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoGetKSP()`
1244025e9500SJason Sarich @*/
1245d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoGetLinearSolveIterations(Tao tao, PetscInt *lits)
1246d71ae5a4SJacob Faibussowitsch {
1247025e9500SJason Sarich   PetscFunctionBegin;
1248025e9500SJason Sarich   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
12494f572ea9SToby Isaac   PetscAssertPointer(lits, 2);
12502d9aa51bSJason Sarich   *lits = tao->ksp_tot_its;
12513ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
1252025e9500SJason Sarich }
1253025e9500SJason Sarich 
1254a7e14dcfSSatish Balay /*@
1255a7e14dcfSSatish Balay   TaoGetLineSearch - Gets the line search used by the optimization solver.
1256a7e14dcfSSatish Balay 
1257a7e14dcfSSatish Balay   Not Collective
1258a7e14dcfSSatish Balay 
125947450a7bSBarry Smith   Input Parameter:
126047450a7bSBarry Smith . tao - the `Tao` solver
1261a7e14dcfSSatish Balay 
126247450a7bSBarry Smith   Output Parameter:
1263a7e14dcfSSatish Balay . ls - the line search used in the optimization solver
1264a7e14dcfSSatish Balay 
1265a7e14dcfSSatish Balay   Level: intermediate
1266a7e14dcfSSatish Balay 
12671cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoLineSearch`, `TaoLineSearchType`
1268a7e14dcfSSatish Balay @*/
1269d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoGetLineSearch(Tao tao, TaoLineSearch *ls)
1270d71ae5a4SJacob Faibussowitsch {
1271a7e14dcfSSatish Balay   PetscFunctionBegin;
127294511df7SStefano Zampini   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
12734f572ea9SToby Isaac   PetscAssertPointer(ls, 2);
1274a7e14dcfSSatish Balay   *ls = tao->linesearch;
12753ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
1276a7e14dcfSSatish Balay }
1277a7e14dcfSSatish Balay 
1278a7e14dcfSSatish Balay /*@
1279a7e14dcfSSatish Balay   TaoAddLineSearchCounts - Adds the number of function evaluations spent
1280a7e14dcfSSatish Balay   in the line search to the running total.
1281a7e14dcfSSatish Balay 
1282a7e14dcfSSatish Balay   Input Parameters:
12832fe279fdSBarry Smith . tao - the `Tao` solver
1284a7e14dcfSSatish Balay 
1285a7e14dcfSSatish Balay   Level: developer
1286a7e14dcfSSatish Balay 
12871cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoGetLineSearch()`, `TaoLineSearchApply()`
1288a7e14dcfSSatish Balay @*/
1289d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoAddLineSearchCounts(Tao tao)
1290d71ae5a4SJacob Faibussowitsch {
1291a7e14dcfSSatish Balay   PetscBool flg;
1292a7e14dcfSSatish Balay   PetscInt  nfeval, ngeval, nfgeval;
129347a47007SBarry Smith 
1294a7e14dcfSSatish Balay   PetscFunctionBegin;
1295441846f8SBarry Smith   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
1296a7e14dcfSSatish Balay   if (tao->linesearch) {
12979566063dSJacob Faibussowitsch     PetscCall(TaoLineSearchIsUsingTaoRoutines(tao->linesearch, &flg));
129894ae4db5SBarry Smith     if (!flg) {
12999566063dSJacob Faibussowitsch       PetscCall(TaoLineSearchGetNumberFunctionEvaluations(tao->linesearch, &nfeval, &ngeval, &nfgeval));
1300a7e14dcfSSatish Balay       tao->nfuncs += nfeval;
1301a7e14dcfSSatish Balay       tao->ngrads += ngeval;
1302a7e14dcfSSatish Balay       tao->nfuncgrads += nfgeval;
1303a7e14dcfSSatish Balay     }
1304a7e14dcfSSatish Balay   }
13053ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
1306a7e14dcfSSatish Balay }
1307a7e14dcfSSatish Balay 
1308a7e14dcfSSatish Balay /*@
130947450a7bSBarry Smith   TaoGetSolution - Returns the vector with the current solution from the `Tao` object
1310a7e14dcfSSatish Balay 
1311a7e14dcfSSatish Balay   Not Collective
1312a7e14dcfSSatish Balay 
1313a7e14dcfSSatish Balay   Input Parameter:
131447450a7bSBarry Smith . tao - the `Tao` context
1315a7e14dcfSSatish Balay 
1316a7e14dcfSSatish Balay   Output Parameter:
1317a7e14dcfSSatish Balay . X - the current solution
1318a7e14dcfSSatish Balay 
1319a7e14dcfSSatish Balay   Level: intermediate
1320a7e14dcfSSatish Balay 
132165ba42b6SBarry Smith   Note:
132265ba42b6SBarry Smith   The returned vector will be the same object that was passed into `TaoSetSolution()`
132365ba42b6SBarry Smith 
13241cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoSetSolution()`, `TaoSolve()`
1325a7e14dcfSSatish Balay @*/
1326d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoGetSolution(Tao tao, Vec *X)
1327d71ae5a4SJacob Faibussowitsch {
1328a7e14dcfSSatish Balay   PetscFunctionBegin;
1329441846f8SBarry Smith   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
13304f572ea9SToby Isaac   PetscAssertPointer(X, 2);
1331a7e14dcfSSatish Balay   *X = tao->solution;
13323ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
1333a7e14dcfSSatish Balay }
1334a7e14dcfSSatish Balay 
1335a7e14dcfSSatish Balay /*@
133647450a7bSBarry Smith   TaoResetStatistics - Initialize the statistics collected by the `Tao` object.
1337a7e14dcfSSatish Balay   These statistics include the iteration number, residual norms, and convergence status.
1338a7e14dcfSSatish Balay   This routine gets called before solving each optimization problem.
1339a7e14dcfSSatish Balay 
1340c3339decSBarry Smith   Collective
1341a7e14dcfSSatish Balay 
134247450a7bSBarry Smith   Input Parameter:
1343e056e8ceSJacob Faibussowitsch . tao - the `Tao` context
1344a7e14dcfSSatish Balay 
1345a7e14dcfSSatish Balay   Level: developer
1346a7e14dcfSSatish Balay 
13471cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoCreate()`, `TaoSolve()`
1348a7e14dcfSSatish Balay @*/
1349d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoResetStatistics(Tao tao)
1350d71ae5a4SJacob Faibussowitsch {
1351a7e14dcfSSatish Balay   PetscFunctionBegin;
1352441846f8SBarry Smith   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
1353a7e14dcfSSatish Balay   tao->niter        = 0;
1354a7e14dcfSSatish Balay   tao->nfuncs       = 0;
1355a7e14dcfSSatish Balay   tao->nfuncgrads   = 0;
1356a7e14dcfSSatish Balay   tao->ngrads       = 0;
1357a7e14dcfSSatish Balay   tao->nhess        = 0;
1358a7e14dcfSSatish Balay   tao->njac         = 0;
1359a7e14dcfSSatish Balay   tao->nconstraints = 0;
1360a7e14dcfSSatish Balay   tao->ksp_its      = 0;
1361ae93cb3cSJason Sarich   tao->ksp_tot_its  = 0;
1362a7e14dcfSSatish Balay   tao->reason       = TAO_CONTINUE_ITERATING;
1363a7e14dcfSSatish Balay   tao->residual     = 0.0;
1364a7e14dcfSSatish Balay   tao->cnorm        = 0.0;
1365a7e14dcfSSatish Balay   tao->step         = 0.0;
1366a7e14dcfSSatish Balay   tao->lsflag       = PETSC_FALSE;
1367a7e14dcfSSatish Balay   if (tao->hist_reset) tao->hist_len = 0;
13683ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
1369a7e14dcfSSatish Balay }
1370a7e14dcfSSatish Balay 
1371a7e14dcfSSatish Balay /*@C
1372e1e80dc8SAlp Dener   TaoSetUpdate - Sets the general-purpose update function called
13732fe279fdSBarry Smith   at the beginning of every iteration of the optimization algorithm. Called after the new solution and the gradient
1374e1e80dc8SAlp Dener   is determined, but before the Hessian is computed (if applicable).
1375e1e80dc8SAlp Dener 
1376c3339decSBarry Smith   Logically Collective
1377e1e80dc8SAlp Dener 
1378e1e80dc8SAlp Dener   Input Parameters:
13792fe279fdSBarry Smith + tao  - The `Tao` solver context
1380a2b725a8SWilliam Gropp - func - The function
1381e1e80dc8SAlp Dener 
138220f4b53cSBarry Smith   Calling sequence of `func`:
138320f4b53cSBarry Smith + tao - the optimizer context
1384e056e8ceSJacob Faibussowitsch - ctx - The current step of the iteration
1385e1e80dc8SAlp Dener 
1386e1e80dc8SAlp Dener   Level: advanced
1387e1e80dc8SAlp Dener 
13881cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoSolve()`
1389e1e80dc8SAlp Dener @*/
1390d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoSetUpdate(Tao tao, PetscErrorCode (*func)(Tao, PetscInt, void *), void *ctx)
1391d71ae5a4SJacob Faibussowitsch {
1392e1e80dc8SAlp Dener   PetscFunctionBegin;
1393e1e80dc8SAlp Dener   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
1394e1e80dc8SAlp Dener   tao->ops->update = func;
1395e1e80dc8SAlp Dener   tao->user_update = ctx;
13963ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
1397e1e80dc8SAlp Dener }
1398e1e80dc8SAlp Dener 
1399e1e80dc8SAlp Dener /*@C
1400a7e14dcfSSatish Balay   TaoSetConvergenceTest - Sets the function that is to be used to test
1401a7e14dcfSSatish Balay   for convergence o fthe iterative minimization solution.  The new convergence
140265ba42b6SBarry Smith   testing routine will replace Tao's default convergence test.
1403a7e14dcfSSatish Balay 
1404c3339decSBarry Smith   Logically Collective
1405a7e14dcfSSatish Balay 
1406a7e14dcfSSatish Balay   Input Parameters:
140747450a7bSBarry Smith + tao  - the `Tao` object
1408a7e14dcfSSatish Balay . conv - the routine to test for convergence
1409a7e14dcfSSatish Balay - ctx  - [optional] context for private data for the convergence routine
141047450a7bSBarry Smith         (may be `NULL`)
1411a7e14dcfSSatish Balay 
141220f4b53cSBarry Smith   Calling sequence of `conv`:
141347450a7bSBarry Smith + tao - the `Tao` object
1414a7e14dcfSSatish Balay - ctx - [optional] convergence context
1415a7e14dcfSSatish Balay 
141647450a7bSBarry Smith   Level: advanced
141747450a7bSBarry Smith 
141865ba42b6SBarry Smith   Note:
141965ba42b6SBarry Smith   The new convergence testing routine should call `TaoSetConvergedReason()`.
1420a7e14dcfSSatish Balay 
14211cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoSolve()`, `TaoSetConvergedReason()`, `TaoGetSolutionStatus()`, `TaoGetTolerances()`, `TaoSetMonitor`
1422a7e14dcfSSatish Balay @*/
1423d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoSetConvergenceTest(Tao tao, PetscErrorCode (*conv)(Tao, void *), void *ctx)
1424d71ae5a4SJacob Faibussowitsch {
1425a7e14dcfSSatish Balay   PetscFunctionBegin;
1426441846f8SBarry Smith   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
142794511df7SStefano Zampini   tao->ops->convergencetest = conv;
142894511df7SStefano Zampini   tao->cnvP                 = ctx;
14293ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
1430a7e14dcfSSatish Balay }
1431a7e14dcfSSatish Balay 
1432a7e14dcfSSatish Balay /*@C
143365ba42b6SBarry Smith   TaoSetMonitor - Sets an additional function that is to be used at every
1434a7e14dcfSSatish Balay   iteration of the solver to display the iteration's
1435a7e14dcfSSatish Balay   progress.
1436a7e14dcfSSatish Balay 
1437c3339decSBarry Smith   Logically Collective
1438a7e14dcfSSatish Balay 
1439a7e14dcfSSatish Balay   Input Parameters:
144047450a7bSBarry Smith + tao  - the `Tao` solver context
14412fe279fdSBarry Smith . func - monitoring routine
14422fe279fdSBarry Smith . ctx  - [optional] user-defined context for private data for the monitor routine (may be `NULL`)
14432fe279fdSBarry Smith - dest - [optional] function to destroy the context when the `Tao` is destroyed
1444a7e14dcfSSatish Balay 
14452fe279fdSBarry Smith   Calling sequence of `func`:
144647450a7bSBarry Smith + tao - the `Tao` solver context
14472fe279fdSBarry Smith - ctx - [optional] monitoring context
14482fe279fdSBarry Smith 
14492fe279fdSBarry Smith   Calling sequence of `dest`:
14502fe279fdSBarry Smith . ctx - monitoring context
1451a7e14dcfSSatish Balay 
1452a7e14dcfSSatish Balay   Options Database Keys:
145365ba42b6SBarry Smith + -tao_monitor          - sets the default monitor `TaoMonitorDefault()`
1454a7e14dcfSSatish Balay . -tao_smonitor         - sets short monitor
1455a7e14dcfSSatish Balay . -tao_cmonitor         - same as smonitor plus constraint norm
1456a7e14dcfSSatish Balay . -tao_view_solution    - view solution at each iteration
1457a7e14dcfSSatish Balay . -tao_view_gradient    - view gradient at each iteration
14584a48860cSAlp Dener . -tao_view_ls_residual - view least-squares residual vector at each iteration
1459*b2dc45ddSPierre Jolivet - -tao_monitor_cancel   - cancels all monitors that have been hardwired into a code by calls to TaoSetMonitor(), but does not cancel those set via the options database.
1460a7e14dcfSSatish Balay 
146147450a7bSBarry Smith   Level: intermediate
146247450a7bSBarry Smith 
146387497f52SBarry Smith   Notes:
1464a7e14dcfSSatish Balay   Several different monitoring routines may be set by calling
146565ba42b6SBarry Smith   `TaoSetMonitor()` multiple times; all will be called in the
1466a7e14dcfSSatish Balay   order in which they were set.
1467a7e14dcfSSatish Balay 
1468e056e8ceSJacob Faibussowitsch   Fortran Notes:
146995452b02SPatrick Sanan   Only one monitor function may be set
1470a7e14dcfSSatish Balay 
1471*b2dc45ddSPierre Jolivet .seealso: [](ch_tao), `Tao`, `TaoSolve()`, `TaoMonitorDefault()`, `TaoMonitorCancel()`, `TaoSetDestroyRoutine()`, `TaoView()`
1472a7e14dcfSSatish Balay @*/
1473d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoSetMonitor(Tao tao, PetscErrorCode (*func)(Tao, void *), void *ctx, PetscErrorCode (*dest)(void **))
1474d71ae5a4SJacob Faibussowitsch {
147508d19d1fSJason Sarich   PetscInt  i;
14768732526dSAlp Dener   PetscBool identical;
147708d19d1fSJason Sarich 
1478a7e14dcfSSatish Balay   PetscFunctionBegin;
1479441846f8SBarry Smith   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
14803c859ba3SBarry Smith   PetscCheck(tao->numbermonitors < MAXTAOMONITORS, PetscObjectComm((PetscObject)tao), PETSC_ERR_SUP, "Cannot attach another monitor -- max=%d", MAXTAOMONITORS);
148108d19d1fSJason Sarich 
148208d19d1fSJason Sarich   for (i = 0; i < tao->numbermonitors; i++) {
14839566063dSJacob Faibussowitsch     PetscCall(PetscMonitorCompare((PetscErrorCode(*)(void))func, ctx, dest, (PetscErrorCode(*)(void))tao->monitor[i], tao->monitorcontext[i], tao->monitordestroy[i], &identical));
14843ba16761SJacob Faibussowitsch     if (identical) PetscFunctionReturn(PETSC_SUCCESS);
148508d19d1fSJason Sarich   }
1486a7e14dcfSSatish Balay   tao->monitor[tao->numbermonitors]        = func;
14878732526dSAlp Dener   tao->monitorcontext[tao->numbermonitors] = (void *)ctx;
1488a7e14dcfSSatish Balay   tao->monitordestroy[tao->numbermonitors] = dest;
1489a7e14dcfSSatish Balay   ++tao->numbermonitors;
14903ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
1491a7e14dcfSSatish Balay }
1492a7e14dcfSSatish Balay 
1493a7e14dcfSSatish Balay /*@
1494*b2dc45ddSPierre Jolivet   TaoMonitorCancel - Clears all the monitor functions for a `Tao` object.
1495a7e14dcfSSatish Balay 
1496c3339decSBarry Smith   Logically Collective
1497a7e14dcfSSatish Balay 
149847450a7bSBarry Smith   Input Parameter:
149947450a7bSBarry Smith . tao - the `Tao` solver context
1500a7e14dcfSSatish Balay 
15013c7db156SBarry Smith   Options Database Key:
1502*b2dc45ddSPierre Jolivet . -tao_monitor_cancel - cancels all monitors that have been hardwired
150365ba42b6SBarry Smith     into a code by calls to `TaoSetMonitor()`, but does not cancel those
1504a7e14dcfSSatish Balay     set via the options database
1505a7e14dcfSSatish Balay 
1506a7e14dcfSSatish Balay   Level: advanced
1507a7e14dcfSSatish Balay 
150847450a7bSBarry Smith   Note:
150947450a7bSBarry Smith   There is no way to clear one specific monitor from a `Tao` object.
151047450a7bSBarry Smith 
15111cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoMonitorDefault()`, `TaoSetMonitor()`
1512a7e14dcfSSatish Balay @*/
1513*b2dc45ddSPierre Jolivet PetscErrorCode TaoMonitorCancel(Tao tao)
1514d71ae5a4SJacob Faibussowitsch {
1515a7e14dcfSSatish Balay   PetscInt i;
151647a47007SBarry Smith 
1517a7e14dcfSSatish Balay   PetscFunctionBegin;
1518441846f8SBarry Smith   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
1519a7e14dcfSSatish Balay   for (i = 0; i < tao->numbermonitors; i++) {
152048a46eb9SPierre Jolivet     if (tao->monitordestroy[i]) PetscCall((*tao->monitordestroy[i])(&tao->monitorcontext[i]));
1521a7e14dcfSSatish Balay   }
1522a7e14dcfSSatish Balay   tao->numbermonitors = 0;
15233ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
1524a7e14dcfSSatish Balay }
1525a7e14dcfSSatish Balay 
15267fab98ebSJason Sarich /*@
152747450a7bSBarry Smith   TaoMonitorDefault - Default routine for monitoring progress of `TaoSolve()`
1528a7e14dcfSSatish Balay 
1529c3339decSBarry Smith   Collective
1530a7e14dcfSSatish Balay 
1531a7e14dcfSSatish Balay   Input Parameters:
153247450a7bSBarry Smith + tao - the `Tao` context
153347450a7bSBarry Smith - ctx - `PetscViewer` context or `NULL`
1534a7e14dcfSSatish Balay 
153547450a7bSBarry Smith   Options Database Key:
1536147403d9SBarry Smith . -tao_monitor - turn on default monitoring
1537a7e14dcfSSatish Balay 
1538a7e14dcfSSatish Balay   Level: advanced
1539a7e14dcfSSatish Balay 
154047450a7bSBarry Smith   Note:
154147450a7bSBarry Smith   This monitor prints the function value and gradient
154247450a7bSBarry Smith   norm at each iteration.
154347450a7bSBarry Smith 
15441cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoDefaultSMonitor()`, `TaoSetMonitor()`
1545a7e14dcfSSatish Balay @*/
1546d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoMonitorDefault(Tao tao, void *ctx)
1547d71ae5a4SJacob Faibussowitsch {
154863b15415SAlp Dener   PetscInt    its, tabs;
1549a7e14dcfSSatish Balay   PetscReal   fct, gnorm;
15508163d661SBarry Smith   PetscViewer viewer = (PetscViewer)ctx;
155147a47007SBarry Smith 
1552a7e14dcfSSatish Balay   PetscFunctionBegin;
155394511df7SStefano Zampini   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
15548163d661SBarry Smith   PetscValidHeaderSpecific(viewer, PETSC_VIEWER_CLASSID, 2);
1555a7e14dcfSSatish Balay   its   = tao->niter;
1556a7e14dcfSSatish Balay   fct   = tao->fc;
1557a7e14dcfSSatish Balay   gnorm = tao->residual;
15589566063dSJacob Faibussowitsch   PetscCall(PetscViewerASCIIGetTab(viewer, &tabs));
15599566063dSJacob Faibussowitsch   PetscCall(PetscViewerASCIISetTab(viewer, ((PetscObject)tao)->tablevel));
1560494bef23SAlp Dener   if (its == 0 && ((PetscObject)tao)->prefix && !tao->header_printed) {
15619566063dSJacob Faibussowitsch     PetscCall(PetscViewerASCIIPrintf(viewer, "  Iteration information for %s solve.\n", ((PetscObject)tao)->prefix));
1562494bef23SAlp Dener     tao->header_printed = PETSC_TRUE;
156363b15415SAlp Dener   }
156463a3b9bcSJacob Faibussowitsch   PetscCall(PetscViewerASCIIPrintf(viewer, "%3" PetscInt_FMT " TAO,", its));
15659566063dSJacob Faibussowitsch   PetscCall(PetscViewerASCIIPrintf(viewer, "  Function value: %g,", (double)fct));
156608d19d1fSJason Sarich   if (gnorm >= PETSC_INFINITY) {
15679566063dSJacob Faibussowitsch     PetscCall(PetscViewerASCIIPrintf(viewer, "  Residual: Inf \n"));
156808d19d1fSJason Sarich   } else {
15699566063dSJacob Faibussowitsch     PetscCall(PetscViewerASCIIPrintf(viewer, "  Residual: %g \n", (double)gnorm));
157008d19d1fSJason Sarich   }
15719566063dSJacob Faibussowitsch   PetscCall(PetscViewerASCIISetTab(viewer, tabs));
15723ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
1573a7e14dcfSSatish Balay }
1574a7e14dcfSSatish Balay 
15757fab98ebSJason Sarich /*@
157647450a7bSBarry Smith   TaoDefaultGMonitor - Default routine for monitoring progress of `TaoSolve()` with extra detail on the globalization method.
15778d5ead36SAlp Dener 
1578c3339decSBarry Smith   Collective
15798d5ead36SAlp Dener 
15808d5ead36SAlp Dener   Input Parameters:
158147450a7bSBarry Smith + tao - the `Tao` context
158247450a7bSBarry Smith - ctx - `PetscViewer` context or `NULL`
15838d5ead36SAlp Dener 
158447450a7bSBarry Smith   Options Database Key:
1585147403d9SBarry Smith . -tao_gmonitor - turn on monitoring with globalization information
15868d5ead36SAlp Dener 
15878d5ead36SAlp Dener   Level: advanced
15888d5ead36SAlp Dener 
158947450a7bSBarry Smith   Note:
159047450a7bSBarry Smith   This monitor prints the function value and gradient norm at each
159147450a7bSBarry Smith   iteration, as well as the step size and trust radius. Note that the
159247450a7bSBarry Smith   step size and trust radius may be the same for some algorithms.
159347450a7bSBarry Smith 
15941cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoDefaultSMonitor()`, `TaoSetMonitor()`
15958d5ead36SAlp Dener @*/
1596d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoDefaultGMonitor(Tao tao, void *ctx)
1597d71ae5a4SJacob Faibussowitsch {
15988d5ead36SAlp Dener   PetscInt    its, tabs;
15998d5ead36SAlp Dener   PetscReal   fct, gnorm, stp, tr;
16008d5ead36SAlp Dener   PetscViewer viewer = (PetscViewer)ctx;
16018d5ead36SAlp Dener 
16028d5ead36SAlp Dener   PetscFunctionBegin;
160394511df7SStefano Zampini   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
16048d5ead36SAlp Dener   PetscValidHeaderSpecific(viewer, PETSC_VIEWER_CLASSID, 2);
16058d5ead36SAlp Dener   its   = tao->niter;
16068d5ead36SAlp Dener   fct   = tao->fc;
16078d5ead36SAlp Dener   gnorm = tao->residual;
16088d5ead36SAlp Dener   stp   = tao->step;
16098d5ead36SAlp Dener   tr    = tao->trust;
16109566063dSJacob Faibussowitsch   PetscCall(PetscViewerASCIIGetTab(viewer, &tabs));
16119566063dSJacob Faibussowitsch   PetscCall(PetscViewerASCIISetTab(viewer, ((PetscObject)tao)->tablevel));
1612494bef23SAlp Dener   if (its == 0 && ((PetscObject)tao)->prefix && !tao->header_printed) {
16139566063dSJacob Faibussowitsch     PetscCall(PetscViewerASCIIPrintf(viewer, "  Iteration information for %s solve.\n", ((PetscObject)tao)->prefix));
1614494bef23SAlp Dener     tao->header_printed = PETSC_TRUE;
16158d5ead36SAlp Dener   }
161663a3b9bcSJacob Faibussowitsch   PetscCall(PetscViewerASCIIPrintf(viewer, "%3" PetscInt_FMT " TAO,", its));
16179566063dSJacob Faibussowitsch   PetscCall(PetscViewerASCIIPrintf(viewer, "  Function value: %g,", (double)fct));
16188d5ead36SAlp Dener   if (gnorm >= PETSC_INFINITY) {
16199566063dSJacob Faibussowitsch     PetscCall(PetscViewerASCIIPrintf(viewer, "  Residual: Inf,"));
16208d5ead36SAlp Dener   } else {
16219566063dSJacob Faibussowitsch     PetscCall(PetscViewerASCIIPrintf(viewer, "  Residual: %g,", (double)gnorm));
16228d5ead36SAlp Dener   }
16239566063dSJacob Faibussowitsch   PetscCall(PetscViewerASCIIPrintf(viewer, "  Step: %g,  Trust: %g\n", (double)stp, (double)tr));
16249566063dSJacob Faibussowitsch   PetscCall(PetscViewerASCIISetTab(viewer, tabs));
16253ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
16268d5ead36SAlp Dener }
16278d5ead36SAlp Dener 
16288d5ead36SAlp Dener /*@
162947450a7bSBarry Smith   TaoDefaultSMonitor - Default routine for monitoring progress of `TaoSolve()`
1630a7e14dcfSSatish Balay 
1631c3339decSBarry Smith   Collective
1632a7e14dcfSSatish Balay 
1633a7e14dcfSSatish Balay   Input Parameters:
163447450a7bSBarry Smith + tao - the `Tao` context
163547450a7bSBarry Smith - ctx - `PetscViewer` context of type `PETSCVIEWERASCII`
1636a7e14dcfSSatish Balay 
163747450a7bSBarry Smith   Options Database Key:
1638147403d9SBarry Smith . -tao_smonitor - turn on default short monitoring
1639a7e14dcfSSatish Balay 
1640a7e14dcfSSatish Balay   Level: advanced
1641a7e14dcfSSatish Balay 
164247450a7bSBarry Smith   Note:
164347450a7bSBarry Smith   Same as `TaoMonitorDefault()` except
164447450a7bSBarry Smith   it prints fewer digits of the residual as the residual gets smaller.
164547450a7bSBarry Smith   This is because the later digits are meaningless and are often
164647450a7bSBarry Smith   different on different machines; by using this routine different
164747450a7bSBarry Smith   machines will usually generate the same output.
164847450a7bSBarry Smith 
16491cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoMonitorDefault()`, `TaoSetMonitor()`
1650a7e14dcfSSatish Balay @*/
1651d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoDefaultSMonitor(Tao tao, void *ctx)
1652d71ae5a4SJacob Faibussowitsch {
16532c9b8b83SAlp Dener   PetscInt    its, tabs;
1654a7e14dcfSSatish Balay   PetscReal   fct, gnorm;
16554d4332d5SBarry Smith   PetscViewer viewer = (PetscViewer)ctx;
165647a47007SBarry Smith 
1657a7e14dcfSSatish Balay   PetscFunctionBegin;
165894511df7SStefano Zampini   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
16594d4332d5SBarry Smith   PetscValidHeaderSpecific(viewer, PETSC_VIEWER_CLASSID, 2);
1660a7e14dcfSSatish Balay   its   = tao->niter;
1661a7e14dcfSSatish Balay   fct   = tao->fc;
1662a7e14dcfSSatish Balay   gnorm = tao->residual;
16639566063dSJacob Faibussowitsch   PetscCall(PetscViewerASCIIGetTab(viewer, &tabs));
16649566063dSJacob Faibussowitsch   PetscCall(PetscViewerASCIISetTab(viewer, ((PetscObject)tao)->tablevel));
166563a3b9bcSJacob Faibussowitsch   PetscCall(PetscViewerASCIIPrintf(viewer, "iter = %3" PetscInt_FMT ",", its));
16669566063dSJacob Faibussowitsch   PetscCall(PetscViewerASCIIPrintf(viewer, " Function value %g,", (double)fct));
1667d393f493SBarry Smith   if (gnorm >= PETSC_INFINITY) {
16689566063dSJacob Faibussowitsch     PetscCall(PetscViewerASCIIPrintf(viewer, " Residual: Inf \n"));
166908d19d1fSJason Sarich   } else if (gnorm > 1.e-6) {
16709566063dSJacob Faibussowitsch     PetscCall(PetscViewerASCIIPrintf(viewer, " Residual: %g \n", (double)gnorm));
1671a7e14dcfSSatish Balay   } else if (gnorm > 1.e-11) {
16729566063dSJacob Faibussowitsch     PetscCall(PetscViewerASCIIPrintf(viewer, " Residual: < 1.0e-6 \n"));
1673a7e14dcfSSatish Balay   } else {
16749566063dSJacob Faibussowitsch     PetscCall(PetscViewerASCIIPrintf(viewer, " Residual: < 1.0e-11 \n"));
1675a7e14dcfSSatish Balay   }
16769566063dSJacob Faibussowitsch   PetscCall(PetscViewerASCIISetTab(viewer, tabs));
16773ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
1678a7e14dcfSSatish Balay }
1679a7e14dcfSSatish Balay 
16807fab98ebSJason Sarich /*@
168165ba42b6SBarry Smith   TaoDefaultCMonitor - same as `TaoMonitorDefault()` except
168247450a7bSBarry Smith   it prints the norm of the constraint function.
1683a7e14dcfSSatish Balay 
1684c3339decSBarry Smith   Collective
1685a7e14dcfSSatish Balay 
1686a7e14dcfSSatish Balay   Input Parameters:
168747450a7bSBarry Smith + tao - the `Tao` context
168847450a7bSBarry Smith - ctx - `PetscViewer` context or `NULL`
1689a7e14dcfSSatish Balay 
169047450a7bSBarry Smith   Options Database Key:
1691147403d9SBarry Smith . -tao_cmonitor - monitor the constraints
1692a7e14dcfSSatish Balay 
1693a7e14dcfSSatish Balay   Level: advanced
1694a7e14dcfSSatish Balay 
16951cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoMonitorDefault()`, `TaoSetMonitor()`
1696a7e14dcfSSatish Balay @*/
1697d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoDefaultCMonitor(Tao tao, void *ctx)
1698d71ae5a4SJacob Faibussowitsch {
16992c9b8b83SAlp Dener   PetscInt    its, tabs;
1700a7e14dcfSSatish Balay   PetscReal   fct, gnorm;
1701d393f493SBarry Smith   PetscViewer viewer = (PetscViewer)ctx;
1702a7e14dcfSSatish Balay 
1703a7e14dcfSSatish Balay   PetscFunctionBegin;
170494511df7SStefano Zampini   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
1705d393f493SBarry Smith   PetscValidHeaderSpecific(viewer, PETSC_VIEWER_CLASSID, 2);
1706a7e14dcfSSatish Balay   its   = tao->niter;
1707a7e14dcfSSatish Balay   fct   = tao->fc;
1708a7e14dcfSSatish Balay   gnorm = tao->residual;
17099566063dSJacob Faibussowitsch   PetscCall(PetscViewerASCIIGetTab(viewer, &tabs));
17109566063dSJacob Faibussowitsch   PetscCall(PetscViewerASCIISetTab(viewer, ((PetscObject)tao)->tablevel));
171163a3b9bcSJacob Faibussowitsch   PetscCall(PetscViewerASCIIPrintf(viewer, "iter = %" PetscInt_FMT ",", its));
17129566063dSJacob Faibussowitsch   PetscCall(PetscViewerASCIIPrintf(viewer, " Function value: %g,", (double)fct));
17139566063dSJacob Faibussowitsch   PetscCall(PetscViewerASCIIPrintf(viewer, "  Residual: %g ", (double)gnorm));
17149566063dSJacob Faibussowitsch   PetscCall(PetscViewerASCIIPrintf(viewer, "  Constraint: %g \n", (double)tao->cnorm));
17159566063dSJacob Faibussowitsch   PetscCall(PetscViewerASCIISetTab(viewer, tabs));
17163ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
1717a7e14dcfSSatish Balay }
1718a7e14dcfSSatish Balay 
1719a7e14dcfSSatish Balay /*@C
172047450a7bSBarry Smith   TaoSolutionMonitor - Views the solution at each iteration of `TaoSolve()`
1721a7e14dcfSSatish Balay 
1722c3339decSBarry Smith   Collective
1723a7e14dcfSSatish Balay 
1724a7e14dcfSSatish Balay   Input Parameters:
172547450a7bSBarry Smith + tao - the `Tao` context
172647450a7bSBarry Smith - ctx - `PetscViewer` context or `NULL`
1727a7e14dcfSSatish Balay 
172847450a7bSBarry Smith   Options Database Key:
1729147403d9SBarry Smith . -tao_view_solution - view the solution
1730a7e14dcfSSatish Balay 
1731a7e14dcfSSatish Balay   Level: advanced
1732a7e14dcfSSatish Balay 
17331cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoDefaultSMonitor()`, `TaoSetMonitor()`
1734a7e14dcfSSatish Balay @*/
1735d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoSolutionMonitor(Tao tao, void *ctx)
1736d71ae5a4SJacob Faibussowitsch {
1737feb237baSPierre Jolivet   PetscViewer viewer = (PetscViewer)ctx;
173847a47007SBarry Smith 
1739a7e14dcfSSatish Balay   PetscFunctionBegin;
174094511df7SStefano Zampini   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
1741d393f493SBarry Smith   PetscValidHeaderSpecific(viewer, PETSC_VIEWER_CLASSID, 2);
17429566063dSJacob Faibussowitsch   PetscCall(VecView(tao->solution, viewer));
17433ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
1744a7e14dcfSSatish Balay }
1745a7e14dcfSSatish Balay 
1746a7e14dcfSSatish Balay /*@C
174747450a7bSBarry Smith   TaoGradientMonitor - Views the gradient at each iteration of `TaoSolve()`
1748a7e14dcfSSatish Balay 
1749c3339decSBarry Smith   Collective
1750a7e14dcfSSatish Balay 
1751a7e14dcfSSatish Balay   Input Parameters:
175247450a7bSBarry Smith + tao - the `Tao` context
175347450a7bSBarry Smith - ctx - `PetscViewer` context or `NULL`
1754a7e14dcfSSatish Balay 
175547450a7bSBarry Smith   Options Database Key:
1756147403d9SBarry Smith . -tao_view_gradient - view the gradient at each iteration
1757a7e14dcfSSatish Balay 
1758a7e14dcfSSatish Balay   Level: advanced
1759a7e14dcfSSatish Balay 
17601cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoDefaultSMonitor()`, `TaoSetMonitor()`
1761a7e14dcfSSatish Balay @*/
1762d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoGradientMonitor(Tao tao, void *ctx)
1763d71ae5a4SJacob Faibussowitsch {
1764d393f493SBarry Smith   PetscViewer viewer = (PetscViewer)ctx;
176547a47007SBarry Smith 
1766a7e14dcfSSatish Balay   PetscFunctionBegin;
176794511df7SStefano Zampini   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
1768d393f493SBarry Smith   PetscValidHeaderSpecific(viewer, PETSC_VIEWER_CLASSID, 2);
17699566063dSJacob Faibussowitsch   PetscCall(VecView(tao->gradient, viewer));
17703ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
1771a7e14dcfSSatish Balay }
1772a7e14dcfSSatish Balay 
1773a7e14dcfSSatish Balay /*@C
177447450a7bSBarry Smith   TaoStepDirectionMonitor - Views the step-direction at each iteration of `TaoSolve()`
1775a7e14dcfSSatish Balay 
1776c3339decSBarry Smith   Collective
1777a7e14dcfSSatish Balay 
1778a7e14dcfSSatish Balay   Input Parameters:
177947450a7bSBarry Smith + tao - the `Tao` context
178047450a7bSBarry Smith - ctx - `PetscViewer` context or `NULL`
1781a7e14dcfSSatish Balay 
178247450a7bSBarry Smith   Options Database Key:
178347450a7bSBarry Smith . -tao_view_stepdirection - view the step direction vector at each iteration
1784a7e14dcfSSatish Balay 
1785a7e14dcfSSatish Balay   Level: advanced
1786a7e14dcfSSatish Balay 
17871cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoDefaultSMonitor()`, `TaoSetMonitor()`
1788a7e14dcfSSatish Balay @*/
1789d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoStepDirectionMonitor(Tao tao, void *ctx)
1790d71ae5a4SJacob Faibussowitsch {
1791d393f493SBarry Smith   PetscViewer viewer = (PetscViewer)ctx;
1792d393f493SBarry Smith 
1793a7e14dcfSSatish Balay   PetscFunctionBegin;
179494511df7SStefano Zampini   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
1795d393f493SBarry Smith   PetscValidHeaderSpecific(viewer, PETSC_VIEWER_CLASSID, 2);
17969566063dSJacob Faibussowitsch   PetscCall(VecView(tao->stepdirection, viewer));
17973ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
1798a7e14dcfSSatish Balay }
1799a7e14dcfSSatish Balay 
1800a7e14dcfSSatish Balay /*@C
180147450a7bSBarry Smith   TaoDrawSolutionMonitor - Plots the solution at each iteration of `TaoSolve()`
1802a7e14dcfSSatish Balay 
1803c3339decSBarry Smith   Collective
1804a7e14dcfSSatish Balay 
1805a7e14dcfSSatish Balay   Input Parameters:
180647450a7bSBarry Smith + tao - the `Tao` context
180765ba42b6SBarry Smith - ctx - `TaoMonitorDraw` context
1808a7e14dcfSSatish Balay 
180947450a7bSBarry Smith   Options Database Key:
1810147403d9SBarry Smith . -tao_draw_solution - draw the solution at each iteration
1811a7e14dcfSSatish Balay 
1812a7e14dcfSSatish Balay   Level: advanced
1813a7e14dcfSSatish Balay 
18141cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoSolutionMonitor()`, `TaoSetMonitor()`, `TaoDrawGradientMonitor`, `TaoMonitorDraw`
1815a7e14dcfSSatish Balay @*/
1816d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoDrawSolutionMonitor(Tao tao, void *ctx)
1817d71ae5a4SJacob Faibussowitsch {
1818e882e171SHong Zhang   TaoMonitorDrawCtx ictx = (TaoMonitorDrawCtx)ctx;
181947a47007SBarry Smith 
1820a7e14dcfSSatish Balay   PetscFunctionBegin;
182194511df7SStefano Zampini   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
18223ba16761SJacob Faibussowitsch   if (!(((ictx->howoften > 0) && (!(tao->niter % ictx->howoften))) || ((ictx->howoften == -1) && tao->reason))) PetscFunctionReturn(PETSC_SUCCESS);
18239566063dSJacob Faibussowitsch   PetscCall(VecView(tao->solution, ictx->viewer));
18243ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
1825a7e14dcfSSatish Balay }
1826a7e14dcfSSatish Balay 
1827a7e14dcfSSatish Balay /*@C
182847450a7bSBarry Smith   TaoDrawGradientMonitor - Plots the gradient at each iteration of `TaoSolve()`
1829a7e14dcfSSatish Balay 
1830c3339decSBarry Smith   Collective
1831a7e14dcfSSatish Balay 
1832a7e14dcfSSatish Balay   Input Parameters:
183347450a7bSBarry Smith + tao - the `Tao` context
183465ba42b6SBarry Smith - ctx - `PetscViewer` context
1835a7e14dcfSSatish Balay 
183647450a7bSBarry Smith   Options Database Key:
1837147403d9SBarry Smith . -tao_draw_gradient - draw the gradient at each iteration
1838a7e14dcfSSatish Balay 
1839a7e14dcfSSatish Balay   Level: advanced
1840a7e14dcfSSatish Balay 
18411cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoGradientMonitor()`, `TaoSetMonitor()`, `TaoDrawSolutionMonitor`
1842a7e14dcfSSatish Balay @*/
1843d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoDrawGradientMonitor(Tao tao, void *ctx)
1844d71ae5a4SJacob Faibussowitsch {
1845e882e171SHong Zhang   TaoMonitorDrawCtx ictx = (TaoMonitorDrawCtx)ctx;
184647a47007SBarry Smith 
1847a7e14dcfSSatish Balay   PetscFunctionBegin;
184894511df7SStefano Zampini   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
18493ba16761SJacob Faibussowitsch   if (!(((ictx->howoften > 0) && (!(tao->niter % ictx->howoften))) || ((ictx->howoften == -1) && tao->reason))) PetscFunctionReturn(PETSC_SUCCESS);
18509566063dSJacob Faibussowitsch   PetscCall(VecView(tao->gradient, ictx->viewer));
18513ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
1852a7e14dcfSSatish Balay }
1853a7e14dcfSSatish Balay 
1854a7e14dcfSSatish Balay /*@C
185547450a7bSBarry Smith   TaoDrawStepMonitor - Plots the step direction at each iteration of `TaoSolve()`
1856a7e14dcfSSatish Balay 
1857c3339decSBarry Smith   Collective
1858a7e14dcfSSatish Balay 
1859a7e14dcfSSatish Balay   Input Parameters:
186047450a7bSBarry Smith + tao - the `Tao` context
186147450a7bSBarry Smith - ctx - the `PetscViewer` context
1862a7e14dcfSSatish Balay 
186347450a7bSBarry Smith   Options Database Key:
1864147403d9SBarry Smith . -tao_draw_step - draw the step direction at each iteration
1865a7e14dcfSSatish Balay 
1866a7e14dcfSSatish Balay   Level: advanced
1867a7e14dcfSSatish Balay 
18681cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoSetMonitor()`, `TaoDrawSolutionMonitor`
1869a7e14dcfSSatish Balay @*/
1870d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoDrawStepMonitor(Tao tao, void *ctx)
1871d71ae5a4SJacob Faibussowitsch {
187294511df7SStefano Zampini   PetscViewer viewer = (PetscViewer)ctx;
187347a47007SBarry Smith 
1874a7e14dcfSSatish Balay   PetscFunctionBegin;
187594511df7SStefano Zampini   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
187694511df7SStefano Zampini   PetscValidHeaderSpecific(viewer, PETSC_VIEWER_CLASSID, 2);
18779566063dSJacob Faibussowitsch   PetscCall(VecView(tao->stepdirection, viewer));
18783ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
1879a7e14dcfSSatish Balay }
1880a7e14dcfSSatish Balay 
1881a7e14dcfSSatish Balay /*@C
188247450a7bSBarry Smith   TaoResidualMonitor - Views the least-squares residual at each iteration of `TaoSolve()`
1883a7e14dcfSSatish Balay 
1884c3339decSBarry Smith   Collective
1885a7e14dcfSSatish Balay 
1886a7e14dcfSSatish Balay   Input Parameters:
188747450a7bSBarry Smith + tao - the `Tao` context
188847450a7bSBarry Smith - ctx - the `PetscViewer` context or `NULL`
1889a7e14dcfSSatish Balay 
189047450a7bSBarry Smith   Options Database Key:
189147450a7bSBarry Smith . -tao_view_ls_residual - view the residual at each iteration
1892a7e14dcfSSatish Balay 
1893a7e14dcfSSatish Balay   Level: advanced
1894a7e14dcfSSatish Balay 
18951cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoDefaultSMonitor()`, `TaoSetMonitor()`
1896a7e14dcfSSatish Balay @*/
1897d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoResidualMonitor(Tao tao, void *ctx)
1898d71ae5a4SJacob Faibussowitsch {
1899d393f493SBarry Smith   PetscViewer viewer = (PetscViewer)ctx;
190047a47007SBarry Smith 
1901a7e14dcfSSatish Balay   PetscFunctionBegin;
190294511df7SStefano Zampini   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
1903d393f493SBarry Smith   PetscValidHeaderSpecific(viewer, PETSC_VIEWER_CLASSID, 2);
19049566063dSJacob Faibussowitsch   PetscCall(VecView(tao->ls_res, viewer));
19053ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
1906a7e14dcfSSatish Balay }
1907a7e14dcfSSatish Balay 
19087fab98ebSJason Sarich /*@
1909a7e14dcfSSatish Balay   TaoDefaultConvergenceTest - Determines whether the solver should continue iterating
1910a7e14dcfSSatish Balay   or terminate.
1911a7e14dcfSSatish Balay 
1912c3339decSBarry Smith   Collective
1913a7e14dcfSSatish Balay 
1914a7e14dcfSSatish Balay   Input Parameters:
191547450a7bSBarry Smith + tao   - the `Tao` context
1916a7e14dcfSSatish Balay - dummy - unused dummy context
1917a7e14dcfSSatish Balay 
191847450a7bSBarry Smith   Level: developer
191947450a7bSBarry Smith 
1920a7e14dcfSSatish Balay   Notes:
1921a7e14dcfSSatish Balay   This routine checks the residual in the optimality conditions, the
1922a7e14dcfSSatish Balay   relative residual in the optimity conditions, the number of function
1923a7e14dcfSSatish Balay   evaluations, and the function value to test convergence.  Some
1924a7e14dcfSSatish Balay   solvers may use different convergence routines.
1925a7e14dcfSSatish Balay 
19261cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoSetTolerances()`, `TaoGetConvergedReason()`, `TaoSetConvergedReason()`
1927a7e14dcfSSatish Balay @*/
1928d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoDefaultConvergenceTest(Tao tao, void *dummy)
1929d71ae5a4SJacob Faibussowitsch {
1930a7e14dcfSSatish Balay   PetscInt           niter = tao->niter, nfuncs = PetscMax(tao->nfuncs, tao->nfuncgrads);
1931a7e14dcfSSatish Balay   PetscInt           max_funcs = tao->max_funcs;
1932a7e14dcfSSatish Balay   PetscReal          gnorm = tao->residual, gnorm0 = tao->gnorm0;
1933a7e14dcfSSatish Balay   PetscReal          f = tao->fc, steptol = tao->steptol, trradius = tao->step;
1934a7e14dcfSSatish Balay   PetscReal          gatol = tao->gatol, grtol = tao->grtol, gttol = tao->gttol;
1935e52336cbSBarry Smith   PetscReal          catol = tao->catol, crtol = tao->crtol;
1936e52336cbSBarry Smith   PetscReal          fmin = tao->fmin, cnorm = tao->cnorm;
1937e4cb33bbSBarry Smith   TaoConvergedReason reason = tao->reason;
1938a7e14dcfSSatish Balay 
1939a7e14dcfSSatish Balay   PetscFunctionBegin;
1940441846f8SBarry Smith   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
19413ba16761SJacob Faibussowitsch   if (reason != TAO_CONTINUE_ITERATING) PetscFunctionReturn(PETSC_SUCCESS);
1942a7e14dcfSSatish Balay 
1943a7e14dcfSSatish Balay   if (PetscIsInfOrNanReal(f)) {
19449566063dSJacob Faibussowitsch     PetscCall(PetscInfo(tao, "Failed to converged, function value is Inf or NaN\n"));
1945a7e14dcfSSatish Balay     reason = TAO_DIVERGED_NAN;
1946a7e14dcfSSatish Balay   } else if (f <= fmin && cnorm <= catol) {
19479566063dSJacob Faibussowitsch     PetscCall(PetscInfo(tao, "Converged due to function value %g < minimum function value %g\n", (double)f, (double)fmin));
1948a7e14dcfSSatish Balay     reason = TAO_CONVERGED_MINF;
1949a7e14dcfSSatish Balay   } else if (gnorm <= gatol && cnorm <= catol) {
19509566063dSJacob Faibussowitsch     PetscCall(PetscInfo(tao, "Converged due to residual norm ||g(X)||=%g < %g\n", (double)gnorm, (double)gatol));
1951a7e14dcfSSatish Balay     reason = TAO_CONVERGED_GATOL;
1952a7e14dcfSSatish Balay   } else if (f != 0 && PetscAbsReal(gnorm / f) <= grtol && cnorm <= crtol) {
19539566063dSJacob Faibussowitsch     PetscCall(PetscInfo(tao, "Converged due to residual ||g(X)||/|f(X)| =%g < %g\n", (double)(gnorm / f), (double)grtol));
1954a7e14dcfSSatish Balay     reason = TAO_CONVERGED_GRTOL;
1955e1d16bb9SBarry Smith   } else if (gnorm0 != 0 && ((gttol == 0 && gnorm == 0) || gnorm / gnorm0 < gttol) && cnorm <= crtol) {
19569566063dSJacob Faibussowitsch     PetscCall(PetscInfo(tao, "Converged due to relative residual norm ||g(X)||/||g(X0)|| = %g < %g\n", (double)(gnorm / gnorm0), (double)gttol));
1957a7e14dcfSSatish Balay     reason = TAO_CONVERGED_GTTOL;
195894511df7SStefano Zampini   } else if (max_funcs >= 0 && nfuncs > max_funcs) {
19599566063dSJacob Faibussowitsch     PetscCall(PetscInfo(tao, "Exceeded maximum number of function evaluations: %" PetscInt_FMT " > %" PetscInt_FMT "\n", nfuncs, max_funcs));
1960a7e14dcfSSatish Balay     reason = TAO_DIVERGED_MAXFCN;
1961a7e14dcfSSatish Balay   } else if (tao->lsflag != 0) {
19629566063dSJacob Faibussowitsch     PetscCall(PetscInfo(tao, "Tao Line Search failure.\n"));
1963a7e14dcfSSatish Balay     reason = TAO_DIVERGED_LS_FAILURE;
1964a7e14dcfSSatish Balay   } else if (trradius < steptol && niter > 0) {
19659566063dSJacob Faibussowitsch     PetscCall(PetscInfo(tao, "Trust region/step size too small: %g < %g\n", (double)trradius, (double)steptol));
1966a7e14dcfSSatish Balay     reason = TAO_CONVERGED_STEPTOL;
1967e031d6f5SAlp Dener   } else if (niter >= tao->max_it) {
196863a3b9bcSJacob Faibussowitsch     PetscCall(PetscInfo(tao, "Exceeded maximum number of iterations: %" PetscInt_FMT " > %" PetscInt_FMT "\n", niter, tao->max_it));
1969a7e14dcfSSatish Balay     reason = TAO_DIVERGED_MAXITS;
1970a7e14dcfSSatish Balay   } else {
1971a7e14dcfSSatish Balay     reason = TAO_CONTINUE_ITERATING;
1972a7e14dcfSSatish Balay   }
1973a7e14dcfSSatish Balay   tao->reason = reason;
19743ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
1975a7e14dcfSSatish Balay }
1976a7e14dcfSSatish Balay 
1977a7e14dcfSSatish Balay /*@C
1978a7e14dcfSSatish Balay   TaoSetOptionsPrefix - Sets the prefix used for searching for all
197965ba42b6SBarry Smith   Tao options in the database.
1980a7e14dcfSSatish Balay 
1981c3339decSBarry Smith   Logically Collective
1982a7e14dcfSSatish Balay 
1983a7e14dcfSSatish Balay   Input Parameters:
198447450a7bSBarry Smith + tao - the `Tao` context
1985e056e8ceSJacob Faibussowitsch - p   - the prefix string to prepend to all Tao option requests
1986a7e14dcfSSatish Balay 
19872fe279fdSBarry Smith   Level: advanced
19882fe279fdSBarry Smith 
1989a7e14dcfSSatish Balay   Notes:
1990a7e14dcfSSatish Balay   A hyphen (-) must NOT be given at the beginning of the prefix name.
1991a7e14dcfSSatish Balay   The first character of all runtime options is AUTOMATICALLY the hyphen.
1992a7e14dcfSSatish Balay 
1993a7e14dcfSSatish Balay   For example, to distinguish between the runtime options for two
199465ba42b6SBarry Smith   different Tao solvers, one could call
1995a7e14dcfSSatish Balay .vb
1996a7e14dcfSSatish Balay       TaoSetOptionsPrefix(tao1,"sys1_")
1997a7e14dcfSSatish Balay       TaoSetOptionsPrefix(tao2,"sys2_")
1998a7e14dcfSSatish Balay .ve
1999a7e14dcfSSatish Balay 
2000a7e14dcfSSatish Balay   This would enable use of different options for each system, such as
2001a7e14dcfSSatish Balay .vb
20029fa2b5dcSStefano Zampini       -sys1_tao_method blmvm -sys1_tao_grtol 1.e-3
20039fa2b5dcSStefano Zampini       -sys2_tao_method lmvm  -sys2_tao_grtol 1.e-4
2004a7e14dcfSSatish Balay .ve
2005a7e14dcfSSatish Balay 
20061cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoSetFromOptions()`, `TaoAppendOptionsPrefix()`, `TaoGetOptionsPrefix()`
2007a7e14dcfSSatish Balay @*/
2008d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoSetOptionsPrefix(Tao tao, const char p[])
2009d71ae5a4SJacob Faibussowitsch {
2010a7e14dcfSSatish Balay   PetscFunctionBegin;
201194511df7SStefano Zampini   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
20129566063dSJacob Faibussowitsch   PetscCall(PetscObjectSetOptionsPrefix((PetscObject)tao, p));
20131baa6e33SBarry Smith   if (tao->linesearch) PetscCall(TaoLineSearchSetOptionsPrefix(tao->linesearch, p));
20141baa6e33SBarry Smith   if (tao->ksp) PetscCall(KSPSetOptionsPrefix(tao->ksp, p));
20153ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
2016a7e14dcfSSatish Balay }
2017a7e14dcfSSatish Balay 
2018a7e14dcfSSatish Balay /*@C
201947450a7bSBarry Smith   TaoAppendOptionsPrefix - Appends to the prefix used for searching for all Tao options in the database.
2020a7e14dcfSSatish Balay 
2021c3339decSBarry Smith   Logically Collective
2022a7e14dcfSSatish Balay 
2023a7e14dcfSSatish Balay   Input Parameters:
202447450a7bSBarry Smith + tao - the `Tao` solver context
2025e056e8ceSJacob Faibussowitsch - p   - the prefix string to prepend to all `Tao` option requests
2026a7e14dcfSSatish Balay 
20272fe279fdSBarry Smith   Level: advanced
20282fe279fdSBarry Smith 
202965ba42b6SBarry Smith   Note:
2030a7e14dcfSSatish Balay   A hyphen (-) must NOT be given at the beginning of the prefix name.
203165ba42b6SBarry Smith   The first character of all runtime options is automatically the hyphen.
2032a7e14dcfSSatish Balay 
20331cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoSetFromOptions()`, `TaoSetOptionsPrefix()`, `TaoGetOptionsPrefix()`
2034a7e14dcfSSatish Balay @*/
2035d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoAppendOptionsPrefix(Tao tao, const char p[])
2036d71ae5a4SJacob Faibussowitsch {
2037a7e14dcfSSatish Balay   PetscFunctionBegin;
203894511df7SStefano Zampini   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
20399566063dSJacob Faibussowitsch   PetscCall(PetscObjectAppendOptionsPrefix((PetscObject)tao, p));
20401baa6e33SBarry Smith   if (tao->linesearch) PetscCall(PetscObjectAppendOptionsPrefix((PetscObject)tao->linesearch, p));
20411baa6e33SBarry Smith   if (tao->ksp) PetscCall(KSPAppendOptionsPrefix(tao->ksp, p));
20423ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
2043a7e14dcfSSatish Balay }
2044a7e14dcfSSatish Balay 
2045a7e14dcfSSatish Balay /*@C
2046a7e14dcfSSatish Balay   TaoGetOptionsPrefix - Gets the prefix used for searching for all
204765ba42b6SBarry Smith   Tao options in the database
2048a7e14dcfSSatish Balay 
2049a7e14dcfSSatish Balay   Not Collective
2050a7e14dcfSSatish Balay 
205147450a7bSBarry Smith   Input Parameter:
205247450a7bSBarry Smith . tao - the `Tao` context
2053a7e14dcfSSatish Balay 
205447450a7bSBarry Smith   Output Parameter:
2055e056e8ceSJacob Faibussowitsch . p - pointer to the prefix string used is returned
2056a7e14dcfSSatish Balay 
2057e056e8ceSJacob Faibussowitsch   Fortran Notes:
205847450a7bSBarry Smith   Pass in a string 'prefix' of sufficient length to hold the prefix.
2059a7e14dcfSSatish Balay 
2060a7e14dcfSSatish Balay   Level: advanced
2061a7e14dcfSSatish Balay 
20621cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoSetFromOptions()`, `TaoSetOptionsPrefix()`, `TaoAppendOptionsPrefix()`
2063a7e14dcfSSatish Balay @*/
2064d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoGetOptionsPrefix(Tao tao, const char *p[])
2065d71ae5a4SJacob Faibussowitsch {
206694511df7SStefano Zampini   PetscFunctionBegin;
206794511df7SStefano Zampini   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
20689566063dSJacob Faibussowitsch   PetscCall(PetscObjectGetOptionsPrefix((PetscObject)tao, p));
20693ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
2070a7e14dcfSSatish Balay }
2071a7e14dcfSSatish Balay 
2072a7e14dcfSSatish Balay /*@C
207347450a7bSBarry Smith   TaoSetType - Sets the `TaoType` for the minimization solver.
2074a7e14dcfSSatish Balay 
2075c3339decSBarry Smith   Collective
2076a7e14dcfSSatish Balay 
2077a7e14dcfSSatish Balay   Input Parameters:
2078e056e8ceSJacob Faibussowitsch + tao  - the `Tao` solver context
2079a7e14dcfSSatish Balay - type - a known method
2080a7e14dcfSSatish Balay 
2081a7e14dcfSSatish Balay   Options Database Key:
208258417fe7SBarry Smith . -tao_type <type> - Sets the method; use -help for a list
208358417fe7SBarry Smith    of available methods (for instance, "-tao_type lmvm" or "-tao_type tron")
2084a7e14dcfSSatish Balay 
2085a7e14dcfSSatish Balay   Level: intermediate
2086a7e14dcfSSatish Balay 
20871cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoCreate()`, `TaoGetType()`, `TaoType`
2088a7e14dcfSSatish Balay @*/
2089d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoSetType(Tao tao, TaoType type)
2090d71ae5a4SJacob Faibussowitsch {
2091441846f8SBarry Smith   PetscErrorCode (*create_xxx)(Tao);
2092a7e14dcfSSatish Balay   PetscBool issame;
2093a7e14dcfSSatish Balay 
2094a7e14dcfSSatish Balay   PetscFunctionBegin;
2095441846f8SBarry Smith   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
2096a7e14dcfSSatish Balay 
20979566063dSJacob Faibussowitsch   PetscCall(PetscObjectTypeCompare((PetscObject)tao, type, &issame));
20983ba16761SJacob Faibussowitsch   if (issame) PetscFunctionReturn(PETSC_SUCCESS);
2099a7e14dcfSSatish Balay 
21009566063dSJacob Faibussowitsch   PetscCall(PetscFunctionListFind(TaoList, type, (void (**)(void)) & create_xxx));
21013c859ba3SBarry Smith   PetscCheck(create_xxx, PetscObjectComm((PetscObject)tao), PETSC_ERR_ARG_UNKNOWN_TYPE, "Unable to find requested Tao type %s", type);
2102a7e14dcfSSatish Balay 
2103a7e14dcfSSatish Balay   /* Destroy the existing solver information */
2104dbbe0bcdSBarry Smith   PetscTryTypeMethod(tao, destroy);
21051baa6e33SBarry Smith   PetscCall(KSPDestroy(&tao->ksp));
21069566063dSJacob Faibussowitsch   PetscCall(TaoLineSearchDestroy(&tao->linesearch));
210783c8fe1dSLisandro Dalcin   tao->ops->setup          = NULL;
210883c8fe1dSLisandro Dalcin   tao->ops->solve          = NULL;
210983c8fe1dSLisandro Dalcin   tao->ops->view           = NULL;
211083c8fe1dSLisandro Dalcin   tao->ops->setfromoptions = NULL;
211183c8fe1dSLisandro Dalcin   tao->ops->destroy        = NULL;
2112a7e14dcfSSatish Balay 
2113a7e14dcfSSatish Balay   tao->setupcalled = PETSC_FALSE;
2114a7e14dcfSSatish Balay 
21159566063dSJacob Faibussowitsch   PetscCall((*create_xxx)(tao));
21169566063dSJacob Faibussowitsch   PetscCall(PetscObjectChangeTypeName((PetscObject)tao, type));
21173ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
2118a7e14dcfSSatish Balay }
211947a47007SBarry Smith 
212067be906fSBarry Smith /*@C
212147450a7bSBarry Smith   TaoRegister - Adds a method to the Tao package for minimization.
2122a7e14dcfSSatish Balay 
212320f4b53cSBarry Smith   Not Collective
2124a7e14dcfSSatish Balay 
2125a7e14dcfSSatish Balay   Input Parameters:
2126a7e14dcfSSatish Balay + sname - name of a new user-defined solver
2127a7e14dcfSSatish Balay - func  - routine to Create method context
2128a7e14dcfSSatish Balay 
2129e056e8ceSJacob Faibussowitsch   Example Usage:
2130a7e14dcfSSatish Balay .vb
2131441846f8SBarry Smith    TaoRegister("my_solver", MySolverCreate);
2132a7e14dcfSSatish Balay .ve
2133a7e14dcfSSatish Balay 
2134a7e14dcfSSatish Balay   Then, your solver can be chosen with the procedural interface via
2135a7e14dcfSSatish Balay $     TaoSetType(tao, "my_solver")
2136a7e14dcfSSatish Balay   or at runtime via the option
213758417fe7SBarry Smith $     -tao_type my_solver
2138a7e14dcfSSatish Balay 
2139a7e14dcfSSatish Balay   Level: advanced
2140a7e14dcfSSatish Balay 
214167be906fSBarry Smith   Note:
214267be906fSBarry Smith   `TaoRegister()` may be called multiple times to add several user-defined solvers.
214367be906fSBarry Smith 
21441cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoSetType()`, `TaoRegisterAll()`, `TaoRegisterDestroy()`
214567be906fSBarry Smith @*/
2146d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoRegister(const char sname[], PetscErrorCode (*func)(Tao))
2147d71ae5a4SJacob Faibussowitsch {
2148a7e14dcfSSatish Balay   PetscFunctionBegin;
21499566063dSJacob Faibussowitsch   PetscCall(TaoInitializePackage());
21509566063dSJacob Faibussowitsch   PetscCall(PetscFunctionListAdd(&TaoList, sname, (void (*)(void))func));
21513ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
2152a7e14dcfSSatish Balay }
2153a7e14dcfSSatish Balay 
2154a7e14dcfSSatish Balay /*@C
2155441846f8SBarry Smith   TaoRegisterDestroy - Frees the list of minimization solvers that were
215647450a7bSBarry Smith   registered by `TaoRegister()`.
2157a7e14dcfSSatish Balay 
2158a7e14dcfSSatish Balay   Not Collective
2159a7e14dcfSSatish Balay 
2160a7e14dcfSSatish Balay   Level: advanced
2161a7e14dcfSSatish Balay 
21621cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoRegisterAll()`, `TaoRegister()`
2163a7e14dcfSSatish Balay @*/
2164d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoRegisterDestroy(void)
2165d71ae5a4SJacob Faibussowitsch {
2166a7e14dcfSSatish Balay   PetscFunctionBegin;
21679566063dSJacob Faibussowitsch   PetscCall(PetscFunctionListDestroy(&TaoList));
2168441846f8SBarry Smith   TaoRegisterAllCalled = PETSC_FALSE;
21693ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
2170a7e14dcfSSatish Balay }
217147a47007SBarry Smith 
21728931d482SJason Sarich /*@
217347450a7bSBarry Smith   TaoGetIterationNumber - Gets the number of `TaoSolve()` iterations completed
21748931d482SJason Sarich   at this time.
21758931d482SJason Sarich 
21768931d482SJason Sarich   Not Collective
21778931d482SJason Sarich 
21788931d482SJason Sarich   Input Parameter:
217947450a7bSBarry Smith . tao - the `Tao` context
21808931d482SJason Sarich 
21818931d482SJason Sarich   Output Parameter:
21828931d482SJason Sarich . iter - iteration number
21838931d482SJason Sarich 
21848931d482SJason Sarich   Notes:
21858931d482SJason Sarich   For example, during the computation of iteration 2 this would return 1.
21868931d482SJason Sarich 
21878931d482SJason Sarich   Level: intermediate
21888931d482SJason Sarich 
21891cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoGetLinearSolveIterations()`, `TaoGetResidualNorm()`, `TaoGetObjective()`
21908931d482SJason Sarich @*/
2191d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoGetIterationNumber(Tao tao, PetscInt *iter)
2192d71ae5a4SJacob Faibussowitsch {
21938931d482SJason Sarich   PetscFunctionBegin;
21948931d482SJason Sarich   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
21954f572ea9SToby Isaac   PetscAssertPointer(iter, 2);
21968931d482SJason Sarich   *iter = tao->niter;
21973ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
21988931d482SJason Sarich }
21998931d482SJason Sarich 
22008931d482SJason Sarich /*@
220147450a7bSBarry Smith   TaoGetResidualNorm - Gets the current value of the norm of the residual (gradient)
220279f5d8caSBarry Smith   at this time.
220379f5d8caSBarry Smith 
220479f5d8caSBarry Smith   Not Collective
220579f5d8caSBarry Smith 
220679f5d8caSBarry Smith   Input Parameter:
220747450a7bSBarry Smith . tao - the `Tao` context
220879f5d8caSBarry Smith 
220979f5d8caSBarry Smith   Output Parameter:
221079f5d8caSBarry Smith . value - the current value
221179f5d8caSBarry Smith 
221279f5d8caSBarry Smith   Level: intermediate
221379f5d8caSBarry Smith 
2214e056e8ceSJacob Faibussowitsch   Developer Notes:
221567be906fSBarry Smith   This is the 2-norm of the residual, we cannot use `TaoGetGradientNorm()` because that has
221647450a7bSBarry Smith   a different meaning. For some reason `Tao` sometimes calls the gradient the residual.
221779f5d8caSBarry Smith 
22181cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoGetLinearSolveIterations()`, `TaoGetIterationNumber()`, `TaoGetObjective()`
221979f5d8caSBarry Smith @*/
2220d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoGetResidualNorm(Tao tao, PetscReal *value)
2221d71ae5a4SJacob Faibussowitsch {
222279f5d8caSBarry Smith   PetscFunctionBegin;
222379f5d8caSBarry Smith   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
22244f572ea9SToby Isaac   PetscAssertPointer(value, 2);
222579f5d8caSBarry Smith   *value = tao->residual;
22263ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
222779f5d8caSBarry Smith }
222879f5d8caSBarry Smith 
222979f5d8caSBarry Smith /*@
22308931d482SJason Sarich   TaoSetIterationNumber - Sets the current iteration number.
22318931d482SJason Sarich 
2232c3339decSBarry Smith   Logically Collective
22338931d482SJason Sarich 
2234d8d19677SJose E. Roman   Input Parameters:
223547450a7bSBarry Smith + tao  - the `Tao` context
2236a2b725a8SWilliam Gropp - iter - iteration number
22378931d482SJason Sarich 
22388931d482SJason Sarich   Level: developer
22398931d482SJason Sarich 
22401cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoGetLinearSolveIterations()`
22418931d482SJason Sarich @*/
2242d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoSetIterationNumber(Tao tao, PetscInt iter)
2243d71ae5a4SJacob Faibussowitsch {
22448931d482SJason Sarich   PetscFunctionBegin;
22458931d482SJason Sarich   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
224694511df7SStefano Zampini   PetscValidLogicalCollectiveInt(tao, iter, 2);
22479566063dSJacob Faibussowitsch   PetscCall(PetscObjectSAWsTakeAccess((PetscObject)tao));
22488931d482SJason Sarich   tao->niter = iter;
22499566063dSJacob Faibussowitsch   PetscCall(PetscObjectSAWsGrantAccess((PetscObject)tao));
22503ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
22518931d482SJason Sarich }
22528931d482SJason Sarich 
22538931d482SJason Sarich /*@
225447450a7bSBarry Smith   TaoGetTotalIterationNumber - Gets the total number of `TaoSolve()` iterations
22558931d482SJason Sarich   completed. This number keeps accumulating if multiple solves
225647450a7bSBarry Smith   are called with the `Tao` object.
22578931d482SJason Sarich 
22588931d482SJason Sarich   Not Collective
22598931d482SJason Sarich 
22608931d482SJason Sarich   Input Parameter:
226147450a7bSBarry Smith . tao - the `Tao` context
22628931d482SJason Sarich 
22638931d482SJason Sarich   Output Parameter:
226447450a7bSBarry Smith . iter - number of iterations
22658931d482SJason Sarich 
22668931d482SJason Sarich   Level: intermediate
22678931d482SJason Sarich 
226820f4b53cSBarry Smith   Note:
226967be906fSBarry Smith   The total iteration count is updated after each solve, if there is a current
227047450a7bSBarry Smith   `TaoSolve()` in progress then those iterations are not included in the count
227167be906fSBarry Smith 
22721cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoGetLinearSolveIterations()`
22738931d482SJason Sarich @*/
2274d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoGetTotalIterationNumber(Tao tao, PetscInt *iter)
2275d71ae5a4SJacob Faibussowitsch {
22768931d482SJason Sarich   PetscFunctionBegin;
22778931d482SJason Sarich   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
22784f572ea9SToby Isaac   PetscAssertPointer(iter, 2);
22798931d482SJason Sarich   *iter = tao->ntotalits;
22803ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
22818931d482SJason Sarich }
22828931d482SJason Sarich 
22838931d482SJason Sarich /*@
22848931d482SJason Sarich   TaoSetTotalIterationNumber - Sets the current total iteration number.
22858931d482SJason Sarich 
2286c3339decSBarry Smith   Logically Collective
22878931d482SJason Sarich 
2288d8d19677SJose E. Roman   Input Parameters:
228947450a7bSBarry Smith + tao  - the `Tao` context
229047450a7bSBarry Smith - iter - the iteration number
22918931d482SJason Sarich 
22928931d482SJason Sarich   Level: developer
22938931d482SJason Sarich 
22941cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoGetLinearSolveIterations()`
22958931d482SJason Sarich @*/
2296d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoSetTotalIterationNumber(Tao tao, PetscInt iter)
2297d71ae5a4SJacob Faibussowitsch {
22988931d482SJason Sarich   PetscFunctionBegin;
22998931d482SJason Sarich   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
230094511df7SStefano Zampini   PetscValidLogicalCollectiveInt(tao, iter, 2);
23019566063dSJacob Faibussowitsch   PetscCall(PetscObjectSAWsTakeAccess((PetscObject)tao));
23028931d482SJason Sarich   tao->ntotalits = iter;
23039566063dSJacob Faibussowitsch   PetscCall(PetscObjectSAWsGrantAccess((PetscObject)tao));
23043ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
23058931d482SJason Sarich }
23068931d482SJason Sarich 
2307a7e14dcfSSatish Balay /*@
230847450a7bSBarry Smith   TaoSetConvergedReason - Sets the termination flag on a `Tao` object
2309a7e14dcfSSatish Balay 
2310c3339decSBarry Smith   Logically Collective
2311a7e14dcfSSatish Balay 
2312a7e14dcfSSatish Balay   Input Parameters:
231347450a7bSBarry Smith + tao    - the `Tao` context
231447450a7bSBarry Smith - reason - the `TaoConvergedReason`
2315a7e14dcfSSatish Balay 
2316a7e14dcfSSatish Balay   Level: intermediate
2317a7e14dcfSSatish Balay 
23181cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoConvergedReason`
2319a7e14dcfSSatish Balay @*/
2320d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoSetConvergedReason(Tao tao, TaoConvergedReason reason)
2321d71ae5a4SJacob Faibussowitsch {
2322a7e14dcfSSatish Balay   PetscFunctionBegin;
232394511df7SStefano Zampini   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
232494511df7SStefano Zampini   PetscValidLogicalCollectiveEnum(tao, reason, 2);
2325a7e14dcfSSatish Balay   tao->reason = reason;
23263ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
2327a7e14dcfSSatish Balay }
2328a7e14dcfSSatish Balay 
2329a7e14dcfSSatish Balay /*@
233047450a7bSBarry Smith   TaoGetConvergedReason - Gets the reason the `TaoSolve()` was stopped.
2331a7e14dcfSSatish Balay 
2332a7e14dcfSSatish Balay   Not Collective
2333a7e14dcfSSatish Balay 
2334a7e14dcfSSatish Balay   Input Parameter:
233547450a7bSBarry Smith . tao - the `Tao` solver context
2336a7e14dcfSSatish Balay 
2337a7e14dcfSSatish Balay   Output Parameter:
233847450a7bSBarry Smith . reason - value of `TaoConvergedReason`
2339a7e14dcfSSatish Balay 
2340a7e14dcfSSatish Balay   Level: intermediate
2341a7e14dcfSSatish Balay 
23421cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoConvergedReason`, `TaoSetConvergenceTest()`, `TaoSetTolerances()`
2343a7e14dcfSSatish Balay @*/
2344d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoGetConvergedReason(Tao tao, TaoConvergedReason *reason)
2345d71ae5a4SJacob Faibussowitsch {
2346a7e14dcfSSatish Balay   PetscFunctionBegin;
2347441846f8SBarry Smith   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
23484f572ea9SToby Isaac   PetscAssertPointer(reason, 2);
2349a7e14dcfSSatish Balay   *reason = tao->reason;
23503ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
2351a7e14dcfSSatish Balay }
2352a7e14dcfSSatish Balay 
2353a7e14dcfSSatish Balay /*@
2354a7e14dcfSSatish Balay   TaoGetSolutionStatus - Get the current iterate, objective value,
235547450a7bSBarry Smith   residual, infeasibility, and termination from a `Tao` object
2356a7e14dcfSSatish Balay 
2357a7e14dcfSSatish Balay   Not Collective
2358a7e14dcfSSatish Balay 
2359f899ff85SJose E. Roman   Input Parameter:
236047450a7bSBarry Smith . tao - the `Tao` context
2361a7e14dcfSSatish Balay 
2362a7e14dcfSSatish Balay   Output Parameters:
2363e056e8ceSJacob Faibussowitsch + its    - the current iterate number (>=0)
2364a7e14dcfSSatish Balay . f      - the current function value
236558417fe7SBarry Smith . gnorm  - the square of the gradient norm, duality gap, or other measure indicating distance from optimality.
2366a7e14dcfSSatish Balay . cnorm  - the infeasibility of the current solution with regard to the constraints.
2367a7e14dcfSSatish Balay . xdiff  - the step length or trust region radius of the most recent iterate.
236865ba42b6SBarry Smith - reason - The termination reason, which can equal `TAO_CONTINUE_ITERATING`
2369a7e14dcfSSatish Balay 
2370a7e14dcfSSatish Balay   Level: intermediate
2371a7e14dcfSSatish Balay 
237265ba42b6SBarry Smith   Notes:
237365ba42b6SBarry Smith   Tao returns the values set by the solvers in the routine `TaoMonitor()`.
2374a7e14dcfSSatish Balay 
237567be906fSBarry Smith   If any of the output arguments are set to `NULL`, no corresponding value will be returned.
2376a7e14dcfSSatish Balay 
23771cc06b55SBarry Smith .seealso: [](ch_tao), `TaoMonitor()`, `TaoGetConvergedReason()`
2378a7e14dcfSSatish Balay @*/
2379d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoGetSolutionStatus(Tao tao, PetscInt *its, PetscReal *f, PetscReal *gnorm, PetscReal *cnorm, PetscReal *xdiff, TaoConvergedReason *reason)
2380d71ae5a4SJacob Faibussowitsch {
2381a7e14dcfSSatish Balay   PetscFunctionBegin;
238294511df7SStefano Zampini   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
2383a7e14dcfSSatish Balay   if (its) *its = tao->niter;
2384a7e14dcfSSatish Balay   if (f) *f = tao->fc;
2385a7e14dcfSSatish Balay   if (gnorm) *gnorm = tao->residual;
2386a7e14dcfSSatish Balay   if (cnorm) *cnorm = tao->cnorm;
2387a7e14dcfSSatish Balay   if (reason) *reason = tao->reason;
2388a7e14dcfSSatish Balay   if (xdiff) *xdiff = tao->step;
23893ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
2390a7e14dcfSSatish Balay }
2391a7e14dcfSSatish Balay 
2392a7e14dcfSSatish Balay /*@C
239347450a7bSBarry Smith   TaoGetType - Gets the current `TaoType` being used in the `Tao` object
2394a7e14dcfSSatish Balay 
2395a7e14dcfSSatish Balay   Not Collective
2396a7e14dcfSSatish Balay 
2397a7e14dcfSSatish Balay   Input Parameter:
239847450a7bSBarry Smith . tao - the `Tao` solver context
2399a7e14dcfSSatish Balay 
2400a7e14dcfSSatish Balay   Output Parameter:
240147450a7bSBarry Smith . type - the `TaoType`
2402a7e14dcfSSatish Balay 
2403a7e14dcfSSatish Balay   Level: intermediate
2404a7e14dcfSSatish Balay 
24051cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoType`, `TaoSetType()`
2406a7e14dcfSSatish Balay @*/
2407d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoGetType(Tao tao, TaoType *type)
2408d71ae5a4SJacob Faibussowitsch {
2409a7e14dcfSSatish Balay   PetscFunctionBegin;
2410441846f8SBarry Smith   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
24114f572ea9SToby Isaac   PetscAssertPointer(type, 2);
2412a7e14dcfSSatish Balay   *type = ((PetscObject)tao)->type_name;
24133ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
2414a7e14dcfSSatish Balay }
2415a7e14dcfSSatish Balay 
2416a7e14dcfSSatish Balay /*@C
2417a7e14dcfSSatish Balay   TaoMonitor - Monitor the solver and the current solution.  This
2418a7e14dcfSSatish Balay   routine will record the iteration number and residual statistics,
2419670c031eSStefano Zampini   and call any monitors specified by the user.
2420a7e14dcfSSatish Balay 
2421a7e14dcfSSatish Balay   Input Parameters:
242247450a7bSBarry Smith + tao        - the `Tao` context
2423a7e14dcfSSatish Balay . its        - the current iterate number (>=0)
2424a7e14dcfSSatish Balay . f          - the current objective function value
24251cb3f120SPierre Jolivet . res        - the gradient norm, square root of the duality gap, or other measure indicating distance from optimality.  This measure will be recorded and
2426a7e14dcfSSatish Balay           used for some termination tests.
2427a7e14dcfSSatish Balay . cnorm      - the infeasibility of the current solution with regard to the constraints.
2428a7e14dcfSSatish Balay - steplength - multiple of the step direction added to the previous iterate.
2429a7e14dcfSSatish Balay 
2430a7e14dcfSSatish Balay   Options Database Key:
2431a7e14dcfSSatish Balay . -tao_monitor - Use the default monitor, which prints statistics to standard output
2432a7e14dcfSSatish Balay 
2433a7e14dcfSSatish Balay   Level: developer
2434a7e14dcfSSatish Balay 
24351cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoGetConvergedReason()`, `TaoMonitorDefault()`, `TaoSetMonitor()`
2436a7e14dcfSSatish Balay @*/
2437d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoMonitor(Tao tao, PetscInt its, PetscReal f, PetscReal res, PetscReal cnorm, PetscReal steplength)
2438d71ae5a4SJacob Faibussowitsch {
243947a47007SBarry Smith   PetscInt i;
244047a47007SBarry Smith 
2441a7e14dcfSSatish Balay   PetscFunctionBegin;
2442441846f8SBarry Smith   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
2443a7e14dcfSSatish Balay   tao->fc       = f;
2444a7e14dcfSSatish Balay   tao->residual = res;
2445a7e14dcfSSatish Balay   tao->cnorm    = cnorm;
2446a7e14dcfSSatish Balay   tao->step     = steplength;
2447e52336cbSBarry Smith   if (!its) {
244894511df7SStefano Zampini     tao->cnorm0 = cnorm;
244994511df7SStefano Zampini     tao->gnorm0 = res;
2450a7e14dcfSSatish Balay   }
2451bfdcf5a2SStefano Zampini   PetscCall(VecLockReadPush(tao->solution));
245248a46eb9SPierre Jolivet   for (i = 0; i < tao->numbermonitors; i++) PetscCall((*tao->monitor[i])(tao, tao->monitorcontext[i]));
2453bfdcf5a2SStefano Zampini   PetscCall(VecLockReadPop(tao->solution));
24543ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
2455a7e14dcfSSatish Balay }
2456a7e14dcfSSatish Balay 
2457a7e14dcfSSatish Balay /*@
2458ae93cb3cSJason Sarich   TaoSetConvergenceHistory - Sets the array used to hold the convergence history.
2459a7e14dcfSSatish Balay 
2460c3339decSBarry Smith   Logically Collective
2461a7e14dcfSSatish Balay 
2462a7e14dcfSSatish Balay   Input Parameters:
246347450a7bSBarry Smith + tao   - the `Tao` solver context
2464a7e14dcfSSatish Balay . obj   - array to hold objective value history
2465a7e14dcfSSatish Balay . resid - array to hold residual history
2466a7e14dcfSSatish Balay . cnorm - array to hold constraint violation history
2467be1558d8SJason Sarich . lits  - integer array holds the number of linear iterations for each Tao iteration
246867be906fSBarry Smith . na    - size of `obj`, `resid`, and `cnorm`
246965ba42b6SBarry Smith - reset - `PETSC_TRUE` indicates each new minimization resets the history counter to zero,
2470a7e14dcfSSatish Balay            else it continues storing new values for new minimizations after the old ones
2471a7e14dcfSSatish Balay 
247267be906fSBarry Smith   Level: intermediate
247367be906fSBarry Smith 
2474a7e14dcfSSatish Balay   Notes:
247547450a7bSBarry Smith   If set, `Tao` will fill the given arrays with the indicated
2476ae93cb3cSJason Sarich   information at each iteration.  If 'obj','resid','cnorm','lits' are
247767be906fSBarry Smith   *all* `NULL` then space (using size `na`, or 1000 if na is `PETSC_DECIDE` or
247865ba42b6SBarry Smith   `PETSC_DEFAULT`) is allocated for the history.
247967be906fSBarry Smith   If not all are `NULL`, then only the non-`NULL` information categories
2480ae93cb3cSJason Sarich   will be stored, the others will be ignored.
2481ae93cb3cSJason Sarich 
2482ae93cb3cSJason Sarich   Any convergence information after iteration number 'na' will not be stored.
2483a7e14dcfSSatish Balay 
2484a7e14dcfSSatish Balay   This routine is useful, e.g., when running a code for purposes
2485a7e14dcfSSatish Balay   of accurate performance monitoring, when no I/O should be done
2486a7e14dcfSSatish Balay   during the section of code that is being timed.
2487a7e14dcfSSatish Balay 
24881cc06b55SBarry Smith .seealso: [](ch_tao), `TaoGetConvergenceHistory()`
2489a7e14dcfSSatish Balay @*/
2490d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoSetConvergenceHistory(Tao tao, PetscReal obj[], PetscReal resid[], PetscReal cnorm[], PetscInt lits[], PetscInt na, PetscBool reset)
2491d71ae5a4SJacob Faibussowitsch {
2492a7e14dcfSSatish Balay   PetscFunctionBegin;
2493441846f8SBarry Smith   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
24944f572ea9SToby Isaac   if (obj) PetscAssertPointer(obj, 2);
24954f572ea9SToby Isaac   if (resid) PetscAssertPointer(resid, 3);
24964f572ea9SToby Isaac   if (cnorm) PetscAssertPointer(cnorm, 4);
24974f572ea9SToby Isaac   if (lits) PetscAssertPointer(lits, 5);
2498ae93cb3cSJason Sarich 
2499ae93cb3cSJason Sarich   if (na == PETSC_DECIDE || na == PETSC_DEFAULT) na = 1000;
2500ae93cb3cSJason Sarich   if (!obj && !resid && !cnorm && !lits) {
25019566063dSJacob Faibussowitsch     PetscCall(PetscCalloc4(na, &obj, na, &resid, na, &cnorm, na, &lits));
2502ae93cb3cSJason Sarich     tao->hist_malloc = PETSC_TRUE;
2503ae93cb3cSJason Sarich   }
2504ae93cb3cSJason Sarich 
2505a7e14dcfSSatish Balay   tao->hist_obj   = obj;
2506a7e14dcfSSatish Balay   tao->hist_resid = resid;
2507a7e14dcfSSatish Balay   tao->hist_cnorm = cnorm;
2508ae93cb3cSJason Sarich   tao->hist_lits  = lits;
2509a7e14dcfSSatish Balay   tao->hist_max   = na;
2510a7e14dcfSSatish Balay   tao->hist_reset = reset;
2511ae93cb3cSJason Sarich   tao->hist_len   = 0;
25123ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
2513a7e14dcfSSatish Balay }
2514a7e14dcfSSatish Balay 
2515a7e14dcfSSatish Balay /*@C
251665ba42b6SBarry Smith   TaoGetConvergenceHistory - Gets the arrays used that hold the convergence history.
2517a7e14dcfSSatish Balay 
2518c3339decSBarry Smith   Collective
2519a7e14dcfSSatish Balay 
2520a7e14dcfSSatish Balay   Input Parameter:
252147450a7bSBarry Smith . tao - the `Tao` context
2522a7e14dcfSSatish Balay 
2523a7e14dcfSSatish Balay   Output Parameters:
2524a7e14dcfSSatish Balay + obj   - array used to hold objective value history
2525a7e14dcfSSatish Balay . resid - array used to hold residual history
2526a7e14dcfSSatish Balay . cnorm - array used to hold constraint violation history
2527ae93cb3cSJason Sarich . lits  - integer array used to hold linear solver iteration count
252867be906fSBarry Smith - nhist - size of `obj`, `resid`, `cnorm`, and `lits`
252967be906fSBarry Smith 
253067be906fSBarry Smith   Level: advanced
2531a7e14dcfSSatish Balay 
2532a7e14dcfSSatish Balay   Notes:
253365ba42b6SBarry Smith   This routine must be preceded by calls to `TaoSetConvergenceHistory()`
253465ba42b6SBarry Smith   and `TaoSolve()`, otherwise it returns useless information.
2535ae93cb3cSJason Sarich 
2536a7e14dcfSSatish Balay   This routine is useful, e.g., when running a code for purposes
2537a7e14dcfSSatish Balay   of accurate performance monitoring, when no I/O should be done
2538a7e14dcfSSatish Balay   during the section of code that is being timed.
2539a7e14dcfSSatish Balay 
2540e056e8ceSJacob Faibussowitsch   Fortran Notes:
254167be906fSBarry Smith   The calling sequence is
254267be906fSBarry Smith .vb
254367be906fSBarry Smith    call TaoGetConvergenceHistory(Tao tao, PetscInt nhist, PetscErrorCode ierr)
254467be906fSBarry Smith .ve
2545a7e14dcfSSatish Balay 
25461cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoSolve()`, `TaoSetConvergenceHistory()`
2547a7e14dcfSSatish Balay @*/
2548d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoGetConvergenceHistory(Tao tao, PetscReal **obj, PetscReal **resid, PetscReal **cnorm, PetscInt **lits, PetscInt *nhist)
2549d71ae5a4SJacob Faibussowitsch {
2550a7e14dcfSSatish Balay   PetscFunctionBegin;
2551441846f8SBarry Smith   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
2552a7e14dcfSSatish Balay   if (obj) *obj = tao->hist_obj;
2553a7e14dcfSSatish Balay   if (cnorm) *cnorm = tao->hist_cnorm;
2554a7e14dcfSSatish Balay   if (resid) *resid = tao->hist_resid;
25555b494b05SBarry Smith   if (lits) *lits = tao->hist_lits;
2556a7e14dcfSSatish Balay   if (nhist) *nhist = tao->hist_len;
25573ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
2558a7e14dcfSSatish Balay }
2559a7e14dcfSSatish Balay 
2560a7e14dcfSSatish Balay /*@
256147450a7bSBarry Smith   TaoSetApplicationContext - Sets the optional user-defined context for a `Tao` solver.
2562a7e14dcfSSatish Balay 
2563c3339decSBarry Smith   Logically Collective
2564a7e14dcfSSatish Balay 
2565a7e14dcfSSatish Balay   Input Parameters:
256647450a7bSBarry Smith + tao  - the `Tao` context
2567a7e14dcfSSatish Balay - usrP - optional user context
2568a7e14dcfSSatish Balay 
2569a7e14dcfSSatish Balay   Level: intermediate
2570a7e14dcfSSatish Balay 
257142747ad1SJacob Faibussowitsch .seealso: [](ch_tao), `Tao`, `TaoGetApplicationContext()`
2572a7e14dcfSSatish Balay @*/
2573d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoSetApplicationContext(Tao tao, void *usrP)
2574d71ae5a4SJacob Faibussowitsch {
2575a7e14dcfSSatish Balay   PetscFunctionBegin;
2576441846f8SBarry Smith   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
2577a7e14dcfSSatish Balay   tao->user = usrP;
25783ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
2579a7e14dcfSSatish Balay }
2580a7e14dcfSSatish Balay 
2581a7e14dcfSSatish Balay /*@
258247450a7bSBarry Smith   TaoGetApplicationContext - Gets the user-defined context for a `Tao` solver
2583a7e14dcfSSatish Balay 
2584a7e14dcfSSatish Balay   Not Collective
2585a7e14dcfSSatish Balay 
2586a7e14dcfSSatish Balay   Input Parameter:
258747450a7bSBarry Smith . tao - the `Tao` context
2588a7e14dcfSSatish Balay 
2589a7e14dcfSSatish Balay   Output Parameter:
2590a7e14dcfSSatish Balay . usrP - user context
2591a7e14dcfSSatish Balay 
2592a7e14dcfSSatish Balay   Level: intermediate
2593a7e14dcfSSatish Balay 
25941cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoSetApplicationContext()`
2595a7e14dcfSSatish Balay @*/
2596d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoGetApplicationContext(Tao tao, void *usrP)
2597d71ae5a4SJacob Faibussowitsch {
2598a7e14dcfSSatish Balay   PetscFunctionBegin;
2599441846f8SBarry Smith   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
26004f572ea9SToby Isaac   PetscAssertPointer(usrP, 2);
2601a7e14dcfSSatish Balay   *(void **)usrP = tao->user;
26023ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
2603a7e14dcfSSatish Balay }
2604a9603a14SPatrick Farrell 
2605a9603a14SPatrick Farrell /*@
260665ba42b6SBarry Smith   TaoSetGradientNorm - Sets the matrix used to define the norm that measures the size of the gradient.
2607a9603a14SPatrick Farrell 
2608c3339decSBarry Smith   Collective
2609a9603a14SPatrick Farrell 
2610a9603a14SPatrick Farrell   Input Parameters:
261147450a7bSBarry Smith + tao - the `Tao` context
261265ba42b6SBarry Smith - M   - matrix that defines the norm
2613a9603a14SPatrick Farrell 
2614a9603a14SPatrick Farrell   Level: beginner
2615a9603a14SPatrick Farrell 
26161cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoGetGradientNorm()`, `TaoGradientNorm()`
2617a9603a14SPatrick Farrell @*/
2618d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoSetGradientNorm(Tao tao, Mat M)
2619d71ae5a4SJacob Faibussowitsch {
2620a9603a14SPatrick Farrell   PetscFunctionBegin;
2621a9603a14SPatrick Farrell   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
262294511df7SStefano Zampini   PetscValidHeaderSpecific(M, MAT_CLASSID, 2);
26239566063dSJacob Faibussowitsch   PetscCall(PetscObjectReference((PetscObject)M));
26249566063dSJacob Faibussowitsch   PetscCall(MatDestroy(&tao->gradient_norm));
26259566063dSJacob Faibussowitsch   PetscCall(VecDestroy(&tao->gradient_norm_tmp));
2626a9603a14SPatrick Farrell   tao->gradient_norm = M;
26279566063dSJacob Faibussowitsch   PetscCall(MatCreateVecs(M, NULL, &tao->gradient_norm_tmp));
26283ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
2629a9603a14SPatrick Farrell }
2630a9603a14SPatrick Farrell 
2631a9603a14SPatrick Farrell /*@
263265ba42b6SBarry Smith   TaoGetGradientNorm - Returns the matrix used to define the norm used for measuring the size of the gradient.
2633a9603a14SPatrick Farrell 
2634a9603a14SPatrick Farrell   Not Collective
2635a9603a14SPatrick Farrell 
2636a9603a14SPatrick Farrell   Input Parameter:
263747450a7bSBarry Smith . tao - the `Tao` context
2638a9603a14SPatrick Farrell 
2639a9603a14SPatrick Farrell   Output Parameter:
2640a9603a14SPatrick Farrell . M - gradient norm
2641a9603a14SPatrick Farrell 
2642a9603a14SPatrick Farrell   Level: beginner
2643a9603a14SPatrick Farrell 
26441cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoSetGradientNorm()`, `TaoGradientNorm()`
2645a9603a14SPatrick Farrell @*/
2646d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoGetGradientNorm(Tao tao, Mat *M)
2647d71ae5a4SJacob Faibussowitsch {
2648a9603a14SPatrick Farrell   PetscFunctionBegin;
2649a9603a14SPatrick Farrell   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
26504f572ea9SToby Isaac   PetscAssertPointer(M, 2);
2651a9603a14SPatrick Farrell   *M = tao->gradient_norm;
26523ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
2653a9603a14SPatrick Farrell }
2654a9603a14SPatrick Farrell 
265505aca2cdSSatish Balay /*@C
265647450a7bSBarry Smith   TaoGradientNorm - Compute the norm using the `NormType`, the user has selected
2657a9603a14SPatrick Farrell 
2658c3339decSBarry Smith   Collective
2659a9603a14SPatrick Farrell 
2660d8d19677SJose E. Roman   Input Parameters:
266147450a7bSBarry Smith + tao      - the `Tao` context
2662a9603a14SPatrick Farrell . gradient - the gradient to be computed
26633b242c63SJacob Faibussowitsch - type     - the norm type
2664a9603a14SPatrick Farrell 
2665a9603a14SPatrick Farrell   Output Parameter:
2666a9603a14SPatrick Farrell . gnorm - the gradient norm
2667a9603a14SPatrick Farrell 
266847450a7bSBarry Smith   Level: advanced
2669a9603a14SPatrick Farrell 
26701cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoSetGradientNorm()`, `TaoGetGradientNorm()`
2671a9603a14SPatrick Farrell @*/
2672d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoGradientNorm(Tao tao, Vec gradient, NormType type, PetscReal *gnorm)
2673d71ae5a4SJacob Faibussowitsch {
2674a9603a14SPatrick Farrell   PetscFunctionBegin;
26758854b543SStefano Zampini   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
26768854b543SStefano Zampini   PetscValidHeaderSpecific(gradient, VEC_CLASSID, 2);
26778854b543SStefano Zampini   PetscValidLogicalCollectiveEnum(tao, type, 3);
26784f572ea9SToby Isaac   PetscAssertPointer(gnorm, 4);
2679a9603a14SPatrick Farrell   if (tao->gradient_norm) {
2680680e2bc7SPatrick Farrell     PetscScalar gnorms;
2681680e2bc7SPatrick Farrell 
26823c859ba3SBarry Smith     PetscCheck(type == NORM_2, PetscObjectComm((PetscObject)gradient), PETSC_ERR_ARG_WRONG, "Norm type must be NORM_2 if an inner product for the gradient norm is set.");
26839566063dSJacob Faibussowitsch     PetscCall(MatMult(tao->gradient_norm, gradient, tao->gradient_norm_tmp));
26849566063dSJacob Faibussowitsch     PetscCall(VecDot(gradient, tao->gradient_norm_tmp, &gnorms));
26857bb79937SPatrick Farrell     *gnorm = PetscRealPart(PetscSqrtScalar(gnorms));
2686a9603a14SPatrick Farrell   } else {
26879566063dSJacob Faibussowitsch     PetscCall(VecNorm(gradient, type, gnorm));
2688a9603a14SPatrick Farrell   }
26893ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
2690a9603a14SPatrick Farrell }
2691a9603a14SPatrick Farrell 
2692e882e171SHong Zhang /*@C
269347450a7bSBarry Smith   TaoMonitorDrawCtxCreate - Creates the monitor context for `TaoMonitorDrawSolution()`
2694a9603a14SPatrick Farrell 
2695c3339decSBarry Smith   Collective
2696e882e171SHong Zhang 
26972fe279fdSBarry Smith   Input Parameters:
26982fe279fdSBarry Smith + comm     - the communicator to share the context
26992fe279fdSBarry Smith . host     - the name of the X Windows host that will display the monitor
27002fe279fdSBarry Smith . label    - the label to put at the top of the display window
27012fe279fdSBarry Smith . x        - the horizontal coordinate of the lower left corner of the window to open
27022fe279fdSBarry Smith . y        - the vertical coordinate of the lower left corner of the window to open
27032fe279fdSBarry Smith . m        - the width of the window
27042fe279fdSBarry Smith . n        - the height of the window
27052fe279fdSBarry Smith - howoften - how many `Tao` iterations between displaying the monitor information
27062fe279fdSBarry Smith 
2707d5b43468SJose E. Roman   Output Parameter:
2708e882e171SHong Zhang . ctx - the monitor context
2709e882e171SHong Zhang 
27103c7db156SBarry Smith   Options Database Key:
2711e882e171SHong Zhang . -tao_draw_solution_initial - show initial guess as well as current solution
2712e882e171SHong Zhang 
2713e882e171SHong Zhang   Level: intermediate
2714e882e171SHong Zhang 
27151cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoMonitorSet()`, `TaoMonitorDefault()`, `VecView()`, `TaoMonitorDrawCtx()`
2716e882e171SHong Zhang @*/
2717d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoMonitorDrawCtxCreate(MPI_Comm comm, const char host[], const char label[], int x, int y, int m, int n, PetscInt howoften, TaoMonitorDrawCtx *ctx)
2718d71ae5a4SJacob Faibussowitsch {
2719e882e171SHong Zhang   PetscFunctionBegin;
27209566063dSJacob Faibussowitsch   PetscCall(PetscNew(ctx));
27219566063dSJacob Faibussowitsch   PetscCall(PetscViewerDrawOpen(comm, host, label, x, y, m, n, &(*ctx)->viewer));
27229566063dSJacob Faibussowitsch   PetscCall(PetscViewerSetFromOptions((*ctx)->viewer));
2723e882e171SHong Zhang   (*ctx)->howoften = howoften;
27243ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
2725e882e171SHong Zhang }
2726e882e171SHong Zhang 
2727e882e171SHong Zhang /*@C
272865ba42b6SBarry Smith   TaoMonitorDrawCtxDestroy - Destroys the monitor context for `TaoMonitorDrawSolution()`
2729e882e171SHong Zhang 
2730c3339decSBarry Smith   Collective
2731e882e171SHong Zhang 
273247450a7bSBarry Smith   Input Parameter:
2733e056e8ceSJacob Faibussowitsch . ictx - the monitor context
2734e882e171SHong Zhang 
2735e882e171SHong Zhang   Level: intermediate
2736e882e171SHong Zhang 
27371cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoMonitorSet()`, `TaoMonitorDefault()`, `VecView()`, `TaoMonitorDrawSolution()`
2738e882e171SHong Zhang @*/
2739d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoMonitorDrawCtxDestroy(TaoMonitorDrawCtx *ictx)
2740d71ae5a4SJacob Faibussowitsch {
2741e882e171SHong Zhang   PetscFunctionBegin;
27429566063dSJacob Faibussowitsch   PetscCall(PetscViewerDestroy(&(*ictx)->viewer));
27439566063dSJacob Faibussowitsch   PetscCall(PetscFree(*ictx));
27443ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
2745e882e171SHong Zhang }
2746