/* multifit/work.c

 * 

 * Copyright (C) 2000, 2007, 2009 Brian Gough

 * 

 * 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.

 */

#include <config.h>

#include <gsl/gsl_errno.h>

#include <gsl/gsl_multifit.h>

gsl_multifit_linear_workspace *

gsl_multifit_linear_alloc (const size_t nmax, const size_t pmax)

{

  gsl_multifit_linear_workspace *w;

  w = calloc (1, sizeof (gsl_multifit_linear_workspace));

  if (w == 0)

    {

      GSL_ERROR_VAL ("failed to allocate space for multifit_linear struct",

                     GSL_ENOMEM, 0);

    }

  w->nmax = nmax;                     /* max number of observations */

  w->pmax = pmax;                     /* max number of parameters */

  w->n = 0;

  w->p = 0;

  w->rcond = 0.0;

  w->A = gsl_matrix_alloc (nmax, pmax);

  if (w->A == 0)

    {

      gsl_multifit_linear_free(w);

      GSL_ERROR_VAL ("failed to allocate space for A", GSL_ENOMEM, 0);

    }

  w->Q = gsl_matrix_alloc (pmax, pmax);

  if (w->Q == 0)

    {

      gsl_multifit_linear_free(w);

      GSL_ERROR_VAL ("failed to allocate space for Q", GSL_ENOMEM, 0);

    }

  w->QSI = gsl_matrix_alloc (pmax, pmax);

  if (w->QSI == 0)

    {

      gsl_multifit_linear_free(w);

      GSL_ERROR_VAL ("failed to allocate space for QSI", GSL_ENOMEM, 0);

    }

  w->S = gsl_vector_alloc (pmax);

  if (w->S == 0)

    {

      gsl_multifit_linear_free(w);

      GSL_ERROR_VAL ("failed to allocate space for S", GSL_ENOMEM, 0);

    }

  w->t = gsl_vector_alloc (nmax);

  if (w->t == 0)

    {

      gsl_multifit_linear_free(w);

      GSL_ERROR_VAL ("failed to allocate space for t", GSL_ENOMEM, 0);

    }

  w->xt = gsl_vector_calloc (pmax);

  if (w->xt == 0)

    {

      gsl_multifit_linear_free(w);

      GSL_ERROR_VAL ("failed to allocate space for xt", GSL_ENOMEM, 0);

    }

  w->D = gsl_vector_calloc (pmax);

  if (w->D == 0)

    {

      gsl_multifit_linear_free(w);

      GSL_ERROR_VAL ("failed to allocate space for D", GSL_ENOMEM, 0);

    }

  return w;

}

void

gsl_multifit_linear_free (gsl_multifit_linear_workspace * w)

{

  RETURN_IF_NULL (w);

  if (w->A)

    gsl_matrix_free (w->A);

  if (w->Q)

    gsl_matrix_free (w->Q);

  if (w->QSI)

    gsl_matrix_free (w->QSI);

  if (w->S)

    gsl_vector_free (w->S);

  if (w->t)

    gsl_vector_free (w->t);

  if (w->xt)

    gsl_vector_free (w->xt);

  if (w->D)

    gsl_vector_free (w->D);

  free (w);

}