new algorithm to parse expressions + distribution of code between lparser

and lcode.

lcode.c

@@ -1,16 +1,23 @@
 /*
-** $Id: $
+** $Id: lcode.c,v 1.1 2000/02/22 13:30:11 roberto Exp roberto $
 ** Code generator for Lua
 ** See Copyright Notice in lua.h
 */
 
 
 #include "lcode.h"
+#include "ldo.h"
 #include "llex.h"
 #include "lmem.h"
 #include "lobject.h"
 #include "lopcodes.h"
 #include "lparser.h"
+#include "lstring.h"
+
+
+void luaK_error (LexState *ls, const char *msg) {
+  luaX_error(ls, msg, ls->token);
+}
 
 
 static Instruction *last_i (FuncState *fs) {
@@ -94,3 +101,185 @@ void luaK_fixjump (LexState *ls, int pc, int dest) {
 }
 
 
+void luaK_deltastack (LexState *ls, int delta) {
+  FuncState *fs = ls->fs;
+  fs->stacksize += delta;
+  if (delta > 0 && fs->stacksize > fs->f->maxstacksize) {
+    if (fs->stacksize > MAXSTACK)
+      luaK_error(ls, "function or expression too complex");
+    fs->f->maxstacksize = fs->stacksize;
+  }
+}
+
+
+static int aux_code (LexState *ls, OpCode op, Instruction i, int delta) {
+  luaK_deltastack(ls, delta);
+  return luaK_code(ls, SET_OPCODE(i, op));
+}
+
+
+int luaK_0 (LexState *ls, OpCode op, int delta) {
+  return aux_code(ls, op, 0, delta);
+}
+
+
+int luaK_U (LexState *ls, OpCode op, int u, int delta) {
+  Instruction i = SETARG_U(0, u);
+  return aux_code(ls, op, i, delta);
+}
+
+
+int luaK_S (LexState *ls, OpCode op, int s, int delta) {
+  Instruction i = SETARG_S(0, s);
+  return aux_code(ls, op, i, delta);
+}
+
+
+int luaK_AB (LexState *ls, OpCode op, int a, int b, int delta) {
+  Instruction i = SETARG_A(0, a);
+  i = SETARG_B(i, b);
+  return aux_code(ls, op, i, delta);
+}
+
+
+int luaK_kstr (LexState *ls, int c) {
+  return luaK_U(ls, PUSHSTRING, c, 1);
+}
+
+
+#ifndef LOOKBACKNUMS
+#define LOOKBACKNUMS	20	/* arbitrary limit */
+#endif
+
+static int real_constant (LexState *ls, real r) {
+  /* check whether `r' has appeared within the last LIM entries */
+  TProtoFunc *f = ls->fs->f;
+  int c = f->nknum;
+  int lim = c < LOOKBACKNUMS ? 0 : c-LOOKBACKNUMS;
+  while (--c >= lim)
+    if (f->knum[c] == r) return c;
+  /* not found; create a new entry */
+  luaM_growvector(ls->L, f->knum, f->nknum, 1, real, constantEM, MAXARG_U);
+  c = f->nknum++;
+  f->knum[c] = r;
+  return c;
+}
+
+
+int luaK_number (LexState *ls, real f) {
+  if (f <= (real)MAXARG_S && (int)f == f)
+    return luaK_S(ls, PUSHINT, (int)f, 1);  /* f has a short integer value */
+  else
+    return luaK_U(ls, PUSHNUM, real_constant(ls, f), 1);
+}
+
+
+int luaK_adjuststack (LexState *ls, int n) {
+  if (n > 0)
+    return luaK_U(ls, POP, n, -n);
+  else if (n < 0)
+    return luaK_U(ls, PUSHNIL, (-n)-1, -n);
+  else return 0;
+}
+
+
+int luaK_iscall (LexState *ls, int pc) {
+  return (GET_OPCODE(ls->fs->f->code[pc]) == CALL);
+}
+
+
+void luaK_setcallreturns (LexState *ls, int pc, int nresults) {
+  if (luaK_iscall(ls, pc)) {  /* expression is a function call? */
+    Instruction *i = &ls->fs->f->code[pc];
+    int old_nresults = GETARG_B(*i);
+    if (old_nresults != MULT_RET)
+      luaK_deltastack(ls, -old_nresults);  /* pop old nresults */
+    *i = SETARG_B(*i, nresults);  /* set nresults */
+    if (nresults != MULT_RET)
+      luaK_deltastack(ls, nresults);  /* push results */
+  }
+}
+
+
+static void assertglobal (LexState *ls, int index) {
+  luaS_assertglobal(ls->L, ls->fs->f->kstr[index]);
+}
+
+
+void luaK_2stack (LexState *ls, expdesc *var) {
+  switch (var->k) {
+    case VLOCAL:
+      var->info = luaK_U(ls, PUSHLOCAL, var->info, 1);
+      break;
+    case VGLOBAL:
+      assertglobal(ls, var->info);  /* make sure that there is a global */
+      var->info = luaK_U(ls, GETGLOBAL, var->info, 1);
+      break;
+    case VINDEXED:
+      var->info = luaK_0(ls, GETTABLE, -1);
+      break;
+    case VEXP:
+      luaK_setcallreturns(ls, var->info, 1);  /* call must return 1 value */
+      break;
+  }
+  var->k = VEXP;
+}
+
+
+void luaK_storevar (LexState *ls, const expdesc *var) {
+  switch (var->k) {
+    case VLOCAL:
+      luaK_U(ls, SETLOCAL, var->info, -1);
+      break;
+    case VGLOBAL:
+      luaK_U(ls, SETGLOBAL, var->info, -1);
+      assertglobal(ls, var->info);  /* make sure that there is a global */
+      break;
+    case VINDEXED:
+      luaK_0(ls, SETTABLEPOP, -3);
+      break;
+    default:
+      LUA_INTERNALERROR(ls->L, "invalid var kind to store");
+  }
+}
+
+
+void luaK_prefix (LexState *ls, int op, expdesc *v) {
+  luaK_2stack(ls, v);
+  if (op == '-')
+    v->info = luaK_0(ls, MINUSOP, 0);
+  else
+    v->info = luaK_0(ls, NOTOP, 0);
+}
+
+
+void luaK_infix (LexState *ls, expdesc *v) {
+  luaK_2stack(ls, v);
+  if (ls->token == AND)
+    v->info = luaK_0(ls, ONFJMP, -1);
+  else if (ls->token == OR)
+    v->info = luaK_0(ls, ONTJMP, -1);
+} 
+
+
+void luaK_posfix (LexState *ls, int op, expdesc *v1, expdesc *v2) {
+  luaK_2stack(ls, v2);
+  switch (op) {
+    case AND:  case OR:
+      luaK_fixjump(ls, v1->info, ls->fs->pc);
+      break;
+    case '+': v1->info = luaK_0(ls, ADDOP, -1); break;
+    case '-': v1->info = luaK_0(ls, SUBOP, -1); break;
+    case '*': v1->info = luaK_0(ls, MULTOP, -1); break;
+    case '/': v1->info = luaK_0(ls, DIVOP, -1); break;
+    case '^': v1->info = luaK_0(ls, POWOP, -1); break;
+    case CONC: v1->info = luaK_0(ls, CONCOP, -1); break;
+    case EQ: v1->info = luaK_0(ls, EQOP, -1); break;
+    case NE: v1->info = luaK_0(ls, NEQOP, -1); break;
+    case '>': v1->info = luaK_0(ls, GTOP, -1); break;
+    case '<': v1->info = luaK_0(ls, LTOP, -1); break;
+    case GE: v1->info = luaK_0(ls, GEOP, -1); break;
+    case LE: v1->info = luaK_0(ls, LEOP, -1); break;
+  }
+}
+

lcode.h

@@ -1,5 +1,5 @@
 /*
-** $Id: $
+** $Id: lcode.h,v 1.1 2000/02/22 13:31:19 roberto Exp roberto $
 ** Code generator for Lua
 ** See Copyright Notice in lua.h
 */
@@ -9,11 +9,29 @@
 
 #include "llex.h"
 #include "lobject.h"
+#include "lopcodes.h"
+#include "lparser.h"
 
 
+void luaK_error (LexState *ls, const char *msg);
 int luaK_primitivecode (LexState *ls, Instruction i);
 int luaK_code (LexState *ls, Instruction i);
 void luaK_fixjump (LexState *ls, int pc, int dest);
+void luaK_deltastack (LexState *ls, int delta);
+int luaK_0 (LexState *ls, OpCode op, int delta);
+int luaK_U (LexState *ls, OpCode op, int u, int delta);
+int luaK_S (LexState *ls, OpCode op, int s, int delta);
+int luaK_AB (LexState *ls, OpCode op, int a, int b, int delta);
+int luaK_kstr (LexState *ls, int c);
+int luaK_number (LexState *ls, real f);
+int luaK_adjuststack (LexState *ls, int n);
+int luaK_iscall (LexState *ls, int pc);
+void luaK_setcallreturns (LexState *ls, int pc, int nresults);
+void luaK_2stack (LexState *ls, expdesc *var);
+void luaK_storevar (LexState *ls, const expdesc *var);
+void luaK_prefix (LexState *ls, int op, expdesc *v);
+void luaK_infix (LexState *ls, expdesc *v);
+void luaK_posfix (LexState *ls, int op, expdesc *v1, expdesc *v2);
 
 
 #endif

lparser.c

@@ -1,5 +1,5 @@
 /*
-** $Id: lparser.c,v 1.59 2000/02/14 16:51:08 roberto Exp roberto $
+** $Id: lparser.c,v 1.60 2000/02/22 13:30:11 roberto Exp roberto $
 ** LL(1) Parser and code generator for Lua
 ** See Copyright Notice in lua.h
 */
@@ -51,10 +51,10 @@ typedef struct constdesc {
 /*
 ** prototypes for recursive non-terminal functions
 */
-static void body (LexState *ls, int needself, int line);
+static int body (LexState *ls, int needself, int line);
 static void chunk (LexState *ls);
-static void constructor (LexState *ls);
-static void expr (LexState *ls, vardesc *v);
+static int constructor (LexState *ls);
+static void expr (LexState *ls, expdesc *v);
 static void exp1 (LexState *ls);
 
 
@@ -63,21 +63,16 @@ static void next (LexState *ls) {
 }
 
 
-static void luaY_error (LexState *ls, const char *msg) {
-  luaX_error(ls, msg, ls->token);
-}
-
-
 static void error_expected (LexState *ls, int token) {
   char buff[100], t[TOKEN_LEN];
   luaX_token2str(token, t);
   sprintf(buff, "`%.20s' expected", t);
-  luaY_error(ls, buff);
+  luaK_error(ls, buff);
 }
 
 
 static void error_unexpected (LexState *ls) {
-  luaY_error(ls, "unexpected token");
+  luaK_error(ls, "unexpected token");
 }
 
 
@@ -91,7 +86,7 @@ static void error_unmatched (LexState *ls, int what, int who, int where) {
     luaX_token2str(who, t_who);
     sprintf(buff, "`%.20s' expected (to close `%.20s' at line %d)",
             t_what, t_who, where);
-    luaY_error(ls, buff);
+    luaK_error(ls, buff);
   }
 }
 
@@ -115,56 +110,14 @@ static void checklimit (LexState *ls, int val, int limit, const char *msg) {
   if (val > limit) {
     char buff[100];
     sprintf(buff, "too many %.50s (limit=%d)", msg, limit);
-    luaY_error(ls, buff);
+    luaK_error(ls, buff);
   }
 }
 
 
-
-static void deltastack (LexState *ls, int delta) {
-  FuncState *fs = ls->fs;
-  fs->stacksize += delta;
-  if (delta > 0 && fs->stacksize > fs->f->maxstacksize) {
-    checklimit(ls, fs->stacksize, MAXSTACK, "temporaries or local variables");
-    fs->f->maxstacksize = fs->stacksize;
-  }
-}
-
-
-static int aux_code (LexState *ls, OpCode op, Instruction i, int delta) {
-  deltastack(ls, delta);
-  return luaK_code(ls, SET_OPCODE(i, op));
-}
-
-
-static int code_0 (LexState *ls, OpCode op, int delta) {
-  return aux_code(ls, op, 0, delta);
-}
-
-
-
-static int code_U (LexState *ls, OpCode op, int u, int delta) {
-  Instruction i = SETARG_U(0, u);
-  return aux_code(ls, op, i, delta);
-}
-
-
-static int code_S (LexState *ls, OpCode op, int s, int delta) {
-  Instruction i = SETARG_S(0, s);
-  return aux_code(ls, op, i, delta);
-}
-
-
-static int code_AB (LexState *ls, OpCode op, int a, int b, int delta) {
-  Instruction i = SETARG_A(0, a);
-  i = SETARG_B(i, b);
-  return aux_code(ls, op, i, delta);
-}
-
-
 static void check_debugline (LexState *ls) {
   if (ls->L->debug && ls->linenumber != ls->fs->lastsetline) {
-    code_U(ls, SETLINE, ls->linenumber, 0);
+    luaK_U(ls, SETLINE, ls->linenumber, 0);
     ls->fs->lastsetline = ls->linenumber;
   }
 }
@@ -178,16 +131,6 @@ static void check_match (LexState *ls, int what, int who, int where) {
 }
 
 
-static void code_kstr (LexState *ls, int c) {
-  code_U(ls, PUSHSTRING, c, 1);
-}
-
-
-static void assertglobal (LexState *ls, int index) {
-  luaS_assertglobal(ls->L, ls->fs->f->kstr[index]);
-}
-
-
 static int string_constant (LexState *ls, FuncState *fs, TaggedString *s) {
   TProtoFunc *f = fs->f;
   int c = s->constindex;
@@ -202,39 +145,15 @@ static int string_constant (LexState *ls, FuncState *fs, TaggedString *s) {
 }
 
 
-static void code_string (LexState *ls, TaggedString *s) {
-  code_kstr(ls, string_constant(ls, ls->fs, s));
-}
-
-
-#define LIM 20
-static int real_constant (LexState *ls, real r) {
-  /* check whether `r' has appeared within the last LIM entries */
-  TProtoFunc *f = ls->fs->f;
-  int c = f->nknum;
-  int lim = c < LIM ? 0 : c-LIM;
-  while (--c >= lim)
-    if (f->knum[c] == r) return c;
-  /* not found; create a new entry */
-  luaM_growvector(ls->L, f->knum, f->nknum, 1, real, constantEM, MAXARG_U);
-  c = f->nknum++;
-  f->knum[c] = r;
-  return c;
-}
-
-
-static void code_number (LexState *ls, real f) {
-  if (f <= (real)MAXARG_S && (int)f == f)
-    code_S(ls, PUSHINT, (int)f, 1);  /* f has a short integer value */
-  else
-    code_U(ls, PUSHNUM, real_constant(ls, f), 1);
+static int code_string (LexState *ls, TaggedString *s) {
+  return luaK_kstr(ls, string_constant(ls, ls->fs, s));
 }
 
 
 static int checkname (LexState *ls) {
   int sc;
   if (ls->token != NAME)
-    luaY_error(ls, "<name> expected");
+    luaK_error(ls, "<name> expected");
   sc = string_constant(ls, ls->fs, ls->seminfo.ts);
   next(ls);
   return sc;
@@ -295,7 +214,7 @@ static int aux_localname (FuncState *fs, TaggedString *n) {
 }
 
 
-static void singlevar (LexState *ls, TaggedString *n, vardesc *var, int prev) {
+static void singlevar (LexState *ls, TaggedString *n, expdesc *var, int prev) {
   FuncState *fs = prev ? ls->fs->prev : ls->fs;
   int i = aux_localname(fs, n);
   if (i >= 0) {  /* local value? */
@@ -315,7 +234,7 @@ static void singlevar (LexState *ls, TaggedString *n, vardesc *var, int prev) {
 
 static int indexupvalue (LexState *ls, TaggedString *n) {
   FuncState *fs = ls->fs;
-  vardesc v;
+  expdesc v;
   int i;
   singlevar(ls, n, &v, 1);
   for (i=0; i<fs->nupvalues; i++) {
@@ -330,48 +249,30 @@ static int indexupvalue (LexState *ls, TaggedString *n) {
 }
 
 
-static void pushupvalue (LexState *ls, TaggedString *n) {
+static int pushupvalue (LexState *ls, TaggedString *n) {
   if (ls->fs->prev == NULL)
     luaX_syntaxerror(ls, "cannot access upvalue in main", n->str);
   if (aux_localname(ls->fs, n) >= 0)
     luaX_syntaxerror(ls, "cannot access an upvalue in current scope", n->str);
-  code_U(ls, PUSHUPVALUE, indexupvalue(ls, n), 1);
-}
-
-
-static void adjuststack (LexState *ls, int n) {
-  if (n > 0)
-    code_U(ls, POP, n, -n);
-  else if (n < 0)
-    code_U(ls, PUSHNIL, (-n)-1, -n);
-}
-
-
-static void close_call (LexState *ls, int pc, int nresults) {
-  if (pc > 0) {  /* expression is an open function call? */
-    Instruction *i = &ls->fs->f->code[pc];
-    *i = SETARG_B(*i, nresults);  /* set nresults */
-    if (nresults != MULT_RET)
-      deltastack(ls, nresults);  /* push results */
-  }
+  return luaK_U(ls, PUSHUPVALUE, indexupvalue(ls, n), 1);
 }
 
 
 static void adjust_mult_assign (LexState *ls, int nvars, listdesc *d) {
   int diff = d->n - nvars;
-  if (d->pc == 0) {  /* list is closed */
+  if (d->n == 0 || !luaK_iscall(ls, d->pc)) {  /* list is empty or closed */
     /* push or pop eventual difference between list lengths */
-    adjuststack(ls, diff);
+    luaK_adjuststack(ls, diff);
   }
-  else {  /* must correct function call */
+  else {  /* list ends in a function call; must correct it */
     diff--;  /* do not count function call itself */
     if (diff <= 0) {  /* more variables than values? */
       /* function call must provide extra values */
-      close_call(ls, d->pc, -diff);
+      luaK_setcallreturns(ls, d->pc, -diff);
     }
     else {  /* more values than variables */
-      close_call(ls, d->pc, 0);  /* call should provide no value */
-      adjuststack(ls, diff);  /* pop eventual extra values */
+      luaK_setcallreturns(ls, d->pc, 0);  /* call should provide no value */
+      luaK_adjuststack(ls, diff);  /* pop eventual extra values */
     }
   }
 }
@@ -385,15 +286,15 @@ static void code_args (LexState *ls, int nparams, int dots) {
   fs->f->numparams = nparams;
   fs->f->is_vararg = dots;
   if (!dots)
-    deltastack(ls, nparams);
+    luaK_deltastack(ls, nparams);
   else {
-    deltastack(ls, nparams+1);
+    luaK_deltastack(ls, nparams+1);
     add_localvar(ls, luaS_newfixed(ls->L, "arg"));
   }
 }
 
 
-static int getvarname (LexState *ls, vardesc *var) {
+static int getvarname (LexState *ls, expdesc *var) {
   switch (var->k) {
     case VGLOBAL:
       return var->info;
@@ -407,55 +308,16 @@ static int getvarname (LexState *ls, vardesc *var) {
 }
 
 
-static void close_exp (LexState *ls, vardesc *var) {
-  switch (var->k) {
-    case VLOCAL:
-      code_U(ls, PUSHLOCAL, var->info, 1);
-      break;
-    case VGLOBAL:
-      code_U(ls, GETGLOBAL, var->info, 1);
-      assertglobal(ls, var->info);  /* make sure that there is a global */
-      break;
-    case VINDEXED:
-      code_0(ls, GETTABLE, -1);
-      break;
-    case VEXP:
-      close_call(ls, var->info, 1);  /* call must return 1 value */
-      break;
-  }
-  var->k = VEXP;
-  var->info = 0;  /* now this is a closed expression */
-}
-
-
-static void storevar (LexState *ls, const vardesc *var) {
-  switch (var->k) {
-    case VLOCAL:
-      code_U(ls, SETLOCAL, var->info, -1);
-      break;
-    case VGLOBAL:
-      code_U(ls, SETGLOBAL, var->info, -1);
-      assertglobal(ls, var->info);  /* make sure that there is a global */
-      break;
-    case VINDEXED:
-      code_0(ls, SETTABLEPOP, -3);
-      break;
-    default:
-      LUA_INTERNALERROR(ls->L, "invalid var kind to store");
-  }
-}
-
-
-static void func_onstack (LexState *ls, FuncState *func) {
+static int func_onstack (LexState *ls, FuncState *func) {
   TProtoFunc *f = ls->fs->f;
   int i;
   for (i=0; i<func->nupvalues; i++)
-    close_exp(ls, &func->upvalues[i]);
+    luaK_2stack(ls, &func->upvalues[i]);
   luaM_growvector(ls->L, f->kproto, f->nkproto, 1, TProtoFunc *,
                   constantEM, MAXARG_A);
   f->kproto[f->nkproto++] = func->f;
-  deltastack(ls, 1);  /* CLOSURE puts one extra element (before popping) */
-  code_AB(ls, CLOSURE, f->nkproto-1, func->nupvalues, -func->nupvalues);
+  luaK_deltastack(ls, 1);  /* CLOSURE puts one extra element before popping */
+  return luaK_AB(ls, CLOSURE, f->nkproto-1, func->nupvalues, -func->nupvalues);
 }
 
 
@@ -487,7 +349,7 @@ static void init_state (LexState *ls, FuncState *fs, TaggedString *source) {
 static void close_func (LexState *ls) {
   FuncState *fs = ls->fs;
   TProtoFunc *f = fs->f;
-  code_0(ls, ENDCODE, 0);
+  luaK_0(ls, ENDCODE, 0);
   luaM_reallocvector(ls->L, f->code, fs->pc, Instruction);
   luaM_reallocvector(ls->L, f->kstr, f->nkstr, TaggedString *);
   luaM_reallocvector(ls->L, f->knum, f->nknum, real);
@@ -509,7 +371,7 @@ TProtoFunc *luaY_parser (lua_State *L, ZIO *z) {
   next(&lexstate);  /* read first token */
   chunk(&lexstate);
   if (lexstate.token != EOS)
-    luaY_error(&lexstate, "<eof> expected");
+    luaK_error(&lexstate, "<eof> expected");
   close_func(&lexstate);
   return funcstate.f;
 }
@@ -522,21 +384,18 @@ TProtoFunc *luaY_parser (lua_State *L, ZIO *z) {
 
 
 static void explist1 (LexState *ls, listdesc *d) {
-  vardesc v;
+  expdesc v;
   expr(ls, &v);
   d->n = 1;
   while (ls->token == ',') {
     d->n++;
-    close_exp(ls, &v);
+    luaK_2stack(ls, &v);
     next(ls);
     expr(ls, &v);
   }
-  if (v.k == VEXP)
-    d->pc = v.info;
-  else {
-    close_exp(ls, &v);
-    d->pc = 0;
-  }
+  luaK_2stack(ls, &v);
+  luaK_setcallreturns(ls, v.info, MULT_RET);  /* default for explists */
+  d->pc = v.info;
 }
 
 
@@ -544,7 +403,6 @@ static void explist (LexState *ls, listdesc *d) {
   switch (ls->token) {
     case ELSE: case ELSEIF: case END: case UNTIL:
     case EOS: case ';': case ')':
-      d->pc = 0;
       d->n = 0;
       break;
 
@@ -564,7 +422,10 @@ static int funcparams (LexState *ls, int slf) {
       next(ls);
       explist(ls, &e);
       check_match(ls, ')', '(', line);
-      close_call(ls, e.pc, MULT_RET);  /* close 1 for old semantics */
+#ifdef LUA_COMPAT_ARGRET
+      if (e.n > 0)  /* arg list is not empty? */
+        luaK_setcallreturns(ls, e.pc, 1);  /* last call returns only 1 value */
+#endif
       break;
     }
 
@@ -578,27 +439,27 @@ static int funcparams (LexState *ls, int slf) {
       break;
 
     default:
-      luaY_error(ls, "function arguments expected");
+      luaK_error(ls, "function arguments expected");
       break;
   }
   fs->stacksize = slevel;  /* call will remove func and params */
-  return code_AB(ls, CALL, slevel, 0, 0);
+  return luaK_AB(ls, CALL, slevel, 0, 0);
 }
 
 
-static void var_or_func_tail (LexState *ls, vardesc *v) {
+static void var_or_func_tail (LexState *ls, expdesc *v) {
   for (;;) {
     switch (ls->token) {
       case '.':  /* var_or_func_tail -> '.' NAME */
         next(ls);
-        close_exp(ls, v);  /* `v' must be on stack */
-        code_kstr(ls, checkname(ls));
+        luaK_2stack(ls, v);  /* `v' must be on stack */
+        luaK_kstr(ls, checkname(ls));
         v->k = VINDEXED;
         break;
 
       case '[':  /* var_or_func_tail -> '[' exp1 ']' */
         next(ls);
-        close_exp(ls, v);  /* `v' must be on stack */
+        luaK_2stack(ls, v);  /* `v' must be on stack */
         exp1(ls);
         check(ls, ']');
         v->k = VINDEXED;
@@ -608,15 +469,15 @@ static void var_or_func_tail (LexState *ls, vardesc *v) {
         int name;
         next(ls);
         name = checkname(ls);
-        close_exp(ls, v);  /* `v' must be on stack */
-        code_U(ls, PUSHSELF, name, 1);
+        luaK_2stack(ls, v);  /* `v' must be on stack */
+        luaK_U(ls, PUSHSELF, name, 1);
         v->k = VEXP;
         v->info = funcparams(ls, 1);
         break;
       }
 
       case '(': case STRING: case '{':  /* var_or_func_tail -> funcparams */
-        close_exp(ls, v);  /* `v' must be on stack */
+        luaK_2stack(ls, v);  /* `v' must be on stack */
         v->k = VEXP;
         v->info = funcparams(ls, 0);
         break;
@@ -627,12 +488,11 @@ static void var_or_func_tail (LexState *ls, vardesc *v) {
 }
 
 
-static void var_or_func (LexState *ls, vardesc *v) {
+static void var_or_func (LexState *ls, expdesc *v) {
   /* var_or_func -> ['%'] NAME var_or_func_tail */
   if (optional(ls, '%')) {  /* upvalue? */
-    pushupvalue(ls, str_checkname(ls));
     v->k = VEXP;
-    v->info = 0;  /* closed expression */
+    v->info = pushupvalue(ls, str_checkname(ls));
   }
   else  /* variable name */
     singlevar(ls, str_checkname(ls), v, 0);
@@ -652,7 +512,7 @@ static void recfield (LexState *ls) {
   /* recfield -> (NAME | '['exp1']') = exp1 */
   switch (ls->token) {
     case NAME:
-      code_kstr(ls, checkname(ls));
+      luaK_kstr(ls, checkname(ls));
       break;
 
     case '[':
@@ -661,7 +521,7 @@ static void recfield (LexState *ls) {
       check(ls, ']');
       break;
 
-    default: luaY_error(ls, "<name> or `[' expected");
+    default: luaK_error(ls, "<name> or `[' expected");
   }
   check(ls, '=');
   exp1(ls);
@@ -679,12 +539,12 @@ static int recfields (LexState *ls) {
     recfield(ls);
     n++;
     if (++mod_n == RFIELDS_PER_FLUSH) {
-      code_U(ls, SETMAP, RFIELDS_PER_FLUSH-1, -2*RFIELDS_PER_FLUSH);
+      luaK_U(ls, SETMAP, RFIELDS_PER_FLUSH-1, -2*RFIELDS_PER_FLUSH);
       mod_n = 0;
     }
   }
   if (mod_n)
-    code_U(ls, SETMAP, mod_n-1, -2*mod_n);
+    luaK_U(ls, SETMAP, mod_n-1, -2*mod_n);
   return n;
 }
 
@@ -702,13 +562,13 @@ static int listfields (LexState *ls) {
     checklimit(ls, n, MAXARG_A*LFIELDS_PER_FLUSH,
                "items in a list initializer");
     if (++mod_n == LFIELDS_PER_FLUSH) {
-      code_AB(ls, SETLIST, n/LFIELDS_PER_FLUSH - 1, LFIELDS_PER_FLUSH-1,
+      luaK_AB(ls, SETLIST, n/LFIELDS_PER_FLUSH - 1, LFIELDS_PER_FLUSH-1,
               -LFIELDS_PER_FLUSH);
       mod_n = 0;
     }
   }
   if (mod_n > 0)
-    code_AB(ls, SETLIST, n/LFIELDS_PER_FLUSH, mod_n-1, -mod_n);
+    luaK_AB(ls, SETLIST, n/LFIELDS_PER_FLUSH, mod_n-1, -mod_n);
   return n;
 }
 
@@ -722,17 +582,17 @@ static void constructor_part (LexState *ls, constdesc *cd) {
       return;
 
     case NAME: {
-      vardesc v;
+      expdesc v;
       expr(ls, &v);
       if (ls->token == '=') {
-      code_kstr(ls, getvarname(ls, &v));
+      luaK_kstr(ls, getvarname(ls, &v));
       next(ls);  /* skip '=' */
       exp1(ls);
         cd->n = recfields(ls);
         cd->k = 1;  /* record */
       }
       else {
-        close_exp(ls, &v);
+        luaK_2stack(ls, &v);
         cd->n = listfields(ls);
         cd->k = 0;  /* list */
       }
@@ -754,10 +614,10 @@ static void constructor_part (LexState *ls, constdesc *cd) {
 }
 
 
-static void constructor (LexState *ls) {
+static int constructor (LexState *ls) {
   /* constructor -> '{' constructor_part [';' constructor_part] '}' */
   int line = ls->linenumber;
-  int pc = code_U(ls, CREATETABLE, 0, 1);
+  int pc = luaK_U(ls, CREATETABLE, 0, 1);
   int nelems;
   constdesc cd;
   check(ls, '{');
@@ -768,12 +628,13 @@ static void constructor (LexState *ls) {
     next(ls);
     constructor_part(ls, &other_cd);
     if (cd.k == other_cd.k)  /* repeated parts? */
-      luaY_error(ls, "invalid constructor syntax");
+      luaK_error(ls, "invalid constructor syntax");
     nelems += other_cd.n;
   }
   check_match(ls, '}', '{', line);
   /* set initial table size */
   ls->fs->f->code[pc] = SETARG_U(ls->fs->f->code[pc], nelems);
+  return pc;
 }
 
 /* }====================================================================== */
@@ -783,101 +644,39 @@ static void constructor (LexState *ls) {
 
 /*
 ** {======================================================================
-** For parsing expressions, we use a classic stack with priorities.
-** Each binary operator is represented by an index: EQ=2, NE=3, ... '^'=13.
-** The unary NOT is 0 and UNMINUS is 1.
+** Expression parsing
 ** =======================================================================
 */
 
-#define INDNOT		0
-#define INDMINUS	1
-
-/* code of first binary operator */
-#define FIRSTBIN	2
-
-/* code for power operator (last operator)
-** '^' needs special treatment because it is right associative
-*/
-#define POW	13
-
-
-static const int priority [POW+1] =  {5, 5, 1, 1, 1, 1, 1, 1, 2, 3, 3, 4, 4, 6};
-
-static const OpCode opcodes [POW+1] = {NOTOP, MINUSOP, EQOP, NEQOP, GTOP,
-  LTOP, LEOP, GEOP, CONCOP, ADDOP, SUBOP, MULTOP, DIVOP, POWOP};
-
-#define MAXOPS	20  /* op's stack size (arbitrary limit) */
-
-typedef struct stack_op {
-  int ops[MAXOPS];
-  int top;
-} stack_op;
-
-
-/*
-** returns the index of a binary operator
-*/
-static int binop (int op) {
-  switch (op) {
-    case EQ: return FIRSTBIN;
-    case  NE: return FIRSTBIN+1;
-    case  '>': return FIRSTBIN+2;
-    case  '<': return FIRSTBIN+3;
-    case  LE: return FIRSTBIN+4;
-    case  GE: return FIRSTBIN+5;
-    case  CONC: return FIRSTBIN+6;
-    case  '+': return FIRSTBIN+7;
-    case  '-': return FIRSTBIN+8;
-    case  '*': return FIRSTBIN+9;
-    case  '/': return FIRSTBIN+10;
-    case  '^': return FIRSTBIN+11;
-    default: return -1;
-  }
-}
-
-
-static void push (LexState *ls, stack_op *s, int op) {
-  if (s->top >= MAXOPS)
-    luaY_error(ls, "expression too complex");
-  s->ops[s->top++] = op;
-}
-
-
-static void pop_to (LexState *ls, stack_op *s, int prio) {
-  int op;
-  while (s->top > 0 && priority[(op=s->ops[s->top-1])] >= prio) {
-    code_0(ls, opcodes[op], op<FIRSTBIN?0:-1);
-    s->top--;
-  }
-}
 
-static void simpleexp (LexState *ls, vardesc *v) {
+static void simpleexp (LexState *ls, expdesc *v) {
   check_debugline(ls);
   switch (ls->token) {
     case NUMBER: {  /* simpleexp -> NUMBER */
       real r = ls->seminfo.r;
       next(ls);
-      code_number(ls, r);
+      v->info = luaK_number(ls, r);
       break;
     }
 
     case STRING:  /* simpleexp -> STRING */
-      code_string(ls, ls->seminfo.ts);  /* must use 'seminfo' before `next' */
+      /* must use 'seminfo' before `next' */
+      v->info = code_string(ls, ls->seminfo.ts);
       next(ls);
       break;
 
     case NIL:  /* simpleexp -> NIL */
-      adjuststack(ls, -1);
+      v->info = luaK_adjuststack(ls, -1);
       next(ls);
       break;
 
     case '{':  /* simpleexp -> constructor */
-      constructor(ls);
+      v->info = constructor(ls);
       break;
 
     case FUNCTION:  /* simpleexp -> FUNCTION body */
       next(ls);
-      body(ls, 0, ls->linenumber);
+      v->info = body(ls, 0, ls->linenumber);
       break;
 
     case '(':  /* simpleexp -> '(' expr ')' */
@@ -891,67 +690,75 @@ static void simpleexp (LexState *ls, vardesc *v) {
       return;
 
     default:
-      luaY_error(ls, "<expression> expected");
+      luaK_error(ls, "<expression> expected");
       return;
   }
-  v->k = VEXP; v->info = 0;
+  v->k = VEXP;
 }
 
 
-static void prefixexp (LexState *ls, vardesc *v, stack_op *s) {
-  /* prefixexp -> {NOT | '-'} simpleexp */
-  while (ls->token == NOT || ls->token == '-') {
-    push(ls, s, (ls->token==NOT)?INDNOT:INDMINUS);
-    next(ls);
-  }
-  simpleexp(ls, v);
+static void exp1 (LexState *ls) {
+  expdesc v;
+  expr(ls, &v);
+  luaK_2stack(ls, &v);
 }
 
 
-static void arith_exp (LexState *ls, vardesc *v) {
-  stack_op s;
-  int op;
-  s.top = 0;
-  prefixexp(ls, v, &s);
-  while ((op = binop(ls->token)) >= 0) {
-    close_exp(ls, v);
-    /* '^' is right associative, so must 'simulate' a higher priority */
-    pop_to(ls, &s, (op == POW)?priority[op]+1:priority[op]);
-    push(ls, &s, op);
-    next(ls);
-    prefixexp(ls, v, &s);
-    close_exp(ls, v);
-  }
-  if (s.top > 0) {
-    close_exp(ls, v);
-    pop_to(ls, &s, 0);
+/*
+** gets priorities of an operator. Returns the priority to the left, and
+** sets `rp' to the priority to the right.
+*/
+static int get_priority (int op, int *rp) {
+  switch (op) {
+    case AND:  case OR:
+      *rp = 1; return 1;
+    case EQ:  case  NE:
+    case  '>':  case  '<':  case  LE:  case  GE:
+      *rp = 2; return 2;
+    case  CONC:
+      *rp = 3; return 3;
+    case  '+':  case  '-':
+      *rp = 4; return 4;
+    case  '*':  case  '/':
+      *rp = 5; return 5;
+    /* priority 6 is for unary operators */
+    case  '^':
+      *rp = 7; return 8;  /* right associative */
+    default:
+      *rp = -1; return -1;
   }
 }
 
 
-static void exp1 (LexState *ls) {
-  vardesc v;
-  expr(ls, &v);
-  close_exp(ls, &v);
-}
-
-
-static void expr (LexState *ls, vardesc *v) {
-  /* expr -> arith_exp {(AND | OR) arith_exp} */
-  arith_exp(ls, v);
-  while (ls->token == AND || ls->token == OR) {
-    OpCode op = (ls->token == AND) ? ONFJMP : ONTJMP;
-    int pc;
-    close_exp(ls, v);
+/*
+** expr -> simplexep | (NOT | '-') expr | expr binop expr
+** where `binop' is any binary operator with a priority higher than `limit'
+*/
+static void operator_expr (LexState *ls, expdesc *v, int limit) {
+  int rp;
+  if (ls->token == '-' || ls->token == NOT) {
+    int op = ls->token;  /* operator */
     next(ls);
-    pc = code_S(ls, op, 0, -1);
-    arith_exp(ls, v);
-    close_exp(ls, v);
-    luaK_fixjump(ls, pc, ls->fs->pc);
+    operator_expr(ls, v, 6);  /* 6 == priority of NOT and unary `-' */
+    luaK_prefix(ls, op, v);
+  }
+  else simpleexp(ls, v);
+  /* expand while following operators have a priority higher than `limit' */
+  while (get_priority(ls->token, &rp) > limit) {
+    int op = ls->token;  /* operator */
+    expdesc v2;
+    luaK_infix(ls, v);
+    next(ls);
+    operator_expr(ls, &v2, rp);
+    luaK_posfix(ls, op, v, &v2);
   }
 }
 
 
+static void expr (LexState *ls, expdesc *v) {
+  operator_expr(ls, v, -1);
+}
+
 /* }==================================================================== */
 
 
@@ -967,21 +774,21 @@ static void block (LexState *ls) {
   FuncState *fs = ls->fs;
   int nlocalvar = fs->nlocalvar;
   chunk(ls);
-  adjuststack(ls, fs->nlocalvar - nlocalvar);
+  luaK_adjuststack(ls, fs->nlocalvar - nlocalvar);
   for (; fs->nlocalvar > nlocalvar; fs->nlocalvar--)
     luaI_unregisterlocalvar(ls, fs->lastsetline);
 }
 
 
-static int assignment (LexState *ls, vardesc *v, int nvars) {
+static int assignment (LexState *ls, expdesc *v, int nvars) {
   int left = 0;
   checklimit(ls, nvars, MAXVARSLH, "variables in a multiple assignment");
   if (ls->token == ',') {  /* assignment -> ',' NAME assignment */
-    vardesc nv;
+    expdesc nv;
     next(ls);
     var_or_func(ls, &nv);
     if (nv.k == VEXP)
-      luaY_error(ls, "syntax error");
+      luaK_error(ls, "syntax error");
     left = assignment(ls, &nv, nvars+1);
   }
   else {  /* assignment -> '=' explist1 */
@@ -994,10 +801,10 @@ static int assignment (LexState *ls, vardesc *v, int nvars) {
   }
   if (v->k != VINDEXED || left+(nvars-1) == 0) {
     /* global/local var or indexed var without values in between */
-    storevar(ls, v);
+    luaK_storevar(ls, v);
   }
   else {  /* indexed var with values in between*/
-    code_U(ls, SETTABLE, left+(nvars-1), -1);
+    luaK_U(ls, SETTABLE, left+(nvars-1), -1);
     left += 2;  /* table&index are not popped, because they aren't on top */
   }
   return left;
@@ -1021,20 +828,20 @@ static void whilestat (LexState *ls, int line) {
   cond_size = fs->pc - while_init;
   /* save condition (to move it to after body) */
   if (cond_size > MAX_WHILE_EXP)
-    luaY_error(ls, "while condition too complex");
+    luaK_error(ls, "while condition too complex");
   for (i=0; i<cond_size; i++) buffer[i] = fs->f->code[while_init+i];
   /* go back to state prior condition */
   fs->pc = while_init;
-  deltastack(ls, -1);
-  code_S(ls, JMP, 0, 0);  /* initial jump to condition */
+  luaK_deltastack(ls, -1);
+  luaK_S(ls, JMP, 0, 0);  /* initial jump to condition */
   check(ls, DO);
   block(ls);
   check_match(ls, END, WHILE, line);
   luaK_fixjump(ls, while_init, fs->pc);
   /* copy condition to new position, and correct stack */
   for (i=0; i<cond_size; i++) luaK_primitivecode(ls, buffer[i]);
-  deltastack(ls, 1);
-  luaK_fixjump(ls, code_S(ls, IFTJMP, 0, -1), while_init+1);
+  luaK_deltastack(ls, 1);
+  luaK_fixjump(ls, luaK_S(ls, IFTJMP, 0, -1), while_init+1);
 }
 
 
@@ -1046,7 +853,7 @@ static void repeatstat (LexState *ls, int line) {
   block(ls);
   check_match(ls, UNTIL, REPEAT, line);
   exp1(ls);
-  luaK_fixjump(ls, code_S(ls, IFFJMP, 0, -1), repeat_init);
+  luaK_fixjump(ls, luaK_S(ls, IFFJMP, 0, -1), repeat_init);
 }
 
 
@@ -1068,10 +875,8 @@ static void decinit (LexState *ls, listdesc *d) {
     next(ls);
     explist1(ls, d);
   }
-  else {
+  else
     d->n = 0;
-    d->pc = 0;
-  }
 }
 
 
@@ -1089,15 +894,15 @@ static void localstat (LexState *ls) {
 }
 
 
-static int funcname (LexState *ls, vardesc *v) {
+static int funcname (LexState *ls, expdesc *v) {
   /* funcname -> NAME [':' NAME | '.' NAME] */
   int needself = 0;
   singlevar(ls, str_checkname(ls), v, 0);
   if (ls->token == ':' || ls->token == '.') {
     needself = (ls->token == ':');
     next(ls);
-    close_exp(ls, v);
-    code_kstr(ls, checkname(ls));
+    luaK_2stack(ls, v);
+    luaK_kstr(ls, checkname(ls));
     v->k = VINDEXED;
   }
   return needself;
@@ -1107,31 +912,31 @@ static int funcname (LexState *ls, vardesc *v) {
 static int funcstat (LexState *ls, int line) {
   /* funcstat -> FUNCTION funcname body */
   int needself;
-  vardesc v;
+  expdesc v;
   if (ls->fs->prev)  /* inside other function? */
     return 0;
   check_debugline(ls);
   next(ls);
   needself = funcname(ls, &v);
   body(ls, needself, line);
-  storevar(ls, &v);
+  luaK_storevar(ls, &v);
   return 1;
 }
 
 
 static void namestat (LexState *ls) {
   /* stat -> func | ['%'] NAME assignment */
-  vardesc v;
+  expdesc v;
   check_debugline(ls);
   var_or_func(ls, &v);
   if (v.k == VEXP) {  /* stat -> func */
-    if (v.info == 0)  /* is just an upper value? */
-      luaY_error(ls, "syntax error");
-    close_call(ls, v.info, 0);  /* call statement uses no results */
+    if (!luaK_iscall(ls, v.info))  /* is just an upper value? */
+      luaK_error(ls, "syntax error");
+    luaK_setcallreturns(ls, v.info, 0);  /* call statement uses no results */
   }
   else {  /* stat -> ['%'] NAME assignment */
     int left = assignment(ls, &v, 1);
-    adjuststack(ls, left);  /* remove eventual garbage left on stack */
+    luaK_adjuststack(ls, left);  /* remove eventual garbage left on stack */
   }
 }
 
@@ -1144,10 +949,10 @@ static void ifpart (LexState *ls, int line) {
   int elseinit;
   next(ls);  /* skip IF or ELSEIF */
   exp1(ls);  /* cond */
-  c = code_S(ls, IFFJMP, 0, -1);  /* jump `then' if `cond' is false */
+  c = luaK_S(ls, IFFJMP, 0, -1);  /* jump `then' if `cond' is false */
   check(ls, THEN);
   block(ls);  /* `then' part */
-  je = code_S(ls, JMP, 0, 0);  /* jump `else' part after `then' */
+  je = luaK_S(ls, JMP, 0, 0);  /* jump `else' part after `then' */
   elseinit = fs->pc;
   if (ls->token == ELSEIF)
     ifpart(ls, line);
@@ -1234,20 +1039,20 @@ static void parlist (LexState *ls) {
           case NAME:  /* tailparlist -> NAME [',' tailparlist] */
             goto init;
 
-          default: luaY_error(ls, "<name> or `...' expected");
+          default: luaK_error(ls, "<name> or `...' expected");
         }
       }
       break;
 
     case ')': break;  /* parlist -> empty */
 
-    default: luaY_error(ls, "<name> or `...' expected");
+    default: luaK_error(ls, "<name> or `...' expected");
   }
   code_args(ls, nparams, dots);
 }
 
 
-static void body (LexState *ls, int needself, int line) {
+static int body (LexState *ls, int needself, int line) {
   /* body ->  '(' parlist ')' chunk END */
   FuncState new_fs;
   init_state(ls, &new_fs, ls->fs->f->source);
@@ -1260,7 +1065,7 @@ static void body (LexState *ls, int needself, int line) {
   chunk(ls);
   check_match(ls, END, FUNCTION, line);
   close_func(ls);
-  func_onstack(ls, &new_fs);
+  return func_onstack(ls, &new_fs);
 }
 
 
@@ -1271,8 +1076,7 @@ static void ret (LexState *ls) {
     check_debugline(ls);
     next(ls);
     explist(ls, &e); 
-    close_call(ls, e.pc, MULT_RET);
-    code_U(ls, RETCODE, ls->fs->nlocalvar, 0);
+    luaK_U(ls, RETCODE, ls->fs->nlocalvar, 0);
     ls->fs->stacksize = ls->fs->nlocalvar;  /* removes all temp values */
     optional(ls, ';');
   }

lparser.h

@@ -1,5 +1,5 @@
 /*
-** $Id: lparser.h,v 1.5 1999/11/22 13:12:07 roberto Exp roberto $
+** $Id: lparser.h,v 1.6 2000/02/22 13:30:11 roberto Exp roberto $
 ** LL(1) Parser and code generator for Lua
 ** See Copyright Notice in lua.h
 */
@@ -50,21 +50,19 @@
 
 
 /*
-** Variable descriptor:
-** must include an `exp' option because LL(1) cannot distinguish
-** between variables, upvalues and function calls on first sight.
+** Expression descriptor
 */
 typedef enum {
   VGLOBAL,  /* info is constant index of global name */
   VLOCAL,   /* info is stack index */
   VINDEXED, /* no info (table and index are on the stack) */
-  VEXP      /* info is pc index of a call (or 0 if exp is closed) */
-} varkind;
+  VEXP      /* info is pc index of exp main operator */
+} expkind;
 
-typedef struct vardesc {
-  varkind k;
+typedef struct expdesc {
+  expkind k;
   int info;
-} vardesc;
+} expdesc;
 
 
 /* state needed to generate code for a given function */
@@ -78,7 +76,7 @@ typedef struct FuncState {
   int nupvalues;  /* number of upvalues */
   int nvars;  /* number of entries in f->locvars (-1 if no debug information) */
   int lastsetline;  /* line where last SETLINE was issued */
-  vardesc upvalues[MAXUPVALUES];  /* upvalues */
+  expdesc upvalues[MAXUPVALUES];  /* upvalues */
   TaggedString *localvar[MAXLOCALS];  /* store local variable names */
 } FuncState;