diff options
Diffstat (limited to 'gsl-1.9/multifit/work.c')
-rw-r--r-- | gsl-1.9/multifit/work.c | 133 |
1 files changed, 133 insertions, 0 deletions
diff --git a/gsl-1.9/multifit/work.c b/gsl-1.9/multifit/work.c new file mode 100644 index 0000000..f05e3ff --- /dev/null +++ b/gsl-1.9/multifit/work.c @@ -0,0 +1,133 @@ +/* multifit/work.c + * + * Copyright (C) 2000 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 2 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 (size_t n, size_t p) +{ + gsl_multifit_linear_workspace *w; + + w = (gsl_multifit_linear_workspace *) + malloc (sizeof (gsl_multifit_linear_workspace)); + + if (w == 0) + { + GSL_ERROR_VAL ("failed to allocate space for multifit_linear struct", + GSL_ENOMEM, 0); + } + + w->n = n; /* number of observations */ + w->p = p; /* number of parameters */ + + w->A = gsl_matrix_alloc (n, p); + + if (w->A == 0) + { + free (w); + GSL_ERROR_VAL ("failed to allocate space for A", GSL_ENOMEM, 0); + } + + w->Q = gsl_matrix_alloc (p, p); + + if (w->Q == 0) + { + gsl_matrix_free (w->A); + free (w); + GSL_ERROR_VAL ("failed to allocate space for Q", GSL_ENOMEM, 0); + } + + w->QSI = gsl_matrix_alloc (p, p); + + if (w->QSI == 0) + { + gsl_matrix_free (w->Q); + gsl_matrix_free (w->A); + free (w); + GSL_ERROR_VAL ("failed to allocate space for QSI", GSL_ENOMEM, 0); + } + + w->S = gsl_vector_alloc (p); + + if (w->S == 0) + { + gsl_matrix_free (w->QSI); + gsl_matrix_free (w->Q); + gsl_matrix_free (w->A); + free (w); + GSL_ERROR_VAL ("failed to allocate space for S", GSL_ENOMEM, 0); + } + + w->t = gsl_vector_alloc (n); + + if (w->t == 0) + { + gsl_vector_free (w->S); + gsl_matrix_free (w->QSI); + gsl_matrix_free (w->Q); + gsl_matrix_free (w->A); + free (w); + GSL_ERROR_VAL ("failed to allocate space for t", GSL_ENOMEM, 0); + } + + w->xt = gsl_vector_calloc (p); + + if (w->xt == 0) + { + gsl_vector_free (w->t); + gsl_vector_free (w->S); + gsl_matrix_free (w->QSI); + gsl_matrix_free (w->Q); + gsl_matrix_free (w->A); + free (w); + GSL_ERROR_VAL ("failed to allocate space for xt", GSL_ENOMEM, 0); + } + + w->D = gsl_vector_calloc (p); + + if (w->D == 0) + { + gsl_vector_free (w->D); + gsl_vector_free (w->t); + gsl_vector_free (w->S); + gsl_matrix_free (w->QSI); + gsl_matrix_free (w->Q); + gsl_matrix_free (w->A); + free (w); + GSL_ERROR_VAL ("failed to allocate space for xt", GSL_ENOMEM, 0); + } + + return w; +} + +void +gsl_multifit_linear_free (gsl_multifit_linear_workspace * work) +{ + gsl_matrix_free (work->A); + gsl_matrix_free (work->Q); + gsl_matrix_free (work->QSI); + gsl_vector_free (work->S); + gsl_vector_free (work->t); + gsl_vector_free (work->xt); + gsl_vector_free (work->D); + free (work); +} + |