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: }