Index + value iteration for arrays 🦚

This commit is contained in:
Laurenz 2021-02-24 18:54:06 +01:00
parent e7cc35073f
commit dae3dad540
4 changed files with 18 additions and 5 deletions

View File

@ -460,6 +460,9 @@ impl Eval for ExprFor {
(ForPattern::Value(v), Value::Array(array)) => {
iter!(for (v => value) in array.into_iter());
}
(ForPattern::KeyValue(i, v), Value::Array(array)) => {
iter!(for (i => idx, v => value) in array.into_iter().enumerate());
}
(ForPattern::Value(v), Value::Dict(dict)) => {
iter!(for (v => value) in dict.into_iter().map(|p| p.1));
}
@ -467,8 +470,7 @@ impl Eval for ExprFor {
iter!(for (k => key, v => value) in dict.into_iter());
}
(ForPattern::KeyValue(_, _), Value::Str(_))
| (ForPattern::KeyValue(_, _), Value::Array(_)) => {
(ForPattern::KeyValue(_, _), Value::Str(_)) => {
ctx.diag(error!(self.pattern.span(), "mismatched pattern"));
}

View File

@ -590,6 +590,12 @@ primitive! { ValueTemplate: "template", Value::Template }
primitive! { ValueFunc: "function", Value::Func }
primitive! { ValueArgs: "arguments", Value::Args }
impl From<usize> for Value {
fn from(v: usize) -> Self {
Self::Int(v as i64)
}
}
impl From<&str> for Value {
fn from(v: &str) -> Self {
Self::Str(v.to_string())

View File

@ -735,7 +735,7 @@ mod tests {
// Simple values.
test_value(Value::None, "none");
test_value(false, "false");
test_value(12, "12");
test_value(12i64, "12");
test_value(3.14, "3.14");
test_value(Length::pt(5.5), "5.5pt");
test_value(Angle::deg(90.0), "90.0deg");

View File

@ -9,6 +9,11 @@
out += (v,)
}
// Indices and values of array.
#for i, v in ("1", "2", "3") {
test(repr(i + 1), v)
}
// Values of dictionary.
#for v in (a: 4, b: 5) {
out += (v,)
@ -23,8 +28,8 @@
#test(out, (1, 2, 3, 4, 5, "a", 6, "b", 7))
---
// Keys and values of array.
// Keys and values of strings.
// Error: 6-10 mismatched pattern
#for k, v in (-1, -2, -3) {
#for k, v in "hi" {
dont-care
}