Add support for persistent USB device memory.

[bmusb] / bmusb.h

diff --git a/bmusb.h b/bmusb.h
index 17b5bc082d7a4dc266d25d80d93dcf1e3123b242..4c7254bcb5dea6bb773381c510565cea37035839 100644 (file)
--- a/bmusb.h
+++ b/bmusb.h
@@ -6,12 +6,14 @@
 #include <condition_variable>
 #include <deque>
 #include <functional>
+#include <map>
 #include <mutex>
 #include <stack>
 #include <string>
 #include <thread>
 #include <vector>
 
+struct libusb_device_handle;
 struct libusb_transfer;
 
 // An interface for frame allocators; if you do not specify one
@@ -78,6 +80,16 @@ private:
        std::stack<std::unique_ptr<uint8_t[]>> freelist;  // All of size <frame_size>.
 };
 
+// Represents an input mode you can tune a card to.
+struct VideoMode {
+       std::string name;
+       bool autodetect = false;  // If true, all the remaining fields are irrelevant.
+       unsigned width = 0, height = 0;
+       unsigned frame_rate_num = 0, frame_rate_den = 0;
+       bool interlaced = false;
+};
+
+// Represents the format of an actual frame coming in.
 struct VideoFormat {
        uint16_t id = 0;  // For debugging/logging only.
        unsigned width = 0, height = 0, second_field_start = 0;
@@ -102,6 +114,18 @@ class CaptureInterface {
  public:
        virtual ~CaptureInterface() {}
 
+       virtual std::map<uint32_t, VideoMode> get_available_video_modes() const = 0;
+       virtual uint32_t get_current_video_mode() const = 0;
+       virtual void set_video_mode(uint32_t video_mode_id) = 0;
+
+       virtual std::map<uint32_t, std::string> get_available_video_inputs() const = 0;
+       virtual void set_video_input(uint32_t video_input_id) = 0;
+       virtual uint32_t get_current_video_input() const = 0;
+
+       virtual std::map<uint32_t, std::string> get_available_audio_inputs() const = 0;
+       virtual void set_audio_input(uint32_t audio_input_id) = 0;
+       virtual uint32_t get_current_audio_input() const = 0;
+
        // Does not take ownership.
        virtual void set_video_frame_allocator(FrameAllocator *allocator) = 0;
 
@@ -137,10 +161,25 @@ class BMUSBCapture : public CaptureInterface {
 
        ~BMUSBCapture() {}
 
+       std::map<uint32_t, VideoMode> get_available_video_modes() const override;
+       uint32_t get_current_video_mode() const override;
+       void set_video_mode(uint32_t video_mode_id) override;
+
+       virtual std::map<uint32_t, std::string> get_available_video_inputs() const override;
+       virtual void set_video_input(uint32_t video_input_id) override;
+       virtual uint32_t get_current_video_input() const override { return current_video_input; }
+
+       virtual std::map<uint32_t, std::string> get_available_audio_inputs() const override;
+       virtual void set_audio_input(uint32_t audio_input_id) override;
+       virtual uint32_t get_current_audio_input() const override { return current_audio_input; }
+
        // Does not take ownership.
        void set_video_frame_allocator(FrameAllocator *allocator) override
        {
                video_frame_allocator = allocator;
+               if (owned_video_frame_allocator.get() != allocator) {
+                       owned_video_frame_allocator.reset();
+               }
        }
 
        FrameAllocator *get_video_frame_allocator() override
@@ -152,6 +191,9 @@ class BMUSBCapture : public CaptureInterface {
        void set_audio_frame_allocator(FrameAllocator *allocator) override
        {
                audio_frame_allocator = allocator;
+               if (owned_audio_frame_allocator.get() != allocator) {
+                       owned_audio_frame_allocator.reset();
+               }
        }
 
        FrameAllocator *get_audio_frame_allocator() override
@@ -201,6 +243,8 @@ class BMUSBCapture : public CaptureInterface {
        static void usb_thread_func();
        static void cb_xfr(struct libusb_transfer *xfr);
 
+       void update_capture_mode();
+
        std::string description;
 
        FrameAllocator::Frame current_video_frame;
@@ -213,6 +257,8 @@ class BMUSBCapture : public CaptureInterface {
 
        FrameAllocator *video_frame_allocator = nullptr;
        FrameAllocator *audio_frame_allocator = nullptr;
+       std::unique_ptr<FrameAllocator> owned_video_frame_allocator;
+       std::unique_ptr<FrameAllocator> owned_audio_frame_allocator;
        frame_callback_t frame_callback = nullptr;
 
        std::thread dequeue_thread;
@@ -229,6 +275,10 @@ class BMUSBCapture : public CaptureInterface {
        int card_index;
        std::vector<libusb_transfer *> iso_xfrs;
        int assumed_frame_width = 1280;
+
+       libusb_device_handle *devh = nullptr;
+       uint32_t current_video_input = 0x00000000;  // HDMI/SDI.
+       uint32_t current_audio_input = 0x00000000;  // Embedded.
 };
 
 // Get details for the given video format; returns false if detection was incomplete.