2020-12-20 15:09:22 +00:00
|
|
|
use gst::element_error;
|
2017-11-19 03:18:34 +00:00
|
|
|
use gst::prelude::*;
|
|
|
|
|
|
|
|
#[path = "../examples-common.rs"]
|
|
|
|
mod examples_common;
|
|
|
|
|
|
|
|
use std::env;
|
|
|
|
|
2020-05-02 23:23:38 +00:00
|
|
|
use anyhow::Error;
|
|
|
|
use derive_more::{Display, Error};
|
2017-11-19 03:18:34 +00:00
|
|
|
|
2020-05-02 23:23:38 +00:00
|
|
|
#[derive(Debug, Display, Error)]
|
|
|
|
#[display(fmt = "No such pad {} in {}", _0, _1)]
|
2017-11-19 03:18:34 +00:00
|
|
|
struct NoSuchPad(&'static str, String);
|
|
|
|
|
2020-05-02 23:23:38 +00:00
|
|
|
#[derive(Debug, Display, Error)]
|
|
|
|
#[display(fmt = "Usage: {} URI FEC_PERCENTAGE", _0)]
|
|
|
|
struct UsageError(#[error(not(source))] String);
|
2017-11-19 03:18:34 +00:00
|
|
|
|
2020-05-02 23:23:38 +00:00
|
|
|
#[derive(Debug, Display, Error)]
|
|
|
|
#[display(fmt = "Received error from {}: {} (debug: {:?})", src, error, debug)]
|
2017-11-19 03:18:34 +00:00
|
|
|
struct ErrorMessage {
|
|
|
|
src: String,
|
|
|
|
error: String,
|
|
|
|
debug: Option<String>,
|
2020-05-02 23:23:38 +00:00
|
|
|
source: glib::Error,
|
2017-11-19 03:18:34 +00:00
|
|
|
}
|
|
|
|
|
2021-04-20 10:23:24 +00:00
|
|
|
fn static_pad(element: &gst::Element, pad_name: &'static str) -> Result<gst::Pad, Error> {
|
2021-04-20 10:24:17 +00:00
|
|
|
match element.static_pad(pad_name) {
|
2017-11-19 03:18:34 +00:00
|
|
|
Some(pad) => Ok(pad),
|
|
|
|
None => {
|
2021-04-11 19:39:50 +00:00
|
|
|
let element_name = element.name();
|
2018-12-09 16:09:20 +00:00
|
|
|
Err(Error::from(NoSuchPad(pad_name, element_name.to_string())))
|
2017-11-19 03:18:34 +00:00
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2021-04-20 10:23:24 +00:00
|
|
|
fn request_pad(element: &gst::Element, pad_name: &'static str) -> Result<gst::Pad, Error> {
|
2021-04-20 13:12:36 +00:00
|
|
|
match element.request_pad_simple(pad_name) {
|
2017-11-19 03:18:34 +00:00
|
|
|
Some(pad) => Ok(pad),
|
|
|
|
None => {
|
2021-04-11 19:39:50 +00:00
|
|
|
let element_name = element.name();
|
2018-12-09 16:09:20 +00:00
|
|
|
Err(Error::from(NoSuchPad(pad_name, element_name.to_string())))
|
2017-11-19 03:18:34 +00:00
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
fn connect_decodebin_pad(src_pad: &gst::Pad, sink: &gst::Element) -> Result<(), Error> {
|
2021-07-30 10:19:24 +00:00
|
|
|
let sinkpad = static_pad(sink, "sink")?;
|
2019-01-08 16:13:37 +00:00
|
|
|
src_pad.link(&sinkpad)?;
|
2017-11-19 03:18:34 +00:00
|
|
|
|
|
|
|
Ok(())
|
|
|
|
}
|
|
|
|
|
|
|
|
fn make_fec_encoder(fec_percentage: u32) -> Result<gst::Element, Error> {
|
2022-10-19 11:13:57 +00:00
|
|
|
let fecenc = gst::ElementFactory::make("rtpulpfecenc")
|
|
|
|
.property("pt", 100u32)
|
|
|
|
.property("multipacket", true)
|
|
|
|
.property("percentage", fec_percentage)
|
|
|
|
.build()?;
|
2017-11-19 03:18:34 +00:00
|
|
|
|
|
|
|
Ok(fecenc)
|
|
|
|
}
|
|
|
|
|
|
|
|
fn example_main() -> Result<(), Error> {
|
|
|
|
gst::init()?;
|
|
|
|
|
|
|
|
let args: Vec<_> = env::args().collect();
|
|
|
|
|
|
|
|
if args.len() != 3 {
|
|
|
|
return Err(Error::from(UsageError(args[0].clone())));
|
|
|
|
}
|
|
|
|
|
|
|
|
let uri = &args[1];
|
|
|
|
let fec_percentage = args[2].parse::<u32>()?;
|
|
|
|
|
2022-10-19 11:13:57 +00:00
|
|
|
let video_caps = gst::Caps::builder("video/x-raw").build();
|
|
|
|
|
2022-10-22 14:24:21 +00:00
|
|
|
let pipeline = gst::Pipeline::default();
|
2022-10-19 11:13:57 +00:00
|
|
|
let src = gst::ElementFactory::make("uridecodebin")
|
|
|
|
.property_from_str("pattern", "ball")
|
|
|
|
.property("expose-all-streams", false)
|
|
|
|
.property("caps", video_caps)
|
|
|
|
.property("uri", uri)
|
|
|
|
.build()?;
|
|
|
|
let conv = gst::ElementFactory::make("videoconvert").build()?;
|
|
|
|
let q1 = gst::ElementFactory::make("queue").build()?;
|
|
|
|
let enc = gst::ElementFactory::make("vp8enc")
|
|
|
|
.property("keyframe-max-dist", 30i32)
|
|
|
|
.property("threads", 12i32)
|
|
|
|
.property("cpu-used", -16i32)
|
|
|
|
.property("deadline", 1i64)
|
|
|
|
.property_from_str("error-resilient", "default")
|
|
|
|
.build()?;
|
|
|
|
let q2 = gst::ElementFactory::make("queue").build()?;
|
|
|
|
let pay = gst::ElementFactory::make("rtpvp8pay").build()?;
|
|
|
|
let rtpbin = gst::ElementFactory::make("rtpbin").build()?;
|
|
|
|
let sink = gst::ElementFactory::make("udpsink")
|
|
|
|
.property("host", "127.0.0.1")
|
|
|
|
.property("sync", true)
|
|
|
|
.build()?;
|
2017-11-19 03:18:34 +00:00
|
|
|
|
|
|
|
pipeline.add_many(&[&src, &conv, &q1, &enc, &q2, &pay, &rtpbin, &sink])?;
|
|
|
|
|
|
|
|
conv.link(&q1)?;
|
|
|
|
q1.link(&enc)?;
|
|
|
|
enc.link(&pay)?;
|
|
|
|
pay.link(&q2)?;
|
|
|
|
|
|
|
|
rtpbin.connect("request-fec-encoder", false, move |values| {
|
2019-08-11 07:33:34 +00:00
|
|
|
let rtpbin = values[0]
|
|
|
|
.get::<gst::Element>()
|
2021-04-20 07:19:02 +00:00
|
|
|
.expect("rtpbin \"request-fec-encoder\" signal values[0]");
|
2017-11-19 03:18:34 +00:00
|
|
|
|
|
|
|
match make_fec_encoder(fec_percentage) {
|
|
|
|
Ok(elem) => Some(elem.to_value()),
|
|
|
|
Err(err) => {
|
2020-12-20 15:09:22 +00:00
|
|
|
element_error!(
|
2017-11-19 03:18:34 +00:00
|
|
|
rtpbin,
|
|
|
|
gst::LibraryError::Failed,
|
|
|
|
("Failed to make FEC encoder"),
|
|
|
|
["{}", err]
|
|
|
|
);
|
|
|
|
None
|
|
|
|
}
|
|
|
|
}
|
2021-11-06 17:09:27 +00:00
|
|
|
});
|
2017-11-19 03:18:34 +00:00
|
|
|
|
2021-04-20 10:24:17 +00:00
|
|
|
let srcpad = static_pad(&q2, "src")?;
|
|
|
|
let sinkpad = request_pad(&rtpbin, "send_rtp_sink_0")?;
|
2019-01-08 16:13:37 +00:00
|
|
|
srcpad.link(&sinkpad)?;
|
2017-11-19 03:18:34 +00:00
|
|
|
|
2021-04-20 10:24:17 +00:00
|
|
|
let srcpad = static_pad(&rtpbin, "send_rtp_src_0")?;
|
|
|
|
let sinkpad = static_pad(&sink, "sink")?;
|
2019-01-08 16:13:37 +00:00
|
|
|
srcpad.link(&sinkpad)?;
|
2017-11-19 03:18:34 +00:00
|
|
|
|
2019-12-22 07:59:23 +00:00
|
|
|
src.connect_pad_added(
|
2021-07-30 10:19:24 +00:00
|
|
|
move |decodebin, src_pad| match connect_decodebin_pad(src_pad, &conv) {
|
2017-11-19 03:18:34 +00:00
|
|
|
Ok(_) => (),
|
|
|
|
Err(err) => {
|
2020-12-20 15:09:22 +00:00
|
|
|
element_error!(
|
2017-11-19 03:18:34 +00:00
|
|
|
decodebin,
|
|
|
|
gst::LibraryError::Failed,
|
|
|
|
("Failed to link decodebin srcpad"),
|
|
|
|
["{}", err]
|
|
|
|
);
|
|
|
|
}
|
2019-12-22 07:59:23 +00:00
|
|
|
},
|
|
|
|
);
|
2017-11-19 03:18:34 +00:00
|
|
|
|
|
|
|
let bus = pipeline
|
2021-04-11 19:39:50 +00:00
|
|
|
.bus()
|
2017-11-19 03:18:34 +00:00
|
|
|
.expect("Pipeline without bus. Shouldn't happen!");
|
|
|
|
|
2019-01-08 16:13:37 +00:00
|
|
|
pipeline
|
|
|
|
.set_state(gst::State::Playing)
|
|
|
|
.expect("Unable to set the pipeline to the `Playing` state");
|
2017-11-19 03:18:34 +00:00
|
|
|
|
2021-04-28 22:29:13 +00:00
|
|
|
for msg in bus.iter_timed(gst::ClockTime::NONE) {
|
2017-11-19 03:18:34 +00:00
|
|
|
use gst::MessageView;
|
|
|
|
|
|
|
|
match msg.view() {
|
|
|
|
MessageView::Eos(..) => break,
|
|
|
|
MessageView::Error(err) => {
|
2019-01-08 16:13:37 +00:00
|
|
|
pipeline
|
|
|
|
.set_state(gst::State::Null)
|
|
|
|
.expect("Unable to set the pipeline to the `Null` state");
|
2018-07-27 10:07:24 +00:00
|
|
|
|
2018-02-22 10:18:37 +00:00
|
|
|
return Err(ErrorMessage {
|
2018-07-27 10:36:40 +00:00
|
|
|
src: msg
|
2021-04-11 19:39:50 +00:00
|
|
|
.src()
|
|
|
|
.map(|s| String::from(s.path_string()))
|
2018-07-20 07:21:06 +00:00
|
|
|
.unwrap_or_else(|| String::from("None")),
|
2021-04-11 19:39:50 +00:00
|
|
|
error: err.error().to_string(),
|
|
|
|
debug: err.debug(),
|
|
|
|
source: err.error(),
|
2018-10-08 12:02:23 +00:00
|
|
|
}
|
|
|
|
.into());
|
2017-11-19 03:18:34 +00:00
|
|
|
}
|
2019-02-28 08:54:32 +00:00
|
|
|
MessageView::StateChanged(s) => {
|
2021-04-11 19:39:50 +00:00
|
|
|
if let Some(element) = msg.src() {
|
|
|
|
if element == pipeline && s.current() == gst::State::Playing {
|
2018-10-08 12:02:23 +00:00
|
|
|
eprintln!("PLAYING");
|
|
|
|
gst::debug_bin_to_dot_file(
|
|
|
|
&pipeline,
|
|
|
|
gst::DebugGraphDetails::all(),
|
|
|
|
"server-playing",
|
|
|
|
);
|
|
|
|
}
|
|
|
|
}
|
2019-02-28 08:54:32 +00:00
|
|
|
}
|
2017-11-19 03:18:34 +00:00
|
|
|
_ => (),
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2019-01-08 16:13:37 +00:00
|
|
|
pipeline
|
|
|
|
.set_state(gst::State::Null)
|
|
|
|
.expect("Unable to set the pipeline to the `Null` state");
|
2017-11-19 03:18:34 +00:00
|
|
|
|
2018-04-23 17:46:29 +00:00
|
|
|
Ok(())
|
2017-11-19 03:18:34 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
fn main() {
|
|
|
|
match examples_common::run(example_main) {
|
|
|
|
Ok(r) => r,
|
|
|
|
Err(e) => eprintln!("Error! {}", e),
|
|
|
|
}
|
|
|
|
}
|