X-Git-Url: https://git.sesse.net/?a=blobdiff_plain;f=nageru%2Ftheme.h;h=8e4537351905ae02354010a0a51b86cc2c4dccc3;hb=f9024d141398e69e7b4011becd3ebbe37eaa1776;hp=7c6b3243c85e5d61a1bd7ef42c77d64e7444c0a0;hpb=337e2d06624b4b46eb2e7e5365e2ece219f9f100;p=nageru

diff --git a/nageru/theme.h b/nageru/theme.h
index 7c6b324..8e45373 100644
--- a/nageru/theme.h
+++ b/nageru/theme.h
@@ -2,6 +2,7 @@
 #define _THEME_H 1
 
 #include <lua.hpp>
+#include <movit/effect.h>
 #include <movit/flat_input.h>
 #include <movit/ycbcr_input.h>
 #include <stdbool.h>
@@ -39,6 +40,7 @@ enum EffectType {
 
 	IDENTITY_EFFECT,
 	WHITE_BALANCE_EFFECT,
+	AUTO_WHITE_BALANCE_EFFECT,  // Same as WHITE_BALANCE_EFFECT, but sets its value automatically.
 	RESAMPLE_EFFECT,
 	PADDING_EFFECT,
 	INTEGRAL_PADDING_EFFECT,
@@ -46,7 +48,9 @@ enum EffectType {
 	RESIZE_EFFECT,
 	MULTIPLY_EFFECT,
 	MIX_EFFECT,
-	LIFT_GAMMA_GAIN_EFFECT
+	LIFT_GAMMA_GAIN_EFFECT,
+
+	NO_EFFECT_TYPE
 };
 
 // An EffectBlueprint refers to an Effect before it's being added to the graph.
@@ -100,9 +104,13 @@ public:
 	std::string get_channel_name(unsigned channel);
 	int get_channel_signal(unsigned channel);
 	bool get_supports_set_wb(unsigned channel);
-	void set_wb(unsigned channel, double r, double g, double b);
+	void set_wb(unsigned channel, float r, float g, float b);
+	void set_wb_for_signal(int signal, float r, float g, float b);
+	movit::RGBTriplet get_white_balance_for_signal(int signal);
 	std::string get_channel_color(unsigned channel);
 
+	std::unordered_map<int, movit::RGBTriplet> white_balance_for_signal;
+
 	std::vector<std::string> get_transition_names(float t);
 
 	void transition_clicked(int transition_num, float t);
@@ -147,10 +155,31 @@ public:
 #endif
 
 	struct MenuEntry {
+		MenuEntry(const std::string &text, lua_State *L, int lua_ref, unsigned flags)
+			: text(text), is_submenu(false), entry{L, lua_ref, flags} {}
+		MenuEntry(const std::string &text, std::vector<std::unique_ptr<MenuEntry>> submenu)
+			: text(text), is_submenu(true), submenu(std::move(submenu)) {}
+		~MenuEntry();
+
+		static constexpr unsigned CHECKABLE = 1;
+		static constexpr unsigned CHECKED = 2;
+
 		std::string text;
-		int lua_ref;
+		bool is_submenu;
+
+		union {
+			// is_submenu = false.
+			struct {
+				lua_State *L;
+				int lua_ref;
+				unsigned flags;
+			} entry;
+
+			// is_submenu = true.
+			std::vector<std::unique_ptr<MenuEntry>> submenu;
+		};
 	};
-	std::vector<MenuEntry> get_theme_menu() { return theme_menu; }  // Can be empty for no menu.
+	MenuEntry *get_theme_menu() { return theme_menu.get(); }  // Can be empty for no menu.
 	void theme_menu_entry_clicked(int lua_ref);
 
 	// Will be invoked every time the theme sets a new menu.
@@ -160,11 +189,14 @@ public:
 		theme_menu_callback = callback;
 	}
 
+	std::string format_status_line(const std::string &disk_space_left_text, double file_length_seconds);
+
 private:
-	void register_constants();
-	void register_class(const char *class_name, const luaL_Reg *funcs);
+	void register_globals();
+	void register_class(const char *class_name, const luaL_Reg *funcs, EffectType effect_type = NO_EFFECT_TYPE);
 	int set_theme_menu(lua_State *L);
 	Chain get_chain_from_effect_chain(movit::EffectChain *effect_chain, unsigned num, const InputState &input_state);
+	void call_lua_wb_callback(unsigned channel, float r, float g, float b);
 
 	std::string theme_path;
 
@@ -172,8 +204,9 @@ private:
 	lua_State *L;  // Protected by <m>.
 	const InputState *input_state = nullptr;  // Protected by <m>. Only set temporarily, during chain setup.
 	movit::ResourcePool *resource_pool;
-	int num_channels;
+	int num_channels = -1;
 	unsigned num_cards;
+	bool startup_finished = false;
 
 	std::mutex map_m;
 	std::map<int, int> signal_to_card_mapping;  // Protected by <map_m>.
@@ -195,12 +228,20 @@ private:
 		html_signal_connections;
 #endif
 
-	std::vector<MenuEntry> theme_menu;
+	std::unique_ptr<MenuEntry> theme_menu;
 	std::function<void()> theme_menu_callback;
 
+	std::map<unsigned, std::string> channel_names;  // Set using Nageru.set_channel_name(). Protected by <m>.
+	std::map<unsigned, int> channel_signals;  // Set using Nageru.set_channel_signal(). Protected by <m>.
+	std::map<unsigned, bool> channel_supports_wb;  // Set using Nageru.set_supports_wb(). Protected by <m>.
+
 	friend class LiveInputWrapper;
 	friend class Scene;
 	friend int ThemeMenu_set(lua_State *L);
+	friend int Nageru_set_channel_name(lua_State *L);
+	friend int Nageru_set_num_channels(lua_State *L);
+	friend int Nageru_set_channel_signal(lua_State *L);
+	friend int Nageru_set_supports_wb(lua_State *L);
 };
 
 // LiveInputWrapper is a facade on top of an YCbCrInput, exposed to