webmetro/src/chunk.rs

use std::rc::Rc;

#[derive(Clone)]
pub enum Chunk<B: AsRef<[u8]> = Vec<u8>> {
    Headers {
        bytes: Rc<B>
    },
    ClusterHead {
        keyframe: bool,
        start: u64,
        end: u64,
        // space for a Cluster tag and a Timecode tag
        bytes: [u8;16]
    },
    ClusterBody {
        bytes: Rc<B>
    }
}

impl<B: AsRef<[u8]>> AsRef<[u8]> for Chunk<B> {
    fn as_ref(&self) -> &[u8] {
        match self {
            &Chunk::Headers {ref bytes, ..} => bytes.as_ref().as_ref(),
            &Chunk::ClusterHead {ref bytes, ..} => bytes,
            &Chunk::ClusterBody {ref bytes, ..} => bytes.as_ref().as_ref()
        }
    }
}

#[cfg(test)]
mod tests {

    use chunk::*;
    use std::io::Cursor;
    use webm::*;

    #[test]
    fn enough_space_for_header() {
        let mut chunk: Chunk = Chunk::ClusterHead {
            keyframe: false,
            start: 0,
            end: 0,
            bytes: [0;16]
        };
        if let Chunk::ClusterHead {ref mut bytes, ..} = chunk {
            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();
        }
    }
}