physis/
uwb.rs

1// SPDX-FileCopyrightText: 2025 Joshua Goins <josh@redstrate.com>
2// SPDX-License-Identifier: GPL-3.0-or-later
3
4use std::io::Cursor;
5
6use crate::ByteBuffer;
7use crate::ByteSpan;
8use binrw::BinRead;
9use binrw::BinWrite;
10use binrw::binrw;
11
12#[binrw]
13#[derive(Debug)]
14#[brw(little)]
15#[brw(magic = b"UWB1")]
16pub struct Uwb {
17    /// Including this header
18    pub file_size: u32,
19    /// Number of UWC's
20    #[br(temp)]
21    #[bw(calc = uwcs.len() as u32)]
22    uwc_count: u32,
23    #[br(count = uwc_count)]
24    pub uwcs: Vec<Uwc>,
25}
26
27#[binrw]
28#[derive(Debug)]
29#[brw(little)]
30#[brw(magic = b"UWC1")]
31pub struct Uwc {
32    /// Including this header
33    pub file_size: u32,
34    // TODO: figure out what this is
35    #[br(count = file_size - 8)]
36    pub unk1: Vec<u8>,
37}
38
39impl Uwb {
40    /// Reads an existing UWB file
41    pub fn from_existing(buffer: ByteSpan) -> Option<Self> {
42        let mut cursor = Cursor::new(buffer);
43        Uwb::read(&mut cursor).ok()
44    }
45
46    pub fn write_to_buffer(&self) -> Option<ByteBuffer> {
47        let mut buffer = ByteBuffer::new();
48
49        {
50            let mut cursor = Cursor::new(&mut buffer);
51            self.write_le(&mut cursor).ok()?;
52        }
53
54        Some(buffer)
55    }
56}
57
58#[cfg(test)]
59mod tests {
60    use std::fs::read;
61    use std::path::PathBuf;
62
63    use super::*;
64
65    #[test]
66    fn test_invalid() {
67        let mut d = PathBuf::from(env!("CARGO_MANIFEST_DIR"));
68        d.push("resources/tests");
69        d.push("random");
70
71        // Feeding it invalid data should not panic
72        Uwb::from_existing(&read(d).unwrap());
73    }
74}