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

use crate::PickingPluginState;
use bevy::prelude::*;

/// Tracks the current selection state to be used with change tracking in the events system. Meshes
/// with [Selection] will have selection state managed.
///
/// # Requirements
///
/// An entity with the [Selection] component must also have an [Interaction] component.
#[derive(Debug, Copy, Clone)]
pub struct Selection {
    selected: bool,
}
impl Selection {
    pub fn selected(&self) -> bool {
        self.selected
    }
    /// Set the selection state.
    pub fn set_selected(&mut self, selected: bool) {
        self.selected = selected;
    }
}
impl Default for Selection {
    fn default() -> Self {
        Selection { selected: false }
    }
}

/// Marker struct used to mark pickable entities you don't want to trigger a deselection event when pickedy. This is useful for gizmos or other pickable UI entities.
#[derive(Debug, Copy, Clone)]
pub struct NoDeselect;

#[allow(clippy::too_many_arguments)]
pub fn mesh_selection(
    state: Res<PickingPluginState>,
    mouse_button_input: Res<Input<MouseButton>>,
    touches_input: Res<Touches>,
    keyboard_input: Res<Input<KeyCode>>,
    query_changed: Query<&Interaction, (Changed<Interaction>, Without<NoDeselect>)>,
    mut query_all: Query<(&mut Selection, &Interaction)>,
    node_query: Query<&Interaction, With<Node>>,
    no_deselect_query: Query<&Interaction, With<NoDeselect>>,
) {
    if state.paused_for_ui || !state.enabled {
        return;
    }

    // Check if something has been clicked on
    let mut new_selection = false;
    for interaction in query_changed.iter() {
        if *interaction == Interaction::Clicked {
            new_selection = true;
        }
    }

    if keyboard_input.pressed(KeyCode::LControl) && keyboard_input.pressed(KeyCode::A) {
        // The user has hit ctrl+a, select all the things!
        for (mut selection, _interaction) in &mut query_all.iter_mut() {
            if !selection.selected {
                selection.selected = true;
            }
        }
    } else if new_selection {
        // Some pickable mesh has been clicked on - figure out what to select or deselect
        for (mut selection, interaction) in &mut query_all.iter_mut() {
            if selection.selected
                && *interaction != Interaction::Clicked
                && !keyboard_input.pressed(KeyCode::LControl)
            {
                // In this case, the entity is currently marked as selected, but it was not clicked
                // on (interaction), and lctrl was not being held, so it should be deselected.
                selection.selected = false;
            } else if *interaction == Interaction::Clicked
                && keyboard_input.pressed(KeyCode::LControl)
            {
                selection.selected = !selection.selected
            } else if !selection.selected && *interaction == Interaction::Clicked {
                selection.selected = true;
            }
        }
    } else {
        // This branch deselects everything if the user clicks, in empty space. Deselection is not
        // run if the UI or an item tagged with `NoDeselect` was clicked on.
        let mut ui_not_clicked = true;
        for interaction in node_query.iter() {
            // Check if anything in the UI is being interacted with
            if *interaction == Interaction::Clicked && !keyboard_input.pressed(KeyCode::LControl) {
                ui_not_clicked = false;
            }
        }
        let mut no_deselect_not_clicked = true;
        for interaction in no_deselect_query.iter() {
            if *interaction == Interaction::Clicked && !keyboard_input.pressed(KeyCode::LControl) {
                no_deselect_not_clicked = false;
            }
        }
        let mouse_clicked =
            mouse_button_input.just_pressed(MouseButton::Left) || touches_input.just_released(0);
        if mouse_clicked && ui_not_clicked && no_deselect_not_clicked {
            for (mut selection, _interaction) in &mut query_all.iter_mut() {
                if selection.selected {
                    selection.selected = false;
                }
            }
        }
    }
}