#ifndef __LUA_TEMPLATES_H
#define __LUA_TEMPLATES_H

/*
LUA_TEMPLATES.H

	Copyright (C) 2008 by Gregory Smith
 
	This program is free software; you can redistribute it and/or modify
	it under the terms of the GNU General Public License as published by
	the Free Software Foundation; either version 3 of the License, or
	(at your option) any later version.

	This program is distributed in the hope that it will be useful,
	but WITHOUT ANY WARRANTY; without even the implied warranty of
	MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
	GNU General Public License for more details.

	This license is contained in the file "COPYING",
	which is included with this source code; it is available online at
	http://www.gnu.org/licenses/gpl.html

	Templates to help create the Lua/C interface
*/

#include "cseries.h"

#ifdef HAVE_LUA
extern "C"
{
#include "lua.h"
#include "lauxlib.h"
#include "lualib.h"
}

#include <boost/function.hpp>
#include "lua_script.h"
#include "lua_mnemonics.h" // for lang_def and mnemonics
#include <sstream>
#include <map>

static inline int luaL_typerror(lua_State* L, int narg, const char* tname)
{
	const char *msg = lua_pushfstring(L, "%s expected, got %s",
					  tname, luaL_typename(L, narg));
	return luaL_argerror(L, narg, msg);
}

// helper functions for finding the script's data path
// not in lua_script.h because they expose lua_State
extern void L_Set_Search_Path(lua_State* L, const std::string& path);
extern std::string L_Get_Search_Path(lua_State* L);

extern bool L_Get_Proper_Item_Accounting(lua_State* L);
extern void L_Set_Proper_Item_Accounting(lua_State* L, bool value);

// pushes a function that returns the parameterized function
template<lua_CFunction f>
int L_TableFunction(lua_State *L)
{
	lua_pushcfunction(L, f);
	return 1;
}

template<char *name, typename index_t = int16>
class L_Class {
public:

	index_t m_index;
	typedef index_t index_type;

	static void Register(lua_State *L, const luaL_Reg get[] = 0, const luaL_Reg set[] = 0, const luaL_Reg metatable[] = 0);
	static L_Class *Push(lua_State *L, index_t index);
	static index_t Index(lua_State *L, int index);
	static bool Is(lua_State *L, int index);
	static void Invalidate(lua_State *L, index_t index);
	static boost::function<bool (index_t)> Valid;
	struct ValidRange
	{
		ValidRange(int32 max_index) : m_max(max_index) {}
		bool operator() (int32 index)
		{
			return (index >= 0 && index < m_max);
		}

		int32 m_max;
	};

	// ghs: codewarrior chokes on this:
	//	template<index_t max_index> static bool ValidRange(index_t index) { return index >= 0 && index < max_index; }
private:
	// C functions for Lua
	static int _index(lua_State *L);
	static int _get(lua_State *L);
	static int _set(lua_State *L);
	static int _is(lua_State *L);
	static int _new(lua_State *L);
	static int _tostring(lua_State *L);

protected:
	// registry keys
	static void _push_get_methods_key(lua_State *L) {
		lua_pushlightuserdata(L, (void *) (&name[1]));
	}

	static void _push_set_methods_key(lua_State *L) {
		lua_pushlightuserdata(L, (void *) (&name[2]));
	}

	static void _push_instances_key(lua_State *L) {
		lua_pushlightuserdata(L, (void *) (&name[3]));
	}

	// special tables
	static void _push_custom_fields_table(lua_State *L);
};

struct always_valid
{
	bool operator()(int32 x) { return true; }
};

template<char *name, typename index_t>
boost::function<bool (index_t)> L_Class<name, index_t>::Valid = always_valid();

template<char *name, typename index_t>
void L_Class<name, index_t>::Register(lua_State *L, const luaL_Reg get[], const luaL_Reg set[], const luaL_Reg metatable[])
{
	// create the metatable itself
	luaL_newmetatable(L, name);

	// Lua 5.1 doesn't do this reverse mapping any more--do it ourselves
	lua_pushvalue(L, -1);
	lua_pushstring(L, name);
	lua_settable(L, LUA_REGISTRYINDEX);

	// register metatable get
	lua_pushcfunction(L, _get);
	lua_setfield(L, -2, "__index");

	// register metatable set
	lua_pushcfunction(L, _set);
	lua_setfield(L, -2, "__newindex");

	// register metatable tostring
	lua_pushcfunction(L, _tostring);
	lua_setfield(L, -2, "__tostring");

	lua_pushcfunction(L, _new);
	lua_setfield(L, -2, "__new");

	if (metatable)
		luaL_setfuncs(L, metatable, 0);
	
	// clear the stack
	lua_pop(L, 1);
	
	// register get methods
	_push_get_methods_key(L);
	lua_newtable(L);

	// always want index
	lua_pushcfunction(L, _index);
	lua_setfield(L, -2, "index");

	if (get)
		luaL_setfuncs(L, get, 0);
	lua_settable(L, LUA_REGISTRYINDEX);

	// register set methods
	_push_set_methods_key(L);
	lua_newtable(L);

	if (set)
		luaL_setfuncs(L, set, 0);
	lua_settable(L, LUA_REGISTRYINDEX);
		
	// register a table for instances
	_push_instances_key(L);
	lua_newtable(L);
	lua_settable(L, LUA_REGISTRYINDEX);

	// register is_
	lua_pushcfunction(L, _is);
	std::string is_name = "is_" + std::string(name);
	lua_setglobal(L, is_name.c_str());
}

template<char *name, typename index_t>
L_Class<name, index_t> *L_Class<name, index_t>::Push(lua_State *L, index_t index)
{
	L_Class<name, index_t>* t = 0;

	if (!Valid(index))
	{
		lua_pushnil(L);
		return 0;
	}

	// look it up in the index table
	_push_instances_key(L);
	lua_gettable(L, LUA_REGISTRYINDEX);

	lua_pushnumber(L, index);
	lua_gettable(L, -2);

	if (lua_isnil(L, -1)) 
	{
		lua_pop(L, 1);

		// create an instance
		t = static_cast<L_Class<name, index_t> *>(lua_newuserdata(L, sizeof(L_Class<name, index_t>)));
		luaL_getmetatable(L, name);
		lua_setmetatable(L, -2);
		t->m_index = index;

		// insert it into the instance table
		lua_pushnumber(L, index);
		lua_pushvalue(L, -2);
		lua_settable(L, -4);

	}
	else
	{
		t = static_cast<L_Class<name, index_t> *>(lua_touserdata(L, -1));
	}

	// remove the instance table
	lua_remove(L, -2);
		
	return t;
}

template<char *name, typename index_t>
index_t L_Class<name, index_t>::Index(lua_State *L, int index)
{
	L_Class<name, index_t> *t = static_cast<L_Class<name, index_t> *>(lua_touserdata(L, index));
	if (!t) luaL_typerror(L, index, name);
	return t->m_index;
}

template<char *name, typename index_t>
bool L_Class<name, index_t>::Is(lua_State *L, int index)
{
	L_Class<name, index_t>* t = static_cast<L_Class<name, index_t>*>(lua_touserdata(L, index));
	if (!t) return false;

	if (lua_getmetatable(L, index))
	{
		lua_getfield(L, LUA_REGISTRYINDEX, name);
		if (lua_rawequal(L, -1, -2))
		{
			lua_pop(L, 2);
			return true;
		}
		else
		{
			lua_pop(L, 2);
			return false;
		}
	}

	return false;
}

template<char *name, typename index_t>
void L_Class<name, index_t>::Invalidate(lua_State *L, index_t index)
{
	// remove it from the index table
	_push_instances_key(L);
	lua_gettable(L, LUA_REGISTRYINDEX);

	lua_pushnumber(L, index);
	lua_pushnil(L);
	lua_settable(L, -3);
	lua_pop(L, 1);

	// clear custom fields
	lua_pushlightuserdata(L, (void *) (L_Persistent_Table_Key()));
	lua_gettable(L, LUA_REGISTRYINDEX);
	lua_getfield(L, -1, name);
	if (!lua_isnil(L, -1))
	{
		lua_pushnumber(L, index);
		lua_pushnil(L);
		lua_settable(L, -3);
	}
	lua_pop(L, 2);
}

template<char *name, typename index_t>
int L_Class<name, index_t>::_index(lua_State *L)
{
	lua_pushnumber(L, Index(L, 1));
	return 1;
}

template<char *name, typename index_t>
int L_Class<name, index_t>::_is(lua_State *L)
{
	lua_pushboolean(L, Is(L, 1));
	return 1;
}

template<char *name, typename index_t>
int L_Class<name, index_t>::_get(lua_State *L)
{
	if (lua_isstring(L, 2))
	{
		luaL_checktype(L, 1, LUA_TUSERDATA);
		luaL_checkudata(L, 1, name);
		if (!Valid(Index(L, 1)) && strcmp(lua_tostring(L, 2), "valid") != 0 && strcmp(lua_tostring(L, 2), "index") != 0)
			luaL_error(L, "invalid object");

		if (lua_tostring(L, 2)[0] == '_')
		{
			_push_custom_fields_table(L);
			lua_pushnumber(L, Index(L, 1));
			lua_gettable(L, -2);
			if (lua_istable(L, -1))
			{
				lua_pushvalue(L, 2);
				lua_gettable(L, -2);
				lua_remove(L, -2);
			}
			else 
			{
				lua_pop(L, 1);
				lua_pushnil(L);
			}

			lua_remove(L, -2);
		}
		else
		{
			// pop the get table
			_push_get_methods_key(L);
			lua_gettable(L, LUA_REGISTRYINDEX);

			// get the function from that table
			lua_pushvalue(L, 2);
			lua_gettable(L, -2);
			lua_remove(L, -2);
		
			if (lua_isfunction(L, -1))
			{
				// execute the function with table as our argument
				lua_pushvalue(L, 1);
				if (lua_pcall(L, 1, 1, 0) == LUA_ERRRUN)
				{
					// report the error as being on this line
					luaL_where(L, 1);
					lua_pushvalue(L, -2);
					lua_concat(L, 2);
					lua_error(L);
				}
			}
			else
			{
				lua_pop(L, 1);
				lua_pushnil(L);
			}
		}
	}
	else
	{
		lua_pushnil(L);
	}

	return 1;
}

template<char *name, typename index_t>
int L_Class<name, index_t>::_new(lua_State *L)
{
	if (lua_isnumber(L, 1))
	{
		index_t index = static_cast<index_t>(lua_tonumber(L, 1));
		lua_pop(L, 1);
		Push(L, index);
		return 1;
	}
	else
	{
		return 0;
	}
}

template<char *name, typename index_t>
int L_Class<name, index_t>::_set(lua_State *L)
{
	luaL_checktype(L, 1, LUA_TUSERDATA);
	luaL_checkudata(L, 1, name);

	if (lua_isstring(L, 2) && lua_tostring(L, 2)[0] == '_')
	{
		_push_custom_fields_table(L);
		lua_pushnumber(L, Index(L, 1));
		lua_gettable(L, -2);
		if (lua_istable(L, -1)) 
		{
			lua_pushvalue(L, 2);
			lua_pushvalue(L, 3);
			lua_settable(L, -3);
			lua_pop(L, 1);
		}
		else
		{
			lua_pop(L, 1);

			lua_newtable(L);
			
			lua_pushnumber(L, Index(L, 1));
			lua_pushvalue(L, -2);
			lua_settable(L, -4);
			
			lua_pushvalue(L, 2);
			lua_pushvalue(L, 3);
			lua_settable(L, -3);
			lua_pop(L, 1);
		}

		lua_pop(L, 1);
	}
	else
	{
		// pop the set table
		_push_set_methods_key(L);
		lua_gettable(L, LUA_REGISTRYINDEX);
		
		// get the function from that table
		lua_pushvalue(L, 2);
		lua_gettable(L, -2);
		
		if (lua_isnil(L, -1))
		{
			luaL_error(L, "no such index");
		}
		
		// execute the function with table, value as our arguments
		lua_pushvalue(L, 1);
		lua_pushvalue(L, 3);
		if (lua_pcall(L, 2, 0, 0) == LUA_ERRRUN)
		{
			// report the error as being on this line
			luaL_where(L, 1);
			lua_pushvalue(L, -2);
			lua_concat(L, 2);
			lua_error(L);
		}
		
		lua_pop(L, 1);
	}

	return 0;
}

template<char *name, typename index_t>
int L_Class<name, index_t>::_tostring(lua_State *L)
{
	std::ostringstream s;
	s << name << " " << Index(L, 1);
	lua_pushstring(L, s.str().c_str());
	return 1;
}


template<char *name, typename index_t>
void L_Class<name, index_t>::_push_custom_fields_table(lua_State *L)
{
	lua_pushlightuserdata(L, (void *) (L_Persistent_Table_Key()));
	lua_gettable(L, LUA_REGISTRYINDEX);

	lua_getfield(L, -1, name);
	if (lua_isnil(L, -1))
	{
		// remove the nil
		lua_pop(L, 1);

		// push a new table
		lua_newtable(L);
		lua_pushvalue(L, -1);
		lua_setfield(L, -3, name);
	}
	
	lua_remove(L, -2);
}


// enum classes define equality with numbers
template<char *name, typename index_t = int16>
class L_Enum : public L_Class<name, index_t>
{
public:
	static void Register(lua_State *L, const luaL_Reg get[] = 0, const luaL_Reg set[] = 0, const luaL_Reg metatable[] = 0, const lang_def mnemonics[] = 0);
	static index_t ToIndex(lua_State *L, int index);
	static void PushMnemonicTable(lua_State *L);
protected:
	static bool _lookup(lua_State *L, int index, index_t& to);
	static int _equals(lua_State *L);
private:
	static int _get_mnemonic(lua_State *L);
	static int _set_mnemonic(lua_State *L);

	static void _push_mnemonic_key(lua_State *L) {
		lua_pushlightuserdata(L, (void *) (&name[4]));
	}
};

// the justification for this specialization is long
template<char *name, typename index_t = int16>
class L_LazyEnum : public L_Enum<name, index_t>
{
public:
	static index_t ToIndex(lua_State *L, int index) {
		index_t to;
		if(lua_isnil(L, index)) return -1;
		else if(L_Enum<name, index_t>::_lookup(L, index, to)) return to;
		else {
			std::string error;
			if(lua_isnumber(L, index) || lua_isstring(L, index))
				error = std::string(name) + ": invalid index";
			else
				error = std::string(name) + ": incorrect argument type";
			return luaL_error(L, error.c_str());
		}
	}
};

template<char *name, typename index_t>
void L_Enum<name, index_t>::Register(lua_State *L, const luaL_Reg get[], const luaL_Reg set[], const luaL_Reg metatable[], const lang_def mnemonics[])
{
	L_Class<name, index_t>::Register(L, get, set, 0);

	// add mnemonic accessors to get/set tables
	L_Class<name, index_t>::_push_get_methods_key(L);
	lua_gettable(L, LUA_REGISTRYINDEX);
	lua_pushcfunction(L, _get_mnemonic);
	lua_setfield(L, -2, "mnemonic");
	lua_pop(L, 1);

	L_Class<name, index_t>::_push_set_methods_key(L);
	lua_gettable(L, LUA_REGISTRYINDEX);
	lua_pushcfunction(L, _set_mnemonic);
	lua_setfield(L, -2, "mnemonic");
	lua_pop(L, 1);

	// add __eq and __tostring to metatable
	luaL_getmetatable(L, name);

	lua_pushcfunction(L, _equals);
	lua_setfield(L, -2, "__eq");

	lua_pushcfunction(L, _get_mnemonic);
	lua_setfield(L, -2, "__tostring");

	if (metatable)
		luaL_setfuncs(L, metatable, 0);

	lua_pop(L, 1);

	// load mnemonics to mnemonic table
	if (mnemonics)
	{
		_push_mnemonic_key(L);
		lua_newtable(L);
		
		const lang_def *mnemonic = mnemonics;
		while (mnemonic->name)
		{
			lua_pushstring(L, mnemonic->name);
			lua_pushnumber(L, mnemonic->value);
			lua_settable(L, -3);

			lua_pushnumber(L, mnemonic->value);
			lua_pushstring(L, mnemonic->name);
			lua_settable(L, -3);
			
			mnemonic++;
		}

		lua_settable(L, LUA_REGISTRYINDEX);
	}
}

template<char *name, typename index_t>
void L_Enum<name, index_t>::PushMnemonicTable(lua_State *L)
{
	_push_mnemonic_key(L);
	lua_gettable(L, LUA_REGISTRYINDEX);
}

template<char *name, typename index_t>
bool L_Enum<name, index_t>::_lookup(lua_State *L, int index, index_t& to)
{
	if (L_Class<name, index_t>::Is(L, index))
	{
		to = L_Class<name, index_t>::Index(L, index);
		return true;
	}
	else if (lua_isnumber(L, index))
	{
		to = static_cast<index_t>(lua_tonumber(L, index));
		return L_Class<name, index_t>::Valid(to);
	}
	else if (lua_isstring(L, index))
	{
		// look for mnemonic
		PushMnemonicTable(L);

		if (lua_istable(L, -1))
		{
			lua_pushvalue(L, index > 0 ? index : index - 1);
			lua_gettable(L, -2);
			if (lua_isnumber(L, -1))
			{
				to = static_cast<index_t>(lua_tonumber(L, -1));
				lua_pop(L, 2);
				return (L_Class<name, index_t>::Valid(to));
			}
			else
			{
				lua_pop(L, 2);
				return false;
			}
		}
		else
		{
			lua_pop(L, 1);
			return false;
		}
	}
	else
	{
		return false;
	}
}

template<char *name, typename index_t>
index_t L_Enum<name, index_t>::ToIndex(lua_State *L, int index)
{
	index_t to;
	if (_lookup(L, index, to))
	{
		return to;
	}
	else
	{
		std::string error;
		if (lua_isnumber(L, index) || lua_isstring(L, index))
		{
			error = std::string(name) + ": invalid index";
		}
		else
		{
			error = std::string(name) + ": incorrect argument type";
		}
		return luaL_error(L, error.c_str());
	}
}

template<char *name, typename index_t>
int L_Enum<name, index_t>::_equals(lua_State *L)
{
	index_t a, b;
	lua_pushboolean(L, _lookup(L, 1, a) && _lookup(L, 2, b) && (a == b));
	return 1;
}

template<char *name, typename index_t>
int L_Enum<name, index_t>::_get_mnemonic(lua_State *L)
{
	PushMnemonicTable(L);
	lua_pushnumber(L, ToIndex(L, 1));
	lua_gettable(L, -2);
	if (lua_isstring(L, -1))
	{
		lua_remove(L, -2);
		return 1;
	}
	else
	{
		lua_pop(L, 2);
		return 0;
	}
}

template<char *name, typename index_t>
int L_Enum<name, index_t>::_set_mnemonic(lua_State *L)
{
	if (!lua_isstring(L, 2))
		return luaL_error(L, "mnemonic: incorrect argument type");

	index_t index = ToIndex(L, 1);
	const char *new_mnemonic = lua_tostring(L, 2);

	PushMnemonicTable(L);
	
	// look up the old mnemonic
	lua_pushnumber(L, index);
	lua_gettable(L, -2);
	if (lua_isstring(L, -1))
	{
		// if it exists, remove the string key
		lua_pushnil(L);
		lua_settable(L, -3);
		
	}
	else
	{
		lua_pop(L, 1);
	}
	
	// update string key
	lua_pushstring(L, new_mnemonic);
	lua_pushnumber(L, index);
	lua_settable(L, -3);

	// update index key
	lua_pushnumber(L, index);
	lua_pushstring(L, new_mnemonic);
	lua_settable(L, -3);

	lua_pop(L, 1);
	return 0;
}

template<char *name, class T>
class L_Container {
public:
	static void Register(lua_State *L, const luaL_Reg methods[] = 0, const luaL_Reg metatable[] = 0);
	static boost::function<typename T::index_type (void)> Length;
	struct ConstantLength
	{
		ConstantLength(int32 length) : m_length(length) {}
		int32 operator() (void) { return m_length; }
		int32 m_length;
	};
private:
	static int _get(lua_State *);
	static int _set(lua_State *);
	static int _call(lua_State *);
	static int _iterator(lua_State *);
	static int _length(lua_State *);

protected:
	static void _push_methods_key(lua_State *L) {
		lua_pushlightuserdata(L, (void *) (&name[1]));
	}
};

template<char *name, class T>
boost::function<typename T::index_type (void)> L_Container<name, T>::Length = ConstantLength(1);

template<char *name, class T>
void L_Container<name, T>::Register(lua_State *L, const luaL_Reg methods[], const luaL_Reg metatable[])
{
	lua_newuserdata(L, 0);

	luaL_newmetatable(L, name);
	
	lua_pushcfunction(L, _get);
	lua_setfield(L, -2, "__index");
	
	lua_pushcfunction(L, _set);
	lua_setfield(L, -2, "__newindex");
	
	lua_pushcfunction(L, _call);
	lua_setfield(L, -2, "__call");
	
	lua_pushcfunction(L, _length);
	lua_setfield(L, -2, "__len");

	if (metatable)
		luaL_setfuncs(L, metatable, 0);
	
	lua_setmetatable(L, -2);

	lua_setglobal(L, name);

	_push_methods_key(L);
	lua_newtable(L);
	if (methods)
		luaL_setfuncs(L, methods, 0);
	lua_settable(L, LUA_REGISTRYINDEX);
}

template<char *name, class T>
int L_Container<name, T>::_get(lua_State *L)
{
	if (lua_isnumber(L, 2))
	{
		int32 index = static_cast<int32>(lua_tonumber(L, 2));
		if (!T::Valid(index))
		{
			lua_pushnil(L);
		}
		else
		{
			T::Push(L, index);
		}
	}
	else
	{
		// get the function from methods
		_push_methods_key(L);
		lua_gettable(L, LUA_REGISTRYINDEX);

		lua_pushvalue(L, 2);
		lua_gettable(L, -2);
	}

	return 1;
}

template<char *name, class T>
int L_Container<name, T>::_set(lua_State *L)
{
	std::string error = std::string(name) + " is read-only";
	return luaL_error(L, error.c_str());
}

template<char *name, class T>
int L_Container<name, T>::_iterator(lua_State *L)
{
	int32 index = static_cast<int32>(lua_tonumber(L, lua_upvalueindex(1)));
	while (index < Length())
	{
		if (T::Valid(index))

		{
			T::Push(L, index);
			lua_pushnumber(L, ++index);
			lua_replace(L, lua_upvalueindex(1));
			return 1;
		}
		else
		{
			++index;
		}
	}

	lua_pushnil(L);
	return 1;
}

template<char *name, class T>
int L_Container<name, T>::_call(lua_State *L)
{
	lua_pushnumber(L, 0);
	lua_pushcclosure(L, _iterator, 1);
	return 1;
}

template<char *name, class T>
int L_Container<name, T>::_length(lua_State *L)
{
	lua_pushnumber(L, Length());
	return 1;
}

// enum containers will be able to look up by strings
template<char *name, class T>
class L_EnumContainer : public L_Container<name, T>
{
public:
	static void Register(lua_State *L, const luaL_Reg methods[] = 0, const luaL_Reg metatable[] = 0);
private:
	static int _get(lua_State *);
};

template <char *name, class T>
void L_EnumContainer<name, T>::Register(lua_State *L, const luaL_Reg methods[], const luaL_Reg metatable[])
{
	L_Container<name, T>::Register(L, methods, metatable);
	
	luaL_getmetatable(L, name);

	lua_pushcfunction(L, _get);
	lua_setfield(L, -2, "__index");

	lua_pop(L, 1);
}

template <char *name, class T>
int L_EnumContainer<name, T>::_get(lua_State *L)
{
	if (lua_isnumber(L, 2))
	{
		int32 index = static_cast<int32>(lua_tonumber(L, 2));
		if (!T::Valid(index))
		{
			lua_pushnil(L);
			return 1;
		}
		else
		{
			T::Push(L, index);
			return 1;
		}
	}
	else if (lua_isstring(L, 2))
	{
		// try mnemonics
		T::PushMnemonicTable(L);

		if (lua_istable(L, -1))
		{
			lua_pushvalue(L, 2);
			lua_gettable(L, -2);
			if (lua_isnumber(L, -1))
			{
				int32 index = static_cast<int32>(lua_tonumber(L, -1));
				lua_pop(L, 2);
				T::Push(L, index);
				return 1;
			}
			else
			{
				// should not happen
				lua_pop(L, 2);
			}
		} 
		else
		{
			lua_pop(L, 1);
		}

	}

	// get the function from methods
	L_Container<name, T>::_push_methods_key(L);
	lua_gettable(L, LUA_REGISTRYINDEX);
	
	lua_pushvalue(L, 2);
	lua_gettable(L, -2);
	
	return 1;
}

// object classes hold an object_t as well as numeric index_t
template<char *name, typename object_t, typename index_t = int16>
class L_ObjectClass : L_Class<name, index_t> {
public:
	static L_ObjectClass<name, object_t, index_t> *Push(lua_State *L, object_t object);
	static object_t ObjectAtIndex(lua_State *L, index_t index);
	static object_t Object(lua_State *L, int index);
	static void Invalidate(lua_State *L, index_t index);

	static void Register(lua_State *L, const luaL_Reg get[] = 0, const luaL_Reg set[] = 0, const luaL_Reg metatable[] = 0) {
		return L_Class<name, index_t>::Register(L, get, set, metatable);
	}
	static index_t Index(lua_State *L, int index) {
		return L_Class<name, index_t>::Index(L, index);
	}
	static bool Is(lua_State *L, int index) {
		return L_Class<name, index_t>::Is(L, index);
	}
	static boost::function<bool (index_t)> Valid;
	
	static map<index_t, object_t> _objects;
};

template<char *name, typename object_t, typename index_t>
map<index_t, object_t> L_ObjectClass<name, object_t, index_t>::_objects;

template<char *name, typename object_t, typename index_t>
struct object_valid
{
	bool operator()(index_t x) {
			return (L_ObjectClass<name, object_t, index_t>::_objects.find(x) !=
						  L_ObjectClass<name, object_t, index_t>::_objects.end());
	}
};

template<char *name, typename object_t, typename index_t>
boost::function<bool (index_t)> L_ObjectClass<name, object_t, index_t>::Valid = object_valid<name, object_t, index_t>();


template<char *name, typename object_t, typename index_t>
L_ObjectClass<name, object_t, index_t> *L_ObjectClass<name, object_t, index_t>::Push(lua_State *L, object_t object)
{
	// find unused index in our map
	index_t idx = 0;
	while (_objects.find(++idx) != _objects.end()) { }
	
	_objects[idx] = object;
	
	// create an instance
	L_ObjectClass<name, object_t, index_t> *t = static_cast<L_ObjectClass<name, object_t, index_t> *>(lua_newuserdata(L, sizeof(L_ObjectClass<name, object_t, index_t>)));
	luaL_getmetatable(L, name);
	lua_setmetatable(L, -2);
	t->m_index = idx;
	
	return t;
}

template<char *name, typename object_t, typename index_t>
object_t L_ObjectClass<name, object_t, index_t>::ObjectAtIndex(lua_State *L, index_t index)
{
	if (_objects.find(index) == _objects.end())
		luaL_typerror(L, index, name);
	
	return _objects[index];
}

template<char *name, typename object_t, typename index_t>
object_t L_ObjectClass<name, object_t, index_t>::Object(lua_State *L, int index)
{
	return ObjectAtIndex(L, L_Class<name, index_t>::Index(L, index));
}

template<char *name, typename object_t, typename index_t>
void L_ObjectClass<name, object_t, index_t>::Invalidate(lua_State *L, index_t index)
{
	// remove index from our object map
	_objects.erase(index);
}


#endif

#endif
