improve PPtr functions

6 files changed, 42 insertions, 10 deletions

diff --git a/src/bin/meshes.rs b/src/bin/meshes.rs
index d59b496..e3758c4 100644
--- a/src/bin/meshes.rs
+++ b/src/bin/meshes.rs
@@ -16,7 +16,6 @@ fn main() -> anyhow::Result<()> {
     env_logger::init_from_env("LOG");
     let file = || BufReader::new(File::open(args().nth(1).unwrap()).unwrap());
     let mut fs = UnityFS::open(file())?;
-    // let mut fs2 = UnityFS::open(file())?;
 
     let mut i = 0;
     create_dir_all("/tmp/a").unwrap();
diff --git a/src/bin/typegraph.rs b/src/bin/typegraph.rs
index 7425778..ea55e05 100644
--- a/src/bin/typegraph.rs
+++ b/src/bin/typegraph.rs
@@ -41,7 +41,10 @@ fn main() -> anyhow::Result<()> {
                         if let Some(inner) = c.type_string.strip_prefix("PPtr<") {
                             c.type_string = inner.strip_suffix(">").unwrap().to_owned();
                             f = true;
-                        } else if matches!(c.type_string.as_str(), "Array" | "pair" | "map") {
+                        } else if matches!(
+                            c.type_string.as_str(),
+                            "Array" | "pair" | "map" | "vector"
+                        ) {
                             nc.extend(c.children);
                             f = true
                         } else {
@@ -105,7 +108,6 @@ fn is_primitive(s: &str) -> bool {
             | "SInt64"
             | "bool"
             | "float"
-            | "vector"
             | "string"
             | "float3"
             | "float4"
@@ -113,6 +115,9 @@ fn is_primitive(s: &str) -> bool {
             | "Vector2f"
             | "Vector3f"
             | "Vector4f"
+            | "Matrix4x4f"
+            | "ColorRGBA"
+            | "Rectf"
             | "Quaternionf"
             | "xform"
     )
diff --git a/src/classes/mesh.rs b/src/classes/mesh.rs
index b8b9e90..f0b2173 100644
--- a/src/classes/mesh.rs
+++ b/src/classes/mesh.rs
@@ -85,6 +85,7 @@ impl FromValue for Mesh {
         })
     }
 }
+
 impl Mesh {
     pub fn read_indecies(&self) -> Vec<[u32; 3]> {
         if self.index_format == 0 {
diff --git a/src/classes/pptr.rs b/src/classes/pptr.rs
index d6e4bb6..30f37ad 100644
--- a/src/classes/pptr.rs
+++ b/src/classes/pptr.rs
@@ -1,11 +1,17 @@
-use super::HValue;
-use crate::object::{Value, parser::FromValue};
+use crate::{
+    object::{Value, parser::FromValue},
+    serialized_file::SerializedFile,
+};
 use anyhow::{Result, anyhow, bail};
+use log::debug;
 use serde::Serialize;
-use std::marker::PhantomData;
+use std::{
+    io::{Read, Seek},
+    marker::PhantomData,
+};
 
 #[derive(Debug, Serialize)]
-pub struct PPtr<T = HValue> {
+pub struct PPtr<T = Value> {
     #[serde(skip, default)]
     _class: PhantomData<T>,
     pub class: String,
@@ -36,8 +42,8 @@ impl<T> FromValue for PPtr<T> {
     }
 }
 
-impl PPtr {
-    pub fn cast<T>(self) -> PPtr<T> {
+impl<T: FromValue> PPtr<T> {
+    pub fn cast<U>(self) -> PPtr<U> {
         PPtr {
             _class: PhantomData,
             class: self.class,
@@ -45,4 +51,20 @@ impl PPtr {
             path_id: self.path_id,
         }
     }
+    pub fn is_null(&self) -> bool {
+        self.path_id == 0 && self.file_id == 0
+    }
+    pub fn load(&self, file: &mut SerializedFile<impl Read + Seek>) -> Result<T> {
+        debug!(
+            "loading PPtr<{}> file_id={} path_id={}",
+            self.class, self.file_id, self.path_id
+        );
+        let ob = file
+            .objects
+            .iter()
+            .find(|o| o.path_id == self.path_id)
+            .ok_or(anyhow!("object with path id {} not found", self.path_id))?
+            .to_owned();
+        file.read_object(ob)?.parse()
+    }
 }
diff --git a/src/classes/transform.rs b/src/classes/transform.rs
index 2ec515a..880eb9a 100644
--- a/src/classes/transform.rs
+++ b/src/classes/transform.rs
@@ -29,7 +29,7 @@ impl FromValue for Transform {
                 .as_vector()
                 .unwrap()
                 .into_iter()
-                .map(|e| PPtr::from_value(e).unwrap().cast())
+                .map(|e| PPtr::from_value(e).unwrap())
                 .collect(),
         })
     }
diff --git a/src/object/parser.rs b/src/object/parser.rs
index 2e5d027..b9cc1d4 100644
--- a/src/object/parser.rs
+++ b/src/object/parser.rs
@@ -44,6 +44,11 @@ impl Fields {
     }
 }
 
+impl FromValue for Value {
+    fn from_value(v: Value) -> Result<Self> {
+        Ok(v)
+    }
+}
 impl FromValue for u8 {
     fn from_value(v: Value) -> anyhow::Result<Self> {
         v.as_u8().ok_or(anyhow!("expected u8"))