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 <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 <libpspp/alloc.h>
  31 #include <libpspp/hash.h>
  32 #include <libpspp/message.h>
  33 #include <libpspp/str.h>
  34
  35 #include "gettext.h"
  36 #define _(msgid) gettext (msgid)
  37 \f
  38 /* Declarations. */
  39
  40 static int do_value_labels (int);
  41 static int verify_val_labs (struct variable **vars, size_t var_cnt);
  42 static void erase_labels (struct variable **vars, size_t var_cnt);
  43 static int get_label (struct variable **vars, size_t var_cnt);
  44 \f
  45 /* Stubs. */
  46
  47 int
  48 cmd_value_labels (void)
  49 {
  50   return do_value_labels (1);
  51 }
  52
  53 int
  54 cmd_add_value_labels (void)
  55 {
  56   return do_value_labels (0);
  57 }
  58 \f
  59 /* Do it. */
  60
  61 static int
  62 do_value_labels (int erase)
  63 {
  64   struct variable **vars; /* Variable list. */
  65   size_t var_cnt;         /* Number of variables. */
  66   int parse_err=0;        /* true if error parsing variables */
  67
  68   lex_match ('/');
  69
  70   while (token != '.')
  71     {
  72       parse_err = !parse_variables (default_dict, &vars, &var_cnt,
  73                                     PV_SAME_TYPE) ;
  74       if (var_cnt < 1)
  75         {
  76           free(vars);
  77           return CMD_FAILURE;
  78         }
  79       if (!verify_val_labs (vars, var_cnt))
  80         goto lossage;
  81       if (erase)
  82         erase_labels (vars, var_cnt);
  83       while (token != '/' && token != '.')
  84         if (!get_label (vars, var_cnt))
  85           goto lossage;
  86
  87       if (token != '/')
  88         {
  89         free (vars);
  90         break;
  91         }
  92
  93       lex_get ();
  94
  95       free (vars);
  96     }
  97
  98   if (parse_err)
  99     return CMD_FAILURE;
 100
 101   return lex_end_of_command ();
 102
 103  lossage:
 104   free (vars);
 105   return CMD_FAILURE;
 106 }
 107
 108 /* Verifies that none of the VAR_CNT variables in VARS are long
 109    string variables. */
 110 static int
 111 verify_val_labs (struct variable **vars, size_t var_cnt)
 112 {
 113   size_t i;
 114
 115   for (i = 0; i < var_cnt; i++)
 116     {
 117       struct variable *vp = vars[i];
 118
 119       if (vp->type == ALPHA && vp->width > MAX_SHORT_STRING)
 120         {
 121           msg (SE, _("It is not possible to assign value labels to long "
 122                      "string variables such as %s."), vp->name);
 123           return 0;
 124         }
 125     }
 126   return 1;
 127 }
 128
 129 /* Erases all the labels for the VAR_CNT variables in VARS. */
 130 static void
 131 erase_labels (struct variable **vars, size_t var_cnt)
 132 {
 133   size_t i;
 134
 135   /* Erase old value labels if desired. */
 136   for (i = 0; i < var_cnt; i++)
 137     val_labs_clear (vars[i]->val_labs);
 138 }
 139
 140 /* Parse all the labels for the VAR_CNT variables in VARS and add
 141    the specified labels to those variables.  */
 142 static int
 143 get_label (struct variable **vars, size_t var_cnt)
 144 {
 145   /* Parse all the labels and add them to the variables. */
 146   do
 147     {
 148       union value value;
 149       char *label;
 150       size_t i;
 151
 152       /* Set value. */
 153       if (vars[0]->type == ALPHA)
 154         {
 155           if (token != T_STRING)
 156             {
 157               lex_error (_("expecting string"));
 158               return 0;
 159             }
 160           buf_copy_str_rpad (value.s, MAX_SHORT_STRING, ds_c_str (&tokstr));
 161         }
 162       else
 163         {
 164           if (!lex_is_number ())
 165             {
 166               lex_error (_("expecting integer"));
 167               return 0;
 168             }
 169           if (!lex_is_integer ())
 170             msg (SW, _("Value label `%g' is not integer."), tokval);
 171           value.f = tokval;
 172         }
 173       lex_get ();
 174
 175       /* Set label. */
 176       if (!lex_force_string ())
 177         return 0;
 178       if (ds_length (&tokstr) > 60)
 179         {
 180           msg (SW, _("Truncating value label to 60 characters."));
 181           ds_truncate (&tokstr, 60);
 182         }
 183       label = ds_c_str (&tokstr);
 184
 185       for (i = 0; i < var_cnt; i++)
 186         val_labs_replace (vars[i]->val_labs, value, label);
 187
 188       lex_get ();
 189     }
 190   while (token != '/' && token != '.');
 191
 192   return 1;
 193 }