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
|
use crate::packets::{Object, Packet, Resource};
use glam::Vec3;
use std::collections::HashMap;
pub struct SceneTree {
pub objects: HashMap<Object, ObjectData>,
}
pub struct ObjectData {
pub pos: Vec3,
pub rot: Vec3,
pub parent: Object,
pub pose: Vec<f32>,
pub res: Resource,
}
impl Default for SceneTree {
fn default() -> Self {
Self {
objects: Default::default(),
}
}
}
impl SceneTree {
pub fn update(&mut self, p: &Packet) {
match p {
Packet::Add(object, res) => {
self.objects.insert(*object, ObjectData {
parent: Object(0),
pos: Vec3::ZERO,
rot: Vec3::ZERO,
pose: Vec::new(),
res: *res,
});
}
Packet::Remove(object) => {
self.objects.remove(&object);
}
Packet::Position(object, pos, rot) => {
if let Some(o) = self.objects.get_mut(&object) {
o.pos = *pos;
o.rot = *rot;
}
}
Packet::Pose(object, pose) => {
if let Some(o) = self.objects.get_mut(&object) {
o.pose = pose.to_vec();
}
}
Packet::Parent(parent, child) => {
if let Some(o) = self.objects.get_mut(&parent) {
o.parent = *child
}
}
_ => (),
}
}
pub fn prime_client(&self) -> impl Iterator<Item = Packet> {
self.objects
.iter()
.map(|(object, data)| {
[
Packet::Add(*object, data.res),
Packet::Parent(*object, data.parent),
Packet::Position(*object, data.pos, data.rot),
Packet::Pose(*object, data.pose.clone()),
]
})
.flatten()
}
}
|
