parent
17266ecb07
commit
5ddb70940e
|
@ -344,19 +344,16 @@ fn (mut g Gen) gen_str_for_map(info table.Map, styp string, str_fn_name string)
|
|||
g.auto_str_funcs.writeln('\t\tstrings__Builder_write(&sb, _SLIT(": "));')
|
||||
if val_sym.kind == .function {
|
||||
g.auto_str_funcs.writeln('\t\tstrings__Builder_write(&sb, ${elem_str_fn_name}());')
|
||||
} else if val_sym.kind == .string {
|
||||
g.auto_str_funcs.writeln('\t\tstrings__Builder_write(&sb, _STR("\'%.*s\\000\'", 2, *($val_styp*)DenseArray_value(&m.key_values, i)));')
|
||||
} else if val_sym.kind == .struct_ && !val_sym.has_method('str') {
|
||||
g.auto_str_funcs.writeln('\t\tstrings__Builder_write(&sb, indent_${elem_str_fn_name}(*($val_styp*)DenseArray_value(&m.key_values, i), indent_count));')
|
||||
} else if val_sym.kind in [.f32, .f64] {
|
||||
g.auto_str_funcs.writeln('\t\tstrings__Builder_write(&sb, _STR("%g", 1, *($val_styp*)DenseArray_value(&m.key_values, i)));')
|
||||
} else if val_sym.kind == .rune {
|
||||
g.auto_str_funcs.writeln('\t\tstrings__Builder_write(&sb, _STR("`%.*s\\000`", 2, ${elem_str_fn_name}(*($val_styp*)DenseArray_value(&m.key_values, i))));')
|
||||
} else {
|
||||
g.auto_str_funcs.writeln('\t\t$val_styp it = *($val_styp*)DenseArray_value(&m.key_values, i);')
|
||||
if val_sym.kind == .string {
|
||||
g.auto_str_funcs.writeln('\t\tstrings__Builder_write(&sb, _STR("\'%.*s\\000\'", 2, it));')
|
||||
} else if val_sym.kind == .struct_ && !val_sym.has_method('str') {
|
||||
g.auto_str_funcs.writeln('\t\tstrings__Builder_write(&sb, indent_${elem_str_fn_name}(it, indent_count));')
|
||||
} else if val_sym.kind in [.f32, .f64] {
|
||||
g.auto_str_funcs.writeln('\t\tstrings__Builder_write(&sb, _STR("%g", 1, it));')
|
||||
} else if val_sym.kind == .rune {
|
||||
g.auto_str_funcs.writeln('\t\tstrings__Builder_write(&sb, _STR("`%.*s\\000`", 2, ${elem_str_fn_name}(it)));')
|
||||
} else {
|
||||
g.auto_str_funcs.writeln('\t\tstrings__Builder_write(&sb, ${elem_str_fn_name}(it));')
|
||||
}
|
||||
g.auto_str_funcs.writeln('\t\tstrings__Builder_write(&sb, ${elem_str_fn_name}(*($val_styp*)DenseArray_value(&m.key_values, i)));')
|
||||
}
|
||||
g.auto_str_funcs.writeln('\t\tif (i != m.key_values.len-1) {')
|
||||
g.auto_str_funcs.writeln('\t\t\tstrings__Builder_write(&sb, _SLIT(", "));')
|
||||
|
|
|
@ -4110,7 +4110,7 @@ fn (mut g Gen) index_expr(node ast.IndexExpr) {
|
|||
g.is_array_set = true
|
||||
g.write('map_set_1(')
|
||||
} else {
|
||||
g.write('*(($elem_type_str*)map_get_and_set_1(')
|
||||
g.write('(*(($elem_type_str*)map_get_and_set_1(')
|
||||
}
|
||||
if !left_is_ptr || node.left_type.has_flag(.shared_f) {
|
||||
g.write('&')
|
||||
|
@ -4133,7 +4133,7 @@ fn (mut g Gen) index_expr(node ast.IndexExpr) {
|
|||
}
|
||||
if g.assign_op != .assign && info.value_type != table.string_type {
|
||||
zero := g.type_default(info.value_type)
|
||||
g.write('$zero }))')
|
||||
g.write('$zero })))')
|
||||
}
|
||||
} else if (g.inside_map_postfix || g.inside_map_infix) ||
|
||||
(g.is_assign_lhs && !g.is_array_set && get_and_set_types)
|
||||
|
|
|
@ -23,3 +23,29 @@ fn test_for_in_mut_val_of_fixed_array() {
|
|||
println(arr)
|
||||
assert arr == [2, 4, 6]!
|
||||
}
|
||||
|
||||
fn foo3(mut m map[string][3]int){
|
||||
for i in 0..m['hello'].len {
|
||||
m['hello'][i] *= 2
|
||||
}
|
||||
}
|
||||
|
||||
fn test_fn_mut_val_of_map() {
|
||||
mut m := {'hello': [1,2,3]!}
|
||||
foo3(mut m)
|
||||
println(m)
|
||||
assert '$m' == "{'hello': [2, 4, 6]}"
|
||||
}
|
||||
|
||||
fn foo4(mut m map[string][3]int){
|
||||
for _, mut j in m['hello'] {
|
||||
j *= 2
|
||||
}
|
||||
}
|
||||
|
||||
fn test_for_in_mut_val_of_map() {
|
||||
mut m := {'hello':[1,2,3]!}
|
||||
foo4(mut m)
|
||||
println(m)
|
||||
assert '$m' == "{'hello': [2, 4, 6]}"
|
||||
}
|
||||
|
|
|
@ -36,4 +36,7 @@ fn test_interpolation_map_to_string() {
|
|||
assert s.contains("y: 'abc'")
|
||||
assert s.contains("}, '2': Test{")
|
||||
assert s.contains("y: 'def'")
|
||||
|
||||
f := {'hello': [1,2,3]!}
|
||||
assert '$f' == "{'hello': [1, 2, 3]}"
|
||||
}
|
||||
|
|
Loading…
Reference in New Issue