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

use ecs::phase::START as START_PHASE;
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,)>,
    enemy_query: Query<(Health, EnemyName)>,
)
{
    for (attack_strength,) in &attacker_query {
        for (mut health, enemy_name) in &enemy_query {
            let damage = match *attack_strength {
                AttackStrength::Strong => 20,
                AttackStrength::Weak => 10,
            };

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

                health.health = 0;

                continue;
            }

            health.health -= damage;

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

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

    world.register_system(*START_PHASE, 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.step();
}