Fix memory leak in AGGREGATE.
[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 <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 }