X-Git-Url: https://pintos-os.org/cgi-bin/gitweb.cgi?a=blobdiff_plain;f=src%2Fmath%2Flinreg%2Flinreg.h;h=c5f5ef10ba24e944e91f548b37bf01621a85358a;hb=13d1b533c6ce0545a4b06e5e30c73211591034c5;hp=67a422ea3fcaed0ab7497606b7f0920929de9496;hpb=e08ba3a0b4be7ebf4f049fa4ead1f0f1508bd7f8;p=pspp-builds.git

diff --git a/src/math/linreg/linreg.h b/src/math/linreg/linreg.h
index 67a422ea..c5f5ef10 100644
--- a/src/math/linreg/linreg.h
+++ b/src/math/linreg/linreg.h
@@ -26,7 +26,7 @@
 #include <gsl/gsl_matrix.h>
 
 struct variable;
-struct pspp_linreg_coeff;
+struct pspp_coeff;
 union value;
 
 enum
@@ -42,8 +42,6 @@ enum
  */
 struct pspp_linreg_opts_struct
 {
-  int resid;			/* Should the residuals be returned? */
-
   int get_depvar_mean_std;
   int *get_indep_mean_std;	/* Array of booleans
 				   dictating which
@@ -103,13 +101,13 @@ struct pspp_linreg_cache_struct
   int n_coeffs;
 
   /*
-    The variable struct is ignored during estimation. It is here so
-    the calling procedure can find the variable used in the model.
+     The variable struct is ignored during estimation. It is here so
+     the calling procedure can find the variable used in the model.
    */
   const struct variable *depvar;
 
   gsl_vector *residuals;
-  struct pspp_linreg_coeff *coeff;
+  struct pspp_coeff **coeff;
   int method;			/* Method to use to estimate parameters. */
   /*
      Means and standard deviations of the variables.
@@ -141,7 +139,7 @@ struct pspp_linreg_cache_struct
 				   variables, i.e. \sum (x[i] - mean(x))^2. */
   double ssy;			/* Centered sums of squares for dependent
 				   variable. 
-				*/
+				 */
   /*
      Covariance matrix of the parameter estimates.
    */
@@ -163,8 +161,14 @@ struct pspp_linreg_cache_struct
 		     const void *, int);
   double (*residual) (const struct variable **,
 		      const union value **,
-		      const union value *,
-		      const void *, int);
+		      const union value *, const void *, int);
+  /*
+     Returns pointers to the variables used in the model.
+   */
+  int (*get_vars) (const void *, struct variable **);
+  struct variable *resid;
+  struct variable *pred;
+
 };
 
 typedef struct pspp_linreg_cache_struct pspp_linreg_cache;
@@ -178,7 +182,7 @@ typedef struct pspp_linreg_cache_struct pspp_linreg_cache;
  */
 pspp_linreg_cache *pspp_linreg_cache_alloc (size_t n, size_t p);
 
-bool pspp_linreg_cache_free (pspp_linreg_cache * c);
+bool pspp_linreg_cache_free (void *);
 
 /*
   Fit the linear model via least squares. All pointers passed to pspp_linreg
@@ -195,4 +199,8 @@ pspp_linreg_predict (const struct variable **, const union value **,
 double
 pspp_linreg_residual (const struct variable **, const union value **,
 		      const union value *, const void *, int);
+/*
+  All variables used in the model.
+ */
+int pspp_linreg_get_vars (const void *, struct variable **);
 #endif