void SundialsIda::printStats(clock_t dt)
{
long int nst, nje, nre, netf, ncfn, nge, npe, nps;
int retval;
retval = IDAGetNumSteps(sundialsMem, &nst);
check_flag(&retval, "IDAGetNumSteps", 1);
retval = IDAGetNumResEvals(sundialsMem, &nre);
check_flag(&retval, "IDAGetNumResEvals", 1);
retval = IDAGetNumErrTestFails(sundialsMem, &netf);
check_flag(&retval, "IDAGetNumErrTestFails", 1);
retval = IDAGetNumNonlinSolvConvFails(sundialsMem, &ncfn);
check_flag(&retval, "IDAGetNumNonlinSolvConvFails", 1);
if (findRoots) {
retval = IDAGetNumGEvals(sundialsMem, &nge);
check_flag(&retval, "IDAGetNumGEvals", 1);
}
retval = IDASpilsGetNumJtimesEvals(sundialsMem, &nje);
check_flag(&retval, "IDASpilsGetNumJtimesEvals", 1);
retval = IDASpilsGetNumPrecEvals(sundialsMem, &npe);
check_flag(&retval, "IDASpilsGetPrecEvals", 1);
retval = IDASpilsGetNumPrecSolves(sundialsMem, &nps);
check_flag(&retval, "IDASpilsGetNumPrecSolves", 1);
printf("\nIDA Solver Statistics: \n\n");
printf("Number of steps = %ld\n", nst);
printf("Number of residual evaluations = %ld\n", nre);
printf("Number of error test failures = %ld\n", netf);
printf("Number of nonlinear conv. failures = %ld\n", ncfn);
if (findRoots) {
printf("Number of root fn. evaluations = %ld\n", nge);
}
printf("Number of J-v Evaluations = %ld\n", nje);
printf("Number of preconditioner evals. = %ld\n", npe);
printf("Number of preconditioner solves = %ld\n", nps);
retval = IDAGetNumSteps(sundialsMem, &nst);
if (dt == 0) {
logFile.write(format("IDA solver took %i steps.\n") % nst);
} else {
logFile.write(format("IDA solver took %i steps in %f seconds.\n") %
nst % (((double) dt)/CLOCKS_PER_SEC));
}
}
static void PrintOutput(void *mem, realtype t, N_Vector uu)
{
realtype hused, umax;
long int nst, nni, nje, nre, nreLS, nli, npe, nps;
int kused, ier;
umax = N_VMaxNorm(uu);
ier = IDAGetLastOrder(mem, &kused);
check_flag(&ier, "IDAGetLastOrder", 1);
ier = IDAGetNumSteps(mem, &nst);
check_flag(&ier, "IDAGetNumSteps", 1);
ier = IDAGetNumNonlinSolvIters(mem, &nni);
check_flag(&ier, "IDAGetNumNonlinSolvIters", 1);
ier = IDAGetNumResEvals(mem, &nre);
check_flag(&ier, "IDAGetNumResEvals", 1);
ier = IDAGetLastStep(mem, &hused);
check_flag(&ier, "IDAGetLastStep", 1);
ier = IDASpilsGetNumJtimesEvals(mem, &nje);
check_flag(&ier, "IDASpilsGetNumJtimesEvals", 1);
ier = IDASpilsGetNumLinIters(mem, &nli);
check_flag(&ier, "IDASpilsGetNumLinIters", 1);
ier = IDASpilsGetNumResEvals(mem, &nreLS);
check_flag(&ier, "IDASpilsGetNumResEvals", 1);
ier = IDASpilsGetNumPrecEvals(mem, &npe);
check_flag(&ier, "IDASpilsGetPrecEvals", 1);
ier = IDASpilsGetNumPrecSolves(mem, &nps);
check_flag(&ier, "IDASpilsGetNumPrecSolves", 1);
#if defined(SUNDIALS_EXTENDED_PRECISION)
printf(" %5.2Lf %13.5Le %d %3ld %3ld %3ld %4ld %4ld %9.2Le %3ld %3ld\n",
t, umax, kused, nst, nni, nje, nre, nreLS, hused, npe, nps);
#elif defined(SUNDIALS_DOUBLE_PRECISION)
printf(" %5.2f %13.5e %d %3ld %3ld %3ld %4ld %4ld %9.2e %3ld %3ld\n",
t, umax, kused, nst, nni, nje, nre, nreLS, hused, npe, nps);
#else
printf(" %5.2f %13.5e %d %3ld %3ld %3ld %4ld %4ld %9.2e %3ld %3ld\n",
t, umax, kused, nst, nni, nje, nre, nreLS, hused, npe, nps);
#endif
}
void FIDA_SOLVE(realtype *tout, realtype *tret, realtype *yret,
realtype *ypret, int *itask, int *ier)
{
int klast, kcur;
*ier = 0;
/* Attach user data to vectors */
N_VSetArrayPointer(yret, F2C_IDA_vec);
N_VSetArrayPointer(ypret, F2C_IDA_ypvec);
*ier = IDASolve(IDA_idamem, *tout, tret, F2C_IDA_vec, F2C_IDA_ypvec, *itask);
/* Reset data pointers */
N_VSetArrayPointer(NULL, F2C_IDA_vec);
N_VSetArrayPointer(NULL, F2C_IDA_ypvec);
/* Set optional outputs */
IDAGetWorkSpace(IDA_idamem,
&IDA_iout[0], /* LENRW */
&IDA_iout[1]); /* LENIW */
IDAGetIntegratorStats(IDA_idamem,
&IDA_iout[2], /* NST */
&IDA_iout[3], /* NRE */
&IDA_iout[7], /* NSETUPS */
&IDA_iout[4], /* NETF */
&klast, /* KLAST */
&kcur, /* KCUR */
&IDA_rout[0], /* HINUSED */
&IDA_rout[1], /* HLAST */
&IDA_rout[2], /* HCUR */
&IDA_rout[3]); /* TCUR */
IDA_iout[8] = (long int) klast;
IDA_iout[9] = (long int) kcur;
IDAGetNonlinSolvStats(IDA_idamem,
&IDA_iout[6], /* NNI */
&IDA_iout[5]); /* NCFN */
IDAGetNumBacktrackOps(IDA_idamem,
&IDA_iout[10]); /* NBCKTRK */
IDAGetTolScaleFactor(IDA_idamem,
&IDA_rout[4]); /* TOLSFAC */
/* Root finding is on */
if (IDA_nrtfn != 0)
IDAGetNumGEvals(IDA_idamem, &IDA_iout[11]); /* NGE */
switch(IDA_ls) {
case IDA_LS_DENSE:
case IDA_LS_BAND:
case IDA_LS_LAPACKDENSE:
case IDA_LS_LAPACKBAND:
IDADlsGetWorkSpace(IDA_idamem, &IDA_iout[12], &IDA_iout[13]); /* LENRWLS, LENIWLS */
IDADlsGetLastFlag(IDA_idamem, &IDA_iout[14]); /* LSTF */
IDADlsGetNumResEvals(IDA_idamem, &IDA_iout[15]); /* NRE */
IDADlsGetNumJacEvals(IDA_idamem, &IDA_iout[16]); /* NJE */
break;
case IDA_LS_KLU:
case IDA_LS_SUPERLUMT:
IDASlsGetLastFlag(IDA_idamem, &IDA_iout[14]); /* LSTF */
IDASlsGetNumJacEvals(IDA_idamem, &IDA_iout[16]); /* NJE */
break;
case IDA_LS_SPGMR:
case IDA_LS_SPBCG:
case IDA_LS_SPTFQMR:
IDASpilsGetWorkSpace(IDA_idamem, &IDA_iout[12], &IDA_iout[13]); /* LENRWLS, LENIWLS */
IDASpilsGetLastFlag(IDA_idamem, &IDA_iout[14]); /* LSTF */
IDASpilsGetNumResEvals(IDA_idamem, &IDA_iout[15]); /* NRE */
IDASpilsGetNumJtimesEvals(IDA_idamem, &IDA_iout[16]); /* NJE */
IDASpilsGetNumPrecEvals(IDA_idamem, &IDA_iout[17]); /* NPE */
IDASpilsGetNumPrecSolves(IDA_idamem, &IDA_iout[18]); /* NPS */
IDASpilsGetNumLinIters(IDA_idamem, &IDA_iout[19]); /* NLI */
IDASpilsGetNumConvFails(IDA_idamem, &IDA_iout[20]); /* NCFL */
break;
}
return;
}
