Heed the Exif white point when playing back (MJPEG) video.

[nageru] / nageru / theme.h

diff --git a/nageru/theme.h b/nageru/theme.h
index d955e643458773eefa3527176d088072744264fa..8e4537351905ae02354010a0a51b86cc2c4dccc3 100644 (file)
--- 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,
@@ -102,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);
@@ -149,12 +155,15 @@ public:
 #endif
 
        struct MenuEntry {
-               MenuEntry(const std::string &text, lua_State *L, int lua_ref)
-                       : text(text), is_submenu(false), entry{L, lua_ref} {}
+               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;
                bool is_submenu;
 
@@ -163,6 +172,7 @@ public:
                        struct {
                                lua_State *L;
                                int lua_ref;
+                               unsigned flags;
                        } entry;
 
                        // is_submenu = true.
@@ -179,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_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;
 
@@ -219,12 +232,16 @@ private:
        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