cube_rs 0.4.7

Universal GameCube file format tool
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28

use crate::{rarc::Rarc, virtual_fs::VirtualFile};
use std::io::Cursor;
use yaz0::{Error as Yaz0Error, Yaz0Archive, Yaz0Writer};

/// Extracts an (optionally Yaz0 compressed) SZS archive into a list of files with
/// their respective paths and raw contents.
pub fn extract_szs(data: Vec<u8>) -> Result<Vec<VirtualFile>, Yaz0Error> {
    let arc = if &data[..4] == b"Yaz0" {
        Yaz0Archive::new(Cursor::new(data))?.decompress()?
    } else {
        data
    };
    let rarc = Rarc::parse(arc.as_slice()).expect("Rarc decompression error!");
    Ok(rarc
        .files()
        .map(|(path, bytes)| VirtualFile {
            path,
            bytes: bytes.to_vec(),
        })
        .collect())
}

pub fn yaz0_compress(bytes: &[u8]) -> Result<Vec<u8>, Yaz0Error> {
    let mut out = Vec::new();
    let yaz0_writer = Yaz0Writer::new(&mut out);
    yaz0_writer.compress_and_write(bytes, yaz0::CompressionLevel::Lookahead { quality: 10 })?;
    Ok(out)
}