From 8a75d68421b57b6578faef41a4c82c197b7c42d0 Mon Sep 17 00:00:00 2001
From: yuyi <yuyi98@163.com>
Date: Wed, 27 Apr 2022 01:58:47 +0800
Subject: [PATCH] cgen: split up array_init() (#14178)

---
 vlib/v/gen/c/array.v | 367 ++++++++++++++++++++++---------------------
 1 file changed, 189 insertions(+), 178 deletions(-)

diff --git a/vlib/v/gen/c/array.v b/vlib/v/gen/c/array.v
index fe072c5b44..fa32b168e3 100644
--- a/vlib/v/gen/c/array.v
+++ b/vlib/v/gen/c/array.v
@@ -22,182 +22,165 @@ fn (mut g Gen) array_init(node ast.ArrayInit) {
 		array_styp = g.typ(array_type.typ)
 		g.write('HEAP($array_styp, ')
 	}
+	len := node.exprs.len
 	if array_type.unaliased_sym.kind == .array_fixed {
-		if node.has_it {
-			g.inside_lambda = true
-			tmp := g.new_tmp_var()
-			mut s := g.go_before_stmt(0)
-			s_ends_with_ln := s.ends_with('\n')
-			s = s.trim_space()
-			ret_typ := g.typ(node.typ)
-			elem_typ := g.typ(node.elem_type)
-			g.empty_line = true
-			g.write('$ret_typ $tmp =')
-			g.write('{')
-			if node.has_val {
-				for i, expr in node.exprs {
-					if expr.is_auto_deref_var() {
-						g.write('*')
-					}
-					g.write('0')
-					if i != node.exprs.len - 1 {
-						g.write(', ')
-					}
-				}
-			} else if node.has_default {
-				g.write('0')
-				info := array_type.unaliased_sym.info as ast.ArrayFixed
-				for _ in 1 .. info.size {
+		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('0')
 				}
-			} else {
-				g.write('0')
 			}
-			g.write('}')
-			g.writeln(';')
-			g.writeln('{')
-			g.indent++
-			g.writeln('$elem_typ* pelem = ($elem_typ*)$tmp;')
-			g.writeln('int _len = (int)sizeof($tmp) / sizeof($elem_typ);')
-			g.writeln('for(int it=0; it<_len; it++, pelem++) {')
-			g.indent++
-			g.write('*pelem = ')
-			g.expr(node.default_expr)
-			g.writeln(';')
-			g.indent--
-			g.writeln('}')
-			g.indent--
-			g.writeln('}')
-			if s_ends_with_ln {
-				g.writeln(s)
-			} else {
-				g.write(s)
-			}
-			g.write(tmp)
-			g.inside_lambda = false
-			return
 		}
-		need_tmp_var := g.inside_call && !g.inside_struct_init && node.exprs.len == 0
-		mut stmt_str := ''
-		mut tmp_var := ''
-		if need_tmp_var {
-			tmp_var = g.new_tmp_var()
-			stmt_str = g.go_before_stmt(0)
-			ret_typ := g.typ(node.typ)
-			g.empty_line = true
-			g.write('$ret_typ $tmp_var = ')
+		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 {
+		g.inside_lambda = true
+		tmp := g.new_tmp_var()
+		mut s := g.go_before_stmt(0)
+		s_ends_with_ln := s.ends_with('\n')
+		s = s.trim_space()
+		ret_typ := g.typ(node.typ)
+		elem_typ := g.typ(node.elem_type)
+		g.empty_line = true
+		g.write('$ret_typ $tmp =')
 		g.write('{')
 		if node.has_val {
 			for i, expr in node.exprs {
 				if expr.is_auto_deref_var() {
 					g.write('*')
 				}
-				g.expr(expr)
+				g.write('0')
 				if i != node.exprs.len - 1 {
 					g.write(', ')
 				}
 			}
 		} else if node.has_default {
-			g.expr(node.default_expr)
+			g.write('0')
 			info := array_type.unaliased_sym.info as ast.ArrayFixed
 			for _ in 1 .. info.size {
 				g.write(', ')
-				g.expr(node.default_expr)
+				g.write('0')
 			}
 		} else {
 			g.write('0')
 		}
 		g.write('}')
-		if need_tmp_var {
-			g.writeln(';')
-			g.write(stmt_str)
-			g.write(tmp_var)
+		g.writeln(';')
+		g.writeln('{')
+		g.indent++
+		g.writeln('$elem_typ* pelem = ($elem_typ*)$tmp;')
+		g.writeln('int _len = (int)sizeof($tmp) / sizeof($elem_typ);')
+		g.writeln('for(int it=0; it<_len; it++, pelem++) {')
+		g.indent++
+		g.write('*pelem = ')
+		g.expr(node.default_expr)
+		g.writeln(';')
+		g.indent--
+		g.writeln('}')
+		g.indent--
+		g.writeln('}')
+		if s_ends_with_ln {
+			g.writeln(s)
+		} else {
+			g.write(s)
 		}
+		g.write(tmp)
+		g.inside_lambda = false
 		return
 	}
+	need_tmp_var := g.inside_call && !g.inside_struct_init && node.exprs.len == 0
+	mut stmt_str := ''
+	mut tmp_var := ''
+	if need_tmp_var {
+		tmp_var = g.new_tmp_var()
+		stmt_str = g.go_before_stmt(0)
+		ret_typ := g.typ(node.typ)
+		g.empty_line = true
+		g.write('$ret_typ $tmp_var = ')
+	}
+	g.write('{')
+	if node.has_val {
+		for i, expr in node.exprs {
+			if expr.is_auto_deref_var() {
+				g.write('*')
+			}
+			g.expr(expr)
+			if i != node.exprs.len - 1 {
+				g.write(', ')
+			}
+		}
+	} else if node.has_default {
+		g.expr(node.default_expr)
+		info := array_type.unaliased_sym.info as ast.ArrayFixed
+		for _ in 1 .. info.size {
+			g.write(', ')
+			g.expr(node.default_expr)
+		}
+	} else {
+		g.write('0')
+	}
+	g.write('}')
+	if need_tmp_var {
+		g.writeln(';')
+		g.write(stmt_str)
+		g.write(tmp_var)
+	}
+}
+
+// `[]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)
 	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_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
-			g.inside_lambda = true
-			tmp := g.new_tmp_var()
-			mut s := g.go_before_stmt(0)
-			s_ends_with_ln := s.ends_with('\n')
-			s = s.trim_space()
-			ret_typ := g.typ(node.typ)
-			elem_typ := g.typ(node.elem_type)
-			g.empty_line = true
-			g.write('$ret_typ $tmp =')
-			if is_default_array {
-				g.write('__new_array_with_array_default${noscan}(')
-			} else if is_default_map {
-				g.write('__new_array_with_map_default${noscan}(')
-			} else {
-				g.write('__new_array_with_default${noscan}(')
-			}
-			if node.has_len {
-				g.expr(node.len_expr)
-				g.write(', ')
-			} else {
-				g.write('0, ')
-			}
-			if node.has_cap {
-				g.expr(node.cap_expr)
-				g.write(', ')
-			} else {
-				g.write('0, ')
-			}
-			if elem_type.unaliased_sym.kind == .function || g.is_empty_struct(elem_type) {
-				g.write('sizeof(voidptr), ')
-			} else {
-				g.write('sizeof($elem_styp), ')
-			}
-			if is_default_array {
-				g.write('($elem_styp[]){')
-				g.expr(node.default_expr)
-				g.write('}[0])')
-			} else if node.has_len && node.elem_type == ast.string_type {
-				g.write('&($elem_styp[]){')
-				g.write('_SLIT("")')
-				g.write('})')
-			} else if node.has_len && elem_type.unaliased_sym.kind in [.array, .map] {
-				g.write('(voidptr)&($elem_styp[]){')
-				g.write(g.type_default(node.elem_type))
-				g.write('}[0])')
-			} else {
-				g.write('0)')
-			}
-			if g.is_shared {
-				g.write('}, sizeof($shared_styp))')
-			} else if is_amp {
-				g.write(')')
-			}
-			g.writeln(';')
-			g.writeln('{')
-			g.indent++
-			g.writeln('$elem_typ* pelem = ($elem_typ*)${tmp}.data;')
-			g.writeln('for(int it=0; it<${tmp}.len; it++, pelem++) {')
-			g.indent++
-			g.write('*pelem = ')
-			g.expr(node.default_expr)
-			g.writeln(';')
-			g.indent--
-			g.writeln('}')
-			g.indent--
-			g.writeln('}')
-			if s_ends_with_ln {
-				g.writeln(s)
-			} else {
-				g.write(s)
-			}
-			g.write(tmp)
-			g.inside_lambda = false
-			return
-		}
+	is_default_array := elem_type.unaliased_sym.kind == .array && 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
+		g.inside_lambda = true
+		tmp := g.new_tmp_var()
+		mut s := g.go_before_stmt(0)
+		s_ends_with_ln := s.ends_with('\n')
+		s = s.trim_space()
+		ret_typ := g.typ(node.typ)
+		elem_typ := g.typ(node.elem_type)
+		g.empty_line = true
+		g.write('$ret_typ $tmp =')
 		if is_default_array {
 			g.write('__new_array_with_array_default${noscan}(')
 		} else if is_default_map {
@@ -222,14 +205,10 @@ fn (mut g Gen) array_init(node ast.ArrayInit) {
 		} else {
 			g.write('sizeof($elem_styp), ')
 		}
-		if is_default_array || is_default_map {
+		if is_default_array {
 			g.write('($elem_styp[]){')
 			g.expr(node.default_expr)
 			g.write('}[0])')
-		} else if node.has_default {
-			g.write('&($elem_styp[]){')
-			g.expr_with_cast(node.default_expr, node.default_type, node.elem_type)
-			g.write('})')
 		} else if node.has_len && node.elem_type == ast.string_type {
 			g.write('&($elem_styp[]){')
 			g.write('_SLIT("")')
@@ -246,39 +225,71 @@ fn (mut g Gen) array_init(node ast.ArrayInit) {
 		} else if is_amp {
 			g.write(')')
 		}
+		g.writeln(';')
+		g.writeln('{')
+		g.indent++
+		g.writeln('$elem_typ* pelem = ($elem_typ*)${tmp}.data;')
+		g.writeln('for(int it=0; it<${tmp}.len; it++, pelem++) {')
+		g.indent++
+		g.write('*pelem = ')
+		g.expr(node.default_expr)
+		g.writeln(';')
+		g.indent--
+		g.writeln('}')
+		g.indent--
+		g.writeln('}')
+		if s_ends_with_ln {
+			g.writeln(s)
+		} else {
+			g.write(s)
+		}
+		g.write(tmp)
+		g.inside_lambda = false
 		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]){')
+	if is_default_array {
+		g.write('__new_array_with_array_default${noscan}(')
+	} else if is_default_map {
+		g.write('__new_array_with_map_default${noscan}(')
 	} else {
-		g.write('new_array_from_c_array${noscan}($len, $len, sizeof($elem_styp), _MOV(($elem_styp[$len]){')
+		g.write('__new_array_with_default${noscan}(')
 	}
-	if len > 8 {
-		g.writeln('')
-		g.write('\t\t')
+	if node.has_len {
+		g.expr(node.len_expr)
+		g.write(', ')
+	} else {
+		g.write('0, ')
 	}
-	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(', ')
-			}
-		}
+	if node.has_cap {
+		g.expr(node.cap_expr)
+		g.write(', ')
+	} else {
+		g.write('0, ')
+	}
+	if elem_type.unaliased_sym.kind == .function || g.is_empty_struct(elem_type) {
+		g.write('sizeof(voidptr), ')
+	} else {
+		g.write('sizeof($elem_styp), ')
+	}
+	if is_default_array || is_default_map {
+		g.write('($elem_styp[]){')
+		g.expr(node.default_expr)
+		g.write('}[0])')
+	} else if node.has_default {
+		g.write('&($elem_styp[]){')
+		g.expr_with_cast(node.default_expr, node.default_type, node.elem_type)
+		g.write('})')
+	} else if node.has_len && node.elem_type == ast.string_type {
+		g.write('&($elem_styp[]){')
+		g.write('_SLIT("")')
+		g.write('})')
+	} else if node.has_len && elem_type.unaliased_sym.kind in [.array, .map] {
+		g.write('(voidptr)&($elem_styp[]){')
+		g.write(g.type_default(node.elem_type))
+		g.write('}[0])')
+	} else {
+		g.write('0)')
 	}
-	g.write('}))')
 	if g.is_shared {
 		g.write('}, sizeof($shared_styp))')
 	} else if is_amp {