/**
* Tests that a given interpolation type reproduces the data points it is given,
* and then tests that it correctly reproduces additional values.
*
* @param xarr the x values of the points that define the function
* @param yarr the y values of the points that define the function
* @param zarr the values of the function at the points specified by xarr and yarr
* @param xsize the length of xarr
* @param ysize the length of yarr
* @param xval the x values of additional points at which to calculate interpolated values
* @param yval the y values of additional points at which to calculate interpolated values
* @param zval the expected results of the additional interpolations
* @param zxval the expected results of the x derivative calculations
* @param zyval the expected results of the y derivative calculations
* @param zxxval the expected results of the xx derivative calculations
* @param zyyval the expected results of the yy derivative calculations
* @param zxyval the expected results of the xy derivative calculations
* @param test_size the length of xval, yval, zval, etc.
* @param T the interpolation type
*/
int test_interp2d(const double xarr[], const double yarr[], const double zarr[], // interpolation data
size_t xsize, size_t ysize, // sizes of xarr and yarr
const double xval[], const double yval[], // test points
const double zval[], // expected results
const double zxval[], const double zyval[],
const double zxxval[], const double zyyval[], const double zxyval[],
size_t test_size, // number of test points
const interp2d_type* T) {
gsl_interp_accel *xa, *ya;
int status = 0;
size_t xi, yi, zi, i;
xa = gsl_interp_accel_alloc();
ya = gsl_interp_accel_alloc();
interp2d* interp = interp2d_alloc(T, xsize, ysize);
interp2d_spline* interp_s = interp2d_spline_alloc(T, xsize, ysize);
unsigned int min_size = interp2d_type_min_size(T);
gsl_test_int(min_size, T->min_size, "interp2d_type_min_size on %s", interp2d_name(interp));
interp2d_init(interp, xarr, yarr, zarr, xsize, ysize);
interp2d_spline_init(interp_s, xarr, yarr, zarr, xsize, ysize);
// First check that the interpolation reproduces the given points
for (xi = 0; xi < xsize; xi++) {
double x = xarr[xi];
for (yi = 0; yi < ysize; yi++) {
double y = yarr[yi];
zi = INDEX_2D(xi, yi, xsize, ysize);
test_single_low_level(&interp2d_eval, &interp2d_eval_e, interp, xarr, yarr, zarr, x, y, xa, ya, zarr, zi);
test_single_low_level(&interp2d_eval_no_boundary_check, &interp2d_eval_e_no_boundary_check, interp, xarr, yarr, zarr, x, y, xa, ya, zarr, zi);
test_single_high_level(&interp2d_spline_eval, &interp2d_spline_eval_e, interp_s, x, y, xa, ya, zarr, zi);
}
}
// Then check additional points provided
for (i = 0; i < test_size; i++) {
double x = xval[i];
double y = yval[i];
test_single_low_level(&interp2d_eval, &interp2d_eval_e, interp, xarr, yarr, zarr, x, y, xa, ya, zval, i);
test_single_low_level(&interp2d_eval_deriv_x, &interp2d_eval_deriv_x_e, interp, xarr, yarr, zarr, x, y, xa, ya, zxval, i);
test_single_low_level(&interp2d_eval_deriv_y, &interp2d_eval_deriv_y_e, interp, xarr, yarr, zarr, x, y, xa, ya, zyval, i);
test_single_low_level(&interp2d_eval_deriv_xx,&interp2d_eval_deriv_xx_e, interp, xarr, yarr, zarr, x, y, xa, ya, zxxval, i);
test_single_low_level(&interp2d_eval_deriv_yy,&interp2d_eval_deriv_yy_e, interp, xarr, yarr, zarr, x, y, xa, ya, zyyval, i);
test_single_low_level(&interp2d_eval_deriv_xy,&interp2d_eval_deriv_xy_e, interp, xarr, yarr, zarr, x, y, xa, ya, zxyval, i);
test_single_high_level(&interp2d_spline_eval, &interp2d_spline_eval_e, interp_s, x, y, xa, ya, zval, i);
test_single_high_level(&interp2d_spline_eval_deriv_x, &interp2d_spline_eval_deriv_x_e, interp_s, x, y, xa, ya, zxval, i);
test_single_high_level(&interp2d_spline_eval_deriv_y, &interp2d_spline_eval_deriv_y_e, interp_s, x, y, xa, ya, zyval, i);
test_single_high_level(&interp2d_spline_eval_deriv_xx,&interp2d_spline_eval_deriv_xx_e, interp_s, x, y, xa, ya, zxxval, i);
test_single_high_level(&interp2d_spline_eval_deriv_yy,&interp2d_spline_eval_deriv_yy_e, interp_s, x, y, xa, ya, zyyval, i);
test_single_high_level(&interp2d_spline_eval_deriv_xy,&interp2d_spline_eval_deriv_xy_e, interp_s, x, y, xa, ya, zxyval, i);
test_single_low_level(&interp2d_eval_no_boundary_check, &interp2d_eval_e_no_boundary_check, interp, xarr, yarr, zarr, x, y, xa, ya, zval, i);
}
gsl_interp_accel_free(xa);
gsl_interp_accel_free(ya);
interp2d_free(interp);
return status;
}
/*
* Tests that a given interpolation type reproduces the data points
* it is given, and then tests that it correctly reproduces additional
* values.
*/
static int
test_interp2d(
const double xarr[], const double yarr[], const double zarr[], /* data */
size_t xsize, size_t ysize, /* array sizes */
const double xval[], const double yval[], /* test points */
const double zval[], /* expected results */
const double zxval[], const double zyval[],
const double zxxval[], const double zyyval[], const double zxyval[],
size_t test_size, /* number of test points */
const gsl_interp2d_type * T)
{
gsl_interp_accel *xa = gsl_interp_accel_alloc();
gsl_interp_accel *ya = gsl_interp_accel_alloc();
int status = 0;
size_t xi, yi, zi, i;
gsl_interp2d * interp = gsl_interp2d_alloc(T, xsize, ysize);
gsl_spline2d * interp_s = gsl_spline2d_alloc(T, xsize, ysize);
unsigned int min_size = gsl_interp2d_type_min_size(T);
gsl_test_int(min_size, T->min_size, "gsl_interp2d_type_min_size on %s", gsl_interp2d_name(interp));
gsl_interp2d_init(interp, xarr, yarr, zarr, xsize, ysize);
gsl_spline2d_init(interp_s, xarr, yarr, zarr, xsize, ysize);
/* First check that the interpolation reproduces the given points */
for (xi = 0; xi < xsize; xi++)
{
double x = xarr[xi];
for (yi = 0; yi < ysize; yi++)
{
double y = yarr[yi];
zi = gsl_interp2d_idx(interp, xi, yi);
test_single_low_level(&gsl_interp2d_eval, &gsl_interp2d_eval_e,
interp, xarr, yarr, zarr, x, y,
xa, ya, zarr, zi);
test_single_low_level(&gsl_interp2d_eval_extrap,
&gsl_interp2d_eval_e_extrap, interp,
xarr, yarr, zarr, x, y, xa, ya, zarr, zi);
test_single_high_level(&gsl_spline2d_eval, &gsl_spline2d_eval_e,
interp_s, x, y, xa, ya, zarr, zi);
}
}
/* Then check additional points provided */
for (i = 0; i < test_size; i++)
{
double x = xval[i];
double y = yval[i];
test_single_low_level(&gsl_interp2d_eval, &gsl_interp2d_eval_e, interp, xarr, yarr, zarr, x, y, xa, ya, zval, i);
test_single_low_level(&gsl_interp2d_eval_deriv_x, &gsl_interp2d_eval_deriv_x_e, interp, xarr, yarr, zarr, x, y, xa, ya, zxval, i);
test_single_low_level(&gsl_interp2d_eval_deriv_y, &gsl_interp2d_eval_deriv_y_e, interp, xarr, yarr, zarr, x, y, xa, ya, zyval, i);
test_single_low_level(&gsl_interp2d_eval_deriv_xx,&gsl_interp2d_eval_deriv_xx_e, interp, xarr, yarr, zarr, x, y, xa, ya, zxxval, i);
test_single_low_level(&gsl_interp2d_eval_deriv_yy,&gsl_interp2d_eval_deriv_yy_e, interp, xarr, yarr, zarr, x, y, xa, ya, zyyval, i);
test_single_low_level(&gsl_interp2d_eval_deriv_xy,&gsl_interp2d_eval_deriv_xy_e, interp, xarr, yarr, zarr, x, y, xa, ya, zxyval, i);
test_single_high_level(&gsl_spline2d_eval, &gsl_spline2d_eval_e, interp_s, x, y, xa, ya, zval, i);
test_single_high_level(&gsl_spline2d_eval_deriv_x, &gsl_spline2d_eval_deriv_x_e, interp_s, x, y, xa, ya, zxval, i);
test_single_high_level(&gsl_spline2d_eval_deriv_y, &gsl_spline2d_eval_deriv_y_e, interp_s, x, y, xa, ya, zyval, i);
test_single_high_level(&gsl_spline2d_eval_deriv_xx,&gsl_spline2d_eval_deriv_xx_e, interp_s, x, y, xa, ya, zxxval, i);
test_single_high_level(&gsl_spline2d_eval_deriv_yy,&gsl_spline2d_eval_deriv_yy_e, interp_s, x, y, xa, ya, zyyval, i);
test_single_high_level(&gsl_spline2d_eval_deriv_xy,&gsl_spline2d_eval_deriv_xy_e, interp_s, x, y, xa, ya, zxyval, i);
test_single_low_level(&gsl_interp2d_eval_extrap, &gsl_interp2d_eval_e_extrap, interp, xarr, yarr, zarr, x, y, xa, ya, zval, i);
}
gsl_interp_accel_free(xa);
gsl_interp_accel_free(ya);
gsl_interp2d_free(interp);
gsl_spline2d_free(interp_s);
return status;
}
