static int CVSpgmrSolve(CVodeMem cv_mem, N_Vector b, N_Vector weight,
N_Vector ynow, N_Vector fnow)
{
realtype bnorm, res_norm;
CVSpgmrMem cvspgmr_mem;
int nli_inc, nps_inc, ier;
cvspgmr_mem = (CVSpgmrMem) lmem;
/* Test norm(b); if small, return x = 0 or x = b */
deltar = delt*tq[4];
bnorm = N_VWrmsNorm(b, weight);
if (bnorm <= deltar) {
if (mnewt > 0) N_VConst(ZERO, b);
return(0);
}
/* Set vectors ycur and fcur for use by the Atimes and Psolve routines */
ycur = ynow;
fcur = fnow;
/* Set inputs delta and initial guess x = 0 to SpgmrSolve */
delta = deltar * sqrtN;
N_VConst(ZERO, x);
/* Call SpgmrSolve and copy x to b */
ier = SpgmrSolve(spgmr_mem, cv_mem, x, b, pretype, gstype, delta, 0,
cv_mem, weight, weight, CVSpgmrAtimes, CVSpgmrPSolve,
&res_norm, &nli_inc, &nps_inc);
N_VScale(ONE, x, b);
/* Increment counters nli, nps, and ncfl */
nli += nli_inc;
nps += nps_inc;
if (ier != 0) ncfl++;
/* Set return value to -1, 0, or 1 */
last_flag = ier;
if (ier < 0) return(-1);
if ((ier == SPGMR_SUCCESS) ||
((ier == SPGMR_RES_REDUCED) && (mnewt == 0)))
return(0);
return(1);
}
static int IDASpgmrSolve(IDAMem IDA_mem, N_Vector bb, N_Vector weight,
N_Vector yy_now, N_Vector yp_now, N_Vector rr_now)
{
IDASpilsMem idaspils_mem;
SpgmrMem spgmr_mem;
int pretype, nli_inc, nps_inc, retval;
realtype res_norm;
idaspils_mem = (IDASpilsMem) lmem;
spgmr_mem = (SpgmrMem) spils_mem;
/* Set SpgmrSolve convergence test constant epslin, in terms of the
Newton convergence test constant epsNewt and safety factors. The factor
sqrt(Neq) assures that the GMRES convergence test is applied to the
WRMS norm of the residual vector, rather than the weighted L2 norm. */
epslin = sqrtN*eplifac*epsNewt;
/* Set vectors ycur, ypcur, and rcur for use by the Atimes and Psolve */
ycur = yy_now;
ypcur = yp_now;
rcur = rr_now;
/* Set SpgmrSolve inputs pretype and initial guess xx = 0. */
pretype = (psolve == NULL) ? PREC_NONE : PREC_LEFT;
N_VConst(ZERO, xx);
/* Call SpgmrSolve and copy xx to bb. */
retval = SpgmrSolve(spgmr_mem, IDA_mem, xx, bb, pretype, gstype, epslin,
maxrs, IDA_mem, weight, weight, IDASpilsAtimes,
IDASpilsPSolve, &res_norm, &nli_inc, &nps_inc);
if (nli_inc == 0) N_VScale(ONE, SPGMR_VTEMP(spgmr_mem), bb);
else N_VScale(ONE, xx, bb);
/* Increment counters nli, nps, and return if successful. */
nli += nli_inc;
nps += nps_inc;
if (retval != SPGMR_SUCCESS) ncfl++;
/* Interpret return value from SpgmrSolve */
last_flag = retval;
switch(retval) {
case SPGMR_SUCCESS:
return(0);
break;
case SPGMR_RES_REDUCED:
return(1);
break;
case SPGMR_CONV_FAIL:
return(1);
break;
case SPGMR_QRFACT_FAIL:
return(1);
break;
case SPGMR_PSOLVE_FAIL_REC:
return(1);
break;
case SPGMR_ATIMES_FAIL_REC:
return(1);
break;
case SPGMR_MEM_NULL:
return(-1);
break;
case SPGMR_ATIMES_FAIL_UNREC:
IDAProcessError(IDA_mem, SPGMR_ATIMES_FAIL_UNREC, "IDASPGMR", "IDASpgmrSolve", MSGS_JTIMES_FAILED);
return(-1);
break;
case SPGMR_PSOLVE_FAIL_UNREC:
IDAProcessError(IDA_mem, SPGMR_PSOLVE_FAIL_UNREC, "IDASPGMR", "IDASpgmrSolve", MSGS_PSOLVE_FAILED);
return(-1);
break;
case SPGMR_GS_FAIL:
return(-1);
break;
case SPGMR_QRSOL_FAIL:
return(-1);
break;
}
return(0);
}
