mirror of
https://gitlab.freedesktop.org/gstreamer/gstreamer-rs.git
synced 2024-07-01 04:20:35 +00:00
BufferList: serde impl
This commit is contained in:
parent
862afe5cc7
commit
0f1d3bea89
|
@ -193,3 +193,195 @@ impl<'a> DoubleEndedIterator for Iter<'a> {
|
||||||
}
|
}
|
||||||
|
|
||||||
impl<'a> ExactSizeIterator for Iter<'a> {}
|
impl<'a> ExactSizeIterator for Iter<'a> {}
|
||||||
|
|
||||||
|
#[cfg(feature = "ser_de")]
|
||||||
|
pub(crate) mod serde {
|
||||||
|
use serde::de::{Deserialize, Deserializer, SeqAccess, Visitor};
|
||||||
|
use serde::ser::{Serialize, Serializer, SerializeSeq};
|
||||||
|
|
||||||
|
use std::fmt;
|
||||||
|
|
||||||
|
use Buffer;
|
||||||
|
use BufferList;
|
||||||
|
use BufferListRef;
|
||||||
|
|
||||||
|
impl Serialize for BufferListRef {
|
||||||
|
fn serialize<S: Serializer>(&self, serializer: S) -> Result<S::Ok, S::Error> {
|
||||||
|
let mut iter = self.iter();
|
||||||
|
let (remaining, _) = iter.size_hint();
|
||||||
|
if remaining > 0 {
|
||||||
|
let mut seq = serializer.serialize_seq(Some(remaining))?;
|
||||||
|
while let Some(ref buffer) = iter.next() {
|
||||||
|
seq.serialize_element(buffer)?;
|
||||||
|
}
|
||||||
|
seq.end()
|
||||||
|
} else {
|
||||||
|
let seq = serializer.serialize_seq(None)?;
|
||||||
|
seq.end()
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
impl Serialize for BufferList {
|
||||||
|
fn serialize<S: Serializer>(&self, serializer: S) -> Result<S::Ok, S::Error> {
|
||||||
|
self.as_ref().serialize(serializer)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
struct BufferListVisitor;
|
||||||
|
impl<'de> Visitor<'de> for BufferListVisitor {
|
||||||
|
type Value = BufferList;
|
||||||
|
|
||||||
|
fn expecting(&self, formatter: &mut fmt::Formatter) -> fmt::Result {
|
||||||
|
formatter.write_str("a sequence of Buffers")
|
||||||
|
}
|
||||||
|
|
||||||
|
fn visit_seq<A: SeqAccess<'de>>(self, mut seq: A) -> Result<Self::Value, A::Error> {
|
||||||
|
let mut buffer_list = BufferList::new();
|
||||||
|
{
|
||||||
|
let buffer_list = buffer_list.get_mut().unwrap();
|
||||||
|
while let Some(buffer) = seq.next_element::<Buffer>()? {
|
||||||
|
buffer_list.add(buffer);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
Ok(buffer_list)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
impl<'de> Deserialize<'de> for BufferList {
|
||||||
|
fn deserialize<D: Deserializer<'de>>(deserializer: D) -> Result<Self, D::Error> {
|
||||||
|
deserializer.deserialize_seq(BufferListVisitor)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
#[cfg(test)]
|
||||||
|
mod tests {
|
||||||
|
#[cfg(feature = "ser_de")]
|
||||||
|
#[test]
|
||||||
|
fn test_serialize() {
|
||||||
|
extern crate ron;
|
||||||
|
|
||||||
|
use Buffer;
|
||||||
|
use BufferList;
|
||||||
|
|
||||||
|
::init().unwrap();
|
||||||
|
|
||||||
|
let mut buffer_list = BufferList::new();
|
||||||
|
{
|
||||||
|
let buffer_list = buffer_list.get_mut().unwrap();
|
||||||
|
|
||||||
|
let mut buffer = Buffer::from_slice(vec![1, 2, 3, 4]).unwrap();
|
||||||
|
{
|
||||||
|
let buffer = buffer.get_mut().unwrap();
|
||||||
|
buffer.set_pts(1.into());
|
||||||
|
buffer.set_offset(0);
|
||||||
|
buffer.set_offset_end(4);
|
||||||
|
buffer.set_duration(4.into());
|
||||||
|
}
|
||||||
|
buffer_list.add(buffer);
|
||||||
|
|
||||||
|
let mut buffer = Buffer::from_slice(vec![5, 6]).unwrap();
|
||||||
|
{
|
||||||
|
let buffer = buffer.get_mut().unwrap();
|
||||||
|
buffer.set_pts(5.into());
|
||||||
|
buffer.set_offset(4);
|
||||||
|
buffer.set_offset_end(6);
|
||||||
|
buffer.set_duration(2.into());
|
||||||
|
}
|
||||||
|
buffer_list.add(buffer);
|
||||||
|
}
|
||||||
|
|
||||||
|
// don't use newlines
|
||||||
|
let mut pretty_config = ron::ser::PrettyConfig::default();
|
||||||
|
pretty_config.new_line = "".to_string();
|
||||||
|
|
||||||
|
let res = ron::ser::to_string_pretty(&buffer_list, pretty_config);
|
||||||
|
assert_eq!(
|
||||||
|
Ok(
|
||||||
|
concat!(
|
||||||
|
"[",
|
||||||
|
" (",
|
||||||
|
" pts: Some(1),",
|
||||||
|
" dts: None,",
|
||||||
|
" duration: Some(4),",
|
||||||
|
" offset: 0,",
|
||||||
|
" offset_end: 4,",
|
||||||
|
" flags: (",
|
||||||
|
" bits: 0,",
|
||||||
|
" ),",
|
||||||
|
" buffer: \"AQIDBA==\",",
|
||||||
|
" ),",
|
||||||
|
" (",
|
||||||
|
" pts: Some(5),",
|
||||||
|
" dts: None,",
|
||||||
|
" duration: Some(2),",
|
||||||
|
" offset: 4,",
|
||||||
|
" offset_end: 6,",
|
||||||
|
" flags: (",
|
||||||
|
" bits: 0,",
|
||||||
|
" ),",
|
||||||
|
" buffer: \"BQY=\",",
|
||||||
|
" ),",
|
||||||
|
"]"
|
||||||
|
)
|
||||||
|
.to_owned()
|
||||||
|
),
|
||||||
|
res,
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
#[cfg(feature = "ser_de")]
|
||||||
|
#[test]
|
||||||
|
fn test_deserialize() {
|
||||||
|
extern crate ron;
|
||||||
|
|
||||||
|
use BufferList;
|
||||||
|
|
||||||
|
::init().unwrap();
|
||||||
|
|
||||||
|
let buffer_list_ron = r#"
|
||||||
|
[
|
||||||
|
(
|
||||||
|
pts: Some(1),
|
||||||
|
dts: None,
|
||||||
|
duration: Some(4),
|
||||||
|
offset: 0,
|
||||||
|
offset_end: 4,
|
||||||
|
flags: (
|
||||||
|
bits: 0,
|
||||||
|
),
|
||||||
|
buffer: "AQIDBA==",
|
||||||
|
),
|
||||||
|
(
|
||||||
|
pts: Some(5),
|
||||||
|
dts: None,
|
||||||
|
duration: Some(2),
|
||||||
|
offset: 4,
|
||||||
|
offset_end: 6,
|
||||||
|
flags: (
|
||||||
|
bits: 0,
|
||||||
|
),
|
||||||
|
buffer: "BQY=",
|
||||||
|
),
|
||||||
|
]
|
||||||
|
"#;
|
||||||
|
|
||||||
|
let buffer_list: BufferList = ron::de::from_str(buffer_list_ron).unwrap();
|
||||||
|
let mut iter = buffer_list.iter();
|
||||||
|
let buffer = iter.next().unwrap();
|
||||||
|
assert_eq!(buffer.get_pts(), 1.into());
|
||||||
|
assert_eq!(buffer.get_dts(), None.into());
|
||||||
|
{
|
||||||
|
let data = buffer.map_readable().unwrap();
|
||||||
|
assert_eq!(data.as_slice(), vec![1, 2, 3, 4].as_slice());
|
||||||
|
}
|
||||||
|
|
||||||
|
let buffer = iter.next().unwrap();
|
||||||
|
assert_eq!(buffer.get_pts(), 5.into());
|
||||||
|
{
|
||||||
|
let data = buffer.map_readable().unwrap();
|
||||||
|
assert_eq!(data.as_slice(), vec![5, 6].as_slice());
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
Loading…
Reference in a new issue