Remove "Written by Ben Pfaff <blp@gnu.org>" lines everywhere.
[pspp-builds.git] / src / language / dictionary / vector.c
1 /* PSPP - computes sample statistics.
2    Copyright (C) 1997-9, 2000 Free Software Foundation, Inc.
3
4    This program is free software; you can redistribute it and/or
5    modify it under the terms of the GNU General Public License as
6    published by the Free Software Foundation; either version 2 of the
7    License, or (at your option) any later version.
8
9    This program is distributed in the hope that it will be useful, but
10    WITHOUT ANY WARRANTY; without even the implied warranty of
11    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
12    General Public License for more details.
13
14    You should have received a copy of the GNU General Public License
15    along with this program; if not, write to the Free Software
16    Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA
17    02110-1301, USA. */
18
19 #include <config.h>
20
21 #include <stdlib.h>
22
23 #include <data/procedure.h>
24 #include <data/dictionary.h>
25 #include <data/variable.h>
26 #include <language/command.h>
27 #include <language/lexer/lexer.h>
28 #include <language/lexer/variable-parser.h>
29 #include <libpspp/alloc.h>
30 #include <libpspp/assertion.h>
31 #include <libpspp/message.h>
32 #include <libpspp/misc.h>
33 #include <libpspp/str.h>
34
35 #include "gettext.h"
36 #define _(msgid) gettext (msgid)
37
38 int
39 cmd_vector (struct lexer *lexer, struct dataset *ds)
40 {
41   /* Just to be different, points to a set of null terminated strings
42      containing the names of the vectors to be created.  The list
43      itself is terminated by a empty string.  So a list of three
44      elements, A B C, would look like this: "A\0B\0C\0\0". */
45   char *vecnames;
46
47   /* vecnames iterators. */
48   char *cp, *cp2;
49
50   /* Maximum allocated position for vecnames, plus one position. */
51   char *endp = NULL;
52
53   struct dictionary *dict = dataset_dict (ds);
54
55   cp = vecnames = xmalloc (256);
56   endp = &vecnames[256];
57   do
58     {
59       /* Get the name(s) of the new vector(s). */
60       if (!lex_force_id (lexer))
61         return CMD_CASCADING_FAILURE;
62       while (lex_token (lexer) == T_ID)
63         {
64           if (cp + 16 > endp)
65             {
66               char *old_vecnames = vecnames;
67               vecnames = xrealloc (vecnames, endp - vecnames + 256);
68               cp = (cp - old_vecnames) + vecnames;
69               endp = (endp - old_vecnames) + vecnames + 256;
70             }
71
72           for (cp2 = cp; cp2 < cp; cp2 += strlen (cp))
73             if (!strcasecmp (cp2, lex_tokid (lexer)))
74               {
75                 msg (SE, _("Vector name %s is given twice."), lex_tokid (lexer));
76                 goto fail;
77               }
78
79           if (dict_lookup_vector (dict, lex_tokid (lexer)))
80             {
81               msg (SE, _("There is already a vector with name %s."), lex_tokid (lexer));
82               goto fail;
83             }
84
85           cp = stpcpy (cp, lex_tokid (lexer)) + 1;
86           lex_get (lexer);
87           lex_match (lexer, ',');
88         }
89       *cp++ = 0;
90
91       /* Now that we have the names it's time to check for the short
92          or long forms. */
93       if (lex_match (lexer, '='))
94         {
95           /* Long form. */
96           struct variable **v;
97           size_t nv;
98
99           if (strchr (vecnames, '\0')[1])
100             {
101               /* There's more than one vector name. */
102               msg (SE, _("A slash must be used to separate each vector "
103                          "specification when using the long form.  Commands "
104                          "such as VECTOR A,B=Q1 TO Q20 are not supported."));
105               goto fail;
106             }
107
108           if (!parse_variables (lexer, dict, &v, &nv,
109                                 PV_SAME_WIDTH | PV_DUPLICATE))
110             goto fail;
111
112           dict_create_vector (dict, vecnames, v, nv);
113           free (v);
114         }
115       else if (lex_match (lexer, '('))
116         {
117           int i;
118
119           /* Maximum number of digits in a number to add to the base
120              vecname. */
121           int ndig;
122
123           /* Name of an individual variable to be created. */
124           char name[SHORT_NAME_LEN + 1];
125
126           /* Vector variables. */
127           struct variable **v;
128           int nv;
129
130           if (!lex_force_int (lexer))
131             return CMD_CASCADING_FAILURE;
132           nv = lex_integer (lexer);
133           lex_get (lexer);
134           if (nv <= 0)
135             {
136               msg (SE, _("Vectors must have at least one element."));
137               goto fail;
138             }
139           if (!lex_force_match (lexer, ')'))
140             goto fail;
141
142           /* First check that all the generated variable names
143              are LONG_NAME_LEN characters or shorter. */
144           ndig = intlog10 (nv);
145           for (cp = vecnames; *cp;)
146             {
147               int len = strlen (cp);
148               if (len + ndig > LONG_NAME_LEN)
149                 {
150                   msg (SE, _("%s%d is too long for a variable name."), cp, nv);
151                   goto fail;
152                 }
153               cp += len + 1;
154             }
155
156           /* Next check that none of the variables exist. */
157           for (cp = vecnames; *cp;)
158             {
159               for (i = 0; i < nv; i++)
160                 {
161                   sprintf (name, "%s%d", cp, i + 1);
162                   if (dict_lookup_var (dict, name))
163                     {
164                       msg (SE, _("There is already a variable named %s."),
165                            name);
166                       goto fail;
167                     }
168                 }
169               cp += strlen (cp) + 1;
170             }
171
172           /* Finally create the variables and vectors. */
173           v = xmalloc (nv * sizeof *v);
174           for (cp = vecnames; *cp;)
175             {
176               for (i = 0; i < nv; i++)
177                 {
178                   sprintf (name, "%s%d", cp, i + 1);
179                   v[i] = dict_create_var_assert (dict, name, 0);
180                 }
181               if (!dict_create_vector (dict, cp, v, nv))
182                 NOT_REACHED ();
183               cp += strlen (cp) + 1;
184             }
185           free (v);
186         }
187       else
188         {
189           msg (SE, _("The syntax for this command does not match "
190                "the expected syntax for either the long form "
191                "or the short form of VECTOR."));
192           goto fail;
193         }
194
195       free (vecnames);
196       vecnames = NULL;
197     }
198   while (lex_match (lexer, '/'));
199
200   if (lex_token (lexer) != '.')
201     {
202       lex_error (lexer, _("expecting end of command"));
203       goto fail;
204     }
205   return CMD_SUCCESS;
206
207 fail:
208   free (vecnames);
209   return CMD_FAILURE;
210 }