xref: /petsc/src/tao/interface/taosolver.c (revision a1cb98fac0cdf0eb4d3e8a0c8b58f3fe8f800bc6) !
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;
290f0abf79SStefano Zampini   if (!snes_ewdummy) PetscFunctionReturn(0);
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;
370f0abf79SStefano Zampini   PetscFunctionReturn(0);
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;
450f0abf79SStefano Zampini   if (!snes_ewdummy) PetscFunctionReturn(0);
460f0abf79SStefano Zampini   PetscCall(KSPPostSolve_SNESEW(ksp, b, x, snes_ewdummy));
470f0abf79SStefano Zampini   PetscFunctionReturn(0);
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;
560f0abf79SStefano Zampini   if (!tao->ksp) PetscFunctionReturn(0);
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;
650f0abf79SStefano Zampini     PetscCall(SNESEWSetFromOptions_Private(kctx, PetscObjectComm((PetscObject)tao), ewprefix));
660f0abf79SStefano Zampini   } else PetscCall(SNESDestroy(&tao->snes_ewdummy));
670f0abf79SStefano Zampini   PetscFunctionReturn(0);
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:
79441846f8SBarry Smith . newtao - the new Tao context
80a7e14dcfSSatish Balay 
81a7e14dcfSSatish Balay   Available methods include:
8265ba42b6SBarry Smith +    `TAONLS` - nls Newton's method with line search for unconstrained minimization
8365ba42b6SBarry Smith .    `TAONTR` - ntr Newton's method with trust region for unconstrained minimization
8465ba42b6SBarry Smith .    `TAONTL` - ntl Newton's method with trust region, line search for unconstrained minimization
8565ba42b6SBarry Smith .    `TAOLMVM` - lmvm Limited memory variable metric method for unconstrained minimization
8665ba42b6SBarry Smith .    `TAOCG` - cg Nonlinear conjugate gradient method for unconstrained minimization
8765ba42b6SBarry Smith .    `TAONM` - nm Nelder-Mead algorithm for derivate-free unconstrained minimization
8865ba42b6SBarry Smith .    `TAOTRON` - tron Newton Trust Region method for bound constrained minimization
8965ba42b6SBarry Smith .    `TAOGPCG` - gpcg Newton Trust Region method for quadratic bound constrained minimization
9065ba42b6SBarry Smith .    `TAOBLMVM` - blmvm Limited memory variable metric method for bound constrained minimization
9165ba42b6SBarry Smith .    `TAOLCL` - lcl Linearly constrained Lagrangian method for pde-constrained minimization
9265ba42b6SBarry Smith -    `TAOPOUNDERS` - pounders Model-based algorithm for nonlinear least squares
93a7e14dcfSSatish Balay 
94a7e14dcfSSatish Balay    Options Database Keys:
9565ba42b6SBarry Smith .   -tao_type - select which method Tao should use
96a7e14dcfSSatish Balay 
97a7e14dcfSSatish Balay    Level: beginner
98a7e14dcfSSatish Balay 
9965ba42b6SBarry Smith .seealso: `Tao`, `TaoSolve()`, `TaoDestroy()`, `TAOSetFromOptions()`, `TAOSetType()`
100a7e14dcfSSatish Balay @*/
101d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoCreate(MPI_Comm comm, Tao *newtao)
102d71ae5a4SJacob Faibussowitsch {
103441846f8SBarry Smith   Tao tao;
104a7e14dcfSSatish Balay 
105a7e14dcfSSatish Balay   PetscFunctionBegin;
106a7e14dcfSSatish Balay   PetscValidPointer(newtao, 2);
1079566063dSJacob Faibussowitsch   PetscCall(TaoInitializePackage());
1089566063dSJacob Faibussowitsch   PetscCall(TaoLineSearchInitializePackage());
1099566063dSJacob Faibussowitsch   PetscCall(PetscHeaderCreate(tao, TAO_CLASSID, "Tao", "Optimization solver", "Tao", comm, TaoDestroy, TaoView));
110a7e14dcfSSatish Balay 
11194511df7SStefano Zampini   /* Set non-NULL defaults */
11294511df7SStefano Zampini   tao->ops->convergencetest = TaoDefaultConvergenceTest;
113a7e14dcfSSatish Balay 
114a7e14dcfSSatish Balay   tao->max_it    = 10000;
11594511df7SStefano Zampini   tao->max_funcs = -1;
1166f4723b1SBarry Smith #if defined(PETSC_USE_REAL_SINGLE)
1176f4723b1SBarry Smith   tao->gatol = 1e-5;
1186f4723b1SBarry Smith   tao->grtol = 1e-5;
1196f1f5f59SAlp Dener   tao->crtol = 1e-5;
1206f1f5f59SAlp Dener   tao->catol = 1e-5;
1216f4723b1SBarry Smith #else
122a7e14dcfSSatish Balay   tao->gatol = 1e-8;
123a7e14dcfSSatish Balay   tao->grtol = 1e-8;
1246f1f5f59SAlp Dener   tao->crtol = 1e-8;
1256f1f5f59SAlp Dener   tao->catol = 1e-8;
1266f4723b1SBarry Smith #endif
1276552cf8aSJason Sarich   tao->gttol   = 0.0;
1284f1535bcSTodd Munson   tao->steptol = 0.0;
129e270355aSBarry Smith   tao->trust0  = PETSC_INFINITY;
1306f4723b1SBarry Smith   tao->fmin    = PETSC_NINFINITY;
1319fa2b5dcSStefano Zampini 
132a7e14dcfSSatish Balay   tao->hist_reset = PETSC_TRUE;
133a7e14dcfSSatish Balay 
1349566063dSJacob Faibussowitsch   PetscCall(TaoResetStatistics(tao));
135a7e14dcfSSatish Balay   *newtao = tao;
136a7e14dcfSSatish Balay   PetscFunctionReturn(0);
137a7e14dcfSSatish Balay }
138a7e14dcfSSatish Balay 
139a7e14dcfSSatish Balay /*@
140a7e14dcfSSatish Balay   TaoSolve - Solves an optimization problem min F(x) s.t. l <= x <= u
141a7e14dcfSSatish Balay 
14265ba42b6SBarry Smith   Collective on tao
143a7e14dcfSSatish Balay 
144a7e14dcfSSatish Balay   Input Parameters:
145441846f8SBarry Smith . tao - the Tao context
146a7e14dcfSSatish Balay 
147a7e14dcfSSatish Balay   Notes:
14865ba42b6SBarry Smith   The user must set up the Tao with calls to `TaoSetSolution()`, `TaoSetObjective()`, `TaoSetGradient()`, and (if using 2nd order method) `TaoSetHessian()`.
149a7e14dcfSSatish Balay 
15065ba42b6SBarry Smith   You should call `TaoGetConvergedReason()` or run with -tao_converged_reason to determine if the optimization algorithm actually succeeded or
151a35d58b8SBarry Smith   why it failed.
152a35d58b8SBarry Smith 
153a7e14dcfSSatish Balay   Level: beginner
154a7e14dcfSSatish Balay 
15565ba42b6SBarry Smith .seealso: `Tao`, `TaoCreate()`, `TaoSetObjective()`, `TaoSetGradient()`, `TaoSetHessian()`, `TaoGetConvergedReason()`, `TaoSetUp()`
156a7e14dcfSSatish Balay  @*/
157d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoSolve(Tao tao)
158d71ae5a4SJacob Faibussowitsch {
159e2379f4fSBarry Smith   static PetscBool set = PETSC_FALSE;
16047a47007SBarry Smith 
161a7e14dcfSSatish Balay   PetscFunctionBegin;
162441846f8SBarry Smith   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
163d0609cedSBarry Smith   PetscCall(PetscCitationsRegister("@TechReport{tao-user-ref,\n"
164e2379f4fSBarry Smith                                    "title   = {Toolkit for Advanced Optimization (TAO) Users Manual},\n"
165e2379f4fSBarry Smith                                    "author  = {Todd Munson and Jason Sarich and Stefan Wild and Steve Benson and Lois Curfman McInnes},\n"
166e2379f4fSBarry Smith                                    "Institution = {Argonne National Laboratory},\n"
167e2379f4fSBarry Smith                                    "Year   = 2014,\n"
168e2379f4fSBarry Smith                                    "Number = {ANL/MCS-TM-322 - Revision 3.5},\n"
1699371c9d4SSatish Balay                                    "url    = {https://www.mcs.anl.gov/research/projects/tao/}\n}\n",
1709371c9d4SSatish Balay                                    &set));
171494bef23SAlp Dener   tao->header_printed = PETSC_FALSE;
1729566063dSJacob Faibussowitsch   PetscCall(TaoSetUp(tao));
1739566063dSJacob Faibussowitsch   PetscCall(TaoResetStatistics(tao));
1741baa6e33SBarry Smith   if (tao->linesearch) PetscCall(TaoLineSearchReset(tao->linesearch));
175a7e14dcfSSatish Balay 
1769566063dSJacob Faibussowitsch   PetscCall(PetscLogEventBegin(TAO_Solve, tao, 0, 0, 0));
177dbbe0bcdSBarry Smith   PetscTryTypeMethod(tao, solve);
1789566063dSJacob Faibussowitsch   PetscCall(PetscLogEventEnd(TAO_Solve, tao, 0, 0, 0));
179a7e14dcfSSatish Balay 
1809566063dSJacob Faibussowitsch   PetscCall(VecViewFromOptions(tao->solution, (PetscObject)tao, "-tao_view_solution"));
1817cf37e64SBarry Smith 
1828931d482SJason Sarich   tao->ntotalits += tao->niter;
1839566063dSJacob Faibussowitsch   PetscCall(TaoViewFromOptions(tao, NULL, "-tao_view"));
184a7e14dcfSSatish Balay 
185a7e14dcfSSatish Balay   if (tao->printreason) {
186a7e14dcfSSatish Balay     if (tao->reason > 0) {
18763a3b9bcSJacob Faibussowitsch       PetscCall(PetscPrintf(((PetscObject)tao)->comm, "TAO solve converged due to %s iterations %" PetscInt_FMT "\n", TaoConvergedReasons[tao->reason], tao->niter));
188a7e14dcfSSatish Balay     } else {
18963a3b9bcSJacob Faibussowitsch       PetscCall(PetscPrintf(((PetscObject)tao)->comm, "TAO solve did not converge due to %s iteration %" PetscInt_FMT "\n", TaoConvergedReasons[tao->reason], tao->niter));
190a7e14dcfSSatish Balay     }
191a7e14dcfSSatish Balay   }
192a7e14dcfSSatish Balay   PetscFunctionReturn(0);
193a7e14dcfSSatish Balay }
194a7e14dcfSSatish Balay 
195a7e14dcfSSatish Balay /*@
196a7e14dcfSSatish Balay   TaoSetUp - Sets up the internal data structures for the later use
197a7e14dcfSSatish Balay   of a Tao solver
198a7e14dcfSSatish Balay 
199a7e14dcfSSatish Balay   Collective on tao
200a7e14dcfSSatish Balay 
201a7e14dcfSSatish Balay   Input Parameters:
20265ba42b6SBarry Smith . tao - the Tao context
203a7e14dcfSSatish Balay 
204a7e14dcfSSatish Balay   Notes:
20565ba42b6SBarry Smith   The user will not need to explicitly call `TaoSetUp()`, as it will
20665ba42b6SBarry Smith   automatically be called in `TaoSolve()`.  However, if the user
20765ba42b6SBarry Smith   desires to call it explicitly, it should come after `TaoCreate()`
20865ba42b6SBarry Smith   and any TaoSetSomething() routines, but before `TaoSolve()`.
209a7e14dcfSSatish Balay 
210a7e14dcfSSatish Balay   Level: advanced
211a7e14dcfSSatish Balay 
21265ba42b6SBarry Smith .seealso: `Tao`, `TaoCreate()`, `TaoSolve()`
213a7e14dcfSSatish Balay @*/
214d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoSetUp(Tao tao)
215d71ae5a4SJacob Faibussowitsch {
216a7e14dcfSSatish Balay   PetscFunctionBegin;
217441846f8SBarry Smith   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
218a7e14dcfSSatish Balay   if (tao->setupcalled) PetscFunctionReturn(0);
2190f0abf79SStefano Zampini   PetscCall(TaoSetUpEW_Private(tao));
2203c859ba3SBarry Smith   PetscCheck(tao->solution, PetscObjectComm((PetscObject)tao), PETSC_ERR_ARG_WRONGSTATE, "Must call TaoSetSolution");
221dbbe0bcdSBarry Smith   PetscTryTypeMethod(tao, setup);
222a7e14dcfSSatish Balay   tao->setupcalled = PETSC_TRUE;
223a7e14dcfSSatish Balay   PetscFunctionReturn(0);
224a7e14dcfSSatish Balay }
225a7e14dcfSSatish Balay 
2261fb7b255SJunchao Zhang /*@C
22765ba42b6SBarry Smith   TaoDestroy - Destroys the Tao context that was created with `TaoCreate()`
228a7e14dcfSSatish Balay 
22965ba42b6SBarry Smith   Collective on tao
230a7e14dcfSSatish Balay 
231a7e14dcfSSatish Balay   Input Parameter:
232441846f8SBarry Smith . tao - the Tao context
233a7e14dcfSSatish Balay 
234a7e14dcfSSatish Balay   Level: beginner
235a7e14dcfSSatish Balay 
23665ba42b6SBarry Smith .seealso: `Tao`, `TaoCreate()`, `TaoSolve()`
237a7e14dcfSSatish Balay @*/
238d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoDestroy(Tao *tao)
239d71ae5a4SJacob Faibussowitsch {
240a7e14dcfSSatish Balay   PetscFunctionBegin;
241a7e14dcfSSatish Balay   if (!*tao) PetscFunctionReturn(0);
242441846f8SBarry Smith   PetscValidHeaderSpecific(*tao, TAO_CLASSID, 1);
2439371c9d4SSatish Balay   if (--((PetscObject)*tao)->refct > 0) {
2449371c9d4SSatish Balay     *tao = NULL;
2459371c9d4SSatish Balay     PetscFunctionReturn(0);
246a7e14dcfSSatish Balay   }
2479371c9d4SSatish Balay 
24848a46eb9SPierre Jolivet   if ((*tao)->ops->destroy) PetscCall((*((*tao))->ops->destroy)(*tao));
2499566063dSJacob Faibussowitsch   PetscCall(KSPDestroy(&(*tao)->ksp));
2500f0abf79SStefano Zampini   PetscCall(SNESDestroy(&(*tao)->snes_ewdummy));
2519566063dSJacob Faibussowitsch   PetscCall(TaoLineSearchDestroy(&(*tao)->linesearch));
252a7e14dcfSSatish Balay 
253a7e14dcfSSatish Balay   if ((*tao)->ops->convergencedestroy) {
2549566063dSJacob Faibussowitsch     PetscCall((*(*tao)->ops->convergencedestroy)((*tao)->cnvP));
25548a46eb9SPierre Jolivet     if ((*tao)->jacobian_state_inv) PetscCall(MatDestroy(&(*tao)->jacobian_state_inv));
256a7e14dcfSSatish Balay   }
2579566063dSJacob Faibussowitsch   PetscCall(VecDestroy(&(*tao)->solution));
2589566063dSJacob Faibussowitsch   PetscCall(VecDestroy(&(*tao)->gradient));
2599566063dSJacob Faibussowitsch   PetscCall(VecDestroy(&(*tao)->ls_res));
260a7e14dcfSSatish Balay 
261a9603a14SPatrick Farrell   if ((*tao)->gradient_norm) {
2629566063dSJacob Faibussowitsch     PetscCall(PetscObjectDereference((PetscObject)(*tao)->gradient_norm));
2639566063dSJacob Faibussowitsch     PetscCall(VecDestroy(&(*tao)->gradient_norm_tmp));
264a9603a14SPatrick Farrell   }
265a9603a14SPatrick Farrell 
2669566063dSJacob Faibussowitsch   PetscCall(VecDestroy(&(*tao)->XL));
2679566063dSJacob Faibussowitsch   PetscCall(VecDestroy(&(*tao)->XU));
2689566063dSJacob Faibussowitsch   PetscCall(VecDestroy(&(*tao)->IL));
2699566063dSJacob Faibussowitsch   PetscCall(VecDestroy(&(*tao)->IU));
2709566063dSJacob Faibussowitsch   PetscCall(VecDestroy(&(*tao)->DE));
2719566063dSJacob Faibussowitsch   PetscCall(VecDestroy(&(*tao)->DI));
2729566063dSJacob Faibussowitsch   PetscCall(VecDestroy(&(*tao)->constraints));
2739566063dSJacob Faibussowitsch   PetscCall(VecDestroy(&(*tao)->constraints_equality));
2749566063dSJacob Faibussowitsch   PetscCall(VecDestroy(&(*tao)->constraints_inequality));
2759566063dSJacob Faibussowitsch   PetscCall(VecDestroy(&(*tao)->stepdirection));
2769566063dSJacob Faibussowitsch   PetscCall(MatDestroy(&(*tao)->hessian_pre));
2779566063dSJacob Faibussowitsch   PetscCall(MatDestroy(&(*tao)->hessian));
2789566063dSJacob Faibussowitsch   PetscCall(MatDestroy(&(*tao)->ls_jac));
2799566063dSJacob Faibussowitsch   PetscCall(MatDestroy(&(*tao)->ls_jac_pre));
2809566063dSJacob Faibussowitsch   PetscCall(MatDestroy(&(*tao)->jacobian_pre));
2819566063dSJacob Faibussowitsch   PetscCall(MatDestroy(&(*tao)->jacobian));
2829566063dSJacob Faibussowitsch   PetscCall(MatDestroy(&(*tao)->jacobian_state_pre));
2839566063dSJacob Faibussowitsch   PetscCall(MatDestroy(&(*tao)->jacobian_state));
2849566063dSJacob Faibussowitsch   PetscCall(MatDestroy(&(*tao)->jacobian_state_inv));
2859566063dSJacob Faibussowitsch   PetscCall(MatDestroy(&(*tao)->jacobian_design));
2869566063dSJacob Faibussowitsch   PetscCall(MatDestroy(&(*tao)->jacobian_equality));
2879566063dSJacob Faibussowitsch   PetscCall(MatDestroy(&(*tao)->jacobian_equality_pre));
2889566063dSJacob Faibussowitsch   PetscCall(MatDestroy(&(*tao)->jacobian_inequality));
2899566063dSJacob Faibussowitsch   PetscCall(MatDestroy(&(*tao)->jacobian_inequality_pre));
2909566063dSJacob Faibussowitsch   PetscCall(ISDestroy(&(*tao)->state_is));
2919566063dSJacob Faibussowitsch   PetscCall(ISDestroy(&(*tao)->design_is));
2929566063dSJacob Faibussowitsch   PetscCall(VecDestroy(&(*tao)->res_weights_v));
2939566063dSJacob Faibussowitsch   PetscCall(TaoCancelMonitors(*tao));
29448a46eb9SPierre Jolivet   if ((*tao)->hist_malloc) PetscCall(PetscFree4((*tao)->hist_obj, (*tao)->hist_resid, (*tao)->hist_cnorm, (*tao)->hist_lits));
295737f463aSAlp Dener   if ((*tao)->res_weights_n) {
2969566063dSJacob Faibussowitsch     PetscCall(PetscFree((*tao)->res_weights_rows));
2979566063dSJacob Faibussowitsch     PetscCall(PetscFree((*tao)->res_weights_cols));
2989566063dSJacob Faibussowitsch     PetscCall(PetscFree((*tao)->res_weights_w));
299125ddc8aSJason Sarich   }
3009566063dSJacob Faibussowitsch   PetscCall(PetscHeaderDestroy(tao));
301a7e14dcfSSatish Balay   PetscFunctionReturn(0);
302a7e14dcfSSatish Balay }
303a7e14dcfSSatish Balay 
304a7e14dcfSSatish Balay /*@
30565ba42b6SBarry Smith    TaoKSPSetUseEW - Sets `SNES` use Eisenstat-Walker method for
3060f0abf79SStefano Zampini    computing relative tolerance for linear solvers.
3070f0abf79SStefano Zampini 
30865ba42b6SBarry Smith    Logically Collective on tao
3090f0abf79SStefano Zampini 
3100f0abf79SStefano Zampini    Input Parameters:
3110f0abf79SStefano Zampini +  tao - Tao context
31265ba42b6SBarry Smith -  flag - `PETSC_TRUE` or `PETSC_FALSE`
3130f0abf79SStefano Zampini 
3140f0abf79SStefano Zampini    Notes:
31565ba42b6SBarry Smith    See `SNESKSPSetUseEW()` for customization details.
3160f0abf79SStefano Zampini 
3170f0abf79SStefano Zampini    Level: advanced
3180f0abf79SStefano Zampini 
3190f0abf79SStefano Zampini    Reference:
3200f0abf79SStefano Zampini    S. C. Eisenstat and H. F. Walker, "Choosing the forcing terms in an
3210f0abf79SStefano Zampini    inexact Newton method", SISC 17 (1), pp.16-32, 1996.
3220f0abf79SStefano Zampini 
32365ba42b6SBarry Smith .seealso: `Tao`, `SNESKSPSetUseEW()`
3240f0abf79SStefano Zampini @*/
325d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoKSPSetUseEW(Tao tao, PetscBool flag)
326d71ae5a4SJacob Faibussowitsch {
3270f0abf79SStefano Zampini   PetscFunctionBegin;
3280f0abf79SStefano Zampini   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
3290f0abf79SStefano Zampini   PetscValidLogicalCollectiveBool(tao, flag, 2);
3300f0abf79SStefano Zampini   tao->ksp_ewconv = flag;
3310f0abf79SStefano Zampini   PetscFunctionReturn(0);
3320f0abf79SStefano Zampini }
3330f0abf79SStefano Zampini 
3340f0abf79SStefano Zampini /*@
335441846f8SBarry Smith   TaoSetFromOptions - Sets various Tao parameters from user
336a7e14dcfSSatish Balay   options.
337a7e14dcfSSatish Balay 
33865ba42b6SBarry Smith   Collective on tao
339a7e14dcfSSatish Balay 
34001d2d390SJose E. Roman   Input Parameter:
341441846f8SBarry Smith . tao - the Tao solver context
342a7e14dcfSSatish Balay 
343a7e14dcfSSatish Balay   options Database Keys:
34465ba42b6SBarry Smith + -tao_type <type> - The algorithm that Tao uses (lmvm, nls, etc.)
345a7e14dcfSSatish Balay . -tao_gatol <gatol> - absolute error tolerance for ||gradient||
346a7e14dcfSSatish Balay . -tao_grtol <grtol> - relative error tolerance for ||gradient||
347a7e14dcfSSatish Balay . -tao_gttol <gttol> - reduction of ||gradient|| relative to initial gradient
348a7e14dcfSSatish Balay . -tao_max_it <max> - sets maximum number of iterations
349a7e14dcfSSatish Balay . -tao_max_funcs <max> - sets maximum number of function evaluations
350a7e14dcfSSatish Balay . -tao_fmin <fmin> - stop if function value reaches fmin
351a7e14dcfSSatish Balay . -tao_steptol <tol> - stop if trust region radius less than <tol>
352a7e14dcfSSatish Balay . -tao_trust0 <t> - initial trust region radius
353a7e14dcfSSatish Balay . -tao_monitor - prints function value and residual at each iteration
354a7e14dcfSSatish Balay . -tao_smonitor - same as tao_monitor, but truncates very small values
355a7e14dcfSSatish Balay . -tao_cmonitor - prints function value, residual, and constraint norm at each iteration
356a7e14dcfSSatish Balay . -tao_view_solution - prints solution vector at each iteration
3574a48860cSAlp Dener . -tao_view_ls_residual - prints least-squares residual vector at each iteration
358147403d9SBarry Smith . -tao_view_stepdirection - prints step direction vector at each iteration
359a7e14dcfSSatish Balay . -tao_view_gradient - prints gradient vector at each iteration
360a7e14dcfSSatish Balay . -tao_draw_solution - graphically view solution vector at each iteration
361a7e14dcfSSatish Balay . -tao_draw_step - graphically view step vector at each iteration
362a7e14dcfSSatish Balay . -tao_draw_gradient - graphically view gradient at each iteration
363a7e14dcfSSatish Balay . -tao_fd_gradient - use gradient computed with finite differences
364f4c1ad5cSStefano Zampini . -tao_fd_hessian - use hessian computed with finite differences
365f4c1ad5cSStefano Zampini . -tao_mf_hessian - use matrix-free hessian computed with finite differences
366a7e14dcfSSatish Balay . -tao_cancelmonitors - cancels all monitors (except those set with command line)
367441846f8SBarry Smith . -tao_view - prints information about the Tao after solving
36865ba42b6SBarry Smith - -tao_converged_reason - prints the reason Tao stopped iterating
369a7e14dcfSSatish Balay 
370a7e14dcfSSatish Balay   Notes:
371a7e14dcfSSatish Balay   To see all options, run your program with the -help option or consult the
37265ba42b6SBarry Smith  user's manual. Should be called after `TaoCreate()` but before `TaoSolve()`
373a7e14dcfSSatish Balay 
374a7e14dcfSSatish Balay   Level: beginner
37565ba42b6SBarry Smith 
37665ba42b6SBarry Smith .seealso: `Tao`, `TaoCreate()`, `TaoSolve()`
377a7e14dcfSSatish Balay @*/
378d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoSetFromOptions(Tao tao)
379d71ae5a4SJacob Faibussowitsch {
380b625d6c7SJed Brown   TaoType     default_type = TAOLMVM;
381a7e14dcfSSatish Balay   char        type[256], monfilename[PETSC_MAX_PATH_LEN];
382a7e14dcfSSatish Balay   PetscViewer monviewer;
383a7e14dcfSSatish Balay   PetscBool   flg;
384a7e14dcfSSatish Balay   MPI_Comm    comm;
385a7e14dcfSSatish Balay 
386a7e14dcfSSatish Balay   PetscFunctionBegin;
387441846f8SBarry Smith   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
3889566063dSJacob Faibussowitsch   PetscCall(PetscObjectGetComm((PetscObject)tao, &comm));
389125ddc8aSJason Sarich 
39060b70c5cSStefano Zampini   if (((PetscObject)tao)->type_name) default_type = ((PetscObject)tao)->type_name;
39160b70c5cSStefano Zampini 
392a7e14dcfSSatish Balay   /* So no warnings are given about unused options */
3939566063dSJacob Faibussowitsch   PetscCall(PetscOptionsHasName(((PetscObject)tao)->options, ((PetscObject)tao)->prefix, "-tao_ls_type", &flg));
394a7e14dcfSSatish Balay 
395d0609cedSBarry Smith   PetscObjectOptionsBegin((PetscObject)tao);
396a7e14dcfSSatish Balay   /* Check for type from options */
3979566063dSJacob Faibussowitsch   PetscCall(PetscOptionsFList("-tao_type", "Tao Solver type", "TaoSetType", TaoList, default_type, type, 256, &flg));
398a7e14dcfSSatish Balay   if (flg) {
3999566063dSJacob Faibussowitsch     PetscCall(TaoSetType(tao, type));
400a7e14dcfSSatish Balay   } else if (!((PetscObject)tao)->type_name) {
4019566063dSJacob Faibussowitsch     PetscCall(TaoSetType(tao, default_type));
402a7e14dcfSSatish Balay   }
403a7e14dcfSSatish Balay 
40460b70c5cSStefano Zampini   /* Tao solvers do not set the prefix, set it here if not yet done
40560b70c5cSStefano Zampini      We do it after SetType since solver may have been changed */
40660b70c5cSStefano Zampini   if (tao->linesearch) {
40760b70c5cSStefano Zampini     const char *prefix;
40860b70c5cSStefano Zampini     PetscCall(TaoLineSearchGetOptionsPrefix(tao->linesearch, &prefix));
40960b70c5cSStefano Zampini     if (!prefix) PetscCall(TaoLineSearchSetOptionsPrefix(tao->linesearch, ((PetscObject)(tao))->prefix));
41060b70c5cSStefano Zampini   }
41160b70c5cSStefano Zampini 
4129566063dSJacob Faibussowitsch   PetscCall(PetscOptionsReal("-tao_catol", "Stop if constraints violations within", "TaoSetConstraintTolerances", tao->catol, &tao->catol, &flg));
4136552cf8aSJason Sarich   if (flg) tao->catol_changed = PETSC_TRUE;
4146aad120cSJose E. Roman   PetscCall(PetscOptionsReal("-tao_crtol", "Stop if relative constraint violations within", "TaoSetConstraintTolerances", tao->crtol, &tao->crtol, &flg));
4156552cf8aSJason Sarich   if (flg) tao->crtol_changed = PETSC_TRUE;
4169566063dSJacob Faibussowitsch   PetscCall(PetscOptionsReal("-tao_gatol", "Stop if norm of gradient less than", "TaoSetTolerances", tao->gatol, &tao->gatol, &flg));
4176552cf8aSJason Sarich   if (flg) tao->gatol_changed = PETSC_TRUE;
4189566063dSJacob Faibussowitsch   PetscCall(PetscOptionsReal("-tao_grtol", "Stop if norm of gradient divided by the function value is less than", "TaoSetTolerances", tao->grtol, &tao->grtol, &flg));
4196552cf8aSJason Sarich   if (flg) tao->grtol_changed = PETSC_TRUE;
4209566063dSJacob 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));
4216552cf8aSJason Sarich   if (flg) tao->gttol_changed = PETSC_TRUE;
4229566063dSJacob Faibussowitsch   PetscCall(PetscOptionsInt("-tao_max_it", "Stop if iteration number exceeds", "TaoSetMaximumIterations", tao->max_it, &tao->max_it, &flg));
4236552cf8aSJason Sarich   if (flg) tao->max_it_changed = PETSC_TRUE;
4249566063dSJacob Faibussowitsch   PetscCall(PetscOptionsInt("-tao_max_funcs", "Stop if number of function evaluations exceeds", "TaoSetMaximumFunctionEvaluations", tao->max_funcs, &tao->max_funcs, &flg));
4256552cf8aSJason Sarich   if (flg) tao->max_funcs_changed = PETSC_TRUE;
4269566063dSJacob Faibussowitsch   PetscCall(PetscOptionsReal("-tao_fmin", "Stop if function less than", "TaoSetFunctionLowerBound", tao->fmin, &tao->fmin, &flg));
4276552cf8aSJason Sarich   if (flg) tao->fmin_changed = PETSC_TRUE;
4289566063dSJacob Faibussowitsch   PetscCall(PetscOptionsReal("-tao_steptol", "Stop if step size or trust region radius less than", "", tao->steptol, &tao->steptol, &flg));
4296552cf8aSJason Sarich   if (flg) tao->steptol_changed = PETSC_TRUE;
4309566063dSJacob Faibussowitsch   PetscCall(PetscOptionsReal("-tao_trust0", "Initial trust region radius", "TaoSetTrustRegionRadius", tao->trust0, &tao->trust0, &flg));
4316552cf8aSJason Sarich   if (flg) tao->trust0_changed = PETSC_TRUE;
4329566063dSJacob Faibussowitsch   PetscCall(PetscOptionsString("-tao_view_solution", "view solution 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, TaoSolutionMonitor, monviewer, (PetscErrorCode(*)(void **))PetscViewerDestroy));
436a7e14dcfSSatish Balay   }
437a7e14dcfSSatish Balay 
43865ba42b6SBarry Smith   PetscCall(PetscOptionsBool("-tao_converged_reason", "Print reason for Tao converged", "TaoSolve", tao->printreason, &tao->printreason, NULL));
4399566063dSJacob Faibussowitsch   PetscCall(PetscOptionsString("-tao_view_gradient", "view gradient vector after each evaluation", "TaoSetMonitor", "stdout", monfilename, sizeof(monfilename), &flg));
440a7e14dcfSSatish Balay   if (flg) {
4419566063dSJacob Faibussowitsch     PetscCall(PetscViewerASCIIOpen(comm, monfilename, &monviewer));
4429566063dSJacob Faibussowitsch     PetscCall(TaoSetMonitor(tao, TaoGradientMonitor, monviewer, (PetscErrorCode(*)(void **))PetscViewerDestroy));
443a7e14dcfSSatish Balay   }
444a7e14dcfSSatish Balay 
4459566063dSJacob Faibussowitsch   PetscCall(PetscOptionsString("-tao_view_stepdirection", "view step direction vector after each iteration", "TaoSetMonitor", "stdout", monfilename, sizeof(monfilename), &flg));
446a7e14dcfSSatish Balay   if (flg) {
4479566063dSJacob Faibussowitsch     PetscCall(PetscViewerASCIIOpen(comm, monfilename, &monviewer));
4489566063dSJacob Faibussowitsch     PetscCall(TaoSetMonitor(tao, TaoStepDirectionMonitor, monviewer, (PetscErrorCode(*)(void **))PetscViewerDestroy));
449a7e14dcfSSatish Balay   }
450a7e14dcfSSatish Balay 
4519566063dSJacob Faibussowitsch   PetscCall(PetscOptionsString("-tao_view_residual", "view least-squares residual vector after each evaluation", "TaoSetMonitor", "stdout", monfilename, sizeof(monfilename), &flg));
452a7e14dcfSSatish Balay   if (flg) {
4539566063dSJacob Faibussowitsch     PetscCall(PetscViewerASCIIOpen(comm, monfilename, &monviewer));
4549566063dSJacob Faibussowitsch     PetscCall(TaoSetMonitor(tao, TaoResidualMonitor, monviewer, (PetscErrorCode(*)(void **))PetscViewerDestroy));
455a7e14dcfSSatish Balay   }
456a7e14dcfSSatish Balay 
4579566063dSJacob Faibussowitsch   PetscCall(PetscOptionsString("-tao_monitor", "Use the default convergence monitor", "TaoSetMonitor", "stdout", monfilename, sizeof(monfilename), &flg));
458a7e14dcfSSatish Balay   if (flg) {
4599566063dSJacob Faibussowitsch     PetscCall(PetscViewerASCIIOpen(comm, monfilename, &monviewer));
4609566063dSJacob Faibussowitsch     PetscCall(TaoSetMonitor(tao, TaoMonitorDefault, monviewer, (PetscErrorCode(*)(void **))PetscViewerDestroy));
461a7e14dcfSSatish Balay   }
462a7e14dcfSSatish Balay 
4639566063dSJacob Faibussowitsch   PetscCall(PetscOptionsString("-tao_gmonitor", "Use the convergence monitor with extra globalization info", "TaoSetMonitor", "stdout", monfilename, sizeof(monfilename), &flg));
4648d5ead36SAlp Dener   if (flg) {
4659566063dSJacob Faibussowitsch     PetscCall(PetscViewerASCIIOpen(comm, monfilename, &monviewer));
4669566063dSJacob Faibussowitsch     PetscCall(TaoSetMonitor(tao, TaoDefaultGMonitor, monviewer, (PetscErrorCode(*)(void **))PetscViewerDestroy));
4678d5ead36SAlp Dener   }
4688d5ead36SAlp Dener 
4699566063dSJacob Faibussowitsch   PetscCall(PetscOptionsString("-tao_smonitor", "Use the short convergence monitor", "TaoSetMonitor", "stdout", monfilename, sizeof(monfilename), &flg));
470a7e14dcfSSatish Balay   if (flg) {
4719566063dSJacob Faibussowitsch     PetscCall(PetscViewerASCIIOpen(comm, monfilename, &monviewer));
4729566063dSJacob Faibussowitsch     PetscCall(TaoSetMonitor(tao, TaoDefaultSMonitor, monviewer, (PetscErrorCode(*)(void **))PetscViewerDestroy));
473a7e14dcfSSatish Balay   }
474a7e14dcfSSatish Balay 
4759566063dSJacob Faibussowitsch   PetscCall(PetscOptionsString("-tao_cmonitor", "Use the default convergence monitor with constraint norm", "TaoSetMonitor", "stdout", monfilename, sizeof(monfilename), &flg));
476a7e14dcfSSatish Balay   if (flg) {
4779566063dSJacob Faibussowitsch     PetscCall(PetscViewerASCIIOpen(comm, monfilename, &monviewer));
4789566063dSJacob Faibussowitsch     PetscCall(TaoSetMonitor(tao, TaoDefaultCMonitor, monviewer, (PetscErrorCode(*)(void **))PetscViewerDestroy));
479a7e14dcfSSatish Balay   }
480a7e14dcfSSatish Balay 
4818afaa268SBarry Smith   flg = PETSC_FALSE;
4829566063dSJacob Faibussowitsch   PetscCall(PetscOptionsBool("-tao_cancelmonitors", "cancel all monitors and call any registered destroy routines", "TaoCancelMonitors", flg, &flg, NULL));
4839566063dSJacob Faibussowitsch   if (flg) PetscCall(TaoCancelMonitors(tao));
484a7e14dcfSSatish Balay 
4858afaa268SBarry Smith   flg = PETSC_FALSE;
4869566063dSJacob Faibussowitsch   PetscCall(PetscOptionsBool("-tao_draw_solution", "Plot solution vector at each iteration", "TaoSetMonitor", flg, &flg, NULL));
487a7e14dcfSSatish Balay   if (flg) {
488e882e171SHong Zhang     TaoMonitorDrawCtx drawctx;
489e882e171SHong Zhang     PetscInt          howoften = 1;
4909566063dSJacob Faibussowitsch     PetscCall(TaoMonitorDrawCtxCreate(PetscObjectComm((PetscObject)tao), NULL, NULL, PETSC_DECIDE, PETSC_DECIDE, 300, 300, howoften, &drawctx));
4919566063dSJacob Faibussowitsch     PetscCall(TaoSetMonitor(tao, TaoDrawSolutionMonitor, drawctx, (PetscErrorCode(*)(void **))TaoMonitorDrawCtxDestroy));
492a7e14dcfSSatish Balay   }
493a7e14dcfSSatish Balay 
4948afaa268SBarry Smith   flg = PETSC_FALSE;
4959566063dSJacob Faibussowitsch   PetscCall(PetscOptionsBool("-tao_draw_step", "plots step direction at each iteration", "TaoSetMonitor", flg, &flg, NULL));
4961baa6e33SBarry Smith   if (flg) PetscCall(TaoSetMonitor(tao, TaoDrawStepMonitor, NULL, NULL));
497a7e14dcfSSatish Balay 
4988afaa268SBarry Smith   flg = PETSC_FALSE;
4999566063dSJacob Faibussowitsch   PetscCall(PetscOptionsBool("-tao_draw_gradient", "plots gradient at each iteration", "TaoSetMonitor", flg, &flg, NULL));
500a7e14dcfSSatish Balay   if (flg) {
501e882e171SHong Zhang     TaoMonitorDrawCtx drawctx;
502e882e171SHong Zhang     PetscInt          howoften = 1;
5039566063dSJacob Faibussowitsch     PetscCall(TaoMonitorDrawCtxCreate(PetscObjectComm((PetscObject)tao), NULL, NULL, PETSC_DECIDE, PETSC_DECIDE, 300, 300, howoften, &drawctx));
5049566063dSJacob Faibussowitsch     PetscCall(TaoSetMonitor(tao, TaoDrawGradientMonitor, drawctx, (PetscErrorCode(*)(void **))TaoMonitorDrawCtxDestroy));
505a7e14dcfSSatish Balay   }
5068afaa268SBarry Smith   flg = PETSC_FALSE;
5079566063dSJacob Faibussowitsch   PetscCall(PetscOptionsBool("-tao_fd_gradient", "compute gradient using finite differences", "TaoDefaultComputeGradient", flg, &flg, NULL));
5081baa6e33SBarry Smith   if (flg) PetscCall(TaoSetGradient(tao, NULL, TaoDefaultComputeGradient, NULL));
509f4c1ad5cSStefano Zampini   flg = PETSC_FALSE;
5109566063dSJacob Faibussowitsch   PetscCall(PetscOptionsBool("-tao_fd_hessian", "compute hessian using finite differences", "TaoDefaultComputeHessian", flg, &flg, NULL));
511f4c1ad5cSStefano Zampini   if (flg) {
512f4c1ad5cSStefano Zampini     Mat H;
513f4c1ad5cSStefano Zampini 
5149566063dSJacob Faibussowitsch     PetscCall(MatCreate(PetscObjectComm((PetscObject)tao), &H));
5159566063dSJacob Faibussowitsch     PetscCall(MatSetType(H, MATAIJ));
5169566063dSJacob Faibussowitsch     PetscCall(TaoSetHessian(tao, H, H, TaoDefaultComputeHessian, NULL));
5179566063dSJacob Faibussowitsch     PetscCall(MatDestroy(&H));
518f4c1ad5cSStefano Zampini   }
519f4c1ad5cSStefano Zampini   flg = PETSC_FALSE;
5209566063dSJacob Faibussowitsch   PetscCall(PetscOptionsBool("-tao_mf_hessian", "compute matrix-free hessian using finite differences", "TaoDefaultComputeHessianMFFD", flg, &flg, NULL));
521f4c1ad5cSStefano Zampini   if (flg) {
522f4c1ad5cSStefano Zampini     Mat H;
523f4c1ad5cSStefano Zampini 
5249566063dSJacob Faibussowitsch     PetscCall(MatCreate(PetscObjectComm((PetscObject)tao), &H));
5259566063dSJacob Faibussowitsch     PetscCall(TaoSetHessian(tao, H, H, TaoDefaultComputeHessianMFFD, NULL));
5269566063dSJacob Faibussowitsch     PetscCall(MatDestroy(&H));
527f4c1ad5cSStefano Zampini   }
528414d97d3SAlp Dener   flg = PETSC_FALSE;
5299566063dSJacob Faibussowitsch   PetscCall(PetscOptionsBool("-tao_recycle_history", "enable recycling/re-using information from the previous TaoSolve() call for some algorithms", "TaoSetRecycleHistory", flg, &flg, NULL));
5301baa6e33SBarry Smith   if (flg) PetscCall(TaoSetRecycleHistory(tao, PETSC_TRUE));
5319566063dSJacob Faibussowitsch   PetscCall(PetscOptionsEnum("-tao_subset_type", "subset type", "", TaoSubSetTypes, (PetscEnum)tao->subset_type, (PetscEnum *)&tao->subset_type, NULL));
532a7e14dcfSSatish Balay 
5330f0abf79SStefano Zampini   if (tao->ksp) {
5340f0abf79SStefano Zampini     PetscCall(PetscOptionsBool("-tao_ksp_ew", "Use Eisentat-Walker linear system convergence test", "TaoKSPSetUseEW", tao->ksp_ewconv, &tao->ksp_ewconv, NULL));
5350f0abf79SStefano Zampini     PetscCall(TaoKSPSetUseEW(tao, tao->ksp_ewconv));
5360f0abf79SStefano Zampini   }
5370f0abf79SStefano Zampini 
538dbbe0bcdSBarry Smith   PetscTryTypeMethod(tao, setfromoptions, PetscOptionsObject);
53960b70c5cSStefano Zampini 
54060b70c5cSStefano Zampini   /* process any options handlers added with PetscObjectAddOptionsHandler() */
54160b70c5cSStefano Zampini   PetscCall(PetscObjectProcessOptionsHandlers((PetscObject)tao, PetscOptionsObject));
542d0609cedSBarry Smith   PetscOptionsEnd();
54360b70c5cSStefano Zampini 
54460b70c5cSStefano Zampini   if (tao->linesearch) PetscCall(TaoLineSearchSetFromOptions(tao->linesearch));
545a7e14dcfSSatish Balay   PetscFunctionReturn(0);
546a7e14dcfSSatish Balay }
547a7e14dcfSSatish Balay 
548a7e14dcfSSatish Balay /*@C
54965ba42b6SBarry Smith    TaoViewFromOptions - View a Tao options from the options database
550fe2efc57SMark 
55165ba42b6SBarry Smith    Collective on tao
552fe2efc57SMark 
553fe2efc57SMark    Input Parameters:
554fe2efc57SMark +  A - the  Tao context
555736c3998SJose E. Roman .  obj - Optional object
556736c3998SJose E. Roman -  name - command line option
557fe2efc57SMark 
558fe2efc57SMark    Level: intermediate
559db781477SPatrick Sanan .seealso: `Tao`, `TaoView`, `PetscObjectViewFromOptions()`, `TaoCreate()`
560fe2efc57SMark @*/
561d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoViewFromOptions(Tao A, PetscObject obj, const char name[])
562d71ae5a4SJacob Faibussowitsch {
563fe2efc57SMark   PetscFunctionBegin;
564fe2efc57SMark   PetscValidHeaderSpecific(A, TAO_CLASSID, 1);
5659566063dSJacob Faibussowitsch   PetscCall(PetscObjectViewFromOptions((PetscObject)A, obj, name));
566fe2efc57SMark   PetscFunctionReturn(0);
567fe2efc57SMark }
568fe2efc57SMark 
569fe2efc57SMark /*@C
57065ba42b6SBarry Smith   TaoView - Prints information about the Tao object
571a7e14dcfSSatish Balay 
57265ba42b6SBarry Smith   Collective on tao
573a7e14dcfSSatish Balay 
574a7e14dcfSSatish Balay   InputParameters:
575441846f8SBarry Smith + tao - the Tao context
576a7e14dcfSSatish Balay - viewer - visualization context
577a7e14dcfSSatish Balay 
578a7e14dcfSSatish Balay   Options Database Key:
57965ba42b6SBarry Smith . -tao_view - Calls `TaoView()` at the end of `TaoSolve()`
580a7e14dcfSSatish Balay 
581a7e14dcfSSatish Balay   Notes:
582a7e14dcfSSatish Balay   The available visualization contexts include
58365ba42b6SBarry Smith +     `PETSC_VIEWER_STDOUT_SELF` - standard output (default)
58465ba42b6SBarry Smith -     `PETSC_VIEWER_STDOUT_WORLD` - synchronized standard
585a7e14dcfSSatish Balay          output where only the first processor opens
586a7e14dcfSSatish Balay          the file.  All other processors send their
587a7e14dcfSSatish Balay          data to the first processor to print.
588a7e14dcfSSatish Balay 
589a7e14dcfSSatish Balay   Level: beginner
590a7e14dcfSSatish Balay 
591db781477SPatrick Sanan .seealso: `PetscViewerASCIIOpen()`
592a7e14dcfSSatish Balay @*/
593d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoView(Tao tao, PetscViewer viewer)
594d71ae5a4SJacob Faibussowitsch {
595a7e14dcfSSatish Balay   PetscBool isascii, isstring;
596b625d6c7SJed Brown   TaoType   type;
59747a47007SBarry Smith 
598a7e14dcfSSatish Balay   PetscFunctionBegin;
599441846f8SBarry Smith   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
60048a46eb9SPierre Jolivet   if (!viewer) PetscCall(PetscViewerASCIIGetStdout(((PetscObject)tao)->comm, &viewer));
601a7e14dcfSSatish Balay   PetscValidHeaderSpecific(viewer, PETSC_VIEWER_CLASSID, 2);
602a7e14dcfSSatish Balay   PetscCheckSameComm(tao, 1, viewer, 2);
603a7e14dcfSSatish Balay 
6049566063dSJacob Faibussowitsch   PetscCall(PetscObjectTypeCompare((PetscObject)viewer, PETSCVIEWERASCII, &isascii));
6059566063dSJacob Faibussowitsch   PetscCall(PetscObjectTypeCompare((PetscObject)viewer, PETSCVIEWERSTRING, &isstring));
606a7e14dcfSSatish Balay   if (isascii) {
6079566063dSJacob Faibussowitsch     PetscCall(PetscObjectPrintClassNamePrefixType((PetscObject)tao, viewer));
608a7e14dcfSSatish Balay 
609a7e14dcfSSatish Balay     if (tao->ops->view) {
6109566063dSJacob Faibussowitsch       PetscCall(PetscViewerASCIIPushTab(viewer));
611dbbe0bcdSBarry Smith       PetscUseTypeMethod(tao, view, viewer);
6129566063dSJacob Faibussowitsch       PetscCall(PetscViewerASCIIPopTab(viewer));
613a7e14dcfSSatish Balay     }
6148d3f3ddaSAlp Dener     if (tao->linesearch) {
6159566063dSJacob Faibussowitsch       PetscCall(PetscViewerASCIIPushTab(viewer));
6169566063dSJacob Faibussowitsch       PetscCall(TaoLineSearchView(tao->linesearch, viewer));
6179566063dSJacob Faibussowitsch       PetscCall(PetscViewerASCIIPopTab(viewer));
6188d3f3ddaSAlp Dener     }
619a7e14dcfSSatish Balay     if (tao->ksp) {
6209566063dSJacob Faibussowitsch       PetscCall(PetscViewerASCIIPushTab(viewer));
6219566063dSJacob Faibussowitsch       PetscCall(KSPView(tao->ksp, viewer));
62263a3b9bcSJacob Faibussowitsch       PetscCall(PetscViewerASCIIPrintf(viewer, "total KSP iterations: %" PetscInt_FMT "\n", tao->ksp_tot_its));
6239566063dSJacob Faibussowitsch       PetscCall(PetscViewerASCIIPopTab(viewer));
624a7e14dcfSSatish Balay     }
62519b5c101SAlp Dener 
6269566063dSJacob Faibussowitsch     PetscCall(PetscViewerASCIIPushTab(viewer));
62719b5c101SAlp Dener 
62848a46eb9SPierre Jolivet     if (tao->XL || tao->XU) PetscCall(PetscViewerASCIIPrintf(viewer, "Active Set subset type: %s\n", TaoSubSetTypes[tao->subset_type]));
629a7e14dcfSSatish Balay 
6309566063dSJacob Faibussowitsch     PetscCall(PetscViewerASCIIPrintf(viewer, "convergence tolerances: gatol=%g,", (double)tao->gatol));
6319566063dSJacob Faibussowitsch     PetscCall(PetscViewerASCIIPrintf(viewer, " steptol=%g,", (double)tao->steptol));
6329566063dSJacob Faibussowitsch     PetscCall(PetscViewerASCIIPrintf(viewer, " gttol=%g\n", (double)tao->gttol));
6339566063dSJacob Faibussowitsch     PetscCall(PetscViewerASCIIPrintf(viewer, "Residual in Function/Gradient:=%g\n", (double)tao->residual));
634a7e14dcfSSatish Balay 
6356246e8cdSAlp Dener     if (tao->constrained) {
6369566063dSJacob Faibussowitsch       PetscCall(PetscViewerASCIIPrintf(viewer, "convergence tolerances:"));
6379566063dSJacob Faibussowitsch       PetscCall(PetscViewerASCIIPrintf(viewer, " catol=%g,", (double)tao->catol));
6389566063dSJacob Faibussowitsch       PetscCall(PetscViewerASCIIPrintf(viewer, " crtol=%g\n", (double)tao->crtol));
6399566063dSJacob Faibussowitsch       PetscCall(PetscViewerASCIIPrintf(viewer, "Residual in Constraints:=%g\n", (double)tao->cnorm));
640a7e14dcfSSatish Balay     }
641a7e14dcfSSatish Balay 
642a7e14dcfSSatish Balay     if (tao->trust < tao->steptol) {
6439566063dSJacob Faibussowitsch       PetscCall(PetscViewerASCIIPrintf(viewer, "convergence tolerances: steptol=%g\n", (double)tao->steptol));
6449566063dSJacob Faibussowitsch       PetscCall(PetscViewerASCIIPrintf(viewer, "Final trust region radius:=%g\n", (double)tao->trust));
645a7e14dcfSSatish Balay     }
646a7e14dcfSSatish Balay 
64748a46eb9SPierre Jolivet     if (tao->fmin > -1.e25) PetscCall(PetscViewerASCIIPrintf(viewer, "convergence tolerances: function minimum=%g\n", (double)tao->fmin));
6489566063dSJacob Faibussowitsch     PetscCall(PetscViewerASCIIPrintf(viewer, "Objective value=%g\n", (double)tao->fc));
649a7e14dcfSSatish Balay 
65063a3b9bcSJacob Faibussowitsch     PetscCall(PetscViewerASCIIPrintf(viewer, "total number of iterations=%" PetscInt_FMT ",          ", tao->niter));
65163a3b9bcSJacob Faibussowitsch     PetscCall(PetscViewerASCIIPrintf(viewer, "              (max: %" PetscInt_FMT ")\n", tao->max_it));
652a7e14dcfSSatish Balay 
653a7e14dcfSSatish Balay     if (tao->nfuncs > 0) {
65463a3b9bcSJacob Faibussowitsch       PetscCall(PetscViewerASCIIPrintf(viewer, "total number of function evaluations=%" PetscInt_FMT ",", tao->nfuncs));
65563a3b9bcSJacob Faibussowitsch       PetscCall(PetscViewerASCIIPrintf(viewer, "                max: %" PetscInt_FMT "\n", tao->max_funcs));
656a7e14dcfSSatish Balay     }
657a7e14dcfSSatish Balay     if (tao->ngrads > 0) {
65863a3b9bcSJacob Faibussowitsch       PetscCall(PetscViewerASCIIPrintf(viewer, "total number of gradient evaluations=%" PetscInt_FMT ",", tao->ngrads));
65963a3b9bcSJacob Faibussowitsch       PetscCall(PetscViewerASCIIPrintf(viewer, "                max: %" PetscInt_FMT "\n", tao->max_funcs));
660a7e14dcfSSatish Balay     }
661a7e14dcfSSatish Balay     if (tao->nfuncgrads > 0) {
66263a3b9bcSJacob Faibussowitsch       PetscCall(PetscViewerASCIIPrintf(viewer, "total number of function/gradient evaluations=%" PetscInt_FMT ",", tao->nfuncgrads));
66363a3b9bcSJacob Faibussowitsch       PetscCall(PetscViewerASCIIPrintf(viewer, "    (max: %" PetscInt_FMT ")\n", tao->max_funcs));
664a7e14dcfSSatish Balay     }
66548a46eb9SPierre Jolivet     if (tao->nhess > 0) PetscCall(PetscViewerASCIIPrintf(viewer, "total number of Hessian evaluations=%" PetscInt_FMT "\n", tao->nhess));
66648a46eb9SPierre Jolivet     if (tao->nconstraints > 0) PetscCall(PetscViewerASCIIPrintf(viewer, "total number of constraint function evaluations=%" PetscInt_FMT "\n", tao->nconstraints));
66748a46eb9SPierre Jolivet     if (tao->njac > 0) PetscCall(PetscViewerASCIIPrintf(viewer, "total number of Jacobian evaluations=%" PetscInt_FMT "\n", tao->njac));
668a7e14dcfSSatish Balay 
669a7e14dcfSSatish Balay     if (tao->reason > 0) {
6709566063dSJacob Faibussowitsch       PetscCall(PetscViewerASCIIPrintf(viewer, "Solution converged: "));
671a7e14dcfSSatish Balay       switch (tao->reason) {
672d71ae5a4SJacob Faibussowitsch       case TAO_CONVERGED_GATOL:
673d71ae5a4SJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, " ||g(X)|| <= gatol\n"));
674d71ae5a4SJacob Faibussowitsch         break;
675d71ae5a4SJacob Faibussowitsch       case TAO_CONVERGED_GRTOL:
676d71ae5a4SJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, " ||g(X)||/|f(X)| <= grtol\n"));
677d71ae5a4SJacob Faibussowitsch         break;
678d71ae5a4SJacob Faibussowitsch       case TAO_CONVERGED_GTTOL:
679d71ae5a4SJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, " ||g(X)||/||g(X0)|| <= gttol\n"));
680d71ae5a4SJacob Faibussowitsch         break;
681d71ae5a4SJacob Faibussowitsch       case TAO_CONVERGED_STEPTOL:
682d71ae5a4SJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, " Steptol -- step size small\n"));
683d71ae5a4SJacob Faibussowitsch         break;
684d71ae5a4SJacob Faibussowitsch       case TAO_CONVERGED_MINF:
685d71ae5a4SJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, " Minf --  f < fmin\n"));
686d71ae5a4SJacob Faibussowitsch         break;
687d71ae5a4SJacob Faibussowitsch       case TAO_CONVERGED_USER:
688d71ae5a4SJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, " User Terminated\n"));
689d71ae5a4SJacob Faibussowitsch         break;
690d71ae5a4SJacob Faibussowitsch       default:
691d71ae5a4SJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, "\n"));
692d71ae5a4SJacob Faibussowitsch         break;
693a7e14dcfSSatish Balay       }
694a7e14dcfSSatish Balay     } else {
6959566063dSJacob Faibussowitsch       PetscCall(PetscViewerASCIIPrintf(viewer, "Solver terminated: %d", tao->reason));
696a7e14dcfSSatish Balay       switch (tao->reason) {
697d71ae5a4SJacob Faibussowitsch       case TAO_DIVERGED_MAXITS:
698d71ae5a4SJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, " Maximum Iterations\n"));
699d71ae5a4SJacob Faibussowitsch         break;
700d71ae5a4SJacob Faibussowitsch       case TAO_DIVERGED_NAN:
701d71ae5a4SJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, " NAN or Inf encountered\n"));
702d71ae5a4SJacob Faibussowitsch         break;
703d71ae5a4SJacob Faibussowitsch       case TAO_DIVERGED_MAXFCN:
704d71ae5a4SJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, " Maximum Function Evaluations\n"));
705d71ae5a4SJacob Faibussowitsch         break;
706d71ae5a4SJacob Faibussowitsch       case TAO_DIVERGED_LS_FAILURE:
707d71ae5a4SJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, " Line Search Failure\n"));
708d71ae5a4SJacob Faibussowitsch         break;
709d71ae5a4SJacob Faibussowitsch       case TAO_DIVERGED_TR_REDUCTION:
710d71ae5a4SJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, " Trust Region too small\n"));
711d71ae5a4SJacob Faibussowitsch         break;
712d71ae5a4SJacob Faibussowitsch       case TAO_DIVERGED_USER:
713d71ae5a4SJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, " User Terminated\n"));
714d71ae5a4SJacob Faibussowitsch         break;
715d71ae5a4SJacob Faibussowitsch       default:
716d71ae5a4SJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer, "\n"));
717d71ae5a4SJacob Faibussowitsch         break;
718a7e14dcfSSatish Balay       }
719a7e14dcfSSatish Balay     }
7209566063dSJacob Faibussowitsch     PetscCall(PetscViewerASCIIPopTab(viewer));
721a7e14dcfSSatish Balay   } else if (isstring) {
7229566063dSJacob Faibussowitsch     PetscCall(TaoGetType(tao, &type));
7239566063dSJacob Faibussowitsch     PetscCall(PetscViewerStringSPrintf(viewer, " %-3.3s", type));
724a7e14dcfSSatish Balay   }
725a7e14dcfSSatish Balay   PetscFunctionReturn(0);
726a7e14dcfSSatish Balay }
727a7e14dcfSSatish Balay 
728a7e14dcfSSatish Balay /*@
729414d97d3SAlp Dener   TaoSetRecycleHistory - Sets the boolean flag to enable/disable re-using
73065ba42b6SBarry Smith   iterate information from the previous `TaoSolve()`. This feature is disabled by
731414d97d3SAlp Dener   default.
732414d97d3SAlp Dener 
73365ba42b6SBarry Smith   For conjugate gradient methods (`TAOBNCG`), this re-uses the latest search direction
73465ba42b6SBarry Smith   from the previous `TaoSolve()` call when computing the first search direction in a
735414d97d3SAlp Dener   new solution. By default, CG methods set the first search direction to the
736414d97d3SAlp Dener   negative gradient.
737414d97d3SAlp Dener 
73865ba42b6SBarry Smith   For quasi-Newton family of methods (`TAOBQNLS`, `TAOBQNKLS`, `TAOBQNKTR`, `TAOBQNKTL`), this re-uses
73965ba42b6SBarry Smith   the accumulated quasi-Newton Hessian approximation from the previous `TaoSolve()`
740414d97d3SAlp Dener   call. By default, QN family of methods reset the initial Hessian approximation to
741414d97d3SAlp Dener   the identity matrix.
742414d97d3SAlp Dener 
743414d97d3SAlp Dener   For any other algorithm, this setting has no effect.
744414d97d3SAlp Dener 
74565ba42b6SBarry Smith   Logically collective on tao
746414d97d3SAlp Dener 
747414d97d3SAlp Dener   Input Parameters:
748414d97d3SAlp Dener + tao - the Tao context
749414d97d3SAlp Dener - recycle - boolean flag
750414d97d3SAlp Dener 
751414d97d3SAlp Dener   Options Database Keys:
752147403d9SBarry Smith . -tao_recycle_history <true,false> - reuse the history
753414d97d3SAlp Dener 
754414d97d3SAlp Dener   Level: intermediate
755414d97d3SAlp Dener 
75665ba42b6SBarry Smith .seealso: `TaoGetRecycleHistory()`, `TAOBNCG`, `TAOBQNLS`, `TAOBQNKLS`, `TAOBQNKTR`, `TAOBQNKTL`
757414d97d3SAlp Dener 
758414d97d3SAlp Dener @*/
759d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoSetRecycleHistory(Tao tao, PetscBool recycle)
760d71ae5a4SJacob Faibussowitsch {
761414d97d3SAlp Dener   PetscFunctionBegin;
762414d97d3SAlp Dener   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
76394511df7SStefano Zampini   PetscValidLogicalCollectiveBool(tao, recycle, 2);
764414d97d3SAlp Dener   tao->recycle = recycle;
765414d97d3SAlp Dener   PetscFunctionReturn(0);
766414d97d3SAlp Dener }
767414d97d3SAlp Dener 
768414d97d3SAlp Dener /*@
769414d97d3SAlp Dener   TaoGetRecycleHistory - Retrieve the boolean flag for re-using iterate information
77065ba42b6SBarry Smith   from the previous `TaoSolve()`. This feature is disabled by default.
771414d97d3SAlp Dener 
77265ba42b6SBarry Smith   Logically collective on tao
773414d97d3SAlp Dener 
774414d97d3SAlp Dener   Input Parameters:
775147403d9SBarry Smith . tao - the Tao context
776414d97d3SAlp Dener 
777414d97d3SAlp Dener   Output Parameters:
778147403d9SBarry Smith . recycle - boolean flag
779414d97d3SAlp Dener 
780414d97d3SAlp Dener   Level: intermediate
781414d97d3SAlp Dener 
78265ba42b6SBarry Smith .seealso: `TaoSetRecycleHistory()`, `TAOBNCG`, `TAOBQNLS`, `TAOBQNKLS`, `TAOBQNKTR`, `TAOBQNKTL`
783414d97d3SAlp Dener 
784414d97d3SAlp Dener @*/
785d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoGetRecycleHistory(Tao tao, PetscBool *recycle)
786d71ae5a4SJacob Faibussowitsch {
787414d97d3SAlp Dener   PetscFunctionBegin;
788414d97d3SAlp Dener   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
78994511df7SStefano Zampini   PetscValidBoolPointer(recycle, 2);
790414d97d3SAlp Dener   *recycle = tao->recycle;
791414d97d3SAlp Dener   PetscFunctionReturn(0);
792414d97d3SAlp Dener }
793414d97d3SAlp Dener 
794414d97d3SAlp Dener /*@
79565ba42b6SBarry Smith   TaoSetTolerances - Sets parameters used in Tao convergence tests
796a7e14dcfSSatish Balay 
79765ba42b6SBarry Smith   Logically collective on tao
798a7e14dcfSSatish Balay 
799a7e14dcfSSatish Balay   Input Parameters:
800441846f8SBarry Smith + tao - the Tao context
801a7e14dcfSSatish Balay . gatol - stop if norm of gradient is less than this
802a7e14dcfSSatish Balay . grtol - stop if relative norm of gradient is less than this
803a7e14dcfSSatish Balay - gttol - stop if norm of gradient is reduced by this factor
804a7e14dcfSSatish Balay 
805a7e14dcfSSatish Balay   Options Database Keys:
806e52336cbSBarry Smith + -tao_gatol <gatol> - Sets gatol
807a7e14dcfSSatish Balay . -tao_grtol <grtol> - Sets grtol
808a7e14dcfSSatish Balay - -tao_gttol <gttol> - Sets gttol
809a7e14dcfSSatish Balay 
810a7e14dcfSSatish Balay   Stopping Criteria:
811a7e14dcfSSatish Balay $ ||g(X)||                            <= gatol
812a7e14dcfSSatish Balay $ ||g(X)|| / |f(X)|                   <= grtol
813a7e14dcfSSatish Balay $ ||g(X)|| / ||g(X0)||                <= gttol
814a7e14dcfSSatish Balay 
815a7e14dcfSSatish Balay   Notes:
816a7e14dcfSSatish Balay   Use PETSC_DEFAULT to leave one or more tolerances unchanged.
817a7e14dcfSSatish Balay 
818a7e14dcfSSatish Balay   Level: beginner
819a7e14dcfSSatish Balay 
820db781477SPatrick Sanan .seealso: `TaoGetTolerances()`
821a7e14dcfSSatish Balay 
822a7e14dcfSSatish Balay @*/
823d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoSetTolerances(Tao tao, PetscReal gatol, PetscReal grtol, PetscReal gttol)
824d71ae5a4SJacob Faibussowitsch {
825a7e14dcfSSatish Balay   PetscFunctionBegin;
826441846f8SBarry Smith   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
82794511df7SStefano Zampini   PetscValidLogicalCollectiveReal(tao, gatol, 2);
82894511df7SStefano Zampini   PetscValidLogicalCollectiveReal(tao, grtol, 3);
82994511df7SStefano Zampini   PetscValidLogicalCollectiveReal(tao, gttol, 4);
830a7e14dcfSSatish Balay 
831a7e14dcfSSatish Balay   if (gatol != PETSC_DEFAULT) {
832a7e14dcfSSatish Balay     if (gatol < 0) {
8339566063dSJacob Faibussowitsch       PetscCall(PetscInfo(tao, "Tried to set negative gatol -- ignored.\n"));
834a7e14dcfSSatish Balay     } else {
835a7e14dcfSSatish Balay       tao->gatol         = PetscMax(0, gatol);
8366552cf8aSJason Sarich       tao->gatol_changed = PETSC_TRUE;
837a7e14dcfSSatish Balay     }
838a7e14dcfSSatish Balay   }
839a7e14dcfSSatish Balay 
840a7e14dcfSSatish Balay   if (grtol != PETSC_DEFAULT) {
841a7e14dcfSSatish Balay     if (grtol < 0) {
8429566063dSJacob Faibussowitsch       PetscCall(PetscInfo(tao, "Tried to set negative grtol -- ignored.\n"));
843a7e14dcfSSatish Balay     } else {
844a7e14dcfSSatish Balay       tao->grtol         = PetscMax(0, grtol);
8456552cf8aSJason Sarich       tao->grtol_changed = PETSC_TRUE;
846a7e14dcfSSatish Balay     }
847a7e14dcfSSatish Balay   }
848a7e14dcfSSatish Balay 
849a7e14dcfSSatish Balay   if (gttol != PETSC_DEFAULT) {
850a7e14dcfSSatish Balay     if (gttol < 0) {
8519566063dSJacob Faibussowitsch       PetscCall(PetscInfo(tao, "Tried to set negative gttol -- ignored.\n"));
852a7e14dcfSSatish Balay     } else {
853a7e14dcfSSatish Balay       tao->gttol         = PetscMax(0, gttol);
8546552cf8aSJason Sarich       tao->gttol_changed = PETSC_TRUE;
855a7e14dcfSSatish Balay     }
856a7e14dcfSSatish Balay   }
857a7e14dcfSSatish Balay   PetscFunctionReturn(0);
858a7e14dcfSSatish Balay }
859a7e14dcfSSatish Balay 
860a7e14dcfSSatish Balay /*@
86165ba42b6SBarry Smith   TaoSetConstraintTolerances - Sets constraint tolerance parameters used in Tao convergence tests
862a7e14dcfSSatish Balay 
86365ba42b6SBarry Smith   Logically collective on tao
864a7e14dcfSSatish Balay 
865a7e14dcfSSatish Balay   Input Parameters:
866441846f8SBarry Smith + tao - the Tao context
867e52336cbSBarry Smith . catol - absolute constraint tolerance, constraint norm must be less than catol for used for gatol convergence criteria
8686aad120cSJose E. Roman - crtol - relative constraint tolerance, constraint norm must be less than crtol for used for gatol, gttol convergence criteria
869a7e14dcfSSatish Balay 
870a7e14dcfSSatish Balay   Options Database Keys:
871a7e14dcfSSatish Balay + -tao_catol <catol> - Sets catol
872a7e14dcfSSatish Balay - -tao_crtol <crtol> - Sets crtol
873a7e14dcfSSatish Balay 
8746552cf8aSJason Sarich   Notes:
8756552cf8aSJason Sarich   Use PETSC_DEFAULT to leave any tolerance unchanged.
8766552cf8aSJason Sarich 
877a7e14dcfSSatish Balay   Level: intermediate
878a7e14dcfSSatish Balay 
879db781477SPatrick Sanan .seealso: `TaoGetTolerances()`, `TaoGetConstraintTolerances()`, `TaoSetTolerances()`
880a7e14dcfSSatish Balay 
881a7e14dcfSSatish Balay @*/
882d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoSetConstraintTolerances(Tao tao, PetscReal catol, PetscReal crtol)
883d71ae5a4SJacob Faibussowitsch {
884a7e14dcfSSatish Balay   PetscFunctionBegin;
885441846f8SBarry Smith   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
88694511df7SStefano Zampini   PetscValidLogicalCollectiveReal(tao, catol, 2);
88794511df7SStefano Zampini   PetscValidLogicalCollectiveReal(tao, crtol, 3);
888a7e14dcfSSatish Balay 
889a7e14dcfSSatish Balay   if (catol != PETSC_DEFAULT) {
890a7e14dcfSSatish Balay     if (catol < 0) {
8919566063dSJacob Faibussowitsch       PetscCall(PetscInfo(tao, "Tried to set negative catol -- ignored.\n"));
892a7e14dcfSSatish Balay     } else {
893a7e14dcfSSatish Balay       tao->catol         = PetscMax(0, catol);
8946552cf8aSJason Sarich       tao->catol_changed = PETSC_TRUE;
895a7e14dcfSSatish Balay     }
896a7e14dcfSSatish Balay   }
897a7e14dcfSSatish Balay 
898a7e14dcfSSatish Balay   if (crtol != PETSC_DEFAULT) {
899a7e14dcfSSatish Balay     if (crtol < 0) {
9009566063dSJacob Faibussowitsch       PetscCall(PetscInfo(tao, "Tried to set negative crtol -- ignored.\n"));
901a7e14dcfSSatish Balay     } else {
902a7e14dcfSSatish Balay       tao->crtol         = PetscMax(0, crtol);
9036552cf8aSJason Sarich       tao->crtol_changed = PETSC_TRUE;
904a7e14dcfSSatish Balay     }
905a7e14dcfSSatish Balay   }
906a7e14dcfSSatish Balay   PetscFunctionReturn(0);
907a7e14dcfSSatish Balay }
908a7e14dcfSSatish Balay 
90958417fe7SBarry Smith /*@
91065ba42b6SBarry Smith   TaoGetConstraintTolerances - Gets constraint tolerance parameters used in Tao  convergence tests
91158417fe7SBarry Smith 
91258417fe7SBarry Smith   Not ollective
91358417fe7SBarry Smith 
91458417fe7SBarry Smith   Input Parameter:
91558417fe7SBarry Smith . tao - the Tao context
91658417fe7SBarry Smith 
917d8d19677SJose E. Roman   Output Parameters:
918e52336cbSBarry Smith + catol - absolute constraint tolerance, constraint norm must be less than catol for used for gatol convergence criteria
9196aad120cSJose E. Roman - crtol - relative constraint tolerance, constraint norm must be less than crtol for used for gatol, gttol convergence criteria
92058417fe7SBarry Smith 
92158417fe7SBarry Smith   Level: intermediate
92258417fe7SBarry Smith 
923db781477SPatrick Sanan .seealso: `TaoGetTolerances()`, `TaoSetTolerances()`, `TaoSetConstraintTolerances()`
92458417fe7SBarry Smith 
92558417fe7SBarry Smith @*/
926d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoGetConstraintTolerances(Tao tao, PetscReal *catol, PetscReal *crtol)
927d71ae5a4SJacob Faibussowitsch {
92858417fe7SBarry Smith   PetscFunctionBegin;
92958417fe7SBarry Smith   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
93058417fe7SBarry Smith   if (catol) *catol = tao->catol;
93158417fe7SBarry Smith   if (crtol) *crtol = tao->crtol;
93258417fe7SBarry Smith   PetscFunctionReturn(0);
93358417fe7SBarry Smith }
93458417fe7SBarry Smith 
935a7e14dcfSSatish Balay /*@
936a7e14dcfSSatish Balay    TaoSetFunctionLowerBound - Sets a bound on the solution objective value.
937a7e14dcfSSatish Balay    When an approximate solution with an objective value below this number
938a7e14dcfSSatish Balay    has been found, the solver will terminate.
939a7e14dcfSSatish Balay 
94065ba42b6SBarry Smith    Logically Collective on tao
941a7e14dcfSSatish Balay 
942a7e14dcfSSatish Balay    Input Parameters:
943441846f8SBarry Smith +  tao - the Tao solver context
944a7e14dcfSSatish Balay -  fmin - the tolerance
945a7e14dcfSSatish Balay 
946a7e14dcfSSatish Balay    Options Database Keys:
947a7e14dcfSSatish Balay .    -tao_fmin <fmin> - sets the minimum function value
948a7e14dcfSSatish Balay 
949a7e14dcfSSatish Balay    Level: intermediate
950a7e14dcfSSatish Balay 
951db781477SPatrick Sanan .seealso: `TaoSetTolerances()`
952a7e14dcfSSatish Balay @*/
953d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoSetFunctionLowerBound(Tao tao, PetscReal fmin)
954d71ae5a4SJacob Faibussowitsch {
955a7e14dcfSSatish Balay   PetscFunctionBegin;
956441846f8SBarry Smith   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
95794511df7SStefano Zampini   PetscValidLogicalCollectiveReal(tao, fmin, 2);
958a7e14dcfSSatish Balay   tao->fmin         = fmin;
9596552cf8aSJason Sarich   tao->fmin_changed = PETSC_TRUE;
960a7e14dcfSSatish Balay   PetscFunctionReturn(0);
961a7e14dcfSSatish Balay }
962a7e14dcfSSatish Balay 
963a7e14dcfSSatish Balay /*@
9648e96d397SJason Sarich    TaoGetFunctionLowerBound - Gets the bound on the solution objective value.
965a7e14dcfSSatish Balay    When an approximate solution with an objective value below this number
966a7e14dcfSSatish Balay    has been found, the solver will terminate.
967a7e14dcfSSatish Balay 
96865ba42b6SBarry Smith    Not collective on tao
969a7e14dcfSSatish Balay 
970a7e14dcfSSatish Balay    Input Parameters:
971441846f8SBarry Smith .  tao - the Tao solver context
972a7e14dcfSSatish Balay 
973a7e14dcfSSatish Balay    OutputParameters:
974a7e14dcfSSatish Balay .  fmin - the minimum function value
975a7e14dcfSSatish Balay 
976a7e14dcfSSatish Balay    Level: intermediate
977a7e14dcfSSatish Balay 
978db781477SPatrick Sanan .seealso: `TaoSetFunctionLowerBound()`
979a7e14dcfSSatish Balay @*/
980d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoGetFunctionLowerBound(Tao tao, PetscReal *fmin)
981d71ae5a4SJacob Faibussowitsch {
982a7e14dcfSSatish Balay   PetscFunctionBegin;
983441846f8SBarry Smith   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
98494511df7SStefano Zampini   PetscValidRealPointer(fmin, 2);
985a7e14dcfSSatish Balay   *fmin = tao->fmin;
986a7e14dcfSSatish Balay   PetscFunctionReturn(0);
987a7e14dcfSSatish Balay }
988a7e14dcfSSatish Balay 
989a7e14dcfSSatish Balay /*@
990a7e14dcfSSatish Balay    TaoSetMaximumFunctionEvaluations - Sets a maximum number of
991a7e14dcfSSatish Balay    function evaluations.
992a7e14dcfSSatish Balay 
99365ba42b6SBarry Smith    Logically Collective on tao
994a7e14dcfSSatish Balay 
995a7e14dcfSSatish Balay    Input Parameters:
996441846f8SBarry Smith +  tao - the Tao solver context
997a7e14dcfSSatish Balay -  nfcn - the maximum number of function evaluations (>=0)
998a7e14dcfSSatish Balay 
999a7e14dcfSSatish Balay    Options Database Keys:
1000a7e14dcfSSatish Balay .    -tao_max_funcs <nfcn> - sets the maximum number of function evaluations
1001a7e14dcfSSatish Balay 
1002a7e14dcfSSatish Balay    Level: intermediate
1003a7e14dcfSSatish Balay 
1004db781477SPatrick Sanan .seealso: `TaoSetTolerances()`, `TaoSetMaximumIterations()`
1005a7e14dcfSSatish Balay @*/
1006a7e14dcfSSatish Balay 
1007d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoSetMaximumFunctionEvaluations(Tao tao, PetscInt nfcn)
1008d71ae5a4SJacob Faibussowitsch {
1009a7e14dcfSSatish Balay   PetscFunctionBegin;
1010441846f8SBarry Smith   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
101194511df7SStefano Zampini   PetscValidLogicalCollectiveInt(tao, nfcn, 2);
10129371c9d4SSatish Balay   if (nfcn >= 0) {
10139371c9d4SSatish Balay     tao->max_funcs = PetscMax(0, nfcn);
10149371c9d4SSatish Balay   } else {
10159371c9d4SSatish Balay     tao->max_funcs = -1;
10169371c9d4SSatish Balay   }
10176552cf8aSJason Sarich   tao->max_funcs_changed = PETSC_TRUE;
1018a7e14dcfSSatish Balay   PetscFunctionReturn(0);
1019a7e14dcfSSatish Balay }
1020a7e14dcfSSatish Balay 
1021a7e14dcfSSatish Balay /*@
102265ba42b6SBarry Smith    TaoGetMaximumFunctionEvaluations - Gets a maximum number of
1023a7e14dcfSSatish Balay    function evaluations.
1024a7e14dcfSSatish Balay 
102565ba42b6SBarry Smith    Logically Collective on tao
1026a7e14dcfSSatish Balay 
1027a7e14dcfSSatish Balay    Input Parameters:
1028441846f8SBarry Smith .  tao - the Tao solver context
1029a7e14dcfSSatish Balay 
1030a7e14dcfSSatish Balay    Output Parameters:
1031a7e14dcfSSatish Balay .  nfcn - the maximum number of function evaluations
1032a7e14dcfSSatish Balay 
1033a7e14dcfSSatish Balay    Level: intermediate
1034a7e14dcfSSatish Balay 
1035db781477SPatrick Sanan .seealso: `TaoSetMaximumFunctionEvaluations()`, `TaoGetMaximumIterations()`
1036a7e14dcfSSatish Balay @*/
1037a7e14dcfSSatish Balay 
1038d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoGetMaximumFunctionEvaluations(Tao tao, PetscInt *nfcn)
1039d71ae5a4SJacob Faibussowitsch {
1040a7e14dcfSSatish Balay   PetscFunctionBegin;
1041441846f8SBarry Smith   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
104294511df7SStefano Zampini   PetscValidIntPointer(nfcn, 2);
1043a7e14dcfSSatish Balay   *nfcn = tao->max_funcs;
1044a7e14dcfSSatish Balay   PetscFunctionReturn(0);
1045a7e14dcfSSatish Balay }
1046a7e14dcfSSatish Balay 
1047770232b9SCe Qin /*@
1048770232b9SCe Qin    TaoGetCurrentFunctionEvaluations - Get current number of
1049770232b9SCe Qin    function evaluations.
1050770232b9SCe Qin 
1051770232b9SCe Qin    Not Collective
1052770232b9SCe Qin 
1053770232b9SCe Qin    Input Parameters:
1054770232b9SCe Qin .  tao - the Tao solver context
1055770232b9SCe Qin 
1056770232b9SCe Qin    Output Parameters:
105794511df7SStefano Zampini .  nfuncs - the current number of function evaluations (maximum between gradient and function evaluations)
1058770232b9SCe Qin 
1059770232b9SCe Qin    Level: intermediate
1060770232b9SCe Qin 
1061db781477SPatrick Sanan .seealso: `TaoSetMaximumFunctionEvaluations()`, `TaoGetMaximumFunctionEvaluations()`, `TaoGetMaximumIterations()`
1062770232b9SCe Qin @*/
1063770232b9SCe Qin 
1064d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoGetCurrentFunctionEvaluations(Tao tao, PetscInt *nfuncs)
1065d71ae5a4SJacob Faibussowitsch {
1066770232b9SCe Qin   PetscFunctionBegin;
1067770232b9SCe Qin   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
106894511df7SStefano Zampini   PetscValidIntPointer(nfuncs, 2);
1069770232b9SCe Qin   *nfuncs = PetscMax(tao->nfuncs, tao->nfuncgrads);
1070770232b9SCe Qin   PetscFunctionReturn(0);
1071770232b9SCe Qin }
1072770232b9SCe Qin 
1073a7e14dcfSSatish Balay /*@
1074a7e14dcfSSatish Balay    TaoSetMaximumIterations - Sets a maximum number of iterates.
1075a7e14dcfSSatish Balay 
107665ba42b6SBarry Smith    Logically Collective on tao
1077a7e14dcfSSatish Balay 
1078a7e14dcfSSatish Balay    Input Parameters:
1079441846f8SBarry Smith +  tao - the Tao solver context
1080a7e14dcfSSatish Balay -  maxits - the maximum number of iterates (>=0)
1081a7e14dcfSSatish Balay 
1082a7e14dcfSSatish Balay    Options Database Keys:
1083a7e14dcfSSatish Balay .    -tao_max_it <its> - sets the maximum number of iterations
1084a7e14dcfSSatish Balay 
1085a7e14dcfSSatish Balay    Level: intermediate
1086a7e14dcfSSatish Balay 
1087db781477SPatrick Sanan .seealso: `TaoSetTolerances()`, `TaoSetMaximumFunctionEvaluations()`
1088a7e14dcfSSatish Balay @*/
1089d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoSetMaximumIterations(Tao tao, PetscInt maxits)
1090d71ae5a4SJacob Faibussowitsch {
1091a7e14dcfSSatish Balay   PetscFunctionBegin;
1092441846f8SBarry Smith   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
109394511df7SStefano Zampini   PetscValidLogicalCollectiveInt(tao, maxits, 2);
109447a47007SBarry Smith   tao->max_it         = PetscMax(0, maxits);
10956552cf8aSJason Sarich   tao->max_it_changed = PETSC_TRUE;
1096a7e14dcfSSatish Balay   PetscFunctionReturn(0);
1097a7e14dcfSSatish Balay }
1098a7e14dcfSSatish Balay 
1099a7e14dcfSSatish Balay /*@
110065ba42b6SBarry Smith    TaoGetMaximumIterations - Gets a maximum number of iterates that will be used
1101a7e14dcfSSatish Balay 
1102a7e14dcfSSatish Balay    Not Collective
1103a7e14dcfSSatish Balay 
1104a7e14dcfSSatish Balay    Input Parameters:
1105441846f8SBarry Smith .  tao - the Tao solver context
1106a7e14dcfSSatish Balay 
1107a7e14dcfSSatish Balay    Output Parameters:
1108a7e14dcfSSatish Balay .  maxits - the maximum number of iterates
1109a7e14dcfSSatish Balay 
1110a7e14dcfSSatish Balay    Level: intermediate
1111a7e14dcfSSatish Balay 
1112db781477SPatrick Sanan .seealso: `TaoSetMaximumIterations()`, `TaoGetMaximumFunctionEvaluations()`
1113a7e14dcfSSatish Balay @*/
1114d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoGetMaximumIterations(Tao tao, PetscInt *maxits)
1115d71ae5a4SJacob Faibussowitsch {
1116a7e14dcfSSatish Balay   PetscFunctionBegin;
1117441846f8SBarry Smith   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
111894511df7SStefano Zampini   PetscValidIntPointer(maxits, 2);
1119a7e14dcfSSatish Balay   *maxits = tao->max_it;
1120a7e14dcfSSatish Balay   PetscFunctionReturn(0);
1121a7e14dcfSSatish Balay }
1122a7e14dcfSSatish Balay 
1123a7e14dcfSSatish Balay /*@
1124a7e14dcfSSatish Balay    TaoSetInitialTrustRegionRadius - Sets the initial trust region radius.
1125a7e14dcfSSatish Balay 
112665ba42b6SBarry Smith    Logically collective on tao
1127a7e14dcfSSatish Balay 
1128d8d19677SJose E. Roman    Input Parameters:
112965ba42b6SBarry Smith +  tao - a Tao optimization solver
1130a7e14dcfSSatish Balay -  radius - the trust region radius
1131a7e14dcfSSatish Balay 
1132a7e14dcfSSatish Balay    Level: intermediate
1133a7e14dcfSSatish Balay 
1134a7e14dcfSSatish Balay    Options Database Key:
1135a7e14dcfSSatish Balay .  -tao_trust0 <t0> - sets initial trust region radius
1136a7e14dcfSSatish Balay 
113765ba42b6SBarry Smith .seealso: `TaoGetTrustRegionRadius()`, `TaoSetTrustRegionTolerance()`, `TAONTR`
1138a7e14dcfSSatish Balay @*/
1139d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoSetInitialTrustRegionRadius(Tao tao, PetscReal radius)
1140d71ae5a4SJacob Faibussowitsch {
1141a7e14dcfSSatish Balay   PetscFunctionBegin;
1142441846f8SBarry Smith   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
114394511df7SStefano Zampini   PetscValidLogicalCollectiveReal(tao, radius, 2);
1144a7e14dcfSSatish Balay   tao->trust0         = PetscMax(0.0, radius);
11456552cf8aSJason Sarich   tao->trust0_changed = PETSC_TRUE;
1146a7e14dcfSSatish Balay   PetscFunctionReturn(0);
1147a7e14dcfSSatish Balay }
1148a7e14dcfSSatish Balay 
1149a7e14dcfSSatish Balay /*@
115065ba42b6SBarry Smith    TaoGetInitialTrustRegionRadius - Gets the initial trust region radius.
1151a7e14dcfSSatish Balay 
1152a7e14dcfSSatish Balay    Not Collective
1153a7e14dcfSSatish Balay 
1154a7e14dcfSSatish Balay    Input Parameter:
115565ba42b6SBarry Smith .  tao - a Tao optimization solver
1156a7e14dcfSSatish Balay 
1157a7e14dcfSSatish Balay    Output Parameter:
1158a7e14dcfSSatish Balay .  radius - the trust region radius
1159a7e14dcfSSatish Balay 
1160a7e14dcfSSatish Balay    Level: intermediate
1161a7e14dcfSSatish Balay 
116265ba42b6SBarry Smith .seealso: `TaoSetInitialTrustRegionRadius()`, `TaoGetCurrentTrustRegionRadius()`, `TAONTR`
1163a7e14dcfSSatish Balay @*/
1164d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoGetInitialTrustRegionRadius(Tao tao, PetscReal *radius)
1165d71ae5a4SJacob Faibussowitsch {
1166a7e14dcfSSatish Balay   PetscFunctionBegin;
1167441846f8SBarry Smith   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
116894511df7SStefano Zampini   PetscValidRealPointer(radius, 2);
1169a7e14dcfSSatish Balay   *radius = tao->trust0;
1170a7e14dcfSSatish Balay   PetscFunctionReturn(0);
1171a7e14dcfSSatish Balay }
1172a7e14dcfSSatish Balay 
1173a7e14dcfSSatish Balay /*@
1174a7e14dcfSSatish Balay    TaoGetCurrentTrustRegionRadius - Gets the current trust region radius.
1175a7e14dcfSSatish Balay 
1176a7e14dcfSSatish Balay    Not Collective
1177a7e14dcfSSatish Balay 
1178a7e14dcfSSatish Balay    Input Parameter:
117965ba42b6SBarry Smith .  tao - a Tao optimization solver
1180a7e14dcfSSatish Balay 
1181a7e14dcfSSatish Balay    Output Parameter:
1182a7e14dcfSSatish Balay .  radius - the trust region radius
1183a7e14dcfSSatish Balay 
1184a7e14dcfSSatish Balay    Level: intermediate
1185a7e14dcfSSatish Balay 
118665ba42b6SBarry Smith .seealso: `TaoSetInitialTrustRegionRadius()`, `TaoGetInitialTrustRegionRadius()`, `TAONTR`
1187a7e14dcfSSatish Balay @*/
1188d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoGetCurrentTrustRegionRadius(Tao tao, PetscReal *radius)
1189d71ae5a4SJacob Faibussowitsch {
1190a7e14dcfSSatish Balay   PetscFunctionBegin;
1191441846f8SBarry Smith   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
119294511df7SStefano Zampini   PetscValidRealPointer(radius, 2);
1193a7e14dcfSSatish Balay   *radius = tao->trust;
1194a7e14dcfSSatish Balay   PetscFunctionReturn(0);
1195a7e14dcfSSatish Balay }
1196a7e14dcfSSatish Balay 
1197a7e14dcfSSatish Balay /*@
1198a7e14dcfSSatish Balay   TaoGetTolerances - gets the current values of tolerances
1199a7e14dcfSSatish Balay 
1200a7e14dcfSSatish Balay   Not Collective
1201a7e14dcfSSatish Balay 
1202f899ff85SJose E. Roman   Input Parameter:
1203441846f8SBarry Smith . tao - the Tao context
1204a7e14dcfSSatish Balay 
1205a7e14dcfSSatish Balay   Output Parameters:
1206e52336cbSBarry Smith + gatol - stop if norm of gradient is less than this
1207a7e14dcfSSatish Balay . grtol - stop if relative norm of gradient is less than this
1208a7e14dcfSSatish Balay - gttol - stop if norm of gradient is reduced by a this factor
1209a7e14dcfSSatish Balay 
1210a7e14dcfSSatish Balay   Level: intermediate
1211*a1cb98faSBarry Smith 
1212*a1cb98faSBarry Smith   Note:
1213*a1cb98faSBarry Smith   NULL can be used as an argument if not all tolerances values are needed
1214*a1cb98faSBarry Smith 
1215*a1cb98faSBarry Smith .seealso: `Tao`, `TaoSetTolerances()`
1216a7e14dcfSSatish Balay @*/
1217d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoGetTolerances(Tao tao, PetscReal *gatol, PetscReal *grtol, PetscReal *gttol)
1218d71ae5a4SJacob Faibussowitsch {
1219a7e14dcfSSatish Balay   PetscFunctionBegin;
1220441846f8SBarry Smith   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
1221a7e14dcfSSatish Balay   if (gatol) *gatol = tao->gatol;
1222a7e14dcfSSatish Balay   if (grtol) *grtol = tao->grtol;
1223a7e14dcfSSatish Balay   if (gttol) *gttol = tao->gttol;
1224a7e14dcfSSatish Balay   PetscFunctionReturn(0);
1225a7e14dcfSSatish Balay }
1226a7e14dcfSSatish Balay 
1227a7e14dcfSSatish Balay /*@
1228a7e14dcfSSatish Balay   TaoGetKSP - Gets the linear solver used by the optimization solver.
122965ba42b6SBarry Smith   Application writers should use `TaoGetKSP()` if they need direct access
123065ba42b6SBarry Smith   to the PETSc `KSP` object.
1231a7e14dcfSSatish Balay 
1232a7e14dcfSSatish Balay   Not Collective
1233a7e14dcfSSatish Balay 
1234a7e14dcfSSatish Balay    Input Parameters:
123565ba42b6SBarry Smith .  tao - the Tao solver
1236a7e14dcfSSatish Balay 
1237a7e14dcfSSatish Balay    Output Parameters:
1238a7e14dcfSSatish Balay .  ksp - the KSP linear solver used in the optimization solver
1239a7e14dcfSSatish Balay 
1240a7e14dcfSSatish Balay    Level: intermediate
1241a7e14dcfSSatish Balay 
124265ba42b6SBarry Smith .seealso: `Tao`, `KSP`
1243a7e14dcfSSatish Balay @*/
1244d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoGetKSP(Tao tao, KSP *ksp)
1245d71ae5a4SJacob Faibussowitsch {
1246a7e14dcfSSatish Balay   PetscFunctionBegin;
124794511df7SStefano Zampini   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
124894511df7SStefano Zampini   PetscValidPointer(ksp, 2);
1249a7e14dcfSSatish Balay   *ksp = tao->ksp;
1250a7e14dcfSSatish Balay   PetscFunctionReturn(0);
1251a7e14dcfSSatish Balay }
1252a7e14dcfSSatish Balay 
1253025e9500SJason Sarich /*@
1254025e9500SJason Sarich    TaoGetLinearSolveIterations - Gets the total number of linear iterations
125565ba42b6SBarry Smith    used by the Tao solver
1256025e9500SJason Sarich 
1257025e9500SJason Sarich    Not Collective
1258025e9500SJason Sarich 
1259025e9500SJason Sarich    Input Parameter:
126065ba42b6SBarry Smith .  tao - Tao context
1261025e9500SJason Sarich 
1262025e9500SJason Sarich    Output Parameter:
1263025e9500SJason Sarich .  lits - number of linear iterations
1264025e9500SJason Sarich 
1265025e9500SJason Sarich    Notes:
1266025e9500SJason Sarich    This counter is reset to zero for each successive call to TaoSolve()
1267025e9500SJason Sarich 
1268025e9500SJason Sarich    Level: intermediate
1269025e9500SJason Sarich 
127065ba42b6SBarry Smith .seealso: `Tao`, `TaoGetKSP()`
1271025e9500SJason Sarich @*/
1272d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoGetLinearSolveIterations(Tao tao, PetscInt *lits)
1273d71ae5a4SJacob Faibussowitsch {
1274025e9500SJason Sarich   PetscFunctionBegin;
1275025e9500SJason Sarich   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
1276025e9500SJason Sarich   PetscValidIntPointer(lits, 2);
12772d9aa51bSJason Sarich   *lits = tao->ksp_tot_its;
1278025e9500SJason Sarich   PetscFunctionReturn(0);
1279025e9500SJason Sarich }
1280025e9500SJason Sarich 
1281a7e14dcfSSatish Balay /*@
1282a7e14dcfSSatish Balay   TaoGetLineSearch - Gets the line search used by the optimization solver.
128365ba42b6SBarry Smith   Application writers should use `TaoGetLineSearch()` if they need direct access
1284a7e14dcfSSatish Balay   to the TaoLineSearch object.
1285a7e14dcfSSatish Balay 
1286a7e14dcfSSatish Balay   Not Collective
1287a7e14dcfSSatish Balay 
1288a7e14dcfSSatish Balay    Input Parameters:
128965ba42b6SBarry Smith .  tao - the Tao solver
1290a7e14dcfSSatish Balay 
1291a7e14dcfSSatish Balay    Output Parameters:
1292a7e14dcfSSatish Balay .  ls - the line search used in the optimization solver
1293a7e14dcfSSatish Balay 
1294a7e14dcfSSatish Balay    Level: intermediate
1295a7e14dcfSSatish Balay 
1296a7e14dcfSSatish Balay @*/
1297d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoGetLineSearch(Tao tao, TaoLineSearch *ls)
1298d71ae5a4SJacob Faibussowitsch {
1299a7e14dcfSSatish Balay   PetscFunctionBegin;
130094511df7SStefano Zampini   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
130194511df7SStefano Zampini   PetscValidPointer(ls, 2);
1302a7e14dcfSSatish Balay   *ls = tao->linesearch;
1303a7e14dcfSSatish Balay   PetscFunctionReturn(0);
1304a7e14dcfSSatish Balay }
1305a7e14dcfSSatish Balay 
1306a7e14dcfSSatish Balay /*@
1307a7e14dcfSSatish Balay   TaoAddLineSearchCounts - Adds the number of function evaluations spent
1308a7e14dcfSSatish Balay   in the line search to the running total.
1309a7e14dcfSSatish Balay 
1310a7e14dcfSSatish Balay    Input Parameters:
131165ba42b6SBarry Smith +  tao - the Tao solver
1312a7e14dcfSSatish Balay -  ls - the line search used in the optimization solver
1313a7e14dcfSSatish Balay 
1314a7e14dcfSSatish Balay    Level: developer
1315a7e14dcfSSatish Balay 
131665ba42b6SBarry Smith .seealso: `TaoGetLineSearch()`, `TaoLineSearchApply()`
1317a7e14dcfSSatish Balay @*/
1318d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoAddLineSearchCounts(Tao tao)
1319d71ae5a4SJacob Faibussowitsch {
1320a7e14dcfSSatish Balay   PetscBool flg;
1321a7e14dcfSSatish Balay   PetscInt  nfeval, ngeval, nfgeval;
132247a47007SBarry Smith 
1323a7e14dcfSSatish Balay   PetscFunctionBegin;
1324441846f8SBarry Smith   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
1325a7e14dcfSSatish Balay   if (tao->linesearch) {
13269566063dSJacob Faibussowitsch     PetscCall(TaoLineSearchIsUsingTaoRoutines(tao->linesearch, &flg));
132794ae4db5SBarry Smith     if (!flg) {
13289566063dSJacob Faibussowitsch       PetscCall(TaoLineSearchGetNumberFunctionEvaluations(tao->linesearch, &nfeval, &ngeval, &nfgeval));
1329a7e14dcfSSatish Balay       tao->nfuncs += nfeval;
1330a7e14dcfSSatish Balay       tao->ngrads += ngeval;
1331a7e14dcfSSatish Balay       tao->nfuncgrads += nfgeval;
1332a7e14dcfSSatish Balay     }
1333a7e14dcfSSatish Balay   }
1334a7e14dcfSSatish Balay   PetscFunctionReturn(0);
1335a7e14dcfSSatish Balay }
1336a7e14dcfSSatish Balay 
1337a7e14dcfSSatish Balay /*@
133865ba42b6SBarry Smith   TaoGetSolution - Returns the vector with the current Tao solution
1339a7e14dcfSSatish Balay 
1340a7e14dcfSSatish Balay   Not Collective
1341a7e14dcfSSatish Balay 
1342a7e14dcfSSatish Balay   Input Parameter:
1343441846f8SBarry Smith . tao - the Tao context
1344a7e14dcfSSatish Balay 
1345a7e14dcfSSatish Balay   Output Parameter:
1346a7e14dcfSSatish Balay . X - the current solution
1347a7e14dcfSSatish Balay 
1348a7e14dcfSSatish Balay   Level: intermediate
1349a7e14dcfSSatish Balay 
135065ba42b6SBarry Smith   Note:
135165ba42b6SBarry Smith   The returned vector will be the same object that was passed into `TaoSetSolution()`
135265ba42b6SBarry Smith 
135365ba42b6SBarry Smith .seealso: `Tao`, `TaoSetSolution()`, `TaoSolve()`
1354a7e14dcfSSatish Balay @*/
1355d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoGetSolution(Tao tao, Vec *X)
1356d71ae5a4SJacob Faibussowitsch {
1357a7e14dcfSSatish Balay   PetscFunctionBegin;
1358441846f8SBarry Smith   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
135994511df7SStefano Zampini   PetscValidPointer(X, 2);
1360a7e14dcfSSatish Balay   *X = tao->solution;
1361a7e14dcfSSatish Balay   PetscFunctionReturn(0);
1362a7e14dcfSSatish Balay }
1363a7e14dcfSSatish Balay 
1364a7e14dcfSSatish Balay /*@
136565ba42b6SBarry Smith    TaoResetStatistics - Initialize the statistics used by Tao for all of the solvers.
1366a7e14dcfSSatish Balay    These statistics include the iteration number, residual norms, and convergence status.
1367a7e14dcfSSatish Balay    This routine gets called before solving each optimization problem.
1368a7e14dcfSSatish Balay 
136965ba42b6SBarry Smith    Collective on tao
1370a7e14dcfSSatish Balay 
1371a7e14dcfSSatish Balay    Input Parameters:
1372441846f8SBarry Smith .  solver - the Tao context
1373a7e14dcfSSatish Balay 
1374a7e14dcfSSatish Balay    Level: developer
1375a7e14dcfSSatish Balay 
137665ba42b6SBarry Smith .seealso: `Tao`, `TaoCreate()`, `TaoSolve()`
1377a7e14dcfSSatish Balay @*/
1378d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoResetStatistics(Tao tao)
1379d71ae5a4SJacob Faibussowitsch {
1380a7e14dcfSSatish Balay   PetscFunctionBegin;
1381441846f8SBarry Smith   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
1382a7e14dcfSSatish Balay   tao->niter        = 0;
1383a7e14dcfSSatish Balay   tao->nfuncs       = 0;
1384a7e14dcfSSatish Balay   tao->nfuncgrads   = 0;
1385a7e14dcfSSatish Balay   tao->ngrads       = 0;
1386a7e14dcfSSatish Balay   tao->nhess        = 0;
1387a7e14dcfSSatish Balay   tao->njac         = 0;
1388a7e14dcfSSatish Balay   tao->nconstraints = 0;
1389a7e14dcfSSatish Balay   tao->ksp_its      = 0;
1390ae93cb3cSJason Sarich   tao->ksp_tot_its  = 0;
1391a7e14dcfSSatish Balay   tao->reason       = TAO_CONTINUE_ITERATING;
1392a7e14dcfSSatish Balay   tao->residual     = 0.0;
1393a7e14dcfSSatish Balay   tao->cnorm        = 0.0;
1394a7e14dcfSSatish Balay   tao->step         = 0.0;
1395a7e14dcfSSatish Balay   tao->lsflag       = PETSC_FALSE;
1396a7e14dcfSSatish Balay   if (tao->hist_reset) tao->hist_len = 0;
1397a7e14dcfSSatish Balay   PetscFunctionReturn(0);
1398a7e14dcfSSatish Balay }
1399a7e14dcfSSatish Balay 
1400a7e14dcfSSatish Balay /*@C
1401e1e80dc8SAlp Dener   TaoSetUpdate - Sets the general-purpose update function called
140265ba42b6SBarry Smith   at the beginning of every iteration of the optimization algorithm. Specifically
1403e1e80dc8SAlp Dener   it is called at the top of every iteration, after the new solution and the gradient
1404e1e80dc8SAlp Dener   is determined, but before the Hessian is computed (if applicable).
1405e1e80dc8SAlp Dener 
140665ba42b6SBarry Smith   Logically Collective on tao
1407e1e80dc8SAlp Dener 
1408e1e80dc8SAlp Dener   Input Parameters:
1409a2b725a8SWilliam Gropp + tao - The tao solver context
1410a2b725a8SWilliam Gropp - func - The function
1411e1e80dc8SAlp Dener 
1412e1e80dc8SAlp Dener   Calling sequence of func:
1413a2b725a8SWilliam Gropp $ func (Tao tao, PetscInt step);
1414e1e80dc8SAlp Dener 
1415e1e80dc8SAlp Dener . step - The current step of the iteration
1416e1e80dc8SAlp Dener 
1417e1e80dc8SAlp Dener   Level: advanced
1418e1e80dc8SAlp Dener 
1419*a1cb98faSBarry Smith .seealso: `Tao`, `TaoSolve()`
1420e1e80dc8SAlp Dener @*/
1421d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoSetUpdate(Tao tao, PetscErrorCode (*func)(Tao, PetscInt, void *), void *ctx)
1422d71ae5a4SJacob Faibussowitsch {
1423e1e80dc8SAlp Dener   PetscFunctionBegin;
1424e1e80dc8SAlp Dener   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
1425e1e80dc8SAlp Dener   tao->ops->update = func;
1426e1e80dc8SAlp Dener   tao->user_update = ctx;
1427e1e80dc8SAlp Dener   PetscFunctionReturn(0);
1428e1e80dc8SAlp Dener }
1429e1e80dc8SAlp Dener 
1430e1e80dc8SAlp Dener /*@C
1431a7e14dcfSSatish Balay   TaoSetConvergenceTest - Sets the function that is to be used to test
1432a7e14dcfSSatish Balay   for convergence o fthe iterative minimization solution.  The new convergence
143365ba42b6SBarry Smith   testing routine will replace Tao's default convergence test.
1434a7e14dcfSSatish Balay 
143565ba42b6SBarry Smith   Logically Collective on tao
1436a7e14dcfSSatish Balay 
1437a7e14dcfSSatish Balay   Input Parameters:
1438441846f8SBarry Smith + tao - the Tao object
1439a7e14dcfSSatish Balay . conv - the routine to test for convergence
1440a7e14dcfSSatish Balay - ctx - [optional] context for private data for the convergence routine
144147a47007SBarry Smith         (may be NULL)
1442a7e14dcfSSatish Balay 
1443a7e14dcfSSatish Balay   Calling sequence of conv:
1444441846f8SBarry Smith $   PetscErrorCode conv(Tao tao, void *ctx)
1445a7e14dcfSSatish Balay 
1446441846f8SBarry Smith + tao - the Tao object
1447a7e14dcfSSatish Balay - ctx - [optional] convergence context
1448a7e14dcfSSatish Balay 
144965ba42b6SBarry Smith   Note:
145065ba42b6SBarry Smith   The new convergence testing routine should call `TaoSetConvergedReason()`.
1451a7e14dcfSSatish Balay 
1452a7e14dcfSSatish Balay   Level: advanced
1453a7e14dcfSSatish Balay 
145465ba42b6SBarry Smith .seealso: `Tao`, `TaoSolve()`, `TaoSetConvergedReason()`, `TaoGetSolutionStatus()`, `TaoGetTolerances()`, `TaoSetMonitor`
1455a7e14dcfSSatish Balay 
1456a7e14dcfSSatish Balay @*/
1457d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoSetConvergenceTest(Tao tao, PetscErrorCode (*conv)(Tao, void *), void *ctx)
1458d71ae5a4SJacob Faibussowitsch {
1459a7e14dcfSSatish Balay   PetscFunctionBegin;
1460441846f8SBarry Smith   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
146194511df7SStefano Zampini   tao->ops->convergencetest = conv;
146294511df7SStefano Zampini   tao->cnvP                 = ctx;
1463a7e14dcfSSatish Balay   PetscFunctionReturn(0);
1464a7e14dcfSSatish Balay }
1465a7e14dcfSSatish Balay 
1466a7e14dcfSSatish Balay /*@C
146765ba42b6SBarry Smith    TaoSetMonitor - Sets an additional function that is to be used at every
1468a7e14dcfSSatish Balay    iteration of the solver to display the iteration's
1469a7e14dcfSSatish Balay    progress.
1470a7e14dcfSSatish Balay 
147165ba42b6SBarry Smith    Logically Collective on tao
1472a7e14dcfSSatish Balay 
1473a7e14dcfSSatish Balay    Input Parameters:
1474441846f8SBarry Smith +  tao - the Tao solver context
1475a7e14dcfSSatish Balay .  mymonitor - monitoring routine
1476a7e14dcfSSatish Balay -  mctx - [optional] user-defined context for private data for the
147747a47007SBarry Smith           monitor routine (may be NULL)
1478a7e14dcfSSatish Balay 
1479a7e14dcfSSatish Balay    Calling sequence of mymonitor:
1480147403d9SBarry Smith .vb
1481147403d9SBarry Smith      PetscErrorCode mymonitor(Tao tao,void *mctx)
1482147403d9SBarry Smith .ve
1483a7e14dcfSSatish Balay 
1484441846f8SBarry Smith +    tao - the Tao solver context
1485a7e14dcfSSatish Balay -    mctx - [optional] monitoring context
1486a7e14dcfSSatish Balay 
1487a7e14dcfSSatish Balay    Options Database Keys:
148865ba42b6SBarry Smith +    -tao_monitor        - sets the default monitor `TaoMonitorDefault()`
1489a7e14dcfSSatish Balay .    -tao_smonitor       - sets short monitor
1490a7e14dcfSSatish Balay .    -tao_cmonitor       - same as smonitor plus constraint norm
1491a7e14dcfSSatish Balay .    -tao_view_solution   - view solution at each iteration
1492a7e14dcfSSatish Balay .    -tao_view_gradient   - view gradient at each iteration
14934a48860cSAlp Dener .    -tao_view_ls_residual - view least-squares residual vector at each iteration
1494a7e14dcfSSatish 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.
1495a7e14dcfSSatish Balay 
149687497f52SBarry Smith    Notes:
1497a7e14dcfSSatish Balay    Several different monitoring routines may be set by calling
149865ba42b6SBarry Smith    `TaoSetMonitor()` multiple times; all will be called in the
1499a7e14dcfSSatish Balay    order in which they were set.
1500a7e14dcfSSatish Balay 
150165ba42b6SBarry Smith    Fortran Note:
150295452b02SPatrick Sanan     Only one monitor function may be set
1503a7e14dcfSSatish Balay 
1504a7e14dcfSSatish Balay    Level: intermediate
1505a7e14dcfSSatish Balay 
150665ba42b6SBarry Smith .seealso: `Tao`, `TaoSolve()`, `TaoMonitorDefault()`, `TaoCancelMonitors()`, `TaoSetDestroyRoutine()`, `TaoView()`
1507a7e14dcfSSatish Balay @*/
1508d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoSetMonitor(Tao tao, PetscErrorCode (*func)(Tao, void *), void *ctx, PetscErrorCode (*dest)(void **))
1509d71ae5a4SJacob Faibussowitsch {
151008d19d1fSJason Sarich   PetscInt  i;
15118732526dSAlp Dener   PetscBool identical;
151208d19d1fSJason Sarich 
1513a7e14dcfSSatish Balay   PetscFunctionBegin;
1514441846f8SBarry Smith   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
15153c859ba3SBarry Smith   PetscCheck(tao->numbermonitors < MAXTAOMONITORS, PetscObjectComm((PetscObject)tao), PETSC_ERR_SUP, "Cannot attach another monitor -- max=%d", MAXTAOMONITORS);
151608d19d1fSJason Sarich 
151708d19d1fSJason Sarich   for (i = 0; i < tao->numbermonitors; i++) {
15189566063dSJacob Faibussowitsch     PetscCall(PetscMonitorCompare((PetscErrorCode(*)(void))func, ctx, dest, (PetscErrorCode(*)(void))tao->monitor[i], tao->monitorcontext[i], tao->monitordestroy[i], &identical));
15198732526dSAlp Dener     if (identical) PetscFunctionReturn(0);
152008d19d1fSJason Sarich   }
1521a7e14dcfSSatish Balay   tao->monitor[tao->numbermonitors]        = func;
15228732526dSAlp Dener   tao->monitorcontext[tao->numbermonitors] = (void *)ctx;
1523a7e14dcfSSatish Balay   tao->monitordestroy[tao->numbermonitors] = dest;
1524a7e14dcfSSatish Balay   ++tao->numbermonitors;
1525a7e14dcfSSatish Balay   PetscFunctionReturn(0);
1526a7e14dcfSSatish Balay }
1527a7e14dcfSSatish Balay 
1528a7e14dcfSSatish Balay /*@
1529441846f8SBarry Smith    TaoCancelMonitors - Clears all the monitor functions for a Tao object.
1530a7e14dcfSSatish Balay 
153165ba42b6SBarry Smith    Logically Collective on tao
1532a7e14dcfSSatish Balay 
1533a7e14dcfSSatish Balay    Input Parameters:
1534441846f8SBarry Smith .  tao - the Tao solver context
1535a7e14dcfSSatish Balay 
1536a7e14dcfSSatish Balay    Options Database:
1537a7e14dcfSSatish Balay .  -tao_cancelmonitors - cancels all monitors that have been hardwired
153865ba42b6SBarry Smith     into a code by calls to `TaoSetMonitor()`, but does not cancel those
1539a7e14dcfSSatish Balay     set via the options database
1540a7e14dcfSSatish Balay 
1541a7e14dcfSSatish Balay    Notes:
1542441846f8SBarry Smith    There is no way to clear one specific monitor from a Tao object.
1543a7e14dcfSSatish Balay 
1544a7e14dcfSSatish Balay    Level: advanced
1545a7e14dcfSSatish Balay 
154665ba42b6SBarry Smith .seealso: `Tao`, `TaoMonitorDefault()`, `TaoSetMonitor()`
1547a7e14dcfSSatish Balay @*/
1548d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoCancelMonitors(Tao tao)
1549d71ae5a4SJacob Faibussowitsch {
1550a7e14dcfSSatish Balay   PetscInt i;
155147a47007SBarry Smith 
1552a7e14dcfSSatish Balay   PetscFunctionBegin;
1553441846f8SBarry Smith   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
1554a7e14dcfSSatish Balay   for (i = 0; i < tao->numbermonitors; i++) {
155548a46eb9SPierre Jolivet     if (tao->monitordestroy[i]) PetscCall((*tao->monitordestroy[i])(&tao->monitorcontext[i]));
1556a7e14dcfSSatish Balay   }
1557a7e14dcfSSatish Balay   tao->numbermonitors = 0;
1558a7e14dcfSSatish Balay   PetscFunctionReturn(0);
1559a7e14dcfSSatish Balay }
1560a7e14dcfSSatish Balay 
15617fab98ebSJason Sarich /*@
156298ea980cSBarry Smith    TaoMonitorDefault - Default routine for monitoring progress of the
1563441846f8SBarry Smith    Tao solvers (default).  This monitor prints the function value and gradient
1564a7e14dcfSSatish Balay    norm at each iteration.  It can be turned on from the command line using the
1565a7e14dcfSSatish Balay    -tao_monitor option
1566a7e14dcfSSatish Balay 
156765ba42b6SBarry Smith    Collective on tao
1568a7e14dcfSSatish Balay 
1569a7e14dcfSSatish Balay    Input Parameters:
1570441846f8SBarry Smith +  tao - the Tao context
157165ba42b6SBarry Smith -  ctx - `PetscViewer` context or NULL
1572a7e14dcfSSatish Balay 
1573a7e14dcfSSatish Balay    Options Database Keys:
1574147403d9SBarry Smith .  -tao_monitor - turn on default monitoring
1575a7e14dcfSSatish Balay 
1576a7e14dcfSSatish Balay    Level: advanced
1577a7e14dcfSSatish Balay 
1578db781477SPatrick Sanan .seealso: `TaoDefaultSMonitor()`, `TaoSetMonitor()`
1579a7e14dcfSSatish Balay @*/
1580d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoMonitorDefault(Tao tao, void *ctx)
1581d71ae5a4SJacob Faibussowitsch {
158263b15415SAlp Dener   PetscInt    its, tabs;
1583a7e14dcfSSatish Balay   PetscReal   fct, gnorm;
15848163d661SBarry Smith   PetscViewer viewer = (PetscViewer)ctx;
158547a47007SBarry Smith 
1586a7e14dcfSSatish Balay   PetscFunctionBegin;
158794511df7SStefano Zampini   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
15888163d661SBarry Smith   PetscValidHeaderSpecific(viewer, PETSC_VIEWER_CLASSID, 2);
1589a7e14dcfSSatish Balay   its   = tao->niter;
1590a7e14dcfSSatish Balay   fct   = tao->fc;
1591a7e14dcfSSatish Balay   gnorm = tao->residual;
15929566063dSJacob Faibussowitsch   PetscCall(PetscViewerASCIIGetTab(viewer, &tabs));
15939566063dSJacob Faibussowitsch   PetscCall(PetscViewerASCIISetTab(viewer, ((PetscObject)tao)->tablevel));
1594494bef23SAlp Dener   if (its == 0 && ((PetscObject)tao)->prefix && !tao->header_printed) {
15959566063dSJacob Faibussowitsch     PetscCall(PetscViewerASCIIPrintf(viewer, "  Iteration information for %s solve.\n", ((PetscObject)tao)->prefix));
1596494bef23SAlp Dener     tao->header_printed = PETSC_TRUE;
159763b15415SAlp Dener   }
159863a3b9bcSJacob Faibussowitsch   PetscCall(PetscViewerASCIIPrintf(viewer, "%3" PetscInt_FMT " TAO,", its));
15999566063dSJacob Faibussowitsch   PetscCall(PetscViewerASCIIPrintf(viewer, "  Function value: %g,", (double)fct));
160008d19d1fSJason Sarich   if (gnorm >= PETSC_INFINITY) {
16019566063dSJacob Faibussowitsch     PetscCall(PetscViewerASCIIPrintf(viewer, "  Residual: Inf \n"));
160208d19d1fSJason Sarich   } else {
16039566063dSJacob Faibussowitsch     PetscCall(PetscViewerASCIIPrintf(viewer, "  Residual: %g \n", (double)gnorm));
160408d19d1fSJason Sarich   }
16059566063dSJacob Faibussowitsch   PetscCall(PetscViewerASCIISetTab(viewer, tabs));
1606a7e14dcfSSatish Balay   PetscFunctionReturn(0);
1607a7e14dcfSSatish Balay }
1608a7e14dcfSSatish Balay 
16097fab98ebSJason Sarich /*@
16108d5ead36SAlp Dener    TaoDefaultGMonitor - Default routine for monitoring progress of the
16118d5ead36SAlp Dener    Tao solvers (default) with extra detail on the globalization method.
16128d5ead36SAlp Dener    This monitor prints the function value and gradient norm at each
16138d5ead36SAlp Dener    iteration, as well as the step size and trust radius. Note that the
16148d5ead36SAlp Dener    step size and trust radius may be the same for some algorithms.
16158d5ead36SAlp Dener    It can be turned on from the command line using the
16168d5ead36SAlp Dener    -tao_gmonitor option
16178d5ead36SAlp Dener 
161865ba42b6SBarry Smith    Collective on tao
16198d5ead36SAlp Dener 
16208d5ead36SAlp Dener    Input Parameters:
16218d5ead36SAlp Dener +  tao - the Tao context
162265ba42b6SBarry Smith -  ctx - `PetscViewer` context or NULL
16238d5ead36SAlp Dener 
16248d5ead36SAlp Dener    Options Database Keys:
1625147403d9SBarry Smith .  -tao_gmonitor - turn on monitoring with globalization information
16268d5ead36SAlp Dener 
16278d5ead36SAlp Dener    Level: advanced
16288d5ead36SAlp Dener 
1629db781477SPatrick Sanan .seealso: `TaoDefaultSMonitor()`, `TaoSetMonitor()`
16308d5ead36SAlp Dener @*/
1631d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoDefaultGMonitor(Tao tao, void *ctx)
1632d71ae5a4SJacob Faibussowitsch {
16338d5ead36SAlp Dener   PetscInt    its, tabs;
16348d5ead36SAlp Dener   PetscReal   fct, gnorm, stp, tr;
16358d5ead36SAlp Dener   PetscViewer viewer = (PetscViewer)ctx;
16368d5ead36SAlp Dener 
16378d5ead36SAlp Dener   PetscFunctionBegin;
163894511df7SStefano Zampini   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
16398d5ead36SAlp Dener   PetscValidHeaderSpecific(viewer, PETSC_VIEWER_CLASSID, 2);
16408d5ead36SAlp Dener   its   = tao->niter;
16418d5ead36SAlp Dener   fct   = tao->fc;
16428d5ead36SAlp Dener   gnorm = tao->residual;
16438d5ead36SAlp Dener   stp   = tao->step;
16448d5ead36SAlp Dener   tr    = tao->trust;
16459566063dSJacob Faibussowitsch   PetscCall(PetscViewerASCIIGetTab(viewer, &tabs));
16469566063dSJacob Faibussowitsch   PetscCall(PetscViewerASCIISetTab(viewer, ((PetscObject)tao)->tablevel));
1647494bef23SAlp Dener   if (its == 0 && ((PetscObject)tao)->prefix && !tao->header_printed) {
16489566063dSJacob Faibussowitsch     PetscCall(PetscViewerASCIIPrintf(viewer, "  Iteration information for %s solve.\n", ((PetscObject)tao)->prefix));
1649494bef23SAlp Dener     tao->header_printed = PETSC_TRUE;
16508d5ead36SAlp Dener   }
165163a3b9bcSJacob Faibussowitsch   PetscCall(PetscViewerASCIIPrintf(viewer, "%3" PetscInt_FMT " TAO,", its));
16529566063dSJacob Faibussowitsch   PetscCall(PetscViewerASCIIPrintf(viewer, "  Function value: %g,", (double)fct));
16538d5ead36SAlp Dener   if (gnorm >= PETSC_INFINITY) {
16549566063dSJacob Faibussowitsch     PetscCall(PetscViewerASCIIPrintf(viewer, "  Residual: Inf,"));
16558d5ead36SAlp Dener   } else {
16569566063dSJacob Faibussowitsch     PetscCall(PetscViewerASCIIPrintf(viewer, "  Residual: %g,", (double)gnorm));
16578d5ead36SAlp Dener   }
16589566063dSJacob Faibussowitsch   PetscCall(PetscViewerASCIIPrintf(viewer, "  Step: %g,  Trust: %g\n", (double)stp, (double)tr));
16599566063dSJacob Faibussowitsch   PetscCall(PetscViewerASCIISetTab(viewer, tabs));
16608d5ead36SAlp Dener   PetscFunctionReturn(0);
16618d5ead36SAlp Dener }
16628d5ead36SAlp Dener 
16638d5ead36SAlp Dener /*@
1664a7e14dcfSSatish Balay    TaoDefaultSMonitor - Default routine for monitoring progress of the
166565ba42b6SBarry Smith    solver. Same as `TaoMonitorDefault()` except
1666a7e14dcfSSatish Balay    it prints fewer digits of the residual as the residual gets smaller.
1667a7e14dcfSSatish Balay    This is because the later digits are meaningless and are often
1668a7e14dcfSSatish Balay    different on different machines; by using this routine different
1669a7e14dcfSSatish Balay    machines will usually generate the same output. It can be turned on
1670a7e14dcfSSatish Balay    by using the -tao_smonitor option
1671a7e14dcfSSatish Balay 
167265ba42b6SBarry Smith    Collective on tao
1673a7e14dcfSSatish Balay 
1674a7e14dcfSSatish Balay    Input Parameters:
1675441846f8SBarry Smith +  tao - the Tao context
16764d4332d5SBarry Smith -  ctx - PetscViewer context of type ASCII
1677a7e14dcfSSatish Balay 
1678a7e14dcfSSatish Balay    Options Database Keys:
1679147403d9SBarry Smith .  -tao_smonitor - turn on default short monitoring
1680a7e14dcfSSatish Balay 
1681a7e14dcfSSatish Balay    Level: advanced
1682a7e14dcfSSatish Balay 
1683db781477SPatrick Sanan .seealso: `TaoMonitorDefault()`, `TaoSetMonitor()`
1684a7e14dcfSSatish Balay @*/
1685d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoDefaultSMonitor(Tao tao, void *ctx)
1686d71ae5a4SJacob Faibussowitsch {
16872c9b8b83SAlp Dener   PetscInt    its, tabs;
1688a7e14dcfSSatish Balay   PetscReal   fct, gnorm;
16894d4332d5SBarry Smith   PetscViewer viewer = (PetscViewer)ctx;
169047a47007SBarry Smith 
1691a7e14dcfSSatish Balay   PetscFunctionBegin;
169294511df7SStefano Zampini   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
16934d4332d5SBarry Smith   PetscValidHeaderSpecific(viewer, PETSC_VIEWER_CLASSID, 2);
1694a7e14dcfSSatish Balay   its   = tao->niter;
1695a7e14dcfSSatish Balay   fct   = tao->fc;
1696a7e14dcfSSatish Balay   gnorm = tao->residual;
16979566063dSJacob Faibussowitsch   PetscCall(PetscViewerASCIIGetTab(viewer, &tabs));
16989566063dSJacob Faibussowitsch   PetscCall(PetscViewerASCIISetTab(viewer, ((PetscObject)tao)->tablevel));
169963a3b9bcSJacob Faibussowitsch   PetscCall(PetscViewerASCIIPrintf(viewer, "iter = %3" PetscInt_FMT ",", its));
17009566063dSJacob Faibussowitsch   PetscCall(PetscViewerASCIIPrintf(viewer, " Function value %g,", (double)fct));
1701d393f493SBarry Smith   if (gnorm >= PETSC_INFINITY) {
17029566063dSJacob Faibussowitsch     PetscCall(PetscViewerASCIIPrintf(viewer, " Residual: Inf \n"));
170308d19d1fSJason Sarich   } else if (gnorm > 1.e-6) {
17049566063dSJacob Faibussowitsch     PetscCall(PetscViewerASCIIPrintf(viewer, " Residual: %g \n", (double)gnorm));
1705a7e14dcfSSatish Balay   } else if (gnorm > 1.e-11) {
17069566063dSJacob Faibussowitsch     PetscCall(PetscViewerASCIIPrintf(viewer, " Residual: < 1.0e-6 \n"));
1707a7e14dcfSSatish Balay   } else {
17089566063dSJacob Faibussowitsch     PetscCall(PetscViewerASCIIPrintf(viewer, " Residual: < 1.0e-11 \n"));
1709a7e14dcfSSatish Balay   }
17109566063dSJacob Faibussowitsch   PetscCall(PetscViewerASCIISetTab(viewer, tabs));
1711a7e14dcfSSatish Balay   PetscFunctionReturn(0);
1712a7e14dcfSSatish Balay }
1713a7e14dcfSSatish Balay 
17147fab98ebSJason Sarich /*@
171565ba42b6SBarry Smith    TaoDefaultCMonitor - same as `TaoMonitorDefault()` except
1716a7e14dcfSSatish Balay    it prints the norm of the constraints function. It can be turned on
1717a7e14dcfSSatish Balay    from the command line using the -tao_cmonitor option
1718a7e14dcfSSatish Balay 
171965ba42b6SBarry Smith    Collective on tao
1720a7e14dcfSSatish Balay 
1721a7e14dcfSSatish Balay    Input Parameters:
1722441846f8SBarry Smith +  tao - the Tao context
172365ba42b6SBarry Smith -  ctx - `PetscViewer` context or NULL
1724a7e14dcfSSatish Balay 
1725a7e14dcfSSatish Balay    Options Database Keys:
1726147403d9SBarry Smith .  -tao_cmonitor - monitor the constraints
1727a7e14dcfSSatish Balay 
1728a7e14dcfSSatish Balay    Level: advanced
1729a7e14dcfSSatish Balay 
1730db781477SPatrick Sanan .seealso: `TaoMonitorDefault()`, `TaoSetMonitor()`
1731a7e14dcfSSatish Balay @*/
1732d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoDefaultCMonitor(Tao tao, void *ctx)
1733d71ae5a4SJacob Faibussowitsch {
17342c9b8b83SAlp Dener   PetscInt    its, tabs;
1735a7e14dcfSSatish Balay   PetscReal   fct, gnorm;
1736d393f493SBarry Smith   PetscViewer viewer = (PetscViewer)ctx;
1737a7e14dcfSSatish Balay 
1738a7e14dcfSSatish Balay   PetscFunctionBegin;
173994511df7SStefano Zampini   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
1740d393f493SBarry Smith   PetscValidHeaderSpecific(viewer, PETSC_VIEWER_CLASSID, 2);
1741a7e14dcfSSatish Balay   its   = tao->niter;
1742a7e14dcfSSatish Balay   fct   = tao->fc;
1743a7e14dcfSSatish Balay   gnorm = tao->residual;
17449566063dSJacob Faibussowitsch   PetscCall(PetscViewerASCIIGetTab(viewer, &tabs));
17459566063dSJacob Faibussowitsch   PetscCall(PetscViewerASCIISetTab(viewer, ((PetscObject)tao)->tablevel));
174663a3b9bcSJacob Faibussowitsch   PetscCall(PetscViewerASCIIPrintf(viewer, "iter = %" PetscInt_FMT ",", its));
17479566063dSJacob Faibussowitsch   PetscCall(PetscViewerASCIIPrintf(viewer, " Function value: %g,", (double)fct));
17489566063dSJacob Faibussowitsch   PetscCall(PetscViewerASCIIPrintf(viewer, "  Residual: %g ", (double)gnorm));
17499566063dSJacob Faibussowitsch   PetscCall(PetscViewerASCIIPrintf(viewer, "  Constraint: %g \n", (double)tao->cnorm));
17509566063dSJacob Faibussowitsch   PetscCall(PetscViewerASCIISetTab(viewer, tabs));
1751a7e14dcfSSatish Balay   PetscFunctionReturn(0);
1752a7e14dcfSSatish Balay }
1753a7e14dcfSSatish Balay 
1754a7e14dcfSSatish Balay /*@C
1755a7e14dcfSSatish Balay    TaoSolutionMonitor - Views the solution at each iteration
1756a7e14dcfSSatish Balay    It can be turned on from the command line using the
1757a7e14dcfSSatish Balay    -tao_view_solution option
1758a7e14dcfSSatish Balay 
175965ba42b6SBarry Smith    Collective on tao
1760a7e14dcfSSatish Balay 
1761a7e14dcfSSatish Balay    Input Parameters:
1762441846f8SBarry Smith +  tao - the Tao context
176365ba42b6SBarry Smith -  ctx - `PetscViewer` context or NULL
1764a7e14dcfSSatish Balay 
1765a7e14dcfSSatish Balay    Options Database Keys:
1766147403d9SBarry Smith .  -tao_view_solution - view the solution
1767a7e14dcfSSatish Balay 
1768a7e14dcfSSatish Balay    Level: advanced
1769a7e14dcfSSatish Balay 
1770db781477SPatrick Sanan .seealso: `TaoDefaultSMonitor()`, `TaoSetMonitor()`
1771a7e14dcfSSatish Balay @*/
1772d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoSolutionMonitor(Tao tao, void *ctx)
1773d71ae5a4SJacob Faibussowitsch {
1774feb237baSPierre Jolivet   PetscViewer viewer = (PetscViewer)ctx;
177547a47007SBarry Smith 
1776a7e14dcfSSatish Balay   PetscFunctionBegin;
177794511df7SStefano Zampini   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
1778d393f493SBarry Smith   PetscValidHeaderSpecific(viewer, PETSC_VIEWER_CLASSID, 2);
17799566063dSJacob Faibussowitsch   PetscCall(VecView(tao->solution, viewer));
1780a7e14dcfSSatish Balay   PetscFunctionReturn(0);
1781a7e14dcfSSatish Balay }
1782a7e14dcfSSatish Balay 
1783a7e14dcfSSatish Balay /*@C
1784a7e14dcfSSatish Balay    TaoGradientMonitor - Views the gradient at each iteration
1785a7e14dcfSSatish Balay    It can be turned on from the command line using the
1786a7e14dcfSSatish Balay    -tao_view_gradient option
1787a7e14dcfSSatish Balay 
178865ba42b6SBarry Smith    Collective on tao
1789a7e14dcfSSatish Balay 
1790a7e14dcfSSatish Balay    Input Parameters:
1791441846f8SBarry Smith +  tao - the Tao context
179265ba42b6SBarry Smith -  ctx - `PetscViewer` context or NULL
1793a7e14dcfSSatish Balay 
1794a7e14dcfSSatish Balay    Options Database Keys:
1795147403d9SBarry Smith .  -tao_view_gradient - view the gradient at each iteration
1796a7e14dcfSSatish Balay 
1797a7e14dcfSSatish Balay    Level: advanced
1798a7e14dcfSSatish Balay 
1799db781477SPatrick Sanan .seealso: `TaoDefaultSMonitor()`, `TaoSetMonitor()`
1800a7e14dcfSSatish Balay @*/
1801d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoGradientMonitor(Tao tao, void *ctx)
1802d71ae5a4SJacob Faibussowitsch {
1803d393f493SBarry Smith   PetscViewer viewer = (PetscViewer)ctx;
180447a47007SBarry Smith 
1805a7e14dcfSSatish Balay   PetscFunctionBegin;
180694511df7SStefano Zampini   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
1807d393f493SBarry Smith   PetscValidHeaderSpecific(viewer, PETSC_VIEWER_CLASSID, 2);
18089566063dSJacob Faibussowitsch   PetscCall(VecView(tao->gradient, viewer));
1809a7e14dcfSSatish Balay   PetscFunctionReturn(0);
1810a7e14dcfSSatish Balay }
1811a7e14dcfSSatish Balay 
1812a7e14dcfSSatish Balay /*@C
1813147403d9SBarry Smith    TaoStepDirectionMonitor - Views the step-direction at each iteration
1814a7e14dcfSSatish Balay 
181565ba42b6SBarry Smith    Collective on tao
1816a7e14dcfSSatish Balay 
1817a7e14dcfSSatish Balay    Input Parameters:
1818441846f8SBarry Smith +  tao - the Tao context
181965ba42b6SBarry Smith -  ctx - `PetscViewer` context or NULL
1820a7e14dcfSSatish Balay 
1821a7e14dcfSSatish Balay    Options Database Keys:
1822147403d9SBarry Smith .  -tao_view_gradient - view the gradient at each iteration
1823a7e14dcfSSatish Balay 
1824a7e14dcfSSatish Balay    Level: advanced
1825a7e14dcfSSatish Balay 
1826db781477SPatrick Sanan .seealso: `TaoDefaultSMonitor()`, `TaoSetMonitor()`
1827a7e14dcfSSatish Balay @*/
1828d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoStepDirectionMonitor(Tao tao, void *ctx)
1829d71ae5a4SJacob Faibussowitsch {
1830d393f493SBarry Smith   PetscViewer viewer = (PetscViewer)ctx;
1831d393f493SBarry Smith 
1832a7e14dcfSSatish Balay   PetscFunctionBegin;
183394511df7SStefano Zampini   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
1834d393f493SBarry Smith   PetscValidHeaderSpecific(viewer, PETSC_VIEWER_CLASSID, 2);
18359566063dSJacob Faibussowitsch   PetscCall(VecView(tao->stepdirection, viewer));
1836a7e14dcfSSatish Balay   PetscFunctionReturn(0);
1837a7e14dcfSSatish Balay }
1838a7e14dcfSSatish Balay 
1839a7e14dcfSSatish Balay /*@C
1840a7e14dcfSSatish Balay    TaoDrawSolutionMonitor - Plots the solution at each iteration
1841a7e14dcfSSatish Balay    It can be turned on from the command line using the
1842a7e14dcfSSatish Balay    -tao_draw_solution option
1843a7e14dcfSSatish Balay 
184465ba42b6SBarry Smith    Collective on tao
1845a7e14dcfSSatish Balay 
1846a7e14dcfSSatish Balay    Input Parameters:
1847441846f8SBarry Smith +  tao - the Tao context
184865ba42b6SBarry Smith -  ctx - `TaoMonitorDraw` context
1849a7e14dcfSSatish Balay 
1850a7e14dcfSSatish Balay    Options Database Keys:
1851147403d9SBarry Smith .  -tao_draw_solution - draw the solution at each iteration
1852a7e14dcfSSatish Balay 
1853a7e14dcfSSatish Balay    Level: advanced
1854a7e14dcfSSatish Balay 
185565ba42b6SBarry Smith .seealso: `TaoSolutionMonitor()`, `TaoSetMonitor()`, `TaoDrawGradientMonitor`, `TaoMonitorDraw`
1856a7e14dcfSSatish Balay @*/
1857d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoDrawSolutionMonitor(Tao tao, void *ctx)
1858d71ae5a4SJacob Faibussowitsch {
1859e882e171SHong Zhang   TaoMonitorDrawCtx ictx = (TaoMonitorDrawCtx)ctx;
186047a47007SBarry Smith 
1861a7e14dcfSSatish Balay   PetscFunctionBegin;
186294511df7SStefano Zampini   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
1863e882e171SHong Zhang   if (!(((ictx->howoften > 0) && (!(tao->niter % ictx->howoften))) || ((ictx->howoften == -1) && tao->reason))) PetscFunctionReturn(0);
18649566063dSJacob Faibussowitsch   PetscCall(VecView(tao->solution, ictx->viewer));
1865a7e14dcfSSatish Balay   PetscFunctionReturn(0);
1866a7e14dcfSSatish Balay }
1867a7e14dcfSSatish Balay 
1868a7e14dcfSSatish Balay /*@C
1869a7e14dcfSSatish Balay    TaoDrawGradientMonitor - Plots the gradient at each iteration
1870a7e14dcfSSatish Balay    It can be turned on from the command line using the
1871a7e14dcfSSatish Balay    -tao_draw_gradient option
1872a7e14dcfSSatish Balay 
187365ba42b6SBarry Smith    Collective on tao
1874a7e14dcfSSatish Balay 
1875a7e14dcfSSatish Balay    Input Parameters:
1876441846f8SBarry Smith +  tao - the Tao context
187765ba42b6SBarry Smith -  ctx - `PetscViewer` context
1878a7e14dcfSSatish Balay 
1879a7e14dcfSSatish Balay    Options Database Keys:
1880147403d9SBarry Smith .  -tao_draw_gradient - draw the gradient at each iteration
1881a7e14dcfSSatish Balay 
1882a7e14dcfSSatish Balay    Level: advanced
1883a7e14dcfSSatish Balay 
1884db781477SPatrick Sanan .seealso: `TaoGradientMonitor()`, `TaoSetMonitor()`, `TaoDrawSolutionMonitor`
1885a7e14dcfSSatish Balay @*/
1886d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoDrawGradientMonitor(Tao tao, void *ctx)
1887d71ae5a4SJacob Faibussowitsch {
1888e882e171SHong Zhang   TaoMonitorDrawCtx ictx = (TaoMonitorDrawCtx)ctx;
188947a47007SBarry Smith 
1890a7e14dcfSSatish Balay   PetscFunctionBegin;
189194511df7SStefano Zampini   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
1892e882e171SHong Zhang   if (!(((ictx->howoften > 0) && (!(tao->niter % ictx->howoften))) || ((ictx->howoften == -1) && tao->reason))) PetscFunctionReturn(0);
18939566063dSJacob Faibussowitsch   PetscCall(VecView(tao->gradient, ictx->viewer));
1894a7e14dcfSSatish Balay   PetscFunctionReturn(0);
1895a7e14dcfSSatish Balay }
1896a7e14dcfSSatish Balay 
1897a7e14dcfSSatish Balay /*@C
1898a7e14dcfSSatish Balay    TaoDrawStepMonitor - Plots the step direction at each iteration
1899a7e14dcfSSatish Balay 
190065ba42b6SBarry Smith    Collective on tao
1901a7e14dcfSSatish Balay 
1902a7e14dcfSSatish Balay    Input Parameters:
1903441846f8SBarry Smith +  tao - the Tao context
1904f55353a2SBarry Smith -  ctx - PetscViewer context
1905a7e14dcfSSatish Balay 
1906a7e14dcfSSatish Balay    Options Database Keys:
1907147403d9SBarry Smith .  -tao_draw_step - draw the step direction at each iteration
1908a7e14dcfSSatish Balay 
1909a7e14dcfSSatish Balay    Level: advanced
1910a7e14dcfSSatish Balay 
1911db781477SPatrick Sanan .seealso: `TaoSetMonitor()`, `TaoDrawSolutionMonitor`
1912a7e14dcfSSatish Balay @*/
1913d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoDrawStepMonitor(Tao tao, void *ctx)
1914d71ae5a4SJacob Faibussowitsch {
191594511df7SStefano Zampini   PetscViewer viewer = (PetscViewer)ctx;
191647a47007SBarry Smith 
1917a7e14dcfSSatish Balay   PetscFunctionBegin;
191894511df7SStefano Zampini   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
191994511df7SStefano Zampini   PetscValidHeaderSpecific(viewer, PETSC_VIEWER_CLASSID, 2);
19209566063dSJacob Faibussowitsch   PetscCall(VecView(tao->stepdirection, viewer));
1921a7e14dcfSSatish Balay   PetscFunctionReturn(0);
1922a7e14dcfSSatish Balay }
1923a7e14dcfSSatish Balay 
1924a7e14dcfSSatish Balay /*@C
1925737f463aSAlp Dener    TaoResidualMonitor - Views the least-squares residual at each iteration
1926a7e14dcfSSatish Balay 
192765ba42b6SBarry Smith    Collective on tao
1928a7e14dcfSSatish Balay 
1929a7e14dcfSSatish Balay    Input Parameters:
1930441846f8SBarry Smith +  tao - the Tao context
193165ba42b6SBarry Smith -  ctx - `PetscViewer` context or NULL
1932a7e14dcfSSatish Balay 
1933a7e14dcfSSatish Balay    Options Database Keys:
1934147403d9SBarry Smith .  -tao_view_ls_residual - view the least-squares residual at each iteration
1935a7e14dcfSSatish Balay 
1936a7e14dcfSSatish Balay    Level: advanced
1937a7e14dcfSSatish Balay 
1938db781477SPatrick Sanan .seealso: `TaoDefaultSMonitor()`, `TaoSetMonitor()`
1939a7e14dcfSSatish Balay @*/
1940d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoResidualMonitor(Tao tao, void *ctx)
1941d71ae5a4SJacob Faibussowitsch {
1942d393f493SBarry Smith   PetscViewer viewer = (PetscViewer)ctx;
194347a47007SBarry Smith 
1944a7e14dcfSSatish Balay   PetscFunctionBegin;
194594511df7SStefano Zampini   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
1946d393f493SBarry Smith   PetscValidHeaderSpecific(viewer, PETSC_VIEWER_CLASSID, 2);
19479566063dSJacob Faibussowitsch   PetscCall(VecView(tao->ls_res, viewer));
1948a7e14dcfSSatish Balay   PetscFunctionReturn(0);
1949a7e14dcfSSatish Balay }
1950a7e14dcfSSatish Balay 
19517fab98ebSJason Sarich /*@
1952a7e14dcfSSatish Balay    TaoDefaultConvergenceTest - Determines whether the solver should continue iterating
1953a7e14dcfSSatish Balay    or terminate.
1954a7e14dcfSSatish Balay 
195565ba42b6SBarry Smith    Collective on tao
1956a7e14dcfSSatish Balay 
1957a7e14dcfSSatish Balay    Input Parameters:
1958441846f8SBarry Smith +  tao - the Tao context
1959a7e14dcfSSatish Balay -  dummy - unused dummy context
1960a7e14dcfSSatish Balay 
1961a7e14dcfSSatish Balay    Output Parameter:
1962a7e14dcfSSatish Balay .  reason - for terminating
1963a7e14dcfSSatish Balay 
1964a7e14dcfSSatish Balay    Notes:
1965a7e14dcfSSatish Balay    This routine checks the residual in the optimality conditions, the
1966a7e14dcfSSatish Balay    relative residual in the optimity conditions, the number of function
1967a7e14dcfSSatish Balay    evaluations, and the function value to test convergence.  Some
1968a7e14dcfSSatish Balay    solvers may use different convergence routines.
1969a7e14dcfSSatish Balay 
1970a7e14dcfSSatish Balay    Level: developer
1971a7e14dcfSSatish Balay 
1972c2e3fba1SPatrick Sanan .seealso: `TaoSetTolerances()`, `TaoGetConvergedReason()`, `TaoSetConvergedReason()`
1973a7e14dcfSSatish Balay @*/
1974a7e14dcfSSatish Balay 
1975d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoDefaultConvergenceTest(Tao tao, void *dummy)
1976d71ae5a4SJacob Faibussowitsch {
1977a7e14dcfSSatish Balay   PetscInt           niter = tao->niter, nfuncs = PetscMax(tao->nfuncs, tao->nfuncgrads);
1978a7e14dcfSSatish Balay   PetscInt           max_funcs = tao->max_funcs;
1979a7e14dcfSSatish Balay   PetscReal          gnorm = tao->residual, gnorm0 = tao->gnorm0;
1980a7e14dcfSSatish Balay   PetscReal          f = tao->fc, steptol = tao->steptol, trradius = tao->step;
1981a7e14dcfSSatish Balay   PetscReal          gatol = tao->gatol, grtol = tao->grtol, gttol = tao->gttol;
1982e52336cbSBarry Smith   PetscReal          catol = tao->catol, crtol = tao->crtol;
1983e52336cbSBarry Smith   PetscReal          fmin = tao->fmin, cnorm = tao->cnorm;
1984e4cb33bbSBarry Smith   TaoConvergedReason reason = tao->reason;
1985a7e14dcfSSatish Balay 
1986a7e14dcfSSatish Balay   PetscFunctionBegin;
1987441846f8SBarry Smith   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
198894511df7SStefano Zampini   if (reason != TAO_CONTINUE_ITERATING) PetscFunctionReturn(0);
1989a7e14dcfSSatish Balay 
1990a7e14dcfSSatish Balay   if (PetscIsInfOrNanReal(f)) {
19919566063dSJacob Faibussowitsch     PetscCall(PetscInfo(tao, "Failed to converged, function value is Inf or NaN\n"));
1992a7e14dcfSSatish Balay     reason = TAO_DIVERGED_NAN;
1993a7e14dcfSSatish Balay   } else if (f <= fmin && cnorm <= catol) {
19949566063dSJacob Faibussowitsch     PetscCall(PetscInfo(tao, "Converged due to function value %g < minimum function value %g\n", (double)f, (double)fmin));
1995a7e14dcfSSatish Balay     reason = TAO_CONVERGED_MINF;
1996a7e14dcfSSatish Balay   } else if (gnorm <= gatol && cnorm <= catol) {
19979566063dSJacob Faibussowitsch     PetscCall(PetscInfo(tao, "Converged due to residual norm ||g(X)||=%g < %g\n", (double)gnorm, (double)gatol));
1998a7e14dcfSSatish Balay     reason = TAO_CONVERGED_GATOL;
1999a7e14dcfSSatish Balay   } else if (f != 0 && PetscAbsReal(gnorm / f) <= grtol && cnorm <= crtol) {
20009566063dSJacob Faibussowitsch     PetscCall(PetscInfo(tao, "Converged due to residual ||g(X)||/|f(X)| =%g < %g\n", (double)(gnorm / f), (double)grtol));
2001a7e14dcfSSatish Balay     reason = TAO_CONVERGED_GRTOL;
2002e1d16bb9SBarry Smith   } else if (gnorm0 != 0 && ((gttol == 0 && gnorm == 0) || gnorm / gnorm0 < gttol) && cnorm <= crtol) {
20039566063dSJacob Faibussowitsch     PetscCall(PetscInfo(tao, "Converged due to relative residual norm ||g(X)||/||g(X0)|| = %g < %g\n", (double)(gnorm / gnorm0), (double)gttol));
2004a7e14dcfSSatish Balay     reason = TAO_CONVERGED_GTTOL;
200594511df7SStefano Zampini   } else if (max_funcs >= 0 && nfuncs > max_funcs) {
20069566063dSJacob Faibussowitsch     PetscCall(PetscInfo(tao, "Exceeded maximum number of function evaluations: %" PetscInt_FMT " > %" PetscInt_FMT "\n", nfuncs, max_funcs));
2007a7e14dcfSSatish Balay     reason = TAO_DIVERGED_MAXFCN;
2008a7e14dcfSSatish Balay   } else if (tao->lsflag != 0) {
20099566063dSJacob Faibussowitsch     PetscCall(PetscInfo(tao, "Tao Line Search failure.\n"));
2010a7e14dcfSSatish Balay     reason = TAO_DIVERGED_LS_FAILURE;
2011a7e14dcfSSatish Balay   } else if (trradius < steptol && niter > 0) {
20129566063dSJacob Faibussowitsch     PetscCall(PetscInfo(tao, "Trust region/step size too small: %g < %g\n", (double)trradius, (double)steptol));
2013a7e14dcfSSatish Balay     reason = TAO_CONVERGED_STEPTOL;
2014e031d6f5SAlp Dener   } else if (niter >= tao->max_it) {
201563a3b9bcSJacob Faibussowitsch     PetscCall(PetscInfo(tao, "Exceeded maximum number of iterations: %" PetscInt_FMT " > %" PetscInt_FMT "\n", niter, tao->max_it));
2016a7e14dcfSSatish Balay     reason = TAO_DIVERGED_MAXITS;
2017a7e14dcfSSatish Balay   } else {
2018a7e14dcfSSatish Balay     reason = TAO_CONTINUE_ITERATING;
2019a7e14dcfSSatish Balay   }
2020a7e14dcfSSatish Balay   tao->reason = reason;
2021a7e14dcfSSatish Balay   PetscFunctionReturn(0);
2022a7e14dcfSSatish Balay }
2023a7e14dcfSSatish Balay 
2024a7e14dcfSSatish Balay /*@C
2025a7e14dcfSSatish Balay    TaoSetOptionsPrefix - Sets the prefix used for searching for all
202665ba42b6SBarry Smith    Tao options in the database.
2027a7e14dcfSSatish Balay 
202865ba42b6SBarry Smith    Logically Collective on tao
2029a7e14dcfSSatish Balay 
2030a7e14dcfSSatish Balay    Input Parameters:
2031441846f8SBarry Smith +  tao - the Tao context
203265ba42b6SBarry Smith -  prefix - the prefix string to prepend to all Tao option requests
2033a7e14dcfSSatish Balay 
2034a7e14dcfSSatish Balay    Notes:
2035a7e14dcfSSatish Balay    A hyphen (-) must NOT be given at the beginning of the prefix name.
2036a7e14dcfSSatish Balay    The first character of all runtime options is AUTOMATICALLY the hyphen.
2037a7e14dcfSSatish Balay 
2038a7e14dcfSSatish Balay    For example, to distinguish between the runtime options for two
203965ba42b6SBarry Smith    different Tao solvers, one could call
2040a7e14dcfSSatish Balay .vb
2041a7e14dcfSSatish Balay       TaoSetOptionsPrefix(tao1,"sys1_")
2042a7e14dcfSSatish Balay       TaoSetOptionsPrefix(tao2,"sys2_")
2043a7e14dcfSSatish Balay .ve
2044a7e14dcfSSatish Balay 
2045a7e14dcfSSatish Balay    This would enable use of different options for each system, such as
2046a7e14dcfSSatish Balay .vb
20479fa2b5dcSStefano Zampini       -sys1_tao_method blmvm -sys1_tao_grtol 1.e-3
20489fa2b5dcSStefano Zampini       -sys2_tao_method lmvm  -sys2_tao_grtol 1.e-4
2049a7e14dcfSSatish Balay .ve
2050a7e14dcfSSatish Balay 
2051a7e14dcfSSatish Balay    Level: advanced
2052a7e14dcfSSatish Balay 
205365ba42b6SBarry Smith .seealso: `TaoSetFromOptions()`, `TaoAppendOptionsPrefix()`, `TaoGetOptionsPrefix()`
2054a7e14dcfSSatish Balay @*/
2055a7e14dcfSSatish Balay 
2056d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoSetOptionsPrefix(Tao tao, const char p[])
2057d71ae5a4SJacob Faibussowitsch {
2058a7e14dcfSSatish Balay   PetscFunctionBegin;
205994511df7SStefano Zampini   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
20609566063dSJacob Faibussowitsch   PetscCall(PetscObjectSetOptionsPrefix((PetscObject)tao, p));
20611baa6e33SBarry Smith   if (tao->linesearch) PetscCall(TaoLineSearchSetOptionsPrefix(tao->linesearch, p));
20621baa6e33SBarry Smith   if (tao->ksp) PetscCall(KSPSetOptionsPrefix(tao->ksp, p));
2063a7e14dcfSSatish Balay   PetscFunctionReturn(0);
2064a7e14dcfSSatish Balay }
2065a7e14dcfSSatish Balay 
2066a7e14dcfSSatish Balay /*@C
2067a7e14dcfSSatish Balay    TaoAppendOptionsPrefix - Appends to the prefix used for searching for all
206865ba42b6SBarry Smith    Tao options in the database.
2069a7e14dcfSSatish Balay 
207065ba42b6SBarry Smith    Logically Collective on tao
2071a7e14dcfSSatish Balay 
2072a7e14dcfSSatish Balay    Input Parameters:
2073441846f8SBarry Smith +  tao - the Tao solver context
207465ba42b6SBarry Smith -  prefix - the prefix string to prepend to all Tao option requests
2075a7e14dcfSSatish Balay 
207665ba42b6SBarry Smith    Note:
2077a7e14dcfSSatish Balay    A hyphen (-) must NOT be given at the beginning of the prefix name.
207865ba42b6SBarry Smith    The first character of all runtime options is automatically the hyphen.
2079a7e14dcfSSatish Balay 
2080a7e14dcfSSatish Balay    Level: advanced
2081a7e14dcfSSatish Balay 
208265ba42b6SBarry Smith .seealso: `TaoSetFromOptions()`, `TaoSetOptionsPrefix()`, `TaoGetOptionsPrefix()`
2083a7e14dcfSSatish Balay @*/
2084d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoAppendOptionsPrefix(Tao tao, const char p[])
2085d71ae5a4SJacob Faibussowitsch {
2086a7e14dcfSSatish Balay   PetscFunctionBegin;
208794511df7SStefano Zampini   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
20889566063dSJacob Faibussowitsch   PetscCall(PetscObjectAppendOptionsPrefix((PetscObject)tao, p));
20891baa6e33SBarry Smith   if (tao->linesearch) PetscCall(PetscObjectAppendOptionsPrefix((PetscObject)tao->linesearch, p));
20901baa6e33SBarry Smith   if (tao->ksp) PetscCall(KSPAppendOptionsPrefix(tao->ksp, p));
2091a7e14dcfSSatish Balay   PetscFunctionReturn(0);
2092a7e14dcfSSatish Balay }
2093a7e14dcfSSatish Balay 
2094a7e14dcfSSatish Balay /*@C
2095a7e14dcfSSatish Balay   TaoGetOptionsPrefix - Gets the prefix used for searching for all
209665ba42b6SBarry Smith   Tao options in the database
2097a7e14dcfSSatish Balay 
2098a7e14dcfSSatish Balay   Not Collective
2099a7e14dcfSSatish Balay 
2100a7e14dcfSSatish Balay   Input Parameters:
2101441846f8SBarry Smith . tao - the Tao context
2102a7e14dcfSSatish Balay 
2103a7e14dcfSSatish Balay   Output Parameters:
2104a7e14dcfSSatish Balay . prefix - pointer to the prefix string used is returned
2105a7e14dcfSSatish Balay 
210665ba42b6SBarry Smith   Fortran Note:
210795452b02SPatrick Sanan     On the fortran side, the user should pass in a string 'prefix' of
2108a7e14dcfSSatish Balay   sufficient length to hold the prefix.
2109a7e14dcfSSatish Balay 
2110a7e14dcfSSatish Balay   Level: advanced
2111a7e14dcfSSatish Balay 
211265ba42b6SBarry Smith .seealso: `TaoSetFromOptions()`, `TaoSetOptionsPrefix()`, `TaoAppendOptionsPrefix()`
2113a7e14dcfSSatish Balay @*/
2114d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoGetOptionsPrefix(Tao tao, const char *p[])
2115d71ae5a4SJacob Faibussowitsch {
211694511df7SStefano Zampini   PetscFunctionBegin;
211794511df7SStefano Zampini   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
21189566063dSJacob Faibussowitsch   PetscCall(PetscObjectGetOptionsPrefix((PetscObject)tao, p));
211994511df7SStefano Zampini   PetscFunctionReturn(0);
2120a7e14dcfSSatish Balay }
2121a7e14dcfSSatish Balay 
2122a7e14dcfSSatish Balay /*@C
2123a7e14dcfSSatish Balay    TaoSetType - Sets the method for the unconstrained minimization solver.
2124a7e14dcfSSatish Balay 
212565ba42b6SBarry Smith    Collective on tao
2126a7e14dcfSSatish Balay 
2127a7e14dcfSSatish Balay    Input Parameters:
2128441846f8SBarry Smith +  solver - the Tao solver context
2129a7e14dcfSSatish Balay -  type - a known method
2130a7e14dcfSSatish Balay 
2131a7e14dcfSSatish Balay    Options Database Key:
213258417fe7SBarry Smith .  -tao_type <type> - Sets the method; use -help for a list
213358417fe7SBarry Smith    of available methods (for instance, "-tao_type lmvm" or "-tao_type tron")
2134a7e14dcfSSatish Balay 
2135a7e14dcfSSatish Balay    Available methods include:
213665ba42b6SBarry Smith +    `TAONLS` - nls Newton's method with line search for unconstrained minimization
213765ba42b6SBarry Smith .    `TAONTR` - ntr Newton's method with trust region for unconstrained minimization
213865ba42b6SBarry Smith .    `TAONTL` - ntl Newton's method with trust region, line search for unconstrained minimization
213965ba42b6SBarry Smith .    `TAOLMVM` - lmvm Limited memory variable metric method for unconstrained minimization
214065ba42b6SBarry Smith .    `TAOCG` - cg Nonlinear conjugate gradient method for unconstrained minimization
214165ba42b6SBarry Smith .    `TAONM` - nm Nelder-Mead algorithm for derivate-free unconstrained minimization
214265ba42b6SBarry Smith .    `TAOTRON` - tron Newton Trust Region method for bound constrained minimization
214365ba42b6SBarry Smith .    `TAOGPCG` - gpcg Newton Trust Region method for quadratic bound constrained minimization
214465ba42b6SBarry Smith .    `TAOBLMVM` - blmvm Limited memory variable metric method for bound constrained minimization
214565ba42b6SBarry Smith .    `TAOLCL` - lcl Linearly constrained Lagrangian method for pde-constrained minimization
214665ba42b6SBarry Smith -    `TAOPOUNDERS` - pounders Model-based algorithm for nonlinear least squares
2147a7e14dcfSSatish Balay 
2148a7e14dcfSSatish Balay   Level: intermediate
2149a7e14dcfSSatish Balay 
215065ba42b6SBarry Smith .seealso: `Tao`, `TaoCreate()`, `TaoGetType()`, `TaoType`
2151a7e14dcfSSatish Balay 
2152a7e14dcfSSatish Balay @*/
2153d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoSetType(Tao tao, TaoType type)
2154d71ae5a4SJacob Faibussowitsch {
2155441846f8SBarry Smith   PetscErrorCode (*create_xxx)(Tao);
2156a7e14dcfSSatish Balay   PetscBool issame;
2157a7e14dcfSSatish Balay 
2158a7e14dcfSSatish Balay   PetscFunctionBegin;
2159441846f8SBarry Smith   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
2160a7e14dcfSSatish Balay 
21619566063dSJacob Faibussowitsch   PetscCall(PetscObjectTypeCompare((PetscObject)tao, type, &issame));
2162a7e14dcfSSatish Balay   if (issame) PetscFunctionReturn(0);
2163a7e14dcfSSatish Balay 
21649566063dSJacob Faibussowitsch   PetscCall(PetscFunctionListFind(TaoList, type, (void (**)(void)) & create_xxx));
21653c859ba3SBarry Smith   PetscCheck(create_xxx, PetscObjectComm((PetscObject)tao), PETSC_ERR_ARG_UNKNOWN_TYPE, "Unable to find requested Tao type %s", type);
2166a7e14dcfSSatish Balay 
2167a7e14dcfSSatish Balay   /* Destroy the existing solver information */
2168dbbe0bcdSBarry Smith   PetscTryTypeMethod(tao, destroy);
21691baa6e33SBarry Smith   PetscCall(KSPDestroy(&tao->ksp));
21709566063dSJacob Faibussowitsch   PetscCall(TaoLineSearchDestroy(&tao->linesearch));
217183c8fe1dSLisandro Dalcin   tao->ops->setup          = NULL;
217283c8fe1dSLisandro Dalcin   tao->ops->solve          = NULL;
217383c8fe1dSLisandro Dalcin   tao->ops->view           = NULL;
217483c8fe1dSLisandro Dalcin   tao->ops->setfromoptions = NULL;
217583c8fe1dSLisandro Dalcin   tao->ops->destroy        = NULL;
2176a7e14dcfSSatish Balay 
2177a7e14dcfSSatish Balay   tao->setupcalled = PETSC_FALSE;
2178a7e14dcfSSatish Balay 
21799566063dSJacob Faibussowitsch   PetscCall((*create_xxx)(tao));
21809566063dSJacob Faibussowitsch   PetscCall(PetscObjectChangeTypeName((PetscObject)tao, type));
2181a7e14dcfSSatish Balay   PetscFunctionReturn(0);
2182a7e14dcfSSatish Balay }
218347a47007SBarry Smith 
2184a7e14dcfSSatish Balay /*MC
218565ba42b6SBarry Smith    TaoRegister - Adds a method to the Tao package for unconstrained minimization.
2186a7e14dcfSSatish Balay 
2187a7e14dcfSSatish Balay    Synopsis:
21885bd1e576SStefano Zampini    TaoRegister(char *name_solver,char *path,char *name_Create,PetscErrorCode (*routine_Create)(Tao))
2189a7e14dcfSSatish Balay 
2190a7e14dcfSSatish Balay    Not collective
2191a7e14dcfSSatish Balay 
2192a7e14dcfSSatish Balay    Input Parameters:
2193a7e14dcfSSatish Balay +  sname - name of a new user-defined solver
2194a7e14dcfSSatish Balay -  func - routine to Create method context
2195a7e14dcfSSatish Balay 
219665ba42b6SBarry Smith    Note:
219765ba42b6SBarry Smith    `TaoRegister()` may be called multiple times to add several user-defined solvers.
2198a7e14dcfSSatish Balay 
2199a7e14dcfSSatish Balay    Sample usage:
2200a7e14dcfSSatish Balay .vb
2201441846f8SBarry Smith    TaoRegister("my_solver",MySolverCreate);
2202a7e14dcfSSatish Balay .ve
2203a7e14dcfSSatish Balay 
2204a7e14dcfSSatish Balay    Then, your solver can be chosen with the procedural interface via
2205a7e14dcfSSatish Balay $     TaoSetType(tao,"my_solver")
2206a7e14dcfSSatish Balay    or at runtime via the option
220758417fe7SBarry Smith $     -tao_type my_solver
2208a7e14dcfSSatish Balay 
2209a7e14dcfSSatish Balay    Level: advanced
2210a7e14dcfSSatish Balay 
221165ba42b6SBarry Smith .seealso: `Tao`, `TaoSetType()`, `TaoRegisterAll()`, `TaoRegisterDestroy()`
2212a7e14dcfSSatish Balay M*/
2213d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoRegister(const char sname[], PetscErrorCode (*func)(Tao))
2214d71ae5a4SJacob Faibussowitsch {
2215a7e14dcfSSatish Balay   PetscFunctionBegin;
22169566063dSJacob Faibussowitsch   PetscCall(TaoInitializePackage());
22179566063dSJacob Faibussowitsch   PetscCall(PetscFunctionListAdd(&TaoList, sname, (void (*)(void))func));
2218a7e14dcfSSatish Balay   PetscFunctionReturn(0);
2219a7e14dcfSSatish Balay }
2220a7e14dcfSSatish Balay 
2221a7e14dcfSSatish Balay /*@C
2222441846f8SBarry Smith    TaoRegisterDestroy - Frees the list of minimization solvers that were
222365ba42b6SBarry Smith    registered by `TaoRegisterDynamic()`.
2224a7e14dcfSSatish Balay 
2225a7e14dcfSSatish Balay    Not Collective
2226a7e14dcfSSatish Balay 
2227a7e14dcfSSatish Balay    Level: advanced
2228a7e14dcfSSatish Balay 
2229db781477SPatrick Sanan .seealso: `TaoRegisterAll()`, `TaoRegister()`
2230a7e14dcfSSatish Balay @*/
2231d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoRegisterDestroy(void)
2232d71ae5a4SJacob Faibussowitsch {
2233a7e14dcfSSatish Balay   PetscFunctionBegin;
22349566063dSJacob Faibussowitsch   PetscCall(PetscFunctionListDestroy(&TaoList));
2235441846f8SBarry Smith   TaoRegisterAllCalled = PETSC_FALSE;
2236a7e14dcfSSatish Balay   PetscFunctionReturn(0);
2237a7e14dcfSSatish Balay }
223847a47007SBarry Smith 
22398931d482SJason Sarich /*@
22408931d482SJason Sarich    TaoGetIterationNumber - Gets the number of Tao iterations completed
22418931d482SJason Sarich    at this time.
22428931d482SJason Sarich 
22438931d482SJason Sarich    Not Collective
22448931d482SJason Sarich 
22458931d482SJason Sarich    Input Parameter:
22468931d482SJason Sarich .  tao - Tao context
22478931d482SJason Sarich 
22488931d482SJason Sarich    Output Parameter:
22498931d482SJason Sarich .  iter - iteration number
22508931d482SJason Sarich 
22518931d482SJason Sarich    Notes:
22528931d482SJason Sarich    For example, during the computation of iteration 2 this would return 1.
22538931d482SJason Sarich 
22548931d482SJason Sarich    Level: intermediate
22558931d482SJason Sarich 
2256db781477SPatrick Sanan .seealso: `TaoGetLinearSolveIterations()`, `TaoGetResidualNorm()`, `TaoGetObjective()`
22578931d482SJason Sarich @*/
2258d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoGetIterationNumber(Tao tao, PetscInt *iter)
2259d71ae5a4SJacob Faibussowitsch {
22608931d482SJason Sarich   PetscFunctionBegin;
22618931d482SJason Sarich   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
22628931d482SJason Sarich   PetscValidIntPointer(iter, 2);
22638931d482SJason Sarich   *iter = tao->niter;
22648931d482SJason Sarich   PetscFunctionReturn(0);
22658931d482SJason Sarich }
22668931d482SJason Sarich 
22678931d482SJason Sarich /*@
226879f5d8caSBarry Smith    TaoGetResidualNorm - Gets the current value of the norm of the residual
226979f5d8caSBarry Smith    at this time.
227079f5d8caSBarry Smith 
227179f5d8caSBarry Smith    Not Collective
227279f5d8caSBarry Smith 
227379f5d8caSBarry Smith    Input Parameter:
227479f5d8caSBarry Smith .  tao - Tao context
227579f5d8caSBarry Smith 
227679f5d8caSBarry Smith    Output Parameter:
227779f5d8caSBarry Smith .  value - the current value
227879f5d8caSBarry Smith 
227979f5d8caSBarry Smith    Level: intermediate
228079f5d8caSBarry Smith 
228165ba42b6SBarry Smith    Developer Note: This is the 2-norm of the residual, we cannot use `TaoGetGradientNorm()` because that has
228279f5d8caSBarry Smith                    a different meaning. For some reason Tao sometimes calls the gradient the residual.
228379f5d8caSBarry Smith 
2284db781477SPatrick Sanan .seealso: `TaoGetLinearSolveIterations()`, `TaoGetIterationNumber()`, `TaoGetObjective()`
228579f5d8caSBarry Smith @*/
2286d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoGetResidualNorm(Tao tao, PetscReal *value)
2287d71ae5a4SJacob Faibussowitsch {
228879f5d8caSBarry Smith   PetscFunctionBegin;
228979f5d8caSBarry Smith   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
229079f5d8caSBarry Smith   PetscValidRealPointer(value, 2);
229179f5d8caSBarry Smith   *value = tao->residual;
229279f5d8caSBarry Smith   PetscFunctionReturn(0);
229379f5d8caSBarry Smith }
229479f5d8caSBarry Smith 
229579f5d8caSBarry Smith /*@
22968931d482SJason Sarich    TaoSetIterationNumber - Sets the current iteration number.
22978931d482SJason Sarich 
229865ba42b6SBarry Smith    Logically Collective on tao
22998931d482SJason Sarich 
2300d8d19677SJose E. Roman    Input Parameters:
2301a2b725a8SWilliam Gropp +  tao - Tao context
2302a2b725a8SWilliam Gropp -  iter - iteration number
23038931d482SJason Sarich 
23048931d482SJason Sarich    Level: developer
23058931d482SJason Sarich 
2306db781477SPatrick Sanan .seealso: `TaoGetLinearSolveIterations()`
23078931d482SJason Sarich @*/
2308d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoSetIterationNumber(Tao tao, PetscInt iter)
2309d71ae5a4SJacob Faibussowitsch {
23108931d482SJason Sarich   PetscFunctionBegin;
23118931d482SJason Sarich   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
231294511df7SStefano Zampini   PetscValidLogicalCollectiveInt(tao, iter, 2);
23139566063dSJacob Faibussowitsch   PetscCall(PetscObjectSAWsTakeAccess((PetscObject)tao));
23148931d482SJason Sarich   tao->niter = iter;
23159566063dSJacob Faibussowitsch   PetscCall(PetscObjectSAWsGrantAccess((PetscObject)tao));
23168931d482SJason Sarich   PetscFunctionReturn(0);
23178931d482SJason Sarich }
23188931d482SJason Sarich 
23198931d482SJason Sarich /*@
23208931d482SJason Sarich    TaoGetTotalIterationNumber - Gets the total number of Tao iterations
23218931d482SJason Sarich    completed. This number keeps accumulating if multiple solves
23228931d482SJason Sarich    are called with the Tao object.
23238931d482SJason Sarich 
23248931d482SJason Sarich    Not Collective
23258931d482SJason Sarich 
23268931d482SJason Sarich    Input Parameter:
23278931d482SJason Sarich .  tao - Tao context
23288931d482SJason Sarich 
23298931d482SJason Sarich    Output Parameter:
23308931d482SJason Sarich .  iter - iteration number
23318931d482SJason Sarich 
23328931d482SJason Sarich    Notes:
23338931d482SJason Sarich    The total iteration count is updated after each solve, if there is a current
23348931d482SJason Sarich    TaoSolve() in progress then those iterations are not yet counted.
23358931d482SJason Sarich 
23368931d482SJason Sarich    Level: intermediate
23378931d482SJason Sarich 
2338db781477SPatrick Sanan .seealso: `TaoGetLinearSolveIterations()`
23398931d482SJason Sarich @*/
2340d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoGetTotalIterationNumber(Tao tao, PetscInt *iter)
2341d71ae5a4SJacob Faibussowitsch {
23428931d482SJason Sarich   PetscFunctionBegin;
23438931d482SJason Sarich   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
23448931d482SJason Sarich   PetscValidIntPointer(iter, 2);
23458931d482SJason Sarich   *iter = tao->ntotalits;
23468931d482SJason Sarich   PetscFunctionReturn(0);
23478931d482SJason Sarich }
23488931d482SJason Sarich 
23498931d482SJason Sarich /*@
23508931d482SJason Sarich    TaoSetTotalIterationNumber - Sets the current total iteration number.
23518931d482SJason Sarich 
235265ba42b6SBarry Smith    Logically Collective on tao
23538931d482SJason Sarich 
2354d8d19677SJose E. Roman    Input Parameters:
2355a2b725a8SWilliam Gropp +  tao - Tao context
2356a2b725a8SWilliam Gropp -  iter - iteration number
23578931d482SJason Sarich 
23588931d482SJason Sarich    Level: developer
23598931d482SJason Sarich 
2360db781477SPatrick Sanan .seealso: `TaoGetLinearSolveIterations()`
23618931d482SJason Sarich @*/
2362d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoSetTotalIterationNumber(Tao tao, PetscInt iter)
2363d71ae5a4SJacob Faibussowitsch {
23648931d482SJason Sarich   PetscFunctionBegin;
23658931d482SJason Sarich   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
236694511df7SStefano Zampini   PetscValidLogicalCollectiveInt(tao, iter, 2);
23679566063dSJacob Faibussowitsch   PetscCall(PetscObjectSAWsTakeAccess((PetscObject)tao));
23688931d482SJason Sarich   tao->ntotalits = iter;
23699566063dSJacob Faibussowitsch   PetscCall(PetscObjectSAWsGrantAccess((PetscObject)tao));
23708931d482SJason Sarich   PetscFunctionReturn(0);
23718931d482SJason Sarich }
23728931d482SJason Sarich 
2373a7e14dcfSSatish Balay /*@
2374e4cb33bbSBarry Smith   TaoSetConvergedReason - Sets the termination flag on a Tao object
2375a7e14dcfSSatish Balay 
237665ba42b6SBarry Smith   Logically Collective on tao
2377a7e14dcfSSatish Balay 
2378a7e14dcfSSatish Balay   Input Parameters:
2379441846f8SBarry Smith + tao - the Tao context
2380a7e14dcfSSatish Balay - reason - one of
238165ba42b6SBarry Smith $     `TAO_CONVERGED_ATOL` (2),
238265ba42b6SBarry Smith $     `TAO_CONVERGED_RTOL` (3),
238365ba42b6SBarry Smith $     `TAO_CONVERGED_STEPTOL` (4),
238465ba42b6SBarry Smith $     `TAO_CONVERGED_MINF` (5),
238565ba42b6SBarry Smith $     `TAO_CONVERGED_USER` (6),
238665ba42b6SBarry Smith $     `TAO_DIVERGED_MAXITS` (-2),
238765ba42b6SBarry Smith $     `TAO_DIVERGED_NAN` (-4),
238865ba42b6SBarry Smith $     `TAO_DIVERGED_MAXFCN` (-5),
238965ba42b6SBarry Smith $     `TAO_DIVERGED_LS_FAILURE` (-6),
239065ba42b6SBarry Smith $     `TAO_DIVERGED_TR_REDUCTION` (-7),
239165ba42b6SBarry Smith $     `TAO_DIVERGED_USER` (-8),
239265ba42b6SBarry Smith $     `TAO_CONTINUE_ITERATING` (0)
2393a7e14dcfSSatish Balay 
2394a7e14dcfSSatish Balay    Level: intermediate
2395a7e14dcfSSatish Balay 
2396a7e14dcfSSatish Balay @*/
2397d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoSetConvergedReason(Tao tao, TaoConvergedReason reason)
2398d71ae5a4SJacob Faibussowitsch {
2399a7e14dcfSSatish Balay   PetscFunctionBegin;
240094511df7SStefano Zampini   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
240194511df7SStefano Zampini   PetscValidLogicalCollectiveEnum(tao, reason, 2);
2402a7e14dcfSSatish Balay   tao->reason = reason;
2403a7e14dcfSSatish Balay   PetscFunctionReturn(0);
2404a7e14dcfSSatish Balay }
2405a7e14dcfSSatish Balay 
2406a7e14dcfSSatish Balay /*@
2407e4cb33bbSBarry Smith    TaoGetConvergedReason - Gets the reason the Tao iteration was stopped.
2408a7e14dcfSSatish Balay 
2409a7e14dcfSSatish Balay    Not Collective
2410a7e14dcfSSatish Balay 
2411a7e14dcfSSatish Balay    Input Parameter:
2412441846f8SBarry Smith .  tao - the Tao solver context
2413a7e14dcfSSatish Balay 
2414a7e14dcfSSatish Balay    Output Parameter:
2415a7e14dcfSSatish Balay .  reason - one of
241665ba42b6SBarry Smith $  `TAO_CONVERGED_GATOL` (3)           ||g(X)|| < gatol
241765ba42b6SBarry Smith $  `TAO_CONVERGED_GRTOL` (4)           ||g(X)|| / f(X)  < grtol
241865ba42b6SBarry Smith $  `TAO_CONVERGED_GTTOL` (5)           ||g(X)|| / ||g(X0)|| < gttol
241965ba42b6SBarry Smith $  `TAO_CONVERGED_STEPTOL` (6)         step size small
242065ba42b6SBarry Smith $  `TAO_CONVERGED_MINF` (7)            F < F_min
242165ba42b6SBarry Smith $  `TAO_CONVERGED_USER` (8)            User defined
242265ba42b6SBarry Smith $  `TAO_DIVERGED_MAXITS` (-2)          its > maxits
242365ba42b6SBarry Smith $  `TAO_DIVERGED_NAN` (-4)             Numerical problems
242465ba42b6SBarry Smith $  `TAO_DIVERGED_MAXFCN` (-5)          fevals > max_funcsals
242565ba42b6SBarry Smith $  `TAO_DIVERGED_LS_FAILURE` (-6)      line search failure
242665ba42b6SBarry Smith $  `TAO_DIVERGED_TR_REDUCTION` (-7)    trust region failure
242765ba42b6SBarry Smith $  `TAO_DIVERGED_USER` (-8)             (user defined)
242865ba42b6SBarry Smith $  `TAO_CONTINUE_ITERATING` (0)
2429a7e14dcfSSatish Balay 
2430a7e14dcfSSatish Balay    where
2431a7e14dcfSSatish Balay +  X - current solution
2432a7e14dcfSSatish Balay .  X0 - initial guess
2433a7e14dcfSSatish Balay .  f(X) - current function value
2434a7e14dcfSSatish Balay .  f(X*) - true solution (estimated)
2435a7e14dcfSSatish Balay .  g(X) - current gradient
2436a7e14dcfSSatish Balay .  its - current iterate number
2437a7e14dcfSSatish Balay .  maxits - maximum number of iterates
2438a7e14dcfSSatish Balay .  fevals - number of function evaluations
2439a7e14dcfSSatish Balay -  max_funcsals - maximum number of function evaluations
2440a7e14dcfSSatish Balay 
2441a7e14dcfSSatish Balay    Level: intermediate
2442a7e14dcfSSatish Balay 
2443db781477SPatrick Sanan .seealso: `TaoSetConvergenceTest()`, `TaoSetTolerances()`
2444a7e14dcfSSatish Balay 
2445a7e14dcfSSatish Balay @*/
2446d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoGetConvergedReason(Tao tao, TaoConvergedReason *reason)
2447d71ae5a4SJacob Faibussowitsch {
2448a7e14dcfSSatish Balay   PetscFunctionBegin;
2449441846f8SBarry Smith   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
2450a7e14dcfSSatish Balay   PetscValidPointer(reason, 2);
2451a7e14dcfSSatish Balay   *reason = tao->reason;
2452a7e14dcfSSatish Balay   PetscFunctionReturn(0);
2453a7e14dcfSSatish Balay }
2454a7e14dcfSSatish Balay 
2455a7e14dcfSSatish Balay /*@
2456a7e14dcfSSatish Balay    TaoGetSolutionStatus - Get the current iterate, objective value,
2457a7e14dcfSSatish Balay    residual, infeasibility, and termination
2458a7e14dcfSSatish Balay 
2459a7e14dcfSSatish Balay    Not Collective
2460a7e14dcfSSatish Balay 
2461f899ff85SJose E. Roman    Input Parameter:
2462441846f8SBarry Smith .  tao - the Tao context
2463a7e14dcfSSatish Balay 
2464a7e14dcfSSatish Balay    Output Parameters:
2465a7e14dcfSSatish Balay +  iterate - the current iterate number (>=0)
2466a7e14dcfSSatish Balay .  f - the current function value
246758417fe7SBarry Smith .  gnorm - the square of the gradient norm, duality gap, or other measure indicating distance from optimality.
2468a7e14dcfSSatish Balay .  cnorm - the infeasibility of the current solution with regard to the constraints.
2469a7e14dcfSSatish Balay .  xdiff - the step length or trust region radius of the most recent iterate.
247065ba42b6SBarry Smith -  reason - The termination reason, which can equal `TAO_CONTINUE_ITERATING`
2471a7e14dcfSSatish Balay 
2472a7e14dcfSSatish Balay    Level: intermediate
2473a7e14dcfSSatish Balay 
247465ba42b6SBarry Smith    Notes:
247565ba42b6SBarry Smith    Tao returns the values set by the solvers in the routine `TaoMonitor()`.
2476a7e14dcfSSatish Balay 
247747a47007SBarry Smith    If any of the output arguments are set to NULL, no corresponding value will be returned.
2478a7e14dcfSSatish Balay 
2479db781477SPatrick Sanan .seealso: `TaoMonitor()`, `TaoGetConvergedReason()`
2480a7e14dcfSSatish Balay @*/
2481d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoGetSolutionStatus(Tao tao, PetscInt *its, PetscReal *f, PetscReal *gnorm, PetscReal *cnorm, PetscReal *xdiff, TaoConvergedReason *reason)
2482d71ae5a4SJacob Faibussowitsch {
2483a7e14dcfSSatish Balay   PetscFunctionBegin;
248494511df7SStefano Zampini   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
2485a7e14dcfSSatish Balay   if (its) *its = tao->niter;
2486a7e14dcfSSatish Balay   if (f) *f = tao->fc;
2487a7e14dcfSSatish Balay   if (gnorm) *gnorm = tao->residual;
2488a7e14dcfSSatish Balay   if (cnorm) *cnorm = tao->cnorm;
2489a7e14dcfSSatish Balay   if (reason) *reason = tao->reason;
2490a7e14dcfSSatish Balay   if (xdiff) *xdiff = tao->step;
2491a7e14dcfSSatish Balay   PetscFunctionReturn(0);
2492a7e14dcfSSatish Balay }
2493a7e14dcfSSatish Balay 
2494a7e14dcfSSatish Balay /*@C
2495441846f8SBarry Smith    TaoGetType - Gets the current Tao algorithm.
2496a7e14dcfSSatish Balay 
2497a7e14dcfSSatish Balay    Not Collective
2498a7e14dcfSSatish Balay 
2499a7e14dcfSSatish Balay    Input Parameter:
2500441846f8SBarry Smith .  tao - the Tao solver context
2501a7e14dcfSSatish Balay 
2502a7e14dcfSSatish Balay    Output Parameter:
2503441846f8SBarry Smith .  type - Tao method
2504a7e14dcfSSatish Balay 
2505a7e14dcfSSatish Balay    Level: intermediate
2506a7e14dcfSSatish Balay 
250765ba42b6SBarry Smith .seealso: `Tao`, `TaoType`, `TaoSetType()`
2508a7e14dcfSSatish Balay @*/
2509d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoGetType(Tao tao, TaoType *type)
2510d71ae5a4SJacob Faibussowitsch {
2511a7e14dcfSSatish Balay   PetscFunctionBegin;
2512441846f8SBarry Smith   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
2513a7e14dcfSSatish Balay   PetscValidPointer(type, 2);
2514a7e14dcfSSatish Balay   *type = ((PetscObject)tao)->type_name;
2515a7e14dcfSSatish Balay   PetscFunctionReturn(0);
2516a7e14dcfSSatish Balay }
2517a7e14dcfSSatish Balay 
2518a7e14dcfSSatish Balay /*@C
2519a7e14dcfSSatish Balay   TaoMonitor - Monitor the solver and the current solution.  This
2520a7e14dcfSSatish Balay   routine will record the iteration number and residual statistics,
2521670c031eSStefano Zampini   and call any monitors specified by the user.
2522a7e14dcfSSatish Balay 
2523a7e14dcfSSatish Balay    Input Parameters:
2524441846f8SBarry Smith +  tao - the Tao context
2525a7e14dcfSSatish Balay .  its - the current iterate number (>=0)
2526a7e14dcfSSatish Balay .  f - the current objective function value
252758417fe7SBarry Smith .  res - the gradient norm, square root of the duality gap, or other measure indicating distince from optimality.  This measure will be recorded and
2528a7e14dcfSSatish Balay           used for some termination tests.
2529a7e14dcfSSatish Balay .  cnorm - the infeasibility of the current solution with regard to the constraints.
2530a7e14dcfSSatish Balay -  steplength - multiple of the step direction added to the previous iterate.
2531a7e14dcfSSatish Balay 
2532a7e14dcfSSatish Balay    Output Parameters:
253365ba42b6SBarry Smith .  reason - The termination reason, which can equal `TAO_CONTINUE_ITERATING`
2534a7e14dcfSSatish Balay 
2535a7e14dcfSSatish Balay    Options Database Key:
2536a7e14dcfSSatish Balay .  -tao_monitor - Use the default monitor, which prints statistics to standard output
2537a7e14dcfSSatish Balay 
2538a7e14dcfSSatish Balay    Level: developer
2539a7e14dcfSSatish Balay 
2540*a1cb98faSBarry Smith .seealso: `Tao`, `TaoGetConvergedReason()`, `TaoMonitorDefault()`, `TaoSetMonitor()`
2541a7e14dcfSSatish Balay @*/
2542d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoMonitor(Tao tao, PetscInt its, PetscReal f, PetscReal res, PetscReal cnorm, PetscReal steplength)
2543d71ae5a4SJacob Faibussowitsch {
254447a47007SBarry Smith   PetscInt i;
254547a47007SBarry Smith 
2546a7e14dcfSSatish Balay   PetscFunctionBegin;
2547441846f8SBarry Smith   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
2548a7e14dcfSSatish Balay   tao->fc       = f;
2549a7e14dcfSSatish Balay   tao->residual = res;
2550a7e14dcfSSatish Balay   tao->cnorm    = cnorm;
2551a7e14dcfSSatish Balay   tao->step     = steplength;
2552e52336cbSBarry Smith   if (!its) {
255394511df7SStefano Zampini     tao->cnorm0 = cnorm;
255494511df7SStefano Zampini     tao->gnorm0 = res;
2555a7e14dcfSSatish Balay   }
25563c859ba3SBarry Smith   PetscCheck(!PetscIsInfOrNanReal(f) && !PetscIsInfOrNanReal(res), PetscObjectComm((PetscObject)tao), PETSC_ERR_USER, "User provided compute function generated Inf or NaN");
255748a46eb9SPierre Jolivet   for (i = 0; i < tao->numbermonitors; i++) PetscCall((*tao->monitor[i])(tao, tao->monitorcontext[i]));
2558a7e14dcfSSatish Balay   PetscFunctionReturn(0);
2559a7e14dcfSSatish Balay }
2560a7e14dcfSSatish Balay 
2561a7e14dcfSSatish Balay /*@
2562ae93cb3cSJason Sarich    TaoSetConvergenceHistory - Sets the array used to hold the convergence history.
2563a7e14dcfSSatish Balay 
256465ba42b6SBarry Smith    Logically Collective on tao
2565a7e14dcfSSatish Balay 
2566a7e14dcfSSatish Balay    Input Parameters:
2567441846f8SBarry Smith +  tao - the Tao solver context
2568a7e14dcfSSatish Balay .  obj   - array to hold objective value history
2569a7e14dcfSSatish Balay .  resid - array to hold residual history
2570a7e14dcfSSatish Balay .  cnorm - array to hold constraint violation history
2571be1558d8SJason Sarich .  lits - integer array holds the number of linear iterations for each Tao iteration
2572a7e14dcfSSatish Balay .  na  - size of obj, resid, and cnorm
257365ba42b6SBarry Smith -  reset - `PETSC_TRUE` indicates each new minimization resets the history counter to zero,
2574a7e14dcfSSatish Balay            else it continues storing new values for new minimizations after the old ones
2575a7e14dcfSSatish Balay 
2576a7e14dcfSSatish Balay    Notes:
257765ba42b6SBarry Smith    If set, Tao will fill the given arrays with the indicated
2578ae93cb3cSJason Sarich    information at each iteration.  If 'obj','resid','cnorm','lits' are
257965ba42b6SBarry Smith    *all* NULL then space (using size na, or 1000 if na is `PETSC_DECIDE` or
258065ba42b6SBarry Smith    `PETSC_DEFAULT`) is allocated for the history.
2581be1558d8SJason Sarich    If not all are NULL, then only the non-NULL information categories
2582ae93cb3cSJason Sarich    will be stored, the others will be ignored.
2583ae93cb3cSJason Sarich 
2584ae93cb3cSJason Sarich    Any convergence information after iteration number 'na' will not be stored.
2585a7e14dcfSSatish Balay 
2586a7e14dcfSSatish Balay    This routine is useful, e.g., when running a code for purposes
2587a7e14dcfSSatish Balay    of accurate performance monitoring, when no I/O should be done
2588a7e14dcfSSatish Balay    during the section of code that is being timed.
2589a7e14dcfSSatish Balay 
2590a7e14dcfSSatish Balay    Level: intermediate
2591a7e14dcfSSatish Balay 
2592db781477SPatrick Sanan .seealso: `TaoGetConvergenceHistory()`
2593a7e14dcfSSatish Balay 
2594a7e14dcfSSatish Balay @*/
2595d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoSetConvergenceHistory(Tao tao, PetscReal obj[], PetscReal resid[], PetscReal cnorm[], PetscInt lits[], PetscInt na, PetscBool reset)
2596d71ae5a4SJacob Faibussowitsch {
2597a7e14dcfSSatish Balay   PetscFunctionBegin;
2598441846f8SBarry Smith   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
2599064a246eSJacob Faibussowitsch   if (obj) PetscValidRealPointer(obj, 2);
2600064a246eSJacob Faibussowitsch   if (resid) PetscValidRealPointer(resid, 3);
2601064a246eSJacob Faibussowitsch   if (cnorm) PetscValidRealPointer(cnorm, 4);
2602ae93cb3cSJason Sarich   if (lits) PetscValidIntPointer(lits, 5);
2603ae93cb3cSJason Sarich 
2604ae93cb3cSJason Sarich   if (na == PETSC_DECIDE || na == PETSC_DEFAULT) na = 1000;
2605ae93cb3cSJason Sarich   if (!obj && !resid && !cnorm && !lits) {
26069566063dSJacob Faibussowitsch     PetscCall(PetscCalloc4(na, &obj, na, &resid, na, &cnorm, na, &lits));
2607ae93cb3cSJason Sarich     tao->hist_malloc = PETSC_TRUE;
2608ae93cb3cSJason Sarich   }
2609ae93cb3cSJason Sarich 
2610a7e14dcfSSatish Balay   tao->hist_obj   = obj;
2611a7e14dcfSSatish Balay   tao->hist_resid = resid;
2612a7e14dcfSSatish Balay   tao->hist_cnorm = cnorm;
2613ae93cb3cSJason Sarich   tao->hist_lits  = lits;
2614a7e14dcfSSatish Balay   tao->hist_max   = na;
2615a7e14dcfSSatish Balay   tao->hist_reset = reset;
2616ae93cb3cSJason Sarich   tao->hist_len   = 0;
2617a7e14dcfSSatish Balay   PetscFunctionReturn(0);
2618a7e14dcfSSatish Balay }
2619a7e14dcfSSatish Balay 
2620a7e14dcfSSatish Balay /*@C
262165ba42b6SBarry Smith    TaoGetConvergenceHistory - Gets the arrays used that hold the convergence history.
2622a7e14dcfSSatish Balay 
262365ba42b6SBarry Smith    Collective on tao
2624a7e14dcfSSatish Balay 
2625a7e14dcfSSatish Balay    Input Parameter:
2626441846f8SBarry Smith .  tao - the Tao context
2627a7e14dcfSSatish Balay 
2628a7e14dcfSSatish Balay    Output Parameters:
2629a7e14dcfSSatish Balay +  obj   - array used to hold objective value history
2630a7e14dcfSSatish Balay .  resid - array used to hold residual history
2631a7e14dcfSSatish Balay .  cnorm - array used to hold constraint violation history
2632ae93cb3cSJason Sarich .  lits  - integer array used to hold linear solver iteration count
2633dd25e5eeSStefano Zampini -  nhist  - size of obj, resid, cnorm, and lits
2634a7e14dcfSSatish Balay 
2635a7e14dcfSSatish Balay    Notes:
263665ba42b6SBarry Smith     This routine must be preceded by calls to `TaoSetConvergenceHistory()`
263765ba42b6SBarry Smith     and `TaoSolve()`, otherwise it returns useless information.
2638ae93cb3cSJason Sarich 
2639a7e14dcfSSatish Balay     The calling sequence for this routine in Fortran is
2640ae93cb3cSJason Sarich $   call TaoGetConvergenceHistory(Tao tao, PetscInt nhist, PetscErrorCode ierr)
2641a7e14dcfSSatish Balay 
2642a7e14dcfSSatish Balay    This routine is useful, e.g., when running a code for purposes
2643a7e14dcfSSatish Balay    of accurate performance monitoring, when no I/O should be done
2644a7e14dcfSSatish Balay    during the section of code that is being timed.
2645a7e14dcfSSatish Balay 
2646a7e14dcfSSatish Balay    Level: advanced
2647a7e14dcfSSatish Balay 
264865ba42b6SBarry Smith .seealso: `Tao`, `TaoSolve()`, `TaoSetConvergenceHistory()`
2649a7e14dcfSSatish Balay 
2650a7e14dcfSSatish Balay @*/
2651d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoGetConvergenceHistory(Tao tao, PetscReal **obj, PetscReal **resid, PetscReal **cnorm, PetscInt **lits, PetscInt *nhist)
2652d71ae5a4SJacob Faibussowitsch {
2653a7e14dcfSSatish Balay   PetscFunctionBegin;
2654441846f8SBarry Smith   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
2655a7e14dcfSSatish Balay   if (obj) *obj = tao->hist_obj;
2656a7e14dcfSSatish Balay   if (cnorm) *cnorm = tao->hist_cnorm;
2657a7e14dcfSSatish Balay   if (resid) *resid = tao->hist_resid;
2658a7e14dcfSSatish Balay   if (nhist) *nhist = tao->hist_len;
2659a7e14dcfSSatish Balay   PetscFunctionReturn(0);
2660a7e14dcfSSatish Balay }
2661a7e14dcfSSatish Balay 
2662a7e14dcfSSatish Balay /*@
2663a7e14dcfSSatish Balay    TaoSetApplicationContext - Sets the optional user-defined context for
2664a7e14dcfSSatish Balay    a solver.
2665a7e14dcfSSatish Balay 
266665ba42b6SBarry Smith    Logically Collective on tao
2667a7e14dcfSSatish Balay 
2668a7e14dcfSSatish Balay    Input Parameters:
2669441846f8SBarry Smith +  tao  - the Tao context
2670a7e14dcfSSatish Balay -  usrP - optional user context
2671a7e14dcfSSatish Balay 
2672a7e14dcfSSatish Balay    Level: intermediate
2673a7e14dcfSSatish Balay 
267465ba42b6SBarry Smith .seealso: `Tao`, `TaoGetApplicationContext()`, `TaoSetApplicationContext()`
2675a7e14dcfSSatish Balay @*/
2676d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoSetApplicationContext(Tao tao, void *usrP)
2677d71ae5a4SJacob Faibussowitsch {
2678a7e14dcfSSatish Balay   PetscFunctionBegin;
2679441846f8SBarry Smith   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
2680a7e14dcfSSatish Balay   tao->user = usrP;
2681a7e14dcfSSatish Balay   PetscFunctionReturn(0);
2682a7e14dcfSSatish Balay }
2683a7e14dcfSSatish Balay 
2684a7e14dcfSSatish Balay /*@
2685a7e14dcfSSatish Balay    TaoGetApplicationContext - Gets the user-defined context for a
268665ba42b6SBarry Smith    Tao solvers.
2687a7e14dcfSSatish Balay 
2688a7e14dcfSSatish Balay    Not Collective
2689a7e14dcfSSatish Balay 
2690a7e14dcfSSatish Balay    Input Parameter:
2691441846f8SBarry Smith .  tao  - Tao context
2692a7e14dcfSSatish Balay 
2693a7e14dcfSSatish Balay    Output Parameter:
2694a7e14dcfSSatish Balay .  usrP - user context
2695a7e14dcfSSatish Balay 
2696a7e14dcfSSatish Balay    Level: intermediate
2697a7e14dcfSSatish Balay 
2698db781477SPatrick Sanan .seealso: `TaoSetApplicationContext()`
2699a7e14dcfSSatish Balay @*/
2700d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoGetApplicationContext(Tao tao, void *usrP)
2701d71ae5a4SJacob Faibussowitsch {
2702a7e14dcfSSatish Balay   PetscFunctionBegin;
2703441846f8SBarry Smith   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
270494511df7SStefano Zampini   PetscValidPointer(usrP, 2);
2705a7e14dcfSSatish Balay   *(void **)usrP = tao->user;
2706a7e14dcfSSatish Balay   PetscFunctionReturn(0);
2707a7e14dcfSSatish Balay }
2708a9603a14SPatrick Farrell 
2709a9603a14SPatrick Farrell /*@
271065ba42b6SBarry Smith    TaoSetGradientNorm - Sets the matrix used to define the norm that measures the size of the gradient.
2711a9603a14SPatrick Farrell 
2712a9603a14SPatrick Farrell    Collective on tao
2713a9603a14SPatrick Farrell 
2714a9603a14SPatrick Farrell    Input Parameters:
2715a9603a14SPatrick Farrell +  tao  - the Tao context
271665ba42b6SBarry Smith -  M    - matrix that defines the norm
2717a9603a14SPatrick Farrell 
2718a9603a14SPatrick Farrell    Level: beginner
2719a9603a14SPatrick Farrell 
272065ba42b6SBarry Smith .seealso: `Tao`, `TaoGetGradientNorm()`, `TaoGradientNorm()`
2721a9603a14SPatrick Farrell @*/
2722d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoSetGradientNorm(Tao tao, Mat M)
2723d71ae5a4SJacob Faibussowitsch {
2724a9603a14SPatrick Farrell   PetscFunctionBegin;
2725a9603a14SPatrick Farrell   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
272694511df7SStefano Zampini   PetscValidHeaderSpecific(M, MAT_CLASSID, 2);
27279566063dSJacob Faibussowitsch   PetscCall(PetscObjectReference((PetscObject)M));
27289566063dSJacob Faibussowitsch   PetscCall(MatDestroy(&tao->gradient_norm));
27299566063dSJacob Faibussowitsch   PetscCall(VecDestroy(&tao->gradient_norm_tmp));
2730a9603a14SPatrick Farrell   tao->gradient_norm = M;
27319566063dSJacob Faibussowitsch   PetscCall(MatCreateVecs(M, NULL, &tao->gradient_norm_tmp));
2732a9603a14SPatrick Farrell   PetscFunctionReturn(0);
2733a9603a14SPatrick Farrell }
2734a9603a14SPatrick Farrell 
2735a9603a14SPatrick Farrell /*@
273665ba42b6SBarry Smith    TaoGetGradientNorm - Returns the matrix used to define the norm used for measuring the size of the gradient.
2737a9603a14SPatrick Farrell 
2738a9603a14SPatrick Farrell    Not Collective
2739a9603a14SPatrick Farrell 
2740a9603a14SPatrick Farrell    Input Parameter:
2741a9603a14SPatrick Farrell .  tao  - Tao context
2742a9603a14SPatrick Farrell 
2743a9603a14SPatrick Farrell    Output Parameter:
2744a9603a14SPatrick Farrell .  M - gradient norm
2745a9603a14SPatrick Farrell 
2746a9603a14SPatrick Farrell    Level: beginner
2747a9603a14SPatrick Farrell 
274865ba42b6SBarry Smith .seealso: `Tao`, `TaoSetGradientNorm()`, `TaoGradientNorm()`
2749a9603a14SPatrick Farrell @*/
2750d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoGetGradientNorm(Tao tao, Mat *M)
2751d71ae5a4SJacob Faibussowitsch {
2752a9603a14SPatrick Farrell   PetscFunctionBegin;
2753a9603a14SPatrick Farrell   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
275494511df7SStefano Zampini   PetscValidPointer(M, 2);
2755a9603a14SPatrick Farrell   *M = tao->gradient_norm;
2756a9603a14SPatrick Farrell   PetscFunctionReturn(0);
2757a9603a14SPatrick Farrell }
2758a9603a14SPatrick Farrell 
275905aca2cdSSatish Balay /*@C
276065ba42b6SBarry Smith    TaoGradientNorm - Compute the norm with respect to the norm the user has set.
2761a9603a14SPatrick Farrell 
2762a9603a14SPatrick Farrell    Collective on tao
2763a9603a14SPatrick Farrell 
2764d8d19677SJose E. Roman    Input Parameters:
27652653c764SPierre Jolivet +  tao      - the Tao context
2766a9603a14SPatrick Farrell .  gradient - the gradient to be computed
27672653c764SPierre Jolivet -  norm     - the norm type
2768a9603a14SPatrick Farrell 
2769a9603a14SPatrick Farrell    Output Parameter:
2770a9603a14SPatrick Farrell .  gnorm    - the gradient norm
2771a9603a14SPatrick Farrell 
2772a9603a14SPatrick Farrell    Level: developer
2773a9603a14SPatrick Farrell 
277465ba42b6SBarry Smith .seealso: `Tao`, `TaoSetGradientNorm()`, `TaoGetGradientNorm()`
2775a9603a14SPatrick Farrell @*/
2776d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoGradientNorm(Tao tao, Vec gradient, NormType type, PetscReal *gnorm)
2777d71ae5a4SJacob Faibussowitsch {
2778a9603a14SPatrick Farrell   PetscFunctionBegin;
27798854b543SStefano Zampini   PetscValidHeaderSpecific(tao, TAO_CLASSID, 1);
27808854b543SStefano Zampini   PetscValidHeaderSpecific(gradient, VEC_CLASSID, 2);
27818854b543SStefano Zampini   PetscValidLogicalCollectiveEnum(tao, type, 3);
278294511df7SStefano Zampini   PetscValidRealPointer(gnorm, 4);
2783a9603a14SPatrick Farrell   if (tao->gradient_norm) {
2784680e2bc7SPatrick Farrell     PetscScalar gnorms;
2785680e2bc7SPatrick Farrell 
27863c859ba3SBarry 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.");
27879566063dSJacob Faibussowitsch     PetscCall(MatMult(tao->gradient_norm, gradient, tao->gradient_norm_tmp));
27889566063dSJacob Faibussowitsch     PetscCall(VecDot(gradient, tao->gradient_norm_tmp, &gnorms));
27897bb79937SPatrick Farrell     *gnorm = PetscRealPart(PetscSqrtScalar(gnorms));
2790a9603a14SPatrick Farrell   } else {
27919566063dSJacob Faibussowitsch     PetscCall(VecNorm(gradient, type, gnorm));
2792a9603a14SPatrick Farrell   }
2793a9603a14SPatrick Farrell   PetscFunctionReturn(0);
2794a9603a14SPatrick Farrell }
2795a9603a14SPatrick Farrell 
2796e882e171SHong Zhang /*@C
279765ba42b6SBarry Smith    TaoMonitorDrawCtxCreate - Creates the monitor context `TaoMonitorDrawSolution()`
2798a9603a14SPatrick Farrell 
279965ba42b6SBarry Smith    Collective on tao
2800e882e171SHong Zhang 
2801d5b43468SJose E. Roman    Output Parameter:
2802e882e171SHong Zhang .    ctx - the monitor context
2803e882e171SHong Zhang 
2804e882e171SHong Zhang    Options Database:
2805e882e171SHong Zhang .   -tao_draw_solution_initial - show initial guess as well as current solution
2806e882e171SHong Zhang 
2807e882e171SHong Zhang    Level: intermediate
2808e882e171SHong Zhang 
280965ba42b6SBarry Smith .seealso: `Tao`, `TaoMonitorSet()`, `TaoMonitorDefault()`, `VecView()`, `TaoMonitorDrawCtx()`
2810e882e171SHong Zhang @*/
2811d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoMonitorDrawCtxCreate(MPI_Comm comm, const char host[], const char label[], int x, int y, int m, int n, PetscInt howoften, TaoMonitorDrawCtx *ctx)
2812d71ae5a4SJacob Faibussowitsch {
2813e882e171SHong Zhang   PetscFunctionBegin;
28149566063dSJacob Faibussowitsch   PetscCall(PetscNew(ctx));
28159566063dSJacob Faibussowitsch   PetscCall(PetscViewerDrawOpen(comm, host, label, x, y, m, n, &(*ctx)->viewer));
28169566063dSJacob Faibussowitsch   PetscCall(PetscViewerSetFromOptions((*ctx)->viewer));
2817e882e171SHong Zhang   (*ctx)->howoften = howoften;
2818e882e171SHong Zhang   PetscFunctionReturn(0);
2819e882e171SHong Zhang }
2820e882e171SHong Zhang 
2821e882e171SHong Zhang /*@C
282265ba42b6SBarry Smith    TaoMonitorDrawCtxDestroy - Destroys the monitor context for `TaoMonitorDrawSolution()`
2823e882e171SHong Zhang 
282465ba42b6SBarry Smith    Collective on tao
2825e882e171SHong Zhang 
2826e882e171SHong Zhang    Input Parameters:
2827e882e171SHong Zhang .    ctx - the monitor context
2828e882e171SHong Zhang 
2829e882e171SHong Zhang    Level: intermediate
2830e882e171SHong Zhang 
2831db781477SPatrick Sanan .seealso: `TaoMonitorSet()`, `TaoMonitorDefault()`, `VecView()`, `TaoMonitorDrawSolution()`
2832e882e171SHong Zhang @*/
2833d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoMonitorDrawCtxDestroy(TaoMonitorDrawCtx *ictx)
2834d71ae5a4SJacob Faibussowitsch {
2835e882e171SHong Zhang   PetscFunctionBegin;
28369566063dSJacob Faibussowitsch   PetscCall(PetscViewerDestroy(&(*ictx)->viewer));
28379566063dSJacob Faibussowitsch   PetscCall(PetscFree(*ictx));
2838e882e171SHong Zhang   PetscFunctionReturn(0);
2839e882e171SHong Zhang }
2840