xref: /petsc/src/tao/interface/taosolver.c (revision e876fe7549bc2ee6a21cc02cc139aff122c6ae5c)
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));
2839566063dSJacob Faibussowitsch   PetscCall(TaoCancelMonitors(*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
350a7e14dcfSSatish Balay . -tao_cancelmonitors     - 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;
367*e876fe75SStephan Köhler   PetscBool   flg, found;
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;
4639566063dSJacob Faibussowitsch   PetscCall(PetscOptionsBool("-tao_cancelmonitors", "cancel all monitors and call any registered destroy routines", "TaoCancelMonitors", flg, &flg, NULL));
4649566063dSJacob Faibussowitsch   if (flg) PetscCall(TaoCancelMonitors(tao));
465a7e14dcfSSatish Balay 
4668afaa268SBarry Smith   flg = PETSC_FALSE;
4679566063dSJacob Faibussowitsch   PetscCall(PetscOptionsBool("-tao_draw_solution", "Plot solution vector at each iteration", "TaoSetMonitor", flg, &flg, NULL));
468a7e14dcfSSatish Balay   if (flg) {
469e882e171SHong Zhang     TaoMonitorDrawCtx drawctx;
470e882e171SHong Zhang     PetscInt          howoften = 1;
4719566063dSJacob Faibussowitsch     PetscCall(TaoMonitorDrawCtxCreate(PetscObjectComm((PetscObject)tao), NULL, NULL, PETSC_DECIDE, PETSC_DECIDE, 300, 300, howoften, &drawctx));
4729566063dSJacob Faibussowitsch     PetscCall(TaoSetMonitor(tao, TaoDrawSolutionMonitor, drawctx, (PetscErrorCode(*)(void **))TaoMonitorDrawCtxDestroy));
473a7e14dcfSSatish Balay   }
474a7e14dcfSSatish Balay 
4758afaa268SBarry Smith   flg = PETSC_FALSE;
4769566063dSJacob Faibussowitsch   PetscCall(PetscOptionsBool("-tao_draw_step", "plots step direction at each iteration", "TaoSetMonitor", flg, &flg, NULL));
4771baa6e33SBarry Smith   if (flg) PetscCall(TaoSetMonitor(tao, TaoDrawStepMonitor, NULL, NULL));
478a7e14dcfSSatish Balay 
4798afaa268SBarry Smith   flg = PETSC_FALSE;
4809566063dSJacob Faibussowitsch   PetscCall(PetscOptionsBool("-tao_draw_gradient", "plots gradient at each iteration", "TaoSetMonitor", flg, &flg, NULL));
481a7e14dcfSSatish Balay   if (flg) {
482e882e171SHong Zhang     TaoMonitorDrawCtx drawctx;
483e882e171SHong Zhang     PetscInt          howoften = 1;
4849566063dSJacob Faibussowitsch     PetscCall(TaoMonitorDrawCtxCreate(PetscObjectComm((PetscObject)tao), NULL, NULL, PETSC_DECIDE, PETSC_DECIDE, 300, 300, howoften, &drawctx));
4859566063dSJacob Faibussowitsch     PetscCall(TaoSetMonitor(tao, TaoDrawGradientMonitor, drawctx, (PetscErrorCode(*)(void **))TaoMonitorDrawCtxDestroy));
486a7e14dcfSSatish Balay   }
4878afaa268SBarry Smith   flg = PETSC_FALSE;
4889566063dSJacob Faibussowitsch   PetscCall(PetscOptionsBool("-tao_fd_gradient", "compute gradient using finite differences", "TaoDefaultComputeGradient", flg, &flg, NULL));
4891baa6e33SBarry Smith   if (flg) PetscCall(TaoSetGradient(tao, NULL, TaoDefaultComputeGradient, NULL));
490f4c1ad5cSStefano Zampini   flg = PETSC_FALSE;
4919566063dSJacob Faibussowitsch   PetscCall(PetscOptionsBool("-tao_fd_hessian", "compute hessian using finite differences", "TaoDefaultComputeHessian", flg, &flg, NULL));
492f4c1ad5cSStefano Zampini   if (flg) {
493f4c1ad5cSStefano Zampini     Mat H;
494f4c1ad5cSStefano Zampini 
4959566063dSJacob Faibussowitsch     PetscCall(MatCreate(PetscObjectComm((PetscObject)tao), &H));
4969566063dSJacob Faibussowitsch     PetscCall(MatSetType(H, MATAIJ));
4979566063dSJacob Faibussowitsch     PetscCall(TaoSetHessian(tao, H, H, TaoDefaultComputeHessian, NULL));
4989566063dSJacob Faibussowitsch     PetscCall(MatDestroy(&H));
499f4c1ad5cSStefano Zampini   }
500f4c1ad5cSStefano Zampini   flg = PETSC_FALSE;
5019566063dSJacob Faibussowitsch   PetscCall(PetscOptionsBool("-tao_mf_hessian", "compute matrix-free hessian using finite differences", "TaoDefaultComputeHessianMFFD", flg, &flg, NULL));
502f4c1ad5cSStefano Zampini   if (flg) {
503f4c1ad5cSStefano Zampini     Mat H;
504f4c1ad5cSStefano Zampini 
5059566063dSJacob Faibussowitsch     PetscCall(MatCreate(PetscObjectComm((PetscObject)tao), &H));
5069566063dSJacob Faibussowitsch     PetscCall(TaoSetHessian(tao, H, H, TaoDefaultComputeHessianMFFD, NULL));
5079566063dSJacob Faibussowitsch     PetscCall(MatDestroy(&H));
508f4c1ad5cSStefano Zampini   }
509*e876fe75SStephan Köhler   PetscCall(PetscOptionsBool("-tao_recycle_history", "enable recycling/re-using information from the previous TaoSolve() call for some algorithms", "TaoSetRecycleHistory", flg, &flg, &found));
510*e876fe75SStephan Köhler   if (found) PetscCall(TaoSetRecycleHistory(tao, flg));
5119566063dSJacob Faibussowitsch   PetscCall(PetscOptionsEnum("-tao_subset_type", "subset type", "", TaoSubSetTypes, (PetscEnum)tao->subset_type, (PetscEnum *)&tao->subset_type, NULL));
512a7e14dcfSSatish Balay 
5130f0abf79SStefano Zampini   if (tao->ksp) {
5140f0abf79SStefano Zampini     PetscCall(PetscOptionsBool("-tao_ksp_ew", "Use Eisentat-Walker linear system convergence test", "TaoKSPSetUseEW", tao->ksp_ewconv, &tao->ksp_ewconv, NULL));
5150f0abf79SStefano Zampini     PetscCall(TaoKSPSetUseEW(tao, tao->ksp_ewconv));
5160f0abf79SStefano Zampini   }
5170f0abf79SStefano Zampini 
518dbbe0bcdSBarry Smith   PetscTryTypeMethod(tao, setfromoptions, PetscOptionsObject);
51960b70c5cSStefano Zampini 
52060b70c5cSStefano Zampini   /* process any options handlers added with PetscObjectAddOptionsHandler() */
52160b70c5cSStefano Zampini   PetscCall(PetscObjectProcessOptionsHandlers((PetscObject)tao, PetscOptionsObject));
522d0609cedSBarry Smith   PetscOptionsEnd();
52360b70c5cSStefano Zampini 
52460b70c5cSStefano Zampini   if (tao->linesearch) PetscCall(TaoLineSearchSetFromOptions(tao->linesearch));
5253ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
526a7e14dcfSSatish Balay }
527a7e14dcfSSatish Balay 
528a7e14dcfSSatish Balay /*@C
52947450a7bSBarry Smith   TaoViewFromOptions - View a `Tao` object based on values in the options database
530fe2efc57SMark 
531c3339decSBarry Smith   Collective
532fe2efc57SMark 
533fe2efc57SMark   Input Parameters:
53447450a7bSBarry Smith + A    - the  `Tao` context
53547450a7bSBarry Smith . obj  - Optional object that provides the prefix for the options database
536736c3998SJose E. Roman - name - command line option
537fe2efc57SMark 
538fe2efc57SMark   Level: intermediate
53967be906fSBarry Smith 
5401cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoView`, `PetscObjectViewFromOptions()`, `TaoCreate()`
541fe2efc57SMark @*/
542d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoViewFromOptions(Tao A, PetscObject obj, const char name[])
543d71ae5a4SJacob Faibussowitsch {
544fe2efc57SMark   PetscFunctionBegin;
545fe2efc57SMark   PetscValidHeaderSpecific(A, TAO_CLASSID, 1);
5469566063dSJacob Faibussowitsch   PetscCall(PetscObjectViewFromOptions((PetscObject)A, obj, name));
5473ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
548fe2efc57SMark }
549fe2efc57SMark 
550fe2efc57SMark /*@C
55147450a7bSBarry Smith   TaoView - Prints information about the `Tao` object
552a7e14dcfSSatish Balay 
553c3339decSBarry Smith   Collective
554a7e14dcfSSatish Balay 
555a7e14dcfSSatish Balay   Input Parameters:
55647450a7bSBarry Smith + tao    - the `Tao` context
557a7e14dcfSSatish Balay - viewer - visualization context
558a7e14dcfSSatish Balay 
559a7e14dcfSSatish Balay   Options Database Key:
56065ba42b6SBarry Smith . -tao_view - Calls `TaoView()` at the end of `TaoSolve()`
561a7e14dcfSSatish Balay 
56267be906fSBarry Smith   Level: beginner
56367be906fSBarry Smith 
564a7e14dcfSSatish Balay   Notes:
565a7e14dcfSSatish Balay   The available visualization contexts include
56665ba42b6SBarry Smith +     `PETSC_VIEWER_STDOUT_SELF` - standard output (default)
56765ba42b6SBarry Smith -     `PETSC_VIEWER_STDOUT_WORLD` - synchronized standard
568a7e14dcfSSatish Balay   output where only the first processor opens
569a7e14dcfSSatish Balay   the file.  All other processors send their
570a7e14dcfSSatish Balay   data to the first processor to print.
571a7e14dcfSSatish Balay 
5721cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `PetscViewerASCIIOpen()`
573a7e14dcfSSatish Balay @*/
574d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoView(Tao tao, PetscViewer viewer)
575d71ae5a4SJacob Faibussowitsch {
576a7e14dcfSSatish Balay   PetscBool isascii, isstring;
577b625d6c7SJed Brown   TaoType   type;
57847a47007SBarry Smith 
579a7e14dcfSSatish Balay   PetscFunctionBegin;
580441846f8SBarry Smith   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
58148a46eb9SPierre Jolivet   if (!viewer) PetscCall(PetscViewerASCIIGetStdout(((PetscObject)tao)->comm, &viewer));
582a7e14dcfSSatish Balay   PetscValidHeaderSpecific(viewer, PETSC_VIEWER_CLASSID, 2);
583a7e14dcfSSatish Balay   PetscCheckSameComm(tao, 1, viewer, 2);
584a7e14dcfSSatish Balay 
5859566063dSJacob Faibussowitsch   PetscCall(PetscObjectTypeCompare((PetscObject)viewer, PETSCVIEWERASCII, &isascii));
5869566063dSJacob Faibussowitsch   PetscCall(PetscObjectTypeCompare((PetscObject)viewer, PETSCVIEWERSTRING, &isstring));
587a7e14dcfSSatish Balay   if (isascii) {
5889566063dSJacob Faibussowitsch     PetscCall(PetscObjectPrintClassNamePrefixType((PetscObject)tao, viewer));
589a7e14dcfSSatish Balay 
590a7e14dcfSSatish Balay     if (tao->ops->view) {
5919566063dSJacob Faibussowitsch       PetscCall(PetscViewerASCIIPushTab(viewer));
592dbbe0bcdSBarry Smith       PetscUseTypeMethod(tao, view, viewer);
5939566063dSJacob Faibussowitsch       PetscCall(PetscViewerASCIIPopTab(viewer));
594a7e14dcfSSatish Balay     }
5958d3f3ddaSAlp Dener     if (tao->linesearch) {
5969566063dSJacob Faibussowitsch       PetscCall(PetscViewerASCIIPushTab(viewer));
5979566063dSJacob Faibussowitsch       PetscCall(TaoLineSearchView(tao->linesearch, viewer));
5989566063dSJacob Faibussowitsch       PetscCall(PetscViewerASCIIPopTab(viewer));
5998d3f3ddaSAlp Dener     }
600a7e14dcfSSatish Balay     if (tao->ksp) {
6019566063dSJacob Faibussowitsch       PetscCall(PetscViewerASCIIPushTab(viewer));
6029566063dSJacob Faibussowitsch       PetscCall(KSPView(tao->ksp, viewer));
60363a3b9bcSJacob Faibussowitsch       PetscCall(PetscViewerASCIIPrintf(viewer, "total KSP iterations: %" PetscInt_FMT "\n", tao->ksp_tot_its));
6049566063dSJacob Faibussowitsch       PetscCall(PetscViewerASCIIPopTab(viewer));
605a7e14dcfSSatish Balay     }
60619b5c101SAlp Dener 
6079566063dSJacob Faibussowitsch     PetscCall(PetscViewerASCIIPushTab(viewer));
60819b5c101SAlp Dener 
60948a46eb9SPierre Jolivet     if (tao->XL || tao->XU) PetscCall(PetscViewerASCIIPrintf(viewer, "Active Set subset type: %s\n", TaoSubSetTypes[tao->subset_type]));
610a7e14dcfSSatish Balay 
6119566063dSJacob Faibussowitsch     PetscCall(PetscViewerASCIIPrintf(viewer, "convergence tolerances: gatol=%g,", (double)tao->gatol));
6129566063dSJacob Faibussowitsch     PetscCall(PetscViewerASCIIPrintf(viewer, " steptol=%g,", (double)tao->steptol));
6139566063dSJacob Faibussowitsch     PetscCall(PetscViewerASCIIPrintf(viewer, " gttol=%g\n", (double)tao->gttol));
6149566063dSJacob Faibussowitsch     PetscCall(PetscViewerASCIIPrintf(viewer, "Residual in Function/Gradient:=%g\n", (double)tao->residual));
615a7e14dcfSSatish Balay 
6166246e8cdSAlp Dener     if (tao->constrained) {
6179566063dSJacob Faibussowitsch       PetscCall(PetscViewerASCIIPrintf(viewer, "convergence tolerances:"));
6189566063dSJacob Faibussowitsch       PetscCall(PetscViewerASCIIPrintf(viewer, " catol=%g,", (double)tao->catol));
6199566063dSJacob Faibussowitsch       PetscCall(PetscViewerASCIIPrintf(viewer, " crtol=%g\n", (double)tao->crtol));
6209566063dSJacob Faibussowitsch       PetscCall(PetscViewerASCIIPrintf(viewer, "Residual in Constraints:=%g\n", (double)tao->cnorm));
621a7e14dcfSSatish Balay     }
622a7e14dcfSSatish Balay 
623a7e14dcfSSatish Balay     if (tao->trust < tao->steptol) {
6249566063dSJacob Faibussowitsch       PetscCall(PetscViewerASCIIPrintf(viewer, "convergence tolerances: steptol=%g\n", (double)tao->steptol));
6259566063dSJacob Faibussowitsch       PetscCall(PetscViewerASCIIPrintf(viewer, "Final trust region radius:=%g\n", (double)tao->trust));
626a7e14dcfSSatish Balay     }
627a7e14dcfSSatish Balay 
62848a46eb9SPierre Jolivet     if (tao->fmin > -1.e25) PetscCall(PetscViewerASCIIPrintf(viewer, "convergence tolerances: function minimum=%g\n", (double)tao->fmin));
6299566063dSJacob Faibussowitsch     PetscCall(PetscViewerASCIIPrintf(viewer, "Objective value=%g\n", (double)tao->fc));
630a7e14dcfSSatish Balay 
63163a3b9bcSJacob Faibussowitsch     PetscCall(PetscViewerASCIIPrintf(viewer, "total number of iterations=%" PetscInt_FMT ",          ", tao->niter));
63263a3b9bcSJacob Faibussowitsch     PetscCall(PetscViewerASCIIPrintf(viewer, "              (max: %" PetscInt_FMT ")\n", tao->max_it));
633a7e14dcfSSatish Balay 
634a7e14dcfSSatish Balay     if (tao->nfuncs > 0) {
63563a3b9bcSJacob Faibussowitsch       PetscCall(PetscViewerASCIIPrintf(viewer, "total number of function evaluations=%" PetscInt_FMT ",", tao->nfuncs));
63663a3b9bcSJacob Faibussowitsch       PetscCall(PetscViewerASCIIPrintf(viewer, "                max: %" PetscInt_FMT "\n", tao->max_funcs));
637a7e14dcfSSatish Balay     }
638a7e14dcfSSatish Balay     if (tao->ngrads > 0) {
63963a3b9bcSJacob Faibussowitsch       PetscCall(PetscViewerASCIIPrintf(viewer, "total number of gradient evaluations=%" PetscInt_FMT ",", tao->ngrads));
64063a3b9bcSJacob Faibussowitsch       PetscCall(PetscViewerASCIIPrintf(viewer, "                max: %" PetscInt_FMT "\n", tao->max_funcs));
641a7e14dcfSSatish Balay     }
642a7e14dcfSSatish Balay     if (tao->nfuncgrads > 0) {
64363a3b9bcSJacob Faibussowitsch       PetscCall(PetscViewerASCIIPrintf(viewer, "total number of function/gradient evaluations=%" PetscInt_FMT ",", tao->nfuncgrads));
64463a3b9bcSJacob Faibussowitsch       PetscCall(PetscViewerASCIIPrintf(viewer, "    (max: %" PetscInt_FMT ")\n", tao->max_funcs));
645a7e14dcfSSatish Balay     }
64648a46eb9SPierre Jolivet     if (tao->nhess > 0) PetscCall(PetscViewerASCIIPrintf(viewer, "total number of Hessian evaluations=%" PetscInt_FMT "\n", tao->nhess));
64748a46eb9SPierre Jolivet     if (tao->nconstraints > 0) PetscCall(PetscViewerASCIIPrintf(viewer, "total number of constraint function evaluations=%" PetscInt_FMT "\n", tao->nconstraints));
64848a46eb9SPierre Jolivet     if (tao->njac > 0) PetscCall(PetscViewerASCIIPrintf(viewer, "total number of Jacobian evaluations=%" PetscInt_FMT "\n", tao->njac));
649a7e14dcfSSatish Balay 
650a7e14dcfSSatish Balay     if (tao->reason > 0) {
6519566063dSJacob Faibussowitsch       PetscCall(PetscViewerASCIIPrintf(viewer, "Solution converged: "));
652a7e14dcfSSatish Balay       switch (tao->reason) {
653d71ae5a4SJacob Faibussowitsch       case TAO_CONVERGED_GATOL:
654d71ae5a4SJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, " ||g(X)|| <= gatol\n"));
655d71ae5a4SJacob Faibussowitsch         break;
656d71ae5a4SJacob Faibussowitsch       case TAO_CONVERGED_GRTOL:
657d71ae5a4SJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, " ||g(X)||/|f(X)| <= grtol\n"));
658d71ae5a4SJacob Faibussowitsch         break;
659d71ae5a4SJacob Faibussowitsch       case TAO_CONVERGED_GTTOL:
660d71ae5a4SJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, " ||g(X)||/||g(X0)|| <= gttol\n"));
661d71ae5a4SJacob Faibussowitsch         break;
662d71ae5a4SJacob Faibussowitsch       case TAO_CONVERGED_STEPTOL:
663d71ae5a4SJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, " Steptol -- step size small\n"));
664d71ae5a4SJacob Faibussowitsch         break;
665d71ae5a4SJacob Faibussowitsch       case TAO_CONVERGED_MINF:
666d71ae5a4SJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, " Minf --  f < fmin\n"));
667d71ae5a4SJacob Faibussowitsch         break;
668d71ae5a4SJacob Faibussowitsch       case TAO_CONVERGED_USER:
669d71ae5a4SJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, " User Terminated\n"));
670d71ae5a4SJacob Faibussowitsch         break;
671d71ae5a4SJacob Faibussowitsch       default:
672d71ae5a4SJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, "\n"));
673d71ae5a4SJacob Faibussowitsch         break;
674a7e14dcfSSatish Balay       }
675a7e14dcfSSatish Balay     } else {
6769566063dSJacob Faibussowitsch       PetscCall(PetscViewerASCIIPrintf(viewer, "Solver terminated: %d", tao->reason));
677a7e14dcfSSatish Balay       switch (tao->reason) {
678d71ae5a4SJacob Faibussowitsch       case TAO_DIVERGED_MAXITS:
679d71ae5a4SJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, " Maximum Iterations\n"));
680d71ae5a4SJacob Faibussowitsch         break;
681d71ae5a4SJacob Faibussowitsch       case TAO_DIVERGED_NAN:
682d71ae5a4SJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, " NAN or Inf encountered\n"));
683d71ae5a4SJacob Faibussowitsch         break;
684d71ae5a4SJacob Faibussowitsch       case TAO_DIVERGED_MAXFCN:
685d71ae5a4SJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, " Maximum Function Evaluations\n"));
686d71ae5a4SJacob Faibussowitsch         break;
687d71ae5a4SJacob Faibussowitsch       case TAO_DIVERGED_LS_FAILURE:
688d71ae5a4SJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, " Line Search Failure\n"));
689d71ae5a4SJacob Faibussowitsch         break;
690d71ae5a4SJacob Faibussowitsch       case TAO_DIVERGED_TR_REDUCTION:
691d71ae5a4SJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, " Trust Region too small\n"));
692d71ae5a4SJacob Faibussowitsch         break;
693d71ae5a4SJacob Faibussowitsch       case TAO_DIVERGED_USER:
694d71ae5a4SJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, " User Terminated\n"));
695d71ae5a4SJacob Faibussowitsch         break;
696d71ae5a4SJacob Faibussowitsch       default:
697d71ae5a4SJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, "\n"));
698d71ae5a4SJacob Faibussowitsch         break;
699a7e14dcfSSatish Balay       }
700a7e14dcfSSatish Balay     }
7019566063dSJacob Faibussowitsch     PetscCall(PetscViewerASCIIPopTab(viewer));
702a7e14dcfSSatish Balay   } else if (isstring) {
7039566063dSJacob Faibussowitsch     PetscCall(TaoGetType(tao, &type));
7049566063dSJacob Faibussowitsch     PetscCall(PetscViewerStringSPrintf(viewer, " %-3.3s", type));
705a7e14dcfSSatish Balay   }
7063ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
707a7e14dcfSSatish Balay }
708a7e14dcfSSatish Balay 
709a7e14dcfSSatish Balay /*@
710414d97d3SAlp Dener   TaoSetRecycleHistory - Sets the boolean flag to enable/disable re-using
71165ba42b6SBarry Smith   iterate information from the previous `TaoSolve()`. This feature is disabled by
712414d97d3SAlp Dener   default.
713414d97d3SAlp Dener 
71467be906fSBarry Smith   Logically Collective
71567be906fSBarry Smith 
71667be906fSBarry Smith   Input Parameters:
71747450a7bSBarry Smith + tao     - the `Tao` context
71867be906fSBarry Smith - recycle - boolean flag
71967be906fSBarry Smith 
72047450a7bSBarry Smith   Options Database Key:
72167be906fSBarry Smith . -tao_recycle_history <true,false> - reuse the history
72267be906fSBarry Smith 
72367be906fSBarry Smith   Level: intermediate
72467be906fSBarry Smith 
72567be906fSBarry Smith   Notes:
72665ba42b6SBarry Smith   For conjugate gradient methods (`TAOBNCG`), this re-uses the latest search direction
72765ba42b6SBarry Smith   from the previous `TaoSolve()` call when computing the first search direction in a
728414d97d3SAlp Dener   new solution. By default, CG methods set the first search direction to the
729414d97d3SAlp Dener   negative gradient.
730414d97d3SAlp Dener 
73165ba42b6SBarry Smith   For quasi-Newton family of methods (`TAOBQNLS`, `TAOBQNKLS`, `TAOBQNKTR`, `TAOBQNKTL`), this re-uses
73265ba42b6SBarry Smith   the accumulated quasi-Newton Hessian approximation from the previous `TaoSolve()`
733414d97d3SAlp Dener   call. By default, QN family of methods reset the initial Hessian approximation to
734414d97d3SAlp Dener   the identity matrix.
735414d97d3SAlp Dener 
736414d97d3SAlp Dener   For any other algorithm, this setting has no effect.
737414d97d3SAlp Dener 
7381cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoGetRecycleHistory()`, `TAOBNCG`, `TAOBQNLS`, `TAOBQNKLS`, `TAOBQNKTR`, `TAOBQNKTL`
739414d97d3SAlp Dener @*/
740d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoSetRecycleHistory(Tao tao, PetscBool recycle)
741d71ae5a4SJacob Faibussowitsch {
742414d97d3SAlp Dener   PetscFunctionBegin;
743414d97d3SAlp Dener   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
74494511df7SStefano Zampini   PetscValidLogicalCollectiveBool(tao, recycle, 2);
745414d97d3SAlp Dener   tao->recycle = recycle;
7463ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
747414d97d3SAlp Dener }
748414d97d3SAlp Dener 
749414d97d3SAlp Dener /*@
750414d97d3SAlp Dener   TaoGetRecycleHistory - Retrieve the boolean flag for re-using iterate information
75165ba42b6SBarry Smith   from the previous `TaoSolve()`. This feature is disabled by default.
752414d97d3SAlp Dener 
75367be906fSBarry Smith   Logically Collective
754414d97d3SAlp Dener 
75547450a7bSBarry Smith   Input Parameter:
75647450a7bSBarry Smith . tao - the `Tao` context
757414d97d3SAlp Dener 
75847450a7bSBarry Smith   Output Parameter:
759147403d9SBarry Smith . recycle - boolean flag
760414d97d3SAlp Dener 
761414d97d3SAlp Dener   Level: intermediate
762414d97d3SAlp Dener 
7631cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoSetRecycleHistory()`, `TAOBNCG`, `TAOBQNLS`, `TAOBQNKLS`, `TAOBQNKTR`, `TAOBQNKTL`
764414d97d3SAlp Dener @*/
765d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoGetRecycleHistory(Tao tao, PetscBool *recycle)
766d71ae5a4SJacob Faibussowitsch {
767414d97d3SAlp Dener   PetscFunctionBegin;
768414d97d3SAlp Dener   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
7694f572ea9SToby Isaac   PetscAssertPointer(recycle, 2);
770414d97d3SAlp Dener   *recycle = tao->recycle;
7713ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
772414d97d3SAlp Dener }
773414d97d3SAlp Dener 
774414d97d3SAlp Dener /*@
77547450a7bSBarry Smith   TaoSetTolerances - Sets parameters used in `TaoSolve()` convergence tests
776a7e14dcfSSatish Balay 
77767be906fSBarry Smith   Logically Collective
778a7e14dcfSSatish Balay 
779a7e14dcfSSatish Balay   Input Parameters:
78047450a7bSBarry Smith + tao   - the `Tao` context
781a7e14dcfSSatish Balay . gatol - stop if norm of gradient is less than this
782a7e14dcfSSatish Balay . grtol - stop if relative norm of gradient is less than this
783a7e14dcfSSatish Balay - gttol - stop if norm of gradient is reduced by this factor
784a7e14dcfSSatish Balay 
785a7e14dcfSSatish Balay   Options Database Keys:
786e52336cbSBarry Smith + -tao_gatol <gatol> - Sets gatol
787a7e14dcfSSatish Balay . -tao_grtol <grtol> - Sets grtol
788a7e14dcfSSatish Balay - -tao_gttol <gttol> - Sets gttol
789a7e14dcfSSatish Balay 
7903b242c63SJacob Faibussowitsch   Stopping Criteria\:
79167be906fSBarry Smith .vb
79267be906fSBarry Smith   ||g(X)||                            <= gatol
79367be906fSBarry Smith   ||g(X)|| / |f(X)|                   <= grtol
79467be906fSBarry Smith   ||g(X)|| / ||g(X0)||                <= gttol
79567be906fSBarry Smith .ve
796a7e14dcfSSatish Balay 
797a7e14dcfSSatish Balay   Level: beginner
798a7e14dcfSSatish Balay 
79967be906fSBarry Smith   Note:
80067be906fSBarry Smith   Use `PETSC_DEFAULT` to leave one or more tolerances unchanged.
801a7e14dcfSSatish Balay 
8021cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoConvergedReason`, `TaoGetTolerances()`
803a7e14dcfSSatish Balay @*/
804d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoSetTolerances(Tao tao, PetscReal gatol, PetscReal grtol, PetscReal gttol)
805d71ae5a4SJacob Faibussowitsch {
806a7e14dcfSSatish Balay   PetscFunctionBegin;
807441846f8SBarry Smith   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
80894511df7SStefano Zampini   PetscValidLogicalCollectiveReal(tao, gatol, 2);
80994511df7SStefano Zampini   PetscValidLogicalCollectiveReal(tao, grtol, 3);
81094511df7SStefano Zampini   PetscValidLogicalCollectiveReal(tao, gttol, 4);
811a7e14dcfSSatish Balay 
81213bcc0bdSJacob Faibussowitsch   if (gatol != (PetscReal)PETSC_DEFAULT) {
813a7e14dcfSSatish Balay     if (gatol < 0) {
8149566063dSJacob Faibussowitsch       PetscCall(PetscInfo(tao, "Tried to set negative gatol -- ignored.\n"));
815a7e14dcfSSatish Balay     } else {
816a7e14dcfSSatish Balay       tao->gatol         = PetscMax(0, gatol);
8176552cf8aSJason Sarich       tao->gatol_changed = PETSC_TRUE;
818a7e14dcfSSatish Balay     }
819a7e14dcfSSatish Balay   }
820a7e14dcfSSatish Balay 
82113bcc0bdSJacob Faibussowitsch   if (grtol != (PetscReal)PETSC_DEFAULT) {
822a7e14dcfSSatish Balay     if (grtol < 0) {
8239566063dSJacob Faibussowitsch       PetscCall(PetscInfo(tao, "Tried to set negative grtol -- ignored.\n"));
824a7e14dcfSSatish Balay     } else {
825a7e14dcfSSatish Balay       tao->grtol         = PetscMax(0, grtol);
8266552cf8aSJason Sarich       tao->grtol_changed = PETSC_TRUE;
827a7e14dcfSSatish Balay     }
828a7e14dcfSSatish Balay   }
829a7e14dcfSSatish Balay 
83013bcc0bdSJacob Faibussowitsch   if (gttol != (PetscReal)PETSC_DEFAULT) {
831a7e14dcfSSatish Balay     if (gttol < 0) {
8329566063dSJacob Faibussowitsch       PetscCall(PetscInfo(tao, "Tried to set negative gttol -- ignored.\n"));
833a7e14dcfSSatish Balay     } else {
834a7e14dcfSSatish Balay       tao->gttol         = PetscMax(0, gttol);
8356552cf8aSJason Sarich       tao->gttol_changed = PETSC_TRUE;
836a7e14dcfSSatish Balay     }
837a7e14dcfSSatish Balay   }
8383ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
839a7e14dcfSSatish Balay }
840a7e14dcfSSatish Balay 
841a7e14dcfSSatish Balay /*@
84247450a7bSBarry Smith   TaoSetConstraintTolerances - Sets constraint tolerance parameters used in `TaoSolve()` convergence tests
843a7e14dcfSSatish Balay 
84467be906fSBarry Smith   Logically Collective
845a7e14dcfSSatish Balay 
846a7e14dcfSSatish Balay   Input Parameters:
84747450a7bSBarry Smith + tao   - the `Tao` context
84867be906fSBarry Smith . catol - absolute constraint tolerance, constraint norm must be less than `catol` for used for gatol convergence criteria
84967be906fSBarry Smith - crtol - relative constraint tolerance, constraint norm must be less than `crtol` for used for gatol, gttol convergence criteria
850a7e14dcfSSatish Balay 
851a7e14dcfSSatish Balay   Options Database Keys:
852a7e14dcfSSatish Balay + -tao_catol <catol> - Sets catol
853a7e14dcfSSatish Balay - -tao_crtol <crtol> - Sets crtol
854a7e14dcfSSatish Balay 
855a7e14dcfSSatish Balay   Level: intermediate
856a7e14dcfSSatish Balay 
85767be906fSBarry Smith   Notes:
85867be906fSBarry Smith   Use `PETSC_DEFAULT` to leave any tolerance unchanged.
859a7e14dcfSSatish Balay 
8601cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoConvergedReason`, `TaoGetTolerances()`, `TaoGetConstraintTolerances()`, `TaoSetTolerances()`
861a7e14dcfSSatish Balay @*/
862d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoSetConstraintTolerances(Tao tao, PetscReal catol, PetscReal crtol)
863d71ae5a4SJacob Faibussowitsch {
864a7e14dcfSSatish Balay   PetscFunctionBegin;
865441846f8SBarry Smith   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
86694511df7SStefano Zampini   PetscValidLogicalCollectiveReal(tao, catol, 2);
86794511df7SStefano Zampini   PetscValidLogicalCollectiveReal(tao, crtol, 3);
868a7e14dcfSSatish Balay 
86913bcc0bdSJacob Faibussowitsch   if (catol != (PetscReal)PETSC_DEFAULT) {
870a7e14dcfSSatish Balay     if (catol < 0) {
8719566063dSJacob Faibussowitsch       PetscCall(PetscInfo(tao, "Tried to set negative catol -- ignored.\n"));
872a7e14dcfSSatish Balay     } else {
873a7e14dcfSSatish Balay       tao->catol         = PetscMax(0, catol);
8746552cf8aSJason Sarich       tao->catol_changed = PETSC_TRUE;
875a7e14dcfSSatish Balay     }
876a7e14dcfSSatish Balay   }
877a7e14dcfSSatish Balay 
87813bcc0bdSJacob Faibussowitsch   if (crtol != (PetscReal)PETSC_DEFAULT) {
879a7e14dcfSSatish Balay     if (crtol < 0) {
8809566063dSJacob Faibussowitsch       PetscCall(PetscInfo(tao, "Tried to set negative crtol -- ignored.\n"));
881a7e14dcfSSatish Balay     } else {
882a7e14dcfSSatish Balay       tao->crtol         = PetscMax(0, crtol);
8836552cf8aSJason Sarich       tao->crtol_changed = PETSC_TRUE;
884a7e14dcfSSatish Balay     }
885a7e14dcfSSatish Balay   }
8863ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
887a7e14dcfSSatish Balay }
888a7e14dcfSSatish Balay 
88958417fe7SBarry Smith /*@
89047450a7bSBarry Smith   TaoGetConstraintTolerances - Gets constraint tolerance parameters used in `TaoSolve()` convergence tests
89158417fe7SBarry Smith 
89267be906fSBarry Smith   Not Collective
89358417fe7SBarry Smith 
89458417fe7SBarry Smith   Input Parameter:
89547450a7bSBarry Smith . tao - the `Tao` context
89658417fe7SBarry Smith 
897d8d19677SJose E. Roman   Output Parameters:
89867be906fSBarry Smith + catol - absolute constraint tolerance, constraint norm must be less than `catol` for used for gatol convergence criteria
89967be906fSBarry Smith - crtol - relative constraint tolerance, constraint norm must be less than `crtol` for used for gatol, gttol convergence criteria
90058417fe7SBarry Smith 
90158417fe7SBarry Smith   Level: intermediate
90258417fe7SBarry Smith 
9031cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoConvergedReasons`,`TaoGetTolerances()`, `TaoSetTolerances()`, `TaoSetConstraintTolerances()`
90458417fe7SBarry Smith @*/
905d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoGetConstraintTolerances(Tao tao, PetscReal *catol, PetscReal *crtol)
906d71ae5a4SJacob Faibussowitsch {
90758417fe7SBarry Smith   PetscFunctionBegin;
90858417fe7SBarry Smith   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
90958417fe7SBarry Smith   if (catol) *catol = tao->catol;
91058417fe7SBarry Smith   if (crtol) *crtol = tao->crtol;
9113ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
91258417fe7SBarry Smith }
91358417fe7SBarry Smith 
914a7e14dcfSSatish Balay /*@
915a7e14dcfSSatish Balay   TaoSetFunctionLowerBound - Sets a bound on the solution objective value.
916a7e14dcfSSatish Balay   When an approximate solution with an objective value below this number
917a7e14dcfSSatish Balay   has been found, the solver will terminate.
918a7e14dcfSSatish Balay 
919c3339decSBarry Smith   Logically Collective
920a7e14dcfSSatish Balay 
921a7e14dcfSSatish Balay   Input Parameters:
922441846f8SBarry Smith + tao  - the Tao solver context
923a7e14dcfSSatish Balay - fmin - the tolerance
924a7e14dcfSSatish Balay 
92547450a7bSBarry Smith   Options Database Key:
926a7e14dcfSSatish Balay . -tao_fmin <fmin> - sets the minimum function value
927a7e14dcfSSatish Balay 
928a7e14dcfSSatish Balay   Level: intermediate
929a7e14dcfSSatish Balay 
9301cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoConvergedReason`, `TaoSetTolerances()`
931a7e14dcfSSatish Balay @*/
932d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoSetFunctionLowerBound(Tao tao, PetscReal fmin)
933d71ae5a4SJacob Faibussowitsch {
934a7e14dcfSSatish Balay   PetscFunctionBegin;
935441846f8SBarry Smith   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
93694511df7SStefano Zampini   PetscValidLogicalCollectiveReal(tao, fmin, 2);
937a7e14dcfSSatish Balay   tao->fmin         = fmin;
9386552cf8aSJason Sarich   tao->fmin_changed = PETSC_TRUE;
9393ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
940a7e14dcfSSatish Balay }
941a7e14dcfSSatish Balay 
942a7e14dcfSSatish Balay /*@
9438e96d397SJason Sarich   TaoGetFunctionLowerBound - Gets the bound on the solution objective value.
944a7e14dcfSSatish Balay   When an approximate solution with an objective value below this number
945a7e14dcfSSatish Balay   has been found, the solver will terminate.
946a7e14dcfSSatish Balay 
94767be906fSBarry Smith   Not Collective
948a7e14dcfSSatish Balay 
94947450a7bSBarry Smith   Input Parameter:
95047450a7bSBarry Smith . tao - the `Tao` solver context
951a7e14dcfSSatish Balay 
95247450a7bSBarry Smith   Output Parameter:
953a7e14dcfSSatish Balay . fmin - the minimum function value
954a7e14dcfSSatish Balay 
955a7e14dcfSSatish Balay   Level: intermediate
956a7e14dcfSSatish Balay 
9571cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoConvergedReason`, `TaoSetFunctionLowerBound()`
958a7e14dcfSSatish Balay @*/
959d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoGetFunctionLowerBound(Tao tao, PetscReal *fmin)
960d71ae5a4SJacob Faibussowitsch {
961a7e14dcfSSatish Balay   PetscFunctionBegin;
962441846f8SBarry Smith   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
9634f572ea9SToby Isaac   PetscAssertPointer(fmin, 2);
964a7e14dcfSSatish Balay   *fmin = tao->fmin;
9653ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
966a7e14dcfSSatish Balay }
967a7e14dcfSSatish Balay 
968a7e14dcfSSatish Balay /*@
96947450a7bSBarry Smith   TaoSetMaximumFunctionEvaluations - Sets a maximum number of function evaluations allowed for a `TaoSolve()`.
970a7e14dcfSSatish Balay 
971c3339decSBarry Smith   Logically Collective
972a7e14dcfSSatish Balay 
973a7e14dcfSSatish Balay   Input Parameters:
97447450a7bSBarry Smith + tao  - the `Tao` solver context
975a7e14dcfSSatish Balay - nfcn - the maximum number of function evaluations (>=0)
976a7e14dcfSSatish Balay 
97747450a7bSBarry Smith   Options Database Key:
978a7e14dcfSSatish Balay . -tao_max_funcs <nfcn> - sets the maximum number of function evaluations
979a7e14dcfSSatish Balay 
980a7e14dcfSSatish Balay   Level: intermediate
981a7e14dcfSSatish Balay 
9821cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoSetTolerances()`, `TaoSetMaximumIterations()`
983a7e14dcfSSatish Balay @*/
984d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoSetMaximumFunctionEvaluations(Tao tao, PetscInt nfcn)
985d71ae5a4SJacob Faibussowitsch {
986a7e14dcfSSatish Balay   PetscFunctionBegin;
987441846f8SBarry Smith   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
98894511df7SStefano Zampini   PetscValidLogicalCollectiveInt(tao, nfcn, 2);
9899371c9d4SSatish Balay   if (nfcn >= 0) {
9909371c9d4SSatish Balay     tao->max_funcs = PetscMax(0, nfcn);
9919371c9d4SSatish Balay   } else {
9929371c9d4SSatish Balay     tao->max_funcs = -1;
9939371c9d4SSatish Balay   }
9946552cf8aSJason Sarich   tao->max_funcs_changed = PETSC_TRUE;
9953ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
996a7e14dcfSSatish Balay }
997a7e14dcfSSatish Balay 
998a7e14dcfSSatish Balay /*@
99947450a7bSBarry Smith   TaoGetMaximumFunctionEvaluations - Gets a maximum number of function evaluations allowed for a `TaoSolve()`
1000a7e14dcfSSatish Balay 
1001c3339decSBarry Smith   Logically Collective
1002a7e14dcfSSatish Balay 
100347450a7bSBarry Smith   Input Parameter:
100447450a7bSBarry Smith . tao - the `Tao` solver context
1005a7e14dcfSSatish Balay 
100647450a7bSBarry Smith   Output Parameter:
1007a7e14dcfSSatish Balay . nfcn - the maximum number of function evaluations
1008a7e14dcfSSatish Balay 
1009a7e14dcfSSatish Balay   Level: intermediate
1010a7e14dcfSSatish Balay 
10111cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoSetMaximumFunctionEvaluations()`, `TaoGetMaximumIterations()`
1012a7e14dcfSSatish Balay @*/
1013d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoGetMaximumFunctionEvaluations(Tao tao, PetscInt *nfcn)
1014d71ae5a4SJacob Faibussowitsch {
1015a7e14dcfSSatish Balay   PetscFunctionBegin;
1016441846f8SBarry Smith   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
10174f572ea9SToby Isaac   PetscAssertPointer(nfcn, 2);
1018a7e14dcfSSatish Balay   *nfcn = tao->max_funcs;
10193ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
1020a7e14dcfSSatish Balay }
1021a7e14dcfSSatish Balay 
1022770232b9SCe Qin /*@
102347450a7bSBarry Smith   TaoGetCurrentFunctionEvaluations - Get current number of function evaluations used by a `Tao` object
1024770232b9SCe Qin 
1025770232b9SCe Qin   Not Collective
1026770232b9SCe Qin 
102747450a7bSBarry Smith   Input Parameter:
102847450a7bSBarry Smith . tao - the `Tao` solver context
1029770232b9SCe Qin 
103047450a7bSBarry Smith   Output Parameter:
103194511df7SStefano Zampini . nfuncs - the current number of function evaluations (maximum between gradient and function evaluations)
1032770232b9SCe Qin 
1033770232b9SCe Qin   Level: intermediate
1034770232b9SCe Qin 
10351cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoSetMaximumFunctionEvaluations()`, `TaoGetMaximumFunctionEvaluations()`, `TaoGetMaximumIterations()`
1036770232b9SCe Qin @*/
1037d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoGetCurrentFunctionEvaluations(Tao tao, PetscInt *nfuncs)
1038d71ae5a4SJacob Faibussowitsch {
1039770232b9SCe Qin   PetscFunctionBegin;
1040770232b9SCe Qin   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
10414f572ea9SToby Isaac   PetscAssertPointer(nfuncs, 2);
1042770232b9SCe Qin   *nfuncs = PetscMax(tao->nfuncs, tao->nfuncgrads);
10433ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
1044770232b9SCe Qin }
1045770232b9SCe Qin 
1046a7e14dcfSSatish Balay /*@
104747450a7bSBarry Smith   TaoSetMaximumIterations - Sets a maximum number of iterates to be used in `TaoSolve()`
1048a7e14dcfSSatish Balay 
1049c3339decSBarry Smith   Logically Collective
1050a7e14dcfSSatish Balay 
1051a7e14dcfSSatish Balay   Input Parameters:
105247450a7bSBarry Smith + tao    - the `Tao` solver context
1053a7e14dcfSSatish Balay - maxits - the maximum number of iterates (>=0)
1054a7e14dcfSSatish Balay 
105547450a7bSBarry Smith   Options Database Key:
1056a7e14dcfSSatish Balay . -tao_max_it <its> - sets the maximum number of iterations
1057a7e14dcfSSatish Balay 
1058a7e14dcfSSatish Balay   Level: intermediate
1059a7e14dcfSSatish Balay 
10601cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoSetTolerances()`, `TaoSetMaximumFunctionEvaluations()`
1061a7e14dcfSSatish Balay @*/
1062d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoSetMaximumIterations(Tao tao, PetscInt maxits)
1063d71ae5a4SJacob Faibussowitsch {
1064a7e14dcfSSatish Balay   PetscFunctionBegin;
1065441846f8SBarry Smith   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
106694511df7SStefano Zampini   PetscValidLogicalCollectiveInt(tao, maxits, 2);
106747a47007SBarry Smith   tao->max_it         = PetscMax(0, maxits);
10686552cf8aSJason Sarich   tao->max_it_changed = PETSC_TRUE;
10693ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
1070a7e14dcfSSatish Balay }
1071a7e14dcfSSatish Balay 
1072a7e14dcfSSatish Balay /*@
107365ba42b6SBarry Smith   TaoGetMaximumIterations - Gets a maximum number of iterates that will be used
1074a7e14dcfSSatish Balay 
1075a7e14dcfSSatish Balay   Not Collective
1076a7e14dcfSSatish Balay 
107747450a7bSBarry Smith   Input Parameter:
107847450a7bSBarry Smith . tao - the `Tao` solver context
1079a7e14dcfSSatish Balay 
108047450a7bSBarry Smith   Output Parameter:
1081a7e14dcfSSatish Balay . maxits - the maximum number of iterates
1082a7e14dcfSSatish Balay 
1083a7e14dcfSSatish Balay   Level: intermediate
1084a7e14dcfSSatish Balay 
10851cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoSetMaximumIterations()`, `TaoGetMaximumFunctionEvaluations()`
1086a7e14dcfSSatish Balay @*/
1087d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoGetMaximumIterations(Tao tao, PetscInt *maxits)
1088d71ae5a4SJacob Faibussowitsch {
1089a7e14dcfSSatish Balay   PetscFunctionBegin;
1090441846f8SBarry Smith   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
10914f572ea9SToby Isaac   PetscAssertPointer(maxits, 2);
1092a7e14dcfSSatish Balay   *maxits = tao->max_it;
10933ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
1094a7e14dcfSSatish Balay }
1095a7e14dcfSSatish Balay 
1096a7e14dcfSSatish Balay /*@
1097a7e14dcfSSatish Balay   TaoSetInitialTrustRegionRadius - Sets the initial trust region radius.
1098a7e14dcfSSatish Balay 
109967be906fSBarry Smith   Logically Collective
1100a7e14dcfSSatish Balay 
1101d8d19677SJose E. Roman   Input Parameters:
110247450a7bSBarry Smith + tao    - a `Tao` optimization solver
1103a7e14dcfSSatish Balay - radius - the trust region radius
1104a7e14dcfSSatish Balay 
1105a7e14dcfSSatish Balay   Options Database Key:
1106a7e14dcfSSatish Balay . -tao_trust0 <t0> - sets initial trust region radius
1107a7e14dcfSSatish Balay 
110847450a7bSBarry Smith   Level: intermediate
110947450a7bSBarry Smith 
11101cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoGetTrustRegionRadius()`, `TaoSetTrustRegionTolerance()`, `TAONTR`
1111a7e14dcfSSatish Balay @*/
1112d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoSetInitialTrustRegionRadius(Tao tao, PetscReal radius)
1113d71ae5a4SJacob Faibussowitsch {
1114a7e14dcfSSatish Balay   PetscFunctionBegin;
1115441846f8SBarry Smith   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
111694511df7SStefano Zampini   PetscValidLogicalCollectiveReal(tao, radius, 2);
1117a7e14dcfSSatish Balay   tao->trust0         = PetscMax(0.0, radius);
11186552cf8aSJason Sarich   tao->trust0_changed = PETSC_TRUE;
11193ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
1120a7e14dcfSSatish Balay }
1121a7e14dcfSSatish Balay 
1122a7e14dcfSSatish Balay /*@
112365ba42b6SBarry Smith   TaoGetInitialTrustRegionRadius - Gets the initial trust region radius.
1124a7e14dcfSSatish Balay 
1125a7e14dcfSSatish Balay   Not Collective
1126a7e14dcfSSatish Balay 
1127a7e14dcfSSatish Balay   Input Parameter:
112847450a7bSBarry Smith . tao - a `Tao` optimization solver
1129a7e14dcfSSatish Balay 
1130a7e14dcfSSatish Balay   Output Parameter:
1131a7e14dcfSSatish Balay . radius - the trust region radius
1132a7e14dcfSSatish Balay 
1133a7e14dcfSSatish Balay   Level: intermediate
1134a7e14dcfSSatish Balay 
11351cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoSetInitialTrustRegionRadius()`, `TaoGetCurrentTrustRegionRadius()`, `TAONTR`
1136a7e14dcfSSatish Balay @*/
1137d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoGetInitialTrustRegionRadius(Tao tao, PetscReal *radius)
1138d71ae5a4SJacob Faibussowitsch {
1139a7e14dcfSSatish Balay   PetscFunctionBegin;
1140441846f8SBarry Smith   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
11414f572ea9SToby Isaac   PetscAssertPointer(radius, 2);
1142a7e14dcfSSatish Balay   *radius = tao->trust0;
11433ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
1144a7e14dcfSSatish Balay }
1145a7e14dcfSSatish Balay 
1146a7e14dcfSSatish Balay /*@
1147a7e14dcfSSatish Balay   TaoGetCurrentTrustRegionRadius - Gets the current trust region radius.
1148a7e14dcfSSatish Balay 
1149a7e14dcfSSatish Balay   Not Collective
1150a7e14dcfSSatish Balay 
1151a7e14dcfSSatish Balay   Input Parameter:
115247450a7bSBarry Smith . tao - a `Tao` optimization solver
1153a7e14dcfSSatish Balay 
1154a7e14dcfSSatish Balay   Output Parameter:
1155a7e14dcfSSatish Balay . radius - the trust region radius
1156a7e14dcfSSatish Balay 
1157a7e14dcfSSatish Balay   Level: intermediate
1158a7e14dcfSSatish Balay 
11591cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoSetInitialTrustRegionRadius()`, `TaoGetInitialTrustRegionRadius()`, `TAONTR`
1160a7e14dcfSSatish Balay @*/
1161d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoGetCurrentTrustRegionRadius(Tao tao, PetscReal *radius)
1162d71ae5a4SJacob Faibussowitsch {
1163a7e14dcfSSatish Balay   PetscFunctionBegin;
1164441846f8SBarry Smith   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
11654f572ea9SToby Isaac   PetscAssertPointer(radius, 2);
1166a7e14dcfSSatish Balay   *radius = tao->trust;
11673ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
1168a7e14dcfSSatish Balay }
1169a7e14dcfSSatish Balay 
1170a7e14dcfSSatish Balay /*@
117147450a7bSBarry Smith   TaoGetTolerances - gets the current values of some tolerances used for the convergence testing of `TaoSolve()`
1172a7e14dcfSSatish Balay 
1173a7e14dcfSSatish Balay   Not Collective
1174a7e14dcfSSatish Balay 
1175f899ff85SJose E. Roman   Input Parameter:
117647450a7bSBarry Smith . tao - the `Tao` context
1177a7e14dcfSSatish Balay 
1178a7e14dcfSSatish Balay   Output Parameters:
1179e52336cbSBarry Smith + gatol - stop if norm of gradient is less than this
1180a7e14dcfSSatish Balay . grtol - stop if relative norm of gradient is less than this
1181a7e14dcfSSatish Balay - gttol - stop if norm of gradient is reduced by a this factor
1182a7e14dcfSSatish Balay 
1183a7e14dcfSSatish Balay   Level: intermediate
1184a1cb98faSBarry Smith 
1185a1cb98faSBarry Smith   Note:
118647450a7bSBarry Smith   `NULL` can be used as an argument if not all tolerances values are needed
1187a1cb98faSBarry Smith 
11881cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoSetTolerances()`
1189a7e14dcfSSatish Balay @*/
1190d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoGetTolerances(Tao tao, PetscReal *gatol, PetscReal *grtol, PetscReal *gttol)
1191d71ae5a4SJacob Faibussowitsch {
1192a7e14dcfSSatish Balay   PetscFunctionBegin;
1193441846f8SBarry Smith   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
1194a7e14dcfSSatish Balay   if (gatol) *gatol = tao->gatol;
1195a7e14dcfSSatish Balay   if (grtol) *grtol = tao->grtol;
1196a7e14dcfSSatish Balay   if (gttol) *gttol = tao->gttol;
11973ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
1198a7e14dcfSSatish Balay }
1199a7e14dcfSSatish Balay 
1200a7e14dcfSSatish Balay /*@
1201a7e14dcfSSatish Balay   TaoGetKSP - Gets the linear solver used by the optimization solver.
1202a7e14dcfSSatish Balay 
1203a7e14dcfSSatish Balay   Not Collective
1204a7e14dcfSSatish Balay 
120547450a7bSBarry Smith   Input Parameter:
120647450a7bSBarry Smith . tao - the `Tao` solver
1207a7e14dcfSSatish Balay 
120847450a7bSBarry Smith   Output Parameter:
120947450a7bSBarry Smith . ksp - the `KSP` linear solver used in the optimization solver
1210a7e14dcfSSatish Balay 
1211a7e14dcfSSatish Balay   Level: intermediate
1212a7e14dcfSSatish Balay 
12131cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `KSP`
1214a7e14dcfSSatish Balay @*/
1215d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoGetKSP(Tao tao, KSP *ksp)
1216d71ae5a4SJacob Faibussowitsch {
1217a7e14dcfSSatish Balay   PetscFunctionBegin;
121894511df7SStefano Zampini   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
12194f572ea9SToby Isaac   PetscAssertPointer(ksp, 2);
1220a7e14dcfSSatish Balay   *ksp = tao->ksp;
12213ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
1222a7e14dcfSSatish Balay }
1223a7e14dcfSSatish Balay 
1224025e9500SJason Sarich /*@
1225025e9500SJason Sarich   TaoGetLinearSolveIterations - Gets the total number of linear iterations
122647450a7bSBarry Smith   used by the `Tao` solver
1227025e9500SJason Sarich 
1228025e9500SJason Sarich   Not Collective
1229025e9500SJason Sarich 
1230025e9500SJason Sarich   Input Parameter:
123147450a7bSBarry Smith . tao - the `Tao` context
1232025e9500SJason Sarich 
1233025e9500SJason Sarich   Output Parameter:
1234025e9500SJason Sarich . lits - number of linear iterations
1235025e9500SJason Sarich 
1236025e9500SJason Sarich   Level: intermediate
1237025e9500SJason Sarich 
123847450a7bSBarry Smith   Note:
123947450a7bSBarry Smith   This counter is reset to zero for each successive call to `TaoSolve()`
124047450a7bSBarry Smith 
12411cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoGetKSP()`
1242025e9500SJason Sarich @*/
1243d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoGetLinearSolveIterations(Tao tao, PetscInt *lits)
1244d71ae5a4SJacob Faibussowitsch {
1245025e9500SJason Sarich   PetscFunctionBegin;
1246025e9500SJason Sarich   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
12474f572ea9SToby Isaac   PetscAssertPointer(lits, 2);
12482d9aa51bSJason Sarich   *lits = tao->ksp_tot_its;
12493ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
1250025e9500SJason Sarich }
1251025e9500SJason Sarich 
1252a7e14dcfSSatish Balay /*@
1253a7e14dcfSSatish Balay   TaoGetLineSearch - Gets the line search used by the optimization solver.
1254a7e14dcfSSatish Balay 
1255a7e14dcfSSatish Balay   Not Collective
1256a7e14dcfSSatish Balay 
125747450a7bSBarry Smith   Input Parameter:
125847450a7bSBarry Smith . tao - the `Tao` solver
1259a7e14dcfSSatish Balay 
126047450a7bSBarry Smith   Output Parameter:
1261a7e14dcfSSatish Balay . ls - the line search used in the optimization solver
1262a7e14dcfSSatish Balay 
1263a7e14dcfSSatish Balay   Level: intermediate
1264a7e14dcfSSatish Balay 
12651cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoLineSearch`, `TaoLineSearchType`
1266a7e14dcfSSatish Balay @*/
1267d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoGetLineSearch(Tao tao, TaoLineSearch *ls)
1268d71ae5a4SJacob Faibussowitsch {
1269a7e14dcfSSatish Balay   PetscFunctionBegin;
127094511df7SStefano Zampini   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
12714f572ea9SToby Isaac   PetscAssertPointer(ls, 2);
1272a7e14dcfSSatish Balay   *ls = tao->linesearch;
12733ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
1274a7e14dcfSSatish Balay }
1275a7e14dcfSSatish Balay 
1276a7e14dcfSSatish Balay /*@
1277a7e14dcfSSatish Balay   TaoAddLineSearchCounts - Adds the number of function evaluations spent
1278a7e14dcfSSatish Balay   in the line search to the running total.
1279a7e14dcfSSatish Balay 
1280a7e14dcfSSatish Balay   Input Parameters:
12812fe279fdSBarry Smith . tao - the `Tao` solver
1282a7e14dcfSSatish Balay 
1283a7e14dcfSSatish Balay   Level: developer
1284a7e14dcfSSatish Balay 
12851cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoGetLineSearch()`, `TaoLineSearchApply()`
1286a7e14dcfSSatish Balay @*/
1287d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoAddLineSearchCounts(Tao tao)
1288d71ae5a4SJacob Faibussowitsch {
1289a7e14dcfSSatish Balay   PetscBool flg;
1290a7e14dcfSSatish Balay   PetscInt  nfeval, ngeval, nfgeval;
129147a47007SBarry Smith 
1292a7e14dcfSSatish Balay   PetscFunctionBegin;
1293441846f8SBarry Smith   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
1294a7e14dcfSSatish Balay   if (tao->linesearch) {
12959566063dSJacob Faibussowitsch     PetscCall(TaoLineSearchIsUsingTaoRoutines(tao->linesearch, &flg));
129694ae4db5SBarry Smith     if (!flg) {
12979566063dSJacob Faibussowitsch       PetscCall(TaoLineSearchGetNumberFunctionEvaluations(tao->linesearch, &nfeval, &ngeval, &nfgeval));
1298a7e14dcfSSatish Balay       tao->nfuncs += nfeval;
1299a7e14dcfSSatish Balay       tao->ngrads += ngeval;
1300a7e14dcfSSatish Balay       tao->nfuncgrads += nfgeval;
1301a7e14dcfSSatish Balay     }
1302a7e14dcfSSatish Balay   }
13033ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
1304a7e14dcfSSatish Balay }
1305a7e14dcfSSatish Balay 
1306a7e14dcfSSatish Balay /*@
130747450a7bSBarry Smith   TaoGetSolution - Returns the vector with the current solution from the `Tao` object
1308a7e14dcfSSatish Balay 
1309a7e14dcfSSatish Balay   Not Collective
1310a7e14dcfSSatish Balay 
1311a7e14dcfSSatish Balay   Input Parameter:
131247450a7bSBarry Smith . tao - the `Tao` context
1313a7e14dcfSSatish Balay 
1314a7e14dcfSSatish Balay   Output Parameter:
1315a7e14dcfSSatish Balay . X - the current solution
1316a7e14dcfSSatish Balay 
1317a7e14dcfSSatish Balay   Level: intermediate
1318a7e14dcfSSatish Balay 
131965ba42b6SBarry Smith   Note:
132065ba42b6SBarry Smith   The returned vector will be the same object that was passed into `TaoSetSolution()`
132165ba42b6SBarry Smith 
13221cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoSetSolution()`, `TaoSolve()`
1323a7e14dcfSSatish Balay @*/
1324d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoGetSolution(Tao tao, Vec *X)
1325d71ae5a4SJacob Faibussowitsch {
1326a7e14dcfSSatish Balay   PetscFunctionBegin;
1327441846f8SBarry Smith   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
13284f572ea9SToby Isaac   PetscAssertPointer(X, 2);
1329a7e14dcfSSatish Balay   *X = tao->solution;
13303ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
1331a7e14dcfSSatish Balay }
1332a7e14dcfSSatish Balay 
1333a7e14dcfSSatish Balay /*@
133447450a7bSBarry Smith   TaoResetStatistics - Initialize the statistics collected by the `Tao` object.
1335a7e14dcfSSatish Balay   These statistics include the iteration number, residual norms, and convergence status.
1336a7e14dcfSSatish Balay   This routine gets called before solving each optimization problem.
1337a7e14dcfSSatish Balay 
1338c3339decSBarry Smith   Collective
1339a7e14dcfSSatish Balay 
134047450a7bSBarry Smith   Input Parameter:
1341e056e8ceSJacob Faibussowitsch . tao - the `Tao` context
1342a7e14dcfSSatish Balay 
1343a7e14dcfSSatish Balay   Level: developer
1344a7e14dcfSSatish Balay 
13451cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoCreate()`, `TaoSolve()`
1346a7e14dcfSSatish Balay @*/
1347d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoResetStatistics(Tao tao)
1348d71ae5a4SJacob Faibussowitsch {
1349a7e14dcfSSatish Balay   PetscFunctionBegin;
1350441846f8SBarry Smith   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
1351a7e14dcfSSatish Balay   tao->niter        = 0;
1352a7e14dcfSSatish Balay   tao->nfuncs       = 0;
1353a7e14dcfSSatish Balay   tao->nfuncgrads   = 0;
1354a7e14dcfSSatish Balay   tao->ngrads       = 0;
1355a7e14dcfSSatish Balay   tao->nhess        = 0;
1356a7e14dcfSSatish Balay   tao->njac         = 0;
1357a7e14dcfSSatish Balay   tao->nconstraints = 0;
1358a7e14dcfSSatish Balay   tao->ksp_its      = 0;
1359ae93cb3cSJason Sarich   tao->ksp_tot_its  = 0;
1360a7e14dcfSSatish Balay   tao->reason       = TAO_CONTINUE_ITERATING;
1361a7e14dcfSSatish Balay   tao->residual     = 0.0;
1362a7e14dcfSSatish Balay   tao->cnorm        = 0.0;
1363a7e14dcfSSatish Balay   tao->step         = 0.0;
1364a7e14dcfSSatish Balay   tao->lsflag       = PETSC_FALSE;
1365a7e14dcfSSatish Balay   if (tao->hist_reset) tao->hist_len = 0;
13663ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
1367a7e14dcfSSatish Balay }
1368a7e14dcfSSatish Balay 
1369a7e14dcfSSatish Balay /*@C
1370e1e80dc8SAlp Dener   TaoSetUpdate - Sets the general-purpose update function called
13712fe279fdSBarry Smith   at the beginning of every iteration of the optimization algorithm. Called after the new solution and the gradient
1372e1e80dc8SAlp Dener   is determined, but before the Hessian is computed (if applicable).
1373e1e80dc8SAlp Dener 
1374c3339decSBarry Smith   Logically Collective
1375e1e80dc8SAlp Dener 
1376e1e80dc8SAlp Dener   Input Parameters:
13772fe279fdSBarry Smith + tao  - The `Tao` solver context
1378a2b725a8SWilliam Gropp - func - The function
1379e1e80dc8SAlp Dener 
138020f4b53cSBarry Smith   Calling sequence of `func`:
138120f4b53cSBarry Smith + tao - the optimizer context
1382e056e8ceSJacob Faibussowitsch - ctx - The current step of the iteration
1383e1e80dc8SAlp Dener 
1384e1e80dc8SAlp Dener   Level: advanced
1385e1e80dc8SAlp Dener 
13861cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoSolve()`
1387e1e80dc8SAlp Dener @*/
1388d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoSetUpdate(Tao tao, PetscErrorCode (*func)(Tao, PetscInt, void *), void *ctx)
1389d71ae5a4SJacob Faibussowitsch {
1390e1e80dc8SAlp Dener   PetscFunctionBegin;
1391e1e80dc8SAlp Dener   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
1392e1e80dc8SAlp Dener   tao->ops->update = func;
1393e1e80dc8SAlp Dener   tao->user_update = ctx;
13943ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
1395e1e80dc8SAlp Dener }
1396e1e80dc8SAlp Dener 
1397e1e80dc8SAlp Dener /*@C
1398a7e14dcfSSatish Balay   TaoSetConvergenceTest - Sets the function that is to be used to test
1399a7e14dcfSSatish Balay   for convergence o fthe iterative minimization solution.  The new convergence
140065ba42b6SBarry Smith   testing routine will replace Tao's default convergence test.
1401a7e14dcfSSatish Balay 
1402c3339decSBarry Smith   Logically Collective
1403a7e14dcfSSatish Balay 
1404a7e14dcfSSatish Balay   Input Parameters:
140547450a7bSBarry Smith + tao  - the `Tao` object
1406a7e14dcfSSatish Balay . conv - the routine to test for convergence
1407a7e14dcfSSatish Balay - ctx  - [optional] context for private data for the convergence routine
140847450a7bSBarry Smith         (may be `NULL`)
1409a7e14dcfSSatish Balay 
141020f4b53cSBarry Smith   Calling sequence of `conv`:
141147450a7bSBarry Smith + tao - the `Tao` object
1412a7e14dcfSSatish Balay - ctx - [optional] convergence context
1413a7e14dcfSSatish Balay 
141447450a7bSBarry Smith   Level: advanced
141547450a7bSBarry Smith 
141665ba42b6SBarry Smith   Note:
141765ba42b6SBarry Smith   The new convergence testing routine should call `TaoSetConvergedReason()`.
1418a7e14dcfSSatish Balay 
14191cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoSolve()`, `TaoSetConvergedReason()`, `TaoGetSolutionStatus()`, `TaoGetTolerances()`, `TaoSetMonitor`
1420a7e14dcfSSatish Balay @*/
1421d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoSetConvergenceTest(Tao tao, PetscErrorCode (*conv)(Tao, void *), void *ctx)
1422d71ae5a4SJacob Faibussowitsch {
1423a7e14dcfSSatish Balay   PetscFunctionBegin;
1424441846f8SBarry Smith   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
142594511df7SStefano Zampini   tao->ops->convergencetest = conv;
142694511df7SStefano Zampini   tao->cnvP                 = ctx;
14273ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
1428a7e14dcfSSatish Balay }
1429a7e14dcfSSatish Balay 
1430a7e14dcfSSatish Balay /*@C
143165ba42b6SBarry Smith   TaoSetMonitor - Sets an additional function that is to be used at every
1432a7e14dcfSSatish Balay   iteration of the solver to display the iteration's
1433a7e14dcfSSatish Balay   progress.
1434a7e14dcfSSatish Balay 
1435c3339decSBarry Smith   Logically Collective
1436a7e14dcfSSatish Balay 
1437a7e14dcfSSatish Balay   Input Parameters:
143847450a7bSBarry Smith + tao  - the `Tao` solver context
14392fe279fdSBarry Smith . func - monitoring routine
14402fe279fdSBarry Smith . ctx  - [optional] user-defined context for private data for the monitor routine (may be `NULL`)
14412fe279fdSBarry Smith - dest - [optional] function to destroy the context when the `Tao` is destroyed
1442a7e14dcfSSatish Balay 
14432fe279fdSBarry Smith   Calling sequence of `func`:
144447450a7bSBarry Smith + tao - the `Tao` solver context
14452fe279fdSBarry Smith - ctx - [optional] monitoring context
14462fe279fdSBarry Smith 
14472fe279fdSBarry Smith   Calling sequence of `dest`:
14482fe279fdSBarry Smith . ctx - monitoring context
1449a7e14dcfSSatish Balay 
1450a7e14dcfSSatish Balay   Options Database Keys:
145165ba42b6SBarry Smith + -tao_monitor          - sets the default monitor `TaoMonitorDefault()`
1452a7e14dcfSSatish Balay . -tao_smonitor         - sets short monitor
1453a7e14dcfSSatish Balay . -tao_cmonitor         - same as smonitor plus constraint norm
1454a7e14dcfSSatish Balay . -tao_view_solution    - view solution at each iteration
1455a7e14dcfSSatish Balay . -tao_view_gradient    - view gradient at each iteration
14564a48860cSAlp Dener . -tao_view_ls_residual - view least-squares residual vector at each iteration
1457a7e14dcfSSatish Balay - -tao_cancelmonitors   - cancels all monitors that have been hardwired into a code by calls to TaoSetMonitor(), but does not cancel those set via the options database.
1458a7e14dcfSSatish Balay 
145947450a7bSBarry Smith   Level: intermediate
146047450a7bSBarry Smith 
146187497f52SBarry Smith   Notes:
1462a7e14dcfSSatish Balay   Several different monitoring routines may be set by calling
146365ba42b6SBarry Smith   `TaoSetMonitor()` multiple times; all will be called in the
1464a7e14dcfSSatish Balay   order in which they were set.
1465a7e14dcfSSatish Balay 
1466e056e8ceSJacob Faibussowitsch   Fortran Notes:
146795452b02SPatrick Sanan   Only one monitor function may be set
1468a7e14dcfSSatish Balay 
14691cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoSolve()`, `TaoMonitorDefault()`, `TaoCancelMonitors()`, `TaoSetDestroyRoutine()`, `TaoView()`
1470a7e14dcfSSatish Balay @*/
1471d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoSetMonitor(Tao tao, PetscErrorCode (*func)(Tao, void *), void *ctx, PetscErrorCode (*dest)(void **))
1472d71ae5a4SJacob Faibussowitsch {
147308d19d1fSJason Sarich   PetscInt  i;
14748732526dSAlp Dener   PetscBool identical;
147508d19d1fSJason Sarich 
1476a7e14dcfSSatish Balay   PetscFunctionBegin;
1477441846f8SBarry Smith   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
14783c859ba3SBarry Smith   PetscCheck(tao->numbermonitors < MAXTAOMONITORS, PetscObjectComm((PetscObject)tao), PETSC_ERR_SUP, "Cannot attach another monitor -- max=%d", MAXTAOMONITORS);
147908d19d1fSJason Sarich 
148008d19d1fSJason Sarich   for (i = 0; i < tao->numbermonitors; i++) {
14819566063dSJacob Faibussowitsch     PetscCall(PetscMonitorCompare((PetscErrorCode(*)(void))func, ctx, dest, (PetscErrorCode(*)(void))tao->monitor[i], tao->monitorcontext[i], tao->monitordestroy[i], &identical));
14823ba16761SJacob Faibussowitsch     if (identical) PetscFunctionReturn(PETSC_SUCCESS);
148308d19d1fSJason Sarich   }
1484a7e14dcfSSatish Balay   tao->monitor[tao->numbermonitors]        = func;
14858732526dSAlp Dener   tao->monitorcontext[tao->numbermonitors] = (void *)ctx;
1486a7e14dcfSSatish Balay   tao->monitordestroy[tao->numbermonitors] = dest;
1487a7e14dcfSSatish Balay   ++tao->numbermonitors;
14883ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
1489a7e14dcfSSatish Balay }
1490a7e14dcfSSatish Balay 
1491a7e14dcfSSatish Balay /*@
149247450a7bSBarry Smith   TaoCancelMonitors - Clears all the monitor functions for a `Tao` object.
1493a7e14dcfSSatish Balay 
1494c3339decSBarry Smith   Logically Collective
1495a7e14dcfSSatish Balay 
149647450a7bSBarry Smith   Input Parameter:
149747450a7bSBarry Smith . tao - the `Tao` solver context
1498a7e14dcfSSatish Balay 
14993c7db156SBarry Smith   Options Database Key:
1500a7e14dcfSSatish Balay . -tao_cancelmonitors - cancels all monitors that have been hardwired
150165ba42b6SBarry Smith     into a code by calls to `TaoSetMonitor()`, but does not cancel those
1502a7e14dcfSSatish Balay     set via the options database
1503a7e14dcfSSatish Balay 
1504a7e14dcfSSatish Balay   Level: advanced
1505a7e14dcfSSatish Balay 
150647450a7bSBarry Smith   Note:
150747450a7bSBarry Smith   There is no way to clear one specific monitor from a `Tao` object.
150847450a7bSBarry Smith 
15091cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoMonitorDefault()`, `TaoSetMonitor()`
1510a7e14dcfSSatish Balay @*/
1511d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoCancelMonitors(Tao tao)
1512d71ae5a4SJacob Faibussowitsch {
1513a7e14dcfSSatish Balay   PetscInt i;
151447a47007SBarry Smith 
1515a7e14dcfSSatish Balay   PetscFunctionBegin;
1516441846f8SBarry Smith   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
1517a7e14dcfSSatish Balay   for (i = 0; i < tao->numbermonitors; i++) {
151848a46eb9SPierre Jolivet     if (tao->monitordestroy[i]) PetscCall((*tao->monitordestroy[i])(&tao->monitorcontext[i]));
1519a7e14dcfSSatish Balay   }
1520a7e14dcfSSatish Balay   tao->numbermonitors = 0;
15213ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
1522a7e14dcfSSatish Balay }
1523a7e14dcfSSatish Balay 
15247fab98ebSJason Sarich /*@
152547450a7bSBarry Smith   TaoMonitorDefault - Default routine for monitoring progress of `TaoSolve()`
1526a7e14dcfSSatish Balay 
1527c3339decSBarry Smith   Collective
1528a7e14dcfSSatish Balay 
1529a7e14dcfSSatish Balay   Input Parameters:
153047450a7bSBarry Smith + tao - the `Tao` context
153147450a7bSBarry Smith - ctx - `PetscViewer` context or `NULL`
1532a7e14dcfSSatish Balay 
153347450a7bSBarry Smith   Options Database Key:
1534147403d9SBarry Smith . -tao_monitor - turn on default monitoring
1535a7e14dcfSSatish Balay 
1536a7e14dcfSSatish Balay   Level: advanced
1537a7e14dcfSSatish Balay 
153847450a7bSBarry Smith   Note:
153947450a7bSBarry Smith   This monitor prints the function value and gradient
154047450a7bSBarry Smith   norm at each iteration.
154147450a7bSBarry Smith 
15421cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoDefaultSMonitor()`, `TaoSetMonitor()`
1543a7e14dcfSSatish Balay @*/
1544d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoMonitorDefault(Tao tao, void *ctx)
1545d71ae5a4SJacob Faibussowitsch {
154663b15415SAlp Dener   PetscInt    its, tabs;
1547a7e14dcfSSatish Balay   PetscReal   fct, gnorm;
15488163d661SBarry Smith   PetscViewer viewer = (PetscViewer)ctx;
154947a47007SBarry Smith 
1550a7e14dcfSSatish Balay   PetscFunctionBegin;
155194511df7SStefano Zampini   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
15528163d661SBarry Smith   PetscValidHeaderSpecific(viewer, PETSC_VIEWER_CLASSID, 2);
1553a7e14dcfSSatish Balay   its   = tao->niter;
1554a7e14dcfSSatish Balay   fct   = tao->fc;
1555a7e14dcfSSatish Balay   gnorm = tao->residual;
15569566063dSJacob Faibussowitsch   PetscCall(PetscViewerASCIIGetTab(viewer, &tabs));
15579566063dSJacob Faibussowitsch   PetscCall(PetscViewerASCIISetTab(viewer, ((PetscObject)tao)->tablevel));
1558494bef23SAlp Dener   if (its == 0 && ((PetscObject)tao)->prefix && !tao->header_printed) {
15599566063dSJacob Faibussowitsch     PetscCall(PetscViewerASCIIPrintf(viewer, "  Iteration information for %s solve.\n", ((PetscObject)tao)->prefix));
1560494bef23SAlp Dener     tao->header_printed = PETSC_TRUE;
156163b15415SAlp Dener   }
156263a3b9bcSJacob Faibussowitsch   PetscCall(PetscViewerASCIIPrintf(viewer, "%3" PetscInt_FMT " TAO,", its));
15639566063dSJacob Faibussowitsch   PetscCall(PetscViewerASCIIPrintf(viewer, "  Function value: %g,", (double)fct));
156408d19d1fSJason Sarich   if (gnorm >= PETSC_INFINITY) {
15659566063dSJacob Faibussowitsch     PetscCall(PetscViewerASCIIPrintf(viewer, "  Residual: Inf \n"));
156608d19d1fSJason Sarich   } else {
15679566063dSJacob Faibussowitsch     PetscCall(PetscViewerASCIIPrintf(viewer, "  Residual: %g \n", (double)gnorm));
156808d19d1fSJason Sarich   }
15699566063dSJacob Faibussowitsch   PetscCall(PetscViewerASCIISetTab(viewer, tabs));
15703ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
1571a7e14dcfSSatish Balay }
1572a7e14dcfSSatish Balay 
15737fab98ebSJason Sarich /*@
157447450a7bSBarry Smith   TaoDefaultGMonitor - Default routine for monitoring progress of `TaoSolve()` with extra detail on the globalization method.
15758d5ead36SAlp Dener 
1576c3339decSBarry Smith   Collective
15778d5ead36SAlp Dener 
15788d5ead36SAlp Dener   Input Parameters:
157947450a7bSBarry Smith + tao - the `Tao` context
158047450a7bSBarry Smith - ctx - `PetscViewer` context or `NULL`
15818d5ead36SAlp Dener 
158247450a7bSBarry Smith   Options Database Key:
1583147403d9SBarry Smith . -tao_gmonitor - turn on monitoring with globalization information
15848d5ead36SAlp Dener 
15858d5ead36SAlp Dener   Level: advanced
15868d5ead36SAlp Dener 
158747450a7bSBarry Smith   Note:
158847450a7bSBarry Smith   This monitor prints the function value and gradient norm at each
158947450a7bSBarry Smith   iteration, as well as the step size and trust radius. Note that the
159047450a7bSBarry Smith   step size and trust radius may be the same for some algorithms.
159147450a7bSBarry Smith 
15921cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoDefaultSMonitor()`, `TaoSetMonitor()`
15938d5ead36SAlp Dener @*/
1594d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoDefaultGMonitor(Tao tao, void *ctx)
1595d71ae5a4SJacob Faibussowitsch {
15968d5ead36SAlp Dener   PetscInt    its, tabs;
15978d5ead36SAlp Dener   PetscReal   fct, gnorm, stp, tr;
15988d5ead36SAlp Dener   PetscViewer viewer = (PetscViewer)ctx;
15998d5ead36SAlp Dener 
16008d5ead36SAlp Dener   PetscFunctionBegin;
160194511df7SStefano Zampini   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
16028d5ead36SAlp Dener   PetscValidHeaderSpecific(viewer, PETSC_VIEWER_CLASSID, 2);
16038d5ead36SAlp Dener   its   = tao->niter;
16048d5ead36SAlp Dener   fct   = tao->fc;
16058d5ead36SAlp Dener   gnorm = tao->residual;
16068d5ead36SAlp Dener   stp   = tao->step;
16078d5ead36SAlp Dener   tr    = tao->trust;
16089566063dSJacob Faibussowitsch   PetscCall(PetscViewerASCIIGetTab(viewer, &tabs));
16099566063dSJacob Faibussowitsch   PetscCall(PetscViewerASCIISetTab(viewer, ((PetscObject)tao)->tablevel));
1610494bef23SAlp Dener   if (its == 0 && ((PetscObject)tao)->prefix && !tao->header_printed) {
16119566063dSJacob Faibussowitsch     PetscCall(PetscViewerASCIIPrintf(viewer, "  Iteration information for %s solve.\n", ((PetscObject)tao)->prefix));
1612494bef23SAlp Dener     tao->header_printed = PETSC_TRUE;
16138d5ead36SAlp Dener   }
161463a3b9bcSJacob Faibussowitsch   PetscCall(PetscViewerASCIIPrintf(viewer, "%3" PetscInt_FMT " TAO,", its));
16159566063dSJacob Faibussowitsch   PetscCall(PetscViewerASCIIPrintf(viewer, "  Function value: %g,", (double)fct));
16168d5ead36SAlp Dener   if (gnorm >= PETSC_INFINITY) {
16179566063dSJacob Faibussowitsch     PetscCall(PetscViewerASCIIPrintf(viewer, "  Residual: Inf,"));
16188d5ead36SAlp Dener   } else {
16199566063dSJacob Faibussowitsch     PetscCall(PetscViewerASCIIPrintf(viewer, "  Residual: %g,", (double)gnorm));
16208d5ead36SAlp Dener   }
16219566063dSJacob Faibussowitsch   PetscCall(PetscViewerASCIIPrintf(viewer, "  Step: %g,  Trust: %g\n", (double)stp, (double)tr));
16229566063dSJacob Faibussowitsch   PetscCall(PetscViewerASCIISetTab(viewer, tabs));
16233ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
16248d5ead36SAlp Dener }
16258d5ead36SAlp Dener 
16268d5ead36SAlp Dener /*@
162747450a7bSBarry Smith   TaoDefaultSMonitor - Default routine for monitoring progress of `TaoSolve()`
1628a7e14dcfSSatish Balay 
1629c3339decSBarry Smith   Collective
1630a7e14dcfSSatish Balay 
1631a7e14dcfSSatish Balay   Input Parameters:
163247450a7bSBarry Smith + tao - the `Tao` context
163347450a7bSBarry Smith - ctx - `PetscViewer` context of type `PETSCVIEWERASCII`
1634a7e14dcfSSatish Balay 
163547450a7bSBarry Smith   Options Database Key:
1636147403d9SBarry Smith . -tao_smonitor - turn on default short monitoring
1637a7e14dcfSSatish Balay 
1638a7e14dcfSSatish Balay   Level: advanced
1639a7e14dcfSSatish Balay 
164047450a7bSBarry Smith   Note:
164147450a7bSBarry Smith   Same as `TaoMonitorDefault()` except
164247450a7bSBarry Smith   it prints fewer digits of the residual as the residual gets smaller.
164347450a7bSBarry Smith   This is because the later digits are meaningless and are often
164447450a7bSBarry Smith   different on different machines; by using this routine different
164547450a7bSBarry Smith   machines will usually generate the same output.
164647450a7bSBarry Smith 
16471cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoMonitorDefault()`, `TaoSetMonitor()`
1648a7e14dcfSSatish Balay @*/
1649d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoDefaultSMonitor(Tao tao, void *ctx)
1650d71ae5a4SJacob Faibussowitsch {
16512c9b8b83SAlp Dener   PetscInt    its, tabs;
1652a7e14dcfSSatish Balay   PetscReal   fct, gnorm;
16534d4332d5SBarry Smith   PetscViewer viewer = (PetscViewer)ctx;
165447a47007SBarry Smith 
1655a7e14dcfSSatish Balay   PetscFunctionBegin;
165694511df7SStefano Zampini   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
16574d4332d5SBarry Smith   PetscValidHeaderSpecific(viewer, PETSC_VIEWER_CLASSID, 2);
1658a7e14dcfSSatish Balay   its   = tao->niter;
1659a7e14dcfSSatish Balay   fct   = tao->fc;
1660a7e14dcfSSatish Balay   gnorm = tao->residual;
16619566063dSJacob Faibussowitsch   PetscCall(PetscViewerASCIIGetTab(viewer, &tabs));
16629566063dSJacob Faibussowitsch   PetscCall(PetscViewerASCIISetTab(viewer, ((PetscObject)tao)->tablevel));
166363a3b9bcSJacob Faibussowitsch   PetscCall(PetscViewerASCIIPrintf(viewer, "iter = %3" PetscInt_FMT ",", its));
16649566063dSJacob Faibussowitsch   PetscCall(PetscViewerASCIIPrintf(viewer, " Function value %g,", (double)fct));
1665d393f493SBarry Smith   if (gnorm >= PETSC_INFINITY) {
16669566063dSJacob Faibussowitsch     PetscCall(PetscViewerASCIIPrintf(viewer, " Residual: Inf \n"));
166708d19d1fSJason Sarich   } else if (gnorm > 1.e-6) {
16689566063dSJacob Faibussowitsch     PetscCall(PetscViewerASCIIPrintf(viewer, " Residual: %g \n", (double)gnorm));
1669a7e14dcfSSatish Balay   } else if (gnorm > 1.e-11) {
16709566063dSJacob Faibussowitsch     PetscCall(PetscViewerASCIIPrintf(viewer, " Residual: < 1.0e-6 \n"));
1671a7e14dcfSSatish Balay   } else {
16729566063dSJacob Faibussowitsch     PetscCall(PetscViewerASCIIPrintf(viewer, " Residual: < 1.0e-11 \n"));
1673a7e14dcfSSatish Balay   }
16749566063dSJacob Faibussowitsch   PetscCall(PetscViewerASCIISetTab(viewer, tabs));
16753ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
1676a7e14dcfSSatish Balay }
1677a7e14dcfSSatish Balay 
16787fab98ebSJason Sarich /*@
167965ba42b6SBarry Smith   TaoDefaultCMonitor - same as `TaoMonitorDefault()` except
168047450a7bSBarry Smith   it prints the norm of the constraint function.
1681a7e14dcfSSatish Balay 
1682c3339decSBarry Smith   Collective
1683a7e14dcfSSatish Balay 
1684a7e14dcfSSatish Balay   Input Parameters:
168547450a7bSBarry Smith + tao - the `Tao` context
168647450a7bSBarry Smith - ctx - `PetscViewer` context or `NULL`
1687a7e14dcfSSatish Balay 
168847450a7bSBarry Smith   Options Database Key:
1689147403d9SBarry Smith . -tao_cmonitor - monitor the constraints
1690a7e14dcfSSatish Balay 
1691a7e14dcfSSatish Balay   Level: advanced
1692a7e14dcfSSatish Balay 
16931cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoMonitorDefault()`, `TaoSetMonitor()`
1694a7e14dcfSSatish Balay @*/
1695d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoDefaultCMonitor(Tao tao, void *ctx)
1696d71ae5a4SJacob Faibussowitsch {
16972c9b8b83SAlp Dener   PetscInt    its, tabs;
1698a7e14dcfSSatish Balay   PetscReal   fct, gnorm;
1699d393f493SBarry Smith   PetscViewer viewer = (PetscViewer)ctx;
1700a7e14dcfSSatish Balay 
1701a7e14dcfSSatish Balay   PetscFunctionBegin;
170294511df7SStefano Zampini   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
1703d393f493SBarry Smith   PetscValidHeaderSpecific(viewer, PETSC_VIEWER_CLASSID, 2);
1704a7e14dcfSSatish Balay   its   = tao->niter;
1705a7e14dcfSSatish Balay   fct   = tao->fc;
1706a7e14dcfSSatish Balay   gnorm = tao->residual;
17079566063dSJacob Faibussowitsch   PetscCall(PetscViewerASCIIGetTab(viewer, &tabs));
17089566063dSJacob Faibussowitsch   PetscCall(PetscViewerASCIISetTab(viewer, ((PetscObject)tao)->tablevel));
170963a3b9bcSJacob Faibussowitsch   PetscCall(PetscViewerASCIIPrintf(viewer, "iter = %" PetscInt_FMT ",", its));
17109566063dSJacob Faibussowitsch   PetscCall(PetscViewerASCIIPrintf(viewer, " Function value: %g,", (double)fct));
17119566063dSJacob Faibussowitsch   PetscCall(PetscViewerASCIIPrintf(viewer, "  Residual: %g ", (double)gnorm));
17129566063dSJacob Faibussowitsch   PetscCall(PetscViewerASCIIPrintf(viewer, "  Constraint: %g \n", (double)tao->cnorm));
17139566063dSJacob Faibussowitsch   PetscCall(PetscViewerASCIISetTab(viewer, tabs));
17143ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
1715a7e14dcfSSatish Balay }
1716a7e14dcfSSatish Balay 
1717a7e14dcfSSatish Balay /*@C
171847450a7bSBarry Smith   TaoSolutionMonitor - Views the solution at each iteration of `TaoSolve()`
1719a7e14dcfSSatish Balay 
1720c3339decSBarry Smith   Collective
1721a7e14dcfSSatish Balay 
1722a7e14dcfSSatish Balay   Input Parameters:
172347450a7bSBarry Smith + tao - the `Tao` context
172447450a7bSBarry Smith - ctx - `PetscViewer` context or `NULL`
1725a7e14dcfSSatish Balay 
172647450a7bSBarry Smith   Options Database Key:
1727147403d9SBarry Smith . -tao_view_solution - view the solution
1728a7e14dcfSSatish Balay 
1729a7e14dcfSSatish Balay   Level: advanced
1730a7e14dcfSSatish Balay 
17311cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoDefaultSMonitor()`, `TaoSetMonitor()`
1732a7e14dcfSSatish Balay @*/
1733d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoSolutionMonitor(Tao tao, void *ctx)
1734d71ae5a4SJacob Faibussowitsch {
1735feb237baSPierre Jolivet   PetscViewer viewer = (PetscViewer)ctx;
173647a47007SBarry Smith 
1737a7e14dcfSSatish Balay   PetscFunctionBegin;
173894511df7SStefano Zampini   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
1739d393f493SBarry Smith   PetscValidHeaderSpecific(viewer, PETSC_VIEWER_CLASSID, 2);
17409566063dSJacob Faibussowitsch   PetscCall(VecView(tao->solution, viewer));
17413ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
1742a7e14dcfSSatish Balay }
1743a7e14dcfSSatish Balay 
1744a7e14dcfSSatish Balay /*@C
174547450a7bSBarry Smith   TaoGradientMonitor - Views the gradient at each iteration of `TaoSolve()`
1746a7e14dcfSSatish Balay 
1747c3339decSBarry Smith   Collective
1748a7e14dcfSSatish Balay 
1749a7e14dcfSSatish Balay   Input Parameters:
175047450a7bSBarry Smith + tao - the `Tao` context
175147450a7bSBarry Smith - ctx - `PetscViewer` context or `NULL`
1752a7e14dcfSSatish Balay 
175347450a7bSBarry Smith   Options Database Key:
1754147403d9SBarry Smith . -tao_view_gradient - view the gradient at each iteration
1755a7e14dcfSSatish Balay 
1756a7e14dcfSSatish Balay   Level: advanced
1757a7e14dcfSSatish Balay 
17581cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoDefaultSMonitor()`, `TaoSetMonitor()`
1759a7e14dcfSSatish Balay @*/
1760d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoGradientMonitor(Tao tao, void *ctx)
1761d71ae5a4SJacob Faibussowitsch {
1762d393f493SBarry Smith   PetscViewer viewer = (PetscViewer)ctx;
176347a47007SBarry Smith 
1764a7e14dcfSSatish Balay   PetscFunctionBegin;
176594511df7SStefano Zampini   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
1766d393f493SBarry Smith   PetscValidHeaderSpecific(viewer, PETSC_VIEWER_CLASSID, 2);
17679566063dSJacob Faibussowitsch   PetscCall(VecView(tao->gradient, viewer));
17683ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
1769a7e14dcfSSatish Balay }
1770a7e14dcfSSatish Balay 
1771a7e14dcfSSatish Balay /*@C
177247450a7bSBarry Smith   TaoStepDirectionMonitor - Views the step-direction at each iteration of `TaoSolve()`
1773a7e14dcfSSatish Balay 
1774c3339decSBarry Smith   Collective
1775a7e14dcfSSatish Balay 
1776a7e14dcfSSatish Balay   Input Parameters:
177747450a7bSBarry Smith + tao - the `Tao` context
177847450a7bSBarry Smith - ctx - `PetscViewer` context or `NULL`
1779a7e14dcfSSatish Balay 
178047450a7bSBarry Smith   Options Database Key:
178147450a7bSBarry Smith . -tao_view_stepdirection - view the step direction vector at each iteration
1782a7e14dcfSSatish Balay 
1783a7e14dcfSSatish Balay   Level: advanced
1784a7e14dcfSSatish Balay 
17851cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoDefaultSMonitor()`, `TaoSetMonitor()`
1786a7e14dcfSSatish Balay @*/
1787d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoStepDirectionMonitor(Tao tao, void *ctx)
1788d71ae5a4SJacob Faibussowitsch {
1789d393f493SBarry Smith   PetscViewer viewer = (PetscViewer)ctx;
1790d393f493SBarry Smith 
1791a7e14dcfSSatish Balay   PetscFunctionBegin;
179294511df7SStefano Zampini   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
1793d393f493SBarry Smith   PetscValidHeaderSpecific(viewer, PETSC_VIEWER_CLASSID, 2);
17949566063dSJacob Faibussowitsch   PetscCall(VecView(tao->stepdirection, viewer));
17953ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
1796a7e14dcfSSatish Balay }
1797a7e14dcfSSatish Balay 
1798a7e14dcfSSatish Balay /*@C
179947450a7bSBarry Smith   TaoDrawSolutionMonitor - Plots the solution at each iteration of `TaoSolve()`
1800a7e14dcfSSatish Balay 
1801c3339decSBarry Smith   Collective
1802a7e14dcfSSatish Balay 
1803a7e14dcfSSatish Balay   Input Parameters:
180447450a7bSBarry Smith + tao - the `Tao` context
180565ba42b6SBarry Smith - ctx - `TaoMonitorDraw` context
1806a7e14dcfSSatish Balay 
180747450a7bSBarry Smith   Options Database Key:
1808147403d9SBarry Smith . -tao_draw_solution - draw the solution at each iteration
1809a7e14dcfSSatish Balay 
1810a7e14dcfSSatish Balay   Level: advanced
1811a7e14dcfSSatish Balay 
18121cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoSolutionMonitor()`, `TaoSetMonitor()`, `TaoDrawGradientMonitor`, `TaoMonitorDraw`
1813a7e14dcfSSatish Balay @*/
1814d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoDrawSolutionMonitor(Tao tao, void *ctx)
1815d71ae5a4SJacob Faibussowitsch {
1816e882e171SHong Zhang   TaoMonitorDrawCtx ictx = (TaoMonitorDrawCtx)ctx;
181747a47007SBarry Smith 
1818a7e14dcfSSatish Balay   PetscFunctionBegin;
181994511df7SStefano Zampini   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
18203ba16761SJacob Faibussowitsch   if (!(((ictx->howoften > 0) && (!(tao->niter % ictx->howoften))) || ((ictx->howoften == -1) && tao->reason))) PetscFunctionReturn(PETSC_SUCCESS);
18219566063dSJacob Faibussowitsch   PetscCall(VecView(tao->solution, ictx->viewer));
18223ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
1823a7e14dcfSSatish Balay }
1824a7e14dcfSSatish Balay 
1825a7e14dcfSSatish Balay /*@C
182647450a7bSBarry Smith   TaoDrawGradientMonitor - Plots the gradient at each iteration of `TaoSolve()`
1827a7e14dcfSSatish Balay 
1828c3339decSBarry Smith   Collective
1829a7e14dcfSSatish Balay 
1830a7e14dcfSSatish Balay   Input Parameters:
183147450a7bSBarry Smith + tao - the `Tao` context
183265ba42b6SBarry Smith - ctx - `PetscViewer` context
1833a7e14dcfSSatish Balay 
183447450a7bSBarry Smith   Options Database Key:
1835147403d9SBarry Smith . -tao_draw_gradient - draw the gradient at each iteration
1836a7e14dcfSSatish Balay 
1837a7e14dcfSSatish Balay   Level: advanced
1838a7e14dcfSSatish Balay 
18391cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoGradientMonitor()`, `TaoSetMonitor()`, `TaoDrawSolutionMonitor`
1840a7e14dcfSSatish Balay @*/
1841d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoDrawGradientMonitor(Tao tao, void *ctx)
1842d71ae5a4SJacob Faibussowitsch {
1843e882e171SHong Zhang   TaoMonitorDrawCtx ictx = (TaoMonitorDrawCtx)ctx;
184447a47007SBarry Smith 
1845a7e14dcfSSatish Balay   PetscFunctionBegin;
184694511df7SStefano Zampini   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
18473ba16761SJacob Faibussowitsch   if (!(((ictx->howoften > 0) && (!(tao->niter % ictx->howoften))) || ((ictx->howoften == -1) && tao->reason))) PetscFunctionReturn(PETSC_SUCCESS);
18489566063dSJacob Faibussowitsch   PetscCall(VecView(tao->gradient, ictx->viewer));
18493ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
1850a7e14dcfSSatish Balay }
1851a7e14dcfSSatish Balay 
1852a7e14dcfSSatish Balay /*@C
185347450a7bSBarry Smith   TaoDrawStepMonitor - Plots the step direction at each iteration of `TaoSolve()`
1854a7e14dcfSSatish Balay 
1855c3339decSBarry Smith   Collective
1856a7e14dcfSSatish Balay 
1857a7e14dcfSSatish Balay   Input Parameters:
185847450a7bSBarry Smith + tao - the `Tao` context
185947450a7bSBarry Smith - ctx - the `PetscViewer` context
1860a7e14dcfSSatish Balay 
186147450a7bSBarry Smith   Options Database Key:
1862147403d9SBarry Smith . -tao_draw_step - draw the step direction at each iteration
1863a7e14dcfSSatish Balay 
1864a7e14dcfSSatish Balay   Level: advanced
1865a7e14dcfSSatish Balay 
18661cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoSetMonitor()`, `TaoDrawSolutionMonitor`
1867a7e14dcfSSatish Balay @*/
1868d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoDrawStepMonitor(Tao tao, void *ctx)
1869d71ae5a4SJacob Faibussowitsch {
187094511df7SStefano Zampini   PetscViewer viewer = (PetscViewer)ctx;
187147a47007SBarry Smith 
1872a7e14dcfSSatish Balay   PetscFunctionBegin;
187394511df7SStefano Zampini   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
187494511df7SStefano Zampini   PetscValidHeaderSpecific(viewer, PETSC_VIEWER_CLASSID, 2);
18759566063dSJacob Faibussowitsch   PetscCall(VecView(tao->stepdirection, viewer));
18763ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
1877a7e14dcfSSatish Balay }
1878a7e14dcfSSatish Balay 
1879a7e14dcfSSatish Balay /*@C
188047450a7bSBarry Smith   TaoResidualMonitor - Views the least-squares residual at each iteration of `TaoSolve()`
1881a7e14dcfSSatish Balay 
1882c3339decSBarry Smith   Collective
1883a7e14dcfSSatish Balay 
1884a7e14dcfSSatish Balay   Input Parameters:
188547450a7bSBarry Smith + tao - the `Tao` context
188647450a7bSBarry Smith - ctx - the `PetscViewer` context or `NULL`
1887a7e14dcfSSatish Balay 
188847450a7bSBarry Smith   Options Database Key:
188947450a7bSBarry Smith . -tao_view_ls_residual - view the residual at each iteration
1890a7e14dcfSSatish Balay 
1891a7e14dcfSSatish Balay   Level: advanced
1892a7e14dcfSSatish Balay 
18931cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoDefaultSMonitor()`, `TaoSetMonitor()`
1894a7e14dcfSSatish Balay @*/
1895d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoResidualMonitor(Tao tao, void *ctx)
1896d71ae5a4SJacob Faibussowitsch {
1897d393f493SBarry Smith   PetscViewer viewer = (PetscViewer)ctx;
189847a47007SBarry Smith 
1899a7e14dcfSSatish Balay   PetscFunctionBegin;
190094511df7SStefano Zampini   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
1901d393f493SBarry Smith   PetscValidHeaderSpecific(viewer, PETSC_VIEWER_CLASSID, 2);
19029566063dSJacob Faibussowitsch   PetscCall(VecView(tao->ls_res, viewer));
19033ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
1904a7e14dcfSSatish Balay }
1905a7e14dcfSSatish Balay 
19067fab98ebSJason Sarich /*@
1907a7e14dcfSSatish Balay   TaoDefaultConvergenceTest - Determines whether the solver should continue iterating
1908a7e14dcfSSatish Balay   or terminate.
1909a7e14dcfSSatish Balay 
1910c3339decSBarry Smith   Collective
1911a7e14dcfSSatish Balay 
1912a7e14dcfSSatish Balay   Input Parameters:
191347450a7bSBarry Smith + tao   - the `Tao` context
1914a7e14dcfSSatish Balay - dummy - unused dummy context
1915a7e14dcfSSatish Balay 
191647450a7bSBarry Smith   Level: developer
191747450a7bSBarry Smith 
1918a7e14dcfSSatish Balay   Notes:
1919a7e14dcfSSatish Balay   This routine checks the residual in the optimality conditions, the
1920a7e14dcfSSatish Balay   relative residual in the optimity conditions, the number of function
1921a7e14dcfSSatish Balay   evaluations, and the function value to test convergence.  Some
1922a7e14dcfSSatish Balay   solvers may use different convergence routines.
1923a7e14dcfSSatish Balay 
19241cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoSetTolerances()`, `TaoGetConvergedReason()`, `TaoSetConvergedReason()`
1925a7e14dcfSSatish Balay @*/
1926d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoDefaultConvergenceTest(Tao tao, void *dummy)
1927d71ae5a4SJacob Faibussowitsch {
1928a7e14dcfSSatish Balay   PetscInt           niter = tao->niter, nfuncs = PetscMax(tao->nfuncs, tao->nfuncgrads);
1929a7e14dcfSSatish Balay   PetscInt           max_funcs = tao->max_funcs;
1930a7e14dcfSSatish Balay   PetscReal          gnorm = tao->residual, gnorm0 = tao->gnorm0;
1931a7e14dcfSSatish Balay   PetscReal          f = tao->fc, steptol = tao->steptol, trradius = tao->step;
1932a7e14dcfSSatish Balay   PetscReal          gatol = tao->gatol, grtol = tao->grtol, gttol = tao->gttol;
1933e52336cbSBarry Smith   PetscReal          catol = tao->catol, crtol = tao->crtol;
1934e52336cbSBarry Smith   PetscReal          fmin = tao->fmin, cnorm = tao->cnorm;
1935e4cb33bbSBarry Smith   TaoConvergedReason reason = tao->reason;
1936a7e14dcfSSatish Balay 
1937a7e14dcfSSatish Balay   PetscFunctionBegin;
1938441846f8SBarry Smith   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
19393ba16761SJacob Faibussowitsch   if (reason != TAO_CONTINUE_ITERATING) PetscFunctionReturn(PETSC_SUCCESS);
1940a7e14dcfSSatish Balay 
1941a7e14dcfSSatish Balay   if (PetscIsInfOrNanReal(f)) {
19429566063dSJacob Faibussowitsch     PetscCall(PetscInfo(tao, "Failed to converged, function value is Inf or NaN\n"));
1943a7e14dcfSSatish Balay     reason = TAO_DIVERGED_NAN;
1944a7e14dcfSSatish Balay   } else if (f <= fmin && cnorm <= catol) {
19459566063dSJacob Faibussowitsch     PetscCall(PetscInfo(tao, "Converged due to function value %g < minimum function value %g\n", (double)f, (double)fmin));
1946a7e14dcfSSatish Balay     reason = TAO_CONVERGED_MINF;
1947a7e14dcfSSatish Balay   } else if (gnorm <= gatol && cnorm <= catol) {
19489566063dSJacob Faibussowitsch     PetscCall(PetscInfo(tao, "Converged due to residual norm ||g(X)||=%g < %g\n", (double)gnorm, (double)gatol));
1949a7e14dcfSSatish Balay     reason = TAO_CONVERGED_GATOL;
1950a7e14dcfSSatish Balay   } else if (f != 0 && PetscAbsReal(gnorm / f) <= grtol && cnorm <= crtol) {
19519566063dSJacob Faibussowitsch     PetscCall(PetscInfo(tao, "Converged due to residual ||g(X)||/|f(X)| =%g < %g\n", (double)(gnorm / f), (double)grtol));
1952a7e14dcfSSatish Balay     reason = TAO_CONVERGED_GRTOL;
1953e1d16bb9SBarry Smith   } else if (gnorm0 != 0 && ((gttol == 0 && gnorm == 0) || gnorm / gnorm0 < gttol) && cnorm <= crtol) {
19549566063dSJacob Faibussowitsch     PetscCall(PetscInfo(tao, "Converged due to relative residual norm ||g(X)||/||g(X0)|| = %g < %g\n", (double)(gnorm / gnorm0), (double)gttol));
1955a7e14dcfSSatish Balay     reason = TAO_CONVERGED_GTTOL;
195694511df7SStefano Zampini   } else if (max_funcs >= 0 && nfuncs > max_funcs) {
19579566063dSJacob Faibussowitsch     PetscCall(PetscInfo(tao, "Exceeded maximum number of function evaluations: %" PetscInt_FMT " > %" PetscInt_FMT "\n", nfuncs, max_funcs));
1958a7e14dcfSSatish Balay     reason = TAO_DIVERGED_MAXFCN;
1959a7e14dcfSSatish Balay   } else if (tao->lsflag != 0) {
19609566063dSJacob Faibussowitsch     PetscCall(PetscInfo(tao, "Tao Line Search failure.\n"));
1961a7e14dcfSSatish Balay     reason = TAO_DIVERGED_LS_FAILURE;
1962a7e14dcfSSatish Balay   } else if (trradius < steptol && niter > 0) {
19639566063dSJacob Faibussowitsch     PetscCall(PetscInfo(tao, "Trust region/step size too small: %g < %g\n", (double)trradius, (double)steptol));
1964a7e14dcfSSatish Balay     reason = TAO_CONVERGED_STEPTOL;
1965e031d6f5SAlp Dener   } else if (niter >= tao->max_it) {
196663a3b9bcSJacob Faibussowitsch     PetscCall(PetscInfo(tao, "Exceeded maximum number of iterations: %" PetscInt_FMT " > %" PetscInt_FMT "\n", niter, tao->max_it));
1967a7e14dcfSSatish Balay     reason = TAO_DIVERGED_MAXITS;
1968a7e14dcfSSatish Balay   } else {
1969a7e14dcfSSatish Balay     reason = TAO_CONTINUE_ITERATING;
1970a7e14dcfSSatish Balay   }
1971a7e14dcfSSatish Balay   tao->reason = reason;
19723ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
1973a7e14dcfSSatish Balay }
1974a7e14dcfSSatish Balay 
1975a7e14dcfSSatish Balay /*@C
1976a7e14dcfSSatish Balay   TaoSetOptionsPrefix - Sets the prefix used for searching for all
197765ba42b6SBarry Smith   Tao options in the database.
1978a7e14dcfSSatish Balay 
1979c3339decSBarry Smith   Logically Collective
1980a7e14dcfSSatish Balay 
1981a7e14dcfSSatish Balay   Input Parameters:
198247450a7bSBarry Smith + tao - the `Tao` context
1983e056e8ceSJacob Faibussowitsch - p   - the prefix string to prepend to all Tao option requests
1984a7e14dcfSSatish Balay 
19852fe279fdSBarry Smith   Level: advanced
19862fe279fdSBarry Smith 
1987a7e14dcfSSatish Balay   Notes:
1988a7e14dcfSSatish Balay   A hyphen (-) must NOT be given at the beginning of the prefix name.
1989a7e14dcfSSatish Balay   The first character of all runtime options is AUTOMATICALLY the hyphen.
1990a7e14dcfSSatish Balay 
1991a7e14dcfSSatish Balay   For example, to distinguish between the runtime options for two
199265ba42b6SBarry Smith   different Tao solvers, one could call
1993a7e14dcfSSatish Balay .vb
1994a7e14dcfSSatish Balay       TaoSetOptionsPrefix(tao1,"sys1_")
1995a7e14dcfSSatish Balay       TaoSetOptionsPrefix(tao2,"sys2_")
1996a7e14dcfSSatish Balay .ve
1997a7e14dcfSSatish Balay 
1998a7e14dcfSSatish Balay   This would enable use of different options for each system, such as
1999a7e14dcfSSatish Balay .vb
20009fa2b5dcSStefano Zampini       -sys1_tao_method blmvm -sys1_tao_grtol 1.e-3
20019fa2b5dcSStefano Zampini       -sys2_tao_method lmvm  -sys2_tao_grtol 1.e-4
2002a7e14dcfSSatish Balay .ve
2003a7e14dcfSSatish Balay 
20041cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoSetFromOptions()`, `TaoAppendOptionsPrefix()`, `TaoGetOptionsPrefix()`
2005a7e14dcfSSatish Balay @*/
2006d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoSetOptionsPrefix(Tao tao, const char p[])
2007d71ae5a4SJacob Faibussowitsch {
2008a7e14dcfSSatish Balay   PetscFunctionBegin;
200994511df7SStefano Zampini   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
20109566063dSJacob Faibussowitsch   PetscCall(PetscObjectSetOptionsPrefix((PetscObject)tao, p));
20111baa6e33SBarry Smith   if (tao->linesearch) PetscCall(TaoLineSearchSetOptionsPrefix(tao->linesearch, p));
20121baa6e33SBarry Smith   if (tao->ksp) PetscCall(KSPSetOptionsPrefix(tao->ksp, p));
20133ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
2014a7e14dcfSSatish Balay }
2015a7e14dcfSSatish Balay 
2016a7e14dcfSSatish Balay /*@C
201747450a7bSBarry Smith   TaoAppendOptionsPrefix - Appends to the prefix used for searching for all Tao options in the database.
2018a7e14dcfSSatish Balay 
2019c3339decSBarry Smith   Logically Collective
2020a7e14dcfSSatish Balay 
2021a7e14dcfSSatish Balay   Input Parameters:
202247450a7bSBarry Smith + tao - the `Tao` solver context
2023e056e8ceSJacob Faibussowitsch - p   - the prefix string to prepend to all `Tao` option requests
2024a7e14dcfSSatish Balay 
20252fe279fdSBarry Smith   Level: advanced
20262fe279fdSBarry Smith 
202765ba42b6SBarry Smith   Note:
2028a7e14dcfSSatish Balay   A hyphen (-) must NOT be given at the beginning of the prefix name.
202965ba42b6SBarry Smith   The first character of all runtime options is automatically the hyphen.
2030a7e14dcfSSatish Balay 
20311cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoSetFromOptions()`, `TaoSetOptionsPrefix()`, `TaoGetOptionsPrefix()`
2032a7e14dcfSSatish Balay @*/
2033d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoAppendOptionsPrefix(Tao tao, const char p[])
2034d71ae5a4SJacob Faibussowitsch {
2035a7e14dcfSSatish Balay   PetscFunctionBegin;
203694511df7SStefano Zampini   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
20379566063dSJacob Faibussowitsch   PetscCall(PetscObjectAppendOptionsPrefix((PetscObject)tao, p));
20381baa6e33SBarry Smith   if (tao->linesearch) PetscCall(PetscObjectAppendOptionsPrefix((PetscObject)tao->linesearch, p));
20391baa6e33SBarry Smith   if (tao->ksp) PetscCall(KSPAppendOptionsPrefix(tao->ksp, p));
20403ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
2041a7e14dcfSSatish Balay }
2042a7e14dcfSSatish Balay 
2043a7e14dcfSSatish Balay /*@C
2044a7e14dcfSSatish Balay   TaoGetOptionsPrefix - Gets the prefix used for searching for all
204565ba42b6SBarry Smith   Tao options in the database
2046a7e14dcfSSatish Balay 
2047a7e14dcfSSatish Balay   Not Collective
2048a7e14dcfSSatish Balay 
204947450a7bSBarry Smith   Input Parameter:
205047450a7bSBarry Smith . tao - the `Tao` context
2051a7e14dcfSSatish Balay 
205247450a7bSBarry Smith   Output Parameter:
2053e056e8ceSJacob Faibussowitsch . p - pointer to the prefix string used is returned
2054a7e14dcfSSatish Balay 
2055e056e8ceSJacob Faibussowitsch   Fortran Notes:
205647450a7bSBarry Smith   Pass in a string 'prefix' of sufficient length to hold the prefix.
2057a7e14dcfSSatish Balay 
2058a7e14dcfSSatish Balay   Level: advanced
2059a7e14dcfSSatish Balay 
20601cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoSetFromOptions()`, `TaoSetOptionsPrefix()`, `TaoAppendOptionsPrefix()`
2061a7e14dcfSSatish Balay @*/
2062d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoGetOptionsPrefix(Tao tao, const char *p[])
2063d71ae5a4SJacob Faibussowitsch {
206494511df7SStefano Zampini   PetscFunctionBegin;
206594511df7SStefano Zampini   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
20669566063dSJacob Faibussowitsch   PetscCall(PetscObjectGetOptionsPrefix((PetscObject)tao, p));
20673ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
2068a7e14dcfSSatish Balay }
2069a7e14dcfSSatish Balay 
2070a7e14dcfSSatish Balay /*@C
207147450a7bSBarry Smith   TaoSetType - Sets the `TaoType` for the minimization solver.
2072a7e14dcfSSatish Balay 
2073c3339decSBarry Smith   Collective
2074a7e14dcfSSatish Balay 
2075a7e14dcfSSatish Balay   Input Parameters:
2076e056e8ceSJacob Faibussowitsch + tao  - the `Tao` solver context
2077a7e14dcfSSatish Balay - type - a known method
2078a7e14dcfSSatish Balay 
2079a7e14dcfSSatish Balay   Options Database Key:
208058417fe7SBarry Smith . -tao_type <type> - Sets the method; use -help for a list
208158417fe7SBarry Smith    of available methods (for instance, "-tao_type lmvm" or "-tao_type tron")
2082a7e14dcfSSatish Balay 
2083a7e14dcfSSatish Balay   Level: intermediate
2084a7e14dcfSSatish Balay 
20851cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoCreate()`, `TaoGetType()`, `TaoType`
2086a7e14dcfSSatish Balay @*/
2087d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoSetType(Tao tao, TaoType type)
2088d71ae5a4SJacob Faibussowitsch {
2089441846f8SBarry Smith   PetscErrorCode (*create_xxx)(Tao);
2090a7e14dcfSSatish Balay   PetscBool issame;
2091a7e14dcfSSatish Balay 
2092a7e14dcfSSatish Balay   PetscFunctionBegin;
2093441846f8SBarry Smith   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
2094a7e14dcfSSatish Balay 
20959566063dSJacob Faibussowitsch   PetscCall(PetscObjectTypeCompare((PetscObject)tao, type, &issame));
20963ba16761SJacob Faibussowitsch   if (issame) PetscFunctionReturn(PETSC_SUCCESS);
2097a7e14dcfSSatish Balay 
20989566063dSJacob Faibussowitsch   PetscCall(PetscFunctionListFind(TaoList, type, (void (**)(void)) & create_xxx));
20993c859ba3SBarry Smith   PetscCheck(create_xxx, PetscObjectComm((PetscObject)tao), PETSC_ERR_ARG_UNKNOWN_TYPE, "Unable to find requested Tao type %s", type);
2100a7e14dcfSSatish Balay 
2101a7e14dcfSSatish Balay   /* Destroy the existing solver information */
2102dbbe0bcdSBarry Smith   PetscTryTypeMethod(tao, destroy);
21031baa6e33SBarry Smith   PetscCall(KSPDestroy(&tao->ksp));
21049566063dSJacob Faibussowitsch   PetscCall(TaoLineSearchDestroy(&tao->linesearch));
210583c8fe1dSLisandro Dalcin   tao->ops->setup          = NULL;
210683c8fe1dSLisandro Dalcin   tao->ops->solve          = NULL;
210783c8fe1dSLisandro Dalcin   tao->ops->view           = NULL;
210883c8fe1dSLisandro Dalcin   tao->ops->setfromoptions = NULL;
210983c8fe1dSLisandro Dalcin   tao->ops->destroy        = NULL;
2110a7e14dcfSSatish Balay 
2111a7e14dcfSSatish Balay   tao->setupcalled = PETSC_FALSE;
2112a7e14dcfSSatish Balay 
21139566063dSJacob Faibussowitsch   PetscCall((*create_xxx)(tao));
21149566063dSJacob Faibussowitsch   PetscCall(PetscObjectChangeTypeName((PetscObject)tao, type));
21153ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
2116a7e14dcfSSatish Balay }
211747a47007SBarry Smith 
211867be906fSBarry Smith /*@C
211947450a7bSBarry Smith   TaoRegister - Adds a method to the Tao package for minimization.
2120a7e14dcfSSatish Balay 
212120f4b53cSBarry Smith   Not Collective
2122a7e14dcfSSatish Balay 
2123a7e14dcfSSatish Balay   Input Parameters:
2124a7e14dcfSSatish Balay + sname - name of a new user-defined solver
2125a7e14dcfSSatish Balay - func  - routine to Create method context
2126a7e14dcfSSatish Balay 
2127e056e8ceSJacob Faibussowitsch   Example Usage:
2128a7e14dcfSSatish Balay .vb
2129441846f8SBarry Smith    TaoRegister("my_solver", MySolverCreate);
2130a7e14dcfSSatish Balay .ve
2131a7e14dcfSSatish Balay 
2132a7e14dcfSSatish Balay   Then, your solver can be chosen with the procedural interface via
2133a7e14dcfSSatish Balay $     TaoSetType(tao, "my_solver")
2134a7e14dcfSSatish Balay   or at runtime via the option
213558417fe7SBarry Smith $     -tao_type my_solver
2136a7e14dcfSSatish Balay 
2137a7e14dcfSSatish Balay   Level: advanced
2138a7e14dcfSSatish Balay 
213967be906fSBarry Smith   Note:
214067be906fSBarry Smith   `TaoRegister()` may be called multiple times to add several user-defined solvers.
214167be906fSBarry Smith 
21421cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoSetType()`, `TaoRegisterAll()`, `TaoRegisterDestroy()`
214367be906fSBarry Smith @*/
2144d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoRegister(const char sname[], PetscErrorCode (*func)(Tao))
2145d71ae5a4SJacob Faibussowitsch {
2146a7e14dcfSSatish Balay   PetscFunctionBegin;
21479566063dSJacob Faibussowitsch   PetscCall(TaoInitializePackage());
21489566063dSJacob Faibussowitsch   PetscCall(PetscFunctionListAdd(&TaoList, sname, (void (*)(void))func));
21493ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
2150a7e14dcfSSatish Balay }
2151a7e14dcfSSatish Balay 
2152a7e14dcfSSatish Balay /*@C
2153441846f8SBarry Smith   TaoRegisterDestroy - Frees the list of minimization solvers that were
215447450a7bSBarry Smith   registered by `TaoRegister()`.
2155a7e14dcfSSatish Balay 
2156a7e14dcfSSatish Balay   Not Collective
2157a7e14dcfSSatish Balay 
2158a7e14dcfSSatish Balay   Level: advanced
2159a7e14dcfSSatish Balay 
21601cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoRegisterAll()`, `TaoRegister()`
2161a7e14dcfSSatish Balay @*/
2162d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoRegisterDestroy(void)
2163d71ae5a4SJacob Faibussowitsch {
2164a7e14dcfSSatish Balay   PetscFunctionBegin;
21659566063dSJacob Faibussowitsch   PetscCall(PetscFunctionListDestroy(&TaoList));
2166441846f8SBarry Smith   TaoRegisterAllCalled = PETSC_FALSE;
21673ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
2168a7e14dcfSSatish Balay }
216947a47007SBarry Smith 
21708931d482SJason Sarich /*@
217147450a7bSBarry Smith   TaoGetIterationNumber - Gets the number of `TaoSolve()` iterations completed
21728931d482SJason Sarich   at this time.
21738931d482SJason Sarich 
21748931d482SJason Sarich   Not Collective
21758931d482SJason Sarich 
21768931d482SJason Sarich   Input Parameter:
217747450a7bSBarry Smith . tao - the `Tao` context
21788931d482SJason Sarich 
21798931d482SJason Sarich   Output Parameter:
21808931d482SJason Sarich . iter - iteration number
21818931d482SJason Sarich 
21828931d482SJason Sarich   Notes:
21838931d482SJason Sarich   For example, during the computation of iteration 2 this would return 1.
21848931d482SJason Sarich 
21858931d482SJason Sarich   Level: intermediate
21868931d482SJason Sarich 
21871cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoGetLinearSolveIterations()`, `TaoGetResidualNorm()`, `TaoGetObjective()`
21888931d482SJason Sarich @*/
2189d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoGetIterationNumber(Tao tao, PetscInt *iter)
2190d71ae5a4SJacob Faibussowitsch {
21918931d482SJason Sarich   PetscFunctionBegin;
21928931d482SJason Sarich   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
21934f572ea9SToby Isaac   PetscAssertPointer(iter, 2);
21948931d482SJason Sarich   *iter = tao->niter;
21953ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
21968931d482SJason Sarich }
21978931d482SJason Sarich 
21988931d482SJason Sarich /*@
219947450a7bSBarry Smith   TaoGetResidualNorm - Gets the current value of the norm of the residual (gradient)
220079f5d8caSBarry Smith   at this time.
220179f5d8caSBarry Smith 
220279f5d8caSBarry Smith   Not Collective
220379f5d8caSBarry Smith 
220479f5d8caSBarry Smith   Input Parameter:
220547450a7bSBarry Smith . tao - the `Tao` context
220679f5d8caSBarry Smith 
220779f5d8caSBarry Smith   Output Parameter:
220879f5d8caSBarry Smith . value - the current value
220979f5d8caSBarry Smith 
221079f5d8caSBarry Smith   Level: intermediate
221179f5d8caSBarry Smith 
2212e056e8ceSJacob Faibussowitsch   Developer Notes:
221367be906fSBarry Smith   This is the 2-norm of the residual, we cannot use `TaoGetGradientNorm()` because that has
221447450a7bSBarry Smith   a different meaning. For some reason `Tao` sometimes calls the gradient the residual.
221579f5d8caSBarry Smith 
22161cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoGetLinearSolveIterations()`, `TaoGetIterationNumber()`, `TaoGetObjective()`
221779f5d8caSBarry Smith @*/
2218d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoGetResidualNorm(Tao tao, PetscReal *value)
2219d71ae5a4SJacob Faibussowitsch {
222079f5d8caSBarry Smith   PetscFunctionBegin;
222179f5d8caSBarry Smith   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
22224f572ea9SToby Isaac   PetscAssertPointer(value, 2);
222379f5d8caSBarry Smith   *value = tao->residual;
22243ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
222579f5d8caSBarry Smith }
222679f5d8caSBarry Smith 
222779f5d8caSBarry Smith /*@
22288931d482SJason Sarich   TaoSetIterationNumber - Sets the current iteration number.
22298931d482SJason Sarich 
2230c3339decSBarry Smith   Logically Collective
22318931d482SJason Sarich 
2232d8d19677SJose E. Roman   Input Parameters:
223347450a7bSBarry Smith + tao  - the `Tao` context
2234a2b725a8SWilliam Gropp - iter - iteration number
22358931d482SJason Sarich 
22368931d482SJason Sarich   Level: developer
22378931d482SJason Sarich 
22381cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoGetLinearSolveIterations()`
22398931d482SJason Sarich @*/
2240d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoSetIterationNumber(Tao tao, PetscInt iter)
2241d71ae5a4SJacob Faibussowitsch {
22428931d482SJason Sarich   PetscFunctionBegin;
22438931d482SJason Sarich   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
224494511df7SStefano Zampini   PetscValidLogicalCollectiveInt(tao, iter, 2);
22459566063dSJacob Faibussowitsch   PetscCall(PetscObjectSAWsTakeAccess((PetscObject)tao));
22468931d482SJason Sarich   tao->niter = iter;
22479566063dSJacob Faibussowitsch   PetscCall(PetscObjectSAWsGrantAccess((PetscObject)tao));
22483ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
22498931d482SJason Sarich }
22508931d482SJason Sarich 
22518931d482SJason Sarich /*@
225247450a7bSBarry Smith   TaoGetTotalIterationNumber - Gets the total number of `TaoSolve()` iterations
22538931d482SJason Sarich   completed. This number keeps accumulating if multiple solves
225447450a7bSBarry Smith   are called with the `Tao` object.
22558931d482SJason Sarich 
22568931d482SJason Sarich   Not Collective
22578931d482SJason Sarich 
22588931d482SJason Sarich   Input Parameter:
225947450a7bSBarry Smith . tao - the `Tao` context
22608931d482SJason Sarich 
22618931d482SJason Sarich   Output Parameter:
226247450a7bSBarry Smith . iter - number of iterations
22638931d482SJason Sarich 
22648931d482SJason Sarich   Level: intermediate
22658931d482SJason Sarich 
226620f4b53cSBarry Smith   Note:
226767be906fSBarry Smith   The total iteration count is updated after each solve, if there is a current
226847450a7bSBarry Smith   `TaoSolve()` in progress then those iterations are not included in the count
226967be906fSBarry Smith 
22701cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoGetLinearSolveIterations()`
22718931d482SJason Sarich @*/
2272d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoGetTotalIterationNumber(Tao tao, PetscInt *iter)
2273d71ae5a4SJacob Faibussowitsch {
22748931d482SJason Sarich   PetscFunctionBegin;
22758931d482SJason Sarich   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
22764f572ea9SToby Isaac   PetscAssertPointer(iter, 2);
22778931d482SJason Sarich   *iter = tao->ntotalits;
22783ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
22798931d482SJason Sarich }
22808931d482SJason Sarich 
22818931d482SJason Sarich /*@
22828931d482SJason Sarich   TaoSetTotalIterationNumber - Sets the current total iteration number.
22838931d482SJason Sarich 
2284c3339decSBarry Smith   Logically Collective
22858931d482SJason Sarich 
2286d8d19677SJose E. Roman   Input Parameters:
228747450a7bSBarry Smith + tao  - the `Tao` context
228847450a7bSBarry Smith - iter - the iteration number
22898931d482SJason Sarich 
22908931d482SJason Sarich   Level: developer
22918931d482SJason Sarich 
22921cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoGetLinearSolveIterations()`
22938931d482SJason Sarich @*/
2294d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoSetTotalIterationNumber(Tao tao, PetscInt iter)
2295d71ae5a4SJacob Faibussowitsch {
22968931d482SJason Sarich   PetscFunctionBegin;
22978931d482SJason Sarich   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
229894511df7SStefano Zampini   PetscValidLogicalCollectiveInt(tao, iter, 2);
22999566063dSJacob Faibussowitsch   PetscCall(PetscObjectSAWsTakeAccess((PetscObject)tao));
23008931d482SJason Sarich   tao->ntotalits = iter;
23019566063dSJacob Faibussowitsch   PetscCall(PetscObjectSAWsGrantAccess((PetscObject)tao));
23023ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
23038931d482SJason Sarich }
23048931d482SJason Sarich 
2305a7e14dcfSSatish Balay /*@
230647450a7bSBarry Smith   TaoSetConvergedReason - Sets the termination flag on a `Tao` object
2307a7e14dcfSSatish Balay 
2308c3339decSBarry Smith   Logically Collective
2309a7e14dcfSSatish Balay 
2310a7e14dcfSSatish Balay   Input Parameters:
231147450a7bSBarry Smith + tao    - the `Tao` context
231247450a7bSBarry Smith - reason - the `TaoConvergedReason`
2313a7e14dcfSSatish Balay 
2314a7e14dcfSSatish Balay   Level: intermediate
2315a7e14dcfSSatish Balay 
23161cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoConvergedReason`
2317a7e14dcfSSatish Balay @*/
2318d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoSetConvergedReason(Tao tao, TaoConvergedReason reason)
2319d71ae5a4SJacob Faibussowitsch {
2320a7e14dcfSSatish Balay   PetscFunctionBegin;
232194511df7SStefano Zampini   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
232294511df7SStefano Zampini   PetscValidLogicalCollectiveEnum(tao, reason, 2);
2323a7e14dcfSSatish Balay   tao->reason = reason;
23243ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
2325a7e14dcfSSatish Balay }
2326a7e14dcfSSatish Balay 
2327a7e14dcfSSatish Balay /*@
232847450a7bSBarry Smith   TaoGetConvergedReason - Gets the reason the `TaoSolve()` was stopped.
2329a7e14dcfSSatish Balay 
2330a7e14dcfSSatish Balay   Not Collective
2331a7e14dcfSSatish Balay 
2332a7e14dcfSSatish Balay   Input Parameter:
233347450a7bSBarry Smith . tao - the `Tao` solver context
2334a7e14dcfSSatish Balay 
2335a7e14dcfSSatish Balay   Output Parameter:
233647450a7bSBarry Smith . reason - value of `TaoConvergedReason`
2337a7e14dcfSSatish Balay 
2338a7e14dcfSSatish Balay   Level: intermediate
2339a7e14dcfSSatish Balay 
23401cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoConvergedReason`, `TaoSetConvergenceTest()`, `TaoSetTolerances()`
2341a7e14dcfSSatish Balay @*/
2342d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoGetConvergedReason(Tao tao, TaoConvergedReason *reason)
2343d71ae5a4SJacob Faibussowitsch {
2344a7e14dcfSSatish Balay   PetscFunctionBegin;
2345441846f8SBarry Smith   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
23464f572ea9SToby Isaac   PetscAssertPointer(reason, 2);
2347a7e14dcfSSatish Balay   *reason = tao->reason;
23483ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
2349a7e14dcfSSatish Balay }
2350a7e14dcfSSatish Balay 
2351a7e14dcfSSatish Balay /*@
2352a7e14dcfSSatish Balay   TaoGetSolutionStatus - Get the current iterate, objective value,
235347450a7bSBarry Smith   residual, infeasibility, and termination from a `Tao` object
2354a7e14dcfSSatish Balay 
2355a7e14dcfSSatish Balay   Not Collective
2356a7e14dcfSSatish Balay 
2357f899ff85SJose E. Roman   Input Parameter:
235847450a7bSBarry Smith . tao - the `Tao` context
2359a7e14dcfSSatish Balay 
2360a7e14dcfSSatish Balay   Output Parameters:
2361e056e8ceSJacob Faibussowitsch + its    - the current iterate number (>=0)
2362a7e14dcfSSatish Balay . f      - the current function value
236358417fe7SBarry Smith . gnorm  - the square of the gradient norm, duality gap, or other measure indicating distance from optimality.
2364a7e14dcfSSatish Balay . cnorm  - the infeasibility of the current solution with regard to the constraints.
2365a7e14dcfSSatish Balay . xdiff  - the step length or trust region radius of the most recent iterate.
236665ba42b6SBarry Smith - reason - The termination reason, which can equal `TAO_CONTINUE_ITERATING`
2367a7e14dcfSSatish Balay 
2368a7e14dcfSSatish Balay   Level: intermediate
2369a7e14dcfSSatish Balay 
237065ba42b6SBarry Smith   Notes:
237165ba42b6SBarry Smith   Tao returns the values set by the solvers in the routine `TaoMonitor()`.
2372a7e14dcfSSatish Balay 
237367be906fSBarry Smith   If any of the output arguments are set to `NULL`, no corresponding value will be returned.
2374a7e14dcfSSatish Balay 
23751cc06b55SBarry Smith .seealso: [](ch_tao), `TaoMonitor()`, `TaoGetConvergedReason()`
2376a7e14dcfSSatish Balay @*/
2377d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoGetSolutionStatus(Tao tao, PetscInt *its, PetscReal *f, PetscReal *gnorm, PetscReal *cnorm, PetscReal *xdiff, TaoConvergedReason *reason)
2378d71ae5a4SJacob Faibussowitsch {
2379a7e14dcfSSatish Balay   PetscFunctionBegin;
238094511df7SStefano Zampini   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
2381a7e14dcfSSatish Balay   if (its) *its = tao->niter;
2382a7e14dcfSSatish Balay   if (f) *f = tao->fc;
2383a7e14dcfSSatish Balay   if (gnorm) *gnorm = tao->residual;
2384a7e14dcfSSatish Balay   if (cnorm) *cnorm = tao->cnorm;
2385a7e14dcfSSatish Balay   if (reason) *reason = tao->reason;
2386a7e14dcfSSatish Balay   if (xdiff) *xdiff = tao->step;
23873ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
2388a7e14dcfSSatish Balay }
2389a7e14dcfSSatish Balay 
2390a7e14dcfSSatish Balay /*@C
239147450a7bSBarry Smith   TaoGetType - Gets the current `TaoType` being used in the `Tao` object
2392a7e14dcfSSatish Balay 
2393a7e14dcfSSatish Balay   Not Collective
2394a7e14dcfSSatish Balay 
2395a7e14dcfSSatish Balay   Input Parameter:
239647450a7bSBarry Smith . tao - the `Tao` solver context
2397a7e14dcfSSatish Balay 
2398a7e14dcfSSatish Balay   Output Parameter:
239947450a7bSBarry Smith . type - the `TaoType`
2400a7e14dcfSSatish Balay 
2401a7e14dcfSSatish Balay   Level: intermediate
2402a7e14dcfSSatish Balay 
24031cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoType`, `TaoSetType()`
2404a7e14dcfSSatish Balay @*/
2405d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoGetType(Tao tao, TaoType *type)
2406d71ae5a4SJacob Faibussowitsch {
2407a7e14dcfSSatish Balay   PetscFunctionBegin;
2408441846f8SBarry Smith   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
24094f572ea9SToby Isaac   PetscAssertPointer(type, 2);
2410a7e14dcfSSatish Balay   *type = ((PetscObject)tao)->type_name;
24113ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
2412a7e14dcfSSatish Balay }
2413a7e14dcfSSatish Balay 
2414a7e14dcfSSatish Balay /*@C
2415a7e14dcfSSatish Balay   TaoMonitor - Monitor the solver and the current solution.  This
2416a7e14dcfSSatish Balay   routine will record the iteration number and residual statistics,
2417670c031eSStefano Zampini   and call any monitors specified by the user.
2418a7e14dcfSSatish Balay 
2419a7e14dcfSSatish Balay   Input Parameters:
242047450a7bSBarry Smith + tao        - the `Tao` context
2421a7e14dcfSSatish Balay . its        - the current iterate number (>=0)
2422a7e14dcfSSatish Balay . f          - the current objective function value
24231cb3f120SPierre Jolivet . res        - the gradient norm, square root of the duality gap, or other measure indicating distance from optimality.  This measure will be recorded and
2424a7e14dcfSSatish Balay           used for some termination tests.
2425a7e14dcfSSatish Balay . cnorm      - the infeasibility of the current solution with regard to the constraints.
2426a7e14dcfSSatish Balay - steplength - multiple of the step direction added to the previous iterate.
2427a7e14dcfSSatish Balay 
2428a7e14dcfSSatish Balay   Options Database Key:
2429a7e14dcfSSatish Balay . -tao_monitor - Use the default monitor, which prints statistics to standard output
2430a7e14dcfSSatish Balay 
2431a7e14dcfSSatish Balay   Level: developer
2432a7e14dcfSSatish Balay 
24331cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoGetConvergedReason()`, `TaoMonitorDefault()`, `TaoSetMonitor()`
2434a7e14dcfSSatish Balay @*/
2435d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoMonitor(Tao tao, PetscInt its, PetscReal f, PetscReal res, PetscReal cnorm, PetscReal steplength)
2436d71ae5a4SJacob Faibussowitsch {
243747a47007SBarry Smith   PetscInt i;
243847a47007SBarry Smith 
2439a7e14dcfSSatish Balay   PetscFunctionBegin;
2440441846f8SBarry Smith   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
2441a7e14dcfSSatish Balay   tao->fc       = f;
2442a7e14dcfSSatish Balay   tao->residual = res;
2443a7e14dcfSSatish Balay   tao->cnorm    = cnorm;
2444a7e14dcfSSatish Balay   tao->step     = steplength;
2445e52336cbSBarry Smith   if (!its) {
244694511df7SStefano Zampini     tao->cnorm0 = cnorm;
244794511df7SStefano Zampini     tao->gnorm0 = res;
2448a7e14dcfSSatish Balay   }
2449bfdcf5a2SStefano Zampini   PetscCall(VecLockReadPush(tao->solution));
245048a46eb9SPierre Jolivet   for (i = 0; i < tao->numbermonitors; i++) PetscCall((*tao->monitor[i])(tao, tao->monitorcontext[i]));
2451bfdcf5a2SStefano Zampini   PetscCall(VecLockReadPop(tao->solution));
24523ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
2453a7e14dcfSSatish Balay }
2454a7e14dcfSSatish Balay 
2455a7e14dcfSSatish Balay /*@
2456ae93cb3cSJason Sarich   TaoSetConvergenceHistory - Sets the array used to hold the convergence history.
2457a7e14dcfSSatish Balay 
2458c3339decSBarry Smith   Logically Collective
2459a7e14dcfSSatish Balay 
2460a7e14dcfSSatish Balay   Input Parameters:
246147450a7bSBarry Smith + tao   - the `Tao` solver context
2462a7e14dcfSSatish Balay . obj   - array to hold objective value history
2463a7e14dcfSSatish Balay . resid - array to hold residual history
2464a7e14dcfSSatish Balay . cnorm - array to hold constraint violation history
2465be1558d8SJason Sarich . lits  - integer array holds the number of linear iterations for each Tao iteration
246667be906fSBarry Smith . na    - size of `obj`, `resid`, and `cnorm`
246765ba42b6SBarry Smith - reset - `PETSC_TRUE` indicates each new minimization resets the history counter to zero,
2468a7e14dcfSSatish Balay            else it continues storing new values for new minimizations after the old ones
2469a7e14dcfSSatish Balay 
247067be906fSBarry Smith   Level: intermediate
247167be906fSBarry Smith 
2472a7e14dcfSSatish Balay   Notes:
247347450a7bSBarry Smith   If set, `Tao` will fill the given arrays with the indicated
2474ae93cb3cSJason Sarich   information at each iteration.  If 'obj','resid','cnorm','lits' are
247567be906fSBarry Smith   *all* `NULL` then space (using size `na`, or 1000 if na is `PETSC_DECIDE` or
247665ba42b6SBarry Smith   `PETSC_DEFAULT`) is allocated for the history.
247767be906fSBarry Smith   If not all are `NULL`, then only the non-`NULL` information categories
2478ae93cb3cSJason Sarich   will be stored, the others will be ignored.
2479ae93cb3cSJason Sarich 
2480ae93cb3cSJason Sarich   Any convergence information after iteration number 'na' will not be stored.
2481a7e14dcfSSatish Balay 
2482a7e14dcfSSatish Balay   This routine is useful, e.g., when running a code for purposes
2483a7e14dcfSSatish Balay   of accurate performance monitoring, when no I/O should be done
2484a7e14dcfSSatish Balay   during the section of code that is being timed.
2485a7e14dcfSSatish Balay 
24861cc06b55SBarry Smith .seealso: [](ch_tao), `TaoGetConvergenceHistory()`
2487a7e14dcfSSatish Balay @*/
2488d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoSetConvergenceHistory(Tao tao, PetscReal obj[], PetscReal resid[], PetscReal cnorm[], PetscInt lits[], PetscInt na, PetscBool reset)
2489d71ae5a4SJacob Faibussowitsch {
2490a7e14dcfSSatish Balay   PetscFunctionBegin;
2491441846f8SBarry Smith   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
24924f572ea9SToby Isaac   if (obj) PetscAssertPointer(obj, 2);
24934f572ea9SToby Isaac   if (resid) PetscAssertPointer(resid, 3);
24944f572ea9SToby Isaac   if (cnorm) PetscAssertPointer(cnorm, 4);
24954f572ea9SToby Isaac   if (lits) PetscAssertPointer(lits, 5);
2496ae93cb3cSJason Sarich 
2497ae93cb3cSJason Sarich   if (na == PETSC_DECIDE || na == PETSC_DEFAULT) na = 1000;
2498ae93cb3cSJason Sarich   if (!obj && !resid && !cnorm && !lits) {
24999566063dSJacob Faibussowitsch     PetscCall(PetscCalloc4(na, &obj, na, &resid, na, &cnorm, na, &lits));
2500ae93cb3cSJason Sarich     tao->hist_malloc = PETSC_TRUE;
2501ae93cb3cSJason Sarich   }
2502ae93cb3cSJason Sarich 
2503a7e14dcfSSatish Balay   tao->hist_obj   = obj;
2504a7e14dcfSSatish Balay   tao->hist_resid = resid;
2505a7e14dcfSSatish Balay   tao->hist_cnorm = cnorm;
2506ae93cb3cSJason Sarich   tao->hist_lits  = lits;
2507a7e14dcfSSatish Balay   tao->hist_max   = na;
2508a7e14dcfSSatish Balay   tao->hist_reset = reset;
2509ae93cb3cSJason Sarich   tao->hist_len   = 0;
25103ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
2511a7e14dcfSSatish Balay }
2512a7e14dcfSSatish Balay 
2513a7e14dcfSSatish Balay /*@C
251465ba42b6SBarry Smith   TaoGetConvergenceHistory - Gets the arrays used that hold the convergence history.
2515a7e14dcfSSatish Balay 
2516c3339decSBarry Smith   Collective
2517a7e14dcfSSatish Balay 
2518a7e14dcfSSatish Balay   Input Parameter:
251947450a7bSBarry Smith . tao - the `Tao` context
2520a7e14dcfSSatish Balay 
2521a7e14dcfSSatish Balay   Output Parameters:
2522a7e14dcfSSatish Balay + obj   - array used to hold objective value history
2523a7e14dcfSSatish Balay . resid - array used to hold residual history
2524a7e14dcfSSatish Balay . cnorm - array used to hold constraint violation history
2525ae93cb3cSJason Sarich . lits  - integer array used to hold linear solver iteration count
252667be906fSBarry Smith - nhist - size of `obj`, `resid`, `cnorm`, and `lits`
252767be906fSBarry Smith 
252867be906fSBarry Smith   Level: advanced
2529a7e14dcfSSatish Balay 
2530a7e14dcfSSatish Balay   Notes:
253165ba42b6SBarry Smith   This routine must be preceded by calls to `TaoSetConvergenceHistory()`
253265ba42b6SBarry Smith   and `TaoSolve()`, otherwise it returns useless information.
2533ae93cb3cSJason Sarich 
2534a7e14dcfSSatish Balay   This routine is useful, e.g., when running a code for purposes
2535a7e14dcfSSatish Balay   of accurate performance monitoring, when no I/O should be done
2536a7e14dcfSSatish Balay   during the section of code that is being timed.
2537a7e14dcfSSatish Balay 
2538e056e8ceSJacob Faibussowitsch   Fortran Notes:
253967be906fSBarry Smith   The calling sequence is
254067be906fSBarry Smith .vb
254167be906fSBarry Smith    call TaoGetConvergenceHistory(Tao tao, PetscInt nhist, PetscErrorCode ierr)
254267be906fSBarry Smith .ve
2543a7e14dcfSSatish Balay 
25441cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoSolve()`, `TaoSetConvergenceHistory()`
2545a7e14dcfSSatish Balay @*/
2546d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoGetConvergenceHistory(Tao tao, PetscReal **obj, PetscReal **resid, PetscReal **cnorm, PetscInt **lits, PetscInt *nhist)
2547d71ae5a4SJacob Faibussowitsch {
2548a7e14dcfSSatish Balay   PetscFunctionBegin;
2549441846f8SBarry Smith   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
2550a7e14dcfSSatish Balay   if (obj) *obj = tao->hist_obj;
2551a7e14dcfSSatish Balay   if (cnorm) *cnorm = tao->hist_cnorm;
2552a7e14dcfSSatish Balay   if (resid) *resid = tao->hist_resid;
25535b494b05SBarry Smith   if (lits) *lits = tao->hist_lits;
2554a7e14dcfSSatish Balay   if (nhist) *nhist = tao->hist_len;
25553ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
2556a7e14dcfSSatish Balay }
2557a7e14dcfSSatish Balay 
2558a7e14dcfSSatish Balay /*@
255947450a7bSBarry Smith   TaoSetApplicationContext - Sets the optional user-defined context for a `Tao` solver.
2560a7e14dcfSSatish Balay 
2561c3339decSBarry Smith   Logically Collective
2562a7e14dcfSSatish Balay 
2563a7e14dcfSSatish Balay   Input Parameters:
256447450a7bSBarry Smith + tao  - the `Tao` context
2565a7e14dcfSSatish Balay - usrP - optional user context
2566a7e14dcfSSatish Balay 
2567a7e14dcfSSatish Balay   Level: intermediate
2568a7e14dcfSSatish Balay 
256942747ad1SJacob Faibussowitsch .seealso: [](ch_tao), `Tao`, `TaoGetApplicationContext()`
2570a7e14dcfSSatish Balay @*/
2571d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoSetApplicationContext(Tao tao, void *usrP)
2572d71ae5a4SJacob Faibussowitsch {
2573a7e14dcfSSatish Balay   PetscFunctionBegin;
2574441846f8SBarry Smith   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
2575a7e14dcfSSatish Balay   tao->user = usrP;
25763ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
2577a7e14dcfSSatish Balay }
2578a7e14dcfSSatish Balay 
2579a7e14dcfSSatish Balay /*@
258047450a7bSBarry Smith   TaoGetApplicationContext - Gets the user-defined context for a `Tao` solver
2581a7e14dcfSSatish Balay 
2582a7e14dcfSSatish Balay   Not Collective
2583a7e14dcfSSatish Balay 
2584a7e14dcfSSatish Balay   Input Parameter:
258547450a7bSBarry Smith . tao - the `Tao` context
2586a7e14dcfSSatish Balay 
2587a7e14dcfSSatish Balay   Output Parameter:
2588a7e14dcfSSatish Balay . usrP - user context
2589a7e14dcfSSatish Balay 
2590a7e14dcfSSatish Balay   Level: intermediate
2591a7e14dcfSSatish Balay 
25921cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoSetApplicationContext()`
2593a7e14dcfSSatish Balay @*/
2594d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoGetApplicationContext(Tao tao, void *usrP)
2595d71ae5a4SJacob Faibussowitsch {
2596a7e14dcfSSatish Balay   PetscFunctionBegin;
2597441846f8SBarry Smith   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
25984f572ea9SToby Isaac   PetscAssertPointer(usrP, 2);
2599a7e14dcfSSatish Balay   *(void **)usrP = tao->user;
26003ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
2601a7e14dcfSSatish Balay }
2602a9603a14SPatrick Farrell 
2603a9603a14SPatrick Farrell /*@
260465ba42b6SBarry Smith   TaoSetGradientNorm - Sets the matrix used to define the norm that measures the size of the gradient.
2605a9603a14SPatrick Farrell 
2606c3339decSBarry Smith   Collective
2607a9603a14SPatrick Farrell 
2608a9603a14SPatrick Farrell   Input Parameters:
260947450a7bSBarry Smith + tao - the `Tao` context
261065ba42b6SBarry Smith - M   - matrix that defines the norm
2611a9603a14SPatrick Farrell 
2612a9603a14SPatrick Farrell   Level: beginner
2613a9603a14SPatrick Farrell 
26141cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoGetGradientNorm()`, `TaoGradientNorm()`
2615a9603a14SPatrick Farrell @*/
2616d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoSetGradientNorm(Tao tao, Mat M)
2617d71ae5a4SJacob Faibussowitsch {
2618a9603a14SPatrick Farrell   PetscFunctionBegin;
2619a9603a14SPatrick Farrell   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
262094511df7SStefano Zampini   PetscValidHeaderSpecific(M, MAT_CLASSID, 2);
26219566063dSJacob Faibussowitsch   PetscCall(PetscObjectReference((PetscObject)M));
26229566063dSJacob Faibussowitsch   PetscCall(MatDestroy(&tao->gradient_norm));
26239566063dSJacob Faibussowitsch   PetscCall(VecDestroy(&tao->gradient_norm_tmp));
2624a9603a14SPatrick Farrell   tao->gradient_norm = M;
26259566063dSJacob Faibussowitsch   PetscCall(MatCreateVecs(M, NULL, &tao->gradient_norm_tmp));
26263ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
2627a9603a14SPatrick Farrell }
2628a9603a14SPatrick Farrell 
2629a9603a14SPatrick Farrell /*@
263065ba42b6SBarry Smith   TaoGetGradientNorm - Returns the matrix used to define the norm used for measuring the size of the gradient.
2631a9603a14SPatrick Farrell 
2632a9603a14SPatrick Farrell   Not Collective
2633a9603a14SPatrick Farrell 
2634a9603a14SPatrick Farrell   Input Parameter:
263547450a7bSBarry Smith . tao - the `Tao` context
2636a9603a14SPatrick Farrell 
2637a9603a14SPatrick Farrell   Output Parameter:
2638a9603a14SPatrick Farrell . M - gradient norm
2639a9603a14SPatrick Farrell 
2640a9603a14SPatrick Farrell   Level: beginner
2641a9603a14SPatrick Farrell 
26421cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoSetGradientNorm()`, `TaoGradientNorm()`
2643a9603a14SPatrick Farrell @*/
2644d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoGetGradientNorm(Tao tao, Mat *M)
2645d71ae5a4SJacob Faibussowitsch {
2646a9603a14SPatrick Farrell   PetscFunctionBegin;
2647a9603a14SPatrick Farrell   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
26484f572ea9SToby Isaac   PetscAssertPointer(M, 2);
2649a9603a14SPatrick Farrell   *M = tao->gradient_norm;
26503ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
2651a9603a14SPatrick Farrell }
2652a9603a14SPatrick Farrell 
265305aca2cdSSatish Balay /*@C
265447450a7bSBarry Smith   TaoGradientNorm - Compute the norm using the `NormType`, the user has selected
2655a9603a14SPatrick Farrell 
2656c3339decSBarry Smith   Collective
2657a9603a14SPatrick Farrell 
2658d8d19677SJose E. Roman   Input Parameters:
265947450a7bSBarry Smith + tao      - the `Tao` context
2660a9603a14SPatrick Farrell . gradient - the gradient to be computed
26613b242c63SJacob Faibussowitsch - type     - the norm type
2662a9603a14SPatrick Farrell 
2663a9603a14SPatrick Farrell   Output Parameter:
2664a9603a14SPatrick Farrell . gnorm - the gradient norm
2665a9603a14SPatrick Farrell 
266647450a7bSBarry Smith   Level: advanced
2667a9603a14SPatrick Farrell 
26681cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoSetGradientNorm()`, `TaoGetGradientNorm()`
2669a9603a14SPatrick Farrell @*/
2670d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoGradientNorm(Tao tao, Vec gradient, NormType type, PetscReal *gnorm)
2671d71ae5a4SJacob Faibussowitsch {
2672a9603a14SPatrick Farrell   PetscFunctionBegin;
26738854b543SStefano Zampini   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
26748854b543SStefano Zampini   PetscValidHeaderSpecific(gradient, VEC_CLASSID, 2);
26758854b543SStefano Zampini   PetscValidLogicalCollectiveEnum(tao, type, 3);
26764f572ea9SToby Isaac   PetscAssertPointer(gnorm, 4);
2677a9603a14SPatrick Farrell   if (tao->gradient_norm) {
2678680e2bc7SPatrick Farrell     PetscScalar gnorms;
2679680e2bc7SPatrick Farrell 
26803c859ba3SBarry 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.");
26819566063dSJacob Faibussowitsch     PetscCall(MatMult(tao->gradient_norm, gradient, tao->gradient_norm_tmp));
26829566063dSJacob Faibussowitsch     PetscCall(VecDot(gradient, tao->gradient_norm_tmp, &gnorms));
26837bb79937SPatrick Farrell     *gnorm = PetscRealPart(PetscSqrtScalar(gnorms));
2684a9603a14SPatrick Farrell   } else {
26859566063dSJacob Faibussowitsch     PetscCall(VecNorm(gradient, type, gnorm));
2686a9603a14SPatrick Farrell   }
26873ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
2688a9603a14SPatrick Farrell }
2689a9603a14SPatrick Farrell 
2690e882e171SHong Zhang /*@C
269147450a7bSBarry Smith   TaoMonitorDrawCtxCreate - Creates the monitor context for `TaoMonitorDrawSolution()`
2692a9603a14SPatrick Farrell 
2693c3339decSBarry Smith   Collective
2694e882e171SHong Zhang 
26952fe279fdSBarry Smith   Input Parameters:
26962fe279fdSBarry Smith + comm     - the communicator to share the context
26972fe279fdSBarry Smith . host     - the name of the X Windows host that will display the monitor
26982fe279fdSBarry Smith . label    - the label to put at the top of the display window
26992fe279fdSBarry Smith . x        - the horizontal coordinate of the lower left corner of the window to open
27002fe279fdSBarry Smith . y        - the vertical coordinate of the lower left corner of the window to open
27012fe279fdSBarry Smith . m        - the width of the window
27022fe279fdSBarry Smith . n        - the height of the window
27032fe279fdSBarry Smith - howoften - how many `Tao` iterations between displaying the monitor information
27042fe279fdSBarry Smith 
2705d5b43468SJose E. Roman   Output Parameter:
2706e882e171SHong Zhang . ctx - the monitor context
2707e882e171SHong Zhang 
27083c7db156SBarry Smith   Options Database Key:
2709e882e171SHong Zhang . -tao_draw_solution_initial - show initial guess as well as current solution
2710e882e171SHong Zhang 
2711e882e171SHong Zhang   Level: intermediate
2712e882e171SHong Zhang 
27131cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoMonitorSet()`, `TaoMonitorDefault()`, `VecView()`, `TaoMonitorDrawCtx()`
2714e882e171SHong Zhang @*/
2715d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoMonitorDrawCtxCreate(MPI_Comm comm, const char host[], const char label[], int x, int y, int m, int n, PetscInt howoften, TaoMonitorDrawCtx *ctx)
2716d71ae5a4SJacob Faibussowitsch {
2717e882e171SHong Zhang   PetscFunctionBegin;
27189566063dSJacob Faibussowitsch   PetscCall(PetscNew(ctx));
27199566063dSJacob Faibussowitsch   PetscCall(PetscViewerDrawOpen(comm, host, label, x, y, m, n, &(*ctx)->viewer));
27209566063dSJacob Faibussowitsch   PetscCall(PetscViewerSetFromOptions((*ctx)->viewer));
2721e882e171SHong Zhang   (*ctx)->howoften = howoften;
27223ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
2723e882e171SHong Zhang }
2724e882e171SHong Zhang 
2725e882e171SHong Zhang /*@C
272665ba42b6SBarry Smith   TaoMonitorDrawCtxDestroy - Destroys the monitor context for `TaoMonitorDrawSolution()`
2727e882e171SHong Zhang 
2728c3339decSBarry Smith   Collective
2729e882e171SHong Zhang 
273047450a7bSBarry Smith   Input Parameter:
2731e056e8ceSJacob Faibussowitsch . ictx - the monitor context
2732e882e171SHong Zhang 
2733e882e171SHong Zhang   Level: intermediate
2734e882e171SHong Zhang 
27351cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoMonitorSet()`, `TaoMonitorDefault()`, `VecView()`, `TaoMonitorDrawSolution()`
2736e882e171SHong Zhang @*/
2737d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoMonitorDrawCtxDestroy(TaoMonitorDrawCtx *ictx)
2738d71ae5a4SJacob Faibussowitsch {
2739e882e171SHong Zhang   PetscFunctionBegin;
27409566063dSJacob Faibussowitsch   PetscCall(PetscViewerDestroy(&(*ictx)->viewer));
27419566063dSJacob Faibussowitsch   PetscCall(PetscFree(*ictx));
27423ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
2743e882e171SHong Zhang }
2744