summaryrefslogtreecommitdiff
path: root/ecs/examples/multiple_queries.rs
blob: 9a699c2bbb1ee69a397b7e30387fc368b9f68995 (plain)
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
use std::fmt::Display;

use ecs::{Component, Query, World};

#[derive(Component)]
struct Health
{
    health: u32,
}

#[derive(Component)]
enum AttackStrength
{
    Strong,
    Weak,
}

#[derive(Component)]
struct EnemyName
{
    name: String,
}

impl Display for EnemyName
{
    fn fmt(&self, formatter: &mut std::fmt::Formatter<'_>) -> std::fmt::Result
    {
        self.name.fmt(formatter)
    }
}

fn do_attacks(
    attacker_query: Query<(AttackStrength,)>,
    mut enemy_query: Query<(Health, EnemyName)>,
)
{
    for (attack_strength,) in &attacker_query {
        for (health, enemy_name) in enemy_query.iter_mut() {
            let damage = match attack_strength {
                AttackStrength::Strong => 20,
                AttackStrength::Weak => 10,
            };

            if health.health <= damage {
                println!("Enemy '{enemy_name}' died");

                health.health = 0;

                continue;
            }

            health.health -= damage;

            println!("Enemy '{enemy_name}' took {damage} damage");
        }
    }
}

#[derive(Debug, PartialEq, Eq, Hash)]
enum Event
{
    Start,
}

fn main()
{
    let mut world = World::<Event>::new();

    world.register_system(Event::Start, do_attacks);

    world.create_entity((
        Health { health: 100 },
        EnemyName { name: "Big spider".to_string() },
    ));

    world.create_entity((
        Health { health: 30 },
        EnemyName { name: "Small goblin".to_string() },
    ));

    world.create_entity((
        Health { health: 30 },
        EnemyName { name: "Headcrab".to_string() },
    ));

    world.create_entity((AttackStrength::Strong,));
    world.create_entity((AttackStrength::Weak,));

    world.emit(&Event::Start);
}