pintos-os.org Git - pspp/blob - src/language/dictionary/attributes.c

   1 /* PSPP - a program for statistical analysis.
   2    Copyright (C) 2008 Free Software Foundation, Inc.
   3
   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.
   8
   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.
  13
  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/>. */
  16
  17 #include <config.h>
  18
  19 #include <stdlib.h>
  20
  21 #include <data/attributes.h>
  22 #include <data/dictionary.h>
  23 #include <data/procedure.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>
  29
  30 #include "xalloc.h"
  31
  32 #include "gettext.h"
  33 #define _(msgid) gettext (msgid)
  34
  35 static enum cmd_result parse_attributes (struct lexer *, struct attrset **,
  36                                          size_t n);
  37
  38 /* Parses the DATAFILE ATTRIBUTE command. */
  39 int
  40 cmd_datafile_attribute (struct lexer *lexer, struct dataset *ds)
  41 {
  42   struct attrset *set = dict_get_attributes (dataset_dict (ds));
  43   return parse_attributes (lexer, &set, 1);
  44 }
  45
  46 /* Parses the VARIABLE ATTRIBUTE command. */
  47 int
  48 cmd_variable_attribute (struct lexer *lexer, struct dataset *ds)
  49 {
  50   do
  51     {
  52       struct variable **vars;
  53       struct attrset **sets;
  54       size_t n_vars, i;
  55       bool ok;
  56
  57       if (!lex_force_match_id (lexer, "VARIABLES")
  58           || !lex_force_match (lexer, '=')
  59           || !parse_variables (lexer, dataset_dict (ds), &vars, &n_vars,
  60                                PV_NONE))
  61         return CMD_FAILURE;
  62
  63       sets = xmalloc (n_vars * sizeof *sets);
  64       for (i = 0; i < n_vars; i++)
  65         sets[i] = var_get_attributes (vars[i]);
  66
  67       ok = parse_attributes (lexer, sets, n_vars);
  68       free (vars);
  69       free (sets);
  70       if (!ok)
  71         return CMD_FAILURE;
  72     }
  73   while (lex_match (lexer, '/'));
  74
  75   return lex_end_of_command (lexer);
  76 }
  77
  78 static bool
  79 match_subcommand (struct lexer *lexer, const char *keyword)
  80 {
  81   if (lex_token (lexer) == T_ID
  82       && lex_id_match (ss_cstr (lex_tokid (lexer)), ss_cstr (keyword))
  83       && lex_look_ahead (lexer) == '=')
  84     {
  85       lex_get (lexer);          /* Skip keyword. */
  86       lex_get (lexer);          /* Skip '='. */
  87       return true;
  88     }
  89   else
  90     return false;
  91 }
  92
  93 static bool
  94 parse_attribute_name (struct lexer *lexer, char name[VAR_NAME_LEN + 1],
  95                       size_t *index)
  96 {
  97   if (!lex_force_id (lexer))
  98     return false;
  99   strcpy (name, lex_tokid (lexer));
 100   lex_get (lexer);
 101
 102   if (lex_match (lexer, '['))
 103     {
 104       if (!lex_force_int (lexer))
 105         return false;
 106       if (lex_integer (lexer) < 1 || lex_integer (lexer) > 65535)
 107         {
 108           msg (SE, _("Attribute array index must be between 1 and 65535."));
 109           return false;
 110         }
 111       *index = lex_integer (lexer);
 112       lex_get (lexer);
 113       if (!lex_force_match (lexer, ']'))
 114         return false;
 115     }
 116   else
 117     *index = 0;
 118   return true;
 119 }
 120
 121 static bool
 122 add_attribute (struct lexer *lexer, struct attrset **sets, size_t n)
 123 {
 124   char name[VAR_NAME_LEN + 1];
 125   size_t index, i;
 126   char *value;
 127
 128   if (!parse_attribute_name (lexer, name, &index)
 129       || !lex_force_match (lexer, '(')
 130       || !lex_force_string (lexer))
 131     return false;
 132   value = ds_cstr (lex_tokstr (lexer));
 133
 134   for (i = 0; i < n; i++)
 135     {
 136       struct attribute *attr = attrset_lookup (sets[i], name);
 137       if (attr == NULL)
 138         {
 139           attr = attribute_create (name);
 140           attrset_add (sets[i], attr);
 141         }
 142       attribute_set_value (attr, index ? index - 1 : 0, value);
 143     }
 144
 145   lex_get (lexer);
 146   return lex_force_match (lexer, ')');
 147 }
 148
 149 static bool
 150 delete_attribute (struct lexer *lexer, struct attrset **sets, size_t n)
 151 {
 152   char name[VAR_NAME_LEN + 1];
 153   size_t index, i;
 154
 155   if (!parse_attribute_name (lexer, name, &index))
 156     return false;
 157
 158   for (i = 0; i < n; i++)
 159     {
 160       struct attrset *set = sets[i];
 161       if (index == 0)
 162         attrset_delete (set, name);
 163       else
 164         {
 165           struct attribute *attr = attrset_lookup (set, name);
 166           if (attr != NULL)
 167             {
 168               attribute_del_value (attr, index - 1);
 169               if (attribute_get_n_values (attr) == 0)
 170                 attrset_delete (set, name);
 171             }
 172         }
 173     }
 174   return true;
 175 }
 176
 177 static enum cmd_result
 178 parse_attributes (struct lexer *lexer, struct attrset **sets, size_t n)
 179 {
 180   enum { UNKNOWN, ADD, DELETE } command = UNKNOWN;
 181   do
 182     {
 183       if (match_subcommand (lexer, "ATTRIBUTE"))
 184         command = ADD;
 185       else if (match_subcommand (lexer, "DELETE"))
 186         command = DELETE;
 187       else if (command == UNKNOWN)
 188         {
 189           lex_error (lexer, _("expecting ATTRIBUTE= or DELETE="));
 190           return CMD_FAILURE;
 191         }
 192
 193       if (!(command == ADD
 194             ? add_attribute (lexer, sets, n)
 195             : delete_attribute (lexer, sets, n)))
 196         return CMD_FAILURE;
 197     }
 198   while (lex_token (lexer) != '/' && lex_token (lexer) != '.');
 199   return CMD_SUCCESS;
 200 }