37 lines
1.3 KiB
Rust
37 lines
1.3 KiB
Rust
pub mod components;
|
|
pub mod hierarchy;
|
|
pub mod transform_propagate_system;
|
|
|
|
pub mod prelude {
|
|
pub use crate::{components::*, hierarchy::*, TransformPlugin};
|
|
}
|
|
|
|
use bevy_app::{prelude::*, startup_stage};
|
|
use bevy_ecs::IntoSystem;
|
|
use bevy_reflect::RegisterTypeBuilder;
|
|
use prelude::{parent_update_system, Children, GlobalTransform, Parent, PreviousParent, Transform};
|
|
|
|
#[derive(Default)]
|
|
pub struct TransformPlugin;
|
|
|
|
impl Plugin for TransformPlugin {
|
|
fn build(&self, app: &mut AppBuilder) {
|
|
app.register_type::<Children>()
|
|
.register_type::<Parent>()
|
|
.register_type::<PreviousParent>()
|
|
.register_type::<Transform>()
|
|
.register_type::<GlobalTransform>()
|
|
// add transform systems to startup so the first update is "correct"
|
|
.add_startup_system_to_stage(startup_stage::POST_STARTUP, parent_update_system.system())
|
|
.add_startup_system_to_stage(
|
|
startup_stage::POST_STARTUP,
|
|
transform_propagate_system::transform_propagate_system.system(),
|
|
)
|
|
.add_system_to_stage(stage::POST_UPDATE, parent_update_system.system())
|
|
.add_system_to_stage(
|
|
stage::POST_UPDATE,
|
|
transform_propagate_system::transform_propagate_system.system(),
|
|
);
|
|
}
|
|
}
|