xref: /petsc/src/tao/interface/taosolver.c (revision c2e3fba1fe1cda7e6350bbca19c4ed35ce95940a)
1441846f8SBarry Smith #define TAO_DLL
2a7e14dcfSSatish Balay 
3af0996ceSBarry Smith #include <petsc/private/taoimpl.h> /*I "petsctao.h" I*/
4a7e14dcfSSatish Balay 
5441846f8SBarry Smith PetscBool TaoRegisterAllCalled = PETSC_FALSE;
6441846f8SBarry Smith PetscFunctionList TaoList = NULL;
7a7e14dcfSSatish Balay 
8441846f8SBarry Smith PetscClassId TAO_CLASSID;
90ebee16dSLisandro Dalcin 
100ebee16dSLisandro Dalcin PetscLogEvent TAO_Solve;
110ebee16dSLisandro Dalcin PetscLogEvent TAO_ObjectiveEval;
120ebee16dSLisandro Dalcin PetscLogEvent TAO_GradientEval;
130ebee16dSLisandro Dalcin PetscLogEvent TAO_ObjGradEval;
140ebee16dSLisandro Dalcin PetscLogEvent TAO_HessianEval;
150ebee16dSLisandro Dalcin PetscLogEvent TAO_JacobianEval;
160ebee16dSLisandro Dalcin PetscLogEvent TAO_ConstraintsEval;
17a7e14dcfSSatish Balay 
1883c8fe1dSLisandro Dalcin const char *TaoSubSetTypes[] = {"subvec","mask","matrixfree","TaoSubSetType","TAO_SUBSET_",NULL};
19a7e14dcfSSatish Balay 
20e882e171SHong Zhang struct _n_TaoMonitorDrawCtx {
21e882e171SHong Zhang   PetscViewer viewer;
22e882e171SHong Zhang   PetscInt    howoften;  /* when > 0 uses iteration % howoften, when negative only final solution plotted */
23e882e171SHong Zhang };
24e882e171SHong Zhang 
25a7e14dcfSSatish Balay /*@
26a7e14dcfSSatish Balay   TaoCreate - Creates a TAO solver
27a7e14dcfSSatish Balay 
28d083f849SBarry Smith   Collective
29a7e14dcfSSatish Balay 
30a7e14dcfSSatish Balay   Input Parameter:
31a7e14dcfSSatish Balay . comm - MPI communicator
32a7e14dcfSSatish Balay 
33a7e14dcfSSatish Balay   Output Parameter:
34441846f8SBarry Smith . newtao - the new Tao context
35a7e14dcfSSatish Balay 
36a7e14dcfSSatish Balay   Available methods include:
3758417fe7SBarry Smith +    nls - Newton's method with line search for unconstrained minimization
3858417fe7SBarry Smith .    ntr - Newton's method with trust region for unconstrained minimization
3958417fe7SBarry Smith .    ntl - Newton's method with trust region, line search for unconstrained minimization
4058417fe7SBarry Smith .    lmvm - Limited memory variable metric method for unconstrained minimization
4158417fe7SBarry Smith .    cg - Nonlinear conjugate gradient method for unconstrained minimization
4258417fe7SBarry Smith .    nm - Nelder-Mead algorithm for derivate-free unconstrained minimization
4358417fe7SBarry Smith .    tron - Newton Trust Region method for bound constrained minimization
4458417fe7SBarry Smith .    gpcg - Newton Trust Region method for quadratic bound constrained minimization
4558417fe7SBarry Smith .    blmvm - Limited memory variable metric method for bound constrained minimization
4658417fe7SBarry Smith .    lcl - Linearly constrained Lagrangian method for pde-constrained minimization
4758417fe7SBarry Smith -    pounders - Model-based algorithm for nonlinear least squares
48a7e14dcfSSatish Balay 
49a7e14dcfSSatish Balay    Options Database Keys:
5058417fe7SBarry Smith .   -tao_type - select which method TAO should use
51a7e14dcfSSatish Balay 
52a7e14dcfSSatish Balay    Level: beginner
53a7e14dcfSSatish Balay 
54db781477SPatrick Sanan .seealso: `TaoSolve()`, `TaoDestroy()`
55a7e14dcfSSatish Balay @*/
56441846f8SBarry Smith PetscErrorCode TaoCreate(MPI_Comm comm, Tao *newtao)
57a7e14dcfSSatish Balay {
58441846f8SBarry Smith   Tao            tao;
59a7e14dcfSSatish Balay 
60a7e14dcfSSatish Balay   PetscFunctionBegin;
61a7e14dcfSSatish Balay   PetscValidPointer(newtao,2);
629566063dSJacob Faibussowitsch   PetscCall(TaoInitializePackage());
639566063dSJacob Faibussowitsch   PetscCall(TaoLineSearchInitializePackage());
649566063dSJacob Faibussowitsch   PetscCall(PetscHeaderCreate(tao,TAO_CLASSID,"Tao","Optimization solver","Tao",comm,TaoDestroy,TaoView));
65a7e14dcfSSatish Balay 
6694511df7SStefano Zampini   /* Set non-NULL defaults */
6794511df7SStefano Zampini   tao->ops->convergencetest = TaoDefaultConvergenceTest;
68a7e14dcfSSatish Balay 
69a7e14dcfSSatish Balay   tao->max_it    = 10000;
7094511df7SStefano Zampini   tao->max_funcs = -1;
716f4723b1SBarry Smith #if defined(PETSC_USE_REAL_SINGLE)
726f4723b1SBarry Smith   tao->gatol     = 1e-5;
736f4723b1SBarry Smith   tao->grtol     = 1e-5;
746f1f5f59SAlp Dener   tao->crtol     = 1e-5;
756f1f5f59SAlp Dener   tao->catol     = 1e-5;
766f4723b1SBarry Smith #else
77a7e14dcfSSatish Balay   tao->gatol     = 1e-8;
78a7e14dcfSSatish Balay   tao->grtol     = 1e-8;
796f1f5f59SAlp Dener   tao->crtol     = 1e-8;
806f1f5f59SAlp Dener   tao->catol     = 1e-8;
816f4723b1SBarry Smith #endif
826552cf8aSJason Sarich   tao->gttol     = 0.0;
834f1535bcSTodd Munson   tao->steptol   = 0.0;
84e270355aSBarry Smith   tao->trust0    = PETSC_INFINITY;
856f4723b1SBarry Smith   tao->fmin      = PETSC_NINFINITY;
869fa2b5dcSStefano Zampini 
87a7e14dcfSSatish Balay   tao->hist_reset  = PETSC_TRUE;
88a7e14dcfSSatish Balay 
899566063dSJacob Faibussowitsch   PetscCall(TaoResetStatistics(tao));
90a7e14dcfSSatish Balay   *newtao = tao;
91a7e14dcfSSatish Balay   PetscFunctionReturn(0);
92a7e14dcfSSatish Balay }
93a7e14dcfSSatish Balay 
94a7e14dcfSSatish Balay /*@
95a7e14dcfSSatish Balay   TaoSolve - Solves an optimization problem min F(x) s.t. l <= x <= u
96a7e14dcfSSatish Balay 
97441846f8SBarry Smith   Collective on Tao
98a7e14dcfSSatish Balay 
99a7e14dcfSSatish Balay   Input Parameters:
100441846f8SBarry Smith . tao - the Tao context
101a7e14dcfSSatish Balay 
102a7e14dcfSSatish Balay   Notes:
103a82e8c82SStefano Zampini   The user must set up the Tao with calls to TaoSetSolution(),
104a82e8c82SStefano Zampini   TaoSetObjective(),
105a82e8c82SStefano Zampini   TaoSetGradient(), and (if using 2nd order method) TaoSetHessian().
106a7e14dcfSSatish Balay 
107a35d58b8SBarry Smith   You should call TaoGetConvergedReason() or run with -tao_converged_reason to determine if the optimization algorithm actually succeeded or
108a35d58b8SBarry Smith   why it failed.
109a35d58b8SBarry Smith 
110a7e14dcfSSatish Balay   Level: beginner
111a7e14dcfSSatish Balay 
112db781477SPatrick Sanan .seealso: `TaoCreate()`, `TaoSetObjective()`, `TaoSetGradient()`, `TaoSetHessian()`, `TaoGetConvergedReason()`
113a7e14dcfSSatish Balay  @*/
114441846f8SBarry Smith PetscErrorCode TaoSolve(Tao tao)
115a7e14dcfSSatish Balay {
116e2379f4fSBarry Smith   static PetscBool set = PETSC_FALSE;
11747a47007SBarry Smith 
118a7e14dcfSSatish Balay   PetscFunctionBegin;
119441846f8SBarry Smith   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
120d0609cedSBarry Smith   PetscCall(PetscCitationsRegister("@TechReport{tao-user-ref,\n"
121e2379f4fSBarry Smith                                    "title   = {Toolkit for Advanced Optimization (TAO) Users Manual},\n"
122e2379f4fSBarry Smith                                    "author  = {Todd Munson and Jason Sarich and Stefan Wild and Steve Benson and Lois Curfman McInnes},\n"
123e2379f4fSBarry Smith                                    "Institution = {Argonne National Laboratory},\n"
124e2379f4fSBarry Smith                                    "Year   = 2014,\n"
125e2379f4fSBarry Smith                                    "Number = {ANL/MCS-TM-322 - Revision 3.5},\n"
126d0609cedSBarry Smith                                    "url    = {https://www.mcs.anl.gov/research/projects/tao/}\n}\n",&set));
127494bef23SAlp Dener   tao->header_printed = PETSC_FALSE;
1289566063dSJacob Faibussowitsch   PetscCall(TaoSetUp(tao));
1299566063dSJacob Faibussowitsch   PetscCall(TaoResetStatistics(tao));
130a7e14dcfSSatish Balay   if (tao->linesearch) {
1319566063dSJacob Faibussowitsch     PetscCall(TaoLineSearchReset(tao->linesearch));
132a7e14dcfSSatish Balay   }
133a7e14dcfSSatish Balay 
1349566063dSJacob Faibussowitsch   PetscCall(PetscLogEventBegin(TAO_Solve,tao,0,0,0));
1359566063dSJacob Faibussowitsch   if (tao->ops->solve) PetscCall((*tao->ops->solve)(tao));
1369566063dSJacob Faibussowitsch   PetscCall(PetscLogEventEnd(TAO_Solve,tao,0,0,0));
137a7e14dcfSSatish Balay 
1389566063dSJacob Faibussowitsch   PetscCall(VecViewFromOptions(tao->solution,(PetscObject)tao,"-tao_view_solution"));
1397cf37e64SBarry Smith 
1408931d482SJason Sarich   tao->ntotalits += tao->niter;
1419566063dSJacob Faibussowitsch   PetscCall(TaoViewFromOptions(tao,NULL,"-tao_view"));
142a7e14dcfSSatish Balay 
143a7e14dcfSSatish Balay   if (tao->printreason) {
144a7e14dcfSSatish Balay     if (tao->reason > 0) {
14563a3b9bcSJacob Faibussowitsch       PetscCall(PetscPrintf(((PetscObject)tao)->comm,"TAO solve converged due to %s iterations %" PetscInt_FMT "\n",TaoConvergedReasons[tao->reason],tao->niter));
146a7e14dcfSSatish Balay     } else {
14763a3b9bcSJacob Faibussowitsch       PetscCall(PetscPrintf(((PetscObject)tao)->comm,"TAO solve did not converge due to %s iteration %" PetscInt_FMT "\n",TaoConvergedReasons[tao->reason],tao->niter));
148a7e14dcfSSatish Balay     }
149a7e14dcfSSatish Balay   }
150a7e14dcfSSatish Balay   PetscFunctionReturn(0);
151a7e14dcfSSatish Balay }
152a7e14dcfSSatish Balay 
153a7e14dcfSSatish Balay /*@
154a7e14dcfSSatish Balay   TaoSetUp - Sets up the internal data structures for the later use
155a7e14dcfSSatish Balay   of a Tao solver
156a7e14dcfSSatish Balay 
157a7e14dcfSSatish Balay   Collective on tao
158a7e14dcfSSatish Balay 
159a7e14dcfSSatish Balay   Input Parameters:
160a7e14dcfSSatish Balay . tao - the TAO context
161a7e14dcfSSatish Balay 
162a7e14dcfSSatish Balay   Notes:
163a7e14dcfSSatish Balay   The user will not need to explicitly call TaoSetUp(), as it will
164a7e14dcfSSatish Balay   automatically be called in TaoSolve().  However, if the user
165a7e14dcfSSatish Balay   desires to call it explicitly, it should come after TaoCreate()
166a7e14dcfSSatish Balay   and any TaoSetSomething() routines, but before TaoSolve().
167a7e14dcfSSatish Balay 
168a7e14dcfSSatish Balay   Level: advanced
169a7e14dcfSSatish Balay 
170db781477SPatrick Sanan .seealso: `TaoCreate()`, `TaoSolve()`
171a7e14dcfSSatish Balay @*/
172441846f8SBarry Smith PetscErrorCode TaoSetUp(Tao tao)
173a7e14dcfSSatish Balay {
174a7e14dcfSSatish Balay   PetscFunctionBegin;
175441846f8SBarry Smith   PetscValidHeaderSpecific(tao, TAO_CLASSID,1);
176a7e14dcfSSatish Balay   if (tao->setupcalled) PetscFunctionReturn(0);
1773c859ba3SBarry Smith   PetscCheck(tao->solution,PetscObjectComm((PetscObject)tao),PETSC_ERR_ARG_WRONGSTATE,"Must call TaoSetSolution");
178a7e14dcfSSatish Balay   if (tao->ops->setup) {
1799566063dSJacob Faibussowitsch     PetscCall((*tao->ops->setup)(tao));
180a7e14dcfSSatish Balay   }
181a7e14dcfSSatish Balay   tao->setupcalled = PETSC_TRUE;
182a7e14dcfSSatish Balay   PetscFunctionReturn(0);
183a7e14dcfSSatish Balay }
184a7e14dcfSSatish Balay 
1851fb7b255SJunchao Zhang /*@C
186a7e14dcfSSatish Balay   TaoDestroy - Destroys the TAO context that was created with
187a7e14dcfSSatish Balay   TaoCreate()
188a7e14dcfSSatish Balay 
189441846f8SBarry Smith   Collective on Tao
190a7e14dcfSSatish Balay 
191a7e14dcfSSatish Balay   Input Parameter:
192441846f8SBarry Smith . tao - the Tao context
193a7e14dcfSSatish Balay 
194a7e14dcfSSatish Balay   Level: beginner
195a7e14dcfSSatish Balay 
196db781477SPatrick Sanan .seealso: `TaoCreate()`, `TaoSolve()`
197a7e14dcfSSatish Balay @*/
198441846f8SBarry Smith PetscErrorCode TaoDestroy(Tao *tao)
199a7e14dcfSSatish Balay {
200a7e14dcfSSatish Balay   PetscFunctionBegin;
201a7e14dcfSSatish Balay   if (!*tao) PetscFunctionReturn(0);
202441846f8SBarry Smith   PetscValidHeaderSpecific(*tao,TAO_CLASSID,1);
20383c8fe1dSLisandro Dalcin   if (--((PetscObject)*tao)->refct > 0) {*tao = NULL;PetscFunctionReturn(0);}
204a7e14dcfSSatish Balay 
205a7e14dcfSSatish Balay   if ((*tao)->ops->destroy) {
2069566063dSJacob Faibussowitsch     PetscCall((*((*tao))->ops->destroy)(*tao));
207a7e14dcfSSatish Balay   }
2089566063dSJacob Faibussowitsch   PetscCall(KSPDestroy(&(*tao)->ksp));
2099566063dSJacob Faibussowitsch   PetscCall(TaoLineSearchDestroy(&(*tao)->linesearch));
210a7e14dcfSSatish Balay 
211a7e14dcfSSatish Balay   if ((*tao)->ops->convergencedestroy) {
2129566063dSJacob Faibussowitsch     PetscCall((*(*tao)->ops->convergencedestroy)((*tao)->cnvP));
213a7e14dcfSSatish Balay     if ((*tao)->jacobian_state_inv) {
2149566063dSJacob Faibussowitsch       PetscCall(MatDestroy(&(*tao)->jacobian_state_inv));
215a7e14dcfSSatish Balay     }
216a7e14dcfSSatish Balay   }
2179566063dSJacob Faibussowitsch   PetscCall(VecDestroy(&(*tao)->solution));
2189566063dSJacob Faibussowitsch   PetscCall(VecDestroy(&(*tao)->gradient));
2199566063dSJacob Faibussowitsch   PetscCall(VecDestroy(&(*tao)->ls_res));
220a7e14dcfSSatish Balay 
221a9603a14SPatrick Farrell   if ((*tao)->gradient_norm) {
2229566063dSJacob Faibussowitsch     PetscCall(PetscObjectDereference((PetscObject)(*tao)->gradient_norm));
2239566063dSJacob Faibussowitsch     PetscCall(VecDestroy(&(*tao)->gradient_norm_tmp));
224a9603a14SPatrick Farrell   }
225a9603a14SPatrick Farrell 
2269566063dSJacob Faibussowitsch   PetscCall(VecDestroy(&(*tao)->XL));
2279566063dSJacob Faibussowitsch   PetscCall(VecDestroy(&(*tao)->XU));
2289566063dSJacob Faibussowitsch   PetscCall(VecDestroy(&(*tao)->IL));
2299566063dSJacob Faibussowitsch   PetscCall(VecDestroy(&(*tao)->IU));
2309566063dSJacob Faibussowitsch   PetscCall(VecDestroy(&(*tao)->DE));
2319566063dSJacob Faibussowitsch   PetscCall(VecDestroy(&(*tao)->DI));
2329566063dSJacob Faibussowitsch   PetscCall(VecDestroy(&(*tao)->constraints));
2339566063dSJacob Faibussowitsch   PetscCall(VecDestroy(&(*tao)->constraints_equality));
2349566063dSJacob Faibussowitsch   PetscCall(VecDestroy(&(*tao)->constraints_inequality));
2359566063dSJacob Faibussowitsch   PetscCall(VecDestroy(&(*tao)->stepdirection));
2369566063dSJacob Faibussowitsch   PetscCall(MatDestroy(&(*tao)->hessian_pre));
2379566063dSJacob Faibussowitsch   PetscCall(MatDestroy(&(*tao)->hessian));
2389566063dSJacob Faibussowitsch   PetscCall(MatDestroy(&(*tao)->ls_jac));
2399566063dSJacob Faibussowitsch   PetscCall(MatDestroy(&(*tao)->ls_jac_pre));
2409566063dSJacob Faibussowitsch   PetscCall(MatDestroy(&(*tao)->jacobian_pre));
2419566063dSJacob Faibussowitsch   PetscCall(MatDestroy(&(*tao)->jacobian));
2429566063dSJacob Faibussowitsch   PetscCall(MatDestroy(&(*tao)->jacobian_state_pre));
2439566063dSJacob Faibussowitsch   PetscCall(MatDestroy(&(*tao)->jacobian_state));
2449566063dSJacob Faibussowitsch   PetscCall(MatDestroy(&(*tao)->jacobian_state_inv));
2459566063dSJacob Faibussowitsch   PetscCall(MatDestroy(&(*tao)->jacobian_design));
2469566063dSJacob Faibussowitsch   PetscCall(MatDestroy(&(*tao)->jacobian_equality));
2479566063dSJacob Faibussowitsch   PetscCall(MatDestroy(&(*tao)->jacobian_equality_pre));
2489566063dSJacob Faibussowitsch   PetscCall(MatDestroy(&(*tao)->jacobian_inequality));
2499566063dSJacob Faibussowitsch   PetscCall(MatDestroy(&(*tao)->jacobian_inequality_pre));
2509566063dSJacob Faibussowitsch   PetscCall(ISDestroy(&(*tao)->state_is));
2519566063dSJacob Faibussowitsch   PetscCall(ISDestroy(&(*tao)->design_is));
2529566063dSJacob Faibussowitsch   PetscCall(VecDestroy(&(*tao)->res_weights_v));
2539566063dSJacob Faibussowitsch   PetscCall(TaoCancelMonitors(*tao));
254ae93cb3cSJason Sarich   if ((*tao)->hist_malloc) {
2559566063dSJacob Faibussowitsch     PetscCall(PetscFree4((*tao)->hist_obj,(*tao)->hist_resid,(*tao)->hist_cnorm,(*tao)->hist_lits));
256ae93cb3cSJason Sarich   }
257737f463aSAlp Dener   if ((*tao)->res_weights_n) {
2589566063dSJacob Faibussowitsch     PetscCall(PetscFree((*tao)->res_weights_rows));
2599566063dSJacob Faibussowitsch     PetscCall(PetscFree((*tao)->res_weights_cols));
2609566063dSJacob Faibussowitsch     PetscCall(PetscFree((*tao)->res_weights_w));
261125ddc8aSJason Sarich   }
2629566063dSJacob Faibussowitsch   PetscCall(PetscHeaderDestroy(tao));
263a7e14dcfSSatish Balay   PetscFunctionReturn(0);
264a7e14dcfSSatish Balay }
265a7e14dcfSSatish Balay 
266a7e14dcfSSatish Balay /*@
267441846f8SBarry Smith   TaoSetFromOptions - Sets various Tao parameters from user
268a7e14dcfSSatish Balay   options.
269a7e14dcfSSatish Balay 
270441846f8SBarry Smith   Collective on Tao
271a7e14dcfSSatish Balay 
27201d2d390SJose E. Roman   Input Parameter:
273441846f8SBarry Smith . tao - the Tao solver context
274a7e14dcfSSatish Balay 
275a7e14dcfSSatish Balay   options Database Keys:
27658417fe7SBarry Smith + -tao_type <type> - The algorithm that TAO uses (lmvm, nls, etc.)
277a7e14dcfSSatish Balay . -tao_gatol <gatol> - absolute error tolerance for ||gradient||
278a7e14dcfSSatish Balay . -tao_grtol <grtol> - relative error tolerance for ||gradient||
279a7e14dcfSSatish Balay . -tao_gttol <gttol> - reduction of ||gradient|| relative to initial gradient
280a7e14dcfSSatish Balay . -tao_max_it <max> - sets maximum number of iterations
281a7e14dcfSSatish Balay . -tao_max_funcs <max> - sets maximum number of function evaluations
282a7e14dcfSSatish Balay . -tao_fmin <fmin> - stop if function value reaches fmin
283a7e14dcfSSatish Balay . -tao_steptol <tol> - stop if trust region radius less than <tol>
284a7e14dcfSSatish Balay . -tao_trust0 <t> - initial trust region radius
285a7e14dcfSSatish Balay . -tao_monitor - prints function value and residual at each iteration
286a7e14dcfSSatish Balay . -tao_smonitor - same as tao_monitor, but truncates very small values
287a7e14dcfSSatish Balay . -tao_cmonitor - prints function value, residual, and constraint norm at each iteration
288a7e14dcfSSatish Balay . -tao_view_solution - prints solution vector at each iteration
2894a48860cSAlp Dener . -tao_view_ls_residual - prints least-squares residual vector at each iteration
290147403d9SBarry Smith . -tao_view_stepdirection - prints step direction vector at each iteration
291a7e14dcfSSatish Balay . -tao_view_gradient - prints gradient vector at each iteration
292a7e14dcfSSatish Balay . -tao_draw_solution - graphically view solution vector at each iteration
293a7e14dcfSSatish Balay . -tao_draw_step - graphically view step vector at each iteration
294a7e14dcfSSatish Balay . -tao_draw_gradient - graphically view gradient at each iteration
295a7e14dcfSSatish Balay . -tao_fd_gradient - use gradient computed with finite differences
296f4c1ad5cSStefano Zampini . -tao_fd_hessian - use hessian computed with finite differences
297f4c1ad5cSStefano Zampini . -tao_mf_hessian - use matrix-free hessian computed with finite differences
298a7e14dcfSSatish Balay . -tao_cancelmonitors - cancels all monitors (except those set with command line)
299441846f8SBarry Smith . -tao_view - prints information about the Tao after solving
300a7e14dcfSSatish Balay - -tao_converged_reason - prints the reason TAO stopped iterating
301a7e14dcfSSatish Balay 
302a7e14dcfSSatish Balay   Notes:
303a7e14dcfSSatish Balay   To see all options, run your program with the -help option or consult the
304a7e14dcfSSatish Balay   user's manual. Should be called after TaoCreate() but before TaoSolve()
305a7e14dcfSSatish Balay 
306a7e14dcfSSatish Balay   Level: beginner
307a7e14dcfSSatish Balay @*/
308441846f8SBarry Smith PetscErrorCode TaoSetFromOptions(Tao tao)
309a7e14dcfSSatish Balay {
310b625d6c7SJed Brown   TaoType        default_type = TAOLMVM;
311a7e14dcfSSatish Balay   char           type[256], monfilename[PETSC_MAX_PATH_LEN];
312a7e14dcfSSatish Balay   PetscViewer    monviewer;
313a7e14dcfSSatish Balay   PetscBool      flg;
314a7e14dcfSSatish Balay   MPI_Comm       comm;
315a7e14dcfSSatish Balay 
316a7e14dcfSSatish Balay   PetscFunctionBegin;
317441846f8SBarry Smith   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
3189566063dSJacob Faibussowitsch   PetscCall(PetscObjectGetComm((PetscObject)tao,&comm));
319125ddc8aSJason Sarich 
320a7e14dcfSSatish Balay   /* So no warnings are given about unused options */
3219566063dSJacob Faibussowitsch   PetscCall(PetscOptionsHasName(((PetscObject)tao)->options,((PetscObject)tao)->prefix,"-tao_ls_type",&flg));
322a7e14dcfSSatish Balay 
323d0609cedSBarry Smith   PetscObjectOptionsBegin((PetscObject)tao);
324a7e14dcfSSatish Balay   {
325a82e8c82SStefano Zampini     if (((PetscObject)tao)->type_name) default_type = ((PetscObject)tao)->type_name;
326a7e14dcfSSatish Balay     /* Check for type from options */
3279566063dSJacob Faibussowitsch     PetscCall(PetscOptionsFList("-tao_type","Tao Solver type","TaoSetType",TaoList,default_type,type,256,&flg));
328a7e14dcfSSatish Balay     if (flg) {
3299566063dSJacob Faibussowitsch       PetscCall(TaoSetType(tao,type));
330a7e14dcfSSatish Balay     } else if (!((PetscObject)tao)->type_name) {
3319566063dSJacob Faibussowitsch       PetscCall(TaoSetType(tao,default_type));
332a7e14dcfSSatish Balay     }
333a7e14dcfSSatish Balay 
3349566063dSJacob Faibussowitsch     PetscCall(PetscOptionsReal("-tao_catol","Stop if constraints violations within","TaoSetConstraintTolerances",tao->catol,&tao->catol,&flg));
3356552cf8aSJason Sarich     if (flg) tao->catol_changed = PETSC_TRUE;
3369566063dSJacob Faibussowitsch     PetscCall(PetscOptionsReal("-tao_crtol","Stop if relative contraint violations within","TaoSetConstraintTolerances",tao->crtol,&tao->crtol,&flg));
3376552cf8aSJason Sarich     if (flg) tao->crtol_changed = PETSC_TRUE;
3389566063dSJacob Faibussowitsch     PetscCall(PetscOptionsReal("-tao_gatol","Stop if norm of gradient less than","TaoSetTolerances",tao->gatol,&tao->gatol,&flg));
3396552cf8aSJason Sarich     if (flg) tao->gatol_changed = PETSC_TRUE;
3409566063dSJacob Faibussowitsch     PetscCall(PetscOptionsReal("-tao_grtol","Stop if norm of gradient divided by the function value is less than","TaoSetTolerances",tao->grtol,&tao->grtol,&flg));
3416552cf8aSJason Sarich     if (flg) tao->grtol_changed = PETSC_TRUE;
3429566063dSJacob 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));
3436552cf8aSJason Sarich     if (flg) tao->gttol_changed = PETSC_TRUE;
3449566063dSJacob Faibussowitsch     PetscCall(PetscOptionsInt("-tao_max_it","Stop if iteration number exceeds","TaoSetMaximumIterations",tao->max_it,&tao->max_it,&flg));
3456552cf8aSJason Sarich     if (flg) tao->max_it_changed = PETSC_TRUE;
3469566063dSJacob Faibussowitsch     PetscCall(PetscOptionsInt("-tao_max_funcs","Stop if number of function evaluations exceeds","TaoSetMaximumFunctionEvaluations",tao->max_funcs,&tao->max_funcs,&flg));
3476552cf8aSJason Sarich     if (flg) tao->max_funcs_changed = PETSC_TRUE;
3489566063dSJacob Faibussowitsch     PetscCall(PetscOptionsReal("-tao_fmin","Stop if function less than","TaoSetFunctionLowerBound",tao->fmin,&tao->fmin,&flg));
3496552cf8aSJason Sarich     if (flg) tao->fmin_changed = PETSC_TRUE;
3509566063dSJacob Faibussowitsch     PetscCall(PetscOptionsReal("-tao_steptol","Stop if step size or trust region radius less than","",tao->steptol,&tao->steptol,&flg));
3516552cf8aSJason Sarich     if (flg) tao->steptol_changed = PETSC_TRUE;
3529566063dSJacob Faibussowitsch     PetscCall(PetscOptionsReal("-tao_trust0","Initial trust region radius","TaoSetTrustRegionRadius",tao->trust0,&tao->trust0,&flg));
3536552cf8aSJason Sarich     if (flg) tao->trust0_changed = PETSC_TRUE;
3549566063dSJacob Faibussowitsch     PetscCall(PetscOptionsString("-tao_view_solution","view solution vector after each evaluation","TaoSetMonitor","stdout",monfilename,sizeof(monfilename),&flg));
355a7e14dcfSSatish Balay     if (flg) {
3569566063dSJacob Faibussowitsch       PetscCall(PetscViewerASCIIOpen(comm,monfilename,&monviewer));
3579566063dSJacob Faibussowitsch       PetscCall(TaoSetMonitor(tao,TaoSolutionMonitor,monviewer,(PetscErrorCode (*)(void**))PetscViewerDestroy));
358a7e14dcfSSatish Balay     }
359a7e14dcfSSatish Balay 
3609566063dSJacob Faibussowitsch     PetscCall(PetscOptionsBool("-tao_converged_reason","Print reason for TAO converged","TaoSolve",tao->printreason,&tao->printreason,NULL));
3619566063dSJacob Faibussowitsch     PetscCall(PetscOptionsString("-tao_view_gradient","view gradient vector after each evaluation","TaoSetMonitor","stdout",monfilename,sizeof(monfilename),&flg));
362a7e14dcfSSatish Balay     if (flg) {
3639566063dSJacob Faibussowitsch       PetscCall(PetscViewerASCIIOpen(comm,monfilename,&monviewer));
3649566063dSJacob Faibussowitsch       PetscCall(TaoSetMonitor(tao,TaoGradientMonitor,monviewer,(PetscErrorCode (*)(void**))PetscViewerDestroy));
365a7e14dcfSSatish Balay     }
366a7e14dcfSSatish Balay 
3679566063dSJacob Faibussowitsch     PetscCall(PetscOptionsString("-tao_view_stepdirection","view step direction vector after each iteration","TaoSetMonitor","stdout",monfilename,sizeof(monfilename),&flg));
368a7e14dcfSSatish Balay     if (flg) {
3699566063dSJacob Faibussowitsch       PetscCall(PetscViewerASCIIOpen(comm,monfilename,&monviewer));
3709566063dSJacob Faibussowitsch       PetscCall(TaoSetMonitor(tao,TaoStepDirectionMonitor,monviewer,(PetscErrorCode (*)(void**))PetscViewerDestroy));
371a7e14dcfSSatish Balay     }
372a7e14dcfSSatish Balay 
3739566063dSJacob Faibussowitsch     PetscCall(PetscOptionsString("-tao_view_residual","view least-squares residual vector after each evaluation","TaoSetMonitor","stdout",monfilename,sizeof(monfilename),&flg));
374a7e14dcfSSatish Balay     if (flg) {
3759566063dSJacob Faibussowitsch       PetscCall(PetscViewerASCIIOpen(comm,monfilename,&monviewer));
3769566063dSJacob Faibussowitsch       PetscCall(TaoSetMonitor(tao,TaoResidualMonitor,monviewer,(PetscErrorCode (*)(void**))PetscViewerDestroy));
377a7e14dcfSSatish Balay     }
378a7e14dcfSSatish Balay 
3799566063dSJacob Faibussowitsch     PetscCall(PetscOptionsString("-tao_monitor","Use the default convergence monitor","TaoSetMonitor","stdout",monfilename,sizeof(monfilename),&flg));
380a7e14dcfSSatish Balay     if (flg) {
3819566063dSJacob Faibussowitsch       PetscCall(PetscViewerASCIIOpen(comm,monfilename,&monviewer));
3829566063dSJacob Faibussowitsch       PetscCall(TaoSetMonitor(tao,TaoMonitorDefault,monviewer,(PetscErrorCode (*)(void**))PetscViewerDestroy));
383a7e14dcfSSatish Balay     }
384a7e14dcfSSatish Balay 
3859566063dSJacob Faibussowitsch     PetscCall(PetscOptionsString("-tao_gmonitor","Use the convergence monitor with extra globalization info","TaoSetMonitor","stdout",monfilename,sizeof(monfilename),&flg));
3868d5ead36SAlp Dener     if (flg) {
3879566063dSJacob Faibussowitsch       PetscCall(PetscViewerASCIIOpen(comm,monfilename,&monviewer));
3889566063dSJacob Faibussowitsch       PetscCall(TaoSetMonitor(tao,TaoDefaultGMonitor,monviewer,(PetscErrorCode (*)(void**))PetscViewerDestroy));
3898d5ead36SAlp Dener     }
3908d5ead36SAlp Dener 
3919566063dSJacob Faibussowitsch     PetscCall(PetscOptionsString("-tao_smonitor","Use the short convergence monitor","TaoSetMonitor","stdout",monfilename,sizeof(monfilename),&flg));
392a7e14dcfSSatish Balay     if (flg) {
3939566063dSJacob Faibussowitsch       PetscCall(PetscViewerASCIIOpen(comm,monfilename,&monviewer));
3949566063dSJacob Faibussowitsch       PetscCall(TaoSetMonitor(tao,TaoDefaultSMonitor,monviewer,(PetscErrorCode (*)(void**))PetscViewerDestroy));
395a7e14dcfSSatish Balay     }
396a7e14dcfSSatish Balay 
3979566063dSJacob Faibussowitsch     PetscCall(PetscOptionsString("-tao_cmonitor","Use the default convergence monitor with constraint norm","TaoSetMonitor","stdout",monfilename,sizeof(monfilename),&flg));
398a7e14dcfSSatish Balay     if (flg) {
3999566063dSJacob Faibussowitsch       PetscCall(PetscViewerASCIIOpen(comm,monfilename,&monviewer));
4009566063dSJacob Faibussowitsch       PetscCall(TaoSetMonitor(tao,TaoDefaultCMonitor,monviewer,(PetscErrorCode (*)(void**))PetscViewerDestroy));
401a7e14dcfSSatish Balay     }
402a7e14dcfSSatish Balay 
4038afaa268SBarry Smith     flg = PETSC_FALSE;
4049566063dSJacob Faibussowitsch     PetscCall(PetscOptionsBool("-tao_cancelmonitors","cancel all monitors and call any registered destroy routines","TaoCancelMonitors",flg,&flg,NULL));
4059566063dSJacob Faibussowitsch     if (flg) PetscCall(TaoCancelMonitors(tao));
406a7e14dcfSSatish Balay 
4078afaa268SBarry Smith     flg = PETSC_FALSE;
4089566063dSJacob Faibussowitsch     PetscCall(PetscOptionsBool("-tao_draw_solution","Plot solution vector at each iteration","TaoSetMonitor",flg,&flg,NULL));
409a7e14dcfSSatish Balay     if (flg) {
410e882e171SHong Zhang       TaoMonitorDrawCtx drawctx;
411e882e171SHong Zhang       PetscInt          howoften = 1;
4129566063dSJacob Faibussowitsch       PetscCall(TaoMonitorDrawCtxCreate(PetscObjectComm((PetscObject)tao),NULL,NULL,PETSC_DECIDE,PETSC_DECIDE,300,300,howoften,&drawctx));
4139566063dSJacob Faibussowitsch       PetscCall(TaoSetMonitor(tao,TaoDrawSolutionMonitor,drawctx,(PetscErrorCode (*)(void**))TaoMonitorDrawCtxDestroy));
414a7e14dcfSSatish Balay     }
415a7e14dcfSSatish Balay 
4168afaa268SBarry Smith     flg = PETSC_FALSE;
4179566063dSJacob Faibussowitsch     PetscCall(PetscOptionsBool("-tao_draw_step","plots step direction at each iteration","TaoSetMonitor",flg,&flg,NULL));
418a7e14dcfSSatish Balay     if (flg) {
4199566063dSJacob Faibussowitsch       PetscCall(TaoSetMonitor(tao,TaoDrawStepMonitor,NULL,NULL));
420a7e14dcfSSatish Balay     }
421a7e14dcfSSatish Balay 
4228afaa268SBarry Smith     flg = PETSC_FALSE;
4239566063dSJacob Faibussowitsch     PetscCall(PetscOptionsBool("-tao_draw_gradient","plots gradient at each iteration","TaoSetMonitor",flg,&flg,NULL));
424a7e14dcfSSatish Balay     if (flg) {
425e882e171SHong Zhang       TaoMonitorDrawCtx drawctx;
426e882e171SHong Zhang       PetscInt          howoften = 1;
4279566063dSJacob Faibussowitsch       PetscCall(TaoMonitorDrawCtxCreate(PetscObjectComm((PetscObject)tao),NULL,NULL,PETSC_DECIDE,PETSC_DECIDE,300,300,howoften,&drawctx));
4289566063dSJacob Faibussowitsch       PetscCall(TaoSetMonitor(tao,TaoDrawGradientMonitor,drawctx,(PetscErrorCode (*)(void**))TaoMonitorDrawCtxDestroy));
429a7e14dcfSSatish Balay     }
4308afaa268SBarry Smith     flg = PETSC_FALSE;
4319566063dSJacob Faibussowitsch     PetscCall(PetscOptionsBool("-tao_fd_gradient","compute gradient using finite differences","TaoDefaultComputeGradient",flg,&flg,NULL));
432a7e14dcfSSatish Balay     if (flg) {
4339566063dSJacob Faibussowitsch       PetscCall(TaoSetGradient(tao,NULL,TaoDefaultComputeGradient,NULL));
434a7e14dcfSSatish Balay     }
435f4c1ad5cSStefano Zampini     flg = PETSC_FALSE;
4369566063dSJacob Faibussowitsch     PetscCall(PetscOptionsBool("-tao_fd_hessian","compute hessian using finite differences","TaoDefaultComputeHessian",flg,&flg,NULL));
437f4c1ad5cSStefano Zampini     if (flg) {
438f4c1ad5cSStefano Zampini       Mat H;
439f4c1ad5cSStefano Zampini 
4409566063dSJacob Faibussowitsch       PetscCall(MatCreate(PetscObjectComm((PetscObject)tao),&H));
4419566063dSJacob Faibussowitsch       PetscCall(MatSetType(H,MATAIJ));
4429566063dSJacob Faibussowitsch       PetscCall(TaoSetHessian(tao,H,H,TaoDefaultComputeHessian,NULL));
4439566063dSJacob Faibussowitsch       PetscCall(MatDestroy(&H));
444f4c1ad5cSStefano Zampini     }
445f4c1ad5cSStefano Zampini     flg = PETSC_FALSE;
4469566063dSJacob Faibussowitsch     PetscCall(PetscOptionsBool("-tao_mf_hessian","compute matrix-free hessian using finite differences","TaoDefaultComputeHessianMFFD",flg,&flg,NULL));
447f4c1ad5cSStefano Zampini     if (flg) {
448f4c1ad5cSStefano Zampini       Mat H;
449f4c1ad5cSStefano Zampini 
4509566063dSJacob Faibussowitsch       PetscCall(MatCreate(PetscObjectComm((PetscObject)tao),&H));
4519566063dSJacob Faibussowitsch       PetscCall(TaoSetHessian(tao,H,H,TaoDefaultComputeHessianMFFD,NULL));
4529566063dSJacob Faibussowitsch       PetscCall(MatDestroy(&H));
453f4c1ad5cSStefano Zampini     }
454414d97d3SAlp Dener     flg = PETSC_FALSE;
4559566063dSJacob Faibussowitsch     PetscCall(PetscOptionsBool("-tao_recycle_history","enable recycling/re-using information from the previous TaoSolve() call for some algorithms","TaoSetRecycleHistory",flg,&flg,NULL));
456414d97d3SAlp Dener     if (flg) {
4579566063dSJacob Faibussowitsch       PetscCall(TaoSetRecycleHistory(tao,PETSC_TRUE));
458414d97d3SAlp Dener     }
4599566063dSJacob Faibussowitsch     PetscCall(PetscOptionsEnum("-tao_subset_type","subset type","",TaoSubSetTypes,(PetscEnum)tao->subset_type,(PetscEnum*)&tao->subset_type,NULL));
460a7e14dcfSSatish Balay 
461a82e8c82SStefano Zampini     if (tao->linesearch) {
4629566063dSJacob Faibussowitsch       PetscCall(TaoLineSearchSetFromOptions(tao->linesearch));
463a82e8c82SStefano Zampini     }
464a82e8c82SStefano Zampini 
465a7e14dcfSSatish Balay     if (tao->ops->setfromoptions) {
4669566063dSJacob Faibussowitsch       PetscCall((*tao->ops->setfromoptions)(PetscOptionsObject,tao));
467a7e14dcfSSatish Balay     }
468a7e14dcfSSatish Balay   }
469d0609cedSBarry Smith   PetscOptionsEnd();
470a7e14dcfSSatish Balay   PetscFunctionReturn(0);
471a7e14dcfSSatish Balay }
472a7e14dcfSSatish Balay 
473a7e14dcfSSatish Balay /*@C
474fe2efc57SMark    TaoViewFromOptions - View from Options
475fe2efc57SMark 
476fe2efc57SMark    Collective on Tao
477fe2efc57SMark 
478fe2efc57SMark    Input Parameters:
479fe2efc57SMark +  A - the  Tao context
480736c3998SJose E. Roman .  obj - Optional object
481736c3998SJose E. Roman -  name - command line option
482fe2efc57SMark 
483fe2efc57SMark    Level: intermediate
484db781477SPatrick Sanan .seealso: `Tao`, `TaoView`, `PetscObjectViewFromOptions()`, `TaoCreate()`
485fe2efc57SMark @*/
486fe2efc57SMark PetscErrorCode  TaoViewFromOptions(Tao A,PetscObject obj,const char name[])
487fe2efc57SMark {
488fe2efc57SMark   PetscFunctionBegin;
489fe2efc57SMark   PetscValidHeaderSpecific(A,TAO_CLASSID,1);
4909566063dSJacob Faibussowitsch   PetscCall(PetscObjectViewFromOptions((PetscObject)A,obj,name));
491fe2efc57SMark   PetscFunctionReturn(0);
492fe2efc57SMark }
493fe2efc57SMark 
494fe2efc57SMark /*@C
495441846f8SBarry Smith   TaoView - Prints information about the Tao
496a7e14dcfSSatish Balay 
497441846f8SBarry Smith   Collective on Tao
498a7e14dcfSSatish Balay 
499a7e14dcfSSatish Balay   InputParameters:
500441846f8SBarry Smith + tao - the Tao context
501a7e14dcfSSatish Balay - viewer - visualization context
502a7e14dcfSSatish Balay 
503a7e14dcfSSatish Balay   Options Database Key:
504a7e14dcfSSatish Balay . -tao_view - Calls TaoView() at the end of TaoSolve()
505a7e14dcfSSatish Balay 
506a7e14dcfSSatish Balay   Notes:
507a7e14dcfSSatish Balay   The available visualization contexts include
508a7e14dcfSSatish Balay +     PETSC_VIEWER_STDOUT_SELF - standard output (default)
509a7e14dcfSSatish Balay -     PETSC_VIEWER_STDOUT_WORLD - synchronized standard
510a7e14dcfSSatish Balay          output where only the first processor opens
511a7e14dcfSSatish Balay          the file.  All other processors send their
512a7e14dcfSSatish Balay          data to the first processor to print.
513a7e14dcfSSatish Balay 
514a7e14dcfSSatish Balay   Level: beginner
515a7e14dcfSSatish Balay 
516db781477SPatrick Sanan .seealso: `PetscViewerASCIIOpen()`
517a7e14dcfSSatish Balay @*/
518441846f8SBarry Smith PetscErrorCode TaoView(Tao tao, PetscViewer viewer)
519a7e14dcfSSatish Balay {
520a7e14dcfSSatish Balay   PetscBool           isascii,isstring;
521b625d6c7SJed Brown   TaoType             type;
52247a47007SBarry Smith 
523a7e14dcfSSatish Balay   PetscFunctionBegin;
524441846f8SBarry Smith   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
525a7e14dcfSSatish Balay   if (!viewer) {
5269566063dSJacob Faibussowitsch     PetscCall(PetscViewerASCIIGetStdout(((PetscObject)tao)->comm,&viewer));
527a7e14dcfSSatish Balay   }
528a7e14dcfSSatish Balay   PetscValidHeaderSpecific(viewer,PETSC_VIEWER_CLASSID,2);
529a7e14dcfSSatish Balay   PetscCheckSameComm(tao,1,viewer,2);
530a7e14dcfSSatish Balay 
5319566063dSJacob Faibussowitsch   PetscCall(PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERASCII,&isascii));
5329566063dSJacob Faibussowitsch   PetscCall(PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERSTRING,&isstring));
533a7e14dcfSSatish Balay   if (isascii) {
5349566063dSJacob Faibussowitsch     PetscCall(PetscObjectPrintClassNamePrefixType((PetscObject)tao,viewer));
535a7e14dcfSSatish Balay 
536a7e14dcfSSatish Balay     if (tao->ops->view) {
5379566063dSJacob Faibussowitsch       PetscCall(PetscViewerASCIIPushTab(viewer));
5389566063dSJacob Faibussowitsch       PetscCall((*tao->ops->view)(tao,viewer));
5399566063dSJacob Faibussowitsch       PetscCall(PetscViewerASCIIPopTab(viewer));
540a7e14dcfSSatish Balay     }
5418d3f3ddaSAlp Dener     if (tao->linesearch) {
5429566063dSJacob Faibussowitsch       PetscCall(PetscViewerASCIIPushTab(viewer));
5439566063dSJacob Faibussowitsch       PetscCall(TaoLineSearchView(tao->linesearch,viewer));
5449566063dSJacob Faibussowitsch       PetscCall(PetscViewerASCIIPopTab(viewer));
5458d3f3ddaSAlp Dener     }
546a7e14dcfSSatish Balay     if (tao->ksp) {
5479566063dSJacob Faibussowitsch       PetscCall(PetscViewerASCIIPushTab(viewer));
5489566063dSJacob Faibussowitsch       PetscCall(KSPView(tao->ksp,viewer));
54963a3b9bcSJacob Faibussowitsch       PetscCall(PetscViewerASCIIPrintf(viewer,"total KSP iterations: %" PetscInt_FMT "\n",tao->ksp_tot_its));
5509566063dSJacob Faibussowitsch       PetscCall(PetscViewerASCIIPopTab(viewer));
551a7e14dcfSSatish Balay     }
55219b5c101SAlp Dener 
5539566063dSJacob Faibussowitsch     PetscCall(PetscViewerASCIIPushTab(viewer));
55419b5c101SAlp Dener 
555a7e14dcfSSatish Balay     if (tao->XL || tao->XU) {
5569566063dSJacob Faibussowitsch       PetscCall(PetscViewerASCIIPrintf(viewer,"Active Set subset type: %s\n",TaoSubSetTypes[tao->subset_type]));
557a7e14dcfSSatish Balay     }
558a7e14dcfSSatish Balay 
5599566063dSJacob Faibussowitsch     PetscCall(PetscViewerASCIIPrintf(viewer,"convergence tolerances: gatol=%g,",(double)tao->gatol));
5609566063dSJacob Faibussowitsch     PetscCall(PetscViewerASCIIPrintf(viewer," steptol=%g,",(double)tao->steptol));
5619566063dSJacob Faibussowitsch     PetscCall(PetscViewerASCIIPrintf(viewer," gttol=%g\n",(double)tao->gttol));
5629566063dSJacob Faibussowitsch     PetscCall(PetscViewerASCIIPrintf(viewer,"Residual in Function/Gradient:=%g\n",(double)tao->residual));
563a7e14dcfSSatish Balay 
5646246e8cdSAlp Dener     if (tao->constrained) {
5659566063dSJacob Faibussowitsch       PetscCall(PetscViewerASCIIPrintf(viewer,"convergence tolerances:"));
5669566063dSJacob Faibussowitsch       PetscCall(PetscViewerASCIIPrintf(viewer," catol=%g,",(double)tao->catol));
5679566063dSJacob Faibussowitsch       PetscCall(PetscViewerASCIIPrintf(viewer," crtol=%g\n",(double)tao->crtol));
5689566063dSJacob Faibussowitsch       PetscCall(PetscViewerASCIIPrintf(viewer,"Residual in Constraints:=%g\n",(double)tao->cnorm));
569a7e14dcfSSatish Balay     }
570a7e14dcfSSatish Balay 
571a7e14dcfSSatish Balay     if (tao->trust < tao->steptol) {
5729566063dSJacob Faibussowitsch       PetscCall(PetscViewerASCIIPrintf(viewer,"convergence tolerances: steptol=%g\n",(double)tao->steptol));
5739566063dSJacob Faibussowitsch       PetscCall(PetscViewerASCIIPrintf(viewer,"Final trust region radius:=%g\n",(double)tao->trust));
574a7e14dcfSSatish Balay     }
575a7e14dcfSSatish Balay 
576a7e14dcfSSatish Balay     if (tao->fmin>-1.e25) {
5779566063dSJacob Faibussowitsch       PetscCall(PetscViewerASCIIPrintf(viewer,"convergence tolerances: function minimum=%g\n",(double)tao->fmin));
578a7e14dcfSSatish Balay     }
5799566063dSJacob Faibussowitsch     PetscCall(PetscViewerASCIIPrintf(viewer,"Objective value=%g\n",(double)tao->fc));
580a7e14dcfSSatish Balay 
58163a3b9bcSJacob Faibussowitsch     PetscCall(PetscViewerASCIIPrintf(viewer,"total number of iterations=%" PetscInt_FMT ",          ",tao->niter));
58263a3b9bcSJacob Faibussowitsch     PetscCall(PetscViewerASCIIPrintf(viewer,"              (max: %" PetscInt_FMT ")\n",tao->max_it));
583a7e14dcfSSatish Balay 
584a7e14dcfSSatish Balay     if (tao->nfuncs>0) {
58563a3b9bcSJacob Faibussowitsch       PetscCall(PetscViewerASCIIPrintf(viewer,"total number of function evaluations=%" PetscInt_FMT ",",tao->nfuncs));
58663a3b9bcSJacob Faibussowitsch       PetscCall(PetscViewerASCIIPrintf(viewer,"                max: %" PetscInt_FMT "\n",tao->max_funcs));
587a7e14dcfSSatish Balay     }
588a7e14dcfSSatish Balay     if (tao->ngrads>0) {
58963a3b9bcSJacob Faibussowitsch       PetscCall(PetscViewerASCIIPrintf(viewer,"total number of gradient evaluations=%" PetscInt_FMT ",",tao->ngrads));
59063a3b9bcSJacob Faibussowitsch       PetscCall(PetscViewerASCIIPrintf(viewer,"                max: %" PetscInt_FMT "\n",tao->max_funcs));
591a7e14dcfSSatish Balay     }
592a7e14dcfSSatish Balay     if (tao->nfuncgrads>0) {
59363a3b9bcSJacob Faibussowitsch       PetscCall(PetscViewerASCIIPrintf(viewer,"total number of function/gradient evaluations=%" PetscInt_FMT ",",tao->nfuncgrads));
59463a3b9bcSJacob Faibussowitsch       PetscCall(PetscViewerASCIIPrintf(viewer,"    (max: %" PetscInt_FMT ")\n",tao->max_funcs));
595a7e14dcfSSatish Balay     }
596a7e14dcfSSatish Balay     if (tao->nhess>0) {
59763a3b9bcSJacob Faibussowitsch       PetscCall(PetscViewerASCIIPrintf(viewer,"total number of Hessian evaluations=%" PetscInt_FMT "\n",tao->nhess));
598a7e14dcfSSatish Balay     }
599a7e14dcfSSatish Balay     if (tao->nconstraints>0) {
60063a3b9bcSJacob Faibussowitsch       PetscCall(PetscViewerASCIIPrintf(viewer,"total number of constraint function evaluations=%" PetscInt_FMT "\n",tao->nconstraints));
601a7e14dcfSSatish Balay     }
602a7e14dcfSSatish Balay     if (tao->njac>0) {
60363a3b9bcSJacob Faibussowitsch       PetscCall(PetscViewerASCIIPrintf(viewer,"total number of Jacobian evaluations=%" PetscInt_FMT "\n",tao->njac));
604a7e14dcfSSatish Balay     }
605a7e14dcfSSatish Balay 
606a7e14dcfSSatish Balay     if (tao->reason>0) {
6079566063dSJacob Faibussowitsch       PetscCall(PetscViewerASCIIPrintf(viewer,    "Solution converged: "));
608a7e14dcfSSatish Balay       switch (tao->reason) {
609a7e14dcfSSatish Balay       case TAO_CONVERGED_GATOL:
6109566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer," ||g(X)|| <= gatol\n"));
611a7e14dcfSSatish Balay         break;
612a7e14dcfSSatish Balay       case TAO_CONVERGED_GRTOL:
6139566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer," ||g(X)||/|f(X)| <= grtol\n"));
614a7e14dcfSSatish Balay         break;
615a7e14dcfSSatish Balay       case TAO_CONVERGED_GTTOL:
6169566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer," ||g(X)||/||g(X0)|| <= gttol\n"));
617a7e14dcfSSatish Balay         break;
618a7e14dcfSSatish Balay       case TAO_CONVERGED_STEPTOL:
6199566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer," Steptol -- step size small\n"));
620a7e14dcfSSatish Balay         break;
621a7e14dcfSSatish Balay       case TAO_CONVERGED_MINF:
6229566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer," Minf --  f < fmin\n"));
623a7e14dcfSSatish Balay         break;
624a7e14dcfSSatish Balay       case TAO_CONVERGED_USER:
6259566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer," User Terminated\n"));
626a7e14dcfSSatish Balay         break;
627a7e14dcfSSatish Balay       default:
6289566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer,"\n"));
629a7e14dcfSSatish Balay         break;
630a7e14dcfSSatish Balay       }
631a7e14dcfSSatish Balay     } else {
6329566063dSJacob Faibussowitsch       PetscCall(PetscViewerASCIIPrintf(viewer,"Solver terminated: %d",tao->reason));
633a7e14dcfSSatish Balay       switch (tao->reason) {
634a7e14dcfSSatish Balay       case TAO_DIVERGED_MAXITS:
6359566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer," Maximum Iterations\n"));
636a7e14dcfSSatish Balay         break;
637a7e14dcfSSatish Balay       case TAO_DIVERGED_NAN:
6389566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer," NAN or Inf encountered\n"));
639a7e14dcfSSatish Balay         break;
640a7e14dcfSSatish Balay       case TAO_DIVERGED_MAXFCN:
6419566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer," Maximum Function Evaluations\n"));
642a7e14dcfSSatish Balay         break;
643a7e14dcfSSatish Balay       case TAO_DIVERGED_LS_FAILURE:
6449566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer," Line Search Failure\n"));
645a7e14dcfSSatish Balay         break;
646a7e14dcfSSatish Balay       case TAO_DIVERGED_TR_REDUCTION:
6479566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer," Trust Region too small\n"));
648a7e14dcfSSatish Balay         break;
649a7e14dcfSSatish Balay       case TAO_DIVERGED_USER:
6509566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer," User Terminated\n"));
651a7e14dcfSSatish Balay         break;
652a7e14dcfSSatish Balay       default:
6539566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer,"\n"));
654a7e14dcfSSatish Balay         break;
655a7e14dcfSSatish Balay       }
656a7e14dcfSSatish Balay     }
6579566063dSJacob Faibussowitsch     PetscCall(PetscViewerASCIIPopTab(viewer));
658a7e14dcfSSatish Balay   } else if (isstring) {
6599566063dSJacob Faibussowitsch     PetscCall(TaoGetType(tao,&type));
6609566063dSJacob Faibussowitsch     PetscCall(PetscViewerStringSPrintf(viewer," %-3.3s",type));
661a7e14dcfSSatish Balay   }
662a7e14dcfSSatish Balay   PetscFunctionReturn(0);
663a7e14dcfSSatish Balay }
664a7e14dcfSSatish Balay 
665a7e14dcfSSatish Balay /*@
666414d97d3SAlp Dener   TaoSetRecycleHistory - Sets the boolean flag to enable/disable re-using
667414d97d3SAlp Dener   iterate information from the previous TaoSolve(). This feature is disabled by
668414d97d3SAlp Dener   default.
669414d97d3SAlp Dener 
670414d97d3SAlp Dener   For conjugate gradient methods (BNCG), this re-uses the latest search direction
671414d97d3SAlp Dener   from the previous TaoSolve() call when computing the first search direction in a
672414d97d3SAlp Dener   new solution. By default, CG methods set the first search direction to the
673414d97d3SAlp Dener   negative gradient.
674414d97d3SAlp Dener 
675414d97d3SAlp Dener   For quasi-Newton family of methods (BQNLS, BQNKLS, BQNKTR, BQNKTL), this re-uses
676414d97d3SAlp Dener   the accumulated quasi-Newton Hessian approximation from the previous TaoSolve()
677414d97d3SAlp Dener   call. By default, QN family of methods reset the initial Hessian approximation to
678414d97d3SAlp Dener   the identity matrix.
679414d97d3SAlp Dener 
680414d97d3SAlp Dener   For any other algorithm, this setting has no effect.
681414d97d3SAlp Dener 
682414d97d3SAlp Dener   Logically collective on Tao
683414d97d3SAlp Dener 
684414d97d3SAlp Dener   Input Parameters:
685414d97d3SAlp Dener + tao - the Tao context
686414d97d3SAlp Dener - recycle - boolean flag
687414d97d3SAlp Dener 
688414d97d3SAlp Dener   Options Database Keys:
689147403d9SBarry Smith . -tao_recycle_history <true,false> - reuse the history
690414d97d3SAlp Dener 
691414d97d3SAlp Dener   Level: intermediate
692414d97d3SAlp Dener 
693db781477SPatrick Sanan .seealso: `TaoSetRecycleHistory()`, `TAOBNCG`, `TAOBQNLS`, `TAOBQNKLS`, `TAOBQNKTR`, `TAOBQNKTL`
694414d97d3SAlp Dener 
695414d97d3SAlp Dener @*/
696414d97d3SAlp Dener PetscErrorCode TaoSetRecycleHistory(Tao tao, PetscBool recycle)
697414d97d3SAlp Dener {
698414d97d3SAlp Dener   PetscFunctionBegin;
699414d97d3SAlp Dener   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
70094511df7SStefano Zampini   PetscValidLogicalCollectiveBool(tao,recycle,2);
701414d97d3SAlp Dener   tao->recycle = recycle;
702414d97d3SAlp Dener   PetscFunctionReturn(0);
703414d97d3SAlp Dener }
704414d97d3SAlp Dener 
705414d97d3SAlp Dener /*@
706414d97d3SAlp Dener   TaoGetRecycleHistory - Retrieve the boolean flag for re-using iterate information
707414d97d3SAlp Dener   from the previous TaoSolve(). This feature is disabled by default.
708414d97d3SAlp Dener 
709414d97d3SAlp Dener   Logically collective on Tao
710414d97d3SAlp Dener 
711414d97d3SAlp Dener   Input Parameters:
712147403d9SBarry Smith . tao - the Tao context
713414d97d3SAlp Dener 
714414d97d3SAlp Dener   Output Parameters:
715147403d9SBarry Smith . recycle - boolean flag
716414d97d3SAlp Dener 
717414d97d3SAlp Dener   Level: intermediate
718414d97d3SAlp Dener 
719db781477SPatrick Sanan .seealso: `TaoGetRecycleHistory()`, `TAOBNCG`, `TAOBQNLS`, `TAOBQNKLS`, `TAOBQNKTR`, `TAOBQNKTL`
720414d97d3SAlp Dener 
721414d97d3SAlp Dener @*/
722414d97d3SAlp Dener PetscErrorCode TaoGetRecycleHistory(Tao tao, PetscBool *recycle)
723414d97d3SAlp Dener {
724414d97d3SAlp Dener   PetscFunctionBegin;
725414d97d3SAlp Dener   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
72694511df7SStefano Zampini   PetscValidBoolPointer(recycle,2);
727414d97d3SAlp Dener   *recycle = tao->recycle;
728414d97d3SAlp Dener   PetscFunctionReturn(0);
729414d97d3SAlp Dener }
730414d97d3SAlp Dener 
731414d97d3SAlp Dener /*@
732a7e14dcfSSatish Balay   TaoSetTolerances - Sets parameters used in TAO convergence tests
733a7e14dcfSSatish Balay 
734441846f8SBarry Smith   Logically collective on Tao
735a7e14dcfSSatish Balay 
736a7e14dcfSSatish Balay   Input Parameters:
737441846f8SBarry Smith + tao - the Tao context
738a7e14dcfSSatish Balay . gatol - stop if norm of gradient is less than this
739a7e14dcfSSatish Balay . grtol - stop if relative norm of gradient is less than this
740a7e14dcfSSatish Balay - gttol - stop if norm of gradient is reduced by this factor
741a7e14dcfSSatish Balay 
742a7e14dcfSSatish Balay   Options Database Keys:
743e52336cbSBarry Smith + -tao_gatol <gatol> - Sets gatol
744a7e14dcfSSatish Balay . -tao_grtol <grtol> - Sets grtol
745a7e14dcfSSatish Balay - -tao_gttol <gttol> - Sets gttol
746a7e14dcfSSatish Balay 
747a7e14dcfSSatish Balay   Stopping Criteria:
748a7e14dcfSSatish Balay $ ||g(X)||                            <= gatol
749a7e14dcfSSatish Balay $ ||g(X)|| / |f(X)|                   <= grtol
750a7e14dcfSSatish Balay $ ||g(X)|| / ||g(X0)||                <= gttol
751a7e14dcfSSatish Balay 
752a7e14dcfSSatish Balay   Notes:
753a7e14dcfSSatish Balay   Use PETSC_DEFAULT to leave one or more tolerances unchanged.
754a7e14dcfSSatish Balay 
755a7e14dcfSSatish Balay   Level: beginner
756a7e14dcfSSatish Balay 
757db781477SPatrick Sanan .seealso: `TaoGetTolerances()`
758a7e14dcfSSatish Balay 
759a7e14dcfSSatish Balay @*/
760e52336cbSBarry Smith PetscErrorCode TaoSetTolerances(Tao tao, PetscReal gatol, PetscReal grtol, PetscReal gttol)
761a7e14dcfSSatish Balay {
762a7e14dcfSSatish Balay   PetscFunctionBegin;
763441846f8SBarry Smith   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
76494511df7SStefano Zampini   PetscValidLogicalCollectiveReal(tao,gatol,2);
76594511df7SStefano Zampini   PetscValidLogicalCollectiveReal(tao,grtol,3);
76694511df7SStefano Zampini   PetscValidLogicalCollectiveReal(tao,gttol,4);
767a7e14dcfSSatish Balay 
768a7e14dcfSSatish Balay   if (gatol != PETSC_DEFAULT) {
769a7e14dcfSSatish Balay     if (gatol<0) {
7709566063dSJacob Faibussowitsch       PetscCall(PetscInfo(tao,"Tried to set negative gatol -- ignored.\n"));
771a7e14dcfSSatish Balay     } else {
772a7e14dcfSSatish Balay       tao->gatol = PetscMax(0,gatol);
7736552cf8aSJason Sarich       tao->gatol_changed = PETSC_TRUE;
774a7e14dcfSSatish Balay     }
775a7e14dcfSSatish Balay   }
776a7e14dcfSSatish Balay 
777a7e14dcfSSatish Balay   if (grtol != PETSC_DEFAULT) {
778a7e14dcfSSatish Balay     if (grtol<0) {
7799566063dSJacob Faibussowitsch       PetscCall(PetscInfo(tao,"Tried to set negative grtol -- ignored.\n"));
780a7e14dcfSSatish Balay     } else {
781a7e14dcfSSatish Balay       tao->grtol = PetscMax(0,grtol);
7826552cf8aSJason Sarich       tao->grtol_changed = PETSC_TRUE;
783a7e14dcfSSatish Balay     }
784a7e14dcfSSatish Balay   }
785a7e14dcfSSatish Balay 
786a7e14dcfSSatish Balay   if (gttol != PETSC_DEFAULT) {
787a7e14dcfSSatish Balay     if (gttol<0) {
7889566063dSJacob Faibussowitsch       PetscCall(PetscInfo(tao,"Tried to set negative gttol -- ignored.\n"));
789a7e14dcfSSatish Balay     } else {
790a7e14dcfSSatish Balay       tao->gttol = PetscMax(0,gttol);
7916552cf8aSJason Sarich       tao->gttol_changed = PETSC_TRUE;
792a7e14dcfSSatish Balay     }
793a7e14dcfSSatish Balay   }
794a7e14dcfSSatish Balay   PetscFunctionReturn(0);
795a7e14dcfSSatish Balay }
796a7e14dcfSSatish Balay 
797a7e14dcfSSatish Balay /*@
79858417fe7SBarry Smith   TaoSetConstraintTolerances - Sets constraint tolerance parameters used in TAO  convergence tests
799a7e14dcfSSatish Balay 
800441846f8SBarry Smith   Logically collective on Tao
801a7e14dcfSSatish Balay 
802a7e14dcfSSatish Balay   Input Parameters:
803441846f8SBarry Smith + tao - the Tao context
804e52336cbSBarry Smith . catol - absolute constraint tolerance, constraint norm must be less than catol for used for gatol convergence criteria
805e52336cbSBarry Smith - crtol - relative contraint tolerance, constraint norm must be less than crtol for used for gatol, gttol convergence criteria
806a7e14dcfSSatish Balay 
807a7e14dcfSSatish Balay   Options Database Keys:
808a7e14dcfSSatish Balay + -tao_catol <catol> - Sets catol
809a7e14dcfSSatish Balay - -tao_crtol <crtol> - Sets crtol
810a7e14dcfSSatish Balay 
8116552cf8aSJason Sarich   Notes:
8126552cf8aSJason Sarich   Use PETSC_DEFAULT to leave any tolerance unchanged.
8136552cf8aSJason Sarich 
814a7e14dcfSSatish Balay   Level: intermediate
815a7e14dcfSSatish Balay 
816db781477SPatrick Sanan .seealso: `TaoGetTolerances()`, `TaoGetConstraintTolerances()`, `TaoSetTolerances()`
817a7e14dcfSSatish Balay 
818a7e14dcfSSatish Balay @*/
819441846f8SBarry Smith PetscErrorCode TaoSetConstraintTolerances(Tao tao, PetscReal catol, PetscReal crtol)
820a7e14dcfSSatish Balay {
821a7e14dcfSSatish Balay   PetscFunctionBegin;
822441846f8SBarry Smith   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
82394511df7SStefano Zampini   PetscValidLogicalCollectiveReal(tao,catol,2);
82494511df7SStefano Zampini   PetscValidLogicalCollectiveReal(tao,crtol,3);
825a7e14dcfSSatish Balay 
826a7e14dcfSSatish Balay   if (catol != PETSC_DEFAULT) {
827a7e14dcfSSatish Balay     if (catol<0) {
8289566063dSJacob Faibussowitsch       PetscCall(PetscInfo(tao,"Tried to set negative catol -- ignored.\n"));
829a7e14dcfSSatish Balay     } else {
830a7e14dcfSSatish Balay       tao->catol = PetscMax(0,catol);
8316552cf8aSJason Sarich       tao->catol_changed = PETSC_TRUE;
832a7e14dcfSSatish Balay     }
833a7e14dcfSSatish Balay   }
834a7e14dcfSSatish Balay 
835a7e14dcfSSatish Balay   if (crtol != PETSC_DEFAULT) {
836a7e14dcfSSatish Balay     if (crtol<0) {
8379566063dSJacob Faibussowitsch       PetscCall(PetscInfo(tao,"Tried to set negative crtol -- ignored.\n"));
838a7e14dcfSSatish Balay     } else {
839a7e14dcfSSatish Balay       tao->crtol = PetscMax(0,crtol);
8406552cf8aSJason Sarich       tao->crtol_changed = PETSC_TRUE;
841a7e14dcfSSatish Balay     }
842a7e14dcfSSatish Balay   }
843a7e14dcfSSatish Balay   PetscFunctionReturn(0);
844a7e14dcfSSatish Balay }
845a7e14dcfSSatish Balay 
84658417fe7SBarry Smith /*@
84758417fe7SBarry Smith   TaoGetConstraintTolerances - Gets constraint tolerance parameters used in TAO  convergence tests
84858417fe7SBarry Smith 
84958417fe7SBarry Smith   Not ollective
85058417fe7SBarry Smith 
85158417fe7SBarry Smith   Input Parameter:
85258417fe7SBarry Smith . tao - the Tao context
85358417fe7SBarry Smith 
854d8d19677SJose E. Roman   Output Parameters:
855e52336cbSBarry Smith + catol - absolute constraint tolerance, constraint norm must be less than catol for used for gatol convergence criteria
856e52336cbSBarry Smith - crtol - relative contraint tolerance, constraint norm must be less than crtol for used for gatol, gttol convergence criteria
85758417fe7SBarry Smith 
85858417fe7SBarry Smith   Level: intermediate
85958417fe7SBarry Smith 
860db781477SPatrick Sanan .seealso: `TaoGetTolerances()`, `TaoSetTolerances()`, `TaoSetConstraintTolerances()`
86158417fe7SBarry Smith 
86258417fe7SBarry Smith @*/
86358417fe7SBarry Smith PetscErrorCode TaoGetConstraintTolerances(Tao tao, PetscReal *catol, PetscReal *crtol)
86458417fe7SBarry Smith {
86558417fe7SBarry Smith   PetscFunctionBegin;
86658417fe7SBarry Smith   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
86758417fe7SBarry Smith   if (catol) *catol = tao->catol;
86858417fe7SBarry Smith   if (crtol) *crtol = tao->crtol;
86958417fe7SBarry Smith   PetscFunctionReturn(0);
87058417fe7SBarry Smith }
87158417fe7SBarry Smith 
872a7e14dcfSSatish Balay /*@
873a7e14dcfSSatish Balay    TaoSetFunctionLowerBound - Sets a bound on the solution objective value.
874a7e14dcfSSatish Balay    When an approximate solution with an objective value below this number
875a7e14dcfSSatish Balay    has been found, the solver will terminate.
876a7e14dcfSSatish Balay 
877441846f8SBarry Smith    Logically Collective on Tao
878a7e14dcfSSatish Balay 
879a7e14dcfSSatish Balay    Input Parameters:
880441846f8SBarry Smith +  tao - the Tao solver context
881a7e14dcfSSatish Balay -  fmin - the tolerance
882a7e14dcfSSatish Balay 
883a7e14dcfSSatish Balay    Options Database Keys:
884a7e14dcfSSatish Balay .    -tao_fmin <fmin> - sets the minimum function value
885a7e14dcfSSatish Balay 
886a7e14dcfSSatish Balay    Level: intermediate
887a7e14dcfSSatish Balay 
888db781477SPatrick Sanan .seealso: `TaoSetTolerances()`
889a7e14dcfSSatish Balay @*/
890441846f8SBarry Smith PetscErrorCode TaoSetFunctionLowerBound(Tao tao,PetscReal fmin)
891a7e14dcfSSatish Balay {
892a7e14dcfSSatish Balay   PetscFunctionBegin;
893441846f8SBarry Smith   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
89494511df7SStefano Zampini   PetscValidLogicalCollectiveReal(tao,fmin,2);
895a7e14dcfSSatish Balay   tao->fmin = fmin;
8966552cf8aSJason Sarich   tao->fmin_changed = PETSC_TRUE;
897a7e14dcfSSatish Balay   PetscFunctionReturn(0);
898a7e14dcfSSatish Balay }
899a7e14dcfSSatish Balay 
900a7e14dcfSSatish Balay /*@
9018e96d397SJason Sarich    TaoGetFunctionLowerBound - Gets the bound on the solution objective value.
902a7e14dcfSSatish Balay    When an approximate solution with an objective value below this number
903a7e14dcfSSatish Balay    has been found, the solver will terminate.
904a7e14dcfSSatish Balay 
905441846f8SBarry Smith    Not collective on Tao
906a7e14dcfSSatish Balay 
907a7e14dcfSSatish Balay    Input Parameters:
908441846f8SBarry Smith .  tao - the Tao solver context
909a7e14dcfSSatish Balay 
910a7e14dcfSSatish Balay    OutputParameters:
911a7e14dcfSSatish Balay .  fmin - the minimum function value
912a7e14dcfSSatish Balay 
913a7e14dcfSSatish Balay    Level: intermediate
914a7e14dcfSSatish Balay 
915db781477SPatrick Sanan .seealso: `TaoSetFunctionLowerBound()`
916a7e14dcfSSatish Balay @*/
917441846f8SBarry Smith PetscErrorCode TaoGetFunctionLowerBound(Tao tao,PetscReal *fmin)
918a7e14dcfSSatish Balay {
919a7e14dcfSSatish Balay   PetscFunctionBegin;
920441846f8SBarry Smith   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
92194511df7SStefano Zampini   PetscValidRealPointer(fmin,2);
922a7e14dcfSSatish Balay   *fmin = tao->fmin;
923a7e14dcfSSatish Balay   PetscFunctionReturn(0);
924a7e14dcfSSatish Balay }
925a7e14dcfSSatish Balay 
926a7e14dcfSSatish Balay /*@
927a7e14dcfSSatish Balay    TaoSetMaximumFunctionEvaluations - Sets a maximum number of
928a7e14dcfSSatish Balay    function evaluations.
929a7e14dcfSSatish Balay 
930441846f8SBarry Smith    Logically Collective on Tao
931a7e14dcfSSatish Balay 
932a7e14dcfSSatish Balay    Input Parameters:
933441846f8SBarry Smith +  tao - the Tao solver context
934a7e14dcfSSatish Balay -  nfcn - the maximum number of function evaluations (>=0)
935a7e14dcfSSatish Balay 
936a7e14dcfSSatish Balay    Options Database Keys:
937a7e14dcfSSatish Balay .    -tao_max_funcs <nfcn> - sets the maximum number of function evaluations
938a7e14dcfSSatish Balay 
939a7e14dcfSSatish Balay    Level: intermediate
940a7e14dcfSSatish Balay 
941db781477SPatrick Sanan .seealso: `TaoSetTolerances()`, `TaoSetMaximumIterations()`
942a7e14dcfSSatish Balay @*/
943a7e14dcfSSatish Balay 
944441846f8SBarry Smith PetscErrorCode TaoSetMaximumFunctionEvaluations(Tao tao,PetscInt nfcn)
945a7e14dcfSSatish Balay {
946a7e14dcfSSatish Balay   PetscFunctionBegin;
947441846f8SBarry Smith   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
94894511df7SStefano Zampini   PetscValidLogicalCollectiveInt(tao,nfcn,2);
94994511df7SStefano Zampini   if (nfcn >= 0) { tao->max_funcs = PetscMax(0,nfcn); }
95094511df7SStefano Zampini   else { tao->max_funcs = -1; }
9516552cf8aSJason Sarich   tao->max_funcs_changed = PETSC_TRUE;
952a7e14dcfSSatish Balay   PetscFunctionReturn(0);
953a7e14dcfSSatish Balay }
954a7e14dcfSSatish Balay 
955a7e14dcfSSatish Balay /*@
956a7e14dcfSSatish Balay    TaoGetMaximumFunctionEvaluations - Sets a maximum number of
957a7e14dcfSSatish Balay    function evaluations.
958a7e14dcfSSatish Balay 
959a7e14dcfSSatish Balay    Not Collective
960a7e14dcfSSatish Balay 
961a7e14dcfSSatish Balay    Input Parameters:
962441846f8SBarry Smith .  tao - the Tao solver context
963a7e14dcfSSatish Balay 
964a7e14dcfSSatish Balay    Output Parameters:
965a7e14dcfSSatish Balay .  nfcn - the maximum number of function evaluations
966a7e14dcfSSatish Balay 
967a7e14dcfSSatish Balay    Level: intermediate
968a7e14dcfSSatish Balay 
969db781477SPatrick Sanan .seealso: `TaoSetMaximumFunctionEvaluations()`, `TaoGetMaximumIterations()`
970a7e14dcfSSatish Balay @*/
971a7e14dcfSSatish Balay 
972441846f8SBarry Smith PetscErrorCode TaoGetMaximumFunctionEvaluations(Tao tao,PetscInt *nfcn)
973a7e14dcfSSatish Balay {
974a7e14dcfSSatish Balay   PetscFunctionBegin;
975441846f8SBarry Smith   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
97694511df7SStefano Zampini   PetscValidIntPointer(nfcn,2);
977a7e14dcfSSatish Balay   *nfcn = tao->max_funcs;
978a7e14dcfSSatish Balay   PetscFunctionReturn(0);
979a7e14dcfSSatish Balay }
980a7e14dcfSSatish Balay 
981770232b9SCe Qin /*@
982770232b9SCe Qin    TaoGetCurrentFunctionEvaluations - Get current number of
983770232b9SCe Qin    function evaluations.
984770232b9SCe Qin 
985770232b9SCe Qin    Not Collective
986770232b9SCe Qin 
987770232b9SCe Qin    Input Parameters:
988770232b9SCe Qin .  tao - the Tao solver context
989770232b9SCe Qin 
990770232b9SCe Qin    Output Parameters:
99194511df7SStefano Zampini .  nfuncs - the current number of function evaluations (maximum between gradient and function evaluations)
992770232b9SCe Qin 
993770232b9SCe Qin    Level: intermediate
994770232b9SCe Qin 
995db781477SPatrick Sanan .seealso: `TaoSetMaximumFunctionEvaluations()`, `TaoGetMaximumFunctionEvaluations()`, `TaoGetMaximumIterations()`
996770232b9SCe Qin @*/
997770232b9SCe Qin 
998770232b9SCe Qin PetscErrorCode TaoGetCurrentFunctionEvaluations(Tao tao,PetscInt *nfuncs)
999770232b9SCe Qin {
1000770232b9SCe Qin   PetscFunctionBegin;
1001770232b9SCe Qin   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
100294511df7SStefano Zampini   PetscValidIntPointer(nfuncs,2);
1003770232b9SCe Qin   *nfuncs = PetscMax(tao->nfuncs,tao->nfuncgrads);
1004770232b9SCe Qin   PetscFunctionReturn(0);
1005770232b9SCe Qin }
1006770232b9SCe Qin 
1007a7e14dcfSSatish Balay /*@
1008a7e14dcfSSatish Balay    TaoSetMaximumIterations - Sets a maximum number of iterates.
1009a7e14dcfSSatish Balay 
1010441846f8SBarry Smith    Logically Collective on Tao
1011a7e14dcfSSatish Balay 
1012a7e14dcfSSatish Balay    Input Parameters:
1013441846f8SBarry Smith +  tao - the Tao solver context
1014a7e14dcfSSatish Balay -  maxits - the maximum number of iterates (>=0)
1015a7e14dcfSSatish Balay 
1016a7e14dcfSSatish Balay    Options Database Keys:
1017a7e14dcfSSatish Balay .    -tao_max_it <its> - sets the maximum number of iterations
1018a7e14dcfSSatish Balay 
1019a7e14dcfSSatish Balay    Level: intermediate
1020a7e14dcfSSatish Balay 
1021db781477SPatrick Sanan .seealso: `TaoSetTolerances()`, `TaoSetMaximumFunctionEvaluations()`
1022a7e14dcfSSatish Balay @*/
1023441846f8SBarry Smith PetscErrorCode TaoSetMaximumIterations(Tao tao,PetscInt maxits)
1024a7e14dcfSSatish Balay {
1025a7e14dcfSSatish Balay   PetscFunctionBegin;
1026441846f8SBarry Smith   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
102794511df7SStefano Zampini   PetscValidLogicalCollectiveInt(tao,maxits,2);
102847a47007SBarry Smith   tao->max_it = PetscMax(0,maxits);
10296552cf8aSJason Sarich   tao->max_it_changed = PETSC_TRUE;
1030a7e14dcfSSatish Balay   PetscFunctionReturn(0);
1031a7e14dcfSSatish Balay }
1032a7e14dcfSSatish Balay 
1033a7e14dcfSSatish Balay /*@
1034a7e14dcfSSatish Balay    TaoGetMaximumIterations - Sets a maximum number of iterates.
1035a7e14dcfSSatish Balay 
1036a7e14dcfSSatish Balay    Not Collective
1037a7e14dcfSSatish Balay 
1038a7e14dcfSSatish Balay    Input Parameters:
1039441846f8SBarry Smith .  tao - the Tao solver context
1040a7e14dcfSSatish Balay 
1041a7e14dcfSSatish Balay    Output Parameters:
1042a7e14dcfSSatish Balay .  maxits - the maximum number of iterates
1043a7e14dcfSSatish Balay 
1044a7e14dcfSSatish Balay    Level: intermediate
1045a7e14dcfSSatish Balay 
1046db781477SPatrick Sanan .seealso: `TaoSetMaximumIterations()`, `TaoGetMaximumFunctionEvaluations()`
1047a7e14dcfSSatish Balay @*/
1048441846f8SBarry Smith PetscErrorCode TaoGetMaximumIterations(Tao tao,PetscInt *maxits)
1049a7e14dcfSSatish Balay {
1050a7e14dcfSSatish Balay   PetscFunctionBegin;
1051441846f8SBarry Smith   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
105294511df7SStefano Zampini   PetscValidIntPointer(maxits,2);
1053a7e14dcfSSatish Balay   *maxits = tao->max_it;
1054a7e14dcfSSatish Balay   PetscFunctionReturn(0);
1055a7e14dcfSSatish Balay }
1056a7e14dcfSSatish Balay 
1057a7e14dcfSSatish Balay /*@
1058a7e14dcfSSatish Balay    TaoSetInitialTrustRegionRadius - Sets the initial trust region radius.
1059a7e14dcfSSatish Balay 
1060441846f8SBarry Smith    Logically collective on Tao
1061a7e14dcfSSatish Balay 
1062d8d19677SJose E. Roman    Input Parameters:
1063a7e14dcfSSatish Balay +  tao - a TAO optimization solver
1064a7e14dcfSSatish Balay -  radius - the trust region radius
1065a7e14dcfSSatish Balay 
1066a7e14dcfSSatish Balay    Level: intermediate
1067a7e14dcfSSatish Balay 
1068a7e14dcfSSatish Balay    Options Database Key:
1069a7e14dcfSSatish Balay .  -tao_trust0 <t0> - sets initial trust region radius
1070a7e14dcfSSatish Balay 
1071db781477SPatrick Sanan .seealso: `TaoGetTrustRegionRadius()`, `TaoSetTrustRegionTolerance()`
1072a7e14dcfSSatish Balay @*/
1073441846f8SBarry Smith PetscErrorCode TaoSetInitialTrustRegionRadius(Tao tao, PetscReal radius)
1074a7e14dcfSSatish Balay {
1075a7e14dcfSSatish Balay   PetscFunctionBegin;
1076441846f8SBarry Smith   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
107794511df7SStefano Zampini   PetscValidLogicalCollectiveReal(tao,radius,2);
1078a7e14dcfSSatish Balay   tao->trust0 = PetscMax(0.0,radius);
10796552cf8aSJason Sarich   tao->trust0_changed = PETSC_TRUE;
1080a7e14dcfSSatish Balay   PetscFunctionReturn(0);
1081a7e14dcfSSatish Balay }
1082a7e14dcfSSatish Balay 
1083a7e14dcfSSatish Balay /*@
1084a7e14dcfSSatish Balay    TaoGetInitialTrustRegionRadius - Sets the initial trust region radius.
1085a7e14dcfSSatish Balay 
1086a7e14dcfSSatish Balay    Not Collective
1087a7e14dcfSSatish Balay 
1088a7e14dcfSSatish Balay    Input Parameter:
1089a7e14dcfSSatish Balay .  tao - a TAO optimization solver
1090a7e14dcfSSatish Balay 
1091a7e14dcfSSatish Balay    Output Parameter:
1092a7e14dcfSSatish Balay .  radius - the trust region radius
1093a7e14dcfSSatish Balay 
1094a7e14dcfSSatish Balay    Level: intermediate
1095a7e14dcfSSatish Balay 
1096db781477SPatrick Sanan .seealso: `TaoSetInitialTrustRegionRadius()`, `TaoGetCurrentTrustRegionRadius()`
1097a7e14dcfSSatish Balay @*/
1098441846f8SBarry Smith PetscErrorCode TaoGetInitialTrustRegionRadius(Tao tao, PetscReal *radius)
1099a7e14dcfSSatish Balay {
1100a7e14dcfSSatish Balay   PetscFunctionBegin;
1101441846f8SBarry Smith   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
110294511df7SStefano Zampini   PetscValidRealPointer(radius,2);
1103a7e14dcfSSatish Balay   *radius = tao->trust0;
1104a7e14dcfSSatish Balay   PetscFunctionReturn(0);
1105a7e14dcfSSatish Balay }
1106a7e14dcfSSatish Balay 
1107a7e14dcfSSatish Balay /*@
1108a7e14dcfSSatish Balay    TaoGetCurrentTrustRegionRadius - Gets the current trust region radius.
1109a7e14dcfSSatish Balay 
1110a7e14dcfSSatish Balay    Not Collective
1111a7e14dcfSSatish Balay 
1112a7e14dcfSSatish Balay    Input Parameter:
1113a7e14dcfSSatish Balay .  tao - a TAO optimization solver
1114a7e14dcfSSatish Balay 
1115a7e14dcfSSatish Balay    Output Parameter:
1116a7e14dcfSSatish Balay .  radius - the trust region radius
1117a7e14dcfSSatish Balay 
1118a7e14dcfSSatish Balay    Level: intermediate
1119a7e14dcfSSatish Balay 
1120db781477SPatrick Sanan .seealso: `TaoSetInitialTrustRegionRadius()`, `TaoGetInitialTrustRegionRadius()`
1121a7e14dcfSSatish Balay @*/
1122441846f8SBarry Smith PetscErrorCode TaoGetCurrentTrustRegionRadius(Tao tao, PetscReal *radius)
1123a7e14dcfSSatish Balay {
1124a7e14dcfSSatish Balay   PetscFunctionBegin;
1125441846f8SBarry Smith   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
112694511df7SStefano Zampini   PetscValidRealPointer(radius,2);
1127a7e14dcfSSatish Balay   *radius = tao->trust;
1128a7e14dcfSSatish Balay   PetscFunctionReturn(0);
1129a7e14dcfSSatish Balay }
1130a7e14dcfSSatish Balay 
1131a7e14dcfSSatish Balay /*@
1132a7e14dcfSSatish Balay   TaoGetTolerances - gets the current values of tolerances
1133a7e14dcfSSatish Balay 
1134a7e14dcfSSatish Balay   Not Collective
1135a7e14dcfSSatish Balay 
1136f899ff85SJose E. Roman   Input Parameter:
1137441846f8SBarry Smith . tao - the Tao context
1138a7e14dcfSSatish Balay 
1139a7e14dcfSSatish Balay   Output Parameters:
1140e52336cbSBarry Smith + gatol - stop if norm of gradient is less than this
1141a7e14dcfSSatish Balay . grtol - stop if relative norm of gradient is less than this
1142a7e14dcfSSatish Balay - gttol - stop if norm of gradient is reduced by a this factor
1143a7e14dcfSSatish Balay 
114447a47007SBarry Smith   Note: NULL can be used as an argument if not all tolerances values are needed
1145a7e14dcfSSatish Balay 
1146db781477SPatrick Sanan .seealso `TaoSetTolerances()`
1147a7e14dcfSSatish Balay 
1148a7e14dcfSSatish Balay   Level: intermediate
1149a7e14dcfSSatish Balay @*/
1150e52336cbSBarry Smith PetscErrorCode TaoGetTolerances(Tao tao, PetscReal *gatol, PetscReal *grtol, PetscReal *gttol)
1151a7e14dcfSSatish Balay {
1152a7e14dcfSSatish Balay   PetscFunctionBegin;
1153441846f8SBarry Smith   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
1154a7e14dcfSSatish Balay   if (gatol) *gatol = tao->gatol;
1155a7e14dcfSSatish Balay   if (grtol) *grtol = tao->grtol;
1156a7e14dcfSSatish Balay   if (gttol) *gttol = tao->gttol;
1157a7e14dcfSSatish Balay   PetscFunctionReturn(0);
1158a7e14dcfSSatish Balay }
1159a7e14dcfSSatish Balay 
1160a7e14dcfSSatish Balay /*@
1161a7e14dcfSSatish Balay   TaoGetKSP - Gets the linear solver used by the optimization solver.
1162a7e14dcfSSatish Balay   Application writers should use TaoGetKSP if they need direct access
1163a7e14dcfSSatish Balay   to the PETSc KSP object.
1164a7e14dcfSSatish Balay 
1165a7e14dcfSSatish Balay   Not Collective
1166a7e14dcfSSatish Balay 
1167a7e14dcfSSatish Balay    Input Parameters:
1168a7e14dcfSSatish Balay .  tao - the TAO solver
1169a7e14dcfSSatish Balay 
1170a7e14dcfSSatish Balay    Output Parameters:
1171a7e14dcfSSatish Balay .  ksp - the KSP linear solver used in the optimization solver
1172a7e14dcfSSatish Balay 
1173a7e14dcfSSatish Balay    Level: intermediate
1174a7e14dcfSSatish Balay 
1175a7e14dcfSSatish Balay @*/
1176441846f8SBarry Smith PetscErrorCode TaoGetKSP(Tao tao, KSP *ksp)
117747a47007SBarry Smith {
1178a7e14dcfSSatish Balay   PetscFunctionBegin;
117994511df7SStefano Zampini   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
118094511df7SStefano Zampini   PetscValidPointer(ksp,2);
1181a7e14dcfSSatish Balay   *ksp = tao->ksp;
1182a7e14dcfSSatish Balay   PetscFunctionReturn(0);
1183a7e14dcfSSatish Balay }
1184a7e14dcfSSatish Balay 
1185025e9500SJason Sarich /*@
1186025e9500SJason Sarich    TaoGetLinearSolveIterations - Gets the total number of linear iterations
1187025e9500SJason Sarich    used by the TAO solver
1188025e9500SJason Sarich 
1189025e9500SJason Sarich    Not Collective
1190025e9500SJason Sarich 
1191025e9500SJason Sarich    Input Parameter:
1192025e9500SJason Sarich .  tao - TAO context
1193025e9500SJason Sarich 
1194025e9500SJason Sarich    Output Parameter:
1195025e9500SJason Sarich .  lits - number of linear iterations
1196025e9500SJason Sarich 
1197025e9500SJason Sarich    Notes:
1198025e9500SJason Sarich    This counter is reset to zero for each successive call to TaoSolve()
1199025e9500SJason Sarich 
1200025e9500SJason Sarich    Level: intermediate
1201025e9500SJason Sarich 
1202db781477SPatrick Sanan .seealso: `TaoGetKSP()`
1203025e9500SJason Sarich @*/
1204025e9500SJason Sarich PetscErrorCode TaoGetLinearSolveIterations(Tao tao, PetscInt *lits)
1205025e9500SJason Sarich {
1206025e9500SJason Sarich   PetscFunctionBegin;
1207025e9500SJason Sarich   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
1208025e9500SJason Sarich   PetscValidIntPointer(lits,2);
12092d9aa51bSJason Sarich   *lits = tao->ksp_tot_its;
1210025e9500SJason Sarich   PetscFunctionReturn(0);
1211025e9500SJason Sarich }
1212025e9500SJason Sarich 
1213a7e14dcfSSatish Balay /*@
1214a7e14dcfSSatish Balay   TaoGetLineSearch - Gets the line search used by the optimization solver.
1215a7e14dcfSSatish Balay   Application writers should use TaoGetLineSearch if they need direct access
1216a7e14dcfSSatish Balay   to the TaoLineSearch object.
1217a7e14dcfSSatish Balay 
1218a7e14dcfSSatish Balay   Not Collective
1219a7e14dcfSSatish Balay 
1220a7e14dcfSSatish Balay    Input Parameters:
1221a7e14dcfSSatish Balay .  tao - the TAO solver
1222a7e14dcfSSatish Balay 
1223a7e14dcfSSatish Balay    Output Parameters:
1224a7e14dcfSSatish Balay .  ls - the line search used in the optimization solver
1225a7e14dcfSSatish Balay 
1226a7e14dcfSSatish Balay    Level: intermediate
1227a7e14dcfSSatish Balay 
1228a7e14dcfSSatish Balay @*/
1229441846f8SBarry Smith PetscErrorCode TaoGetLineSearch(Tao tao, TaoLineSearch *ls)
123047a47007SBarry Smith {
1231a7e14dcfSSatish Balay   PetscFunctionBegin;
123294511df7SStefano Zampini   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
123394511df7SStefano Zampini   PetscValidPointer(ls,2);
1234a7e14dcfSSatish Balay   *ls = tao->linesearch;
1235a7e14dcfSSatish Balay   PetscFunctionReturn(0);
1236a7e14dcfSSatish Balay }
1237a7e14dcfSSatish Balay 
1238a7e14dcfSSatish Balay /*@
1239a7e14dcfSSatish Balay   TaoAddLineSearchCounts - Adds the number of function evaluations spent
1240a7e14dcfSSatish Balay   in the line search to the running total.
1241a7e14dcfSSatish Balay 
1242a7e14dcfSSatish Balay    Input Parameters:
1243a7e14dcfSSatish Balay +  tao - the TAO solver
1244a7e14dcfSSatish Balay -  ls - the line search used in the optimization solver
1245a7e14dcfSSatish Balay 
1246a7e14dcfSSatish Balay    Level: developer
1247a7e14dcfSSatish Balay 
1248db781477SPatrick Sanan .seealso: `TaoLineSearchApply()`
1249a7e14dcfSSatish Balay @*/
1250441846f8SBarry Smith PetscErrorCode TaoAddLineSearchCounts(Tao tao)
125147a47007SBarry Smith {
1252a7e14dcfSSatish Balay   PetscBool      flg;
1253a7e14dcfSSatish Balay   PetscInt       nfeval,ngeval,nfgeval;
125447a47007SBarry Smith 
1255a7e14dcfSSatish Balay   PetscFunctionBegin;
1256441846f8SBarry Smith   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
1257a7e14dcfSSatish Balay   if (tao->linesearch) {
12589566063dSJacob Faibussowitsch     PetscCall(TaoLineSearchIsUsingTaoRoutines(tao->linesearch,&flg));
125994ae4db5SBarry Smith     if (!flg) {
12609566063dSJacob Faibussowitsch       PetscCall(TaoLineSearchGetNumberFunctionEvaluations(tao->linesearch,&nfeval,&ngeval,&nfgeval));
1261a7e14dcfSSatish Balay       tao->nfuncs += nfeval;
1262a7e14dcfSSatish Balay       tao->ngrads += ngeval;
1263a7e14dcfSSatish Balay       tao->nfuncgrads += nfgeval;
1264a7e14dcfSSatish Balay     }
1265a7e14dcfSSatish Balay   }
1266a7e14dcfSSatish Balay   PetscFunctionReturn(0);
1267a7e14dcfSSatish Balay }
1268a7e14dcfSSatish Balay 
1269a7e14dcfSSatish Balay /*@
1270a82e8c82SStefano Zampini   TaoGetSolution - Returns the vector with the current TAO solution
1271a7e14dcfSSatish Balay 
1272a7e14dcfSSatish Balay   Not Collective
1273a7e14dcfSSatish Balay 
1274a7e14dcfSSatish Balay   Input Parameter:
1275441846f8SBarry Smith . tao - the Tao context
1276a7e14dcfSSatish Balay 
1277a7e14dcfSSatish Balay   Output Parameter:
1278a7e14dcfSSatish Balay . X - the current solution
1279a7e14dcfSSatish Balay 
1280a7e14dcfSSatish Balay   Level: intermediate
1281a7e14dcfSSatish Balay 
1282a82e8c82SStefano Zampini   Note:  The returned vector will be the same object that was passed into TaoSetSolution()
1283a7e14dcfSSatish Balay @*/
1284a82e8c82SStefano Zampini PetscErrorCode TaoGetSolution(Tao tao, Vec *X)
1285a7e14dcfSSatish Balay {
1286a7e14dcfSSatish Balay   PetscFunctionBegin;
1287441846f8SBarry Smith   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
128894511df7SStefano Zampini   PetscValidPointer(X,2);
1289a7e14dcfSSatish Balay   *X = tao->solution;
1290a7e14dcfSSatish Balay   PetscFunctionReturn(0);
1291a7e14dcfSSatish Balay }
1292a7e14dcfSSatish Balay 
1293a7e14dcfSSatish Balay /*@
1294a7e14dcfSSatish Balay    TaoResetStatistics - Initialize the statistics used by TAO for all of the solvers.
1295a7e14dcfSSatish Balay    These statistics include the iteration number, residual norms, and convergence status.
1296a7e14dcfSSatish Balay    This routine gets called before solving each optimization problem.
1297a7e14dcfSSatish Balay 
1298441846f8SBarry Smith    Collective on Tao
1299a7e14dcfSSatish Balay 
1300a7e14dcfSSatish Balay    Input Parameters:
1301441846f8SBarry Smith .  solver - the Tao context
1302a7e14dcfSSatish Balay 
1303a7e14dcfSSatish Balay    Level: developer
1304a7e14dcfSSatish Balay 
1305db781477SPatrick Sanan .seealso: `TaoCreate()`, `TaoSolve()`
1306a7e14dcfSSatish Balay @*/
1307441846f8SBarry Smith PetscErrorCode TaoResetStatistics(Tao tao)
1308a7e14dcfSSatish Balay {
1309a7e14dcfSSatish Balay   PetscFunctionBegin;
1310441846f8SBarry Smith   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
1311a7e14dcfSSatish Balay   tao->niter        = 0;
1312a7e14dcfSSatish Balay   tao->nfuncs       = 0;
1313a7e14dcfSSatish Balay   tao->nfuncgrads   = 0;
1314a7e14dcfSSatish Balay   tao->ngrads       = 0;
1315a7e14dcfSSatish Balay   tao->nhess        = 0;
1316a7e14dcfSSatish Balay   tao->njac         = 0;
1317a7e14dcfSSatish Balay   tao->nconstraints = 0;
1318a7e14dcfSSatish Balay   tao->ksp_its      = 0;
1319ae93cb3cSJason Sarich   tao->ksp_tot_its  = 0;
1320a7e14dcfSSatish Balay   tao->reason       = TAO_CONTINUE_ITERATING;
1321a7e14dcfSSatish Balay   tao->residual     = 0.0;
1322a7e14dcfSSatish Balay   tao->cnorm        = 0.0;
1323a7e14dcfSSatish Balay   tao->step         = 0.0;
1324a7e14dcfSSatish Balay   tao->lsflag       = PETSC_FALSE;
1325a7e14dcfSSatish Balay   if (tao->hist_reset) tao->hist_len = 0;
1326a7e14dcfSSatish Balay   PetscFunctionReturn(0);
1327a7e14dcfSSatish Balay }
1328a7e14dcfSSatish Balay 
1329a7e14dcfSSatish Balay /*@C
1330e1e80dc8SAlp Dener   TaoSetUpdate - Sets the general-purpose update function called
1331e1e80dc8SAlp Dener   at the beginning of every iteration of the nonlinear solve. Specifically
1332e1e80dc8SAlp Dener   it is called at the top of every iteration, after the new solution and the gradient
1333e1e80dc8SAlp Dener   is determined, but before the Hessian is computed (if applicable).
1334e1e80dc8SAlp Dener 
1335e1e80dc8SAlp Dener   Logically Collective on Tao
1336e1e80dc8SAlp Dener 
1337e1e80dc8SAlp Dener   Input Parameters:
1338a2b725a8SWilliam Gropp + tao - The tao solver context
1339a2b725a8SWilliam Gropp - func - The function
1340e1e80dc8SAlp Dener 
1341e1e80dc8SAlp Dener   Calling sequence of func:
1342a2b725a8SWilliam Gropp $ func (Tao tao, PetscInt step);
1343e1e80dc8SAlp Dener 
1344e1e80dc8SAlp Dener . step - The current step of the iteration
1345e1e80dc8SAlp Dener 
1346e1e80dc8SAlp Dener   Level: advanced
1347e1e80dc8SAlp Dener 
1348db781477SPatrick Sanan .seealso `TaoSolve()`
1349e1e80dc8SAlp Dener @*/
13508fcddce6SStefano Zampini PetscErrorCode TaoSetUpdate(Tao tao, PetscErrorCode (*func)(Tao, PetscInt, void*), void *ctx)
1351e1e80dc8SAlp Dener {
1352e1e80dc8SAlp Dener   PetscFunctionBegin;
1353e1e80dc8SAlp Dener   PetscValidHeaderSpecific(tao, TAO_CLASSID,1);
1354e1e80dc8SAlp Dener   tao->ops->update = func;
1355e1e80dc8SAlp Dener   tao->user_update = ctx;
1356e1e80dc8SAlp Dener   PetscFunctionReturn(0);
1357e1e80dc8SAlp Dener }
1358e1e80dc8SAlp Dener 
1359e1e80dc8SAlp Dener /*@C
1360a7e14dcfSSatish Balay   TaoSetConvergenceTest - Sets the function that is to be used to test
1361a7e14dcfSSatish Balay   for convergence o fthe iterative minimization solution.  The new convergence
1362a7e14dcfSSatish Balay   testing routine will replace TAO's default convergence test.
1363a7e14dcfSSatish Balay 
1364441846f8SBarry Smith   Logically Collective on Tao
1365a7e14dcfSSatish Balay 
1366a7e14dcfSSatish Balay   Input Parameters:
1367441846f8SBarry Smith + tao - the Tao object
1368a7e14dcfSSatish Balay . conv - the routine to test for convergence
1369a7e14dcfSSatish Balay - ctx - [optional] context for private data for the convergence routine
137047a47007SBarry Smith         (may be NULL)
1371a7e14dcfSSatish Balay 
1372a7e14dcfSSatish Balay   Calling sequence of conv:
1373441846f8SBarry Smith $   PetscErrorCode conv(Tao tao, void *ctx)
1374a7e14dcfSSatish Balay 
1375441846f8SBarry Smith + tao - the Tao object
1376a7e14dcfSSatish Balay - ctx - [optional] convergence context
1377a7e14dcfSSatish Balay 
1378e4cb33bbSBarry Smith   Note: The new convergence testing routine should call TaoSetConvergedReason().
1379a7e14dcfSSatish Balay 
1380a7e14dcfSSatish Balay   Level: advanced
1381a7e14dcfSSatish Balay 
1382db781477SPatrick Sanan .seealso: `TaoSetConvergedReason()`, `TaoGetSolutionStatus()`, `TaoGetTolerances()`, `TaoSetMonitor`
1383a7e14dcfSSatish Balay 
1384a7e14dcfSSatish Balay @*/
1385441846f8SBarry Smith PetscErrorCode TaoSetConvergenceTest(Tao tao, PetscErrorCode (*conv)(Tao, void*), void *ctx)
1386a7e14dcfSSatish Balay {
1387a7e14dcfSSatish Balay   PetscFunctionBegin;
1388441846f8SBarry Smith   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
138994511df7SStefano Zampini   tao->ops->convergencetest = conv;
139094511df7SStefano Zampini   tao->cnvP = ctx;
1391a7e14dcfSSatish Balay   PetscFunctionReturn(0);
1392a7e14dcfSSatish Balay }
1393a7e14dcfSSatish Balay 
1394a7e14dcfSSatish Balay /*@C
1395a7e14dcfSSatish Balay    TaoSetMonitor - Sets an ADDITIONAL function that is to be used at every
1396a7e14dcfSSatish Balay    iteration of the solver to display the iteration's
1397a7e14dcfSSatish Balay    progress.
1398a7e14dcfSSatish Balay 
1399441846f8SBarry Smith    Logically Collective on Tao
1400a7e14dcfSSatish Balay 
1401a7e14dcfSSatish Balay    Input Parameters:
1402441846f8SBarry Smith +  tao - the Tao solver context
1403a7e14dcfSSatish Balay .  mymonitor - monitoring routine
1404a7e14dcfSSatish Balay -  mctx - [optional] user-defined context for private data for the
140547a47007SBarry Smith           monitor routine (may be NULL)
1406a7e14dcfSSatish Balay 
1407a7e14dcfSSatish Balay    Calling sequence of mymonitor:
1408147403d9SBarry Smith .vb
1409147403d9SBarry Smith      PetscErrorCode mymonitor(Tao tao,void *mctx)
1410147403d9SBarry Smith .ve
1411a7e14dcfSSatish Balay 
1412441846f8SBarry Smith +    tao - the Tao solver context
1413a7e14dcfSSatish Balay -    mctx - [optional] monitoring context
1414a7e14dcfSSatish Balay 
1415a7e14dcfSSatish Balay    Options Database Keys:
141698ea980cSBarry Smith +    -tao_monitor        - sets TaoMonitorDefault()
1417a7e14dcfSSatish Balay .    -tao_smonitor       - sets short monitor
1418a7e14dcfSSatish Balay .    -tao_cmonitor       - same as smonitor plus constraint norm
1419a7e14dcfSSatish Balay .    -tao_view_solution   - view solution at each iteration
1420a7e14dcfSSatish Balay .    -tao_view_gradient   - view gradient at each iteration
14214a48860cSAlp Dener .    -tao_view_ls_residual - view least-squares residual vector at each iteration
1422a7e14dcfSSatish 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.
1423a7e14dcfSSatish Balay 
1424a7e14dcfSSatish Balay    Notes:
1425a7e14dcfSSatish Balay    Several different monitoring routines may be set by calling
1426a7e14dcfSSatish Balay    TaoSetMonitor() multiple times; all will be called in the
1427a7e14dcfSSatish Balay    order in which they were set.
1428a7e14dcfSSatish Balay 
142995452b02SPatrick Sanan    Fortran Notes:
143095452b02SPatrick Sanan     Only one monitor function may be set
1431a7e14dcfSSatish Balay 
1432a7e14dcfSSatish Balay    Level: intermediate
1433a7e14dcfSSatish Balay 
1434db781477SPatrick Sanan .seealso: `TaoMonitorDefault()`, `TaoCancelMonitors()`, `TaoSetDestroyRoutine()`
1435a7e14dcfSSatish Balay @*/
1436441846f8SBarry Smith PetscErrorCode TaoSetMonitor(Tao tao, PetscErrorCode (*func)(Tao, void*), void *ctx,PetscErrorCode (*dest)(void**))
1437a7e14dcfSSatish Balay {
143808d19d1fSJason Sarich   PetscInt       i;
14398732526dSAlp Dener   PetscBool      identical;
144008d19d1fSJason Sarich 
1441a7e14dcfSSatish Balay   PetscFunctionBegin;
1442441846f8SBarry Smith   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
14433c859ba3SBarry Smith   PetscCheck(tao->numbermonitors < MAXTAOMONITORS,PetscObjectComm((PetscObject)tao),PETSC_ERR_SUP,"Cannot attach another monitor -- max=%d",MAXTAOMONITORS);
144408d19d1fSJason Sarich 
144508d19d1fSJason Sarich   for (i=0; i<tao->numbermonitors;i++) {
14469566063dSJacob Faibussowitsch     PetscCall(PetscMonitorCompare((PetscErrorCode (*)(void))func,ctx,dest,(PetscErrorCode (*)(void))tao->monitor[i],tao->monitorcontext[i],tao->monitordestroy[i],&identical));
14478732526dSAlp Dener     if (identical) PetscFunctionReturn(0);
144808d19d1fSJason Sarich   }
1449a7e14dcfSSatish Balay   tao->monitor[tao->numbermonitors] = func;
14508732526dSAlp Dener   tao->monitorcontext[tao->numbermonitors] = (void*)ctx;
1451a7e14dcfSSatish Balay   tao->monitordestroy[tao->numbermonitors] = dest;
1452a7e14dcfSSatish Balay   ++tao->numbermonitors;
1453a7e14dcfSSatish Balay   PetscFunctionReturn(0);
1454a7e14dcfSSatish Balay }
1455a7e14dcfSSatish Balay 
1456a7e14dcfSSatish Balay /*@
1457441846f8SBarry Smith    TaoCancelMonitors - Clears all the monitor functions for a Tao object.
1458a7e14dcfSSatish Balay 
1459441846f8SBarry Smith    Logically Collective on Tao
1460a7e14dcfSSatish Balay 
1461a7e14dcfSSatish Balay    Input Parameters:
1462441846f8SBarry Smith .  tao - the Tao solver context
1463a7e14dcfSSatish Balay 
1464a7e14dcfSSatish Balay    Options Database:
1465a7e14dcfSSatish Balay .  -tao_cancelmonitors - cancels all monitors that have been hardwired
1466a7e14dcfSSatish Balay     into a code by calls to TaoSetMonitor(), but does not cancel those
1467a7e14dcfSSatish Balay     set via the options database
1468a7e14dcfSSatish Balay 
1469a7e14dcfSSatish Balay    Notes:
1470441846f8SBarry Smith    There is no way to clear one specific monitor from a Tao object.
1471a7e14dcfSSatish Balay 
1472a7e14dcfSSatish Balay    Level: advanced
1473a7e14dcfSSatish Balay 
1474db781477SPatrick Sanan .seealso: `TaoMonitorDefault()`, `TaoSetMonitor()`
1475a7e14dcfSSatish Balay @*/
1476441846f8SBarry Smith PetscErrorCode TaoCancelMonitors(Tao tao)
1477a7e14dcfSSatish Balay {
1478a7e14dcfSSatish Balay   PetscInt       i;
147947a47007SBarry Smith 
1480a7e14dcfSSatish Balay   PetscFunctionBegin;
1481441846f8SBarry Smith   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
1482a7e14dcfSSatish Balay   for (i=0;i<tao->numbermonitors;i++) {
1483a7e14dcfSSatish Balay     if (tao->monitordestroy[i]) {
14849566063dSJacob Faibussowitsch       PetscCall((*tao->monitordestroy[i])(&tao->monitorcontext[i]));
1485a7e14dcfSSatish Balay     }
1486a7e14dcfSSatish Balay   }
1487a7e14dcfSSatish Balay   tao->numbermonitors = 0;
1488a7e14dcfSSatish Balay   PetscFunctionReturn(0);
1489a7e14dcfSSatish Balay }
1490a7e14dcfSSatish Balay 
14917fab98ebSJason Sarich /*@
149298ea980cSBarry Smith    TaoMonitorDefault - Default routine for monitoring progress of the
1493441846f8SBarry Smith    Tao solvers (default).  This monitor prints the function value and gradient
1494a7e14dcfSSatish Balay    norm at each iteration.  It can be turned on from the command line using the
1495a7e14dcfSSatish Balay    -tao_monitor option
1496a7e14dcfSSatish Balay 
1497441846f8SBarry Smith    Collective on Tao
1498a7e14dcfSSatish Balay 
1499a7e14dcfSSatish Balay    Input Parameters:
1500441846f8SBarry Smith +  tao - the Tao context
150147a47007SBarry Smith -  ctx - PetscViewer context or NULL
1502a7e14dcfSSatish Balay 
1503a7e14dcfSSatish Balay    Options Database Keys:
1504147403d9SBarry Smith .  -tao_monitor - turn on default monitoring
1505a7e14dcfSSatish Balay 
1506a7e14dcfSSatish Balay    Level: advanced
1507a7e14dcfSSatish Balay 
1508db781477SPatrick Sanan .seealso: `TaoDefaultSMonitor()`, `TaoSetMonitor()`
1509a7e14dcfSSatish Balay @*/
151098ea980cSBarry Smith PetscErrorCode TaoMonitorDefault(Tao tao, void *ctx)
1511a7e14dcfSSatish Balay {
151263b15415SAlp Dener   PetscInt       its, tabs;
1513a7e14dcfSSatish Balay   PetscReal      fct,gnorm;
15148163d661SBarry Smith   PetscViewer    viewer = (PetscViewer)ctx;
151547a47007SBarry Smith 
1516a7e14dcfSSatish Balay   PetscFunctionBegin;
151794511df7SStefano Zampini   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
15188163d661SBarry Smith   PetscValidHeaderSpecific(viewer,PETSC_VIEWER_CLASSID,2);
1519a7e14dcfSSatish Balay   its = tao->niter;
1520a7e14dcfSSatish Balay   fct = tao->fc;
1521a7e14dcfSSatish Balay   gnorm = tao->residual;
15229566063dSJacob Faibussowitsch   PetscCall(PetscViewerASCIIGetTab(viewer, &tabs));
15239566063dSJacob Faibussowitsch   PetscCall(PetscViewerASCIISetTab(viewer, ((PetscObject)tao)->tablevel));
1524494bef23SAlp Dener   if (its == 0 && ((PetscObject)tao)->prefix && !tao->header_printed) {
15259566063dSJacob Faibussowitsch      PetscCall(PetscViewerASCIIPrintf(viewer,"  Iteration information for %s solve.\n",((PetscObject)tao)->prefix));
1526494bef23SAlp Dener      tao->header_printed = PETSC_TRUE;
152763b15415SAlp Dener    }
152863a3b9bcSJacob Faibussowitsch   PetscCall(PetscViewerASCIIPrintf(viewer,"%3" PetscInt_FMT " TAO,",its));
15299566063dSJacob Faibussowitsch   PetscCall(PetscViewerASCIIPrintf(viewer,"  Function value: %g,",(double)fct));
153008d19d1fSJason Sarich   if (gnorm >= PETSC_INFINITY) {
15319566063dSJacob Faibussowitsch     PetscCall(PetscViewerASCIIPrintf(viewer,"  Residual: Inf \n"));
153208d19d1fSJason Sarich   } else {
15339566063dSJacob Faibussowitsch     PetscCall(PetscViewerASCIIPrintf(viewer,"  Residual: %g \n",(double)gnorm));
153408d19d1fSJason Sarich   }
15359566063dSJacob Faibussowitsch   PetscCall(PetscViewerASCIISetTab(viewer, tabs));
1536a7e14dcfSSatish Balay   PetscFunctionReturn(0);
1537a7e14dcfSSatish Balay }
1538a7e14dcfSSatish Balay 
15397fab98ebSJason Sarich /*@
15408d5ead36SAlp Dener    TaoDefaultGMonitor - Default routine for monitoring progress of the
15418d5ead36SAlp Dener    Tao solvers (default) with extra detail on the globalization method.
15428d5ead36SAlp Dener    This monitor prints the function value and gradient norm at each
15438d5ead36SAlp Dener    iteration, as well as the step size and trust radius. Note that the
15448d5ead36SAlp Dener    step size and trust radius may be the same for some algorithms.
15458d5ead36SAlp Dener    It can be turned on from the command line using the
15468d5ead36SAlp Dener    -tao_gmonitor option
15478d5ead36SAlp Dener 
15488d5ead36SAlp Dener    Collective on Tao
15498d5ead36SAlp Dener 
15508d5ead36SAlp Dener    Input Parameters:
15518d5ead36SAlp Dener +  tao - the Tao context
15528d5ead36SAlp Dener -  ctx - PetscViewer context or NULL
15538d5ead36SAlp Dener 
15548d5ead36SAlp Dener    Options Database Keys:
1555147403d9SBarry Smith .  -tao_gmonitor - turn on monitoring with globalization information
15568d5ead36SAlp Dener 
15578d5ead36SAlp Dener    Level: advanced
15588d5ead36SAlp Dener 
1559db781477SPatrick Sanan .seealso: `TaoDefaultSMonitor()`, `TaoSetMonitor()`
15608d5ead36SAlp Dener @*/
15618d5ead36SAlp Dener PetscErrorCode TaoDefaultGMonitor(Tao tao, void *ctx)
15628d5ead36SAlp Dener {
15638d5ead36SAlp Dener   PetscInt       its, tabs;
15648d5ead36SAlp Dener   PetscReal      fct,gnorm,stp,tr;
15658d5ead36SAlp Dener   PetscViewer    viewer = (PetscViewer)ctx;
15668d5ead36SAlp Dener 
15678d5ead36SAlp Dener   PetscFunctionBegin;
156894511df7SStefano Zampini   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
15698d5ead36SAlp Dener   PetscValidHeaderSpecific(viewer,PETSC_VIEWER_CLASSID,2);
15708d5ead36SAlp Dener   its = tao->niter;
15718d5ead36SAlp Dener   fct = tao->fc;
15728d5ead36SAlp Dener   gnorm = tao->residual;
15738d5ead36SAlp Dener   stp = tao->step;
15748d5ead36SAlp Dener   tr = tao->trust;
15759566063dSJacob Faibussowitsch   PetscCall(PetscViewerASCIIGetTab(viewer, &tabs));
15769566063dSJacob Faibussowitsch   PetscCall(PetscViewerASCIISetTab(viewer, ((PetscObject)tao)->tablevel));
1577494bef23SAlp Dener   if (its == 0 && ((PetscObject)tao)->prefix && !tao->header_printed) {
15789566063dSJacob Faibussowitsch      PetscCall(PetscViewerASCIIPrintf(viewer,"  Iteration information for %s solve.\n",((PetscObject)tao)->prefix));
1579494bef23SAlp Dener      tao->header_printed = PETSC_TRUE;
15808d5ead36SAlp Dener    }
158163a3b9bcSJacob Faibussowitsch   PetscCall(PetscViewerASCIIPrintf(viewer,"%3" PetscInt_FMT " TAO,",its));
15829566063dSJacob Faibussowitsch   PetscCall(PetscViewerASCIIPrintf(viewer,"  Function value: %g,",(double)fct));
15838d5ead36SAlp Dener   if (gnorm >= PETSC_INFINITY) {
15849566063dSJacob Faibussowitsch     PetscCall(PetscViewerASCIIPrintf(viewer,"  Residual: Inf,"));
15858d5ead36SAlp Dener   } else {
15869566063dSJacob Faibussowitsch     PetscCall(PetscViewerASCIIPrintf(viewer,"  Residual: %g,",(double)gnorm));
15878d5ead36SAlp Dener   }
15889566063dSJacob Faibussowitsch   PetscCall(PetscViewerASCIIPrintf(viewer,"  Step: %g,  Trust: %g\n",(double)stp,(double)tr));
15899566063dSJacob Faibussowitsch   PetscCall(PetscViewerASCIISetTab(viewer, tabs));
15908d5ead36SAlp Dener   PetscFunctionReturn(0);
15918d5ead36SAlp Dener }
15928d5ead36SAlp Dener 
15938d5ead36SAlp Dener /*@
1594a7e14dcfSSatish Balay    TaoDefaultSMonitor - Default routine for monitoring progress of the
159598ea980cSBarry Smith    solver. Same as TaoMonitorDefault() except
1596a7e14dcfSSatish Balay    it prints fewer digits of the residual as the residual gets smaller.
1597a7e14dcfSSatish Balay    This is because the later digits are meaningless and are often
1598a7e14dcfSSatish Balay    different on different machines; by using this routine different
1599a7e14dcfSSatish Balay    machines will usually generate the same output. It can be turned on
1600a7e14dcfSSatish Balay    by using the -tao_smonitor option
1601a7e14dcfSSatish Balay 
1602441846f8SBarry Smith    Collective on Tao
1603a7e14dcfSSatish Balay 
1604a7e14dcfSSatish Balay    Input Parameters:
1605441846f8SBarry Smith +  tao - the Tao context
16064d4332d5SBarry Smith -  ctx - PetscViewer context of type ASCII
1607a7e14dcfSSatish Balay 
1608a7e14dcfSSatish Balay    Options Database Keys:
1609147403d9SBarry Smith .  -tao_smonitor - turn on default short monitoring
1610a7e14dcfSSatish Balay 
1611a7e14dcfSSatish Balay    Level: advanced
1612a7e14dcfSSatish Balay 
1613db781477SPatrick Sanan .seealso: `TaoMonitorDefault()`, `TaoSetMonitor()`
1614a7e14dcfSSatish Balay @*/
1615441846f8SBarry Smith PetscErrorCode TaoDefaultSMonitor(Tao tao, void *ctx)
1616a7e14dcfSSatish Balay {
16172c9b8b83SAlp Dener   PetscInt       its, tabs;
1618a7e14dcfSSatish Balay   PetscReal      fct,gnorm;
16194d4332d5SBarry Smith   PetscViewer    viewer = (PetscViewer)ctx;
162047a47007SBarry Smith 
1621a7e14dcfSSatish Balay   PetscFunctionBegin;
162294511df7SStefano Zampini   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
16234d4332d5SBarry Smith   PetscValidHeaderSpecific(viewer,PETSC_VIEWER_CLASSID,2);
1624a7e14dcfSSatish Balay   its = tao->niter;
1625a7e14dcfSSatish Balay   fct = tao->fc;
1626a7e14dcfSSatish Balay   gnorm = tao->residual;
16279566063dSJacob Faibussowitsch   PetscCall(PetscViewerASCIIGetTab(viewer, &tabs));
16289566063dSJacob Faibussowitsch   PetscCall(PetscViewerASCIISetTab(viewer, ((PetscObject)tao)->tablevel));
162963a3b9bcSJacob Faibussowitsch   PetscCall(PetscViewerASCIIPrintf(viewer,"iter = %3" PetscInt_FMT ",",its));
16309566063dSJacob Faibussowitsch   PetscCall(PetscViewerASCIIPrintf(viewer," Function value %g,",(double)fct));
1631d393f493SBarry Smith   if (gnorm >= PETSC_INFINITY) {
16329566063dSJacob Faibussowitsch     PetscCall(PetscViewerASCIIPrintf(viewer," Residual: Inf \n"));
163308d19d1fSJason Sarich   } else if (gnorm > 1.e-6) {
16349566063dSJacob Faibussowitsch     PetscCall(PetscViewerASCIIPrintf(viewer," Residual: %g \n",(double)gnorm));
1635a7e14dcfSSatish Balay   } else if (gnorm > 1.e-11) {
16369566063dSJacob Faibussowitsch     PetscCall(PetscViewerASCIIPrintf(viewer," Residual: < 1.0e-6 \n"));
1637a7e14dcfSSatish Balay   } else {
16389566063dSJacob Faibussowitsch     PetscCall(PetscViewerASCIIPrintf(viewer," Residual: < 1.0e-11 \n"));
1639a7e14dcfSSatish Balay   }
16409566063dSJacob Faibussowitsch   PetscCall(PetscViewerASCIISetTab(viewer, tabs));
1641a7e14dcfSSatish Balay   PetscFunctionReturn(0);
1642a7e14dcfSSatish Balay }
1643a7e14dcfSSatish Balay 
16447fab98ebSJason Sarich /*@
164598ea980cSBarry Smith    TaoDefaultCMonitor - same as TaoMonitorDefault() except
1646a7e14dcfSSatish Balay    it prints the norm of the constraints function. It can be turned on
1647a7e14dcfSSatish Balay    from the command line using the -tao_cmonitor option
1648a7e14dcfSSatish Balay 
1649441846f8SBarry Smith    Collective on Tao
1650a7e14dcfSSatish Balay 
1651a7e14dcfSSatish Balay    Input Parameters:
1652441846f8SBarry Smith +  tao - the Tao context
165347a47007SBarry Smith -  ctx - PetscViewer context or NULL
1654a7e14dcfSSatish Balay 
1655a7e14dcfSSatish Balay    Options Database Keys:
1656147403d9SBarry Smith .  -tao_cmonitor - monitor the constraints
1657a7e14dcfSSatish Balay 
1658a7e14dcfSSatish Balay    Level: advanced
1659a7e14dcfSSatish Balay 
1660db781477SPatrick Sanan .seealso: `TaoMonitorDefault()`, `TaoSetMonitor()`
1661a7e14dcfSSatish Balay @*/
1662441846f8SBarry Smith PetscErrorCode TaoDefaultCMonitor(Tao tao, void *ctx)
1663a7e14dcfSSatish Balay {
16642c9b8b83SAlp Dener   PetscInt       its, tabs;
1665a7e14dcfSSatish Balay   PetscReal      fct,gnorm;
1666d393f493SBarry Smith   PetscViewer    viewer = (PetscViewer)ctx;
1667a7e14dcfSSatish Balay 
1668a7e14dcfSSatish Balay   PetscFunctionBegin;
166994511df7SStefano Zampini   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
1670d393f493SBarry Smith   PetscValidHeaderSpecific(viewer,PETSC_VIEWER_CLASSID,2);
1671a7e14dcfSSatish Balay   its = tao->niter;
1672a7e14dcfSSatish Balay   fct = tao->fc;
1673a7e14dcfSSatish Balay   gnorm = tao->residual;
16749566063dSJacob Faibussowitsch   PetscCall(PetscViewerASCIIGetTab(viewer, &tabs));
16759566063dSJacob Faibussowitsch   PetscCall(PetscViewerASCIISetTab(viewer, ((PetscObject)tao)->tablevel));
167663a3b9bcSJacob Faibussowitsch   PetscCall(PetscViewerASCIIPrintf(viewer,"iter = %" PetscInt_FMT ",",its));
16779566063dSJacob Faibussowitsch   PetscCall(PetscViewerASCIIPrintf(viewer," Function value: %g,",(double)fct));
16789566063dSJacob Faibussowitsch   PetscCall(PetscViewerASCIIPrintf(viewer,"  Residual: %g ",(double)gnorm));
16799566063dSJacob Faibussowitsch   PetscCall(PetscViewerASCIIPrintf(viewer,"  Constraint: %g \n",(double)tao->cnorm));
16809566063dSJacob Faibussowitsch   PetscCall(PetscViewerASCIISetTab(viewer, tabs));
1681a7e14dcfSSatish Balay   PetscFunctionReturn(0);
1682a7e14dcfSSatish Balay }
1683a7e14dcfSSatish Balay 
1684a7e14dcfSSatish Balay /*@C
1685a7e14dcfSSatish Balay    TaoSolutionMonitor - Views the solution at each iteration
1686a7e14dcfSSatish Balay    It can be turned on from the command line using the
1687a7e14dcfSSatish Balay    -tao_view_solution option
1688a7e14dcfSSatish Balay 
1689441846f8SBarry Smith    Collective on Tao
1690a7e14dcfSSatish Balay 
1691a7e14dcfSSatish Balay    Input Parameters:
1692441846f8SBarry Smith +  tao - the Tao context
169347a47007SBarry Smith -  ctx - PetscViewer context or NULL
1694a7e14dcfSSatish Balay 
1695a7e14dcfSSatish Balay    Options Database Keys:
1696147403d9SBarry Smith .  -tao_view_solution - view the solution
1697a7e14dcfSSatish Balay 
1698a7e14dcfSSatish Balay    Level: advanced
1699a7e14dcfSSatish Balay 
1700db781477SPatrick Sanan .seealso: `TaoDefaultSMonitor()`, `TaoSetMonitor()`
1701a7e14dcfSSatish Balay @*/
1702441846f8SBarry Smith PetscErrorCode TaoSolutionMonitor(Tao tao, void *ctx)
1703a7e14dcfSSatish Balay {
1704feb237baSPierre Jolivet   PetscViewer    viewer  = (PetscViewer)ctx;
170547a47007SBarry Smith 
1706a7e14dcfSSatish Balay   PetscFunctionBegin;
170794511df7SStefano Zampini   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
1708d393f493SBarry Smith   PetscValidHeaderSpecific(viewer,PETSC_VIEWER_CLASSID,2);
17099566063dSJacob Faibussowitsch   PetscCall(VecView(tao->solution,viewer));
1710a7e14dcfSSatish Balay   PetscFunctionReturn(0);
1711a7e14dcfSSatish Balay }
1712a7e14dcfSSatish Balay 
1713a7e14dcfSSatish Balay /*@C
1714a7e14dcfSSatish Balay    TaoGradientMonitor - Views the gradient at each iteration
1715a7e14dcfSSatish Balay    It can be turned on from the command line using the
1716a7e14dcfSSatish Balay    -tao_view_gradient option
1717a7e14dcfSSatish Balay 
1718441846f8SBarry Smith    Collective on Tao
1719a7e14dcfSSatish Balay 
1720a7e14dcfSSatish Balay    Input Parameters:
1721441846f8SBarry Smith +  tao - the Tao context
172247a47007SBarry Smith -  ctx - PetscViewer context or NULL
1723a7e14dcfSSatish Balay 
1724a7e14dcfSSatish Balay    Options Database Keys:
1725147403d9SBarry Smith .  -tao_view_gradient - view the gradient at each iteration
1726a7e14dcfSSatish Balay 
1727a7e14dcfSSatish Balay    Level: advanced
1728a7e14dcfSSatish Balay 
1729db781477SPatrick Sanan .seealso: `TaoDefaultSMonitor()`, `TaoSetMonitor()`
1730a7e14dcfSSatish Balay @*/
1731441846f8SBarry Smith PetscErrorCode TaoGradientMonitor(Tao tao, void *ctx)
1732a7e14dcfSSatish Balay {
1733d393f493SBarry Smith   PetscViewer    viewer = (PetscViewer)ctx;
173447a47007SBarry Smith 
1735a7e14dcfSSatish Balay   PetscFunctionBegin;
173694511df7SStefano Zampini   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
1737d393f493SBarry Smith   PetscValidHeaderSpecific(viewer,PETSC_VIEWER_CLASSID,2);
17389566063dSJacob Faibussowitsch   PetscCall(VecView(tao->gradient,viewer));
1739a7e14dcfSSatish Balay   PetscFunctionReturn(0);
1740a7e14dcfSSatish Balay }
1741a7e14dcfSSatish Balay 
1742a7e14dcfSSatish Balay /*@C
1743147403d9SBarry Smith    TaoStepDirectionMonitor - Views the step-direction at each iteration
1744a7e14dcfSSatish Balay 
1745441846f8SBarry Smith    Collective on Tao
1746a7e14dcfSSatish Balay 
1747a7e14dcfSSatish Balay    Input Parameters:
1748441846f8SBarry Smith +  tao - the Tao context
174947a47007SBarry Smith -  ctx - PetscViewer context or NULL
1750a7e14dcfSSatish Balay 
1751a7e14dcfSSatish Balay    Options Database Keys:
1752147403d9SBarry Smith .  -tao_view_gradient - view the gradient at each iteration
1753a7e14dcfSSatish Balay 
1754a7e14dcfSSatish Balay    Level: advanced
1755a7e14dcfSSatish Balay 
1756db781477SPatrick Sanan .seealso: `TaoDefaultSMonitor()`, `TaoSetMonitor()`
1757a7e14dcfSSatish Balay @*/
1758441846f8SBarry Smith PetscErrorCode TaoStepDirectionMonitor(Tao tao, void *ctx)
1759a7e14dcfSSatish Balay {
1760d393f493SBarry Smith   PetscViewer    viewer = (PetscViewer)ctx;
1761d393f493SBarry Smith 
1762a7e14dcfSSatish Balay   PetscFunctionBegin;
176394511df7SStefano Zampini   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
1764d393f493SBarry Smith   PetscValidHeaderSpecific(viewer,PETSC_VIEWER_CLASSID,2);
17659566063dSJacob Faibussowitsch   PetscCall(VecView(tao->stepdirection,viewer));
1766a7e14dcfSSatish Balay   PetscFunctionReturn(0);
1767a7e14dcfSSatish Balay }
1768a7e14dcfSSatish Balay 
1769a7e14dcfSSatish Balay /*@C
1770a7e14dcfSSatish Balay    TaoDrawSolutionMonitor - Plots the solution at each iteration
1771a7e14dcfSSatish Balay    It can be turned on from the command line using the
1772a7e14dcfSSatish Balay    -tao_draw_solution option
1773a7e14dcfSSatish Balay 
1774441846f8SBarry Smith    Collective on Tao
1775a7e14dcfSSatish Balay 
1776a7e14dcfSSatish Balay    Input Parameters:
1777441846f8SBarry Smith +  tao - the Tao context
1778e882e171SHong Zhang -  ctx - TaoMonitorDraw context
1779a7e14dcfSSatish Balay 
1780a7e14dcfSSatish Balay    Options Database Keys:
1781147403d9SBarry Smith .  -tao_draw_solution - draw the solution at each iteration
1782a7e14dcfSSatish Balay 
1783a7e14dcfSSatish Balay    Level: advanced
1784a7e14dcfSSatish Balay 
1785db781477SPatrick Sanan .seealso: `TaoSolutionMonitor()`, `TaoSetMonitor()`, `TaoDrawGradientMonitor`
1786a7e14dcfSSatish Balay @*/
1787441846f8SBarry Smith PetscErrorCode TaoDrawSolutionMonitor(Tao tao, void *ctx)
1788a7e14dcfSSatish Balay {
1789e882e171SHong Zhang   TaoMonitorDrawCtx ictx = (TaoMonitorDrawCtx)ctx;
179047a47007SBarry Smith 
1791a7e14dcfSSatish Balay   PetscFunctionBegin;
179294511df7SStefano Zampini   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
1793e882e171SHong Zhang   if (!(((ictx->howoften > 0) && (!(tao->niter % ictx->howoften))) || ((ictx->howoften == -1) && tao->reason))) PetscFunctionReturn(0);
17949566063dSJacob Faibussowitsch   PetscCall(VecView(tao->solution,ictx->viewer));
1795a7e14dcfSSatish Balay   PetscFunctionReturn(0);
1796a7e14dcfSSatish Balay }
1797a7e14dcfSSatish Balay 
1798a7e14dcfSSatish Balay /*@C
1799a7e14dcfSSatish Balay    TaoDrawGradientMonitor - Plots the gradient at each iteration
1800a7e14dcfSSatish Balay    It can be turned on from the command line using the
1801a7e14dcfSSatish Balay    -tao_draw_gradient option
1802a7e14dcfSSatish Balay 
1803441846f8SBarry Smith    Collective on Tao
1804a7e14dcfSSatish Balay 
1805a7e14dcfSSatish Balay    Input Parameters:
1806441846f8SBarry Smith +  tao - the Tao context
1807f55353a2SBarry Smith -  ctx - PetscViewer context
1808a7e14dcfSSatish Balay 
1809a7e14dcfSSatish Balay    Options Database Keys:
1810147403d9SBarry Smith .  -tao_draw_gradient - draw the gradient at each iteration
1811a7e14dcfSSatish Balay 
1812a7e14dcfSSatish Balay    Level: advanced
1813a7e14dcfSSatish Balay 
1814db781477SPatrick Sanan .seealso: `TaoGradientMonitor()`, `TaoSetMonitor()`, `TaoDrawSolutionMonitor`
1815a7e14dcfSSatish Balay @*/
1816441846f8SBarry Smith PetscErrorCode TaoDrawGradientMonitor(Tao tao, void *ctx)
1817a7e14dcfSSatish Balay {
1818e882e171SHong Zhang   TaoMonitorDrawCtx ictx = (TaoMonitorDrawCtx)ctx;
181947a47007SBarry Smith 
1820a7e14dcfSSatish Balay   PetscFunctionBegin;
182194511df7SStefano Zampini   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
1822e882e171SHong Zhang   if (!(((ictx->howoften > 0) && (!(tao->niter % ictx->howoften))) || ((ictx->howoften == -1) && tao->reason))) PetscFunctionReturn(0);
18239566063dSJacob Faibussowitsch   PetscCall(VecView(tao->gradient,ictx->viewer));
1824a7e14dcfSSatish Balay   PetscFunctionReturn(0);
1825a7e14dcfSSatish Balay }
1826a7e14dcfSSatish Balay 
1827a7e14dcfSSatish Balay /*@C
1828a7e14dcfSSatish Balay    TaoDrawStepMonitor - Plots the step direction at each iteration
1829a7e14dcfSSatish Balay 
1830441846f8SBarry Smith    Collective on Tao
1831a7e14dcfSSatish Balay 
1832a7e14dcfSSatish Balay    Input Parameters:
1833441846f8SBarry Smith +  tao - the Tao context
1834f55353a2SBarry Smith -  ctx - PetscViewer context
1835a7e14dcfSSatish Balay 
1836a7e14dcfSSatish Balay    Options Database Keys:
1837147403d9SBarry Smith .  -tao_draw_step - draw the step direction at each iteration
1838a7e14dcfSSatish Balay 
1839a7e14dcfSSatish Balay    Level: advanced
1840a7e14dcfSSatish Balay 
1841db781477SPatrick Sanan .seealso: `TaoSetMonitor()`, `TaoDrawSolutionMonitor`
1842a7e14dcfSSatish Balay @*/
1843441846f8SBarry Smith PetscErrorCode TaoDrawStepMonitor(Tao tao, void *ctx)
1844a7e14dcfSSatish Balay {
184594511df7SStefano Zampini   PetscViewer    viewer = (PetscViewer)ctx;
184647a47007SBarry Smith 
1847a7e14dcfSSatish Balay   PetscFunctionBegin;
184894511df7SStefano Zampini   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
184994511df7SStefano Zampini   PetscValidHeaderSpecific(viewer,PETSC_VIEWER_CLASSID,2);
18509566063dSJacob Faibussowitsch   PetscCall(VecView(tao->stepdirection,viewer));
1851a7e14dcfSSatish Balay   PetscFunctionReturn(0);
1852a7e14dcfSSatish Balay }
1853a7e14dcfSSatish Balay 
1854a7e14dcfSSatish Balay /*@C
1855737f463aSAlp Dener    TaoResidualMonitor - Views the least-squares residual at each iteration
1856a7e14dcfSSatish Balay 
1857441846f8SBarry Smith    Collective on Tao
1858a7e14dcfSSatish Balay 
1859a7e14dcfSSatish Balay    Input Parameters:
1860441846f8SBarry Smith +  tao - the Tao context
186147a47007SBarry Smith -  ctx - PetscViewer context or NULL
1862a7e14dcfSSatish Balay 
1863a7e14dcfSSatish Balay    Options Database Keys:
1864147403d9SBarry Smith .  -tao_view_ls_residual - view the least-squares residual at each iteration
1865a7e14dcfSSatish Balay 
1866a7e14dcfSSatish Balay    Level: advanced
1867a7e14dcfSSatish Balay 
1868db781477SPatrick Sanan .seealso: `TaoDefaultSMonitor()`, `TaoSetMonitor()`
1869a7e14dcfSSatish Balay @*/
1870737f463aSAlp Dener PetscErrorCode TaoResidualMonitor(Tao tao, void *ctx)
1871a7e14dcfSSatish Balay {
1872d393f493SBarry Smith   PetscViewer    viewer  = (PetscViewer)ctx;
187347a47007SBarry Smith 
1874a7e14dcfSSatish Balay   PetscFunctionBegin;
187594511df7SStefano Zampini   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
1876d393f493SBarry Smith   PetscValidHeaderSpecific(viewer,PETSC_VIEWER_CLASSID,2);
18779566063dSJacob Faibussowitsch   PetscCall(VecView(tao->ls_res,viewer));
1878a7e14dcfSSatish Balay   PetscFunctionReturn(0);
1879a7e14dcfSSatish Balay }
1880a7e14dcfSSatish Balay 
18817fab98ebSJason Sarich /*@
1882a7e14dcfSSatish Balay    TaoDefaultConvergenceTest - Determines whether the solver should continue iterating
1883a7e14dcfSSatish Balay    or terminate.
1884a7e14dcfSSatish Balay 
1885441846f8SBarry Smith    Collective on Tao
1886a7e14dcfSSatish Balay 
1887a7e14dcfSSatish Balay    Input Parameters:
1888441846f8SBarry Smith +  tao - the Tao context
1889a7e14dcfSSatish Balay -  dummy - unused dummy context
1890a7e14dcfSSatish Balay 
1891a7e14dcfSSatish Balay    Output Parameter:
1892a7e14dcfSSatish Balay .  reason - for terminating
1893a7e14dcfSSatish Balay 
1894a7e14dcfSSatish Balay    Notes:
1895a7e14dcfSSatish Balay    This routine checks the residual in the optimality conditions, the
1896a7e14dcfSSatish Balay    relative residual in the optimity conditions, the number of function
1897a7e14dcfSSatish Balay    evaluations, and the function value to test convergence.  Some
1898a7e14dcfSSatish Balay    solvers may use different convergence routines.
1899a7e14dcfSSatish Balay 
1900a7e14dcfSSatish Balay    Level: developer
1901a7e14dcfSSatish Balay 
1902*c2e3fba1SPatrick Sanan .seealso: `TaoSetTolerances()`, `TaoGetConvergedReason()`, `TaoSetConvergedReason()`
1903a7e14dcfSSatish Balay @*/
1904a7e14dcfSSatish Balay 
1905441846f8SBarry Smith PetscErrorCode TaoDefaultConvergenceTest(Tao tao,void *dummy)
1906a7e14dcfSSatish Balay {
1907a7e14dcfSSatish Balay   PetscInt           niter=tao->niter, nfuncs=PetscMax(tao->nfuncs,tao->nfuncgrads);
1908a7e14dcfSSatish Balay   PetscInt           max_funcs=tao->max_funcs;
1909a7e14dcfSSatish Balay   PetscReal          gnorm=tao->residual, gnorm0=tao->gnorm0;
1910a7e14dcfSSatish Balay   PetscReal          f=tao->fc, steptol=tao->steptol,trradius=tao->step;
1911a7e14dcfSSatish Balay   PetscReal          gatol=tao->gatol,grtol=tao->grtol,gttol=tao->gttol;
1912e52336cbSBarry Smith   PetscReal          catol=tao->catol,crtol=tao->crtol;
1913e52336cbSBarry Smith   PetscReal          fmin=tao->fmin, cnorm=tao->cnorm;
1914e4cb33bbSBarry Smith   TaoConvergedReason reason=tao->reason;
1915a7e14dcfSSatish Balay 
1916a7e14dcfSSatish Balay   PetscFunctionBegin;
1917441846f8SBarry Smith   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
191894511df7SStefano Zampini   if (reason != TAO_CONTINUE_ITERATING) PetscFunctionReturn(0);
1919a7e14dcfSSatish Balay 
1920a7e14dcfSSatish Balay   if (PetscIsInfOrNanReal(f)) {
19219566063dSJacob Faibussowitsch     PetscCall(PetscInfo(tao,"Failed to converged, function value is Inf or NaN\n"));
1922a7e14dcfSSatish Balay     reason = TAO_DIVERGED_NAN;
1923a7e14dcfSSatish Balay   } else if (f <= fmin && cnorm <=catol) {
19249566063dSJacob Faibussowitsch     PetscCall(PetscInfo(tao,"Converged due to function value %g < minimum function value %g\n", (double)f,(double)fmin));
1925a7e14dcfSSatish Balay     reason = TAO_CONVERGED_MINF;
1926a7e14dcfSSatish Balay   } else if (gnorm<= gatol && cnorm <=catol) {
19279566063dSJacob Faibussowitsch     PetscCall(PetscInfo(tao,"Converged due to residual norm ||g(X)||=%g < %g\n",(double)gnorm,(double)gatol));
1928a7e14dcfSSatish Balay     reason = TAO_CONVERGED_GATOL;
1929a7e14dcfSSatish Balay   } else if (f!=0 && PetscAbsReal(gnorm/f) <= grtol && cnorm <= crtol) {
19309566063dSJacob Faibussowitsch     PetscCall(PetscInfo(tao,"Converged due to residual ||g(X)||/|f(X)| =%g < %g\n",(double)(gnorm/f),(double)grtol));
1931a7e14dcfSSatish Balay     reason = TAO_CONVERGED_GRTOL;
1932e1d16bb9SBarry Smith   } else if (gnorm0 != 0 && ((gttol == 0 && gnorm == 0) || gnorm/gnorm0 < gttol) && cnorm <= crtol) {
19339566063dSJacob Faibussowitsch     PetscCall(PetscInfo(tao,"Converged due to relative residual norm ||g(X)||/||g(X0)|| = %g < %g\n",(double)(gnorm/gnorm0),(double)gttol));
1934a7e14dcfSSatish Balay     reason = TAO_CONVERGED_GTTOL;
193594511df7SStefano Zampini   } else if (max_funcs >=0 && nfuncs > max_funcs) {
19369566063dSJacob Faibussowitsch     PetscCall(PetscInfo(tao,"Exceeded maximum number of function evaluations: %" PetscInt_FMT " > %" PetscInt_FMT "\n", nfuncs,max_funcs));
1937a7e14dcfSSatish Balay     reason = TAO_DIVERGED_MAXFCN;
1938a7e14dcfSSatish Balay   } else if (tao->lsflag != 0) {
19399566063dSJacob Faibussowitsch     PetscCall(PetscInfo(tao,"Tao Line Search failure.\n"));
1940a7e14dcfSSatish Balay     reason = TAO_DIVERGED_LS_FAILURE;
1941a7e14dcfSSatish Balay   } else if (trradius < steptol && niter > 0) {
19429566063dSJacob Faibussowitsch     PetscCall(PetscInfo(tao,"Trust region/step size too small: %g < %g\n", (double)trradius,(double)steptol));
1943a7e14dcfSSatish Balay     reason = TAO_CONVERGED_STEPTOL;
1944e031d6f5SAlp Dener   } else if (niter >= tao->max_it) {
194563a3b9bcSJacob Faibussowitsch     PetscCall(PetscInfo(tao,"Exceeded maximum number of iterations: %" PetscInt_FMT " > %" PetscInt_FMT "\n",niter,tao->max_it));
1946a7e14dcfSSatish Balay     reason = TAO_DIVERGED_MAXITS;
1947a7e14dcfSSatish Balay   } else {
1948a7e14dcfSSatish Balay     reason = TAO_CONTINUE_ITERATING;
1949a7e14dcfSSatish Balay   }
1950a7e14dcfSSatish Balay   tao->reason = reason;
1951a7e14dcfSSatish Balay   PetscFunctionReturn(0);
1952a7e14dcfSSatish Balay }
1953a7e14dcfSSatish Balay 
1954a7e14dcfSSatish Balay /*@C
1955a7e14dcfSSatish Balay    TaoSetOptionsPrefix - Sets the prefix used for searching for all
1956a7e14dcfSSatish Balay    TAO options in the database.
1957a7e14dcfSSatish Balay 
1958441846f8SBarry Smith    Logically Collective on Tao
1959a7e14dcfSSatish Balay 
1960a7e14dcfSSatish Balay    Input Parameters:
1961441846f8SBarry Smith +  tao - the Tao context
1962a7e14dcfSSatish Balay -  prefix - the prefix string to prepend to all TAO option requests
1963a7e14dcfSSatish Balay 
1964a7e14dcfSSatish Balay    Notes:
1965a7e14dcfSSatish Balay    A hyphen (-) must NOT be given at the beginning of the prefix name.
1966a7e14dcfSSatish Balay    The first character of all runtime options is AUTOMATICALLY the hyphen.
1967a7e14dcfSSatish Balay 
1968a7e14dcfSSatish Balay    For example, to distinguish between the runtime options for two
1969a7e14dcfSSatish Balay    different TAO solvers, one could call
1970a7e14dcfSSatish Balay .vb
1971a7e14dcfSSatish Balay       TaoSetOptionsPrefix(tao1,"sys1_")
1972a7e14dcfSSatish Balay       TaoSetOptionsPrefix(tao2,"sys2_")
1973a7e14dcfSSatish Balay .ve
1974a7e14dcfSSatish Balay 
1975a7e14dcfSSatish Balay    This would enable use of different options for each system, such as
1976a7e14dcfSSatish Balay .vb
19779fa2b5dcSStefano Zampini       -sys1_tao_method blmvm -sys1_tao_grtol 1.e-3
19789fa2b5dcSStefano Zampini       -sys2_tao_method lmvm  -sys2_tao_grtol 1.e-4
1979a7e14dcfSSatish Balay .ve
1980a7e14dcfSSatish Balay 
1981a7e14dcfSSatish Balay    Level: advanced
1982a7e14dcfSSatish Balay 
1983db781477SPatrick Sanan .seealso: `TaoAppendOptionsPrefix()`, `TaoGetOptionsPrefix()`
1984a7e14dcfSSatish Balay @*/
1985a7e14dcfSSatish Balay 
1986441846f8SBarry Smith PetscErrorCode TaoSetOptionsPrefix(Tao tao, const char p[])
1987a7e14dcfSSatish Balay {
1988a7e14dcfSSatish Balay   PetscFunctionBegin;
198994511df7SStefano Zampini   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
19909566063dSJacob Faibussowitsch   PetscCall(PetscObjectSetOptionsPrefix((PetscObject)tao,p));
1991a7e14dcfSSatish Balay   if (tao->linesearch) {
19929566063dSJacob Faibussowitsch     PetscCall(TaoLineSearchSetOptionsPrefix(tao->linesearch,p));
1993a7e14dcfSSatish Balay   }
1994a7e14dcfSSatish Balay   if (tao->ksp) {
19959566063dSJacob Faibussowitsch     PetscCall(KSPSetOptionsPrefix(tao->ksp,p));
1996a7e14dcfSSatish Balay   }
1997a7e14dcfSSatish Balay   PetscFunctionReturn(0);
1998a7e14dcfSSatish Balay }
1999a7e14dcfSSatish Balay 
2000a7e14dcfSSatish Balay /*@C
2001a7e14dcfSSatish Balay    TaoAppendOptionsPrefix - Appends to the prefix used for searching for all
2002a7e14dcfSSatish Balay    TAO options in the database.
2003a7e14dcfSSatish Balay 
2004441846f8SBarry Smith    Logically Collective on Tao
2005a7e14dcfSSatish Balay 
2006a7e14dcfSSatish Balay    Input Parameters:
2007441846f8SBarry Smith +  tao - the Tao solver context
2008a7e14dcfSSatish Balay -  prefix - the prefix string to prepend to all TAO option requests
2009a7e14dcfSSatish Balay 
2010a7e14dcfSSatish Balay    Notes:
2011a7e14dcfSSatish Balay    A hyphen (-) must NOT be given at the beginning of the prefix name.
2012a7e14dcfSSatish Balay    The first character of all runtime options is AUTOMATICALLY the hyphen.
2013a7e14dcfSSatish Balay 
2014a7e14dcfSSatish Balay    Level: advanced
2015a7e14dcfSSatish Balay 
2016db781477SPatrick Sanan .seealso: `TaoSetOptionsPrefix()`, `TaoGetOptionsPrefix()`
2017a7e14dcfSSatish Balay @*/
2018441846f8SBarry Smith PetscErrorCode TaoAppendOptionsPrefix(Tao tao, const char p[])
2019a7e14dcfSSatish Balay {
2020a7e14dcfSSatish Balay   PetscFunctionBegin;
202194511df7SStefano Zampini   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
20229566063dSJacob Faibussowitsch   PetscCall(PetscObjectAppendOptionsPrefix((PetscObject)tao,p));
2023a7e14dcfSSatish Balay   if (tao->linesearch) {
20249566063dSJacob Faibussowitsch     PetscCall(PetscObjectAppendOptionsPrefix((PetscObject)tao->linesearch,p));
2025a7e14dcfSSatish Balay   }
2026a7e14dcfSSatish Balay   if (tao->ksp) {
20279566063dSJacob Faibussowitsch     PetscCall(KSPAppendOptionsPrefix(tao->ksp,p));
2028a7e14dcfSSatish Balay   }
2029a7e14dcfSSatish Balay   PetscFunctionReturn(0);
2030a7e14dcfSSatish Balay }
2031a7e14dcfSSatish Balay 
2032a7e14dcfSSatish Balay /*@C
2033a7e14dcfSSatish Balay   TaoGetOptionsPrefix - Gets the prefix used for searching for all
2034a7e14dcfSSatish Balay   TAO options in the database
2035a7e14dcfSSatish Balay 
2036a7e14dcfSSatish Balay   Not Collective
2037a7e14dcfSSatish Balay 
2038a7e14dcfSSatish Balay   Input Parameters:
2039441846f8SBarry Smith . tao - the Tao context
2040a7e14dcfSSatish Balay 
2041a7e14dcfSSatish Balay   Output Parameters:
2042a7e14dcfSSatish Balay . prefix - pointer to the prefix string used is returned
2043a7e14dcfSSatish Balay 
204495452b02SPatrick Sanan   Notes:
204595452b02SPatrick Sanan     On the fortran side, the user should pass in a string 'prefix' of
2046a7e14dcfSSatish Balay   sufficient length to hold the prefix.
2047a7e14dcfSSatish Balay 
2048a7e14dcfSSatish Balay   Level: advanced
2049a7e14dcfSSatish Balay 
2050db781477SPatrick Sanan .seealso: `TaoSetOptionsPrefix()`, `TaoAppendOptionsPrefix()`
2051a7e14dcfSSatish Balay @*/
2052441846f8SBarry Smith PetscErrorCode TaoGetOptionsPrefix(Tao tao, const char *p[])
2053a7e14dcfSSatish Balay {
205494511df7SStefano Zampini   PetscFunctionBegin;
205594511df7SStefano Zampini   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
20569566063dSJacob Faibussowitsch   PetscCall(PetscObjectGetOptionsPrefix((PetscObject)tao,p));
205794511df7SStefano Zampini   PetscFunctionReturn(0);
2058a7e14dcfSSatish Balay }
2059a7e14dcfSSatish Balay 
2060a7e14dcfSSatish Balay /*@C
2061a7e14dcfSSatish Balay    TaoSetType - Sets the method for the unconstrained minimization solver.
2062a7e14dcfSSatish Balay 
2063441846f8SBarry Smith    Collective on Tao
2064a7e14dcfSSatish Balay 
2065a7e14dcfSSatish Balay    Input Parameters:
2066441846f8SBarry Smith +  solver - the Tao solver context
2067a7e14dcfSSatish Balay -  type - a known method
2068a7e14dcfSSatish Balay 
2069a7e14dcfSSatish Balay    Options Database Key:
207058417fe7SBarry Smith .  -tao_type <type> - Sets the method; use -help for a list
207158417fe7SBarry Smith    of available methods (for instance, "-tao_type lmvm" or "-tao_type tron")
2072a7e14dcfSSatish Balay 
2073a7e14dcfSSatish Balay    Available methods include:
207458417fe7SBarry Smith +    nls - Newton's method with line search for unconstrained minimization
207558417fe7SBarry Smith .    ntr - Newton's method with trust region for unconstrained minimization
207658417fe7SBarry Smith .    ntl - Newton's method with trust region, line search for unconstrained minimization
207758417fe7SBarry Smith .    lmvm - Limited memory variable metric method for unconstrained minimization
207858417fe7SBarry Smith .    cg - Nonlinear conjugate gradient method for unconstrained minimization
207958417fe7SBarry Smith .    nm - Nelder-Mead algorithm for derivate-free unconstrained minimization
208058417fe7SBarry Smith .    tron - Newton Trust Region method for bound constrained minimization
208158417fe7SBarry Smith .    gpcg - Newton Trust Region method for quadratic bound constrained minimization
208258417fe7SBarry Smith .    blmvm - Limited memory variable metric method for bound constrained minimization
20831eb8069cSJason Sarich -    pounders - Model-based algorithm pounder extended for nonlinear least squares
2084a7e14dcfSSatish Balay 
2085a7e14dcfSSatish Balay   Level: intermediate
2086a7e14dcfSSatish Balay 
2087db781477SPatrick Sanan .seealso: `TaoCreate()`, `TaoGetType()`, `TaoType`
2088a7e14dcfSSatish Balay 
2089a7e14dcfSSatish Balay @*/
2090b625d6c7SJed Brown PetscErrorCode TaoSetType(Tao tao, TaoType type)
2091a7e14dcfSSatish Balay {
2092441846f8SBarry Smith   PetscErrorCode (*create_xxx)(Tao);
2093a7e14dcfSSatish Balay   PetscBool      issame;
2094a7e14dcfSSatish Balay 
2095a7e14dcfSSatish Balay   PetscFunctionBegin;
2096441846f8SBarry Smith   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
2097a7e14dcfSSatish Balay 
20989566063dSJacob Faibussowitsch   PetscCall(PetscObjectTypeCompare((PetscObject)tao,type,&issame));
2099a7e14dcfSSatish Balay   if (issame) PetscFunctionReturn(0);
2100a7e14dcfSSatish Balay 
21019566063dSJacob Faibussowitsch   PetscCall(PetscFunctionListFind(TaoList, type, (void(**)(void))&create_xxx));
21023c859ba3SBarry Smith   PetscCheck(create_xxx,PetscObjectComm((PetscObject)tao),PETSC_ERR_ARG_UNKNOWN_TYPE,"Unable to find requested Tao type %s",type);
2103a7e14dcfSSatish Balay 
2104a7e14dcfSSatish Balay   /* Destroy the existing solver information */
2105a7e14dcfSSatish Balay   if (tao->ops->destroy) {
21069566063dSJacob Faibussowitsch     PetscCall((*tao->ops->destroy)(tao));
2107a7e14dcfSSatish Balay   }
21089566063dSJacob Faibussowitsch   PetscCall(KSPDestroy(&tao->ksp));
21099566063dSJacob Faibussowitsch   PetscCall(TaoLineSearchDestroy(&tao->linesearch));
21109566063dSJacob Faibussowitsch   PetscCall(VecDestroy(&tao->gradient));
21119566063dSJacob Faibussowitsch   PetscCall(VecDestroy(&tao->stepdirection));
2112a7e14dcfSSatish Balay 
211383c8fe1dSLisandro Dalcin   tao->ops->setup = NULL;
211483c8fe1dSLisandro Dalcin   tao->ops->solve = NULL;
211583c8fe1dSLisandro Dalcin   tao->ops->view  = NULL;
211683c8fe1dSLisandro Dalcin   tao->ops->setfromoptions = NULL;
211783c8fe1dSLisandro Dalcin   tao->ops->destroy = NULL;
2118a7e14dcfSSatish Balay 
2119a7e14dcfSSatish Balay   tao->setupcalled = PETSC_FALSE;
2120a7e14dcfSSatish Balay 
21219566063dSJacob Faibussowitsch   PetscCall((*create_xxx)(tao));
21229566063dSJacob Faibussowitsch   PetscCall(PetscObjectChangeTypeName((PetscObject)tao,type));
2123a7e14dcfSSatish Balay   PetscFunctionReturn(0);
2124a7e14dcfSSatish Balay }
212547a47007SBarry Smith 
2126a7e14dcfSSatish Balay /*MC
2127441846f8SBarry Smith    TaoRegister - Adds a method to the TAO package for unconstrained minimization.
2128a7e14dcfSSatish Balay 
2129a7e14dcfSSatish Balay    Synopsis:
21305bd1e576SStefano Zampini    TaoRegister(char *name_solver,char *path,char *name_Create,PetscErrorCode (*routine_Create)(Tao))
2131a7e14dcfSSatish Balay 
2132a7e14dcfSSatish Balay    Not collective
2133a7e14dcfSSatish Balay 
2134a7e14dcfSSatish Balay    Input Parameters:
2135a7e14dcfSSatish Balay +  sname - name of a new user-defined solver
2136a7e14dcfSSatish Balay -  func - routine to Create method context
2137a7e14dcfSSatish Balay 
2138a7e14dcfSSatish Balay    Notes:
2139441846f8SBarry Smith    TaoRegister() may be called multiple times to add several user-defined solvers.
2140a7e14dcfSSatish Balay 
2141a7e14dcfSSatish Balay    Sample usage:
2142a7e14dcfSSatish Balay .vb
2143441846f8SBarry Smith    TaoRegister("my_solver",MySolverCreate);
2144a7e14dcfSSatish Balay .ve
2145a7e14dcfSSatish Balay 
2146a7e14dcfSSatish Balay    Then, your solver can be chosen with the procedural interface via
2147a7e14dcfSSatish Balay $     TaoSetType(tao,"my_solver")
2148a7e14dcfSSatish Balay    or at runtime via the option
214958417fe7SBarry Smith $     -tao_type my_solver
2150a7e14dcfSSatish Balay 
2151a7e14dcfSSatish Balay    Level: advanced
2152a7e14dcfSSatish Balay 
2153db781477SPatrick Sanan .seealso: `TaoRegisterAll()`, `TaoRegisterDestroy()`
2154a7e14dcfSSatish Balay M*/
2155441846f8SBarry Smith PetscErrorCode TaoRegister(const char sname[], PetscErrorCode (*func)(Tao))
2156a7e14dcfSSatish Balay {
2157a7e14dcfSSatish Balay   PetscFunctionBegin;
21589566063dSJacob Faibussowitsch   PetscCall(TaoInitializePackage());
21599566063dSJacob Faibussowitsch   PetscCall(PetscFunctionListAdd(&TaoList,sname,(void (*)(void))func));
2160a7e14dcfSSatish Balay   PetscFunctionReturn(0);
2161a7e14dcfSSatish Balay }
2162a7e14dcfSSatish Balay 
2163a7e14dcfSSatish Balay /*@C
2164441846f8SBarry Smith    TaoRegisterDestroy - Frees the list of minimization solvers that were
2165441846f8SBarry Smith    registered by TaoRegisterDynamic().
2166a7e14dcfSSatish Balay 
2167a7e14dcfSSatish Balay    Not Collective
2168a7e14dcfSSatish Balay 
2169a7e14dcfSSatish Balay    Level: advanced
2170a7e14dcfSSatish Balay 
2171db781477SPatrick Sanan .seealso: `TaoRegisterAll()`, `TaoRegister()`
2172a7e14dcfSSatish Balay @*/
2173441846f8SBarry Smith PetscErrorCode TaoRegisterDestroy(void)
2174a7e14dcfSSatish Balay {
2175a7e14dcfSSatish Balay   PetscFunctionBegin;
21769566063dSJacob Faibussowitsch   PetscCall(PetscFunctionListDestroy(&TaoList));
2177441846f8SBarry Smith   TaoRegisterAllCalled = PETSC_FALSE;
2178a7e14dcfSSatish Balay   PetscFunctionReturn(0);
2179a7e14dcfSSatish Balay }
218047a47007SBarry Smith 
21818931d482SJason Sarich /*@
21828931d482SJason Sarich    TaoGetIterationNumber - Gets the number of Tao iterations completed
21838931d482SJason Sarich    at this time.
21848931d482SJason Sarich 
21858931d482SJason Sarich    Not Collective
21868931d482SJason Sarich 
21878931d482SJason Sarich    Input Parameter:
21888931d482SJason Sarich .  tao - Tao context
21898931d482SJason Sarich 
21908931d482SJason Sarich    Output Parameter:
21918931d482SJason Sarich .  iter - iteration number
21928931d482SJason Sarich 
21938931d482SJason Sarich    Notes:
21948931d482SJason Sarich    For example, during the computation of iteration 2 this would return 1.
21958931d482SJason Sarich 
21968931d482SJason Sarich    Level: intermediate
21978931d482SJason Sarich 
2198db781477SPatrick Sanan .seealso: `TaoGetLinearSolveIterations()`, `TaoGetResidualNorm()`, `TaoGetObjective()`
21998931d482SJason Sarich @*/
22008931d482SJason Sarich PetscErrorCode  TaoGetIterationNumber(Tao tao,PetscInt *iter)
22018931d482SJason Sarich {
22028931d482SJason Sarich   PetscFunctionBegin;
22038931d482SJason Sarich   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
22048931d482SJason Sarich   PetscValidIntPointer(iter,2);
22058931d482SJason Sarich   *iter = tao->niter;
22068931d482SJason Sarich   PetscFunctionReturn(0);
22078931d482SJason Sarich }
22088931d482SJason Sarich 
22098931d482SJason Sarich /*@
221079f5d8caSBarry Smith    TaoGetResidualNorm - Gets the current value of the norm of the residual
221179f5d8caSBarry Smith    at this time.
221279f5d8caSBarry Smith 
221379f5d8caSBarry Smith    Not Collective
221479f5d8caSBarry Smith 
221579f5d8caSBarry Smith    Input Parameter:
221679f5d8caSBarry Smith .  tao - Tao context
221779f5d8caSBarry Smith 
221879f5d8caSBarry Smith    Output Parameter:
221979f5d8caSBarry Smith .  value - the current value
222079f5d8caSBarry Smith 
222179f5d8caSBarry Smith    Level: intermediate
222279f5d8caSBarry Smith 
222379f5d8caSBarry Smith    Developer Note: This is the 2-norm of the residual, we cannot use TaoGetGradientNorm() because that has
222479f5d8caSBarry Smith                    a different meaning. For some reason Tao sometimes calls the gradient the residual.
222579f5d8caSBarry Smith 
2226db781477SPatrick Sanan .seealso: `TaoGetLinearSolveIterations()`, `TaoGetIterationNumber()`, `TaoGetObjective()`
222779f5d8caSBarry Smith @*/
222879f5d8caSBarry Smith PetscErrorCode TaoGetResidualNorm(Tao tao,PetscReal *value)
222979f5d8caSBarry Smith {
223079f5d8caSBarry Smith   PetscFunctionBegin;
223179f5d8caSBarry Smith   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
223279f5d8caSBarry Smith   PetscValidRealPointer(value,2);
223379f5d8caSBarry Smith   *value = tao->residual;
223479f5d8caSBarry Smith   PetscFunctionReturn(0);
223579f5d8caSBarry Smith }
223679f5d8caSBarry Smith 
223779f5d8caSBarry Smith /*@
22388931d482SJason Sarich    TaoSetIterationNumber - Sets the current iteration number.
22398931d482SJason Sarich 
224094511df7SStefano Zampini    Logically Collective on Tao
22418931d482SJason Sarich 
2242d8d19677SJose E. Roman    Input Parameters:
2243a2b725a8SWilliam Gropp +  tao - Tao context
2244a2b725a8SWilliam Gropp -  iter - iteration number
22458931d482SJason Sarich 
22468931d482SJason Sarich    Level: developer
22478931d482SJason Sarich 
2248db781477SPatrick Sanan .seealso: `TaoGetLinearSolveIterations()`
22498931d482SJason Sarich @*/
22508931d482SJason Sarich PetscErrorCode  TaoSetIterationNumber(Tao tao,PetscInt iter)
22518931d482SJason Sarich {
22528931d482SJason Sarich   PetscFunctionBegin;
22538931d482SJason Sarich   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
225494511df7SStefano Zampini   PetscValidLogicalCollectiveInt(tao,iter,2);
22559566063dSJacob Faibussowitsch   PetscCall(PetscObjectSAWsTakeAccess((PetscObject)tao));
22568931d482SJason Sarich   tao->niter = iter;
22579566063dSJacob Faibussowitsch   PetscCall(PetscObjectSAWsGrantAccess((PetscObject)tao));
22588931d482SJason Sarich   PetscFunctionReturn(0);
22598931d482SJason Sarich }
22608931d482SJason Sarich 
22618931d482SJason Sarich /*@
22628931d482SJason Sarich    TaoGetTotalIterationNumber - Gets the total number of Tao iterations
22638931d482SJason Sarich    completed. This number keeps accumulating if multiple solves
22648931d482SJason Sarich    are called with the Tao object.
22658931d482SJason Sarich 
22668931d482SJason Sarich    Not Collective
22678931d482SJason Sarich 
22688931d482SJason Sarich    Input Parameter:
22698931d482SJason Sarich .  tao - Tao context
22708931d482SJason Sarich 
22718931d482SJason Sarich    Output Parameter:
22728931d482SJason Sarich .  iter - iteration number
22738931d482SJason Sarich 
22748931d482SJason Sarich    Notes:
22758931d482SJason Sarich    The total iteration count is updated after each solve, if there is a current
22768931d482SJason Sarich    TaoSolve() in progress then those iterations are not yet counted.
22778931d482SJason Sarich 
22788931d482SJason Sarich    Level: intermediate
22798931d482SJason Sarich 
2280db781477SPatrick Sanan .seealso: `TaoGetLinearSolveIterations()`
22818931d482SJason Sarich @*/
22828931d482SJason Sarich PetscErrorCode  TaoGetTotalIterationNumber(Tao tao,PetscInt *iter)
22838931d482SJason Sarich {
22848931d482SJason Sarich   PetscFunctionBegin;
22858931d482SJason Sarich   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
22868931d482SJason Sarich   PetscValidIntPointer(iter,2);
22878931d482SJason Sarich   *iter = tao->ntotalits;
22888931d482SJason Sarich   PetscFunctionReturn(0);
22898931d482SJason Sarich }
22908931d482SJason Sarich 
22918931d482SJason Sarich /*@
22928931d482SJason Sarich    TaoSetTotalIterationNumber - Sets the current total iteration number.
22938931d482SJason Sarich 
229494511df7SStefano Zampini    Logically Collective on Tao
22958931d482SJason Sarich 
2296d8d19677SJose E. Roman    Input Parameters:
2297a2b725a8SWilliam Gropp +  tao - Tao context
2298a2b725a8SWilliam Gropp -  iter - iteration number
22998931d482SJason Sarich 
23008931d482SJason Sarich    Level: developer
23018931d482SJason Sarich 
2302db781477SPatrick Sanan .seealso: `TaoGetLinearSolveIterations()`
23038931d482SJason Sarich @*/
23048931d482SJason Sarich PetscErrorCode  TaoSetTotalIterationNumber(Tao tao,PetscInt iter)
23058931d482SJason Sarich {
23068931d482SJason Sarich   PetscFunctionBegin;
23078931d482SJason Sarich   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
230894511df7SStefano Zampini   PetscValidLogicalCollectiveInt(tao,iter,2);
23099566063dSJacob Faibussowitsch   PetscCall(PetscObjectSAWsTakeAccess((PetscObject)tao));
23108931d482SJason Sarich   tao->ntotalits = iter;
23119566063dSJacob Faibussowitsch   PetscCall(PetscObjectSAWsGrantAccess((PetscObject)tao));
23128931d482SJason Sarich   PetscFunctionReturn(0);
23138931d482SJason Sarich }
23148931d482SJason Sarich 
2315a7e14dcfSSatish Balay /*@
2316e4cb33bbSBarry Smith   TaoSetConvergedReason - Sets the termination flag on a Tao object
2317a7e14dcfSSatish Balay 
2318441846f8SBarry Smith   Logically Collective on Tao
2319a7e14dcfSSatish Balay 
2320a7e14dcfSSatish Balay   Input Parameters:
2321441846f8SBarry Smith + tao - the Tao context
2322a7e14dcfSSatish Balay - reason - one of
2323a7e14dcfSSatish Balay $     TAO_CONVERGED_ATOL (2),
2324a7e14dcfSSatish Balay $     TAO_CONVERGED_RTOL (3),
2325a7e14dcfSSatish Balay $     TAO_CONVERGED_STEPTOL (4),
2326a7e14dcfSSatish Balay $     TAO_CONVERGED_MINF (5),
2327a7e14dcfSSatish Balay $     TAO_CONVERGED_USER (6),
2328a7e14dcfSSatish Balay $     TAO_DIVERGED_MAXITS (-2),
2329a7e14dcfSSatish Balay $     TAO_DIVERGED_NAN (-4),
2330a7e14dcfSSatish Balay $     TAO_DIVERGED_MAXFCN (-5),
2331a7e14dcfSSatish Balay $     TAO_DIVERGED_LS_FAILURE (-6),
2332a7e14dcfSSatish Balay $     TAO_DIVERGED_TR_REDUCTION (-7),
2333a7e14dcfSSatish Balay $     TAO_DIVERGED_USER (-8),
2334a7e14dcfSSatish Balay $     TAO_CONTINUE_ITERATING (0)
2335a7e14dcfSSatish Balay 
2336a7e14dcfSSatish Balay    Level: intermediate
2337a7e14dcfSSatish Balay 
2338a7e14dcfSSatish Balay @*/
2339e4cb33bbSBarry Smith PetscErrorCode TaoSetConvergedReason(Tao tao, TaoConvergedReason reason)
2340a7e14dcfSSatish Balay {
2341a7e14dcfSSatish Balay   PetscFunctionBegin;
234294511df7SStefano Zampini   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
234394511df7SStefano Zampini   PetscValidLogicalCollectiveEnum(tao,reason,2);
2344a7e14dcfSSatish Balay   tao->reason = reason;
2345a7e14dcfSSatish Balay   PetscFunctionReturn(0);
2346a7e14dcfSSatish Balay }
2347a7e14dcfSSatish Balay 
2348a7e14dcfSSatish Balay /*@
2349e4cb33bbSBarry Smith    TaoGetConvergedReason - Gets the reason the Tao iteration was stopped.
2350a7e14dcfSSatish Balay 
2351a7e14dcfSSatish Balay    Not Collective
2352a7e14dcfSSatish Balay 
2353a7e14dcfSSatish Balay    Input Parameter:
2354441846f8SBarry Smith .  tao - the Tao solver context
2355a7e14dcfSSatish Balay 
2356a7e14dcfSSatish Balay    Output Parameter:
2357a7e14dcfSSatish Balay .  reason - one of
2358a7e14dcfSSatish Balay $  TAO_CONVERGED_GATOL (3)           ||g(X)|| < gatol
2359a7e14dcfSSatish Balay $  TAO_CONVERGED_GRTOL (4)           ||g(X)|| / f(X)  < grtol
2360a7e14dcfSSatish Balay $  TAO_CONVERGED_GTTOL (5)           ||g(X)|| / ||g(X0)|| < gttol
2361a7e14dcfSSatish Balay $  TAO_CONVERGED_STEPTOL (6)         step size small
2362a7e14dcfSSatish Balay $  TAO_CONVERGED_MINF (7)            F < F_min
2363a7e14dcfSSatish Balay $  TAO_CONVERGED_USER (8)            User defined
2364a7e14dcfSSatish Balay $  TAO_DIVERGED_MAXITS (-2)          its > maxits
2365a7e14dcfSSatish Balay $  TAO_DIVERGED_NAN (-4)             Numerical problems
2366a7e14dcfSSatish Balay $  TAO_DIVERGED_MAXFCN (-5)          fevals > max_funcsals
2367a7e14dcfSSatish Balay $  TAO_DIVERGED_LS_FAILURE (-6)      line search failure
2368a7e14dcfSSatish Balay $  TAO_DIVERGED_TR_REDUCTION (-7)    trust region failure
2369a7e14dcfSSatish Balay $  TAO_DIVERGED_USER(-8)             (user defined)
2370a7e14dcfSSatish Balay $  TAO_CONTINUE_ITERATING (0)
2371a7e14dcfSSatish Balay 
2372a7e14dcfSSatish Balay    where
2373a7e14dcfSSatish Balay +  X - current solution
2374a7e14dcfSSatish Balay .  X0 - initial guess
2375a7e14dcfSSatish Balay .  f(X) - current function value
2376a7e14dcfSSatish Balay .  f(X*) - true solution (estimated)
2377a7e14dcfSSatish Balay .  g(X) - current gradient
2378a7e14dcfSSatish Balay .  its - current iterate number
2379a7e14dcfSSatish Balay .  maxits - maximum number of iterates
2380a7e14dcfSSatish Balay .  fevals - number of function evaluations
2381a7e14dcfSSatish Balay -  max_funcsals - maximum number of function evaluations
2382a7e14dcfSSatish Balay 
2383a7e14dcfSSatish Balay    Level: intermediate
2384a7e14dcfSSatish Balay 
2385db781477SPatrick Sanan .seealso: `TaoSetConvergenceTest()`, `TaoSetTolerances()`
2386a7e14dcfSSatish Balay 
2387a7e14dcfSSatish Balay @*/
2388e4cb33bbSBarry Smith PetscErrorCode TaoGetConvergedReason(Tao tao, TaoConvergedReason *reason)
2389a7e14dcfSSatish Balay {
2390a7e14dcfSSatish Balay   PetscFunctionBegin;
2391441846f8SBarry Smith   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
2392a7e14dcfSSatish Balay   PetscValidPointer(reason,2);
2393a7e14dcfSSatish Balay   *reason = tao->reason;
2394a7e14dcfSSatish Balay   PetscFunctionReturn(0);
2395a7e14dcfSSatish Balay }
2396a7e14dcfSSatish Balay 
2397a7e14dcfSSatish Balay /*@
2398a7e14dcfSSatish Balay    TaoGetSolutionStatus - Get the current iterate, objective value,
2399a7e14dcfSSatish Balay    residual, infeasibility, and termination
2400a7e14dcfSSatish Balay 
2401a7e14dcfSSatish Balay    Not Collective
2402a7e14dcfSSatish Balay 
2403f899ff85SJose E. Roman    Input Parameter:
2404441846f8SBarry Smith .  tao - the Tao context
2405a7e14dcfSSatish Balay 
2406a7e14dcfSSatish Balay    Output Parameters:
2407a7e14dcfSSatish Balay +  iterate - the current iterate number (>=0)
2408a7e14dcfSSatish Balay .  f - the current function value
240958417fe7SBarry Smith .  gnorm - the square of the gradient norm, duality gap, or other measure indicating distance from optimality.
2410a7e14dcfSSatish Balay .  cnorm - the infeasibility of the current solution with regard to the constraints.
2411a7e14dcfSSatish Balay .  xdiff - the step length or trust region radius of the most recent iterate.
2412a7e14dcfSSatish Balay -  reason - The termination reason, which can equal TAO_CONTINUE_ITERATING
2413a7e14dcfSSatish Balay 
2414a7e14dcfSSatish Balay    Level: intermediate
2415a7e14dcfSSatish Balay 
2416a7e14dcfSSatish Balay    Note:
2417a7e14dcfSSatish Balay    TAO returns the values set by the solvers in the routine TaoMonitor().
2418a7e14dcfSSatish Balay 
2419a7e14dcfSSatish Balay    Note:
242047a47007SBarry Smith    If any of the output arguments are set to NULL, no corresponding value will be returned.
2421a7e14dcfSSatish Balay 
2422db781477SPatrick Sanan .seealso: `TaoMonitor()`, `TaoGetConvergedReason()`
2423a7e14dcfSSatish Balay @*/
2424e4cb33bbSBarry Smith PetscErrorCode TaoGetSolutionStatus(Tao tao, PetscInt *its, PetscReal *f, PetscReal *gnorm, PetscReal *cnorm, PetscReal *xdiff, TaoConvergedReason *reason)
2425a7e14dcfSSatish Balay {
2426a7e14dcfSSatish Balay   PetscFunctionBegin;
242794511df7SStefano Zampini   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
2428a7e14dcfSSatish Balay   if (its) *its = tao->niter;
2429a7e14dcfSSatish Balay   if (f) *f = tao->fc;
2430a7e14dcfSSatish Balay   if (gnorm) *gnorm = tao->residual;
2431a7e14dcfSSatish Balay   if (cnorm) *cnorm = tao->cnorm;
2432a7e14dcfSSatish Balay   if (reason) *reason = tao->reason;
2433a7e14dcfSSatish Balay   if (xdiff) *xdiff = tao->step;
2434a7e14dcfSSatish Balay   PetscFunctionReturn(0);
2435a7e14dcfSSatish Balay }
2436a7e14dcfSSatish Balay 
2437a7e14dcfSSatish Balay /*@C
2438441846f8SBarry Smith    TaoGetType - Gets the current Tao algorithm.
2439a7e14dcfSSatish Balay 
2440a7e14dcfSSatish Balay    Not Collective
2441a7e14dcfSSatish Balay 
2442a7e14dcfSSatish Balay    Input Parameter:
2443441846f8SBarry Smith .  tao - the Tao solver context
2444a7e14dcfSSatish Balay 
2445a7e14dcfSSatish Balay    Output Parameter:
2446441846f8SBarry Smith .  type - Tao method
2447a7e14dcfSSatish Balay 
2448a7e14dcfSSatish Balay    Level: intermediate
2449a7e14dcfSSatish Balay 
2450a7e14dcfSSatish Balay @*/
2451b625d6c7SJed Brown PetscErrorCode TaoGetType(Tao tao,TaoType *type)
2452a7e14dcfSSatish Balay {
2453a7e14dcfSSatish Balay   PetscFunctionBegin;
2454441846f8SBarry Smith   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
2455a7e14dcfSSatish Balay   PetscValidPointer(type,2);
2456a7e14dcfSSatish Balay   *type = ((PetscObject)tao)->type_name;
2457a7e14dcfSSatish Balay   PetscFunctionReturn(0);
2458a7e14dcfSSatish Balay }
2459a7e14dcfSSatish Balay 
2460a7e14dcfSSatish Balay /*@C
2461a7e14dcfSSatish Balay   TaoMonitor - Monitor the solver and the current solution.  This
2462a7e14dcfSSatish Balay   routine will record the iteration number and residual statistics,
2463a7e14dcfSSatish Balay   call any monitors specified by the user, and calls the convergence-check routine.
2464a7e14dcfSSatish Balay 
2465a7e14dcfSSatish Balay    Input Parameters:
2466441846f8SBarry Smith +  tao - the Tao context
2467a7e14dcfSSatish Balay .  its - the current iterate number (>=0)
2468a7e14dcfSSatish Balay .  f - the current objective function value
246958417fe7SBarry Smith .  res - the gradient norm, square root of the duality gap, or other measure indicating distince from optimality.  This measure will be recorded and
2470a7e14dcfSSatish Balay           used for some termination tests.
2471a7e14dcfSSatish Balay .  cnorm - the infeasibility of the current solution with regard to the constraints.
2472a7e14dcfSSatish Balay -  steplength - multiple of the step direction added to the previous iterate.
2473a7e14dcfSSatish Balay 
2474a7e14dcfSSatish Balay    Output Parameters:
2475a7e14dcfSSatish Balay .  reason - The termination reason, which can equal TAO_CONTINUE_ITERATING
2476a7e14dcfSSatish Balay 
2477a7e14dcfSSatish Balay    Options Database Key:
2478a7e14dcfSSatish Balay .  -tao_monitor - Use the default monitor, which prints statistics to standard output
2479a7e14dcfSSatish Balay 
2480db781477SPatrick Sanan .seealso `TaoGetConvergedReason()`, `TaoMonitorDefault()`, `TaoSetMonitor()`
2481a7e14dcfSSatish Balay 
2482a7e14dcfSSatish Balay    Level: developer
2483a7e14dcfSSatish Balay 
2484a7e14dcfSSatish Balay @*/
24853ecd9318SAlp Dener PetscErrorCode TaoMonitor(Tao tao, PetscInt its, PetscReal f, PetscReal res, PetscReal cnorm, PetscReal steplength)
2486a7e14dcfSSatish Balay {
248747a47007SBarry Smith   PetscInt       i;
248847a47007SBarry Smith 
2489a7e14dcfSSatish Balay   PetscFunctionBegin;
2490441846f8SBarry Smith   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
2491a7e14dcfSSatish Balay   tao->fc = f;
2492a7e14dcfSSatish Balay   tao->residual = res;
2493a7e14dcfSSatish Balay   tao->cnorm = cnorm;
2494a7e14dcfSSatish Balay   tao->step = steplength;
2495e52336cbSBarry Smith   if (!its) {
249694511df7SStefano Zampini     tao->cnorm0 = cnorm;
249794511df7SStefano Zampini     tao->gnorm0 = res;
2498a7e14dcfSSatish Balay   }
24993c859ba3SBarry Smith   PetscCheck(!PetscIsInfOrNanReal(f) && !PetscIsInfOrNanReal(res),PetscObjectComm((PetscObject)tao),PETSC_ERR_USER, "User provided compute function generated Inf or NaN");
2500a7e14dcfSSatish Balay   for (i=0;i<tao->numbermonitors;i++) {
25019566063dSJacob Faibussowitsch     PetscCall((*tao->monitor[i])(tao,tao->monitorcontext[i]));
2502a7e14dcfSSatish Balay   }
2503a7e14dcfSSatish Balay   PetscFunctionReturn(0);
2504a7e14dcfSSatish Balay }
2505a7e14dcfSSatish Balay 
2506a7e14dcfSSatish Balay /*@
2507ae93cb3cSJason Sarich    TaoSetConvergenceHistory - Sets the array used to hold the convergence history.
2508a7e14dcfSSatish Balay 
2509441846f8SBarry Smith    Logically Collective on Tao
2510a7e14dcfSSatish Balay 
2511a7e14dcfSSatish Balay    Input Parameters:
2512441846f8SBarry Smith +  tao - the Tao solver context
2513a7e14dcfSSatish Balay .  obj   - array to hold objective value history
2514a7e14dcfSSatish Balay .  resid - array to hold residual history
2515a7e14dcfSSatish Balay .  cnorm - array to hold constraint violation history
2516be1558d8SJason Sarich .  lits - integer array holds the number of linear iterations for each Tao iteration
2517a7e14dcfSSatish Balay .  na  - size of obj, resid, and cnorm
2518a7e14dcfSSatish Balay -  reset - PetscTrue indicates each new minimization resets the history counter to zero,
2519a7e14dcfSSatish Balay            else it continues storing new values for new minimizations after the old ones
2520a7e14dcfSSatish Balay 
2521a7e14dcfSSatish Balay    Notes:
2522a7e14dcfSSatish Balay    If set, TAO will fill the given arrays with the indicated
2523ae93cb3cSJason Sarich    information at each iteration.  If 'obj','resid','cnorm','lits' are
2524ae93cb3cSJason Sarich    *all* NULL then space (using size na, or 1000 if na is PETSC_DECIDE or
2525ae93cb3cSJason Sarich    PETSC_DEFAULT) is allocated for the history.
2526be1558d8SJason Sarich    If not all are NULL, then only the non-NULL information categories
2527ae93cb3cSJason Sarich    will be stored, the others will be ignored.
2528ae93cb3cSJason Sarich 
2529ae93cb3cSJason Sarich    Any convergence information after iteration number 'na' will not be stored.
2530a7e14dcfSSatish Balay 
2531a7e14dcfSSatish Balay    This routine is useful, e.g., when running a code for purposes
2532a7e14dcfSSatish Balay    of accurate performance monitoring, when no I/O should be done
2533a7e14dcfSSatish Balay    during the section of code that is being timed.
2534a7e14dcfSSatish Balay 
2535a7e14dcfSSatish Balay    Level: intermediate
2536a7e14dcfSSatish Balay 
2537db781477SPatrick Sanan .seealso: `TaoGetConvergenceHistory()`
2538a7e14dcfSSatish Balay 
2539a7e14dcfSSatish Balay @*/
25401b266c99SBarry Smith PetscErrorCode TaoSetConvergenceHistory(Tao tao, PetscReal obj[], PetscReal resid[], PetscReal cnorm[], PetscInt lits[], PetscInt na,PetscBool reset)
2541a7e14dcfSSatish Balay {
2542a7e14dcfSSatish Balay   PetscFunctionBegin;
2543441846f8SBarry Smith   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
2544064a246eSJacob Faibussowitsch   if (obj) PetscValidRealPointer(obj,2);
2545064a246eSJacob Faibussowitsch   if (resid) PetscValidRealPointer(resid,3);
2546064a246eSJacob Faibussowitsch   if (cnorm) PetscValidRealPointer(cnorm,4);
2547ae93cb3cSJason Sarich   if (lits) PetscValidIntPointer(lits,5);
2548ae93cb3cSJason Sarich 
2549ae93cb3cSJason Sarich   if (na == PETSC_DECIDE || na == PETSC_DEFAULT) na = 1000;
2550ae93cb3cSJason Sarich   if (!obj && !resid && !cnorm && !lits) {
25519566063dSJacob Faibussowitsch     PetscCall(PetscCalloc4(na,&obj,na,&resid,na,&cnorm,na,&lits));
2552ae93cb3cSJason Sarich     tao->hist_malloc = PETSC_TRUE;
2553ae93cb3cSJason Sarich   }
2554ae93cb3cSJason Sarich 
2555a7e14dcfSSatish Balay   tao->hist_obj = obj;
2556a7e14dcfSSatish Balay   tao->hist_resid = resid;
2557a7e14dcfSSatish Balay   tao->hist_cnorm = cnorm;
2558ae93cb3cSJason Sarich   tao->hist_lits = lits;
2559a7e14dcfSSatish Balay   tao->hist_max   = na;
2560a7e14dcfSSatish Balay   tao->hist_reset = reset;
2561ae93cb3cSJason Sarich   tao->hist_len = 0;
2562a7e14dcfSSatish Balay   PetscFunctionReturn(0);
2563a7e14dcfSSatish Balay }
2564a7e14dcfSSatish Balay 
2565a7e14dcfSSatish Balay /*@C
2566ae93cb3cSJason Sarich    TaoGetConvergenceHistory - Gets the arrays used to hold the convergence history.
2567a7e14dcfSSatish Balay 
2568441846f8SBarry Smith    Collective on Tao
2569a7e14dcfSSatish Balay 
2570a7e14dcfSSatish Balay    Input Parameter:
2571441846f8SBarry Smith .  tao - the Tao context
2572a7e14dcfSSatish Balay 
2573a7e14dcfSSatish Balay    Output Parameters:
2574a7e14dcfSSatish Balay +  obj   - array used to hold objective value history
2575a7e14dcfSSatish Balay .  resid - array used to hold residual history
2576a7e14dcfSSatish Balay .  cnorm - array used to hold constraint violation history
2577ae93cb3cSJason Sarich .  lits  - integer array used to hold linear solver iteration count
2578dd25e5eeSStefano Zampini -  nhist  - size of obj, resid, cnorm, and lits
2579a7e14dcfSSatish Balay 
2580a7e14dcfSSatish Balay    Notes:
2581ae93cb3cSJason Sarich     This routine must be preceded by calls to TaoSetConvergenceHistory()
2582ae93cb3cSJason Sarich     and TaoSolve(), otherwise it returns useless information.
2583ae93cb3cSJason Sarich 
2584a7e14dcfSSatish Balay     The calling sequence for this routine in Fortran is
2585ae93cb3cSJason Sarich $   call TaoGetConvergenceHistory(Tao tao, PetscInt nhist, PetscErrorCode ierr)
2586a7e14dcfSSatish Balay 
2587a7e14dcfSSatish Balay    This routine is useful, e.g., when running a code for purposes
2588a7e14dcfSSatish Balay    of accurate performance monitoring, when no I/O should be done
2589a7e14dcfSSatish Balay    during the section of code that is being timed.
2590a7e14dcfSSatish Balay 
2591a7e14dcfSSatish Balay    Level: advanced
2592a7e14dcfSSatish Balay 
2593db781477SPatrick Sanan .seealso: `TaoSetConvergenceHistory()`
2594a7e14dcfSSatish Balay 
2595a7e14dcfSSatish Balay @*/
2596ae93cb3cSJason Sarich PetscErrorCode TaoGetConvergenceHistory(Tao tao, PetscReal **obj, PetscReal **resid, PetscReal **cnorm, PetscInt **lits, PetscInt *nhist)
2597a7e14dcfSSatish Balay {
2598a7e14dcfSSatish Balay   PetscFunctionBegin;
2599441846f8SBarry Smith   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
2600a7e14dcfSSatish Balay   if (obj)   *obj   = tao->hist_obj;
2601a7e14dcfSSatish Balay   if (cnorm) *cnorm = tao->hist_cnorm;
2602a7e14dcfSSatish Balay   if (resid) *resid = tao->hist_resid;
2603a7e14dcfSSatish Balay   if (nhist) *nhist = tao->hist_len;
2604a7e14dcfSSatish Balay   PetscFunctionReturn(0);
2605a7e14dcfSSatish Balay }
2606a7e14dcfSSatish Balay 
2607a7e14dcfSSatish Balay /*@
2608a7e14dcfSSatish Balay    TaoSetApplicationContext - Sets the optional user-defined context for
2609a7e14dcfSSatish Balay    a solver.
2610a7e14dcfSSatish Balay 
2611441846f8SBarry Smith    Logically Collective on Tao
2612a7e14dcfSSatish Balay 
2613a7e14dcfSSatish Balay    Input Parameters:
2614441846f8SBarry Smith +  tao  - the Tao context
2615a7e14dcfSSatish Balay -  usrP - optional user context
2616a7e14dcfSSatish Balay 
2617a7e14dcfSSatish Balay    Level: intermediate
2618a7e14dcfSSatish Balay 
2619db781477SPatrick Sanan .seealso: `TaoGetApplicationContext()`, `TaoSetApplicationContext()`
2620a7e14dcfSSatish Balay @*/
2621441846f8SBarry Smith PetscErrorCode  TaoSetApplicationContext(Tao tao,void *usrP)
2622a7e14dcfSSatish Balay {
2623a7e14dcfSSatish Balay   PetscFunctionBegin;
2624441846f8SBarry Smith   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
2625a7e14dcfSSatish Balay   tao->user = usrP;
2626a7e14dcfSSatish Balay   PetscFunctionReturn(0);
2627a7e14dcfSSatish Balay }
2628a7e14dcfSSatish Balay 
2629a7e14dcfSSatish Balay /*@
2630a7e14dcfSSatish Balay    TaoGetApplicationContext - Gets the user-defined context for a
2631a7e14dcfSSatish Balay    TAO solvers.
2632a7e14dcfSSatish Balay 
2633a7e14dcfSSatish Balay    Not Collective
2634a7e14dcfSSatish Balay 
2635a7e14dcfSSatish Balay    Input Parameter:
2636441846f8SBarry Smith .  tao  - Tao context
2637a7e14dcfSSatish Balay 
2638a7e14dcfSSatish Balay    Output Parameter:
2639a7e14dcfSSatish Balay .  usrP - user context
2640a7e14dcfSSatish Balay 
2641a7e14dcfSSatish Balay    Level: intermediate
2642a7e14dcfSSatish Balay 
2643db781477SPatrick Sanan .seealso: `TaoSetApplicationContext()`
2644a7e14dcfSSatish Balay @*/
2645441846f8SBarry Smith PetscErrorCode  TaoGetApplicationContext(Tao tao,void *usrP)
2646a7e14dcfSSatish Balay {
2647a7e14dcfSSatish Balay   PetscFunctionBegin;
2648441846f8SBarry Smith   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
264994511df7SStefano Zampini   PetscValidPointer(usrP,2);
2650a7e14dcfSSatish Balay   *(void**)usrP = tao->user;
2651a7e14dcfSSatish Balay   PetscFunctionReturn(0);
2652a7e14dcfSSatish Balay }
2653a9603a14SPatrick Farrell 
2654a9603a14SPatrick Farrell /*@
2655a9603a14SPatrick Farrell    TaoSetGradientNorm - Sets the matrix used to define the inner product that measures the size of the gradient.
2656a9603a14SPatrick Farrell 
2657a9603a14SPatrick Farrell    Collective on tao
2658a9603a14SPatrick Farrell 
2659a9603a14SPatrick Farrell    Input Parameters:
2660a9603a14SPatrick Farrell +  tao  - the Tao context
2661a9603a14SPatrick Farrell -  M    - gradient norm
2662a9603a14SPatrick Farrell 
2663a9603a14SPatrick Farrell    Level: beginner
2664a9603a14SPatrick Farrell 
2665db781477SPatrick Sanan .seealso: `TaoGetGradientNorm()`, `TaoGradientNorm()`
2666a9603a14SPatrick Farrell @*/
2667a9603a14SPatrick Farrell PetscErrorCode  TaoSetGradientNorm(Tao tao, Mat M)
2668a9603a14SPatrick Farrell {
2669a9603a14SPatrick Farrell   PetscFunctionBegin;
2670a9603a14SPatrick Farrell   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
267194511df7SStefano Zampini   PetscValidHeaderSpecific(M,MAT_CLASSID,2);
26729566063dSJacob Faibussowitsch   PetscCall(PetscObjectReference((PetscObject)M));
26739566063dSJacob Faibussowitsch   PetscCall(MatDestroy(&tao->gradient_norm));
26749566063dSJacob Faibussowitsch   PetscCall(VecDestroy(&tao->gradient_norm_tmp));
2675a9603a14SPatrick Farrell   tao->gradient_norm = M;
26769566063dSJacob Faibussowitsch   PetscCall(MatCreateVecs(M, NULL, &tao->gradient_norm_tmp));
2677a9603a14SPatrick Farrell   PetscFunctionReturn(0);
2678a9603a14SPatrick Farrell }
2679a9603a14SPatrick Farrell 
2680a9603a14SPatrick Farrell /*@
2681a9603a14SPatrick Farrell    TaoGetGradientNorm - Returns the matrix used to define the inner product for measuring the size of the gradient.
2682a9603a14SPatrick Farrell 
2683a9603a14SPatrick Farrell    Not Collective
2684a9603a14SPatrick Farrell 
2685a9603a14SPatrick Farrell    Input Parameter:
2686a9603a14SPatrick Farrell .  tao  - Tao context
2687a9603a14SPatrick Farrell 
2688a9603a14SPatrick Farrell    Output Parameter:
2689a9603a14SPatrick Farrell .  M - gradient norm
2690a9603a14SPatrick Farrell 
2691a9603a14SPatrick Farrell    Level: beginner
2692a9603a14SPatrick Farrell 
2693db781477SPatrick Sanan .seealso: `TaoSetGradientNorm()`, `TaoGradientNorm()`
2694a9603a14SPatrick Farrell @*/
2695a9603a14SPatrick Farrell PetscErrorCode  TaoGetGradientNorm(Tao tao, Mat *M)
2696a9603a14SPatrick Farrell {
2697a9603a14SPatrick Farrell   PetscFunctionBegin;
2698a9603a14SPatrick Farrell   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
269994511df7SStefano Zampini   PetscValidPointer(M,2);
2700a9603a14SPatrick Farrell   *M = tao->gradient_norm;
2701a9603a14SPatrick Farrell   PetscFunctionReturn(0);
2702a9603a14SPatrick Farrell }
2703a9603a14SPatrick Farrell 
270405aca2cdSSatish Balay /*@C
2705a9603a14SPatrick Farrell    TaoGradientNorm - Compute the norm with respect to the inner product the user has set.
2706a9603a14SPatrick Farrell 
2707a9603a14SPatrick Farrell    Collective on tao
2708a9603a14SPatrick Farrell 
2709d8d19677SJose E. Roman    Input Parameters:
27102653c764SPierre Jolivet +  tao      - the Tao context
2711a9603a14SPatrick Farrell .  gradient - the gradient to be computed
27122653c764SPierre Jolivet -  norm     - the norm type
2713a9603a14SPatrick Farrell 
2714a9603a14SPatrick Farrell    Output Parameter:
2715a9603a14SPatrick Farrell .  gnorm    - the gradient norm
2716a9603a14SPatrick Farrell 
2717a9603a14SPatrick Farrell    Level: developer
2718a9603a14SPatrick Farrell 
2719db781477SPatrick Sanan .seealso: `TaoSetGradientNorm()`, `TaoGetGradientNorm()`
2720a9603a14SPatrick Farrell @*/
2721a9603a14SPatrick Farrell PetscErrorCode  TaoGradientNorm(Tao tao, Vec gradient, NormType type, PetscReal *gnorm)
2722a9603a14SPatrick Farrell {
2723a9603a14SPatrick Farrell   PetscFunctionBegin;
27248854b543SStefano Zampini   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
27258854b543SStefano Zampini   PetscValidHeaderSpecific(gradient,VEC_CLASSID,2);
27268854b543SStefano Zampini   PetscValidLogicalCollectiveEnum(tao,type,3);
272794511df7SStefano Zampini   PetscValidRealPointer(gnorm,4);
2728a9603a14SPatrick Farrell   if (tao->gradient_norm) {
2729680e2bc7SPatrick Farrell     PetscScalar gnorms;
2730680e2bc7SPatrick Farrell 
27313c859ba3SBarry 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.");
27329566063dSJacob Faibussowitsch     PetscCall(MatMult(tao->gradient_norm, gradient, tao->gradient_norm_tmp));
27339566063dSJacob Faibussowitsch     PetscCall(VecDot(gradient, tao->gradient_norm_tmp, &gnorms));
27347bb79937SPatrick Farrell     *gnorm = PetscRealPart(PetscSqrtScalar(gnorms));
2735a9603a14SPatrick Farrell   } else {
27369566063dSJacob Faibussowitsch     PetscCall(VecNorm(gradient, type, gnorm));
2737a9603a14SPatrick Farrell   }
2738a9603a14SPatrick Farrell   PetscFunctionReturn(0);
2739a9603a14SPatrick Farrell }
2740a9603a14SPatrick Farrell 
2741e882e171SHong Zhang /*@C
2742e882e171SHong Zhang    TaoMonitorDrawCtxCreate - Creates the monitor context for TaoMonitorDrawCtx
2743a9603a14SPatrick Farrell 
2744e882e171SHong Zhang    Collective on Tao
2745e882e171SHong Zhang 
2746e882e171SHong Zhang    Output Patameter:
2747e882e171SHong Zhang .    ctx - the monitor context
2748e882e171SHong Zhang 
2749e882e171SHong Zhang    Options Database:
2750e882e171SHong Zhang .   -tao_draw_solution_initial - show initial guess as well as current solution
2751e882e171SHong Zhang 
2752e882e171SHong Zhang    Level: intermediate
2753e882e171SHong Zhang 
2754db781477SPatrick Sanan .seealso: `TaoMonitorSet()`, `TaoMonitorDefault()`, `VecView()`, `TaoMonitorDrawCtx()`
2755e882e171SHong Zhang @*/
2756e882e171SHong Zhang PetscErrorCode  TaoMonitorDrawCtxCreate(MPI_Comm comm,const char host[],const char label[],int x,int y,int m,int n,PetscInt howoften,TaoMonitorDrawCtx *ctx)
2757e882e171SHong Zhang {
2758e882e171SHong Zhang   PetscFunctionBegin;
27599566063dSJacob Faibussowitsch   PetscCall(PetscNew(ctx));
27609566063dSJacob Faibussowitsch   PetscCall(PetscViewerDrawOpen(comm,host,label,x,y,m,n,&(*ctx)->viewer));
27619566063dSJacob Faibussowitsch   PetscCall(PetscViewerSetFromOptions((*ctx)->viewer));
2762e882e171SHong Zhang   (*ctx)->howoften = howoften;
2763e882e171SHong Zhang   PetscFunctionReturn(0);
2764e882e171SHong Zhang }
2765e882e171SHong Zhang 
2766e882e171SHong Zhang /*@C
2767e882e171SHong Zhang    TaoMonitorDrawCtxDestroy - Destroys the monitor context for TaoMonitorDrawSolution()
2768e882e171SHong Zhang 
2769e882e171SHong Zhang    Collective on Tao
2770e882e171SHong Zhang 
2771e882e171SHong Zhang    Input Parameters:
2772e882e171SHong Zhang .    ctx - the monitor context
2773e882e171SHong Zhang 
2774e882e171SHong Zhang    Level: intermediate
2775e882e171SHong Zhang 
2776db781477SPatrick Sanan .seealso: `TaoMonitorSet()`, `TaoMonitorDefault()`, `VecView()`, `TaoMonitorDrawSolution()`
2777e882e171SHong Zhang @*/
2778e882e171SHong Zhang PetscErrorCode  TaoMonitorDrawCtxDestroy(TaoMonitorDrawCtx *ictx)
2779e882e171SHong Zhang {
2780e882e171SHong Zhang   PetscFunctionBegin;
27819566063dSJacob Faibussowitsch   PetscCall(PetscViewerDestroy(&(*ictx)->viewer));
27829566063dSJacob Faibussowitsch   PetscCall(PetscFree(*ictx));
2783e882e171SHong Zhang   PetscFunctionReturn(0);
2784e882e171SHong Zhang }
2785