2 * Copyright 2008-2009 Katholieke Universiteit Leuven
4 * Use of this software is governed by the GNU LGPLv2.1 license
6 * Written by Sven Verdoolaege, K.U.Leuven, Departement
7 * Computerwetenschappen, Celestijnenlaan 200A, B-3001 Leuven, Belgium
14 #include "isl_stream.h"
16 static struct isl_token *isl_token_new(struct isl_ctx *ctx,
17 int line, int col, unsigned on_new_line)
19 struct isl_token *tok = isl_alloc_type(ctx, struct isl_token);
24 tok->on_new_line = on_new_line;
28 void isl_token_free(struct isl_token *tok)
32 if (tok->type == ISL_TOKEN_VALUE)
33 isl_int_clear(tok->u.v);
34 else if (tok->type == ISL_TOKEN_IDENT)
39 void isl_stream_error(struct isl_stream *s, struct isl_token *tok, char *msg)
41 int line = tok ? tok->line : s->line;
42 int col = tok ? tok->col : s->col;
43 fprintf(stderr, "syntax error (%d, %d): %s\n", line, col, msg);
46 fprintf(stderr, "got '%c'\n", tok->type);
48 fprintf(stderr, "got token type %d\n", tok->type);
52 static struct isl_stream* isl_stream_new(struct isl_ctx *ctx)
55 struct isl_stream *s = isl_alloc_type(ctx, struct isl_stream);
63 s->buffer = isl_alloc_array(ctx, char, s->size);
71 for (i = 0; i < 5; ++i)
80 struct isl_stream* isl_stream_new_file(struct isl_ctx *ctx, FILE *file)
82 struct isl_stream *s = isl_stream_new(ctx);
89 struct isl_stream* isl_stream_new_str(struct isl_ctx *ctx, const char *str)
91 struct isl_stream *s = isl_stream_new(ctx);
96 static int isl_stream_getc(struct isl_stream *s)
121 static void isl_stream_ungetc(struct isl_stream *s, int c)
130 static int isl_stream_push_char(struct isl_stream *s, int c)
132 if (s->len >= s->size) {
133 s->size = (3*s->size)/2;
134 s->buffer = isl_realloc_array(ctx, s->buffer, char, s->size);
138 s->buffer[s->len++] = c;
142 void isl_stream_push_token(struct isl_stream *s, struct isl_token *tok)
144 isl_assert(s->ctx, s->n_token < 5, return);
145 s->tokens[s->n_token++] = tok;
148 struct isl_token *isl_stream_next_token(struct isl_stream *s)
151 struct isl_token *tok = NULL;
153 int old_line = s->line;
156 return s->tokens[--s->n_token];
160 /* skip spaces and comment lines */
161 while ((c = isl_stream_getc(s)) != -1) {
163 while ((c = isl_stream_getc(s)) != -1 && c != '\n')
168 } else if (!isspace(c))
190 tok = isl_token_new(s->ctx, line, col, old_line != line);
193 tok->type = (enum isl_token_type)c;
198 if ((c = isl_stream_getc(s)) == '>') {
199 tok = isl_token_new(s->ctx, line, col, old_line != line);
202 tok->type = ISL_TOKEN_TO;
206 isl_stream_ungetc(s, c);
208 if (c == '-' || isdigit(c)) {
209 tok = isl_token_new(s->ctx, line, col, old_line != line);
212 tok->type = ISL_TOKEN_VALUE;
213 isl_int_init(tok->u.v);
214 if (isl_stream_push_char(s, c))
216 while ((c = isl_stream_getc(s)) != -1 && isdigit(c))
217 if (isl_stream_push_char(s, c))
220 isl_stream_ungetc(s, c);
221 if (s->len == 1 && s->buffer[0] == '-')
222 isl_int_set_si(tok->u.v, -1);
224 isl_stream_push_char(s, '\0');
225 isl_int_read(tok->u.v, s->buffer);
230 tok = isl_token_new(s->ctx, line, col, old_line != line);
233 isl_stream_push_char(s, c);
234 while ((c = isl_stream_getc(s)) != -1 && isalnum(c))
235 isl_stream_push_char(s, c);
237 isl_stream_ungetc(s, c);
238 isl_stream_push_char(s, '\0');
239 if (!strcasecmp(s->buffer, "exists"))
240 tok->type = ISL_TOKEN_EXISTS;
241 else if (!strcasecmp(s->buffer, "and"))
242 tok->type = ISL_TOKEN_AND;
243 else if (!strcasecmp(s->buffer, "or"))
244 tok->type = ISL_TOKEN_OR;
246 tok->type = ISL_TOKEN_IDENT;
247 tok->u.s = strdup(s->buffer);
253 tok = isl_token_new(s->ctx, line, col, old_line != line);
256 if ((c = isl_stream_getc(s)) == '=') {
257 tok->type = ISL_TOKEN_GE;
261 isl_stream_ungetc(s, c);
262 tok->type = ISL_TOKEN_GT;
267 tok = isl_token_new(s->ctx, line, col, old_line != line);
270 if ((c = isl_stream_getc(s)) == '=') {
271 tok->type = ISL_TOKEN_LE;
275 isl_stream_ungetc(s, c);
276 tok->type = ISL_TOKEN_LT;
280 tok = isl_token_new(s->ctx, line, col, old_line != line);
283 tok->type = ISL_TOKEN_AND;
284 if ((c = isl_stream_getc(s)) != '&' && c != -1)
285 isl_stream_ungetc(s, c);
289 tok = isl_token_new(s->ctx, line, col, old_line != line);
292 tok->type = ISL_TOKEN_OR;
293 if ((c = isl_stream_getc(s)) != '|' && c != -1)
294 isl_stream_ungetc(s, c);
298 tok = isl_token_new(s->ctx, line, col, old_line != line);
301 tok->type = ISL_TOKEN_UNKNOWN;
308 int isl_stream_eat(struct isl_stream *s, int type)
310 struct isl_token *tok;
312 tok = isl_stream_next_token(s);
315 if (tok->type == type) {
319 isl_stream_error(s, tok, "expecting other token");
320 isl_stream_push_token(s, tok);
324 void isl_stream_free(struct isl_stream *s)
329 if (s->n_token != 0) {
330 struct isl_token *tok = isl_stream_next_token(s);
331 isl_stream_error(s, tok, "unexpected token");
334 isl_ctx_deref(s->ctx);