]> git.sesse.net Git - nageru/blobdiff - nageru/theme.h
Heed the Exif white point when playing back (MJPEG) video.
[nageru] / nageru / theme.h
index 3d57750c9d9cd9ac39c272a1b66997637f15edbf..8e4537351905ae02354010a0a51b86cc2c4dccc3 100644 (file)
@@ -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>
 #include <vector>
 
 #include "bmusb/bmusb.h"
+#include "defs.h"
 #include "ref_counted_frame.h"
 #include "tweaked_inputs.h"
 
+class Scene;
 class CEFCapture;
 class FFmpegCapture;
 class LiveInputWrapper;
@@ -27,6 +30,58 @@ class EffectChain;
 class ResourcePool;
 }  // namespace movit
 
+enum EffectType {
+       // LIVE_INPUT_* also covers CEF and video inputs.
+       LIVE_INPUT_YCBCR,
+       LIVE_INPUT_YCBCR_WITH_DEINTERLACE,
+       LIVE_INPUT_YCBCR_PLANAR,
+       LIVE_INPUT_BGRA,
+       IMAGE_INPUT,
+
+       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,
+       OVERLAY_EFFECT,
+       RESIZE_EFFECT,
+       MULTIPLY_EFFECT,
+       MIX_EFFECT,
+       LIFT_GAMMA_GAIN_EFFECT,
+
+       NO_EFFECT_TYPE
+};
+
+// An EffectBlueprint refers to an Effect before it's being added to the graph.
+// It contains enough information to instantiate the effect, including any
+// parameters that were set before it was added to the graph. Once it is
+// instantiated, it forwards its calls on to the real Effect instead.
+struct EffectBlueprint {
+       EffectBlueprint(EffectType effect_type) : effect_type(effect_type) {}
+
+       EffectType effect_type;
+       std::map<std::string, int> int_parameters;
+       std::map<std::string, float> float_parameters;
+       std::map<std::string, std::array<float, 3>> vec3_parameters;
+       std::map<std::string, std::array<float, 4>> vec4_parameters;
+
+       movit::Effect *effect = nullptr;  // Gets filled out when it's instantiated.
+};
+
+// Contains basically the same data as InputState, but does not hold on to
+// a reference to the frames. This is important so that we can release them
+// without having to wait for Lua's GC.
+struct InputStateInfo {
+       explicit InputStateInfo(const InputState& input_state);
+
+       unsigned last_width[MAX_VIDEO_CARDS], last_height[MAX_VIDEO_CARDS];
+       bool last_interlaced[MAX_VIDEO_CARDS], last_has_signal[MAX_VIDEO_CARDS], last_is_connected[MAX_VIDEO_CARDS];
+       unsigned last_frame_rate_nom[MAX_VIDEO_CARDS], last_frame_rate_den[MAX_VIDEO_CARDS];
+       bool has_last_subtitle[MAX_VIDEO_CARDS];
+       std::string last_subtitle[MAX_VIDEO_CARDS];
+};
+
 class Theme {
 public:
        Theme(const std::string &filename, const std::vector<std::string> &search_dirs, movit::ResourcePool *resource_pool, unsigned num_cards);
@@ -49,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);
@@ -96,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.
@@ -109,10 +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;
 
@@ -120,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>.
@@ -143,11 +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
@@ -184,4 +278,12 @@ private:
        bool user_connectable;
 };
 
+// Utility functions used by Scene.
+void add_outputs_and_finalize(movit::EffectChain *chain, bool is_main_chain);
+Theme *get_theme_updata(lua_State* L);
+bool checkbool(lua_State* L, int idx);
+std::string checkstdstring(lua_State *L, int index);
+movit::Effect *instantiate_effect(movit::EffectChain *chain, EffectType effect_type);
+void print_warning(lua_State* L, const char *format, ...);
+
 #endif  // !defined(_THEME_H)