mech_core/
kind.rs

1use crate::*;
2
3// Kind -----------------------------------------------------------------------
4
5#[derive(Clone, Debug, PartialEq, Eq, Hash)]
6pub enum Kind {
7  Any,
8  None,
9  Atom(u64),
10  Empty,
11  Enum(u64),
12  //Fsm(Vec<Kind>,Vec<Kind>),
13  //Function(Vec<Kind>,Vec<Kind>),
14  Id,
15  Index,
16  Map(Box<Kind>,Box<Kind>),
17  Matrix(Box<Kind>,Vec<usize>),
18  Option(Box<Kind>),
19  Record((Vec<(String,Kind)>)),
20  Reference(Box<Kind>),
21  Scalar(u64),
22  Set(Box<Kind>,Option<usize>),
23  Table(Vec<(String,Kind)>,usize),
24  Tuple(Vec<Kind>),
25}
26
27impl Kind {
28
29  #[cfg(feature = "kind_annotation")]
30  pub fn to_value(&self, kinds: &KindTable) -> MResult<Value> {
31    let value_kind = self.to_value_kind(kinds)?;
32    Ok(Value::Kind(value_kind))
33  }
34
35  #[cfg(feature = "kind_annotation")]
36  pub fn to_value_kind(&self, kinds: &KindTable) -> MResult<ValueKind> {
37    match self {
38      Kind::None => Ok(ValueKind::None),
39      Kind::Any => Ok(ValueKind::Any),
40      Kind::Atom(id) => Ok(ValueKind::Atom(*id)),
41      Kind::Empty => Ok(ValueKind::Empty),
42      Kind::Enum(id) => Ok(ValueKind::Enum(*id)),
43      Kind::Id => Ok(ValueKind::Id),
44      Kind::Index => Ok(ValueKind::Index),
45      Kind::Map(keys, vals) => {
46        let key_knd = keys.to_value_kind(kinds)?;
47        let val_knd = vals.to_value_kind(kinds)?;
48        Ok(ValueKind::Map(Box::new(key_knd), Box::new(val_knd)))
49      },
50      Kind::Matrix(knd, size) => {
51        let val_knd = knd.to_value_kind(kinds)?;
52        Ok(ValueKind::Matrix(Box::new(val_knd), size.clone()))
53      },
54      Kind::Option(knd) => {
55        let val_knd = knd.to_value_kind(kinds)?;
56        Ok(ValueKind::Option(Box::new(val_knd)))
57      },
58      Kind::Record(elements) => {
59        let val_knds: Vec<(String, ValueKind)> = elements.iter()
60          .map(|(id, k)| k.to_value_kind(kinds).map(|kind| (id.clone(), kind)))
61          .collect::<MResult<_>>()?;
62        Ok(ValueKind::Record(val_knds))
63      },
64      Kind::Reference(kind) => {
65        let val_knd = kind.to_value_kind(kinds)?;
66        Ok(ValueKind::Reference(Box::new(val_knd)))
67      },
68      Kind::Scalar(id) => {
69        match kinds.get(id).cloned() {
70          Some(val_knd) => Ok(val_knd),
71          None => Err(
72            MechError2::new(
73              UndefinedKindError { kind_id: *id },
74              None,
75            )
76            .with_compiler_loc()
77          ),
78        }
79      },
80      Kind::Set(kind, size) => {
81        let val_knd = kind.to_value_kind(kinds)?;
82        Ok(ValueKind::Set(Box::new(val_knd), *size))
83      },
84      Kind::Table(elements, size) => {
85        let val_knds: Vec<(String, ValueKind)> = elements.iter()
86          .map(|(id, k)| k.to_value_kind(kinds).map(|kind| (id.clone(), kind)))
87          .collect::<MResult<_>>()?;
88        Ok(ValueKind::Table(val_knds, *size))
89      },
90      Kind::Tuple(elements) => {
91        let val_knds = elements.iter().map(|k| k.to_value_kind(kinds)).collect::<MResult<Vec<ValueKind>>>()?;
92        Ok(ValueKind::Tuple(val_knds))
93      }
94    }
95  }
96}