expose extras from gltf nodes

bevyengine · Jul 3, 2021 · 5d72d2a · 5d72d2a
1 parent b52edc1
commit 5d72d2a
Show file tree

Hide file tree

Showing 3 changed files with 16 additions and 2 deletions.
diff --git a/crates/bevy_gltf/Cargo.toml b/crates/bevy_gltf/Cargo.toml
@@ -27,7 +27,7 @@ bevy_scene = { path = "../bevy_scene", version = "0.5.0" }
 bevy_log = { path = "../bevy_log", version = "0.5.0" }
 
 # other
-gltf = { version = "0.16.0", default-features = false, features = ["utils", "names", "KHR_materials_unlit"] }
+gltf = { version = "0.16.0", default-features = false, features = ["utils", "names", "KHR_materials_unlit", "extras"] }
 thiserror = "1.0"
 anyhow = "1.0"
 base64 = "0.13.0"

diff --git a/crates/bevy_gltf/src/lib.rs b/crates/bevy_gltf/src/lib.rs
@@ -5,8 +5,9 @@ pub use loader::*;
 
 use bevy_app::prelude::*;
 use bevy_asset::{AddAsset, Handle};
+use bevy_ecs::reflect::ReflectComponent;
 use bevy_pbr::prelude::StandardMaterial;
-use bevy_reflect::TypeUuid;
+use bevy_reflect::{Reflect, TypeUuid};
 use bevy_render::mesh::Mesh;
 use bevy_scene::Scene;
 
@@ -17,6 +18,7 @@ pub struct GltfPlugin;
 impl Plugin for GltfPlugin {
     fn build(&self, app: &mut AppBuilder) {
         app.init_asset_loader::<GltfLoader>()
+            .register_type::<GltfExtras>()
             .add_asset::<Gltf>()
             .add_asset::<GltfNode>()
             .add_asset::<GltfPrimitive>()
@@ -58,3 +60,9 @@ pub struct GltfPrimitive {
     pub mesh: Handle<Mesh>,
     pub material: Option<Handle<StandardMaterial>>,
 }
+
+#[derive(Debug, Reflect, Default)]
+#[reflect(Component)]
+pub struct GltfExtras {
+    pub value: String,
+}
diff --git a/crates/bevy_gltf/src/loader.rs b/crates/bevy_gltf/src/loader.rs
@@ -459,6 +459,12 @@ fn load_node(
         node.insert(Name::new(name.to_string()));
     }
 
+    if let Some(extras) = gltf_node.extras() {
+        node.insert(super::GltfExtras {
+            value: extras.get().to_string(),
+        });
+    }
+
     // create camera node
     if let Some(camera) = gltf_node.camera() {
         node.insert(VisibleEntities {