X-Git-Url: https://git.sesse.net/?a=blobdiff_plain;f=theme.cpp;h=7240cd4c0f2ce3b7d4c2cb571f0b0c2891575a1b;hb=f19abc8c6ab711a8de92f3c23d241c4c372a4469;hp=4ddea3a25915713191719c0da8c412d49933ac02;hpb=8ec5630a8e7ec089d2e39697d51642278d1977d7;p=nageru

diff --git a/theme.cpp b/theme.cpp
index 4ddea3a..7240cd4 100644
--- a/theme.cpp
+++ b/theme.cpp
@@ -28,13 +28,76 @@
 #include <utility>
 
 #include "defs.h"
+#ifdef HAVE_CEF
 #include "cef_capture.h"
+#endif
 #include "ffmpeg_capture.h"
 #include "flags.h"
 #include "image_input.h"
 #include "input_state.h"
 #include "pbo_frame_allocator.h"
 
+#if !defined LUA_VERSION_NUM || LUA_VERSION_NUM==501
+
+// Compatibility shims for LuaJIT 2.0 (LuaJIT 2.1 implements the entire Lua 5.2 API).
+// Adapted from https://github.com/keplerproject/lua-compat-5.2/blob/master/c-api/compat-5.2.c
+// and licensed as follows:
+//
+// The MIT License (MIT)
+//
+// Copyright (c) 2013 Hisham Muhammad
+//
+// Permission is hereby granted, free of charge, to any person obtaining a copy of
+// this software and associated documentation files (the "Software"), to deal in
+// the Software without restriction, including without limitation the rights to
+// use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies of
+// the Software, and to permit persons to whom the Software is furnished to do so,
+// subject to the following conditions:
+//
+// The above copyright notice and this permission notice shall be included in all
+// copies or substantial portions of the Software.
+//
+// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS
+// FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR
+// COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER
+// IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN
+// CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
+
+/*
+** Adapted from Lua 5.2.0
+*/
+void luaL_setfuncs(lua_State *L, const luaL_Reg *l, int nup) {
+	luaL_checkstack(L, nup+1, "too many upvalues");
+	for (; l->name != NULL; l++) {  /* fill the table with given functions */
+		int i;
+		lua_pushstring(L, l->name);
+		for (i = 0; i < nup; i++)  /* copy upvalues to the top */
+			lua_pushvalue(L, -(nup + 1));
+		lua_pushcclosure(L, l->func, nup);  /* closure with those upvalues */
+		lua_settable(L, -(nup + 3)); /* table must be below the upvalues, the name and the closure */
+	}
+	lua_pop(L, nup);  /* remove upvalues */
+}
+
+void *luaL_testudata(lua_State *L, int i, const char *tname) {
+	void *p = lua_touserdata(L, i);
+	luaL_checkstack(L, 2, "not enough stack slots");
+	if (p == NULL || !lua_getmetatable(L, i))
+		return NULL;
+	else {
+		int res = 0;
+		luaL_getmetatable(L, tname);
+		res = lua_rawequal(L, -1, -2);
+		lua_pop(L, 2);
+		if (!res)
+			p = NULL;
+	}
+	return p;
+}
+
+#endif
+
 class Mixer;
 
 namespace movit {
@@ -46,6 +109,18 @@ using namespace movit;
 
 extern Mixer *global_mixer;
 
+Theme *get_theme_updata(lua_State* L)
+{
+	luaL_checktype(L, lua_upvalueindex(1), LUA_TLIGHTUSERDATA);
+	return (Theme *)lua_touserdata(L, lua_upvalueindex(1));
+}
+
+int ThemeMenu_set(lua_State *L)
+{
+	Theme *theme = get_theme_updata(L);
+	return theme->set_theme_menu(L);
+}
+
 namespace {
 
 // Contains basically the same data as InputState, but does not hold on to
@@ -134,12 +209,6 @@ int wrap_lua_object_nonowned(lua_State* L, const char *class_name, Args&&... arg
 	return 1;
 }
 
-Theme *get_theme_updata(lua_State* L)
-{	
-	luaL_checktype(L, lua_upvalueindex(1), LUA_TLIGHTUSERDATA);
-	return (Theme *)lua_touserdata(L, lua_upvalueindex(1));
-}
-
 Effect *get_effect(lua_State *L, int idx)
 {
 	if (luaL_testudata(L, idx, "WhiteBalanceEffect") ||
@@ -233,6 +302,7 @@ int EffectChain_add_video_input(lua_State* L)
 	return ret;
 }
 
+#ifdef HAVE_CEF
 int EffectChain_add_html_input(lua_State* L)
 {
 	assert(lua_gettop(L) == 2);
@@ -254,6 +324,7 @@ int EffectChain_add_html_input(lua_State* L)
 	}
 	return ret;
 }
+#endif
 
 int EffectChain_add_effect(lua_State* L)
 {
@@ -411,6 +482,7 @@ int VideoInput_get_signal_num(lua_State* L)
 
 int HTMLInput_new(lua_State* L)
 {
+#ifdef HAVE_CEF
 	assert(lua_gettop(L) == 1);
 	string url = checkstdstring(L, 1);
 	int ret = wrap_lua_object_nonowned<CEFCapture>(L, "HTMLInput", url, global_flags.width, global_flags.height);
@@ -420,8 +492,14 @@ int HTMLInput_new(lua_State* L)
 		theme->register_html_input(*capture);
 	}
 	return ret;
+#else
+	fprintf(stderr, "This version of Nageru has been compiled without CEF support.\n");
+	fprintf(stderr, "HTMLInput is not available.\n");
+	exit(1);
+#endif
 }
 
+#ifdef HAVE_CEF
 int HTMLInput_set_url(lua_State* L)
 {
 	assert(lua_gettop(L) == 2);
@@ -439,6 +517,24 @@ int HTMLInput_reload(lua_State* L)
 	return 0;
 }
 
+int HTMLInput_set_max_fps(lua_State* L)
+{
+	assert(lua_gettop(L) == 2);
+	CEFCapture **video_input = (CEFCapture **)luaL_checkudata(L, 1, "HTMLInput");
+	int max_fps = lrint(luaL_checknumber(L, 2));
+	(*video_input)->set_max_fps(max_fps);
+	return 0;
+}
+
+int HTMLInput_execute_javascript_async(lua_State* L)
+{
+	assert(lua_gettop(L) == 2);
+	CEFCapture **video_input = (CEFCapture **)luaL_checkudata(L, 1, "HTMLInput");
+	string js = checkstdstring(L, 2);
+	(*video_input)->execute_javascript_async(js);
+	return 0;
+}
+
 int HTMLInput_get_signal_num(lua_State* L)
 {
 	assert(lua_gettop(L) == 1);
@@ -446,6 +542,7 @@ int HTMLInput_get_signal_num(lua_State* L)
 	lua_pushnumber(L, -1 - (*video_input)->get_card_index());
 	return 1;
 }
+#endif
 
 int WhiteBalanceEffect_new(lua_State* L)
 {
@@ -627,7 +724,9 @@ const luaL_Reg EffectChain_funcs[] = {
 	{ "__gc", EffectChain_gc },
 	{ "add_live_input", EffectChain_add_live_input },
 	{ "add_video_input", EffectChain_add_video_input },
+#ifdef HAVE_CEF
 	{ "add_html_input", EffectChain_add_html_input },
+#endif
 	{ "add_effect", EffectChain_add_effect },
 	{ "finalize", EffectChain_finalize },
 	{ NULL, NULL }
@@ -656,11 +755,14 @@ const luaL_Reg VideoInput_funcs[] = {
 };
 
 const luaL_Reg HTMLInput_funcs[] = {
-	// TODO: execute_javascript, perhaps set_max_fps?
 	{ "new", HTMLInput_new },
+#ifdef HAVE_CEF
 	{ "set_url", HTMLInput_set_url },
 	{ "reload", HTMLInput_reload },
+	{ "set_max_fps", HTMLInput_set_max_fps },
+	{ "execute_javascript_async", HTMLInput_execute_javascript_async },
 	{ "get_signal_num", HTMLInput_get_signal_num },
+#endif
 	{ NULL, NULL }
 };
 
@@ -747,6 +849,11 @@ const luaL_Reg InputStateInfo_funcs[] = {
 	{ NULL, NULL }
 };
 
+const luaL_Reg ThemeMenu_funcs[] = {
+	{ "set", ThemeMenu_set },
+	{ NULL, NULL }
+};
+
 }  // namespace
 
 LiveInputWrapper::LiveInputWrapper(Theme *theme, EffectChain *chain, bmusb::PixelFormat pixel_format, bool override_bounce, bool deinterlace)
@@ -987,6 +1094,7 @@ Theme::Theme(const string &filename, const vector<string> &search_dirs, Resource
 	register_class("MultiplyEffect", MultiplyEffect_funcs);
 	register_class("MixEffect", MixEffect_funcs);
 	register_class("InputStateInfo", InputStateInfo_funcs);
+	register_class("ThemeMenu", ThemeMenu_funcs);
 
 	// Run script. Search through all directories until we find a file that will load
 	// (as in, does not return LUA_ERRFILE); then run it. We store load errors
@@ -994,8 +1102,21 @@ Theme::Theme(const string &filename, const vector<string> &search_dirs, Resource
 	lua_settop(L, 0);
 	vector<string> errors;
 	bool success = false;
-	for (size_t i = 0; i < search_dirs.size(); ++i) {
-		string path = search_dirs[i] + "/" + filename;
+
+	vector<string> real_search_dirs;
+	if (!filename.empty() && filename[0] == '/') {
+		real_search_dirs.push_back("");
+	} else {
+		real_search_dirs = search_dirs;
+	}
+
+	for (const string &dir : real_search_dirs) {
+		string path;
+		if (dir.empty()) {
+			path = filename;
+		} else {
+			path = dir + "/" + filename;
+		}
 		int err = luaL_loadfile(L, path.c_str());
 		if (err == 0) {
 			// Success; actually call the code.
@@ -1305,3 +1426,41 @@ void Theme::channel_clicked(int preview_num)
 	}
 	assert(lua_gettop(L) == 0);
 }
+
+int Theme::set_theme_menu(lua_State *L)
+{
+	for (const Theme::MenuEntry &entry : theme_menu) {
+		luaL_unref(L, LUA_REGISTRYINDEX, entry.lua_ref);
+	}
+	theme_menu.clear();
+
+	int num_elements = lua_gettop(L);
+	for (int i = 1; i <= num_elements; ++i) {
+		lua_rawgeti(L, i, 1);
+		const string text = checkstdstring(L, -1);
+		lua_pop(L, 1);
+
+		lua_rawgeti(L, i, 2);
+		luaL_checktype(L, -1, LUA_TFUNCTION);
+		int ref = luaL_ref(L, LUA_REGISTRYINDEX);
+
+		theme_menu.push_back(MenuEntry{ text, ref });
+	}
+	lua_pop(L, num_elements);
+	assert(lua_gettop(L) == 0);
+
+	if (theme_menu_callback != nullptr) {
+		theme_menu_callback();
+	}
+
+	return 0;
+}
+
+void Theme::theme_menu_entry_clicked(int lua_ref)
+{
+	lua_rawgeti(L, LUA_REGISTRYINDEX, lua_ref);
+	if (lua_pcall(L, 0, 0, 0) != 0) {
+		fprintf(stderr, "error running menu callback: %s\n", lua_tostring(L, -1));
+		exit(1);
+	}
+}