5 #include <movit/flat_input.h>
6 #include <movit/ycbcr_input.h>
12 #include <unordered_map>
15 #include "bmusb/bmusb.h"
17 #include "ref_counted_frame.h"
18 #include "tweaked_inputs.h"
23 class LiveInputWrapper;
33 // LIVE_INPUT_* also covers CEF and video inputs.
35 LIVE_INPUT_YCBCR_WITH_DEINTERLACE,
36 LIVE_INPUT_YCBCR_PLANAR,
44 INTEGRAL_PADDING_EFFECT,
49 LIFT_GAMMA_GAIN_EFFECT,
54 // An EffectBlueprint refers to an Effect before it's being added to the graph.
55 // It contains enough information to instantiate the effect, including any
56 // parameters that were set before it was added to the graph. Once it is
57 // instantiated, it forwards its calls on to the real Effect instead.
58 struct EffectBlueprint {
59 EffectBlueprint(EffectType effect_type) : effect_type(effect_type) {}
61 EffectType effect_type;
62 std::map<std::string, int> int_parameters;
63 std::map<std::string, float> float_parameters;
64 std::map<std::string, std::array<float, 3>> vec3_parameters;
65 std::map<std::string, std::array<float, 4>> vec4_parameters;
67 movit::Effect *effect = nullptr; // Gets filled out when it's instantiated.
70 // Contains basically the same data as InputState, but does not hold on to
71 // a reference to the frames. This is important so that we can release them
72 // without having to wait for Lua's GC.
73 struct InputStateInfo {
74 explicit InputStateInfo(const InputState& input_state);
76 unsigned last_width[MAX_VIDEO_CARDS], last_height[MAX_VIDEO_CARDS];
77 bool last_interlaced[MAX_VIDEO_CARDS], last_has_signal[MAX_VIDEO_CARDS], last_is_connected[MAX_VIDEO_CARDS];
78 unsigned last_frame_rate_nom[MAX_VIDEO_CARDS], last_frame_rate_den[MAX_VIDEO_CARDS];
79 bool has_last_subtitle[MAX_VIDEO_CARDS];
80 std::string last_subtitle[MAX_VIDEO_CARDS];
85 Theme(const std::string &filename, const std::vector<std::string> &search_dirs, movit::ResourcePool *resource_pool, unsigned num_cards);
89 movit::EffectChain *chain;
90 std::function<void()> setup_chain;
92 // FRAME_HISTORY frames for each input, in order. Will contain duplicates
93 // for non-interlaced inputs.
94 std::vector<RefCountedFrame> input_frames;
97 Chain get_chain(unsigned num, float t, unsigned width, unsigned height, const InputState &input_state);
99 int get_num_channels() const { return num_channels; }
100 int map_signal(int signal_num);
101 void set_signal_mapping(int signal_num, int card_num);
102 std::string get_channel_name(unsigned channel);
103 int get_channel_signal(unsigned channel);
104 bool get_supports_set_wb(unsigned channel);
105 void set_wb(unsigned channel, double r, double g, double b);
106 std::string get_channel_color(unsigned channel);
108 std::vector<std::string> get_transition_names(float t);
110 void transition_clicked(int transition_num, float t);
111 void channel_clicked(int preview_num);
113 movit::ResourcePool *get_resource_pool() const { return resource_pool; }
115 // Should be called as part of VideoInput.new() only.
116 void register_video_input(FFmpegCapture *capture)
118 video_inputs.push_back(capture);
121 std::vector<FFmpegCapture *> get_video_inputs() const
127 // Should be called as part of HTMLInput.new() only.
128 void register_html_input(CEFCapture *capture)
130 html_inputs.push_back(capture);
133 std::vector<CEFCapture *> get_html_inputs() const
139 void register_video_signal_connection(movit::EffectChain *chain, LiveInputWrapper *live_input, FFmpegCapture *capture)
141 video_signal_connections[chain].emplace_back(VideoSignalConnection { live_input, capture });
145 void register_html_signal_connection(movit::EffectChain *chain, LiveInputWrapper *live_input, CEFCapture *capture)
147 html_signal_connections[chain].emplace_back(CEFSignalConnection { live_input, capture });
155 std::vector<MenuEntry> get_theme_menu() { return theme_menu; } // Can be empty for no menu.
156 void theme_menu_entry_clicked(int lua_ref);
158 // Will be invoked every time the theme sets a new menu.
159 // Is not invoked for a menu that exists at the time of the callback.
160 void set_theme_menu_callback(std::function<void()> callback)
162 theme_menu_callback = callback;
166 void register_constants();
167 void register_class(const char *class_name, const luaL_Reg *funcs, EffectType effect_type = NO_EFFECT_TYPE);
168 int set_theme_menu(lua_State *L);
169 Chain get_chain_from_effect_chain(movit::EffectChain *effect_chain, unsigned num, const InputState &input_state);
171 std::string theme_path;
174 lua_State *L; // Protected by <m>.
175 const InputState *input_state = nullptr; // Protected by <m>. Only set temporarily, during chain setup.
176 movit::ResourcePool *resource_pool;
181 std::map<int, int> signal_to_card_mapping; // Protected by <map_m>.
183 std::vector<FFmpegCapture *> video_inputs;
184 struct VideoSignalConnection {
185 LiveInputWrapper *wrapper;
186 FFmpegCapture *source;
188 std::unordered_map<movit::EffectChain *, std::vector<VideoSignalConnection>>
189 video_signal_connections;
191 std::vector<CEFCapture *> html_inputs;
192 struct CEFSignalConnection {
193 LiveInputWrapper *wrapper;
196 std::unordered_map<movit::EffectChain *, std::vector<CEFSignalConnection>>
197 html_signal_connections;
200 std::vector<MenuEntry> theme_menu;
201 std::function<void()> theme_menu_callback;
203 friend class LiveInputWrapper;
205 friend int ThemeMenu_set(lua_State *L);
208 // LiveInputWrapper is a facade on top of an YCbCrInput, exposed to
209 // the Lua code. It contains a function (connect_signal()) intended
210 // to be called during chain setup, that picks out the current frame
211 // (in the form of a set of textures) from the input state given by
212 // the mixer, and communicates that state over to the actual YCbCrInput.
213 class LiveInputWrapper {
215 // Note: <override_bounce> is irrelevant for PixelFormat_8BitBGRA.
216 LiveInputWrapper(Theme *theme, movit::EffectChain *chain, bmusb::PixelFormat pixel_format, bool override_bounce, bool deinterlace, bool user_connectable);
218 bool connect_signal(int signal_num); // Must be called with the theme's <m> lock held, since it accesses theme->input_state. Returns false on error.
219 void connect_signal_raw(int signal_num, const InputState &input_state);
220 movit::Effect *get_effect() const
223 return deinterlace_effect;
224 } else if (pixel_format == bmusb::PixelFormat_8BitBGRA) {
225 return rgba_inputs[0];
227 return ycbcr_inputs[0];
232 Theme *theme; // Not owned by us.
233 bmusb::PixelFormat pixel_format;
234 movit::YCbCrFormat input_ycbcr_format;
235 std::vector<movit::YCbCrInput *> ycbcr_inputs; // Multiple ones if deinterlacing. Owned by the chain.
236 std::vector<movit::FlatInput *> rgba_inputs; // Multiple ones if deinterlacing. Owned by the chain.
237 movit::Effect *deinterlace_effect = nullptr; // Owned by the chain.
239 bool user_connectable;
242 // Utility functions used by Scene.
243 void add_outputs_and_finalize(movit::EffectChain *chain, bool is_main_chain);
244 Theme *get_theme_updata(lua_State* L);
245 bool checkbool(lua_State* L, int idx);
246 std::string checkstdstring(lua_State *L, int index);
247 movit::Effect *instantiate_effect(movit::EffectChain *chain, EffectType effect_type);
248 void print_warning(lua_State* L, const char *format, ...);
250 #endif // !defined(_THEME_H)