static PairsFloatsCacheEntry *
collect_float_pairs_ce (GnmValue const *vx, GnmValue const *vy,
GnmEvalPos const *ep, CollectFlags flags)
{
PairsFloatsCacheEntry *ce = g_new0 (PairsFloatsCacheEntry, 1);
GSList *missing0 = NULL, *missing1 = NULL;
int n0, n1;
ce->flags = flags;
ce->data_x = collect_floats_value_with_info (vx, ep, flags,
&n0, &missing0, &ce->error);
if (ce->error)
goto err;
ce->data_y = collect_floats_value_with_info (vy, ep, flags,
&n1, &missing1, &ce->error);
if (ce->error)
goto err;
if (n0 != n1) {
ce->n = -1;
goto err;
}
if (missing0 || missing1) {
missing0 = gnm_slist_sort_merge (missing0, missing1);
missing1 = NULL;
gnm_strip_missing (ce->data_x, &n0, missing0);
gnm_strip_missing (ce->data_y, &n1, missing0);
}
ce->n = n0;
err:
if (ce->n <= 0) {
g_free (ce->data_x);
ce->data_x = NULL;
g_free (ce->data_y);
ce->data_y = NULL;
}
g_slist_free (missing0);
g_slist_free (missing1);
return ce;
}
static GnmValue *
gnumeric_periodogram (GnmFuncEvalInfo *ei, GnmValue const * const *argv)
{
gnm_float *ord, *absc;
int filter, interp;
int n0, n1, nb;
GnmValue *error = NULL;
GnmValue *res;
CollectFlags flags;
GnmEvalPos const * const ep = ei->pos;
GnmValue const * const Pt = argv[0];
int i;
GSList *missing0 = NULL, *missing1 = NULL;
gnm_complex *in, *out = NULL;
int const cols = value_area_get_width (Pt, ep);
int const rows = value_area_get_height (Pt, ep);
if (cols == 1)
nb=rows;
else {
if (rows == 1)
nb=cols;
else
nb=0;
}
if (nb == 0) {
res = value_new_error_std (ei->pos, GNM_ERROR_VALUE);
return res;
}
flags=COLLECT_IGNORE_BLANKS | COLLECT_IGNORE_STRINGS | COLLECT_IGNORE_BOOLS;
ord = collect_floats_value_with_info (argv[0], ei->pos, flags,
&n0, &missing0, &error);
if (error) {
g_slist_free (missing0);
return error;
}
if (n0 == 0) {
res = value_new_error_std (ei->pos, GNM_ERROR_VALUE);
return res;
}
if (argv[1]) {
filter = (int) gnm_floor (value_get_as_float (argv[1]));
if (filter < 0 || filter > FILTER_WELCH) {
g_slist_free (missing0);
g_free (ord);
res = value_new_error_std (ei->pos, GNM_ERROR_VALUE);
return res;
}
} else
filter = FILTER_NONE;
if (argv[2]) {
gnm_float *interpolated, *new_ord, start, incr;
int n2;
INTERPPROC(interpproc) = NULL;
absc = collect_floats_value_with_info (argv[2], ei->pos, flags,
&n1, &missing1, &error);
if (n1 == 1) {
g_slist_free (missing1);
g_free (absc);
goto no_absc;
}
if (error) {
g_slist_free (missing0);
g_slist_free (missing1);
g_free (absc);
return error;
}
if (n1 == 0) {
g_slist_free (missing0);
g_slist_free (missing1);
g_free (absc);
g_free (ord);
return value_new_error_std (ei->pos, GNM_ERROR_VALUE);
}
if (argv[3]) {
interp = (int) gnm_floor (value_get_as_float (argv[3]));
if (interp < 0 || interp > INTERPOLATION_SPLINE_AVG) {
g_slist_free (missing0);
g_slist_free (missing1);
g_free (absc);
g_free (ord);
return error;
}
} else
interp = INTERPOLATION_LINEAR;
if (missing0 || missing1) {
GSList *missing = gnm_slist_sort_merge (missing0, missing1);
gnm_strip_missing (ord, &n0, missing);
gnm_strip_missing (absc, &n1, missing);
g_slist_free (missing);
if (n0 != n1)
g_warning ("This should not happen. n0=%d n1=%d\n",
n0, n1);
}
n0 = n1 = MIN (n0, n1);
/* here we test if there is abscissas are always increasing, if not,
an error is returned */
if (n0 < 2 || !gnm_range_increasing (absc, n0) || n0 == 0) {
g_free (absc);
g_free (ord);
return value_new_error_std (ei->pos, GNM_ERROR_VALUE);
}
if (argv[4]) {
n1 = (int) gnm_floor (value_get_as_float (argv[4]));
if (n1 < n0) {
g_free (absc);
g_free (ord);
return value_new_error_std (ei->pos, GNM_ERROR_VALUE);
}
nb = 1;
while (nb < n1)
nb *= 2;
} else {
n1 = 1;
while (n1 < n0)
n1 *= 2;
nb = n1;
}
incr = (absc[n0 - 1] - absc[0]) / n1;
switch (interp) {
case INTERPOLATION_LINEAR:
interpproc = linear_interpolation;
start = absc[0];
n2 = n1;
break;
case INTERPOLATION_LINEAR_AVG:
interpproc = linear_averaging;
start = absc[0] - incr / 2.;
n2 = n1 + 1;
break;
case INTERPOLATION_STAIRCASE:
interpproc = staircase_interpolation;
start = absc[0];
n2 = n1;
break;
case INTERPOLATION_STAIRCASE_AVG:
interpproc = staircase_averaging;
start = absc[0] - incr / 2.;
n2 = n1 + 1;
break;
case INTERPOLATION_SPLINE:
interpproc = spline_interpolation;
start = absc[0];
n2 = n1;
break;
case INTERPOLATION_SPLINE_AVG:
interpproc = spline_averaging;
start = absc[0] - incr / 2.;
n2 = n1 + 1;
break;
default:
g_free (absc);
g_free (ord);
return value_new_error_std (ei->pos, GNM_ERROR_NA);
}
interpolated = g_new (gnm_float, n1);
for (i = 0; i < n2; i++)
interpolated[i] = start + i * incr;
new_ord = interpproc (absc, ord, n0, interpolated, n1);
g_free (ord);
ord = new_ord;
if (ord == NULL) {
g_free (absc);
g_free (interpolated);
return value_new_error_std (ei->pos, GNM_ERROR_NA);
}
n0 = n1;
} else {
no_absc:
/* we have no interpolation to apply, so just take the values */
if (missing0) {
gnm_strip_missing (ord, &n0, missing0);
g_slist_free (missing0);
}
nb = 1;
while (nb < n0)
nb *= 2;
}
/* Now apply the filter if any */
if (filter != FILTER_NONE) {
gnm_float factor;
switch (filter) {
case FILTER_BARTLETT:
factor = n0 / 2.;
for (i = 0; i < n0; i++)
ord[i] *= 1. - gnm_abs ((i / factor - 1));
break;
case FILTER_HANN:
factor = 2. * M_PIgnum / n0;
for (i = 0; i < n0; i++)
ord[i] *= 0.5 * (1 - gnm_cos (factor * i));
break;
case FILTER_WELCH:
factor = n0 / 2.;
for (i = 0; i < n0; i++)
ord[i] *= 1. - (i / factor - 1.) * (i / factor - 1.);
break;
}
}
/* Transform and return the result */
in = g_new0 (gnm_complex, nb);
for (i = 0; i < n0; i++){
in[i].re = ord[i];
}
g_free (ord);
gnm_fourier_fft (in, nb, 1, &out, FALSE);
g_free (in);
nb /= 2;
if (out && nb > 0) {
res = value_new_array_non_init (1 , nb);
res->v_array.vals[0] = g_new (GnmValue *, nb);
for (i = 0; i < nb; i++)
res->v_array.vals[0][i] =
value_new_float (gnm_sqrt (
out[i].re * out[i].re +
out[i].im * out[i].im));
g_free (out);
} else
static GnmValue *
gnumeric_interpolation (GnmFuncEvalInfo *ei, GnmValue const * const *argv)
{
gnm_float *vals0, *vals1, *vals2, *fres;
int n0, n2;
int interp;
GnmValue *error = NULL;
GnmValue *res;
CollectFlags flags;
GnmEvalPos const * const ep = ei->pos;
GnmValue const * const PtInterpolation = argv[2];
int r, i;
GSList *missing2 = NULL, *missing;
INTERPPROC interpproc = NULL;
gboolean constp = FALSE;
/* argv[2] */
int const cols = value_area_get_width (PtInterpolation, ep);
int const rows = value_area_get_height (PtInterpolation, ep);
if (rows == 0 || cols != 1) {
res = value_new_error_std (ei->pos, GNM_ERROR_VALUE);
return res;
}
flags = COLLECT_IGNORE_BLANKS | COLLECT_IGNORE_STRINGS | COLLECT_IGNORE_BOOLS | COLLECT_IGNORE_ERRORS;
vals2 = collect_floats_value_with_info (PtInterpolation, ei->pos, flags,
&n2, &missing2, &error);
if (error) {
g_slist_free (missing2);
return error;
}
/* argv[3] */
if (argv[3]) {
interp = (int) gnm_floor (value_get_as_float (argv[3]));
if (interp < 0 || interp > INTERPOLATION_SPLINE_AVG) {
g_slist_free (missing2);
g_free (vals2);
return value_new_error_VALUE (ei->pos);
}
} else
interp = INTERPOLATION_LINEAR;
switch (interp) {
case INTERPOLATION_LINEAR:
interpproc = linear_interpolation;
break;
case INTERPOLATION_LINEAR_AVG:
interpproc = linear_averaging;
n2--;
break;
case INTERPOLATION_STAIRCASE:
interpproc = staircase_interpolation;
break;
case INTERPOLATION_STAIRCASE_AVG:
interpproc = staircase_averaging;
n2--;
break;
case INTERPOLATION_SPLINE:
interpproc = spline_interpolation;
break;
case INTERPOLATION_SPLINE_AVG:
interpproc = spline_averaging;
n2--;
break;
}
if (n2 <= 0) {
g_slist_free (missing2);
g_free (vals2);
return value_new_error_std (ei->pos, GNM_ERROR_VALUE);
}
/* argv[0] & argv[1] */
flags = COLLECT_IGNORE_BLANKS | COLLECT_IGNORE_STRINGS | COLLECT_IGNORE_BOOLS;
error = collect_float_pairs (argv[0], argv[1], ei->pos, flags, &vals0, &vals1,
&n0, &constp);
if (error) {
g_slist_free (missing2);
g_free (vals2);
return error;
}
/* Check whether the abscissae are increasing, if not order them */
if (!gnm_range_increasing (vals0, n0)) {
gboolean switched = TRUE;
if (constp) {
vals0 = g_memdup (vals0, sizeof(gnm_float) * n0);
vals1 = g_memdup (vals1, sizeof(gnm_float) * n0);
constp = FALSE;
}
while (switched) {
gnm_float *val;
switched = FALSE;
for (i = 1, val = vals0; i < n0; i++, val++) {
if (*val == *(val + 1)) {
res = value_new_error_std (ei->pos, GNM_ERROR_VALUE) ;
goto done;
}
if (*val > *(val + 1)) {
gnm_float v = *val;
*val = *(val + 1);
*(val + 1) = v;
v = *(vals1 + i);
*(vals1 + i) = *(vals1 + i - 1);
*(vals1 + i - 1) = v;
switched = TRUE;
}
}
}
}
{
int n = n2;
if (missing2)
gnm_strip_missing (vals2, &n, missing2);
res = value_new_array_non_init (1 , n2);
i = 0;
res->v_array.vals[0] = g_new (GnmValue *, n2);
fres = interpproc (vals0, vals1, n0, vals2, n);
missing = missing2;
if (fres) {
i = 0;
for (r = 0 ; r < n2; ++r)
if (missing && r == GPOINTER_TO_INT (missing->data)) {
missing = missing->next;
res->v_array.vals[0][r] = value_new_error_std (ei->pos, GNM_ERROR_VALUE);
} else
res->v_array.vals[0][r] = value_new_float (fres[i++]);
g_free (fres);
} else {
for( r = 0 ; r < n2; ++r)
res->v_array.vals[0][r] = value_new_error_std (ei->pos, GNM_ERROR_VALUE);
}
}
done:
g_slist_free (missing2);
if (!constp) {
g_free (vals0);
g_free (vals1);
}
g_free (vals2);
return res;
}
