bevy/examples/scene/properties.rs

106 lines
3.5 KiB
Rust

use bevy::{
prelude::*,
property::{ron::deserialize_dynamic_properties, PropertyTypeRegistry},
type_registry::TypeRegistry,
};
use serde::{Deserialize, Serialize};
fn main() {
App::build()
.add_default_plugins()
// If you need to deserialize custom property types, register them like this:
.register_property_type::<Test>()
.register_property_type::<Nested>()
.register_property_type::<CustomProperty>()
.add_startup_system(setup.system())
.run();
}
#[derive(Properties, Default)]
pub struct Test {
a: usize,
custom: CustomProperty,
nested: Nested,
}
#[derive(Properties, Default)]
pub struct Nested {
b: usize,
}
#[derive(Serialize, Deserialize, Default, Clone, Property)]
pub struct CustomProperty {
a: usize,
}
fn setup(type_registry: Res<TypeRegistry>) {
let mut test = Test {
a: 1,
custom: CustomProperty { a: 10 },
nested: Nested { b: 8 },
};
// You can set a property value like this. The type must match exactly or this will fail.
test.set_prop_val::<usize>("a", 2);
assert_eq!(test.a, 2);
// You can also set properties dynamically. set_prop accepts any type that implements Property
let x: u32 = 3;
test.set_prop("a", &x);
assert_eq!(test.a, 3);
// DynamicProperties also implements the Properties trait.
let mut patch = DynamicProperties::map();
patch.set::<usize>("a", 4);
// You can "apply" Properties on top of other Properties. This will only set properties with the same name and type.
// You can use this to "patch" your components with new values.
test.apply(&patch);
assert_eq!(test.a, 4);
// All properties can be serialized.
// If you #[derive(Properties)] your type doesn't even need to directly implement the Serde trait!
let registry = type_registry.property.read().unwrap();
let ron_string = serialize_ron(&test, &registry).unwrap();
println!("{}\n", ron_string);
// Dynamic properties can be deserialized
let dynamic_properties = deserialize_dynamic_properties(&ron_string, &registry).unwrap();
let round_tripped = serialize_ron(&dynamic_properties, &registry).unwrap();
println!("{}", round_tripped);
assert_eq!(ron_string, round_tripped);
// This means you can patch Properties with dynamic properties deserialized from a string
test.apply(&dynamic_properties);
// Properties can also be sequences.
// Sequences from std::collections (Vec, VecDeque) already implement the Properties trait
let mut seq = vec![1u32, 2u32];
let mut patch = DynamicProperties::seq();
patch.push(Box::new(3u32), None);
seq.apply(&patch);
assert_eq!(seq[0], 3);
let ron_string = serialize_ron(&patch, &registry).unwrap();
println!("{}\n", ron_string);
let dynamic_properties = deserialize_dynamic_properties(&ron_string, &registry).unwrap();
let round_tripped = serialize_ron(&dynamic_properties, &registry).unwrap();
assert_eq!(ron_string, round_tripped);
}
fn serialize_ron<T>(property: &T, registry: &PropertyTypeRegistry) -> Result<String, ron::Error>
where
T: Property,
{
let pretty_config = ron::ser::PrettyConfig::default().with_decimal_floats(true);
let mut buf = Vec::new();
let mut serializer = ron::ser::Serializer::new(&mut buf, Some(pretty_config), false)?;
property
.serializable(registry)
.borrow()
.serialize(&mut serializer)?;
let ron_string = String::from_utf8(buf).unwrap();
Ok(ron_string)
}