static void PrintFinalStats(void *mem)
{
int retval;
long int nst, nni, nje, nre, nreLS, netf, ncfn, nge;
retval = IDAGetNumSteps(mem, &nst);
check_flag(&retval, "IDAGetNumSteps", 1);
retval = IDAGetNumResEvals(mem, &nre);
check_flag(&retval, "IDAGetNumResEvals", 1);
retval = IDADlsGetNumJacEvals(mem, &nje);
check_flag(&retval, "IDADlsGetNumJacEvals", 1);
retval = IDAGetNumNonlinSolvIters(mem, &nni);
check_flag(&retval, "IDAGetNumNonlinSolvIters", 1);
retval = IDAGetNumErrTestFails(mem, &netf);
check_flag(&retval, "IDAGetNumErrTestFails", 1);
retval = IDAGetNumNonlinSolvConvFails(mem, &ncfn);
check_flag(&retval, "IDAGetNumNonlinSolvConvFails", 1);
retval = IDADlsGetNumResEvals(mem, &nreLS);
check_flag(&retval, "IDADlsGetNumResEvals", 1);
retval = IDAGetNumGEvals(mem, &nge);
check_flag(&retval, "IDAGetNumGEvals", 1);
printf("\nFinal Run Statistics: \n\n");
printf("Number of steps = %ld\n", nst);
printf("Number of residual evaluations = %ld\n", nre+nreLS);
printf("Number of Jacobian evaluations = %ld\n", nje);
printf("Number of nonlinear iterations = %ld\n", nni);
printf("Number of error test failures = %ld\n", netf);
printf("Number of nonlinear conv. failures = %ld\n", ncfn);
printf("Number of root fn. evaluations = %ld\n", nge);
}
CAMLprim value sundials_ml_ida_dls_get_num_jac_evals(value ida_solver, value njevals) {
CAMLparam2(ida_solver, njevals);
long int _njevals;
const int ret = IDADlsGetNumJacEvals(IDA_MEM(ida_solver), &_njevals);
Store_field(njevals, 0, Val_int(_njevals));
CAMLreturn(Val_int(ret));
}
static void PrintFinalStats(void *mem)
{
long int nst, nre, nreLS, nni, nje, netf, ncfn;
int flag;
flag = IDAGetNumSteps(mem, &nst);
check_flag(&flag, "IDAGetNumSteps", 1);
flag = IDAGetNumNonlinSolvIters(mem, &nni);
check_flag(&flag, "IDAGetNumNonlinSolvIters", 1);
flag = IDAGetNumResEvals(mem, &nre);
check_flag(&flag, "IDAGetNumResEvals", 1);
flag = IDAGetNumErrTestFails(mem, &netf);
check_flag(&flag, "IDAGetNumErrTestFails", 1);
flag = IDAGetNumNonlinSolvConvFails(mem, &ncfn);
check_flag(&flag, "IDAGetNumNonlinSolvConvFails", 1);
flag = IDADlsGetNumJacEvals(mem, &nje);
check_flag(&flag, "IDADlsGetNumJacEvals", 1);
flag = IDADlsGetNumResEvals(mem, &nreLS);
check_flag(&flag, "IDADlsGetNumResEvals", 1);
printf("-----------------------------------------------------------\n");
printf("Final run statistics: \n\n");
printf("Number of steps = %ld\n", nst);
printf("Number of residual evaluations = %ld\n", nre+nreLS);
printf("Number of Jacobian evaluations = %ld\n", nje);
printf("Number of nonlinear iterations = %ld\n", nni);
printf("Number of error test failures = %ld\n", netf);
printf("Number of nonlinear conv. failures = %ld\n", ncfn);
}
static void PrintOutput(void *mem, realtype t, N_Vector uu)
{
int ier;
realtype umax, hused;
long int nst, nni, nje, nre, nreLS;
int kused;
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 = IDADlsGetNumJacEvals(mem, &nje);
check_flag(&ier, "IDADlsGetNumJacEvals", 1);
ier = IDADlsGetNumResEvals(mem, &nreLS);
check_flag(&ier, "IDADlsGetNumResEvals", 1);
#if defined(SUNDIALS_EXTENDED_PRECISION)
printf(" %5.2Lf %13.5Le %d %3ld %3ld %3ld %4ld %4ld %9.2Le \n",
t, umax, kused, nst, nni, nje, nre, nreLS, hused);
#elif defined(SUNDIALS_DOUBLE_PRECISION)
printf(" %5.2f %13.5e %d %3ld %3ld %3ld %4ld %4ld %9.2e \n",
t, umax, kused, nst, nni, nje, nre, nreLS, hused);
#else
printf(" %5.2f %13.5e %d %3ld %3ld %3ld %4ld %4ld %9.2e \n",
t, umax, kused, nst, nni, nje, nre, nreLS, hused);
#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;
}
