From ed6ed7a62217369544f3e31ef9a886f459f0c21b Mon Sep 17 00:00:00 2001
From: metamuffin <metamuffin@disroot.org>
Date: Sat, 15 Mar 2025 21:31:40 +0100
Subject: assetbundle struct to abstract over unityfs and serializedfile

---
 src/serialized_file.rs | 7 +++++++
 1 file changed, 7 insertions(+)

(limited to 'src/serialized_file.rs')

diff --git a/src/serialized_file.rs b/src/serialized_file.rs
index a6514ee..aba72b9 100644
--- a/src/serialized_file.rs
+++ b/src/serialized_file.rs
@@ -345,6 +345,13 @@ impl<T: Read + Seek> SerializedFile<T> {
             .ok_or(anyhow!("type tree missing"))?;
         Value::read(typetree, self.endianness, &mut self.file)
     }
+
+    pub fn all_objects_of_class(&self, class_name: &str) -> impl Iterator<Item = &ObjectInfo> {
+        self.objects.iter().filter(move |o| {
+            self.get_object_type_tree(&o)
+                .map_or(false, |t| t.type_string == class_name)
+        })
+    }
 }
 
 impl TypeTreeNode {
-- 
cgit v1.2.3-70-g09d2