src/language/dictionary/numeric.c

   1 /* PSPP - computes sample statistics.
   2    Copyright (C) 1997-9, 2000, 2006 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/dictionary.h>
  24 #include <data/procedure.h>
  25 #include <data/variable.h>
  26 #include <language/command.h>
  27 #include <language/lexer/format-parser.h>
  28 #include <language/lexer/lexer.h>
  29 #include <language/lexer/variable-parser.h>
  30 #include <libpspp/assertion.h>
  31 #include <libpspp/message.h>
  32 #include <libpspp/str.h>
  33
  34 #include "gettext.h"
  35 #define _(msgid) gettext (msgid)
  36
  37 /* Parses the NUMERIC command. */
  38 int
  39 cmd_numeric (struct lexer *lexer, struct dataset *ds)
  40 {
  41   size_t i;
  42
  43   /* Names of variables to create. */
  44   char **v;
  45   size_t nv;
  46
  47   /* Format spec for variables to create.  f.type==-1 if default is to
  48      be used. */
  49   struct fmt_spec f;
  50
  51   do
  52     {
  53       if (!parse_DATA_LIST_vars (lexer, &v, &nv, PV_NONE))
  54         return CMD_FAILURE;
  55
  56       /* Get the optional format specification. */
  57       if (lex_match (lexer, '('))
  58         {
  59           if (!parse_format_specifier (lexer, &f))
  60             goto fail;
  61           if (fmt_is_string (f.type))
  62             {
  63               char str[FMT_STRING_LEN_MAX + 1];
  64               msg (SE, _("Format type %s may not be used with a numeric "
  65                          "variable."), fmt_to_string (&f, str));
  66               goto fail;
  67             }
  68
  69           if (!lex_match (lexer, ')'))
  70             {
  71               msg (SE, _("`)' expected after output format."));
  72               goto fail;
  73             }
  74         }
  75       else
  76         f.type = -1;
  77
  78       /* Create each variable. */
  79       for (i = 0; i < nv; i++)
  80         {
  81           struct variable *new_var = dict_create_var (dataset_dict (ds), v[i], 0);
  82           if (!new_var)
  83             msg (SE, _("There is already a variable named %s."), v[i]);
  84           else
  85             {
  86               if (f.type != -1)
  87                 var_set_both_formats (new_var, &f);
  88             }
  89         }
  90
  91       /* Clean up. */
  92       for (i = 0; i < nv; i++)
  93         free (v[i]);
  94       free (v);
  95     }
  96   while (lex_match (lexer, '/'));
  97
  98   return lex_end_of_command (lexer);
  99
 100   /* If we have an error at a point where cleanup is required,
 101      flow-of-control comes here. */
 102 fail:
 103   for (i = 0; i < nv; i++)
 104     free (v[i]);
 105   free (v);
 106   return CMD_FAILURE;
 107 }
 108
 109 /* Parses the STRING command. */
 110 int
 111 cmd_string (struct lexer *lexer, struct dataset *ds)
 112 {
 113   size_t i;
 114
 115   /* Names of variables to create. */
 116   char **v;
 117   size_t nv;
 118
 119   /* Format spec for variables to create. */
 120   struct fmt_spec f;
 121
 122   /* Width of variables to create. */
 123   int width;
 124
 125   do
 126     {
 127       if (!parse_DATA_LIST_vars (lexer, &v, &nv, PV_NONE))
 128         return CMD_FAILURE;
 129
 130       if (!lex_force_match (lexer, '(')
 131           || !parse_format_specifier (lexer, &f)
 132           || !lex_force_match (lexer, ')'))
 133         goto fail;
 134       if (!fmt_is_string (f.type))
 135         {
 136           char str[FMT_STRING_LEN_MAX + 1];
 137           msg (SE, _("Format type %s may not be used with a string "
 138                      "variable."), fmt_to_string (&f, str));
 139           goto fail;
 140         }
 141       if (!fmt_check_output (&f))
 142         goto fail;
 143
 144       width = fmt_var_width (&f);
 145
 146       /* Create each variable. */
 147       for (i = 0; i < nv; i++)
 148         {
 149           struct variable *new_var = dict_create_var (dataset_dict (ds), v[i],
 150                                                       width);
 151           if (!new_var)
 152             msg (SE, _("There is already a variable named %s."), v[i]);
 153           else
 154             var_set_both_formats (new_var, &f);
 155         }
 156
 157       /* Clean up. */
 158       for (i = 0; i < nv; i++)
 159         free (v[i]);
 160       free (v);
 161     }
 162   while (lex_match (lexer, '/'));
 163
 164   return lex_end_of_command (lexer);
 165
 166   /* If we have an error at a point where cleanup is required,
 167      flow-of-control comes here. */
 168 fail:
 169   for (i = 0; i < nv; i++)
 170     free (v[i]);
 171   free (v);
 172   return CMD_FAILURE;
 173 }
 174
 175 /* Parses the LEAVE command. */
 176 int
 177 cmd_leave (struct lexer *lexer, struct dataset *ds)
 178 {
 179   struct variable **v;
 180   size_t nv;
 181
 182   size_t i;
 183
 184   if (!parse_variables (lexer, dataset_dict (ds), &v, &nv, PV_NONE))
 185     return CMD_CASCADING_FAILURE;
 186   for (i = 0; i < nv; i++)
 187     var_set_leave (v[i], true);
 188   free (v);
 189
 190   return lex_end_of_command (lexer);
 191 }