1 /* PSPP - a program for statistical analysis.
2 Copyright (C) 1997-9, 2000, 2009-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/>. */
21 #include "data/dataset.h"
22 #include "data/transformations.h"
23 #include "language/command.h"
24 #include "language/data-io/inpt-pgm.h"
25 #include "language/expressions/public.h"
26 #include "language/lexer/lexer.h"
27 #include "libpspp/compiler.h"
28 #include "libpspp/message.h"
29 #include "libpspp/str.h"
31 #include "gl/xalloc.h"
34 #define _(msgid) gettext (msgid)
36 /* A conditional clause. */
39 struct msg_location *location;
40 struct expression *condition; /* Test expression; NULL for ELSE clause. */
41 struct trns_chain xforms;
44 /* DO IF transformation. */
47 struct clause *clauses; /* Clauses. */
48 size_t n_clauses; /* Number of clauses. */
50 const struct trns_chain *resume;
54 static const struct trns_class do_if_trns_class;
57 start_clause (struct lexer *lexer, struct dataset *ds,
58 bool condition, struct do_if_trns *do_if,
59 size_t *allocated_clauses, bool *ok)
61 if (*ok && do_if->n_clauses > 0
62 && !do_if->clauses[do_if->n_clauses - 1].condition)
65 lex_ofs_error (lexer, 0, 1,
66 _("ELSE IF is not allowed following ELSE "
67 "within DO IF...END IF."));
69 lex_ofs_error (lexer, 0, 0,
70 _("Only one ELSE is allowed within DO IF...END IF."));
72 msg_at (SN, do_if->clauses[do_if->n_clauses - 1].location,
73 _("This is the location of the previous ELSE clause."));
75 msg_at (SN, do_if->clauses[0].location,
76 _("This is the location of the DO IF command."));
81 if (do_if->n_clauses >= *allocated_clauses)
82 do_if->clauses = x2nrealloc (do_if->clauses, allocated_clauses,
83 sizeof *do_if->clauses);
84 struct clause *clause = &do_if->clauses[do_if->n_clauses++];
86 *clause = (struct clause) { .location = NULL };
89 clause->condition = expr_parse_bool (lexer, ds);
90 if (!clause->condition)
91 lex_discard_rest_of_command (lexer);
93 clause->location = lex_ofs_location (lexer, 0, lex_ofs (lexer));
95 lex_end_of_command (lexer);
98 proc_push_transformations (ds);
102 finish_clause (struct dataset *ds, struct do_if_trns *do_if)
104 struct clause *clause = &do_if->clauses[do_if->n_clauses - 1];
105 proc_pop_transformations (ds, &clause->xforms);
110 cmd_do_if (struct lexer *lexer, struct dataset *ds)
112 struct do_if_trns *do_if = xmalloc (sizeof *do_if);
113 *do_if = (struct do_if_trns) { .n_clauses = 0 };
115 size_t allocated_clauses = 0;
118 start_clause (lexer, ds, true, do_if, &allocated_clauses, &ok);
119 while (!lex_match_phrase (lexer, "END IF"))
121 if (lex_token (lexer) == T_STOP)
123 lex_error (lexer, NULL);
126 else if (lex_match_phrase (lexer, "ELSE IF"))
128 finish_clause (ds, do_if);
129 start_clause (lexer, ds, true, do_if, &allocated_clauses, &ok);
131 else if (lex_match_id (lexer, "ELSE"))
133 finish_clause (ds, do_if);
134 start_clause (lexer, ds, false, do_if, &allocated_clauses, &ok);
137 cmd_parse_in_state (lexer, ds,
139 ? CMD_STATE_NESTED_INPUT_PROGRAM
140 : CMD_STATE_NESTED_DATA));
142 finish_clause (ds, do_if);
144 add_transformation (ds, &do_if_trns_class, do_if);
146 return ok ? CMD_SUCCESS : CMD_FAILURE;
150 cmd_inside_do_if (struct lexer *lexer UNUSED, struct dataset *ds UNUSED)
152 msg (SE, _("This command cannot appear outside DO IF...END IF."));
156 static const struct trns_chain *
157 do_if_find_clause (const struct do_if_trns *do_if,
158 struct ccase *c, casenumber case_num)
160 for (size_t i = 0; i < do_if->n_clauses; i++)
162 const struct clause *clause = &do_if->clauses[i];
163 if (!clause->condition)
164 return &clause->xforms;
166 double boolean = expr_evaluate_num (clause->condition, c, case_num);
168 return boolean == SYSMIS ? NULL : &clause->xforms;
173 static enum trns_result
174 do_if_trns_proc (void *do_if_, struct ccase **c, casenumber case_num)
176 struct do_if_trns *do_if = do_if_;
178 const struct trns_chain *chain;
182 chain = do_if->resume;
184 do_if->resume = NULL;
189 chain = do_if_find_clause (do_if, *c, case_num);
191 return TRNS_CONTINUE;
195 for (size_t i = start; i < chain->n; i++)
197 const struct transformation *trns = &chain->xforms[i];
198 enum trns_result r = trns->class->execute (trns->aux, c, case_num);
211 do_if->resume = chain;
216 return TRNS_CONTINUE;
220 do_if_trns_free (void *do_if_)
222 struct do_if_trns *do_if = do_if_;
224 for (size_t i = 0; i < do_if->n_clauses; i++)
226 struct clause *clause = &do_if->clauses[i];
228 msg_location_destroy (clause->location);
229 expr_free (clause->condition);
231 trns_chain_uninit (&clause->xforms);
233 free (do_if->clauses);
238 static const struct trns_class do_if_trns_class = {
240 .execute = do_if_trns_proc,
241 .destroy = do_if_trns_free,