Add the GPU memory metrics to the Grafana dashboard.

[nageru] / theme.h

diff --git a/theme.h b/theme.h
index 71e44bd42cac49b4b9691f358c8f85f545f7addf..0a239953baebf515f99c9b831e75ccdf6fc1b822 100644 (file)
--- a/theme.h
+++ b/theme.h
@@ -9,12 +9,14 @@
 #include <map>
 #include <mutex>
 #include <string>
+#include <unordered_map>
 #include <vector>
 
 #include "bmusb/bmusb.h"
 #include "ref_counted_frame.h"
 #include "tweaked_inputs.h"
 
+class CEFCapture;
 class FFmpegCapture;
 class LiveInputWrapper;
 struct InputState;
@@ -34,7 +36,8 @@ public:
                movit::EffectChain *chain;
                std::function<void()> setup_chain;
 
-               // May have duplicates.
+               // FRAME_HISTORY frames for each input, in order. Will contain duplicates
+               // for non-interlaced inputs.
                std::vector<RefCountedFrame> input_frames;
        };
 
@@ -67,23 +70,55 @@ public:
                return video_inputs;
        }
 
-       void register_signal_connection(LiveInputWrapper *live_input, FFmpegCapture *capture)
+#ifdef HAVE_CEF
+       // Should be called as part of HTMLInput.new() only.
+       void register_html_input(CEFCapture *capture)
        {
-               signal_connections.emplace_back(live_input, capture);
+               html_inputs.push_back(capture);
        }
 
-       std::vector<std::pair<LiveInputWrapper *, FFmpegCapture *>> get_signal_connections() const
+       std::vector<CEFCapture *> get_html_inputs() const
        {
-               return signal_connections;
+               return html_inputs;
+       }
+#endif
+
+       void register_video_signal_connection(movit::EffectChain *chain, LiveInputWrapper *live_input, FFmpegCapture *capture)
+       {
+               video_signal_connections[chain].emplace_back(VideoSignalConnection { live_input, capture });
+       }
+
+#ifdef HAVE_CEF
+       void register_html_signal_connection(movit::EffectChain *chain, LiveInputWrapper *live_input, CEFCapture *capture)
+       {
+               html_signal_connections[chain].emplace_back(CEFSignalConnection { live_input, capture });
+       }
+#endif
+
+       struct MenuEntry {
+               std::string text;
+               int lua_ref;
+       };
+       std::vector<MenuEntry> get_theme_menu() { return theme_menu; }  // 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.
+       // Is not invoked for a menu that exists at the time of the callback.
+       void set_theme_menu_callback(std::function<void()> callback)
+       {
+               theme_menu_callback = callback;
        }
 
 private:
        void register_constants();
        void register_class(const char *class_name, const luaL_Reg *funcs);
+       int set_theme_menu(lua_State *L);
+
+       std::string theme_path;
 
        std::mutex m;
        lua_State *L;  // Protected by <m>.
-       const InputState *input_state;  // Protected by <m>. Only set temporarily, during chain setup.
+       const InputState *input_state = nullptr;  // Protected by <m>. Only set temporarily, during chain setup.
        movit::ResourcePool *resource_pool;
        int num_channels;
        unsigned num_cards;
@@ -92,9 +127,27 @@ private:
        std::map<int, int> signal_to_card_mapping;  // Protected by <map_m>.
 
        std::vector<FFmpegCapture *> video_inputs;
-       std::vector<std::pair<LiveInputWrapper *, FFmpegCapture *>> signal_connections;
+       struct VideoSignalConnection {
+               LiveInputWrapper *wrapper;
+               FFmpegCapture *source;
+       };
+       std::unordered_map<movit::EffectChain *, std::vector<VideoSignalConnection>>
+                video_signal_connections;
+#ifdef HAVE_CEF
+       std::vector<CEFCapture *> html_inputs;
+       struct CEFSignalConnection {
+               LiveInputWrapper *wrapper;
+               CEFCapture *source;
+       };
+       std::unordered_map<movit::EffectChain *, std::vector<CEFSignalConnection>>
+               html_signal_connections;
+#endif
+
+       std::vector<MenuEntry> theme_menu;
+       std::function<void()> theme_menu_callback;
 
        friend class LiveInputWrapper;
+       friend int ThemeMenu_set(lua_State *L);
 };
 
 // LiveInputWrapper is a facade on top of an YCbCrInput, exposed to
@@ -105,10 +158,10 @@ private:
 class LiveInputWrapper {
 public:
        // Note: <override_bounce> is irrelevant for PixelFormat_8BitBGRA.
-       LiveInputWrapper(Theme *theme, movit::EffectChain *chain, bmusb::PixelFormat pixel_format, bool override_bounce, bool deinterlace);
+       LiveInputWrapper(Theme *theme, movit::EffectChain *chain, bmusb::PixelFormat pixel_format, bool override_bounce, bool deinterlace, bool user_connectable);
 
-       void connect_signal(int signal_num);
-       void connect_signal_raw(int signal_num);
+       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.
+       void connect_signal_raw(int signal_num, const InputState &input_state);
        movit::Effect *get_effect() const
        {
                if (deinterlace) {
@@ -128,6 +181,7 @@ private:
        std::vector<movit::FlatInput *> rgba_inputs;  // Multiple ones if deinterlacing. Owned by the chain.
        movit::Effect *deinterlace_effect = nullptr;  // Owned by the chain.
        bool deinterlace;
+       bool user_connectable;
 };
 
 #endif  // !defined(_THEME_H)