freya_core/
layers.rs

1use std::ops::{
2    Deref,
3    DerefMut,
4};
5
6use rustc_hash::{
7    FxHashMap,
8    FxHashSet,
9};
10
11use crate::node_id::NodeId;
12
13#[derive(Default, Clone, Debug, PartialEq)]
14pub struct Layers(FxHashMap<i16, FxHashSet<NodeId>>);
15
16impl Layers {
17    /// Insert the given [NodeId] in the given layer. Will create an entry for the layer if missing.
18    pub fn insert_node_in_layer(&mut self, node_id: NodeId, layer_n: i16) {
19        let layer = self.0.entry(layer_n).or_default();
20        layer.insert(node_id);
21    }
22
23    /// Remove the [NodeId] from the given layer. Will remove the entry of the layer if it becomes empty.
24    pub fn remove_node_from_layer(&mut self, node_id: &NodeId, layer_n: i16) {
25        let layer = self.0.get_mut(&layer_n);
26        if let Some(layer) = layer {
27            layer.remove(node_id);
28
29            if layer.is_empty() {
30                self.0.remove(&layer_n);
31            }
32        }
33    }
34}
35
36impl Deref for Layers {
37    type Target = FxHashMap<i16, FxHashSet<NodeId>>;
38
39    fn deref(&self) -> &Self::Target {
40        &self.0
41    }
42}
43
44impl DerefMut for Layers {
45    fn deref_mut(&mut self) -> &mut Self::Target {
46        &mut self.0
47    }
48}
freya_core/layers.rs

freya_core/
layers.rs
