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

use super::*;
use bevy::prelude::*;
use std::collections::HashSet;

/// Meshes with `SelectableMesh` will have selection state managed
#[derive(Debug)]
pub struct SelectablePickMesh {
    selected: HashSet<Group>,
}

impl SelectablePickMesh {
    pub fn new() -> Self {
        SelectablePickMesh::default()
    }
    pub fn selected(&self, group: &Group) -> bool {
        self.selected.get(group).is_some()
    }
}

impl Default for SelectablePickMesh {
    fn default() -> Self {
        SelectablePickMesh {
            selected: HashSet::new(),
        }
    }
}

/// Update all entities with the groups they are selected in.
pub fn select_mesh(
    // Resources
    mouse_button_inputs: Res<Input<MouseButton>>,
    // Queries
    mut query: Query<(&mut SelectablePickMesh, &InteractableMesh)>,
) {
    if mouse_button_inputs.just_pressed(MouseButton::Left) {
        // Update Selections
        for (mut selectable, interactable) in &mut query.iter_mut() {
            selectable.selected = interactable.groups_just_pressed(MouseButton::Left);
        }
    }
}