1 /* PSPP - a program for statistical analysis.
2 Copyright (C) 1997-9, 2000, 2006, 2007, 2009, 2010, 2011, 2012 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/expressions/private.h"
24 #include "libpspp/assertion.h"
25 #include "libpspp/message.h"
26 #include "language/expressions/helpers.h"
27 #include "language/lexer/format-parser.h"
28 #include "language/lexer/value-parser.h"
29 #include "libpspp/pool.h"
30 #include "output/driver.h"
35 expr_evaluate (struct expression *e, const struct ccase *c, int case_idx,
38 struct dataset *ds = e->ds;
39 union operation_data *op = e->ops;
41 double *ns = e->number_stack;
42 struct substring *ss = e->string_stack;
44 /* Without a dictionary/dataset, the expression can't refer to variables,
45 and you don't need to specify a case when you evaluate the
46 expression. With a dictionary/dataset, the expression can refer
47 to variables, so you must specify a case when you evaluate the
49 assert ((c != NULL) == (e->ds != NULL));
51 pool_clear (e->eval_pool);
55 assert (op < e->ops + e->n_ops);
56 switch (op++->operation)
65 const struct substring *s = &op++->string;
66 *ss++ = copy_string (e, s->string, s->length);
70 case OP_return_number:
71 *(double *) result = isfinite (ns[-1]) ? ns[-1] : SYSMIS;
74 case OP_return_string:
75 *(struct substring *) result = ss[-1];
78 #include "evaluate.inc"
87 expr_evaluate_num (struct expression *e, const struct ccase *c, int case_idx)
91 assert (e->type == OP_number || e->type == OP_boolean);
92 expr_evaluate (e, c, case_idx, &d);
97 expr_evaluate_str (struct expression *e, const struct ccase *c, int case_idx,
98 char *dst, size_t dst_size)
102 assert (e->type == OP_string);
103 assert ((dst == NULL) == (dst_size == 0));
104 expr_evaluate (e, c, case_idx, &s);
106 buf_copy_rpad (dst, dst_size, s.string, s.length, ' ');
109 #include "language/lexer/lexer.h"
110 #include "language/command.h"
112 static bool default_optimize = true;
115 cmd_debug_evaluate (struct lexer *lexer, struct dataset *dsother UNUSED)
117 bool optimize = default_optimize;
118 int retval = CMD_FAILURE;
119 bool dump_postfix = false;
120 bool set_defaults = false;
122 struct ccase *c = NULL;
124 struct dataset *ds = NULL;
128 struct fmt_spec format;
129 bool has_format = false;
131 struct expression *expr;
133 struct dictionary *d = NULL;
137 if (lex_match_id (lexer, "NOOPTIMIZE"))
139 else if (lex_match_id (lexer, "OPTIMIZE"))
141 else if (lex_match_id (lexer, "POSTFIX"))
143 else if (lex_match_id (lexer, "SET"))
145 else if (lex_match (lexer, T_LPAREN))
149 if (!lex_force_id (lexer))
151 int name_ofs = lex_ofs (lexer);
152 name = xstrdup (lex_tokcstr (lexer));
155 if (!lex_force_match (lexer, T_EQUALS))
160 if (lex_is_number (lexer))
163 value.f = lex_number (lexer);
166 else if (lex_match_id (lexer, "SYSMIS"))
171 else if (lex_is_string (lexer))
173 width = ss_length (lex_tokss (lexer));
174 value.s = CHAR_CAST (uint8_t *, ss_xstrdup (lex_tokss (lexer)));
179 lex_error (lexer, _("Syntax error expecting number or string."));
185 ds = dataset_create (NULL, "");
186 d = dataset_dict (ds);
189 v = dict_create_var (d, name, width);
192 lex_ofs_error (lexer, name_ofs, name_ofs,
193 _("Duplicate variable name %s."), name);
194 value_destroy (&value, width);
200 if (lex_match_id (lexer, "MISSING"))
202 struct missing_values mv;
203 mv_init (&mv, width);
204 mv_add_value (&mv, &value);
205 var_set_missing_values (v, &mv);
210 c = case_create (dict_get_proto (d));
212 c = case_unshare_and_resize (c, dict_get_proto (d));
213 value_swap (case_data_rw (c, v), &value);
214 value_destroy (&value, width);
216 if (!lex_force_match (lexer, T_RPAREN))
219 else if (lex_match_id (lexer, "VECTOR"))
221 struct variable **vars;
223 dict_get_vars_mutable (d, &vars, &n, 0);
224 dict_create_vector_assert (d, "V", vars, n);
227 else if (lex_match_id (lexer, "FORMAT"))
229 lex_match (lexer, T_EQUALS);
230 if (!parse_format_specifier (lexer, &format))
232 char *error = fmt_check_output__ (format);
234 error = fmt_check_type_compat__ (format, NULL, VAL_NUMERIC);
237 lex_next_error (lexer, -1, -1, "%s", error);
249 retval = CMD_SUCCESS;
250 default_optimize = optimize;
254 if (!lex_force_match (lexer, T_SLASH))
257 for (size_t i = 1; ; i++)
258 if (lex_next_token (lexer, i) == T_ENDCMD)
260 title = lex_next_representation (lexer, 0, i - 1);
264 expr = expr_parse_any (lexer, ds, optimize);
265 if (!expr || lex_end_of_command (lexer) != CMD_SUCCESS)
269 output_log ("%s => error", title);
274 expr_debug_print_postfix (expr);
279 case OP_num_vec_elem:
281 double d = expr_evaluate_num (expr, c, 0);
284 char *output = data_out (&(const union value) { .f = d },
286 settings_get_fmt_settings ());
287 output_log ("%s => %s", title, output);
290 else if (d == SYSMIS)
291 output_log ("%s => sysmis", title);
293 output_log ("%s => %.2f", title, d);
299 double b = expr_evaluate_num (expr, c, 0);
300 output_log ("%s => %s", title,
301 b == SYSMIS ? "sysmis" : b == 0.0 ? "false" : "true");
307 struct substring out;
308 expr_evaluate (expr, c, 0, &out);
309 output_log ("%s => \"%.*s\"", title, (int) out.length, out.string);
318 retval = CMD_SUCCESS;
321 dataset_destroy (ds);
332 expr_debug_print_postfix (const struct expression *e)
334 struct string s = DS_EMPTY_INITIALIZER;
336 for (size_t i = 0; i < e->n_ops; i++)
338 union operation_data *op = &e->ops[i];
340 ds_put_byte (&s, ' ');
341 switch (e->op_types[i])
344 if (op->operation == OP_return_number)
345 ds_put_cstr (&s, "return_number");
346 else if (op->operation == OP_return_string)
347 ds_put_cstr (&s, "return_string");
348 else if (is_function (op->operation))
349 ds_put_format (&s, "%s", operations[op->operation].prototype);
350 else if (is_composite (op->operation))
351 ds_put_format (&s, "%s", operations[op->operation].name);
353 ds_put_format (&s, "%s:", operations[op->operation].name);
356 if (op->number != SYSMIS)
357 ds_put_format (&s, "n<%g>", op->number);
359 ds_put_cstr (&s, "n<SYSMIS>");
362 ds_put_cstr (&s, "s<");
363 ds_put_substring (&s, op->string);
364 ds_put_byte (&s, '>');
368 char str[FMT_STRING_LEN_MAX + 1];
369 fmt_to_string (op->format, str);
370 ds_put_format (&s, "f<%s>", str);
374 ds_put_format (&s, "v<%s>", var_get_name (op->variable));
377 ds_put_format (&s, "vec<%s>", vector_get_name (op->vector));
380 ds_put_format (&s, "i<%d>", op->integer);
383 ds_put_cstr (&s, "expr_node");
389 output_log_nocopy (ds_steal_cstr (&s));