X-Git-Url: https://git.sesse.net/?a=blobdiff_plain;f=nageru%2Fmixer.h;h=3ad74f8dd7a878b2f18c9f490104ff96830ca80c;hb=02a4d8ccf0b7e29e857e46208adb4938f78c45df;hp=b4ed76f4575ff0403b4865a470af8925d548ddd1;hpb=eeda8995329601f9f4e35047358400833eeae68e;p=nageru diff --git a/nageru/mixer.h b/nageru/mixer.h index b4ed76f..3ad74f8 100644 --- a/nageru/mixer.h +++ b/nageru/mixer.h @@ -23,11 +23,13 @@ #include #include +#include #include #include "audio_mixer.h" #include "bmusb/bmusb.h" #include "defs.h" +#include "ffmpeg_capture.h" #include "shared/httpd.h" #include "input_state.h" #include "libusb.h" @@ -130,14 +132,12 @@ private: class QueueLengthPolicy { public: QueueLengthPolicy() {} - void reset(unsigned card_index) { - this->card_index = card_index; - } void register_metrics(const std::vector> &labels); void unregister_metrics(const std::vector> &labels); // Call after picking out a frame, so 0 means starvation. + // Note that the policy has no memory; everything is given in as parameters. void update_policy(std::chrono::steady_clock::time_point now, std::chrono::steady_clock::time_point expected_next_frame, int64_t input_frame_duration, @@ -147,7 +147,6 @@ public: unsigned get_safe_queue_length() const { return safe_queue_length; } private: - unsigned card_index; // For debugging and metrics only. unsigned safe_queue_length = 0; // Can never go below zero. // Metrics. @@ -157,7 +156,7 @@ private: class Mixer { public: // The surface format is used for offscreen destinations for OpenGL contexts we need. - Mixer(const QSurfaceFormat &format, unsigned num_cards); + Mixer(const QSurfaceFormat &format); ~Mixer(); void start(); void quit(); @@ -250,14 +249,14 @@ public: return theme->get_channel_color(channel); } - int get_channel_signal(unsigned channel) const + int map_channel_to_signal(unsigned channel) const { - return theme->get_channel_signal(channel); + return theme->map_channel_to_signal(channel); } - int map_signal(unsigned channel) + int map_signal_to_card(int signal) { - return theme->map_signal(channel); + return theme->map_signal_to_card(signal); } unsigned get_master_clock() const @@ -288,6 +287,11 @@ public: theme->set_wb(channel, r, g, b); } + std::string format_status_line(const std::string &disk_space_left_text, double file_length_seconds) + { + return theme->format_status_line(disk_space_left_text, file_length_seconds); + } + // Note: You can also get this through the global variable global_audio_mixer. AudioMixer *get_audio_mixer() { return audio_mixer.get(); } const AudioMixer *get_audio_mixer() const { return audio_mixer.get(); } @@ -297,10 +301,8 @@ public: should_cut = true; } - unsigned get_num_cards() const { return num_cards; } - std::string get_card_description(unsigned card_index) const { - assert(card_index < num_cards); + assert(card_index < MAX_VIDEO_CARDS); return cards[card_index].capture->get_description(); } @@ -310,7 +312,7 @@ public: // the card's actual name. std::string get_output_card_description(unsigned card_index) const { assert(card_can_be_used_as_output(card_index)); - assert(card_index < num_cards); + assert(card_index < MAX_VIDEO_CARDS); if (cards[card_index].parked_capture) { return cards[card_index].parked_capture->get_description(); } else { @@ -319,59 +321,87 @@ public: } bool card_can_be_used_as_output(unsigned card_index) const { - assert(card_index < num_cards); - return cards[card_index].output != nullptr; + assert(card_index < MAX_VIDEO_CARDS); + return cards[card_index].output != nullptr && cards[card_index].capture != nullptr; + } + + bool card_is_cef(unsigned card_index) const { + assert(card_index < MAX_VIDEO_CARDS); + return cards[card_index].type == CardType::CEF_INPUT; } bool card_is_ffmpeg(unsigned card_index) const { - assert(card_index < num_cards + num_video_inputs); - return cards[card_index].type == CardType::FFMPEG_INPUT; + assert(card_index < MAX_VIDEO_CARDS); + if (cards[card_index].type != CardType::FFMPEG_INPUT) { + return false; + } +#ifdef HAVE_SRT + // SRT inputs are more like regular inputs than FFmpeg inputs, + // so show them as such. (This allows the user to right-click + // to select a different input.) + return static_cast(cards[card_index].capture.get())->get_srt_sock() == -1; +#else + return true; +#endif + } + + bool card_is_active(unsigned card_index) const { + assert(card_index < MAX_VIDEO_CARDS); + std::lock_guard lock(card_mutex); + return cards[card_index].capture != nullptr; + } + + void force_card_active(unsigned card_index) + { + // handle_hotplugged_cards() will pick this up. + std::lock_guard lock(card_mutex); + cards[card_index].force_active = true; } std::map get_available_video_modes(unsigned card_index) const { - assert(card_index < num_cards); + assert(card_index < MAX_VIDEO_CARDS); return cards[card_index].capture->get_available_video_modes(); } uint32_t get_current_video_mode(unsigned card_index) const { - assert(card_index < num_cards); + assert(card_index < MAX_VIDEO_CARDS); return cards[card_index].capture->get_current_video_mode(); } void set_video_mode(unsigned card_index, uint32_t mode) { - assert(card_index < num_cards); + assert(card_index < MAX_VIDEO_CARDS); cards[card_index].capture->set_video_mode(mode); } void start_mode_scanning(unsigned card_index); std::map get_available_video_inputs(unsigned card_index) const { - assert(card_index < num_cards); + assert(card_index < MAX_VIDEO_CARDS); return cards[card_index].capture->get_available_video_inputs(); } uint32_t get_current_video_input(unsigned card_index) const { - assert(card_index < num_cards); + assert(card_index < MAX_VIDEO_CARDS); return cards[card_index].capture->get_current_video_input(); } void set_video_input(unsigned card_index, uint32_t input) { - assert(card_index < num_cards); + assert(card_index < MAX_VIDEO_CARDS); cards[card_index].capture->set_video_input(input); } std::map get_available_audio_inputs(unsigned card_index) const { - assert(card_index < num_cards); + assert(card_index < MAX_VIDEO_CARDS); return cards[card_index].capture->get_available_audio_inputs(); } uint32_t get_current_audio_input(unsigned card_index) const { - assert(card_index < num_cards); + assert(card_index < MAX_VIDEO_CARDS); return cards[card_index].capture->get_current_audio_input(); } void set_audio_input(unsigned card_index, uint32_t input) { - assert(card_index < num_cards); + assert(card_index < MAX_VIDEO_CARDS); cards[card_index].capture->set_audio_input(input); } @@ -413,7 +443,7 @@ public: return httpd.get_num_connected_clients(); } - std::vector get_theme_menu() { return theme->get_theme_menu(); } + Theme::MenuEntry *get_theme_menu() { return theme->get_theme_menu(); } void theme_menu_entry_clicked(int lua_ref) { return theme->theme_menu_entry_clicked(lua_ref); } @@ -427,17 +457,15 @@ public: private: struct CaptureCard; - enum class CardType { - LIVE_CARD, - FAKE_CAPTURE, - FFMPEG_INPUT, - CEF_INPUT, - }; - void configure_card(unsigned card_index, bmusb::CaptureInterface *capture, CardType card_type, DeckLinkOutput *output); + void configure_card(unsigned card_index, bmusb::CaptureInterface *capture, CardType card_type, DeckLinkOutput *output, bool is_srt_card); void set_output_card_internal(int card_index); // Should only be called from the mixer thread. void bm_frame(unsigned card_index, uint16_t timecode, bmusb::FrameAllocator::Frame video_frame, size_t video_offset, bmusb::VideoFormat video_format, bmusb::FrameAllocator::Frame audio_frame, size_t audio_offset, bmusb::AudioFormat audio_format); + void upload_texture_for_frame( + int field, bmusb::VideoFormat video_format, + size_t y_offset, size_t cbcr_offset, size_t video_offset, + PBOFrameAllocator::Userdata *userdata); void bm_hotplug_add(libusb_device *dev); void bm_hotplug_remove(unsigned card_index); void place_rectangle(movit::Effect *resample_effect, movit::Effect *padding_effect, float x0, float y0, float x1, float y1); @@ -448,15 +476,18 @@ private: void render_one_frame(int64_t duration); void audio_thread_func(); void release_display_frame(DisplayFrame *frame); +#ifdef HAVE_SRT + void start_srt(); +#endif double pts() { return double(pts_int) / TIMEBASE; } void trim_queue(CaptureCard *card, size_t safe_queue_length); std::pair get_channels_json(); std::pair get_channel_color_http(unsigned channel_idx); HTTPD httpd; - unsigned num_cards, num_video_inputs, num_html_inputs = 0; + unsigned num_video_inputs, num_html_inputs = 0; - QSurface *mixer_surface, *h264_encoder_surface, *decklink_output_surface; + QSurface *mixer_surface, *h264_encoder_surface, *decklink_output_surface, *image_update_surface; std::unique_ptr resource_pool; std::unique_ptr theme; std::atomic audio_source_channel{0}; @@ -495,11 +526,24 @@ private: // frame rate is integer, will always stay zero. unsigned fractional_samples = 0; + // Monotonic counter that lets us know which slot was last turned into + // a fake capture. Used for SRT re-plugging. + unsigned fake_capture_counter = 0; + mutable std::mutex card_mutex; bool has_bmusb_thread = false; struct CaptureCard { + // If nullptr, the card is inactive, and will be hidden in the UI. + // Only fake capture cards can be inactive. std::unique_ptr capture; + // If true, card must always be active (typically because it's one of the + // first cards, or because the theme has explicitly asked for it). + bool force_active = false; bool is_fake_capture; + // If is_fake_capture is true, contains a monotonic timer value for when + // it was last changed. Otherwise undefined. Used for SRT re-plugging. + int fake_capture_counter; + std::string last_srt_stream_id = ""; // Used for SRT re-plugging. CardType type; std::unique_ptr output; @@ -529,21 +573,22 @@ private: int64_t length; // In TIMEBASE units. bool interlaced; unsigned field; // Which field (0 or 1) of the frame to use. Always 0 for progressive. - std::function upload_func; // Needs to be called to actually upload the texture to OpenGL. + bool texture_uploaded = false; unsigned dropped_frames = 0; // Number of dropped frames before this one. std::chrono::steady_clock::time_point received_timestamp = std::chrono::steady_clock::time_point::min(); + movit::RGBTriplet neutral_color{1.0f, 1.0f, 1.0f}; - // Used for MJPEG encoding. (upload_func packs everything it needs - // into the functor, but would otherwise also use these.) + // Used for MJPEG encoding, and texture upload. // width=0 or height=0 means a broken frame, ie., do not upload. bmusb::VideoFormat video_format; - size_t y_offset, cbcr_offset; + size_t video_offset, y_offset, cbcr_offset; }; std::deque new_frames; std::condition_variable new_frames_changed; // Set whenever new_frames is changed. - QueueLengthPolicy queue_length_policy; // Refers to the "new_frames" queue. + std::vector new_raw_audio; + int last_timecode = -1; // Unwrapped. JitterHistory jitter_history; @@ -565,10 +610,58 @@ private: std::atomic metric_input_frame_rate_nom{-1}; std::atomic metric_input_frame_rate_den{-1}; std::atomic metric_input_sample_rate_hz{-1}; + + // SRT metrics. + std::atomic metric_srt_uptime_seconds{0.0 / 0.0}; + std::atomic metric_srt_send_duration_seconds{0.0 / 0.0}; + std::atomic metric_srt_sent_bytes{-1}; + std::atomic metric_srt_received_bytes{-1}; + std::atomic metric_srt_sent_packets_normal{-1}; + std::atomic metric_srt_received_packets_normal{-1}; + std::atomic metric_srt_sent_packets_lost{-1}; + std::atomic metric_srt_received_packets_lost{-1}; + std::atomic metric_srt_sent_packets_retransmitted{-1}; + std::atomic metric_srt_sent_bytes_retransmitted{-1}; + std::atomic metric_srt_sent_packets_ack{-1}; + std::atomic metric_srt_received_packets_ack{-1}; + std::atomic metric_srt_sent_packets_nak{-1}; + std::atomic metric_srt_received_packets_nak{-1}; + std::atomic metric_srt_sent_packets_dropped{-1}; + std::atomic metric_srt_received_packets_dropped{-1}; + std::atomic metric_srt_sent_bytes_dropped{-1}; + std::atomic metric_srt_received_bytes_dropped{-1}; + std::atomic metric_srt_received_packets_undecryptable{-1}; + std::atomic metric_srt_received_bytes_undecryptable{-1}; + + std::atomic metric_srt_filter_received_extra_packets{-1}; + std::atomic metric_srt_filter_received_rebuilt_packets{-1}; + std::atomic metric_srt_filter_received_lost_packets{-1}; + + std::atomic metric_srt_packet_sending_period_seconds{0.0 / 0.0}; + std::atomic metric_srt_flow_window_packets{-1}; + std::atomic metric_srt_congestion_window_packets{-1}; + std::atomic metric_srt_flight_size_packets{-1}; + std::atomic metric_srt_rtt_seconds{0.0 / 0.0}; + std::atomic metric_srt_estimated_bandwidth_bits_per_second{0.0 / 0.0}; + std::atomic metric_srt_bandwidth_ceiling_bits_per_second{0.0 / 0.0}; + std::atomic metric_srt_send_buffer_available_bytes{-1}; + std::atomic metric_srt_receive_buffer_available_bytes{-1}; + std::atomic metric_srt_mss_bytes{-1}; + std::atomic metric_srt_sender_unacked_packets{-1}; + std::atomic metric_srt_sender_unacked_bytes{-1}; + std::atomic metric_srt_sender_unacked_timespan_seconds{0.0 / 0.0}; + std::atomic metric_srt_sender_delivery_delay_seconds{0.0 / 0.0}; + std::atomic metric_srt_receiver_unacked_packets{-1}; + std::atomic metric_srt_receiver_unacked_bytes{-1}; + std::atomic metric_srt_receiver_unacked_timespan_seconds{0.0 / 0.0}; + std::atomic metric_srt_receiver_delivery_delay_seconds{0.0 / 0.0}; + std::atomic metric_srt_filter_sent_packets{-1}; + }; JitterHistory output_jitter_history; CaptureCard cards[MAX_VIDEO_CARDS]; // Protected by . YCbCrInterpretation ycbcr_interpretation[MAX_VIDEO_CARDS]; // Protected by . + movit::RGBTriplet last_received_neutral_color[MAX_VIDEO_CARDS]; // Used by the mixer thread only. Constructor-initialiezd. std::unique_ptr audio_mixer; // Same as global_audio_mixer (see audio_mixer.h). bool input_card_is_master_clock(unsigned card_index, unsigned master_card_index) const; struct OutputFrameInfo { @@ -578,7 +671,14 @@ private: bool is_preroll; std::chrono::steady_clock::time_point frame_timestamp; }; - OutputFrameInfo 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 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], std::vector raw_audio[MAX_VIDEO_CARDS]); + +#ifdef HAVE_SRT + void update_srt_stats(int srt_sock, Mixer::CaptureCard *card); +#endif + + std::string description_for_card(unsigned card_index); + static bool is_srt_card(const CaptureCard *card); InputState input_state; @@ -586,6 +686,9 @@ private: // Protected by its own mutex. std::mutex hotplug_mutex; std::vector hotplugged_cards; +#ifdef HAVE_SRT + std::vector hotplugged_srt_cards; +#endif class OutputChannel { public: @@ -618,6 +721,9 @@ private: std::thread mixer_thread; std::thread audio_thread; +#ifdef HAVE_SRT + std::thread srt_thread; +#endif std::atomic should_quit{false}; std::atomic should_cut{false};