--- /dev/null
+/* PSPP - a program for statistical analysis.
+ Copyright (C) 1997-9, 2000, 2007, 2009, 2010 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 <http://www.gnu.org/licenses/>. */
+
+#include <config.h>
+
+#include <data/case.h>
+#include <data/casegrouper.h>
+#include <data/casereader.h>
+
+#include <libpspp/ll.h>
+
+#include <language/lexer/lexer.h>
+#include <language/lexer/variable-parser.h>
+#include <language/lexer/value-parser.h>
+#include <language/command.h>
+
+#include <data/procedure.h>
+#include <data/dictionary.h>
+
+
+#include <language/dictionary/split-file.h>
+#include <libpspp/hash.h>
+#include <libpspp/taint.h>
+#include <math/group-proc.h>
+#include <math/levene.h>
+#include <libpspp/misc.h>
+
+#include <output/tab.h>
+
+#include <gsl/gsl_cdf.h>
+#include <math.h>
+#include <data/format.h>
+
+#include <libpspp/message.h>
+
+#include "gettext.h"
+#define _(msgid) gettext (msgid)
+
+enum missing_type
+ {
+ MISS_LISTWISE,
+ MISS_ANALYSIS,
+ };
+
+enum statistics
+ {
+ STATS_DESCRIPTIVES = 0x0001,
+ STATS_HOMOGENEITY = 0x0002
+ };
+
+struct coeff_node
+{
+ struct ll ll;
+ double coeff;
+};
+
+
+struct contrasts_node
+{
+ struct ll ll;
+ struct ll_list coefficient_list;
+
+ bool bad_count; /* True if the number of coefficients does not equal the number of groups */
+};
+
+struct oneway
+{
+ size_t n_vars;
+ const struct variable **vars;
+
+ const struct variable *indep_var;
+
+ enum statistics stats;
+
+ enum missing_type missing_type;
+ enum mv_class exclude;
+
+ /* The number of distinct values of the independent variable, when all
+ missing values are disregarded */
+ int actual_number_of_groups;
+
+ /* A hash table containing all the distinct values of the independent
+ variable */
+ struct hsh_table *group_hash;
+
+ /* List of contrasts */
+ struct ll_list contrast_list;
+};
+
+/* Routines to show the output tables */
+static void show_anova_table (const struct oneway *);
+static void show_descriptives (const struct oneway *, const struct dictionary *dict);
+static void show_homogeneity (const struct oneway *);
+
+static void output_oneway (const struct oneway *, const struct dictionary *dict);
+static void run_oneway (struct oneway *cmd, struct casereader *input, const struct dataset *ds);
+
+int
+cmd_oneway (struct lexer *lexer, struct dataset *ds)
+{
+ const struct dictionary *dict = dataset_dict (ds);
+
+ struct oneway oneway ;
+ oneway.n_vars = 0;
+ oneway.vars = NULL;
+ oneway.indep_var = NULL;
+ oneway.stats = 0;
+ oneway.missing_type = MISS_ANALYSIS;
+ oneway.exclude = MV_ANY;
+ oneway.actual_number_of_groups = 0;
+ oneway.group_hash = NULL;
+
+ ll_init (&oneway.contrast_list);
+
+
+ if ( lex_match (lexer, '/'))
+ {
+ if (!lex_force_match_id (lexer, "VARIABLES"))
+ {
+ goto error;
+ }
+ lex_match (lexer, '=');
+ }
+
+ if (!parse_variables_const (lexer, dict,
+ &oneway.vars, &oneway.n_vars,
+ PV_NO_DUPLICATE | PV_NUMERIC))
+ goto error;
+
+ lex_force_match (lexer, T_BY);
+
+ oneway.indep_var = parse_variable_const (lexer, dict);
+
+ while (lex_token (lexer) != '.')
+ {
+ lex_match (lexer, '/');
+
+ if (lex_match_id (lexer, "STATISTICS"))
+ {
+ lex_match (lexer, '=');
+ while (lex_token (lexer) != '.' && lex_token (lexer) != '/')
+ {
+ if (lex_match_id (lexer, "DESCRIPTIVES"))
+ {
+ oneway.stats |= STATS_DESCRIPTIVES;
+ }
+ else if (lex_match_id (lexer, "HOMOGENEITY"))
+ {
+ oneway.stats |= STATS_HOMOGENEITY;
+ }
+ else
+ {
+ lex_error (lexer, NULL);
+ goto error;
+ }
+ }
+ }
+ else if (lex_match_id (lexer, "CONTRAST"))
+ {
+ struct contrasts_node *cl = xzalloc (sizeof *cl);
+
+ struct ll_list *coefficient_list = &cl->coefficient_list;
+ lex_match (lexer, '=');
+
+ ll_init (coefficient_list);
+
+ while (lex_token (lexer) != '.' && lex_token (lexer) != '/')
+ {
+ union value val;
+ if ( parse_value (lexer, &val, 0))
+ {
+ struct coeff_node *cc = xmalloc (sizeof *cc);
+ cc->coeff = val.f;
+
+ ll_push_tail (coefficient_list, &cc->ll);
+ }
+ else
+ {
+ lex_error (lexer, NULL);
+ goto error;
+ }
+ }
+
+ ll_push_tail (&oneway.contrast_list, &cl->ll);
+ }
+ else if (lex_match_id (lexer, "MISSING"))
+ {
+ lex_match (lexer, '=');
+ while (lex_token (lexer) != '.' && lex_token (lexer) != '/')
+ {
+ if (lex_match_id (lexer, "INCLUDE"))
+ {
+ oneway.exclude = MV_SYSTEM;
+ }
+ else if (lex_match_id (lexer, "EXCLUDE"))
+ {
+ oneway.exclude = MV_ANY;
+ }
+ else if (lex_match_id (lexer, "LISTWISE"))
+ {
+ oneway.missing_type = MISS_LISTWISE;
+ }
+ else if (lex_match_id (lexer, "ANALYSIS"))
+ {
+ oneway.missing_type = MISS_ANALYSIS;
+ }
+ else
+ {
+ lex_error (lexer, NULL);
+ goto error;
+ }
+ }
+ }
+ else
+ {
+ lex_error (lexer, NULL);
+ goto error;
+ }
+ }
+
+
+ {
+ struct casegrouper *grouper;
+ struct casereader *group;
+ bool ok;
+
+ grouper = casegrouper_create_splits (proc_open (ds), dict);
+ while (casegrouper_get_next_group (grouper, &group))
+ run_oneway (&oneway, group, ds);
+ ok = casegrouper_destroy (grouper);
+ ok = proc_commit (ds) && ok;
+ }
+
+ free (oneway.vars);
+ return CMD_SUCCESS;
+
+ error:
+ free (oneway.vars);
+ return CMD_FAILURE;
+}
+
+
+\f
+
+static int
+compare_double_3way (const void *a_, const void *b_, const void *aux UNUSED)
+{
+ const double *a = a_;
+ const double *b = b_;
+ return *a < *b ? -1 : *a > *b;
+}
+
+static unsigned
+do_hash_double (const void *value_, const void *aux UNUSED)
+{
+ const double *value = value_;
+ return hash_double (*value, 0);
+}
+
+static void
+free_double (void *value_, const void *aux UNUSED)
+{
+ double *value = value_;
+ free (value);
+}
+
+static void postcalc (const struct oneway *cmd);
+static void precalc (const struct oneway *cmd);
+
+
+static void
+run_oneway (struct oneway *cmd,
+ struct casereader *input,
+ const struct dataset *ds)
+{
+ struct taint *taint;
+ struct dictionary *dict = dataset_dict (ds);
+ struct casereader *reader;
+ struct ccase *c;
+
+ c = casereader_peek (input, 0);
+ if (c == NULL)
+ {
+ casereader_destroy (input);
+ return;
+ }
+ output_split_file_values (ds, c);
+ case_unref (c);
+
+ taint = taint_clone (casereader_get_taint (input));
+
+ cmd->group_hash = hsh_create (4,
+ compare_double_3way,
+ do_hash_double,
+ free_double,
+ cmd->indep_var);
+
+ precalc (cmd);
+
+ input = casereader_create_filter_missing (input, &cmd->indep_var, 1,
+ cmd->exclude, NULL, NULL);
+ if (cmd->missing_type == MISS_LISTWISE)
+ input = casereader_create_filter_missing (input, cmd->vars, cmd->n_vars,
+ cmd->exclude, NULL, NULL);
+ input = casereader_create_filter_weight (input, dict, NULL, NULL);
+
+ reader = casereader_clone (input);
+ for (; (c = casereader_read (reader)) != NULL; case_unref (c))
+ {
+ size_t i;
+
+ const double weight = dict_get_case_weight (dict, c, NULL);
+
+ const union value *indep_val = case_data (c, cmd->indep_var);
+ void **p = hsh_probe (cmd->group_hash, &indep_val->f);
+ if (*p == NULL)
+ {
+ double *value = *p = xmalloc (sizeof *value);
+ *value = indep_val->f;
+ }
+
+ for (i = 0; i < cmd->n_vars; ++i)
+ {
+ const struct variable *v = cmd->vars[i];
+
+ const union value *val = case_data (c, v);
+
+ struct group_proc *gp = group_proc_get (cmd->vars[i]);
+ struct hsh_table *group_hash = gp->group_hash;
+
+ struct group_statistics *gs;
+
+ gs = hsh_find (group_hash, indep_val );
+
+ if ( ! gs )
+ {
+ gs = xmalloc (sizeof *gs);
+ gs->id = *indep_val;
+ gs->sum = 0;
+ gs->n = 0;
+ gs->ssq = 0;
+ gs->sum_diff = 0;
+ gs->minimum = DBL_MAX;
+ gs->maximum = -DBL_MAX;
+
+ hsh_insert ( group_hash, gs );
+ }
+
+ if (!var_is_value_missing (v, val, cmd->exclude))
+ {
+ struct group_statistics *totals = &gp->ugs;
+
+ totals->n += weight;
+ totals->sum += weight * val->f;
+ totals->ssq += weight * pow2 (val->f);
+
+ if ( val->f * weight < totals->minimum )
+ totals->minimum = val->f * weight;
+
+ if ( val->f * weight > totals->maximum )
+ totals->maximum = val->f * weight;
+
+ gs->n += weight;
+ gs->sum += weight * val->f;
+ gs->ssq += weight * pow2 (val->f);
+
+ if ( val->f * weight < gs->minimum )
+ gs->minimum = val->f * weight;
+
+ if ( val->f * weight > gs->maximum )
+ gs->maximum = val->f * weight;
+ }
+
+ gp->n_groups = hsh_count (group_hash );
+ }
+
+ }
+ casereader_destroy (reader);
+
+ postcalc (cmd);
+
+ if ( cmd->stats & STATS_HOMOGENEITY )
+ levene (dict, casereader_clone (input), cmd->indep_var,
+ cmd->n_vars, cmd->vars, cmd->exclude);
+
+ casereader_destroy (input);
+
+ cmd->actual_number_of_groups = hsh_count (cmd->group_hash);
+
+ if (!taint_has_tainted_successor (taint))
+ output_oneway (cmd, dict);
+
+ taint_destroy (taint);
+}
+
+/* Pre calculations */
+static void
+precalc (const struct oneway *cmd)
+{
+ size_t i = 0;
+
+ for (i = 0; i < cmd->n_vars; ++i)
+ {
+ struct group_proc *gp = group_proc_get (cmd->vars[i]);
+ struct group_statistics *totals = &gp->ugs;
+
+ /* Create a hash for each of the dependent variables.
+ The hash contains a group_statistics structure,
+ and is keyed by value of the independent variable */
+
+ gp->group_hash = hsh_create (4, compare_group, hash_group,
+ (hsh_free_func *) free_group,
+ cmd->indep_var);
+
+ totals->sum = 0;
+ totals->n = 0;
+ totals->ssq = 0;
+ totals->sum_diff = 0;
+ totals->maximum = -DBL_MAX;
+ totals->minimum = DBL_MAX;
+ }
+}
+
+/* Post calculations for the ONEWAY command */
+static void
+postcalc (const struct oneway *cmd)
+{
+ size_t i = 0;
+
+ for (i = 0; i < cmd->n_vars; ++i)
+ {
+ struct group_proc *gp = group_proc_get (cmd->vars[i]);
+ struct hsh_table *group_hash = gp->group_hash;
+ struct group_statistics *totals = &gp->ugs;
+
+ struct hsh_iterator g;
+ struct group_statistics *gs;
+
+ for (gs = hsh_first (group_hash, &g);
+ gs != 0;
+ gs = hsh_next (group_hash, &g))
+ {
+ gs->mean = gs->sum / gs->n;
+ gs->s_std_dev = sqrt (gs->ssq / gs->n - pow2 (gs->mean));
+
+ gs->std_dev = sqrt (
+ gs->n / (gs->n - 1) *
+ ( gs->ssq / gs->n - pow2 (gs->mean))
+ );
+
+ gs->se_mean = gs->std_dev / sqrt (gs->n);
+ gs->mean_diff = gs->sum_diff / gs->n;
+ }
+
+ totals->mean = totals->sum / totals->n;
+ totals->std_dev = sqrt (
+ totals->n / (totals->n - 1) *
+ (totals->ssq / totals->n - pow2 (totals->mean))
+ );
+
+ totals->se_mean = totals->std_dev / sqrt (totals->n);
+ }
+}
+
+static void show_contrast_coeffs (const struct oneway *cmd);
+static void show_contrast_tests (const struct oneway *cmd);
+
+static void
+output_oneway (const struct oneway *cmd, const struct dictionary *dict)
+{
+ size_t i = 0;
+
+ /* Check the sanity of the given contrast values */
+ struct contrasts_node *coeff_list = NULL;
+ ll_for_each (coeff_list, struct contrasts_node, ll, &cmd->contrast_list)
+ {
+ struct coeff_node *cn = NULL;
+ double sum = 0;
+ struct ll_list *cl = &coeff_list->coefficient_list;
+ ++i;
+
+ if (ll_count (cl) != cmd->actual_number_of_groups)
+ {
+ msg (SW,
+ _("Number of contrast coefficients must equal the number of groups"));
+ coeff_list->bad_count = true;
+ continue;
+ }
+
+ ll_for_each (cn, struct coeff_node, ll, cl)
+ sum += cn->coeff;
+
+ if ( sum != 0.0 )
+ msg (SW, _("Coefficients for contrast %zu do not total zero"), i);
+ }
+
+ if (cmd->stats & STATS_DESCRIPTIVES)
+ show_descriptives (cmd, dict);
+
+ if (cmd->stats & STATS_HOMOGENEITY)
+ show_homogeneity (cmd);
+
+ show_anova_table (cmd);
+
+
+ if (ll_count (&cmd->contrast_list) > 0)
+ {
+ show_contrast_coeffs (cmd);
+ show_contrast_tests (cmd);
+ }
+
+
+ /* Clean up */
+ for (i = 0; i < cmd->n_vars; ++i )
+ {
+ struct hsh_table *group_hash = group_proc_get (cmd->vars[i])->group_hash;
+
+ hsh_destroy (group_hash);
+ }
+
+ hsh_destroy (cmd->group_hash);
+}
+
+
+/* Show the ANOVA table */
+static void
+show_anova_table (const struct oneway *cmd)
+{
+ size_t i;
+ int n_cols =7;
+ size_t n_rows = cmd->n_vars * 3 + 1;
+
+ struct tab_table *t = tab_create (n_cols, n_rows);
+
+ tab_headers (t, 2, 0, 1, 0);
+
+ tab_box (t,
+ TAL_2, TAL_2,
+ -1, TAL_1,
+ 0, 0,
+ n_cols - 1, n_rows - 1);
+
+ tab_hline (t, TAL_2, 0, n_cols - 1, 1 );
+ tab_vline (t, TAL_2, 2, 0, n_rows - 1);
+ tab_vline (t, TAL_0, 1, 0, 0);
+
+ tab_text (t, 2, 0, TAB_CENTER | TAT_TITLE, _("Sum of Squares"));
+ tab_text (t, 3, 0, TAB_CENTER | TAT_TITLE, _("df"));
+ tab_text (t, 4, 0, TAB_CENTER | TAT_TITLE, _("Mean Square"));
+ tab_text (t, 5, 0, TAB_CENTER | TAT_TITLE, _("F"));
+ tab_text (t, 6, 0, TAB_CENTER | TAT_TITLE, _("Significance"));
+
+
+ for (i = 0; i < cmd->n_vars; ++i)
+ {
+ struct group_statistics *totals = &group_proc_get (cmd->vars[i])->ugs;
+ struct hsh_table *group_hash = group_proc_get (cmd->vars[i])->group_hash;
+ struct hsh_iterator g;
+ struct group_statistics *gs;
+ double ssa = 0;
+ const char *s = var_to_string (cmd->vars[i]);
+
+ for (gs = hsh_first (group_hash, &g);
+ gs != 0;
+ gs = hsh_next (group_hash, &g))
+ {
+ ssa += pow2 (gs->sum) / gs->n;
+ }
+
+ ssa -= pow2 (totals->sum) / totals->n;
+
+ tab_text (t, 0, i * 3 + 1, TAB_LEFT | TAT_TITLE, s);
+ tab_text (t, 1, i * 3 + 1, TAB_LEFT | TAT_TITLE, _("Between Groups"));
+ tab_text (t, 1, i * 3 + 2, TAB_LEFT | TAT_TITLE, _("Within Groups"));
+ tab_text (t, 1, i * 3 + 3, TAB_LEFT | TAT_TITLE, _("Total"));
+
+ if (i > 0)
+ tab_hline (t, TAL_1, 0, n_cols - 1, i * 3 + 1);
+
+ {
+ struct group_proc *gp = group_proc_get (cmd->vars[i]);
+ const double sst = totals->ssq - pow2 (totals->sum) / totals->n;
+ const double df1 = gp->n_groups - 1;
+ const double df2 = totals->n - gp->n_groups;
+ const double msa = ssa / df1;
+
+ gp->mse = (sst - ssa) / df2;
+
+
+ /* Sums of Squares */
+ tab_double (t, 2, i * 3 + 1, 0, ssa, NULL);
+ tab_double (t, 2, i * 3 + 3, 0, sst, NULL);
+ tab_double (t, 2, i * 3 + 2, 0, sst - ssa, NULL);
+
+
+ /* Degrees of freedom */
+ tab_fixed (t, 3, i * 3 + 1, 0, df1, 4, 0);
+ tab_fixed (t, 3, i * 3 + 2, 0, df2, 4, 0);
+ tab_fixed (t, 3, i * 3 + 3, 0, totals->n - 1, 4, 0);
+
+ /* Mean Squares */
+ tab_double (t, 4, i * 3 + 1, TAB_RIGHT, msa, NULL);
+ tab_double (t, 4, i * 3 + 2, TAB_RIGHT, gp->mse, NULL);
+
+ {
+ const double F = msa / gp->mse ;
+
+ /* The F value */
+ tab_double (t, 5, i * 3 + 1, 0, F, NULL);
+
+ /* The significance */
+ tab_double (t, 6, i * 3 + 1, 0, gsl_cdf_fdist_Q (F, df1, df2), NULL);
+ }
+ }
+ }
+
+
+ tab_title (t, _("ANOVA"));
+ tab_submit (t);
+}
+
+
+/* Show the descriptives table */
+static void
+show_descriptives (const struct oneway *cmd, const struct dictionary *dict)
+{
+ size_t v;
+ int n_cols = 10;
+ struct tab_table *t;
+ int row;
+
+ const double confidence = 0.95;
+ const double q = (1.0 - confidence) / 2.0;
+
+ const struct variable *wv = dict_get_weight (dict);
+ const struct fmt_spec *wfmt = wv ? var_get_print_format (wv) : & F_8_0;
+
+ int n_rows = 2;
+
+ for ( v = 0; v < cmd->n_vars; ++v )
+ n_rows += group_proc_get (cmd->vars[v])->n_groups + 1;
+
+ t = tab_create (n_cols, n_rows);
+ tab_headers (t, 2, 0, 2, 0);
+
+
+ /* Put a frame around the entire box, and vertical lines inside */
+ tab_box (t,
+ TAL_2, TAL_2,
+ -1, TAL_1,
+ 0, 0,
+ n_cols - 1, n_rows - 1);
+
+ /* Underline headers */
+ tab_hline (t, TAL_2, 0, n_cols - 1, 2);
+ tab_vline (t, TAL_2, 2, 0, n_rows - 1);
+
+ tab_text (t, 2, 1, TAB_CENTER | TAT_TITLE, _("N"));
+ tab_text (t, 3, 1, TAB_CENTER | TAT_TITLE, _("Mean"));
+ tab_text (t, 4, 1, TAB_CENTER | TAT_TITLE, _("Std. Deviation"));
+ tab_text (t, 5, 1, TAB_CENTER | TAT_TITLE, _("Std. Error"));
+
+
+ tab_vline (t, TAL_0, 7, 0, 0);
+ tab_hline (t, TAL_1, 6, 7, 1);
+ tab_joint_text_format (t, 6, 0, 7, 0, TAB_CENTER | TAT_TITLE,
+ _("%g%% Confidence Interval for Mean"),
+ confidence*100.0);
+
+ tab_text (t, 6, 1, TAB_CENTER | TAT_TITLE, _("Lower Bound"));
+ tab_text (t, 7, 1, TAB_CENTER | TAT_TITLE, _("Upper Bound"));
+
+ tab_text (t, 8, 1, TAB_CENTER | TAT_TITLE, _("Minimum"));
+ tab_text (t, 9, 1, TAB_CENTER | TAT_TITLE, _("Maximum"));
+
+
+ tab_title (t, _("Descriptives"));
+
+
+ row = 2;
+ for (v = 0; v < cmd->n_vars; ++v)
+ {
+ double T;
+ double std_error;
+
+ struct group_proc *gp = group_proc_get (cmd->vars[v]);
+
+ struct group_statistics *gs;
+ struct group_statistics *totals = &gp->ugs;
+
+ const char *s = var_to_string (cmd->vars[v]);
+ const struct fmt_spec *fmt = var_get_print_format (cmd->vars[v]);
+
+ struct group_statistics *const *gs_array =
+ (struct group_statistics *const *) hsh_sort (gp->group_hash);
+ int count = 0;
+
+ tab_text (t, 0, row, TAB_LEFT | TAT_TITLE, s);
+ if ( v > 0)
+ tab_hline (t, TAL_1, 0, n_cols - 1, row);
+
+ for (count = 0; count < hsh_count (gp->group_hash); ++count)
+ {
+ struct string vstr;
+ ds_init_empty (&vstr);
+ gs = gs_array[count];
+
+ var_append_value_name (cmd->indep_var, &gs->id, &vstr);
+
+ tab_text (t, 1, row + count,
+ TAB_LEFT | TAT_TITLE,
+ ds_cstr (&vstr));
+
+ ds_destroy (&vstr);
+
+ /* Now fill in the numbers ... */
+
+ tab_fixed (t, 2, row + count, 0, gs->n, 8, 0);
+
+ tab_double (t, 3, row + count, 0, gs->mean, NULL);
+
+ tab_double (t, 4, row + count, 0, gs->std_dev, NULL);
+
+ std_error = gs->std_dev / sqrt (gs->n) ;
+ tab_double (t, 5, row + count, 0,
+ std_error, NULL);
+
+ /* Now the confidence interval */
+
+ T = gsl_cdf_tdist_Qinv (q, gs->n - 1);
+
+ tab_double (t, 6, row + count, 0,
+ gs->mean - T * std_error, NULL);
+
+ tab_double (t, 7, row + count, 0,
+ gs->mean + T * std_error, NULL);
+
+ /* Min and Max */
+
+ tab_double (t, 8, row + count, 0, gs->minimum, fmt);
+ tab_double (t, 9, row + count, 0, gs->maximum, fmt);
+ }
+
+ tab_text (t, 1, row + count,
+ TAB_LEFT | TAT_TITLE, _("Total"));
+
+ tab_double (t, 2, row + count, 0, totals->n, wfmt);
+
+ tab_double (t, 3, row + count, 0, totals->mean, NULL);
+
+ tab_double (t, 4, row + count, 0, totals->std_dev, NULL);
+
+ std_error = totals->std_dev / sqrt (totals->n) ;
+
+ tab_double (t, 5, row + count, 0, std_error, NULL);
+
+ /* Now the confidence interval */
+
+ T = gsl_cdf_tdist_Qinv (q, totals->n - 1);
+
+ tab_double (t, 6, row + count, 0,
+ totals->mean - T * std_error, NULL);
+
+ tab_double (t, 7, row + count, 0,
+ totals->mean + T * std_error, NULL);
+
+ /* Min and Max */
+
+ tab_double (t, 8, row + count, 0, totals->minimum, fmt);
+ tab_double (t, 9, row + count, 0, totals->maximum, fmt);
+
+ row += gp->n_groups + 1;
+ }
+
+ tab_submit (t);
+}
+
+/* Show the homogeneity table */
+static void
+show_homogeneity (const struct oneway *cmd)
+{
+ size_t v;
+ int n_cols = 5;
+ size_t n_rows = cmd->n_vars + 1;
+
+ struct tab_table *t;
+
+
+ t = tab_create (n_cols, n_rows);
+ tab_headers (t, 1, 0, 1, 0);
+
+
+ /* Put a frame around the entire box, and vertical lines inside */
+ tab_box (t,
+ TAL_2, TAL_2,
+ -1, TAL_1,
+ 0, 0,
+ n_cols - 1, n_rows - 1);
+
+
+ tab_hline (t, TAL_2, 0, n_cols - 1, 1);
+ tab_vline (t, TAL_2, 1, 0, n_rows - 1);
+
+
+ tab_text (t, 1, 0, TAB_CENTER | TAT_TITLE, _("Levene Statistic"));
+ tab_text (t, 2, 0, TAB_CENTER | TAT_TITLE, _("df1"));
+ tab_text (t, 3, 0, TAB_CENTER | TAT_TITLE, _("df2"));
+ tab_text (t, 4, 0, TAB_CENTER | TAT_TITLE, _("Significance"));
+
+ tab_title (t, _("Test of Homogeneity of Variances"));
+
+ for (v = 0; v < cmd->n_vars; ++v)
+ {
+ double F;
+ const struct variable *var = cmd->vars[v];
+ const struct group_proc *gp = group_proc_get (cmd->vars[v]);
+ const char *s = var_to_string (var);
+ const struct group_statistics *totals = &gp->ugs;
+
+ const double df1 = gp->n_groups - 1;
+ const double df2 = totals->n - gp->n_groups;
+
+ tab_text (t, 0, v + 1, TAB_LEFT | TAT_TITLE, s);
+
+ F = gp->levene;
+ tab_double (t, 1, v + 1, TAB_RIGHT, F, NULL);
+ tab_fixed (t, 2, v + 1, TAB_RIGHT, df1, 8, 0);
+ tab_fixed (t, 3, v + 1, TAB_RIGHT, df2, 8, 0);
+
+ /* Now the significance */
+ tab_double (t, 4, v + 1, TAB_RIGHT,gsl_cdf_fdist_Q (F, df1, df2), NULL);
+ }
+
+ tab_submit (t);
+}
+
+
+/* Show the contrast coefficients table */
+static void
+show_contrast_coeffs (const struct oneway *cmd)
+{
+ int c_num = 0;
+ struct ll *cli;
+
+ int n_contrasts = ll_count (&cmd->contrast_list);
+ int n_cols = 2 + cmd->actual_number_of_groups;
+ int n_rows = 2 + n_contrasts;
+
+ void *const *group_values;
+
+ struct tab_table *t;
+
+ t = tab_create (n_cols, n_rows);
+ tab_headers (t, 2, 0, 2, 0);
+
+ /* Put a frame around the entire box, and vertical lines inside */
+ tab_box (t,
+ TAL_2, TAL_2,
+ -1, TAL_1,
+ 0, 0,
+ n_cols - 1, n_rows - 1);
+
+ tab_box (t,
+ -1, -1,
+ TAL_0, TAL_0,
+ 2, 0,
+ n_cols - 1, 0);
+
+ tab_box (t,
+ -1, -1,
+ TAL_0, TAL_0,
+ 0, 0,
+ 1, 1);
+
+ tab_hline (t, TAL_1, 2, n_cols - 1, 1);
+ tab_hline (t, TAL_2, 0, n_cols - 1, 2);
+
+ tab_vline (t, TAL_2, 2, 0, n_rows - 1);
+
+ tab_title (t, _("Contrast Coefficients"));
+
+ tab_text (t, 0, 2, TAB_LEFT | TAT_TITLE, _("Contrast"));
+
+
+ tab_joint_text (t, 2, 0, n_cols - 1, 0, TAB_CENTER | TAT_TITLE,
+ var_to_string (cmd->indep_var));
+
+ group_values = hsh_sort (cmd->group_hash);
+
+ for ( cli = ll_head (&cmd->contrast_list);
+ cli != ll_null (&cmd->contrast_list);
+ cli = ll_next (cli))
+ {
+ int count = 0;
+ struct contrasts_node *cn = ll_data (cli, struct contrasts_node, ll);
+ struct ll *coeffi = ll_head (&cn->coefficient_list);
+
+ tab_text_format (t, 1, c_num + 2, TAB_CENTER, "%d", c_num + 1);
+
+ for (count = 0;
+ count < hsh_count (cmd->group_hash) && coeffi != ll_null (&cn->coefficient_list);
+ ++count)
+ {
+ double *group_value_p;
+ union value group_value;
+ struct string vstr;
+
+ ds_init_empty (&vstr);
+
+ group_value_p = group_values[count];
+ group_value.f = *group_value_p;
+ var_append_value_name (cmd->indep_var, &group_value, &vstr);
+
+ tab_text (t, count + 2, 1, TAB_CENTER | TAT_TITLE,
+ ds_cstr (&vstr));
+
+ ds_destroy (&vstr);
+
+ if (cn->bad_count)
+ tab_text (t, count + 2, c_num + 2, TAB_RIGHT, "?" );
+ else
+ {
+ struct coeff_node *coeffn = ll_data (coeffi, struct coeff_node, ll);
+
+ tab_text_format (t, count + 2, c_num + 2, TAB_RIGHT, "%g", coeffn->coeff);
+ }
+
+ coeffi = ll_next (coeffi);
+ }
+ ++c_num;
+ }
+
+ tab_submit (t);
+}
+
+
+/* Show the results of the contrast tests */
+static void
+show_contrast_tests (const struct oneway *cmd)
+{
+ int n_contrasts = ll_count (&cmd->contrast_list);
+ size_t v;
+ int n_cols = 8;
+ size_t n_rows = 1 + cmd->n_vars * 2 * n_contrasts;
+
+ struct tab_table *t;
+
+ t = tab_create (n_cols, n_rows);
+ tab_headers (t, 3, 0, 1, 0);
+
+ /* Put a frame around the entire box, and vertical lines inside */
+ tab_box (t,
+ TAL_2, TAL_2,
+ -1, TAL_1,
+ 0, 0,
+ n_cols - 1, n_rows - 1);
+
+ tab_box (t,
+ -1, -1,
+ TAL_0, TAL_0,
+ 0, 0,
+ 2, 0);
+
+ tab_hline (t, TAL_2, 0, n_cols - 1, 1);
+ tab_vline (t, TAL_2, 3, 0, n_rows - 1);
+
+
+ tab_title (t, _("Contrast Tests"));
+
+ tab_text (t, 2, 0, TAB_CENTER | TAT_TITLE, _("Contrast"));
+ tab_text (t, 3, 0, TAB_CENTER | TAT_TITLE, _("Value of Contrast"));
+ tab_text (t, 4, 0, TAB_CENTER | TAT_TITLE, _("Std. Error"));
+ tab_text (t, 5, 0, TAB_CENTER | TAT_TITLE, _("t"));
+ tab_text (t, 6, 0, TAB_CENTER | TAT_TITLE, _("df"));
+ tab_text (t, 7, 0, TAB_CENTER | TAT_TITLE, _("Sig. (2-tailed)"));
+
+ for (v = 0; v < cmd->n_vars; ++v)
+ {
+ struct ll *cli;
+ int i = 0;
+ int lines_per_variable = 2 * n_contrasts;
+
+ tab_text (t, 0, (v * lines_per_variable) + 1, TAB_LEFT | TAT_TITLE,
+ var_to_string (cmd->vars[v]));
+
+ for ( cli = ll_head (&cmd->contrast_list);
+ cli != ll_null (&cmd->contrast_list);
+ ++i, cli = ll_next (cli))
+ {
+ struct contrasts_node *cn = ll_data (cli, struct contrasts_node, ll);
+ struct ll *coeffi = ll_head (&cn->coefficient_list);
+ int ci;
+ double contrast_value = 0.0;
+ double coef_msq = 0.0;
+ struct group_proc *grp_data = group_proc_get (cmd->vars[v]);
+ struct hsh_table *group_hash = grp_data->group_hash;
+
+ void *const *group_stat_array;
+
+ double T;
+ double std_error_contrast;
+ double df;
+ double sec_vneq = 0.0;
+
+ /* Note: The calculation of the degrees of freedom in the
+ "variances not equal" case is painfull!!
+ The following formula may help to understand it:
+ \frac{\left (\sum_{i=1}^k{c_i^2\frac{s_i^2}{n_i}}\right)^2}
+ {
+ \sum_{i=1}^k\left (
+ \frac{\left (c_i^2\frac{s_i^2}{n_i}\right)^2} {n_i-1}
+ \right)
+ }
+ */
+
+ double df_denominator = 0.0;
+ double df_numerator = 0.0;
+ if ( i == 0 )
+ {
+ tab_text (t, 1, (v * lines_per_variable) + i + 1,
+ TAB_LEFT | TAT_TITLE,
+ _("Assume equal variances"));
+
+ tab_text (t, 1, (v * lines_per_variable) + i + 1 + n_contrasts,
+ TAB_LEFT | TAT_TITLE,
+ _("Does not assume equal"));
+ }
+
+ tab_text_format (t, 2, (v * lines_per_variable) + i + 1,
+ TAB_CENTER | TAT_TITLE, "%d", i + 1);
+
+
+ tab_text_format (t, 2,
+ (v * lines_per_variable) + i + 1 + n_contrasts,
+ TAB_CENTER | TAT_TITLE, "%d", i + 1);
+
+ if (cn->bad_count)
+ continue;
+
+ group_stat_array = hsh_sort (group_hash);
+
+ for (ci = 0;
+ coeffi != ll_null (&cn->coefficient_list) &&
+ ci < hsh_count (group_hash);
+ ++ci, coeffi = ll_next (coeffi))
+ {
+ struct coeff_node *cn = ll_data (coeffi, struct coeff_node, ll);
+ const double coef = cn->coeff;
+ struct group_statistics *gs = group_stat_array[ci];
+
+ const double winv = pow2 (gs->std_dev) / gs->n;
+
+ contrast_value += coef * gs->mean;
+
+ coef_msq += (coef * coef) / gs->n;
+
+ sec_vneq += (coef * coef) * pow2 (gs->std_dev) /gs->n;
+
+ df_numerator += (coef * coef) * winv;
+ df_denominator += pow2((coef * coef) * winv) / (gs->n - 1);
+ }
+
+ sec_vneq = sqrt (sec_vneq);
+
+ df_numerator = pow2 (df_numerator);
+
+ tab_double (t, 3, (v * lines_per_variable) + i + 1,
+ TAB_RIGHT, contrast_value, NULL);
+
+ tab_double (t, 3, (v * lines_per_variable) + i + 1 +
+ n_contrasts,
+ TAB_RIGHT, contrast_value, NULL);
+
+ std_error_contrast = sqrt (grp_data->mse * coef_msq);
+
+ /* Std. Error */
+ tab_double (t, 4, (v * lines_per_variable) + i + 1,
+ TAB_RIGHT, std_error_contrast,
+ NULL);
+
+ T = fabs (contrast_value / std_error_contrast);
+
+ /* T Statistic */
+
+ tab_double (t, 5, (v * lines_per_variable) + i + 1,
+ TAB_RIGHT, T,
+ NULL);
+
+ df = grp_data->ugs.n - grp_data->n_groups;
+
+ /* Degrees of Freedom */
+ tab_fixed (t, 6, (v * lines_per_variable) + i + 1,
+ TAB_RIGHT, df,
+ 8, 0);
+
+
+ /* Significance TWO TAILED !!*/
+ tab_double (t, 7, (v * lines_per_variable) + i + 1,
+ TAB_RIGHT, 2 * gsl_cdf_tdist_Q (T, df),
+ NULL);
+
+ /* Now for the Variances NOT Equal case */
+
+ /* Std. Error */
+ tab_double (t, 4,
+ (v * lines_per_variable) + i + 1 + n_contrasts,
+ TAB_RIGHT, sec_vneq,
+ NULL);
+
+ T = contrast_value / sec_vneq;
+ tab_double (t, 5,
+ (v * lines_per_variable) + i + 1 + n_contrasts,
+ TAB_RIGHT, T,
+ NULL);
+
+ df = df_numerator / df_denominator;
+
+ tab_double (t, 6,
+ (v * lines_per_variable) + i + 1 + n_contrasts,
+ TAB_RIGHT, df,
+ NULL);
+
+ /* The Significance */
+ tab_double (t, 7, (v * lines_per_variable) + i + 1 + n_contrasts,
+ TAB_RIGHT, 2 * gsl_cdf_tdist_Q (T,df),
+ NULL);
+ }
+
+ if ( v > 0 )
+ tab_hline (t, TAL_1, 0, n_cols - 1, (v * lines_per_variable) + 1);
+ }
+
+ tab_submit (t);
+}
+
+
+++ /dev/null
-/* PSPP - a program for statistical analysis.
- Copyright (C) 1997-9, 2000, 2007, 2009 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 <http://www.gnu.org/licenses/>. */
-
-#include <config.h>
-
-#include <gsl/gsl_cdf.h>
-#include <math.h>
-#include <stdio.h>
-#include <stdlib.h>
-
-#include <data/case.h>
-#include <data/casegrouper.h>
-#include <data/casereader.h>
-#include <data/dictionary.h>
-#include <data/procedure.h>
-#include <data/value-labels.h>
-#include <data/variable.h>
-#include <language/command.h>
-#include <language/dictionary/split-file.h>
-#include <language/lexer/lexer.h>
-#include <libpspp/compiler.h>
-#include <libpspp/hash.h>
-#include <libpspp/message.h>
-#include <libpspp/misc.h>
-#include <libpspp/str.h>
-#include <libpspp/taint.h>
-#include <math/group-proc.h>
-#include <math/group.h>
-#include <math/levene.h>
-#include <output/tab.h>
-#include "sort-criteria.h"
-#include <data/format.h>
-
-#include "xalloc.h"
-
-#include "gettext.h"
-#define _(msgid) gettext (msgid)
-
-/* (headers) */
-
-/* (specification)
- "ONEWAY" (oneway_):
- *^variables=custom;
- missing=miss:!analysis/listwise,
- incl:include/!exclude;
- +contrast= double list;
- +statistics[st_]=descriptives,homogeneity.
-*/
-/* (declarations) */
-/* (functions) */
-
-static struct cmd_oneway cmd;
-
-/* The independent variable */
-static const struct variable *indep_var;
-
-/* Number of dependent variables */
-static size_t n_vars;
-
-/* The dependent variables */
-static const struct variable **vars;
-
-
-/* A hash table containing all the distinct values of the independent
- variables */
-static struct hsh_table *global_group_hash;
-
-/* The number of distinct values of the independent variable, when all
- missing values are disregarded */
-static int ostensible_number_of_groups = -1;
-
-
-static void run_oneway (struct cmd_oneway *, struct casereader *,
- const struct dataset *);
-
-
-/* Routines to show the output tables */
-static void show_anova_table(void);
-static void show_descriptives (const struct dictionary *dict);
-static void show_homogeneity(void);
-
-static void show_contrast_coeffs (short *);
-static void show_contrast_tests (short *);
-
-
-enum stat_table_t {STAT_DESC = 1, STAT_HOMO = 2};
-
-static enum stat_table_t stat_tables;
-
-static void output_oneway (const struct dictionary *dict);
-
-
-int
-cmd_oneway (struct lexer *lexer, struct dataset *ds)
-{
- struct casegrouper *grouper;
- struct casereader *group;
- int i;
- bool ok;
-
- if ( !parse_oneway (lexer, ds, &cmd, NULL))
- return CMD_FAILURE;
-
- /* What statistics were requested */
- if ( cmd.sbc_statistics)
- {
-
- for (i = 0; i < ONEWAY_ST_count; ++i)
- {
- if (! cmd.a_statistics[i]) continue;
-
- switch (i)
- {
- case ONEWAY_ST_DESCRIPTIVES:
- stat_tables |= STAT_DESC;
- break;
- case ONEWAY_ST_HOMOGENEITY:
- stat_tables |= STAT_HOMO;
- break;
- }
- }
- }
-
- /* Data pass. FIXME: error handling. */
- grouper = casegrouper_create_splits (proc_open (ds), dataset_dict (ds));
- while (casegrouper_get_next_group (grouper, &group))
- run_oneway (&cmd, group, ds);
- ok = casegrouper_destroy (grouper);
- ok = proc_commit (ds) && ok;
-
- free (vars);
- free_oneway (&cmd);
-
- return ok ? CMD_SUCCESS : CMD_CASCADING_FAILURE;
-}
-
-
-static void
-output_oneway (const struct dictionary *dict)
-{
- size_t i;
- short *bad_contrast;
-
- bad_contrast = xnmalloc (cmd.sbc_contrast, sizeof *bad_contrast);
-
- /* Check the sanity of the given contrast values */
- for (i = 0; i < cmd.sbc_contrast; ++i)
- {
- int j;
- double sum = 0;
-
- bad_contrast[i] = 0;
- if (subc_list_double_count (&cmd.dl_contrast[i]) !=
- ostensible_number_of_groups)
- {
- msg (SW,
- _("Number of contrast coefficients must equal the number of groups"));
- bad_contrast[i] = 1;
- continue;
- }
-
- for (j = 0; j < ostensible_number_of_groups; ++j)
- sum += subc_list_double_at (&cmd.dl_contrast[i], j);
-
- if ( sum != 0.0 )
- msg (SW, _("Coefficients for contrast %zu do not total zero"), i + 1);
- }
-
- if ( stat_tables & STAT_DESC )
- show_descriptives (dict);
-
- if ( stat_tables & STAT_HOMO )
- show_homogeneity ();
-
- show_anova_table ();
-
- if (cmd.sbc_contrast )
- {
- show_contrast_coeffs (bad_contrast);
- show_contrast_tests (bad_contrast);
- }
-
- free (bad_contrast);
-
- /* Clean up */
- for (i = 0; i < n_vars; ++i )
- {
- struct hsh_table *group_hash = group_proc_get (vars[i])->group_hash;
-
- hsh_destroy (group_hash);
- }
-
- hsh_destroy (global_group_hash);
-}
-
-
-/* Parser for the variables sub command */
-static int
-oneway_custom_variables (struct lexer *lexer,
- struct dataset *ds, struct cmd_oneway *cmd UNUSED,
- void *aux UNUSED)
-{
- struct dictionary *dict = dataset_dict (ds);
-
- lex_match (lexer, '=');
-
- if ((lex_token (lexer) != T_ID ||
- dict_lookup_var (dict, lex_tokid (lexer)) == NULL)
- && lex_token (lexer) != T_ALL)
- return 2;
-
- if (!parse_variables_const (lexer, dict, &vars, &n_vars,
- PV_DUPLICATE
- | PV_NUMERIC | PV_NO_SCRATCH) )
- {
- free (vars);
- return 0;
- }
-
- assert (n_vars);
-
- if ( ! lex_match (lexer, T_BY))
- return 2;
-
- indep_var = parse_variable (lexer, dict);
-
- if ( !indep_var )
- {
- msg (SE, _("`%s' is not a variable name"), lex_tokid (lexer));
- return 0;
- }
-
- return 1;
-}
-
-
-/* Show the ANOVA table */
-static void
-show_anova_table (void)
-{
- size_t i;
- int n_cols =7;
- size_t n_rows = n_vars * 3 + 1;
-
- struct tab_table *t;
-
-
- t = tab_create (n_cols, n_rows);
- tab_headers (t, 2, 0, 1, 0);
-
- tab_box (t,
- TAL_2, TAL_2,
- -1, TAL_1,
- 0, 0,
- n_cols - 1, n_rows - 1);
-
- tab_hline (t, TAL_2, 0, n_cols - 1, 1 );
- tab_vline (t, TAL_2, 2, 0, n_rows - 1);
- tab_vline (t, TAL_0, 1, 0, 0);
-
- tab_text (t, 2, 0, TAB_CENTER | TAT_TITLE, _("Sum of Squares"));
- tab_text (t, 3, 0, TAB_CENTER | TAT_TITLE, _("df"));
- tab_text (t, 4, 0, TAB_CENTER | TAT_TITLE, _("Mean Square"));
- tab_text (t, 5, 0, TAB_CENTER | TAT_TITLE, _("F"));
- tab_text (t, 6, 0, TAB_CENTER | TAT_TITLE, _("Significance"));
-
-
- for (i = 0; i < n_vars; ++i)
- {
- struct group_statistics *totals = &group_proc_get (vars[i])->ugs;
- struct hsh_table *group_hash = group_proc_get (vars[i])->group_hash;
- struct hsh_iterator g;
- struct group_statistics *gs;
- double ssa = 0;
- const char *s = var_to_string (vars[i]);
-
- for (gs = hsh_first (group_hash, &g);
- gs != 0;
- gs = hsh_next (group_hash, &g))
- {
- ssa += pow2 (gs->sum) / gs->n;
- }
-
- ssa -= pow2 (totals->sum) / totals->n;
-
- tab_text (t, 0, i * 3 + 1, TAB_LEFT | TAT_TITLE, s);
- tab_text (t, 1, i * 3 + 1, TAB_LEFT | TAT_TITLE, _("Between Groups"));
- tab_text (t, 1, i * 3 + 2, TAB_LEFT | TAT_TITLE, _("Within Groups"));
- tab_text (t, 1, i * 3 + 3, TAB_LEFT | TAT_TITLE, _("Total"));
-
- if (i > 0)
- tab_hline (t, TAL_1, 0, n_cols - 1, i * 3 + 1);
-
- {
- struct group_proc *gp = group_proc_get (vars[i]);
- const double sst = totals->ssq - pow2 (totals->sum) / totals->n;
- const double df1 = gp->n_groups - 1;
- const double df2 = totals->n - gp->n_groups;
- const double msa = ssa / df1;
-
- gp->mse = (sst - ssa) / df2;
-
-
- /* Sums of Squares */
- tab_double (t, 2, i * 3 + 1, 0, ssa, NULL);
- tab_double (t, 2, i * 3 + 3, 0, sst, NULL);
- tab_double (t, 2, i * 3 + 2, 0, sst - ssa, NULL);
-
-
- /* Degrees of freedom */
- tab_fixed (t, 3, i * 3 + 1, 0, df1, 4, 0);
- tab_fixed (t, 3, i * 3 + 2, 0, df2, 4, 0);
- tab_fixed (t, 3, i * 3 + 3, 0, totals->n - 1, 4, 0);
-
- /* Mean Squares */
- tab_double (t, 4, i * 3 + 1, TAB_RIGHT, msa, NULL);
- tab_double (t, 4, i * 3 + 2, TAB_RIGHT, gp->mse, NULL);
-
- {
- const double F = msa / gp->mse ;
-
- /* The F value */
- tab_double (t, 5, i * 3 + 1, 0, F, NULL);
-
- /* The significance */
- tab_double (t, 6, i * 3 + 1, 0, gsl_cdf_fdist_Q (F, df1, df2), NULL);
- }
- }
- }
-
-
- tab_title (t, _("ANOVA"));
- tab_submit (t);
-}
-
-
-/* Show the descriptives table */
-static void
-show_descriptives (const struct dictionary *dict)
-{
- size_t v;
- int n_cols = 10;
- struct tab_table *t;
- int row;
-
- const double confidence = 0.95;
- const double q = (1.0 - confidence) / 2.0;
-
- const struct variable *wv = dict_get_weight (dict);
- const struct fmt_spec *wfmt = wv ? var_get_print_format (wv) : & F_8_0;
-
- int n_rows = 2;
-
- for ( v = 0; v < n_vars; ++v )
- n_rows += group_proc_get (vars[v])->n_groups + 1;
-
- t = tab_create (n_cols, n_rows);
- tab_headers (t, 2, 0, 2, 0);
-
-
- /* Put a frame around the entire box, and vertical lines inside */
- tab_box (t,
- TAL_2, TAL_2,
- -1, TAL_1,
- 0, 0,
- n_cols - 1, n_rows - 1);
-
- /* Underline headers */
- tab_hline (t, TAL_2, 0, n_cols - 1, 2);
- tab_vline (t, TAL_2, 2, 0, n_rows - 1);
-
- tab_text (t, 2, 1, TAB_CENTER | TAT_TITLE, _("N"));
- tab_text (t, 3, 1, TAB_CENTER | TAT_TITLE, _("Mean"));
- tab_text (t, 4, 1, TAB_CENTER | TAT_TITLE, _("Std. Deviation"));
- tab_text (t, 5, 1, TAB_CENTER | TAT_TITLE, _("Std. Error"));
-
-
- tab_vline (t, TAL_0, 7, 0, 0);
- tab_hline (t, TAL_1, 6, 7, 1);
- tab_joint_text_format (t, 6, 0, 7, 0, TAB_CENTER | TAT_TITLE,
- _("%g%% Confidence Interval for Mean"),
- confidence*100.0);
-
- tab_text (t, 6, 1, TAB_CENTER | TAT_TITLE, _("Lower Bound"));
- tab_text (t, 7, 1, TAB_CENTER | TAT_TITLE, _("Upper Bound"));
-
- tab_text (t, 8, 1, TAB_CENTER | TAT_TITLE, _("Minimum"));
- tab_text (t, 9, 1, TAB_CENTER | TAT_TITLE, _("Maximum"));
-
-
- tab_title (t, _("Descriptives"));
-
-
- row = 2;
- for (v = 0; v < n_vars; ++v)
- {
- double T;
- double std_error;
-
- struct group_proc *gp = group_proc_get (vars[v]);
-
- struct group_statistics *gs;
- struct group_statistics *totals = &gp->ugs;
-
- const char *s = var_to_string (vars[v]);
- const struct fmt_spec *fmt = var_get_print_format (vars[v]);
-
- struct group_statistics *const *gs_array =
- (struct group_statistics *const *) hsh_sort (gp->group_hash);
- int count = 0;
-
- tab_text (t, 0, row, TAB_LEFT | TAT_TITLE, s);
- if ( v > 0)
- tab_hline (t, TAL_1, 0, n_cols - 1, row);
-
- for (count = 0; count < hsh_count (gp->group_hash); ++count)
- {
- struct string vstr;
- ds_init_empty (&vstr);
- gs = gs_array[count];
-
- var_append_value_name (indep_var, &gs->id, &vstr);
-
- tab_text (t, 1, row + count,
- TAB_LEFT | TAT_TITLE,
- ds_cstr (&vstr));
-
- ds_destroy (&vstr);
-
- /* Now fill in the numbers ... */
-
- tab_fixed (t, 2, row + count, 0, gs->n, 8, 0);
-
- tab_double (t, 3, row + count, 0, gs->mean, NULL);
-
- tab_double (t, 4, row + count, 0, gs->std_dev, NULL);
-
- std_error = gs->std_dev / sqrt (gs->n) ;
- tab_double (t, 5, row + count, 0,
- std_error, NULL);
-
- /* Now the confidence interval */
-
- T = gsl_cdf_tdist_Qinv (q, gs->n - 1);
-
- tab_double (t, 6, row + count, 0,
- gs->mean - T * std_error, NULL);
-
- tab_double (t, 7, row + count, 0,
- gs->mean + T * std_error, NULL);
-
- /* Min and Max */
-
- tab_double (t, 8, row + count, 0, gs->minimum, fmt);
- tab_double (t, 9, row + count, 0, gs->maximum, fmt);
- }
-
- tab_text (t, 1, row + count,
- TAB_LEFT | TAT_TITLE, _("Total"));
-
- tab_double (t, 2, row + count, 0, totals->n, wfmt);
-
- tab_double (t, 3, row + count, 0, totals->mean, NULL);
-
- tab_double (t, 4, row + count, 0, totals->std_dev, NULL);
-
- std_error = totals->std_dev / sqrt (totals->n) ;
-
- tab_double (t, 5, row + count, 0, std_error, NULL);
-
- /* Now the confidence interval */
-
- T = gsl_cdf_tdist_Qinv (q, totals->n - 1);
-
- tab_double (t, 6, row + count, 0,
- totals->mean - T * std_error, NULL);
-
- tab_double (t, 7, row + count, 0,
- totals->mean + T * std_error, NULL);
-
- /* Min and Max */
-
- tab_double (t, 8, row + count, 0, totals->minimum, fmt);
- tab_double (t, 9, row + count, 0, totals->maximum, fmt);
-
- row += gp->n_groups + 1;
- }
-
- tab_submit (t);
-}
-
-/* Show the homogeneity table */
-static void
-show_homogeneity (void)
-{
- size_t v;
- int n_cols = 5;
- size_t n_rows = n_vars + 1;
-
- struct tab_table *t;
-
-
- t = tab_create (n_cols, n_rows);
- tab_headers (t, 1, 0, 1, 0);
-
-
- /* Put a frame around the entire box, and vertical lines inside */
- tab_box (t,
- TAL_2, TAL_2,
- -1, TAL_1,
- 0, 0,
- n_cols - 1, n_rows - 1);
-
-
- tab_hline (t, TAL_2, 0, n_cols - 1, 1);
- tab_vline (t, TAL_2, 1, 0, n_rows - 1);
-
-
- tab_text (t, 1, 0, TAB_CENTER | TAT_TITLE, _("Levene Statistic"));
- tab_text (t, 2, 0, TAB_CENTER | TAT_TITLE, _("df1"));
- tab_text (t, 3, 0, TAB_CENTER | TAT_TITLE, _("df2"));
- tab_text (t, 4, 0, TAB_CENTER | TAT_TITLE, _("Significance"));
-
- tab_title (t, _("Test of Homogeneity of Variances"));
-
- for (v = 0; v < n_vars; ++v)
- {
- double F;
- const struct variable *var = vars[v];
- const struct group_proc *gp = group_proc_get (vars[v]);
- const char *s = var_to_string (var);
- const struct group_statistics *totals = &gp->ugs;
-
- const double df1 = gp->n_groups - 1;
- const double df2 = totals->n - gp->n_groups;
-
- tab_text (t, 0, v + 1, TAB_LEFT | TAT_TITLE, s);
-
- F = gp->levene;
- tab_double (t, 1, v + 1, TAB_RIGHT, F, NULL);
- tab_fixed (t, 2, v + 1, TAB_RIGHT, df1, 8, 0);
- tab_fixed (t, 3, v + 1, TAB_RIGHT, df2, 8, 0);
-
- /* Now the significance */
- tab_double (t, 4, v + 1, TAB_RIGHT,gsl_cdf_fdist_Q (F, df1, df2), NULL);
- }
-
- tab_submit (t);
-}
-
-
-/* Show the contrast coefficients table */
-static void
-show_contrast_coeffs (short *bad_contrast)
-{
- int n_cols = 2 + ostensible_number_of_groups;
- int n_rows = 2 + cmd.sbc_contrast;
- int count = 0;
- void *const *group_values;
-
- struct tab_table *t;
-
- t = tab_create (n_cols, n_rows);
- tab_headers (t, 2, 0, 2, 0);
-
- /* Put a frame around the entire box, and vertical lines inside */
- tab_box (t,
- TAL_2, TAL_2,
- -1, TAL_1,
- 0, 0,
- n_cols - 1, n_rows - 1);
-
- tab_box (t,
- -1, -1,
- TAL_0, TAL_0,
- 2, 0,
- n_cols - 1, 0);
-
- tab_box (t,
- -1, -1,
- TAL_0, TAL_0,
- 0, 0,
- 1, 1);
-
- tab_hline (t, TAL_1, 2, n_cols - 1, 1);
- tab_hline (t, TAL_2, 0, n_cols - 1, 2);
-
- tab_vline (t, TAL_2, 2, 0, n_rows - 1);
-
- tab_title (t, _("Contrast Coefficients"));
-
- tab_text (t, 0, 2, TAB_LEFT | TAT_TITLE, _("Contrast"));
-
-
- tab_joint_text (t, 2, 0, n_cols - 1, 0, TAB_CENTER | TAT_TITLE,
- var_to_string (indep_var));
-
- group_values = hsh_sort (global_group_hash);
- for (count = 0;
- count < hsh_count (global_group_hash);
- ++count)
- {
- double *group_value_p;
- union value group_value;
- int i;
- struct string vstr;
-
- ds_init_empty (&vstr);
-
- group_value_p = group_values[count];
- group_value.f = *group_value_p;
- var_append_value_name (indep_var, &group_value, &vstr);
-
- tab_text (t, count + 2, 1, TAB_CENTER | TAT_TITLE,
- ds_cstr (&vstr));
-
- ds_destroy (&vstr);
-
-
- for (i = 0; i < cmd.sbc_contrast; ++i )
- {
- tab_text_format (t, 1, i + 2, TAB_CENTER, "%d", i + 1);
-
- if ( bad_contrast[i] )
- tab_text (t, count + 2, i + 2, TAB_RIGHT, "?" );
- else
- tab_text_format (t, count + 2, i + 2, TAB_RIGHT, "%g",
- subc_list_double_at (&cmd.dl_contrast[i], count));
- }
- }
-
- tab_submit (t);
-}
-
-
-/* Show the results of the contrast tests */
-static void
-show_contrast_tests (short *bad_contrast)
-{
- size_t v;
- int n_cols = 8;
- size_t n_rows = 1 + n_vars * 2 * cmd.sbc_contrast;
-
- struct tab_table *t;
-
- t = tab_create (n_cols, n_rows);
- tab_headers (t, 3, 0, 1, 0);
-
- /* Put a frame around the entire box, and vertical lines inside */
- tab_box (t,
- TAL_2, TAL_2,
- -1, TAL_1,
- 0, 0,
- n_cols - 1, n_rows - 1);
-
- tab_box (t,
- -1, -1,
- TAL_0, TAL_0,
- 0, 0,
- 2, 0);
-
- tab_hline (t, TAL_2, 0, n_cols - 1, 1);
- tab_vline (t, TAL_2, 3, 0, n_rows - 1);
-
-
- tab_title (t, _("Contrast Tests"));
-
- tab_text (t, 2, 0, TAB_CENTER | TAT_TITLE, _("Contrast"));
- tab_text (t, 3, 0, TAB_CENTER | TAT_TITLE, _("Value of Contrast"));
- tab_text (t, 4, 0, TAB_CENTER | TAT_TITLE, _("Std. Error"));
- tab_text (t, 5, 0, TAB_CENTER | TAT_TITLE, _("t"));
- tab_text (t, 6, 0, TAB_CENTER | TAT_TITLE, _("df"));
- tab_text (t, 7, 0, TAB_CENTER | TAT_TITLE, _("Sig. (2-tailed)"));
-
- for (v = 0; v < n_vars; ++v)
- {
- int i;
- int lines_per_variable = 2 * cmd.sbc_contrast;
-
-
- tab_text (t, 0, (v * lines_per_variable) + 1, TAB_LEFT | TAT_TITLE,
- var_to_string (vars[v]));
-
- for (i = 0; i < cmd.sbc_contrast; ++i)
- {
- int ci;
- double contrast_value = 0.0;
- double coef_msq = 0.0;
- struct group_proc *grp_data = group_proc_get (vars[v]);
- struct hsh_table *group_hash = grp_data->group_hash;
-
- void *const *group_stat_array;
-
- double T;
- double std_error_contrast;
- double df;
- double sec_vneq = 0.0;
-
-
- /* Note: The calculation of the degrees of freedom in the
- "variances not equal" case is painfull!!
- The following formula may help to understand it:
- \frac{\left (\sum_{i=1}^k{c_i^2\frac{s_i^2}{n_i}}\right)^2}
- {
- \sum_{i=1}^k\left (
- \frac{\left (c_i^2\frac{s_i^2}{n_i}\right)^2} {n_i-1}
- \right)
- }
- */
-
- double df_denominator = 0.0;
- double df_numerator = 0.0;
- if ( i == 0 )
- {
- tab_text (t, 1, (v * lines_per_variable) + i + 1,
- TAB_LEFT | TAT_TITLE,
- _("Assume equal variances"));
-
- tab_text (t, 1, (v * lines_per_variable) + i + 1 + cmd.sbc_contrast,
- TAB_LEFT | TAT_TITLE,
- _("Does not assume equal"));
- }
-
- tab_text_format (t, 2, (v * lines_per_variable) + i + 1,
- TAB_CENTER | TAT_TITLE, "%d", i + 1);
-
-
- tab_text_format (t, 2,
- (v * lines_per_variable) + i + 1 + cmd.sbc_contrast,
- TAB_CENTER | TAT_TITLE, "%d", i + 1);
-
-
- if ( bad_contrast[i])
- continue;
-
- group_stat_array = hsh_sort (group_hash);
-
- for (ci = 0; ci < hsh_count (group_hash); ++ci)
- {
- const double coef = subc_list_double_at (&cmd.dl_contrast[i], ci);
- struct group_statistics *gs = group_stat_array[ci];
-
- const double winv = pow2 (gs->std_dev) / gs->n;
-
- contrast_value += coef * gs->mean;
-
- coef_msq += (coef * coef) / gs->n;
-
- sec_vneq += (coef * coef) * pow2 (gs->std_dev) /gs->n;
-
- df_numerator += (coef * coef) * winv;
- df_denominator += pow2((coef * coef) * winv) / (gs->n - 1);
- }
- sec_vneq = sqrt (sec_vneq);
-
- df_numerator = pow2 (df_numerator);
-
- tab_double (t, 3, (v * lines_per_variable) + i + 1,
- TAB_RIGHT, contrast_value, NULL);
-
- tab_double (t, 3, (v * lines_per_variable) + i + 1 +
- cmd.sbc_contrast,
- TAB_RIGHT, contrast_value, NULL);
-
- std_error_contrast = sqrt (grp_data->mse * coef_msq);
-
- /* Std. Error */
- tab_double (t, 4, (v * lines_per_variable) + i + 1,
- TAB_RIGHT, std_error_contrast,
- NULL);
-
- T = fabs (contrast_value / std_error_contrast);
-
- /* T Statistic */
-
- tab_double (t, 5, (v * lines_per_variable) + i + 1,
- TAB_RIGHT, T,
- NULL);
-
- df = grp_data->ugs.n - grp_data->n_groups;
-
- /* Degrees of Freedom */
- tab_fixed (t, 6, (v * lines_per_variable) + i + 1,
- TAB_RIGHT, df,
- 8, 0);
-
-
- /* Significance TWO TAILED !!*/
- tab_double (t, 7, (v * lines_per_variable) + i + 1,
- TAB_RIGHT, 2 * gsl_cdf_tdist_Q (T, df),
- NULL);
-
- /* Now for the Variances NOT Equal case */
-
- /* Std. Error */
- tab_double (t, 4,
- (v * lines_per_variable) + i + 1 + cmd.sbc_contrast,
- TAB_RIGHT, sec_vneq,
- NULL);
-
- T = contrast_value / sec_vneq;
- tab_double (t, 5,
- (v * lines_per_variable) + i + 1 + cmd.sbc_contrast,
- TAB_RIGHT, T,
- NULL);
-
- df = df_numerator / df_denominator;
-
- tab_double (t, 6,
- (v * lines_per_variable) + i + 1 + cmd.sbc_contrast,
- TAB_RIGHT, df,
- NULL);
-
- /* The Significance */
-
- tab_double (t, 7, (v * lines_per_variable) + i + 1 + cmd.sbc_contrast,
- TAB_RIGHT, 2 * gsl_cdf_tdist_Q (T,df),
- NULL);
- }
-
- if ( v > 0 )
- tab_hline (t, TAL_1, 0, n_cols - 1, (v * lines_per_variable) + 1);
- }
-
- tab_submit (t);
-}
-
-
-/* ONEWAY ANOVA Calculations */
-
-static void postcalc (struct cmd_oneway *cmd UNUSED);
-
-static void precalc (struct cmd_oneway *cmd UNUSED);
-
-
-
-/* Pre calculations */
-static void
-precalc (struct cmd_oneway *cmd UNUSED)
-{
- size_t i = 0;
-
- for (i = 0; i < n_vars; ++i)
- {
- struct group_proc *gp = group_proc_get (vars[i]);
- struct group_statistics *totals = &gp->ugs;
-
- /* Create a hash for each of the dependent variables.
- The hash contains a group_statistics structure,
- and is keyed by value of the independent variable */
-
- gp->group_hash = hsh_create (4, compare_group, hash_group,
- (hsh_free_func *) free_group,
- indep_var);
-
- totals->sum = 0;
- totals->n = 0;
- totals->ssq = 0;
- totals->sum_diff = 0;
- totals->maximum = -DBL_MAX;
- totals->minimum = DBL_MAX;
- }
-}
-
-static int
-compare_double_3way (const void *a_, const void *b_, const void *aux UNUSED)
-{
- const double *a = a_;
- const double *b = b_;
- return *a < *b ? -1 : *a > *b;
-}
-
-static unsigned
-do_hash_double (const void *value_, const void *aux UNUSED)
-{
- const double *value = value_;
- return hash_double (*value, 0);
-}
-
-static void
-free_double (void *value_, const void *aux UNUSED)
-{
- double *value = value_;
- free (value);
-}
-
-static void
-run_oneway (struct cmd_oneway *cmd,
- struct casereader *input,
- const struct dataset *ds)
-{
- struct taint *taint;
- struct dictionary *dict = dataset_dict (ds);
- enum mv_class exclude;
- struct casereader *reader;
- struct ccase *c;
-
- c = casereader_peek (input, 0);
- if (c == NULL)
- {
- casereader_destroy (input);
- return;
- }
- output_split_file_values (ds, c);
- case_unref (c);
-
- taint = taint_clone (casereader_get_taint (input));
-
- global_group_hash = hsh_create (4,
- compare_double_3way,
- do_hash_double,
- free_double,
- indep_var);
-
- precalc (cmd);
-
- exclude = cmd->incl != ONEWAY_INCLUDE ? MV_ANY : MV_SYSTEM;
- input = casereader_create_filter_missing (input, &indep_var, 1,
- exclude, NULL, NULL);
- if (cmd->miss == ONEWAY_LISTWISE)
- input = casereader_create_filter_missing (input, vars, n_vars,
- exclude, NULL, NULL);
- input = casereader_create_filter_weight (input, dict, NULL, NULL);
-
- reader = casereader_clone (input);
- for (; (c = casereader_read (reader)) != NULL; case_unref (c))
- {
- size_t i;
-
- const double weight = dict_get_case_weight (dict, c, NULL);
-
- const union value *indep_val = case_data (c, indep_var);
- void **p = hsh_probe (global_group_hash, &indep_val->f);
- if (*p == NULL)
- {
- double *value = *p = xmalloc (sizeof *value);
- *value = indep_val->f;
- }
-
- for (i = 0; i < n_vars; ++i)
- {
- const struct variable *v = vars[i];
-
- const union value *val = case_data (c, v);
-
- struct group_proc *gp = group_proc_get (vars[i]);
- struct hsh_table *group_hash = gp->group_hash;
-
- struct group_statistics *gs;
-
- gs = hsh_find (group_hash, indep_val );
-
- if ( ! gs )
- {
- gs = xmalloc (sizeof *gs);
- gs->id = *indep_val;
- gs->sum = 0;
- gs->n = 0;
- gs->ssq = 0;
- gs->sum_diff = 0;
- gs->minimum = DBL_MAX;
- gs->maximum = -DBL_MAX;
-
- hsh_insert ( group_hash, gs );
- }
-
- if (!var_is_value_missing (v, val, exclude))
- {
- struct group_statistics *totals = &gp->ugs;
-
- totals->n += weight;
- totals->sum += weight * val->f;
- totals->ssq += weight * pow2 (val->f);
-
- if ( val->f * weight < totals->minimum )
- totals->minimum = val->f * weight;
-
- if ( val->f * weight > totals->maximum )
- totals->maximum = val->f * weight;
-
- gs->n += weight;
- gs->sum += weight * val->f;
- gs->ssq += weight * pow2 (val->f);
-
- if ( val->f * weight < gs->minimum )
- gs->minimum = val->f * weight;
-
- if ( val->f * weight > gs->maximum )
- gs->maximum = val->f * weight;
- }
-
- gp->n_groups = hsh_count ( group_hash );
- }
-
- }
- casereader_destroy (reader);
-
- postcalc (cmd);
-
-
- if ( stat_tables & STAT_HOMO )
- levene (dict, casereader_clone (input), indep_var, n_vars, vars, exclude);
-
- casereader_destroy (input);
-
- ostensible_number_of_groups = hsh_count (global_group_hash);
-
- if (!taint_has_tainted_successor (taint))
- output_oneway (dict);
-
- taint_destroy (taint);
-}
-
-
-/* Post calculations for the ONEWAY command */
-void
-postcalc ( struct cmd_oneway *cmd UNUSED )
-{
- size_t i = 0;
-
- for (i = 0; i < n_vars; ++i)
- {
- struct group_proc *gp = group_proc_get (vars[i]);
- struct hsh_table *group_hash = gp->group_hash;
- struct group_statistics *totals = &gp->ugs;
-
- struct hsh_iterator g;
- struct group_statistics *gs;
-
- for (gs = hsh_first (group_hash, &g);
- gs != 0;
- gs = hsh_next (group_hash, &g))
- {
- gs->mean = gs->sum / gs->n;
- gs->s_std_dev = sqrt (gs->ssq / gs->n - pow2 (gs->mean));
-
- gs->std_dev = sqrt (
- gs->n / (gs->n - 1) *
- ( gs->ssq / gs->n - pow2 (gs->mean))
- );
-
- gs->se_mean = gs->std_dev / sqrt (gs->n);
- gs->mean_diff = gs->sum_diff / gs->n;
- }
-
- totals->mean = totals->sum / totals->n;
- totals->std_dev = sqrt (
- totals->n / (totals->n - 1) *
- (totals->ssq / totals->n - pow2 (totals->mean))
- );
-
- totals->se_mean = totals->std_dev / sqrt (totals->n);
- }
-}
-
-/*
- Local Variables:
- mode: c
- End:
-*/
projects / pspp / commitdiff