1 /* PSPP - a program for statistical analysis.
2 Copyright (C) 2008, 2010, 2011 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/attributes.h"
22 #include "data/dataset.h"
23 #include "data/dictionary.h"
24 #include "data/variable.h"
25 #include "language/command.h"
26 #include "language/lexer/lexer.h"
27 #include "language/lexer/variable-parser.h"
28 #include "libpspp/message.h"
30 #include "gl/xalloc.h"
33 #define _(msgid) gettext (msgid)
35 static enum cmd_result parse_attributes (struct lexer *,
36 const char *dict_encoding,
37 struct attrset **, size_t n);
39 /* Parses the DATAFILE ATTRIBUTE command. */
41 cmd_datafile_attribute (struct lexer *lexer, struct dataset *ds)
43 struct dictionary *dict = dataset_dict (ds);
44 struct attrset *set = dict_get_attributes (dict);
45 return parse_attributes (lexer, dict_get_encoding (dict), &set, 1);
48 /* Parses the VARIABLE ATTRIBUTE command. */
50 cmd_variable_attribute (struct lexer *lexer, struct dataset *ds)
52 struct dictionary *dict = dataset_dict (ds);
53 const char *dict_encoding = dict_get_encoding (dict);
57 struct variable **vars;
58 struct attrset **sets;
62 if (!lex_force_match_phrase (lexer, "VARIABLES=")
63 || !parse_variables (lexer, dict, &vars, &n_vars, PV_NONE))
66 sets = xmalloc (n_vars * sizeof *sets);
67 for (i = 0; i < n_vars; i++)
68 sets[i] = var_get_attributes (vars[i]);
70 ok = parse_attributes (lexer, dict_encoding, sets, n_vars);
76 while (lex_match (lexer, T_SLASH));
81 /* Parses an attribute name and verifies that it is valid in DICT_ENCODING,
82 optionally followed by an index inside square brackets. Returns the
83 attribute name or NULL if there was a parse error. Stores the index into
86 parse_attribute_name (struct lexer *lexer, const char *dict_encoding,
89 if (!lex_force_id (lexer))
91 char *error = id_is_valid__ (lex_tokcstr (lexer), dict_encoding);
94 lex_error (lexer, "%s", error);
98 char *name = xstrdup (lex_tokcstr (lexer));
101 if (lex_match (lexer, T_LBRACK))
103 if (!lex_force_int_range (lexer, NULL, 1, 65535))
105 *index = lex_integer (lexer);
107 if (!lex_force_match (lexer, T_RBRACK))
120 add_attribute (struct lexer *lexer, const char *dict_encoding,
121 struct attrset **sets, size_t n)
127 name = parse_attribute_name (lexer, dict_encoding, &index);
130 if (!lex_force_match (lexer, T_LPAREN) || !lex_force_string (lexer))
135 value = lex_tokcstr (lexer);
137 for (i = 0; i < n; i++)
139 struct attribute *attr = attrset_lookup (sets[i], name);
142 attr = attribute_create (name);
143 attrset_add (sets[i], attr);
145 attribute_set_value (attr, index ? index - 1 : 0, value);
150 return lex_force_match (lexer, T_RPAREN);
154 delete_attribute (struct lexer *lexer, const char *dict_encoding,
155 struct attrset **sets, size_t n)
160 name = parse_attribute_name (lexer, dict_encoding, &index);
164 for (i = 0; i < n; i++)
166 struct attrset *set = sets[i];
168 attrset_delete (set, name);
171 struct attribute *attr = attrset_lookup (set, name);
174 attribute_del_value (attr, index - 1);
175 if (attribute_get_n_values (attr) == 0)
176 attrset_delete (set, name);
185 static enum cmd_result
186 parse_attributes (struct lexer *lexer, const char *dict_encoding,
187 struct attrset **sets, size_t n)
189 enum { UNKNOWN, ADD, DELETE } command = UNKNOWN;
192 if (lex_match_phrase (lexer, "ATTRIBUTE="))
194 else if (lex_match_phrase (lexer, "DELETE="))
196 else if (command == UNKNOWN)
198 lex_error_expecting (lexer, "ATTRIBUTE=", "DELETE=");
203 ? add_attribute (lexer, dict_encoding, sets, n)
204 : delete_attribute (lexer, dict_encoding, sets, n)))
207 while (lex_token (lexer) != T_SLASH && lex_token (lexer) != T_ENDCMD);