๐Ÿ“ฆ bevyengine / bevy

๐Ÿ“„ external_source_external_thread.rs ยท 75 lines
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//! How to use an external thread to run an infinite task and communicate with a channel.

use bevy::prelude::*;
// Using crossbeam_channel instead of std as std `Receiver` is `!Sync`
use crossbeam_channel::{bounded, Receiver};
use rand::{Rng, SeedableRng};
use rand_chacha::ChaCha8Rng;

fn main() {
    App::new()
        .add_message::<StreamMessage>()
        .add_plugins(DefaultPlugins)
        .add_systems(Startup, setup)
        .add_systems(Update, (spawn_text, move_text))
        .add_systems(FixedUpdate, read_stream)
        .insert_resource(Time::<Fixed>::from_seconds(0.5))
        .run();
}

#[derive(Resource, Deref)]
struct StreamReceiver(Receiver<u32>);

#[derive(Message)]
struct StreamMessage(u32);

fn setup(mut commands: Commands) {
    commands.spawn(Camera2d);

    let (tx, rx) = bounded::<u32>(1);
    std::thread::spawn(move || {
        // We're seeding the PRNG here to make this example deterministic for testing purposes.
        // This isn't strictly required in practical use unless you need your app to be deterministic.
        let mut rng = ChaCha8Rng::seed_from_u64(19878367467713);
        loop {
            // Everything here happens in another thread
            // This is where you could connect to an external data source

            // This will block until the previous value has been read in system `read_stream`
            tx.send(rng.random_range(0..2000)).unwrap();
        }
    });

    commands.insert_resource(StreamReceiver(rx));
}

// This system reads from the receiver and sends messages in the ECS
fn read_stream(receiver: Res<StreamReceiver>, mut events: MessageWriter<StreamMessage>) {
    for from_stream in receiver.try_iter() {
        events.write(StreamMessage(from_stream));
    }
}

fn spawn_text(mut commands: Commands, mut reader: MessageReader<StreamMessage>) {
    for (per_frame, message) in reader.read().enumerate() {
        commands.spawn((
            Text2d::new(message.0.to_string()),
            TextLayout::new_with_justify(Justify::Center),
            Transform::from_xyz(per_frame as f32 * 100.0, 300.0, 0.0),
        ));
    }
}

fn move_text(
    mut commands: Commands,
    mut texts: Query<(Entity, &mut Transform), With<Text2d>>,
    time: Res<Time>,
) {
    for (entity, mut position) in &mut texts {
        position.translation -= Vec3::new(0.0, 100.0 * time.delta_secs(), 0.0);
        if position.translation.y < -300.0 {
            commands.entity(entity).despawn();
        }
    }
}