Changed DFM from open-at-first-access to explicit-open. Before,
[pspp-builds.git] / src / vector.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 <assert.h>
22 #include <stdlib.h>
23 #include "alloc.h"
24 #include "command.h"
25 #include "error.h"
26 #include "lexer.h"
27 #include "misc.h"
28 #include "str.h"
29 #include "var.h"
30
31 int
32 cmd_vector (void)
33 {
34   /* Just to be different, points to a set of null terminated strings
35      containing the names of the vectors to be created.  The list
36      itself is terminated by a empty string.  So a list of three
37      elements, A B C, would look like this: "A\0B\0C\0\0". */
38   char *vecnames;
39
40   /* vecnames iterators. */
41   char *cp, *cp2;
42
43   /* Maximum allocated position for vecnames, plus one position. */
44   char *endp = NULL;
45
46   cp = vecnames = xmalloc (256);
47   endp = &vecnames[256];
48   do
49     {
50       /* Get the name(s) of the new vector(s). */
51       if (!lex_force_id ())
52         return CMD_FAILURE;
53       while (token == T_ID)
54         {
55           if (cp + 16 > endp)
56             {
57               char *old_vecnames = vecnames;
58               vecnames = xrealloc (vecnames, endp - vecnames + 256);
59               cp = (cp - old_vecnames) + vecnames;
60               endp = (endp - old_vecnames) + vecnames + 256;
61             }
62
63           for (cp2 = cp; cp2 < cp; cp2 += strlen (cp))
64             if (!strcmp (cp2, tokid))
65               {
66                 msg (SE, _("Vector name %s is given twice."), tokid);
67                 goto fail;
68               }
69
70           if (dict_lookup_vector (default_dict, tokid))
71             {
72               msg (SE, _("There is already a vector with name %s."), tokid);
73               goto fail;
74             }
75
76           cp = stpcpy (cp, tokid) + 1;
77           lex_get ();
78           lex_match (',');
79         }
80       *cp++ = 0;
81
82       /* Now that we have the names it's time to check for the short
83          or long forms. */
84       if (lex_match ('='))
85         {
86           /* Long form. */
87           struct variable **v;
88           int nv;
89
90           if (strchr (vecnames, '\0')[1])
91             {
92               /* There's more than one vector name. */
93               msg (SE, _("A slash must be used to separate each vector "
94                    "specification when using the long form.  Commands "
95                    "such as VECTOR A,B=Q1 TO Q20 are not supported."));
96               goto fail;
97             }
98
99           if (!parse_variables (default_dict, &v, &nv,
100                                 PV_SAME_TYPE | PV_DUPLICATE))
101             goto fail;
102
103           dict_create_vector (default_dict, vecnames, v, nv);
104           free (v);
105         }
106       else if (lex_match ('('))
107         {
108           int i;
109
110           /* Maximum number of digits in a number to add to the base
111              vecname. */
112           int ndig;
113
114           /* Name of an individual variable to be created. */
115           char name[9];
116
117           /* Vector variables. */
118           struct variable **v;
119           int nv;
120
121           if (!lex_force_int ())
122             return CMD_FAILURE;
123           nv = lex_integer ();
124           lex_get ();
125           if (nv <= 0)
126             {
127               msg (SE, _("Vectors must have at least one element."));
128               goto fail;
129             }
130           if (!lex_force_match (')'))
131             goto fail;
132
133           /* First check that all the generated variable names are 8
134              characters or shorter. */
135           ndig = intlog10 (nv);
136           for (cp = vecnames; *cp;)
137             {
138               int len = strlen (cp);
139               if (len + ndig > 8)
140                 {
141                   msg (SE, _("%s%d is too long for a variable name."), cp, nv);
142                   goto fail;
143                 }
144               cp += len + 1;
145             }
146
147           /* Next check that none of the variables exist. */
148           for (cp = vecnames; *cp;)
149             {
150               for (i = 0; i < nv; i++)
151                 {
152                   sprintf (name, "%s%d", cp, i + 1);
153                   if (dict_lookup_var (default_dict, name))
154                     {
155                       msg (SE, _("There is already a variable named %s."), name);
156                       goto fail;
157                     }
158                 }
159               cp += strlen (cp) + 1;
160             }
161
162           /* Finally create the variables and vectors. */
163           v = xmalloc (nv * sizeof *v);
164           for (cp = vecnames; *cp;)
165             {
166               for (i = 0; i < nv; i++)
167                 {
168                   sprintf (name, "%s%d", cp, i + 1);
169                   v[i] = dict_create_var_assert (default_dict, name, 0);
170                 }
171               if (!dict_create_vector (default_dict, cp, v, nv))
172                 assert (0);
173               cp += strlen (cp) + 1;
174             }
175           free (v);
176         }
177       else
178         {
179           msg (SE, _("The syntax for this command does not match "
180                "the expected syntax for either the long form "
181                "or the short form of VECTOR."));
182           goto fail;
183         }
184
185       free (vecnames);
186       vecnames = NULL;
187     }
188   while (lex_match ('/'));
189
190   if (token != '.')
191     {
192       lex_error (_("expecting end of command"));
193       goto fail;
194     }
195   return CMD_SUCCESS;
196
197 fail:
198   free (vecnames);
199   return CMD_PART_SUCCESS_MAYBE;
200 }