1use crate::*;
2
3#[derive(Clone, Debug, PartialEq, Eq, Hash)]
6pub enum Kind {
7 Any,
8 None,
9 Atom(u64),
10 Empty,
11 Enum(u64),
12 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}