1 files changed, 60 insertions, 4 deletions

diff --git a/Source/3rdParty/Luax/luax_runtime.h b/Source/3rdParty/Luax/luax_runtime.h
index 94d8ecc..2414c2e 100644
--- a/Source/3rdParty/Luax/luax_runtime.h
+++ b/Source/3rdParty/Luax/luax_runtime.h
@@ -1,13 +1,69 @@
 #ifndef __LUAX_RUNTIME_H__
 #define __LUAX_RUNTIME_H__
 
-namespace Luax 
+#include <map>
+
+#include "luax_config.h"
+#include "luax_state.h"
+
+namespace Luax
 {
 
-	enum RegisterIndex
+	///
+	/// ����lua_state��ص�context����һϵ�д����ļ��ϣ�����Ҳû��ϵ����Ҫ��Ϊ�˽�Լ�ڴ档
+	///
+	class Context
 	{
-		LUAX_OBJECT_INDEX = 1, 
-		LUAX_OBJECT_MODULE = 2,
+	public:
+		Context(lua_State* L);
+		~Context();
+
+		LuaxState    state;          // lua state
+		LuaxRefTable strongRefTable; // strong ref table
+		LuaxRefTable weakRefTable;   // weak ref table
+
+		size_t objectCount;          // ͳ�������ڴ�state�д�����ʵ��
+
+	};
+
+	///
+	/// ͳһ�����������е�lua states��
+	///
+	class LuaxRuntime
+	{
+	public:
+
+		static LuaxRuntime& Get();
+
+		///
+		/// ����һ���µ�lua_State���������������ص�lua_State*��һ��8\4�ֽڵ�key��
+		///
+		lua_State* Open();
+
+		///
+		/// �ر�lua_State�������runtime��ɾ����
+		///
+		void Close(lua_State* L);
+
+		bool          HasLuaxState(lua_State* L);
+		LuaxState&    GetLuaxState(lua_State* L);
+		LuaxRefTable& GetStrongRefTable(lua_State* L);
+		LuaxRefTable& GetWeaksRefTable(lua_State* L);
+
+		Context& operator[](lua_State* L);
+
+	private:
+
+		LuaxRuntime();
+		~LuaxRuntime();
+
+		static LuaxRuntime* mRuntime;
+
+		///
+		/// ��lua_state�ҵ�context
+		///
+		std::map<lua_State*, Context> mContexts;
+
 	};
 
 }