gst-plugins-rs/video/closedcaption/tests/mcc_parse.rs

217 lines
6.2 KiB
Rust
Raw Normal View History

// Copyright (C) 2018 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
// 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.
#[macro_use]
extern crate pretty_assertions;
use gst::prelude::*;
2019-03-22 14:08:54 +00:00
use gst::EventView;
use rand::{Rng, SeedableRng};
2019-03-22 14:08:54 +00:00
use std::path::PathBuf;
fn init() {
use std::sync::Once;
static INIT: Once = Once::new();
INIT.call_once(|| {
gst::init().unwrap();
gstrsclosedcaption::plugin_register_static().expect("mccparse test");
});
}
/// Randomized test passing buffers of arbitrary sizes to the parser
#[test]
fn test_parse() {
init();
let mut data = include_bytes!("captions-test_708.mcc").as_ref();
let mut rnd = if let Ok(seed) = std::env::var("MCC_PARSE_TEST_SEED") {
rand::rngs::SmallRng::seed_from_u64(
seed.parse::<u64>()
.expect("MCC_PARSE_TEST_SEED has to contain a 64 bit integer seed"),
)
} else {
let seed = rand::random::<u64>();
println!("seed {}", seed);
rand::rngs::SmallRng::seed_from_u64(seed)
};
let mut h = gst_check::Harness::new("mccparse");
h.set_src_caps_str("application/x-mcc, version=(int) 1");
let mut input_len = 0;
let mut output_len = 0;
let mut checksum = 0u32;
let mut expected_timecode = None;
while !data.is_empty() {
let l = if data.len() == 1 {
1
} else {
rnd.gen_range(1, data.len())
};
let buf = gst::Buffer::from_mut_slice(Vec::from(&data[0..l]));
input_len += buf.get_size();
assert_eq!(h.push(buf), Ok(gst::FlowSuccess::Ok));
while let Some(buf) = h.try_pull() {
output_len += buf.get_size();
checksum = checksum.wrapping_add(
buf.map_readable()
.unwrap()
.iter()
.fold(0u32, |s, v| s.wrapping_add(*v as u32)),
);
let tc_meta = buf
.get_meta::<gst_video::VideoTimeCodeMeta>()
.expect("No timecode meta");
if let Some(ref timecode) = expected_timecode {
assert_eq!(&tc_meta.get_tc(), timecode);
} else {
expected_timecode = Some(tc_meta.get_tc());
}
if let Some(ref mut tc) = expected_timecode {
tc.increment_frame();
}
}
data = &data[l..];
}
h.push_event(gst::Event::new_eos().build());
while let Some(buf) = h.try_pull() {
output_len += buf.get_size();
checksum = checksum.wrapping_add(
buf.map_readable()
.unwrap()
.iter()
.fold(0u32, |s, v| s.wrapping_add(*v as u32)),
);
let tc_meta = buf
.get_meta::<gst_video::VideoTimeCodeMeta>()
.expect("No timecode meta");
if let Some(ref timecode) = expected_timecode {
assert_eq!(&tc_meta.get_tc(), timecode);
} else {
expected_timecode = Some(tc_meta.get_tc());
}
if let Some(ref mut tc) = expected_timecode {
tc.increment_frame();
}
}
assert!(expected_timecode.is_some());
assert_eq!(input_len, 28_818);
assert_eq!(output_len, 42_383);
assert_eq!(checksum, 3_988_480);
let caps = h
.get_sinkpad()
.expect("harness has no sinkpad")
.get_current_caps()
.expect("pad has no caps");
assert_eq!(
caps,
gst::Caps::builder("closedcaption/x-cea-708")
.field("format", &"cdp")
.field("framerate", &gst::Fraction::new(30000, 1001))
.build()
);
}
2019-03-22 14:08:54 +00:00
#[test]
fn test_pull() {
init();
let mut path = PathBuf::from(env!("CARGO_MANIFEST_DIR"));
path.push("tests/captions-test_708.mcc");
let mut h = gst_check::Harness::new_parse(&format!("filesrc location={:?} ! mccparse", path));
h.play();
/* Let's first pull until EOS */
loop {
let mut done = false;
while h.events_in_queue() != 0 {
let event = h.pull_event();
2019-12-18 05:50:10 +00:00
if let Ok(event) = event {
2019-03-22 14:08:54 +00:00
match event.view() {
EventView::Eos(_) => {
done = true;
break;
}
_ => (),
}
}
}
while h.buffers_in_queue() != 0 {
let _ = h.pull();
}
if done {
break;
}
}
/* Now seek and check that we receive buffers with appropriate PTS */
h.push_upstream_event(
gst::Event::new_seek(
1.0,
gst::SeekFlags::FLUSH,
gst::SeekType::Set,
gst::GenericFormattedValue::Time(gst::SECOND),
2019-03-22 14:08:54 +00:00
gst::SeekType::Set,
gst::GenericFormattedValue::Time(2 * gst::SECOND),
2019-03-22 14:08:54 +00:00
)
.build(),
);
loop {
let mut done = false;
while h.buffers_in_queue() != 0 {
2019-12-18 05:50:10 +00:00
if let Ok(buffer) = h.pull() {
2019-03-22 14:08:54 +00:00
let pts = buffer.get_pts();
assert!(pts > gst::SECOND && pts < 2 * gst::SECOND);
}
}
while h.events_in_queue() != 0 {
let event = h.pull_event();
2019-12-18 05:50:10 +00:00
if let Ok(event) = event {
2019-03-22 14:08:54 +00:00
match event.view() {
EventView::Eos(_) => {
done = true;
break;
}
_ => (),
}
}
}
if done {
break;
}
}
}