|
Packit |
67cb25 |
#define enso_N 168
|
|
Packit |
67cb25 |
#define enso_P 9
|
|
Packit |
67cb25 |
|
|
Packit |
67cb25 |
static double enso_x0a[enso_P] = { 11.0, 3.0, 0.5, 40.0, -0.7, -1.3, 25.0, -0.3, 1.4 };
|
|
Packit |
67cb25 |
static double enso_x0b[enso_P] = { 10.0, 3.0, 0.5, 44.0, -1.5, 0.5, 26.0, -0.1, 1.5 };
|
|
Packit |
67cb25 |
static double enso_epsrel = 1.0e-3;
|
|
Packit |
67cb25 |
|
|
Packit |
67cb25 |
static double enso_J[enso_N * enso_P];
|
|
Packit |
67cb25 |
|
|
Packit |
67cb25 |
static double enso_sigma[enso_P] = {
|
|
Packit |
67cb25 |
1.7488832467E-01, 2.4310052139E-01, 2.4354686618E-01, 9.4408025976E-01, 2.8078369611E-01,
|
|
Packit |
67cb25 |
4.8073701119E-01, 4.1612939130E-01, 5.1460022911E-01, 2.5434468893E-01
|
|
Packit |
67cb25 |
};
|
|
Packit |
67cb25 |
|
|
Packit |
67cb25 |
static double enso_F[enso_N] = {
|
|
Packit |
67cb25 |
12.90000, 11.30000, 10.60000, 11.20000, 10.90000,
|
|
Packit |
67cb25 |
7.500000, 7.700000, 11.70000, 12.90000, 14.30000,
|
|
Packit |
67cb25 |
10.90000, 13.70000, 17.10000, 14.00000, 15.30000,
|
|
Packit |
67cb25 |
8.500000, 5.700000, 5.500000, 7.600000, 8.600000,
|
|
Packit |
67cb25 |
7.300000, 7.600000, 12.70000, 11.00000, 12.70000,
|
|
Packit |
67cb25 |
12.90000, 13.00000, 10.90000, 10.400000, 10.200000,
|
|
Packit |
67cb25 |
8.000000, 10.90000, 13.60000, 10.500000, 9.200000,
|
|
Packit |
67cb25 |
12.40000, 12.70000, 13.30000, 10.100000, 7.800000,
|
|
Packit |
67cb25 |
4.800000, 3.000000, 2.500000, 6.300000, 9.700000,
|
|
Packit |
67cb25 |
11.60000, 8.600000, 12.40000, 10.500000, 13.30000,
|
|
Packit |
67cb25 |
10.400000, 8.100000, 3.700000, 10.70000, 5.100000,
|
|
Packit |
67cb25 |
10.400000, 10.90000, 11.70000, 11.40000, 13.70000,
|
|
Packit |
67cb25 |
14.10000, 14.00000, 12.50000, 6.300000, 9.600000,
|
|
Packit |
67cb25 |
11.70000, 5.000000, 10.80000, 12.70000, 10.80000,
|
|
Packit |
67cb25 |
11.80000, 12.60000, 15.70000, 12.60000, 14.80000,
|
|
Packit |
67cb25 |
7.800000, 7.100000, 11.20000, 8.100000, 6.400000,
|
|
Packit |
67cb25 |
5.200000, 12.00000, 10.200000, 12.70000, 10.200000,
|
|
Packit |
67cb25 |
14.70000, 12.20000, 7.100000, 5.700000, 6.700000,
|
|
Packit |
67cb25 |
3.900000, 8.500000, 8.300000, 10.80000, 16.70000,
|
|
Packit |
67cb25 |
12.60000, 12.50000, 12.50000, 9.800000, 7.200000,
|
|
Packit |
67cb25 |
4.100000, 10.60000, 10.100000, 10.100000, 11.90000,
|
|
Packit |
67cb25 |
13.60000, 16.30000, 17.60000, 15.50000, 16.00000,
|
|
Packit |
67cb25 |
15.20000, 11.20000, 14.30000, 14.50000, 8.500000,
|
|
Packit |
67cb25 |
12.00000, 12.70000, 11.30000, 14.50000, 15.10000,
|
|
Packit |
67cb25 |
10.400000, 11.50000, 13.40000, 7.500000, 0.6000000,
|
|
Packit |
67cb25 |
0.3000000, 5.500000, 5.000000, 4.600000, 8.200000,
|
|
Packit |
67cb25 |
9.900000, 9.200000, 12.50000, 10.90000, 9.900000,
|
|
Packit |
67cb25 |
8.900000, 7.600000, 9.500000, 8.400000, 10.70000,
|
|
Packit |
67cb25 |
13.60000, 13.70000, 13.70000, 16.50000, 16.80000,
|
|
Packit |
67cb25 |
17.10000, 15.40000, 9.500000, 6.100000, 10.100000,
|
|
Packit |
67cb25 |
9.300000, 5.300000, 11.20000, 16.60000, 15.60000,
|
|
Packit |
67cb25 |
12.00000, 11.50000, 8.600000, 13.80000, 8.700000,
|
|
Packit |
67cb25 |
8.600000, 8.600000, 8.700000, 12.80000, 13.20000,
|
|
Packit |
67cb25 |
14.00000, 13.40000, 14.80000
|
|
Packit |
67cb25 |
};
|
|
Packit |
67cb25 |
|
|
Packit |
67cb25 |
static void
|
|
Packit |
67cb25 |
enso_checksol(const double x[], const double sumsq,
|
|
Packit |
67cb25 |
const double epsrel, const char *sname,
|
|
Packit |
67cb25 |
const char *pname)
|
|
Packit |
67cb25 |
{
|
|
Packit |
67cb25 |
size_t i;
|
|
Packit |
67cb25 |
const double sumsq_exact = 7.8853978668E+02;
|
|
Packit |
67cb25 |
const double enso_x[enso_P] = {
|
|
Packit |
67cb25 |
1.0510749193E+01, 3.0762128085E+00, 5.3280138227E-01,
|
|
Packit |
67cb25 |
4.4311088700E+01, -1.6231428586E+00, 5.2554493756E-01,
|
|
Packit |
67cb25 |
2.6887614440E+01, 2.1232288488E-01, 1.4966870418E+00 };
|
|
Packit |
67cb25 |
|
|
Packit |
67cb25 |
gsl_test_rel(sumsq, sumsq_exact, epsrel, "%s/%s sumsq",
|
|
Packit |
67cb25 |
sname, pname);
|
|
Packit |
67cb25 |
|
|
Packit |
67cb25 |
for (i = 0; i < enso_P; ++i)
|
|
Packit |
67cb25 |
{
|
|
Packit |
67cb25 |
gsl_test_rel(x[i], enso_x[i], epsrel, "%s/%s i=%zu",
|
|
Packit |
67cb25 |
sname, pname, i);
|
|
Packit |
67cb25 |
}
|
|
Packit |
67cb25 |
}
|
|
Packit |
67cb25 |
|
|
Packit |
67cb25 |
|
|
Packit |
67cb25 |
static int
|
|
Packit |
67cb25 |
enso_f (const gsl_vector * x, void *params, gsl_vector * f)
|
|
Packit |
67cb25 |
{
|
|
Packit |
67cb25 |
double b[enso_P];
|
|
Packit |
67cb25 |
size_t i;
|
|
Packit |
67cb25 |
|
|
Packit |
67cb25 |
for (i = 0; i < enso_P; i++)
|
|
Packit |
67cb25 |
{
|
|
Packit |
67cb25 |
b[i] = gsl_vector_get(x, i);
|
|
Packit |
67cb25 |
}
|
|
Packit |
67cb25 |
|
|
Packit |
67cb25 |
for (i = 0; i < enso_N; i++)
|
|
Packit |
67cb25 |
{
|
|
Packit |
67cb25 |
double t = (i + 1.0);
|
|
Packit |
67cb25 |
double y;
|
|
Packit |
67cb25 |
y = b[0];
|
|
Packit |
67cb25 |
y += b[1] * cos(2*M_PI*t/12);
|
|
Packit |
67cb25 |
y += b[2] * sin(2*M_PI*t/12);
|
|
Packit |
67cb25 |
y += b[4] * cos(2*M_PI*t/b[3]);
|
|
Packit |
67cb25 |
y += b[5] * sin(2*M_PI*t/b[3]);
|
|
Packit |
67cb25 |
y += b[7] * cos(2*M_PI*t/b[6]);
|
|
Packit |
67cb25 |
y += b[8] * sin(2*M_PI*t/b[6]);
|
|
Packit |
67cb25 |
|
|
Packit |
67cb25 |
gsl_vector_set (f, i, enso_F[i] - y);
|
|
Packit |
67cb25 |
}
|
|
Packit |
67cb25 |
|
|
Packit |
67cb25 |
(void)params; /* avoid unused parameter warning */
|
|
Packit |
67cb25 |
|
|
Packit |
67cb25 |
return GSL_SUCCESS;
|
|
Packit |
67cb25 |
}
|
|
Packit |
67cb25 |
|
|
Packit |
67cb25 |
static int
|
|
Packit |
67cb25 |
enso_df (CBLAS_TRANSPOSE_t TransJ, const gsl_vector * x,
|
|
Packit |
67cb25 |
const gsl_vector * u, void * params, gsl_vector * v,
|
|
Packit |
67cb25 |
gsl_matrix * JTJ)
|
|
Packit |
67cb25 |
{
|
|
Packit |
67cb25 |
gsl_matrix_view J = gsl_matrix_view_array(enso_J, enso_N, enso_P);
|
|
Packit |
67cb25 |
double b[enso_P];
|
|
Packit |
67cb25 |
size_t i;
|
|
Packit |
67cb25 |
|
|
Packit |
67cb25 |
for (i = 0; i < enso_P; i++)
|
|
Packit |
67cb25 |
{
|
|
Packit |
67cb25 |
b[i] = gsl_vector_get(x, i);
|
|
Packit |
67cb25 |
}
|
|
Packit |
67cb25 |
|
|
Packit |
67cb25 |
for (i = 0; i < enso_N; i++)
|
|
Packit |
67cb25 |
{
|
|
Packit |
67cb25 |
double t = (i + 1.0);
|
|
Packit |
67cb25 |
|
|
Packit |
67cb25 |
gsl_matrix_set (&J.matrix, i, 0, -1.0);
|
|
Packit |
67cb25 |
gsl_matrix_set (&J.matrix, i, 1, -cos(2*M_PI*t/12));
|
|
Packit |
67cb25 |
gsl_matrix_set (&J.matrix, i, 2, -sin(2*M_PI*t/12));
|
|
Packit |
67cb25 |
gsl_matrix_set (&J.matrix, i, 3,
|
|
Packit |
67cb25 |
-b[4]*(2*M_PI*t/(b[3]*b[3]))*sin(2*M_PI*t/b[3])
|
|
Packit |
67cb25 |
+b[5]*(2*M_PI*t/(b[3]*b[3]))*cos(2*M_PI*t/b[3]));
|
|
Packit |
67cb25 |
gsl_matrix_set (&J.matrix, i, 4, -cos(2*M_PI*t/b[3]));
|
|
Packit |
67cb25 |
gsl_matrix_set (&J.matrix, i, 5, -sin(2*M_PI*t/b[3]));
|
|
Packit |
67cb25 |
gsl_matrix_set (&J.matrix, i, 6,
|
|
Packit |
67cb25 |
-b[7] * (2*M_PI*t/(b[6]*b[6])) * sin(2*M_PI*t/b[6])
|
|
Packit |
67cb25 |
+b[8] * (2*M_PI*t/(b[6]*b[6])) * cos(2*M_PI*t/b[6]));
|
|
Packit |
67cb25 |
gsl_matrix_set (&J.matrix, i, 7, -cos(2*M_PI*t/b[6]));
|
|
Packit |
67cb25 |
gsl_matrix_set (&J.matrix, i, 8, -sin(2*M_PI*t/b[6]));
|
|
Packit |
67cb25 |
}
|
|
Packit |
67cb25 |
|
|
Packit |
67cb25 |
if (v)
|
|
Packit |
67cb25 |
gsl_blas_dgemv(TransJ, 1.0, &J.matrix, u, 0.0, v);
|
|
Packit |
67cb25 |
|
|
Packit |
67cb25 |
if (JTJ)
|
|
Packit |
67cb25 |
gsl_blas_dsyrk(CblasLower, CblasTrans, 1.0, &J.matrix, 0.0, JTJ);
|
|
Packit |
67cb25 |
|
|
Packit |
67cb25 |
(void)params; /* avoid unused parameter warning */
|
|
Packit |
67cb25 |
|
|
Packit |
67cb25 |
return GSL_SUCCESS;
|
|
Packit |
67cb25 |
}
|
|
Packit |
67cb25 |
|
|
Packit |
67cb25 |
static gsl_multilarge_nlinear_fdf enso_func =
|
|
Packit |
67cb25 |
{
|
|
Packit |
67cb25 |
enso_f,
|
|
Packit |
67cb25 |
enso_df,
|
|
Packit |
67cb25 |
NULL, /* analytic expression too complex */
|
|
Packit |
67cb25 |
enso_N,
|
|
Packit |
67cb25 |
enso_P,
|
|
Packit |
67cb25 |
NULL,
|
|
Packit |
67cb25 |
0,
|
|
Packit |
67cb25 |
0,
|
|
Packit |
67cb25 |
0,
|
|
Packit |
67cb25 |
0
|
|
Packit |
67cb25 |
};
|
|
Packit |
67cb25 |
|
|
Packit |
67cb25 |
static test_fdf_problem ensoa_problem =
|
|
Packit |
67cb25 |
{
|
|
Packit |
67cb25 |
"nist-ENSOa",
|
|
Packit |
67cb25 |
enso_x0a,
|
|
Packit |
67cb25 |
enso_sigma,
|
|
Packit |
67cb25 |
&enso_epsrel,
|
|
Packit |
67cb25 |
&enso_checksol,
|
|
Packit |
67cb25 |
&enso_func
|
|
Packit |
67cb25 |
};
|
|
Packit |
67cb25 |
|
|
Packit |
67cb25 |
static test_fdf_problem ensob_problem =
|
|
Packit |
67cb25 |
{
|
|
Packit |
67cb25 |
"nist-ENSOb",
|
|
Packit |
67cb25 |
enso_x0b,
|
|
Packit |
67cb25 |
enso_sigma,
|
|
Packit |
67cb25 |
&enso_epsrel,
|
|
Packit |
67cb25 |
&enso_checksol,
|
|
Packit |
67cb25 |
&enso_func
|
|
Packit |
67cb25 |
};
|