tangent128/webmetro
1
0
Fork 0
Code Issues 1 Pull requests Releases Packages Wiki Activity

Move Chunk timecode updating into the struct impl proper.

Browse source
This commit is contained in:
Tangent 128 2017-09-25 01:41:56 -04:00
parent d0f1cde464
commit c3f0de71fa
1 changed files with 17 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

24
src/chunk.rs
View file

@ -1,4 +1,6 @@
use std::io::Cursor;
use std::rc::Rc; use std::rc::Rc;
use webm::*;
#[derive(Clone)] #[derive(Clone)]
pub enum Chunk<B: AsRef<[u8]> = Vec<u8>> { pub enum Chunk<B: AsRef<[u8]> = Vec<u8>> {
@ -17,6 +19,20 @@ pub enum Chunk<B: AsRef<[u8]> = Vec<u8>> {
} }
} }
impl<B: AsRef<[u8]>> Chunk<B> {
pub fn update_timecode(&mut self, timecode: u64) {
if let &mut Chunk::ClusterHead {ref mut start, ref mut end, ref mut bytes, ..} = self {
let delta = *end - *start;
*start = timecode;
*end = *start + delta;
let mut cursor = Cursor::new(bytes as &mut [u8]);
// buffer is sized so these should never fail
encode_webm_element(&WebmElement::Cluster, &mut cursor).unwrap();
encode_webm_element(&WebmElement::Timecode(timecode), &mut cursor).unwrap();
}
}
}
impl<B: AsRef<[u8]>> AsRef<[u8]> for Chunk<B> { impl<B: AsRef<[u8]>> AsRef<[u8]> for Chunk<B> {
fn as_ref(&self) -> &[u8] { fn as_ref(&self) -> &[u8] {
match self { match self {
@ -31,8 +47,6 @@ impl<B: AsRef<[u8]>> AsRef<[u8]> for Chunk<B> {
mod tests { mod tests {
use chunk::*; use chunk::*;
use std::io::Cursor;
use webm::*;
#[test] #[test]
fn enough_space_for_header() { fn enough_space_for_header() {
@ -42,10 +56,6 @@ mod tests {
end: 0, end: 0,
bytes: [0;16] bytes: [0;16]
}; };
if let Chunk::ClusterHead {ref mut bytes, ..} = chunk { chunk.update_timecode(u64::max_value());
let mut cursor = Cursor::new(bytes as &mut [u8]);
encode_webm_element(&WebmElement::Cluster, &mut cursor).unwrap();
encode_webm_element(&WebmElement::Timecode(u64::max_value()), &mut cursor).unwrap();
}
} }
} }