--- /dev/null
+/* PSPP - a program for statistical analysis.
+ Copyright (C) 2011, 2015 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 "libpspp/message.h"
+
+#include "data/casegrouper.h"
+#include "data/casereader.h"
+#include "data/dictionary.h"
+#include "data/dataset.h"
+#include "data/missing-values.h"
+
+#include "language/lexer/lexer.h"
+#include "language/command.h"
+#include "language/lexer/variable-parser.h"
+#include "language/lexer/value-parser.h"
+
+#include "gettext.h"
+#define _(msgid) gettext (msgid)
+
+#include "t-test.h"
+
+int
+cmd_t_test (struct lexer *lexer, struct dataset *ds)
+{
+ bool ok = false;
+
+ /* Variables pertaining to the paired mode */
+ const struct variable **v1 = NULL;
+ size_t n_v1 = 0;
+ const struct variable **v2 = NULL;
+ size_t n_v2 = 0;
+
+ size_t n_pairs = 0;
+ vp *pairs = NULL;
+
+ /* One sample mode */
+ double testval = SYSMIS;
+
+ /* Independent samples mode */
+ const struct variable *gvar;
+ union value gval0;
+ union value gval1;
+ int gval_width = -1;
+ bool cut = false;
+
+ const struct dictionary *dict = dataset_dict (ds);
+ struct tt tt = {
+ .wv = dict_get_weight (dict),
+ .dict = dict,
+ .confidence = 0.95,
+ .exclude = MV_ANY,
+ .missing_type = MISS_ANALYSIS,
+ .n_vars = 0,
+ .vars = NULL,
+ };
+
+ lex_match (lexer, T_EQUALS);
+
+ int mode_count = 0;
+ while (lex_token (lexer) != T_ENDCMD)
+ {
+ lex_match (lexer, T_SLASH);
+ if (lex_match_id (lexer, "TESTVAL"))
+ {
+ mode_count++;
+ tt.mode = MODE_SINGLE;
+ lex_match (lexer, T_EQUALS);
+ if (!lex_force_num (lexer))
+ goto exit;
+ testval = lex_number (lexer);
+ lex_get (lexer);
+ }
+ else if (lex_match_id (lexer, "GROUPS"))
+ {
+ mode_count++;
+ cut = false;
+ tt.mode = MODE_INDEP;
+ lex_match (lexer, T_EQUALS);
+
+ int groups_start = lex_ofs (lexer);
+ gvar = parse_variable (lexer, dict);
+ if (!gvar)
+ goto exit;
+
+ gval_width = var_get_width (gvar);
+ value_init (&gval0, gval_width);
+ value_init (&gval1, gval_width);
+
+ int n;
+ if (lex_match (lexer, T_LPAREN))
+ {
+ if (!parse_value (lexer, &gval0, gvar))
+ goto exit;
+ if (lex_token (lexer) != T_RPAREN)
+ {
+ lex_match (lexer, T_COMMA);
+ if (!parse_value (lexer, &gval1, gvar))
+ goto exit;
+ cut = false;
+ n = 2;
+ }
+ else
+ {
+ cut = true;
+ n = 1;
+ }
+
+ if (!lex_force_match (lexer, T_RPAREN))
+ goto exit;
+ }
+ else
+ {
+ gval0.f = 1.0;
+ gval1.f = 2.0;
+ cut = false;
+ n = 0;
+ }
+ int groups_end = lex_ofs (lexer) - 1;
+
+ if (n != 2 && var_is_alpha (gvar))
+ {
+ lex_ofs_error (lexer, groups_start, groups_end,
+ _("When applying %s to a string variable, two "
+ "values must be specified."), "GROUPS");
+ goto exit;
+ }
+ }
+ else if (lex_match_id (lexer, "PAIRS"))
+ {
+ bool with = false;
+ bool paired = false;
+
+ if (tt.n_vars > 0)
+ {
+ lex_next_error (lexer, -1, -1,
+ _("%s subcommand may not be used with %s."),
+ "VARIABLES", "PAIRS");
+ goto exit;
+ }
+
+ mode_count++;
+ tt.mode = MODE_PAIRED;
+ lex_match (lexer, T_EQUALS);
+
+ int vars_start = lex_ofs (lexer);
+ if (!parse_variables_const (lexer, dict,
+ &v1, &n_v1,
+ PV_NO_DUPLICATE | PV_NUMERIC))
+ goto exit;
+
+ if (lex_match (lexer, T_WITH))
+ {
+ with = true;
+ if (!parse_variables_const (lexer, dict,
+ &v2, &n_v2,
+ PV_NO_DUPLICATE | PV_NUMERIC))
+ goto exit;
+ int vars_end = lex_ofs (lexer) - 1;
+
+ if (lex_match_phrase (lexer, "(PAIRED)"))
+ {
+ paired = true;
+ if (n_v1 != n_v2)
+ {
+ lex_ofs_error (lexer, vars_start, vars_end,
+ _("PAIRED was specified, but the number "
+ "of variables preceding WITH (%zu) "
+ "does not match the number following "
+ "(%zu)."),
+ n_v1, n_v2);
+ goto exit;
+ }
+ }
+ }
+
+ n_pairs = (paired ? n_v1
+ : with ? n_v1 * n_v2
+ : (n_v1 * (n_v1 - 1)) / 2.0);
+ pairs = xcalloc (n_pairs, sizeof *pairs);
+
+ if (paired)
+ {
+ for (size_t i = 0; i < n_v1; ++i)
+ {
+ vp *pair = &pairs[i];
+ (*pair)[0] = v1[i];
+ (*pair)[1] = v2[i];
+ }
+ }
+ else if (with)
+ {
+ int x = 0;
+ for (size_t i = 0; i < n_v1; ++i)
+ for (size_t j = 0; j < n_v2; ++j)
+ {
+ vp *pair = &pairs[x++];
+ (*pair)[0] = v1[i];
+ (*pair)[1] = v2[j];
+ }
+ }
+ else
+ {
+ int x = 0;
+ for (size_t i = 0; i < n_v1; ++i)
+ for (size_t j = i + 1; j < n_v1; ++j)
+ {
+ vp *pair = &pairs[x++];
+ (*pair)[0] = v1[i];
+ (*pair)[1] = v1[j];
+ }
+ }
+ }
+ else if (lex_match_id (lexer, "VARIABLES"))
+ {
+ if (mode_count && tt.mode == MODE_PAIRED)
+ {
+ lex_next_error (lexer, -1, -1,
+ _("%s subcommand may not be used with %s."),
+ "VARIABLES", "PAIRS");
+ goto exit;
+ }
+
+ lex_match (lexer, T_EQUALS);
+
+ if (!parse_variables_const (lexer, dict,
+ &tt.vars,
+ &tt.n_vars,
+ PV_NO_DUPLICATE | PV_NUMERIC))
+ goto exit;
+ }
+ else if (lex_match_id (lexer, "MISSING"))
+ {
+ lex_match (lexer, T_EQUALS);
+ while (lex_token (lexer) != T_ENDCMD && lex_token (lexer) != T_SLASH)
+ {
+ if (lex_match_id (lexer, "INCLUDE"))
+ tt.exclude = MV_SYSTEM;
+ else if (lex_match_id (lexer, "EXCLUDE"))
+ tt.exclude = MV_ANY;
+ else if (lex_match_id (lexer, "LISTWISE"))
+ tt.missing_type = MISS_LISTWISE;
+ else if (lex_match_id (lexer, "ANALYSIS"))
+ tt.missing_type = MISS_ANALYSIS;
+ else
+ {
+ lex_error_expecting (lexer, "INCLUDE", "EXCLUDE",
+ "LISTWISE", "ANALYSIS");
+ goto exit;
+ }
+ lex_match (lexer, T_COMMA);
+ }
+ }
+ else if (lex_match_id (lexer, "CRITERIA"))
+ {
+ lex_match (lexer, T_EQUALS);
+ if (!lex_match_id (lexer, "CIN") && !lex_match_id (lexer, "CI"))
+ {
+ lex_error_expecting (lexer, "CIN", "CI");
+ goto exit;
+ }
+ if (!lex_force_match (lexer, T_LPAREN))
+ goto exit;
+ if (!lex_force_num (lexer))
+ goto exit;
+ tt.confidence = lex_number (lexer);
+ lex_get (lexer);
+ if (!lex_force_match (lexer, T_RPAREN))
+ goto exit;
+ }
+ else
+ {
+ lex_error_expecting (lexer, "TESTVAL", "GROUPS", "PAIRS",
+ "VARIABLES", "MISSING", "CRITERIA");
+ goto exit;
+ }
+ }
+
+ if (mode_count != 1)
+ {
+ msg (SE, _("Exactly one of TESTVAL, GROUPS and PAIRS subcommands "
+ "must be specified."));
+ goto exit;
+ }
+
+ if (tt.n_vars == 0 && tt.mode != MODE_PAIRED)
+ {
+ lex_sbc_missing (lexer, "VARIABLES");
+ goto exit;
+ }
+
+ struct casereader *group;
+ struct casegrouper *grouper = casegrouper_create_splits (proc_open (ds), dict);
+ while (casegrouper_get_next_group (grouper, &group))
+ switch (tt.mode)
+ {
+ case MODE_SINGLE:
+ if (tt.missing_type == MISS_LISTWISE)
+ group = casereader_create_filter_missing (group, tt.vars, tt.n_vars,
+ tt.exclude, NULL, NULL);
+ one_sample_run (&tt, testval, group);
+ break;
+
+ case MODE_PAIRED:
+ if (tt.missing_type == MISS_LISTWISE)
+ {
+ group = casereader_create_filter_missing (group, v1, n_v1,
+ tt.exclude, NULL, NULL);
+ group = casereader_create_filter_missing (group, v2, n_v2,
+ tt.exclude, NULL, NULL);
+ }
+ paired_run (&tt, n_pairs, pairs, group);
+ break;
+
+ case MODE_INDEP:
+ if (tt.missing_type == MISS_LISTWISE)
+ {
+ group = casereader_create_filter_missing (group, tt.vars, tt.n_vars,
+ tt.exclude, NULL, NULL);
+ group = casereader_create_filter_missing (group, &gvar, 1,
+ tt.exclude, NULL, NULL);
+ }
+ indep_run (&tt, gvar, cut, &gval0, &gval1, group);
+ break;
+ }
+
+ ok = casegrouper_destroy (grouper);
+ ok = proc_commit (ds) && ok;
+
+exit:
+ if (gval_width != -1)
+ {
+ value_destroy (&gval0, gval_width);
+ value_destroy (&gval1, gval_width);
+ }
+ free (pairs);
+ free (v1);
+ free (v2);
+ free (tt.vars);
+
+ return ok ? CMD_SUCCESS : CMD_FAILURE;
+}
+
projects / pspp / blobdiff