xref: /petsc/src/tao/unconstrained/impls/nls/nls.c (revision 3c859ba3a04a72e8efcb87bc7ffc046a6cbab413)
1ba92ff59SBarry Smith #include <petsctaolinesearch.h>
21daac38eSTodd Munson #include <../src/tao/unconstrained/impls/nls/nlsimpl.h>
3a7e14dcfSSatish Balay 
4aaa7dc30SBarry Smith #include <petscksp.h>
5a7e14dcfSSatish Balay 
6a7e14dcfSSatish Balay #define NLS_INIT_CONSTANT         0
7a7e14dcfSSatish Balay #define NLS_INIT_DIRECTION        1
8a7e14dcfSSatish Balay #define NLS_INIT_INTERPOLATION    2
9a7e14dcfSSatish Balay #define NLS_INIT_TYPES            3
10a7e14dcfSSatish Balay 
11a7e14dcfSSatish Balay #define NLS_UPDATE_STEP           0
12a7e14dcfSSatish Balay #define NLS_UPDATE_REDUCTION      1
13a7e14dcfSSatish Balay #define NLS_UPDATE_INTERPOLATION  2
14a7e14dcfSSatish Balay #define NLS_UPDATE_TYPES          3
15a7e14dcfSSatish Balay 
1687f595a5SBarry Smith static const char *NLS_INIT[64] = {"constant", "direction", "interpolation"};
17a7e14dcfSSatish Balay 
1887f595a5SBarry Smith static const char *NLS_UPDATE[64] = {"step", "reduction", "interpolation"};
19a7e14dcfSSatish Balay 
201daac38eSTodd Munson /*
21a7e14dcfSSatish Balay  Implements Newton's Method with a line search approach for solving
22a7e14dcfSSatish Balay  unconstrained minimization problems.  A More'-Thuente line search
23a7e14dcfSSatish Balay  is used to guarantee that the bfgs preconditioner remains positive
24a7e14dcfSSatish Balay  definite.
25a7e14dcfSSatish Balay 
26a7e14dcfSSatish Balay  The method can shift the Hessian matrix.  The shifting procedure is
27a7e14dcfSSatish Balay  adapted from the PATH algorithm for solving complementarity
28a7e14dcfSSatish Balay  problems.
29a7e14dcfSSatish Balay 
30a7e14dcfSSatish Balay  The linear system solve should be done with a conjugate gradient
311daac38eSTodd Munson  method, although any method can be used.
321daac38eSTodd Munson */
33a7e14dcfSSatish Balay 
34a7e14dcfSSatish Balay #define NLS_NEWTON              0
35a7e14dcfSSatish Balay #define NLS_BFGS                1
360c51296cSAlp Dener #define NLS_GRADIENT            2
37a7e14dcfSSatish Balay 
38441846f8SBarry Smith static PetscErrorCode TaoSolve_NLS(Tao tao)
39a7e14dcfSSatish Balay {
40a7e14dcfSSatish Balay   PetscErrorCode               ierr;
41a7e14dcfSSatish Balay   TAO_NLS                      *nlsP = (TAO_NLS *)tao->data;
421daac38eSTodd Munson   KSPType                      ksp_type;
430ad3a497SAlp Dener   PetscBool                    is_nash,is_stcg,is_gltr,is_bfgs,is_jacobi,is_symmetric,sym_set;
44a7e14dcfSSatish Balay   KSPConvergedReason           ksp_reason;
451daac38eSTodd Munson   PC                           pc;
461daac38eSTodd Munson   TaoLineSearchConvergedReason ls_reason;
47a7e14dcfSSatish Balay 
48a7e14dcfSSatish Balay   PetscReal                    fmin, ftrial, f_full, prered, actred, kappa, sigma;
49a7e14dcfSSatish Balay   PetscReal                    tau, tau_1, tau_2, tau_max, tau_min, max_radius;
50a7e14dcfSSatish Balay   PetscReal                    f, fold, gdx, gnorm, pert;
51a7e14dcfSSatish Balay   PetscReal                    step = 1.0;
52a7e14dcfSSatish Balay   PetscReal                    norm_d = 0.0, e_min;
53a7e14dcfSSatish Balay 
54a7e14dcfSSatish Balay   PetscInt                     stepType;
55a7e14dcfSSatish Balay   PetscInt                     bfgsUpdates = 0;
56a7e14dcfSSatish Balay   PetscInt                     n,N,kspits;
57b4690188SBarry Smith   PetscInt                     needH = 1;
58a7e14dcfSSatish Balay 
59a7e14dcfSSatish Balay   PetscInt                     i_max = 5;
60a7e14dcfSSatish Balay   PetscInt                     j_max = 1;
61a7e14dcfSSatish Balay   PetscInt                     i, j;
62a7e14dcfSSatish Balay 
63a7e14dcfSSatish Balay   PetscFunctionBegin;
64a7e14dcfSSatish Balay   if (tao->XL || tao->XU || tao->ops->computebounds) {
659dcef436SStefano Zampini     ierr = PetscInfo(tao,"WARNING: Variable bounds have been set but will be ignored by nls algorithm\n");CHKERRQ(ierr);
66a7e14dcfSSatish Balay   }
67a7e14dcfSSatish Balay 
68a7e14dcfSSatish Balay   /* Initialized variables */
69a7e14dcfSSatish Balay   pert = nlsP->sval;
70a7e14dcfSSatish Balay 
712d9aa51bSJason Sarich   /* Number of times ksp stopped because of these reasons */
72a7e14dcfSSatish Balay   nlsP->ksp_atol = 0;
73a7e14dcfSSatish Balay   nlsP->ksp_rtol = 0;
74a7e14dcfSSatish Balay   nlsP->ksp_dtol = 0;
75a7e14dcfSSatish Balay   nlsP->ksp_ctol = 0;
76a7e14dcfSSatish Balay   nlsP->ksp_negc = 0;
77a7e14dcfSSatish Balay   nlsP->ksp_iter = 0;
78a7e14dcfSSatish Balay   nlsP->ksp_othr = 0;
79a7e14dcfSSatish Balay 
80a7e14dcfSSatish Balay   /* Initialize trust-region radius when using nash, stcg, or gltr
81ba7fe8fbSTodd Munson      Command automatically ignored for other methods
82ba7fe8fbSTodd Munson      Will be reset during the first iteration
83ba7fe8fbSTodd Munson   */
841daac38eSTodd Munson   ierr = KSPGetType(tao->ksp,&ksp_type);CHKERRQ(ierr);
8505de396fSBarry Smith   ierr = PetscStrcmp(ksp_type,KSPNASH,&is_nash);CHKERRQ(ierr);
8605de396fSBarry Smith   ierr = PetscStrcmp(ksp_type,KSPSTCG,&is_stcg);CHKERRQ(ierr);
8705de396fSBarry Smith   ierr = PetscStrcmp(ksp_type,KSPGLTR,&is_gltr);CHKERRQ(ierr);
881daac38eSTodd Munson 
89ba7fe8fbSTodd Munson   ierr = KSPCGSetRadius(tao->ksp,nlsP->max_radius);CHKERRQ(ierr);
90a7e14dcfSSatish Balay 
911daac38eSTodd Munson   if (is_nash || is_stcg || is_gltr) {
92*3c859ba3SBarry Smith     PetscCheck(tao->trust0 >= 0.0,PetscObjectComm((PetscObject)tao),PETSC_ERR_ARG_OUTOFRANGE,"Initial radius negative");
93a7e14dcfSSatish Balay     tao->trust = tao->trust0;
94a7e14dcfSSatish Balay     tao->trust = PetscMax(tao->trust, nlsP->min_radius);
95a7e14dcfSSatish Balay     tao->trust = PetscMin(tao->trust, nlsP->max_radius);
96a7e14dcfSSatish Balay   }
97a7e14dcfSSatish Balay 
98a7e14dcfSSatish Balay   /* Check convergence criteria */
99a7e14dcfSSatish Balay   ierr = TaoComputeObjectiveAndGradient(tao, tao->solution, &f, tao->gradient);CHKERRQ(ierr);
100a9603a14SPatrick Farrell   ierr = TaoGradientNorm(tao, tao->gradient,NORM_2,&gnorm);CHKERRQ(ierr);
101*3c859ba3SBarry Smith   PetscCheck(!PetscIsInfOrNanReal(f) && !PetscIsInfOrNanReal(gnorm),PetscObjectComm((PetscObject)tao),PETSC_ERR_USER, "User provided compute function generated Inf or NaN");
102a7e14dcfSSatish Balay 
1033ecd9318SAlp Dener   tao->reason = TAO_CONTINUE_ITERATING;
1043ecd9318SAlp Dener   ierr = TaoLogConvergenceHistory(tao,f,gnorm,0.0,tao->ksp_its);CHKERRQ(ierr);
1053ecd9318SAlp Dener   ierr = TaoMonitor(tao,tao->niter,f,gnorm,0.0,step);CHKERRQ(ierr);
1063ecd9318SAlp Dener   ierr = (*tao->ops->convergencetest)(tao,tao->cnvP);CHKERRQ(ierr);
1073ecd9318SAlp Dener   if (tao->reason != TAO_CONTINUE_ITERATING) PetscFunctionReturn(0);
108a7e14dcfSSatish Balay 
1090c51296cSAlp Dener   /* Allocate the vectors needed for the BFGS approximation */
110a7e14dcfSSatish Balay   ierr = KSPGetPC(tao->ksp, &pc);CHKERRQ(ierr);
1110c51296cSAlp Dener   ierr = PetscObjectTypeCompare((PetscObject)pc, PCLMVM, &is_bfgs);CHKERRQ(ierr);
1120c51296cSAlp Dener   ierr = PetscObjectTypeCompare((PetscObject)pc, PCJACOBI, &is_jacobi);CHKERRQ(ierr);
1130c51296cSAlp Dener   if (is_bfgs) {
1140c51296cSAlp Dener     nlsP->bfgs_pre = pc;
1150c51296cSAlp Dener     ierr = PCLMVMGetMatLMVM(nlsP->bfgs_pre, &nlsP->M);CHKERRQ(ierr);
1160c51296cSAlp Dener     ierr = VecGetLocalSize(tao->solution, &n);CHKERRQ(ierr);
1170c51296cSAlp Dener     ierr = VecGetSize(tao->solution, &N);CHKERRQ(ierr);
1180c51296cSAlp Dener     ierr = MatSetSizes(nlsP->M, n, n, N, N);CHKERRQ(ierr);
1190c51296cSAlp Dener     ierr = MatLMVMAllocate(nlsP->M, tao->solution, tao->gradient);CHKERRQ(ierr);
1200ad3a497SAlp Dener     ierr = MatIsSymmetricKnown(nlsP->M, &sym_set, &is_symmetric);CHKERRQ(ierr);
121*3c859ba3SBarry Smith     PetscCheck(sym_set && is_symmetric,PetscObjectComm((PetscObject)tao), PETSC_ERR_ARG_INCOMP, "LMVM matrix in the LMVM preconditioner must be symmetric.");
1220c51296cSAlp Dener   } else if (is_jacobi) {
123baa89ecbSBarry Smith     ierr = PCJacobiSetUseAbs(pc,PETSC_TRUE);CHKERRQ(ierr);
124a7e14dcfSSatish Balay   }
125a7e14dcfSSatish Balay 
126a7e14dcfSSatish Balay   /* Initialize trust-region radius.  The initialization is only performed
127a7e14dcfSSatish Balay      when we are using Nash, Steihaug-Toint or the Generalized Lanczos method. */
1281daac38eSTodd Munson   if (is_nash || is_stcg || is_gltr) {
129a7e14dcfSSatish Balay     switch (nlsP->init_type) {
130a7e14dcfSSatish Balay     case NLS_INIT_CONSTANT:
131a7e14dcfSSatish Balay       /* Use the initial radius specified */
132a7e14dcfSSatish Balay       break;
133a7e14dcfSSatish Balay 
134a7e14dcfSSatish Balay     case NLS_INIT_INTERPOLATION:
135a7e14dcfSSatish Balay       /* Use the initial radius specified */
136a7e14dcfSSatish Balay       max_radius = 0.0;
137a7e14dcfSSatish Balay 
138a7e14dcfSSatish Balay       for (j = 0; j < j_max; ++j) {
139a7e14dcfSSatish Balay         fmin = f;
140a7e14dcfSSatish Balay         sigma = 0.0;
141a7e14dcfSSatish Balay 
142a7e14dcfSSatish Balay         if (needH) {
143ffad9901SBarry Smith           ierr  = TaoComputeHessian(tao, tao->solution,tao->hessian,tao->hessian_pre);CHKERRQ(ierr);
144a7e14dcfSSatish Balay           needH = 0;
145a7e14dcfSSatish Balay         }
146a7e14dcfSSatish Balay 
147a7e14dcfSSatish Balay         for (i = 0; i < i_max; ++i) {
148a7e14dcfSSatish Balay           ierr = VecCopy(tao->solution,nlsP->W);CHKERRQ(ierr);
149a7e14dcfSSatish Balay           ierr = VecAXPY(nlsP->W,-tao->trust/gnorm,tao->gradient);CHKERRQ(ierr);
150a7e14dcfSSatish Balay           ierr = TaoComputeObjective(tao, nlsP->W, &ftrial);CHKERRQ(ierr);
151a7e14dcfSSatish Balay           if (PetscIsInfOrNanReal(ftrial)) {
152a7e14dcfSSatish Balay             tau = nlsP->gamma1_i;
15387f595a5SBarry Smith           } else {
154a7e14dcfSSatish Balay             if (ftrial < fmin) {
155a7e14dcfSSatish Balay               fmin = ftrial;
156a7e14dcfSSatish Balay               sigma = -tao->trust / gnorm;
157a7e14dcfSSatish Balay             }
158a7e14dcfSSatish Balay 
159a7e14dcfSSatish Balay             ierr = MatMult(tao->hessian, tao->gradient, nlsP->D);CHKERRQ(ierr);
160a7e14dcfSSatish Balay             ierr = VecDot(tao->gradient, nlsP->D, &prered);CHKERRQ(ierr);
161a7e14dcfSSatish Balay 
162a7e14dcfSSatish Balay             prered = tao->trust * (gnorm - 0.5 * tao->trust * prered / (gnorm * gnorm));
163a7e14dcfSSatish Balay             actred = f - ftrial;
16487f595a5SBarry Smith             if ((PetscAbsScalar(actred) <= nlsP->epsilon) && (PetscAbsScalar(prered) <= nlsP->epsilon)) {
165a7e14dcfSSatish Balay               kappa = 1.0;
16687f595a5SBarry Smith             } else {
167a7e14dcfSSatish Balay               kappa = actred / prered;
168a7e14dcfSSatish Balay             }
169a7e14dcfSSatish Balay 
170a7e14dcfSSatish Balay             tau_1 = nlsP->theta_i * gnorm * tao->trust / (nlsP->theta_i * gnorm * tao->trust + (1.0 - nlsP->theta_i) * prered - actred);
171a7e14dcfSSatish Balay             tau_2 = nlsP->theta_i * gnorm * tao->trust / (nlsP->theta_i * gnorm * tao->trust - (1.0 + nlsP->theta_i) * prered + actred);
172a7e14dcfSSatish Balay             tau_min = PetscMin(tau_1, tau_2);
173a7e14dcfSSatish Balay             tau_max = PetscMax(tau_1, tau_2);
174a7e14dcfSSatish Balay 
17518cfbf8eSSatish Balay             if (PetscAbsScalar(kappa - (PetscReal)1.0) <= nlsP->mu1_i) {
176a7e14dcfSSatish Balay               /* Great agreement */
177a7e14dcfSSatish Balay               max_radius = PetscMax(max_radius, tao->trust);
178a7e14dcfSSatish Balay 
179a7e14dcfSSatish Balay               if (tau_max < 1.0) {
180a7e14dcfSSatish Balay                 tau = nlsP->gamma3_i;
18187f595a5SBarry Smith               } else if (tau_max > nlsP->gamma4_i) {
182a7e14dcfSSatish Balay                 tau = nlsP->gamma4_i;
18387f595a5SBarry Smith               } else if (tau_1 >= 1.0 && tau_1 <= nlsP->gamma4_i && tau_2 < 1.0) {
184a7e14dcfSSatish Balay                 tau = tau_1;
18587f595a5SBarry Smith               } else if (tau_2 >= 1.0 && tau_2 <= nlsP->gamma4_i && tau_1 < 1.0) {
186a7e14dcfSSatish Balay                 tau = tau_2;
18787f595a5SBarry Smith               } else {
188a7e14dcfSSatish Balay                 tau = tau_max;
189a7e14dcfSSatish Balay               }
19018cfbf8eSSatish Balay             } else if (PetscAbsScalar(kappa - (PetscReal)1.0) <= nlsP->mu2_i) {
191a7e14dcfSSatish Balay               /* Good agreement */
192a7e14dcfSSatish Balay               max_radius = PetscMax(max_radius, tao->trust);
193a7e14dcfSSatish Balay 
194a7e14dcfSSatish Balay               if (tau_max < nlsP->gamma2_i) {
195a7e14dcfSSatish Balay                 tau = nlsP->gamma2_i;
19687f595a5SBarry Smith               } else if (tau_max > nlsP->gamma3_i) {
197a7e14dcfSSatish Balay                 tau = nlsP->gamma3_i;
19887f595a5SBarry Smith               } else {
199a7e14dcfSSatish Balay                 tau = tau_max;
200a7e14dcfSSatish Balay               }
20187f595a5SBarry Smith             } else {
202a7e14dcfSSatish Balay               /* Not good agreement */
203a7e14dcfSSatish Balay               if (tau_min > 1.0) {
204a7e14dcfSSatish Balay                 tau = nlsP->gamma2_i;
20587f595a5SBarry Smith               } else if (tau_max < nlsP->gamma1_i) {
206a7e14dcfSSatish Balay                 tau = nlsP->gamma1_i;
20787f595a5SBarry Smith               } else if ((tau_min < nlsP->gamma1_i) && (tau_max >= 1.0)) {
208a7e14dcfSSatish Balay                 tau = nlsP->gamma1_i;
20987f595a5SBarry Smith               } else if ((tau_1 >= nlsP->gamma1_i) && (tau_1 < 1.0) && ((tau_2 < nlsP->gamma1_i) || (tau_2 >= 1.0))) {
210a7e14dcfSSatish Balay                 tau = tau_1;
21187f595a5SBarry Smith               } else if ((tau_2 >= nlsP->gamma1_i) && (tau_2 < 1.0) && ((tau_1 < nlsP->gamma1_i) || (tau_2 >= 1.0))) {
212a7e14dcfSSatish Balay                 tau = tau_2;
21387f595a5SBarry Smith               } else {
214a7e14dcfSSatish Balay                 tau = tau_max;
215a7e14dcfSSatish Balay               }
216a7e14dcfSSatish Balay             }
217a7e14dcfSSatish Balay           }
218a7e14dcfSSatish Balay           tao->trust = tau * tao->trust;
219a7e14dcfSSatish Balay         }
220a7e14dcfSSatish Balay 
221a7e14dcfSSatish Balay         if (fmin < f) {
222a7e14dcfSSatish Balay           f = fmin;
223a7e14dcfSSatish Balay           ierr = VecAXPY(tao->solution,sigma,tao->gradient);CHKERRQ(ierr);
224a7e14dcfSSatish Balay           ierr = TaoComputeGradient(tao,tao->solution,tao->gradient);CHKERRQ(ierr);
225a7e14dcfSSatish Balay 
226a9603a14SPatrick Farrell           ierr = TaoGradientNorm(tao, tao->gradient,NORM_2,&gnorm);CHKERRQ(ierr);
227*3c859ba3SBarry Smith           PetscCheck(!PetscIsInfOrNanReal(gnorm),PetscObjectComm((PetscObject)tao),PETSC_ERR_USER, "User provided compute gradient generated Inf or NaN");
228a7e14dcfSSatish Balay           needH = 1;
229a7e14dcfSSatish Balay 
2303ecd9318SAlp Dener           ierr = TaoLogConvergenceHistory(tao,f,gnorm,0.0,tao->ksp_its);CHKERRQ(ierr);
2313ecd9318SAlp Dener           ierr = TaoMonitor(tao,tao->niter,f,gnorm,0.0,step);CHKERRQ(ierr);
2323ecd9318SAlp Dener           ierr = (*tao->ops->convergencetest)(tao,tao->cnvP);CHKERRQ(ierr);
2333ecd9318SAlp Dener           if (tao->reason != TAO_CONTINUE_ITERATING) PetscFunctionReturn(0);
234a7e14dcfSSatish Balay         }
235a7e14dcfSSatish Balay       }
236a7e14dcfSSatish Balay       tao->trust = PetscMax(tao->trust, max_radius);
237a7e14dcfSSatish Balay 
238a7e14dcfSSatish Balay       /* Modify the radius if it is too large or small */
239a7e14dcfSSatish Balay       tao->trust = PetscMax(tao->trust, nlsP->min_radius);
240a7e14dcfSSatish Balay       tao->trust = PetscMin(tao->trust, nlsP->max_radius);
241a7e14dcfSSatish Balay       break;
242a7e14dcfSSatish Balay 
243a7e14dcfSSatish Balay     default:
244a7e14dcfSSatish Balay       /* Norm of the first direction will initialize radius */
245a7e14dcfSSatish Balay       tao->trust = 0.0;
246a7e14dcfSSatish Balay       break;
247a7e14dcfSSatish Balay     }
248a7e14dcfSSatish Balay   }
249a7e14dcfSSatish Balay 
250a7e14dcfSSatish Balay   /* Set counter for gradient/reset steps*/
251a7e14dcfSSatish Balay   nlsP->newt = 0;
252a7e14dcfSSatish Balay   nlsP->bfgs = 0;
253a7e14dcfSSatish Balay   nlsP->grad = 0;
254a7e14dcfSSatish Balay 
255a7e14dcfSSatish Balay   /* Have not converged; continue with Newton method */
2563ecd9318SAlp Dener   while (tao->reason == TAO_CONTINUE_ITERATING) {
257e1e80dc8SAlp Dener     /* Call general purpose update function */
258e1e80dc8SAlp Dener     if (tao->ops->update) {
2598fcddce6SStefano Zampini       ierr = (*tao->ops->update)(tao, tao->niter, tao->user_update);CHKERRQ(ierr);
260e1e80dc8SAlp Dener     }
2618931d482SJason Sarich     ++tao->niter;
262ae93cb3cSJason Sarich     tao->ksp_its = 0;
263a7e14dcfSSatish Balay 
264a7e14dcfSSatish Balay     /* Compute the Hessian */
265a7e14dcfSSatish Balay     if (needH) {
266ffad9901SBarry Smith       ierr = TaoComputeHessian(tao,tao->solution,tao->hessian,tao->hessian_pre);CHKERRQ(ierr);
267a7e14dcfSSatish Balay     }
268a7e14dcfSSatish Balay 
269a7e14dcfSSatish Balay     /* Shift the Hessian matrix */
270a7e14dcfSSatish Balay     if (pert > 0) {
271302440fdSBarry Smith       ierr = MatShift(tao->hessian, pert);CHKERRQ(ierr);
272a7e14dcfSSatish Balay       if (tao->hessian != tao->hessian_pre) {
273a7e14dcfSSatish Balay         ierr = MatShift(tao->hessian_pre, pert);CHKERRQ(ierr);
274a7e14dcfSSatish Balay       }
275a7e14dcfSSatish Balay     }
276a7e14dcfSSatish Balay 
2770c51296cSAlp Dener     if (nlsP->bfgs_pre) {
278a7e14dcfSSatish Balay       ierr = MatLMVMUpdate(nlsP->M, tao->solution, tao->gradient);CHKERRQ(ierr);
279a7e14dcfSSatish Balay       ++bfgsUpdates;
280a7e14dcfSSatish Balay     }
281a7e14dcfSSatish Balay 
282a7e14dcfSSatish Balay     /* Solve the Newton system of equations */
28323ee1639SBarry Smith     ierr = KSPSetOperators(tao->ksp,tao->hessian,tao->hessian_pre);CHKERRQ(ierr);
2841daac38eSTodd Munson     if (is_nash || is_stcg || is_gltr) {
285ba7fe8fbSTodd Munson       ierr = KSPCGSetRadius(tao->ksp,nlsP->max_radius);CHKERRQ(ierr);
286a7e14dcfSSatish Balay       ierr = KSPSolve(tao->ksp, tao->gradient, nlsP->D);CHKERRQ(ierr);
287a7e14dcfSSatish Balay       ierr = KSPGetIterationNumber(tao->ksp,&kspits);CHKERRQ(ierr);
288a7e14dcfSSatish Balay       tao->ksp_its += kspits;
289ae93cb3cSJason Sarich       tao->ksp_tot_its += kspits;
290ba7fe8fbSTodd Munson       ierr = KSPCGGetNormD(tao->ksp,&norm_d);CHKERRQ(ierr);
291a7e14dcfSSatish Balay 
292a7e14dcfSSatish Balay       if (0.0 == tao->trust) {
293a7e14dcfSSatish Balay         /* Radius was uninitialized; use the norm of the direction */
294a7e14dcfSSatish Balay         if (norm_d > 0.0) {
295a7e14dcfSSatish Balay           tao->trust = norm_d;
296a7e14dcfSSatish Balay 
297a7e14dcfSSatish Balay           /* Modify the radius if it is too large or small */
298a7e14dcfSSatish Balay           tao->trust = PetscMax(tao->trust, nlsP->min_radius);
299a7e14dcfSSatish Balay           tao->trust = PetscMin(tao->trust, nlsP->max_radius);
30087f595a5SBarry Smith         } else {
301a7e14dcfSSatish Balay           /* The direction was bad; set radius to default value and re-solve
302a7e14dcfSSatish Balay              the trust-region subproblem to get a direction */
303a7e14dcfSSatish Balay           tao->trust = tao->trust0;
304a7e14dcfSSatish Balay 
305a7e14dcfSSatish Balay           /* Modify the radius if it is too large or small */
306a7e14dcfSSatish Balay           tao->trust = PetscMax(tao->trust, nlsP->min_radius);
307a7e14dcfSSatish Balay           tao->trust = PetscMin(tao->trust, nlsP->max_radius);
308a7e14dcfSSatish Balay 
309ba7fe8fbSTodd Munson           ierr = KSPCGSetRadius(tao->ksp,nlsP->max_radius);CHKERRQ(ierr);
310a7e14dcfSSatish Balay           ierr = KSPSolve(tao->ksp, tao->gradient, nlsP->D);CHKERRQ(ierr);
311a7e14dcfSSatish Balay           ierr = KSPGetIterationNumber(tao->ksp,&kspits);CHKERRQ(ierr);
312a7e14dcfSSatish Balay           tao->ksp_its += kspits;
313ae93cb3cSJason Sarich           tao->ksp_tot_its += kspits;
314ba7fe8fbSTodd Munson           ierr = KSPCGGetNormD(tao->ksp,&norm_d);CHKERRQ(ierr);
315a7e14dcfSSatish Balay 
316*3c859ba3SBarry Smith           PetscCheck(norm_d != 0.0,PETSC_COMM_SELF,PETSC_ERR_PLIB, "Initial direction zero");
317a7e14dcfSSatish Balay         }
318a7e14dcfSSatish Balay       }
31987f595a5SBarry Smith     } else {
320a7e14dcfSSatish Balay       ierr = KSPSolve(tao->ksp, tao->gradient, nlsP->D);CHKERRQ(ierr);
321a7e14dcfSSatish Balay       ierr = KSPGetIterationNumber(tao->ksp, &kspits);CHKERRQ(ierr);
322a7e14dcfSSatish Balay       tao->ksp_its += kspits;
323ae93cb3cSJason Sarich       tao->ksp_tot_its += kspits;
324a7e14dcfSSatish Balay     }
325a7e14dcfSSatish Balay     ierr = VecScale(nlsP->D, -1.0);CHKERRQ(ierr);
326a7e14dcfSSatish Balay     ierr = KSPGetConvergedReason(tao->ksp, &ksp_reason);CHKERRQ(ierr);
3270c51296cSAlp Dener     if ((KSP_DIVERGED_INDEFINITE_PC == ksp_reason) &&  (nlsP->bfgs_pre)) {
328a7e14dcfSSatish Balay       /* Preconditioner is numerically indefinite; reset the
329a7e14dcfSSatish Balay          approximate if using BFGS preconditioning. */
330cd929ea3SAlp Dener       ierr = MatLMVMReset(nlsP->M, PETSC_FALSE);CHKERRQ(ierr);
331a7e14dcfSSatish Balay       ierr = MatLMVMUpdate(nlsP->M, tao->solution, tao->gradient);CHKERRQ(ierr);
332a7e14dcfSSatish Balay       bfgsUpdates = 1;
333a7e14dcfSSatish Balay     }
334a7e14dcfSSatish Balay 
335a7e14dcfSSatish Balay     if (KSP_CONVERGED_ATOL == ksp_reason) {
336a7e14dcfSSatish Balay       ++nlsP->ksp_atol;
33787f595a5SBarry Smith     } else if (KSP_CONVERGED_RTOL == ksp_reason) {
338a7e14dcfSSatish Balay       ++nlsP->ksp_rtol;
33987f595a5SBarry Smith     } else if (KSP_CONVERGED_CG_CONSTRAINED == ksp_reason) {
340a7e14dcfSSatish Balay       ++nlsP->ksp_ctol;
34187f595a5SBarry Smith     } else if (KSP_CONVERGED_CG_NEG_CURVE == ksp_reason) {
342a7e14dcfSSatish Balay       ++nlsP->ksp_negc;
34387f595a5SBarry Smith     } else if (KSP_DIVERGED_DTOL == ksp_reason) {
344a7e14dcfSSatish Balay       ++nlsP->ksp_dtol;
34587f595a5SBarry Smith     } else if (KSP_DIVERGED_ITS == ksp_reason) {
346a7e14dcfSSatish Balay       ++nlsP->ksp_iter;
34787f595a5SBarry Smith     } else {
348a7e14dcfSSatish Balay       ++nlsP->ksp_othr;
349a7e14dcfSSatish Balay     }
350a7e14dcfSSatish Balay 
351a7e14dcfSSatish Balay     /* Check for success (descent direction) */
352a7e14dcfSSatish Balay     ierr = VecDot(nlsP->D, tao->gradient, &gdx);CHKERRQ(ierr);
353a7e14dcfSSatish Balay     if ((gdx >= 0.0) || PetscIsInfOrNanReal(gdx)) {
354a7e14dcfSSatish Balay       /* Newton step is not descent or direction produced Inf or NaN
355a7e14dcfSSatish Balay          Update the perturbation for next time */
356a7e14dcfSSatish Balay       if (pert <= 0.0) {
357a7e14dcfSSatish Balay         /* Initialize the perturbation */
358a7e14dcfSSatish Balay         pert = PetscMin(nlsP->imax, PetscMax(nlsP->imin, nlsP->imfac * gnorm));
3591daac38eSTodd Munson         if (is_gltr) {
36005de396fSBarry Smith           ierr = KSPGLTRGetMinEig(tao->ksp,&e_min);CHKERRQ(ierr);
361a7e14dcfSSatish Balay           pert = PetscMax(pert, -e_min);
362a7e14dcfSSatish Balay         }
36387f595a5SBarry Smith       } else {
364a7e14dcfSSatish Balay         /* Increase the perturbation */
365a7e14dcfSSatish Balay         pert = PetscMin(nlsP->pmax, PetscMax(nlsP->pgfac * pert, nlsP->pmgfac * gnorm));
366a7e14dcfSSatish Balay       }
367a7e14dcfSSatish Balay 
3680c51296cSAlp Dener       if (!nlsP->bfgs_pre) {
369a7e14dcfSSatish Balay         /* We don't have the bfgs matrix around and updated
370a7e14dcfSSatish Balay            Must use gradient direction in this case */
371a7e14dcfSSatish Balay         ierr = VecCopy(tao->gradient, nlsP->D);CHKERRQ(ierr);
372a7e14dcfSSatish Balay         ierr = VecScale(nlsP->D, -1.0);CHKERRQ(ierr);
373a7e14dcfSSatish Balay         ++nlsP->grad;
374a7e14dcfSSatish Balay         stepType = NLS_GRADIENT;
37587f595a5SBarry Smith       } else {
376a7e14dcfSSatish Balay         /* Attempt to use the BFGS direction */
377cd929ea3SAlp Dener         ierr = MatSolve(nlsP->M, tao->gradient, nlsP->D);CHKERRQ(ierr);
378a7e14dcfSSatish Balay         ierr = VecScale(nlsP->D, -1.0);CHKERRQ(ierr);
379a7e14dcfSSatish Balay 
380a7e14dcfSSatish Balay         /* Check for success (descent direction) */
381a7e14dcfSSatish Balay         ierr = VecDot(tao->gradient, nlsP->D, &gdx);CHKERRQ(ierr);
382a7e14dcfSSatish Balay         if ((gdx >= 0) || PetscIsInfOrNanReal(gdx)) {
383a7e14dcfSSatish Balay           /* BFGS direction is not descent or direction produced not a number
384a7e14dcfSSatish Balay              We can assert bfgsUpdates > 1 in this case because
385a7e14dcfSSatish Balay              the first solve produces the scaled gradient direction,
386a7e14dcfSSatish Balay              which is guaranteed to be descent */
387a7e14dcfSSatish Balay 
388a7e14dcfSSatish Balay           /* Use steepest descent direction (scaled) */
389cd929ea3SAlp Dener           ierr = MatLMVMReset(nlsP->M, PETSC_FALSE);CHKERRQ(ierr);
390a7e14dcfSSatish Balay           ierr = MatLMVMUpdate(nlsP->M, tao->solution, tao->gradient);CHKERRQ(ierr);
391cd929ea3SAlp Dener           ierr = MatSolve(nlsP->M, tao->gradient, nlsP->D);CHKERRQ(ierr);
392a7e14dcfSSatish Balay           ierr = VecScale(nlsP->D, -1.0);CHKERRQ(ierr);
393a7e14dcfSSatish Balay 
394a7e14dcfSSatish Balay           bfgsUpdates = 1;
3950c51296cSAlp Dener           ++nlsP->grad;
3960c51296cSAlp Dener           stepType = NLS_GRADIENT;
39787f595a5SBarry Smith         } else {
3980c51296cSAlp Dener           ierr = MatLMVMGetUpdateCount(nlsP->M, &bfgsUpdates);CHKERRQ(ierr);
399a7e14dcfSSatish Balay           if (1 == bfgsUpdates) {
400a7e14dcfSSatish Balay             /* The first BFGS direction is always the scaled gradient */
4010c51296cSAlp Dener             ++nlsP->grad;
4020c51296cSAlp Dener             stepType = NLS_GRADIENT;
40387f595a5SBarry Smith           } else {
404a7e14dcfSSatish Balay             ++nlsP->bfgs;
405a7e14dcfSSatish Balay             stepType = NLS_BFGS;
406a7e14dcfSSatish Balay           }
407a7e14dcfSSatish Balay         }
408a7e14dcfSSatish Balay       }
40987f595a5SBarry Smith     } else {
410a7e14dcfSSatish Balay       /* Computed Newton step is descent */
411a7e14dcfSSatish Balay       switch (ksp_reason) {
412a7e14dcfSSatish Balay       case KSP_DIVERGED_NANORINF:
413a7e14dcfSSatish Balay       case KSP_DIVERGED_BREAKDOWN:
414a7e14dcfSSatish Balay       case KSP_DIVERGED_INDEFINITE_MAT:
415a7e14dcfSSatish Balay       case KSP_DIVERGED_INDEFINITE_PC:
416a7e14dcfSSatish Balay       case KSP_CONVERGED_CG_NEG_CURVE:
417a7e14dcfSSatish Balay         /* Matrix or preconditioner is indefinite; increase perturbation */
418a7e14dcfSSatish Balay         if (pert <= 0.0) {
419a7e14dcfSSatish Balay           /* Initialize the perturbation */
420a7e14dcfSSatish Balay           pert = PetscMin(nlsP->imax, PetscMax(nlsP->imin, nlsP->imfac * gnorm));
4211daac38eSTodd Munson           if (is_gltr) {
42205de396fSBarry Smith             ierr = KSPGLTRGetMinEig(tao->ksp, &e_min);CHKERRQ(ierr);
423a7e14dcfSSatish Balay             pert = PetscMax(pert, -e_min);
424a7e14dcfSSatish Balay           }
42587f595a5SBarry Smith         } else {
426a7e14dcfSSatish Balay           /* Increase the perturbation */
427a7e14dcfSSatish Balay           pert = PetscMin(nlsP->pmax, PetscMax(nlsP->pgfac * pert, nlsP->pmgfac * gnorm));
428a7e14dcfSSatish Balay         }
429a7e14dcfSSatish Balay         break;
430a7e14dcfSSatish Balay 
431a7e14dcfSSatish Balay       default:
432a7e14dcfSSatish Balay         /* Newton step computation is good; decrease perturbation */
433a7e14dcfSSatish Balay         pert = PetscMin(nlsP->psfac * pert, nlsP->pmsfac * gnorm);
434a7e14dcfSSatish Balay         if (pert < nlsP->pmin) {
435a7e14dcfSSatish Balay           pert = 0.0;
436a7e14dcfSSatish Balay         }
437a7e14dcfSSatish Balay         break;
438a7e14dcfSSatish Balay       }
439a7e14dcfSSatish Balay 
440a7e14dcfSSatish Balay       ++nlsP->newt;
441a7e14dcfSSatish Balay       stepType = NLS_NEWTON;
442a7e14dcfSSatish Balay     }
443a7e14dcfSSatish Balay 
444a7e14dcfSSatish Balay     /* Perform the linesearch */
445a7e14dcfSSatish Balay     fold = f;
446a7e14dcfSSatish Balay     ierr = VecCopy(tao->solution, nlsP->Xold);CHKERRQ(ierr);
447a7e14dcfSSatish Balay     ierr = VecCopy(tao->gradient, nlsP->Gold);CHKERRQ(ierr);
448a7e14dcfSSatish Balay 
449a7e14dcfSSatish Balay     ierr = TaoLineSearchApply(tao->linesearch, tao->solution, &f, tao->gradient, nlsP->D, &step, &ls_reason);CHKERRQ(ierr);
450a7e14dcfSSatish Balay     ierr = TaoAddLineSearchCounts(tao);CHKERRQ(ierr);
451a7e14dcfSSatish Balay 
45287f595a5SBarry Smith     while (ls_reason != TAOLINESEARCH_SUCCESS && ls_reason != TAOLINESEARCH_SUCCESS_USER && stepType != NLS_GRADIENT) {      /* Linesearch failed */
453a7e14dcfSSatish Balay       f = fold;
454a7e14dcfSSatish Balay       ierr = VecCopy(nlsP->Xold, tao->solution);CHKERRQ(ierr);
455a7e14dcfSSatish Balay       ierr = VecCopy(nlsP->Gold, tao->gradient);CHKERRQ(ierr);
456a7e14dcfSSatish Balay 
457a7e14dcfSSatish Balay       switch(stepType) {
458a7e14dcfSSatish Balay       case NLS_NEWTON:
459a7e14dcfSSatish Balay         /* Failed to obtain acceptable iterate with Newton 1step
460a7e14dcfSSatish Balay            Update the perturbation for next time */
461a7e14dcfSSatish Balay         if (pert <= 0.0) {
462a7e14dcfSSatish Balay           /* Initialize the perturbation */
463a7e14dcfSSatish Balay           pert = PetscMin(nlsP->imax, PetscMax(nlsP->imin, nlsP->imfac * gnorm));
4641daac38eSTodd Munson           if (is_gltr) {
46505de396fSBarry Smith             ierr = KSPGLTRGetMinEig(tao->ksp,&e_min);CHKERRQ(ierr);
466a7e14dcfSSatish Balay             pert = PetscMax(pert, -e_min);
467a7e14dcfSSatish Balay           }
46887f595a5SBarry Smith         } else {
469a7e14dcfSSatish Balay           /* Increase the perturbation */
470a7e14dcfSSatish Balay           pert = PetscMin(nlsP->pmax, PetscMax(nlsP->pgfac * pert, nlsP->pmgfac * gnorm));
471a7e14dcfSSatish Balay         }
472a7e14dcfSSatish Balay 
4730c51296cSAlp Dener         if (!nlsP->bfgs_pre) {
474a7e14dcfSSatish Balay           /* We don't have the bfgs matrix around and being updated
475a7e14dcfSSatish Balay              Must use gradient direction in this case */
476a7e14dcfSSatish Balay           ierr = VecCopy(tao->gradient, nlsP->D);CHKERRQ(ierr);
477a7e14dcfSSatish Balay           ++nlsP->grad;
478a7e14dcfSSatish Balay           stepType = NLS_GRADIENT;
47987f595a5SBarry Smith         } else {
480a7e14dcfSSatish Balay           /* Attempt to use the BFGS direction */
481cd929ea3SAlp Dener           ierr = MatSolve(nlsP->M, tao->gradient, nlsP->D);CHKERRQ(ierr);
482a7e14dcfSSatish Balay           /* Check for success (descent direction) */
483a7e14dcfSSatish Balay           ierr = VecDot(tao->solution, nlsP->D, &gdx);CHKERRQ(ierr);
484a7e14dcfSSatish Balay           if ((gdx <= 0) || PetscIsInfOrNanReal(gdx)) {
485a7e14dcfSSatish Balay             /* BFGS direction is not descent or direction produced not a number
486a7e14dcfSSatish Balay                We can assert bfgsUpdates > 1 in this case
487a7e14dcfSSatish Balay                Use steepest descent direction (scaled) */
488cd929ea3SAlp Dener             ierr = MatLMVMReset(nlsP->M, PETSC_FALSE);CHKERRQ(ierr);
489a7e14dcfSSatish Balay             ierr = MatLMVMUpdate(nlsP->M, tao->solution, tao->gradient);CHKERRQ(ierr);
490cd929ea3SAlp Dener             ierr = MatSolve(nlsP->M, tao->gradient, nlsP->D);CHKERRQ(ierr);
491a7e14dcfSSatish Balay 
492a7e14dcfSSatish Balay             bfgsUpdates = 1;
4930c51296cSAlp Dener             ++nlsP->grad;
4940c51296cSAlp Dener             stepType = NLS_GRADIENT;
49587f595a5SBarry Smith           } else {
496a7e14dcfSSatish Balay             if (1 == bfgsUpdates) {
497a7e14dcfSSatish Balay               /* The first BFGS direction is always the scaled gradient */
4980c51296cSAlp Dener               ++nlsP->grad;
4990c51296cSAlp Dener               stepType = NLS_GRADIENT;
50087f595a5SBarry Smith             } else {
501a7e14dcfSSatish Balay               ++nlsP->bfgs;
502a7e14dcfSSatish Balay               stepType = NLS_BFGS;
503a7e14dcfSSatish Balay             }
504a7e14dcfSSatish Balay           }
505a7e14dcfSSatish Balay         }
506a7e14dcfSSatish Balay         break;
507a7e14dcfSSatish Balay 
508a7e14dcfSSatish Balay       case NLS_BFGS:
509a7e14dcfSSatish Balay         /* Can only enter if pc_type == NLS_PC_BFGS
510a7e14dcfSSatish Balay            Failed to obtain acceptable iterate with BFGS step
511a7e14dcfSSatish Balay            Attempt to use the scaled gradient direction */
512cd929ea3SAlp Dener         ierr = MatLMVMReset(nlsP->M, PETSC_FALSE);CHKERRQ(ierr);
513a7e14dcfSSatish Balay         ierr = MatLMVMUpdate(nlsP->M, tao->solution, tao->gradient);CHKERRQ(ierr);
514cd929ea3SAlp Dener         ierr = MatSolve(nlsP->M, tao->gradient, nlsP->D);CHKERRQ(ierr);
515a7e14dcfSSatish Balay 
516a7e14dcfSSatish Balay         bfgsUpdates = 1;
517a7e14dcfSSatish Balay         ++nlsP->grad;
518a7e14dcfSSatish Balay         stepType = NLS_GRADIENT;
519a7e14dcfSSatish Balay         break;
520a7e14dcfSSatish Balay       }
521a7e14dcfSSatish Balay       ierr = VecScale(nlsP->D, -1.0);CHKERRQ(ierr);
522a7e14dcfSSatish Balay 
523a7e14dcfSSatish Balay       ierr = TaoLineSearchApply(tao->linesearch, tao->solution, &f, tao->gradient, nlsP->D, &step, &ls_reason);CHKERRQ(ierr);
524a7e14dcfSSatish Balay       ierr = TaoLineSearchGetFullStepObjective(tao->linesearch, &f_full);CHKERRQ(ierr);
525a7e14dcfSSatish Balay       ierr = TaoAddLineSearchCounts(tao);CHKERRQ(ierr);
526a7e14dcfSSatish Balay     }
527a7e14dcfSSatish Balay 
52887f595a5SBarry Smith     if (ls_reason != TAOLINESEARCH_SUCCESS && ls_reason != TAOLINESEARCH_SUCCESS_USER) {
529a7e14dcfSSatish Balay       /* Failed to find an improving point */
530a7e14dcfSSatish Balay       f = fold;
531a7e14dcfSSatish Balay       ierr = VecCopy(nlsP->Xold, tao->solution);CHKERRQ(ierr);
532a7e14dcfSSatish Balay       ierr = VecCopy(nlsP->Gold, tao->gradient);CHKERRQ(ierr);
533a7e14dcfSSatish Balay       step = 0.0;
534a7e14dcfSSatish Balay       tao->reason = TAO_DIVERGED_LS_FAILURE;
535a7e14dcfSSatish Balay       break;
536a7e14dcfSSatish Balay     }
537a7e14dcfSSatish Balay 
538a7e14dcfSSatish Balay     /* Update trust region radius */
5391daac38eSTodd Munson     if (is_nash || is_stcg || is_gltr) {
540a7e14dcfSSatish Balay       switch(nlsP->update_type) {
541a7e14dcfSSatish Balay       case NLS_UPDATE_STEP:
542a7e14dcfSSatish Balay         if (stepType == NLS_NEWTON) {
543a7e14dcfSSatish Balay           if (step < nlsP->nu1) {
544a7e14dcfSSatish Balay             /* Very bad step taken; reduce radius */
545a7e14dcfSSatish Balay             tao->trust = nlsP->omega1 * PetscMin(norm_d, tao->trust);
54687f595a5SBarry Smith           } else if (step < nlsP->nu2) {
547a7e14dcfSSatish Balay             /* Reasonably bad step taken; reduce radius */
548a7e14dcfSSatish Balay             tao->trust = nlsP->omega2 * PetscMin(norm_d, tao->trust);
54987f595a5SBarry Smith           } else if (step < nlsP->nu3) {
550a7e14dcfSSatish Balay             /*  Reasonable step was taken; leave radius alone */
551a7e14dcfSSatish Balay             if (nlsP->omega3 < 1.0) {
552a7e14dcfSSatish Balay               tao->trust = nlsP->omega3 * PetscMin(norm_d, tao->trust);
55387f595a5SBarry Smith             } else if (nlsP->omega3 > 1.0) {
554a7e14dcfSSatish Balay               tao->trust = PetscMax(nlsP->omega3 * norm_d, tao->trust);
555a7e14dcfSSatish Balay             }
55687f595a5SBarry Smith           } else if (step < nlsP->nu4) {
557a7e14dcfSSatish Balay             /*  Full step taken; increase the radius */
558a7e14dcfSSatish Balay             tao->trust = PetscMax(nlsP->omega4 * norm_d, tao->trust);
55987f595a5SBarry Smith           } else {
560a7e14dcfSSatish Balay             /*  More than full step taken; increase the radius */
561a7e14dcfSSatish Balay             tao->trust = PetscMax(nlsP->omega5 * norm_d, tao->trust);
562a7e14dcfSSatish Balay           }
56387f595a5SBarry Smith         } else {
564a7e14dcfSSatish Balay           /*  Newton step was not good; reduce the radius */
565a7e14dcfSSatish Balay           tao->trust = nlsP->omega1 * PetscMin(norm_d, tao->trust);
566a7e14dcfSSatish Balay         }
567a7e14dcfSSatish Balay         break;
568a7e14dcfSSatish Balay 
569a7e14dcfSSatish Balay       case NLS_UPDATE_REDUCTION:
570a7e14dcfSSatish Balay         if (stepType == NLS_NEWTON) {
571a7e14dcfSSatish Balay           /*  Get predicted reduction */
572a7e14dcfSSatish Balay 
573ba7fe8fbSTodd Munson           ierr = KSPCGGetObjFcn(tao->ksp,&prered);CHKERRQ(ierr);
574a7e14dcfSSatish Balay           if (prered >= 0.0) {
575a7e14dcfSSatish Balay             /*  The predicted reduction has the wrong sign.  This cannot */
576a7e14dcfSSatish Balay             /*  happen in infinite precision arithmetic.  Step should */
577a7e14dcfSSatish Balay             /*  be rejected! */
578a7e14dcfSSatish Balay             tao->trust = nlsP->alpha1 * PetscMin(tao->trust, norm_d);
57987f595a5SBarry Smith           } else {
580a7e14dcfSSatish Balay             if (PetscIsInfOrNanReal(f_full)) {
581a7e14dcfSSatish Balay               tao->trust = nlsP->alpha1 * PetscMin(tao->trust, norm_d);
58287f595a5SBarry Smith             } else {
583a7e14dcfSSatish Balay               /*  Compute and actual reduction */
584a7e14dcfSSatish Balay               actred = fold - f_full;
585a7e14dcfSSatish Balay               prered = -prered;
586a7e14dcfSSatish Balay               if ((PetscAbsScalar(actred) <= nlsP->epsilon) &&
587a7e14dcfSSatish Balay                   (PetscAbsScalar(prered) <= nlsP->epsilon)) {
588a7e14dcfSSatish Balay                 kappa = 1.0;
58987f595a5SBarry Smith               } else {
590a7e14dcfSSatish Balay                 kappa = actred / prered;
591a7e14dcfSSatish Balay               }
592a7e14dcfSSatish Balay 
593a7e14dcfSSatish Balay               /*  Accept of reject the step and update radius */
594a7e14dcfSSatish Balay               if (kappa < nlsP->eta1) {
595a7e14dcfSSatish Balay                 /*  Very bad step */
596a7e14dcfSSatish Balay                 tao->trust = nlsP->alpha1 * PetscMin(tao->trust, norm_d);
59787f595a5SBarry Smith               } else if (kappa < nlsP->eta2) {
598a7e14dcfSSatish Balay                 /*  Marginal bad step */
599a7e14dcfSSatish Balay                 tao->trust = nlsP->alpha2 * PetscMin(tao->trust, norm_d);
60087f595a5SBarry Smith               } else if (kappa < nlsP->eta3) {
601a7e14dcfSSatish Balay                 /*  Reasonable step */
602a7e14dcfSSatish Balay                 if (nlsP->alpha3 < 1.0) {
603a7e14dcfSSatish Balay                   tao->trust = nlsP->alpha3 * PetscMin(norm_d, tao->trust);
60487f595a5SBarry Smith                 } else if (nlsP->alpha3 > 1.0) {
605a7e14dcfSSatish Balay                   tao->trust = PetscMax(nlsP->alpha3 * norm_d, tao->trust);
606a7e14dcfSSatish Balay                 }
60787f595a5SBarry Smith               } else if (kappa < nlsP->eta4) {
608a7e14dcfSSatish Balay                 /*  Good step */
609a7e14dcfSSatish Balay                 tao->trust = PetscMax(nlsP->alpha4 * norm_d, tao->trust);
61087f595a5SBarry Smith               } else {
611a7e14dcfSSatish Balay                 /*  Very good step */
612a7e14dcfSSatish Balay                 tao->trust = PetscMax(nlsP->alpha5 * norm_d, tao->trust);
613a7e14dcfSSatish Balay               }
614a7e14dcfSSatish Balay             }
615a7e14dcfSSatish Balay           }
61687f595a5SBarry Smith         } else {
617a7e14dcfSSatish Balay           /*  Newton step was not good; reduce the radius */
618a7e14dcfSSatish Balay           tao->trust = nlsP->alpha1 * PetscMin(norm_d, tao->trust);
619a7e14dcfSSatish Balay         }
620a7e14dcfSSatish Balay         break;
621a7e14dcfSSatish Balay 
622a7e14dcfSSatish Balay       default:
623a7e14dcfSSatish Balay         if (stepType == NLS_NEWTON) {
624ba7fe8fbSTodd Munson           ierr = KSPCGGetObjFcn(tao->ksp,&prered);CHKERRQ(ierr);
625a7e14dcfSSatish Balay           if (prered >= 0.0) {
626a7e14dcfSSatish Balay             /*  The predicted reduction has the wrong sign.  This cannot */
627a7e14dcfSSatish Balay             /*  happen in infinite precision arithmetic.  Step should */
628a7e14dcfSSatish Balay             /*  be rejected! */
629a7e14dcfSSatish Balay             tao->trust = nlsP->gamma1 * PetscMin(tao->trust, norm_d);
63087f595a5SBarry Smith           } else {
631a7e14dcfSSatish Balay             if (PetscIsInfOrNanReal(f_full)) {
632a7e14dcfSSatish Balay               tao->trust = nlsP->gamma1 * PetscMin(tao->trust, norm_d);
63387f595a5SBarry Smith             } else {
634a7e14dcfSSatish Balay               actred = fold - f_full;
635a7e14dcfSSatish Balay               prered = -prered;
63687f595a5SBarry Smith               if ((PetscAbsScalar(actred) <= nlsP->epsilon) && (PetscAbsScalar(prered) <= nlsP->epsilon)) {
637a7e14dcfSSatish Balay                 kappa = 1.0;
63887f595a5SBarry Smith               } else {
639a7e14dcfSSatish Balay                 kappa = actred / prered;
640a7e14dcfSSatish Balay               }
641a7e14dcfSSatish Balay 
642a7e14dcfSSatish Balay               tau_1 = nlsP->theta * gdx / (nlsP->theta * gdx - (1.0 - nlsP->theta) * prered + actred);
643a7e14dcfSSatish Balay               tau_2 = nlsP->theta * gdx / (nlsP->theta * gdx + (1.0 + nlsP->theta) * prered - actred);
644a7e14dcfSSatish Balay               tau_min = PetscMin(tau_1, tau_2);
645a7e14dcfSSatish Balay               tau_max = PetscMax(tau_1, tau_2);
646a7e14dcfSSatish Balay 
647a7e14dcfSSatish Balay               if (kappa >= 1.0 - nlsP->mu1) {
648a7e14dcfSSatish Balay                 /*  Great agreement */
649a7e14dcfSSatish Balay                 if (tau_max < 1.0) {
650a7e14dcfSSatish Balay                   tao->trust = PetscMax(tao->trust, nlsP->gamma3 * norm_d);
65187f595a5SBarry Smith                 } else if (tau_max > nlsP->gamma4) {
652a7e14dcfSSatish Balay                   tao->trust = PetscMax(tao->trust, nlsP->gamma4 * norm_d);
65387f595a5SBarry Smith                 } else {
654a7e14dcfSSatish Balay                   tao->trust = PetscMax(tao->trust, tau_max * norm_d);
655a7e14dcfSSatish Balay                 }
65687f595a5SBarry Smith               } else if (kappa >= 1.0 - nlsP->mu2) {
657a7e14dcfSSatish Balay                 /*  Good agreement */
658a7e14dcfSSatish Balay 
659a7e14dcfSSatish Balay                 if (tau_max < nlsP->gamma2) {
660a7e14dcfSSatish Balay                   tao->trust = nlsP->gamma2 * PetscMin(tao->trust, norm_d);
66187f595a5SBarry Smith                 } else if (tau_max > nlsP->gamma3) {
662a7e14dcfSSatish Balay                   tao->trust = PetscMax(tao->trust, nlsP->gamma3 * norm_d);
66387f595a5SBarry Smith                 } else if (tau_max < 1.0) {
664a7e14dcfSSatish Balay                   tao->trust = tau_max * PetscMin(tao->trust, norm_d);
66587f595a5SBarry Smith                 } else {
666a7e14dcfSSatish Balay                   tao->trust = PetscMax(tao->trust, tau_max * norm_d);
667a7e14dcfSSatish Balay                 }
66887f595a5SBarry Smith               } else {
669a7e14dcfSSatish Balay                 /*  Not good agreement */
670a7e14dcfSSatish Balay                 if (tau_min > 1.0) {
671a7e14dcfSSatish Balay                   tao->trust = nlsP->gamma2 * PetscMin(tao->trust, norm_d);
67287f595a5SBarry Smith                 } else if (tau_max < nlsP->gamma1) {
673a7e14dcfSSatish Balay                   tao->trust = nlsP->gamma1 * PetscMin(tao->trust, norm_d);
67487f595a5SBarry Smith                 } else if ((tau_min < nlsP->gamma1) && (tau_max >= 1.0)) {
675a7e14dcfSSatish Balay                   tao->trust = nlsP->gamma1 * PetscMin(tao->trust, norm_d);
67687f595a5SBarry Smith                 } else if ((tau_1 >= nlsP->gamma1) && (tau_1 < 1.0) && ((tau_2 < nlsP->gamma1) || (tau_2 >= 1.0))) {
677a7e14dcfSSatish Balay                   tao->trust = tau_1 * PetscMin(tao->trust, norm_d);
67887f595a5SBarry Smith                 } else if ((tau_2 >= nlsP->gamma1) && (tau_2 < 1.0) && ((tau_1 < nlsP->gamma1) || (tau_2 >= 1.0))) {
679a7e14dcfSSatish Balay                   tao->trust = tau_2 * PetscMin(tao->trust, norm_d);
68087f595a5SBarry Smith                 } else {
681a7e14dcfSSatish Balay                   tao->trust = tau_max * PetscMin(tao->trust, norm_d);
682a7e14dcfSSatish Balay                 }
683a7e14dcfSSatish Balay               }
684a7e14dcfSSatish Balay             }
685a7e14dcfSSatish Balay           }
68687f595a5SBarry Smith         } else {
687a7e14dcfSSatish Balay           /*  Newton step was not good; reduce the radius */
688a7e14dcfSSatish Balay           tao->trust = nlsP->gamma1 * PetscMin(norm_d, tao->trust);
689a7e14dcfSSatish Balay         }
690a7e14dcfSSatish Balay         break;
691a7e14dcfSSatish Balay       }
692a7e14dcfSSatish Balay 
693a7e14dcfSSatish Balay       /*  The radius may have been increased; modify if it is too large */
694a7e14dcfSSatish Balay       tao->trust = PetscMin(tao->trust, nlsP->max_radius);
695a7e14dcfSSatish Balay     }
696a7e14dcfSSatish Balay 
697a7e14dcfSSatish Balay     /*  Check for termination */
698a9603a14SPatrick Farrell     ierr = TaoGradientNorm(tao, tao->gradient,NORM_2,&gnorm);CHKERRQ(ierr);
699*3c859ba3SBarry Smith     PetscCheck(!PetscIsInfOrNanReal(f) && !PetscIsInfOrNanReal(gnorm),PetscObjectComm((PetscObject)tao),PETSC_ERR_USER,"User provided compute function generated Not-a-Number");
700a7e14dcfSSatish Balay     needH = 1;
7013ecd9318SAlp Dener     ierr = TaoLogConvergenceHistory(tao,f,gnorm,0.0,tao->ksp_its);CHKERRQ(ierr);
7023ecd9318SAlp Dener     ierr = TaoMonitor(tao,tao->niter,f,gnorm,0.0,step);CHKERRQ(ierr);
7033ecd9318SAlp Dener     ierr = (*tao->ops->convergencetest)(tao,tao->cnvP);CHKERRQ(ierr);
704a7e14dcfSSatish Balay   }
705a7e14dcfSSatish Balay   PetscFunctionReturn(0);
706a7e14dcfSSatish Balay }
707a7e14dcfSSatish Balay 
708a7e14dcfSSatish Balay /* ---------------------------------------------------------- */
709441846f8SBarry Smith static PetscErrorCode TaoSetUp_NLS(Tao tao)
710a7e14dcfSSatish Balay {
711a7e14dcfSSatish Balay   TAO_NLS        *nlsP = (TAO_NLS *)tao->data;
712a7e14dcfSSatish Balay   PetscErrorCode ierr;
713a7e14dcfSSatish Balay 
714a7e14dcfSSatish Balay   PetscFunctionBegin;
715a7e14dcfSSatish Balay   if (!tao->gradient) {ierr = VecDuplicate(tao->solution,&tao->gradient);CHKERRQ(ierr);}
716a7e14dcfSSatish Balay   if (!tao->stepdirection) {ierr = VecDuplicate(tao->solution,&tao->stepdirection);CHKERRQ(ierr);}
717a7e14dcfSSatish Balay   if (!nlsP->W) {ierr = VecDuplicate(tao->solution,&nlsP->W);CHKERRQ(ierr);}
718a7e14dcfSSatish Balay   if (!nlsP->D) {ierr = VecDuplicate(tao->solution,&nlsP->D);CHKERRQ(ierr);}
719a7e14dcfSSatish Balay   if (!nlsP->Xold) {ierr = VecDuplicate(tao->solution,&nlsP->Xold);CHKERRQ(ierr);}
720a7e14dcfSSatish Balay   if (!nlsP->Gold) {ierr = VecDuplicate(tao->solution,&nlsP->Gold);CHKERRQ(ierr);}
72183c8fe1dSLisandro Dalcin   nlsP->M = NULL;
72283c8fe1dSLisandro Dalcin   nlsP->bfgs_pre = NULL;
723a7e14dcfSSatish Balay   PetscFunctionReturn(0);
724a7e14dcfSSatish Balay }
725a7e14dcfSSatish Balay 
726a7e14dcfSSatish Balay /*------------------------------------------------------------*/
727441846f8SBarry Smith static PetscErrorCode TaoDestroy_NLS(Tao tao)
728a7e14dcfSSatish Balay {
729a7e14dcfSSatish Balay   TAO_NLS        *nlsP = (TAO_NLS *)tao->data;
730a7e14dcfSSatish Balay   PetscErrorCode ierr;
731a7e14dcfSSatish Balay 
732a7e14dcfSSatish Balay   PetscFunctionBegin;
733a7e14dcfSSatish Balay   if (tao->setupcalled) {
734a7e14dcfSSatish Balay     ierr = VecDestroy(&nlsP->D);CHKERRQ(ierr);
735a7e14dcfSSatish Balay     ierr = VecDestroy(&nlsP->W);CHKERRQ(ierr);
736a7e14dcfSSatish Balay     ierr = VecDestroy(&nlsP->Xold);CHKERRQ(ierr);
737a7e14dcfSSatish Balay     ierr = VecDestroy(&nlsP->Gold);CHKERRQ(ierr);
738a7e14dcfSSatish Balay   }
739a7e14dcfSSatish Balay   ierr = PetscFree(tao->data);CHKERRQ(ierr);
740a7e14dcfSSatish Balay   PetscFunctionReturn(0);
741a7e14dcfSSatish Balay }
742a7e14dcfSSatish Balay 
743a7e14dcfSSatish Balay /*------------------------------------------------------------*/
7444416b707SBarry Smith static PetscErrorCode TaoSetFromOptions_NLS(PetscOptionItems *PetscOptionsObject,Tao tao)
745a7e14dcfSSatish Balay {
746a7e14dcfSSatish Balay   TAO_NLS        *nlsP = (TAO_NLS *)tao->data;
747a7e14dcfSSatish Balay   PetscErrorCode ierr;
748a7e14dcfSSatish Balay 
749a7e14dcfSSatish Balay   PetscFunctionBegin;
7501a1499c8SBarry Smith   ierr = PetscOptionsHead(PetscOptionsObject,"Newton line search method for unconstrained optimization");CHKERRQ(ierr);
75183c8fe1dSLisandro Dalcin   ierr = PetscOptionsEList("-tao_nls_init_type", "radius initialization type", "", NLS_INIT, NLS_INIT_TYPES, NLS_INIT[nlsP->init_type], &nlsP->init_type, NULL);CHKERRQ(ierr);
75283c8fe1dSLisandro Dalcin   ierr = PetscOptionsEList("-tao_nls_update_type", "radius update type", "", NLS_UPDATE, NLS_UPDATE_TYPES, NLS_UPDATE[nlsP->update_type], &nlsP->update_type, NULL);CHKERRQ(ierr);
75394ae4db5SBarry Smith   ierr = PetscOptionsReal("-tao_nls_sval", "perturbation starting value", "", nlsP->sval, &nlsP->sval,NULL);CHKERRQ(ierr);
75494ae4db5SBarry Smith   ierr = PetscOptionsReal("-tao_nls_imin", "minimum initial perturbation", "", nlsP->imin, &nlsP->imin,NULL);CHKERRQ(ierr);
75594ae4db5SBarry Smith   ierr = PetscOptionsReal("-tao_nls_imax", "maximum initial perturbation", "", nlsP->imax, &nlsP->imax,NULL);CHKERRQ(ierr);
75694ae4db5SBarry Smith   ierr = PetscOptionsReal("-tao_nls_imfac", "initial merit factor", "", nlsP->imfac, &nlsP->imfac,NULL);CHKERRQ(ierr);
75794ae4db5SBarry Smith   ierr = PetscOptionsReal("-tao_nls_pmin", "minimum perturbation", "", nlsP->pmin, &nlsP->pmin,NULL);CHKERRQ(ierr);
75894ae4db5SBarry Smith   ierr = PetscOptionsReal("-tao_nls_pmax", "maximum perturbation", "", nlsP->pmax, &nlsP->pmax,NULL);CHKERRQ(ierr);
75994ae4db5SBarry Smith   ierr = PetscOptionsReal("-tao_nls_pgfac", "growth factor", "", nlsP->pgfac, &nlsP->pgfac,NULL);CHKERRQ(ierr);
76094ae4db5SBarry Smith   ierr = PetscOptionsReal("-tao_nls_psfac", "shrink factor", "", nlsP->psfac, &nlsP->psfac,NULL);CHKERRQ(ierr);
76194ae4db5SBarry Smith   ierr = PetscOptionsReal("-tao_nls_pmgfac", "merit growth factor", "", nlsP->pmgfac, &nlsP->pmgfac,NULL);CHKERRQ(ierr);
76294ae4db5SBarry Smith   ierr = PetscOptionsReal("-tao_nls_pmsfac", "merit shrink factor", "", nlsP->pmsfac, &nlsP->pmsfac,NULL);CHKERRQ(ierr);
76394ae4db5SBarry Smith   ierr = PetscOptionsReal("-tao_nls_eta1", "poor steplength; reduce radius", "", nlsP->eta1, &nlsP->eta1,NULL);CHKERRQ(ierr);
76494ae4db5SBarry Smith   ierr = PetscOptionsReal("-tao_nls_eta2", "reasonable steplength; leave radius alone", "", nlsP->eta2, &nlsP->eta2,NULL);CHKERRQ(ierr);
76594ae4db5SBarry Smith   ierr = PetscOptionsReal("-tao_nls_eta3", "good steplength; increase radius", "", nlsP->eta3, &nlsP->eta3,NULL);CHKERRQ(ierr);
76694ae4db5SBarry Smith   ierr = PetscOptionsReal("-tao_nls_eta4", "excellent steplength; greatly increase radius", "", nlsP->eta4, &nlsP->eta4,NULL);CHKERRQ(ierr);
76794ae4db5SBarry Smith   ierr = PetscOptionsReal("-tao_nls_alpha1", "", "", nlsP->alpha1, &nlsP->alpha1,NULL);CHKERRQ(ierr);
76894ae4db5SBarry Smith   ierr = PetscOptionsReal("-tao_nls_alpha2", "", "", nlsP->alpha2, &nlsP->alpha2,NULL);CHKERRQ(ierr);
76994ae4db5SBarry Smith   ierr = PetscOptionsReal("-tao_nls_alpha3", "", "", nlsP->alpha3, &nlsP->alpha3,NULL);CHKERRQ(ierr);
77094ae4db5SBarry Smith   ierr = PetscOptionsReal("-tao_nls_alpha4", "", "", nlsP->alpha4, &nlsP->alpha4,NULL);CHKERRQ(ierr);
77194ae4db5SBarry Smith   ierr = PetscOptionsReal("-tao_nls_alpha5", "", "", nlsP->alpha5, &nlsP->alpha5,NULL);CHKERRQ(ierr);
77294ae4db5SBarry Smith   ierr = PetscOptionsReal("-tao_nls_nu1", "poor steplength; reduce radius", "", nlsP->nu1, &nlsP->nu1,NULL);CHKERRQ(ierr);
77394ae4db5SBarry Smith   ierr = PetscOptionsReal("-tao_nls_nu2", "reasonable steplength; leave radius alone", "", nlsP->nu2, &nlsP->nu2,NULL);CHKERRQ(ierr);
77494ae4db5SBarry Smith   ierr = PetscOptionsReal("-tao_nls_nu3", "good steplength; increase radius", "", nlsP->nu3, &nlsP->nu3,NULL);CHKERRQ(ierr);
77594ae4db5SBarry Smith   ierr = PetscOptionsReal("-tao_nls_nu4", "excellent steplength; greatly increase radius", "", nlsP->nu4, &nlsP->nu4,NULL);CHKERRQ(ierr);
77694ae4db5SBarry Smith   ierr = PetscOptionsReal("-tao_nls_omega1", "", "", nlsP->omega1, &nlsP->omega1,NULL);CHKERRQ(ierr);
77794ae4db5SBarry Smith   ierr = PetscOptionsReal("-tao_nls_omega2", "", "", nlsP->omega2, &nlsP->omega2,NULL);CHKERRQ(ierr);
77894ae4db5SBarry Smith   ierr = PetscOptionsReal("-tao_nls_omega3", "", "", nlsP->omega3, &nlsP->omega3,NULL);CHKERRQ(ierr);
77994ae4db5SBarry Smith   ierr = PetscOptionsReal("-tao_nls_omega4", "", "", nlsP->omega4, &nlsP->omega4,NULL);CHKERRQ(ierr);
78094ae4db5SBarry Smith   ierr = PetscOptionsReal("-tao_nls_omega5", "", "", nlsP->omega5, &nlsP->omega5,NULL);CHKERRQ(ierr);
78194ae4db5SBarry Smith   ierr = PetscOptionsReal("-tao_nls_mu1_i", "", "", nlsP->mu1_i, &nlsP->mu1_i,NULL);CHKERRQ(ierr);
78294ae4db5SBarry Smith   ierr = PetscOptionsReal("-tao_nls_mu2_i", "", "", nlsP->mu2_i, &nlsP->mu2_i,NULL);CHKERRQ(ierr);
78394ae4db5SBarry Smith   ierr = PetscOptionsReal("-tao_nls_gamma1_i", "", "", nlsP->gamma1_i, &nlsP->gamma1_i,NULL);CHKERRQ(ierr);
78494ae4db5SBarry Smith   ierr = PetscOptionsReal("-tao_nls_gamma2_i", "", "", nlsP->gamma2_i, &nlsP->gamma2_i,NULL);CHKERRQ(ierr);
78594ae4db5SBarry Smith   ierr = PetscOptionsReal("-tao_nls_gamma3_i", "", "", nlsP->gamma3_i, &nlsP->gamma3_i,NULL);CHKERRQ(ierr);
78694ae4db5SBarry Smith   ierr = PetscOptionsReal("-tao_nls_gamma4_i", "", "", nlsP->gamma4_i, &nlsP->gamma4_i,NULL);CHKERRQ(ierr);
78794ae4db5SBarry Smith   ierr = PetscOptionsReal("-tao_nls_theta_i", "", "", nlsP->theta_i, &nlsP->theta_i,NULL);CHKERRQ(ierr);
78894ae4db5SBarry Smith   ierr = PetscOptionsReal("-tao_nls_mu1", "", "", nlsP->mu1, &nlsP->mu1,NULL);CHKERRQ(ierr);
78994ae4db5SBarry Smith   ierr = PetscOptionsReal("-tao_nls_mu2", "", "", nlsP->mu2, &nlsP->mu2,NULL);CHKERRQ(ierr);
79094ae4db5SBarry Smith   ierr = PetscOptionsReal("-tao_nls_gamma1", "", "", nlsP->gamma1, &nlsP->gamma1,NULL);CHKERRQ(ierr);
79194ae4db5SBarry Smith   ierr = PetscOptionsReal("-tao_nls_gamma2", "", "", nlsP->gamma2, &nlsP->gamma2,NULL);CHKERRQ(ierr);
79294ae4db5SBarry Smith   ierr = PetscOptionsReal("-tao_nls_gamma3", "", "", nlsP->gamma3, &nlsP->gamma3,NULL);CHKERRQ(ierr);
79394ae4db5SBarry Smith   ierr = PetscOptionsReal("-tao_nls_gamma4", "", "", nlsP->gamma4, &nlsP->gamma4,NULL);CHKERRQ(ierr);
79494ae4db5SBarry Smith   ierr = PetscOptionsReal("-tao_nls_theta", "", "", nlsP->theta, &nlsP->theta,NULL);CHKERRQ(ierr);
79594ae4db5SBarry Smith   ierr = PetscOptionsReal("-tao_nls_min_radius", "lower bound on initial radius", "", nlsP->min_radius, &nlsP->min_radius,NULL);CHKERRQ(ierr);
79694ae4db5SBarry Smith   ierr = PetscOptionsReal("-tao_nls_max_radius", "upper bound on radius", "", nlsP->max_radius, &nlsP->max_radius,NULL);CHKERRQ(ierr);
79794ae4db5SBarry Smith   ierr = PetscOptionsReal("-tao_nls_epsilon", "tolerance used when computing actual and predicted reduction", "", nlsP->epsilon, &nlsP->epsilon,NULL);CHKERRQ(ierr);
798a7e14dcfSSatish Balay   ierr = PetscOptionsTail();CHKERRQ(ierr);
799a7e14dcfSSatish Balay   ierr = TaoLineSearchSetFromOptions(tao->linesearch);CHKERRQ(ierr);
800a7e14dcfSSatish Balay   ierr = KSPSetFromOptions(tao->ksp);CHKERRQ(ierr);
801a7e14dcfSSatish Balay   PetscFunctionReturn(0);
802a7e14dcfSSatish Balay }
803a7e14dcfSSatish Balay 
804a7e14dcfSSatish Balay /*------------------------------------------------------------*/
805441846f8SBarry Smith static PetscErrorCode TaoView_NLS(Tao tao, PetscViewer viewer)
806a7e14dcfSSatish Balay {
807a7e14dcfSSatish Balay   TAO_NLS        *nlsP = (TAO_NLS *)tao->data;
808a7e14dcfSSatish Balay   PetscBool      isascii;
809a7e14dcfSSatish Balay   PetscErrorCode ierr;
810a7e14dcfSSatish Balay 
811a7e14dcfSSatish Balay   PetscFunctionBegin;
812a7e14dcfSSatish Balay   ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERASCII,&isascii);CHKERRQ(ierr);
813a7e14dcfSSatish Balay   if (isascii) {
814a7e14dcfSSatish Balay     ierr = PetscViewerASCIIPushTab(viewer);CHKERRQ(ierr);
815a7e14dcfSSatish Balay     ierr = PetscViewerASCIIPrintf(viewer, "Newton steps: %D\n", nlsP->newt);CHKERRQ(ierr);
816a7e14dcfSSatish Balay     ierr = PetscViewerASCIIPrintf(viewer, "BFGS steps: %D\n", nlsP->bfgs);CHKERRQ(ierr);
817a7e14dcfSSatish Balay     ierr = PetscViewerASCIIPrintf(viewer, "Gradient steps: %D\n", nlsP->grad);CHKERRQ(ierr);
818a7e14dcfSSatish Balay 
819a7e14dcfSSatish Balay     ierr = PetscViewerASCIIPrintf(viewer, "nls ksp atol: %D\n", nlsP->ksp_atol);CHKERRQ(ierr);
820a7e14dcfSSatish Balay     ierr = PetscViewerASCIIPrintf(viewer, "nls ksp rtol: %D\n", nlsP->ksp_rtol);CHKERRQ(ierr);
821a7e14dcfSSatish Balay     ierr = PetscViewerASCIIPrintf(viewer, "nls ksp ctol: %D\n", nlsP->ksp_ctol);CHKERRQ(ierr);
822a7e14dcfSSatish Balay     ierr = PetscViewerASCIIPrintf(viewer, "nls ksp negc: %D\n", nlsP->ksp_negc);CHKERRQ(ierr);
823a7e14dcfSSatish Balay     ierr = PetscViewerASCIIPrintf(viewer, "nls ksp dtol: %D\n", nlsP->ksp_dtol);CHKERRQ(ierr);
824a7e14dcfSSatish Balay     ierr = PetscViewerASCIIPrintf(viewer, "nls ksp iter: %D\n", nlsP->ksp_iter);CHKERRQ(ierr);
825a7e14dcfSSatish Balay     ierr = PetscViewerASCIIPrintf(viewer, "nls ksp othr: %D\n", nlsP->ksp_othr);CHKERRQ(ierr);
826a7e14dcfSSatish Balay     ierr = PetscViewerASCIIPopTab(viewer);CHKERRQ(ierr);
827a7e14dcfSSatish Balay   }
828a7e14dcfSSatish Balay   PetscFunctionReturn(0);
829a7e14dcfSSatish Balay }
830a7e14dcfSSatish Balay 
831a7e14dcfSSatish Balay /* ---------------------------------------------------------- */
8324aa34175SJason Sarich /*MC
8334aa34175SJason Sarich   TAONLS - Newton's method with linesearch for unconstrained minimization.
8344aa34175SJason Sarich   At each iteration, the Newton line search method solves the symmetric
8354aa34175SJason Sarich   system of equations to obtain the step diretion dk:
8364aa34175SJason Sarich               Hk dk = -gk
8374aa34175SJason Sarich   a More-Thuente line search is applied on the direction dk to approximately
8384aa34175SJason Sarich   solve
8394aa34175SJason Sarich               min_t f(xk + t d_k)
8404aa34175SJason Sarich 
8414aa34175SJason Sarich     Options Database Keys:
8429d0a60b2SAlp Dener + -tao_nls_init_type - "constant","direction","interpolation"
8434aa34175SJason Sarich . -tao_nls_update_type - "step","direction","interpolation"
8444aa34175SJason Sarich . -tao_nls_sval - perturbation starting value
8454aa34175SJason Sarich . -tao_nls_imin - minimum initial perturbation
8464aa34175SJason Sarich . -tao_nls_imax - maximum initial perturbation
8474aa34175SJason Sarich . -tao_nls_pmin - minimum perturbation
8484aa34175SJason Sarich . -tao_nls_pmax - maximum perturbation
8494aa34175SJason Sarich . -tao_nls_pgfac - growth factor
8504aa34175SJason Sarich . -tao_nls_psfac - shrink factor
8514aa34175SJason Sarich . -tao_nls_imfac - initial merit factor
8524aa34175SJason Sarich . -tao_nls_pmgfac - merit growth factor
8534aa34175SJason Sarich . -tao_nls_pmsfac - merit shrink factor
8544aa34175SJason Sarich . -tao_nls_eta1 - poor steplength; reduce radius
8554aa34175SJason Sarich . -tao_nls_eta2 - reasonable steplength; leave radius
8564aa34175SJason Sarich . -tao_nls_eta3 - good steplength; increase readius
8574aa34175SJason Sarich . -tao_nls_eta4 - excellent steplength; greatly increase radius
8584aa34175SJason Sarich . -tao_nls_alpha1 - alpha1 reduction
8594aa34175SJason Sarich . -tao_nls_alpha2 - alpha2 reduction
8604aa34175SJason Sarich . -tao_nls_alpha3 - alpha3 reduction
8614aa34175SJason Sarich . -tao_nls_alpha4 - alpha4 reduction
8624aa34175SJason Sarich . -tao_nls_alpha - alpha5 reduction
8634aa34175SJason Sarich . -tao_nls_mu1 - mu1 interpolation update
8644aa34175SJason Sarich . -tao_nls_mu2 - mu2 interpolation update
8654aa34175SJason Sarich . -tao_nls_gamma1 - gamma1 interpolation update
8664aa34175SJason Sarich . -tao_nls_gamma2 - gamma2 interpolation update
8674aa34175SJason Sarich . -tao_nls_gamma3 - gamma3 interpolation update
8684aa34175SJason Sarich . -tao_nls_gamma4 - gamma4 interpolation update
8694aa34175SJason Sarich . -tao_nls_theta - theta interpolation update
8704aa34175SJason Sarich . -tao_nls_omega1 - omega1 step update
8714aa34175SJason Sarich . -tao_nls_omega2 - omega2 step update
8724aa34175SJason Sarich . -tao_nls_omega3 - omega3 step update
8734aa34175SJason Sarich . -tao_nls_omega4 - omega4 step update
8744aa34175SJason Sarich . -tao_nls_omega5 - omega5 step update
8751522df2eSJason Sarich . -tao_nls_mu1_i -  mu1 interpolation init factor
8761522df2eSJason Sarich . -tao_nls_mu2_i -  mu2 interpolation init factor
8771522df2eSJason Sarich . -tao_nls_gamma1_i -  gamma1 interpolation init factor
8781522df2eSJason Sarich . -tao_nls_gamma2_i -  gamma2 interpolation init factor
8791522df2eSJason Sarich . -tao_nls_gamma3_i -  gamma3 interpolation init factor
8801522df2eSJason Sarich . -tao_nls_gamma4_i -  gamma4 interpolation init factor
8811522df2eSJason Sarich - -tao_nls_theta_i -  theta interpolation init factor
8821eb8069cSJason Sarich 
8831eb8069cSJason Sarich   Level: beginner
8841522df2eSJason Sarich M*/
8854aa34175SJason Sarich 
886728e0ed0SBarry Smith PETSC_EXTERN PetscErrorCode TaoCreate_NLS(Tao tao)
887a7e14dcfSSatish Balay {
888a7e14dcfSSatish Balay   TAO_NLS        *nlsP;
8898caf6e8cSBarry Smith   const char     *morethuente_type = TAOLINESEARCHMT;
890a7e14dcfSSatish Balay   PetscErrorCode ierr;
891a7e14dcfSSatish Balay 
892a7e14dcfSSatish Balay   PetscFunctionBegin;
8933c9e27cfSGeoffrey Irving   ierr = PetscNewLog(tao,&nlsP);CHKERRQ(ierr);
894a7e14dcfSSatish Balay 
895a7e14dcfSSatish Balay   tao->ops->setup = TaoSetUp_NLS;
896a7e14dcfSSatish Balay   tao->ops->solve = TaoSolve_NLS;
897a7e14dcfSSatish Balay   tao->ops->view = TaoView_NLS;
898a7e14dcfSSatish Balay   tao->ops->setfromoptions = TaoSetFromOptions_NLS;
899a7e14dcfSSatish Balay   tao->ops->destroy = TaoDestroy_NLS;
900a7e14dcfSSatish Balay 
9016552cf8aSJason Sarich   /* Override default settings (unless already changed) */
9026552cf8aSJason Sarich   if (!tao->max_it_changed) tao->max_it = 50;
9036552cf8aSJason Sarich   if (!tao->trust0_changed) tao->trust0 = 100.0;
9046552cf8aSJason Sarich 
905a7e14dcfSSatish Balay   tao->data = (void*)nlsP;
906a7e14dcfSSatish Balay 
907a7e14dcfSSatish Balay   nlsP->sval   = 0.0;
908a7e14dcfSSatish Balay   nlsP->imin   = 1.0e-4;
909a7e14dcfSSatish Balay   nlsP->imax   = 1.0e+2;
910a7e14dcfSSatish Balay   nlsP->imfac  = 1.0e-1;
911a7e14dcfSSatish Balay 
912a7e14dcfSSatish Balay   nlsP->pmin   = 1.0e-12;
913a7e14dcfSSatish Balay   nlsP->pmax   = 1.0e+2;
914a7e14dcfSSatish Balay   nlsP->pgfac  = 1.0e+1;
915a7e14dcfSSatish Balay   nlsP->psfac  = 4.0e-1;
916a7e14dcfSSatish Balay   nlsP->pmgfac = 1.0e-1;
917a7e14dcfSSatish Balay   nlsP->pmsfac = 1.0e-1;
918a7e14dcfSSatish Balay 
919a7e14dcfSSatish Balay   /*  Default values for trust-region radius update based on steplength */
920a7e14dcfSSatish Balay   nlsP->nu1 = 0.25;
921a7e14dcfSSatish Balay   nlsP->nu2 = 0.50;
922a7e14dcfSSatish Balay   nlsP->nu3 = 1.00;
923a7e14dcfSSatish Balay   nlsP->nu4 = 1.25;
924a7e14dcfSSatish Balay 
925a7e14dcfSSatish Balay   nlsP->omega1 = 0.25;
926a7e14dcfSSatish Balay   nlsP->omega2 = 0.50;
927a7e14dcfSSatish Balay   nlsP->omega3 = 1.00;
928a7e14dcfSSatish Balay   nlsP->omega4 = 2.00;
929a7e14dcfSSatish Balay   nlsP->omega5 = 4.00;
930a7e14dcfSSatish Balay 
931a7e14dcfSSatish Balay   /*  Default values for trust-region radius update based on reduction */
932a7e14dcfSSatish Balay   nlsP->eta1 = 1.0e-4;
933a7e14dcfSSatish Balay   nlsP->eta2 = 0.25;
934a7e14dcfSSatish Balay   nlsP->eta3 = 0.50;
935a7e14dcfSSatish Balay   nlsP->eta4 = 0.90;
936a7e14dcfSSatish Balay 
937a7e14dcfSSatish Balay   nlsP->alpha1 = 0.25;
938a7e14dcfSSatish Balay   nlsP->alpha2 = 0.50;
939a7e14dcfSSatish Balay   nlsP->alpha3 = 1.00;
940a7e14dcfSSatish Balay   nlsP->alpha4 = 2.00;
941a7e14dcfSSatish Balay   nlsP->alpha5 = 4.00;
942a7e14dcfSSatish Balay 
943a7e14dcfSSatish Balay   /*  Default values for trust-region radius update based on interpolation */
944a7e14dcfSSatish Balay   nlsP->mu1 = 0.10;
945a7e14dcfSSatish Balay   nlsP->mu2 = 0.50;
946a7e14dcfSSatish Balay 
947a7e14dcfSSatish Balay   nlsP->gamma1 = 0.25;
948a7e14dcfSSatish Balay   nlsP->gamma2 = 0.50;
949a7e14dcfSSatish Balay   nlsP->gamma3 = 2.00;
950a7e14dcfSSatish Balay   nlsP->gamma4 = 4.00;
951a7e14dcfSSatish Balay 
952a7e14dcfSSatish Balay   nlsP->theta = 0.05;
953a7e14dcfSSatish Balay 
954a7e14dcfSSatish Balay   /*  Default values for trust region initialization based on interpolation */
955a7e14dcfSSatish Balay   nlsP->mu1_i = 0.35;
956a7e14dcfSSatish Balay   nlsP->mu2_i = 0.50;
957a7e14dcfSSatish Balay 
958a7e14dcfSSatish Balay   nlsP->gamma1_i = 0.0625;
959a7e14dcfSSatish Balay   nlsP->gamma2_i = 0.5;
960a7e14dcfSSatish Balay   nlsP->gamma3_i = 2.0;
961a7e14dcfSSatish Balay   nlsP->gamma4_i = 5.0;
962a7e14dcfSSatish Balay 
963a7e14dcfSSatish Balay   nlsP->theta_i = 0.25;
964a7e14dcfSSatish Balay 
965a7e14dcfSSatish Balay   /*  Remaining parameters */
966a7e14dcfSSatish Balay   nlsP->min_radius = 1.0e-10;
967a7e14dcfSSatish Balay   nlsP->max_radius = 1.0e10;
968a7e14dcfSSatish Balay   nlsP->epsilon = 1.0e-6;
969a7e14dcfSSatish Balay 
970a7e14dcfSSatish Balay   nlsP->init_type       = NLS_INIT_INTERPOLATION;
971a7e14dcfSSatish Balay   nlsP->update_type     = NLS_UPDATE_STEP;
972a7e14dcfSSatish Balay 
973a7e14dcfSSatish Balay   ierr = TaoLineSearchCreate(((PetscObject)tao)->comm,&tao->linesearch);CHKERRQ(ierr);
97463b15415SAlp Dener   ierr = PetscObjectIncrementTabLevel((PetscObject)tao->linesearch,(PetscObject)tao,1);CHKERRQ(ierr);
975a7e14dcfSSatish Balay   ierr = TaoLineSearchSetType(tao->linesearch,morethuente_type);CHKERRQ(ierr);
976441846f8SBarry Smith   ierr = TaoLineSearchUseTaoRoutines(tao->linesearch,tao);CHKERRQ(ierr);
9775d527766SPatrick Farrell   ierr = TaoLineSearchSetOptionsPrefix(tao->linesearch,tao->hdr.prefix);CHKERRQ(ierr);
978a7e14dcfSSatish Balay 
979a7e14dcfSSatish Balay   /*  Set linear solver to default for symmetric matrices */
980a7e14dcfSSatish Balay   ierr = KSPCreate(((PetscObject)tao)->comm,&tao->ksp);CHKERRQ(ierr);
98163b15415SAlp Dener   ierr = PetscObjectIncrementTabLevel((PetscObject)tao->ksp,(PetscObject)tao,1);CHKERRQ(ierr);
9825d527766SPatrick Farrell   ierr = KSPSetOptionsPrefix(tao->ksp,tao->hdr.prefix);CHKERRQ(ierr);
983cbf034f8SAlp Dener   ierr = KSPAppendOptionsPrefix(tao->ksp,"tao_nls_");CHKERRQ(ierr);
98405de396fSBarry Smith   ierr = KSPSetType(tao->ksp,KSPSTCG);CHKERRQ(ierr);
985a7e14dcfSSatish Balay   PetscFunctionReturn(0);
986a7e14dcfSSatish Balay }
987