transcriberbin: notify passthrough at the appropriate time

We want to enable passthrough internally, and only notify that
internally it has been enabled once the transcriber has been unlinked.

This way applications connected to the notify handler can synchronously
update the properties and attempt to disable passthrough again.

Doing so properly requires a refactoring of the transition to the
passthrough state, with the currently set passthrough mode maintained
separately from the target passthrough state.

This commit also finishes the work left incomplete in
17d7997137 by moving the passthrough
property to the sink pad class, making each transcriber passthrough
state independent from the others.

Also adds an example to demonstrate the behavior

Part-of: <https://gitlab.freedesktop.org/gstreamer/gst-plugins-rs/-/merge_requests/1856>
This commit is contained in:
Mathieu Duponchelle 2024-10-16 19:37:39 +02:00 committed by GStreamer Marge Bot
parent 5f8e8b4873
commit 635a83a8d7
5 changed files with 478 additions and 166 deletions

1
Cargo.lock generated
View file

@ -2481,6 +2481,7 @@ dependencies = [
"cea608-types", "cea608-types",
"cea708-types", "cea708-types",
"chrono", "chrono",
"clap",
"either", "either",
"gst-plugin-version-helper", "gst-plugin-version-helper",
"gstreamer", "gstreamer",

View file

@ -7759,18 +7759,6 @@
"type": "GstTranscriberBinMuxMethod", "type": "GstTranscriberBinMuxMethod",
"writable": true "writable": true
}, },
"passthrough": {
"blurb": "Whether transcription should occur",
"conditionally-available": false,
"construct": false,
"construct-only": false,
"controllable": false,
"default": "false",
"mutable": "playing",
"readable": true,
"type": "gboolean",
"writable": true
},
"transcriber": { "transcriber": {
"blurb": "The transcriber element to use", "blurb": "The transcriber element to use",
"conditionally-available": false, "conditionally-available": false,
@ -8153,6 +8141,18 @@
"type": "GstTtToCea608Mode", "type": "GstTtToCea608Mode",
"writable": true "writable": true
}, },
"passthrough": {
"blurb": "Whether transcription should occur",
"conditionally-available": false,
"construct": false,
"construct-only": false,
"controllable": false,
"default": "false",
"mutable": "playing",
"readable": true,
"type": "gboolean",
"writable": true
},
"transcriber": { "transcriber": {
"blurb": "The transcriber element to use", "blurb": "The transcriber element to use",
"conditionally-available": false, "conditionally-available": false,

View file

@ -33,6 +33,7 @@ bitstream-io = "2.3"
pretty_assertions = "1" pretty_assertions = "1"
rand = { version = "0.8", features = ["small_rng"] } rand = { version = "0.8", features = ["small_rng"] }
gst-check.workspace = true gst-check.workspace = true
clap = { version = "4", features = ["derive"] }
[lib] [lib]
name = "gstrsclosedcaption" name = "gstrsclosedcaption"
@ -62,3 +63,6 @@ import_library = false
[package.metadata.capi.pkg_config] [package.metadata.capi.pkg_config]
requires_private = "gstreamer-1.0, gstreamer-base-1.0, gstreamer-video-1.0, gobject-2.0, glib-2.0, gmodule-2.0, pango, pangocairo, cairo-gobject" requires_private = "gstreamer-1.0, gstreamer-base-1.0, gstreamer-video-1.0, gobject-2.0, glib-2.0, gmodule-2.0, pango, pangocairo, cairo-gobject"
[[example]]
name = "passthrough-notify"

View file

@ -0,0 +1,184 @@
// This example creates a pipeline that will decode a file, pipe its audio
// and video streams through transcriberbin, and display the result
// with closed captions overlaid.
//
// At first the (AWS) transcriber will not have its access keys set, which
// means it should automatically go to passthrough=true.
//
// At this point we set the access keys and disable passthrough again.
//
// The expected result is for the terminal to display
// "Access key set, disabling passthrough" and for closed captions to be
// overlaid over the video.
use anyhow::Error;
use clap::Parser;
use gst::glib;
use gst::prelude::*;
#[derive(Debug, Default, Clone, clap::Parser)]
struct Args {
#[clap(long, help = "URI to transcribe")]
pub uri: String,
#[clap(long, help = "Access key ID")]
pub access_key_id: String,
#[clap(long, help = "Secret access key")]
pub secret_access_key: String,
}
fn link_video_stream(
pipeline: &gst::Pipeline,
transcriberbin: &gst::Element,
pad: &gst::Pad,
) -> Result<(), Error> {
let conv = gst::ElementFactory::make("videoconvert").build()?;
pipeline.add(&conv)?;
conv.sync_state_with_parent()?;
pad.link(&conv.static_pad("sink").unwrap())?;
conv.link_pads(None, transcriberbin, Some("sink_video"))?;
Ok(())
}
fn link_audio_stream(
pipeline: &gst::Pipeline,
transcriberbin: &gst::Element,
pad: &gst::Pad,
) -> Result<(), Error> {
let conv = gst::ElementFactory::make("audioconvert").build()?;
pipeline.add(&conv)?;
conv.sync_state_with_parent()?;
pad.link(&conv.static_pad("sink").unwrap())?;
conv.link_pads(None, transcriberbin, Some("sink_audio"))?;
Ok(())
}
fn main() -> Result<(), Error> {
let args = Args::parse();
gst::init()?;
let pipeline = gst::Pipeline::builder().build();
let uridecodebin = gst::ElementFactory::make("uridecodebin")
.property("uri", &args.uri)
.build()?;
let transcriberbin = gst::ElementFactory::make("transcriberbin").build()?;
let asink = gst::ElementFactory::make("fakesink").build()?;
let overlay = gst::ElementFactory::make("cea608overlay").build()?;
let vconv = gst::ElementFactory::make("videoconvert").build()?;
let vsink = gst::ElementFactory::make("autovideosink").build()?;
uridecodebin.connect_pad_added(glib::clone!(
#[weak]
pipeline,
#[weak]
transcriberbin,
move |_element, pad| {
if pad
.current_caps()
.map(|c| c.structure(0).unwrap().name().starts_with("video/"))
.unwrap_or(false)
{
link_video_stream(&pipeline, &transcriberbin, pad)
.expect("Failed to link video stream");
} else {
link_audio_stream(&pipeline, &transcriberbin, pad)
.expect("Failed to link audio stream");
}
}
));
transcriberbin
.static_pad("sink_audio")
.unwrap()
.connect_closure(
"notify::passthrough",
false,
glib::closure!(
#[strong]
args,
move |pad: &gst::Pad, _pspec: &gst::glib::ParamSpec| {
let passthrough = pad.property::<bool>("passthrough");
if passthrough {
let transcriber = pad.property::<gst::Element>("transcriber");
transcriber.set_property("access-key", &args.access_key_id);
transcriber.set_property("secret-access-key", &args.secret_access_key);
eprintln!(
"Access key set, disabling passthrough, transcriber state: {:?}",
transcriber.state(gst::ClockTime::NONE)
);
pad.set_property("passthrough", false);
}
}
),
);
pipeline.add_many([
&uridecodebin,
&transcriberbin,
&asink,
&overlay,
&vconv,
&vsink,
])?;
transcriberbin.link_pads(Some("src_audio"), &asink, None)?;
transcriberbin.link_pads(Some("src_video"), &overlay, None)?;
gst::Element::link_many([&overlay, &vconv, &vsink])?;
pipeline.set_state(gst::State::Playing)?;
let bus = pipeline.bus().expect("Pipeline should have a bus");
for msg in bus.iter_timed(gst::ClockTime::NONE) {
use gst::MessageView;
match msg.view() {
MessageView::Eos(..) => {
println!("EOS");
break;
}
MessageView::StateChanged(sc) => {
if msg.src() == Some(pipeline.upcast_ref()) {
pipeline.debug_to_dot_file(
gst::DebugGraphDetails::all(),
format!("{}-{:?}-{:?}", pipeline.name(), sc.old(), sc.current()),
);
}
}
MessageView::Error(err) => {
pipeline.debug_to_dot_file(gst::DebugGraphDetails::ALL, "error");
pipeline.set_state(gst::State::Null)?;
eprintln!(
"Got error from {}: {} ({})",
msg.src()
.map(|s| String::from(s.path_string()))
.unwrap_or_else(|| "None".into()),
err.error(),
err.debug().unwrap_or_else(|| "".into()),
);
break;
}
_ => (),
}
}
pipeline.set_state(gst::State::Null)?;
Ok(())
}

View file

@ -38,6 +38,12 @@ const DEFAULT_MUX_METHOD: MuxMethod = MuxMethod::Cea608;
const CEAX08MUX_LATENCY: gst::ClockTime = gst::ClockTime::from_mseconds(100); const CEAX08MUX_LATENCY: gst::ClockTime = gst::ClockTime::from_mseconds(100);
enum TargetPassthroughState {
None,
Enabled,
Disabled,
}
/* One per language, including original */ /* One per language, including original */
struct TranscriptionChannel { struct TranscriptionChannel {
bin: gst::Bin, bin: gst::Bin,
@ -79,7 +85,6 @@ impl TranscriptionChannel {
struct State { struct State {
mux_method: MuxMethod, mux_method: MuxMethod,
framerate: Option<gst::Fraction>, framerate: Option<gst::Fraction>,
tearing_down: usize,
internal_bin: gst::Bin, internal_bin: gst::Bin,
video_queue: gst::Element, video_queue: gst::Element,
ccmux: gst::Element, ccmux: gst::Element,
@ -96,7 +101,6 @@ struct Settings {
cc_caps: gst::Caps, cc_caps: gst::Caps,
latency: gst::ClockTime, latency: gst::ClockTime,
translate_latency: gst::ClockTime, translate_latency: gst::ClockTime,
passthrough: bool,
accumulate_time: gst::ClockTime, accumulate_time: gst::ClockTime,
caption_source: CaptionSource, caption_source: CaptionSource,
mux_method: MuxMethod, mux_method: MuxMethod,
@ -108,7 +112,6 @@ impl Default for Settings {
cc_caps: gst::Caps::builder("closedcaption/x-cea-608") cc_caps: gst::Caps::builder("closedcaption/x-cea-608")
.field("format", "raw") .field("format", "raw")
.build(), .build(),
passthrough: DEFAULT_PASSTHROUGH,
latency: DEFAULT_LATENCY, latency: DEFAULT_LATENCY,
translate_latency: DEFAULT_TRANSLATE_LATENCY, translate_latency: DEFAULT_TRANSLATE_LATENCY,
accumulate_time: DEFAULT_ACCUMULATE, accumulate_time: DEFAULT_ACCUMULATE,
@ -536,35 +539,32 @@ impl TranscriberBin {
.set_property("max-size-time", max_size_time); .set_property("max-size-time", max_size_time);
} }
if !settings.passthrough { gst::debug!(
gst::debug!( CAT,
CAT, imp = self,
imp = self, "Linking transcription bins and synchronizing state"
"Linking transcription bins and synchronizing state" );
); state
state .transcription_bin
.link_pads(Some("src"), &state.cccombiner, Some("caption"))
.unwrap();
state.transcription_bin.set_locked_state(false);
state.transcription_bin.sync_state_with_parent().unwrap();
for pad in state.audio_sink_pads.values() {
let ps = pad.imp().state.lock().unwrap();
let pad_state = ps.as_ref().unwrap();
pad_state.transcription_bin.set_locked_state(false);
pad_state
.transcription_bin .transcription_bin
.link_pads(Some("src"), &state.cccombiner, Some("caption")) .sync_state_with_parent()
.unwrap(); .unwrap();
let transcription_sink_pad = state.transcription_bin.static_pad(&pad.name()).unwrap();
state.transcription_bin.set_locked_state(false); // Might be linked already if "translation-languages" is set
state.transcription_bin.sync_state_with_parent().unwrap(); if transcription_sink_pad.peer().is_none() {
let audio_tee_pad = pad_state.audio_tee.request_pad_simple("src_%u").unwrap();
for pad in state.audio_sink_pads.values() { audio_tee_pad.link(&transcription_sink_pad).unwrap();
let ps = pad.imp().state.lock().unwrap();
let pad_state = ps.as_ref().unwrap();
pad_state.transcription_bin.set_locked_state(false);
pad_state
.transcription_bin
.sync_state_with_parent()
.unwrap();
let transcription_sink_pad =
state.transcription_bin.static_pad(&pad.name()).unwrap();
// Might be linked already if "translation-languages" is set
if transcription_sink_pad.peer().is_none() {
let audio_tee_pad = pad_state.audio_tee.request_pad_simple("src_%u").unwrap();
audio_tee_pad.link(&transcription_sink_pad).unwrap();
}
} }
} }
@ -576,92 +576,161 @@ impl TranscriberBin {
} }
} }
fn disable_transcription_bin(&self, state: &mut State) { fn disable_transcription_bin(
// At this point, we want to check whether passthrough &self,
// has been unset in the meantime pad: &super::TranscriberSinkPad,
let passthrough = self.settings.lock().unwrap().passthrough; state: &mut State,
pad_state: &mut TranscriberSinkPadState,
) {
gst::debug!(CAT, imp = self, "disabling transcription bin");
if passthrough { let bin_sink_pad = state.transcription_bin.static_pad(&pad.name()).unwrap();
gst::debug!(CAT, imp = self, "disabling transcription bin"); if let Some(audio_tee_pad) = bin_sink_pad.peer() {
audio_tee_pad.unlink(&bin_sink_pad).unwrap();
for pad in state.audio_sink_pads.values() { pad_state.audio_tee.release_request_pad(&audio_tee_pad);
let ps = pad.imp().state.lock().unwrap();
let pad_state = ps.as_ref().unwrap();
let bin_sink_pad = state.transcription_bin.static_pad(&pad.name()).unwrap();
if let Some(audio_tee_pad) = bin_sink_pad.peer() {
audio_tee_pad.unlink(&bin_sink_pad).unwrap();
pad_state.audio_tee.release_request_pad(&audio_tee_pad);
}
}
let bin_src_pad = state.transcription_bin.static_pad("src").unwrap();
if let Some(cccombiner_pad) = bin_src_pad.peer() {
bin_src_pad.unlink(&cccombiner_pad).unwrap();
state.cccombiner.release_request_pad(&cccombiner_pad);
}
state.transcription_bin.set_locked_state(true);
state.transcription_bin.set_state(gst::State::Null).unwrap();
} }
for channel in pad_state.transcription_channels.values() {
let srcpad = pad_state
.transcription_bin
.static_pad(&format!("src_{}", channel.language))
.unwrap();
if let Some(sinkpad) = state.ccmux.static_pad(&channel.ccmux_pad_name) {
srcpad.unlink(&sinkpad).unwrap();
state.ccmux.release_request_pad(&sinkpad);
}
}
pad_state.transcription_bin.set_locked_state(true);
pad_state
.transcription_bin
.set_state(gst::State::Null)
.unwrap();
} }
fn block_and_update(&self, passthrough: bool) { fn enable_transcription_bin(
let mut s = self.state.lock().unwrap(); &self,
sinkpad: &TranscriberSinkPad,
state: &mut State,
pad_state: &mut TranscriberSinkPadState,
) {
for channel in pad_state.transcription_channels.values() {
let srcpad = pad_state
.transcription_bin
.static_pad(&format!("src_{}", channel.language))
.unwrap();
let sinkpad = state
.ccmux
.static_pad(&channel.ccmux_pad_name)
.unwrap_or_else(|| {
state
.ccmux
.request_pad_simple(&channel.ccmux_pad_name)
.unwrap()
});
srcpad.link(&sinkpad).unwrap();
}
pad_state.transcription_bin.set_locked_state(false);
pad_state
.transcription_bin
.sync_state_with_parent()
.unwrap();
let audio_tee_pad = pad_state.audio_tee.request_pad_simple("src_%u").unwrap();
let transcription_sink_pad = state
.transcription_bin
.static_pad(&sinkpad.obj().name())
.unwrap();
audio_tee_pad.link(&transcription_sink_pad).unwrap();
pad_state.target_passthrough_state = TargetPassthroughState::None;
}
if let Some(ref mut state) = s.as_mut() { fn block_and_update(
if passthrough { &self,
state.tearing_down = state.audio_sink_pads.len(); sinkpad: &TranscriberSinkPad,
let sinkpads = state.audio_sink_pads.clone(); mut s: std::sync::MutexGuard<Option<State>>,
mut ps: std::sync::MutexGuard<Result<TranscriberSinkPadState, Error>>,
) {
let Some(ref mut state) = s.as_mut() else {
return;
};
let Ok(ref mut pad_state) = ps.as_mut() else {
return;
};
match pad_state.target_passthrough_state {
TargetPassthroughState::Enabled => {
drop(s); drop(s);
for sinkpad in sinkpads.values() { drop(ps);
let imp_weak = self.downgrade(); let imp_weak = self.downgrade();
let _ = sinkpad.add_probe( let _ = sinkpad.obj().add_probe(
gst::PadProbeType::IDLE gst::PadProbeType::IDLE
| gst::PadProbeType::BUFFER | gst::PadProbeType::BUFFER
| gst::PadProbeType::EVENT_DOWNSTREAM, | gst::PadProbeType::EVENT_DOWNSTREAM,
move |_pad, _info| { move |pad, _info| {
let Some(imp) = imp_weak.upgrade() else { let Some(imp) = imp_weak.upgrade() else {
return gst::PadProbeReturn::Remove; return gst::PadProbeReturn::Remove;
}; };
let mut s = imp.state.lock().unwrap(); let mut s = imp.state.lock().unwrap();
if let Some(ref mut state) = s.as_mut() { let Some(ref mut state) = s.as_mut() else {
state.tearing_down -= 1; return gst::PadProbeReturn::Remove;
if state.tearing_down == 0 { };
imp.disable_transcription_bin(state);
let pad_imp = pad
.downcast_ref::<super::TranscriberSinkPad>()
.unwrap()
.imp();
let mut ps = pad_imp.state.lock().unwrap();
let Ok(ref mut pad_state) = ps.as_mut() else {
return gst::PadProbeReturn::Remove;
};
match pad_state.target_passthrough_state {
TargetPassthroughState::Enabled => {
imp.disable_transcription_bin(pad, state, pad_state);
pad_state.target_passthrough_state = TargetPassthroughState::None;
// Now that we are done, make sure that this is reflected in our settings
let notify = {
let mut pad_settings = pad_imp.settings.lock().unwrap();
let old_passthrough = pad_settings.passthrough;
pad_settings.passthrough = true;
!old_passthrough
};
if notify {
// We cannot notify from this thread, as this could cause
// the user to reset the passthrough property, which would
// in turn hang at link time (the probe would fire again
// and deadlock)
let pad_weak = pad.downgrade();
imp.obj().call_async(move |_| {
let Some(pad) = pad_weak.upgrade() else {
return;
};
pad.notify("passthrough");
});
} }
} }
TargetPassthroughState::Disabled => {
// Now at this point, even though we initially blocked in
// order to enable passthrough, the user may instead have
// requested disabling it in the meantime.
imp.enable_transcription_bin(pad_imp, state, pad_state);
}
_ => (),
}
gst::PadProbeReturn::Remove gst::PadProbeReturn::Remove
}, },
); );
}
} else if state.tearing_down > 0 {
// Do nothing, wait for the previous transcription bin
// to finish tearing down
} else {
state
.transcription_bin
.link_pads(Some("src"), &state.cccombiner, Some("caption"))
.unwrap();
state.transcription_bin.set_locked_state(false);
state.transcription_bin.sync_state_with_parent().unwrap();
for pad in state.audio_sink_pads.values() {
let ps = pad.imp().state.lock().unwrap();
let pad_state = ps.as_ref().unwrap();
pad_state.transcription_bin.set_locked_state(false);
pad_state
.transcription_bin
.sync_state_with_parent()
.unwrap();
let audio_tee_pad = pad_state.audio_tee.request_pad_simple("src_%u").unwrap();
let transcription_sink_pad =
state.transcription_bin.static_pad(&pad.name()).unwrap();
audio_tee_pad.link(&transcription_sink_pad).unwrap();
}
} }
TargetPassthroughState::Disabled => {
self.enable_transcription_bin(sinkpad, state, pad_state);
}
_ => (),
} }
} }
@ -857,7 +926,7 @@ impl TranscriberBin {
} }
if lang_code_only { if lang_code_only {
if !settings.passthrough { if !pad_settings.passthrough {
gst::debug!(CAT, imp = self, "Syncing state with parent"); gst::debug!(CAT, imp = self, "Syncing state with parent");
drop(settings); drop(settings);
@ -942,7 +1011,7 @@ impl TranscriberBin {
self.setup_cc_mode(&pad.obj(), pad_state, state.mux_method, pad_settings.mode); self.setup_cc_mode(&pad.obj(), pad_state, state.mux_method, pad_settings.mode);
if !settings.passthrough { if !pad_settings.passthrough {
gst::debug!(CAT, imp = self, "Syncing state with parent"); gst::debug!(CAT, imp = self, "Syncing state with parent");
let audio_tee_pad = pad_state.audio_tee.request_pad_simple("src_%u").unwrap(); let audio_tee_pad = pad_state.audio_tee.request_pad_simple("src_%u").unwrap();
@ -1036,6 +1105,16 @@ impl TranscriberBin {
false false
} }
fn all_sinks_are_passthrough(&self, state: &State) -> bool {
for pad in state.audio_sink_pads.values() {
let pad_settings = pad.imp().settings.lock().unwrap();
if !pad_settings.passthrough {
return false;
}
}
true
}
#[allow(clippy::single_match)] #[allow(clippy::single_match)]
fn src_query(&self, pad: &gst::Pad, query: &mut gst::QueryRef) -> bool { fn src_query(&self, pad: &gst::Pad, query: &mut gst::QueryRef) -> bool {
use gst::QueryViewMut; use gst::QueryViewMut;
@ -1051,16 +1130,20 @@ impl TranscriberBin {
if ret { if ret {
let (_, mut min, _) = upstream_query.result(); let (_, mut min, _) = upstream_query.result();
let state = self.state.lock().unwrap(); let state = self.state.lock().unwrap();
let (received_framerate, translating) = { let (received_framerate, translating, all_passthrough) = {
if let Some(state) = state.as_ref() { if let Some(state) = state.as_ref() {
(state.framerate, self.any_sink_is_translating(state)) (
state.framerate,
self.any_sink_is_translating(state),
self.all_sinks_are_passthrough(state),
)
} else { } else {
(None, false) (None, false, true)
} }
}; };
let settings = self.settings.lock().unwrap(); let settings = self.settings.lock().unwrap();
if settings.passthrough || received_framerate.is_none() { if all_passthrough || received_framerate.is_none() {
min += settings.latency + settings.accumulate_time + CEAX08MUX_LATENCY; min += settings.latency + settings.accumulate_time + CEAX08MUX_LATENCY;
if translating { if translating {
@ -1103,12 +1186,23 @@ impl TranscriberBin {
let settings = self.settings.lock().unwrap(); let settings = self.settings.lock().unwrap();
let mux_method = settings.mux_method; let mux_method = settings.mux_method;
let has_force_live = |factory_name: &str| {
gst::ElementFactory::find(factory_name)
.and_then(|f| f.load().ok())
.map(|f| f.element_type())
.and_then(glib::Class::<gst::Element>::from_type)
.map(|k| k.has_property("force-live", Some(bool::static_type())))
.unwrap_or(false)
};
let ccmux = match mux_method { let ccmux = match mux_method {
MuxMethod::Cea608 => gst::ElementFactory::make("cea608mux") MuxMethod::Cea608 => gst::ElementFactory::make("cea608mux")
.property_from_str("start-time-selection", "first") .property_from_str("start-time-selection", "first")
.property_if("force-live", true, has_force_live("cea608mux"))
.build()?, .build()?,
MuxMethod::Cea708 => gst::ElementFactory::make("cea708mux") MuxMethod::Cea708 => gst::ElementFactory::make("cea708mux")
.property_from_str("start-time-selection", "first") .property_from_str("start-time-selection", "first")
.property_if("force-live", true, has_force_live("cea708mux"))
.build()?, .build()?,
}; };
let ccmux_filter = gst::ElementFactory::make("capsfilter").build()?; let ccmux_filter = gst::ElementFactory::make("capsfilter").build()?;
@ -1142,7 +1236,6 @@ impl TranscriberBin {
transcription_bin, transcription_bin,
cccapsfilter, cccapsfilter,
transcription_valve, transcription_valve,
tearing_down: 0,
audio_serial: 0, audio_serial: 0,
audio_sink_pads, audio_sink_pads,
}) })
@ -1273,12 +1366,6 @@ impl ObjectImpl for TranscriberBin {
fn properties() -> &'static [glib::ParamSpec] { fn properties() -> &'static [glib::ParamSpec] {
static PROPERTIES: LazyLock<Vec<glib::ParamSpec>> = LazyLock::new(|| { static PROPERTIES: LazyLock<Vec<glib::ParamSpec>> = LazyLock::new(|| {
vec![ vec![
glib::ParamSpecBoolean::builder("passthrough")
.nick("Passthrough")
.blurb("Whether transcription should occur")
.default_value(DEFAULT_PASSTHROUGH)
.mutable_playing()
.build(),
glib::ParamSpecUInt::builder("latency") glib::ParamSpecUInt::builder("latency")
.nick("Latency") .nick("Latency")
.blurb("Amount of milliseconds to allow the transcriber") .blurb("Amount of milliseconds to allow the transcriber")
@ -1346,18 +1433,6 @@ impl ObjectImpl for TranscriberBin {
fn set_property(&self, _id: usize, value: &glib::Value, pspec: &glib::ParamSpec) { fn set_property(&self, _id: usize, value: &glib::Value, pspec: &glib::ParamSpec) {
match pspec.name() { match pspec.name() {
"passthrough" => {
let mut settings = self.settings.lock().unwrap();
let old_passthrough = settings.passthrough;
let new_passthrough = value.get().expect("type checked upstream");
settings.passthrough = new_passthrough;
if old_passthrough != new_passthrough {
drop(settings);
self.block_and_update(new_passthrough);
}
}
"latency" => { "latency" => {
let mut settings = self.settings.lock().unwrap(); let mut settings = self.settings.lock().unwrap();
settings.latency = gst::ClockTime::from_mseconds( settings.latency = gst::ClockTime::from_mseconds(
@ -1437,10 +1512,6 @@ impl ObjectImpl for TranscriberBin {
fn property(&self, _id: usize, pspec: &glib::ParamSpec) -> glib::Value { fn property(&self, _id: usize, pspec: &glib::ParamSpec) -> glib::Value {
match pspec.name() { match pspec.name() {
"passthrough" => {
let settings = self.settings.lock().unwrap();
settings.passthrough.to_value()
}
"latency" => { "latency" => {
let settings = self.settings.lock().unwrap(); let settings = self.settings.lock().unwrap();
(settings.latency.mseconds() as u32).to_value() (settings.latency.mseconds() as u32).to_value()
@ -1798,27 +1869,42 @@ impl BinImpl for TranscriberBin {
match msg.view() { match msg.view() {
MessageView::Error(m) => { MessageView::Error(m) => {
let pad = self let mut s = self.state.lock().unwrap();
.audio_sinkpad let Some(state) = s.as_mut() else {
.downcast_ref::<super::TranscriberSinkPad>() drop(s);
.unwrap(); self.parent_handle_message(msg);
let ps = pad.imp().state.lock().unwrap(); return;
let pad_state = ps.as_ref().unwrap(); };
if msg.src() == pad_state.transcriber.as_ref().map(|t| t.upcast_ref()) { let mut handled = false;
gst::error!( for pad in state.audio_sink_pads.values() {
CAT, let mut ps = pad.imp().state.lock().unwrap();
imp = self, let Ok(pad_state) = ps.as_mut() else {
"Transcriber has posted an error ({m:?}), going back to passthrough", continue;
); };
drop(ps); if msg.src() == pad_state.transcriber.as_ref().map(|t| t.upcast_ref()) {
self.settings.lock().unwrap().passthrough = true; gst::error!(
self.obj().notify("passthrough"); CAT,
self.obj().call_async(move |bin| { imp = self,
let thiz = bin.imp(); "Transcriber has posted an error ({m:?}), going back to passthrough",
thiz.block_and_update(true); );
}); pad_state.target_passthrough_state = TargetPassthroughState::Enabled;
} else { let pad_weak = pad.downgrade();
drop(ps); self.obj().call_async(move |bin| {
let Some(pad) = pad_weak.upgrade() else {
return;
};
let thiz = bin.imp();
let s = thiz.state.lock().unwrap();
let ps = pad.imp().state.lock().unwrap();
thiz.block_and_update(pad.imp(), s, ps);
});
handled = true;
break;
}
}
if !handled {
drop(s);
self.parent_handle_message(msg); self.parent_handle_message(msg);
} }
} }
@ -1832,6 +1918,7 @@ struct TranscriberSinkPadSettings {
translation_languages: Option<gst::Structure>, translation_languages: Option<gst::Structure>,
language_code: String, language_code: String,
mode: Cea608Mode, mode: Cea608Mode,
passthrough: bool,
} }
impl Default for TranscriberSinkPadSettings { impl Default for TranscriberSinkPadSettings {
@ -1840,6 +1927,7 @@ impl Default for TranscriberSinkPadSettings {
translation_languages: None, translation_languages: None,
language_code: String::from(DEFAULT_INPUT_LANG_CODE), language_code: String::from(DEFAULT_INPUT_LANG_CODE),
mode: DEFAULT_MODE, mode: DEFAULT_MODE,
passthrough: DEFAULT_PASSTHROUGH,
} }
} }
} }
@ -1855,6 +1943,7 @@ struct TranscriberSinkPadState {
queue_passthrough: gst::Element, queue_passthrough: gst::Element,
transcription_channels: HashMap<String, TranscriptionChannel>, transcription_channels: HashMap<String, TranscriptionChannel>,
srcpad_name: Option<String>, srcpad_name: Option<String>,
target_passthrough_state: TargetPassthroughState,
} }
impl TranscriberSinkPadState { impl TranscriberSinkPadState {
@ -1882,6 +1971,7 @@ impl TranscriberSinkPadState {
queue_passthrough: gst::ElementFactory::make("queue").build()?, queue_passthrough: gst::ElementFactory::make("queue").build()?,
transcription_channels: HashMap::new(), transcription_channels: HashMap::new(),
srcpad_name: None, srcpad_name: None,
target_passthrough_state: TargetPassthroughState::None,
}) })
} }
} }
@ -1909,6 +1999,12 @@ impl ObjectImpl for TranscriberSinkPad {
fn properties() -> &'static [glib::ParamSpec] { fn properties() -> &'static [glib::ParamSpec] {
static PROPERTIES: LazyLock<Vec<glib::ParamSpec>> = LazyLock::new(|| { static PROPERTIES: LazyLock<Vec<glib::ParamSpec>> = LazyLock::new(|| {
vec![ vec![
glib::ParamSpecBoolean::builder("passthrough")
.nick("Passthrough")
.blurb("Whether transcription should occur")
.default_value(DEFAULT_PASSTHROUGH)
.mutable_playing()
.build(),
glib::ParamSpecBoxed::builder::<gst::Structure>("translation-languages") glib::ParamSpecBoxed::builder::<gst::Structure>("translation-languages")
.nick("Translation languages") .nick("Translation languages")
.blurb("A map of language codes to caption channels, e.g. translation-languages=\"languages, transcript={CC1, 708_1}, fr={708_2, CC3}\" will map the French translation to CC1/service 1 and the original transcript to CC3/service 2") .blurb("A map of language codes to caption channels, e.g. translation-languages=\"languages, transcript={CC1, 708_1}, fr={708_2, CC3}\" will map the French translation to CC1/service 1 and the original transcript to CC3/service 2")
@ -1938,6 +2034,29 @@ impl ObjectImpl for TranscriberSinkPad {
fn set_property(&self, _id: usize, value: &glib::Value, pspec: &glib::ParamSpec) { fn set_property(&self, _id: usize, value: &glib::Value, pspec: &glib::ParamSpec) {
match pspec.name() { match pspec.name() {
"passthrough" => {
let Some(parent) = self.obj().parent().and_downcast::<super::TranscriberBin>()
else {
return;
};
let s = parent.imp().state.lock().unwrap();
let mut ps = self.state.lock().unwrap();
let mut pad_settings = self.settings.lock().unwrap();
pad_settings.passthrough = value.get().expect("type checked upstream");
let Ok(ref mut pad_state) = ps.as_mut() else {
return;
};
if pad_settings.passthrough {
pad_state.target_passthrough_state = TargetPassthroughState::Enabled;
} else {
pad_state.target_passthrough_state = TargetPassthroughState::Disabled;
}
drop(pad_settings);
parent.imp().block_and_update(self, s, ps);
}
"translation-languages" => { "translation-languages" => {
let mut settings = self.settings.lock().unwrap(); let mut settings = self.settings.lock().unwrap();
settings.translation_languages = value settings.translation_languages = value
@ -2038,6 +2157,10 @@ impl ObjectImpl for TranscriberSinkPad {
fn property(&self, _id: usize, pspec: &glib::ParamSpec) -> glib::Value { fn property(&self, _id: usize, pspec: &glib::ParamSpec) -> glib::Value {
match pspec.name() { match pspec.name() {
"passthrough" => {
let settings = self.settings.lock().unwrap();
settings.passthrough.to_value()
}
"translation-languages" => { "translation-languages" => {
let settings = self.settings.lock().unwrap(); let settings = self.settings.lock().unwrap();
settings.translation_languages.to_value() settings.translation_languages.to_value()