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