/* ode-initval/control.c
 * 
 * Copyright (C) 1996, 1997, 1998, 1999, 2000 Gerard Jungman
 * 
 * This program is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation; either version 3 of the License, or (at
 * your option) any later version.
 * 
 * This program is distributed in the hope that it will be useful, but
 * WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
 * General Public License for more details.
 * 
 * You should have received a copy of the GNU General Public License
 * along with this program; if not, write to the Free Software
 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
 */

/* Author:  G. Jungman */

#include <config.h>
#include <stdlib.h>
#include <gsl/gsl_math.h>
#include <gsl/gsl_errno.h>
#include <gsl/gsl_odeiv2.h>

gsl_odeiv2_control *
gsl_odeiv2_control_alloc (const gsl_odeiv2_control_type * T)
{
  gsl_odeiv2_control *c =
    (gsl_odeiv2_control *) malloc (sizeof (gsl_odeiv2_control));

  if (c == 0)
    {
      GSL_ERROR_NULL ("failed to allocate space for control struct",
                      GSL_ENOMEM);
    };

  c->type = T;
  c->state = c->type->alloc ();

  if (c->state == 0)
    {
      free (c);                 /* exception in constructor, avoid memory leak */

      GSL_ERROR_NULL ("failed to allocate space for control state",
                      GSL_ENOMEM);
    };

  return c;
}

int
gsl_odeiv2_control_init (gsl_odeiv2_control * c,
                         double eps_abs, double eps_rel,
                         double a_y, double a_dydt)
{
  return c->type->init (c->state, eps_abs, eps_rel, a_y, a_dydt);
}

void
gsl_odeiv2_control_free (gsl_odeiv2_control * c)
{
  RETURN_IF_NULL (c);
  c->type->free (c->state);
  free (c);
}

const char *
gsl_odeiv2_control_name (const gsl_odeiv2_control * c)
{
  return c->type->name;
}

int
gsl_odeiv2_control_hadjust (gsl_odeiv2_control * c, gsl_odeiv2_step * s,
                            const double y[], const double yerr[],
                            const double dydt[], double *h)
{
  return c->type->hadjust (c->state, s->dimension, s->type->order (s->state),
                           y, yerr, dydt, h);
}

int
gsl_odeiv2_control_errlevel (gsl_odeiv2_control * c, const double y,
                             const double dydt, const double h,
                             const size_t ind, double *errlev)
{
  return c->type->errlevel (c->state, y, dydt, h, ind, errlev);
}

int
gsl_odeiv2_control_set_driver (gsl_odeiv2_control * c,
                               const gsl_odeiv2_driver * d)
{
  if (d != NULL)
    {
      c->type->set_driver (c->state, d);
    }
  else
    {
      GSL_ERROR_NULL ("driver pointer is null", GSL_EFAULT);
    }

  return GSL_SUCCESS;
}