xref: /petsc/src/tao/interface/taosolver.c (revision e1e80dc898c3d5fa028e909e947004e745fb92d9)
1441846f8SBarry Smith #define TAO_DLL
2a7e14dcfSSatish Balay 
3af0996ceSBarry Smith #include <petsc/private/taoimpl.h> /*I "petsctao.h" I*/
4a7e14dcfSSatish Balay 
5441846f8SBarry Smith PetscBool TaoRegisterAllCalled = PETSC_FALSE;
6441846f8SBarry Smith PetscFunctionList TaoList = NULL;
7a7e14dcfSSatish Balay 
8441846f8SBarry Smith PetscClassId TAO_CLASSID;
90ebee16dSLisandro Dalcin 
100ebee16dSLisandro Dalcin PetscLogEvent TAO_Solve;
110ebee16dSLisandro Dalcin PetscLogEvent TAO_ObjectiveEval;
120ebee16dSLisandro Dalcin PetscLogEvent TAO_GradientEval;
130ebee16dSLisandro Dalcin PetscLogEvent TAO_ObjGradEval;
140ebee16dSLisandro Dalcin PetscLogEvent TAO_HessianEval;
150ebee16dSLisandro Dalcin PetscLogEvent TAO_JacobianEval;
160ebee16dSLisandro Dalcin PetscLogEvent TAO_ConstraintsEval;
17a7e14dcfSSatish Balay 
185b0d99aeSBarry Smith const char *TaoSubSetTypes[] = {"subvec","mask","matrixfree","TaoSubSetType","TAO_SUBSET_",0};
19a7e14dcfSSatish Balay 
20e882e171SHong Zhang struct _n_TaoMonitorDrawCtx {
21e882e171SHong Zhang   PetscViewer viewer;
22e882e171SHong Zhang   PetscInt    howoften;  /* when > 0 uses iteration % howoften, when negative only final solution plotted */
23e882e171SHong Zhang };
24e882e171SHong Zhang 
25a7e14dcfSSatish Balay /*@
26a7e14dcfSSatish Balay   TaoCreate - Creates a TAO solver
27a7e14dcfSSatish Balay 
28a7e14dcfSSatish Balay   Collective on MPI_Comm
29a7e14dcfSSatish Balay 
30a7e14dcfSSatish Balay   Input Parameter:
31a7e14dcfSSatish Balay . comm - MPI communicator
32a7e14dcfSSatish Balay 
33a7e14dcfSSatish Balay   Output Parameter:
34441846f8SBarry Smith . newtao - the new Tao context
35a7e14dcfSSatish Balay 
36a7e14dcfSSatish Balay   Available methods include:
3758417fe7SBarry Smith +    nls - Newton's method with line search for unconstrained minimization
3858417fe7SBarry Smith .    ntr - Newton's method with trust region for unconstrained minimization
3958417fe7SBarry Smith .    ntl - Newton's method with trust region, line search for unconstrained minimization
4058417fe7SBarry Smith .    lmvm - Limited memory variable metric method for unconstrained minimization
4158417fe7SBarry Smith .    cg - Nonlinear conjugate gradient method for unconstrained minimization
4258417fe7SBarry Smith .    nm - Nelder-Mead algorithm for derivate-free unconstrained minimization
4358417fe7SBarry Smith .    tron - Newton Trust Region method for bound constrained minimization
4458417fe7SBarry Smith .    gpcg - Newton Trust Region method for quadratic bound constrained minimization
4558417fe7SBarry Smith .    blmvm - Limited memory variable metric method for bound constrained minimization
4658417fe7SBarry Smith .    lcl - Linearly constrained Lagrangian method for pde-constrained minimization
4758417fe7SBarry Smith -    pounders - Model-based algorithm for nonlinear least squares
48a7e14dcfSSatish Balay 
49a7e14dcfSSatish Balay    Options Database Keys:
5058417fe7SBarry Smith .   -tao_type - select which method TAO should use
51a7e14dcfSSatish Balay 
52a7e14dcfSSatish Balay    Level: beginner
53a7e14dcfSSatish Balay 
54a7e14dcfSSatish Balay .seealso: TaoSolve(), TaoDestroy()
55a7e14dcfSSatish Balay @*/
56441846f8SBarry Smith PetscErrorCode TaoCreate(MPI_Comm comm, Tao *newtao)
57a7e14dcfSSatish Balay {
58a7e14dcfSSatish Balay   PetscErrorCode ierr;
59441846f8SBarry Smith   Tao            tao;
60a7e14dcfSSatish Balay 
61a7e14dcfSSatish Balay   PetscFunctionBegin;
62a7e14dcfSSatish Balay   PetscValidPointer(newtao,2);
6347a47007SBarry Smith   *newtao = NULL;
64a7e14dcfSSatish Balay 
65a7e14dcfSSatish Balay   ierr = TaoInitializePackage();CHKERRQ(ierr);
66a7e14dcfSSatish Balay   ierr = TaoLineSearchInitializePackage();CHKERRQ(ierr);
67a7e14dcfSSatish Balay 
6873107ff1SLisandro Dalcin   ierr = PetscHeaderCreate(tao,TAO_CLASSID,"Tao","Optimization solver","Tao",comm,TaoDestroy,TaoView);CHKERRQ(ierr);
69a7e14dcfSSatish Balay   tao->ops->computeobjective=0;
70a7e14dcfSSatish Balay   tao->ops->computeobjectiveandgradient=0;
71a7e14dcfSSatish Balay   tao->ops->computegradient=0;
72a7e14dcfSSatish Balay   tao->ops->computehessian=0;
734a48860cSAlp Dener   tao->ops->computeresidual=0;
744a48860cSAlp Dener   tao->ops->computeresidualjacobian=0;
75a7e14dcfSSatish Balay   tao->ops->computeconstraints=0;
76a7e14dcfSSatish Balay   tao->ops->computejacobian=0;
77a7e14dcfSSatish Balay   tao->ops->computejacobianequality=0;
78a7e14dcfSSatish Balay   tao->ops->computejacobianinequality=0;
79a7e14dcfSSatish Balay   tao->ops->computeequalityconstraints=0;
80a7e14dcfSSatish Balay   tao->ops->computeinequalityconstraints=0;
81a7e14dcfSSatish Balay   tao->ops->convergencetest=TaoDefaultConvergenceTest;
82a7e14dcfSSatish Balay   tao->ops->convergencedestroy=0;
83a7e14dcfSSatish Balay   tao->ops->computedual=0;
84a7e14dcfSSatish Balay   tao->ops->setup=0;
85a7e14dcfSSatish Balay   tao->ops->solve=0;
86a7e14dcfSSatish Balay   tao->ops->view=0;
87a7e14dcfSSatish Balay   tao->ops->setfromoptions=0;
88a7e14dcfSSatish Balay   tao->ops->destroy=0;
89a7e14dcfSSatish Balay 
9047a47007SBarry Smith   tao->solution=NULL;
9147a47007SBarry Smith   tao->gradient=NULL;
924a48860cSAlp Dener   tao->ls_res = NULL;
934a48860cSAlp Dener   tao->ls_jac = NULL;
9447a47007SBarry Smith   tao->constraints=NULL;
9547a47007SBarry Smith   tao->constraints_equality=NULL;
9647a47007SBarry Smith   tao->constraints_inequality=NULL;
97737f463aSAlp Dener   tao->res_weights_v=NULL;
98737f463aSAlp Dener   tao->res_weights_w=NULL;
9947a47007SBarry Smith   tao->stepdirection=NULL;
1008931d482SJason Sarich   tao->niter=0;
1018931d482SJason Sarich   tao->ntotalits=0;
10247a47007SBarry Smith   tao->XL = NULL;
10347a47007SBarry Smith   tao->XU = NULL;
10447a47007SBarry Smith   tao->IL = NULL;
10547a47007SBarry Smith   tao->IU = NULL;
10647a47007SBarry Smith   tao->DI = NULL;
10747a47007SBarry Smith   tao->DE = NULL;
108a9603a14SPatrick Farrell   tao->gradient_norm = NULL;
109a9603a14SPatrick Farrell   tao->gradient_norm_tmp = NULL;
11047a47007SBarry Smith   tao->hessian = NULL;
11147a47007SBarry Smith   tao->hessian_pre = NULL;
11247a47007SBarry Smith   tao->jacobian = NULL;
11347a47007SBarry Smith   tao->jacobian_pre = NULL;
11447a47007SBarry Smith   tao->jacobian_state = NULL;
11547a47007SBarry Smith   tao->jacobian_state_pre = NULL;
11647a47007SBarry Smith   tao->jacobian_state_inv = NULL;
11747a47007SBarry Smith   tao->jacobian_design = NULL;
11847a47007SBarry Smith   tao->jacobian_design_pre = NULL;
11947a47007SBarry Smith   tao->jacobian_equality = NULL;
12047a47007SBarry Smith   tao->jacobian_equality_pre = NULL;
12147a47007SBarry Smith   tao->jacobian_inequality = NULL;
12247a47007SBarry Smith   tao->jacobian_inequality_pre = NULL;
12347a47007SBarry Smith   tao->state_is = NULL;
12447a47007SBarry Smith   tao->design_is = NULL;
125a7e14dcfSSatish Balay 
126a7e14dcfSSatish Balay   tao->max_it     = 10000;
127a7e14dcfSSatish Balay   tao->max_funcs   = 10000;
1286f4723b1SBarry Smith #if defined(PETSC_USE_REAL_SINGLE)
1296f4723b1SBarry Smith   tao->gatol       = 1e-5;
1306f4723b1SBarry Smith   tao->grtol       = 1e-5;
1316f4723b1SBarry Smith #else
132a7e14dcfSSatish Balay   tao->gatol       = 1e-8;
133a7e14dcfSSatish Balay   tao->grtol       = 1e-8;
1346f4723b1SBarry Smith #endif
135a7e14dcfSSatish Balay   tao->crtol       = 0.0;
1366552cf8aSJason Sarich   tao->catol       = 0.0;
1376552cf8aSJason Sarich   tao->gttol       = 0.0;
1384f1535bcSTodd Munson   tao->steptol     = 0.0;
139e270355aSBarry Smith   tao->trust0      = PETSC_INFINITY;
1406f4723b1SBarry Smith   tao->fmin        = PETSC_NINFINITY;
141ae93cb3cSJason Sarich   tao->hist_malloc = PETSC_FALSE;
142a7e14dcfSSatish Balay   tao->hist_reset = PETSC_TRUE;
143a7e14dcfSSatish Balay   tao->hist_max = 0;
144a7e14dcfSSatish Balay   tao->hist_len = 0;
14547a47007SBarry Smith   tao->hist_obj = NULL;
14647a47007SBarry Smith   tao->hist_resid = NULL;
14747a47007SBarry Smith   tao->hist_cnorm = NULL;
148ae93cb3cSJason Sarich   tao->hist_lits = NULL;
149a7e14dcfSSatish Balay 
150a7e14dcfSSatish Balay   tao->numbermonitors=0;
151a7e14dcfSSatish Balay   tao->viewsolution=PETSC_FALSE;
152a7e14dcfSSatish Balay   tao->viewhessian=PETSC_FALSE;
153a7e14dcfSSatish Balay   tao->viewgradient=PETSC_FALSE;
154a7e14dcfSSatish Balay   tao->viewjacobian=PETSC_FALSE;
155a7e14dcfSSatish Balay   tao->viewconstraints = PETSC_FALSE;
156cd929ea3SAlp Dener 
157cd929ea3SAlp Dener   tao->bounded = PETSC_FALSE;
158a7e14dcfSSatish Balay 
1596552cf8aSJason Sarich   /* These flags prevents algorithms from overriding user options */
1606552cf8aSJason Sarich   tao->max_it_changed   =PETSC_FALSE;
1616552cf8aSJason Sarich   tao->max_funcs_changed=PETSC_FALSE;
1626552cf8aSJason Sarich   tao->gatol_changed    =PETSC_FALSE;
1636552cf8aSJason Sarich   tao->grtol_changed    =PETSC_FALSE;
1646552cf8aSJason Sarich   tao->gttol_changed    =PETSC_FALSE;
1656552cf8aSJason Sarich   tao->steptol_changed  =PETSC_FALSE;
1666552cf8aSJason Sarich   tao->trust0_changed   =PETSC_FALSE;
1676552cf8aSJason Sarich   tao->fmin_changed     =PETSC_FALSE;
1686552cf8aSJason Sarich   tao->catol_changed    =PETSC_FALSE;
1696552cf8aSJason Sarich   tao->crtol_changed    =PETSC_FALSE;
170a7e14dcfSSatish Balay   ierr = TaoResetStatistics(tao);CHKERRQ(ierr);
171a7e14dcfSSatish Balay   *newtao = tao;
172a7e14dcfSSatish Balay   PetscFunctionReturn(0);
173a7e14dcfSSatish Balay }
174a7e14dcfSSatish Balay 
175a7e14dcfSSatish Balay /*@
176a7e14dcfSSatish Balay   TaoSolve - Solves an optimization problem min F(x) s.t. l <= x <= u
177a7e14dcfSSatish Balay 
178441846f8SBarry Smith   Collective on Tao
179a7e14dcfSSatish Balay 
180a7e14dcfSSatish Balay   Input Parameters:
181441846f8SBarry Smith . tao - the Tao context
182a7e14dcfSSatish Balay 
183a7e14dcfSSatish Balay   Notes:
184441846f8SBarry Smith   The user must set up the Tao with calls to TaoSetInitialVector(),
185a7e14dcfSSatish Balay   TaoSetObjectiveRoutine(),
186a7e14dcfSSatish Balay   TaoSetGradientRoutine(), and (if using 2nd order method) TaoSetHessianRoutine().
187a7e14dcfSSatish Balay 
188a35d58b8SBarry Smith   You should call TaoGetConvergedReason() or run with -tao_converged_reason to determine if the optimization algorithm actually succeeded or
189a35d58b8SBarry Smith   why it failed.
190a35d58b8SBarry Smith 
191a7e14dcfSSatish Balay   Level: beginner
192a7e14dcfSSatish Balay 
193a35d58b8SBarry Smith .seealso: TaoCreate(), TaoSetObjectiveRoutine(), TaoSetGradientRoutine(), TaoSetHessianRoutine(), TaoGetConvergedReason()
194a7e14dcfSSatish Balay  @*/
195441846f8SBarry Smith PetscErrorCode TaoSolve(Tao tao)
196a7e14dcfSSatish Balay {
197a7e14dcfSSatish Balay   PetscErrorCode   ierr;
198e2379f4fSBarry Smith   static PetscBool set = PETSC_FALSE;
19947a47007SBarry Smith 
200a7e14dcfSSatish Balay   PetscFunctionBegin;
201441846f8SBarry Smith   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
202e2379f4fSBarry Smith   ierr = PetscCitationsRegister("@TechReport{tao-user-ref,\n"
203e2379f4fSBarry Smith                                 "title   = {Toolkit for Advanced Optimization (TAO) Users Manual},\n"
204e2379f4fSBarry Smith                                 "author  = {Todd Munson and Jason Sarich and Stefan Wild and Steve Benson and Lois Curfman McInnes},\n"
205e2379f4fSBarry Smith                                 "Institution = {Argonne National Laboratory},\n"
206e2379f4fSBarry Smith                                 "Year   = 2014,\n"
207e2379f4fSBarry Smith                                 "Number = {ANL/MCS-TM-322 - Revision 3.5},\n"
208302440fdSBarry Smith                                 "url    = {http://www.mcs.anl.gov/tao}\n}\n",&set);CHKERRQ(ierr);
209e2379f4fSBarry Smith 
210a7e14dcfSSatish Balay   ierr = TaoSetUp(tao);CHKERRQ(ierr);
211a7e14dcfSSatish Balay   ierr = TaoResetStatistics(tao);CHKERRQ(ierr);
212a7e14dcfSSatish Balay   if (tao->linesearch) {
213a7e14dcfSSatish Balay     ierr = TaoLineSearchReset(tao->linesearch);CHKERRQ(ierr);
214a7e14dcfSSatish Balay   }
215a7e14dcfSSatish Balay 
2160ebee16dSLisandro Dalcin   ierr = PetscLogEventBegin(TAO_Solve,tao,0,0,0);CHKERRQ(ierr);
217a7e14dcfSSatish Balay   if (tao->ops->solve){ ierr = (*tao->ops->solve)(tao);CHKERRQ(ierr); }
2180ebee16dSLisandro Dalcin   ierr = PetscLogEventEnd(TAO_Solve,tao,0,0,0);CHKERRQ(ierr);
219a7e14dcfSSatish Balay 
2207cf37e64SBarry Smith   ierr = VecViewFromOptions(tao->solution,(PetscObject)tao,"-tao_view_solution");CHKERRQ(ierr);
2217cf37e64SBarry Smith 
2228931d482SJason Sarich   tao->ntotalits += tao->niter;
223fbe0838dSJason Sarich   ierr = TaoViewFromOptions(tao,NULL,"-tao_view");CHKERRQ(ierr);
224a7e14dcfSSatish Balay 
225a7e14dcfSSatish Balay   if (tao->printreason) {
226a7e14dcfSSatish Balay     if (tao->reason > 0) {
2278931d482SJason Sarich       ierr = PetscPrintf(((PetscObject)tao)->comm,"TAO solve converged due to %s iterations %D\n",TaoConvergedReasons[tao->reason],tao->niter);CHKERRQ(ierr);
228a7e14dcfSSatish Balay     } else {
2298931d482SJason Sarich       ierr = PetscPrintf(((PetscObject)tao)->comm,"TAO solve did not converge due to %s iteration %D\n",TaoConvergedReasons[tao->reason],tao->niter);CHKERRQ(ierr);
230a7e14dcfSSatish Balay     }
231a7e14dcfSSatish Balay   }
232a7e14dcfSSatish Balay   PetscFunctionReturn(0);
233a7e14dcfSSatish Balay }
234a7e14dcfSSatish Balay 
235a7e14dcfSSatish Balay /*@
236a7e14dcfSSatish Balay   TaoSetUp - Sets up the internal data structures for the later use
237a7e14dcfSSatish Balay   of a Tao solver
238a7e14dcfSSatish Balay 
239a7e14dcfSSatish Balay   Collective on tao
240a7e14dcfSSatish Balay 
241a7e14dcfSSatish Balay   Input Parameters:
242a7e14dcfSSatish Balay . tao - the TAO context
243a7e14dcfSSatish Balay 
244a7e14dcfSSatish Balay   Notes:
245a7e14dcfSSatish Balay   The user will not need to explicitly call TaoSetUp(), as it will
246a7e14dcfSSatish Balay   automatically be called in TaoSolve().  However, if the user
247a7e14dcfSSatish Balay   desires to call it explicitly, it should come after TaoCreate()
248a7e14dcfSSatish Balay   and any TaoSetSomething() routines, but before TaoSolve().
249a7e14dcfSSatish Balay 
250a7e14dcfSSatish Balay   Level: advanced
251a7e14dcfSSatish Balay 
252a7e14dcfSSatish Balay .seealso: TaoCreate(), TaoSolve()
253a7e14dcfSSatish Balay @*/
254441846f8SBarry Smith PetscErrorCode TaoSetUp(Tao tao)
255a7e14dcfSSatish Balay {
256a7e14dcfSSatish Balay   PetscErrorCode ierr;
25747a47007SBarry Smith 
258a7e14dcfSSatish Balay   PetscFunctionBegin;
259441846f8SBarry Smith   PetscValidHeaderSpecific(tao, TAO_CLASSID,1);
260a7e14dcfSSatish Balay   if (tao->setupcalled) PetscFunctionReturn(0);
261a7e14dcfSSatish Balay 
26247a47007SBarry Smith   if (!tao->solution) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"Must call TaoSetInitialVector");
263a7e14dcfSSatish Balay   if (tao->ops->setup) {
264a7e14dcfSSatish Balay     ierr = (*tao->ops->setup)(tao);CHKERRQ(ierr);
265a7e14dcfSSatish Balay   }
266a7e14dcfSSatish Balay   tao->setupcalled = PETSC_TRUE;
267a7e14dcfSSatish Balay   PetscFunctionReturn(0);
268a7e14dcfSSatish Balay }
269a7e14dcfSSatish Balay 
270a7e14dcfSSatish Balay /*@
271a7e14dcfSSatish Balay   TaoDestroy - Destroys the TAO context that was created with
272a7e14dcfSSatish Balay   TaoCreate()
273a7e14dcfSSatish Balay 
274441846f8SBarry Smith   Collective on Tao
275a7e14dcfSSatish Balay 
276a7e14dcfSSatish Balay   Input Parameter:
277441846f8SBarry Smith . tao - the Tao context
278a7e14dcfSSatish Balay 
279a7e14dcfSSatish Balay   Level: beginner
280a7e14dcfSSatish Balay 
281a7e14dcfSSatish Balay .seealso: TaoCreate(), TaoSolve()
282a7e14dcfSSatish Balay @*/
283441846f8SBarry Smith PetscErrorCode TaoDestroy(Tao *tao)
284a7e14dcfSSatish Balay {
285a7e14dcfSSatish Balay   PetscErrorCode ierr;
28647a47007SBarry Smith 
287a7e14dcfSSatish Balay   PetscFunctionBegin;
288a7e14dcfSSatish Balay   if (!*tao) PetscFunctionReturn(0);
289441846f8SBarry Smith   PetscValidHeaderSpecific(*tao,TAO_CLASSID,1);
290a7e14dcfSSatish Balay   if (--((PetscObject)*tao)->refct > 0) {*tao=0;PetscFunctionReturn(0);}
291a7e14dcfSSatish Balay 
292a7e14dcfSSatish Balay   if ((*tao)->ops->destroy) {
293a7e14dcfSSatish Balay     ierr = (*((*tao))->ops->destroy)(*tao);CHKERRQ(ierr);
294a7e14dcfSSatish Balay   }
295a7e14dcfSSatish Balay   ierr = KSPDestroy(&(*tao)->ksp);CHKERRQ(ierr);
296a7e14dcfSSatish Balay   ierr = TaoLineSearchDestroy(&(*tao)->linesearch);CHKERRQ(ierr);
297a7e14dcfSSatish Balay 
298a7e14dcfSSatish Balay   if ((*tao)->ops->convergencedestroy) {
299a7e14dcfSSatish Balay     ierr = (*(*tao)->ops->convergencedestroy)((*tao)->cnvP);CHKERRQ(ierr);
300a7e14dcfSSatish Balay     if ((*tao)->jacobian_state_inv) {
301a7e14dcfSSatish Balay       ierr = MatDestroy(&(*tao)->jacobian_state_inv);CHKERRQ(ierr);
302a7e14dcfSSatish Balay     }
303a7e14dcfSSatish Balay   }
304a7e14dcfSSatish Balay   ierr = VecDestroy(&(*tao)->solution);CHKERRQ(ierr);
305a7e14dcfSSatish Balay   ierr = VecDestroy(&(*tao)->gradient);CHKERRQ(ierr);
3064a48860cSAlp Dener   ierr = VecDestroy(&(*tao)->ls_res);CHKERRQ(ierr);
307a7e14dcfSSatish Balay 
308a9603a14SPatrick Farrell   if ((*tao)->gradient_norm) {
309a9603a14SPatrick Farrell     ierr = PetscObjectDereference((PetscObject)(*tao)->gradient_norm);CHKERRQ(ierr);
310a9603a14SPatrick Farrell     ierr = VecDestroy(&(*tao)->gradient_norm_tmp);CHKERRQ(ierr);
311a9603a14SPatrick Farrell   }
312a9603a14SPatrick Farrell 
313a7e14dcfSSatish Balay   ierr = VecDestroy(&(*tao)->XL);CHKERRQ(ierr);
314a7e14dcfSSatish Balay   ierr = VecDestroy(&(*tao)->XU);CHKERRQ(ierr);
315a7e14dcfSSatish Balay   ierr = VecDestroy(&(*tao)->IL);CHKERRQ(ierr);
316a7e14dcfSSatish Balay   ierr = VecDestroy(&(*tao)->IU);CHKERRQ(ierr);
317a7e14dcfSSatish Balay   ierr = VecDestroy(&(*tao)->DE);CHKERRQ(ierr);
318a7e14dcfSSatish Balay   ierr = VecDestroy(&(*tao)->DI);CHKERRQ(ierr);
319a7e14dcfSSatish Balay   ierr = VecDestroy(&(*tao)->constraints_equality);CHKERRQ(ierr);
320a7e14dcfSSatish Balay   ierr = VecDestroy(&(*tao)->constraints_inequality);CHKERRQ(ierr);
321a7e14dcfSSatish Balay   ierr = VecDestroy(&(*tao)->stepdirection);CHKERRQ(ierr);
322a7e14dcfSSatish Balay   ierr = MatDestroy(&(*tao)->hessian_pre);CHKERRQ(ierr);
323a7e14dcfSSatish Balay   ierr = MatDestroy(&(*tao)->hessian);CHKERRQ(ierr);
3244a48860cSAlp Dener   ierr = MatDestroy(&(*tao)->ls_jac);CHKERRQ(ierr);
3254a48860cSAlp Dener   ierr = MatDestroy(&(*tao)->ls_jac_pre);CHKERRQ(ierr);
326a7e14dcfSSatish Balay   ierr = MatDestroy(&(*tao)->jacobian_pre);CHKERRQ(ierr);
327a7e14dcfSSatish Balay   ierr = MatDestroy(&(*tao)->jacobian);CHKERRQ(ierr);
328a7e14dcfSSatish Balay   ierr = MatDestroy(&(*tao)->jacobian_state_pre);CHKERRQ(ierr);
329a7e14dcfSSatish Balay   ierr = MatDestroy(&(*tao)->jacobian_state);CHKERRQ(ierr);
330a7e14dcfSSatish Balay   ierr = MatDestroy(&(*tao)->jacobian_state_inv);CHKERRQ(ierr);
331067f330cSJason Sarich   ierr = MatDestroy(&(*tao)->jacobian_design);CHKERRQ(ierr);
332a7e14dcfSSatish Balay   ierr = MatDestroy(&(*tao)->jacobian_equality);CHKERRQ(ierr);
333a7e14dcfSSatish Balay   ierr = MatDestroy(&(*tao)->jacobian_equality_pre);CHKERRQ(ierr);
334a7e14dcfSSatish Balay   ierr = MatDestroy(&(*tao)->jacobian_inequality);CHKERRQ(ierr);
335a7e14dcfSSatish Balay   ierr = MatDestroy(&(*tao)->jacobian_inequality_pre);CHKERRQ(ierr);
336a7e14dcfSSatish Balay   ierr = ISDestroy(&(*tao)->state_is);CHKERRQ(ierr);
337a7e14dcfSSatish Balay   ierr = ISDestroy(&(*tao)->design_is);CHKERRQ(ierr);
338737f463aSAlp Dener   ierr = VecDestroy(&(*tao)->res_weights_v);CHKERRQ(ierr);
339a7e14dcfSSatish Balay   ierr = TaoCancelMonitors(*tao);CHKERRQ(ierr);
340ae93cb3cSJason Sarich   if ((*tao)->hist_malloc) {
341ae93cb3cSJason Sarich     ierr = PetscFree((*tao)->hist_obj);CHKERRQ(ierr);
342ae93cb3cSJason Sarich     ierr = PetscFree((*tao)->hist_resid);CHKERRQ(ierr);
343ae93cb3cSJason Sarich     ierr = PetscFree((*tao)->hist_cnorm);CHKERRQ(ierr);
344ae93cb3cSJason Sarich     ierr = PetscFree((*tao)->hist_lits);CHKERRQ(ierr);
345ae93cb3cSJason Sarich   }
346737f463aSAlp Dener   if ((*tao)->res_weights_n) {
347737f463aSAlp Dener     ierr = PetscFree((*tao)->res_weights_rows);CHKERRQ(ierr);
348737f463aSAlp Dener     ierr = PetscFree((*tao)->res_weights_cols);CHKERRQ(ierr);
349737f463aSAlp Dener     ierr = PetscFree((*tao)->res_weights_w);CHKERRQ(ierr);
350125ddc8aSJason Sarich   }
351a7e14dcfSSatish Balay   ierr = PetscHeaderDestroy(tao);CHKERRQ(ierr);
352a7e14dcfSSatish Balay   PetscFunctionReturn(0);
353a7e14dcfSSatish Balay }
354a7e14dcfSSatish Balay 
355a7e14dcfSSatish Balay /*@
356441846f8SBarry Smith   TaoSetFromOptions - Sets various Tao parameters from user
357a7e14dcfSSatish Balay   options.
358a7e14dcfSSatish Balay 
359441846f8SBarry Smith   Collective on Tao
360a7e14dcfSSatish Balay 
361a7e14dcfSSatish Balay   Input Paremeter:
362441846f8SBarry Smith . tao - the Tao solver context
363a7e14dcfSSatish Balay 
364a7e14dcfSSatish Balay   options Database Keys:
36558417fe7SBarry Smith + -tao_type <type> - The algorithm that TAO uses (lmvm, nls, etc.)
366a7e14dcfSSatish Balay . -tao_gatol <gatol> - absolute error tolerance for ||gradient||
367a7e14dcfSSatish Balay . -tao_grtol <grtol> - relative error tolerance for ||gradient||
368a7e14dcfSSatish Balay . -tao_gttol <gttol> - reduction of ||gradient|| relative to initial gradient
369a7e14dcfSSatish Balay . -tao_max_it <max> - sets maximum number of iterations
370a7e14dcfSSatish Balay . -tao_max_funcs <max> - sets maximum number of function evaluations
371a7e14dcfSSatish Balay . -tao_fmin <fmin> - stop if function value reaches fmin
372a7e14dcfSSatish Balay . -tao_steptol <tol> - stop if trust region radius less than <tol>
373a7e14dcfSSatish Balay . -tao_trust0 <t> - initial trust region radius
374a7e14dcfSSatish Balay . -tao_monitor - prints function value and residual at each iteration
375a7e14dcfSSatish Balay . -tao_smonitor - same as tao_monitor, but truncates very small values
376a7e14dcfSSatish Balay . -tao_cmonitor - prints function value, residual, and constraint norm at each iteration
377a7e14dcfSSatish Balay . -tao_view_solution - prints solution vector at each iteration
3784a48860cSAlp Dener . -tao_view_ls_residual - prints least-squares residual vector at each iteration
379a7e14dcfSSatish Balay . -tao_view_step - prints step direction vector at each iteration
380a7e14dcfSSatish Balay . -tao_view_gradient - prints gradient vector at each iteration
381a7e14dcfSSatish Balay . -tao_draw_solution - graphically view solution vector at each iteration
382a7e14dcfSSatish Balay . -tao_draw_step - graphically view step vector at each iteration
383a7e14dcfSSatish Balay . -tao_draw_gradient - graphically view gradient at each iteration
384a7e14dcfSSatish Balay . -tao_fd_gradient - use gradient computed with finite differences
385f4c1ad5cSStefano Zampini . -tao_fd_hessian - use hessian computed with finite differences
386f4c1ad5cSStefano Zampini . -tao_mf_hessian - use matrix-free hessian computed with finite differences
387a7e14dcfSSatish Balay . -tao_cancelmonitors - cancels all monitors (except those set with command line)
388441846f8SBarry Smith . -tao_view - prints information about the Tao after solving
389a7e14dcfSSatish Balay - -tao_converged_reason - prints the reason TAO stopped iterating
390a7e14dcfSSatish Balay 
391a7e14dcfSSatish Balay   Notes:
392a7e14dcfSSatish Balay   To see all options, run your program with the -help option or consult the
393a7e14dcfSSatish Balay   user's manual. Should be called after TaoCreate() but before TaoSolve()
394a7e14dcfSSatish Balay 
395a7e14dcfSSatish Balay   Level: beginner
396a7e14dcfSSatish Balay @*/
397441846f8SBarry Smith PetscErrorCode TaoSetFromOptions(Tao tao)
398a7e14dcfSSatish Balay {
399a7e14dcfSSatish Balay   PetscErrorCode ierr;
400b625d6c7SJed Brown   TaoType        default_type = TAOLMVM;
401a7e14dcfSSatish Balay   char           type[256], monfilename[PETSC_MAX_PATH_LEN];
402a7e14dcfSSatish Balay   PetscViewer    monviewer;
403a7e14dcfSSatish Balay   PetscBool      flg;
404a7e14dcfSSatish Balay   MPI_Comm       comm;
405a7e14dcfSSatish Balay 
406a7e14dcfSSatish Balay   PetscFunctionBegin;
407441846f8SBarry Smith   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
408a7e14dcfSSatish Balay   ierr = PetscObjectGetComm((PetscObject)tao,&comm);CHKERRQ(ierr);
409125ddc8aSJason Sarich 
410a7e14dcfSSatish Balay   /* So no warnings are given about unused options */
411c5929fdfSBarry Smith   ierr = PetscOptionsHasName(((PetscObject)tao)->options,((PetscObject)tao)->prefix,"-tao_ls_type",&flg);CHKERRQ(ierr);
412a7e14dcfSSatish Balay 
413a7e14dcfSSatish Balay   ierr = PetscObjectOptionsBegin((PetscObject)tao);CHKERRQ(ierr);
414a7e14dcfSSatish Balay   {
415441846f8SBarry Smith     ierr = TaoRegisterAll();CHKERRQ(ierr);
416a7e14dcfSSatish Balay     if (((PetscObject)tao)->type_name) {
417a7e14dcfSSatish Balay       default_type = ((PetscObject)tao)->type_name;
418a7e14dcfSSatish Balay     }
419a7e14dcfSSatish Balay     /* Check for type from options */
420441846f8SBarry Smith     ierr = PetscOptionsFList("-tao_type","Tao Solver type","TaoSetType",TaoList,default_type,type,256,&flg);CHKERRQ(ierr);
421a7e14dcfSSatish Balay     if (flg) {
422a7e14dcfSSatish Balay       ierr = TaoSetType(tao,type);CHKERRQ(ierr);
423a7e14dcfSSatish Balay     } else if (!((PetscObject)tao)->type_name) {
424302440fdSBarry Smith       ierr = TaoSetType(tao,default_type);CHKERRQ(ierr);
425a7e14dcfSSatish Balay     }
426a7e14dcfSSatish Balay 
4276552cf8aSJason Sarich     ierr = PetscOptionsReal("-tao_catol","Stop if constraints violations within","TaoSetConstraintTolerances",tao->catol,&tao->catol,&flg);CHKERRQ(ierr);
4286552cf8aSJason Sarich     if (flg) tao->catol_changed=PETSC_TRUE;
4296552cf8aSJason Sarich     ierr = PetscOptionsReal("-tao_crtol","Stop if relative contraint violations within","TaoSetConstraintTolerances",tao->crtol,&tao->crtol,&flg);CHKERRQ(ierr);
4306552cf8aSJason Sarich     if (flg) tao->crtol_changed=PETSC_TRUE;
4316552cf8aSJason Sarich     ierr = PetscOptionsReal("-tao_gatol","Stop if norm of gradient less than","TaoSetTolerances",tao->gatol,&tao->gatol,&flg);CHKERRQ(ierr);
4326552cf8aSJason Sarich     if (flg) tao->gatol_changed=PETSC_TRUE;
4336552cf8aSJason Sarich     ierr = PetscOptionsReal("-tao_grtol","Stop if norm of gradient divided by the function value is less than","TaoSetTolerances",tao->grtol,&tao->grtol,&flg);CHKERRQ(ierr);
4346552cf8aSJason Sarich     if (flg) tao->grtol_changed=PETSC_TRUE;
4356552cf8aSJason Sarich     ierr = PetscOptionsReal("-tao_gttol","Stop if the norm of the gradient is less than the norm of the initial gradient times tol","TaoSetTolerances",tao->gttol,&tao->gttol,&flg);CHKERRQ(ierr);
4366552cf8aSJason Sarich     if (flg) tao->gttol_changed=PETSC_TRUE;
4376552cf8aSJason Sarich     ierr = PetscOptionsInt("-tao_max_it","Stop if iteration number exceeds","TaoSetMaximumIterations",tao->max_it,&tao->max_it,&flg);CHKERRQ(ierr);
4386552cf8aSJason Sarich     if (flg) tao->max_it_changed=PETSC_TRUE;
4396552cf8aSJason Sarich     ierr = PetscOptionsInt("-tao_max_funcs","Stop if number of function evaluations exceeds","TaoSetMaximumFunctionEvaluations",tao->max_funcs,&tao->max_funcs,&flg);CHKERRQ(ierr);
4406552cf8aSJason Sarich     if (flg) tao->max_funcs_changed=PETSC_TRUE;
4416552cf8aSJason Sarich     ierr = PetscOptionsReal("-tao_fmin","Stop if function less than","TaoSetFunctionLowerBound",tao->fmin,&tao->fmin,&flg);CHKERRQ(ierr);
4426552cf8aSJason Sarich     if (flg) tao->fmin_changed=PETSC_TRUE;
4436552cf8aSJason Sarich     ierr = PetscOptionsReal("-tao_steptol","Stop if step size or trust region radius less than","",tao->steptol,&tao->steptol,&flg);CHKERRQ(ierr);
4446552cf8aSJason Sarich     if (flg) tao->steptol_changed=PETSC_TRUE;
4456552cf8aSJason Sarich     ierr = PetscOptionsReal("-tao_trust0","Initial trust region radius","TaoSetTrustRegionRadius",tao->trust0,&tao->trust0,&flg);CHKERRQ(ierr);
4466552cf8aSJason Sarich     if (flg) tao->trust0_changed=PETSC_TRUE;
447a7e14dcfSSatish Balay     ierr = PetscOptionsString("-tao_view_solution","view solution vector after each evaluation","TaoSetMonitor","stdout",monfilename,PETSC_MAX_PATH_LEN,&flg);CHKERRQ(ierr);
448a7e14dcfSSatish Balay     if (flg) {
449a7e14dcfSSatish Balay       ierr = PetscViewerASCIIOpen(comm,monfilename,&monviewer);CHKERRQ(ierr);
450a7e14dcfSSatish Balay       ierr = TaoSetMonitor(tao,TaoSolutionMonitor,monviewer,(PetscErrorCode (*)(void**))PetscViewerDestroy);CHKERRQ(ierr);
451a7e14dcfSSatish Balay     }
452a7e14dcfSSatish Balay 
4538afaa268SBarry Smith     ierr = PetscOptionsBool("-tao_converged_reason","Print reason for TAO converged","TaoSolve",tao->printreason,&tao->printreason,NULL);CHKERRQ(ierr);
454a7e14dcfSSatish Balay     ierr = PetscOptionsString("-tao_view_gradient","view gradient vector after each evaluation","TaoSetMonitor","stdout",monfilename,PETSC_MAX_PATH_LEN,&flg);CHKERRQ(ierr);
455a7e14dcfSSatish Balay     if (flg) {
456a7e14dcfSSatish Balay       ierr = PetscViewerASCIIOpen(comm,monfilename,&monviewer);CHKERRQ(ierr);
457a7e14dcfSSatish Balay       ierr = TaoSetMonitor(tao,TaoGradientMonitor,monviewer,(PetscErrorCode (*)(void**))PetscViewerDestroy);CHKERRQ(ierr);
458a7e14dcfSSatish Balay     }
459a7e14dcfSSatish Balay 
460a7e14dcfSSatish Balay     ierr = PetscOptionsString("-tao_view_stepdirection","view step direction vector after each iteration","TaoSetMonitor","stdout",monfilename,PETSC_MAX_PATH_LEN,&flg);CHKERRQ(ierr);
461a7e14dcfSSatish Balay     if (flg) {
462a7e14dcfSSatish Balay       ierr = PetscViewerASCIIOpen(comm,monfilename,&monviewer);CHKERRQ(ierr);
463a7e14dcfSSatish Balay       ierr = TaoSetMonitor(tao,TaoStepDirectionMonitor,monviewer,(PetscErrorCode (*)(void**))PetscViewerDestroy);CHKERRQ(ierr);
464a7e14dcfSSatish Balay     }
465a7e14dcfSSatish Balay 
466737f463aSAlp Dener     ierr = PetscOptionsString("-tao_view_residual","view least-squares residual vector after each evaluation","TaoSetMonitor","stdout",monfilename,PETSC_MAX_PATH_LEN,&flg);CHKERRQ(ierr);
467a7e14dcfSSatish Balay     if (flg) {
468a7e14dcfSSatish Balay       ierr = PetscViewerASCIIOpen(comm,monfilename,&monviewer);CHKERRQ(ierr);
469737f463aSAlp Dener       ierr = TaoSetMonitor(tao,TaoResidualMonitor,monviewer,(PetscErrorCode (*)(void**))PetscViewerDestroy);CHKERRQ(ierr);
470a7e14dcfSSatish Balay     }
471a7e14dcfSSatish Balay 
472a7e14dcfSSatish Balay     ierr = PetscOptionsString("-tao_monitor","Use the default convergence monitor","TaoSetMonitor","stdout",monfilename,PETSC_MAX_PATH_LEN,&flg);CHKERRQ(ierr);
473a7e14dcfSSatish Balay     if (flg) {
474a7e14dcfSSatish Balay       ierr = PetscViewerASCIIOpen(comm,monfilename,&monviewer);CHKERRQ(ierr);
47598ea980cSBarry Smith       ierr = TaoSetMonitor(tao,TaoMonitorDefault,monviewer,(PetscErrorCode (*)(void**))PetscViewerDestroy);CHKERRQ(ierr);
476a7e14dcfSSatish Balay     }
477a7e14dcfSSatish Balay 
4788d5ead36SAlp Dener     ierr = PetscOptionsString("-tao_gmonitor","Use the convergence monitor with extra globalization info","TaoSetMonitor","stdout",monfilename,PETSC_MAX_PATH_LEN,&flg);CHKERRQ(ierr);
4798d5ead36SAlp Dener     if (flg) {
4808d5ead36SAlp Dener       ierr = PetscViewerASCIIOpen(comm,monfilename,&monviewer);CHKERRQ(ierr);
4818d5ead36SAlp Dener       ierr = TaoSetMonitor(tao,TaoDefaultGMonitor,monviewer,(PetscErrorCode (*)(void**))PetscViewerDestroy);CHKERRQ(ierr);
4828d5ead36SAlp Dener     }
4838d5ead36SAlp Dener 
484a7e14dcfSSatish Balay     ierr = PetscOptionsString("-tao_smonitor","Use the short convergence monitor","TaoSetMonitor","stdout",monfilename,PETSC_MAX_PATH_LEN,&flg);CHKERRQ(ierr);
485a7e14dcfSSatish Balay     if (flg) {
486a7e14dcfSSatish Balay       ierr = PetscViewerASCIIOpen(comm,monfilename,&monviewer);CHKERRQ(ierr);
487a7e14dcfSSatish Balay       ierr = TaoSetMonitor(tao,TaoDefaultSMonitor,monviewer,(PetscErrorCode (*)(void**))PetscViewerDestroy);CHKERRQ(ierr);
488a7e14dcfSSatish Balay     }
489a7e14dcfSSatish Balay 
490a7e14dcfSSatish Balay     ierr = PetscOptionsString("-tao_cmonitor","Use the default convergence monitor with constraint norm","TaoSetMonitor","stdout",monfilename,PETSC_MAX_PATH_LEN,&flg);CHKERRQ(ierr);
491a7e14dcfSSatish Balay     if (flg) {
492a7e14dcfSSatish Balay       ierr = PetscViewerASCIIOpen(comm,monfilename,&monviewer);CHKERRQ(ierr);
493a7e14dcfSSatish Balay       ierr = TaoSetMonitor(tao,TaoDefaultCMonitor,monviewer,(PetscErrorCode (*)(void**))PetscViewerDestroy);CHKERRQ(ierr);
494a7e14dcfSSatish Balay     }
495a7e14dcfSSatish Balay 
496a7e14dcfSSatish Balay 
4978afaa268SBarry Smith     flg = PETSC_FALSE;
4988afaa268SBarry Smith     ierr = PetscOptionsBool("-tao_cancelmonitors","cancel all monitors and call any registered destroy routines","TaoCancelMonitors",flg,&flg,NULL);CHKERRQ(ierr);
499a7e14dcfSSatish Balay     if (flg) {ierr = TaoCancelMonitors(tao);CHKERRQ(ierr);}
500a7e14dcfSSatish Balay 
5018afaa268SBarry Smith     flg = PETSC_FALSE;
5028afaa268SBarry Smith     ierr = PetscOptionsBool("-tao_draw_solution","Plot solution vector at each iteration","TaoSetMonitor",flg,&flg,NULL);CHKERRQ(ierr);
503a7e14dcfSSatish Balay     if (flg) {
504e882e171SHong Zhang       TaoMonitorDrawCtx drawctx;
505e882e171SHong Zhang       PetscInt          howoften = 1;
506e882e171SHong Zhang       ierr = TaoMonitorDrawCtxCreate(PetscObjectComm((PetscObject)tao),0,0,PETSC_DECIDE,PETSC_DECIDE,300,300,howoften,&drawctx);CHKERRQ(ierr);
507e882e171SHong Zhang       ierr = TaoSetMonitor(tao,TaoDrawSolutionMonitor,drawctx,(PetscErrorCode (*)(void**))TaoMonitorDrawCtxDestroy);CHKERRQ(ierr);
508a7e14dcfSSatish Balay     }
509a7e14dcfSSatish Balay 
5108afaa268SBarry Smith     flg = PETSC_FALSE;
5118afaa268SBarry Smith     ierr = PetscOptionsBool("-tao_draw_step","plots step direction at each iteration","TaoSetMonitor",flg,&flg,NULL);CHKERRQ(ierr);
512a7e14dcfSSatish Balay     if (flg) {
51347a47007SBarry Smith       ierr = TaoSetMonitor(tao,TaoDrawStepMonitor,NULL,NULL);CHKERRQ(ierr);
514a7e14dcfSSatish Balay     }
515a7e14dcfSSatish Balay 
5168afaa268SBarry Smith     flg = PETSC_FALSE;
5178afaa268SBarry Smith     ierr = PetscOptionsBool("-tao_draw_gradient","plots gradient at each iteration","TaoSetMonitor",flg,&flg,NULL);CHKERRQ(ierr);
518a7e14dcfSSatish Balay     if (flg) {
519e882e171SHong Zhang       TaoMonitorDrawCtx drawctx;
520e882e171SHong Zhang       PetscInt          howoften = 1;
521e882e171SHong Zhang       ierr = TaoMonitorDrawCtxCreate(PetscObjectComm((PetscObject)tao),0,0,PETSC_DECIDE,PETSC_DECIDE,300,300,howoften,&drawctx);CHKERRQ(ierr);
522e882e171SHong Zhang       ierr = TaoSetMonitor(tao,TaoDrawGradientMonitor,drawctx,(PetscErrorCode (*)(void**))TaoMonitorDrawCtxDestroy);CHKERRQ(ierr);
523a7e14dcfSSatish Balay     }
5248afaa268SBarry Smith     flg = PETSC_FALSE;
5258afaa268SBarry Smith     ierr = PetscOptionsBool("-tao_fd_gradient","compute gradient using finite differences","TaoDefaultComputeGradient",flg,&flg,NULL);CHKERRQ(ierr);
526a7e14dcfSSatish Balay     if (flg) {
52747a47007SBarry Smith       ierr = TaoSetGradientRoutine(tao,TaoDefaultComputeGradient,NULL);CHKERRQ(ierr);
528a7e14dcfSSatish Balay     }
529f4c1ad5cSStefano Zampini     flg = PETSC_FALSE;
530f4c1ad5cSStefano Zampini     ierr = PetscOptionsBool("-tao_fd_hessian","compute hessian using finite differences","TaoDefaultComputeHessian",flg,&flg,NULL);CHKERRQ(ierr);
531f4c1ad5cSStefano Zampini     if (flg) {
532f4c1ad5cSStefano Zampini       Mat H;
533f4c1ad5cSStefano Zampini 
534f4c1ad5cSStefano Zampini       ierr = MatCreate(PetscObjectComm((PetscObject)tao),&H);CHKERRQ(ierr);
535f4c1ad5cSStefano Zampini       ierr = MatSetType(H,MATAIJ);CHKERRQ(ierr);
536f4c1ad5cSStefano Zampini       ierr = TaoSetHessianRoutine(tao,H,H,TaoDefaultComputeHessian,NULL);CHKERRQ(ierr);
537f4c1ad5cSStefano Zampini       ierr = MatDestroy(&H);CHKERRQ(ierr);
538f4c1ad5cSStefano Zampini     }
539f4c1ad5cSStefano Zampini     flg = PETSC_FALSE;
540f4c1ad5cSStefano Zampini     ierr = PetscOptionsBool("-tao_mf_hessian","compute matrix-free hessian using finite differences","TaoDefaultComputeHessianMFFD",flg,&flg,NULL);CHKERRQ(ierr);
541f4c1ad5cSStefano Zampini     if (flg) {
542f4c1ad5cSStefano Zampini       Mat H;
543f4c1ad5cSStefano Zampini 
544f4c1ad5cSStefano Zampini       ierr = MatCreate(PetscObjectComm((PetscObject)tao),&H);CHKERRQ(ierr);
545f4c1ad5cSStefano Zampini       ierr = TaoSetHessianRoutine(tao,H,H,TaoDefaultComputeHessianMFFD,NULL);CHKERRQ(ierr);
546f4c1ad5cSStefano Zampini       ierr = MatDestroy(&H);CHKERRQ(ierr);
547f4c1ad5cSStefano Zampini     }
548f4c1ad5cSStefano Zampini     ierr = PetscOptionsEnum("-tao_subset_type","subset type","",TaoSubSetTypes,(PetscEnum)tao->subset_type,(PetscEnum*)&tao->subset_type,NULL);CHKERRQ(ierr);
549a7e14dcfSSatish Balay 
550a7e14dcfSSatish Balay     if (tao->ops->setfromoptions) {
5511a1499c8SBarry Smith       ierr = (*tao->ops->setfromoptions)(PetscOptionsObject,tao);CHKERRQ(ierr);
552a7e14dcfSSatish Balay     }
553a7e14dcfSSatish Balay   }
554a7e14dcfSSatish Balay   ierr = PetscOptionsEnd();CHKERRQ(ierr);
555a7e14dcfSSatish Balay   PetscFunctionReturn(0);
556a7e14dcfSSatish Balay }
557a7e14dcfSSatish Balay 
558a7e14dcfSSatish Balay /*@C
559441846f8SBarry Smith   TaoView - Prints information about the Tao
560a7e14dcfSSatish Balay 
561441846f8SBarry Smith   Collective on Tao
562a7e14dcfSSatish Balay 
563a7e14dcfSSatish Balay   InputParameters:
564441846f8SBarry Smith + tao - the Tao context
565a7e14dcfSSatish Balay - viewer - visualization context
566a7e14dcfSSatish Balay 
567a7e14dcfSSatish Balay   Options Database Key:
568a7e14dcfSSatish Balay . -tao_view - Calls TaoView() at the end of TaoSolve()
569a7e14dcfSSatish Balay 
570a7e14dcfSSatish Balay   Notes:
571a7e14dcfSSatish Balay   The available visualization contexts include
572a7e14dcfSSatish Balay +     PETSC_VIEWER_STDOUT_SELF - standard output (default)
573a7e14dcfSSatish Balay -     PETSC_VIEWER_STDOUT_WORLD - synchronized standard
574a7e14dcfSSatish Balay          output where only the first processor opens
575a7e14dcfSSatish Balay          the file.  All other processors send their
576a7e14dcfSSatish Balay          data to the first processor to print.
577a7e14dcfSSatish Balay 
578a7e14dcfSSatish Balay   Level: beginner
579a7e14dcfSSatish Balay 
580a7e14dcfSSatish Balay .seealso: PetscViewerASCIIOpen()
581a7e14dcfSSatish Balay @*/
582441846f8SBarry Smith PetscErrorCode TaoView(Tao tao, PetscViewer viewer)
583a7e14dcfSSatish Balay {
584a7e14dcfSSatish Balay   PetscErrorCode      ierr;
585a7e14dcfSSatish Balay   PetscBool           isascii,isstring;
586b625d6c7SJed Brown   TaoType             type;
58747a47007SBarry Smith 
588a7e14dcfSSatish Balay   PetscFunctionBegin;
589441846f8SBarry Smith   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
590a7e14dcfSSatish Balay   if (!viewer) {
591a7e14dcfSSatish Balay     ierr = PetscViewerASCIIGetStdout(((PetscObject)tao)->comm,&viewer);CHKERRQ(ierr);
592a7e14dcfSSatish Balay   }
593a7e14dcfSSatish Balay   PetscValidHeaderSpecific(viewer,PETSC_VIEWER_CLASSID,2);
594a7e14dcfSSatish Balay   PetscCheckSameComm(tao,1,viewer,2);
595a7e14dcfSSatish Balay 
596a7e14dcfSSatish Balay   ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERASCII,&isascii);CHKERRQ(ierr);
597a7e14dcfSSatish Balay   ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERSTRING,&isstring);CHKERRQ(ierr);
598a7e14dcfSSatish Balay   if (isascii) {
599a7e14dcfSSatish Balay     ierr = PetscObjectPrintClassNamePrefixType((PetscObject)tao,viewer);CHKERRQ(ierr);
600a7e14dcfSSatish Balay 
601a7e14dcfSSatish Balay     if (tao->ops->view) {
602a7e14dcfSSatish Balay       ierr = PetscViewerASCIIPushTab(viewer);CHKERRQ(ierr);
603a7e14dcfSSatish Balay       ierr = (*tao->ops->view)(tao,viewer);CHKERRQ(ierr);
604a7e14dcfSSatish Balay       ierr = PetscViewerASCIIPopTab(viewer);CHKERRQ(ierr);
605a7e14dcfSSatish Balay     }
6068d3f3ddaSAlp Dener     if (tao->linesearch) {
60785928a98SStefano Zampini       ierr = PetscViewerASCIIPushTab(viewer);CHKERRQ(ierr);
6088d3f3ddaSAlp Dener       ierr = TaoLineSearchView(tao->linesearch,viewer);CHKERRQ(ierr);
60985928a98SStefano Zampini       ierr = PetscViewerASCIIPopTab(viewer);CHKERRQ(ierr);
6108d3f3ddaSAlp Dener     }
611a7e14dcfSSatish Balay     if (tao->ksp) {
61219b5c101SAlp Dener       ierr = PetscViewerASCIIPushTab(viewer);CHKERRQ(ierr);
61385928a98SStefano Zampini       ierr = KSPView(tao->ksp,viewer);CHKERRQ(ierr);
614ae93cb3cSJason Sarich       ierr = PetscViewerASCIIPrintf(viewer,"total KSP iterations: %D\n",tao->ksp_tot_its);CHKERRQ(ierr);
61519b5c101SAlp Dener       ierr = PetscViewerASCIIPopTab(viewer);CHKERRQ(ierr);
616a7e14dcfSSatish Balay     }
61719b5c101SAlp Dener 
61819b5c101SAlp Dener     ierr = PetscViewerASCIIPushTab(viewer);CHKERRQ(ierr);
61919b5c101SAlp Dener 
620a7e14dcfSSatish Balay     if (tao->XL || tao->XU) {
621302440fdSBarry Smith       ierr = PetscViewerASCIIPrintf(viewer,"Active Set subset type: %s\n",TaoSubSetTypes[tao->subset_type]);CHKERRQ(ierr);
622a7e14dcfSSatish Balay     }
623a7e14dcfSSatish Balay 
62447a47007SBarry Smith     ierr = PetscViewerASCIIPrintf(viewer,"convergence tolerances: gatol=%g,",(double)tao->gatol);CHKERRQ(ierr);
62547a47007SBarry Smith     ierr = PetscViewerASCIIPrintf(viewer," steptol=%g,",(double)tao->steptol);CHKERRQ(ierr);
62647a47007SBarry Smith     ierr = PetscViewerASCIIPrintf(viewer," gttol=%g\n",(double)tao->gttol);CHKERRQ(ierr);
62747a47007SBarry Smith     ierr = PetscViewerASCIIPrintf(viewer,"Residual in Function/Gradient:=%g\n",(double)tao->residual);CHKERRQ(ierr);
628a7e14dcfSSatish Balay 
629a7e14dcfSSatish Balay     if (tao->cnorm>0 || tao->catol>0 || tao->crtol>0){
630a7e14dcfSSatish Balay       ierr=PetscViewerASCIIPrintf(viewer,"convergence tolerances:");CHKERRQ(ierr);
63147a47007SBarry Smith       ierr=PetscViewerASCIIPrintf(viewer," catol=%g,",(double)tao->catol);CHKERRQ(ierr);
63247a47007SBarry Smith       ierr=PetscViewerASCIIPrintf(viewer," crtol=%g\n",(double)tao->crtol);CHKERRQ(ierr);
63347a47007SBarry Smith       ierr = PetscViewerASCIIPrintf(viewer,"Residual in Constraints:=%g\n",(double)tao->cnorm);CHKERRQ(ierr);
634a7e14dcfSSatish Balay     }
635a7e14dcfSSatish Balay 
636a7e14dcfSSatish Balay     if (tao->trust < tao->steptol){
63747a47007SBarry Smith       ierr=PetscViewerASCIIPrintf(viewer,"convergence tolerances: steptol=%g\n",(double)tao->steptol);CHKERRQ(ierr);
63847a47007SBarry Smith       ierr=PetscViewerASCIIPrintf(viewer,"Final trust region radius:=%g\n",(double)tao->trust);CHKERRQ(ierr);
639a7e14dcfSSatish Balay     }
640a7e14dcfSSatish Balay 
641a7e14dcfSSatish Balay     if (tao->fmin>-1.e25){
64247a47007SBarry Smith       ierr=PetscViewerASCIIPrintf(viewer,"convergence tolerances: function minimum=%g\n",(double)tao->fmin);CHKERRQ(ierr);
643a7e14dcfSSatish Balay     }
64447a47007SBarry Smith     ierr = PetscViewerASCIIPrintf(viewer,"Objective value=%g\n",(double)tao->fc);CHKERRQ(ierr);
645a7e14dcfSSatish Balay 
64647a47007SBarry Smith     ierr = PetscViewerASCIIPrintf(viewer,"total number of iterations=%D,          ",tao->niter);CHKERRQ(ierr);
647a7e14dcfSSatish Balay     ierr = PetscViewerASCIIPrintf(viewer,"              (max: %D)\n",tao->max_it);CHKERRQ(ierr);
648a7e14dcfSSatish Balay 
649a7e14dcfSSatish Balay     if (tao->nfuncs>0){
65047a47007SBarry Smith       ierr = PetscViewerASCIIPrintf(viewer,"total number of function evaluations=%D,",tao->nfuncs);CHKERRQ(ierr);
65147a47007SBarry Smith       ierr = PetscViewerASCIIPrintf(viewer,"                max: %D\n",tao->max_funcs);CHKERRQ(ierr);
652a7e14dcfSSatish Balay     }
653a7e14dcfSSatish Balay     if (tao->ngrads>0){
65447a47007SBarry Smith       ierr = PetscViewerASCIIPrintf(viewer,"total number of gradient evaluations=%D,",tao->ngrads);CHKERRQ(ierr);
65547a47007SBarry Smith       ierr = PetscViewerASCIIPrintf(viewer,"                max: %D\n",tao->max_funcs);CHKERRQ(ierr);
656a7e14dcfSSatish Balay     }
657a7e14dcfSSatish Balay     if (tao->nfuncgrads>0){
65847a47007SBarry Smith       ierr = PetscViewerASCIIPrintf(viewer,"total number of function/gradient evaluations=%D,",tao->nfuncgrads);CHKERRQ(ierr);
65947a47007SBarry Smith       ierr = PetscViewerASCIIPrintf(viewer,"    (max: %D)\n",tao->max_funcs);CHKERRQ(ierr);
660a7e14dcfSSatish Balay     }
661a7e14dcfSSatish Balay     if (tao->nhess>0){
66247a47007SBarry Smith       ierr = PetscViewerASCIIPrintf(viewer,"total number of Hessian evaluations=%D\n",tao->nhess);CHKERRQ(ierr);
663a7e14dcfSSatish Balay     }
664a7e14dcfSSatish Balay     /*  if (tao->linear_its>0){
66547a47007SBarry Smith      ierr = PetscViewerASCIIPrintf(viewer,"  total Krylov method iterations=%D\n",tao->linear_its);CHKERRQ(ierr);
666a7e14dcfSSatish Balay      }*/
667a7e14dcfSSatish Balay     if (tao->nconstraints>0){
66847a47007SBarry Smith       ierr = PetscViewerASCIIPrintf(viewer,"total number of constraint function evaluations=%D\n",tao->nconstraints);CHKERRQ(ierr);
669a7e14dcfSSatish Balay     }
670a7e14dcfSSatish Balay     if (tao->njac>0){
67147a47007SBarry Smith       ierr = PetscViewerASCIIPrintf(viewer,"total number of Jacobian evaluations=%D\n",tao->njac);CHKERRQ(ierr);
672a7e14dcfSSatish Balay     }
673a7e14dcfSSatish Balay 
674a7e14dcfSSatish Balay     if (tao->reason>0){
675a7e14dcfSSatish Balay       ierr = PetscViewerASCIIPrintf(viewer,    "Solution converged: ");CHKERRQ(ierr);
676a7e14dcfSSatish Balay       switch (tao->reason) {
677a7e14dcfSSatish Balay       case TAO_CONVERGED_GATOL:
678a7e14dcfSSatish Balay         ierr = PetscViewerASCIIPrintf(viewer," ||g(X)|| <= gatol\n");CHKERRQ(ierr);
679a7e14dcfSSatish Balay         break;
680a7e14dcfSSatish Balay       case TAO_CONVERGED_GRTOL:
681a7e14dcfSSatish Balay         ierr = PetscViewerASCIIPrintf(viewer," ||g(X)||/|f(X)| <= grtol\n");CHKERRQ(ierr);
682a7e14dcfSSatish Balay         break;
683a7e14dcfSSatish Balay       case TAO_CONVERGED_GTTOL:
684a7e14dcfSSatish Balay         ierr = PetscViewerASCIIPrintf(viewer," ||g(X)||/||g(X0)|| <= gttol\n");CHKERRQ(ierr);
685a7e14dcfSSatish Balay         break;
686a7e14dcfSSatish Balay       case TAO_CONVERGED_STEPTOL:
687a7e14dcfSSatish Balay         ierr = PetscViewerASCIIPrintf(viewer," Steptol -- step size small\n");CHKERRQ(ierr);
688a7e14dcfSSatish Balay         break;
689a7e14dcfSSatish Balay       case TAO_CONVERGED_MINF:
690a7e14dcfSSatish Balay         ierr = PetscViewerASCIIPrintf(viewer," Minf --  f < fmin\n");CHKERRQ(ierr);
691a7e14dcfSSatish Balay         break;
692a7e14dcfSSatish Balay       case TAO_CONVERGED_USER:
693a7e14dcfSSatish Balay         ierr = PetscViewerASCIIPrintf(viewer," User Terminated\n");CHKERRQ(ierr);
694a7e14dcfSSatish Balay         break;
695a7e14dcfSSatish Balay       default:
696a7e14dcfSSatish Balay         ierr = PetscViewerASCIIPrintf(viewer,"\n");CHKERRQ(ierr);
697a7e14dcfSSatish Balay         break;
698a7e14dcfSSatish Balay       }
699a7e14dcfSSatish Balay 
700a7e14dcfSSatish Balay     } else {
70147cfc4faSBarry Smith       ierr = PetscViewerASCIIPrintf(viewer,"Solver terminated: %d",tao->reason);CHKERRQ(ierr);
702a7e14dcfSSatish Balay       switch (tao->reason) {
703a7e14dcfSSatish Balay       case TAO_DIVERGED_MAXITS:
70447a47007SBarry Smith         ierr = PetscViewerASCIIPrintf(viewer," Maximum Iterations\n");CHKERRQ(ierr);
705a7e14dcfSSatish Balay         break;
706a7e14dcfSSatish Balay       case TAO_DIVERGED_NAN:
70747a47007SBarry Smith         ierr = PetscViewerASCIIPrintf(viewer," NAN or Inf encountered\n");CHKERRQ(ierr);
708a7e14dcfSSatish Balay         break;
709a7e14dcfSSatish Balay       case TAO_DIVERGED_MAXFCN:
71047a47007SBarry Smith         ierr = PetscViewerASCIIPrintf(viewer," Maximum Function Evaluations\n");CHKERRQ(ierr);
711a7e14dcfSSatish Balay         break;
712a7e14dcfSSatish Balay       case TAO_DIVERGED_LS_FAILURE:
71347a47007SBarry Smith         ierr = PetscViewerASCIIPrintf(viewer," Line Search Failure\n");CHKERRQ(ierr);
714a7e14dcfSSatish Balay         break;
715a7e14dcfSSatish Balay       case TAO_DIVERGED_TR_REDUCTION:
71647a47007SBarry Smith         ierr = PetscViewerASCIIPrintf(viewer," Trust Region too small\n");CHKERRQ(ierr);
717a7e14dcfSSatish Balay         break;
718a7e14dcfSSatish Balay       case TAO_DIVERGED_USER:
71947a47007SBarry Smith         ierr = PetscViewerASCIIPrintf(viewer," User Terminated\n");CHKERRQ(ierr);
720a7e14dcfSSatish Balay         break;
721a7e14dcfSSatish Balay       default:
722a7e14dcfSSatish Balay         ierr = PetscViewerASCIIPrintf(viewer,"\n");CHKERRQ(ierr);
723a7e14dcfSSatish Balay         break;
724a7e14dcfSSatish Balay       }
725a7e14dcfSSatish Balay     }
726a7e14dcfSSatish Balay     ierr = PetscViewerASCIIPopTab(viewer);CHKERRQ(ierr);
727a7e14dcfSSatish Balay   } else if (isstring) {
728a7e14dcfSSatish Balay     ierr = TaoGetType(tao,&type);CHKERRQ(ierr);
729a7e14dcfSSatish Balay     ierr = PetscViewerStringSPrintf(viewer," %-3.3s",type);CHKERRQ(ierr);
730a7e14dcfSSatish Balay   }
731a7e14dcfSSatish Balay   PetscFunctionReturn(0);
732a7e14dcfSSatish Balay }
733a7e14dcfSSatish Balay 
734a7e14dcfSSatish Balay /*@
735a7e14dcfSSatish Balay   TaoSetTolerances - Sets parameters used in TAO convergence tests
736a7e14dcfSSatish Balay 
737441846f8SBarry Smith   Logically collective on Tao
738a7e14dcfSSatish Balay 
739a7e14dcfSSatish Balay   Input Parameters:
740441846f8SBarry Smith + tao - the Tao context
741a7e14dcfSSatish Balay . gatol - stop if norm of gradient is less than this
742a7e14dcfSSatish Balay . grtol - stop if relative norm of gradient is less than this
743a7e14dcfSSatish Balay - gttol - stop if norm of gradient is reduced by this factor
744a7e14dcfSSatish Balay 
745a7e14dcfSSatish Balay   Options Database Keys:
746e52336cbSBarry Smith + -tao_gatol <gatol> - Sets gatol
747a7e14dcfSSatish Balay . -tao_grtol <grtol> - Sets grtol
748a7e14dcfSSatish Balay - -tao_gttol <gttol> - Sets gttol
749a7e14dcfSSatish Balay 
750a7e14dcfSSatish Balay   Stopping Criteria:
751a7e14dcfSSatish Balay $ ||g(X)||                            <= gatol
752a7e14dcfSSatish Balay $ ||g(X)|| / |f(X)|                   <= grtol
753a7e14dcfSSatish Balay $ ||g(X)|| / ||g(X0)||                <= gttol
754a7e14dcfSSatish Balay 
755a7e14dcfSSatish Balay   Notes:
756a7e14dcfSSatish Balay   Use PETSC_DEFAULT to leave one or more tolerances unchanged.
757a7e14dcfSSatish Balay 
758a7e14dcfSSatish Balay   Level: beginner
759a7e14dcfSSatish Balay 
760a7e14dcfSSatish Balay .seealso: TaoGetTolerances()
761a7e14dcfSSatish Balay 
762a7e14dcfSSatish Balay @*/
763e52336cbSBarry Smith PetscErrorCode TaoSetTolerances(Tao tao, PetscReal gatol, PetscReal grtol, PetscReal gttol)
764a7e14dcfSSatish Balay {
765a7e14dcfSSatish Balay   PetscErrorCode ierr;
76647a47007SBarry Smith 
767a7e14dcfSSatish Balay   PetscFunctionBegin;
768441846f8SBarry Smith   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
769a7e14dcfSSatish Balay 
770a7e14dcfSSatish Balay   if (gatol != PETSC_DEFAULT) {
771a7e14dcfSSatish Balay     if (gatol<0) {
772955c1f14SBarry Smith       ierr = PetscInfo(tao,"Tried to set negative gatol -- ignored.\n");CHKERRQ(ierr);
773a7e14dcfSSatish Balay     } else {
774a7e14dcfSSatish Balay       tao->gatol = PetscMax(0,gatol);
7756552cf8aSJason Sarich       tao->gatol_changed=PETSC_TRUE;
776a7e14dcfSSatish Balay     }
777a7e14dcfSSatish Balay   }
778a7e14dcfSSatish Balay 
779a7e14dcfSSatish Balay   if (grtol != PETSC_DEFAULT) {
780a7e14dcfSSatish Balay     if (grtol<0) {
781955c1f14SBarry Smith       ierr = PetscInfo(tao,"Tried to set negative grtol -- ignored.\n");CHKERRQ(ierr);
782a7e14dcfSSatish Balay     } else {
783a7e14dcfSSatish Balay       tao->grtol = PetscMax(0,grtol);
7846552cf8aSJason Sarich       tao->grtol_changed=PETSC_TRUE;
785a7e14dcfSSatish Balay     }
786a7e14dcfSSatish Balay   }
787a7e14dcfSSatish Balay 
788a7e14dcfSSatish Balay   if (gttol != PETSC_DEFAULT) {
789a7e14dcfSSatish Balay     if (gttol<0) {
790955c1f14SBarry Smith       ierr = PetscInfo(tao,"Tried to set negative gttol -- ignored.\n");CHKERRQ(ierr);
791a7e14dcfSSatish Balay     } else {
792a7e14dcfSSatish Balay       tao->gttol = PetscMax(0,gttol);
7936552cf8aSJason Sarich       tao->gttol_changed=PETSC_TRUE;
794a7e14dcfSSatish Balay     }
795a7e14dcfSSatish Balay   }
796a7e14dcfSSatish Balay   PetscFunctionReturn(0);
797a7e14dcfSSatish Balay }
798a7e14dcfSSatish Balay 
799a7e14dcfSSatish Balay /*@
80058417fe7SBarry Smith   TaoSetConstraintTolerances - Sets constraint tolerance parameters used in TAO  convergence tests
801a7e14dcfSSatish Balay 
802441846f8SBarry Smith   Logically collective on Tao
803a7e14dcfSSatish Balay 
804a7e14dcfSSatish Balay   Input Parameters:
805441846f8SBarry Smith + tao - the Tao context
806e52336cbSBarry Smith . catol - absolute constraint tolerance, constraint norm must be less than catol for used for gatol convergence criteria
807e52336cbSBarry Smith - crtol - relative contraint tolerance, constraint norm must be less than crtol for used for gatol, gttol convergence criteria
808a7e14dcfSSatish Balay 
809a7e14dcfSSatish Balay   Options Database Keys:
810a7e14dcfSSatish Balay + -tao_catol <catol> - Sets catol
811a7e14dcfSSatish Balay - -tao_crtol <crtol> - Sets crtol
812a7e14dcfSSatish Balay 
8136552cf8aSJason Sarich   Notes:
8146552cf8aSJason Sarich   Use PETSC_DEFAULT to leave any tolerance unchanged.
8156552cf8aSJason Sarich 
816a7e14dcfSSatish Balay   Level: intermediate
817a7e14dcfSSatish Balay 
81858417fe7SBarry Smith .seealso: TaoGetTolerances(), TaoGetConstraintTolerances(), TaoSetTolerances()
819a7e14dcfSSatish Balay 
820a7e14dcfSSatish Balay @*/
821441846f8SBarry Smith PetscErrorCode TaoSetConstraintTolerances(Tao tao, PetscReal catol, PetscReal crtol)
822a7e14dcfSSatish Balay {
823a7e14dcfSSatish Balay   PetscErrorCode ierr;
82447a47007SBarry Smith 
825a7e14dcfSSatish Balay   PetscFunctionBegin;
826441846f8SBarry Smith   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
827a7e14dcfSSatish Balay 
828a7e14dcfSSatish Balay   if (catol != PETSC_DEFAULT) {
829a7e14dcfSSatish Balay     if (catol<0) {
830955c1f14SBarry Smith       ierr = PetscInfo(tao,"Tried to set negative catol -- ignored.\n");CHKERRQ(ierr);
831a7e14dcfSSatish Balay     } else {
832a7e14dcfSSatish Balay       tao->catol = PetscMax(0,catol);
8336552cf8aSJason Sarich       tao->catol_changed=PETSC_TRUE;
834a7e14dcfSSatish Balay     }
835a7e14dcfSSatish Balay   }
836a7e14dcfSSatish Balay 
837a7e14dcfSSatish Balay   if (crtol != PETSC_DEFAULT) {
838a7e14dcfSSatish Balay     if (crtol<0) {
839955c1f14SBarry Smith       ierr = PetscInfo(tao,"Tried to set negative crtol -- ignored.\n");CHKERRQ(ierr);
840a7e14dcfSSatish Balay     } else {
841a7e14dcfSSatish Balay       tao->crtol = PetscMax(0,crtol);
8426552cf8aSJason Sarich       tao->crtol_changed=PETSC_TRUE;
843a7e14dcfSSatish Balay     }
844a7e14dcfSSatish Balay   }
845a7e14dcfSSatish Balay   PetscFunctionReturn(0);
846a7e14dcfSSatish Balay }
847a7e14dcfSSatish Balay 
84858417fe7SBarry Smith /*@
84958417fe7SBarry Smith   TaoGetConstraintTolerances - Gets constraint tolerance parameters used in TAO  convergence tests
85058417fe7SBarry Smith 
85158417fe7SBarry Smith   Not ollective
85258417fe7SBarry Smith 
85358417fe7SBarry Smith   Input Parameter:
85458417fe7SBarry Smith . tao - the Tao context
85558417fe7SBarry Smith 
85658417fe7SBarry Smith   Output Parameter:
857e52336cbSBarry Smith + catol - absolute constraint tolerance, constraint norm must be less than catol for used for gatol convergence criteria
858e52336cbSBarry Smith - crtol - relative contraint tolerance, constraint norm must be less than crtol for used for gatol, gttol convergence criteria
85958417fe7SBarry Smith 
86058417fe7SBarry Smith   Level: intermediate
86158417fe7SBarry Smith 
86258417fe7SBarry Smith .seealso: TaoGetTolerances(), TaoSetTolerances(), TaoSetConstraintTolerances()
86358417fe7SBarry Smith 
86458417fe7SBarry Smith @*/
86558417fe7SBarry Smith PetscErrorCode TaoGetConstraintTolerances(Tao tao, PetscReal *catol, PetscReal *crtol)
86658417fe7SBarry Smith {
86758417fe7SBarry Smith   PetscFunctionBegin;
86858417fe7SBarry Smith   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
86958417fe7SBarry Smith   if (catol) *catol = tao->catol;
87058417fe7SBarry Smith   if (crtol) *crtol = tao->crtol;
87158417fe7SBarry Smith   PetscFunctionReturn(0);
87258417fe7SBarry Smith }
87358417fe7SBarry Smith 
874a7e14dcfSSatish Balay /*@
875a7e14dcfSSatish Balay    TaoSetFunctionLowerBound - Sets a bound on the solution objective value.
876a7e14dcfSSatish Balay    When an approximate solution with an objective value below this number
877a7e14dcfSSatish Balay    has been found, the solver will terminate.
878a7e14dcfSSatish Balay 
879441846f8SBarry Smith    Logically Collective on Tao
880a7e14dcfSSatish Balay 
881a7e14dcfSSatish Balay    Input Parameters:
882441846f8SBarry Smith +  tao - the Tao solver context
883a7e14dcfSSatish Balay -  fmin - the tolerance
884a7e14dcfSSatish Balay 
885a7e14dcfSSatish Balay    Options Database Keys:
886a7e14dcfSSatish Balay .    -tao_fmin <fmin> - sets the minimum function value
887a7e14dcfSSatish Balay 
888a7e14dcfSSatish Balay    Level: intermediate
889a7e14dcfSSatish Balay 
890a7e14dcfSSatish Balay .seealso: TaoSetTolerances()
891a7e14dcfSSatish Balay @*/
892441846f8SBarry Smith PetscErrorCode TaoSetFunctionLowerBound(Tao tao,PetscReal fmin)
893a7e14dcfSSatish Balay {
894a7e14dcfSSatish Balay   PetscFunctionBegin;
895441846f8SBarry Smith   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
896a7e14dcfSSatish Balay   tao->fmin = fmin;
8976552cf8aSJason Sarich   tao->fmin_changed=PETSC_TRUE;
898a7e14dcfSSatish Balay   PetscFunctionReturn(0);
899a7e14dcfSSatish Balay }
900a7e14dcfSSatish Balay 
901a7e14dcfSSatish Balay /*@
9028e96d397SJason Sarich    TaoGetFunctionLowerBound - Gets the bound on the solution objective value.
903a7e14dcfSSatish Balay    When an approximate solution with an objective value below this number
904a7e14dcfSSatish Balay    has been found, the solver will terminate.
905a7e14dcfSSatish Balay 
906441846f8SBarry Smith    Not collective on Tao
907a7e14dcfSSatish Balay 
908a7e14dcfSSatish Balay    Input Parameters:
909441846f8SBarry Smith .  tao - the Tao solver context
910a7e14dcfSSatish Balay 
911a7e14dcfSSatish Balay    OutputParameters:
912a7e14dcfSSatish Balay .  fmin - the minimum function value
913a7e14dcfSSatish Balay 
914a7e14dcfSSatish Balay    Level: intermediate
915a7e14dcfSSatish Balay 
916a7e14dcfSSatish Balay .seealso: TaoSetFunctionLowerBound()
917a7e14dcfSSatish Balay @*/
918441846f8SBarry Smith PetscErrorCode TaoGetFunctionLowerBound(Tao tao,PetscReal *fmin)
919a7e14dcfSSatish Balay {
920a7e14dcfSSatish Balay   PetscFunctionBegin;
921441846f8SBarry Smith   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
922a7e14dcfSSatish Balay   *fmin = tao->fmin;
923a7e14dcfSSatish Balay   PetscFunctionReturn(0);
924a7e14dcfSSatish Balay }
925a7e14dcfSSatish Balay 
926a7e14dcfSSatish Balay /*@
927a7e14dcfSSatish Balay    TaoSetMaximumFunctionEvaluations - Sets a maximum number of
928a7e14dcfSSatish Balay    function evaluations.
929a7e14dcfSSatish Balay 
930441846f8SBarry Smith    Logically Collective on Tao
931a7e14dcfSSatish Balay 
932a7e14dcfSSatish Balay    Input Parameters:
933441846f8SBarry Smith +  tao - the Tao solver context
934a7e14dcfSSatish Balay -  nfcn - the maximum number of function evaluations (>=0)
935a7e14dcfSSatish Balay 
936a7e14dcfSSatish Balay    Options Database Keys:
937a7e14dcfSSatish Balay .    -tao_max_funcs <nfcn> - sets the maximum number of function evaluations
938a7e14dcfSSatish Balay 
939a7e14dcfSSatish Balay    Level: intermediate
940a7e14dcfSSatish Balay 
941a7e14dcfSSatish Balay .seealso: TaoSetTolerances(), TaoSetMaximumIterations()
942a7e14dcfSSatish Balay @*/
943a7e14dcfSSatish Balay 
944441846f8SBarry Smith PetscErrorCode TaoSetMaximumFunctionEvaluations(Tao tao,PetscInt nfcn)
945a7e14dcfSSatish Balay {
946a7e14dcfSSatish Balay   PetscFunctionBegin;
947441846f8SBarry Smith   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
94847a47007SBarry Smith   tao->max_funcs = PetscMax(0,nfcn);
9496552cf8aSJason Sarich   tao->max_funcs_changed=PETSC_TRUE;
950a7e14dcfSSatish Balay   PetscFunctionReturn(0);
951a7e14dcfSSatish Balay }
952a7e14dcfSSatish Balay 
953a7e14dcfSSatish Balay /*@
954a7e14dcfSSatish Balay    TaoGetMaximumFunctionEvaluations - Sets a maximum number of
955a7e14dcfSSatish Balay    function evaluations.
956a7e14dcfSSatish Balay 
957a7e14dcfSSatish Balay    Not Collective
958a7e14dcfSSatish Balay 
959a7e14dcfSSatish Balay    Input Parameters:
960441846f8SBarry Smith .  tao - the Tao solver context
961a7e14dcfSSatish Balay 
962a7e14dcfSSatish Balay    Output Parameters:
963a7e14dcfSSatish Balay .  nfcn - the maximum number of function evaluations
964a7e14dcfSSatish Balay 
965a7e14dcfSSatish Balay    Level: intermediate
966a7e14dcfSSatish Balay 
967a7e14dcfSSatish Balay .seealso: TaoSetMaximumFunctionEvaluations(), TaoGetMaximumIterations()
968a7e14dcfSSatish Balay @*/
969a7e14dcfSSatish Balay 
970441846f8SBarry Smith PetscErrorCode TaoGetMaximumFunctionEvaluations(Tao tao,PetscInt *nfcn)
971a7e14dcfSSatish Balay {
972a7e14dcfSSatish Balay   PetscFunctionBegin;
973441846f8SBarry Smith   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
974a7e14dcfSSatish Balay   *nfcn = tao->max_funcs;
975a7e14dcfSSatish Balay   PetscFunctionReturn(0);
976a7e14dcfSSatish Balay }
977a7e14dcfSSatish Balay 
978770232b9SCe Qin /*@
979770232b9SCe Qin    TaoGetCurrentFunctionEvaluations - Get current number of
980770232b9SCe Qin    function evaluations.
981770232b9SCe Qin 
982770232b9SCe Qin    Not Collective
983770232b9SCe Qin 
984770232b9SCe Qin    Input Parameters:
985770232b9SCe Qin .  tao - the Tao solver context
986770232b9SCe Qin 
987770232b9SCe Qin    Output Parameters:
988770232b9SCe Qin .  nfuncs - the current number of function evaluations
989770232b9SCe Qin 
990770232b9SCe Qin    Level: intermediate
991770232b9SCe Qin 
992770232b9SCe Qin .seealso: TaoSetMaximumFunctionEvaluations(), TaoGetMaximumFunctionEvaluations(), TaoGetMaximumIterations()
993770232b9SCe Qin @*/
994770232b9SCe Qin 
995770232b9SCe Qin PetscErrorCode TaoGetCurrentFunctionEvaluations(Tao tao,PetscInt *nfuncs)
996770232b9SCe Qin {
997770232b9SCe Qin   PetscFunctionBegin;
998770232b9SCe Qin   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
999770232b9SCe Qin   *nfuncs=PetscMax(tao->nfuncs,tao->nfuncgrads);
1000770232b9SCe Qin   PetscFunctionReturn(0);
1001770232b9SCe Qin }
1002770232b9SCe Qin 
1003a7e14dcfSSatish Balay /*@
1004a7e14dcfSSatish Balay    TaoSetMaximumIterations - Sets a maximum number of iterates.
1005a7e14dcfSSatish Balay 
1006441846f8SBarry Smith    Logically Collective on Tao
1007a7e14dcfSSatish Balay 
1008a7e14dcfSSatish Balay    Input Parameters:
1009441846f8SBarry Smith +  tao - the Tao solver context
1010a7e14dcfSSatish Balay -  maxits - the maximum number of iterates (>=0)
1011a7e14dcfSSatish Balay 
1012a7e14dcfSSatish Balay    Options Database Keys:
1013a7e14dcfSSatish Balay .    -tao_max_it <its> - sets the maximum number of iterations
1014a7e14dcfSSatish Balay 
1015a7e14dcfSSatish Balay    Level: intermediate
1016a7e14dcfSSatish Balay 
1017a7e14dcfSSatish Balay .seealso: TaoSetTolerances(), TaoSetMaximumFunctionEvaluations()
1018a7e14dcfSSatish Balay @*/
1019441846f8SBarry Smith PetscErrorCode TaoSetMaximumIterations(Tao tao,PetscInt maxits)
1020a7e14dcfSSatish Balay {
1021a7e14dcfSSatish Balay   PetscFunctionBegin;
1022441846f8SBarry Smith   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
102347a47007SBarry Smith   tao->max_it = PetscMax(0,maxits);
10246552cf8aSJason Sarich   tao->max_it_changed=PETSC_TRUE;
1025a7e14dcfSSatish Balay   PetscFunctionReturn(0);
1026a7e14dcfSSatish Balay }
1027a7e14dcfSSatish Balay 
1028a7e14dcfSSatish Balay /*@
1029a7e14dcfSSatish Balay    TaoGetMaximumIterations - Sets a maximum number of iterates.
1030a7e14dcfSSatish Balay 
1031a7e14dcfSSatish Balay    Not Collective
1032a7e14dcfSSatish Balay 
1033a7e14dcfSSatish Balay    Input Parameters:
1034441846f8SBarry Smith .  tao - the Tao solver context
1035a7e14dcfSSatish Balay 
1036a7e14dcfSSatish Balay    Output Parameters:
1037a7e14dcfSSatish Balay .  maxits - the maximum number of iterates
1038a7e14dcfSSatish Balay 
1039a7e14dcfSSatish Balay    Level: intermediate
1040a7e14dcfSSatish Balay 
1041a7e14dcfSSatish Balay .seealso: TaoSetMaximumIterations(), TaoGetMaximumFunctionEvaluations()
1042a7e14dcfSSatish Balay @*/
1043441846f8SBarry Smith PetscErrorCode TaoGetMaximumIterations(Tao tao,PetscInt *maxits)
1044a7e14dcfSSatish Balay {
1045a7e14dcfSSatish Balay   PetscFunctionBegin;
1046441846f8SBarry Smith   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
1047a7e14dcfSSatish Balay   *maxits = tao->max_it;
1048a7e14dcfSSatish Balay   PetscFunctionReturn(0);
1049a7e14dcfSSatish Balay }
1050a7e14dcfSSatish Balay 
1051a7e14dcfSSatish Balay /*@
1052a7e14dcfSSatish Balay    TaoSetInitialTrustRegionRadius - Sets the initial trust region radius.
1053a7e14dcfSSatish Balay 
1054441846f8SBarry Smith    Logically collective on Tao
1055a7e14dcfSSatish Balay 
1056a7e14dcfSSatish Balay    Input Parameter:
1057a7e14dcfSSatish Balay +  tao - a TAO optimization solver
1058a7e14dcfSSatish Balay -  radius - the trust region radius
1059a7e14dcfSSatish Balay 
1060a7e14dcfSSatish Balay    Level: intermediate
1061a7e14dcfSSatish Balay 
1062a7e14dcfSSatish Balay    Options Database Key:
1063a7e14dcfSSatish Balay .  -tao_trust0 <t0> - sets initial trust region radius
1064a7e14dcfSSatish Balay 
1065a7e14dcfSSatish Balay .seealso: TaoGetTrustRegionRadius(), TaoSetTrustRegionTolerance()
1066a7e14dcfSSatish Balay @*/
1067441846f8SBarry Smith PetscErrorCode TaoSetInitialTrustRegionRadius(Tao tao, PetscReal radius)
1068a7e14dcfSSatish Balay {
1069a7e14dcfSSatish Balay   PetscFunctionBegin;
1070441846f8SBarry Smith   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
1071a7e14dcfSSatish Balay   tao->trust0 = PetscMax(0.0,radius);
10726552cf8aSJason Sarich   tao->trust0_changed=PETSC_TRUE;
1073a7e14dcfSSatish Balay   PetscFunctionReturn(0);
1074a7e14dcfSSatish Balay }
1075a7e14dcfSSatish Balay 
1076a7e14dcfSSatish Balay /*@
1077a7e14dcfSSatish Balay    TaoGetInitialTrustRegionRadius - Sets the initial trust region radius.
1078a7e14dcfSSatish Balay 
1079a7e14dcfSSatish Balay    Not Collective
1080a7e14dcfSSatish Balay 
1081a7e14dcfSSatish Balay    Input Parameter:
1082a7e14dcfSSatish Balay .  tao - a TAO optimization solver
1083a7e14dcfSSatish Balay 
1084a7e14dcfSSatish Balay    Output Parameter:
1085a7e14dcfSSatish Balay .  radius - the trust region radius
1086a7e14dcfSSatish Balay 
1087a7e14dcfSSatish Balay    Level: intermediate
1088a7e14dcfSSatish Balay 
1089a7e14dcfSSatish Balay .seealso: TaoSetInitialTrustRegionRadius(), TaoGetCurrentTrustRegionRadius()
1090a7e14dcfSSatish Balay @*/
1091441846f8SBarry Smith PetscErrorCode TaoGetInitialTrustRegionRadius(Tao tao, PetscReal *radius)
1092a7e14dcfSSatish Balay {
1093a7e14dcfSSatish Balay   PetscFunctionBegin;
1094441846f8SBarry Smith   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
1095a7e14dcfSSatish Balay   *radius = tao->trust0;
1096a7e14dcfSSatish Balay   PetscFunctionReturn(0);
1097a7e14dcfSSatish Balay }
1098a7e14dcfSSatish Balay 
1099a7e14dcfSSatish Balay /*@
1100a7e14dcfSSatish Balay    TaoGetCurrentTrustRegionRadius - Gets the current trust region radius.
1101a7e14dcfSSatish Balay 
1102a7e14dcfSSatish Balay    Not Collective
1103a7e14dcfSSatish Balay 
1104a7e14dcfSSatish Balay    Input Parameter:
1105a7e14dcfSSatish Balay .  tao - a TAO optimization solver
1106a7e14dcfSSatish Balay 
1107a7e14dcfSSatish Balay    Output Parameter:
1108a7e14dcfSSatish Balay .  radius - the trust region radius
1109a7e14dcfSSatish Balay 
1110a7e14dcfSSatish Balay    Level: intermediate
1111a7e14dcfSSatish Balay 
1112a7e14dcfSSatish Balay .seealso: TaoSetInitialTrustRegionRadius(), TaoGetInitialTrustRegionRadius()
1113a7e14dcfSSatish Balay @*/
1114441846f8SBarry Smith PetscErrorCode TaoGetCurrentTrustRegionRadius(Tao tao, PetscReal *radius)
1115a7e14dcfSSatish Balay {
1116a7e14dcfSSatish Balay   PetscFunctionBegin;
1117441846f8SBarry Smith   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
1118a7e14dcfSSatish Balay   *radius = tao->trust;
1119a7e14dcfSSatish Balay   PetscFunctionReturn(0);
1120a7e14dcfSSatish Balay }
1121a7e14dcfSSatish Balay 
1122a7e14dcfSSatish Balay /*@
1123a7e14dcfSSatish Balay   TaoGetTolerances - gets the current values of tolerances
1124a7e14dcfSSatish Balay 
1125a7e14dcfSSatish Balay   Not Collective
1126a7e14dcfSSatish Balay 
1127a7e14dcfSSatish Balay   Input Parameters:
1128441846f8SBarry Smith . tao - the Tao context
1129a7e14dcfSSatish Balay 
1130a7e14dcfSSatish Balay   Output Parameters:
1131e52336cbSBarry Smith + gatol - stop if norm of gradient is less than this
1132a7e14dcfSSatish Balay . grtol - stop if relative norm of gradient is less than this
1133a7e14dcfSSatish Balay - gttol - stop if norm of gradient is reduced by a this factor
1134a7e14dcfSSatish Balay 
113547a47007SBarry Smith   Note: NULL can be used as an argument if not all tolerances values are needed
1136a7e14dcfSSatish Balay 
1137a7e14dcfSSatish Balay .seealso TaoSetTolerances()
1138a7e14dcfSSatish Balay 
1139a7e14dcfSSatish Balay   Level: intermediate
1140a7e14dcfSSatish Balay @*/
1141e52336cbSBarry Smith PetscErrorCode TaoGetTolerances(Tao tao, PetscReal *gatol, PetscReal *grtol, PetscReal *gttol)
1142a7e14dcfSSatish Balay {
1143a7e14dcfSSatish Balay   PetscFunctionBegin;
1144441846f8SBarry Smith   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
1145a7e14dcfSSatish Balay   if (gatol) *gatol=tao->gatol;
1146a7e14dcfSSatish Balay   if (grtol) *grtol=tao->grtol;
1147a7e14dcfSSatish Balay   if (gttol) *gttol=tao->gttol;
1148a7e14dcfSSatish Balay   PetscFunctionReturn(0);
1149a7e14dcfSSatish Balay }
1150a7e14dcfSSatish Balay 
1151a7e14dcfSSatish Balay /*@
1152a7e14dcfSSatish Balay   TaoGetKSP - Gets the linear solver used by the optimization solver.
1153a7e14dcfSSatish Balay   Application writers should use TaoGetKSP if they need direct access
1154a7e14dcfSSatish Balay   to the PETSc KSP object.
1155a7e14dcfSSatish Balay 
1156a7e14dcfSSatish Balay   Not Collective
1157a7e14dcfSSatish Balay 
1158a7e14dcfSSatish Balay    Input Parameters:
1159a7e14dcfSSatish Balay .  tao - the TAO solver
1160a7e14dcfSSatish Balay 
1161a7e14dcfSSatish Balay    Output Parameters:
1162a7e14dcfSSatish Balay .  ksp - the KSP linear solver used in the optimization solver
1163a7e14dcfSSatish Balay 
1164a7e14dcfSSatish Balay    Level: intermediate
1165a7e14dcfSSatish Balay 
1166a7e14dcfSSatish Balay @*/
1167441846f8SBarry Smith PetscErrorCode TaoGetKSP(Tao tao, KSP *ksp)
116847a47007SBarry Smith {
1169a7e14dcfSSatish Balay   PetscFunctionBegin;
1170a7e14dcfSSatish Balay   *ksp = tao->ksp;
1171a7e14dcfSSatish Balay   PetscFunctionReturn(0);
1172a7e14dcfSSatish Balay }
1173a7e14dcfSSatish Balay 
1174025e9500SJason Sarich /*@
1175025e9500SJason Sarich    TaoGetLinearSolveIterations - Gets the total number of linear iterations
1176025e9500SJason Sarich    used by the TAO solver
1177025e9500SJason Sarich 
1178025e9500SJason Sarich    Not Collective
1179025e9500SJason Sarich 
1180025e9500SJason Sarich    Input Parameter:
1181025e9500SJason Sarich .  tao - TAO context
1182025e9500SJason Sarich 
1183025e9500SJason Sarich    Output Parameter:
1184025e9500SJason Sarich .  lits - number of linear iterations
1185025e9500SJason Sarich 
1186025e9500SJason Sarich    Notes:
1187025e9500SJason Sarich    This counter is reset to zero for each successive call to TaoSolve()
1188025e9500SJason Sarich 
1189025e9500SJason Sarich    Level: intermediate
1190025e9500SJason Sarich 
1191025e9500SJason Sarich .keywords: TAO
1192025e9500SJason Sarich 
1193025e9500SJason Sarich .seealso:  TaoGetKSP()
1194025e9500SJason Sarich @*/
1195025e9500SJason Sarich PetscErrorCode  TaoGetLinearSolveIterations(Tao tao,PetscInt *lits)
1196025e9500SJason Sarich {
1197025e9500SJason Sarich   PetscFunctionBegin;
1198025e9500SJason Sarich   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
1199025e9500SJason Sarich   PetscValidIntPointer(lits,2);
12002d9aa51bSJason Sarich   *lits = tao->ksp_tot_its;
1201025e9500SJason Sarich   PetscFunctionReturn(0);
1202025e9500SJason Sarich }
1203025e9500SJason Sarich 
1204a7e14dcfSSatish Balay /*@
1205a7e14dcfSSatish Balay   TaoGetLineSearch - Gets the line search used by the optimization solver.
1206a7e14dcfSSatish Balay   Application writers should use TaoGetLineSearch if they need direct access
1207a7e14dcfSSatish Balay   to the TaoLineSearch object.
1208a7e14dcfSSatish Balay 
1209a7e14dcfSSatish Balay   Not Collective
1210a7e14dcfSSatish Balay 
1211a7e14dcfSSatish Balay    Input Parameters:
1212a7e14dcfSSatish Balay .  tao - the TAO solver
1213a7e14dcfSSatish Balay 
1214a7e14dcfSSatish Balay    Output Parameters:
1215a7e14dcfSSatish Balay .  ls - the line search used in the optimization solver
1216a7e14dcfSSatish Balay 
1217a7e14dcfSSatish Balay    Level: intermediate
1218a7e14dcfSSatish Balay 
1219a7e14dcfSSatish Balay @*/
1220441846f8SBarry Smith PetscErrorCode TaoGetLineSearch(Tao tao, TaoLineSearch *ls)
122147a47007SBarry Smith {
1222a7e14dcfSSatish Balay   PetscFunctionBegin;
1223a7e14dcfSSatish Balay   *ls = tao->linesearch;
1224a7e14dcfSSatish Balay   PetscFunctionReturn(0);
1225a7e14dcfSSatish Balay }
1226a7e14dcfSSatish Balay 
1227a7e14dcfSSatish Balay /*@
1228a7e14dcfSSatish Balay   TaoAddLineSearchCounts - Adds the number of function evaluations spent
1229a7e14dcfSSatish Balay   in the line search to the running total.
1230a7e14dcfSSatish Balay 
1231a7e14dcfSSatish Balay    Input Parameters:
1232a7e14dcfSSatish Balay +  tao - the TAO solver
1233a7e14dcfSSatish Balay -  ls - the line search used in the optimization solver
1234a7e14dcfSSatish Balay 
1235a7e14dcfSSatish Balay    Level: developer
1236a7e14dcfSSatish Balay 
1237a7e14dcfSSatish Balay .seealso: TaoLineSearchApply()
1238a7e14dcfSSatish Balay @*/
1239441846f8SBarry Smith PetscErrorCode TaoAddLineSearchCounts(Tao tao)
124047a47007SBarry Smith {
1241a7e14dcfSSatish Balay   PetscErrorCode ierr;
1242a7e14dcfSSatish Balay   PetscBool      flg;
1243a7e14dcfSSatish Balay   PetscInt       nfeval,ngeval,nfgeval;
124447a47007SBarry Smith 
1245a7e14dcfSSatish Balay   PetscFunctionBegin;
1246441846f8SBarry Smith   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
1247a7e14dcfSSatish Balay   if (tao->linesearch) {
1248302440fdSBarry Smith     ierr = TaoLineSearchIsUsingTaoRoutines(tao->linesearch,&flg);CHKERRQ(ierr);
124994ae4db5SBarry Smith     if (!flg) {
125047a47007SBarry Smith       ierr = TaoLineSearchGetNumberFunctionEvaluations(tao->linesearch,&nfeval,&ngeval,&nfgeval);CHKERRQ(ierr);
1251a7e14dcfSSatish Balay       tao->nfuncs+=nfeval;
1252a7e14dcfSSatish Balay       tao->ngrads+=ngeval;
1253a7e14dcfSSatish Balay       tao->nfuncgrads+=nfgeval;
1254a7e14dcfSSatish Balay     }
1255a7e14dcfSSatish Balay   }
1256a7e14dcfSSatish Balay   PetscFunctionReturn(0);
1257a7e14dcfSSatish Balay }
1258a7e14dcfSSatish Balay 
1259a7e14dcfSSatish Balay /*@
1260a7e14dcfSSatish Balay   TaoGetSolutionVector - Returns the vector with the current TAO solution
1261a7e14dcfSSatish Balay 
1262a7e14dcfSSatish Balay   Not Collective
1263a7e14dcfSSatish Balay 
1264a7e14dcfSSatish Balay   Input Parameter:
1265441846f8SBarry Smith . tao - the Tao context
1266a7e14dcfSSatish Balay 
1267a7e14dcfSSatish Balay   Output Parameter:
1268a7e14dcfSSatish Balay . X - the current solution
1269a7e14dcfSSatish Balay 
1270a7e14dcfSSatish Balay   Level: intermediate
1271a7e14dcfSSatish Balay 
1272a7e14dcfSSatish Balay   Note:  The returned vector will be the same object that was passed into TaoSetInitialVector()
1273a7e14dcfSSatish Balay @*/
1274441846f8SBarry Smith PetscErrorCode TaoGetSolutionVector(Tao tao, Vec *X)
1275a7e14dcfSSatish Balay {
1276a7e14dcfSSatish Balay   PetscFunctionBegin;
1277441846f8SBarry Smith   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
1278a7e14dcfSSatish Balay   *X = tao->solution;
1279a7e14dcfSSatish Balay   PetscFunctionReturn(0);
1280a7e14dcfSSatish Balay }
1281a7e14dcfSSatish Balay 
1282a7e14dcfSSatish Balay /*@
1283a7e14dcfSSatish Balay   TaoGetGradientVector - Returns the vector with the current TAO gradient
1284a7e14dcfSSatish Balay 
1285a7e14dcfSSatish Balay   Not Collective
1286a7e14dcfSSatish Balay 
1287a7e14dcfSSatish Balay   Input Parameter:
1288441846f8SBarry Smith . tao - the Tao context
1289a7e14dcfSSatish Balay 
1290a7e14dcfSSatish Balay   Output Parameter:
1291a7e14dcfSSatish Balay . G - the current solution
1292a7e14dcfSSatish Balay 
1293a7e14dcfSSatish Balay   Level: intermediate
1294a7e14dcfSSatish Balay @*/
1295441846f8SBarry Smith PetscErrorCode TaoGetGradientVector(Tao tao, Vec *G)
1296a7e14dcfSSatish Balay {
1297a7e14dcfSSatish Balay   PetscFunctionBegin;
1298441846f8SBarry Smith   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
1299a7e14dcfSSatish Balay   *G = tao->gradient;
1300a7e14dcfSSatish Balay   PetscFunctionReturn(0);
1301a7e14dcfSSatish Balay }
1302a7e14dcfSSatish Balay 
1303a7e14dcfSSatish Balay /*@
1304a7e14dcfSSatish Balay    TaoResetStatistics - Initialize the statistics used by TAO for all of the solvers.
1305a7e14dcfSSatish Balay    These statistics include the iteration number, residual norms, and convergence status.
1306a7e14dcfSSatish Balay    This routine gets called before solving each optimization problem.
1307a7e14dcfSSatish Balay 
1308441846f8SBarry Smith    Collective on Tao
1309a7e14dcfSSatish Balay 
1310a7e14dcfSSatish Balay    Input Parameters:
1311441846f8SBarry Smith .  solver - the Tao context
1312a7e14dcfSSatish Balay 
1313a7e14dcfSSatish Balay    Level: developer
1314a7e14dcfSSatish Balay 
1315a7e14dcfSSatish Balay .seealso: TaoCreate(), TaoSolve()
1316a7e14dcfSSatish Balay @*/
1317441846f8SBarry Smith PetscErrorCode TaoResetStatistics(Tao tao)
1318a7e14dcfSSatish Balay {
1319a7e14dcfSSatish Balay   PetscFunctionBegin;
1320441846f8SBarry Smith   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
1321a7e14dcfSSatish Balay   tao->niter        = 0;
1322a7e14dcfSSatish Balay   tao->nfuncs       = 0;
1323a7e14dcfSSatish Balay   tao->nfuncgrads   = 0;
1324a7e14dcfSSatish Balay   tao->ngrads       = 0;
1325a7e14dcfSSatish Balay   tao->nhess        = 0;
1326a7e14dcfSSatish Balay   tao->njac         = 0;
1327a7e14dcfSSatish Balay   tao->nconstraints = 0;
1328a7e14dcfSSatish Balay   tao->ksp_its      = 0;
1329ae93cb3cSJason Sarich   tao->ksp_tot_its  = 0;
1330a7e14dcfSSatish Balay   tao->reason       = TAO_CONTINUE_ITERATING;
1331a7e14dcfSSatish Balay   tao->residual     = 0.0;
1332a7e14dcfSSatish Balay   tao->cnorm        = 0.0;
1333a7e14dcfSSatish Balay   tao->step         = 0.0;
1334a7e14dcfSSatish Balay   tao->lsflag       = PETSC_FALSE;
1335a7e14dcfSSatish Balay   if (tao->hist_reset) tao->hist_len=0;
1336a7e14dcfSSatish Balay   PetscFunctionReturn(0);
1337a7e14dcfSSatish Balay }
1338a7e14dcfSSatish Balay 
1339a7e14dcfSSatish Balay /*@C
1340*e1e80dc8SAlp Dener   TaoSetUpdate - Sets the general-purpose update function called
1341*e1e80dc8SAlp Dener   at the beginning of every iteration of the nonlinear solve. Specifically
1342*e1e80dc8SAlp Dener   it is called at the top of every iteration, after the new solution and the gradient
1343*e1e80dc8SAlp Dener   is determined, but before the Hessian is computed (if applicable).
1344*e1e80dc8SAlp Dener 
1345*e1e80dc8SAlp Dener   Logically Collective on Tao
1346*e1e80dc8SAlp Dener 
1347*e1e80dc8SAlp Dener   Input Parameters:
1348*e1e80dc8SAlp Dener . tao - The tao solver context
1349*e1e80dc8SAlp Dener . func - The function
1350*e1e80dc8SAlp Dener 
1351*e1e80dc8SAlp Dener   Calling sequence of func:
1352*e1e80dc8SAlp Dener . func (Tao tao, PetscInt step);
1353*e1e80dc8SAlp Dener 
1354*e1e80dc8SAlp Dener . step - The current step of the iteration
1355*e1e80dc8SAlp Dener 
1356*e1e80dc8SAlp Dener   Level: advanced
1357*e1e80dc8SAlp Dener 
1358*e1e80dc8SAlp Dener .keywords: Tao, update
1359*e1e80dc8SAlp Dener 
1360*e1e80dc8SAlp Dener .seealso TaoSolve()
1361*e1e80dc8SAlp Dener @*/
1362*e1e80dc8SAlp Dener PetscErrorCode  TaoSetUpdate(Tao tao, PetscErrorCode (*func)(Tao, PetscInt), void *ctx)
1363*e1e80dc8SAlp Dener {
1364*e1e80dc8SAlp Dener   PetscFunctionBegin;
1365*e1e80dc8SAlp Dener   PetscValidHeaderSpecific(tao, TAO_CLASSID,1);
1366*e1e80dc8SAlp Dener   tao->ops->update = func;
1367*e1e80dc8SAlp Dener   tao->user_update = ctx;
1368*e1e80dc8SAlp Dener   PetscFunctionReturn(0);
1369*e1e80dc8SAlp Dener }
1370*e1e80dc8SAlp Dener 
1371*e1e80dc8SAlp Dener /*@C
1372a7e14dcfSSatish Balay   TaoSetConvergenceTest - Sets the function that is to be used to test
1373a7e14dcfSSatish Balay   for convergence o fthe iterative minimization solution.  The new convergence
1374a7e14dcfSSatish Balay   testing routine will replace TAO's default convergence test.
1375a7e14dcfSSatish Balay 
1376441846f8SBarry Smith   Logically Collective on Tao
1377a7e14dcfSSatish Balay 
1378a7e14dcfSSatish Balay   Input Parameters:
1379441846f8SBarry Smith + tao - the Tao object
1380a7e14dcfSSatish Balay . conv - the routine to test for convergence
1381a7e14dcfSSatish Balay - ctx - [optional] context for private data for the convergence routine
138247a47007SBarry Smith         (may be NULL)
1383a7e14dcfSSatish Balay 
1384a7e14dcfSSatish Balay   Calling sequence of conv:
1385441846f8SBarry Smith $   PetscErrorCode conv(Tao tao, void *ctx)
1386a7e14dcfSSatish Balay 
1387441846f8SBarry Smith + tao - the Tao object
1388a7e14dcfSSatish Balay - ctx - [optional] convergence context
1389a7e14dcfSSatish Balay 
1390e4cb33bbSBarry Smith   Note: The new convergence testing routine should call TaoSetConvergedReason().
1391a7e14dcfSSatish Balay 
1392a7e14dcfSSatish Balay   Level: advanced
1393a7e14dcfSSatish Balay 
1394e4cb33bbSBarry Smith .seealso: TaoSetConvergedReason(), TaoGetSolutionStatus(), TaoGetTolerances(), TaoSetMonitor
1395a7e14dcfSSatish Balay 
1396a7e14dcfSSatish Balay @*/
1397441846f8SBarry Smith PetscErrorCode TaoSetConvergenceTest(Tao tao, PetscErrorCode (*conv)(Tao,void*), void *ctx)
1398a7e14dcfSSatish Balay {
1399a7e14dcfSSatish Balay   PetscFunctionBegin;
1400441846f8SBarry Smith   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
1401a7e14dcfSSatish Balay   (tao)->ops->convergencetest = conv;
1402a7e14dcfSSatish Balay   (tao)->cnvP = ctx;
1403a7e14dcfSSatish Balay   PetscFunctionReturn(0);
1404a7e14dcfSSatish Balay }
1405a7e14dcfSSatish Balay 
1406a7e14dcfSSatish Balay /*@C
1407a7e14dcfSSatish Balay    TaoSetMonitor - Sets an ADDITIONAL function that is to be used at every
1408a7e14dcfSSatish Balay    iteration of the solver to display the iteration's
1409a7e14dcfSSatish Balay    progress.
1410a7e14dcfSSatish Balay 
1411441846f8SBarry Smith    Logically Collective on Tao
1412a7e14dcfSSatish Balay 
1413a7e14dcfSSatish Balay    Input Parameters:
1414441846f8SBarry Smith +  tao - the Tao solver context
1415a7e14dcfSSatish Balay .  mymonitor - monitoring routine
1416a7e14dcfSSatish Balay -  mctx - [optional] user-defined context for private data for the
141747a47007SBarry Smith           monitor routine (may be NULL)
1418a7e14dcfSSatish Balay 
1419a7e14dcfSSatish Balay    Calling sequence of mymonitor:
1420441846f8SBarry Smith $     int mymonitor(Tao tao,void *mctx)
1421a7e14dcfSSatish Balay 
1422441846f8SBarry Smith +    tao - the Tao solver context
1423a7e14dcfSSatish Balay -    mctx - [optional] monitoring context
1424a7e14dcfSSatish Balay 
1425a7e14dcfSSatish Balay 
1426a7e14dcfSSatish Balay    Options Database Keys:
142798ea980cSBarry Smith +    -tao_monitor        - sets TaoMonitorDefault()
1428a7e14dcfSSatish Balay .    -tao_smonitor       - sets short monitor
1429a7e14dcfSSatish Balay .    -tao_cmonitor       - same as smonitor plus constraint norm
1430a7e14dcfSSatish Balay .    -tao_view_solution   - view solution at each iteration
1431a7e14dcfSSatish Balay .    -tao_view_gradient   - view gradient at each iteration
14324a48860cSAlp Dener .    -tao_view_ls_residual - view least-squares residual vector at each iteration
1433a7e14dcfSSatish 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.
1434a7e14dcfSSatish Balay 
1435a7e14dcfSSatish Balay 
1436a7e14dcfSSatish Balay    Notes:
1437a7e14dcfSSatish Balay    Several different monitoring routines may be set by calling
1438a7e14dcfSSatish Balay    TaoSetMonitor() multiple times; all will be called in the
1439a7e14dcfSSatish Balay    order in which they were set.
1440a7e14dcfSSatish Balay 
144195452b02SPatrick Sanan    Fortran Notes:
144295452b02SPatrick Sanan     Only one monitor function may be set
1443a7e14dcfSSatish Balay 
1444a7e14dcfSSatish Balay    Level: intermediate
1445a7e14dcfSSatish Balay 
144698ea980cSBarry Smith .seealso: TaoMonitorDefault(), TaoCancelMonitors(),  TaoSetDestroyRoutine()
1447a7e14dcfSSatish Balay @*/
1448441846f8SBarry Smith PetscErrorCode TaoSetMonitor(Tao tao, PetscErrorCode (*func)(Tao, void*), void *ctx,PetscErrorCode (*dest)(void**))
1449a7e14dcfSSatish Balay {
145008d19d1fSJason Sarich   PetscErrorCode ierr;
145108d19d1fSJason Sarich   PetscInt       i;
14528732526dSAlp Dener   PetscBool      identical;
145308d19d1fSJason Sarich 
1454a7e14dcfSSatish Balay   PetscFunctionBegin;
1455441846f8SBarry Smith   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
145647a47007SBarry Smith   if (tao->numbermonitors >= MAXTAOMONITORS) SETERRQ1(PETSC_COMM_SELF,1,"Cannot attach another monitor -- max=",MAXTAOMONITORS);
145708d19d1fSJason Sarich 
145808d19d1fSJason Sarich   for (i=0; i<tao->numbermonitors;i++) {
14598732526dSAlp Dener     ierr = PetscMonitorCompare((PetscErrorCode (*)(void))func,ctx,dest,(PetscErrorCode (*)(void))tao->monitor[i],tao->monitorcontext[i],tao->monitordestroy[i],&identical);CHKERRQ(ierr);
14608732526dSAlp Dener     if (identical) PetscFunctionReturn(0);
146108d19d1fSJason Sarich   }
1462a7e14dcfSSatish Balay   tao->monitor[tao->numbermonitors] = func;
14638732526dSAlp Dener   tao->monitorcontext[tao->numbermonitors] = (void*)ctx;
1464a7e14dcfSSatish Balay   tao->monitordestroy[tao->numbermonitors] = dest;
1465a7e14dcfSSatish Balay   ++tao->numbermonitors;
1466a7e14dcfSSatish Balay   PetscFunctionReturn(0);
1467a7e14dcfSSatish Balay }
1468a7e14dcfSSatish Balay 
1469a7e14dcfSSatish Balay /*@
1470441846f8SBarry Smith    TaoCancelMonitors - Clears all the monitor functions for a Tao object.
1471a7e14dcfSSatish Balay 
1472441846f8SBarry Smith    Logically Collective on Tao
1473a7e14dcfSSatish Balay 
1474a7e14dcfSSatish Balay    Input Parameters:
1475441846f8SBarry Smith .  tao - the Tao solver context
1476a7e14dcfSSatish Balay 
1477a7e14dcfSSatish Balay    Options Database:
1478a7e14dcfSSatish Balay .  -tao_cancelmonitors - cancels all monitors that have been hardwired
1479a7e14dcfSSatish Balay     into a code by calls to TaoSetMonitor(), but does not cancel those
1480a7e14dcfSSatish Balay     set via the options database
1481a7e14dcfSSatish Balay 
1482a7e14dcfSSatish Balay    Notes:
1483441846f8SBarry Smith    There is no way to clear one specific monitor from a Tao object.
1484a7e14dcfSSatish Balay 
1485a7e14dcfSSatish Balay    Level: advanced
1486a7e14dcfSSatish Balay 
148798ea980cSBarry Smith .seealso: TaoMonitorDefault(), TaoSetMonitor()
1488a7e14dcfSSatish Balay @*/
1489441846f8SBarry Smith PetscErrorCode TaoCancelMonitors(Tao tao)
1490a7e14dcfSSatish Balay {
1491a7e14dcfSSatish Balay   PetscInt       i;
1492a7e14dcfSSatish Balay   PetscErrorCode ierr;
149347a47007SBarry Smith 
1494a7e14dcfSSatish Balay   PetscFunctionBegin;
1495441846f8SBarry Smith   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
1496a7e14dcfSSatish Balay   for (i=0;i<tao->numbermonitors;i++) {
1497a7e14dcfSSatish Balay     if (tao->monitordestroy[i]) {
1498a7e14dcfSSatish Balay       ierr = (*tao->monitordestroy[i])(&tao->monitorcontext[i]);CHKERRQ(ierr);
1499a7e14dcfSSatish Balay     }
1500a7e14dcfSSatish Balay   }
1501a7e14dcfSSatish Balay   tao->numbermonitors=0;
1502a7e14dcfSSatish Balay   PetscFunctionReturn(0);
1503a7e14dcfSSatish Balay }
1504a7e14dcfSSatish Balay 
15057fab98ebSJason Sarich /*@
150698ea980cSBarry Smith    TaoMonitorDefault - Default routine for monitoring progress of the
1507441846f8SBarry Smith    Tao solvers (default).  This monitor prints the function value and gradient
1508a7e14dcfSSatish Balay    norm at each iteration.  It can be turned on from the command line using the
1509a7e14dcfSSatish Balay    -tao_monitor option
1510a7e14dcfSSatish Balay 
1511441846f8SBarry Smith    Collective on Tao
1512a7e14dcfSSatish Balay 
1513a7e14dcfSSatish Balay    Input Parameters:
1514441846f8SBarry Smith +  tao - the Tao context
151547a47007SBarry Smith -  ctx - PetscViewer context or NULL
1516a7e14dcfSSatish Balay 
1517a7e14dcfSSatish Balay    Options Database Keys:
1518a7e14dcfSSatish Balay .  -tao_monitor
1519a7e14dcfSSatish Balay 
1520a7e14dcfSSatish Balay    Level: advanced
1521a7e14dcfSSatish Balay 
1522a7e14dcfSSatish Balay .seealso: TaoDefaultSMonitor(), TaoSetMonitor()
1523a7e14dcfSSatish Balay @*/
152498ea980cSBarry Smith PetscErrorCode TaoMonitorDefault(Tao tao, void *ctx)
1525a7e14dcfSSatish Balay {
1526a7e14dcfSSatish Balay   PetscErrorCode ierr;
152763b15415SAlp Dener   PetscInt       its, tabs;
1528a7e14dcfSSatish Balay   PetscReal      fct,gnorm;
15298163d661SBarry Smith   PetscViewer    viewer = (PetscViewer)ctx;
153047a47007SBarry Smith 
1531a7e14dcfSSatish Balay   PetscFunctionBegin;
15328163d661SBarry Smith   PetscValidHeaderSpecific(viewer,PETSC_VIEWER_CLASSID,2);
1533a7e14dcfSSatish Balay   its=tao->niter;
1534a7e14dcfSSatish Balay   fct=tao->fc;
1535a7e14dcfSSatish Balay   gnorm=tao->residual;
153663b15415SAlp Dener   ierr = PetscViewerASCIIGetTab(viewer, &tabs);CHKERRQ(ierr);
153763b15415SAlp Dener   ierr = PetscViewerASCIISetTab(viewer, ((PetscObject)tao)->tablevel);CHKERRQ(ierr);
153863b15415SAlp Dener   if (its == 0 && ((PetscObject)tao)->prefix) {
153963b15415SAlp Dener      ierr = PetscViewerASCIIPrintf(viewer,"  Iteration information for %s solve.\n",((PetscObject)tao)->prefix);CHKERRQ(ierr);
154063b15415SAlp Dener    }
154198ea980cSBarry Smith   ierr=PetscViewerASCIIPrintf(viewer,"%3D TAO,",its);CHKERRQ(ierr);
154247a47007SBarry Smith   ierr=PetscViewerASCIIPrintf(viewer,"  Function value: %g,",(double)fct);CHKERRQ(ierr);
154308d19d1fSJason Sarich   if (gnorm >= PETSC_INFINITY) {
154408d19d1fSJason Sarich     ierr=PetscViewerASCIIPrintf(viewer,"  Residual: Inf \n");CHKERRQ(ierr);
154508d19d1fSJason Sarich   } else {
154647a47007SBarry Smith     ierr=PetscViewerASCIIPrintf(viewer,"  Residual: %g \n",(double)gnorm);CHKERRQ(ierr);
154708d19d1fSJason Sarich   }
154863b15415SAlp Dener   ierr = PetscViewerASCIISetTab(viewer, tabs);CHKERRQ(ierr);
1549a7e14dcfSSatish Balay   PetscFunctionReturn(0);
1550a7e14dcfSSatish Balay }
1551a7e14dcfSSatish Balay 
15527fab98ebSJason Sarich /*@
15538d5ead36SAlp Dener    TaoDefaultGMonitor - Default routine for monitoring progress of the
15548d5ead36SAlp Dener    Tao solvers (default) with extra detail on the globalization method.
15558d5ead36SAlp Dener    This monitor prints the function value and gradient norm at each
15568d5ead36SAlp Dener    iteration, as well as the step size and trust radius. Note that the
15578d5ead36SAlp Dener    step size and trust radius may be the same for some algorithms.
15588d5ead36SAlp Dener    It can be turned on from the command line using the
15598d5ead36SAlp Dener    -tao_gmonitor option
15608d5ead36SAlp Dener 
15618d5ead36SAlp Dener    Collective on Tao
15628d5ead36SAlp Dener 
15638d5ead36SAlp Dener    Input Parameters:
15648d5ead36SAlp Dener +  tao - the Tao context
15658d5ead36SAlp Dener -  ctx - PetscViewer context or NULL
15668d5ead36SAlp Dener 
15678d5ead36SAlp Dener    Options Database Keys:
15688d5ead36SAlp Dener .  -tao_monitor
15698d5ead36SAlp Dener 
15708d5ead36SAlp Dener    Level: advanced
15718d5ead36SAlp Dener 
15728d5ead36SAlp Dener .seealso: TaoDefaultSMonitor(), TaoSetMonitor()
15738d5ead36SAlp Dener @*/
15748d5ead36SAlp Dener PetscErrorCode TaoDefaultGMonitor(Tao tao, void *ctx)
15758d5ead36SAlp Dener {
15768d5ead36SAlp Dener   PetscErrorCode ierr;
15778d5ead36SAlp Dener   PetscInt       its, tabs;
15788d5ead36SAlp Dener   PetscReal      fct,gnorm,stp,tr;
15798d5ead36SAlp Dener   PetscViewer    viewer = (PetscViewer)ctx;
15808d5ead36SAlp Dener 
15818d5ead36SAlp Dener   PetscFunctionBegin;
15828d5ead36SAlp Dener   PetscValidHeaderSpecific(viewer,PETSC_VIEWER_CLASSID,2);
15838d5ead36SAlp Dener   its=tao->niter;
15848d5ead36SAlp Dener   fct=tao->fc;
15858d5ead36SAlp Dener   gnorm=tao->residual;
15868d5ead36SAlp Dener   stp=tao->step;
15878d5ead36SAlp Dener   tr=tao->trust;
15888d5ead36SAlp Dener   ierr = PetscViewerASCIIGetTab(viewer, &tabs);CHKERRQ(ierr);
15898d5ead36SAlp Dener   ierr = PetscViewerASCIISetTab(viewer, ((PetscObject)tao)->tablevel);CHKERRQ(ierr);
15908d5ead36SAlp Dener   if (its == 0 && ((PetscObject)tao)->prefix) {
15918d5ead36SAlp Dener      ierr = PetscViewerASCIIPrintf(viewer,"  Iteration information for %s solve.\n",((PetscObject)tao)->prefix);CHKERRQ(ierr);
15928d5ead36SAlp Dener    }
15938d5ead36SAlp Dener   ierr=PetscViewerASCIIPrintf(viewer,"%3D TAO,",its);CHKERRQ(ierr);
15948d5ead36SAlp Dener   ierr=PetscViewerASCIIPrintf(viewer,"  Function value: %g,",(double)fct);CHKERRQ(ierr);
15958d5ead36SAlp Dener   if (gnorm >= PETSC_INFINITY) {
15968d5ead36SAlp Dener     ierr=PetscViewerASCIIPrintf(viewer,"  Residual: Inf,");CHKERRQ(ierr);
15978d5ead36SAlp Dener   } else {
15988d5ead36SAlp Dener     ierr=PetscViewerASCIIPrintf(viewer,"  Residual: %g,",(double)gnorm);CHKERRQ(ierr);
15998d5ead36SAlp Dener   }
16008d5ead36SAlp Dener   ierr = PetscViewerASCIIPrintf(viewer,"  Step: %g,  Trust: %g\n",(double)stp,(double)tr);CHKERRQ(ierr);
16018d5ead36SAlp Dener   ierr = PetscViewerASCIISetTab(viewer, tabs);CHKERRQ(ierr);
16028d5ead36SAlp Dener   PetscFunctionReturn(0);
16038d5ead36SAlp Dener }
16048d5ead36SAlp Dener 
16058d5ead36SAlp Dener /*@
1606a7e14dcfSSatish Balay    TaoDefaultSMonitor - Default routine for monitoring progress of the
160798ea980cSBarry Smith    solver. Same as TaoMonitorDefault() except
1608a7e14dcfSSatish Balay    it prints fewer digits of the residual as the residual gets smaller.
1609a7e14dcfSSatish Balay    This is because the later digits are meaningless and are often
1610a7e14dcfSSatish Balay    different on different machines; by using this routine different
1611a7e14dcfSSatish Balay    machines will usually generate the same output. It can be turned on
1612a7e14dcfSSatish Balay    by using the -tao_smonitor option
1613a7e14dcfSSatish Balay 
1614441846f8SBarry Smith    Collective on Tao
1615a7e14dcfSSatish Balay 
1616a7e14dcfSSatish Balay    Input Parameters:
1617441846f8SBarry Smith +  tao - the Tao context
16184d4332d5SBarry Smith -  ctx - PetscViewer context of type ASCII
1619a7e14dcfSSatish Balay 
1620a7e14dcfSSatish Balay    Options Database Keys:
1621a7e14dcfSSatish Balay .  -tao_smonitor
1622a7e14dcfSSatish Balay 
1623a7e14dcfSSatish Balay    Level: advanced
1624a7e14dcfSSatish Balay 
162598ea980cSBarry Smith .seealso: TaoMonitorDefault(), TaoSetMonitor()
1626a7e14dcfSSatish Balay @*/
1627441846f8SBarry Smith PetscErrorCode TaoDefaultSMonitor(Tao tao, void *ctx)
1628a7e14dcfSSatish Balay {
1629a7e14dcfSSatish Balay   PetscErrorCode ierr;
16302c9b8b83SAlp Dener   PetscInt       its, tabs;
1631a7e14dcfSSatish Balay   PetscReal      fct,gnorm;
16324d4332d5SBarry Smith   PetscViewer    viewer = (PetscViewer)ctx;
163347a47007SBarry Smith 
1634a7e14dcfSSatish Balay   PetscFunctionBegin;
16354d4332d5SBarry Smith   PetscValidHeaderSpecific(viewer,PETSC_VIEWER_CLASSID,2);
1636a7e14dcfSSatish Balay   its=tao->niter;
1637a7e14dcfSSatish Balay   fct=tao->fc;
1638a7e14dcfSSatish Balay   gnorm=tao->residual;
16392c9b8b83SAlp Dener   ierr = PetscViewerASCIIGetTab(viewer, &tabs);CHKERRQ(ierr);
16402c9b8b83SAlp Dener   ierr = PetscViewerASCIISetTab(viewer, ((PetscObject)tao)->tablevel);CHKERRQ(ierr);
1641a7e14dcfSSatish Balay   ierr=PetscViewerASCIIPrintf(viewer,"iter = %3D,",its);CHKERRQ(ierr);
164247a47007SBarry Smith   ierr=PetscViewerASCIIPrintf(viewer," Function value %g,",(double)fct);CHKERRQ(ierr);
1643d393f493SBarry Smith   if (gnorm >= PETSC_INFINITY) {
164408d19d1fSJason Sarich     ierr=PetscViewerASCIIPrintf(viewer," Residual: Inf \n");CHKERRQ(ierr);
164508d19d1fSJason Sarich   } else if (gnorm > 1.e-6) {
164647a47007SBarry Smith     ierr=PetscViewerASCIIPrintf(viewer," Residual: %g \n",(double)gnorm);CHKERRQ(ierr);
1647a7e14dcfSSatish Balay   } else if (gnorm > 1.e-11) {
1648a7e14dcfSSatish Balay     ierr=PetscViewerASCIIPrintf(viewer," Residual: < 1.0e-6 \n");CHKERRQ(ierr);
1649a7e14dcfSSatish Balay   } else {
1650a7e14dcfSSatish Balay     ierr=PetscViewerASCIIPrintf(viewer," Residual: < 1.0e-11 \n");CHKERRQ(ierr);
1651a7e14dcfSSatish Balay   }
16522c9b8b83SAlp Dener   ierr = PetscViewerASCIISetTab(viewer, tabs);CHKERRQ(ierr);
1653a7e14dcfSSatish Balay   PetscFunctionReturn(0);
1654a7e14dcfSSatish Balay }
1655a7e14dcfSSatish Balay 
16567fab98ebSJason Sarich /*@
165798ea980cSBarry Smith    TaoDefaultCMonitor - same as TaoMonitorDefault() except
1658a7e14dcfSSatish Balay    it prints the norm of the constraints function. It can be turned on
1659a7e14dcfSSatish Balay    from the command line using the -tao_cmonitor option
1660a7e14dcfSSatish Balay 
1661441846f8SBarry Smith    Collective on Tao
1662a7e14dcfSSatish Balay 
1663a7e14dcfSSatish Balay    Input Parameters:
1664441846f8SBarry Smith +  tao - the Tao context
166547a47007SBarry Smith -  ctx - PetscViewer context or NULL
1666a7e14dcfSSatish Balay 
1667a7e14dcfSSatish Balay    Options Database Keys:
1668a7e14dcfSSatish Balay .  -tao_cmonitor
1669a7e14dcfSSatish Balay 
1670a7e14dcfSSatish Balay    Level: advanced
1671a7e14dcfSSatish Balay 
167298ea980cSBarry Smith .seealso: TaoMonitorDefault(), TaoSetMonitor()
1673a7e14dcfSSatish Balay @*/
1674441846f8SBarry Smith PetscErrorCode TaoDefaultCMonitor(Tao tao, void *ctx)
1675a7e14dcfSSatish Balay {
1676a7e14dcfSSatish Balay   PetscErrorCode ierr;
16772c9b8b83SAlp Dener   PetscInt       its, tabs;
1678a7e14dcfSSatish Balay   PetscReal      fct,gnorm;
1679d393f493SBarry Smith   PetscViewer    viewer = (PetscViewer)ctx;
1680a7e14dcfSSatish Balay 
1681a7e14dcfSSatish Balay   PetscFunctionBegin;
1682d393f493SBarry Smith   PetscValidHeaderSpecific(viewer,PETSC_VIEWER_CLASSID,2);
1683a7e14dcfSSatish Balay   its=tao->niter;
1684a7e14dcfSSatish Balay   fct=tao->fc;
1685a7e14dcfSSatish Balay   gnorm=tao->residual;
16862c9b8b83SAlp Dener   ierr = PetscViewerASCIIGetTab(viewer, &tabs);CHKERRQ(ierr);
16872c9b8b83SAlp Dener   ierr = PetscViewerASCIISetTab(viewer, ((PetscObject)tao)->tablevel);CHKERRQ(ierr);
1688a7e14dcfSSatish Balay   ierr=PetscViewerASCIIPrintf(viewer,"iter = %D,",its);CHKERRQ(ierr);
168947a47007SBarry Smith   ierr=PetscViewerASCIIPrintf(viewer," Function value: %g,",(double)fct);CHKERRQ(ierr);
169047a47007SBarry Smith   ierr=PetscViewerASCIIPrintf(viewer,"  Residual: %g ",(double)gnorm);CHKERRQ(ierr);
169147a47007SBarry Smith   ierr = PetscViewerASCIIPrintf(viewer,"  Constraint: %g \n",(double)tao->cnorm);CHKERRQ(ierr);
16922c9b8b83SAlp Dener   ierr = PetscViewerASCIISetTab(viewer, tabs);CHKERRQ(ierr);
1693a7e14dcfSSatish Balay   PetscFunctionReturn(0);
1694a7e14dcfSSatish Balay }
1695a7e14dcfSSatish Balay 
1696a7e14dcfSSatish Balay /*@C
1697a7e14dcfSSatish Balay    TaoSolutionMonitor - Views the solution at each iteration
1698a7e14dcfSSatish Balay    It can be turned on from the command line using the
1699a7e14dcfSSatish Balay    -tao_view_solution option
1700a7e14dcfSSatish Balay 
1701441846f8SBarry Smith    Collective on Tao
1702a7e14dcfSSatish Balay 
1703a7e14dcfSSatish Balay    Input Parameters:
1704441846f8SBarry Smith +  tao - the Tao context
170547a47007SBarry Smith -  ctx - PetscViewer context or NULL
1706a7e14dcfSSatish Balay 
1707a7e14dcfSSatish Balay    Options Database Keys:
1708a7e14dcfSSatish Balay .  -tao_view_solution
1709a7e14dcfSSatish Balay 
1710a7e14dcfSSatish Balay    Level: advanced
1711a7e14dcfSSatish Balay 
1712a7e14dcfSSatish Balay .seealso: TaoDefaultSMonitor(), TaoSetMonitor()
1713a7e14dcfSSatish Balay @*/
1714441846f8SBarry Smith PetscErrorCode TaoSolutionMonitor(Tao tao, void *ctx)
1715a7e14dcfSSatish Balay {
1716a7e14dcfSSatish Balay   PetscErrorCode ierr;
1717d393f493SBarry Smith   PetscViewer    viewer  = (PetscViewer)ctx;;
171847a47007SBarry Smith 
1719a7e14dcfSSatish Balay   PetscFunctionBegin;
1720d393f493SBarry Smith   PetscValidHeaderSpecific(viewer,PETSC_VIEWER_CLASSID,2);
1721a7e14dcfSSatish Balay   ierr = VecView(tao->solution, viewer);CHKERRQ(ierr);
1722a7e14dcfSSatish Balay   PetscFunctionReturn(0);
1723a7e14dcfSSatish Balay }
1724a7e14dcfSSatish Balay 
1725a7e14dcfSSatish Balay /*@C
1726a7e14dcfSSatish Balay    TaoGradientMonitor - Views the gradient at each iteration
1727a7e14dcfSSatish Balay    It can be turned on from the command line using the
1728a7e14dcfSSatish Balay    -tao_view_gradient option
1729a7e14dcfSSatish Balay 
1730441846f8SBarry Smith    Collective on Tao
1731a7e14dcfSSatish Balay 
1732a7e14dcfSSatish Balay    Input Parameters:
1733441846f8SBarry Smith +  tao - the Tao context
173447a47007SBarry Smith -  ctx - PetscViewer context or NULL
1735a7e14dcfSSatish Balay 
1736a7e14dcfSSatish Balay    Options Database Keys:
1737a7e14dcfSSatish Balay .  -tao_view_gradient
1738a7e14dcfSSatish Balay 
1739a7e14dcfSSatish Balay    Level: advanced
1740a7e14dcfSSatish Balay 
1741a7e14dcfSSatish Balay .seealso: TaoDefaultSMonitor(), TaoSetMonitor()
1742a7e14dcfSSatish Balay @*/
1743441846f8SBarry Smith PetscErrorCode TaoGradientMonitor(Tao tao, void *ctx)
1744a7e14dcfSSatish Balay {
1745a7e14dcfSSatish Balay   PetscErrorCode ierr;
1746d393f493SBarry Smith   PetscViewer    viewer = (PetscViewer)ctx;
174747a47007SBarry Smith 
1748a7e14dcfSSatish Balay   PetscFunctionBegin;
1749d393f493SBarry Smith   PetscValidHeaderSpecific(viewer,PETSC_VIEWER_CLASSID,2);
1750a7e14dcfSSatish Balay   ierr = VecView(tao->gradient, viewer);CHKERRQ(ierr);
1751a7e14dcfSSatish Balay   PetscFunctionReturn(0);
1752a7e14dcfSSatish Balay }
1753a7e14dcfSSatish Balay 
1754a7e14dcfSSatish Balay /*@C
1755a7e14dcfSSatish Balay    TaoStepDirectionMonitor - Views the gradient at each iteration
1756a7e14dcfSSatish Balay    It can be turned on from the command line using the
1757a7e14dcfSSatish Balay    -tao_view_gradient option
1758a7e14dcfSSatish Balay 
1759441846f8SBarry Smith    Collective on Tao
1760a7e14dcfSSatish Balay 
1761a7e14dcfSSatish Balay    Input Parameters:
1762441846f8SBarry Smith +  tao - the Tao context
176347a47007SBarry Smith -  ctx - PetscViewer context or NULL
1764a7e14dcfSSatish Balay 
1765a7e14dcfSSatish Balay    Options Database Keys:
1766a7e14dcfSSatish Balay .  -tao_view_gradient
1767a7e14dcfSSatish Balay 
1768a7e14dcfSSatish Balay    Level: advanced
1769a7e14dcfSSatish Balay 
1770a7e14dcfSSatish Balay .seealso: TaoDefaultSMonitor(), TaoSetMonitor()
1771a7e14dcfSSatish Balay @*/
1772441846f8SBarry Smith PetscErrorCode TaoStepDirectionMonitor(Tao tao, void *ctx)
1773a7e14dcfSSatish Balay {
1774a7e14dcfSSatish Balay   PetscErrorCode ierr;
1775d393f493SBarry Smith   PetscViewer    viewer = (PetscViewer)ctx;
1776d393f493SBarry Smith 
1777a7e14dcfSSatish Balay   PetscFunctionBegin;
1778d393f493SBarry Smith   PetscValidHeaderSpecific(viewer,PETSC_VIEWER_CLASSID,2);
1779a7e14dcfSSatish Balay   ierr = VecView(tao->stepdirection, viewer);CHKERRQ(ierr);
1780a7e14dcfSSatish Balay   PetscFunctionReturn(0);
1781a7e14dcfSSatish Balay }
1782a7e14dcfSSatish Balay 
1783a7e14dcfSSatish Balay /*@C
1784a7e14dcfSSatish Balay    TaoDrawSolutionMonitor - Plots the solution at each iteration
1785a7e14dcfSSatish Balay    It can be turned on from the command line using the
1786a7e14dcfSSatish Balay    -tao_draw_solution option
1787a7e14dcfSSatish Balay 
1788441846f8SBarry Smith    Collective on Tao
1789a7e14dcfSSatish Balay 
1790a7e14dcfSSatish Balay    Input Parameters:
1791441846f8SBarry Smith +  tao - the Tao context
1792e882e171SHong Zhang -  ctx - TaoMonitorDraw context
1793a7e14dcfSSatish Balay 
1794a7e14dcfSSatish Balay    Options Database Keys:
1795a7e14dcfSSatish Balay .  -tao_draw_solution
1796a7e14dcfSSatish Balay 
1797a7e14dcfSSatish Balay    Level: advanced
1798a7e14dcfSSatish Balay 
1799a7e14dcfSSatish Balay .seealso: TaoSolutionMonitor(), TaoSetMonitor(), TaoDrawGradientMonitor
1800a7e14dcfSSatish Balay @*/
1801441846f8SBarry Smith PetscErrorCode TaoDrawSolutionMonitor(Tao tao, void *ctx)
1802a7e14dcfSSatish Balay {
1803a7e14dcfSSatish Balay   PetscErrorCode    ierr;
1804e882e171SHong Zhang   TaoMonitorDrawCtx ictx = (TaoMonitorDrawCtx)ctx;
180547a47007SBarry Smith 
1806a7e14dcfSSatish Balay   PetscFunctionBegin;
1807e882e171SHong Zhang   if (!(((ictx->howoften > 0) && (!(tao->niter % ictx->howoften))) || ((ictx->howoften == -1) && tao->reason))) PetscFunctionReturn(0);
1808e882e171SHong Zhang   ierr = VecView(tao->solution,ictx->viewer);CHKERRQ(ierr);
1809a7e14dcfSSatish Balay   PetscFunctionReturn(0);
1810a7e14dcfSSatish Balay }
1811a7e14dcfSSatish Balay 
1812a7e14dcfSSatish Balay /*@C
1813a7e14dcfSSatish Balay    TaoDrawGradientMonitor - Plots the gradient at each iteration
1814a7e14dcfSSatish Balay    It can be turned on from the command line using the
1815a7e14dcfSSatish Balay    -tao_draw_gradient option
1816a7e14dcfSSatish Balay 
1817441846f8SBarry Smith    Collective on Tao
1818a7e14dcfSSatish Balay 
1819a7e14dcfSSatish Balay    Input Parameters:
1820441846f8SBarry Smith +  tao - the Tao context
1821f55353a2SBarry Smith -  ctx - PetscViewer context
1822a7e14dcfSSatish Balay 
1823a7e14dcfSSatish Balay    Options Database Keys:
1824a7e14dcfSSatish Balay .  -tao_draw_gradient
1825a7e14dcfSSatish Balay 
1826a7e14dcfSSatish Balay    Level: advanced
1827a7e14dcfSSatish Balay 
1828a7e14dcfSSatish Balay .seealso: TaoGradientMonitor(), TaoSetMonitor(), TaoDrawSolutionMonitor
1829a7e14dcfSSatish Balay @*/
1830441846f8SBarry Smith PetscErrorCode TaoDrawGradientMonitor(Tao tao, void *ctx)
1831a7e14dcfSSatish Balay {
1832a7e14dcfSSatish Balay   PetscErrorCode    ierr;
1833e882e171SHong Zhang   TaoMonitorDrawCtx ictx = (TaoMonitorDrawCtx)ctx;
183447a47007SBarry Smith 
1835a7e14dcfSSatish Balay   PetscFunctionBegin;
1836e882e171SHong Zhang   if (!(((ictx->howoften > 0) && (!(tao->niter % ictx->howoften))) || ((ictx->howoften == -1) && tao->reason))) PetscFunctionReturn(0);
1837e882e171SHong Zhang   ierr = VecView(tao->gradient,ictx->viewer);CHKERRQ(ierr);
1838a7e14dcfSSatish Balay   PetscFunctionReturn(0);
1839a7e14dcfSSatish Balay }
1840a7e14dcfSSatish Balay 
1841a7e14dcfSSatish Balay /*@C
1842a7e14dcfSSatish Balay    TaoDrawStepMonitor - Plots the step direction at each iteration
1843a7e14dcfSSatish Balay    It can be turned on from the command line using the
1844a7e14dcfSSatish Balay    -tao_draw_step option
1845a7e14dcfSSatish Balay 
1846441846f8SBarry Smith    Collective on Tao
1847a7e14dcfSSatish Balay 
1848a7e14dcfSSatish Balay    Input Parameters:
1849441846f8SBarry Smith +  tao - the Tao context
1850f55353a2SBarry Smith -  ctx - PetscViewer context
1851a7e14dcfSSatish Balay 
1852a7e14dcfSSatish Balay    Options Database Keys:
1853a7e14dcfSSatish Balay .  -tao_draw_step
1854a7e14dcfSSatish Balay 
1855a7e14dcfSSatish Balay    Level: advanced
1856a7e14dcfSSatish Balay 
1857a7e14dcfSSatish Balay .seealso: TaoSetMonitor(), TaoDrawSolutionMonitor
1858a7e14dcfSSatish Balay @*/
1859441846f8SBarry Smith PetscErrorCode TaoDrawStepMonitor(Tao tao, void *ctx)
1860a7e14dcfSSatish Balay {
1861a7e14dcfSSatish Balay   PetscErrorCode ierr;
1862a7e14dcfSSatish Balay   PetscViewer    viewer = (PetscViewer)(ctx);
186347a47007SBarry Smith 
1864a7e14dcfSSatish Balay   PetscFunctionBegin;
1865a7e14dcfSSatish Balay   ierr = VecView(tao->stepdirection, viewer);CHKERRQ(ierr);
1866a7e14dcfSSatish Balay   PetscFunctionReturn(0);
1867a7e14dcfSSatish Balay }
1868a7e14dcfSSatish Balay 
1869a7e14dcfSSatish Balay /*@C
1870737f463aSAlp Dener    TaoResidualMonitor - Views the least-squares residual at each iteration
1871a7e14dcfSSatish Balay    It can be turned on from the command line using the
18724a48860cSAlp Dener    -tao_view_ls_residual option
1873a7e14dcfSSatish Balay 
1874441846f8SBarry Smith    Collective on Tao
1875a7e14dcfSSatish Balay 
1876a7e14dcfSSatish Balay    Input Parameters:
1877441846f8SBarry Smith +  tao - the Tao context
187847a47007SBarry Smith -  ctx - PetscViewer context or NULL
1879a7e14dcfSSatish Balay 
1880a7e14dcfSSatish Balay    Options Database Keys:
18814a48860cSAlp Dener .  -tao_view_ls_residual
1882a7e14dcfSSatish Balay 
1883a7e14dcfSSatish Balay    Level: advanced
1884a7e14dcfSSatish Balay 
1885a7e14dcfSSatish Balay .seealso: TaoDefaultSMonitor(), TaoSetMonitor()
1886a7e14dcfSSatish Balay @*/
1887737f463aSAlp Dener PetscErrorCode TaoResidualMonitor(Tao tao, void *ctx)
1888a7e14dcfSSatish Balay {
1889a7e14dcfSSatish Balay   PetscErrorCode ierr;
1890d393f493SBarry Smith   PetscViewer    viewer  = (PetscViewer)ctx;
189147a47007SBarry Smith 
1892a7e14dcfSSatish Balay   PetscFunctionBegin;
1893d393f493SBarry Smith   PetscValidHeaderSpecific(viewer,PETSC_VIEWER_CLASSID,2);
18944a48860cSAlp Dener   ierr = VecView(tao->ls_res,viewer);CHKERRQ(ierr);
1895a7e14dcfSSatish Balay   PetscFunctionReturn(0);
1896a7e14dcfSSatish Balay }
1897a7e14dcfSSatish Balay 
18987fab98ebSJason Sarich /*@
1899a7e14dcfSSatish Balay    TaoDefaultConvergenceTest - Determines whether the solver should continue iterating
1900a7e14dcfSSatish Balay    or terminate.
1901a7e14dcfSSatish Balay 
1902441846f8SBarry Smith    Collective on Tao
1903a7e14dcfSSatish Balay 
1904a7e14dcfSSatish Balay    Input Parameters:
1905441846f8SBarry Smith +  tao - the Tao context
1906a7e14dcfSSatish Balay -  dummy - unused dummy context
1907a7e14dcfSSatish Balay 
1908a7e14dcfSSatish Balay    Output Parameter:
1909a7e14dcfSSatish Balay .  reason - for terminating
1910a7e14dcfSSatish Balay 
1911a7e14dcfSSatish Balay    Notes:
1912a7e14dcfSSatish Balay    This routine checks the residual in the optimality conditions, the
1913a7e14dcfSSatish Balay    relative residual in the optimity conditions, the number of function
1914a7e14dcfSSatish Balay    evaluations, and the function value to test convergence.  Some
1915a7e14dcfSSatish Balay    solvers may use different convergence routines.
1916a7e14dcfSSatish Balay 
1917a7e14dcfSSatish Balay    Level: developer
1918a7e14dcfSSatish Balay 
1919e4cb33bbSBarry Smith .seealso: TaoSetTolerances(),TaoGetConvergedReason(),TaoSetConvergedReason()
1920a7e14dcfSSatish Balay @*/
1921a7e14dcfSSatish Balay 
1922441846f8SBarry Smith PetscErrorCode TaoDefaultConvergenceTest(Tao tao,void *dummy)
1923a7e14dcfSSatish Balay {
1924a7e14dcfSSatish Balay   PetscInt           niter=tao->niter, nfuncs=PetscMax(tao->nfuncs,tao->nfuncgrads);
1925a7e14dcfSSatish Balay   PetscInt           max_funcs=tao->max_funcs;
1926a7e14dcfSSatish Balay   PetscReal          gnorm=tao->residual, gnorm0=tao->gnorm0;
1927a7e14dcfSSatish Balay   PetscReal          f=tao->fc, steptol=tao->steptol,trradius=tao->step;
1928a7e14dcfSSatish Balay   PetscReal          gatol=tao->gatol,grtol=tao->grtol,gttol=tao->gttol;
1929e52336cbSBarry Smith   PetscReal          catol=tao->catol,crtol=tao->crtol;
1930e52336cbSBarry Smith   PetscReal          fmin=tao->fmin, cnorm=tao->cnorm;
1931e4cb33bbSBarry Smith   TaoConvergedReason reason=tao->reason;
1932a7e14dcfSSatish Balay   PetscErrorCode     ierr;
1933a7e14dcfSSatish Balay 
1934a7e14dcfSSatish Balay   PetscFunctionBegin;
1935441846f8SBarry Smith   PetscValidHeaderSpecific(tao, TAO_CLASSID,1);
1936a7e14dcfSSatish Balay   if (reason != TAO_CONTINUE_ITERATING) {
1937a7e14dcfSSatish Balay     PetscFunctionReturn(0);
1938a7e14dcfSSatish Balay   }
1939a7e14dcfSSatish Balay 
1940a7e14dcfSSatish Balay   if (PetscIsInfOrNanReal(f)) {
1941a7e14dcfSSatish Balay     ierr = PetscInfo(tao,"Failed to converged, function value is Inf or NaN\n");CHKERRQ(ierr);
1942a7e14dcfSSatish Balay     reason = TAO_DIVERGED_NAN;
1943a7e14dcfSSatish Balay   } else if (f <= fmin && cnorm <=catol) {
194447a47007SBarry Smith     ierr = PetscInfo2(tao,"Converged due to function value %g < minimum function value %g\n", (double)f,(double)fmin);CHKERRQ(ierr);
1945a7e14dcfSSatish Balay     reason = TAO_CONVERGED_MINF;
1946a7e14dcfSSatish Balay   } else if (gnorm<= gatol && cnorm <=catol) {
194747a47007SBarry Smith     ierr = PetscInfo2(tao,"Converged due to residual norm ||g(X)||=%g < %g\n",(double)gnorm,(double)gatol);CHKERRQ(ierr);
1948a7e14dcfSSatish Balay     reason = TAO_CONVERGED_GATOL;
1949a7e14dcfSSatish Balay   } else if ( f!=0 && PetscAbsReal(gnorm/f) <= grtol && cnorm <= crtol) {
195047a47007SBarry Smith     ierr = PetscInfo2(tao,"Converged due to residual ||g(X)||/|f(X)| =%g < %g\n",(double)(gnorm/f),(double)grtol);CHKERRQ(ierr);
1951a7e14dcfSSatish Balay     reason = TAO_CONVERGED_GRTOL;
1952e1d16bb9SBarry Smith   } else if (gnorm0 != 0 && ((gttol == 0 && gnorm == 0) || gnorm/gnorm0 < gttol) && cnorm <= crtol) {
195347a47007SBarry Smith     ierr = PetscInfo2(tao,"Converged due to relative residual norm ||g(X)||/||g(X0)|| = %g < %g\n",(double)(gnorm/gnorm0),(double)gttol);CHKERRQ(ierr);
1954a7e14dcfSSatish Balay     reason = TAO_CONVERGED_GTTOL;
1955a7e14dcfSSatish Balay   } else if (nfuncs > max_funcs){
1956a7e14dcfSSatish Balay     ierr = PetscInfo2(tao,"Exceeded maximum number of function evaluations: %D > %D\n", nfuncs,max_funcs);CHKERRQ(ierr);
1957a7e14dcfSSatish Balay     reason = TAO_DIVERGED_MAXFCN;
1958a7e14dcfSSatish Balay   } else if ( tao->lsflag != 0 ){
1959a7e14dcfSSatish Balay     ierr = PetscInfo(tao,"Tao Line Search failure.\n");CHKERRQ(ierr);
1960a7e14dcfSSatish Balay     reason = TAO_DIVERGED_LS_FAILURE;
1961a7e14dcfSSatish Balay   } else if (trradius < steptol && niter > 0){
196247a47007SBarry Smith     ierr = PetscInfo2(tao,"Trust region/step size too small: %g < %g\n", (double)trradius,(double)steptol);CHKERRQ(ierr);
1963a7e14dcfSSatish Balay     reason = TAO_CONVERGED_STEPTOL;
1964e031d6f5SAlp Dener   } else if (niter >= tao->max_it) {
1965a7e14dcfSSatish Balay     ierr = PetscInfo2(tao,"Exceeded maximum number of iterations: %D > %D\n",niter,tao->max_it);CHKERRQ(ierr);
1966a7e14dcfSSatish Balay     reason = TAO_DIVERGED_MAXITS;
1967a7e14dcfSSatish Balay   } else {
1968a7e14dcfSSatish Balay     reason = TAO_CONTINUE_ITERATING;
1969a7e14dcfSSatish Balay   }
1970a7e14dcfSSatish Balay   tao->reason = reason;
1971a7e14dcfSSatish Balay   PetscFunctionReturn(0);
1972a7e14dcfSSatish Balay }
1973a7e14dcfSSatish Balay 
1974a7e14dcfSSatish Balay /*@C
1975a7e14dcfSSatish Balay    TaoSetOptionsPrefix - Sets the prefix used for searching for all
1976a7e14dcfSSatish Balay    TAO options in the database.
1977a7e14dcfSSatish Balay 
1978a7e14dcfSSatish Balay 
1979441846f8SBarry Smith    Logically Collective on Tao
1980a7e14dcfSSatish Balay 
1981a7e14dcfSSatish Balay    Input Parameters:
1982441846f8SBarry Smith +  tao - the Tao context
1983a7e14dcfSSatish Balay -  prefix - the prefix string to prepend to all TAO option requests
1984a7e14dcfSSatish Balay 
1985a7e14dcfSSatish Balay    Notes:
1986a7e14dcfSSatish Balay    A hyphen (-) must NOT be given at the beginning of the prefix name.
1987a7e14dcfSSatish Balay    The first character of all runtime options is AUTOMATICALLY the hyphen.
1988a7e14dcfSSatish Balay 
1989a7e14dcfSSatish Balay    For example, to distinguish between the runtime options for two
1990a7e14dcfSSatish Balay    different TAO solvers, one could call
1991a7e14dcfSSatish Balay .vb
1992a7e14dcfSSatish Balay       TaoSetOptionsPrefix(tao1,"sys1_")
1993a7e14dcfSSatish Balay       TaoSetOptionsPrefix(tao2,"sys2_")
1994a7e14dcfSSatish Balay .ve
1995a7e14dcfSSatish Balay 
1996a7e14dcfSSatish Balay    This would enable use of different options for each system, such as
1997a7e14dcfSSatish Balay .vb
1998a7e14dcfSSatish Balay       -sys1_tao_method blmvm -sys1_tao_gtol 1.e-3
1999a7e14dcfSSatish Balay       -sys2_tao_method lmvm  -sys2_tao_gtol 1.e-4
2000a7e14dcfSSatish Balay .ve
2001a7e14dcfSSatish Balay 
2002a7e14dcfSSatish Balay 
2003a7e14dcfSSatish Balay    Level: advanced
2004a7e14dcfSSatish Balay 
2005a7e14dcfSSatish Balay .seealso: TaoAppendOptionsPrefix(), TaoGetOptionsPrefix()
2006a7e14dcfSSatish Balay @*/
2007a7e14dcfSSatish Balay 
2008441846f8SBarry Smith PetscErrorCode TaoSetOptionsPrefix(Tao tao, const char p[])
2009a7e14dcfSSatish Balay {
2010a7e14dcfSSatish Balay   PetscErrorCode ierr;
201147a47007SBarry Smith 
2012a7e14dcfSSatish Balay   PetscFunctionBegin;
2013a7e14dcfSSatish Balay   ierr = PetscObjectSetOptionsPrefix((PetscObject)tao,p);CHKERRQ(ierr);
2014a7e14dcfSSatish Balay   if (tao->linesearch) {
2015a7e14dcfSSatish Balay     ierr = TaoLineSearchSetOptionsPrefix(tao->linesearch,p);CHKERRQ(ierr);
2016a7e14dcfSSatish Balay   }
2017a7e14dcfSSatish Balay   if (tao->ksp) {
2018a7e14dcfSSatish Balay     ierr = KSPSetOptionsPrefix(tao->ksp,p);CHKERRQ(ierr);
2019a7e14dcfSSatish Balay   }
2020a7e14dcfSSatish Balay   PetscFunctionReturn(0);
2021a7e14dcfSSatish Balay }
2022a7e14dcfSSatish Balay 
2023a7e14dcfSSatish Balay /*@C
2024a7e14dcfSSatish Balay    TaoAppendOptionsPrefix - Appends to the prefix used for searching for all
2025a7e14dcfSSatish Balay    TAO options in the database.
2026a7e14dcfSSatish Balay 
2027a7e14dcfSSatish Balay 
2028441846f8SBarry Smith    Logically Collective on Tao
2029a7e14dcfSSatish Balay 
2030a7e14dcfSSatish Balay    Input Parameters:
2031441846f8SBarry Smith +  tao - the Tao solver context
2032a7e14dcfSSatish Balay -  prefix - the prefix string to prepend to all TAO option requests
2033a7e14dcfSSatish Balay 
2034a7e14dcfSSatish Balay    Notes:
2035a7e14dcfSSatish Balay    A hyphen (-) must NOT be given at the beginning of the prefix name.
2036a7e14dcfSSatish Balay    The first character of all runtime options is AUTOMATICALLY the hyphen.
2037a7e14dcfSSatish Balay 
2038a7e14dcfSSatish Balay 
2039a7e14dcfSSatish Balay    Level: advanced
2040a7e14dcfSSatish Balay 
2041a7e14dcfSSatish Balay .seealso: TaoSetOptionsPrefix(), TaoGetOptionsPrefix()
2042a7e14dcfSSatish Balay @*/
2043441846f8SBarry Smith PetscErrorCode TaoAppendOptionsPrefix(Tao tao, const char p[])
2044a7e14dcfSSatish Balay {
2045a7e14dcfSSatish Balay   PetscErrorCode ierr;
204647a47007SBarry Smith 
2047a7e14dcfSSatish Balay   PetscFunctionBegin;
2048a7e14dcfSSatish Balay   ierr = PetscObjectAppendOptionsPrefix((PetscObject)tao,p);CHKERRQ(ierr);
2049a7e14dcfSSatish Balay   if (tao->linesearch) {
2050a7e14dcfSSatish Balay     ierr = TaoLineSearchSetOptionsPrefix(tao->linesearch,p);CHKERRQ(ierr);
2051a7e14dcfSSatish Balay   }
2052a7e14dcfSSatish Balay   if (tao->ksp) {
2053a7e14dcfSSatish Balay     ierr = KSPSetOptionsPrefix(tao->ksp,p);CHKERRQ(ierr);
2054a7e14dcfSSatish Balay   }
2055a7e14dcfSSatish Balay   PetscFunctionReturn(0);
2056a7e14dcfSSatish Balay }
2057a7e14dcfSSatish Balay 
2058a7e14dcfSSatish Balay /*@C
2059a7e14dcfSSatish Balay   TaoGetOptionsPrefix - Gets the prefix used for searching for all
2060a7e14dcfSSatish Balay   TAO options in the database
2061a7e14dcfSSatish Balay 
2062a7e14dcfSSatish Balay   Not Collective
2063a7e14dcfSSatish Balay 
2064a7e14dcfSSatish Balay   Input Parameters:
2065441846f8SBarry Smith . tao - the Tao context
2066a7e14dcfSSatish Balay 
2067a7e14dcfSSatish Balay   Output Parameters:
2068a7e14dcfSSatish Balay . prefix - pointer to the prefix string used is returned
2069a7e14dcfSSatish Balay 
207095452b02SPatrick Sanan   Notes:
207195452b02SPatrick Sanan     On the fortran side, the user should pass in a string 'prefix' of
2072a7e14dcfSSatish Balay   sufficient length to hold the prefix.
2073a7e14dcfSSatish Balay 
2074a7e14dcfSSatish Balay   Level: advanced
2075a7e14dcfSSatish Balay 
2076a7e14dcfSSatish Balay .seealso: TaoSetOptionsPrefix(), TaoAppendOptionsPrefix()
2077a7e14dcfSSatish Balay @*/
2078441846f8SBarry Smith PetscErrorCode TaoGetOptionsPrefix(Tao tao, const char *p[])
2079a7e14dcfSSatish Balay {
208047a47007SBarry Smith    return PetscObjectGetOptionsPrefix((PetscObject)tao,p);
2081a7e14dcfSSatish Balay }
2082a7e14dcfSSatish Balay 
2083a7e14dcfSSatish Balay /*@C
2084a7e14dcfSSatish Balay    TaoSetType - Sets the method for the unconstrained minimization solver.
2085a7e14dcfSSatish Balay 
2086441846f8SBarry Smith    Collective on Tao
2087a7e14dcfSSatish Balay 
2088a7e14dcfSSatish Balay    Input Parameters:
2089441846f8SBarry Smith +  solver - the Tao solver context
2090a7e14dcfSSatish Balay -  type - a known method
2091a7e14dcfSSatish Balay 
2092a7e14dcfSSatish Balay    Options Database Key:
209358417fe7SBarry Smith .  -tao_type <type> - Sets the method; use -help for a list
209458417fe7SBarry Smith    of available methods (for instance, "-tao_type lmvm" or "-tao_type tron")
2095a7e14dcfSSatish Balay 
2096a7e14dcfSSatish Balay    Available methods include:
209758417fe7SBarry Smith +    nls - Newton's method with line search for unconstrained minimization
209858417fe7SBarry Smith .    ntr - Newton's method with trust region for unconstrained minimization
209958417fe7SBarry Smith .    ntl - Newton's method with trust region, line search for unconstrained minimization
210058417fe7SBarry Smith .    lmvm - Limited memory variable metric method for unconstrained minimization
210158417fe7SBarry Smith .    cg - Nonlinear conjugate gradient method for unconstrained minimization
210258417fe7SBarry Smith .    nm - Nelder-Mead algorithm for derivate-free unconstrained minimization
210358417fe7SBarry Smith .    tron - Newton Trust Region method for bound constrained minimization
210458417fe7SBarry Smith .    gpcg - Newton Trust Region method for quadratic bound constrained minimization
210558417fe7SBarry Smith .    blmvm - Limited memory variable metric method for bound constrained minimization
21061eb8069cSJason Sarich -    pounders - Model-based algorithm pounder extended for nonlinear least squares
2107a7e14dcfSSatish Balay 
2108a7e14dcfSSatish Balay   Level: intermediate
2109a7e14dcfSSatish Balay 
21101eb8069cSJason Sarich .seealso: TaoCreate(), TaoGetType(), TaoType
2111a7e14dcfSSatish Balay 
2112a7e14dcfSSatish Balay @*/
2113b625d6c7SJed Brown PetscErrorCode TaoSetType(Tao tao, TaoType type)
2114a7e14dcfSSatish Balay {
2115a7e14dcfSSatish Balay   PetscErrorCode ierr;
2116441846f8SBarry Smith   PetscErrorCode (*create_xxx)(Tao);
2117a7e14dcfSSatish Balay   PetscBool      issame;
2118a7e14dcfSSatish Balay 
2119a7e14dcfSSatish Balay   PetscFunctionBegin;
2120441846f8SBarry Smith   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
2121a7e14dcfSSatish Balay 
2122a7e14dcfSSatish Balay   ierr = PetscObjectTypeCompare((PetscObject)tao,type,&issame);CHKERRQ(ierr);
2123a7e14dcfSSatish Balay   if (issame) PetscFunctionReturn(0);
2124a7e14dcfSSatish Balay 
2125441846f8SBarry Smith   ierr = PetscFunctionListFind(TaoList, type, (void(**)(void))&create_xxx);CHKERRQ(ierr);
2126441846f8SBarry Smith   if (!create_xxx) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_UNKNOWN_TYPE,"Unable to find requested Tao type %s",type);
2127a7e14dcfSSatish Balay 
2128a7e14dcfSSatish Balay   /* Destroy the existing solver information */
2129a7e14dcfSSatish Balay   if (tao->ops->destroy) {
2130a7e14dcfSSatish Balay     ierr = (*tao->ops->destroy)(tao);CHKERRQ(ierr);
2131a7e14dcfSSatish Balay   }
2132a7e14dcfSSatish Balay   ierr = KSPDestroy(&tao->ksp);CHKERRQ(ierr);
2133a7e14dcfSSatish Balay   ierr = TaoLineSearchDestroy(&tao->linesearch);CHKERRQ(ierr);
2134a7e14dcfSSatish Balay   ierr = VecDestroy(&tao->gradient);CHKERRQ(ierr);
2135a7e14dcfSSatish Balay   ierr = VecDestroy(&tao->stepdirection);CHKERRQ(ierr);
2136a7e14dcfSSatish Balay 
2137a7e14dcfSSatish Balay   tao->ops->setup = 0;
2138a7e14dcfSSatish Balay   tao->ops->solve = 0;
2139a7e14dcfSSatish Balay   tao->ops->view  = 0;
2140a7e14dcfSSatish Balay   tao->ops->setfromoptions = 0;
2141a7e14dcfSSatish Balay   tao->ops->destroy = 0;
2142a7e14dcfSSatish Balay 
2143a7e14dcfSSatish Balay   tao->setupcalled = PETSC_FALSE;
2144a7e14dcfSSatish Balay 
2145a7e14dcfSSatish Balay   ierr = (*create_xxx)(tao);CHKERRQ(ierr);
2146a7e14dcfSSatish Balay   ierr = PetscObjectChangeTypeName((PetscObject)tao,type);CHKERRQ(ierr);
2147a7e14dcfSSatish Balay   PetscFunctionReturn(0);
2148a7e14dcfSSatish Balay }
214947a47007SBarry Smith 
2150a7e14dcfSSatish Balay /*MC
2151441846f8SBarry Smith    TaoRegister - Adds a method to the TAO package for unconstrained minimization.
2152a7e14dcfSSatish Balay 
2153a7e14dcfSSatish Balay    Synopsis:
2154441846f8SBarry Smith    TaoRegister(char *name_solver,char *path,char *name_Create,int (*routine_Create)(Tao))
2155a7e14dcfSSatish Balay 
2156a7e14dcfSSatish Balay    Not collective
2157a7e14dcfSSatish Balay 
2158a7e14dcfSSatish Balay    Input Parameters:
2159a7e14dcfSSatish Balay +  sname - name of a new user-defined solver
2160a7e14dcfSSatish Balay -  func - routine to Create method context
2161a7e14dcfSSatish Balay 
2162a7e14dcfSSatish Balay    Notes:
2163441846f8SBarry Smith    TaoRegister() may be called multiple times to add several user-defined solvers.
2164a7e14dcfSSatish Balay 
2165a7e14dcfSSatish Balay    Sample usage:
2166a7e14dcfSSatish Balay .vb
2167441846f8SBarry Smith    TaoRegister("my_solver",MySolverCreate);
2168a7e14dcfSSatish Balay .ve
2169a7e14dcfSSatish Balay 
2170a7e14dcfSSatish Balay    Then, your solver can be chosen with the procedural interface via
2171a7e14dcfSSatish Balay $     TaoSetType(tao,"my_solver")
2172a7e14dcfSSatish Balay    or at runtime via the option
217358417fe7SBarry Smith $     -tao_type my_solver
2174a7e14dcfSSatish Balay 
2175a7e14dcfSSatish Balay    Level: advanced
2176a7e14dcfSSatish Balay 
2177441846f8SBarry Smith .seealso: TaoRegisterAll(), TaoRegisterDestroy()
2178a7e14dcfSSatish Balay M*/
2179441846f8SBarry Smith PetscErrorCode TaoRegister(const char sname[], PetscErrorCode (*func)(Tao))
2180a7e14dcfSSatish Balay {
2181a7e14dcfSSatish Balay   PetscErrorCode ierr;
2182a7e14dcfSSatish Balay 
2183a7e14dcfSSatish Balay   PetscFunctionBegin;
21841d36bdfdSBarry Smith   ierr = TaoInitializePackage();CHKERRQ(ierr);
2185441846f8SBarry Smith   ierr = PetscFunctionListAdd(&TaoList,sname, (void (*)(void))func);CHKERRQ(ierr);
2186a7e14dcfSSatish Balay   PetscFunctionReturn(0);
2187a7e14dcfSSatish Balay }
2188a7e14dcfSSatish Balay 
2189a7e14dcfSSatish Balay /*@C
2190441846f8SBarry Smith    TaoRegisterDestroy - Frees the list of minimization solvers that were
2191441846f8SBarry Smith    registered by TaoRegisterDynamic().
2192a7e14dcfSSatish Balay 
2193a7e14dcfSSatish Balay    Not Collective
2194a7e14dcfSSatish Balay 
2195a7e14dcfSSatish Balay    Level: advanced
2196a7e14dcfSSatish Balay 
2197441846f8SBarry Smith .seealso: TaoRegisterAll(), TaoRegister()
2198a7e14dcfSSatish Balay @*/
2199441846f8SBarry Smith PetscErrorCode TaoRegisterDestroy(void)
2200a7e14dcfSSatish Balay {
2201a7e14dcfSSatish Balay   PetscErrorCode ierr;
2202a7e14dcfSSatish Balay   PetscFunctionBegin;
2203441846f8SBarry Smith   ierr = PetscFunctionListDestroy(&TaoList);CHKERRQ(ierr);
2204441846f8SBarry Smith   TaoRegisterAllCalled = PETSC_FALSE;
2205a7e14dcfSSatish Balay   PetscFunctionReturn(0);
2206a7e14dcfSSatish Balay }
220747a47007SBarry Smith 
22088931d482SJason Sarich /*@
22098931d482SJason Sarich    TaoGetIterationNumber - Gets the number of Tao iterations completed
22108931d482SJason Sarich    at this time.
22118931d482SJason Sarich 
22128931d482SJason Sarich    Not Collective
22138931d482SJason Sarich 
22148931d482SJason Sarich    Input Parameter:
22158931d482SJason Sarich .  tao - Tao context
22168931d482SJason Sarich 
22178931d482SJason Sarich    Output Parameter:
22188931d482SJason Sarich .  iter - iteration number
22198931d482SJason Sarich 
22208931d482SJason Sarich    Notes:
22218931d482SJason Sarich    For example, during the computation of iteration 2 this would return 1.
22228931d482SJason Sarich 
22238931d482SJason Sarich 
22248931d482SJason Sarich    Level: intermediate
22258931d482SJason Sarich 
22268931d482SJason Sarich .keywords: Tao, nonlinear, get, iteration, number,
22278931d482SJason Sarich 
222879f5d8caSBarry Smith .seealso:   TaoGetLinearSolveIterations(), TaoGetResidualNorm(), TaoGetObjective()
22298931d482SJason Sarich @*/
22308931d482SJason Sarich PetscErrorCode  TaoGetIterationNumber(Tao tao,PetscInt *iter)
22318931d482SJason Sarich {
22328931d482SJason Sarich   PetscFunctionBegin;
22338931d482SJason Sarich   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
22348931d482SJason Sarich   PetscValidIntPointer(iter,2);
22358931d482SJason Sarich   *iter = tao->niter;
22368931d482SJason Sarich   PetscFunctionReturn(0);
22378931d482SJason Sarich }
22388931d482SJason Sarich 
22398931d482SJason Sarich /*@
224079f5d8caSBarry Smith    TaoGetObjective - Gets the current value of the objective function
224179f5d8caSBarry Smith    at this time.
224279f5d8caSBarry Smith 
224379f5d8caSBarry Smith    Not Collective
224479f5d8caSBarry Smith 
224579f5d8caSBarry Smith    Input Parameter:
224679f5d8caSBarry Smith .  tao - Tao context
224779f5d8caSBarry Smith 
224879f5d8caSBarry Smith    Output Parameter:
224979f5d8caSBarry Smith .  value - the current value
225079f5d8caSBarry Smith 
225179f5d8caSBarry Smith    Level: intermediate
225279f5d8caSBarry Smith 
225379f5d8caSBarry Smith .keywords: Tao, nonlinear, get, iteration, number,
225479f5d8caSBarry Smith 
225579f5d8caSBarry Smith .seealso:   TaoGetLinearSolveIterations(), TaoGetIterationNumber(), TaoGetResidualNorm()
225679f5d8caSBarry Smith @*/
225779f5d8caSBarry Smith PetscErrorCode  TaoGetObjective(Tao tao,PetscReal *value)
225879f5d8caSBarry Smith {
225979f5d8caSBarry Smith   PetscFunctionBegin;
226079f5d8caSBarry Smith   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
226179f5d8caSBarry Smith   PetscValidRealPointer(value,2);
226279f5d8caSBarry Smith   *value = tao->fc;
226379f5d8caSBarry Smith   PetscFunctionReturn(0);
226479f5d8caSBarry Smith }
226579f5d8caSBarry Smith 
226679f5d8caSBarry Smith /*@
226779f5d8caSBarry Smith    TaoGetResidualNorm - Gets the current value of the norm of the residual
226879f5d8caSBarry Smith    at this time.
226979f5d8caSBarry Smith 
227079f5d8caSBarry Smith    Not Collective
227179f5d8caSBarry Smith 
227279f5d8caSBarry Smith    Input Parameter:
227379f5d8caSBarry Smith .  tao - Tao context
227479f5d8caSBarry Smith 
227579f5d8caSBarry Smith    Output Parameter:
227679f5d8caSBarry Smith .  value - the current value
227779f5d8caSBarry Smith 
227879f5d8caSBarry Smith    Level: intermediate
227979f5d8caSBarry Smith 
228079f5d8caSBarry Smith    Developer Note: This is the 2-norm of the residual, we cannot use TaoGetGradientNorm() because that has
228179f5d8caSBarry Smith                    a different meaning. For some reason Tao sometimes calls the gradient the residual.
228279f5d8caSBarry Smith 
228379f5d8caSBarry Smith .keywords: Tao, nonlinear, get, iteration, number,
228479f5d8caSBarry Smith 
228579f5d8caSBarry Smith .seealso:   TaoGetLinearSolveIterations(), TaoGetIterationNumber(), TaoGetObjective()
228679f5d8caSBarry Smith @*/
228779f5d8caSBarry Smith PetscErrorCode  TaoGetResidualNorm(Tao tao,PetscReal *value)
228879f5d8caSBarry Smith {
228979f5d8caSBarry Smith   PetscFunctionBegin;
229079f5d8caSBarry Smith   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
229179f5d8caSBarry Smith   PetscValidRealPointer(value,2);
229279f5d8caSBarry Smith   *value = tao->residual;
229379f5d8caSBarry Smith   PetscFunctionReturn(0);
229479f5d8caSBarry Smith }
229579f5d8caSBarry Smith 
229679f5d8caSBarry Smith /*@
22978931d482SJason Sarich    TaoSetIterationNumber - Sets the current iteration number.
22988931d482SJason Sarich 
22998931d482SJason Sarich    Not Collective
23008931d482SJason Sarich 
23018931d482SJason Sarich    Input Parameter:
23028931d482SJason Sarich .  tao - Tao context
23038931d482SJason Sarich .  iter - iteration number
23048931d482SJason Sarich 
23058931d482SJason Sarich    Level: developer
23068931d482SJason Sarich 
23078931d482SJason Sarich .keywords: Tao, nonlinear, set, iteration, number,
23088931d482SJason Sarich 
23098931d482SJason Sarich .seealso:   TaoGetLinearSolveIterations()
23108931d482SJason Sarich @*/
23118931d482SJason Sarich PetscErrorCode  TaoSetIterationNumber(Tao tao,PetscInt iter)
23128931d482SJason Sarich {
23138931d482SJason Sarich   PetscErrorCode ierr;
23148931d482SJason Sarich 
23158931d482SJason Sarich   PetscFunctionBegin;
23168931d482SJason Sarich   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
23178931d482SJason Sarich   ierr       = PetscObjectSAWsTakeAccess((PetscObject)tao);CHKERRQ(ierr);
23188931d482SJason Sarich   tao->niter = iter;
23198931d482SJason Sarich   ierr       = PetscObjectSAWsGrantAccess((PetscObject)tao);CHKERRQ(ierr);
23208931d482SJason Sarich   PetscFunctionReturn(0);
23218931d482SJason Sarich }
23228931d482SJason Sarich 
23238931d482SJason Sarich /*@
23248931d482SJason Sarich    TaoGetTotalIterationNumber - Gets the total number of Tao iterations
23258931d482SJason Sarich    completed. This number keeps accumulating if multiple solves
23268931d482SJason Sarich    are called with the Tao object.
23278931d482SJason Sarich 
23288931d482SJason Sarich    Not Collective
23298931d482SJason Sarich 
23308931d482SJason Sarich    Input Parameter:
23318931d482SJason Sarich .  tao - Tao context
23328931d482SJason Sarich 
23338931d482SJason Sarich    Output Parameter:
23348931d482SJason Sarich .  iter - iteration number
23358931d482SJason Sarich 
23368931d482SJason Sarich    Notes:
23378931d482SJason Sarich    The total iteration count is updated after each solve, if there is a current
23388931d482SJason Sarich    TaoSolve() in progress then those iterations are not yet counted.
23398931d482SJason Sarich 
23408931d482SJason Sarich    Level: intermediate
23418931d482SJason Sarich 
23428931d482SJason Sarich .keywords: Tao, nonlinear, get, iteration, number,
23438931d482SJason Sarich 
23448931d482SJason Sarich .seealso:   TaoGetLinearSolveIterations()
23458931d482SJason Sarich @*/
23468931d482SJason Sarich PetscErrorCode  TaoGetTotalIterationNumber(Tao tao,PetscInt *iter)
23478931d482SJason Sarich {
23488931d482SJason Sarich   PetscFunctionBegin;
23498931d482SJason Sarich   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
23508931d482SJason Sarich   PetscValidIntPointer(iter,2);
23518931d482SJason Sarich   *iter = tao->ntotalits;
23528931d482SJason Sarich   PetscFunctionReturn(0);
23538931d482SJason Sarich }
23548931d482SJason Sarich 
23558931d482SJason Sarich /*@
23568931d482SJason Sarich    TaoSetTotalIterationNumber - Sets the current total iteration number.
23578931d482SJason Sarich 
23588931d482SJason Sarich    Not Collective
23598931d482SJason Sarich 
23608931d482SJason Sarich    Input Parameter:
23618931d482SJason Sarich .  tao - Tao context
23628931d482SJason Sarich .  iter - iteration number
23638931d482SJason Sarich 
23648931d482SJason Sarich    Level: developer
23658931d482SJason Sarich 
23668931d482SJason Sarich .keywords: Tao, nonlinear, set, iteration, number,
23678931d482SJason Sarich 
23688931d482SJason Sarich .seealso:   TaoGetLinearSolveIterations()
23698931d482SJason Sarich @*/
23708931d482SJason Sarich PetscErrorCode  TaoSetTotalIterationNumber(Tao tao,PetscInt iter)
23718931d482SJason Sarich {
23728931d482SJason Sarich   PetscErrorCode ierr;
23738931d482SJason Sarich 
23748931d482SJason Sarich   PetscFunctionBegin;
23758931d482SJason Sarich   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
23768931d482SJason Sarich   ierr       = PetscObjectSAWsTakeAccess((PetscObject)tao);CHKERRQ(ierr);
23778931d482SJason Sarich   tao->ntotalits = iter;
23788931d482SJason Sarich   ierr       = PetscObjectSAWsGrantAccess((PetscObject)tao);CHKERRQ(ierr);
23798931d482SJason Sarich   PetscFunctionReturn(0);
23808931d482SJason Sarich }
23818931d482SJason Sarich 
2382a7e14dcfSSatish Balay /*@
2383e4cb33bbSBarry Smith   TaoSetConvergedReason - Sets the termination flag on a Tao object
2384a7e14dcfSSatish Balay 
2385441846f8SBarry Smith   Logically Collective on Tao
2386a7e14dcfSSatish Balay 
2387a7e14dcfSSatish Balay   Input Parameters:
2388441846f8SBarry Smith + tao - the Tao context
2389a7e14dcfSSatish Balay - reason - one of
2390a7e14dcfSSatish Balay $     TAO_CONVERGED_ATOL (2),
2391a7e14dcfSSatish Balay $     TAO_CONVERGED_RTOL (3),
2392a7e14dcfSSatish Balay $     TAO_CONVERGED_STEPTOL (4),
2393a7e14dcfSSatish Balay $     TAO_CONVERGED_MINF (5),
2394a7e14dcfSSatish Balay $     TAO_CONVERGED_USER (6),
2395a7e14dcfSSatish Balay $     TAO_DIVERGED_MAXITS (-2),
2396a7e14dcfSSatish Balay $     TAO_DIVERGED_NAN (-4),
2397a7e14dcfSSatish Balay $     TAO_DIVERGED_MAXFCN (-5),
2398a7e14dcfSSatish Balay $     TAO_DIVERGED_LS_FAILURE (-6),
2399a7e14dcfSSatish Balay $     TAO_DIVERGED_TR_REDUCTION (-7),
2400a7e14dcfSSatish Balay $     TAO_DIVERGED_USER (-8),
2401a7e14dcfSSatish Balay $     TAO_CONTINUE_ITERATING (0)
2402a7e14dcfSSatish Balay 
2403a7e14dcfSSatish Balay    Level: intermediate
2404a7e14dcfSSatish Balay 
2405a7e14dcfSSatish Balay @*/
2406e4cb33bbSBarry Smith PetscErrorCode TaoSetConvergedReason(Tao tao, TaoConvergedReason reason)
2407a7e14dcfSSatish Balay {
2408441846f8SBarry Smith   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
2409a7e14dcfSSatish Balay   PetscFunctionBegin;
2410a7e14dcfSSatish Balay   tao->reason = reason;
2411a7e14dcfSSatish Balay   PetscFunctionReturn(0);
2412a7e14dcfSSatish Balay }
2413a7e14dcfSSatish Balay 
2414a7e14dcfSSatish Balay /*@
2415e4cb33bbSBarry Smith    TaoGetConvergedReason - Gets the reason the Tao iteration was stopped.
2416a7e14dcfSSatish Balay 
2417a7e14dcfSSatish Balay    Not Collective
2418a7e14dcfSSatish Balay 
2419a7e14dcfSSatish Balay    Input Parameter:
2420441846f8SBarry Smith .  tao - the Tao solver context
2421a7e14dcfSSatish Balay 
2422a7e14dcfSSatish Balay    Output Parameter:
2423a7e14dcfSSatish Balay .  reason - one of
2424a7e14dcfSSatish Balay $  TAO_CONVERGED_GATOL (3)           ||g(X)|| < gatol
2425a7e14dcfSSatish Balay $  TAO_CONVERGED_GRTOL (4)           ||g(X)|| / f(X)  < grtol
2426a7e14dcfSSatish Balay $  TAO_CONVERGED_GTTOL (5)           ||g(X)|| / ||g(X0)|| < gttol
2427a7e14dcfSSatish Balay $  TAO_CONVERGED_STEPTOL (6)         step size small
2428a7e14dcfSSatish Balay $  TAO_CONVERGED_MINF (7)            F < F_min
2429a7e14dcfSSatish Balay $  TAO_CONVERGED_USER (8)            User defined
2430a7e14dcfSSatish Balay $  TAO_DIVERGED_MAXITS (-2)          its > maxits
2431a7e14dcfSSatish Balay $  TAO_DIVERGED_NAN (-4)             Numerical problems
2432a7e14dcfSSatish Balay $  TAO_DIVERGED_MAXFCN (-5)          fevals > max_funcsals
2433a7e14dcfSSatish Balay $  TAO_DIVERGED_LS_FAILURE (-6)      line search failure
2434a7e14dcfSSatish Balay $  TAO_DIVERGED_TR_REDUCTION (-7)    trust region failure
2435a7e14dcfSSatish Balay $  TAO_DIVERGED_USER(-8)             (user defined)
2436a7e14dcfSSatish Balay  $  TAO_CONTINUE_ITERATING (0)
2437a7e14dcfSSatish Balay 
2438a7e14dcfSSatish Balay    where
2439a7e14dcfSSatish Balay +  X - current solution
2440a7e14dcfSSatish Balay .  X0 - initial guess
2441a7e14dcfSSatish Balay .  f(X) - current function value
2442a7e14dcfSSatish Balay .  f(X*) - true solution (estimated)
2443a7e14dcfSSatish Balay .  g(X) - current gradient
2444a7e14dcfSSatish Balay .  its - current iterate number
2445a7e14dcfSSatish Balay .  maxits - maximum number of iterates
2446a7e14dcfSSatish Balay .  fevals - number of function evaluations
2447a7e14dcfSSatish Balay -  max_funcsals - maximum number of function evaluations
2448a7e14dcfSSatish Balay 
2449a7e14dcfSSatish Balay    Level: intermediate
2450a7e14dcfSSatish Balay 
2451a7e14dcfSSatish Balay .seealso: TaoSetConvergenceTest(), TaoSetTolerances()
2452a7e14dcfSSatish Balay 
2453a7e14dcfSSatish Balay @*/
2454e4cb33bbSBarry Smith PetscErrorCode TaoGetConvergedReason(Tao tao, TaoConvergedReason *reason)
2455a7e14dcfSSatish Balay {
2456a7e14dcfSSatish Balay   PetscFunctionBegin;
2457441846f8SBarry Smith   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
2458a7e14dcfSSatish Balay   PetscValidPointer(reason,2);
2459a7e14dcfSSatish Balay   *reason = tao->reason;
2460a7e14dcfSSatish Balay   PetscFunctionReturn(0);
2461a7e14dcfSSatish Balay }
2462a7e14dcfSSatish Balay 
2463a7e14dcfSSatish Balay /*@
2464a7e14dcfSSatish Balay   TaoGetSolutionStatus - Get the current iterate, objective value,
2465a7e14dcfSSatish Balay   residual, infeasibility, and termination
2466a7e14dcfSSatish Balay 
2467a7e14dcfSSatish Balay   Not Collective
2468a7e14dcfSSatish Balay 
2469a7e14dcfSSatish Balay    Input Parameters:
2470441846f8SBarry Smith .  tao - the Tao context
2471a7e14dcfSSatish Balay 
2472a7e14dcfSSatish Balay    Output Parameters:
2473a7e14dcfSSatish Balay +  iterate - the current iterate number (>=0)
2474a7e14dcfSSatish Balay .  f - the current function value
247558417fe7SBarry Smith .  gnorm - the square of the gradient norm, duality gap, or other measure indicating distance from optimality.
2476a7e14dcfSSatish Balay .  cnorm - the infeasibility of the current solution with regard to the constraints.
2477a7e14dcfSSatish Balay .  xdiff - the step length or trust region radius of the most recent iterate.
2478a7e14dcfSSatish Balay -  reason - The termination reason, which can equal TAO_CONTINUE_ITERATING
2479a7e14dcfSSatish Balay 
2480a7e14dcfSSatish Balay    Level: intermediate
2481a7e14dcfSSatish Balay 
2482a7e14dcfSSatish Balay    Note:
2483a7e14dcfSSatish Balay    TAO returns the values set by the solvers in the routine TaoMonitor().
2484a7e14dcfSSatish Balay 
2485a7e14dcfSSatish Balay    Note:
248647a47007SBarry Smith    If any of the output arguments are set to NULL, no corresponding value will be returned.
2487a7e14dcfSSatish Balay 
2488e4cb33bbSBarry Smith .seealso: TaoMonitor(), TaoGetConvergedReason()
2489a7e14dcfSSatish Balay @*/
2490e4cb33bbSBarry Smith PetscErrorCode TaoGetSolutionStatus(Tao tao, PetscInt *its, PetscReal *f, PetscReal *gnorm, PetscReal *cnorm, PetscReal *xdiff, TaoConvergedReason *reason)
2491a7e14dcfSSatish Balay {
2492a7e14dcfSSatish Balay   PetscFunctionBegin;
2493a7e14dcfSSatish Balay   if (its) *its=tao->niter;
2494a7e14dcfSSatish Balay   if (f) *f=tao->fc;
2495a7e14dcfSSatish Balay   if (gnorm) *gnorm=tao->residual;
2496a7e14dcfSSatish Balay   if (cnorm) *cnorm=tao->cnorm;
2497a7e14dcfSSatish Balay   if (reason) *reason=tao->reason;
2498a7e14dcfSSatish Balay   if (xdiff) *xdiff=tao->step;
2499a7e14dcfSSatish Balay   PetscFunctionReturn(0);
2500a7e14dcfSSatish Balay }
2501a7e14dcfSSatish Balay 
2502a7e14dcfSSatish Balay /*@C
2503441846f8SBarry Smith    TaoGetType - Gets the current Tao algorithm.
2504a7e14dcfSSatish Balay 
2505a7e14dcfSSatish Balay    Not Collective
2506a7e14dcfSSatish Balay 
2507a7e14dcfSSatish Balay    Input Parameter:
2508441846f8SBarry Smith .  tao - the Tao solver context
2509a7e14dcfSSatish Balay 
2510a7e14dcfSSatish Balay    Output Parameter:
2511441846f8SBarry Smith .  type - Tao method
2512a7e14dcfSSatish Balay 
2513a7e14dcfSSatish Balay    Level: intermediate
2514a7e14dcfSSatish Balay 
2515a7e14dcfSSatish Balay @*/
2516b625d6c7SJed Brown PetscErrorCode TaoGetType(Tao tao,TaoType *type)
2517a7e14dcfSSatish Balay {
2518a7e14dcfSSatish Balay   PetscFunctionBegin;
2519441846f8SBarry Smith   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
2520a7e14dcfSSatish Balay   PetscValidPointer(type,2);
2521a7e14dcfSSatish Balay   *type=((PetscObject)tao)->type_name;
2522a7e14dcfSSatish Balay   PetscFunctionReturn(0);
2523a7e14dcfSSatish Balay }
2524a7e14dcfSSatish Balay 
2525a7e14dcfSSatish Balay /*@C
2526a7e14dcfSSatish Balay   TaoMonitor - Monitor the solver and the current solution.  This
2527a7e14dcfSSatish Balay   routine will record the iteration number and residual statistics,
2528a7e14dcfSSatish Balay   call any monitors specified by the user, and calls the convergence-check routine.
2529a7e14dcfSSatish Balay 
2530a7e14dcfSSatish Balay    Input Parameters:
2531441846f8SBarry Smith +  tao - the Tao context
2532a7e14dcfSSatish Balay .  its - the current iterate number (>=0)
2533a7e14dcfSSatish Balay .  f - the current objective function value
253458417fe7SBarry Smith .  res - the gradient norm, square root of the duality gap, or other measure indicating distince from optimality.  This measure will be recorded and
2535a7e14dcfSSatish Balay           used for some termination tests.
2536a7e14dcfSSatish Balay .  cnorm - the infeasibility of the current solution with regard to the constraints.
2537a7e14dcfSSatish Balay -  steplength - multiple of the step direction added to the previous iterate.
2538a7e14dcfSSatish Balay 
2539a7e14dcfSSatish Balay    Output Parameters:
2540a7e14dcfSSatish Balay .  reason - The termination reason, which can equal TAO_CONTINUE_ITERATING
2541a7e14dcfSSatish Balay 
2542a7e14dcfSSatish Balay    Options Database Key:
2543a7e14dcfSSatish Balay .  -tao_monitor - Use the default monitor, which prints statistics to standard output
2544a7e14dcfSSatish Balay 
254598ea980cSBarry Smith .seealso TaoGetConvergedReason(), TaoMonitorDefault(), TaoSetMonitor()
2546a7e14dcfSSatish Balay 
2547a7e14dcfSSatish Balay    Level: developer
2548a7e14dcfSSatish Balay 
2549a7e14dcfSSatish Balay @*/
25503ecd9318SAlp Dener PetscErrorCode TaoMonitor(Tao tao, PetscInt its, PetscReal f, PetscReal res, PetscReal cnorm, PetscReal steplength)
2551a7e14dcfSSatish Balay {
2552a7e14dcfSSatish Balay   PetscErrorCode ierr;
255347a47007SBarry Smith   PetscInt       i;
255447a47007SBarry Smith 
2555a7e14dcfSSatish Balay   PetscFunctionBegin;
2556441846f8SBarry Smith   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
2557a7e14dcfSSatish Balay   tao->fc = f;
2558a7e14dcfSSatish Balay   tao->residual = res;
2559a7e14dcfSSatish Balay   tao->cnorm = cnorm;
2560a7e14dcfSSatish Balay   tao->step = steplength;
2561e52336cbSBarry Smith   if (!its) {
2562a7e14dcfSSatish Balay     tao->cnorm0 = cnorm; tao->gnorm0 = res;
2563a7e14dcfSSatish Balay   }
256447a47007SBarry Smith   if (PetscIsInfOrNanReal(f) || PetscIsInfOrNanReal(res)) SETERRQ(PETSC_COMM_SELF,1, "User provided compute function generated Inf or NaN");
2565a7e14dcfSSatish Balay   for (i=0;i<tao->numbermonitors;i++) {
2566a7e14dcfSSatish Balay     ierr = (*tao->monitor[i])(tao,tao->monitorcontext[i]);CHKERRQ(ierr);
2567a7e14dcfSSatish Balay   }
2568a7e14dcfSSatish Balay   PetscFunctionReturn(0);
2569a7e14dcfSSatish Balay }
2570a7e14dcfSSatish Balay 
2571a7e14dcfSSatish Balay /*@
2572ae93cb3cSJason Sarich    TaoSetConvergenceHistory - Sets the array used to hold the convergence history.
2573a7e14dcfSSatish Balay 
2574441846f8SBarry Smith    Logically Collective on Tao
2575a7e14dcfSSatish Balay 
2576a7e14dcfSSatish Balay    Input Parameters:
2577441846f8SBarry Smith +  tao - the Tao solver context
2578a7e14dcfSSatish Balay .  obj   - array to hold objective value history
2579a7e14dcfSSatish Balay .  resid - array to hold residual history
2580a7e14dcfSSatish Balay .  cnorm - array to hold constraint violation history
2581be1558d8SJason Sarich .  lits - integer array holds the number of linear iterations for each Tao iteration
2582a7e14dcfSSatish Balay .  na  - size of obj, resid, and cnorm
2583a7e14dcfSSatish Balay -  reset - PetscTrue indicates each new minimization resets the history counter to zero,
2584a7e14dcfSSatish Balay            else it continues storing new values for new minimizations after the old ones
2585a7e14dcfSSatish Balay 
2586a7e14dcfSSatish Balay    Notes:
2587a7e14dcfSSatish Balay    If set, TAO will fill the given arrays with the indicated
2588ae93cb3cSJason Sarich    information at each iteration.  If 'obj','resid','cnorm','lits' are
2589ae93cb3cSJason Sarich    *all* NULL then space (using size na, or 1000 if na is PETSC_DECIDE or
2590ae93cb3cSJason Sarich    PETSC_DEFAULT) is allocated for the history.
2591be1558d8SJason Sarich    If not all are NULL, then only the non-NULL information categories
2592ae93cb3cSJason Sarich    will be stored, the others will be ignored.
2593ae93cb3cSJason Sarich 
2594ae93cb3cSJason Sarich    Any convergence information after iteration number 'na' will not be stored.
2595a7e14dcfSSatish Balay 
2596a7e14dcfSSatish Balay    This routine is useful, e.g., when running a code for purposes
2597a7e14dcfSSatish Balay    of accurate performance monitoring, when no I/O should be done
2598a7e14dcfSSatish Balay    during the section of code that is being timed.
2599a7e14dcfSSatish Balay 
2600a7e14dcfSSatish Balay    Level: intermediate
2601a7e14dcfSSatish Balay 
2602ae93cb3cSJason Sarich .seealso: TaoGetConvergenceHistory()
2603a7e14dcfSSatish Balay 
2604a7e14dcfSSatish Balay @*/
26051b266c99SBarry Smith PetscErrorCode TaoSetConvergenceHistory(Tao tao, PetscReal obj[], PetscReal resid[], PetscReal cnorm[], PetscInt lits[], PetscInt na,PetscBool reset)
2606a7e14dcfSSatish Balay {
2607ae93cb3cSJason Sarich   PetscErrorCode ierr;
2608ae93cb3cSJason Sarich 
2609a7e14dcfSSatish Balay   PetscFunctionBegin;
2610441846f8SBarry Smith   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
2611ae93cb3cSJason Sarich   if (obj) PetscValidScalarPointer(obj,2);
2612ae93cb3cSJason Sarich   if (resid) PetscValidScalarPointer(resid,3);
2613ae93cb3cSJason Sarich   if (cnorm) PetscValidScalarPointer(cnorm,4);
2614ae93cb3cSJason Sarich   if (lits) PetscValidIntPointer(lits,5);
2615ae93cb3cSJason Sarich 
2616ae93cb3cSJason Sarich   if (na == PETSC_DECIDE || na == PETSC_DEFAULT) na = 1000;
2617ae93cb3cSJason Sarich   if (!obj && !resid && !cnorm && !lits) {
2618ae93cb3cSJason Sarich     ierr = PetscCalloc1(na,&obj);CHKERRQ(ierr);
2619ae93cb3cSJason Sarich     ierr = PetscCalloc1(na,&resid);CHKERRQ(ierr);
2620ae93cb3cSJason Sarich     ierr = PetscCalloc1(na,&cnorm);CHKERRQ(ierr);
2621ae93cb3cSJason Sarich     ierr = PetscCalloc1(na,&lits);CHKERRQ(ierr);
2622ae93cb3cSJason Sarich     tao->hist_malloc=PETSC_TRUE;
2623ae93cb3cSJason Sarich   }
2624ae93cb3cSJason Sarich 
2625a7e14dcfSSatish Balay   tao->hist_obj = obj;
2626a7e14dcfSSatish Balay   tao->hist_resid = resid;
2627a7e14dcfSSatish Balay   tao->hist_cnorm = cnorm;
2628ae93cb3cSJason Sarich   tao->hist_lits = lits;
2629a7e14dcfSSatish Balay   tao->hist_max   = na;
2630a7e14dcfSSatish Balay   tao->hist_reset = reset;
2631ae93cb3cSJason Sarich   tao->hist_len = 0;
2632a7e14dcfSSatish Balay   PetscFunctionReturn(0);
2633a7e14dcfSSatish Balay }
2634a7e14dcfSSatish Balay 
2635a7e14dcfSSatish Balay /*@C
2636ae93cb3cSJason Sarich    TaoGetConvergenceHistory - Gets the arrays used to hold the convergence history.
2637a7e14dcfSSatish Balay 
2638441846f8SBarry Smith    Collective on Tao
2639a7e14dcfSSatish Balay 
2640a7e14dcfSSatish Balay    Input Parameter:
2641441846f8SBarry Smith .  tao - the Tao context
2642a7e14dcfSSatish Balay 
2643a7e14dcfSSatish Balay    Output Parameters:
2644a7e14dcfSSatish Balay +  obj   - array used to hold objective value history
2645a7e14dcfSSatish Balay .  resid - array used to hold residual history
2646a7e14dcfSSatish Balay .  cnorm - array used to hold constraint violation history
2647ae93cb3cSJason Sarich .  lits  - integer array used to hold linear solver iteration count
2648ae93cb3cSJason Sarich -  nhist  - size of obj, resid, cnorm, and lits (will be less than or equal to na given in TaoSetHistory)
2649a7e14dcfSSatish Balay 
2650a7e14dcfSSatish Balay    Notes:
2651ae93cb3cSJason Sarich     This routine must be preceded by calls to TaoSetConvergenceHistory()
2652ae93cb3cSJason Sarich     and TaoSolve(), otherwise it returns useless information.
2653ae93cb3cSJason Sarich 
2654a7e14dcfSSatish Balay     The calling sequence for this routine in Fortran is
2655ae93cb3cSJason Sarich $   call TaoGetConvergenceHistory(Tao tao, PetscInt nhist, PetscErrorCode ierr)
2656a7e14dcfSSatish Balay 
2657a7e14dcfSSatish Balay    This routine is useful, e.g., when running a code for purposes
2658a7e14dcfSSatish Balay    of accurate performance monitoring, when no I/O should be done
2659a7e14dcfSSatish Balay    during the section of code that is being timed.
2660a7e14dcfSSatish Balay 
2661a7e14dcfSSatish Balay    Level: advanced
2662a7e14dcfSSatish Balay 
2663ae93cb3cSJason Sarich .seealso: TaoSetConvergenceHistory()
2664a7e14dcfSSatish Balay 
2665a7e14dcfSSatish Balay @*/
2666ae93cb3cSJason Sarich PetscErrorCode TaoGetConvergenceHistory(Tao tao, PetscReal **obj, PetscReal **resid, PetscReal **cnorm, PetscInt **lits, PetscInt *nhist)
2667a7e14dcfSSatish Balay {
2668a7e14dcfSSatish Balay   PetscFunctionBegin;
2669441846f8SBarry Smith   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
2670a7e14dcfSSatish Balay   if (obj)   *obj   = tao->hist_obj;
2671a7e14dcfSSatish Balay   if (cnorm) *cnorm = tao->hist_cnorm;
2672a7e14dcfSSatish Balay   if (resid) *resid = tao->hist_resid;
2673a7e14dcfSSatish Balay   if (nhist) *nhist   = tao->hist_len;
2674a7e14dcfSSatish Balay   PetscFunctionReturn(0);
2675a7e14dcfSSatish Balay }
2676a7e14dcfSSatish Balay 
2677a7e14dcfSSatish Balay /*@
2678a7e14dcfSSatish Balay    TaoSetApplicationContext - Sets the optional user-defined context for
2679a7e14dcfSSatish Balay    a solver.
2680a7e14dcfSSatish Balay 
2681441846f8SBarry Smith    Logically Collective on Tao
2682a7e14dcfSSatish Balay 
2683a7e14dcfSSatish Balay    Input Parameters:
2684441846f8SBarry Smith +  tao  - the Tao context
2685a7e14dcfSSatish Balay -  usrP - optional user context
2686a7e14dcfSSatish Balay 
2687a7e14dcfSSatish Balay    Level: intermediate
2688a7e14dcfSSatish Balay 
2689a7e14dcfSSatish Balay .seealso: TaoGetApplicationContext(), TaoSetApplicationContext()
2690a7e14dcfSSatish Balay @*/
2691441846f8SBarry Smith PetscErrorCode  TaoSetApplicationContext(Tao tao,void *usrP)
2692a7e14dcfSSatish Balay {
2693a7e14dcfSSatish Balay   PetscFunctionBegin;
2694441846f8SBarry Smith   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
2695a7e14dcfSSatish Balay   tao->user = usrP;
2696a7e14dcfSSatish Balay   PetscFunctionReturn(0);
2697a7e14dcfSSatish Balay }
2698a7e14dcfSSatish Balay 
2699a7e14dcfSSatish Balay /*@
2700a7e14dcfSSatish Balay    TaoGetApplicationContext - Gets the user-defined context for a
2701a7e14dcfSSatish Balay    TAO solvers.
2702a7e14dcfSSatish Balay 
2703a7e14dcfSSatish Balay    Not Collective
2704a7e14dcfSSatish Balay 
2705a7e14dcfSSatish Balay    Input Parameter:
2706441846f8SBarry Smith .  tao  - Tao context
2707a7e14dcfSSatish Balay 
2708a7e14dcfSSatish Balay    Output Parameter:
2709a7e14dcfSSatish Balay .  usrP - user context
2710a7e14dcfSSatish Balay 
2711a7e14dcfSSatish Balay    Level: intermediate
2712a7e14dcfSSatish Balay 
2713a7e14dcfSSatish Balay .seealso: TaoSetApplicationContext()
2714a7e14dcfSSatish Balay @*/
2715441846f8SBarry Smith PetscErrorCode  TaoGetApplicationContext(Tao tao,void *usrP)
2716a7e14dcfSSatish Balay {
2717a7e14dcfSSatish Balay   PetscFunctionBegin;
2718441846f8SBarry Smith   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
2719a7e14dcfSSatish Balay   *(void**)usrP = tao->user;
2720a7e14dcfSSatish Balay   PetscFunctionReturn(0);
2721a7e14dcfSSatish Balay }
2722a9603a14SPatrick Farrell 
2723a9603a14SPatrick Farrell /*@
2724a9603a14SPatrick Farrell    TaoSetGradientNorm - Sets the matrix used to define the inner product that measures the size of the gradient.
2725a9603a14SPatrick Farrell 
2726a9603a14SPatrick Farrell    Collective on tao
2727a9603a14SPatrick Farrell 
2728a9603a14SPatrick Farrell    Input Parameters:
2729a9603a14SPatrick Farrell +  tao  - the Tao context
2730a9603a14SPatrick Farrell -  M    - gradient norm
2731a9603a14SPatrick Farrell 
2732a9603a14SPatrick Farrell    Level: beginner
2733a9603a14SPatrick Farrell 
2734a9603a14SPatrick Farrell .seealso: TaoGetGradientNorm(), TaoGradientNorm()
2735a9603a14SPatrick Farrell @*/
2736a9603a14SPatrick Farrell PetscErrorCode  TaoSetGradientNorm(Tao tao, Mat M)
2737a9603a14SPatrick Farrell {
2738a9603a14SPatrick Farrell   PetscErrorCode ierr;
2739a9603a14SPatrick Farrell 
2740a9603a14SPatrick Farrell   PetscFunctionBegin;
2741a9603a14SPatrick Farrell   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
2742a9603a14SPatrick Farrell 
2743a9603a14SPatrick Farrell   if (tao->gradient_norm) {
2744a9603a14SPatrick Farrell     ierr = PetscObjectDereference((PetscObject)tao->gradient_norm);CHKERRQ(ierr);
2745a9603a14SPatrick Farrell     ierr = VecDestroy(&tao->gradient_norm_tmp);CHKERRQ(ierr);
2746a9603a14SPatrick Farrell   }
2747a9603a14SPatrick Farrell 
2748a9603a14SPatrick Farrell   ierr = PetscObjectReference((PetscObject)M);CHKERRQ(ierr);
2749a9603a14SPatrick Farrell   tao->gradient_norm = M;
2750a9603a14SPatrick Farrell   ierr = MatCreateVecs(M, NULL, &tao->gradient_norm_tmp);CHKERRQ(ierr);
2751a9603a14SPatrick Farrell   PetscFunctionReturn(0);
2752a9603a14SPatrick Farrell }
2753a9603a14SPatrick Farrell 
2754a9603a14SPatrick Farrell /*@
2755a9603a14SPatrick Farrell    TaoGetGradientNorm - Returns the matrix used to define the inner product for measuring the size of the gradient.
2756a9603a14SPatrick Farrell 
2757a9603a14SPatrick Farrell    Not Collective
2758a9603a14SPatrick Farrell 
2759a9603a14SPatrick Farrell    Input Parameter:
2760a9603a14SPatrick Farrell .  tao  - Tao context
2761a9603a14SPatrick Farrell 
2762a9603a14SPatrick Farrell    Output Parameter:
2763a9603a14SPatrick Farrell .  M - gradient norm
2764a9603a14SPatrick Farrell 
2765a9603a14SPatrick Farrell    Level: beginner
2766a9603a14SPatrick Farrell 
2767a9603a14SPatrick Farrell .seealso: TaoSetGradientNorm(), TaoGradientNorm()
2768a9603a14SPatrick Farrell @*/
2769a9603a14SPatrick Farrell PetscErrorCode  TaoGetGradientNorm(Tao tao, Mat *M)
2770a9603a14SPatrick Farrell {
2771a9603a14SPatrick Farrell   PetscFunctionBegin;
2772a9603a14SPatrick Farrell   PetscValidHeaderSpecific(tao,TAO_CLASSID,1);
2773a9603a14SPatrick Farrell   *M = tao->gradient_norm;
2774a9603a14SPatrick Farrell   PetscFunctionReturn(0);
2775a9603a14SPatrick Farrell }
2776a9603a14SPatrick Farrell 
2777a9603a14SPatrick Farrell /*c
2778a9603a14SPatrick Farrell    TaoGradientNorm - Compute the norm with respect to the inner product the user has set.
2779a9603a14SPatrick Farrell 
2780a9603a14SPatrick Farrell    Collective on tao
2781a9603a14SPatrick Farrell 
2782a9603a14SPatrick Farrell    Input Parameter:
2783a9603a14SPatrick Farrell .  tao      - the Tao context
2784a9603a14SPatrick Farrell .  gradient - the gradient to be computed
2785a9603a14SPatrick Farrell .  norm     - the norm type
2786a9603a14SPatrick Farrell 
2787a9603a14SPatrick Farrell    Output Parameter:
2788a9603a14SPatrick Farrell .  gnorm    - the gradient norm
2789a9603a14SPatrick Farrell 
2790a9603a14SPatrick Farrell    Level: developer
2791a9603a14SPatrick Farrell 
2792a9603a14SPatrick Farrell .seealso: TaoSetGradientNorm(), TaoGetGradientNorm()
2793a9603a14SPatrick Farrell @*/
2794a9603a14SPatrick Farrell PetscErrorCode  TaoGradientNorm(Tao tao, Vec gradient, NormType type, PetscReal *gnorm)
2795a9603a14SPatrick Farrell {
2796a9603a14SPatrick Farrell   PetscErrorCode ierr;
2797a9603a14SPatrick Farrell 
2798a9603a14SPatrick Farrell   PetscFunctionBegin;
2799a9603a14SPatrick Farrell   PetscValidHeaderSpecific(gradient,VEC_CLASSID,1);
2800a9603a14SPatrick Farrell 
2801a9603a14SPatrick Farrell   if (tao->gradient_norm) {
2802680e2bc7SPatrick Farrell     PetscScalar gnorms;
2803680e2bc7SPatrick Farrell 
2804a9603a14SPatrick Farrell     if (type != NORM_2) SETERRQ(PetscObjectComm((PetscObject)gradient), PETSC_ERR_ARG_WRONGSTATE, "Norm type must be NORM_2 if an inner product for the gradient norm is set.");
2805a9603a14SPatrick Farrell     ierr = MatMult(tao->gradient_norm, gradient, tao->gradient_norm_tmp);CHKERRQ(ierr);
2806680e2bc7SPatrick Farrell     ierr = VecDot(gradient, tao->gradient_norm_tmp, &gnorms);CHKERRQ(ierr);
28077bb79937SPatrick Farrell     *gnorm = PetscRealPart(PetscSqrtScalar(gnorms));
2808a9603a14SPatrick Farrell   } else {
2809a9603a14SPatrick Farrell     ierr = VecNorm(gradient, type, gnorm);CHKERRQ(ierr);
2810a9603a14SPatrick Farrell   }
2811a9603a14SPatrick Farrell   PetscFunctionReturn(0);
2812a9603a14SPatrick Farrell }
2813a9603a14SPatrick Farrell 
2814e882e171SHong Zhang /*@C
2815e882e171SHong Zhang    TaoMonitorDrawCtxCreate - Creates the monitor context for TaoMonitorDrawCtx
2816a9603a14SPatrick Farrell 
2817e882e171SHong Zhang    Collective on Tao
2818e882e171SHong Zhang 
2819e882e171SHong Zhang    Output Patameter:
2820e882e171SHong Zhang .    ctx - the monitor context
2821e882e171SHong Zhang 
2822e882e171SHong Zhang    Options Database:
2823e882e171SHong Zhang .   -tao_draw_solution_initial - show initial guess as well as current solution
2824e882e171SHong Zhang 
2825e882e171SHong Zhang    Level: intermediate
2826e882e171SHong Zhang 
2827e882e171SHong Zhang .keywords: Tao,  vector, monitor, view
2828e882e171SHong Zhang 
2829e882e171SHong Zhang .seealso: TaoMonitorSet(), TaoMonitorDefault(), VecView(), TaoMonitorDrawCtx()
2830e882e171SHong Zhang @*/
2831e882e171SHong Zhang PetscErrorCode  TaoMonitorDrawCtxCreate(MPI_Comm comm,const char host[],const char label[],int x,int y,int m,int n,PetscInt howoften,TaoMonitorDrawCtx *ctx)
2832e882e171SHong Zhang {
2833e882e171SHong Zhang   PetscErrorCode   ierr;
2834e882e171SHong Zhang 
2835e882e171SHong Zhang   PetscFunctionBegin;
2836e882e171SHong Zhang   ierr = PetscNew(ctx);CHKERRQ(ierr);
2837e882e171SHong Zhang   ierr = PetscViewerDrawOpen(comm,host,label,x,y,m,n,&(*ctx)->viewer);CHKERRQ(ierr);
2838e882e171SHong Zhang   ierr = PetscViewerSetFromOptions((*ctx)->viewer);CHKERRQ(ierr);
2839e882e171SHong Zhang   (*ctx)->howoften = howoften;
2840e882e171SHong Zhang   PetscFunctionReturn(0);
2841e882e171SHong Zhang }
2842e882e171SHong Zhang 
2843e882e171SHong Zhang /*@C
2844e882e171SHong Zhang    TaoMonitorDrawCtxDestroy - Destroys the monitor context for TaoMonitorDrawSolution()
2845e882e171SHong Zhang 
2846e882e171SHong Zhang    Collective on Tao
2847e882e171SHong Zhang 
2848e882e171SHong Zhang    Input Parameters:
2849e882e171SHong Zhang .    ctx - the monitor context
2850e882e171SHong Zhang 
2851e882e171SHong Zhang    Level: intermediate
2852e882e171SHong Zhang 
2853e882e171SHong Zhang .keywords: Tao,  vector, monitor, view
2854e882e171SHong Zhang 
2855e882e171SHong Zhang .seealso: TaoMonitorSet(), TaoMonitorDefault(), VecView(), TaoMonitorDrawSolution()
2856e882e171SHong Zhang @*/
2857e882e171SHong Zhang PetscErrorCode  TaoMonitorDrawCtxDestroy(TaoMonitorDrawCtx *ictx)
2858e882e171SHong Zhang {
2859e882e171SHong Zhang   PetscErrorCode ierr;
2860e882e171SHong Zhang 
2861e882e171SHong Zhang   PetscFunctionBegin;
2862e882e171SHong Zhang   ierr = PetscViewerDestroy(&(*ictx)->viewer);CHKERRQ(ierr);
2863e882e171SHong Zhang   ierr = PetscFree(*ictx);CHKERRQ(ierr);
2864e882e171SHong Zhang   PetscFunctionReturn(0);
2865e882e171SHong Zhang }
2866