cosmic/widget/segmented_button/model/
selection.rs

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
108
// Copyright 2022 System76 <info@system76.com>
// SPDX-License-Identifier: MPL-2.0

//! Describes logic specific to the single-select and multi-select modes of a model.

use super::{Entity, Model};
use std::collections::HashSet;

/// Describes a type that has selectable items.
pub trait Selectable {
    /// Activate an item.
    fn activate(&mut self, id: Entity);

    /// Deactivate an item.
    fn deactivate(&mut self, id: Entity);

    /// Checks if the item is active.
    fn is_active(&self, id: Entity) -> bool;
}

/// [`Model<SingleSelect>`] Ensures that only one key may be selected.
#[derive(Debug, Default)]
pub struct SingleSelect {
    pub active: Entity,
}

impl Selectable for Model<SingleSelect> {
    fn activate(&mut self, id: Entity) {
        if !self.items.contains_key(id) {
            return;
        }

        self.selection.active = id;
    }

    fn deactivate(&mut self, id: Entity) {
        if id == self.selection.active {
            self.selection.active = Entity::default();
        }
    }

    fn is_active(&self, id: Entity) -> bool {
        self.selection.active == id
    }
}

impl Model<SingleSelect> {
    /// Get an immutable reference to the data associated with the active item.
    #[must_use]
    pub fn active_data<Data: 'static>(&self) -> Option<&Data> {
        self.data(self.active())
    }

    /// Get a mutable reference to the data associated with the active item.
    #[must_use]
    pub fn active_data_mut<Data: 'static>(&mut self) -> Option<&mut Data> {
        self.data_mut(self.active())
    }

    /// Deactivates the active item.
    pub fn deactivate(&mut self) {
        Selectable::deactivate(self, Entity::default());
    }

    /// The ID of the active item.
    #[must_use]
    pub fn active(&self) -> Entity {
        self.selection.active
    }
}

/// [`Model<MultiSelect>`] permits multiple keys to be active at a time.
#[derive(Debug, Default)]
pub struct MultiSelect {
    pub active: HashSet<Entity>,
}

impl Selectable for Model<MultiSelect> {
    fn activate(&mut self, id: Entity) {
        if !self.items.contains_key(id) {
            return;
        }

        if !self.selection.active.insert(id) {
            self.selection.active.remove(&id);
        }
    }

    fn deactivate(&mut self, id: Entity) {
        self.selection.active.remove(&id);
    }

    fn is_active(&self, id: Entity) -> bool {
        self.selection.active.contains(&id)
    }
}

impl Model<MultiSelect> {
    /// Deactivates the item in the model.
    pub fn deactivate(&mut self, id: Entity) {
        Selectable::deactivate(self, id);
    }

    /// The IDs of the active items.
    pub fn active(&self) -> impl Iterator<Item = Entity> + '_ {
        self.selection.active.iter().copied()
    }
}