#include <stdlib.h>
#include <string.h>
-#include <libpspp/alloc.h>
#include <libpspp/message.h>
#include <data/variable.h>
#include <data/category.h>
#include <gsl/gsl_vector.h>
#include <gsl/gsl_matrix.h>
+#include "xalloc.h"
+
#define DM_COLUMN_NOT_FOUND -1
#define DM_INDEX_NOT_FOUND -3
dm = xmalloc (sizeof *dm);
dm->vars = xnmalloc (n_variables, sizeof *dm->vars);
+ dm->n_cases = xnmalloc (n_variables, sizeof (*dm->n_cases));
dm->n_vars = n_variables;
for (i = 0; i < n_variables; i++)
{
+ dm->n_cases[i] = 0;
v = v_variables[i];
assert ((dm->vars + i) != NULL);
(dm->vars + i)->v = v; /* Allows us to look up the variable from
dm->m = gsl_matrix_calloc (n_data, n_cols);
col = 0;
+
return dm;
}
{
free (dm->vars);
gsl_matrix_free (dm->m);
+ free (dm->n_cases);
free (dm);
}
assert (col != DM_COLUMN_NOT_FOUND);
gsl_matrix_set (dm->m, row, col, val->f);
}
+
+struct design_matrix *
+design_matrix_clone (const struct design_matrix *dm)
+{
+ struct design_matrix *result;
+ size_t i;
+
+ assert (dm != NULL);
+ result = xmalloc (sizeof *result);
+ result->vars = xnmalloc (dm->n_vars, sizeof *dm->vars);
+ result->n_vars = dm->n_vars;
+ result->m = gsl_matrix_alloc (dm->m->size1, dm->m->size2);
+
+ gsl_matrix_memcpy (result->m, dm->m);
+ for (i = 0; i < result->n_vars; i++)
+ {
+ result->vars[i] = dm->vars[i];
+ }
+ return result;
+}
+
+/*
+ Increment the number of cases for V.
+ */
+void
+design_matrix_increment_case_count (struct design_matrix *dm, const struct variable *v)
+{
+ size_t i;
+ assert (dm != NULL);
+ assert (dm->n_cases != NULL);
+ assert (v != NULL);
+ i = design_matrix_var_to_column (dm, v);
+ dm->n_cases[i]++;
+}
+
+/*
+ Set the number of cases for V.
+ */
+void
+design_matrix_set_case_count (struct design_matrix *dm, const struct variable *v, size_t n)
+{
+ size_t i;
+ assert (dm != NULL);
+ assert (dm->n_cases != NULL);
+ assert (v != NULL);
+ i = design_matrix_var_to_column (dm, v);
+ dm->n_cases[i] = n;
+}
+
+/*
+ Get the number of cases for V.
+ */
+size_t
+design_matrix_get_case_count (const struct design_matrix *dm, const struct variable *v)
+{
+ size_t i;
+ assert (dm != NULL);
+ assert (dm->n_cases != NULL);
+ assert (v != NULL);
+ i = design_matrix_var_to_column (dm, v);
+ return dm->n_cases[i];
+}
+
+