src/language/dictionary/value-labels.c

   1 /* PSPP - computes sample statistics.
   2    Copyright (C) 1997-9, 2000 Free Software Foundation, Inc.
   3    Written by Ben Pfaff <blp@gnu.org>.
   4
   5    This program is free software; you can redistribute it and/or
   6    modify it under the terms of the GNU General Public License as
   7    published by the Free Software Foundation; either version 2 of the
   8    License, or (at your option) any later version.
   9
  10    This program is distributed in the hope that it will be useful, but
  11    WITHOUT ANY WARRANTY; without even the implied warranty of
  12    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  13    General Public License for more details.
  14
  15    You should have received a copy of the GNU General Public License
  16    along with this program; if not, write to the Free Software
  17    Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA
  18    02110-1301, USA. */
  19
  20 #include <config.h>
  21
  22 #include <stdio.h>
  23 #include <stdlib.h>
  24
  25 #include <data/procedure.h>
  26 #include <data/value-labels.h>
  27 #include <data/variable.h>
  28 #include <language/command.h>
  29 #include <language/lexer/lexer.h>
  30 #include <language/lexer/variable-parser.h>
  31 #include <libpspp/alloc.h>
  32 #include <libpspp/hash.h>
  33 #include <libpspp/message.h>
  34 #include <libpspp/str.h>
  35
  36 #include "gettext.h"
  37 #define _(msgid) gettext (msgid)
  38 \f
  39 /* Declarations. */
  40
  41 static int do_value_labels (int);
  42 static int verify_val_labs (struct variable **vars, size_t var_cnt);
  43 static void erase_labels (struct variable **vars, size_t var_cnt);
  44 static int get_label (struct variable **vars, size_t var_cnt);
  45 \f
  46 /* Stubs. */
  47
  48 int
  49 cmd_value_labels (void)
  50 {
  51   return do_value_labels (1);
  52 }
  53
  54 int
  55 cmd_add_value_labels (void)
  56 {
  57   return do_value_labels (0);
  58 }
  59 \f
  60 /* Do it. */
  61
  62 static int
  63 do_value_labels (int erase)
  64 {
  65   struct variable **vars; /* Variable list. */
  66   size_t var_cnt;         /* Number of variables. */
  67   int parse_err=0;        /* true if error parsing variables */
  68
  69   lex_match ('/');
  70
  71   while (token != '.')
  72     {
  73       parse_err = !parse_variables (default_dict, &vars, &var_cnt,
  74                                     PV_SAME_TYPE) ;
  75       if (var_cnt < 1)
  76         {
  77           free(vars);
  78           return CMD_FAILURE;
  79         }
  80       if (!verify_val_labs (vars, var_cnt))
  81         goto lossage;
  82       if (erase)
  83         erase_labels (vars, var_cnt);
  84       while (token != '/' && token != '.')
  85         if (!get_label (vars, var_cnt))
  86           goto lossage;
  87
  88       if (token != '/')
  89         {
  90         free (vars);
  91         break;
  92         }
  93
  94       lex_get ();
  95
  96       free (vars);
  97     }
  98
  99   if (parse_err)
 100     return CMD_FAILURE;
 101
 102   return lex_end_of_command ();
 103
 104  lossage:
 105   free (vars);
 106   return CMD_FAILURE;
 107 }
 108
 109 /* Verifies that none of the VAR_CNT variables in VARS are long
 110    string variables. */
 111 static int
 112 verify_val_labs (struct variable **vars, size_t var_cnt)
 113 {
 114   size_t i;
 115
 116   for (i = 0; i < var_cnt; i++)
 117     {
 118       struct variable *vp = vars[i];
 119
 120       if (vp->type == ALPHA && vp->width > MAX_SHORT_STRING)
 121         {
 122           msg (SE, _("It is not possible to assign value labels to long "
 123                      "string variables such as %s."), vp->name);
 124           return 0;
 125         }
 126     }
 127   return 1;
 128 }
 129
 130 /* Erases all the labels for the VAR_CNT variables in VARS. */
 131 static void
 132 erase_labels (struct variable **vars, size_t var_cnt)
 133 {
 134   size_t i;
 135
 136   /* Erase old value labels if desired. */
 137   for (i = 0; i < var_cnt; i++)
 138     val_labs_clear (vars[i]->val_labs);
 139 }
 140
 141 /* Parse all the labels for the VAR_CNT variables in VARS and add
 142    the specified labels to those variables.  */
 143 static int
 144 get_label (struct variable **vars, size_t var_cnt)
 145 {
 146   /* Parse all the labels and add them to the variables. */
 147   do
 148     {
 149       union value value;
 150       char *label;
 151       size_t i;
 152
 153       /* Set value. */
 154       if (vars[0]->type == ALPHA)
 155         {
 156           if (token != T_STRING)
 157             {
 158               lex_error (_("expecting string"));
 159               return 0;
 160             }
 161           buf_copy_str_rpad (value.s, MAX_SHORT_STRING, ds_cstr (&tokstr));
 162         }
 163       else
 164         {
 165           if (!lex_is_number ())
 166             {
 167               lex_error (_("expecting integer"));
 168               return 0;
 169             }
 170           if (!lex_is_integer ())
 171             msg (SW, _("Value label `%g' is not integer."), tokval);
 172           value.f = tokval;
 173         }
 174       lex_get ();
 175
 176       /* Set label. */
 177       if (!lex_force_string ())
 178         return 0;
 179       if (ds_length (&tokstr) > 60)
 180         {
 181           msg (SW, _("Truncating value label to 60 characters."));
 182           ds_truncate (&tokstr, 60);
 183         }
 184       label = ds_cstr (&tokstr);
 185
 186       for (i = 0; i < var_cnt; i++)
 187         val_labs_replace (vars[i]->val_labs, value, label);
 188
 189       lex_get ();
 190     }
 191   while (token != '/' && token != '.');
 192
 193   return 1;
 194 }