gst-plugins-rs/utils/fallbackswitch/examples/gtk_fallbackswitch.rs
2022-10-22 19:50:24 +03:00

210 lines
6.5 KiB
Rust

// Copyright (C) 2019 Sebastian Dröge <sebastian@centricular.com>
//
// This Source Code Form is subject to the terms of the Mozilla Public License, v2.0.
// If a copy of the MPL was not distributed with this file, You can obtain one at
// <https://mozilla.org/MPL/2.0/>.
//
// SPDX-License-Identifier: MPL-2.0
use gio::prelude::*;
use gst::glib;
use gst::prelude::*;
use gtk::prelude::*;
use gtk::Inhibit;
use std::cell::RefCell;
const MAIN_PIPELINE: &str = "videotestsrc is-live=true pattern=ball";
const FALLBACK_PIPELINE: &str = "videotestsrc is-live=true pattern=snow";
//const MAIN_PIPELINE: &str = "videotestsrc is-live=true pattern=ball ! x264enc tune=zerolatency";
//const FALLBACK_PIPELINE: &str = "videotestsrc is-live=true pattern=snow ! x264enc tune=zerolatency";
fn create_pipeline() -> (gst::Pipeline, gst::Pad, gst::Element) {
let pipeline = gst::Pipeline::default();
let video_src = gst::parse_bin_from_description(MAIN_PIPELINE, true)
.unwrap()
.upcast();
let fallback_video_src = gst::parse_bin_from_description(FALLBACK_PIPELINE, true)
.unwrap()
.upcast();
let fallbackswitch = gst::ElementFactory::make("fallbackswitch")
.property("timeout", gst::ClockTime::SECOND)
.build()
.unwrap();
let decodebin = gst::ElementFactory::make("decodebin").build().unwrap();
let videoconvert = gst::ElementFactory::make("videoconvert").build().unwrap();
let videoconvert_clone = videoconvert.clone();
decodebin.connect_pad_added(move |_, pad| {
let caps = pad.current_caps().unwrap();
let s = caps.structure(0).unwrap();
let sinkpad = videoconvert_clone.static_pad("sink").unwrap();
if s.name() == "video/x-raw" && !sinkpad.is_linked() {
pad.link(&sinkpad).unwrap();
}
});
let video_sink = gst::ElementFactory::make("gtk4paintablesink")
.build()
.unwrap();
pipeline
.add_many(&[
&video_src,
&fallback_video_src,
&fallbackswitch,
&decodebin,
&videoconvert,
&video_sink,
])
.unwrap();
/* The first pad requested will be automatically preferred */
video_src
.link_pads(Some("src"), &fallbackswitch, Some("sink_%u"))
.unwrap();
fallback_video_src
.link_pads(Some("src"), &fallbackswitch, Some("sink_%u"))
.unwrap();
fallbackswitch
.link_pads(Some("src"), &decodebin, Some("sink"))
.unwrap();
videoconvert
.link_pads(Some("src"), &video_sink, Some("sink"))
.unwrap();
(pipeline, video_src.static_pad("src").unwrap(), video_sink)
}
fn create_ui(app: &gtk::Application) {
let (pipeline, video_src_pad, video_sink) = create_pipeline();
let window = gtk::ApplicationWindow::new(app);
window.set_default_size(320, 240);
let vbox = gtk::Box::new(gtk::Orientation::Vertical, 0);
let picture = gtk::Picture::new();
let paintable = video_sink.property::<gtk::gdk::Paintable>("paintable");
picture.set_paintable(Some(&paintable));
vbox.append(&picture);
let position_label = gtk::Label::new(Some("Position: 00:00:00"));
vbox.append(&position_label);
let drop_button = gtk::ToggleButton::with_label("Drop Signal");
vbox.append(&drop_button);
window.set_child(Some(&vbox));
window.show();
app.add_window(&window);
let video_sink_weak = video_sink.downgrade();
let timeout_id = glib::timeout_add_local(std::time::Duration::from_millis(100), move || {
let video_sink = match video_sink_weak.upgrade() {
Some(video_sink) => video_sink,
None => return glib::Continue(true),
};
let position = video_sink
.query_position::<gst::ClockTime>()
.unwrap_or(gst::ClockTime::ZERO);
position_label.set_text(&format!("Position: {:.1}", position));
glib::Continue(true)
});
let video_src_pad_weak = video_src_pad.downgrade();
let drop_id = RefCell::new(None);
drop_button.connect_toggled(move |drop_button| {
let video_src_pad = match video_src_pad_weak.upgrade() {
Some(video_src_pad) => video_src_pad,
None => return,
};
let drop = drop_button.is_active();
if drop {
let mut drop_id = drop_id.borrow_mut();
if drop_id.is_none() {
*drop_id = video_src_pad
.add_probe(gst::PadProbeType::BUFFER, |_, _| gst::PadProbeReturn::Drop);
}
} else if let Some(drop_id) = drop_id.borrow_mut().take() {
video_src_pad.remove_probe(drop_id);
}
});
let app_weak = app.downgrade();
window.connect_close_request(move |_| {
let app = match app_weak.upgrade() {
Some(app) => app,
None => return Inhibit(false),
};
app.quit();
Inhibit(false)
});
let bus = pipeline.bus().unwrap();
let app_weak = app.downgrade();
bus.add_watch_local(move |_, msg| {
use gst::MessageView;
let app = match app_weak.upgrade() {
Some(app) => app,
None => return glib::Continue(false),
};
match msg.view() {
MessageView::Eos(..) => app.quit(),
MessageView::Error(err) => {
println!(
"Error from {:?}: {} ({:?})",
msg.src().map(|s| s.path_string()),
err.error(),
err.debug()
);
app.quit();
}
_ => (),
};
glib::Continue(true)
})
.expect("Failed to add bus watch");
pipeline.set_state(gst::State::Playing).unwrap();
// Pipeline reference is owned by the closure below, so will be
// destroyed once the app is destroyed
let timeout_id = RefCell::new(Some(timeout_id));
app.connect_shutdown(move |_| {
pipeline.set_state(gst::State::Null).unwrap();
bus.remove_watch().unwrap();
if let Some(timeout_id) = timeout_id.borrow_mut().take() {
timeout_id.remove();
}
});
}
fn main() {
gst::init().unwrap();
gtk::init().unwrap();
gstfallbackswitch::plugin_register_static().expect("Failed to register fallbackswitch plugin");
gstgtk4::plugin_register_static().expect("Failed to register gtk4paintablesink plugin");
let app = gtk::Application::new(None, gio::ApplicationFlags::FLAGS_NONE);
app.connect_activate(create_ui);
app.run();
}