mirrors/gst-plugins-rs
mirrors/gst-plugins-rs
1
1
Fork 0
mirror of https://gitlab.freedesktop.org/gstreamer/gst-plugins-rs.git synced 2024-11-26 21:41:03 +00:00
Code Issues Projects Releases Wiki Activity

threadshare: Refactor infrastructure

Browse source 
The biggest changes are
- Many functions are not asynchronous anymore as it would be difficult
  to run them correctly with our mix of synchronous C code and Rust
  code.
- The pad context and its corresponding custom event are gone and
  instead thread local storage and task local storage are used. This
  makes it easier to correctly pass it through the different layers
  of Rust and C code and back.
- Sink events have a different function for serialized and oob events,
  src events are handled correctly by default now by simply forwarding
  them.
- Task::prepare() has a separate variant that takes a preparation
  function as this is a very common task.
- The task loop function can signal via its return value if it wants to
  be called again or not.
This commit is contained in:
Sebastian Dröge 2020-03-05 17:59:13 +02:00
parent 3ea465907d
commit e729324cce
7 changed files with 1122 additions and 1167 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
gst-plugin-threadshare/Cargo.toml
View file

@ -21,10 +21,9 @@ gst-net = { package = "gstreamer-net", git = "https://gitlab.freedesktop.org/gst
gst-rtp = { package = "gstreamer-rtp", git = "https://gitlab.freedesktop.org/gstreamer/gstreamer-rs" }
gstreamer-sys = { git = "https://gitlab.freedesktop.org/gstreamer/gstreamer-rs-sys" }
pin-project = "0.4"
tokio = { git = "https://github.com/fengalin/tokio", tag = "tokio-0.2.12-throttling", features = ["io-util", "macros", "rt-core", "sync", "stream", "time", "tcp", "udp"] }
tokio = { git = "https://github.com/fengalin/tokio", tag = "tokio-0.2.12-throttling", features = ["io-util", "macros", "rt-core", "sync", "stream", "time", "tcp", "udp", "rt-util"] }
futures = "0.3"
lazy_static = "1.0"
either = "1.0"
rand = "0.7"
net2 = "0.2"

351
gst-plugin-threadshare/src/runtime/executor.rs
View file

@ -1,4 +1,4 @@
// Copyright (C) 2018-2019 Sebastian Dröge <sebastian@centricular.com>
// Copyright (C) 2018-2020 Sebastian Dröge <sebastian@centricular.com>
// Copyright (C) 2019-2020 François Laignel <fengalin@free.fr>
//
// This library is free software; you can redistribute it and/or
@ -37,18 +37,14 @@
use futures::channel::oneshot;
use futures::future::BoxFuture;
use futures::prelude::*;
use futures::stream::futures_unordered::FuturesUnordered;
use glib;
use glib::{glib_boxed_derive_traits, glib_boxed_type};
use gst;
use gst::{gst_debug, gst_log, gst_trace};
use gst::{gst_debug, gst_log, gst_trace, gst_warning};
use lazy_static::lazy_static;
use std::cell::RefCell;
use std::collections::HashMap;
use std::collections::{HashMap, VecDeque};
use std::fmt;
use std::io;
use std::mem;
@ -77,6 +73,10 @@ lazy_static! {
thread_local!(static CURRENT_THREAD_CONTEXT: RefCell<Option<ContextWeak>> = RefCell::new(None));
tokio::task_local! {
static CURRENT_TASK_ID: TaskId;
}
/// Blocks on `future`.
///
/// IO & time related `Future`s must be handled within their own [`Context`].
@ -188,19 +188,17 @@ impl Drop for ContextShutdown {
}
#[derive(Clone, Copy, Eq, PartialEq, Hash, Debug)]
pub struct TaskQueueId(u64);
pub struct TaskId(u64);
impl glib::subclass::boxed::BoxedType for TaskQueueId {
const NAME: &'static str = "TsTaskQueueId";
pub type SubTaskOutput = Result<(), gst::FlowError>;
pub struct SubTaskQueue(VecDeque<BoxFuture<'static, SubTaskOutput>>);
glib_boxed_type!();
impl fmt::Debug for SubTaskQueue {
fn fmt(&self, fmt: &mut fmt::Formatter<'_>) -> fmt::Result {
fmt.debug_tuple("SubTaskQueue").finish()
}
}
glib_boxed_derive_traits!(TaskQueueId);
pub type TaskOutput = Result<(), gst::FlowError>;
type TaskQueue = FuturesUnordered<BoxFuture<'static, TaskOutput>>;
pub struct JoinError(tokio::task::JoinError);
impl fmt::Display for JoinError {
@ -224,14 +222,31 @@ impl From<tokio::task::JoinError> for JoinError {
}
/// Wrapper for the underlying runtime JoinHandle implementation.
pub struct JoinHandle<T>(tokio::task::JoinHandle<T>);
pub struct JoinHandle<T> {
join_handle: tokio::task::JoinHandle<T>,
context: ContextWeak,
task_id: TaskId,
}
unsafe impl<T: Send> Send for JoinHandle<T> {}
unsafe impl<T: Send> Sync for JoinHandle<T> {}
impl<T> From<tokio::task::JoinHandle<T>> for JoinHandle<T> {
fn from(src: tokio::task::JoinHandle<T>) -> Self {
JoinHandle(src)
impl<T> JoinHandle<T> {
pub fn is_current(&self) -> bool {
if let Some((context, task_id)) = Context::current_task() {
let self_context = self.context.upgrade();
self_context.map(|c| c == context).unwrap_or(false) && task_id == self.task_id
} else {
false
}
}
pub fn context(&self) -> Option<Context> {
self.context.upgrade()
}
pub fn task_id(&self) -> TaskId {
self.task_id
}
}
@ -241,16 +256,25 @@ impl<T> Future for JoinHandle<T> {
type Output = Result<T, JoinError>;
fn poll(mut self: Pin<&mut Self>, cx: &mut std::task::Context<'_>) -> Poll<Self::Output> {
self.as_mut().0.poll_unpin(cx).map_err(JoinError::from)
if self.as_ref().is_current() {
panic!("Trying to join task {:?} from itself", self.as_ref());
}
self.as_mut()
.join_handle
.poll_unpin(cx)
.map_err(JoinError::from)
}
}
impl<T> fmt::Debug for JoinHandle<T>
where
T: fmt::Debug,
{
impl<T> fmt::Debug for JoinHandle<T> {
fn fmt(&self, fmt: &mut fmt::Formatter<'_>) -> fmt::Result {
fmt.debug_struct("JoinHandle").finish()
let context_name = self.context.upgrade().map(|c| String::from(c.name()));
fmt.debug_struct("JoinHandle")
.field("context", &context_name)
.field("task_id", &self.task_id)
.finish()
}
}
@ -260,7 +284,7 @@ struct ContextInner {
handle: Mutex<tokio::runtime::Handle>,
// Only used for dropping
shutdown: ContextShutdown,
task_queues: Mutex<(u64, HashMap<u64, TaskQueue>)>,
task_queues: Mutex<(u64, HashMap<u64, SubTaskQueue>)>,
}
impl Drop for ContextInner {
@ -295,6 +319,14 @@ impl ContextWeak {
#[derive(Clone, Debug)]
pub struct Context(Arc<ContextInner>);
impl PartialEq for Context {
fn eq(&self, other: &Self) -> bool {
Arc::ptr_eq(&self.0, &other.0)
}
}
impl Eq for Context {}
impl Context {
pub fn acquire(context_name: &str, wait: u32) -> Result<Self, io::Error> {
let mut contexts = CONTEXTS.lock().unwrap();
@ -317,15 +349,6 @@ impl Context {
ContextWeak(Arc::downgrade(&self.0))
}
pub fn acquire_task_queue_id(&self) -> TaskQueueId {
let mut task_queues = self.0.task_queues.lock().unwrap();
let id = task_queues.0;
task_queues.0 += 1;
task_queues.1.insert(id, FuturesUnordered::new());
TaskQueueId(id)
}
pub fn name(&self) -> &str {
self.0.name.as_str()
}
@ -345,6 +368,21 @@ impl Context {
})
}
/// Returns the `TaskId` running on current thread, if any.
pub fn current_task() -> Option<(Context, TaskId)> {
CURRENT_THREAD_CONTEXT.with(|cur_ctx| {
cur_ctx
.borrow()
.as_ref()
.and_then(|ctx_weak| ctx_weak.upgrade())
.and_then(|ctx| {
let task_id = ctx.enter(|| CURRENT_TASK_ID.try_with(|task_id| *task_id).ok());
task_id.map(move |task_id| (ctx, task_id))
})
})
}
pub fn enter<F, R>(&self, f: F) -> R
where
F: FnOnce() -> R,
@ -357,55 +395,147 @@ impl Context {
Fut: Future + Send + 'static,
Fut::Output: Send + 'static,
{
self.0.handle.lock().unwrap().spawn(future).into()
}
pub fn release_task_queue(&self, id: TaskQueueId) -> Option<TaskQueue> {
let mut task_queues = self.0.task_queues.lock().unwrap();
task_queues.1.remove(&id.0)
}
let id = task_queues.0;
task_queues.0 += 1;
task_queues.1.insert(id, SubTaskQueue(VecDeque::new()));
pub fn add_task<T>(&self, id: TaskQueueId, task: T) -> Result<(), ()>
where
T: Future<Output = TaskOutput> + Send + 'static,
{
let mut task_queues = self.0.task_queues.lock().unwrap();
match task_queues.1.get_mut(&id.0) {
Some(task_queue) => {
task_queue.push(task.boxed());
Ok(())
let id = TaskId(id);
gst_trace!(
RUNTIME_CAT,
"Spawning new task {:?} on context {}",
id,
self.0.name
);
let join_handle = self.0.handle.lock().unwrap().spawn(async move {
let ctx = Context::current().unwrap();
gst_trace!(
RUNTIME_CAT,
"Running task {:?} on context {}",
id,
ctx.0.name
);
let res = CURRENT_TASK_ID.scope(id, future).await;
// Remove task from the list
{
let mut task_queues = ctx.0.task_queues.lock().unwrap();
if let Some(task_queue) = task_queues.1.remove(&id.0) {
let l = task_queue.0.len();
if l > 0 {
gst_warning!(
RUNTIME_CAT,
"Task {:?} on context {} has {} pending sub tasks",
id,
ctx.0.name,
l
);
}
}
}
None => Err(()),
gst_trace!(RUNTIME_CAT, "Task {:?} on context {} done", id, ctx.0.name);
res
});
JoinHandle {
join_handle,
context: self.downgrade(),
task_id: id,
}
}
pub fn clear_task_queue(&self, id: TaskQueueId) {
let mut task_queues = self.0.task_queues.lock().unwrap();
let task_queue = task_queues.1.get_mut(&id.0).unwrap();
*task_queue = FuturesUnordered::new();
}
pub fn drain_task_queue(&self, id: TaskQueueId) -> Option<impl Future<Output = TaskOutput>> {
let task_queue = {
let mut task_queues = self.0.task_queues.lock().unwrap();
let task_queue = task_queues.1.get_mut(&id.0).unwrap();
mem::replace(task_queue, FuturesUnordered::new())
pub fn current_has_sub_tasks() -> bool {
let (ctx, task_id) = match Context::current_task() {
Some(task) => task,
None => {
gst_trace!(RUNTIME_CAT, "No current task");
return false;
}
};
if !task_queue.is_empty() {
gst_log!(
RUNTIME_CAT,
"Scheduling {} tasks from {:?} on '{}'",
task_queue.len(),
id,
self.0.name,
);
let task_queues = ctx.0.task_queues.lock().unwrap();
task_queues
.1
.get(&task_id.0)
.map(|t| !t.0.is_empty())
.unwrap_or(false)
}
Some(task_queue.try_for_each(|_| future::ok(())))
} else {
None
pub fn add_sub_task<T>(sub_task: T) -> Result<(), T>
where
T: Future<Output = SubTaskOutput> + Send + 'static,
{
let (ctx, task_id) = match Context::current_task() {
Some(task) => task,
None => {
gst_trace!(RUNTIME_CAT, "No current task");
return Err(sub_task);
}
};
let mut task_queues = ctx.0.task_queues.lock().unwrap();
match task_queues.1.get_mut(&task_id.0) {
Some(task_queue) => {
gst_trace!(
RUNTIME_CAT,
"Adding subtask to {:?} on context {}",
task_id,
ctx.0.name
);
task_queue.0.push_back(sub_task.boxed());
Ok(())
}
None => {
gst_trace!(RUNTIME_CAT, "Task was removed in the meantime");
Err(sub_task)
}
}
}
pub fn drain_sub_tasks() -> impl Future<Output = SubTaskOutput> + Send + 'static {
async {
let (ctx, task_id) = match Context::current_task() {
Some(task) => task,
None => return Ok(()),
};
ctx.drain_sub_tasks_internal(task_id).await
}
}
fn drain_sub_tasks_internal(
&self,
id: TaskId,
) -> impl Future<Output = SubTaskOutput> + Send + 'static {
let mut task_queue = {
let mut task_queues = self.0.task_queues.lock().unwrap();
if let Some(task_queue) = task_queues.1.get_mut(&id.0) {
mem::replace(task_queue, SubTaskQueue(VecDeque::new()))
} else {
SubTaskQueue(VecDeque::new())
}
};
let name = self.0.name.clone();
async move {
if !task_queue.0.is_empty() {
gst_log!(
RUNTIME_CAT,
"Scheduling draining {} sub tasks from {:?} on '{}'",
task_queue.0.len(),
id,
&name,
);
for task in task_queue.0.drain(..) {
task.await?;
}
}
Ok(())
}
}
}
@ -432,46 +562,57 @@ mod tests {
const DELAY: Duration = Duration::from_millis(SLEEP_DURATION_MS as u64 * 10);
#[tokio::test]
async fn user_drain_pending_tasks() {
async fn drain_sub_tasks() {
// Setup
gst::init().unwrap();
let context = Context::acquire("user_drain_task_queue", SLEEP_DURATION_MS).unwrap();
let queue_id = context.acquire_task_queue_id();
let context = Context::acquire("drain_sub_tasks", SLEEP_DURATION_MS).unwrap();
let (sender, mut receiver) = mpsc::channel(1);
let sender: Arc<Mutex<mpsc::Sender<Item>>> = Arc::new(Mutex::new(sender));
let join_handle = context.spawn(async move {
let (sender, mut receiver) = mpsc::channel(1);
let sender: Arc<Mutex<mpsc::Sender<Item>>> = Arc::new(Mutex::new(sender));
let ctx_weak = context.downgrade();
let add_task = move |item| {
let sender_task = Arc::clone(&sender);
let context = ctx_weak.upgrade().unwrap();
context.add_task(queue_id, async move {
sender_task
.lock()
.await
.send(item)
.await
.map_err(|_| gst::FlowError::Error)
})
};
let add_sub_task = move |item| {
let sender = sender.clone();
Context::add_sub_task(async move {
sender
.lock()
.await
.send(item)
.await
.map_err(|_| gst::FlowError::Error)
})
};
// Tests
assert!(context.drain_task_queue(queue_id).is_none());
// Tests
add_task(0).unwrap();
receiver.try_next().unwrap_err();
// Drain empty queue
let drain_fut = Context::drain_sub_tasks();
drain_fut.await.unwrap();
let drain = context.drain_task_queue(queue_id).unwrap();
// Add a subtask
add_sub_task(0).map_err(drop).unwrap();
// User triggered drain
receiver.try_next().unwrap_err();
// Check that it was not executed yet
receiver.try_next().unwrap_err();
drain.await.unwrap();
assert_eq!(receiver.try_next().unwrap(), Some(0));
// Drain it now and check that it was executed
let drain_fut = Context::drain_sub_tasks();
drain_fut.await.unwrap();
assert_eq!(receiver.try_next().unwrap(), Some(0));
add_task(1).unwrap();
receiver.try_next().unwrap_err();
// Add another task and check that it's not executed yet
add_sub_task(1).map_err(drop).unwrap();
receiver.try_next().unwrap_err();
// Return the receiver
receiver
});
let mut receiver = join_handle.await.unwrap();
// The last sub task should be simply dropped at this point
assert_eq!(receiver.try_next().unwrap(), None);
}
#[tokio::test]

7
gst-plugin-threadshare/src/runtime/mod.rs
View file

@ -44,20 +44,17 @@
//! [`PadSink`]: pad/struct.PadSink.html
pub mod executor;
pub use executor::{Context, JoinHandle, TaskOutput};
pub use executor::{Context, JoinHandle, SubTaskOutput};
pub mod pad;
pub use pad::{PadSink, PadSinkRef, PadSinkWeak, PadSrc, PadSrcRef, PadSrcWeak};
pub mod pad_context;
pub use pad_context::{PadContext, PadContextRef, PadContextWeak};
pub mod task;
pub mod prelude {
pub use super::pad::{PadSinkHandler, PadSrcHandler};
}
pub mod task;
pub mod time;
use gst;

621
gst-plugin-threadshare/src/runtime/pad.rs
View file

@ -1,4 +1,5 @@
// Copyright (C) 2019-2020 François Laignel <fengalin@free.fr>
// Copyright (C) 2020 Sebastian Dröge <sebastian@centricular.com>
//
// This library is free software; you can redistribute it and/or
// modify it under the terms of the GNU Library General Public
@ -59,21 +60,22 @@
//! `Element A` & `Element B` can also be linked to non-threadshare `Element`s in which case, they
//! operate in a regular synchronous way.
//!
//! Note that only operations on the streaming thread (serialized events, buffers, serialized
//! queries) are handled from the `PadContext` and asynchronously, everything else operates
//! blocking.
//!
//! [`PadSink`]: struct.PadSink.html
//! [`PadSrc`]: struct.PadSrc.html
//! [`Context`]: ../executor/struct.Context.html
use either::Either;
use futures::future;
use futures::future::BoxFuture;
use futures::lock::{Mutex, MutexGuard};
use futures::prelude::*;
use gst;
use gst::prelude::*;
use gst::subclass::prelude::*;
use gst::{gst_debug, gst_error, gst_fixme, gst_log, gst_loggable_error, gst_warning};
use gst::{gst_debug, gst_error, gst_fixme, gst_log, gst_loggable_error};
use gst::{FlowError, FlowSuccess};
use std::fmt;
@ -81,8 +83,7 @@ use std::marker::PhantomData;
use std::sync;
use std::sync::{Arc, Weak};
use super::executor::{self, Context, JoinHandle, TaskOutput};
use super::pad_context::{PadContext, PadContextWeak};
use super::executor::Context;
use super::task::Task;
use super::RUNTIME_CAT;
@ -109,10 +110,13 @@ impl fmt::Display for PadContextError {
impl std::error::Error for PadContextError {}
#[inline]
fn event_ret_to_event_full_res(ret: bool, event: &gst::Event) -> Result<FlowSuccess, FlowError> {
fn event_ret_to_event_full_res(
ret: bool,
event_type: gst::EventType,
) -> Result<FlowSuccess, FlowError> {
if ret {
Ok(FlowSuccess::Ok)
} else if let gst::EventView::Caps(_) = event.view() {
} else if event_type == gst::EventType::Caps {
Err(FlowError::NotNegotiated)
} else {
Err(FlowError::Error)
@ -120,17 +124,17 @@ fn event_ret_to_event_full_res(ret: bool, event: &gst::Event) -> Result<FlowSucc
}
#[inline]
fn event_to_event_full(
ret: Either<bool, BoxFuture<'static, bool>>,
event: gst::Event,
) -> Either<Result<FlowSuccess, FlowError>, BoxFuture<'static, Result<FlowSuccess, FlowError>>> {
match ret {
Either::Left(ret) => Either::Left(event_ret_to_event_full_res(ret, &event)),
Either::Right(fut) => Either::Right(
fut.map(move |ret| event_ret_to_event_full_res(ret, &event))
.boxed(),
),
}
fn event_to_event_full(ret: bool, event_type: gst::EventType) -> Result<FlowSuccess, FlowError> {
event_ret_to_event_full_res(ret, event_type)
}
#[inline]
fn event_to_event_full_serialized(
ret: BoxFuture<'static, bool>,
event_type: gst::EventType,
) -> BoxFuture<'static, Result<FlowSuccess, FlowError>> {
ret.map(move |ret| event_ret_to_event_full_res(ret, event_type))
.boxed()
}
/// A trait to define `handler`s for [`PadSrc`] callbacks.
@ -140,7 +144,7 @@ fn event_to_event_full(
/// [`PadSrc`]: struct.PadSrc.html
/// [`pad` module]: index.html
pub trait PadSrcHandler: Clone + Send + Sync + 'static {
type ElementImpl: ElementImpl;
type ElementImpl: ElementImpl + ObjectSubclass;
fn src_activate(
&self,
@ -189,24 +193,9 @@ pub trait PadSrcHandler: Clone + Send + Sync + 'static {
_imp: &Self::ElementImpl,
element: &gst::Element,
event: gst::Event,
) -> Either<bool, BoxFuture<'static, bool>> {
if event.is_serialized() {
let pad_weak = pad.downgrade();
let element = element.clone();
Either::Right(
async move {
let pad = pad_weak.upgrade().expect("PadSrc no longer exists");
gst_log!(RUNTIME_CAT, obj: pad.gst_pad(), "Handling {:?}", event);
pad.gst_pad().event_default(Some(&element), event)
}
.boxed(),
)
} else {
gst_log!(RUNTIME_CAT, obj: pad.gst_pad(), "Handling {:?}", event);
Either::Left(pad.gst_pad().event_default(Some(element), event))
}
) -> bool {
gst_log!(RUNTIME_CAT, obj: pad.gst_pad(), "Handling {:?}", event);
pad.gst_pad().event_default(Some(element), event)
}
fn src_event_full(
@ -215,12 +204,11 @@ pub trait PadSrcHandler: Clone + Send + Sync + 'static {
imp: &Self::ElementImpl,
element: &gst::Element,
event: gst::Event,
) -> Either<Result<FlowSuccess, FlowError>, BoxFuture<'static, Result<FlowSuccess, FlowError>>>
{
) -> Result<FlowSuccess, FlowError> {
// default is to dispatch to `src_event`
// (as implemented in `gst_pad_send_event_unchecked`)
let event_clone = event.clone();
event_to_event_full(self.src_event(pad, imp, element, event), event_clone)
let event_type = event.get_type();
event_to_event_full(self.src_event(pad, imp, element, event), event_type)
}
fn src_query(
@ -242,15 +230,12 @@ pub trait PadSrcHandler: Clone + Send + Sync + 'static {
}
#[derive(Default, Debug)]
pub struct PadSrcState {
is_initialized: bool,
}
struct PadSrcState;
#[derive(Debug)]
struct PadSrcInner {
state: Mutex<PadSrcState>,
state: sync::Mutex<PadSrcState>,
gst_pad: gst::Pad,
pad_context: sync::RwLock<Option<PadContext>>,
task: Task,
}
@ -261,26 +246,11 @@ impl PadSrcInner {
}
PadSrcInner {
state: Mutex::new(PadSrcState::default()),
state: sync::Mutex::new(PadSrcState::default()),
gst_pad,
pad_context: sync::RwLock::new(None),
task: Task::default(),
}
}
fn has_pad_context(&self) -> bool {
self.pad_context.read().unwrap().as_ref().is_some()
}
}
impl Drop for PadSrcInner {
fn drop(&mut self) {
// Check invariant which can't be held automatically in `PadSrc`
// because `drop` can't be `async`
if self.has_pad_context() {
panic!("Missing call to `PadSrc::unprepare`");
}
}
}
/// A [`PadSrc`] which can be moved in [`handler`]s functions and `Future`s.
@ -330,28 +300,20 @@ impl<'a> PadSrcRef<'a> {
self.strong.gst_pad()
}
pub async fn lock_state(&'a self) -> MutexGuard<'a, PadSrcState> {
self.strong.lock_state().await
}
pub fn pad_context(&self) -> PadContextWeak {
self.strong.pad_context()
}
/// Spawns `future` using current [`PadContext`].
///
/// # Panics
///
/// This function panics if the `PadSrc` is not prepared.
///
/// [`PadContext`]: ../struct.PadContext.html
pub fn spawn<Fut>(&self, future: Fut) -> JoinHandle<Fut::Output>
where
Fut: Future + Send + 'static,
Fut::Output: Send + 'static,
{
self.strong.spawn(future)
}
///// Spawns `future` using current [`PadContext`].
/////
///// # Panics
/////
///// This function panics if the `PadSrc` is not prepared.
/////
///// [`PadContext`]: ../struct.PadContext.html
//pub fn spawn<Fut>(&self, future: Fut) -> JoinHandle<Fut::Output>
//where
// Fut: Future + Send + 'static,
// Fut::Output: Send + 'static,
//{
// self.strong.spawn(future)
//}
pub fn downgrade(&self) -> PadSrcWeak {
self.strong.downgrade()
@ -373,21 +335,22 @@ impl<'a> PadSrcRef<'a> {
///
/// The `Task` will loop on the provided `func`.
/// The execution occurs on the `Task`'s context.
pub async fn start_task<F, Fut>(&self, func: F)
pub fn start_task<F, Fut>(&self, func: F)
where
F: (FnMut() -> Fut) + Send + 'static,
Fut: Future<Output = ()> + Send + 'static,
Fut: Future<Output = glib::Continue> + Send + 'static,
{
self.strong.start_task(func).await;
self.strong.start_task(func);
}
/// Pauses the `Started` `Pad` `Task`.
pub async fn pause_task(&self) {
let _ = self.strong.pause_task().await;
/// Cancels the `Started` `Pad` `Task`.
pub fn cancel_task(&self) {
self.strong.cancel_task();
}
pub async fn stop_task(&self) {
self.strong.stop_task().await;
/// Stops the `Started` `Pad` `Task`.
pub fn stop_task(&self) {
self.strong.stop_task();
}
fn activate_mode_hook(
@ -407,12 +370,6 @@ impl<'a> PadSrcRef<'a> {
));
}
if !active {
executor::block_on(async {
self.strong.lock_state().await.is_initialized = false;
});
}
Ok(())
}
}
@ -430,97 +387,28 @@ impl PadSrcStrong {
&self.0.gst_pad
}
#[inline]
async fn lock_state(&self) -> MutexGuard<'_, PadSrcState> {
self.0.state.lock().await
}
#[inline]
fn pad_context_priv(&self) -> sync::RwLockReadGuard<'_, Option<PadContext>> {
self.0.pad_context.read().unwrap()
}
#[inline]
fn pad_context(&self) -> PadContextWeak {
self.pad_context_priv()
.as_ref()
.expect("PadContext not initialized")
.downgrade()
}
#[inline]
pub fn spawn<Fut>(&self, future: Fut) -> JoinHandle<Fut::Output>
where
Fut: Future + Send + 'static,
Fut::Output: Send + 'static,
{
let pad_ctx = self.pad_context_priv();
pad_ctx
.as_ref()
.expect("PadContext not initialized")
.spawn(future)
}
//#[inline]
//fn spawn<Fut>(&self, future: Fut) -> JoinHandle<Fut::Output>
//where
// Fut: Future + Send + 'static,
// Fut::Output: Send + 'static,
//{
// let pad_ctx = self.pad_context_priv();
// pad_ctx
// .as_ref()
// .expect("PadContext not initialized")
// .spawn(future)
//}
#[inline]
fn downgrade(&self) -> PadSrcWeak {
PadSrcWeak(Arc::downgrade(&self.0))
}
fn push_prelude(
&self,
state: &mut MutexGuard<'_, PadSrcState>,
) -> Result<FlowSuccess, FlowError> {
if !state.is_initialized || self.gst_pad().check_reconfigure() {
if !self.push_pad_context_event() {
return Err(FlowError::Error);
}
if !state.is_initialized {
// Get rid of reconfigure flag
self.gst_pad().check_reconfigure();
state.is_initialized = true;
}
}
Ok(FlowSuccess::Ok)
}
#[inline]
fn push_pad_context_event(&self) -> bool {
let pad_ctx = self.pad_context_priv();
let pad_ctx = pad_ctx.as_ref().unwrap();
gst_log!(
RUNTIME_CAT,
obj: self.gst_pad(),
"Pushing PadContext Event {}",
pad_ctx,
);
let ret = self.gst_pad().push_event(pad_ctx.new_sticky_event());
if !ret {
gst_error!(RUNTIME_CAT,
obj: self.gst_pad(),
"Failed to push PadContext sticky event to PadSrc",
);
}
ret
}
fn drain_pending_tasks(&self) -> Option<impl Future<Output = TaskOutput>> {
self.pad_context_priv()
.as_ref()
.unwrap()
.drain_pending_tasks()
}
#[inline]
async fn push(&self, buffer: gst::Buffer) -> Result<FlowSuccess, FlowError> {
let mut state = self.lock_state().await;
gst_log!(RUNTIME_CAT, obj: self.gst_pad(), "Pushing {:?}", buffer);
self.push_prelude(&mut state)?;
let success = self.gst_pad().push(buffer).map_err(|err| {
gst_error!(RUNTIME_CAT,
obj: self.gst_pad(),
@ -530,91 +418,63 @@ impl PadSrcStrong {
err
})?;
if let Some(pending_tasks) = self.drain_pending_tasks() {
gst_log!(RUNTIME_CAT, obj: self.gst_pad(), "Processing pending tasks (push)");
pending_tasks.await?;
}
gst_log!(RUNTIME_CAT, obj: self.gst_pad(), "Processing any pending sub tasks");
Context::drain_sub_tasks().await?;
Ok(success)
}
#[inline]
async fn push_list(&self, list: gst::BufferList) -> Result<FlowSuccess, FlowError> {
let mut state = self.lock_state().await;
gst_log!(RUNTIME_CAT, obj: self.gst_pad(), "Pushing {:?}", list);
self.push_prelude(&mut state)?;
let success = self.gst_pad().push_list(list).map_err(|err| {
gst_error!(
RUNTIME_CAT,
obj: self.gst_pad(),
"Failed to push BufferList to PadSrc: {:?} ({})",
"Failed to push BufferList to PadSrc: {:?}",
err,
self.pad_context_priv().as_ref().unwrap(),
);
err
})?;
if let Some(pending_tasks) = self.drain_pending_tasks() {
gst_log!(RUNTIME_CAT, obj: self.gst_pad(), "Processing pending tasks (push_list)");
pending_tasks.await?;
}
gst_log!(RUNTIME_CAT, obj: self.gst_pad(), "Processing any pending sub tasks");
Context::drain_sub_tasks().await?;
Ok(success)
}
#[inline]
async fn push_event(&self, event: gst::Event) -> bool {
let mut state = self.lock_state().await;
gst_log!(RUNTIME_CAT, obj: self.gst_pad(), "Pushing {:?}", event);
let was_handled = if PadContext::is_pad_context_event(&event) {
// Push our own PadContext
if !self.push_pad_context_event() {
return false;
}
let was_handled = self.gst_pad().push_event(event);
// Get rid of reconfigure flag
self.gst_pad().check_reconfigure();
state.is_initialized = true;
true
} else {
gst_log!(RUNTIME_CAT, obj: self.gst_pad(), "Pushing {:?}", event);
if self.push_prelude(&mut state).is_err() {
return false;
}
self.gst_pad().push_event(event)
};
if let Some(pending_tasks) = self.drain_pending_tasks() {
gst_log!(RUNTIME_CAT, obj: self.gst_pad(), "Processing pending tasks (push_event)");
if pending_tasks.await.is_err() {
return false;
}
gst_log!(RUNTIME_CAT, obj: self.gst_pad(), "Processing any pending sub tasks");
if Context::drain_sub_tasks().await.is_err() {
return false;
}
was_handled
}
#[inline]
async fn start_task<F, Fut>(&self, func: F)
fn start_task<F, Fut>(&self, func: F)
where
F: (FnMut() -> Fut) + Send + 'static,
Fut: Future<Output = ()> + Send + 'static,
Fut: Future<Output = glib::Continue> + Send + 'static,
{
self.0.task.start(func).await;
self.0.task.start(func);
}
#[inline]
pub async fn pause_task(&self) -> BoxFuture<'static, ()> {
self.0.task.pause().await
fn cancel_task(&self) {
self.0.task.cancel();
}
#[inline]
async fn stop_task(&self) {
self.0.task.stop().await;
fn stop_task(&self) {
self.0.task.stop();
}
}
@ -670,28 +530,20 @@ impl PadSrc {
self.gst_pad().check_reconfigure()
}
pub async fn lock_state(&self) -> MutexGuard<'_, PadSrcState> {
self.0.lock_state().await
}
pub fn pad_context(&self) -> PadContextWeak {
self.0.pad_context()
}
/// Spawns `future` using current [`PadContext`].
///
/// # Panics
///
/// This function panics if the `PadSrc` is not prepared.
///
/// [`PadContext`]: ../struct.PadContext.html
pub fn spawn<Fut>(&self, future: Fut) -> JoinHandle<Fut::Output>
where
Fut: Future + Send + 'static,
Fut::Output: Send + 'static,
{
self.0.spawn(future)
}
///// Spawns `future` using current [`PadContext`].
/////
///// # Panics
/////
///// This function panics if the `PadSrc` is not prepared.
/////
///// [`PadContext`]: ../struct.PadContext.html
//pub fn spawn<Fut>(&self, future: Fut) -> JoinHandle<Fut::Output>
//where
// Fut: Future + Send + 'static,
// Fut::Output: Send + 'static,
//{
// self.0.spawn(future)
//}
fn init_pad_functions<H: PadSrcHandler>(&self, handler: &H) {
let handler_clone = handler.clone();
@ -749,15 +601,7 @@ impl PadSrc {
|| Err(FlowError::Error),
move |imp, element| {
let this_ref = this_weak.upgrade().expect("PadSrc no longer exists");
match handler.src_event_full(&this_ref, imp, &element, event) {
Either::Left(res) => res,
Either::Right(_fut) => {
// See these threads:
// https://gitlab.freedesktop.org/gstreamer/gst-plugins-rs/merge_requests/240#note_378446
// https://gitlab.freedesktop.org/gstreamer/gst-plugins-rs/merge_requests/240#note_378454
unimplemented!("Future handling in src_event*");
}
}
handler.src_event_full(&this_ref, imp, &element, event)
},
)
});
@ -784,26 +628,35 @@ impl PadSrc {
});
}
pub async fn prepare<H: PadSrcHandler>(
pub fn prepare<H: PadSrcHandler>(
&self,
context: Context,
handler: &H,
) -> Result<(), PadContextError> {
let _state = self.lock_state().await;
) -> Result<(), super::task::TaskError> {
let _state = (self.0).0.state.lock().unwrap();
gst_log!(RUNTIME_CAT, obj: self.gst_pad(), "Preparing");
if (self.0).0.has_pad_context() {
return Err(PadContextError::ActiveContext);
}
(self.0).0.task.prepare(context)?;
(self.0)
.0
.task
.prepare(context.clone())
.await
.map_err(|_| PadContextError::ActiveTask)?;
self.init_pad_functions(handler);
*(self.0).0.pad_context.write().unwrap() = Some(PadContext::new(context.clone()));
Ok(())
}
pub fn prepare_with_func<H: PadSrcHandler, F, Fut>(
&self,
context: Context,
handler: &H,
prepare_func: F,
) -> Result<(), super::task::TaskError>
where
F: (FnOnce() -> Fut) + Send + 'static,
Fut: Future<Output = ()> + Send + 'static,
{
let _state = (self.0).0.state.lock().unwrap();
gst_log!(RUNTIME_CAT, obj: self.gst_pad(), "Preparing");
(self.0).0.task.prepare_with_func(context, prepare_func)?;
self.init_pad_functions(handler);
@ -811,15 +664,14 @@ impl PadSrc {
}
/// Releases the resources held by this `PadSrc`.
pub async fn unprepare(&self) -> Result<(), PadContextError> {
let _state = self.lock_state().await;
pub fn unprepare(&self) -> Result<(), PadContextError> {
let _state = (self.0).0.state.lock().unwrap();
gst_log!(RUNTIME_CAT, obj: self.gst_pad(), "Unpreparing");
(self.0)
.0
.task
.unprepare()
.await
.map_err(|_| PadContextError::ActiveTask)?;
self.gst_pad()
@ -829,12 +681,10 @@ impl PadSrc {
self.gst_pad()
.set_event_function(move |_gst_pad, _parent, _event| false);
self.gst_pad()
.set_event_full_function(move |_gst_pad, _parent, _event| Err(FlowError::Error));
.set_event_full_function(move |_gst_pad, _parent, _event| Err(FlowError::Flushing));
self.gst_pad()
.set_query_function(move |_gst_pad, _parent, _query| false);
*(self.0).0.pad_context.write().unwrap() = None;
Ok(())
}
@ -854,21 +704,20 @@ impl PadSrc {
///
/// The `Task` will loop on the provided `func`.
/// The execution occurs on the `Task`'s context.
pub async fn start_task<F, Fut>(&self, func: F)
pub fn start_task<F, Fut>(&self, func: F)
where
F: (FnMut() -> Fut) + Send + 'static,
Fut: Future<Output = ()> + Send + 'static,
Fut: Future<Output = glib::Continue> + Send + 'static,
{
self.0.start_task(func).await;
self.0.start_task(func);
}
/// Pauses the `Started` `Pad` `task`.
pub async fn pause_task(&self) -> BoxFuture<'static, ()> {
self.0.pause_task().await
pub fn cancel_task(&self) {
self.0.cancel_task();
}
pub async fn stop_task(&self) {
self.0.stop_task().await;
pub fn stop_task(&self) {
self.0.stop_task();
}
}
@ -879,7 +728,7 @@ impl PadSrc {
/// [`PadSink`]: struct.PadSink.html
/// [`pad` module]: index.html
pub trait PadSinkHandler: Clone + Send + Sync + 'static {
type ElementImpl: ElementImpl;
type ElementImpl: ElementImpl + ObjectSubclass;
fn sink_activate(
&self,
@ -948,24 +797,30 @@ pub trait PadSinkHandler: Clone + Send + Sync + 'static {
_imp: &Self::ElementImpl,
element: &gst::Element,
event: gst::Event,
) -> Either<bool, BoxFuture<'static, bool>> {
if event.is_serialized() {
let pad_weak = pad.downgrade();
let element = element.clone();
) -> bool {
assert!(!event.is_serialized());
gst_log!(RUNTIME_CAT, obj: pad.gst_pad(), "Handling {:?}", event);
pad.gst_pad().event_default(Some(element), event)
}
Either::Right(
async move {
let pad = pad_weak.upgrade().expect("PadSink no longer exists");
gst_log!(RUNTIME_CAT, obj: pad.gst_pad(), "Handling {:?}", event);
fn sink_event_serialized(
&self,
pad: &PadSinkRef,
_imp: &Self::ElementImpl,
element: &gst::Element,
event: gst::Event,
) -> BoxFuture<'static, bool> {
assert!(event.is_serialized());
let pad_weak = pad.downgrade();
let element = element.clone();
pad.gst_pad().event_default(Some(&element), event)
}
.boxed(),
)
} else {
async move {
let pad = pad_weak.upgrade().expect("PadSink no longer exists");
gst_log!(RUNTIME_CAT, obj: pad.gst_pad(), "Handling {:?}", event);
Either::Left(pad.gst_pad().event_default(Some(element), event))
pad.gst_pad().event_default(Some(&element), event)
}
.boxed()
}
fn sink_event_full(
@ -974,12 +829,29 @@ pub trait PadSinkHandler: Clone + Send + Sync + 'static {
imp: &Self::ElementImpl,
element: &gst::Element,
event: gst::Event,
) -> Either<Result<FlowSuccess, FlowError>, BoxFuture<'static, Result<FlowSuccess, FlowError>>>
{
) -> Result<FlowSuccess, FlowError> {
assert!(!event.is_serialized());
// default is to dispatch to `sink_event`
// (as implemented in `gst_pad_send_event_unchecked`)
let event_clone = event.clone();
event_to_event_full(self.sink_event(pad, imp, element, event), event_clone)
let event_type = event.get_type();
event_to_event_full(self.sink_event(pad, imp, element, event), event_type)
}
fn sink_event_full_serialized(
&self,
pad: &PadSinkRef,
imp: &Self::ElementImpl,
element: &gst::Element,
event: gst::Event,
) -> BoxFuture<'static, Result<FlowSuccess, FlowError>> {
assert!(event.is_serialized());
// default is to dispatch to `sink_event`
// (as implemented in `gst_pad_send_event_unchecked`)
let event_type = event.get_type();
event_to_event_full_serialized(
self.sink_event_serialized(pad, imp, element, event),
event_type,
)
}
fn sink_query(
@ -989,12 +861,13 @@ pub trait PadSinkHandler: Clone + Send + Sync + 'static {
element: &gst::Element,
query: &mut gst::QueryRef,
) -> bool {
gst_log!(RUNTIME_CAT, obj: pad.gst_pad(), "Handling {:?}", query);
if query.is_serialized() {
gst_log!(RUNTIME_CAT, obj: pad.gst_pad(), "Dropping {:?}", query);
// FIXME serialized queries should be handled with the dataflow
// but we can't return a `Future` because we couldn't honor QueryRef's lifetime
false
} else {
gst_log!(RUNTIME_CAT, obj: pad.gst_pad(), "Handling {:?}", query);
pad.gst_pad().query_default(Some(element), query)
}
}
@ -1003,7 +876,6 @@ pub trait PadSinkHandler: Clone + Send + Sync + 'static {
#[derive(Debug)]
struct PadSinkInner {
gst_pad: gst::Pad,
pad_context: sync::RwLock<Option<PadContextWeak>>,
}
impl PadSinkInner {
@ -1012,10 +884,7 @@ impl PadSinkInner {
panic!("Wrong pad direction for PadSink");
}
PadSinkInner {
gst_pad,
pad_context: sync::RwLock::new(None),
}
PadSinkInner { gst_pad }
}
}
@ -1065,10 +934,6 @@ impl<'a> PadSinkRef<'a> {
self.strong.gst_pad()
}
pub fn pad_context(&self) -> Option<PadContextWeak> {
self.strong.pad_context()
}
pub fn downgrade(&self) -> PadSinkWeak {
self.strong.downgrade()
}
@ -1097,32 +962,9 @@ impl<'a> PadSinkRef<'a> {
&self,
fut: impl Future<Output = Result<FlowSuccess, FlowError>> + Send + 'static,
) -> Result<FlowSuccess, FlowError> {
if Context::is_context_thread() {
match self.pad_context().as_ref() {
Some(pad_ctx_weak) => {
pad_ctx_weak
.upgrade()
.expect("PadContext no longer exists")
.add_pending_task(fut.map(|res| res.map(drop)));
Ok(FlowSuccess::Ok)
}
None => {
// This can happen when an upstream element forwards the PadContext sticky event
// after the StreamStart event. While the upstream element should be fixed,
// we have no other solution but blocking the `Context`.
// See https://gitlab.freedesktop.org/gstreamer/gst-plugins-rs/issues/94
gst_warning!(
RUNTIME_CAT,
obj: self.gst_pad(),
"Operating on a Context without a PadContext. An upstream element should be fixed.",
);
// Note: we don't use `crate::runtime::executor::block_on` here
// because `Context::is_context_thread()` is checked in the `if`
// statement above.
futures::executor::block_on(fut)
}
}
} else {
// First try to add it as a sub task to the current task, if any
if let Err(fut) = Context::add_sub_task(fut.map(|res| res.map(drop))) {
// FIXME: update comments below
// Not on a context thread: execute the Future immediately.
//
// - If there is no PadContext, we don't have any other options.
@ -1134,7 +976,9 @@ impl<'a> PadSinkRef<'a> {
// Note: we don't use `crate::runtime::executor::block_on` here
// because `Context::is_context_thread()` is checked in the `if`
// statement above.
futures::executor::block_on(fut)
futures::executor::block_on(fut.map(|res| res.map(|_| gst::FlowSuccess::Ok)))
} else {
Ok(gst::FlowSuccess::Ok)
}
}
}
@ -1151,10 +995,6 @@ impl PadSinkStrong {
&self.0.gst_pad
}
fn pad_context(&self) -> Option<PadContextWeak> {
self.0.pad_context.read().unwrap().clone()
}
fn downgrade(&self) -> PadSinkWeak {
PadSinkWeak(Arc::downgrade(&self.0))
}
@ -1200,10 +1040,6 @@ impl PadSink {
self.0.gst_pad()
}
pub fn pad_context(&self) -> Option<PadContextWeak> {
self.0.pad_context()
}
pub fn downgrade(&self) -> PadSinkWeak {
self.0.downgrade()
}
@ -1265,9 +1101,25 @@ impl PadSink {
parent,
|| Err(FlowError::Error),
move |imp, element| {
let this_ref = this_weak.upgrade().expect("PadSink no longer exists");
let chain_fut = handler.sink_chain(&this_ref, imp, &element, buffer);
this_ref.handle_future(chain_fut)
if Context::current_has_sub_tasks() {
let this_weak = this_weak.clone();
let handler = handler.clone();
let element = element.clone();
let delayed_fut = async move {
let imp =
<H::ElementImpl as ObjectSubclass>::from_instance(&element);
let this_ref =
this_weak.upgrade().ok_or(gst::FlowError::Flushing)?;
handler.sink_chain(&this_ref, imp, &element, buffer).await
};
let _ = Context::add_sub_task(delayed_fut.map(|res| res.map(drop)));
Ok(gst::FlowSuccess::Ok)
} else {
let this_ref = this_weak.upgrade().expect("PadSink no longer exists");
let chain_fut = handler.sink_chain(&this_ref, imp, &element, buffer);
this_ref.handle_future(chain_fut)
}
},
)
});
@ -1282,10 +1134,28 @@ impl PadSink {
parent,
|| Err(FlowError::Error),
move |imp, element| {
let this_ref = this_weak.upgrade().expect("PadSink no longer exists");
let chain_list_fut =
handler.sink_chain_list(&this_ref, imp, &element, list);
this_ref.handle_future(chain_list_fut)
if Context::current_has_sub_tasks() {
let this_weak = this_weak.clone();
let handler = handler.clone();
let element = element.clone();
let delayed_fut = async move {
let imp =
<H::ElementImpl as ObjectSubclass>::from_instance(&element);
let this_ref =
this_weak.upgrade().ok_or(gst::FlowError::Flushing)?;
handler
.sink_chain_list(&this_ref, imp, &element, list)
.await
};
let _ = Context::add_sub_task(delayed_fut.map(|res| res.map(drop)));
Ok(gst::FlowSuccess::Ok)
} else {
let this_ref = this_weak.upgrade().expect("PadSink no longer exists");
let chain_list_fut =
handler.sink_chain_list(&this_ref, imp, &element, list);
this_ref.handle_future(chain_list_fut)
}
},
)
});
@ -1295,7 +1165,7 @@ impl PadSink {
let handler_clone = handler.clone();
let this_weak = self.downgrade();
self.gst_pad()
.set_event_full_function(move |gst_pad, parent, event| {
.set_event_full_function(move |_gst_pad, parent, event| {
let handler = handler_clone.clone();
let this_weak = this_weak.clone();
H::ElementImpl::catch_panic_pad_function(
@ -1303,14 +1173,31 @@ impl PadSink {
|| Err(FlowError::Error),
move |imp, element| {
let this_ref = this_weak.upgrade().expect("PadSink no longer exists");
if let Some(received_pc) = PadContext::check_pad_context_event(&event) {
gst_log!(RUNTIME_CAT, obj: gst_pad, "Received {:?}", received_pc);
*this_ref.strong.0.pad_context.write().unwrap() = Some(received_pc);
}
if event.is_serialized() {
if Context::current_has_sub_tasks() {
let this_weak = this_weak.clone();
let handler = handler.clone();
let element = element.clone();
let delayed_fut = async move {
let imp =
<H::ElementImpl as ObjectSubclass>::from_instance(&element);
let this_ref =
this_weak.upgrade().ok_or(gst::FlowError::Flushing)?;
match handler.sink_event_full(&this_ref, imp, &element, event) {
Either::Left(ret) => ret,
Either::Right(fut) => this_ref.handle_future(fut),
handler
.sink_event_full_serialized(&this_ref, imp, &element, event)
.await
};
let _ = Context::add_sub_task(delayed_fut.map(|res| res.map(drop)));
Ok(gst::FlowSuccess::Ok)
} else {
let event_fut = handler
.sink_event_full_serialized(&this_ref, imp, &element, event);
this_ref.handle_future(event_fut)
}
} else {
handler.sink_event_full(&this_ref, imp, &element, event)
}
},
)
@ -1338,23 +1225,23 @@ impl PadSink {
});
}
pub async fn prepare<H: PadSinkHandler>(&self, handler: &H) {
pub fn prepare<H: PadSinkHandler>(&self, handler: &H) {
gst_log!(RUNTIME_CAT, obj: self.gst_pad(), "Preparing");
self.init_pad_functions(handler);
}
/// Releases the resources held by this `PadSink`.
pub async fn unprepare(&self) {
pub fn unprepare(&self) {
gst_log!(RUNTIME_CAT, obj: self.gst_pad(), "Unpreparing");
self.gst_pad()
.set_chain_function(move |_gst_pad, _parent, _buffer| Err(FlowError::Error));
.set_chain_function(move |_gst_pad, _parent, _buffer| Err(FlowError::Flushing));
self.gst_pad()
.set_chain_list_function(move |_gst_pad, _parent, _list| Err(FlowError::Error));
.set_chain_list_function(move |_gst_pad, _parent, _list| Err(FlowError::Flushing));
self.gst_pad()
.set_event_function(move |_gst_pad, _parent, _event| false);
self.gst_pad()
.set_event_full_function(move |_gst_pad, _parent, _event| Err(FlowError::Error));
.set_event_full_function(move |_gst_pad, _parent, _event| Err(FlowError::Flushing));
self.gst_pad()
.set_query_function(move |_gst_pad, _parent, _query| false);
}

254
gst-plugin-threadshare/src/runtime/pad_context.rs
View file

@ -1,254 +0,0 @@
// Copyright (C) 2019 François Laignel <fengalin@free.fr>
//
// This library is free software; you can redistribute it and/or
// modify it under the terms of the GNU Library General Public
// License as published by the Free Software Foundation; either
// version 2 of the License, or (at your option) any later version.
//
// This library is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
// Library General Public License for more details.
//
// You should have received a copy of the GNU Library General Public
// License along with this library; if not, write to the
// Free Software Foundation, Inc., 51 Franklin Street, Suite 500,
// Boston, MA 02110-1335, USA.
//! Types that allow `Pad`s to operate within the threadshare runtime.
use futures::prelude::*;
use glib;
use glib::{glib_boxed_derive_traits, glib_boxed_type};
use std::marker::PhantomData;
use super::executor::{Context, ContextWeak, JoinHandle, TaskOutput, TaskQueueId};
#[derive(Clone)]
pub struct PadContextWeak {
context_weak: ContextWeak,
queue_id: TaskQueueId,
}
impl PadContextWeak {
pub fn upgrade(&self) -> Option<PadContextRef> {
self.context_weak
.upgrade()
.map(|inner| PadContextRef::new(inner, self.queue_id))
}
}
impl glib::subclass::boxed::BoxedType for PadContextWeak {
const NAME: &'static str = "TsPadContext";
glib_boxed_type!();
}
glib_boxed_derive_traits!(PadContextWeak);
impl std::fmt::Debug for PadContextWeak {
fn fmt(&self, f: &mut std::fmt::Formatter) -> std::fmt::Result {
match self.context_weak.upgrade() {
Some(context) => write!(
f,
"PadContext {{ context: '{}'), {:?} }}",
context.name(),
self.queue_id
),
None => write!(
f,
"PadContext {{ context: _NO LONGER AVAILABLE_, {:?} }}",
self.queue_id
),
}
}
}
#[derive(Debug)]
pub struct PadContextRef<'a> {
strong: PadContextStrong,
phantom: PhantomData<&'a PadContextStrong>,
}
impl<'a> PadContextRef<'a> {
fn new(context: Context, queue_id: TaskQueueId) -> Self {
PadContextRef {
strong: PadContextStrong { context, queue_id },
phantom: PhantomData,
}
}
}
impl<'a> PadContextRef<'a> {
pub fn downgrade(&self) -> PadContextWeak {
self.strong.downgrade()
}
pub fn spawn<Fut>(&self, future: Fut) -> JoinHandle<Fut::Output>
where
Fut: Future + Send + 'static,
Fut::Output: Send + 'static,
{
self.strong.context.spawn(future)
}
pub fn add_pending_task<T>(&self, task: T)
where
T: Future<Output = TaskOutput> + Send + 'static,
{
self.strong.add_pending_task(task);
}
pub fn drain_pending_tasks(&self) -> Option<impl Future<Output = TaskOutput>> {
self.strong.drain_pending_tasks()
}
pub fn clear_pending_tasks(&self) {
self.strong.clear_pending_tasks();
}
pub fn context(&self) -> &Context {
&self.strong.context
}
}
impl std::fmt::Display for PadContextRef<'_> {
fn fmt(&self, f: &mut std::fmt::Formatter) -> std::fmt::Result {
self.strong.fmt(f)
}
}
#[derive(Debug)]
struct PadContextStrong {
context: Context,
queue_id: TaskQueueId,
}
impl PadContextStrong {
#[inline]
pub fn downgrade(&self) -> PadContextWeak {
PadContextWeak {
context_weak: self.context.downgrade(),
queue_id: self.queue_id,
}
}
#[inline]
fn add_pending_task<T>(&self, task: T)
where
T: Future<Output = TaskOutput> + Send + 'static,
{
self.context
.add_task(self.queue_id, task)
.expect("TaskQueueId controlled by TaskContext");
}
#[inline]
fn drain_pending_tasks(&self) -> Option<impl Future<Output = TaskOutput>> {
self.context.drain_task_queue(self.queue_id)
}
#[inline]
fn clear_pending_tasks(&self) {
self.context.clear_task_queue(self.queue_id);
}
}
impl std::fmt::Display for PadContextStrong {
fn fmt(&self, f: &mut std::fmt::Formatter) -> std::fmt::Result {
write!(f, "Context('{}'), {:?}", self.context.name(), self.queue_id)
}
}
/// A wrapper on a [`Context`] with additional features for [`PadSrc`] & [`PadSink`].
///
/// [`Context`]: ../executor/struct.Context.html
/// [`PadSrc`]: ../pad/struct.PadSrc.html
/// [`PadSink`]: ../pad/struct.PadSink.html
#[derive(Debug)]
pub struct PadContext(PadContextStrong);
impl PadContext {
pub fn new(context: Context) -> Self {
PadContext(PadContextStrong {
queue_id: context.acquire_task_queue_id(),
context,
})
}
pub fn downgrade(&self) -> PadContextWeak {
self.0.downgrade()
}
pub fn as_ref(&self) -> PadContextRef<'_> {
PadContextRef::new(self.0.context.clone(), self.0.queue_id)
}
pub fn spawn<Fut>(&self, future: Fut) -> JoinHandle<Fut::Output>
where
Fut: Future + Send + 'static,
Fut::Output: Send + 'static,
{
self.0.context.spawn(future)
}
pub fn drain_pending_tasks(&self) -> Option<impl Future<Output = TaskOutput>> {
self.0.drain_pending_tasks()
}
pub fn clear_pending_tasks(&self) {
self.0.clear_pending_tasks();
}
pub(super) fn new_sticky_event(&self) -> gst::Event {
let s = gst::Structure::new("ts-pad-context", &[("pad-context", &self.downgrade())]);
gst::Event::new_custom_downstream_sticky(s).build()
}
#[inline]
pub fn is_pad_context_sticky_event(event: &gst::event::CustomDownstreamSticky) -> bool {
event.get_structure().unwrap().get_name() == "ts-pad-context"
}
#[inline]
pub fn is_pad_context_event(event: &gst::Event) -> bool {
if let gst::EventView::CustomDownstreamSticky(e) = event.view() {
return Self::is_pad_context_sticky_event(&e);
}
false
}
pub fn check_pad_context_event(event: &gst::Event) -> Option<PadContextWeak> {
if let gst::EventView::CustomDownstreamSticky(e) = event.view() {
if Self::is_pad_context_sticky_event(&e) {
let s = e.get_structure().unwrap();
let pad_context = s
.get::<&PadContextWeak>("pad-context")
.expect("event field")
.expect("missing event field")
.clone();
Some(pad_context)
} else {
None
}
} else {
None
}
}
}
impl Drop for PadContext {
fn drop(&mut self) {
self.0.context.release_task_queue(self.0.queue_id);
}
}
impl std::fmt::Display for PadContext {
fn fmt(&self, f: &mut std::fmt::Formatter) -> std::fmt::Result {
self.0.fmt(f)
}
}

408
gst-plugin-threadshare/src/runtime/task.rs
View file

@ -1,4 +1,5 @@
// Copyright (C) 2019 François Laignel <fengalin@free.fr>
// Copyright (C) 2020 Sebastian Dröge <sebastian@centricular.com>
//
// This library is free software; you can redistribute it and/or
// modify it under the terms of the GNU Library General Public
@ -17,15 +18,14 @@
//! An execution loop to run asynchronous processing.
use futures::future::{self, abortable, AbortHandle, Aborted, BoxFuture};
use futures::lock::Mutex;
use futures::future::{abortable, AbortHandle, Aborted};
use futures::prelude::*;
use gst::TaskState;
use gst::{gst_debug, gst_log, gst_trace, gst_warning};
use std::fmt;
use std::sync::Arc;
use std::sync::{Arc, Mutex};
use super::{Context, JoinHandle, RUNTIME_CAT};
@ -48,6 +48,8 @@ impl std::error::Error for TaskError {}
struct TaskInner {
context: Option<Context>,
state: TaskState,
prepare_handle: Option<JoinHandle<Result<Result<(), gst::FlowError>, Aborted>>>,
prepare_abort_handle: Option<AbortHandle>,
abort_handle: Option<AbortHandle>,
loop_handle: Option<JoinHandle<Result<(), Aborted>>>,
}
@ -57,6 +59,8 @@ impl Default for TaskInner {
TaskInner {
context: None,
state: TaskState::Stopped,
prepare_handle: None,
prepare_abort_handle: None,
abort_handle: None,
loop_handle: None,
}
@ -86,41 +90,159 @@ impl Default for Task {
}
impl Task {
pub async fn prepare(&self, context: Context) -> Result<(), TaskError> {
let mut inner = self.0.lock().await;
pub fn prepare_with_func<F, Fut>(
&self,
context: Context,
prepare_func: F,
) -> Result<(), TaskError>
where
F: (FnOnce() -> Fut) + Send + 'static,
Fut: Future<Output = ()> + Send + 'static,
{
gst_debug!(RUNTIME_CAT, "Preparing task");
let mut inner = self.0.lock().unwrap();
if inner.state != TaskState::Stopped {
return Err(TaskError::ActiveTask);
}
// Spawn prepare function in the background
let task_weak = Arc::downgrade(&self.0);
let (prepare_fut, prepare_abort_handle) = abortable(async move {
gst_trace!(RUNTIME_CAT, "Calling task prepare function");
prepare_func().await;
gst_trace!(RUNTIME_CAT, "Task prepare function finished");
Context::drain_sub_tasks().await?;
// Once the prepare function is finished we can forget the corresponding
// handles so that unprepare and friends don't have to block on it anymore
if let Some(task_inner) = task_weak.upgrade() {
let mut inner = task_inner.lock().unwrap();
inner.prepare_abort_handle = None;
inner.prepare_handle = None;
}
gst_trace!(RUNTIME_CAT, "Task fully prepared");
Ok(())
});
let prepare_handle = context.spawn(prepare_fut);
inner.prepare_handle = Some(prepare_handle);
inner.prepare_abort_handle = Some(prepare_abort_handle);
inner.context = Some(context);
gst_debug!(RUNTIME_CAT, "Task prepared");
Ok(())
}
pub async fn unprepare(&self) -> Result<(), TaskError> {
let mut inner = self.0.lock().await;
pub fn prepare(&self, context: Context) -> Result<(), TaskError> {
gst_debug!(RUNTIME_CAT, "Preparing task");
let mut inner = self.0.lock().unwrap();
if inner.state != TaskState::Stopped {
return Err(TaskError::ActiveTask);
}
inner.context = None;
inner.prepare_handle = None;
inner.prepare_abort_handle = None;
inner.context = Some(context);
gst_debug!(RUNTIME_CAT, "Task prepared");
Ok(())
}
pub async fn state(&self) -> TaskState {
self.0.lock().await.state
pub fn unprepare(&self) -> Result<(), TaskError> {
gst_debug!(RUNTIME_CAT, "Unpreparing task");
let mut inner = self.0.lock().unwrap();
if inner.state != TaskState::Stopped {
return Err(TaskError::ActiveTask);
}
// Abort any pending preparation
if let Some(abort_handle) = inner.prepare_abort_handle.take() {
abort_handle.abort();
}
let prepare_handle = inner.prepare_handle.take();
let context = inner.context.take().unwrap();
drop(inner);
if let Some(prepare_handle) = prepare_handle {
if let Some((cur_context, cur_task_id)) = Context::current_task() {
if prepare_handle.is_current() {
// This would deadlock!
gst_warning!(
RUNTIME_CAT,
"Trying to stop task {:?} from itself, not waiting",
prepare_handle
);
} else if cur_context == context {
// This is ok: as we're on the same thread and the prepare function is aborted
// this means that it won't ever be called, and we're not inside it here
gst_debug!(
RUNTIME_CAT,
"Asynchronously waiting for task {:?} on the same context",
prepare_handle
);
let _ = Context::add_sub_task(async move {
let _ = prepare_handle.await;
Ok(())
});
} else {
// This is suboptimal but we can't really do this asynchronously as otherwise
// it might be started again before it's actually stopped.
gst_warning!(
RUNTIME_CAT,
"Synchronously waiting for task {:?} on task {:?} on context {}",
prepare_handle,
cur_task_id,
cur_context.name()
);
let _ = futures::executor::block_on(prepare_handle);
}
} else {
gst_debug!(
RUNTIME_CAT,
"Synchronously waiting for task {:?}",
prepare_handle
);
let _ = futures::executor::block_on(prepare_handle);
}
}
gst_debug!(RUNTIME_CAT, "Task unprepared");
Ok(())
}
pub fn state(&self) -> TaskState {
self.0.lock().unwrap().state
}
pub fn context(&self) -> Option<Context> {
self.0.lock().unwrap().context.as_ref().cloned()
}
/// `Starts` the `Task`.
///
/// The `Task` will loop on the provided @func.
/// The execution occurs on the `Task`'s context.
pub async fn start<F, Fut>(&self, mut func: F)
pub fn start<F, Fut>(&self, mut func: F)
where
F: (FnMut() -> Fut) + Send + 'static,
Fut: Future<Output = ()> + Send + 'static,
Fut: Future<Output = glib::Continue> + Send + 'static,
{
let inner_clone = Arc::clone(&self.0);
let mut inner = self.0.lock().await;
let mut inner = self.0.lock().unwrap();
match inner.state {
TaskState::Started => {
gst_log!(RUNTIME_CAT, "Task already Started");
@ -132,18 +254,85 @@ impl Task {
gst_debug!(RUNTIME_CAT, "Starting Task");
let (loop_fut, abort_handle) = abortable(async move {
loop {
func().await;
let prepare_handle = inner.prepare_handle.take();
match inner_clone.lock().await.state {
// If the task was only cancelled and not actually stopped yet then
// wait for that to happen as first thing in the new task.
let loop_handle = inner.loop_handle.take();
let (loop_fut, abort_handle) = abortable(async move {
let task_id = Context::current_task().unwrap().1;
if let Some(loop_handle) = loop_handle {
gst_trace!(
RUNTIME_CAT,
"Waiting for previous loop to finish before starting"
);
let _ = loop_handle.await;
}
// First await on the prepare function, if any
if let Some(prepare_handle) = prepare_handle {
gst_trace!(RUNTIME_CAT, "Waiting for prepare before starting");
let res = prepare_handle.await;
if res.is_err() {
gst_warning!(RUNTIME_CAT, "Preparing failed");
return;
}
}
gst_trace!(RUNTIME_CAT, "Starting task loop");
// Then loop as long as we're actually running
loop {
match inner_clone.lock().unwrap().state {
TaskState::Started => (),
TaskState::Paused | TaskState::Stopped => {
gst_trace!(RUNTIME_CAT, "Stopping task loop");
break;
}
other => unreachable!("Unexpected Task state {:?}", other),
}
if func().await == glib::Continue(false) {
let mut inner = inner_clone.lock().unwrap();
// Make sure to only reset the state if this is still the correct task
// and no new task was started in the meantime
if inner.state == TaskState::Started
&& inner
.loop_handle
.as_ref()
.map(|h| h.task_id() == task_id)
.unwrap_or(false)
{
gst_trace!(RUNTIME_CAT, "Pausing task loop");
inner.state = TaskState::Paused;
}
break;
}
}
// Once the loop function is finished we can forget the corresponding
// handles so that unprepare and friends don't have to block on it anymore
{
let mut inner = inner_clone.lock().unwrap();
// Make sure to only reset the state if this is still the correct task
// and no new task was started in the meantime
if inner
.loop_handle
.as_ref()
.map(|h| h.task_id() == task_id)
.unwrap_or(false)
{
inner.abort_handle = None;
inner.loop_handle = None;
}
}
gst_trace!(RUNTIME_CAT, "Task loop finished");
});
let loop_handle = inner
@ -159,38 +348,27 @@ impl Task {
gst_debug!(RUNTIME_CAT, "Task Started");
}
/// Pauses the `Started` `Task`.
pub async fn pause(&self) -> BoxFuture<'static, ()> {
let mut inner = self.0.lock().await;
match inner.state {
TaskState::Started => {
gst_log!(RUNTIME_CAT, "Pausing Task");
inner.state = TaskState::Paused;
let loop_handle = inner.loop_handle.take().unwrap();
async move {
let _ = loop_handle.await;
gst_log!(RUNTIME_CAT, "Task Paused");
}
.boxed()
}
TaskState::Paused => {
gst_trace!(RUNTIME_CAT, "Task already Paused");
future::ready(()).boxed()
}
other => {
gst_warning!(RUNTIME_CAT, "Attempting to pause Task in state {:?}", other,);
future::ready(()).boxed()
}
/// Cancels the `Task` so that it stops running as soon as possible.
pub fn cancel(&self) {
let mut inner = self.0.lock().unwrap();
if inner.state != TaskState::Started {
gst_log!(RUNTIME_CAT, "Task already paused or stopped");
return;
}
gst_debug!(RUNTIME_CAT, "Cancelling Task");
// Abort any still running loop function
if let Some(abort_handle) = inner.abort_handle.take() {
abort_handle.abort();
}
inner.state = TaskState::Paused;
}
pub async fn stop(&self) {
let mut inner = self.0.lock().await;
/// Stops the `Started` `Task` and wait for it to finish.
pub fn stop(&self) {
let mut inner = self.0.lock().unwrap();
if inner.state == TaskState::Stopped {
gst_log!(RUNTIME_CAT, "Task already stopped");
return;
@ -198,23 +376,69 @@ impl Task {
gst_debug!(RUNTIME_CAT, "Stopping Task");
inner.state = TaskState::Stopped;
// Abort any still running loop function
if let Some(abort_handle) = inner.abort_handle.take() {
abort_handle.abort();
}
if let Some(loop_handle) = inner.loop_handle.take() {
let _ = loop_handle.await;
// And now wait for it to actually stop
let loop_handle = inner.loop_handle.take();
let context = inner.context.as_ref().unwrap().clone();
drop(inner);
if let Some(loop_handle) = loop_handle {
if let Some((cur_context, cur_task_id)) = Context::current_task() {
if loop_handle.is_current() {
// This would deadlock!
gst_warning!(
RUNTIME_CAT,
"Trying to stop task {:?} from itself, not waiting",
loop_handle
);
} else if cur_context == context {
// This is ok: as we're on the same thread and the loop function is aborted
// this means that it won't ever be called, and we're not inside it here
gst_debug!(
RUNTIME_CAT,
"Asynchronously waiting for task {:?} on the same context",
loop_handle
);
let _ = Context::add_sub_task(async move {
let _ = loop_handle.await;
Ok(())
});
} else {
// This is suboptimal but we can't really do this asynchronously as otherwise
// it might be started again before it's actually stopped.
gst_warning!(
RUNTIME_CAT,
"Synchronously waiting for task {:?} on task {:?} on context {}",
loop_handle,
cur_task_id,
cur_context.name()
);
let _ = futures::executor::block_on(loop_handle);
}
} else {
gst_debug!(
RUNTIME_CAT,
"Synchronously waiting for task {:?}",
loop_handle
);
let _ = futures::executor::block_on(loop_handle);
}
}
inner.state = TaskState::Stopped;
gst_debug!(RUNTIME_CAT, "Task Stopped");
gst_debug!(RUNTIME_CAT, "Task stopped");
}
}
#[cfg(test)]
mod tests {
use futures::channel::mpsc;
use futures::channel::{mpsc, oneshot};
use futures::lock::Mutex;
use std::sync::Arc;
@ -230,7 +454,7 @@ mod tests {
let context = Context::acquire("task", 2).unwrap();
let task = Task::default();
task.prepare(context).await.unwrap();
task.prepare(context).unwrap();
let (mut sender, receiver) = mpsc::channel(0);
let receiver = Arc::new(Mutex::new(receiver));
@ -241,28 +465,86 @@ mod tests {
async move {
gst_debug!(RUNTIME_CAT, "task test: awaiting receiver");
match receiver.lock().await.next().await {
Some(_) => gst_debug!(RUNTIME_CAT, "task test: item received"),
None => gst_debug!(RUNTIME_CAT, "task test: channel complete"),
Some(_) => {
gst_debug!(RUNTIME_CAT, "task test: item received");
glib::Continue(true)
}
None => {
gst_debug!(RUNTIME_CAT, "task test: channel complete");
glib::Continue(false)
}
}
}
})
.await;
});
gst_debug!(RUNTIME_CAT, "task test: sending item");
sender.send(()).await.unwrap();
gst_debug!(RUNTIME_CAT, "task test: item sent");
gst_debug!(RUNTIME_CAT, "task test: pausing");
let pause_completion = task.pause().await;
gst_debug!(RUNTIME_CAT, "task test: dropping sender");
drop(sender);
gst_debug!(RUNTIME_CAT, "task test: stopping");
task.stop();
gst_debug!(RUNTIME_CAT, "task test: stopped");
task.unprepare().unwrap();
gst_debug!(RUNTIME_CAT, "task test: unprepared");
}
#[tokio::test]
async fn task_with_prepare_func() {
gst::init().unwrap();
let context = Context::acquire("task_with_prepare_func", 2).unwrap();
let task = Task::default();
let (prepare_sender, prepare_receiver) = oneshot::channel();
task.prepare_with_func(context, move || async move {
prepare_sender.send(()).unwrap();
})
.unwrap();
let (mut sender, receiver) = mpsc::channel(0);
let receiver = Arc::new(Mutex::new(receiver));
let mut prepare_receiver = Some(prepare_receiver);
gst_debug!(RUNTIME_CAT, "task test: starting");
task.start(move || {
if let Some(mut prepare_receiver) = prepare_receiver.take() {
assert_eq!(prepare_receiver.try_recv().unwrap(), Some(()));
}
let receiver = Arc::clone(&receiver);
async move {
gst_debug!(RUNTIME_CAT, "task test: awaiting receiver");
match receiver.lock().await.next().await {
Some(_) => {
gst_debug!(RUNTIME_CAT, "task test: item received");
glib::Continue(true)
}
None => {
gst_debug!(RUNTIME_CAT, "task test: channel complete");
glib::Continue(false)
}
}
}
});
gst_debug!(RUNTIME_CAT, "task test: sending item");
sender.send(()).await.unwrap();
gst_debug!(RUNTIME_CAT, "task test: item sent");
gst_debug!(RUNTIME_CAT, "task test: dropping sender");
drop(sender);
gst_debug!(RUNTIME_CAT, "task test: awaiting pause completion");
pause_completion.await;
gst_debug!(RUNTIME_CAT, "task test: stopping");
task.stop().await;
task.stop();
gst_debug!(RUNTIME_CAT, "task test: stopped");
task.unprepare().unwrap();
gst_debug!(RUNTIME_CAT, "task test: unprepared");
}
}

645
gst-plugin-threadshare/tests/pad.rs
View file

@ -1,4 +1,5 @@
// Copyright (C) 2019-2020 François Laignel <fengalin@free.fr>
// Copyright (C) 2020 Sebastian Dröge <sebastian@centricular.com>
//
// This library is free software; you can redistribute it and/or
// modify it under the terms of the GNU Library General Public
@ -15,11 +16,9 @@
// Free Software Foundation, Inc., 51 Franklin Street, Suite 500,
// Boston, MA 02110-1335, USA.
use either::Either;
use futures::channel::mpsc;
use futures::future::BoxFuture;
use futures::lock::Mutex;
use futures::lock::Mutex as FutMutex;
use futures::prelude::*;
use glib;
@ -34,13 +33,11 @@ use gst::{gst_debug, gst_error_msg, gst_log};
use lazy_static::lazy_static;
use std::boxed::Box;
use std::sync::{self, Arc};
use std::sync::atomic::{AtomicBool, Ordering};
use std::sync::{Arc, Mutex};
use gstthreadshare::runtime::executor::block_on;
use gstthreadshare::runtime::prelude::*;
use gstthreadshare::runtime::{
self, Context, JoinHandle, PadContext, PadSink, PadSinkRef, PadSrc, PadSrcRef,
};
use gstthreadshare::runtime::{Context, PadSink, PadSinkRef, PadSrc, PadSrcRef};
const DEFAULT_CONTEXT: &str = "";
const THROTTLING_DURATION: u32 = 2;
@ -68,7 +65,7 @@ static SRC_PROPERTIES: [glib::subclass::Property; 1] =
)
})];
#[derive(Debug)]
#[derive(Clone, Debug)]
struct Settings {
context: String,
}
@ -82,135 +79,79 @@ lazy_static! {
}
#[derive(Clone, Debug, Default)]
struct PadSrcHandlerTest {
flush_join_handle: Arc<sync::Mutex<Option<JoinHandle<Result<(), ()>>>>>,
}
struct PadSrcHandlerTest;
impl PadSrcHandlerTest {
async fn start_task(&self, pad: PadSrcRef<'_>, receiver: mpsc::Receiver<Item>) {
fn start_task(&self, pad: PadSrcRef<'_>, receiver: mpsc::Receiver<Item>) {
gst_debug!(SRC_CAT, obj: pad.gst_pad(), "SrcPad task starting");
let pad_weak = pad.downgrade();
let receiver = Arc::new(Mutex::new(receiver));
let receiver = Arc::new(FutMutex::new(receiver));
pad.start_task(move || {
let pad_weak = pad_weak.clone();
let receiver = Arc::clone(&receiver);
async move {
let item = receiver.lock().await.next().await;
let pad = pad_weak.upgrade().expect("PadSrc no longer exists");
let item = match item {
Some(item) => item,
None => {
gst_debug!(SRC_CAT, obj: pad.gst_pad(), "SrcPad channel aborted");
pad.pause_task().await;
return;
let item = {
let mut receiver = receiver.lock().await;
match receiver.next().await {
Some(item) => item,
None => {
gst_debug!(SRC_CAT, obj: pad.gst_pad(), "SrcPad channel aborted");
return glib::Continue(false);
}
}
};
Self::push_item(pad, item).await;
// We could also check here first if we're flushing but as we're not doing anything
// complicated below we can just defer that to the pushing function
match Self::push_item(pad, item).await {
Ok(_) => glib::Continue(true),
Err(gst::FlowError::Flushing) => glib::Continue(false),
Err(err) => panic!("Got error {:?}", err),
}
}
})
.await;
});
}
async fn push_item(pad: PadSrcRef<'_>, item: Item) {
async fn push_item(pad: PadSrcRef<'_>, item: Item) -> Result<gst::FlowSuccess, gst::FlowError> {
match item {
Item::Event(event) => {
pad.push_event(event).await;
Ok(gst::FlowSuccess::Ok)
}
Item::Buffer(buffer) => {
pad.push(buffer).await.unwrap();
}
Item::BufferList(list) => {
pad.push_list(list).await.unwrap();
}
Item::Buffer(buffer) => pad.push(buffer).await,
Item::BufferList(list) => pad.push_list(list).await,
}
gst_debug!(SRC_CAT, obj: pad.gst_pad(), "SrcPad handled an Item");
}
}
impl PadSrcHandler for PadSrcHandlerTest {
type ElementImpl = ElementSrcTest;
fn src_activatemode(
&self,
_pad: &PadSrcRef,
_elem_src_test: &ElementSrcTest,
_element: &gst::Element,
mode: gst::PadMode,
active: bool,
) -> Result<(), gst::LoggableError> {
gst_debug!(SRC_CAT, "SrcPad activatemode {:?}, {}", mode, active);
Ok(())
}
fn src_event(
&self,
pad: &PadSrcRef,
_elem_src_test: &ElementSrcTest,
elem_src_test: &ElementSrcTest,
element: &gst::Element,
event: gst::Event,
) -> Either<bool, BoxFuture<'static, bool>> {
) -> bool {
gst_log!(SRC_CAT, obj: pad.gst_pad(), "Handling {:?}", event);
let ret = match event.view() {
EventView::FlushStart(..) => {
let mut flush_join_handle = self.flush_join_handle.lock().unwrap();
if flush_join_handle.is_none() {
let element = element.clone();
let pad_weak = pad.downgrade();
*flush_join_handle = Some(pad.spawn(async move {
let res = ElementSrcTest::from_instance(&element)
.pause(&element)
.await;
let pad = pad_weak.upgrade().unwrap();
if res.is_ok() {
gst_debug!(SRC_CAT, obj: pad.gst_pad(), "FlushStart complete");
} else {
gst_debug!(SRC_CAT, obj: pad.gst_pad(), "FlushStart failed");
}
res
}));
} else {
gst_debug!(SRC_CAT, obj: pad.gst_pad(), "FlushStart ignored: previous Flush in progress");
}
// Cancel the task so that it finishes ASAP
// and clear the sender
elem_src_test.pause(element).unwrap();
true
}
EventView::Qos(..) | EventView::Reconfigure(..) | EventView::Latency(..) => true,
EventView::FlushStop(..) => {
let element = element.clone();
let flush_join_handle_weak = Arc::downgrade(&self.flush_join_handle);
let pad_weak = pad.downgrade();
let fut = async move {
let mut ret = false;
let pad = pad_weak.upgrade().unwrap();
let flush_join_handle = flush_join_handle_weak.upgrade().unwrap();
let flush_join_handle = flush_join_handle.lock().unwrap().take();
if let Some(flush_join_handle) = flush_join_handle {
gst_debug!(SRC_CAT, obj: pad.gst_pad(), "Waiting for FlushStart to complete");
if let Ok(Ok(())) = flush_join_handle.await {
ret = ElementSrcTest::from_instance(&element)
.start(&element)
.await
.is_ok();
gst_debug!(SRC_CAT, obj: pad.gst_pad(), "FlushStop complete");
} else {
gst_debug!(SRC_CAT, obj: pad.gst_pad(), "FlushStop aborted: FlushStart failed");
}
} else {
gst_debug!(SRC_CAT, obj: pad.gst_pad(), "FlushStop ignored: no Flush in progress");
}
ret
}
.boxed();
return Either::Right(fut);
elem_src_test.flush_stop(&element);
true
}
_ => false,
};
@ -221,18 +162,7 @@ impl PadSrcHandler for PadSrcHandlerTest {
gst_log!(SRC_CAT, obj: pad.gst_pad(), "Didn't handle {:?}", event);
}
Either::Left(ret)
}
}
#[derive(Debug)]
struct ElementSrcState {
sender: Option<mpsc::Sender<Item>>,
}
impl Default for ElementSrcState {
fn default() -> Self {
ElementSrcState { sender: None }
ret
}
}
@ -240,13 +170,13 @@ impl Default for ElementSrcState {
struct ElementSrcTest {
src_pad: PadSrc,
src_pad_handler: PadSrcHandlerTest,
state: Mutex<ElementSrcState>,
sender: Mutex<Option<mpsc::Sender<Item>>>,
settings: Mutex<Settings>,
}
impl ElementSrcTest {
async fn try_push(&self, item: Item) -> Result<(), Item> {
match self.state.lock().await.sender.as_mut() {
fn try_push(&self, item: Item) -> Result<(), Item> {
match self.sender.lock().unwrap().as_mut() {
Some(sender) => sender
.try_send(item)
.map_err(mpsc::TrySendError::into_inner),
@ -254,21 +184,19 @@ impl ElementSrcTest {
}
}
async fn prepare(&self, element: &gst::Element) -> Result<(), gst::ErrorMessage> {
let _state = self.state.lock().await;
fn prepare(&self, element: &gst::Element) -> Result<(), gst::ErrorMessage> {
gst_debug!(SRC_CAT, obj: element, "Preparing");
let context = Context::acquire(&self.settings.lock().await.context, THROTTLING_DURATION)
.map_err(|err| {
gst_error_msg!(
gst::ResourceError::OpenRead,
["Failed to acquire Context: {}", err]
)
})?;
let settings = self.settings.lock().unwrap().clone();
let context = Context::acquire(&settings.context, THROTTLING_DURATION).map_err(|err| {
gst_error_msg!(
gst::ResourceError::OpenRead,
["Failed to acquire Context: {}", err]
)
})?;
self.src_pad
.prepare(context, &self.src_pad_handler)
.await
.map_err(|err| {
gst_error_msg!(
gst::ResourceError::OpenRead,
@ -281,52 +209,89 @@ impl ElementSrcTest {
Ok(())
}
async fn unprepare(&self, element: &gst::Element) -> Result<(), ()> {
let _state = self.state.lock().await;
fn unprepare(&self, element: &gst::Element) -> Result<(), ()> {
gst_debug!(SRC_CAT, obj: element, "Unpreparing");
self.src_pad.stop_task().await;
let _ = self.src_pad.unprepare().await;
self.src_pad.unprepare().unwrap();
gst_debug!(SRC_CAT, obj: element, "Unprepared");
Ok(())
}
async fn start(&self, element: &gst::Element) -> Result<(), ()> {
let mut state = self.state.lock().await;
fn start(&self, element: &gst::Element) -> Result<(), ()> {
let mut sender = self.sender.lock().unwrap();
if sender.is_some() {
gst_debug!(SRC_CAT, obj: element, "Already started");
return Err(());
}
gst_debug!(SRC_CAT, obj: element, "Starting");
let (sender, receiver) = mpsc::channel(1);
state.sender = Some(sender);
self.src_pad_handler
.start_task(self.src_pad.as_ref(), receiver)
.await;
self.start_unchecked(&mut sender);
gst_debug!(SRC_CAT, obj: element, "Started");
Ok(())
}
async fn pause(&self, element: &gst::Element) -> Result<(), ()> {
let pause_completion = {
let mut state = self.state.lock().await;
gst_debug!(SRC_CAT, obj: element, "Pausing");
fn flush_stop(&self, element: &gst::Element) {
// Keep the lock on the `sender` until `flush_stop` is complete
// so as to prevent race conditions due to concurrent state transitions.
// Note that this won't deadlock as `sender` is not used
// within the `src_pad`'s `Task`.
let mut sender = self.sender.lock().unwrap();
if sender.is_some() {
gst_debug!(SRC_CAT, obj: element, "Already started");
return;
}
let pause_completion = self.src_pad.pause_task().await;
// Prevent subsequent items from being enqueued
state.sender = None;
gst_debug!(SRC_CAT, obj: element, "Stopping Flush");
pause_completion
};
// Stop it so we wait for it to actually finish
self.src_pad.stop_task();
gst_debug!(SRC_CAT, obj: element, "Waiting for Task Pause to complete");
pause_completion.await;
// And then start it again
self.start_unchecked(&mut sender);
gst_debug!(SRC_CAT, obj: element, "Stopped Flush");
}
fn start_unchecked(&self, sender: &mut Option<mpsc::Sender<Item>>) {
// Start the task and set up the sender. We only accept
// data in Playing
let (sender_new, receiver) = mpsc::channel(1);
*sender = Some(sender_new);
self.src_pad_handler
.start_task(self.src_pad.as_ref(), receiver);
}
fn pause(&self, element: &gst::Element) -> Result<(), ()> {
let mut sender = self.sender.lock().unwrap();
gst_debug!(SRC_CAT, obj: element, "Pausing");
// Cancel task, we only accept data in Playing
self.src_pad.cancel_task();
// Prevent subsequent items from being enqueued
*sender = None;
gst_debug!(SRC_CAT, obj: element, "Paused");
Ok(())
}
fn stop(&self, element: &gst::Element) -> Result<(), ()> {
gst_debug!(SRC_CAT, obj: element, "Stopping");
// Now stop the task if it was still running, blocking
// until this has actually happened
self.src_pad.stop_task();
gst_debug!(SRC_CAT, obj: element, "Stopped");
Ok(())
}
}
impl ObjectSubclass for ElementSrcTest {
@ -369,7 +334,7 @@ impl ObjectSubclass for ElementSrcTest {
ElementSrcTest {
src_pad,
src_pad_handler: PadSrcHandlerTest::default(),
state: Mutex::new(ElementSrcState::default()),
sender: Mutex::new(None),
settings: Mutex::new(settings),
}
}
@ -388,7 +353,7 @@ impl ObjectImpl for ElementSrcTest {
.expect("type checked upstream")
.unwrap_or_else(|| "".into());
runtime::executor::block_on(self.settings.lock()).context = context;
self.settings.lock().unwrap().context = context;
}
_ => unimplemented!(),
}
@ -412,18 +377,16 @@ impl ElementImpl for ElementSrcTest {
match transition {
gst::StateChange::NullToReady => {
runtime::executor::block_on(self.prepare(element)).map_err(|err| {
self.prepare(element).map_err(|err| {
element.post_error_message(&err);
gst::StateChangeError
})?;
}
gst::StateChange::PlayingToPaused => {
runtime::executor::block_on(self.pause(element))
.map_err(|_| gst::StateChangeError)?;
self.pause(element).map_err(|_| gst::StateChangeError)?;
}
gst::StateChange::ReadyToNull => {
runtime::executor::block_on(self.unprepare(element))
.map_err(|_| gst::StateChangeError)?;
self.unprepare(element).map_err(|_| gst::StateChangeError)?;
}
_ => (),
}
@ -431,11 +394,13 @@ impl ElementImpl for ElementSrcTest {
let mut success = self.parent_change_state(element, transition)?;
match transition {
gst::StateChange::PausedToPlaying => {
runtime::executor::block_on(self.start(element))
.map_err(|_| gst::StateChangeError)?;
gst::StateChange::PausedToReady => {
self.stop(element).map_err(|_| gst::StateChangeError)?;
}
gst::StateChange::ReadyToPaused => {
gst::StateChange::PausedToPlaying => {
self.start(element).map_err(|_| gst::StateChangeError)?;
}
gst::StateChange::ReadyToPaused | gst::StateChange::PlayingToPaused => {
success = gst::StateChangeSuccess::NoPreroll;
}
_ => (),
@ -443,6 +408,26 @@ impl ElementImpl for ElementSrcTest {
Ok(success)
}
fn send_event(&self, element: &gst::Element, event: gst::Event) -> bool {
match event.view() {
EventView::FlushStart(..) => {
// Cancel the task so that it finishes ASAP
// and clear the sender
self.pause(element).unwrap();
}
EventView::FlushStop(..) => {
self.flush_stop(element);
}
_ => (),
}
if !event.is_serialized() {
self.src_pad.gst_pad().push_event(event)
} else {
self.try_push(Item::Event(event)).is_ok()
}
}
}
// Sink
@ -476,21 +461,12 @@ static SINK_PROPERTIES: [glib::subclass::Property; 1] =
)
})];
#[derive(Debug)]
struct PadSinkHandlerTestInner {
flush_join_handle: sync::Mutex<Option<JoinHandle<()>>>,
context: Context,
}
#[derive(Clone, Debug)]
struct PadSinkHandlerTest(Arc<PadSinkHandlerTestInner>);
struct PadSinkHandlerTest;
impl Default for PadSinkHandlerTest {
fn default() -> Self {
PadSinkHandlerTest(Arc::new(PadSinkHandlerTestInner {
flush_join_handle: sync::Mutex::new(None),
context: Context::acquire("PadSinkHandlerTest", THROTTLING_DURATION).unwrap(),
}))
PadSinkHandlerTest
}
}
@ -532,82 +508,57 @@ impl PadSinkHandler for PadSinkHandlerTest {
}
fn sink_event(
&self,
pad: &PadSinkRef,
elem_sink_test: &ElementSinkTest,
element: &gst::Element,
event: gst::Event,
) -> bool {
gst_debug!(SINK_CAT, obj: pad.gst_pad(), "Handling non-serialized {:?}", event);
let ret = match event.view() {
EventView::FlushStart(..) => {
elem_sink_test.stop(&element);
true
}
_ => false,
};
// Should forward item here
ret
}
fn sink_event_serialized(
&self,
pad: &PadSinkRef,
_elem_sink_test: &ElementSinkTest,
element: &gst::Element,
event: gst::Event,
) -> Either<bool, BoxFuture<'static, bool>> {
if event.is_serialized() {
gst_log!(SINK_CAT, obj: pad.gst_pad(), "Handling serialized {:?}", event);
) -> BoxFuture<'static, bool> {
gst_log!(SINK_CAT, obj: pad.gst_pad(), "Handling serialized {:?}", event);
let pad_weak = pad.downgrade();
let element = element.clone();
let inner_weak = Arc::downgrade(&self.0);
let element = element.clone();
async move {
let elem_sink_test = ElementSinkTest::from_instance(&element);
Either::Right(async move {
let elem_sink_test = ElementSinkTest::from_instance(&element);
if let EventView::FlushStop(..) = event.view() {
elem_sink_test.start(&element);
}
if let EventView::FlushStop(..) = event.view() {
let pad = pad_weak
.upgrade()
.expect("PadSink no longer exists in sink_event");
let inner = inner_weak.upgrade().unwrap();
let flush_join_handle = inner.flush_join_handle.lock().unwrap().take();
if let Some(flush_join_handle) = flush_join_handle {
gst_debug!(SINK_CAT, obj: pad.gst_pad(), "Waiting for FlushStart to complete");
if let Ok(()) = flush_join_handle.await {
elem_sink_test.start(&element).await;
} else {
gst_debug!(SINK_CAT, obj: pad.gst_pad(), "FlushStop ignored: FlushStart failed to complete");
}
} else {
gst_debug!(SINK_CAT, obj: pad.gst_pad(), "FlushStop ignored: no Flush in progress");
}
}
elem_sink_test.forward_item(&element, Item::Event(event)).await.is_ok()
}.boxed())
} else {
gst_debug!(SINK_CAT, obj: pad.gst_pad(), "Handling non-serialized {:?}", event);
let ret = match event.view() {
EventView::FlushStart(..) => {
let mut flush_join_handle = self.0.flush_join_handle.lock().unwrap();
if flush_join_handle.is_none() {
gst_log!(SINK_CAT, obj: pad.gst_pad(), "Handling {:?}", event);
let element = element.clone();
*flush_join_handle = Some(self.0.context.spawn(async move {
ElementSinkTest::from_instance(&element)
.stop(&element)
.await;
}));
} else {
gst_debug!(SINK_CAT, obj: pad.gst_pad(), "FlushStart ignored: previous Flush in progress");
}
true
}
_ => false,
};
// Should forward item here
Either::Left(ret)
elem_sink_test
.forward_item(&element, Item::Event(event))
.await
.is_ok()
}
.boxed()
}
}
#[derive(Debug, Default)]
struct ElementSinkState {
flushing: bool,
sender: Option<mpsc::Sender<Item>>,
}
#[derive(Debug)]
struct ElementSinkTest {
sink_pad: PadSink,
state: Mutex<ElementSinkState>,
flushing: AtomicBool,
sender: FutMutex<Option<mpsc::Sender<Item>>>,
}
impl ElementSinkTest {
@ -616,17 +567,17 @@ impl ElementSinkTest {
element: &gst::Element,
item: Item,
) -> Result<gst::FlowSuccess, gst::FlowError> {
let mut state = self.state.lock().await;
if !state.flushing {
if !self.flushing.load(Ordering::SeqCst) {
gst_debug!(SINK_CAT, obj: element, "Fowarding {:?}", item);
state
.sender
self.sender
.lock()
.await
.as_mut()
.expect("Item Sender not set")
.send(item)
.await
.map(|_| gst::FlowSuccess::Ok)
.map_err(|_| gst::FlowError::CustomError)
.map_err(|_| gst::FlowError::Error)
} else {
gst_debug!(
SINK_CAT,
@ -638,16 +589,15 @@ impl ElementSinkTest {
}
}
async fn start(&self, element: &gst::Element) {
fn start(&self, element: &gst::Element) {
gst_debug!(SINK_CAT, obj: element, "Starting");
let mut state = self.state.lock().await;
state.flushing = false;
self.flushing.store(false, Ordering::SeqCst);
gst_debug!(SINK_CAT, obj: element, "Started");
}
async fn stop(&self, element: &gst::Element) {
fn stop(&self, element: &gst::Element) {
gst_debug!(SINK_CAT, obj: element, "Stopping");
self.state.lock().await.flushing = true;
self.flushing.store(true, Ordering::SeqCst);
gst_debug!(SINK_CAT, obj: element, "Stopped");
}
}
@ -687,48 +637,6 @@ impl ObjectSubclass for ElementSinkTest {
klass.add_pad_template(sink_pad_template);
klass.install_properties(&SINK_PROPERTIES);
klass.add_signal_with_class_handler(
"flush-start",
glib::SignalFlags::RUN_LAST | glib::SignalFlags::ACTION,
&[],
bool::static_type(),
|_, args| {
let element = args[0]
.get::<gst::Element>()
.expect("signal arg")
.expect("missing signal arg");
let this = Self::from_instance(&element);
gst_debug!(SINK_CAT, obj: &element, "Pushing FlushStart");
Some(
this.sink_pad
.gst_pad()
.push_event(gst::Event::new_flush_start().build())
.to_value(),
)
},
);
klass.add_signal_with_class_handler(
"flush-stop",
glib::SignalFlags::RUN_LAST | glib::SignalFlags::ACTION,
&[],
bool::static_type(),
|_, args| {
let element = args[0]
.get::<gst::Element>()
.expect("signal arg")
.expect("missing signal arg");
let this = Self::from_instance(&element);
gst_debug!(SINK_CAT, obj: &element, "Pushing FlushStop");
Some(
this.sink_pad
.gst_pad()
.push_event(gst::Event::new_flush_stop(true).build())
.to_value(),
)
},
);
}
fn new_with_class(klass: &glib::subclass::simple::ClassStruct<Self>) -> Self {
@ -737,7 +645,8 @@ impl ObjectSubclass for ElementSinkTest {
ElementSinkTest {
sink_pad,
state: Mutex::new(ElementSinkState::default()),
flushing: AtomicBool::new(true),
sender: FutMutex::new(None),
}
}
}
@ -755,7 +664,7 @@ impl ObjectImpl for ElementSinkTest {
.expect("type checked upstream")
.expect("ItemSender not found")
.clone();
runtime::executor::block_on(self.state.lock()).sender = Some(sender);
*futures::executor::block_on(self.sender.lock()) = Some(sender);
}
_ => unimplemented!(),
}
@ -779,13 +688,13 @@ impl ElementImpl for ElementSinkTest {
match transition {
gst::StateChange::NullToReady => {
runtime::executor::block_on(self.sink_pad.prepare(&PadSinkHandlerTest::default()));
self.sink_pad.prepare(&PadSinkHandlerTest::default());
}
gst::StateChange::PausedToReady => {
runtime::executor::block_on(self.stop(element));
self.stop(element);
}
gst::StateChange::ReadyToNull => {
runtime::executor::block_on(self.sink_pad.unprepare());
self.sink_pad.unprepare();
}
_ => (),
}
@ -793,7 +702,7 @@ impl ElementImpl for ElementSinkTest {
let success = self.parent_change_state(element, transition)?;
if transition == gst::StateChange::ReadyToPaused {
runtime::executor::block_on(self.start(element));
self.start(element);
}
Ok(success)
@ -863,26 +772,15 @@ fn nominal_scenario(
pipeline.set_state(gst::State::Playing).unwrap();
// Initial events
block_on(
elem_src_test.try_push(Item::Event(
elem_src_test
.try_push(Item::Event(
gst::Event::new_stream_start(scenario_name)
.group_id(gst::GroupId::next())
.build(),
)),
)
.unwrap();
))
.unwrap();
match block_on(receiver.next()).unwrap() {
Item::Event(event) => match event.view() {
EventView::CustomDownstreamSticky(e) => {
assert!(PadContext::is_pad_context_sticky_event(&e))
}
other => panic!("Unexpected event {:?}", other),
},
other => panic!("Unexpected item {:?}", other),
}
match block_on(receiver.next()).unwrap() {
match futures::executor::block_on(receiver.next()).unwrap() {
Item::Event(event) => match event.view() {
EventView::StreamStart(_) => (),
other => panic!("Unexpected event {:?}", other),
@ -890,12 +788,13 @@ fn nominal_scenario(
other => panic!("Unexpected item {:?}", other),
}
block_on(elem_src_test.try_push(Item::Event(
gst::Event::new_segment(&gst::FormattedSegment::<gst::format::Time>::new()).build(),
)))
.unwrap();
elem_src_test
.try_push(Item::Event(
gst::Event::new_segment(&gst::FormattedSegment::<gst::format::Time>::new()).build(),
))
.unwrap();
match block_on(receiver.next()).unwrap() {
match futures::executor::block_on(receiver.next()).unwrap() {
Item::Event(event) => match event.view() {
EventView::Segment(_) => (),
other => panic!("Unexpected event {:?}", other),
@ -904,10 +803,11 @@ fn nominal_scenario(
}
// Buffer
block_on(elem_src_test.try_push(Item::Buffer(gst::Buffer::from_slice(vec![1, 2, 3, 4]))))
elem_src_test
.try_push(Item::Buffer(gst::Buffer::from_slice(vec![1, 2, 3, 4])))
.unwrap();
match block_on(receiver.next()).unwrap() {
match futures::executor::block_on(receiver.next()).unwrap() {
Item::Buffer(buffer) => {
let data = buffer.map_readable().unwrap();
assert_eq!(data.as_slice(), vec![1, 2, 3, 4].as_slice());
@ -920,9 +820,9 @@ fn nominal_scenario(
list.get_mut()
.unwrap()
.add(gst::Buffer::from_slice(vec![1, 2, 3, 4]));
block_on(elem_src_test.try_push(Item::BufferList(list))).unwrap();
elem_src_test.try_push(Item::BufferList(list)).unwrap();
match block_on(receiver.next()).unwrap() {
match futures::executor::block_on(receiver.next()).unwrap() {
Item::BufferList(_) => (),
other => panic!("Unexpected item {:?}", other),
}
@ -931,7 +831,8 @@ fn nominal_scenario(
pipeline.set_state(gst::State::Paused).unwrap();
// Items not longer accepted
block_on(elem_src_test.try_push(Item::Buffer(gst::Buffer::from_slice(vec![1, 2, 3, 4]))))
elem_src_test
.try_push(Item::Buffer(gst::Buffer::from_slice(vec![1, 2, 3, 4])))
.unwrap_err();
// Nothing forwarded
@ -944,11 +845,10 @@ fn nominal_scenario(
receiver.try_next().unwrap_err();
// Flush
block_on(elem_src_test.try_push(Item::Event(gst::Event::new_flush_start().build()))).unwrap();
block_on(elem_src_test.try_push(Item::Event(gst::Event::new_flush_stop(false).build())))
.unwrap();
src_element.send_event(gst::Event::new_flush_start().build());
src_element.send_event(gst::Event::new_flush_stop(true).build());
match block_on(receiver.next()).unwrap() {
match futures::executor::block_on(receiver.next()).unwrap() {
Item::Event(event) => match event.view() {
EventView::FlushStop(_) => (),
other => panic!("Unexpected event {:?}", other),
@ -957,9 +857,11 @@ fn nominal_scenario(
}
// EOS
block_on(elem_src_test.try_push(Item::Event(gst::Event::new_eos().build()))).unwrap();
elem_src_test
.try_push(Item::Event(gst::Event::new_eos().build()))
.unwrap();
match block_on(receiver.next()).unwrap() {
match futures::executor::block_on(receiver.next()).unwrap() {
Item::Event(event) => match event.view() {
EventView::Eos(_) => (),
other => panic!("Unexpected event {:?}", other),
@ -970,14 +872,15 @@ fn nominal_scenario(
pipeline.set_state(gst::State::Ready).unwrap();
// Receiver was dropped when stopping => can't send anymore
block_on(
elem_src_test.try_push(Item::Event(
elem_src_test
.try_push(Item::Event(
gst::Event::new_stream_start(&format!("{}_past_stop", scenario_name))
.group_id(gst::GroupId::next())
.build(),
)),
)
.unwrap_err();
))
.unwrap_err();
pipeline.set_state(gst::State::Null).unwrap();
}
#[test]
@ -989,24 +892,24 @@ fn src_sink_nominal() {
nominal_scenario(&name, pipeline, src_element, receiver);
}
#[test]
fn src_tsqueue_sink_nominal() {
init();
let name = "src_tsqueue_sink";
let ts_queue = gst::ElementFactory::make("ts-queue", Some("ts-queue")).unwrap();
ts_queue
.set_property("context", &format!("{}_queue", name))
.unwrap();
ts_queue
.set_property("context-wait", &THROTTLING_DURATION)
.unwrap();
let (pipeline, src_element, _sink_element, receiver) = setup(name, Some(ts_queue), None);
nominal_scenario(&name, pipeline, src_element, receiver);
}
// #[test]
// fn src_tsqueue_sink_nominal() {
// init();
//
// let name = "src_tsqueue_sink";
//
// let ts_queue = gst::ElementFactory::make("ts-queue", Some("ts-queue")).unwrap();
// ts_queue
// .set_property("context", &format!("{}_queue", name))
// .unwrap();
// ts_queue
// .set_property("context-wait", &THROTTLING_DURATION)
// .unwrap();
//
// let (pipeline, src_element, _sink_element, receiver) = setup(name, Some(ts_queue), None);
//
// nominal_scenario(&name, pipeline, src_element, receiver);
// }
#[test]
fn src_queue_sink_nominal() {
@ -1020,30 +923,30 @@ fn src_queue_sink_nominal() {
nominal_scenario(&name, pipeline, src_element, receiver);
}
#[test]
fn src_tsproxy_sink_nominal() {
init();
let name = "src_tsproxy_sink";
let ts_proxy_sink = gst::ElementFactory::make("ts-proxysink", Some("ts-proxysink")).unwrap();
ts_proxy_sink
.set_property("proxy-context", &format!("{}_proxy_context", name))
.unwrap();
let ts_proxy_src = gst::ElementFactory::make("ts-proxysrc", Some("ts-proxysrc")).unwrap();
ts_proxy_src
.set_property("proxy-context", &format!("{}_proxy_context", name))
.unwrap();
ts_proxy_src
.set_property("context", &format!("{}_context", name))
.unwrap();
ts_proxy_src
.set_property("context-wait", &THROTTLING_DURATION)
.unwrap();
let (pipeline, src_element, _sink_element, receiver) =
setup(name, Some(ts_proxy_sink), Some(ts_proxy_src));
nominal_scenario(&name, pipeline, src_element, receiver);
}
// #[test]
// fn src_tsproxy_sink_nominal() {
// init();
//
// let name = "src_tsproxy_sink";
//
// let ts_proxy_sink = gst::ElementFactory::make("ts-proxysink", Some("ts-proxysink")).unwrap();
// ts_proxy_sink
// .set_property("proxy-context", &format!("{}_proxy_context", name))
// .unwrap();
//
// let ts_proxy_src = gst::ElementFactory::make("ts-proxysrc", Some("ts-proxysrc")).unwrap();
// ts_proxy_src
// .set_property("proxy-context", &format!("{}_proxy_context", name))
// .unwrap();
// ts_proxy_src
// .set_property("context", &format!("{}_context", name))
// .unwrap();
// ts_proxy_src
// .set_property("context-wait", &THROTTLING_DURATION)
// .unwrap();
//
// let (pipeline, src_element, _sink_element, receiver) =
// setup(name, Some(ts_proxy_sink), Some(ts_proxy_src));
//
// nominal_scenario(&name, pipeline, src_element, receiver);
// }