From 7ca91b6f33bb260bc966e853c832b4339c1c2ae0 Mon Sep 17 00:00:00 2001 From: metamuffin Date: Thu, 12 Jan 2023 22:05:32 +0100 Subject: implementing my own ebml library --- remuxer/src/format/ebml/write.rs | 38 ++++++++++++++++++++++++++++++++++++++ 1 file changed, 38 insertions(+) create mode 100644 remuxer/src/format/ebml/write.rs (limited to 'remuxer/src/format/ebml/write.rs') diff --git a/remuxer/src/format/ebml/write.rs b/remuxer/src/format/ebml/write.rs new file mode 100644 index 0000000..1dee225 --- /dev/null +++ b/remuxer/src/format/ebml/write.rs @@ -0,0 +1,38 @@ +use anyhow::{bail, Result}; +use std::io::Write; + +pub struct EbmlWriter { + inner: T, + position: usize, +} + +impl EbmlWriter { + pub fn new(inner: T, position: usize) -> Self { + Self { inner, position } + } + + pub fn write(&mut self, data: &[u8]) -> Result<()> { + self.inner.write_all(data)?; + self.position += data.len(); + Ok(()) + } + + pub fn write_vint(&mut self, i: u64) -> Result<()> { + if i > (1 << 56) - 1 { + bail!("vint does not fit"); + } + let mut len = 1; + while len <= 8 { + if i < (1 << ((7 * len) - 1)) { + break; + } + len += 1; + } + let mut bytes = i.to_be_bytes(); + bytes[0] |= 1 << (8 - len); + self.write(&bytes) + } + +} + + -- cgit v1.2.3-70-g09d2