pintos-os.org Git - pspp/blob - tests/language/lexer/scan-test.c

   1 /* PSPP - a program for statistical analysis.
   2    Copyright (C) 2010, 2011 Free Software Foundation, Inc.
   3
   4    This program is free software: you can redistribute it and/or modify
   5    it under the terms of the GNU General Public License as published by
   6    the Free Software Foundation, either version 3 of the License, or
   7    (at your option) any later version.
   8
   9    This program is distributed in the hope that it will be useful,
  10    but WITHOUT ANY WARRANTY; without even the implied warranty of
  11    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  12    GNU 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, see <http://www.gnu.org/licenses/>. */
  16
  17 #include <config.h>
  18
  19 #include <ctype.h>
  20 #include <errno.h>
  21 #include <getopt.h>
  22 #include <limits.h>
  23 #include <stdio.h>
  24 #include <stdlib.h>
  25 #include <string.h>
  26
  27 #include "libpspp/assertion.h"
  28 #include "libpspp/compiler.h"
  29 #include "libpspp/misc.h"
  30 #include "language/lexer/scan.h"
  31 #include "language/lexer/token.h"
  32
  33 #include "gl/error.h"
  34 #include "gl/ftoastr.h"
  35 #include "gl/progname.h"
  36 #include "gl/read-file.h"
  37 #include "gl/xalloc.h"
  38
  39 /* -a/--auto, -b/--batch, -i/--interactive: syntax mode. */
  40 static enum segmenter_mode mode = SEG_MODE_AUTO;
  41
  42 static const char *parse_options (int argc, char **argv);
  43 static void usage (void) NO_RETURN;
  44
  45 int
  46 main (int argc, char *argv[])
  47 {
  48   const char *file_name;
  49   size_t length;
  50   char *input;
  51
  52   struct string_lexer slex;
  53   bool more;
  54
  55   set_program_name (argv[0]);
  56   file_name = parse_options (argc, argv);
  57
  58   /* Read from stdin into 'input'.  Ensure that 'input' ends in a new-line
  59      followed by a null byte. */
  60   input = (!strcmp (file_name, "-")
  61            ? fread_file (stdin, &length)
  62            : read_file (file_name, &length));
  63   if (input == NULL)
  64     error (EXIT_FAILURE, errno, "reading %s failed", file_name);
  65   input = xrealloc (input, length + 3);
  66   if (length == 0 || input[length - 1] != '\n')
  67     input[length++] = '\n';
  68   input[length++] = '\0';
  69
  70   string_lexer_init (&slex, input, mode);
  71   do
  72     {
  73       struct token token;
  74
  75       more = string_lexer_next (&slex, &token);
  76
  77       printf ("%s", scan_type_to_string (token.type));
  78       if (token.number != 0.0)
  79         {
  80           char s[DBL_BUFSIZE_BOUND];
  81
  82           dtoastr (s, sizeof s, 0, 0, token.number);
  83           printf (" %s", s);
  84         }
  85       if (token.string.string != NULL || token.string.length > 0)
  86         printf (" \"%.*s\"", (int) token.string.length, token.string.string);
  87       printf ("\n");
  88
  89       token_destroy (&token);
  90     }
  91   while (more);
  92
  93   free (input);
  94
  95   return 0;
  96 }
  97
  98 static const char *
  99 parse_options (int argc, char **argv)
 100 {
 101   for (;;)
 102     {
 103       static const struct option options[] =
 104         {
 105           {"auto", no_argument, NULL, 'a'},
 106           {"batch", no_argument, NULL, 'b'},
 107           {"interactive", no_argument, NULL, 'i'},
 108           {"help", no_argument, NULL, 'h'},
 109           {NULL, 0, NULL, 0},
 110         };
 111
 112       int c = getopt_long (argc, argv, "abih", options, NULL);
 113       if (c == -1)
 114         break;
 115
 116       switch (c)
 117         {
 118         case 'a':
 119           mode = SEG_MODE_AUTO;
 120           break;
 121
 122         case 'b':
 123           mode = SEG_MODE_BATCH;
 124           break;
 125
 126         case 'i':
 127           mode = SEG_MODE_INTERACTIVE;
 128           break;
 129
 130         case 'h':
 131           usage ();
 132
 133         case 0:
 134           break;
 135
 136         case '?':
 137           exit (EXIT_FAILURE);
 138           break;
 139
 140         default:
 141           NOT_REACHED ();
 142         }
 143
 144     }
 145
 146   if (optind + 1 != argc)
 147     error (1, 0, "exactly one non-option argument required; "
 148            "use --help for help");
 149   return argv[optind];
 150 }
 151
 152 static void
 153 usage (void)
 154 {
 155   printf ("\
 156 %s, to test breaking PSPP syntax into tokens\n\
 157 usage: %s [OPTIONS] INPUT\n\
 158 \n\
 159 Options:\n\
 160   -1, --one-segment   feed one segment at a time\n\
 161   -a, --auto          use \"auto\" syntax mode\n\
 162   -b, --batch         use \"batch\" syntax mode\n\
 163   -i, --interactive   use \"interactive\" syntax mode (default)\n\
 164   -v, --verbose       include rows and column numbers in output\n\
 165   -h, --help          print this help message\n",
 166           program_name, program_name);
 167   exit (EXIT_SUCCESS);
 168 }