mirror of
https://gitlab.freedesktop.org/gstreamer/gst-plugins-rs.git
synced 2024-06-08 17:29:22 +00:00
Add initial version of lewton based Vorbis decoder
This commit is contained in:
Sebastian Dröge
parent
63eab12132
commit
d0978473d3
|
|
@ -15,6 +15,7 @@ members = [
|
|||
"gst-plugin-rav1e",
|
||||
"gst-plugin-s3",
|
||||
"gst-plugin-fallbackswitch",
|
||||
"gst-plugin-lewton",
|
||||
]
|
||||
|
||||
[profile.release]
|
||||
|
|
|
|||
25
gst-plugin-lewton/Cargo.toml
Normal file
25
gst-plugin-lewton/Cargo.toml
Normal file
|
|
@ -0,0 +1,25 @@
|
|||
[package]
|
||||
name = "gst-plugin-lewton"
|
||||
version = "0.6.0"
|
||||
authors = ["Sebastian Dröge <sebastian@centricular.com>"]
|
||||
repository = "https://gitlab.freedesktop.org/gstreamer/gst-plugin-rs"
|
||||
license = "MIT/Apache-2.0"
|
||||
description = "lewton Vorbis Decoder Plugin"
|
||||
edition = "2018"
|
||||
|
||||
[dependencies]
|
||||
glib = { git = "https://github.com/gtk-rs/glib" }
|
||||
gstreamer = { git = "https://gitlab.freedesktop.org/gstreamer/gstreamer-rs" }
|
||||
gstreamer-audio = { git = "https://gitlab.freedesktop.org/gstreamer/gstreamer-rs" }
|
||||
gstreamer-check = { git = "https://gitlab.freedesktop.org/gstreamer/gstreamer-rs" }
|
||||
lewton = { version = "0.9", default-features = false }
|
||||
byte-slice-cast = "0.3"
|
||||
atomic_refcell = "0.1"
|
||||
|
||||
[lib]
|
||||
name = "gstlewton"
|
||||
crate-type = ["cdylib", "rlib"]
|
||||
path = "src/lib.rs"
|
||||
|
||||
[build-dependencies]
|
||||
gst-plugin-version-helper = { path="../gst-plugin-version-helper" }
|
||||
5
gst-plugin-lewton/build.rs
Normal file
5
gst-plugin-lewton/build.rs
Normal file
|
|
@ -0,0 +1,5 @@
|
|||
extern crate gst_plugin_version_helper;
|
||||
|
||||
fn main() {
|
||||
gst_plugin_version_helper::get_info()
|
||||
}
|
||||
662
gst-plugin-lewton/src/lewtondec.rs
Normal file
662
gst-plugin-lewton/src/lewtondec.rs
Normal file
|
|
@ -0,0 +1,662 @@
|
|||
// 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.
|
||||
|
||||
use glib;
|
||||
use glib::subclass;
|
||||
use glib::subclass::prelude::*;
|
||||
use gst;
|
||||
use gst::subclass::prelude::*;
|
||||
use gst_audio::prelude::*;
|
||||
use gst_audio::subclass::prelude::*;
|
||||
use gstreamer_audio as gst_audio;
|
||||
|
||||
use atomic_refcell::AtomicRefCell;
|
||||
|
||||
use byte_slice_cast::*;
|
||||
|
||||
struct State {
|
||||
header_bufs: (
|
||||
Option<gst::Buffer>,
|
||||
Option<gst::Buffer>,
|
||||
Option<gst::Buffer>,
|
||||
),
|
||||
headerset: Option<lewton::header::HeaderSet>,
|
||||
pwr: lewton::audio::PreviousWindowRight,
|
||||
audio_info: Option<gst_audio::AudioInfo>,
|
||||
reorder_map: Option<[usize; 8]>,
|
||||
}
|
||||
|
||||
struct LewtonDec {
|
||||
cat: gst::DebugCategory,
|
||||
state: AtomicRefCell<Option<State>>,
|
||||
}
|
||||
|
||||
impl ObjectSubclass for LewtonDec {
|
||||
const NAME: &'static str = "LewtonDec";
|
||||
type ParentType = gst_audio::AudioDecoder;
|
||||
type Instance = gst::subclass::ElementInstanceStruct<Self>;
|
||||
type Class = subclass::simple::ClassStruct<Self>;
|
||||
|
||||
glib_object_subclass!();
|
||||
|
||||
fn new() -> Self {
|
||||
Self {
|
||||
cat: gst::DebugCategory::new(
|
||||
"lewtondec",
|
||||
gst::DebugColorFlags::empty(),
|
||||
Some("lewton Vorbis decoder"),
|
||||
),
|
||||
state: AtomicRefCell::new(None),
|
||||
}
|
||||
}
|
||||
|
||||
fn class_init(klass: &mut subclass::simple::ClassStruct<Self>) {
|
||||
klass.set_metadata(
|
||||
"lewton Vorbis decoder",
|
||||
"Decoder/Audio",
|
||||
"lewton Vorbis decoder",
|
||||
"Sebastian Dröge <sebastian@centricular.com>",
|
||||
);
|
||||
|
||||
let sink_caps = gst::Caps::new_simple("audio/x-vorbis", &[]);
|
||||
let sink_pad_template = gst::PadTemplate::new(
|
||||
"sink",
|
||||
gst::PadDirection::Sink,
|
||||
gst::PadPresence::Always,
|
||||
&sink_caps,
|
||||
)
|
||||
.unwrap();
|
||||
klass.add_pad_template(sink_pad_template);
|
||||
|
||||
let src_caps = gst::Caps::new_simple(
|
||||
"audio/x-raw",
|
||||
&[
|
||||
("format", &gst_audio::AUDIO_FORMAT_F32.to_string()),
|
||||
("rate", &gst::IntRange::<i32>::new(1, std::i32::MAX)),
|
||||
("channels", &gst::IntRange::<i32>::new(1, 255)),
|
||||
("layout", &"interleaved"),
|
||||
],
|
||||
);
|
||||
let src_pad_template = gst::PadTemplate::new(
|
||||
"src",
|
||||
gst::PadDirection::Src,
|
||||
gst::PadPresence::Always,
|
||||
&src_caps,
|
||||
)
|
||||
.unwrap();
|
||||
klass.add_pad_template(src_pad_template);
|
||||
}
|
||||
}
|
||||
|
||||
impl ObjectImpl for LewtonDec {
|
||||
glib_object_impl!();
|
||||
}
|
||||
|
||||
impl ElementImpl for LewtonDec {}
|
||||
|
||||
impl AudioDecoderImpl for LewtonDec {
|
||||
fn stop(&self, _element: &gst_audio::AudioDecoder) -> Result<(), gst::ErrorMessage> {
|
||||
*self.state.borrow_mut() = None;
|
||||
|
||||
Ok(())
|
||||
}
|
||||
|
||||
fn start(&self, _element: &gst_audio::AudioDecoder) -> Result<(), gst::ErrorMessage> {
|
||||
*self.state.borrow_mut() = Some(State {
|
||||
header_bufs: (None, None, None),
|
||||
headerset: None,
|
||||
pwr: lewton::audio::PreviousWindowRight::new(),
|
||||
audio_info: None,
|
||||
reorder_map: None,
|
||||
});
|
||||
|
||||
Ok(())
|
||||
}
|
||||
|
||||
fn set_format(
|
||||
&self,
|
||||
element: &gst_audio::AudioDecoder,
|
||||
caps: &gst::Caps,
|
||||
) -> Result<(), gst::LoggableError> {
|
||||
gst_debug!(self.cat, obj: element, "Setting format {:?}", caps);
|
||||
|
||||
// When the caps are changing we require new headers
|
||||
let mut state_guard = self.state.borrow_mut();
|
||||
*state_guard = Some(State {
|
||||
header_bufs: (None, None, None),
|
||||
headerset: None,
|
||||
pwr: lewton::audio::PreviousWindowRight::new(),
|
||||
audio_info: None,
|
||||
reorder_map: None,
|
||||
});
|
||||
|
||||
let mut state = state_guard.as_mut().unwrap();
|
||||
|
||||
let s = caps.get_structure(0).unwrap();
|
||||
if let Ok(Some(streamheaders)) = s.get_optional::<gst::Array>("streamheader") {
|
||||
let streamheaders = streamheaders.as_slice();
|
||||
if streamheaders.len() < 3 {
|
||||
gst_debug!(
|
||||
self.cat,
|
||||
obj: element,
|
||||
"Not enough streamheaders, trying in-band"
|
||||
);
|
||||
return Ok(());
|
||||
}
|
||||
|
||||
let ident_buf = streamheaders[0].get::<gst::Buffer>();
|
||||
let comment_buf = streamheaders[1].get::<gst::Buffer>();
|
||||
let setup_buf = streamheaders[2].get::<gst::Buffer>();
|
||||
if let (Ok(Some(ident_buf)), Ok(Some(comment_buf)), Ok(Some(setup_buf))) =
|
||||
(ident_buf, comment_buf, setup_buf)
|
||||
{
|
||||
gst_debug!(self.cat, obj: element, "Got streamheader buffers");
|
||||
state.header_bufs = (Some(ident_buf), Some(comment_buf), Some(setup_buf));
|
||||
}
|
||||
}
|
||||
|
||||
Ok(())
|
||||
}
|
||||
|
||||
fn flush(&self, element: &gst_audio::AudioDecoder, _hard: bool) {
|
||||
gst_debug!(self.cat, obj: element, "Flushing");
|
||||
|
||||
let mut state_guard = self.state.borrow_mut();
|
||||
if let Some(ref mut state) = *state_guard {
|
||||
state.pwr = lewton::audio::PreviousWindowRight::new();
|
||||
}
|
||||
}
|
||||
|
||||
fn handle_frame(
|
||||
&self,
|
||||
element: &gst_audio::AudioDecoder,
|
||||
inbuf: Option<&gst::Buffer>,
|
||||
) -> Result<gst::FlowSuccess, gst::FlowError> {
|
||||
gst_debug!(self.cat, obj: element, "Handling buffer {:?}", inbuf);
|
||||
|
||||
let inbuf = match inbuf {
|
||||
None => return Ok(gst::FlowSuccess::Ok),
|
||||
Some(inbuf) => inbuf,
|
||||
};
|
||||
|
||||
let inmap = inbuf.map_readable().ok_or_else(|| {
|
||||
gst_error!(self.cat, obj: element, "Failed to buffer readable");
|
||||
gst::FlowError::Error
|
||||
})?;
|
||||
|
||||
let mut state_guard = self.state.borrow_mut();
|
||||
let state = state_guard.as_mut().ok_or(gst::FlowError::NotNegotiated)?;
|
||||
|
||||
// Ignore empty packets unless we have no headers yet
|
||||
if inmap.len() == 0 {
|
||||
element.finish_frame(None, 1)?;
|
||||
|
||||
if state.headerset.is_some() {
|
||||
return Ok(gst::FlowSuccess::Ok);
|
||||
} else {
|
||||
gst_error!(
|
||||
self.cat,
|
||||
obj: element,
|
||||
"Got empty packet before all headers"
|
||||
);
|
||||
return Err(gst::FlowError::Error);
|
||||
}
|
||||
}
|
||||
|
||||
// If this is a header packet then handle it
|
||||
if inmap[0] & 0x01 == 0x01 {
|
||||
return self.handle_header(element, state, inbuf, inmap.as_ref());
|
||||
}
|
||||
|
||||
// If it's a data packet then try to initialize the headerset now if we didn't yet
|
||||
if state.headerset.is_none() {
|
||||
self.initialize(element, state)?;
|
||||
}
|
||||
|
||||
self.handle_data(element, state, inmap.as_ref())
|
||||
}
|
||||
}
|
||||
|
||||
impl LewtonDec {
|
||||
fn handle_header(
|
||||
&self,
|
||||
element: &gst_audio::AudioDecoder,
|
||||
state: &mut State,
|
||||
inbuf: &gst::Buffer,
|
||||
indata: &[u8],
|
||||
) -> Result<gst::FlowSuccess, gst::FlowError> {
|
||||
// ident header
|
||||
if indata[0] == 0x01 {
|
||||
gst_debug!(self.cat, obj: element, "Got ident header buffer");
|
||||
state.header_bufs = (Some(inbuf.clone()), None, None);
|
||||
} else if indata[0] == 0x03 {
|
||||
// comment header
|
||||
if state.header_bufs.0.is_none() {
|
||||
gst_warning!(
|
||||
self.cat,
|
||||
obj: element,
|
||||
"Got comment header before ident header"
|
||||
);
|
||||
} else {
|
||||
gst_debug!(self.cat, obj: element, "Got comment header buffer");
|
||||
state.header_bufs.1 = Some(inbuf.clone());
|
||||
}
|
||||
} else if indata[0] == 0x05 {
|
||||
// setup header
|
||||
if state.header_bufs.0.is_none() || state.header_bufs.1.is_none() {
|
||||
gst_warning!(
|
||||
self.cat,
|
||||
obj: element,
|
||||
"Got setup header before ident/comment header"
|
||||
);
|
||||
} else {
|
||||
gst_debug!(self.cat, obj: element, "Got setup header buffer");
|
||||
state.header_bufs.2 = Some(inbuf.clone());
|
||||
}
|
||||
}
|
||||
|
||||
element.finish_frame(None, 1)
|
||||
}
|
||||
|
||||
fn initialize(
|
||||
&self,
|
||||
element: &gst_audio::AudioDecoder,
|
||||
state: &mut State,
|
||||
) -> Result<(), gst::FlowError> {
|
||||
let (ident_buf, comment_buf, setup_buf) = match state.header_bufs {
|
||||
(Some(ref ident_buf), Some(ref comment_buf), Some(ref setup_buf)) => {
|
||||
(ident_buf, comment_buf, setup_buf)
|
||||
}
|
||||
_ => {
|
||||
gst_element_error!(
|
||||
element,
|
||||
gst::StreamError::Decode,
|
||||
["Got no headers before data packets"]
|
||||
);
|
||||
return Err(gst::FlowError::NotNegotiated);
|
||||
}
|
||||
};
|
||||
|
||||
// First try to parse the headers
|
||||
let ident_map = ident_buf.map_readable().ok_or_else(|| {
|
||||
gst_error!(
|
||||
self.cat,
|
||||
obj: element,
|
||||
"Failed to map ident buffer readable"
|
||||
);
|
||||
gst::FlowError::Error
|
||||
})?;
|
||||
let ident = lewton::header::read_header_ident(ident_map.as_ref()).map_err(|err| {
|
||||
gst_element_error!(
|
||||
element,
|
||||
gst::StreamError::Decode,
|
||||
["Failed to parse ident header: {:?}", err]
|
||||
);
|
||||
gst::FlowError::Error
|
||||
})?;
|
||||
|
||||
let comment_map = comment_buf.map_readable().ok_or_else(|| {
|
||||
gst_error!(
|
||||
self.cat,
|
||||
obj: element,
|
||||
"Failed to map comment buffer readable"
|
||||
);
|
||||
gst::FlowError::Error
|
||||
})?;
|
||||
let comment = lewton::header::read_header_comment(comment_map.as_ref()).map_err(|err| {
|
||||
gst_element_error!(
|
||||
element,
|
||||
gst::StreamError::Decode,
|
||||
["Failed to parse comment header: {:?}", err]
|
||||
);
|
||||
gst::FlowError::Error
|
||||
})?;
|
||||
|
||||
let setup_map = setup_buf.map_readable().ok_or_else(|| {
|
||||
gst_error!(
|
||||
self.cat,
|
||||
obj: element,
|
||||
"Failed to map setup buffer readable"
|
||||
);
|
||||
gst::FlowError::Error
|
||||
})?;
|
||||
let setup = lewton::header::read_header_setup(
|
||||
setup_map.as_ref(),
|
||||
ident.audio_channels,
|
||||
(ident.blocksize_0, ident.blocksize_1),
|
||||
)
|
||||
.map_err(|err| {
|
||||
gst_element_error!(
|
||||
element,
|
||||
gst::StreamError::Decode,
|
||||
["Failed to parse setup header: {:?}", err]
|
||||
);
|
||||
gst::FlowError::Error
|
||||
})?;
|
||||
|
||||
let mut audio_info = gst_audio::AudioInfo::new(
|
||||
gst_audio::AUDIO_FORMAT_F32,
|
||||
ident.audio_sample_rate,
|
||||
ident.audio_channels as u32,
|
||||
);
|
||||
|
||||
// For 1-8 channels there are defined channel positions, so initialize
|
||||
// everything accordingly and set the channel positions in the output
|
||||
// format
|
||||
let mut reorder_map = None;
|
||||
if ident.audio_channels > 1 && ident.audio_channels < 9 {
|
||||
let channels = ident.audio_channels as usize;
|
||||
let from = &VORBIS_CHANNEL_POSITIONS[channels - 1][..channels];
|
||||
let to = &GST_VORBIS_CHANNEL_POSITIONS[channels - 1][..channels];
|
||||
|
||||
audio_info = audio_info.positions(to);
|
||||
|
||||
let mut map = [0; 8];
|
||||
if let Err(_) = gst_audio::get_channel_reorder_map(from, to, &mut map[..channels]) {
|
||||
gst_error!(
|
||||
self.cat,
|
||||
obj: element,
|
||||
"Failed to generate channel reorder map from {:?} to {:?}",
|
||||
from,
|
||||
to,
|
||||
);
|
||||
} else {
|
||||
// If the reorder map is not the identity matrix
|
||||
if !map[..channels].iter().enumerate().all(|(c1, c2)| c1 == *c2) {
|
||||
reorder_map = Some(map);
|
||||
}
|
||||
}
|
||||
}
|
||||
let audio_info = audio_info.build().unwrap();
|
||||
|
||||
gst_debug!(
|
||||
self.cat,
|
||||
obj: element,
|
||||
"Successfully parsed headers: {:?}",
|
||||
audio_info
|
||||
);
|
||||
state.headerset = Some((ident, comment, setup));
|
||||
state.audio_info = Some(audio_info.clone());
|
||||
state.reorder_map = reorder_map;
|
||||
|
||||
element.set_output_format(&audio_info)?;
|
||||
element.negotiate()?;
|
||||
|
||||
Ok(())
|
||||
}
|
||||
|
||||
fn handle_data(
|
||||
&self,
|
||||
element: &gst_audio::AudioDecoder,
|
||||
state: &mut State,
|
||||
indata: &[u8],
|
||||
) -> Result<gst::FlowSuccess, gst::FlowError> {
|
||||
// We ensured above that we have headers here now
|
||||
let headerset = state.headerset.as_ref().unwrap();
|
||||
let audio_info = state.audio_info.as_ref().unwrap();
|
||||
|
||||
// Decode the input packet
|
||||
let decoded = lewton::audio::read_audio_packet_generic::<
|
||||
lewton::samples::InterleavedSamples<f32>,
|
||||
>(&headerset.0, &headerset.2, indata, &mut state.pwr)
|
||||
.map_err(|err| {
|
||||
gst_element_error!(
|
||||
element,
|
||||
gst::StreamError::Decode,
|
||||
["Failed to decode packet: {:?}", err]
|
||||
);
|
||||
gst::FlowError::Error
|
||||
})?;
|
||||
|
||||
if decoded.channel_count != audio_info.channels() as usize {
|
||||
gst_element_error!(
|
||||
element,
|
||||
gst::StreamError::Decode,
|
||||
[
|
||||
"Channel count mismatch (got {}, expected {})",
|
||||
decoded.channel_count,
|
||||
audio_info.channels()
|
||||
]
|
||||
);
|
||||
return Err(gst::FlowError::Error);
|
||||
}
|
||||
|
||||
let sample_count = decoded.samples.len() / audio_info.channels() as usize;
|
||||
gst_debug!(
|
||||
self.cat,
|
||||
obj: element,
|
||||
"Got {} decoded samples",
|
||||
sample_count
|
||||
);
|
||||
|
||||
if sample_count == 0 {
|
||||
return element.finish_frame(None, 1);
|
||||
}
|
||||
|
||||
let mut outbuf = element
|
||||
.allocate_output_buffer(sample_count as usize * audio_info.bpf() as usize)
|
||||
.ok_or_else(|| {
|
||||
gst_element_error!(
|
||||
element,
|
||||
gst::StreamError::Decode,
|
||||
["Failed to allocate output buffer"]
|
||||
);
|
||||
gst::FlowError::Error
|
||||
})?;
|
||||
|
||||
// And copy the decoded data into our output buffer, if needed
|
||||
// while reordering the channels to the GStreamer channel order
|
||||
{
|
||||
let outbuf = outbuf.get_mut().unwrap();
|
||||
let mut outmap = outbuf.map_writable().ok_or_else(|| {
|
||||
gst_element_error!(
|
||||
element,
|
||||
gst::StreamError::Decode,
|
||||
["Failed to map output buffer writable"]
|
||||
);
|
||||
gst::FlowError::Error
|
||||
})?;
|
||||
|
||||
let outdata = outmap.as_mut_slice_of::<f32>().unwrap();
|
||||
let channels = audio_info.channels() as usize;
|
||||
if let Some(ref reorder_map) = state.reorder_map {
|
||||
assert!(reorder_map.len() >= channels);
|
||||
assert!(reorder_map[..channels].iter().all(|c| *c < channels));
|
||||
|
||||
for (output, input) in outdata
|
||||
.chunks_exact_mut(channels)
|
||||
.zip(decoded.samples.chunks_exact(channels))
|
||||
{
|
||||
for (c, s) in input.iter().enumerate() {
|
||||
output[reorder_map[c]] = *s;
|
||||
}
|
||||
}
|
||||
} else {
|
||||
outdata.copy_from_slice(&decoded.samples);
|
||||
}
|
||||
}
|
||||
|
||||
element.finish_frame(Some(outbuf), 1)
|
||||
}
|
||||
}
|
||||
|
||||
pub fn register(plugin: &gst::Plugin) -> Result<(), glib::BoolError> {
|
||||
gst::Element::register(
|
||||
Some(plugin),
|
||||
"lewtondec",
|
||||
gst::Rank::Marginal,
|
||||
LewtonDec::get_type(),
|
||||
)
|
||||
}
|
||||
|
||||
// http://www.xiph.org/vorbis/doc/Vorbis_I_spec.html#x1-800004.3.9
|
||||
const VORBIS_CHANNEL_POSITIONS: [[gst_audio::AudioChannelPosition; 8]; 8] = [
|
||||
[
|
||||
gst_audio::AudioChannelPosition::Mono,
|
||||
gst_audio::AudioChannelPosition::Invalid,
|
||||
gst_audio::AudioChannelPosition::Invalid,
|
||||
gst_audio::AudioChannelPosition::Invalid,
|
||||
gst_audio::AudioChannelPosition::Invalid,
|
||||
gst_audio::AudioChannelPosition::Invalid,
|
||||
gst_audio::AudioChannelPosition::Invalid,
|
||||
gst_audio::AudioChannelPosition::Invalid,
|
||||
],
|
||||
[
|
||||
gst_audio::AudioChannelPosition::FrontLeft,
|
||||
gst_audio::AudioChannelPosition::FrontRight,
|
||||
gst_audio::AudioChannelPosition::Invalid,
|
||||
gst_audio::AudioChannelPosition::Invalid,
|
||||
gst_audio::AudioChannelPosition::Invalid,
|
||||
gst_audio::AudioChannelPosition::Invalid,
|
||||
gst_audio::AudioChannelPosition::Invalid,
|
||||
gst_audio::AudioChannelPosition::Invalid,
|
||||
],
|
||||
[
|
||||
gst_audio::AudioChannelPosition::FrontLeft,
|
||||
gst_audio::AudioChannelPosition::FrontCenter,
|
||||
gst_audio::AudioChannelPosition::FrontRight,
|
||||
gst_audio::AudioChannelPosition::Invalid,
|
||||
gst_audio::AudioChannelPosition::Invalid,
|
||||
gst_audio::AudioChannelPosition::Invalid,
|
||||
gst_audio::AudioChannelPosition::Invalid,
|
||||
gst_audio::AudioChannelPosition::Invalid,
|
||||
],
|
||||
[
|
||||
gst_audio::AudioChannelPosition::FrontLeft,
|
||||
gst_audio::AudioChannelPosition::FrontRight,
|
||||
gst_audio::AudioChannelPosition::RearLeft,
|
||||
gst_audio::AudioChannelPosition::RearRight,
|
||||
gst_audio::AudioChannelPosition::Invalid,
|
||||
gst_audio::AudioChannelPosition::Invalid,
|
||||
gst_audio::AudioChannelPosition::Invalid,
|
||||
gst_audio::AudioChannelPosition::Invalid,
|
||||
],
|
||||
[
|
||||
gst_audio::AudioChannelPosition::FrontLeft,
|
||||
gst_audio::AudioChannelPosition::FrontCenter,
|
||||
gst_audio::AudioChannelPosition::FrontRight,
|
||||
gst_audio::AudioChannelPosition::RearLeft,
|
||||
gst_audio::AudioChannelPosition::RearRight,
|
||||
gst_audio::AudioChannelPosition::Invalid,
|
||||
gst_audio::AudioChannelPosition::Invalid,
|
||||
gst_audio::AudioChannelPosition::Invalid,
|
||||
],
|
||||
[
|
||||
gst_audio::AudioChannelPosition::FrontLeft,
|
||||
gst_audio::AudioChannelPosition::FrontCenter,
|
||||
gst_audio::AudioChannelPosition::FrontRight,
|
||||
gst_audio::AudioChannelPosition::RearLeft,
|
||||
gst_audio::AudioChannelPosition::RearRight,
|
||||
gst_audio::AudioChannelPosition::Lfe1,
|
||||
gst_audio::AudioChannelPosition::Invalid,
|
||||
gst_audio::AudioChannelPosition::Invalid,
|
||||
],
|
||||
[
|
||||
gst_audio::AudioChannelPosition::FrontLeft,
|
||||
gst_audio::AudioChannelPosition::FrontCenter,
|
||||
gst_audio::AudioChannelPosition::FrontRight,
|
||||
gst_audio::AudioChannelPosition::SideLeft,
|
||||
gst_audio::AudioChannelPosition::SideRight,
|
||||
gst_audio::AudioChannelPosition::RearCenter,
|
||||
gst_audio::AudioChannelPosition::Lfe1,
|
||||
gst_audio::AudioChannelPosition::Invalid,
|
||||
],
|
||||
[
|
||||
gst_audio::AudioChannelPosition::FrontLeft,
|
||||
gst_audio::AudioChannelPosition::FrontCenter,
|
||||
gst_audio::AudioChannelPosition::FrontRight,
|
||||
gst_audio::AudioChannelPosition::SideLeft,
|
||||
gst_audio::AudioChannelPosition::SideRight,
|
||||
gst_audio::AudioChannelPosition::RearLeft,
|
||||
gst_audio::AudioChannelPosition::RearRight,
|
||||
gst_audio::AudioChannelPosition::Lfe1,
|
||||
],
|
||||
];
|
||||
|
||||
const GST_VORBIS_CHANNEL_POSITIONS: [[gst_audio::AudioChannelPosition; 8]; 8] = [
|
||||
[
|
||||
gst_audio::AudioChannelPosition::Mono,
|
||||
gst_audio::AudioChannelPosition::Invalid,
|
||||
gst_audio::AudioChannelPosition::Invalid,
|
||||
gst_audio::AudioChannelPosition::Invalid,
|
||||
gst_audio::AudioChannelPosition::Invalid,
|
||||
gst_audio::AudioChannelPosition::Invalid,
|
||||
gst_audio::AudioChannelPosition::Invalid,
|
||||
gst_audio::AudioChannelPosition::Invalid,
|
||||
],
|
||||
[
|
||||
gst_audio::AudioChannelPosition::FrontLeft,
|
||||
gst_audio::AudioChannelPosition::FrontRight,
|
||||
gst_audio::AudioChannelPosition::Invalid,
|
||||
gst_audio::AudioChannelPosition::Invalid,
|
||||
gst_audio::AudioChannelPosition::Invalid,
|
||||
gst_audio::AudioChannelPosition::Invalid,
|
||||
gst_audio::AudioChannelPosition::Invalid,
|
||||
gst_audio::AudioChannelPosition::Invalid,
|
||||
],
|
||||
[
|
||||
gst_audio::AudioChannelPosition::FrontLeft,
|
||||
gst_audio::AudioChannelPosition::FrontRight,
|
||||
gst_audio::AudioChannelPosition::FrontCenter,
|
||||
gst_audio::AudioChannelPosition::Invalid,
|
||||
gst_audio::AudioChannelPosition::Invalid,
|
||||
gst_audio::AudioChannelPosition::Invalid,
|
||||
gst_audio::AudioChannelPosition::Invalid,
|
||||
gst_audio::AudioChannelPosition::Invalid,
|
||||
],
|
||||
[
|
||||
gst_audio::AudioChannelPosition::FrontLeft,
|
||||
gst_audio::AudioChannelPosition::FrontRight,
|
||||
gst_audio::AudioChannelPosition::RearLeft,
|
||||
gst_audio::AudioChannelPosition::RearRight,
|
||||
gst_audio::AudioChannelPosition::Invalid,
|
||||
gst_audio::AudioChannelPosition::Invalid,
|
||||
gst_audio::AudioChannelPosition::Invalid,
|
||||
gst_audio::AudioChannelPosition::Invalid,
|
||||
],
|
||||
[
|
||||
gst_audio::AudioChannelPosition::FrontLeft,
|
||||
gst_audio::AudioChannelPosition::FrontRight,
|
||||
gst_audio::AudioChannelPosition::FrontCenter,
|
||||
gst_audio::AudioChannelPosition::RearLeft,
|
||||
gst_audio::AudioChannelPosition::RearRight,
|
||||
gst_audio::AudioChannelPosition::Invalid,
|
||||
gst_audio::AudioChannelPosition::Invalid,
|
||||
gst_audio::AudioChannelPosition::Invalid,
|
||||
],
|
||||
[
|
||||
gst_audio::AudioChannelPosition::FrontLeft,
|
||||
gst_audio::AudioChannelPosition::FrontRight,
|
||||
gst_audio::AudioChannelPosition::FrontCenter,
|
||||
gst_audio::AudioChannelPosition::Lfe1,
|
||||
gst_audio::AudioChannelPosition::RearLeft,
|
||||
gst_audio::AudioChannelPosition::RearRight,
|
||||
gst_audio::AudioChannelPosition::Invalid,
|
||||
gst_audio::AudioChannelPosition::Invalid,
|
||||
],
|
||||
[
|
||||
gst_audio::AudioChannelPosition::FrontLeft,
|
||||
gst_audio::AudioChannelPosition::FrontRight,
|
||||
gst_audio::AudioChannelPosition::FrontCenter,
|
||||
gst_audio::AudioChannelPosition::Lfe1,
|
||||
gst_audio::AudioChannelPosition::RearCenter,
|
||||
gst_audio::AudioChannelPosition::SideLeft,
|
||||
gst_audio::AudioChannelPosition::SideRight,
|
||||
gst_audio::AudioChannelPosition::Invalid,
|
||||
],
|
||||
[
|
||||
gst_audio::AudioChannelPosition::FrontLeft,
|
||||
gst_audio::AudioChannelPosition::FrontRight,
|
||||
gst_audio::AudioChannelPosition::FrontCenter,
|
||||
gst_audio::AudioChannelPosition::Lfe1,
|
||||
gst_audio::AudioChannelPosition::RearLeft,
|
||||
gst_audio::AudioChannelPosition::RearRight,
|
||||
gst_audio::AudioChannelPosition::SideLeft,
|
||||
gst_audio::AudioChannelPosition::SideRight,
|
||||
],
|
||||
];
|
||||
32
gst-plugin-lewton/src/lib.rs
Normal file
32
gst-plugin-lewton/src/lib.rs
Normal file
|
|
@ -0,0 +1,32 @@
|
|||
// 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.
|
||||
|
||||
#![crate_type = "cdylib"]
|
||||
|
||||
#[macro_use]
|
||||
extern crate glib;
|
||||
#[macro_use]
|
||||
extern crate gstreamer as gst;
|
||||
|
||||
mod lewtondec;
|
||||
|
||||
fn plugin_init(plugin: &gst::Plugin) -> Result<(), glib::BoolError> {
|
||||
lewtondec::register(plugin)
|
||||
}
|
||||
|
||||
gst_plugin_define!(
|
||||
lewton,
|
||||
env!("CARGO_PKG_DESCRIPTION"),
|
||||
plugin_init,
|
||||
concat!(env!("CARGO_PKG_VERSION"), "-", env!("COMMIT_ID")),
|
||||
"MIT/X11",
|
||||
env!("CARGO_PKG_NAME"),
|
||||
env!("CARGO_PKG_NAME"),
|
||||
env!("CARGO_PKG_REPOSITORY"),
|
||||
env!("BUILD_REL_DATE")
|
||||
);
|
||||
107
gst-plugin-lewton/tests/lewtondec.rs
Normal file
107
gst-plugin-lewton/tests/lewtondec.rs
Normal file
|
|
@ -0,0 +1,107 @@
|
|||
// 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_string())
|
||||
.field("rate", &44_100i32)
|
||||
.field("channels", &1i32)
|
||||
.field("layout", &"interleaved")
|
||||
.build()
|
||||
);
|
||||
}
|
||||
BIN
gst-plugin-lewton/tests/test.vorbis
Normal file
BIN
gst-plugin-lewton/tests/test.vorbis
Normal file
Binary file not shown.
Loading…
Reference in a new issue