X-Git-Url: https://git.sesse.net/?a=blobdiff_plain;f=nageru%2Ftheme.h;h=2be6689f794b1f57cb47aa986b4789621f088032;hb=0490c42536a845dd6d718a7c58cfb3e3ab02b1ab;hp=37113b2b954d88a607087ca8e691df1aa082a8fd;hpb=948d715655a84b93d8292e64731ea3c32b45deb7;p=nageru

diff --git a/nageru/theme.h b/nageru/theme.h
index 37113b2..2be6689 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,
@@ -97,14 +99,18 @@ public:
 	Chain get_chain(unsigned num, float t, unsigned width, unsigned height, const InputState &input_state);
 
 	int get_num_channels() const { return num_channels; }
-	int map_signal(int signal_num);
-	void set_signal_mapping(int signal_num, int card_num);
+	int map_signal_to_card(int signal_num);
+	void set_signal_mapping(int signal_num, int card_idx);
 	std::string get_channel_name(unsigned channel);
-	int get_channel_signal(unsigned channel);
+	int map_channel_to_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);
@@ -190,6 +196,7 @@ private:
 	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;