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

use fxhash::FxHashSet;
use std::any::TypeId;

/// TypeID based map for storing different types
#[derive(Default)]
pub struct TypeSet {
    map: FxHashSet<TypeId>,
}

impl TypeSet {
    /// Create new empty map (wrapper to default)
    #[inline]
    pub fn new() -> Self {
        Self::default()
    }

    /// Insert a value to map
    pub fn insert<T: 'static>(&mut self) {
        self.map.insert(TypeId::of::<T>());
    }

    /// Check if map contains specific type
    #[inline]
    pub fn contains<T: 'static>(&self) -> bool {
        self.map.contains(&TypeId::of::<T>())
    }

    /// Remove entry from map
    pub fn remove<T: 'static>(&mut self) -> bool {
        self.map
            .remove(&TypeId::of::<T>())
    }

    /// Clear whole map
    #[inline]
    pub fn clear(&mut self) {
        self.map.clear()
    }
}