X-Git-Url: https://pintos-os.org/cgi-bin/gitweb.cgi?a=blobdiff_plain;ds=sidebyside;f=src%2Fmath%2Fcoefficient.c;h=fed45b64675cd3e763775ad1ac38838d7325addb;hb=b5b474193e450bba97610065df0518c08074a7fb;hp=eb2ee3174e42721e9b74147807453dc09f64caab;hpb=b57bac17a6057cc0642e6ecea792e058596f92bb;p=pspp-builds.git diff --git a/src/math/coefficient.c b/src/math/coefficient.c index eb2ee317..fed45b64 100644 --- a/src/math/coefficient.c +++ b/src/math/coefficient.c @@ -1,28 +1,24 @@ -/* - src/math/coefficient.c - - Copyright (C) 2005 Free Software Foundation, Inc. Written by Jason H Stover. - - 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 02111-1307, USA. -*/ +/* PSPP - a program for statistical analysis. + Copyright (C) 2005 Free Software Foundation, Inc. + + 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, see . */ /* Accessor functions for matching coefficients and variables. */ +#include #include -#include #include "src/math/design-matrix.h" #include @@ -50,7 +46,7 @@ pspp_coeff_free (struct pspp_coeff *c) /* Initialize the variable and value pointers inside the - coefficient structures for the linear model. + coefficient structures for the model. */ void pspp_coeff_init (struct pspp_coeff ** c, const struct design_matrix *X) @@ -58,27 +54,27 @@ pspp_coeff_init (struct pspp_coeff ** c, const struct design_matrix *X) size_t i; int n_vals = 1; + assert (c != NULL); for (i = 0; i < X->m->size2; i++) { c[i] = xmalloc (sizeof (*c[i])); c[i]->n_vars = n_vals; /* Currently, no procedures allow interactions. This line will have to change when procedures that allow - interaction terms are written. + interaction terms are written. */ c[i]->v_info = xnmalloc (c[i]->n_vars, sizeof (*c[i]->v_info)); assert (c[i]->v_info != NULL); - c[i]->v_info->v = - (const struct variable *) design_matrix_col_to_var (X, i); + c[i]->v_info->v = design_matrix_col_to_var (X, i); - if (c[i]->v_info->v->type == ALPHA) + if (var_is_alpha (c[i]->v_info->v)) { size_t k; k = design_matrix_var_to_column (X, c[i]->v_info->v); assert (k <= i); k = i - k; c[i]->v_info->val = - cat_subscript_to_value (k, (struct variable *) c[i]->v_info->v); + cat_subscript_to_value (k, c[i]->v_info->v); } } } @@ -162,14 +158,14 @@ pspp_coeff_get_value (struct pspp_coeff *c, { return NULL; } - if (v->type == NUMERIC) + if (var_is_numeric (v)) { return NULL; } while (i < c->n_vars) { candidate = pspp_coeff_get_var (c, i); - if (v->index == candidate->index) + if (v == candidate) { return (c->v_info + i)->val; } @@ -178,62 +174,3 @@ pspp_coeff_get_value (struct pspp_coeff *c, return NULL; } -/* - Which coefficient is associated with V? The VAL argument is relevant - only to categorical variables. - */ -const struct pspp_coeff * -pspp_linreg_get_coeff (const pspp_linreg_cache * c, - const struct variable *v, const union value *val) -{ - int i = 1; - struct pspp_coeff *result = NULL; - const struct variable *tmp = NULL; - - if (c == NULL) - { - return NULL; - } - if (c->coeff == NULL || c->n_indeps == 0 || v == NULL) - { - return NULL; - } - - result = c->coeff[i]; - tmp = pspp_coeff_get_var (result, 0); - while (tmp->index != v->index && i < c->n_coeffs) - { - result = c->coeff[i]; - tmp = pspp_coeff_get_var (result, 0); - i++; - } - if (i > c->n_coeffs) - { - return NULL; - } - if (v->type == NUMERIC) - { - return result; - } - else if (val != NULL) - { - /* - If v is categorical, we need to ensure the coefficient - matches the VAL. - */ - while (tmp->index != v->index && i < c->n_coeffs - && compare_values (pspp_coeff_get_value (result, tmp), - val, v->width)) - { /* FIX THIS */ - i++; - result = c->coeff[i]; - tmp = pspp_coeff_get_var (result, 0); - } - if (i == c->n_coeffs) - { - return NULL; - } - return result; - } - return NULL; -}