xref: /petsc/src/tao/interface/taosolver.c (revision 8ebe3e4e9e00d86ece2e9fcd0cc84910b0ad437c)
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 {
58a7e14dcfSSatish Balay   PetscErrorCode ierr;
59441846f8SBarry Smith   Tao            tao;
60a7e14dcfSSatish Balay 
61a7e14dcfSSatish Balay   PetscFunctionBegin;
62a7e14dcfSSatish Balay   PetscValidPointer(newtao,2);
63a7e14dcfSSatish Balay   ierr = TaoInitializePackage();CHKERRQ(ierr);
64a7e14dcfSSatish Balay   ierr = TaoLineSearchInitializePackage();CHKERRQ(ierr);
6573107ff1SLisandro Dalcin   ierr = PetscHeaderCreate(tao,TAO_CLASSID,"Tao","Optimization solver","Tao",comm,TaoDestroy,TaoView);CHKERRQ(ierr);
66a7e14dcfSSatish Balay 
6794511df7SStefano Zampini   /* Set non-NULL defaults */
6894511df7SStefano Zampini   tao->ops->convergencetest = TaoDefaultConvergenceTest;
69a7e14dcfSSatish Balay 
70a7e14dcfSSatish Balay   tao->max_it    = 10000;
7194511df7SStefano Zampini   tao->max_funcs = -1;
726f4723b1SBarry Smith #if defined(PETSC_USE_REAL_SINGLE)
736f4723b1SBarry Smith   tao->gatol     = 1e-5;
746f4723b1SBarry Smith   tao->grtol     = 1e-5;
756f1f5f59SAlp Dener   tao->crtol     = 1e-5;
766f1f5f59SAlp Dener   tao->catol     = 1e-5;
776f4723b1SBarry Smith #else
78a7e14dcfSSatish Balay   tao->gatol     = 1e-8;
79a7e14dcfSSatish Balay   tao->grtol     = 1e-8;
806f1f5f59SAlp Dener   tao->crtol     = 1e-8;
816f1f5f59SAlp Dener   tao->catol     = 1e-8;
826f4723b1SBarry Smith #endif
836552cf8aSJason Sarich   tao->gttol     = 0.0;
844f1535bcSTodd Munson   tao->steptol   = 0.0;
85e270355aSBarry Smith   tao->trust0    = PETSC_INFINITY;
866f4723b1SBarry Smith   tao->fmin      = PETSC_NINFINITY;
879fa2b5dcSStefano Zampini 
88a7e14dcfSSatish Balay   tao->hist_reset  = PETSC_TRUE;
89a7e14dcfSSatish Balay 
90a7e14dcfSSatish Balay   ierr = TaoResetStatistics(tao);CHKERRQ(ierr);
91a7e14dcfSSatish Balay   *newtao = tao;
92a7e14dcfSSatish Balay   PetscFunctionReturn(0);
93a7e14dcfSSatish Balay }
94a7e14dcfSSatish Balay 
95a7e14dcfSSatish Balay /*@
96a7e14dcfSSatish Balay   TaoSolve - Solves an optimization problem min F(x) s.t. l <= x <= u
97a7e14dcfSSatish Balay 
98441846f8SBarry Smith   Collective on Tao
99a7e14dcfSSatish Balay 
100a7e14dcfSSatish Balay   Input Parameters:
101441846f8SBarry Smith . tao - the Tao context
102a7e14dcfSSatish Balay 
103a7e14dcfSSatish Balay   Notes:
104441846f8SBarry Smith   The user must set up the Tao with calls to TaoSetInitialVector(),
105a7e14dcfSSatish Balay   TaoSetObjectiveRoutine(),
106a7e14dcfSSatish Balay   TaoSetGradientRoutine(), and (if using 2nd order method) TaoSetHessianRoutine().
107a7e14dcfSSatish Balay 
108a35d58b8SBarry Smith   You should call TaoGetConvergedReason() or run with -tao_converged_reason to determine if the optimization algorithm actually succeeded or
109a35d58b8SBarry Smith   why it failed.
110a35d58b8SBarry Smith 
111a7e14dcfSSatish Balay   Level: beginner
112a7e14dcfSSatish Balay 
113a35d58b8SBarry Smith .seealso: TaoCreate(), TaoSetObjectiveRoutine(), TaoSetGradientRoutine(), TaoSetHessianRoutine(), TaoGetConvergedReason()
114a7e14dcfSSatish Balay  @*/
115441846f8SBarry Smith PetscErrorCode TaoSolve(Tao tao)
116a7e14dcfSSatish Balay {
117a7e14dcfSSatish Balay   PetscErrorCode   ierr;
118e2379f4fSBarry Smith   static PetscBool set = PETSC_FALSE;
11947a47007SBarry Smith 
120a7e14dcfSSatish Balay   PetscFunctionBegin;
121441846f8SBarry Smith   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
122e2379f4fSBarry Smith   ierr = PetscCitationsRegister("@TechReport{tao-user-ref,\n"
123e2379f4fSBarry Smith                                 "title   = {Toolkit for Advanced Optimization (TAO) Users Manual},\n"
124e2379f4fSBarry Smith                                 "author  = {Todd Munson and Jason Sarich and Stefan Wild and Steve Benson and Lois Curfman McInnes},\n"
125e2379f4fSBarry Smith                                 "Institution = {Argonne National Laboratory},\n"
126e2379f4fSBarry Smith                                 "Year   = 2014,\n"
127e2379f4fSBarry Smith                                 "Number = {ANL/MCS-TM-322 - Revision 3.5},\n"
128a8d69d7bSBarry Smith                                 "url    = {https://www.mcs.anl.gov/research/projects/tao/}\n}\n",&set);CHKERRQ(ierr);
129494bef23SAlp Dener   tao->header_printed = PETSC_FALSE;
130a7e14dcfSSatish Balay   ierr = TaoSetUp(tao);CHKERRQ(ierr);
131a7e14dcfSSatish Balay   ierr = TaoResetStatistics(tao);CHKERRQ(ierr);
132a7e14dcfSSatish Balay   if (tao->linesearch) {
133a7e14dcfSSatish Balay     ierr = TaoLineSearchReset(tao->linesearch);CHKERRQ(ierr);
134a7e14dcfSSatish Balay   }
135a7e14dcfSSatish Balay 
1360ebee16dSLisandro Dalcin   ierr = PetscLogEventBegin(TAO_Solve,tao,0,0,0);CHKERRQ(ierr);
137a7e14dcfSSatish Balay   if (tao->ops->solve) { ierr = (*tao->ops->solve)(tao);CHKERRQ(ierr); }
1380ebee16dSLisandro Dalcin   ierr = PetscLogEventEnd(TAO_Solve,tao,0,0,0);CHKERRQ(ierr);
139a7e14dcfSSatish Balay 
1407cf37e64SBarry Smith   ierr = VecViewFromOptions(tao->solution,(PetscObject)tao,"-tao_view_solution");CHKERRQ(ierr);
1417cf37e64SBarry Smith 
1428931d482SJason Sarich   tao->ntotalits += tao->niter;
143fbe0838dSJason Sarich   ierr = TaoViewFromOptions(tao,NULL,"-tao_view");CHKERRQ(ierr);
144a7e14dcfSSatish Balay 
145a7e14dcfSSatish Balay   if (tao->printreason) {
146a7e14dcfSSatish Balay     if (tao->reason > 0) {
1478931d482SJason Sarich       ierr = PetscPrintf(((PetscObject)tao)->comm,"TAO solve converged due to %s iterations %D\n",TaoConvergedReasons[tao->reason],tao->niter);CHKERRQ(ierr);
148a7e14dcfSSatish Balay     } else {
1498931d482SJason Sarich       ierr = PetscPrintf(((PetscObject)tao)->comm,"TAO solve did not converge due to %s iteration %D\n",TaoConvergedReasons[tao->reason],tao->niter);CHKERRQ(ierr);
150a7e14dcfSSatish Balay     }
151a7e14dcfSSatish Balay   }
152a7e14dcfSSatish Balay   PetscFunctionReturn(0);
153a7e14dcfSSatish Balay }
154a7e14dcfSSatish Balay 
155a7e14dcfSSatish Balay /*@
156a7e14dcfSSatish Balay   TaoSetUp - Sets up the internal data structures for the later use
157a7e14dcfSSatish Balay   of a Tao solver
158a7e14dcfSSatish Balay 
159a7e14dcfSSatish Balay   Collective on tao
160a7e14dcfSSatish Balay 
161a7e14dcfSSatish Balay   Input Parameters:
162a7e14dcfSSatish Balay . tao - the TAO context
163a7e14dcfSSatish Balay 
164a7e14dcfSSatish Balay   Notes:
165a7e14dcfSSatish Balay   The user will not need to explicitly call TaoSetUp(), as it will
166a7e14dcfSSatish Balay   automatically be called in TaoSolve().  However, if the user
167a7e14dcfSSatish Balay   desires to call it explicitly, it should come after TaoCreate()
168a7e14dcfSSatish Balay   and any TaoSetSomething() routines, but before TaoSolve().
169a7e14dcfSSatish Balay 
170a7e14dcfSSatish Balay   Level: advanced
171a7e14dcfSSatish Balay 
172a7e14dcfSSatish Balay .seealso: TaoCreate(), TaoSolve()
173a7e14dcfSSatish Balay @*/
174441846f8SBarry Smith PetscErrorCode TaoSetUp(Tao tao)
175a7e14dcfSSatish Balay {
176a7e14dcfSSatish Balay   PetscErrorCode ierr;
17747a47007SBarry Smith 
178a7e14dcfSSatish Balay   PetscFunctionBegin;
179441846f8SBarry Smith   PetscValidHeaderSpecific(tao, TAO_CLASSID,1);
180a7e14dcfSSatish Balay   if (tao->setupcalled) PetscFunctionReturn(0);
181691b26d3SBarry Smith   if (!tao->solution) SETERRQ(PetscObjectComm((PetscObject)tao),PETSC_ERR_ARG_WRONGSTATE,"Must call TaoSetInitialVector");
182a7e14dcfSSatish Balay   if (tao->ops->setup) {
183a7e14dcfSSatish Balay     ierr = (*tao->ops->setup)(tao);CHKERRQ(ierr);
184a7e14dcfSSatish Balay   }
185a7e14dcfSSatish Balay   tao->setupcalled = PETSC_TRUE;
186a7e14dcfSSatish Balay   PetscFunctionReturn(0);
187a7e14dcfSSatish Balay }
188a7e14dcfSSatish Balay 
1891fb7b255SJunchao Zhang /*@C
190a7e14dcfSSatish Balay   TaoDestroy - Destroys the TAO context that was created with
191a7e14dcfSSatish Balay   TaoCreate()
192a7e14dcfSSatish Balay 
193441846f8SBarry Smith   Collective on Tao
194a7e14dcfSSatish Balay 
195a7e14dcfSSatish Balay   Input Parameter:
196441846f8SBarry Smith . tao - the Tao context
197a7e14dcfSSatish Balay 
198a7e14dcfSSatish Balay   Level: beginner
199a7e14dcfSSatish Balay 
200a7e14dcfSSatish Balay .seealso: TaoCreate(), TaoSolve()
201a7e14dcfSSatish Balay @*/
202441846f8SBarry Smith PetscErrorCode TaoDestroy(Tao *tao)
203a7e14dcfSSatish Balay {
204a7e14dcfSSatish Balay   PetscErrorCode ierr;
20547a47007SBarry Smith 
206a7e14dcfSSatish Balay   PetscFunctionBegin;
207a7e14dcfSSatish Balay   if (!*tao) PetscFunctionReturn(0);
208441846f8SBarry Smith   PetscValidHeaderSpecific(*tao,TAO_CLASSID,1);
20983c8fe1dSLisandro Dalcin   if (--((PetscObject)*tao)->refct > 0) {*tao = NULL;PetscFunctionReturn(0);}
210a7e14dcfSSatish Balay 
211a7e14dcfSSatish Balay   if ((*tao)->ops->destroy) {
212a7e14dcfSSatish Balay     ierr = (*((*tao))->ops->destroy)(*tao);CHKERRQ(ierr);
213a7e14dcfSSatish Balay   }
214a7e14dcfSSatish Balay   ierr = KSPDestroy(&(*tao)->ksp);CHKERRQ(ierr);
215a7e14dcfSSatish Balay   ierr = TaoLineSearchDestroy(&(*tao)->linesearch);CHKERRQ(ierr);
216a7e14dcfSSatish Balay 
217a7e14dcfSSatish Balay   if ((*tao)->ops->convergencedestroy) {
218a7e14dcfSSatish Balay     ierr = (*(*tao)->ops->convergencedestroy)((*tao)->cnvP);CHKERRQ(ierr);
219a7e14dcfSSatish Balay     if ((*tao)->jacobian_state_inv) {
220a7e14dcfSSatish Balay       ierr = MatDestroy(&(*tao)->jacobian_state_inv);CHKERRQ(ierr);
221a7e14dcfSSatish Balay     }
222a7e14dcfSSatish Balay   }
223a7e14dcfSSatish Balay   ierr = VecDestroy(&(*tao)->solution);CHKERRQ(ierr);
224a7e14dcfSSatish Balay   ierr = VecDestroy(&(*tao)->gradient);CHKERRQ(ierr);
2254a48860cSAlp Dener   ierr = VecDestroy(&(*tao)->ls_res);CHKERRQ(ierr);
226a7e14dcfSSatish Balay 
227a9603a14SPatrick Farrell   if ((*tao)->gradient_norm) {
228a9603a14SPatrick Farrell     ierr = PetscObjectDereference((PetscObject)(*tao)->gradient_norm);CHKERRQ(ierr);
229a9603a14SPatrick Farrell     ierr = VecDestroy(&(*tao)->gradient_norm_tmp);CHKERRQ(ierr);
230a9603a14SPatrick Farrell   }
231a9603a14SPatrick Farrell 
232a7e14dcfSSatish Balay   ierr = VecDestroy(&(*tao)->XL);CHKERRQ(ierr);
233a7e14dcfSSatish Balay   ierr = VecDestroy(&(*tao)->XU);CHKERRQ(ierr);
234a7e14dcfSSatish Balay   ierr = VecDestroy(&(*tao)->IL);CHKERRQ(ierr);
235a7e14dcfSSatish Balay   ierr = VecDestroy(&(*tao)->IU);CHKERRQ(ierr);
236a7e14dcfSSatish Balay   ierr = VecDestroy(&(*tao)->DE);CHKERRQ(ierr);
237a7e14dcfSSatish Balay   ierr = VecDestroy(&(*tao)->DI);CHKERRQ(ierr);
238a7e14dcfSSatish Balay   ierr = VecDestroy(&(*tao)->constraints_equality);CHKERRQ(ierr);
239a7e14dcfSSatish Balay   ierr = VecDestroy(&(*tao)->constraints_inequality);CHKERRQ(ierr);
240a7e14dcfSSatish Balay   ierr = VecDestroy(&(*tao)->stepdirection);CHKERRQ(ierr);
241a7e14dcfSSatish Balay   ierr = MatDestroy(&(*tao)->hessian_pre);CHKERRQ(ierr);
242a7e14dcfSSatish Balay   ierr = MatDestroy(&(*tao)->hessian);CHKERRQ(ierr);
2434a48860cSAlp Dener   ierr = MatDestroy(&(*tao)->ls_jac);CHKERRQ(ierr);
2444a48860cSAlp Dener   ierr = MatDestroy(&(*tao)->ls_jac_pre);CHKERRQ(ierr);
245a7e14dcfSSatish Balay   ierr = MatDestroy(&(*tao)->jacobian_pre);CHKERRQ(ierr);
246a7e14dcfSSatish Balay   ierr = MatDestroy(&(*tao)->jacobian);CHKERRQ(ierr);
247a7e14dcfSSatish Balay   ierr = MatDestroy(&(*tao)->jacobian_state_pre);CHKERRQ(ierr);
248a7e14dcfSSatish Balay   ierr = MatDestroy(&(*tao)->jacobian_state);CHKERRQ(ierr);
249a7e14dcfSSatish Balay   ierr = MatDestroy(&(*tao)->jacobian_state_inv);CHKERRQ(ierr);
250067f330cSJason Sarich   ierr = MatDestroy(&(*tao)->jacobian_design);CHKERRQ(ierr);
251a7e14dcfSSatish Balay   ierr = MatDestroy(&(*tao)->jacobian_equality);CHKERRQ(ierr);
252a7e14dcfSSatish Balay   ierr = MatDestroy(&(*tao)->jacobian_equality_pre);CHKERRQ(ierr);
253a7e14dcfSSatish Balay   ierr = MatDestroy(&(*tao)->jacobian_inequality);CHKERRQ(ierr);
254a7e14dcfSSatish Balay   ierr = MatDestroy(&(*tao)->jacobian_inequality_pre);CHKERRQ(ierr);
255a7e14dcfSSatish Balay   ierr = ISDestroy(&(*tao)->state_is);CHKERRQ(ierr);
256a7e14dcfSSatish Balay   ierr = ISDestroy(&(*tao)->design_is);CHKERRQ(ierr);
257737f463aSAlp Dener   ierr = VecDestroy(&(*tao)->res_weights_v);CHKERRQ(ierr);
258a7e14dcfSSatish Balay   ierr = TaoCancelMonitors(*tao);CHKERRQ(ierr);
259ae93cb3cSJason Sarich   if ((*tao)->hist_malloc) {
260071fcb05SBarry Smith     ierr = PetscFree4((*tao)->hist_obj,(*tao)->hist_resid,(*tao)->hist_cnorm,(*tao)->hist_lits);CHKERRQ(ierr);
261ae93cb3cSJason Sarich   }
262737f463aSAlp Dener   if ((*tao)->res_weights_n) {
263737f463aSAlp Dener     ierr = PetscFree((*tao)->res_weights_rows);CHKERRQ(ierr);
264737f463aSAlp Dener     ierr = PetscFree((*tao)->res_weights_cols);CHKERRQ(ierr);
265737f463aSAlp Dener     ierr = PetscFree((*tao)->res_weights_w);CHKERRQ(ierr);
266125ddc8aSJason Sarich   }
267a7e14dcfSSatish Balay   ierr = PetscHeaderDestroy(tao);CHKERRQ(ierr);
268a7e14dcfSSatish Balay   PetscFunctionReturn(0);
269a7e14dcfSSatish Balay }
270a7e14dcfSSatish Balay 
271a7e14dcfSSatish Balay /*@
272441846f8SBarry Smith   TaoSetFromOptions - Sets various Tao parameters from user
273a7e14dcfSSatish Balay   options.
274a7e14dcfSSatish Balay 
275441846f8SBarry Smith   Collective on Tao
276a7e14dcfSSatish Balay 
27701d2d390SJose E. Roman   Input Parameter:
278441846f8SBarry Smith . tao - the Tao solver context
279a7e14dcfSSatish Balay 
280a7e14dcfSSatish Balay   options Database Keys:
28158417fe7SBarry Smith + -tao_type <type> - The algorithm that TAO uses (lmvm, nls, etc.)
282a7e14dcfSSatish Balay . -tao_gatol <gatol> - absolute error tolerance for ||gradient||
283a7e14dcfSSatish Balay . -tao_grtol <grtol> - relative error tolerance for ||gradient||
284a7e14dcfSSatish Balay . -tao_gttol <gttol> - reduction of ||gradient|| relative to initial gradient
285a7e14dcfSSatish Balay . -tao_max_it <max> - sets maximum number of iterations
286a7e14dcfSSatish Balay . -tao_max_funcs <max> - sets maximum number of function evaluations
287a7e14dcfSSatish Balay . -tao_fmin <fmin> - stop if function value reaches fmin
288a7e14dcfSSatish Balay . -tao_steptol <tol> - stop if trust region radius less than <tol>
289a7e14dcfSSatish Balay . -tao_trust0 <t> - initial trust region radius
290a7e14dcfSSatish Balay . -tao_monitor - prints function value and residual at each iteration
291a7e14dcfSSatish Balay . -tao_smonitor - same as tao_monitor, but truncates very small values
292a7e14dcfSSatish Balay . -tao_cmonitor - prints function value, residual, and constraint norm at each iteration
293a7e14dcfSSatish Balay . -tao_view_solution - prints solution vector at each iteration
2944a48860cSAlp Dener . -tao_view_ls_residual - prints least-squares residual vector at each iteration
295a7e14dcfSSatish Balay . -tao_view_step - prints step direction vector at each iteration
296a7e14dcfSSatish Balay . -tao_view_gradient - prints gradient vector at each iteration
297a7e14dcfSSatish Balay . -tao_draw_solution - graphically view solution vector at each iteration
298a7e14dcfSSatish Balay . -tao_draw_step - graphically view step vector at each iteration
299a7e14dcfSSatish Balay . -tao_draw_gradient - graphically view gradient at each iteration
300a7e14dcfSSatish Balay . -tao_fd_gradient - use gradient computed with finite differences
301f4c1ad5cSStefano Zampini . -tao_fd_hessian - use hessian computed with finite differences
302f4c1ad5cSStefano Zampini . -tao_mf_hessian - use matrix-free hessian computed with finite differences
303a7e14dcfSSatish Balay . -tao_cancelmonitors - cancels all monitors (except those set with command line)
304441846f8SBarry Smith . -tao_view - prints information about the Tao after solving
305a7e14dcfSSatish Balay - -tao_converged_reason - prints the reason TAO stopped iterating
306a7e14dcfSSatish Balay 
307a7e14dcfSSatish Balay   Notes:
308a7e14dcfSSatish Balay   To see all options, run your program with the -help option or consult the
309a7e14dcfSSatish Balay   user's manual. Should be called after TaoCreate() but before TaoSolve()
310a7e14dcfSSatish Balay 
311a7e14dcfSSatish Balay   Level: beginner
312a7e14dcfSSatish Balay @*/
313441846f8SBarry Smith PetscErrorCode TaoSetFromOptions(Tao tao)
314a7e14dcfSSatish Balay {
315a7e14dcfSSatish Balay   PetscErrorCode ierr;
316b625d6c7SJed Brown   TaoType        default_type = TAOLMVM;
317a7e14dcfSSatish Balay   char           type[256], monfilename[PETSC_MAX_PATH_LEN];
318a7e14dcfSSatish Balay   PetscViewer    monviewer;
319a7e14dcfSSatish Balay   PetscBool      flg;
320a7e14dcfSSatish Balay   MPI_Comm       comm;
321a7e14dcfSSatish Balay 
322a7e14dcfSSatish Balay   PetscFunctionBegin;
323441846f8SBarry Smith   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
324a7e14dcfSSatish Balay   ierr = PetscObjectGetComm((PetscObject)tao,&comm);CHKERRQ(ierr);
325125ddc8aSJason Sarich 
326a7e14dcfSSatish Balay   /* So no warnings are given about unused options */
327c5929fdfSBarry Smith   ierr = PetscOptionsHasName(((PetscObject)tao)->options,((PetscObject)tao)->prefix,"-tao_ls_type",&flg);CHKERRQ(ierr);
328a7e14dcfSSatish Balay 
329a7e14dcfSSatish Balay   ierr = PetscObjectOptionsBegin((PetscObject)tao);CHKERRQ(ierr);
330a7e14dcfSSatish Balay   {
331441846f8SBarry Smith     ierr = TaoRegisterAll();CHKERRQ(ierr);
332a7e14dcfSSatish Balay     if (((PetscObject)tao)->type_name) {
333a7e14dcfSSatish Balay       default_type = ((PetscObject)tao)->type_name;
334a7e14dcfSSatish Balay     }
335a7e14dcfSSatish Balay     /* Check for type from options */
336441846f8SBarry Smith     ierr = PetscOptionsFList("-tao_type","Tao Solver type","TaoSetType",TaoList,default_type,type,256,&flg);CHKERRQ(ierr);
337a7e14dcfSSatish Balay     if (flg) {
338a7e14dcfSSatish Balay       ierr = TaoSetType(tao,type);CHKERRQ(ierr);
339a7e14dcfSSatish Balay     } else if (!((PetscObject)tao)->type_name) {
340302440fdSBarry Smith       ierr = TaoSetType(tao,default_type);CHKERRQ(ierr);
341a7e14dcfSSatish Balay     }
342a7e14dcfSSatish Balay 
3436552cf8aSJason Sarich     ierr = PetscOptionsReal("-tao_catol","Stop if constraints violations within","TaoSetConstraintTolerances",tao->catol,&tao->catol,&flg);CHKERRQ(ierr);
3446552cf8aSJason Sarich     if (flg) tao->catol_changed = PETSC_TRUE;
3456552cf8aSJason Sarich     ierr = PetscOptionsReal("-tao_crtol","Stop if relative contraint violations within","TaoSetConstraintTolerances",tao->crtol,&tao->crtol,&flg);CHKERRQ(ierr);
3466552cf8aSJason Sarich     if (flg) tao->crtol_changed = PETSC_TRUE;
3476552cf8aSJason Sarich     ierr = PetscOptionsReal("-tao_gatol","Stop if norm of gradient less than","TaoSetTolerances",tao->gatol,&tao->gatol,&flg);CHKERRQ(ierr);
3486552cf8aSJason Sarich     if (flg) tao->gatol_changed = PETSC_TRUE;
3496552cf8aSJason Sarich     ierr = PetscOptionsReal("-tao_grtol","Stop if norm of gradient divided by the function value is less than","TaoSetTolerances",tao->grtol,&tao->grtol,&flg);CHKERRQ(ierr);
3506552cf8aSJason Sarich     if (flg) tao->grtol_changed = PETSC_TRUE;
3516552cf8aSJason Sarich     ierr = 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);CHKERRQ(ierr);
3526552cf8aSJason Sarich     if (flg) tao->gttol_changed = PETSC_TRUE;
3536552cf8aSJason Sarich     ierr = PetscOptionsInt("-tao_max_it","Stop if iteration number exceeds","TaoSetMaximumIterations",tao->max_it,&tao->max_it,&flg);CHKERRQ(ierr);
3546552cf8aSJason Sarich     if (flg) tao->max_it_changed = PETSC_TRUE;
3556552cf8aSJason Sarich     ierr = PetscOptionsInt("-tao_max_funcs","Stop if number of function evaluations exceeds","TaoSetMaximumFunctionEvaluations",tao->max_funcs,&tao->max_funcs,&flg);CHKERRQ(ierr);
3566552cf8aSJason Sarich     if (flg) tao->max_funcs_changed = PETSC_TRUE;
3576552cf8aSJason Sarich     ierr = PetscOptionsReal("-tao_fmin","Stop if function less than","TaoSetFunctionLowerBound",tao->fmin,&tao->fmin,&flg);CHKERRQ(ierr);
3586552cf8aSJason Sarich     if (flg) tao->fmin_changed = PETSC_TRUE;
3596552cf8aSJason Sarich     ierr = PetscOptionsReal("-tao_steptol","Stop if step size or trust region radius less than","",tao->steptol,&tao->steptol,&flg);CHKERRQ(ierr);
3606552cf8aSJason Sarich     if (flg) tao->steptol_changed = PETSC_TRUE;
3616552cf8aSJason Sarich     ierr = PetscOptionsReal("-tao_trust0","Initial trust region radius","TaoSetTrustRegionRadius",tao->trust0,&tao->trust0,&flg);CHKERRQ(ierr);
3626552cf8aSJason Sarich     if (flg) tao->trust0_changed = PETSC_TRUE;
363589a23caSBarry Smith     ierr = PetscOptionsString("-tao_view_solution","view solution vector after each evaluation","TaoSetMonitor","stdout",monfilename,sizeof(monfilename),&flg);CHKERRQ(ierr);
364a7e14dcfSSatish Balay     if (flg) {
365a7e14dcfSSatish Balay       ierr = PetscViewerASCIIOpen(comm,monfilename,&monviewer);CHKERRQ(ierr);
366a7e14dcfSSatish Balay       ierr = TaoSetMonitor(tao,TaoSolutionMonitor,monviewer,(PetscErrorCode (*)(void**))PetscViewerDestroy);CHKERRQ(ierr);
367a7e14dcfSSatish Balay     }
368a7e14dcfSSatish Balay 
3698afaa268SBarry Smith     ierr = PetscOptionsBool("-tao_converged_reason","Print reason for TAO converged","TaoSolve",tao->printreason,&tao->printreason,NULL);CHKERRQ(ierr);
370589a23caSBarry Smith     ierr = PetscOptionsString("-tao_view_gradient","view gradient vector after each evaluation","TaoSetMonitor","stdout",monfilename,sizeof(monfilename),&flg);CHKERRQ(ierr);
371a7e14dcfSSatish Balay     if (flg) {
372a7e14dcfSSatish Balay       ierr = PetscViewerASCIIOpen(comm,monfilename,&monviewer);CHKERRQ(ierr);
373a7e14dcfSSatish Balay       ierr = TaoSetMonitor(tao,TaoGradientMonitor,monviewer,(PetscErrorCode (*)(void**))PetscViewerDestroy);CHKERRQ(ierr);
374a7e14dcfSSatish Balay     }
375a7e14dcfSSatish Balay 
376589a23caSBarry Smith     ierr = PetscOptionsString("-tao_view_stepdirection","view step direction vector after each iteration","TaoSetMonitor","stdout",monfilename,sizeof(monfilename),&flg);CHKERRQ(ierr);
377a7e14dcfSSatish Balay     if (flg) {
378a7e14dcfSSatish Balay       ierr = PetscViewerASCIIOpen(comm,monfilename,&monviewer);CHKERRQ(ierr);
379a7e14dcfSSatish Balay       ierr = TaoSetMonitor(tao,TaoStepDirectionMonitor,monviewer,(PetscErrorCode (*)(void**))PetscViewerDestroy);CHKERRQ(ierr);
380a7e14dcfSSatish Balay     }
381a7e14dcfSSatish Balay 
382589a23caSBarry Smith     ierr = PetscOptionsString("-tao_view_residual","view least-squares residual vector after each evaluation","TaoSetMonitor","stdout",monfilename,sizeof(monfilename),&flg);CHKERRQ(ierr);
383a7e14dcfSSatish Balay     if (flg) {
384a7e14dcfSSatish Balay       ierr = PetscViewerASCIIOpen(comm,monfilename,&monviewer);CHKERRQ(ierr);
385737f463aSAlp Dener       ierr = TaoSetMonitor(tao,TaoResidualMonitor,monviewer,(PetscErrorCode (*)(void**))PetscViewerDestroy);CHKERRQ(ierr);
386a7e14dcfSSatish Balay     }
387a7e14dcfSSatish Balay 
388589a23caSBarry Smith     ierr = PetscOptionsString("-tao_monitor","Use the default convergence monitor","TaoSetMonitor","stdout",monfilename,sizeof(monfilename),&flg);CHKERRQ(ierr);
389a7e14dcfSSatish Balay     if (flg) {
390a7e14dcfSSatish Balay       ierr = PetscViewerASCIIOpen(comm,monfilename,&monviewer);CHKERRQ(ierr);
39198ea980cSBarry Smith       ierr = TaoSetMonitor(tao,TaoMonitorDefault,monviewer,(PetscErrorCode (*)(void**))PetscViewerDestroy);CHKERRQ(ierr);
392a7e14dcfSSatish Balay     }
393a7e14dcfSSatish Balay 
394589a23caSBarry Smith     ierr = PetscOptionsString("-tao_gmonitor","Use the convergence monitor with extra globalization info","TaoSetMonitor","stdout",monfilename,sizeof(monfilename),&flg);CHKERRQ(ierr);
3958d5ead36SAlp Dener     if (flg) {
3968d5ead36SAlp Dener       ierr = PetscViewerASCIIOpen(comm,monfilename,&monviewer);CHKERRQ(ierr);
3978d5ead36SAlp Dener       ierr = TaoSetMonitor(tao,TaoDefaultGMonitor,monviewer,(PetscErrorCode (*)(void**))PetscViewerDestroy);CHKERRQ(ierr);
3988d5ead36SAlp Dener     }
3998d5ead36SAlp Dener 
400589a23caSBarry Smith     ierr = PetscOptionsString("-tao_smonitor","Use the short convergence monitor","TaoSetMonitor","stdout",monfilename,sizeof(monfilename),&flg);CHKERRQ(ierr);
401a7e14dcfSSatish Balay     if (flg) {
402a7e14dcfSSatish Balay       ierr = PetscViewerASCIIOpen(comm,monfilename,&monviewer);CHKERRQ(ierr);
403a7e14dcfSSatish Balay       ierr = TaoSetMonitor(tao,TaoDefaultSMonitor,monviewer,(PetscErrorCode (*)(void**))PetscViewerDestroy);CHKERRQ(ierr);
404a7e14dcfSSatish Balay     }
405a7e14dcfSSatish Balay 
406589a23caSBarry Smith     ierr = PetscOptionsString("-tao_cmonitor","Use the default convergence monitor with constraint norm","TaoSetMonitor","stdout",monfilename,sizeof(monfilename),&flg);CHKERRQ(ierr);
407a7e14dcfSSatish Balay     if (flg) {
408a7e14dcfSSatish Balay       ierr = PetscViewerASCIIOpen(comm,monfilename,&monviewer);CHKERRQ(ierr);
409a7e14dcfSSatish Balay       ierr = TaoSetMonitor(tao,TaoDefaultCMonitor,monviewer,(PetscErrorCode (*)(void**))PetscViewerDestroy);CHKERRQ(ierr);
410a7e14dcfSSatish Balay     }
411a7e14dcfSSatish Balay 
4128afaa268SBarry Smith     flg = PETSC_FALSE;
4138afaa268SBarry Smith     ierr = PetscOptionsBool("-tao_cancelmonitors","cancel all monitors and call any registered destroy routines","TaoCancelMonitors",flg,&flg,NULL);CHKERRQ(ierr);
414a7e14dcfSSatish Balay     if (flg) {ierr = TaoCancelMonitors(tao);CHKERRQ(ierr);}
415a7e14dcfSSatish Balay 
4168afaa268SBarry Smith     flg = PETSC_FALSE;
4178afaa268SBarry Smith     ierr = PetscOptionsBool("-tao_draw_solution","Plot solution vector at each iteration","TaoSetMonitor",flg,&flg,NULL);CHKERRQ(ierr);
418a7e14dcfSSatish Balay     if (flg) {
419e882e171SHong Zhang       TaoMonitorDrawCtx drawctx;
420e882e171SHong Zhang       PetscInt          howoften = 1;
42183c8fe1dSLisandro Dalcin       ierr = TaoMonitorDrawCtxCreate(PetscObjectComm((PetscObject)tao),NULL,NULL,PETSC_DECIDE,PETSC_DECIDE,300,300,howoften,&drawctx);CHKERRQ(ierr);
422e882e171SHong Zhang       ierr = TaoSetMonitor(tao,TaoDrawSolutionMonitor,drawctx,(PetscErrorCode (*)(void**))TaoMonitorDrawCtxDestroy);CHKERRQ(ierr);
423a7e14dcfSSatish Balay     }
424a7e14dcfSSatish Balay 
4258afaa268SBarry Smith     flg = PETSC_FALSE;
4268afaa268SBarry Smith     ierr = PetscOptionsBool("-tao_draw_step","plots step direction at each iteration","TaoSetMonitor",flg,&flg,NULL);CHKERRQ(ierr);
427a7e14dcfSSatish Balay     if (flg) {
42847a47007SBarry Smith       ierr = TaoSetMonitor(tao,TaoDrawStepMonitor,NULL,NULL);CHKERRQ(ierr);
429a7e14dcfSSatish Balay     }
430a7e14dcfSSatish Balay 
4318afaa268SBarry Smith     flg = PETSC_FALSE;
4328afaa268SBarry Smith     ierr = PetscOptionsBool("-tao_draw_gradient","plots gradient at each iteration","TaoSetMonitor",flg,&flg,NULL);CHKERRQ(ierr);
433a7e14dcfSSatish Balay     if (flg) {
434e882e171SHong Zhang       TaoMonitorDrawCtx drawctx;
435e882e171SHong Zhang       PetscInt          howoften = 1;
43683c8fe1dSLisandro Dalcin       ierr = TaoMonitorDrawCtxCreate(PetscObjectComm((PetscObject)tao),NULL,NULL,PETSC_DECIDE,PETSC_DECIDE,300,300,howoften,&drawctx);CHKERRQ(ierr);
437e882e171SHong Zhang       ierr = TaoSetMonitor(tao,TaoDrawGradientMonitor,drawctx,(PetscErrorCode (*)(void**))TaoMonitorDrawCtxDestroy);CHKERRQ(ierr);
438a7e14dcfSSatish Balay     }
4398afaa268SBarry Smith     flg = PETSC_FALSE;
4408afaa268SBarry Smith     ierr = PetscOptionsBool("-tao_fd_gradient","compute gradient using finite differences","TaoDefaultComputeGradient",flg,&flg,NULL);CHKERRQ(ierr);
441a7e14dcfSSatish Balay     if (flg) {
44247a47007SBarry Smith       ierr = TaoSetGradientRoutine(tao,TaoDefaultComputeGradient,NULL);CHKERRQ(ierr);
443a7e14dcfSSatish Balay     }
444f4c1ad5cSStefano Zampini     flg = PETSC_FALSE;
445f4c1ad5cSStefano Zampini     ierr = PetscOptionsBool("-tao_fd_hessian","compute hessian using finite differences","TaoDefaultComputeHessian",flg,&flg,NULL);CHKERRQ(ierr);
446f4c1ad5cSStefano Zampini     if (flg) {
447f4c1ad5cSStefano Zampini       Mat H;
448f4c1ad5cSStefano Zampini 
449f4c1ad5cSStefano Zampini       ierr = MatCreate(PetscObjectComm((PetscObject)tao),&H);CHKERRQ(ierr);
450f4c1ad5cSStefano Zampini       ierr = MatSetType(H,MATAIJ);CHKERRQ(ierr);
451f4c1ad5cSStefano Zampini       ierr = TaoSetHessianRoutine(tao,H,H,TaoDefaultComputeHessian,NULL);CHKERRQ(ierr);
452f4c1ad5cSStefano Zampini       ierr = MatDestroy(&H);CHKERRQ(ierr);
453f4c1ad5cSStefano Zampini     }
454f4c1ad5cSStefano Zampini     flg = PETSC_FALSE;
455f4c1ad5cSStefano Zampini     ierr = PetscOptionsBool("-tao_mf_hessian","compute matrix-free hessian using finite differences","TaoDefaultComputeHessianMFFD",flg,&flg,NULL);CHKERRQ(ierr);
456f4c1ad5cSStefano Zampini     if (flg) {
457f4c1ad5cSStefano Zampini       Mat H;
458f4c1ad5cSStefano Zampini 
459f4c1ad5cSStefano Zampini       ierr = MatCreate(PetscObjectComm((PetscObject)tao),&H);CHKERRQ(ierr);
460f4c1ad5cSStefano Zampini       ierr = TaoSetHessianRoutine(tao,H,H,TaoDefaultComputeHessianMFFD,NULL);CHKERRQ(ierr);
461f4c1ad5cSStefano Zampini       ierr = MatDestroy(&H);CHKERRQ(ierr);
462f4c1ad5cSStefano Zampini     }
463414d97d3SAlp Dener     flg = PETSC_FALSE;
464414d97d3SAlp Dener     ierr = PetscOptionsBool("-tao_recycle_history","enable recycling/re-using information from the previous TaoSolve() call for some algorithms","TaoSetRecycleHistory",flg,&flg,NULL);CHKERRQ(ierr);
465414d97d3SAlp Dener     if (flg) {
466414d97d3SAlp Dener       ierr = TaoSetRecycleHistory(tao, PETSC_TRUE);CHKERRQ(ierr);
467414d97d3SAlp Dener     }
468f4c1ad5cSStefano Zampini     ierr = PetscOptionsEnum("-tao_subset_type","subset type","",TaoSubSetTypes,(PetscEnum)tao->subset_type,(PetscEnum*)&tao->subset_type,NULL);CHKERRQ(ierr);
469a7e14dcfSSatish Balay 
470a7e14dcfSSatish Balay     if (tao->ops->setfromoptions) {
4711a1499c8SBarry Smith       ierr = (*tao->ops->setfromoptions)(PetscOptionsObject,tao);CHKERRQ(ierr);
472a7e14dcfSSatish Balay     }
473a7e14dcfSSatish Balay   }
474a7e14dcfSSatish Balay   ierr = PetscOptionsEnd();CHKERRQ(ierr);
475a7e14dcfSSatish Balay   PetscFunctionReturn(0);
476a7e14dcfSSatish Balay }
477a7e14dcfSSatish Balay 
478a7e14dcfSSatish Balay /*@C
479fe2efc57SMark    TaoViewFromOptions - View from Options
480fe2efc57SMark 
481fe2efc57SMark    Collective on Tao
482fe2efc57SMark 
483fe2efc57SMark    Input Parameters:
484fe2efc57SMark +  A - the  Tao context
485736c3998SJose E. Roman .  obj - Optional object
486736c3998SJose E. Roman -  name - command line option
487fe2efc57SMark 
488fe2efc57SMark    Level: intermediate
489fe2efc57SMark .seealso:  Tao, TaoView, PetscObjectViewFromOptions(), TaoCreate()
490fe2efc57SMark @*/
491fe2efc57SMark PetscErrorCode  TaoViewFromOptions(Tao A,PetscObject obj,const char name[])
492fe2efc57SMark {
493fe2efc57SMark   PetscErrorCode ierr;
494fe2efc57SMark 
495fe2efc57SMark   PetscFunctionBegin;
496fe2efc57SMark   PetscValidHeaderSpecific(A,TAO_CLASSID,1);
497fe2efc57SMark   ierr = PetscObjectViewFromOptions((PetscObject)A,obj,name);CHKERRQ(ierr);
498fe2efc57SMark   PetscFunctionReturn(0);
499fe2efc57SMark }
500fe2efc57SMark 
501fe2efc57SMark /*@C
502441846f8SBarry Smith   TaoView - Prints information about the Tao
503a7e14dcfSSatish Balay 
504441846f8SBarry Smith   Collective on Tao
505a7e14dcfSSatish Balay 
506a7e14dcfSSatish Balay   InputParameters:
507441846f8SBarry Smith + tao - the Tao context
508a7e14dcfSSatish Balay - viewer - visualization context
509a7e14dcfSSatish Balay 
510a7e14dcfSSatish Balay   Options Database Key:
511a7e14dcfSSatish Balay . -tao_view - Calls TaoView() at the end of TaoSolve()
512a7e14dcfSSatish Balay 
513a7e14dcfSSatish Balay   Notes:
514a7e14dcfSSatish Balay   The available visualization contexts include
515a7e14dcfSSatish Balay +     PETSC_VIEWER_STDOUT_SELF - standard output (default)
516a7e14dcfSSatish Balay -     PETSC_VIEWER_STDOUT_WORLD - synchronized standard
517a7e14dcfSSatish Balay          output where only the first processor opens
518a7e14dcfSSatish Balay          the file.  All other processors send their
519a7e14dcfSSatish Balay          data to the first processor to print.
520a7e14dcfSSatish Balay 
521a7e14dcfSSatish Balay   Level: beginner
522a7e14dcfSSatish Balay 
523a7e14dcfSSatish Balay .seealso: PetscViewerASCIIOpen()
524a7e14dcfSSatish Balay @*/
525441846f8SBarry Smith PetscErrorCode TaoView(Tao tao, PetscViewer viewer)
526a7e14dcfSSatish Balay {
527a7e14dcfSSatish Balay   PetscErrorCode      ierr;
528a7e14dcfSSatish Balay   PetscBool           isascii,isstring;
529b625d6c7SJed Brown   TaoType             type;
53047a47007SBarry Smith 
531a7e14dcfSSatish Balay   PetscFunctionBegin;
532441846f8SBarry Smith   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
533a7e14dcfSSatish Balay   if (!viewer) {
534a7e14dcfSSatish Balay     ierr = PetscViewerASCIIGetStdout(((PetscObject)tao)->comm,&viewer);CHKERRQ(ierr);
535a7e14dcfSSatish Balay   }
536a7e14dcfSSatish Balay   PetscValidHeaderSpecific(viewer,PETSC_VIEWER_CLASSID,2);
537a7e14dcfSSatish Balay   PetscCheckSameComm(tao,1,viewer,2);
538a7e14dcfSSatish Balay 
539a7e14dcfSSatish Balay   ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERASCII,&isascii);CHKERRQ(ierr);
540a7e14dcfSSatish Balay   ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERSTRING,&isstring);CHKERRQ(ierr);
541a7e14dcfSSatish Balay   if (isascii) {
542a7e14dcfSSatish Balay     ierr = PetscObjectPrintClassNamePrefixType((PetscObject)tao,viewer);CHKERRQ(ierr);
543a7e14dcfSSatish Balay 
544a7e14dcfSSatish Balay     if (tao->ops->view) {
545a7e14dcfSSatish Balay       ierr = PetscViewerASCIIPushTab(viewer);CHKERRQ(ierr);
546a7e14dcfSSatish Balay       ierr = (*tao->ops->view)(tao,viewer);CHKERRQ(ierr);
547a7e14dcfSSatish Balay       ierr = PetscViewerASCIIPopTab(viewer);CHKERRQ(ierr);
548a7e14dcfSSatish Balay     }
5498d3f3ddaSAlp Dener     if (tao->linesearch) {
55085928a98SStefano Zampini       ierr = PetscViewerASCIIPushTab(viewer);CHKERRQ(ierr);
5518d3f3ddaSAlp Dener       ierr = TaoLineSearchView(tao->linesearch,viewer);CHKERRQ(ierr);
55285928a98SStefano Zampini       ierr = PetscViewerASCIIPopTab(viewer);CHKERRQ(ierr);
5538d3f3ddaSAlp Dener     }
554a7e14dcfSSatish Balay     if (tao->ksp) {
55519b5c101SAlp Dener       ierr = PetscViewerASCIIPushTab(viewer);CHKERRQ(ierr);
55685928a98SStefano Zampini       ierr = KSPView(tao->ksp,viewer);CHKERRQ(ierr);
557ae93cb3cSJason Sarich       ierr = PetscViewerASCIIPrintf(viewer,"total KSP iterations: %D\n",tao->ksp_tot_its);CHKERRQ(ierr);
55819b5c101SAlp Dener       ierr = PetscViewerASCIIPopTab(viewer);CHKERRQ(ierr);
559a7e14dcfSSatish Balay     }
56019b5c101SAlp Dener 
56119b5c101SAlp Dener     ierr = PetscViewerASCIIPushTab(viewer);CHKERRQ(ierr);
56219b5c101SAlp Dener 
563a7e14dcfSSatish Balay     if (tao->XL || tao->XU) {
564302440fdSBarry Smith       ierr = PetscViewerASCIIPrintf(viewer,"Active Set subset type: %s\n",TaoSubSetTypes[tao->subset_type]);CHKERRQ(ierr);
565a7e14dcfSSatish Balay     }
566a7e14dcfSSatish Balay 
56747a47007SBarry Smith     ierr = PetscViewerASCIIPrintf(viewer,"convergence tolerances: gatol=%g,",(double)tao->gatol);CHKERRQ(ierr);
56847a47007SBarry Smith     ierr = PetscViewerASCIIPrintf(viewer," steptol=%g,",(double)tao->steptol);CHKERRQ(ierr);
56947a47007SBarry Smith     ierr = PetscViewerASCIIPrintf(viewer," gttol=%g\n",(double)tao->gttol);CHKERRQ(ierr);
57047a47007SBarry Smith     ierr = PetscViewerASCIIPrintf(viewer,"Residual in Function/Gradient:=%g\n",(double)tao->residual);CHKERRQ(ierr);
571a7e14dcfSSatish Balay 
5726246e8cdSAlp Dener     if (tao->constrained) {
573a7e14dcfSSatish Balay       ierr = PetscViewerASCIIPrintf(viewer,"convergence tolerances:");CHKERRQ(ierr);
57447a47007SBarry Smith       ierr = PetscViewerASCIIPrintf(viewer," catol=%g,",(double)tao->catol);CHKERRQ(ierr);
57547a47007SBarry Smith       ierr = PetscViewerASCIIPrintf(viewer," crtol=%g\n",(double)tao->crtol);CHKERRQ(ierr);
57647a47007SBarry Smith       ierr = PetscViewerASCIIPrintf(viewer,"Residual in Constraints:=%g\n",(double)tao->cnorm);CHKERRQ(ierr);
577a7e14dcfSSatish Balay     }
578a7e14dcfSSatish Balay 
579a7e14dcfSSatish Balay     if (tao->trust < tao->steptol) {
58047a47007SBarry Smith       ierr = PetscViewerASCIIPrintf(viewer,"convergence tolerances: steptol=%g\n",(double)tao->steptol);CHKERRQ(ierr);
58147a47007SBarry Smith       ierr = PetscViewerASCIIPrintf(viewer,"Final trust region radius:=%g\n",(double)tao->trust);CHKERRQ(ierr);
582a7e14dcfSSatish Balay     }
583a7e14dcfSSatish Balay 
584a7e14dcfSSatish Balay     if (tao->fmin>-1.e25) {
58547a47007SBarry Smith       ierr = PetscViewerASCIIPrintf(viewer,"convergence tolerances: function minimum=%g\n",(double)tao->fmin);CHKERRQ(ierr);
586a7e14dcfSSatish Balay     }
58747a47007SBarry Smith     ierr = PetscViewerASCIIPrintf(viewer,"Objective value=%g\n",(double)tao->fc);CHKERRQ(ierr);
588a7e14dcfSSatish Balay 
58947a47007SBarry Smith     ierr = PetscViewerASCIIPrintf(viewer,"total number of iterations=%D,          ",tao->niter);CHKERRQ(ierr);
590a7e14dcfSSatish Balay     ierr = PetscViewerASCIIPrintf(viewer,"              (max: %D)\n",tao->max_it);CHKERRQ(ierr);
591a7e14dcfSSatish Balay 
592a7e14dcfSSatish Balay     if (tao->nfuncs>0) {
59347a47007SBarry Smith       ierr = PetscViewerASCIIPrintf(viewer,"total number of function evaluations=%D,",tao->nfuncs);CHKERRQ(ierr);
59447a47007SBarry Smith       ierr = PetscViewerASCIIPrintf(viewer,"                max: %D\n",tao->max_funcs);CHKERRQ(ierr);
595a7e14dcfSSatish Balay     }
596a7e14dcfSSatish Balay     if (tao->ngrads>0) {
59747a47007SBarry Smith       ierr = PetscViewerASCIIPrintf(viewer,"total number of gradient evaluations=%D,",tao->ngrads);CHKERRQ(ierr);
59847a47007SBarry Smith       ierr = PetscViewerASCIIPrintf(viewer,"                max: %D\n",tao->max_funcs);CHKERRQ(ierr);
599a7e14dcfSSatish Balay     }
600a7e14dcfSSatish Balay     if (tao->nfuncgrads>0) {
60147a47007SBarry Smith       ierr = PetscViewerASCIIPrintf(viewer,"total number of function/gradient evaluations=%D,",tao->nfuncgrads);CHKERRQ(ierr);
60247a47007SBarry Smith       ierr = PetscViewerASCIIPrintf(viewer,"    (max: %D)\n",tao->max_funcs);CHKERRQ(ierr);
603a7e14dcfSSatish Balay     }
604a7e14dcfSSatish Balay     if (tao->nhess>0) {
60547a47007SBarry Smith       ierr = PetscViewerASCIIPrintf(viewer,"total number of Hessian evaluations=%D\n",tao->nhess);CHKERRQ(ierr);
606a7e14dcfSSatish Balay     }
607a7e14dcfSSatish Balay     if (tao->nconstraints>0) {
60847a47007SBarry Smith       ierr = PetscViewerASCIIPrintf(viewer,"total number of constraint function evaluations=%D\n",tao->nconstraints);CHKERRQ(ierr);
609a7e14dcfSSatish Balay     }
610a7e14dcfSSatish Balay     if (tao->njac>0) {
61147a47007SBarry Smith       ierr = PetscViewerASCIIPrintf(viewer,"total number of Jacobian evaluations=%D\n",tao->njac);CHKERRQ(ierr);
612a7e14dcfSSatish Balay     }
613a7e14dcfSSatish Balay 
614a7e14dcfSSatish Balay     if (tao->reason>0) {
615a7e14dcfSSatish Balay       ierr = PetscViewerASCIIPrintf(viewer,    "Solution converged: ");CHKERRQ(ierr);
616a7e14dcfSSatish Balay       switch (tao->reason) {
617a7e14dcfSSatish Balay       case TAO_CONVERGED_GATOL:
618a7e14dcfSSatish Balay         ierr = PetscViewerASCIIPrintf(viewer," ||g(X)|| <= gatol\n");CHKERRQ(ierr);
619a7e14dcfSSatish Balay         break;
620a7e14dcfSSatish Balay       case TAO_CONVERGED_GRTOL:
621a7e14dcfSSatish Balay         ierr = PetscViewerASCIIPrintf(viewer," ||g(X)||/|f(X)| <= grtol\n");CHKERRQ(ierr);
622a7e14dcfSSatish Balay         break;
623a7e14dcfSSatish Balay       case TAO_CONVERGED_GTTOL:
624a7e14dcfSSatish Balay         ierr = PetscViewerASCIIPrintf(viewer," ||g(X)||/||g(X0)|| <= gttol\n");CHKERRQ(ierr);
625a7e14dcfSSatish Balay         break;
626a7e14dcfSSatish Balay       case TAO_CONVERGED_STEPTOL:
627a7e14dcfSSatish Balay         ierr = PetscViewerASCIIPrintf(viewer," Steptol -- step size small\n");CHKERRQ(ierr);
628a7e14dcfSSatish Balay         break;
629a7e14dcfSSatish Balay       case TAO_CONVERGED_MINF:
630a7e14dcfSSatish Balay         ierr = PetscViewerASCIIPrintf(viewer," Minf --  f < fmin\n");CHKERRQ(ierr);
631a7e14dcfSSatish Balay         break;
632a7e14dcfSSatish Balay       case TAO_CONVERGED_USER:
633a7e14dcfSSatish Balay         ierr = PetscViewerASCIIPrintf(viewer," User Terminated\n");CHKERRQ(ierr);
634a7e14dcfSSatish Balay         break;
635a7e14dcfSSatish Balay       default:
636a7e14dcfSSatish Balay         ierr = PetscViewerASCIIPrintf(viewer,"\n");CHKERRQ(ierr);
637a7e14dcfSSatish Balay         break;
638a7e14dcfSSatish Balay       }
639a7e14dcfSSatish Balay     } else {
64047cfc4faSBarry Smith       ierr = PetscViewerASCIIPrintf(viewer,"Solver terminated: %d",tao->reason);CHKERRQ(ierr);
641a7e14dcfSSatish Balay       switch (tao->reason) {
642a7e14dcfSSatish Balay       case TAO_DIVERGED_MAXITS:
64347a47007SBarry Smith         ierr = PetscViewerASCIIPrintf(viewer," Maximum Iterations\n");CHKERRQ(ierr);
644a7e14dcfSSatish Balay         break;
645a7e14dcfSSatish Balay       case TAO_DIVERGED_NAN:
64647a47007SBarry Smith         ierr = PetscViewerASCIIPrintf(viewer," NAN or Inf encountered\n");CHKERRQ(ierr);
647a7e14dcfSSatish Balay         break;
648a7e14dcfSSatish Balay       case TAO_DIVERGED_MAXFCN:
64947a47007SBarry Smith         ierr = PetscViewerASCIIPrintf(viewer," Maximum Function Evaluations\n");CHKERRQ(ierr);
650a7e14dcfSSatish Balay         break;
651a7e14dcfSSatish Balay       case TAO_DIVERGED_LS_FAILURE:
65247a47007SBarry Smith         ierr = PetscViewerASCIIPrintf(viewer," Line Search Failure\n");CHKERRQ(ierr);
653a7e14dcfSSatish Balay         break;
654a7e14dcfSSatish Balay       case TAO_DIVERGED_TR_REDUCTION:
65547a47007SBarry Smith         ierr = PetscViewerASCIIPrintf(viewer," Trust Region too small\n");CHKERRQ(ierr);
656a7e14dcfSSatish Balay         break;
657a7e14dcfSSatish Balay       case TAO_DIVERGED_USER:
65847a47007SBarry Smith         ierr = PetscViewerASCIIPrintf(viewer," User Terminated\n");CHKERRQ(ierr);
659a7e14dcfSSatish Balay         break;
660a7e14dcfSSatish Balay       default:
661a7e14dcfSSatish Balay         ierr = PetscViewerASCIIPrintf(viewer,"\n");CHKERRQ(ierr);
662a7e14dcfSSatish Balay         break;
663a7e14dcfSSatish Balay       }
664a7e14dcfSSatish Balay     }
665a7e14dcfSSatish Balay     ierr = PetscViewerASCIIPopTab(viewer);CHKERRQ(ierr);
666a7e14dcfSSatish Balay   } else if (isstring) {
667a7e14dcfSSatish Balay     ierr = TaoGetType(tao,&type);CHKERRQ(ierr);
668a7e14dcfSSatish Balay     ierr = PetscViewerStringSPrintf(viewer," %-3.3s",type);CHKERRQ(ierr);
669a7e14dcfSSatish Balay   }
670a7e14dcfSSatish Balay   PetscFunctionReturn(0);
671a7e14dcfSSatish Balay }
672a7e14dcfSSatish Balay 
673a7e14dcfSSatish Balay /*@
674414d97d3SAlp Dener   TaoSetRecycleHistory - Sets the boolean flag to enable/disable re-using
675414d97d3SAlp Dener   iterate information from the previous TaoSolve(). This feature is disabled by
676414d97d3SAlp Dener   default.
677414d97d3SAlp Dener 
678414d97d3SAlp Dener   For conjugate gradient methods (BNCG), this re-uses the latest search direction
679414d97d3SAlp Dener   from the previous TaoSolve() call when computing the first search direction in a
680414d97d3SAlp Dener   new solution. By default, CG methods set the first search direction to the
681414d97d3SAlp Dener   negative gradient.
682414d97d3SAlp Dener 
683414d97d3SAlp Dener   For quasi-Newton family of methods (BQNLS, BQNKLS, BQNKTR, BQNKTL), this re-uses
684414d97d3SAlp Dener   the accumulated quasi-Newton Hessian approximation from the previous TaoSolve()
685414d97d3SAlp Dener   call. By default, QN family of methods reset the initial Hessian approximation to
686414d97d3SAlp Dener   the identity matrix.
687414d97d3SAlp Dener 
688414d97d3SAlp Dener   For any other algorithm, this setting has no effect.
689414d97d3SAlp Dener 
690414d97d3SAlp Dener   Logically collective on Tao
691414d97d3SAlp Dener 
692414d97d3SAlp Dener   Input Parameters:
693414d97d3SAlp Dener + tao - the Tao context
694414d97d3SAlp Dener - recycle - boolean flag
695414d97d3SAlp Dener 
696414d97d3SAlp Dener   Options Database Keys:
697414d97d3SAlp Dener . -tao_recycle_history
698414d97d3SAlp Dener 
699414d97d3SAlp Dener   Level: intermediate
700414d97d3SAlp Dener 
701414d97d3SAlp Dener .seealso: TaoSetRecycleHistory(), TAOBNCG, TAOBQNLS, TAOBQNKLS, TAOBQNKTR, TAOBQNKTL
702414d97d3SAlp Dener 
703414d97d3SAlp Dener @*/
704414d97d3SAlp Dener PetscErrorCode TaoSetRecycleHistory(Tao tao, PetscBool recycle)
705414d97d3SAlp Dener {
706414d97d3SAlp Dener   PetscFunctionBegin;
707414d97d3SAlp Dener   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
70894511df7SStefano Zampini   PetscValidLogicalCollectiveBool(tao,recycle,2);
709414d97d3SAlp Dener   tao->recycle = recycle;
710414d97d3SAlp Dener   PetscFunctionReturn(0);
711414d97d3SAlp Dener }
712414d97d3SAlp Dener 
713414d97d3SAlp Dener /*@
714414d97d3SAlp Dener   TaoGetRecycleHistory - Retrieve the boolean flag for re-using iterate information
715414d97d3SAlp Dener   from the previous TaoSolve(). This feature is disabled by default.
716414d97d3SAlp Dener 
717414d97d3SAlp Dener   Logically collective on Tao
718414d97d3SAlp Dener 
719414d97d3SAlp Dener   Input Parameters:
720414d97d3SAlp Dener , tao - the Tao context
721414d97d3SAlp Dener 
722414d97d3SAlp Dener   Output Parameters:
723414d97d3SAlp Dener , recycle - boolean flag
724414d97d3SAlp Dener 
725414d97d3SAlp Dener   Options Database Keys:
726414d97d3SAlp Dener . -tao_recycle_history
727414d97d3SAlp Dener 
728414d97d3SAlp Dener   Level: intermediate
729414d97d3SAlp Dener 
730414d97d3SAlp Dener .seealso: TaoGetRecycleHistory(), TAOBNCG, TAOBQNLS, TAOBQNKLS, TAOBQNKTR, TAOBQNKTL
731414d97d3SAlp Dener 
732414d97d3SAlp Dener @*/
733414d97d3SAlp Dener PetscErrorCode TaoGetRecycleHistory(Tao tao, PetscBool *recycle)
734414d97d3SAlp Dener {
735414d97d3SAlp Dener   PetscFunctionBegin;
736414d97d3SAlp Dener   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
73794511df7SStefano Zampini   PetscValidBoolPointer(recycle,2);
738414d97d3SAlp Dener   *recycle = tao->recycle;
739414d97d3SAlp Dener   PetscFunctionReturn(0);
740414d97d3SAlp Dener }
741414d97d3SAlp Dener 
742414d97d3SAlp Dener /*@
743a7e14dcfSSatish Balay   TaoSetTolerances - Sets parameters used in TAO convergence tests
744a7e14dcfSSatish Balay 
745441846f8SBarry Smith   Logically collective on Tao
746a7e14dcfSSatish Balay 
747a7e14dcfSSatish Balay   Input Parameters:
748441846f8SBarry Smith + tao - the Tao context
749a7e14dcfSSatish Balay . gatol - stop if norm of gradient is less than this
750a7e14dcfSSatish Balay . grtol - stop if relative norm of gradient is less than this
751a7e14dcfSSatish Balay - gttol - stop if norm of gradient is reduced by this factor
752a7e14dcfSSatish Balay 
753a7e14dcfSSatish Balay   Options Database Keys:
754e52336cbSBarry Smith + -tao_gatol <gatol> - Sets gatol
755a7e14dcfSSatish Balay . -tao_grtol <grtol> - Sets grtol
756a7e14dcfSSatish Balay - -tao_gttol <gttol> - Sets gttol
757a7e14dcfSSatish Balay 
758a7e14dcfSSatish Balay   Stopping Criteria:
759a7e14dcfSSatish Balay $ ||g(X)||                            <= gatol
760a7e14dcfSSatish Balay $ ||g(X)|| / |f(X)|                   <= grtol
761a7e14dcfSSatish Balay $ ||g(X)|| / ||g(X0)||                <= gttol
762a7e14dcfSSatish Balay 
763a7e14dcfSSatish Balay   Notes:
764a7e14dcfSSatish Balay   Use PETSC_DEFAULT to leave one or more tolerances unchanged.
765a7e14dcfSSatish Balay 
766a7e14dcfSSatish Balay   Level: beginner
767a7e14dcfSSatish Balay 
768a7e14dcfSSatish Balay .seealso: TaoGetTolerances()
769a7e14dcfSSatish Balay 
770a7e14dcfSSatish Balay @*/
771e52336cbSBarry Smith PetscErrorCode TaoSetTolerances(Tao tao, PetscReal gatol, PetscReal grtol, PetscReal gttol)
772a7e14dcfSSatish Balay {
773a7e14dcfSSatish Balay   PetscErrorCode ierr;
77447a47007SBarry Smith 
775a7e14dcfSSatish Balay   PetscFunctionBegin;
776441846f8SBarry Smith   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
77794511df7SStefano Zampini   PetscValidLogicalCollectiveReal(tao,gatol,2);
77894511df7SStefano Zampini   PetscValidLogicalCollectiveReal(tao,grtol,3);
77994511df7SStefano Zampini   PetscValidLogicalCollectiveReal(tao,gttol,4);
780a7e14dcfSSatish Balay 
781a7e14dcfSSatish Balay   if (gatol != PETSC_DEFAULT) {
782a7e14dcfSSatish Balay     if (gatol<0) {
783955c1f14SBarry Smith       ierr = PetscInfo(tao,"Tried to set negative gatol -- ignored.\n");CHKERRQ(ierr);
784a7e14dcfSSatish Balay     } else {
785a7e14dcfSSatish Balay       tao->gatol = PetscMax(0,gatol);
7866552cf8aSJason Sarich       tao->gatol_changed = PETSC_TRUE;
787a7e14dcfSSatish Balay     }
788a7e14dcfSSatish Balay   }
789a7e14dcfSSatish Balay 
790a7e14dcfSSatish Balay   if (grtol != PETSC_DEFAULT) {
791a7e14dcfSSatish Balay     if (grtol<0) {
792955c1f14SBarry Smith       ierr = PetscInfo(tao,"Tried to set negative grtol -- ignored.\n");CHKERRQ(ierr);
793a7e14dcfSSatish Balay     } else {
794a7e14dcfSSatish Balay       tao->grtol = PetscMax(0,grtol);
7956552cf8aSJason Sarich       tao->grtol_changed = PETSC_TRUE;
796a7e14dcfSSatish Balay     }
797a7e14dcfSSatish Balay   }
798a7e14dcfSSatish Balay 
799a7e14dcfSSatish Balay   if (gttol != PETSC_DEFAULT) {
800a7e14dcfSSatish Balay     if (gttol<0) {
801955c1f14SBarry Smith       ierr = PetscInfo(tao,"Tried to set negative gttol -- ignored.\n");CHKERRQ(ierr);
802a7e14dcfSSatish Balay     } else {
803a7e14dcfSSatish Balay       tao->gttol = PetscMax(0,gttol);
8046552cf8aSJason Sarich       tao->gttol_changed = PETSC_TRUE;
805a7e14dcfSSatish Balay     }
806a7e14dcfSSatish Balay   }
807a7e14dcfSSatish Balay   PetscFunctionReturn(0);
808a7e14dcfSSatish Balay }
809a7e14dcfSSatish Balay 
810a7e14dcfSSatish Balay /*@
81158417fe7SBarry Smith   TaoSetConstraintTolerances - Sets constraint tolerance parameters used in TAO  convergence tests
812a7e14dcfSSatish Balay 
813441846f8SBarry Smith   Logically collective on Tao
814a7e14dcfSSatish Balay 
815a7e14dcfSSatish Balay   Input Parameters:
816441846f8SBarry Smith + tao - the Tao context
817e52336cbSBarry Smith . catol - absolute constraint tolerance, constraint norm must be less than catol for used for gatol convergence criteria
818e52336cbSBarry Smith - crtol - relative contraint tolerance, constraint norm must be less than crtol for used for gatol, gttol convergence criteria
819a7e14dcfSSatish Balay 
820a7e14dcfSSatish Balay   Options Database Keys:
821a7e14dcfSSatish Balay + -tao_catol <catol> - Sets catol
822a7e14dcfSSatish Balay - -tao_crtol <crtol> - Sets crtol
823a7e14dcfSSatish Balay 
8246552cf8aSJason Sarich   Notes:
8256552cf8aSJason Sarich   Use PETSC_DEFAULT to leave any tolerance unchanged.
8266552cf8aSJason Sarich 
827a7e14dcfSSatish Balay   Level: intermediate
828a7e14dcfSSatish Balay 
82958417fe7SBarry Smith .seealso: TaoGetTolerances(), TaoGetConstraintTolerances(), TaoSetTolerances()
830a7e14dcfSSatish Balay 
831a7e14dcfSSatish Balay @*/
832441846f8SBarry Smith PetscErrorCode TaoSetConstraintTolerances(Tao tao, PetscReal catol, PetscReal crtol)
833a7e14dcfSSatish Balay {
834a7e14dcfSSatish Balay   PetscErrorCode ierr;
83547a47007SBarry Smith 
836a7e14dcfSSatish Balay   PetscFunctionBegin;
837441846f8SBarry Smith   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
83894511df7SStefano Zampini   PetscValidLogicalCollectiveReal(tao,catol,2);
83994511df7SStefano Zampini   PetscValidLogicalCollectiveReal(tao,crtol,3);
840a7e14dcfSSatish Balay 
841a7e14dcfSSatish Balay   if (catol != PETSC_DEFAULT) {
842a7e14dcfSSatish Balay     if (catol<0) {
843955c1f14SBarry Smith       ierr = PetscInfo(tao,"Tried to set negative catol -- ignored.\n");CHKERRQ(ierr);
844a7e14dcfSSatish Balay     } else {
845a7e14dcfSSatish Balay       tao->catol = PetscMax(0,catol);
8466552cf8aSJason Sarich       tao->catol_changed = PETSC_TRUE;
847a7e14dcfSSatish Balay     }
848a7e14dcfSSatish Balay   }
849a7e14dcfSSatish Balay 
850a7e14dcfSSatish Balay   if (crtol != PETSC_DEFAULT) {
851a7e14dcfSSatish Balay     if (crtol<0) {
852955c1f14SBarry Smith       ierr = PetscInfo(tao,"Tried to set negative crtol -- ignored.\n");CHKERRQ(ierr);
853a7e14dcfSSatish Balay     } else {
854a7e14dcfSSatish Balay       tao->crtol = PetscMax(0,crtol);
8556552cf8aSJason Sarich       tao->crtol_changed = PETSC_TRUE;
856a7e14dcfSSatish Balay     }
857a7e14dcfSSatish Balay   }
858a7e14dcfSSatish Balay   PetscFunctionReturn(0);
859a7e14dcfSSatish Balay }
860a7e14dcfSSatish Balay 
86158417fe7SBarry Smith /*@
86258417fe7SBarry Smith   TaoGetConstraintTolerances - Gets constraint tolerance parameters used in TAO  convergence tests
86358417fe7SBarry Smith 
86458417fe7SBarry Smith   Not ollective
86558417fe7SBarry Smith 
86658417fe7SBarry Smith   Input Parameter:
86758417fe7SBarry Smith . tao - the Tao context
86858417fe7SBarry Smith 
869d8d19677SJose E. Roman   Output Parameters:
870e52336cbSBarry Smith + catol - absolute constraint tolerance, constraint norm must be less than catol for used for gatol convergence criteria
871e52336cbSBarry Smith - crtol - relative contraint tolerance, constraint norm must be less than crtol for used for gatol, gttol convergence criteria
87258417fe7SBarry Smith 
87358417fe7SBarry Smith   Level: intermediate
87458417fe7SBarry Smith 
87558417fe7SBarry Smith .seealso: TaoGetTolerances(), TaoSetTolerances(), TaoSetConstraintTolerances()
87658417fe7SBarry Smith 
87758417fe7SBarry Smith @*/
87858417fe7SBarry Smith PetscErrorCode TaoGetConstraintTolerances(Tao tao, PetscReal *catol, PetscReal *crtol)
87958417fe7SBarry Smith {
88058417fe7SBarry Smith   PetscFunctionBegin;
88158417fe7SBarry Smith   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
88258417fe7SBarry Smith   if (catol) *catol = tao->catol;
88358417fe7SBarry Smith   if (crtol) *crtol = tao->crtol;
88458417fe7SBarry Smith   PetscFunctionReturn(0);
88558417fe7SBarry Smith }
88658417fe7SBarry Smith 
887a7e14dcfSSatish Balay /*@
888a7e14dcfSSatish Balay    TaoSetFunctionLowerBound - Sets a bound on the solution objective value.
889a7e14dcfSSatish Balay    When an approximate solution with an objective value below this number
890a7e14dcfSSatish Balay    has been found, the solver will terminate.
891a7e14dcfSSatish Balay 
892441846f8SBarry Smith    Logically Collective on Tao
893a7e14dcfSSatish Balay 
894a7e14dcfSSatish Balay    Input Parameters:
895441846f8SBarry Smith +  tao - the Tao solver context
896a7e14dcfSSatish Balay -  fmin - the tolerance
897a7e14dcfSSatish Balay 
898a7e14dcfSSatish Balay    Options Database Keys:
899a7e14dcfSSatish Balay .    -tao_fmin <fmin> - sets the minimum function value
900a7e14dcfSSatish Balay 
901a7e14dcfSSatish Balay    Level: intermediate
902a7e14dcfSSatish Balay 
903a7e14dcfSSatish Balay .seealso: TaoSetTolerances()
904a7e14dcfSSatish Balay @*/
905441846f8SBarry Smith PetscErrorCode TaoSetFunctionLowerBound(Tao tao,PetscReal fmin)
906a7e14dcfSSatish Balay {
907a7e14dcfSSatish Balay   PetscFunctionBegin;
908441846f8SBarry Smith   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
90994511df7SStefano Zampini   PetscValidLogicalCollectiveReal(tao,fmin,2);
910a7e14dcfSSatish Balay   tao->fmin = fmin;
9116552cf8aSJason Sarich   tao->fmin_changed = PETSC_TRUE;
912a7e14dcfSSatish Balay   PetscFunctionReturn(0);
913a7e14dcfSSatish Balay }
914a7e14dcfSSatish Balay 
915a7e14dcfSSatish Balay /*@
9168e96d397SJason Sarich    TaoGetFunctionLowerBound - Gets the bound on the solution objective value.
917a7e14dcfSSatish Balay    When an approximate solution with an objective value below this number
918a7e14dcfSSatish Balay    has been found, the solver will terminate.
919a7e14dcfSSatish Balay 
920441846f8SBarry Smith    Not collective on Tao
921a7e14dcfSSatish Balay 
922a7e14dcfSSatish Balay    Input Parameters:
923441846f8SBarry Smith .  tao - the Tao solver context
924a7e14dcfSSatish Balay 
925a7e14dcfSSatish Balay    OutputParameters:
926a7e14dcfSSatish Balay .  fmin - the minimum function value
927a7e14dcfSSatish Balay 
928a7e14dcfSSatish Balay    Level: intermediate
929a7e14dcfSSatish Balay 
930a7e14dcfSSatish Balay .seealso: TaoSetFunctionLowerBound()
931a7e14dcfSSatish Balay @*/
932441846f8SBarry Smith PetscErrorCode TaoGetFunctionLowerBound(Tao tao,PetscReal *fmin)
933a7e14dcfSSatish Balay {
934a7e14dcfSSatish Balay   PetscFunctionBegin;
935441846f8SBarry Smith   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
93694511df7SStefano Zampini   PetscValidRealPointer(fmin,2);
937a7e14dcfSSatish Balay   *fmin = tao->fmin;
938a7e14dcfSSatish Balay   PetscFunctionReturn(0);
939a7e14dcfSSatish Balay }
940a7e14dcfSSatish Balay 
941a7e14dcfSSatish Balay /*@
942a7e14dcfSSatish Balay    TaoSetMaximumFunctionEvaluations - Sets a maximum number of
943a7e14dcfSSatish Balay    function evaluations.
944a7e14dcfSSatish Balay 
945441846f8SBarry Smith    Logically Collective on Tao
946a7e14dcfSSatish Balay 
947a7e14dcfSSatish Balay    Input Parameters:
948441846f8SBarry Smith +  tao - the Tao solver context
949a7e14dcfSSatish Balay -  nfcn - the maximum number of function evaluations (>=0)
950a7e14dcfSSatish Balay 
951a7e14dcfSSatish Balay    Options Database Keys:
952a7e14dcfSSatish Balay .    -tao_max_funcs <nfcn> - sets the maximum number of function evaluations
953a7e14dcfSSatish Balay 
954a7e14dcfSSatish Balay    Level: intermediate
955a7e14dcfSSatish Balay 
956a7e14dcfSSatish Balay .seealso: TaoSetTolerances(), TaoSetMaximumIterations()
957a7e14dcfSSatish Balay @*/
958a7e14dcfSSatish Balay 
959441846f8SBarry Smith PetscErrorCode TaoSetMaximumFunctionEvaluations(Tao tao,PetscInt nfcn)
960a7e14dcfSSatish Balay {
961a7e14dcfSSatish Balay   PetscFunctionBegin;
962441846f8SBarry Smith   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
96394511df7SStefano Zampini   PetscValidLogicalCollectiveInt(tao,nfcn,2);
96494511df7SStefano Zampini   if (nfcn >= 0) { tao->max_funcs = PetscMax(0,nfcn); }
96594511df7SStefano Zampini   else { tao->max_funcs = -1; }
9666552cf8aSJason Sarich   tao->max_funcs_changed = PETSC_TRUE;
967a7e14dcfSSatish Balay   PetscFunctionReturn(0);
968a7e14dcfSSatish Balay }
969a7e14dcfSSatish Balay 
970a7e14dcfSSatish Balay /*@
971a7e14dcfSSatish Balay    TaoGetMaximumFunctionEvaluations - Sets a maximum number of
972a7e14dcfSSatish Balay    function evaluations.
973a7e14dcfSSatish Balay 
974a7e14dcfSSatish Balay    Not Collective
975a7e14dcfSSatish Balay 
976a7e14dcfSSatish Balay    Input Parameters:
977441846f8SBarry Smith .  tao - the Tao solver context
978a7e14dcfSSatish Balay 
979a7e14dcfSSatish Balay    Output Parameters:
980a7e14dcfSSatish Balay .  nfcn - the maximum number of function evaluations
981a7e14dcfSSatish Balay 
982a7e14dcfSSatish Balay    Level: intermediate
983a7e14dcfSSatish Balay 
984a7e14dcfSSatish Balay .seealso: TaoSetMaximumFunctionEvaluations(), TaoGetMaximumIterations()
985a7e14dcfSSatish Balay @*/
986a7e14dcfSSatish Balay 
987441846f8SBarry Smith PetscErrorCode TaoGetMaximumFunctionEvaluations(Tao tao,PetscInt *nfcn)
988a7e14dcfSSatish Balay {
989a7e14dcfSSatish Balay   PetscFunctionBegin;
990441846f8SBarry Smith   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
99194511df7SStefano Zampini   PetscValidIntPointer(nfcn,2);
992a7e14dcfSSatish Balay   *nfcn = tao->max_funcs;
993a7e14dcfSSatish Balay   PetscFunctionReturn(0);
994a7e14dcfSSatish Balay }
995a7e14dcfSSatish Balay 
996770232b9SCe Qin /*@
997770232b9SCe Qin    TaoGetCurrentFunctionEvaluations - Get current number of
998770232b9SCe Qin    function evaluations.
999770232b9SCe Qin 
1000770232b9SCe Qin    Not Collective
1001770232b9SCe Qin 
1002770232b9SCe Qin    Input Parameters:
1003770232b9SCe Qin .  tao - the Tao solver context
1004770232b9SCe Qin 
1005770232b9SCe Qin    Output Parameters:
100694511df7SStefano Zampini .  nfuncs - the current number of function evaluations (maximum between gradient and function evaluations)
1007770232b9SCe Qin 
1008770232b9SCe Qin    Level: intermediate
1009770232b9SCe Qin 
1010770232b9SCe Qin .seealso: TaoSetMaximumFunctionEvaluations(), TaoGetMaximumFunctionEvaluations(), TaoGetMaximumIterations()
1011770232b9SCe Qin @*/
1012770232b9SCe Qin 
1013770232b9SCe Qin PetscErrorCode TaoGetCurrentFunctionEvaluations(Tao tao,PetscInt *nfuncs)
1014770232b9SCe Qin {
1015770232b9SCe Qin   PetscFunctionBegin;
1016770232b9SCe Qin   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
101794511df7SStefano Zampini   PetscValidIntPointer(nfuncs,2);
1018770232b9SCe Qin   *nfuncs = PetscMax(tao->nfuncs,tao->nfuncgrads);
1019770232b9SCe Qin   PetscFunctionReturn(0);
1020770232b9SCe Qin }
1021770232b9SCe Qin 
1022a7e14dcfSSatish Balay /*@
1023a7e14dcfSSatish Balay    TaoSetMaximumIterations - Sets a maximum number of iterates.
1024a7e14dcfSSatish Balay 
1025441846f8SBarry Smith    Logically Collective on Tao
1026a7e14dcfSSatish Balay 
1027a7e14dcfSSatish Balay    Input Parameters:
1028441846f8SBarry Smith +  tao - the Tao solver context
1029a7e14dcfSSatish Balay -  maxits - the maximum number of iterates (>=0)
1030a7e14dcfSSatish Balay 
1031a7e14dcfSSatish Balay    Options Database Keys:
1032a7e14dcfSSatish Balay .    -tao_max_it <its> - sets the maximum number of iterations
1033a7e14dcfSSatish Balay 
1034a7e14dcfSSatish Balay    Level: intermediate
1035a7e14dcfSSatish Balay 
1036a7e14dcfSSatish Balay .seealso: TaoSetTolerances(), TaoSetMaximumFunctionEvaluations()
1037a7e14dcfSSatish Balay @*/
1038441846f8SBarry Smith PetscErrorCode TaoSetMaximumIterations(Tao tao,PetscInt maxits)
1039a7e14dcfSSatish Balay {
1040a7e14dcfSSatish Balay   PetscFunctionBegin;
1041441846f8SBarry Smith   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
104294511df7SStefano Zampini   PetscValidLogicalCollectiveInt(tao,maxits,2);
104347a47007SBarry Smith   tao->max_it = PetscMax(0,maxits);
10446552cf8aSJason Sarich   tao->max_it_changed = PETSC_TRUE;
1045a7e14dcfSSatish Balay   PetscFunctionReturn(0);
1046a7e14dcfSSatish Balay }
1047a7e14dcfSSatish Balay 
1048a7e14dcfSSatish Balay /*@
1049a7e14dcfSSatish Balay    TaoGetMaximumIterations - Sets a maximum number of iterates.
1050a7e14dcfSSatish Balay 
1051a7e14dcfSSatish Balay    Not Collective
1052a7e14dcfSSatish Balay 
1053a7e14dcfSSatish Balay    Input Parameters:
1054441846f8SBarry Smith .  tao - the Tao solver context
1055a7e14dcfSSatish Balay 
1056a7e14dcfSSatish Balay    Output Parameters:
1057a7e14dcfSSatish Balay .  maxits - the maximum number of iterates
1058a7e14dcfSSatish Balay 
1059a7e14dcfSSatish Balay    Level: intermediate
1060a7e14dcfSSatish Balay 
1061a7e14dcfSSatish Balay .seealso: TaoSetMaximumIterations(), TaoGetMaximumFunctionEvaluations()
1062a7e14dcfSSatish Balay @*/
1063441846f8SBarry Smith PetscErrorCode TaoGetMaximumIterations(Tao tao,PetscInt *maxits)
1064a7e14dcfSSatish Balay {
1065a7e14dcfSSatish Balay   PetscFunctionBegin;
1066441846f8SBarry Smith   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
106794511df7SStefano Zampini   PetscValidIntPointer(maxits,2);
1068a7e14dcfSSatish Balay   *maxits = tao->max_it;
1069a7e14dcfSSatish Balay   PetscFunctionReturn(0);
1070a7e14dcfSSatish Balay }
1071a7e14dcfSSatish Balay 
1072a7e14dcfSSatish Balay /*@
1073a7e14dcfSSatish Balay    TaoSetInitialTrustRegionRadius - Sets the initial trust region radius.
1074a7e14dcfSSatish Balay 
1075441846f8SBarry Smith    Logically collective on Tao
1076a7e14dcfSSatish Balay 
1077d8d19677SJose E. Roman    Input Parameters:
1078a7e14dcfSSatish Balay +  tao - a TAO optimization solver
1079a7e14dcfSSatish Balay -  radius - the trust region radius
1080a7e14dcfSSatish Balay 
1081a7e14dcfSSatish Balay    Level: intermediate
1082a7e14dcfSSatish Balay 
1083a7e14dcfSSatish Balay    Options Database Key:
1084a7e14dcfSSatish Balay .  -tao_trust0 <t0> - sets initial trust region radius
1085a7e14dcfSSatish Balay 
1086a7e14dcfSSatish Balay .seealso: TaoGetTrustRegionRadius(), TaoSetTrustRegionTolerance()
1087a7e14dcfSSatish Balay @*/
1088441846f8SBarry Smith PetscErrorCode TaoSetInitialTrustRegionRadius(Tao tao, PetscReal radius)
1089a7e14dcfSSatish Balay {
1090a7e14dcfSSatish Balay   PetscFunctionBegin;
1091441846f8SBarry Smith   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
109294511df7SStefano Zampini   PetscValidLogicalCollectiveReal(tao,radius,2);
1093a7e14dcfSSatish Balay   tao->trust0 = PetscMax(0.0,radius);
10946552cf8aSJason Sarich   tao->trust0_changed = PETSC_TRUE;
1095a7e14dcfSSatish Balay   PetscFunctionReturn(0);
1096a7e14dcfSSatish Balay }
1097a7e14dcfSSatish Balay 
1098a7e14dcfSSatish Balay /*@
1099a7e14dcfSSatish Balay    TaoGetInitialTrustRegionRadius - Sets the initial trust region radius.
1100a7e14dcfSSatish Balay 
1101a7e14dcfSSatish Balay    Not Collective
1102a7e14dcfSSatish Balay 
1103a7e14dcfSSatish Balay    Input Parameter:
1104a7e14dcfSSatish Balay .  tao - a TAO optimization solver
1105a7e14dcfSSatish Balay 
1106a7e14dcfSSatish Balay    Output Parameter:
1107a7e14dcfSSatish Balay .  radius - the trust region radius
1108a7e14dcfSSatish Balay 
1109a7e14dcfSSatish Balay    Level: intermediate
1110a7e14dcfSSatish Balay 
1111a7e14dcfSSatish Balay .seealso: TaoSetInitialTrustRegionRadius(), TaoGetCurrentTrustRegionRadius()
1112a7e14dcfSSatish Balay @*/
1113441846f8SBarry Smith PetscErrorCode TaoGetInitialTrustRegionRadius(Tao tao, PetscReal *radius)
1114a7e14dcfSSatish Balay {
1115a7e14dcfSSatish Balay   PetscFunctionBegin;
1116441846f8SBarry Smith   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
111794511df7SStefano Zampini   PetscValidRealPointer(radius,2);
1118a7e14dcfSSatish Balay   *radius = tao->trust0;
1119a7e14dcfSSatish Balay   PetscFunctionReturn(0);
1120a7e14dcfSSatish Balay }
1121a7e14dcfSSatish Balay 
1122a7e14dcfSSatish Balay /*@
1123a7e14dcfSSatish Balay    TaoGetCurrentTrustRegionRadius - Gets the current trust region radius.
1124a7e14dcfSSatish Balay 
1125a7e14dcfSSatish Balay    Not Collective
1126a7e14dcfSSatish Balay 
1127a7e14dcfSSatish Balay    Input Parameter:
1128a7e14dcfSSatish Balay .  tao - a TAO optimization solver
1129a7e14dcfSSatish Balay 
1130a7e14dcfSSatish Balay    Output Parameter:
1131a7e14dcfSSatish Balay .  radius - the trust region radius
1132a7e14dcfSSatish Balay 
1133a7e14dcfSSatish Balay    Level: intermediate
1134a7e14dcfSSatish Balay 
1135a7e14dcfSSatish Balay .seealso: TaoSetInitialTrustRegionRadius(), TaoGetInitialTrustRegionRadius()
1136a7e14dcfSSatish Balay @*/
1137441846f8SBarry Smith PetscErrorCode TaoGetCurrentTrustRegionRadius(Tao tao, PetscReal *radius)
1138a7e14dcfSSatish Balay {
1139a7e14dcfSSatish Balay   PetscFunctionBegin;
1140441846f8SBarry Smith   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
114194511df7SStefano Zampini   PetscValidRealPointer(radius,2);
1142a7e14dcfSSatish Balay   *radius = tao->trust;
1143a7e14dcfSSatish Balay   PetscFunctionReturn(0);
1144a7e14dcfSSatish Balay }
1145a7e14dcfSSatish Balay 
1146a7e14dcfSSatish Balay /*@
1147a7e14dcfSSatish Balay   TaoGetTolerances - gets the current values of tolerances
1148a7e14dcfSSatish Balay 
1149a7e14dcfSSatish Balay   Not Collective
1150a7e14dcfSSatish Balay 
1151f899ff85SJose E. Roman   Input Parameter:
1152441846f8SBarry Smith . tao - the Tao context
1153a7e14dcfSSatish Balay 
1154a7e14dcfSSatish Balay   Output Parameters:
1155e52336cbSBarry Smith + gatol - stop if norm of gradient is less than this
1156a7e14dcfSSatish Balay . grtol - stop if relative norm of gradient is less than this
1157a7e14dcfSSatish Balay - gttol - stop if norm of gradient is reduced by a this factor
1158a7e14dcfSSatish Balay 
115947a47007SBarry Smith   Note: NULL can be used as an argument if not all tolerances values are needed
1160a7e14dcfSSatish Balay 
1161a7e14dcfSSatish Balay .seealso TaoSetTolerances()
1162a7e14dcfSSatish Balay 
1163a7e14dcfSSatish Balay   Level: intermediate
1164a7e14dcfSSatish Balay @*/
1165e52336cbSBarry Smith PetscErrorCode TaoGetTolerances(Tao tao, PetscReal *gatol, PetscReal *grtol, PetscReal *gttol)
1166a7e14dcfSSatish Balay {
1167a7e14dcfSSatish Balay   PetscFunctionBegin;
1168441846f8SBarry Smith   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
1169a7e14dcfSSatish Balay   if (gatol) *gatol = tao->gatol;
1170a7e14dcfSSatish Balay   if (grtol) *grtol = tao->grtol;
1171a7e14dcfSSatish Balay   if (gttol) *gttol = tao->gttol;
1172a7e14dcfSSatish Balay   PetscFunctionReturn(0);
1173a7e14dcfSSatish Balay }
1174a7e14dcfSSatish Balay 
1175a7e14dcfSSatish Balay /*@
1176a7e14dcfSSatish Balay   TaoGetKSP - Gets the linear solver used by the optimization solver.
1177a7e14dcfSSatish Balay   Application writers should use TaoGetKSP if they need direct access
1178a7e14dcfSSatish Balay   to the PETSc KSP object.
1179a7e14dcfSSatish Balay 
1180a7e14dcfSSatish Balay   Not Collective
1181a7e14dcfSSatish Balay 
1182a7e14dcfSSatish Balay    Input Parameters:
1183a7e14dcfSSatish Balay .  tao - the TAO solver
1184a7e14dcfSSatish Balay 
1185a7e14dcfSSatish Balay    Output Parameters:
1186a7e14dcfSSatish Balay .  ksp - the KSP linear solver used in the optimization solver
1187a7e14dcfSSatish Balay 
1188a7e14dcfSSatish Balay    Level: intermediate
1189a7e14dcfSSatish Balay 
1190a7e14dcfSSatish Balay @*/
1191441846f8SBarry Smith PetscErrorCode TaoGetKSP(Tao tao, KSP *ksp)
119247a47007SBarry Smith {
1193a7e14dcfSSatish Balay   PetscFunctionBegin;
119494511df7SStefano Zampini   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
119594511df7SStefano Zampini   PetscValidPointer(ksp,2);
1196a7e14dcfSSatish Balay   *ksp = tao->ksp;
1197a7e14dcfSSatish Balay   PetscFunctionReturn(0);
1198a7e14dcfSSatish Balay }
1199a7e14dcfSSatish Balay 
1200025e9500SJason Sarich /*@
1201025e9500SJason Sarich    TaoGetLinearSolveIterations - Gets the total number of linear iterations
1202025e9500SJason Sarich    used by the TAO solver
1203025e9500SJason Sarich 
1204025e9500SJason Sarich    Not Collective
1205025e9500SJason Sarich 
1206025e9500SJason Sarich    Input Parameter:
1207025e9500SJason Sarich .  tao - TAO context
1208025e9500SJason Sarich 
1209025e9500SJason Sarich    Output Parameter:
1210025e9500SJason Sarich .  lits - number of linear iterations
1211025e9500SJason Sarich 
1212025e9500SJason Sarich    Notes:
1213025e9500SJason Sarich    This counter is reset to zero for each successive call to TaoSolve()
1214025e9500SJason Sarich 
1215025e9500SJason Sarich    Level: intermediate
1216025e9500SJason Sarich 
1217025e9500SJason Sarich .seealso:  TaoGetKSP()
1218025e9500SJason Sarich @*/
1219025e9500SJason Sarich PetscErrorCode TaoGetLinearSolveIterations(Tao tao, PetscInt *lits)
1220025e9500SJason Sarich {
1221025e9500SJason Sarich   PetscFunctionBegin;
1222025e9500SJason Sarich   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
1223025e9500SJason Sarich   PetscValidIntPointer(lits,2);
12242d9aa51bSJason Sarich   *lits = tao->ksp_tot_its;
1225025e9500SJason Sarich   PetscFunctionReturn(0);
1226025e9500SJason Sarich }
1227025e9500SJason Sarich 
1228a7e14dcfSSatish Balay /*@
1229a7e14dcfSSatish Balay   TaoGetLineSearch - Gets the line search used by the optimization solver.
1230a7e14dcfSSatish Balay   Application writers should use TaoGetLineSearch if they need direct access
1231a7e14dcfSSatish Balay   to the TaoLineSearch object.
1232a7e14dcfSSatish Balay 
1233a7e14dcfSSatish Balay   Not Collective
1234a7e14dcfSSatish Balay 
1235a7e14dcfSSatish Balay    Input Parameters:
1236a7e14dcfSSatish Balay .  tao - the TAO solver
1237a7e14dcfSSatish Balay 
1238a7e14dcfSSatish Balay    Output Parameters:
1239a7e14dcfSSatish Balay .  ls - the line search used in the optimization solver
1240a7e14dcfSSatish Balay 
1241a7e14dcfSSatish Balay    Level: intermediate
1242a7e14dcfSSatish Balay 
1243a7e14dcfSSatish Balay @*/
1244441846f8SBarry Smith PetscErrorCode TaoGetLineSearch(Tao tao, TaoLineSearch *ls)
124547a47007SBarry Smith {
1246a7e14dcfSSatish Balay   PetscFunctionBegin;
124794511df7SStefano Zampini   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
124894511df7SStefano Zampini   PetscValidPointer(ls,2);
1249a7e14dcfSSatish Balay   *ls = tao->linesearch;
1250a7e14dcfSSatish Balay   PetscFunctionReturn(0);
1251a7e14dcfSSatish Balay }
1252a7e14dcfSSatish Balay 
1253a7e14dcfSSatish Balay /*@
1254a7e14dcfSSatish Balay   TaoAddLineSearchCounts - Adds the number of function evaluations spent
1255a7e14dcfSSatish Balay   in the line search to the running total.
1256a7e14dcfSSatish Balay 
1257a7e14dcfSSatish Balay    Input Parameters:
1258a7e14dcfSSatish Balay +  tao - the TAO solver
1259a7e14dcfSSatish Balay -  ls - the line search used in the optimization solver
1260a7e14dcfSSatish Balay 
1261a7e14dcfSSatish Balay    Level: developer
1262a7e14dcfSSatish Balay 
1263a7e14dcfSSatish Balay .seealso: TaoLineSearchApply()
1264a7e14dcfSSatish Balay @*/
1265441846f8SBarry Smith PetscErrorCode TaoAddLineSearchCounts(Tao tao)
126647a47007SBarry Smith {
1267a7e14dcfSSatish Balay   PetscErrorCode ierr;
1268a7e14dcfSSatish Balay   PetscBool      flg;
1269a7e14dcfSSatish Balay   PetscInt       nfeval,ngeval,nfgeval;
127047a47007SBarry Smith 
1271a7e14dcfSSatish Balay   PetscFunctionBegin;
1272441846f8SBarry Smith   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
1273a7e14dcfSSatish Balay   if (tao->linesearch) {
1274302440fdSBarry Smith     ierr = TaoLineSearchIsUsingTaoRoutines(tao->linesearch,&flg);CHKERRQ(ierr);
127594ae4db5SBarry Smith     if (!flg) {
127647a47007SBarry Smith       ierr = TaoLineSearchGetNumberFunctionEvaluations(tao->linesearch,&nfeval,&ngeval,&nfgeval);CHKERRQ(ierr);
1277a7e14dcfSSatish Balay       tao->nfuncs += nfeval;
1278a7e14dcfSSatish Balay       tao->ngrads += ngeval;
1279a7e14dcfSSatish Balay       tao->nfuncgrads += nfgeval;
1280a7e14dcfSSatish Balay     }
1281a7e14dcfSSatish Balay   }
1282a7e14dcfSSatish Balay   PetscFunctionReturn(0);
1283a7e14dcfSSatish Balay }
1284a7e14dcfSSatish Balay 
1285a7e14dcfSSatish Balay /*@
1286a7e14dcfSSatish Balay   TaoGetSolutionVector - Returns the vector with the current TAO solution
1287a7e14dcfSSatish Balay 
1288a7e14dcfSSatish Balay   Not Collective
1289a7e14dcfSSatish Balay 
1290a7e14dcfSSatish Balay   Input Parameter:
1291441846f8SBarry Smith . tao - the Tao context
1292a7e14dcfSSatish Balay 
1293a7e14dcfSSatish Balay   Output Parameter:
1294a7e14dcfSSatish Balay . X - the current solution
1295a7e14dcfSSatish Balay 
1296a7e14dcfSSatish Balay   Level: intermediate
1297a7e14dcfSSatish Balay 
1298a7e14dcfSSatish Balay   Note:  The returned vector will be the same object that was passed into TaoSetInitialVector()
1299a7e14dcfSSatish Balay @*/
1300441846f8SBarry Smith PetscErrorCode TaoGetSolutionVector(Tao tao, Vec *X)
1301a7e14dcfSSatish Balay {
1302a7e14dcfSSatish Balay   PetscFunctionBegin;
1303441846f8SBarry Smith   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
130494511df7SStefano Zampini   PetscValidPointer(X,2);
1305a7e14dcfSSatish Balay   *X = tao->solution;
1306a7e14dcfSSatish Balay   PetscFunctionReturn(0);
1307a7e14dcfSSatish Balay }
1308a7e14dcfSSatish Balay 
1309a7e14dcfSSatish Balay /*@
1310a7e14dcfSSatish Balay   TaoGetGradientVector - Returns the vector with the current TAO gradient
1311a7e14dcfSSatish Balay 
1312a7e14dcfSSatish Balay   Not Collective
1313a7e14dcfSSatish Balay 
1314a7e14dcfSSatish Balay   Input Parameter:
1315441846f8SBarry Smith . tao - the Tao context
1316a7e14dcfSSatish Balay 
1317a7e14dcfSSatish Balay   Output Parameter:
1318a7e14dcfSSatish Balay . G - the current solution
1319a7e14dcfSSatish Balay 
1320a7e14dcfSSatish Balay   Level: intermediate
1321a7e14dcfSSatish Balay @*/
1322441846f8SBarry Smith PetscErrorCode TaoGetGradientVector(Tao tao, Vec *G)
1323a7e14dcfSSatish Balay {
1324a7e14dcfSSatish Balay   PetscFunctionBegin;
1325441846f8SBarry Smith   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
132694511df7SStefano Zampini   PetscValidPointer(G,2);
1327a7e14dcfSSatish Balay   *G = tao->gradient;
1328a7e14dcfSSatish Balay   PetscFunctionReturn(0);
1329a7e14dcfSSatish Balay }
1330a7e14dcfSSatish Balay 
1331a7e14dcfSSatish Balay /*@
1332a7e14dcfSSatish Balay    TaoResetStatistics - Initialize the statistics used by TAO for all of the solvers.
1333a7e14dcfSSatish Balay    These statistics include the iteration number, residual norms, and convergence status.
1334a7e14dcfSSatish Balay    This routine gets called before solving each optimization problem.
1335a7e14dcfSSatish Balay 
1336441846f8SBarry Smith    Collective on Tao
1337a7e14dcfSSatish Balay 
1338a7e14dcfSSatish Balay    Input Parameters:
1339441846f8SBarry Smith .  solver - the Tao context
1340a7e14dcfSSatish Balay 
1341a7e14dcfSSatish Balay    Level: developer
1342a7e14dcfSSatish Balay 
1343a7e14dcfSSatish Balay .seealso: TaoCreate(), TaoSolve()
1344a7e14dcfSSatish Balay @*/
1345441846f8SBarry Smith PetscErrorCode TaoResetStatistics(Tao tao)
1346a7e14dcfSSatish Balay {
1347a7e14dcfSSatish Balay   PetscFunctionBegin;
1348441846f8SBarry Smith   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
1349a7e14dcfSSatish Balay   tao->niter        = 0;
1350a7e14dcfSSatish Balay   tao->nfuncs       = 0;
1351a7e14dcfSSatish Balay   tao->nfuncgrads   = 0;
1352a7e14dcfSSatish Balay   tao->ngrads       = 0;
1353a7e14dcfSSatish Balay   tao->nhess        = 0;
1354a7e14dcfSSatish Balay   tao->njac         = 0;
1355a7e14dcfSSatish Balay   tao->nconstraints = 0;
1356a7e14dcfSSatish Balay   tao->ksp_its      = 0;
1357ae93cb3cSJason Sarich   tao->ksp_tot_its  = 0;
1358a7e14dcfSSatish Balay   tao->reason       = TAO_CONTINUE_ITERATING;
1359a7e14dcfSSatish Balay   tao->residual     = 0.0;
1360a7e14dcfSSatish Balay   tao->cnorm        = 0.0;
1361a7e14dcfSSatish Balay   tao->step         = 0.0;
1362a7e14dcfSSatish Balay   tao->lsflag       = PETSC_FALSE;
1363a7e14dcfSSatish Balay   if (tao->hist_reset) tao->hist_len = 0;
1364a7e14dcfSSatish Balay   PetscFunctionReturn(0);
1365a7e14dcfSSatish Balay }
1366a7e14dcfSSatish Balay 
1367a7e14dcfSSatish Balay /*@C
1368e1e80dc8SAlp Dener   TaoSetUpdate - Sets the general-purpose update function called
1369e1e80dc8SAlp Dener   at the beginning of every iteration of the nonlinear solve. Specifically
1370e1e80dc8SAlp Dener   it is called at the top of every iteration, after the new solution and the gradient
1371e1e80dc8SAlp Dener   is determined, but before the Hessian is computed (if applicable).
1372e1e80dc8SAlp Dener 
1373e1e80dc8SAlp Dener   Logically Collective on Tao
1374e1e80dc8SAlp Dener 
1375e1e80dc8SAlp Dener   Input Parameters:
1376a2b725a8SWilliam Gropp + tao - The tao solver context
1377a2b725a8SWilliam Gropp - func - The function
1378e1e80dc8SAlp Dener 
1379e1e80dc8SAlp Dener   Calling sequence of func:
1380a2b725a8SWilliam Gropp $ func (Tao tao, PetscInt step);
1381e1e80dc8SAlp Dener 
1382e1e80dc8SAlp Dener . step - The current step of the iteration
1383e1e80dc8SAlp Dener 
1384e1e80dc8SAlp Dener   Level: advanced
1385e1e80dc8SAlp Dener 
1386e1e80dc8SAlp Dener .seealso TaoSolve()
1387e1e80dc8SAlp Dener @*/
13888fcddce6SStefano Zampini PetscErrorCode  TaoSetUpdate(Tao tao, PetscErrorCode (*func)(Tao, PetscInt,void*), void *ctx)
1389e1e80dc8SAlp Dener {
1390e1e80dc8SAlp Dener   PetscFunctionBegin;
1391e1e80dc8SAlp Dener   PetscValidHeaderSpecific(tao, TAO_CLASSID,1);
1392e1e80dc8SAlp Dener   tao->ops->update = func;
1393e1e80dc8SAlp Dener   tao->user_update = ctx;
1394e1e80dc8SAlp Dener   PetscFunctionReturn(0);
1395e1e80dc8SAlp Dener }
1396e1e80dc8SAlp Dener 
1397e1e80dc8SAlp Dener /*@C
1398a7e14dcfSSatish Balay   TaoSetConvergenceTest - Sets the function that is to be used to test
1399a7e14dcfSSatish Balay   for convergence o fthe iterative minimization solution.  The new convergence
1400a7e14dcfSSatish Balay   testing routine will replace TAO's default convergence test.
1401a7e14dcfSSatish Balay 
1402441846f8SBarry Smith   Logically Collective on Tao
1403a7e14dcfSSatish Balay 
1404a7e14dcfSSatish Balay   Input Parameters:
1405441846f8SBarry Smith + tao - the Tao object
1406a7e14dcfSSatish Balay . conv - the routine to test for convergence
1407a7e14dcfSSatish Balay - ctx - [optional] context for private data for the convergence routine
140847a47007SBarry Smith         (may be NULL)
1409a7e14dcfSSatish Balay 
1410a7e14dcfSSatish Balay   Calling sequence of conv:
1411441846f8SBarry Smith $   PetscErrorCode conv(Tao tao, void *ctx)
1412a7e14dcfSSatish Balay 
1413441846f8SBarry Smith + tao - the Tao object
1414a7e14dcfSSatish Balay - ctx - [optional] convergence context
1415a7e14dcfSSatish Balay 
1416e4cb33bbSBarry Smith   Note: The new convergence testing routine should call TaoSetConvergedReason().
1417a7e14dcfSSatish Balay 
1418a7e14dcfSSatish Balay   Level: advanced
1419a7e14dcfSSatish Balay 
1420e4cb33bbSBarry Smith .seealso: TaoSetConvergedReason(), TaoGetSolutionStatus(), TaoGetTolerances(), TaoSetMonitor
1421a7e14dcfSSatish Balay 
1422a7e14dcfSSatish Balay @*/
1423441846f8SBarry Smith PetscErrorCode TaoSetConvergenceTest(Tao tao, PetscErrorCode (*conv)(Tao,void*), void *ctx)
1424a7e14dcfSSatish Balay {
1425a7e14dcfSSatish Balay   PetscFunctionBegin;
1426441846f8SBarry Smith   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
142794511df7SStefano Zampini   tao->ops->convergencetest = conv;
142894511df7SStefano Zampini   tao->cnvP = ctx;
1429a7e14dcfSSatish Balay   PetscFunctionReturn(0);
1430a7e14dcfSSatish Balay }
1431a7e14dcfSSatish Balay 
1432a7e14dcfSSatish Balay /*@C
1433a7e14dcfSSatish Balay    TaoSetMonitor - Sets an ADDITIONAL function that is to be used at every
1434a7e14dcfSSatish Balay    iteration of the solver to display the iteration's
1435a7e14dcfSSatish Balay    progress.
1436a7e14dcfSSatish Balay 
1437441846f8SBarry Smith    Logically Collective on Tao
1438a7e14dcfSSatish Balay 
1439a7e14dcfSSatish Balay    Input Parameters:
1440441846f8SBarry Smith +  tao - the Tao solver context
1441a7e14dcfSSatish Balay .  mymonitor - monitoring routine
1442a7e14dcfSSatish Balay -  mctx - [optional] user-defined context for private data for the
144347a47007SBarry Smith           monitor routine (may be NULL)
1444a7e14dcfSSatish Balay 
1445a7e14dcfSSatish Balay    Calling sequence of mymonitor:
14465bd1e576SStefano Zampini $     PetscErrorCode mymonitor(Tao tao,void *mctx)
1447a7e14dcfSSatish Balay 
1448441846f8SBarry Smith +    tao - the Tao solver context
1449a7e14dcfSSatish Balay -    mctx - [optional] monitoring context
1450a7e14dcfSSatish Balay 
1451a7e14dcfSSatish Balay    Options Database Keys:
145298ea980cSBarry Smith +    -tao_monitor        - sets TaoMonitorDefault()
1453a7e14dcfSSatish Balay .    -tao_smonitor       - sets short monitor
1454a7e14dcfSSatish Balay .    -tao_cmonitor       - same as smonitor plus constraint norm
1455a7e14dcfSSatish Balay .    -tao_view_solution   - view solution at each iteration
1456a7e14dcfSSatish Balay .    -tao_view_gradient   - view gradient at each iteration
14574a48860cSAlp Dener .    -tao_view_ls_residual - view least-squares residual vector at each iteration
1458a7e14dcfSSatish Balay -    -tao_cancelmonitors - cancels all monitors that have been hardwired into a code by calls to TaoSetMonitor(), but does not cancel those set via the options database.
1459a7e14dcfSSatish Balay 
1460a7e14dcfSSatish Balay    Notes:
1461a7e14dcfSSatish Balay    Several different monitoring routines may be set by calling
1462a7e14dcfSSatish Balay    TaoSetMonitor() multiple times; all will be called in the
1463a7e14dcfSSatish Balay    order in which they were set.
1464a7e14dcfSSatish Balay 
146595452b02SPatrick Sanan    Fortran Notes:
146695452b02SPatrick Sanan     Only one monitor function may be set
1467a7e14dcfSSatish Balay 
1468a7e14dcfSSatish Balay    Level: intermediate
1469a7e14dcfSSatish Balay 
147098ea980cSBarry Smith .seealso: TaoMonitorDefault(), TaoCancelMonitors(),  TaoSetDestroyRoutine()
1471a7e14dcfSSatish Balay @*/
1472441846f8SBarry Smith PetscErrorCode TaoSetMonitor(Tao tao, PetscErrorCode (*func)(Tao, void*), void *ctx,PetscErrorCode (*dest)(void**))
1473a7e14dcfSSatish Balay {
147408d19d1fSJason Sarich   PetscErrorCode ierr;
147508d19d1fSJason Sarich   PetscInt       i;
14768732526dSAlp Dener   PetscBool      identical;
147708d19d1fSJason Sarich 
1478a7e14dcfSSatish Balay   PetscFunctionBegin;
1479441846f8SBarry Smith   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
1480d8185827SBarry Smith   if (tao->numbermonitors >= MAXTAOMONITORS) SETERRQ1(PetscObjectComm((PetscObject)tao),PETSC_ERR_SUP,"Cannot attach another monitor -- max=%d",MAXTAOMONITORS);
148108d19d1fSJason Sarich 
148208d19d1fSJason Sarich   for (i=0; i<tao->numbermonitors;i++) {
14838732526dSAlp Dener     ierr = PetscMonitorCompare((PetscErrorCode (*)(void))func,ctx,dest,(PetscErrorCode (*)(void))tao->monitor[i],tao->monitorcontext[i],tao->monitordestroy[i],&identical);CHKERRQ(ierr);
14848732526dSAlp Dener     if (identical) PetscFunctionReturn(0);
148508d19d1fSJason Sarich   }
1486a7e14dcfSSatish Balay   tao->monitor[tao->numbermonitors] = func;
14878732526dSAlp Dener   tao->monitorcontext[tao->numbermonitors] = (void*)ctx;
1488a7e14dcfSSatish Balay   tao->monitordestroy[tao->numbermonitors] = dest;
1489a7e14dcfSSatish Balay   ++tao->numbermonitors;
1490a7e14dcfSSatish Balay   PetscFunctionReturn(0);
1491a7e14dcfSSatish Balay }
1492a7e14dcfSSatish Balay 
1493a7e14dcfSSatish Balay /*@
1494441846f8SBarry Smith    TaoCancelMonitors - Clears all the monitor functions for a Tao object.
1495a7e14dcfSSatish Balay 
1496441846f8SBarry Smith    Logically Collective on Tao
1497a7e14dcfSSatish Balay 
1498a7e14dcfSSatish Balay    Input Parameters:
1499441846f8SBarry Smith .  tao - the Tao solver context
1500a7e14dcfSSatish Balay 
1501a7e14dcfSSatish Balay    Options Database:
1502a7e14dcfSSatish Balay .  -tao_cancelmonitors - cancels all monitors that have been hardwired
1503a7e14dcfSSatish Balay     into a code by calls to TaoSetMonitor(), but does not cancel those
1504a7e14dcfSSatish Balay     set via the options database
1505a7e14dcfSSatish Balay 
1506a7e14dcfSSatish Balay    Notes:
1507441846f8SBarry Smith    There is no way to clear one specific monitor from a Tao object.
1508a7e14dcfSSatish Balay 
1509a7e14dcfSSatish Balay    Level: advanced
1510a7e14dcfSSatish Balay 
151198ea980cSBarry Smith .seealso: TaoMonitorDefault(), TaoSetMonitor()
1512a7e14dcfSSatish Balay @*/
1513441846f8SBarry Smith PetscErrorCode TaoCancelMonitors(Tao tao)
1514a7e14dcfSSatish Balay {
1515a7e14dcfSSatish Balay   PetscInt       i;
1516a7e14dcfSSatish Balay   PetscErrorCode ierr;
151747a47007SBarry Smith 
1518a7e14dcfSSatish Balay   PetscFunctionBegin;
1519441846f8SBarry Smith   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
1520a7e14dcfSSatish Balay   for (i=0;i<tao->numbermonitors;i++) {
1521a7e14dcfSSatish Balay     if (tao->monitordestroy[i]) {
1522a7e14dcfSSatish Balay       ierr = (*tao->monitordestroy[i])(&tao->monitorcontext[i]);CHKERRQ(ierr);
1523a7e14dcfSSatish Balay     }
1524a7e14dcfSSatish Balay   }
1525a7e14dcfSSatish Balay   tao->numbermonitors = 0;
1526a7e14dcfSSatish Balay   PetscFunctionReturn(0);
1527a7e14dcfSSatish Balay }
1528a7e14dcfSSatish Balay 
15297fab98ebSJason Sarich /*@
153098ea980cSBarry Smith    TaoMonitorDefault - Default routine for monitoring progress of the
1531441846f8SBarry Smith    Tao solvers (default).  This monitor prints the function value and gradient
1532a7e14dcfSSatish Balay    norm at each iteration.  It can be turned on from the command line using the
1533a7e14dcfSSatish Balay    -tao_monitor option
1534a7e14dcfSSatish Balay 
1535441846f8SBarry Smith    Collective on Tao
1536a7e14dcfSSatish Balay 
1537a7e14dcfSSatish Balay    Input Parameters:
1538441846f8SBarry Smith +  tao - the Tao context
153947a47007SBarry Smith -  ctx - PetscViewer context or NULL
1540a7e14dcfSSatish Balay 
1541a7e14dcfSSatish Balay    Options Database Keys:
1542a7e14dcfSSatish Balay .  -tao_monitor
1543a7e14dcfSSatish Balay 
1544a7e14dcfSSatish Balay    Level: advanced
1545a7e14dcfSSatish Balay 
1546a7e14dcfSSatish Balay .seealso: TaoDefaultSMonitor(), TaoSetMonitor()
1547a7e14dcfSSatish Balay @*/
154898ea980cSBarry Smith PetscErrorCode TaoMonitorDefault(Tao tao, void *ctx)
1549a7e14dcfSSatish Balay {
1550a7e14dcfSSatish Balay   PetscErrorCode ierr;
155163b15415SAlp Dener   PetscInt       its, tabs;
1552a7e14dcfSSatish Balay   PetscReal      fct,gnorm;
15538163d661SBarry Smith   PetscViewer    viewer = (PetscViewer)ctx;
155447a47007SBarry Smith 
1555a7e14dcfSSatish Balay   PetscFunctionBegin;
155694511df7SStefano Zampini   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
15578163d661SBarry Smith   PetscValidHeaderSpecific(viewer,PETSC_VIEWER_CLASSID,2);
1558a7e14dcfSSatish Balay   its = tao->niter;
1559a7e14dcfSSatish Balay   fct = tao->fc;
1560a7e14dcfSSatish Balay   gnorm = tao->residual;
156163b15415SAlp Dener   ierr = PetscViewerASCIIGetTab(viewer, &tabs);CHKERRQ(ierr);
156263b15415SAlp Dener   ierr = PetscViewerASCIISetTab(viewer, ((PetscObject)tao)->tablevel);CHKERRQ(ierr);
1563494bef23SAlp Dener   if (its == 0 && ((PetscObject)tao)->prefix && !tao->header_printed) {
156463b15415SAlp Dener      ierr = PetscViewerASCIIPrintf(viewer,"  Iteration information for %s solve.\n",((PetscObject)tao)->prefix);CHKERRQ(ierr);
1565494bef23SAlp Dener      tao->header_printed = PETSC_TRUE;
156663b15415SAlp Dener    }
156798ea980cSBarry Smith   ierr = PetscViewerASCIIPrintf(viewer,"%3D TAO,",its);CHKERRQ(ierr);
156847a47007SBarry Smith   ierr = PetscViewerASCIIPrintf(viewer,"  Function value: %g,",(double)fct);CHKERRQ(ierr);
156908d19d1fSJason Sarich   if (gnorm >= PETSC_INFINITY) {
157008d19d1fSJason Sarich     ierr = PetscViewerASCIIPrintf(viewer,"  Residual: Inf \n");CHKERRQ(ierr);
157108d19d1fSJason Sarich   } else {
157247a47007SBarry Smith     ierr = PetscViewerASCIIPrintf(viewer,"  Residual: %g \n",(double)gnorm);CHKERRQ(ierr);
157308d19d1fSJason Sarich   }
157463b15415SAlp Dener   ierr = PetscViewerASCIISetTab(viewer, tabs);CHKERRQ(ierr);
1575a7e14dcfSSatish Balay   PetscFunctionReturn(0);
1576a7e14dcfSSatish Balay }
1577a7e14dcfSSatish Balay 
15787fab98ebSJason Sarich /*@
15798d5ead36SAlp Dener    TaoDefaultGMonitor - Default routine for monitoring progress of the
15808d5ead36SAlp Dener    Tao solvers (default) with extra detail on the globalization method.
15818d5ead36SAlp Dener    This monitor prints the function value and gradient norm at each
15828d5ead36SAlp Dener    iteration, as well as the step size and trust radius. Note that the
15838d5ead36SAlp Dener    step size and trust radius may be the same for some algorithms.
15848d5ead36SAlp Dener    It can be turned on from the command line using the
15858d5ead36SAlp Dener    -tao_gmonitor option
15868d5ead36SAlp Dener 
15878d5ead36SAlp Dener    Collective on Tao
15888d5ead36SAlp Dener 
15898d5ead36SAlp Dener    Input Parameters:
15908d5ead36SAlp Dener +  tao - the Tao context
15918d5ead36SAlp Dener -  ctx - PetscViewer context or NULL
15928d5ead36SAlp Dener 
15938d5ead36SAlp Dener    Options Database Keys:
15948d5ead36SAlp Dener .  -tao_monitor
15958d5ead36SAlp Dener 
15968d5ead36SAlp Dener    Level: advanced
15978d5ead36SAlp Dener 
15988d5ead36SAlp Dener .seealso: TaoDefaultSMonitor(), TaoSetMonitor()
15998d5ead36SAlp Dener @*/
16008d5ead36SAlp Dener PetscErrorCode TaoDefaultGMonitor(Tao tao, void *ctx)
16018d5ead36SAlp Dener {
16028d5ead36SAlp Dener   PetscErrorCode ierr;
16038d5ead36SAlp Dener   PetscInt       its, tabs;
16048d5ead36SAlp Dener   PetscReal      fct,gnorm,stp,tr;
16058d5ead36SAlp Dener   PetscViewer    viewer = (PetscViewer)ctx;
16068d5ead36SAlp Dener 
16078d5ead36SAlp Dener   PetscFunctionBegin;
160894511df7SStefano Zampini   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
16098d5ead36SAlp Dener   PetscValidHeaderSpecific(viewer,PETSC_VIEWER_CLASSID,2);
16108d5ead36SAlp Dener   its = tao->niter;
16118d5ead36SAlp Dener   fct = tao->fc;
16128d5ead36SAlp Dener   gnorm = tao->residual;
16138d5ead36SAlp Dener   stp = tao->step;
16148d5ead36SAlp Dener   tr = tao->trust;
16158d5ead36SAlp Dener   ierr = PetscViewerASCIIGetTab(viewer, &tabs);CHKERRQ(ierr);
16168d5ead36SAlp Dener   ierr = PetscViewerASCIISetTab(viewer, ((PetscObject)tao)->tablevel);CHKERRQ(ierr);
1617494bef23SAlp Dener   if (its == 0 && ((PetscObject)tao)->prefix && !tao->header_printed) {
16188d5ead36SAlp Dener      ierr = PetscViewerASCIIPrintf(viewer,"  Iteration information for %s solve.\n",((PetscObject)tao)->prefix);CHKERRQ(ierr);
1619494bef23SAlp Dener      tao->header_printed = PETSC_TRUE;
16208d5ead36SAlp Dener    }
16218d5ead36SAlp Dener   ierr = PetscViewerASCIIPrintf(viewer,"%3D TAO,",its);CHKERRQ(ierr);
16228d5ead36SAlp Dener   ierr = PetscViewerASCIIPrintf(viewer,"  Function value: %g,",(double)fct);CHKERRQ(ierr);
16238d5ead36SAlp Dener   if (gnorm >= PETSC_INFINITY) {
16248d5ead36SAlp Dener     ierr = PetscViewerASCIIPrintf(viewer,"  Residual: Inf,");CHKERRQ(ierr);
16258d5ead36SAlp Dener   } else {
16268d5ead36SAlp Dener     ierr = PetscViewerASCIIPrintf(viewer,"  Residual: %g,",(double)gnorm);CHKERRQ(ierr);
16278d5ead36SAlp Dener   }
16288d5ead36SAlp Dener   ierr = PetscViewerASCIIPrintf(viewer,"  Step: %g,  Trust: %g\n",(double)stp,(double)tr);CHKERRQ(ierr);
16298d5ead36SAlp Dener   ierr = PetscViewerASCIISetTab(viewer, tabs);CHKERRQ(ierr);
16308d5ead36SAlp Dener   PetscFunctionReturn(0);
16318d5ead36SAlp Dener }
16328d5ead36SAlp Dener 
16338d5ead36SAlp Dener /*@
1634a7e14dcfSSatish Balay    TaoDefaultSMonitor - Default routine for monitoring progress of the
163598ea980cSBarry Smith    solver. Same as TaoMonitorDefault() except
1636a7e14dcfSSatish Balay    it prints fewer digits of the residual as the residual gets smaller.
1637a7e14dcfSSatish Balay    This is because the later digits are meaningless and are often
1638a7e14dcfSSatish Balay    different on different machines; by using this routine different
1639a7e14dcfSSatish Balay    machines will usually generate the same output. It can be turned on
1640a7e14dcfSSatish Balay    by using the -tao_smonitor option
1641a7e14dcfSSatish Balay 
1642441846f8SBarry Smith    Collective on Tao
1643a7e14dcfSSatish Balay 
1644a7e14dcfSSatish Balay    Input Parameters:
1645441846f8SBarry Smith +  tao - the Tao context
16464d4332d5SBarry Smith -  ctx - PetscViewer context of type ASCII
1647a7e14dcfSSatish Balay 
1648a7e14dcfSSatish Balay    Options Database Keys:
1649a7e14dcfSSatish Balay .  -tao_smonitor
1650a7e14dcfSSatish Balay 
1651a7e14dcfSSatish Balay    Level: advanced
1652a7e14dcfSSatish Balay 
165398ea980cSBarry Smith .seealso: TaoMonitorDefault(), TaoSetMonitor()
1654a7e14dcfSSatish Balay @*/
1655441846f8SBarry Smith PetscErrorCode TaoDefaultSMonitor(Tao tao, void *ctx)
1656a7e14dcfSSatish Balay {
1657a7e14dcfSSatish Balay   PetscErrorCode ierr;
16582c9b8b83SAlp Dener   PetscInt       its, tabs;
1659a7e14dcfSSatish Balay   PetscReal      fct,gnorm;
16604d4332d5SBarry Smith   PetscViewer    viewer = (PetscViewer)ctx;
166147a47007SBarry Smith 
1662a7e14dcfSSatish Balay   PetscFunctionBegin;
166394511df7SStefano Zampini   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
16644d4332d5SBarry Smith   PetscValidHeaderSpecific(viewer,PETSC_VIEWER_CLASSID,2);
1665a7e14dcfSSatish Balay   its = tao->niter;
1666a7e14dcfSSatish Balay   fct = tao->fc;
1667a7e14dcfSSatish Balay   gnorm = tao->residual;
16682c9b8b83SAlp Dener   ierr = PetscViewerASCIIGetTab(viewer, &tabs);CHKERRQ(ierr);
16692c9b8b83SAlp Dener   ierr = PetscViewerASCIISetTab(viewer, ((PetscObject)tao)->tablevel);CHKERRQ(ierr);
1670a7e14dcfSSatish Balay   ierr = PetscViewerASCIIPrintf(viewer,"iter = %3D,",its);CHKERRQ(ierr);
167147a47007SBarry Smith   ierr = PetscViewerASCIIPrintf(viewer," Function value %g,",(double)fct);CHKERRQ(ierr);
1672d393f493SBarry Smith   if (gnorm >= PETSC_INFINITY) {
167308d19d1fSJason Sarich     ierr = PetscViewerASCIIPrintf(viewer," Residual: Inf \n");CHKERRQ(ierr);
167408d19d1fSJason Sarich   } else if (gnorm > 1.e-6) {
167547a47007SBarry Smith     ierr = PetscViewerASCIIPrintf(viewer," Residual: %g \n",(double)gnorm);CHKERRQ(ierr);
1676a7e14dcfSSatish Balay   } else if (gnorm > 1.e-11) {
1677a7e14dcfSSatish Balay     ierr = PetscViewerASCIIPrintf(viewer," Residual: < 1.0e-6 \n");CHKERRQ(ierr);
1678a7e14dcfSSatish Balay   } else {
1679a7e14dcfSSatish Balay     ierr = PetscViewerASCIIPrintf(viewer," Residual: < 1.0e-11 \n");CHKERRQ(ierr);
1680a7e14dcfSSatish Balay   }
16812c9b8b83SAlp Dener   ierr = PetscViewerASCIISetTab(viewer, tabs);CHKERRQ(ierr);
1682a7e14dcfSSatish Balay   PetscFunctionReturn(0);
1683a7e14dcfSSatish Balay }
1684a7e14dcfSSatish Balay 
16857fab98ebSJason Sarich /*@
168698ea980cSBarry Smith    TaoDefaultCMonitor - same as TaoMonitorDefault() except
1687a7e14dcfSSatish Balay    it prints the norm of the constraints function. It can be turned on
1688a7e14dcfSSatish Balay    from the command line using the -tao_cmonitor option
1689a7e14dcfSSatish Balay 
1690441846f8SBarry Smith    Collective on Tao
1691a7e14dcfSSatish Balay 
1692a7e14dcfSSatish Balay    Input Parameters:
1693441846f8SBarry Smith +  tao - the Tao context
169447a47007SBarry Smith -  ctx - PetscViewer context or NULL
1695a7e14dcfSSatish Balay 
1696a7e14dcfSSatish Balay    Options Database Keys:
1697a7e14dcfSSatish Balay .  -tao_cmonitor
1698a7e14dcfSSatish Balay 
1699a7e14dcfSSatish Balay    Level: advanced
1700a7e14dcfSSatish Balay 
170198ea980cSBarry Smith .seealso: TaoMonitorDefault(), TaoSetMonitor()
1702a7e14dcfSSatish Balay @*/
1703441846f8SBarry Smith PetscErrorCode TaoDefaultCMonitor(Tao tao, void *ctx)
1704a7e14dcfSSatish Balay {
1705a7e14dcfSSatish Balay   PetscErrorCode ierr;
17062c9b8b83SAlp Dener   PetscInt       its, tabs;
1707a7e14dcfSSatish Balay   PetscReal      fct,gnorm;
1708d393f493SBarry Smith   PetscViewer    viewer = (PetscViewer)ctx;
1709a7e14dcfSSatish Balay 
1710a7e14dcfSSatish Balay   PetscFunctionBegin;
171194511df7SStefano Zampini   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
1712d393f493SBarry Smith   PetscValidHeaderSpecific(viewer,PETSC_VIEWER_CLASSID,2);
1713a7e14dcfSSatish Balay   its = tao->niter;
1714a7e14dcfSSatish Balay   fct = tao->fc;
1715a7e14dcfSSatish Balay   gnorm = tao->residual;
17162c9b8b83SAlp Dener   ierr = PetscViewerASCIIGetTab(viewer, &tabs);CHKERRQ(ierr);
17172c9b8b83SAlp Dener   ierr = PetscViewerASCIISetTab(viewer, ((PetscObject)tao)->tablevel);CHKERRQ(ierr);
1718a7e14dcfSSatish Balay   ierr = PetscViewerASCIIPrintf(viewer,"iter = %D,",its);CHKERRQ(ierr);
171947a47007SBarry Smith   ierr = PetscViewerASCIIPrintf(viewer," Function value: %g,",(double)fct);CHKERRQ(ierr);
172047a47007SBarry Smith   ierr = PetscViewerASCIIPrintf(viewer,"  Residual: %g ",(double)gnorm);CHKERRQ(ierr);
172147a47007SBarry Smith   ierr = PetscViewerASCIIPrintf(viewer,"  Constraint: %g \n",(double)tao->cnorm);CHKERRQ(ierr);
17222c9b8b83SAlp Dener   ierr = PetscViewerASCIISetTab(viewer, tabs);CHKERRQ(ierr);
1723a7e14dcfSSatish Balay   PetscFunctionReturn(0);
1724a7e14dcfSSatish Balay }
1725a7e14dcfSSatish Balay 
1726a7e14dcfSSatish Balay /*@C
1727a7e14dcfSSatish Balay    TaoSolutionMonitor - Views the solution at each iteration
1728a7e14dcfSSatish Balay    It can be turned on from the command line using the
1729a7e14dcfSSatish Balay    -tao_view_solution option
1730a7e14dcfSSatish Balay 
1731441846f8SBarry Smith    Collective on Tao
1732a7e14dcfSSatish Balay 
1733a7e14dcfSSatish Balay    Input Parameters:
1734441846f8SBarry Smith +  tao - the Tao context
173547a47007SBarry Smith -  ctx - PetscViewer context or NULL
1736a7e14dcfSSatish Balay 
1737a7e14dcfSSatish Balay    Options Database Keys:
1738a7e14dcfSSatish Balay .  -tao_view_solution
1739a7e14dcfSSatish Balay 
1740a7e14dcfSSatish Balay    Level: advanced
1741a7e14dcfSSatish Balay 
1742a7e14dcfSSatish Balay .seealso: TaoDefaultSMonitor(), TaoSetMonitor()
1743a7e14dcfSSatish Balay @*/
1744441846f8SBarry Smith PetscErrorCode TaoSolutionMonitor(Tao tao, void *ctx)
1745a7e14dcfSSatish Balay {
1746a7e14dcfSSatish Balay   PetscErrorCode ierr;
1747feb237baSPierre Jolivet   PetscViewer    viewer  = (PetscViewer)ctx;
174847a47007SBarry Smith 
1749a7e14dcfSSatish Balay   PetscFunctionBegin;
175094511df7SStefano Zampini   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
1751d393f493SBarry Smith   PetscValidHeaderSpecific(viewer,PETSC_VIEWER_CLASSID,2);
1752a7e14dcfSSatish Balay   ierr = VecView(tao->solution,viewer);CHKERRQ(ierr);
1753a7e14dcfSSatish Balay   PetscFunctionReturn(0);
1754a7e14dcfSSatish Balay }
1755a7e14dcfSSatish Balay 
1756a7e14dcfSSatish Balay /*@C
1757a7e14dcfSSatish Balay    TaoGradientMonitor - Views the gradient at each iteration
1758a7e14dcfSSatish Balay    It can be turned on from the command line using the
1759a7e14dcfSSatish Balay    -tao_view_gradient option
1760a7e14dcfSSatish Balay 
1761441846f8SBarry Smith    Collective on Tao
1762a7e14dcfSSatish Balay 
1763a7e14dcfSSatish Balay    Input Parameters:
1764441846f8SBarry Smith +  tao - the Tao context
176547a47007SBarry Smith -  ctx - PetscViewer context or NULL
1766a7e14dcfSSatish Balay 
1767a7e14dcfSSatish Balay    Options Database Keys:
1768a7e14dcfSSatish Balay .  -tao_view_gradient
1769a7e14dcfSSatish Balay 
1770a7e14dcfSSatish Balay    Level: advanced
1771a7e14dcfSSatish Balay 
1772a7e14dcfSSatish Balay .seealso: TaoDefaultSMonitor(), TaoSetMonitor()
1773a7e14dcfSSatish Balay @*/
1774441846f8SBarry Smith PetscErrorCode TaoGradientMonitor(Tao tao, void *ctx)
1775a7e14dcfSSatish Balay {
1776a7e14dcfSSatish Balay   PetscErrorCode ierr;
1777d393f493SBarry Smith   PetscViewer    viewer = (PetscViewer)ctx;
177847a47007SBarry Smith 
1779a7e14dcfSSatish Balay   PetscFunctionBegin;
178094511df7SStefano Zampini   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
1781d393f493SBarry Smith   PetscValidHeaderSpecific(viewer,PETSC_VIEWER_CLASSID,2);
1782a7e14dcfSSatish Balay   ierr = VecView(tao->gradient,viewer);CHKERRQ(ierr);
1783a7e14dcfSSatish Balay   PetscFunctionReturn(0);
1784a7e14dcfSSatish Balay }
1785a7e14dcfSSatish Balay 
1786a7e14dcfSSatish Balay /*@C
1787a7e14dcfSSatish Balay    TaoStepDirectionMonitor - Views the gradient at each iteration
1788a7e14dcfSSatish Balay    It can be turned on from the command line using the
1789a7e14dcfSSatish Balay    -tao_view_gradient option
1790a7e14dcfSSatish Balay 
1791441846f8SBarry Smith    Collective on Tao
1792a7e14dcfSSatish Balay 
1793a7e14dcfSSatish Balay    Input Parameters:
1794441846f8SBarry Smith +  tao - the Tao context
179547a47007SBarry Smith -  ctx - PetscViewer context or NULL
1796a7e14dcfSSatish Balay 
1797a7e14dcfSSatish Balay    Options Database Keys:
1798a7e14dcfSSatish Balay .  -tao_view_gradient
1799a7e14dcfSSatish Balay 
1800a7e14dcfSSatish Balay    Level: advanced
1801a7e14dcfSSatish Balay 
1802a7e14dcfSSatish Balay .seealso: TaoDefaultSMonitor(), TaoSetMonitor()
1803a7e14dcfSSatish Balay @*/
1804441846f8SBarry Smith PetscErrorCode TaoStepDirectionMonitor(Tao tao, void *ctx)
1805a7e14dcfSSatish Balay {
1806a7e14dcfSSatish Balay   PetscErrorCode ierr;
1807d393f493SBarry Smith   PetscViewer    viewer = (PetscViewer)ctx;
1808d393f493SBarry Smith 
1809a7e14dcfSSatish Balay   PetscFunctionBegin;
181094511df7SStefano Zampini   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
1811d393f493SBarry Smith   PetscValidHeaderSpecific(viewer,PETSC_VIEWER_CLASSID,2);
1812a7e14dcfSSatish Balay   ierr = VecView(tao->stepdirection,viewer);CHKERRQ(ierr);
1813a7e14dcfSSatish Balay   PetscFunctionReturn(0);
1814a7e14dcfSSatish Balay }
1815a7e14dcfSSatish Balay 
1816a7e14dcfSSatish Balay /*@C
1817a7e14dcfSSatish Balay    TaoDrawSolutionMonitor - Plots the solution at each iteration
1818a7e14dcfSSatish Balay    It can be turned on from the command line using the
1819a7e14dcfSSatish Balay    -tao_draw_solution option
1820a7e14dcfSSatish Balay 
1821441846f8SBarry Smith    Collective on Tao
1822a7e14dcfSSatish Balay 
1823a7e14dcfSSatish Balay    Input Parameters:
1824441846f8SBarry Smith +  tao - the Tao context
1825e882e171SHong Zhang -  ctx - TaoMonitorDraw context
1826a7e14dcfSSatish Balay 
1827a7e14dcfSSatish Balay    Options Database Keys:
1828a7e14dcfSSatish Balay .  -tao_draw_solution
1829a7e14dcfSSatish Balay 
1830a7e14dcfSSatish Balay    Level: advanced
1831a7e14dcfSSatish Balay 
1832a7e14dcfSSatish Balay .seealso: TaoSolutionMonitor(), TaoSetMonitor(), TaoDrawGradientMonitor
1833a7e14dcfSSatish Balay @*/
1834441846f8SBarry Smith PetscErrorCode TaoDrawSolutionMonitor(Tao tao, void *ctx)
1835a7e14dcfSSatish Balay {
1836a7e14dcfSSatish Balay   PetscErrorCode    ierr;
1837e882e171SHong Zhang   TaoMonitorDrawCtx ictx = (TaoMonitorDrawCtx)ctx;
183847a47007SBarry Smith 
1839a7e14dcfSSatish Balay   PetscFunctionBegin;
184094511df7SStefano Zampini   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
1841e882e171SHong Zhang   if (!(((ictx->howoften > 0) && (!(tao->niter % ictx->howoften))) || ((ictx->howoften == -1) && tao->reason))) PetscFunctionReturn(0);
1842e882e171SHong Zhang   ierr = VecView(tao->solution,ictx->viewer);CHKERRQ(ierr);
1843a7e14dcfSSatish Balay   PetscFunctionReturn(0);
1844a7e14dcfSSatish Balay }
1845a7e14dcfSSatish Balay 
1846a7e14dcfSSatish Balay /*@C
1847a7e14dcfSSatish Balay    TaoDrawGradientMonitor - Plots the gradient at each iteration
1848a7e14dcfSSatish Balay    It can be turned on from the command line using the
1849a7e14dcfSSatish Balay    -tao_draw_gradient option
1850a7e14dcfSSatish Balay 
1851441846f8SBarry Smith    Collective on Tao
1852a7e14dcfSSatish Balay 
1853a7e14dcfSSatish Balay    Input Parameters:
1854441846f8SBarry Smith +  tao - the Tao context
1855f55353a2SBarry Smith -  ctx - PetscViewer context
1856a7e14dcfSSatish Balay 
1857a7e14dcfSSatish Balay    Options Database Keys:
1858a7e14dcfSSatish Balay .  -tao_draw_gradient
1859a7e14dcfSSatish Balay 
1860a7e14dcfSSatish Balay    Level: advanced
1861a7e14dcfSSatish Balay 
1862a7e14dcfSSatish Balay .seealso: TaoGradientMonitor(), TaoSetMonitor(), TaoDrawSolutionMonitor
1863a7e14dcfSSatish Balay @*/
1864441846f8SBarry Smith PetscErrorCode TaoDrawGradientMonitor(Tao tao, void *ctx)
1865a7e14dcfSSatish Balay {
1866a7e14dcfSSatish Balay   PetscErrorCode    ierr;
1867e882e171SHong Zhang   TaoMonitorDrawCtx ictx = (TaoMonitorDrawCtx)ctx;
186847a47007SBarry Smith 
1869a7e14dcfSSatish Balay   PetscFunctionBegin;
187094511df7SStefano Zampini   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
1871e882e171SHong Zhang   if (!(((ictx->howoften > 0) && (!(tao->niter % ictx->howoften))) || ((ictx->howoften == -1) && tao->reason))) PetscFunctionReturn(0);
1872e882e171SHong Zhang   ierr = VecView(tao->gradient,ictx->viewer);CHKERRQ(ierr);
1873a7e14dcfSSatish Balay   PetscFunctionReturn(0);
1874a7e14dcfSSatish Balay }
1875a7e14dcfSSatish Balay 
1876a7e14dcfSSatish Balay /*@C
1877a7e14dcfSSatish Balay    TaoDrawStepMonitor - Plots the step direction at each iteration
1878a7e14dcfSSatish Balay    It can be turned on from the command line using the
1879a7e14dcfSSatish Balay    -tao_draw_step option
1880a7e14dcfSSatish Balay 
1881441846f8SBarry Smith    Collective on Tao
1882a7e14dcfSSatish Balay 
1883a7e14dcfSSatish Balay    Input Parameters:
1884441846f8SBarry Smith +  tao - the Tao context
1885f55353a2SBarry Smith -  ctx - PetscViewer context
1886a7e14dcfSSatish Balay 
1887a7e14dcfSSatish Balay    Options Database Keys:
1888a7e14dcfSSatish Balay .  -tao_draw_step
1889a7e14dcfSSatish Balay 
1890a7e14dcfSSatish Balay    Level: advanced
1891a7e14dcfSSatish Balay 
1892a7e14dcfSSatish Balay .seealso: TaoSetMonitor(), TaoDrawSolutionMonitor
1893a7e14dcfSSatish Balay @*/
1894441846f8SBarry Smith PetscErrorCode TaoDrawStepMonitor(Tao tao, void *ctx)
1895a7e14dcfSSatish Balay {
1896a7e14dcfSSatish Balay   PetscErrorCode ierr;
189794511df7SStefano Zampini   PetscViewer    viewer = (PetscViewer)ctx;
189847a47007SBarry Smith 
1899a7e14dcfSSatish Balay   PetscFunctionBegin;
190094511df7SStefano Zampini   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
190194511df7SStefano Zampini   PetscValidHeaderSpecific(viewer,PETSC_VIEWER_CLASSID,2);
1902a7e14dcfSSatish Balay   ierr = VecView(tao->stepdirection,viewer);CHKERRQ(ierr);
1903a7e14dcfSSatish Balay   PetscFunctionReturn(0);
1904a7e14dcfSSatish Balay }
1905a7e14dcfSSatish Balay 
1906a7e14dcfSSatish Balay /*@C
1907737f463aSAlp Dener    TaoResidualMonitor - Views the least-squares residual at each iteration
1908a7e14dcfSSatish Balay    It can be turned on from the command line using the
19094a48860cSAlp Dener    -tao_view_ls_residual option
1910a7e14dcfSSatish Balay 
1911441846f8SBarry Smith    Collective on Tao
1912a7e14dcfSSatish Balay 
1913a7e14dcfSSatish Balay    Input Parameters:
1914441846f8SBarry Smith +  tao - the Tao context
191547a47007SBarry Smith -  ctx - PetscViewer context or NULL
1916a7e14dcfSSatish Balay 
1917a7e14dcfSSatish Balay    Options Database Keys:
19184a48860cSAlp Dener .  -tao_view_ls_residual
1919a7e14dcfSSatish Balay 
1920a7e14dcfSSatish Balay    Level: advanced
1921a7e14dcfSSatish Balay 
1922a7e14dcfSSatish Balay .seealso: TaoDefaultSMonitor(), TaoSetMonitor()
1923a7e14dcfSSatish Balay @*/
1924737f463aSAlp Dener PetscErrorCode TaoResidualMonitor(Tao tao, void *ctx)
1925a7e14dcfSSatish Balay {
1926a7e14dcfSSatish Balay   PetscErrorCode ierr;
1927d393f493SBarry Smith   PetscViewer    viewer  = (PetscViewer)ctx;
192847a47007SBarry Smith 
1929a7e14dcfSSatish Balay   PetscFunctionBegin;
193094511df7SStefano Zampini   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
1931d393f493SBarry Smith   PetscValidHeaderSpecific(viewer,PETSC_VIEWER_CLASSID,2);
19324a48860cSAlp Dener   ierr = VecView(tao->ls_res,viewer);CHKERRQ(ierr);
1933a7e14dcfSSatish Balay   PetscFunctionReturn(0);
1934a7e14dcfSSatish Balay }
1935a7e14dcfSSatish Balay 
19367fab98ebSJason Sarich /*@
1937a7e14dcfSSatish Balay    TaoDefaultConvergenceTest - Determines whether the solver should continue iterating
1938a7e14dcfSSatish Balay    or terminate.
1939a7e14dcfSSatish Balay 
1940441846f8SBarry Smith    Collective on Tao
1941a7e14dcfSSatish Balay 
1942a7e14dcfSSatish Balay    Input Parameters:
1943441846f8SBarry Smith +  tao - the Tao context
1944a7e14dcfSSatish Balay -  dummy - unused dummy context
1945a7e14dcfSSatish Balay 
1946a7e14dcfSSatish Balay    Output Parameter:
1947a7e14dcfSSatish Balay .  reason - for terminating
1948a7e14dcfSSatish Balay 
1949a7e14dcfSSatish Balay    Notes:
1950a7e14dcfSSatish Balay    This routine checks the residual in the optimality conditions, the
1951a7e14dcfSSatish Balay    relative residual in the optimity conditions, the number of function
1952a7e14dcfSSatish Balay    evaluations, and the function value to test convergence.  Some
1953a7e14dcfSSatish Balay    solvers may use different convergence routines.
1954a7e14dcfSSatish Balay 
1955a7e14dcfSSatish Balay    Level: developer
1956a7e14dcfSSatish Balay 
1957e4cb33bbSBarry Smith .seealso: TaoSetTolerances(),TaoGetConvergedReason(),TaoSetConvergedReason()
1958a7e14dcfSSatish Balay @*/
1959a7e14dcfSSatish Balay 
1960441846f8SBarry Smith PetscErrorCode TaoDefaultConvergenceTest(Tao tao,void *dummy)
1961a7e14dcfSSatish Balay {
1962a7e14dcfSSatish Balay   PetscInt           niter=tao->niter, nfuncs=PetscMax(tao->nfuncs,tao->nfuncgrads);
1963a7e14dcfSSatish Balay   PetscInt           max_funcs=tao->max_funcs;
1964a7e14dcfSSatish Balay   PetscReal          gnorm=tao->residual, gnorm0=tao->gnorm0;
1965a7e14dcfSSatish Balay   PetscReal          f=tao->fc, steptol=tao->steptol,trradius=tao->step;
1966a7e14dcfSSatish Balay   PetscReal          gatol=tao->gatol,grtol=tao->grtol,gttol=tao->gttol;
1967e52336cbSBarry Smith   PetscReal          catol=tao->catol,crtol=tao->crtol;
1968e52336cbSBarry Smith   PetscReal          fmin=tao->fmin, cnorm=tao->cnorm;
1969e4cb33bbSBarry Smith   TaoConvergedReason reason=tao->reason;
1970a7e14dcfSSatish Balay   PetscErrorCode     ierr;
1971a7e14dcfSSatish Balay 
1972a7e14dcfSSatish Balay   PetscFunctionBegin;
1973441846f8SBarry Smith   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
197494511df7SStefano Zampini   if (reason != TAO_CONTINUE_ITERATING) PetscFunctionReturn(0);
1975a7e14dcfSSatish Balay 
1976a7e14dcfSSatish Balay   if (PetscIsInfOrNanReal(f)) {
1977a7e14dcfSSatish Balay     ierr = PetscInfo(tao,"Failed to converged, function value is Inf or NaN\n");CHKERRQ(ierr);
1978a7e14dcfSSatish Balay     reason = TAO_DIVERGED_NAN;
1979a7e14dcfSSatish Balay   } else if (f <= fmin && cnorm <=catol) {
198047a47007SBarry Smith     ierr = PetscInfo2(tao,"Converged due to function value %g < minimum function value %g\n", (double)f,(double)fmin);CHKERRQ(ierr);
1981a7e14dcfSSatish Balay     reason = TAO_CONVERGED_MINF;
1982a7e14dcfSSatish Balay   } else if (gnorm<= gatol && cnorm <=catol) {
198347a47007SBarry Smith     ierr = PetscInfo2(tao,"Converged due to residual norm ||g(X)||=%g < %g\n",(double)gnorm,(double)gatol);CHKERRQ(ierr);
1984a7e14dcfSSatish Balay     reason = TAO_CONVERGED_GATOL;
1985a7e14dcfSSatish Balay   } else if (f!=0 && PetscAbsReal(gnorm/f) <= grtol && cnorm <= crtol) {
198647a47007SBarry Smith     ierr = PetscInfo2(tao,"Converged due to residual ||g(X)||/|f(X)| =%g < %g\n",(double)(gnorm/f),(double)grtol);CHKERRQ(ierr);
1987a7e14dcfSSatish Balay     reason = TAO_CONVERGED_GRTOL;
1988e1d16bb9SBarry Smith   } else if (gnorm0 != 0 && ((gttol == 0 && gnorm == 0) || gnorm/gnorm0 < gttol) && cnorm <= crtol) {
198947a47007SBarry Smith     ierr = PetscInfo2(tao,"Converged due to relative residual norm ||g(X)||/||g(X0)|| = %g < %g\n",(double)(gnorm/gnorm0),(double)gttol);CHKERRQ(ierr);
1990a7e14dcfSSatish Balay     reason = TAO_CONVERGED_GTTOL;
199194511df7SStefano Zampini   } else if (max_funcs >=0 && nfuncs > max_funcs) {
1992a7e14dcfSSatish Balay     ierr = PetscInfo2(tao,"Exceeded maximum number of function evaluations: %D > %D\n", nfuncs,max_funcs);CHKERRQ(ierr);
1993a7e14dcfSSatish Balay     reason = TAO_DIVERGED_MAXFCN;
1994a7e14dcfSSatish Balay   } else if (tao->lsflag != 0) {
1995a7e14dcfSSatish Balay     ierr = PetscInfo(tao,"Tao Line Search failure.\n");CHKERRQ(ierr);
1996a7e14dcfSSatish Balay     reason = TAO_DIVERGED_LS_FAILURE;
1997a7e14dcfSSatish Balay   } else if (trradius < steptol && niter > 0) {
199847a47007SBarry Smith     ierr = PetscInfo2(tao,"Trust region/step size too small: %g < %g\n", (double)trradius,(double)steptol);CHKERRQ(ierr);
1999a7e14dcfSSatish Balay     reason = TAO_CONVERGED_STEPTOL;
2000e031d6f5SAlp Dener   } else if (niter >= tao->max_it) {
2001a7e14dcfSSatish Balay     ierr = PetscInfo2(tao,"Exceeded maximum number of iterations: %D > %D\n",niter,tao->max_it);CHKERRQ(ierr);
2002a7e14dcfSSatish Balay     reason = TAO_DIVERGED_MAXITS;
2003a7e14dcfSSatish Balay   } else {
2004a7e14dcfSSatish Balay     reason = TAO_CONTINUE_ITERATING;
2005a7e14dcfSSatish Balay   }
2006a7e14dcfSSatish Balay   tao->reason = reason;
2007a7e14dcfSSatish Balay   PetscFunctionReturn(0);
2008a7e14dcfSSatish Balay }
2009a7e14dcfSSatish Balay 
2010a7e14dcfSSatish Balay /*@C
2011a7e14dcfSSatish Balay    TaoSetOptionsPrefix - Sets the prefix used for searching for all
2012a7e14dcfSSatish Balay    TAO options in the database.
2013a7e14dcfSSatish Balay 
2014441846f8SBarry Smith    Logically Collective on Tao
2015a7e14dcfSSatish Balay 
2016a7e14dcfSSatish Balay    Input Parameters:
2017441846f8SBarry Smith +  tao - the Tao context
2018a7e14dcfSSatish Balay -  prefix - the prefix string to prepend to all TAO option requests
2019a7e14dcfSSatish Balay 
2020a7e14dcfSSatish Balay    Notes:
2021a7e14dcfSSatish Balay    A hyphen (-) must NOT be given at the beginning of the prefix name.
2022a7e14dcfSSatish Balay    The first character of all runtime options is AUTOMATICALLY the hyphen.
2023a7e14dcfSSatish Balay 
2024a7e14dcfSSatish Balay    For example, to distinguish between the runtime options for two
2025a7e14dcfSSatish Balay    different TAO solvers, one could call
2026a7e14dcfSSatish Balay .vb
2027a7e14dcfSSatish Balay       TaoSetOptionsPrefix(tao1,"sys1_")
2028a7e14dcfSSatish Balay       TaoSetOptionsPrefix(tao2,"sys2_")
2029a7e14dcfSSatish Balay .ve
2030a7e14dcfSSatish Balay 
2031a7e14dcfSSatish Balay    This would enable use of different options for each system, such as
2032a7e14dcfSSatish Balay .vb
20339fa2b5dcSStefano Zampini       -sys1_tao_method blmvm -sys1_tao_grtol 1.e-3
20349fa2b5dcSStefano Zampini       -sys2_tao_method lmvm  -sys2_tao_grtol 1.e-4
2035a7e14dcfSSatish Balay .ve
2036a7e14dcfSSatish Balay 
2037a7e14dcfSSatish Balay    Level: advanced
2038a7e14dcfSSatish Balay 
2039a7e14dcfSSatish Balay .seealso: TaoAppendOptionsPrefix(), TaoGetOptionsPrefix()
2040a7e14dcfSSatish Balay @*/
2041a7e14dcfSSatish Balay 
2042441846f8SBarry Smith PetscErrorCode TaoSetOptionsPrefix(Tao tao, const char p[])
2043a7e14dcfSSatish Balay {
2044a7e14dcfSSatish Balay   PetscErrorCode ierr;
204547a47007SBarry Smith 
2046a7e14dcfSSatish Balay   PetscFunctionBegin;
204794511df7SStefano Zampini   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
2048a7e14dcfSSatish Balay   ierr = PetscObjectSetOptionsPrefix((PetscObject)tao,p);CHKERRQ(ierr);
2049a7e14dcfSSatish Balay   if (tao->linesearch) {
2050a7e14dcfSSatish Balay     ierr = TaoLineSearchSetOptionsPrefix(tao->linesearch,p);CHKERRQ(ierr);
2051a7e14dcfSSatish Balay   }
2052a7e14dcfSSatish Balay   if (tao->ksp) {
2053a7e14dcfSSatish Balay     ierr = KSPSetOptionsPrefix(tao->ksp,p);CHKERRQ(ierr);
2054a7e14dcfSSatish Balay   }
2055a7e14dcfSSatish Balay   PetscFunctionReturn(0);
2056a7e14dcfSSatish Balay }
2057a7e14dcfSSatish Balay 
2058a7e14dcfSSatish Balay /*@C
2059a7e14dcfSSatish Balay    TaoAppendOptionsPrefix - Appends to the prefix used for searching for all
2060a7e14dcfSSatish Balay    TAO options in the database.
2061a7e14dcfSSatish Balay 
2062441846f8SBarry Smith    Logically Collective on Tao
2063a7e14dcfSSatish Balay 
2064a7e14dcfSSatish Balay    Input Parameters:
2065441846f8SBarry Smith +  tao - the Tao solver context
2066a7e14dcfSSatish Balay -  prefix - the prefix string to prepend to all TAO option requests
2067a7e14dcfSSatish Balay 
2068a7e14dcfSSatish Balay    Notes:
2069a7e14dcfSSatish Balay    A hyphen (-) must NOT be given at the beginning of the prefix name.
2070a7e14dcfSSatish Balay    The first character of all runtime options is AUTOMATICALLY the hyphen.
2071a7e14dcfSSatish Balay 
2072a7e14dcfSSatish Balay    Level: advanced
2073a7e14dcfSSatish Balay 
2074a7e14dcfSSatish Balay .seealso: TaoSetOptionsPrefix(), TaoGetOptionsPrefix()
2075a7e14dcfSSatish Balay @*/
2076441846f8SBarry Smith PetscErrorCode TaoAppendOptionsPrefix(Tao tao, const char p[])
2077a7e14dcfSSatish Balay {
2078a7e14dcfSSatish Balay   PetscErrorCode ierr;
207947a47007SBarry Smith 
2080a7e14dcfSSatish Balay   PetscFunctionBegin;
208194511df7SStefano Zampini   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
2082a7e14dcfSSatish Balay   ierr = PetscObjectAppendOptionsPrefix((PetscObject)tao,p);CHKERRQ(ierr);
2083a7e14dcfSSatish Balay   if (tao->linesearch) {
2084*8ebe3e4eSStefano Zampini     ierr = PetscObjectAppendOptionsPrefix((PetscObject)tao->linesearch,p);CHKERRQ(ierr);
2085a7e14dcfSSatish Balay   }
2086a7e14dcfSSatish Balay   if (tao->ksp) {
2087*8ebe3e4eSStefano Zampini     ierr = KSPAppendOptionsPrefix(tao->ksp,p);CHKERRQ(ierr);
2088a7e14dcfSSatish Balay   }
2089a7e14dcfSSatish Balay   PetscFunctionReturn(0);
2090a7e14dcfSSatish Balay }
2091a7e14dcfSSatish Balay 
2092a7e14dcfSSatish Balay /*@C
2093a7e14dcfSSatish Balay   TaoGetOptionsPrefix - Gets the prefix used for searching for all
2094a7e14dcfSSatish Balay   TAO options in the database
2095a7e14dcfSSatish Balay 
2096a7e14dcfSSatish Balay   Not Collective
2097a7e14dcfSSatish Balay 
2098a7e14dcfSSatish Balay   Input Parameters:
2099441846f8SBarry Smith . tao - the Tao context
2100a7e14dcfSSatish Balay 
2101a7e14dcfSSatish Balay   Output Parameters:
2102a7e14dcfSSatish Balay . prefix - pointer to the prefix string used is returned
2103a7e14dcfSSatish Balay 
210495452b02SPatrick Sanan   Notes:
210595452b02SPatrick Sanan     On the fortran side, the user should pass in a string 'prefix' of
2106a7e14dcfSSatish Balay   sufficient length to hold the prefix.
2107a7e14dcfSSatish Balay 
2108a7e14dcfSSatish Balay   Level: advanced
2109a7e14dcfSSatish Balay 
2110a7e14dcfSSatish Balay .seealso: TaoSetOptionsPrefix(), TaoAppendOptionsPrefix()
2111a7e14dcfSSatish Balay @*/
2112441846f8SBarry Smith PetscErrorCode TaoGetOptionsPrefix(Tao tao, const char *p[])
2113a7e14dcfSSatish Balay {
211494511df7SStefano Zampini   PetscErrorCode ierr;
211594511df7SStefano Zampini 
211694511df7SStefano Zampini   PetscFunctionBegin;
211794511df7SStefano Zampini   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
211894511df7SStefano Zampini   ierr = PetscObjectGetOptionsPrefix((PetscObject)tao,p);CHKERRQ(ierr);
211994511df7SStefano Zampini   PetscFunctionReturn(0);
2120a7e14dcfSSatish Balay }
2121a7e14dcfSSatish Balay 
2122a7e14dcfSSatish Balay /*@C
2123a7e14dcfSSatish Balay    TaoSetType - Sets the method for the unconstrained minimization solver.
2124a7e14dcfSSatish Balay 
2125441846f8SBarry Smith    Collective on Tao
2126a7e14dcfSSatish Balay 
2127a7e14dcfSSatish Balay    Input Parameters:
2128441846f8SBarry Smith +  solver - the Tao solver context
2129a7e14dcfSSatish Balay -  type - a known method
2130a7e14dcfSSatish Balay 
2131a7e14dcfSSatish Balay    Options Database Key:
213258417fe7SBarry Smith .  -tao_type <type> - Sets the method; use -help for a list
213358417fe7SBarry Smith    of available methods (for instance, "-tao_type lmvm" or "-tao_type tron")
2134a7e14dcfSSatish Balay 
2135a7e14dcfSSatish Balay    Available methods include:
213658417fe7SBarry Smith +    nls - Newton's method with line search for unconstrained minimization
213758417fe7SBarry Smith .    ntr - Newton's method with trust region for unconstrained minimization
213858417fe7SBarry Smith .    ntl - Newton's method with trust region, line search for unconstrained minimization
213958417fe7SBarry Smith .    lmvm - Limited memory variable metric method for unconstrained minimization
214058417fe7SBarry Smith .    cg - Nonlinear conjugate gradient method for unconstrained minimization
214158417fe7SBarry Smith .    nm - Nelder-Mead algorithm for derivate-free unconstrained minimization
214258417fe7SBarry Smith .    tron - Newton Trust Region method for bound constrained minimization
214358417fe7SBarry Smith .    gpcg - Newton Trust Region method for quadratic bound constrained minimization
214458417fe7SBarry Smith .    blmvm - Limited memory variable metric method for bound constrained minimization
21451eb8069cSJason Sarich -    pounders - Model-based algorithm pounder extended for nonlinear least squares
2146a7e14dcfSSatish Balay 
2147a7e14dcfSSatish Balay   Level: intermediate
2148a7e14dcfSSatish Balay 
21491eb8069cSJason Sarich .seealso: TaoCreate(), TaoGetType(), TaoType
2150a7e14dcfSSatish Balay 
2151a7e14dcfSSatish Balay @*/
2152b625d6c7SJed Brown PetscErrorCode TaoSetType(Tao tao, TaoType type)
2153a7e14dcfSSatish Balay {
2154a7e14dcfSSatish Balay   PetscErrorCode ierr;
2155441846f8SBarry Smith   PetscErrorCode (*create_xxx)(Tao);
2156a7e14dcfSSatish Balay   PetscBool      issame;
2157a7e14dcfSSatish Balay 
2158a7e14dcfSSatish Balay   PetscFunctionBegin;
2159441846f8SBarry Smith   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
2160a7e14dcfSSatish Balay 
2161a7e14dcfSSatish Balay   ierr = PetscObjectTypeCompare((PetscObject)tao,type,&issame);CHKERRQ(ierr);
2162a7e14dcfSSatish Balay   if (issame) PetscFunctionReturn(0);
2163a7e14dcfSSatish Balay 
2164441846f8SBarry Smith   ierr = PetscFunctionListFind(TaoList, type, (void(**)(void))&create_xxx);CHKERRQ(ierr);
2165691b26d3SBarry Smith   if (!create_xxx) SETERRQ1(PetscObjectComm((PetscObject)tao),PETSC_ERR_ARG_UNKNOWN_TYPE,"Unable to find requested Tao type %s",type);
2166a7e14dcfSSatish Balay 
2167a7e14dcfSSatish Balay   /* Destroy the existing solver information */
2168a7e14dcfSSatish Balay   if (tao->ops->destroy) {
2169a7e14dcfSSatish Balay     ierr = (*tao->ops->destroy)(tao);CHKERRQ(ierr);
2170a7e14dcfSSatish Balay   }
2171a7e14dcfSSatish Balay   ierr = KSPDestroy(&tao->ksp);CHKERRQ(ierr);
2172a7e14dcfSSatish Balay   ierr = TaoLineSearchDestroy(&tao->linesearch);CHKERRQ(ierr);
2173a7e14dcfSSatish Balay   ierr = VecDestroy(&tao->gradient);CHKERRQ(ierr);
2174a7e14dcfSSatish Balay   ierr = VecDestroy(&tao->stepdirection);CHKERRQ(ierr);
2175a7e14dcfSSatish Balay 
217683c8fe1dSLisandro Dalcin   tao->ops->setup = NULL;
217783c8fe1dSLisandro Dalcin   tao->ops->solve = NULL;
217883c8fe1dSLisandro Dalcin   tao->ops->view  = NULL;
217983c8fe1dSLisandro Dalcin   tao->ops->setfromoptions = NULL;
218083c8fe1dSLisandro Dalcin   tao->ops->destroy = NULL;
2181a7e14dcfSSatish Balay 
2182a7e14dcfSSatish Balay   tao->setupcalled = PETSC_FALSE;
2183a7e14dcfSSatish Balay 
2184a7e14dcfSSatish Balay   ierr = (*create_xxx)(tao);CHKERRQ(ierr);
2185a7e14dcfSSatish Balay   ierr = PetscObjectChangeTypeName((PetscObject)tao,type);CHKERRQ(ierr);
2186a7e14dcfSSatish Balay   PetscFunctionReturn(0);
2187a7e14dcfSSatish Balay }
218847a47007SBarry Smith 
2189a7e14dcfSSatish Balay /*MC
2190441846f8SBarry Smith    TaoRegister - Adds a method to the TAO package for unconstrained minimization.
2191a7e14dcfSSatish Balay 
2192a7e14dcfSSatish Balay    Synopsis:
21935bd1e576SStefano Zampini    TaoRegister(char *name_solver,char *path,char *name_Create,PetscErrorCode (*routine_Create)(Tao))
2194a7e14dcfSSatish Balay 
2195a7e14dcfSSatish Balay    Not collective
2196a7e14dcfSSatish Balay 
2197a7e14dcfSSatish Balay    Input Parameters:
2198a7e14dcfSSatish Balay +  sname - name of a new user-defined solver
2199a7e14dcfSSatish Balay -  func - routine to Create method context
2200a7e14dcfSSatish Balay 
2201a7e14dcfSSatish Balay    Notes:
2202441846f8SBarry Smith    TaoRegister() may be called multiple times to add several user-defined solvers.
2203a7e14dcfSSatish Balay 
2204a7e14dcfSSatish Balay    Sample usage:
2205a7e14dcfSSatish Balay .vb
2206441846f8SBarry Smith    TaoRegister("my_solver",MySolverCreate);
2207a7e14dcfSSatish Balay .ve
2208a7e14dcfSSatish Balay 
2209a7e14dcfSSatish Balay    Then, your solver can be chosen with the procedural interface via
2210a7e14dcfSSatish Balay $     TaoSetType(tao,"my_solver")
2211a7e14dcfSSatish Balay    or at runtime via the option
221258417fe7SBarry Smith $     -tao_type my_solver
2213a7e14dcfSSatish Balay 
2214a7e14dcfSSatish Balay    Level: advanced
2215a7e14dcfSSatish Balay 
2216441846f8SBarry Smith .seealso: TaoRegisterAll(), TaoRegisterDestroy()
2217a7e14dcfSSatish Balay M*/
2218441846f8SBarry Smith PetscErrorCode TaoRegister(const char sname[], PetscErrorCode (*func)(Tao))
2219a7e14dcfSSatish Balay {
2220a7e14dcfSSatish Balay   PetscErrorCode ierr;
2221a7e14dcfSSatish Balay 
2222a7e14dcfSSatish Balay   PetscFunctionBegin;
22231d36bdfdSBarry Smith   ierr = TaoInitializePackage();CHKERRQ(ierr);
2224441846f8SBarry Smith   ierr = PetscFunctionListAdd(&TaoList,sname, (void (*)(void))func);CHKERRQ(ierr);
2225a7e14dcfSSatish Balay   PetscFunctionReturn(0);
2226a7e14dcfSSatish Balay }
2227a7e14dcfSSatish Balay 
2228a7e14dcfSSatish Balay /*@C
2229441846f8SBarry Smith    TaoRegisterDestroy - Frees the list of minimization solvers that were
2230441846f8SBarry Smith    registered by TaoRegisterDynamic().
2231a7e14dcfSSatish Balay 
2232a7e14dcfSSatish Balay    Not Collective
2233a7e14dcfSSatish Balay 
2234a7e14dcfSSatish Balay    Level: advanced
2235a7e14dcfSSatish Balay 
2236441846f8SBarry Smith .seealso: TaoRegisterAll(), TaoRegister()
2237a7e14dcfSSatish Balay @*/
2238441846f8SBarry Smith PetscErrorCode TaoRegisterDestroy(void)
2239a7e14dcfSSatish Balay {
2240a7e14dcfSSatish Balay   PetscErrorCode ierr;
224194511df7SStefano Zampini 
2242a7e14dcfSSatish Balay   PetscFunctionBegin;
2243441846f8SBarry Smith   ierr = PetscFunctionListDestroy(&TaoList);CHKERRQ(ierr);
2244441846f8SBarry Smith   TaoRegisterAllCalled = PETSC_FALSE;
2245a7e14dcfSSatish Balay   PetscFunctionReturn(0);
2246a7e14dcfSSatish Balay }
224747a47007SBarry Smith 
22488931d482SJason Sarich /*@
22498931d482SJason Sarich    TaoGetIterationNumber - Gets the number of Tao iterations completed
22508931d482SJason Sarich    at this time.
22518931d482SJason Sarich 
22528931d482SJason Sarich    Not Collective
22538931d482SJason Sarich 
22548931d482SJason Sarich    Input Parameter:
22558931d482SJason Sarich .  tao - Tao context
22568931d482SJason Sarich 
22578931d482SJason Sarich    Output Parameter:
22588931d482SJason Sarich .  iter - iteration number
22598931d482SJason Sarich 
22608931d482SJason Sarich    Notes:
22618931d482SJason Sarich    For example, during the computation of iteration 2 this would return 1.
22628931d482SJason Sarich 
22638931d482SJason Sarich    Level: intermediate
22648931d482SJason Sarich 
226579f5d8caSBarry Smith .seealso:   TaoGetLinearSolveIterations(), TaoGetResidualNorm(), TaoGetObjective()
22668931d482SJason Sarich @*/
22678931d482SJason Sarich PetscErrorCode  TaoGetIterationNumber(Tao tao,PetscInt *iter)
22688931d482SJason Sarich {
22698931d482SJason Sarich   PetscFunctionBegin;
22708931d482SJason Sarich   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
22718931d482SJason Sarich   PetscValidIntPointer(iter,2);
22728931d482SJason Sarich   *iter = tao->niter;
22738931d482SJason Sarich   PetscFunctionReturn(0);
22748931d482SJason Sarich }
22758931d482SJason Sarich 
22768931d482SJason Sarich /*@
227779f5d8caSBarry Smith    TaoGetObjective - Gets the current value of the objective function
227879f5d8caSBarry Smith    at this time.
227979f5d8caSBarry Smith 
228079f5d8caSBarry Smith    Not Collective
228179f5d8caSBarry Smith 
228279f5d8caSBarry Smith    Input Parameter:
228379f5d8caSBarry Smith .  tao - Tao context
228479f5d8caSBarry Smith 
228579f5d8caSBarry Smith    Output Parameter:
228679f5d8caSBarry Smith .  value - the current value
228779f5d8caSBarry Smith 
228879f5d8caSBarry Smith    Level: intermediate
228979f5d8caSBarry Smith 
229079f5d8caSBarry Smith .seealso:   TaoGetLinearSolveIterations(), TaoGetIterationNumber(), TaoGetResidualNorm()
229179f5d8caSBarry Smith @*/
229279f5d8caSBarry Smith PetscErrorCode  TaoGetObjective(Tao tao,PetscReal *value)
229379f5d8caSBarry Smith {
229479f5d8caSBarry Smith   PetscFunctionBegin;
229579f5d8caSBarry Smith   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
229679f5d8caSBarry Smith   PetscValidRealPointer(value,2);
229779f5d8caSBarry Smith   *value = tao->fc;
229879f5d8caSBarry Smith   PetscFunctionReturn(0);
229979f5d8caSBarry Smith }
230079f5d8caSBarry Smith 
230179f5d8caSBarry Smith /*@
230279f5d8caSBarry Smith    TaoGetResidualNorm - Gets the current value of the norm of the residual
230379f5d8caSBarry Smith    at this time.
230479f5d8caSBarry Smith 
230579f5d8caSBarry Smith    Not Collective
230679f5d8caSBarry Smith 
230779f5d8caSBarry Smith    Input Parameter:
230879f5d8caSBarry Smith .  tao - Tao context
230979f5d8caSBarry Smith 
231079f5d8caSBarry Smith    Output Parameter:
231179f5d8caSBarry Smith .  value - the current value
231279f5d8caSBarry Smith 
231379f5d8caSBarry Smith    Level: intermediate
231479f5d8caSBarry Smith 
231579f5d8caSBarry Smith    Developer Note: This is the 2-norm of the residual, we cannot use TaoGetGradientNorm() because that has
231679f5d8caSBarry Smith                    a different meaning. For some reason Tao sometimes calls the gradient the residual.
231779f5d8caSBarry Smith 
231879f5d8caSBarry Smith .seealso:   TaoGetLinearSolveIterations(), TaoGetIterationNumber(), TaoGetObjective()
231979f5d8caSBarry Smith @*/
232079f5d8caSBarry Smith PetscErrorCode  TaoGetResidualNorm(Tao tao,PetscReal *value)
232179f5d8caSBarry Smith {
232279f5d8caSBarry Smith   PetscFunctionBegin;
232379f5d8caSBarry Smith   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
232479f5d8caSBarry Smith   PetscValidRealPointer(value,2);
232579f5d8caSBarry Smith   *value = tao->residual;
232679f5d8caSBarry Smith   PetscFunctionReturn(0);
232779f5d8caSBarry Smith }
232879f5d8caSBarry Smith 
232979f5d8caSBarry Smith /*@
23308931d482SJason Sarich    TaoSetIterationNumber - Sets the current iteration number.
23318931d482SJason Sarich 
233294511df7SStefano Zampini    Logically Collective on Tao
23338931d482SJason Sarich 
2334d8d19677SJose E. Roman    Input Parameters:
2335a2b725a8SWilliam Gropp +  tao - Tao context
2336a2b725a8SWilliam Gropp -  iter - iteration number
23378931d482SJason Sarich 
23388931d482SJason Sarich    Level: developer
23398931d482SJason Sarich 
23408931d482SJason Sarich .seealso:   TaoGetLinearSolveIterations()
23418931d482SJason Sarich @*/
23428931d482SJason Sarich PetscErrorCode  TaoSetIterationNumber(Tao tao,PetscInt iter)
23438931d482SJason Sarich {
23448931d482SJason Sarich   PetscErrorCode ierr;
23458931d482SJason Sarich 
23468931d482SJason Sarich   PetscFunctionBegin;
23478931d482SJason Sarich   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
234894511df7SStefano Zampini   PetscValidLogicalCollectiveInt(tao,iter,2);
23498931d482SJason Sarich   ierr = PetscObjectSAWsTakeAccess((PetscObject)tao);CHKERRQ(ierr);
23508931d482SJason Sarich   tao->niter = iter;
23518931d482SJason Sarich   ierr = PetscObjectSAWsGrantAccess((PetscObject)tao);CHKERRQ(ierr);
23528931d482SJason Sarich   PetscFunctionReturn(0);
23538931d482SJason Sarich }
23548931d482SJason Sarich 
23558931d482SJason Sarich /*@
23568931d482SJason Sarich    TaoGetTotalIterationNumber - Gets the total number of Tao iterations
23578931d482SJason Sarich    completed. This number keeps accumulating if multiple solves
23588931d482SJason Sarich    are called with the Tao object.
23598931d482SJason Sarich 
23608931d482SJason Sarich    Not Collective
23618931d482SJason Sarich 
23628931d482SJason Sarich    Input Parameter:
23638931d482SJason Sarich .  tao - Tao context
23648931d482SJason Sarich 
23658931d482SJason Sarich    Output Parameter:
23668931d482SJason Sarich .  iter - iteration number
23678931d482SJason Sarich 
23688931d482SJason Sarich    Notes:
23698931d482SJason Sarich    The total iteration count is updated after each solve, if there is a current
23708931d482SJason Sarich    TaoSolve() in progress then those iterations are not yet counted.
23718931d482SJason Sarich 
23728931d482SJason Sarich    Level: intermediate
23738931d482SJason Sarich 
23748931d482SJason Sarich .seealso:   TaoGetLinearSolveIterations()
23758931d482SJason Sarich @*/
23768931d482SJason Sarich PetscErrorCode  TaoGetTotalIterationNumber(Tao tao,PetscInt *iter)
23778931d482SJason Sarich {
23788931d482SJason Sarich   PetscFunctionBegin;
23798931d482SJason Sarich   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
23808931d482SJason Sarich   PetscValidIntPointer(iter,2);
23818931d482SJason Sarich   *iter = tao->ntotalits;
23828931d482SJason Sarich   PetscFunctionReturn(0);
23838931d482SJason Sarich }
23848931d482SJason Sarich 
23858931d482SJason Sarich /*@
23868931d482SJason Sarich    TaoSetTotalIterationNumber - Sets the current total iteration number.
23878931d482SJason Sarich 
238894511df7SStefano Zampini    Logically Collective on Tao
23898931d482SJason Sarich 
2390d8d19677SJose E. Roman    Input Parameters:
2391a2b725a8SWilliam Gropp +  tao - Tao context
2392a2b725a8SWilliam Gropp -  iter - iteration number
23938931d482SJason Sarich 
23948931d482SJason Sarich    Level: developer
23958931d482SJason Sarich 
23968931d482SJason Sarich .seealso:   TaoGetLinearSolveIterations()
23978931d482SJason Sarich @*/
23988931d482SJason Sarich PetscErrorCode  TaoSetTotalIterationNumber(Tao tao,PetscInt iter)
23998931d482SJason Sarich {
24008931d482SJason Sarich   PetscErrorCode ierr;
24018931d482SJason Sarich 
24028931d482SJason Sarich   PetscFunctionBegin;
24038931d482SJason Sarich   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
240494511df7SStefano Zampini   PetscValidLogicalCollectiveInt(tao,iter,2);
24058931d482SJason Sarich   ierr = PetscObjectSAWsTakeAccess((PetscObject)tao);CHKERRQ(ierr);
24068931d482SJason Sarich   tao->ntotalits = iter;
24078931d482SJason Sarich   ierr = PetscObjectSAWsGrantAccess((PetscObject)tao);CHKERRQ(ierr);
24088931d482SJason Sarich   PetscFunctionReturn(0);
24098931d482SJason Sarich }
24108931d482SJason Sarich 
2411a7e14dcfSSatish Balay /*@
2412e4cb33bbSBarry Smith   TaoSetConvergedReason - Sets the termination flag on a Tao object
2413a7e14dcfSSatish Balay 
2414441846f8SBarry Smith   Logically Collective on Tao
2415a7e14dcfSSatish Balay 
2416a7e14dcfSSatish Balay   Input Parameters:
2417441846f8SBarry Smith + tao - the Tao context
2418a7e14dcfSSatish Balay - reason - one of
2419a7e14dcfSSatish Balay $     TAO_CONVERGED_ATOL (2),
2420a7e14dcfSSatish Balay $     TAO_CONVERGED_RTOL (3),
2421a7e14dcfSSatish Balay $     TAO_CONVERGED_STEPTOL (4),
2422a7e14dcfSSatish Balay $     TAO_CONVERGED_MINF (5),
2423a7e14dcfSSatish Balay $     TAO_CONVERGED_USER (6),
2424a7e14dcfSSatish Balay $     TAO_DIVERGED_MAXITS (-2),
2425a7e14dcfSSatish Balay $     TAO_DIVERGED_NAN (-4),
2426a7e14dcfSSatish Balay $     TAO_DIVERGED_MAXFCN (-5),
2427a7e14dcfSSatish Balay $     TAO_DIVERGED_LS_FAILURE (-6),
2428a7e14dcfSSatish Balay $     TAO_DIVERGED_TR_REDUCTION (-7),
2429a7e14dcfSSatish Balay $     TAO_DIVERGED_USER (-8),
2430a7e14dcfSSatish Balay $     TAO_CONTINUE_ITERATING (0)
2431a7e14dcfSSatish Balay 
2432a7e14dcfSSatish Balay    Level: intermediate
2433a7e14dcfSSatish Balay 
2434a7e14dcfSSatish Balay @*/
2435e4cb33bbSBarry Smith PetscErrorCode TaoSetConvergedReason(Tao tao, TaoConvergedReason reason)
2436a7e14dcfSSatish Balay {
2437a7e14dcfSSatish Balay   PetscFunctionBegin;
243894511df7SStefano Zampini   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
243994511df7SStefano Zampini   PetscValidLogicalCollectiveEnum(tao,reason,2);
2440a7e14dcfSSatish Balay   tao->reason = reason;
2441a7e14dcfSSatish Balay   PetscFunctionReturn(0);
2442a7e14dcfSSatish Balay }
2443a7e14dcfSSatish Balay 
2444a7e14dcfSSatish Balay /*@
2445e4cb33bbSBarry Smith    TaoGetConvergedReason - Gets the reason the Tao iteration was stopped.
2446a7e14dcfSSatish Balay 
2447a7e14dcfSSatish Balay    Not Collective
2448a7e14dcfSSatish Balay 
2449a7e14dcfSSatish Balay    Input Parameter:
2450441846f8SBarry Smith .  tao - the Tao solver context
2451a7e14dcfSSatish Balay 
2452a7e14dcfSSatish Balay    Output Parameter:
2453a7e14dcfSSatish Balay .  reason - one of
2454a7e14dcfSSatish Balay $  TAO_CONVERGED_GATOL (3)           ||g(X)|| < gatol
2455a7e14dcfSSatish Balay $  TAO_CONVERGED_GRTOL (4)           ||g(X)|| / f(X)  < grtol
2456a7e14dcfSSatish Balay $  TAO_CONVERGED_GTTOL (5)           ||g(X)|| / ||g(X0)|| < gttol
2457a7e14dcfSSatish Balay $  TAO_CONVERGED_STEPTOL (6)         step size small
2458a7e14dcfSSatish Balay $  TAO_CONVERGED_MINF (7)            F < F_min
2459a7e14dcfSSatish Balay $  TAO_CONVERGED_USER (8)            User defined
2460a7e14dcfSSatish Balay $  TAO_DIVERGED_MAXITS (-2)          its > maxits
2461a7e14dcfSSatish Balay $  TAO_DIVERGED_NAN (-4)             Numerical problems
2462a7e14dcfSSatish Balay $  TAO_DIVERGED_MAXFCN (-5)          fevals > max_funcsals
2463a7e14dcfSSatish Balay $  TAO_DIVERGED_LS_FAILURE (-6)      line search failure
2464a7e14dcfSSatish Balay $  TAO_DIVERGED_TR_REDUCTION (-7)    trust region failure
2465a7e14dcfSSatish Balay $  TAO_DIVERGED_USER(-8)             (user defined)
2466a7e14dcfSSatish Balay $  TAO_CONTINUE_ITERATING (0)
2467a7e14dcfSSatish Balay 
2468a7e14dcfSSatish Balay    where
2469a7e14dcfSSatish Balay +  X - current solution
2470a7e14dcfSSatish Balay .  X0 - initial guess
2471a7e14dcfSSatish Balay .  f(X) - current function value
2472a7e14dcfSSatish Balay .  f(X*) - true solution (estimated)
2473a7e14dcfSSatish Balay .  g(X) - current gradient
2474a7e14dcfSSatish Balay .  its - current iterate number
2475a7e14dcfSSatish Balay .  maxits - maximum number of iterates
2476a7e14dcfSSatish Balay .  fevals - number of function evaluations
2477a7e14dcfSSatish Balay -  max_funcsals - maximum number of function evaluations
2478a7e14dcfSSatish Balay 
2479a7e14dcfSSatish Balay    Level: intermediate
2480a7e14dcfSSatish Balay 
2481a7e14dcfSSatish Balay .seealso: TaoSetConvergenceTest(), TaoSetTolerances()
2482a7e14dcfSSatish Balay 
2483a7e14dcfSSatish Balay @*/
2484e4cb33bbSBarry Smith PetscErrorCode TaoGetConvergedReason(Tao tao, TaoConvergedReason *reason)
2485a7e14dcfSSatish Balay {
2486a7e14dcfSSatish Balay   PetscFunctionBegin;
2487441846f8SBarry Smith   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
2488a7e14dcfSSatish Balay   PetscValidPointer(reason,2);
2489a7e14dcfSSatish Balay   *reason = tao->reason;
2490a7e14dcfSSatish Balay   PetscFunctionReturn(0);
2491a7e14dcfSSatish Balay }
2492a7e14dcfSSatish Balay 
2493a7e14dcfSSatish Balay /*@
2494a7e14dcfSSatish Balay   TaoGetSolutionStatus - Get the current iterate, objective value,
2495a7e14dcfSSatish Balay   residual, infeasibility, and termination
2496a7e14dcfSSatish Balay 
2497a7e14dcfSSatish Balay   Not Collective
2498a7e14dcfSSatish Balay 
2499f899ff85SJose E. Roman    Input Parameter:
2500441846f8SBarry Smith .  tao - the Tao context
2501a7e14dcfSSatish Balay 
2502a7e14dcfSSatish Balay    Output Parameters:
2503a7e14dcfSSatish Balay +  iterate - the current iterate number (>=0)
2504a7e14dcfSSatish Balay .  f - the current function value
250558417fe7SBarry Smith .  gnorm - the square of the gradient norm, duality gap, or other measure indicating distance from optimality.
2506a7e14dcfSSatish Balay .  cnorm - the infeasibility of the current solution with regard to the constraints.
2507a7e14dcfSSatish Balay .  xdiff - the step length or trust region radius of the most recent iterate.
2508a7e14dcfSSatish Balay -  reason - The termination reason, which can equal TAO_CONTINUE_ITERATING
2509a7e14dcfSSatish Balay 
2510a7e14dcfSSatish Balay    Level: intermediate
2511a7e14dcfSSatish Balay 
2512a7e14dcfSSatish Balay    Note:
2513a7e14dcfSSatish Balay    TAO returns the values set by the solvers in the routine TaoMonitor().
2514a7e14dcfSSatish Balay 
2515a7e14dcfSSatish Balay    Note:
251647a47007SBarry Smith    If any of the output arguments are set to NULL, no corresponding value will be returned.
2517a7e14dcfSSatish Balay 
2518e4cb33bbSBarry Smith .seealso: TaoMonitor(), TaoGetConvergedReason()
2519a7e14dcfSSatish Balay @*/
2520e4cb33bbSBarry Smith PetscErrorCode TaoGetSolutionStatus(Tao tao, PetscInt *its, PetscReal *f, PetscReal *gnorm, PetscReal *cnorm, PetscReal *xdiff, TaoConvergedReason *reason)
2521a7e14dcfSSatish Balay {
2522a7e14dcfSSatish Balay   PetscFunctionBegin;
252394511df7SStefano Zampini   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
2524a7e14dcfSSatish Balay   if (its) *its = tao->niter;
2525a7e14dcfSSatish Balay   if (f) *f = tao->fc;
2526a7e14dcfSSatish Balay   if (gnorm) *gnorm = tao->residual;
2527a7e14dcfSSatish Balay   if (cnorm) *cnorm = tao->cnorm;
2528a7e14dcfSSatish Balay   if (reason) *reason = tao->reason;
2529a7e14dcfSSatish Balay   if (xdiff) *xdiff = tao->step;
2530a7e14dcfSSatish Balay   PetscFunctionReturn(0);
2531a7e14dcfSSatish Balay }
2532a7e14dcfSSatish Balay 
2533a7e14dcfSSatish Balay /*@C
2534441846f8SBarry Smith    TaoGetType - Gets the current Tao algorithm.
2535a7e14dcfSSatish Balay 
2536a7e14dcfSSatish Balay    Not Collective
2537a7e14dcfSSatish Balay 
2538a7e14dcfSSatish Balay    Input Parameter:
2539441846f8SBarry Smith .  tao - the Tao solver context
2540a7e14dcfSSatish Balay 
2541a7e14dcfSSatish Balay    Output Parameter:
2542441846f8SBarry Smith .  type - Tao method
2543a7e14dcfSSatish Balay 
2544a7e14dcfSSatish Balay    Level: intermediate
2545a7e14dcfSSatish Balay 
2546a7e14dcfSSatish Balay @*/
2547b625d6c7SJed Brown PetscErrorCode TaoGetType(Tao tao,TaoType *type)
2548a7e14dcfSSatish Balay {
2549a7e14dcfSSatish Balay   PetscFunctionBegin;
2550441846f8SBarry Smith   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
2551a7e14dcfSSatish Balay   PetscValidPointer(type,2);
2552a7e14dcfSSatish Balay   *type = ((PetscObject)tao)->type_name;
2553a7e14dcfSSatish Balay   PetscFunctionReturn(0);
2554a7e14dcfSSatish Balay }
2555a7e14dcfSSatish Balay 
2556a7e14dcfSSatish Balay /*@C
2557a7e14dcfSSatish Balay   TaoMonitor - Monitor the solver and the current solution.  This
2558a7e14dcfSSatish Balay   routine will record the iteration number and residual statistics,
2559a7e14dcfSSatish Balay   call any monitors specified by the user, and calls the convergence-check routine.
2560a7e14dcfSSatish Balay 
2561a7e14dcfSSatish Balay    Input Parameters:
2562441846f8SBarry Smith +  tao - the Tao context
2563a7e14dcfSSatish Balay .  its - the current iterate number (>=0)
2564a7e14dcfSSatish Balay .  f - the current objective function value
256558417fe7SBarry Smith .  res - the gradient norm, square root of the duality gap, or other measure indicating distince from optimality.  This measure will be recorded and
2566a7e14dcfSSatish Balay           used for some termination tests.
2567a7e14dcfSSatish Balay .  cnorm - the infeasibility of the current solution with regard to the constraints.
2568a7e14dcfSSatish Balay -  steplength - multiple of the step direction added to the previous iterate.
2569a7e14dcfSSatish Balay 
2570a7e14dcfSSatish Balay    Output Parameters:
2571a7e14dcfSSatish Balay .  reason - The termination reason, which can equal TAO_CONTINUE_ITERATING
2572a7e14dcfSSatish Balay 
2573a7e14dcfSSatish Balay    Options Database Key:
2574a7e14dcfSSatish Balay .  -tao_monitor - Use the default monitor, which prints statistics to standard output
2575a7e14dcfSSatish Balay 
257698ea980cSBarry Smith .seealso TaoGetConvergedReason(), TaoMonitorDefault(), TaoSetMonitor()
2577a7e14dcfSSatish Balay 
2578a7e14dcfSSatish Balay    Level: developer
2579a7e14dcfSSatish Balay 
2580a7e14dcfSSatish Balay @*/
25813ecd9318SAlp Dener PetscErrorCode TaoMonitor(Tao tao, PetscInt its, PetscReal f, PetscReal res, PetscReal cnorm, PetscReal steplength)
2582a7e14dcfSSatish Balay {
2583a7e14dcfSSatish Balay   PetscErrorCode ierr;
258447a47007SBarry Smith   PetscInt       i;
258547a47007SBarry Smith 
2586a7e14dcfSSatish Balay   PetscFunctionBegin;
2587441846f8SBarry Smith   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
2588a7e14dcfSSatish Balay   tao->fc = f;
2589a7e14dcfSSatish Balay   tao->residual = res;
2590a7e14dcfSSatish Balay   tao->cnorm = cnorm;
2591a7e14dcfSSatish Balay   tao->step = steplength;
2592e52336cbSBarry Smith   if (!its) {
259394511df7SStefano Zampini     tao->cnorm0 = cnorm;
259494511df7SStefano Zampini     tao->gnorm0 = res;
2595a7e14dcfSSatish Balay   }
2596691b26d3SBarry Smith   if (PetscIsInfOrNanReal(f) || PetscIsInfOrNanReal(res)) SETERRQ(PetscObjectComm((PetscObject)tao),PETSC_ERR_USER, "User provided compute function generated Inf or NaN");
2597a7e14dcfSSatish Balay   for (i=0;i<tao->numbermonitors;i++) {
2598a7e14dcfSSatish Balay     ierr = (*tao->monitor[i])(tao,tao->monitorcontext[i]);CHKERRQ(ierr);
2599a7e14dcfSSatish Balay   }
2600a7e14dcfSSatish Balay   PetscFunctionReturn(0);
2601a7e14dcfSSatish Balay }
2602a7e14dcfSSatish Balay 
2603a7e14dcfSSatish Balay /*@
2604ae93cb3cSJason Sarich    TaoSetConvergenceHistory - Sets the array used to hold the convergence history.
2605a7e14dcfSSatish Balay 
2606441846f8SBarry Smith    Logically Collective on Tao
2607a7e14dcfSSatish Balay 
2608a7e14dcfSSatish Balay    Input Parameters:
2609441846f8SBarry Smith +  tao - the Tao solver context
2610a7e14dcfSSatish Balay .  obj   - array to hold objective value history
2611a7e14dcfSSatish Balay .  resid - array to hold residual history
2612a7e14dcfSSatish Balay .  cnorm - array to hold constraint violation history
2613be1558d8SJason Sarich .  lits - integer array holds the number of linear iterations for each Tao iteration
2614a7e14dcfSSatish Balay .  na  - size of obj, resid, and cnorm
2615a7e14dcfSSatish Balay -  reset - PetscTrue indicates each new minimization resets the history counter to zero,
2616a7e14dcfSSatish Balay            else it continues storing new values for new minimizations after the old ones
2617a7e14dcfSSatish Balay 
2618a7e14dcfSSatish Balay    Notes:
2619a7e14dcfSSatish Balay    If set, TAO will fill the given arrays with the indicated
2620ae93cb3cSJason Sarich    information at each iteration.  If 'obj','resid','cnorm','lits' are
2621ae93cb3cSJason Sarich    *all* NULL then space (using size na, or 1000 if na is PETSC_DECIDE or
2622ae93cb3cSJason Sarich    PETSC_DEFAULT) is allocated for the history.
2623be1558d8SJason Sarich    If not all are NULL, then only the non-NULL information categories
2624ae93cb3cSJason Sarich    will be stored, the others will be ignored.
2625ae93cb3cSJason Sarich 
2626ae93cb3cSJason Sarich    Any convergence information after iteration number 'na' will not be stored.
2627a7e14dcfSSatish Balay 
2628a7e14dcfSSatish Balay    This routine is useful, e.g., when running a code for purposes
2629a7e14dcfSSatish Balay    of accurate performance monitoring, when no I/O should be done
2630a7e14dcfSSatish Balay    during the section of code that is being timed.
2631a7e14dcfSSatish Balay 
2632a7e14dcfSSatish Balay    Level: intermediate
2633a7e14dcfSSatish Balay 
2634ae93cb3cSJason Sarich .seealso: TaoGetConvergenceHistory()
2635a7e14dcfSSatish Balay 
2636a7e14dcfSSatish Balay @*/
26371b266c99SBarry Smith PetscErrorCode TaoSetConvergenceHistory(Tao tao, PetscReal obj[], PetscReal resid[], PetscReal cnorm[], PetscInt lits[], PetscInt na,PetscBool reset)
2638a7e14dcfSSatish Balay {
2639ae93cb3cSJason Sarich   PetscErrorCode ierr;
2640ae93cb3cSJason Sarich 
2641a7e14dcfSSatish Balay   PetscFunctionBegin;
2642441846f8SBarry Smith   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
2643064a246eSJacob Faibussowitsch   if (obj) PetscValidRealPointer(obj,2);
2644064a246eSJacob Faibussowitsch   if (resid) PetscValidRealPointer(resid,3);
2645064a246eSJacob Faibussowitsch   if (cnorm) PetscValidRealPointer(cnorm,4);
2646ae93cb3cSJason Sarich   if (lits) PetscValidIntPointer(lits,5);
2647ae93cb3cSJason Sarich 
2648ae93cb3cSJason Sarich   if (na == PETSC_DECIDE || na == PETSC_DEFAULT) na = 1000;
2649ae93cb3cSJason Sarich   if (!obj && !resid && !cnorm && !lits) {
2650071fcb05SBarry Smith     ierr = PetscCalloc4(na,&obj,na,&resid,na,&cnorm,na,&lits);CHKERRQ(ierr);
2651ae93cb3cSJason Sarich     tao->hist_malloc = PETSC_TRUE;
2652ae93cb3cSJason Sarich   }
2653ae93cb3cSJason Sarich 
2654a7e14dcfSSatish Balay   tao->hist_obj = obj;
2655a7e14dcfSSatish Balay   tao->hist_resid = resid;
2656a7e14dcfSSatish Balay   tao->hist_cnorm = cnorm;
2657ae93cb3cSJason Sarich   tao->hist_lits = lits;
2658a7e14dcfSSatish Balay   tao->hist_max   = na;
2659a7e14dcfSSatish Balay   tao->hist_reset = reset;
2660ae93cb3cSJason Sarich   tao->hist_len = 0;
2661a7e14dcfSSatish Balay   PetscFunctionReturn(0);
2662a7e14dcfSSatish Balay }
2663a7e14dcfSSatish Balay 
2664a7e14dcfSSatish Balay /*@C
2665ae93cb3cSJason Sarich    TaoGetConvergenceHistory - Gets the arrays used to hold the convergence history.
2666a7e14dcfSSatish Balay 
2667441846f8SBarry Smith    Collective on Tao
2668a7e14dcfSSatish Balay 
2669a7e14dcfSSatish Balay    Input Parameter:
2670441846f8SBarry Smith .  tao - the Tao context
2671a7e14dcfSSatish Balay 
2672a7e14dcfSSatish Balay    Output Parameters:
2673a7e14dcfSSatish Balay +  obj   - array used to hold objective value history
2674a7e14dcfSSatish Balay .  resid - array used to hold residual history
2675a7e14dcfSSatish Balay .  cnorm - array used to hold constraint violation history
2676ae93cb3cSJason Sarich .  lits  - integer array used to hold linear solver iteration count
2677dd25e5eeSStefano Zampini -  nhist  - size of obj, resid, cnorm, and lits
2678a7e14dcfSSatish Balay 
2679a7e14dcfSSatish Balay    Notes:
2680ae93cb3cSJason Sarich     This routine must be preceded by calls to TaoSetConvergenceHistory()
2681ae93cb3cSJason Sarich     and TaoSolve(), otherwise it returns useless information.
2682ae93cb3cSJason Sarich 
2683a7e14dcfSSatish Balay     The calling sequence for this routine in Fortran is
2684ae93cb3cSJason Sarich $   call TaoGetConvergenceHistory(Tao tao, PetscInt nhist, PetscErrorCode ierr)
2685a7e14dcfSSatish Balay 
2686a7e14dcfSSatish Balay    This routine is useful, e.g., when running a code for purposes
2687a7e14dcfSSatish Balay    of accurate performance monitoring, when no I/O should be done
2688a7e14dcfSSatish Balay    during the section of code that is being timed.
2689a7e14dcfSSatish Balay 
2690a7e14dcfSSatish Balay    Level: advanced
2691a7e14dcfSSatish Balay 
2692ae93cb3cSJason Sarich .seealso: TaoSetConvergenceHistory()
2693a7e14dcfSSatish Balay 
2694a7e14dcfSSatish Balay @*/
2695ae93cb3cSJason Sarich PetscErrorCode TaoGetConvergenceHistory(Tao tao, PetscReal **obj, PetscReal **resid, PetscReal **cnorm, PetscInt **lits, PetscInt *nhist)
2696a7e14dcfSSatish Balay {
2697a7e14dcfSSatish Balay   PetscFunctionBegin;
2698441846f8SBarry Smith   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
2699a7e14dcfSSatish Balay   if (obj)   *obj   = tao->hist_obj;
2700a7e14dcfSSatish Balay   if (cnorm) *cnorm = tao->hist_cnorm;
2701a7e14dcfSSatish Balay   if (resid) *resid = tao->hist_resid;
2702a7e14dcfSSatish Balay   if (nhist) *nhist = tao->hist_len;
2703a7e14dcfSSatish Balay   PetscFunctionReturn(0);
2704a7e14dcfSSatish Balay }
2705a7e14dcfSSatish Balay 
2706a7e14dcfSSatish Balay /*@
2707a7e14dcfSSatish Balay    TaoSetApplicationContext - Sets the optional user-defined context for
2708a7e14dcfSSatish Balay    a solver.
2709a7e14dcfSSatish Balay 
2710441846f8SBarry Smith    Logically Collective on Tao
2711a7e14dcfSSatish Balay 
2712a7e14dcfSSatish Balay    Input Parameters:
2713441846f8SBarry Smith +  tao  - the Tao context
2714a7e14dcfSSatish Balay -  usrP - optional user context
2715a7e14dcfSSatish Balay 
2716a7e14dcfSSatish Balay    Level: intermediate
2717a7e14dcfSSatish Balay 
2718a7e14dcfSSatish Balay .seealso: TaoGetApplicationContext(), TaoSetApplicationContext()
2719a7e14dcfSSatish Balay @*/
2720441846f8SBarry Smith PetscErrorCode  TaoSetApplicationContext(Tao tao,void *usrP)
2721a7e14dcfSSatish Balay {
2722a7e14dcfSSatish Balay   PetscFunctionBegin;
2723441846f8SBarry Smith   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
2724a7e14dcfSSatish Balay   tao->user = usrP;
2725a7e14dcfSSatish Balay   PetscFunctionReturn(0);
2726a7e14dcfSSatish Balay }
2727a7e14dcfSSatish Balay 
2728a7e14dcfSSatish Balay /*@
2729a7e14dcfSSatish Balay    TaoGetApplicationContext - Gets the user-defined context for a
2730a7e14dcfSSatish Balay    TAO solvers.
2731a7e14dcfSSatish Balay 
2732a7e14dcfSSatish Balay    Not Collective
2733a7e14dcfSSatish Balay 
2734a7e14dcfSSatish Balay    Input Parameter:
2735441846f8SBarry Smith .  tao  - Tao context
2736a7e14dcfSSatish Balay 
2737a7e14dcfSSatish Balay    Output Parameter:
2738a7e14dcfSSatish Balay .  usrP - user context
2739a7e14dcfSSatish Balay 
2740a7e14dcfSSatish Balay    Level: intermediate
2741a7e14dcfSSatish Balay 
2742a7e14dcfSSatish Balay .seealso: TaoSetApplicationContext()
2743a7e14dcfSSatish Balay @*/
2744441846f8SBarry Smith PetscErrorCode  TaoGetApplicationContext(Tao tao,void *usrP)
2745a7e14dcfSSatish Balay {
2746a7e14dcfSSatish Balay   PetscFunctionBegin;
2747441846f8SBarry Smith   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
274894511df7SStefano Zampini   PetscValidPointer(usrP,2);
2749a7e14dcfSSatish Balay   *(void**)usrP = tao->user;
2750a7e14dcfSSatish Balay   PetscFunctionReturn(0);
2751a7e14dcfSSatish Balay }
2752a9603a14SPatrick Farrell 
2753a9603a14SPatrick Farrell /*@
2754a9603a14SPatrick Farrell    TaoSetGradientNorm - Sets the matrix used to define the inner product that measures the size of the gradient.
2755a9603a14SPatrick Farrell 
2756a9603a14SPatrick Farrell    Collective on tao
2757a9603a14SPatrick Farrell 
2758a9603a14SPatrick Farrell    Input Parameters:
2759a9603a14SPatrick Farrell +  tao  - the Tao context
2760a9603a14SPatrick Farrell -  M    - gradient norm
2761a9603a14SPatrick Farrell 
2762a9603a14SPatrick Farrell    Level: beginner
2763a9603a14SPatrick Farrell 
2764a9603a14SPatrick Farrell .seealso: TaoGetGradientNorm(), TaoGradientNorm()
2765a9603a14SPatrick Farrell @*/
2766a9603a14SPatrick Farrell PetscErrorCode  TaoSetGradientNorm(Tao tao, Mat M)
2767a9603a14SPatrick Farrell {
2768a9603a14SPatrick Farrell   PetscErrorCode ierr;
2769a9603a14SPatrick Farrell 
2770a9603a14SPatrick Farrell   PetscFunctionBegin;
2771a9603a14SPatrick Farrell   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
277294511df7SStefano Zampini   PetscValidHeaderSpecific(M,MAT_CLASSID,2);
2773a9603a14SPatrick Farrell   ierr = PetscObjectReference((PetscObject)M);CHKERRQ(ierr);
27748854b543SStefano Zampini   ierr = MatDestroy(&tao->gradient_norm);CHKERRQ(ierr);
27758854b543SStefano Zampini   ierr = VecDestroy(&tao->gradient_norm_tmp);CHKERRQ(ierr);
2776a9603a14SPatrick Farrell   tao->gradient_norm = M;
2777a9603a14SPatrick Farrell   ierr = MatCreateVecs(M, NULL, &tao->gradient_norm_tmp);CHKERRQ(ierr);
2778a9603a14SPatrick Farrell   PetscFunctionReturn(0);
2779a9603a14SPatrick Farrell }
2780a9603a14SPatrick Farrell 
2781a9603a14SPatrick Farrell /*@
2782a9603a14SPatrick Farrell    TaoGetGradientNorm - Returns the matrix used to define the inner product for measuring the size of the gradient.
2783a9603a14SPatrick Farrell 
2784a9603a14SPatrick Farrell    Not Collective
2785a9603a14SPatrick Farrell 
2786a9603a14SPatrick Farrell    Input Parameter:
2787a9603a14SPatrick Farrell .  tao  - Tao context
2788a9603a14SPatrick Farrell 
2789a9603a14SPatrick Farrell    Output Parameter:
2790a9603a14SPatrick Farrell .  M - gradient norm
2791a9603a14SPatrick Farrell 
2792a9603a14SPatrick Farrell    Level: beginner
2793a9603a14SPatrick Farrell 
2794a9603a14SPatrick Farrell .seealso: TaoSetGradientNorm(), TaoGradientNorm()
2795a9603a14SPatrick Farrell @*/
2796a9603a14SPatrick Farrell PetscErrorCode  TaoGetGradientNorm(Tao tao, Mat *M)
2797a9603a14SPatrick Farrell {
2798a9603a14SPatrick Farrell   PetscFunctionBegin;
2799a9603a14SPatrick Farrell   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
280094511df7SStefano Zampini   PetscValidPointer(M,2);
2801a9603a14SPatrick Farrell   *M = tao->gradient_norm;
2802a9603a14SPatrick Farrell   PetscFunctionReturn(0);
2803a9603a14SPatrick Farrell }
2804a9603a14SPatrick Farrell 
280505aca2cdSSatish Balay /*@C
2806a9603a14SPatrick Farrell    TaoGradientNorm - Compute the norm with respect to the inner product the user has set.
2807a9603a14SPatrick Farrell 
2808a9603a14SPatrick Farrell    Collective on tao
2809a9603a14SPatrick Farrell 
2810d8d19677SJose E. Roman    Input Parameters:
28112653c764SPierre Jolivet +  tao      - the Tao context
2812a9603a14SPatrick Farrell .  gradient - the gradient to be computed
28132653c764SPierre Jolivet -  norm     - the norm type
2814a9603a14SPatrick Farrell 
2815a9603a14SPatrick Farrell    Output Parameter:
2816a9603a14SPatrick Farrell .  gnorm    - the gradient norm
2817a9603a14SPatrick Farrell 
2818a9603a14SPatrick Farrell    Level: developer
2819a9603a14SPatrick Farrell 
2820a9603a14SPatrick Farrell .seealso: TaoSetGradientNorm(), TaoGetGradientNorm()
2821a9603a14SPatrick Farrell @*/
2822a9603a14SPatrick Farrell PetscErrorCode  TaoGradientNorm(Tao tao, Vec gradient, NormType type, PetscReal *gnorm)
2823a9603a14SPatrick Farrell {
2824a9603a14SPatrick Farrell   PetscErrorCode ierr;
2825a9603a14SPatrick Farrell 
2826a9603a14SPatrick Farrell   PetscFunctionBegin;
28278854b543SStefano Zampini   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
28288854b543SStefano Zampini   PetscValidHeaderSpecific(gradient,VEC_CLASSID,2);
28298854b543SStefano Zampini   PetscValidLogicalCollectiveEnum(tao,type,3);
283094511df7SStefano Zampini   PetscValidRealPointer(gnorm,4);
2831a9603a14SPatrick Farrell   if (tao->gradient_norm) {
2832680e2bc7SPatrick Farrell     PetscScalar gnorms;
2833680e2bc7SPatrick Farrell 
28348854b543SStefano Zampini     if (type != NORM_2) SETERRQ(PetscObjectComm((PetscObject)gradient), PETSC_ERR_ARG_WRONG, "Norm type must be NORM_2 if an inner product for the gradient norm is set.");
2835a9603a14SPatrick Farrell     ierr = MatMult(tao->gradient_norm, gradient, tao->gradient_norm_tmp);CHKERRQ(ierr);
2836680e2bc7SPatrick Farrell     ierr = VecDot(gradient, tao->gradient_norm_tmp, &gnorms);CHKERRQ(ierr);
28377bb79937SPatrick Farrell     *gnorm = PetscRealPart(PetscSqrtScalar(gnorms));
2838a9603a14SPatrick Farrell   } else {
2839a9603a14SPatrick Farrell     ierr = VecNorm(gradient, type, gnorm);CHKERRQ(ierr);
2840a9603a14SPatrick Farrell   }
2841a9603a14SPatrick Farrell   PetscFunctionReturn(0);
2842a9603a14SPatrick Farrell }
2843a9603a14SPatrick Farrell 
2844e882e171SHong Zhang /*@C
2845e882e171SHong Zhang    TaoMonitorDrawCtxCreate - Creates the monitor context for TaoMonitorDrawCtx
2846a9603a14SPatrick Farrell 
2847e882e171SHong Zhang    Collective on Tao
2848e882e171SHong Zhang 
2849e882e171SHong Zhang    Output Patameter:
2850e882e171SHong Zhang .    ctx - the monitor context
2851e882e171SHong Zhang 
2852e882e171SHong Zhang    Options Database:
2853e882e171SHong Zhang .   -tao_draw_solution_initial - show initial guess as well as current solution
2854e882e171SHong Zhang 
2855e882e171SHong Zhang    Level: intermediate
2856e882e171SHong Zhang 
2857e882e171SHong Zhang .seealso: TaoMonitorSet(), TaoMonitorDefault(), VecView(), TaoMonitorDrawCtx()
2858e882e171SHong Zhang @*/
2859e882e171SHong Zhang PetscErrorCode  TaoMonitorDrawCtxCreate(MPI_Comm comm,const char host[],const char label[],int x,int y,int m,int n,PetscInt howoften,TaoMonitorDrawCtx *ctx)
2860e882e171SHong Zhang {
2861e882e171SHong Zhang   PetscErrorCode ierr;
2862e882e171SHong Zhang 
2863e882e171SHong Zhang   PetscFunctionBegin;
2864e882e171SHong Zhang   ierr = PetscNew(ctx);CHKERRQ(ierr);
2865e882e171SHong Zhang   ierr = PetscViewerDrawOpen(comm,host,label,x,y,m,n,&(*ctx)->viewer);CHKERRQ(ierr);
2866e882e171SHong Zhang   ierr = PetscViewerSetFromOptions((*ctx)->viewer);CHKERRQ(ierr);
2867e882e171SHong Zhang   (*ctx)->howoften = howoften;
2868e882e171SHong Zhang   PetscFunctionReturn(0);
2869e882e171SHong Zhang }
2870e882e171SHong Zhang 
2871e882e171SHong Zhang /*@C
2872e882e171SHong Zhang    TaoMonitorDrawCtxDestroy - Destroys the monitor context for TaoMonitorDrawSolution()
2873e882e171SHong Zhang 
2874e882e171SHong Zhang    Collective on Tao
2875e882e171SHong Zhang 
2876e882e171SHong Zhang    Input Parameters:
2877e882e171SHong Zhang .    ctx - the monitor context
2878e882e171SHong Zhang 
2879e882e171SHong Zhang    Level: intermediate
2880e882e171SHong Zhang 
2881e882e171SHong Zhang .seealso: TaoMonitorSet(), TaoMonitorDefault(), VecView(), TaoMonitorDrawSolution()
2882e882e171SHong Zhang @*/
2883e882e171SHong Zhang PetscErrorCode  TaoMonitorDrawCtxDestroy(TaoMonitorDrawCtx *ictx)
2884e882e171SHong Zhang {
2885e882e171SHong Zhang   PetscErrorCode ierr;
2886e882e171SHong Zhang 
2887e882e171SHong Zhang   PetscFunctionBegin;
2888e882e171SHong Zhang   ierr = PetscViewerDestroy(&(*ictx)->viewer);CHKERRQ(ierr);
2889e882e171SHong Zhang   ierr = PetscFree(*ictx);CHKERRQ(ierr);
2890e882e171SHong Zhang   PetscFunctionReturn(0);
2891e882e171SHong Zhang }
2892