beet_spatial 0.0.8

Spatial actions built upon beet_flow
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46

use crate::prelude::*;
use beet_core::prelude::*;
use std::f32::consts::PI;

/// Rotate an entity to face its [`Velocity`] in 2D space
#[derive(Default, Component, Reflect)]
#[reflect(Default, Component)]
pub struct RotateToVelocity2d;


pub(crate) fn rotate_to_velocity_2d(
	mut query: Query<(&mut Transform, &Velocity), With<RotateToVelocity2d>>,
) {
	for (mut transform, velocity) in query.iter_mut() {
		let Some(dir) = velocity.0.try_normalize() else {
			continue;
		};
		transform.rotation =
			Quat::from_rotation_z(f32::atan2(dir.y, dir.x) - PI * 0.5);
	}
}


/// Rotate an entity to face its [`Velocity`] in 3D space
/// If the velocity is zero, this does nothing
#[derive(Component, Deref, DerefMut, Reflect)]
#[reflect(Default, Component)]
pub struct RotateToVelocity3d(pub f32);

impl Default for RotateToVelocity3d {
	fn default() -> Self { Self(5.) }
}

pub(crate) fn rotate_to_velocity_3d(
	time: When<Res<Time>>,
	mut query: Query<(&mut Transform, &Velocity, &RotateToVelocity3d)>,
) {
	for (mut transform, velocity, rotate) in query.iter_mut() {
		let Some(dir) = velocity.0.try_normalize() else {
			continue;
		};
		transform.rotation = transform
			.rotation
			.slerp(Quat::look_at(dir), **rotate * time.delta_secs());
	}
}