mirror of
https://gitlab.freedesktop.org/gstreamer/gst-plugins-rs.git
synced 2024-12-25 11:30:29 +00:00
107 lines
3.1 KiB
Rust
107 lines
3.1 KiB
Rust
// Copyright (C) 2019 Sebastian Dröge <sebastian@centricular.com>
|
|
//
|
|
// Licensed under the Apache License, Version 2.0 <LICENSE-APACHE or
|
|
// http://www.apache.org/licenses/LICENSE-2.0> or the MIT license
|
|
// <LICENSE-MIT or http://opensource.org/licenses/MIT>, at your
|
|
// option. This file may not be copied, modified, or distributed
|
|
// except according to those terms.
|
|
|
|
extern crate glib;
|
|
extern crate gstreamer as gst;
|
|
use gst::prelude::*;
|
|
extern crate gstreamer_audio as gst_audio;
|
|
extern crate gstreamer_check as gst_check;
|
|
|
|
extern crate gstlewton;
|
|
|
|
fn init() {
|
|
use std::sync::Once;
|
|
static INIT: Once = Once::new();
|
|
|
|
INIT.call_once(|| {
|
|
gst::init().unwrap();
|
|
gstlewton::plugin_register_static().expect("lewton test");
|
|
});
|
|
}
|
|
|
|
#[test]
|
|
fn test_with_streamheader() {
|
|
run_test(false);
|
|
}
|
|
|
|
#[test]
|
|
fn test_with_inline_headers() {
|
|
run_test(true);
|
|
}
|
|
|
|
fn run_test(inline_headers: bool) {
|
|
let data = include_bytes!("test.vorbis");
|
|
let packet_sizes = [30, 99, 3189, 43, 20, 56, 56, 21, 20, 22, 21, 22, 22, 43];
|
|
let packet_offsets = packet_sizes
|
|
.iter()
|
|
.scan(0, |state, &size| {
|
|
*state += size;
|
|
Some(*state)
|
|
})
|
|
.collect::<Vec<usize>>();
|
|
let decoded_samples = [0usize, 128, 576, 1472, 128, 128, 128, 128, 128, 128, 128];
|
|
|
|
init();
|
|
|
|
let mut h = gst_check::Harness::new("lewtondec");
|
|
h.play();
|
|
|
|
if inline_headers {
|
|
let caps = gst::Caps::builder("audio/x-vorbis").build();
|
|
h.set_src_caps(caps);
|
|
} else {
|
|
let caps = gst::Caps::builder("audio/x-vorbis")
|
|
.field(
|
|
"streamheader",
|
|
&gst::Array::new(&[
|
|
&gst::Buffer::from_slice(&data[0..packet_offsets[0]]),
|
|
&gst::Buffer::from_slice(&data[packet_offsets[0]..packet_offsets[1]]),
|
|
&gst::Buffer::from_slice(&data[packet_offsets[1]..packet_offsets[2]]),
|
|
]),
|
|
)
|
|
.build();
|
|
h.set_src_caps(caps);
|
|
}
|
|
|
|
let packet_offsets_iter = std::iter::once(&0).chain(packet_offsets.iter());
|
|
let skip = if inline_headers { 0 } else { 3 };
|
|
|
|
for (offset_start, offset_end) in packet_offsets_iter
|
|
.clone()
|
|
.skip(skip)
|
|
.zip(packet_offsets_iter.clone().skip(skip + 1))
|
|
{
|
|
let buffer = gst::Buffer::from_slice(&data[*offset_start..*offset_end]);
|
|
h.push(buffer).unwrap();
|
|
}
|
|
|
|
h.push_event(gst::Event::new_eos().build());
|
|
|
|
for samples in &decoded_samples {
|
|
if *samples == 0 {
|
|
continue;
|
|
}
|
|
let buffer = h.pull().unwrap();
|
|
assert_eq!(buffer.get_size(), 4 * samples);
|
|
}
|
|
|
|
let caps = h
|
|
.get_sinkpad()
|
|
.expect("harness has no sinkpad")
|
|
.get_current_caps()
|
|
.expect("pad has no caps");
|
|
assert_eq!(
|
|
caps,
|
|
gst::Caps::builder("audio/x-raw")
|
|
.field("format", &gst_audio::AUDIO_FORMAT_F32.to_str())
|
|
.field("rate", &44_100i32)
|
|
.field("channels", &1i32)
|
|
.field("layout", &"interleaved")
|
|
.build()
|
|
);
|
|
}
|