mirror of
https://github.com/vlang/v.git
synced 2023-08-10 21:13:21 +03:00
cgen: add gen_str_for_multi_return
This commit is contained in:
parent
7f225f2eaa
commit
81c19517d1
@ -272,7 +272,7 @@ pub fn (c &Checker) get_default_fmt(ftyp, typ table.Type) byte {
|
||||
} else {
|
||||
sym := c.table.get_type_symbol(ftyp)
|
||||
if ftyp in [table.string_type, table.bool_type] || sym.kind in [.enum_, .array, .array_fixed,
|
||||
.struct_, .map] || ftyp.has_flag(.optional) || sym.has_method('str') {
|
||||
.struct_, .map, .multi_return] || ftyp.has_flag(.optional) || sym.has_method('str') {
|
||||
return `s`
|
||||
} else {
|
||||
return `_`
|
||||
|
@ -3050,7 +3050,7 @@ fn (mut g Gen) gen_expr_to_string(expr ast.Expr, etype table.Type) ?bool {
|
||||
g.enum_expr(expr)
|
||||
g.write('")')
|
||||
}
|
||||
} else if sym_has_str_method || sym.kind in [.array, .array_fixed, .map, .struct_] {
|
||||
} else if sym_has_str_method || sym.kind in [.array, .array_fixed, .map, .struct_, .multi_return] {
|
||||
is_p := etype.is_ptr()
|
||||
val_type := if is_p { etype.deref() } else { etype }
|
||||
str_fn_name := g.gen_str_for_type(val_type)
|
||||
@ -3855,6 +3855,7 @@ fn (mut g Gen) gen_str_for_type_with_styp(typ table.Type, styp string) string {
|
||||
table.Enum { g.gen_str_for_enum(it, styp, str_fn_name) }
|
||||
table.Struct { g.gen_str_for_struct(it, styp, str_fn_name) }
|
||||
table.Map { g.gen_str_for_map(it, styp, str_fn_name) }
|
||||
table.MultiReturn { g.gen_str_for_multi_return(it, styp, str_fn_name) }
|
||||
else { verror("could not generate string method $str_fn_name for type \'$styp\'") }
|
||||
}
|
||||
}
|
||||
@ -4153,6 +4154,53 @@ fn (mut g Gen) gen_str_for_varg(styp, str_fn_name string, has_str_method bool) {
|
||||
g.auto_str_funcs.writeln('}')
|
||||
}
|
||||
|
||||
fn (mut g Gen) gen_str_for_multi_return(info table.MultiReturn, styp, str_fn_name string) {
|
||||
for typ in info.types {
|
||||
sym := g.table.get_type_symbol(typ)
|
||||
if !sym.has_method('str') {
|
||||
field_styp := g.typ(typ)
|
||||
g.gen_str_for_type_with_styp(typ, field_styp)
|
||||
}
|
||||
}
|
||||
g.type_definitions.writeln('string ${str_fn_name}($styp a); // auto')
|
||||
g.auto_str_funcs.writeln('string ${str_fn_name}($styp a) {')
|
||||
g.auto_str_funcs.writeln('\tstrings__Builder sb = strings__new_builder($info.types.len * 10);')
|
||||
g.auto_str_funcs.writeln('\tstrings__Builder_write(&sb, tos_lit("("));')
|
||||
for i, typ in info.types {
|
||||
sym := g.table.get_type_symbol(typ)
|
||||
field_styp := g.typ(typ)
|
||||
is_arg_ptr := typ.is_ptr()
|
||||
sym_has_str_method, str_method_expects_ptr, _ := sym.str_method_info()
|
||||
mut arg_str_fn_name := ''
|
||||
if sym_has_str_method {
|
||||
arg_str_fn_name = if is_arg_ptr { field_styp.replace('*', '') + '_str' } else { field_styp + '_str' }
|
||||
} else {
|
||||
arg_str_fn_name = styp_to_str_fn_name(field_styp)
|
||||
}
|
||||
if sym.kind == .struct_ && !sym_has_str_method {
|
||||
g.auto_str_funcs.writeln('\tstrings__Builder_write(&sb, ${str_fn_name}(a.arg$i,0));')
|
||||
} else if sym.kind in [.f32, .f64] {
|
||||
g.auto_str_funcs.writeln('\tstrings__Builder_write(&sb, _STR("%g", 1, a.arg$i));')
|
||||
} else if sym.kind == .string {
|
||||
g.auto_str_funcs.writeln('\tstrings__Builder_write(&sb, _STR("\'%.*s\\000\'", 2, a.arg$i));')
|
||||
} else {
|
||||
if (str_method_expects_ptr && is_arg_ptr) || (!str_method_expects_ptr && !is_arg_ptr) {
|
||||
g.auto_str_funcs.writeln('\tstrings__Builder_write(&sb, ${arg_str_fn_name}(a.arg$i));')
|
||||
} else if str_method_expects_ptr && !is_arg_ptr {
|
||||
g.auto_str_funcs.writeln('\tstrings__Builder_write(&sb, ${arg_str_fn_name}(&a.arg$i));')
|
||||
} else if !str_method_expects_ptr && is_arg_ptr {
|
||||
g.auto_str_funcs.writeln('\tstrings__Builder_write(&sb, ${arg_str_fn_name}(*a.arg$i));')
|
||||
}
|
||||
}
|
||||
if i != info.types.len - 1 {
|
||||
g.auto_str_funcs.writeln('\tstrings__Builder_write(&sb, tos_lit(", "));')
|
||||
}
|
||||
}
|
||||
g.auto_str_funcs.writeln('\tstrings__Builder_write(&sb, tos_lit(")"));')
|
||||
g.auto_str_funcs.writeln('\treturn strings__Builder_str(&sb);')
|
||||
g.auto_str_funcs.writeln('}')
|
||||
}
|
||||
|
||||
fn (g Gen) type_to_fmt(typ table.Type) string {
|
||||
sym := g.table.get_type_symbol(typ)
|
||||
if sym.kind in [.struct_, .array, .array_fixed, .map] {
|
||||
|
48
vlib/v/tests/string_interpolation_multi_return_test.v
Normal file
48
vlib/v/tests/string_interpolation_multi_return_test.v
Normal file
@ -0,0 +1,48 @@
|
||||
|
||||
struct RetTest {}
|
||||
|
||||
fn (r RetTest) return_multi_1() (int, string) {
|
||||
return 1, 'aaa'
|
||||
}
|
||||
|
||||
fn (r RetTest) return_multi_2() (string, bool, []int) {
|
||||
return 'aaa', true, [1,2,3]
|
||||
}
|
||||
|
||||
fn (r RetTest) return_multi_3() (f32, bool, []string) {
|
||||
return 2.2, false, ['a', 'b', 'c']
|
||||
}
|
||||
|
||||
fn return_multi_4() (int, string) {
|
||||
return 1, 'aaa'
|
||||
}
|
||||
|
||||
fn return_multi_5() (string, bool, []int) {
|
||||
return 'aaa', true, [1,2,3]
|
||||
}
|
||||
|
||||
fn return_multi_6() (f32, bool, []string) {
|
||||
return 2.2, false, ['a', 'b', 'c']
|
||||
}
|
||||
|
||||
fn test_multi_return_interpolation() {
|
||||
r := RetTest{}
|
||||
|
||||
s1 := '${r.return_multi_1()}'
|
||||
assert s1 == "(1, 'aaa')"
|
||||
|
||||
s2 := '${r.return_multi_2()}'
|
||||
assert s2 == "('aaa', true, [1, 2, 3])"
|
||||
|
||||
s3 := '${r.return_multi_3()}'
|
||||
assert s3 == "(2.2, false, ['a', 'b', 'c'])"
|
||||
|
||||
s4 := '${return_multi_4()}'
|
||||
assert s4 == "(1, 'aaa')"
|
||||
|
||||
s5 := '${return_multi_5()}'
|
||||
assert s5 == "('aaa', true, [1, 2, 3])"
|
||||
|
||||
s6 := '${return_multi_6()}'
|
||||
assert s6 == "(2.2, false, ['a', 'b', 'c'])"
|
||||
}
|
Loading…
Reference in New Issue
Block a user