Constness and global namespace patrol.
[pspp] / 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 (dataset_dict (current_dataset), &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 }