| 1 | // Aseprite | 
| 2 | // Copyright (C) 2018-2019  Igara Studio S.A. | 
| 3 | // Copyright (C) 2018  David Capello | 
| 4 | // | 
| 5 | // This program is distributed under the terms of | 
| 6 | // the End-User License Agreement for Aseprite. | 
| 7 |  | 
| 8 | #ifndef APP_SCRIPT_LUACPP_H_INCLUDED | 
| 9 | #define APP_SCRIPT_LUACPP_H_INCLUDED | 
| 10 | #pragma once | 
| 11 |  | 
| 12 | extern "C"  { | 
| 13 |   #include "lua.h" | 
| 14 |   #include "lualib.h" | 
| 15 |   #include "lauxlib.h" | 
| 16 | } | 
| 17 |  | 
| 18 | #include "base/debug.h" | 
| 19 |  | 
| 20 | #include <functional> | 
| 21 | #include <type_traits> | 
| 22 |  | 
| 23 | namespace app { | 
| 24 | namespace script { | 
| 25 |  | 
| 26 | #if LUA_TNONE != -1 | 
| 27 |   #error Invalid LUA_TNONE value | 
| 28 | #endif | 
| 29 | #if LUA_TNIL != 0 | 
| 30 |   #error Invalid LUA_TNIL value | 
| 31 | #endif | 
| 32 | #define VALID_LUATYPE(type) ((type) > 0) | 
| 33 |  | 
| 34 | // Some of these auxiliary methods are based on code from the Skia | 
| 35 | // library (SkLua.cpp file) by Google Inc. | 
| 36 |  | 
| 37 | template <typename T> const char* get_mtname(); | 
| 38 | #define DEF_MTNAME(T)                         \ | 
| 39 |   template <> const char* get_mtname<T>() {   \ | 
| 40 |     return #T;                                \ | 
| 41 |   } | 
| 42 |  | 
| 43 | #define DEF_MTNAME_ALIAS(T, ALIAS)              \ | 
| 44 |   template <> const char* get_mtname<ALIAS>() { \ | 
| 45 |     return #T;                                  \ | 
| 46 |   } | 
| 47 |  | 
| 48 | template <typename T, typename... Args> T* push_new(lua_State* L, Args&&... args) { | 
| 49 |   T* addr = (T*)lua_newuserdata(L, sizeof(T)); | 
| 50 |   new (addr) T(std::forward<Args>(args)...); | 
| 51 |   luaL_getmetatable(L, get_mtname<T>()); | 
| 52 |   lua_setmetatable(L, -2); | 
| 53 |   return addr; | 
| 54 | } | 
| 55 |  | 
| 56 | template <typename T> void push_obj(lua_State* L, const T& obj) { | 
| 57 |   new (lua_newuserdata(L, sizeof(T))) T(obj); | 
| 58 |   luaL_getmetatable(L, get_mtname<T>()); | 
| 59 |   lua_setmetatable(L, -2); | 
| 60 | } | 
| 61 |  | 
| 62 | template <typename T> T* push_ptr(lua_State* L, T* ptr) { | 
| 63 |   *(T**)lua_newuserdata(L, sizeof(T*)) = ptr; | 
| 64 |   luaL_getmetatable(L, get_mtname<T>()); | 
| 65 |   lua_setmetatable(L, -2); | 
| 66 |   return ptr; | 
| 67 | } | 
| 68 |  | 
| 69 | template <typename T> T* get_ptr(lua_State* L, int index) { | 
| 70 |   return *(T**)luaL_checkudata(L, index, get_mtname<T>()); | 
| 71 | } | 
| 72 |  | 
| 73 | template <typename T> T* get_obj(lua_State* L, int index) { | 
| 74 |   T* ptr = (T*)luaL_checkudata(L, index, get_mtname<T>()); | 
| 75 |   ASSERT(typeid(*ptr) == typeid(T)); | 
| 76 |   return ptr; | 
| 77 | } | 
| 78 |  | 
| 79 | // Returns nil if the index doesn't have the given metatable | 
| 80 | template <typename T> T* may_get_ptr(lua_State* L, int index) { | 
| 81 |   T** ptr = (T**)luaL_testudata(L, index, get_mtname<T>()); | 
| 82 |   if (ptr) { | 
| 83 |     ASSERT(typeid(**ptr) == typeid(T)); | 
| 84 |     return *ptr; | 
| 85 |   } | 
| 86 |   else | 
| 87 |     return nullptr; | 
| 88 | } | 
| 89 |  | 
| 90 | // Returns nil if the index doesn't have the given metatable | 
| 91 | template <typename T> T* may_get_obj(lua_State* L, int index) { | 
| 92 |   return (T*)luaL_testudata(L, index, get_mtname<T>()); | 
| 93 | } | 
| 94 |  | 
| 95 | inline bool lua2bool(lua_State* L, int index) { | 
| 96 |   return !!lua_toboolean(L, index); | 
| 97 | } | 
| 98 |  | 
| 99 | template<typename T> | 
| 100 | inline void setfield_integer(lua_State* L, const char* key, const T& value) { | 
| 101 |   lua_pushinteger(L, int(value)); | 
| 102 |   lua_setfield(L, -2, key); | 
| 103 | } | 
| 104 |  | 
| 105 | #define REG_CLASS(L, T) {                        \ | 
| 106 |     luaL_newmetatable(L, get_mtname<T>());       \ | 
| 107 |     lua_getglobal(L, "__generic_mt_index");      \ | 
| 108 |     lua_setfield(L, -2, "__index");              \ | 
| 109 |     lua_getglobal(L, "__generic_mt_newindex");   \ | 
| 110 |     lua_setfield(L, -2, "__newindex");           \ | 
| 111 |     luaL_setfuncs(L, T##_methods, 0);            \ | 
| 112 |     lua_pop(L, 1);                               \ | 
| 113 |   } | 
| 114 |  | 
| 115 | #define REG_CLASS_NEW(L, T) {                    \ | 
| 116 |     lua_pushcfunction(L, T##_new);               \ | 
| 117 |     lua_setglobal(L, #T);                        \ | 
| 118 |   } | 
| 119 |  | 
| 120 | struct Property { | 
| 121 |   const char* name; | 
| 122 |   lua_CFunction getter; | 
| 123 |   lua_CFunction setter; | 
| 124 | }; | 
| 125 |  | 
| 126 | void run_mt_index_code(lua_State* L); | 
| 127 | void create_mt_getters_setters(lua_State* L, | 
| 128 |                                const char* tname, | 
| 129 |                                const Property* properties); | 
| 130 |  | 
| 131 | bool lua_is_key_true(lua_State* L, int tableIndex, const char* keyName); | 
| 132 |  | 
| 133 | #define REG_CLASS_PROPERTIES(L, T) {                                \ | 
| 134 |     luaL_getmetatable(L, get_mtname<T>());                          \ | 
| 135 |     create_mt_getters_setters(L, get_mtname<T>(), T##_properties);  \ | 
| 136 |     lua_pop(L, 1);                                                  \ | 
| 137 |   } | 
| 138 |  | 
| 139 | } // namespace script | 
| 140 | } // namespace app | 
| 141 |  | 
| 142 | #endif | 
| 143 |  |