xref: /petsc/src/tao/interface/taosolver.c (revision a958fbfc1c07da5d8abfa22584ccb9c44e85e9ad)
1af0996ceSBarry Smith #include <petsc/private/taoimpl.h> /*I "petsctao.h" I*/
20f0abf79SStefano Zampini #include <petsc/private/snesimpl.h>
3a7e14dcfSSatish Balay 
4441846f8SBarry Smith PetscBool TaoRegisterAllCalled = PETSC_FALSE;
5441846f8SBarry Smith PetscFunctionList TaoList = NULL;
6a7e14dcfSSatish Balay 
7441846f8SBarry Smith PetscClassId TAO_CLASSID;
80ebee16dSLisandro Dalcin 
90ebee16dSLisandro Dalcin PetscLogEvent TAO_Solve;
100ebee16dSLisandro Dalcin PetscLogEvent TAO_ObjectiveEval;
110ebee16dSLisandro Dalcin PetscLogEvent TAO_GradientEval;
120ebee16dSLisandro Dalcin PetscLogEvent TAO_ObjGradEval;
130ebee16dSLisandro Dalcin PetscLogEvent TAO_HessianEval;
140ebee16dSLisandro Dalcin PetscLogEvent TAO_JacobianEval;
150ebee16dSLisandro Dalcin PetscLogEvent TAO_ConstraintsEval;
16a7e14dcfSSatish Balay 
1783c8fe1dSLisandro Dalcin const char *TaoSubSetTypes[] = {"subvec","mask","matrixfree","TaoSubSetType","TAO_SUBSET_",NULL};
18a7e14dcfSSatish Balay 
19e882e171SHong Zhang struct _n_TaoMonitorDrawCtx {
20e882e171SHong Zhang   PetscViewer viewer;
21e882e171SHong Zhang   PetscInt    howoften;  /* when > 0 uses iteration % howoften, when negative only final solution plotted */
22e882e171SHong Zhang };
23e882e171SHong Zhang 
240f0abf79SStefano Zampini static PetscErrorCode KSPPreSolve_TAOEW_Private(KSP ksp, Vec b, Vec x, Tao tao)
250f0abf79SStefano Zampini {
260f0abf79SStefano Zampini   SNES snes_ewdummy = tao->snes_ewdummy;
270f0abf79SStefano Zampini 
280f0abf79SStefano Zampini   PetscFunctionBegin;
290f0abf79SStefano Zampini   if (!snes_ewdummy) PetscFunctionReturn(0);
300f0abf79SStefano Zampini   /* populate snes_ewdummy struct values used in KSPPreSolve_SNESEW */
310f0abf79SStefano Zampini   snes_ewdummy->vec_func = b;
320f0abf79SStefano Zampini   snes_ewdummy->rtol = tao->gttol;
330f0abf79SStefano Zampini   snes_ewdummy->iter = tao->niter;
340f0abf79SStefano Zampini   PetscCall(VecNorm(b,NORM_2,&snes_ewdummy->norm));
350f0abf79SStefano Zampini   PetscCall(KSPPreSolve_SNESEW(ksp,b,x,snes_ewdummy));
360f0abf79SStefano Zampini   snes_ewdummy->vec_func = NULL;
370f0abf79SStefano Zampini   PetscFunctionReturn(0);
380f0abf79SStefano Zampini }
390f0abf79SStefano Zampini 
400f0abf79SStefano Zampini static PetscErrorCode KSPPostSolve_TAOEW_Private(KSP ksp, Vec b, Vec x, Tao tao)
410f0abf79SStefano Zampini {
420f0abf79SStefano Zampini   SNES snes_ewdummy = tao->snes_ewdummy;
430f0abf79SStefano Zampini 
440f0abf79SStefano Zampini   PetscFunctionBegin;
450f0abf79SStefano Zampini   if (!snes_ewdummy) PetscFunctionReturn(0);
460f0abf79SStefano Zampini   PetscCall(KSPPostSolve_SNESEW(ksp,b,x,snes_ewdummy));
470f0abf79SStefano Zampini   PetscFunctionReturn(0);
480f0abf79SStefano Zampini }
490f0abf79SStefano Zampini 
500f0abf79SStefano Zampini static PetscErrorCode TaoSetUpEW_Private(Tao tao)
510f0abf79SStefano Zampini {
520f0abf79SStefano Zampini   SNESKSPEW  *kctx;
530f0abf79SStefano Zampini   const char *ewprefix;
540f0abf79SStefano Zampini 
550f0abf79SStefano Zampini   PetscFunctionBegin;
560f0abf79SStefano Zampini   if (!tao->ksp) PetscFunctionReturn(0);
570f0abf79SStefano Zampini   if (tao->ksp_ewconv) {
580f0abf79SStefano Zampini     if (!tao->snes_ewdummy) PetscCall(SNESCreate(PetscObjectComm((PetscObject)tao),&tao->snes_ewdummy));
590f0abf79SStefano Zampini     tao->snes_ewdummy->ksp_ewconv = PETSC_TRUE;
600f0abf79SStefano Zampini     PetscCall(KSPSetPreSolve(tao->ksp,(PetscErrorCode (*)(KSP,Vec,Vec,void*))KSPPreSolve_TAOEW_Private,tao));
610f0abf79SStefano Zampini     PetscCall(KSPSetPostSolve(tao->ksp,(PetscErrorCode (*)(KSP,Vec,Vec,void*))KSPPostSolve_TAOEW_Private,tao));
620f0abf79SStefano Zampini 
630f0abf79SStefano Zampini     PetscCall(KSPGetOptionsPrefix(tao->ksp,&ewprefix));
640f0abf79SStefano Zampini     kctx = (SNESKSPEW*)tao->snes_ewdummy->kspconvctx;
650f0abf79SStefano Zampini     PetscCall(SNESEWSetFromOptions_Private(kctx,PetscObjectComm((PetscObject)tao),ewprefix));
660f0abf79SStefano Zampini   } else PetscCall(SNESDestroy(&tao->snes_ewdummy));
670f0abf79SStefano Zampini   PetscFunctionReturn(0);
680f0abf79SStefano Zampini }
690f0abf79SStefano Zampini 
70a7e14dcfSSatish Balay /*@
71a7e14dcfSSatish Balay   TaoCreate - Creates a TAO solver
72a7e14dcfSSatish Balay 
73d083f849SBarry Smith   Collective
74a7e14dcfSSatish Balay 
75a7e14dcfSSatish Balay   Input Parameter:
76a7e14dcfSSatish Balay . comm - MPI communicator
77a7e14dcfSSatish Balay 
78a7e14dcfSSatish Balay   Output Parameter:
79441846f8SBarry Smith . newtao - the new Tao context
80a7e14dcfSSatish Balay 
81a7e14dcfSSatish Balay   Available methods include:
8258417fe7SBarry Smith +    nls - Newton's method with line search for unconstrained minimization
8358417fe7SBarry Smith .    ntr - Newton's method with trust region for unconstrained minimization
8458417fe7SBarry Smith .    ntl - Newton's method with trust region, line search for unconstrained minimization
8558417fe7SBarry Smith .    lmvm - Limited memory variable metric method for unconstrained minimization
8658417fe7SBarry Smith .    cg - Nonlinear conjugate gradient method for unconstrained minimization
8758417fe7SBarry Smith .    nm - Nelder-Mead algorithm for derivate-free unconstrained minimization
8858417fe7SBarry Smith .    tron - Newton Trust Region method for bound constrained minimization
8958417fe7SBarry Smith .    gpcg - Newton Trust Region method for quadratic bound constrained minimization
9058417fe7SBarry Smith .    blmvm - Limited memory variable metric method for bound constrained minimization
9158417fe7SBarry Smith .    lcl - Linearly constrained Lagrangian method for pde-constrained minimization
9258417fe7SBarry Smith -    pounders - Model-based algorithm for nonlinear least squares
93a7e14dcfSSatish Balay 
94a7e14dcfSSatish Balay    Options Database Keys:
9558417fe7SBarry Smith .   -tao_type - select which method TAO should use
96a7e14dcfSSatish Balay 
97a7e14dcfSSatish Balay    Level: beginner
98a7e14dcfSSatish Balay 
99db781477SPatrick Sanan .seealso: `TaoSolve()`, `TaoDestroy()`
100a7e14dcfSSatish Balay @*/
101441846f8SBarry Smith PetscErrorCode TaoCreate(MPI_Comm comm, Tao *newtao)
102a7e14dcfSSatish Balay {
103441846f8SBarry Smith   Tao            tao;
104a7e14dcfSSatish Balay 
105a7e14dcfSSatish Balay   PetscFunctionBegin;
106a7e14dcfSSatish Balay   PetscValidPointer(newtao,2);
1079566063dSJacob Faibussowitsch   PetscCall(TaoInitializePackage());
1089566063dSJacob Faibussowitsch   PetscCall(TaoLineSearchInitializePackage());
1099566063dSJacob Faibussowitsch   PetscCall(PetscHeaderCreate(tao,TAO_CLASSID,"Tao","Optimization solver","Tao",comm,TaoDestroy,TaoView));
110a7e14dcfSSatish Balay 
11194511df7SStefano Zampini   /* Set non-NULL defaults */
11294511df7SStefano Zampini   tao->ops->convergencetest = TaoDefaultConvergenceTest;
113a7e14dcfSSatish Balay 
114a7e14dcfSSatish Balay   tao->max_it    = 10000;
11594511df7SStefano Zampini   tao->max_funcs = -1;
1166f4723b1SBarry Smith #if defined(PETSC_USE_REAL_SINGLE)
1176f4723b1SBarry Smith   tao->gatol     = 1e-5;
1186f4723b1SBarry Smith   tao->grtol     = 1e-5;
1196f1f5f59SAlp Dener   tao->crtol     = 1e-5;
1206f1f5f59SAlp Dener   tao->catol     = 1e-5;
1216f4723b1SBarry Smith #else
122a7e14dcfSSatish Balay   tao->gatol     = 1e-8;
123a7e14dcfSSatish Balay   tao->grtol     = 1e-8;
1246f1f5f59SAlp Dener   tao->crtol     = 1e-8;
1256f1f5f59SAlp Dener   tao->catol     = 1e-8;
1266f4723b1SBarry Smith #endif
1276552cf8aSJason Sarich   tao->gttol     = 0.0;
1284f1535bcSTodd Munson   tao->steptol   = 0.0;
129e270355aSBarry Smith   tao->trust0    = PETSC_INFINITY;
1306f4723b1SBarry Smith   tao->fmin      = PETSC_NINFINITY;
1319fa2b5dcSStefano Zampini 
132a7e14dcfSSatish Balay   tao->hist_reset  = PETSC_TRUE;
133a7e14dcfSSatish Balay 
1349566063dSJacob Faibussowitsch   PetscCall(TaoResetStatistics(tao));
135a7e14dcfSSatish Balay   *newtao = tao;
136a7e14dcfSSatish Balay   PetscFunctionReturn(0);
137a7e14dcfSSatish Balay }
138a7e14dcfSSatish Balay 
139a7e14dcfSSatish Balay /*@
140a7e14dcfSSatish Balay   TaoSolve - Solves an optimization problem min F(x) s.t. l <= x <= u
141a7e14dcfSSatish Balay 
142441846f8SBarry Smith   Collective on Tao
143a7e14dcfSSatish Balay 
144a7e14dcfSSatish Balay   Input Parameters:
145441846f8SBarry Smith . tao - the Tao context
146a7e14dcfSSatish Balay 
147a7e14dcfSSatish Balay   Notes:
148a82e8c82SStefano Zampini   The user must set up the Tao with calls to TaoSetSolution(),
149a82e8c82SStefano Zampini   TaoSetObjective(),
150a82e8c82SStefano Zampini   TaoSetGradient(), and (if using 2nd order method) TaoSetHessian().
151a7e14dcfSSatish Balay 
152a35d58b8SBarry Smith   You should call TaoGetConvergedReason() or run with -tao_converged_reason to determine if the optimization algorithm actually succeeded or
153a35d58b8SBarry Smith   why it failed.
154a35d58b8SBarry Smith 
155a7e14dcfSSatish Balay   Level: beginner
156a7e14dcfSSatish Balay 
157db781477SPatrick Sanan .seealso: `TaoCreate()`, `TaoSetObjective()`, `TaoSetGradient()`, `TaoSetHessian()`, `TaoGetConvergedReason()`
158a7e14dcfSSatish Balay  @*/
159441846f8SBarry Smith PetscErrorCode TaoSolve(Tao tao)
160a7e14dcfSSatish Balay {
161e2379f4fSBarry Smith   static PetscBool set = PETSC_FALSE;
16247a47007SBarry Smith 
163a7e14dcfSSatish Balay   PetscFunctionBegin;
164441846f8SBarry Smith   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
165d0609cedSBarry Smith   PetscCall(PetscCitationsRegister("@TechReport{tao-user-ref,\n"
166e2379f4fSBarry Smith                                    "title   = {Toolkit for Advanced Optimization (TAO) Users Manual},\n"
167e2379f4fSBarry Smith                                    "author  = {Todd Munson and Jason Sarich and Stefan Wild and Steve Benson and Lois Curfman McInnes},\n"
168e2379f4fSBarry Smith                                    "Institution = {Argonne National Laboratory},\n"
169e2379f4fSBarry Smith                                    "Year   = 2014,\n"
170e2379f4fSBarry Smith                                    "Number = {ANL/MCS-TM-322 - Revision 3.5},\n"
171d0609cedSBarry Smith                                    "url    = {https://www.mcs.anl.gov/research/projects/tao/}\n}\n",&set));
172494bef23SAlp Dener   tao->header_printed = PETSC_FALSE;
1739566063dSJacob Faibussowitsch   PetscCall(TaoSetUp(tao));
1749566063dSJacob Faibussowitsch   PetscCall(TaoResetStatistics(tao));
175a7e14dcfSSatish Balay   if (tao->linesearch) {
1769566063dSJacob Faibussowitsch     PetscCall(TaoLineSearchReset(tao->linesearch));
177a7e14dcfSSatish Balay   }
178a7e14dcfSSatish Balay 
1799566063dSJacob Faibussowitsch   PetscCall(PetscLogEventBegin(TAO_Solve,tao,0,0,0));
1809566063dSJacob Faibussowitsch   if (tao->ops->solve) PetscCall((*tao->ops->solve)(tao));
1819566063dSJacob Faibussowitsch   PetscCall(PetscLogEventEnd(TAO_Solve,tao,0,0,0));
182a7e14dcfSSatish Balay 
1839566063dSJacob Faibussowitsch   PetscCall(VecViewFromOptions(tao->solution,(PetscObject)tao,"-tao_view_solution"));
1847cf37e64SBarry Smith 
1858931d482SJason Sarich   tao->ntotalits += tao->niter;
1869566063dSJacob Faibussowitsch   PetscCall(TaoViewFromOptions(tao,NULL,"-tao_view"));
187a7e14dcfSSatish Balay 
188a7e14dcfSSatish Balay   if (tao->printreason) {
189a7e14dcfSSatish Balay     if (tao->reason > 0) {
19063a3b9bcSJacob Faibussowitsch       PetscCall(PetscPrintf(((PetscObject)tao)->comm,"TAO solve converged due to %s iterations %" PetscInt_FMT "\n",TaoConvergedReasons[tao->reason],tao->niter));
191a7e14dcfSSatish Balay     } else {
19263a3b9bcSJacob Faibussowitsch       PetscCall(PetscPrintf(((PetscObject)tao)->comm,"TAO solve did not converge due to %s iteration %" PetscInt_FMT "\n",TaoConvergedReasons[tao->reason],tao->niter));
193a7e14dcfSSatish Balay     }
194a7e14dcfSSatish Balay   }
195a7e14dcfSSatish Balay   PetscFunctionReturn(0);
196a7e14dcfSSatish Balay }
197a7e14dcfSSatish Balay 
198a7e14dcfSSatish Balay /*@
199a7e14dcfSSatish Balay   TaoSetUp - Sets up the internal data structures for the later use
200a7e14dcfSSatish Balay   of a Tao solver
201a7e14dcfSSatish Balay 
202a7e14dcfSSatish Balay   Collective on tao
203a7e14dcfSSatish Balay 
204a7e14dcfSSatish Balay   Input Parameters:
205a7e14dcfSSatish Balay . tao - the TAO context
206a7e14dcfSSatish Balay 
207a7e14dcfSSatish Balay   Notes:
208a7e14dcfSSatish Balay   The user will not need to explicitly call TaoSetUp(), as it will
209a7e14dcfSSatish Balay   automatically be called in TaoSolve().  However, if the user
210a7e14dcfSSatish Balay   desires to call it explicitly, it should come after TaoCreate()
211a7e14dcfSSatish Balay   and any TaoSetSomething() routines, but before TaoSolve().
212a7e14dcfSSatish Balay 
213a7e14dcfSSatish Balay   Level: advanced
214a7e14dcfSSatish Balay 
215db781477SPatrick Sanan .seealso: `TaoCreate()`, `TaoSolve()`
216a7e14dcfSSatish Balay @*/
217441846f8SBarry Smith PetscErrorCode TaoSetUp(Tao tao)
218a7e14dcfSSatish Balay {
219a7e14dcfSSatish Balay   PetscFunctionBegin;
220441846f8SBarry Smith   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
221a7e14dcfSSatish Balay   if (tao->setupcalled) PetscFunctionReturn(0);
2220f0abf79SStefano Zampini   PetscCall(TaoSetUpEW_Private(tao));
2233c859ba3SBarry Smith   PetscCheck(tao->solution,PetscObjectComm((PetscObject)tao),PETSC_ERR_ARG_WRONGSTATE,"Must call TaoSetSolution");
2240f0abf79SStefano Zampini   if (tao->ops->setup) PetscCall((*tao->ops->setup)(tao));
225a7e14dcfSSatish Balay   tao->setupcalled = PETSC_TRUE;
226a7e14dcfSSatish Balay   PetscFunctionReturn(0);
227a7e14dcfSSatish Balay }
228a7e14dcfSSatish Balay 
2291fb7b255SJunchao Zhang /*@C
230a7e14dcfSSatish Balay   TaoDestroy - Destroys the TAO context that was created with
231a7e14dcfSSatish Balay   TaoCreate()
232a7e14dcfSSatish Balay 
233441846f8SBarry Smith   Collective on Tao
234a7e14dcfSSatish Balay 
235a7e14dcfSSatish Balay   Input Parameter:
236441846f8SBarry Smith . tao - the Tao context
237a7e14dcfSSatish Balay 
238a7e14dcfSSatish Balay   Level: beginner
239a7e14dcfSSatish Balay 
240db781477SPatrick Sanan .seealso: `TaoCreate()`, `TaoSolve()`
241a7e14dcfSSatish Balay @*/
242441846f8SBarry Smith PetscErrorCode TaoDestroy(Tao *tao)
243a7e14dcfSSatish Balay {
244a7e14dcfSSatish Balay   PetscFunctionBegin;
245a7e14dcfSSatish Balay   if (!*tao) PetscFunctionReturn(0);
246441846f8SBarry Smith   PetscValidHeaderSpecific(*tao,TAO_CLASSID,1);
24783c8fe1dSLisandro Dalcin   if (--((PetscObject)*tao)->refct > 0) {*tao = NULL;PetscFunctionReturn(0);}
248a7e14dcfSSatish Balay 
249a7e14dcfSSatish Balay   if ((*tao)->ops->destroy) {
2509566063dSJacob Faibussowitsch     PetscCall((*((*tao))->ops->destroy)(*tao));
251a7e14dcfSSatish Balay   }
2529566063dSJacob Faibussowitsch   PetscCall(KSPDestroy(&(*tao)->ksp));
2530f0abf79SStefano Zampini   PetscCall(SNESDestroy(&(*tao)->snes_ewdummy));
2549566063dSJacob Faibussowitsch   PetscCall(TaoLineSearchDestroy(&(*tao)->linesearch));
255a7e14dcfSSatish Balay 
256a7e14dcfSSatish Balay   if ((*tao)->ops->convergencedestroy) {
2579566063dSJacob Faibussowitsch     PetscCall((*(*tao)->ops->convergencedestroy)((*tao)->cnvP));
258a7e14dcfSSatish Balay     if ((*tao)->jacobian_state_inv) {
2599566063dSJacob Faibussowitsch       PetscCall(MatDestroy(&(*tao)->jacobian_state_inv));
260a7e14dcfSSatish Balay     }
261a7e14dcfSSatish Balay   }
2629566063dSJacob Faibussowitsch   PetscCall(VecDestroy(&(*tao)->solution));
2639566063dSJacob Faibussowitsch   PetscCall(VecDestroy(&(*tao)->gradient));
2649566063dSJacob Faibussowitsch   PetscCall(VecDestroy(&(*tao)->ls_res));
265a7e14dcfSSatish Balay 
266a9603a14SPatrick Farrell   if ((*tao)->gradient_norm) {
2679566063dSJacob Faibussowitsch     PetscCall(PetscObjectDereference((PetscObject)(*tao)->gradient_norm));
2689566063dSJacob Faibussowitsch     PetscCall(VecDestroy(&(*tao)->gradient_norm_tmp));
269a9603a14SPatrick Farrell   }
270a9603a14SPatrick Farrell 
2719566063dSJacob Faibussowitsch   PetscCall(VecDestroy(&(*tao)->XL));
2729566063dSJacob Faibussowitsch   PetscCall(VecDestroy(&(*tao)->XU));
2739566063dSJacob Faibussowitsch   PetscCall(VecDestroy(&(*tao)->IL));
2749566063dSJacob Faibussowitsch   PetscCall(VecDestroy(&(*tao)->IU));
2759566063dSJacob Faibussowitsch   PetscCall(VecDestroy(&(*tao)->DE));
2769566063dSJacob Faibussowitsch   PetscCall(VecDestroy(&(*tao)->DI));
2779566063dSJacob Faibussowitsch   PetscCall(VecDestroy(&(*tao)->constraints));
2789566063dSJacob Faibussowitsch   PetscCall(VecDestroy(&(*tao)->constraints_equality));
2799566063dSJacob Faibussowitsch   PetscCall(VecDestroy(&(*tao)->constraints_inequality));
2809566063dSJacob Faibussowitsch   PetscCall(VecDestroy(&(*tao)->stepdirection));
2819566063dSJacob Faibussowitsch   PetscCall(MatDestroy(&(*tao)->hessian_pre));
2829566063dSJacob Faibussowitsch   PetscCall(MatDestroy(&(*tao)->hessian));
2839566063dSJacob Faibussowitsch   PetscCall(MatDestroy(&(*tao)->ls_jac));
2849566063dSJacob Faibussowitsch   PetscCall(MatDestroy(&(*tao)->ls_jac_pre));
2859566063dSJacob Faibussowitsch   PetscCall(MatDestroy(&(*tao)->jacobian_pre));
2869566063dSJacob Faibussowitsch   PetscCall(MatDestroy(&(*tao)->jacobian));
2879566063dSJacob Faibussowitsch   PetscCall(MatDestroy(&(*tao)->jacobian_state_pre));
2889566063dSJacob Faibussowitsch   PetscCall(MatDestroy(&(*tao)->jacobian_state));
2899566063dSJacob Faibussowitsch   PetscCall(MatDestroy(&(*tao)->jacobian_state_inv));
2909566063dSJacob Faibussowitsch   PetscCall(MatDestroy(&(*tao)->jacobian_design));
2919566063dSJacob Faibussowitsch   PetscCall(MatDestroy(&(*tao)->jacobian_equality));
2929566063dSJacob Faibussowitsch   PetscCall(MatDestroy(&(*tao)->jacobian_equality_pre));
2939566063dSJacob Faibussowitsch   PetscCall(MatDestroy(&(*tao)->jacobian_inequality));
2949566063dSJacob Faibussowitsch   PetscCall(MatDestroy(&(*tao)->jacobian_inequality_pre));
2959566063dSJacob Faibussowitsch   PetscCall(ISDestroy(&(*tao)->state_is));
2969566063dSJacob Faibussowitsch   PetscCall(ISDestroy(&(*tao)->design_is));
2979566063dSJacob Faibussowitsch   PetscCall(VecDestroy(&(*tao)->res_weights_v));
2989566063dSJacob Faibussowitsch   PetscCall(TaoCancelMonitors(*tao));
299ae93cb3cSJason Sarich   if ((*tao)->hist_malloc) {
3009566063dSJacob Faibussowitsch     PetscCall(PetscFree4((*tao)->hist_obj,(*tao)->hist_resid,(*tao)->hist_cnorm,(*tao)->hist_lits));
301ae93cb3cSJason Sarich   }
302737f463aSAlp Dener   if ((*tao)->res_weights_n) {
3039566063dSJacob Faibussowitsch     PetscCall(PetscFree((*tao)->res_weights_rows));
3049566063dSJacob Faibussowitsch     PetscCall(PetscFree((*tao)->res_weights_cols));
3059566063dSJacob Faibussowitsch     PetscCall(PetscFree((*tao)->res_weights_w));
306125ddc8aSJason Sarich   }
3079566063dSJacob Faibussowitsch   PetscCall(PetscHeaderDestroy(tao));
308a7e14dcfSSatish Balay   PetscFunctionReturn(0);
309a7e14dcfSSatish Balay }
310a7e14dcfSSatish Balay 
311a7e14dcfSSatish Balay /*@
3120f0abf79SStefano Zampini    TaoKSPSetUseEW - Sets SNES use Eisenstat-Walker method for
3130f0abf79SStefano Zampini    computing relative tolerance for linear solvers.
3140f0abf79SStefano Zampini 
3150f0abf79SStefano Zampini    Logically Collective on TAO
3160f0abf79SStefano Zampini 
3170f0abf79SStefano Zampini    Input Parameters:
3180f0abf79SStefano Zampini +  tao - Tao context
3190f0abf79SStefano Zampini -  flag - PETSC_TRUE or PETSC_FALSE
3200f0abf79SStefano Zampini 
3210f0abf79SStefano Zampini    Notes:
3220f0abf79SStefano Zampini    See SNESKSPSetUseEW() for customization details.
3230f0abf79SStefano Zampini 
3240f0abf79SStefano Zampini    Level: advanced
3250f0abf79SStefano Zampini 
3260f0abf79SStefano Zampini    Reference:
3270f0abf79SStefano Zampini    S. C. Eisenstat and H. F. Walker, "Choosing the forcing terms in an
3280f0abf79SStefano Zampini    inexact Newton method", SISC 17 (1), pp.16-32, 1996.
3290f0abf79SStefano Zampini 
3300f0abf79SStefano Zampini .seealso: `SNESKSPSetUseEW()`
3310f0abf79SStefano Zampini @*/
3320f0abf79SStefano Zampini PetscErrorCode  TaoKSPSetUseEW(Tao tao,PetscBool flag)
3330f0abf79SStefano Zampini {
3340f0abf79SStefano Zampini   PetscFunctionBegin;
3350f0abf79SStefano Zampini   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
3360f0abf79SStefano Zampini   PetscValidLogicalCollectiveBool(tao,flag,2);
3370f0abf79SStefano Zampini   tao->ksp_ewconv = flag;
3380f0abf79SStefano Zampini   PetscFunctionReturn(0);
3390f0abf79SStefano Zampini }
3400f0abf79SStefano Zampini 
3410f0abf79SStefano Zampini /*@
342441846f8SBarry Smith   TaoSetFromOptions - Sets various Tao parameters from user
343a7e14dcfSSatish Balay   options.
344a7e14dcfSSatish Balay 
345441846f8SBarry Smith   Collective on Tao
346a7e14dcfSSatish Balay 
34701d2d390SJose E. Roman   Input Parameter:
348441846f8SBarry Smith . tao - the Tao solver context
349a7e14dcfSSatish Balay 
350a7e14dcfSSatish Balay   options Database Keys:
35158417fe7SBarry Smith + -tao_type <type> - The algorithm that TAO uses (lmvm, nls, etc.)
352a7e14dcfSSatish Balay . -tao_gatol <gatol> - absolute error tolerance for ||gradient||
353a7e14dcfSSatish Balay . -tao_grtol <grtol> - relative error tolerance for ||gradient||
354a7e14dcfSSatish Balay . -tao_gttol <gttol> - reduction of ||gradient|| relative to initial gradient
355a7e14dcfSSatish Balay . -tao_max_it <max> - sets maximum number of iterations
356a7e14dcfSSatish Balay . -tao_max_funcs <max> - sets maximum number of function evaluations
357a7e14dcfSSatish Balay . -tao_fmin <fmin> - stop if function value reaches fmin
358a7e14dcfSSatish Balay . -tao_steptol <tol> - stop if trust region radius less than <tol>
359a7e14dcfSSatish Balay . -tao_trust0 <t> - initial trust region radius
360a7e14dcfSSatish Balay . -tao_monitor - prints function value and residual at each iteration
361a7e14dcfSSatish Balay . -tao_smonitor - same as tao_monitor, but truncates very small values
362a7e14dcfSSatish Balay . -tao_cmonitor - prints function value, residual, and constraint norm at each iteration
363a7e14dcfSSatish Balay . -tao_view_solution - prints solution vector at each iteration
3644a48860cSAlp Dener . -tao_view_ls_residual - prints least-squares residual vector at each iteration
365147403d9SBarry Smith . -tao_view_stepdirection - prints step direction vector at each iteration
366a7e14dcfSSatish Balay . -tao_view_gradient - prints gradient vector at each iteration
367a7e14dcfSSatish Balay . -tao_draw_solution - graphically view solution vector at each iteration
368a7e14dcfSSatish Balay . -tao_draw_step - graphically view step vector at each iteration
369a7e14dcfSSatish Balay . -tao_draw_gradient - graphically view gradient at each iteration
370a7e14dcfSSatish Balay . -tao_fd_gradient - use gradient computed with finite differences
371f4c1ad5cSStefano Zampini . -tao_fd_hessian - use hessian computed with finite differences
372f4c1ad5cSStefano Zampini . -tao_mf_hessian - use matrix-free hessian computed with finite differences
373a7e14dcfSSatish Balay . -tao_cancelmonitors - cancels all monitors (except those set with command line)
374441846f8SBarry Smith . -tao_view - prints information about the Tao after solving
375a7e14dcfSSatish Balay - -tao_converged_reason - prints the reason TAO stopped iterating
376a7e14dcfSSatish Balay 
377a7e14dcfSSatish Balay   Notes:
378a7e14dcfSSatish Balay   To see all options, run your program with the -help option or consult the
379a7e14dcfSSatish Balay   user's manual. Should be called after TaoCreate() but before TaoSolve()
380a7e14dcfSSatish Balay 
381a7e14dcfSSatish Balay   Level: beginner
382a7e14dcfSSatish Balay @*/
383441846f8SBarry Smith PetscErrorCode TaoSetFromOptions(Tao tao)
384a7e14dcfSSatish Balay {
385b625d6c7SJed Brown   TaoType        default_type = TAOLMVM;
386a7e14dcfSSatish Balay   char           type[256], monfilename[PETSC_MAX_PATH_LEN];
387a7e14dcfSSatish Balay   PetscViewer    monviewer;
388a7e14dcfSSatish Balay   PetscBool      flg;
389a7e14dcfSSatish Balay   MPI_Comm       comm;
390a7e14dcfSSatish Balay 
391a7e14dcfSSatish Balay   PetscFunctionBegin;
392441846f8SBarry Smith   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
3939566063dSJacob Faibussowitsch   PetscCall(PetscObjectGetComm((PetscObject)tao,&comm));
394125ddc8aSJason Sarich 
395a7e14dcfSSatish Balay   /* So no warnings are given about unused options */
3969566063dSJacob Faibussowitsch   PetscCall(PetscOptionsHasName(((PetscObject)tao)->options,((PetscObject)tao)->prefix,"-tao_ls_type",&flg));
397a7e14dcfSSatish Balay 
398d0609cedSBarry Smith   PetscObjectOptionsBegin((PetscObject)tao);
399a7e14dcfSSatish Balay   {
400a82e8c82SStefano Zampini     if (((PetscObject)tao)->type_name) default_type = ((PetscObject)tao)->type_name;
401a7e14dcfSSatish Balay     /* Check for type from options */
4029566063dSJacob Faibussowitsch     PetscCall(PetscOptionsFList("-tao_type","Tao Solver type","TaoSetType",TaoList,default_type,type,256,&flg));
403a7e14dcfSSatish Balay     if (flg) {
4049566063dSJacob Faibussowitsch       PetscCall(TaoSetType(tao,type));
405a7e14dcfSSatish Balay     } else if (!((PetscObject)tao)->type_name) {
4069566063dSJacob Faibussowitsch       PetscCall(TaoSetType(tao,default_type));
407a7e14dcfSSatish Balay     }
408a7e14dcfSSatish Balay 
4099566063dSJacob Faibussowitsch     PetscCall(PetscOptionsReal("-tao_catol","Stop if constraints violations within","TaoSetConstraintTolerances",tao->catol,&tao->catol,&flg));
4106552cf8aSJason Sarich     if (flg) tao->catol_changed = PETSC_TRUE;
4116aad120cSJose E. Roman     PetscCall(PetscOptionsReal("-tao_crtol","Stop if relative constraint violations within","TaoSetConstraintTolerances",tao->crtol,&tao->crtol,&flg));
4126552cf8aSJason Sarich     if (flg) tao->crtol_changed = PETSC_TRUE;
4139566063dSJacob Faibussowitsch     PetscCall(PetscOptionsReal("-tao_gatol","Stop if norm of gradient less than","TaoSetTolerances",tao->gatol,&tao->gatol,&flg));
4146552cf8aSJason Sarich     if (flg) tao->gatol_changed = PETSC_TRUE;
4159566063dSJacob Faibussowitsch     PetscCall(PetscOptionsReal("-tao_grtol","Stop if norm of gradient divided by the function value is less than","TaoSetTolerances",tao->grtol,&tao->grtol,&flg));
4166552cf8aSJason Sarich     if (flg) tao->grtol_changed = PETSC_TRUE;
4179566063dSJacob Faibussowitsch     PetscCall(PetscOptionsReal("-tao_gttol","Stop if the norm of the gradient is less than the norm of the initial gradient times tol","TaoSetTolerances",tao->gttol,&tao->gttol,&flg));
4186552cf8aSJason Sarich     if (flg) tao->gttol_changed = PETSC_TRUE;
4199566063dSJacob Faibussowitsch     PetscCall(PetscOptionsInt("-tao_max_it","Stop if iteration number exceeds","TaoSetMaximumIterations",tao->max_it,&tao->max_it,&flg));
4206552cf8aSJason Sarich     if (flg) tao->max_it_changed = PETSC_TRUE;
4219566063dSJacob Faibussowitsch     PetscCall(PetscOptionsInt("-tao_max_funcs","Stop if number of function evaluations exceeds","TaoSetMaximumFunctionEvaluations",tao->max_funcs,&tao->max_funcs,&flg));
4226552cf8aSJason Sarich     if (flg) tao->max_funcs_changed = PETSC_TRUE;
4239566063dSJacob Faibussowitsch     PetscCall(PetscOptionsReal("-tao_fmin","Stop if function less than","TaoSetFunctionLowerBound",tao->fmin,&tao->fmin,&flg));
4246552cf8aSJason Sarich     if (flg) tao->fmin_changed = PETSC_TRUE;
4259566063dSJacob Faibussowitsch     PetscCall(PetscOptionsReal("-tao_steptol","Stop if step size or trust region radius less than","",tao->steptol,&tao->steptol,&flg));
4266552cf8aSJason Sarich     if (flg) tao->steptol_changed = PETSC_TRUE;
4279566063dSJacob Faibussowitsch     PetscCall(PetscOptionsReal("-tao_trust0","Initial trust region radius","TaoSetTrustRegionRadius",tao->trust0,&tao->trust0,&flg));
4286552cf8aSJason Sarich     if (flg) tao->trust0_changed = PETSC_TRUE;
4299566063dSJacob Faibussowitsch     PetscCall(PetscOptionsString("-tao_view_solution","view solution vector after each evaluation","TaoSetMonitor","stdout",monfilename,sizeof(monfilename),&flg));
430a7e14dcfSSatish Balay     if (flg) {
4319566063dSJacob Faibussowitsch       PetscCall(PetscViewerASCIIOpen(comm,monfilename,&monviewer));
4329566063dSJacob Faibussowitsch       PetscCall(TaoSetMonitor(tao,TaoSolutionMonitor,monviewer,(PetscErrorCode (*)(void**))PetscViewerDestroy));
433a7e14dcfSSatish Balay     }
434a7e14dcfSSatish Balay 
4359566063dSJacob Faibussowitsch     PetscCall(PetscOptionsBool("-tao_converged_reason","Print reason for TAO converged","TaoSolve",tao->printreason,&tao->printreason,NULL));
4369566063dSJacob Faibussowitsch     PetscCall(PetscOptionsString("-tao_view_gradient","view gradient vector after each evaluation","TaoSetMonitor","stdout",monfilename,sizeof(monfilename),&flg));
437a7e14dcfSSatish Balay     if (flg) {
4389566063dSJacob Faibussowitsch       PetscCall(PetscViewerASCIIOpen(comm,monfilename,&monviewer));
4399566063dSJacob Faibussowitsch       PetscCall(TaoSetMonitor(tao,TaoGradientMonitor,monviewer,(PetscErrorCode (*)(void**))PetscViewerDestroy));
440a7e14dcfSSatish Balay     }
441a7e14dcfSSatish Balay 
4429566063dSJacob Faibussowitsch     PetscCall(PetscOptionsString("-tao_view_stepdirection","view step direction vector after each iteration","TaoSetMonitor","stdout",monfilename,sizeof(monfilename),&flg));
443a7e14dcfSSatish Balay     if (flg) {
4449566063dSJacob Faibussowitsch       PetscCall(PetscViewerASCIIOpen(comm,monfilename,&monviewer));
4459566063dSJacob Faibussowitsch       PetscCall(TaoSetMonitor(tao,TaoStepDirectionMonitor,monviewer,(PetscErrorCode (*)(void**))PetscViewerDestroy));
446a7e14dcfSSatish Balay     }
447a7e14dcfSSatish Balay 
4489566063dSJacob Faibussowitsch     PetscCall(PetscOptionsString("-tao_view_residual","view least-squares residual vector after each evaluation","TaoSetMonitor","stdout",monfilename,sizeof(monfilename),&flg));
449a7e14dcfSSatish Balay     if (flg) {
4509566063dSJacob Faibussowitsch       PetscCall(PetscViewerASCIIOpen(comm,monfilename,&monviewer));
4519566063dSJacob Faibussowitsch       PetscCall(TaoSetMonitor(tao,TaoResidualMonitor,monviewer,(PetscErrorCode (*)(void**))PetscViewerDestroy));
452a7e14dcfSSatish Balay     }
453a7e14dcfSSatish Balay 
4549566063dSJacob Faibussowitsch     PetscCall(PetscOptionsString("-tao_monitor","Use the default convergence monitor","TaoSetMonitor","stdout",monfilename,sizeof(monfilename),&flg));
455a7e14dcfSSatish Balay     if (flg) {
4569566063dSJacob Faibussowitsch       PetscCall(PetscViewerASCIIOpen(comm,monfilename,&monviewer));
4579566063dSJacob Faibussowitsch       PetscCall(TaoSetMonitor(tao,TaoMonitorDefault,monviewer,(PetscErrorCode (*)(void**))PetscViewerDestroy));
458a7e14dcfSSatish Balay     }
459a7e14dcfSSatish Balay 
4609566063dSJacob Faibussowitsch     PetscCall(PetscOptionsString("-tao_gmonitor","Use the convergence monitor with extra globalization info","TaoSetMonitor","stdout",monfilename,sizeof(monfilename),&flg));
4618d5ead36SAlp Dener     if (flg) {
4629566063dSJacob Faibussowitsch       PetscCall(PetscViewerASCIIOpen(comm,monfilename,&monviewer));
4639566063dSJacob Faibussowitsch       PetscCall(TaoSetMonitor(tao,TaoDefaultGMonitor,monviewer,(PetscErrorCode (*)(void**))PetscViewerDestroy));
4648d5ead36SAlp Dener     }
4658d5ead36SAlp Dener 
4669566063dSJacob Faibussowitsch     PetscCall(PetscOptionsString("-tao_smonitor","Use the short convergence monitor","TaoSetMonitor","stdout",monfilename,sizeof(monfilename),&flg));
467a7e14dcfSSatish Balay     if (flg) {
4689566063dSJacob Faibussowitsch       PetscCall(PetscViewerASCIIOpen(comm,monfilename,&monviewer));
4699566063dSJacob Faibussowitsch       PetscCall(TaoSetMonitor(tao,TaoDefaultSMonitor,monviewer,(PetscErrorCode (*)(void**))PetscViewerDestroy));
470a7e14dcfSSatish Balay     }
471a7e14dcfSSatish Balay 
4729566063dSJacob Faibussowitsch     PetscCall(PetscOptionsString("-tao_cmonitor","Use the default convergence monitor with constraint norm","TaoSetMonitor","stdout",monfilename,sizeof(monfilename),&flg));
473a7e14dcfSSatish Balay     if (flg) {
4749566063dSJacob Faibussowitsch       PetscCall(PetscViewerASCIIOpen(comm,monfilename,&monviewer));
4759566063dSJacob Faibussowitsch       PetscCall(TaoSetMonitor(tao,TaoDefaultCMonitor,monviewer,(PetscErrorCode (*)(void**))PetscViewerDestroy));
476a7e14dcfSSatish Balay     }
477a7e14dcfSSatish Balay 
4788afaa268SBarry Smith     flg = PETSC_FALSE;
4799566063dSJacob Faibussowitsch     PetscCall(PetscOptionsBool("-tao_cancelmonitors","cancel all monitors and call any registered destroy routines","TaoCancelMonitors",flg,&flg,NULL));
4809566063dSJacob Faibussowitsch     if (flg) PetscCall(TaoCancelMonitors(tao));
481a7e14dcfSSatish Balay 
4828afaa268SBarry Smith     flg = PETSC_FALSE;
4839566063dSJacob Faibussowitsch     PetscCall(PetscOptionsBool("-tao_draw_solution","Plot solution vector at each iteration","TaoSetMonitor",flg,&flg,NULL));
484a7e14dcfSSatish Balay     if (flg) {
485e882e171SHong Zhang       TaoMonitorDrawCtx drawctx;
486e882e171SHong Zhang       PetscInt          howoften = 1;
4879566063dSJacob Faibussowitsch       PetscCall(TaoMonitorDrawCtxCreate(PetscObjectComm((PetscObject)tao),NULL,NULL,PETSC_DECIDE,PETSC_DECIDE,300,300,howoften,&drawctx));
4889566063dSJacob Faibussowitsch       PetscCall(TaoSetMonitor(tao,TaoDrawSolutionMonitor,drawctx,(PetscErrorCode (*)(void**))TaoMonitorDrawCtxDestroy));
489a7e14dcfSSatish Balay     }
490a7e14dcfSSatish Balay 
4918afaa268SBarry Smith     flg = PETSC_FALSE;
4929566063dSJacob Faibussowitsch     PetscCall(PetscOptionsBool("-tao_draw_step","plots step direction at each iteration","TaoSetMonitor",flg,&flg,NULL));
493a7e14dcfSSatish Balay     if (flg) {
4949566063dSJacob Faibussowitsch       PetscCall(TaoSetMonitor(tao,TaoDrawStepMonitor,NULL,NULL));
495a7e14dcfSSatish Balay     }
496a7e14dcfSSatish Balay 
4978afaa268SBarry Smith     flg = PETSC_FALSE;
4989566063dSJacob Faibussowitsch     PetscCall(PetscOptionsBool("-tao_draw_gradient","plots gradient at each iteration","TaoSetMonitor",flg,&flg,NULL));
499a7e14dcfSSatish Balay     if (flg) {
500e882e171SHong Zhang       TaoMonitorDrawCtx drawctx;
501e882e171SHong Zhang       PetscInt          howoften = 1;
5029566063dSJacob Faibussowitsch       PetscCall(TaoMonitorDrawCtxCreate(PetscObjectComm((PetscObject)tao),NULL,NULL,PETSC_DECIDE,PETSC_DECIDE,300,300,howoften,&drawctx));
5039566063dSJacob Faibussowitsch       PetscCall(TaoSetMonitor(tao,TaoDrawGradientMonitor,drawctx,(PetscErrorCode (*)(void**))TaoMonitorDrawCtxDestroy));
504a7e14dcfSSatish Balay     }
5058afaa268SBarry Smith     flg = PETSC_FALSE;
5069566063dSJacob Faibussowitsch     PetscCall(PetscOptionsBool("-tao_fd_gradient","compute gradient using finite differences","TaoDefaultComputeGradient",flg,&flg,NULL));
507a7e14dcfSSatish Balay     if (flg) {
5089566063dSJacob Faibussowitsch       PetscCall(TaoSetGradient(tao,NULL,TaoDefaultComputeGradient,NULL));
509a7e14dcfSSatish Balay     }
510f4c1ad5cSStefano Zampini     flg = PETSC_FALSE;
5119566063dSJacob Faibussowitsch     PetscCall(PetscOptionsBool("-tao_fd_hessian","compute hessian using finite differences","TaoDefaultComputeHessian",flg,&flg,NULL));
512f4c1ad5cSStefano Zampini     if (flg) {
513f4c1ad5cSStefano Zampini       Mat H;
514f4c1ad5cSStefano Zampini 
5159566063dSJacob Faibussowitsch       PetscCall(MatCreate(PetscObjectComm((PetscObject)tao),&H));
5169566063dSJacob Faibussowitsch       PetscCall(MatSetType(H,MATAIJ));
5179566063dSJacob Faibussowitsch       PetscCall(TaoSetHessian(tao,H,H,TaoDefaultComputeHessian,NULL));
5189566063dSJacob Faibussowitsch       PetscCall(MatDestroy(&H));
519f4c1ad5cSStefano Zampini     }
520f4c1ad5cSStefano Zampini     flg = PETSC_FALSE;
5219566063dSJacob Faibussowitsch     PetscCall(PetscOptionsBool("-tao_mf_hessian","compute matrix-free hessian using finite differences","TaoDefaultComputeHessianMFFD",flg,&flg,NULL));
522f4c1ad5cSStefano Zampini     if (flg) {
523f4c1ad5cSStefano Zampini       Mat H;
524f4c1ad5cSStefano Zampini 
5259566063dSJacob Faibussowitsch       PetscCall(MatCreate(PetscObjectComm((PetscObject)tao),&H));
5269566063dSJacob Faibussowitsch       PetscCall(TaoSetHessian(tao,H,H,TaoDefaultComputeHessianMFFD,NULL));
5279566063dSJacob Faibussowitsch       PetscCall(MatDestroy(&H));
528f4c1ad5cSStefano Zampini     }
529414d97d3SAlp Dener     flg = PETSC_FALSE;
5309566063dSJacob Faibussowitsch     PetscCall(PetscOptionsBool("-tao_recycle_history","enable recycling/re-using information from the previous TaoSolve() call for some algorithms","TaoSetRecycleHistory",flg,&flg,NULL));
531414d97d3SAlp Dener     if (flg) {
5329566063dSJacob Faibussowitsch       PetscCall(TaoSetRecycleHistory(tao,PETSC_TRUE));
533414d97d3SAlp Dener     }
5349566063dSJacob Faibussowitsch     PetscCall(PetscOptionsEnum("-tao_subset_type","subset type","",TaoSubSetTypes,(PetscEnum)tao->subset_type,(PetscEnum*)&tao->subset_type,NULL));
535a7e14dcfSSatish Balay 
5360f0abf79SStefano Zampini     if (tao->ksp) {
5370f0abf79SStefano Zampini       PetscCall(PetscOptionsBool("-tao_ksp_ew","Use Eisentat-Walker linear system convergence test","TaoKSPSetUseEW",tao->ksp_ewconv,&tao->ksp_ewconv,NULL));
5380f0abf79SStefano Zampini       PetscCall(TaoKSPSetUseEW(tao,tao->ksp_ewconv));
5390f0abf79SStefano Zampini     }
5400f0abf79SStefano Zampini 
541a82e8c82SStefano Zampini     if (tao->linesearch) {
5429566063dSJacob Faibussowitsch       PetscCall(TaoLineSearchSetFromOptions(tao->linesearch));
543a82e8c82SStefano Zampini     }
544a82e8c82SStefano Zampini 
545a7e14dcfSSatish Balay     if (tao->ops->setfromoptions) {
5469566063dSJacob Faibussowitsch       PetscCall((*tao->ops->setfromoptions)(PetscOptionsObject,tao));
547a7e14dcfSSatish Balay     }
548a7e14dcfSSatish Balay   }
549d0609cedSBarry Smith   PetscOptionsEnd();
550a7e14dcfSSatish Balay   PetscFunctionReturn(0);
551a7e14dcfSSatish Balay }
552a7e14dcfSSatish Balay 
553a7e14dcfSSatish Balay /*@C
554fe2efc57SMark    TaoViewFromOptions - View from Options
555fe2efc57SMark 
556fe2efc57SMark    Collective on Tao
557fe2efc57SMark 
558fe2efc57SMark    Input Parameters:
559fe2efc57SMark +  A - the  Tao context
560736c3998SJose E. Roman .  obj - Optional object
561736c3998SJose E. Roman -  name - command line option
562fe2efc57SMark 
563fe2efc57SMark    Level: intermediate
564db781477SPatrick Sanan .seealso: `Tao`, `TaoView`, `PetscObjectViewFromOptions()`, `TaoCreate()`
565fe2efc57SMark @*/
566fe2efc57SMark PetscErrorCode  TaoViewFromOptions(Tao A,PetscObject obj,const char name[])
567fe2efc57SMark {
568fe2efc57SMark   PetscFunctionBegin;
569fe2efc57SMark   PetscValidHeaderSpecific(A,TAO_CLASSID,1);
5709566063dSJacob Faibussowitsch   PetscCall(PetscObjectViewFromOptions((PetscObject)A,obj,name));
571fe2efc57SMark   PetscFunctionReturn(0);
572fe2efc57SMark }
573fe2efc57SMark 
574fe2efc57SMark /*@C
575441846f8SBarry Smith   TaoView - Prints information about the Tao
576a7e14dcfSSatish Balay 
577441846f8SBarry Smith   Collective on Tao
578a7e14dcfSSatish Balay 
579a7e14dcfSSatish Balay   InputParameters:
580441846f8SBarry Smith + tao - the Tao context
581a7e14dcfSSatish Balay - viewer - visualization context
582a7e14dcfSSatish Balay 
583a7e14dcfSSatish Balay   Options Database Key:
584a7e14dcfSSatish Balay . -tao_view - Calls TaoView() at the end of TaoSolve()
585a7e14dcfSSatish Balay 
586a7e14dcfSSatish Balay   Notes:
587a7e14dcfSSatish Balay   The available visualization contexts include
588a7e14dcfSSatish Balay +     PETSC_VIEWER_STDOUT_SELF - standard output (default)
589a7e14dcfSSatish Balay -     PETSC_VIEWER_STDOUT_WORLD - synchronized standard
590a7e14dcfSSatish Balay          output where only the first processor opens
591a7e14dcfSSatish Balay          the file.  All other processors send their
592a7e14dcfSSatish Balay          data to the first processor to print.
593a7e14dcfSSatish Balay 
594a7e14dcfSSatish Balay   Level: beginner
595a7e14dcfSSatish Balay 
596db781477SPatrick Sanan .seealso: `PetscViewerASCIIOpen()`
597a7e14dcfSSatish Balay @*/
598441846f8SBarry Smith PetscErrorCode TaoView(Tao tao, PetscViewer viewer)
599a7e14dcfSSatish Balay {
600a7e14dcfSSatish Balay   PetscBool           isascii,isstring;
601b625d6c7SJed Brown   TaoType             type;
60247a47007SBarry Smith 
603a7e14dcfSSatish Balay   PetscFunctionBegin;
604441846f8SBarry Smith   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
605a7e14dcfSSatish Balay   if (!viewer) {
6069566063dSJacob Faibussowitsch     PetscCall(PetscViewerASCIIGetStdout(((PetscObject)tao)->comm,&viewer));
607a7e14dcfSSatish Balay   }
608a7e14dcfSSatish Balay   PetscValidHeaderSpecific(viewer,PETSC_VIEWER_CLASSID,2);
609a7e14dcfSSatish Balay   PetscCheckSameComm(tao,1,viewer,2);
610a7e14dcfSSatish Balay 
6119566063dSJacob Faibussowitsch   PetscCall(PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERASCII,&isascii));
6129566063dSJacob Faibussowitsch   PetscCall(PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERSTRING,&isstring));
613a7e14dcfSSatish Balay   if (isascii) {
6149566063dSJacob Faibussowitsch     PetscCall(PetscObjectPrintClassNamePrefixType((PetscObject)tao,viewer));
615a7e14dcfSSatish Balay 
616a7e14dcfSSatish Balay     if (tao->ops->view) {
6179566063dSJacob Faibussowitsch       PetscCall(PetscViewerASCIIPushTab(viewer));
6189566063dSJacob Faibussowitsch       PetscCall((*tao->ops->view)(tao,viewer));
6199566063dSJacob Faibussowitsch       PetscCall(PetscViewerASCIIPopTab(viewer));
620a7e14dcfSSatish Balay     }
6218d3f3ddaSAlp Dener     if (tao->linesearch) {
6229566063dSJacob Faibussowitsch       PetscCall(PetscViewerASCIIPushTab(viewer));
6239566063dSJacob Faibussowitsch       PetscCall(TaoLineSearchView(tao->linesearch,viewer));
6249566063dSJacob Faibussowitsch       PetscCall(PetscViewerASCIIPopTab(viewer));
6258d3f3ddaSAlp Dener     }
626a7e14dcfSSatish Balay     if (tao->ksp) {
6279566063dSJacob Faibussowitsch       PetscCall(PetscViewerASCIIPushTab(viewer));
6289566063dSJacob Faibussowitsch       PetscCall(KSPView(tao->ksp,viewer));
62963a3b9bcSJacob Faibussowitsch       PetscCall(PetscViewerASCIIPrintf(viewer,"total KSP iterations: %" PetscInt_FMT "\n",tao->ksp_tot_its));
6309566063dSJacob Faibussowitsch       PetscCall(PetscViewerASCIIPopTab(viewer));
631a7e14dcfSSatish Balay     }
63219b5c101SAlp Dener 
6339566063dSJacob Faibussowitsch     PetscCall(PetscViewerASCIIPushTab(viewer));
63419b5c101SAlp Dener 
635a7e14dcfSSatish Balay     if (tao->XL || tao->XU) {
6369566063dSJacob Faibussowitsch       PetscCall(PetscViewerASCIIPrintf(viewer,"Active Set subset type: %s\n",TaoSubSetTypes[tao->subset_type]));
637a7e14dcfSSatish Balay     }
638a7e14dcfSSatish Balay 
6399566063dSJacob Faibussowitsch     PetscCall(PetscViewerASCIIPrintf(viewer,"convergence tolerances: gatol=%g,",(double)tao->gatol));
6409566063dSJacob Faibussowitsch     PetscCall(PetscViewerASCIIPrintf(viewer," steptol=%g,",(double)tao->steptol));
6419566063dSJacob Faibussowitsch     PetscCall(PetscViewerASCIIPrintf(viewer," gttol=%g\n",(double)tao->gttol));
6429566063dSJacob Faibussowitsch     PetscCall(PetscViewerASCIIPrintf(viewer,"Residual in Function/Gradient:=%g\n",(double)tao->residual));
643a7e14dcfSSatish Balay 
6446246e8cdSAlp Dener     if (tao->constrained) {
6459566063dSJacob Faibussowitsch       PetscCall(PetscViewerASCIIPrintf(viewer,"convergence tolerances:"));
6469566063dSJacob Faibussowitsch       PetscCall(PetscViewerASCIIPrintf(viewer," catol=%g,",(double)tao->catol));
6479566063dSJacob Faibussowitsch       PetscCall(PetscViewerASCIIPrintf(viewer," crtol=%g\n",(double)tao->crtol));
6489566063dSJacob Faibussowitsch       PetscCall(PetscViewerASCIIPrintf(viewer,"Residual in Constraints:=%g\n",(double)tao->cnorm));
649a7e14dcfSSatish Balay     }
650a7e14dcfSSatish Balay 
651a7e14dcfSSatish Balay     if (tao->trust < tao->steptol) {
6529566063dSJacob Faibussowitsch       PetscCall(PetscViewerASCIIPrintf(viewer,"convergence tolerances: steptol=%g\n",(double)tao->steptol));
6539566063dSJacob Faibussowitsch       PetscCall(PetscViewerASCIIPrintf(viewer,"Final trust region radius:=%g\n",(double)tao->trust));
654a7e14dcfSSatish Balay     }
655a7e14dcfSSatish Balay 
656a7e14dcfSSatish Balay     if (tao->fmin>-1.e25) {
6579566063dSJacob Faibussowitsch       PetscCall(PetscViewerASCIIPrintf(viewer,"convergence tolerances: function minimum=%g\n",(double)tao->fmin));
658a7e14dcfSSatish Balay     }
6599566063dSJacob Faibussowitsch     PetscCall(PetscViewerASCIIPrintf(viewer,"Objective value=%g\n",(double)tao->fc));
660a7e14dcfSSatish Balay 
66163a3b9bcSJacob Faibussowitsch     PetscCall(PetscViewerASCIIPrintf(viewer,"total number of iterations=%" PetscInt_FMT ",          ",tao->niter));
66263a3b9bcSJacob Faibussowitsch     PetscCall(PetscViewerASCIIPrintf(viewer,"              (max: %" PetscInt_FMT ")\n",tao->max_it));
663a7e14dcfSSatish Balay 
664a7e14dcfSSatish Balay     if (tao->nfuncs>0) {
66563a3b9bcSJacob Faibussowitsch       PetscCall(PetscViewerASCIIPrintf(viewer,"total number of function evaluations=%" PetscInt_FMT ",",tao->nfuncs));
66663a3b9bcSJacob Faibussowitsch       PetscCall(PetscViewerASCIIPrintf(viewer,"                max: %" PetscInt_FMT "\n",tao->max_funcs));
667a7e14dcfSSatish Balay     }
668a7e14dcfSSatish Balay     if (tao->ngrads>0) {
66963a3b9bcSJacob Faibussowitsch       PetscCall(PetscViewerASCIIPrintf(viewer,"total number of gradient evaluations=%" PetscInt_FMT ",",tao->ngrads));
67063a3b9bcSJacob Faibussowitsch       PetscCall(PetscViewerASCIIPrintf(viewer,"                max: %" PetscInt_FMT "\n",tao->max_funcs));
671a7e14dcfSSatish Balay     }
672a7e14dcfSSatish Balay     if (tao->nfuncgrads>0) {
67363a3b9bcSJacob Faibussowitsch       PetscCall(PetscViewerASCIIPrintf(viewer,"total number of function/gradient evaluations=%" PetscInt_FMT ",",tao->nfuncgrads));
67463a3b9bcSJacob Faibussowitsch       PetscCall(PetscViewerASCIIPrintf(viewer,"    (max: %" PetscInt_FMT ")\n",tao->max_funcs));
675a7e14dcfSSatish Balay     }
676a7e14dcfSSatish Balay     if (tao->nhess>0) {
67763a3b9bcSJacob Faibussowitsch       PetscCall(PetscViewerASCIIPrintf(viewer,"total number of Hessian evaluations=%" PetscInt_FMT "\n",tao->nhess));
678a7e14dcfSSatish Balay     }
679a7e14dcfSSatish Balay     if (tao->nconstraints>0) {
68063a3b9bcSJacob Faibussowitsch       PetscCall(PetscViewerASCIIPrintf(viewer,"total number of constraint function evaluations=%" PetscInt_FMT "\n",tao->nconstraints));
681a7e14dcfSSatish Balay     }
682a7e14dcfSSatish Balay     if (tao->njac>0) {
68363a3b9bcSJacob Faibussowitsch       PetscCall(PetscViewerASCIIPrintf(viewer,"total number of Jacobian evaluations=%" PetscInt_FMT "\n",tao->njac));
684a7e14dcfSSatish Balay     }
685a7e14dcfSSatish Balay 
686a7e14dcfSSatish Balay     if (tao->reason>0) {
6879566063dSJacob Faibussowitsch       PetscCall(PetscViewerASCIIPrintf(viewer,    "Solution converged: "));
688a7e14dcfSSatish Balay       switch (tao->reason) {
689a7e14dcfSSatish Balay       case TAO_CONVERGED_GATOL:
6909566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer," ||g(X)|| <= gatol\n"));
691a7e14dcfSSatish Balay         break;
692a7e14dcfSSatish Balay       case TAO_CONVERGED_GRTOL:
6939566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer," ||g(X)||/|f(X)| <= grtol\n"));
694a7e14dcfSSatish Balay         break;
695a7e14dcfSSatish Balay       case TAO_CONVERGED_GTTOL:
6969566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer," ||g(X)||/||g(X0)|| <= gttol\n"));
697a7e14dcfSSatish Balay         break;
698a7e14dcfSSatish Balay       case TAO_CONVERGED_STEPTOL:
6999566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer," Steptol -- step size small\n"));
700a7e14dcfSSatish Balay         break;
701a7e14dcfSSatish Balay       case TAO_CONVERGED_MINF:
7029566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer," Minf --  f < fmin\n"));
703a7e14dcfSSatish Balay         break;
704a7e14dcfSSatish Balay       case TAO_CONVERGED_USER:
7059566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer," User Terminated\n"));
706a7e14dcfSSatish Balay         break;
707a7e14dcfSSatish Balay       default:
7089566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer,"\n"));
709a7e14dcfSSatish Balay         break;
710a7e14dcfSSatish Balay       }
711a7e14dcfSSatish Balay     } else {
7129566063dSJacob Faibussowitsch       PetscCall(PetscViewerASCIIPrintf(viewer,"Solver terminated: %d",tao->reason));
713a7e14dcfSSatish Balay       switch (tao->reason) {
714a7e14dcfSSatish Balay       case TAO_DIVERGED_MAXITS:
7159566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer," Maximum Iterations\n"));
716a7e14dcfSSatish Balay         break;
717a7e14dcfSSatish Balay       case TAO_DIVERGED_NAN:
7189566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer," NAN or Inf encountered\n"));
719a7e14dcfSSatish Balay         break;
720a7e14dcfSSatish Balay       case TAO_DIVERGED_MAXFCN:
7219566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer," Maximum Function Evaluations\n"));
722a7e14dcfSSatish Balay         break;
723a7e14dcfSSatish Balay       case TAO_DIVERGED_LS_FAILURE:
7249566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer," Line Search Failure\n"));
725a7e14dcfSSatish Balay         break;
726a7e14dcfSSatish Balay       case TAO_DIVERGED_TR_REDUCTION:
7279566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer," Trust Region too small\n"));
728a7e14dcfSSatish Balay         break;
729a7e14dcfSSatish Balay       case TAO_DIVERGED_USER:
7309566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer," User Terminated\n"));
731a7e14dcfSSatish Balay         break;
732a7e14dcfSSatish Balay       default:
7339566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer,"\n"));
734a7e14dcfSSatish Balay         break;
735a7e14dcfSSatish Balay       }
736a7e14dcfSSatish Balay     }
7379566063dSJacob Faibussowitsch     PetscCall(PetscViewerASCIIPopTab(viewer));
738a7e14dcfSSatish Balay   } else if (isstring) {
7399566063dSJacob Faibussowitsch     PetscCall(TaoGetType(tao,&type));
7409566063dSJacob Faibussowitsch     PetscCall(PetscViewerStringSPrintf(viewer," %-3.3s",type));
741a7e14dcfSSatish Balay   }
742a7e14dcfSSatish Balay   PetscFunctionReturn(0);
743a7e14dcfSSatish Balay }
744a7e14dcfSSatish Balay 
745a7e14dcfSSatish Balay /*@
746414d97d3SAlp Dener   TaoSetRecycleHistory - Sets the boolean flag to enable/disable re-using
747414d97d3SAlp Dener   iterate information from the previous TaoSolve(). This feature is disabled by
748414d97d3SAlp Dener   default.
749414d97d3SAlp Dener 
750414d97d3SAlp Dener   For conjugate gradient methods (BNCG), this re-uses the latest search direction
751414d97d3SAlp Dener   from the previous TaoSolve() call when computing the first search direction in a
752414d97d3SAlp Dener   new solution. By default, CG methods set the first search direction to the
753414d97d3SAlp Dener   negative gradient.
754414d97d3SAlp Dener 
755414d97d3SAlp Dener   For quasi-Newton family of methods (BQNLS, BQNKLS, BQNKTR, BQNKTL), this re-uses
756414d97d3SAlp Dener   the accumulated quasi-Newton Hessian approximation from the previous TaoSolve()
757414d97d3SAlp Dener   call. By default, QN family of methods reset the initial Hessian approximation to
758414d97d3SAlp Dener   the identity matrix.
759414d97d3SAlp Dener 
760414d97d3SAlp Dener   For any other algorithm, this setting has no effect.
761414d97d3SAlp Dener 
762414d97d3SAlp Dener   Logically collective on Tao
763414d97d3SAlp Dener 
764414d97d3SAlp Dener   Input Parameters:
765414d97d3SAlp Dener + tao - the Tao context
766414d97d3SAlp Dener - recycle - boolean flag
767414d97d3SAlp Dener 
768414d97d3SAlp Dener   Options Database Keys:
769147403d9SBarry Smith . -tao_recycle_history <true,false> - reuse the history
770414d97d3SAlp Dener 
771414d97d3SAlp Dener   Level: intermediate
772414d97d3SAlp Dener 
773db781477SPatrick Sanan .seealso: `TaoSetRecycleHistory()`, `TAOBNCG`, `TAOBQNLS`, `TAOBQNKLS`, `TAOBQNKTR`, `TAOBQNKTL`
774414d97d3SAlp Dener 
775414d97d3SAlp Dener @*/
776414d97d3SAlp Dener PetscErrorCode TaoSetRecycleHistory(Tao tao, PetscBool recycle)
777414d97d3SAlp Dener {
778414d97d3SAlp Dener   PetscFunctionBegin;
779414d97d3SAlp Dener   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
78094511df7SStefano Zampini   PetscValidLogicalCollectiveBool(tao,recycle,2);
781414d97d3SAlp Dener   tao->recycle = recycle;
782414d97d3SAlp Dener   PetscFunctionReturn(0);
783414d97d3SAlp Dener }
784414d97d3SAlp Dener 
785414d97d3SAlp Dener /*@
786414d97d3SAlp Dener   TaoGetRecycleHistory - Retrieve the boolean flag for re-using iterate information
787414d97d3SAlp Dener   from the previous TaoSolve(). This feature is disabled by default.
788414d97d3SAlp Dener 
789414d97d3SAlp Dener   Logically collective on Tao
790414d97d3SAlp Dener 
791414d97d3SAlp Dener   Input Parameters:
792147403d9SBarry Smith . tao - the Tao context
793414d97d3SAlp Dener 
794414d97d3SAlp Dener   Output Parameters:
795147403d9SBarry Smith . recycle - boolean flag
796414d97d3SAlp Dener 
797414d97d3SAlp Dener   Level: intermediate
798414d97d3SAlp Dener 
799db781477SPatrick Sanan .seealso: `TaoGetRecycleHistory()`, `TAOBNCG`, `TAOBQNLS`, `TAOBQNKLS`, `TAOBQNKTR`, `TAOBQNKTL`
800414d97d3SAlp Dener 
801414d97d3SAlp Dener @*/
802414d97d3SAlp Dener PetscErrorCode TaoGetRecycleHistory(Tao tao, PetscBool *recycle)
803414d97d3SAlp Dener {
804414d97d3SAlp Dener   PetscFunctionBegin;
805414d97d3SAlp Dener   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
80694511df7SStefano Zampini   PetscValidBoolPointer(recycle,2);
807414d97d3SAlp Dener   *recycle = tao->recycle;
808414d97d3SAlp Dener   PetscFunctionReturn(0);
809414d97d3SAlp Dener }
810414d97d3SAlp Dener 
811414d97d3SAlp Dener /*@
812a7e14dcfSSatish Balay   TaoSetTolerances - Sets parameters used in TAO convergence tests
813a7e14dcfSSatish Balay 
814441846f8SBarry Smith   Logically collective on Tao
815a7e14dcfSSatish Balay 
816a7e14dcfSSatish Balay   Input Parameters:
817441846f8SBarry Smith + tao - the Tao context
818a7e14dcfSSatish Balay . gatol - stop if norm of gradient is less than this
819a7e14dcfSSatish Balay . grtol - stop if relative norm of gradient is less than this
820a7e14dcfSSatish Balay - gttol - stop if norm of gradient is reduced by this factor
821a7e14dcfSSatish Balay 
822a7e14dcfSSatish Balay   Options Database Keys:
823e52336cbSBarry Smith + -tao_gatol <gatol> - Sets gatol
824a7e14dcfSSatish Balay . -tao_grtol <grtol> - Sets grtol
825a7e14dcfSSatish Balay - -tao_gttol <gttol> - Sets gttol
826a7e14dcfSSatish Balay 
827a7e14dcfSSatish Balay   Stopping Criteria:
828a7e14dcfSSatish Balay $ ||g(X)||                            <= gatol
829a7e14dcfSSatish Balay $ ||g(X)|| / |f(X)|                   <= grtol
830a7e14dcfSSatish Balay $ ||g(X)|| / ||g(X0)||                <= gttol
831a7e14dcfSSatish Balay 
832a7e14dcfSSatish Balay   Notes:
833a7e14dcfSSatish Balay   Use PETSC_DEFAULT to leave one or more tolerances unchanged.
834a7e14dcfSSatish Balay 
835a7e14dcfSSatish Balay   Level: beginner
836a7e14dcfSSatish Balay 
837db781477SPatrick Sanan .seealso: `TaoGetTolerances()`
838a7e14dcfSSatish Balay 
839a7e14dcfSSatish Balay @*/
840e52336cbSBarry Smith PetscErrorCode TaoSetTolerances(Tao tao, PetscReal gatol, PetscReal grtol, PetscReal gttol)
841a7e14dcfSSatish Balay {
842a7e14dcfSSatish Balay   PetscFunctionBegin;
843441846f8SBarry Smith   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
84494511df7SStefano Zampini   PetscValidLogicalCollectiveReal(tao,gatol,2);
84594511df7SStefano Zampini   PetscValidLogicalCollectiveReal(tao,grtol,3);
84694511df7SStefano Zampini   PetscValidLogicalCollectiveReal(tao,gttol,4);
847a7e14dcfSSatish Balay 
848a7e14dcfSSatish Balay   if (gatol != PETSC_DEFAULT) {
849a7e14dcfSSatish Balay     if (gatol<0) {
8509566063dSJacob Faibussowitsch       PetscCall(PetscInfo(tao,"Tried to set negative gatol -- ignored.\n"));
851a7e14dcfSSatish Balay     } else {
852a7e14dcfSSatish Balay       tao->gatol = PetscMax(0,gatol);
8536552cf8aSJason Sarich       tao->gatol_changed = PETSC_TRUE;
854a7e14dcfSSatish Balay     }
855a7e14dcfSSatish Balay   }
856a7e14dcfSSatish Balay 
857a7e14dcfSSatish Balay   if (grtol != PETSC_DEFAULT) {
858a7e14dcfSSatish Balay     if (grtol<0) {
8599566063dSJacob Faibussowitsch       PetscCall(PetscInfo(tao,"Tried to set negative grtol -- ignored.\n"));
860a7e14dcfSSatish Balay     } else {
861a7e14dcfSSatish Balay       tao->grtol = PetscMax(0,grtol);
8626552cf8aSJason Sarich       tao->grtol_changed = PETSC_TRUE;
863a7e14dcfSSatish Balay     }
864a7e14dcfSSatish Balay   }
865a7e14dcfSSatish Balay 
866a7e14dcfSSatish Balay   if (gttol != PETSC_DEFAULT) {
867a7e14dcfSSatish Balay     if (gttol<0) {
8689566063dSJacob Faibussowitsch       PetscCall(PetscInfo(tao,"Tried to set negative gttol -- ignored.\n"));
869a7e14dcfSSatish Balay     } else {
870a7e14dcfSSatish Balay       tao->gttol = PetscMax(0,gttol);
8716552cf8aSJason Sarich       tao->gttol_changed = PETSC_TRUE;
872a7e14dcfSSatish Balay     }
873a7e14dcfSSatish Balay   }
874a7e14dcfSSatish Balay   PetscFunctionReturn(0);
875a7e14dcfSSatish Balay }
876a7e14dcfSSatish Balay 
877a7e14dcfSSatish Balay /*@
87858417fe7SBarry Smith   TaoSetConstraintTolerances - Sets constraint tolerance parameters used in TAO  convergence tests
879a7e14dcfSSatish Balay 
880441846f8SBarry Smith   Logically collective on Tao
881a7e14dcfSSatish Balay 
882a7e14dcfSSatish Balay   Input Parameters:
883441846f8SBarry Smith + tao - the Tao context
884e52336cbSBarry Smith . catol - absolute constraint tolerance, constraint norm must be less than catol for used for gatol convergence criteria
8856aad120cSJose E. Roman - crtol - relative constraint tolerance, constraint norm must be less than crtol for used for gatol, gttol convergence criteria
886a7e14dcfSSatish Balay 
887a7e14dcfSSatish Balay   Options Database Keys:
888a7e14dcfSSatish Balay + -tao_catol <catol> - Sets catol
889a7e14dcfSSatish Balay - -tao_crtol <crtol> - Sets crtol
890a7e14dcfSSatish Balay 
8916552cf8aSJason Sarich   Notes:
8926552cf8aSJason Sarich   Use PETSC_DEFAULT to leave any tolerance unchanged.
8936552cf8aSJason Sarich 
894a7e14dcfSSatish Balay   Level: intermediate
895a7e14dcfSSatish Balay 
896db781477SPatrick Sanan .seealso: `TaoGetTolerances()`, `TaoGetConstraintTolerances()`, `TaoSetTolerances()`
897a7e14dcfSSatish Balay 
898a7e14dcfSSatish Balay @*/
899441846f8SBarry Smith PetscErrorCode TaoSetConstraintTolerances(Tao tao, PetscReal catol, PetscReal crtol)
900a7e14dcfSSatish Balay {
901a7e14dcfSSatish Balay   PetscFunctionBegin;
902441846f8SBarry Smith   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
90394511df7SStefano Zampini   PetscValidLogicalCollectiveReal(tao,catol,2);
90494511df7SStefano Zampini   PetscValidLogicalCollectiveReal(tao,crtol,3);
905a7e14dcfSSatish Balay 
906a7e14dcfSSatish Balay   if (catol != PETSC_DEFAULT) {
907a7e14dcfSSatish Balay     if (catol<0) {
9089566063dSJacob Faibussowitsch       PetscCall(PetscInfo(tao,"Tried to set negative catol -- ignored.\n"));
909a7e14dcfSSatish Balay     } else {
910a7e14dcfSSatish Balay       tao->catol = PetscMax(0,catol);
9116552cf8aSJason Sarich       tao->catol_changed = PETSC_TRUE;
912a7e14dcfSSatish Balay     }
913a7e14dcfSSatish Balay   }
914a7e14dcfSSatish Balay 
915a7e14dcfSSatish Balay   if (crtol != PETSC_DEFAULT) {
916a7e14dcfSSatish Balay     if (crtol<0) {
9179566063dSJacob Faibussowitsch       PetscCall(PetscInfo(tao,"Tried to set negative crtol -- ignored.\n"));
918a7e14dcfSSatish Balay     } else {
919a7e14dcfSSatish Balay       tao->crtol = PetscMax(0,crtol);
9206552cf8aSJason Sarich       tao->crtol_changed = PETSC_TRUE;
921a7e14dcfSSatish Balay     }
922a7e14dcfSSatish Balay   }
923a7e14dcfSSatish Balay   PetscFunctionReturn(0);
924a7e14dcfSSatish Balay }
925a7e14dcfSSatish Balay 
92658417fe7SBarry Smith /*@
92758417fe7SBarry Smith   TaoGetConstraintTolerances - Gets constraint tolerance parameters used in TAO  convergence tests
92858417fe7SBarry Smith 
92958417fe7SBarry Smith   Not ollective
93058417fe7SBarry Smith 
93158417fe7SBarry Smith   Input Parameter:
93258417fe7SBarry Smith . tao - the Tao context
93358417fe7SBarry Smith 
934d8d19677SJose E. Roman   Output Parameters:
935e52336cbSBarry Smith + catol - absolute constraint tolerance, constraint norm must be less than catol for used for gatol convergence criteria
9366aad120cSJose E. Roman - crtol - relative constraint tolerance, constraint norm must be less than crtol for used for gatol, gttol convergence criteria
93758417fe7SBarry Smith 
93858417fe7SBarry Smith   Level: intermediate
93958417fe7SBarry Smith 
940db781477SPatrick Sanan .seealso: `TaoGetTolerances()`, `TaoSetTolerances()`, `TaoSetConstraintTolerances()`
94158417fe7SBarry Smith 
94258417fe7SBarry Smith @*/
94358417fe7SBarry Smith PetscErrorCode TaoGetConstraintTolerances(Tao tao, PetscReal *catol, PetscReal *crtol)
94458417fe7SBarry Smith {
94558417fe7SBarry Smith   PetscFunctionBegin;
94658417fe7SBarry Smith   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
94758417fe7SBarry Smith   if (catol) *catol = tao->catol;
94858417fe7SBarry Smith   if (crtol) *crtol = tao->crtol;
94958417fe7SBarry Smith   PetscFunctionReturn(0);
95058417fe7SBarry Smith }
95158417fe7SBarry Smith 
952a7e14dcfSSatish Balay /*@
953a7e14dcfSSatish Balay    TaoSetFunctionLowerBound - Sets a bound on the solution objective value.
954a7e14dcfSSatish Balay    When an approximate solution with an objective value below this number
955a7e14dcfSSatish Balay    has been found, the solver will terminate.
956a7e14dcfSSatish Balay 
957441846f8SBarry Smith    Logically Collective on Tao
958a7e14dcfSSatish Balay 
959a7e14dcfSSatish Balay    Input Parameters:
960441846f8SBarry Smith +  tao - the Tao solver context
961a7e14dcfSSatish Balay -  fmin - the tolerance
962a7e14dcfSSatish Balay 
963a7e14dcfSSatish Balay    Options Database Keys:
964a7e14dcfSSatish Balay .    -tao_fmin <fmin> - sets the minimum function value
965a7e14dcfSSatish Balay 
966a7e14dcfSSatish Balay    Level: intermediate
967a7e14dcfSSatish Balay 
968db781477SPatrick Sanan .seealso: `TaoSetTolerances()`
969a7e14dcfSSatish Balay @*/
970441846f8SBarry Smith PetscErrorCode TaoSetFunctionLowerBound(Tao tao,PetscReal fmin)
971a7e14dcfSSatish Balay {
972a7e14dcfSSatish Balay   PetscFunctionBegin;
973441846f8SBarry Smith   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
97494511df7SStefano Zampini   PetscValidLogicalCollectiveReal(tao,fmin,2);
975a7e14dcfSSatish Balay   tao->fmin = fmin;
9766552cf8aSJason Sarich   tao->fmin_changed = PETSC_TRUE;
977a7e14dcfSSatish Balay   PetscFunctionReturn(0);
978a7e14dcfSSatish Balay }
979a7e14dcfSSatish Balay 
980a7e14dcfSSatish Balay /*@
9818e96d397SJason Sarich    TaoGetFunctionLowerBound - Gets the bound on the solution objective value.
982a7e14dcfSSatish Balay    When an approximate solution with an objective value below this number
983a7e14dcfSSatish Balay    has been found, the solver will terminate.
984a7e14dcfSSatish Balay 
985441846f8SBarry Smith    Not collective on Tao
986a7e14dcfSSatish Balay 
987a7e14dcfSSatish Balay    Input Parameters:
988441846f8SBarry Smith .  tao - the Tao solver context
989a7e14dcfSSatish Balay 
990a7e14dcfSSatish Balay    OutputParameters:
991a7e14dcfSSatish Balay .  fmin - the minimum function value
992a7e14dcfSSatish Balay 
993a7e14dcfSSatish Balay    Level: intermediate
994a7e14dcfSSatish Balay 
995db781477SPatrick Sanan .seealso: `TaoSetFunctionLowerBound()`
996a7e14dcfSSatish Balay @*/
997441846f8SBarry Smith PetscErrorCode TaoGetFunctionLowerBound(Tao tao,PetscReal *fmin)
998a7e14dcfSSatish Balay {
999a7e14dcfSSatish Balay   PetscFunctionBegin;
1000441846f8SBarry Smith   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
100194511df7SStefano Zampini   PetscValidRealPointer(fmin,2);
1002a7e14dcfSSatish Balay   *fmin = tao->fmin;
1003a7e14dcfSSatish Balay   PetscFunctionReturn(0);
1004a7e14dcfSSatish Balay }
1005a7e14dcfSSatish Balay 
1006a7e14dcfSSatish Balay /*@
1007a7e14dcfSSatish Balay    TaoSetMaximumFunctionEvaluations - Sets a maximum number of
1008a7e14dcfSSatish Balay    function evaluations.
1009a7e14dcfSSatish Balay 
1010441846f8SBarry Smith    Logically Collective on Tao
1011a7e14dcfSSatish Balay 
1012a7e14dcfSSatish Balay    Input Parameters:
1013441846f8SBarry Smith +  tao - the Tao solver context
1014a7e14dcfSSatish Balay -  nfcn - the maximum number of function evaluations (>=0)
1015a7e14dcfSSatish Balay 
1016a7e14dcfSSatish Balay    Options Database Keys:
1017a7e14dcfSSatish Balay .    -tao_max_funcs <nfcn> - sets the maximum number of function evaluations
1018a7e14dcfSSatish Balay 
1019a7e14dcfSSatish Balay    Level: intermediate
1020a7e14dcfSSatish Balay 
1021db781477SPatrick Sanan .seealso: `TaoSetTolerances()`, `TaoSetMaximumIterations()`
1022a7e14dcfSSatish Balay @*/
1023a7e14dcfSSatish Balay 
1024441846f8SBarry Smith PetscErrorCode TaoSetMaximumFunctionEvaluations(Tao tao,PetscInt nfcn)
1025a7e14dcfSSatish Balay {
1026a7e14dcfSSatish Balay   PetscFunctionBegin;
1027441846f8SBarry Smith   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
102894511df7SStefano Zampini   PetscValidLogicalCollectiveInt(tao,nfcn,2);
102994511df7SStefano Zampini   if (nfcn >= 0) { tao->max_funcs = PetscMax(0,nfcn); }
103094511df7SStefano Zampini   else { tao->max_funcs = -1; }
10316552cf8aSJason Sarich   tao->max_funcs_changed = PETSC_TRUE;
1032a7e14dcfSSatish Balay   PetscFunctionReturn(0);
1033a7e14dcfSSatish Balay }
1034a7e14dcfSSatish Balay 
1035a7e14dcfSSatish Balay /*@
1036a7e14dcfSSatish Balay    TaoGetMaximumFunctionEvaluations - Sets a maximum number of
1037a7e14dcfSSatish Balay    function evaluations.
1038a7e14dcfSSatish Balay 
1039a7e14dcfSSatish Balay    Not Collective
1040a7e14dcfSSatish Balay 
1041a7e14dcfSSatish Balay    Input Parameters:
1042441846f8SBarry Smith .  tao - the Tao solver context
1043a7e14dcfSSatish Balay 
1044a7e14dcfSSatish Balay    Output Parameters:
1045a7e14dcfSSatish Balay .  nfcn - the maximum number of function evaluations
1046a7e14dcfSSatish Balay 
1047a7e14dcfSSatish Balay    Level: intermediate
1048a7e14dcfSSatish Balay 
1049db781477SPatrick Sanan .seealso: `TaoSetMaximumFunctionEvaluations()`, `TaoGetMaximumIterations()`
1050a7e14dcfSSatish Balay @*/
1051a7e14dcfSSatish Balay 
1052441846f8SBarry Smith PetscErrorCode TaoGetMaximumFunctionEvaluations(Tao tao,PetscInt *nfcn)
1053a7e14dcfSSatish Balay {
1054a7e14dcfSSatish Balay   PetscFunctionBegin;
1055441846f8SBarry Smith   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
105694511df7SStefano Zampini   PetscValidIntPointer(nfcn,2);
1057a7e14dcfSSatish Balay   *nfcn = tao->max_funcs;
1058a7e14dcfSSatish Balay   PetscFunctionReturn(0);
1059a7e14dcfSSatish Balay }
1060a7e14dcfSSatish Balay 
1061770232b9SCe Qin /*@
1062770232b9SCe Qin    TaoGetCurrentFunctionEvaluations - Get current number of
1063770232b9SCe Qin    function evaluations.
1064770232b9SCe Qin 
1065770232b9SCe Qin    Not Collective
1066770232b9SCe Qin 
1067770232b9SCe Qin    Input Parameters:
1068770232b9SCe Qin .  tao - the Tao solver context
1069770232b9SCe Qin 
1070770232b9SCe Qin    Output Parameters:
107194511df7SStefano Zampini .  nfuncs - the current number of function evaluations (maximum between gradient and function evaluations)
1072770232b9SCe Qin 
1073770232b9SCe Qin    Level: intermediate
1074770232b9SCe Qin 
1075db781477SPatrick Sanan .seealso: `TaoSetMaximumFunctionEvaluations()`, `TaoGetMaximumFunctionEvaluations()`, `TaoGetMaximumIterations()`
1076770232b9SCe Qin @*/
1077770232b9SCe Qin 
1078770232b9SCe Qin PetscErrorCode TaoGetCurrentFunctionEvaluations(Tao tao,PetscInt *nfuncs)
1079770232b9SCe Qin {
1080770232b9SCe Qin   PetscFunctionBegin;
1081770232b9SCe Qin   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
108294511df7SStefano Zampini   PetscValidIntPointer(nfuncs,2);
1083770232b9SCe Qin   *nfuncs = PetscMax(tao->nfuncs,tao->nfuncgrads);
1084770232b9SCe Qin   PetscFunctionReturn(0);
1085770232b9SCe Qin }
1086770232b9SCe Qin 
1087a7e14dcfSSatish Balay /*@
1088a7e14dcfSSatish Balay    TaoSetMaximumIterations - Sets a maximum number of iterates.
1089a7e14dcfSSatish Balay 
1090441846f8SBarry Smith    Logically Collective on Tao
1091a7e14dcfSSatish Balay 
1092a7e14dcfSSatish Balay    Input Parameters:
1093441846f8SBarry Smith +  tao - the Tao solver context
1094a7e14dcfSSatish Balay -  maxits - the maximum number of iterates (>=0)
1095a7e14dcfSSatish Balay 
1096a7e14dcfSSatish Balay    Options Database Keys:
1097a7e14dcfSSatish Balay .    -tao_max_it <its> - sets the maximum number of iterations
1098a7e14dcfSSatish Balay 
1099a7e14dcfSSatish Balay    Level: intermediate
1100a7e14dcfSSatish Balay 
1101db781477SPatrick Sanan .seealso: `TaoSetTolerances()`, `TaoSetMaximumFunctionEvaluations()`
1102a7e14dcfSSatish Balay @*/
1103441846f8SBarry Smith PetscErrorCode TaoSetMaximumIterations(Tao tao,PetscInt maxits)
1104a7e14dcfSSatish Balay {
1105a7e14dcfSSatish Balay   PetscFunctionBegin;
1106441846f8SBarry Smith   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
110794511df7SStefano Zampini   PetscValidLogicalCollectiveInt(tao,maxits,2);
110847a47007SBarry Smith   tao->max_it = PetscMax(0,maxits);
11096552cf8aSJason Sarich   tao->max_it_changed = PETSC_TRUE;
1110a7e14dcfSSatish Balay   PetscFunctionReturn(0);
1111a7e14dcfSSatish Balay }
1112a7e14dcfSSatish Balay 
1113a7e14dcfSSatish Balay /*@
1114a7e14dcfSSatish Balay    TaoGetMaximumIterations - Sets a maximum number of iterates.
1115a7e14dcfSSatish Balay 
1116a7e14dcfSSatish Balay    Not Collective
1117a7e14dcfSSatish Balay 
1118a7e14dcfSSatish Balay    Input Parameters:
1119441846f8SBarry Smith .  tao - the Tao solver context
1120a7e14dcfSSatish Balay 
1121a7e14dcfSSatish Balay    Output Parameters:
1122a7e14dcfSSatish Balay .  maxits - the maximum number of iterates
1123a7e14dcfSSatish Balay 
1124a7e14dcfSSatish Balay    Level: intermediate
1125a7e14dcfSSatish Balay 
1126db781477SPatrick Sanan .seealso: `TaoSetMaximumIterations()`, `TaoGetMaximumFunctionEvaluations()`
1127a7e14dcfSSatish Balay @*/
1128441846f8SBarry Smith PetscErrorCode TaoGetMaximumIterations(Tao tao,PetscInt *maxits)
1129a7e14dcfSSatish Balay {
1130a7e14dcfSSatish Balay   PetscFunctionBegin;
1131441846f8SBarry Smith   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
113294511df7SStefano Zampini   PetscValidIntPointer(maxits,2);
1133a7e14dcfSSatish Balay   *maxits = tao->max_it;
1134a7e14dcfSSatish Balay   PetscFunctionReturn(0);
1135a7e14dcfSSatish Balay }
1136a7e14dcfSSatish Balay 
1137a7e14dcfSSatish Balay /*@
1138a7e14dcfSSatish Balay    TaoSetInitialTrustRegionRadius - Sets the initial trust region radius.
1139a7e14dcfSSatish Balay 
1140441846f8SBarry Smith    Logically collective on Tao
1141a7e14dcfSSatish Balay 
1142d8d19677SJose E. Roman    Input Parameters:
1143a7e14dcfSSatish Balay +  tao - a TAO optimization solver
1144a7e14dcfSSatish Balay -  radius - the trust region radius
1145a7e14dcfSSatish Balay 
1146a7e14dcfSSatish Balay    Level: intermediate
1147a7e14dcfSSatish Balay 
1148a7e14dcfSSatish Balay    Options Database Key:
1149a7e14dcfSSatish Balay .  -tao_trust0 <t0> - sets initial trust region radius
1150a7e14dcfSSatish Balay 
1151db781477SPatrick Sanan .seealso: `TaoGetTrustRegionRadius()`, `TaoSetTrustRegionTolerance()`
1152a7e14dcfSSatish Balay @*/
1153441846f8SBarry Smith PetscErrorCode TaoSetInitialTrustRegionRadius(Tao tao, PetscReal radius)
1154a7e14dcfSSatish Balay {
1155a7e14dcfSSatish Balay   PetscFunctionBegin;
1156441846f8SBarry Smith   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
115794511df7SStefano Zampini   PetscValidLogicalCollectiveReal(tao,radius,2);
1158a7e14dcfSSatish Balay   tao->trust0 = PetscMax(0.0,radius);
11596552cf8aSJason Sarich   tao->trust0_changed = PETSC_TRUE;
1160a7e14dcfSSatish Balay   PetscFunctionReturn(0);
1161a7e14dcfSSatish Balay }
1162a7e14dcfSSatish Balay 
1163a7e14dcfSSatish Balay /*@
1164a7e14dcfSSatish Balay    TaoGetInitialTrustRegionRadius - Sets the initial trust region radius.
1165a7e14dcfSSatish Balay 
1166a7e14dcfSSatish Balay    Not Collective
1167a7e14dcfSSatish Balay 
1168a7e14dcfSSatish Balay    Input Parameter:
1169a7e14dcfSSatish Balay .  tao - a TAO optimization solver
1170a7e14dcfSSatish Balay 
1171a7e14dcfSSatish Balay    Output Parameter:
1172a7e14dcfSSatish Balay .  radius - the trust region radius
1173a7e14dcfSSatish Balay 
1174a7e14dcfSSatish Balay    Level: intermediate
1175a7e14dcfSSatish Balay 
1176db781477SPatrick Sanan .seealso: `TaoSetInitialTrustRegionRadius()`, `TaoGetCurrentTrustRegionRadius()`
1177a7e14dcfSSatish Balay @*/
1178441846f8SBarry Smith PetscErrorCode TaoGetInitialTrustRegionRadius(Tao tao, PetscReal *radius)
1179a7e14dcfSSatish Balay {
1180a7e14dcfSSatish Balay   PetscFunctionBegin;
1181441846f8SBarry Smith   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
118294511df7SStefano Zampini   PetscValidRealPointer(radius,2);
1183a7e14dcfSSatish Balay   *radius = tao->trust0;
1184a7e14dcfSSatish Balay   PetscFunctionReturn(0);
1185a7e14dcfSSatish Balay }
1186a7e14dcfSSatish Balay 
1187a7e14dcfSSatish Balay /*@
1188a7e14dcfSSatish Balay    TaoGetCurrentTrustRegionRadius - Gets the current trust region radius.
1189a7e14dcfSSatish Balay 
1190a7e14dcfSSatish Balay    Not Collective
1191a7e14dcfSSatish Balay 
1192a7e14dcfSSatish Balay    Input Parameter:
1193a7e14dcfSSatish Balay .  tao - a TAO optimization solver
1194a7e14dcfSSatish Balay 
1195a7e14dcfSSatish Balay    Output Parameter:
1196a7e14dcfSSatish Balay .  radius - the trust region radius
1197a7e14dcfSSatish Balay 
1198a7e14dcfSSatish Balay    Level: intermediate
1199a7e14dcfSSatish Balay 
1200db781477SPatrick Sanan .seealso: `TaoSetInitialTrustRegionRadius()`, `TaoGetInitialTrustRegionRadius()`
1201a7e14dcfSSatish Balay @*/
1202441846f8SBarry Smith PetscErrorCode TaoGetCurrentTrustRegionRadius(Tao tao, PetscReal *radius)
1203a7e14dcfSSatish Balay {
1204a7e14dcfSSatish Balay   PetscFunctionBegin;
1205441846f8SBarry Smith   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
120694511df7SStefano Zampini   PetscValidRealPointer(radius,2);
1207a7e14dcfSSatish Balay   *radius = tao->trust;
1208a7e14dcfSSatish Balay   PetscFunctionReturn(0);
1209a7e14dcfSSatish Balay }
1210a7e14dcfSSatish Balay 
1211a7e14dcfSSatish Balay /*@
1212a7e14dcfSSatish Balay   TaoGetTolerances - gets the current values of tolerances
1213a7e14dcfSSatish Balay 
1214a7e14dcfSSatish Balay   Not Collective
1215a7e14dcfSSatish Balay 
1216f899ff85SJose E. Roman   Input Parameter:
1217441846f8SBarry Smith . tao - the Tao context
1218a7e14dcfSSatish Balay 
1219a7e14dcfSSatish Balay   Output Parameters:
1220e52336cbSBarry Smith + gatol - stop if norm of gradient is less than this
1221a7e14dcfSSatish Balay . grtol - stop if relative norm of gradient is less than this
1222a7e14dcfSSatish Balay - gttol - stop if norm of gradient is reduced by a this factor
1223a7e14dcfSSatish Balay 
122447a47007SBarry Smith   Note: NULL can be used as an argument if not all tolerances values are needed
1225a7e14dcfSSatish Balay 
1226db781477SPatrick Sanan .seealso `TaoSetTolerances()`
1227a7e14dcfSSatish Balay 
1228a7e14dcfSSatish Balay   Level: intermediate
1229a7e14dcfSSatish Balay @*/
1230e52336cbSBarry Smith PetscErrorCode TaoGetTolerances(Tao tao, PetscReal *gatol, PetscReal *grtol, PetscReal *gttol)
1231a7e14dcfSSatish Balay {
1232a7e14dcfSSatish Balay   PetscFunctionBegin;
1233441846f8SBarry Smith   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
1234a7e14dcfSSatish Balay   if (gatol) *gatol = tao->gatol;
1235a7e14dcfSSatish Balay   if (grtol) *grtol = tao->grtol;
1236a7e14dcfSSatish Balay   if (gttol) *gttol = tao->gttol;
1237a7e14dcfSSatish Balay   PetscFunctionReturn(0);
1238a7e14dcfSSatish Balay }
1239a7e14dcfSSatish Balay 
1240a7e14dcfSSatish Balay /*@
1241a7e14dcfSSatish Balay   TaoGetKSP - Gets the linear solver used by the optimization solver.
1242a7e14dcfSSatish Balay   Application writers should use TaoGetKSP if they need direct access
1243a7e14dcfSSatish Balay   to the PETSc KSP object.
1244a7e14dcfSSatish Balay 
1245a7e14dcfSSatish Balay   Not Collective
1246a7e14dcfSSatish Balay 
1247a7e14dcfSSatish Balay    Input Parameters:
1248a7e14dcfSSatish Balay .  tao - the TAO solver
1249a7e14dcfSSatish Balay 
1250a7e14dcfSSatish Balay    Output Parameters:
1251a7e14dcfSSatish Balay .  ksp - the KSP linear solver used in the optimization solver
1252a7e14dcfSSatish Balay 
1253a7e14dcfSSatish Balay    Level: intermediate
1254a7e14dcfSSatish Balay 
1255a7e14dcfSSatish Balay @*/
1256441846f8SBarry Smith PetscErrorCode TaoGetKSP(Tao tao, KSP *ksp)
125747a47007SBarry Smith {
1258a7e14dcfSSatish Balay   PetscFunctionBegin;
125994511df7SStefano Zampini   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
126094511df7SStefano Zampini   PetscValidPointer(ksp,2);
1261a7e14dcfSSatish Balay   *ksp = tao->ksp;
1262a7e14dcfSSatish Balay   PetscFunctionReturn(0);
1263a7e14dcfSSatish Balay }
1264a7e14dcfSSatish Balay 
1265025e9500SJason Sarich /*@
1266025e9500SJason Sarich    TaoGetLinearSolveIterations - Gets the total number of linear iterations
1267025e9500SJason Sarich    used by the TAO solver
1268025e9500SJason Sarich 
1269025e9500SJason Sarich    Not Collective
1270025e9500SJason Sarich 
1271025e9500SJason Sarich    Input Parameter:
1272025e9500SJason Sarich .  tao - TAO context
1273025e9500SJason Sarich 
1274025e9500SJason Sarich    Output Parameter:
1275025e9500SJason Sarich .  lits - number of linear iterations
1276025e9500SJason Sarich 
1277025e9500SJason Sarich    Notes:
1278025e9500SJason Sarich    This counter is reset to zero for each successive call to TaoSolve()
1279025e9500SJason Sarich 
1280025e9500SJason Sarich    Level: intermediate
1281025e9500SJason Sarich 
1282db781477SPatrick Sanan .seealso: `TaoGetKSP()`
1283025e9500SJason Sarich @*/
1284025e9500SJason Sarich PetscErrorCode TaoGetLinearSolveIterations(Tao tao, PetscInt *lits)
1285025e9500SJason Sarich {
1286025e9500SJason Sarich   PetscFunctionBegin;
1287025e9500SJason Sarich   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
1288025e9500SJason Sarich   PetscValidIntPointer(lits,2);
12892d9aa51bSJason Sarich   *lits = tao->ksp_tot_its;
1290025e9500SJason Sarich   PetscFunctionReturn(0);
1291025e9500SJason Sarich }
1292025e9500SJason Sarich 
1293a7e14dcfSSatish Balay /*@
1294a7e14dcfSSatish Balay   TaoGetLineSearch - Gets the line search used by the optimization solver.
1295a7e14dcfSSatish Balay   Application writers should use TaoGetLineSearch if they need direct access
1296a7e14dcfSSatish Balay   to the TaoLineSearch object.
1297a7e14dcfSSatish Balay 
1298a7e14dcfSSatish Balay   Not Collective
1299a7e14dcfSSatish Balay 
1300a7e14dcfSSatish Balay    Input Parameters:
1301a7e14dcfSSatish Balay .  tao - the TAO solver
1302a7e14dcfSSatish Balay 
1303a7e14dcfSSatish Balay    Output Parameters:
1304a7e14dcfSSatish Balay .  ls - the line search used in the optimization solver
1305a7e14dcfSSatish Balay 
1306a7e14dcfSSatish Balay    Level: intermediate
1307a7e14dcfSSatish Balay 
1308a7e14dcfSSatish Balay @*/
1309441846f8SBarry Smith PetscErrorCode TaoGetLineSearch(Tao tao, TaoLineSearch *ls)
131047a47007SBarry Smith {
1311a7e14dcfSSatish Balay   PetscFunctionBegin;
131294511df7SStefano Zampini   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
131394511df7SStefano Zampini   PetscValidPointer(ls,2);
1314a7e14dcfSSatish Balay   *ls = tao->linesearch;
1315a7e14dcfSSatish Balay   PetscFunctionReturn(0);
1316a7e14dcfSSatish Balay }
1317a7e14dcfSSatish Balay 
1318a7e14dcfSSatish Balay /*@
1319a7e14dcfSSatish Balay   TaoAddLineSearchCounts - Adds the number of function evaluations spent
1320a7e14dcfSSatish Balay   in the line search to the running total.
1321a7e14dcfSSatish Balay 
1322a7e14dcfSSatish Balay    Input Parameters:
1323a7e14dcfSSatish Balay +  tao - the TAO solver
1324a7e14dcfSSatish Balay -  ls - the line search used in the optimization solver
1325a7e14dcfSSatish Balay 
1326a7e14dcfSSatish Balay    Level: developer
1327a7e14dcfSSatish Balay 
1328db781477SPatrick Sanan .seealso: `TaoLineSearchApply()`
1329a7e14dcfSSatish Balay @*/
1330441846f8SBarry Smith PetscErrorCode TaoAddLineSearchCounts(Tao tao)
133147a47007SBarry Smith {
1332a7e14dcfSSatish Balay   PetscBool      flg;
1333a7e14dcfSSatish Balay   PetscInt       nfeval,ngeval,nfgeval;
133447a47007SBarry Smith 
1335a7e14dcfSSatish Balay   PetscFunctionBegin;
1336441846f8SBarry Smith   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
1337a7e14dcfSSatish Balay   if (tao->linesearch) {
13389566063dSJacob Faibussowitsch     PetscCall(TaoLineSearchIsUsingTaoRoutines(tao->linesearch,&flg));
133994ae4db5SBarry Smith     if (!flg) {
13409566063dSJacob Faibussowitsch       PetscCall(TaoLineSearchGetNumberFunctionEvaluations(tao->linesearch,&nfeval,&ngeval,&nfgeval));
1341a7e14dcfSSatish Balay       tao->nfuncs += nfeval;
1342a7e14dcfSSatish Balay       tao->ngrads += ngeval;
1343a7e14dcfSSatish Balay       tao->nfuncgrads += nfgeval;
1344a7e14dcfSSatish Balay     }
1345a7e14dcfSSatish Balay   }
1346a7e14dcfSSatish Balay   PetscFunctionReturn(0);
1347a7e14dcfSSatish Balay }
1348a7e14dcfSSatish Balay 
1349a7e14dcfSSatish Balay /*@
1350a82e8c82SStefano Zampini   TaoGetSolution - Returns the vector with the current TAO solution
1351a7e14dcfSSatish Balay 
1352a7e14dcfSSatish Balay   Not Collective
1353a7e14dcfSSatish Balay 
1354a7e14dcfSSatish Balay   Input Parameter:
1355441846f8SBarry Smith . tao - the Tao context
1356a7e14dcfSSatish Balay 
1357a7e14dcfSSatish Balay   Output Parameter:
1358a7e14dcfSSatish Balay . X - the current solution
1359a7e14dcfSSatish Balay 
1360a7e14dcfSSatish Balay   Level: intermediate
1361a7e14dcfSSatish Balay 
1362a82e8c82SStefano Zampini   Note:  The returned vector will be the same object that was passed into TaoSetSolution()
1363a7e14dcfSSatish Balay @*/
1364a82e8c82SStefano Zampini PetscErrorCode TaoGetSolution(Tao tao, Vec *X)
1365a7e14dcfSSatish Balay {
1366a7e14dcfSSatish Balay   PetscFunctionBegin;
1367441846f8SBarry Smith   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
136894511df7SStefano Zampini   PetscValidPointer(X,2);
1369a7e14dcfSSatish Balay   *X = tao->solution;
1370a7e14dcfSSatish Balay   PetscFunctionReturn(0);
1371a7e14dcfSSatish Balay }
1372a7e14dcfSSatish Balay 
1373a7e14dcfSSatish Balay /*@
1374a7e14dcfSSatish Balay    TaoResetStatistics - Initialize the statistics used by TAO for all of the solvers.
1375a7e14dcfSSatish Balay    These statistics include the iteration number, residual norms, and convergence status.
1376a7e14dcfSSatish Balay    This routine gets called before solving each optimization problem.
1377a7e14dcfSSatish Balay 
1378441846f8SBarry Smith    Collective on Tao
1379a7e14dcfSSatish Balay 
1380a7e14dcfSSatish Balay    Input Parameters:
1381441846f8SBarry Smith .  solver - the Tao context
1382a7e14dcfSSatish Balay 
1383a7e14dcfSSatish Balay    Level: developer
1384a7e14dcfSSatish Balay 
1385db781477SPatrick Sanan .seealso: `TaoCreate()`, `TaoSolve()`
1386a7e14dcfSSatish Balay @*/
1387441846f8SBarry Smith PetscErrorCode TaoResetStatistics(Tao tao)
1388a7e14dcfSSatish Balay {
1389a7e14dcfSSatish Balay   PetscFunctionBegin;
1390441846f8SBarry Smith   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
1391a7e14dcfSSatish Balay   tao->niter        = 0;
1392a7e14dcfSSatish Balay   tao->nfuncs       = 0;
1393a7e14dcfSSatish Balay   tao->nfuncgrads   = 0;
1394a7e14dcfSSatish Balay   tao->ngrads       = 0;
1395a7e14dcfSSatish Balay   tao->nhess        = 0;
1396a7e14dcfSSatish Balay   tao->njac         = 0;
1397a7e14dcfSSatish Balay   tao->nconstraints = 0;
1398a7e14dcfSSatish Balay   tao->ksp_its      = 0;
1399ae93cb3cSJason Sarich   tao->ksp_tot_its  = 0;
1400a7e14dcfSSatish Balay   tao->reason       = TAO_CONTINUE_ITERATING;
1401a7e14dcfSSatish Balay   tao->residual     = 0.0;
1402a7e14dcfSSatish Balay   tao->cnorm        = 0.0;
1403a7e14dcfSSatish Balay   tao->step         = 0.0;
1404a7e14dcfSSatish Balay   tao->lsflag       = PETSC_FALSE;
1405a7e14dcfSSatish Balay   if (tao->hist_reset) tao->hist_len = 0;
1406a7e14dcfSSatish Balay   PetscFunctionReturn(0);
1407a7e14dcfSSatish Balay }
1408a7e14dcfSSatish Balay 
1409a7e14dcfSSatish Balay /*@C
1410e1e80dc8SAlp Dener   TaoSetUpdate - Sets the general-purpose update function called
1411e1e80dc8SAlp Dener   at the beginning of every iteration of the nonlinear solve. Specifically
1412e1e80dc8SAlp Dener   it is called at the top of every iteration, after the new solution and the gradient
1413e1e80dc8SAlp Dener   is determined, but before the Hessian is computed (if applicable).
1414e1e80dc8SAlp Dener 
1415e1e80dc8SAlp Dener   Logically Collective on Tao
1416e1e80dc8SAlp Dener 
1417e1e80dc8SAlp Dener   Input Parameters:
1418a2b725a8SWilliam Gropp + tao - The tao solver context
1419a2b725a8SWilliam Gropp - func - The function
1420e1e80dc8SAlp Dener 
1421e1e80dc8SAlp Dener   Calling sequence of func:
1422a2b725a8SWilliam Gropp $ func (Tao tao, PetscInt step);
1423e1e80dc8SAlp Dener 
1424e1e80dc8SAlp Dener . step - The current step of the iteration
1425e1e80dc8SAlp Dener 
1426e1e80dc8SAlp Dener   Level: advanced
1427e1e80dc8SAlp Dener 
1428db781477SPatrick Sanan .seealso `TaoSolve()`
1429e1e80dc8SAlp Dener @*/
14308fcddce6SStefano Zampini PetscErrorCode TaoSetUpdate(Tao tao, PetscErrorCode (*func)(Tao, PetscInt, void*), void *ctx)
1431e1e80dc8SAlp Dener {
1432e1e80dc8SAlp Dener   PetscFunctionBegin;
1433e1e80dc8SAlp Dener   PetscValidHeaderSpecific(tao, TAO_CLASSID,1);
1434e1e80dc8SAlp Dener   tao->ops->update = func;
1435e1e80dc8SAlp Dener   tao->user_update = ctx;
1436e1e80dc8SAlp Dener   PetscFunctionReturn(0);
1437e1e80dc8SAlp Dener }
1438e1e80dc8SAlp Dener 
1439e1e80dc8SAlp Dener /*@C
1440a7e14dcfSSatish Balay   TaoSetConvergenceTest - Sets the function that is to be used to test
1441a7e14dcfSSatish Balay   for convergence o fthe iterative minimization solution.  The new convergence
1442a7e14dcfSSatish Balay   testing routine will replace TAO's default convergence test.
1443a7e14dcfSSatish Balay 
1444441846f8SBarry Smith   Logically Collective on Tao
1445a7e14dcfSSatish Balay 
1446a7e14dcfSSatish Balay   Input Parameters:
1447441846f8SBarry Smith + tao - the Tao object
1448a7e14dcfSSatish Balay . conv - the routine to test for convergence
1449a7e14dcfSSatish Balay - ctx - [optional] context for private data for the convergence routine
145047a47007SBarry Smith         (may be NULL)
1451a7e14dcfSSatish Balay 
1452a7e14dcfSSatish Balay   Calling sequence of conv:
1453441846f8SBarry Smith $   PetscErrorCode conv(Tao tao, void *ctx)
1454a7e14dcfSSatish Balay 
1455441846f8SBarry Smith + tao - the Tao object
1456a7e14dcfSSatish Balay - ctx - [optional] convergence context
1457a7e14dcfSSatish Balay 
1458e4cb33bbSBarry Smith   Note: The new convergence testing routine should call TaoSetConvergedReason().
1459a7e14dcfSSatish Balay 
1460a7e14dcfSSatish Balay   Level: advanced
1461a7e14dcfSSatish Balay 
1462db781477SPatrick Sanan .seealso: `TaoSetConvergedReason()`, `TaoGetSolutionStatus()`, `TaoGetTolerances()`, `TaoSetMonitor`
1463a7e14dcfSSatish Balay 
1464a7e14dcfSSatish Balay @*/
1465441846f8SBarry Smith PetscErrorCode TaoSetConvergenceTest(Tao tao, PetscErrorCode (*conv)(Tao, void*), void *ctx)
1466a7e14dcfSSatish Balay {
1467a7e14dcfSSatish Balay   PetscFunctionBegin;
1468441846f8SBarry Smith   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
146994511df7SStefano Zampini   tao->ops->convergencetest = conv;
147094511df7SStefano Zampini   tao->cnvP = ctx;
1471a7e14dcfSSatish Balay   PetscFunctionReturn(0);
1472a7e14dcfSSatish Balay }
1473a7e14dcfSSatish Balay 
1474a7e14dcfSSatish Balay /*@C
1475a7e14dcfSSatish Balay    TaoSetMonitor - Sets an ADDITIONAL function that is to be used at every
1476a7e14dcfSSatish Balay    iteration of the solver to display the iteration's
1477a7e14dcfSSatish Balay    progress.
1478a7e14dcfSSatish Balay 
1479441846f8SBarry Smith    Logically Collective on Tao
1480a7e14dcfSSatish Balay 
1481a7e14dcfSSatish Balay    Input Parameters:
1482441846f8SBarry Smith +  tao - the Tao solver context
1483a7e14dcfSSatish Balay .  mymonitor - monitoring routine
1484a7e14dcfSSatish Balay -  mctx - [optional] user-defined context for private data for the
148547a47007SBarry Smith           monitor routine (may be NULL)
1486a7e14dcfSSatish Balay 
1487a7e14dcfSSatish Balay    Calling sequence of mymonitor:
1488147403d9SBarry Smith .vb
1489147403d9SBarry Smith      PetscErrorCode mymonitor(Tao tao,void *mctx)
1490147403d9SBarry Smith .ve
1491a7e14dcfSSatish Balay 
1492441846f8SBarry Smith +    tao - the Tao solver context
1493a7e14dcfSSatish Balay -    mctx - [optional] monitoring context
1494a7e14dcfSSatish Balay 
1495a7e14dcfSSatish Balay    Options Database Keys:
149698ea980cSBarry Smith +    -tao_monitor        - sets TaoMonitorDefault()
1497a7e14dcfSSatish Balay .    -tao_smonitor       - sets short monitor
1498a7e14dcfSSatish Balay .    -tao_cmonitor       - same as smonitor plus constraint norm
1499a7e14dcfSSatish Balay .    -tao_view_solution   - view solution at each iteration
1500a7e14dcfSSatish Balay .    -tao_view_gradient   - view gradient at each iteration
15014a48860cSAlp Dener .    -tao_view_ls_residual - view least-squares residual vector at each iteration
1502a7e14dcfSSatish 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.
1503a7e14dcfSSatish Balay 
1504a7e14dcfSSatish Balay    Notes:
1505a7e14dcfSSatish Balay    Several different monitoring routines may be set by calling
1506a7e14dcfSSatish Balay    TaoSetMonitor() multiple times; all will be called in the
1507a7e14dcfSSatish Balay    order in which they were set.
1508a7e14dcfSSatish Balay 
150995452b02SPatrick Sanan    Fortran Notes:
151095452b02SPatrick Sanan     Only one monitor function may be set
1511a7e14dcfSSatish Balay 
1512a7e14dcfSSatish Balay    Level: intermediate
1513a7e14dcfSSatish Balay 
1514db781477SPatrick Sanan .seealso: `TaoMonitorDefault()`, `TaoCancelMonitors()`, `TaoSetDestroyRoutine()`
1515a7e14dcfSSatish Balay @*/
1516441846f8SBarry Smith PetscErrorCode TaoSetMonitor(Tao tao, PetscErrorCode (*func)(Tao, void*), void *ctx,PetscErrorCode (*dest)(void**))
1517a7e14dcfSSatish Balay {
151808d19d1fSJason Sarich   PetscInt       i;
15198732526dSAlp Dener   PetscBool      identical;
152008d19d1fSJason Sarich 
1521a7e14dcfSSatish Balay   PetscFunctionBegin;
1522441846f8SBarry Smith   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
15233c859ba3SBarry Smith   PetscCheck(tao->numbermonitors < MAXTAOMONITORS,PetscObjectComm((PetscObject)tao),PETSC_ERR_SUP,"Cannot attach another monitor -- max=%d",MAXTAOMONITORS);
152408d19d1fSJason Sarich 
152508d19d1fSJason Sarich   for (i=0; i<tao->numbermonitors;i++) {
15269566063dSJacob Faibussowitsch     PetscCall(PetscMonitorCompare((PetscErrorCode (*)(void))func,ctx,dest,(PetscErrorCode (*)(void))tao->monitor[i],tao->monitorcontext[i],tao->monitordestroy[i],&identical));
15278732526dSAlp Dener     if (identical) PetscFunctionReturn(0);
152808d19d1fSJason Sarich   }
1529a7e14dcfSSatish Balay   tao->monitor[tao->numbermonitors] = func;
15308732526dSAlp Dener   tao->monitorcontext[tao->numbermonitors] = (void*)ctx;
1531a7e14dcfSSatish Balay   tao->monitordestroy[tao->numbermonitors] = dest;
1532a7e14dcfSSatish Balay   ++tao->numbermonitors;
1533a7e14dcfSSatish Balay   PetscFunctionReturn(0);
1534a7e14dcfSSatish Balay }
1535a7e14dcfSSatish Balay 
1536a7e14dcfSSatish Balay /*@
1537441846f8SBarry Smith    TaoCancelMonitors - Clears all the monitor functions for a Tao object.
1538a7e14dcfSSatish Balay 
1539441846f8SBarry Smith    Logically Collective on Tao
1540a7e14dcfSSatish Balay 
1541a7e14dcfSSatish Balay    Input Parameters:
1542441846f8SBarry Smith .  tao - the Tao solver context
1543a7e14dcfSSatish Balay 
1544a7e14dcfSSatish Balay    Options Database:
1545a7e14dcfSSatish Balay .  -tao_cancelmonitors - cancels all monitors that have been hardwired
1546a7e14dcfSSatish Balay     into a code by calls to TaoSetMonitor(), but does not cancel those
1547a7e14dcfSSatish Balay     set via the options database
1548a7e14dcfSSatish Balay 
1549a7e14dcfSSatish Balay    Notes:
1550441846f8SBarry Smith    There is no way to clear one specific monitor from a Tao object.
1551a7e14dcfSSatish Balay 
1552a7e14dcfSSatish Balay    Level: advanced
1553a7e14dcfSSatish Balay 
1554db781477SPatrick Sanan .seealso: `TaoMonitorDefault()`, `TaoSetMonitor()`
1555a7e14dcfSSatish Balay @*/
1556441846f8SBarry Smith PetscErrorCode TaoCancelMonitors(Tao tao)
1557a7e14dcfSSatish Balay {
1558a7e14dcfSSatish Balay   PetscInt       i;
155947a47007SBarry Smith 
1560a7e14dcfSSatish Balay   PetscFunctionBegin;
1561441846f8SBarry Smith   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
1562a7e14dcfSSatish Balay   for (i=0;i<tao->numbermonitors;i++) {
1563a7e14dcfSSatish Balay     if (tao->monitordestroy[i]) {
15649566063dSJacob Faibussowitsch       PetscCall((*tao->monitordestroy[i])(&tao->monitorcontext[i]));
1565a7e14dcfSSatish Balay     }
1566a7e14dcfSSatish Balay   }
1567a7e14dcfSSatish Balay   tao->numbermonitors = 0;
1568a7e14dcfSSatish Balay   PetscFunctionReturn(0);
1569a7e14dcfSSatish Balay }
1570a7e14dcfSSatish Balay 
15717fab98ebSJason Sarich /*@
157298ea980cSBarry Smith    TaoMonitorDefault - Default routine for monitoring progress of the
1573441846f8SBarry Smith    Tao solvers (default).  This monitor prints the function value and gradient
1574a7e14dcfSSatish Balay    norm at each iteration.  It can be turned on from the command line using the
1575a7e14dcfSSatish Balay    -tao_monitor option
1576a7e14dcfSSatish Balay 
1577441846f8SBarry Smith    Collective on Tao
1578a7e14dcfSSatish Balay 
1579a7e14dcfSSatish Balay    Input Parameters:
1580441846f8SBarry Smith +  tao - the Tao context
158147a47007SBarry Smith -  ctx - PetscViewer context or NULL
1582a7e14dcfSSatish Balay 
1583a7e14dcfSSatish Balay    Options Database Keys:
1584147403d9SBarry Smith .  -tao_monitor - turn on default monitoring
1585a7e14dcfSSatish Balay 
1586a7e14dcfSSatish Balay    Level: advanced
1587a7e14dcfSSatish Balay 
1588db781477SPatrick Sanan .seealso: `TaoDefaultSMonitor()`, `TaoSetMonitor()`
1589a7e14dcfSSatish Balay @*/
159098ea980cSBarry Smith PetscErrorCode TaoMonitorDefault(Tao tao, void *ctx)
1591a7e14dcfSSatish Balay {
159263b15415SAlp Dener   PetscInt       its, tabs;
1593a7e14dcfSSatish Balay   PetscReal      fct,gnorm;
15948163d661SBarry Smith   PetscViewer    viewer = (PetscViewer)ctx;
159547a47007SBarry Smith 
1596a7e14dcfSSatish Balay   PetscFunctionBegin;
159794511df7SStefano Zampini   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
15988163d661SBarry Smith   PetscValidHeaderSpecific(viewer,PETSC_VIEWER_CLASSID,2);
1599a7e14dcfSSatish Balay   its = tao->niter;
1600a7e14dcfSSatish Balay   fct = tao->fc;
1601a7e14dcfSSatish Balay   gnorm = tao->residual;
16029566063dSJacob Faibussowitsch   PetscCall(PetscViewerASCIIGetTab(viewer, &tabs));
16039566063dSJacob Faibussowitsch   PetscCall(PetscViewerASCIISetTab(viewer, ((PetscObject)tao)->tablevel));
1604494bef23SAlp Dener   if (its == 0 && ((PetscObject)tao)->prefix && !tao->header_printed) {
16059566063dSJacob Faibussowitsch      PetscCall(PetscViewerASCIIPrintf(viewer,"  Iteration information for %s solve.\n",((PetscObject)tao)->prefix));
1606494bef23SAlp Dener      tao->header_printed = PETSC_TRUE;
160763b15415SAlp Dener    }
160863a3b9bcSJacob Faibussowitsch   PetscCall(PetscViewerASCIIPrintf(viewer,"%3" PetscInt_FMT " TAO,",its));
16099566063dSJacob Faibussowitsch   PetscCall(PetscViewerASCIIPrintf(viewer,"  Function value: %g,",(double)fct));
161008d19d1fSJason Sarich   if (gnorm >= PETSC_INFINITY) {
16119566063dSJacob Faibussowitsch     PetscCall(PetscViewerASCIIPrintf(viewer,"  Residual: Inf \n"));
161208d19d1fSJason Sarich   } else {
16139566063dSJacob Faibussowitsch     PetscCall(PetscViewerASCIIPrintf(viewer,"  Residual: %g \n",(double)gnorm));
161408d19d1fSJason Sarich   }
16159566063dSJacob Faibussowitsch   PetscCall(PetscViewerASCIISetTab(viewer, tabs));
1616a7e14dcfSSatish Balay   PetscFunctionReturn(0);
1617a7e14dcfSSatish Balay }
1618a7e14dcfSSatish Balay 
16197fab98ebSJason Sarich /*@
16208d5ead36SAlp Dener    TaoDefaultGMonitor - Default routine for monitoring progress of the
16218d5ead36SAlp Dener    Tao solvers (default) with extra detail on the globalization method.
16228d5ead36SAlp Dener    This monitor prints the function value and gradient norm at each
16238d5ead36SAlp Dener    iteration, as well as the step size and trust radius. Note that the
16248d5ead36SAlp Dener    step size and trust radius may be the same for some algorithms.
16258d5ead36SAlp Dener    It can be turned on from the command line using the
16268d5ead36SAlp Dener    -tao_gmonitor option
16278d5ead36SAlp Dener 
16288d5ead36SAlp Dener    Collective on Tao
16298d5ead36SAlp Dener 
16308d5ead36SAlp Dener    Input Parameters:
16318d5ead36SAlp Dener +  tao - the Tao context
16328d5ead36SAlp Dener -  ctx - PetscViewer context or NULL
16338d5ead36SAlp Dener 
16348d5ead36SAlp Dener    Options Database Keys:
1635147403d9SBarry Smith .  -tao_gmonitor - turn on monitoring with globalization information
16368d5ead36SAlp Dener 
16378d5ead36SAlp Dener    Level: advanced
16388d5ead36SAlp Dener 
1639db781477SPatrick Sanan .seealso: `TaoDefaultSMonitor()`, `TaoSetMonitor()`
16408d5ead36SAlp Dener @*/
16418d5ead36SAlp Dener PetscErrorCode TaoDefaultGMonitor(Tao tao, void *ctx)
16428d5ead36SAlp Dener {
16438d5ead36SAlp Dener   PetscInt       its, tabs;
16448d5ead36SAlp Dener   PetscReal      fct,gnorm,stp,tr;
16458d5ead36SAlp Dener   PetscViewer    viewer = (PetscViewer)ctx;
16468d5ead36SAlp Dener 
16478d5ead36SAlp Dener   PetscFunctionBegin;
164894511df7SStefano Zampini   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
16498d5ead36SAlp Dener   PetscValidHeaderSpecific(viewer,PETSC_VIEWER_CLASSID,2);
16508d5ead36SAlp Dener   its = tao->niter;
16518d5ead36SAlp Dener   fct = tao->fc;
16528d5ead36SAlp Dener   gnorm = tao->residual;
16538d5ead36SAlp Dener   stp = tao->step;
16548d5ead36SAlp Dener   tr = tao->trust;
16559566063dSJacob Faibussowitsch   PetscCall(PetscViewerASCIIGetTab(viewer, &tabs));
16569566063dSJacob Faibussowitsch   PetscCall(PetscViewerASCIISetTab(viewer, ((PetscObject)tao)->tablevel));
1657494bef23SAlp Dener   if (its == 0 && ((PetscObject)tao)->prefix && !tao->header_printed) {
16589566063dSJacob Faibussowitsch      PetscCall(PetscViewerASCIIPrintf(viewer,"  Iteration information for %s solve.\n",((PetscObject)tao)->prefix));
1659494bef23SAlp Dener      tao->header_printed = PETSC_TRUE;
16608d5ead36SAlp Dener    }
166163a3b9bcSJacob Faibussowitsch   PetscCall(PetscViewerASCIIPrintf(viewer,"%3" PetscInt_FMT " TAO,",its));
16629566063dSJacob Faibussowitsch   PetscCall(PetscViewerASCIIPrintf(viewer,"  Function value: %g,",(double)fct));
16638d5ead36SAlp Dener   if (gnorm >= PETSC_INFINITY) {
16649566063dSJacob Faibussowitsch     PetscCall(PetscViewerASCIIPrintf(viewer,"  Residual: Inf,"));
16658d5ead36SAlp Dener   } else {
16669566063dSJacob Faibussowitsch     PetscCall(PetscViewerASCIIPrintf(viewer,"  Residual: %g,",(double)gnorm));
16678d5ead36SAlp Dener   }
16689566063dSJacob Faibussowitsch   PetscCall(PetscViewerASCIIPrintf(viewer,"  Step: %g,  Trust: %g\n",(double)stp,(double)tr));
16699566063dSJacob Faibussowitsch   PetscCall(PetscViewerASCIISetTab(viewer, tabs));
16708d5ead36SAlp Dener   PetscFunctionReturn(0);
16718d5ead36SAlp Dener }
16728d5ead36SAlp Dener 
16738d5ead36SAlp Dener /*@
1674a7e14dcfSSatish Balay    TaoDefaultSMonitor - Default routine for monitoring progress of the
167598ea980cSBarry Smith    solver. Same as TaoMonitorDefault() except
1676a7e14dcfSSatish Balay    it prints fewer digits of the residual as the residual gets smaller.
1677a7e14dcfSSatish Balay    This is because the later digits are meaningless and are often
1678a7e14dcfSSatish Balay    different on different machines; by using this routine different
1679a7e14dcfSSatish Balay    machines will usually generate the same output. It can be turned on
1680a7e14dcfSSatish Balay    by using the -tao_smonitor option
1681a7e14dcfSSatish Balay 
1682441846f8SBarry Smith    Collective on Tao
1683a7e14dcfSSatish Balay 
1684a7e14dcfSSatish Balay    Input Parameters:
1685441846f8SBarry Smith +  tao - the Tao context
16864d4332d5SBarry Smith -  ctx - PetscViewer context of type ASCII
1687a7e14dcfSSatish Balay 
1688a7e14dcfSSatish Balay    Options Database Keys:
1689147403d9SBarry Smith .  -tao_smonitor - turn on default short monitoring
1690a7e14dcfSSatish Balay 
1691a7e14dcfSSatish Balay    Level: advanced
1692a7e14dcfSSatish Balay 
1693db781477SPatrick Sanan .seealso: `TaoMonitorDefault()`, `TaoSetMonitor()`
1694a7e14dcfSSatish Balay @*/
1695441846f8SBarry Smith PetscErrorCode TaoDefaultSMonitor(Tao tao, void *ctx)
1696a7e14dcfSSatish Balay {
16972c9b8b83SAlp Dener   PetscInt       its, tabs;
1698a7e14dcfSSatish Balay   PetscReal      fct,gnorm;
16994d4332d5SBarry Smith   PetscViewer    viewer = (PetscViewer)ctx;
170047a47007SBarry Smith 
1701a7e14dcfSSatish Balay   PetscFunctionBegin;
170294511df7SStefano Zampini   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
17034d4332d5SBarry Smith   PetscValidHeaderSpecific(viewer,PETSC_VIEWER_CLASSID,2);
1704a7e14dcfSSatish Balay   its = tao->niter;
1705a7e14dcfSSatish Balay   fct = tao->fc;
1706a7e14dcfSSatish Balay   gnorm = tao->residual;
17079566063dSJacob Faibussowitsch   PetscCall(PetscViewerASCIIGetTab(viewer, &tabs));
17089566063dSJacob Faibussowitsch   PetscCall(PetscViewerASCIISetTab(viewer, ((PetscObject)tao)->tablevel));
170963a3b9bcSJacob Faibussowitsch   PetscCall(PetscViewerASCIIPrintf(viewer,"iter = %3" PetscInt_FMT ",",its));
17109566063dSJacob Faibussowitsch   PetscCall(PetscViewerASCIIPrintf(viewer," Function value %g,",(double)fct));
1711d393f493SBarry Smith   if (gnorm >= PETSC_INFINITY) {
17129566063dSJacob Faibussowitsch     PetscCall(PetscViewerASCIIPrintf(viewer," Residual: Inf \n"));
171308d19d1fSJason Sarich   } else if (gnorm > 1.e-6) {
17149566063dSJacob Faibussowitsch     PetscCall(PetscViewerASCIIPrintf(viewer," Residual: %g \n",(double)gnorm));
1715a7e14dcfSSatish Balay   } else if (gnorm > 1.e-11) {
17169566063dSJacob Faibussowitsch     PetscCall(PetscViewerASCIIPrintf(viewer," Residual: < 1.0e-6 \n"));
1717a7e14dcfSSatish Balay   } else {
17189566063dSJacob Faibussowitsch     PetscCall(PetscViewerASCIIPrintf(viewer," Residual: < 1.0e-11 \n"));
1719a7e14dcfSSatish Balay   }
17209566063dSJacob Faibussowitsch   PetscCall(PetscViewerASCIISetTab(viewer, tabs));
1721a7e14dcfSSatish Balay   PetscFunctionReturn(0);
1722a7e14dcfSSatish Balay }
1723a7e14dcfSSatish Balay 
17247fab98ebSJason Sarich /*@
172598ea980cSBarry Smith    TaoDefaultCMonitor - same as TaoMonitorDefault() except
1726a7e14dcfSSatish Balay    it prints the norm of the constraints function. It can be turned on
1727a7e14dcfSSatish Balay    from the command line using the -tao_cmonitor option
1728a7e14dcfSSatish Balay 
1729441846f8SBarry Smith    Collective on Tao
1730a7e14dcfSSatish Balay 
1731a7e14dcfSSatish Balay    Input Parameters:
1732441846f8SBarry Smith +  tao - the Tao context
173347a47007SBarry Smith -  ctx - PetscViewer context or NULL
1734a7e14dcfSSatish Balay 
1735a7e14dcfSSatish Balay    Options Database Keys:
1736147403d9SBarry Smith .  -tao_cmonitor - monitor the constraints
1737a7e14dcfSSatish Balay 
1738a7e14dcfSSatish Balay    Level: advanced
1739a7e14dcfSSatish Balay 
1740db781477SPatrick Sanan .seealso: `TaoMonitorDefault()`, `TaoSetMonitor()`
1741a7e14dcfSSatish Balay @*/
1742441846f8SBarry Smith PetscErrorCode TaoDefaultCMonitor(Tao tao, void *ctx)
1743a7e14dcfSSatish Balay {
17442c9b8b83SAlp Dener   PetscInt       its, tabs;
1745a7e14dcfSSatish Balay   PetscReal      fct,gnorm;
1746d393f493SBarry Smith   PetscViewer    viewer = (PetscViewer)ctx;
1747a7e14dcfSSatish Balay 
1748a7e14dcfSSatish Balay   PetscFunctionBegin;
174994511df7SStefano Zampini   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
1750d393f493SBarry Smith   PetscValidHeaderSpecific(viewer,PETSC_VIEWER_CLASSID,2);
1751a7e14dcfSSatish Balay   its = tao->niter;
1752a7e14dcfSSatish Balay   fct = tao->fc;
1753a7e14dcfSSatish Balay   gnorm = tao->residual;
17549566063dSJacob Faibussowitsch   PetscCall(PetscViewerASCIIGetTab(viewer, &tabs));
17559566063dSJacob Faibussowitsch   PetscCall(PetscViewerASCIISetTab(viewer, ((PetscObject)tao)->tablevel));
175663a3b9bcSJacob Faibussowitsch   PetscCall(PetscViewerASCIIPrintf(viewer,"iter = %" PetscInt_FMT ",",its));
17579566063dSJacob Faibussowitsch   PetscCall(PetscViewerASCIIPrintf(viewer," Function value: %g,",(double)fct));
17589566063dSJacob Faibussowitsch   PetscCall(PetscViewerASCIIPrintf(viewer,"  Residual: %g ",(double)gnorm));
17599566063dSJacob Faibussowitsch   PetscCall(PetscViewerASCIIPrintf(viewer,"  Constraint: %g \n",(double)tao->cnorm));
17609566063dSJacob Faibussowitsch   PetscCall(PetscViewerASCIISetTab(viewer, tabs));
1761a7e14dcfSSatish Balay   PetscFunctionReturn(0);
1762a7e14dcfSSatish Balay }
1763a7e14dcfSSatish Balay 
1764a7e14dcfSSatish Balay /*@C
1765a7e14dcfSSatish Balay    TaoSolutionMonitor - Views the solution at each iteration
1766a7e14dcfSSatish Balay    It can be turned on from the command line using the
1767a7e14dcfSSatish Balay    -tao_view_solution option
1768a7e14dcfSSatish Balay 
1769441846f8SBarry Smith    Collective on Tao
1770a7e14dcfSSatish Balay 
1771a7e14dcfSSatish Balay    Input Parameters:
1772441846f8SBarry Smith +  tao - the Tao context
177347a47007SBarry Smith -  ctx - PetscViewer context or NULL
1774a7e14dcfSSatish Balay 
1775a7e14dcfSSatish Balay    Options Database Keys:
1776147403d9SBarry Smith .  -tao_view_solution - view the solution
1777a7e14dcfSSatish Balay 
1778a7e14dcfSSatish Balay    Level: advanced
1779a7e14dcfSSatish Balay 
1780db781477SPatrick Sanan .seealso: `TaoDefaultSMonitor()`, `TaoSetMonitor()`
1781a7e14dcfSSatish Balay @*/
1782441846f8SBarry Smith PetscErrorCode TaoSolutionMonitor(Tao tao, void *ctx)
1783a7e14dcfSSatish Balay {
1784feb237baSPierre Jolivet   PetscViewer    viewer  = (PetscViewer)ctx;
178547a47007SBarry Smith 
1786a7e14dcfSSatish Balay   PetscFunctionBegin;
178794511df7SStefano Zampini   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
1788d393f493SBarry Smith   PetscValidHeaderSpecific(viewer,PETSC_VIEWER_CLASSID,2);
17899566063dSJacob Faibussowitsch   PetscCall(VecView(tao->solution,viewer));
1790a7e14dcfSSatish Balay   PetscFunctionReturn(0);
1791a7e14dcfSSatish Balay }
1792a7e14dcfSSatish Balay 
1793a7e14dcfSSatish Balay /*@C
1794a7e14dcfSSatish Balay    TaoGradientMonitor - Views the gradient at each iteration
1795a7e14dcfSSatish Balay    It can be turned on from the command line using the
1796a7e14dcfSSatish Balay    -tao_view_gradient option
1797a7e14dcfSSatish Balay 
1798441846f8SBarry Smith    Collective on Tao
1799a7e14dcfSSatish Balay 
1800a7e14dcfSSatish Balay    Input Parameters:
1801441846f8SBarry Smith +  tao - the Tao context
180247a47007SBarry Smith -  ctx - PetscViewer context or NULL
1803a7e14dcfSSatish Balay 
1804a7e14dcfSSatish Balay    Options Database Keys:
1805147403d9SBarry Smith .  -tao_view_gradient - view the gradient at each iteration
1806a7e14dcfSSatish Balay 
1807a7e14dcfSSatish Balay    Level: advanced
1808a7e14dcfSSatish Balay 
1809db781477SPatrick Sanan .seealso: `TaoDefaultSMonitor()`, `TaoSetMonitor()`
1810a7e14dcfSSatish Balay @*/
1811441846f8SBarry Smith PetscErrorCode TaoGradientMonitor(Tao tao, void *ctx)
1812a7e14dcfSSatish Balay {
1813d393f493SBarry Smith   PetscViewer    viewer = (PetscViewer)ctx;
181447a47007SBarry Smith 
1815a7e14dcfSSatish Balay   PetscFunctionBegin;
181694511df7SStefano Zampini   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
1817d393f493SBarry Smith   PetscValidHeaderSpecific(viewer,PETSC_VIEWER_CLASSID,2);
18189566063dSJacob Faibussowitsch   PetscCall(VecView(tao->gradient,viewer));
1819a7e14dcfSSatish Balay   PetscFunctionReturn(0);
1820a7e14dcfSSatish Balay }
1821a7e14dcfSSatish Balay 
1822a7e14dcfSSatish Balay /*@C
1823147403d9SBarry Smith    TaoStepDirectionMonitor - Views the step-direction at each iteration
1824a7e14dcfSSatish Balay 
1825441846f8SBarry Smith    Collective on Tao
1826a7e14dcfSSatish Balay 
1827a7e14dcfSSatish Balay    Input Parameters:
1828441846f8SBarry Smith +  tao - the Tao context
182947a47007SBarry Smith -  ctx - PetscViewer context or NULL
1830a7e14dcfSSatish Balay 
1831a7e14dcfSSatish Balay    Options Database Keys:
1832147403d9SBarry Smith .  -tao_view_gradient - view the gradient at each iteration
1833a7e14dcfSSatish Balay 
1834a7e14dcfSSatish Balay    Level: advanced
1835a7e14dcfSSatish Balay 
1836db781477SPatrick Sanan .seealso: `TaoDefaultSMonitor()`, `TaoSetMonitor()`
1837a7e14dcfSSatish Balay @*/
1838441846f8SBarry Smith PetscErrorCode TaoStepDirectionMonitor(Tao tao, void *ctx)
1839a7e14dcfSSatish Balay {
1840d393f493SBarry Smith   PetscViewer    viewer = (PetscViewer)ctx;
1841d393f493SBarry Smith 
1842a7e14dcfSSatish Balay   PetscFunctionBegin;
184394511df7SStefano Zampini   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
1844d393f493SBarry Smith   PetscValidHeaderSpecific(viewer,PETSC_VIEWER_CLASSID,2);
18459566063dSJacob Faibussowitsch   PetscCall(VecView(tao->stepdirection,viewer));
1846a7e14dcfSSatish Balay   PetscFunctionReturn(0);
1847a7e14dcfSSatish Balay }
1848a7e14dcfSSatish Balay 
1849a7e14dcfSSatish Balay /*@C
1850a7e14dcfSSatish Balay    TaoDrawSolutionMonitor - Plots the solution at each iteration
1851a7e14dcfSSatish Balay    It can be turned on from the command line using the
1852a7e14dcfSSatish Balay    -tao_draw_solution option
1853a7e14dcfSSatish Balay 
1854441846f8SBarry Smith    Collective on Tao
1855a7e14dcfSSatish Balay 
1856a7e14dcfSSatish Balay    Input Parameters:
1857441846f8SBarry Smith +  tao - the Tao context
1858e882e171SHong Zhang -  ctx - TaoMonitorDraw context
1859a7e14dcfSSatish Balay 
1860a7e14dcfSSatish Balay    Options Database Keys:
1861147403d9SBarry Smith .  -tao_draw_solution - draw the solution at each iteration
1862a7e14dcfSSatish Balay 
1863a7e14dcfSSatish Balay    Level: advanced
1864a7e14dcfSSatish Balay 
1865db781477SPatrick Sanan .seealso: `TaoSolutionMonitor()`, `TaoSetMonitor()`, `TaoDrawGradientMonitor`
1866a7e14dcfSSatish Balay @*/
1867441846f8SBarry Smith PetscErrorCode TaoDrawSolutionMonitor(Tao tao, void *ctx)
1868a7e14dcfSSatish Balay {
1869e882e171SHong Zhang   TaoMonitorDrawCtx ictx = (TaoMonitorDrawCtx)ctx;
187047a47007SBarry Smith 
1871a7e14dcfSSatish Balay   PetscFunctionBegin;
187294511df7SStefano Zampini   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
1873e882e171SHong Zhang   if (!(((ictx->howoften > 0) && (!(tao->niter % ictx->howoften))) || ((ictx->howoften == -1) && tao->reason))) PetscFunctionReturn(0);
18749566063dSJacob Faibussowitsch   PetscCall(VecView(tao->solution,ictx->viewer));
1875a7e14dcfSSatish Balay   PetscFunctionReturn(0);
1876a7e14dcfSSatish Balay }
1877a7e14dcfSSatish Balay 
1878a7e14dcfSSatish Balay /*@C
1879a7e14dcfSSatish Balay    TaoDrawGradientMonitor - Plots the gradient at each iteration
1880a7e14dcfSSatish Balay    It can be turned on from the command line using the
1881a7e14dcfSSatish Balay    -tao_draw_gradient option
1882a7e14dcfSSatish Balay 
1883441846f8SBarry Smith    Collective on Tao
1884a7e14dcfSSatish Balay 
1885a7e14dcfSSatish Balay    Input Parameters:
1886441846f8SBarry Smith +  tao - the Tao context
1887f55353a2SBarry Smith -  ctx - PetscViewer context
1888a7e14dcfSSatish Balay 
1889a7e14dcfSSatish Balay    Options Database Keys:
1890147403d9SBarry Smith .  -tao_draw_gradient - draw the gradient at each iteration
1891a7e14dcfSSatish Balay 
1892a7e14dcfSSatish Balay    Level: advanced
1893a7e14dcfSSatish Balay 
1894db781477SPatrick Sanan .seealso: `TaoGradientMonitor()`, `TaoSetMonitor()`, `TaoDrawSolutionMonitor`
1895a7e14dcfSSatish Balay @*/
1896441846f8SBarry Smith PetscErrorCode TaoDrawGradientMonitor(Tao tao, void *ctx)
1897a7e14dcfSSatish Balay {
1898e882e171SHong Zhang   TaoMonitorDrawCtx ictx = (TaoMonitorDrawCtx)ctx;
189947a47007SBarry Smith 
1900a7e14dcfSSatish Balay   PetscFunctionBegin;
190194511df7SStefano Zampini   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
1902e882e171SHong Zhang   if (!(((ictx->howoften > 0) && (!(tao->niter % ictx->howoften))) || ((ictx->howoften == -1) && tao->reason))) PetscFunctionReturn(0);
19039566063dSJacob Faibussowitsch   PetscCall(VecView(tao->gradient,ictx->viewer));
1904a7e14dcfSSatish Balay   PetscFunctionReturn(0);
1905a7e14dcfSSatish Balay }
1906a7e14dcfSSatish Balay 
1907a7e14dcfSSatish Balay /*@C
1908a7e14dcfSSatish Balay    TaoDrawStepMonitor - Plots the step direction at each iteration
1909a7e14dcfSSatish Balay 
1910441846f8SBarry Smith    Collective on Tao
1911a7e14dcfSSatish Balay 
1912a7e14dcfSSatish Balay    Input Parameters:
1913441846f8SBarry Smith +  tao - the Tao context
1914f55353a2SBarry Smith -  ctx - PetscViewer context
1915a7e14dcfSSatish Balay 
1916a7e14dcfSSatish Balay    Options Database Keys:
1917147403d9SBarry Smith .  -tao_draw_step - draw the step direction at each iteration
1918a7e14dcfSSatish Balay 
1919a7e14dcfSSatish Balay    Level: advanced
1920a7e14dcfSSatish Balay 
1921db781477SPatrick Sanan .seealso: `TaoSetMonitor()`, `TaoDrawSolutionMonitor`
1922a7e14dcfSSatish Balay @*/
1923441846f8SBarry Smith PetscErrorCode TaoDrawStepMonitor(Tao tao, void *ctx)
1924a7e14dcfSSatish Balay {
192594511df7SStefano Zampini   PetscViewer    viewer = (PetscViewer)ctx;
192647a47007SBarry Smith 
1927a7e14dcfSSatish Balay   PetscFunctionBegin;
192894511df7SStefano Zampini   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
192994511df7SStefano Zampini   PetscValidHeaderSpecific(viewer,PETSC_VIEWER_CLASSID,2);
19309566063dSJacob Faibussowitsch   PetscCall(VecView(tao->stepdirection,viewer));
1931a7e14dcfSSatish Balay   PetscFunctionReturn(0);
1932a7e14dcfSSatish Balay }
1933a7e14dcfSSatish Balay 
1934a7e14dcfSSatish Balay /*@C
1935737f463aSAlp Dener    TaoResidualMonitor - Views the least-squares residual at each iteration
1936a7e14dcfSSatish Balay 
1937441846f8SBarry Smith    Collective on Tao
1938a7e14dcfSSatish Balay 
1939a7e14dcfSSatish Balay    Input Parameters:
1940441846f8SBarry Smith +  tao - the Tao context
194147a47007SBarry Smith -  ctx - PetscViewer context or NULL
1942a7e14dcfSSatish Balay 
1943a7e14dcfSSatish Balay    Options Database Keys:
1944147403d9SBarry Smith .  -tao_view_ls_residual - view the least-squares residual at each iteration
1945a7e14dcfSSatish Balay 
1946a7e14dcfSSatish Balay    Level: advanced
1947a7e14dcfSSatish Balay 
1948db781477SPatrick Sanan .seealso: `TaoDefaultSMonitor()`, `TaoSetMonitor()`
1949a7e14dcfSSatish Balay @*/
1950737f463aSAlp Dener PetscErrorCode TaoResidualMonitor(Tao tao, void *ctx)
1951a7e14dcfSSatish Balay {
1952d393f493SBarry Smith   PetscViewer    viewer  = (PetscViewer)ctx;
195347a47007SBarry Smith 
1954a7e14dcfSSatish Balay   PetscFunctionBegin;
195594511df7SStefano Zampini   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
1956d393f493SBarry Smith   PetscValidHeaderSpecific(viewer,PETSC_VIEWER_CLASSID,2);
19579566063dSJacob Faibussowitsch   PetscCall(VecView(tao->ls_res,viewer));
1958a7e14dcfSSatish Balay   PetscFunctionReturn(0);
1959a7e14dcfSSatish Balay }
1960a7e14dcfSSatish Balay 
19617fab98ebSJason Sarich /*@
1962a7e14dcfSSatish Balay    TaoDefaultConvergenceTest - Determines whether the solver should continue iterating
1963a7e14dcfSSatish Balay    or terminate.
1964a7e14dcfSSatish Balay 
1965441846f8SBarry Smith    Collective on Tao
1966a7e14dcfSSatish Balay 
1967a7e14dcfSSatish Balay    Input Parameters:
1968441846f8SBarry Smith +  tao - the Tao context
1969a7e14dcfSSatish Balay -  dummy - unused dummy context
1970a7e14dcfSSatish Balay 
1971a7e14dcfSSatish Balay    Output Parameter:
1972a7e14dcfSSatish Balay .  reason - for terminating
1973a7e14dcfSSatish Balay 
1974a7e14dcfSSatish Balay    Notes:
1975a7e14dcfSSatish Balay    This routine checks the residual in the optimality conditions, the
1976a7e14dcfSSatish Balay    relative residual in the optimity conditions, the number of function
1977a7e14dcfSSatish Balay    evaluations, and the function value to test convergence.  Some
1978a7e14dcfSSatish Balay    solvers may use different convergence routines.
1979a7e14dcfSSatish Balay 
1980a7e14dcfSSatish Balay    Level: developer
1981a7e14dcfSSatish Balay 
1982c2e3fba1SPatrick Sanan .seealso: `TaoSetTolerances()`, `TaoGetConvergedReason()`, `TaoSetConvergedReason()`
1983a7e14dcfSSatish Balay @*/
1984a7e14dcfSSatish Balay 
1985441846f8SBarry Smith PetscErrorCode TaoDefaultConvergenceTest(Tao tao,void *dummy)
1986a7e14dcfSSatish Balay {
1987a7e14dcfSSatish Balay   PetscInt           niter=tao->niter, nfuncs=PetscMax(tao->nfuncs,tao->nfuncgrads);
1988a7e14dcfSSatish Balay   PetscInt           max_funcs=tao->max_funcs;
1989a7e14dcfSSatish Balay   PetscReal          gnorm=tao->residual, gnorm0=tao->gnorm0;
1990a7e14dcfSSatish Balay   PetscReal          f=tao->fc, steptol=tao->steptol,trradius=tao->step;
1991a7e14dcfSSatish Balay   PetscReal          gatol=tao->gatol,grtol=tao->grtol,gttol=tao->gttol;
1992e52336cbSBarry Smith   PetscReal          catol=tao->catol,crtol=tao->crtol;
1993e52336cbSBarry Smith   PetscReal          fmin=tao->fmin, cnorm=tao->cnorm;
1994e4cb33bbSBarry Smith   TaoConvergedReason reason=tao->reason;
1995a7e14dcfSSatish Balay 
1996a7e14dcfSSatish Balay   PetscFunctionBegin;
1997441846f8SBarry Smith   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
199894511df7SStefano Zampini   if (reason != TAO_CONTINUE_ITERATING) PetscFunctionReturn(0);
1999a7e14dcfSSatish Balay 
2000a7e14dcfSSatish Balay   if (PetscIsInfOrNanReal(f)) {
20019566063dSJacob Faibussowitsch     PetscCall(PetscInfo(tao,"Failed to converged, function value is Inf or NaN\n"));
2002a7e14dcfSSatish Balay     reason = TAO_DIVERGED_NAN;
2003a7e14dcfSSatish Balay   } else if (f <= fmin && cnorm <=catol) {
20049566063dSJacob Faibussowitsch     PetscCall(PetscInfo(tao,"Converged due to function value %g < minimum function value %g\n", (double)f,(double)fmin));
2005a7e14dcfSSatish Balay     reason = TAO_CONVERGED_MINF;
2006a7e14dcfSSatish Balay   } else if (gnorm<= gatol && cnorm <=catol) {
20079566063dSJacob Faibussowitsch     PetscCall(PetscInfo(tao,"Converged due to residual norm ||g(X)||=%g < %g\n",(double)gnorm,(double)gatol));
2008a7e14dcfSSatish Balay     reason = TAO_CONVERGED_GATOL;
2009a7e14dcfSSatish Balay   } else if (f!=0 && PetscAbsReal(gnorm/f) <= grtol && cnorm <= crtol) {
20109566063dSJacob Faibussowitsch     PetscCall(PetscInfo(tao,"Converged due to residual ||g(X)||/|f(X)| =%g < %g\n",(double)(gnorm/f),(double)grtol));
2011a7e14dcfSSatish Balay     reason = TAO_CONVERGED_GRTOL;
2012e1d16bb9SBarry Smith   } else if (gnorm0 != 0 && ((gttol == 0 && gnorm == 0) || gnorm/gnorm0 < gttol) && cnorm <= crtol) {
20139566063dSJacob Faibussowitsch     PetscCall(PetscInfo(tao,"Converged due to relative residual norm ||g(X)||/||g(X0)|| = %g < %g\n",(double)(gnorm/gnorm0),(double)gttol));
2014a7e14dcfSSatish Balay     reason = TAO_CONVERGED_GTTOL;
201594511df7SStefano Zampini   } else if (max_funcs >=0 && nfuncs > max_funcs) {
20169566063dSJacob Faibussowitsch     PetscCall(PetscInfo(tao,"Exceeded maximum number of function evaluations: %" PetscInt_FMT " > %" PetscInt_FMT "\n", nfuncs,max_funcs));
2017a7e14dcfSSatish Balay     reason = TAO_DIVERGED_MAXFCN;
2018a7e14dcfSSatish Balay   } else if (tao->lsflag != 0) {
20199566063dSJacob Faibussowitsch     PetscCall(PetscInfo(tao,"Tao Line Search failure.\n"));
2020a7e14dcfSSatish Balay     reason = TAO_DIVERGED_LS_FAILURE;
2021a7e14dcfSSatish Balay   } else if (trradius < steptol && niter > 0) {
20229566063dSJacob Faibussowitsch     PetscCall(PetscInfo(tao,"Trust region/step size too small: %g < %g\n", (double)trradius,(double)steptol));
2023a7e14dcfSSatish Balay     reason = TAO_CONVERGED_STEPTOL;
2024e031d6f5SAlp Dener   } else if (niter >= tao->max_it) {
202563a3b9bcSJacob Faibussowitsch     PetscCall(PetscInfo(tao,"Exceeded maximum number of iterations: %" PetscInt_FMT " > %" PetscInt_FMT "\n",niter,tao->max_it));
2026a7e14dcfSSatish Balay     reason = TAO_DIVERGED_MAXITS;
2027a7e14dcfSSatish Balay   } else {
2028a7e14dcfSSatish Balay     reason = TAO_CONTINUE_ITERATING;
2029a7e14dcfSSatish Balay   }
2030a7e14dcfSSatish Balay   tao->reason = reason;
2031a7e14dcfSSatish Balay   PetscFunctionReturn(0);
2032a7e14dcfSSatish Balay }
2033a7e14dcfSSatish Balay 
2034a7e14dcfSSatish Balay /*@C
2035a7e14dcfSSatish Balay    TaoSetOptionsPrefix - Sets the prefix used for searching for all
2036a7e14dcfSSatish Balay    TAO options in the database.
2037a7e14dcfSSatish Balay 
2038441846f8SBarry Smith    Logically Collective on Tao
2039a7e14dcfSSatish Balay 
2040a7e14dcfSSatish Balay    Input Parameters:
2041441846f8SBarry Smith +  tao - the Tao context
2042a7e14dcfSSatish Balay -  prefix - the prefix string to prepend to all TAO option requests
2043a7e14dcfSSatish Balay 
2044a7e14dcfSSatish Balay    Notes:
2045a7e14dcfSSatish Balay    A hyphen (-) must NOT be given at the beginning of the prefix name.
2046a7e14dcfSSatish Balay    The first character of all runtime options is AUTOMATICALLY the hyphen.
2047a7e14dcfSSatish Balay 
2048a7e14dcfSSatish Balay    For example, to distinguish between the runtime options for two
2049a7e14dcfSSatish Balay    different TAO solvers, one could call
2050a7e14dcfSSatish Balay .vb
2051a7e14dcfSSatish Balay       TaoSetOptionsPrefix(tao1,"sys1_")
2052a7e14dcfSSatish Balay       TaoSetOptionsPrefix(tao2,"sys2_")
2053a7e14dcfSSatish Balay .ve
2054a7e14dcfSSatish Balay 
2055a7e14dcfSSatish Balay    This would enable use of different options for each system, such as
2056a7e14dcfSSatish Balay .vb
20579fa2b5dcSStefano Zampini       -sys1_tao_method blmvm -sys1_tao_grtol 1.e-3
20589fa2b5dcSStefano Zampini       -sys2_tao_method lmvm  -sys2_tao_grtol 1.e-4
2059a7e14dcfSSatish Balay .ve
2060a7e14dcfSSatish Balay 
2061a7e14dcfSSatish Balay    Level: advanced
2062a7e14dcfSSatish Balay 
2063db781477SPatrick Sanan .seealso: `TaoAppendOptionsPrefix()`, `TaoGetOptionsPrefix()`
2064a7e14dcfSSatish Balay @*/
2065a7e14dcfSSatish Balay 
2066441846f8SBarry Smith PetscErrorCode TaoSetOptionsPrefix(Tao tao, const char p[])
2067a7e14dcfSSatish Balay {
2068a7e14dcfSSatish Balay   PetscFunctionBegin;
206994511df7SStefano Zampini   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
20709566063dSJacob Faibussowitsch   PetscCall(PetscObjectSetOptionsPrefix((PetscObject)tao,p));
2071a7e14dcfSSatish Balay   if (tao->linesearch) {
20729566063dSJacob Faibussowitsch     PetscCall(TaoLineSearchSetOptionsPrefix(tao->linesearch,p));
2073a7e14dcfSSatish Balay   }
2074a7e14dcfSSatish Balay   if (tao->ksp) {
20759566063dSJacob Faibussowitsch     PetscCall(KSPSetOptionsPrefix(tao->ksp,p));
2076a7e14dcfSSatish Balay   }
2077a7e14dcfSSatish Balay   PetscFunctionReturn(0);
2078a7e14dcfSSatish Balay }
2079a7e14dcfSSatish Balay 
2080a7e14dcfSSatish Balay /*@C
2081a7e14dcfSSatish Balay    TaoAppendOptionsPrefix - Appends to the prefix used for searching for all
2082a7e14dcfSSatish Balay    TAO options in the database.
2083a7e14dcfSSatish Balay 
2084441846f8SBarry Smith    Logically Collective on Tao
2085a7e14dcfSSatish Balay 
2086a7e14dcfSSatish Balay    Input Parameters:
2087441846f8SBarry Smith +  tao - the Tao solver context
2088a7e14dcfSSatish Balay -  prefix - the prefix string to prepend to all TAO option requests
2089a7e14dcfSSatish Balay 
2090a7e14dcfSSatish Balay    Notes:
2091a7e14dcfSSatish Balay    A hyphen (-) must NOT be given at the beginning of the prefix name.
2092a7e14dcfSSatish Balay    The first character of all runtime options is AUTOMATICALLY the hyphen.
2093a7e14dcfSSatish Balay 
2094a7e14dcfSSatish Balay    Level: advanced
2095a7e14dcfSSatish Balay 
2096db781477SPatrick Sanan .seealso: `TaoSetOptionsPrefix()`, `TaoGetOptionsPrefix()`
2097a7e14dcfSSatish Balay @*/
2098441846f8SBarry Smith PetscErrorCode TaoAppendOptionsPrefix(Tao tao, const char p[])
2099a7e14dcfSSatish Balay {
2100a7e14dcfSSatish Balay   PetscFunctionBegin;
210194511df7SStefano Zampini   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
21029566063dSJacob Faibussowitsch   PetscCall(PetscObjectAppendOptionsPrefix((PetscObject)tao,p));
2103a7e14dcfSSatish Balay   if (tao->linesearch) {
21049566063dSJacob Faibussowitsch     PetscCall(PetscObjectAppendOptionsPrefix((PetscObject)tao->linesearch,p));
2105a7e14dcfSSatish Balay   }
2106a7e14dcfSSatish Balay   if (tao->ksp) {
21079566063dSJacob Faibussowitsch     PetscCall(KSPAppendOptionsPrefix(tao->ksp,p));
2108a7e14dcfSSatish Balay   }
2109a7e14dcfSSatish Balay   PetscFunctionReturn(0);
2110a7e14dcfSSatish Balay }
2111a7e14dcfSSatish Balay 
2112a7e14dcfSSatish Balay /*@C
2113a7e14dcfSSatish Balay   TaoGetOptionsPrefix - Gets the prefix used for searching for all
2114a7e14dcfSSatish Balay   TAO options in the database
2115a7e14dcfSSatish Balay 
2116a7e14dcfSSatish Balay   Not Collective
2117a7e14dcfSSatish Balay 
2118a7e14dcfSSatish Balay   Input Parameters:
2119441846f8SBarry Smith . tao - the Tao context
2120a7e14dcfSSatish Balay 
2121a7e14dcfSSatish Balay   Output Parameters:
2122a7e14dcfSSatish Balay . prefix - pointer to the prefix string used is returned
2123a7e14dcfSSatish Balay 
212495452b02SPatrick Sanan   Notes:
212595452b02SPatrick Sanan     On the fortran side, the user should pass in a string 'prefix' of
2126a7e14dcfSSatish Balay   sufficient length to hold the prefix.
2127a7e14dcfSSatish Balay 
2128a7e14dcfSSatish Balay   Level: advanced
2129a7e14dcfSSatish Balay 
2130db781477SPatrick Sanan .seealso: `TaoSetOptionsPrefix()`, `TaoAppendOptionsPrefix()`
2131a7e14dcfSSatish Balay @*/
2132441846f8SBarry Smith PetscErrorCode TaoGetOptionsPrefix(Tao tao, const char *p[])
2133a7e14dcfSSatish Balay {
213494511df7SStefano Zampini   PetscFunctionBegin;
213594511df7SStefano Zampini   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
21369566063dSJacob Faibussowitsch   PetscCall(PetscObjectGetOptionsPrefix((PetscObject)tao,p));
213794511df7SStefano Zampini   PetscFunctionReturn(0);
2138a7e14dcfSSatish Balay }
2139a7e14dcfSSatish Balay 
2140a7e14dcfSSatish Balay /*@C
2141a7e14dcfSSatish Balay    TaoSetType - Sets the method for the unconstrained minimization solver.
2142a7e14dcfSSatish Balay 
2143441846f8SBarry Smith    Collective on Tao
2144a7e14dcfSSatish Balay 
2145a7e14dcfSSatish Balay    Input Parameters:
2146441846f8SBarry Smith +  solver - the Tao solver context
2147a7e14dcfSSatish Balay -  type - a known method
2148a7e14dcfSSatish Balay 
2149a7e14dcfSSatish Balay    Options Database Key:
215058417fe7SBarry Smith .  -tao_type <type> - Sets the method; use -help for a list
215158417fe7SBarry Smith    of available methods (for instance, "-tao_type lmvm" or "-tao_type tron")
2152a7e14dcfSSatish Balay 
2153a7e14dcfSSatish Balay    Available methods include:
215458417fe7SBarry Smith +    nls - Newton's method with line search for unconstrained minimization
215558417fe7SBarry Smith .    ntr - Newton's method with trust region for unconstrained minimization
215658417fe7SBarry Smith .    ntl - Newton's method with trust region, line search for unconstrained minimization
215758417fe7SBarry Smith .    lmvm - Limited memory variable metric method for unconstrained minimization
215858417fe7SBarry Smith .    cg - Nonlinear conjugate gradient method for unconstrained minimization
215958417fe7SBarry Smith .    nm - Nelder-Mead algorithm for derivate-free unconstrained minimization
216058417fe7SBarry Smith .    tron - Newton Trust Region method for bound constrained minimization
216158417fe7SBarry Smith .    gpcg - Newton Trust Region method for quadratic bound constrained minimization
216258417fe7SBarry Smith .    blmvm - Limited memory variable metric method for bound constrained minimization
21631eb8069cSJason Sarich -    pounders - Model-based algorithm pounder extended for nonlinear least squares
2164a7e14dcfSSatish Balay 
2165a7e14dcfSSatish Balay   Level: intermediate
2166a7e14dcfSSatish Balay 
2167db781477SPatrick Sanan .seealso: `TaoCreate()`, `TaoGetType()`, `TaoType`
2168a7e14dcfSSatish Balay 
2169a7e14dcfSSatish Balay @*/
2170b625d6c7SJed Brown PetscErrorCode TaoSetType(Tao tao, TaoType type)
2171a7e14dcfSSatish Balay {
2172441846f8SBarry Smith   PetscErrorCode (*create_xxx)(Tao);
2173a7e14dcfSSatish Balay   PetscBool      issame;
2174a7e14dcfSSatish Balay 
2175a7e14dcfSSatish Balay   PetscFunctionBegin;
2176441846f8SBarry Smith   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
2177a7e14dcfSSatish Balay 
21789566063dSJacob Faibussowitsch   PetscCall(PetscObjectTypeCompare((PetscObject)tao,type,&issame));
2179a7e14dcfSSatish Balay   if (issame) PetscFunctionReturn(0);
2180a7e14dcfSSatish Balay 
21819566063dSJacob Faibussowitsch   PetscCall(PetscFunctionListFind(TaoList, type, (void(**)(void))&create_xxx));
21823c859ba3SBarry Smith   PetscCheck(create_xxx,PetscObjectComm((PetscObject)tao),PETSC_ERR_ARG_UNKNOWN_TYPE,"Unable to find requested Tao type %s",type);
2183a7e14dcfSSatish Balay 
2184a7e14dcfSSatish Balay   /* Destroy the existing solver information */
2185*a958fbfcSStefano Zampini   if (tao->ops->destroy) PetscCall((*tao->ops->destroy)(tao));
21869566063dSJacob Faibussowitsch   PetscCall(TaoLineSearchDestroy(&tao->linesearch));
218783c8fe1dSLisandro Dalcin   tao->ops->setup          = NULL;
218883c8fe1dSLisandro Dalcin   tao->ops->solve          = NULL;
218983c8fe1dSLisandro Dalcin   tao->ops->view           = NULL;
219083c8fe1dSLisandro Dalcin   tao->ops->setfromoptions = NULL;
219183c8fe1dSLisandro Dalcin   tao->ops->destroy        = NULL;
2192a7e14dcfSSatish Balay 
2193a7e14dcfSSatish Balay   tao->setupcalled = PETSC_FALSE;
2194a7e14dcfSSatish Balay 
21959566063dSJacob Faibussowitsch   PetscCall((*create_xxx)(tao));
21969566063dSJacob Faibussowitsch   PetscCall(PetscObjectChangeTypeName((PetscObject)tao,type));
2197a7e14dcfSSatish Balay   PetscFunctionReturn(0);
2198a7e14dcfSSatish Balay }
219947a47007SBarry Smith 
2200a7e14dcfSSatish Balay /*MC
2201441846f8SBarry Smith    TaoRegister - Adds a method to the TAO package for unconstrained minimization.
2202a7e14dcfSSatish Balay 
2203a7e14dcfSSatish Balay    Synopsis:
22045bd1e576SStefano Zampini    TaoRegister(char *name_solver,char *path,char *name_Create,PetscErrorCode (*routine_Create)(Tao))
2205a7e14dcfSSatish Balay 
2206a7e14dcfSSatish Balay    Not collective
2207a7e14dcfSSatish Balay 
2208a7e14dcfSSatish Balay    Input Parameters:
2209a7e14dcfSSatish Balay +  sname - name of a new user-defined solver
2210a7e14dcfSSatish Balay -  func - routine to Create method context
2211a7e14dcfSSatish Balay 
2212a7e14dcfSSatish Balay    Notes:
2213441846f8SBarry Smith    TaoRegister() may be called multiple times to add several user-defined solvers.
2214a7e14dcfSSatish Balay 
2215a7e14dcfSSatish Balay    Sample usage:
2216a7e14dcfSSatish Balay .vb
2217441846f8SBarry Smith    TaoRegister("my_solver",MySolverCreate);
2218a7e14dcfSSatish Balay .ve
2219a7e14dcfSSatish Balay 
2220a7e14dcfSSatish Balay    Then, your solver can be chosen with the procedural interface via
2221a7e14dcfSSatish Balay $     TaoSetType(tao,"my_solver")
2222a7e14dcfSSatish Balay    or at runtime via the option
222358417fe7SBarry Smith $     -tao_type my_solver
2224a7e14dcfSSatish Balay 
2225a7e14dcfSSatish Balay    Level: advanced
2226a7e14dcfSSatish Balay 
2227db781477SPatrick Sanan .seealso: `TaoRegisterAll()`, `TaoRegisterDestroy()`
2228a7e14dcfSSatish Balay M*/
2229441846f8SBarry Smith PetscErrorCode TaoRegister(const char sname[], PetscErrorCode (*func)(Tao))
2230a7e14dcfSSatish Balay {
2231a7e14dcfSSatish Balay   PetscFunctionBegin;
22329566063dSJacob Faibussowitsch   PetscCall(TaoInitializePackage());
22339566063dSJacob Faibussowitsch   PetscCall(PetscFunctionListAdd(&TaoList,sname,(void (*)(void))func));
2234a7e14dcfSSatish Balay   PetscFunctionReturn(0);
2235a7e14dcfSSatish Balay }
2236a7e14dcfSSatish Balay 
2237a7e14dcfSSatish Balay /*@C
2238441846f8SBarry Smith    TaoRegisterDestroy - Frees the list of minimization solvers that were
2239441846f8SBarry Smith    registered by TaoRegisterDynamic().
2240a7e14dcfSSatish Balay 
2241a7e14dcfSSatish Balay    Not Collective
2242a7e14dcfSSatish Balay 
2243a7e14dcfSSatish Balay    Level: advanced
2244a7e14dcfSSatish Balay 
2245db781477SPatrick Sanan .seealso: `TaoRegisterAll()`, `TaoRegister()`
2246a7e14dcfSSatish Balay @*/
2247441846f8SBarry Smith PetscErrorCode TaoRegisterDestroy(void)
2248a7e14dcfSSatish Balay {
2249a7e14dcfSSatish Balay   PetscFunctionBegin;
22509566063dSJacob Faibussowitsch   PetscCall(PetscFunctionListDestroy(&TaoList));
2251441846f8SBarry Smith   TaoRegisterAllCalled = PETSC_FALSE;
2252a7e14dcfSSatish Balay   PetscFunctionReturn(0);
2253a7e14dcfSSatish Balay }
225447a47007SBarry Smith 
22558931d482SJason Sarich /*@
22568931d482SJason Sarich    TaoGetIterationNumber - Gets the number of Tao iterations completed
22578931d482SJason Sarich    at this time.
22588931d482SJason Sarich 
22598931d482SJason Sarich    Not Collective
22608931d482SJason Sarich 
22618931d482SJason Sarich    Input Parameter:
22628931d482SJason Sarich .  tao - Tao context
22638931d482SJason Sarich 
22648931d482SJason Sarich    Output Parameter:
22658931d482SJason Sarich .  iter - iteration number
22668931d482SJason Sarich 
22678931d482SJason Sarich    Notes:
22688931d482SJason Sarich    For example, during the computation of iteration 2 this would return 1.
22698931d482SJason Sarich 
22708931d482SJason Sarich    Level: intermediate
22718931d482SJason Sarich 
2272db781477SPatrick Sanan .seealso: `TaoGetLinearSolveIterations()`, `TaoGetResidualNorm()`, `TaoGetObjective()`
22738931d482SJason Sarich @*/
22748931d482SJason Sarich PetscErrorCode  TaoGetIterationNumber(Tao tao,PetscInt *iter)
22758931d482SJason Sarich {
22768931d482SJason Sarich   PetscFunctionBegin;
22778931d482SJason Sarich   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
22788931d482SJason Sarich   PetscValidIntPointer(iter,2);
22798931d482SJason Sarich   *iter = tao->niter;
22808931d482SJason Sarich   PetscFunctionReturn(0);
22818931d482SJason Sarich }
22828931d482SJason Sarich 
22838931d482SJason Sarich /*@
228479f5d8caSBarry Smith    TaoGetResidualNorm - Gets the current value of the norm of the residual
228579f5d8caSBarry Smith    at this time.
228679f5d8caSBarry Smith 
228779f5d8caSBarry Smith    Not Collective
228879f5d8caSBarry Smith 
228979f5d8caSBarry Smith    Input Parameter:
229079f5d8caSBarry Smith .  tao - Tao context
229179f5d8caSBarry Smith 
229279f5d8caSBarry Smith    Output Parameter:
229379f5d8caSBarry Smith .  value - the current value
229479f5d8caSBarry Smith 
229579f5d8caSBarry Smith    Level: intermediate
229679f5d8caSBarry Smith 
229779f5d8caSBarry Smith    Developer Note: This is the 2-norm of the residual, we cannot use TaoGetGradientNorm() because that has
229879f5d8caSBarry Smith                    a different meaning. For some reason Tao sometimes calls the gradient the residual.
229979f5d8caSBarry Smith 
2300db781477SPatrick Sanan .seealso: `TaoGetLinearSolveIterations()`, `TaoGetIterationNumber()`, `TaoGetObjective()`
230179f5d8caSBarry Smith @*/
230279f5d8caSBarry Smith PetscErrorCode TaoGetResidualNorm(Tao tao,PetscReal *value)
230379f5d8caSBarry Smith {
230479f5d8caSBarry Smith   PetscFunctionBegin;
230579f5d8caSBarry Smith   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
230679f5d8caSBarry Smith   PetscValidRealPointer(value,2);
230779f5d8caSBarry Smith   *value = tao->residual;
230879f5d8caSBarry Smith   PetscFunctionReturn(0);
230979f5d8caSBarry Smith }
231079f5d8caSBarry Smith 
231179f5d8caSBarry Smith /*@
23128931d482SJason Sarich    TaoSetIterationNumber - Sets the current iteration number.
23138931d482SJason Sarich 
231494511df7SStefano Zampini    Logically Collective on Tao
23158931d482SJason Sarich 
2316d8d19677SJose E. Roman    Input Parameters:
2317a2b725a8SWilliam Gropp +  tao - Tao context
2318a2b725a8SWilliam Gropp -  iter - iteration number
23198931d482SJason Sarich 
23208931d482SJason Sarich    Level: developer
23218931d482SJason Sarich 
2322db781477SPatrick Sanan .seealso: `TaoGetLinearSolveIterations()`
23238931d482SJason Sarich @*/
23248931d482SJason Sarich PetscErrorCode  TaoSetIterationNumber(Tao tao,PetscInt iter)
23258931d482SJason Sarich {
23268931d482SJason Sarich   PetscFunctionBegin;
23278931d482SJason Sarich   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
232894511df7SStefano Zampini   PetscValidLogicalCollectiveInt(tao,iter,2);
23299566063dSJacob Faibussowitsch   PetscCall(PetscObjectSAWsTakeAccess((PetscObject)tao));
23308931d482SJason Sarich   tao->niter = iter;
23319566063dSJacob Faibussowitsch   PetscCall(PetscObjectSAWsGrantAccess((PetscObject)tao));
23328931d482SJason Sarich   PetscFunctionReturn(0);
23338931d482SJason Sarich }
23348931d482SJason Sarich 
23358931d482SJason Sarich /*@
23368931d482SJason Sarich    TaoGetTotalIterationNumber - Gets the total number of Tao iterations
23378931d482SJason Sarich    completed. This number keeps accumulating if multiple solves
23388931d482SJason Sarich    are called with the Tao object.
23398931d482SJason Sarich 
23408931d482SJason Sarich    Not Collective
23418931d482SJason Sarich 
23428931d482SJason Sarich    Input Parameter:
23438931d482SJason Sarich .  tao - Tao context
23448931d482SJason Sarich 
23458931d482SJason Sarich    Output Parameter:
23468931d482SJason Sarich .  iter - iteration number
23478931d482SJason Sarich 
23488931d482SJason Sarich    Notes:
23498931d482SJason Sarich    The total iteration count is updated after each solve, if there is a current
23508931d482SJason Sarich    TaoSolve() in progress then those iterations are not yet counted.
23518931d482SJason Sarich 
23528931d482SJason Sarich    Level: intermediate
23538931d482SJason Sarich 
2354db781477SPatrick Sanan .seealso: `TaoGetLinearSolveIterations()`
23558931d482SJason Sarich @*/
23568931d482SJason Sarich PetscErrorCode  TaoGetTotalIterationNumber(Tao tao,PetscInt *iter)
23578931d482SJason Sarich {
23588931d482SJason Sarich   PetscFunctionBegin;
23598931d482SJason Sarich   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
23608931d482SJason Sarich   PetscValidIntPointer(iter,2);
23618931d482SJason Sarich   *iter = tao->ntotalits;
23628931d482SJason Sarich   PetscFunctionReturn(0);
23638931d482SJason Sarich }
23648931d482SJason Sarich 
23658931d482SJason Sarich /*@
23668931d482SJason Sarich    TaoSetTotalIterationNumber - Sets the current total iteration number.
23678931d482SJason Sarich 
236894511df7SStefano Zampini    Logically Collective on Tao
23698931d482SJason Sarich 
2370d8d19677SJose E. Roman    Input Parameters:
2371a2b725a8SWilliam Gropp +  tao - Tao context
2372a2b725a8SWilliam Gropp -  iter - iteration number
23738931d482SJason Sarich 
23748931d482SJason Sarich    Level: developer
23758931d482SJason Sarich 
2376db781477SPatrick Sanan .seealso: `TaoGetLinearSolveIterations()`
23778931d482SJason Sarich @*/
23788931d482SJason Sarich PetscErrorCode  TaoSetTotalIterationNumber(Tao tao,PetscInt iter)
23798931d482SJason Sarich {
23808931d482SJason Sarich   PetscFunctionBegin;
23818931d482SJason Sarich   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
238294511df7SStefano Zampini   PetscValidLogicalCollectiveInt(tao,iter,2);
23839566063dSJacob Faibussowitsch   PetscCall(PetscObjectSAWsTakeAccess((PetscObject)tao));
23848931d482SJason Sarich   tao->ntotalits = iter;
23859566063dSJacob Faibussowitsch   PetscCall(PetscObjectSAWsGrantAccess((PetscObject)tao));
23868931d482SJason Sarich   PetscFunctionReturn(0);
23878931d482SJason Sarich }
23888931d482SJason Sarich 
2389a7e14dcfSSatish Balay /*@
2390e4cb33bbSBarry Smith   TaoSetConvergedReason - Sets the termination flag on a Tao object
2391a7e14dcfSSatish Balay 
2392441846f8SBarry Smith   Logically Collective on Tao
2393a7e14dcfSSatish Balay 
2394a7e14dcfSSatish Balay   Input Parameters:
2395441846f8SBarry Smith + tao - the Tao context
2396a7e14dcfSSatish Balay - reason - one of
2397a7e14dcfSSatish Balay $     TAO_CONVERGED_ATOL (2),
2398a7e14dcfSSatish Balay $     TAO_CONVERGED_RTOL (3),
2399a7e14dcfSSatish Balay $     TAO_CONVERGED_STEPTOL (4),
2400a7e14dcfSSatish Balay $     TAO_CONVERGED_MINF (5),
2401a7e14dcfSSatish Balay $     TAO_CONVERGED_USER (6),
2402a7e14dcfSSatish Balay $     TAO_DIVERGED_MAXITS (-2),
2403a7e14dcfSSatish Balay $     TAO_DIVERGED_NAN (-4),
2404a7e14dcfSSatish Balay $     TAO_DIVERGED_MAXFCN (-5),
2405a7e14dcfSSatish Balay $     TAO_DIVERGED_LS_FAILURE (-6),
2406a7e14dcfSSatish Balay $     TAO_DIVERGED_TR_REDUCTION (-7),
2407a7e14dcfSSatish Balay $     TAO_DIVERGED_USER (-8),
2408a7e14dcfSSatish Balay $     TAO_CONTINUE_ITERATING (0)
2409a7e14dcfSSatish Balay 
2410a7e14dcfSSatish Balay    Level: intermediate
2411a7e14dcfSSatish Balay 
2412a7e14dcfSSatish Balay @*/
2413e4cb33bbSBarry Smith PetscErrorCode TaoSetConvergedReason(Tao tao, TaoConvergedReason reason)
2414a7e14dcfSSatish Balay {
2415a7e14dcfSSatish Balay   PetscFunctionBegin;
241694511df7SStefano Zampini   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
241794511df7SStefano Zampini   PetscValidLogicalCollectiveEnum(tao,reason,2);
2418a7e14dcfSSatish Balay   tao->reason = reason;
2419a7e14dcfSSatish Balay   PetscFunctionReturn(0);
2420a7e14dcfSSatish Balay }
2421a7e14dcfSSatish Balay 
2422a7e14dcfSSatish Balay /*@
2423e4cb33bbSBarry Smith    TaoGetConvergedReason - Gets the reason the Tao iteration was stopped.
2424a7e14dcfSSatish Balay 
2425a7e14dcfSSatish Balay    Not Collective
2426a7e14dcfSSatish Balay 
2427a7e14dcfSSatish Balay    Input Parameter:
2428441846f8SBarry Smith .  tao - the Tao solver context
2429a7e14dcfSSatish Balay 
2430a7e14dcfSSatish Balay    Output Parameter:
2431a7e14dcfSSatish Balay .  reason - one of
2432a7e14dcfSSatish Balay $  TAO_CONVERGED_GATOL (3)           ||g(X)|| < gatol
2433a7e14dcfSSatish Balay $  TAO_CONVERGED_GRTOL (4)           ||g(X)|| / f(X)  < grtol
2434a7e14dcfSSatish Balay $  TAO_CONVERGED_GTTOL (5)           ||g(X)|| / ||g(X0)|| < gttol
2435a7e14dcfSSatish Balay $  TAO_CONVERGED_STEPTOL (6)         step size small
2436a7e14dcfSSatish Balay $  TAO_CONVERGED_MINF (7)            F < F_min
2437a7e14dcfSSatish Balay $  TAO_CONVERGED_USER (8)            User defined
2438a7e14dcfSSatish Balay $  TAO_DIVERGED_MAXITS (-2)          its > maxits
2439a7e14dcfSSatish Balay $  TAO_DIVERGED_NAN (-4)             Numerical problems
2440a7e14dcfSSatish Balay $  TAO_DIVERGED_MAXFCN (-5)          fevals > max_funcsals
2441a7e14dcfSSatish Balay $  TAO_DIVERGED_LS_FAILURE (-6)      line search failure
2442a7e14dcfSSatish Balay $  TAO_DIVERGED_TR_REDUCTION (-7)    trust region failure
2443a7e14dcfSSatish Balay $  TAO_DIVERGED_USER(-8)             (user defined)
2444a7e14dcfSSatish Balay $  TAO_CONTINUE_ITERATING (0)
2445a7e14dcfSSatish Balay 
2446a7e14dcfSSatish Balay    where
2447a7e14dcfSSatish Balay +  X - current solution
2448a7e14dcfSSatish Balay .  X0 - initial guess
2449a7e14dcfSSatish Balay .  f(X) - current function value
2450a7e14dcfSSatish Balay .  f(X*) - true solution (estimated)
2451a7e14dcfSSatish Balay .  g(X) - current gradient
2452a7e14dcfSSatish Balay .  its - current iterate number
2453a7e14dcfSSatish Balay .  maxits - maximum number of iterates
2454a7e14dcfSSatish Balay .  fevals - number of function evaluations
2455a7e14dcfSSatish Balay -  max_funcsals - maximum number of function evaluations
2456a7e14dcfSSatish Balay 
2457a7e14dcfSSatish Balay    Level: intermediate
2458a7e14dcfSSatish Balay 
2459db781477SPatrick Sanan .seealso: `TaoSetConvergenceTest()`, `TaoSetTolerances()`
2460a7e14dcfSSatish Balay 
2461a7e14dcfSSatish Balay @*/
2462e4cb33bbSBarry Smith PetscErrorCode TaoGetConvergedReason(Tao tao, TaoConvergedReason *reason)
2463a7e14dcfSSatish Balay {
2464a7e14dcfSSatish Balay   PetscFunctionBegin;
2465441846f8SBarry Smith   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
2466a7e14dcfSSatish Balay   PetscValidPointer(reason,2);
2467a7e14dcfSSatish Balay   *reason = tao->reason;
2468a7e14dcfSSatish Balay   PetscFunctionReturn(0);
2469a7e14dcfSSatish Balay }
2470a7e14dcfSSatish Balay 
2471a7e14dcfSSatish Balay /*@
2472a7e14dcfSSatish Balay    TaoGetSolutionStatus - Get the current iterate, objective value,
2473a7e14dcfSSatish Balay    residual, infeasibility, and termination
2474a7e14dcfSSatish Balay 
2475a7e14dcfSSatish Balay    Not Collective
2476a7e14dcfSSatish Balay 
2477f899ff85SJose E. Roman    Input Parameter:
2478441846f8SBarry Smith .  tao - the Tao context
2479a7e14dcfSSatish Balay 
2480a7e14dcfSSatish Balay    Output Parameters:
2481a7e14dcfSSatish Balay +  iterate - the current iterate number (>=0)
2482a7e14dcfSSatish Balay .  f - the current function value
248358417fe7SBarry Smith .  gnorm - the square of the gradient norm, duality gap, or other measure indicating distance from optimality.
2484a7e14dcfSSatish Balay .  cnorm - the infeasibility of the current solution with regard to the constraints.
2485a7e14dcfSSatish Balay .  xdiff - the step length or trust region radius of the most recent iterate.
2486a7e14dcfSSatish Balay -  reason - The termination reason, which can equal TAO_CONTINUE_ITERATING
2487a7e14dcfSSatish Balay 
2488a7e14dcfSSatish Balay    Level: intermediate
2489a7e14dcfSSatish Balay 
2490a7e14dcfSSatish Balay    Note:
2491a7e14dcfSSatish Balay    TAO returns the values set by the solvers in the routine TaoMonitor().
2492a7e14dcfSSatish Balay 
2493a7e14dcfSSatish Balay    Note:
249447a47007SBarry Smith    If any of the output arguments are set to NULL, no corresponding value will be returned.
2495a7e14dcfSSatish Balay 
2496db781477SPatrick Sanan .seealso: `TaoMonitor()`, `TaoGetConvergedReason()`
2497a7e14dcfSSatish Balay @*/
2498e4cb33bbSBarry Smith PetscErrorCode TaoGetSolutionStatus(Tao tao, PetscInt *its, PetscReal *f, PetscReal *gnorm, PetscReal *cnorm, PetscReal *xdiff, TaoConvergedReason *reason)
2499a7e14dcfSSatish Balay {
2500a7e14dcfSSatish Balay   PetscFunctionBegin;
250194511df7SStefano Zampini   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
2502a7e14dcfSSatish Balay   if (its) *its = tao->niter;
2503a7e14dcfSSatish Balay   if (f) *f = tao->fc;
2504a7e14dcfSSatish Balay   if (gnorm) *gnorm = tao->residual;
2505a7e14dcfSSatish Balay   if (cnorm) *cnorm = tao->cnorm;
2506a7e14dcfSSatish Balay   if (reason) *reason = tao->reason;
2507a7e14dcfSSatish Balay   if (xdiff) *xdiff = tao->step;
2508a7e14dcfSSatish Balay   PetscFunctionReturn(0);
2509a7e14dcfSSatish Balay }
2510a7e14dcfSSatish Balay 
2511a7e14dcfSSatish Balay /*@C
2512441846f8SBarry Smith    TaoGetType - Gets the current Tao algorithm.
2513a7e14dcfSSatish Balay 
2514a7e14dcfSSatish Balay    Not Collective
2515a7e14dcfSSatish Balay 
2516a7e14dcfSSatish Balay    Input Parameter:
2517441846f8SBarry Smith .  tao - the Tao solver context
2518a7e14dcfSSatish Balay 
2519a7e14dcfSSatish Balay    Output Parameter:
2520441846f8SBarry Smith .  type - Tao method
2521a7e14dcfSSatish Balay 
2522a7e14dcfSSatish Balay    Level: intermediate
2523a7e14dcfSSatish Balay 
2524a7e14dcfSSatish Balay @*/
2525b625d6c7SJed Brown PetscErrorCode TaoGetType(Tao tao,TaoType *type)
2526a7e14dcfSSatish Balay {
2527a7e14dcfSSatish Balay   PetscFunctionBegin;
2528441846f8SBarry Smith   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
2529a7e14dcfSSatish Balay   PetscValidPointer(type,2);
2530a7e14dcfSSatish Balay   *type = ((PetscObject)tao)->type_name;
2531a7e14dcfSSatish Balay   PetscFunctionReturn(0);
2532a7e14dcfSSatish Balay }
2533a7e14dcfSSatish Balay 
2534a7e14dcfSSatish Balay /*@C
2535a7e14dcfSSatish Balay   TaoMonitor - Monitor the solver and the current solution.  This
2536a7e14dcfSSatish Balay   routine will record the iteration number and residual statistics,
2537a7e14dcfSSatish Balay   call any monitors specified by the user, and calls the convergence-check routine.
2538a7e14dcfSSatish Balay 
2539a7e14dcfSSatish Balay    Input Parameters:
2540441846f8SBarry Smith +  tao - the Tao context
2541a7e14dcfSSatish Balay .  its - the current iterate number (>=0)
2542a7e14dcfSSatish Balay .  f - the current objective function value
254358417fe7SBarry Smith .  res - the gradient norm, square root of the duality gap, or other measure indicating distince from optimality.  This measure will be recorded and
2544a7e14dcfSSatish Balay           used for some termination tests.
2545a7e14dcfSSatish Balay .  cnorm - the infeasibility of the current solution with regard to the constraints.
2546a7e14dcfSSatish Balay -  steplength - multiple of the step direction added to the previous iterate.
2547a7e14dcfSSatish Balay 
2548a7e14dcfSSatish Balay    Output Parameters:
2549a7e14dcfSSatish Balay .  reason - The termination reason, which can equal TAO_CONTINUE_ITERATING
2550a7e14dcfSSatish Balay 
2551a7e14dcfSSatish Balay    Options Database Key:
2552a7e14dcfSSatish Balay .  -tao_monitor - Use the default monitor, which prints statistics to standard output
2553a7e14dcfSSatish Balay 
2554db781477SPatrick Sanan .seealso `TaoGetConvergedReason()`, `TaoMonitorDefault()`, `TaoSetMonitor()`
2555a7e14dcfSSatish Balay 
2556a7e14dcfSSatish Balay    Level: developer
2557a7e14dcfSSatish Balay 
2558a7e14dcfSSatish Balay @*/
25593ecd9318SAlp Dener PetscErrorCode TaoMonitor(Tao tao, PetscInt its, PetscReal f, PetscReal res, PetscReal cnorm, PetscReal steplength)
2560a7e14dcfSSatish Balay {
256147a47007SBarry Smith   PetscInt       i;
256247a47007SBarry Smith 
2563a7e14dcfSSatish Balay   PetscFunctionBegin;
2564441846f8SBarry Smith   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
2565a7e14dcfSSatish Balay   tao->fc = f;
2566a7e14dcfSSatish Balay   tao->residual = res;
2567a7e14dcfSSatish Balay   tao->cnorm = cnorm;
2568a7e14dcfSSatish Balay   tao->step = steplength;
2569e52336cbSBarry Smith   if (!its) {
257094511df7SStefano Zampini     tao->cnorm0 = cnorm;
257194511df7SStefano Zampini     tao->gnorm0 = res;
2572a7e14dcfSSatish Balay   }
25733c859ba3SBarry Smith   PetscCheck(!PetscIsInfOrNanReal(f) && !PetscIsInfOrNanReal(res),PetscObjectComm((PetscObject)tao),PETSC_ERR_USER, "User provided compute function generated Inf or NaN");
2574a7e14dcfSSatish Balay   for (i=0;i<tao->numbermonitors;i++) {
25759566063dSJacob Faibussowitsch     PetscCall((*tao->monitor[i])(tao,tao->monitorcontext[i]));
2576a7e14dcfSSatish Balay   }
2577a7e14dcfSSatish Balay   PetscFunctionReturn(0);
2578a7e14dcfSSatish Balay }
2579a7e14dcfSSatish Balay 
2580a7e14dcfSSatish Balay /*@
2581ae93cb3cSJason Sarich    TaoSetConvergenceHistory - Sets the array used to hold the convergence history.
2582a7e14dcfSSatish Balay 
2583441846f8SBarry Smith    Logically Collective on Tao
2584a7e14dcfSSatish Balay 
2585a7e14dcfSSatish Balay    Input Parameters:
2586441846f8SBarry Smith +  tao - the Tao solver context
2587a7e14dcfSSatish Balay .  obj   - array to hold objective value history
2588a7e14dcfSSatish Balay .  resid - array to hold residual history
2589a7e14dcfSSatish Balay .  cnorm - array to hold constraint violation history
2590be1558d8SJason Sarich .  lits - integer array holds the number of linear iterations for each Tao iteration
2591a7e14dcfSSatish Balay .  na  - size of obj, resid, and cnorm
2592a7e14dcfSSatish Balay -  reset - PetscTrue indicates each new minimization resets the history counter to zero,
2593a7e14dcfSSatish Balay            else it continues storing new values for new minimizations after the old ones
2594a7e14dcfSSatish Balay 
2595a7e14dcfSSatish Balay    Notes:
2596a7e14dcfSSatish Balay    If set, TAO will fill the given arrays with the indicated
2597ae93cb3cSJason Sarich    information at each iteration.  If 'obj','resid','cnorm','lits' are
2598ae93cb3cSJason Sarich    *all* NULL then space (using size na, or 1000 if na is PETSC_DECIDE or
2599ae93cb3cSJason Sarich    PETSC_DEFAULT) is allocated for the history.
2600be1558d8SJason Sarich    If not all are NULL, then only the non-NULL information categories
2601ae93cb3cSJason Sarich    will be stored, the others will be ignored.
2602ae93cb3cSJason Sarich 
2603ae93cb3cSJason Sarich    Any convergence information after iteration number 'na' will not be stored.
2604a7e14dcfSSatish Balay 
2605a7e14dcfSSatish Balay    This routine is useful, e.g., when running a code for purposes
2606a7e14dcfSSatish Balay    of accurate performance monitoring, when no I/O should be done
2607a7e14dcfSSatish Balay    during the section of code that is being timed.
2608a7e14dcfSSatish Balay 
2609a7e14dcfSSatish Balay    Level: intermediate
2610a7e14dcfSSatish Balay 
2611db781477SPatrick Sanan .seealso: `TaoGetConvergenceHistory()`
2612a7e14dcfSSatish Balay 
2613a7e14dcfSSatish Balay @*/
26141b266c99SBarry Smith PetscErrorCode TaoSetConvergenceHistory(Tao tao, PetscReal obj[], PetscReal resid[], PetscReal cnorm[], PetscInt lits[], PetscInt na,PetscBool reset)
2615a7e14dcfSSatish Balay {
2616a7e14dcfSSatish Balay   PetscFunctionBegin;
2617441846f8SBarry Smith   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
2618064a246eSJacob Faibussowitsch   if (obj) PetscValidRealPointer(obj,2);
2619064a246eSJacob Faibussowitsch   if (resid) PetscValidRealPointer(resid,3);
2620064a246eSJacob Faibussowitsch   if (cnorm) PetscValidRealPointer(cnorm,4);
2621ae93cb3cSJason Sarich   if (lits) PetscValidIntPointer(lits,5);
2622ae93cb3cSJason Sarich 
2623ae93cb3cSJason Sarich   if (na == PETSC_DECIDE || na == PETSC_DEFAULT) na = 1000;
2624ae93cb3cSJason Sarich   if (!obj && !resid && !cnorm && !lits) {
26259566063dSJacob Faibussowitsch     PetscCall(PetscCalloc4(na,&obj,na,&resid,na,&cnorm,na,&lits));
2626ae93cb3cSJason Sarich     tao->hist_malloc = PETSC_TRUE;
2627ae93cb3cSJason Sarich   }
2628ae93cb3cSJason Sarich 
2629a7e14dcfSSatish Balay   tao->hist_obj = obj;
2630a7e14dcfSSatish Balay   tao->hist_resid = resid;
2631a7e14dcfSSatish Balay   tao->hist_cnorm = cnorm;
2632ae93cb3cSJason Sarich   tao->hist_lits = lits;
2633a7e14dcfSSatish Balay   tao->hist_max   = na;
2634a7e14dcfSSatish Balay   tao->hist_reset = reset;
2635ae93cb3cSJason Sarich   tao->hist_len = 0;
2636a7e14dcfSSatish Balay   PetscFunctionReturn(0);
2637a7e14dcfSSatish Balay }
2638a7e14dcfSSatish Balay 
2639a7e14dcfSSatish Balay /*@C
2640ae93cb3cSJason Sarich    TaoGetConvergenceHistory - Gets the arrays used to hold the convergence history.
2641a7e14dcfSSatish Balay 
2642441846f8SBarry Smith    Collective on Tao
2643a7e14dcfSSatish Balay 
2644a7e14dcfSSatish Balay    Input Parameter:
2645441846f8SBarry Smith .  tao - the Tao context
2646a7e14dcfSSatish Balay 
2647a7e14dcfSSatish Balay    Output Parameters:
2648a7e14dcfSSatish Balay +  obj   - array used to hold objective value history
2649a7e14dcfSSatish Balay .  resid - array used to hold residual history
2650a7e14dcfSSatish Balay .  cnorm - array used to hold constraint violation history
2651ae93cb3cSJason Sarich .  lits  - integer array used to hold linear solver iteration count
2652dd25e5eeSStefano Zampini -  nhist  - size of obj, resid, cnorm, and lits
2653a7e14dcfSSatish Balay 
2654a7e14dcfSSatish Balay    Notes:
2655ae93cb3cSJason Sarich     This routine must be preceded by calls to TaoSetConvergenceHistory()
2656ae93cb3cSJason Sarich     and TaoSolve(), otherwise it returns useless information.
2657ae93cb3cSJason Sarich 
2658a7e14dcfSSatish Balay     The calling sequence for this routine in Fortran is
2659ae93cb3cSJason Sarich $   call TaoGetConvergenceHistory(Tao tao, PetscInt nhist, PetscErrorCode ierr)
2660a7e14dcfSSatish Balay 
2661a7e14dcfSSatish Balay    This routine is useful, e.g., when running a code for purposes
2662a7e14dcfSSatish Balay    of accurate performance monitoring, when no I/O should be done
2663a7e14dcfSSatish Balay    during the section of code that is being timed.
2664a7e14dcfSSatish Balay 
2665a7e14dcfSSatish Balay    Level: advanced
2666a7e14dcfSSatish Balay 
2667db781477SPatrick Sanan .seealso: `TaoSetConvergenceHistory()`
2668a7e14dcfSSatish Balay 
2669a7e14dcfSSatish Balay @*/
2670ae93cb3cSJason Sarich PetscErrorCode TaoGetConvergenceHistory(Tao tao, PetscReal **obj, PetscReal **resid, PetscReal **cnorm, PetscInt **lits, PetscInt *nhist)
2671a7e14dcfSSatish Balay {
2672a7e14dcfSSatish Balay   PetscFunctionBegin;
2673441846f8SBarry Smith   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
2674a7e14dcfSSatish Balay   if (obj)   *obj   = tao->hist_obj;
2675a7e14dcfSSatish Balay   if (cnorm) *cnorm = tao->hist_cnorm;
2676a7e14dcfSSatish Balay   if (resid) *resid = tao->hist_resid;
2677a7e14dcfSSatish Balay   if (nhist) *nhist = tao->hist_len;
2678a7e14dcfSSatish Balay   PetscFunctionReturn(0);
2679a7e14dcfSSatish Balay }
2680a7e14dcfSSatish Balay 
2681a7e14dcfSSatish Balay /*@
2682a7e14dcfSSatish Balay    TaoSetApplicationContext - Sets the optional user-defined context for
2683a7e14dcfSSatish Balay    a solver.
2684a7e14dcfSSatish Balay 
2685441846f8SBarry Smith    Logically Collective on Tao
2686a7e14dcfSSatish Balay 
2687a7e14dcfSSatish Balay    Input Parameters:
2688441846f8SBarry Smith +  tao  - the Tao context
2689a7e14dcfSSatish Balay -  usrP - optional user context
2690a7e14dcfSSatish Balay 
2691a7e14dcfSSatish Balay    Level: intermediate
2692a7e14dcfSSatish Balay 
2693db781477SPatrick Sanan .seealso: `TaoGetApplicationContext()`, `TaoSetApplicationContext()`
2694a7e14dcfSSatish Balay @*/
2695441846f8SBarry Smith PetscErrorCode  TaoSetApplicationContext(Tao tao,void *usrP)
2696a7e14dcfSSatish Balay {
2697a7e14dcfSSatish Balay   PetscFunctionBegin;
2698441846f8SBarry Smith   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
2699a7e14dcfSSatish Balay   tao->user = usrP;
2700a7e14dcfSSatish Balay   PetscFunctionReturn(0);
2701a7e14dcfSSatish Balay }
2702a7e14dcfSSatish Balay 
2703a7e14dcfSSatish Balay /*@
2704a7e14dcfSSatish Balay    TaoGetApplicationContext - Gets the user-defined context for a
2705a7e14dcfSSatish Balay    TAO solvers.
2706a7e14dcfSSatish Balay 
2707a7e14dcfSSatish Balay    Not Collective
2708a7e14dcfSSatish Balay 
2709a7e14dcfSSatish Balay    Input Parameter:
2710441846f8SBarry Smith .  tao  - Tao context
2711a7e14dcfSSatish Balay 
2712a7e14dcfSSatish Balay    Output Parameter:
2713a7e14dcfSSatish Balay .  usrP - user context
2714a7e14dcfSSatish Balay 
2715a7e14dcfSSatish Balay    Level: intermediate
2716a7e14dcfSSatish Balay 
2717db781477SPatrick Sanan .seealso: `TaoSetApplicationContext()`
2718a7e14dcfSSatish Balay @*/
2719441846f8SBarry Smith PetscErrorCode  TaoGetApplicationContext(Tao tao,void *usrP)
2720a7e14dcfSSatish Balay {
2721a7e14dcfSSatish Balay   PetscFunctionBegin;
2722441846f8SBarry Smith   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
272394511df7SStefano Zampini   PetscValidPointer(usrP,2);
2724a7e14dcfSSatish Balay   *(void**)usrP = tao->user;
2725a7e14dcfSSatish Balay   PetscFunctionReturn(0);
2726a7e14dcfSSatish Balay }
2727a9603a14SPatrick Farrell 
2728a9603a14SPatrick Farrell /*@
2729a9603a14SPatrick Farrell    TaoSetGradientNorm - Sets the matrix used to define the inner product that measures the size of the gradient.
2730a9603a14SPatrick Farrell 
2731a9603a14SPatrick Farrell    Collective on tao
2732a9603a14SPatrick Farrell 
2733a9603a14SPatrick Farrell    Input Parameters:
2734a9603a14SPatrick Farrell +  tao  - the Tao context
2735a9603a14SPatrick Farrell -  M    - gradient norm
2736a9603a14SPatrick Farrell 
2737a9603a14SPatrick Farrell    Level: beginner
2738a9603a14SPatrick Farrell 
2739db781477SPatrick Sanan .seealso: `TaoGetGradientNorm()`, `TaoGradientNorm()`
2740a9603a14SPatrick Farrell @*/
2741a9603a14SPatrick Farrell PetscErrorCode  TaoSetGradientNorm(Tao tao, Mat M)
2742a9603a14SPatrick Farrell {
2743a9603a14SPatrick Farrell   PetscFunctionBegin;
2744a9603a14SPatrick Farrell   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
274594511df7SStefano Zampini   PetscValidHeaderSpecific(M,MAT_CLASSID,2);
27469566063dSJacob Faibussowitsch   PetscCall(PetscObjectReference((PetscObject)M));
27479566063dSJacob Faibussowitsch   PetscCall(MatDestroy(&tao->gradient_norm));
27489566063dSJacob Faibussowitsch   PetscCall(VecDestroy(&tao->gradient_norm_tmp));
2749a9603a14SPatrick Farrell   tao->gradient_norm = M;
27509566063dSJacob Faibussowitsch   PetscCall(MatCreateVecs(M, NULL, &tao->gradient_norm_tmp));
2751a9603a14SPatrick Farrell   PetscFunctionReturn(0);
2752a9603a14SPatrick Farrell }
2753a9603a14SPatrick Farrell 
2754a9603a14SPatrick Farrell /*@
2755a9603a14SPatrick Farrell    TaoGetGradientNorm - Returns the matrix used to define the inner product for measuring the size of the gradient.
2756a9603a14SPatrick Farrell 
2757a9603a14SPatrick Farrell    Not Collective
2758a9603a14SPatrick Farrell 
2759a9603a14SPatrick Farrell    Input Parameter:
2760a9603a14SPatrick Farrell .  tao  - Tao context
2761a9603a14SPatrick Farrell 
2762a9603a14SPatrick Farrell    Output Parameter:
2763a9603a14SPatrick Farrell .  M - gradient norm
2764a9603a14SPatrick Farrell 
2765a9603a14SPatrick Farrell    Level: beginner
2766a9603a14SPatrick Farrell 
2767db781477SPatrick Sanan .seealso: `TaoSetGradientNorm()`, `TaoGradientNorm()`
2768a9603a14SPatrick Farrell @*/
2769a9603a14SPatrick Farrell PetscErrorCode  TaoGetGradientNorm(Tao tao, Mat *M)
2770a9603a14SPatrick Farrell {
2771a9603a14SPatrick Farrell   PetscFunctionBegin;
2772a9603a14SPatrick Farrell   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
277394511df7SStefano Zampini   PetscValidPointer(M,2);
2774a9603a14SPatrick Farrell   *M = tao->gradient_norm;
2775a9603a14SPatrick Farrell   PetscFunctionReturn(0);
2776a9603a14SPatrick Farrell }
2777a9603a14SPatrick Farrell 
277805aca2cdSSatish Balay /*@C
2779a9603a14SPatrick Farrell    TaoGradientNorm - Compute the norm with respect to the inner product the user has set.
2780a9603a14SPatrick Farrell 
2781a9603a14SPatrick Farrell    Collective on tao
2782a9603a14SPatrick Farrell 
2783d8d19677SJose E. Roman    Input Parameters:
27842653c764SPierre Jolivet +  tao      - the Tao context
2785a9603a14SPatrick Farrell .  gradient - the gradient to be computed
27862653c764SPierre Jolivet -  norm     - the norm type
2787a9603a14SPatrick Farrell 
2788a9603a14SPatrick Farrell    Output Parameter:
2789a9603a14SPatrick Farrell .  gnorm    - the gradient norm
2790a9603a14SPatrick Farrell 
2791a9603a14SPatrick Farrell    Level: developer
2792a9603a14SPatrick Farrell 
2793db781477SPatrick Sanan .seealso: `TaoSetGradientNorm()`, `TaoGetGradientNorm()`
2794a9603a14SPatrick Farrell @*/
2795a9603a14SPatrick Farrell PetscErrorCode  TaoGradientNorm(Tao tao, Vec gradient, NormType type, PetscReal *gnorm)
2796a9603a14SPatrick Farrell {
2797a9603a14SPatrick Farrell   PetscFunctionBegin;
27988854b543SStefano Zampini   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
27998854b543SStefano Zampini   PetscValidHeaderSpecific(gradient,VEC_CLASSID,2);
28008854b543SStefano Zampini   PetscValidLogicalCollectiveEnum(tao,type,3);
280194511df7SStefano Zampini   PetscValidRealPointer(gnorm,4);
2802a9603a14SPatrick Farrell   if (tao->gradient_norm) {
2803680e2bc7SPatrick Farrell     PetscScalar gnorms;
2804680e2bc7SPatrick Farrell 
28053c859ba3SBarry Smith     PetscCheck(type == NORM_2,PetscObjectComm((PetscObject)gradient), PETSC_ERR_ARG_WRONG, "Norm type must be NORM_2 if an inner product for the gradient norm is set.");
28069566063dSJacob Faibussowitsch     PetscCall(MatMult(tao->gradient_norm, gradient, tao->gradient_norm_tmp));
28079566063dSJacob Faibussowitsch     PetscCall(VecDot(gradient, tao->gradient_norm_tmp, &gnorms));
28087bb79937SPatrick Farrell     *gnorm = PetscRealPart(PetscSqrtScalar(gnorms));
2809a9603a14SPatrick Farrell   } else {
28109566063dSJacob Faibussowitsch     PetscCall(VecNorm(gradient, type, gnorm));
2811a9603a14SPatrick Farrell   }
2812a9603a14SPatrick Farrell   PetscFunctionReturn(0);
2813a9603a14SPatrick Farrell }
2814a9603a14SPatrick Farrell 
2815e882e171SHong Zhang /*@C
2816e882e171SHong Zhang    TaoMonitorDrawCtxCreate - Creates the monitor context for TaoMonitorDrawCtx
2817a9603a14SPatrick Farrell 
2818e882e171SHong Zhang    Collective on Tao
2819e882e171SHong Zhang 
2820e882e171SHong Zhang    Output Patameter:
2821e882e171SHong Zhang .    ctx - the monitor context
2822e882e171SHong Zhang 
2823e882e171SHong Zhang    Options Database:
2824e882e171SHong Zhang .   -tao_draw_solution_initial - show initial guess as well as current solution
2825e882e171SHong Zhang 
2826e882e171SHong Zhang    Level: intermediate
2827e882e171SHong Zhang 
2828db781477SPatrick Sanan .seealso: `TaoMonitorSet()`, `TaoMonitorDefault()`, `VecView()`, `TaoMonitorDrawCtx()`
2829e882e171SHong Zhang @*/
2830e882e171SHong Zhang PetscErrorCode  TaoMonitorDrawCtxCreate(MPI_Comm comm,const char host[],const char label[],int x,int y,int m,int n,PetscInt howoften,TaoMonitorDrawCtx *ctx)
2831e882e171SHong Zhang {
2832e882e171SHong Zhang   PetscFunctionBegin;
28339566063dSJacob Faibussowitsch   PetscCall(PetscNew(ctx));
28349566063dSJacob Faibussowitsch   PetscCall(PetscViewerDrawOpen(comm,host,label,x,y,m,n,&(*ctx)->viewer));
28359566063dSJacob Faibussowitsch   PetscCall(PetscViewerSetFromOptions((*ctx)->viewer));
2836e882e171SHong Zhang   (*ctx)->howoften = howoften;
2837e882e171SHong Zhang   PetscFunctionReturn(0);
2838e882e171SHong Zhang }
2839e882e171SHong Zhang 
2840e882e171SHong Zhang /*@C
2841e882e171SHong Zhang    TaoMonitorDrawCtxDestroy - Destroys the monitor context for TaoMonitorDrawSolution()
2842e882e171SHong Zhang 
2843e882e171SHong Zhang    Collective on Tao
2844e882e171SHong Zhang 
2845e882e171SHong Zhang    Input Parameters:
2846e882e171SHong Zhang .    ctx - the monitor context
2847e882e171SHong Zhang 
2848e882e171SHong Zhang    Level: intermediate
2849e882e171SHong Zhang 
2850db781477SPatrick Sanan .seealso: `TaoMonitorSet()`, `TaoMonitorDefault()`, `VecView()`, `TaoMonitorDrawSolution()`
2851e882e171SHong Zhang @*/
2852e882e171SHong Zhang PetscErrorCode  TaoMonitorDrawCtxDestroy(TaoMonitorDrawCtx *ictx)
2853e882e171SHong Zhang {
2854e882e171SHong Zhang   PetscFunctionBegin;
28559566063dSJacob Faibussowitsch   PetscCall(PetscViewerDestroy(&(*ictx)->viewer));
28569566063dSJacob Faibussowitsch   PetscCall(PetscFree(*ictx));
2857e882e171SHong Zhang   PetscFunctionReturn(0);
2858e882e171SHong Zhang }
2859