1 /* PSPP - a program for statistical analysis. -*-c-*-
2 Copyright (C) 2006, 2008, 2009, 2010, 2011, 2016 Free Software Foundation, Inc.
4 This program is free software: you can redistribute it and/or modify
5 it under the terms of the GNU General Public License as published by
6 the Free Software Foundation, either version 3 of the License, or
7 (at your option) any later version.
9 This program is distributed in the hope that it will be useful,
10 but WITHOUT ANY WARRANTY; without even the implied warranty of
11 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
12 GNU General Public License for more details.
14 You should have received a copy of the GNU General Public License
15 along with this program. If not, see <http://www.gnu.org/licenses/>. */
19 #include "language/stats/npar.h"
24 #include "data/case.h"
25 #include "data/casegrouper.h"
26 #include "data/casereader.h"
27 #include "data/dataset.h"
28 #include "data/dictionary.h"
29 #include "data/settings.h"
30 #include "data/variable.h"
31 #include "language/command.h"
32 #include "language/lexer/lexer.h"
33 #include "language/lexer/value-parser.h"
34 #include "language/lexer/variable-parser.h"
35 #include "language/stats/binomial.h"
36 #include "language/stats/chisquare.h"
37 #include "language/stats/ks-one-sample.h"
38 #include "language/stats/cochran.h"
39 #include "language/stats/friedman.h"
40 #include "language/stats/jonckheere-terpstra.h"
41 #include "language/stats/kruskal-wallis.h"
42 #include "language/stats/mann-whitney.h"
43 #include "language/stats/mcnemar.h"
44 #include "language/stats/median.h"
45 #include "language/stats/npar-summary.h"
46 #include "language/stats/runs.h"
47 #include "language/stats/sign.h"
48 #include "language/stats/wilcoxon.h"
49 #include "libpspp/array.h"
50 #include "libpspp/assertion.h"
51 #include "libpspp/cast.h"
52 #include "libpspp/hash-functions.h"
53 #include "libpspp/hmapx.h"
54 #include "libpspp/message.h"
55 #include "libpspp/pool.h"
56 #include "libpspp/str.h"
57 #include "libpspp/taint.h"
58 #include "math/moments.h"
60 #include "gl/xalloc.h"
63 #define _(msgid) gettext (msgid)
65 /* Settings for subcommand specifiers. */
72 /* Array indices for STATISTICS subcommand. */
75 NPAR_ST_DESCRIPTIVES = 0,
76 NPAR_ST_QUARTILES = 1,
81 /* NPAR TESTS structure. */
84 /* Count variables indicating how many
85 of the subcommands have been given. */
99 int jonckheere_terpstra;
104 /* How missing values should be treated */
107 /* Which statistics have been requested */
108 int a_statistics[NPAR_ST_count];
115 struct npar_test **test;
118 const struct variable **vv; /* Compendium of all variables
119 (those mentioned on ANY subcommand */
120 int n_vars; /* Number of variables in vv */
122 enum mv_class filter; /* Missing values to filter. */
124 bool descriptives; /* Descriptive statistics should be calculated */
125 bool quartiles; /* Quartiles should be calculated */
127 bool exact; /* Whether exact calculations have been requested */
128 double timer; /* Maximum time (in minutes) to wait for exact calculations */
132 /* Prototype for custom subcommands of NPAR TESTS. */
133 static int npar_chisquare (struct lexer *, struct dataset *, struct npar_specs *);
134 static int npar_binomial (struct lexer *, struct dataset *, struct npar_specs *);
135 static int npar_ks_one_sample (struct lexer *, struct dataset *, struct npar_specs *);
136 static int npar_runs (struct lexer *, struct dataset *, struct npar_specs *);
137 static int npar_friedman (struct lexer *, struct dataset *, struct npar_specs *);
138 static int npar_kendall (struct lexer *, struct dataset *, struct npar_specs *);
139 static int npar_cochran (struct lexer *, struct dataset *, struct npar_specs *);
140 static int npar_wilcoxon (struct lexer *, struct dataset *, struct npar_specs *);
141 static int npar_sign (struct lexer *, struct dataset *, struct npar_specs *);
142 static int npar_kruskal_wallis (struct lexer *, struct dataset *, struct npar_specs *);
143 static int npar_jonckheere_terpstra (struct lexer *, struct dataset *, struct npar_specs *);
144 static int npar_mann_whitney (struct lexer *, struct dataset *, struct npar_specs *);
145 static int npar_mcnemar (struct lexer *, struct dataset *, struct npar_specs *);
146 static int npar_median (struct lexer *, struct dataset *, struct npar_specs *);
148 static int npar_method (struct lexer *, struct npar_specs *);
150 /* Command parsing functions. */
151 static int parse_npar_tests (struct lexer *lexer, struct dataset *ds, struct cmd_npar_tests *p,
152 struct npar_specs *npar_specs);
155 parse_npar_tests (struct lexer *lexer, struct dataset *ds, struct cmd_npar_tests *npt,
156 struct npar_specs *nps)
161 npt->ks_one_sample = 0;
167 npt->kruskal_wallis = 0;
168 npt->mann_whitney = 0;
171 npt->jonckheere_terpstra = 0;
173 npt->miss = MISS_ANALYSIS;
178 memset (npt->a_statistics, 0, sizeof npt->a_statistics);
181 if (lex_match_id (lexer, "COCHRAN"))
184 switch (npar_cochran (lexer, ds, nps))
191 lex_error (lexer, NULL);
197 else if (lex_match_id (lexer, "FRIEDMAN"))
200 switch (npar_friedman (lexer, ds, nps))
207 lex_error (lexer, NULL);
213 else if (lex_match_id (lexer, "KENDALL"))
216 switch (npar_kendall (lexer, ds, nps))
223 lex_error (lexer, NULL);
229 else if (lex_match_id (lexer, "RUNS"))
232 switch (npar_runs (lexer, ds, nps))
239 lex_error (lexer, NULL);
245 else if (lex_match_id (lexer, "CHISQUARE"))
247 lex_match (lexer, T_EQUALS);
249 switch (npar_chisquare (lexer, ds, nps))
256 lex_error (lexer, NULL);
264 else if (lex_match_id (lexer, "BINOMIAL"))
266 lex_match (lexer, T_EQUALS);
268 switch (npar_binomial (lexer, ds, nps))
275 lex_error (lexer, NULL);
281 else if (lex_match_phrase (lexer, "K-S") ||
282 lex_match_phrase (lexer, "KOLMOGOROV-SMIRNOV"))
284 lex_match (lexer, T_EQUALS);
285 npt->ks_one_sample++;
286 switch (npar_ks_one_sample (lexer, ds, nps))
293 lex_error (lexer, NULL);
299 else if (lex_match_phrase (lexer, "J-T") ||
300 lex_match_phrase (lexer, "JONCKHEERE-TERPSTRA"))
302 lex_match (lexer, T_EQUALS);
303 npt->jonckheere_terpstra++;
304 switch (npar_jonckheere_terpstra (lexer, ds, nps))
311 lex_error (lexer, NULL);
317 else if (lex_match_phrase (lexer, "K-W") ||
318 lex_match_phrase (lexer, "KRUSKAL-WALLIS"))
320 lex_match (lexer, T_EQUALS);
321 npt->kruskal_wallis++;
322 switch (npar_kruskal_wallis (lexer, ds, nps))
329 lex_error (lexer, NULL);
335 else if (lex_match_phrase (lexer, "MCNEMAR"))
337 lex_match (lexer, T_EQUALS);
339 switch (npar_mcnemar (lexer, ds, nps))
346 lex_error (lexer, NULL);
352 else if (lex_match_phrase (lexer, "M-W") ||
353 lex_match_phrase (lexer, "MANN-WHITNEY"))
355 lex_match (lexer, T_EQUALS);
357 switch (npar_mann_whitney (lexer, ds, nps))
364 lex_error (lexer, NULL);
370 else if (lex_match_phrase (lexer, "MEDIAN"))
374 switch (npar_median (lexer, ds, nps))
381 lex_error (lexer, NULL);
387 else if (lex_match_id (lexer, "WILCOXON"))
389 lex_match (lexer, T_EQUALS);
391 switch (npar_wilcoxon (lexer, ds, nps))
398 lex_error (lexer, NULL);
404 else if (lex_match_id (lexer, "SIGN"))
406 lex_match (lexer, T_EQUALS);
408 switch (npar_sign (lexer, ds, nps))
415 lex_error (lexer, NULL);
421 else if (lex_match_id (lexer, "MISSING"))
423 lex_match (lexer, T_EQUALS);
425 if (npt->missing > 1)
427 lex_sbc_only_once (lexer, "MISSING");
430 while (lex_token (lexer) != T_SLASH && lex_token (lexer) != T_ENDCMD)
432 if (lex_match_id (lexer, "ANALYSIS"))
433 npt->miss = MISS_ANALYSIS;
434 else if (lex_match_id (lexer, "LISTWISE"))
435 npt->miss = MISS_LISTWISE;
436 else if (lex_match_id (lexer, "INCLUDE"))
437 nps->filter = MV_SYSTEM;
438 else if (lex_match_id (lexer, "EXCLUDE"))
439 nps->filter = MV_ANY;
442 lex_error (lexer, NULL);
445 lex_match (lexer, T_COMMA);
448 else if (lex_match_id (lexer, "METHOD"))
450 lex_match (lexer, T_EQUALS);
454 lex_sbc_only_once (lexer, "METHOD");
457 switch (npar_method (lexer, nps))
464 lex_error (lexer, NULL);
470 else if (lex_match_id (lexer, "STATISTICS"))
472 lex_match (lexer, T_EQUALS);
474 while (lex_token (lexer) != T_SLASH && lex_token (lexer) != T_ENDCMD)
476 if (lex_match_id (lexer, "DESCRIPTIVES"))
477 npt->a_statistics[NPAR_ST_DESCRIPTIVES] = 1;
478 else if (lex_match_id (lexer, "QUARTILES"))
479 npt->a_statistics[NPAR_ST_QUARTILES] = 1;
480 else if (lex_match (lexer, T_ALL))
481 npt->a_statistics[NPAR_ST_ALL] = 1;
484 lex_error (lexer, NULL);
487 lex_match (lexer, T_COMMA);
490 else if (settings_get_syntax () != COMPATIBLE && lex_match_id (lexer, "ALGORITHM"))
492 lex_match (lexer, T_EQUALS);
493 if (lex_match_id (lexer, "COMPATIBLE"))
494 settings_set_cmd_algorithm (COMPATIBLE);
495 else if (lex_match_id (lexer, "ENHANCED"))
496 settings_set_cmd_algorithm (ENHANCED);
498 if (!lex_match (lexer, T_SLASH))
501 if (lex_token (lexer) != T_ENDCMD)
503 lex_error (lexer, _("Syntax error expecting end of command."));
514 static void one_sample_insert_variables (const struct npar_test *test,
517 static void two_sample_insert_variables (const struct npar_test *test,
520 static void n_sample_insert_variables (const struct npar_test *test,
524 npar_execute (struct casereader *input,
525 const struct npar_specs *specs,
526 const struct dataset *ds)
529 struct descriptives *summary_descriptives = NULL;
531 for (t = 0 ; t < specs->n_tests; ++t)
533 const struct npar_test *test = specs->test[t];
534 if (NULL == test->execute)
536 msg (SW, _("%s subcommand not currently implemented."), "NPAR");
539 test->execute (ds, casereader_clone (input), specs->filter, test, specs->exact, specs->timer);
542 if (specs->descriptives && specs->n_vars > 0)
544 summary_descriptives = xnmalloc (sizeof (*summary_descriptives),
547 npar_summary_calc_descriptives (summary_descriptives,
548 casereader_clone (input),
550 specs->vv, specs->n_vars,
554 if ((specs->descriptives || specs->quartiles)
555 && !taint_has_tainted_successor (casereader_get_taint (input)))
556 do_summary_box (summary_descriptives, specs->vv, specs->n_vars,
557 dict_get_weight_format (dataset_dict (ds)));
559 free (summary_descriptives);
560 casereader_destroy (input);
564 cmd_npar_tests (struct lexer *lexer, struct dataset *ds)
566 struct cmd_npar_tests cmd;
569 struct npar_specs npar_specs = {0, 0, 0, 0, 0, 0, 0, 0, 0, 0};
570 struct casegrouper *grouper;
571 struct casereader *input, *group;
572 struct hmapx var_map = HMAPX_INITIALIZER (var_map);
575 npar_specs.pool = pool_create ();
576 npar_specs.filter = MV_ANY;
577 npar_specs.n_vars = -1;
578 npar_specs.vv = NULL;
580 if (! parse_npar_tests (lexer, ds, &cmd, &npar_specs))
582 pool_destroy (npar_specs.pool);
586 for (i = 0; i < npar_specs.n_tests; ++i)
588 const struct npar_test *test = npar_specs.test[i];
589 test->insert_variables (test, &var_map);
593 struct hmapx_node *node;
594 struct variable *var;
595 npar_specs.n_vars = 0;
597 HMAPX_FOR_EACH (var, node, &var_map)
599 npar_specs.n_vars ++;
600 npar_specs.vv = pool_nrealloc (npar_specs.pool, npar_specs.vv, npar_specs.n_vars, sizeof (*npar_specs.vv));
601 npar_specs.vv[npar_specs.n_vars - 1] = var;
605 sort (npar_specs.vv, npar_specs.n_vars, sizeof (*npar_specs.vv),
606 compare_var_ptrs_by_name, NULL);
612 for (i = 0 ; i < NPAR_ST_count; ++i)
614 if (cmd.a_statistics[i])
618 case NPAR_ST_DESCRIPTIVES:
619 npar_specs.descriptives = true;
621 case NPAR_ST_QUARTILES:
622 npar_specs.quartiles = true;
625 npar_specs.quartiles = true;
626 npar_specs.descriptives = true;
635 input = proc_open (ds);
636 if (cmd.miss == MISS_LISTWISE)
638 input = casereader_create_filter_missing (input,
646 grouper = casegrouper_create_splits (input, dataset_dict (ds));
647 while (casegrouper_get_next_group (grouper, &group))
648 npar_execute (group, &npar_specs, ds);
649 ok = casegrouper_destroy (grouper);
650 ok = proc_commit (ds) && ok;
652 pool_destroy (npar_specs.pool);
653 hmapx_destroy (&var_map);
655 return ok ? CMD_SUCCESS : CMD_CASCADING_FAILURE;
659 npar_runs (struct lexer *lexer, struct dataset *ds,
660 struct npar_specs *specs)
662 struct runs_test *rt = pool_alloc (specs->pool, sizeof (*rt));
663 struct one_sample_test *tp = &rt->parent;
664 struct npar_test *nt = &tp->parent;
666 nt->execute = runs_execute;
667 nt->insert_variables = one_sample_insert_variables;
669 if (lex_force_match (lexer, T_LPAREN))
671 if (lex_match_id (lexer, "MEAN"))
673 rt->cp_mode = CP_MEAN;
675 else if (lex_match_id (lexer, "MEDIAN"))
677 rt->cp_mode = CP_MEDIAN;
679 else if (lex_match_id (lexer, "MODE"))
681 rt->cp_mode = CP_MODE;
683 else if (lex_is_number (lexer))
685 rt->cutpoint = lex_number (lexer);
686 rt->cp_mode = CP_CUSTOM;
691 lex_error (lexer, _("Syntax error expecting %s, %s, %s or a number."),
692 "MEAN", "MEDIAN", "MODE");
696 if (! lex_force_match (lexer, T_RPAREN))
699 if (! lex_force_match (lexer, T_EQUALS))
702 if (!parse_variables_const_pool (lexer, specs->pool, dataset_dict (ds),
703 &tp->vars, &tp->n_vars,
704 PV_NO_SCRATCH | PV_NO_DUPLICATE | PV_NUMERIC))
711 specs->test = pool_realloc (specs->pool,
713 sizeof (*specs->test) * specs->n_tests);
715 specs->test[specs->n_tests - 1] = nt;
721 npar_friedman (struct lexer *lexer, struct dataset *ds,
722 struct npar_specs *specs)
724 struct friedman_test *ft = pool_alloc (specs->pool, sizeof (*ft));
725 struct one_sample_test *ost = &ft->parent;
726 struct npar_test *nt = &ost->parent;
728 ft->kendalls_w = false;
729 nt->execute = friedman_execute;
730 nt->insert_variables = one_sample_insert_variables;
732 lex_match (lexer, T_EQUALS);
734 if (!parse_variables_const_pool (lexer, specs->pool, dataset_dict (ds),
735 &ost->vars, &ost->n_vars,
736 PV_NO_SCRATCH | PV_NO_DUPLICATE | PV_NUMERIC))
742 specs->test = pool_realloc (specs->pool,
744 sizeof (*specs->test) * specs->n_tests);
746 specs->test[specs->n_tests - 1] = nt;
752 npar_kendall (struct lexer *lexer, struct dataset *ds,
753 struct npar_specs *specs)
755 struct friedman_test *kt = pool_alloc (specs->pool, sizeof (*kt));
756 struct one_sample_test *ost = &kt->parent;
757 struct npar_test *nt = &ost->parent;
759 kt->kendalls_w = true;
760 nt->execute = friedman_execute;
761 nt->insert_variables = one_sample_insert_variables;
763 lex_match (lexer, T_EQUALS);
765 if (!parse_variables_const_pool (lexer, specs->pool, dataset_dict (ds),
766 &ost->vars, &ost->n_vars,
767 PV_NO_SCRATCH | PV_NO_DUPLICATE | PV_NUMERIC))
773 specs->test = pool_realloc (specs->pool,
775 sizeof (*specs->test) * specs->n_tests);
777 specs->test[specs->n_tests - 1] = nt;
784 npar_cochran (struct lexer *lexer, struct dataset *ds,
785 struct npar_specs *specs)
787 struct one_sample_test *ft = pool_alloc (specs->pool, sizeof (*ft));
788 struct npar_test *nt = &ft->parent;
790 nt->execute = cochran_execute;
791 nt->insert_variables = one_sample_insert_variables;
793 lex_match (lexer, T_EQUALS);
795 if (!parse_variables_const_pool (lexer, specs->pool, dataset_dict (ds),
796 &ft->vars, &ft->n_vars,
797 PV_NO_SCRATCH | PV_NO_DUPLICATE | PV_NUMERIC))
803 specs->test = pool_realloc (specs->pool,
805 sizeof (*specs->test) * specs->n_tests);
807 specs->test[specs->n_tests - 1] = nt;
814 npar_chisquare (struct lexer *lexer, struct dataset *ds,
815 struct npar_specs *specs)
817 struct chisquare_test *cstp = pool_alloc (specs->pool, sizeof (*cstp));
818 struct one_sample_test *tp = &cstp->parent;
819 struct npar_test *nt = &tp->parent;
822 nt->execute = chisquare_execute;
823 nt->insert_variables = one_sample_insert_variables;
825 if (!parse_variables_const_pool (lexer, specs->pool, dataset_dict (ds),
826 &tp->vars, &tp->n_vars,
827 PV_NO_SCRATCH | PV_NO_DUPLICATE))
832 cstp->ranged = false;
834 if (lex_match (lexer, T_LPAREN))
837 if (! lex_force_num (lexer)) return 0;
838 cstp->lo = lex_number (lexer);
840 if (! lex_force_match (lexer, T_COMMA)) return 0;
841 if (! lex_force_num (lexer)) return 0;
842 cstp->hi = lex_number (lexer);
843 if (cstp->lo >= cstp->hi)
846 _("The specified value of HI (%d) is "
847 "lower than the specified value of LO (%d)"),
852 if (! lex_force_match (lexer, T_RPAREN)) return 0;
855 cstp->n_expected = 0;
856 cstp->expected = NULL;
857 if (lex_match_phrase (lexer, "/EXPECTED"))
859 if (! lex_force_match (lexer, T_EQUALS)) return 0;
860 if (! lex_match_id (lexer, "EQUAL"))
864 while (lex_is_number (lexer))
868 f = lex_number (lexer);
870 if (lex_match (lexer, T_ASTERISK))
873 if (!lex_force_num (lexer))
875 f = lex_number (lexer);
878 lex_match (lexer, T_COMMA);
880 cstp->n_expected += n;
881 cstp->expected = pool_realloc (specs->pool,
885 for (i = cstp->n_expected - n ;
886 i < cstp->n_expected;
888 cstp->expected[i] = f;
894 if (cstp->ranged && cstp->n_expected > 0 &&
895 cstp->n_expected != cstp->hi - cstp->lo + 1)
898 _("%d expected values were given, but the specified "
899 "range (%d-%d) requires exactly %d values."),
900 cstp->n_expected, cstp->lo, cstp->hi,
901 cstp->hi - cstp->lo +1);
906 specs->test = pool_realloc (specs->pool,
908 sizeof (*specs->test) * specs->n_tests);
910 specs->test[specs->n_tests - 1] = nt;
917 npar_binomial (struct lexer *lexer, struct dataset *ds,
918 struct npar_specs *specs)
920 struct binomial_test *btp = pool_alloc (specs->pool, sizeof (*btp));
921 struct one_sample_test *tp = &btp->parent;
922 struct npar_test *nt = &tp->parent;
925 nt->execute = binomial_execute;
926 nt->insert_variables = one_sample_insert_variables;
928 btp->category1 = btp->category2 = btp->cutpoint = SYSMIS;
932 if (lex_match (lexer, T_LPAREN))
935 if (lex_force_num (lexer))
937 btp->p = lex_number (lexer);
939 if (!lex_force_match (lexer, T_RPAREN))
949 if (!lex_force_match (lexer, T_EQUALS))
953 if (parse_variables_const_pool (lexer, specs->pool, dataset_dict (ds),
954 &tp->vars, &tp->n_vars,
955 PV_NUMERIC | PV_NO_SCRATCH | PV_NO_DUPLICATE))
957 if (lex_match (lexer, T_LPAREN))
959 if (! lex_force_num (lexer))
961 btp->category1 = lex_number (lexer);
963 if (lex_match (lexer, T_COMMA))
965 if (! lex_force_num (lexer)) return 2;
966 btp->category2 = lex_number (lexer);
971 btp->cutpoint = btp->category1;
974 if (! lex_force_match (lexer, T_RPAREN))
985 specs->test = pool_realloc (specs->pool,
987 sizeof (*specs->test) * specs->n_tests);
989 specs->test[specs->n_tests - 1] = nt;
997 ks_one_sample_parse_params (struct lexer *lexer, struct ks_one_sample_test *kst, int params)
999 assert (params == 1 || params == 2);
1001 if (lex_is_number (lexer))
1003 kst->p[0] = lex_number (lexer);
1008 lex_match (lexer, T_COMMA);
1009 if (lex_force_num (lexer))
1011 kst->p[1] = lex_number (lexer);
1019 npar_ks_one_sample (struct lexer *lexer, struct dataset *ds, struct npar_specs *specs)
1021 struct ks_one_sample_test *kst = pool_alloc (specs->pool, sizeof (*kst));
1022 struct one_sample_test *tp = &kst->parent;
1023 struct npar_test *nt = &tp->parent;
1025 nt->execute = ks_one_sample_execute;
1026 nt->insert_variables = one_sample_insert_variables;
1028 kst->p[0] = kst->p[1] = SYSMIS;
1030 if (! lex_force_match (lexer, T_LPAREN))
1033 if (lex_match_id (lexer, "NORMAL"))
1035 kst->dist = KS_NORMAL;
1036 ks_one_sample_parse_params (lexer, kst, 2);
1038 else if (lex_match_id (lexer, "POISSON"))
1040 kst->dist = KS_POISSON;
1041 ks_one_sample_parse_params (lexer, kst, 1);
1043 else if (lex_match_id (lexer, "UNIFORM"))
1045 kst->dist = KS_UNIFORM;
1046 ks_one_sample_parse_params (lexer, kst, 2);
1048 else if (lex_match_id (lexer, "EXPONENTIAL"))
1050 kst->dist = KS_EXPONENTIAL;
1051 ks_one_sample_parse_params (lexer, kst, 1);
1056 if (! lex_force_match (lexer, T_RPAREN))
1059 lex_match (lexer, T_EQUALS);
1061 if (! parse_variables_const_pool (lexer, specs->pool, dataset_dict (ds),
1062 &tp->vars, &tp->n_vars,
1063 PV_NUMERIC | PV_NO_SCRATCH | PV_NO_DUPLICATE))
1067 specs->test = pool_realloc (specs->pool,
1069 sizeof (*specs->test) * specs->n_tests);
1071 specs->test[specs->n_tests - 1] = nt;
1078 parse_two_sample_related_test (struct lexer *lexer,
1079 const struct dictionary *dict,
1080 struct two_sample_test *test_parameters,
1084 bool paired = false;
1086 const struct variable **vlist1;
1089 const struct variable **vlist2;
1092 test_parameters->parent.insert_variables = two_sample_insert_variables;
1094 if (!parse_variables_const_pool (lexer, pool,
1097 PV_NUMERIC | PV_NO_SCRATCH | PV_DUPLICATE))
1100 if (lex_match (lexer, T_WITH))
1103 if (!parse_variables_const_pool (lexer, pool, dict,
1105 PV_NUMERIC | PV_NO_SCRATCH | PV_DUPLICATE))
1108 paired = (lex_match (lexer, T_LPAREN) &&
1109 lex_match_id (lexer, "PAIRED") && lex_match (lexer, T_RPAREN));
1117 if (n_vlist1 != n_vlist2)
1119 msg (SE, _("PAIRED was specified but the number of variables "
1120 "preceding WITH (%zu) did not match the number "
1121 "following (%zu)."), n_vlist1, n_vlist2);
1125 test_parameters->n_pairs = n_vlist1 ;
1129 test_parameters->n_pairs = n_vlist1 * n_vlist2;
1134 test_parameters->n_pairs = (n_vlist1 * (n_vlist1 - 1)) / 2 ;
1137 test_parameters->pairs =
1138 pool_alloc (pool, sizeof (variable_pair) * test_parameters->n_pairs);
1145 assert (n_vlist1 == n_vlist2);
1146 for (i = 0 ; i < n_vlist1; ++i)
1148 test_parameters->pairs[n][0] = vlist1[i];
1149 test_parameters->pairs[n][1] = vlist2[i];
1156 for (i = 0 ; i < n_vlist1; ++i)
1158 for (j = 0 ; j < n_vlist2; ++j)
1160 test_parameters->pairs[n][0] = vlist1[i];
1161 test_parameters->pairs[n][1] = vlist2[j];
1170 for (i = 0 ; i < n_vlist1 - 1; ++i)
1172 for (j = i + 1 ; j < n_vlist1; ++j)
1174 assert (n < test_parameters->n_pairs);
1175 test_parameters->pairs[n][0] = vlist1[i];
1176 test_parameters->pairs[n][1] = vlist1[j];
1182 assert (n == test_parameters->n_pairs);
1189 parse_n_sample_related_test (struct lexer *lexer,
1190 const struct dictionary *dict,
1191 struct n_sample_test *nst,
1195 if (!parse_variables_const_pool (lexer, pool,
1197 &nst->vars, &nst->n_vars,
1198 PV_NUMERIC | PV_NO_SCRATCH | PV_NO_DUPLICATE))
1201 if (! lex_force_match (lexer, T_BY))
1204 nst->indep_var = parse_variable_const (lexer, dict);
1205 if (!nst->indep_var)
1208 if (! lex_force_match (lexer, T_LPAREN))
1211 value_init (&nst->val1, var_get_width (nst->indep_var));
1212 if (! parse_value (lexer, &nst->val1, nst->indep_var))
1214 value_destroy (&nst->val1, var_get_width (nst->indep_var));
1218 lex_match (lexer, T_COMMA);
1220 value_init (&nst->val2, var_get_width (nst->indep_var));
1221 if (! parse_value (lexer, &nst->val2, nst->indep_var))
1223 value_destroy (&nst->val2, var_get_width (nst->indep_var));
1227 if (! lex_force_match (lexer, T_RPAREN))
1234 npar_wilcoxon (struct lexer *lexer,
1236 struct npar_specs *specs)
1238 struct two_sample_test *tp = pool_alloc (specs->pool, sizeof (*tp));
1239 struct npar_test *nt = &tp->parent;
1240 nt->execute = wilcoxon_execute;
1242 if (!parse_two_sample_related_test (lexer, dataset_dict (ds),
1247 specs->test = pool_realloc (specs->pool,
1249 sizeof (*specs->test) * specs->n_tests);
1250 specs->test[specs->n_tests - 1] = nt;
1257 npar_mann_whitney (struct lexer *lexer,
1259 struct npar_specs *specs)
1261 struct n_sample_test *tp = pool_alloc (specs->pool, sizeof (*tp));
1262 struct npar_test *nt = &tp->parent;
1264 nt->insert_variables = n_sample_insert_variables;
1265 nt->execute = mann_whitney_execute;
1267 if (!parse_n_sample_related_test (lexer, dataset_dict (ds),
1272 specs->test = pool_realloc (specs->pool,
1274 sizeof (*specs->test) * specs->n_tests);
1275 specs->test[specs->n_tests - 1] = nt;
1282 npar_median (struct lexer *lexer,
1284 struct npar_specs *specs)
1286 struct median_test *mt = pool_alloc (specs->pool, sizeof (*mt));
1287 struct n_sample_test *tp = &mt->parent;
1288 struct npar_test *nt = &tp->parent;
1290 mt->median = SYSMIS;
1292 if (lex_match (lexer, T_LPAREN) && lex_force_num (lexer))
1294 mt->median = lex_number (lexer);
1296 if (! lex_force_match (lexer, T_RPAREN))
1300 lex_match (lexer, T_EQUALS);
1302 nt->insert_variables = n_sample_insert_variables;
1303 nt->execute = median_execute;
1305 if (!parse_n_sample_related_test (lexer, dataset_dict (ds),
1310 specs->test = pool_realloc (specs->pool,
1312 sizeof (*specs->test) * specs->n_tests);
1313 specs->test[specs->n_tests - 1] = nt;
1320 npar_sign (struct lexer *lexer, struct dataset *ds,
1321 struct npar_specs *specs)
1323 struct two_sample_test *tp = pool_alloc (specs->pool, sizeof (*tp));
1324 struct npar_test *nt = &tp->parent;
1326 nt->execute = sign_execute;
1328 if (!parse_two_sample_related_test (lexer, dataset_dict (ds),
1333 specs->test = pool_realloc (specs->pool,
1335 sizeof (*specs->test) * specs->n_tests);
1336 specs->test[specs->n_tests - 1] = nt;
1343 npar_mcnemar (struct lexer *lexer, struct dataset *ds,
1344 struct npar_specs *specs)
1346 struct two_sample_test *tp = pool_alloc (specs->pool, sizeof (*tp));
1347 struct npar_test *nt = &tp->parent;
1349 nt->execute = mcnemar_execute;
1351 if (!parse_two_sample_related_test (lexer, dataset_dict (ds),
1356 specs->test = pool_realloc (specs->pool,
1358 sizeof (*specs->test) * specs->n_tests);
1359 specs->test[specs->n_tests - 1] = nt;
1366 npar_jonckheere_terpstra (struct lexer *lexer, struct dataset *ds,
1367 struct npar_specs *specs)
1369 struct n_sample_test *tp = pool_alloc (specs->pool, sizeof (*tp));
1370 struct npar_test *nt = &tp->parent;
1372 nt->insert_variables = n_sample_insert_variables;
1374 nt->execute = jonckheere_terpstra_execute;
1376 if (!parse_n_sample_related_test (lexer, dataset_dict (ds),
1381 specs->test = pool_realloc (specs->pool,
1383 sizeof (*specs->test) * specs->n_tests);
1384 specs->test[specs->n_tests - 1] = nt;
1390 npar_kruskal_wallis (struct lexer *lexer, struct dataset *ds,
1391 struct npar_specs *specs)
1393 struct n_sample_test *tp = pool_alloc (specs->pool, sizeof (*tp));
1394 struct npar_test *nt = &tp->parent;
1396 nt->insert_variables = n_sample_insert_variables;
1398 nt->execute = kruskal_wallis_execute;
1400 if (!parse_n_sample_related_test (lexer, dataset_dict (ds),
1405 specs->test = pool_realloc (specs->pool,
1407 sizeof (*specs->test) * specs->n_tests);
1408 specs->test[specs->n_tests - 1] = nt;
1414 insert_variable_into_map (struct hmapx *var_map, const struct variable *var)
1416 size_t hash = hash_pointer (var, 0);
1417 struct hmapx_node *node;
1418 const struct variable *v = NULL;
1420 HMAPX_FOR_EACH_WITH_HASH (v, node, hash, var_map)
1426 hmapx_insert (var_map, CONST_CAST (struct variable *, var), hash);
1429 /* Insert the variables for TEST into VAR_MAP */
1431 one_sample_insert_variables (const struct npar_test *test,
1432 struct hmapx *var_map)
1435 const struct one_sample_test *ost = UP_CAST (test, const struct one_sample_test, parent);
1437 for (i = 0 ; i < ost->n_vars ; ++i)
1438 insert_variable_into_map (var_map, ost->vars[i]);
1443 two_sample_insert_variables (const struct npar_test *test,
1444 struct hmapx *var_map)
1447 const struct two_sample_test *tst = UP_CAST (test, const struct two_sample_test, parent);
1449 for (i = 0 ; i < tst->n_pairs ; ++i)
1451 variable_pair *pair = &tst->pairs[i];
1453 insert_variable_into_map (var_map, (*pair)[0]);
1454 insert_variable_into_map (var_map, (*pair)[1]);
1459 n_sample_insert_variables (const struct npar_test *test,
1460 struct hmapx *var_map)
1463 const struct n_sample_test *tst = UP_CAST (test, const struct n_sample_test, parent);
1465 for (i = 0 ; i < tst->n_vars ; ++i)
1466 insert_variable_into_map (var_map, tst->vars[i]);
1468 insert_variable_into_map (var_map, tst->indep_var);
1473 npar_method (struct lexer *lexer, struct npar_specs *specs)
1475 if (lex_match_id (lexer, "EXACT"))
1477 specs->exact = true;
1479 if (lex_match_id (lexer, "TIMER"))
1483 if (lex_match (lexer, T_LPAREN))
1485 if (lex_force_num (lexer))
1487 specs->timer = lex_number (lexer);
1490 if (lex_force_match (lexer, T_RPAREN))