]> git.sesse.net Git - bmusb/blobdiff - bmusb.cpp
Add a (commented-out) useful AVX2 debugging function.
[bmusb] / bmusb.cpp
index c3914ac83ca29482a3e2ec503c8266daa3337a95..78e6b412d1438cdd41354c671ca3c117aa371c57 100644 (file)
--- a/bmusb.cpp
+++ b/bmusb.cpp
@@ -45,6 +45,9 @@ using namespace std::placeholders;
 
 FILE *audiofp;
 
+thread usb_thread;
+atomic<bool> should_quit;
+
 FrameAllocator::~FrameAllocator() {}
 
 #define NUM_QUEUED_FRAMES 8
@@ -280,6 +283,49 @@ void add_to_frame(FrameAllocator::Frame *current_frame, const char *frame_type_n
 
 #ifdef __SSE2__
 
+#if 0
+void avx2_dump(const char *name, __m256i n)
+{
+       printf("%-10s:", name);
+       printf(" %02x", _mm256_extract_epi8(n, 0));
+       printf(" %02x", _mm256_extract_epi8(n, 1));
+       printf(" %02x", _mm256_extract_epi8(n, 2));
+       printf(" %02x", _mm256_extract_epi8(n, 3));
+       printf(" %02x", _mm256_extract_epi8(n, 4));
+       printf(" %02x", _mm256_extract_epi8(n, 5));
+       printf(" %02x", _mm256_extract_epi8(n, 6));
+       printf(" %02x", _mm256_extract_epi8(n, 7));
+       printf(" ");
+       printf(" %02x", _mm256_extract_epi8(n, 8));
+       printf(" %02x", _mm256_extract_epi8(n, 9));
+       printf(" %02x", _mm256_extract_epi8(n, 10));
+       printf(" %02x", _mm256_extract_epi8(n, 11));
+       printf(" %02x", _mm256_extract_epi8(n, 12));
+       printf(" %02x", _mm256_extract_epi8(n, 13));
+       printf(" %02x", _mm256_extract_epi8(n, 14));
+       printf(" %02x", _mm256_extract_epi8(n, 15));
+       printf(" ");
+       printf(" %02x", _mm256_extract_epi8(n, 16));
+       printf(" %02x", _mm256_extract_epi8(n, 17));
+       printf(" %02x", _mm256_extract_epi8(n, 18));
+       printf(" %02x", _mm256_extract_epi8(n, 19));
+       printf(" %02x", _mm256_extract_epi8(n, 20));
+       printf(" %02x", _mm256_extract_epi8(n, 21));
+       printf(" %02x", _mm256_extract_epi8(n, 22));
+       printf(" %02x", _mm256_extract_epi8(n, 23));
+       printf(" ");
+       printf(" %02x", _mm256_extract_epi8(n, 24));
+       printf(" %02x", _mm256_extract_epi8(n, 25));
+       printf(" %02x", _mm256_extract_epi8(n, 26));
+       printf(" %02x", _mm256_extract_epi8(n, 27));
+       printf(" %02x", _mm256_extract_epi8(n, 28));
+       printf(" %02x", _mm256_extract_epi8(n, 29));
+       printf(" %02x", _mm256_extract_epi8(n, 30));
+       printf(" %02x", _mm256_extract_epi8(n, 31));
+       printf("\n");
+}
+#endif
+
 // Does a memcpy and memchr in one to reduce processing time.
 // Note that the benefit is somewhat limited if your L3 cache is small,
 // as you'll (unfortunately) spend most of the time loading the data
@@ -335,32 +381,39 @@ const uint8_t *add_to_frame_fastpath(FrameAllocator::Frame *current_frame, const
 #if __AVX2__
        const __m256i needle = _mm256_set1_epi8(sync_char);
 
-       const __m256i *in = (const __m256i *)aligned_start;
+       const __restrict __m256i *in = (const __m256i *)aligned_start;
        if (current_frame->interleaved) {
-               __m256i *out1 = (__m256i *)(current_frame->data + (current_frame->len + 1) / 2);
-               __m256i *out2 = (__m256i *)(current_frame->data2 + current_frame->len / 2);
+               __restrict __m256i *out1 = (__m256i *)(current_frame->data + (current_frame->len + 1) / 2);
+               __restrict __m256i *out2 = (__m256i *)(current_frame->data2 + current_frame->len / 2);
                if (current_frame->len % 2 == 1) {
                        swap(out1, out2);
                }
 
-               __m256i mask_lower_byte = _mm256_set1_epi16(0x00ff);
+               __m256i shuffle_cw = _mm256_set_epi8(
+                       15, 13, 11, 9, 7, 5, 3, 1, 14, 12, 10, 8, 6, 4, 2, 0,
+                       15, 13, 11, 9, 7, 5, 3, 1, 14, 12, 10, 8, 6, 4, 2, 0);
                while (in < (const __m256i *)limit) {
-                       __m256i data1 = _mm256_load_si256(in);
-                       __m256i data2 = _mm256_load_si256(in + 1);
-                       __m256i data1_lo = _mm256_and_si256(data1, mask_lower_byte);
-                       __m256i data2_lo = _mm256_and_si256(data2, mask_lower_byte);
-                       __m256i data1_hi = _mm256_srli_epi16(data1, 8);
-                       __m256i data2_hi = _mm256_srli_epi16(data2, 8);
-                       __m256i lo = _mm256_packus_epi16(data1_lo, data2_lo);
-                       lo = _mm256_permute4x64_epi64(lo, 0b11011000);
-                       _mm256_storeu_si256(out1, lo);  // Store as early as possible, even if the data isn't used.
-                       __m256i hi = _mm256_packus_epi16(data1_hi, data2_hi);
-                       hi = _mm256_permute4x64_epi64(hi, 0b11011000);
-                       _mm256_storeu_si256(out2, hi);
+                       // Note: For brevity, comments show lanes as if they were 2x64-bit (they're actually 2x128).
+                       __m256i data1 = _mm256_stream_load_si256(in);         // AaBbCcDd EeFfGgHh
+                       __m256i data2 = _mm256_stream_load_si256(in + 1);     // IiJjKkLl MmNnOoPp
+
                        __m256i found1 = _mm256_cmpeq_epi8(data1, needle);
                        __m256i found2 = _mm256_cmpeq_epi8(data2, needle);
-                       if (!_mm256_testz_si256(found1, found1) ||
-                           !_mm256_testz_si256(found2, found2)) {
+                       __m256i found = _mm256_or_si256(found1, found2);
+
+                       data1 = _mm256_shuffle_epi8(data1, shuffle_cw);       // ABCDabcd EFGHefgh
+                       data2 = _mm256_shuffle_epi8(data2, shuffle_cw);       // IJKLijkl MNOPmnop
+               
+                       data1 = _mm256_permute4x64_epi64(data1, 0b11011000);  // ABCDEFGH abcdefgh
+                       data2 = _mm256_permute4x64_epi64(data2, 0b11011000);  // IJKLMNOP ijklmnop
+
+                       __m256i lo = _mm256_permute2x128_si256(data1, data2, 0b00100000);
+                       __m256i hi = _mm256_permute2x128_si256(data1, data2, 0b00110001);
+
+                       _mm256_storeu_si256(out1, lo);  // Store as early as possible, even if the data isn't used.
+                       _mm256_storeu_si256(out2, hi);
+
+                       if (!_mm256_testz_si256(found, found)) {
                                break;
                        }
 
@@ -567,7 +620,7 @@ void BMUSBCapture::usb_thread_func()
        }
 }
 
-void BMUSBCapture::start_bm_capture()
+void BMUSBCapture::configure_card()
 {
        if (video_frame_allocator == nullptr) {
                set_video_frame_allocator(new MallocFrameAllocator(FRAME_SIZE));  // FIXME: leak.
@@ -579,7 +632,6 @@ void BMUSBCapture::start_bm_capture()
 
        int rc;
        struct libusb_transfer *xfr;
-       vector<libusb_transfer *> iso_xfrs;
 
        rc = libusb_init(nullptr);
        if (rc < 0) {
@@ -587,7 +639,9 @@ void BMUSBCapture::start_bm_capture()
                exit(1);
        }
 
-       struct libusb_device_handle *devh = libusb_open_device_with_vid_pid(nullptr, 0x1edb, 0xbd3b);
+       //struct libusb_device_handle *devh = libusb_open_device_with_vid_pid(nullptr, 0x1edb, 0xbd3b);
+       //struct libusb_device_handle *devh = libusb_open_device_with_vid_pid(nullptr, 0x1edb, 0xbd4f);
+       struct libusb_device_handle *devh = libusb_open_device_with_vid_pid(nullptr, vid, pid);
        if (!devh) {
                fprintf(stderr, "Error finding USB device\n");
                exit(1);
@@ -686,8 +740,9 @@ void BMUSBCapture::start_bm_capture()
        //    36 can be set to 0 with no apparent effect (all of this tested on both video and audio),
        //    but the driver sets it to 0x8036802a at some point.
        //
-       //    0 is written to during firmware upgrade. Probably best to stay out of it unless you know
-       //    what you're doing.
+       //    all of this is on request 214/215. other requests (192, 219,
+       //    222, 223, 224) are used for firmware upgrade. Probably best to
+       //    stay out of it unless you know what you're doing.
        //
        //
        // register 16:
@@ -709,59 +764,6 @@ void BMUSBCapture::start_bm_capture()
        static const ctrl ctrls[] = {
                { LIBUSB_ENDPOINT_IN,  214, 16, 0 },
                { LIBUSB_ENDPOINT_IN,  214,  0, 0 },
-               { LIBUSB_ENDPOINT_IN,  214,  0, 0 },
-               { LIBUSB_ENDPOINT_IN,  214,  4, 0 },
-               { LIBUSB_ENDPOINT_IN,  214, 12, 0 },
-               { LIBUSB_ENDPOINT_IN,  214, 16, 0 },
-               { LIBUSB_ENDPOINT_IN,  214, 20, 0 },
-               { LIBUSB_ENDPOINT_IN,  214, 24, 0 },
-               { LIBUSB_ENDPOINT_IN,  214, 28, 0 },
-               { LIBUSB_ENDPOINT_IN,  215, 32, 0 },
-               { LIBUSB_ENDPOINT_IN,  214, 36, 0 },
-               { LIBUSB_ENDPOINT_IN,  214, 40, 0 },
-               { LIBUSB_ENDPOINT_IN,  216, 44, 0 },
-               { LIBUSB_ENDPOINT_IN,  214, 48, 0 },
-               { LIBUSB_ENDPOINT_IN,  214, 52, 0 },
-               { LIBUSB_ENDPOINT_IN,  214, 12, 0 },
-               { LIBUSB_ENDPOINT_IN,  214, 40, 0 },
-               { LIBUSB_ENDPOINT_IN,  214, 12, 0 },
-               { LIBUSB_ENDPOINT_IN,  214, 12, 0 },
-               { LIBUSB_ENDPOINT_IN,  214, 12, 0 },
-               { LIBUSB_ENDPOINT_IN,  214, 40, 0 },
-               { LIBUSB_ENDPOINT_IN,  214, 12, 0 },
-               { LIBUSB_ENDPOINT_IN,  214, 12, 0 },
-               { LIBUSB_ENDPOINT_IN,  214, 12, 0 },
-               { LIBUSB_ENDPOINT_IN,  214, 40, 0 },
-               { LIBUSB_ENDPOINT_IN,  214, 12, 0 },
-               { LIBUSB_ENDPOINT_IN,  214, 24, 0 },
-               { LIBUSB_ENDPOINT_IN,  214, 12, 0 },
-               { LIBUSB_ENDPOINT_IN,  214, 12, 0 },
-               { LIBUSB_ENDPOINT_IN,  214, 40, 0 },
-               { LIBUSB_ENDPOINT_IN,  214, 12, 0 },
-               { LIBUSB_ENDPOINT_IN,  214, 12, 0 },
-               { LIBUSB_ENDPOINT_IN,  214, 12, 0 },
-               { LIBUSB_ENDPOINT_IN,  214, 40, 0 },
-               { LIBUSB_ENDPOINT_IN,  214, 12, 0 },
-               { LIBUSB_ENDPOINT_IN,  214, 12, 0 },
-               { LIBUSB_ENDPOINT_IN,  214, 12, 0 },
-               { LIBUSB_ENDPOINT_IN,  214, 40, 0 },
-               { LIBUSB_ENDPOINT_IN,  214, 24, 0 },
-               { LIBUSB_ENDPOINT_IN,  214, 12, 0 },
-               { LIBUSB_ENDPOINT_IN,  214, 12, 0 },
-               { LIBUSB_ENDPOINT_IN,  214, 12, 0 },
-               { LIBUSB_ENDPOINT_IN,  214, 40, 0 },
-               { LIBUSB_ENDPOINT_IN,  214, 12, 0 },
-               { LIBUSB_ENDPOINT_IN,  214, 12, 0 },
-               { LIBUSB_ENDPOINT_IN,  214, 12, 0 },
-               { LIBUSB_ENDPOINT_IN,  214, 40, 0 },
-               { LIBUSB_ENDPOINT_IN,  214, 12, 0 },  // packet 354
-               { LIBUSB_ENDPOINT_IN,  214, 24, 0 },
-               { LIBUSB_ENDPOINT_IN,  214, 12, 0 },
-               { LIBUSB_ENDPOINT_IN,  214, 12, 0 },
-               { LIBUSB_ENDPOINT_IN,  214, 40, 0 },
-               // more...
-               //{ LIBUSB_ENDPOINT_OUT, 215,  0, 0x80000100 },
-               //{ LIBUSB_ENDPOINT_OUT, 215,  0, 0x09000000 },  // wow, some kind of mode
 
                // seems to capture on HDMI, clearing the 0x20000000 bit seems to activate 10-bit
                // capture (v210).
@@ -771,46 +773,10 @@ void BMUSBCapture::start_bm_capture()
                // 0x3c000000 = composite video? (analog audio)
                // 0x3e000000 = s-video? (analog audio)
                { LIBUSB_ENDPOINT_OUT, 215,  0, 0x29000000 },
+               //{ LIBUSB_ENDPOINT_OUT, 215,  0, 0x80000100 },
                //{ LIBUSB_ENDPOINT_OUT, 215,  0, 0x09000000 },
-
-               //{ LIBUSB_ENDPOINT_OUT, 215, 28, 0xffffffff },
-               //{ LIBUSB_ENDPOINT_OUT, 215, 32, 0xffffffff },
-               //{ LIBUSB_ENDPOINT_OUT, 215, 28, 0x40404040 },
-               //{ LIBUSB_ENDPOINT_OUT, 215, 32, 0x40404040 },
-               //{ LIBUSB_ENDPOINT_OUT, 215, 36, 0x8036802a },
                { LIBUSB_ENDPOINT_OUT, 215, 24, 0x73c60001 },  // latch for frame start?
-               //{ LIBUSB_ENDPOINT_OUT, 215, 24, 0x13370001 },  // latch for frame start?
                { LIBUSB_ENDPOINT_IN,  214, 24, 0 },  // 
-               //{ LIBUSB_ENDPOINT_OUT, 215,  4, 0x00000000 },  // appears to have no e fect
-               //{ LIBUSB_ENDPOINT_OUT, 215,  8, 0x00000000 },  // appears to have no effect
-               //{ LIBUSB_ENDPOINT_OUT, 215, 20, 0x00000000 },  // appears to have no effect
-               //{ LIBUSB_ENDPOINT_OUT, 215, 28, 0x00000000 },  // appears to have no effect
-               //{ LIBUSB_ENDPOINT_OUT, 215, 32, 0x00000000 },  // appears to have no effect
-               //{ LIBUSB_ENDPOINT_OUT, 215, 36, 0x00000000 },  // appears to have no effect
-#if 0
-               { LIBUSB_ENDPOINT_OUT, 215,  0 },
-               { LIBUSB_ENDPOINT_OUT, 215,  0 },
-               { LIBUSB_ENDPOINT_OUT, 215, 28 },
-               { LIBUSB_ENDPOINT_OUT, 215, 32 },
-               { LIBUSB_ENDPOINT_OUT, 215, 36 },
-               { LIBUSB_ENDPOINT_OUT, 215, 24 },
-               { LIBUSB_ENDPOINT_OUT, 215, 24 },
-               { LIBUSB_ENDPOINT_OUT, 215, 24 },
-               { LIBUSB_ENDPOINT_OUT, 215, 24 },
-               { LIBUSB_ENDPOINT_OUT, 215, 24 },
-               { LIBUSB_ENDPOINT_OUT, 215, 24 },
-               { LIBUSB_ENDPOINT_OUT, 215, 24 },
-               { LIBUSB_ENDPOINT_OUT, 215, 24 },
-               { LIBUSB_ENDPOINT_OUT, 215, 24 },
-               { LIBUSB_ENDPOINT_OUT, 215, 24 },
-               { LIBUSB_ENDPOINT_OUT, 215,  0 },
-               { LIBUSB_ENDPOINT_OUT, 215, 24 },
-               { LIBUSB_ENDPOINT_OUT, 215, 24 },
-               { LIBUSB_ENDPOINT_OUT, 215, 24 },
-               { LIBUSB_ENDPOINT_OUT, 215, 24 },
-               { LIBUSB_ENDPOINT_OUT, 215, 24 },
-               { LIBUSB_ENDPOINT_OUT, 215, 24 },
-#endif
        };
 
        for (unsigned req = 0; req < sizeof(ctrls) / sizeof(ctrls[0]); ++req) {
@@ -910,7 +876,7 @@ void BMUSBCapture::start_bm_capture()
                                        size &= ~1023;
                                        size += 1024;
                                }
-                               num_iso_pack = (2 << 20) / size;  // 2 MB.
+                               num_iso_pack = (2 << 18) / size;  // 512 kB.
                                printf("Picking %d packets of 0x%x bytes each\n", num_iso_pack, size);
                        } else {
                                size = 0xc0;
@@ -933,24 +899,24 @@ void BMUSBCapture::start_bm_capture()
                        iso_xfrs.push_back(xfr);
                }
        }
+}
 
-       {
-               int i = 0;
-               for (libusb_transfer *xfr : iso_xfrs) {
-                       rc = libusb_submit_transfer(xfr);
-                       ++i;
-                       if (rc < 0) {
-                               //printf("num_bytes=%d\n", num_bytes);
-                               fprintf(stderr, "Error submitting iso to endpoint 0x%02x, number %d: %s\n",
-                                       xfr->endpoint, i, libusb_error_name(rc));
-                               exit(1);
-                       }
+void BMUSBCapture::start_bm_capture()
+{
+       printf("starting capture\n");
+       int i = 0;
+       for (libusb_transfer *xfr : iso_xfrs) {
+               printf("submitting transfer...\n");
+               int rc = libusb_submit_transfer(xfr);
+               ++i;
+               if (rc < 0) {
+                       //printf("num_bytes=%d\n", num_bytes);
+                       fprintf(stderr, "Error submitting iso to endpoint 0x%02x, number %d: %s\n",
+                               xfr->endpoint, i, libusb_error_name(rc));
+                       exit(1);
                }
        }
 
-       should_quit = false;
-       usb_thread = thread(&BMUSBCapture::usb_thread_func, this);
-
 
 #if 0
        libusb_release_interface(devh, 0);
@@ -962,7 +928,13 @@ out:
 #endif
 }
 
-void BMUSBCapture::stop_bm_capture()
+void BMUSBCapture::start_bm_thread()
+{
+       should_quit = false;
+       usb_thread = thread(&BMUSBCapture::usb_thread_func);
+}
+
+void BMUSBCapture::stop_bm_thread()
 {
        should_quit = true;
        usb_thread.join();