]> git.sesse.net Git - nageru/blobdiff - mixer.cpp
Refactor upload_func by splitting out some helper functions.
[nageru] / mixer.cpp
index b28da27e90f3781180dab3068b433135b739a1f2..81d4b73e5a2425027332071e02635596e4060fb7 100644 (file)
--- a/mixer.cpp
+++ b/mixer.cpp
@@ -45,6 +45,7 @@
 #include "ref_counted_gl_sync.h"
 #include "resampling_queue.h"
 #include "timebase.h"
+#include "timecode_renderer.h"
 #include "video_encoder.h"
 
 class IDeckLink;
@@ -76,27 +77,79 @@ void insert_new_frame(RefCountedFrame frame, unsigned field_num, bool interlaced
        }
 }
 
+void ensure_texture_resolution(PBOFrameAllocator::Userdata *userdata, unsigned field, unsigned width, unsigned height)
+{
+       if (userdata->tex_y[field] == 0 ||
+           userdata->tex_cbcr[field] == 0 ||
+           width != userdata->last_width[field] ||
+           height != userdata->last_height[field]) {
+               size_t cbcr_width = width / 2;
+
+               // We changed resolution since last use of this texture, so we need to create
+               // a new object. Note that this each card has its own PBOFrameAllocator,
+               // we don't need to worry about these flip-flopping between resolutions.
+               glBindTexture(GL_TEXTURE_2D, userdata->tex_cbcr[field]);
+               check_error();
+               glTexImage2D(GL_TEXTURE_2D, 0, GL_RG8, cbcr_width, height, 0, GL_RG, GL_UNSIGNED_BYTE, nullptr);
+               check_error();
+               glBindTexture(GL_TEXTURE_2D, userdata->tex_y[field]);
+               check_error();
+               glTexImage2D(GL_TEXTURE_2D, 0, GL_R8, width, height, 0, GL_RED, GL_UNSIGNED_BYTE, nullptr);
+               check_error();
+               userdata->last_width[field] = width;
+               userdata->last_height[field] = height;
+       }
+}
+
+void upload_texture(GLuint tex, GLuint width, GLuint height, GLuint stride, bool interlaced_stride, GLenum format, GLintptr offset)
+{
+       if (interlaced_stride) {
+               stride *= 2;
+       }
+       if (global_flags.flush_pbos) {
+               glFlushMappedBufferRange(GL_PIXEL_UNPACK_BUFFER, offset, stride * height);
+               check_error();
+       }
+
+       glBindTexture(GL_TEXTURE_2D, tex);
+       check_error();
+       if (interlaced_stride) {
+               glPixelStorei(GL_UNPACK_ROW_LENGTH, width * 2);
+               check_error();
+       } else {
+               glPixelStorei(GL_UNPACK_ROW_LENGTH, 0);
+               check_error();
+       }
+
+       glTexSubImage2D(GL_TEXTURE_2D, 0, 0, 0, width, height, format, GL_UNSIGNED_BYTE, BUFFER_OFFSET(offset));
+       check_error();
+       glBindTexture(GL_TEXTURE_2D, 0);
+       check_error();
+       glPixelStorei(GL_UNPACK_ROW_LENGTH, 0);
+       check_error();
+}
+
 }  // namespace
 
-void QueueLengthPolicy::update_policy(int queue_length)
+void QueueLengthPolicy::update_policy(unsigned queue_length)
 {
-       if (queue_length < 0) {  // Starvation.
-               if (been_at_safe_point_since_last_starvation && safe_queue_length < 5) {
+       if (queue_length == 0) {  // Starvation.
+               if (been_at_safe_point_since_last_starvation && safe_queue_length < unsigned(global_flags.max_input_queue_frames)) {
                        ++safe_queue_length;
-                       fprintf(stderr, "Card %u: Starvation, increasing safe limit to %u frames\n",
+                       fprintf(stderr, "Card %u: Starvation, increasing safe limit to %u frame(s)\n",
                                card_index, safe_queue_length);
                }
                frames_with_at_least_one = 0;
                been_at_safe_point_since_last_starvation = false;
                return;
        }
-       if (queue_length > 0) {
-               if (queue_length >= int(safe_queue_length)) {
+       if (queue_length >= 1) {
+               if (queue_length >= safe_queue_length) {
                        been_at_safe_point_since_last_starvation = true;
                }
-               if (++frames_with_at_least_one >= 1000 && safe_queue_length > 0) {
+               if (++frames_with_at_least_one >= 1000 && safe_queue_length > 1) {
                        --safe_queue_length;
-                       fprintf(stderr, "Card %u: Spare frames for more than 1000 frames, reducing safe limit to %u frames\n",
+                       fprintf(stderr, "Card %u: Spare frames for more than 1000 frames, reducing safe limit to %u frame(s)\n",
                                card_index, safe_queue_length);
                        frames_with_at_least_one = 0;
                }
@@ -199,6 +252,10 @@ Mixer::Mixer(const QSurfaceFormat &format, unsigned num_cards)
 
        chroma_subsampler.reset(new ChromaSubsampler(resource_pool.get()));
 
+       timecode_renderer.reset(new TimecodeRenderer(resource_pool.get(), global_flags.width, global_flags.height));
+       display_timecode_in_stream = global_flags.display_timecode_in_stream;
+       display_timecode_on_stdout = global_flags.display_timecode_on_stdout;
+
        if (global_flags.enable_alsa_output) {
                alsa.reset(new ALSAOutput(OUTPUT_FREQUENCY, /*num_channels=*/2));
        }
@@ -249,7 +306,7 @@ void Mixer::configure_card(unsigned card_index, CaptureInterface *capture, bool
        if (card->surface == nullptr) {
                card->surface = create_surface_with_same_format(mixer_surface);
        }
-       while (!card->new_frames.empty()) card->new_frames.pop();
+       while (!card->new_frames.empty()) card->new_frames.pop_front();
        card->last_timecode = -1;
        card->capture->configure_card();
 
@@ -297,7 +354,8 @@ void Mixer::set_output_card_internal(int card_index)
                configure_card(card_index, fake_capture, /*is_fake_capture=*/true, card->output.release());
                card->queue_length_policy.reset(card_index);
                card->capture->start_bm_capture();
-               card->output->start_output(bmdModeHD720p5994, pts_int);  // FIXME
+               desired_output_video_mode = output_video_mode = card->output->pick_video_mode(desired_output_video_mode);
+               card->output->start_output(desired_output_video_mode, pts_int);
        }
        output_card_index = card_index;
 }
@@ -412,7 +470,8 @@ void Mixer::bm_frame(unsigned card_index, uint16_t timecode,
                        new_frame.length = frame_length;
                        new_frame.interlaced = false;
                        new_frame.dropped_frames = dropped_frames;
-                       card->new_frames.push(move(new_frame));
+                       new_frame.received_timestamp = video_frame.received_timestamp;
+                       card->new_frames.push_back(move(new_frame));
                        card->new_frames_changed.notify_all();
                }
                return;
@@ -464,69 +523,19 @@ void Mixer::bm_frame(unsigned card_index, uint16_t timecode,
                        } else {
                                field_start_line = video_format.extra_lines_top;
                        }
-
-                       if (userdata->tex_y[field] == 0 ||
-                           userdata->tex_cbcr[field] == 0 ||
-                           video_format.width != userdata->last_width[field] ||
-                           video_format.height != userdata->last_height[field]) {
-                               // We changed resolution since last use of this texture, so we need to create
-                               // a new object. Note that this each card has its own PBOFrameAllocator,
-                               // we don't need to worry about these flip-flopping between resolutions.
-                               glBindTexture(GL_TEXTURE_2D, userdata->tex_cbcr[field]);
-                               check_error();
-                               glTexImage2D(GL_TEXTURE_2D, 0, GL_RG8, cbcr_width, video_format.height, 0, GL_RG, GL_UNSIGNED_BYTE, nullptr);
-                               check_error();
-                               glBindTexture(GL_TEXTURE_2D, userdata->tex_y[field]);
-                               check_error();
-                               glTexImage2D(GL_TEXTURE_2D, 0, GL_R8, video_format.width, video_format.height, 0, GL_RED, GL_UNSIGNED_BYTE, nullptr);
-                               check_error();
-                               userdata->last_width[field] = video_format.width;
-                               userdata->last_height[field] = video_format.height;
-                       }
-
-                       GLuint pbo = userdata->pbo;
-                       check_error();
-                       glBindBuffer(GL_PIXEL_UNPACK_BUFFER, pbo);
-                       check_error();
-
                        size_t field_y_start = y_offset + video_format.width * field_start_line;
                        size_t field_cbcr_start = cbcr_offset + cbcr_width * field_start_line * sizeof(uint16_t);
 
-                       if (global_flags.flush_pbos) {
-                               glFlushMappedBufferRange(GL_PIXEL_UNPACK_BUFFER, field_y_start, video_format.width * video_format.height);
-                               check_error();
-                               glFlushMappedBufferRange(GL_PIXEL_UNPACK_BUFFER, field_cbcr_start, cbcr_width * video_format.height * sizeof(uint16_t));
-                               check_error();
-                       }
+                       ensure_texture_resolution(userdata, field, video_format.width, video_format.height);
 
-                       glBindTexture(GL_TEXTURE_2D, userdata->tex_cbcr[field]);
-                       check_error();
-                       if (interlaced_stride) {
-                               glPixelStorei(GL_UNPACK_ROW_LENGTH, cbcr_width * 2);
-                               check_error();
-                       } else {
-                               glPixelStorei(GL_UNPACK_ROW_LENGTH, 0);
-                               check_error();
-                       }
-                       glTexSubImage2D(GL_TEXTURE_2D, 0, 0, 0, cbcr_width, video_format.height, GL_RG, GL_UNSIGNED_BYTE, BUFFER_OFFSET(field_cbcr_start));
-                       check_error();
-                       glBindTexture(GL_TEXTURE_2D, userdata->tex_y[field]);
-                       check_error();
-                       if (interlaced_stride) {
-                               glPixelStorei(GL_UNPACK_ROW_LENGTH, video_format.width * 2);
-                               check_error();
-                       } else {
-                               glPixelStorei(GL_UNPACK_ROW_LENGTH, 0);
-                               check_error();
-                       }
-                       glTexSubImage2D(GL_TEXTURE_2D, 0, 0, 0, video_format.width, video_format.height, GL_RED, GL_UNSIGNED_BYTE, BUFFER_OFFSET(field_y_start));
-                       check_error();
-                       glBindTexture(GL_TEXTURE_2D, 0);
+                       glBindBuffer(GL_PIXEL_UNPACK_BUFFER, userdata->pbo);
                        check_error();
+
+                       upload_texture(userdata->tex_y[field], video_format.width, video_format.height, video_format.width, interlaced_stride, GL_RED, field_y_start);
+                       upload_texture(userdata->tex_cbcr[field], cbcr_width, video_format.height, cbcr_width * sizeof(uint16_t), interlaced_stride, GL_RG, field_cbcr_start);
+
                        glBindBuffer(GL_PIXEL_UNPACK_BUFFER, 0);
                        check_error();
-                       glPixelStorei(GL_UNPACK_ROW_LENGTH, 0);
-                       check_error();
                };
 
                if (field == 1) {
@@ -551,7 +560,7 @@ void Mixer::bm_frame(unsigned card_index, uint16_t timecode,
                        new_frame.upload_func = upload_func;
                        new_frame.dropped_frames = dropped_frames;
                        new_frame.received_timestamp = video_frame.received_timestamp;  // Ignore the audio timestamp.
-                       card->new_frames.push(move(new_frame));
+                       card->new_frames.push_back(move(new_frame));
                        card->new_frames_changed.notify_all();
                }
        }
@@ -590,13 +599,19 @@ void Mixer::thread_func()
        steady_clock::time_point start, now;
        start = steady_clock::now();
 
-       int frame = 0;
        int stats_dropped_frames = 0;
 
        while (!should_quit) {
                if (desired_output_card_index != output_card_index) {
                        set_output_card_internal(desired_output_card_index);
                }
+               if (output_card_index != -1 &&
+                   desired_output_video_mode != output_video_mode) {
+                       DeckLinkOutput *output = cards[output_card_index].output.get();
+                       output->end_output();
+                       desired_output_video_mode = output_video_mode = output->pick_video_mode(desired_output_video_mode);
+                       output->start_output(desired_output_video_mode, pts_int);
+               }
 
                CaptureCard::NewFrame new_frames[MAX_VIDEO_CARDS];
                bool has_new_frame[MAX_VIDEO_CARDS] = { false };
@@ -657,15 +672,15 @@ void Mixer::thread_func()
 
                int64_t frame_duration = output_frame_info.frame_duration;
                render_one_frame(frame_duration);
-               ++frame;
+               ++frame_num;
                pts_int += frame_duration;
 
                now = steady_clock::now();
                double elapsed = duration<double>(now - start).count();
-               if (frame % 100 == 0) {
+               if (frame_num % 100 == 0) {
                        printf("%d frames (%d dropped) in %.3f seconds = %.1f fps (%.1f ms/frame)",
-                               frame, stats_dropped_frames, elapsed, frame / elapsed,
-                               1e3 * elapsed / frame);
+                               frame_num, stats_dropped_frames, elapsed, frame_num / elapsed,
+                               1e3 * elapsed / frame_num);
                //      chain->print_phase_timing();
 
                        // Check our memory usage, to see if we are close to our mlockall()
@@ -702,7 +717,7 @@ void Mixer::thread_func()
 
 
                if (should_cut.exchange(false)) {  // Test and clear.
-                       video_encoder->do_cut(frame);
+                       video_encoder->do_cut(frame_num);
                }
 
 #if 0
@@ -731,6 +746,47 @@ bool Mixer::input_card_is_master_clock(unsigned card_index, unsigned master_card
        return (card_index == master_card_index);
 }
 
+void Mixer::trim_queue(CaptureCard *card, unsigned card_index)
+{
+       // Count the number of frames in the queue, including any frames
+       // we dropped. It's hard to know exactly how we should deal with
+       // dropped (corrupted) input frames; they don't help our goal of
+       // avoiding starvation, but they still add to the problem of latency.
+       // Since dropped frames is going to mean a bump in the signal anyway,
+       // we err on the side of having more stable latency instead.
+       unsigned queue_length = 0;
+       for (const CaptureCard::NewFrame &frame : card->new_frames) {
+               queue_length += frame.dropped_frames + 1;
+       }
+       card->queue_length_policy.update_policy(queue_length);
+
+       // If needed, drop frames until the queue is below the safe limit.
+       // We prefer to drop from the head, because all else being equal,
+       // we'd like more recent frames (less latency).
+       unsigned dropped_frames = 0;
+       while (queue_length > card->queue_length_policy.get_safe_queue_length()) {
+               assert(!card->new_frames.empty());
+               assert(queue_length > card->new_frames.front().dropped_frames);
+               queue_length -= card->new_frames.front().dropped_frames;
+
+               if (queue_length <= card->queue_length_policy.get_safe_queue_length()) {
+                       // No need to drop anything.
+                       break;
+               }
+
+               card->new_frames.pop_front();
+               card->new_frames_changed.notify_all();
+               --queue_length;
+               ++dropped_frames;
+       }
+
+       if (dropped_frames > 0) {
+               fprintf(stderr, "Card %u dropped %u frame(s) to keep latency down.\n",
+                       card_index, dropped_frames);
+       }
+}
+
+
 Mixer::OutputFrameInfo Mixer::get_one_frame_from_each_card(unsigned master_card_index, bool master_card_is_output, CaptureCard::NewFrame new_frames[MAX_VIDEO_CARDS], bool has_new_frame[MAX_VIDEO_CARDS])
 {
        OutputFrameInfo output_frame_info;
@@ -765,28 +821,20 @@ start:
 
        for (unsigned card_index = 0; card_index < num_cards; ++card_index) {
                CaptureCard *card = &cards[card_index];
-               if (card->new_frames.empty()) {
-                       assert(!input_card_is_master_clock(card_index, master_card_index));
-                       card->queue_length_policy.update_policy(-1);
-                       continue;
-               }
-               new_frames[card_index] = move(card->new_frames.front());
-               has_new_frame[card_index] = true;
-               card->new_frames.pop();
-               card->new_frames_changed.notify_all();
-
                if (input_card_is_master_clock(card_index, master_card_index)) {
                        // We don't use the queue length policy for the master card,
                        // but we will if it stops being the master. Thus, clear out
                        // the policy in case we switch in the future.
                        card->queue_length_policy.reset(card_index);
+                       assert(!card->new_frames.empty());
                } else {
-                       // If we have excess frames compared to the policy for this card,
-                       // drop frames from the head.
-                       card->queue_length_policy.update_policy(card->new_frames.size());
-                       while (card->new_frames.size() > card->queue_length_policy.get_safe_queue_length()) {
-                               card->new_frames.pop();
-                       }
+                       trim_queue(card, card_index);
+               }
+               if (!card->new_frames.empty()) {
+                       new_frames[card_index] = move(card->new_frames.front());
+                       has_new_frame[card_index] = true;
+                       card->new_frames.pop_front();
+                       card->new_frames_changed.notify_all();
                }
        }
 
@@ -888,6 +936,12 @@ void Mixer::schedule_audio_resampling_tasks(unsigned dropped_frames, int num_sam
 
 void Mixer::render_one_frame(int64_t duration)
 {
+       // Determine the time code for this frame before we start rendering.
+       string timecode_text = timecode_renderer->get_timecode_text(double(pts_int) / TIMEBASE, frame_num);
+       if (display_timecode_on_stdout) {
+               printf("Timecode: '%s'\n", timecode_text.c_str());
+       }
+
        // Get the main chain from the theme, and set its state immediately.
        Theme::Chain theme_main_chain = theme->get_chain(0, pts(), global_flags.width, global_flags.height, input_state);
        EffectChain *chain = theme_main_chain.chain;
@@ -904,6 +958,12 @@ void Mixer::render_one_frame(int64_t duration)
        GLuint fbo = resource_pool->create_fbo(y_tex, cbcr_full_tex, rgba_tex);
        check_error();
        chain->render_to_fbo(fbo, global_flags.width, global_flags.height);
+
+       if (display_timecode_in_stream) {
+               // Render the timecode on top.
+               timecode_renderer->render_timecode(fbo, timecode_text);
+       }
+
        resource_pool->release_fbo(fbo);
 
        chroma_subsampler->subsample_chroma(cbcr_full_tex, global_flags.width, global_flags.height, cbcr_tex);
@@ -1034,6 +1094,13 @@ void Mixer::start_mode_scanning(unsigned card_index)
        last_mode_scan_change[card_index] = steady_clock::now();
 }
 
+map<uint32_t, bmusb::VideoMode> Mixer::get_available_output_video_modes() const
+{
+       assert(desired_output_card_index != -1);
+       unique_lock<mutex> lock(card_mutex);
+       return cards[desired_output_card_index].output->get_available_video_modes();
+}
+
 Mixer::OutputChannel::~OutputChannel()
 {
        if (has_current_frame) {