usdz 0.0.1

USDZ file format parser
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
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88

mod zip;
use zip::*;
mod usd;
pub use usd::*;

pub struct UsdzFile {
    pub zip_file: ZipFile,
}

impl UsdzFile {
    pub fn parse(buffer: &[u8]) -> Result<UsdzFile, &'static str> {
        let zip_file = parse_zip_file(buffer)?;
        Ok(UsdzFile { zip_file })
    }

    pub fn get_files(&self) -> Vec<String> {
        let mut files = Vec::new();
        for part in &self.zip_file.parts {
            match part {
                ZipFilePart::LocalFileHeader(local_file_header) => {
                    files.push(local_file_header.file_name.clone());
                }
                _ => {}
            }
        }
        files
    }

    pub fn get_file_data(&self, file_name: &str) -> Option<Vec<u8>> {
        for part in &self.zip_file.parts {
            match part {
                ZipFilePart::LocalFileHeader(local_file_header) => {
                    if local_file_header.file_name == file_name {
                        return local_file_header.uncompressed_data.clone();
                    }
                }
                _ => {}
            }
        }
        None
    }
}

#[cfg(test)]
mod tests {
    use super::*;

    #[test]
    fn basic_unzip() {
        let buffer = include_bytes!("test_usdz/basic.usdz").to_vec();
        let zip_file = parse_zip_file(&buffer);
        assert_eq!(zip_file.is_ok(), true);
    }

    #[test]
    fn parse_usd() {
        let buffer = include_bytes!("./test_usdz/basic.usdz").to_vec();
        let zip_file = UsdzFile::parse(&buffer).unwrap();
        let files = zip_file.get_files();
        assert_eq!(files.len(), 1);
        let first_file = files.get(0).unwrap();
        assert_eq!(first_file, "basic/basic.usd");
        let file_data = zip_file.get_file_data(first_file).unwrap();
        let usd = Usd::parse(&file_data).unwrap();
        assert_eq!(usd.parts.len(), 2);
        if let UsdPart::Comment(comment) = &usd.parts[0] {
            assert_eq!(comment.0, "usda 1.0");
        } else {
            panic!("Expected comment");
        }
        if let UsdPart::Node(node) = &usd.parts[1] {
            assert_eq!(node.name, "hello");
            assert_eq!(node.kind, "Xform");
            assert_eq!(node.properties.len(), 0);
            assert_eq!(node.children.len(), 1);
            if let UsdPart::Node(child) = &node.children[0] {
                assert_eq!(child.name, "world");
                assert_eq!(child.kind, "Sphere");
                assert_eq!(child.properties.len(), 0);
                assert_eq!(child.children.len(), 0);
            } else {
                panic!("Expected node");
            }
        } else {
            panic!("Expected node");
        }
    }
}