use std::ops::{
Deref,
DerefMut,
};
use rustc_hash::{
FxHashMap,
FxHashSet,
};
use crate::node_id::NodeId;
#[derive(Clone, Copy, Debug, PartialEq)]
pub enum Layer {
Relative(i16),
Overlay,
}
impl Default for Layer {
fn default() -> Self {
Layer::Relative(0)
}
}
impl From<i16> for Layer {
fn from(value: i16) -> Self {
Layer::Relative(value)
}
}
#[derive(Default, Clone, Debug, PartialEq)]
pub struct Layers(FxHashMap<i16, FxHashSet<NodeId>>);
impl Layers {
pub fn insert_node_in_layer(&mut self, node_id: NodeId, layer_n: i16) {
let layer = self.0.entry(layer_n).or_default();
layer.insert(node_id);
}
pub fn remove_node_from_layer(&mut self, node_id: &NodeId, layer_n: i16) {
let layer = self.0.get_mut(&layer_n);
if let Some(layer) = layer {
layer.remove(node_id);
if layer.is_empty() {
self.0.remove(&layer_n);
}
}
}
}
impl Deref for Layers {
type Target = FxHashMap<i16, FxHashSet<NodeId>>;
fn deref(&self) -> &Self::Target {
&self.0
}
}
impl DerefMut for Layers {
fn deref_mut(&mut self) -> &mut Self::Target {
&mut self.0
}
}