diff options
Diffstat (limited to 'src/s_stream/mod.rs')
-rw-r--r-- | src/s_stream/mod.rs | 88 |
1 files changed, 88 insertions, 0 deletions
diff --git a/src/s_stream/mod.rs b/src/s_stream/mod.rs new file mode 100644 index 0000000..454adb8 --- /dev/null +++ b/src/s_stream/mod.rs @@ -0,0 +1,88 @@ +// Copyright 2024 Gabriel Bjørnager Jensen. +// +// This file is part of bzipper. +// +// bzipper is free software: you can redistribute +// it and/or modify it under the terms of the GNU +// Lesser General Public License as published by +// the Free Software Foundation, either version 3 +// of the License, or (at your option) any later +// version. +// +// bzipper is distributed in the hope that it will +// be useful, but WITHOUT ANY WARRANTY; without +// even the implied warranty of MERCHANTABILITY or +// FITNESS FOR A PARTICULAR PURPOSE. See the GNU +// Lesser General Public License for more details. +// +// You should have received a copy of the GNU Less- +// er General Public License along with bzipper. If +// not, see <https://www.gnu.org/licenses/>. + +use crate::{DStream, Serialise}; + +use std::fmt::{Debug, Formatter}; +use std::mem::size_of; + +/// Byte stream for serialisation. +/// +/// The bytes themselves are contained by the type. +/// The stream can +#[derive(Clone, Eq, PartialEq)] +pub struct SStream(pub(in crate) Vec<u8>); + +impl SStream { + /// Constructs a new, empty byte stream. + #[inline(always)] + #[must_use] + pub const fn new() -> Self { Self(Vec::new()) } + + /// Extends the byte stream. + #[inline(always)] + pub fn append(&mut self, extra: &[u8]) { + self.0.extend(extra); + } + + /// Converts the stream to a `DStream` object. + /// + /// The returned object references the original stream. + #[inline(always)] + #[must_use] + pub fn as_d_stream(&self) -> DStream { DStream::new(&self.0) } +} + +impl AsRef<[u8]> for SStream { + #[inline(always)] + fn as_ref(&self) -> &[u8] { self.0.as_ref() } +} + +impl Debug for SStream { + fn fmt(&self, f: &mut Formatter) -> std::fmt::Result { + write!(f, "[")?; + + for v in &self.0 { write!(f, "{v:#02X},")? }; + + write!(f, "]")?; + + Ok(()) + } +} + +impl Default for SStream { + #[inline(always)] + fn default() -> Self { Self::new() } +} + +impl<T: Serialise> From<&T> for SStream { + fn from(value: &T) -> Self { + let mut stream = Self(Vec::with_capacity(size_of::<T>())); + value.serialise(&mut stream); + + stream + } +} + +impl From<SStream> for Box<[u8]> { + #[inline(always)] + fn from(value: SStream) -> Self { value.0.into_boxed_slice() } +} |