lib/linreg/coefficient.c

   1 /* lib/linreg/coefficient.c
   2
   3    Copyright (C) 2005 Free Software Foundation, Inc.
   4    Written by Jason H Stover.
   5
   6    This program is free software; you can redistribute it and/or modify
   7    it under the terms of the GNU General Public License as published by
   8    the Free Software Foundation; either version 2 of the License, or (at
   9    your option) any later version.
  10
  11    This program is distributed in the hope that it will be useful, but
  12    WITHOUT ANY WARRANTY; without even the implied warranty of
  13    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  14    General Public License for more details.
  15
  16    You should have received a copy of the GNU General Public License
  17    along with this program; if not, write to the Free Software
  18    Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA
  19    02111-1307, USA.
  20  */
  21
  22 /*
  23   Accessor functions for matching coefficients and variables.
  24  */
  25 #include <assert.h>
  26 #include "pspp_linreg.h"
  27 #include <gl/xalloc.h>
  28
  29
  30 struct varinfo
  31 {
  32   const struct variable *v; /* Variable associated with this
  33                                coefficient. Note this variable may not
  34                                be unique. In other words, a
  35                                coefficient structure may have other
  36                                v_info's, each with its own variable.
  37                              */
  38   const union value *val; /* Value of the variable v which this
  39                              varinfo refers to. This member is relevant
  40                              only to categorical variables.
  41                           */
  42 };
  43
  44 void pspp_linreg_coeff_free (struct pspp_linreg_coeff *c)
  45 {
  46   free (c);
  47 }
  48
  49 /*
  50   Initialize the variable and value pointers inside the
  51   coefficient structures for the linear model.
  52  */
  53 void
  54 pspp_linreg_coeff_init (pspp_linreg_cache *c, struct design_matrix *X)
  55 {
  56   size_t i;
  57   size_t j;
  58   int n_vals = 1;
  59   struct pspp_linreg_coeff *coeff;
  60
  61   c->coeff = xnmalloc (X->m->size2 + 1, sizeof (*c->coeff));
  62   for (i = 0; i < X->m->size2; i++)
  63     {
  64       j = i + 1;                /* The first coefficient is the intercept. */
  65       coeff = c->coeff + j;
  66       coeff->n_vars = n_vals; /* Currently, no procedures allow interactions.
  67                                  This will have to change when procedures that
  68                                  allow interaction terms are written.
  69                               */
  70       coeff->v_info = xnmalloc (coeff->n_vars, sizeof (*coeff->v_info));
  71       assert (coeff->v_info != NULL);
  72       coeff->v_info->v = (const struct variable *) design_matrix_col_to_var (X, i);
  73
  74       if (coeff->v_info->v->type == ALPHA)
  75         {
  76           size_t k;
  77           k = design_matrix_var_to_column (X, coeff->v_info->v);
  78           assert (k <= i);
  79           k = i - k;
  80           coeff->v_info->val = cat_subscript_to_value (k, (struct variable *) coeff->v_info->v);
  81         }
  82     }
  83 }
  84 void
  85 pspp_linreg_coeff_set_estimate (struct pspp_linreg_coeff *c,
  86                                 double estimate)
  87 {
  88   c->estimate = estimate;
  89 }
  90 void
  91 pspp_linreg_coeff_set_std_err (struct pspp_linreg_coeff *c,
  92                                double std_err)
  93 {
  94   c->std_err = std_err;
  95 }
  96 /*
  97   How many variables are associated with this coefficient?
  98  */
  99 int
 100 pspp_linreg_coeff_get_n_vars (struct pspp_linreg_coeff *c)
 101 {
 102   return c->n_vars;
 103 }
 104 /*
 105   Which variable does this coefficient match?
 106  */
 107 const struct variable *
 108 pspp_linreg_coeff_get_var (struct pspp_linreg_coeff *c, int i)
 109 {
 110   assert (i < c->n_vars);
 111   return (c->v_info + i)->v;
 112 }
 113 /*
 114    Which value is associated with this coefficient/variable comination?
 115 */
 116 const union value *
 117 pspp_linreg_coeff_get_value (struct pspp_linreg_coeff *c,
 118                              const struct variable *v)
 119 {
 120   int i = 0;
 121   const struct variable *candidate;
 122
 123   while (i < c->n_vars)
 124     {
 125       candidate = pspp_linreg_coeff_get_var (c, i);
 126       if (v->index == candidate->index)
 127         {
 128           return (c->v_info + i)->val;
 129         }
 130       i++;
 131     }
 132   return NULL;
 133 }