Blame multifit_nlinear/test_penalty1.c

Packit 67cb25
#define penalty1_P         10
Packit 67cb25
#define penalty1_N         (penalty1_P + 1)
Packit 67cb25
Packit 67cb25
static double penalty1_x0[penalty1_P] = { 1.0, 2.0, 3.0, 4.0, 5.0,
Packit 67cb25
                                          6.0, 7.0, 8.0, 9.0, 10.0 };
Packit 67cb25
static double penalty1_epsrel = 1.0e-12;
Packit 67cb25
Packit 67cb25
static void
Packit 67cb25
penalty1_checksol(const double x[], const double sumsq,
Packit 67cb25
                  const double epsrel, const char *sname,
Packit 67cb25
                  const char *pname)
Packit 67cb25
{
Packit 67cb25
  const double sumsq_exact = 7.08765146709037993e-05;
Packit 67cb25
Packit 67cb25
  gsl_test_rel(sumsq, sumsq_exact, epsrel, "%s/%s sumsq",
Packit 67cb25
               sname, pname);
Packit 67cb25
Packit 67cb25
  (void)x; /* avoid unused parameter warning */
Packit 67cb25
}
Packit 67cb25
Packit 67cb25
static int
Packit 67cb25
penalty1_f (const gsl_vector * x, void *params, gsl_vector * f)
Packit 67cb25
{
Packit 67cb25
  const double alpha = 1.0e-5;
Packit 67cb25
  const double sqrt_alpha = sqrt(alpha);
Packit 67cb25
  size_t i;
Packit 67cb25
  double sum = 0.0;
Packit 67cb25
Packit 67cb25
  for (i = 0; i < penalty1_P; ++i)
Packit 67cb25
    {
Packit 67cb25
      double xi = gsl_vector_get(x, i);
Packit 67cb25
Packit 67cb25
      gsl_vector_set(f, i, sqrt_alpha*(xi - 1.0));
Packit 67cb25
Packit 67cb25
      sum += xi * xi;
Packit 67cb25
    }
Packit 67cb25
Packit 67cb25
  gsl_vector_set(f, penalty1_N - 1, sum - 0.25);
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
penalty1_df (const gsl_vector * x, void *params, gsl_matrix * J)
Packit 67cb25
{
Packit 67cb25
  const double alpha = 1.0e-5;
Packit 67cb25
  const double sqrt_alpha = sqrt(alpha);
Packit 67cb25
  size_t i;
Packit 67cb25
  gsl_matrix_view m = gsl_matrix_submatrix(J, 0, 0, penalty1_P, penalty1_P);
Packit 67cb25
  gsl_vector_view diag = gsl_matrix_diagonal(&m.matrix);
Packit 67cb25
Packit 67cb25
  gsl_matrix_set_zero(&m.matrix);
Packit 67cb25
  gsl_vector_set_all(&diag.vector, sqrt_alpha);
Packit 67cb25
Packit 67cb25
  for (i = 0; i < penalty1_P; ++i)
Packit 67cb25
    {
Packit 67cb25
      double xi = gsl_vector_get(x, i);
Packit 67cb25
      gsl_matrix_set(J, penalty1_N - 1, i, 2.0 * xi);
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
penalty1_fvv (const gsl_vector * x, const gsl_vector * v,
Packit 67cb25
              void *params, gsl_vector * fvv)
Packit 67cb25
{
Packit 67cb25
  double u;
Packit 67cb25
Packit 67cb25
  gsl_vector_set_zero(fvv);
Packit 67cb25
Packit 67cb25
  gsl_blas_ddot(v, v, &u);
Packit 67cb25
  gsl_vector_set(fvv, penalty1_N - 1, 2.0 * u);
Packit 67cb25
Packit 67cb25
  (void)x;      /* avoid unused parameter warning */
Packit 67cb25
  (void)params; /* avoid unused parameter warning */
Packit 67cb25
Packit 67cb25
  return GSL_SUCCESS;
Packit 67cb25
}
Packit 67cb25
Packit 67cb25
static gsl_multifit_nlinear_fdf penalty1_func =
Packit 67cb25
{
Packit 67cb25
  penalty1_f,
Packit 67cb25
  penalty1_df,
Packit 67cb25
  penalty1_fvv,
Packit 67cb25
  penalty1_N,
Packit 67cb25
  penalty1_P,
Packit 67cb25
  NULL,
Packit 67cb25
  0,
Packit 67cb25
  0,
Packit 67cb25
  0
Packit 67cb25
};
Packit 67cb25
Packit 67cb25
static test_fdf_problem penalty1_problem =
Packit 67cb25
{
Packit 67cb25
  "penalty1",
Packit 67cb25
  penalty1_x0,
Packit 67cb25
  NULL,
Packit 67cb25
  NULL,
Packit 67cb25
  &penalty1_epsrel,
Packit 67cb25
  &penalty1_checksol,
Packit 67cb25
  &penalty1_func
Packit 67cb25
};