#pragma prototyped
#include <ast.h>
#include <ctype.h>
#define getchr(ex) (*(ex)->nextchr++)
#define peekchr(ex) (*(ex)->nextchr)
#define ungetchr(ex) ((ex)->nextchr--)
#define error(ex,msg) return(seterror(ex,msg))
typedef struct
{
char* nextchr;
char* errchr;
char* errmsg;
long (*convert)(const char*, char**, void*);
void* handle;
} Expr_t;
static long
seterror(register Expr_t* ex, char* msg)
{
if (!ex->errmsg) ex->errmsg = msg;
ex->errchr = ex->nextchr;
ex->nextchr = "";
return(0);
}
static long
expr(register Expr_t* ex, register int precedence)
{
register int c;
register long n;
register long x;
char* pos;
int operand = 1;
while (c = getchr(ex), isspace(c));
switch (c)
{
case 0:
ungetchr(ex);
if (!precedence) return(0);
error(ex, "more tokens expected");
case '-':
n = -expr(ex, 13);
break;
case '+':
n = expr(ex, 13);
break;
case '!':
n = !expr(ex, 13);
break;
case '~':
n = ~expr(ex, 13);
break;
default:
ungetchr(ex);
n = 0;
operand = 0;
break;
}
for (;;)
{
switch (c = getchr(ex))
{
case 0:
goto done;
case ')':
if (!precedence) error(ex, "too many )'s");
goto done;
case '(':
n = expr(ex, 1);
if (getchr(ex) != ')')
{
ungetchr(ex);
error(ex, "closing ) expected");
}
gotoperand:
if (operand) error(ex, "operator expected");
operand = 1;
continue;
case '?':
if (precedence > 1) goto done;
if (peekchr(ex) == ':')
{
getchr(ex);
x = expr(ex, 2);
if (!n) n = x;
}
else
{
x = expr(ex, 2);
if (getchr(ex) != ':')
{
ungetchr(ex);
error(ex, ": expected for ? operator");
}
if (n)
{
n = x;
expr(ex, 2);
}
else n = expr(ex, 2);
}
break;
case ':':
goto done;
case '|':
if (peekchr(ex) == '|')
{
if (precedence > 2) goto done;
getchr(ex);
x = expr(ex, 3);
n = n || x;
}
else
{
if (precedence > 4) goto done;
x = expr(ex, 5);
n |= x;
}
break;
case '^':
if (precedence > 5) goto done;
x = expr(ex, 6);
n ^= x;
break;
case '&':
if (peekchr(ex) == '&')
{
if (precedence > 3) goto done;
getchr(ex);
x = expr(ex, 4);
n = n && x;
}
else
{
if (precedence > 6) goto done;
x = expr(ex, 7);
n &= x;
}
break;
case '=':
case '!':
if (peekchr(ex) != '=') error(ex, "operator syntax error");
if (precedence > 7) goto done;
getchr(ex);
x = expr(ex, 8);
if (c == '=') n = n == x;
else n = n != x;
break;
case '<':
case '>':
if (peekchr(ex) == c)
{
if (precedence > 9) goto done;
getchr(ex);
x = expr(ex, 10);
if (c == '<') n <<= x;
else n >>= x;
}
else
{
if (precedence > 8) goto done;
if (peekchr(ex) == '=')
{
getchr(ex);
x = expr(ex, 9);
if (c == '<') n = n <= x;
else n = n >= x;
}
else
{
x = expr(ex, 9);
if (c == '<') n = n < x;
else n = n > x;
}
}
break;
case '+':
case '-':
if (precedence > 10) goto done;
x = expr(ex, 11);
if (c == '+') n += x;
else n -= x;
break;
case '*':
case '/':
case '%':
if (precedence > 11) goto done;
x = expr(ex, 12);
if (c == '*') n *= x;
else if (x == 0) error(ex, "divide by zero");
else if (c == '/') n /= x;
else n %= x;
break;
default:
if (isspace(c)) continue;
pos = --ex->nextchr;
if (isdigit(c)) n = strton(ex->nextchr, &ex->nextchr, NiL, 0);
else if (ex->convert) n = (*ex->convert)(ex->nextchr, &ex->nextchr, ex->handle);
if (ex->nextchr == pos) error(ex, "syntax error");
goto gotoperand;
}
if (ex->errmsg) return(0);
if (!operand) error(ex, "operand expected");
}
done:
ungetchr(ex);
if (!operand) error(ex, "operand expected");
return(n);
}
long
strexpr(const char* s, char** end, long(*convert)(const char*, char**, void*), void* handle)
{
long n;
Expr_t ex;
ex.nextchr = (char*)s;
ex.errmsg = 0;
ex.convert = convert;
ex.handle = handle;
n = expr(&ex, 0);
if (peekchr(&ex) == ':')
seterror(&ex, "invalid use of :");
if (ex.errmsg)
{
if (convert) (*convert)(NiL, &ex.errmsg, handle);
ex.nextchr = ex.errchr;
n = 0;
}
if (end) *end = ex.nextchr;
return(n);
}