/* ** $Id: lbuiltin.c,v 1.9 1997/11/26 18:53:45 roberto Exp roberto $ ** Built-in functions ** See Copyright Notice in lua.h */ #include #include #include "lapi.h" #include "lauxlib.h" #include "lbuiltin.h" #include "ldo.h" #include "lfunc.h" #include "lmem.h" #include "lobject.h" #include "lstate.h" #include "lstring.h" #include "ltable.h" #include "ltm.h" #include "lua.h" static lua_Object tablearg (int arg) { lua_Object o = lua_getparam(arg); luaL_arg_check(lua_istable(o), arg, "table expected"); return o; } static lua_Object functionarg (int arg) { lua_Object o = lua_getparam(arg); luaL_arg_check(lua_isfunction(o), arg, "function expected"); return o; } static void pushstring (TaggedString *s) { TObject o; o.ttype = LUA_T_STRING; o.value.ts = s; luaA_pushobject(&o); } static void nextvar (void) { TObject *o = luaA_Address(luaL_nonnullarg(1)); TaggedString *g; if (ttype(o) == LUA_T_NIL) g = (TaggedString *)L->rootglobal.next; else { luaL_arg_check(ttype(o) == LUA_T_STRING, 1, "variable name expected"); g = tsvalue(o); /* check whether name is in global var list */ luaL_arg_check((GCnode *)g != g->head.next, 1, "variable name expected"); g = (TaggedString *)g->head.next; } while (g && g->u.globalval.ttype == LUA_T_NIL) /* skip globals with nil */ g = (TaggedString *)g->head.next; if (g) { pushstring(g); luaA_pushobject(&g->u.globalval); } } static void foreachvar (void) { TObject f = *luaA_Address(functionarg(1)); GCnode *g; StkId name = L->Cstack.base++; /* place to keep var name (to avoid GC) */ ttype(L->stack.stack+name) = LUA_T_NIL; L->stack.top++; for (g = L->rootglobal.next; g; g = g->next) { TaggedString *s = (TaggedString *)g; if (s->u.globalval.ttype != LUA_T_NIL) { ttype(L->stack.stack+name) = LUA_T_STRING; tsvalue(L->stack.stack+name) = s; /* keep s on stack to avoid GC */ luaA_pushobject(&f); pushstring(s); luaA_pushobject(&s->u.globalval); luaD_call((L->stack.top-L->stack.stack)-2, 1); if (ttype(L->stack.top-1) != LUA_T_NIL) return; L->stack.top--; } } } static void next (void) { lua_Object o = tablearg(1); lua_Object r = luaL_nonnullarg(2); Node *n = luaH_next(luaA_Address(o), luaA_Address(r)); if (n) { luaA_pushobject(&n->ref); luaA_pushobject(&n->val); } } static void foreach (void) { TObject t = *luaA_Address(tablearg(1)); TObject f = *luaA_Address(functionarg(2)); int i; for (i=0; inhash; i++) { Node *nd = &(avalue(&t)->node[i]); if (ttype(ref(nd)) != LUA_T_NIL && ttype(val(nd)) != LUA_T_NIL) { luaA_pushobject(&f); luaA_pushobject(ref(nd)); luaA_pushobject(val(nd)); luaD_call((L->stack.top-L->stack.stack)-2, 1); if (ttype(L->stack.top-1) != LUA_T_NIL) return; L->stack.top--; } } } static void internaldostring (void) { if (lua_getparam(2) != LUA_NOOBJECT) lua_error("invalid 2nd argument (probably obsolete code)"); if (lua_dostring(luaL_check_string(1)) == 0) if (luaA_passresults() == 0) lua_pushuserdata(NULL); /* at least one result to signal no errors */ } static void internaldofile (void) { char *fname = luaL_opt_string(1, NULL); if (lua_dofile(fname) == 0) if (luaA_passresults() == 0) lua_pushuserdata(NULL); /* at least one result to signal no errors */ } static char *to_string (lua_Object obj) { char *buff = luaM_buffer(30); TObject *o = luaA_Address(obj); switch (ttype(o)) { case LUA_T_NUMBER: case LUA_T_STRING: return lua_getstring(obj); case LUA_T_ARRAY: { sprintf(buff, "table: %p", (void *)o->value.a); return buff; } case LUA_T_FUNCTION: { sprintf(buff, "function: %p", (void *)o->value.cl); return buff; } case LUA_T_USERDATA: { sprintf(buff, "userdata: %p", o->value.ts->u.d.v); return buff; } case LUA_T_NIL: return "nil"; default: return ""; } } static void bi_tostring (void) { lua_pushstring(to_string(lua_getparam(1))); } static void luaI_print (void) { int i = 1; lua_Object obj; while ((obj = lua_getparam(i++)) != LUA_NOOBJECT) printf("%s\t", to_string(obj)); printf("\n"); } static void luaI_type (void) { lua_Object o = luaL_nonnullarg(1); lua_pushstring(luaO_typenames[-ttype(luaA_Address(o))]); lua_pushnumber(lua_tag(o)); } static void lua_obj2number (void) { lua_Object o = lua_getparam(1); if (lua_isnumber(o)) lua_pushnumber(lua_getnumber(o)); } static void luaI_error (void) { char *s = lua_getstring(lua_getparam(1)); if (s == NULL) s = "(no message)"; lua_error(s); } static void luaI_assert (void) { lua_Object p = lua_getparam(1); if (p == LUA_NOOBJECT || lua_isnil(p)) luaL_verror("assertion failed! %.100s", luaL_opt_string(2, "")); } static void setglobal (void) { lua_Object value = luaL_nonnullarg(2); lua_pushobject(value); lua_setglobal(luaL_check_string(1)); lua_pushobject(value); /* return given value */ } static void rawsetglobal (void) { lua_Object value = luaL_nonnullarg(2); lua_pushobject(value); lua_rawsetglobal(luaL_check_string(1)); lua_pushobject(value); /* return given value */ } static void getglobal (void) { lua_pushobject(lua_getglobal(luaL_check_string(1))); } static void rawgetglobal (void) { lua_pushobject(lua_rawgetglobal(luaL_check_string(1))); } static void luatag (void) { lua_pushnumber(lua_tag(lua_getparam(1))); } static int getnarg (lua_Object table) { lua_Object temp; /* temp = table.n */ lua_pushobject(table); lua_pushstring("n"); temp = lua_rawgettable(); return (lua_isnumber(temp) ? lua_getnumber(temp) : MAX_WORD); } static void luaI_call (void) { lua_Object f = luaL_nonnullarg(1); lua_Object arg = tablearg(2); char *options = luaL_opt_string(3, ""); lua_Object err = lua_getparam(4); int narg = getnarg(arg); int i, status; if (err != LUA_NOOBJECT) { /* set new error method */ lua_pushobject(err); err = lua_seterrormethod(); } /* push arg[1...n] */ for (i=0; i