1 files changed, 7 insertions, 2 deletions
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"
     )