beet_net 0.0.3

A very flexible AI behavior library for games and robotics.
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
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116

use crate::prelude::*;
use bevy::prelude::*;
use forky_core::ResultTEExt;


pub fn handle_incoming_commands(
	mut commands: Commands,
	mut registrations: ResMut<ReplicateRegistry>,
	incoming: Res<MessageIncoming>,
) {
	for msg in incoming.iter() {
		match msg {
			Message::Spawn { entity } => {
				let local = commands.spawn_empty().id();
				registrations.entities.insert(*entity, local);
			}
			Message::Despawn { entity } => {
				commands.entity(*entity).despawn();
			}
			Message::Add {
				entity,
				reg_id,
				payload,
			} => {
				if let Some((entity, fns)) =
					registrations.entity_fns(*entity, *reg_id)
				{
					(fns.insert)(&mut commands.entity(entity), &payload)
						.ok_or(|e| log::error!("{e}"));
				}
			}
			Message::Change {
				entity,
				reg_id,
				payload,
			} => {
				if let Some((entity, fns)) =
					registrations.entity_fns(*entity, *reg_id)
				{
					(fns.change)(&mut commands.entity(entity), payload)
						.ok_or(|e| log::error!("{e}"));
				}
			}
			Message::Remove { entity, reg_id } => {
				if let Some((entity, fns)) =
					registrations.entity_fns(*entity, *reg_id)
				{
					(fns.remove)(&mut commands.entity(entity));
				}
			}
			Message::InsertResource { reg_id, payload } => {
				if let Some(fns) =
					registrations.incoming_resource_fns.get(reg_id)
				{
					(fns.insert)(&mut commands, payload)
						.ok_or(|e| log::error!("{e}"));
				}
			}
			Message::ChangeResource { reg_id, payload } => {
				if let Some(fns) =
					registrations.incoming_resource_fns.get(reg_id)
				{
					(fns.change)(&mut commands, payload)
						.ok_or(|e| log::error!("{e}"));
				}
			}
			Message::RemoveResource { reg_id } => {
				if let Some(fns) =
					registrations.incoming_resource_fns.get(reg_id)
				{
					(fns.remove)(&mut commands);
				}
			}
			Message::SendObserver { reg_id, payload } => {
				if let Some(fns) =
					registrations.incoming_observer_fns.get(reg_id)
				{
					(fns.send)(&mut commands, payload)
						.ok_or(|e| log::error!("{e}"));
				}
			}
			Message::SendEvent {
				reg_id: _,
				payload: _,
			} => {
				// events require world access
			}
		}
	}
}

pub fn handle_incoming_world(world: &mut World) {
	let registrations = world.resource::<ReplicateRegistry>();
	let events = world
		.resource::<MessageIncoming>()
		.iter()
		.filter_map(|msg| match msg {
			Message::SendEvent { reg_id, payload } => {
				if let Some(fns) = registrations.incoming_event_fns.get(reg_id)
				{
					Some((fns.clone(), payload.clone()))
				} else {
					None
				}
			}
			_ => {
				// all other messages are handled by `handle_incoming_commands`
				None
			}
		})
		.collect::<Vec<_>>();

	for (fns, payload) in events {
		(fns.send)(world, &payload).ok_or(|e| log::error!("{e}"));
	}
}