xref: /petsc/src/tao/interface/taosolver.c (revision 6aad120caa16b1027d343a5f30f73d01448e4dc0)
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 
54a7e14dcfSSatish Balay .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 
112a82e8c82SStefano Zampini .seealso: TaoCreate(), TaoSetObjective(), TaoSetGradient(), TaoSetHessian(), TaoGetConvergedReason()
113a7e14dcfSSatish Balay  @*/
114441846f8SBarry Smith PetscErrorCode TaoSolve(Tao tao)
115a7e14dcfSSatish Balay {
116a7e14dcfSSatish Balay   PetscErrorCode   ierr;
117e2379f4fSBarry Smith   static PetscBool set = PETSC_FALSE;
11847a47007SBarry Smith 
119a7e14dcfSSatish Balay   PetscFunctionBegin;
120441846f8SBarry Smith   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
121e2379f4fSBarry Smith   ierr = PetscCitationsRegister("@TechReport{tao-user-ref,\n"
122e2379f4fSBarry Smith                                 "title   = {Toolkit for Advanced Optimization (TAO) Users Manual},\n"
123e2379f4fSBarry Smith                                 "author  = {Todd Munson and Jason Sarich and Stefan Wild and Steve Benson and Lois Curfman McInnes},\n"
124e2379f4fSBarry Smith                                 "Institution = {Argonne National Laboratory},\n"
125e2379f4fSBarry Smith                                 "Year   = 2014,\n"
126e2379f4fSBarry Smith                                 "Number = {ANL/MCS-TM-322 - Revision 3.5},\n"
1279566063dSJacob Faibussowitsch                                 "url    = {https://www.mcs.anl.gov/research/projects/tao/}\n}\n",&set);PetscCall(ierr);
128494bef23SAlp Dener   tao->header_printed = PETSC_FALSE;
1299566063dSJacob Faibussowitsch   PetscCall(TaoSetUp(tao));
1309566063dSJacob Faibussowitsch   PetscCall(TaoResetStatistics(tao));
131a7e14dcfSSatish Balay   if (tao->linesearch) {
1329566063dSJacob Faibussowitsch     PetscCall(TaoLineSearchReset(tao->linesearch));
133a7e14dcfSSatish Balay   }
134a7e14dcfSSatish Balay 
1359566063dSJacob Faibussowitsch   PetscCall(PetscLogEventBegin(TAO_Solve,tao,0,0,0));
1369566063dSJacob Faibussowitsch   if (tao->ops->solve) PetscCall((*tao->ops->solve)(tao));
1379566063dSJacob Faibussowitsch   PetscCall(PetscLogEventEnd(TAO_Solve,tao,0,0,0));
138a7e14dcfSSatish Balay 
1399566063dSJacob Faibussowitsch   PetscCall(VecViewFromOptions(tao->solution,(PetscObject)tao,"-tao_view_solution"));
1407cf37e64SBarry Smith 
1418931d482SJason Sarich   tao->ntotalits += tao->niter;
1429566063dSJacob Faibussowitsch   PetscCall(TaoViewFromOptions(tao,NULL,"-tao_view"));
143a7e14dcfSSatish Balay 
144a7e14dcfSSatish Balay   if (tao->printreason) {
145a7e14dcfSSatish Balay     if (tao->reason > 0) {
1469566063dSJacob Faibussowitsch       PetscCall(PetscPrintf(((PetscObject)tao)->comm,"TAO solve converged due to %s iterations %D\n",TaoConvergedReasons[tao->reason],tao->niter));
147a7e14dcfSSatish Balay     } else {
1489566063dSJacob Faibussowitsch       PetscCall(PetscPrintf(((PetscObject)tao)->comm,"TAO solve did not converge due to %s iteration %D\n",TaoConvergedReasons[tao->reason],tao->niter));
149a7e14dcfSSatish Balay     }
150a7e14dcfSSatish Balay   }
151a7e14dcfSSatish Balay   PetscFunctionReturn(0);
152a7e14dcfSSatish Balay }
153a7e14dcfSSatish Balay 
154a7e14dcfSSatish Balay /*@
155a7e14dcfSSatish Balay   TaoSetUp - Sets up the internal data structures for the later use
156a7e14dcfSSatish Balay   of a Tao solver
157a7e14dcfSSatish Balay 
158a7e14dcfSSatish Balay   Collective on tao
159a7e14dcfSSatish Balay 
160a7e14dcfSSatish Balay   Input Parameters:
161a7e14dcfSSatish Balay . tao - the TAO context
162a7e14dcfSSatish Balay 
163a7e14dcfSSatish Balay   Notes:
164a7e14dcfSSatish Balay   The user will not need to explicitly call TaoSetUp(), as it will
165a7e14dcfSSatish Balay   automatically be called in TaoSolve().  However, if the user
166a7e14dcfSSatish Balay   desires to call it explicitly, it should come after TaoCreate()
167a7e14dcfSSatish Balay   and any TaoSetSomething() routines, but before TaoSolve().
168a7e14dcfSSatish Balay 
169a7e14dcfSSatish Balay   Level: advanced
170a7e14dcfSSatish Balay 
171a7e14dcfSSatish Balay .seealso: TaoCreate(), TaoSolve()
172a7e14dcfSSatish Balay @*/
173441846f8SBarry Smith PetscErrorCode TaoSetUp(Tao tao)
174a7e14dcfSSatish Balay {
175a7e14dcfSSatish Balay   PetscFunctionBegin;
176441846f8SBarry Smith   PetscValidHeaderSpecific(tao, TAO_CLASSID,1);
177a7e14dcfSSatish Balay   if (tao->setupcalled) PetscFunctionReturn(0);
1783c859ba3SBarry Smith   PetscCheck(tao->solution,PetscObjectComm((PetscObject)tao),PETSC_ERR_ARG_WRONGSTATE,"Must call TaoSetSolution");
179a7e14dcfSSatish Balay   if (tao->ops->setup) {
1809566063dSJacob Faibussowitsch     PetscCall((*tao->ops->setup)(tao));
181a7e14dcfSSatish Balay   }
182a7e14dcfSSatish Balay   tao->setupcalled = PETSC_TRUE;
183a7e14dcfSSatish Balay   PetscFunctionReturn(0);
184a7e14dcfSSatish Balay }
185a7e14dcfSSatish Balay 
1861fb7b255SJunchao Zhang /*@C
187a7e14dcfSSatish Balay   TaoDestroy - Destroys the TAO context that was created with
188a7e14dcfSSatish Balay   TaoCreate()
189a7e14dcfSSatish Balay 
190441846f8SBarry Smith   Collective on Tao
191a7e14dcfSSatish Balay 
192a7e14dcfSSatish Balay   Input Parameter:
193441846f8SBarry Smith . tao - the Tao context
194a7e14dcfSSatish Balay 
195a7e14dcfSSatish Balay   Level: beginner
196a7e14dcfSSatish Balay 
197a7e14dcfSSatish Balay .seealso: TaoCreate(), TaoSolve()
198a7e14dcfSSatish Balay @*/
199441846f8SBarry Smith PetscErrorCode TaoDestroy(Tao *tao)
200a7e14dcfSSatish Balay {
201a7e14dcfSSatish Balay   PetscFunctionBegin;
202a7e14dcfSSatish Balay   if (!*tao) PetscFunctionReturn(0);
203441846f8SBarry Smith   PetscValidHeaderSpecific(*tao,TAO_CLASSID,1);
20483c8fe1dSLisandro Dalcin   if (--((PetscObject)*tao)->refct > 0) {*tao = NULL;PetscFunctionReturn(0);}
205a7e14dcfSSatish Balay 
206a7e14dcfSSatish Balay   if ((*tao)->ops->destroy) {
2079566063dSJacob Faibussowitsch     PetscCall((*((*tao))->ops->destroy)(*tao));
208a7e14dcfSSatish Balay   }
2099566063dSJacob Faibussowitsch   PetscCall(KSPDestroy(&(*tao)->ksp));
2109566063dSJacob Faibussowitsch   PetscCall(TaoLineSearchDestroy(&(*tao)->linesearch));
211a7e14dcfSSatish Balay 
212a7e14dcfSSatish Balay   if ((*tao)->ops->convergencedestroy) {
2139566063dSJacob Faibussowitsch     PetscCall((*(*tao)->ops->convergencedestroy)((*tao)->cnvP));
214a7e14dcfSSatish Balay     if ((*tao)->jacobian_state_inv) {
2159566063dSJacob Faibussowitsch       PetscCall(MatDestroy(&(*tao)->jacobian_state_inv));
216a7e14dcfSSatish Balay     }
217a7e14dcfSSatish Balay   }
2189566063dSJacob Faibussowitsch   PetscCall(VecDestroy(&(*tao)->solution));
2199566063dSJacob Faibussowitsch   PetscCall(VecDestroy(&(*tao)->gradient));
2209566063dSJacob Faibussowitsch   PetscCall(VecDestroy(&(*tao)->ls_res));
221a7e14dcfSSatish Balay 
222a9603a14SPatrick Farrell   if ((*tao)->gradient_norm) {
2239566063dSJacob Faibussowitsch     PetscCall(PetscObjectDereference((PetscObject)(*tao)->gradient_norm));
2249566063dSJacob Faibussowitsch     PetscCall(VecDestroy(&(*tao)->gradient_norm_tmp));
225a9603a14SPatrick Farrell   }
226a9603a14SPatrick Farrell 
2279566063dSJacob Faibussowitsch   PetscCall(VecDestroy(&(*tao)->XL));
2289566063dSJacob Faibussowitsch   PetscCall(VecDestroy(&(*tao)->XU));
2299566063dSJacob Faibussowitsch   PetscCall(VecDestroy(&(*tao)->IL));
2309566063dSJacob Faibussowitsch   PetscCall(VecDestroy(&(*tao)->IU));
2319566063dSJacob Faibussowitsch   PetscCall(VecDestroy(&(*tao)->DE));
2329566063dSJacob Faibussowitsch   PetscCall(VecDestroy(&(*tao)->DI));
2339566063dSJacob Faibussowitsch   PetscCall(VecDestroy(&(*tao)->constraints));
2349566063dSJacob Faibussowitsch   PetscCall(VecDestroy(&(*tao)->constraints_equality));
2359566063dSJacob Faibussowitsch   PetscCall(VecDestroy(&(*tao)->constraints_inequality));
2369566063dSJacob Faibussowitsch   PetscCall(VecDestroy(&(*tao)->stepdirection));
2379566063dSJacob Faibussowitsch   PetscCall(MatDestroy(&(*tao)->hessian_pre));
2389566063dSJacob Faibussowitsch   PetscCall(MatDestroy(&(*tao)->hessian));
2399566063dSJacob Faibussowitsch   PetscCall(MatDestroy(&(*tao)->ls_jac));
2409566063dSJacob Faibussowitsch   PetscCall(MatDestroy(&(*tao)->ls_jac_pre));
2419566063dSJacob Faibussowitsch   PetscCall(MatDestroy(&(*tao)->jacobian_pre));
2429566063dSJacob Faibussowitsch   PetscCall(MatDestroy(&(*tao)->jacobian));
2439566063dSJacob Faibussowitsch   PetscCall(MatDestroy(&(*tao)->jacobian_state_pre));
2449566063dSJacob Faibussowitsch   PetscCall(MatDestroy(&(*tao)->jacobian_state));
2459566063dSJacob Faibussowitsch   PetscCall(MatDestroy(&(*tao)->jacobian_state_inv));
2469566063dSJacob Faibussowitsch   PetscCall(MatDestroy(&(*tao)->jacobian_design));
2479566063dSJacob Faibussowitsch   PetscCall(MatDestroy(&(*tao)->jacobian_equality));
2489566063dSJacob Faibussowitsch   PetscCall(MatDestroy(&(*tao)->jacobian_equality_pre));
2499566063dSJacob Faibussowitsch   PetscCall(MatDestroy(&(*tao)->jacobian_inequality));
2509566063dSJacob Faibussowitsch   PetscCall(MatDestroy(&(*tao)->jacobian_inequality_pre));
2519566063dSJacob Faibussowitsch   PetscCall(ISDestroy(&(*tao)->state_is));
2529566063dSJacob Faibussowitsch   PetscCall(ISDestroy(&(*tao)->design_is));
2539566063dSJacob Faibussowitsch   PetscCall(VecDestroy(&(*tao)->res_weights_v));
2549566063dSJacob Faibussowitsch   PetscCall(TaoCancelMonitors(*tao));
255ae93cb3cSJason Sarich   if ((*tao)->hist_malloc) {
2569566063dSJacob Faibussowitsch     PetscCall(PetscFree4((*tao)->hist_obj,(*tao)->hist_resid,(*tao)->hist_cnorm,(*tao)->hist_lits));
257ae93cb3cSJason Sarich   }
258737f463aSAlp Dener   if ((*tao)->res_weights_n) {
2599566063dSJacob Faibussowitsch     PetscCall(PetscFree((*tao)->res_weights_rows));
2609566063dSJacob Faibussowitsch     PetscCall(PetscFree((*tao)->res_weights_cols));
2619566063dSJacob Faibussowitsch     PetscCall(PetscFree((*tao)->res_weights_w));
262125ddc8aSJason Sarich   }
2639566063dSJacob Faibussowitsch   PetscCall(PetscHeaderDestroy(tao));
264a7e14dcfSSatish Balay   PetscFunctionReturn(0);
265a7e14dcfSSatish Balay }
266a7e14dcfSSatish Balay 
267a7e14dcfSSatish Balay /*@
268441846f8SBarry Smith   TaoSetFromOptions - Sets various Tao parameters from user
269a7e14dcfSSatish Balay   options.
270a7e14dcfSSatish Balay 
271441846f8SBarry Smith   Collective on Tao
272a7e14dcfSSatish Balay 
27301d2d390SJose E. Roman   Input Parameter:
274441846f8SBarry Smith . tao - the Tao solver context
275a7e14dcfSSatish Balay 
276a7e14dcfSSatish Balay   options Database Keys:
27758417fe7SBarry Smith + -tao_type <type> - The algorithm that TAO uses (lmvm, nls, etc.)
278a7e14dcfSSatish Balay . -tao_gatol <gatol> - absolute error tolerance for ||gradient||
279a7e14dcfSSatish Balay . -tao_grtol <grtol> - relative error tolerance for ||gradient||
280a7e14dcfSSatish Balay . -tao_gttol <gttol> - reduction of ||gradient|| relative to initial gradient
281a7e14dcfSSatish Balay . -tao_max_it <max> - sets maximum number of iterations
282a7e14dcfSSatish Balay . -tao_max_funcs <max> - sets maximum number of function evaluations
283a7e14dcfSSatish Balay . -tao_fmin <fmin> - stop if function value reaches fmin
284a7e14dcfSSatish Balay . -tao_steptol <tol> - stop if trust region radius less than <tol>
285a7e14dcfSSatish Balay . -tao_trust0 <t> - initial trust region radius
286a7e14dcfSSatish Balay . -tao_monitor - prints function value and residual at each iteration
287a7e14dcfSSatish Balay . -tao_smonitor - same as tao_monitor, but truncates very small values
288a7e14dcfSSatish Balay . -tao_cmonitor - prints function value, residual, and constraint norm at each iteration
289a7e14dcfSSatish Balay . -tao_view_solution - prints solution vector at each iteration
2904a48860cSAlp Dener . -tao_view_ls_residual - prints least-squares residual vector at each iteration
291147403d9SBarry Smith . -tao_view_stepdirection - prints step direction vector at each iteration
292a7e14dcfSSatish Balay . -tao_view_gradient - prints gradient vector at each iteration
293a7e14dcfSSatish Balay . -tao_draw_solution - graphically view solution vector at each iteration
294a7e14dcfSSatish Balay . -tao_draw_step - graphically view step vector at each iteration
295a7e14dcfSSatish Balay . -tao_draw_gradient - graphically view gradient at each iteration
296a7e14dcfSSatish Balay . -tao_fd_gradient - use gradient computed with finite differences
297f4c1ad5cSStefano Zampini . -tao_fd_hessian - use hessian computed with finite differences
298f4c1ad5cSStefano Zampini . -tao_mf_hessian - use matrix-free hessian computed with finite differences
299a7e14dcfSSatish Balay . -tao_cancelmonitors - cancels all monitors (except those set with command line)
300441846f8SBarry Smith . -tao_view - prints information about the Tao after solving
301a7e14dcfSSatish Balay - -tao_converged_reason - prints the reason TAO stopped iterating
302a7e14dcfSSatish Balay 
303a7e14dcfSSatish Balay   Notes:
304a7e14dcfSSatish Balay   To see all options, run your program with the -help option or consult the
305a7e14dcfSSatish Balay   user's manual. Should be called after TaoCreate() but before TaoSolve()
306a7e14dcfSSatish Balay 
307a7e14dcfSSatish Balay   Level: beginner
308a7e14dcfSSatish Balay @*/
309441846f8SBarry Smith PetscErrorCode TaoSetFromOptions(Tao tao)
310a7e14dcfSSatish Balay {
311a7e14dcfSSatish Balay   PetscErrorCode ierr;
312b625d6c7SJed Brown   TaoType        default_type = TAOLMVM;
313a7e14dcfSSatish Balay   char           type[256], monfilename[PETSC_MAX_PATH_LEN];
314a7e14dcfSSatish Balay   PetscViewer    monviewer;
315a7e14dcfSSatish Balay   PetscBool      flg;
316a7e14dcfSSatish Balay   MPI_Comm       comm;
317a7e14dcfSSatish Balay 
318a7e14dcfSSatish Balay   PetscFunctionBegin;
319441846f8SBarry Smith   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
3209566063dSJacob Faibussowitsch   PetscCall(PetscObjectGetComm((PetscObject)tao,&comm));
321125ddc8aSJason Sarich 
322a7e14dcfSSatish Balay   /* So no warnings are given about unused options */
3239566063dSJacob Faibussowitsch   PetscCall(PetscOptionsHasName(((PetscObject)tao)->options,((PetscObject)tao)->prefix,"-tao_ls_type",&flg));
324a7e14dcfSSatish Balay 
3259566063dSJacob Faibussowitsch   ierr = PetscObjectOptionsBegin((PetscObject)tao);PetscCall(ierr);
326a7e14dcfSSatish Balay   {
327a82e8c82SStefano Zampini     if (((PetscObject)tao)->type_name) default_type = ((PetscObject)tao)->type_name;
328a7e14dcfSSatish Balay     /* Check for type from options */
3299566063dSJacob Faibussowitsch     PetscCall(PetscOptionsFList("-tao_type","Tao Solver type","TaoSetType",TaoList,default_type,type,256,&flg));
330a7e14dcfSSatish Balay     if (flg) {
3319566063dSJacob Faibussowitsch       PetscCall(TaoSetType(tao,type));
332a7e14dcfSSatish Balay     } else if (!((PetscObject)tao)->type_name) {
3339566063dSJacob Faibussowitsch       PetscCall(TaoSetType(tao,default_type));
334a7e14dcfSSatish Balay     }
335a7e14dcfSSatish Balay 
3369566063dSJacob Faibussowitsch     PetscCall(PetscOptionsReal("-tao_catol","Stop if constraints violations within","TaoSetConstraintTolerances",tao->catol,&tao->catol,&flg));
3376552cf8aSJason Sarich     if (flg) tao->catol_changed = PETSC_TRUE;
338*6aad120cSJose E. Roman     PetscCall(PetscOptionsReal("-tao_crtol","Stop if relative constraint violations within","TaoSetConstraintTolerances",tao->crtol,&tao->crtol,&flg));
3396552cf8aSJason Sarich     if (flg) tao->crtol_changed = PETSC_TRUE;
3409566063dSJacob Faibussowitsch     PetscCall(PetscOptionsReal("-tao_gatol","Stop if norm of gradient less than","TaoSetTolerances",tao->gatol,&tao->gatol,&flg));
3416552cf8aSJason Sarich     if (flg) tao->gatol_changed = PETSC_TRUE;
3429566063dSJacob Faibussowitsch     PetscCall(PetscOptionsReal("-tao_grtol","Stop if norm of gradient divided by the function value is less than","TaoSetTolerances",tao->grtol,&tao->grtol,&flg));
3436552cf8aSJason Sarich     if (flg) tao->grtol_changed = PETSC_TRUE;
3449566063dSJacob 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));
3456552cf8aSJason Sarich     if (flg) tao->gttol_changed = PETSC_TRUE;
3469566063dSJacob Faibussowitsch     PetscCall(PetscOptionsInt("-tao_max_it","Stop if iteration number exceeds","TaoSetMaximumIterations",tao->max_it,&tao->max_it,&flg));
3476552cf8aSJason Sarich     if (flg) tao->max_it_changed = PETSC_TRUE;
3489566063dSJacob Faibussowitsch     PetscCall(PetscOptionsInt("-tao_max_funcs","Stop if number of function evaluations exceeds","TaoSetMaximumFunctionEvaluations",tao->max_funcs,&tao->max_funcs,&flg));
3496552cf8aSJason Sarich     if (flg) tao->max_funcs_changed = PETSC_TRUE;
3509566063dSJacob Faibussowitsch     PetscCall(PetscOptionsReal("-tao_fmin","Stop if function less than","TaoSetFunctionLowerBound",tao->fmin,&tao->fmin,&flg));
3516552cf8aSJason Sarich     if (flg) tao->fmin_changed = PETSC_TRUE;
3529566063dSJacob Faibussowitsch     PetscCall(PetscOptionsReal("-tao_steptol","Stop if step size or trust region radius less than","",tao->steptol,&tao->steptol,&flg));
3536552cf8aSJason Sarich     if (flg) tao->steptol_changed = PETSC_TRUE;
3549566063dSJacob Faibussowitsch     PetscCall(PetscOptionsReal("-tao_trust0","Initial trust region radius","TaoSetTrustRegionRadius",tao->trust0,&tao->trust0,&flg));
3556552cf8aSJason Sarich     if (flg) tao->trust0_changed = PETSC_TRUE;
3569566063dSJacob Faibussowitsch     PetscCall(PetscOptionsString("-tao_view_solution","view solution vector after each evaluation","TaoSetMonitor","stdout",monfilename,sizeof(monfilename),&flg));
357a7e14dcfSSatish Balay     if (flg) {
3589566063dSJacob Faibussowitsch       PetscCall(PetscViewerASCIIOpen(comm,monfilename,&monviewer));
3599566063dSJacob Faibussowitsch       PetscCall(TaoSetMonitor(tao,TaoSolutionMonitor,monviewer,(PetscErrorCode (*)(void**))PetscViewerDestroy));
360a7e14dcfSSatish Balay     }
361a7e14dcfSSatish Balay 
3629566063dSJacob Faibussowitsch     PetscCall(PetscOptionsBool("-tao_converged_reason","Print reason for TAO converged","TaoSolve",tao->printreason,&tao->printreason,NULL));
3639566063dSJacob Faibussowitsch     PetscCall(PetscOptionsString("-tao_view_gradient","view gradient vector after each evaluation","TaoSetMonitor","stdout",monfilename,sizeof(monfilename),&flg));
364a7e14dcfSSatish Balay     if (flg) {
3659566063dSJacob Faibussowitsch       PetscCall(PetscViewerASCIIOpen(comm,monfilename,&monviewer));
3669566063dSJacob Faibussowitsch       PetscCall(TaoSetMonitor(tao,TaoGradientMonitor,monviewer,(PetscErrorCode (*)(void**))PetscViewerDestroy));
367a7e14dcfSSatish Balay     }
368a7e14dcfSSatish Balay 
3699566063dSJacob Faibussowitsch     PetscCall(PetscOptionsString("-tao_view_stepdirection","view step direction vector after each iteration","TaoSetMonitor","stdout",monfilename,sizeof(monfilename),&flg));
370a7e14dcfSSatish Balay     if (flg) {
3719566063dSJacob Faibussowitsch       PetscCall(PetscViewerASCIIOpen(comm,monfilename,&monviewer));
3729566063dSJacob Faibussowitsch       PetscCall(TaoSetMonitor(tao,TaoStepDirectionMonitor,monviewer,(PetscErrorCode (*)(void**))PetscViewerDestroy));
373a7e14dcfSSatish Balay     }
374a7e14dcfSSatish Balay 
3759566063dSJacob Faibussowitsch     PetscCall(PetscOptionsString("-tao_view_residual","view least-squares residual vector after each evaluation","TaoSetMonitor","stdout",monfilename,sizeof(monfilename),&flg));
376a7e14dcfSSatish Balay     if (flg) {
3779566063dSJacob Faibussowitsch       PetscCall(PetscViewerASCIIOpen(comm,monfilename,&monviewer));
3789566063dSJacob Faibussowitsch       PetscCall(TaoSetMonitor(tao,TaoResidualMonitor,monviewer,(PetscErrorCode (*)(void**))PetscViewerDestroy));
379a7e14dcfSSatish Balay     }
380a7e14dcfSSatish Balay 
3819566063dSJacob Faibussowitsch     PetscCall(PetscOptionsString("-tao_monitor","Use the default convergence monitor","TaoSetMonitor","stdout",monfilename,sizeof(monfilename),&flg));
382a7e14dcfSSatish Balay     if (flg) {
3839566063dSJacob Faibussowitsch       PetscCall(PetscViewerASCIIOpen(comm,monfilename,&monviewer));
3849566063dSJacob Faibussowitsch       PetscCall(TaoSetMonitor(tao,TaoMonitorDefault,monviewer,(PetscErrorCode (*)(void**))PetscViewerDestroy));
385a7e14dcfSSatish Balay     }
386a7e14dcfSSatish Balay 
3879566063dSJacob Faibussowitsch     PetscCall(PetscOptionsString("-tao_gmonitor","Use the convergence monitor with extra globalization info","TaoSetMonitor","stdout",monfilename,sizeof(monfilename),&flg));
3888d5ead36SAlp Dener     if (flg) {
3899566063dSJacob Faibussowitsch       PetscCall(PetscViewerASCIIOpen(comm,monfilename,&monviewer));
3909566063dSJacob Faibussowitsch       PetscCall(TaoSetMonitor(tao,TaoDefaultGMonitor,monviewer,(PetscErrorCode (*)(void**))PetscViewerDestroy));
3918d5ead36SAlp Dener     }
3928d5ead36SAlp Dener 
3939566063dSJacob Faibussowitsch     PetscCall(PetscOptionsString("-tao_smonitor","Use the short convergence monitor","TaoSetMonitor","stdout",monfilename,sizeof(monfilename),&flg));
394a7e14dcfSSatish Balay     if (flg) {
3959566063dSJacob Faibussowitsch       PetscCall(PetscViewerASCIIOpen(comm,monfilename,&monviewer));
3969566063dSJacob Faibussowitsch       PetscCall(TaoSetMonitor(tao,TaoDefaultSMonitor,monviewer,(PetscErrorCode (*)(void**))PetscViewerDestroy));
397a7e14dcfSSatish Balay     }
398a7e14dcfSSatish Balay 
3999566063dSJacob Faibussowitsch     PetscCall(PetscOptionsString("-tao_cmonitor","Use the default convergence monitor with constraint norm","TaoSetMonitor","stdout",monfilename,sizeof(monfilename),&flg));
400a7e14dcfSSatish Balay     if (flg) {
4019566063dSJacob Faibussowitsch       PetscCall(PetscViewerASCIIOpen(comm,monfilename,&monviewer));
4029566063dSJacob Faibussowitsch       PetscCall(TaoSetMonitor(tao,TaoDefaultCMonitor,monviewer,(PetscErrorCode (*)(void**))PetscViewerDestroy));
403a7e14dcfSSatish Balay     }
404a7e14dcfSSatish Balay 
4058afaa268SBarry Smith     flg = PETSC_FALSE;
4069566063dSJacob Faibussowitsch     PetscCall(PetscOptionsBool("-tao_cancelmonitors","cancel all monitors and call any registered destroy routines","TaoCancelMonitors",flg,&flg,NULL));
4079566063dSJacob Faibussowitsch     if (flg) PetscCall(TaoCancelMonitors(tao));
408a7e14dcfSSatish Balay 
4098afaa268SBarry Smith     flg = PETSC_FALSE;
4109566063dSJacob Faibussowitsch     PetscCall(PetscOptionsBool("-tao_draw_solution","Plot solution vector at each iteration","TaoSetMonitor",flg,&flg,NULL));
411a7e14dcfSSatish Balay     if (flg) {
412e882e171SHong Zhang       TaoMonitorDrawCtx drawctx;
413e882e171SHong Zhang       PetscInt          howoften = 1;
4149566063dSJacob Faibussowitsch       PetscCall(TaoMonitorDrawCtxCreate(PetscObjectComm((PetscObject)tao),NULL,NULL,PETSC_DECIDE,PETSC_DECIDE,300,300,howoften,&drawctx));
4159566063dSJacob Faibussowitsch       PetscCall(TaoSetMonitor(tao,TaoDrawSolutionMonitor,drawctx,(PetscErrorCode (*)(void**))TaoMonitorDrawCtxDestroy));
416a7e14dcfSSatish Balay     }
417a7e14dcfSSatish Balay 
4188afaa268SBarry Smith     flg = PETSC_FALSE;
4199566063dSJacob Faibussowitsch     PetscCall(PetscOptionsBool("-tao_draw_step","plots step direction at each iteration","TaoSetMonitor",flg,&flg,NULL));
420a7e14dcfSSatish Balay     if (flg) {
4219566063dSJacob Faibussowitsch       PetscCall(TaoSetMonitor(tao,TaoDrawStepMonitor,NULL,NULL));
422a7e14dcfSSatish Balay     }
423a7e14dcfSSatish Balay 
4248afaa268SBarry Smith     flg = PETSC_FALSE;
4259566063dSJacob Faibussowitsch     PetscCall(PetscOptionsBool("-tao_draw_gradient","plots gradient at each iteration","TaoSetMonitor",flg,&flg,NULL));
426a7e14dcfSSatish Balay     if (flg) {
427e882e171SHong Zhang       TaoMonitorDrawCtx drawctx;
428e882e171SHong Zhang       PetscInt          howoften = 1;
4299566063dSJacob Faibussowitsch       PetscCall(TaoMonitorDrawCtxCreate(PetscObjectComm((PetscObject)tao),NULL,NULL,PETSC_DECIDE,PETSC_DECIDE,300,300,howoften,&drawctx));
4309566063dSJacob Faibussowitsch       PetscCall(TaoSetMonitor(tao,TaoDrawGradientMonitor,drawctx,(PetscErrorCode (*)(void**))TaoMonitorDrawCtxDestroy));
431a7e14dcfSSatish Balay     }
4328afaa268SBarry Smith     flg = PETSC_FALSE;
4339566063dSJacob Faibussowitsch     PetscCall(PetscOptionsBool("-tao_fd_gradient","compute gradient using finite differences","TaoDefaultComputeGradient",flg,&flg,NULL));
434a7e14dcfSSatish Balay     if (flg) {
4359566063dSJacob Faibussowitsch       PetscCall(TaoSetGradient(tao,NULL,TaoDefaultComputeGradient,NULL));
436a7e14dcfSSatish Balay     }
437f4c1ad5cSStefano Zampini     flg = PETSC_FALSE;
4389566063dSJacob Faibussowitsch     PetscCall(PetscOptionsBool("-tao_fd_hessian","compute hessian using finite differences","TaoDefaultComputeHessian",flg,&flg,NULL));
439f4c1ad5cSStefano Zampini     if (flg) {
440f4c1ad5cSStefano Zampini       Mat H;
441f4c1ad5cSStefano Zampini 
4429566063dSJacob Faibussowitsch       PetscCall(MatCreate(PetscObjectComm((PetscObject)tao),&H));
4439566063dSJacob Faibussowitsch       PetscCall(MatSetType(H,MATAIJ));
4449566063dSJacob Faibussowitsch       PetscCall(TaoSetHessian(tao,H,H,TaoDefaultComputeHessian,NULL));
4459566063dSJacob Faibussowitsch       PetscCall(MatDestroy(&H));
446f4c1ad5cSStefano Zampini     }
447f4c1ad5cSStefano Zampini     flg = PETSC_FALSE;
4489566063dSJacob Faibussowitsch     PetscCall(PetscOptionsBool("-tao_mf_hessian","compute matrix-free hessian using finite differences","TaoDefaultComputeHessianMFFD",flg,&flg,NULL));
449f4c1ad5cSStefano Zampini     if (flg) {
450f4c1ad5cSStefano Zampini       Mat H;
451f4c1ad5cSStefano Zampini 
4529566063dSJacob Faibussowitsch       PetscCall(MatCreate(PetscObjectComm((PetscObject)tao),&H));
4539566063dSJacob Faibussowitsch       PetscCall(TaoSetHessian(tao,H,H,TaoDefaultComputeHessianMFFD,NULL));
4549566063dSJacob Faibussowitsch       PetscCall(MatDestroy(&H));
455f4c1ad5cSStefano Zampini     }
456414d97d3SAlp Dener     flg = PETSC_FALSE;
4579566063dSJacob Faibussowitsch     PetscCall(PetscOptionsBool("-tao_recycle_history","enable recycling/re-using information from the previous TaoSolve() call for some algorithms","TaoSetRecycleHistory",flg,&flg,NULL));
458414d97d3SAlp Dener     if (flg) {
4599566063dSJacob Faibussowitsch       PetscCall(TaoSetRecycleHistory(tao,PETSC_TRUE));
460414d97d3SAlp Dener     }
4619566063dSJacob Faibussowitsch     PetscCall(PetscOptionsEnum("-tao_subset_type","subset type","",TaoSubSetTypes,(PetscEnum)tao->subset_type,(PetscEnum*)&tao->subset_type,NULL));
462a7e14dcfSSatish Balay 
463a82e8c82SStefano Zampini     if (tao->linesearch) {
4649566063dSJacob Faibussowitsch       PetscCall(TaoLineSearchSetFromOptions(tao->linesearch));
465a82e8c82SStefano Zampini     }
466a82e8c82SStefano Zampini 
467a7e14dcfSSatish Balay     if (tao->ops->setfromoptions) {
4689566063dSJacob Faibussowitsch       PetscCall((*tao->ops->setfromoptions)(PetscOptionsObject,tao));
469a7e14dcfSSatish Balay     }
470a7e14dcfSSatish Balay   }
4719566063dSJacob Faibussowitsch   ierr = PetscOptionsEnd();PetscCall(ierr);
472a7e14dcfSSatish Balay   PetscFunctionReturn(0);
473a7e14dcfSSatish Balay }
474a7e14dcfSSatish Balay 
475a7e14dcfSSatish Balay /*@C
476fe2efc57SMark    TaoViewFromOptions - View from Options
477fe2efc57SMark 
478fe2efc57SMark    Collective on Tao
479fe2efc57SMark 
480fe2efc57SMark    Input Parameters:
481fe2efc57SMark +  A - the  Tao context
482736c3998SJose E. Roman .  obj - Optional object
483736c3998SJose E. Roman -  name - command line option
484fe2efc57SMark 
485fe2efc57SMark    Level: intermediate
486fe2efc57SMark .seealso:  Tao, TaoView, PetscObjectViewFromOptions(), TaoCreate()
487fe2efc57SMark @*/
488fe2efc57SMark PetscErrorCode  TaoViewFromOptions(Tao A,PetscObject obj,const char name[])
489fe2efc57SMark {
490fe2efc57SMark   PetscFunctionBegin;
491fe2efc57SMark   PetscValidHeaderSpecific(A,TAO_CLASSID,1);
4929566063dSJacob Faibussowitsch   PetscCall(PetscObjectViewFromOptions((PetscObject)A,obj,name));
493fe2efc57SMark   PetscFunctionReturn(0);
494fe2efc57SMark }
495fe2efc57SMark 
496fe2efc57SMark /*@C
497441846f8SBarry Smith   TaoView - Prints information about the Tao
498a7e14dcfSSatish Balay 
499441846f8SBarry Smith   Collective on Tao
500a7e14dcfSSatish Balay 
501a7e14dcfSSatish Balay   InputParameters:
502441846f8SBarry Smith + tao - the Tao context
503a7e14dcfSSatish Balay - viewer - visualization context
504a7e14dcfSSatish Balay 
505a7e14dcfSSatish Balay   Options Database Key:
506a7e14dcfSSatish Balay . -tao_view - Calls TaoView() at the end of TaoSolve()
507a7e14dcfSSatish Balay 
508a7e14dcfSSatish Balay   Notes:
509a7e14dcfSSatish Balay   The available visualization contexts include
510a7e14dcfSSatish Balay +     PETSC_VIEWER_STDOUT_SELF - standard output (default)
511a7e14dcfSSatish Balay -     PETSC_VIEWER_STDOUT_WORLD - synchronized standard
512a7e14dcfSSatish Balay          output where only the first processor opens
513a7e14dcfSSatish Balay          the file.  All other processors send their
514a7e14dcfSSatish Balay          data to the first processor to print.
515a7e14dcfSSatish Balay 
516a7e14dcfSSatish Balay   Level: beginner
517a7e14dcfSSatish Balay 
518a7e14dcfSSatish Balay .seealso: PetscViewerASCIIOpen()
519a7e14dcfSSatish Balay @*/
520441846f8SBarry Smith PetscErrorCode TaoView(Tao tao, PetscViewer viewer)
521a7e14dcfSSatish Balay {
522a7e14dcfSSatish Balay   PetscBool           isascii,isstring;
523b625d6c7SJed Brown   TaoType             type;
52447a47007SBarry Smith 
525a7e14dcfSSatish Balay   PetscFunctionBegin;
526441846f8SBarry Smith   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
527a7e14dcfSSatish Balay   if (!viewer) {
5289566063dSJacob Faibussowitsch     PetscCall(PetscViewerASCIIGetStdout(((PetscObject)tao)->comm,&viewer));
529a7e14dcfSSatish Balay   }
530a7e14dcfSSatish Balay   PetscValidHeaderSpecific(viewer,PETSC_VIEWER_CLASSID,2);
531a7e14dcfSSatish Balay   PetscCheckSameComm(tao,1,viewer,2);
532a7e14dcfSSatish Balay 
5339566063dSJacob Faibussowitsch   PetscCall(PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERASCII,&isascii));
5349566063dSJacob Faibussowitsch   PetscCall(PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERSTRING,&isstring));
535a7e14dcfSSatish Balay   if (isascii) {
5369566063dSJacob Faibussowitsch     PetscCall(PetscObjectPrintClassNamePrefixType((PetscObject)tao,viewer));
537a7e14dcfSSatish Balay 
538a7e14dcfSSatish Balay     if (tao->ops->view) {
5399566063dSJacob Faibussowitsch       PetscCall(PetscViewerASCIIPushTab(viewer));
5409566063dSJacob Faibussowitsch       PetscCall((*tao->ops->view)(tao,viewer));
5419566063dSJacob Faibussowitsch       PetscCall(PetscViewerASCIIPopTab(viewer));
542a7e14dcfSSatish Balay     }
5438d3f3ddaSAlp Dener     if (tao->linesearch) {
5449566063dSJacob Faibussowitsch       PetscCall(PetscViewerASCIIPushTab(viewer));
5459566063dSJacob Faibussowitsch       PetscCall(TaoLineSearchView(tao->linesearch,viewer));
5469566063dSJacob Faibussowitsch       PetscCall(PetscViewerASCIIPopTab(viewer));
5478d3f3ddaSAlp Dener     }
548a7e14dcfSSatish Balay     if (tao->ksp) {
5499566063dSJacob Faibussowitsch       PetscCall(PetscViewerASCIIPushTab(viewer));
5509566063dSJacob Faibussowitsch       PetscCall(KSPView(tao->ksp,viewer));
5519566063dSJacob Faibussowitsch       PetscCall(PetscViewerASCIIPrintf(viewer,"total KSP iterations: %D\n",tao->ksp_tot_its));
5529566063dSJacob Faibussowitsch       PetscCall(PetscViewerASCIIPopTab(viewer));
553a7e14dcfSSatish Balay     }
55419b5c101SAlp Dener 
5559566063dSJacob Faibussowitsch     PetscCall(PetscViewerASCIIPushTab(viewer));
55619b5c101SAlp Dener 
557a7e14dcfSSatish Balay     if (tao->XL || tao->XU) {
5589566063dSJacob Faibussowitsch       PetscCall(PetscViewerASCIIPrintf(viewer,"Active Set subset type: %s\n",TaoSubSetTypes[tao->subset_type]));
559a7e14dcfSSatish Balay     }
560a7e14dcfSSatish Balay 
5619566063dSJacob Faibussowitsch     PetscCall(PetscViewerASCIIPrintf(viewer,"convergence tolerances: gatol=%g,",(double)tao->gatol));
5629566063dSJacob Faibussowitsch     PetscCall(PetscViewerASCIIPrintf(viewer," steptol=%g,",(double)tao->steptol));
5639566063dSJacob Faibussowitsch     PetscCall(PetscViewerASCIIPrintf(viewer," gttol=%g\n",(double)tao->gttol));
5649566063dSJacob Faibussowitsch     PetscCall(PetscViewerASCIIPrintf(viewer,"Residual in Function/Gradient:=%g\n",(double)tao->residual));
565a7e14dcfSSatish Balay 
5666246e8cdSAlp Dener     if (tao->constrained) {
5679566063dSJacob Faibussowitsch       PetscCall(PetscViewerASCIIPrintf(viewer,"convergence tolerances:"));
5689566063dSJacob Faibussowitsch       PetscCall(PetscViewerASCIIPrintf(viewer," catol=%g,",(double)tao->catol));
5699566063dSJacob Faibussowitsch       PetscCall(PetscViewerASCIIPrintf(viewer," crtol=%g\n",(double)tao->crtol));
5709566063dSJacob Faibussowitsch       PetscCall(PetscViewerASCIIPrintf(viewer,"Residual in Constraints:=%g\n",(double)tao->cnorm));
571a7e14dcfSSatish Balay     }
572a7e14dcfSSatish Balay 
573a7e14dcfSSatish Balay     if (tao->trust < tao->steptol) {
5749566063dSJacob Faibussowitsch       PetscCall(PetscViewerASCIIPrintf(viewer,"convergence tolerances: steptol=%g\n",(double)tao->steptol));
5759566063dSJacob Faibussowitsch       PetscCall(PetscViewerASCIIPrintf(viewer,"Final trust region radius:=%g\n",(double)tao->trust));
576a7e14dcfSSatish Balay     }
577a7e14dcfSSatish Balay 
578a7e14dcfSSatish Balay     if (tao->fmin>-1.e25) {
5799566063dSJacob Faibussowitsch       PetscCall(PetscViewerASCIIPrintf(viewer,"convergence tolerances: function minimum=%g\n",(double)tao->fmin));
580a7e14dcfSSatish Balay     }
5819566063dSJacob Faibussowitsch     PetscCall(PetscViewerASCIIPrintf(viewer,"Objective value=%g\n",(double)tao->fc));
582a7e14dcfSSatish Balay 
5839566063dSJacob Faibussowitsch     PetscCall(PetscViewerASCIIPrintf(viewer,"total number of iterations=%D,          ",tao->niter));
5849566063dSJacob Faibussowitsch     PetscCall(PetscViewerASCIIPrintf(viewer,"              (max: %D)\n",tao->max_it));
585a7e14dcfSSatish Balay 
586a7e14dcfSSatish Balay     if (tao->nfuncs>0) {
5879566063dSJacob Faibussowitsch       PetscCall(PetscViewerASCIIPrintf(viewer,"total number of function evaluations=%D,",tao->nfuncs));
5889566063dSJacob Faibussowitsch       PetscCall(PetscViewerASCIIPrintf(viewer,"                max: %D\n",tao->max_funcs));
589a7e14dcfSSatish Balay     }
590a7e14dcfSSatish Balay     if (tao->ngrads>0) {
5919566063dSJacob Faibussowitsch       PetscCall(PetscViewerASCIIPrintf(viewer,"total number of gradient evaluations=%D,",tao->ngrads));
5929566063dSJacob Faibussowitsch       PetscCall(PetscViewerASCIIPrintf(viewer,"                max: %D\n",tao->max_funcs));
593a7e14dcfSSatish Balay     }
594a7e14dcfSSatish Balay     if (tao->nfuncgrads>0) {
5959566063dSJacob Faibussowitsch       PetscCall(PetscViewerASCIIPrintf(viewer,"total number of function/gradient evaluations=%D,",tao->nfuncgrads));
5969566063dSJacob Faibussowitsch       PetscCall(PetscViewerASCIIPrintf(viewer,"    (max: %D)\n",tao->max_funcs));
597a7e14dcfSSatish Balay     }
598a7e14dcfSSatish Balay     if (tao->nhess>0) {
5999566063dSJacob Faibussowitsch       PetscCall(PetscViewerASCIIPrintf(viewer,"total number of Hessian evaluations=%D\n",tao->nhess));
600a7e14dcfSSatish Balay     }
601a7e14dcfSSatish Balay     if (tao->nconstraints>0) {
6029566063dSJacob Faibussowitsch       PetscCall(PetscViewerASCIIPrintf(viewer,"total number of constraint function evaluations=%D\n",tao->nconstraints));
603a7e14dcfSSatish Balay     }
604a7e14dcfSSatish Balay     if (tao->njac>0) {
6059566063dSJacob Faibussowitsch       PetscCall(PetscViewerASCIIPrintf(viewer,"total number of Jacobian evaluations=%D\n",tao->njac));
606a7e14dcfSSatish Balay     }
607a7e14dcfSSatish Balay 
608a7e14dcfSSatish Balay     if (tao->reason>0) {
6099566063dSJacob Faibussowitsch       PetscCall(PetscViewerASCIIPrintf(viewer,    "Solution converged: "));
610a7e14dcfSSatish Balay       switch (tao->reason) {
611a7e14dcfSSatish Balay       case TAO_CONVERGED_GATOL:
6129566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer," ||g(X)|| <= gatol\n"));
613a7e14dcfSSatish Balay         break;
614a7e14dcfSSatish Balay       case TAO_CONVERGED_GRTOL:
6159566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer," ||g(X)||/|f(X)| <= grtol\n"));
616a7e14dcfSSatish Balay         break;
617a7e14dcfSSatish Balay       case TAO_CONVERGED_GTTOL:
6189566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer," ||g(X)||/||g(X0)|| <= gttol\n"));
619a7e14dcfSSatish Balay         break;
620a7e14dcfSSatish Balay       case TAO_CONVERGED_STEPTOL:
6219566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer," Steptol -- step size small\n"));
622a7e14dcfSSatish Balay         break;
623a7e14dcfSSatish Balay       case TAO_CONVERGED_MINF:
6249566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer," Minf --  f < fmin\n"));
625a7e14dcfSSatish Balay         break;
626a7e14dcfSSatish Balay       case TAO_CONVERGED_USER:
6279566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer," User Terminated\n"));
628a7e14dcfSSatish Balay         break;
629a7e14dcfSSatish Balay       default:
6309566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer,"\n"));
631a7e14dcfSSatish Balay         break;
632a7e14dcfSSatish Balay       }
633a7e14dcfSSatish Balay     } else {
6349566063dSJacob Faibussowitsch       PetscCall(PetscViewerASCIIPrintf(viewer,"Solver terminated: %d",tao->reason));
635a7e14dcfSSatish Balay       switch (tao->reason) {
636a7e14dcfSSatish Balay       case TAO_DIVERGED_MAXITS:
6379566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer," Maximum Iterations\n"));
638a7e14dcfSSatish Balay         break;
639a7e14dcfSSatish Balay       case TAO_DIVERGED_NAN:
6409566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer," NAN or Inf encountered\n"));
641a7e14dcfSSatish Balay         break;
642a7e14dcfSSatish Balay       case TAO_DIVERGED_MAXFCN:
6439566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer," Maximum Function Evaluations\n"));
644a7e14dcfSSatish Balay         break;
645a7e14dcfSSatish Balay       case TAO_DIVERGED_LS_FAILURE:
6469566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer," Line Search Failure\n"));
647a7e14dcfSSatish Balay         break;
648a7e14dcfSSatish Balay       case TAO_DIVERGED_TR_REDUCTION:
6499566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer," Trust Region too small\n"));
650a7e14dcfSSatish Balay         break;
651a7e14dcfSSatish Balay       case TAO_DIVERGED_USER:
6529566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer," User Terminated\n"));
653a7e14dcfSSatish Balay         break;
654a7e14dcfSSatish Balay       default:
6559566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer,"\n"));
656a7e14dcfSSatish Balay         break;
657a7e14dcfSSatish Balay       }
658a7e14dcfSSatish Balay     }
6599566063dSJacob Faibussowitsch     PetscCall(PetscViewerASCIIPopTab(viewer));
660a7e14dcfSSatish Balay   } else if (isstring) {
6619566063dSJacob Faibussowitsch     PetscCall(TaoGetType(tao,&type));
6629566063dSJacob Faibussowitsch     PetscCall(PetscViewerStringSPrintf(viewer," %-3.3s",type));
663a7e14dcfSSatish Balay   }
664a7e14dcfSSatish Balay   PetscFunctionReturn(0);
665a7e14dcfSSatish Balay }
666a7e14dcfSSatish Balay 
667a7e14dcfSSatish Balay /*@
668414d97d3SAlp Dener   TaoSetRecycleHistory - Sets the boolean flag to enable/disable re-using
669414d97d3SAlp Dener   iterate information from the previous TaoSolve(). This feature is disabled by
670414d97d3SAlp Dener   default.
671414d97d3SAlp Dener 
672414d97d3SAlp Dener   For conjugate gradient methods (BNCG), this re-uses the latest search direction
673414d97d3SAlp Dener   from the previous TaoSolve() call when computing the first search direction in a
674414d97d3SAlp Dener   new solution. By default, CG methods set the first search direction to the
675414d97d3SAlp Dener   negative gradient.
676414d97d3SAlp Dener 
677414d97d3SAlp Dener   For quasi-Newton family of methods (BQNLS, BQNKLS, BQNKTR, BQNKTL), this re-uses
678414d97d3SAlp Dener   the accumulated quasi-Newton Hessian approximation from the previous TaoSolve()
679414d97d3SAlp Dener   call. By default, QN family of methods reset the initial Hessian approximation to
680414d97d3SAlp Dener   the identity matrix.
681414d97d3SAlp Dener 
682414d97d3SAlp Dener   For any other algorithm, this setting has no effect.
683414d97d3SAlp Dener 
684414d97d3SAlp Dener   Logically collective on Tao
685414d97d3SAlp Dener 
686414d97d3SAlp Dener   Input Parameters:
687414d97d3SAlp Dener + tao - the Tao context
688414d97d3SAlp Dener - recycle - boolean flag
689414d97d3SAlp Dener 
690414d97d3SAlp Dener   Options Database Keys:
691147403d9SBarry Smith . -tao_recycle_history <true,false> - reuse the history
692414d97d3SAlp Dener 
693414d97d3SAlp Dener   Level: intermediate
694414d97d3SAlp Dener 
695414d97d3SAlp Dener .seealso: TaoSetRecycleHistory(), TAOBNCG, TAOBQNLS, TAOBQNKLS, TAOBQNKTR, TAOBQNKTL
696414d97d3SAlp Dener 
697414d97d3SAlp Dener @*/
698414d97d3SAlp Dener PetscErrorCode TaoSetRecycleHistory(Tao tao, PetscBool recycle)
699414d97d3SAlp Dener {
700414d97d3SAlp Dener   PetscFunctionBegin;
701414d97d3SAlp Dener   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
70294511df7SStefano Zampini   PetscValidLogicalCollectiveBool(tao,recycle,2);
703414d97d3SAlp Dener   tao->recycle = recycle;
704414d97d3SAlp Dener   PetscFunctionReturn(0);
705414d97d3SAlp Dener }
706414d97d3SAlp Dener 
707414d97d3SAlp Dener /*@
708414d97d3SAlp Dener   TaoGetRecycleHistory - Retrieve the boolean flag for re-using iterate information
709414d97d3SAlp Dener   from the previous TaoSolve(). This feature is disabled by default.
710414d97d3SAlp Dener 
711414d97d3SAlp Dener   Logically collective on Tao
712414d97d3SAlp Dener 
713414d97d3SAlp Dener   Input Parameters:
714147403d9SBarry Smith . tao - the Tao context
715414d97d3SAlp Dener 
716414d97d3SAlp Dener   Output Parameters:
717147403d9SBarry Smith . recycle - boolean flag
718414d97d3SAlp Dener 
719414d97d3SAlp Dener   Level: intermediate
720414d97d3SAlp Dener 
721414d97d3SAlp Dener .seealso: TaoGetRecycleHistory(), TAOBNCG, TAOBQNLS, TAOBQNKLS, TAOBQNKTR, TAOBQNKTL
722414d97d3SAlp Dener 
723414d97d3SAlp Dener @*/
724414d97d3SAlp Dener PetscErrorCode TaoGetRecycleHistory(Tao tao, PetscBool *recycle)
725414d97d3SAlp Dener {
726414d97d3SAlp Dener   PetscFunctionBegin;
727414d97d3SAlp Dener   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
72894511df7SStefano Zampini   PetscValidBoolPointer(recycle,2);
729414d97d3SAlp Dener   *recycle = tao->recycle;
730414d97d3SAlp Dener   PetscFunctionReturn(0);
731414d97d3SAlp Dener }
732414d97d3SAlp Dener 
733414d97d3SAlp Dener /*@
734a7e14dcfSSatish Balay   TaoSetTolerances - Sets parameters used in TAO convergence tests
735a7e14dcfSSatish Balay 
736441846f8SBarry Smith   Logically collective on Tao
737a7e14dcfSSatish Balay 
738a7e14dcfSSatish Balay   Input Parameters:
739441846f8SBarry Smith + tao - the Tao context
740a7e14dcfSSatish Balay . gatol - stop if norm of gradient is less than this
741a7e14dcfSSatish Balay . grtol - stop if relative norm of gradient is less than this
742a7e14dcfSSatish Balay - gttol - stop if norm of gradient is reduced by this factor
743a7e14dcfSSatish Balay 
744a7e14dcfSSatish Balay   Options Database Keys:
745e52336cbSBarry Smith + -tao_gatol <gatol> - Sets gatol
746a7e14dcfSSatish Balay . -tao_grtol <grtol> - Sets grtol
747a7e14dcfSSatish Balay - -tao_gttol <gttol> - Sets gttol
748a7e14dcfSSatish Balay 
749a7e14dcfSSatish Balay   Stopping Criteria:
750a7e14dcfSSatish Balay $ ||g(X)||                            <= gatol
751a7e14dcfSSatish Balay $ ||g(X)|| / |f(X)|                   <= grtol
752a7e14dcfSSatish Balay $ ||g(X)|| / ||g(X0)||                <= gttol
753a7e14dcfSSatish Balay 
754a7e14dcfSSatish Balay   Notes:
755a7e14dcfSSatish Balay   Use PETSC_DEFAULT to leave one or more tolerances unchanged.
756a7e14dcfSSatish Balay 
757a7e14dcfSSatish Balay   Level: beginner
758a7e14dcfSSatish Balay 
759a7e14dcfSSatish Balay .seealso: TaoGetTolerances()
760a7e14dcfSSatish Balay 
761a7e14dcfSSatish Balay @*/
762e52336cbSBarry Smith PetscErrorCode TaoSetTolerances(Tao tao, PetscReal gatol, PetscReal grtol, PetscReal gttol)
763a7e14dcfSSatish Balay {
764a7e14dcfSSatish Balay   PetscFunctionBegin;
765441846f8SBarry Smith   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
76694511df7SStefano Zampini   PetscValidLogicalCollectiveReal(tao,gatol,2);
76794511df7SStefano Zampini   PetscValidLogicalCollectiveReal(tao,grtol,3);
76894511df7SStefano Zampini   PetscValidLogicalCollectiveReal(tao,gttol,4);
769a7e14dcfSSatish Balay 
770a7e14dcfSSatish Balay   if (gatol != PETSC_DEFAULT) {
771a7e14dcfSSatish Balay     if (gatol<0) {
7729566063dSJacob Faibussowitsch       PetscCall(PetscInfo(tao,"Tried to set negative gatol -- ignored.\n"));
773a7e14dcfSSatish Balay     } else {
774a7e14dcfSSatish Balay       tao->gatol = PetscMax(0,gatol);
7756552cf8aSJason Sarich       tao->gatol_changed = PETSC_TRUE;
776a7e14dcfSSatish Balay     }
777a7e14dcfSSatish Balay   }
778a7e14dcfSSatish Balay 
779a7e14dcfSSatish Balay   if (grtol != PETSC_DEFAULT) {
780a7e14dcfSSatish Balay     if (grtol<0) {
7819566063dSJacob Faibussowitsch       PetscCall(PetscInfo(tao,"Tried to set negative grtol -- ignored.\n"));
782a7e14dcfSSatish Balay     } else {
783a7e14dcfSSatish Balay       tao->grtol = PetscMax(0,grtol);
7846552cf8aSJason Sarich       tao->grtol_changed = PETSC_TRUE;
785a7e14dcfSSatish Balay     }
786a7e14dcfSSatish Balay   }
787a7e14dcfSSatish Balay 
788a7e14dcfSSatish Balay   if (gttol != PETSC_DEFAULT) {
789a7e14dcfSSatish Balay     if (gttol<0) {
7909566063dSJacob Faibussowitsch       PetscCall(PetscInfo(tao,"Tried to set negative gttol -- ignored.\n"));
791a7e14dcfSSatish Balay     } else {
792a7e14dcfSSatish Balay       tao->gttol = PetscMax(0,gttol);
7936552cf8aSJason Sarich       tao->gttol_changed = PETSC_TRUE;
794a7e14dcfSSatish Balay     }
795a7e14dcfSSatish Balay   }
796a7e14dcfSSatish Balay   PetscFunctionReturn(0);
797a7e14dcfSSatish Balay }
798a7e14dcfSSatish Balay 
799a7e14dcfSSatish Balay /*@
80058417fe7SBarry Smith   TaoSetConstraintTolerances - Sets constraint tolerance parameters used in TAO  convergence tests
801a7e14dcfSSatish Balay 
802441846f8SBarry Smith   Logically collective on Tao
803a7e14dcfSSatish Balay 
804a7e14dcfSSatish Balay   Input Parameters:
805441846f8SBarry Smith + tao - the Tao context
806e52336cbSBarry Smith . catol - absolute constraint tolerance, constraint norm must be less than catol for used for gatol convergence criteria
807*6aad120cSJose E. Roman - crtol - relative constraint tolerance, constraint norm must be less than crtol for used for gatol, gttol convergence criteria
808a7e14dcfSSatish Balay 
809a7e14dcfSSatish Balay   Options Database Keys:
810a7e14dcfSSatish Balay + -tao_catol <catol> - Sets catol
811a7e14dcfSSatish Balay - -tao_crtol <crtol> - Sets crtol
812a7e14dcfSSatish Balay 
8136552cf8aSJason Sarich   Notes:
8146552cf8aSJason Sarich   Use PETSC_DEFAULT to leave any tolerance unchanged.
8156552cf8aSJason Sarich 
816a7e14dcfSSatish Balay   Level: intermediate
817a7e14dcfSSatish Balay 
81858417fe7SBarry Smith .seealso: TaoGetTolerances(), TaoGetConstraintTolerances(), TaoSetTolerances()
819a7e14dcfSSatish Balay 
820a7e14dcfSSatish Balay @*/
821441846f8SBarry Smith PetscErrorCode TaoSetConstraintTolerances(Tao tao, PetscReal catol, PetscReal crtol)
822a7e14dcfSSatish Balay {
823a7e14dcfSSatish Balay   PetscFunctionBegin;
824441846f8SBarry Smith   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
82594511df7SStefano Zampini   PetscValidLogicalCollectiveReal(tao,catol,2);
82694511df7SStefano Zampini   PetscValidLogicalCollectiveReal(tao,crtol,3);
827a7e14dcfSSatish Balay 
828a7e14dcfSSatish Balay   if (catol != PETSC_DEFAULT) {
829a7e14dcfSSatish Balay     if (catol<0) {
8309566063dSJacob Faibussowitsch       PetscCall(PetscInfo(tao,"Tried to set negative catol -- ignored.\n"));
831a7e14dcfSSatish Balay     } else {
832a7e14dcfSSatish Balay       tao->catol = PetscMax(0,catol);
8336552cf8aSJason Sarich       tao->catol_changed = PETSC_TRUE;
834a7e14dcfSSatish Balay     }
835a7e14dcfSSatish Balay   }
836a7e14dcfSSatish Balay 
837a7e14dcfSSatish Balay   if (crtol != PETSC_DEFAULT) {
838a7e14dcfSSatish Balay     if (crtol<0) {
8399566063dSJacob Faibussowitsch       PetscCall(PetscInfo(tao,"Tried to set negative crtol -- ignored.\n"));
840a7e14dcfSSatish Balay     } else {
841a7e14dcfSSatish Balay       tao->crtol = PetscMax(0,crtol);
8426552cf8aSJason Sarich       tao->crtol_changed = PETSC_TRUE;
843a7e14dcfSSatish Balay     }
844a7e14dcfSSatish Balay   }
845a7e14dcfSSatish Balay   PetscFunctionReturn(0);
846a7e14dcfSSatish Balay }
847a7e14dcfSSatish Balay 
84858417fe7SBarry Smith /*@
84958417fe7SBarry Smith   TaoGetConstraintTolerances - Gets constraint tolerance parameters used in TAO  convergence tests
85058417fe7SBarry Smith 
85158417fe7SBarry Smith   Not ollective
85258417fe7SBarry Smith 
85358417fe7SBarry Smith   Input Parameter:
85458417fe7SBarry Smith . tao - the Tao context
85558417fe7SBarry Smith 
856d8d19677SJose E. Roman   Output Parameters:
857e52336cbSBarry Smith + catol - absolute constraint tolerance, constraint norm must be less than catol for used for gatol convergence criteria
858*6aad120cSJose E. Roman - crtol - relative constraint tolerance, constraint norm must be less than crtol for used for gatol, gttol convergence criteria
85958417fe7SBarry Smith 
86058417fe7SBarry Smith   Level: intermediate
86158417fe7SBarry Smith 
86258417fe7SBarry Smith .seealso: TaoGetTolerances(), TaoSetTolerances(), TaoSetConstraintTolerances()
86358417fe7SBarry Smith 
86458417fe7SBarry Smith @*/
86558417fe7SBarry Smith PetscErrorCode TaoGetConstraintTolerances(Tao tao, PetscReal *catol, PetscReal *crtol)
86658417fe7SBarry Smith {
86758417fe7SBarry Smith   PetscFunctionBegin;
86858417fe7SBarry Smith   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
86958417fe7SBarry Smith   if (catol) *catol = tao->catol;
87058417fe7SBarry Smith   if (crtol) *crtol = tao->crtol;
87158417fe7SBarry Smith   PetscFunctionReturn(0);
87258417fe7SBarry Smith }
87358417fe7SBarry Smith 
874a7e14dcfSSatish Balay /*@
875a7e14dcfSSatish Balay    TaoSetFunctionLowerBound - Sets a bound on the solution objective value.
876a7e14dcfSSatish Balay    When an approximate solution with an objective value below this number
877a7e14dcfSSatish Balay    has been found, the solver will terminate.
878a7e14dcfSSatish Balay 
879441846f8SBarry Smith    Logically Collective on Tao
880a7e14dcfSSatish Balay 
881a7e14dcfSSatish Balay    Input Parameters:
882441846f8SBarry Smith +  tao - the Tao solver context
883a7e14dcfSSatish Balay -  fmin - the tolerance
884a7e14dcfSSatish Balay 
885a7e14dcfSSatish Balay    Options Database Keys:
886a7e14dcfSSatish Balay .    -tao_fmin <fmin> - sets the minimum function value
887a7e14dcfSSatish Balay 
888a7e14dcfSSatish Balay    Level: intermediate
889a7e14dcfSSatish Balay 
890a7e14dcfSSatish Balay .seealso: TaoSetTolerances()
891a7e14dcfSSatish Balay @*/
892441846f8SBarry Smith PetscErrorCode TaoSetFunctionLowerBound(Tao tao,PetscReal fmin)
893a7e14dcfSSatish Balay {
894a7e14dcfSSatish Balay   PetscFunctionBegin;
895441846f8SBarry Smith   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
89694511df7SStefano Zampini   PetscValidLogicalCollectiveReal(tao,fmin,2);
897a7e14dcfSSatish Balay   tao->fmin = fmin;
8986552cf8aSJason Sarich   tao->fmin_changed = PETSC_TRUE;
899a7e14dcfSSatish Balay   PetscFunctionReturn(0);
900a7e14dcfSSatish Balay }
901a7e14dcfSSatish Balay 
902a7e14dcfSSatish Balay /*@
9038e96d397SJason Sarich    TaoGetFunctionLowerBound - Gets the bound on the solution objective value.
904a7e14dcfSSatish Balay    When an approximate solution with an objective value below this number
905a7e14dcfSSatish Balay    has been found, the solver will terminate.
906a7e14dcfSSatish Balay 
907441846f8SBarry Smith    Not collective on Tao
908a7e14dcfSSatish Balay 
909a7e14dcfSSatish Balay    Input Parameters:
910441846f8SBarry Smith .  tao - the Tao solver context
911a7e14dcfSSatish Balay 
912a7e14dcfSSatish Balay    OutputParameters:
913a7e14dcfSSatish Balay .  fmin - the minimum function value
914a7e14dcfSSatish Balay 
915a7e14dcfSSatish Balay    Level: intermediate
916a7e14dcfSSatish Balay 
917a7e14dcfSSatish Balay .seealso: TaoSetFunctionLowerBound()
918a7e14dcfSSatish Balay @*/
919441846f8SBarry Smith PetscErrorCode TaoGetFunctionLowerBound(Tao tao,PetscReal *fmin)
920a7e14dcfSSatish Balay {
921a7e14dcfSSatish Balay   PetscFunctionBegin;
922441846f8SBarry Smith   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
92394511df7SStefano Zampini   PetscValidRealPointer(fmin,2);
924a7e14dcfSSatish Balay   *fmin = tao->fmin;
925a7e14dcfSSatish Balay   PetscFunctionReturn(0);
926a7e14dcfSSatish Balay }
927a7e14dcfSSatish Balay 
928a7e14dcfSSatish Balay /*@
929a7e14dcfSSatish Balay    TaoSetMaximumFunctionEvaluations - Sets a maximum number of
930a7e14dcfSSatish Balay    function evaluations.
931a7e14dcfSSatish Balay 
932441846f8SBarry Smith    Logically Collective on Tao
933a7e14dcfSSatish Balay 
934a7e14dcfSSatish Balay    Input Parameters:
935441846f8SBarry Smith +  tao - the Tao solver context
936a7e14dcfSSatish Balay -  nfcn - the maximum number of function evaluations (>=0)
937a7e14dcfSSatish Balay 
938a7e14dcfSSatish Balay    Options Database Keys:
939a7e14dcfSSatish Balay .    -tao_max_funcs <nfcn> - sets the maximum number of function evaluations
940a7e14dcfSSatish Balay 
941a7e14dcfSSatish Balay    Level: intermediate
942a7e14dcfSSatish Balay 
943a7e14dcfSSatish Balay .seealso: TaoSetTolerances(), TaoSetMaximumIterations()
944a7e14dcfSSatish Balay @*/
945a7e14dcfSSatish Balay 
946441846f8SBarry Smith PetscErrorCode TaoSetMaximumFunctionEvaluations(Tao tao,PetscInt nfcn)
947a7e14dcfSSatish Balay {
948a7e14dcfSSatish Balay   PetscFunctionBegin;
949441846f8SBarry Smith   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
95094511df7SStefano Zampini   PetscValidLogicalCollectiveInt(tao,nfcn,2);
95194511df7SStefano Zampini   if (nfcn >= 0) { tao->max_funcs = PetscMax(0,nfcn); }
95294511df7SStefano Zampini   else { tao->max_funcs = -1; }
9536552cf8aSJason Sarich   tao->max_funcs_changed = PETSC_TRUE;
954a7e14dcfSSatish Balay   PetscFunctionReturn(0);
955a7e14dcfSSatish Balay }
956a7e14dcfSSatish Balay 
957a7e14dcfSSatish Balay /*@
958a7e14dcfSSatish Balay    TaoGetMaximumFunctionEvaluations - Sets a maximum number of
959a7e14dcfSSatish Balay    function evaluations.
960a7e14dcfSSatish Balay 
961a7e14dcfSSatish Balay    Not Collective
962a7e14dcfSSatish Balay 
963a7e14dcfSSatish Balay    Input Parameters:
964441846f8SBarry Smith .  tao - the Tao solver context
965a7e14dcfSSatish Balay 
966a7e14dcfSSatish Balay    Output Parameters:
967a7e14dcfSSatish Balay .  nfcn - the maximum number of function evaluations
968a7e14dcfSSatish Balay 
969a7e14dcfSSatish Balay    Level: intermediate
970a7e14dcfSSatish Balay 
971a7e14dcfSSatish Balay .seealso: TaoSetMaximumFunctionEvaluations(), TaoGetMaximumIterations()
972a7e14dcfSSatish Balay @*/
973a7e14dcfSSatish Balay 
974441846f8SBarry Smith PetscErrorCode TaoGetMaximumFunctionEvaluations(Tao tao,PetscInt *nfcn)
975a7e14dcfSSatish Balay {
976a7e14dcfSSatish Balay   PetscFunctionBegin;
977441846f8SBarry Smith   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
97894511df7SStefano Zampini   PetscValidIntPointer(nfcn,2);
979a7e14dcfSSatish Balay   *nfcn = tao->max_funcs;
980a7e14dcfSSatish Balay   PetscFunctionReturn(0);
981a7e14dcfSSatish Balay }
982a7e14dcfSSatish Balay 
983770232b9SCe Qin /*@
984770232b9SCe Qin    TaoGetCurrentFunctionEvaluations - Get current number of
985770232b9SCe Qin    function evaluations.
986770232b9SCe Qin 
987770232b9SCe Qin    Not Collective
988770232b9SCe Qin 
989770232b9SCe Qin    Input Parameters:
990770232b9SCe Qin .  tao - the Tao solver context
991770232b9SCe Qin 
992770232b9SCe Qin    Output Parameters:
99394511df7SStefano Zampini .  nfuncs - the current number of function evaluations (maximum between gradient and function evaluations)
994770232b9SCe Qin 
995770232b9SCe Qin    Level: intermediate
996770232b9SCe Qin 
997770232b9SCe Qin .seealso: TaoSetMaximumFunctionEvaluations(), TaoGetMaximumFunctionEvaluations(), TaoGetMaximumIterations()
998770232b9SCe Qin @*/
999770232b9SCe Qin 
1000770232b9SCe Qin PetscErrorCode TaoGetCurrentFunctionEvaluations(Tao tao,PetscInt *nfuncs)
1001770232b9SCe Qin {
1002770232b9SCe Qin   PetscFunctionBegin;
1003770232b9SCe Qin   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
100494511df7SStefano Zampini   PetscValidIntPointer(nfuncs,2);
1005770232b9SCe Qin   *nfuncs = PetscMax(tao->nfuncs,tao->nfuncgrads);
1006770232b9SCe Qin   PetscFunctionReturn(0);
1007770232b9SCe Qin }
1008770232b9SCe Qin 
1009a7e14dcfSSatish Balay /*@
1010a7e14dcfSSatish Balay    TaoSetMaximumIterations - Sets a maximum number of iterates.
1011a7e14dcfSSatish Balay 
1012441846f8SBarry Smith    Logically Collective on Tao
1013a7e14dcfSSatish Balay 
1014a7e14dcfSSatish Balay    Input Parameters:
1015441846f8SBarry Smith +  tao - the Tao solver context
1016a7e14dcfSSatish Balay -  maxits - the maximum number of iterates (>=0)
1017a7e14dcfSSatish Balay 
1018a7e14dcfSSatish Balay    Options Database Keys:
1019a7e14dcfSSatish Balay .    -tao_max_it <its> - sets the maximum number of iterations
1020a7e14dcfSSatish Balay 
1021a7e14dcfSSatish Balay    Level: intermediate
1022a7e14dcfSSatish Balay 
1023a7e14dcfSSatish Balay .seealso: TaoSetTolerances(), TaoSetMaximumFunctionEvaluations()
1024a7e14dcfSSatish Balay @*/
1025441846f8SBarry Smith PetscErrorCode TaoSetMaximumIterations(Tao tao,PetscInt maxits)
1026a7e14dcfSSatish Balay {
1027a7e14dcfSSatish Balay   PetscFunctionBegin;
1028441846f8SBarry Smith   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
102994511df7SStefano Zampini   PetscValidLogicalCollectiveInt(tao,maxits,2);
103047a47007SBarry Smith   tao->max_it = PetscMax(0,maxits);
10316552cf8aSJason Sarich   tao->max_it_changed = PETSC_TRUE;
1032a7e14dcfSSatish Balay   PetscFunctionReturn(0);
1033a7e14dcfSSatish Balay }
1034a7e14dcfSSatish Balay 
1035a7e14dcfSSatish Balay /*@
1036a7e14dcfSSatish Balay    TaoGetMaximumIterations - Sets a maximum number of iterates.
1037a7e14dcfSSatish Balay 
1038a7e14dcfSSatish Balay    Not Collective
1039a7e14dcfSSatish Balay 
1040a7e14dcfSSatish Balay    Input Parameters:
1041441846f8SBarry Smith .  tao - the Tao solver context
1042a7e14dcfSSatish Balay 
1043a7e14dcfSSatish Balay    Output Parameters:
1044a7e14dcfSSatish Balay .  maxits - the maximum number of iterates
1045a7e14dcfSSatish Balay 
1046a7e14dcfSSatish Balay    Level: intermediate
1047a7e14dcfSSatish Balay 
1048a7e14dcfSSatish Balay .seealso: TaoSetMaximumIterations(), TaoGetMaximumFunctionEvaluations()
1049a7e14dcfSSatish Balay @*/
1050441846f8SBarry Smith PetscErrorCode TaoGetMaximumIterations(Tao tao,PetscInt *maxits)
1051a7e14dcfSSatish Balay {
1052a7e14dcfSSatish Balay   PetscFunctionBegin;
1053441846f8SBarry Smith   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
105494511df7SStefano Zampini   PetscValidIntPointer(maxits,2);
1055a7e14dcfSSatish Balay   *maxits = tao->max_it;
1056a7e14dcfSSatish Balay   PetscFunctionReturn(0);
1057a7e14dcfSSatish Balay }
1058a7e14dcfSSatish Balay 
1059a7e14dcfSSatish Balay /*@
1060a7e14dcfSSatish Balay    TaoSetInitialTrustRegionRadius - Sets the initial trust region radius.
1061a7e14dcfSSatish Balay 
1062441846f8SBarry Smith    Logically collective on Tao
1063a7e14dcfSSatish Balay 
1064d8d19677SJose E. Roman    Input Parameters:
1065a7e14dcfSSatish Balay +  tao - a TAO optimization solver
1066a7e14dcfSSatish Balay -  radius - the trust region radius
1067a7e14dcfSSatish Balay 
1068a7e14dcfSSatish Balay    Level: intermediate
1069a7e14dcfSSatish Balay 
1070a7e14dcfSSatish Balay    Options Database Key:
1071a7e14dcfSSatish Balay .  -tao_trust0 <t0> - sets initial trust region radius
1072a7e14dcfSSatish Balay 
1073a7e14dcfSSatish Balay .seealso: TaoGetTrustRegionRadius(), TaoSetTrustRegionTolerance()
1074a7e14dcfSSatish Balay @*/
1075441846f8SBarry Smith PetscErrorCode TaoSetInitialTrustRegionRadius(Tao tao, PetscReal radius)
1076a7e14dcfSSatish Balay {
1077a7e14dcfSSatish Balay   PetscFunctionBegin;
1078441846f8SBarry Smith   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
107994511df7SStefano Zampini   PetscValidLogicalCollectiveReal(tao,radius,2);
1080a7e14dcfSSatish Balay   tao->trust0 = PetscMax(0.0,radius);
10816552cf8aSJason Sarich   tao->trust0_changed = PETSC_TRUE;
1082a7e14dcfSSatish Balay   PetscFunctionReturn(0);
1083a7e14dcfSSatish Balay }
1084a7e14dcfSSatish Balay 
1085a7e14dcfSSatish Balay /*@
1086a7e14dcfSSatish Balay    TaoGetInitialTrustRegionRadius - Sets the initial trust region radius.
1087a7e14dcfSSatish Balay 
1088a7e14dcfSSatish Balay    Not Collective
1089a7e14dcfSSatish Balay 
1090a7e14dcfSSatish Balay    Input Parameter:
1091a7e14dcfSSatish Balay .  tao - a TAO optimization solver
1092a7e14dcfSSatish Balay 
1093a7e14dcfSSatish Balay    Output Parameter:
1094a7e14dcfSSatish Balay .  radius - the trust region radius
1095a7e14dcfSSatish Balay 
1096a7e14dcfSSatish Balay    Level: intermediate
1097a7e14dcfSSatish Balay 
1098a7e14dcfSSatish Balay .seealso: TaoSetInitialTrustRegionRadius(), TaoGetCurrentTrustRegionRadius()
1099a7e14dcfSSatish Balay @*/
1100441846f8SBarry Smith PetscErrorCode TaoGetInitialTrustRegionRadius(Tao tao, PetscReal *radius)
1101a7e14dcfSSatish Balay {
1102a7e14dcfSSatish Balay   PetscFunctionBegin;
1103441846f8SBarry Smith   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
110494511df7SStefano Zampini   PetscValidRealPointer(radius,2);
1105a7e14dcfSSatish Balay   *radius = tao->trust0;
1106a7e14dcfSSatish Balay   PetscFunctionReturn(0);
1107a7e14dcfSSatish Balay }
1108a7e14dcfSSatish Balay 
1109a7e14dcfSSatish Balay /*@
1110a7e14dcfSSatish Balay    TaoGetCurrentTrustRegionRadius - Gets the current trust region radius.
1111a7e14dcfSSatish Balay 
1112a7e14dcfSSatish Balay    Not Collective
1113a7e14dcfSSatish Balay 
1114a7e14dcfSSatish Balay    Input Parameter:
1115a7e14dcfSSatish Balay .  tao - a TAO optimization solver
1116a7e14dcfSSatish Balay 
1117a7e14dcfSSatish Balay    Output Parameter:
1118a7e14dcfSSatish Balay .  radius - the trust region radius
1119a7e14dcfSSatish Balay 
1120a7e14dcfSSatish Balay    Level: intermediate
1121a7e14dcfSSatish Balay 
1122a7e14dcfSSatish Balay .seealso: TaoSetInitialTrustRegionRadius(), TaoGetInitialTrustRegionRadius()
1123a7e14dcfSSatish Balay @*/
1124441846f8SBarry Smith PetscErrorCode TaoGetCurrentTrustRegionRadius(Tao tao, PetscReal *radius)
1125a7e14dcfSSatish Balay {
1126a7e14dcfSSatish Balay   PetscFunctionBegin;
1127441846f8SBarry Smith   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
112894511df7SStefano Zampini   PetscValidRealPointer(radius,2);
1129a7e14dcfSSatish Balay   *radius = tao->trust;
1130a7e14dcfSSatish Balay   PetscFunctionReturn(0);
1131a7e14dcfSSatish Balay }
1132a7e14dcfSSatish Balay 
1133a7e14dcfSSatish Balay /*@
1134a7e14dcfSSatish Balay   TaoGetTolerances - gets the current values of tolerances
1135a7e14dcfSSatish Balay 
1136a7e14dcfSSatish Balay   Not Collective
1137a7e14dcfSSatish Balay 
1138f899ff85SJose E. Roman   Input Parameter:
1139441846f8SBarry Smith . tao - the Tao context
1140a7e14dcfSSatish Balay 
1141a7e14dcfSSatish Balay   Output Parameters:
1142e52336cbSBarry Smith + gatol - stop if norm of gradient is less than this
1143a7e14dcfSSatish Balay . grtol - stop if relative norm of gradient is less than this
1144a7e14dcfSSatish Balay - gttol - stop if norm of gradient is reduced by a this factor
1145a7e14dcfSSatish Balay 
114647a47007SBarry Smith   Note: NULL can be used as an argument if not all tolerances values are needed
1147a7e14dcfSSatish Balay 
1148a7e14dcfSSatish Balay .seealso TaoSetTolerances()
1149a7e14dcfSSatish Balay 
1150a7e14dcfSSatish Balay   Level: intermediate
1151a7e14dcfSSatish Balay @*/
1152e52336cbSBarry Smith PetscErrorCode TaoGetTolerances(Tao tao, PetscReal *gatol, PetscReal *grtol, PetscReal *gttol)
1153a7e14dcfSSatish Balay {
1154a7e14dcfSSatish Balay   PetscFunctionBegin;
1155441846f8SBarry Smith   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
1156a7e14dcfSSatish Balay   if (gatol) *gatol = tao->gatol;
1157a7e14dcfSSatish Balay   if (grtol) *grtol = tao->grtol;
1158a7e14dcfSSatish Balay   if (gttol) *gttol = tao->gttol;
1159a7e14dcfSSatish Balay   PetscFunctionReturn(0);
1160a7e14dcfSSatish Balay }
1161a7e14dcfSSatish Balay 
1162a7e14dcfSSatish Balay /*@
1163a7e14dcfSSatish Balay   TaoGetKSP - Gets the linear solver used by the optimization solver.
1164a7e14dcfSSatish Balay   Application writers should use TaoGetKSP if they need direct access
1165a7e14dcfSSatish Balay   to the PETSc KSP object.
1166a7e14dcfSSatish Balay 
1167a7e14dcfSSatish Balay   Not Collective
1168a7e14dcfSSatish Balay 
1169a7e14dcfSSatish Balay    Input Parameters:
1170a7e14dcfSSatish Balay .  tao - the TAO solver
1171a7e14dcfSSatish Balay 
1172a7e14dcfSSatish Balay    Output Parameters:
1173a7e14dcfSSatish Balay .  ksp - the KSP linear solver used in the optimization solver
1174a7e14dcfSSatish Balay 
1175a7e14dcfSSatish Balay    Level: intermediate
1176a7e14dcfSSatish Balay 
1177a7e14dcfSSatish Balay @*/
1178441846f8SBarry Smith PetscErrorCode TaoGetKSP(Tao tao, KSP *ksp)
117947a47007SBarry Smith {
1180a7e14dcfSSatish Balay   PetscFunctionBegin;
118194511df7SStefano Zampini   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
118294511df7SStefano Zampini   PetscValidPointer(ksp,2);
1183a7e14dcfSSatish Balay   *ksp = tao->ksp;
1184a7e14dcfSSatish Balay   PetscFunctionReturn(0);
1185a7e14dcfSSatish Balay }
1186a7e14dcfSSatish Balay 
1187025e9500SJason Sarich /*@
1188025e9500SJason Sarich    TaoGetLinearSolveIterations - Gets the total number of linear iterations
1189025e9500SJason Sarich    used by the TAO solver
1190025e9500SJason Sarich 
1191025e9500SJason Sarich    Not Collective
1192025e9500SJason Sarich 
1193025e9500SJason Sarich    Input Parameter:
1194025e9500SJason Sarich .  tao - TAO context
1195025e9500SJason Sarich 
1196025e9500SJason Sarich    Output Parameter:
1197025e9500SJason Sarich .  lits - number of linear iterations
1198025e9500SJason Sarich 
1199025e9500SJason Sarich    Notes:
1200025e9500SJason Sarich    This counter is reset to zero for each successive call to TaoSolve()
1201025e9500SJason Sarich 
1202025e9500SJason Sarich    Level: intermediate
1203025e9500SJason Sarich 
1204025e9500SJason Sarich .seealso:  TaoGetKSP()
1205025e9500SJason Sarich @*/
1206025e9500SJason Sarich PetscErrorCode TaoGetLinearSolveIterations(Tao tao, PetscInt *lits)
1207025e9500SJason Sarich {
1208025e9500SJason Sarich   PetscFunctionBegin;
1209025e9500SJason Sarich   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
1210025e9500SJason Sarich   PetscValidIntPointer(lits,2);
12112d9aa51bSJason Sarich   *lits = tao->ksp_tot_its;
1212025e9500SJason Sarich   PetscFunctionReturn(0);
1213025e9500SJason Sarich }
1214025e9500SJason Sarich 
1215a7e14dcfSSatish Balay /*@
1216a7e14dcfSSatish Balay   TaoGetLineSearch - Gets the line search used by the optimization solver.
1217a7e14dcfSSatish Balay   Application writers should use TaoGetLineSearch if they need direct access
1218a7e14dcfSSatish Balay   to the TaoLineSearch object.
1219a7e14dcfSSatish Balay 
1220a7e14dcfSSatish Balay   Not Collective
1221a7e14dcfSSatish Balay 
1222a7e14dcfSSatish Balay    Input Parameters:
1223a7e14dcfSSatish Balay .  tao - the TAO solver
1224a7e14dcfSSatish Balay 
1225a7e14dcfSSatish Balay    Output Parameters:
1226a7e14dcfSSatish Balay .  ls - the line search used in the optimization solver
1227a7e14dcfSSatish Balay 
1228a7e14dcfSSatish Balay    Level: intermediate
1229a7e14dcfSSatish Balay 
1230a7e14dcfSSatish Balay @*/
1231441846f8SBarry Smith PetscErrorCode TaoGetLineSearch(Tao tao, TaoLineSearch *ls)
123247a47007SBarry Smith {
1233a7e14dcfSSatish Balay   PetscFunctionBegin;
123494511df7SStefano Zampini   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
123594511df7SStefano Zampini   PetscValidPointer(ls,2);
1236a7e14dcfSSatish Balay   *ls = tao->linesearch;
1237a7e14dcfSSatish Balay   PetscFunctionReturn(0);
1238a7e14dcfSSatish Balay }
1239a7e14dcfSSatish Balay 
1240a7e14dcfSSatish Balay /*@
1241a7e14dcfSSatish Balay   TaoAddLineSearchCounts - Adds the number of function evaluations spent
1242a7e14dcfSSatish Balay   in the line search to the running total.
1243a7e14dcfSSatish Balay 
1244a7e14dcfSSatish Balay    Input Parameters:
1245a7e14dcfSSatish Balay +  tao - the TAO solver
1246a7e14dcfSSatish Balay -  ls - the line search used in the optimization solver
1247a7e14dcfSSatish Balay 
1248a7e14dcfSSatish Balay    Level: developer
1249a7e14dcfSSatish Balay 
1250a7e14dcfSSatish Balay .seealso: TaoLineSearchApply()
1251a7e14dcfSSatish Balay @*/
1252441846f8SBarry Smith PetscErrorCode TaoAddLineSearchCounts(Tao tao)
125347a47007SBarry Smith {
1254a7e14dcfSSatish Balay   PetscBool      flg;
1255a7e14dcfSSatish Balay   PetscInt       nfeval,ngeval,nfgeval;
125647a47007SBarry Smith 
1257a7e14dcfSSatish Balay   PetscFunctionBegin;
1258441846f8SBarry Smith   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
1259a7e14dcfSSatish Balay   if (tao->linesearch) {
12609566063dSJacob Faibussowitsch     PetscCall(TaoLineSearchIsUsingTaoRoutines(tao->linesearch,&flg));
126194ae4db5SBarry Smith     if (!flg) {
12629566063dSJacob Faibussowitsch       PetscCall(TaoLineSearchGetNumberFunctionEvaluations(tao->linesearch,&nfeval,&ngeval,&nfgeval));
1263a7e14dcfSSatish Balay       tao->nfuncs += nfeval;
1264a7e14dcfSSatish Balay       tao->ngrads += ngeval;
1265a7e14dcfSSatish Balay       tao->nfuncgrads += nfgeval;
1266a7e14dcfSSatish Balay     }
1267a7e14dcfSSatish Balay   }
1268a7e14dcfSSatish Balay   PetscFunctionReturn(0);
1269a7e14dcfSSatish Balay }
1270a7e14dcfSSatish Balay 
1271a7e14dcfSSatish Balay /*@
1272a82e8c82SStefano Zampini   TaoGetSolution - Returns the vector with the current TAO solution
1273a7e14dcfSSatish Balay 
1274a7e14dcfSSatish Balay   Not Collective
1275a7e14dcfSSatish Balay 
1276a7e14dcfSSatish Balay   Input Parameter:
1277441846f8SBarry Smith . tao - the Tao context
1278a7e14dcfSSatish Balay 
1279a7e14dcfSSatish Balay   Output Parameter:
1280a7e14dcfSSatish Balay . X - the current solution
1281a7e14dcfSSatish Balay 
1282a7e14dcfSSatish Balay   Level: intermediate
1283a7e14dcfSSatish Balay 
1284a82e8c82SStefano Zampini   Note:  The returned vector will be the same object that was passed into TaoSetSolution()
1285a7e14dcfSSatish Balay @*/
1286a82e8c82SStefano Zampini PetscErrorCode TaoGetSolution(Tao tao, Vec *X)
1287a7e14dcfSSatish Balay {
1288a7e14dcfSSatish Balay   PetscFunctionBegin;
1289441846f8SBarry Smith   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
129094511df7SStefano Zampini   PetscValidPointer(X,2);
1291a7e14dcfSSatish Balay   *X = tao->solution;
1292a7e14dcfSSatish Balay   PetscFunctionReturn(0);
1293a7e14dcfSSatish Balay }
1294a7e14dcfSSatish Balay 
1295a7e14dcfSSatish Balay /*@
1296a7e14dcfSSatish Balay    TaoResetStatistics - Initialize the statistics used by TAO for all of the solvers.
1297a7e14dcfSSatish Balay    These statistics include the iteration number, residual norms, and convergence status.
1298a7e14dcfSSatish Balay    This routine gets called before solving each optimization problem.
1299a7e14dcfSSatish Balay 
1300441846f8SBarry Smith    Collective on Tao
1301a7e14dcfSSatish Balay 
1302a7e14dcfSSatish Balay    Input Parameters:
1303441846f8SBarry Smith .  solver - the Tao context
1304a7e14dcfSSatish Balay 
1305a7e14dcfSSatish Balay    Level: developer
1306a7e14dcfSSatish Balay 
1307a7e14dcfSSatish Balay .seealso: TaoCreate(), TaoSolve()
1308a7e14dcfSSatish Balay @*/
1309441846f8SBarry Smith PetscErrorCode TaoResetStatistics(Tao tao)
1310a7e14dcfSSatish Balay {
1311a7e14dcfSSatish Balay   PetscFunctionBegin;
1312441846f8SBarry Smith   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
1313a7e14dcfSSatish Balay   tao->niter        = 0;
1314a7e14dcfSSatish Balay   tao->nfuncs       = 0;
1315a7e14dcfSSatish Balay   tao->nfuncgrads   = 0;
1316a7e14dcfSSatish Balay   tao->ngrads       = 0;
1317a7e14dcfSSatish Balay   tao->nhess        = 0;
1318a7e14dcfSSatish Balay   tao->njac         = 0;
1319a7e14dcfSSatish Balay   tao->nconstraints = 0;
1320a7e14dcfSSatish Balay   tao->ksp_its      = 0;
1321ae93cb3cSJason Sarich   tao->ksp_tot_its  = 0;
1322a7e14dcfSSatish Balay   tao->reason       = TAO_CONTINUE_ITERATING;
1323a7e14dcfSSatish Balay   tao->residual     = 0.0;
1324a7e14dcfSSatish Balay   tao->cnorm        = 0.0;
1325a7e14dcfSSatish Balay   tao->step         = 0.0;
1326a7e14dcfSSatish Balay   tao->lsflag       = PETSC_FALSE;
1327a7e14dcfSSatish Balay   if (tao->hist_reset) tao->hist_len = 0;
1328a7e14dcfSSatish Balay   PetscFunctionReturn(0);
1329a7e14dcfSSatish Balay }
1330a7e14dcfSSatish Balay 
1331a7e14dcfSSatish Balay /*@C
1332e1e80dc8SAlp Dener   TaoSetUpdate - Sets the general-purpose update function called
1333e1e80dc8SAlp Dener   at the beginning of every iteration of the nonlinear solve. Specifically
1334e1e80dc8SAlp Dener   it is called at the top of every iteration, after the new solution and the gradient
1335e1e80dc8SAlp Dener   is determined, but before the Hessian is computed (if applicable).
1336e1e80dc8SAlp Dener 
1337e1e80dc8SAlp Dener   Logically Collective on Tao
1338e1e80dc8SAlp Dener 
1339e1e80dc8SAlp Dener   Input Parameters:
1340a2b725a8SWilliam Gropp + tao - The tao solver context
1341a2b725a8SWilliam Gropp - func - The function
1342e1e80dc8SAlp Dener 
1343e1e80dc8SAlp Dener   Calling sequence of func:
1344a2b725a8SWilliam Gropp $ func (Tao tao, PetscInt step);
1345e1e80dc8SAlp Dener 
1346e1e80dc8SAlp Dener . step - The current step of the iteration
1347e1e80dc8SAlp Dener 
1348e1e80dc8SAlp Dener   Level: advanced
1349e1e80dc8SAlp Dener 
1350e1e80dc8SAlp Dener .seealso TaoSolve()
1351e1e80dc8SAlp Dener @*/
13528fcddce6SStefano Zampini PetscErrorCode TaoSetUpdate(Tao tao, PetscErrorCode (*func)(Tao, PetscInt, void*), void *ctx)
1353e1e80dc8SAlp Dener {
1354e1e80dc8SAlp Dener   PetscFunctionBegin;
1355e1e80dc8SAlp Dener   PetscValidHeaderSpecific(tao, TAO_CLASSID,1);
1356e1e80dc8SAlp Dener   tao->ops->update = func;
1357e1e80dc8SAlp Dener   tao->user_update = ctx;
1358e1e80dc8SAlp Dener   PetscFunctionReturn(0);
1359e1e80dc8SAlp Dener }
1360e1e80dc8SAlp Dener 
1361e1e80dc8SAlp Dener /*@C
1362a7e14dcfSSatish Balay   TaoSetConvergenceTest - Sets the function that is to be used to test
1363a7e14dcfSSatish Balay   for convergence o fthe iterative minimization solution.  The new convergence
1364a7e14dcfSSatish Balay   testing routine will replace TAO's default convergence test.
1365a7e14dcfSSatish Balay 
1366441846f8SBarry Smith   Logically Collective on Tao
1367a7e14dcfSSatish Balay 
1368a7e14dcfSSatish Balay   Input Parameters:
1369441846f8SBarry Smith + tao - the Tao object
1370a7e14dcfSSatish Balay . conv - the routine to test for convergence
1371a7e14dcfSSatish Balay - ctx - [optional] context for private data for the convergence routine
137247a47007SBarry Smith         (may be NULL)
1373a7e14dcfSSatish Balay 
1374a7e14dcfSSatish Balay   Calling sequence of conv:
1375441846f8SBarry Smith $   PetscErrorCode conv(Tao tao, void *ctx)
1376a7e14dcfSSatish Balay 
1377441846f8SBarry Smith + tao - the Tao object
1378a7e14dcfSSatish Balay - ctx - [optional] convergence context
1379a7e14dcfSSatish Balay 
1380e4cb33bbSBarry Smith   Note: The new convergence testing routine should call TaoSetConvergedReason().
1381a7e14dcfSSatish Balay 
1382a7e14dcfSSatish Balay   Level: advanced
1383a7e14dcfSSatish Balay 
1384e4cb33bbSBarry Smith .seealso: TaoSetConvergedReason(), TaoGetSolutionStatus(), TaoGetTolerances(), TaoSetMonitor
1385a7e14dcfSSatish Balay 
1386a7e14dcfSSatish Balay @*/
1387441846f8SBarry Smith PetscErrorCode TaoSetConvergenceTest(Tao tao, PetscErrorCode (*conv)(Tao, void*), void *ctx)
1388a7e14dcfSSatish Balay {
1389a7e14dcfSSatish Balay   PetscFunctionBegin;
1390441846f8SBarry Smith   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
139194511df7SStefano Zampini   tao->ops->convergencetest = conv;
139294511df7SStefano Zampini   tao->cnvP = ctx;
1393a7e14dcfSSatish Balay   PetscFunctionReturn(0);
1394a7e14dcfSSatish Balay }
1395a7e14dcfSSatish Balay 
1396a7e14dcfSSatish Balay /*@C
1397a7e14dcfSSatish Balay    TaoSetMonitor - Sets an ADDITIONAL function that is to be used at every
1398a7e14dcfSSatish Balay    iteration of the solver to display the iteration's
1399a7e14dcfSSatish Balay    progress.
1400a7e14dcfSSatish Balay 
1401441846f8SBarry Smith    Logically Collective on Tao
1402a7e14dcfSSatish Balay 
1403a7e14dcfSSatish Balay    Input Parameters:
1404441846f8SBarry Smith +  tao - the Tao solver context
1405a7e14dcfSSatish Balay .  mymonitor - monitoring routine
1406a7e14dcfSSatish Balay -  mctx - [optional] user-defined context for private data for the
140747a47007SBarry Smith           monitor routine (may be NULL)
1408a7e14dcfSSatish Balay 
1409a7e14dcfSSatish Balay    Calling sequence of mymonitor:
1410147403d9SBarry Smith .vb
1411147403d9SBarry Smith      PetscErrorCode mymonitor(Tao tao,void *mctx)
1412147403d9SBarry Smith .ve
1413a7e14dcfSSatish Balay 
1414441846f8SBarry Smith +    tao - the Tao solver context
1415a7e14dcfSSatish Balay -    mctx - [optional] monitoring context
1416a7e14dcfSSatish Balay 
1417a7e14dcfSSatish Balay    Options Database Keys:
141898ea980cSBarry Smith +    -tao_monitor        - sets TaoMonitorDefault()
1419a7e14dcfSSatish Balay .    -tao_smonitor       - sets short monitor
1420a7e14dcfSSatish Balay .    -tao_cmonitor       - same as smonitor plus constraint norm
1421a7e14dcfSSatish Balay .    -tao_view_solution   - view solution at each iteration
1422a7e14dcfSSatish Balay .    -tao_view_gradient   - view gradient at each iteration
14234a48860cSAlp Dener .    -tao_view_ls_residual - view least-squares residual vector at each iteration
1424a7e14dcfSSatish 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.
1425a7e14dcfSSatish Balay 
1426a7e14dcfSSatish Balay    Notes:
1427a7e14dcfSSatish Balay    Several different monitoring routines may be set by calling
1428a7e14dcfSSatish Balay    TaoSetMonitor() multiple times; all will be called in the
1429a7e14dcfSSatish Balay    order in which they were set.
1430a7e14dcfSSatish Balay 
143195452b02SPatrick Sanan    Fortran Notes:
143295452b02SPatrick Sanan     Only one monitor function may be set
1433a7e14dcfSSatish Balay 
1434a7e14dcfSSatish Balay    Level: intermediate
1435a7e14dcfSSatish Balay 
143698ea980cSBarry Smith .seealso: TaoMonitorDefault(), TaoCancelMonitors(),  TaoSetDestroyRoutine()
1437a7e14dcfSSatish Balay @*/
1438441846f8SBarry Smith PetscErrorCode TaoSetMonitor(Tao tao, PetscErrorCode (*func)(Tao, void*), void *ctx,PetscErrorCode (*dest)(void**))
1439a7e14dcfSSatish Balay {
144008d19d1fSJason Sarich   PetscInt       i;
14418732526dSAlp Dener   PetscBool      identical;
144208d19d1fSJason Sarich 
1443a7e14dcfSSatish Balay   PetscFunctionBegin;
1444441846f8SBarry Smith   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
14453c859ba3SBarry Smith   PetscCheck(tao->numbermonitors < MAXTAOMONITORS,PetscObjectComm((PetscObject)tao),PETSC_ERR_SUP,"Cannot attach another monitor -- max=%d",MAXTAOMONITORS);
144608d19d1fSJason Sarich 
144708d19d1fSJason Sarich   for (i=0; i<tao->numbermonitors;i++) {
14489566063dSJacob Faibussowitsch     PetscCall(PetscMonitorCompare((PetscErrorCode (*)(void))func,ctx,dest,(PetscErrorCode (*)(void))tao->monitor[i],tao->monitorcontext[i],tao->monitordestroy[i],&identical));
14498732526dSAlp Dener     if (identical) PetscFunctionReturn(0);
145008d19d1fSJason Sarich   }
1451a7e14dcfSSatish Balay   tao->monitor[tao->numbermonitors] = func;
14528732526dSAlp Dener   tao->monitorcontext[tao->numbermonitors] = (void*)ctx;
1453a7e14dcfSSatish Balay   tao->monitordestroy[tao->numbermonitors] = dest;
1454a7e14dcfSSatish Balay   ++tao->numbermonitors;
1455a7e14dcfSSatish Balay   PetscFunctionReturn(0);
1456a7e14dcfSSatish Balay }
1457a7e14dcfSSatish Balay 
1458a7e14dcfSSatish Balay /*@
1459441846f8SBarry Smith    TaoCancelMonitors - Clears all the monitor functions for a Tao object.
1460a7e14dcfSSatish Balay 
1461441846f8SBarry Smith    Logically Collective on Tao
1462a7e14dcfSSatish Balay 
1463a7e14dcfSSatish Balay    Input Parameters:
1464441846f8SBarry Smith .  tao - the Tao solver context
1465a7e14dcfSSatish Balay 
1466a7e14dcfSSatish Balay    Options Database:
1467a7e14dcfSSatish Balay .  -tao_cancelmonitors - cancels all monitors that have been hardwired
1468a7e14dcfSSatish Balay     into a code by calls to TaoSetMonitor(), but does not cancel those
1469a7e14dcfSSatish Balay     set via the options database
1470a7e14dcfSSatish Balay 
1471a7e14dcfSSatish Balay    Notes:
1472441846f8SBarry Smith    There is no way to clear one specific monitor from a Tao object.
1473a7e14dcfSSatish Balay 
1474a7e14dcfSSatish Balay    Level: advanced
1475a7e14dcfSSatish Balay 
147698ea980cSBarry Smith .seealso: TaoMonitorDefault(), TaoSetMonitor()
1477a7e14dcfSSatish Balay @*/
1478441846f8SBarry Smith PetscErrorCode TaoCancelMonitors(Tao tao)
1479a7e14dcfSSatish Balay {
1480a7e14dcfSSatish Balay   PetscInt       i;
148147a47007SBarry Smith 
1482a7e14dcfSSatish Balay   PetscFunctionBegin;
1483441846f8SBarry Smith   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
1484a7e14dcfSSatish Balay   for (i=0;i<tao->numbermonitors;i++) {
1485a7e14dcfSSatish Balay     if (tao->monitordestroy[i]) {
14869566063dSJacob Faibussowitsch       PetscCall((*tao->monitordestroy[i])(&tao->monitorcontext[i]));
1487a7e14dcfSSatish Balay     }
1488a7e14dcfSSatish Balay   }
1489a7e14dcfSSatish Balay   tao->numbermonitors = 0;
1490a7e14dcfSSatish Balay   PetscFunctionReturn(0);
1491a7e14dcfSSatish Balay }
1492a7e14dcfSSatish Balay 
14937fab98ebSJason Sarich /*@
149498ea980cSBarry Smith    TaoMonitorDefault - Default routine for monitoring progress of the
1495441846f8SBarry Smith    Tao solvers (default).  This monitor prints the function value and gradient
1496a7e14dcfSSatish Balay    norm at each iteration.  It can be turned on from the command line using the
1497a7e14dcfSSatish Balay    -tao_monitor option
1498a7e14dcfSSatish Balay 
1499441846f8SBarry Smith    Collective on Tao
1500a7e14dcfSSatish Balay 
1501a7e14dcfSSatish Balay    Input Parameters:
1502441846f8SBarry Smith +  tao - the Tao context
150347a47007SBarry Smith -  ctx - PetscViewer context or NULL
1504a7e14dcfSSatish Balay 
1505a7e14dcfSSatish Balay    Options Database Keys:
1506147403d9SBarry Smith .  -tao_monitor - turn on default monitoring
1507a7e14dcfSSatish Balay 
1508a7e14dcfSSatish Balay    Level: advanced
1509a7e14dcfSSatish Balay 
1510a7e14dcfSSatish Balay .seealso: TaoDefaultSMonitor(), TaoSetMonitor()
1511a7e14dcfSSatish Balay @*/
151298ea980cSBarry Smith PetscErrorCode TaoMonitorDefault(Tao tao, void *ctx)
1513a7e14dcfSSatish Balay {
151463b15415SAlp Dener   PetscInt       its, tabs;
1515a7e14dcfSSatish Balay   PetscReal      fct,gnorm;
15168163d661SBarry Smith   PetscViewer    viewer = (PetscViewer)ctx;
151747a47007SBarry Smith 
1518a7e14dcfSSatish Balay   PetscFunctionBegin;
151994511df7SStefano Zampini   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
15208163d661SBarry Smith   PetscValidHeaderSpecific(viewer,PETSC_VIEWER_CLASSID,2);
1521a7e14dcfSSatish Balay   its = tao->niter;
1522a7e14dcfSSatish Balay   fct = tao->fc;
1523a7e14dcfSSatish Balay   gnorm = tao->residual;
15249566063dSJacob Faibussowitsch   PetscCall(PetscViewerASCIIGetTab(viewer, &tabs));
15259566063dSJacob Faibussowitsch   PetscCall(PetscViewerASCIISetTab(viewer, ((PetscObject)tao)->tablevel));
1526494bef23SAlp Dener   if (its == 0 && ((PetscObject)tao)->prefix && !tao->header_printed) {
15279566063dSJacob Faibussowitsch      PetscCall(PetscViewerASCIIPrintf(viewer,"  Iteration information for %s solve.\n",((PetscObject)tao)->prefix));
1528494bef23SAlp Dener      tao->header_printed = PETSC_TRUE;
152963b15415SAlp Dener    }
15309566063dSJacob Faibussowitsch   PetscCall(PetscViewerASCIIPrintf(viewer,"%3D TAO,",its));
15319566063dSJacob Faibussowitsch   PetscCall(PetscViewerASCIIPrintf(viewer,"  Function value: %g,",(double)fct));
153208d19d1fSJason Sarich   if (gnorm >= PETSC_INFINITY) {
15339566063dSJacob Faibussowitsch     PetscCall(PetscViewerASCIIPrintf(viewer,"  Residual: Inf \n"));
153408d19d1fSJason Sarich   } else {
15359566063dSJacob Faibussowitsch     PetscCall(PetscViewerASCIIPrintf(viewer,"  Residual: %g \n",(double)gnorm));
153608d19d1fSJason Sarich   }
15379566063dSJacob Faibussowitsch   PetscCall(PetscViewerASCIISetTab(viewer, tabs));
1538a7e14dcfSSatish Balay   PetscFunctionReturn(0);
1539a7e14dcfSSatish Balay }
1540a7e14dcfSSatish Balay 
15417fab98ebSJason Sarich /*@
15428d5ead36SAlp Dener    TaoDefaultGMonitor - Default routine for monitoring progress of the
15438d5ead36SAlp Dener    Tao solvers (default) with extra detail on the globalization method.
15448d5ead36SAlp Dener    This monitor prints the function value and gradient norm at each
15458d5ead36SAlp Dener    iteration, as well as the step size and trust radius. Note that the
15468d5ead36SAlp Dener    step size and trust radius may be the same for some algorithms.
15478d5ead36SAlp Dener    It can be turned on from the command line using the
15488d5ead36SAlp Dener    -tao_gmonitor option
15498d5ead36SAlp Dener 
15508d5ead36SAlp Dener    Collective on Tao
15518d5ead36SAlp Dener 
15528d5ead36SAlp Dener    Input Parameters:
15538d5ead36SAlp Dener +  tao - the Tao context
15548d5ead36SAlp Dener -  ctx - PetscViewer context or NULL
15558d5ead36SAlp Dener 
15568d5ead36SAlp Dener    Options Database Keys:
1557147403d9SBarry Smith .  -tao_gmonitor - turn on monitoring with globalization information
15588d5ead36SAlp Dener 
15598d5ead36SAlp Dener    Level: advanced
15608d5ead36SAlp Dener 
15618d5ead36SAlp Dener .seealso: TaoDefaultSMonitor(), TaoSetMonitor()
15628d5ead36SAlp Dener @*/
15638d5ead36SAlp Dener PetscErrorCode TaoDefaultGMonitor(Tao tao, void *ctx)
15648d5ead36SAlp Dener {
15658d5ead36SAlp Dener   PetscInt       its, tabs;
15668d5ead36SAlp Dener   PetscReal      fct,gnorm,stp,tr;
15678d5ead36SAlp Dener   PetscViewer    viewer = (PetscViewer)ctx;
15688d5ead36SAlp Dener 
15698d5ead36SAlp Dener   PetscFunctionBegin;
157094511df7SStefano Zampini   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
15718d5ead36SAlp Dener   PetscValidHeaderSpecific(viewer,PETSC_VIEWER_CLASSID,2);
15728d5ead36SAlp Dener   its = tao->niter;
15738d5ead36SAlp Dener   fct = tao->fc;
15748d5ead36SAlp Dener   gnorm = tao->residual;
15758d5ead36SAlp Dener   stp = tao->step;
15768d5ead36SAlp Dener   tr = tao->trust;
15779566063dSJacob Faibussowitsch   PetscCall(PetscViewerASCIIGetTab(viewer, &tabs));
15789566063dSJacob Faibussowitsch   PetscCall(PetscViewerASCIISetTab(viewer, ((PetscObject)tao)->tablevel));
1579494bef23SAlp Dener   if (its == 0 && ((PetscObject)tao)->prefix && !tao->header_printed) {
15809566063dSJacob Faibussowitsch      PetscCall(PetscViewerASCIIPrintf(viewer,"  Iteration information for %s solve.\n",((PetscObject)tao)->prefix));
1581494bef23SAlp Dener      tao->header_printed = PETSC_TRUE;
15828d5ead36SAlp Dener    }
15839566063dSJacob Faibussowitsch   PetscCall(PetscViewerASCIIPrintf(viewer,"%3D TAO,",its));
15849566063dSJacob Faibussowitsch   PetscCall(PetscViewerASCIIPrintf(viewer,"  Function value: %g,",(double)fct));
15858d5ead36SAlp Dener   if (gnorm >= PETSC_INFINITY) {
15869566063dSJacob Faibussowitsch     PetscCall(PetscViewerASCIIPrintf(viewer,"  Residual: Inf,"));
15878d5ead36SAlp Dener   } else {
15889566063dSJacob Faibussowitsch     PetscCall(PetscViewerASCIIPrintf(viewer,"  Residual: %g,",(double)gnorm));
15898d5ead36SAlp Dener   }
15909566063dSJacob Faibussowitsch   PetscCall(PetscViewerASCIIPrintf(viewer,"  Step: %g,  Trust: %g\n",(double)stp,(double)tr));
15919566063dSJacob Faibussowitsch   PetscCall(PetscViewerASCIISetTab(viewer, tabs));
15928d5ead36SAlp Dener   PetscFunctionReturn(0);
15938d5ead36SAlp Dener }
15948d5ead36SAlp Dener 
15958d5ead36SAlp Dener /*@
1596a7e14dcfSSatish Balay    TaoDefaultSMonitor - Default routine for monitoring progress of the
159798ea980cSBarry Smith    solver. Same as TaoMonitorDefault() except
1598a7e14dcfSSatish Balay    it prints fewer digits of the residual as the residual gets smaller.
1599a7e14dcfSSatish Balay    This is because the later digits are meaningless and are often
1600a7e14dcfSSatish Balay    different on different machines; by using this routine different
1601a7e14dcfSSatish Balay    machines will usually generate the same output. It can be turned on
1602a7e14dcfSSatish Balay    by using the -tao_smonitor option
1603a7e14dcfSSatish Balay 
1604441846f8SBarry Smith    Collective on Tao
1605a7e14dcfSSatish Balay 
1606a7e14dcfSSatish Balay    Input Parameters:
1607441846f8SBarry Smith +  tao - the Tao context
16084d4332d5SBarry Smith -  ctx - PetscViewer context of type ASCII
1609a7e14dcfSSatish Balay 
1610a7e14dcfSSatish Balay    Options Database Keys:
1611147403d9SBarry Smith .  -tao_smonitor - turn on default short monitoring
1612a7e14dcfSSatish Balay 
1613a7e14dcfSSatish Balay    Level: advanced
1614a7e14dcfSSatish Balay 
161598ea980cSBarry Smith .seealso: TaoMonitorDefault(), TaoSetMonitor()
1616a7e14dcfSSatish Balay @*/
1617441846f8SBarry Smith PetscErrorCode TaoDefaultSMonitor(Tao tao, void *ctx)
1618a7e14dcfSSatish Balay {
16192c9b8b83SAlp Dener   PetscInt       its, tabs;
1620a7e14dcfSSatish Balay   PetscReal      fct,gnorm;
16214d4332d5SBarry Smith   PetscViewer    viewer = (PetscViewer)ctx;
162247a47007SBarry Smith 
1623a7e14dcfSSatish Balay   PetscFunctionBegin;
162494511df7SStefano Zampini   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
16254d4332d5SBarry Smith   PetscValidHeaderSpecific(viewer,PETSC_VIEWER_CLASSID,2);
1626a7e14dcfSSatish Balay   its = tao->niter;
1627a7e14dcfSSatish Balay   fct = tao->fc;
1628a7e14dcfSSatish Balay   gnorm = tao->residual;
16299566063dSJacob Faibussowitsch   PetscCall(PetscViewerASCIIGetTab(viewer, &tabs));
16309566063dSJacob Faibussowitsch   PetscCall(PetscViewerASCIISetTab(viewer, ((PetscObject)tao)->tablevel));
16319566063dSJacob Faibussowitsch   PetscCall(PetscViewerASCIIPrintf(viewer,"iter = %3D,",its));
16329566063dSJacob Faibussowitsch   PetscCall(PetscViewerASCIIPrintf(viewer," Function value %g,",(double)fct));
1633d393f493SBarry Smith   if (gnorm >= PETSC_INFINITY) {
16349566063dSJacob Faibussowitsch     PetscCall(PetscViewerASCIIPrintf(viewer," Residual: Inf \n"));
163508d19d1fSJason Sarich   } else if (gnorm > 1.e-6) {
16369566063dSJacob Faibussowitsch     PetscCall(PetscViewerASCIIPrintf(viewer," Residual: %g \n",(double)gnorm));
1637a7e14dcfSSatish Balay   } else if (gnorm > 1.e-11) {
16389566063dSJacob Faibussowitsch     PetscCall(PetscViewerASCIIPrintf(viewer," Residual: < 1.0e-6 \n"));
1639a7e14dcfSSatish Balay   } else {
16409566063dSJacob Faibussowitsch     PetscCall(PetscViewerASCIIPrintf(viewer," Residual: < 1.0e-11 \n"));
1641a7e14dcfSSatish Balay   }
16429566063dSJacob Faibussowitsch   PetscCall(PetscViewerASCIISetTab(viewer, tabs));
1643a7e14dcfSSatish Balay   PetscFunctionReturn(0);
1644a7e14dcfSSatish Balay }
1645a7e14dcfSSatish Balay 
16467fab98ebSJason Sarich /*@
164798ea980cSBarry Smith    TaoDefaultCMonitor - same as TaoMonitorDefault() except
1648a7e14dcfSSatish Balay    it prints the norm of the constraints function. It can be turned on
1649a7e14dcfSSatish Balay    from the command line using the -tao_cmonitor option
1650a7e14dcfSSatish Balay 
1651441846f8SBarry Smith    Collective on Tao
1652a7e14dcfSSatish Balay 
1653a7e14dcfSSatish Balay    Input Parameters:
1654441846f8SBarry Smith +  tao - the Tao context
165547a47007SBarry Smith -  ctx - PetscViewer context or NULL
1656a7e14dcfSSatish Balay 
1657a7e14dcfSSatish Balay    Options Database Keys:
1658147403d9SBarry Smith .  -tao_cmonitor - monitor the constraints
1659a7e14dcfSSatish Balay 
1660a7e14dcfSSatish Balay    Level: advanced
1661a7e14dcfSSatish Balay 
166298ea980cSBarry Smith .seealso: TaoMonitorDefault(), TaoSetMonitor()
1663a7e14dcfSSatish Balay @*/
1664441846f8SBarry Smith PetscErrorCode TaoDefaultCMonitor(Tao tao, void *ctx)
1665a7e14dcfSSatish Balay {
16662c9b8b83SAlp Dener   PetscInt       its, tabs;
1667a7e14dcfSSatish Balay   PetscReal      fct,gnorm;
1668d393f493SBarry Smith   PetscViewer    viewer = (PetscViewer)ctx;
1669a7e14dcfSSatish Balay 
1670a7e14dcfSSatish Balay   PetscFunctionBegin;
167194511df7SStefano Zampini   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
1672d393f493SBarry Smith   PetscValidHeaderSpecific(viewer,PETSC_VIEWER_CLASSID,2);
1673a7e14dcfSSatish Balay   its = tao->niter;
1674a7e14dcfSSatish Balay   fct = tao->fc;
1675a7e14dcfSSatish Balay   gnorm = tao->residual;
16769566063dSJacob Faibussowitsch   PetscCall(PetscViewerASCIIGetTab(viewer, &tabs));
16779566063dSJacob Faibussowitsch   PetscCall(PetscViewerASCIISetTab(viewer, ((PetscObject)tao)->tablevel));
16789566063dSJacob Faibussowitsch   PetscCall(PetscViewerASCIIPrintf(viewer,"iter = %D,",its));
16799566063dSJacob Faibussowitsch   PetscCall(PetscViewerASCIIPrintf(viewer," Function value: %g,",(double)fct));
16809566063dSJacob Faibussowitsch   PetscCall(PetscViewerASCIIPrintf(viewer,"  Residual: %g ",(double)gnorm));
16819566063dSJacob Faibussowitsch   PetscCall(PetscViewerASCIIPrintf(viewer,"  Constraint: %g \n",(double)tao->cnorm));
16829566063dSJacob Faibussowitsch   PetscCall(PetscViewerASCIISetTab(viewer, tabs));
1683a7e14dcfSSatish Balay   PetscFunctionReturn(0);
1684a7e14dcfSSatish Balay }
1685a7e14dcfSSatish Balay 
1686a7e14dcfSSatish Balay /*@C
1687a7e14dcfSSatish Balay    TaoSolutionMonitor - Views the solution at each iteration
1688a7e14dcfSSatish Balay    It can be turned on from the command line using the
1689a7e14dcfSSatish Balay    -tao_view_solution option
1690a7e14dcfSSatish Balay 
1691441846f8SBarry Smith    Collective on Tao
1692a7e14dcfSSatish Balay 
1693a7e14dcfSSatish Balay    Input Parameters:
1694441846f8SBarry Smith +  tao - the Tao context
169547a47007SBarry Smith -  ctx - PetscViewer context or NULL
1696a7e14dcfSSatish Balay 
1697a7e14dcfSSatish Balay    Options Database Keys:
1698147403d9SBarry Smith .  -tao_view_solution - view the solution
1699a7e14dcfSSatish Balay 
1700a7e14dcfSSatish Balay    Level: advanced
1701a7e14dcfSSatish Balay 
1702a7e14dcfSSatish Balay .seealso: TaoDefaultSMonitor(), TaoSetMonitor()
1703a7e14dcfSSatish Balay @*/
1704441846f8SBarry Smith PetscErrorCode TaoSolutionMonitor(Tao tao, void *ctx)
1705a7e14dcfSSatish Balay {
1706feb237baSPierre Jolivet   PetscViewer    viewer  = (PetscViewer)ctx;
170747a47007SBarry Smith 
1708a7e14dcfSSatish Balay   PetscFunctionBegin;
170994511df7SStefano Zampini   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
1710d393f493SBarry Smith   PetscValidHeaderSpecific(viewer,PETSC_VIEWER_CLASSID,2);
17119566063dSJacob Faibussowitsch   PetscCall(VecView(tao->solution,viewer));
1712a7e14dcfSSatish Balay   PetscFunctionReturn(0);
1713a7e14dcfSSatish Balay }
1714a7e14dcfSSatish Balay 
1715a7e14dcfSSatish Balay /*@C
1716a7e14dcfSSatish Balay    TaoGradientMonitor - Views the gradient at each iteration
1717a7e14dcfSSatish Balay    It can be turned on from the command line using the
1718a7e14dcfSSatish Balay    -tao_view_gradient option
1719a7e14dcfSSatish Balay 
1720441846f8SBarry Smith    Collective on Tao
1721a7e14dcfSSatish Balay 
1722a7e14dcfSSatish Balay    Input Parameters:
1723441846f8SBarry Smith +  tao - the Tao context
172447a47007SBarry Smith -  ctx - PetscViewer context or NULL
1725a7e14dcfSSatish Balay 
1726a7e14dcfSSatish Balay    Options Database Keys:
1727147403d9SBarry Smith .  -tao_view_gradient - view the gradient at each iteration
1728a7e14dcfSSatish Balay 
1729a7e14dcfSSatish Balay    Level: advanced
1730a7e14dcfSSatish Balay 
1731a7e14dcfSSatish Balay .seealso: TaoDefaultSMonitor(), TaoSetMonitor()
1732a7e14dcfSSatish Balay @*/
1733441846f8SBarry Smith PetscErrorCode TaoGradientMonitor(Tao tao, void *ctx)
1734a7e14dcfSSatish Balay {
1735d393f493SBarry Smith   PetscViewer    viewer = (PetscViewer)ctx;
173647a47007SBarry Smith 
1737a7e14dcfSSatish Balay   PetscFunctionBegin;
173894511df7SStefano Zampini   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
1739d393f493SBarry Smith   PetscValidHeaderSpecific(viewer,PETSC_VIEWER_CLASSID,2);
17409566063dSJacob Faibussowitsch   PetscCall(VecView(tao->gradient,viewer));
1741a7e14dcfSSatish Balay   PetscFunctionReturn(0);
1742a7e14dcfSSatish Balay }
1743a7e14dcfSSatish Balay 
1744a7e14dcfSSatish Balay /*@C
1745147403d9SBarry Smith    TaoStepDirectionMonitor - Views the step-direction at each iteration
1746a7e14dcfSSatish Balay 
1747441846f8SBarry Smith    Collective on Tao
1748a7e14dcfSSatish Balay 
1749a7e14dcfSSatish Balay    Input Parameters:
1750441846f8SBarry Smith +  tao - the Tao context
175147a47007SBarry Smith -  ctx - PetscViewer context or NULL
1752a7e14dcfSSatish Balay 
1753a7e14dcfSSatish Balay    Options Database Keys:
1754147403d9SBarry Smith .  -tao_view_gradient - view the gradient at each iteration
1755a7e14dcfSSatish Balay 
1756a7e14dcfSSatish Balay    Level: advanced
1757a7e14dcfSSatish Balay 
1758a7e14dcfSSatish Balay .seealso: TaoDefaultSMonitor(), TaoSetMonitor()
1759a7e14dcfSSatish Balay @*/
1760441846f8SBarry Smith PetscErrorCode TaoStepDirectionMonitor(Tao tao, void *ctx)
1761a7e14dcfSSatish Balay {
1762d393f493SBarry Smith   PetscViewer    viewer = (PetscViewer)ctx;
1763d393f493SBarry Smith 
1764a7e14dcfSSatish Balay   PetscFunctionBegin;
176594511df7SStefano Zampini   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
1766d393f493SBarry Smith   PetscValidHeaderSpecific(viewer,PETSC_VIEWER_CLASSID,2);
17679566063dSJacob Faibussowitsch   PetscCall(VecView(tao->stepdirection,viewer));
1768a7e14dcfSSatish Balay   PetscFunctionReturn(0);
1769a7e14dcfSSatish Balay }
1770a7e14dcfSSatish Balay 
1771a7e14dcfSSatish Balay /*@C
1772a7e14dcfSSatish Balay    TaoDrawSolutionMonitor - Plots the solution at each iteration
1773a7e14dcfSSatish Balay    It can be turned on from the command line using the
1774a7e14dcfSSatish Balay    -tao_draw_solution option
1775a7e14dcfSSatish Balay 
1776441846f8SBarry Smith    Collective on Tao
1777a7e14dcfSSatish Balay 
1778a7e14dcfSSatish Balay    Input Parameters:
1779441846f8SBarry Smith +  tao - the Tao context
1780e882e171SHong Zhang -  ctx - TaoMonitorDraw context
1781a7e14dcfSSatish Balay 
1782a7e14dcfSSatish Balay    Options Database Keys:
1783147403d9SBarry Smith .  -tao_draw_solution - draw the solution at each iteration
1784a7e14dcfSSatish Balay 
1785a7e14dcfSSatish Balay    Level: advanced
1786a7e14dcfSSatish Balay 
1787a7e14dcfSSatish Balay .seealso: TaoSolutionMonitor(), TaoSetMonitor(), TaoDrawGradientMonitor
1788a7e14dcfSSatish Balay @*/
1789441846f8SBarry Smith PetscErrorCode TaoDrawSolutionMonitor(Tao tao, void *ctx)
1790a7e14dcfSSatish Balay {
1791e882e171SHong Zhang   TaoMonitorDrawCtx ictx = (TaoMonitorDrawCtx)ctx;
179247a47007SBarry Smith 
1793a7e14dcfSSatish Balay   PetscFunctionBegin;
179494511df7SStefano Zampini   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
1795e882e171SHong Zhang   if (!(((ictx->howoften > 0) && (!(tao->niter % ictx->howoften))) || ((ictx->howoften == -1) && tao->reason))) PetscFunctionReturn(0);
17969566063dSJacob Faibussowitsch   PetscCall(VecView(tao->solution,ictx->viewer));
1797a7e14dcfSSatish Balay   PetscFunctionReturn(0);
1798a7e14dcfSSatish Balay }
1799a7e14dcfSSatish Balay 
1800a7e14dcfSSatish Balay /*@C
1801a7e14dcfSSatish Balay    TaoDrawGradientMonitor - Plots the gradient at each iteration
1802a7e14dcfSSatish Balay    It can be turned on from the command line using the
1803a7e14dcfSSatish Balay    -tao_draw_gradient option
1804a7e14dcfSSatish Balay 
1805441846f8SBarry Smith    Collective on Tao
1806a7e14dcfSSatish Balay 
1807a7e14dcfSSatish Balay    Input Parameters:
1808441846f8SBarry Smith +  tao - the Tao context
1809f55353a2SBarry Smith -  ctx - PetscViewer context
1810a7e14dcfSSatish Balay 
1811a7e14dcfSSatish Balay    Options Database Keys:
1812147403d9SBarry Smith .  -tao_draw_gradient - draw the gradient at each iteration
1813a7e14dcfSSatish Balay 
1814a7e14dcfSSatish Balay    Level: advanced
1815a7e14dcfSSatish Balay 
1816a7e14dcfSSatish Balay .seealso: TaoGradientMonitor(), TaoSetMonitor(), TaoDrawSolutionMonitor
1817a7e14dcfSSatish Balay @*/
1818441846f8SBarry Smith PetscErrorCode TaoDrawGradientMonitor(Tao tao, void *ctx)
1819a7e14dcfSSatish Balay {
1820e882e171SHong Zhang   TaoMonitorDrawCtx ictx = (TaoMonitorDrawCtx)ctx;
182147a47007SBarry Smith 
1822a7e14dcfSSatish Balay   PetscFunctionBegin;
182394511df7SStefano Zampini   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
1824e882e171SHong Zhang   if (!(((ictx->howoften > 0) && (!(tao->niter % ictx->howoften))) || ((ictx->howoften == -1) && tao->reason))) PetscFunctionReturn(0);
18259566063dSJacob Faibussowitsch   PetscCall(VecView(tao->gradient,ictx->viewer));
1826a7e14dcfSSatish Balay   PetscFunctionReturn(0);
1827a7e14dcfSSatish Balay }
1828a7e14dcfSSatish Balay 
1829a7e14dcfSSatish Balay /*@C
1830a7e14dcfSSatish Balay    TaoDrawStepMonitor - Plots the step direction at each iteration
1831a7e14dcfSSatish Balay 
1832441846f8SBarry Smith    Collective on Tao
1833a7e14dcfSSatish Balay 
1834a7e14dcfSSatish Balay    Input Parameters:
1835441846f8SBarry Smith +  tao - the Tao context
1836f55353a2SBarry Smith -  ctx - PetscViewer context
1837a7e14dcfSSatish Balay 
1838a7e14dcfSSatish Balay    Options Database Keys:
1839147403d9SBarry Smith .  -tao_draw_step - draw the step direction at each iteration
1840a7e14dcfSSatish Balay 
1841a7e14dcfSSatish Balay    Level: advanced
1842a7e14dcfSSatish Balay 
1843a7e14dcfSSatish Balay .seealso: TaoSetMonitor(), TaoDrawSolutionMonitor
1844a7e14dcfSSatish Balay @*/
1845441846f8SBarry Smith PetscErrorCode TaoDrawStepMonitor(Tao tao, void *ctx)
1846a7e14dcfSSatish Balay {
184794511df7SStefano Zampini   PetscViewer    viewer = (PetscViewer)ctx;
184847a47007SBarry Smith 
1849a7e14dcfSSatish Balay   PetscFunctionBegin;
185094511df7SStefano Zampini   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
185194511df7SStefano Zampini   PetscValidHeaderSpecific(viewer,PETSC_VIEWER_CLASSID,2);
18529566063dSJacob Faibussowitsch   PetscCall(VecView(tao->stepdirection,viewer));
1853a7e14dcfSSatish Balay   PetscFunctionReturn(0);
1854a7e14dcfSSatish Balay }
1855a7e14dcfSSatish Balay 
1856a7e14dcfSSatish Balay /*@C
1857737f463aSAlp Dener    TaoResidualMonitor - Views the least-squares residual at each iteration
1858a7e14dcfSSatish Balay 
1859441846f8SBarry Smith    Collective on Tao
1860a7e14dcfSSatish Balay 
1861a7e14dcfSSatish Balay    Input Parameters:
1862441846f8SBarry Smith +  tao - the Tao context
186347a47007SBarry Smith -  ctx - PetscViewer context or NULL
1864a7e14dcfSSatish Balay 
1865a7e14dcfSSatish Balay    Options Database Keys:
1866147403d9SBarry Smith .  -tao_view_ls_residual - view the least-squares residual at each iteration
1867a7e14dcfSSatish Balay 
1868a7e14dcfSSatish Balay    Level: advanced
1869a7e14dcfSSatish Balay 
1870a7e14dcfSSatish Balay .seealso: TaoDefaultSMonitor(), TaoSetMonitor()
1871a7e14dcfSSatish Balay @*/
1872737f463aSAlp Dener PetscErrorCode TaoResidualMonitor(Tao tao, void *ctx)
1873a7e14dcfSSatish Balay {
1874d393f493SBarry Smith   PetscViewer    viewer  = (PetscViewer)ctx;
187547a47007SBarry Smith 
1876a7e14dcfSSatish Balay   PetscFunctionBegin;
187794511df7SStefano Zampini   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
1878d393f493SBarry Smith   PetscValidHeaderSpecific(viewer,PETSC_VIEWER_CLASSID,2);
18799566063dSJacob Faibussowitsch   PetscCall(VecView(tao->ls_res,viewer));
1880a7e14dcfSSatish Balay   PetscFunctionReturn(0);
1881a7e14dcfSSatish Balay }
1882a7e14dcfSSatish Balay 
18837fab98ebSJason Sarich /*@
1884a7e14dcfSSatish Balay    TaoDefaultConvergenceTest - Determines whether the solver should continue iterating
1885a7e14dcfSSatish Balay    or terminate.
1886a7e14dcfSSatish Balay 
1887441846f8SBarry Smith    Collective on Tao
1888a7e14dcfSSatish Balay 
1889a7e14dcfSSatish Balay    Input Parameters:
1890441846f8SBarry Smith +  tao - the Tao context
1891a7e14dcfSSatish Balay -  dummy - unused dummy context
1892a7e14dcfSSatish Balay 
1893a7e14dcfSSatish Balay    Output Parameter:
1894a7e14dcfSSatish Balay .  reason - for terminating
1895a7e14dcfSSatish Balay 
1896a7e14dcfSSatish Balay    Notes:
1897a7e14dcfSSatish Balay    This routine checks the residual in the optimality conditions, the
1898a7e14dcfSSatish Balay    relative residual in the optimity conditions, the number of function
1899a7e14dcfSSatish Balay    evaluations, and the function value to test convergence.  Some
1900a7e14dcfSSatish Balay    solvers may use different convergence routines.
1901a7e14dcfSSatish Balay 
1902a7e14dcfSSatish Balay    Level: developer
1903a7e14dcfSSatish Balay 
1904e4cb33bbSBarry Smith .seealso: TaoSetTolerances(),TaoGetConvergedReason(),TaoSetConvergedReason()
1905a7e14dcfSSatish Balay @*/
1906a7e14dcfSSatish Balay 
1907441846f8SBarry Smith PetscErrorCode TaoDefaultConvergenceTest(Tao tao,void *dummy)
1908a7e14dcfSSatish Balay {
1909a7e14dcfSSatish Balay   PetscInt           niter=tao->niter, nfuncs=PetscMax(tao->nfuncs,tao->nfuncgrads);
1910a7e14dcfSSatish Balay   PetscInt           max_funcs=tao->max_funcs;
1911a7e14dcfSSatish Balay   PetscReal          gnorm=tao->residual, gnorm0=tao->gnorm0;
1912a7e14dcfSSatish Balay   PetscReal          f=tao->fc, steptol=tao->steptol,trradius=tao->step;
1913a7e14dcfSSatish Balay   PetscReal          gatol=tao->gatol,grtol=tao->grtol,gttol=tao->gttol;
1914e52336cbSBarry Smith   PetscReal          catol=tao->catol,crtol=tao->crtol;
1915e52336cbSBarry Smith   PetscReal          fmin=tao->fmin, cnorm=tao->cnorm;
1916e4cb33bbSBarry Smith   TaoConvergedReason reason=tao->reason;
1917a7e14dcfSSatish Balay 
1918a7e14dcfSSatish Balay   PetscFunctionBegin;
1919441846f8SBarry Smith   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
192094511df7SStefano Zampini   if (reason != TAO_CONTINUE_ITERATING) PetscFunctionReturn(0);
1921a7e14dcfSSatish Balay 
1922a7e14dcfSSatish Balay   if (PetscIsInfOrNanReal(f)) {
19239566063dSJacob Faibussowitsch     PetscCall(PetscInfo(tao,"Failed to converged, function value is Inf or NaN\n"));
1924a7e14dcfSSatish Balay     reason = TAO_DIVERGED_NAN;
1925a7e14dcfSSatish Balay   } else if (f <= fmin && cnorm <=catol) {
19269566063dSJacob Faibussowitsch     PetscCall(PetscInfo(tao,"Converged due to function value %g < minimum function value %g\n", (double)f,(double)fmin));
1927a7e14dcfSSatish Balay     reason = TAO_CONVERGED_MINF;
1928a7e14dcfSSatish Balay   } else if (gnorm<= gatol && cnorm <=catol) {
19299566063dSJacob Faibussowitsch     PetscCall(PetscInfo(tao,"Converged due to residual norm ||g(X)||=%g < %g\n",(double)gnorm,(double)gatol));
1930a7e14dcfSSatish Balay     reason = TAO_CONVERGED_GATOL;
1931a7e14dcfSSatish Balay   } else if (f!=0 && PetscAbsReal(gnorm/f) <= grtol && cnorm <= crtol) {
19329566063dSJacob Faibussowitsch     PetscCall(PetscInfo(tao,"Converged due to residual ||g(X)||/|f(X)| =%g < %g\n",(double)(gnorm/f),(double)grtol));
1933a7e14dcfSSatish Balay     reason = TAO_CONVERGED_GRTOL;
1934e1d16bb9SBarry Smith   } else if (gnorm0 != 0 && ((gttol == 0 && gnorm == 0) || gnorm/gnorm0 < gttol) && cnorm <= crtol) {
19359566063dSJacob Faibussowitsch     PetscCall(PetscInfo(tao,"Converged due to relative residual norm ||g(X)||/||g(X0)|| = %g < %g\n",(double)(gnorm/gnorm0),(double)gttol));
1936a7e14dcfSSatish Balay     reason = TAO_CONVERGED_GTTOL;
193794511df7SStefano Zampini   } else if (max_funcs >=0 && nfuncs > max_funcs) {
19389566063dSJacob Faibussowitsch     PetscCall(PetscInfo(tao,"Exceeded maximum number of function evaluations: %" PetscInt_FMT " > %" PetscInt_FMT "\n", nfuncs,max_funcs));
1939a7e14dcfSSatish Balay     reason = TAO_DIVERGED_MAXFCN;
1940a7e14dcfSSatish Balay   } else if (tao->lsflag != 0) {
19419566063dSJacob Faibussowitsch     PetscCall(PetscInfo(tao,"Tao Line Search failure.\n"));
1942a7e14dcfSSatish Balay     reason = TAO_DIVERGED_LS_FAILURE;
1943a7e14dcfSSatish Balay   } else if (trradius < steptol && niter > 0) {
19449566063dSJacob Faibussowitsch     PetscCall(PetscInfo(tao,"Trust region/step size too small: %g < %g\n", (double)trradius,(double)steptol));
1945a7e14dcfSSatish Balay     reason = TAO_CONVERGED_STEPTOL;
1946e031d6f5SAlp Dener   } else if (niter >= tao->max_it) {
19479566063dSJacob Faibussowitsch     PetscCall(PetscInfo(tao,"Exceeded maximum number of iterations: %D > %D\n",niter,tao->max_it));
1948a7e14dcfSSatish Balay     reason = TAO_DIVERGED_MAXITS;
1949a7e14dcfSSatish Balay   } else {
1950a7e14dcfSSatish Balay     reason = TAO_CONTINUE_ITERATING;
1951a7e14dcfSSatish Balay   }
1952a7e14dcfSSatish Balay   tao->reason = reason;
1953a7e14dcfSSatish Balay   PetscFunctionReturn(0);
1954a7e14dcfSSatish Balay }
1955a7e14dcfSSatish Balay 
1956a7e14dcfSSatish Balay /*@C
1957a7e14dcfSSatish Balay    TaoSetOptionsPrefix - Sets the prefix used for searching for all
1958a7e14dcfSSatish Balay    TAO options in the database.
1959a7e14dcfSSatish Balay 
1960441846f8SBarry Smith    Logically Collective on Tao
1961a7e14dcfSSatish Balay 
1962a7e14dcfSSatish Balay    Input Parameters:
1963441846f8SBarry Smith +  tao - the Tao context
1964a7e14dcfSSatish Balay -  prefix - the prefix string to prepend to all TAO option requests
1965a7e14dcfSSatish Balay 
1966a7e14dcfSSatish Balay    Notes:
1967a7e14dcfSSatish Balay    A hyphen (-) must NOT be given at the beginning of the prefix name.
1968a7e14dcfSSatish Balay    The first character of all runtime options is AUTOMATICALLY the hyphen.
1969a7e14dcfSSatish Balay 
1970a7e14dcfSSatish Balay    For example, to distinguish between the runtime options for two
1971a7e14dcfSSatish Balay    different TAO solvers, one could call
1972a7e14dcfSSatish Balay .vb
1973a7e14dcfSSatish Balay       TaoSetOptionsPrefix(tao1,"sys1_")
1974a7e14dcfSSatish Balay       TaoSetOptionsPrefix(tao2,"sys2_")
1975a7e14dcfSSatish Balay .ve
1976a7e14dcfSSatish Balay 
1977a7e14dcfSSatish Balay    This would enable use of different options for each system, such as
1978a7e14dcfSSatish Balay .vb
19799fa2b5dcSStefano Zampini       -sys1_tao_method blmvm -sys1_tao_grtol 1.e-3
19809fa2b5dcSStefano Zampini       -sys2_tao_method lmvm  -sys2_tao_grtol 1.e-4
1981a7e14dcfSSatish Balay .ve
1982a7e14dcfSSatish Balay 
1983a7e14dcfSSatish Balay    Level: advanced
1984a7e14dcfSSatish Balay 
1985a7e14dcfSSatish Balay .seealso: TaoAppendOptionsPrefix(), TaoGetOptionsPrefix()
1986a7e14dcfSSatish Balay @*/
1987a7e14dcfSSatish Balay 
1988441846f8SBarry Smith PetscErrorCode TaoSetOptionsPrefix(Tao tao, const char p[])
1989a7e14dcfSSatish Balay {
1990a7e14dcfSSatish Balay   PetscFunctionBegin;
199194511df7SStefano Zampini   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
19929566063dSJacob Faibussowitsch   PetscCall(PetscObjectSetOptionsPrefix((PetscObject)tao,p));
1993a7e14dcfSSatish Balay   if (tao->linesearch) {
19949566063dSJacob Faibussowitsch     PetscCall(TaoLineSearchSetOptionsPrefix(tao->linesearch,p));
1995a7e14dcfSSatish Balay   }
1996a7e14dcfSSatish Balay   if (tao->ksp) {
19979566063dSJacob Faibussowitsch     PetscCall(KSPSetOptionsPrefix(tao->ksp,p));
1998a7e14dcfSSatish Balay   }
1999a7e14dcfSSatish Balay   PetscFunctionReturn(0);
2000a7e14dcfSSatish Balay }
2001a7e14dcfSSatish Balay 
2002a7e14dcfSSatish Balay /*@C
2003a7e14dcfSSatish Balay    TaoAppendOptionsPrefix - Appends to the prefix used for searching for all
2004a7e14dcfSSatish Balay    TAO options in the database.
2005a7e14dcfSSatish Balay 
2006441846f8SBarry Smith    Logically Collective on Tao
2007a7e14dcfSSatish Balay 
2008a7e14dcfSSatish Balay    Input Parameters:
2009441846f8SBarry Smith +  tao - the Tao solver context
2010a7e14dcfSSatish Balay -  prefix - the prefix string to prepend to all TAO option requests
2011a7e14dcfSSatish Balay 
2012a7e14dcfSSatish Balay    Notes:
2013a7e14dcfSSatish Balay    A hyphen (-) must NOT be given at the beginning of the prefix name.
2014a7e14dcfSSatish Balay    The first character of all runtime options is AUTOMATICALLY the hyphen.
2015a7e14dcfSSatish Balay 
2016a7e14dcfSSatish Balay    Level: advanced
2017a7e14dcfSSatish Balay 
2018a7e14dcfSSatish Balay .seealso: TaoSetOptionsPrefix(), TaoGetOptionsPrefix()
2019a7e14dcfSSatish Balay @*/
2020441846f8SBarry Smith PetscErrorCode TaoAppendOptionsPrefix(Tao tao, const char p[])
2021a7e14dcfSSatish Balay {
2022a7e14dcfSSatish Balay   PetscFunctionBegin;
202394511df7SStefano Zampini   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
20249566063dSJacob Faibussowitsch   PetscCall(PetscObjectAppendOptionsPrefix((PetscObject)tao,p));
2025a7e14dcfSSatish Balay   if (tao->linesearch) {
20269566063dSJacob Faibussowitsch     PetscCall(PetscObjectAppendOptionsPrefix((PetscObject)tao->linesearch,p));
2027a7e14dcfSSatish Balay   }
2028a7e14dcfSSatish Balay   if (tao->ksp) {
20299566063dSJacob Faibussowitsch     PetscCall(KSPAppendOptionsPrefix(tao->ksp,p));
2030a7e14dcfSSatish Balay   }
2031a7e14dcfSSatish Balay   PetscFunctionReturn(0);
2032a7e14dcfSSatish Balay }
2033a7e14dcfSSatish Balay 
2034a7e14dcfSSatish Balay /*@C
2035a7e14dcfSSatish Balay   TaoGetOptionsPrefix - Gets the prefix used for searching for all
2036a7e14dcfSSatish Balay   TAO options in the database
2037a7e14dcfSSatish Balay 
2038a7e14dcfSSatish Balay   Not Collective
2039a7e14dcfSSatish Balay 
2040a7e14dcfSSatish Balay   Input Parameters:
2041441846f8SBarry Smith . tao - the Tao context
2042a7e14dcfSSatish Balay 
2043a7e14dcfSSatish Balay   Output Parameters:
2044a7e14dcfSSatish Balay . prefix - pointer to the prefix string used is returned
2045a7e14dcfSSatish Balay 
204695452b02SPatrick Sanan   Notes:
204795452b02SPatrick Sanan     On the fortran side, the user should pass in a string 'prefix' of
2048a7e14dcfSSatish Balay   sufficient length to hold the prefix.
2049a7e14dcfSSatish Balay 
2050a7e14dcfSSatish Balay   Level: advanced
2051a7e14dcfSSatish Balay 
2052a7e14dcfSSatish Balay .seealso: TaoSetOptionsPrefix(), TaoAppendOptionsPrefix()
2053a7e14dcfSSatish Balay @*/
2054441846f8SBarry Smith PetscErrorCode TaoGetOptionsPrefix(Tao tao, const char *p[])
2055a7e14dcfSSatish Balay {
205694511df7SStefano Zampini   PetscFunctionBegin;
205794511df7SStefano Zampini   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
20589566063dSJacob Faibussowitsch   PetscCall(PetscObjectGetOptionsPrefix((PetscObject)tao,p));
205994511df7SStefano Zampini   PetscFunctionReturn(0);
2060a7e14dcfSSatish Balay }
2061a7e14dcfSSatish Balay 
2062a7e14dcfSSatish Balay /*@C
2063a7e14dcfSSatish Balay    TaoSetType - Sets the method for the unconstrained minimization solver.
2064a7e14dcfSSatish Balay 
2065441846f8SBarry Smith    Collective on Tao
2066a7e14dcfSSatish Balay 
2067a7e14dcfSSatish Balay    Input Parameters:
2068441846f8SBarry Smith +  solver - the Tao solver context
2069a7e14dcfSSatish Balay -  type - a known method
2070a7e14dcfSSatish Balay 
2071a7e14dcfSSatish Balay    Options Database Key:
207258417fe7SBarry Smith .  -tao_type <type> - Sets the method; use -help for a list
207358417fe7SBarry Smith    of available methods (for instance, "-tao_type lmvm" or "-tao_type tron")
2074a7e14dcfSSatish Balay 
2075a7e14dcfSSatish Balay    Available methods include:
207658417fe7SBarry Smith +    nls - Newton's method with line search for unconstrained minimization
207758417fe7SBarry Smith .    ntr - Newton's method with trust region for unconstrained minimization
207858417fe7SBarry Smith .    ntl - Newton's method with trust region, line search for unconstrained minimization
207958417fe7SBarry Smith .    lmvm - Limited memory variable metric method for unconstrained minimization
208058417fe7SBarry Smith .    cg - Nonlinear conjugate gradient method for unconstrained minimization
208158417fe7SBarry Smith .    nm - Nelder-Mead algorithm for derivate-free unconstrained minimization
208258417fe7SBarry Smith .    tron - Newton Trust Region method for bound constrained minimization
208358417fe7SBarry Smith .    gpcg - Newton Trust Region method for quadratic bound constrained minimization
208458417fe7SBarry Smith .    blmvm - Limited memory variable metric method for bound constrained minimization
20851eb8069cSJason Sarich -    pounders - Model-based algorithm pounder extended for nonlinear least squares
2086a7e14dcfSSatish Balay 
2087a7e14dcfSSatish Balay   Level: intermediate
2088a7e14dcfSSatish Balay 
20891eb8069cSJason Sarich .seealso: TaoCreate(), TaoGetType(), TaoType
2090a7e14dcfSSatish Balay 
2091a7e14dcfSSatish Balay @*/
2092b625d6c7SJed Brown PetscErrorCode TaoSetType(Tao tao, TaoType type)
2093a7e14dcfSSatish Balay {
2094441846f8SBarry Smith   PetscErrorCode (*create_xxx)(Tao);
2095a7e14dcfSSatish Balay   PetscBool      issame;
2096a7e14dcfSSatish Balay 
2097a7e14dcfSSatish Balay   PetscFunctionBegin;
2098441846f8SBarry Smith   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
2099a7e14dcfSSatish Balay 
21009566063dSJacob Faibussowitsch   PetscCall(PetscObjectTypeCompare((PetscObject)tao,type,&issame));
2101a7e14dcfSSatish Balay   if (issame) PetscFunctionReturn(0);
2102a7e14dcfSSatish Balay 
21039566063dSJacob Faibussowitsch   PetscCall(PetscFunctionListFind(TaoList, type, (void(**)(void))&create_xxx));
21043c859ba3SBarry Smith   PetscCheck(create_xxx,PetscObjectComm((PetscObject)tao),PETSC_ERR_ARG_UNKNOWN_TYPE,"Unable to find requested Tao type %s",type);
2105a7e14dcfSSatish Balay 
2106a7e14dcfSSatish Balay   /* Destroy the existing solver information */
2107a7e14dcfSSatish Balay   if (tao->ops->destroy) {
21089566063dSJacob Faibussowitsch     PetscCall((*tao->ops->destroy)(tao));
2109a7e14dcfSSatish Balay   }
21109566063dSJacob Faibussowitsch   PetscCall(KSPDestroy(&tao->ksp));
21119566063dSJacob Faibussowitsch   PetscCall(TaoLineSearchDestroy(&tao->linesearch));
21129566063dSJacob Faibussowitsch   PetscCall(VecDestroy(&tao->gradient));
21139566063dSJacob Faibussowitsch   PetscCall(VecDestroy(&tao->stepdirection));
2114a7e14dcfSSatish Balay 
211583c8fe1dSLisandro Dalcin   tao->ops->setup = NULL;
211683c8fe1dSLisandro Dalcin   tao->ops->solve = NULL;
211783c8fe1dSLisandro Dalcin   tao->ops->view  = NULL;
211883c8fe1dSLisandro Dalcin   tao->ops->setfromoptions = NULL;
211983c8fe1dSLisandro Dalcin   tao->ops->destroy = NULL;
2120a7e14dcfSSatish Balay 
2121a7e14dcfSSatish Balay   tao->setupcalled = PETSC_FALSE;
2122a7e14dcfSSatish Balay 
21239566063dSJacob Faibussowitsch   PetscCall((*create_xxx)(tao));
21249566063dSJacob Faibussowitsch   PetscCall(PetscObjectChangeTypeName((PetscObject)tao,type));
2125a7e14dcfSSatish Balay   PetscFunctionReturn(0);
2126a7e14dcfSSatish Balay }
212747a47007SBarry Smith 
2128a7e14dcfSSatish Balay /*MC
2129441846f8SBarry Smith    TaoRegister - Adds a method to the TAO package for unconstrained minimization.
2130a7e14dcfSSatish Balay 
2131a7e14dcfSSatish Balay    Synopsis:
21325bd1e576SStefano Zampini    TaoRegister(char *name_solver,char *path,char *name_Create,PetscErrorCode (*routine_Create)(Tao))
2133a7e14dcfSSatish Balay 
2134a7e14dcfSSatish Balay    Not collective
2135a7e14dcfSSatish Balay 
2136a7e14dcfSSatish Balay    Input Parameters:
2137a7e14dcfSSatish Balay +  sname - name of a new user-defined solver
2138a7e14dcfSSatish Balay -  func - routine to Create method context
2139a7e14dcfSSatish Balay 
2140a7e14dcfSSatish Balay    Notes:
2141441846f8SBarry Smith    TaoRegister() may be called multiple times to add several user-defined solvers.
2142a7e14dcfSSatish Balay 
2143a7e14dcfSSatish Balay    Sample usage:
2144a7e14dcfSSatish Balay .vb
2145441846f8SBarry Smith    TaoRegister("my_solver",MySolverCreate);
2146a7e14dcfSSatish Balay .ve
2147a7e14dcfSSatish Balay 
2148a7e14dcfSSatish Balay    Then, your solver can be chosen with the procedural interface via
2149a7e14dcfSSatish Balay $     TaoSetType(tao,"my_solver")
2150a7e14dcfSSatish Balay    or at runtime via the option
215158417fe7SBarry Smith $     -tao_type my_solver
2152a7e14dcfSSatish Balay 
2153a7e14dcfSSatish Balay    Level: advanced
2154a7e14dcfSSatish Balay 
2155441846f8SBarry Smith .seealso: TaoRegisterAll(), TaoRegisterDestroy()
2156a7e14dcfSSatish Balay M*/
2157441846f8SBarry Smith PetscErrorCode TaoRegister(const char sname[], PetscErrorCode (*func)(Tao))
2158a7e14dcfSSatish Balay {
2159a7e14dcfSSatish Balay   PetscFunctionBegin;
21609566063dSJacob Faibussowitsch   PetscCall(TaoInitializePackage());
21619566063dSJacob Faibussowitsch   PetscCall(PetscFunctionListAdd(&TaoList,sname,(void (*)(void))func));
2162a7e14dcfSSatish Balay   PetscFunctionReturn(0);
2163a7e14dcfSSatish Balay }
2164a7e14dcfSSatish Balay 
2165a7e14dcfSSatish Balay /*@C
2166441846f8SBarry Smith    TaoRegisterDestroy - Frees the list of minimization solvers that were
2167441846f8SBarry Smith    registered by TaoRegisterDynamic().
2168a7e14dcfSSatish Balay 
2169a7e14dcfSSatish Balay    Not Collective
2170a7e14dcfSSatish Balay 
2171a7e14dcfSSatish Balay    Level: advanced
2172a7e14dcfSSatish Balay 
2173441846f8SBarry Smith .seealso: TaoRegisterAll(), TaoRegister()
2174a7e14dcfSSatish Balay @*/
2175441846f8SBarry Smith PetscErrorCode TaoRegisterDestroy(void)
2176a7e14dcfSSatish Balay {
2177a7e14dcfSSatish Balay   PetscFunctionBegin;
21789566063dSJacob Faibussowitsch   PetscCall(PetscFunctionListDestroy(&TaoList));
2179441846f8SBarry Smith   TaoRegisterAllCalled = PETSC_FALSE;
2180a7e14dcfSSatish Balay   PetscFunctionReturn(0);
2181a7e14dcfSSatish Balay }
218247a47007SBarry Smith 
21838931d482SJason Sarich /*@
21848931d482SJason Sarich    TaoGetIterationNumber - Gets the number of Tao iterations completed
21858931d482SJason Sarich    at this time.
21868931d482SJason Sarich 
21878931d482SJason Sarich    Not Collective
21888931d482SJason Sarich 
21898931d482SJason Sarich    Input Parameter:
21908931d482SJason Sarich .  tao - Tao context
21918931d482SJason Sarich 
21928931d482SJason Sarich    Output Parameter:
21938931d482SJason Sarich .  iter - iteration number
21948931d482SJason Sarich 
21958931d482SJason Sarich    Notes:
21968931d482SJason Sarich    For example, during the computation of iteration 2 this would return 1.
21978931d482SJason Sarich 
21988931d482SJason Sarich    Level: intermediate
21998931d482SJason Sarich 
220079f5d8caSBarry Smith .seealso:   TaoGetLinearSolveIterations(), TaoGetResidualNorm(), TaoGetObjective()
22018931d482SJason Sarich @*/
22028931d482SJason Sarich PetscErrorCode  TaoGetIterationNumber(Tao tao,PetscInt *iter)
22038931d482SJason Sarich {
22048931d482SJason Sarich   PetscFunctionBegin;
22058931d482SJason Sarich   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
22068931d482SJason Sarich   PetscValidIntPointer(iter,2);
22078931d482SJason Sarich   *iter = tao->niter;
22088931d482SJason Sarich   PetscFunctionReturn(0);
22098931d482SJason Sarich }
22108931d482SJason Sarich 
22118931d482SJason Sarich /*@
221279f5d8caSBarry Smith    TaoGetResidualNorm - Gets the current value of the norm of the residual
221379f5d8caSBarry Smith    at this time.
221479f5d8caSBarry Smith 
221579f5d8caSBarry Smith    Not Collective
221679f5d8caSBarry Smith 
221779f5d8caSBarry Smith    Input Parameter:
221879f5d8caSBarry Smith .  tao - Tao context
221979f5d8caSBarry Smith 
222079f5d8caSBarry Smith    Output Parameter:
222179f5d8caSBarry Smith .  value - the current value
222279f5d8caSBarry Smith 
222379f5d8caSBarry Smith    Level: intermediate
222479f5d8caSBarry Smith 
222579f5d8caSBarry Smith    Developer Note: This is the 2-norm of the residual, we cannot use TaoGetGradientNorm() because that has
222679f5d8caSBarry Smith                    a different meaning. For some reason Tao sometimes calls the gradient the residual.
222779f5d8caSBarry Smith 
222879f5d8caSBarry Smith .seealso:   TaoGetLinearSolveIterations(), TaoGetIterationNumber(), TaoGetObjective()
222979f5d8caSBarry Smith @*/
223079f5d8caSBarry Smith PetscErrorCode TaoGetResidualNorm(Tao tao,PetscReal *value)
223179f5d8caSBarry Smith {
223279f5d8caSBarry Smith   PetscFunctionBegin;
223379f5d8caSBarry Smith   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
223479f5d8caSBarry Smith   PetscValidRealPointer(value,2);
223579f5d8caSBarry Smith   *value = tao->residual;
223679f5d8caSBarry Smith   PetscFunctionReturn(0);
223779f5d8caSBarry Smith }
223879f5d8caSBarry Smith 
223979f5d8caSBarry Smith /*@
22408931d482SJason Sarich    TaoSetIterationNumber - Sets the current iteration number.
22418931d482SJason Sarich 
224294511df7SStefano Zampini    Logically Collective on Tao
22438931d482SJason Sarich 
2244d8d19677SJose E. Roman    Input Parameters:
2245a2b725a8SWilliam Gropp +  tao - Tao context
2246a2b725a8SWilliam Gropp -  iter - iteration number
22478931d482SJason Sarich 
22488931d482SJason Sarich    Level: developer
22498931d482SJason Sarich 
22508931d482SJason Sarich .seealso:   TaoGetLinearSolveIterations()
22518931d482SJason Sarich @*/
22528931d482SJason Sarich PetscErrorCode  TaoSetIterationNumber(Tao tao,PetscInt iter)
22538931d482SJason Sarich {
22548931d482SJason Sarich   PetscFunctionBegin;
22558931d482SJason Sarich   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
225694511df7SStefano Zampini   PetscValidLogicalCollectiveInt(tao,iter,2);
22579566063dSJacob Faibussowitsch   PetscCall(PetscObjectSAWsTakeAccess((PetscObject)tao));
22588931d482SJason Sarich   tao->niter = iter;
22599566063dSJacob Faibussowitsch   PetscCall(PetscObjectSAWsGrantAccess((PetscObject)tao));
22608931d482SJason Sarich   PetscFunctionReturn(0);
22618931d482SJason Sarich }
22628931d482SJason Sarich 
22638931d482SJason Sarich /*@
22648931d482SJason Sarich    TaoGetTotalIterationNumber - Gets the total number of Tao iterations
22658931d482SJason Sarich    completed. This number keeps accumulating if multiple solves
22668931d482SJason Sarich    are called with the Tao object.
22678931d482SJason Sarich 
22688931d482SJason Sarich    Not Collective
22698931d482SJason Sarich 
22708931d482SJason Sarich    Input Parameter:
22718931d482SJason Sarich .  tao - Tao context
22728931d482SJason Sarich 
22738931d482SJason Sarich    Output Parameter:
22748931d482SJason Sarich .  iter - iteration number
22758931d482SJason Sarich 
22768931d482SJason Sarich    Notes:
22778931d482SJason Sarich    The total iteration count is updated after each solve, if there is a current
22788931d482SJason Sarich    TaoSolve() in progress then those iterations are not yet counted.
22798931d482SJason Sarich 
22808931d482SJason Sarich    Level: intermediate
22818931d482SJason Sarich 
22828931d482SJason Sarich .seealso:   TaoGetLinearSolveIterations()
22838931d482SJason Sarich @*/
22848931d482SJason Sarich PetscErrorCode  TaoGetTotalIterationNumber(Tao tao,PetscInt *iter)
22858931d482SJason Sarich {
22868931d482SJason Sarich   PetscFunctionBegin;
22878931d482SJason Sarich   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
22888931d482SJason Sarich   PetscValidIntPointer(iter,2);
22898931d482SJason Sarich   *iter = tao->ntotalits;
22908931d482SJason Sarich   PetscFunctionReturn(0);
22918931d482SJason Sarich }
22928931d482SJason Sarich 
22938931d482SJason Sarich /*@
22948931d482SJason Sarich    TaoSetTotalIterationNumber - Sets the current total iteration number.
22958931d482SJason Sarich 
229694511df7SStefano Zampini    Logically Collective on Tao
22978931d482SJason Sarich 
2298d8d19677SJose E. Roman    Input Parameters:
2299a2b725a8SWilliam Gropp +  tao - Tao context
2300a2b725a8SWilliam Gropp -  iter - iteration number
23018931d482SJason Sarich 
23028931d482SJason Sarich    Level: developer
23038931d482SJason Sarich 
23048931d482SJason Sarich .seealso:   TaoGetLinearSolveIterations()
23058931d482SJason Sarich @*/
23068931d482SJason Sarich PetscErrorCode  TaoSetTotalIterationNumber(Tao tao,PetscInt iter)
23078931d482SJason Sarich {
23088931d482SJason Sarich   PetscFunctionBegin;
23098931d482SJason Sarich   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
231094511df7SStefano Zampini   PetscValidLogicalCollectiveInt(tao,iter,2);
23119566063dSJacob Faibussowitsch   PetscCall(PetscObjectSAWsTakeAccess((PetscObject)tao));
23128931d482SJason Sarich   tao->ntotalits = iter;
23139566063dSJacob Faibussowitsch   PetscCall(PetscObjectSAWsGrantAccess((PetscObject)tao));
23148931d482SJason Sarich   PetscFunctionReturn(0);
23158931d482SJason Sarich }
23168931d482SJason Sarich 
2317a7e14dcfSSatish Balay /*@
2318e4cb33bbSBarry Smith   TaoSetConvergedReason - Sets the termination flag on a Tao object
2319a7e14dcfSSatish Balay 
2320441846f8SBarry Smith   Logically Collective on Tao
2321a7e14dcfSSatish Balay 
2322a7e14dcfSSatish Balay   Input Parameters:
2323441846f8SBarry Smith + tao - the Tao context
2324a7e14dcfSSatish Balay - reason - one of
2325a7e14dcfSSatish Balay $     TAO_CONVERGED_ATOL (2),
2326a7e14dcfSSatish Balay $     TAO_CONVERGED_RTOL (3),
2327a7e14dcfSSatish Balay $     TAO_CONVERGED_STEPTOL (4),
2328a7e14dcfSSatish Balay $     TAO_CONVERGED_MINF (5),
2329a7e14dcfSSatish Balay $     TAO_CONVERGED_USER (6),
2330a7e14dcfSSatish Balay $     TAO_DIVERGED_MAXITS (-2),
2331a7e14dcfSSatish Balay $     TAO_DIVERGED_NAN (-4),
2332a7e14dcfSSatish Balay $     TAO_DIVERGED_MAXFCN (-5),
2333a7e14dcfSSatish Balay $     TAO_DIVERGED_LS_FAILURE (-6),
2334a7e14dcfSSatish Balay $     TAO_DIVERGED_TR_REDUCTION (-7),
2335a7e14dcfSSatish Balay $     TAO_DIVERGED_USER (-8),
2336a7e14dcfSSatish Balay $     TAO_CONTINUE_ITERATING (0)
2337a7e14dcfSSatish Balay 
2338a7e14dcfSSatish Balay    Level: intermediate
2339a7e14dcfSSatish Balay 
2340a7e14dcfSSatish Balay @*/
2341e4cb33bbSBarry Smith PetscErrorCode TaoSetConvergedReason(Tao tao, TaoConvergedReason reason)
2342a7e14dcfSSatish Balay {
2343a7e14dcfSSatish Balay   PetscFunctionBegin;
234494511df7SStefano Zampini   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
234594511df7SStefano Zampini   PetscValidLogicalCollectiveEnum(tao,reason,2);
2346a7e14dcfSSatish Balay   tao->reason = reason;
2347a7e14dcfSSatish Balay   PetscFunctionReturn(0);
2348a7e14dcfSSatish Balay }
2349a7e14dcfSSatish Balay 
2350a7e14dcfSSatish Balay /*@
2351e4cb33bbSBarry Smith    TaoGetConvergedReason - Gets the reason the Tao iteration was stopped.
2352a7e14dcfSSatish Balay 
2353a7e14dcfSSatish Balay    Not Collective
2354a7e14dcfSSatish Balay 
2355a7e14dcfSSatish Balay    Input Parameter:
2356441846f8SBarry Smith .  tao - the Tao solver context
2357a7e14dcfSSatish Balay 
2358a7e14dcfSSatish Balay    Output Parameter:
2359a7e14dcfSSatish Balay .  reason - one of
2360a7e14dcfSSatish Balay $  TAO_CONVERGED_GATOL (3)           ||g(X)|| < gatol
2361a7e14dcfSSatish Balay $  TAO_CONVERGED_GRTOL (4)           ||g(X)|| / f(X)  < grtol
2362a7e14dcfSSatish Balay $  TAO_CONVERGED_GTTOL (5)           ||g(X)|| / ||g(X0)|| < gttol
2363a7e14dcfSSatish Balay $  TAO_CONVERGED_STEPTOL (6)         step size small
2364a7e14dcfSSatish Balay $  TAO_CONVERGED_MINF (7)            F < F_min
2365a7e14dcfSSatish Balay $  TAO_CONVERGED_USER (8)            User defined
2366a7e14dcfSSatish Balay $  TAO_DIVERGED_MAXITS (-2)          its > maxits
2367a7e14dcfSSatish Balay $  TAO_DIVERGED_NAN (-4)             Numerical problems
2368a7e14dcfSSatish Balay $  TAO_DIVERGED_MAXFCN (-5)          fevals > max_funcsals
2369a7e14dcfSSatish Balay $  TAO_DIVERGED_LS_FAILURE (-6)      line search failure
2370a7e14dcfSSatish Balay $  TAO_DIVERGED_TR_REDUCTION (-7)    trust region failure
2371a7e14dcfSSatish Balay $  TAO_DIVERGED_USER(-8)             (user defined)
2372a7e14dcfSSatish Balay $  TAO_CONTINUE_ITERATING (0)
2373a7e14dcfSSatish Balay 
2374a7e14dcfSSatish Balay    where
2375a7e14dcfSSatish Balay +  X - current solution
2376a7e14dcfSSatish Balay .  X0 - initial guess
2377a7e14dcfSSatish Balay .  f(X) - current function value
2378a7e14dcfSSatish Balay .  f(X*) - true solution (estimated)
2379a7e14dcfSSatish Balay .  g(X) - current gradient
2380a7e14dcfSSatish Balay .  its - current iterate number
2381a7e14dcfSSatish Balay .  maxits - maximum number of iterates
2382a7e14dcfSSatish Balay .  fevals - number of function evaluations
2383a7e14dcfSSatish Balay -  max_funcsals - maximum number of function evaluations
2384a7e14dcfSSatish Balay 
2385a7e14dcfSSatish Balay    Level: intermediate
2386a7e14dcfSSatish Balay 
2387a7e14dcfSSatish Balay .seealso: TaoSetConvergenceTest(), TaoSetTolerances()
2388a7e14dcfSSatish Balay 
2389a7e14dcfSSatish Balay @*/
2390e4cb33bbSBarry Smith PetscErrorCode TaoGetConvergedReason(Tao tao, TaoConvergedReason *reason)
2391a7e14dcfSSatish Balay {
2392a7e14dcfSSatish Balay   PetscFunctionBegin;
2393441846f8SBarry Smith   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
2394a7e14dcfSSatish Balay   PetscValidPointer(reason,2);
2395a7e14dcfSSatish Balay   *reason = tao->reason;
2396a7e14dcfSSatish Balay   PetscFunctionReturn(0);
2397a7e14dcfSSatish Balay }
2398a7e14dcfSSatish Balay 
2399a7e14dcfSSatish Balay /*@
2400a7e14dcfSSatish Balay    TaoGetSolutionStatus - Get the current iterate, objective value,
2401a7e14dcfSSatish Balay    residual, infeasibility, and termination
2402a7e14dcfSSatish Balay 
2403a7e14dcfSSatish Balay    Not Collective
2404a7e14dcfSSatish Balay 
2405f899ff85SJose E. Roman    Input Parameter:
2406441846f8SBarry Smith .  tao - the Tao context
2407a7e14dcfSSatish Balay 
2408a7e14dcfSSatish Balay    Output Parameters:
2409a7e14dcfSSatish Balay +  iterate - the current iterate number (>=0)
2410a7e14dcfSSatish Balay .  f - the current function value
241158417fe7SBarry Smith .  gnorm - the square of the gradient norm, duality gap, or other measure indicating distance from optimality.
2412a7e14dcfSSatish Balay .  cnorm - the infeasibility of the current solution with regard to the constraints.
2413a7e14dcfSSatish Balay .  xdiff - the step length or trust region radius of the most recent iterate.
2414a7e14dcfSSatish Balay -  reason - The termination reason, which can equal TAO_CONTINUE_ITERATING
2415a7e14dcfSSatish Balay 
2416a7e14dcfSSatish Balay    Level: intermediate
2417a7e14dcfSSatish Balay 
2418a7e14dcfSSatish Balay    Note:
2419a7e14dcfSSatish Balay    TAO returns the values set by the solvers in the routine TaoMonitor().
2420a7e14dcfSSatish Balay 
2421a7e14dcfSSatish Balay    Note:
242247a47007SBarry Smith    If any of the output arguments are set to NULL, no corresponding value will be returned.
2423a7e14dcfSSatish Balay 
2424e4cb33bbSBarry Smith .seealso: TaoMonitor(), TaoGetConvergedReason()
2425a7e14dcfSSatish Balay @*/
2426e4cb33bbSBarry Smith PetscErrorCode TaoGetSolutionStatus(Tao tao, PetscInt *its, PetscReal *f, PetscReal *gnorm, PetscReal *cnorm, PetscReal *xdiff, TaoConvergedReason *reason)
2427a7e14dcfSSatish Balay {
2428a7e14dcfSSatish Balay   PetscFunctionBegin;
242994511df7SStefano Zampini   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
2430a7e14dcfSSatish Balay   if (its) *its = tao->niter;
2431a7e14dcfSSatish Balay   if (f) *f = tao->fc;
2432a7e14dcfSSatish Balay   if (gnorm) *gnorm = tao->residual;
2433a7e14dcfSSatish Balay   if (cnorm) *cnorm = tao->cnorm;
2434a7e14dcfSSatish Balay   if (reason) *reason = tao->reason;
2435a7e14dcfSSatish Balay   if (xdiff) *xdiff = tao->step;
2436a7e14dcfSSatish Balay   PetscFunctionReturn(0);
2437a7e14dcfSSatish Balay }
2438a7e14dcfSSatish Balay 
2439a7e14dcfSSatish Balay /*@C
2440441846f8SBarry Smith    TaoGetType - Gets the current Tao algorithm.
2441a7e14dcfSSatish Balay 
2442a7e14dcfSSatish Balay    Not Collective
2443a7e14dcfSSatish Balay 
2444a7e14dcfSSatish Balay    Input Parameter:
2445441846f8SBarry Smith .  tao - the Tao solver context
2446a7e14dcfSSatish Balay 
2447a7e14dcfSSatish Balay    Output Parameter:
2448441846f8SBarry Smith .  type - Tao method
2449a7e14dcfSSatish Balay 
2450a7e14dcfSSatish Balay    Level: intermediate
2451a7e14dcfSSatish Balay 
2452a7e14dcfSSatish Balay @*/
2453b625d6c7SJed Brown PetscErrorCode TaoGetType(Tao tao,TaoType *type)
2454a7e14dcfSSatish Balay {
2455a7e14dcfSSatish Balay   PetscFunctionBegin;
2456441846f8SBarry Smith   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
2457a7e14dcfSSatish Balay   PetscValidPointer(type,2);
2458a7e14dcfSSatish Balay   *type = ((PetscObject)tao)->type_name;
2459a7e14dcfSSatish Balay   PetscFunctionReturn(0);
2460a7e14dcfSSatish Balay }
2461a7e14dcfSSatish Balay 
2462a7e14dcfSSatish Balay /*@C
2463a7e14dcfSSatish Balay   TaoMonitor - Monitor the solver and the current solution.  This
2464a7e14dcfSSatish Balay   routine will record the iteration number and residual statistics,
2465a7e14dcfSSatish Balay   call any monitors specified by the user, and calls the convergence-check routine.
2466a7e14dcfSSatish Balay 
2467a7e14dcfSSatish Balay    Input Parameters:
2468441846f8SBarry Smith +  tao - the Tao context
2469a7e14dcfSSatish Balay .  its - the current iterate number (>=0)
2470a7e14dcfSSatish Balay .  f - the current objective function value
247158417fe7SBarry Smith .  res - the gradient norm, square root of the duality gap, or other measure indicating distince from optimality.  This measure will be recorded and
2472a7e14dcfSSatish Balay           used for some termination tests.
2473a7e14dcfSSatish Balay .  cnorm - the infeasibility of the current solution with regard to the constraints.
2474a7e14dcfSSatish Balay -  steplength - multiple of the step direction added to the previous iterate.
2475a7e14dcfSSatish Balay 
2476a7e14dcfSSatish Balay    Output Parameters:
2477a7e14dcfSSatish Balay .  reason - The termination reason, which can equal TAO_CONTINUE_ITERATING
2478a7e14dcfSSatish Balay 
2479a7e14dcfSSatish Balay    Options Database Key:
2480a7e14dcfSSatish Balay .  -tao_monitor - Use the default monitor, which prints statistics to standard output
2481a7e14dcfSSatish Balay 
248298ea980cSBarry Smith .seealso TaoGetConvergedReason(), TaoMonitorDefault(), TaoSetMonitor()
2483a7e14dcfSSatish Balay 
2484a7e14dcfSSatish Balay    Level: developer
2485a7e14dcfSSatish Balay 
2486a7e14dcfSSatish Balay @*/
24873ecd9318SAlp Dener PetscErrorCode TaoMonitor(Tao tao, PetscInt its, PetscReal f, PetscReal res, PetscReal cnorm, PetscReal steplength)
2488a7e14dcfSSatish Balay {
248947a47007SBarry Smith   PetscInt       i;
249047a47007SBarry Smith 
2491a7e14dcfSSatish Balay   PetscFunctionBegin;
2492441846f8SBarry Smith   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
2493a7e14dcfSSatish Balay   tao->fc = f;
2494a7e14dcfSSatish Balay   tao->residual = res;
2495a7e14dcfSSatish Balay   tao->cnorm = cnorm;
2496a7e14dcfSSatish Balay   tao->step = steplength;
2497e52336cbSBarry Smith   if (!its) {
249894511df7SStefano Zampini     tao->cnorm0 = cnorm;
249994511df7SStefano Zampini     tao->gnorm0 = res;
2500a7e14dcfSSatish Balay   }
25013c859ba3SBarry Smith   PetscCheck(!PetscIsInfOrNanReal(f) && !PetscIsInfOrNanReal(res),PetscObjectComm((PetscObject)tao),PETSC_ERR_USER, "User provided compute function generated Inf or NaN");
2502a7e14dcfSSatish Balay   for (i=0;i<tao->numbermonitors;i++) {
25039566063dSJacob Faibussowitsch     PetscCall((*tao->monitor[i])(tao,tao->monitorcontext[i]));
2504a7e14dcfSSatish Balay   }
2505a7e14dcfSSatish Balay   PetscFunctionReturn(0);
2506a7e14dcfSSatish Balay }
2507a7e14dcfSSatish Balay 
2508a7e14dcfSSatish Balay /*@
2509ae93cb3cSJason Sarich    TaoSetConvergenceHistory - Sets the array used to hold the convergence history.
2510a7e14dcfSSatish Balay 
2511441846f8SBarry Smith    Logically Collective on Tao
2512a7e14dcfSSatish Balay 
2513a7e14dcfSSatish Balay    Input Parameters:
2514441846f8SBarry Smith +  tao - the Tao solver context
2515a7e14dcfSSatish Balay .  obj   - array to hold objective value history
2516a7e14dcfSSatish Balay .  resid - array to hold residual history
2517a7e14dcfSSatish Balay .  cnorm - array to hold constraint violation history
2518be1558d8SJason Sarich .  lits - integer array holds the number of linear iterations for each Tao iteration
2519a7e14dcfSSatish Balay .  na  - size of obj, resid, and cnorm
2520a7e14dcfSSatish Balay -  reset - PetscTrue indicates each new minimization resets the history counter to zero,
2521a7e14dcfSSatish Balay            else it continues storing new values for new minimizations after the old ones
2522a7e14dcfSSatish Balay 
2523a7e14dcfSSatish Balay    Notes:
2524a7e14dcfSSatish Balay    If set, TAO will fill the given arrays with the indicated
2525ae93cb3cSJason Sarich    information at each iteration.  If 'obj','resid','cnorm','lits' are
2526ae93cb3cSJason Sarich    *all* NULL then space (using size na, or 1000 if na is PETSC_DECIDE or
2527ae93cb3cSJason Sarich    PETSC_DEFAULT) is allocated for the history.
2528be1558d8SJason Sarich    If not all are NULL, then only the non-NULL information categories
2529ae93cb3cSJason Sarich    will be stored, the others will be ignored.
2530ae93cb3cSJason Sarich 
2531ae93cb3cSJason Sarich    Any convergence information after iteration number 'na' will not be stored.
2532a7e14dcfSSatish Balay 
2533a7e14dcfSSatish Balay    This routine is useful, e.g., when running a code for purposes
2534a7e14dcfSSatish Balay    of accurate performance monitoring, when no I/O should be done
2535a7e14dcfSSatish Balay    during the section of code that is being timed.
2536a7e14dcfSSatish Balay 
2537a7e14dcfSSatish Balay    Level: intermediate
2538a7e14dcfSSatish Balay 
2539ae93cb3cSJason Sarich .seealso: TaoGetConvergenceHistory()
2540a7e14dcfSSatish Balay 
2541a7e14dcfSSatish Balay @*/
25421b266c99SBarry Smith PetscErrorCode TaoSetConvergenceHistory(Tao tao, PetscReal obj[], PetscReal resid[], PetscReal cnorm[], PetscInt lits[], PetscInt na,PetscBool reset)
2543a7e14dcfSSatish Balay {
2544a7e14dcfSSatish Balay   PetscFunctionBegin;
2545441846f8SBarry Smith   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
2546064a246eSJacob Faibussowitsch   if (obj) PetscValidRealPointer(obj,2);
2547064a246eSJacob Faibussowitsch   if (resid) PetscValidRealPointer(resid,3);
2548064a246eSJacob Faibussowitsch   if (cnorm) PetscValidRealPointer(cnorm,4);
2549ae93cb3cSJason Sarich   if (lits) PetscValidIntPointer(lits,5);
2550ae93cb3cSJason Sarich 
2551ae93cb3cSJason Sarich   if (na == PETSC_DECIDE || na == PETSC_DEFAULT) na = 1000;
2552ae93cb3cSJason Sarich   if (!obj && !resid && !cnorm && !lits) {
25539566063dSJacob Faibussowitsch     PetscCall(PetscCalloc4(na,&obj,na,&resid,na,&cnorm,na,&lits));
2554ae93cb3cSJason Sarich     tao->hist_malloc = PETSC_TRUE;
2555ae93cb3cSJason Sarich   }
2556ae93cb3cSJason Sarich 
2557a7e14dcfSSatish Balay   tao->hist_obj = obj;
2558a7e14dcfSSatish Balay   tao->hist_resid = resid;
2559a7e14dcfSSatish Balay   tao->hist_cnorm = cnorm;
2560ae93cb3cSJason Sarich   tao->hist_lits = lits;
2561a7e14dcfSSatish Balay   tao->hist_max   = na;
2562a7e14dcfSSatish Balay   tao->hist_reset = reset;
2563ae93cb3cSJason Sarich   tao->hist_len = 0;
2564a7e14dcfSSatish Balay   PetscFunctionReturn(0);
2565a7e14dcfSSatish Balay }
2566a7e14dcfSSatish Balay 
2567a7e14dcfSSatish Balay /*@C
2568ae93cb3cSJason Sarich    TaoGetConvergenceHistory - Gets the arrays used to hold the convergence history.
2569a7e14dcfSSatish Balay 
2570441846f8SBarry Smith    Collective on Tao
2571a7e14dcfSSatish Balay 
2572a7e14dcfSSatish Balay    Input Parameter:
2573441846f8SBarry Smith .  tao - the Tao context
2574a7e14dcfSSatish Balay 
2575a7e14dcfSSatish Balay    Output Parameters:
2576a7e14dcfSSatish Balay +  obj   - array used to hold objective value history
2577a7e14dcfSSatish Balay .  resid - array used to hold residual history
2578a7e14dcfSSatish Balay .  cnorm - array used to hold constraint violation history
2579ae93cb3cSJason Sarich .  lits  - integer array used to hold linear solver iteration count
2580dd25e5eeSStefano Zampini -  nhist  - size of obj, resid, cnorm, and lits
2581a7e14dcfSSatish Balay 
2582a7e14dcfSSatish Balay    Notes:
2583ae93cb3cSJason Sarich     This routine must be preceded by calls to TaoSetConvergenceHistory()
2584ae93cb3cSJason Sarich     and TaoSolve(), otherwise it returns useless information.
2585ae93cb3cSJason Sarich 
2586a7e14dcfSSatish Balay     The calling sequence for this routine in Fortran is
2587ae93cb3cSJason Sarich $   call TaoGetConvergenceHistory(Tao tao, PetscInt nhist, PetscErrorCode ierr)
2588a7e14dcfSSatish Balay 
2589a7e14dcfSSatish Balay    This routine is useful, e.g., when running a code for purposes
2590a7e14dcfSSatish Balay    of accurate performance monitoring, when no I/O should be done
2591a7e14dcfSSatish Balay    during the section of code that is being timed.
2592a7e14dcfSSatish Balay 
2593a7e14dcfSSatish Balay    Level: advanced
2594a7e14dcfSSatish Balay 
2595ae93cb3cSJason Sarich .seealso: TaoSetConvergenceHistory()
2596a7e14dcfSSatish Balay 
2597a7e14dcfSSatish Balay @*/
2598ae93cb3cSJason Sarich PetscErrorCode TaoGetConvergenceHistory(Tao tao, PetscReal **obj, PetscReal **resid, PetscReal **cnorm, PetscInt **lits, PetscInt *nhist)
2599a7e14dcfSSatish Balay {
2600a7e14dcfSSatish Balay   PetscFunctionBegin;
2601441846f8SBarry Smith   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
2602a7e14dcfSSatish Balay   if (obj)   *obj   = tao->hist_obj;
2603a7e14dcfSSatish Balay   if (cnorm) *cnorm = tao->hist_cnorm;
2604a7e14dcfSSatish Balay   if (resid) *resid = tao->hist_resid;
2605a7e14dcfSSatish Balay   if (nhist) *nhist = tao->hist_len;
2606a7e14dcfSSatish Balay   PetscFunctionReturn(0);
2607a7e14dcfSSatish Balay }
2608a7e14dcfSSatish Balay 
2609a7e14dcfSSatish Balay /*@
2610a7e14dcfSSatish Balay    TaoSetApplicationContext - Sets the optional user-defined context for
2611a7e14dcfSSatish Balay    a solver.
2612a7e14dcfSSatish Balay 
2613441846f8SBarry Smith    Logically Collective on Tao
2614a7e14dcfSSatish Balay 
2615a7e14dcfSSatish Balay    Input Parameters:
2616441846f8SBarry Smith +  tao  - the Tao context
2617a7e14dcfSSatish Balay -  usrP - optional user context
2618a7e14dcfSSatish Balay 
2619a7e14dcfSSatish Balay    Level: intermediate
2620a7e14dcfSSatish Balay 
2621a7e14dcfSSatish Balay .seealso: TaoGetApplicationContext(), TaoSetApplicationContext()
2622a7e14dcfSSatish Balay @*/
2623441846f8SBarry Smith PetscErrorCode  TaoSetApplicationContext(Tao tao,void *usrP)
2624a7e14dcfSSatish Balay {
2625a7e14dcfSSatish Balay   PetscFunctionBegin;
2626441846f8SBarry Smith   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
2627a7e14dcfSSatish Balay   tao->user = usrP;
2628a7e14dcfSSatish Balay   PetscFunctionReturn(0);
2629a7e14dcfSSatish Balay }
2630a7e14dcfSSatish Balay 
2631a7e14dcfSSatish Balay /*@
2632a7e14dcfSSatish Balay    TaoGetApplicationContext - Gets the user-defined context for a
2633a7e14dcfSSatish Balay    TAO solvers.
2634a7e14dcfSSatish Balay 
2635a7e14dcfSSatish Balay    Not Collective
2636a7e14dcfSSatish Balay 
2637a7e14dcfSSatish Balay    Input Parameter:
2638441846f8SBarry Smith .  tao  - Tao context
2639a7e14dcfSSatish Balay 
2640a7e14dcfSSatish Balay    Output Parameter:
2641a7e14dcfSSatish Balay .  usrP - user context
2642a7e14dcfSSatish Balay 
2643a7e14dcfSSatish Balay    Level: intermediate
2644a7e14dcfSSatish Balay 
2645a7e14dcfSSatish Balay .seealso: TaoSetApplicationContext()
2646a7e14dcfSSatish Balay @*/
2647441846f8SBarry Smith PetscErrorCode  TaoGetApplicationContext(Tao tao,void *usrP)
2648a7e14dcfSSatish Balay {
2649a7e14dcfSSatish Balay   PetscFunctionBegin;
2650441846f8SBarry Smith   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
265194511df7SStefano Zampini   PetscValidPointer(usrP,2);
2652a7e14dcfSSatish Balay   *(void**)usrP = tao->user;
2653a7e14dcfSSatish Balay   PetscFunctionReturn(0);
2654a7e14dcfSSatish Balay }
2655a9603a14SPatrick Farrell 
2656a9603a14SPatrick Farrell /*@
2657a9603a14SPatrick Farrell    TaoSetGradientNorm - Sets the matrix used to define the inner product that measures the size of the gradient.
2658a9603a14SPatrick Farrell 
2659a9603a14SPatrick Farrell    Collective on tao
2660a9603a14SPatrick Farrell 
2661a9603a14SPatrick Farrell    Input Parameters:
2662a9603a14SPatrick Farrell +  tao  - the Tao context
2663a9603a14SPatrick Farrell -  M    - gradient norm
2664a9603a14SPatrick Farrell 
2665a9603a14SPatrick Farrell    Level: beginner
2666a9603a14SPatrick Farrell 
2667a9603a14SPatrick Farrell .seealso: TaoGetGradientNorm(), TaoGradientNorm()
2668a9603a14SPatrick Farrell @*/
2669a9603a14SPatrick Farrell PetscErrorCode  TaoSetGradientNorm(Tao tao, Mat M)
2670a9603a14SPatrick Farrell {
2671a9603a14SPatrick Farrell   PetscFunctionBegin;
2672a9603a14SPatrick Farrell   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
267394511df7SStefano Zampini   PetscValidHeaderSpecific(M,MAT_CLASSID,2);
26749566063dSJacob Faibussowitsch   PetscCall(PetscObjectReference((PetscObject)M));
26759566063dSJacob Faibussowitsch   PetscCall(MatDestroy(&tao->gradient_norm));
26769566063dSJacob Faibussowitsch   PetscCall(VecDestroy(&tao->gradient_norm_tmp));
2677a9603a14SPatrick Farrell   tao->gradient_norm = M;
26789566063dSJacob Faibussowitsch   PetscCall(MatCreateVecs(M, NULL, &tao->gradient_norm_tmp));
2679a9603a14SPatrick Farrell   PetscFunctionReturn(0);
2680a9603a14SPatrick Farrell }
2681a9603a14SPatrick Farrell 
2682a9603a14SPatrick Farrell /*@
2683a9603a14SPatrick Farrell    TaoGetGradientNorm - Returns the matrix used to define the inner product for measuring the size of the gradient.
2684a9603a14SPatrick Farrell 
2685a9603a14SPatrick Farrell    Not Collective
2686a9603a14SPatrick Farrell 
2687a9603a14SPatrick Farrell    Input Parameter:
2688a9603a14SPatrick Farrell .  tao  - Tao context
2689a9603a14SPatrick Farrell 
2690a9603a14SPatrick Farrell    Output Parameter:
2691a9603a14SPatrick Farrell .  M - gradient norm
2692a9603a14SPatrick Farrell 
2693a9603a14SPatrick Farrell    Level: beginner
2694a9603a14SPatrick Farrell 
2695a9603a14SPatrick Farrell .seealso: TaoSetGradientNorm(), TaoGradientNorm()
2696a9603a14SPatrick Farrell @*/
2697a9603a14SPatrick Farrell PetscErrorCode  TaoGetGradientNorm(Tao tao, Mat *M)
2698a9603a14SPatrick Farrell {
2699a9603a14SPatrick Farrell   PetscFunctionBegin;
2700a9603a14SPatrick Farrell   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
270194511df7SStefano Zampini   PetscValidPointer(M,2);
2702a9603a14SPatrick Farrell   *M = tao->gradient_norm;
2703a9603a14SPatrick Farrell   PetscFunctionReturn(0);
2704a9603a14SPatrick Farrell }
2705a9603a14SPatrick Farrell 
270605aca2cdSSatish Balay /*@C
2707a9603a14SPatrick Farrell    TaoGradientNorm - Compute the norm with respect to the inner product the user has set.
2708a9603a14SPatrick Farrell 
2709a9603a14SPatrick Farrell    Collective on tao
2710a9603a14SPatrick Farrell 
2711d8d19677SJose E. Roman    Input Parameters:
27122653c764SPierre Jolivet +  tao      - the Tao context
2713a9603a14SPatrick Farrell .  gradient - the gradient to be computed
27142653c764SPierre Jolivet -  norm     - the norm type
2715a9603a14SPatrick Farrell 
2716a9603a14SPatrick Farrell    Output Parameter:
2717a9603a14SPatrick Farrell .  gnorm    - the gradient norm
2718a9603a14SPatrick Farrell 
2719a9603a14SPatrick Farrell    Level: developer
2720a9603a14SPatrick Farrell 
2721a9603a14SPatrick Farrell .seealso: TaoSetGradientNorm(), TaoGetGradientNorm()
2722a9603a14SPatrick Farrell @*/
2723a9603a14SPatrick Farrell PetscErrorCode  TaoGradientNorm(Tao tao, Vec gradient, NormType type, PetscReal *gnorm)
2724a9603a14SPatrick Farrell {
2725a9603a14SPatrick Farrell   PetscFunctionBegin;
27268854b543SStefano Zampini   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
27278854b543SStefano Zampini   PetscValidHeaderSpecific(gradient,VEC_CLASSID,2);
27288854b543SStefano Zampini   PetscValidLogicalCollectiveEnum(tao,type,3);
272994511df7SStefano Zampini   PetscValidRealPointer(gnorm,4);
2730a9603a14SPatrick Farrell   if (tao->gradient_norm) {
2731680e2bc7SPatrick Farrell     PetscScalar gnorms;
2732680e2bc7SPatrick Farrell 
27333c859ba3SBarry 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.");
27349566063dSJacob Faibussowitsch     PetscCall(MatMult(tao->gradient_norm, gradient, tao->gradient_norm_tmp));
27359566063dSJacob Faibussowitsch     PetscCall(VecDot(gradient, tao->gradient_norm_tmp, &gnorms));
27367bb79937SPatrick Farrell     *gnorm = PetscRealPart(PetscSqrtScalar(gnorms));
2737a9603a14SPatrick Farrell   } else {
27389566063dSJacob Faibussowitsch     PetscCall(VecNorm(gradient, type, gnorm));
2739a9603a14SPatrick Farrell   }
2740a9603a14SPatrick Farrell   PetscFunctionReturn(0);
2741a9603a14SPatrick Farrell }
2742a9603a14SPatrick Farrell 
2743e882e171SHong Zhang /*@C
2744e882e171SHong Zhang    TaoMonitorDrawCtxCreate - Creates the monitor context for TaoMonitorDrawCtx
2745a9603a14SPatrick Farrell 
2746e882e171SHong Zhang    Collective on Tao
2747e882e171SHong Zhang 
2748e882e171SHong Zhang    Output Patameter:
2749e882e171SHong Zhang .    ctx - the monitor context
2750e882e171SHong Zhang 
2751e882e171SHong Zhang    Options Database:
2752e882e171SHong Zhang .   -tao_draw_solution_initial - show initial guess as well as current solution
2753e882e171SHong Zhang 
2754e882e171SHong Zhang    Level: intermediate
2755e882e171SHong Zhang 
2756e882e171SHong Zhang .seealso: TaoMonitorSet(), TaoMonitorDefault(), VecView(), TaoMonitorDrawCtx()
2757e882e171SHong Zhang @*/
2758e882e171SHong Zhang PetscErrorCode  TaoMonitorDrawCtxCreate(MPI_Comm comm,const char host[],const char label[],int x,int y,int m,int n,PetscInt howoften,TaoMonitorDrawCtx *ctx)
2759e882e171SHong Zhang {
2760e882e171SHong Zhang   PetscFunctionBegin;
27619566063dSJacob Faibussowitsch   PetscCall(PetscNew(ctx));
27629566063dSJacob Faibussowitsch   PetscCall(PetscViewerDrawOpen(comm,host,label,x,y,m,n,&(*ctx)->viewer));
27639566063dSJacob Faibussowitsch   PetscCall(PetscViewerSetFromOptions((*ctx)->viewer));
2764e882e171SHong Zhang   (*ctx)->howoften = howoften;
2765e882e171SHong Zhang   PetscFunctionReturn(0);
2766e882e171SHong Zhang }
2767e882e171SHong Zhang 
2768e882e171SHong Zhang /*@C
2769e882e171SHong Zhang    TaoMonitorDrawCtxDestroy - Destroys the monitor context for TaoMonitorDrawSolution()
2770e882e171SHong Zhang 
2771e882e171SHong Zhang    Collective on Tao
2772e882e171SHong Zhang 
2773e882e171SHong Zhang    Input Parameters:
2774e882e171SHong Zhang .    ctx - the monitor context
2775e882e171SHong Zhang 
2776e882e171SHong Zhang    Level: intermediate
2777e882e171SHong Zhang 
2778e882e171SHong Zhang .seealso: TaoMonitorSet(), TaoMonitorDefault(), VecView(), TaoMonitorDrawSolution()
2779e882e171SHong Zhang @*/
2780e882e171SHong Zhang PetscErrorCode  TaoMonitorDrawCtxDestroy(TaoMonitorDrawCtx *ictx)
2781e882e171SHong Zhang {
2782e882e171SHong Zhang   PetscFunctionBegin;
27839566063dSJacob Faibussowitsch   PetscCall(PetscViewerDestroy(&(*ictx)->viewer));
27849566063dSJacob Faibussowitsch   PetscCall(PetscFree(*ictx));
2785e882e171SHong Zhang   PetscFunctionReturn(0);
2786e882e171SHong Zhang }
2787