mirrors/gstreamer-rs
mirrors/gstreamer-rs
1
1
Fork 1
mirror of https://gitlab.freedesktop.org/gstreamer/gstreamer-rs.git synced 2024-12-22 16:16:37 +00:00
Code Issues Projects Releases Wiki Activity

utils/streamproducer: add API to not forward eos to consumers

Browse source 
User may want to plug another producer when one is eos, in such case we
don't want to propagate eos to consumers.
This commit is contained in:
Guillaume Desmottes 2022-05-12 15:17:13 +02:00 committed by Sebastian Dröge
parent 351f070c8e
commit f68efd0cab
1 changed files with 25 additions and 9 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

34
gstreamer-utils/src/streamproducer.rs
View file

@ -115,6 +115,11 @@ impl StreamProducer {
self.consumers.lock().unwrap().discard = false;
}
/// configure if EOS from appsrc should be forwarded to all the consumers
pub fn set_forward_eos(&self, forward_eos: bool) {
self.consumers.lock().unwrap().forward_eos = forward_eos;
}
/// Get the GStreamer `appsink` wrapped by this producer
pub fn appsink(&self) -> &gst_app::AppSink {
&self.appsink
@ -143,6 +148,7 @@ impl<'a> From<&'a gst_app::AppSink> for StreamProducer {
latency_updated: false,
consumers: HashMap::new(),
discard: true,
forward_eos: true,
}));
appsink.set_callbacks(
@ -241,17 +247,25 @@ impl<'a> From<&'a gst_app::AppSink> for StreamProducer {
Ok(gst::FlowSuccess::Ok)
}))
.eos(glib::clone!(@strong consumers => move |_| {
let current_consumers = consumers
.eos(glib::clone!(@strong consumers => move |appsink| {
let stream_consumers = consumers
.lock()
.unwrap()
.consumers
.values()
.map(|c| c.appsrc.clone())
.collect::<Vec<_>>();
.unwrap();
for consumer in current_consumers {
let _ = consumer.end_of_stream();
if stream_consumers.forward_eos {
let current_consumers = stream_consumers
.consumers
.values()
.map(|c| c.appsrc.clone())
.collect::<Vec<_>>();
drop(stream_consumers);
for consumer in current_consumers {
gst::debug!(CAT, obj: appsink, "set EOS on consumer {}", consumer.name());
let _ = consumer.end_of_stream();
}
} else {
gst::debug!(CAT, obj: appsink, "don't forward EOS to consumers");
}
}))
.build(),
@ -292,6 +306,8 @@ struct StreamConsumers {
consumers: HashMap<gst_app::AppSrc, StreamConsumer>,
/// Whether appsrc samples should be forwarded to consumers yet
discard: bool,
/// Whether appsrc EOS should be forwarded to consumers
forward_eos: bool,
}
/// Wrapper around a consumer's `appsrc`