ast_description_lang/collections/
mod.rs

1use indexmap::IndexSet;
2use std::fmt::Display;
3use std::hash::Hash;
4use std::marker::PhantomData;
5
6mod impls;
7
8#[derive(Debug, Clone)]
9pub struct NamedSet<'ni, T: NamedItem<'ni>>(IndexSet<Wrapper<'ni, T>>);
10
11#[derive(Debug, Clone)]
12struct Wrapper<'ni, T: NamedItem<'ni>>(T, PhantomData<&'ni ()>);
13
14pub trait NamedItem<'ni> {
15  type Name: Hash + Ord + Copy + Display;
16  type Unnamed: Unnamed<'ni, Named = Self>;
17
18  fn name(&self) -> Self::Name;
19  fn dummy(name: Self::Name) -> Self;
20}
21
22pub trait Unnamed<'ni> {
23  type Named: NamedItem<'ni>;
24
25  fn add_name(self, name: <Self::Named as NamedItem<'ni>>::Name) -> Self::Named;
26}
27
28impl<'ni, T: NamedItem<'ni>> NamedSet<'ni, T> {
29  pub fn insert(&mut self, item: T) -> bool {
30    self.0.insert(Wrapper::new(item))
31  }
32
33  pub fn get(&self, name: T::Name) -> Option<&T> {
34    self
35      .0
36      .get(&Wrapper::new(T::dummy(name)))
37      .map(|wrapper| &wrapper.0)
38  }
39
40  pub fn get_index(&self, index: usize) -> Option<&T> {
41    self.0.get_index(index).map(|wrapper| &wrapper.0)
42  }
43
44  pub fn index_of(&self, name: T::Name) -> Option<usize> {
45    self.0.get_index_of(&Wrapper::new(T::dummy(name)))
46  }
47
48  pub fn contains(&self, name: T::Name) -> bool {
49    self.0.contains(&Wrapper(T::dummy(name), PhantomData))
50  }
51
52  pub fn iter(&'ni self) -> impl Iterator<Item = &'ni T> + 'ni {
53    self.0.iter().map(|wrapper| &wrapper.0)
54  }
55}
56
57impl<'ni, T: NamedItem<'ni>> Wrapper<'ni, T> {
58  fn new(item: T) -> Self {
59    Self(item, PhantomData)
60  }
61}
ast_description_lang/collections/mod.rs

ast_description_lang/collections/
mod.rs
