Initial vendor packages
Signed-off-by: Valentin Popov <valentin@popov.link>
This commit is contained in:
159
vendor/weezl/bin/lzw.rs
vendored
Normal file
159
vendor/weezl/bin/lzw.rs
vendored
Normal file
@ -0,0 +1,159 @@
|
||||
#![forbid(unsafe_code)]
|
||||
use std::path::PathBuf;
|
||||
use std::{env, ffi, fs, io, process};
|
||||
|
||||
extern crate weezl;
|
||||
use weezl::{decode as delzw, encode as enlzw, BitOrder};
|
||||
|
||||
fn main() {
|
||||
let args = env::args_os().skip(1);
|
||||
let flags = Flags::from_args(args).unwrap_or_else(|ParamError| explain());
|
||||
|
||||
let out = io::stdout();
|
||||
let out = out.lock();
|
||||
|
||||
let mut files = flags.files;
|
||||
let input = files.pop().unwrap_or_else(explain);
|
||||
if !files.is_empty() {
|
||||
return explain();
|
||||
}
|
||||
let operation = flags.operation.unwrap_or_else(explain);
|
||||
let min_code = if flags.min_code < 2 || flags.min_code > 12 {
|
||||
return explain();
|
||||
} else {
|
||||
flags.min_code
|
||||
};
|
||||
let bit_order = flags.bit_order;
|
||||
|
||||
let result = match (input, operation) {
|
||||
(Input::File(file), Operation::Encode) => (|| {
|
||||
let data = fs::File::open(file)?;
|
||||
let file = io::BufReader::with_capacity(1 << 26, data);
|
||||
|
||||
let mut encoder = enlzw::Encoder::new(bit_order, min_code);
|
||||
encoder.into_stream(out).encode_all(file).status
|
||||
})(),
|
||||
(Input::Stdin, Operation::Encode) => {
|
||||
let input = io::BufReader::with_capacity(1 << 26, io::stdin());
|
||||
let mut encoder = enlzw::Encoder::new(bit_order, min_code);
|
||||
encoder.into_stream(out).encode_all(input).status
|
||||
}
|
||||
(Input::File(file), Operation::Decode) => (|| {
|
||||
let data = fs::File::open(file)?;
|
||||
let file = io::BufReader::with_capacity(1 << 26, data);
|
||||
|
||||
let mut decoder = delzw::Decoder::new(bit_order, min_code);
|
||||
decoder.into_stream(out).decode_all(file).status
|
||||
})(),
|
||||
(Input::Stdin, Operation::Decode) => {
|
||||
let input = io::BufReader::with_capacity(1 << 26, io::stdin());
|
||||
let mut decoder = delzw::Decoder::new(bit_order, min_code);
|
||||
decoder.into_stream(out).decode_all(input).status
|
||||
}
|
||||
};
|
||||
|
||||
result.expect("Operation Failed: ");
|
||||
}
|
||||
|
||||
struct Flags {
|
||||
files: Vec<Input>,
|
||||
operation: Option<Operation>,
|
||||
min_code: u8,
|
||||
bit_order: BitOrder,
|
||||
}
|
||||
|
||||
struct ParamError;
|
||||
|
||||
enum Input {
|
||||
File(PathBuf),
|
||||
Stdin,
|
||||
}
|
||||
|
||||
enum Operation {
|
||||
Encode,
|
||||
Decode,
|
||||
}
|
||||
|
||||
fn explain<T>() -> T {
|
||||
println!(
|
||||
"Usage: lzw [-e|-d] <file>\n\
|
||||
Arguments:\n\
|
||||
-e\t operation encode (default)\n\
|
||||
-d\t operation decode\n\
|
||||
<file>\tfilepath or '-' for stdin"
|
||||
);
|
||||
process::exit(1);
|
||||
}
|
||||
|
||||
impl Default for Flags {
|
||||
fn default() -> Flags {
|
||||
Flags {
|
||||
files: vec![],
|
||||
operation: None,
|
||||
min_code: 8,
|
||||
bit_order: BitOrder::Msb,
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
impl Flags {
|
||||
fn from_args(mut args: impl Iterator<Item = ffi::OsString>) -> Result<Self, ParamError> {
|
||||
let mut flags = Flags::default();
|
||||
let mut operation = None;
|
||||
loop {
|
||||
match args.next().as_ref().and_then(|s| s.to_str()) {
|
||||
Some("-d") | Some("--decode") => {
|
||||
if operation.is_some() {
|
||||
return Err(ParamError);
|
||||
}
|
||||
operation = Some(Operation::Decode);
|
||||
}
|
||||
Some("-e") | Some("--encode") => {
|
||||
if operation.is_some() {
|
||||
return Err(ParamError);
|
||||
}
|
||||
operation = Some(Operation::Encode);
|
||||
}
|
||||
Some("-w") | Some("--word-bits") => match args.next() {
|
||||
None => return Err(ParamError),
|
||||
Some(bits) => {
|
||||
let st = bits.to_str().ok_or(ParamError)?;
|
||||
flags.min_code = st.parse().ok().ok_or(ParamError)?;
|
||||
}
|
||||
},
|
||||
Some("-le") | Some("--little-endian") => {
|
||||
flags.bit_order = BitOrder::Lsb;
|
||||
}
|
||||
Some("-be") | Some("--big-endian") | Some("-ne") | Some("--network-endian") => {
|
||||
flags.bit_order = BitOrder::Msb;
|
||||
}
|
||||
Some("-") => {
|
||||
flags.files.push(Input::Stdin);
|
||||
}
|
||||
Some(other) if other.starts_with('-') => {
|
||||
// Reserved for future use.
|
||||
// -a: self-describing archive format, similar to actual compress
|
||||
// -b: maximum bits
|
||||
// -v: verbosity
|
||||
// some compress compatibility mode? Probably through arg(0) though.
|
||||
return Err(ParamError);
|
||||
}
|
||||
Some(file) => {
|
||||
flags.files.push(Input::File(file.into()));
|
||||
}
|
||||
None => break,
|
||||
};
|
||||
}
|
||||
|
||||
flags.files.extend(args.map(|file| {
|
||||
if let Some("-") = file.to_str() {
|
||||
Input::Stdin
|
||||
} else {
|
||||
Input::File(file.into())
|
||||
}
|
||||
}));
|
||||
|
||||
flags.operation = operation;
|
||||
Ok(flags)
|
||||
}
|
||||
}
|
Reference in New Issue
Block a user