xref: /petsc/src/tao/interface/taosolver.c (revision 1baa6e3354bfe224b33a0c158f482508792a8a6e)
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));
175*1baa6e33SBarry Smith   if (tao->linesearch) PetscCall(TaoLineSearchReset(tao->linesearch));
176a7e14dcfSSatish Balay 
1779566063dSJacob Faibussowitsch   PetscCall(PetscLogEventBegin(TAO_Solve,tao,0,0,0));
1789566063dSJacob Faibussowitsch   if (tao->ops->solve) PetscCall((*tao->ops->solve)(tao));
1799566063dSJacob Faibussowitsch   PetscCall(PetscLogEventEnd(TAO_Solve,tao,0,0,0));
180a7e14dcfSSatish Balay 
1819566063dSJacob Faibussowitsch   PetscCall(VecViewFromOptions(tao->solution,(PetscObject)tao,"-tao_view_solution"));
1827cf37e64SBarry Smith 
1838931d482SJason Sarich   tao->ntotalits += tao->niter;
1849566063dSJacob Faibussowitsch   PetscCall(TaoViewFromOptions(tao,NULL,"-tao_view"));
185a7e14dcfSSatish Balay 
186a7e14dcfSSatish Balay   if (tao->printreason) {
187a7e14dcfSSatish Balay     if (tao->reason > 0) {
18863a3b9bcSJacob Faibussowitsch       PetscCall(PetscPrintf(((PetscObject)tao)->comm,"TAO solve converged due to %s iterations %" PetscInt_FMT "\n",TaoConvergedReasons[tao->reason],tao->niter));
189a7e14dcfSSatish Balay     } else {
19063a3b9bcSJacob Faibussowitsch       PetscCall(PetscPrintf(((PetscObject)tao)->comm,"TAO solve did not converge due to %s iteration %" PetscInt_FMT "\n",TaoConvergedReasons[tao->reason],tao->niter));
191a7e14dcfSSatish Balay     }
192a7e14dcfSSatish Balay   }
193a7e14dcfSSatish Balay   PetscFunctionReturn(0);
194a7e14dcfSSatish Balay }
195a7e14dcfSSatish Balay 
196a7e14dcfSSatish Balay /*@
197a7e14dcfSSatish Balay   TaoSetUp - Sets up the internal data structures for the later use
198a7e14dcfSSatish Balay   of a Tao solver
199a7e14dcfSSatish Balay 
200a7e14dcfSSatish Balay   Collective on tao
201a7e14dcfSSatish Balay 
202a7e14dcfSSatish Balay   Input Parameters:
203a7e14dcfSSatish Balay . tao - the TAO context
204a7e14dcfSSatish Balay 
205a7e14dcfSSatish Balay   Notes:
206a7e14dcfSSatish Balay   The user will not need to explicitly call TaoSetUp(), as it will
207a7e14dcfSSatish Balay   automatically be called in TaoSolve().  However, if the user
208a7e14dcfSSatish Balay   desires to call it explicitly, it should come after TaoCreate()
209a7e14dcfSSatish Balay   and any TaoSetSomething() routines, but before TaoSolve().
210a7e14dcfSSatish Balay 
211a7e14dcfSSatish Balay   Level: advanced
212a7e14dcfSSatish Balay 
213db781477SPatrick Sanan .seealso: `TaoCreate()`, `TaoSolve()`
214a7e14dcfSSatish Balay @*/
215441846f8SBarry Smith PetscErrorCode TaoSetUp(Tao tao)
216a7e14dcfSSatish Balay {
217a7e14dcfSSatish Balay   PetscFunctionBegin;
218441846f8SBarry Smith   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
219a7e14dcfSSatish Balay   if (tao->setupcalled) PetscFunctionReturn(0);
2200f0abf79SStefano Zampini   PetscCall(TaoSetUpEW_Private(tao));
2213c859ba3SBarry Smith   PetscCheck(tao->solution,PetscObjectComm((PetscObject)tao),PETSC_ERR_ARG_WRONGSTATE,"Must call TaoSetSolution");
2220f0abf79SStefano Zampini   if (tao->ops->setup) PetscCall((*tao->ops->setup)(tao));
223a7e14dcfSSatish Balay   tao->setupcalled = PETSC_TRUE;
224a7e14dcfSSatish Balay   PetscFunctionReturn(0);
225a7e14dcfSSatish Balay }
226a7e14dcfSSatish Balay 
2271fb7b255SJunchao Zhang /*@C
228a7e14dcfSSatish Balay   TaoDestroy - Destroys the TAO context that was created with
229a7e14dcfSSatish Balay   TaoCreate()
230a7e14dcfSSatish Balay 
231441846f8SBarry Smith   Collective on Tao
232a7e14dcfSSatish Balay 
233a7e14dcfSSatish Balay   Input Parameter:
234441846f8SBarry Smith . tao - the Tao context
235a7e14dcfSSatish Balay 
236a7e14dcfSSatish Balay   Level: beginner
237a7e14dcfSSatish Balay 
238db781477SPatrick Sanan .seealso: `TaoCreate()`, `TaoSolve()`
239a7e14dcfSSatish Balay @*/
240441846f8SBarry Smith PetscErrorCode TaoDestroy(Tao *tao)
241a7e14dcfSSatish Balay {
242a7e14dcfSSatish Balay   PetscFunctionBegin;
243a7e14dcfSSatish Balay   if (!*tao) PetscFunctionReturn(0);
244441846f8SBarry Smith   PetscValidHeaderSpecific(*tao,TAO_CLASSID,1);
24583c8fe1dSLisandro Dalcin   if (--((PetscObject)*tao)->refct > 0) {*tao = NULL;PetscFunctionReturn(0);}
246a7e14dcfSSatish Balay 
247a7e14dcfSSatish Balay   if ((*tao)->ops->destroy) {
2489566063dSJacob Faibussowitsch     PetscCall((*((*tao))->ops->destroy)(*tao));
249a7e14dcfSSatish Balay   }
2509566063dSJacob Faibussowitsch   PetscCall(KSPDestroy(&(*tao)->ksp));
2510f0abf79SStefano Zampini   PetscCall(SNESDestroy(&(*tao)->snes_ewdummy));
2529566063dSJacob Faibussowitsch   PetscCall(TaoLineSearchDestroy(&(*tao)->linesearch));
253a7e14dcfSSatish Balay 
254a7e14dcfSSatish Balay   if ((*tao)->ops->convergencedestroy) {
2559566063dSJacob Faibussowitsch     PetscCall((*(*tao)->ops->convergencedestroy)((*tao)->cnvP));
256a7e14dcfSSatish Balay     if ((*tao)->jacobian_state_inv) {
2579566063dSJacob Faibussowitsch       PetscCall(MatDestroy(&(*tao)->jacobian_state_inv));
258a7e14dcfSSatish Balay     }
259a7e14dcfSSatish Balay   }
2609566063dSJacob Faibussowitsch   PetscCall(VecDestroy(&(*tao)->solution));
2619566063dSJacob Faibussowitsch   PetscCall(VecDestroy(&(*tao)->gradient));
2629566063dSJacob Faibussowitsch   PetscCall(VecDestroy(&(*tao)->ls_res));
263a7e14dcfSSatish Balay 
264a9603a14SPatrick Farrell   if ((*tao)->gradient_norm) {
2659566063dSJacob Faibussowitsch     PetscCall(PetscObjectDereference((PetscObject)(*tao)->gradient_norm));
2669566063dSJacob Faibussowitsch     PetscCall(VecDestroy(&(*tao)->gradient_norm_tmp));
267a9603a14SPatrick Farrell   }
268a9603a14SPatrick Farrell 
2699566063dSJacob Faibussowitsch   PetscCall(VecDestroy(&(*tao)->XL));
2709566063dSJacob Faibussowitsch   PetscCall(VecDestroy(&(*tao)->XU));
2719566063dSJacob Faibussowitsch   PetscCall(VecDestroy(&(*tao)->IL));
2729566063dSJacob Faibussowitsch   PetscCall(VecDestroy(&(*tao)->IU));
2739566063dSJacob Faibussowitsch   PetscCall(VecDestroy(&(*tao)->DE));
2749566063dSJacob Faibussowitsch   PetscCall(VecDestroy(&(*tao)->DI));
2759566063dSJacob Faibussowitsch   PetscCall(VecDestroy(&(*tao)->constraints));
2769566063dSJacob Faibussowitsch   PetscCall(VecDestroy(&(*tao)->constraints_equality));
2779566063dSJacob Faibussowitsch   PetscCall(VecDestroy(&(*tao)->constraints_inequality));
2789566063dSJacob Faibussowitsch   PetscCall(VecDestroy(&(*tao)->stepdirection));
2799566063dSJacob Faibussowitsch   PetscCall(MatDestroy(&(*tao)->hessian_pre));
2809566063dSJacob Faibussowitsch   PetscCall(MatDestroy(&(*tao)->hessian));
2819566063dSJacob Faibussowitsch   PetscCall(MatDestroy(&(*tao)->ls_jac));
2829566063dSJacob Faibussowitsch   PetscCall(MatDestroy(&(*tao)->ls_jac_pre));
2839566063dSJacob Faibussowitsch   PetscCall(MatDestroy(&(*tao)->jacobian_pre));
2849566063dSJacob Faibussowitsch   PetscCall(MatDestroy(&(*tao)->jacobian));
2859566063dSJacob Faibussowitsch   PetscCall(MatDestroy(&(*tao)->jacobian_state_pre));
2869566063dSJacob Faibussowitsch   PetscCall(MatDestroy(&(*tao)->jacobian_state));
2879566063dSJacob Faibussowitsch   PetscCall(MatDestroy(&(*tao)->jacobian_state_inv));
2889566063dSJacob Faibussowitsch   PetscCall(MatDestroy(&(*tao)->jacobian_design));
2899566063dSJacob Faibussowitsch   PetscCall(MatDestroy(&(*tao)->jacobian_equality));
2909566063dSJacob Faibussowitsch   PetscCall(MatDestroy(&(*tao)->jacobian_equality_pre));
2919566063dSJacob Faibussowitsch   PetscCall(MatDestroy(&(*tao)->jacobian_inequality));
2929566063dSJacob Faibussowitsch   PetscCall(MatDestroy(&(*tao)->jacobian_inequality_pre));
2939566063dSJacob Faibussowitsch   PetscCall(ISDestroy(&(*tao)->state_is));
2949566063dSJacob Faibussowitsch   PetscCall(ISDestroy(&(*tao)->design_is));
2959566063dSJacob Faibussowitsch   PetscCall(VecDestroy(&(*tao)->res_weights_v));
2969566063dSJacob Faibussowitsch   PetscCall(TaoCancelMonitors(*tao));
297ae93cb3cSJason Sarich   if ((*tao)->hist_malloc) {
2989566063dSJacob Faibussowitsch     PetscCall(PetscFree4((*tao)->hist_obj,(*tao)->hist_resid,(*tao)->hist_cnorm,(*tao)->hist_lits));
299ae93cb3cSJason Sarich   }
300737f463aSAlp Dener   if ((*tao)->res_weights_n) {
3019566063dSJacob Faibussowitsch     PetscCall(PetscFree((*tao)->res_weights_rows));
3029566063dSJacob Faibussowitsch     PetscCall(PetscFree((*tao)->res_weights_cols));
3039566063dSJacob Faibussowitsch     PetscCall(PetscFree((*tao)->res_weights_w));
304125ddc8aSJason Sarich   }
3059566063dSJacob Faibussowitsch   PetscCall(PetscHeaderDestroy(tao));
306a7e14dcfSSatish Balay   PetscFunctionReturn(0);
307a7e14dcfSSatish Balay }
308a7e14dcfSSatish Balay 
309a7e14dcfSSatish Balay /*@
3100f0abf79SStefano Zampini    TaoKSPSetUseEW - Sets SNES use Eisenstat-Walker method for
3110f0abf79SStefano Zampini    computing relative tolerance for linear solvers.
3120f0abf79SStefano Zampini 
3130f0abf79SStefano Zampini    Logically Collective on TAO
3140f0abf79SStefano Zampini 
3150f0abf79SStefano Zampini    Input Parameters:
3160f0abf79SStefano Zampini +  tao - Tao context
3170f0abf79SStefano Zampini -  flag - PETSC_TRUE or PETSC_FALSE
3180f0abf79SStefano Zampini 
3190f0abf79SStefano Zampini    Notes:
3200f0abf79SStefano Zampini    See SNESKSPSetUseEW() for customization details.
3210f0abf79SStefano Zampini 
3220f0abf79SStefano Zampini    Level: advanced
3230f0abf79SStefano Zampini 
3240f0abf79SStefano Zampini    Reference:
3250f0abf79SStefano Zampini    S. C. Eisenstat and H. F. Walker, "Choosing the forcing terms in an
3260f0abf79SStefano Zampini    inexact Newton method", SISC 17 (1), pp.16-32, 1996.
3270f0abf79SStefano Zampini 
3280f0abf79SStefano Zampini .seealso: `SNESKSPSetUseEW()`
3290f0abf79SStefano Zampini @*/
3300f0abf79SStefano Zampini PetscErrorCode  TaoKSPSetUseEW(Tao tao,PetscBool flag)
3310f0abf79SStefano Zampini {
3320f0abf79SStefano Zampini   PetscFunctionBegin;
3330f0abf79SStefano Zampini   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
3340f0abf79SStefano Zampini   PetscValidLogicalCollectiveBool(tao,flag,2);
3350f0abf79SStefano Zampini   tao->ksp_ewconv = flag;
3360f0abf79SStefano Zampini   PetscFunctionReturn(0);
3370f0abf79SStefano Zampini }
3380f0abf79SStefano Zampini 
3390f0abf79SStefano Zampini /*@
340441846f8SBarry Smith   TaoSetFromOptions - Sets various Tao parameters from user
341a7e14dcfSSatish Balay   options.
342a7e14dcfSSatish Balay 
343441846f8SBarry Smith   Collective on Tao
344a7e14dcfSSatish Balay 
34501d2d390SJose E. Roman   Input Parameter:
346441846f8SBarry Smith . tao - the Tao solver context
347a7e14dcfSSatish Balay 
348a7e14dcfSSatish Balay   options Database Keys:
34958417fe7SBarry Smith + -tao_type <type> - The algorithm that TAO uses (lmvm, nls, etc.)
350a7e14dcfSSatish Balay . -tao_gatol <gatol> - absolute error tolerance for ||gradient||
351a7e14dcfSSatish Balay . -tao_grtol <grtol> - relative error tolerance for ||gradient||
352a7e14dcfSSatish Balay . -tao_gttol <gttol> - reduction of ||gradient|| relative to initial gradient
353a7e14dcfSSatish Balay . -tao_max_it <max> - sets maximum number of iterations
354a7e14dcfSSatish Balay . -tao_max_funcs <max> - sets maximum number of function evaluations
355a7e14dcfSSatish Balay . -tao_fmin <fmin> - stop if function value reaches fmin
356a7e14dcfSSatish Balay . -tao_steptol <tol> - stop if trust region radius less than <tol>
357a7e14dcfSSatish Balay . -tao_trust0 <t> - initial trust region radius
358a7e14dcfSSatish Balay . -tao_monitor - prints function value and residual at each iteration
359a7e14dcfSSatish Balay . -tao_smonitor - same as tao_monitor, but truncates very small values
360a7e14dcfSSatish Balay . -tao_cmonitor - prints function value, residual, and constraint norm at each iteration
361a7e14dcfSSatish Balay . -tao_view_solution - prints solution vector at each iteration
3624a48860cSAlp Dener . -tao_view_ls_residual - prints least-squares residual vector at each iteration
363147403d9SBarry Smith . -tao_view_stepdirection - prints step direction vector at each iteration
364a7e14dcfSSatish Balay . -tao_view_gradient - prints gradient vector at each iteration
365a7e14dcfSSatish Balay . -tao_draw_solution - graphically view solution vector at each iteration
366a7e14dcfSSatish Balay . -tao_draw_step - graphically view step vector at each iteration
367a7e14dcfSSatish Balay . -tao_draw_gradient - graphically view gradient at each iteration
368a7e14dcfSSatish Balay . -tao_fd_gradient - use gradient computed with finite differences
369f4c1ad5cSStefano Zampini . -tao_fd_hessian - use hessian computed with finite differences
370f4c1ad5cSStefano Zampini . -tao_mf_hessian - use matrix-free hessian computed with finite differences
371a7e14dcfSSatish Balay . -tao_cancelmonitors - cancels all monitors (except those set with command line)
372441846f8SBarry Smith . -tao_view - prints information about the Tao after solving
373a7e14dcfSSatish Balay - -tao_converged_reason - prints the reason TAO stopped iterating
374a7e14dcfSSatish Balay 
375a7e14dcfSSatish Balay   Notes:
376a7e14dcfSSatish Balay   To see all options, run your program with the -help option or consult the
377a7e14dcfSSatish Balay   user's manual. Should be called after TaoCreate() but before TaoSolve()
378a7e14dcfSSatish Balay 
379a7e14dcfSSatish Balay   Level: beginner
380a7e14dcfSSatish Balay @*/
381441846f8SBarry Smith PetscErrorCode TaoSetFromOptions(Tao tao)
382a7e14dcfSSatish Balay {
383b625d6c7SJed Brown   TaoType        default_type = TAOLMVM;
384a7e14dcfSSatish Balay   char           type[256], monfilename[PETSC_MAX_PATH_LEN];
385a7e14dcfSSatish Balay   PetscViewer    monviewer;
386a7e14dcfSSatish Balay   PetscBool      flg;
387a7e14dcfSSatish Balay   MPI_Comm       comm;
388a7e14dcfSSatish Balay 
389a7e14dcfSSatish Balay   PetscFunctionBegin;
390441846f8SBarry Smith   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
3919566063dSJacob Faibussowitsch   PetscCall(PetscObjectGetComm((PetscObject)tao,&comm));
392125ddc8aSJason Sarich 
393a7e14dcfSSatish Balay   /* So no warnings are given about unused options */
3949566063dSJacob Faibussowitsch   PetscCall(PetscOptionsHasName(((PetscObject)tao)->options,((PetscObject)tao)->prefix,"-tao_ls_type",&flg));
395a7e14dcfSSatish Balay 
396d0609cedSBarry Smith   PetscObjectOptionsBegin((PetscObject)tao);
397a7e14dcfSSatish Balay   {
398a82e8c82SStefano Zampini     if (((PetscObject)tao)->type_name) default_type = ((PetscObject)tao)->type_name;
399a7e14dcfSSatish Balay     /* Check for type from options */
4009566063dSJacob Faibussowitsch     PetscCall(PetscOptionsFList("-tao_type","Tao Solver type","TaoSetType",TaoList,default_type,type,256,&flg));
401a7e14dcfSSatish Balay     if (flg) {
4029566063dSJacob Faibussowitsch       PetscCall(TaoSetType(tao,type));
403a7e14dcfSSatish Balay     } else if (!((PetscObject)tao)->type_name) {
4049566063dSJacob Faibussowitsch       PetscCall(TaoSetType(tao,default_type));
405a7e14dcfSSatish Balay     }
406a7e14dcfSSatish Balay 
4079566063dSJacob Faibussowitsch     PetscCall(PetscOptionsReal("-tao_catol","Stop if constraints violations within","TaoSetConstraintTolerances",tao->catol,&tao->catol,&flg));
4086552cf8aSJason Sarich     if (flg) tao->catol_changed = PETSC_TRUE;
4096aad120cSJose E. Roman     PetscCall(PetscOptionsReal("-tao_crtol","Stop if relative constraint violations within","TaoSetConstraintTolerances",tao->crtol,&tao->crtol,&flg));
4106552cf8aSJason Sarich     if (flg) tao->crtol_changed = PETSC_TRUE;
4119566063dSJacob Faibussowitsch     PetscCall(PetscOptionsReal("-tao_gatol","Stop if norm of gradient less than","TaoSetTolerances",tao->gatol,&tao->gatol,&flg));
4126552cf8aSJason Sarich     if (flg) tao->gatol_changed = PETSC_TRUE;
4139566063dSJacob Faibussowitsch     PetscCall(PetscOptionsReal("-tao_grtol","Stop if norm of gradient divided by the function value is less than","TaoSetTolerances",tao->grtol,&tao->grtol,&flg));
4146552cf8aSJason Sarich     if (flg) tao->grtol_changed = PETSC_TRUE;
4159566063dSJacob 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));
4166552cf8aSJason Sarich     if (flg) tao->gttol_changed = PETSC_TRUE;
4179566063dSJacob Faibussowitsch     PetscCall(PetscOptionsInt("-tao_max_it","Stop if iteration number exceeds","TaoSetMaximumIterations",tao->max_it,&tao->max_it,&flg));
4186552cf8aSJason Sarich     if (flg) tao->max_it_changed = PETSC_TRUE;
4199566063dSJacob Faibussowitsch     PetscCall(PetscOptionsInt("-tao_max_funcs","Stop if number of function evaluations exceeds","TaoSetMaximumFunctionEvaluations",tao->max_funcs,&tao->max_funcs,&flg));
4206552cf8aSJason Sarich     if (flg) tao->max_funcs_changed = PETSC_TRUE;
4219566063dSJacob Faibussowitsch     PetscCall(PetscOptionsReal("-tao_fmin","Stop if function less than","TaoSetFunctionLowerBound",tao->fmin,&tao->fmin,&flg));
4226552cf8aSJason Sarich     if (flg) tao->fmin_changed = PETSC_TRUE;
4239566063dSJacob Faibussowitsch     PetscCall(PetscOptionsReal("-tao_steptol","Stop if step size or trust region radius less than","",tao->steptol,&tao->steptol,&flg));
4246552cf8aSJason Sarich     if (flg) tao->steptol_changed = PETSC_TRUE;
4259566063dSJacob Faibussowitsch     PetscCall(PetscOptionsReal("-tao_trust0","Initial trust region radius","TaoSetTrustRegionRadius",tao->trust0,&tao->trust0,&flg));
4266552cf8aSJason Sarich     if (flg) tao->trust0_changed = PETSC_TRUE;
4279566063dSJacob Faibussowitsch     PetscCall(PetscOptionsString("-tao_view_solution","view solution vector after each evaluation","TaoSetMonitor","stdout",monfilename,sizeof(monfilename),&flg));
428a7e14dcfSSatish Balay     if (flg) {
4299566063dSJacob Faibussowitsch       PetscCall(PetscViewerASCIIOpen(comm,monfilename,&monviewer));
4309566063dSJacob Faibussowitsch       PetscCall(TaoSetMonitor(tao,TaoSolutionMonitor,monviewer,(PetscErrorCode (*)(void**))PetscViewerDestroy));
431a7e14dcfSSatish Balay     }
432a7e14dcfSSatish Balay 
4339566063dSJacob Faibussowitsch     PetscCall(PetscOptionsBool("-tao_converged_reason","Print reason for TAO converged","TaoSolve",tao->printreason,&tao->printreason,NULL));
4349566063dSJacob Faibussowitsch     PetscCall(PetscOptionsString("-tao_view_gradient","view gradient vector after each evaluation","TaoSetMonitor","stdout",monfilename,sizeof(monfilename),&flg));
435a7e14dcfSSatish Balay     if (flg) {
4369566063dSJacob Faibussowitsch       PetscCall(PetscViewerASCIIOpen(comm,monfilename,&monviewer));
4379566063dSJacob Faibussowitsch       PetscCall(TaoSetMonitor(tao,TaoGradientMonitor,monviewer,(PetscErrorCode (*)(void**))PetscViewerDestroy));
438a7e14dcfSSatish Balay     }
439a7e14dcfSSatish Balay 
4409566063dSJacob Faibussowitsch     PetscCall(PetscOptionsString("-tao_view_stepdirection","view step direction vector after each iteration","TaoSetMonitor","stdout",monfilename,sizeof(monfilename),&flg));
441a7e14dcfSSatish Balay     if (flg) {
4429566063dSJacob Faibussowitsch       PetscCall(PetscViewerASCIIOpen(comm,monfilename,&monviewer));
4439566063dSJacob Faibussowitsch       PetscCall(TaoSetMonitor(tao,TaoStepDirectionMonitor,monviewer,(PetscErrorCode (*)(void**))PetscViewerDestroy));
444a7e14dcfSSatish Balay     }
445a7e14dcfSSatish Balay 
4469566063dSJacob Faibussowitsch     PetscCall(PetscOptionsString("-tao_view_residual","view least-squares residual vector after each evaluation","TaoSetMonitor","stdout",monfilename,sizeof(monfilename),&flg));
447a7e14dcfSSatish Balay     if (flg) {
4489566063dSJacob Faibussowitsch       PetscCall(PetscViewerASCIIOpen(comm,monfilename,&monviewer));
4499566063dSJacob Faibussowitsch       PetscCall(TaoSetMonitor(tao,TaoResidualMonitor,monviewer,(PetscErrorCode (*)(void**))PetscViewerDestroy));
450a7e14dcfSSatish Balay     }
451a7e14dcfSSatish Balay 
4529566063dSJacob Faibussowitsch     PetscCall(PetscOptionsString("-tao_monitor","Use the default convergence monitor","TaoSetMonitor","stdout",monfilename,sizeof(monfilename),&flg));
453a7e14dcfSSatish Balay     if (flg) {
4549566063dSJacob Faibussowitsch       PetscCall(PetscViewerASCIIOpen(comm,monfilename,&monviewer));
4559566063dSJacob Faibussowitsch       PetscCall(TaoSetMonitor(tao,TaoMonitorDefault,monviewer,(PetscErrorCode (*)(void**))PetscViewerDestroy));
456a7e14dcfSSatish Balay     }
457a7e14dcfSSatish Balay 
4589566063dSJacob Faibussowitsch     PetscCall(PetscOptionsString("-tao_gmonitor","Use the convergence monitor with extra globalization info","TaoSetMonitor","stdout",monfilename,sizeof(monfilename),&flg));
4598d5ead36SAlp Dener     if (flg) {
4609566063dSJacob Faibussowitsch       PetscCall(PetscViewerASCIIOpen(comm,monfilename,&monviewer));
4619566063dSJacob Faibussowitsch       PetscCall(TaoSetMonitor(tao,TaoDefaultGMonitor,monviewer,(PetscErrorCode (*)(void**))PetscViewerDestroy));
4628d5ead36SAlp Dener     }
4638d5ead36SAlp Dener 
4649566063dSJacob Faibussowitsch     PetscCall(PetscOptionsString("-tao_smonitor","Use the short convergence monitor","TaoSetMonitor","stdout",monfilename,sizeof(monfilename),&flg));
465a7e14dcfSSatish Balay     if (flg) {
4669566063dSJacob Faibussowitsch       PetscCall(PetscViewerASCIIOpen(comm,monfilename,&monviewer));
4679566063dSJacob Faibussowitsch       PetscCall(TaoSetMonitor(tao,TaoDefaultSMonitor,monviewer,(PetscErrorCode (*)(void**))PetscViewerDestroy));
468a7e14dcfSSatish Balay     }
469a7e14dcfSSatish Balay 
4709566063dSJacob Faibussowitsch     PetscCall(PetscOptionsString("-tao_cmonitor","Use the default convergence monitor with constraint norm","TaoSetMonitor","stdout",monfilename,sizeof(monfilename),&flg));
471a7e14dcfSSatish Balay     if (flg) {
4729566063dSJacob Faibussowitsch       PetscCall(PetscViewerASCIIOpen(comm,monfilename,&monviewer));
4739566063dSJacob Faibussowitsch       PetscCall(TaoSetMonitor(tao,TaoDefaultCMonitor,monviewer,(PetscErrorCode (*)(void**))PetscViewerDestroy));
474a7e14dcfSSatish Balay     }
475a7e14dcfSSatish Balay 
4768afaa268SBarry Smith     flg = PETSC_FALSE;
4779566063dSJacob Faibussowitsch     PetscCall(PetscOptionsBool("-tao_cancelmonitors","cancel all monitors and call any registered destroy routines","TaoCancelMonitors",flg,&flg,NULL));
4789566063dSJacob Faibussowitsch     if (flg) PetscCall(TaoCancelMonitors(tao));
479a7e14dcfSSatish Balay 
4808afaa268SBarry Smith     flg = PETSC_FALSE;
4819566063dSJacob Faibussowitsch     PetscCall(PetscOptionsBool("-tao_draw_solution","Plot solution vector at each iteration","TaoSetMonitor",flg,&flg,NULL));
482a7e14dcfSSatish Balay     if (flg) {
483e882e171SHong Zhang       TaoMonitorDrawCtx drawctx;
484e882e171SHong Zhang       PetscInt          howoften = 1;
4859566063dSJacob Faibussowitsch       PetscCall(TaoMonitorDrawCtxCreate(PetscObjectComm((PetscObject)tao),NULL,NULL,PETSC_DECIDE,PETSC_DECIDE,300,300,howoften,&drawctx));
4869566063dSJacob Faibussowitsch       PetscCall(TaoSetMonitor(tao,TaoDrawSolutionMonitor,drawctx,(PetscErrorCode (*)(void**))TaoMonitorDrawCtxDestroy));
487a7e14dcfSSatish Balay     }
488a7e14dcfSSatish Balay 
4898afaa268SBarry Smith     flg = PETSC_FALSE;
4909566063dSJacob Faibussowitsch     PetscCall(PetscOptionsBool("-tao_draw_step","plots step direction at each iteration","TaoSetMonitor",flg,&flg,NULL));
491*1baa6e33SBarry Smith     if (flg) PetscCall(TaoSetMonitor(tao,TaoDrawStepMonitor,NULL,NULL));
492a7e14dcfSSatish Balay 
4938afaa268SBarry Smith     flg = PETSC_FALSE;
4949566063dSJacob Faibussowitsch     PetscCall(PetscOptionsBool("-tao_draw_gradient","plots gradient at each iteration","TaoSetMonitor",flg,&flg,NULL));
495a7e14dcfSSatish Balay     if (flg) {
496e882e171SHong Zhang       TaoMonitorDrawCtx drawctx;
497e882e171SHong Zhang       PetscInt          howoften = 1;
4989566063dSJacob Faibussowitsch       PetscCall(TaoMonitorDrawCtxCreate(PetscObjectComm((PetscObject)tao),NULL,NULL,PETSC_DECIDE,PETSC_DECIDE,300,300,howoften,&drawctx));
4999566063dSJacob Faibussowitsch       PetscCall(TaoSetMonitor(tao,TaoDrawGradientMonitor,drawctx,(PetscErrorCode (*)(void**))TaoMonitorDrawCtxDestroy));
500a7e14dcfSSatish Balay     }
5018afaa268SBarry Smith     flg = PETSC_FALSE;
5029566063dSJacob Faibussowitsch     PetscCall(PetscOptionsBool("-tao_fd_gradient","compute gradient using finite differences","TaoDefaultComputeGradient",flg,&flg,NULL));
503*1baa6e33SBarry Smith     if (flg) PetscCall(TaoSetGradient(tao,NULL,TaoDefaultComputeGradient,NULL));
504f4c1ad5cSStefano Zampini     flg = PETSC_FALSE;
5059566063dSJacob Faibussowitsch     PetscCall(PetscOptionsBool("-tao_fd_hessian","compute hessian using finite differences","TaoDefaultComputeHessian",flg,&flg,NULL));
506f4c1ad5cSStefano Zampini     if (flg) {
507f4c1ad5cSStefano Zampini       Mat H;
508f4c1ad5cSStefano Zampini 
5099566063dSJacob Faibussowitsch       PetscCall(MatCreate(PetscObjectComm((PetscObject)tao),&H));
5109566063dSJacob Faibussowitsch       PetscCall(MatSetType(H,MATAIJ));
5119566063dSJacob Faibussowitsch       PetscCall(TaoSetHessian(tao,H,H,TaoDefaultComputeHessian,NULL));
5129566063dSJacob Faibussowitsch       PetscCall(MatDestroy(&H));
513f4c1ad5cSStefano Zampini     }
514f4c1ad5cSStefano Zampini     flg = PETSC_FALSE;
5159566063dSJacob Faibussowitsch     PetscCall(PetscOptionsBool("-tao_mf_hessian","compute matrix-free hessian using finite differences","TaoDefaultComputeHessianMFFD",flg,&flg,NULL));
516f4c1ad5cSStefano Zampini     if (flg) {
517f4c1ad5cSStefano Zampini       Mat H;
518f4c1ad5cSStefano Zampini 
5199566063dSJacob Faibussowitsch       PetscCall(MatCreate(PetscObjectComm((PetscObject)tao),&H));
5209566063dSJacob Faibussowitsch       PetscCall(TaoSetHessian(tao,H,H,TaoDefaultComputeHessianMFFD,NULL));
5219566063dSJacob Faibussowitsch       PetscCall(MatDestroy(&H));
522f4c1ad5cSStefano Zampini     }
523414d97d3SAlp Dener     flg = PETSC_FALSE;
5249566063dSJacob Faibussowitsch     PetscCall(PetscOptionsBool("-tao_recycle_history","enable recycling/re-using information from the previous TaoSolve() call for some algorithms","TaoSetRecycleHistory",flg,&flg,NULL));
525*1baa6e33SBarry Smith     if (flg) PetscCall(TaoSetRecycleHistory(tao,PETSC_TRUE));
5269566063dSJacob Faibussowitsch     PetscCall(PetscOptionsEnum("-tao_subset_type","subset type","",TaoSubSetTypes,(PetscEnum)tao->subset_type,(PetscEnum*)&tao->subset_type,NULL));
527a7e14dcfSSatish Balay 
5280f0abf79SStefano Zampini     if (tao->ksp) {
5290f0abf79SStefano Zampini       PetscCall(PetscOptionsBool("-tao_ksp_ew","Use Eisentat-Walker linear system convergence test","TaoKSPSetUseEW",tao->ksp_ewconv,&tao->ksp_ewconv,NULL));
5300f0abf79SStefano Zampini       PetscCall(TaoKSPSetUseEW(tao,tao->ksp_ewconv));
5310f0abf79SStefano Zampini     }
5320f0abf79SStefano Zampini 
533*1baa6e33SBarry Smith     if (tao->linesearch) PetscCall(TaoLineSearchSetFromOptions(tao->linesearch));
534a82e8c82SStefano Zampini 
535*1baa6e33SBarry Smith     if (tao->ops->setfromoptions) PetscCall((*tao->ops->setfromoptions)(PetscOptionsObject,tao));
536a7e14dcfSSatish Balay   }
537d0609cedSBarry Smith   PetscOptionsEnd();
538a7e14dcfSSatish Balay   PetscFunctionReturn(0);
539a7e14dcfSSatish Balay }
540a7e14dcfSSatish Balay 
541a7e14dcfSSatish Balay /*@C
542fe2efc57SMark    TaoViewFromOptions - View from Options
543fe2efc57SMark 
544fe2efc57SMark    Collective on Tao
545fe2efc57SMark 
546fe2efc57SMark    Input Parameters:
547fe2efc57SMark +  A - the  Tao context
548736c3998SJose E. Roman .  obj - Optional object
549736c3998SJose E. Roman -  name - command line option
550fe2efc57SMark 
551fe2efc57SMark    Level: intermediate
552db781477SPatrick Sanan .seealso: `Tao`, `TaoView`, `PetscObjectViewFromOptions()`, `TaoCreate()`
553fe2efc57SMark @*/
554fe2efc57SMark PetscErrorCode  TaoViewFromOptions(Tao A,PetscObject obj,const char name[])
555fe2efc57SMark {
556fe2efc57SMark   PetscFunctionBegin;
557fe2efc57SMark   PetscValidHeaderSpecific(A,TAO_CLASSID,1);
5589566063dSJacob Faibussowitsch   PetscCall(PetscObjectViewFromOptions((PetscObject)A,obj,name));
559fe2efc57SMark   PetscFunctionReturn(0);
560fe2efc57SMark }
561fe2efc57SMark 
562fe2efc57SMark /*@C
563441846f8SBarry Smith   TaoView - Prints information about the Tao
564a7e14dcfSSatish Balay 
565441846f8SBarry Smith   Collective on Tao
566a7e14dcfSSatish Balay 
567a7e14dcfSSatish Balay   InputParameters:
568441846f8SBarry Smith + tao - the Tao context
569a7e14dcfSSatish Balay - viewer - visualization context
570a7e14dcfSSatish Balay 
571a7e14dcfSSatish Balay   Options Database Key:
572a7e14dcfSSatish Balay . -tao_view - Calls TaoView() at the end of TaoSolve()
573a7e14dcfSSatish Balay 
574a7e14dcfSSatish Balay   Notes:
575a7e14dcfSSatish Balay   The available visualization contexts include
576a7e14dcfSSatish Balay +     PETSC_VIEWER_STDOUT_SELF - standard output (default)
577a7e14dcfSSatish Balay -     PETSC_VIEWER_STDOUT_WORLD - synchronized standard
578a7e14dcfSSatish Balay          output where only the first processor opens
579a7e14dcfSSatish Balay          the file.  All other processors send their
580a7e14dcfSSatish Balay          data to the first processor to print.
581a7e14dcfSSatish Balay 
582a7e14dcfSSatish Balay   Level: beginner
583a7e14dcfSSatish Balay 
584db781477SPatrick Sanan .seealso: `PetscViewerASCIIOpen()`
585a7e14dcfSSatish Balay @*/
586441846f8SBarry Smith PetscErrorCode TaoView(Tao tao, PetscViewer viewer)
587a7e14dcfSSatish Balay {
588a7e14dcfSSatish Balay   PetscBool           isascii,isstring;
589b625d6c7SJed Brown   TaoType             type;
59047a47007SBarry Smith 
591a7e14dcfSSatish Balay   PetscFunctionBegin;
592441846f8SBarry Smith   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
593a7e14dcfSSatish Balay   if (!viewer) {
5949566063dSJacob Faibussowitsch     PetscCall(PetscViewerASCIIGetStdout(((PetscObject)tao)->comm,&viewer));
595a7e14dcfSSatish Balay   }
596a7e14dcfSSatish Balay   PetscValidHeaderSpecific(viewer,PETSC_VIEWER_CLASSID,2);
597a7e14dcfSSatish Balay   PetscCheckSameComm(tao,1,viewer,2);
598a7e14dcfSSatish Balay 
5999566063dSJacob Faibussowitsch   PetscCall(PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERASCII,&isascii));
6009566063dSJacob Faibussowitsch   PetscCall(PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERSTRING,&isstring));
601a7e14dcfSSatish Balay   if (isascii) {
6029566063dSJacob Faibussowitsch     PetscCall(PetscObjectPrintClassNamePrefixType((PetscObject)tao,viewer));
603a7e14dcfSSatish Balay 
604a7e14dcfSSatish Balay     if (tao->ops->view) {
6059566063dSJacob Faibussowitsch       PetscCall(PetscViewerASCIIPushTab(viewer));
6069566063dSJacob Faibussowitsch       PetscCall((*tao->ops->view)(tao,viewer));
6079566063dSJacob Faibussowitsch       PetscCall(PetscViewerASCIIPopTab(viewer));
608a7e14dcfSSatish Balay     }
6098d3f3ddaSAlp Dener     if (tao->linesearch) {
6109566063dSJacob Faibussowitsch       PetscCall(PetscViewerASCIIPushTab(viewer));
6119566063dSJacob Faibussowitsch       PetscCall(TaoLineSearchView(tao->linesearch,viewer));
6129566063dSJacob Faibussowitsch       PetscCall(PetscViewerASCIIPopTab(viewer));
6138d3f3ddaSAlp Dener     }
614a7e14dcfSSatish Balay     if (tao->ksp) {
6159566063dSJacob Faibussowitsch       PetscCall(PetscViewerASCIIPushTab(viewer));
6169566063dSJacob Faibussowitsch       PetscCall(KSPView(tao->ksp,viewer));
61763a3b9bcSJacob Faibussowitsch       PetscCall(PetscViewerASCIIPrintf(viewer,"total KSP iterations: %" PetscInt_FMT "\n",tao->ksp_tot_its));
6189566063dSJacob Faibussowitsch       PetscCall(PetscViewerASCIIPopTab(viewer));
619a7e14dcfSSatish Balay     }
62019b5c101SAlp Dener 
6219566063dSJacob Faibussowitsch     PetscCall(PetscViewerASCIIPushTab(viewer));
62219b5c101SAlp Dener 
623a7e14dcfSSatish Balay     if (tao->XL || tao->XU) {
6249566063dSJacob Faibussowitsch       PetscCall(PetscViewerASCIIPrintf(viewer,"Active Set subset type: %s\n",TaoSubSetTypes[tao->subset_type]));
625a7e14dcfSSatish Balay     }
626a7e14dcfSSatish Balay 
6279566063dSJacob Faibussowitsch     PetscCall(PetscViewerASCIIPrintf(viewer,"convergence tolerances: gatol=%g,",(double)tao->gatol));
6289566063dSJacob Faibussowitsch     PetscCall(PetscViewerASCIIPrintf(viewer," steptol=%g,",(double)tao->steptol));
6299566063dSJacob Faibussowitsch     PetscCall(PetscViewerASCIIPrintf(viewer," gttol=%g\n",(double)tao->gttol));
6309566063dSJacob Faibussowitsch     PetscCall(PetscViewerASCIIPrintf(viewer,"Residual in Function/Gradient:=%g\n",(double)tao->residual));
631a7e14dcfSSatish Balay 
6326246e8cdSAlp Dener     if (tao->constrained) {
6339566063dSJacob Faibussowitsch       PetscCall(PetscViewerASCIIPrintf(viewer,"convergence tolerances:"));
6349566063dSJacob Faibussowitsch       PetscCall(PetscViewerASCIIPrintf(viewer," catol=%g,",(double)tao->catol));
6359566063dSJacob Faibussowitsch       PetscCall(PetscViewerASCIIPrintf(viewer," crtol=%g\n",(double)tao->crtol));
6369566063dSJacob Faibussowitsch       PetscCall(PetscViewerASCIIPrintf(viewer,"Residual in Constraints:=%g\n",(double)tao->cnorm));
637a7e14dcfSSatish Balay     }
638a7e14dcfSSatish Balay 
639a7e14dcfSSatish Balay     if (tao->trust < tao->steptol) {
6409566063dSJacob Faibussowitsch       PetscCall(PetscViewerASCIIPrintf(viewer,"convergence tolerances: steptol=%g\n",(double)tao->steptol));
6419566063dSJacob Faibussowitsch       PetscCall(PetscViewerASCIIPrintf(viewer,"Final trust region radius:=%g\n",(double)tao->trust));
642a7e14dcfSSatish Balay     }
643a7e14dcfSSatish Balay 
644a7e14dcfSSatish Balay     if (tao->fmin>-1.e25) {
6459566063dSJacob Faibussowitsch       PetscCall(PetscViewerASCIIPrintf(viewer,"convergence tolerances: function minimum=%g\n",(double)tao->fmin));
646a7e14dcfSSatish Balay     }
6479566063dSJacob Faibussowitsch     PetscCall(PetscViewerASCIIPrintf(viewer,"Objective value=%g\n",(double)tao->fc));
648a7e14dcfSSatish Balay 
64963a3b9bcSJacob Faibussowitsch     PetscCall(PetscViewerASCIIPrintf(viewer,"total number of iterations=%" PetscInt_FMT ",          ",tao->niter));
65063a3b9bcSJacob Faibussowitsch     PetscCall(PetscViewerASCIIPrintf(viewer,"              (max: %" PetscInt_FMT ")\n",tao->max_it));
651a7e14dcfSSatish Balay 
652a7e14dcfSSatish Balay     if (tao->nfuncs>0) {
65363a3b9bcSJacob Faibussowitsch       PetscCall(PetscViewerASCIIPrintf(viewer,"total number of function evaluations=%" PetscInt_FMT ",",tao->nfuncs));
65463a3b9bcSJacob Faibussowitsch       PetscCall(PetscViewerASCIIPrintf(viewer,"                max: %" PetscInt_FMT "\n",tao->max_funcs));
655a7e14dcfSSatish Balay     }
656a7e14dcfSSatish Balay     if (tao->ngrads>0) {
65763a3b9bcSJacob Faibussowitsch       PetscCall(PetscViewerASCIIPrintf(viewer,"total number of gradient evaluations=%" PetscInt_FMT ",",tao->ngrads));
65863a3b9bcSJacob Faibussowitsch       PetscCall(PetscViewerASCIIPrintf(viewer,"                max: %" PetscInt_FMT "\n",tao->max_funcs));
659a7e14dcfSSatish Balay     }
660a7e14dcfSSatish Balay     if (tao->nfuncgrads>0) {
66163a3b9bcSJacob Faibussowitsch       PetscCall(PetscViewerASCIIPrintf(viewer,"total number of function/gradient evaluations=%" PetscInt_FMT ",",tao->nfuncgrads));
66263a3b9bcSJacob Faibussowitsch       PetscCall(PetscViewerASCIIPrintf(viewer,"    (max: %" PetscInt_FMT ")\n",tao->max_funcs));
663a7e14dcfSSatish Balay     }
664a7e14dcfSSatish Balay     if (tao->nhess>0) {
66563a3b9bcSJacob Faibussowitsch       PetscCall(PetscViewerASCIIPrintf(viewer,"total number of Hessian evaluations=%" PetscInt_FMT "\n",tao->nhess));
666a7e14dcfSSatish Balay     }
667a7e14dcfSSatish Balay     if (tao->nconstraints>0) {
66863a3b9bcSJacob Faibussowitsch       PetscCall(PetscViewerASCIIPrintf(viewer,"total number of constraint function evaluations=%" PetscInt_FMT "\n",tao->nconstraints));
669a7e14dcfSSatish Balay     }
670a7e14dcfSSatish Balay     if (tao->njac>0) {
67163a3b9bcSJacob Faibussowitsch       PetscCall(PetscViewerASCIIPrintf(viewer,"total number of Jacobian evaluations=%" PetscInt_FMT "\n",tao->njac));
672a7e14dcfSSatish Balay     }
673a7e14dcfSSatish Balay 
674a7e14dcfSSatish Balay     if (tao->reason>0) {
6759566063dSJacob Faibussowitsch       PetscCall(PetscViewerASCIIPrintf(viewer,    "Solution converged: "));
676a7e14dcfSSatish Balay       switch (tao->reason) {
677a7e14dcfSSatish Balay       case TAO_CONVERGED_GATOL:
6789566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer," ||g(X)|| <= gatol\n"));
679a7e14dcfSSatish Balay         break;
680a7e14dcfSSatish Balay       case TAO_CONVERGED_GRTOL:
6819566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer," ||g(X)||/|f(X)| <= grtol\n"));
682a7e14dcfSSatish Balay         break;
683a7e14dcfSSatish Balay       case TAO_CONVERGED_GTTOL:
6849566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer," ||g(X)||/||g(X0)|| <= gttol\n"));
685a7e14dcfSSatish Balay         break;
686a7e14dcfSSatish Balay       case TAO_CONVERGED_STEPTOL:
6879566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer," Steptol -- step size small\n"));
688a7e14dcfSSatish Balay         break;
689a7e14dcfSSatish Balay       case TAO_CONVERGED_MINF:
6909566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer," Minf --  f < fmin\n"));
691a7e14dcfSSatish Balay         break;
692a7e14dcfSSatish Balay       case TAO_CONVERGED_USER:
6939566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer," User Terminated\n"));
694a7e14dcfSSatish Balay         break;
695a7e14dcfSSatish Balay       default:
6969566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer,"\n"));
697a7e14dcfSSatish Balay         break;
698a7e14dcfSSatish Balay       }
699a7e14dcfSSatish Balay     } else {
7009566063dSJacob Faibussowitsch       PetscCall(PetscViewerASCIIPrintf(viewer,"Solver terminated: %d",tao->reason));
701a7e14dcfSSatish Balay       switch (tao->reason) {
702a7e14dcfSSatish Balay       case TAO_DIVERGED_MAXITS:
7039566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer," Maximum Iterations\n"));
704a7e14dcfSSatish Balay         break;
705a7e14dcfSSatish Balay       case TAO_DIVERGED_NAN:
7069566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer," NAN or Inf encountered\n"));
707a7e14dcfSSatish Balay         break;
708a7e14dcfSSatish Balay       case TAO_DIVERGED_MAXFCN:
7099566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer," Maximum Function Evaluations\n"));
710a7e14dcfSSatish Balay         break;
711a7e14dcfSSatish Balay       case TAO_DIVERGED_LS_FAILURE:
7129566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer," Line Search Failure\n"));
713a7e14dcfSSatish Balay         break;
714a7e14dcfSSatish Balay       case TAO_DIVERGED_TR_REDUCTION:
7159566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer," Trust Region too small\n"));
716a7e14dcfSSatish Balay         break;
717a7e14dcfSSatish Balay       case TAO_DIVERGED_USER:
7189566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer," User Terminated\n"));
719a7e14dcfSSatish Balay         break;
720a7e14dcfSSatish Balay       default:
7219566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer,"\n"));
722a7e14dcfSSatish Balay         break;
723a7e14dcfSSatish Balay       }
724a7e14dcfSSatish Balay     }
7259566063dSJacob Faibussowitsch     PetscCall(PetscViewerASCIIPopTab(viewer));
726a7e14dcfSSatish Balay   } else if (isstring) {
7279566063dSJacob Faibussowitsch     PetscCall(TaoGetType(tao,&type));
7289566063dSJacob Faibussowitsch     PetscCall(PetscViewerStringSPrintf(viewer," %-3.3s",type));
729a7e14dcfSSatish Balay   }
730a7e14dcfSSatish Balay   PetscFunctionReturn(0);
731a7e14dcfSSatish Balay }
732a7e14dcfSSatish Balay 
733a7e14dcfSSatish Balay /*@
734414d97d3SAlp Dener   TaoSetRecycleHistory - Sets the boolean flag to enable/disable re-using
735414d97d3SAlp Dener   iterate information from the previous TaoSolve(). This feature is disabled by
736414d97d3SAlp Dener   default.
737414d97d3SAlp Dener 
738414d97d3SAlp Dener   For conjugate gradient methods (BNCG), this re-uses the latest search direction
739414d97d3SAlp Dener   from the previous TaoSolve() call when computing the first search direction in a
740414d97d3SAlp Dener   new solution. By default, CG methods set the first search direction to the
741414d97d3SAlp Dener   negative gradient.
742414d97d3SAlp Dener 
743414d97d3SAlp Dener   For quasi-Newton family of methods (BQNLS, BQNKLS, BQNKTR, BQNKTL), this re-uses
744414d97d3SAlp Dener   the accumulated quasi-Newton Hessian approximation from the previous TaoSolve()
745414d97d3SAlp Dener   call. By default, QN family of methods reset the initial Hessian approximation to
746414d97d3SAlp Dener   the identity matrix.
747414d97d3SAlp Dener 
748414d97d3SAlp Dener   For any other algorithm, this setting has no effect.
749414d97d3SAlp Dener 
750414d97d3SAlp Dener   Logically collective on Tao
751414d97d3SAlp Dener 
752414d97d3SAlp Dener   Input Parameters:
753414d97d3SAlp Dener + tao - the Tao context
754414d97d3SAlp Dener - recycle - boolean flag
755414d97d3SAlp Dener 
756414d97d3SAlp Dener   Options Database Keys:
757147403d9SBarry Smith . -tao_recycle_history <true,false> - reuse the history
758414d97d3SAlp Dener 
759414d97d3SAlp Dener   Level: intermediate
760414d97d3SAlp Dener 
761db781477SPatrick Sanan .seealso: `TaoSetRecycleHistory()`, `TAOBNCG`, `TAOBQNLS`, `TAOBQNKLS`, `TAOBQNKTR`, `TAOBQNKTL`
762414d97d3SAlp Dener 
763414d97d3SAlp Dener @*/
764414d97d3SAlp Dener PetscErrorCode TaoSetRecycleHistory(Tao tao, PetscBool recycle)
765414d97d3SAlp Dener {
766414d97d3SAlp Dener   PetscFunctionBegin;
767414d97d3SAlp Dener   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
76894511df7SStefano Zampini   PetscValidLogicalCollectiveBool(tao,recycle,2);
769414d97d3SAlp Dener   tao->recycle = recycle;
770414d97d3SAlp Dener   PetscFunctionReturn(0);
771414d97d3SAlp Dener }
772414d97d3SAlp Dener 
773414d97d3SAlp Dener /*@
774414d97d3SAlp Dener   TaoGetRecycleHistory - Retrieve the boolean flag for re-using iterate information
775414d97d3SAlp Dener   from the previous TaoSolve(). This feature is disabled by default.
776414d97d3SAlp Dener 
777414d97d3SAlp Dener   Logically collective on Tao
778414d97d3SAlp Dener 
779414d97d3SAlp Dener   Input Parameters:
780147403d9SBarry Smith . tao - the Tao context
781414d97d3SAlp Dener 
782414d97d3SAlp Dener   Output Parameters:
783147403d9SBarry Smith . recycle - boolean flag
784414d97d3SAlp Dener 
785414d97d3SAlp Dener   Level: intermediate
786414d97d3SAlp Dener 
787db781477SPatrick Sanan .seealso: `TaoGetRecycleHistory()`, `TAOBNCG`, `TAOBQNLS`, `TAOBQNKLS`, `TAOBQNKTR`, `TAOBQNKTL`
788414d97d3SAlp Dener 
789414d97d3SAlp Dener @*/
790414d97d3SAlp Dener PetscErrorCode TaoGetRecycleHistory(Tao tao, PetscBool *recycle)
791414d97d3SAlp Dener {
792414d97d3SAlp Dener   PetscFunctionBegin;
793414d97d3SAlp Dener   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
79494511df7SStefano Zampini   PetscValidBoolPointer(recycle,2);
795414d97d3SAlp Dener   *recycle = tao->recycle;
796414d97d3SAlp Dener   PetscFunctionReturn(0);
797414d97d3SAlp Dener }
798414d97d3SAlp Dener 
799414d97d3SAlp Dener /*@
800a7e14dcfSSatish Balay   TaoSetTolerances - Sets parameters used in TAO convergence tests
801a7e14dcfSSatish Balay 
802441846f8SBarry Smith   Logically collective on Tao
803a7e14dcfSSatish Balay 
804a7e14dcfSSatish Balay   Input Parameters:
805441846f8SBarry Smith + tao - the Tao context
806a7e14dcfSSatish Balay . gatol - stop if norm of gradient is less than this
807a7e14dcfSSatish Balay . grtol - stop if relative norm of gradient is less than this
808a7e14dcfSSatish Balay - gttol - stop if norm of gradient is reduced by this factor
809a7e14dcfSSatish Balay 
810a7e14dcfSSatish Balay   Options Database Keys:
811e52336cbSBarry Smith + -tao_gatol <gatol> - Sets gatol
812a7e14dcfSSatish Balay . -tao_grtol <grtol> - Sets grtol
813a7e14dcfSSatish Balay - -tao_gttol <gttol> - Sets gttol
814a7e14dcfSSatish Balay 
815a7e14dcfSSatish Balay   Stopping Criteria:
816a7e14dcfSSatish Balay $ ||g(X)||                            <= gatol
817a7e14dcfSSatish Balay $ ||g(X)|| / |f(X)|                   <= grtol
818a7e14dcfSSatish Balay $ ||g(X)|| / ||g(X0)||                <= gttol
819a7e14dcfSSatish Balay 
820a7e14dcfSSatish Balay   Notes:
821a7e14dcfSSatish Balay   Use PETSC_DEFAULT to leave one or more tolerances unchanged.
822a7e14dcfSSatish Balay 
823a7e14dcfSSatish Balay   Level: beginner
824a7e14dcfSSatish Balay 
825db781477SPatrick Sanan .seealso: `TaoGetTolerances()`
826a7e14dcfSSatish Balay 
827a7e14dcfSSatish Balay @*/
828e52336cbSBarry Smith PetscErrorCode TaoSetTolerances(Tao tao, PetscReal gatol, PetscReal grtol, PetscReal gttol)
829a7e14dcfSSatish Balay {
830a7e14dcfSSatish Balay   PetscFunctionBegin;
831441846f8SBarry Smith   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
83294511df7SStefano Zampini   PetscValidLogicalCollectiveReal(tao,gatol,2);
83394511df7SStefano Zampini   PetscValidLogicalCollectiveReal(tao,grtol,3);
83494511df7SStefano Zampini   PetscValidLogicalCollectiveReal(tao,gttol,4);
835a7e14dcfSSatish Balay 
836a7e14dcfSSatish Balay   if (gatol != PETSC_DEFAULT) {
837a7e14dcfSSatish Balay     if (gatol<0) {
8389566063dSJacob Faibussowitsch       PetscCall(PetscInfo(tao,"Tried to set negative gatol -- ignored.\n"));
839a7e14dcfSSatish Balay     } else {
840a7e14dcfSSatish Balay       tao->gatol = PetscMax(0,gatol);
8416552cf8aSJason Sarich       tao->gatol_changed = PETSC_TRUE;
842a7e14dcfSSatish Balay     }
843a7e14dcfSSatish Balay   }
844a7e14dcfSSatish Balay 
845a7e14dcfSSatish Balay   if (grtol != PETSC_DEFAULT) {
846a7e14dcfSSatish Balay     if (grtol<0) {
8479566063dSJacob Faibussowitsch       PetscCall(PetscInfo(tao,"Tried to set negative grtol -- ignored.\n"));
848a7e14dcfSSatish Balay     } else {
849a7e14dcfSSatish Balay       tao->grtol = PetscMax(0,grtol);
8506552cf8aSJason Sarich       tao->grtol_changed = PETSC_TRUE;
851a7e14dcfSSatish Balay     }
852a7e14dcfSSatish Balay   }
853a7e14dcfSSatish Balay 
854a7e14dcfSSatish Balay   if (gttol != PETSC_DEFAULT) {
855a7e14dcfSSatish Balay     if (gttol<0) {
8569566063dSJacob Faibussowitsch       PetscCall(PetscInfo(tao,"Tried to set negative gttol -- ignored.\n"));
857a7e14dcfSSatish Balay     } else {
858a7e14dcfSSatish Balay       tao->gttol = PetscMax(0,gttol);
8596552cf8aSJason Sarich       tao->gttol_changed = PETSC_TRUE;
860a7e14dcfSSatish Balay     }
861a7e14dcfSSatish Balay   }
862a7e14dcfSSatish Balay   PetscFunctionReturn(0);
863a7e14dcfSSatish Balay }
864a7e14dcfSSatish Balay 
865a7e14dcfSSatish Balay /*@
86658417fe7SBarry Smith   TaoSetConstraintTolerances - Sets constraint tolerance parameters used in TAO  convergence tests
867a7e14dcfSSatish Balay 
868441846f8SBarry Smith   Logically collective on Tao
869a7e14dcfSSatish Balay 
870a7e14dcfSSatish Balay   Input Parameters:
871441846f8SBarry Smith + tao - the Tao context
872e52336cbSBarry Smith . catol - absolute constraint tolerance, constraint norm must be less than catol for used for gatol convergence criteria
8736aad120cSJose E. Roman - crtol - relative constraint tolerance, constraint norm must be less than crtol for used for gatol, gttol convergence criteria
874a7e14dcfSSatish Balay 
875a7e14dcfSSatish Balay   Options Database Keys:
876a7e14dcfSSatish Balay + -tao_catol <catol> - Sets catol
877a7e14dcfSSatish Balay - -tao_crtol <crtol> - Sets crtol
878a7e14dcfSSatish Balay 
8796552cf8aSJason Sarich   Notes:
8806552cf8aSJason Sarich   Use PETSC_DEFAULT to leave any tolerance unchanged.
8816552cf8aSJason Sarich 
882a7e14dcfSSatish Balay   Level: intermediate
883a7e14dcfSSatish Balay 
884db781477SPatrick Sanan .seealso: `TaoGetTolerances()`, `TaoGetConstraintTolerances()`, `TaoSetTolerances()`
885a7e14dcfSSatish Balay 
886a7e14dcfSSatish Balay @*/
887441846f8SBarry Smith PetscErrorCode TaoSetConstraintTolerances(Tao tao, PetscReal catol, PetscReal crtol)
888a7e14dcfSSatish Balay {
889a7e14dcfSSatish Balay   PetscFunctionBegin;
890441846f8SBarry Smith   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
89194511df7SStefano Zampini   PetscValidLogicalCollectiveReal(tao,catol,2);
89294511df7SStefano Zampini   PetscValidLogicalCollectiveReal(tao,crtol,3);
893a7e14dcfSSatish Balay 
894a7e14dcfSSatish Balay   if (catol != PETSC_DEFAULT) {
895a7e14dcfSSatish Balay     if (catol<0) {
8969566063dSJacob Faibussowitsch       PetscCall(PetscInfo(tao,"Tried to set negative catol -- ignored.\n"));
897a7e14dcfSSatish Balay     } else {
898a7e14dcfSSatish Balay       tao->catol = PetscMax(0,catol);
8996552cf8aSJason Sarich       tao->catol_changed = PETSC_TRUE;
900a7e14dcfSSatish Balay     }
901a7e14dcfSSatish Balay   }
902a7e14dcfSSatish Balay 
903a7e14dcfSSatish Balay   if (crtol != PETSC_DEFAULT) {
904a7e14dcfSSatish Balay     if (crtol<0) {
9059566063dSJacob Faibussowitsch       PetscCall(PetscInfo(tao,"Tried to set negative crtol -- ignored.\n"));
906a7e14dcfSSatish Balay     } else {
907a7e14dcfSSatish Balay       tao->crtol = PetscMax(0,crtol);
9086552cf8aSJason Sarich       tao->crtol_changed = PETSC_TRUE;
909a7e14dcfSSatish Balay     }
910a7e14dcfSSatish Balay   }
911a7e14dcfSSatish Balay   PetscFunctionReturn(0);
912a7e14dcfSSatish Balay }
913a7e14dcfSSatish Balay 
91458417fe7SBarry Smith /*@
91558417fe7SBarry Smith   TaoGetConstraintTolerances - Gets constraint tolerance parameters used in TAO  convergence tests
91658417fe7SBarry Smith 
91758417fe7SBarry Smith   Not ollective
91858417fe7SBarry Smith 
91958417fe7SBarry Smith   Input Parameter:
92058417fe7SBarry Smith . tao - the Tao context
92158417fe7SBarry Smith 
922d8d19677SJose E. Roman   Output Parameters:
923e52336cbSBarry Smith + catol - absolute constraint tolerance, constraint norm must be less than catol for used for gatol convergence criteria
9246aad120cSJose E. Roman - crtol - relative constraint tolerance, constraint norm must be less than crtol for used for gatol, gttol convergence criteria
92558417fe7SBarry Smith 
92658417fe7SBarry Smith   Level: intermediate
92758417fe7SBarry Smith 
928db781477SPatrick Sanan .seealso: `TaoGetTolerances()`, `TaoSetTolerances()`, `TaoSetConstraintTolerances()`
92958417fe7SBarry Smith 
93058417fe7SBarry Smith @*/
93158417fe7SBarry Smith PetscErrorCode TaoGetConstraintTolerances(Tao tao, PetscReal *catol, PetscReal *crtol)
93258417fe7SBarry Smith {
93358417fe7SBarry Smith   PetscFunctionBegin;
93458417fe7SBarry Smith   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
93558417fe7SBarry Smith   if (catol) *catol = tao->catol;
93658417fe7SBarry Smith   if (crtol) *crtol = tao->crtol;
93758417fe7SBarry Smith   PetscFunctionReturn(0);
93858417fe7SBarry Smith }
93958417fe7SBarry Smith 
940a7e14dcfSSatish Balay /*@
941a7e14dcfSSatish Balay    TaoSetFunctionLowerBound - Sets a bound on the solution objective value.
942a7e14dcfSSatish Balay    When an approximate solution with an objective value below this number
943a7e14dcfSSatish Balay    has been found, the solver will terminate.
944a7e14dcfSSatish Balay 
945441846f8SBarry Smith    Logically Collective on Tao
946a7e14dcfSSatish Balay 
947a7e14dcfSSatish Balay    Input Parameters:
948441846f8SBarry Smith +  tao - the Tao solver context
949a7e14dcfSSatish Balay -  fmin - the tolerance
950a7e14dcfSSatish Balay 
951a7e14dcfSSatish Balay    Options Database Keys:
952a7e14dcfSSatish Balay .    -tao_fmin <fmin> - sets the minimum function value
953a7e14dcfSSatish Balay 
954a7e14dcfSSatish Balay    Level: intermediate
955a7e14dcfSSatish Balay 
956db781477SPatrick Sanan .seealso: `TaoSetTolerances()`
957a7e14dcfSSatish Balay @*/
958441846f8SBarry Smith PetscErrorCode TaoSetFunctionLowerBound(Tao tao,PetscReal fmin)
959a7e14dcfSSatish Balay {
960a7e14dcfSSatish Balay   PetscFunctionBegin;
961441846f8SBarry Smith   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
96294511df7SStefano Zampini   PetscValidLogicalCollectiveReal(tao,fmin,2);
963a7e14dcfSSatish Balay   tao->fmin = fmin;
9646552cf8aSJason Sarich   tao->fmin_changed = PETSC_TRUE;
965a7e14dcfSSatish Balay   PetscFunctionReturn(0);
966a7e14dcfSSatish Balay }
967a7e14dcfSSatish Balay 
968a7e14dcfSSatish Balay /*@
9698e96d397SJason Sarich    TaoGetFunctionLowerBound - Gets the bound on the solution objective value.
970a7e14dcfSSatish Balay    When an approximate solution with an objective value below this number
971a7e14dcfSSatish Balay    has been found, the solver will terminate.
972a7e14dcfSSatish Balay 
973441846f8SBarry Smith    Not collective on Tao
974a7e14dcfSSatish Balay 
975a7e14dcfSSatish Balay    Input Parameters:
976441846f8SBarry Smith .  tao - the Tao solver context
977a7e14dcfSSatish Balay 
978a7e14dcfSSatish Balay    OutputParameters:
979a7e14dcfSSatish Balay .  fmin - the minimum function value
980a7e14dcfSSatish Balay 
981a7e14dcfSSatish Balay    Level: intermediate
982a7e14dcfSSatish Balay 
983db781477SPatrick Sanan .seealso: `TaoSetFunctionLowerBound()`
984a7e14dcfSSatish Balay @*/
985441846f8SBarry Smith PetscErrorCode TaoGetFunctionLowerBound(Tao tao,PetscReal *fmin)
986a7e14dcfSSatish Balay {
987a7e14dcfSSatish Balay   PetscFunctionBegin;
988441846f8SBarry Smith   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
98994511df7SStefano Zampini   PetscValidRealPointer(fmin,2);
990a7e14dcfSSatish Balay   *fmin = tao->fmin;
991a7e14dcfSSatish Balay   PetscFunctionReturn(0);
992a7e14dcfSSatish Balay }
993a7e14dcfSSatish Balay 
994a7e14dcfSSatish Balay /*@
995a7e14dcfSSatish Balay    TaoSetMaximumFunctionEvaluations - Sets a maximum number of
996a7e14dcfSSatish Balay    function evaluations.
997a7e14dcfSSatish Balay 
998441846f8SBarry Smith    Logically Collective on Tao
999a7e14dcfSSatish Balay 
1000a7e14dcfSSatish Balay    Input Parameters:
1001441846f8SBarry Smith +  tao - the Tao solver context
1002a7e14dcfSSatish Balay -  nfcn - the maximum number of function evaluations (>=0)
1003a7e14dcfSSatish Balay 
1004a7e14dcfSSatish Balay    Options Database Keys:
1005a7e14dcfSSatish Balay .    -tao_max_funcs <nfcn> - sets the maximum number of function evaluations
1006a7e14dcfSSatish Balay 
1007a7e14dcfSSatish Balay    Level: intermediate
1008a7e14dcfSSatish Balay 
1009db781477SPatrick Sanan .seealso: `TaoSetTolerances()`, `TaoSetMaximumIterations()`
1010a7e14dcfSSatish Balay @*/
1011a7e14dcfSSatish Balay 
1012441846f8SBarry Smith PetscErrorCode TaoSetMaximumFunctionEvaluations(Tao tao,PetscInt nfcn)
1013a7e14dcfSSatish Balay {
1014a7e14dcfSSatish Balay   PetscFunctionBegin;
1015441846f8SBarry Smith   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
101694511df7SStefano Zampini   PetscValidLogicalCollectiveInt(tao,nfcn,2);
101794511df7SStefano Zampini   if (nfcn >= 0) { tao->max_funcs = PetscMax(0,nfcn); }
101894511df7SStefano Zampini   else { tao->max_funcs = -1; }
10196552cf8aSJason Sarich   tao->max_funcs_changed = PETSC_TRUE;
1020a7e14dcfSSatish Balay   PetscFunctionReturn(0);
1021a7e14dcfSSatish Balay }
1022a7e14dcfSSatish Balay 
1023a7e14dcfSSatish Balay /*@
1024a7e14dcfSSatish Balay    TaoGetMaximumFunctionEvaluations - Sets a maximum number of
1025a7e14dcfSSatish Balay    function evaluations.
1026a7e14dcfSSatish Balay 
1027a7e14dcfSSatish Balay    Not Collective
1028a7e14dcfSSatish Balay 
1029a7e14dcfSSatish Balay    Input Parameters:
1030441846f8SBarry Smith .  tao - the Tao solver context
1031a7e14dcfSSatish Balay 
1032a7e14dcfSSatish Balay    Output Parameters:
1033a7e14dcfSSatish Balay .  nfcn - the maximum number of function evaluations
1034a7e14dcfSSatish Balay 
1035a7e14dcfSSatish Balay    Level: intermediate
1036a7e14dcfSSatish Balay 
1037db781477SPatrick Sanan .seealso: `TaoSetMaximumFunctionEvaluations()`, `TaoGetMaximumIterations()`
1038a7e14dcfSSatish Balay @*/
1039a7e14dcfSSatish Balay 
1040441846f8SBarry Smith PetscErrorCode TaoGetMaximumFunctionEvaluations(Tao tao,PetscInt *nfcn)
1041a7e14dcfSSatish Balay {
1042a7e14dcfSSatish Balay   PetscFunctionBegin;
1043441846f8SBarry Smith   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
104494511df7SStefano Zampini   PetscValidIntPointer(nfcn,2);
1045a7e14dcfSSatish Balay   *nfcn = tao->max_funcs;
1046a7e14dcfSSatish Balay   PetscFunctionReturn(0);
1047a7e14dcfSSatish Balay }
1048a7e14dcfSSatish Balay 
1049770232b9SCe Qin /*@
1050770232b9SCe Qin    TaoGetCurrentFunctionEvaluations - Get current number of
1051770232b9SCe Qin    function evaluations.
1052770232b9SCe Qin 
1053770232b9SCe Qin    Not Collective
1054770232b9SCe Qin 
1055770232b9SCe Qin    Input Parameters:
1056770232b9SCe Qin .  tao - the Tao solver context
1057770232b9SCe Qin 
1058770232b9SCe Qin    Output Parameters:
105994511df7SStefano Zampini .  nfuncs - the current number of function evaluations (maximum between gradient and function evaluations)
1060770232b9SCe Qin 
1061770232b9SCe Qin    Level: intermediate
1062770232b9SCe Qin 
1063db781477SPatrick Sanan .seealso: `TaoSetMaximumFunctionEvaluations()`, `TaoGetMaximumFunctionEvaluations()`, `TaoGetMaximumIterations()`
1064770232b9SCe Qin @*/
1065770232b9SCe Qin 
1066770232b9SCe Qin PetscErrorCode TaoGetCurrentFunctionEvaluations(Tao tao,PetscInt *nfuncs)
1067770232b9SCe Qin {
1068770232b9SCe Qin   PetscFunctionBegin;
1069770232b9SCe Qin   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
107094511df7SStefano Zampini   PetscValidIntPointer(nfuncs,2);
1071770232b9SCe Qin   *nfuncs = PetscMax(tao->nfuncs,tao->nfuncgrads);
1072770232b9SCe Qin   PetscFunctionReturn(0);
1073770232b9SCe Qin }
1074770232b9SCe Qin 
1075a7e14dcfSSatish Balay /*@
1076a7e14dcfSSatish Balay    TaoSetMaximumIterations - Sets a maximum number of iterates.
1077a7e14dcfSSatish Balay 
1078441846f8SBarry Smith    Logically Collective on Tao
1079a7e14dcfSSatish Balay 
1080a7e14dcfSSatish Balay    Input Parameters:
1081441846f8SBarry Smith +  tao - the Tao solver context
1082a7e14dcfSSatish Balay -  maxits - the maximum number of iterates (>=0)
1083a7e14dcfSSatish Balay 
1084a7e14dcfSSatish Balay    Options Database Keys:
1085a7e14dcfSSatish Balay .    -tao_max_it <its> - sets the maximum number of iterations
1086a7e14dcfSSatish Balay 
1087a7e14dcfSSatish Balay    Level: intermediate
1088a7e14dcfSSatish Balay 
1089db781477SPatrick Sanan .seealso: `TaoSetTolerances()`, `TaoSetMaximumFunctionEvaluations()`
1090a7e14dcfSSatish Balay @*/
1091441846f8SBarry Smith PetscErrorCode TaoSetMaximumIterations(Tao tao,PetscInt maxits)
1092a7e14dcfSSatish Balay {
1093a7e14dcfSSatish Balay   PetscFunctionBegin;
1094441846f8SBarry Smith   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
109594511df7SStefano Zampini   PetscValidLogicalCollectiveInt(tao,maxits,2);
109647a47007SBarry Smith   tao->max_it = PetscMax(0,maxits);
10976552cf8aSJason Sarich   tao->max_it_changed = PETSC_TRUE;
1098a7e14dcfSSatish Balay   PetscFunctionReturn(0);
1099a7e14dcfSSatish Balay }
1100a7e14dcfSSatish Balay 
1101a7e14dcfSSatish Balay /*@
1102a7e14dcfSSatish Balay    TaoGetMaximumIterations - Sets a maximum number of iterates.
1103a7e14dcfSSatish Balay 
1104a7e14dcfSSatish Balay    Not Collective
1105a7e14dcfSSatish Balay 
1106a7e14dcfSSatish Balay    Input Parameters:
1107441846f8SBarry Smith .  tao - the Tao solver context
1108a7e14dcfSSatish Balay 
1109a7e14dcfSSatish Balay    Output Parameters:
1110a7e14dcfSSatish Balay .  maxits - the maximum number of iterates
1111a7e14dcfSSatish Balay 
1112a7e14dcfSSatish Balay    Level: intermediate
1113a7e14dcfSSatish Balay 
1114db781477SPatrick Sanan .seealso: `TaoSetMaximumIterations()`, `TaoGetMaximumFunctionEvaluations()`
1115a7e14dcfSSatish Balay @*/
1116441846f8SBarry Smith PetscErrorCode TaoGetMaximumIterations(Tao tao,PetscInt *maxits)
1117a7e14dcfSSatish Balay {
1118a7e14dcfSSatish Balay   PetscFunctionBegin;
1119441846f8SBarry Smith   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
112094511df7SStefano Zampini   PetscValidIntPointer(maxits,2);
1121a7e14dcfSSatish Balay   *maxits = tao->max_it;
1122a7e14dcfSSatish Balay   PetscFunctionReturn(0);
1123a7e14dcfSSatish Balay }
1124a7e14dcfSSatish Balay 
1125a7e14dcfSSatish Balay /*@
1126a7e14dcfSSatish Balay    TaoSetInitialTrustRegionRadius - Sets the initial trust region radius.
1127a7e14dcfSSatish Balay 
1128441846f8SBarry Smith    Logically collective on Tao
1129a7e14dcfSSatish Balay 
1130d8d19677SJose E. Roman    Input Parameters:
1131a7e14dcfSSatish Balay +  tao - a TAO optimization solver
1132a7e14dcfSSatish Balay -  radius - the trust region radius
1133a7e14dcfSSatish Balay 
1134a7e14dcfSSatish Balay    Level: intermediate
1135a7e14dcfSSatish Balay 
1136a7e14dcfSSatish Balay    Options Database Key:
1137a7e14dcfSSatish Balay .  -tao_trust0 <t0> - sets initial trust region radius
1138a7e14dcfSSatish Balay 
1139db781477SPatrick Sanan .seealso: `TaoGetTrustRegionRadius()`, `TaoSetTrustRegionTolerance()`
1140a7e14dcfSSatish Balay @*/
1141441846f8SBarry Smith PetscErrorCode TaoSetInitialTrustRegionRadius(Tao tao, PetscReal radius)
1142a7e14dcfSSatish Balay {
1143a7e14dcfSSatish Balay   PetscFunctionBegin;
1144441846f8SBarry Smith   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
114594511df7SStefano Zampini   PetscValidLogicalCollectiveReal(tao,radius,2);
1146a7e14dcfSSatish Balay   tao->trust0 = PetscMax(0.0,radius);
11476552cf8aSJason Sarich   tao->trust0_changed = PETSC_TRUE;
1148a7e14dcfSSatish Balay   PetscFunctionReturn(0);
1149a7e14dcfSSatish Balay }
1150a7e14dcfSSatish Balay 
1151a7e14dcfSSatish Balay /*@
1152a7e14dcfSSatish Balay    TaoGetInitialTrustRegionRadius - Sets the initial trust region radius.
1153a7e14dcfSSatish Balay 
1154a7e14dcfSSatish Balay    Not Collective
1155a7e14dcfSSatish Balay 
1156a7e14dcfSSatish Balay    Input Parameter:
1157a7e14dcfSSatish Balay .  tao - a TAO optimization solver
1158a7e14dcfSSatish Balay 
1159a7e14dcfSSatish Balay    Output Parameter:
1160a7e14dcfSSatish Balay .  radius - the trust region radius
1161a7e14dcfSSatish Balay 
1162a7e14dcfSSatish Balay    Level: intermediate
1163a7e14dcfSSatish Balay 
1164db781477SPatrick Sanan .seealso: `TaoSetInitialTrustRegionRadius()`, `TaoGetCurrentTrustRegionRadius()`
1165a7e14dcfSSatish Balay @*/
1166441846f8SBarry Smith PetscErrorCode TaoGetInitialTrustRegionRadius(Tao tao, PetscReal *radius)
1167a7e14dcfSSatish Balay {
1168a7e14dcfSSatish Balay   PetscFunctionBegin;
1169441846f8SBarry Smith   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
117094511df7SStefano Zampini   PetscValidRealPointer(radius,2);
1171a7e14dcfSSatish Balay   *radius = tao->trust0;
1172a7e14dcfSSatish Balay   PetscFunctionReturn(0);
1173a7e14dcfSSatish Balay }
1174a7e14dcfSSatish Balay 
1175a7e14dcfSSatish Balay /*@
1176a7e14dcfSSatish Balay    TaoGetCurrentTrustRegionRadius - Gets the current trust region radius.
1177a7e14dcfSSatish Balay 
1178a7e14dcfSSatish Balay    Not Collective
1179a7e14dcfSSatish Balay 
1180a7e14dcfSSatish Balay    Input Parameter:
1181a7e14dcfSSatish Balay .  tao - a TAO optimization solver
1182a7e14dcfSSatish Balay 
1183a7e14dcfSSatish Balay    Output Parameter:
1184a7e14dcfSSatish Balay .  radius - the trust region radius
1185a7e14dcfSSatish Balay 
1186a7e14dcfSSatish Balay    Level: intermediate
1187a7e14dcfSSatish Balay 
1188db781477SPatrick Sanan .seealso: `TaoSetInitialTrustRegionRadius()`, `TaoGetInitialTrustRegionRadius()`
1189a7e14dcfSSatish Balay @*/
1190441846f8SBarry Smith PetscErrorCode TaoGetCurrentTrustRegionRadius(Tao tao, PetscReal *radius)
1191a7e14dcfSSatish Balay {
1192a7e14dcfSSatish Balay   PetscFunctionBegin;
1193441846f8SBarry Smith   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
119494511df7SStefano Zampini   PetscValidRealPointer(radius,2);
1195a7e14dcfSSatish Balay   *radius = tao->trust;
1196a7e14dcfSSatish Balay   PetscFunctionReturn(0);
1197a7e14dcfSSatish Balay }
1198a7e14dcfSSatish Balay 
1199a7e14dcfSSatish Balay /*@
1200a7e14dcfSSatish Balay   TaoGetTolerances - gets the current values of tolerances
1201a7e14dcfSSatish Balay 
1202a7e14dcfSSatish Balay   Not Collective
1203a7e14dcfSSatish Balay 
1204f899ff85SJose E. Roman   Input Parameter:
1205441846f8SBarry Smith . tao - the Tao context
1206a7e14dcfSSatish Balay 
1207a7e14dcfSSatish Balay   Output Parameters:
1208e52336cbSBarry Smith + gatol - stop if norm of gradient is less than this
1209a7e14dcfSSatish Balay . grtol - stop if relative norm of gradient is less than this
1210a7e14dcfSSatish Balay - gttol - stop if norm of gradient is reduced by a this factor
1211a7e14dcfSSatish Balay 
121247a47007SBarry Smith   Note: NULL can be used as an argument if not all tolerances values are needed
1213a7e14dcfSSatish Balay 
1214db781477SPatrick Sanan .seealso `TaoSetTolerances()`
1215a7e14dcfSSatish Balay 
1216a7e14dcfSSatish Balay   Level: intermediate
1217a7e14dcfSSatish Balay @*/
1218e52336cbSBarry Smith PetscErrorCode TaoGetTolerances(Tao tao, PetscReal *gatol, PetscReal *grtol, PetscReal *gttol)
1219a7e14dcfSSatish Balay {
1220a7e14dcfSSatish Balay   PetscFunctionBegin;
1221441846f8SBarry Smith   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
1222a7e14dcfSSatish Balay   if (gatol) *gatol = tao->gatol;
1223a7e14dcfSSatish Balay   if (grtol) *grtol = tao->grtol;
1224a7e14dcfSSatish Balay   if (gttol) *gttol = tao->gttol;
1225a7e14dcfSSatish Balay   PetscFunctionReturn(0);
1226a7e14dcfSSatish Balay }
1227a7e14dcfSSatish Balay 
1228a7e14dcfSSatish Balay /*@
1229a7e14dcfSSatish Balay   TaoGetKSP - Gets the linear solver used by the optimization solver.
1230a7e14dcfSSatish Balay   Application writers should use TaoGetKSP if they need direct access
1231a7e14dcfSSatish Balay   to the PETSc KSP object.
1232a7e14dcfSSatish Balay 
1233a7e14dcfSSatish Balay   Not Collective
1234a7e14dcfSSatish Balay 
1235a7e14dcfSSatish Balay    Input Parameters:
1236a7e14dcfSSatish Balay .  tao - the TAO solver
1237a7e14dcfSSatish Balay 
1238a7e14dcfSSatish Balay    Output Parameters:
1239a7e14dcfSSatish Balay .  ksp - the KSP linear solver used in the optimization solver
1240a7e14dcfSSatish Balay 
1241a7e14dcfSSatish Balay    Level: intermediate
1242a7e14dcfSSatish Balay 
1243a7e14dcfSSatish Balay @*/
1244441846f8SBarry Smith PetscErrorCode TaoGetKSP(Tao tao, KSP *ksp)
124547a47007SBarry Smith {
1246a7e14dcfSSatish Balay   PetscFunctionBegin;
124794511df7SStefano Zampini   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
124894511df7SStefano Zampini   PetscValidPointer(ksp,2);
1249a7e14dcfSSatish Balay   *ksp = tao->ksp;
1250a7e14dcfSSatish Balay   PetscFunctionReturn(0);
1251a7e14dcfSSatish Balay }
1252a7e14dcfSSatish Balay 
1253025e9500SJason Sarich /*@
1254025e9500SJason Sarich    TaoGetLinearSolveIterations - Gets the total number of linear iterations
1255025e9500SJason Sarich    used by the TAO solver
1256025e9500SJason Sarich 
1257025e9500SJason Sarich    Not Collective
1258025e9500SJason Sarich 
1259025e9500SJason Sarich    Input Parameter:
1260025e9500SJason Sarich .  tao - TAO context
1261025e9500SJason Sarich 
1262025e9500SJason Sarich    Output Parameter:
1263025e9500SJason Sarich .  lits - number of linear iterations
1264025e9500SJason Sarich 
1265025e9500SJason Sarich    Notes:
1266025e9500SJason Sarich    This counter is reset to zero for each successive call to TaoSolve()
1267025e9500SJason Sarich 
1268025e9500SJason Sarich    Level: intermediate
1269025e9500SJason Sarich 
1270db781477SPatrick Sanan .seealso: `TaoGetKSP()`
1271025e9500SJason Sarich @*/
1272025e9500SJason Sarich PetscErrorCode TaoGetLinearSolveIterations(Tao tao, PetscInt *lits)
1273025e9500SJason Sarich {
1274025e9500SJason Sarich   PetscFunctionBegin;
1275025e9500SJason Sarich   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
1276025e9500SJason Sarich   PetscValidIntPointer(lits,2);
12772d9aa51bSJason Sarich   *lits = tao->ksp_tot_its;
1278025e9500SJason Sarich   PetscFunctionReturn(0);
1279025e9500SJason Sarich }
1280025e9500SJason Sarich 
1281a7e14dcfSSatish Balay /*@
1282a7e14dcfSSatish Balay   TaoGetLineSearch - Gets the line search used by the optimization solver.
1283a7e14dcfSSatish Balay   Application writers should use TaoGetLineSearch if they need direct access
1284a7e14dcfSSatish Balay   to the TaoLineSearch object.
1285a7e14dcfSSatish Balay 
1286a7e14dcfSSatish Balay   Not Collective
1287a7e14dcfSSatish Balay 
1288a7e14dcfSSatish Balay    Input Parameters:
1289a7e14dcfSSatish Balay .  tao - the TAO solver
1290a7e14dcfSSatish Balay 
1291a7e14dcfSSatish Balay    Output Parameters:
1292a7e14dcfSSatish Balay .  ls - the line search used in the optimization solver
1293a7e14dcfSSatish Balay 
1294a7e14dcfSSatish Balay    Level: intermediate
1295a7e14dcfSSatish Balay 
1296a7e14dcfSSatish Balay @*/
1297441846f8SBarry Smith PetscErrorCode TaoGetLineSearch(Tao tao, TaoLineSearch *ls)
129847a47007SBarry Smith {
1299a7e14dcfSSatish Balay   PetscFunctionBegin;
130094511df7SStefano Zampini   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
130194511df7SStefano Zampini   PetscValidPointer(ls,2);
1302a7e14dcfSSatish Balay   *ls = tao->linesearch;
1303a7e14dcfSSatish Balay   PetscFunctionReturn(0);
1304a7e14dcfSSatish Balay }
1305a7e14dcfSSatish Balay 
1306a7e14dcfSSatish Balay /*@
1307a7e14dcfSSatish Balay   TaoAddLineSearchCounts - Adds the number of function evaluations spent
1308a7e14dcfSSatish Balay   in the line search to the running total.
1309a7e14dcfSSatish Balay 
1310a7e14dcfSSatish Balay    Input Parameters:
1311a7e14dcfSSatish Balay +  tao - the TAO solver
1312a7e14dcfSSatish Balay -  ls - the line search used in the optimization solver
1313a7e14dcfSSatish Balay 
1314a7e14dcfSSatish Balay    Level: developer
1315a7e14dcfSSatish Balay 
1316db781477SPatrick Sanan .seealso: `TaoLineSearchApply()`
1317a7e14dcfSSatish Balay @*/
1318441846f8SBarry Smith PetscErrorCode TaoAddLineSearchCounts(Tao tao)
131947a47007SBarry Smith {
1320a7e14dcfSSatish Balay   PetscBool      flg;
1321a7e14dcfSSatish Balay   PetscInt       nfeval,ngeval,nfgeval;
132247a47007SBarry Smith 
1323a7e14dcfSSatish Balay   PetscFunctionBegin;
1324441846f8SBarry Smith   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
1325a7e14dcfSSatish Balay   if (tao->linesearch) {
13269566063dSJacob Faibussowitsch     PetscCall(TaoLineSearchIsUsingTaoRoutines(tao->linesearch,&flg));
132794ae4db5SBarry Smith     if (!flg) {
13289566063dSJacob Faibussowitsch       PetscCall(TaoLineSearchGetNumberFunctionEvaluations(tao->linesearch,&nfeval,&ngeval,&nfgeval));
1329a7e14dcfSSatish Balay       tao->nfuncs += nfeval;
1330a7e14dcfSSatish Balay       tao->ngrads += ngeval;
1331a7e14dcfSSatish Balay       tao->nfuncgrads += nfgeval;
1332a7e14dcfSSatish Balay     }
1333a7e14dcfSSatish Balay   }
1334a7e14dcfSSatish Balay   PetscFunctionReturn(0);
1335a7e14dcfSSatish Balay }
1336a7e14dcfSSatish Balay 
1337a7e14dcfSSatish Balay /*@
1338a82e8c82SStefano Zampini   TaoGetSolution - Returns the vector with the current TAO solution
1339a7e14dcfSSatish Balay 
1340a7e14dcfSSatish Balay   Not Collective
1341a7e14dcfSSatish Balay 
1342a7e14dcfSSatish Balay   Input Parameter:
1343441846f8SBarry Smith . tao - the Tao context
1344a7e14dcfSSatish Balay 
1345a7e14dcfSSatish Balay   Output Parameter:
1346a7e14dcfSSatish Balay . X - the current solution
1347a7e14dcfSSatish Balay 
1348a7e14dcfSSatish Balay   Level: intermediate
1349a7e14dcfSSatish Balay 
1350a82e8c82SStefano Zampini   Note:  The returned vector will be the same object that was passed into TaoSetSolution()
1351a7e14dcfSSatish Balay @*/
1352a82e8c82SStefano Zampini PetscErrorCode TaoGetSolution(Tao tao, Vec *X)
1353a7e14dcfSSatish Balay {
1354a7e14dcfSSatish Balay   PetscFunctionBegin;
1355441846f8SBarry Smith   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
135694511df7SStefano Zampini   PetscValidPointer(X,2);
1357a7e14dcfSSatish Balay   *X = tao->solution;
1358a7e14dcfSSatish Balay   PetscFunctionReturn(0);
1359a7e14dcfSSatish Balay }
1360a7e14dcfSSatish Balay 
1361a7e14dcfSSatish Balay /*@
1362a7e14dcfSSatish Balay    TaoResetStatistics - Initialize the statistics used by TAO for all of the solvers.
1363a7e14dcfSSatish Balay    These statistics include the iteration number, residual norms, and convergence status.
1364a7e14dcfSSatish Balay    This routine gets called before solving each optimization problem.
1365a7e14dcfSSatish Balay 
1366441846f8SBarry Smith    Collective on Tao
1367a7e14dcfSSatish Balay 
1368a7e14dcfSSatish Balay    Input Parameters:
1369441846f8SBarry Smith .  solver - the Tao context
1370a7e14dcfSSatish Balay 
1371a7e14dcfSSatish Balay    Level: developer
1372a7e14dcfSSatish Balay 
1373db781477SPatrick Sanan .seealso: `TaoCreate()`, `TaoSolve()`
1374a7e14dcfSSatish Balay @*/
1375441846f8SBarry Smith PetscErrorCode TaoResetStatistics(Tao tao)
1376a7e14dcfSSatish Balay {
1377a7e14dcfSSatish Balay   PetscFunctionBegin;
1378441846f8SBarry Smith   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
1379a7e14dcfSSatish Balay   tao->niter        = 0;
1380a7e14dcfSSatish Balay   tao->nfuncs       = 0;
1381a7e14dcfSSatish Balay   tao->nfuncgrads   = 0;
1382a7e14dcfSSatish Balay   tao->ngrads       = 0;
1383a7e14dcfSSatish Balay   tao->nhess        = 0;
1384a7e14dcfSSatish Balay   tao->njac         = 0;
1385a7e14dcfSSatish Balay   tao->nconstraints = 0;
1386a7e14dcfSSatish Balay   tao->ksp_its      = 0;
1387ae93cb3cSJason Sarich   tao->ksp_tot_its  = 0;
1388a7e14dcfSSatish Balay   tao->reason       = TAO_CONTINUE_ITERATING;
1389a7e14dcfSSatish Balay   tao->residual     = 0.0;
1390a7e14dcfSSatish Balay   tao->cnorm        = 0.0;
1391a7e14dcfSSatish Balay   tao->step         = 0.0;
1392a7e14dcfSSatish Balay   tao->lsflag       = PETSC_FALSE;
1393a7e14dcfSSatish Balay   if (tao->hist_reset) tao->hist_len = 0;
1394a7e14dcfSSatish Balay   PetscFunctionReturn(0);
1395a7e14dcfSSatish Balay }
1396a7e14dcfSSatish Balay 
1397a7e14dcfSSatish Balay /*@C
1398e1e80dc8SAlp Dener   TaoSetUpdate - Sets the general-purpose update function called
1399e1e80dc8SAlp Dener   at the beginning of every iteration of the nonlinear solve. Specifically
1400e1e80dc8SAlp Dener   it is called at the top of every iteration, after the new solution and the gradient
1401e1e80dc8SAlp Dener   is determined, but before the Hessian is computed (if applicable).
1402e1e80dc8SAlp Dener 
1403e1e80dc8SAlp Dener   Logically Collective on Tao
1404e1e80dc8SAlp Dener 
1405e1e80dc8SAlp Dener   Input Parameters:
1406a2b725a8SWilliam Gropp + tao - The tao solver context
1407a2b725a8SWilliam Gropp - func - The function
1408e1e80dc8SAlp Dener 
1409e1e80dc8SAlp Dener   Calling sequence of func:
1410a2b725a8SWilliam Gropp $ func (Tao tao, PetscInt step);
1411e1e80dc8SAlp Dener 
1412e1e80dc8SAlp Dener . step - The current step of the iteration
1413e1e80dc8SAlp Dener 
1414e1e80dc8SAlp Dener   Level: advanced
1415e1e80dc8SAlp Dener 
1416db781477SPatrick Sanan .seealso `TaoSolve()`
1417e1e80dc8SAlp Dener @*/
14188fcddce6SStefano Zampini PetscErrorCode TaoSetUpdate(Tao tao, PetscErrorCode (*func)(Tao, PetscInt, void*), void *ctx)
1419e1e80dc8SAlp Dener {
1420e1e80dc8SAlp Dener   PetscFunctionBegin;
1421e1e80dc8SAlp Dener   PetscValidHeaderSpecific(tao, TAO_CLASSID,1);
1422e1e80dc8SAlp Dener   tao->ops->update = func;
1423e1e80dc8SAlp Dener   tao->user_update = ctx;
1424e1e80dc8SAlp Dener   PetscFunctionReturn(0);
1425e1e80dc8SAlp Dener }
1426e1e80dc8SAlp Dener 
1427e1e80dc8SAlp Dener /*@C
1428a7e14dcfSSatish Balay   TaoSetConvergenceTest - Sets the function that is to be used to test
1429a7e14dcfSSatish Balay   for convergence o fthe iterative minimization solution.  The new convergence
1430a7e14dcfSSatish Balay   testing routine will replace TAO's default convergence test.
1431a7e14dcfSSatish Balay 
1432441846f8SBarry Smith   Logically Collective on Tao
1433a7e14dcfSSatish Balay 
1434a7e14dcfSSatish Balay   Input Parameters:
1435441846f8SBarry Smith + tao - the Tao object
1436a7e14dcfSSatish Balay . conv - the routine to test for convergence
1437a7e14dcfSSatish Balay - ctx - [optional] context for private data for the convergence routine
143847a47007SBarry Smith         (may be NULL)
1439a7e14dcfSSatish Balay 
1440a7e14dcfSSatish Balay   Calling sequence of conv:
1441441846f8SBarry Smith $   PetscErrorCode conv(Tao tao, void *ctx)
1442a7e14dcfSSatish Balay 
1443441846f8SBarry Smith + tao - the Tao object
1444a7e14dcfSSatish Balay - ctx - [optional] convergence context
1445a7e14dcfSSatish Balay 
1446e4cb33bbSBarry Smith   Note: The new convergence testing routine should call TaoSetConvergedReason().
1447a7e14dcfSSatish Balay 
1448a7e14dcfSSatish Balay   Level: advanced
1449a7e14dcfSSatish Balay 
1450db781477SPatrick Sanan .seealso: `TaoSetConvergedReason()`, `TaoGetSolutionStatus()`, `TaoGetTolerances()`, `TaoSetMonitor`
1451a7e14dcfSSatish Balay 
1452a7e14dcfSSatish Balay @*/
1453441846f8SBarry Smith PetscErrorCode TaoSetConvergenceTest(Tao tao, PetscErrorCode (*conv)(Tao, void*), void *ctx)
1454a7e14dcfSSatish Balay {
1455a7e14dcfSSatish Balay   PetscFunctionBegin;
1456441846f8SBarry Smith   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
145794511df7SStefano Zampini   tao->ops->convergencetest = conv;
145894511df7SStefano Zampini   tao->cnvP = ctx;
1459a7e14dcfSSatish Balay   PetscFunctionReturn(0);
1460a7e14dcfSSatish Balay }
1461a7e14dcfSSatish Balay 
1462a7e14dcfSSatish Balay /*@C
1463a7e14dcfSSatish Balay    TaoSetMonitor - Sets an ADDITIONAL function that is to be used at every
1464a7e14dcfSSatish Balay    iteration of the solver to display the iteration's
1465a7e14dcfSSatish Balay    progress.
1466a7e14dcfSSatish Balay 
1467441846f8SBarry Smith    Logically Collective on Tao
1468a7e14dcfSSatish Balay 
1469a7e14dcfSSatish Balay    Input Parameters:
1470441846f8SBarry Smith +  tao - the Tao solver context
1471a7e14dcfSSatish Balay .  mymonitor - monitoring routine
1472a7e14dcfSSatish Balay -  mctx - [optional] user-defined context for private data for the
147347a47007SBarry Smith           monitor routine (may be NULL)
1474a7e14dcfSSatish Balay 
1475a7e14dcfSSatish Balay    Calling sequence of mymonitor:
1476147403d9SBarry Smith .vb
1477147403d9SBarry Smith      PetscErrorCode mymonitor(Tao tao,void *mctx)
1478147403d9SBarry Smith .ve
1479a7e14dcfSSatish Balay 
1480441846f8SBarry Smith +    tao - the Tao solver context
1481a7e14dcfSSatish Balay -    mctx - [optional] monitoring context
1482a7e14dcfSSatish Balay 
1483a7e14dcfSSatish Balay    Options Database Keys:
148498ea980cSBarry Smith +    -tao_monitor        - sets TaoMonitorDefault()
1485a7e14dcfSSatish Balay .    -tao_smonitor       - sets short monitor
1486a7e14dcfSSatish Balay .    -tao_cmonitor       - same as smonitor plus constraint norm
1487a7e14dcfSSatish Balay .    -tao_view_solution   - view solution at each iteration
1488a7e14dcfSSatish Balay .    -tao_view_gradient   - view gradient at each iteration
14894a48860cSAlp Dener .    -tao_view_ls_residual - view least-squares residual vector at each iteration
1490a7e14dcfSSatish 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.
1491a7e14dcfSSatish Balay 
1492a7e14dcfSSatish Balay    Notes:
1493a7e14dcfSSatish Balay    Several different monitoring routines may be set by calling
1494a7e14dcfSSatish Balay    TaoSetMonitor() multiple times; all will be called in the
1495a7e14dcfSSatish Balay    order in which they were set.
1496a7e14dcfSSatish Balay 
149795452b02SPatrick Sanan    Fortran Notes:
149895452b02SPatrick Sanan     Only one monitor function may be set
1499a7e14dcfSSatish Balay 
1500a7e14dcfSSatish Balay    Level: intermediate
1501a7e14dcfSSatish Balay 
1502db781477SPatrick Sanan .seealso: `TaoMonitorDefault()`, `TaoCancelMonitors()`, `TaoSetDestroyRoutine()`
1503a7e14dcfSSatish Balay @*/
1504441846f8SBarry Smith PetscErrorCode TaoSetMonitor(Tao tao, PetscErrorCode (*func)(Tao, void*), void *ctx,PetscErrorCode (*dest)(void**))
1505a7e14dcfSSatish Balay {
150608d19d1fSJason Sarich   PetscInt       i;
15078732526dSAlp Dener   PetscBool      identical;
150808d19d1fSJason Sarich 
1509a7e14dcfSSatish Balay   PetscFunctionBegin;
1510441846f8SBarry Smith   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
15113c859ba3SBarry Smith   PetscCheck(tao->numbermonitors < MAXTAOMONITORS,PetscObjectComm((PetscObject)tao),PETSC_ERR_SUP,"Cannot attach another monitor -- max=%d",MAXTAOMONITORS);
151208d19d1fSJason Sarich 
151308d19d1fSJason Sarich   for (i=0; i<tao->numbermonitors;i++) {
15149566063dSJacob Faibussowitsch     PetscCall(PetscMonitorCompare((PetscErrorCode (*)(void))func,ctx,dest,(PetscErrorCode (*)(void))tao->monitor[i],tao->monitorcontext[i],tao->monitordestroy[i],&identical));
15158732526dSAlp Dener     if (identical) PetscFunctionReturn(0);
151608d19d1fSJason Sarich   }
1517a7e14dcfSSatish Balay   tao->monitor[tao->numbermonitors] = func;
15188732526dSAlp Dener   tao->monitorcontext[tao->numbermonitors] = (void*)ctx;
1519a7e14dcfSSatish Balay   tao->monitordestroy[tao->numbermonitors] = dest;
1520a7e14dcfSSatish Balay   ++tao->numbermonitors;
1521a7e14dcfSSatish Balay   PetscFunctionReturn(0);
1522a7e14dcfSSatish Balay }
1523a7e14dcfSSatish Balay 
1524a7e14dcfSSatish Balay /*@
1525441846f8SBarry Smith    TaoCancelMonitors - Clears all the monitor functions for a Tao object.
1526a7e14dcfSSatish Balay 
1527441846f8SBarry Smith    Logically Collective on Tao
1528a7e14dcfSSatish Balay 
1529a7e14dcfSSatish Balay    Input Parameters:
1530441846f8SBarry Smith .  tao - the Tao solver context
1531a7e14dcfSSatish Balay 
1532a7e14dcfSSatish Balay    Options Database:
1533a7e14dcfSSatish Balay .  -tao_cancelmonitors - cancels all monitors that have been hardwired
1534a7e14dcfSSatish Balay     into a code by calls to TaoSetMonitor(), but does not cancel those
1535a7e14dcfSSatish Balay     set via the options database
1536a7e14dcfSSatish Balay 
1537a7e14dcfSSatish Balay    Notes:
1538441846f8SBarry Smith    There is no way to clear one specific monitor from a Tao object.
1539a7e14dcfSSatish Balay 
1540a7e14dcfSSatish Balay    Level: advanced
1541a7e14dcfSSatish Balay 
1542db781477SPatrick Sanan .seealso: `TaoMonitorDefault()`, `TaoSetMonitor()`
1543a7e14dcfSSatish Balay @*/
1544441846f8SBarry Smith PetscErrorCode TaoCancelMonitors(Tao tao)
1545a7e14dcfSSatish Balay {
1546a7e14dcfSSatish Balay   PetscInt       i;
154747a47007SBarry Smith 
1548a7e14dcfSSatish Balay   PetscFunctionBegin;
1549441846f8SBarry Smith   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
1550a7e14dcfSSatish Balay   for (i=0;i<tao->numbermonitors;i++) {
1551a7e14dcfSSatish Balay     if (tao->monitordestroy[i]) {
15529566063dSJacob Faibussowitsch       PetscCall((*tao->monitordestroy[i])(&tao->monitorcontext[i]));
1553a7e14dcfSSatish Balay     }
1554a7e14dcfSSatish Balay   }
1555a7e14dcfSSatish Balay   tao->numbermonitors = 0;
1556a7e14dcfSSatish Balay   PetscFunctionReturn(0);
1557a7e14dcfSSatish Balay }
1558a7e14dcfSSatish Balay 
15597fab98ebSJason Sarich /*@
156098ea980cSBarry Smith    TaoMonitorDefault - Default routine for monitoring progress of the
1561441846f8SBarry Smith    Tao solvers (default).  This monitor prints the function value and gradient
1562a7e14dcfSSatish Balay    norm at each iteration.  It can be turned on from the command line using the
1563a7e14dcfSSatish Balay    -tao_monitor option
1564a7e14dcfSSatish Balay 
1565441846f8SBarry Smith    Collective on Tao
1566a7e14dcfSSatish Balay 
1567a7e14dcfSSatish Balay    Input Parameters:
1568441846f8SBarry Smith +  tao - the Tao context
156947a47007SBarry Smith -  ctx - PetscViewer context or NULL
1570a7e14dcfSSatish Balay 
1571a7e14dcfSSatish Balay    Options Database Keys:
1572147403d9SBarry Smith .  -tao_monitor - turn on default monitoring
1573a7e14dcfSSatish Balay 
1574a7e14dcfSSatish Balay    Level: advanced
1575a7e14dcfSSatish Balay 
1576db781477SPatrick Sanan .seealso: `TaoDefaultSMonitor()`, `TaoSetMonitor()`
1577a7e14dcfSSatish Balay @*/
157898ea980cSBarry Smith PetscErrorCode TaoMonitorDefault(Tao tao, void *ctx)
1579a7e14dcfSSatish Balay {
158063b15415SAlp Dener   PetscInt       its, tabs;
1581a7e14dcfSSatish Balay   PetscReal      fct,gnorm;
15828163d661SBarry Smith   PetscViewer    viewer = (PetscViewer)ctx;
158347a47007SBarry Smith 
1584a7e14dcfSSatish Balay   PetscFunctionBegin;
158594511df7SStefano Zampini   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
15868163d661SBarry Smith   PetscValidHeaderSpecific(viewer,PETSC_VIEWER_CLASSID,2);
1587a7e14dcfSSatish Balay   its = tao->niter;
1588a7e14dcfSSatish Balay   fct = tao->fc;
1589a7e14dcfSSatish Balay   gnorm = tao->residual;
15909566063dSJacob Faibussowitsch   PetscCall(PetscViewerASCIIGetTab(viewer, &tabs));
15919566063dSJacob Faibussowitsch   PetscCall(PetscViewerASCIISetTab(viewer, ((PetscObject)tao)->tablevel));
1592494bef23SAlp Dener   if (its == 0 && ((PetscObject)tao)->prefix && !tao->header_printed) {
15939566063dSJacob Faibussowitsch      PetscCall(PetscViewerASCIIPrintf(viewer,"  Iteration information for %s solve.\n",((PetscObject)tao)->prefix));
1594494bef23SAlp Dener      tao->header_printed = PETSC_TRUE;
159563b15415SAlp Dener    }
159663a3b9bcSJacob Faibussowitsch   PetscCall(PetscViewerASCIIPrintf(viewer,"%3" PetscInt_FMT " TAO,",its));
15979566063dSJacob Faibussowitsch   PetscCall(PetscViewerASCIIPrintf(viewer,"  Function value: %g,",(double)fct));
159808d19d1fSJason Sarich   if (gnorm >= PETSC_INFINITY) {
15999566063dSJacob Faibussowitsch     PetscCall(PetscViewerASCIIPrintf(viewer,"  Residual: Inf \n"));
160008d19d1fSJason Sarich   } else {
16019566063dSJacob Faibussowitsch     PetscCall(PetscViewerASCIIPrintf(viewer,"  Residual: %g \n",(double)gnorm));
160208d19d1fSJason Sarich   }
16039566063dSJacob Faibussowitsch   PetscCall(PetscViewerASCIISetTab(viewer, tabs));
1604a7e14dcfSSatish Balay   PetscFunctionReturn(0);
1605a7e14dcfSSatish Balay }
1606a7e14dcfSSatish Balay 
16077fab98ebSJason Sarich /*@
16088d5ead36SAlp Dener    TaoDefaultGMonitor - Default routine for monitoring progress of the
16098d5ead36SAlp Dener    Tao solvers (default) with extra detail on the globalization method.
16108d5ead36SAlp Dener    This monitor prints the function value and gradient norm at each
16118d5ead36SAlp Dener    iteration, as well as the step size and trust radius. Note that the
16128d5ead36SAlp Dener    step size and trust radius may be the same for some algorithms.
16138d5ead36SAlp Dener    It can be turned on from the command line using the
16148d5ead36SAlp Dener    -tao_gmonitor option
16158d5ead36SAlp Dener 
16168d5ead36SAlp Dener    Collective on Tao
16178d5ead36SAlp Dener 
16188d5ead36SAlp Dener    Input Parameters:
16198d5ead36SAlp Dener +  tao - the Tao context
16208d5ead36SAlp Dener -  ctx - PetscViewer context or NULL
16218d5ead36SAlp Dener 
16228d5ead36SAlp Dener    Options Database Keys:
1623147403d9SBarry Smith .  -tao_gmonitor - turn on monitoring with globalization information
16248d5ead36SAlp Dener 
16258d5ead36SAlp Dener    Level: advanced
16268d5ead36SAlp Dener 
1627db781477SPatrick Sanan .seealso: `TaoDefaultSMonitor()`, `TaoSetMonitor()`
16288d5ead36SAlp Dener @*/
16298d5ead36SAlp Dener PetscErrorCode TaoDefaultGMonitor(Tao tao, void *ctx)
16308d5ead36SAlp Dener {
16318d5ead36SAlp Dener   PetscInt       its, tabs;
16328d5ead36SAlp Dener   PetscReal      fct,gnorm,stp,tr;
16338d5ead36SAlp Dener   PetscViewer    viewer = (PetscViewer)ctx;
16348d5ead36SAlp Dener 
16358d5ead36SAlp Dener   PetscFunctionBegin;
163694511df7SStefano Zampini   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
16378d5ead36SAlp Dener   PetscValidHeaderSpecific(viewer,PETSC_VIEWER_CLASSID,2);
16388d5ead36SAlp Dener   its = tao->niter;
16398d5ead36SAlp Dener   fct = tao->fc;
16408d5ead36SAlp Dener   gnorm = tao->residual;
16418d5ead36SAlp Dener   stp = tao->step;
16428d5ead36SAlp Dener   tr = tao->trust;
16439566063dSJacob Faibussowitsch   PetscCall(PetscViewerASCIIGetTab(viewer, &tabs));
16449566063dSJacob Faibussowitsch   PetscCall(PetscViewerASCIISetTab(viewer, ((PetscObject)tao)->tablevel));
1645494bef23SAlp Dener   if (its == 0 && ((PetscObject)tao)->prefix && !tao->header_printed) {
16469566063dSJacob Faibussowitsch      PetscCall(PetscViewerASCIIPrintf(viewer,"  Iteration information for %s solve.\n",((PetscObject)tao)->prefix));
1647494bef23SAlp Dener      tao->header_printed = PETSC_TRUE;
16488d5ead36SAlp Dener    }
164963a3b9bcSJacob Faibussowitsch   PetscCall(PetscViewerASCIIPrintf(viewer,"%3" PetscInt_FMT " TAO,",its));
16509566063dSJacob Faibussowitsch   PetscCall(PetscViewerASCIIPrintf(viewer,"  Function value: %g,",(double)fct));
16518d5ead36SAlp Dener   if (gnorm >= PETSC_INFINITY) {
16529566063dSJacob Faibussowitsch     PetscCall(PetscViewerASCIIPrintf(viewer,"  Residual: Inf,"));
16538d5ead36SAlp Dener   } else {
16549566063dSJacob Faibussowitsch     PetscCall(PetscViewerASCIIPrintf(viewer,"  Residual: %g,",(double)gnorm));
16558d5ead36SAlp Dener   }
16569566063dSJacob Faibussowitsch   PetscCall(PetscViewerASCIIPrintf(viewer,"  Step: %g,  Trust: %g\n",(double)stp,(double)tr));
16579566063dSJacob Faibussowitsch   PetscCall(PetscViewerASCIISetTab(viewer, tabs));
16588d5ead36SAlp Dener   PetscFunctionReturn(0);
16598d5ead36SAlp Dener }
16608d5ead36SAlp Dener 
16618d5ead36SAlp Dener /*@
1662a7e14dcfSSatish Balay    TaoDefaultSMonitor - Default routine for monitoring progress of the
166398ea980cSBarry Smith    solver. Same as TaoMonitorDefault() except
1664a7e14dcfSSatish Balay    it prints fewer digits of the residual as the residual gets smaller.
1665a7e14dcfSSatish Balay    This is because the later digits are meaningless and are often
1666a7e14dcfSSatish Balay    different on different machines; by using this routine different
1667a7e14dcfSSatish Balay    machines will usually generate the same output. It can be turned on
1668a7e14dcfSSatish Balay    by using the -tao_smonitor option
1669a7e14dcfSSatish Balay 
1670441846f8SBarry Smith    Collective on Tao
1671a7e14dcfSSatish Balay 
1672a7e14dcfSSatish Balay    Input Parameters:
1673441846f8SBarry Smith +  tao - the Tao context
16744d4332d5SBarry Smith -  ctx - PetscViewer context of type ASCII
1675a7e14dcfSSatish Balay 
1676a7e14dcfSSatish Balay    Options Database Keys:
1677147403d9SBarry Smith .  -tao_smonitor - turn on default short monitoring
1678a7e14dcfSSatish Balay 
1679a7e14dcfSSatish Balay    Level: advanced
1680a7e14dcfSSatish Balay 
1681db781477SPatrick Sanan .seealso: `TaoMonitorDefault()`, `TaoSetMonitor()`
1682a7e14dcfSSatish Balay @*/
1683441846f8SBarry Smith PetscErrorCode TaoDefaultSMonitor(Tao tao, void *ctx)
1684a7e14dcfSSatish Balay {
16852c9b8b83SAlp Dener   PetscInt       its, tabs;
1686a7e14dcfSSatish Balay   PetscReal      fct,gnorm;
16874d4332d5SBarry Smith   PetscViewer    viewer = (PetscViewer)ctx;
168847a47007SBarry Smith 
1689a7e14dcfSSatish Balay   PetscFunctionBegin;
169094511df7SStefano Zampini   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
16914d4332d5SBarry Smith   PetscValidHeaderSpecific(viewer,PETSC_VIEWER_CLASSID,2);
1692a7e14dcfSSatish Balay   its = tao->niter;
1693a7e14dcfSSatish Balay   fct = tao->fc;
1694a7e14dcfSSatish Balay   gnorm = tao->residual;
16959566063dSJacob Faibussowitsch   PetscCall(PetscViewerASCIIGetTab(viewer, &tabs));
16969566063dSJacob Faibussowitsch   PetscCall(PetscViewerASCIISetTab(viewer, ((PetscObject)tao)->tablevel));
169763a3b9bcSJacob Faibussowitsch   PetscCall(PetscViewerASCIIPrintf(viewer,"iter = %3" PetscInt_FMT ",",its));
16989566063dSJacob Faibussowitsch   PetscCall(PetscViewerASCIIPrintf(viewer," Function value %g,",(double)fct));
1699d393f493SBarry Smith   if (gnorm >= PETSC_INFINITY) {
17009566063dSJacob Faibussowitsch     PetscCall(PetscViewerASCIIPrintf(viewer," Residual: Inf \n"));
170108d19d1fSJason Sarich   } else if (gnorm > 1.e-6) {
17029566063dSJacob Faibussowitsch     PetscCall(PetscViewerASCIIPrintf(viewer," Residual: %g \n",(double)gnorm));
1703a7e14dcfSSatish Balay   } else if (gnorm > 1.e-11) {
17049566063dSJacob Faibussowitsch     PetscCall(PetscViewerASCIIPrintf(viewer," Residual: < 1.0e-6 \n"));
1705a7e14dcfSSatish Balay   } else {
17069566063dSJacob Faibussowitsch     PetscCall(PetscViewerASCIIPrintf(viewer," Residual: < 1.0e-11 \n"));
1707a7e14dcfSSatish Balay   }
17089566063dSJacob Faibussowitsch   PetscCall(PetscViewerASCIISetTab(viewer, tabs));
1709a7e14dcfSSatish Balay   PetscFunctionReturn(0);
1710a7e14dcfSSatish Balay }
1711a7e14dcfSSatish Balay 
17127fab98ebSJason Sarich /*@
171398ea980cSBarry Smith    TaoDefaultCMonitor - same as TaoMonitorDefault() except
1714a7e14dcfSSatish Balay    it prints the norm of the constraints function. It can be turned on
1715a7e14dcfSSatish Balay    from the command line using the -tao_cmonitor option
1716a7e14dcfSSatish Balay 
1717441846f8SBarry Smith    Collective on Tao
1718a7e14dcfSSatish Balay 
1719a7e14dcfSSatish Balay    Input Parameters:
1720441846f8SBarry Smith +  tao - the Tao context
172147a47007SBarry Smith -  ctx - PetscViewer context or NULL
1722a7e14dcfSSatish Balay 
1723a7e14dcfSSatish Balay    Options Database Keys:
1724147403d9SBarry Smith .  -tao_cmonitor - monitor the constraints
1725a7e14dcfSSatish Balay 
1726a7e14dcfSSatish Balay    Level: advanced
1727a7e14dcfSSatish Balay 
1728db781477SPatrick Sanan .seealso: `TaoMonitorDefault()`, `TaoSetMonitor()`
1729a7e14dcfSSatish Balay @*/
1730441846f8SBarry Smith PetscErrorCode TaoDefaultCMonitor(Tao tao, void *ctx)
1731a7e14dcfSSatish Balay {
17322c9b8b83SAlp Dener   PetscInt       its, tabs;
1733a7e14dcfSSatish Balay   PetscReal      fct,gnorm;
1734d393f493SBarry Smith   PetscViewer    viewer = (PetscViewer)ctx;
1735a7e14dcfSSatish Balay 
1736a7e14dcfSSatish Balay   PetscFunctionBegin;
173794511df7SStefano Zampini   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
1738d393f493SBarry Smith   PetscValidHeaderSpecific(viewer,PETSC_VIEWER_CLASSID,2);
1739a7e14dcfSSatish Balay   its = tao->niter;
1740a7e14dcfSSatish Balay   fct = tao->fc;
1741a7e14dcfSSatish Balay   gnorm = tao->residual;
17429566063dSJacob Faibussowitsch   PetscCall(PetscViewerASCIIGetTab(viewer, &tabs));
17439566063dSJacob Faibussowitsch   PetscCall(PetscViewerASCIISetTab(viewer, ((PetscObject)tao)->tablevel));
174463a3b9bcSJacob Faibussowitsch   PetscCall(PetscViewerASCIIPrintf(viewer,"iter = %" PetscInt_FMT ",",its));
17459566063dSJacob Faibussowitsch   PetscCall(PetscViewerASCIIPrintf(viewer," Function value: %g,",(double)fct));
17469566063dSJacob Faibussowitsch   PetscCall(PetscViewerASCIIPrintf(viewer,"  Residual: %g ",(double)gnorm));
17479566063dSJacob Faibussowitsch   PetscCall(PetscViewerASCIIPrintf(viewer,"  Constraint: %g \n",(double)tao->cnorm));
17489566063dSJacob Faibussowitsch   PetscCall(PetscViewerASCIISetTab(viewer, tabs));
1749a7e14dcfSSatish Balay   PetscFunctionReturn(0);
1750a7e14dcfSSatish Balay }
1751a7e14dcfSSatish Balay 
1752a7e14dcfSSatish Balay /*@C
1753a7e14dcfSSatish Balay    TaoSolutionMonitor - Views the solution at each iteration
1754a7e14dcfSSatish Balay    It can be turned on from the command line using the
1755a7e14dcfSSatish Balay    -tao_view_solution option
1756a7e14dcfSSatish Balay 
1757441846f8SBarry Smith    Collective on Tao
1758a7e14dcfSSatish Balay 
1759a7e14dcfSSatish Balay    Input Parameters:
1760441846f8SBarry Smith +  tao - the Tao context
176147a47007SBarry Smith -  ctx - PetscViewer context or NULL
1762a7e14dcfSSatish Balay 
1763a7e14dcfSSatish Balay    Options Database Keys:
1764147403d9SBarry Smith .  -tao_view_solution - view the solution
1765a7e14dcfSSatish Balay 
1766a7e14dcfSSatish Balay    Level: advanced
1767a7e14dcfSSatish Balay 
1768db781477SPatrick Sanan .seealso: `TaoDefaultSMonitor()`, `TaoSetMonitor()`
1769a7e14dcfSSatish Balay @*/
1770441846f8SBarry Smith PetscErrorCode TaoSolutionMonitor(Tao tao, void *ctx)
1771a7e14dcfSSatish Balay {
1772feb237baSPierre Jolivet   PetscViewer    viewer  = (PetscViewer)ctx;
177347a47007SBarry Smith 
1774a7e14dcfSSatish Balay   PetscFunctionBegin;
177594511df7SStefano Zampini   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
1776d393f493SBarry Smith   PetscValidHeaderSpecific(viewer,PETSC_VIEWER_CLASSID,2);
17779566063dSJacob Faibussowitsch   PetscCall(VecView(tao->solution,viewer));
1778a7e14dcfSSatish Balay   PetscFunctionReturn(0);
1779a7e14dcfSSatish Balay }
1780a7e14dcfSSatish Balay 
1781a7e14dcfSSatish Balay /*@C
1782a7e14dcfSSatish Balay    TaoGradientMonitor - Views the gradient at each iteration
1783a7e14dcfSSatish Balay    It can be turned on from the command line using the
1784a7e14dcfSSatish Balay    -tao_view_gradient option
1785a7e14dcfSSatish Balay 
1786441846f8SBarry Smith    Collective on Tao
1787a7e14dcfSSatish Balay 
1788a7e14dcfSSatish Balay    Input Parameters:
1789441846f8SBarry Smith +  tao - the Tao context
179047a47007SBarry Smith -  ctx - PetscViewer context or NULL
1791a7e14dcfSSatish Balay 
1792a7e14dcfSSatish Balay    Options Database Keys:
1793147403d9SBarry Smith .  -tao_view_gradient - view the gradient at each iteration
1794a7e14dcfSSatish Balay 
1795a7e14dcfSSatish Balay    Level: advanced
1796a7e14dcfSSatish Balay 
1797db781477SPatrick Sanan .seealso: `TaoDefaultSMonitor()`, `TaoSetMonitor()`
1798a7e14dcfSSatish Balay @*/
1799441846f8SBarry Smith PetscErrorCode TaoGradientMonitor(Tao tao, void *ctx)
1800a7e14dcfSSatish Balay {
1801d393f493SBarry Smith   PetscViewer    viewer = (PetscViewer)ctx;
180247a47007SBarry Smith 
1803a7e14dcfSSatish Balay   PetscFunctionBegin;
180494511df7SStefano Zampini   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
1805d393f493SBarry Smith   PetscValidHeaderSpecific(viewer,PETSC_VIEWER_CLASSID,2);
18069566063dSJacob Faibussowitsch   PetscCall(VecView(tao->gradient,viewer));
1807a7e14dcfSSatish Balay   PetscFunctionReturn(0);
1808a7e14dcfSSatish Balay }
1809a7e14dcfSSatish Balay 
1810a7e14dcfSSatish Balay /*@C
1811147403d9SBarry Smith    TaoStepDirectionMonitor - Views the step-direction at each iteration
1812a7e14dcfSSatish Balay 
1813441846f8SBarry Smith    Collective on Tao
1814a7e14dcfSSatish Balay 
1815a7e14dcfSSatish Balay    Input Parameters:
1816441846f8SBarry Smith +  tao - the Tao context
181747a47007SBarry Smith -  ctx - PetscViewer context or NULL
1818a7e14dcfSSatish Balay 
1819a7e14dcfSSatish Balay    Options Database Keys:
1820147403d9SBarry Smith .  -tao_view_gradient - view the gradient at each iteration
1821a7e14dcfSSatish Balay 
1822a7e14dcfSSatish Balay    Level: advanced
1823a7e14dcfSSatish Balay 
1824db781477SPatrick Sanan .seealso: `TaoDefaultSMonitor()`, `TaoSetMonitor()`
1825a7e14dcfSSatish Balay @*/
1826441846f8SBarry Smith PetscErrorCode TaoStepDirectionMonitor(Tao tao, void *ctx)
1827a7e14dcfSSatish Balay {
1828d393f493SBarry Smith   PetscViewer    viewer = (PetscViewer)ctx;
1829d393f493SBarry Smith 
1830a7e14dcfSSatish Balay   PetscFunctionBegin;
183194511df7SStefano Zampini   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
1832d393f493SBarry Smith   PetscValidHeaderSpecific(viewer,PETSC_VIEWER_CLASSID,2);
18339566063dSJacob Faibussowitsch   PetscCall(VecView(tao->stepdirection,viewer));
1834a7e14dcfSSatish Balay   PetscFunctionReturn(0);
1835a7e14dcfSSatish Balay }
1836a7e14dcfSSatish Balay 
1837a7e14dcfSSatish Balay /*@C
1838a7e14dcfSSatish Balay    TaoDrawSolutionMonitor - Plots the solution at each iteration
1839a7e14dcfSSatish Balay    It can be turned on from the command line using the
1840a7e14dcfSSatish Balay    -tao_draw_solution option
1841a7e14dcfSSatish Balay 
1842441846f8SBarry Smith    Collective on Tao
1843a7e14dcfSSatish Balay 
1844a7e14dcfSSatish Balay    Input Parameters:
1845441846f8SBarry Smith +  tao - the Tao context
1846e882e171SHong Zhang -  ctx - TaoMonitorDraw context
1847a7e14dcfSSatish Balay 
1848a7e14dcfSSatish Balay    Options Database Keys:
1849147403d9SBarry Smith .  -tao_draw_solution - draw the solution at each iteration
1850a7e14dcfSSatish Balay 
1851a7e14dcfSSatish Balay    Level: advanced
1852a7e14dcfSSatish Balay 
1853db781477SPatrick Sanan .seealso: `TaoSolutionMonitor()`, `TaoSetMonitor()`, `TaoDrawGradientMonitor`
1854a7e14dcfSSatish Balay @*/
1855441846f8SBarry Smith PetscErrorCode TaoDrawSolutionMonitor(Tao tao, void *ctx)
1856a7e14dcfSSatish Balay {
1857e882e171SHong Zhang   TaoMonitorDrawCtx ictx = (TaoMonitorDrawCtx)ctx;
185847a47007SBarry Smith 
1859a7e14dcfSSatish Balay   PetscFunctionBegin;
186094511df7SStefano Zampini   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
1861e882e171SHong Zhang   if (!(((ictx->howoften > 0) && (!(tao->niter % ictx->howoften))) || ((ictx->howoften == -1) && tao->reason))) PetscFunctionReturn(0);
18629566063dSJacob Faibussowitsch   PetscCall(VecView(tao->solution,ictx->viewer));
1863a7e14dcfSSatish Balay   PetscFunctionReturn(0);
1864a7e14dcfSSatish Balay }
1865a7e14dcfSSatish Balay 
1866a7e14dcfSSatish Balay /*@C
1867a7e14dcfSSatish Balay    TaoDrawGradientMonitor - Plots the gradient at each iteration
1868a7e14dcfSSatish Balay    It can be turned on from the command line using the
1869a7e14dcfSSatish Balay    -tao_draw_gradient option
1870a7e14dcfSSatish Balay 
1871441846f8SBarry Smith    Collective on Tao
1872a7e14dcfSSatish Balay 
1873a7e14dcfSSatish Balay    Input Parameters:
1874441846f8SBarry Smith +  tao - the Tao context
1875f55353a2SBarry Smith -  ctx - PetscViewer context
1876a7e14dcfSSatish Balay 
1877a7e14dcfSSatish Balay    Options Database Keys:
1878147403d9SBarry Smith .  -tao_draw_gradient - draw the gradient at each iteration
1879a7e14dcfSSatish Balay 
1880a7e14dcfSSatish Balay    Level: advanced
1881a7e14dcfSSatish Balay 
1882db781477SPatrick Sanan .seealso: `TaoGradientMonitor()`, `TaoSetMonitor()`, `TaoDrawSolutionMonitor`
1883a7e14dcfSSatish Balay @*/
1884441846f8SBarry Smith PetscErrorCode TaoDrawGradientMonitor(Tao tao, void *ctx)
1885a7e14dcfSSatish Balay {
1886e882e171SHong Zhang   TaoMonitorDrawCtx ictx = (TaoMonitorDrawCtx)ctx;
188747a47007SBarry Smith 
1888a7e14dcfSSatish Balay   PetscFunctionBegin;
188994511df7SStefano Zampini   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
1890e882e171SHong Zhang   if (!(((ictx->howoften > 0) && (!(tao->niter % ictx->howoften))) || ((ictx->howoften == -1) && tao->reason))) PetscFunctionReturn(0);
18919566063dSJacob Faibussowitsch   PetscCall(VecView(tao->gradient,ictx->viewer));
1892a7e14dcfSSatish Balay   PetscFunctionReturn(0);
1893a7e14dcfSSatish Balay }
1894a7e14dcfSSatish Balay 
1895a7e14dcfSSatish Balay /*@C
1896a7e14dcfSSatish Balay    TaoDrawStepMonitor - Plots the step direction at each iteration
1897a7e14dcfSSatish Balay 
1898441846f8SBarry Smith    Collective on Tao
1899a7e14dcfSSatish Balay 
1900a7e14dcfSSatish Balay    Input Parameters:
1901441846f8SBarry Smith +  tao - the Tao context
1902f55353a2SBarry Smith -  ctx - PetscViewer context
1903a7e14dcfSSatish Balay 
1904a7e14dcfSSatish Balay    Options Database Keys:
1905147403d9SBarry Smith .  -tao_draw_step - draw the step direction at each iteration
1906a7e14dcfSSatish Balay 
1907a7e14dcfSSatish Balay    Level: advanced
1908a7e14dcfSSatish Balay 
1909db781477SPatrick Sanan .seealso: `TaoSetMonitor()`, `TaoDrawSolutionMonitor`
1910a7e14dcfSSatish Balay @*/
1911441846f8SBarry Smith PetscErrorCode TaoDrawStepMonitor(Tao tao, void *ctx)
1912a7e14dcfSSatish Balay {
191394511df7SStefano Zampini   PetscViewer    viewer = (PetscViewer)ctx;
191447a47007SBarry Smith 
1915a7e14dcfSSatish Balay   PetscFunctionBegin;
191694511df7SStefano Zampini   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
191794511df7SStefano Zampini   PetscValidHeaderSpecific(viewer,PETSC_VIEWER_CLASSID,2);
19189566063dSJacob Faibussowitsch   PetscCall(VecView(tao->stepdirection,viewer));
1919a7e14dcfSSatish Balay   PetscFunctionReturn(0);
1920a7e14dcfSSatish Balay }
1921a7e14dcfSSatish Balay 
1922a7e14dcfSSatish Balay /*@C
1923737f463aSAlp Dener    TaoResidualMonitor - Views the least-squares residual at each iteration
1924a7e14dcfSSatish Balay 
1925441846f8SBarry Smith    Collective on Tao
1926a7e14dcfSSatish Balay 
1927a7e14dcfSSatish Balay    Input Parameters:
1928441846f8SBarry Smith +  tao - the Tao context
192947a47007SBarry Smith -  ctx - PetscViewer context or NULL
1930a7e14dcfSSatish Balay 
1931a7e14dcfSSatish Balay    Options Database Keys:
1932147403d9SBarry Smith .  -tao_view_ls_residual - view the least-squares residual at each iteration
1933a7e14dcfSSatish Balay 
1934a7e14dcfSSatish Balay    Level: advanced
1935a7e14dcfSSatish Balay 
1936db781477SPatrick Sanan .seealso: `TaoDefaultSMonitor()`, `TaoSetMonitor()`
1937a7e14dcfSSatish Balay @*/
1938737f463aSAlp Dener PetscErrorCode TaoResidualMonitor(Tao tao, void *ctx)
1939a7e14dcfSSatish Balay {
1940d393f493SBarry Smith   PetscViewer    viewer  = (PetscViewer)ctx;
194147a47007SBarry Smith 
1942a7e14dcfSSatish Balay   PetscFunctionBegin;
194394511df7SStefano Zampini   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
1944d393f493SBarry Smith   PetscValidHeaderSpecific(viewer,PETSC_VIEWER_CLASSID,2);
19459566063dSJacob Faibussowitsch   PetscCall(VecView(tao->ls_res,viewer));
1946a7e14dcfSSatish Balay   PetscFunctionReturn(0);
1947a7e14dcfSSatish Balay }
1948a7e14dcfSSatish Balay 
19497fab98ebSJason Sarich /*@
1950a7e14dcfSSatish Balay    TaoDefaultConvergenceTest - Determines whether the solver should continue iterating
1951a7e14dcfSSatish Balay    or terminate.
1952a7e14dcfSSatish Balay 
1953441846f8SBarry Smith    Collective on Tao
1954a7e14dcfSSatish Balay 
1955a7e14dcfSSatish Balay    Input Parameters:
1956441846f8SBarry Smith +  tao - the Tao context
1957a7e14dcfSSatish Balay -  dummy - unused dummy context
1958a7e14dcfSSatish Balay 
1959a7e14dcfSSatish Balay    Output Parameter:
1960a7e14dcfSSatish Balay .  reason - for terminating
1961a7e14dcfSSatish Balay 
1962a7e14dcfSSatish Balay    Notes:
1963a7e14dcfSSatish Balay    This routine checks the residual in the optimality conditions, the
1964a7e14dcfSSatish Balay    relative residual in the optimity conditions, the number of function
1965a7e14dcfSSatish Balay    evaluations, and the function value to test convergence.  Some
1966a7e14dcfSSatish Balay    solvers may use different convergence routines.
1967a7e14dcfSSatish Balay 
1968a7e14dcfSSatish Balay    Level: developer
1969a7e14dcfSSatish Balay 
1970c2e3fba1SPatrick Sanan .seealso: `TaoSetTolerances()`, `TaoGetConvergedReason()`, `TaoSetConvergedReason()`
1971a7e14dcfSSatish Balay @*/
1972a7e14dcfSSatish Balay 
1973441846f8SBarry Smith PetscErrorCode TaoDefaultConvergenceTest(Tao tao,void *dummy)
1974a7e14dcfSSatish Balay {
1975a7e14dcfSSatish Balay   PetscInt           niter=tao->niter, nfuncs=PetscMax(tao->nfuncs,tao->nfuncgrads);
1976a7e14dcfSSatish Balay   PetscInt           max_funcs=tao->max_funcs;
1977a7e14dcfSSatish Balay   PetscReal          gnorm=tao->residual, gnorm0=tao->gnorm0;
1978a7e14dcfSSatish Balay   PetscReal          f=tao->fc, steptol=tao->steptol,trradius=tao->step;
1979a7e14dcfSSatish Balay   PetscReal          gatol=tao->gatol,grtol=tao->grtol,gttol=tao->gttol;
1980e52336cbSBarry Smith   PetscReal          catol=tao->catol,crtol=tao->crtol;
1981e52336cbSBarry Smith   PetscReal          fmin=tao->fmin, cnorm=tao->cnorm;
1982e4cb33bbSBarry Smith   TaoConvergedReason reason=tao->reason;
1983a7e14dcfSSatish Balay 
1984a7e14dcfSSatish Balay   PetscFunctionBegin;
1985441846f8SBarry Smith   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
198694511df7SStefano Zampini   if (reason != TAO_CONTINUE_ITERATING) PetscFunctionReturn(0);
1987a7e14dcfSSatish Balay 
1988a7e14dcfSSatish Balay   if (PetscIsInfOrNanReal(f)) {
19899566063dSJacob Faibussowitsch     PetscCall(PetscInfo(tao,"Failed to converged, function value is Inf or NaN\n"));
1990a7e14dcfSSatish Balay     reason = TAO_DIVERGED_NAN;
1991a7e14dcfSSatish Balay   } else if (f <= fmin && cnorm <=catol) {
19929566063dSJacob Faibussowitsch     PetscCall(PetscInfo(tao,"Converged due to function value %g < minimum function value %g\n", (double)f,(double)fmin));
1993a7e14dcfSSatish Balay     reason = TAO_CONVERGED_MINF;
1994a7e14dcfSSatish Balay   } else if (gnorm<= gatol && cnorm <=catol) {
19959566063dSJacob Faibussowitsch     PetscCall(PetscInfo(tao,"Converged due to residual norm ||g(X)||=%g < %g\n",(double)gnorm,(double)gatol));
1996a7e14dcfSSatish Balay     reason = TAO_CONVERGED_GATOL;
1997a7e14dcfSSatish Balay   } else if (f!=0 && PetscAbsReal(gnorm/f) <= grtol && cnorm <= crtol) {
19989566063dSJacob Faibussowitsch     PetscCall(PetscInfo(tao,"Converged due to residual ||g(X)||/|f(X)| =%g < %g\n",(double)(gnorm/f),(double)grtol));
1999a7e14dcfSSatish Balay     reason = TAO_CONVERGED_GRTOL;
2000e1d16bb9SBarry Smith   } else if (gnorm0 != 0 && ((gttol == 0 && gnorm == 0) || gnorm/gnorm0 < gttol) && cnorm <= crtol) {
20019566063dSJacob Faibussowitsch     PetscCall(PetscInfo(tao,"Converged due to relative residual norm ||g(X)||/||g(X0)|| = %g < %g\n",(double)(gnorm/gnorm0),(double)gttol));
2002a7e14dcfSSatish Balay     reason = TAO_CONVERGED_GTTOL;
200394511df7SStefano Zampini   } else if (max_funcs >=0 && nfuncs > max_funcs) {
20049566063dSJacob Faibussowitsch     PetscCall(PetscInfo(tao,"Exceeded maximum number of function evaluations: %" PetscInt_FMT " > %" PetscInt_FMT "\n", nfuncs,max_funcs));
2005a7e14dcfSSatish Balay     reason = TAO_DIVERGED_MAXFCN;
2006a7e14dcfSSatish Balay   } else if (tao->lsflag != 0) {
20079566063dSJacob Faibussowitsch     PetscCall(PetscInfo(tao,"Tao Line Search failure.\n"));
2008a7e14dcfSSatish Balay     reason = TAO_DIVERGED_LS_FAILURE;
2009a7e14dcfSSatish Balay   } else if (trradius < steptol && niter > 0) {
20109566063dSJacob Faibussowitsch     PetscCall(PetscInfo(tao,"Trust region/step size too small: %g < %g\n", (double)trradius,(double)steptol));
2011a7e14dcfSSatish Balay     reason = TAO_CONVERGED_STEPTOL;
2012e031d6f5SAlp Dener   } else if (niter >= tao->max_it) {
201363a3b9bcSJacob Faibussowitsch     PetscCall(PetscInfo(tao,"Exceeded maximum number of iterations: %" PetscInt_FMT " > %" PetscInt_FMT "\n",niter,tao->max_it));
2014a7e14dcfSSatish Balay     reason = TAO_DIVERGED_MAXITS;
2015a7e14dcfSSatish Balay   } else {
2016a7e14dcfSSatish Balay     reason = TAO_CONTINUE_ITERATING;
2017a7e14dcfSSatish Balay   }
2018a7e14dcfSSatish Balay   tao->reason = reason;
2019a7e14dcfSSatish Balay   PetscFunctionReturn(0);
2020a7e14dcfSSatish Balay }
2021a7e14dcfSSatish Balay 
2022a7e14dcfSSatish Balay /*@C
2023a7e14dcfSSatish Balay    TaoSetOptionsPrefix - Sets the prefix used for searching for all
2024a7e14dcfSSatish Balay    TAO options in the database.
2025a7e14dcfSSatish Balay 
2026441846f8SBarry Smith    Logically Collective on Tao
2027a7e14dcfSSatish Balay 
2028a7e14dcfSSatish Balay    Input Parameters:
2029441846f8SBarry Smith +  tao - the Tao context
2030a7e14dcfSSatish Balay -  prefix - the prefix string to prepend to all TAO option requests
2031a7e14dcfSSatish Balay 
2032a7e14dcfSSatish Balay    Notes:
2033a7e14dcfSSatish Balay    A hyphen (-) must NOT be given at the beginning of the prefix name.
2034a7e14dcfSSatish Balay    The first character of all runtime options is AUTOMATICALLY the hyphen.
2035a7e14dcfSSatish Balay 
2036a7e14dcfSSatish Balay    For example, to distinguish between the runtime options for two
2037a7e14dcfSSatish Balay    different TAO solvers, one could call
2038a7e14dcfSSatish Balay .vb
2039a7e14dcfSSatish Balay       TaoSetOptionsPrefix(tao1,"sys1_")
2040a7e14dcfSSatish Balay       TaoSetOptionsPrefix(tao2,"sys2_")
2041a7e14dcfSSatish Balay .ve
2042a7e14dcfSSatish Balay 
2043a7e14dcfSSatish Balay    This would enable use of different options for each system, such as
2044a7e14dcfSSatish Balay .vb
20459fa2b5dcSStefano Zampini       -sys1_tao_method blmvm -sys1_tao_grtol 1.e-3
20469fa2b5dcSStefano Zampini       -sys2_tao_method lmvm  -sys2_tao_grtol 1.e-4
2047a7e14dcfSSatish Balay .ve
2048a7e14dcfSSatish Balay 
2049a7e14dcfSSatish Balay    Level: advanced
2050a7e14dcfSSatish Balay 
2051db781477SPatrick Sanan .seealso: `TaoAppendOptionsPrefix()`, `TaoGetOptionsPrefix()`
2052a7e14dcfSSatish Balay @*/
2053a7e14dcfSSatish Balay 
2054441846f8SBarry Smith PetscErrorCode TaoSetOptionsPrefix(Tao tao, const char p[])
2055a7e14dcfSSatish Balay {
2056a7e14dcfSSatish Balay   PetscFunctionBegin;
205794511df7SStefano Zampini   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
20589566063dSJacob Faibussowitsch   PetscCall(PetscObjectSetOptionsPrefix((PetscObject)tao,p));
2059*1baa6e33SBarry Smith   if (tao->linesearch) PetscCall(TaoLineSearchSetOptionsPrefix(tao->linesearch,p));
2060*1baa6e33SBarry Smith   if (tao->ksp) PetscCall(KSPSetOptionsPrefix(tao->ksp,p));
2061a7e14dcfSSatish Balay   PetscFunctionReturn(0);
2062a7e14dcfSSatish Balay }
2063a7e14dcfSSatish Balay 
2064a7e14dcfSSatish Balay /*@C
2065a7e14dcfSSatish Balay    TaoAppendOptionsPrefix - Appends to the prefix used for searching for all
2066a7e14dcfSSatish Balay    TAO options in the database.
2067a7e14dcfSSatish Balay 
2068441846f8SBarry Smith    Logically Collective on Tao
2069a7e14dcfSSatish Balay 
2070a7e14dcfSSatish Balay    Input Parameters:
2071441846f8SBarry Smith +  tao - the Tao solver context
2072a7e14dcfSSatish Balay -  prefix - the prefix string to prepend to all TAO option requests
2073a7e14dcfSSatish Balay 
2074a7e14dcfSSatish Balay    Notes:
2075a7e14dcfSSatish Balay    A hyphen (-) must NOT be given at the beginning of the prefix name.
2076a7e14dcfSSatish Balay    The first character of all runtime options is AUTOMATICALLY the hyphen.
2077a7e14dcfSSatish Balay 
2078a7e14dcfSSatish Balay    Level: advanced
2079a7e14dcfSSatish Balay 
2080db781477SPatrick Sanan .seealso: `TaoSetOptionsPrefix()`, `TaoGetOptionsPrefix()`
2081a7e14dcfSSatish Balay @*/
2082441846f8SBarry Smith PetscErrorCode TaoAppendOptionsPrefix(Tao tao, const char p[])
2083a7e14dcfSSatish Balay {
2084a7e14dcfSSatish Balay   PetscFunctionBegin;
208594511df7SStefano Zampini   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
20869566063dSJacob Faibussowitsch   PetscCall(PetscObjectAppendOptionsPrefix((PetscObject)tao,p));
2087*1baa6e33SBarry Smith   if (tao->linesearch) PetscCall(PetscObjectAppendOptionsPrefix((PetscObject)tao->linesearch,p));
2088*1baa6e33SBarry Smith   if (tao->ksp) PetscCall(KSPAppendOptionsPrefix(tao->ksp,p));
2089a7e14dcfSSatish Balay   PetscFunctionReturn(0);
2090a7e14dcfSSatish Balay }
2091a7e14dcfSSatish Balay 
2092a7e14dcfSSatish Balay /*@C
2093a7e14dcfSSatish Balay   TaoGetOptionsPrefix - Gets the prefix used for searching for all
2094a7e14dcfSSatish Balay   TAO options in the database
2095a7e14dcfSSatish Balay 
2096a7e14dcfSSatish Balay   Not Collective
2097a7e14dcfSSatish Balay 
2098a7e14dcfSSatish Balay   Input Parameters:
2099441846f8SBarry Smith . tao - the Tao context
2100a7e14dcfSSatish Balay 
2101a7e14dcfSSatish Balay   Output Parameters:
2102a7e14dcfSSatish Balay . prefix - pointer to the prefix string used is returned
2103a7e14dcfSSatish Balay 
210495452b02SPatrick Sanan   Notes:
210595452b02SPatrick Sanan     On the fortran side, the user should pass in a string 'prefix' of
2106a7e14dcfSSatish Balay   sufficient length to hold the prefix.
2107a7e14dcfSSatish Balay 
2108a7e14dcfSSatish Balay   Level: advanced
2109a7e14dcfSSatish Balay 
2110db781477SPatrick Sanan .seealso: `TaoSetOptionsPrefix()`, `TaoAppendOptionsPrefix()`
2111a7e14dcfSSatish Balay @*/
2112441846f8SBarry Smith PetscErrorCode TaoGetOptionsPrefix(Tao tao, const char *p[])
2113a7e14dcfSSatish Balay {
211494511df7SStefano Zampini   PetscFunctionBegin;
211594511df7SStefano Zampini   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
21169566063dSJacob Faibussowitsch   PetscCall(PetscObjectGetOptionsPrefix((PetscObject)tao,p));
211794511df7SStefano Zampini   PetscFunctionReturn(0);
2118a7e14dcfSSatish Balay }
2119a7e14dcfSSatish Balay 
2120a7e14dcfSSatish Balay /*@C
2121a7e14dcfSSatish Balay    TaoSetType - Sets the method for the unconstrained minimization solver.
2122a7e14dcfSSatish Balay 
2123441846f8SBarry Smith    Collective on Tao
2124a7e14dcfSSatish Balay 
2125a7e14dcfSSatish Balay    Input Parameters:
2126441846f8SBarry Smith +  solver - the Tao solver context
2127a7e14dcfSSatish Balay -  type - a known method
2128a7e14dcfSSatish Balay 
2129a7e14dcfSSatish Balay    Options Database Key:
213058417fe7SBarry Smith .  -tao_type <type> - Sets the method; use -help for a list
213158417fe7SBarry Smith    of available methods (for instance, "-tao_type lmvm" or "-tao_type tron")
2132a7e14dcfSSatish Balay 
2133a7e14dcfSSatish Balay    Available methods include:
213458417fe7SBarry Smith +    nls - Newton's method with line search for unconstrained minimization
213558417fe7SBarry Smith .    ntr - Newton's method with trust region for unconstrained minimization
213658417fe7SBarry Smith .    ntl - Newton's method with trust region, line search for unconstrained minimization
213758417fe7SBarry Smith .    lmvm - Limited memory variable metric method for unconstrained minimization
213858417fe7SBarry Smith .    cg - Nonlinear conjugate gradient method for unconstrained minimization
213958417fe7SBarry Smith .    nm - Nelder-Mead algorithm for derivate-free unconstrained minimization
214058417fe7SBarry Smith .    tron - Newton Trust Region method for bound constrained minimization
214158417fe7SBarry Smith .    gpcg - Newton Trust Region method for quadratic bound constrained minimization
214258417fe7SBarry Smith .    blmvm - Limited memory variable metric method for bound constrained minimization
21431eb8069cSJason Sarich -    pounders - Model-based algorithm pounder extended for nonlinear least squares
2144a7e14dcfSSatish Balay 
2145a7e14dcfSSatish Balay   Level: intermediate
2146a7e14dcfSSatish Balay 
2147db781477SPatrick Sanan .seealso: `TaoCreate()`, `TaoGetType()`, `TaoType`
2148a7e14dcfSSatish Balay 
2149a7e14dcfSSatish Balay @*/
2150b625d6c7SJed Brown PetscErrorCode TaoSetType(Tao tao, TaoType type)
2151a7e14dcfSSatish Balay {
2152441846f8SBarry Smith   PetscErrorCode (*create_xxx)(Tao);
2153a7e14dcfSSatish Balay   PetscBool      issame;
2154a7e14dcfSSatish Balay 
2155a7e14dcfSSatish Balay   PetscFunctionBegin;
2156441846f8SBarry Smith   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
2157a7e14dcfSSatish Balay 
21589566063dSJacob Faibussowitsch   PetscCall(PetscObjectTypeCompare((PetscObject)tao,type,&issame));
2159a7e14dcfSSatish Balay   if (issame) PetscFunctionReturn(0);
2160a7e14dcfSSatish Balay 
21619566063dSJacob Faibussowitsch   PetscCall(PetscFunctionListFind(TaoList, type, (void(**)(void))&create_xxx));
21623c859ba3SBarry Smith   PetscCheck(create_xxx,PetscObjectComm((PetscObject)tao),PETSC_ERR_ARG_UNKNOWN_TYPE,"Unable to find requested Tao type %s",type);
2163a7e14dcfSSatish Balay 
2164a7e14dcfSSatish Balay   /* Destroy the existing solver information */
2165a958fbfcSStefano Zampini   if (tao->ops->destroy) PetscCall((*tao->ops->destroy)(tao));
2166*1baa6e33SBarry Smith   PetscCall(KSPDestroy(&tao->ksp));
21679566063dSJacob Faibussowitsch   PetscCall(TaoLineSearchDestroy(&tao->linesearch));
216883c8fe1dSLisandro Dalcin   tao->ops->setup          = NULL;
216983c8fe1dSLisandro Dalcin   tao->ops->solve          = NULL;
217083c8fe1dSLisandro Dalcin   tao->ops->view           = NULL;
217183c8fe1dSLisandro Dalcin   tao->ops->setfromoptions = NULL;
217283c8fe1dSLisandro Dalcin   tao->ops->destroy        = NULL;
2173a7e14dcfSSatish Balay 
2174a7e14dcfSSatish Balay   tao->setupcalled = PETSC_FALSE;
2175a7e14dcfSSatish Balay 
21769566063dSJacob Faibussowitsch   PetscCall((*create_xxx)(tao));
21779566063dSJacob Faibussowitsch   PetscCall(PetscObjectChangeTypeName((PetscObject)tao,type));
2178a7e14dcfSSatish Balay   PetscFunctionReturn(0);
2179a7e14dcfSSatish Balay }
218047a47007SBarry Smith 
2181a7e14dcfSSatish Balay /*MC
2182441846f8SBarry Smith    TaoRegister - Adds a method to the TAO package for unconstrained minimization.
2183a7e14dcfSSatish Balay 
2184a7e14dcfSSatish Balay    Synopsis:
21855bd1e576SStefano Zampini    TaoRegister(char *name_solver,char *path,char *name_Create,PetscErrorCode (*routine_Create)(Tao))
2186a7e14dcfSSatish Balay 
2187a7e14dcfSSatish Balay    Not collective
2188a7e14dcfSSatish Balay 
2189a7e14dcfSSatish Balay    Input Parameters:
2190a7e14dcfSSatish Balay +  sname - name of a new user-defined solver
2191a7e14dcfSSatish Balay -  func - routine to Create method context
2192a7e14dcfSSatish Balay 
2193a7e14dcfSSatish Balay    Notes:
2194441846f8SBarry Smith    TaoRegister() may be called multiple times to add several user-defined solvers.
2195a7e14dcfSSatish Balay 
2196a7e14dcfSSatish Balay    Sample usage:
2197a7e14dcfSSatish Balay .vb
2198441846f8SBarry Smith    TaoRegister("my_solver",MySolverCreate);
2199a7e14dcfSSatish Balay .ve
2200a7e14dcfSSatish Balay 
2201a7e14dcfSSatish Balay    Then, your solver can be chosen with the procedural interface via
2202a7e14dcfSSatish Balay $     TaoSetType(tao,"my_solver")
2203a7e14dcfSSatish Balay    or at runtime via the option
220458417fe7SBarry Smith $     -tao_type my_solver
2205a7e14dcfSSatish Balay 
2206a7e14dcfSSatish Balay    Level: advanced
2207a7e14dcfSSatish Balay 
2208db781477SPatrick Sanan .seealso: `TaoRegisterAll()`, `TaoRegisterDestroy()`
2209a7e14dcfSSatish Balay M*/
2210441846f8SBarry Smith PetscErrorCode TaoRegister(const char sname[], PetscErrorCode (*func)(Tao))
2211a7e14dcfSSatish Balay {
2212a7e14dcfSSatish Balay   PetscFunctionBegin;
22139566063dSJacob Faibussowitsch   PetscCall(TaoInitializePackage());
22149566063dSJacob Faibussowitsch   PetscCall(PetscFunctionListAdd(&TaoList,sname,(void (*)(void))func));
2215a7e14dcfSSatish Balay   PetscFunctionReturn(0);
2216a7e14dcfSSatish Balay }
2217a7e14dcfSSatish Balay 
2218a7e14dcfSSatish Balay /*@C
2219441846f8SBarry Smith    TaoRegisterDestroy - Frees the list of minimization solvers that were
2220441846f8SBarry Smith    registered by TaoRegisterDynamic().
2221a7e14dcfSSatish Balay 
2222a7e14dcfSSatish Balay    Not Collective
2223a7e14dcfSSatish Balay 
2224a7e14dcfSSatish Balay    Level: advanced
2225a7e14dcfSSatish Balay 
2226db781477SPatrick Sanan .seealso: `TaoRegisterAll()`, `TaoRegister()`
2227a7e14dcfSSatish Balay @*/
2228441846f8SBarry Smith PetscErrorCode TaoRegisterDestroy(void)
2229a7e14dcfSSatish Balay {
2230a7e14dcfSSatish Balay   PetscFunctionBegin;
22319566063dSJacob Faibussowitsch   PetscCall(PetscFunctionListDestroy(&TaoList));
2232441846f8SBarry Smith   TaoRegisterAllCalled = PETSC_FALSE;
2233a7e14dcfSSatish Balay   PetscFunctionReturn(0);
2234a7e14dcfSSatish Balay }
223547a47007SBarry Smith 
22368931d482SJason Sarich /*@
22378931d482SJason Sarich    TaoGetIterationNumber - Gets the number of Tao iterations completed
22388931d482SJason Sarich    at this time.
22398931d482SJason Sarich 
22408931d482SJason Sarich    Not Collective
22418931d482SJason Sarich 
22428931d482SJason Sarich    Input Parameter:
22438931d482SJason Sarich .  tao - Tao context
22448931d482SJason Sarich 
22458931d482SJason Sarich    Output Parameter:
22468931d482SJason Sarich .  iter - iteration number
22478931d482SJason Sarich 
22488931d482SJason Sarich    Notes:
22498931d482SJason Sarich    For example, during the computation of iteration 2 this would return 1.
22508931d482SJason Sarich 
22518931d482SJason Sarich    Level: intermediate
22528931d482SJason Sarich 
2253db781477SPatrick Sanan .seealso: `TaoGetLinearSolveIterations()`, `TaoGetResidualNorm()`, `TaoGetObjective()`
22548931d482SJason Sarich @*/
22558931d482SJason Sarich PetscErrorCode  TaoGetIterationNumber(Tao tao,PetscInt *iter)
22568931d482SJason Sarich {
22578931d482SJason Sarich   PetscFunctionBegin;
22588931d482SJason Sarich   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
22598931d482SJason Sarich   PetscValidIntPointer(iter,2);
22608931d482SJason Sarich   *iter = tao->niter;
22618931d482SJason Sarich   PetscFunctionReturn(0);
22628931d482SJason Sarich }
22638931d482SJason Sarich 
22648931d482SJason Sarich /*@
226579f5d8caSBarry Smith    TaoGetResidualNorm - Gets the current value of the norm of the residual
226679f5d8caSBarry Smith    at this time.
226779f5d8caSBarry Smith 
226879f5d8caSBarry Smith    Not Collective
226979f5d8caSBarry Smith 
227079f5d8caSBarry Smith    Input Parameter:
227179f5d8caSBarry Smith .  tao - Tao context
227279f5d8caSBarry Smith 
227379f5d8caSBarry Smith    Output Parameter:
227479f5d8caSBarry Smith .  value - the current value
227579f5d8caSBarry Smith 
227679f5d8caSBarry Smith    Level: intermediate
227779f5d8caSBarry Smith 
227879f5d8caSBarry Smith    Developer Note: This is the 2-norm of the residual, we cannot use TaoGetGradientNorm() because that has
227979f5d8caSBarry Smith                    a different meaning. For some reason Tao sometimes calls the gradient the residual.
228079f5d8caSBarry Smith 
2281db781477SPatrick Sanan .seealso: `TaoGetLinearSolveIterations()`, `TaoGetIterationNumber()`, `TaoGetObjective()`
228279f5d8caSBarry Smith @*/
228379f5d8caSBarry Smith PetscErrorCode TaoGetResidualNorm(Tao tao,PetscReal *value)
228479f5d8caSBarry Smith {
228579f5d8caSBarry Smith   PetscFunctionBegin;
228679f5d8caSBarry Smith   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
228779f5d8caSBarry Smith   PetscValidRealPointer(value,2);
228879f5d8caSBarry Smith   *value = tao->residual;
228979f5d8caSBarry Smith   PetscFunctionReturn(0);
229079f5d8caSBarry Smith }
229179f5d8caSBarry Smith 
229279f5d8caSBarry Smith /*@
22938931d482SJason Sarich    TaoSetIterationNumber - Sets the current iteration number.
22948931d482SJason Sarich 
229594511df7SStefano Zampini    Logically Collective on Tao
22968931d482SJason Sarich 
2297d8d19677SJose E. Roman    Input Parameters:
2298a2b725a8SWilliam Gropp +  tao - Tao context
2299a2b725a8SWilliam Gropp -  iter - iteration number
23008931d482SJason Sarich 
23018931d482SJason Sarich    Level: developer
23028931d482SJason Sarich 
2303db781477SPatrick Sanan .seealso: `TaoGetLinearSolveIterations()`
23048931d482SJason Sarich @*/
23058931d482SJason Sarich PetscErrorCode  TaoSetIterationNumber(Tao tao,PetscInt iter)
23068931d482SJason Sarich {
23078931d482SJason Sarich   PetscFunctionBegin;
23088931d482SJason Sarich   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
230994511df7SStefano Zampini   PetscValidLogicalCollectiveInt(tao,iter,2);
23109566063dSJacob Faibussowitsch   PetscCall(PetscObjectSAWsTakeAccess((PetscObject)tao));
23118931d482SJason Sarich   tao->niter = iter;
23129566063dSJacob Faibussowitsch   PetscCall(PetscObjectSAWsGrantAccess((PetscObject)tao));
23138931d482SJason Sarich   PetscFunctionReturn(0);
23148931d482SJason Sarich }
23158931d482SJason Sarich 
23168931d482SJason Sarich /*@
23178931d482SJason Sarich    TaoGetTotalIterationNumber - Gets the total number of Tao iterations
23188931d482SJason Sarich    completed. This number keeps accumulating if multiple solves
23198931d482SJason Sarich    are called with the Tao object.
23208931d482SJason Sarich 
23218931d482SJason Sarich    Not Collective
23228931d482SJason Sarich 
23238931d482SJason Sarich    Input Parameter:
23248931d482SJason Sarich .  tao - Tao context
23258931d482SJason Sarich 
23268931d482SJason Sarich    Output Parameter:
23278931d482SJason Sarich .  iter - iteration number
23288931d482SJason Sarich 
23298931d482SJason Sarich    Notes:
23308931d482SJason Sarich    The total iteration count is updated after each solve, if there is a current
23318931d482SJason Sarich    TaoSolve() in progress then those iterations are not yet counted.
23328931d482SJason Sarich 
23338931d482SJason Sarich    Level: intermediate
23348931d482SJason Sarich 
2335db781477SPatrick Sanan .seealso: `TaoGetLinearSolveIterations()`
23368931d482SJason Sarich @*/
23378931d482SJason Sarich PetscErrorCode  TaoGetTotalIterationNumber(Tao tao,PetscInt *iter)
23388931d482SJason Sarich {
23398931d482SJason Sarich   PetscFunctionBegin;
23408931d482SJason Sarich   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
23418931d482SJason Sarich   PetscValidIntPointer(iter,2);
23428931d482SJason Sarich   *iter = tao->ntotalits;
23438931d482SJason Sarich   PetscFunctionReturn(0);
23448931d482SJason Sarich }
23458931d482SJason Sarich 
23468931d482SJason Sarich /*@
23478931d482SJason Sarich    TaoSetTotalIterationNumber - Sets the current total iteration number.
23488931d482SJason Sarich 
234994511df7SStefano Zampini    Logically Collective on Tao
23508931d482SJason Sarich 
2351d8d19677SJose E. Roman    Input Parameters:
2352a2b725a8SWilliam Gropp +  tao - Tao context
2353a2b725a8SWilliam Gropp -  iter - iteration number
23548931d482SJason Sarich 
23558931d482SJason Sarich    Level: developer
23568931d482SJason Sarich 
2357db781477SPatrick Sanan .seealso: `TaoGetLinearSolveIterations()`
23588931d482SJason Sarich @*/
23598931d482SJason Sarich PetscErrorCode  TaoSetTotalIterationNumber(Tao tao,PetscInt iter)
23608931d482SJason Sarich {
23618931d482SJason Sarich   PetscFunctionBegin;
23628931d482SJason Sarich   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
236394511df7SStefano Zampini   PetscValidLogicalCollectiveInt(tao,iter,2);
23649566063dSJacob Faibussowitsch   PetscCall(PetscObjectSAWsTakeAccess((PetscObject)tao));
23658931d482SJason Sarich   tao->ntotalits = iter;
23669566063dSJacob Faibussowitsch   PetscCall(PetscObjectSAWsGrantAccess((PetscObject)tao));
23678931d482SJason Sarich   PetscFunctionReturn(0);
23688931d482SJason Sarich }
23698931d482SJason Sarich 
2370a7e14dcfSSatish Balay /*@
2371e4cb33bbSBarry Smith   TaoSetConvergedReason - Sets the termination flag on a Tao object
2372a7e14dcfSSatish Balay 
2373441846f8SBarry Smith   Logically Collective on Tao
2374a7e14dcfSSatish Balay 
2375a7e14dcfSSatish Balay   Input Parameters:
2376441846f8SBarry Smith + tao - the Tao context
2377a7e14dcfSSatish Balay - reason - one of
2378a7e14dcfSSatish Balay $     TAO_CONVERGED_ATOL (2),
2379a7e14dcfSSatish Balay $     TAO_CONVERGED_RTOL (3),
2380a7e14dcfSSatish Balay $     TAO_CONVERGED_STEPTOL (4),
2381a7e14dcfSSatish Balay $     TAO_CONVERGED_MINF (5),
2382a7e14dcfSSatish Balay $     TAO_CONVERGED_USER (6),
2383a7e14dcfSSatish Balay $     TAO_DIVERGED_MAXITS (-2),
2384a7e14dcfSSatish Balay $     TAO_DIVERGED_NAN (-4),
2385a7e14dcfSSatish Balay $     TAO_DIVERGED_MAXFCN (-5),
2386a7e14dcfSSatish Balay $     TAO_DIVERGED_LS_FAILURE (-6),
2387a7e14dcfSSatish Balay $     TAO_DIVERGED_TR_REDUCTION (-7),
2388a7e14dcfSSatish Balay $     TAO_DIVERGED_USER (-8),
2389a7e14dcfSSatish Balay $     TAO_CONTINUE_ITERATING (0)
2390a7e14dcfSSatish Balay 
2391a7e14dcfSSatish Balay    Level: intermediate
2392a7e14dcfSSatish Balay 
2393a7e14dcfSSatish Balay @*/
2394e4cb33bbSBarry Smith PetscErrorCode TaoSetConvergedReason(Tao tao, TaoConvergedReason reason)
2395a7e14dcfSSatish Balay {
2396a7e14dcfSSatish Balay   PetscFunctionBegin;
239794511df7SStefano Zampini   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
239894511df7SStefano Zampini   PetscValidLogicalCollectiveEnum(tao,reason,2);
2399a7e14dcfSSatish Balay   tao->reason = reason;
2400a7e14dcfSSatish Balay   PetscFunctionReturn(0);
2401a7e14dcfSSatish Balay }
2402a7e14dcfSSatish Balay 
2403a7e14dcfSSatish Balay /*@
2404e4cb33bbSBarry Smith    TaoGetConvergedReason - Gets the reason the Tao iteration was stopped.
2405a7e14dcfSSatish Balay 
2406a7e14dcfSSatish Balay    Not Collective
2407a7e14dcfSSatish Balay 
2408a7e14dcfSSatish Balay    Input Parameter:
2409441846f8SBarry Smith .  tao - the Tao solver context
2410a7e14dcfSSatish Balay 
2411a7e14dcfSSatish Balay    Output Parameter:
2412a7e14dcfSSatish Balay .  reason - one of
2413a7e14dcfSSatish Balay $  TAO_CONVERGED_GATOL (3)           ||g(X)|| < gatol
2414a7e14dcfSSatish Balay $  TAO_CONVERGED_GRTOL (4)           ||g(X)|| / f(X)  < grtol
2415a7e14dcfSSatish Balay $  TAO_CONVERGED_GTTOL (5)           ||g(X)|| / ||g(X0)|| < gttol
2416a7e14dcfSSatish Balay $  TAO_CONVERGED_STEPTOL (6)         step size small
2417a7e14dcfSSatish Balay $  TAO_CONVERGED_MINF (7)            F < F_min
2418a7e14dcfSSatish Balay $  TAO_CONVERGED_USER (8)            User defined
2419a7e14dcfSSatish Balay $  TAO_DIVERGED_MAXITS (-2)          its > maxits
2420a7e14dcfSSatish Balay $  TAO_DIVERGED_NAN (-4)             Numerical problems
2421a7e14dcfSSatish Balay $  TAO_DIVERGED_MAXFCN (-5)          fevals > max_funcsals
2422a7e14dcfSSatish Balay $  TAO_DIVERGED_LS_FAILURE (-6)      line search failure
2423a7e14dcfSSatish Balay $  TAO_DIVERGED_TR_REDUCTION (-7)    trust region failure
2424a7e14dcfSSatish Balay $  TAO_DIVERGED_USER(-8)             (user defined)
2425a7e14dcfSSatish Balay $  TAO_CONTINUE_ITERATING (0)
2426a7e14dcfSSatish Balay 
2427a7e14dcfSSatish Balay    where
2428a7e14dcfSSatish Balay +  X - current solution
2429a7e14dcfSSatish Balay .  X0 - initial guess
2430a7e14dcfSSatish Balay .  f(X) - current function value
2431a7e14dcfSSatish Balay .  f(X*) - true solution (estimated)
2432a7e14dcfSSatish Balay .  g(X) - current gradient
2433a7e14dcfSSatish Balay .  its - current iterate number
2434a7e14dcfSSatish Balay .  maxits - maximum number of iterates
2435a7e14dcfSSatish Balay .  fevals - number of function evaluations
2436a7e14dcfSSatish Balay -  max_funcsals - maximum number of function evaluations
2437a7e14dcfSSatish Balay 
2438a7e14dcfSSatish Balay    Level: intermediate
2439a7e14dcfSSatish Balay 
2440db781477SPatrick Sanan .seealso: `TaoSetConvergenceTest()`, `TaoSetTolerances()`
2441a7e14dcfSSatish Balay 
2442a7e14dcfSSatish Balay @*/
2443e4cb33bbSBarry Smith PetscErrorCode TaoGetConvergedReason(Tao tao, TaoConvergedReason *reason)
2444a7e14dcfSSatish Balay {
2445a7e14dcfSSatish Balay   PetscFunctionBegin;
2446441846f8SBarry Smith   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
2447a7e14dcfSSatish Balay   PetscValidPointer(reason,2);
2448a7e14dcfSSatish Balay   *reason = tao->reason;
2449a7e14dcfSSatish Balay   PetscFunctionReturn(0);
2450a7e14dcfSSatish Balay }
2451a7e14dcfSSatish Balay 
2452a7e14dcfSSatish Balay /*@
2453a7e14dcfSSatish Balay    TaoGetSolutionStatus - Get the current iterate, objective value,
2454a7e14dcfSSatish Balay    residual, infeasibility, and termination
2455a7e14dcfSSatish Balay 
2456a7e14dcfSSatish Balay    Not Collective
2457a7e14dcfSSatish Balay 
2458f899ff85SJose E. Roman    Input Parameter:
2459441846f8SBarry Smith .  tao - the Tao context
2460a7e14dcfSSatish Balay 
2461a7e14dcfSSatish Balay    Output Parameters:
2462a7e14dcfSSatish Balay +  iterate - the current iterate number (>=0)
2463a7e14dcfSSatish Balay .  f - the current function value
246458417fe7SBarry Smith .  gnorm - the square of the gradient norm, duality gap, or other measure indicating distance from optimality.
2465a7e14dcfSSatish Balay .  cnorm - the infeasibility of the current solution with regard to the constraints.
2466a7e14dcfSSatish Balay .  xdiff - the step length or trust region radius of the most recent iterate.
2467a7e14dcfSSatish Balay -  reason - The termination reason, which can equal TAO_CONTINUE_ITERATING
2468a7e14dcfSSatish Balay 
2469a7e14dcfSSatish Balay    Level: intermediate
2470a7e14dcfSSatish Balay 
2471a7e14dcfSSatish Balay    Note:
2472a7e14dcfSSatish Balay    TAO returns the values set by the solvers in the routine TaoMonitor().
2473a7e14dcfSSatish Balay 
2474a7e14dcfSSatish Balay    Note:
247547a47007SBarry Smith    If any of the output arguments are set to NULL, no corresponding value will be returned.
2476a7e14dcfSSatish Balay 
2477db781477SPatrick Sanan .seealso: `TaoMonitor()`, `TaoGetConvergedReason()`
2478a7e14dcfSSatish Balay @*/
2479e4cb33bbSBarry Smith PetscErrorCode TaoGetSolutionStatus(Tao tao, PetscInt *its, PetscReal *f, PetscReal *gnorm, PetscReal *cnorm, PetscReal *xdiff, TaoConvergedReason *reason)
2480a7e14dcfSSatish Balay {
2481a7e14dcfSSatish Balay   PetscFunctionBegin;
248294511df7SStefano Zampini   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
2483a7e14dcfSSatish Balay   if (its) *its = tao->niter;
2484a7e14dcfSSatish Balay   if (f) *f = tao->fc;
2485a7e14dcfSSatish Balay   if (gnorm) *gnorm = tao->residual;
2486a7e14dcfSSatish Balay   if (cnorm) *cnorm = tao->cnorm;
2487a7e14dcfSSatish Balay   if (reason) *reason = tao->reason;
2488a7e14dcfSSatish Balay   if (xdiff) *xdiff = tao->step;
2489a7e14dcfSSatish Balay   PetscFunctionReturn(0);
2490a7e14dcfSSatish Balay }
2491a7e14dcfSSatish Balay 
2492a7e14dcfSSatish Balay /*@C
2493441846f8SBarry Smith    TaoGetType - Gets the current Tao algorithm.
2494a7e14dcfSSatish Balay 
2495a7e14dcfSSatish Balay    Not Collective
2496a7e14dcfSSatish Balay 
2497a7e14dcfSSatish Balay    Input Parameter:
2498441846f8SBarry Smith .  tao - the Tao solver context
2499a7e14dcfSSatish Balay 
2500a7e14dcfSSatish Balay    Output Parameter:
2501441846f8SBarry Smith .  type - Tao method
2502a7e14dcfSSatish Balay 
2503a7e14dcfSSatish Balay    Level: intermediate
2504a7e14dcfSSatish Balay 
2505a7e14dcfSSatish Balay @*/
2506b625d6c7SJed Brown PetscErrorCode TaoGetType(Tao tao,TaoType *type)
2507a7e14dcfSSatish Balay {
2508a7e14dcfSSatish Balay   PetscFunctionBegin;
2509441846f8SBarry Smith   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
2510a7e14dcfSSatish Balay   PetscValidPointer(type,2);
2511a7e14dcfSSatish Balay   *type = ((PetscObject)tao)->type_name;
2512a7e14dcfSSatish Balay   PetscFunctionReturn(0);
2513a7e14dcfSSatish Balay }
2514a7e14dcfSSatish Balay 
2515a7e14dcfSSatish Balay /*@C
2516a7e14dcfSSatish Balay   TaoMonitor - Monitor the solver and the current solution.  This
2517a7e14dcfSSatish Balay   routine will record the iteration number and residual statistics,
2518a7e14dcfSSatish Balay   call any monitors specified by the user, and calls the convergence-check routine.
2519a7e14dcfSSatish Balay 
2520a7e14dcfSSatish Balay    Input Parameters:
2521441846f8SBarry Smith +  tao - the Tao context
2522a7e14dcfSSatish Balay .  its - the current iterate number (>=0)
2523a7e14dcfSSatish Balay .  f - the current objective function value
252458417fe7SBarry Smith .  res - the gradient norm, square root of the duality gap, or other measure indicating distince from optimality.  This measure will be recorded and
2525a7e14dcfSSatish Balay           used for some termination tests.
2526a7e14dcfSSatish Balay .  cnorm - the infeasibility of the current solution with regard to the constraints.
2527a7e14dcfSSatish Balay -  steplength - multiple of the step direction added to the previous iterate.
2528a7e14dcfSSatish Balay 
2529a7e14dcfSSatish Balay    Output Parameters:
2530a7e14dcfSSatish Balay .  reason - The termination reason, which can equal TAO_CONTINUE_ITERATING
2531a7e14dcfSSatish Balay 
2532a7e14dcfSSatish Balay    Options Database Key:
2533a7e14dcfSSatish Balay .  -tao_monitor - Use the default monitor, which prints statistics to standard output
2534a7e14dcfSSatish Balay 
2535db781477SPatrick Sanan .seealso `TaoGetConvergedReason()`, `TaoMonitorDefault()`, `TaoSetMonitor()`
2536a7e14dcfSSatish Balay 
2537a7e14dcfSSatish Balay    Level: developer
2538a7e14dcfSSatish Balay 
2539a7e14dcfSSatish Balay @*/
25403ecd9318SAlp Dener PetscErrorCode TaoMonitor(Tao tao, PetscInt its, PetscReal f, PetscReal res, PetscReal cnorm, PetscReal steplength)
2541a7e14dcfSSatish Balay {
254247a47007SBarry Smith   PetscInt       i;
254347a47007SBarry Smith 
2544a7e14dcfSSatish Balay   PetscFunctionBegin;
2545441846f8SBarry Smith   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
2546a7e14dcfSSatish Balay   tao->fc = f;
2547a7e14dcfSSatish Balay   tao->residual = res;
2548a7e14dcfSSatish Balay   tao->cnorm = cnorm;
2549a7e14dcfSSatish Balay   tao->step = steplength;
2550e52336cbSBarry Smith   if (!its) {
255194511df7SStefano Zampini     tao->cnorm0 = cnorm;
255294511df7SStefano Zampini     tao->gnorm0 = res;
2553a7e14dcfSSatish Balay   }
25543c859ba3SBarry Smith   PetscCheck(!PetscIsInfOrNanReal(f) && !PetscIsInfOrNanReal(res),PetscObjectComm((PetscObject)tao),PETSC_ERR_USER, "User provided compute function generated Inf or NaN");
2555a7e14dcfSSatish Balay   for (i=0;i<tao->numbermonitors;i++) {
25569566063dSJacob Faibussowitsch     PetscCall((*tao->monitor[i])(tao,tao->monitorcontext[i]));
2557a7e14dcfSSatish Balay   }
2558a7e14dcfSSatish Balay   PetscFunctionReturn(0);
2559a7e14dcfSSatish Balay }
2560a7e14dcfSSatish Balay 
2561a7e14dcfSSatish Balay /*@
2562ae93cb3cSJason Sarich    TaoSetConvergenceHistory - Sets the array used to hold the convergence history.
2563a7e14dcfSSatish Balay 
2564441846f8SBarry Smith    Logically Collective on Tao
2565a7e14dcfSSatish Balay 
2566a7e14dcfSSatish Balay    Input Parameters:
2567441846f8SBarry Smith +  tao - the Tao solver context
2568a7e14dcfSSatish Balay .  obj   - array to hold objective value history
2569a7e14dcfSSatish Balay .  resid - array to hold residual history
2570a7e14dcfSSatish Balay .  cnorm - array to hold constraint violation history
2571be1558d8SJason Sarich .  lits - integer array holds the number of linear iterations for each Tao iteration
2572a7e14dcfSSatish Balay .  na  - size of obj, resid, and cnorm
2573a7e14dcfSSatish Balay -  reset - PetscTrue indicates each new minimization resets the history counter to zero,
2574a7e14dcfSSatish Balay            else it continues storing new values for new minimizations after the old ones
2575a7e14dcfSSatish Balay 
2576a7e14dcfSSatish Balay    Notes:
2577a7e14dcfSSatish Balay    If set, TAO will fill the given arrays with the indicated
2578ae93cb3cSJason Sarich    information at each iteration.  If 'obj','resid','cnorm','lits' are
2579ae93cb3cSJason Sarich    *all* NULL then space (using size na, or 1000 if na is PETSC_DECIDE or
2580ae93cb3cSJason Sarich    PETSC_DEFAULT) is allocated for the history.
2581be1558d8SJason Sarich    If not all are NULL, then only the non-NULL information categories
2582ae93cb3cSJason Sarich    will be stored, the others will be ignored.
2583ae93cb3cSJason Sarich 
2584ae93cb3cSJason Sarich    Any convergence information after iteration number 'na' will not be stored.
2585a7e14dcfSSatish Balay 
2586a7e14dcfSSatish Balay    This routine is useful, e.g., when running a code for purposes
2587a7e14dcfSSatish Balay    of accurate performance monitoring, when no I/O should be done
2588a7e14dcfSSatish Balay    during the section of code that is being timed.
2589a7e14dcfSSatish Balay 
2590a7e14dcfSSatish Balay    Level: intermediate
2591a7e14dcfSSatish Balay 
2592db781477SPatrick Sanan .seealso: `TaoGetConvergenceHistory()`
2593a7e14dcfSSatish Balay 
2594a7e14dcfSSatish Balay @*/
25951b266c99SBarry Smith PetscErrorCode TaoSetConvergenceHistory(Tao tao, PetscReal obj[], PetscReal resid[], PetscReal cnorm[], PetscInt lits[], PetscInt na,PetscBool reset)
2596a7e14dcfSSatish Balay {
2597a7e14dcfSSatish Balay   PetscFunctionBegin;
2598441846f8SBarry Smith   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
2599064a246eSJacob Faibussowitsch   if (obj) PetscValidRealPointer(obj,2);
2600064a246eSJacob Faibussowitsch   if (resid) PetscValidRealPointer(resid,3);
2601064a246eSJacob Faibussowitsch   if (cnorm) PetscValidRealPointer(cnorm,4);
2602ae93cb3cSJason Sarich   if (lits) PetscValidIntPointer(lits,5);
2603ae93cb3cSJason Sarich 
2604ae93cb3cSJason Sarich   if (na == PETSC_DECIDE || na == PETSC_DEFAULT) na = 1000;
2605ae93cb3cSJason Sarich   if (!obj && !resid && !cnorm && !lits) {
26069566063dSJacob Faibussowitsch     PetscCall(PetscCalloc4(na,&obj,na,&resid,na,&cnorm,na,&lits));
2607ae93cb3cSJason Sarich     tao->hist_malloc = PETSC_TRUE;
2608ae93cb3cSJason Sarich   }
2609ae93cb3cSJason Sarich 
2610a7e14dcfSSatish Balay   tao->hist_obj = obj;
2611a7e14dcfSSatish Balay   tao->hist_resid = resid;
2612a7e14dcfSSatish Balay   tao->hist_cnorm = cnorm;
2613ae93cb3cSJason Sarich   tao->hist_lits = lits;
2614a7e14dcfSSatish Balay   tao->hist_max   = na;
2615a7e14dcfSSatish Balay   tao->hist_reset = reset;
2616ae93cb3cSJason Sarich   tao->hist_len = 0;
2617a7e14dcfSSatish Balay   PetscFunctionReturn(0);
2618a7e14dcfSSatish Balay }
2619a7e14dcfSSatish Balay 
2620a7e14dcfSSatish Balay /*@C
2621ae93cb3cSJason Sarich    TaoGetConvergenceHistory - Gets the arrays used to hold the convergence history.
2622a7e14dcfSSatish Balay 
2623441846f8SBarry Smith    Collective on Tao
2624a7e14dcfSSatish Balay 
2625a7e14dcfSSatish Balay    Input Parameter:
2626441846f8SBarry Smith .  tao - the Tao context
2627a7e14dcfSSatish Balay 
2628a7e14dcfSSatish Balay    Output Parameters:
2629a7e14dcfSSatish Balay +  obj   - array used to hold objective value history
2630a7e14dcfSSatish Balay .  resid - array used to hold residual history
2631a7e14dcfSSatish Balay .  cnorm - array used to hold constraint violation history
2632ae93cb3cSJason Sarich .  lits  - integer array used to hold linear solver iteration count
2633dd25e5eeSStefano Zampini -  nhist  - size of obj, resid, cnorm, and lits
2634a7e14dcfSSatish Balay 
2635a7e14dcfSSatish Balay    Notes:
2636ae93cb3cSJason Sarich     This routine must be preceded by calls to TaoSetConvergenceHistory()
2637ae93cb3cSJason Sarich     and TaoSolve(), otherwise it returns useless information.
2638ae93cb3cSJason Sarich 
2639a7e14dcfSSatish Balay     The calling sequence for this routine in Fortran is
2640ae93cb3cSJason Sarich $   call TaoGetConvergenceHistory(Tao tao, PetscInt nhist, PetscErrorCode ierr)
2641a7e14dcfSSatish Balay 
2642a7e14dcfSSatish Balay    This routine is useful, e.g., when running a code for purposes
2643a7e14dcfSSatish Balay    of accurate performance monitoring, when no I/O should be done
2644a7e14dcfSSatish Balay    during the section of code that is being timed.
2645a7e14dcfSSatish Balay 
2646a7e14dcfSSatish Balay    Level: advanced
2647a7e14dcfSSatish Balay 
2648db781477SPatrick Sanan .seealso: `TaoSetConvergenceHistory()`
2649a7e14dcfSSatish Balay 
2650a7e14dcfSSatish Balay @*/
2651ae93cb3cSJason Sarich PetscErrorCode TaoGetConvergenceHistory(Tao tao, PetscReal **obj, PetscReal **resid, PetscReal **cnorm, PetscInt **lits, PetscInt *nhist)
2652a7e14dcfSSatish Balay {
2653a7e14dcfSSatish Balay   PetscFunctionBegin;
2654441846f8SBarry Smith   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
2655a7e14dcfSSatish Balay   if (obj)   *obj   = tao->hist_obj;
2656a7e14dcfSSatish Balay   if (cnorm) *cnorm = tao->hist_cnorm;
2657a7e14dcfSSatish Balay   if (resid) *resid = tao->hist_resid;
2658a7e14dcfSSatish Balay   if (nhist) *nhist = tao->hist_len;
2659a7e14dcfSSatish Balay   PetscFunctionReturn(0);
2660a7e14dcfSSatish Balay }
2661a7e14dcfSSatish Balay 
2662a7e14dcfSSatish Balay /*@
2663a7e14dcfSSatish Balay    TaoSetApplicationContext - Sets the optional user-defined context for
2664a7e14dcfSSatish Balay    a solver.
2665a7e14dcfSSatish Balay 
2666441846f8SBarry Smith    Logically Collective on Tao
2667a7e14dcfSSatish Balay 
2668a7e14dcfSSatish Balay    Input Parameters:
2669441846f8SBarry Smith +  tao  - the Tao context
2670a7e14dcfSSatish Balay -  usrP - optional user context
2671a7e14dcfSSatish Balay 
2672a7e14dcfSSatish Balay    Level: intermediate
2673a7e14dcfSSatish Balay 
2674db781477SPatrick Sanan .seealso: `TaoGetApplicationContext()`, `TaoSetApplicationContext()`
2675a7e14dcfSSatish Balay @*/
2676441846f8SBarry Smith PetscErrorCode  TaoSetApplicationContext(Tao tao,void *usrP)
2677a7e14dcfSSatish Balay {
2678a7e14dcfSSatish Balay   PetscFunctionBegin;
2679441846f8SBarry Smith   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
2680a7e14dcfSSatish Balay   tao->user = usrP;
2681a7e14dcfSSatish Balay   PetscFunctionReturn(0);
2682a7e14dcfSSatish Balay }
2683a7e14dcfSSatish Balay 
2684a7e14dcfSSatish Balay /*@
2685a7e14dcfSSatish Balay    TaoGetApplicationContext - Gets the user-defined context for a
2686a7e14dcfSSatish Balay    TAO solvers.
2687a7e14dcfSSatish Balay 
2688a7e14dcfSSatish Balay    Not Collective
2689a7e14dcfSSatish Balay 
2690a7e14dcfSSatish Balay    Input Parameter:
2691441846f8SBarry Smith .  tao  - Tao context
2692a7e14dcfSSatish Balay 
2693a7e14dcfSSatish Balay    Output Parameter:
2694a7e14dcfSSatish Balay .  usrP - user context
2695a7e14dcfSSatish Balay 
2696a7e14dcfSSatish Balay    Level: intermediate
2697a7e14dcfSSatish Balay 
2698db781477SPatrick Sanan .seealso: `TaoSetApplicationContext()`
2699a7e14dcfSSatish Balay @*/
2700441846f8SBarry Smith PetscErrorCode  TaoGetApplicationContext(Tao tao,void *usrP)
2701a7e14dcfSSatish Balay {
2702a7e14dcfSSatish Balay   PetscFunctionBegin;
2703441846f8SBarry Smith   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
270494511df7SStefano Zampini   PetscValidPointer(usrP,2);
2705a7e14dcfSSatish Balay   *(void**)usrP = tao->user;
2706a7e14dcfSSatish Balay   PetscFunctionReturn(0);
2707a7e14dcfSSatish Balay }
2708a9603a14SPatrick Farrell 
2709a9603a14SPatrick Farrell /*@
2710a9603a14SPatrick Farrell    TaoSetGradientNorm - Sets the matrix used to define the inner product that measures the size of the gradient.
2711a9603a14SPatrick Farrell 
2712a9603a14SPatrick Farrell    Collective on tao
2713a9603a14SPatrick Farrell 
2714a9603a14SPatrick Farrell    Input Parameters:
2715a9603a14SPatrick Farrell +  tao  - the Tao context
2716a9603a14SPatrick Farrell -  M    - gradient norm
2717a9603a14SPatrick Farrell 
2718a9603a14SPatrick Farrell    Level: beginner
2719a9603a14SPatrick Farrell 
2720db781477SPatrick Sanan .seealso: `TaoGetGradientNorm()`, `TaoGradientNorm()`
2721a9603a14SPatrick Farrell @*/
2722a9603a14SPatrick Farrell PetscErrorCode  TaoSetGradientNorm(Tao tao, Mat M)
2723a9603a14SPatrick Farrell {
2724a9603a14SPatrick Farrell   PetscFunctionBegin;
2725a9603a14SPatrick Farrell   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
272694511df7SStefano Zampini   PetscValidHeaderSpecific(M,MAT_CLASSID,2);
27279566063dSJacob Faibussowitsch   PetscCall(PetscObjectReference((PetscObject)M));
27289566063dSJacob Faibussowitsch   PetscCall(MatDestroy(&tao->gradient_norm));
27299566063dSJacob Faibussowitsch   PetscCall(VecDestroy(&tao->gradient_norm_tmp));
2730a9603a14SPatrick Farrell   tao->gradient_norm = M;
27319566063dSJacob Faibussowitsch   PetscCall(MatCreateVecs(M, NULL, &tao->gradient_norm_tmp));
2732a9603a14SPatrick Farrell   PetscFunctionReturn(0);
2733a9603a14SPatrick Farrell }
2734a9603a14SPatrick Farrell 
2735a9603a14SPatrick Farrell /*@
2736a9603a14SPatrick Farrell    TaoGetGradientNorm - Returns the matrix used to define the inner product for measuring the size of the gradient.
2737a9603a14SPatrick Farrell 
2738a9603a14SPatrick Farrell    Not Collective
2739a9603a14SPatrick Farrell 
2740a9603a14SPatrick Farrell    Input Parameter:
2741a9603a14SPatrick Farrell .  tao  - Tao context
2742a9603a14SPatrick Farrell 
2743a9603a14SPatrick Farrell    Output Parameter:
2744a9603a14SPatrick Farrell .  M - gradient norm
2745a9603a14SPatrick Farrell 
2746a9603a14SPatrick Farrell    Level: beginner
2747a9603a14SPatrick Farrell 
2748db781477SPatrick Sanan .seealso: `TaoSetGradientNorm()`, `TaoGradientNorm()`
2749a9603a14SPatrick Farrell @*/
2750a9603a14SPatrick Farrell PetscErrorCode  TaoGetGradientNorm(Tao tao, Mat *M)
2751a9603a14SPatrick Farrell {
2752a9603a14SPatrick Farrell   PetscFunctionBegin;
2753a9603a14SPatrick Farrell   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
275494511df7SStefano Zampini   PetscValidPointer(M,2);
2755a9603a14SPatrick Farrell   *M = tao->gradient_norm;
2756a9603a14SPatrick Farrell   PetscFunctionReturn(0);
2757a9603a14SPatrick Farrell }
2758a9603a14SPatrick Farrell 
275905aca2cdSSatish Balay /*@C
2760a9603a14SPatrick Farrell    TaoGradientNorm - Compute the norm with respect to the inner product the user has set.
2761a9603a14SPatrick Farrell 
2762a9603a14SPatrick Farrell    Collective on tao
2763a9603a14SPatrick Farrell 
2764d8d19677SJose E. Roman    Input Parameters:
27652653c764SPierre Jolivet +  tao      - the Tao context
2766a9603a14SPatrick Farrell .  gradient - the gradient to be computed
27672653c764SPierre Jolivet -  norm     - the norm type
2768a9603a14SPatrick Farrell 
2769a9603a14SPatrick Farrell    Output Parameter:
2770a9603a14SPatrick Farrell .  gnorm    - the gradient norm
2771a9603a14SPatrick Farrell 
2772a9603a14SPatrick Farrell    Level: developer
2773a9603a14SPatrick Farrell 
2774db781477SPatrick Sanan .seealso: `TaoSetGradientNorm()`, `TaoGetGradientNorm()`
2775a9603a14SPatrick Farrell @*/
2776a9603a14SPatrick Farrell PetscErrorCode  TaoGradientNorm(Tao tao, Vec gradient, NormType type, PetscReal *gnorm)
2777a9603a14SPatrick Farrell {
2778a9603a14SPatrick Farrell   PetscFunctionBegin;
27798854b543SStefano Zampini   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
27808854b543SStefano Zampini   PetscValidHeaderSpecific(gradient,VEC_CLASSID,2);
27818854b543SStefano Zampini   PetscValidLogicalCollectiveEnum(tao,type,3);
278294511df7SStefano Zampini   PetscValidRealPointer(gnorm,4);
2783a9603a14SPatrick Farrell   if (tao->gradient_norm) {
2784680e2bc7SPatrick Farrell     PetscScalar gnorms;
2785680e2bc7SPatrick Farrell 
27863c859ba3SBarry Smith     PetscCheck(type == NORM_2,PetscObjectComm((PetscObject)gradient), PETSC_ERR_ARG_WRONG, "Norm type must be NORM_2 if an inner product for the gradient norm is set.");
27879566063dSJacob Faibussowitsch     PetscCall(MatMult(tao->gradient_norm, gradient, tao->gradient_norm_tmp));
27889566063dSJacob Faibussowitsch     PetscCall(VecDot(gradient, tao->gradient_norm_tmp, &gnorms));
27897bb79937SPatrick Farrell     *gnorm = PetscRealPart(PetscSqrtScalar(gnorms));
2790a9603a14SPatrick Farrell   } else {
27919566063dSJacob Faibussowitsch     PetscCall(VecNorm(gradient, type, gnorm));
2792a9603a14SPatrick Farrell   }
2793a9603a14SPatrick Farrell   PetscFunctionReturn(0);
2794a9603a14SPatrick Farrell }
2795a9603a14SPatrick Farrell 
2796e882e171SHong Zhang /*@C
2797e882e171SHong Zhang    TaoMonitorDrawCtxCreate - Creates the monitor context for TaoMonitorDrawCtx
2798a9603a14SPatrick Farrell 
2799e882e171SHong Zhang    Collective on Tao
2800e882e171SHong Zhang 
2801e882e171SHong Zhang    Output Patameter:
2802e882e171SHong Zhang .    ctx - the monitor context
2803e882e171SHong Zhang 
2804e882e171SHong Zhang    Options Database:
2805e882e171SHong Zhang .   -tao_draw_solution_initial - show initial guess as well as current solution
2806e882e171SHong Zhang 
2807e882e171SHong Zhang    Level: intermediate
2808e882e171SHong Zhang 
2809db781477SPatrick Sanan .seealso: `TaoMonitorSet()`, `TaoMonitorDefault()`, `VecView()`, `TaoMonitorDrawCtx()`
2810e882e171SHong Zhang @*/
2811e882e171SHong Zhang PetscErrorCode  TaoMonitorDrawCtxCreate(MPI_Comm comm,const char host[],const char label[],int x,int y,int m,int n,PetscInt howoften,TaoMonitorDrawCtx *ctx)
2812e882e171SHong Zhang {
2813e882e171SHong Zhang   PetscFunctionBegin;
28149566063dSJacob Faibussowitsch   PetscCall(PetscNew(ctx));
28159566063dSJacob Faibussowitsch   PetscCall(PetscViewerDrawOpen(comm,host,label,x,y,m,n,&(*ctx)->viewer));
28169566063dSJacob Faibussowitsch   PetscCall(PetscViewerSetFromOptions((*ctx)->viewer));
2817e882e171SHong Zhang   (*ctx)->howoften = howoften;
2818e882e171SHong Zhang   PetscFunctionReturn(0);
2819e882e171SHong Zhang }
2820e882e171SHong Zhang 
2821e882e171SHong Zhang /*@C
2822e882e171SHong Zhang    TaoMonitorDrawCtxDestroy - Destroys the monitor context for TaoMonitorDrawSolution()
2823e882e171SHong Zhang 
2824e882e171SHong Zhang    Collective on Tao
2825e882e171SHong Zhang 
2826e882e171SHong Zhang    Input Parameters:
2827e882e171SHong Zhang .    ctx - the monitor context
2828e882e171SHong Zhang 
2829e882e171SHong Zhang    Level: intermediate
2830e882e171SHong Zhang 
2831db781477SPatrick Sanan .seealso: `TaoMonitorSet()`, `TaoMonitorDefault()`, `VecView()`, `TaoMonitorDrawSolution()`
2832e882e171SHong Zhang @*/
2833e882e171SHong Zhang PetscErrorCode  TaoMonitorDrawCtxDestroy(TaoMonitorDrawCtx *ictx)
2834e882e171SHong Zhang {
2835e882e171SHong Zhang   PetscFunctionBegin;
28369566063dSJacob Faibussowitsch   PetscCall(PetscViewerDestroy(&(*ictx)->viewer));
28379566063dSJacob Faibussowitsch   PetscCall(PetscFree(*ictx));
2838e882e171SHong Zhang   PetscFunctionReturn(0);
2839e882e171SHong Zhang }
2840