add

brinkqiang · Dec 1, 2018 · a80c6ea · a80c6ea
1 parent 1e9b137
commit a80c6ea
Showing 1 changed file with 28 additions and 0 deletions.
diff --git a/include/dmlua_opt.h b/include/dmlua_opt.h
@@ -19,6 +19,34 @@ struct LuaReader
     }
 };
 
+template <typename T>
+struct LuaReader<T*>
+{
+    static inline T* Read(lua_State* L, int index)
+    {
+        if (CluaTypeid::Instance().get_name<T>()) {
+            return (T*)tolua_tousertype(L, index, CluaTypeid::Instance().get_name<T>());
+        }
+        else {
+            return (T*)tolua_touserdata(L, index, NULL);
+        }
+    }
+};
+
+template <typename T>
+struct LuaReader<T&>
+{
+    static inline T& Read(lua_State* L, int index)
+    {
+        if (CluaTypeid::Instance().get_name<T>()) {
+            return *(T*)tolua_tousertype(L, index, CluaTypeid::Instance().get_name<T>());
+        }
+        else {
+            return *(T*)tolua_touserdata(L, index, NULL);
+        }
+    }
+};
+
 template <>
 struct LuaReader<void>
 {