Initial Mercurial commit.

src/common/runtime.cpp

@@ -0,0 +1,278 @@
+/**
+* Copyright (c) 2006-2009 LOVE Development Team
+* 
+* This software is provided 'as-is', without any express or implied
+* warranty.  In no event will the authors be held liable for any damages
+* arising from the use of this software.
+* 
+* Permission is granted to anyone to use this software for any purpose,
+* including commercial applications, and to alter it and redistribute it
+* freely, subject to the following restrictions:
+* 
+* 1. The origin of this software must not be misrepresented; you must not
+*    claim that you wrote the original software. If you use this software
+*    in a product, an acknowledgment in the product documentation would be
+*    appreciated but is not required.
+* 2. Altered source versions must be plainly marked as such, and must not be
+*    misrepresented as being the original software.
+* 3. This notice may not be removed or altered from any source distribution.
+**/
+
+#include "runtime.h"
+
+// LOVE
+#include "Module.h"
+#include "Object.h"
+
+// STD
+#include <iostream>
+
+namespace love
+{
+
+	/**
+	* Called when an object is collected. The object is released
+	* once in this function, possibly deleting it.
+	**/
+	static int _wrap__gc(lua_State * L)
+	{
+		userdata * u = (userdata *)lua_touserdata(L, 1);
+		Object * t = (Object *)u->data;
+		t->release();
+		return 0;
+	}
+
+	/**
+	* Special garbage collector for Modules.
+	**/
+	static int _wrap__Module_gc(lua_State * L)
+	{
+		userdata * u = (userdata *)lua_touserdata(L, 1);
+		Module * t = (Module *)u->data;
+		delete t;
+		return 0;
+	}
+
+	void luax_printstack(lua_State * L)
+	{
+		for(int i = 1;i<=lua_gettop(L);i++)
+		{
+			std::cout << i << " - " << luaL_typename(L, i) << std::endl;
+		}
+	}
+
+	bool luax_toboolean(lua_State * L, int idx)
+	{
+		return (lua_toboolean(L, idx) == 1 ? true : false);
+	}
+
+	void luax_pushboolean(lua_State * L, bool b)
+	{
+		lua_pushboolean(L, b ? 1 : 0);
+	}
+
+	bool luax_optboolean(lua_State * L, int idx, bool b)
+	{
+		if(lua_isboolean(L, idx) == 1)
+			return (lua_toboolean(L, idx) == 1 ? true : false);
+		return b;
+	}
+
+	int luax_assert_argc(lua_State * L, int lower)
+	{
+		int argc = lua_gettop(L);
+		if( argc < lower )
+			return luaL_error(L, "Incorrect number of arguments. Got [%d], expected at least [%d]", argc, lower);
+		return 0;
+	}
+
+	int luax_assert_argc(lua_State * L, int lower, int upper)
+	{
+		int argc = lua_gettop(L);
+		if( argc < lower || argc > upper)
+			return luaL_error(L, "Incorrect number of arguments. Got [%d], expected [%d-%d]", argc, lower, upper);
+		return 0;
+	}
+
+	int luax_assert_function(lua_State * L, int n)
+	{
+		if(!lua_isfunction(L, n))
+			return luaL_error(L, "Argument must be of type \"function\".");
+		return 0;
+	}
+
+	int luax_register_gc(lua_State * L, const char * mname, Module * m)
+	{
+		lua_getglobal(L, "love");
+		lua_getfield(L, -1, "__fin");
+
+		userdata * u = (userdata *)lua_newuserdata(L, sizeof(userdata));
+		u->own = true;
+		u->data = m;
+
+		luaL_newmetatable(L, mname);
+		lua_pushcfunction(L, _wrap__Module_gc);
+		lua_setfield(L, -2, "__gc");
+		lua_setmetatable(L, -2);
+
+		lua_setfield(L, -2, mname);
+
+		lua_pop(L, 2); // __fin, love
+
+		return 0;
+	}
+
+	int luax_register_info(lua_State * L, const char * name, 
+		const char * provides, const char * desc, const char * author,
+		lua_CFunction open)
+	{
+		lua_getglobal(L, "love");
+		lua_getfield(L, -1, "__mod");
+		lua_getfield(L, -1, provides);
+
+		// Make sure it exists.
+		if(lua_isnil(L, -1))
+		{
+			lua_pop(L, 1);
+			lua_newtable(L);
+			lua_setfield(L, -2, provides);
+			lua_getfield(L, -1, provides);
+		}
+
+		lua_newtable(L);
+
+		lua_pushstring(L, desc);
+		lua_setfield(L, -2, "description");
+		lua_pushstring(L, author);
+		lua_setfield(L, -2, "author");
+		lua_pushcfunction(L, open);
+		lua_setfield(L, -2, "open");
+
+		lua_setfield(L, -2, name);
+
+		// Pop "provides", __mod, love
+		lua_pop(L, 3);
+		return 0;
+	}
+
+	int luax_register_module(lua_State * L, const luaL_Reg * fn, const lua_CFunction * types)
+	{
+		// Create new table for module.
+		lua_newtable(L);
+
+		// Register all the functions.
+		luaL_register(L, 0, fn);
+
+		// Register types.
+		if(types != 0)
+			for(const lua_CFunction * t = types; *t != 0; t++)
+				(*t)(L);
+
+		return 1;
+	}
+
+	int luax_preload(lua_State * L, lua_CFunction f, const char * name)
+	{
+		lua_getglobal(L, "package");
+		lua_getfield(L, -1, "preload");
+		lua_pushcfunction(L, f);
+		lua_setfield(L, -2, name);
+		lua_pop(L, 2);
+		return 0;
+	}
+
+	int luax_register_type(lua_State * L, const char * tname, const luaL_Reg * fn)
+	{
+		luaL_newmetatable(L, tname);
+
+		// m.__index = m
+		lua_pushvalue(L, -1);
+		lua_setfield(L, -2, "__index");
+
+		// setup gc
+		lua_pushcfunction(L, _wrap__gc);
+		lua_setfield(L, -2, "__gc");
+
+		luaL_register(L, 0, fn);
+		lua_pop(L, 1); // Pops metatable.
+		return 0;
+	}
+
+	int luax_register_searcher(lua_State * L, lua_CFunction f)
+	{
+		// Add the package loader to the package.loaders table.
+		lua_getglobal(L, "package");
+
+		if(lua_isnil(L, -1))
+			return luaL_error(L, "Can't register searcher: package table does not exist.");
+		
+		lua_getfield(L, -1, "loaders");
+		
+		if(lua_isnil(L, -1))
+			return luaL_error(L, "Can't register searcher: package.loaders table does not exist.");
+		
+		int len = lua_objlen(L, -1);
+		lua_pushinteger(L, len+1);
+		lua_pushcfunction(L, f);
+		lua_settable(L, -3);
+		lua_pop(L, 2);
+		return 0;
+	}
+
+	void luax_newtype(lua_State * L, const char * tname, bits flags, void * data, bool own)
+	{
+		userdata * u = (userdata *)lua_newuserdata(L, sizeof(userdata));
+
+		u->data = data;
+		u->flags = flags;
+		u->own = own;
+
+		luaL_newmetatable(L, tname);
+		lua_setmetatable(L, -2);
+	}
+
+	bool luax_istype(lua_State * L, int idx, love::bits type)
+	{
+		if(lua_isuserdata(L, idx) == 0)
+			return false;
+
+		userdata * u = (userdata *)lua_touserdata(L, idx);
+
+		return ((u->flags & type) == type);
+	}
+
+	int luax_getfunction(lua_State * L, const char * mod, const char * fn)
+	{
+		lua_getglobal(L, "love");
+		if(lua_isnil(L, -1)) return luaL_error(L, "Could not find global love!");
+		lua_getfield(L, -1, mod);
+		if(lua_isnil(L, -1)) return luaL_error(L, "Could not find love.%s!", mod);
+		lua_getfield(L, -1, fn);
+		if(lua_isnil(L, -1)) return luaL_error(L, "Could not find love.%s.%s!", mod, fn);
+		
+		lua_remove(L, -2); // remove mod
+		lua_remove(L, -2); // remove fn
+		return 0;
+	}
+
+	int luax_convobj(lua_State * L, int idx, const char * mod, const char * fn)
+	{
+		// Convert string to a file.
+		luax_getfunction(L, mod, fn);
+		lua_pushvalue(L, idx); // The initial argument.
+		lua_call(L, 1, 1); // Call the function, one arg, one return value.
+		lua_replace(L, idx); // Replace the initial argument with the new object.
+		return 0;
+	}
+
+	int luax_strtofile(lua_State * L, int idx)
+	{
+		return luax_convobj(L, idx, "filesystem", "newFile");
+	}
+
+	int luax_filetodata(lua_State * L, int idx)
+	{
+		return luax_convobj(L, idx, "filesystem", "read");
+	}
+
+} // love