use cairo_lang_sierra::ids::ConcreteTypeId;
use std::collections::HashMap;
#[derive(Default)]
pub struct EnumSnapshotVariantsMeta {
map: HashMap<ConcreteTypeId, Vec<ConcreteTypeId>>,
}
impl EnumSnapshotVariantsMeta {
pub fn set_mapping(&mut self, snapshot_id: &ConcreteTypeId, variants: &[ConcreteTypeId]) {
self.map.insert(snapshot_id.clone(), variants.to_vec());
}
pub fn get_variants(&self, snapshot_id: &ConcreteTypeId) -> Option<&Vec<ConcreteTypeId>> {
self.map.get(snapshot_id)
}
}