1 /* PSPP - a program for statistical analysis.
2 Copyright (C) 1997-9, 2000, 2009-2013 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/>. */
21 #include "data/case.h"
22 #include "data/control-stack.h"
23 #include "data/dataset.h"
24 #include "data/transformations.h"
25 #include "data/value.h"
26 #include "language/command.h"
27 #include "language/expressions/public.h"
28 #include "language/lexer/lexer.h"
29 #include "libpspp/compiler.h"
30 #include "libpspp/message.h"
31 #include "libpspp/str.h"
33 #include "gl/xalloc.h"
36 #define _(msgid) gettext (msgid)
38 /* DO IF, ELSE IF, and ELSE are translated as a single
39 transformation that evaluates each condition and jumps to the
40 start of the appropriate block of transformations. Each block
41 of transformations (except for the last) ends with a
42 transformation that jumps past the remaining blocks.
44 So, the following code:
54 is effectively translated like this:
56 IF a GOTO 1, IF b GOTO 2, ELSE GOTO 3.
66 /* A conditional clause. */
69 struct expression *condition; /* Test expression; NULL for ELSE clause. */
70 int target_index; /* Transformation to jump to if true. */
73 /* DO IF transformation. */
76 struct dataset *ds; /* The dataset */
77 struct clause *clauses; /* Clauses. */
78 size_t clause_cnt; /* Number of clauses. */
79 int past_END_IF_index; /* Transformation just past last clause. */
82 static const struct ctl_class do_if_class;
84 static int parse_clause (struct lexer *, struct do_if_trns *, struct dataset *ds);
85 static void add_clause (struct do_if_trns *, struct expression *condition);
86 static void add_else (struct do_if_trns *);
88 static bool has_else (struct do_if_trns *);
89 static bool must_not_have_else (struct do_if_trns *);
90 static void close_do_if (void *do_if);
92 static trns_proc_func do_if_trns_proc, break_trns_proc;
93 static trns_free_func do_if_trns_free;
97 cmd_do_if (struct lexer *lexer, struct dataset *ds)
99 struct do_if_trns *do_if = xmalloc (sizeof *do_if);
100 do_if->clauses = NULL;
101 do_if->clause_cnt = 0;
104 ctl_stack_push (&do_if_class, do_if);
105 add_transformation (ds, do_if_trns_proc, do_if_trns_free, do_if);
107 return parse_clause (lexer, do_if, ds);
112 cmd_else_if (struct lexer *lexer, struct dataset *ds)
114 struct do_if_trns *do_if = ctl_stack_top (&do_if_class);
115 if (do_if == NULL || !must_not_have_else (do_if))
116 return CMD_CASCADING_FAILURE;
117 return parse_clause (lexer, do_if, ds);
122 cmd_else (struct lexer *lexer UNUSED, struct dataset *ds)
124 struct do_if_trns *do_if = ctl_stack_top (&do_if_class);
125 assert (ds == do_if->ds);
126 if (do_if == NULL || !must_not_have_else (do_if))
127 return CMD_CASCADING_FAILURE;
134 cmd_end_if (struct lexer *lexer UNUSED, struct dataset *ds)
136 struct do_if_trns *do_if = ctl_stack_top (&do_if_class);
139 return CMD_CASCADING_FAILURE;
141 assert (ds == do_if->ds);
142 ctl_stack_pop (do_if);
147 /* Closes out DO_IF, by adding a sentinel ELSE clause if
148 necessary and setting past_END_IF_index. */
150 close_do_if (void *do_if_)
152 struct do_if_trns *do_if = do_if_;
154 if (!has_else (do_if))
156 do_if->past_END_IF_index = next_transformation (do_if->ds);
159 /* Adds an ELSE clause to DO_IF pointing to the next
162 add_else (struct do_if_trns *do_if)
164 assert (!has_else (do_if));
165 add_clause (do_if, NULL);
168 /* Returns true if DO_IF does not yet have an ELSE clause.
169 Reports an error and returns false if it does already. */
171 must_not_have_else (struct do_if_trns *do_if)
173 if (has_else (do_if))
175 msg (SE, _("This command may not follow %s in %s ... %s."), "ELSE", "DO IF", "END IF");
182 /* Returns true if DO_IF already has an ELSE clause,
185 has_else (struct do_if_trns *do_if)
187 return (do_if->clause_cnt != 0
188 && do_if->clauses[do_if->clause_cnt - 1].condition == NULL);
191 /* Parses a DO IF or ELSE IF expression and appends the
192 corresponding clause to DO_IF. Checks for end of command and
193 returns a command return code. */
195 parse_clause (struct lexer *lexer, struct do_if_trns *do_if, struct dataset *ds)
197 struct expression *condition;
199 condition = expr_parse (lexer, ds, EXPR_BOOLEAN);
200 if (condition == NULL)
201 return CMD_CASCADING_FAILURE;
203 add_clause (do_if, condition);
208 /* Adds a clause to DO_IF that tests for the given CONDITION and,
209 if true, jumps to the set of transformations produced by
210 following commands. */
212 add_clause (struct do_if_trns *do_if, struct expression *condition)
214 struct clause *clause;
216 if (do_if->clause_cnt > 0)
217 add_transformation (do_if->ds, break_trns_proc, NULL, do_if);
219 do_if->clauses = xnrealloc (do_if->clauses,
220 do_if->clause_cnt + 1, sizeof *do_if->clauses);
221 clause = &do_if->clauses[do_if->clause_cnt++];
222 clause->condition = condition;
223 clause->target_index = next_transformation (do_if->ds);
226 /* DO IF transformation procedure.
227 Checks each clause and jumps to the appropriate
230 do_if_trns_proc (void *do_if_, struct ccase **c, casenumber case_num UNUSED)
232 struct do_if_trns *do_if = do_if_;
233 struct clause *clause;
235 for (clause = do_if->clauses; clause < do_if->clauses + do_if->clause_cnt;
238 if (clause->condition != NULL)
240 double boolean = expr_evaluate_num (clause->condition, *c, case_num);
242 return clause->target_index;
243 else if (boolean == SYSMIS)
244 return do_if->past_END_IF_index;
247 return clause->target_index;
249 return do_if->past_END_IF_index;
252 /* Frees a DO IF transformation. */
254 do_if_trns_free (void *do_if_)
256 struct do_if_trns *do_if = do_if_;
257 struct clause *clause;
259 for (clause = do_if->clauses; clause < do_if->clauses + do_if->clause_cnt;
261 expr_free (clause->condition);
262 free (do_if->clauses);
267 /* Breaks out of a DO IF construct. */
269 break_trns_proc (void *do_if_, struct ccase **c UNUSED,
270 casenumber case_num UNUSED)
272 struct do_if_trns *do_if = do_if_;
274 return do_if->past_END_IF_index;
277 /* DO IF control structure class definition. */
278 static const struct ctl_class do_if_class =