X-Git-Url: https://git.sesse.net/?a=blobdiff_plain;f=fake_capture.cpp;h=5f4f5da4ea5904e9d7186eb9eef98003de1644be;hb=01ddb8f836114c07cff3ca040d9ed2c946b2fdbf;hp=961700eb54158a2b9245059d89f384620dac1f32;hpb=e0837a17b5a497476d67237c768836e51f8a4ce7;p=bmusb diff --git a/fake_capture.cpp b/fake_capture.cpp index 961700e..5f4f5da 100644 --- a/fake_capture.cpp +++ b/fake_capture.cpp @@ -4,6 +4,7 @@ #include "bmusb/fake_capture.h" #include +#include #include #include #include @@ -13,6 +14,7 @@ #if __SSE2__ #include #endif +#include #include #include "bmusb/bmusb.h" @@ -26,6 +28,7 @@ constexpr uint8_t cbs[NUM_COLORS] = { 90, 54, 240, 128 }; constexpr uint8_t crs[NUM_COLORS] = { 240, 34, 110, 128 }; using namespace std; +using namespace std::chrono; namespace bmusb { namespace { @@ -84,10 +87,30 @@ void memset4(uint8_t *s, const uint8_t c[4], size_t n) } } +void memset16(uint8_t *s, const uint32_t c[4], size_t n) +{ + size_t i = 0; +#if __SSE2__ + __m128i cc = *(__m128i *)c; + __m128i *out = (__m128i *)s; + + for ( ; i < (n & ~1); i += 2) { + _mm_storeu_si128(out++, cc); + _mm_storeu_si128(out++, cc); + } + + s = (uint8_t *)out; +#endif + for ( ; i < n; ++i) { + memcpy(s, c, 16); + s += 16; + } +} + } // namespace FakeCapture::FakeCapture(unsigned width, unsigned height, unsigned fps, unsigned audio_sample_frequency, int card_index, bool has_audio) - : width(width), height(height), fps(fps), audio_sample_frequency(audio_sample_frequency) + : width(width), height(height), fps(fps), audio_sample_frequency(audio_sample_frequency), card_index(card_index) { char buf[256]; snprintf(buf, sizeof(buf), "Fake card %d", card_index + 1); @@ -199,6 +222,10 @@ bool timespec_less_than(const timespec &a, const timespec &b) void FakeCapture::producer_thread_func() { + char thread_name[16]; + snprintf(thread_name, sizeof(thread_name), "FakeCapture_%d", card_index); + pthread_setname_np(pthread_self(), thread_name); + uint16_t timecode = 0; if (has_dequeue_callbacks) { @@ -231,6 +258,7 @@ void FakeCapture::producer_thread_func() next_frame = now; } } + steady_clock::time_point timestamp = steady_clock::now(); // Figure out when the next frame is to be, then compute the current one. add_time(1.0 / fps, &next_frame); @@ -238,6 +266,11 @@ void FakeCapture::producer_thread_func() VideoFormat video_format; video_format.width = width; video_format.height = height; + if (current_pixel_format == PixelFormat_10BitYCbCr) { + video_format.stride = (width + 5) / 6 * 4 * sizeof(uint32_t); + } else { + video_format.stride = width * 2; + } video_format.frame_rate_nom = fps; video_format.frame_rate_den = 1; video_format.has_signal = true; @@ -247,31 +280,45 @@ void FakeCapture::producer_thread_func() if (video_frame.data != nullptr) { assert(video_frame.size >= width * height * 2); if (video_frame.interleaved) { + assert(current_pixel_format == PixelFormat_8BitYCbCr); uint8_t cbcr[] = { cb, cr }; memset2(video_frame.data, cbcr, width * height / 2); memset(video_frame.data2, y, width * height); } else { - uint8_t ycbcr[] = { y, cb, y, cr }; - memset4(video_frame.data, ycbcr, width * height / 2); + if (current_pixel_format == PixelFormat_10BitYCbCr) { + // Just use the 8-bit-values shifted left by 2. + // It's not 100% correct, but it's close enough. + uint32_t pix[4]; + pix[0] = (cb << 2) | (y << 12) | (cr << 22); + pix[1] = (y << 2) | (cb << 12) | ( y << 22); + pix[2] = (cr << 2) | (y << 12) | (cb << 22); + pix[3] = (y << 2) | (cr << 12) | ( y << 22); + memset16(video_frame.data, pix, video_format.stride * height / sizeof(pix)); + } else { + uint8_t ycbcr[] = { y, cb, y, cr }; + memset4(video_frame.data, ycbcr, width * height / 2); + } } - video_frame.len = width * height * 2; + video_frame.len = video_format.stride * height; + video_frame.received_timestamp = timestamp; } AudioFormat audio_format; audio_format.bits_per_sample = 32; - audio_format.num_channels = 2; + audio_format.num_channels = 8; FrameAllocator::Frame audio_frame = audio_frame_allocator->alloc_frame(); if (audio_frame.data != nullptr) { const unsigned num_stereo_samples = audio_sample_frequency / fps; - assert(audio_frame.size >= 2 * sizeof(int32_t) * num_stereo_samples); - audio_frame.len = 2 * sizeof(int32_t) * num_stereo_samples; + assert(audio_frame.size >= audio_format.num_channels * sizeof(int32_t) * num_stereo_samples); + audio_frame.len = audio_format.num_channels * sizeof(int32_t) * num_stereo_samples; + audio_frame.received_timestamp = timestamp; if (audio_sin == 0.0f) { // Silence. memset(audio_frame.data, 0, audio_frame.len); } else { - make_tone((int32_t *)audio_frame.data, num_stereo_samples); + make_tone((int32_t *)audio_frame.data, num_stereo_samples, audio_format.num_channels); } } @@ -284,14 +331,15 @@ void FakeCapture::producer_thread_func() } } -void FakeCapture::make_tone(int32_t *out, unsigned num_stereo_samples) +void FakeCapture::make_tone(int32_t *out, unsigned num_stereo_samples, unsigned num_channels) { int32_t *ptr = out; float r = audio_real, i = audio_imag; for (unsigned sample_num = 0; sample_num < num_stereo_samples; ++sample_num) { int32_t s = lrintf(r); - *ptr++ = s; - *ptr++ = s; + for (unsigned i = 0; i < num_channels; ++i) { + *ptr++ = s; + } // Rotate the phaser by one sample. float new_r = r * audio_cos - i * audio_sin;