beet_examples 0.0.8

Bits and pieces for substantial beet examples
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
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101

use beet_core::prelude::*;

#[derive(Default, Component, Reflect)]
#[reflect(Default, Component)]
pub struct FollowCursor2d;

pub fn follow_cursor_2d(
	camera_query: Single<(&Camera, &GlobalTransform)>,
	windows: Single<&Window>,
	mut cursor_query: Populated<&mut Transform, With<FollowCursor2d>>,
) {
	let (camera, camera_transform) = *camera_query;

	let Some(cursor_position) = windows.cursor_position() else {
		return;
	};

	let Ok(point) =
		camera.viewport_to_world_2d(camera_transform, cursor_position)
	else {
		return;
	};

	for mut transform in cursor_query.iter_mut() {
		transform.translation = point.extend(0.);
	}
}


#[derive(Component, Reflect)]
#[reflect(Default, Component)]
pub struct FollowCursor3d {
	pub intersect_point: Vec3,
	pub intersect_plane: InfinitePlane3d,
}

impl FollowCursor3d {
	/// Follows the cursor on the XZ plane with the Y axis as the normal.
	pub const ORIGIN_Y: Self = Self {
		intersect_point: Vec3::ZERO,
		intersect_plane: InfinitePlane3d { normal: Dir3::Y },
	};

	/// Follows the cursor on the XY plane with the Z axis as the normal.
	pub const ORIGIN_Z: Self = Self {
		intersect_point: Vec3::ZERO,
		intersect_plane: InfinitePlane3d { normal: Dir3::Z },
	};


	pub fn new(intersect_point: Vec3, intersect_plane: Vec3) -> Self {
		Self {
			intersect_point,
			intersect_plane: InfinitePlane3d::new(intersect_plane),
		}
	}
	pub fn with_intersect_point(self, intersect_point: Vec3) -> Self {
		Self {
			intersect_point,
			..self
		}
	}
	pub fn with_intersect_plane(self, intersect_plane: Vec3) -> Self {
		Self {
			intersect_plane: InfinitePlane3d::new(intersect_plane),
			..self
		}
	}
}

impl Default for FollowCursor3d {
	fn default() -> Self { Self::ORIGIN_Y }
}

pub fn follow_cursor_3d(
	camera_query: Single<(&Camera, &GlobalTransform)>,
	windows: Single<&Window>,
	mut cursor_query: Populated<(&mut Transform, &FollowCursor3d)>,
) {
	let (camera, camera_transform) = *camera_query;

	let Some(cursor_position) = windows.cursor_position() else {
		return;
	};

	let Ok(ray) = camera.viewport_to_world(camera_transform, cursor_position)
	else {
		return;
	};

	for (mut transform, follow_cursor) in cursor_query.iter_mut() {
		let Some(dist) = ray.intersect_plane(
			follow_cursor.intersect_point,
			follow_cursor.intersect_plane,
		) else {
			continue;
		};
		let point = ray.get_point(dist);
		transform.translation = point;
	}
}