Actual source code: ssfls.c
1: #include "src/complementarity/impls/ssls/ssls.h"
3: /*------------------------------------------------------------*/
6: static int TaoSolve_SSFLS(TAO_SOLVER tao, void *solver)
7: {
8: TAO_SSLS *ssls = (TAO_SSLS *)solver;
9: // TaoLinearSolver *lsolver;
10: TaoVec *x, *l, *u, *ff, *dpsi, *d, *w;
11: TaoMat *J;
12: double psi, psi_full, ndpsi, normd, innerd, t=0;
13: double delta, rho;
14: int iter=0, info;
15: TaoTerminateReason reason;
16: TaoTruth flag;
18: TaoFunctionBegin;
20: // Assume that Setup has been called!
21: // Set the structure for the Jacobian and create a linear solver.
22:
23: delta = ssls->delta;
24: rho = ssls->rho;
26: info = TaoGetSolution(tao, &x); CHKERRQ(info);
27: l=ssls->xl;
28: u=ssls->xu;
29: info = TaoEvaluateVariableBounds(tao,l,u); CHKERRQ(info);
30: info = x->Median(l,x,u); CHKERRQ(info);
31: info = TaoGetJacobian(tao, &J); CHKERRQ(info);
33: ff = ssls->ff;
34: dpsi = ssls->dpsi;
35: d = ssls->d;
36: w = ssls->w;
38: info = x->PointwiseMaximum(x, l); CHKERRQ(info);
39: info = x->PointwiseMinimum(x, u); CHKERRQ(info);
40: info = TaoSetMeritFunction(tao, Tao_SSLS_Function, Tao_SSLS_FunctionGradient,
41: TAO_NULL, TAO_NULL, ssls); CHKERRQ(info);
43: // Calculate the function value and fischer function value at the
44: // current iterate
45: info = TaoComputeMeritFunctionGradient(tao, x, &psi, dpsi); CHKERRQ(info);
46: info = dpsi->Norm2(&ndpsi);
48: while (1) {
49: info=PetscInfo3(tao, "TaoSolve_SSFLS: %d: merit: %5.4e, ndpsi: %5.4e\n",
50: iter, ssls->merit, ndpsi);CHKERRQ(info);
52: // Check the termination criteria
53: info = TaoMonitor(tao,iter++,ssls->merit,ndpsi,0.0,t,&reason);
54: CHKERRQ(info);
55: if (reason!=TAO_CONTINUE_ITERATING) break;
57: // Calculate direction. (Really negative of newton direction. Therefore,
58: // rest of the code uses -d.)
59: info = TaoPreLinearSolve(tao, J); CHKERRQ(info);
60: info = TaoLinearSolve(tao, J, ff, d, &flag); CHKERRQ(info);
61:
62: info = w->CopyFrom(d); CHKERRQ(info);
63: info = w->Negate(); CHKERRQ(info);
64: info = w->BoundGradientProjection(w,l, x, u);
66: info = w->Norm2(&normd); CHKERRQ(info);
67: info = w->Dot(dpsi, &innerd); CHKERRQ(info);
69: // Make sure that we have a descent direction
70: if (innerd >= -delta*pow(normd, rho)) {
71: info = PetscInfo1(tao, "TaoSolve_SSFLS: %d: newton direction not descent\n", iter); CHKERRQ(info);
72: info = d->CopyFrom(dpsi); CHKERRQ(info);
73: info = w->Dot(dpsi, &innerd); CHKERRQ(info);
74: }
75: info = d->Negate(); CHKERRQ(info);
76: innerd = -innerd;
78: t = 1;
79: info = TaoLineSearchApply(tao, x, dpsi, d, w,
80: &psi, &psi_full, &t, &tao->lsflag); CHKERRQ(info);
81: info = dpsi->Norm2(&ndpsi);
82: }
83: TaoFunctionReturn(0);
84: }
86: /* ---------------------------------------------------------- */
90: int TaoCreate_SSFLS(TAO_SOLVER tao)
91: {
92: TAO_SSLS *ssls;
93: int info;
95: TaoFunctionBegin;
97: info = TaoNew(TAO_SSLS,&ssls); CHKERRQ(info);
98: info = PetscLogObjectMemory(tao, sizeof(TAO_SSLS)); CHKERRQ(info);
100: ssls->delta = 1e-10;
101: ssls->rho = 2.1;
103: info=TaoSetTaoSolveRoutine(tao,TaoSolve_SSFLS,(void*)ssls); CHKERRQ(info);
104: info=TaoSetTaoSetUpDownRoutines(tao,TaoSetUp_SSLS,TaoSetDown_SSLS); CHKERRQ(info);
105: info=TaoSetTaoOptionsRoutine(tao,TaoSetOptions_SSLS); CHKERRQ(info);
106: info=TaoSetTaoViewRoutine(tao,TaoView_SSLS); CHKERRQ(info);
108: info = TaoCreateProjectedArmijoLineSearch(tao); CHKERRQ(info);
110: info = TaoSetMaximumIterates(tao,2000); CHKERRQ(info);
111: info = TaoSetMaximumFunctionEvaluations(tao,4000); CHKERRQ(info);
113: info = TaoSetTolerances(tao,0,0,0,0); CHKERRQ(info);
114: info = TaoSetGradientTolerances(tao,1.0e-16,0.0,0.0); CHKERRQ(info);
115: info = TaoSetFunctionLowerBound(tao,1.0e-8); CHKERRQ(info);
117: TaoFunctionReturn(0);
118: }