--- /dev/null
+/* PSPP - NPAR TESTS. -*-c-*-
+
+Copyright (C) 2006 Free Software Foundation, Inc.
+Author: John Darrington <john@darrington.wattle.id.au>,
+
+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
+02110-1301, USA. */
+
+#include <config.h>
+
+#include <language/lexer/lexer.h>
+#include <language/lexer/variable-parser.h>
+#include <language/command.h>
+#include <data/procedure.h>
+#include <libpspp/pool.h>
+#include <libpspp/hash.h>
+
+#include <data/casefilter.h>
+#include <data/case.h>
+#include <data/casefile.h>
+#include <math/moments.h>
+#include <data/dictionary.h>
+#include <language/stats/chisquare.h>
+#include <language/stats/binomial.h>
+#include <math.h>
+
+#include "npar.h"
+#include "npar-summary.h"
+
+#include "gettext.h"
+#define _(msgid) gettext (msgid)
+
+/* (headers) */
+
+/* (specification)
+ "NPAR TESTS" (npar_):
+ +chisquare=custom;
+ +binomial=custom;
+ +wilcoxon=custom;
+ +mcnemar=custom;
+ +sign=custom;
+ +cochran=varlist;
+ +friedman=varlist;
+ +kendall=varlist;
+ missing=miss:!analysis/listwise,
+ incl:include/!exclude;
+ +statistics[st_]=descriptives,quartiles,all.
+*/
+/* (declarations) */
+/* (functions) */
+
+
+static struct cmd_npar_tests cmd;
+
+
+struct npar_specs
+{
+ struct pool *pool;
+ struct npar_test **test;
+ size_t n_tests;
+
+ const struct variable *const* vv; /* Compendium of all variables
+ (those mentioned on ANY subcommand */
+ int n_vars; /* Number of variables in vv */
+
+ struct casefilter *filter; /* The missing value filter */
+
+ bool descriptives; /* Descriptive statistics should be calculated */
+ bool quartiles; /* Quartiles should be calculated */
+};
+
+void one_sample_insert_variables (const struct npar_test *test,
+ struct hsh_table *variables);
+
+static bool
+npar_execute(const struct ccase *first UNUSED,
+ const struct casefile *cf, void *aux,
+ const struct dataset *ds)
+{
+ int t;
+ const struct npar_specs *specs = aux;
+ struct descriptives *summary_descriptives = NULL;
+
+ for ( t = 0 ; t < specs->n_tests; ++t )
+ {
+ const struct npar_test *test = specs->test[t];
+ if ( NULL == test->execute )
+ {
+ msg (SW, _("NPAR subcommand not currently implemented."));
+ continue;
+ }
+ test->execute (ds, cf, specs->filter, test);
+ }
+
+ if ( specs->descriptives )
+ {
+ summary_descriptives = xnmalloc (sizeof (*summary_descriptives),
+ specs->n_vars);
+
+ npar_summary_calc_descriptives (summary_descriptives, cf,
+ specs->filter,
+ dataset_dict (ds),
+ specs->vv, specs->n_vars);
+ }
+
+ if ( specs->descriptives || specs->quartiles )
+ do_summary_box (summary_descriptives, specs->vv, specs->n_vars );
+
+ free (summary_descriptives);
+
+ return true;
+}
+
+
+int
+cmd_npar_tests (struct lexer *lexer, struct dataset *ds)
+{
+ bool ok;
+ int i;
+ struct npar_specs npar_specs = {0, 0, 0, 0, 0, 0, 0, 0};
+ struct hsh_table *var_hash;
+ npar_specs.pool = pool_create ();
+
+ var_hash = hsh_create_pool (npar_specs.pool, 0,
+ compare_vars_by_name, hash_var_by_name,
+ NULL, NULL);
+
+ if ( ! parse_npar_tests (lexer, ds, &cmd, &npar_specs) )
+ {
+ pool_destroy (npar_specs.pool);
+ return CMD_FAILURE;
+ }
+
+ for (i = 0; i < npar_specs.n_tests; ++i )
+ {
+ const struct npar_test *test = npar_specs.test[i];
+ test->insert_variables (test, var_hash);
+ }
+
+ npar_specs.vv = (const struct variable *const *) hsh_data (var_hash);
+ npar_specs.n_vars = hsh_count (var_hash);
+
+ if ( cmd.sbc_statistics )
+ {
+ int i;
+
+ for ( i = 0 ; i < NPAR_ST_count; ++i )
+ {
+ if ( cmd.a_statistics[i] )
+ {
+ switch ( i )
+ {
+ case NPAR_ST_DESCRIPTIVES:
+ npar_specs.descriptives = true;
+ break;
+ case NPAR_ST_QUARTILES:
+ npar_specs.quartiles = true;
+ break;
+ case NPAR_ST_ALL:
+ npar_specs.quartiles = true;
+ npar_specs.descriptives = true;
+ break;
+ default:
+ NOT_REACHED();
+ };
+ }
+ }
+ }
+
+ npar_specs.filter =
+ casefilter_create (cmd.incl == NPAR_EXCLUDE, 0, 0);
+
+ if ( cmd.miss == NPAR_LISTWISE )
+ casefilter_add_variables (npar_specs.filter,
+ npar_specs.vv,
+ npar_specs.n_vars);
+
+ ok = multipass_procedure_with_splits (ds, npar_execute, &npar_specs);
+
+ casefilter_destroy (npar_specs.filter);
+
+ hsh_destroy (var_hash);
+
+ pool_destroy (npar_specs.pool);
+
+ return ok ? CMD_SUCCESS : CMD_CASCADING_FAILURE;
+}
+
+int
+npar_custom_chisquare(struct lexer *lexer, struct dataset *ds, struct cmd_npar_tests *cmd UNUSED, void *aux )
+{
+ struct npar_specs *specs = aux;
+
+ struct chisquare_test *cstp = pool_alloc(specs->pool, sizeof(*cstp));
+ struct one_sample_test *tp = (struct one_sample_test *) cstp;
+
+ ((struct npar_test *)tp)->execute = chisquare_execute;
+ ((struct npar_test *)tp)->insert_variables = one_sample_insert_variables;
+
+ if (!parse_variables_pool (lexer, specs->pool, dataset_dict (ds),
+ &tp->vars, &tp->n_vars,
+ PV_NO_SCRATCH | PV_NO_DUPLICATE))
+ {
+ return 2;
+ }
+
+ cstp->ranged = false;
+
+ if ( lex_match (lexer, '('))
+ {
+ cstp->ranged = true;
+ if ( ! lex_force_num (lexer)) return 0;
+ cstp->lo = lex_integer (lexer);
+ lex_get (lexer);
+ lex_force_match (lexer, ',');
+ if (! lex_force_num (lexer) ) return 0;
+ cstp->hi = lex_integer (lexer);
+ if ( cstp->lo >= cstp->hi )
+ {
+ msg(ME,
+ _("The specified value of HI (%d) is "
+ "lower than the specified value of LO (%d)"),
+ cstp->hi, cstp->lo);
+ return 0;
+ }
+ lex_get (lexer);
+ if (! lex_force_match (lexer, ')')) return 0;
+ }
+
+ cstp->n_expected = 0;
+ cstp->expected = NULL;
+ if ( lex_match (lexer, '/') )
+ {
+ if ( lex_match_id (lexer, "EXPECTED") )
+ {
+ lex_force_match (lexer, '=');
+ if ( ! lex_match_id (lexer, "EQUAL") )
+ {
+ double f;
+ int n;
+ while ( lex_is_number(lexer) )
+ {
+ int i;
+ n = 1;
+ f = lex_number (lexer);
+ lex_get (lexer);
+ if ( lex_match (lexer, '*'))
+ {
+ n = f;
+ f = lex_number (lexer);
+ lex_get (lexer);
+ }
+ lex_match (lexer, ',');
+
+ cstp->n_expected += n;
+ cstp->expected = pool_realloc (specs->pool,
+ cstp->expected,
+ sizeof(double) *
+ cstp->n_expected);
+ for ( i = cstp->n_expected - n ;
+ i < cstp->n_expected;
+ ++i )
+ cstp->expected[i] = f;
+
+ }
+ }
+ }
+ else
+ lex_put_back (lexer, '/');
+ }
+
+ if ( cstp->ranged && cstp->n_expected > 0 &&
+ cstp->n_expected != cstp->hi - cstp->lo + 1 )
+ {
+ msg(ME,
+ _("%d expected values were given, but the specified "
+ "range (%d-%d) requires exactly %d values."),
+ cstp->n_expected, cstp->lo, cstp->hi,
+ cstp->hi - cstp->lo +1);
+ return 0;
+ }
+
+ specs->n_tests++;
+ specs->test = pool_realloc (specs->pool,
+ specs->test,
+ sizeof(*specs->test) * specs->n_tests);
+
+ specs->test[specs->n_tests - 1] = (struct npar_test *) tp;
+
+ return 1;
+}
+
+
+int
+npar_custom_binomial(struct lexer *lexer, struct dataset *ds, struct cmd_npar_tests *cmd UNUSED, void *aux)
+{
+ struct npar_specs *specs = aux;
+ struct binomial_test *btp = pool_alloc(specs->pool, sizeof(*btp));
+ struct one_sample_test *tp = (struct one_sample_test *) btp;
+
+ ((struct npar_test *)tp)->execute = binomial_execute;
+ ((struct npar_test *)tp)->insert_variables = one_sample_insert_variables;
+
+ btp->category1 = btp->category2 = btp->cutpoint = SYSMIS;
+
+ if ( lex_match(lexer, '(') )
+ {
+ if ( lex_force_num (lexer) )
+ {
+ btp->p = lex_number (lexer);
+ lex_get (lexer);
+ lex_force_match (lexer, ')');
+ }
+ else
+ return 0;
+ }
+
+ if ( lex_match (lexer, '=') )
+ {
+ if (parse_variables_pool (lexer, specs->pool, dataset_dict (ds),
+ &tp->vars, &tp->n_vars,
+ PV_NUMERIC | PV_NO_SCRATCH | PV_NO_DUPLICATE) )
+ {
+ if ( lex_match (lexer, '('))
+ {
+ lex_force_num (lexer);
+ btp->category1 = lex_number (lexer);
+ lex_get (lexer);
+ if ( ! lex_force_match (lexer, ',')) return 2;
+ if ( ! lex_force_num (lexer) ) return 2;
+ btp->category2 = lex_number (lexer);
+ lex_get (lexer);
+ lex_force_match (lexer, ')');
+ }
+ }
+ else
+ return 2;
+ }
+ else
+ {
+ if ( lex_match (lexer, '(') )
+ {
+ lex_force_num (lexer);
+ btp->cutpoint = lex_number (lexer);
+ lex_get (lexer);
+ lex_force_match (lexer, ')');
+ }
+ }
+
+ specs->n_tests++;
+ specs->test = pool_realloc (specs->pool,
+ specs->test,
+ sizeof(*specs->test) * specs->n_tests);
+
+ specs->test[specs->n_tests - 1] = (struct npar_test *) tp;
+
+ return 1;
+}
+
+
+bool parse_two_sample_related_test (struct lexer *lexer,
+ const struct dictionary *dict,
+ struct cmd_npar_tests *cmd,
+ struct two_sample_test *test_parameters,
+ struct pool *pool
+ );
+
+
+bool
+parse_two_sample_related_test (struct lexer *lexer,
+ const struct dictionary *dict,
+ struct cmd_npar_tests *cmd UNUSED,
+ struct two_sample_test *test_parameters,
+ struct pool *pool
+ )
+{
+ int n = 0;
+ bool paired = false;
+ bool with = false;
+ struct variable **vlist1;
+ size_t n_vlist1;
+
+ struct variable **vlist2;
+ size_t n_vlist2;
+
+ if (!parse_variables_pool (lexer, pool,
+ dict,
+ &vlist1, &n_vlist1,
+ PV_NUMERIC | PV_NO_SCRATCH | PV_NO_DUPLICATE) )
+ return false;
+
+ if ( lex_match(lexer, T_WITH))
+ {
+ with = true;
+ if ( !parse_variables_pool (lexer, pool, dict,
+ &vlist2, &n_vlist2,
+ PV_NUMERIC | PV_NO_SCRATCH | PV_NO_DUPLICATE) )
+ return false;
+
+ paired = (lex_match (lexer, '(') &&
+ lex_match_id (lexer, "PAIRED") && lex_match (lexer, ')'));
+ }
+
+
+ if ( with )
+ {
+ if (paired)
+ {
+ if ( n_vlist1 != n_vlist2)
+ msg (SE, _("PAIRED was specified but the number of variables "
+ "preceding WITH (%d) did not match the number "
+ "following (%d)."), n_vlist1, n_vlist2);
+
+ test_parameters->n_pairs = n_vlist1 ;
+ }
+ else
+ {
+ test_parameters->n_pairs = n_vlist1 * n_vlist2;
+ }
+ }
+ else
+ {
+ test_parameters->n_pairs = (n_vlist1 * (n_vlist1 - 1)) / 2 ;
+ }
+
+ test_parameters->pairs =
+ pool_alloc (pool, sizeof ( variable_pair) * test_parameters->n_pairs);
+
+ if ( with )
+ {
+ if (paired)
+ {
+ int i;
+ assert (n_vlist1 == n_vlist2);
+ for ( i = 0 ; i < n_vlist1; ++i )
+ {
+ test_parameters->pairs[n][0] = vlist1[i];
+ test_parameters->pairs[n][1] = vlist2[i];
+ n++;
+ }
+ }
+ else
+ {
+ int i,j;
+ for ( i = 0 ; i < n_vlist1; ++i )
+ {
+ for ( j = 0 ; j < n_vlist2; ++j )
+ {
+ test_parameters->pairs[n][0] = vlist1[i];
+ test_parameters->pairs[n][1] = vlist2[j];
+ n++;
+ }
+ }
+ }
+ }
+ else
+ {
+ int i,j;
+ for ( i = 0 ; i < n_vlist1 - 1; ++i )
+ {
+ for ( j = i + 1 ; j < n_vlist1; ++j )
+ {
+ assert ( n < test_parameters->n_pairs);
+ test_parameters->pairs[n][0] = vlist1[i];
+ test_parameters->pairs[n][1] = vlist1[j];
+ n++;
+ }
+ }
+ }
+
+ assert ( n == test_parameters->n_pairs);
+
+ return true;
+}
+
+int
+npar_custom_wilcoxon (struct lexer *lexer,
+ struct dataset *ds,
+ struct cmd_npar_tests *cmd, void *aux )
+{
+ struct npar_specs *specs = aux;
+
+ struct two_sample_test *tp = pool_alloc(specs->pool, sizeof(*tp));
+ ((struct npar_test *)tp)->execute = NULL;
+
+ if (!parse_two_sample_related_test (lexer, dataset_dict (ds), cmd,
+ tp, specs->pool) )
+ return 0;
+
+ specs->n_tests++;
+ specs->test = pool_realloc (specs->pool,
+ specs->test,
+ sizeof(*specs->test) * specs->n_tests);
+ specs->test[specs->n_tests - 1] = (struct npar_test *) tp;
+
+ return 1;
+}
+
+int
+npar_custom_mcnemar (struct lexer *lexer,
+ struct dataset *ds,
+ struct cmd_npar_tests *cmd, void *aux )
+{
+ struct npar_specs *specs = aux;
+
+ struct two_sample_test *tp = pool_alloc(specs->pool, sizeof(*tp));
+ ((struct npar_test *)tp)->execute = NULL;
+
+
+ if (!parse_two_sample_related_test (lexer, dataset_dict (ds),
+ cmd, tp, specs->pool) )
+ return 0;
+
+ specs->n_tests++;
+ specs->test = pool_realloc (specs->pool,
+ specs->test,
+ sizeof(*specs->test) * specs->n_tests);
+ specs->test[specs->n_tests - 1] = (struct npar_test *) tp;
+
+ return 1;
+}
+
+int
+npar_custom_sign (struct lexer *lexer, struct dataset *ds,
+ struct cmd_npar_tests *cmd, void *aux )
+{
+ struct npar_specs *specs = aux;
+
+ struct two_sample_test *tp = pool_alloc(specs->pool, sizeof(*tp));
+ ((struct npar_test *)tp)->execute = NULL;
+
+
+ if (!parse_two_sample_related_test (lexer, dataset_dict (ds), cmd,
+ tp, specs->pool) )
+ return 0;
+
+ specs->n_tests++;
+ specs->test = pool_realloc (specs->pool,
+ specs->test,
+ sizeof(*specs->test) * specs->n_tests);
+ specs->test[specs->n_tests - 1] = (struct npar_test *) tp;
+
+ return 1;
+}
+
+/* Insert the variables for TEST into VAR_HASH */
+void
+one_sample_insert_variables (const struct npar_test *test,
+ struct hsh_table *var_hash)
+{
+ int i;
+ struct one_sample_test *ost = (struct one_sample_test *) test;
+
+ for ( i = 0 ; i < ost->n_vars ; ++i )
+ hsh_insert (var_hash, ost->vars[i]);
+}
+
projects / pspp-builds.git / blobdiff