assetbundle struct to abstract over unityfs and serializedfile

1 files changed, 8 insertions, 26 deletions

diff --git a/exporter/src/bin/meshes.rs b/exporter/src/bin/meshes.rs
index db8f2f8..4cd86e8 100644
--- a/exporter/src/bin/meshes.rs
+++ b/exporter/src/bin/meshes.rs
@@ -1,53 +1,35 @@
 #![feature(array_chunks)]
-use anyhow::anyhow;
+use glam::Vec3;
 use std::{
     env::args,
     fs::{File, create_dir_all},
     io::{BufReader, BufWriter, Write},
 };
 use unity_tools::{
+    assetbundle::AssetBundle,
     classes::mesh::{Mesh, VertexDataChannel},
-    object::parser::FromValue,
-    serialized_file::SerializedFile,
-    unityfs::UnityFS,
 };
 
 fn main() -> anyhow::Result<()> {
     env_logger::init_from_env("LOG");
     let file = BufReader::new(File::open(args().nth(1).unwrap()).unwrap());
-    let fs = UnityFS::open(file)?;
+    let mut bundle = AssetBundle::open(file)?;
 
     let mut i = 0;
     create_dir_all("/tmp/a").unwrap();
 
-    let cabfile = fs
-        .header
-        .nodes()
-        .iter()
-        .find(|n| !n.name.ends_with(".resource") && !n.name.ends_with(".resS"))
-        .ok_or(anyhow!("no CAB file found"))?
-        .to_owned();
-
-    let mut cab = fs.read(&cabfile)?;
-    let mut file = SerializedFile::read(&mut cab)?;
-    for ob in file.objects.clone() {
-        if file.get_object_type_tree(&ob)?.type_string != "Mesh" {
-            continue;
-        }
-        let value = file.read_object(ob)?;
-        let mesh = Mesh::from_value(value).unwrap();
+    for ob in bundle.all_toplevel_of_class("Mesh").collect::<Vec<_>>() {
+        let mesh = ob.load(&mut bundle)?.parse::<Mesh>()?;
         let mut obj = BufWriter::new(File::create(format!(
             "/tmp/a/{}_{i}.obj",
             mesh.name.replace("/", "-").replace(".", "-")
         ))?);
 
-        let (pos_dims, positions) = mesh
+        let positions = mesh
             .vertex_data
-            .read_channel(VertexDataChannel::Position)
+            .read_channel_vec::<Vec3>(VertexDataChannel::Position)?
             .unwrap();
-        assert_eq!(pos_dims, 3);
-
-        for [x, y, z] in positions.array_chunks() {
+        for Vec3 { x, y, z } in positions {
             writeln!(obj, "v {x} {y} {z}")?;
         }
         for [a, b, c] in mesh.read_indecies() {