co' library goes with basic library (and not with tab')
This commit is contained in:
79
lbaselib.c
79
lbaselib.c
@@ -1,5 +1,5 @@
|
|||||||
/*
|
/*
|
||||||
** $Id: lbaselib.c,v 1.65 2002/04/09 19:46:56 roberto Exp roberto $
|
** $Id: lbaselib.c,v 1.66 2002/04/09 20:19:06 roberto Exp roberto $
|
||||||
** Basic library
|
** Basic library
|
||||||
** See Copyright Notice in lua.h
|
** See Copyright Notice in lua.h
|
||||||
*/
|
*/
|
||||||
@@ -421,6 +421,82 @@ static const luaL_reg base_funcs[] = {
|
|||||||
};
|
};
|
||||||
|
|
||||||
|
|
||||||
|
/*
|
||||||
|
** {======================================================
|
||||||
|
** Coroutine library
|
||||||
|
** =======================================================
|
||||||
|
*/
|
||||||
|
|
||||||
|
|
||||||
|
static int luaB_resume (lua_State *L) {
|
||||||
|
lua_State *co = (lua_State *)lua_getfrombox(L, lua_upvalueindex(1));
|
||||||
|
lua_settop(L, 0);
|
||||||
|
if (lua_resume(L, co) != 0)
|
||||||
|
lua_error(L, "error running co-routine");
|
||||||
|
return lua_gettop(L);
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
static int gc_coroutine (lua_State *L) {
|
||||||
|
lua_State *co = (lua_State *)lua_getfrombox(L, 1);
|
||||||
|
lua_closethread(L, co);
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
static int luaB_coroutine (lua_State *L) {
|
||||||
|
lua_State *NL;
|
||||||
|
int ref;
|
||||||
|
int i;
|
||||||
|
int n = lua_gettop(L);
|
||||||
|
luaL_arg_check(L, lua_isfunction(L, 1) && !lua_iscfunction(L, 1), 1,
|
||||||
|
"Lua function expected");
|
||||||
|
NL = lua_newthread(L);
|
||||||
|
if (NL == NULL) lua_error(L, "unable to create new thread");
|
||||||
|
/* move function and arguments from L to NL */
|
||||||
|
for (i=0; i<n; i++) {
|
||||||
|
ref = lua_ref(L, 1);
|
||||||
|
lua_getref(NL, ref);
|
||||||
|
lua_insert(NL, 1);
|
||||||
|
lua_unref(L, ref);
|
||||||
|
}
|
||||||
|
lua_cobegin(NL, n-1);
|
||||||
|
lua_newpointerbox(L, NL);
|
||||||
|
lua_pushliteral(L, "Coroutine");
|
||||||
|
lua_rawget(L, LUA_REGISTRYINDEX);
|
||||||
|
lua_setmetatable(L, -2);
|
||||||
|
lua_pushcclosure(L, luaB_resume, 1);
|
||||||
|
return 1;
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
static int luaB_yield (lua_State *L) {
|
||||||
|
return lua_yield(L, lua_gettop(L));
|
||||||
|
}
|
||||||
|
|
||||||
|
static const luaL_reg co_funcs[] = {
|
||||||
|
{"create", luaB_coroutine},
|
||||||
|
{"yield", luaB_yield},
|
||||||
|
{NULL, NULL}
|
||||||
|
};
|
||||||
|
|
||||||
|
|
||||||
|
static void co_open (lua_State *L) {
|
||||||
|
luaL_opennamedlib(L, "co", co_funcs, 0);
|
||||||
|
/* create metatable for coroutines */
|
||||||
|
lua_pushliteral(L, "Coroutine");
|
||||||
|
lua_newtable(L);
|
||||||
|
lua_pushliteral(L, "__gc");
|
||||||
|
lua_pushcfunction(L, gc_coroutine);
|
||||||
|
lua_rawset(L, -3);
|
||||||
|
lua_rawset(L, LUA_REGISTRYINDEX);
|
||||||
|
}
|
||||||
|
|
||||||
|
/* }====================================================== */
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
static void base_open (lua_State *L) {
|
static void base_open (lua_State *L) {
|
||||||
lua_pushliteral(L, "_G");
|
lua_pushliteral(L, "_G");
|
||||||
lua_pushvalue(L, LUA_GLOBALSINDEX);
|
lua_pushvalue(L, LUA_GLOBALSINDEX);
|
||||||
@@ -434,6 +510,7 @@ static void base_open (lua_State *L) {
|
|||||||
|
|
||||||
LUALIB_API int lua_baselibopen (lua_State *L) {
|
LUALIB_API int lua_baselibopen (lua_State *L) {
|
||||||
base_open(L);
|
base_open(L);
|
||||||
|
co_open(L);
|
||||||
/* `require' needs an empty table as upvalue */
|
/* `require' needs an empty table as upvalue */
|
||||||
lua_newtable(L);
|
lua_newtable(L);
|
||||||
lua_pushcclosure(L, luaB_require, 1);
|
lua_pushcclosure(L, luaB_require, 1);
|
||||||
|
|||||||
78
ltablib.c
78
ltablib.c
@@ -1,5 +1,5 @@
|
|||||||
/*
|
/*
|
||||||
** $Id: $
|
** $Id: ltablib.c,v 1.1 2002/04/09 20:19:06 roberto Exp roberto $
|
||||||
** Library for Table Manipulation
|
** Library for Table Manipulation
|
||||||
** See Copyright Notice in lua.h
|
** See Copyright Notice in lua.h
|
||||||
*/
|
*/
|
||||||
@@ -214,81 +214,6 @@ static int luaB_sort (lua_State *L) {
|
|||||||
/* }====================================================== */
|
/* }====================================================== */
|
||||||
|
|
||||||
|
|
||||||
/*
|
|
||||||
** {======================================================
|
|
||||||
** Coroutine library
|
|
||||||
** =======================================================
|
|
||||||
*/
|
|
||||||
|
|
||||||
|
|
||||||
static int luaB_resume (lua_State *L) {
|
|
||||||
lua_State *co = (lua_State *)lua_getfrombox(L, lua_upvalueindex(1));
|
|
||||||
lua_settop(L, 0);
|
|
||||||
if (lua_resume(L, co) != 0)
|
|
||||||
lua_error(L, "error running co-routine");
|
|
||||||
return lua_gettop(L);
|
|
||||||
}
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
static int gc_coroutine (lua_State *L) {
|
|
||||||
lua_State *co = (lua_State *)lua_getfrombox(L, 1);
|
|
||||||
lua_closethread(L, co);
|
|
||||||
return 0;
|
|
||||||
}
|
|
||||||
|
|
||||||
|
|
||||||
static int luaB_coroutine (lua_State *L) {
|
|
||||||
lua_State *NL;
|
|
||||||
int ref;
|
|
||||||
int i;
|
|
||||||
int n = lua_gettop(L);
|
|
||||||
luaL_arg_check(L, lua_isfunction(L, 1) && !lua_iscfunction(L, 1), 1,
|
|
||||||
"Lua function expected");
|
|
||||||
NL = lua_newthread(L);
|
|
||||||
if (NL == NULL) lua_error(L, "unable to create new thread");
|
|
||||||
/* move function and arguments from L to NL */
|
|
||||||
for (i=0; i<n; i++) {
|
|
||||||
ref = lua_ref(L, 1);
|
|
||||||
lua_getref(NL, ref);
|
|
||||||
lua_insert(NL, 1);
|
|
||||||
lua_unref(L, ref);
|
|
||||||
}
|
|
||||||
lua_cobegin(NL, n-1);
|
|
||||||
lua_newpointerbox(L, NL);
|
|
||||||
lua_pushliteral(L, "Coroutine");
|
|
||||||
lua_rawget(L, LUA_REGISTRYINDEX);
|
|
||||||
lua_setmetatable(L, -2);
|
|
||||||
lua_pushcclosure(L, luaB_resume, 1);
|
|
||||||
return 1;
|
|
||||||
}
|
|
||||||
|
|
||||||
|
|
||||||
static int luaB_yield (lua_State *L) {
|
|
||||||
return lua_yield(L, lua_gettop(L));
|
|
||||||
}
|
|
||||||
|
|
||||||
static const luaL_reg co_funcs[] = {
|
|
||||||
{"create", luaB_coroutine},
|
|
||||||
{"yield", luaB_yield},
|
|
||||||
{NULL, NULL}
|
|
||||||
};
|
|
||||||
|
|
||||||
|
|
||||||
static void co_open (lua_State *L) {
|
|
||||||
luaL_opennamedlib(L, "co", co_funcs, 0);
|
|
||||||
/* create metatable for coroutines */
|
|
||||||
lua_pushliteral(L, "Coroutine");
|
|
||||||
lua_newtable(L);
|
|
||||||
lua_pushliteral(L, "__gc");
|
|
||||||
lua_pushcfunction(L, gc_coroutine);
|
|
||||||
lua_rawset(L, -3);
|
|
||||||
lua_rawset(L, LUA_REGISTRYINDEX);
|
|
||||||
}
|
|
||||||
|
|
||||||
/* }====================================================== */
|
|
||||||
|
|
||||||
|
|
||||||
static const luaL_reg tab_funcs[] = {
|
static const luaL_reg tab_funcs[] = {
|
||||||
{"foreach", luaB_foreach},
|
{"foreach", luaB_foreach},
|
||||||
{"foreachi", luaB_foreachi},
|
{"foreachi", luaB_foreachi},
|
||||||
@@ -302,7 +227,6 @@ static const luaL_reg tab_funcs[] = {
|
|||||||
|
|
||||||
LUALIB_API int lua_tablibopen (lua_State *L) {
|
LUALIB_API int lua_tablibopen (lua_State *L) {
|
||||||
luaL_opennamedlib(L, "tab", tab_funcs, 0);
|
luaL_opennamedlib(L, "tab", tab_funcs, 0);
|
||||||
co_open(L);
|
|
||||||
return 0;
|
return 0;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user