X-Git-Url: https://git.sesse.net/?a=blobdiff_plain;f=bmusb.h;h=4c7254bcb5dea6bb773381c510565cea37035839;hb=653fb04b1f7fd18a3361524aeb4877efca032cdb;hp=e905e05afc82496abca75a5ef22083826f805125;hpb=487a136592e4bd5acda92d35cf52fe4489c69bbf;p=bmusb

diff --git a/bmusb.h b/bmusb.h
index e905e05..4c7254b 100644
--- a/bmusb.h
+++ b/bmusb.h
@@ -13,6 +13,7 @@
 #include <thread>
 #include <vector>
 
+struct libusb_device_handle;
 struct libusb_transfer;
 
 // An interface for frame allocators; if you do not specify one
@@ -114,9 +115,17 @@ class CaptureInterface {
 	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;
 
@@ -153,13 +162,24 @@ 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
@@ -171,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
@@ -220,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;
@@ -232,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;
@@ -248,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.