xref: /petsc/src/snes/impls/vi/vi.c (revision fe835674b7384f9866c09026cc05aa84cd4654a2)
12b4ed282SShri Abhyankar #define PETSCSNES_DLL
22b4ed282SShri Abhyankar 
32b4ed282SShri Abhyankar #include "../src/snes/impls/vi/viimpl.h"
4408da460SShri Abhyankar #include "../include/private/kspimpl.h"
52b4ed282SShri Abhyankar 
69308c137SBarry Smith #undef __FUNCT__
79308c137SBarry Smith #define __FUNCT__ "SNESMonitorVI"
87087cfbeSBarry Smith PetscErrorCode  SNESMonitorVI(SNES snes,PetscInt its,PetscReal fgnorm,void *dummy)
99308c137SBarry Smith {
109308c137SBarry Smith   PetscErrorCode          ierr;
119308c137SBarry Smith   SNES_VI                 *vi = (SNES_VI*)snes->data;
129308c137SBarry Smith   PetscViewerASCIIMonitor viewer = (PetscViewerASCIIMonitor) dummy;
139308c137SBarry Smith   const PetscScalar       *x,*xl,*xu,*f;
1409573ac7SBarry Smith   PetscInt                i,n,act = 0;
159308c137SBarry Smith   PetscReal               rnorm,fnorm;
169308c137SBarry Smith 
179308c137SBarry Smith   PetscFunctionBegin;
189308c137SBarry Smith   if (!dummy) {
199308c137SBarry Smith     ierr = PetscViewerASCIIMonitorCreate(((PetscObject)snes)->comm,"stdout",0,&viewer);CHKERRQ(ierr);
209308c137SBarry Smith   }
219308c137SBarry Smith   ierr = VecGetLocalSize(snes->vec_sol,&n);CHKERRQ(ierr);
229308c137SBarry Smith   ierr = VecGetArrayRead(vi->xl,&xl);CHKERRQ(ierr);
239308c137SBarry Smith   ierr = VecGetArrayRead(vi->xu,&xu);CHKERRQ(ierr);
249308c137SBarry Smith   ierr = VecGetArrayRead(snes->vec_sol,&x);CHKERRQ(ierr);
253f731af5SBarry Smith   ierr = VecGetArrayRead(snes->vec_func,&f);CHKERRQ(ierr);
269308c137SBarry Smith 
279308c137SBarry Smith   rnorm = 0.0;
289308c137SBarry Smith   for (i=0; i<n; i++) {
297b2c10efSShri Abhyankar     if (((x[i] > xl[i] + 1.e-8 || (f[i] < 0.0)) && ((x[i] < xu[i] - 1.e-8) || f[i] > 0.0))) rnorm += f[i]*f[i];
3009573ac7SBarry Smith     else act++;
319308c137SBarry Smith   }
323f731af5SBarry Smith   ierr = VecRestoreArrayRead(snes->vec_func,&f);CHKERRQ(ierr);
339308c137SBarry Smith   ierr = VecRestoreArrayRead(vi->xl,&xl);CHKERRQ(ierr);
349308c137SBarry Smith   ierr = VecRestoreArrayRead(vi->xu,&xu);CHKERRQ(ierr);
359308c137SBarry Smith   ierr = VecRestoreArrayRead(snes->vec_sol,&x);CHKERRQ(ierr);
369308c137SBarry Smith   ierr = MPI_Allreduce(&rnorm,&fnorm,1,MPIU_REAL,MPI_SUM,((PetscObject)snes)->comm);CHKERRQ(ierr);
379308c137SBarry Smith   fnorm = sqrt(fnorm);
3809573ac7SBarry Smith   ierr = PetscViewerASCIIMonitorPrintf(viewer,"%3D SNES VI Function norm %14.12e Active constraints %D\n",its,fnorm,act);CHKERRQ(ierr);
399308c137SBarry Smith   if (!dummy) {
409308c137SBarry Smith     ierr = PetscViewerASCIIMonitorDestroy(viewer);CHKERRQ(ierr);
419308c137SBarry Smith   }
429308c137SBarry Smith   PetscFunctionReturn(0);
439308c137SBarry Smith }
449308c137SBarry Smith 
452b4ed282SShri Abhyankar /*
462b4ed282SShri Abhyankar      Checks if J^T F = 0 which implies we've found a local minimum of the norm of the function,
472b4ed282SShri Abhyankar     || F(u) ||_2 but not a zero, F(u) = 0. In the case when one cannot compute J^T F we use the fact that
482b4ed282SShri Abhyankar     0 = (J^T F)^T W = F^T J W iff W not in the null space of J. Thanks for Jorge More
492b4ed282SShri Abhyankar     for this trick. One assumes that the probability that W is in the null space of J is very, very small.
502b4ed282SShri Abhyankar */
512b4ed282SShri Abhyankar #undef __FUNCT__
522b4ed282SShri Abhyankar #define __FUNCT__ "SNESVICheckLocalMin_Private"
53ace3abfcSBarry Smith PetscErrorCode SNESVICheckLocalMin_Private(SNES snes,Mat A,Vec F,Vec W,PetscReal fnorm,PetscBool *ismin)
542b4ed282SShri Abhyankar {
552b4ed282SShri Abhyankar   PetscReal      a1;
562b4ed282SShri Abhyankar   PetscErrorCode ierr;
57ace3abfcSBarry Smith   PetscBool     hastranspose;
582b4ed282SShri Abhyankar 
592b4ed282SShri Abhyankar   PetscFunctionBegin;
602b4ed282SShri Abhyankar   *ismin = PETSC_FALSE;
612b4ed282SShri Abhyankar   ierr = MatHasOperation(A,MATOP_MULT_TRANSPOSE,&hastranspose);CHKERRQ(ierr);
622b4ed282SShri Abhyankar   if (hastranspose) {
632b4ed282SShri Abhyankar     /* Compute || J^T F|| */
642b4ed282SShri Abhyankar     ierr = MatMultTranspose(A,F,W);CHKERRQ(ierr);
652b4ed282SShri Abhyankar     ierr = VecNorm(W,NORM_2,&a1);CHKERRQ(ierr);
662b4ed282SShri Abhyankar     ierr = PetscInfo1(snes,"|| J^T F|| %G near zero implies found a local minimum\n",a1/fnorm);CHKERRQ(ierr);
672b4ed282SShri Abhyankar     if (a1/fnorm < 1.e-4) *ismin = PETSC_TRUE;
682b4ed282SShri Abhyankar   } else {
692b4ed282SShri Abhyankar     Vec         work;
702b4ed282SShri Abhyankar     PetscScalar result;
712b4ed282SShri Abhyankar     PetscReal   wnorm;
722b4ed282SShri Abhyankar 
732b4ed282SShri Abhyankar     ierr = VecSetRandom(W,PETSC_NULL);CHKERRQ(ierr);
742b4ed282SShri Abhyankar     ierr = VecNorm(W,NORM_2,&wnorm);CHKERRQ(ierr);
752b4ed282SShri Abhyankar     ierr = VecDuplicate(W,&work);CHKERRQ(ierr);
762b4ed282SShri Abhyankar     ierr = MatMult(A,W,work);CHKERRQ(ierr);
772b4ed282SShri Abhyankar     ierr = VecDot(F,work,&result);CHKERRQ(ierr);
782b4ed282SShri Abhyankar     ierr = VecDestroy(work);CHKERRQ(ierr);
792b4ed282SShri Abhyankar     a1   = PetscAbsScalar(result)/(fnorm*wnorm);
802b4ed282SShri Abhyankar     ierr = PetscInfo1(snes,"(F^T J random)/(|| F ||*||random|| %G near zero implies found a local minimum\n",a1);CHKERRQ(ierr);
812b4ed282SShri Abhyankar     if (a1 < 1.e-4) *ismin = PETSC_TRUE;
822b4ed282SShri Abhyankar   }
832b4ed282SShri Abhyankar   PetscFunctionReturn(0);
842b4ed282SShri Abhyankar }
852b4ed282SShri Abhyankar 
862b4ed282SShri Abhyankar /*
872b4ed282SShri Abhyankar      Checks if J^T(F - J*X) = 0
882b4ed282SShri Abhyankar */
892b4ed282SShri Abhyankar #undef __FUNCT__
902b4ed282SShri Abhyankar #define __FUNCT__ "SNESVICheckResidual_Private"
912b4ed282SShri Abhyankar PetscErrorCode SNESVICheckResidual_Private(SNES snes,Mat A,Vec F,Vec X,Vec W1,Vec W2)
922b4ed282SShri Abhyankar {
932b4ed282SShri Abhyankar   PetscReal      a1,a2;
942b4ed282SShri Abhyankar   PetscErrorCode ierr;
95ace3abfcSBarry Smith   PetscBool     hastranspose;
962b4ed282SShri Abhyankar 
972b4ed282SShri Abhyankar   PetscFunctionBegin;
982b4ed282SShri Abhyankar   ierr = MatHasOperation(A,MATOP_MULT_TRANSPOSE,&hastranspose);CHKERRQ(ierr);
992b4ed282SShri Abhyankar   if (hastranspose) {
1002b4ed282SShri Abhyankar     ierr = MatMult(A,X,W1);CHKERRQ(ierr);
1012b4ed282SShri Abhyankar     ierr = VecAXPY(W1,-1.0,F);CHKERRQ(ierr);
1022b4ed282SShri Abhyankar 
1032b4ed282SShri Abhyankar     /* Compute || J^T W|| */
1042b4ed282SShri Abhyankar     ierr = MatMultTranspose(A,W1,W2);CHKERRQ(ierr);
1052b4ed282SShri Abhyankar     ierr = VecNorm(W1,NORM_2,&a1);CHKERRQ(ierr);
1062b4ed282SShri Abhyankar     ierr = VecNorm(W2,NORM_2,&a2);CHKERRQ(ierr);
1072b4ed282SShri Abhyankar     if (a1 != 0.0) {
1082b4ed282SShri Abhyankar       ierr = PetscInfo1(snes,"||J^T(F-Ax)||/||F-AX|| %G near zero implies inconsistent rhs\n",a2/a1);CHKERRQ(ierr);
1092b4ed282SShri Abhyankar     }
1102b4ed282SShri Abhyankar   }
1112b4ed282SShri Abhyankar   PetscFunctionReturn(0);
1122b4ed282SShri Abhyankar }
1132b4ed282SShri Abhyankar 
1142b4ed282SShri Abhyankar /*
1152b4ed282SShri Abhyankar   SNESDefaultConverged_VI - Checks the convergence of the semismooth newton algorithm.
1162b4ed282SShri Abhyankar 
1172b4ed282SShri Abhyankar   Notes:
1182b4ed282SShri Abhyankar   The convergence criterion currently implemented is
1192b4ed282SShri Abhyankar   merit < abstol
1202b4ed282SShri Abhyankar   merit < rtol*merit_initial
1212b4ed282SShri Abhyankar */
1222b4ed282SShri Abhyankar #undef __FUNCT__
1232b4ed282SShri Abhyankar #define __FUNCT__ "SNESDefaultConverged_VI"
1242b4ed282SShri Abhyankar PetscErrorCode SNESDefaultConverged_VI(SNES snes,PetscInt it,PetscReal xnorm,PetscReal gradnorm,PetscReal merit,SNESConvergedReason *reason,void *dummy)
1252b4ed282SShri Abhyankar {
1262b4ed282SShri Abhyankar   PetscErrorCode ierr;
1272b4ed282SShri Abhyankar 
1282b4ed282SShri Abhyankar   PetscFunctionBegin;
1292b4ed282SShri Abhyankar   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
1302b4ed282SShri Abhyankar   PetscValidPointer(reason,6);
1312b4ed282SShri Abhyankar 
1322b4ed282SShri Abhyankar   *reason = SNES_CONVERGED_ITERATING;
1332b4ed282SShri Abhyankar 
1342b4ed282SShri Abhyankar   if (!it) {
1352b4ed282SShri Abhyankar     /* set parameter for default relative tolerance convergence test */
1362b4ed282SShri Abhyankar     snes->ttol = merit*snes->rtol;
1372b4ed282SShri Abhyankar   }
1382b4ed282SShri Abhyankar   if (merit != merit) {
1392b4ed282SShri Abhyankar     ierr = PetscInfo(snes,"Failed to converged, function norm is NaN\n");CHKERRQ(ierr);
1402b4ed282SShri Abhyankar     *reason = SNES_DIVERGED_FNORM_NAN;
1412b4ed282SShri Abhyankar   } else if (merit < snes->abstol) {
1422b4ed282SShri Abhyankar     ierr = PetscInfo2(snes,"Converged due to merit function %G < %G\n",merit,snes->abstol);CHKERRQ(ierr);
1432b4ed282SShri Abhyankar     *reason = SNES_CONVERGED_FNORM_ABS;
1442b4ed282SShri Abhyankar   } else if (snes->nfuncs >= snes->max_funcs) {
1452b4ed282SShri Abhyankar     ierr = PetscInfo2(snes,"Exceeded maximum number of function evaluations: %D > %D\n",snes->nfuncs,snes->max_funcs);CHKERRQ(ierr);
1462b4ed282SShri Abhyankar     *reason = SNES_DIVERGED_FUNCTION_COUNT;
1472b4ed282SShri Abhyankar   }
1482b4ed282SShri Abhyankar 
1492b4ed282SShri Abhyankar   if (it && !*reason) {
1502b4ed282SShri Abhyankar     if (merit < snes->ttol) {
1512b4ed282SShri Abhyankar       ierr = PetscInfo2(snes,"Converged due to merit function %G < %G (relative tolerance)\n",merit,snes->ttol);CHKERRQ(ierr);
1522b4ed282SShri Abhyankar       *reason = SNES_CONVERGED_FNORM_RELATIVE;
1532b4ed282SShri Abhyankar     }
1542b4ed282SShri Abhyankar   }
1552b4ed282SShri Abhyankar   PetscFunctionReturn(0);
1562b4ed282SShri Abhyankar }
1572b4ed282SShri Abhyankar 
1582b4ed282SShri Abhyankar /*
1592b4ed282SShri Abhyankar   SNESVIComputeMeritFunction - Evaluates the merit function for the mixed complementarity problem.
1602b4ed282SShri Abhyankar 
1612b4ed282SShri Abhyankar   Input Parameter:
1622b4ed282SShri Abhyankar . phi - the semismooth function
1632b4ed282SShri Abhyankar 
1642b4ed282SShri Abhyankar   Output Parameter:
1652b4ed282SShri Abhyankar . merit - the merit function
1662b4ed282SShri Abhyankar . phinorm - ||phi||
1672b4ed282SShri Abhyankar 
1682b4ed282SShri Abhyankar   Notes:
1692b4ed282SShri Abhyankar   The merit function for the mixed complementarity problem is defined as
1702b4ed282SShri Abhyankar      merit = 0.5*phi^T*phi
1712b4ed282SShri Abhyankar */
1722b4ed282SShri Abhyankar #undef __FUNCT__
1732b4ed282SShri Abhyankar #define __FUNCT__ "SNESVIComputeMeritFunction"
1742b4ed282SShri Abhyankar static PetscErrorCode SNESVIComputeMeritFunction(Vec phi, PetscReal* merit,PetscReal* phinorm)
1752b4ed282SShri Abhyankar {
1762b4ed282SShri Abhyankar   PetscErrorCode ierr;
1772b4ed282SShri Abhyankar 
1782b4ed282SShri Abhyankar   PetscFunctionBegin;
1792b4ed282SShri Abhyankar   ierr = VecNormBegin(phi,NORM_2,phinorm);
1802b4ed282SShri Abhyankar   ierr = VecNormEnd(phi,NORM_2,phinorm);
1812b4ed282SShri Abhyankar 
1822b4ed282SShri Abhyankar   *merit = 0.5*(*phinorm)*(*phinorm);
1832b4ed282SShri Abhyankar   PetscFunctionReturn(0);
1842b4ed282SShri Abhyankar }
1852b4ed282SShri Abhyankar 
1862b4ed282SShri Abhyankar /*
1872b4ed282SShri Abhyankar   ComputeFischerFunction - Computes the semismooth fischer burmeister function for a mixed complementarity equation.
1882b4ed282SShri Abhyankar 
1892b4ed282SShri Abhyankar   Notes:
1902b4ed282SShri Abhyankar   The Fischer-Burmeister function is defined as
1912b4ed282SShri Abhyankar        ff(a,b) := sqrt(a*a + b*b) - a - b
1922b4ed282SShri Abhyankar   and is used reformulate a complementarity equation  as a semismooth equation.
1932b4ed282SShri Abhyankar */
1942b4ed282SShri Abhyankar 
1952b4ed282SShri Abhyankar #undef __FUNCT__
1962b4ed282SShri Abhyankar #define __FUNCT__ "ComputeFischerFunction"
1972b4ed282SShri Abhyankar static PetscErrorCode ComputeFischerFunction(PetscScalar a, PetscScalar b, PetscScalar* ff)
1982b4ed282SShri Abhyankar {
1992b4ed282SShri Abhyankar   PetscFunctionBegin;
2002b4ed282SShri Abhyankar   *ff = sqrt(a*a + b*b) - a - b;
2012b4ed282SShri Abhyankar   PetscFunctionReturn(0);
2022b4ed282SShri Abhyankar }
2032b4ed282SShri Abhyankar 
2042b4ed282SShri Abhyankar /*
2051f79c099SShri Abhyankar    SNESVIComputeFunction - Reformulates a system of nonlinear equations in mixed complementarity form to a system of nonlinear equations in semismooth form.
2062b4ed282SShri Abhyankar 
2072b4ed282SShri Abhyankar    Input Parameters:
2082b4ed282SShri Abhyankar .  snes - the SNES context
2092b4ed282SShri Abhyankar .  x - current iterate
2102b4ed282SShri Abhyankar .  functx - user defined function context
2112b4ed282SShri Abhyankar 
2122b4ed282SShri Abhyankar    Output Parameters:
2132b4ed282SShri Abhyankar .  phi - Semismooth function
2142b4ed282SShri Abhyankar 
2152b4ed282SShri Abhyankar    Notes:
2162b4ed282SShri Abhyankar    The result of this function is done by cases:
2172b4ed282SShri Abhyankar +  l[i] == -infinity, u[i] == infinity  -- phi[i] = -f[i]
2182b4ed282SShri Abhyankar .  l[i] == -infinity, u[i] finite       -- phi[i] = ss(u[i]-x[i], -f[i])
2192b4ed282SShri Abhyankar .  l[i] finite,       u[i] == infinity  -- phi[i] = ss(x[i]-l[i],  f[i])
2202b4ed282SShri Abhyankar .  l[i] finite < u[i] finite -- phi[i] = phi(x[i]-l[i], ss(u[i]-x[i], -f[u]))
2212b4ed282SShri Abhyankar -  otherwise l[i] == u[i] -- phi[i] = l[i] - x[i]
2222b4ed282SShri Abhyankar    ss is the semismoothing function used to reformulate the nonlinear equation in complementarity
2232b4ed282SShri Abhyankar    form to semismooth form
2242b4ed282SShri Abhyankar 
2252b4ed282SShri Abhyankar */
2262b4ed282SShri Abhyankar #undef __FUNCT__
2271f79c099SShri Abhyankar #define __FUNCT__ "SNESVIComputeFunction"
2281f79c099SShri Abhyankar static PetscErrorCode SNESVIComputeFunction(SNES snes,Vec X,Vec phi,void* functx)
2292b4ed282SShri Abhyankar {
2302b4ed282SShri Abhyankar   PetscErrorCode  ierr;
2312b4ed282SShri Abhyankar   SNES_VI       *vi = (SNES_VI*)snes->data;
2322b4ed282SShri Abhyankar   Vec             Xl = vi->xl,Xu = vi->xu,F = snes->vec_func;
2332b4ed282SShri Abhyankar   PetscScalar     *phi_arr,*x_arr,*f_arr,*l,*u,t;
2342b4ed282SShri Abhyankar   PetscInt        i,nlocal;
2352b4ed282SShri Abhyankar 
2362b4ed282SShri Abhyankar   PetscFunctionBegin;
2372b4ed282SShri Abhyankar   ierr = (*vi->computeuserfunction)(snes,X,F,functx);CHKERRQ(ierr);
2382b4ed282SShri Abhyankar   ierr = VecGetLocalSize(X,&nlocal);CHKERRQ(ierr);
2392b4ed282SShri Abhyankar 
2402b4ed282SShri Abhyankar   ierr = VecGetArray(X,&x_arr);CHKERRQ(ierr);
2412b4ed282SShri Abhyankar   ierr = VecGetArray(F,&f_arr);CHKERRQ(ierr);
2422b4ed282SShri Abhyankar   ierr = VecGetArray(Xl,&l);CHKERRQ(ierr);
2432b4ed282SShri Abhyankar   ierr = VecGetArray(Xu,&u);CHKERRQ(ierr);
2442b4ed282SShri Abhyankar   ierr = VecGetArray(phi,&phi_arr);CHKERRQ(ierr);
2452b4ed282SShri Abhyankar 
2462b4ed282SShri Abhyankar   for (i=0;i < nlocal;i++) {
2472b4ed282SShri Abhyankar     if ((l[i] <= PETSC_VI_NINF) && (u[i] >= PETSC_VI_INF)) {
2482b4ed282SShri Abhyankar       phi_arr[i] = -f_arr[i];
2492b4ed282SShri Abhyankar     }
2502b4ed282SShri Abhyankar     else if (l[i] <= PETSC_VI_NINF) {
2512b4ed282SShri Abhyankar       t = u[i] - x_arr[i];
2522b4ed282SShri Abhyankar       ierr = ComputeFischerFunction(t,-f_arr[i],&phi_arr[i]);CHKERRQ(ierr);
2532b4ed282SShri Abhyankar       phi_arr[i] = -phi_arr[i];
2542b4ed282SShri Abhyankar     }
2552b4ed282SShri Abhyankar     else if (u[i] >= PETSC_VI_INF) {
2562b4ed282SShri Abhyankar       t = x_arr[i] - l[i];
2572b4ed282SShri Abhyankar       ierr = ComputeFischerFunction(t,f_arr[i],&phi_arr[i]);CHKERRQ(ierr);
2582b4ed282SShri Abhyankar     }
2592b4ed282SShri Abhyankar     else if (l[i] == u[i]) {
2602b4ed282SShri Abhyankar       phi_arr[i] = l[i] - x_arr[i];
2612b4ed282SShri Abhyankar     }
2622b4ed282SShri Abhyankar     else {
2632b4ed282SShri Abhyankar       t = u[i] - x_arr[i];
2642b4ed282SShri Abhyankar       ierr = ComputeFischerFunction(t,-f_arr[i],&phi_arr[i]);
2652b4ed282SShri Abhyankar       t = x_arr[i] - l[i];
2662b4ed282SShri Abhyankar       ierr = ComputeFischerFunction(t,phi_arr[i],&phi_arr[i]);
2672b4ed282SShri Abhyankar     }
2682b4ed282SShri Abhyankar   }
2692b4ed282SShri Abhyankar 
2702b4ed282SShri Abhyankar   ierr = VecRestoreArray(X,&x_arr);CHKERRQ(ierr);
2712b4ed282SShri Abhyankar   ierr = VecRestoreArray(F,&f_arr);CHKERRQ(ierr);
2722b4ed282SShri Abhyankar   ierr = VecRestoreArray(Xl,&l);CHKERRQ(ierr);
2732b4ed282SShri Abhyankar   ierr = VecRestoreArray(Xu,&u);CHKERRQ(ierr);
2742b4ed282SShri Abhyankar   ierr = VecRestoreArray(phi,&phi_arr);CHKERRQ(ierr);
2752b4ed282SShri Abhyankar 
2762b4ed282SShri Abhyankar   PetscFunctionReturn(0);
2772b4ed282SShri Abhyankar }
2782b4ed282SShri Abhyankar 
2792b4ed282SShri Abhyankar /*
280a79edbefSShri Abhyankar    SNESVIComputeBsubdifferentialVectors - Computes the diagonal shift (Da) and row scaling (Db) vectors needed for the
281a79edbefSShri Abhyankar                                           the semismooth jacobian.
2822b4ed282SShri Abhyankar */
2832b4ed282SShri Abhyankar #undef __FUNCT__
284a79edbefSShri Abhyankar #define __FUNCT__ "SNESVIComputeBsubdifferentialVectors"
285a79edbefSShri Abhyankar PetscErrorCode SNESVIComputeBsubdifferentialVectors(SNES snes,Vec X,Vec F,Mat jac,Vec Da,Vec Db)
2862b4ed282SShri Abhyankar {
2872b4ed282SShri Abhyankar   PetscErrorCode ierr;
2882b4ed282SShri Abhyankar   SNES_VI      *vi = (SNES_VI*)snes->data;
2892b4ed282SShri Abhyankar   PetscScalar    *l,*u,*x,*f,*da,*db,*z,*t,t1,t2,ci,di,ei;
2902b4ed282SShri Abhyankar   PetscInt       i,nlocal;
2912b4ed282SShri Abhyankar 
2922b4ed282SShri Abhyankar   PetscFunctionBegin;
2932b4ed282SShri Abhyankar 
2942b4ed282SShri Abhyankar   ierr = VecGetArray(X,&x);CHKERRQ(ierr);
2952b4ed282SShri Abhyankar   ierr = VecGetArray(F,&f);CHKERRQ(ierr);
2962b4ed282SShri Abhyankar   ierr = VecGetArray(vi->xl,&l);CHKERRQ(ierr);
2972b4ed282SShri Abhyankar   ierr = VecGetArray(vi->xu,&u);CHKERRQ(ierr);
298a79edbefSShri Abhyankar   ierr = VecGetArray(Da,&da);CHKERRQ(ierr);
299a79edbefSShri Abhyankar   ierr = VecGetArray(Db,&db);CHKERRQ(ierr);
3002b4ed282SShri Abhyankar   ierr = VecGetArray(vi->z,&z);CHKERRQ(ierr);
3012b4ed282SShri Abhyankar 
3022b4ed282SShri Abhyankar   ierr = VecGetLocalSize(X,&nlocal);CHKERRQ(ierr);
3032b4ed282SShri Abhyankar   /* Set the elements of the vector z:
3042b4ed282SShri Abhyankar      z[i] = 1 if (x[i] - l[i],f[i]) = (0,0) or (u[i] - x[i],f[i]) = (0,0)
3052b4ed282SShri Abhyankar      else z[i] = 0
3062b4ed282SShri Abhyankar   */
3072b4ed282SShri Abhyankar   for(i=0;i < nlocal;i++) {
3082b4ed282SShri Abhyankar     da[i] = db[i] = z[i] = 0;
3092b4ed282SShri Abhyankar     if(PetscAbsScalar(f[i]) <= vi->const_tol) {
3102b4ed282SShri Abhyankar       if ((l[i] > PETSC_VI_NINF) && (PetscAbsScalar(x[i]-l[i]) <= vi->const_tol)) {
3112b4ed282SShri Abhyankar 	da[i] = 1;
3122b4ed282SShri Abhyankar 	z[i]  = 1;
3132b4ed282SShri Abhyankar       }
3142b4ed282SShri Abhyankar       if ((u[i] < PETSC_VI_INF) && (PetscAbsScalar(u[i]-x[i]) <= vi->const_tol)) {
3152b4ed282SShri Abhyankar 	db[i] = 1;
3162b4ed282SShri Abhyankar 	z[i]  = 1;
3172b4ed282SShri Abhyankar       }
3182b4ed282SShri Abhyankar     }
3192b4ed282SShri Abhyankar   }
3202b4ed282SShri Abhyankar   ierr = VecRestoreArray(vi->z,&z);CHKERRQ(ierr);
321a79edbefSShri Abhyankar   ierr = MatMult(jac,vi->z,vi->t);CHKERRQ(ierr);
3222b4ed282SShri Abhyankar   ierr = VecGetArray(vi->t,&t);CHKERRQ(ierr);
3232b4ed282SShri Abhyankar   /* Compute the elements of the diagonal perturbation vector Da and row scaling vector Db */
3242b4ed282SShri Abhyankar   for(i=0;i< nlocal;i++) {
3252b4ed282SShri Abhyankar     /* Free variables */
3262b4ed282SShri Abhyankar     if ((l[i] <= PETSC_VI_NINF) && (u[i] >= PETSC_VI_INF)) {
3272b4ed282SShri Abhyankar       da[i] = 0; db[i] = -1;
3282b4ed282SShri Abhyankar     }
3292b4ed282SShri Abhyankar     /* lower bounded variables */
3302b4ed282SShri Abhyankar     else if (u[i] >= PETSC_VI_INF) {
3312b4ed282SShri Abhyankar       if (da[i] >= 1) {
3322b4ed282SShri Abhyankar 	t2 = PetscScalarNorm(1,t[i]);
3332b4ed282SShri Abhyankar 	da[i] = 1/t2 - 1;
3342b4ed282SShri Abhyankar 	db[i] = t[i]/t2 - 1;
3352b4ed282SShri Abhyankar       } else {
3362b4ed282SShri Abhyankar 	t1 = x[i] - l[i];
3372b4ed282SShri Abhyankar 	t2 = PetscScalarNorm(t1,f[i]);
3382b4ed282SShri Abhyankar 	da[i] = t1/t2 - 1;
3392b4ed282SShri Abhyankar 	db[i] = f[i]/t2 - 1;
3402b4ed282SShri Abhyankar       }
3412b4ed282SShri Abhyankar     }
3422b4ed282SShri Abhyankar     /* upper bounded variables */
3432b4ed282SShri Abhyankar     else if (l[i] <= PETSC_VI_NINF) {
3442b4ed282SShri Abhyankar       if (db[i] >= 1) {
3452b4ed282SShri Abhyankar 	t2 = PetscScalarNorm(1,t[i]);
3462b4ed282SShri Abhyankar 	da[i] = -1/t2 -1;
3472b4ed282SShri Abhyankar 	db[i] = -t[i]/t2 - 1;
3482b4ed282SShri Abhyankar       }
3492b4ed282SShri Abhyankar       else {
3502b4ed282SShri Abhyankar 	t1 = u[i] - x[i];
3512b4ed282SShri Abhyankar 	t2 = PetscScalarNorm(t1,f[i]);
3522b4ed282SShri Abhyankar 	da[i] = t1/t2 - 1;
3532b4ed282SShri Abhyankar 	db[i] = -f[i]/t2 - 1;
3542b4ed282SShri Abhyankar       }
3552b4ed282SShri Abhyankar     }
3562b4ed282SShri Abhyankar     /* Fixed variables */
3572b4ed282SShri Abhyankar     else if (l[i] == u[i]) {
3582b4ed282SShri Abhyankar       da[i] = -1;
3592b4ed282SShri Abhyankar       db[i] = 0;
3602b4ed282SShri Abhyankar     }
3612b4ed282SShri Abhyankar     /* Box constrained variables */
3622b4ed282SShri Abhyankar     else {
3632b4ed282SShri Abhyankar       if (db[i] >= 1) {
3642b4ed282SShri Abhyankar 	t2 = PetscScalarNorm(1,t[i]);
3652b4ed282SShri Abhyankar 	ci = 1/t2 + 1;
3662b4ed282SShri Abhyankar 	di = t[i]/t2 + 1;
3672b4ed282SShri Abhyankar       }
3682b4ed282SShri Abhyankar       else {
3692b4ed282SShri Abhyankar 	t1 = x[i] - u[i];
3702b4ed282SShri Abhyankar 	t2 = PetscScalarNorm(t1,f[i]);
3712b4ed282SShri Abhyankar 	ci = t1/t2 + 1;
3722b4ed282SShri Abhyankar 	di = f[i]/t2 + 1;
3732b4ed282SShri Abhyankar       }
3742b4ed282SShri Abhyankar 
3752b4ed282SShri Abhyankar       if (da[i] >= 1) {
3762b4ed282SShri Abhyankar 	t1 = ci + di*t[i];
3772b4ed282SShri Abhyankar 	t2 = PetscScalarNorm(1,t1);
3782b4ed282SShri Abhyankar 	t1 = t1/t2 - 1;
3792b4ed282SShri Abhyankar 	t2 = 1/t2  - 1;
3802b4ed282SShri Abhyankar       }
3812b4ed282SShri Abhyankar       else {
3822b4ed282SShri Abhyankar 	ierr = ComputeFischerFunction(u[i]-x[i],-f[i],&ei);CHKERRQ(ierr);
3832b4ed282SShri Abhyankar 	t2 = PetscScalarNorm(x[i]-l[i],ei);
3842b4ed282SShri Abhyankar 	t1 = ei/t2 - 1;
3852b4ed282SShri Abhyankar 	t2 = (x[i] - l[i])/t2 - 1;
3862b4ed282SShri Abhyankar       }
3872b4ed282SShri Abhyankar 
3882b4ed282SShri Abhyankar       da[i] = t2 + t1*ci;
3892b4ed282SShri Abhyankar       db[i] = t1*di;
3902b4ed282SShri Abhyankar     }
3912b4ed282SShri Abhyankar   }
3922b4ed282SShri Abhyankar   ierr = VecRestoreArray(X,&x);CHKERRQ(ierr);
3932b4ed282SShri Abhyankar   ierr = VecRestoreArray(F,&f);CHKERRQ(ierr);
3942b4ed282SShri Abhyankar   ierr = VecRestoreArray(vi->xl,&l);CHKERRQ(ierr);
3952b4ed282SShri Abhyankar   ierr = VecRestoreArray(vi->xu,&u);CHKERRQ(ierr);
396a79edbefSShri Abhyankar   ierr = VecRestoreArray(Da,&da);CHKERRQ(ierr);
397a79edbefSShri Abhyankar   ierr = VecRestoreArray(Db,&db);CHKERRQ(ierr);
3982b4ed282SShri Abhyankar   ierr = VecRestoreArray(vi->t,&t);CHKERRQ(ierr);
3992b4ed282SShri Abhyankar 
400a79edbefSShri Abhyankar   PetscFunctionReturn(0);
401a79edbefSShri Abhyankar }
402a79edbefSShri Abhyankar 
403a79edbefSShri Abhyankar /*
404a79edbefSShri Abhyankar    SNESVIComputeJacobian - Computes the jacobian of the semismooth function.The Jacobian for the semismooth function is an element of the B-subdifferential of the Fischer-Burmeister function for complementarity problems.
405a79edbefSShri Abhyankar 
406a79edbefSShri Abhyankar    Input Parameters:
407a79edbefSShri Abhyankar .  Da       - Diagonal shift vector for the semismooth jacobian.
408a79edbefSShri Abhyankar .  Db       - Row scaling vector for the semismooth jacobian.
409a79edbefSShri Abhyankar 
410a79edbefSShri Abhyankar    Output Parameters:
411a79edbefSShri Abhyankar .  jac      - semismooth jacobian
412a79edbefSShri Abhyankar .  jac_pre  - optional preconditioning matrix
413a79edbefSShri Abhyankar 
414a79edbefSShri Abhyankar    Notes:
415a79edbefSShri Abhyankar    The semismooth jacobian matrix is given by
416a79edbefSShri Abhyankar    jac = Da + Db*jacfun
417a79edbefSShri Abhyankar    where Db is the row scaling matrix stored as a vector,
418a79edbefSShri Abhyankar          Da is the diagonal perturbation matrix stored as a vector
419a79edbefSShri Abhyankar    and   jacfun is the jacobian of the original nonlinear function.
420a79edbefSShri Abhyankar */
421a79edbefSShri Abhyankar #undef __FUNCT__
422a79edbefSShri Abhyankar #define __FUNCT__ "SNESVIComputeJacobian"
423a79edbefSShri Abhyankar PetscErrorCode SNESVIComputeJacobian(Mat jac, Mat jac_pre,Vec Da, Vec Db)
424a79edbefSShri Abhyankar {
425a79edbefSShri Abhyankar   PetscErrorCode ierr;
426a79edbefSShri Abhyankar 
4272b4ed282SShri Abhyankar   /* Do row scaling  and add diagonal perturbation */
428a79edbefSShri Abhyankar   ierr = MatDiagonalScale(jac,Db,PETSC_NULL);CHKERRQ(ierr);
429a79edbefSShri Abhyankar   ierr = MatDiagonalSet(jac,Da,ADD_VALUES);CHKERRQ(ierr);
430a79edbefSShri Abhyankar   if (jac != jac_pre) { /* If jac and jac_pre are different */
431a79edbefSShri Abhyankar     ierr = MatDiagonalScale(jac_pre,Db,PETSC_NULL);
432a79edbefSShri Abhyankar     ierr = MatDiagonalSet(jac_pre,Da,ADD_VALUES);CHKERRQ(ierr);
4332b4ed282SShri Abhyankar   }
4342b4ed282SShri Abhyankar   PetscFunctionReturn(0);
4352b4ed282SShri Abhyankar }
4362b4ed282SShri Abhyankar 
4372b4ed282SShri Abhyankar /*
438ee657d29SShri Abhyankar    SNESVIComputeMeritFunctionGradient - Computes the gradient of the merit function psi.
439ee657d29SShri Abhyankar 
440ee657d29SShri Abhyankar    Input Parameters:
441ee657d29SShri Abhyankar    phi - semismooth function.
442ee657d29SShri Abhyankar    H   - semismooth jacobian
443ee657d29SShri Abhyankar 
444ee657d29SShri Abhyankar    Output Parameters:
445ee657d29SShri Abhyankar    dpsi - merit function gradient
446ee657d29SShri Abhyankar 
447ee657d29SShri Abhyankar    Notes:
448ee657d29SShri Abhyankar   The merit function gradient is computed as follows
449ee657d29SShri Abhyankar         dpsi = H^T*phi
450ee657d29SShri Abhyankar */
451ee657d29SShri Abhyankar #undef __FUNCT__
452ee657d29SShri Abhyankar #define __FUNCT__ "SNESVIComputeMeritFunctionGradient"
453ee657d29SShri Abhyankar PetscErrorCode SNESVIComputeMeritFunctionGradient(Mat H, Vec phi, Vec dpsi)
454ee657d29SShri Abhyankar {
455ee657d29SShri Abhyankar   PetscErrorCode ierr;
456ee657d29SShri Abhyankar 
457ee657d29SShri Abhyankar   PetscFunctionBegin;
458ee657d29SShri Abhyankar   ierr = MatMultTranspose(H,phi,dpsi);
459ee657d29SShri Abhyankar   PetscFunctionReturn(0);
460ee657d29SShri Abhyankar }
461ee657d29SShri Abhyankar 
462ee657d29SShri Abhyankar /*
4632b4ed282SShri Abhyankar    SNESVIAdjustInitialGuess - Readjusts the initial guess to the SNES solver supplied by the user so that the initial guess lies inside the feasible region .
4642b4ed282SShri Abhyankar 
4652b4ed282SShri Abhyankar    Input Parameters:
4662b4ed282SShri Abhyankar .  lb - lower bound.
4672b4ed282SShri Abhyankar .  ub - upper bound.
4682b4ed282SShri Abhyankar 
4692b4ed282SShri Abhyankar    Output Parameters:
4702b4ed282SShri Abhyankar .  X - the readjusted initial guess.
4712b4ed282SShri Abhyankar 
4722b4ed282SShri Abhyankar    Notes:
4732b4ed282SShri Abhyankar    The readjusted initial guess X[i] = max(max(min(l[i],X[i]),min(X[i],u[i])),min(l[i],u[i]))
4742b4ed282SShri Abhyankar */
4752b4ed282SShri Abhyankar #undef __FUNCT__
4762b4ed282SShri Abhyankar #define __FUNCT__ "SNESVIAdjustInitialGuess"
4772b4ed282SShri Abhyankar PetscErrorCode SNESVIAdjustInitialGuess(Vec X, Vec lb, Vec ub)
4782b4ed282SShri Abhyankar {
4792b4ed282SShri Abhyankar   PetscErrorCode ierr;
4802b4ed282SShri Abhyankar   PetscInt       i,nlocal;
4812b4ed282SShri Abhyankar   PetscScalar    *x,*l,*u;
4822b4ed282SShri Abhyankar 
4832b4ed282SShri Abhyankar   PetscFunctionBegin;
4842b4ed282SShri Abhyankar 
4852b4ed282SShri Abhyankar   ierr = VecGetLocalSize(X,&nlocal);CHKERRQ(ierr);
4862b4ed282SShri Abhyankar   ierr = VecGetArray(X,&x);CHKERRQ(ierr);
4872b4ed282SShri Abhyankar   ierr = VecGetArray(lb,&l);CHKERRQ(ierr);
4882b4ed282SShri Abhyankar   ierr = VecGetArray(ub,&u);CHKERRQ(ierr);
4892b4ed282SShri Abhyankar 
4902b4ed282SShri Abhyankar   for(i = 0; i < nlocal; i++) {
4912b4ed282SShri Abhyankar     x[i] = PetscMax(PetscMax(PetscMin(x[i],l[i]),PetscMin(x[i],u[i])),PetscMin(l[i],u[i]));
4922b4ed282SShri Abhyankar   }
4932b4ed282SShri Abhyankar 
4942b4ed282SShri Abhyankar   ierr = VecRestoreArray(X,&x);CHKERRQ(ierr);
4952b4ed282SShri Abhyankar   ierr = VecRestoreArray(lb,&l);CHKERRQ(ierr);
4962b4ed282SShri Abhyankar   ierr = VecRestoreArray(ub,&u);CHKERRQ(ierr);
4972b4ed282SShri Abhyankar 
4982b4ed282SShri Abhyankar   PetscFunctionReturn(0);
4992b4ed282SShri Abhyankar }
5002b4ed282SShri Abhyankar 
5012b4ed282SShri Abhyankar /*  --------------------------------------------------------------------
5022b4ed282SShri Abhyankar 
5032b4ed282SShri Abhyankar      This file implements a semismooth truncated Newton method with a line search,
5042b4ed282SShri Abhyankar      for solving a system of nonlinear equations in complementarity form, using the KSP, Vec,
5052b4ed282SShri Abhyankar      and Mat interfaces for linear solvers, vectors, and matrices,
5062b4ed282SShri Abhyankar      respectively.
5072b4ed282SShri Abhyankar 
5082b4ed282SShri Abhyankar      The following basic routines are required for each nonlinear solver:
5092b4ed282SShri Abhyankar           SNESCreate_XXX()          - Creates a nonlinear solver context
5102b4ed282SShri Abhyankar           SNESSetFromOptions_XXX()  - Sets runtime options
5112b4ed282SShri Abhyankar           SNESSolve_XXX()           - Solves the nonlinear system
5122b4ed282SShri Abhyankar           SNESDestroy_XXX()         - Destroys the nonlinear solver context
5132b4ed282SShri Abhyankar      The suffix "_XXX" denotes a particular implementation, in this case
5142b4ed282SShri Abhyankar      we use _VI (e.g., SNESCreate_VI, SNESSolve_VI) for solving
5152b4ed282SShri Abhyankar      systems of nonlinear equations with a line search (LS) method.
5162b4ed282SShri Abhyankar      These routines are actually called via the common user interface
5172b4ed282SShri Abhyankar      routines SNESCreate(), SNESSetFromOptions(), SNESSolve(), and
5182b4ed282SShri Abhyankar      SNESDestroy(), so the application code interface remains identical
5192b4ed282SShri Abhyankar      for all nonlinear solvers.
5202b4ed282SShri Abhyankar 
5212b4ed282SShri Abhyankar      Another key routine is:
5222b4ed282SShri Abhyankar           SNESSetUp_XXX()           - Prepares for the use of a nonlinear solver
5232b4ed282SShri Abhyankar      by setting data structures and options.   The interface routine SNESSetUp()
5242b4ed282SShri Abhyankar      is not usually called directly by the user, but instead is called by
5252b4ed282SShri Abhyankar      SNESSolve() if necessary.
5262b4ed282SShri Abhyankar 
5272b4ed282SShri Abhyankar      Additional basic routines are:
5282b4ed282SShri Abhyankar           SNESView_XXX()            - Prints details of runtime options that
5292b4ed282SShri Abhyankar                                       have actually been used.
5302b4ed282SShri Abhyankar      These are called by application codes via the interface routines
5312b4ed282SShri Abhyankar      SNESView().
5322b4ed282SShri Abhyankar 
5332b4ed282SShri Abhyankar      The various types of solvers (preconditioners, Krylov subspace methods,
5342b4ed282SShri Abhyankar      nonlinear solvers, timesteppers) are all organized similarly, so the
5352b4ed282SShri Abhyankar      above description applies to these categories also.
5362b4ed282SShri Abhyankar 
5372b4ed282SShri Abhyankar     -------------------------------------------------------------------- */
5382b4ed282SShri Abhyankar /*
53969c03620SShri Abhyankar    SNESSolveVI_SS - Solves the complementarity problem with a semismooth Newton
5402b4ed282SShri Abhyankar    method using a line search.
5412b4ed282SShri Abhyankar 
5422b4ed282SShri Abhyankar    Input Parameters:
5432b4ed282SShri Abhyankar .  snes - the SNES context
5442b4ed282SShri Abhyankar 
5452b4ed282SShri Abhyankar    Output Parameter:
5462b4ed282SShri Abhyankar .  outits - number of iterations until termination
5472b4ed282SShri Abhyankar 
5482b4ed282SShri Abhyankar    Application Interface Routine: SNESSolve()
5492b4ed282SShri Abhyankar 
5502b4ed282SShri Abhyankar    Notes:
5512b4ed282SShri Abhyankar    This implements essentially a semismooth Newton method with a
5522b4ed282SShri Abhyankar    line search.  By default a cubic backtracking line search
5532b4ed282SShri Abhyankar    is employed, as described in the text "Numerical Methods for
5542b4ed282SShri Abhyankar    Unconstrained Optimization and Nonlinear Equations" by Dennis
5552b4ed282SShri Abhyankar    and Schnabel.
5562b4ed282SShri Abhyankar */
5572b4ed282SShri Abhyankar #undef __FUNCT__
55869c03620SShri Abhyankar #define __FUNCT__ "SNESSolveVI_SS"
55969c03620SShri Abhyankar PetscErrorCode SNESSolveVI_SS(SNES snes)
5602b4ed282SShri Abhyankar {
5612b4ed282SShri Abhyankar   SNES_VI            *vi = (SNES_VI*)snes->data;
5622b4ed282SShri Abhyankar   PetscErrorCode     ierr;
5632b4ed282SShri Abhyankar   PetscInt           maxits,i,lits;
5643b336b49SShri Abhyankar   PetscBool          lssucceed;
5652b4ed282SShri Abhyankar   MatStructure       flg = DIFFERENT_NONZERO_PATTERN;
5662b4ed282SShri Abhyankar   PetscReal          gnorm,xnorm=0,ynorm;
5672b4ed282SShri Abhyankar   Vec                Y,X,F,G,W;
5682b4ed282SShri Abhyankar   KSPConvergedReason kspreason;
5692b4ed282SShri Abhyankar 
5702b4ed282SShri Abhyankar   PetscFunctionBegin;
5712b4ed282SShri Abhyankar   snes->numFailures            = 0;
5722b4ed282SShri Abhyankar   snes->numLinearSolveFailures = 0;
5732b4ed282SShri Abhyankar   snes->reason                 = SNES_CONVERGED_ITERATING;
5742b4ed282SShri Abhyankar 
5752b4ed282SShri Abhyankar   maxits	= snes->max_its;	/* maximum number of iterations */
5762b4ed282SShri Abhyankar   X		= snes->vec_sol;	/* solution vector */
5772b4ed282SShri Abhyankar   F		= snes->vec_func;	/* residual vector */
5782b4ed282SShri Abhyankar   Y		= snes->work[0];	/* work vectors */
5792b4ed282SShri Abhyankar   G		= snes->work[1];
5802b4ed282SShri Abhyankar   W		= snes->work[2];
5812b4ed282SShri Abhyankar 
5822b4ed282SShri Abhyankar   ierr = PetscObjectTakeAccess(snes);CHKERRQ(ierr);
5832b4ed282SShri Abhyankar   snes->iter = 0;
5842b4ed282SShri Abhyankar   snes->norm = 0.0;
5852b4ed282SShri Abhyankar   ierr = PetscObjectGrantAccess(snes);CHKERRQ(ierr);
5862b4ed282SShri Abhyankar 
5872b4ed282SShri Abhyankar   ierr = SNESVIAdjustInitialGuess(X,vi->xl,vi->xu);CHKERRQ(ierr);
5882b4ed282SShri Abhyankar   ierr = SNESComputeFunction(snes,X,vi->phi);CHKERRQ(ierr);
5892b4ed282SShri Abhyankar   if (snes->domainerror) {
5902b4ed282SShri Abhyankar     snes->reason = SNES_DIVERGED_FUNCTION_DOMAIN;
5912b4ed282SShri Abhyankar     PetscFunctionReturn(0);
5922b4ed282SShri Abhyankar   }
5932b4ed282SShri Abhyankar    /* Compute Merit function */
5942b4ed282SShri Abhyankar   ierr = SNESVIComputeMeritFunction(vi->phi,&vi->merit,&vi->phinorm);CHKERRQ(ierr);
5952b4ed282SShri Abhyankar 
5962b4ed282SShri Abhyankar   ierr = VecNormBegin(X,NORM_2,&xnorm);CHKERRQ(ierr);	/* xnorm <- ||x||  */
5972b4ed282SShri Abhyankar   ierr = VecNormEnd(X,NORM_2,&xnorm);CHKERRQ(ierr);
5982b4ed282SShri Abhyankar   if PetscIsInfOrNanReal(vi->merit) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_FP,"User provided compute function generated a Not-a-Number");
5992b4ed282SShri Abhyankar 
6002b4ed282SShri Abhyankar   ierr = PetscObjectTakeAccess(snes);CHKERRQ(ierr);
6012b4ed282SShri Abhyankar   snes->norm = vi->phinorm;
6022b4ed282SShri Abhyankar   ierr = PetscObjectGrantAccess(snes);CHKERRQ(ierr);
6032b4ed282SShri Abhyankar   SNESLogConvHistory(snes,vi->phinorm,0);
6042b4ed282SShri Abhyankar   SNESMonitor(snes,0,vi->phinorm);
6052b4ed282SShri Abhyankar 
6062b4ed282SShri Abhyankar   /* set parameter for default relative tolerance convergence test */
6072b4ed282SShri Abhyankar   snes->ttol = vi->phinorm*snes->rtol;
6082b4ed282SShri Abhyankar   /* test convergence */
6092b4ed282SShri Abhyankar   ierr = (*snes->ops->converged)(snes,0,0.0,0.0,vi->phinorm,&snes->reason,snes->cnvP);CHKERRQ(ierr);
6102b4ed282SShri Abhyankar   if (snes->reason) PetscFunctionReturn(0);
6112b4ed282SShri Abhyankar 
6122b4ed282SShri Abhyankar   for (i=0; i<maxits; i++) {
6132b4ed282SShri Abhyankar 
6142b4ed282SShri Abhyankar     /* Call general purpose update function */
6152b4ed282SShri Abhyankar     if (snes->ops->update) {
6162b4ed282SShri Abhyankar       ierr = (*snes->ops->update)(snes, snes->iter);CHKERRQ(ierr);
6172b4ed282SShri Abhyankar     }
6182b4ed282SShri Abhyankar 
6192b4ed282SShri Abhyankar     /* Solve J Y = Phi, where J is the semismooth jacobian */
620a79edbefSShri Abhyankar     /* Get the nonlinear function jacobian */
6212b4ed282SShri Abhyankar     ierr = SNESComputeJacobian(snes,X,&snes->jacobian,&snes->jacobian_pre,&flg);CHKERRQ(ierr);
622a79edbefSShri Abhyankar     /* Get the diagonal shift and row scaling vectors */
623a79edbefSShri Abhyankar     ierr = SNESVIComputeBsubdifferentialVectors(snes,X,F,snes->jacobian,vi->Da,vi->Db);CHKERRQ(ierr);
624a79edbefSShri Abhyankar     /* Compute the semismooth jacobian */
625a79edbefSShri Abhyankar     ierr = SNESVIComputeJacobian(snes->jacobian,snes->jacobian_pre,vi->Da,vi->Db);CHKERRQ(ierr);
626ee657d29SShri Abhyankar     /* Compute the merit function gradient */
627ee657d29SShri Abhyankar     ierr = SNESVIComputeMeritFunctionGradient(snes->jacobian,vi->phi,vi->dpsi);CHKERRQ(ierr);
6282b4ed282SShri Abhyankar     ierr = KSPSetOperators(snes->ksp,snes->jacobian,snes->jacobian_pre,flg);CHKERRQ(ierr);
6292b4ed282SShri Abhyankar     ierr = SNES_KSPSolve(snes,snes->ksp,vi->phi,Y);CHKERRQ(ierr);
6302b4ed282SShri Abhyankar     ierr = KSPGetConvergedReason(snes->ksp,&kspreason);CHKERRQ(ierr);
6313b336b49SShri Abhyankar 
6323b336b49SShri Abhyankar     if (kspreason < 0) {
6332b4ed282SShri Abhyankar       if (++snes->numLinearSolveFailures >= snes->maxLinearSolveFailures) {
6342b4ed282SShri Abhyankar         ierr = PetscInfo2(snes,"iter=%D, number linear solve failures %D greater than current SNES allowed, stopping solve\n",snes->iter,snes->numLinearSolveFailures);CHKERRQ(ierr);
6352b4ed282SShri Abhyankar         snes->reason = SNES_DIVERGED_LINEAR_SOLVE;
6362b4ed282SShri Abhyankar         break;
6372b4ed282SShri Abhyankar       }
6382b4ed282SShri Abhyankar     }
6392b4ed282SShri Abhyankar     ierr = KSPGetIterationNumber(snes->ksp,&lits);CHKERRQ(ierr);
6402b4ed282SShri Abhyankar     snes->linear_its += lits;
6412b4ed282SShri Abhyankar     ierr = PetscInfo2(snes,"iter=%D, linear solve iterations=%D\n",snes->iter,lits);CHKERRQ(ierr);
6422b4ed282SShri Abhyankar     /*
6432b4ed282SShri Abhyankar     if (vi->precheckstep) {
644ace3abfcSBarry Smith       PetscBool changed_y = PETSC_FALSE;
6452b4ed282SShri Abhyankar       ierr = (*vi->precheckstep)(snes,X,Y,vi->precheck,&changed_y);CHKERRQ(ierr);
6462b4ed282SShri Abhyankar     }
6472b4ed282SShri Abhyankar 
6482b4ed282SShri Abhyankar     if (PetscLogPrintInfo){
6492b4ed282SShri Abhyankar       ierr = SNESVICheckResidual_Private(snes,snes->jacobian,F,Y,G,W);CHKERRQ(ierr);
6502b4ed282SShri Abhyankar     }
6512b4ed282SShri Abhyankar     */
6522b4ed282SShri Abhyankar     /* Compute a (scaled) negative update in the line search routine:
6532b4ed282SShri Abhyankar          Y <- X - lambda*Y
6542b4ed282SShri Abhyankar        and evaluate G = function(Y) (depends on the line search).
6552b4ed282SShri Abhyankar     */
6562b4ed282SShri Abhyankar     ierr = VecCopy(Y,snes->vec_sol_update);CHKERRQ(ierr);
6572b4ed282SShri Abhyankar     ynorm = 1; gnorm = vi->phinorm;
6582b4ed282SShri Abhyankar     ierr = (*vi->LineSearch)(snes,vi->lsP,X,vi->phi,G,Y,W,vi->phinorm,xnorm,&ynorm,&gnorm,&lssucceed);CHKERRQ(ierr);
6592b4ed282SShri Abhyankar     ierr = PetscInfo4(snes,"fnorm=%18.16e, gnorm=%18.16e, ynorm=%18.16e, lssucceed=%d\n",vi->phinorm,gnorm,ynorm,(int)lssucceed);CHKERRQ(ierr);
6602b4ed282SShri Abhyankar     if (snes->reason == SNES_DIVERGED_FUNCTION_COUNT) break;
6612b4ed282SShri Abhyankar     if (snes->domainerror) {
6622b4ed282SShri Abhyankar       snes->reason = SNES_DIVERGED_FUNCTION_DOMAIN;
6632b4ed282SShri Abhyankar       PetscFunctionReturn(0);
6642b4ed282SShri Abhyankar     }
6652b4ed282SShri Abhyankar     if (!lssucceed) {
6662b4ed282SShri Abhyankar       if (++snes->numFailures >= snes->maxFailures) {
667ace3abfcSBarry Smith 	PetscBool ismin;
6682b4ed282SShri Abhyankar         snes->reason = SNES_DIVERGED_LINE_SEARCH;
6692b4ed282SShri Abhyankar         ierr = SNESVICheckLocalMin_Private(snes,snes->jacobian,G,W,gnorm,&ismin);CHKERRQ(ierr);
6702b4ed282SShri Abhyankar         if (ismin) snes->reason = SNES_DIVERGED_LOCAL_MIN;
6712b4ed282SShri Abhyankar         break;
6722b4ed282SShri Abhyankar       }
6732b4ed282SShri Abhyankar     }
6742b4ed282SShri Abhyankar     /* Update function and solution vectors */
6752b4ed282SShri Abhyankar     vi->phinorm = gnorm;
6762b4ed282SShri Abhyankar     vi->merit = 0.5*vi->phinorm*vi->phinorm;
6772b4ed282SShri Abhyankar     ierr = VecCopy(G,vi->phi);CHKERRQ(ierr);
6782b4ed282SShri Abhyankar     ierr = VecCopy(W,X);CHKERRQ(ierr);
6792b4ed282SShri Abhyankar     /* Monitor convergence */
6802b4ed282SShri Abhyankar     ierr = PetscObjectTakeAccess(snes);CHKERRQ(ierr);
6812b4ed282SShri Abhyankar     snes->iter = i+1;
6822b4ed282SShri Abhyankar     snes->norm = vi->phinorm;
6832b4ed282SShri Abhyankar     ierr = PetscObjectGrantAccess(snes);CHKERRQ(ierr);
6842b4ed282SShri Abhyankar     SNESLogConvHistory(snes,snes->norm,lits);
6852b4ed282SShri Abhyankar     SNESMonitor(snes,snes->iter,snes->norm);
6862b4ed282SShri Abhyankar     /* Test for convergence, xnorm = || X || */
6872b4ed282SShri Abhyankar     if (snes->ops->converged != SNESSkipConverged) { ierr = VecNorm(X,NORM_2,&xnorm);CHKERRQ(ierr); }
6882b4ed282SShri Abhyankar     ierr = (*snes->ops->converged)(snes,snes->iter,xnorm,ynorm,vi->phinorm,&snes->reason,snes->cnvP);CHKERRQ(ierr);
6892b4ed282SShri Abhyankar     if (snes->reason) break;
6902b4ed282SShri Abhyankar   }
6912b4ed282SShri Abhyankar   if (i == maxits) {
6922b4ed282SShri Abhyankar     ierr = PetscInfo1(snes,"Maximum number of iterations has been reached: %D\n",maxits);CHKERRQ(ierr);
6932b4ed282SShri Abhyankar     if(!snes->reason) snes->reason = SNES_DIVERGED_MAX_IT;
6942b4ed282SShri Abhyankar   }
6952b4ed282SShri Abhyankar   PetscFunctionReturn(0);
6962b4ed282SShri Abhyankar }
69769c03620SShri Abhyankar 
69869c03620SShri Abhyankar #undef __FUNCT__
69969c03620SShri Abhyankar #define __FUNCT__ "SNESVICreateIndexSets_AS"
700a79edbefSShri Abhyankar PetscErrorCode SNESVICreateIndexSets_AS(SNES snes,Vec Db,PetscReal thresh,IS* ISact,IS* ISinact)
70169c03620SShri Abhyankar {
70269c03620SShri Abhyankar   PetscErrorCode ierr;
70369c03620SShri Abhyankar   PetscInt       i,nlocal,ilow,ihigh,nloc_isact=0,nloc_isinact=0;
70469c03620SShri Abhyankar   PetscInt       *idx_act,*idx_inact,i1=0,i2=0;
70569c03620SShri Abhyankar   PetscScalar    *db;
70669c03620SShri Abhyankar 
70769c03620SShri Abhyankar   PetscFunctionBegin;
70869c03620SShri Abhyankar 
70969c03620SShri Abhyankar   ierr = VecGetLocalSize(Db,&nlocal);CHKERRQ(ierr);
71069c03620SShri Abhyankar   ierr = VecGetOwnershipRange(Db,&ilow,&ihigh);CHKERRQ(ierr);
71169c03620SShri Abhyankar   ierr = VecGetArray(Db,&db);CHKERRQ(ierr);
71269c03620SShri Abhyankar   /* Compute the sizes of the active and inactive sets */
71369c03620SShri Abhyankar   for (i=0; i < nlocal;i++) {
714fca35906SShri Abhyankar     if (PetscAbsScalar(db[i]) <= thresh) nloc_isact++;
715fca35906SShri Abhyankar     else nloc_isinact++;
71669c03620SShri Abhyankar   }
71769c03620SShri Abhyankar   ierr = PetscMalloc(nloc_isact*sizeof(PetscInt),&idx_act);CHKERRQ(ierr);
71869c03620SShri Abhyankar   ierr = PetscMalloc(nloc_isinact*sizeof(PetscInt),&idx_inact);CHKERRQ(ierr);
71969c03620SShri Abhyankar 
72069c03620SShri Abhyankar   /* Creating the indexing arrays */
721fca35906SShri Abhyankar   for(i=0; i < nlocal; i++) {
722fca35906SShri Abhyankar     if (PetscAbsScalar(db[i]) <= thresh) idx_act[i1++] = ilow+i;
723fca35906SShri Abhyankar     else idx_inact[i2++] = ilow+i;
72469c03620SShri Abhyankar   }
72569c03620SShri Abhyankar 
72669c03620SShri Abhyankar   /* Create the index sets */
72769c03620SShri Abhyankar   ierr = ISCreateGeneral(((PetscObject)snes)->comm,nloc_isact,idx_act,PETSC_COPY_VALUES,ISact);CHKERRQ(ierr);
72869c03620SShri Abhyankar   ierr = ISCreateGeneral(((PetscObject)snes)->comm,nloc_isinact,idx_inact,PETSC_COPY_VALUES,ISinact);CHKERRQ(ierr);
72969c03620SShri Abhyankar 
73069c03620SShri Abhyankar   ierr = VecRestoreArray(Db,&db);CHKERRQ(ierr);
73169c03620SShri Abhyankar   ierr = PetscFree(idx_act);CHKERRQ(ierr);
73269c03620SShri Abhyankar   ierr = PetscFree(idx_inact);CHKERRQ(ierr);
73369c03620SShri Abhyankar   PetscFunctionReturn(0);
73469c03620SShri Abhyankar }
73569c03620SShri Abhyankar 
736d950fb63SShri Abhyankar #undef __FUNCT__
737d950fb63SShri Abhyankar #define __FUNCT__ "SNESVICreateIndexSets_RS"
738d950fb63SShri Abhyankar PetscErrorCode SNESVICreateIndexSets_RS(SNES snes,Vec X,Vec Xl, Vec Xu,IS* ISact,IS* ISinact)
739d950fb63SShri Abhyankar {
740d950fb63SShri Abhyankar   PetscErrorCode     ierr;
741d950fb63SShri Abhyankar   PetscInt           i,nlocal,ilow,ihigh,nloc_isact=0,nloc_isinact=0;
742d950fb63SShri Abhyankar   PetscInt          *idx_act,*idx_inact,i1=0,i2=0;
743*fe835674SShri Abhyankar   const PetscScalar *x,*xl,*xu,*f;
744d950fb63SShri Abhyankar   Vec                F = snes->vec_func;
745d950fb63SShri Abhyankar 
746d950fb63SShri Abhyankar   PetscFunctionBegin;
747d950fb63SShri Abhyankar 
748d950fb63SShri Abhyankar   ierr = VecGetLocalSize(X,&nlocal);CHKERRQ(ierr);
749d950fb63SShri Abhyankar   ierr = VecGetOwnershipRange(X,&ilow,&ihigh);CHKERRQ(ierr);
750*fe835674SShri Abhyankar   ierr = VecGetArrayRead(X,&x);CHKERRQ(ierr);
751*fe835674SShri Abhyankar   ierr = VecGetArrayRead(Xl,&xl);CHKERRQ(ierr);
752*fe835674SShri Abhyankar   ierr = VecGetArrayRead(Xu,&xu);CHKERRQ(ierr);
753*fe835674SShri Abhyankar   ierr = VecGetArrayRead(F,&f);CHKERRQ(ierr);
754d950fb63SShri Abhyankar   /* Compute the sizes of the active and inactive sets */
755d950fb63SShri Abhyankar   for (i=0; i < nlocal;i++) {
756*fe835674SShri Abhyankar     if (((x[i] > xl[i] + 1.e-8 || (f[i] < 0.0)) && ((x[i] < xu[i] - 1.e-8) || f[i] > 0.0))) nloc_isinact++;
757*fe835674SShri Abhyankar     else nloc_isact++;
758d950fb63SShri Abhyankar   }
759d950fb63SShri Abhyankar   ierr = PetscMalloc(nloc_isact*sizeof(PetscInt),&idx_act);CHKERRQ(ierr);
760d950fb63SShri Abhyankar   ierr = PetscMalloc(nloc_isinact*sizeof(PetscInt),&idx_inact);CHKERRQ(ierr);
761d950fb63SShri Abhyankar 
762d950fb63SShri Abhyankar   /* Creating the indexing arrays */
763d950fb63SShri Abhyankar   for(i=0; i < nlocal; i++) {
764*fe835674SShri Abhyankar     if (((x[i] > xl[i] + 1.e-8 || (f[i] < 0.0)) && ((x[i] < xu[i] - 1.e-8) || f[i] > 0.0))) idx_inact[i2++] = ilow+i;
765*fe835674SShri Abhyankar     else idx_act[i1++] = ilow+i;
766d950fb63SShri Abhyankar   }
767d950fb63SShri Abhyankar 
768d950fb63SShri Abhyankar   /* Create the index sets */
769d950fb63SShri Abhyankar   ierr = ISCreateGeneral(((PetscObject)snes)->comm,nloc_isact,idx_act,PETSC_COPY_VALUES,ISact);CHKERRQ(ierr);
770d950fb63SShri Abhyankar   ierr = ISCreateGeneral(((PetscObject)snes)->comm,nloc_isinact,idx_inact,PETSC_COPY_VALUES,ISinact);CHKERRQ(ierr);
771d950fb63SShri Abhyankar 
772*fe835674SShri Abhyankar   ierr = VecRestoreArrayRead(X,&x);CHKERRQ(ierr);
773*fe835674SShri Abhyankar   ierr = VecRestoreArrayRead(Xl,&xl);CHKERRQ(ierr);
774*fe835674SShri Abhyankar   ierr = VecRestoreArrayRead(Xu,&xu);CHKERRQ(ierr);
775*fe835674SShri Abhyankar   ierr = VecRestoreArrayRead(F,&f);CHKERRQ(ierr);
776d950fb63SShri Abhyankar   ierr = PetscFree(idx_act);CHKERRQ(ierr);
777d950fb63SShri Abhyankar   ierr = PetscFree(idx_inact);CHKERRQ(ierr);
778d950fb63SShri Abhyankar   PetscFunctionReturn(0);
779d950fb63SShri Abhyankar }
780d950fb63SShri Abhyankar 
781dbd914b8SShri Abhyankar /* Create active and inactive set vectors. The local size of this vector is set and petsc computes the global size */
782dbd914b8SShri Abhyankar #undef __FUNCT__
783*fe835674SShri Abhyankar #define __FUNCT__ "SNESVICreateSubVectors"
784*fe835674SShri Abhyankar PetscErrorCode SNESVICreateSubVectors(SNES snes,PetscInt n,Vec* newv)
785dbd914b8SShri Abhyankar {
786dbd914b8SShri Abhyankar   PetscErrorCode ierr;
787dbd914b8SShri Abhyankar   Vec            v;
788dbd914b8SShri Abhyankar 
789dbd914b8SShri Abhyankar   PetscFunctionBegin;
790dbd914b8SShri Abhyankar   ierr = VecCreate(((PetscObject)snes)->comm,&v);CHKERRQ(ierr);
791dbd914b8SShri Abhyankar   ierr = VecSetSizes(v,n,PETSC_DECIDE);CHKERRQ(ierr);
792dbd914b8SShri Abhyankar   ierr = VecSetFromOptions(v);CHKERRQ(ierr);
793dbd914b8SShri Abhyankar   *newv = v;
794dbd914b8SShri Abhyankar 
795dbd914b8SShri Abhyankar   PetscFunctionReturn(0);
796dbd914b8SShri Abhyankar }
797dbd914b8SShri Abhyankar 
798732bb160SShri Abhyankar /* Resets the snes PC and KSP when the active set sizes change */
799732bb160SShri Abhyankar #undef __FUNCT__
800732bb160SShri Abhyankar #define __FUNCT__ "SNESVIResetPCandKSP"
801732bb160SShri Abhyankar PetscErrorCode SNESVIResetPCandKSP(SNES snes,Mat Amat,Mat Pmat)
802732bb160SShri Abhyankar {
803732bb160SShri Abhyankar   PetscErrorCode ierr;
804732bb160SShri Abhyankar   KSP kspnew,snesksp;
805732bb160SShri Abhyankar   PC  pcnew;
806732bb160SShri Abhyankar   const MatSolverPackage stype;
807dbd914b8SShri Abhyankar 
808732bb160SShri Abhyankar   PetscFunctionBegin;
809732bb160SShri Abhyankar   /* The active and inactive set sizes have changed so need to create a new snes->ksp object */
810732bb160SShri Abhyankar   ierr = SNESGetKSP(snes,&snesksp);CHKERRQ(ierr);
811732bb160SShri Abhyankar   ierr = KSPCreate(((PetscObject)snes)->comm,&kspnew);CHKERRQ(ierr);
812732bb160SShri Abhyankar   /* Copy over snes->ksp info */
813732bb160SShri Abhyankar   kspnew->pc_side = snesksp->pc_side;
814732bb160SShri Abhyankar   kspnew->rtol    = snesksp->rtol;
815732bb160SShri Abhyankar   kspnew->abstol    = snesksp->abstol;
816732bb160SShri Abhyankar   kspnew->max_it  = snesksp->max_it;
817732bb160SShri Abhyankar   ierr = KSPSetType(kspnew,((PetscObject)snesksp)->type_name);CHKERRQ(ierr);
818732bb160SShri Abhyankar   ierr = KSPGetPC(kspnew,&pcnew);CHKERRQ(ierr);
819732bb160SShri Abhyankar   ierr = PCSetType(kspnew->pc,((PetscObject)snesksp->pc)->type_name);CHKERRQ(ierr);
820732bb160SShri Abhyankar   ierr = PCSetOperators(kspnew->pc,Amat,Pmat,DIFFERENT_NONZERO_PATTERN);CHKERRQ(ierr);
821732bb160SShri Abhyankar   ierr = PCFactorGetMatSolverPackage(snesksp->pc,&stype);CHKERRQ(ierr);
822732bb160SShri Abhyankar   ierr = PCFactorSetMatSolverPackage(kspnew->pc,stype);CHKERRQ(ierr);
823732bb160SShri Abhyankar   ierr = KSPDestroy(snesksp);CHKERRQ(ierr);
824732bb160SShri Abhyankar   snes->ksp = kspnew;
825732bb160SShri Abhyankar   ierr = PetscLogObjectParent(snes,kspnew);CHKERRQ(ierr);
826732bb160SShri Abhyankar   ierr = KSPSetFromOptions(kspnew);CHKERRQ(ierr);
827732bb160SShri Abhyankar   PetscFunctionReturn(0);
828732bb160SShri Abhyankar }
82969c03620SShri Abhyankar /* Variational Inequality solver using active set method */
83069c03620SShri Abhyankar #undef __FUNCT__
83169c03620SShri Abhyankar #define __FUNCT__ "SNESSolveVI_AS"
83269c03620SShri Abhyankar PetscErrorCode SNESSolveVI_AS(SNES snes)
83369c03620SShri Abhyankar {
83469c03620SShri Abhyankar   SNES_VI          *vi = (SNES_VI*)snes->data;
83569c03620SShri Abhyankar   PetscErrorCode     ierr;
836408da460SShri Abhyankar   PetscInt           maxits,i,lits,Nis_act=0;
8373b336b49SShri Abhyankar   PetscBool         lssucceed;
83869c03620SShri Abhyankar   MatStructure       flg = DIFFERENT_NONZERO_PATTERN;
83969c03620SShri Abhyankar   PetscReal          gnorm,xnorm=0,ynorm;
84069c03620SShri Abhyankar   Vec                Y,X,F,G,W;
84169c03620SShri Abhyankar   KSPConvergedReason kspreason;
84269c03620SShri Abhyankar 
84369c03620SShri Abhyankar   PetscFunctionBegin;
84469c03620SShri Abhyankar   snes->numFailures            = 0;
84569c03620SShri Abhyankar   snes->numLinearSolveFailures = 0;
84669c03620SShri Abhyankar   snes->reason                 = SNES_CONVERGED_ITERATING;
84769c03620SShri Abhyankar 
84869c03620SShri Abhyankar   maxits	= snes->max_its;	/* maximum number of iterations */
84969c03620SShri Abhyankar   X		= snes->vec_sol;	/* solution vector */
85069c03620SShri Abhyankar   F		= snes->vec_func;	/* residual vector */
85169c03620SShri Abhyankar   Y		= snes->work[0];	/* work vectors */
85269c03620SShri Abhyankar   G		= snes->work[1];
85369c03620SShri Abhyankar   W		= snes->work[2];
85469c03620SShri Abhyankar 
85569c03620SShri Abhyankar   ierr = PetscObjectTakeAccess(snes);CHKERRQ(ierr);
85669c03620SShri Abhyankar   snes->iter = 0;
85769c03620SShri Abhyankar   snes->norm = 0.0;
85869c03620SShri Abhyankar   ierr = PetscObjectGrantAccess(snes);CHKERRQ(ierr);
85969c03620SShri Abhyankar 
86069c03620SShri Abhyankar   ierr = SNESVIAdjustInitialGuess(X,vi->xl,vi->xu);CHKERRQ(ierr);
86169c03620SShri Abhyankar   ierr = SNESComputeFunction(snes,X,vi->phi);CHKERRQ(ierr);
86269c03620SShri Abhyankar   if (snes->domainerror) {
86369c03620SShri Abhyankar     snes->reason = SNES_DIVERGED_FUNCTION_DOMAIN;
86469c03620SShri Abhyankar     PetscFunctionReturn(0);
86569c03620SShri Abhyankar   }
86669c03620SShri Abhyankar    /* Compute Merit function */
86769c03620SShri Abhyankar   ierr = SNESVIComputeMeritFunction(vi->phi,&vi->merit,&vi->phinorm);CHKERRQ(ierr);
86869c03620SShri Abhyankar 
86969c03620SShri Abhyankar   ierr = VecNormBegin(X,NORM_2,&xnorm);CHKERRQ(ierr);	/* xnorm <- ||x||  */
87069c03620SShri Abhyankar   ierr = VecNormEnd(X,NORM_2,&xnorm);CHKERRQ(ierr);
87169c03620SShri Abhyankar   if PetscIsInfOrNanReal(vi->merit) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_FP,"User provided compute function generated a Not-a-Number");
87269c03620SShri Abhyankar 
87369c03620SShri Abhyankar   ierr = PetscObjectTakeAccess(snes);CHKERRQ(ierr);
87469c03620SShri Abhyankar   snes->norm = vi->phinorm;
87569c03620SShri Abhyankar   ierr = PetscObjectGrantAccess(snes);CHKERRQ(ierr);
87669c03620SShri Abhyankar   SNESLogConvHistory(snes,vi->phinorm,0);
87769c03620SShri Abhyankar   SNESMonitor(snes,0,vi->phinorm);
87869c03620SShri Abhyankar 
87969c03620SShri Abhyankar   /* set parameter for default relative tolerance convergence test */
88069c03620SShri Abhyankar   snes->ttol = vi->phinorm*snes->rtol;
88169c03620SShri Abhyankar   /* test convergence */
88269c03620SShri Abhyankar   ierr = (*snes->ops->converged)(snes,0,0.0,0.0,vi->phinorm,&snes->reason,snes->cnvP);CHKERRQ(ierr);
88369c03620SShri Abhyankar   if (snes->reason) PetscFunctionReturn(0);
88469c03620SShri Abhyankar 
88569c03620SShri Abhyankar   for (i=0; i<maxits; i++) {
88669c03620SShri Abhyankar 
887a79edbefSShri Abhyankar     IS                 IS_act,IS_inact; /* _act -> active set _inact -> inactive set */
888a79edbefSShri Abhyankar     PetscReal          thresh,J_norm1;
889dbd914b8SShri Abhyankar     VecScatter         scat_act,scat_inact;
890408da460SShri Abhyankar     PetscInt           nis_act,nis_inact,Nis_act_prev;
891dbd914b8SShri Abhyankar     Vec                Da_act,Da_inact,Db_inact;
892dbd914b8SShri Abhyankar     Vec                Y_act,Y_inact,phi_act,phi_inact;
893d76c674bSShri Abhyankar     Mat                jac_inact_inact,jac_inact_act,prejac_inact_inact;
894a79edbefSShri Abhyankar 
89569c03620SShri Abhyankar     /* Call general purpose update function */
89669c03620SShri Abhyankar     if (snes->ops->update) {
89769c03620SShri Abhyankar       ierr = (*snes->ops->update)(snes, snes->iter);CHKERRQ(ierr);
89869c03620SShri Abhyankar     }
89969c03620SShri Abhyankar     ierr = SNESComputeJacobian(snes,X,&snes->jacobian,&snes->jacobian_pre,&flg);CHKERRQ(ierr);
90069c03620SShri Abhyankar     /* Compute the threshold value for creating active and inactive sets */
90169c03620SShri Abhyankar     ierr = MatNorm(snes->jacobian,NORM_1,&J_norm1);CHKERRQ(ierr);
90269c03620SShri Abhyankar     thresh = PetscMin(vi->merit,1e-2)/(1+J_norm1);
903a79edbefSShri Abhyankar 
904a79edbefSShri Abhyankar     /* Compute B-subdifferential vectors Da and Db */
905a79edbefSShri Abhyankar     ierr = SNESVIComputeBsubdifferentialVectors(snes,X,F,snes->jacobian,vi->Da,vi->Db);CHKERRQ(ierr);
906dbd914b8SShri Abhyankar 
90769c03620SShri Abhyankar     /* Create active and inactive index sets */
90869c03620SShri Abhyankar     ierr = SNESVICreateIndexSets_AS(snes,vi->Db,thresh,&IS_act,&IS_inact);CHKERRQ(ierr);
90969c03620SShri Abhyankar 
910408da460SShri Abhyankar     Nis_act_prev = Nis_act;
911408da460SShri Abhyankar     /* Get sizes of active and inactive sets */
912dbd914b8SShri Abhyankar     ierr = ISGetLocalSize(IS_act,&nis_act);CHKERRQ(ierr);
913dbd914b8SShri Abhyankar     ierr = ISGetLocalSize(IS_inact,&nis_inact);CHKERRQ(ierr);
914408da460SShri Abhyankar     ierr = ISGetSize(IS_act,&Nis_act);CHKERRQ(ierr);
915dbd914b8SShri Abhyankar 
916d76c674bSShri Abhyankar     ierr = PetscPrintf(PETSC_COMM_WORLD,"Size of active set = %d, size of inactive set = %d\n",nis_act,nis_inact);CHKERRQ(ierr);
917d76c674bSShri Abhyankar 
918dbd914b8SShri Abhyankar     /* Create active and inactive set vectors */
919*fe835674SShri Abhyankar     ierr = SNESVICreateSubVectors(snes,nis_act,&Da_act);CHKERRQ(ierr);
920*fe835674SShri Abhyankar     ierr = SNESVICreateSubVectors(snes,nis_inact,&Da_inact);CHKERRQ(ierr);
921*fe835674SShri Abhyankar     ierr = SNESVICreateSubVectors(snes,nis_inact,&Db_inact);CHKERRQ(ierr);
922*fe835674SShri Abhyankar     ierr = SNESVICreateSubVectors(snes,nis_act,&phi_act);CHKERRQ(ierr);
923*fe835674SShri Abhyankar     ierr = SNESVICreateSubVectors(snes,nis_inact,&phi_inact);CHKERRQ(ierr);
924*fe835674SShri Abhyankar     ierr = SNESVICreateSubVectors(snes,nis_act,&Y_act);CHKERRQ(ierr);
925*fe835674SShri Abhyankar     ierr = SNESVICreateSubVectors(snes,nis_inact,&Y_inact);CHKERRQ(ierr);
926dbd914b8SShri Abhyankar 
927dbd914b8SShri Abhyankar     /* Create inactive set submatrices */
928dbd914b8SShri Abhyankar     ierr = MatGetSubMatrix(snes->jacobian,IS_inact,IS_act,MAT_INITIAL_MATRIX,&jac_inact_act);CHKERRQ(ierr);
929dbd914b8SShri Abhyankar     ierr = MatGetSubMatrix(snes->jacobian,IS_inact,IS_inact,MAT_INITIAL_MATRIX,&jac_inact_inact);CHKERRQ(ierr);
930dbd914b8SShri Abhyankar 
931dbd914b8SShri Abhyankar     /* Create scatter contexts */
932dbd914b8SShri Abhyankar     ierr = VecScatterCreate(vi->Da,IS_act,Da_act,PETSC_NULL,&scat_act);CHKERRQ(ierr);
933dbd914b8SShri Abhyankar     ierr = VecScatterCreate(vi->Da,IS_inact,Da_inact,PETSC_NULL,&scat_inact);CHKERRQ(ierr);
934dbd914b8SShri Abhyankar 
935dbd914b8SShri Abhyankar     /* Do a vec scatter to active and inactive set vectors */
936dbd914b8SShri Abhyankar     ierr = VecScatterBegin(scat_act,vi->Da,Da_act,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
937dbd914b8SShri Abhyankar     ierr = VecScatterEnd(scat_act,vi->Da,Da_act,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
938dbd914b8SShri Abhyankar 
939dbd914b8SShri Abhyankar     ierr = VecScatterBegin(scat_inact,vi->Da,Da_inact,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
940dbd914b8SShri Abhyankar     ierr = VecScatterEnd(scat_inact,vi->Da,Da_inact,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
941dbd914b8SShri Abhyankar 
942dbd914b8SShri Abhyankar     ierr = VecScatterBegin(scat_inact,vi->Db,Db_inact,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
943dbd914b8SShri Abhyankar     ierr = VecScatterEnd(scat_inact,vi->Db,Db_inact,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
944dbd914b8SShri Abhyankar 
945dbd914b8SShri Abhyankar     ierr = VecScatterBegin(scat_act,vi->phi,phi_act,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
946dbd914b8SShri Abhyankar     ierr = VecScatterEnd(scat_act,vi->phi,phi_act,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
947dbd914b8SShri Abhyankar 
948dbd914b8SShri Abhyankar     ierr = VecScatterBegin(scat_inact,vi->phi,phi_inact,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
949dbd914b8SShri Abhyankar     ierr = VecScatterEnd(scat_inact,vi->phi,phi_inact,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
950dbd914b8SShri Abhyankar 
951dbd914b8SShri Abhyankar     ierr = VecScatterBegin(scat_act,Y,Y_act,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
952dbd914b8SShri Abhyankar     ierr = VecScatterEnd(scat_act,Y,Y_act,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
953dbd914b8SShri Abhyankar 
954dbd914b8SShri Abhyankar     ierr = VecScatterBegin(scat_inact,Y,Y_inact,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
955dbd914b8SShri Abhyankar     ierr = VecScatterEnd(scat_inact,Y,Y_inact,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
956dbd914b8SShri Abhyankar 
957dbd914b8SShri Abhyankar     /* Active set direction */
958dbd914b8SShri Abhyankar     ierr = VecPointwiseDivide(Y_act,phi_act,Da_act);CHKERRQ(ierr);
959d76c674bSShri Abhyankar     /* inactive set jacobian and preconditioner */
960dbd914b8SShri Abhyankar     ierr = VecPointwiseDivide(Da_inact,Da_inact,Db_inact);CHKERRQ(ierr);
961dbd914b8SShri Abhyankar     ierr = MatDiagonalSet(jac_inact_inact,Da_inact,ADD_VALUES);CHKERRQ(ierr);
962d76c674bSShri Abhyankar     if (snes->jacobian != snes->jacobian_pre) {
963d76c674bSShri Abhyankar       ierr = MatGetSubMatrix(snes->jacobian_pre,IS_inact,IS_inact,MAT_INITIAL_MATRIX,&prejac_inact_inact);CHKERRQ(ierr);
964d76c674bSShri Abhyankar       ierr = MatDiagonalSet(prejac_inact_inact,Da_inact,ADD_VALUES);CHKERRQ(ierr);
965d76c674bSShri Abhyankar     } else prejac_inact_inact = jac_inact_inact;
966d76c674bSShri Abhyankar 
967dbd914b8SShri Abhyankar     /* right hand side */
968dbd914b8SShri Abhyankar     ierr = VecPointwiseDivide(phi_inact,phi_inact,Db_inact);CHKERRQ(ierr);
969dbd914b8SShri Abhyankar     ierr = MatMult(jac_inact_act,Y_act,Db_inact);CHKERRQ(ierr);
970dbd914b8SShri Abhyankar     ierr = VecAXPY(phi_inact,-1.0,Db_inact);CHKERRQ(ierr);
971dbd914b8SShri Abhyankar 
972408da460SShri Abhyankar     if ((i != 0) && (Nis_act != Nis_act_prev)) {
973732bb160SShri Abhyankar       ierr = SNESVIResetPCandKSP(snes,jac_inact_inact,prejac_inact_inact);CHKERRQ(ierr);
974408da460SShri Abhyankar     }
975408da460SShri Abhyankar 
976d76c674bSShri Abhyankar     ierr = KSPSetOperators(snes->ksp,jac_inact_inact,prejac_inact_inact,flg);CHKERRQ(ierr);
977dbd914b8SShri Abhyankar     ierr = SNES_KSPSolve(snes,snes->ksp,phi_inact,Y_inact);CHKERRQ(ierr);
978dbd914b8SShri Abhyankar     ierr = KSPGetConvergedReason(snes->ksp,&kspreason);CHKERRQ(ierr);
9793b336b49SShri Abhyankar     if (kspreason < 0) {
9803b336b49SShri Abhyankar       if (++snes->numLinearSolveFailures >= snes->maxLinearSolveFailures) {
9813b336b49SShri Abhyankar         ierr = PetscInfo2(snes,"iter=%D, number linear solve failures %D greater than current SNES allowed, stopping solve\n",snes->iter,snes->numLinearSolveFailures);CHKERRQ(ierr);
9823b336b49SShri Abhyankar         snes->reason = SNES_DIVERGED_LINEAR_SOLVE;
9833b336b49SShri Abhyankar         break;
9843b336b49SShri Abhyankar       }
9853b336b49SShri Abhyankar      }
986dbd914b8SShri Abhyankar 
987dbd914b8SShri Abhyankar     ierr = VecScatterBegin(scat_act,Y_act,Y,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr);
988dbd914b8SShri Abhyankar     ierr = VecScatterEnd(scat_act,Y_act,Y,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr);
989dbd914b8SShri Abhyankar     ierr = VecScatterBegin(scat_inact,Y_inact,Y,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr);
990dbd914b8SShri Abhyankar     ierr = VecScatterEnd(scat_inact,Y_inact,Y,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr);
991dbd914b8SShri Abhyankar 
992dbd914b8SShri Abhyankar     ierr = VecDestroy(Da_act);CHKERRQ(ierr);
993dbd914b8SShri Abhyankar     ierr = VecDestroy(Da_inact);CHKERRQ(ierr);
994dbd914b8SShri Abhyankar     ierr = VecDestroy(Db_inact);CHKERRQ(ierr);
995dbd914b8SShri Abhyankar     ierr = VecDestroy(phi_act);CHKERRQ(ierr);
996dbd914b8SShri Abhyankar     ierr = VecDestroy(phi_inact);CHKERRQ(ierr);
997dbd914b8SShri Abhyankar     ierr = VecDestroy(Y_act);CHKERRQ(ierr);
998dbd914b8SShri Abhyankar     ierr = VecDestroy(Y_inact);CHKERRQ(ierr);
999dbd914b8SShri Abhyankar     ierr = VecScatterDestroy(scat_act);CHKERRQ(ierr);
1000dbd914b8SShri Abhyankar     ierr = VecScatterDestroy(scat_inact);CHKERRQ(ierr);
1001730c24dcSShri Abhyankar     ierr = ISDestroy(IS_act);CHKERRQ(ierr);
1002730c24dcSShri Abhyankar     ierr = ISDestroy(IS_inact);CHKERRQ(ierr);
1003bac2f86eSShri Abhyankar     ierr = MatDestroy(jac_inact_act);CHKERRQ(ierr);
1004bac2f86eSShri Abhyankar     ierr = MatDestroy(jac_inact_inact);CHKERRQ(ierr);
1005d76c674bSShri Abhyankar     if (snes->jacobian != snes->jacobian_pre) {
1006d76c674bSShri Abhyankar       ierr = MatDestroy(prejac_inact_inact);CHKERRQ(ierr);
1007d76c674bSShri Abhyankar     }
1008730c24dcSShri Abhyankar 
100969c03620SShri Abhyankar     ierr = KSPGetIterationNumber(snes->ksp,&lits);CHKERRQ(ierr);
101069c03620SShri Abhyankar     snes->linear_its += lits;
101169c03620SShri Abhyankar     ierr = PetscInfo2(snes,"iter=%D, linear solve iterations=%D\n",snes->iter,lits);CHKERRQ(ierr);
101269c03620SShri Abhyankar     /*
101369c03620SShri Abhyankar     if (vi->precheckstep) {
101469c03620SShri Abhyankar       PetscBool changed_y = PETSC_FALSE;
101569c03620SShri Abhyankar       ierr = (*vi->precheckstep)(snes,X,Y,vi->precheck,&changed_y);CHKERRQ(ierr);
101669c03620SShri Abhyankar     }
101769c03620SShri Abhyankar 
101869c03620SShri Abhyankar     if (PetscLogPrintInfo){
101969c03620SShri Abhyankar       ierr = SNESVICheckResidual_Private(snes,snes->jacobian,F,Y,G,W);CHKERRQ(ierr);
102069c03620SShri Abhyankar     }
102169c03620SShri Abhyankar     */
102269c03620SShri Abhyankar     /* Compute a (scaled) negative update in the line search routine:
102369c03620SShri Abhyankar          Y <- X - lambda*Y
102469c03620SShri Abhyankar        and evaluate G = function(Y) (depends on the line search).
102569c03620SShri Abhyankar     */
102669c03620SShri Abhyankar     ierr = VecCopy(Y,snes->vec_sol_update);CHKERRQ(ierr);
102769c03620SShri Abhyankar     ynorm = 1; gnorm = vi->phinorm;
10283b336b49SShri Abhyankar     ierr = SNESVIComputeJacobian(snes->jacobian,snes->jacobian_pre,vi->Da,vi->Db);CHKERRQ(ierr);
1029ee657d29SShri Abhyankar     ierr = SNESVIComputeMeritFunctionGradient(snes->jacobian,vi->phi,vi->dpsi);CHKERRQ(ierr);
103069c03620SShri Abhyankar     ierr = (*vi->LineSearch)(snes,vi->lsP,X,vi->phi,G,Y,W,vi->phinorm,xnorm,&ynorm,&gnorm,&lssucceed);CHKERRQ(ierr);
103169c03620SShri Abhyankar     ierr = PetscInfo4(snes,"fnorm=%18.16e, gnorm=%18.16e, ynorm=%18.16e, lssucceed=%d\n",vi->phinorm,gnorm,ynorm,(int)lssucceed);CHKERRQ(ierr);
103269c03620SShri Abhyankar     if (snes->reason == SNES_DIVERGED_FUNCTION_COUNT) break;
103369c03620SShri Abhyankar     if (snes->domainerror) {
103469c03620SShri Abhyankar       snes->reason = SNES_DIVERGED_FUNCTION_DOMAIN;
103569c03620SShri Abhyankar       PetscFunctionReturn(0);
103669c03620SShri Abhyankar     }
103769c03620SShri Abhyankar     if (!lssucceed) {
103869c03620SShri Abhyankar       if (++snes->numFailures >= snes->maxFailures) {
103969c03620SShri Abhyankar 	PetscBool ismin;
104069c03620SShri Abhyankar         snes->reason = SNES_DIVERGED_LINE_SEARCH;
104169c03620SShri Abhyankar         ierr = SNESVICheckLocalMin_Private(snes,snes->jacobian,G,W,gnorm,&ismin);CHKERRQ(ierr);
104269c03620SShri Abhyankar         if (ismin) snes->reason = SNES_DIVERGED_LOCAL_MIN;
104369c03620SShri Abhyankar         break;
104469c03620SShri Abhyankar       }
104569c03620SShri Abhyankar     }
104669c03620SShri Abhyankar     /* Update function and solution vectors */
104769c03620SShri Abhyankar     vi->phinorm = gnorm;
104869c03620SShri Abhyankar     vi->merit = 0.5*vi->phinorm*vi->phinorm;
104969c03620SShri Abhyankar     ierr = VecCopy(G,vi->phi);CHKERRQ(ierr);
105069c03620SShri Abhyankar     ierr = VecCopy(W,X);CHKERRQ(ierr);
105169c03620SShri Abhyankar     /* Monitor convergence */
105269c03620SShri Abhyankar     ierr = PetscObjectTakeAccess(snes);CHKERRQ(ierr);
105369c03620SShri Abhyankar     snes->iter = i+1;
105469c03620SShri Abhyankar     snes->norm = vi->phinorm;
105569c03620SShri Abhyankar     ierr = PetscObjectGrantAccess(snes);CHKERRQ(ierr);
105669c03620SShri Abhyankar     SNESLogConvHistory(snes,snes->norm,lits);
105769c03620SShri Abhyankar     SNESMonitor(snes,snes->iter,snes->norm);
105869c03620SShri Abhyankar     /* Test for convergence, xnorm = || X || */
105969c03620SShri Abhyankar     if (snes->ops->converged != SNESSkipConverged) { ierr = VecNorm(X,NORM_2,&xnorm);CHKERRQ(ierr); }
106069c03620SShri Abhyankar     ierr = (*snes->ops->converged)(snes,snes->iter,xnorm,ynorm,vi->phinorm,&snes->reason,snes->cnvP);CHKERRQ(ierr);
106169c03620SShri Abhyankar     if (snes->reason) break;
106269c03620SShri Abhyankar   }
106369c03620SShri Abhyankar   if (i == maxits) {
106469c03620SShri Abhyankar     ierr = PetscInfo1(snes,"Maximum number of iterations has been reached: %D\n",maxits);CHKERRQ(ierr);
106569c03620SShri Abhyankar     if(!snes->reason) snes->reason = SNES_DIVERGED_MAX_IT;
106669c03620SShri Abhyankar   }
106769c03620SShri Abhyankar   PetscFunctionReturn(0);
106869c03620SShri Abhyankar }
106969c03620SShri Abhyankar 
1070*fe835674SShri Abhyankar #undef __FUNCT__
1071*fe835674SShri Abhyankar #define __FUNCT__ "SNESVIComputeInactiveSetFnorm"
1072*fe835674SShri Abhyankar PetscErrorCode SNESVIComputeInactiveSetFnorm(SNES snes,Vec F,Vec X,PetscScalar *fnorm)
1073*fe835674SShri Abhyankar {
1074*fe835674SShri Abhyankar   PetscErrorCode    ierr;
1075*fe835674SShri Abhyankar   SNES_VI           *vi = (SNES_VI*)snes->data;
1076*fe835674SShri Abhyankar   const PetscScalar *x,*xl,*xu,*f;
1077*fe835674SShri Abhyankar   PetscInt          i,n;
1078*fe835674SShri Abhyankar   PetscReal         rnorm;
1079*fe835674SShri Abhyankar 
1080*fe835674SShri Abhyankar   PetscFunctionBegin;
1081*fe835674SShri Abhyankar 
1082*fe835674SShri Abhyankar   ierr = VecGetLocalSize(X,&n);CHKERRQ(ierr);
1083*fe835674SShri Abhyankar   ierr = VecGetArrayRead(vi->xl,&xl);CHKERRQ(ierr);
1084*fe835674SShri Abhyankar   ierr = VecGetArrayRead(vi->xu,&xu);CHKERRQ(ierr);
1085*fe835674SShri Abhyankar   ierr = VecGetArrayRead(X,&x);CHKERRQ(ierr);
1086*fe835674SShri Abhyankar   ierr = VecGetArrayRead(F,&f);CHKERRQ(ierr);
1087*fe835674SShri Abhyankar 
1088*fe835674SShri Abhyankar   rnorm = 0.0;
1089*fe835674SShri Abhyankar   for (i=0; i<n; i++) {
1090*fe835674SShri Abhyankar     if (((x[i] > xl[i] + 1.e-8 || (f[i] < 0.0)) && ((x[i] < xu[i] - 1.e-8) || f[i] > 0.0))) rnorm += f[i]*f[i];
1091*fe835674SShri Abhyankar   }
1092*fe835674SShri Abhyankar   ierr = VecRestoreArrayRead(F,&f);CHKERRQ(ierr);
1093*fe835674SShri Abhyankar   ierr = VecRestoreArrayRead(vi->xl,&xl);CHKERRQ(ierr);
1094*fe835674SShri Abhyankar   ierr = VecRestoreArrayRead(vi->xu,&xu);CHKERRQ(ierr);
1095*fe835674SShri Abhyankar   ierr = VecRestoreArrayRead(X,&x);CHKERRQ(ierr);
1096*fe835674SShri Abhyankar   ierr = MPI_Allreduce(&rnorm,fnorm,1,MPIU_REAL,MPI_SUM,((PetscObject)snes)->comm);CHKERRQ(ierr);
1097*fe835674SShri Abhyankar   *fnorm = sqrt(*fnorm);
1098*fe835674SShri Abhyankar   PetscFunctionReturn(0);
1099*fe835674SShri Abhyankar }
1100*fe835674SShri Abhyankar 
1101*fe835674SShri Abhyankar /* Variational Inequality solver using reduce space method. No semismooth algorithm is
1102*fe835674SShri Abhyankar    implemented in this algorithm. It basically identifies the active variables and does
1103*fe835674SShri Abhyankar    a linear solve on the inactive variables. */
1104d950fb63SShri Abhyankar #undef __FUNCT__
1105d950fb63SShri Abhyankar #define __FUNCT__ "SNESSolveVI_RS"
1106d950fb63SShri Abhyankar PetscErrorCode SNESSolveVI_RS(SNES snes)
1107d950fb63SShri Abhyankar {
1108d950fb63SShri Abhyankar   SNES_VI          *vi = (SNES_VI*)snes->data;
1109d950fb63SShri Abhyankar   PetscErrorCode    ierr;
1110d950fb63SShri Abhyankar   PetscInt          maxits,i,lits,Nis_act=0;
1111d950fb63SShri Abhyankar   PetscBool         lssucceed;
1112d950fb63SShri Abhyankar   MatStructure      flg = DIFFERENT_NONZERO_PATTERN;
1113*fe835674SShri Abhyankar   PetscReal         fnorm,gnorm,xnorm=0,ynorm;
1114d950fb63SShri Abhyankar   Vec                Y,X,F,G,W;
1115d950fb63SShri Abhyankar   KSPConvergedReason kspreason;
1116d950fb63SShri Abhyankar 
1117d950fb63SShri Abhyankar   PetscFunctionBegin;
1118d950fb63SShri Abhyankar   snes->numFailures            = 0;
1119d950fb63SShri Abhyankar   snes->numLinearSolveFailures = 0;
1120d950fb63SShri Abhyankar   snes->reason                 = SNES_CONVERGED_ITERATING;
1121d950fb63SShri Abhyankar 
1122d950fb63SShri Abhyankar   maxits	= snes->max_its;	/* maximum number of iterations */
1123d950fb63SShri Abhyankar   X		= snes->vec_sol;	/* solution vector */
1124d950fb63SShri Abhyankar   F		= snes->vec_func;	/* residual vector */
1125d950fb63SShri Abhyankar   Y		= snes->work[0];	/* work vectors */
1126d950fb63SShri Abhyankar   G		= snes->work[1];
1127d950fb63SShri Abhyankar   W		= snes->work[2];
1128d950fb63SShri Abhyankar 
1129d950fb63SShri Abhyankar   ierr = PetscObjectTakeAccess(snes);CHKERRQ(ierr);
1130d950fb63SShri Abhyankar   snes->iter = 0;
1131d950fb63SShri Abhyankar   snes->norm = 0.0;
1132d950fb63SShri Abhyankar   ierr = PetscObjectGrantAccess(snes);CHKERRQ(ierr);
1133d950fb63SShri Abhyankar 
1134d950fb63SShri Abhyankar   ierr = SNESVIAdjustInitialGuess(X,vi->xl,vi->xu);CHKERRQ(ierr);
1135*fe835674SShri Abhyankar   ierr = SNESComputeFunction(snes,X,F);CHKERRQ(ierr);
1136d950fb63SShri Abhyankar   if (snes->domainerror) {
1137d950fb63SShri Abhyankar     snes->reason = SNES_DIVERGED_FUNCTION_DOMAIN;
1138d950fb63SShri Abhyankar     PetscFunctionReturn(0);
1139d950fb63SShri Abhyankar   }
1140*fe835674SShri Abhyankar   ierr = SNESVIComputeInactiveSetFnorm(snes,F,X,&fnorm);CHKERRQ(ierr);
1141d950fb63SShri Abhyankar   ierr = VecNormBegin(X,NORM_2,&xnorm);CHKERRQ(ierr);	/* xnorm <- ||x||  */
1142d950fb63SShri Abhyankar   ierr = VecNormEnd(X,NORM_2,&xnorm);CHKERRQ(ierr);
1143*fe835674SShri Abhyankar   if PetscIsInfOrNanReal(fnorm) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_FP,"User provided compute function generated a Not-a-Number");
1144d950fb63SShri Abhyankar 
1145d950fb63SShri Abhyankar   ierr = PetscObjectTakeAccess(snes);CHKERRQ(ierr);
1146*fe835674SShri Abhyankar   snes->norm = fnorm;
1147d950fb63SShri Abhyankar   ierr = PetscObjectGrantAccess(snes);CHKERRQ(ierr);
1148*fe835674SShri Abhyankar   SNESLogConvHistory(snes,fnorm,0);
1149*fe835674SShri Abhyankar   SNESMonitor(snes,0,fnorm);
1150d950fb63SShri Abhyankar 
1151d950fb63SShri Abhyankar   /* set parameter for default relative tolerance convergence test */
1152*fe835674SShri Abhyankar   snes->ttol = fnorm*snes->rtol;
1153d950fb63SShri Abhyankar   /* test convergence */
1154*fe835674SShri Abhyankar   ierr = (*snes->ops->converged)(snes,0,0.0,0.0,fnorm,&snes->reason,snes->cnvP);CHKERRQ(ierr);
1155d950fb63SShri Abhyankar   if (snes->reason) PetscFunctionReturn(0);
1156d950fb63SShri Abhyankar 
1157d950fb63SShri Abhyankar   for (i=0; i<maxits; i++) {
1158d950fb63SShri Abhyankar 
1159d950fb63SShri Abhyankar     IS                 IS_act,IS_inact; /* _act -> active set _inact -> inactive set */
1160d950fb63SShri Abhyankar     VecScatter         scat_act,scat_inact;
1161d950fb63SShri Abhyankar     PetscInt           nis_act,nis_inact,Nis_act_prev;
1162*fe835674SShri Abhyankar     Vec                Y_act,Y_inact,F_inact;
1163d950fb63SShri Abhyankar     Mat                jac_inact_inact,prejac_inact_inact;
1164d950fb63SShri Abhyankar 
1165d950fb63SShri Abhyankar     /* Call general purpose update function */
1166d950fb63SShri Abhyankar     if (snes->ops->update) {
1167d950fb63SShri Abhyankar       ierr = (*snes->ops->update)(snes, snes->iter);CHKERRQ(ierr);
1168d950fb63SShri Abhyankar     }
1169d950fb63SShri Abhyankar     ierr = SNESComputeJacobian(snes,X,&snes->jacobian,&snes->jacobian_pre,&flg);CHKERRQ(ierr);
1170d950fb63SShri Abhyankar     /* Create active and inactive index sets */
1171d950fb63SShri Abhyankar     ierr = SNESVICreateIndexSets_RS(snes,X,vi->xl,vi->xu,&IS_act,&IS_inact);CHKERRQ(ierr);
1172d950fb63SShri Abhyankar 
1173d950fb63SShri Abhyankar     Nis_act_prev = Nis_act;
1174d950fb63SShri Abhyankar     /* Get sizes of active and inactive sets */
1175d950fb63SShri Abhyankar     ierr = ISGetLocalSize(IS_act,&nis_act);CHKERRQ(ierr);
1176d950fb63SShri Abhyankar     ierr = ISGetLocalSize(IS_inact,&nis_inact);CHKERRQ(ierr);
1177d950fb63SShri Abhyankar     ierr = ISGetSize(IS_act,&Nis_act);CHKERRQ(ierr);
1178d950fb63SShri Abhyankar 
1179d950fb63SShri Abhyankar     ierr = PetscPrintf(PETSC_COMM_WORLD,"Size of active set = %d, size of inactive set = %d\n",nis_act,nis_inact);CHKERRQ(ierr);
1180d950fb63SShri Abhyankar 
1181d950fb63SShri Abhyankar     /* Create active and inactive set vectors */
1182*fe835674SShri Abhyankar     ierr = SNESVICreateSubVectors(snes,nis_inact,&F_inact);CHKERRQ(ierr);
1183*fe835674SShri Abhyankar     ierr = SNESVICreateSubVectors(snes,nis_act,&Y_act);CHKERRQ(ierr);
1184*fe835674SShri Abhyankar     ierr = SNESVICreateSubVectors(snes,nis_inact,&Y_inact);CHKERRQ(ierr);
1185d950fb63SShri Abhyankar 
1186d950fb63SShri Abhyankar     /* Create inactive set submatrices */
1187d950fb63SShri Abhyankar     ierr = MatGetSubMatrix(snes->jacobian,IS_inact,IS_inact,MAT_INITIAL_MATRIX,&jac_inact_inact);CHKERRQ(ierr);
1188d950fb63SShri Abhyankar 
1189d950fb63SShri Abhyankar     /* Create scatter contexts */
1190d950fb63SShri Abhyankar     ierr = VecScatterCreate(Y,IS_act,Y_act,PETSC_NULL,&scat_act);CHKERRQ(ierr);
1191d950fb63SShri Abhyankar     ierr = VecScatterCreate(Y,IS_inact,Y_inact,PETSC_NULL,&scat_inact);CHKERRQ(ierr);
1192d950fb63SShri Abhyankar 
1193d950fb63SShri Abhyankar     /* Do a vec scatter to active and inactive set vectors */
1194*fe835674SShri Abhyankar     ierr = VecScatterBegin(scat_inact,F,F_inact,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
1195*fe835674SShri Abhyankar     ierr = VecScatterEnd(scat_inact,F,F_inact,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
1196d950fb63SShri Abhyankar 
1197d950fb63SShri Abhyankar     ierr = VecScatterBegin(scat_act,Y,Y_act,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
1198d950fb63SShri Abhyankar     ierr = VecScatterEnd(scat_act,Y,Y_act,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
1199d950fb63SShri Abhyankar 
1200d950fb63SShri Abhyankar     ierr = VecScatterBegin(scat_inact,Y,Y_inact,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
1201d950fb63SShri Abhyankar     ierr = VecScatterEnd(scat_inact,Y,Y_inact,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
1202d950fb63SShri Abhyankar 
1203d950fb63SShri Abhyankar     /* Active set direction = 0 */
1204d950fb63SShri Abhyankar     ierr = VecSet(Y_act,0);CHKERRQ(ierr);
1205d950fb63SShri Abhyankar     if (snes->jacobian != snes->jacobian_pre) {
1206d950fb63SShri Abhyankar       ierr = MatGetSubMatrix(snes->jacobian_pre,IS_inact,IS_inact,MAT_INITIAL_MATRIX,&prejac_inact_inact);CHKERRQ(ierr);
1207d950fb63SShri Abhyankar     } else prejac_inact_inact = jac_inact_inact;
1208d950fb63SShri Abhyankar 
1209d950fb63SShri Abhyankar     if ((i != 0) && (Nis_act != Nis_act_prev)) {
1210732bb160SShri Abhyankar       ierr = SNESVIResetPCandKSP(snes,jac_inact_inact,prejac_inact_inact);CHKERRQ(ierr);
1211d950fb63SShri Abhyankar     }
1212d950fb63SShri Abhyankar 
1213d950fb63SShri Abhyankar     ierr = KSPSetOperators(snes->ksp,jac_inact_inact,prejac_inact_inact,flg);CHKERRQ(ierr);
1214*fe835674SShri Abhyankar     ierr = SNES_KSPSolve(snes,snes->ksp,F_inact,Y_inact);CHKERRQ(ierr);
1215d950fb63SShri Abhyankar     ierr = KSPGetConvergedReason(snes->ksp,&kspreason);CHKERRQ(ierr);
1216d950fb63SShri Abhyankar     if (kspreason < 0) {
1217d950fb63SShri Abhyankar       if (++snes->numLinearSolveFailures >= snes->maxLinearSolveFailures) {
1218d950fb63SShri Abhyankar         ierr = PetscInfo2(snes,"iter=%D, number linear solve failures %D greater than current SNES allowed, stopping solve\n",snes->iter,snes->numLinearSolveFailures);CHKERRQ(ierr);
1219d950fb63SShri Abhyankar         snes->reason = SNES_DIVERGED_LINEAR_SOLVE;
1220d950fb63SShri Abhyankar         break;
1221d950fb63SShri Abhyankar       }
1222d950fb63SShri Abhyankar      }
1223d950fb63SShri Abhyankar 
1224d950fb63SShri Abhyankar     ierr = VecScatterBegin(scat_act,Y_act,Y,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr);
1225d950fb63SShri Abhyankar     ierr = VecScatterEnd(scat_act,Y_act,Y,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr);
1226d950fb63SShri Abhyankar     ierr = VecScatterBegin(scat_inact,Y_inact,Y,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr);
1227d950fb63SShri Abhyankar     ierr = VecScatterEnd(scat_inact,Y_inact,Y,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr);
1228d950fb63SShri Abhyankar 
1229*fe835674SShri Abhyankar     ierr = VecDestroy(F_inact);CHKERRQ(ierr);
1230d950fb63SShri Abhyankar     ierr = VecDestroy(Y_act);CHKERRQ(ierr);
1231d950fb63SShri Abhyankar     ierr = VecDestroy(Y_inact);CHKERRQ(ierr);
1232d950fb63SShri Abhyankar     ierr = VecScatterDestroy(scat_act);CHKERRQ(ierr);
1233d950fb63SShri Abhyankar     ierr = VecScatterDestroy(scat_inact);CHKERRQ(ierr);
1234d950fb63SShri Abhyankar     ierr = ISDestroy(IS_act);CHKERRQ(ierr);
1235d950fb63SShri Abhyankar     ierr = ISDestroy(IS_inact);CHKERRQ(ierr);
1236d950fb63SShri Abhyankar     ierr = MatDestroy(jac_inact_inact);CHKERRQ(ierr);
1237d950fb63SShri Abhyankar     if (snes->jacobian != snes->jacobian_pre) {
1238d950fb63SShri Abhyankar       ierr = MatDestroy(prejac_inact_inact);CHKERRQ(ierr);
1239d950fb63SShri Abhyankar     }
1240d950fb63SShri Abhyankar 
1241d950fb63SShri Abhyankar     ierr = KSPGetIterationNumber(snes->ksp,&lits);CHKERRQ(ierr);
1242d950fb63SShri Abhyankar     snes->linear_its += lits;
1243d950fb63SShri Abhyankar     ierr = PetscInfo2(snes,"iter=%D, linear solve iterations=%D\n",snes->iter,lits);CHKERRQ(ierr);
1244d950fb63SShri Abhyankar     /*
1245d950fb63SShri Abhyankar     if (vi->precheckstep) {
1246d950fb63SShri Abhyankar       PetscBool changed_y = PETSC_FALSE;
1247d950fb63SShri Abhyankar       ierr = (*vi->precheckstep)(snes,X,Y,vi->precheck,&changed_y);CHKERRQ(ierr);
1248d950fb63SShri Abhyankar     }
1249d950fb63SShri Abhyankar 
1250d950fb63SShri Abhyankar     if (PetscLogPrintInfo){
1251d950fb63SShri Abhyankar       ierr = SNESVICheckResidual_Private(snes,snes->jacobian,F,Y,G,W);CHKERRQ(ierr);
1252d950fb63SShri Abhyankar     }
1253d950fb63SShri Abhyankar     */
1254d950fb63SShri Abhyankar     /* Compute a (scaled) negative update in the line search routine:
1255d950fb63SShri Abhyankar          Y <- X - lambda*Y
1256d950fb63SShri Abhyankar        and evaluate G = function(Y) (depends on the line search).
1257d950fb63SShri Abhyankar     */
1258d950fb63SShri Abhyankar     ierr = VecCopy(Y,snes->vec_sol_update);CHKERRQ(ierr);
1259*fe835674SShri Abhyankar     ynorm = 1; gnorm = fnorm;
1260*fe835674SShri Abhyankar     ierr = (*vi->LineSearch)(snes,vi->lsP,X,F,G,Y,W,fnorm,xnorm,&ynorm,&gnorm,&lssucceed);CHKERRQ(ierr);
1261*fe835674SShri Abhyankar     ierr = PetscInfo4(snes,"fnorm=%18.16e, gnorm=%18.16e, ynorm=%18.16e, lssucceed=%d\n",fnorm,gnorm,ynorm,(int)lssucceed);CHKERRQ(ierr);
12622b4ed282SShri Abhyankar     if (snes->reason == SNES_DIVERGED_FUNCTION_COUNT) break;
12632b4ed282SShri Abhyankar     if (snes->domainerror) {
12642b4ed282SShri Abhyankar       snes->reason = SNES_DIVERGED_FUNCTION_DOMAIN;
12652b4ed282SShri Abhyankar       PetscFunctionReturn(0);
12662b4ed282SShri Abhyankar     }
12672b4ed282SShri Abhyankar     if (!lssucceed) {
12682b4ed282SShri Abhyankar       if (++snes->numFailures >= snes->maxFailures) {
12692b4ed282SShri Abhyankar 	PetscBool ismin;
12702b4ed282SShri Abhyankar         snes->reason = SNES_DIVERGED_LINE_SEARCH;
12712b4ed282SShri Abhyankar         ierr = SNESVICheckLocalMin_Private(snes,snes->jacobian,G,W,gnorm,&ismin);CHKERRQ(ierr);
12722b4ed282SShri Abhyankar         if (ismin) snes->reason = SNES_DIVERGED_LOCAL_MIN;
12732b4ed282SShri Abhyankar         break;
12742b4ed282SShri Abhyankar       }
12752b4ed282SShri Abhyankar     }
12762b4ed282SShri Abhyankar     /* Update function and solution vectors */
1277*fe835674SShri Abhyankar     fnorm = gnorm;
1278*fe835674SShri Abhyankar     ierr = VecCopy(G,F);CHKERRQ(ierr);
12792b4ed282SShri Abhyankar     ierr = VecCopy(W,X);CHKERRQ(ierr);
12802b4ed282SShri Abhyankar     /* Monitor convergence */
12812b4ed282SShri Abhyankar     ierr = PetscObjectTakeAccess(snes);CHKERRQ(ierr);
12822b4ed282SShri Abhyankar     snes->iter = i+1;
1283*fe835674SShri Abhyankar     snes->norm = fnorm;
12842b4ed282SShri Abhyankar     ierr = PetscObjectGrantAccess(snes);CHKERRQ(ierr);
12852b4ed282SShri Abhyankar     SNESLogConvHistory(snes,snes->norm,lits);
12862b4ed282SShri Abhyankar     SNESMonitor(snes,snes->iter,snes->norm);
12872b4ed282SShri Abhyankar     /* Test for convergence, xnorm = || X || */
12882b4ed282SShri Abhyankar     if (snes->ops->converged != SNESSkipConverged) { ierr = VecNorm(X,NORM_2,&xnorm);CHKERRQ(ierr); }
1289*fe835674SShri Abhyankar     ierr = (*snes->ops->converged)(snes,snes->iter,xnorm,ynorm,fnorm,&snes->reason,snes->cnvP);CHKERRQ(ierr);
12902b4ed282SShri Abhyankar     if (snes->reason) break;
12912b4ed282SShri Abhyankar   }
12922b4ed282SShri Abhyankar   if (i == maxits) {
12932b4ed282SShri Abhyankar     ierr = PetscInfo1(snes,"Maximum number of iterations has been reached: %D\n",maxits);CHKERRQ(ierr);
12942b4ed282SShri Abhyankar     if(!snes->reason) snes->reason = SNES_DIVERGED_MAX_IT;
12952b4ed282SShri Abhyankar   }
12962b4ed282SShri Abhyankar   PetscFunctionReturn(0);
12972b4ed282SShri Abhyankar }
12982b4ed282SShri Abhyankar 
12992b4ed282SShri Abhyankar /* -------------------------------------------------------------------------- */
13002b4ed282SShri Abhyankar /*
13012b4ed282SShri Abhyankar    SNESSetUp_VI - Sets up the internal data structures for the later use
13022b4ed282SShri Abhyankar    of the SNESVI nonlinear solver.
13032b4ed282SShri Abhyankar 
13042b4ed282SShri Abhyankar    Input Parameter:
13052b4ed282SShri Abhyankar .  snes - the SNES context
13062b4ed282SShri Abhyankar .  x - the solution vector
13072b4ed282SShri Abhyankar 
13082b4ed282SShri Abhyankar    Application Interface Routine: SNESSetUp()
13092b4ed282SShri Abhyankar 
13102b4ed282SShri Abhyankar    Notes:
13112b4ed282SShri Abhyankar    For basic use of the SNES solvers, the user need not explicitly call
13122b4ed282SShri Abhyankar    SNESSetUp(), since these actions will automatically occur during
13132b4ed282SShri Abhyankar    the call to SNESSolve().
13142b4ed282SShri Abhyankar  */
13152b4ed282SShri Abhyankar #undef __FUNCT__
13162b4ed282SShri Abhyankar #define __FUNCT__ "SNESSetUp_VI"
13172b4ed282SShri Abhyankar PetscErrorCode SNESSetUp_VI(SNES snes)
13182b4ed282SShri Abhyankar {
13192b4ed282SShri Abhyankar   PetscErrorCode ierr;
13202b4ed282SShri Abhyankar   SNES_VI      *vi = (SNES_VI*) snes->data;
13212b4ed282SShri Abhyankar   PetscInt       i_start[3],i_end[3];
13222b4ed282SShri Abhyankar 
13232b4ed282SShri Abhyankar   PetscFunctionBegin;
13242b4ed282SShri Abhyankar   if (!snes->vec_sol_update) {
13252b4ed282SShri Abhyankar     ierr = VecDuplicate(snes->vec_sol,&snes->vec_sol_update);CHKERRQ(ierr);
13262b4ed282SShri Abhyankar     ierr = PetscLogObjectParent(snes,snes->vec_sol_update);CHKERRQ(ierr);
13272b4ed282SShri Abhyankar   }
13282b4ed282SShri Abhyankar   if (!snes->work) {
13292b4ed282SShri Abhyankar     snes->nwork = 3;
13302b4ed282SShri Abhyankar     ierr = VecDuplicateVecs(snes->vec_sol,snes->nwork,&snes->work);CHKERRQ(ierr);
13312b4ed282SShri Abhyankar     ierr = PetscLogObjectParents(snes,snes->nwork,snes->work);CHKERRQ(ierr);
13322b4ed282SShri Abhyankar   }
13332b4ed282SShri Abhyankar 
13342b4ed282SShri Abhyankar   /* If the lower and upper bound on variables are not set, set it to
13352b4ed282SShri Abhyankar      -Inf and Inf */
13362b4ed282SShri Abhyankar   if (!vi->xl && !vi->xu) {
13372b4ed282SShri Abhyankar     vi->usersetxbounds = PETSC_FALSE;
13382b4ed282SShri Abhyankar     ierr = VecDuplicate(snes->vec_sol, &vi->xl); CHKERRQ(ierr);
13392b4ed282SShri Abhyankar     ierr = VecSet(vi->xl,PETSC_VI_NINF);CHKERRQ(ierr);
13402b4ed282SShri Abhyankar     ierr = VecDuplicate(snes->vec_sol, &vi->xu); CHKERRQ(ierr);
13412b4ed282SShri Abhyankar     ierr = VecSet(vi->xu,PETSC_VI_INF);CHKERRQ(ierr);
13422b4ed282SShri Abhyankar   } else {
13432b4ed282SShri Abhyankar     /* Check if lower bound, upper bound and solution vector distribution across the processors is identical */
13442b4ed282SShri Abhyankar     ierr = VecGetOwnershipRange(snes->vec_sol,i_start,i_end);CHKERRQ(ierr);
13452b4ed282SShri Abhyankar     ierr = VecGetOwnershipRange(vi->xl,i_start+1,i_end+1);CHKERRQ(ierr);
13462b4ed282SShri Abhyankar     ierr = VecGetOwnershipRange(vi->xu,i_start+2,i_end+2);CHKERRQ(ierr);
13472b4ed282SShri Abhyankar     if ((i_start[0] != i_start[1]) || (i_start[0] != i_start[2]) || (i_end[0] != i_end[1]) || (i_end[0] != i_end[2]))
13482b4ed282SShri Abhyankar       SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_SIZ,"Distribution of lower bound, upper bound and the solution vector should be identical across all the processors.");
13492b4ed282SShri Abhyankar   }
13502b4ed282SShri Abhyankar 
1351*fe835674SShri Abhyankar   if (snes->ops->solve != SNESSolveVI_RS) {
1352*fe835674SShri Abhyankar     ierr = VecDuplicate(snes->vec_sol, &vi->dpsi);CHKERRQ(ierr);
1353*fe835674SShri Abhyankar     ierr = VecDuplicate(snes->vec_sol, &vi->phi); CHKERRQ(ierr);
1354*fe835674SShri Abhyankar     ierr = VecDuplicate(snes->vec_sol, &vi->Da); CHKERRQ(ierr);
1355*fe835674SShri Abhyankar     ierr = VecDuplicate(snes->vec_sol, &vi->Db); CHKERRQ(ierr);
1356*fe835674SShri Abhyankar     ierr = VecDuplicate(snes->vec_sol, &vi->z);CHKERRQ(ierr);
1357*fe835674SShri Abhyankar     ierr = VecDuplicate(snes->vec_sol, &vi->t); CHKERRQ(ierr);
1358*fe835674SShri Abhyankar 
1359*fe835674SShri Abhyankar 
13602b4ed282SShri Abhyankar     vi->computeuserfunction = snes->ops->computefunction;
13611f79c099SShri Abhyankar     snes->ops->computefunction = SNESVIComputeFunction;
1362*fe835674SShri Abhyankar   }
13632b4ed282SShri Abhyankar   PetscFunctionReturn(0);
13642b4ed282SShri Abhyankar }
13652b4ed282SShri Abhyankar /* -------------------------------------------------------------------------- */
13662b4ed282SShri Abhyankar /*
13672b4ed282SShri Abhyankar    SNESDestroy_VI - Destroys the private SNES_VI context that was created
13682b4ed282SShri Abhyankar    with SNESCreate_VI().
13692b4ed282SShri Abhyankar 
13702b4ed282SShri Abhyankar    Input Parameter:
13712b4ed282SShri Abhyankar .  snes - the SNES context
13722b4ed282SShri Abhyankar 
13732b4ed282SShri Abhyankar    Application Interface Routine: SNESDestroy()
13742b4ed282SShri Abhyankar  */
13752b4ed282SShri Abhyankar #undef __FUNCT__
13762b4ed282SShri Abhyankar #define __FUNCT__ "SNESDestroy_VI"
13772b4ed282SShri Abhyankar PetscErrorCode SNESDestroy_VI(SNES snes)
13782b4ed282SShri Abhyankar {
13792b4ed282SShri Abhyankar   SNES_VI        *vi = (SNES_VI*) snes->data;
13802b4ed282SShri Abhyankar   PetscErrorCode ierr;
13812b4ed282SShri Abhyankar 
13822b4ed282SShri Abhyankar   PetscFunctionBegin;
13832b4ed282SShri Abhyankar   if (snes->vec_sol_update) {
13842b4ed282SShri Abhyankar     ierr = VecDestroy(snes->vec_sol_update);CHKERRQ(ierr);
13852b4ed282SShri Abhyankar     snes->vec_sol_update = PETSC_NULL;
13862b4ed282SShri Abhyankar   }
13872b4ed282SShri Abhyankar   if (snes->nwork) {
13882b4ed282SShri Abhyankar     ierr = VecDestroyVecs(snes->work,snes->nwork);CHKERRQ(ierr);
13892b4ed282SShri Abhyankar     snes->nwork = 0;
13902b4ed282SShri Abhyankar     snes->work  = PETSC_NULL;
13912b4ed282SShri Abhyankar   }
13922b4ed282SShri Abhyankar 
1393*fe835674SShri Abhyankar   if (snes->ops->solve != SNESSolveVI_RS) {
13942b4ed282SShri Abhyankar     /* clear vectors */
1395ee657d29SShri Abhyankar     ierr = VecDestroy(vi->dpsi);CHKERRQ(ierr);
13962b4ed282SShri Abhyankar     ierr = VecDestroy(vi->phi); CHKERRQ(ierr);
13972b4ed282SShri Abhyankar     ierr = VecDestroy(vi->Da); CHKERRQ(ierr);
13982b4ed282SShri Abhyankar     ierr = VecDestroy(vi->Db); CHKERRQ(ierr);
13992b4ed282SShri Abhyankar     ierr = VecDestroy(vi->z); CHKERRQ(ierr);
14002b4ed282SShri Abhyankar     ierr = VecDestroy(vi->t); CHKERRQ(ierr);
14012b4ed282SShri Abhyankar     if (!vi->usersetxbounds) {
14022b4ed282SShri Abhyankar       ierr = VecDestroy(vi->xl); CHKERRQ(ierr);
14032b4ed282SShri Abhyankar       ierr = VecDestroy(vi->xu); CHKERRQ(ierr);
14042b4ed282SShri Abhyankar     }
1405*fe835674SShri Abhyankar   }
1406c92abb8aSShri Abhyankar   if (vi->lsmonitor) {
1407c92abb8aSShri Abhyankar     ierr = PetscViewerASCIIMonitorDestroy(vi->lsmonitor);CHKERRQ(ierr);
14082b4ed282SShri Abhyankar   }
14092b4ed282SShri Abhyankar   ierr = PetscFree(snes->data);CHKERRQ(ierr);
14102b4ed282SShri Abhyankar 
14112b4ed282SShri Abhyankar   /* clear composed functions */
14122b4ed282SShri Abhyankar   ierr = PetscObjectComposeFunctionDynamic((PetscObject)snes,"SNESLineSearchSet_C","",PETSC_NULL);CHKERRQ(ierr);
1413c92abb8aSShri Abhyankar   ierr = PetscObjectComposeFunctionDynamic((PetscObject)snes,"SNESLineSearchSetMonitor_C","",PETSC_NULL);CHKERRQ(ierr);
14142b4ed282SShri Abhyankar   PetscFunctionReturn(0);
14152b4ed282SShri Abhyankar }
14162b4ed282SShri Abhyankar /* -------------------------------------------------------------------------- */
14172b4ed282SShri Abhyankar #undef __FUNCT__
14182b4ed282SShri Abhyankar #define __FUNCT__ "SNESLineSearchNo_VI"
14192b4ed282SShri Abhyankar 
14202b4ed282SShri Abhyankar /*
14212b4ed282SShri Abhyankar   This routine is a copy of SNESLineSearchNo routine in snes/impls/ls/ls.c
14222b4ed282SShri Abhyankar 
14232b4ed282SShri Abhyankar */
1424ace3abfcSBarry Smith PetscErrorCode SNESLineSearchNo_VI(SNES snes,void *lsctx,Vec x,Vec f,Vec g,Vec y,Vec w,PetscReal fnorm,PetscReal xnorm,PetscReal *ynorm,PetscReal *gnorm,PetscBool *flag)
14252b4ed282SShri Abhyankar {
14262b4ed282SShri Abhyankar   PetscErrorCode ierr;
14272b4ed282SShri Abhyankar   SNES_VI        *vi = (SNES_VI*)snes->data;
1428ace3abfcSBarry Smith   PetscBool     changed_w = PETSC_FALSE,changed_y = PETSC_FALSE;
14292b4ed282SShri Abhyankar 
14302b4ed282SShri Abhyankar   PetscFunctionBegin;
14312b4ed282SShri Abhyankar   *flag = PETSC_TRUE;
14322b4ed282SShri Abhyankar   ierr = PetscLogEventBegin(SNES_LineSearch,snes,x,f,g);CHKERRQ(ierr);
14332b4ed282SShri Abhyankar   ierr = VecNorm(y,NORM_2,ynorm);CHKERRQ(ierr);         /* ynorm = || y || */
14342b4ed282SShri Abhyankar   ierr = VecWAXPY(w,-1.0,y,x);CHKERRQ(ierr);            /* w <- x - y   */
14352b4ed282SShri Abhyankar   if (vi->postcheckstep) {
14362b4ed282SShri Abhyankar    ierr = (*vi->postcheckstep)(snes,x,y,w,vi->postcheck,&changed_y,&changed_w);CHKERRQ(ierr);
14372b4ed282SShri Abhyankar   }
14382b4ed282SShri Abhyankar   if (changed_y) {
14392b4ed282SShri Abhyankar     ierr = VecWAXPY(w,-1.0,y,x);CHKERRQ(ierr);            /* w <- x - y   */
14402b4ed282SShri Abhyankar   }
14412b4ed282SShri Abhyankar   ierr = SNESComputeFunction(snes,w,g);CHKERRQ(ierr);
14422b4ed282SShri Abhyankar   if (!snes->domainerror) {
14432b4ed282SShri Abhyankar     ierr = VecNorm(g,NORM_2,gnorm);CHKERRQ(ierr);  /* gnorm = || g || */
14442b4ed282SShri Abhyankar     if PetscIsInfOrNanReal(*gnorm) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_FP,"User provided compute function generated a Not-a-Number");
14452b4ed282SShri Abhyankar   }
1446c92abb8aSShri Abhyankar   if (vi->lsmonitor) {
1447c92abb8aSShri Abhyankar     ierr = PetscViewerASCIIMonitorPrintf(vi->lsmonitor,"    Line search: Using full step: fnorm %G gnorm %G\n",fnorm,*gnorm);CHKERRQ(ierr);
1448c92abb8aSShri Abhyankar   }
14492b4ed282SShri Abhyankar   ierr = PetscLogEventEnd(SNES_LineSearch,snes,x,f,g);CHKERRQ(ierr);
14502b4ed282SShri Abhyankar   PetscFunctionReturn(0);
14512b4ed282SShri Abhyankar }
14522b4ed282SShri Abhyankar 
14532b4ed282SShri Abhyankar /* -------------------------------------------------------------------------- */
14542b4ed282SShri Abhyankar #undef __FUNCT__
14552b4ed282SShri Abhyankar #define __FUNCT__ "SNESLineSearchNoNorms_VI"
14562b4ed282SShri Abhyankar 
14572b4ed282SShri Abhyankar /*
14582b4ed282SShri Abhyankar   This routine is a copy of SNESLineSearchNoNorms in snes/impls/ls/ls.c
14592b4ed282SShri Abhyankar */
1460ace3abfcSBarry Smith PetscErrorCode SNESLineSearchNoNorms_VI(SNES snes,void *lsctx,Vec x,Vec f,Vec g,Vec y,Vec w,PetscReal fnorm,PetscReal xnorm,PetscReal *ynorm,PetscReal *gnorm,PetscBool *flag)
14612b4ed282SShri Abhyankar {
14622b4ed282SShri Abhyankar   PetscErrorCode ierr;
14632b4ed282SShri Abhyankar   SNES_VI        *vi = (SNES_VI*)snes->data;
1464ace3abfcSBarry Smith   PetscBool     changed_w = PETSC_FALSE,changed_y = PETSC_FALSE;
14652b4ed282SShri Abhyankar 
14662b4ed282SShri Abhyankar   PetscFunctionBegin;
14672b4ed282SShri Abhyankar   *flag = PETSC_TRUE;
14682b4ed282SShri Abhyankar   ierr = PetscLogEventBegin(SNES_LineSearch,snes,x,f,g);CHKERRQ(ierr);
14692b4ed282SShri Abhyankar   ierr = VecWAXPY(w,-1.0,y,x);CHKERRQ(ierr);            /* w <- x - y      */
14702b4ed282SShri Abhyankar   if (vi->postcheckstep) {
14712b4ed282SShri Abhyankar    ierr = (*vi->postcheckstep)(snes,x,y,w,vi->postcheck,&changed_y,&changed_w);CHKERRQ(ierr);
14722b4ed282SShri Abhyankar   }
14732b4ed282SShri Abhyankar   if (changed_y) {
14742b4ed282SShri Abhyankar     ierr = VecWAXPY(w,-1.0,y,x);CHKERRQ(ierr);            /* w <- x - y   */
14752b4ed282SShri Abhyankar   }
14762b4ed282SShri Abhyankar 
14772b4ed282SShri Abhyankar   /* don't evaluate function the last time through */
14782b4ed282SShri Abhyankar   if (snes->iter < snes->max_its-1) {
14792b4ed282SShri Abhyankar     ierr = SNESComputeFunction(snes,w,g);CHKERRQ(ierr);
14802b4ed282SShri Abhyankar   }
14812b4ed282SShri Abhyankar   ierr = PetscLogEventEnd(SNES_LineSearch,snes,x,f,g);CHKERRQ(ierr);
14822b4ed282SShri Abhyankar   PetscFunctionReturn(0);
14832b4ed282SShri Abhyankar }
14842b4ed282SShri Abhyankar /* -------------------------------------------------------------------------- */
14852b4ed282SShri Abhyankar #undef __FUNCT__
14862b4ed282SShri Abhyankar #define __FUNCT__ "SNESLineSearchCubic_VI"
14872b4ed282SShri Abhyankar /*
14882b4ed282SShri Abhyankar   This routine is a copy of SNESLineSearchCubic in snes/impls/ls/ls.c
14892b4ed282SShri Abhyankar */
1490ace3abfcSBarry Smith PetscErrorCode SNESLineSearchCubic_VI(SNES snes,void *lsctx,Vec x,Vec f,Vec g,Vec y,Vec w,PetscReal fnorm,PetscReal xnorm,PetscReal *ynorm,PetscReal *gnorm,PetscBool *flag)
14912b4ed282SShri Abhyankar {
14922b4ed282SShri Abhyankar   /*
14932b4ed282SShri Abhyankar      Note that for line search purposes we work with with the related
14942b4ed282SShri Abhyankar      minimization problem:
14952b4ed282SShri Abhyankar         min  z(x):  R^n -> R,
14962b4ed282SShri Abhyankar      where z(x) = .5 * fnorm*fnorm, and fnorm = || f ||_2.
14972b4ed282SShri Abhyankar      This function z(x) is same as the merit function for the complementarity problem.
14982b4ed282SShri Abhyankar    */
14992b4ed282SShri Abhyankar 
15002b4ed282SShri Abhyankar   PetscReal      initslope,lambdaprev,gnormprev,a,b,d,t1,t2,rellength;
15012b4ed282SShri Abhyankar   PetscReal      minlambda,lambda,lambdatemp;
15022b4ed282SShri Abhyankar #if defined(PETSC_USE_COMPLEX)
15032b4ed282SShri Abhyankar   PetscScalar    cinitslope;
15042b4ed282SShri Abhyankar #endif
15052b4ed282SShri Abhyankar   PetscErrorCode ierr;
15062b4ed282SShri Abhyankar   PetscInt       count;
15072b4ed282SShri Abhyankar   SNES_VI      *vi = (SNES_VI*)snes->data;
1508ace3abfcSBarry Smith   PetscBool     changed_w = PETSC_FALSE,changed_y = PETSC_FALSE;
15092b4ed282SShri Abhyankar   MPI_Comm       comm;
15102b4ed282SShri Abhyankar 
15112b4ed282SShri Abhyankar   PetscFunctionBegin;
15122b4ed282SShri Abhyankar   ierr = PetscObjectGetComm((PetscObject)snes,&comm);CHKERRQ(ierr);
15132b4ed282SShri Abhyankar   ierr = PetscLogEventBegin(SNES_LineSearch,snes,x,f,g);CHKERRQ(ierr);
15142b4ed282SShri Abhyankar   *flag   = PETSC_TRUE;
15152b4ed282SShri Abhyankar 
15162b4ed282SShri Abhyankar   ierr = VecNorm(y,NORM_2,ynorm);CHKERRQ(ierr);
15172b4ed282SShri Abhyankar   if (*ynorm == 0.0) {
1518c92abb8aSShri Abhyankar     if (vi->lsmonitor) {
1519c92abb8aSShri Abhyankar       ierr = PetscViewerASCIIMonitorPrintf(vi->lsmonitor,"    Line search: Initial direction and size is 0\n");CHKERRQ(ierr);
15202b4ed282SShri Abhyankar     }
15212b4ed282SShri Abhyankar     *gnorm = fnorm;
15222b4ed282SShri Abhyankar     ierr   = VecCopy(x,w);CHKERRQ(ierr);
15232b4ed282SShri Abhyankar     ierr   = VecCopy(f,g);CHKERRQ(ierr);
15242b4ed282SShri Abhyankar     *flag  = PETSC_FALSE;
15252b4ed282SShri Abhyankar     goto theend1;
15262b4ed282SShri Abhyankar   }
15272b4ed282SShri Abhyankar   if (*ynorm > vi->maxstep) {	/* Step too big, so scale back */
1528c92abb8aSShri Abhyankar     if (vi->lsmonitor) {
1529c92abb8aSShri Abhyankar       ierr = PetscViewerASCIIMonitorPrintf(vi->lsmonitor,"    Line search: Scaling step by %G old ynorm %G\n",vi->maxstep/(*ynorm),*ynorm);CHKERRQ(ierr);
15302b4ed282SShri Abhyankar     }
15312b4ed282SShri Abhyankar     ierr = VecScale(y,vi->maxstep/(*ynorm));CHKERRQ(ierr);
15322b4ed282SShri Abhyankar     *ynorm = vi->maxstep;
15332b4ed282SShri Abhyankar   }
15342b4ed282SShri Abhyankar   ierr      = VecMaxPointwiseDivide(y,x,&rellength);CHKERRQ(ierr);
15352b4ed282SShri Abhyankar   minlambda = vi->minlambda/rellength;
15362b4ed282SShri Abhyankar #if defined(PETSC_USE_COMPLEX)
153789d9e806SShri Abhyankar   ierr = VecDot(vi->dpsi,y,&cinitslope);CHKERRQ(ierr);
15382b4ed282SShri Abhyankar   initslope = PetscRealPart(cinitslope);
15392b4ed282SShri Abhyankar #else
154089d9e806SShri Abhyankar   ierr = VecDot(vi->dpsi,y,&initslope);CHKERRQ(ierr);
15412b4ed282SShri Abhyankar #endif
15422b4ed282SShri Abhyankar   if (initslope > 0.0)  initslope = -initslope;
15432b4ed282SShri Abhyankar   if (initslope == 0.0) initslope = -1.0;
15442b4ed282SShri Abhyankar 
15452b4ed282SShri Abhyankar   ierr = VecWAXPY(w,-1.0,y,x);CHKERRQ(ierr);
15462b4ed282SShri Abhyankar   if (snes->nfuncs >= snes->max_funcs) {
15472b4ed282SShri Abhyankar     ierr  = PetscInfo(snes,"Exceeded maximum function evaluations, while checking full step length!\n");CHKERRQ(ierr);
15482b4ed282SShri Abhyankar     *flag = PETSC_FALSE;
15492b4ed282SShri Abhyankar     snes->reason = SNES_DIVERGED_FUNCTION_COUNT;
15502b4ed282SShri Abhyankar     goto theend1;
15512b4ed282SShri Abhyankar   }
15522b4ed282SShri Abhyankar   ierr = SNESComputeFunction(snes,w,g);CHKERRQ(ierr);
15532b4ed282SShri Abhyankar   if (snes->domainerror) {
15542b4ed282SShri Abhyankar     ierr = PetscLogEventEnd(SNES_LineSearch,snes,x,f,g);CHKERRQ(ierr);
15552b4ed282SShri Abhyankar     PetscFunctionReturn(0);
15562b4ed282SShri Abhyankar   }
15572b4ed282SShri Abhyankar   ierr = VecNorm(g,NORM_2,gnorm);CHKERRQ(ierr);
15582b4ed282SShri Abhyankar   if PetscIsInfOrNanReal(*gnorm) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_FP,"User provided compute function generated a Not-a-Number");
15592b4ed282SShri Abhyankar   ierr = PetscInfo2(snes,"Initial fnorm %G gnorm %G\n",fnorm,*gnorm);CHKERRQ(ierr);
15602b4ed282SShri Abhyankar   if (.5*(*gnorm)*(*gnorm) <= .5*fnorm*fnorm + vi->alpha*initslope) { /* Sufficient reduction */
1561c92abb8aSShri Abhyankar     if (vi->lsmonitor) {
1562c92abb8aSShri Abhyankar       ierr = PetscViewerASCIIMonitorPrintf(vi->lsmonitor,"    Line search: Using full step: fnorm %G gnorm %G\n",fnorm,*gnorm);CHKERRQ(ierr);
15632b4ed282SShri Abhyankar     }
15642b4ed282SShri Abhyankar     goto theend1;
15652b4ed282SShri Abhyankar   }
15662b4ed282SShri Abhyankar 
15672b4ed282SShri Abhyankar   /* Fit points with quadratic */
15682b4ed282SShri Abhyankar   lambda     = 1.0;
15692b4ed282SShri Abhyankar   lambdatemp = -initslope/((*gnorm)*(*gnorm) - fnorm*fnorm - 2.0*initslope);
15702b4ed282SShri Abhyankar   lambdaprev = lambda;
15712b4ed282SShri Abhyankar   gnormprev  = *gnorm;
15722b4ed282SShri Abhyankar   if (lambdatemp > .5*lambda)  lambdatemp = .5*lambda;
15732b4ed282SShri Abhyankar   if (lambdatemp <= .1*lambda) lambda = .1*lambda;
15742b4ed282SShri Abhyankar   else                         lambda = lambdatemp;
15752b4ed282SShri Abhyankar 
15762b4ed282SShri Abhyankar   ierr  = VecWAXPY(w,-lambda,y,x);CHKERRQ(ierr);
15772b4ed282SShri Abhyankar   if (snes->nfuncs >= snes->max_funcs) {
15782b4ed282SShri Abhyankar     ierr  = PetscInfo1(snes,"Exceeded maximum function evaluations, while attempting quadratic backtracking! %D \n",snes->nfuncs);CHKERRQ(ierr);
15792b4ed282SShri Abhyankar     *flag = PETSC_FALSE;
15802b4ed282SShri Abhyankar     snes->reason = SNES_DIVERGED_FUNCTION_COUNT;
15812b4ed282SShri Abhyankar     goto theend1;
15822b4ed282SShri Abhyankar   }
15832b4ed282SShri Abhyankar   ierr = SNESComputeFunction(snes,w,g);CHKERRQ(ierr);
15842b4ed282SShri Abhyankar   if (snes->domainerror) {
15852b4ed282SShri Abhyankar     ierr = PetscLogEventEnd(SNES_LineSearch,snes,x,f,g);CHKERRQ(ierr);
15862b4ed282SShri Abhyankar     PetscFunctionReturn(0);
15872b4ed282SShri Abhyankar   }
15882b4ed282SShri Abhyankar   ierr = VecNorm(g,NORM_2,gnorm);CHKERRQ(ierr);
15892b4ed282SShri Abhyankar   if PetscIsInfOrNanReal(*gnorm) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_FP,"User provided compute function generated a Not-a-Number");
1590c92abb8aSShri Abhyankar   if (vi->lsmonitor) {
1591c92abb8aSShri Abhyankar     ierr = PetscViewerASCIIMonitorPrintf(vi->lsmonitor,"    Line search: gnorm after quadratic fit %G\n",*gnorm);CHKERRQ(ierr);
15922b4ed282SShri Abhyankar   }
15932b4ed282SShri Abhyankar   if (.5*(*gnorm)*(*gnorm) < .5*fnorm*fnorm + lambda*vi->alpha*initslope) { /* sufficient reduction */
1594c92abb8aSShri Abhyankar     if (vi->lsmonitor) {
1595c92abb8aSShri Abhyankar       ierr = PetscViewerASCIIMonitorPrintf(vi->lsmonitor,"    Line search: Quadratically determined step, lambda=%18.16e\n",lambda);CHKERRQ(ierr);
15962b4ed282SShri Abhyankar     }
15972b4ed282SShri Abhyankar     goto theend1;
15982b4ed282SShri Abhyankar   }
15992b4ed282SShri Abhyankar 
16002b4ed282SShri Abhyankar   /* Fit points with cubic */
16012b4ed282SShri Abhyankar   count = 1;
16022b4ed282SShri Abhyankar   while (PETSC_TRUE) {
16032b4ed282SShri Abhyankar     if (lambda <= minlambda) {
1604c92abb8aSShri Abhyankar       if (vi->lsmonitor) {
1605c92abb8aSShri Abhyankar 	ierr = PetscViewerASCIIMonitorPrintf(vi->lsmonitor,"    Line search: unable to find good step length! After %D tries \n",count);CHKERRQ(ierr);
1606c92abb8aSShri Abhyankar 	ierr = PetscViewerASCIIMonitorPrintf(vi->lsmonitor,"    Line search: fnorm=%18.16e, gnorm=%18.16e, ynorm=%18.16e, minlambda=%18.16e, lambda=%18.16e, initial slope=%18.16e\n",fnorm,*gnorm,*ynorm,minlambda,lambda,initslope);CHKERRQ(ierr);
16072b4ed282SShri Abhyankar       }
16082b4ed282SShri Abhyankar       *flag = PETSC_FALSE;
16092b4ed282SShri Abhyankar       break;
16102b4ed282SShri Abhyankar     }
16112b4ed282SShri Abhyankar     t1 = .5*((*gnorm)*(*gnorm) - fnorm*fnorm) - lambda*initslope;
16122b4ed282SShri Abhyankar     t2 = .5*(gnormprev*gnormprev  - fnorm*fnorm) - lambdaprev*initslope;
16132b4ed282SShri Abhyankar     a  = (t1/(lambda*lambda) - t2/(lambdaprev*lambdaprev))/(lambda-lambdaprev);
16142b4ed282SShri Abhyankar     b  = (-lambdaprev*t1/(lambda*lambda) + lambda*t2/(lambdaprev*lambdaprev))/(lambda-lambdaprev);
16152b4ed282SShri Abhyankar     d  = b*b - 3*a*initslope;
16162b4ed282SShri Abhyankar     if (d < 0.0) d = 0.0;
16172b4ed282SShri Abhyankar     if (a == 0.0) {
16182b4ed282SShri Abhyankar       lambdatemp = -initslope/(2.0*b);
16192b4ed282SShri Abhyankar     } else {
16202b4ed282SShri Abhyankar       lambdatemp = (-b + sqrt(d))/(3.0*a);
16212b4ed282SShri Abhyankar     }
16222b4ed282SShri Abhyankar     lambdaprev = lambda;
16232b4ed282SShri Abhyankar     gnormprev  = *gnorm;
16242b4ed282SShri Abhyankar     if (lambdatemp > .5*lambda)  lambdatemp = .5*lambda;
16252b4ed282SShri Abhyankar     if (lambdatemp <= .1*lambda) lambda     = .1*lambda;
16262b4ed282SShri Abhyankar     else                         lambda     = lambdatemp;
16272b4ed282SShri Abhyankar     ierr  = VecWAXPY(w,-lambda,y,x);CHKERRQ(ierr);
16282b4ed282SShri Abhyankar     if (snes->nfuncs >= snes->max_funcs) {
16292b4ed282SShri Abhyankar       ierr = PetscInfo1(snes,"Exceeded maximum function evaluations, while looking for good step length! %D \n",count);CHKERRQ(ierr);
16302b4ed282SShri Abhyankar       ierr = PetscInfo5(snes,"fnorm=%18.16e, gnorm=%18.16e, ynorm=%18.16e, lambda=%18.16e, initial slope=%18.16e\n",fnorm,*gnorm,*ynorm,lambda,initslope);CHKERRQ(ierr);
16312b4ed282SShri Abhyankar       *flag = PETSC_FALSE;
16322b4ed282SShri Abhyankar       snes->reason = SNES_DIVERGED_FUNCTION_COUNT;
16332b4ed282SShri Abhyankar       break;
16342b4ed282SShri Abhyankar     }
16352b4ed282SShri Abhyankar     ierr = SNESComputeFunction(snes,w,g);CHKERRQ(ierr);
16362b4ed282SShri Abhyankar     if (snes->domainerror) {
16372b4ed282SShri Abhyankar       ierr = PetscLogEventEnd(SNES_LineSearch,snes,x,f,g);CHKERRQ(ierr);
16382b4ed282SShri Abhyankar       PetscFunctionReturn(0);
16392b4ed282SShri Abhyankar     }
16402b4ed282SShri Abhyankar     ierr = VecNorm(g,NORM_2,gnorm);CHKERRQ(ierr);
16412b4ed282SShri Abhyankar     if PetscIsInfOrNanReal(*gnorm) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_FP,"User provided compute function generated a Not-a-Number");
16422b4ed282SShri Abhyankar     if (.5*(*gnorm)*(*gnorm) < .5*fnorm*fnorm + lambda*vi->alpha*initslope) { /* is reduction enough? */
1643c92abb8aSShri Abhyankar       if (vi->lsmonitor) {
16442b4ed282SShri Abhyankar 	ierr = PetscPrintf(comm,"    Line search: Cubically determined step, current gnorm %G lambda=%18.16e\n",*gnorm,lambda);CHKERRQ(ierr);
16452b4ed282SShri Abhyankar       }
16462b4ed282SShri Abhyankar       break;
16472b4ed282SShri Abhyankar     } else {
1648c92abb8aSShri Abhyankar       if (vi->lsmonitor) {
16492b4ed282SShri Abhyankar         ierr = PetscPrintf(comm,"    Line search: Cubic step no good, shrinking lambda, current gnorm %G lambda=%18.16e\n",*gnorm,lambda);CHKERRQ(ierr);
16502b4ed282SShri Abhyankar       }
16512b4ed282SShri Abhyankar     }
16522b4ed282SShri Abhyankar     count++;
16532b4ed282SShri Abhyankar   }
16542b4ed282SShri Abhyankar   theend1:
16552b4ed282SShri Abhyankar   /* Optional user-defined check for line search step validity */
16562b4ed282SShri Abhyankar   if (vi->postcheckstep && *flag) {
16572b4ed282SShri Abhyankar     ierr = (*vi->postcheckstep)(snes,x,y,w,vi->postcheck,&changed_y,&changed_w);CHKERRQ(ierr);
16582b4ed282SShri Abhyankar     if (changed_y) {
16592b4ed282SShri Abhyankar       ierr = VecWAXPY(w,-1.0,y,x);CHKERRQ(ierr);
16602b4ed282SShri Abhyankar     }
16612b4ed282SShri Abhyankar     if (changed_y || changed_w) { /* recompute the function if the step has changed */
16622b4ed282SShri Abhyankar       ierr = SNESComputeFunction(snes,w,g);CHKERRQ(ierr);
16632b4ed282SShri Abhyankar       if (snes->domainerror) {
16642b4ed282SShri Abhyankar         ierr = PetscLogEventEnd(SNES_LineSearch,snes,x,f,g);CHKERRQ(ierr);
16652b4ed282SShri Abhyankar         PetscFunctionReturn(0);
16662b4ed282SShri Abhyankar       }
16672b4ed282SShri Abhyankar       ierr = VecNormBegin(g,NORM_2,gnorm);CHKERRQ(ierr);
16682b4ed282SShri Abhyankar       if PetscIsInfOrNanReal(*gnorm) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_FP,"User provided compute function generated a Not-a-Number");
16692b4ed282SShri Abhyankar       ierr = VecNormBegin(y,NORM_2,ynorm);CHKERRQ(ierr);
16702b4ed282SShri Abhyankar       ierr = VecNormEnd(g,NORM_2,gnorm);CHKERRQ(ierr);
16712b4ed282SShri Abhyankar       ierr = VecNormEnd(y,NORM_2,ynorm);CHKERRQ(ierr);
16722b4ed282SShri Abhyankar     }
16732b4ed282SShri Abhyankar   }
16742b4ed282SShri Abhyankar   ierr = PetscLogEventEnd(SNES_LineSearch,snes,x,f,g);CHKERRQ(ierr);
16752b4ed282SShri Abhyankar   PetscFunctionReturn(0);
16762b4ed282SShri Abhyankar }
16772b4ed282SShri Abhyankar /* -------------------------------------------------------------------------- */
1678*fe835674SShri Abhyankar /*
1679*fe835674SShri Abhyankar  Projects vector X onto the variable bounds such that
1680*fe835674SShri Abhyankar    Xl[i] <= X[i] <= Xu[i]  for i = 1..n
1681*fe835674SShri Abhyankar */
1682*fe835674SShri Abhyankar 
1683*fe835674SShri Abhyankar #undef __FUNCT__
1684*fe835674SShri Abhyankar #define __FUNCT__ "SNESVIProjectOntoBounds"
1685*fe835674SShri Abhyankar PetscErrorCode SNESVIProjectOntoBounds(SNES snes,Vec X)
1686*fe835674SShri Abhyankar {
1687*fe835674SShri Abhyankar   PetscErrorCode    ierr;
1688*fe835674SShri Abhyankar   SNES_VI           *vi = (SNES_VI*)snes->data;
1689*fe835674SShri Abhyankar   const PetscScalar *xl,*xu;
1690*fe835674SShri Abhyankar   PetscScalar       *x;
1691*fe835674SShri Abhyankar   PetscInt          i,n;
1692*fe835674SShri Abhyankar 
1693*fe835674SShri Abhyankar   PetscFunctionBegin;
1694*fe835674SShri Abhyankar 
1695*fe835674SShri Abhyankar   ierr = VecGetLocalSize(X,&n);CHKERRQ(ierr);
1696*fe835674SShri Abhyankar   ierr = VecGetArray(X,&x);CHKERRQ(ierr);
1697*fe835674SShri Abhyankar   ierr = VecGetArrayRead(vi->xl,&xl);CHKERRQ(ierr);
1698*fe835674SShri Abhyankar   ierr = VecGetArrayRead(vi->xu,&xu);CHKERRQ(ierr);
1699*fe835674SShri Abhyankar 
1700*fe835674SShri Abhyankar   for(i = 0;i<n;i++) {
1701*fe835674SShri Abhyankar     if (x[i] < xl[i]) x[i] = xl[i];
1702*fe835674SShri Abhyankar     else if (x[i] > xu[i]) x[i] = xu[i];
1703*fe835674SShri Abhyankar   }
1704*fe835674SShri Abhyankar   ierr = VecRestoreArray(X,&x);CHKERRQ(ierr);
1705*fe835674SShri Abhyankar   ierr = VecRestoreArrayRead(vi->xl,&xl);CHKERRQ(ierr);
1706*fe835674SShri Abhyankar   ierr = VecRestoreArrayRead(vi->xu,&xu);CHKERRQ(ierr);
1707*fe835674SShri Abhyankar 
1708*fe835674SShri Abhyankar   PetscFunctionReturn(0);
1709*fe835674SShri Abhyankar }
1710*fe835674SShri Abhyankar 
1711*fe835674SShri Abhyankar /* -------------------------------------------------------------------------- */
1712*fe835674SShri Abhyankar #undef __FUNCT__
1713*fe835674SShri Abhyankar #define __FUNCT__ "SNESLineSearchBoundProjectedCubic_VI"
1714*fe835674SShri Abhyankar /*
1715*fe835674SShri Abhyankar   This routine implements a bound projected cubic line search
1716*fe835674SShri Abhyankar */
1717*fe835674SShri Abhyankar PetscErrorCode SNESLineSearchBoundProjectedCubic_VI(SNES snes,void *lsctx,Vec x,Vec f,Vec g,Vec y,Vec w,PetscReal fnorm,PetscReal xnorm,PetscReal *ynorm,PetscReal *gnorm,PetscBool *flag)
1718*fe835674SShri Abhyankar {
1719*fe835674SShri Abhyankar   PetscReal      initslope,lambdaprev,gnormprev,a,b,d,t1,t2,rellength;
1720*fe835674SShri Abhyankar   PetscReal      minlambda,lambda,lambdatemp;
1721*fe835674SShri Abhyankar #if defined(PETSC_USE_COMPLEX)
1722*fe835674SShri Abhyankar   PetscScalar    cinitslope;
1723*fe835674SShri Abhyankar #endif
1724*fe835674SShri Abhyankar   PetscErrorCode ierr;
1725*fe835674SShri Abhyankar   PetscInt       count;
1726*fe835674SShri Abhyankar   SNES_VI        *vi = (SNES_VI*)snes->data;
1727*fe835674SShri Abhyankar   PetscBool      changed_w = PETSC_FALSE,changed_y = PETSC_FALSE;
1728*fe835674SShri Abhyankar   MPI_Comm       comm;
1729*fe835674SShri Abhyankar 
1730*fe835674SShri Abhyankar   PetscFunctionBegin;
1731*fe835674SShri Abhyankar   ierr = PetscObjectGetComm((PetscObject)snes,&comm);CHKERRQ(ierr);
1732*fe835674SShri Abhyankar   ierr = PetscLogEventBegin(SNES_LineSearch,snes,x,f,g);CHKERRQ(ierr);
1733*fe835674SShri Abhyankar   *flag   = PETSC_TRUE;
1734*fe835674SShri Abhyankar 
1735*fe835674SShri Abhyankar   ierr = VecNorm(y,NORM_2,ynorm);CHKERRQ(ierr);
1736*fe835674SShri Abhyankar   if (*ynorm == 0.0) {
1737*fe835674SShri Abhyankar     if (vi->lsmonitor) {
1738*fe835674SShri Abhyankar       ierr = PetscViewerASCIIMonitorPrintf(vi->lsmonitor,"    Line search: Initial direction and size is 0\n");CHKERRQ(ierr);
1739*fe835674SShri Abhyankar     }
1740*fe835674SShri Abhyankar     *gnorm = fnorm;
1741*fe835674SShri Abhyankar     ierr   = VecCopy(x,w);CHKERRQ(ierr);
1742*fe835674SShri Abhyankar     ierr   = VecCopy(f,g);CHKERRQ(ierr);
1743*fe835674SShri Abhyankar     *flag  = PETSC_FALSE;
1744*fe835674SShri Abhyankar     goto theend1;
1745*fe835674SShri Abhyankar   }
1746*fe835674SShri Abhyankar   if (*ynorm > vi->maxstep) {	/* Step too big, so scale back */
1747*fe835674SShri Abhyankar     if (vi->lsmonitor) {
1748*fe835674SShri Abhyankar       ierr = PetscViewerASCIIMonitorPrintf(vi->lsmonitor,"    Line search: Scaling step by %G old ynorm %G\n",vi->maxstep/(*ynorm),*ynorm);CHKERRQ(ierr);
1749*fe835674SShri Abhyankar     }
1750*fe835674SShri Abhyankar     ierr = VecScale(y,vi->maxstep/(*ynorm));CHKERRQ(ierr);
1751*fe835674SShri Abhyankar     *ynorm = vi->maxstep;
1752*fe835674SShri Abhyankar   }
1753*fe835674SShri Abhyankar   ierr      = VecMaxPointwiseDivide(y,x,&rellength);CHKERRQ(ierr);
1754*fe835674SShri Abhyankar   minlambda = vi->minlambda/rellength;
1755*fe835674SShri Abhyankar   ierr = MatMult(snes->jacobian,y,w);CHKERRQ(ierr);
1756*fe835674SShri Abhyankar #if defined(PETSC_USE_COMPLEX)
1757*fe835674SShri Abhyankar   ierr = VecDot(f,w,&cinitslope);CHKERRQ(ierr);
1758*fe835674SShri Abhyankar   initslope = PetscRealPart(cinitslope);
1759*fe835674SShri Abhyankar #else
1760*fe835674SShri Abhyankar   ierr = VecDot(f,w,&initslope);CHKERRQ(ierr);
1761*fe835674SShri Abhyankar #endif
1762*fe835674SShri Abhyankar   if (initslope > 0.0)  initslope = -initslope;
1763*fe835674SShri Abhyankar   if (initslope == 0.0) initslope = -1.0;
1764*fe835674SShri Abhyankar 
1765*fe835674SShri Abhyankar   ierr = VecWAXPY(w,-1.0,y,x);CHKERRQ(ierr);
1766*fe835674SShri Abhyankar   ierr = SNESVIProjectOntoBounds(snes,w);CHKERRQ(ierr);
1767*fe835674SShri Abhyankar   if (snes->nfuncs >= snes->max_funcs) {
1768*fe835674SShri Abhyankar     ierr  = PetscInfo(snes,"Exceeded maximum function evaluations, while checking full step length!\n");CHKERRQ(ierr);
1769*fe835674SShri Abhyankar     *flag = PETSC_FALSE;
1770*fe835674SShri Abhyankar     snes->reason = SNES_DIVERGED_FUNCTION_COUNT;
1771*fe835674SShri Abhyankar     goto theend1;
1772*fe835674SShri Abhyankar   }
1773*fe835674SShri Abhyankar   ierr = SNESComputeFunction(snes,w,g);CHKERRQ(ierr);
1774*fe835674SShri Abhyankar   ierr = SNESVIComputeInactiveSetFnorm(snes,g,w,gnorm);CHKERRQ(ierr);
1775*fe835674SShri Abhyankar   if (snes->domainerror) {
1776*fe835674SShri Abhyankar     ierr = PetscLogEventEnd(SNES_LineSearch,snes,x,f,g);CHKERRQ(ierr);
1777*fe835674SShri Abhyankar     PetscFunctionReturn(0);
1778*fe835674SShri Abhyankar   }
1779*fe835674SShri Abhyankar   if PetscIsInfOrNanReal(*gnorm) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_FP,"User provided compute function generated a Not-a-Number");
1780*fe835674SShri Abhyankar   ierr = PetscInfo2(snes,"Initial fnorm %G gnorm %G\n",fnorm,*gnorm);CHKERRQ(ierr);
1781*fe835674SShri Abhyankar   if (.5*(*gnorm)*(*gnorm) <= .5*fnorm*fnorm + vi->alpha*initslope) { /* Sufficient reduction */
1782*fe835674SShri Abhyankar     if (vi->lsmonitor) {
1783*fe835674SShri Abhyankar       ierr = PetscViewerASCIIMonitorPrintf(vi->lsmonitor,"    Line search: Using full step: fnorm %G gnorm %G\n",fnorm,*gnorm);CHKERRQ(ierr);
1784*fe835674SShri Abhyankar     }
1785*fe835674SShri Abhyankar     goto theend1;
1786*fe835674SShri Abhyankar   }
1787*fe835674SShri Abhyankar 
1788*fe835674SShri Abhyankar   /* Fit points with quadratic */
1789*fe835674SShri Abhyankar   lambda     = 1.0;
1790*fe835674SShri Abhyankar   lambdatemp = -initslope/((*gnorm)*(*gnorm) - fnorm*fnorm - 2.0*initslope);
1791*fe835674SShri Abhyankar   lambdaprev = lambda;
1792*fe835674SShri Abhyankar   gnormprev  = *gnorm;
1793*fe835674SShri Abhyankar   if (lambdatemp > .5*lambda)  lambdatemp = .5*lambda;
1794*fe835674SShri Abhyankar   if (lambdatemp <= .1*lambda) lambda = .1*lambda;
1795*fe835674SShri Abhyankar   else                         lambda = lambdatemp;
1796*fe835674SShri Abhyankar 
1797*fe835674SShri Abhyankar   ierr  = VecWAXPY(w,-lambda,y,x);CHKERRQ(ierr);
1798*fe835674SShri Abhyankar   ierr = SNESVIProjectOntoBounds(snes,w);CHKERRQ(ierr);
1799*fe835674SShri Abhyankar   if (snes->nfuncs >= snes->max_funcs) {
1800*fe835674SShri Abhyankar     ierr  = PetscInfo1(snes,"Exceeded maximum function evaluations, while attempting quadratic backtracking! %D \n",snes->nfuncs);CHKERRQ(ierr);
1801*fe835674SShri Abhyankar     *flag = PETSC_FALSE;
1802*fe835674SShri Abhyankar     snes->reason = SNES_DIVERGED_FUNCTION_COUNT;
1803*fe835674SShri Abhyankar     goto theend1;
1804*fe835674SShri Abhyankar   }
1805*fe835674SShri Abhyankar   ierr = SNESComputeFunction(snes,w,g);CHKERRQ(ierr);
1806*fe835674SShri Abhyankar   ierr = SNESVIComputeInactiveSetFnorm(snes,g,w,gnorm);CHKERRQ(ierr);
1807*fe835674SShri Abhyankar   if (snes->domainerror) {
1808*fe835674SShri Abhyankar     ierr = PetscLogEventEnd(SNES_LineSearch,snes,x,f,g);CHKERRQ(ierr);
1809*fe835674SShri Abhyankar     PetscFunctionReturn(0);
1810*fe835674SShri Abhyankar   }
1811*fe835674SShri Abhyankar   if PetscIsInfOrNanReal(*gnorm) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_FP,"User provided compute function generated a Not-a-Number");
1812*fe835674SShri Abhyankar   if (vi->lsmonitor) {
1813*fe835674SShri Abhyankar     ierr = PetscViewerASCIIMonitorPrintf(vi->lsmonitor,"    Line search: gnorm after quadratic fit %G\n",*gnorm);CHKERRQ(ierr);
1814*fe835674SShri Abhyankar   }
1815*fe835674SShri Abhyankar   if (.5*(*gnorm)*(*gnorm) < .5*fnorm*fnorm + lambda*vi->alpha*initslope) { /* sufficient reduction */
1816*fe835674SShri Abhyankar     if (vi->lsmonitor) {
1817*fe835674SShri Abhyankar       ierr = PetscViewerASCIIMonitorPrintf(vi->lsmonitor,"    Line search: Quadratically determined step, lambda=%18.16e\n",lambda);CHKERRQ(ierr);
1818*fe835674SShri Abhyankar     }
1819*fe835674SShri Abhyankar     goto theend1;
1820*fe835674SShri Abhyankar   }
1821*fe835674SShri Abhyankar 
1822*fe835674SShri Abhyankar   /* Fit points with cubic */
1823*fe835674SShri Abhyankar   count = 1;
1824*fe835674SShri Abhyankar   while (PETSC_TRUE) {
1825*fe835674SShri Abhyankar     if (lambda <= minlambda) {
1826*fe835674SShri Abhyankar       if (vi->lsmonitor) {
1827*fe835674SShri Abhyankar 	ierr = PetscViewerASCIIMonitorPrintf(vi->lsmonitor,"    Line search: unable to find good step length! After %D tries \n",count);CHKERRQ(ierr);
1828*fe835674SShri Abhyankar 	ierr = PetscViewerASCIIMonitorPrintf(vi->lsmonitor,"    Line search: fnorm=%18.16e, gnorm=%18.16e, ynorm=%18.16e, minlambda=%18.16e, lambda=%18.16e, initial slope=%18.16e\n",fnorm,*gnorm,*ynorm,minlambda,lambda,initslope);CHKERRQ(ierr);
1829*fe835674SShri Abhyankar       }
1830*fe835674SShri Abhyankar       *flag = PETSC_FALSE;
1831*fe835674SShri Abhyankar       break;
1832*fe835674SShri Abhyankar     }
1833*fe835674SShri Abhyankar     t1 = .5*((*gnorm)*(*gnorm) - fnorm*fnorm) - lambda*initslope;
1834*fe835674SShri Abhyankar     t2 = .5*(gnormprev*gnormprev  - fnorm*fnorm) - lambdaprev*initslope;
1835*fe835674SShri Abhyankar     a  = (t1/(lambda*lambda) - t2/(lambdaprev*lambdaprev))/(lambda-lambdaprev);
1836*fe835674SShri Abhyankar     b  = (-lambdaprev*t1/(lambda*lambda) + lambda*t2/(lambdaprev*lambdaprev))/(lambda-lambdaprev);
1837*fe835674SShri Abhyankar     d  = b*b - 3*a*initslope;
1838*fe835674SShri Abhyankar     if (d < 0.0) d = 0.0;
1839*fe835674SShri Abhyankar     if (a == 0.0) {
1840*fe835674SShri Abhyankar       lambdatemp = -initslope/(2.0*b);
1841*fe835674SShri Abhyankar     } else {
1842*fe835674SShri Abhyankar       lambdatemp = (-b + sqrt(d))/(3.0*a);
1843*fe835674SShri Abhyankar     }
1844*fe835674SShri Abhyankar     lambdaprev = lambda;
1845*fe835674SShri Abhyankar     gnormprev  = *gnorm;
1846*fe835674SShri Abhyankar     if (lambdatemp > .5*lambda)  lambdatemp = .5*lambda;
1847*fe835674SShri Abhyankar     if (lambdatemp <= .1*lambda) lambda     = .1*lambda;
1848*fe835674SShri Abhyankar     else                         lambda     = lambdatemp;
1849*fe835674SShri Abhyankar     ierr = VecWAXPY(w,-lambda,y,x);CHKERRQ(ierr);
1850*fe835674SShri Abhyankar     ierr = SNESVIProjectOntoBounds(snes,w);CHKERRQ(ierr);
1851*fe835674SShri Abhyankar     if (snes->nfuncs >= snes->max_funcs) {
1852*fe835674SShri Abhyankar       ierr = PetscInfo1(snes,"Exceeded maximum function evaluations, while looking for good step length! %D \n",count);CHKERRQ(ierr);
1853*fe835674SShri Abhyankar       ierr = PetscInfo5(snes,"fnorm=%18.16e, gnorm=%18.16e, ynorm=%18.16e, lambda=%18.16e, initial slope=%18.16e\n",fnorm,*gnorm,*ynorm,lambda,initslope);CHKERRQ(ierr);
1854*fe835674SShri Abhyankar       *flag = PETSC_FALSE;
1855*fe835674SShri Abhyankar       snes->reason = SNES_DIVERGED_FUNCTION_COUNT;
1856*fe835674SShri Abhyankar       break;
1857*fe835674SShri Abhyankar     }
1858*fe835674SShri Abhyankar     ierr = SNESComputeFunction(snes,w,g);CHKERRQ(ierr);
1859*fe835674SShri Abhyankar     ierr = SNESVIComputeInactiveSetFnorm(snes,g,w,gnorm);CHKERRQ(ierr);
1860*fe835674SShri Abhyankar     if (snes->domainerror) {
1861*fe835674SShri Abhyankar       ierr = PetscLogEventEnd(SNES_LineSearch,snes,x,f,g);CHKERRQ(ierr);
1862*fe835674SShri Abhyankar       PetscFunctionReturn(0);
1863*fe835674SShri Abhyankar     }
1864*fe835674SShri Abhyankar     if PetscIsInfOrNanReal(*gnorm) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_FP,"User provided compute function generated a Not-a-Number");
1865*fe835674SShri Abhyankar     if (.5*(*gnorm)*(*gnorm) < .5*fnorm*fnorm + lambda*vi->alpha*initslope) { /* is reduction enough? */
1866*fe835674SShri Abhyankar       if (vi->lsmonitor) {
1867*fe835674SShri Abhyankar 	ierr = PetscPrintf(comm,"    Line search: Cubically determined step, current gnorm %G lambda=%18.16e\n",*gnorm,lambda);CHKERRQ(ierr);
1868*fe835674SShri Abhyankar       }
1869*fe835674SShri Abhyankar       break;
1870*fe835674SShri Abhyankar     } else {
1871*fe835674SShri Abhyankar       if (vi->lsmonitor) {
1872*fe835674SShri Abhyankar         ierr = PetscPrintf(comm,"    Line search: Cubic step no good, shrinking lambda, current gnorm %G lambda=%18.16e\n",*gnorm,lambda);CHKERRQ(ierr);
1873*fe835674SShri Abhyankar       }
1874*fe835674SShri Abhyankar     }
1875*fe835674SShri Abhyankar     count++;
1876*fe835674SShri Abhyankar   }
1877*fe835674SShri Abhyankar   theend1:
1878*fe835674SShri Abhyankar   /* Optional user-defined check for line search step validity */
1879*fe835674SShri Abhyankar   if (vi->postcheckstep && *flag) {
1880*fe835674SShri Abhyankar     ierr = (*vi->postcheckstep)(snes,x,y,w,vi->postcheck,&changed_y,&changed_w);CHKERRQ(ierr);
1881*fe835674SShri Abhyankar     if (changed_y) {
1882*fe835674SShri Abhyankar       ierr = VecWAXPY(w,-1.0,y,x);CHKERRQ(ierr);
1883*fe835674SShri Abhyankar       ierr = SNESVIProjectOntoBounds(snes,w);CHKERRQ(ierr);
1884*fe835674SShri Abhyankar     }
1885*fe835674SShri Abhyankar     if (changed_y || changed_w) { /* recompute the function if the step has changed */
1886*fe835674SShri Abhyankar       ierr = SNESComputeFunction(snes,w,g);CHKERRQ(ierr);
1887*fe835674SShri Abhyankar       ierr = SNESVIComputeInactiveSetFnorm(snes,g,w,gnorm);CHKERRQ(ierr);
1888*fe835674SShri Abhyankar       if (snes->domainerror) {
1889*fe835674SShri Abhyankar         ierr = PetscLogEventEnd(SNES_LineSearch,snes,x,f,g);CHKERRQ(ierr);
1890*fe835674SShri Abhyankar         PetscFunctionReturn(0);
1891*fe835674SShri Abhyankar       }
1892*fe835674SShri Abhyankar       if PetscIsInfOrNanReal(*gnorm) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_FP,"User provided compute function generated a Not-a-Number");
1893*fe835674SShri Abhyankar       ierr = VecNormBegin(y,NORM_2,ynorm);CHKERRQ(ierr);
1894*fe835674SShri Abhyankar       ierr = VecNormEnd(y,NORM_2,ynorm);CHKERRQ(ierr);
1895*fe835674SShri Abhyankar     }
1896*fe835674SShri Abhyankar   }
1897*fe835674SShri Abhyankar   ierr = PetscLogEventEnd(SNES_LineSearch,snes,x,f,g);CHKERRQ(ierr);
1898*fe835674SShri Abhyankar   PetscFunctionReturn(0);
1899*fe835674SShri Abhyankar }
1900*fe835674SShri Abhyankar 
19012b4ed282SShri Abhyankar #undef __FUNCT__
19022b4ed282SShri Abhyankar #define __FUNCT__ "SNESLineSearchQuadratic_VI"
19032b4ed282SShri Abhyankar /*
19042b4ed282SShri Abhyankar   This routine is a copy of SNESLineSearchQuadratic in snes/impls/ls/ls.c
19052b4ed282SShri Abhyankar */
1906ace3abfcSBarry Smith PetscErrorCode SNESLineSearchQuadratic_VI(SNES snes,void *lsctx,Vec x,Vec f,Vec g,Vec y,Vec w,PetscReal fnorm,PetscReal xnorm,PetscReal *ynorm,PetscReal *gnorm,PetscBool *flag)
19072b4ed282SShri Abhyankar {
19082b4ed282SShri Abhyankar   /*
19092b4ed282SShri Abhyankar      Note that for line search purposes we work with with the related
19102b4ed282SShri Abhyankar      minimization problem:
19112b4ed282SShri Abhyankar         min  z(x):  R^n -> R,
19122b4ed282SShri Abhyankar      where z(x) = .5 * fnorm*fnorm,and fnorm = || f ||_2.
19132b4ed282SShri Abhyankar      z(x) is the same as the merit function for the complementarity problem
19142b4ed282SShri Abhyankar    */
19152b4ed282SShri Abhyankar   PetscReal      initslope,minlambda,lambda,lambdatemp,rellength;
19162b4ed282SShri Abhyankar #if defined(PETSC_USE_COMPLEX)
19172b4ed282SShri Abhyankar   PetscScalar    cinitslope;
19182b4ed282SShri Abhyankar #endif
19192b4ed282SShri Abhyankar   PetscErrorCode ierr;
19202b4ed282SShri Abhyankar   PetscInt       count;
19212b4ed282SShri Abhyankar   SNES_VI        *vi = (SNES_VI*)snes->data;
1922ace3abfcSBarry Smith   PetscBool     changed_w = PETSC_FALSE,changed_y = PETSC_FALSE;
19232b4ed282SShri Abhyankar 
19242b4ed282SShri Abhyankar   PetscFunctionBegin;
19252b4ed282SShri Abhyankar   ierr    = PetscLogEventBegin(SNES_LineSearch,snes,x,f,g);CHKERRQ(ierr);
19262b4ed282SShri Abhyankar   *flag   = PETSC_TRUE;
19272b4ed282SShri Abhyankar 
19282b4ed282SShri Abhyankar   ierr = VecNorm(y,NORM_2,ynorm);CHKERRQ(ierr);
19292b4ed282SShri Abhyankar   if (*ynorm == 0.0) {
1930c92abb8aSShri Abhyankar     if (vi->lsmonitor) {
1931c92abb8aSShri Abhyankar       ierr = PetscViewerASCIIMonitorPrintf(vi->lsmonitor,"Line search: Direction and size is 0\n");CHKERRQ(ierr);
19322b4ed282SShri Abhyankar     }
19332b4ed282SShri Abhyankar     *gnorm = fnorm;
19342b4ed282SShri Abhyankar     ierr   = VecCopy(x,w);CHKERRQ(ierr);
19352b4ed282SShri Abhyankar     ierr   = VecCopy(f,g);CHKERRQ(ierr);
19362b4ed282SShri Abhyankar     *flag  = PETSC_FALSE;
19372b4ed282SShri Abhyankar     goto theend2;
19382b4ed282SShri Abhyankar   }
19392b4ed282SShri Abhyankar   if (*ynorm > vi->maxstep) {	/* Step too big, so scale back */
19402b4ed282SShri Abhyankar     ierr   = VecScale(y,vi->maxstep/(*ynorm));CHKERRQ(ierr);
19412b4ed282SShri Abhyankar     *ynorm = vi->maxstep;
19422b4ed282SShri Abhyankar   }
19432b4ed282SShri Abhyankar   ierr      = VecMaxPointwiseDivide(y,x,&rellength);CHKERRQ(ierr);
19442b4ed282SShri Abhyankar   minlambda = vi->minlambda/rellength;
19452b4ed282SShri Abhyankar #if defined(PETSC_USE_COMPLEX)
194689d9e806SShri Abhyankar   ierr      = VecDot(vi->dpsi,y,&cinitslope);CHKERRQ(ierr);
19472b4ed282SShri Abhyankar   initslope = PetscRealPart(cinitslope);
19482b4ed282SShri Abhyankar #else
194989d9e806SShri Abhyankar   ierr = VecDot(vi->dpsi,y,&initslope);CHKERRQ(ierr);
19502b4ed282SShri Abhyankar #endif
19512b4ed282SShri Abhyankar   if (initslope > 0.0)  initslope = -initslope;
19522b4ed282SShri Abhyankar   if (initslope == 0.0) initslope = -1.0;
19532b4ed282SShri Abhyankar   ierr = PetscInfo1(snes,"Initslope %G \n",initslope);CHKERRQ(ierr);
19542b4ed282SShri Abhyankar 
19552b4ed282SShri Abhyankar   ierr = VecWAXPY(w,-1.0,y,x);CHKERRQ(ierr);
19562b4ed282SShri Abhyankar   if (snes->nfuncs >= snes->max_funcs) {
19572b4ed282SShri Abhyankar     ierr  = PetscInfo(snes,"Exceeded maximum function evaluations, while checking full step length!\n");CHKERRQ(ierr);
19582b4ed282SShri Abhyankar     *flag = PETSC_FALSE;
19592b4ed282SShri Abhyankar     snes->reason = SNES_DIVERGED_FUNCTION_COUNT;
19602b4ed282SShri Abhyankar     goto theend2;
19612b4ed282SShri Abhyankar   }
19622b4ed282SShri Abhyankar   ierr = SNESComputeFunction(snes,w,g);CHKERRQ(ierr);
19632b4ed282SShri Abhyankar   if (snes->domainerror) {
19642b4ed282SShri Abhyankar     ierr = PetscLogEventEnd(SNES_LineSearch,snes,x,f,g);CHKERRQ(ierr);
19652b4ed282SShri Abhyankar     PetscFunctionReturn(0);
19662b4ed282SShri Abhyankar   }
19672b4ed282SShri Abhyankar   ierr = VecNorm(g,NORM_2,gnorm);CHKERRQ(ierr);
19682b4ed282SShri Abhyankar   if PetscIsInfOrNanReal(*gnorm) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_FP,"User provided compute function generated a Not-a-Number");
19692b4ed282SShri Abhyankar   if (.5*(*gnorm)*(*gnorm) <= .5*fnorm*fnorm + vi->alpha*initslope) { /* Sufficient reduction */
1970c92abb8aSShri Abhyankar     if (vi->lsmonitor) {
1971c92abb8aSShri Abhyankar       ierr = PetscViewerASCIIMonitorPrintf(vi->lsmonitor,"    Line search: Using full step: fnorm %G gnorm %G\n",fnorm,*gnorm);CHKERRQ(ierr);
19722b4ed282SShri Abhyankar     }
19732b4ed282SShri Abhyankar     goto theend2;
19742b4ed282SShri Abhyankar   }
19752b4ed282SShri Abhyankar 
19762b4ed282SShri Abhyankar   /* Fit points with quadratic */
19772b4ed282SShri Abhyankar   lambda = 1.0;
19782b4ed282SShri Abhyankar   count = 1;
19792b4ed282SShri Abhyankar   while (PETSC_TRUE) {
19802b4ed282SShri Abhyankar     if (lambda <= minlambda) { /* bad luck; use full step */
1981c92abb8aSShri Abhyankar       if (vi->lsmonitor) {
1982c92abb8aSShri Abhyankar         ierr = PetscViewerASCIIMonitorPrintf(vi->lsmonitor,"Line search: Unable to find good step length! %D \n",count);CHKERRQ(ierr);
1983c92abb8aSShri Abhyankar         ierr = PetscViewerASCIIMonitorPrintf(vi->lsmonitor,"Line search: fnorm=%G, gnorm=%G, ynorm=%G, lambda=%G, initial slope=%G\n",fnorm,*gnorm,*ynorm,lambda,initslope);CHKERRQ(ierr);
19842b4ed282SShri Abhyankar       }
19852b4ed282SShri Abhyankar       ierr = VecCopy(x,w);CHKERRQ(ierr);
19862b4ed282SShri Abhyankar       *flag = PETSC_FALSE;
19872b4ed282SShri Abhyankar       break;
19882b4ed282SShri Abhyankar     }
19892b4ed282SShri Abhyankar     lambdatemp = -initslope/((*gnorm)*(*gnorm) - fnorm*fnorm - 2.0*initslope);
19902b4ed282SShri Abhyankar     if (lambdatemp > .5*lambda)  lambdatemp = .5*lambda;
19912b4ed282SShri Abhyankar     if (lambdatemp <= .1*lambda) lambda     = .1*lambda;
19922b4ed282SShri Abhyankar     else                         lambda     = lambdatemp;
19932b4ed282SShri Abhyankar 
19942b4ed282SShri Abhyankar     ierr = VecWAXPY(w,-lambda,y,x);CHKERRQ(ierr);
19952b4ed282SShri Abhyankar     if (snes->nfuncs >= snes->max_funcs) {
19962b4ed282SShri Abhyankar       ierr  = PetscInfo1(snes,"Exceeded maximum function evaluations, while looking for good step length! %D \n",count);CHKERRQ(ierr);
19972b4ed282SShri Abhyankar       ierr  = PetscInfo5(snes,"fnorm=%18.16e, gnorm=%18.16e, ynorm=%18.16e, lambda=%18.16e, initial slope=%18.16e\n",fnorm,*gnorm,*ynorm,lambda,initslope);CHKERRQ(ierr);
19982b4ed282SShri Abhyankar       *flag = PETSC_FALSE;
19992b4ed282SShri Abhyankar       snes->reason = SNES_DIVERGED_FUNCTION_COUNT;
20002b4ed282SShri Abhyankar       break;
20012b4ed282SShri Abhyankar     }
20022b4ed282SShri Abhyankar     ierr = SNESComputeFunction(snes,w,g);CHKERRQ(ierr);
20032b4ed282SShri Abhyankar     if (snes->domainerror) {
20042b4ed282SShri Abhyankar       ierr = PetscLogEventEnd(SNES_LineSearch,snes,x,f,g);CHKERRQ(ierr);
20052b4ed282SShri Abhyankar       PetscFunctionReturn(0);
20062b4ed282SShri Abhyankar     }
20072b4ed282SShri Abhyankar     ierr = VecNorm(g,NORM_2,gnorm);CHKERRQ(ierr);
20082b4ed282SShri Abhyankar     if PetscIsInfOrNanReal(*gnorm) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_FP,"User provided compute function generated a Not-a-Number");
20092b4ed282SShri Abhyankar     if (.5*(*gnorm)*(*gnorm) < .5*fnorm*fnorm + lambda*vi->alpha*initslope) { /* sufficient reduction */
2010c92abb8aSShri Abhyankar       if (vi->lsmonitor) {
2011c92abb8aSShri Abhyankar         ierr = PetscViewerASCIIMonitorPrintf(vi->lsmonitor,"    Line Search: Quadratically determined step, lambda=%G\n",lambda);CHKERRQ(ierr);
20122b4ed282SShri Abhyankar       }
20132b4ed282SShri Abhyankar       break;
20142b4ed282SShri Abhyankar     }
20152b4ed282SShri Abhyankar     count++;
20162b4ed282SShri Abhyankar   }
20172b4ed282SShri Abhyankar   theend2:
20182b4ed282SShri Abhyankar   /* Optional user-defined check for line search step validity */
20192b4ed282SShri Abhyankar   if (vi->postcheckstep) {
20202b4ed282SShri Abhyankar     ierr = (*vi->postcheckstep)(snes,x,y,w,vi->postcheck,&changed_y,&changed_w);CHKERRQ(ierr);
20212b4ed282SShri Abhyankar     if (changed_y) {
20222b4ed282SShri Abhyankar       ierr = VecWAXPY(w,-1.0,y,x);CHKERRQ(ierr);
20232b4ed282SShri Abhyankar     }
20242b4ed282SShri Abhyankar     if (changed_y || changed_w) { /* recompute the function if the step has changed */
20252b4ed282SShri Abhyankar       ierr = SNESComputeFunction(snes,w,g);
20262b4ed282SShri Abhyankar       if (snes->domainerror) {
20272b4ed282SShri Abhyankar         ierr = PetscLogEventEnd(SNES_LineSearch,snes,x,f,g);CHKERRQ(ierr);
20282b4ed282SShri Abhyankar         PetscFunctionReturn(0);
20292b4ed282SShri Abhyankar       }
20302b4ed282SShri Abhyankar       ierr = VecNormBegin(g,NORM_2,gnorm);CHKERRQ(ierr);
20312b4ed282SShri Abhyankar       ierr = VecNormBegin(y,NORM_2,ynorm);CHKERRQ(ierr);
20322b4ed282SShri Abhyankar       ierr = VecNormEnd(g,NORM_2,gnorm);CHKERRQ(ierr);
20332b4ed282SShri Abhyankar       ierr = VecNormEnd(y,NORM_2,ynorm);CHKERRQ(ierr);
20342b4ed282SShri Abhyankar       if PetscIsInfOrNanReal(*gnorm) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_FP,"User provided compute function generated a Not-a-Number");
20352b4ed282SShri Abhyankar     }
20362b4ed282SShri Abhyankar   }
20372b4ed282SShri Abhyankar   ierr = PetscLogEventEnd(SNES_LineSearch,snes,x,f,g);CHKERRQ(ierr);
20382b4ed282SShri Abhyankar   PetscFunctionReturn(0);
20392b4ed282SShri Abhyankar }
20402b4ed282SShri Abhyankar 
2041ace3abfcSBarry Smith typedef PetscErrorCode (*FCN2)(SNES,void*,Vec,Vec,Vec,Vec,Vec,PetscReal,PetscReal,PetscReal*,PetscReal*,PetscBool*); /* force argument to next function to not be extern C*/
20422b4ed282SShri Abhyankar /* -------------------------------------------------------------------------- */
20432b4ed282SShri Abhyankar EXTERN_C_BEGIN
20442b4ed282SShri Abhyankar #undef __FUNCT__
20452b4ed282SShri Abhyankar #define __FUNCT__ "SNESLineSearchSet_VI"
20467087cfbeSBarry Smith PetscErrorCode  SNESLineSearchSet_VI(SNES snes,FCN2 func,void *lsctx)
20472b4ed282SShri Abhyankar {
20482b4ed282SShri Abhyankar   PetscFunctionBegin;
20492b4ed282SShri Abhyankar   ((SNES_VI *)(snes->data))->LineSearch = func;
20502b4ed282SShri Abhyankar   ((SNES_VI *)(snes->data))->lsP        = lsctx;
20512b4ed282SShri Abhyankar   PetscFunctionReturn(0);
20522b4ed282SShri Abhyankar }
20532b4ed282SShri Abhyankar EXTERN_C_END
20542b4ed282SShri Abhyankar 
20552b4ed282SShri Abhyankar /* -------------------------------------------------------------------------- */
20562b4ed282SShri Abhyankar EXTERN_C_BEGIN
20572b4ed282SShri Abhyankar #undef __FUNCT__
20582b4ed282SShri Abhyankar #define __FUNCT__ "SNESLineSearchSetMonitor_VI"
20597087cfbeSBarry Smith PetscErrorCode  SNESLineSearchSetMonitor_VI(SNES snes,PetscBool flg)
20602b4ed282SShri Abhyankar {
20612b4ed282SShri Abhyankar   SNES_VI        *vi = (SNES_VI*)snes->data;
20622b4ed282SShri Abhyankar   PetscErrorCode ierr;
20632b4ed282SShri Abhyankar 
20642b4ed282SShri Abhyankar   PetscFunctionBegin;
2065c92abb8aSShri Abhyankar   if (flg && !vi->lsmonitor) {
2066c92abb8aSShri Abhyankar     ierr = PetscViewerASCIIMonitorCreate(((PetscObject)snes)->comm,"stdout",((PetscObject)snes)->tablevel,&vi->lsmonitor);CHKERRQ(ierr);
2067c92abb8aSShri Abhyankar   } else if (!flg && vi->lsmonitor) {
2068c92abb8aSShri Abhyankar     ierr = PetscViewerASCIIMonitorDestroy(vi->lsmonitor);CHKERRQ(ierr);
20692b4ed282SShri Abhyankar   }
20702b4ed282SShri Abhyankar   PetscFunctionReturn(0);
20712b4ed282SShri Abhyankar }
20722b4ed282SShri Abhyankar EXTERN_C_END
20732b4ed282SShri Abhyankar 
20742b4ed282SShri Abhyankar /*
20752b4ed282SShri Abhyankar    SNESView_VI - Prints info from the SNESVI data structure.
20762b4ed282SShri Abhyankar 
20772b4ed282SShri Abhyankar    Input Parameters:
20782b4ed282SShri Abhyankar .  SNES - the SNES context
20792b4ed282SShri Abhyankar .  viewer - visualization context
20802b4ed282SShri Abhyankar 
20812b4ed282SShri Abhyankar    Application Interface Routine: SNESView()
20822b4ed282SShri Abhyankar */
20832b4ed282SShri Abhyankar #undef __FUNCT__
20842b4ed282SShri Abhyankar #define __FUNCT__ "SNESView_VI"
20852b4ed282SShri Abhyankar static PetscErrorCode SNESView_VI(SNES snes,PetscViewer viewer)
20862b4ed282SShri Abhyankar {
20872b4ed282SShri Abhyankar   SNES_VI        *vi = (SNES_VI *)snes->data;
208878c4b9d3SShri Abhyankar   const char     *cstr,*tstr;
20892b4ed282SShri Abhyankar   PetscErrorCode ierr;
2090ace3abfcSBarry Smith   PetscBool     iascii;
20912b4ed282SShri Abhyankar 
20922b4ed282SShri Abhyankar   PetscFunctionBegin;
20932b4ed282SShri Abhyankar   ierr = PetscTypeCompare((PetscObject)viewer,PETSCVIEWERASCII,&iascii);CHKERRQ(ierr);
20942b4ed282SShri Abhyankar   if (iascii) {
20952b4ed282SShri Abhyankar     if (vi->LineSearch == SNESLineSearchNo_VI)             cstr = "SNESLineSearchNo";
20962b4ed282SShri Abhyankar     else if (vi->LineSearch == SNESLineSearchQuadratic_VI) cstr = "SNESLineSearchQuadratic";
20972b4ed282SShri Abhyankar     else if (vi->LineSearch == SNESLineSearchCubic_VI)     cstr = "SNESLineSearchCubic";
2098*fe835674SShri Abhyankar     else if (vi->LineSearch == SNESLineSearchBoundProjectedCubic_VI) cstr = "SNESLineSearchBoundProjectedCubic";
20992b4ed282SShri Abhyankar     else                                                cstr = "unknown";
210078c4b9d3SShri Abhyankar     if (snes->ops->solve == SNESSolveVI_SS)      tstr = "Semismooth";
210178c4b9d3SShri Abhyankar     else if (snes->ops->solve == SNESSolveVI_AS)  tstr = "Active Set";
21020a7c7af0SShri Abhyankar     else if (snes->ops->solve == SNESSolveVI_RS) tstr = "Reduced Space";
210378c4b9d3SShri Abhyankar     else                                         tstr = "unknown";
210478c4b9d3SShri Abhyankar     ierr = PetscViewerASCIIPrintf(viewer,"  VI algorithm: %s\n",tstr);CHKERRQ(ierr);
21052b4ed282SShri Abhyankar     ierr = PetscViewerASCIIPrintf(viewer,"  line search variant: %s\n",cstr);CHKERRQ(ierr);
21062b4ed282SShri Abhyankar     ierr = PetscViewerASCIIPrintf(viewer,"  alpha=%G, maxstep=%G, minlambda=%G\n",vi->alpha,vi->maxstep,vi->minlambda);CHKERRQ(ierr);
21072b4ed282SShri Abhyankar   } else {
21082b4ed282SShri Abhyankar     SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"Viewer type %s not supported for SNES EQ VI",((PetscObject)viewer)->type_name);
21092b4ed282SShri Abhyankar   }
21102b4ed282SShri Abhyankar   PetscFunctionReturn(0);
21112b4ed282SShri Abhyankar }
21122b4ed282SShri Abhyankar 
21132b4ed282SShri Abhyankar /*
21142b4ed282SShri Abhyankar    SNESVISetVariableBounds - Sets the lower and upper bounds for the solution vector. xl <= x <= xu.
21152b4ed282SShri Abhyankar 
21162b4ed282SShri Abhyankar    Input Parameters:
21172b4ed282SShri Abhyankar .  snes - the SNES context.
21182b4ed282SShri Abhyankar .  xl   - lower bound.
21192b4ed282SShri Abhyankar .  xu   - upper bound.
21202b4ed282SShri Abhyankar 
21212b4ed282SShri Abhyankar    Notes:
21222b4ed282SShri Abhyankar    If this routine is not called then the lower and upper bounds are set to
212384c105d7SBarry Smith    PETSC_VI_INF and PETSC_VI_NINF respectively during SNESSetUp().
212484c105d7SBarry Smith 
21252b4ed282SShri Abhyankar */
21262b4ed282SShri Abhyankar 
21272b4ed282SShri Abhyankar #undef __FUNCT__
21282b4ed282SShri Abhyankar #define __FUNCT__ "SNESVISetVariableBounds"
212969c03620SShri Abhyankar PetscErrorCode SNESVISetVariableBounds(SNES snes, Vec xl, Vec xu)
21302b4ed282SShri Abhyankar {
21312b4ed282SShri Abhyankar   SNES_VI        *vi = (SNES_VI*)snes->data;
21322b4ed282SShri Abhyankar 
21332b4ed282SShri Abhyankar   PetscFunctionBegin;
21342b4ed282SShri Abhyankar   PetscValidHeaderSpecific(snes,SNES_CLASSID,1);
21352b4ed282SShri Abhyankar   PetscValidHeaderSpecific(xl,VEC_CLASSID,2);
21362b4ed282SShri Abhyankar   PetscValidHeaderSpecific(xu,VEC_CLASSID,3);
21372b4ed282SShri Abhyankar 
21382b4ed282SShri Abhyankar   /* Check if SNESSetFunction is called */
21392b4ed282SShri Abhyankar   if(!snes->vec_func) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"Must call SNESSetFunction() first");
21402b4ed282SShri Abhyankar 
21412b4ed282SShri Abhyankar   /* Check if the vector sizes are compatible for lower and upper bounds */
21422b4ed282SShri Abhyankar   if (xl->map->N != snes->vec_func->map->N) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_INCOMP,"Incompatible vector lengths lower bound = %D solution vector = %D",xl->map->N,snes->vec_func->map->N);
21432b4ed282SShri Abhyankar   if (xu->map->N != snes->vec_func->map->N) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_INCOMP,"Incompatible vector lengths: upper bound = %D solution vector = %D",xu->map->N,snes->vec_func->map->N);
21442b4ed282SShri Abhyankar   vi->usersetxbounds = PETSC_TRUE;
21452b4ed282SShri Abhyankar   vi->xl = xl;
21462b4ed282SShri Abhyankar   vi->xu = xu;
21472b4ed282SShri Abhyankar 
21482b4ed282SShri Abhyankar   PetscFunctionReturn(0);
21492b4ed282SShri Abhyankar }
21502b4ed282SShri Abhyankar /* -------------------------------------------------------------------------- */
21512b4ed282SShri Abhyankar /*
21522b4ed282SShri Abhyankar    SNESSetFromOptions_VI - Sets various parameters for the SNESVI method.
21532b4ed282SShri Abhyankar 
21542b4ed282SShri Abhyankar    Input Parameter:
21552b4ed282SShri Abhyankar .  snes - the SNES context
21562b4ed282SShri Abhyankar 
21572b4ed282SShri Abhyankar    Application Interface Routine: SNESSetFromOptions()
21582b4ed282SShri Abhyankar */
21592b4ed282SShri Abhyankar #undef __FUNCT__
21602b4ed282SShri Abhyankar #define __FUNCT__ "SNESSetFromOptions_VI"
21612b4ed282SShri Abhyankar static PetscErrorCode SNESSetFromOptions_VI(SNES snes)
21622b4ed282SShri Abhyankar {
21632b4ed282SShri Abhyankar   SNES_VI        *vi = (SNES_VI *)snes->data;
21642b4ed282SShri Abhyankar   const char     *lses[] = {"basic","basicnonorms","quadratic","cubic"};
2165d950fb63SShri Abhyankar   const char     *vies[] = {"ss","as","rs"};
21662b4ed282SShri Abhyankar   PetscErrorCode ierr;
21672b4ed282SShri Abhyankar   PetscInt       indx;
216869c03620SShri Abhyankar   PetscBool      flg,set,flg2;
21692b4ed282SShri Abhyankar 
21702b4ed282SShri Abhyankar   PetscFunctionBegin;
21712b4ed282SShri Abhyankar   ierr = PetscOptionsHead("SNES semismooth method options");CHKERRQ(ierr);
21729308c137SBarry Smith   ierr = PetscOptionsBool("-snes_vi_monitor","Monitor all non-active variables","None",PETSC_FALSE,&flg,0);CHKERRQ(ierr);
21739308c137SBarry Smith   if (flg) {
21749308c137SBarry Smith     ierr = SNESMonitorSet(snes,SNESMonitorVI,0,0);CHKERRQ(ierr);
21759308c137SBarry Smith   }
21762b4ed282SShri Abhyankar   ierr = PetscOptionsReal("-snes_vi_alpha","Function norm must decrease by","None",vi->alpha,&vi->alpha,0);CHKERRQ(ierr);
21772b4ed282SShri Abhyankar   ierr = PetscOptionsReal("-snes_vi_maxstep","Step must be less than","None",vi->maxstep,&vi->maxstep,0);CHKERRQ(ierr);
21782b4ed282SShri Abhyankar   ierr = PetscOptionsReal("-snes_vi_minlambda","Minimum lambda allowed","None",vi->minlambda,&vi->minlambda,0);CHKERRQ(ierr);
21792b4ed282SShri Abhyankar   ierr = PetscOptionsReal("-snes_vi_const_tol","constraint tolerance","None",vi->const_tol,&vi->const_tol,0);CHKERRQ(ierr);
2180acfcf0e5SJed Brown   ierr = PetscOptionsBool("-snes_vi_lsmonitor","Print progress of line searches","SNESLineSearchSetMonitor",vi->lsmonitor ? PETSC_TRUE : PETSC_FALSE,&flg,&set);CHKERRQ(ierr);
21812b4ed282SShri Abhyankar   if (set) {ierr = SNESLineSearchSetMonitor(snes,flg);CHKERRQ(ierr);}
2182d950fb63SShri Abhyankar   ierr = PetscOptionsEList("-snes_vi_type","Semismooth algorithm used","",vies,3,"ss",&indx,&flg2);CHKERRQ(ierr);
218369c03620SShri Abhyankar   if (flg2) {
218469c03620SShri Abhyankar     switch (indx) {
218569c03620SShri Abhyankar     case 0:
218669c03620SShri Abhyankar       snes->ops->solve = SNESSolveVI_SS;
218769c03620SShri Abhyankar       break;
218869c03620SShri Abhyankar     case 1:
218969c03620SShri Abhyankar       snes->ops->solve = SNESSolveVI_AS;
219069c03620SShri Abhyankar       break;
2191d950fb63SShri Abhyankar     case 2:
2192d950fb63SShri Abhyankar       snes->ops->solve = SNESSolveVI_RS;
2193d950fb63SShri Abhyankar       break;
219469c03620SShri Abhyankar     }
219569c03620SShri Abhyankar   }
2196*fe835674SShri Abhyankar   if (snes->ops->solve == SNESSolveVI_RS) {
2197*fe835674SShri Abhyankar     ierr = SNESLineSearchSet(snes,SNESLineSearchBoundProjectedCubic_VI,PETSC_NULL);CHKERRQ(ierr);
2198*fe835674SShri Abhyankar   }
2199*fe835674SShri Abhyankar   else {
2200c92abb8aSShri Abhyankar     ierr = PetscOptionsEList("-snes_vi_ls","Line search used","SNESLineSearchSet",lses,4,"cubic",&indx,&flg);CHKERRQ(ierr);
22012b4ed282SShri Abhyankar     if (flg) {
22022b4ed282SShri Abhyankar       switch (indx) {
22032b4ed282SShri Abhyankar       case 0:
22042b4ed282SShri Abhyankar         ierr = SNESLineSearchSet(snes,SNESLineSearchNo_VI,PETSC_NULL);CHKERRQ(ierr);
22052b4ed282SShri Abhyankar         break;
22062b4ed282SShri Abhyankar       case 1:
22072b4ed282SShri Abhyankar         ierr = SNESLineSearchSet(snes,SNESLineSearchNoNorms_VI,PETSC_NULL);CHKERRQ(ierr);
22082b4ed282SShri Abhyankar         break;
22092b4ed282SShri Abhyankar       case 2:
22102b4ed282SShri Abhyankar         ierr = SNESLineSearchSet(snes,SNESLineSearchQuadratic_VI,PETSC_NULL);CHKERRQ(ierr);
22112b4ed282SShri Abhyankar         break;
22122b4ed282SShri Abhyankar       case 3:
22132b4ed282SShri Abhyankar         ierr = SNESLineSearchSet(snes,SNESLineSearchCubic_VI,PETSC_NULL);CHKERRQ(ierr);
22142b4ed282SShri Abhyankar         break;
22152b4ed282SShri Abhyankar       }
22162b4ed282SShri Abhyankar     }
2217*fe835674SShri Abhyankar   }
22182b4ed282SShri Abhyankar   ierr = PetscOptionsTail();CHKERRQ(ierr);
22192b4ed282SShri Abhyankar   PetscFunctionReturn(0);
22202b4ed282SShri Abhyankar }
22212b4ed282SShri Abhyankar /* -------------------------------------------------------------------------- */
22222b4ed282SShri Abhyankar /*MC
22232b4ed282SShri Abhyankar       SNESVI - Semismooth newton method based nonlinear solver that uses a line search
22242b4ed282SShri Abhyankar 
22252b4ed282SShri Abhyankar    Options Database:
22262b4ed282SShri Abhyankar +   -snes_vi [cubic,quadratic,basic,basicnonorms] - Selects line search
22272b4ed282SShri Abhyankar .   -snes_vi_alpha <alpha> - Sets alpha
22282b4ed282SShri Abhyankar .   -snes_vi_maxstep <maxstep> - Sets the maximum stepsize the line search will use (if the 2-norm(y) > maxstep then scale y to be y = (maxstep/2-norm(y)) *y)
22292b4ed282SShri Abhyankar .   -snes_vi_minlambda <minlambda>  - Sets the minimum lambda the line search will use  minlambda / max_i ( y[i]/x[i] )
22302b4ed282SShri Abhyankar -   -snes_vi_monitor - print information about progress of line searches
22312b4ed282SShri Abhyankar 
22322b4ed282SShri Abhyankar 
22332b4ed282SShri Abhyankar    Level: beginner
22342b4ed282SShri Abhyankar 
22352b4ed282SShri Abhyankar .seealso:  SNESCreate(), SNES, SNESSetType(), SNESTR, SNESLineSearchSet(),
22362b4ed282SShri Abhyankar            SNESLineSearchSetPostCheck(), SNESLineSearchNo(), SNESLineSearchCubic(), SNESLineSearchQuadratic(),
22372b4ed282SShri Abhyankar            SNESLineSearchSet(), SNESLineSearchNoNorms(), SNESLineSearchSetPreCheck(), SNESLineSearchSetParams(), SNESLineSearchGetParams()
22382b4ed282SShri Abhyankar 
22392b4ed282SShri Abhyankar M*/
22402b4ed282SShri Abhyankar EXTERN_C_BEGIN
22412b4ed282SShri Abhyankar #undef __FUNCT__
22422b4ed282SShri Abhyankar #define __FUNCT__ "SNESCreate_VI"
22437087cfbeSBarry Smith PetscErrorCode  SNESCreate_VI(SNES snes)
22442b4ed282SShri Abhyankar {
22452b4ed282SShri Abhyankar   PetscErrorCode ierr;
22462b4ed282SShri Abhyankar   SNES_VI      *vi;
22472b4ed282SShri Abhyankar 
22482b4ed282SShri Abhyankar   PetscFunctionBegin;
22492b4ed282SShri Abhyankar   snes->ops->setup	     = SNESSetUp_VI;
225069c03620SShri Abhyankar   snes->ops->solve	     = SNESSolveVI_SS;
22512b4ed282SShri Abhyankar   snes->ops->destroy	     = SNESDestroy_VI;
22522b4ed282SShri Abhyankar   snes->ops->setfromoptions  = SNESSetFromOptions_VI;
22532b4ed282SShri Abhyankar   snes->ops->view            = SNESView_VI;
22542b4ed282SShri Abhyankar   snes->ops->converged       = SNESDefaultConverged_VI;
22552b4ed282SShri Abhyankar 
22562b4ed282SShri Abhyankar   ierr                   = PetscNewLog(snes,SNES_VI,&vi);CHKERRQ(ierr);
22572b4ed282SShri Abhyankar   snes->data    	 = (void*)vi;
22582b4ed282SShri Abhyankar   vi->alpha		 = 1.e-4;
22592b4ed282SShri Abhyankar   vi->maxstep		 = 1.e8;
22602b4ed282SShri Abhyankar   vi->minlambda         = 1.e-12;
22612b4ed282SShri Abhyankar   vi->LineSearch        = SNESLineSearchCubic_VI;
22622b4ed282SShri Abhyankar   vi->lsP               = PETSC_NULL;
22632b4ed282SShri Abhyankar   vi->postcheckstep     = PETSC_NULL;
22642b4ed282SShri Abhyankar   vi->postcheck         = PETSC_NULL;
22652b4ed282SShri Abhyankar   vi->precheckstep      = PETSC_NULL;
22662b4ed282SShri Abhyankar   vi->precheck          = PETSC_NULL;
22672b4ed282SShri Abhyankar   vi->const_tol         =  2.2204460492503131e-16;
22682b4ed282SShri Abhyankar   vi->computessfunction = ComputeFischerFunction;
22692b4ed282SShri Abhyankar 
22702b4ed282SShri Abhyankar   ierr = PetscObjectComposeFunctionDynamic((PetscObject)snes,"SNESLineSearchSetMonitor_C","SNESLineSearchSetMonitor_VI",SNESLineSearchSetMonitor_VI);CHKERRQ(ierr);
22712b4ed282SShri Abhyankar   ierr = PetscObjectComposeFunctionDynamic((PetscObject)snes,"SNESLineSearchSet_C","SNESLineSearchSet_VI",SNESLineSearchSet_VI);CHKERRQ(ierr);
22722b4ed282SShri Abhyankar 
22732b4ed282SShri Abhyankar   PetscFunctionReturn(0);
22742b4ed282SShri Abhyankar }
22752b4ed282SShri Abhyankar EXTERN_C_END
2276