cgen: fix `for char in nums {}`
parent
b2157388d9
commit
d7e7113047
|
@ -690,7 +690,7 @@ fn (mut g Gen) for_in(it ast.ForInStmt) {
|
||||||
g.expr(it.high)
|
g.expr(it.high)
|
||||||
g.writeln('; $i++) {')
|
g.writeln('; $i++) {')
|
||||||
if it.val_var != '_' {
|
if it.val_var != '_' {
|
||||||
g.writeln('\tint $it.val_var = $i;')
|
g.writeln('\tint ${c_name(it.val_var)} = $i;')
|
||||||
}
|
}
|
||||||
g.stmts(it.stmts)
|
g.stmts(it.stmts)
|
||||||
g.writeln('}')
|
g.writeln('}')
|
||||||
|
@ -708,7 +708,7 @@ fn (mut g Gen) for_in(it ast.ForInStmt) {
|
||||||
op_field := if cond_type_is_ptr { '->' } else { '.' }
|
op_field := if cond_type_is_ptr { '->' } else { '.' }
|
||||||
g.writeln('for (int $i = 0; $i < ${atmp}${op_field}len; $i++) {')
|
g.writeln('for (int $i = 0; $i < ${atmp}${op_field}len; $i++) {')
|
||||||
if it.val_var != '_' {
|
if it.val_var != '_' {
|
||||||
g.writeln('\t$styp $it.val_var = (($styp*)${atmp}${op_field}data)[$i];')
|
g.writeln('\t$styp ${c_name(it.val_var)} = (($styp*)${atmp}${op_field}data)[$i];')
|
||||||
}
|
}
|
||||||
g.stmts(it.stmts)
|
g.stmts(it.stmts)
|
||||||
g.writeln('}')
|
g.writeln('}')
|
||||||
|
@ -727,7 +727,7 @@ fn (mut g Gen) for_in(it ast.ForInStmt) {
|
||||||
g.writeln('for (int $idx = 0; $idx < ${keys_tmp}.len; $idx++) {')
|
g.writeln('for (int $idx = 0; $idx < ${keys_tmp}.len; $idx++) {')
|
||||||
g.writeln('\t$key_styp $key = (($key_styp*)${keys_tmp}.data)[$idx];')
|
g.writeln('\t$key_styp $key = (($key_styp*)${keys_tmp}.data)[$idx];')
|
||||||
if it.val_var != '_' {
|
if it.val_var != '_' {
|
||||||
g.write('\t$val_styp $it.val_var = (*($val_styp*)map_get3(')
|
g.write('\t$val_styp ${c_name(it.val_var)} = (*($val_styp*)map_get3(')
|
||||||
g.expr(it.cond)
|
g.expr(it.cond)
|
||||||
g.writeln(', $key, &($val_styp[]){ $zero }));')
|
g.writeln(', $key, &($val_styp[]){ $zero }));')
|
||||||
}
|
}
|
||||||
|
@ -740,7 +740,7 @@ fn (mut g Gen) for_in(it ast.ForInStmt) {
|
||||||
g.write('for (int $i = 0; $i < ')
|
g.write('for (int $i = 0; $i < ')
|
||||||
g.expr(it.cond)
|
g.expr(it.cond)
|
||||||
g.writeln('.len; $i++) {')
|
g.writeln('.len; $i++) {')
|
||||||
g.write('$styp $it.val_var = ')
|
g.write('$styp ${c_name(it.val_var)} = ')
|
||||||
g.expr(it.cond)
|
g.expr(it.cond)
|
||||||
g.writeln('.args[$i];')
|
g.writeln('.args[$i];')
|
||||||
g.stmts(it.stmts)
|
g.stmts(it.stmts)
|
||||||
|
@ -751,7 +751,7 @@ fn (mut g Gen) for_in(it ast.ForInStmt) {
|
||||||
g.expr(it.cond)
|
g.expr(it.cond)
|
||||||
g.writeln('.len; $i++) {')
|
g.writeln('.len; $i++) {')
|
||||||
if it.val_var != '_' {
|
if it.val_var != '_' {
|
||||||
g.write('byte $it.val_var = ')
|
g.write('byte ${c_name(it.val_var)} = ')
|
||||||
g.expr(it.cond)
|
g.expr(it.cond)
|
||||||
g.writeln('.str[$i];')
|
g.writeln('.str[$i];')
|
||||||
}
|
}
|
||||||
|
|
|
@ -0,0 +1,48 @@
|
||||||
|
const (
|
||||||
|
nums = [1, 2, 3]
|
||||||
|
)
|
||||||
|
|
||||||
|
fn test_for_char_in() {
|
||||||
|
mut sum := 0
|
||||||
|
for char in nums {
|
||||||
|
sum += char
|
||||||
|
}
|
||||||
|
assert sum == 6
|
||||||
|
}
|
||||||
|
|
||||||
|
fn test_for_char_in_range() {
|
||||||
|
mut sum := 0
|
||||||
|
for char in 0 .. nums.len {
|
||||||
|
sum += nums[char]
|
||||||
|
}
|
||||||
|
assert sum == 6
|
||||||
|
}
|
||||||
|
|
||||||
|
fn test_for_char_complex() {
|
||||||
|
mut sum := 0
|
||||||
|
for char := 0; char < nums.len; char++ {
|
||||||
|
sum += nums[char]
|
||||||
|
}
|
||||||
|
assert sum == 6
|
||||||
|
}
|
||||||
|
|
||||||
|
fn test_for_char_in_string() {
|
||||||
|
s := 'abcd'
|
||||||
|
mut sum := 0
|
||||||
|
for char in s {
|
||||||
|
sum += char
|
||||||
|
}
|
||||||
|
assert sum == 394 // ascii codes of `a` + `b` + `c` + `d`
|
||||||
|
}
|
||||||
|
|
||||||
|
fn test_for_char_in_map() {
|
||||||
|
m := {
|
||||||
|
'a': 'b'
|
||||||
|
'c': 'd'
|
||||||
|
}
|
||||||
|
mut acc := ''
|
||||||
|
for k, char in m {
|
||||||
|
acc += '$k: $char, '
|
||||||
|
}
|
||||||
|
assert acc == 'a: b, c: d, '
|
||||||
|
}
|
Loading…
Reference in New Issue