cgen: split up array_init() (#14178)
parent
1c48a8d760
commit
be513b4c27
|
@ -22,7 +22,55 @@ fn (mut g Gen) array_init(node ast.ArrayInit) {
|
||||||
array_styp = g.typ(array_type.typ)
|
array_styp = g.typ(array_type.typ)
|
||||||
g.write('HEAP($array_styp, ')
|
g.write('HEAP($array_styp, ')
|
||||||
}
|
}
|
||||||
|
len := node.exprs.len
|
||||||
if array_type.unaliased_sym.kind == .array_fixed {
|
if array_type.unaliased_sym.kind == .array_fixed {
|
||||||
|
g.fixed_array_init(node, array_type)
|
||||||
|
} else if len == 0 {
|
||||||
|
// `[]int{len: 6, cap:10, init:22}`
|
||||||
|
g.array_init_with_fields(node, elem_type, is_amp, shared_styp)
|
||||||
|
} else {
|
||||||
|
// `[1, 2, 3]`
|
||||||
|
elem_styp := g.typ(elem_type.typ)
|
||||||
|
noscan := g.check_noscan(elem_type.typ)
|
||||||
|
if elem_type.unaliased_sym.kind == .function {
|
||||||
|
g.write('new_array_from_c_array($len, $len, sizeof(voidptr), _MOV((voidptr[$len]){')
|
||||||
|
} else if g.is_empty_struct(elem_type) {
|
||||||
|
g.write('new_array_from_c_array${noscan}($len, $len, sizeof(voidptr), _MOV(($elem_styp[$len]){')
|
||||||
|
} else {
|
||||||
|
g.write('new_array_from_c_array${noscan}($len, $len, sizeof($elem_styp), _MOV(($elem_styp[$len]){')
|
||||||
|
}
|
||||||
|
if len > 8 {
|
||||||
|
g.writeln('')
|
||||||
|
g.write('\t\t')
|
||||||
|
}
|
||||||
|
for i, expr in node.exprs {
|
||||||
|
if node.expr_types[i] == ast.string_type && expr !is ast.StringLiteral
|
||||||
|
&& expr !is ast.StringInterLiteral {
|
||||||
|
g.write('string_clone(')
|
||||||
|
g.expr(expr)
|
||||||
|
g.write(')')
|
||||||
|
} else {
|
||||||
|
g.expr_with_cast(expr, node.expr_types[i], node.elem_type)
|
||||||
|
}
|
||||||
|
if i != len - 1 {
|
||||||
|
if i > 0 && i & 7 == 0 { // i > 0 && i % 8 == 0
|
||||||
|
g.writeln(',')
|
||||||
|
g.write('\t\t')
|
||||||
|
} else {
|
||||||
|
g.write(', ')
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
g.write('}))')
|
||||||
|
if g.is_shared {
|
||||||
|
g.write('}, sizeof($shared_styp))')
|
||||||
|
} else if is_amp {
|
||||||
|
g.write(')')
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
fn (mut g Gen) fixed_array_init(node ast.ArrayInit, array_type Type) {
|
||||||
if node.has_it {
|
if node.has_it {
|
||||||
g.inside_lambda = true
|
g.inside_lambda = true
|
||||||
tmp := g.new_tmp_var()
|
tmp := g.new_tmp_var()
|
||||||
|
@ -115,11 +163,12 @@ fn (mut g Gen) array_init(node ast.ArrayInit) {
|
||||||
g.write(stmt_str)
|
g.write(stmt_str)
|
||||||
g.write(tmp_var)
|
g.write(tmp_var)
|
||||||
}
|
}
|
||||||
return
|
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// `[]int{len: 6, cap: 10, init: it * it}`
|
||||||
|
fn (mut g Gen) array_init_with_fields(node ast.ArrayInit, elem_type Type, is_amp bool, shared_styp string) {
|
||||||
elem_styp := g.typ(elem_type.typ)
|
elem_styp := g.typ(elem_type.typ)
|
||||||
noscan := g.check_noscan(elem_type.typ)
|
noscan := g.check_noscan(elem_type.typ)
|
||||||
if node.exprs.len == 0 {
|
|
||||||
is_default_array := elem_type.unaliased_sym.kind == .array && node.has_default
|
is_default_array := elem_type.unaliased_sym.kind == .array && node.has_default
|
||||||
is_default_map := elem_type.unaliased_sym.kind == .map && node.has_default
|
is_default_map := elem_type.unaliased_sym.kind == .map && node.has_default
|
||||||
if node.has_it { // []int{len: 6, init: it * it} when variable it is used in init expression
|
if node.has_it { // []int{len: 6, init: it * it} when variable it is used in init expression
|
||||||
|
@ -246,44 +295,6 @@ fn (mut g Gen) array_init(node ast.ArrayInit) {
|
||||||
} else if is_amp {
|
} else if is_amp {
|
||||||
g.write(')')
|
g.write(')')
|
||||||
}
|
}
|
||||||
return
|
|
||||||
}
|
|
||||||
len := node.exprs.len
|
|
||||||
if elem_type.unaliased_sym.kind == .function {
|
|
||||||
g.write('new_array_from_c_array($len, $len, sizeof(voidptr), _MOV((voidptr[$len]){')
|
|
||||||
} else if g.is_empty_struct(elem_type) {
|
|
||||||
g.write('new_array_from_c_array${noscan}($len, $len, sizeof(voidptr), _MOV(($elem_styp[$len]){')
|
|
||||||
} else {
|
|
||||||
g.write('new_array_from_c_array${noscan}($len, $len, sizeof($elem_styp), _MOV(($elem_styp[$len]){')
|
|
||||||
}
|
|
||||||
if len > 8 {
|
|
||||||
g.writeln('')
|
|
||||||
g.write('\t\t')
|
|
||||||
}
|
|
||||||
for i, expr in node.exprs {
|
|
||||||
if node.expr_types[i] == ast.string_type && expr !is ast.StringLiteral
|
|
||||||
&& expr !is ast.StringInterLiteral {
|
|
||||||
g.write('string_clone(')
|
|
||||||
g.expr(expr)
|
|
||||||
g.write(')')
|
|
||||||
} else {
|
|
||||||
g.expr_with_cast(expr, node.expr_types[i], node.elem_type)
|
|
||||||
}
|
|
||||||
if i != len - 1 {
|
|
||||||
if i > 0 && i & 7 == 0 { // i > 0 && i % 8 == 0
|
|
||||||
g.writeln(',')
|
|
||||||
g.write('\t\t')
|
|
||||||
} else {
|
|
||||||
g.write(', ')
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
g.write('}))')
|
|
||||||
if g.is_shared {
|
|
||||||
g.write('}, sizeof($shared_styp))')
|
|
||||||
} else if is_amp {
|
|
||||||
g.write(')')
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
|
||||||
// `nums.map(it % 2 == 0)`
|
// `nums.map(it % 2 == 0)`
|
||||||
|
|
Loading…
Reference in New Issue