Actual source code: ssils.c

  1: #include "src/complementarity/impls/ssls/ssls.h"

  3: /*------------------------------------------------------------*/
  6: static int TaoSolve_SSILS(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 = TaoSetMeritFunction(tao, Tao_SSLS_Function, Tao_SSLS_FunctionGradient, 
 39:                              TAO_NULL, TAO_NULL, ssls); CHKERRQ(info);

 41:   // Calculate the function value and fischer function value at the 
 42:   // current iterate
 43:   info = TaoComputeMeritFunctionGradient(tao, x, &psi, dpsi); CHKERRQ(info);
 44:   info = dpsi->Norm2(&ndpsi); CHKERRQ(info);

 46:   while (1) {
 47:     info=PetscInfo3(tao, "TaoSolve_SSILS: %d: merit: %5.4e, ndpsi: %5.4e\n",
 48:                        iter, ssls->merit, ndpsi); CHKERRQ(info);
 49:     // Check the termination criteria
 50:     info = TaoMonitor(tao,iter++,ssls->merit,ndpsi,0.0,t,&reason); 
 51:            CHKERRQ(info);
 52:     if (reason!=TAO_CONTINUE_ITERATING) break;

 54:     // Calculate direction.  (Really negative of newton direction.  Therefore,
 55:     // rest of the code uses -d.)
 56:     info = TaoPreLinearSolve(tao, J); CHKERRQ(info);
 57:     info = TaoLinearSolve(tao, J, ff, d, &flag); CHKERRQ(info);
 58:     info = d->Norm2(&normd); CHKERRQ(info);
 59:     info = d->Dot(dpsi, &innerd); CHKERRQ(info);

 61:     // Make sure that we have a descent direction
 62:     if (innerd <= delta*pow(normd, rho)) {
 63:       info = PetscInfo1(tao, "TaoSolve_SSILS: %d: newton direction not descent\n", iter); CHKERRQ(info);
 64:       info = d->CopyFrom(dpsi); CHKERRQ(info);
 65:       info = d->Dot(dpsi, &innerd); CHKERRQ(info);
 66:     }

 68:     info = d->Negate(); CHKERRQ(info);
 69:     innerd = -innerd;

 71:     t = 1;
 72:     info = TaoLineSearchApply(tao, x, dpsi, d, w, 
 73:                               &psi, &psi_full, &t, &tao->lsflag); CHKERRQ(info);
 74:     info = dpsi->Norm2(&ndpsi);
 75:   }

 77:   TaoFunctionReturn(0);
 78: }

 80: /* ---------------------------------------------------------- */
 84: int TaoCreate_SSILS(TAO_SOLVER tao)
 85: {
 86:   TAO_SSLS *ssls;
 87:   int        info;

 89:   TaoFunctionBegin;

 91:   info = TaoNew(TAO_SSLS,&ssls); CHKERRQ(info);
 92:   info = PetscLogObjectMemory(tao, sizeof(TAO_SSLS)); CHKERRQ(info);

 94:   ssls->delta = 1e-10;
 95:   ssls->rho = 2.1;

 97:   info=TaoSetTaoSolveRoutine(tao,TaoSolve_SSILS,(void*)ssls); CHKERRQ(info);
 98:   info=TaoSetTaoSetUpDownRoutines(tao,TaoSetUp_SSLS,TaoSetDown_SSLS); CHKERRQ(info);
 99:   info=TaoSetTaoOptionsRoutine(tao,TaoSetOptions_SSLS); CHKERRQ(info);
100:   info=TaoSetTaoViewRoutine(tao,TaoView_SSLS); CHKERRQ(info);

102:   info = TaoCreateArmijoLineSearch(tao); CHKERRQ(info);

104:   info = TaoSetMaximumIterates(tao,2000); CHKERRQ(info);
105:   info = TaoSetMaximumFunctionEvaluations(tao,4000); CHKERRQ(info);

107:   info = TaoSetTolerances(tao,0,0,0,0); CHKERRQ(info);
108:   info = TaoSetGradientTolerances(tao,1.0e-16,0.0,0.0); CHKERRQ(info);
109:   info = TaoSetFunctionLowerBound(tao,1.0e-8); CHKERRQ(info);
110:   TaoFunctionReturn(0);
111: }