Compare commits

..

No commits in common. "6fa7b62d2676c356300758724b6a2f41fc9cd12b" and "3a07eae54ff7ffee98f134e5721cbad7b322a847" have entirely different histories.

5 changed files with 11 additions and 55 deletions

View file

@ -53,12 +53,7 @@ pub enum Node {
impl Node { impl Node {
fn build(counts: &HashMap<u8, u32>) -> Option<Self> { fn build(counts: &HashMap<u8, u32>) -> Option<Self> {
let branch = match WeightedNode::build(counts).map(WeightedNode::unburden)? { WeightedNode::build(counts).map(WeightedNode::unburden)
leaf @ Node::Leaf { .. } => Self::join(leaf.clone(), leaf),
branch @ Node::Branch { .. } => branch,
};
Some(branch)
} }
/// Write to `buf` the sequence of bits that this tree has assigned `byte`. /// Write to `buf` the sequence of bits that this tree has assigned `byte`.
@ -125,13 +120,6 @@ impl Node {
} }
} }
} }
fn join(path0: Self, path1: Self) -> Self {
Self::Branch {
path0: Box::new(path0),
path1: Box::new(path1),
}
}
} }
impl fmt::Debug for Node { impl fmt::Debug for Node {

View file

@ -1,28 +0,0 @@
use std::io::Read;
use bitvec::prelude::{BitSlice, BitVec};
use crate::CompressionScheme;
#[derive(Debug, Clone, Copy, PartialEq, Eq, Hash)]
pub struct Id;
impl CompressionScheme for Id {
type Header = ();
fn encode(src: &[u8], buf: &mut BitVec) -> Self::Header {
buf.extend_from_bitslice(BitSlice::<u8>::from_slice(src));
}
fn decode(src: &BitSlice, (): &Self::Header, buf: &mut Vec<u8>) {
// TODO: is this actually fine
#[allow(clippy::unbuffered_bytes)]
for byte in src.bytes() {
buf.push(byte.unwrap());
}
}
fn header_size((): &Self::Header) -> usize {
0
}
}

View file

@ -7,7 +7,6 @@ use std::{
mod compression_scheme; mod compression_scheme;
mod freq; mod freq;
mod huffman; mod huffman;
mod id;
mod rle; mod rle;
#[cfg(test)] #[cfg(test)]
mod test; mod test;
@ -16,18 +15,18 @@ use bitvec::vec::BitVec;
pub use compression_scheme::CompressionScheme; pub use compression_scheme::CompressionScheme;
pub use freq::Freq; pub use freq::Freq;
pub use huffman::Huffman; pub use huffman::Huffman;
pub use id::Id;
pub use rle::Rle; pub use rle::Rle;
fn main() -> Result<(), io::Error> { fn main() -> Result<(), io::Error> {
let debug = env::args().any(|arg| arg == "--debug" || arg == "-d"); let debug = env::args().any(|arg| arg == "--debug" || arg == "-d");
let mut buf = Vec::new(); let mut buf = Vec::new();
io::stdin().read_to_end(&mut buf)?; let len_src = io::stdin().read_to_end(&mut buf)?;
println!("Original: {len_src}");
let mut bitbuf = BitVec::new(); let mut bitbuf = BitVec::new();
run::<Id, _>(&buf, &mut bitbuf, "Original", debug);
run::<Rle, _>(&buf, &mut bitbuf, "rle", debug); run::<Rle, _>(&buf, &mut bitbuf, "rle", debug);
run::<Freq, _>(&buf, &mut bitbuf, "freq", debug); run::<Freq, _>(&buf, &mut bitbuf, "freq", debug);
run::<Huffman, _>(&buf, &mut bitbuf, "Huffman", debug); run::<Huffman, _>(&buf, &mut bitbuf, "Huffman", debug);
@ -43,7 +42,7 @@ where
bitbuf.clear(); bitbuf.clear();
let header = Scheme::encode(buf, bitbuf); let header = Scheme::encode(buf, bitbuf);
let len_freq = Scheme::header_size(&header) + bitbuf.len().div_ceil(8); let len_freq = Scheme::header_size(&header) + bitbuf.len().div_ceil(8);
println!("{name:>10}: {len_freq:>10}"); println!("{name}'d: {len_freq}");
if debug { if debug {
eprintln!("{name} header: {header:#?}\n"); eprintln!("{name} header: {header:#?}\n");
} }

View file

@ -1,24 +1,21 @@
#![allow(clippy::needless_pass_by_value)]
use crate::{CompressionScheme, Freq, Huffman, Id, Rle};
use quickcheck_macros::quickcheck; use quickcheck_macros::quickcheck;
use crate::{CompressionScheme, Freq, Huffman, Rle};
#[allow(clippy::needless_pass_by_value)]
#[quickcheck] #[quickcheck]
fn roundtrip_freq(src: Vec<u8>) -> bool { fn roundtrip_freq(src: Vec<u8>) -> bool {
Freq::idempotent_on(&src) Freq::idempotent_on(&src)
} }
#[allow(clippy::needless_pass_by_value)]
#[quickcheck] #[quickcheck]
fn roundtrip_rle(src: Vec<u8>) -> bool { fn roundtrip_rle(src: Vec<u8>) -> bool {
Rle::idempotent_on(&src) Rle::idempotent_on(&src)
} }
#[allow(clippy::needless_pass_by_value)]
#[quickcheck] #[quickcheck]
fn roundtrip_huffman(src: Vec<u8>) -> bool { fn roundtrip_huffman(src: Vec<u8>) -> bool {
Huffman::idempotent_on(&src) Huffman::idempotent_on(&src)
} }
#[quickcheck]
fn roundtrip_id(src: Vec<u8>) -> bool {
Id::idempotent_on(&src)
}