Playlist of subtitles enabled
This commit is contained in:
parent
2340e85b36
commit
44493c73d7
1 changed files with 54 additions and 26 deletions
68
src/main.rs
68
src/main.rs
|
@ -1,5 +1,7 @@
|
||||||
use gst::prelude::*;
|
|
||||||
use anyhow::Error;
|
use anyhow::Error;
|
||||||
|
use gst::prelude::*;
|
||||||
|
use std::sync::atomic::{AtomicBool, Ordering};
|
||||||
|
use std::time::Duration;
|
||||||
|
|
||||||
fn main() -> Result<(), Error> {
|
fn main() -> Result<(), Error> {
|
||||||
gst::init().unwrap();
|
gst::init().unwrap();
|
||||||
|
@ -8,13 +10,16 @@ fn main() -> Result<(), Error> {
|
||||||
// Create a video source with X buffers and burn the captions in the video feed using the captions tooling
|
// Create a video source with X buffers and burn the captions in the video feed using the captions tooling
|
||||||
let pipeline = gst::parse_launch(
|
let pipeline = gst::parse_launch(
|
||||||
"cccombiner name=ccc ! cea608overlay ! autovideosink \
|
"cccombiner name=ccc ! cea608overlay ! autovideosink \
|
||||||
videotestsrc num-buffers=1800 ! video/x-raw,width=1280,height=720,framerate=30/1 ! queue ! ccc.sink \
|
videotestsrc num-buffers=3600 ! video/x-raw,width=1280,height=720,framerate=30/1 ! queue ! ccc.sink \
|
||||||
tttocea608 name=converter mode=1 ! queue ! ccc.caption",
|
tttocea608 name=converter mode=0 ! queue ! ccc.caption",
|
||||||
)?
|
)?
|
||||||
.downcast::<gst::Pipeline>()
|
.downcast::<gst::Pipeline>()
|
||||||
.expect("Expected a gst::Pipeline");
|
.expect("Expected a gst::Pipeline");
|
||||||
|
|
||||||
let text_src = gst::ElementFactory::make("appsrc", Some("text-src")).unwrap().downcast::<gst_app::AppSrc>().unwrap();
|
let text_src = gst::ElementFactory::make("appsrc", Some("text-src"))
|
||||||
|
.unwrap()
|
||||||
|
.downcast::<gst_app::AppSrc>()
|
||||||
|
.unwrap();
|
||||||
text_src.set_is_live(true);
|
text_src.set_is_live(true);
|
||||||
text_src.set_format(gst::Format::Time);
|
text_src.set_format(gst::Format::Time);
|
||||||
text_src.set_caps(Some(
|
text_src.set_caps(Some(
|
||||||
|
@ -22,10 +27,14 @@ fn main() -> Result<(), Error> {
|
||||||
.field("format", "utf8")
|
.field("format", "utf8")
|
||||||
.build(),
|
.build(),
|
||||||
));
|
));
|
||||||
pipeline.add(&text_src).expect("Failed to add elements to pipeline");
|
pipeline
|
||||||
|
.add(&text_src)
|
||||||
|
.expect("Failed to add elements to pipeline");
|
||||||
|
|
||||||
let converter = pipeline.by_name("converter").unwrap();
|
let converter = pipeline.by_name("converter").unwrap();
|
||||||
text_src.link_pads(Some("src"), &converter, Some("sink")).expect("Failed to link elements");
|
text_src
|
||||||
|
.link_pads(Some("src"), &converter, Some("sink"))
|
||||||
|
.expect("Failed to link elements");
|
||||||
|
|
||||||
println!("Starting pipeline..");
|
println!("Starting pipeline..");
|
||||||
|
|
||||||
|
@ -64,42 +73,61 @@ fn main_loop(pipeline: gst::Pipeline) -> Result<(), Error> {
|
||||||
})
|
})
|
||||||
.expect("Failed to add bus watch");
|
.expect("Failed to add bus watch");
|
||||||
|
|
||||||
|
static RUNNING: AtomicBool = AtomicBool::new(true);
|
||||||
|
let subtitles_src_thread = std::thread::spawn({
|
||||||
let pipeline_weak = pipeline.downgrade();
|
let pipeline_weak = pipeline.downgrade();
|
||||||
let timeout_id = glib::timeout_add_local(std::time::Duration::from_secs(20), move || {
|
move || {
|
||||||
let pipeline = match pipeline_weak.upgrade() {
|
std::thread::sleep(Duration::from_secs(2));
|
||||||
Some(pipeline) => pipeline,
|
|
||||||
None => return glib::Continue(true),
|
let subtitles = [
|
||||||
};
|
"Hello people!",
|
||||||
push_buffer(&pipeline);
|
"This is just an example of \nsubtitle",
|
||||||
glib::Continue(true)
|
"We will learn more about \nGStreamer and buffers today",
|
||||||
|
];
|
||||||
|
let mut sub_idx = (0..subtitles.len()).cycle().into_iter();
|
||||||
|
|
||||||
|
while RUNNING.load(Ordering::Relaxed) {
|
||||||
|
if let Some(pipeline) = pipeline_weak.upgrade() {
|
||||||
|
push_buffer(&pipeline, subtitles[sub_idx.next().unwrap_or(0)]);
|
||||||
|
}
|
||||||
|
println!("pushed the buffer.. now waiting.");
|
||||||
|
std::thread::sleep(Duration::from_secs(5));
|
||||||
|
}
|
||||||
|
println!("Stopping subtitles source thread!");
|
||||||
|
}
|
||||||
});
|
});
|
||||||
|
|
||||||
main_loop.run();
|
main_loop.run();
|
||||||
|
|
||||||
pipeline.set_state(gst::State::Null)?;
|
pipeline.set_state(gst::State::Null)?;
|
||||||
|
RUNNING.store(false, Ordering::Relaxed);
|
||||||
|
subtitles_src_thread.join().unwrap();
|
||||||
|
|
||||||
timeout_id.remove();
|
|
||||||
bus.remove_watch().unwrap();
|
bus.remove_watch().unwrap();
|
||||||
|
|
||||||
Ok(())
|
Ok(())
|
||||||
}
|
}
|
||||||
|
|
||||||
fn push_buffer(pipeline: &gst::Pipeline) {
|
fn push_buffer(pipeline: &gst::Pipeline, text: impl AsRef<str>) {
|
||||||
let src = pipeline
|
let src = pipeline
|
||||||
.by_name("text-src")
|
.by_name("text-src")
|
||||||
.unwrap()
|
.unwrap()
|
||||||
.downcast::<gst_app::AppSrc>()
|
.downcast::<gst_app::AppSrc>()
|
||||||
.unwrap();
|
.unwrap();
|
||||||
|
|
||||||
let timestamp = pipeline.query_position::<gst::ClockTime>().unwrap() + gst::ClockTime::from_seconds(1);
|
let present_caption_at =
|
||||||
|
pipeline.query_position::<gst::ClockTime>().unwrap() + gst::ClockTime::from_seconds(1);
|
||||||
|
|
||||||
println!("Trying to publish text buffer NOW! - {}", timestamp.display());
|
println!(
|
||||||
|
"Trying to publish text buffer NOW! - {} > {}",
|
||||||
|
present_caption_at.display(),
|
||||||
|
text.as_ref()
|
||||||
|
);
|
||||||
|
|
||||||
let text = format!("Such nice: {}!", timestamp.display());
|
let mut buffer = gst::Buffer::from_mut_slice(text.as_ref().to_string().into_bytes());
|
||||||
let mut buffer = gst::Buffer::from_mut_slice(text.into_bytes());
|
|
||||||
{
|
{
|
||||||
let buffer = buffer.get_mut().unwrap();
|
let buffer = buffer.get_mut().unwrap();
|
||||||
buffer.set_pts(timestamp);
|
buffer.set_pts(present_caption_at);
|
||||||
buffer.set_duration(gst::ClockTime::from_seconds(1));
|
buffer.set_duration(gst::ClockTime::from_seconds(1));
|
||||||
}
|
}
|
||||||
src.push_buffer(buffer).unwrap();
|
src.push_buffer(buffer).unwrap();
|
||||||
|
|
Loading…
Reference in a new issue