summaryrefslogtreecommitdiff
path: root/gsl-1.9/multifit/work.c
diff options
context:
space:
mode:
Diffstat (limited to 'gsl-1.9/multifit/work.c')
-rw-r--r--gsl-1.9/multifit/work.c133
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);
+}
+