xref: /petsc/src/tao/interface/taosolver.c (revision 1d27aa22b2f6148b2c4e3f06a75e0638d6493e09)
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 /*@
295*1d27aa22SBarry 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;
367a7e14dcfSSatish Balay   PetscBool   flg;
368a7e14dcfSSatish Balay   MPI_Comm    comm;
369a7e14dcfSSatish Balay 
370a7e14dcfSSatish Balay   PetscFunctionBegin;
371441846f8SBarry Smith   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
3729566063dSJacob Faibussowitsch   PetscCall(PetscObjectGetComm((PetscObject)tao, &comm));
373125ddc8aSJason Sarich 
37460b70c5cSStefano Zampini   if (((PetscObject)tao)->type_name) default_type = ((PetscObject)tao)->type_name;
37560b70c5cSStefano Zampini 
376d0609cedSBarry Smith   PetscObjectOptionsBegin((PetscObject)tao);
377a7e14dcfSSatish Balay   /* Check for type from options */
3789566063dSJacob Faibussowitsch   PetscCall(PetscOptionsFList("-tao_type", "Tao Solver type", "TaoSetType", TaoList, default_type, type, 256, &flg));
379a7e14dcfSSatish Balay   if (flg) {
3809566063dSJacob Faibussowitsch     PetscCall(TaoSetType(tao, type));
381a7e14dcfSSatish Balay   } else if (!((PetscObject)tao)->type_name) {
3829566063dSJacob Faibussowitsch     PetscCall(TaoSetType(tao, default_type));
383a7e14dcfSSatish Balay   }
384a7e14dcfSSatish Balay 
38560b70c5cSStefano Zampini   /* Tao solvers do not set the prefix, set it here if not yet done
38660b70c5cSStefano Zampini      We do it after SetType since solver may have been changed */
38760b70c5cSStefano Zampini   if (tao->linesearch) {
38860b70c5cSStefano Zampini     const char *prefix;
38960b70c5cSStefano Zampini     PetscCall(TaoLineSearchGetOptionsPrefix(tao->linesearch, &prefix));
39060b70c5cSStefano Zampini     if (!prefix) PetscCall(TaoLineSearchSetOptionsPrefix(tao->linesearch, ((PetscObject)(tao))->prefix));
39160b70c5cSStefano Zampini   }
39260b70c5cSStefano Zampini 
3939566063dSJacob Faibussowitsch   PetscCall(PetscOptionsReal("-tao_catol", "Stop if constraints violations within", "TaoSetConstraintTolerances", tao->catol, &tao->catol, &flg));
3946552cf8aSJason Sarich   if (flg) tao->catol_changed = PETSC_TRUE;
3956aad120cSJose E. Roman   PetscCall(PetscOptionsReal("-tao_crtol", "Stop if relative constraint violations within", "TaoSetConstraintTolerances", tao->crtol, &tao->crtol, &flg));
3966552cf8aSJason Sarich   if (flg) tao->crtol_changed = PETSC_TRUE;
3979566063dSJacob Faibussowitsch   PetscCall(PetscOptionsReal("-tao_gatol", "Stop if norm of gradient less than", "TaoSetTolerances", tao->gatol, &tao->gatol, &flg));
3986552cf8aSJason Sarich   if (flg) tao->gatol_changed = PETSC_TRUE;
3999566063dSJacob Faibussowitsch   PetscCall(PetscOptionsReal("-tao_grtol", "Stop if norm of gradient divided by the function value is less than", "TaoSetTolerances", tao->grtol, &tao->grtol, &flg));
4006552cf8aSJason Sarich   if (flg) tao->grtol_changed = PETSC_TRUE;
4019566063dSJacob Faibussowitsch   PetscCall(PetscOptionsReal("-tao_gttol", "Stop if the norm of the gradient is less than the norm of the initial gradient times tol", "TaoSetTolerances", tao->gttol, &tao->gttol, &flg));
4026552cf8aSJason Sarich   if (flg) tao->gttol_changed = PETSC_TRUE;
4039566063dSJacob Faibussowitsch   PetscCall(PetscOptionsInt("-tao_max_it", "Stop if iteration number exceeds", "TaoSetMaximumIterations", tao->max_it, &tao->max_it, &flg));
4046552cf8aSJason Sarich   if (flg) tao->max_it_changed = PETSC_TRUE;
4059566063dSJacob Faibussowitsch   PetscCall(PetscOptionsInt("-tao_max_funcs", "Stop if number of function evaluations exceeds", "TaoSetMaximumFunctionEvaluations", tao->max_funcs, &tao->max_funcs, &flg));
4066552cf8aSJason Sarich   if (flg) tao->max_funcs_changed = PETSC_TRUE;
4079566063dSJacob Faibussowitsch   PetscCall(PetscOptionsReal("-tao_fmin", "Stop if function less than", "TaoSetFunctionLowerBound", tao->fmin, &tao->fmin, &flg));
4086552cf8aSJason Sarich   if (flg) tao->fmin_changed = PETSC_TRUE;
4099566063dSJacob Faibussowitsch   PetscCall(PetscOptionsReal("-tao_steptol", "Stop if step size or trust region radius less than", "", tao->steptol, &tao->steptol, &flg));
4106552cf8aSJason Sarich   if (flg) tao->steptol_changed = PETSC_TRUE;
4119566063dSJacob Faibussowitsch   PetscCall(PetscOptionsReal("-tao_trust0", "Initial trust region radius", "TaoSetTrustRegionRadius", tao->trust0, &tao->trust0, &flg));
4126552cf8aSJason Sarich   if (flg) tao->trust0_changed = PETSC_TRUE;
4139566063dSJacob Faibussowitsch   PetscCall(PetscOptionsString("-tao_view_solution", "view solution vector after each evaluation", "TaoSetMonitor", "stdout", monfilename, sizeof(monfilename), &flg));
414a7e14dcfSSatish Balay   if (flg) {
4159566063dSJacob Faibussowitsch     PetscCall(PetscViewerASCIIOpen(comm, monfilename, &monviewer));
4169566063dSJacob Faibussowitsch     PetscCall(TaoSetMonitor(tao, TaoSolutionMonitor, monviewer, (PetscErrorCode(*)(void **))PetscViewerDestroy));
417a7e14dcfSSatish Balay   }
418a7e14dcfSSatish Balay 
41965ba42b6SBarry Smith   PetscCall(PetscOptionsBool("-tao_converged_reason", "Print reason for Tao converged", "TaoSolve", tao->printreason, &tao->printreason, NULL));
4209566063dSJacob Faibussowitsch   PetscCall(PetscOptionsString("-tao_view_gradient", "view gradient vector after each evaluation", "TaoSetMonitor", "stdout", monfilename, sizeof(monfilename), &flg));
421a7e14dcfSSatish Balay   if (flg) {
4229566063dSJacob Faibussowitsch     PetscCall(PetscViewerASCIIOpen(comm, monfilename, &monviewer));
4239566063dSJacob Faibussowitsch     PetscCall(TaoSetMonitor(tao, TaoGradientMonitor, monviewer, (PetscErrorCode(*)(void **))PetscViewerDestroy));
424a7e14dcfSSatish Balay   }
425a7e14dcfSSatish Balay 
4269566063dSJacob Faibussowitsch   PetscCall(PetscOptionsString("-tao_view_stepdirection", "view step direction vector after each iteration", "TaoSetMonitor", "stdout", monfilename, sizeof(monfilename), &flg));
427a7e14dcfSSatish Balay   if (flg) {
4289566063dSJacob Faibussowitsch     PetscCall(PetscViewerASCIIOpen(comm, monfilename, &monviewer));
4299566063dSJacob Faibussowitsch     PetscCall(TaoSetMonitor(tao, TaoStepDirectionMonitor, monviewer, (PetscErrorCode(*)(void **))PetscViewerDestroy));
430a7e14dcfSSatish Balay   }
431a7e14dcfSSatish Balay 
4329566063dSJacob Faibussowitsch   PetscCall(PetscOptionsString("-tao_view_residual", "view least-squares residual vector after each evaluation", "TaoSetMonitor", "stdout", monfilename, sizeof(monfilename), &flg));
433a7e14dcfSSatish Balay   if (flg) {
4349566063dSJacob Faibussowitsch     PetscCall(PetscViewerASCIIOpen(comm, monfilename, &monviewer));
4359566063dSJacob Faibussowitsch     PetscCall(TaoSetMonitor(tao, TaoResidualMonitor, monviewer, (PetscErrorCode(*)(void **))PetscViewerDestroy));
436a7e14dcfSSatish Balay   }
437a7e14dcfSSatish Balay 
4389566063dSJacob Faibussowitsch   PetscCall(PetscOptionsString("-tao_monitor", "Use the default convergence monitor", "TaoSetMonitor", "stdout", monfilename, sizeof(monfilename), &flg));
439a7e14dcfSSatish Balay   if (flg) {
4409566063dSJacob Faibussowitsch     PetscCall(PetscViewerASCIIOpen(comm, monfilename, &monviewer));
4419566063dSJacob Faibussowitsch     PetscCall(TaoSetMonitor(tao, TaoMonitorDefault, monviewer, (PetscErrorCode(*)(void **))PetscViewerDestroy));
442a7e14dcfSSatish Balay   }
443a7e14dcfSSatish Balay 
4449566063dSJacob Faibussowitsch   PetscCall(PetscOptionsString("-tao_gmonitor", "Use the convergence monitor with extra globalization info", "TaoSetMonitor", "stdout", monfilename, sizeof(monfilename), &flg));
4458d5ead36SAlp Dener   if (flg) {
4469566063dSJacob Faibussowitsch     PetscCall(PetscViewerASCIIOpen(comm, monfilename, &monviewer));
4479566063dSJacob Faibussowitsch     PetscCall(TaoSetMonitor(tao, TaoDefaultGMonitor, monviewer, (PetscErrorCode(*)(void **))PetscViewerDestroy));
4488d5ead36SAlp Dener   }
4498d5ead36SAlp Dener 
4509566063dSJacob Faibussowitsch   PetscCall(PetscOptionsString("-tao_smonitor", "Use the short convergence monitor", "TaoSetMonitor", "stdout", monfilename, sizeof(monfilename), &flg));
451a7e14dcfSSatish Balay   if (flg) {
4529566063dSJacob Faibussowitsch     PetscCall(PetscViewerASCIIOpen(comm, monfilename, &monviewer));
4539566063dSJacob Faibussowitsch     PetscCall(TaoSetMonitor(tao, TaoDefaultSMonitor, monviewer, (PetscErrorCode(*)(void **))PetscViewerDestroy));
454a7e14dcfSSatish Balay   }
455a7e14dcfSSatish Balay 
4569566063dSJacob Faibussowitsch   PetscCall(PetscOptionsString("-tao_cmonitor", "Use the default convergence monitor with constraint norm", "TaoSetMonitor", "stdout", monfilename, sizeof(monfilename), &flg));
457a7e14dcfSSatish Balay   if (flg) {
4589566063dSJacob Faibussowitsch     PetscCall(PetscViewerASCIIOpen(comm, monfilename, &monviewer));
4599566063dSJacob Faibussowitsch     PetscCall(TaoSetMonitor(tao, TaoDefaultCMonitor, monviewer, (PetscErrorCode(*)(void **))PetscViewerDestroy));
460a7e14dcfSSatish Balay   }
461a7e14dcfSSatish Balay 
4628afaa268SBarry Smith   flg = PETSC_FALSE;
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   }
509414d97d3SAlp Dener   flg = PETSC_FALSE;
5109566063dSJacob Faibussowitsch   PetscCall(PetscOptionsBool("-tao_recycle_history", "enable recycling/re-using information from the previous TaoSolve() call for some algorithms", "TaoSetRecycleHistory", flg, &flg, NULL));
5111baa6e33SBarry Smith   if (flg) PetscCall(TaoSetRecycleHistory(tao, PETSC_TRUE));
5129566063dSJacob Faibussowitsch   PetscCall(PetscOptionsEnum("-tao_subset_type", "subset type", "", TaoSubSetTypes, (PetscEnum)tao->subset_type, (PetscEnum *)&tao->subset_type, NULL));
513a7e14dcfSSatish Balay 
5140f0abf79SStefano Zampini   if (tao->ksp) {
5150f0abf79SStefano Zampini     PetscCall(PetscOptionsBool("-tao_ksp_ew", "Use Eisentat-Walker linear system convergence test", "TaoKSPSetUseEW", tao->ksp_ewconv, &tao->ksp_ewconv, NULL));
5160f0abf79SStefano Zampini     PetscCall(TaoKSPSetUseEW(tao, tao->ksp_ewconv));
5170f0abf79SStefano Zampini   }
5180f0abf79SStefano Zampini 
519dbbe0bcdSBarry Smith   PetscTryTypeMethod(tao, setfromoptions, PetscOptionsObject);
52060b70c5cSStefano Zampini 
52160b70c5cSStefano Zampini   /* process any options handlers added with PetscObjectAddOptionsHandler() */
52260b70c5cSStefano Zampini   PetscCall(PetscObjectProcessOptionsHandlers((PetscObject)tao, PetscOptionsObject));
523d0609cedSBarry Smith   PetscOptionsEnd();
52460b70c5cSStefano Zampini 
52560b70c5cSStefano Zampini   if (tao->linesearch) PetscCall(TaoLineSearchSetFromOptions(tao->linesearch));
5263ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
527a7e14dcfSSatish Balay }
528a7e14dcfSSatish Balay 
529a7e14dcfSSatish Balay /*@C
53047450a7bSBarry Smith   TaoViewFromOptions - View a `Tao` object based on values in the options database
531fe2efc57SMark 
532c3339decSBarry Smith   Collective
533fe2efc57SMark 
534fe2efc57SMark   Input Parameters:
53547450a7bSBarry Smith + A    - the  `Tao` context
53647450a7bSBarry Smith . obj  - Optional object that provides the prefix for the options database
537736c3998SJose E. Roman - name - command line option
538fe2efc57SMark 
539fe2efc57SMark   Level: intermediate
54067be906fSBarry Smith 
5411cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoView`, `PetscObjectViewFromOptions()`, `TaoCreate()`
542fe2efc57SMark @*/
543d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoViewFromOptions(Tao A, PetscObject obj, const char name[])
544d71ae5a4SJacob Faibussowitsch {
545fe2efc57SMark   PetscFunctionBegin;
546fe2efc57SMark   PetscValidHeaderSpecific(A, TAO_CLASSID, 1);
5479566063dSJacob Faibussowitsch   PetscCall(PetscObjectViewFromOptions((PetscObject)A, obj, name));
5483ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
549fe2efc57SMark }
550fe2efc57SMark 
551fe2efc57SMark /*@C
55247450a7bSBarry Smith   TaoView - Prints information about the `Tao` object
553a7e14dcfSSatish Balay 
554c3339decSBarry Smith   Collective
555a7e14dcfSSatish Balay 
556a7e14dcfSSatish Balay   Input Parameters:
55747450a7bSBarry Smith + tao    - the `Tao` context
558a7e14dcfSSatish Balay - viewer - visualization context
559a7e14dcfSSatish Balay 
560a7e14dcfSSatish Balay   Options Database Key:
56165ba42b6SBarry Smith . -tao_view - Calls `TaoView()` at the end of `TaoSolve()`
562a7e14dcfSSatish Balay 
56367be906fSBarry Smith   Level: beginner
56467be906fSBarry Smith 
565a7e14dcfSSatish Balay   Notes:
566a7e14dcfSSatish Balay   The available visualization contexts include
56765ba42b6SBarry Smith +     `PETSC_VIEWER_STDOUT_SELF` - standard output (default)
56865ba42b6SBarry Smith -     `PETSC_VIEWER_STDOUT_WORLD` - synchronized standard
569a7e14dcfSSatish Balay   output where only the first processor opens
570a7e14dcfSSatish Balay   the file.  All other processors send their
571a7e14dcfSSatish Balay   data to the first processor to print.
572a7e14dcfSSatish Balay 
5731cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `PetscViewerASCIIOpen()`
574a7e14dcfSSatish Balay @*/
575d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoView(Tao tao, PetscViewer viewer)
576d71ae5a4SJacob Faibussowitsch {
577a7e14dcfSSatish Balay   PetscBool isascii, isstring;
578b625d6c7SJed Brown   TaoType   type;
57947a47007SBarry Smith 
580a7e14dcfSSatish Balay   PetscFunctionBegin;
581441846f8SBarry Smith   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
58248a46eb9SPierre Jolivet   if (!viewer) PetscCall(PetscViewerASCIIGetStdout(((PetscObject)tao)->comm, &viewer));
583a7e14dcfSSatish Balay   PetscValidHeaderSpecific(viewer, PETSC_VIEWER_CLASSID, 2);
584a7e14dcfSSatish Balay   PetscCheckSameComm(tao, 1, viewer, 2);
585a7e14dcfSSatish Balay 
5869566063dSJacob Faibussowitsch   PetscCall(PetscObjectTypeCompare((PetscObject)viewer, PETSCVIEWERASCII, &isascii));
5879566063dSJacob Faibussowitsch   PetscCall(PetscObjectTypeCompare((PetscObject)viewer, PETSCVIEWERSTRING, &isstring));
588a7e14dcfSSatish Balay   if (isascii) {
5899566063dSJacob Faibussowitsch     PetscCall(PetscObjectPrintClassNamePrefixType((PetscObject)tao, viewer));
590a7e14dcfSSatish Balay 
591a7e14dcfSSatish Balay     if (tao->ops->view) {
5929566063dSJacob Faibussowitsch       PetscCall(PetscViewerASCIIPushTab(viewer));
593dbbe0bcdSBarry Smith       PetscUseTypeMethod(tao, view, viewer);
5949566063dSJacob Faibussowitsch       PetscCall(PetscViewerASCIIPopTab(viewer));
595a7e14dcfSSatish Balay     }
5968d3f3ddaSAlp Dener     if (tao->linesearch) {
5979566063dSJacob Faibussowitsch       PetscCall(PetscViewerASCIIPushTab(viewer));
5989566063dSJacob Faibussowitsch       PetscCall(TaoLineSearchView(tao->linesearch, viewer));
5999566063dSJacob Faibussowitsch       PetscCall(PetscViewerASCIIPopTab(viewer));
6008d3f3ddaSAlp Dener     }
601a7e14dcfSSatish Balay     if (tao->ksp) {
6029566063dSJacob Faibussowitsch       PetscCall(PetscViewerASCIIPushTab(viewer));
6039566063dSJacob Faibussowitsch       PetscCall(KSPView(tao->ksp, viewer));
60463a3b9bcSJacob Faibussowitsch       PetscCall(PetscViewerASCIIPrintf(viewer, "total KSP iterations: %" PetscInt_FMT "\n", tao->ksp_tot_its));
6059566063dSJacob Faibussowitsch       PetscCall(PetscViewerASCIIPopTab(viewer));
606a7e14dcfSSatish Balay     }
60719b5c101SAlp Dener 
6089566063dSJacob Faibussowitsch     PetscCall(PetscViewerASCIIPushTab(viewer));
60919b5c101SAlp Dener 
61048a46eb9SPierre Jolivet     if (tao->XL || tao->XU) PetscCall(PetscViewerASCIIPrintf(viewer, "Active Set subset type: %s\n", TaoSubSetTypes[tao->subset_type]));
611a7e14dcfSSatish Balay 
6129566063dSJacob Faibussowitsch     PetscCall(PetscViewerASCIIPrintf(viewer, "convergence tolerances: gatol=%g,", (double)tao->gatol));
6139566063dSJacob Faibussowitsch     PetscCall(PetscViewerASCIIPrintf(viewer, " steptol=%g,", (double)tao->steptol));
6149566063dSJacob Faibussowitsch     PetscCall(PetscViewerASCIIPrintf(viewer, " gttol=%g\n", (double)tao->gttol));
6159566063dSJacob Faibussowitsch     PetscCall(PetscViewerASCIIPrintf(viewer, "Residual in Function/Gradient:=%g\n", (double)tao->residual));
616a7e14dcfSSatish Balay 
6176246e8cdSAlp Dener     if (tao->constrained) {
6189566063dSJacob Faibussowitsch       PetscCall(PetscViewerASCIIPrintf(viewer, "convergence tolerances:"));
6199566063dSJacob Faibussowitsch       PetscCall(PetscViewerASCIIPrintf(viewer, " catol=%g,", (double)tao->catol));
6209566063dSJacob Faibussowitsch       PetscCall(PetscViewerASCIIPrintf(viewer, " crtol=%g\n", (double)tao->crtol));
6219566063dSJacob Faibussowitsch       PetscCall(PetscViewerASCIIPrintf(viewer, "Residual in Constraints:=%g\n", (double)tao->cnorm));
622a7e14dcfSSatish Balay     }
623a7e14dcfSSatish Balay 
624a7e14dcfSSatish Balay     if (tao->trust < tao->steptol) {
6259566063dSJacob Faibussowitsch       PetscCall(PetscViewerASCIIPrintf(viewer, "convergence tolerances: steptol=%g\n", (double)tao->steptol));
6269566063dSJacob Faibussowitsch       PetscCall(PetscViewerASCIIPrintf(viewer, "Final trust region radius:=%g\n", (double)tao->trust));
627a7e14dcfSSatish Balay     }
628a7e14dcfSSatish Balay 
62948a46eb9SPierre Jolivet     if (tao->fmin > -1.e25) PetscCall(PetscViewerASCIIPrintf(viewer, "convergence tolerances: function minimum=%g\n", (double)tao->fmin));
6309566063dSJacob Faibussowitsch     PetscCall(PetscViewerASCIIPrintf(viewer, "Objective value=%g\n", (double)tao->fc));
631a7e14dcfSSatish Balay 
63263a3b9bcSJacob Faibussowitsch     PetscCall(PetscViewerASCIIPrintf(viewer, "total number of iterations=%" PetscInt_FMT ",          ", tao->niter));
63363a3b9bcSJacob Faibussowitsch     PetscCall(PetscViewerASCIIPrintf(viewer, "              (max: %" PetscInt_FMT ")\n", tao->max_it));
634a7e14dcfSSatish Balay 
635a7e14dcfSSatish Balay     if (tao->nfuncs > 0) {
63663a3b9bcSJacob Faibussowitsch       PetscCall(PetscViewerASCIIPrintf(viewer, "total number of function evaluations=%" PetscInt_FMT ",", tao->nfuncs));
63763a3b9bcSJacob Faibussowitsch       PetscCall(PetscViewerASCIIPrintf(viewer, "                max: %" PetscInt_FMT "\n", tao->max_funcs));
638a7e14dcfSSatish Balay     }
639a7e14dcfSSatish Balay     if (tao->ngrads > 0) {
64063a3b9bcSJacob Faibussowitsch       PetscCall(PetscViewerASCIIPrintf(viewer, "total number of gradient evaluations=%" PetscInt_FMT ",", tao->ngrads));
64163a3b9bcSJacob Faibussowitsch       PetscCall(PetscViewerASCIIPrintf(viewer, "                max: %" PetscInt_FMT "\n", tao->max_funcs));
642a7e14dcfSSatish Balay     }
643a7e14dcfSSatish Balay     if (tao->nfuncgrads > 0) {
64463a3b9bcSJacob Faibussowitsch       PetscCall(PetscViewerASCIIPrintf(viewer, "total number of function/gradient evaluations=%" PetscInt_FMT ",", tao->nfuncgrads));
64563a3b9bcSJacob Faibussowitsch       PetscCall(PetscViewerASCIIPrintf(viewer, "    (max: %" PetscInt_FMT ")\n", tao->max_funcs));
646a7e14dcfSSatish Balay     }
64748a46eb9SPierre Jolivet     if (tao->nhess > 0) PetscCall(PetscViewerASCIIPrintf(viewer, "total number of Hessian evaluations=%" PetscInt_FMT "\n", tao->nhess));
64848a46eb9SPierre Jolivet     if (tao->nconstraints > 0) PetscCall(PetscViewerASCIIPrintf(viewer, "total number of constraint function evaluations=%" PetscInt_FMT "\n", tao->nconstraints));
64948a46eb9SPierre Jolivet     if (tao->njac > 0) PetscCall(PetscViewerASCIIPrintf(viewer, "total number of Jacobian evaluations=%" PetscInt_FMT "\n", tao->njac));
650a7e14dcfSSatish Balay 
651a7e14dcfSSatish Balay     if (tao->reason > 0) {
6529566063dSJacob Faibussowitsch       PetscCall(PetscViewerASCIIPrintf(viewer, "Solution converged: "));
653a7e14dcfSSatish Balay       switch (tao->reason) {
654d71ae5a4SJacob Faibussowitsch       case TAO_CONVERGED_GATOL:
655d71ae5a4SJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, " ||g(X)|| <= gatol\n"));
656d71ae5a4SJacob Faibussowitsch         break;
657d71ae5a4SJacob Faibussowitsch       case TAO_CONVERGED_GRTOL:
658d71ae5a4SJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, " ||g(X)||/|f(X)| <= grtol\n"));
659d71ae5a4SJacob Faibussowitsch         break;
660d71ae5a4SJacob Faibussowitsch       case TAO_CONVERGED_GTTOL:
661d71ae5a4SJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, " ||g(X)||/||g(X0)|| <= gttol\n"));
662d71ae5a4SJacob Faibussowitsch         break;
663d71ae5a4SJacob Faibussowitsch       case TAO_CONVERGED_STEPTOL:
664d71ae5a4SJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, " Steptol -- step size small\n"));
665d71ae5a4SJacob Faibussowitsch         break;
666d71ae5a4SJacob Faibussowitsch       case TAO_CONVERGED_MINF:
667d71ae5a4SJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, " Minf --  f < fmin\n"));
668d71ae5a4SJacob Faibussowitsch         break;
669d71ae5a4SJacob Faibussowitsch       case TAO_CONVERGED_USER:
670d71ae5a4SJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, " User Terminated\n"));
671d71ae5a4SJacob Faibussowitsch         break;
672d71ae5a4SJacob Faibussowitsch       default:
673d71ae5a4SJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, "\n"));
674d71ae5a4SJacob Faibussowitsch         break;
675a7e14dcfSSatish Balay       }
676a7e14dcfSSatish Balay     } else {
6779566063dSJacob Faibussowitsch       PetscCall(PetscViewerASCIIPrintf(viewer, "Solver terminated: %d", tao->reason));
678a7e14dcfSSatish Balay       switch (tao->reason) {
679d71ae5a4SJacob Faibussowitsch       case TAO_DIVERGED_MAXITS:
680d71ae5a4SJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, " Maximum Iterations\n"));
681d71ae5a4SJacob Faibussowitsch         break;
682d71ae5a4SJacob Faibussowitsch       case TAO_DIVERGED_NAN:
683d71ae5a4SJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, " NAN or Inf encountered\n"));
684d71ae5a4SJacob Faibussowitsch         break;
685d71ae5a4SJacob Faibussowitsch       case TAO_DIVERGED_MAXFCN:
686d71ae5a4SJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, " Maximum Function Evaluations\n"));
687d71ae5a4SJacob Faibussowitsch         break;
688d71ae5a4SJacob Faibussowitsch       case TAO_DIVERGED_LS_FAILURE:
689d71ae5a4SJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, " Line Search Failure\n"));
690d71ae5a4SJacob Faibussowitsch         break;
691d71ae5a4SJacob Faibussowitsch       case TAO_DIVERGED_TR_REDUCTION:
692d71ae5a4SJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, " Trust Region too small\n"));
693d71ae5a4SJacob Faibussowitsch         break;
694d71ae5a4SJacob Faibussowitsch       case TAO_DIVERGED_USER:
695d71ae5a4SJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, " User Terminated\n"));
696d71ae5a4SJacob Faibussowitsch         break;
697d71ae5a4SJacob Faibussowitsch       default:
698d71ae5a4SJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, "\n"));
699d71ae5a4SJacob Faibussowitsch         break;
700a7e14dcfSSatish Balay       }
701a7e14dcfSSatish Balay     }
7029566063dSJacob Faibussowitsch     PetscCall(PetscViewerASCIIPopTab(viewer));
703a7e14dcfSSatish Balay   } else if (isstring) {
7049566063dSJacob Faibussowitsch     PetscCall(TaoGetType(tao, &type));
7059566063dSJacob Faibussowitsch     PetscCall(PetscViewerStringSPrintf(viewer, " %-3.3s", type));
706a7e14dcfSSatish Balay   }
7073ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
708a7e14dcfSSatish Balay }
709a7e14dcfSSatish Balay 
710a7e14dcfSSatish Balay /*@
711414d97d3SAlp Dener   TaoSetRecycleHistory - Sets the boolean flag to enable/disable re-using
71265ba42b6SBarry Smith   iterate information from the previous `TaoSolve()`. This feature is disabled by
713414d97d3SAlp Dener   default.
714414d97d3SAlp Dener 
71567be906fSBarry Smith   Logically Collective
71667be906fSBarry Smith 
71767be906fSBarry Smith   Input Parameters:
71847450a7bSBarry Smith + tao     - the `Tao` context
71967be906fSBarry Smith - recycle - boolean flag
72067be906fSBarry Smith 
72147450a7bSBarry Smith   Options Database Key:
72267be906fSBarry Smith . -tao_recycle_history <true,false> - reuse the history
72367be906fSBarry Smith 
72467be906fSBarry Smith   Level: intermediate
72567be906fSBarry Smith 
72667be906fSBarry Smith   Notes:
72765ba42b6SBarry Smith   For conjugate gradient methods (`TAOBNCG`), this re-uses the latest search direction
72865ba42b6SBarry Smith   from the previous `TaoSolve()` call when computing the first search direction in a
729414d97d3SAlp Dener   new solution. By default, CG methods set the first search direction to the
730414d97d3SAlp Dener   negative gradient.
731414d97d3SAlp Dener 
73265ba42b6SBarry Smith   For quasi-Newton family of methods (`TAOBQNLS`, `TAOBQNKLS`, `TAOBQNKTR`, `TAOBQNKTL`), this re-uses
73365ba42b6SBarry Smith   the accumulated quasi-Newton Hessian approximation from the previous `TaoSolve()`
734414d97d3SAlp Dener   call. By default, QN family of methods reset the initial Hessian approximation to
735414d97d3SAlp Dener   the identity matrix.
736414d97d3SAlp Dener 
737414d97d3SAlp Dener   For any other algorithm, this setting has no effect.
738414d97d3SAlp Dener 
7391cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoGetRecycleHistory()`, `TAOBNCG`, `TAOBQNLS`, `TAOBQNKLS`, `TAOBQNKTR`, `TAOBQNKTL`
740414d97d3SAlp Dener @*/
741d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoSetRecycleHistory(Tao tao, PetscBool recycle)
742d71ae5a4SJacob Faibussowitsch {
743414d97d3SAlp Dener   PetscFunctionBegin;
744414d97d3SAlp Dener   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
74594511df7SStefano Zampini   PetscValidLogicalCollectiveBool(tao, recycle, 2);
746414d97d3SAlp Dener   tao->recycle = recycle;
7473ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
748414d97d3SAlp Dener }
749414d97d3SAlp Dener 
750414d97d3SAlp Dener /*@
751414d97d3SAlp Dener   TaoGetRecycleHistory - Retrieve the boolean flag for re-using iterate information
75265ba42b6SBarry Smith   from the previous `TaoSolve()`. This feature is disabled by default.
753414d97d3SAlp Dener 
75467be906fSBarry Smith   Logically Collective
755414d97d3SAlp Dener 
75647450a7bSBarry Smith   Input Parameter:
75747450a7bSBarry Smith . tao - the `Tao` context
758414d97d3SAlp Dener 
75947450a7bSBarry Smith   Output Parameter:
760147403d9SBarry Smith . recycle - boolean flag
761414d97d3SAlp Dener 
762414d97d3SAlp Dener   Level: intermediate
763414d97d3SAlp Dener 
7641cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoSetRecycleHistory()`, `TAOBNCG`, `TAOBQNLS`, `TAOBQNKLS`, `TAOBQNKTR`, `TAOBQNKTL`
765414d97d3SAlp Dener @*/
766d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoGetRecycleHistory(Tao tao, PetscBool *recycle)
767d71ae5a4SJacob Faibussowitsch {
768414d97d3SAlp Dener   PetscFunctionBegin;
769414d97d3SAlp Dener   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
7704f572ea9SToby Isaac   PetscAssertPointer(recycle, 2);
771414d97d3SAlp Dener   *recycle = tao->recycle;
7723ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
773414d97d3SAlp Dener }
774414d97d3SAlp Dener 
775414d97d3SAlp Dener /*@
77647450a7bSBarry Smith   TaoSetTolerances - Sets parameters used in `TaoSolve()` convergence tests
777a7e14dcfSSatish Balay 
77867be906fSBarry Smith   Logically Collective
779a7e14dcfSSatish Balay 
780a7e14dcfSSatish Balay   Input Parameters:
78147450a7bSBarry Smith + tao   - the `Tao` context
782a7e14dcfSSatish Balay . gatol - stop if norm of gradient is less than this
783a7e14dcfSSatish Balay . grtol - stop if relative norm of gradient is less than this
784a7e14dcfSSatish Balay - gttol - stop if norm of gradient is reduced by this factor
785a7e14dcfSSatish Balay 
786a7e14dcfSSatish Balay   Options Database Keys:
787e52336cbSBarry Smith + -tao_gatol <gatol> - Sets gatol
788a7e14dcfSSatish Balay . -tao_grtol <grtol> - Sets grtol
789a7e14dcfSSatish Balay - -tao_gttol <gttol> - Sets gttol
790a7e14dcfSSatish Balay 
7913b242c63SJacob Faibussowitsch   Stopping Criteria\:
79267be906fSBarry Smith .vb
79367be906fSBarry Smith   ||g(X)||                            <= gatol
79467be906fSBarry Smith   ||g(X)|| / |f(X)|                   <= grtol
79567be906fSBarry Smith   ||g(X)|| / ||g(X0)||                <= gttol
79667be906fSBarry Smith .ve
797a7e14dcfSSatish Balay 
798a7e14dcfSSatish Balay   Level: beginner
799a7e14dcfSSatish Balay 
80067be906fSBarry Smith   Note:
80167be906fSBarry Smith   Use `PETSC_DEFAULT` to leave one or more tolerances unchanged.
802a7e14dcfSSatish Balay 
8031cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoConvergedReason`, `TaoGetTolerances()`
804a7e14dcfSSatish Balay @*/
805d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoSetTolerances(Tao tao, PetscReal gatol, PetscReal grtol, PetscReal gttol)
806d71ae5a4SJacob Faibussowitsch {
807a7e14dcfSSatish Balay   PetscFunctionBegin;
808441846f8SBarry Smith   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
80994511df7SStefano Zampini   PetscValidLogicalCollectiveReal(tao, gatol, 2);
81094511df7SStefano Zampini   PetscValidLogicalCollectiveReal(tao, grtol, 3);
81194511df7SStefano Zampini   PetscValidLogicalCollectiveReal(tao, gttol, 4);
812a7e14dcfSSatish Balay 
81313bcc0bdSJacob Faibussowitsch   if (gatol != (PetscReal)PETSC_DEFAULT) {
814a7e14dcfSSatish Balay     if (gatol < 0) {
8159566063dSJacob Faibussowitsch       PetscCall(PetscInfo(tao, "Tried to set negative gatol -- ignored.\n"));
816a7e14dcfSSatish Balay     } else {
817a7e14dcfSSatish Balay       tao->gatol         = PetscMax(0, gatol);
8186552cf8aSJason Sarich       tao->gatol_changed = PETSC_TRUE;
819a7e14dcfSSatish Balay     }
820a7e14dcfSSatish Balay   }
821a7e14dcfSSatish Balay 
82213bcc0bdSJacob Faibussowitsch   if (grtol != (PetscReal)PETSC_DEFAULT) {
823a7e14dcfSSatish Balay     if (grtol < 0) {
8249566063dSJacob Faibussowitsch       PetscCall(PetscInfo(tao, "Tried to set negative grtol -- ignored.\n"));
825a7e14dcfSSatish Balay     } else {
826a7e14dcfSSatish Balay       tao->grtol         = PetscMax(0, grtol);
8276552cf8aSJason Sarich       tao->grtol_changed = PETSC_TRUE;
828a7e14dcfSSatish Balay     }
829a7e14dcfSSatish Balay   }
830a7e14dcfSSatish Balay 
83113bcc0bdSJacob Faibussowitsch   if (gttol != (PetscReal)PETSC_DEFAULT) {
832a7e14dcfSSatish Balay     if (gttol < 0) {
8339566063dSJacob Faibussowitsch       PetscCall(PetscInfo(tao, "Tried to set negative gttol -- ignored.\n"));
834a7e14dcfSSatish Balay     } else {
835a7e14dcfSSatish Balay       tao->gttol         = PetscMax(0, gttol);
8366552cf8aSJason Sarich       tao->gttol_changed = PETSC_TRUE;
837a7e14dcfSSatish Balay     }
838a7e14dcfSSatish Balay   }
8393ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
840a7e14dcfSSatish Balay }
841a7e14dcfSSatish Balay 
842a7e14dcfSSatish Balay /*@
84347450a7bSBarry Smith   TaoSetConstraintTolerances - Sets constraint tolerance parameters used in `TaoSolve()` convergence tests
844a7e14dcfSSatish Balay 
84567be906fSBarry Smith   Logically Collective
846a7e14dcfSSatish Balay 
847a7e14dcfSSatish Balay   Input Parameters:
84847450a7bSBarry Smith + tao   - the `Tao` context
84967be906fSBarry Smith . catol - absolute constraint tolerance, constraint norm must be less than `catol` for used for gatol convergence criteria
85067be906fSBarry Smith - crtol - relative constraint tolerance, constraint norm must be less than `crtol` for used for gatol, gttol convergence criteria
851a7e14dcfSSatish Balay 
852a7e14dcfSSatish Balay   Options Database Keys:
853a7e14dcfSSatish Balay + -tao_catol <catol> - Sets catol
854a7e14dcfSSatish Balay - -tao_crtol <crtol> - Sets crtol
855a7e14dcfSSatish Balay 
856a7e14dcfSSatish Balay   Level: intermediate
857a7e14dcfSSatish Balay 
85867be906fSBarry Smith   Notes:
85967be906fSBarry Smith   Use `PETSC_DEFAULT` to leave any tolerance unchanged.
860a7e14dcfSSatish Balay 
8611cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoConvergedReason`, `TaoGetTolerances()`, `TaoGetConstraintTolerances()`, `TaoSetTolerances()`
862a7e14dcfSSatish Balay @*/
863d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoSetConstraintTolerances(Tao tao, PetscReal catol, PetscReal crtol)
864d71ae5a4SJacob Faibussowitsch {
865a7e14dcfSSatish Balay   PetscFunctionBegin;
866441846f8SBarry Smith   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
86794511df7SStefano Zampini   PetscValidLogicalCollectiveReal(tao, catol, 2);
86894511df7SStefano Zampini   PetscValidLogicalCollectiveReal(tao, crtol, 3);
869a7e14dcfSSatish Balay 
87013bcc0bdSJacob Faibussowitsch   if (catol != (PetscReal)PETSC_DEFAULT) {
871a7e14dcfSSatish Balay     if (catol < 0) {
8729566063dSJacob Faibussowitsch       PetscCall(PetscInfo(tao, "Tried to set negative catol -- ignored.\n"));
873a7e14dcfSSatish Balay     } else {
874a7e14dcfSSatish Balay       tao->catol         = PetscMax(0, catol);
8756552cf8aSJason Sarich       tao->catol_changed = PETSC_TRUE;
876a7e14dcfSSatish Balay     }
877a7e14dcfSSatish Balay   }
878a7e14dcfSSatish Balay 
87913bcc0bdSJacob Faibussowitsch   if (crtol != (PetscReal)PETSC_DEFAULT) {
880a7e14dcfSSatish Balay     if (crtol < 0) {
8819566063dSJacob Faibussowitsch       PetscCall(PetscInfo(tao, "Tried to set negative crtol -- ignored.\n"));
882a7e14dcfSSatish Balay     } else {
883a7e14dcfSSatish Balay       tao->crtol         = PetscMax(0, crtol);
8846552cf8aSJason Sarich       tao->crtol_changed = PETSC_TRUE;
885a7e14dcfSSatish Balay     }
886a7e14dcfSSatish Balay   }
8873ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
888a7e14dcfSSatish Balay }
889a7e14dcfSSatish Balay 
89058417fe7SBarry Smith /*@
89147450a7bSBarry Smith   TaoGetConstraintTolerances - Gets constraint tolerance parameters used in `TaoSolve()` convergence tests
89258417fe7SBarry Smith 
89367be906fSBarry Smith   Not Collective
89458417fe7SBarry Smith 
89558417fe7SBarry Smith   Input Parameter:
89647450a7bSBarry Smith . tao - the `Tao` context
89758417fe7SBarry Smith 
898d8d19677SJose E. Roman   Output Parameters:
89967be906fSBarry Smith + catol - absolute constraint tolerance, constraint norm must be less than `catol` for used for gatol convergence criteria
90067be906fSBarry Smith - crtol - relative constraint tolerance, constraint norm must be less than `crtol` for used for gatol, gttol convergence criteria
90158417fe7SBarry Smith 
90258417fe7SBarry Smith   Level: intermediate
90358417fe7SBarry Smith 
9041cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoConvergedReasons`,`TaoGetTolerances()`, `TaoSetTolerances()`, `TaoSetConstraintTolerances()`
90558417fe7SBarry Smith @*/
906d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoGetConstraintTolerances(Tao tao, PetscReal *catol, PetscReal *crtol)
907d71ae5a4SJacob Faibussowitsch {
90858417fe7SBarry Smith   PetscFunctionBegin;
90958417fe7SBarry Smith   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
91058417fe7SBarry Smith   if (catol) *catol = tao->catol;
91158417fe7SBarry Smith   if (crtol) *crtol = tao->crtol;
9123ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
91358417fe7SBarry Smith }
91458417fe7SBarry Smith 
915a7e14dcfSSatish Balay /*@
916a7e14dcfSSatish Balay   TaoSetFunctionLowerBound - Sets a bound on the solution objective value.
917a7e14dcfSSatish Balay   When an approximate solution with an objective value below this number
918a7e14dcfSSatish Balay   has been found, the solver will terminate.
919a7e14dcfSSatish Balay 
920c3339decSBarry Smith   Logically Collective
921a7e14dcfSSatish Balay 
922a7e14dcfSSatish Balay   Input Parameters:
923441846f8SBarry Smith + tao  - the Tao solver context
924a7e14dcfSSatish Balay - fmin - the tolerance
925a7e14dcfSSatish Balay 
92647450a7bSBarry Smith   Options Database Key:
927a7e14dcfSSatish Balay . -tao_fmin <fmin> - sets the minimum function value
928a7e14dcfSSatish Balay 
929a7e14dcfSSatish Balay   Level: intermediate
930a7e14dcfSSatish Balay 
9311cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoConvergedReason`, `TaoSetTolerances()`
932a7e14dcfSSatish Balay @*/
933d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoSetFunctionLowerBound(Tao tao, PetscReal fmin)
934d71ae5a4SJacob Faibussowitsch {
935a7e14dcfSSatish Balay   PetscFunctionBegin;
936441846f8SBarry Smith   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
93794511df7SStefano Zampini   PetscValidLogicalCollectiveReal(tao, fmin, 2);
938a7e14dcfSSatish Balay   tao->fmin         = fmin;
9396552cf8aSJason Sarich   tao->fmin_changed = PETSC_TRUE;
9403ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
941a7e14dcfSSatish Balay }
942a7e14dcfSSatish Balay 
943a7e14dcfSSatish Balay /*@
9448e96d397SJason Sarich   TaoGetFunctionLowerBound - Gets the bound on the solution objective value.
945a7e14dcfSSatish Balay   When an approximate solution with an objective value below this number
946a7e14dcfSSatish Balay   has been found, the solver will terminate.
947a7e14dcfSSatish Balay 
94867be906fSBarry Smith   Not Collective
949a7e14dcfSSatish Balay 
95047450a7bSBarry Smith   Input Parameter:
95147450a7bSBarry Smith . tao - the `Tao` solver context
952a7e14dcfSSatish Balay 
95347450a7bSBarry Smith   Output Parameter:
954a7e14dcfSSatish Balay . fmin - the minimum function value
955a7e14dcfSSatish Balay 
956a7e14dcfSSatish Balay   Level: intermediate
957a7e14dcfSSatish Balay 
9581cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoConvergedReason`, `TaoSetFunctionLowerBound()`
959a7e14dcfSSatish Balay @*/
960d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoGetFunctionLowerBound(Tao tao, PetscReal *fmin)
961d71ae5a4SJacob Faibussowitsch {
962a7e14dcfSSatish Balay   PetscFunctionBegin;
963441846f8SBarry Smith   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
9644f572ea9SToby Isaac   PetscAssertPointer(fmin, 2);
965a7e14dcfSSatish Balay   *fmin = tao->fmin;
9663ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
967a7e14dcfSSatish Balay }
968a7e14dcfSSatish Balay 
969a7e14dcfSSatish Balay /*@
97047450a7bSBarry Smith   TaoSetMaximumFunctionEvaluations - Sets a maximum number of function evaluations allowed for a `TaoSolve()`.
971a7e14dcfSSatish Balay 
972c3339decSBarry Smith   Logically Collective
973a7e14dcfSSatish Balay 
974a7e14dcfSSatish Balay   Input Parameters:
97547450a7bSBarry Smith + tao  - the `Tao` solver context
976a7e14dcfSSatish Balay - nfcn - the maximum number of function evaluations (>=0)
977a7e14dcfSSatish Balay 
97847450a7bSBarry Smith   Options Database Key:
979a7e14dcfSSatish Balay . -tao_max_funcs <nfcn> - sets the maximum number of function evaluations
980a7e14dcfSSatish Balay 
981a7e14dcfSSatish Balay   Level: intermediate
982a7e14dcfSSatish Balay 
9831cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoSetTolerances()`, `TaoSetMaximumIterations()`
984a7e14dcfSSatish Balay @*/
985d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoSetMaximumFunctionEvaluations(Tao tao, PetscInt nfcn)
986d71ae5a4SJacob Faibussowitsch {
987a7e14dcfSSatish Balay   PetscFunctionBegin;
988441846f8SBarry Smith   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
98994511df7SStefano Zampini   PetscValidLogicalCollectiveInt(tao, nfcn, 2);
9909371c9d4SSatish Balay   if (nfcn >= 0) {
9919371c9d4SSatish Balay     tao->max_funcs = PetscMax(0, nfcn);
9929371c9d4SSatish Balay   } else {
9939371c9d4SSatish Balay     tao->max_funcs = -1;
9949371c9d4SSatish Balay   }
9956552cf8aSJason Sarich   tao->max_funcs_changed = PETSC_TRUE;
9963ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
997a7e14dcfSSatish Balay }
998a7e14dcfSSatish Balay 
999a7e14dcfSSatish Balay /*@
100047450a7bSBarry Smith   TaoGetMaximumFunctionEvaluations - Gets a maximum number of function evaluations allowed for a `TaoSolve()`
1001a7e14dcfSSatish Balay 
1002c3339decSBarry Smith   Logically Collective
1003a7e14dcfSSatish Balay 
100447450a7bSBarry Smith   Input Parameter:
100547450a7bSBarry Smith . tao - the `Tao` solver context
1006a7e14dcfSSatish Balay 
100747450a7bSBarry Smith   Output Parameter:
1008a7e14dcfSSatish Balay . nfcn - the maximum number of function evaluations
1009a7e14dcfSSatish Balay 
1010a7e14dcfSSatish Balay   Level: intermediate
1011a7e14dcfSSatish Balay 
10121cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoSetMaximumFunctionEvaluations()`, `TaoGetMaximumIterations()`
1013a7e14dcfSSatish Balay @*/
1014d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoGetMaximumFunctionEvaluations(Tao tao, PetscInt *nfcn)
1015d71ae5a4SJacob Faibussowitsch {
1016a7e14dcfSSatish Balay   PetscFunctionBegin;
1017441846f8SBarry Smith   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
10184f572ea9SToby Isaac   PetscAssertPointer(nfcn, 2);
1019a7e14dcfSSatish Balay   *nfcn = tao->max_funcs;
10203ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
1021a7e14dcfSSatish Balay }
1022a7e14dcfSSatish Balay 
1023770232b9SCe Qin /*@
102447450a7bSBarry Smith   TaoGetCurrentFunctionEvaluations - Get current number of function evaluations used by a `Tao` object
1025770232b9SCe Qin 
1026770232b9SCe Qin   Not Collective
1027770232b9SCe Qin 
102847450a7bSBarry Smith   Input Parameter:
102947450a7bSBarry Smith . tao - the `Tao` solver context
1030770232b9SCe Qin 
103147450a7bSBarry Smith   Output Parameter:
103294511df7SStefano Zampini . nfuncs - the current number of function evaluations (maximum between gradient and function evaluations)
1033770232b9SCe Qin 
1034770232b9SCe Qin   Level: intermediate
1035770232b9SCe Qin 
10361cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoSetMaximumFunctionEvaluations()`, `TaoGetMaximumFunctionEvaluations()`, `TaoGetMaximumIterations()`
1037770232b9SCe Qin @*/
1038d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoGetCurrentFunctionEvaluations(Tao tao, PetscInt *nfuncs)
1039d71ae5a4SJacob Faibussowitsch {
1040770232b9SCe Qin   PetscFunctionBegin;
1041770232b9SCe Qin   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
10424f572ea9SToby Isaac   PetscAssertPointer(nfuncs, 2);
1043770232b9SCe Qin   *nfuncs = PetscMax(tao->nfuncs, tao->nfuncgrads);
10443ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
1045770232b9SCe Qin }
1046770232b9SCe Qin 
1047a7e14dcfSSatish Balay /*@
104847450a7bSBarry Smith   TaoSetMaximumIterations - Sets a maximum number of iterates to be used in `TaoSolve()`
1049a7e14dcfSSatish Balay 
1050c3339decSBarry Smith   Logically Collective
1051a7e14dcfSSatish Balay 
1052a7e14dcfSSatish Balay   Input Parameters:
105347450a7bSBarry Smith + tao    - the `Tao` solver context
1054a7e14dcfSSatish Balay - maxits - the maximum number of iterates (>=0)
1055a7e14dcfSSatish Balay 
105647450a7bSBarry Smith   Options Database Key:
1057a7e14dcfSSatish Balay . -tao_max_it <its> - sets the maximum number of iterations
1058a7e14dcfSSatish Balay 
1059a7e14dcfSSatish Balay   Level: intermediate
1060a7e14dcfSSatish Balay 
10611cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoSetTolerances()`, `TaoSetMaximumFunctionEvaluations()`
1062a7e14dcfSSatish Balay @*/
1063d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoSetMaximumIterations(Tao tao, PetscInt maxits)
1064d71ae5a4SJacob Faibussowitsch {
1065a7e14dcfSSatish Balay   PetscFunctionBegin;
1066441846f8SBarry Smith   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
106794511df7SStefano Zampini   PetscValidLogicalCollectiveInt(tao, maxits, 2);
106847a47007SBarry Smith   tao->max_it         = PetscMax(0, maxits);
10696552cf8aSJason Sarich   tao->max_it_changed = PETSC_TRUE;
10703ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
1071a7e14dcfSSatish Balay }
1072a7e14dcfSSatish Balay 
1073a7e14dcfSSatish Balay /*@
107465ba42b6SBarry Smith   TaoGetMaximumIterations - Gets a maximum number of iterates that will be used
1075a7e14dcfSSatish Balay 
1076a7e14dcfSSatish Balay   Not Collective
1077a7e14dcfSSatish Balay 
107847450a7bSBarry Smith   Input Parameter:
107947450a7bSBarry Smith . tao - the `Tao` solver context
1080a7e14dcfSSatish Balay 
108147450a7bSBarry Smith   Output Parameter:
1082a7e14dcfSSatish Balay . maxits - the maximum number of iterates
1083a7e14dcfSSatish Balay 
1084a7e14dcfSSatish Balay   Level: intermediate
1085a7e14dcfSSatish Balay 
10861cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoSetMaximumIterations()`, `TaoGetMaximumFunctionEvaluations()`
1087a7e14dcfSSatish Balay @*/
1088d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoGetMaximumIterations(Tao tao, PetscInt *maxits)
1089d71ae5a4SJacob Faibussowitsch {
1090a7e14dcfSSatish Balay   PetscFunctionBegin;
1091441846f8SBarry Smith   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
10924f572ea9SToby Isaac   PetscAssertPointer(maxits, 2);
1093a7e14dcfSSatish Balay   *maxits = tao->max_it;
10943ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
1095a7e14dcfSSatish Balay }
1096a7e14dcfSSatish Balay 
1097a7e14dcfSSatish Balay /*@
1098a7e14dcfSSatish Balay   TaoSetInitialTrustRegionRadius - Sets the initial trust region radius.
1099a7e14dcfSSatish Balay 
110067be906fSBarry Smith   Logically Collective
1101a7e14dcfSSatish Balay 
1102d8d19677SJose E. Roman   Input Parameters:
110347450a7bSBarry Smith + tao    - a `Tao` optimization solver
1104a7e14dcfSSatish Balay - radius - the trust region radius
1105a7e14dcfSSatish Balay 
1106a7e14dcfSSatish Balay   Options Database Key:
1107a7e14dcfSSatish Balay . -tao_trust0 <t0> - sets initial trust region radius
1108a7e14dcfSSatish Balay 
110947450a7bSBarry Smith   Level: intermediate
111047450a7bSBarry Smith 
11111cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoGetTrustRegionRadius()`, `TaoSetTrustRegionTolerance()`, `TAONTR`
1112a7e14dcfSSatish Balay @*/
1113d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoSetInitialTrustRegionRadius(Tao tao, PetscReal radius)
1114d71ae5a4SJacob Faibussowitsch {
1115a7e14dcfSSatish Balay   PetscFunctionBegin;
1116441846f8SBarry Smith   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
111794511df7SStefano Zampini   PetscValidLogicalCollectiveReal(tao, radius, 2);
1118a7e14dcfSSatish Balay   tao->trust0         = PetscMax(0.0, radius);
11196552cf8aSJason Sarich   tao->trust0_changed = PETSC_TRUE;
11203ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
1121a7e14dcfSSatish Balay }
1122a7e14dcfSSatish Balay 
1123a7e14dcfSSatish Balay /*@
112465ba42b6SBarry Smith   TaoGetInitialTrustRegionRadius - Gets the initial trust region radius.
1125a7e14dcfSSatish Balay 
1126a7e14dcfSSatish Balay   Not Collective
1127a7e14dcfSSatish Balay 
1128a7e14dcfSSatish Balay   Input Parameter:
112947450a7bSBarry Smith . tao - a `Tao` optimization solver
1130a7e14dcfSSatish Balay 
1131a7e14dcfSSatish Balay   Output Parameter:
1132a7e14dcfSSatish Balay . radius - the trust region radius
1133a7e14dcfSSatish Balay 
1134a7e14dcfSSatish Balay   Level: intermediate
1135a7e14dcfSSatish Balay 
11361cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoSetInitialTrustRegionRadius()`, `TaoGetCurrentTrustRegionRadius()`, `TAONTR`
1137a7e14dcfSSatish Balay @*/
1138d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoGetInitialTrustRegionRadius(Tao tao, PetscReal *radius)
1139d71ae5a4SJacob Faibussowitsch {
1140a7e14dcfSSatish Balay   PetscFunctionBegin;
1141441846f8SBarry Smith   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
11424f572ea9SToby Isaac   PetscAssertPointer(radius, 2);
1143a7e14dcfSSatish Balay   *radius = tao->trust0;
11443ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
1145a7e14dcfSSatish Balay }
1146a7e14dcfSSatish Balay 
1147a7e14dcfSSatish Balay /*@
1148a7e14dcfSSatish Balay   TaoGetCurrentTrustRegionRadius - Gets the current trust region radius.
1149a7e14dcfSSatish Balay 
1150a7e14dcfSSatish Balay   Not Collective
1151a7e14dcfSSatish Balay 
1152a7e14dcfSSatish Balay   Input Parameter:
115347450a7bSBarry Smith . tao - a `Tao` optimization solver
1154a7e14dcfSSatish Balay 
1155a7e14dcfSSatish Balay   Output Parameter:
1156a7e14dcfSSatish Balay . radius - the trust region radius
1157a7e14dcfSSatish Balay 
1158a7e14dcfSSatish Balay   Level: intermediate
1159a7e14dcfSSatish Balay 
11601cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoSetInitialTrustRegionRadius()`, `TaoGetInitialTrustRegionRadius()`, `TAONTR`
1161a7e14dcfSSatish Balay @*/
1162d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoGetCurrentTrustRegionRadius(Tao tao, PetscReal *radius)
1163d71ae5a4SJacob Faibussowitsch {
1164a7e14dcfSSatish Balay   PetscFunctionBegin;
1165441846f8SBarry Smith   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
11664f572ea9SToby Isaac   PetscAssertPointer(radius, 2);
1167a7e14dcfSSatish Balay   *radius = tao->trust;
11683ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
1169a7e14dcfSSatish Balay }
1170a7e14dcfSSatish Balay 
1171a7e14dcfSSatish Balay /*@
117247450a7bSBarry Smith   TaoGetTolerances - gets the current values of some tolerances used for the convergence testing of `TaoSolve()`
1173a7e14dcfSSatish Balay 
1174a7e14dcfSSatish Balay   Not Collective
1175a7e14dcfSSatish Balay 
1176f899ff85SJose E. Roman   Input Parameter:
117747450a7bSBarry Smith . tao - the `Tao` context
1178a7e14dcfSSatish Balay 
1179a7e14dcfSSatish Balay   Output Parameters:
1180e52336cbSBarry Smith + gatol - stop if norm of gradient is less than this
1181a7e14dcfSSatish Balay . grtol - stop if relative norm of gradient is less than this
1182a7e14dcfSSatish Balay - gttol - stop if norm of gradient is reduced by a this factor
1183a7e14dcfSSatish Balay 
1184a7e14dcfSSatish Balay   Level: intermediate
1185a1cb98faSBarry Smith 
1186a1cb98faSBarry Smith   Note:
118747450a7bSBarry Smith   `NULL` can be used as an argument if not all tolerances values are needed
1188a1cb98faSBarry Smith 
11891cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoSetTolerances()`
1190a7e14dcfSSatish Balay @*/
1191d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoGetTolerances(Tao tao, PetscReal *gatol, PetscReal *grtol, PetscReal *gttol)
1192d71ae5a4SJacob Faibussowitsch {
1193a7e14dcfSSatish Balay   PetscFunctionBegin;
1194441846f8SBarry Smith   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
1195a7e14dcfSSatish Balay   if (gatol) *gatol = tao->gatol;
1196a7e14dcfSSatish Balay   if (grtol) *grtol = tao->grtol;
1197a7e14dcfSSatish Balay   if (gttol) *gttol = tao->gttol;
11983ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
1199a7e14dcfSSatish Balay }
1200a7e14dcfSSatish Balay 
1201a7e14dcfSSatish Balay /*@
1202a7e14dcfSSatish Balay   TaoGetKSP - Gets the linear solver used by the optimization solver.
1203a7e14dcfSSatish Balay 
1204a7e14dcfSSatish Balay   Not Collective
1205a7e14dcfSSatish Balay 
120647450a7bSBarry Smith   Input Parameter:
120747450a7bSBarry Smith . tao - the `Tao` solver
1208a7e14dcfSSatish Balay 
120947450a7bSBarry Smith   Output Parameter:
121047450a7bSBarry Smith . ksp - the `KSP` linear solver used in the optimization solver
1211a7e14dcfSSatish Balay 
1212a7e14dcfSSatish Balay   Level: intermediate
1213a7e14dcfSSatish Balay 
12141cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `KSP`
1215a7e14dcfSSatish Balay @*/
1216d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoGetKSP(Tao tao, KSP *ksp)
1217d71ae5a4SJacob Faibussowitsch {
1218a7e14dcfSSatish Balay   PetscFunctionBegin;
121994511df7SStefano Zampini   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
12204f572ea9SToby Isaac   PetscAssertPointer(ksp, 2);
1221a7e14dcfSSatish Balay   *ksp = tao->ksp;
12223ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
1223a7e14dcfSSatish Balay }
1224a7e14dcfSSatish Balay 
1225025e9500SJason Sarich /*@
1226025e9500SJason Sarich   TaoGetLinearSolveIterations - Gets the total number of linear iterations
122747450a7bSBarry Smith   used by the `Tao` solver
1228025e9500SJason Sarich 
1229025e9500SJason Sarich   Not Collective
1230025e9500SJason Sarich 
1231025e9500SJason Sarich   Input Parameter:
123247450a7bSBarry Smith . tao - the `Tao` context
1233025e9500SJason Sarich 
1234025e9500SJason Sarich   Output Parameter:
1235025e9500SJason Sarich . lits - number of linear iterations
1236025e9500SJason Sarich 
1237025e9500SJason Sarich   Level: intermediate
1238025e9500SJason Sarich 
123947450a7bSBarry Smith   Note:
124047450a7bSBarry Smith   This counter is reset to zero for each successive call to `TaoSolve()`
124147450a7bSBarry Smith 
12421cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoGetKSP()`
1243025e9500SJason Sarich @*/
1244d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoGetLinearSolveIterations(Tao tao, PetscInt *lits)
1245d71ae5a4SJacob Faibussowitsch {
1246025e9500SJason Sarich   PetscFunctionBegin;
1247025e9500SJason Sarich   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
12484f572ea9SToby Isaac   PetscAssertPointer(lits, 2);
12492d9aa51bSJason Sarich   *lits = tao->ksp_tot_its;
12503ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
1251025e9500SJason Sarich }
1252025e9500SJason Sarich 
1253a7e14dcfSSatish Balay /*@
1254a7e14dcfSSatish Balay   TaoGetLineSearch - Gets the line search used by the optimization solver.
1255a7e14dcfSSatish Balay 
1256a7e14dcfSSatish Balay   Not Collective
1257a7e14dcfSSatish Balay 
125847450a7bSBarry Smith   Input Parameter:
125947450a7bSBarry Smith . tao - the `Tao` solver
1260a7e14dcfSSatish Balay 
126147450a7bSBarry Smith   Output Parameter:
1262a7e14dcfSSatish Balay . ls - the line search used in the optimization solver
1263a7e14dcfSSatish Balay 
1264a7e14dcfSSatish Balay   Level: intermediate
1265a7e14dcfSSatish Balay 
12661cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoLineSearch`, `TaoLineSearchType`
1267a7e14dcfSSatish Balay @*/
1268d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoGetLineSearch(Tao tao, TaoLineSearch *ls)
1269d71ae5a4SJacob Faibussowitsch {
1270a7e14dcfSSatish Balay   PetscFunctionBegin;
127194511df7SStefano Zampini   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
12724f572ea9SToby Isaac   PetscAssertPointer(ls, 2);
1273a7e14dcfSSatish Balay   *ls = tao->linesearch;
12743ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
1275a7e14dcfSSatish Balay }
1276a7e14dcfSSatish Balay 
1277a7e14dcfSSatish Balay /*@
1278a7e14dcfSSatish Balay   TaoAddLineSearchCounts - Adds the number of function evaluations spent
1279a7e14dcfSSatish Balay   in the line search to the running total.
1280a7e14dcfSSatish Balay 
1281a7e14dcfSSatish Balay   Input Parameters:
12822fe279fdSBarry Smith . tao - the `Tao` solver
1283a7e14dcfSSatish Balay 
1284a7e14dcfSSatish Balay   Level: developer
1285a7e14dcfSSatish Balay 
12861cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoGetLineSearch()`, `TaoLineSearchApply()`
1287a7e14dcfSSatish Balay @*/
1288d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoAddLineSearchCounts(Tao tao)
1289d71ae5a4SJacob Faibussowitsch {
1290a7e14dcfSSatish Balay   PetscBool flg;
1291a7e14dcfSSatish Balay   PetscInt  nfeval, ngeval, nfgeval;
129247a47007SBarry Smith 
1293a7e14dcfSSatish Balay   PetscFunctionBegin;
1294441846f8SBarry Smith   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
1295a7e14dcfSSatish Balay   if (tao->linesearch) {
12969566063dSJacob Faibussowitsch     PetscCall(TaoLineSearchIsUsingTaoRoutines(tao->linesearch, &flg));
129794ae4db5SBarry Smith     if (!flg) {
12989566063dSJacob Faibussowitsch       PetscCall(TaoLineSearchGetNumberFunctionEvaluations(tao->linesearch, &nfeval, &ngeval, &nfgeval));
1299a7e14dcfSSatish Balay       tao->nfuncs += nfeval;
1300a7e14dcfSSatish Balay       tao->ngrads += ngeval;
1301a7e14dcfSSatish Balay       tao->nfuncgrads += nfgeval;
1302a7e14dcfSSatish Balay     }
1303a7e14dcfSSatish Balay   }
13043ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
1305a7e14dcfSSatish Balay }
1306a7e14dcfSSatish Balay 
1307a7e14dcfSSatish Balay /*@
130847450a7bSBarry Smith   TaoGetSolution - Returns the vector with the current solution from the `Tao` object
1309a7e14dcfSSatish Balay 
1310a7e14dcfSSatish Balay   Not Collective
1311a7e14dcfSSatish Balay 
1312a7e14dcfSSatish Balay   Input Parameter:
131347450a7bSBarry Smith . tao - the `Tao` context
1314a7e14dcfSSatish Balay 
1315a7e14dcfSSatish Balay   Output Parameter:
1316a7e14dcfSSatish Balay . X - the current solution
1317a7e14dcfSSatish Balay 
1318a7e14dcfSSatish Balay   Level: intermediate
1319a7e14dcfSSatish Balay 
132065ba42b6SBarry Smith   Note:
132165ba42b6SBarry Smith   The returned vector will be the same object that was passed into `TaoSetSolution()`
132265ba42b6SBarry Smith 
13231cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoSetSolution()`, `TaoSolve()`
1324a7e14dcfSSatish Balay @*/
1325d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoGetSolution(Tao tao, Vec *X)
1326d71ae5a4SJacob Faibussowitsch {
1327a7e14dcfSSatish Balay   PetscFunctionBegin;
1328441846f8SBarry Smith   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
13294f572ea9SToby Isaac   PetscAssertPointer(X, 2);
1330a7e14dcfSSatish Balay   *X = tao->solution;
13313ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
1332a7e14dcfSSatish Balay }
1333a7e14dcfSSatish Balay 
1334a7e14dcfSSatish Balay /*@
133547450a7bSBarry Smith   TaoResetStatistics - Initialize the statistics collected by the `Tao` object.
1336a7e14dcfSSatish Balay   These statistics include the iteration number, residual norms, and convergence status.
1337a7e14dcfSSatish Balay   This routine gets called before solving each optimization problem.
1338a7e14dcfSSatish Balay 
1339c3339decSBarry Smith   Collective
1340a7e14dcfSSatish Balay 
134147450a7bSBarry Smith   Input Parameter:
1342e056e8ceSJacob Faibussowitsch . tao - the `Tao` context
1343a7e14dcfSSatish Balay 
1344a7e14dcfSSatish Balay   Level: developer
1345a7e14dcfSSatish Balay 
13461cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoCreate()`, `TaoSolve()`
1347a7e14dcfSSatish Balay @*/
1348d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoResetStatistics(Tao tao)
1349d71ae5a4SJacob Faibussowitsch {
1350a7e14dcfSSatish Balay   PetscFunctionBegin;
1351441846f8SBarry Smith   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
1352a7e14dcfSSatish Balay   tao->niter        = 0;
1353a7e14dcfSSatish Balay   tao->nfuncs       = 0;
1354a7e14dcfSSatish Balay   tao->nfuncgrads   = 0;
1355a7e14dcfSSatish Balay   tao->ngrads       = 0;
1356a7e14dcfSSatish Balay   tao->nhess        = 0;
1357a7e14dcfSSatish Balay   tao->njac         = 0;
1358a7e14dcfSSatish Balay   tao->nconstraints = 0;
1359a7e14dcfSSatish Balay   tao->ksp_its      = 0;
1360ae93cb3cSJason Sarich   tao->ksp_tot_its  = 0;
1361a7e14dcfSSatish Balay   tao->reason       = TAO_CONTINUE_ITERATING;
1362a7e14dcfSSatish Balay   tao->residual     = 0.0;
1363a7e14dcfSSatish Balay   tao->cnorm        = 0.0;
1364a7e14dcfSSatish Balay   tao->step         = 0.0;
1365a7e14dcfSSatish Balay   tao->lsflag       = PETSC_FALSE;
1366a7e14dcfSSatish Balay   if (tao->hist_reset) tao->hist_len = 0;
13673ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
1368a7e14dcfSSatish Balay }
1369a7e14dcfSSatish Balay 
1370a7e14dcfSSatish Balay /*@C
1371e1e80dc8SAlp Dener   TaoSetUpdate - Sets the general-purpose update function called
13722fe279fdSBarry Smith   at the beginning of every iteration of the optimization algorithm. Called after the new solution and the gradient
1373e1e80dc8SAlp Dener   is determined, but before the Hessian is computed (if applicable).
1374e1e80dc8SAlp Dener 
1375c3339decSBarry Smith   Logically Collective
1376e1e80dc8SAlp Dener 
1377e1e80dc8SAlp Dener   Input Parameters:
13782fe279fdSBarry Smith + tao  - The `Tao` solver context
1379a2b725a8SWilliam Gropp - func - The function
1380e1e80dc8SAlp Dener 
138120f4b53cSBarry Smith   Calling sequence of `func`:
138220f4b53cSBarry Smith + tao - the optimizer context
1383e056e8ceSJacob Faibussowitsch - ctx - The current step of the iteration
1384e1e80dc8SAlp Dener 
1385e1e80dc8SAlp Dener   Level: advanced
1386e1e80dc8SAlp Dener 
13871cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoSolve()`
1388e1e80dc8SAlp Dener @*/
1389d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoSetUpdate(Tao tao, PetscErrorCode (*func)(Tao, PetscInt, void *), void *ctx)
1390d71ae5a4SJacob Faibussowitsch {
1391e1e80dc8SAlp Dener   PetscFunctionBegin;
1392e1e80dc8SAlp Dener   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
1393e1e80dc8SAlp Dener   tao->ops->update = func;
1394e1e80dc8SAlp Dener   tao->user_update = ctx;
13953ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
1396e1e80dc8SAlp Dener }
1397e1e80dc8SAlp Dener 
1398e1e80dc8SAlp Dener /*@C
1399a7e14dcfSSatish Balay   TaoSetConvergenceTest - Sets the function that is to be used to test
1400a7e14dcfSSatish Balay   for convergence o fthe iterative minimization solution.  The new convergence
140165ba42b6SBarry Smith   testing routine will replace Tao's default convergence test.
1402a7e14dcfSSatish Balay 
1403c3339decSBarry Smith   Logically Collective
1404a7e14dcfSSatish Balay 
1405a7e14dcfSSatish Balay   Input Parameters:
140647450a7bSBarry Smith + tao  - the `Tao` object
1407a7e14dcfSSatish Balay . conv - the routine to test for convergence
1408a7e14dcfSSatish Balay - ctx  - [optional] context for private data for the convergence routine
140947450a7bSBarry Smith         (may be `NULL`)
1410a7e14dcfSSatish Balay 
141120f4b53cSBarry Smith   Calling sequence of `conv`:
141247450a7bSBarry Smith + tao - the `Tao` object
1413a7e14dcfSSatish Balay - ctx - [optional] convergence context
1414a7e14dcfSSatish Balay 
141547450a7bSBarry Smith   Level: advanced
141647450a7bSBarry Smith 
141765ba42b6SBarry Smith   Note:
141865ba42b6SBarry Smith   The new convergence testing routine should call `TaoSetConvergedReason()`.
1419a7e14dcfSSatish Balay 
14201cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoSolve()`, `TaoSetConvergedReason()`, `TaoGetSolutionStatus()`, `TaoGetTolerances()`, `TaoSetMonitor`
1421a7e14dcfSSatish Balay @*/
1422d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoSetConvergenceTest(Tao tao, PetscErrorCode (*conv)(Tao, void *), void *ctx)
1423d71ae5a4SJacob Faibussowitsch {
1424a7e14dcfSSatish Balay   PetscFunctionBegin;
1425441846f8SBarry Smith   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
142694511df7SStefano Zampini   tao->ops->convergencetest = conv;
142794511df7SStefano Zampini   tao->cnvP                 = ctx;
14283ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
1429a7e14dcfSSatish Balay }
1430a7e14dcfSSatish Balay 
1431a7e14dcfSSatish Balay /*@C
143265ba42b6SBarry Smith   TaoSetMonitor - Sets an additional function that is to be used at every
1433a7e14dcfSSatish Balay   iteration of the solver to display the iteration's
1434a7e14dcfSSatish Balay   progress.
1435a7e14dcfSSatish Balay 
1436c3339decSBarry Smith   Logically Collective
1437a7e14dcfSSatish Balay 
1438a7e14dcfSSatish Balay   Input Parameters:
143947450a7bSBarry Smith + tao  - the `Tao` solver context
14402fe279fdSBarry Smith . func - monitoring routine
14412fe279fdSBarry Smith . ctx  - [optional] user-defined context for private data for the monitor routine (may be `NULL`)
14422fe279fdSBarry Smith - dest - [optional] function to destroy the context when the `Tao` is destroyed
1443a7e14dcfSSatish Balay 
14442fe279fdSBarry Smith   Calling sequence of `func`:
144547450a7bSBarry Smith + tao - the `Tao` solver context
14462fe279fdSBarry Smith - ctx - [optional] monitoring context
14472fe279fdSBarry Smith 
14482fe279fdSBarry Smith   Calling sequence of `dest`:
14492fe279fdSBarry Smith . ctx - monitoring context
1450a7e14dcfSSatish Balay 
1451a7e14dcfSSatish Balay   Options Database Keys:
145265ba42b6SBarry Smith + -tao_monitor          - sets the default monitor `TaoMonitorDefault()`
1453a7e14dcfSSatish Balay . -tao_smonitor         - sets short monitor
1454a7e14dcfSSatish Balay . -tao_cmonitor         - same as smonitor plus constraint norm
1455a7e14dcfSSatish Balay . -tao_view_solution    - view solution at each iteration
1456a7e14dcfSSatish Balay . -tao_view_gradient    - view gradient at each iteration
14574a48860cSAlp Dener . -tao_view_ls_residual - view least-squares residual vector at each iteration
1458a7e14dcfSSatish 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.
1459a7e14dcfSSatish Balay 
146047450a7bSBarry Smith   Level: intermediate
146147450a7bSBarry Smith 
146287497f52SBarry Smith   Notes:
1463a7e14dcfSSatish Balay   Several different monitoring routines may be set by calling
146465ba42b6SBarry Smith   `TaoSetMonitor()` multiple times; all will be called in the
1465a7e14dcfSSatish Balay   order in which they were set.
1466a7e14dcfSSatish Balay 
1467e056e8ceSJacob Faibussowitsch   Fortran Notes:
146895452b02SPatrick Sanan   Only one monitor function may be set
1469a7e14dcfSSatish Balay 
14701cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoSolve()`, `TaoMonitorDefault()`, `TaoCancelMonitors()`, `TaoSetDestroyRoutine()`, `TaoView()`
1471a7e14dcfSSatish Balay @*/
1472d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoSetMonitor(Tao tao, PetscErrorCode (*func)(Tao, void *), void *ctx, PetscErrorCode (*dest)(void **))
1473d71ae5a4SJacob Faibussowitsch {
147408d19d1fSJason Sarich   PetscInt  i;
14758732526dSAlp Dener   PetscBool identical;
147608d19d1fSJason Sarich 
1477a7e14dcfSSatish Balay   PetscFunctionBegin;
1478441846f8SBarry Smith   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
14793c859ba3SBarry Smith   PetscCheck(tao->numbermonitors < MAXTAOMONITORS, PetscObjectComm((PetscObject)tao), PETSC_ERR_SUP, "Cannot attach another monitor -- max=%d", MAXTAOMONITORS);
148008d19d1fSJason Sarich 
148108d19d1fSJason Sarich   for (i = 0; i < tao->numbermonitors; i++) {
14829566063dSJacob Faibussowitsch     PetscCall(PetscMonitorCompare((PetscErrorCode(*)(void))func, ctx, dest, (PetscErrorCode(*)(void))tao->monitor[i], tao->monitorcontext[i], tao->monitordestroy[i], &identical));
14833ba16761SJacob Faibussowitsch     if (identical) PetscFunctionReturn(PETSC_SUCCESS);
148408d19d1fSJason Sarich   }
1485a7e14dcfSSatish Balay   tao->monitor[tao->numbermonitors]        = func;
14868732526dSAlp Dener   tao->monitorcontext[tao->numbermonitors] = (void *)ctx;
1487a7e14dcfSSatish Balay   tao->monitordestroy[tao->numbermonitors] = dest;
1488a7e14dcfSSatish Balay   ++tao->numbermonitors;
14893ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
1490a7e14dcfSSatish Balay }
1491a7e14dcfSSatish Balay 
1492a7e14dcfSSatish Balay /*@
149347450a7bSBarry Smith   TaoCancelMonitors - Clears all the monitor functions for a `Tao` object.
1494a7e14dcfSSatish Balay 
1495c3339decSBarry Smith   Logically Collective
1496a7e14dcfSSatish Balay 
149747450a7bSBarry Smith   Input Parameter:
149847450a7bSBarry Smith . tao - the `Tao` solver context
1499a7e14dcfSSatish Balay 
15003c7db156SBarry Smith   Options Database Key:
1501a7e14dcfSSatish Balay . -tao_cancelmonitors - cancels all monitors that have been hardwired
150265ba42b6SBarry Smith     into a code by calls to `TaoSetMonitor()`, but does not cancel those
1503a7e14dcfSSatish Balay     set via the options database
1504a7e14dcfSSatish Balay 
1505a7e14dcfSSatish Balay   Level: advanced
1506a7e14dcfSSatish Balay 
150747450a7bSBarry Smith   Note:
150847450a7bSBarry Smith   There is no way to clear one specific monitor from a `Tao` object.
150947450a7bSBarry Smith 
15101cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoMonitorDefault()`, `TaoSetMonitor()`
1511a7e14dcfSSatish Balay @*/
1512d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoCancelMonitors(Tao tao)
1513d71ae5a4SJacob Faibussowitsch {
1514a7e14dcfSSatish Balay   PetscInt i;
151547a47007SBarry Smith 
1516a7e14dcfSSatish Balay   PetscFunctionBegin;
1517441846f8SBarry Smith   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
1518a7e14dcfSSatish Balay   for (i = 0; i < tao->numbermonitors; i++) {
151948a46eb9SPierre Jolivet     if (tao->monitordestroy[i]) PetscCall((*tao->monitordestroy[i])(&tao->monitorcontext[i]));
1520a7e14dcfSSatish Balay   }
1521a7e14dcfSSatish Balay   tao->numbermonitors = 0;
15223ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
1523a7e14dcfSSatish Balay }
1524a7e14dcfSSatish Balay 
15257fab98ebSJason Sarich /*@
152647450a7bSBarry Smith   TaoMonitorDefault - Default routine for monitoring progress of `TaoSolve()`
1527a7e14dcfSSatish Balay 
1528c3339decSBarry Smith   Collective
1529a7e14dcfSSatish Balay 
1530a7e14dcfSSatish Balay   Input Parameters:
153147450a7bSBarry Smith + tao - the `Tao` context
153247450a7bSBarry Smith - ctx - `PetscViewer` context or `NULL`
1533a7e14dcfSSatish Balay 
153447450a7bSBarry Smith   Options Database Key:
1535147403d9SBarry Smith . -tao_monitor - turn on default monitoring
1536a7e14dcfSSatish Balay 
1537a7e14dcfSSatish Balay   Level: advanced
1538a7e14dcfSSatish Balay 
153947450a7bSBarry Smith   Note:
154047450a7bSBarry Smith   This monitor prints the function value and gradient
154147450a7bSBarry Smith   norm at each iteration.
154247450a7bSBarry Smith 
15431cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoDefaultSMonitor()`, `TaoSetMonitor()`
1544a7e14dcfSSatish Balay @*/
1545d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoMonitorDefault(Tao tao, void *ctx)
1546d71ae5a4SJacob Faibussowitsch {
154763b15415SAlp Dener   PetscInt    its, tabs;
1548a7e14dcfSSatish Balay   PetscReal   fct, gnorm;
15498163d661SBarry Smith   PetscViewer viewer = (PetscViewer)ctx;
155047a47007SBarry Smith 
1551a7e14dcfSSatish Balay   PetscFunctionBegin;
155294511df7SStefano Zampini   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
15538163d661SBarry Smith   PetscValidHeaderSpecific(viewer, PETSC_VIEWER_CLASSID, 2);
1554a7e14dcfSSatish Balay   its   = tao->niter;
1555a7e14dcfSSatish Balay   fct   = tao->fc;
1556a7e14dcfSSatish Balay   gnorm = tao->residual;
15579566063dSJacob Faibussowitsch   PetscCall(PetscViewerASCIIGetTab(viewer, &tabs));
15589566063dSJacob Faibussowitsch   PetscCall(PetscViewerASCIISetTab(viewer, ((PetscObject)tao)->tablevel));
1559494bef23SAlp Dener   if (its == 0 && ((PetscObject)tao)->prefix && !tao->header_printed) {
15609566063dSJacob Faibussowitsch     PetscCall(PetscViewerASCIIPrintf(viewer, "  Iteration information for %s solve.\n", ((PetscObject)tao)->prefix));
1561494bef23SAlp Dener     tao->header_printed = PETSC_TRUE;
156263b15415SAlp Dener   }
156363a3b9bcSJacob Faibussowitsch   PetscCall(PetscViewerASCIIPrintf(viewer, "%3" PetscInt_FMT " TAO,", its));
15649566063dSJacob Faibussowitsch   PetscCall(PetscViewerASCIIPrintf(viewer, "  Function value: %g,", (double)fct));
156508d19d1fSJason Sarich   if (gnorm >= PETSC_INFINITY) {
15669566063dSJacob Faibussowitsch     PetscCall(PetscViewerASCIIPrintf(viewer, "  Residual: Inf \n"));
156708d19d1fSJason Sarich   } else {
15689566063dSJacob Faibussowitsch     PetscCall(PetscViewerASCIIPrintf(viewer, "  Residual: %g \n", (double)gnorm));
156908d19d1fSJason Sarich   }
15709566063dSJacob Faibussowitsch   PetscCall(PetscViewerASCIISetTab(viewer, tabs));
15713ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
1572a7e14dcfSSatish Balay }
1573a7e14dcfSSatish Balay 
15747fab98ebSJason Sarich /*@
157547450a7bSBarry Smith   TaoDefaultGMonitor - Default routine for monitoring progress of `TaoSolve()` with extra detail on the globalization method.
15768d5ead36SAlp Dener 
1577c3339decSBarry Smith   Collective
15788d5ead36SAlp Dener 
15798d5ead36SAlp Dener   Input Parameters:
158047450a7bSBarry Smith + tao - the `Tao` context
158147450a7bSBarry Smith - ctx - `PetscViewer` context or `NULL`
15828d5ead36SAlp Dener 
158347450a7bSBarry Smith   Options Database Key:
1584147403d9SBarry Smith . -tao_gmonitor - turn on monitoring with globalization information
15858d5ead36SAlp Dener 
15868d5ead36SAlp Dener   Level: advanced
15878d5ead36SAlp Dener 
158847450a7bSBarry Smith   Note:
158947450a7bSBarry Smith   This monitor prints the function value and gradient norm at each
159047450a7bSBarry Smith   iteration, as well as the step size and trust radius. Note that the
159147450a7bSBarry Smith   step size and trust radius may be the same for some algorithms.
159247450a7bSBarry Smith 
15931cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoDefaultSMonitor()`, `TaoSetMonitor()`
15948d5ead36SAlp Dener @*/
1595d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoDefaultGMonitor(Tao tao, void *ctx)
1596d71ae5a4SJacob Faibussowitsch {
15978d5ead36SAlp Dener   PetscInt    its, tabs;
15988d5ead36SAlp Dener   PetscReal   fct, gnorm, stp, tr;
15998d5ead36SAlp Dener   PetscViewer viewer = (PetscViewer)ctx;
16008d5ead36SAlp Dener 
16018d5ead36SAlp Dener   PetscFunctionBegin;
160294511df7SStefano Zampini   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
16038d5ead36SAlp Dener   PetscValidHeaderSpecific(viewer, PETSC_VIEWER_CLASSID, 2);
16048d5ead36SAlp Dener   its   = tao->niter;
16058d5ead36SAlp Dener   fct   = tao->fc;
16068d5ead36SAlp Dener   gnorm = tao->residual;
16078d5ead36SAlp Dener   stp   = tao->step;
16088d5ead36SAlp Dener   tr    = tao->trust;
16099566063dSJacob Faibussowitsch   PetscCall(PetscViewerASCIIGetTab(viewer, &tabs));
16109566063dSJacob Faibussowitsch   PetscCall(PetscViewerASCIISetTab(viewer, ((PetscObject)tao)->tablevel));
1611494bef23SAlp Dener   if (its == 0 && ((PetscObject)tao)->prefix && !tao->header_printed) {
16129566063dSJacob Faibussowitsch     PetscCall(PetscViewerASCIIPrintf(viewer, "  Iteration information for %s solve.\n", ((PetscObject)tao)->prefix));
1613494bef23SAlp Dener     tao->header_printed = PETSC_TRUE;
16148d5ead36SAlp Dener   }
161563a3b9bcSJacob Faibussowitsch   PetscCall(PetscViewerASCIIPrintf(viewer, "%3" PetscInt_FMT " TAO,", its));
16169566063dSJacob Faibussowitsch   PetscCall(PetscViewerASCIIPrintf(viewer, "  Function value: %g,", (double)fct));
16178d5ead36SAlp Dener   if (gnorm >= PETSC_INFINITY) {
16189566063dSJacob Faibussowitsch     PetscCall(PetscViewerASCIIPrintf(viewer, "  Residual: Inf,"));
16198d5ead36SAlp Dener   } else {
16209566063dSJacob Faibussowitsch     PetscCall(PetscViewerASCIIPrintf(viewer, "  Residual: %g,", (double)gnorm));
16218d5ead36SAlp Dener   }
16229566063dSJacob Faibussowitsch   PetscCall(PetscViewerASCIIPrintf(viewer, "  Step: %g,  Trust: %g\n", (double)stp, (double)tr));
16239566063dSJacob Faibussowitsch   PetscCall(PetscViewerASCIISetTab(viewer, tabs));
16243ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
16258d5ead36SAlp Dener }
16268d5ead36SAlp Dener 
16278d5ead36SAlp Dener /*@
162847450a7bSBarry Smith   TaoDefaultSMonitor - Default routine for monitoring progress of `TaoSolve()`
1629a7e14dcfSSatish Balay 
1630c3339decSBarry Smith   Collective
1631a7e14dcfSSatish Balay 
1632a7e14dcfSSatish Balay   Input Parameters:
163347450a7bSBarry Smith + tao - the `Tao` context
163447450a7bSBarry Smith - ctx - `PetscViewer` context of type `PETSCVIEWERASCII`
1635a7e14dcfSSatish Balay 
163647450a7bSBarry Smith   Options Database Key:
1637147403d9SBarry Smith . -tao_smonitor - turn on default short monitoring
1638a7e14dcfSSatish Balay 
1639a7e14dcfSSatish Balay   Level: advanced
1640a7e14dcfSSatish Balay 
164147450a7bSBarry Smith   Note:
164247450a7bSBarry Smith   Same as `TaoMonitorDefault()` except
164347450a7bSBarry Smith   it prints fewer digits of the residual as the residual gets smaller.
164447450a7bSBarry Smith   This is because the later digits are meaningless and are often
164547450a7bSBarry Smith   different on different machines; by using this routine different
164647450a7bSBarry Smith   machines will usually generate the same output.
164747450a7bSBarry Smith 
16481cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoMonitorDefault()`, `TaoSetMonitor()`
1649a7e14dcfSSatish Balay @*/
1650d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoDefaultSMonitor(Tao tao, void *ctx)
1651d71ae5a4SJacob Faibussowitsch {
16522c9b8b83SAlp Dener   PetscInt    its, tabs;
1653a7e14dcfSSatish Balay   PetscReal   fct, gnorm;
16544d4332d5SBarry Smith   PetscViewer viewer = (PetscViewer)ctx;
165547a47007SBarry Smith 
1656a7e14dcfSSatish Balay   PetscFunctionBegin;
165794511df7SStefano Zampini   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
16584d4332d5SBarry Smith   PetscValidHeaderSpecific(viewer, PETSC_VIEWER_CLASSID, 2);
1659a7e14dcfSSatish Balay   its   = tao->niter;
1660a7e14dcfSSatish Balay   fct   = tao->fc;
1661a7e14dcfSSatish Balay   gnorm = tao->residual;
16629566063dSJacob Faibussowitsch   PetscCall(PetscViewerASCIIGetTab(viewer, &tabs));
16639566063dSJacob Faibussowitsch   PetscCall(PetscViewerASCIISetTab(viewer, ((PetscObject)tao)->tablevel));
166463a3b9bcSJacob Faibussowitsch   PetscCall(PetscViewerASCIIPrintf(viewer, "iter = %3" PetscInt_FMT ",", its));
16659566063dSJacob Faibussowitsch   PetscCall(PetscViewerASCIIPrintf(viewer, " Function value %g,", (double)fct));
1666d393f493SBarry Smith   if (gnorm >= PETSC_INFINITY) {
16679566063dSJacob Faibussowitsch     PetscCall(PetscViewerASCIIPrintf(viewer, " Residual: Inf \n"));
166808d19d1fSJason Sarich   } else if (gnorm > 1.e-6) {
16699566063dSJacob Faibussowitsch     PetscCall(PetscViewerASCIIPrintf(viewer, " Residual: %g \n", (double)gnorm));
1670a7e14dcfSSatish Balay   } else if (gnorm > 1.e-11) {
16719566063dSJacob Faibussowitsch     PetscCall(PetscViewerASCIIPrintf(viewer, " Residual: < 1.0e-6 \n"));
1672a7e14dcfSSatish Balay   } else {
16739566063dSJacob Faibussowitsch     PetscCall(PetscViewerASCIIPrintf(viewer, " Residual: < 1.0e-11 \n"));
1674a7e14dcfSSatish Balay   }
16759566063dSJacob Faibussowitsch   PetscCall(PetscViewerASCIISetTab(viewer, tabs));
16763ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
1677a7e14dcfSSatish Balay }
1678a7e14dcfSSatish Balay 
16797fab98ebSJason Sarich /*@
168065ba42b6SBarry Smith   TaoDefaultCMonitor - same as `TaoMonitorDefault()` except
168147450a7bSBarry Smith   it prints the norm of the constraint function.
1682a7e14dcfSSatish Balay 
1683c3339decSBarry Smith   Collective
1684a7e14dcfSSatish Balay 
1685a7e14dcfSSatish Balay   Input Parameters:
168647450a7bSBarry Smith + tao - the `Tao` context
168747450a7bSBarry Smith - ctx - `PetscViewer` context or `NULL`
1688a7e14dcfSSatish Balay 
168947450a7bSBarry Smith   Options Database Key:
1690147403d9SBarry Smith . -tao_cmonitor - monitor the constraints
1691a7e14dcfSSatish Balay 
1692a7e14dcfSSatish Balay   Level: advanced
1693a7e14dcfSSatish Balay 
16941cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoMonitorDefault()`, `TaoSetMonitor()`
1695a7e14dcfSSatish Balay @*/
1696d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoDefaultCMonitor(Tao tao, void *ctx)
1697d71ae5a4SJacob Faibussowitsch {
16982c9b8b83SAlp Dener   PetscInt    its, tabs;
1699a7e14dcfSSatish Balay   PetscReal   fct, gnorm;
1700d393f493SBarry Smith   PetscViewer viewer = (PetscViewer)ctx;
1701a7e14dcfSSatish Balay 
1702a7e14dcfSSatish Balay   PetscFunctionBegin;
170394511df7SStefano Zampini   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
1704d393f493SBarry Smith   PetscValidHeaderSpecific(viewer, PETSC_VIEWER_CLASSID, 2);
1705a7e14dcfSSatish Balay   its   = tao->niter;
1706a7e14dcfSSatish Balay   fct   = tao->fc;
1707a7e14dcfSSatish Balay   gnorm = tao->residual;
17089566063dSJacob Faibussowitsch   PetscCall(PetscViewerASCIIGetTab(viewer, &tabs));
17099566063dSJacob Faibussowitsch   PetscCall(PetscViewerASCIISetTab(viewer, ((PetscObject)tao)->tablevel));
171063a3b9bcSJacob Faibussowitsch   PetscCall(PetscViewerASCIIPrintf(viewer, "iter = %" PetscInt_FMT ",", its));
17119566063dSJacob Faibussowitsch   PetscCall(PetscViewerASCIIPrintf(viewer, " Function value: %g,", (double)fct));
17129566063dSJacob Faibussowitsch   PetscCall(PetscViewerASCIIPrintf(viewer, "  Residual: %g ", (double)gnorm));
17139566063dSJacob Faibussowitsch   PetscCall(PetscViewerASCIIPrintf(viewer, "  Constraint: %g \n", (double)tao->cnorm));
17149566063dSJacob Faibussowitsch   PetscCall(PetscViewerASCIISetTab(viewer, tabs));
17153ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
1716a7e14dcfSSatish Balay }
1717a7e14dcfSSatish Balay 
1718a7e14dcfSSatish Balay /*@C
171947450a7bSBarry Smith   TaoSolutionMonitor - Views the solution at each iteration of `TaoSolve()`
1720a7e14dcfSSatish Balay 
1721c3339decSBarry Smith   Collective
1722a7e14dcfSSatish Balay 
1723a7e14dcfSSatish Balay   Input Parameters:
172447450a7bSBarry Smith + tao - the `Tao` context
172547450a7bSBarry Smith - ctx - `PetscViewer` context or `NULL`
1726a7e14dcfSSatish Balay 
172747450a7bSBarry Smith   Options Database Key:
1728147403d9SBarry Smith . -tao_view_solution - view the solution
1729a7e14dcfSSatish Balay 
1730a7e14dcfSSatish Balay   Level: advanced
1731a7e14dcfSSatish Balay 
17321cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoDefaultSMonitor()`, `TaoSetMonitor()`
1733a7e14dcfSSatish Balay @*/
1734d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoSolutionMonitor(Tao tao, void *ctx)
1735d71ae5a4SJacob Faibussowitsch {
1736feb237baSPierre Jolivet   PetscViewer viewer = (PetscViewer)ctx;
173747a47007SBarry Smith 
1738a7e14dcfSSatish Balay   PetscFunctionBegin;
173994511df7SStefano Zampini   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
1740d393f493SBarry Smith   PetscValidHeaderSpecific(viewer, PETSC_VIEWER_CLASSID, 2);
17419566063dSJacob Faibussowitsch   PetscCall(VecView(tao->solution, viewer));
17423ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
1743a7e14dcfSSatish Balay }
1744a7e14dcfSSatish Balay 
1745a7e14dcfSSatish Balay /*@C
174647450a7bSBarry Smith   TaoGradientMonitor - Views the gradient at each iteration of `TaoSolve()`
1747a7e14dcfSSatish Balay 
1748c3339decSBarry Smith   Collective
1749a7e14dcfSSatish Balay 
1750a7e14dcfSSatish Balay   Input Parameters:
175147450a7bSBarry Smith + tao - the `Tao` context
175247450a7bSBarry Smith - ctx - `PetscViewer` context or `NULL`
1753a7e14dcfSSatish Balay 
175447450a7bSBarry Smith   Options Database Key:
1755147403d9SBarry Smith . -tao_view_gradient - view the gradient at each iteration
1756a7e14dcfSSatish Balay 
1757a7e14dcfSSatish Balay   Level: advanced
1758a7e14dcfSSatish Balay 
17591cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoDefaultSMonitor()`, `TaoSetMonitor()`
1760a7e14dcfSSatish Balay @*/
1761d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoGradientMonitor(Tao tao, void *ctx)
1762d71ae5a4SJacob Faibussowitsch {
1763d393f493SBarry Smith   PetscViewer viewer = (PetscViewer)ctx;
176447a47007SBarry Smith 
1765a7e14dcfSSatish Balay   PetscFunctionBegin;
176694511df7SStefano Zampini   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
1767d393f493SBarry Smith   PetscValidHeaderSpecific(viewer, PETSC_VIEWER_CLASSID, 2);
17689566063dSJacob Faibussowitsch   PetscCall(VecView(tao->gradient, viewer));
17693ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
1770a7e14dcfSSatish Balay }
1771a7e14dcfSSatish Balay 
1772a7e14dcfSSatish Balay /*@C
177347450a7bSBarry Smith   TaoStepDirectionMonitor - Views the step-direction at each iteration of `TaoSolve()`
1774a7e14dcfSSatish Balay 
1775c3339decSBarry Smith   Collective
1776a7e14dcfSSatish Balay 
1777a7e14dcfSSatish Balay   Input Parameters:
177847450a7bSBarry Smith + tao - the `Tao` context
177947450a7bSBarry Smith - ctx - `PetscViewer` context or `NULL`
1780a7e14dcfSSatish Balay 
178147450a7bSBarry Smith   Options Database Key:
178247450a7bSBarry Smith . -tao_view_stepdirection - view the step direction vector at each iteration
1783a7e14dcfSSatish Balay 
1784a7e14dcfSSatish Balay   Level: advanced
1785a7e14dcfSSatish Balay 
17861cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoDefaultSMonitor()`, `TaoSetMonitor()`
1787a7e14dcfSSatish Balay @*/
1788d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoStepDirectionMonitor(Tao tao, void *ctx)
1789d71ae5a4SJacob Faibussowitsch {
1790d393f493SBarry Smith   PetscViewer viewer = (PetscViewer)ctx;
1791d393f493SBarry Smith 
1792a7e14dcfSSatish Balay   PetscFunctionBegin;
179394511df7SStefano Zampini   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
1794d393f493SBarry Smith   PetscValidHeaderSpecific(viewer, PETSC_VIEWER_CLASSID, 2);
17959566063dSJacob Faibussowitsch   PetscCall(VecView(tao->stepdirection, viewer));
17963ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
1797a7e14dcfSSatish Balay }
1798a7e14dcfSSatish Balay 
1799a7e14dcfSSatish Balay /*@C
180047450a7bSBarry Smith   TaoDrawSolutionMonitor - Plots the solution at each iteration of `TaoSolve()`
1801a7e14dcfSSatish Balay 
1802c3339decSBarry Smith   Collective
1803a7e14dcfSSatish Balay 
1804a7e14dcfSSatish Balay   Input Parameters:
180547450a7bSBarry Smith + tao - the `Tao` context
180665ba42b6SBarry Smith - ctx - `TaoMonitorDraw` context
1807a7e14dcfSSatish Balay 
180847450a7bSBarry Smith   Options Database Key:
1809147403d9SBarry Smith . -tao_draw_solution - draw the solution at each iteration
1810a7e14dcfSSatish Balay 
1811a7e14dcfSSatish Balay   Level: advanced
1812a7e14dcfSSatish Balay 
18131cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoSolutionMonitor()`, `TaoSetMonitor()`, `TaoDrawGradientMonitor`, `TaoMonitorDraw`
1814a7e14dcfSSatish Balay @*/
1815d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoDrawSolutionMonitor(Tao tao, void *ctx)
1816d71ae5a4SJacob Faibussowitsch {
1817e882e171SHong Zhang   TaoMonitorDrawCtx ictx = (TaoMonitorDrawCtx)ctx;
181847a47007SBarry Smith 
1819a7e14dcfSSatish Balay   PetscFunctionBegin;
182094511df7SStefano Zampini   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
18213ba16761SJacob Faibussowitsch   if (!(((ictx->howoften > 0) && (!(tao->niter % ictx->howoften))) || ((ictx->howoften == -1) && tao->reason))) PetscFunctionReturn(PETSC_SUCCESS);
18229566063dSJacob Faibussowitsch   PetscCall(VecView(tao->solution, ictx->viewer));
18233ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
1824a7e14dcfSSatish Balay }
1825a7e14dcfSSatish Balay 
1826a7e14dcfSSatish Balay /*@C
182747450a7bSBarry Smith   TaoDrawGradientMonitor - Plots the gradient at each iteration of `TaoSolve()`
1828a7e14dcfSSatish Balay 
1829c3339decSBarry Smith   Collective
1830a7e14dcfSSatish Balay 
1831a7e14dcfSSatish Balay   Input Parameters:
183247450a7bSBarry Smith + tao - the `Tao` context
183365ba42b6SBarry Smith - ctx - `PetscViewer` context
1834a7e14dcfSSatish Balay 
183547450a7bSBarry Smith   Options Database Key:
1836147403d9SBarry Smith . -tao_draw_gradient - draw the gradient at each iteration
1837a7e14dcfSSatish Balay 
1838a7e14dcfSSatish Balay   Level: advanced
1839a7e14dcfSSatish Balay 
18401cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoGradientMonitor()`, `TaoSetMonitor()`, `TaoDrawSolutionMonitor`
1841a7e14dcfSSatish Balay @*/
1842d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoDrawGradientMonitor(Tao tao, void *ctx)
1843d71ae5a4SJacob Faibussowitsch {
1844e882e171SHong Zhang   TaoMonitorDrawCtx ictx = (TaoMonitorDrawCtx)ctx;
184547a47007SBarry Smith 
1846a7e14dcfSSatish Balay   PetscFunctionBegin;
184794511df7SStefano Zampini   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
18483ba16761SJacob Faibussowitsch   if (!(((ictx->howoften > 0) && (!(tao->niter % ictx->howoften))) || ((ictx->howoften == -1) && tao->reason))) PetscFunctionReturn(PETSC_SUCCESS);
18499566063dSJacob Faibussowitsch   PetscCall(VecView(tao->gradient, ictx->viewer));
18503ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
1851a7e14dcfSSatish Balay }
1852a7e14dcfSSatish Balay 
1853a7e14dcfSSatish Balay /*@C
185447450a7bSBarry Smith   TaoDrawStepMonitor - Plots the step direction at each iteration of `TaoSolve()`
1855a7e14dcfSSatish Balay 
1856c3339decSBarry Smith   Collective
1857a7e14dcfSSatish Balay 
1858a7e14dcfSSatish Balay   Input Parameters:
185947450a7bSBarry Smith + tao - the `Tao` context
186047450a7bSBarry Smith - ctx - the `PetscViewer` context
1861a7e14dcfSSatish Balay 
186247450a7bSBarry Smith   Options Database Key:
1863147403d9SBarry Smith . -tao_draw_step - draw the step direction at each iteration
1864a7e14dcfSSatish Balay 
1865a7e14dcfSSatish Balay   Level: advanced
1866a7e14dcfSSatish Balay 
18671cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoSetMonitor()`, `TaoDrawSolutionMonitor`
1868a7e14dcfSSatish Balay @*/
1869d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoDrawStepMonitor(Tao tao, void *ctx)
1870d71ae5a4SJacob Faibussowitsch {
187194511df7SStefano Zampini   PetscViewer viewer = (PetscViewer)ctx;
187247a47007SBarry Smith 
1873a7e14dcfSSatish Balay   PetscFunctionBegin;
187494511df7SStefano Zampini   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
187594511df7SStefano Zampini   PetscValidHeaderSpecific(viewer, PETSC_VIEWER_CLASSID, 2);
18769566063dSJacob Faibussowitsch   PetscCall(VecView(tao->stepdirection, viewer));
18773ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
1878a7e14dcfSSatish Balay }
1879a7e14dcfSSatish Balay 
1880a7e14dcfSSatish Balay /*@C
188147450a7bSBarry Smith   TaoResidualMonitor - Views the least-squares residual at each iteration of `TaoSolve()`
1882a7e14dcfSSatish Balay 
1883c3339decSBarry Smith   Collective
1884a7e14dcfSSatish Balay 
1885a7e14dcfSSatish Balay   Input Parameters:
188647450a7bSBarry Smith + tao - the `Tao` context
188747450a7bSBarry Smith - ctx - the `PetscViewer` context or `NULL`
1888a7e14dcfSSatish Balay 
188947450a7bSBarry Smith   Options Database Key:
189047450a7bSBarry Smith . -tao_view_ls_residual - view the residual at each iteration
1891a7e14dcfSSatish Balay 
1892a7e14dcfSSatish Balay   Level: advanced
1893a7e14dcfSSatish Balay 
18941cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoDefaultSMonitor()`, `TaoSetMonitor()`
1895a7e14dcfSSatish Balay @*/
1896d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoResidualMonitor(Tao tao, void *ctx)
1897d71ae5a4SJacob Faibussowitsch {
1898d393f493SBarry Smith   PetscViewer viewer = (PetscViewer)ctx;
189947a47007SBarry Smith 
1900a7e14dcfSSatish Balay   PetscFunctionBegin;
190194511df7SStefano Zampini   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
1902d393f493SBarry Smith   PetscValidHeaderSpecific(viewer, PETSC_VIEWER_CLASSID, 2);
19039566063dSJacob Faibussowitsch   PetscCall(VecView(tao->ls_res, viewer));
19043ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
1905a7e14dcfSSatish Balay }
1906a7e14dcfSSatish Balay 
19077fab98ebSJason Sarich /*@
1908a7e14dcfSSatish Balay   TaoDefaultConvergenceTest - Determines whether the solver should continue iterating
1909a7e14dcfSSatish Balay   or terminate.
1910a7e14dcfSSatish Balay 
1911c3339decSBarry Smith   Collective
1912a7e14dcfSSatish Balay 
1913a7e14dcfSSatish Balay   Input Parameters:
191447450a7bSBarry Smith + tao   - the `Tao` context
1915a7e14dcfSSatish Balay - dummy - unused dummy context
1916a7e14dcfSSatish Balay 
191747450a7bSBarry Smith   Level: developer
191847450a7bSBarry Smith 
1919a7e14dcfSSatish Balay   Notes:
1920a7e14dcfSSatish Balay   This routine checks the residual in the optimality conditions, the
1921a7e14dcfSSatish Balay   relative residual in the optimity conditions, the number of function
1922a7e14dcfSSatish Balay   evaluations, and the function value to test convergence.  Some
1923a7e14dcfSSatish Balay   solvers may use different convergence routines.
1924a7e14dcfSSatish Balay 
19251cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoSetTolerances()`, `TaoGetConvergedReason()`, `TaoSetConvergedReason()`
1926a7e14dcfSSatish Balay @*/
1927d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoDefaultConvergenceTest(Tao tao, void *dummy)
1928d71ae5a4SJacob Faibussowitsch {
1929a7e14dcfSSatish Balay   PetscInt           niter = tao->niter, nfuncs = PetscMax(tao->nfuncs, tao->nfuncgrads);
1930a7e14dcfSSatish Balay   PetscInt           max_funcs = tao->max_funcs;
1931a7e14dcfSSatish Balay   PetscReal          gnorm = tao->residual, gnorm0 = tao->gnorm0;
1932a7e14dcfSSatish Balay   PetscReal          f = tao->fc, steptol = tao->steptol, trradius = tao->step;
1933a7e14dcfSSatish Balay   PetscReal          gatol = tao->gatol, grtol = tao->grtol, gttol = tao->gttol;
1934e52336cbSBarry Smith   PetscReal          catol = tao->catol, crtol = tao->crtol;
1935e52336cbSBarry Smith   PetscReal          fmin = tao->fmin, cnorm = tao->cnorm;
1936e4cb33bbSBarry Smith   TaoConvergedReason reason = tao->reason;
1937a7e14dcfSSatish Balay 
1938a7e14dcfSSatish Balay   PetscFunctionBegin;
1939441846f8SBarry Smith   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
19403ba16761SJacob Faibussowitsch   if (reason != TAO_CONTINUE_ITERATING) PetscFunctionReturn(PETSC_SUCCESS);
1941a7e14dcfSSatish Balay 
1942a7e14dcfSSatish Balay   if (PetscIsInfOrNanReal(f)) {
19439566063dSJacob Faibussowitsch     PetscCall(PetscInfo(tao, "Failed to converged, function value is Inf or NaN\n"));
1944a7e14dcfSSatish Balay     reason = TAO_DIVERGED_NAN;
1945a7e14dcfSSatish Balay   } else if (f <= fmin && cnorm <= catol) {
19469566063dSJacob Faibussowitsch     PetscCall(PetscInfo(tao, "Converged due to function value %g < minimum function value %g\n", (double)f, (double)fmin));
1947a7e14dcfSSatish Balay     reason = TAO_CONVERGED_MINF;
1948a7e14dcfSSatish Balay   } else if (gnorm <= gatol && cnorm <= catol) {
19499566063dSJacob Faibussowitsch     PetscCall(PetscInfo(tao, "Converged due to residual norm ||g(X)||=%g < %g\n", (double)gnorm, (double)gatol));
1950a7e14dcfSSatish Balay     reason = TAO_CONVERGED_GATOL;
1951a7e14dcfSSatish Balay   } else if (f != 0 && PetscAbsReal(gnorm / f) <= grtol && cnorm <= crtol) {
19529566063dSJacob Faibussowitsch     PetscCall(PetscInfo(tao, "Converged due to residual ||g(X)||/|f(X)| =%g < %g\n", (double)(gnorm / f), (double)grtol));
1953a7e14dcfSSatish Balay     reason = TAO_CONVERGED_GRTOL;
1954e1d16bb9SBarry Smith   } else if (gnorm0 != 0 && ((gttol == 0 && gnorm == 0) || gnorm / gnorm0 < gttol) && cnorm <= crtol) {
19559566063dSJacob Faibussowitsch     PetscCall(PetscInfo(tao, "Converged due to relative residual norm ||g(X)||/||g(X0)|| = %g < %g\n", (double)(gnorm / gnorm0), (double)gttol));
1956a7e14dcfSSatish Balay     reason = TAO_CONVERGED_GTTOL;
195794511df7SStefano Zampini   } else if (max_funcs >= 0 && nfuncs > max_funcs) {
19589566063dSJacob Faibussowitsch     PetscCall(PetscInfo(tao, "Exceeded maximum number of function evaluations: %" PetscInt_FMT " > %" PetscInt_FMT "\n", nfuncs, max_funcs));
1959a7e14dcfSSatish Balay     reason = TAO_DIVERGED_MAXFCN;
1960a7e14dcfSSatish Balay   } else if (tao->lsflag != 0) {
19619566063dSJacob Faibussowitsch     PetscCall(PetscInfo(tao, "Tao Line Search failure.\n"));
1962a7e14dcfSSatish Balay     reason = TAO_DIVERGED_LS_FAILURE;
1963a7e14dcfSSatish Balay   } else if (trradius < steptol && niter > 0) {
19649566063dSJacob Faibussowitsch     PetscCall(PetscInfo(tao, "Trust region/step size too small: %g < %g\n", (double)trradius, (double)steptol));
1965a7e14dcfSSatish Balay     reason = TAO_CONVERGED_STEPTOL;
1966e031d6f5SAlp Dener   } else if (niter >= tao->max_it) {
196763a3b9bcSJacob Faibussowitsch     PetscCall(PetscInfo(tao, "Exceeded maximum number of iterations: %" PetscInt_FMT " > %" PetscInt_FMT "\n", niter, tao->max_it));
1968a7e14dcfSSatish Balay     reason = TAO_DIVERGED_MAXITS;
1969a7e14dcfSSatish Balay   } else {
1970a7e14dcfSSatish Balay     reason = TAO_CONTINUE_ITERATING;
1971a7e14dcfSSatish Balay   }
1972a7e14dcfSSatish Balay   tao->reason = reason;
19733ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
1974a7e14dcfSSatish Balay }
1975a7e14dcfSSatish Balay 
1976a7e14dcfSSatish Balay /*@C
1977a7e14dcfSSatish Balay   TaoSetOptionsPrefix - Sets the prefix used for searching for all
197865ba42b6SBarry Smith   Tao options in the database.
1979a7e14dcfSSatish Balay 
1980c3339decSBarry Smith   Logically Collective
1981a7e14dcfSSatish Balay 
1982a7e14dcfSSatish Balay   Input Parameters:
198347450a7bSBarry Smith + tao - the `Tao` context
1984e056e8ceSJacob Faibussowitsch - p   - the prefix string to prepend to all Tao option requests
1985a7e14dcfSSatish Balay 
19862fe279fdSBarry Smith   Level: advanced
19872fe279fdSBarry Smith 
1988a7e14dcfSSatish Balay   Notes:
1989a7e14dcfSSatish Balay   A hyphen (-) must NOT be given at the beginning of the prefix name.
1990a7e14dcfSSatish Balay   The first character of all runtime options is AUTOMATICALLY the hyphen.
1991a7e14dcfSSatish Balay 
1992a7e14dcfSSatish Balay   For example, to distinguish between the runtime options for two
199365ba42b6SBarry Smith   different Tao solvers, one could call
1994a7e14dcfSSatish Balay .vb
1995a7e14dcfSSatish Balay       TaoSetOptionsPrefix(tao1,"sys1_")
1996a7e14dcfSSatish Balay       TaoSetOptionsPrefix(tao2,"sys2_")
1997a7e14dcfSSatish Balay .ve
1998a7e14dcfSSatish Balay 
1999a7e14dcfSSatish Balay   This would enable use of different options for each system, such as
2000a7e14dcfSSatish Balay .vb
20019fa2b5dcSStefano Zampini       -sys1_tao_method blmvm -sys1_tao_grtol 1.e-3
20029fa2b5dcSStefano Zampini       -sys2_tao_method lmvm  -sys2_tao_grtol 1.e-4
2003a7e14dcfSSatish Balay .ve
2004a7e14dcfSSatish Balay 
20051cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoSetFromOptions()`, `TaoAppendOptionsPrefix()`, `TaoGetOptionsPrefix()`
2006a7e14dcfSSatish Balay @*/
2007d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoSetOptionsPrefix(Tao tao, const char p[])
2008d71ae5a4SJacob Faibussowitsch {
2009a7e14dcfSSatish Balay   PetscFunctionBegin;
201094511df7SStefano Zampini   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
20119566063dSJacob Faibussowitsch   PetscCall(PetscObjectSetOptionsPrefix((PetscObject)tao, p));
20121baa6e33SBarry Smith   if (tao->linesearch) PetscCall(TaoLineSearchSetOptionsPrefix(tao->linesearch, p));
20131baa6e33SBarry Smith   if (tao->ksp) PetscCall(KSPSetOptionsPrefix(tao->ksp, p));
20143ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
2015a7e14dcfSSatish Balay }
2016a7e14dcfSSatish Balay 
2017a7e14dcfSSatish Balay /*@C
201847450a7bSBarry Smith   TaoAppendOptionsPrefix - Appends to the prefix used for searching for all Tao options in the database.
2019a7e14dcfSSatish Balay 
2020c3339decSBarry Smith   Logically Collective
2021a7e14dcfSSatish Balay 
2022a7e14dcfSSatish Balay   Input Parameters:
202347450a7bSBarry Smith + tao - the `Tao` solver context
2024e056e8ceSJacob Faibussowitsch - p   - the prefix string to prepend to all `Tao` option requests
2025a7e14dcfSSatish Balay 
20262fe279fdSBarry Smith   Level: advanced
20272fe279fdSBarry Smith 
202865ba42b6SBarry Smith   Note:
2029a7e14dcfSSatish Balay   A hyphen (-) must NOT be given at the beginning of the prefix name.
203065ba42b6SBarry Smith   The first character of all runtime options is automatically the hyphen.
2031a7e14dcfSSatish Balay 
20321cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoSetFromOptions()`, `TaoSetOptionsPrefix()`, `TaoGetOptionsPrefix()`
2033a7e14dcfSSatish Balay @*/
2034d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoAppendOptionsPrefix(Tao tao, const char p[])
2035d71ae5a4SJacob Faibussowitsch {
2036a7e14dcfSSatish Balay   PetscFunctionBegin;
203794511df7SStefano Zampini   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
20389566063dSJacob Faibussowitsch   PetscCall(PetscObjectAppendOptionsPrefix((PetscObject)tao, p));
20391baa6e33SBarry Smith   if (tao->linesearch) PetscCall(PetscObjectAppendOptionsPrefix((PetscObject)tao->linesearch, p));
20401baa6e33SBarry Smith   if (tao->ksp) PetscCall(KSPAppendOptionsPrefix(tao->ksp, p));
20413ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
2042a7e14dcfSSatish Balay }
2043a7e14dcfSSatish Balay 
2044a7e14dcfSSatish Balay /*@C
2045a7e14dcfSSatish Balay   TaoGetOptionsPrefix - Gets the prefix used for searching for all
204665ba42b6SBarry Smith   Tao options in the database
2047a7e14dcfSSatish Balay 
2048a7e14dcfSSatish Balay   Not Collective
2049a7e14dcfSSatish Balay 
205047450a7bSBarry Smith   Input Parameter:
205147450a7bSBarry Smith . tao - the `Tao` context
2052a7e14dcfSSatish Balay 
205347450a7bSBarry Smith   Output Parameter:
2054e056e8ceSJacob Faibussowitsch . p - pointer to the prefix string used is returned
2055a7e14dcfSSatish Balay 
2056e056e8ceSJacob Faibussowitsch   Fortran Notes:
205747450a7bSBarry Smith   Pass in a string 'prefix' of sufficient length to hold the prefix.
2058a7e14dcfSSatish Balay 
2059a7e14dcfSSatish Balay   Level: advanced
2060a7e14dcfSSatish Balay 
20611cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoSetFromOptions()`, `TaoSetOptionsPrefix()`, `TaoAppendOptionsPrefix()`
2062a7e14dcfSSatish Balay @*/
2063d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoGetOptionsPrefix(Tao tao, const char *p[])
2064d71ae5a4SJacob Faibussowitsch {
206594511df7SStefano Zampini   PetscFunctionBegin;
206694511df7SStefano Zampini   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
20679566063dSJacob Faibussowitsch   PetscCall(PetscObjectGetOptionsPrefix((PetscObject)tao, p));
20683ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
2069a7e14dcfSSatish Balay }
2070a7e14dcfSSatish Balay 
2071a7e14dcfSSatish Balay /*@C
207247450a7bSBarry Smith   TaoSetType - Sets the `TaoType` for the minimization solver.
2073a7e14dcfSSatish Balay 
2074c3339decSBarry Smith   Collective
2075a7e14dcfSSatish Balay 
2076a7e14dcfSSatish Balay   Input Parameters:
2077e056e8ceSJacob Faibussowitsch + tao  - the `Tao` solver context
2078a7e14dcfSSatish Balay - type - a known method
2079a7e14dcfSSatish Balay 
2080a7e14dcfSSatish Balay   Options Database Key:
208158417fe7SBarry Smith . -tao_type <type> - Sets the method; use -help for a list
208258417fe7SBarry Smith    of available methods (for instance, "-tao_type lmvm" or "-tao_type tron")
2083a7e14dcfSSatish Balay 
2084a7e14dcfSSatish Balay   Level: intermediate
2085a7e14dcfSSatish Balay 
20861cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoCreate()`, `TaoGetType()`, `TaoType`
2087a7e14dcfSSatish Balay @*/
2088d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoSetType(Tao tao, TaoType type)
2089d71ae5a4SJacob Faibussowitsch {
2090441846f8SBarry Smith   PetscErrorCode (*create_xxx)(Tao);
2091a7e14dcfSSatish Balay   PetscBool issame;
2092a7e14dcfSSatish Balay 
2093a7e14dcfSSatish Balay   PetscFunctionBegin;
2094441846f8SBarry Smith   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
2095a7e14dcfSSatish Balay 
20969566063dSJacob Faibussowitsch   PetscCall(PetscObjectTypeCompare((PetscObject)tao, type, &issame));
20973ba16761SJacob Faibussowitsch   if (issame) PetscFunctionReturn(PETSC_SUCCESS);
2098a7e14dcfSSatish Balay 
20999566063dSJacob Faibussowitsch   PetscCall(PetscFunctionListFind(TaoList, type, (void (**)(void)) & create_xxx));
21003c859ba3SBarry Smith   PetscCheck(create_xxx, PetscObjectComm((PetscObject)tao), PETSC_ERR_ARG_UNKNOWN_TYPE, "Unable to find requested Tao type %s", type);
2101a7e14dcfSSatish Balay 
2102a7e14dcfSSatish Balay   /* Destroy the existing solver information */
2103dbbe0bcdSBarry Smith   PetscTryTypeMethod(tao, destroy);
21041baa6e33SBarry Smith   PetscCall(KSPDestroy(&tao->ksp));
21059566063dSJacob Faibussowitsch   PetscCall(TaoLineSearchDestroy(&tao->linesearch));
210683c8fe1dSLisandro Dalcin   tao->ops->setup          = NULL;
210783c8fe1dSLisandro Dalcin   tao->ops->solve          = NULL;
210883c8fe1dSLisandro Dalcin   tao->ops->view           = NULL;
210983c8fe1dSLisandro Dalcin   tao->ops->setfromoptions = NULL;
211083c8fe1dSLisandro Dalcin   tao->ops->destroy        = NULL;
2111a7e14dcfSSatish Balay 
2112a7e14dcfSSatish Balay   tao->setupcalled = PETSC_FALSE;
2113a7e14dcfSSatish Balay 
21149566063dSJacob Faibussowitsch   PetscCall((*create_xxx)(tao));
21159566063dSJacob Faibussowitsch   PetscCall(PetscObjectChangeTypeName((PetscObject)tao, type));
21163ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
2117a7e14dcfSSatish Balay }
211847a47007SBarry Smith 
211967be906fSBarry Smith /*@C
212047450a7bSBarry Smith   TaoRegister - Adds a method to the Tao package for minimization.
2121a7e14dcfSSatish Balay 
212220f4b53cSBarry Smith   Not Collective
2123a7e14dcfSSatish Balay 
2124a7e14dcfSSatish Balay   Input Parameters:
2125a7e14dcfSSatish Balay + sname - name of a new user-defined solver
2126a7e14dcfSSatish Balay - func  - routine to Create method context
2127a7e14dcfSSatish Balay 
2128e056e8ceSJacob Faibussowitsch   Example Usage:
2129a7e14dcfSSatish Balay .vb
2130441846f8SBarry Smith    TaoRegister("my_solver", MySolverCreate);
2131a7e14dcfSSatish Balay .ve
2132a7e14dcfSSatish Balay 
2133a7e14dcfSSatish Balay   Then, your solver can be chosen with the procedural interface via
2134a7e14dcfSSatish Balay $     TaoSetType(tao, "my_solver")
2135a7e14dcfSSatish Balay   or at runtime via the option
213658417fe7SBarry Smith $     -tao_type my_solver
2137a7e14dcfSSatish Balay 
2138a7e14dcfSSatish Balay   Level: advanced
2139a7e14dcfSSatish Balay 
214067be906fSBarry Smith   Note:
214167be906fSBarry Smith   `TaoRegister()` may be called multiple times to add several user-defined solvers.
214267be906fSBarry Smith 
21431cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoSetType()`, `TaoRegisterAll()`, `TaoRegisterDestroy()`
214467be906fSBarry Smith @*/
2145d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoRegister(const char sname[], PetscErrorCode (*func)(Tao))
2146d71ae5a4SJacob Faibussowitsch {
2147a7e14dcfSSatish Balay   PetscFunctionBegin;
21489566063dSJacob Faibussowitsch   PetscCall(TaoInitializePackage());
21499566063dSJacob Faibussowitsch   PetscCall(PetscFunctionListAdd(&TaoList, sname, (void (*)(void))func));
21503ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
2151a7e14dcfSSatish Balay }
2152a7e14dcfSSatish Balay 
2153a7e14dcfSSatish Balay /*@C
2154441846f8SBarry Smith   TaoRegisterDestroy - Frees the list of minimization solvers that were
215547450a7bSBarry Smith   registered by `TaoRegister()`.
2156a7e14dcfSSatish Balay 
2157a7e14dcfSSatish Balay   Not Collective
2158a7e14dcfSSatish Balay 
2159a7e14dcfSSatish Balay   Level: advanced
2160a7e14dcfSSatish Balay 
21611cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoRegisterAll()`, `TaoRegister()`
2162a7e14dcfSSatish Balay @*/
2163d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoRegisterDestroy(void)
2164d71ae5a4SJacob Faibussowitsch {
2165a7e14dcfSSatish Balay   PetscFunctionBegin;
21669566063dSJacob Faibussowitsch   PetscCall(PetscFunctionListDestroy(&TaoList));
2167441846f8SBarry Smith   TaoRegisterAllCalled = PETSC_FALSE;
21683ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
2169a7e14dcfSSatish Balay }
217047a47007SBarry Smith 
21718931d482SJason Sarich /*@
217247450a7bSBarry Smith   TaoGetIterationNumber - Gets the number of `TaoSolve()` iterations completed
21738931d482SJason Sarich   at this time.
21748931d482SJason Sarich 
21758931d482SJason Sarich   Not Collective
21768931d482SJason Sarich 
21778931d482SJason Sarich   Input Parameter:
217847450a7bSBarry Smith . tao - the `Tao` context
21798931d482SJason Sarich 
21808931d482SJason Sarich   Output Parameter:
21818931d482SJason Sarich . iter - iteration number
21828931d482SJason Sarich 
21838931d482SJason Sarich   Notes:
21848931d482SJason Sarich   For example, during the computation of iteration 2 this would return 1.
21858931d482SJason Sarich 
21868931d482SJason Sarich   Level: intermediate
21878931d482SJason Sarich 
21881cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoGetLinearSolveIterations()`, `TaoGetResidualNorm()`, `TaoGetObjective()`
21898931d482SJason Sarich @*/
2190d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoGetIterationNumber(Tao tao, PetscInt *iter)
2191d71ae5a4SJacob Faibussowitsch {
21928931d482SJason Sarich   PetscFunctionBegin;
21938931d482SJason Sarich   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
21944f572ea9SToby Isaac   PetscAssertPointer(iter, 2);
21958931d482SJason Sarich   *iter = tao->niter;
21963ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
21978931d482SJason Sarich }
21988931d482SJason Sarich 
21998931d482SJason Sarich /*@
220047450a7bSBarry Smith   TaoGetResidualNorm - Gets the current value of the norm of the residual (gradient)
220179f5d8caSBarry Smith   at this time.
220279f5d8caSBarry Smith 
220379f5d8caSBarry Smith   Not Collective
220479f5d8caSBarry Smith 
220579f5d8caSBarry Smith   Input Parameter:
220647450a7bSBarry Smith . tao - the `Tao` context
220779f5d8caSBarry Smith 
220879f5d8caSBarry Smith   Output Parameter:
220979f5d8caSBarry Smith . value - the current value
221079f5d8caSBarry Smith 
221179f5d8caSBarry Smith   Level: intermediate
221279f5d8caSBarry Smith 
2213e056e8ceSJacob Faibussowitsch   Developer Notes:
221467be906fSBarry Smith   This is the 2-norm of the residual, we cannot use `TaoGetGradientNorm()` because that has
221547450a7bSBarry Smith   a different meaning. For some reason `Tao` sometimes calls the gradient the residual.
221679f5d8caSBarry Smith 
22171cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoGetLinearSolveIterations()`, `TaoGetIterationNumber()`, `TaoGetObjective()`
221879f5d8caSBarry Smith @*/
2219d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoGetResidualNorm(Tao tao, PetscReal *value)
2220d71ae5a4SJacob Faibussowitsch {
222179f5d8caSBarry Smith   PetscFunctionBegin;
222279f5d8caSBarry Smith   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
22234f572ea9SToby Isaac   PetscAssertPointer(value, 2);
222479f5d8caSBarry Smith   *value = tao->residual;
22253ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
222679f5d8caSBarry Smith }
222779f5d8caSBarry Smith 
222879f5d8caSBarry Smith /*@
22298931d482SJason Sarich   TaoSetIterationNumber - Sets the current iteration number.
22308931d482SJason Sarich 
2231c3339decSBarry Smith   Logically Collective
22328931d482SJason Sarich 
2233d8d19677SJose E. Roman   Input Parameters:
223447450a7bSBarry Smith + tao  - the `Tao` context
2235a2b725a8SWilliam Gropp - iter - iteration number
22368931d482SJason Sarich 
22378931d482SJason Sarich   Level: developer
22388931d482SJason Sarich 
22391cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoGetLinearSolveIterations()`
22408931d482SJason Sarich @*/
2241d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoSetIterationNumber(Tao tao, PetscInt iter)
2242d71ae5a4SJacob Faibussowitsch {
22438931d482SJason Sarich   PetscFunctionBegin;
22448931d482SJason Sarich   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
224594511df7SStefano Zampini   PetscValidLogicalCollectiveInt(tao, iter, 2);
22469566063dSJacob Faibussowitsch   PetscCall(PetscObjectSAWsTakeAccess((PetscObject)tao));
22478931d482SJason Sarich   tao->niter = iter;
22489566063dSJacob Faibussowitsch   PetscCall(PetscObjectSAWsGrantAccess((PetscObject)tao));
22493ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
22508931d482SJason Sarich }
22518931d482SJason Sarich 
22528931d482SJason Sarich /*@
225347450a7bSBarry Smith   TaoGetTotalIterationNumber - Gets the total number of `TaoSolve()` iterations
22548931d482SJason Sarich   completed. This number keeps accumulating if multiple solves
225547450a7bSBarry Smith   are called with the `Tao` object.
22568931d482SJason Sarich 
22578931d482SJason Sarich   Not Collective
22588931d482SJason Sarich 
22598931d482SJason Sarich   Input Parameter:
226047450a7bSBarry Smith . tao - the `Tao` context
22618931d482SJason Sarich 
22628931d482SJason Sarich   Output Parameter:
226347450a7bSBarry Smith . iter - number of iterations
22648931d482SJason Sarich 
22658931d482SJason Sarich   Level: intermediate
22668931d482SJason Sarich 
226720f4b53cSBarry Smith   Note:
226867be906fSBarry Smith   The total iteration count is updated after each solve, if there is a current
226947450a7bSBarry Smith   `TaoSolve()` in progress then those iterations are not included in the count
227067be906fSBarry Smith 
22711cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoGetLinearSolveIterations()`
22728931d482SJason Sarich @*/
2273d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoGetTotalIterationNumber(Tao tao, PetscInt *iter)
2274d71ae5a4SJacob Faibussowitsch {
22758931d482SJason Sarich   PetscFunctionBegin;
22768931d482SJason Sarich   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
22774f572ea9SToby Isaac   PetscAssertPointer(iter, 2);
22788931d482SJason Sarich   *iter = tao->ntotalits;
22793ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
22808931d482SJason Sarich }
22818931d482SJason Sarich 
22828931d482SJason Sarich /*@
22838931d482SJason Sarich   TaoSetTotalIterationNumber - Sets the current total iteration number.
22848931d482SJason Sarich 
2285c3339decSBarry Smith   Logically Collective
22868931d482SJason Sarich 
2287d8d19677SJose E. Roman   Input Parameters:
228847450a7bSBarry Smith + tao  - the `Tao` context
228947450a7bSBarry Smith - iter - the iteration number
22908931d482SJason Sarich 
22918931d482SJason Sarich   Level: developer
22928931d482SJason Sarich 
22931cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoGetLinearSolveIterations()`
22948931d482SJason Sarich @*/
2295d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoSetTotalIterationNumber(Tao tao, PetscInt iter)
2296d71ae5a4SJacob Faibussowitsch {
22978931d482SJason Sarich   PetscFunctionBegin;
22988931d482SJason Sarich   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
229994511df7SStefano Zampini   PetscValidLogicalCollectiveInt(tao, iter, 2);
23009566063dSJacob Faibussowitsch   PetscCall(PetscObjectSAWsTakeAccess((PetscObject)tao));
23018931d482SJason Sarich   tao->ntotalits = iter;
23029566063dSJacob Faibussowitsch   PetscCall(PetscObjectSAWsGrantAccess((PetscObject)tao));
23033ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
23048931d482SJason Sarich }
23058931d482SJason Sarich 
2306a7e14dcfSSatish Balay /*@
230747450a7bSBarry Smith   TaoSetConvergedReason - Sets the termination flag on a `Tao` object
2308a7e14dcfSSatish Balay 
2309c3339decSBarry Smith   Logically Collective
2310a7e14dcfSSatish Balay 
2311a7e14dcfSSatish Balay   Input Parameters:
231247450a7bSBarry Smith + tao    - the `Tao` context
231347450a7bSBarry Smith - reason - the `TaoConvergedReason`
2314a7e14dcfSSatish Balay 
2315a7e14dcfSSatish Balay   Level: intermediate
2316a7e14dcfSSatish Balay 
23171cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoConvergedReason`
2318a7e14dcfSSatish Balay @*/
2319d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoSetConvergedReason(Tao tao, TaoConvergedReason reason)
2320d71ae5a4SJacob Faibussowitsch {
2321a7e14dcfSSatish Balay   PetscFunctionBegin;
232294511df7SStefano Zampini   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
232394511df7SStefano Zampini   PetscValidLogicalCollectiveEnum(tao, reason, 2);
2324a7e14dcfSSatish Balay   tao->reason = reason;
23253ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
2326a7e14dcfSSatish Balay }
2327a7e14dcfSSatish Balay 
2328a7e14dcfSSatish Balay /*@
232947450a7bSBarry Smith   TaoGetConvergedReason - Gets the reason the `TaoSolve()` was stopped.
2330a7e14dcfSSatish Balay 
2331a7e14dcfSSatish Balay   Not Collective
2332a7e14dcfSSatish Balay 
2333a7e14dcfSSatish Balay   Input Parameter:
233447450a7bSBarry Smith . tao - the `Tao` solver context
2335a7e14dcfSSatish Balay 
2336a7e14dcfSSatish Balay   Output Parameter:
233747450a7bSBarry Smith . reason - value of `TaoConvergedReason`
2338a7e14dcfSSatish Balay 
2339a7e14dcfSSatish Balay   Level: intermediate
2340a7e14dcfSSatish Balay 
23411cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoConvergedReason`, `TaoSetConvergenceTest()`, `TaoSetTolerances()`
2342a7e14dcfSSatish Balay @*/
2343d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoGetConvergedReason(Tao tao, TaoConvergedReason *reason)
2344d71ae5a4SJacob Faibussowitsch {
2345a7e14dcfSSatish Balay   PetscFunctionBegin;
2346441846f8SBarry Smith   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
23474f572ea9SToby Isaac   PetscAssertPointer(reason, 2);
2348a7e14dcfSSatish Balay   *reason = tao->reason;
23493ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
2350a7e14dcfSSatish Balay }
2351a7e14dcfSSatish Balay 
2352a7e14dcfSSatish Balay /*@
2353a7e14dcfSSatish Balay   TaoGetSolutionStatus - Get the current iterate, objective value,
235447450a7bSBarry Smith   residual, infeasibility, and termination from a `Tao` object
2355a7e14dcfSSatish Balay 
2356a7e14dcfSSatish Balay   Not Collective
2357a7e14dcfSSatish Balay 
2358f899ff85SJose E. Roman   Input Parameter:
235947450a7bSBarry Smith . tao - the `Tao` context
2360a7e14dcfSSatish Balay 
2361a7e14dcfSSatish Balay   Output Parameters:
2362e056e8ceSJacob Faibussowitsch + its    - the current iterate number (>=0)
2363a7e14dcfSSatish Balay . f      - the current function value
236458417fe7SBarry Smith . gnorm  - the square of the gradient norm, duality gap, or other measure indicating distance from optimality.
2365a7e14dcfSSatish Balay . cnorm  - the infeasibility of the current solution with regard to the constraints.
2366a7e14dcfSSatish Balay . xdiff  - the step length or trust region radius of the most recent iterate.
236765ba42b6SBarry Smith - reason - The termination reason, which can equal `TAO_CONTINUE_ITERATING`
2368a7e14dcfSSatish Balay 
2369a7e14dcfSSatish Balay   Level: intermediate
2370a7e14dcfSSatish Balay 
237165ba42b6SBarry Smith   Notes:
237265ba42b6SBarry Smith   Tao returns the values set by the solvers in the routine `TaoMonitor()`.
2373a7e14dcfSSatish Balay 
237467be906fSBarry Smith   If any of the output arguments are set to `NULL`, no corresponding value will be returned.
2375a7e14dcfSSatish Balay 
23761cc06b55SBarry Smith .seealso: [](ch_tao), `TaoMonitor()`, `TaoGetConvergedReason()`
2377a7e14dcfSSatish Balay @*/
2378d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoGetSolutionStatus(Tao tao, PetscInt *its, PetscReal *f, PetscReal *gnorm, PetscReal *cnorm, PetscReal *xdiff, TaoConvergedReason *reason)
2379d71ae5a4SJacob Faibussowitsch {
2380a7e14dcfSSatish Balay   PetscFunctionBegin;
238194511df7SStefano Zampini   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
2382a7e14dcfSSatish Balay   if (its) *its = tao->niter;
2383a7e14dcfSSatish Balay   if (f) *f = tao->fc;
2384a7e14dcfSSatish Balay   if (gnorm) *gnorm = tao->residual;
2385a7e14dcfSSatish Balay   if (cnorm) *cnorm = tao->cnorm;
2386a7e14dcfSSatish Balay   if (reason) *reason = tao->reason;
2387a7e14dcfSSatish Balay   if (xdiff) *xdiff = tao->step;
23883ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
2389a7e14dcfSSatish Balay }
2390a7e14dcfSSatish Balay 
2391a7e14dcfSSatish Balay /*@C
239247450a7bSBarry Smith   TaoGetType - Gets the current `TaoType` being used in the `Tao` object
2393a7e14dcfSSatish Balay 
2394a7e14dcfSSatish Balay   Not Collective
2395a7e14dcfSSatish Balay 
2396a7e14dcfSSatish Balay   Input Parameter:
239747450a7bSBarry Smith . tao - the `Tao` solver context
2398a7e14dcfSSatish Balay 
2399a7e14dcfSSatish Balay   Output Parameter:
240047450a7bSBarry Smith . type - the `TaoType`
2401a7e14dcfSSatish Balay 
2402a7e14dcfSSatish Balay   Level: intermediate
2403a7e14dcfSSatish Balay 
24041cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoType`, `TaoSetType()`
2405a7e14dcfSSatish Balay @*/
2406d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoGetType(Tao tao, TaoType *type)
2407d71ae5a4SJacob Faibussowitsch {
2408a7e14dcfSSatish Balay   PetscFunctionBegin;
2409441846f8SBarry Smith   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
24104f572ea9SToby Isaac   PetscAssertPointer(type, 2);
2411a7e14dcfSSatish Balay   *type = ((PetscObject)tao)->type_name;
24123ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
2413a7e14dcfSSatish Balay }
2414a7e14dcfSSatish Balay 
2415a7e14dcfSSatish Balay /*@C
2416a7e14dcfSSatish Balay   TaoMonitor - Monitor the solver and the current solution.  This
2417a7e14dcfSSatish Balay   routine will record the iteration number and residual statistics,
2418670c031eSStefano Zampini   and call any monitors specified by the user.
2419a7e14dcfSSatish Balay 
2420a7e14dcfSSatish Balay   Input Parameters:
242147450a7bSBarry Smith + tao        - the `Tao` context
2422a7e14dcfSSatish Balay . its        - the current iterate number (>=0)
2423a7e14dcfSSatish Balay . f          - the current objective function value
24241cb3f120SPierre Jolivet . res        - the gradient norm, square root of the duality gap, or other measure indicating distance from optimality.  This measure will be recorded and
2425a7e14dcfSSatish Balay           used for some termination tests.
2426a7e14dcfSSatish Balay . cnorm      - the infeasibility of the current solution with regard to the constraints.
2427a7e14dcfSSatish Balay - steplength - multiple of the step direction added to the previous iterate.
2428a7e14dcfSSatish Balay 
2429a7e14dcfSSatish Balay   Options Database Key:
2430a7e14dcfSSatish Balay . -tao_monitor - Use the default monitor, which prints statistics to standard output
2431a7e14dcfSSatish Balay 
2432a7e14dcfSSatish Balay   Level: developer
2433a7e14dcfSSatish Balay 
24341cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoGetConvergedReason()`, `TaoMonitorDefault()`, `TaoSetMonitor()`
2435a7e14dcfSSatish Balay @*/
2436d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoMonitor(Tao tao, PetscInt its, PetscReal f, PetscReal res, PetscReal cnorm, PetscReal steplength)
2437d71ae5a4SJacob Faibussowitsch {
243847a47007SBarry Smith   PetscInt i;
243947a47007SBarry Smith 
2440a7e14dcfSSatish Balay   PetscFunctionBegin;
2441441846f8SBarry Smith   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
2442a7e14dcfSSatish Balay   tao->fc       = f;
2443a7e14dcfSSatish Balay   tao->residual = res;
2444a7e14dcfSSatish Balay   tao->cnorm    = cnorm;
2445a7e14dcfSSatish Balay   tao->step     = steplength;
2446e52336cbSBarry Smith   if (!its) {
244794511df7SStefano Zampini     tao->cnorm0 = cnorm;
244894511df7SStefano Zampini     tao->gnorm0 = res;
2449a7e14dcfSSatish Balay   }
2450bfdcf5a2SStefano Zampini   PetscCall(VecLockReadPush(tao->solution));
245148a46eb9SPierre Jolivet   for (i = 0; i < tao->numbermonitors; i++) PetscCall((*tao->monitor[i])(tao, tao->monitorcontext[i]));
2452bfdcf5a2SStefano Zampini   PetscCall(VecLockReadPop(tao->solution));
24533ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
2454a7e14dcfSSatish Balay }
2455a7e14dcfSSatish Balay 
2456a7e14dcfSSatish Balay /*@
2457ae93cb3cSJason Sarich   TaoSetConvergenceHistory - Sets the array used to hold the convergence history.
2458a7e14dcfSSatish Balay 
2459c3339decSBarry Smith   Logically Collective
2460a7e14dcfSSatish Balay 
2461a7e14dcfSSatish Balay   Input Parameters:
246247450a7bSBarry Smith + tao   - the `Tao` solver context
2463a7e14dcfSSatish Balay . obj   - array to hold objective value history
2464a7e14dcfSSatish Balay . resid - array to hold residual history
2465a7e14dcfSSatish Balay . cnorm - array to hold constraint violation history
2466be1558d8SJason Sarich . lits  - integer array holds the number of linear iterations for each Tao iteration
246767be906fSBarry Smith . na    - size of `obj`, `resid`, and `cnorm`
246865ba42b6SBarry Smith - reset - `PETSC_TRUE` indicates each new minimization resets the history counter to zero,
2469a7e14dcfSSatish Balay            else it continues storing new values for new minimizations after the old ones
2470a7e14dcfSSatish Balay 
247167be906fSBarry Smith   Level: intermediate
247267be906fSBarry Smith 
2473a7e14dcfSSatish Balay   Notes:
247447450a7bSBarry Smith   If set, `Tao` will fill the given arrays with the indicated
2475ae93cb3cSJason Sarich   information at each iteration.  If 'obj','resid','cnorm','lits' are
247667be906fSBarry Smith   *all* `NULL` then space (using size `na`, or 1000 if na is `PETSC_DECIDE` or
247765ba42b6SBarry Smith   `PETSC_DEFAULT`) is allocated for the history.
247867be906fSBarry Smith   If not all are `NULL`, then only the non-`NULL` information categories
2479ae93cb3cSJason Sarich   will be stored, the others will be ignored.
2480ae93cb3cSJason Sarich 
2481ae93cb3cSJason Sarich   Any convergence information after iteration number 'na' will not be stored.
2482a7e14dcfSSatish Balay 
2483a7e14dcfSSatish Balay   This routine is useful, e.g., when running a code for purposes
2484a7e14dcfSSatish Balay   of accurate performance monitoring, when no I/O should be done
2485a7e14dcfSSatish Balay   during the section of code that is being timed.
2486a7e14dcfSSatish Balay 
24871cc06b55SBarry Smith .seealso: [](ch_tao), `TaoGetConvergenceHistory()`
2488a7e14dcfSSatish Balay @*/
2489d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoSetConvergenceHistory(Tao tao, PetscReal obj[], PetscReal resid[], PetscReal cnorm[], PetscInt lits[], PetscInt na, PetscBool reset)
2490d71ae5a4SJacob Faibussowitsch {
2491a7e14dcfSSatish Balay   PetscFunctionBegin;
2492441846f8SBarry Smith   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
24934f572ea9SToby Isaac   if (obj) PetscAssertPointer(obj, 2);
24944f572ea9SToby Isaac   if (resid) PetscAssertPointer(resid, 3);
24954f572ea9SToby Isaac   if (cnorm) PetscAssertPointer(cnorm, 4);
24964f572ea9SToby Isaac   if (lits) PetscAssertPointer(lits, 5);
2497ae93cb3cSJason Sarich 
2498ae93cb3cSJason Sarich   if (na == PETSC_DECIDE || na == PETSC_DEFAULT) na = 1000;
2499ae93cb3cSJason Sarich   if (!obj && !resid && !cnorm && !lits) {
25009566063dSJacob Faibussowitsch     PetscCall(PetscCalloc4(na, &obj, na, &resid, na, &cnorm, na, &lits));
2501ae93cb3cSJason Sarich     tao->hist_malloc = PETSC_TRUE;
2502ae93cb3cSJason Sarich   }
2503ae93cb3cSJason Sarich 
2504a7e14dcfSSatish Balay   tao->hist_obj   = obj;
2505a7e14dcfSSatish Balay   tao->hist_resid = resid;
2506a7e14dcfSSatish Balay   tao->hist_cnorm = cnorm;
2507ae93cb3cSJason Sarich   tao->hist_lits  = lits;
2508a7e14dcfSSatish Balay   tao->hist_max   = na;
2509a7e14dcfSSatish Balay   tao->hist_reset = reset;
2510ae93cb3cSJason Sarich   tao->hist_len   = 0;
25113ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
2512a7e14dcfSSatish Balay }
2513a7e14dcfSSatish Balay 
2514a7e14dcfSSatish Balay /*@C
251565ba42b6SBarry Smith   TaoGetConvergenceHistory - Gets the arrays used that hold the convergence history.
2516a7e14dcfSSatish Balay 
2517c3339decSBarry Smith   Collective
2518a7e14dcfSSatish Balay 
2519a7e14dcfSSatish Balay   Input Parameter:
252047450a7bSBarry Smith . tao - the `Tao` context
2521a7e14dcfSSatish Balay 
2522a7e14dcfSSatish Balay   Output Parameters:
2523a7e14dcfSSatish Balay + obj   - array used to hold objective value history
2524a7e14dcfSSatish Balay . resid - array used to hold residual history
2525a7e14dcfSSatish Balay . cnorm - array used to hold constraint violation history
2526ae93cb3cSJason Sarich . lits  - integer array used to hold linear solver iteration count
252767be906fSBarry Smith - nhist - size of `obj`, `resid`, `cnorm`, and `lits`
252867be906fSBarry Smith 
252967be906fSBarry Smith   Level: advanced
2530a7e14dcfSSatish Balay 
2531a7e14dcfSSatish Balay   Notes:
253265ba42b6SBarry Smith   This routine must be preceded by calls to `TaoSetConvergenceHistory()`
253365ba42b6SBarry Smith   and `TaoSolve()`, otherwise it returns useless information.
2534ae93cb3cSJason Sarich 
2535a7e14dcfSSatish Balay   This routine is useful, e.g., when running a code for purposes
2536a7e14dcfSSatish Balay   of accurate performance monitoring, when no I/O should be done
2537a7e14dcfSSatish Balay   during the section of code that is being timed.
2538a7e14dcfSSatish Balay 
2539e056e8ceSJacob Faibussowitsch   Fortran Notes:
254067be906fSBarry Smith   The calling sequence is
254167be906fSBarry Smith .vb
254267be906fSBarry Smith    call TaoGetConvergenceHistory(Tao tao, PetscInt nhist, PetscErrorCode ierr)
254367be906fSBarry Smith .ve
2544a7e14dcfSSatish Balay 
25451cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoSolve()`, `TaoSetConvergenceHistory()`
2546a7e14dcfSSatish Balay @*/
2547d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoGetConvergenceHistory(Tao tao, PetscReal **obj, PetscReal **resid, PetscReal **cnorm, PetscInt **lits, PetscInt *nhist)
2548d71ae5a4SJacob Faibussowitsch {
2549a7e14dcfSSatish Balay   PetscFunctionBegin;
2550441846f8SBarry Smith   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
2551a7e14dcfSSatish Balay   if (obj) *obj = tao->hist_obj;
2552a7e14dcfSSatish Balay   if (cnorm) *cnorm = tao->hist_cnorm;
2553a7e14dcfSSatish Balay   if (resid) *resid = tao->hist_resid;
25545b494b05SBarry Smith   if (lits) *lits = tao->hist_lits;
2555a7e14dcfSSatish Balay   if (nhist) *nhist = tao->hist_len;
25563ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
2557a7e14dcfSSatish Balay }
2558a7e14dcfSSatish Balay 
2559a7e14dcfSSatish Balay /*@
256047450a7bSBarry Smith   TaoSetApplicationContext - Sets the optional user-defined context for a `Tao` solver.
2561a7e14dcfSSatish Balay 
2562c3339decSBarry Smith   Logically Collective
2563a7e14dcfSSatish Balay 
2564a7e14dcfSSatish Balay   Input Parameters:
256547450a7bSBarry Smith + tao  - the `Tao` context
2566a7e14dcfSSatish Balay - usrP - optional user context
2567a7e14dcfSSatish Balay 
2568a7e14dcfSSatish Balay   Level: intermediate
2569a7e14dcfSSatish Balay 
257042747ad1SJacob Faibussowitsch .seealso: [](ch_tao), `Tao`, `TaoGetApplicationContext()`
2571a7e14dcfSSatish Balay @*/
2572d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoSetApplicationContext(Tao tao, void *usrP)
2573d71ae5a4SJacob Faibussowitsch {
2574a7e14dcfSSatish Balay   PetscFunctionBegin;
2575441846f8SBarry Smith   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
2576a7e14dcfSSatish Balay   tao->user = usrP;
25773ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
2578a7e14dcfSSatish Balay }
2579a7e14dcfSSatish Balay 
2580a7e14dcfSSatish Balay /*@
258147450a7bSBarry Smith   TaoGetApplicationContext - Gets the user-defined context for a `Tao` solver
2582a7e14dcfSSatish Balay 
2583a7e14dcfSSatish Balay   Not Collective
2584a7e14dcfSSatish Balay 
2585a7e14dcfSSatish Balay   Input Parameter:
258647450a7bSBarry Smith . tao - the `Tao` context
2587a7e14dcfSSatish Balay 
2588a7e14dcfSSatish Balay   Output Parameter:
2589a7e14dcfSSatish Balay . usrP - user context
2590a7e14dcfSSatish Balay 
2591a7e14dcfSSatish Balay   Level: intermediate
2592a7e14dcfSSatish Balay 
25931cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoSetApplicationContext()`
2594a7e14dcfSSatish Balay @*/
2595d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoGetApplicationContext(Tao tao, void *usrP)
2596d71ae5a4SJacob Faibussowitsch {
2597a7e14dcfSSatish Balay   PetscFunctionBegin;
2598441846f8SBarry Smith   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
25994f572ea9SToby Isaac   PetscAssertPointer(usrP, 2);
2600a7e14dcfSSatish Balay   *(void **)usrP = tao->user;
26013ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
2602a7e14dcfSSatish Balay }
2603a9603a14SPatrick Farrell 
2604a9603a14SPatrick Farrell /*@
260565ba42b6SBarry Smith   TaoSetGradientNorm - Sets the matrix used to define the norm that measures the size of the gradient.
2606a9603a14SPatrick Farrell 
2607c3339decSBarry Smith   Collective
2608a9603a14SPatrick Farrell 
2609a9603a14SPatrick Farrell   Input Parameters:
261047450a7bSBarry Smith + tao - the `Tao` context
261165ba42b6SBarry Smith - M   - matrix that defines the norm
2612a9603a14SPatrick Farrell 
2613a9603a14SPatrick Farrell   Level: beginner
2614a9603a14SPatrick Farrell 
26151cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoGetGradientNorm()`, `TaoGradientNorm()`
2616a9603a14SPatrick Farrell @*/
2617d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoSetGradientNorm(Tao tao, Mat M)
2618d71ae5a4SJacob Faibussowitsch {
2619a9603a14SPatrick Farrell   PetscFunctionBegin;
2620a9603a14SPatrick Farrell   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
262194511df7SStefano Zampini   PetscValidHeaderSpecific(M, MAT_CLASSID, 2);
26229566063dSJacob Faibussowitsch   PetscCall(PetscObjectReference((PetscObject)M));
26239566063dSJacob Faibussowitsch   PetscCall(MatDestroy(&tao->gradient_norm));
26249566063dSJacob Faibussowitsch   PetscCall(VecDestroy(&tao->gradient_norm_tmp));
2625a9603a14SPatrick Farrell   tao->gradient_norm = M;
26269566063dSJacob Faibussowitsch   PetscCall(MatCreateVecs(M, NULL, &tao->gradient_norm_tmp));
26273ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
2628a9603a14SPatrick Farrell }
2629a9603a14SPatrick Farrell 
2630a9603a14SPatrick Farrell /*@
263165ba42b6SBarry Smith   TaoGetGradientNorm - Returns the matrix used to define the norm used for measuring the size of the gradient.
2632a9603a14SPatrick Farrell 
2633a9603a14SPatrick Farrell   Not Collective
2634a9603a14SPatrick Farrell 
2635a9603a14SPatrick Farrell   Input Parameter:
263647450a7bSBarry Smith . tao - the `Tao` context
2637a9603a14SPatrick Farrell 
2638a9603a14SPatrick Farrell   Output Parameter:
2639a9603a14SPatrick Farrell . M - gradient norm
2640a9603a14SPatrick Farrell 
2641a9603a14SPatrick Farrell   Level: beginner
2642a9603a14SPatrick Farrell 
26431cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoSetGradientNorm()`, `TaoGradientNorm()`
2644a9603a14SPatrick Farrell @*/
2645d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoGetGradientNorm(Tao tao, Mat *M)
2646d71ae5a4SJacob Faibussowitsch {
2647a9603a14SPatrick Farrell   PetscFunctionBegin;
2648a9603a14SPatrick Farrell   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
26494f572ea9SToby Isaac   PetscAssertPointer(M, 2);
2650a9603a14SPatrick Farrell   *M = tao->gradient_norm;
26513ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
2652a9603a14SPatrick Farrell }
2653a9603a14SPatrick Farrell 
265405aca2cdSSatish Balay /*@C
265547450a7bSBarry Smith   TaoGradientNorm - Compute the norm using the `NormType`, the user has selected
2656a9603a14SPatrick Farrell 
2657c3339decSBarry Smith   Collective
2658a9603a14SPatrick Farrell 
2659d8d19677SJose E. Roman   Input Parameters:
266047450a7bSBarry Smith + tao      - the `Tao` context
2661a9603a14SPatrick Farrell . gradient - the gradient to be computed
26623b242c63SJacob Faibussowitsch - type     - the norm type
2663a9603a14SPatrick Farrell 
2664a9603a14SPatrick Farrell   Output Parameter:
2665a9603a14SPatrick Farrell . gnorm - the gradient norm
2666a9603a14SPatrick Farrell 
266747450a7bSBarry Smith   Level: advanced
2668a9603a14SPatrick Farrell 
26691cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoSetGradientNorm()`, `TaoGetGradientNorm()`
2670a9603a14SPatrick Farrell @*/
2671d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoGradientNorm(Tao tao, Vec gradient, NormType type, PetscReal *gnorm)
2672d71ae5a4SJacob Faibussowitsch {
2673a9603a14SPatrick Farrell   PetscFunctionBegin;
26748854b543SStefano Zampini   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
26758854b543SStefano Zampini   PetscValidHeaderSpecific(gradient, VEC_CLASSID, 2);
26768854b543SStefano Zampini   PetscValidLogicalCollectiveEnum(tao, type, 3);
26774f572ea9SToby Isaac   PetscAssertPointer(gnorm, 4);
2678a9603a14SPatrick Farrell   if (tao->gradient_norm) {
2679680e2bc7SPatrick Farrell     PetscScalar gnorms;
2680680e2bc7SPatrick Farrell 
26813c859ba3SBarry 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.");
26829566063dSJacob Faibussowitsch     PetscCall(MatMult(tao->gradient_norm, gradient, tao->gradient_norm_tmp));
26839566063dSJacob Faibussowitsch     PetscCall(VecDot(gradient, tao->gradient_norm_tmp, &gnorms));
26847bb79937SPatrick Farrell     *gnorm = PetscRealPart(PetscSqrtScalar(gnorms));
2685a9603a14SPatrick Farrell   } else {
26869566063dSJacob Faibussowitsch     PetscCall(VecNorm(gradient, type, gnorm));
2687a9603a14SPatrick Farrell   }
26883ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
2689a9603a14SPatrick Farrell }
2690a9603a14SPatrick Farrell 
2691e882e171SHong Zhang /*@C
269247450a7bSBarry Smith   TaoMonitorDrawCtxCreate - Creates the monitor context for `TaoMonitorDrawSolution()`
2693a9603a14SPatrick Farrell 
2694c3339decSBarry Smith   Collective
2695e882e171SHong Zhang 
26962fe279fdSBarry Smith   Input Parameters:
26972fe279fdSBarry Smith + comm     - the communicator to share the context
26982fe279fdSBarry Smith . host     - the name of the X Windows host that will display the monitor
26992fe279fdSBarry Smith . label    - the label to put at the top of the display window
27002fe279fdSBarry Smith . x        - the horizontal coordinate of the lower left corner of the window to open
27012fe279fdSBarry Smith . y        - the vertical coordinate of the lower left corner of the window to open
27022fe279fdSBarry Smith . m        - the width of the window
27032fe279fdSBarry Smith . n        - the height of the window
27042fe279fdSBarry Smith - howoften - how many `Tao` iterations between displaying the monitor information
27052fe279fdSBarry Smith 
2706d5b43468SJose E. Roman   Output Parameter:
2707e882e171SHong Zhang . ctx - the monitor context
2708e882e171SHong Zhang 
27093c7db156SBarry Smith   Options Database Key:
2710e882e171SHong Zhang . -tao_draw_solution_initial - show initial guess as well as current solution
2711e882e171SHong Zhang 
2712e882e171SHong Zhang   Level: intermediate
2713e882e171SHong Zhang 
27141cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoMonitorSet()`, `TaoMonitorDefault()`, `VecView()`, `TaoMonitorDrawCtx()`
2715e882e171SHong Zhang @*/
2716d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoMonitorDrawCtxCreate(MPI_Comm comm, const char host[], const char label[], int x, int y, int m, int n, PetscInt howoften, TaoMonitorDrawCtx *ctx)
2717d71ae5a4SJacob Faibussowitsch {
2718e882e171SHong Zhang   PetscFunctionBegin;
27199566063dSJacob Faibussowitsch   PetscCall(PetscNew(ctx));
27209566063dSJacob Faibussowitsch   PetscCall(PetscViewerDrawOpen(comm, host, label, x, y, m, n, &(*ctx)->viewer));
27219566063dSJacob Faibussowitsch   PetscCall(PetscViewerSetFromOptions((*ctx)->viewer));
2722e882e171SHong Zhang   (*ctx)->howoften = howoften;
27233ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
2724e882e171SHong Zhang }
2725e882e171SHong Zhang 
2726e882e171SHong Zhang /*@C
272765ba42b6SBarry Smith   TaoMonitorDrawCtxDestroy - Destroys the monitor context for `TaoMonitorDrawSolution()`
2728e882e171SHong Zhang 
2729c3339decSBarry Smith   Collective
2730e882e171SHong Zhang 
273147450a7bSBarry Smith   Input Parameter:
2732e056e8ceSJacob Faibussowitsch . ictx - the monitor context
2733e882e171SHong Zhang 
2734e882e171SHong Zhang   Level: intermediate
2735e882e171SHong Zhang 
27361cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoMonitorSet()`, `TaoMonitorDefault()`, `VecView()`, `TaoMonitorDrawSolution()`
2737e882e171SHong Zhang @*/
2738d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoMonitorDrawCtxDestroy(TaoMonitorDrawCtx *ictx)
2739d71ae5a4SJacob Faibussowitsch {
2740e882e171SHong Zhang   PetscFunctionBegin;
27419566063dSJacob Faibussowitsch   PetscCall(PetscViewerDestroy(&(*ictx)->viewer));
27429566063dSJacob Faibussowitsch   PetscCall(PetscFree(*ictx));
27433ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
2744e882e171SHong Zhang }
2745