]> git.sesse.net Git - bmusb/blobdiff - bmusb.cpp
Add an RGBA pixel type.
[bmusb] / bmusb.cpp
index 4d524422f944d4a8957a1046e4489a997a571f46..f01e0de97e6d1caf9b425f0324ba134e67dae037 100644 (file)
--- a/bmusb.cpp
+++ b/bmusb.cpp
@@ -1,5 +1,5 @@
-// Intensity Shuttle USB3 capture driver, v0.5.4
-// Can download 8-bit and 10-bit UYVY/v210 frames from HDMI, quite stable
+// Intensity Shuttle USB3 capture driver, v0.6.0
+// Can download 8-bit and 10-bit UYVY/v210-ish frames from HDMI, quite stable
 // (can do captures for hours at a time with no drops), except during startup
 // 576p60/720p60/1080i60 works, 1080p60 does not work (firmware limitation)
 // Audio comes out as 8-channel 24-bit raw audio.
@@ -63,14 +63,24 @@ FILE *audiofp;
 thread usb_thread;
 atomic<bool> should_quit;
 
-int find_xfer_size_for_width(int width)
+int v210_stride(int width)
+{
+       return (width + 5) / 6 * 4 * sizeof(uint32_t);
+}
+
+int find_xfer_size_for_width(PixelFormat pixel_format, int width)
 {
        // Video seems to require isochronous packets scaled with the width;
        // seemingly six lines is about right, rounded up to the required 1kB
        // multiple.
-       int size = width * 2 * 6;
        // Note that for 10-bit input, you'll need to increase size accordingly.
-       //size = size * 4 / 3;
+       int stride;
+       if (pixel_format == PixelFormat_10BitYCbCr) {
+               stride = v210_stride(width);
+       } else {
+               stride = width * sizeof(uint16_t);
+       }
+       int size = stride * 6;
        if (size % 1024 != 0) {
                size &= ~1023;
                size += 1024;
@@ -78,10 +88,10 @@ int find_xfer_size_for_width(int width)
        return size;
 }
 
-void change_xfer_size_for_width(int width, libusb_transfer *xfr)
+void change_xfer_size_for_width(PixelFormat pixel_format, int width, libusb_transfer *xfr)
 {
        assert(width >= MIN_WIDTH);
-       size_t size = find_xfer_size_for_width(width);
+       size_t size = find_xfer_size_for_width(pixel_format, width);
        int num_iso_pack = xfr->length / size;
        if (num_iso_pack != xfr->num_iso_packets ||
            size != xfr->iso_packet_desc[0].length) {
@@ -120,7 +130,7 @@ bool decode_video_format(uint16_t video_format, VideoFormat *decoded_video_forma
                decoded_video_format->has_signal = false;
                return true;
        }
-       if ((video_format & 0xe800) != 0xe800) {
+       if ((video_format & 0xe000) != 0xe000) {
                printf("Video format 0x%04x does not appear to be a video format. Assuming 60 Hz.\n",
                        video_format);
                decoded_video_format->width = 0;
@@ -137,10 +147,16 @@ bool decode_video_format(uint16_t video_format, VideoFormat *decoded_video_forma
        decoded_video_format->has_signal = true;
 
        // NTSC (480i59.94, I suppose). A special case, see below.
-       if (video_format == 0xe901 || video_format == 0xe9c1 || video_format == 0xe801) {
+       if ((video_format & ~0x0800) == 0xe101 ||
+           (video_format & ~0x0800) == 0xe1c1 ||
+           (video_format & ~0x0800) == 0xe001) {
                decoded_video_format->width = 720;
                decoded_video_format->height = 480;
-               decoded_video_format->stride = 720 * 2;
+               if (video_format & 0x0800) {
+                       decoded_video_format->stride = 720 * 2;
+               } else {
+                       decoded_video_format->stride = v210_stride(720);
+               }
                decoded_video_format->extra_lines_top = 17;
                decoded_video_format->extra_lines_bottom = 28;
                decoded_video_format->frame_rate_nom = 30000;
@@ -151,10 +167,18 @@ bool decode_video_format(uint16_t video_format, VideoFormat *decoded_video_forma
        }
 
        // PAL (576i50, I suppose). A special case, see below.
-       if (video_format == 0xe909 || video_format == 0xe9c9 || video_format == 0xe809 || video_format == 0xebe9 || video_format == 0xebe1) {
+       if ((video_format & ~0x0800) == 0xe109 ||
+           (video_format & ~0x0800) == 0xe1c9 ||
+           (video_format & ~0x0800) == 0xe009 ||
+           (video_format & ~0x0800) == 0xe3e9 ||
+           (video_format & ~0x0800) == 0xe3e1) {
                decoded_video_format->width = 720;
                decoded_video_format->height = 576;
-               decoded_video_format->stride = 720 * 2;
+               if (video_format & 0x0800) {
+                       decoded_video_format->stride = 720 * 2;
+               } else {
+                       decoded_video_format->stride = v210_stride(720);
+               }
                decoded_video_format->extra_lines_top = 22;
                decoded_video_format->extra_lines_bottom = 27;
                decoded_video_format->frame_rate_nom = 25;
@@ -167,7 +191,8 @@ bool decode_video_format(uint16_t video_format, VideoFormat *decoded_video_forma
        // 0x8 seems to be a flag about availability of deep color on the input,
        // except when it's not (e.g. it's the only difference between NTSC
        // and PAL). Rather confusing. But we clear it here nevertheless, because
-       // usually it doesn't mean anything.
+       // usually it doesn't mean anything. 0x0800 appears to be 8-bit input
+       // (as opposed to 10-bit).
        //
        // 0x4 is a flag I've only seen from the D4. I don't know what it is.
        uint16_t normalized_video_format = video_format & ~0xe80c;
@@ -179,20 +204,24 @@ bool decode_video_format(uint16_t video_format, VideoFormat *decoded_video_forma
                { 0x0103, 1280,  720,   0, 25,  5,    60,    1, false },  // 720p60.
                { 0x0125, 1280,  720,   0, 25,  5,    60,    1, false },  // 720p60.
                { 0x0121, 1280,  720,   0, 25,  5, 60000, 1001, false },  // 720p59.94.
-               { 0x01c3, 1920, 1080,   0, 20, 25,    30,    1, false },  // 1080p30.
+               { 0x01c3, 1920, 1080,   0, 41,  4,    30,    1, false },  // 1080p30.
                { 0x0003, 1920, 1080, 583, 20, 25,    30,    1,  true },  // 1080i60.
-               { 0x01e1, 1920, 1080,   0, 20, 25, 30000, 1001, false },  // 1080p29.97.
+               { 0x01e1, 1920, 1080,   0, 41,  4, 30000, 1001, false },  // 1080p29.97.
                { 0x0021, 1920, 1080, 583, 20, 25, 30000, 1001,  true },  // 1080i59.94.
-               { 0x0063, 1920, 1080,   0, 20, 25,    25,    1, false },  // 1080p25.
+               { 0x0063, 1920, 1080,   0, 41,  4,    25,    1, false },  // 1080p25.
                { 0x0043, 1920, 1080, 583, 20, 25,    25,    1,  true },  // 1080i50.
-               { 0x0083, 1920, 1080,   0, 20, 25,    24,    1, false },  // 1080p24.
-               { 0x00a1, 1920, 1080,   0, 20, 25, 24000, 1001, false },  // 1080p23.98.
+               { 0x0083, 1920, 1080,   0, 41,  4,    24,    1, false },  // 1080p24.
+               { 0x00a1, 1920, 1080,   0, 41,  4, 24000, 1001, false },  // 1080p23.98.
        };
        for (const VideoFormatEntry &entry : entries) {
                if (normalized_video_format == entry.normalized_video_format) {
                        decoded_video_format->width = entry.width;
                        decoded_video_format->height = entry.height;
-                       decoded_video_format->stride = entry.width * 2;
+                       if (video_format & 0x0800) {
+                               decoded_video_format->stride = entry.width * 2;
+                       } else {
+                               decoded_video_format->stride = v210_stride(entry.width);
+                       }
                        decoded_video_format->second_field_start = entry.second_field_start;
                        decoded_video_format->extra_lines_top = entry.extra_lines_top;
                        decoded_video_format->extra_lines_bottom = entry.extra_lines_bottom;
@@ -819,7 +848,7 @@ void BMUSBCapture::cb_xfr(struct libusb_transfer *xfr)
                        decode_packs(xfr, "\x00\x00\xff\xff", 4, &usb->current_video_frame, "video", bind(&BMUSBCapture::start_new_frame, usb, _1));
 
                        // Update the transfer with the new assumed width, if we're in the process of changing formats.
-                       change_xfer_size_for_width(usb->assumed_frame_width, xfr);
+                       change_xfer_size_for_width(usb->current_pixel_format, usb->assumed_frame_width, xfr);
                }
        }
        if (xfr->type == LIBUSB_TRANSFER_TYPE_CONTROL) {
@@ -1049,6 +1078,12 @@ unsigned BMUSBCapture::num_cards()
        return ret;
 }
 
+void BMUSBCapture::set_pixel_format(PixelFormat pixel_format)
+{
+       current_pixel_format = pixel_format;
+       update_capture_mode();
+}
+
 void BMUSBCapture::configure_card()
 {
        if (video_frame_allocator == nullptr) {
@@ -1307,16 +1342,15 @@ void BMUSBCapture::configure_card()
 
        // set up isochronous transfers for audio and video
        for (int e = 3; e <= 4; ++e) {
-               //int num_transfers = (e == 3) ? 6 : 6;
                int num_transfers = 6;
                for (int i = 0; i < num_transfers; ++i) {
                        size_t buf_size;
                        int num_iso_pack, size;
                        if (e == 3) {
                                // Allocate for minimum width (because that will give us the most
-                               // number of packets, so we don't need to reallocated, but we'll
+                               // number of packets, so we don't need to reallocate, but we'll
                                // default to 720p for the first frame.
-                               size = find_xfer_size_for_width(MIN_WIDTH);
+                               size = find_xfer_size_for_width(PixelFormat_8BitYCbCr, MIN_WIDTH);
                                num_iso_pack = USB_VIDEO_TRANSFER_SIZE / size;
                                buf_size = USB_VIDEO_TRANSFER_SIZE;
                        } else {
@@ -1355,7 +1389,7 @@ void BMUSBCapture::configure_card()
                        xfr->user_data = this;
 
                        if (e == 3) {
-                               change_xfer_size_for_width(assumed_frame_width, xfr);
+                               change_xfer_size_for_width(current_pixel_format, assumed_frame_width, xfr);
                        }
 
                        iso_xfrs.push_back(xfr);
@@ -1472,9 +1506,17 @@ void BMUSBCapture::set_audio_input(uint32_t audio_input_id)
 
 void BMUSBCapture::update_capture_mode()
 {
-       // clearing the 0x20000000 bit seems to activate 10-bit capture (v210).
-       // clearing the 0x08000000 bit seems to change the capture format (other source?)
-       uint32_t mode = htonl(0x29000000 | current_video_input | current_audio_input);
+       if (devh == nullptr) {
+               return;
+       }
+
+       // Clearing the 0x08000000 bit seems to change the capture format (other source?).
+       uint32_t mode = htonl(0x09000000 | current_video_input | current_audio_input);
+       if (current_pixel_format == PixelFormat_8BitYCbCr) {
+               mode |= htonl(0x20000000);
+       } else {
+               assert(current_pixel_format == PixelFormat_10BitYCbCr);
+       }
 
        int rc = libusb_control_transfer(devh, LIBUSB_REQUEST_TYPE_VENDOR | LIBUSB_ENDPOINT_OUT,
                /*request=*/215, /*value=*/0, /*index=*/0, (unsigned char *)&mode, sizeof(mode), /*timeout=*/0);