cgen: minor perf fix in index_expr() (#9054)
yuyi
GitHub
parent
31321b68ea
commit
62ccd740f0
|
|
@ -4224,408 +4224,6 @@ fn (mut g Gen) if_expr(node ast.IfExpr) {
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
fn (mut g Gen) index_expr(node ast.IndexExpr) {
|
|
||||||
gen_or := node.or_expr.kind != .absent
|
|
||||||
match node.index {
|
|
||||||
ast.RangeExpr {
|
|
||||||
sym := g.table.get_type_symbol(node.left_type)
|
|
||||||
if sym.kind == .string {
|
|
||||||
g.write('string_substr(')
|
|
||||||
g.expr(node.left)
|
|
||||||
} else if sym.kind == .array {
|
|
||||||
g.write('array_slice(')
|
|
||||||
if node.left_type.is_ptr() {
|
|
||||||
g.write('*')
|
|
||||||
}
|
|
||||||
g.expr(node.left)
|
|
||||||
} else if sym.kind == .array_fixed {
|
|
||||||
// Convert a fixed array to V array when doing `fixed_arr[start..end]`
|
|
||||||
info := sym.info as table.ArrayFixed
|
|
||||||
g.write('array_slice(new_array_from_c_array(')
|
|
||||||
g.write('$info.size')
|
|
||||||
g.write(', $info.size')
|
|
||||||
g.write(', sizeof(')
|
|
||||||
if node.left_type.is_ptr() {
|
|
||||||
g.write('(*')
|
|
||||||
}
|
|
||||||
g.expr(node.left)
|
|
||||||
if node.left_type.is_ptr() {
|
|
||||||
g.write(')')
|
|
||||||
}
|
|
||||||
g.write('[0]), ')
|
|
||||||
if node.left_type.is_ptr() {
|
|
||||||
g.write('*')
|
|
||||||
}
|
|
||||||
g.expr(node.left)
|
|
||||||
g.write(')')
|
|
||||||
} else {
|
|
||||||
g.expr(node.left)
|
|
||||||
}
|
|
||||||
g.write(', ')
|
|
||||||
if node.index.has_low {
|
|
||||||
g.expr(node.index.low)
|
|
||||||
} else {
|
|
||||||
g.write('0')
|
|
||||||
}
|
|
||||||
g.write(', ')
|
|
||||||
if node.index.has_high {
|
|
||||||
g.expr(node.index.high)
|
|
||||||
} else if sym.kind == .array_fixed {
|
|
||||||
info := sym.info as table.ArrayFixed
|
|
||||||
g.write('$info.size')
|
|
||||||
} else if node.left_type.is_ptr() {
|
|
||||||
g.write('(')
|
|
||||||
g.write('*')
|
|
||||||
g.expr(node.left)
|
|
||||||
g.write(')')
|
|
||||||
g.write('.len')
|
|
||||||
} else {
|
|
||||||
g.expr(node.left)
|
|
||||||
g.write('.len')
|
|
||||||
}
|
|
||||||
g.write(')')
|
|
||||||
}
|
|
||||||
else {
|
|
||||||
// indexing
|
|
||||||
sym := g.table.get_final_type_symbol(node.left_type)
|
|
||||||
left_is_ptr := node.left_type.is_ptr()
|
|
||||||
if sym.kind == .array {
|
|
||||||
info := sym.info as table.Array
|
|
||||||
elem_type_str := g.typ(info.elem_type)
|
|
||||||
elem_type := info.elem_type
|
|
||||||
elem_typ := g.table.get_type_symbol(elem_type)
|
|
||||||
// `vals[i].field = x` is an exception and requires `array_get`:
|
|
||||||
// `(*(Val*)array_get(vals, i)).field = x;`
|
|
||||||
is_selector := node.left is ast.SelectorExpr
|
|
||||||
if g.is_assign_lhs && !is_selector && node.is_setter {
|
|
||||||
is_direct_array_access := g.fn_decl != 0 && g.fn_decl.is_direct_arr
|
|
||||||
is_op_assign := g.assign_op != .assign && info.elem_type != table.string_type
|
|
||||||
array_ptr_type_str := match elem_typ.kind {
|
|
||||||
.function { 'voidptr*' }
|
|
||||||
else { '$elem_type_str*' }
|
|
||||||
}
|
|
||||||
if is_direct_array_access {
|
|
||||||
g.write('(($array_ptr_type_str)')
|
|
||||||
} else if is_op_assign {
|
|
||||||
g.write('(*($array_ptr_type_str)array_get(')
|
|
||||||
if left_is_ptr && !node.left_type.has_flag(.shared_f) {
|
|
||||||
g.write('*')
|
|
||||||
}
|
|
||||||
} else {
|
|
||||||
g.is_array_set = true // special handling of assign_op and closing with '})'
|
|
||||||
g.write('array_set(')
|
|
||||||
if !left_is_ptr || node.left_type.has_flag(.shared_f) {
|
|
||||||
g.write('&')
|
|
||||||
}
|
|
||||||
}
|
|
||||||
g.expr(node.left)
|
|
||||||
if node.left_type.has_flag(.shared_f) {
|
|
||||||
if left_is_ptr {
|
|
||||||
g.write('->val')
|
|
||||||
} else {
|
|
||||||
g.write('.val')
|
|
||||||
}
|
|
||||||
}
|
|
||||||
if is_direct_array_access {
|
|
||||||
if left_is_ptr && !node.left_type.has_flag(.shared_f) {
|
|
||||||
g.write('->')
|
|
||||||
} else {
|
|
||||||
g.write('.')
|
|
||||||
}
|
|
||||||
g.write('data)[')
|
|
||||||
g.expr(node.index)
|
|
||||||
g.write(']')
|
|
||||||
} else {
|
|
||||||
g.write(', ')
|
|
||||||
g.expr(node.index)
|
|
||||||
if !is_op_assign {
|
|
||||||
mut need_wrapper := true
|
|
||||||
/*
|
|
||||||
match node.right {
|
|
||||||
ast.EnumVal, ast.Ident {
|
|
||||||
// `&x` is enough for variables and enums
|
|
||||||
// `&(Foo[]){ ... }` is only needed for function calls and literals
|
|
||||||
need_wrapper = false
|
|
||||||
}
|
|
||||||
else {}
|
|
||||||
}
|
|
||||||
*/
|
|
||||||
if need_wrapper {
|
|
||||||
if elem_typ.kind == .function {
|
|
||||||
g.write(', &(voidptr[]) { ')
|
|
||||||
} else {
|
|
||||||
g.write(', &($elem_type_str[]) { ')
|
|
||||||
}
|
|
||||||
} else {
|
|
||||||
g.write(', &')
|
|
||||||
}
|
|
||||||
} else {
|
|
||||||
// `x[0] *= y`
|
|
||||||
g.write('))')
|
|
||||||
}
|
|
||||||
}
|
|
||||||
} else {
|
|
||||||
is_direct_array_access := g.fn_decl != 0 && g.fn_decl.is_direct_arr
|
|
||||||
array_ptr_type_str := match elem_typ.kind {
|
|
||||||
.function { 'voidptr*' }
|
|
||||||
else { '$elem_type_str*' }
|
|
||||||
}
|
|
||||||
needs_clone := info.elem_type == table.string_type_idx && g.is_autofree
|
|
||||||
&& !g.is_assign_lhs
|
|
||||||
is_gen_or_and_assign_rhs := gen_or && !g.discard_or_result
|
|
||||||
cur_line := if is_gen_or_and_assign_rhs {
|
|
||||||
line := g.go_before_stmt(0)
|
|
||||||
g.out.write_string(c.tabs[g.indent])
|
|
||||||
line
|
|
||||||
} else {
|
|
||||||
''
|
|
||||||
}
|
|
||||||
tmp_opt := if gen_or { g.new_tmp_var() } else { '' }
|
|
||||||
tmp_opt_ptr := if gen_or { g.new_tmp_var() } else { '' }
|
|
||||||
if gen_or {
|
|
||||||
g.write('$array_ptr_type_str $tmp_opt_ptr = ($array_ptr_type_str)/*ee elem_ptr_typ */(array_get_with_check(')
|
|
||||||
} else {
|
|
||||||
if needs_clone {
|
|
||||||
g.write('/*2*/string_clone(')
|
|
||||||
}
|
|
||||||
if g.is_fn_index_call {
|
|
||||||
if elem_typ.info is table.FnType {
|
|
||||||
g.write('((')
|
|
||||||
g.write_fn_ptr_decl(&elem_typ.info, '')
|
|
||||||
g.write(')(*($array_ptr_type_str)/*ee elem_typ */array_get(')
|
|
||||||
}
|
|
||||||
if left_is_ptr && !node.left_type.has_flag(.shared_f) {
|
|
||||||
g.write('*')
|
|
||||||
}
|
|
||||||
} else if is_direct_array_access {
|
|
||||||
g.write('(($array_ptr_type_str)')
|
|
||||||
} else {
|
|
||||||
g.write('(*($array_ptr_type_str)/*ee elem_typ */array_get(')
|
|
||||||
if left_is_ptr && !node.left_type.has_flag(.shared_f) {
|
|
||||||
g.write('*')
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
g.expr(node.left)
|
|
||||||
// TODO: test direct_array_access when 'shared' is implemented
|
|
||||||
if node.left_type.has_flag(.shared_f) {
|
|
||||||
if left_is_ptr {
|
|
||||||
g.write('->val')
|
|
||||||
} else {
|
|
||||||
g.write('.val')
|
|
||||||
}
|
|
||||||
}
|
|
||||||
if is_direct_array_access && !gen_or {
|
|
||||||
if left_is_ptr && !node.left_type.has_flag(.shared_f) {
|
|
||||||
g.write('->')
|
|
||||||
} else {
|
|
||||||
g.write('.')
|
|
||||||
}
|
|
||||||
g.write('data)[')
|
|
||||||
g.expr(node.index)
|
|
||||||
g.write(']')
|
|
||||||
} else {
|
|
||||||
g.write(', ')
|
|
||||||
g.expr(node.index)
|
|
||||||
if g.is_fn_index_call {
|
|
||||||
g.write(')))')
|
|
||||||
} else {
|
|
||||||
g.write('))')
|
|
||||||
}
|
|
||||||
}
|
|
||||||
if needs_clone {
|
|
||||||
g.write(')')
|
|
||||||
}
|
|
||||||
if gen_or {
|
|
||||||
g.writeln(';')
|
|
||||||
opt_elem_type := g.typ(elem_type.set_flag(.optional))
|
|
||||||
g.writeln('$opt_elem_type $tmp_opt = {0};')
|
|
||||||
g.writeln('if ($tmp_opt_ptr) {')
|
|
||||||
g.writeln('\t*(($elem_type_str*)&${tmp_opt}.data) = *(($elem_type_str*)$tmp_opt_ptr);')
|
|
||||||
g.writeln('} else {')
|
|
||||||
g.writeln('\t${tmp_opt}.state = 2; ${tmp_opt}.err = (Error){.msg=_SLIT("array index out of range"), .code=0};')
|
|
||||||
g.writeln('}')
|
|
||||||
g.or_block(tmp_opt, node.or_expr, elem_type)
|
|
||||||
g.write('\n$cur_line*($elem_type_str*)${tmp_opt}.data')
|
|
||||||
}
|
|
||||||
}
|
|
||||||
} else if sym.kind == .array_fixed {
|
|
||||||
info := sym.info as table.ArrayFixed
|
|
||||||
elem_type := info.elem_type
|
|
||||||
elem_sym := g.table.get_type_symbol(elem_type)
|
|
||||||
is_fn_index_call := g.is_fn_index_call && elem_sym.info is table.FnType
|
|
||||||
|
|
||||||
if is_fn_index_call {
|
|
||||||
g.write('(*')
|
|
||||||
}
|
|
||||||
if node.left_type.is_ptr() {
|
|
||||||
g.write('(*')
|
|
||||||
g.expr(node.left)
|
|
||||||
g.write(')')
|
|
||||||
} else {
|
|
||||||
g.expr(node.left)
|
|
||||||
}
|
|
||||||
g.write('[')
|
|
||||||
direct := g.fn_decl != 0 && g.fn_decl.is_direct_arr
|
|
||||||
if direct || node.index is ast.IntegerLiteral {
|
|
||||||
g.expr(node.index)
|
|
||||||
} else {
|
|
||||||
// bounds check
|
|
||||||
g.write('v_fixed_index(')
|
|
||||||
g.expr(node.index)
|
|
||||||
g.write(', $info.size)')
|
|
||||||
}
|
|
||||||
g.write(']')
|
|
||||||
if is_fn_index_call {
|
|
||||||
g.write(')')
|
|
||||||
}
|
|
||||||
} else if sym.kind == .map {
|
|
||||||
info := sym.info as table.Map
|
|
||||||
key_type_str := g.typ(info.key_type)
|
|
||||||
elem_type := info.value_type
|
|
||||||
elem_type_str := g.typ(elem_type)
|
|
||||||
elem_typ := g.table.get_type_symbol(elem_type)
|
|
||||||
get_and_set_types := elem_typ.kind in [.struct_, .map]
|
|
||||||
if g.is_assign_lhs && !g.is_array_set && !get_and_set_types {
|
|
||||||
if g.assign_op == .assign || info.value_type == table.string_type {
|
|
||||||
g.is_array_set = true
|
|
||||||
g.write('map_set_1(')
|
|
||||||
} else {
|
|
||||||
if node.is_setter {
|
|
||||||
g.write('(*(($elem_type_str*)map_get_and_set_1(')
|
|
||||||
} else {
|
|
||||||
g.write('(*(($elem_type_str*)map_get_1(')
|
|
||||||
}
|
|
||||||
}
|
|
||||||
if !left_is_ptr || node.left_type.has_flag(.shared_f) {
|
|
||||||
g.write('&')
|
|
||||||
}
|
|
||||||
if node.left is ast.IndexExpr {
|
|
||||||
g.inside_map_index = true
|
|
||||||
g.expr(node.left)
|
|
||||||
g.inside_map_index = false
|
|
||||||
} else {
|
|
||||||
g.expr(node.left)
|
|
||||||
}
|
|
||||||
if node.left_type.has_flag(.shared_f) {
|
|
||||||
if left_is_ptr {
|
|
||||||
g.write('->val')
|
|
||||||
} else {
|
|
||||||
g.write('.val')
|
|
||||||
}
|
|
||||||
}
|
|
||||||
g.write(', &($key_type_str[]){')
|
|
||||||
g.expr(node.index)
|
|
||||||
g.write('}')
|
|
||||||
if elem_typ.kind == .function {
|
|
||||||
g.write(', &(voidptr[]) { ')
|
|
||||||
} else {
|
|
||||||
g.array_set_pos = g.out.len
|
|
||||||
g.write(', &($elem_type_str[]) { ')
|
|
||||||
}
|
|
||||||
if g.assign_op != .assign && info.value_type != table.string_type {
|
|
||||||
zero := g.type_default(info.value_type)
|
|
||||||
g.write('$zero })))')
|
|
||||||
}
|
|
||||||
} else if g.inside_map_postfix || g.inside_map_infix
|
|
||||||
|| g.inside_map_index
|
|
||||||
|| (g.is_assign_lhs && !g.is_array_set && get_and_set_types) {
|
|
||||||
zero := g.type_default(info.value_type)
|
|
||||||
if node.is_setter {
|
|
||||||
g.write('(*($elem_type_str*)map_get_and_set_1(')
|
|
||||||
} else {
|
|
||||||
g.write('(*($elem_type_str*)map_get_1(')
|
|
||||||
}
|
|
||||||
if !left_is_ptr {
|
|
||||||
g.write('&')
|
|
||||||
}
|
|
||||||
g.expr(node.left)
|
|
||||||
g.write(', &($key_type_str[]){')
|
|
||||||
g.expr(node.index)
|
|
||||||
g.write('}, &($elem_type_str[]){ $zero }))')
|
|
||||||
} else {
|
|
||||||
zero := g.type_default(info.value_type)
|
|
||||||
is_gen_or_and_assign_rhs := gen_or && !g.discard_or_result
|
|
||||||
cur_line := if is_gen_or_and_assign_rhs {
|
|
||||||
line := g.go_before_stmt(0)
|
|
||||||
g.out.write_string(c.tabs[g.indent])
|
|
||||||
line
|
|
||||||
} else {
|
|
||||||
''
|
|
||||||
}
|
|
||||||
tmp_opt := if gen_or { g.new_tmp_var() } else { '' }
|
|
||||||
tmp_opt_ptr := if gen_or { g.new_tmp_var() } else { '' }
|
|
||||||
if gen_or {
|
|
||||||
g.write('$elem_type_str* $tmp_opt_ptr = ($elem_type_str*)/*ee elem_ptr_typ */(map_get_1_check(')
|
|
||||||
} else {
|
|
||||||
if g.is_fn_index_call {
|
|
||||||
if elem_typ.info is table.FnType {
|
|
||||||
g.write('((')
|
|
||||||
g.write_fn_ptr_decl(&elem_typ.info, '')
|
|
||||||
g.write(')(*(voidptr*)map_get_1(')
|
|
||||||
}
|
|
||||||
} else if elem_typ.kind == .function {
|
|
||||||
g.write('(*(voidptr*)map_get_1(')
|
|
||||||
} else {
|
|
||||||
g.write('(*($elem_type_str*)map_get_1(')
|
|
||||||
}
|
|
||||||
}
|
|
||||||
if !left_is_ptr || node.left_type.has_flag(.shared_f) {
|
|
||||||
g.write('ADDR(map, ')
|
|
||||||
g.expr(node.left)
|
|
||||||
} else {
|
|
||||||
g.write('(')
|
|
||||||
g.expr(node.left)
|
|
||||||
}
|
|
||||||
if node.left_type.has_flag(.shared_f) {
|
|
||||||
if left_is_ptr {
|
|
||||||
g.write('->val')
|
|
||||||
} else {
|
|
||||||
g.write('.val')
|
|
||||||
}
|
|
||||||
}
|
|
||||||
g.write('), &($key_type_str[]){')
|
|
||||||
g.expr(node.index)
|
|
||||||
g.write('}')
|
|
||||||
if gen_or {
|
|
||||||
g.write('))')
|
|
||||||
} else if g.is_fn_index_call {
|
|
||||||
g.write(', &(voidptr[]){ $zero })))')
|
|
||||||
} else if elem_typ.kind == .function {
|
|
||||||
g.write(', &(voidptr[]){ $zero }))')
|
|
||||||
} else {
|
|
||||||
g.write(', &($elem_type_str[]){ $zero }))')
|
|
||||||
}
|
|
||||||
if gen_or {
|
|
||||||
g.writeln(';')
|
|
||||||
opt_elem_type := g.typ(elem_type.set_flag(.optional))
|
|
||||||
g.writeln('$opt_elem_type $tmp_opt = {0};')
|
|
||||||
g.writeln('if ($tmp_opt_ptr) {')
|
|
||||||
g.writeln('\t*(($elem_type_str*)&${tmp_opt}.data) = *(($elem_type_str*)$tmp_opt_ptr);')
|
|
||||||
g.writeln('} else {')
|
|
||||||
g.writeln('\t${tmp_opt}.state = 2; ${tmp_opt}.err = (Error){.msg=_SLIT("array index out of range"), .code=0};')
|
|
||||||
|
|
||||||
g.writeln('}')
|
|
||||||
g.or_block(tmp_opt, node.or_expr, elem_type)
|
|
||||||
g.write('\n$cur_line*($elem_type_str*)${tmp_opt}.data')
|
|
||||||
}
|
|
||||||
}
|
|
||||||
} else if sym.kind == .string && !node.left_type.is_ptr() {
|
|
||||||
g.write('string_at(')
|
|
||||||
g.expr(node.left)
|
|
||||||
g.write(', ')
|
|
||||||
g.expr(node.index)
|
|
||||||
g.write(')')
|
|
||||||
} else {
|
|
||||||
g.expr(node.left)
|
|
||||||
g.write('[')
|
|
||||||
g.expr(node.index)
|
|
||||||
g.write(']')
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
[inline]
|
[inline]
|
||||||
fn (g &Gen) expr_is_multi_return_call(expr ast.Expr) bool {
|
fn (g &Gen) expr_is_multi_return_call(expr ast.Expr) bool {
|
||||||
match expr {
|
match expr {
|
||||||
|
|
|
||||||
|
|
@ -0,0 +1,421 @@
|
||||||
|
// Copyright (c) 2019-2021 Alexander Medvednikov. All rights reserved.
|
||||||
|
// Use of this source code is governed by an MIT license
|
||||||
|
// that can be found in the LICENSE file.
|
||||||
|
module c
|
||||||
|
|
||||||
|
import v.ast
|
||||||
|
import v.table
|
||||||
|
|
||||||
|
fn (mut g Gen) index_expr(node ast.IndexExpr) {
|
||||||
|
if node.index is ast.RangeExpr {
|
||||||
|
g.range_expr(node, node.index)
|
||||||
|
} else {
|
||||||
|
sym := g.table.get_final_type_symbol(node.left_type)
|
||||||
|
if sym.kind == .array {
|
||||||
|
g.index_of_array(node, sym)
|
||||||
|
} else if sym.kind == .array_fixed {
|
||||||
|
g.index_of_fixed_array(node, sym)
|
||||||
|
} else if sym.kind == .map {
|
||||||
|
g.index_of_map(node, sym)
|
||||||
|
} else if sym.kind == .string && !node.left_type.is_ptr() {
|
||||||
|
g.write('string_at(')
|
||||||
|
g.expr(node.left)
|
||||||
|
g.write(', ')
|
||||||
|
g.expr(node.index)
|
||||||
|
g.write(')')
|
||||||
|
} else {
|
||||||
|
g.expr(node.left)
|
||||||
|
g.write('[')
|
||||||
|
g.expr(node.index)
|
||||||
|
g.write(']')
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
fn (mut g Gen) range_expr(node ast.IndexExpr, range ast.RangeExpr) {
|
||||||
|
sym := g.table.get_type_symbol(node.left_type)
|
||||||
|
if sym.kind == .string {
|
||||||
|
g.write('string_substr(')
|
||||||
|
g.expr(node.left)
|
||||||
|
} else if sym.kind == .array {
|
||||||
|
g.write('array_slice(')
|
||||||
|
if node.left_type.is_ptr() {
|
||||||
|
g.write('*')
|
||||||
|
}
|
||||||
|
g.expr(node.left)
|
||||||
|
} else if sym.kind == .array_fixed {
|
||||||
|
// Convert a fixed array to V array when doing `fixed_arr[start..end]`
|
||||||
|
info := sym.info as table.ArrayFixed
|
||||||
|
g.write('array_slice(new_array_from_c_array(')
|
||||||
|
g.write('$info.size')
|
||||||
|
g.write(', $info.size')
|
||||||
|
g.write(', sizeof(')
|
||||||
|
if node.left_type.is_ptr() {
|
||||||
|
g.write('(*')
|
||||||
|
}
|
||||||
|
g.expr(node.left)
|
||||||
|
if node.left_type.is_ptr() {
|
||||||
|
g.write(')')
|
||||||
|
}
|
||||||
|
g.write('[0]), ')
|
||||||
|
if node.left_type.is_ptr() {
|
||||||
|
g.write('*')
|
||||||
|
}
|
||||||
|
g.expr(node.left)
|
||||||
|
g.write(')')
|
||||||
|
} else {
|
||||||
|
g.expr(node.left)
|
||||||
|
}
|
||||||
|
g.write(', ')
|
||||||
|
if range.has_low {
|
||||||
|
g.expr(range.low)
|
||||||
|
} else {
|
||||||
|
g.write('0')
|
||||||
|
}
|
||||||
|
g.write(', ')
|
||||||
|
if range.has_high {
|
||||||
|
g.expr(range.high)
|
||||||
|
} else if sym.kind == .array_fixed {
|
||||||
|
info := sym.info as table.ArrayFixed
|
||||||
|
g.write('$info.size')
|
||||||
|
} else if node.left_type.is_ptr() {
|
||||||
|
g.write('(')
|
||||||
|
g.write('*')
|
||||||
|
g.expr(node.left)
|
||||||
|
g.write(')')
|
||||||
|
g.write('.len')
|
||||||
|
} else {
|
||||||
|
g.expr(node.left)
|
||||||
|
g.write('.len')
|
||||||
|
}
|
||||||
|
g.write(')')
|
||||||
|
}
|
||||||
|
|
||||||
|
fn (mut g Gen) index_of_array(node ast.IndexExpr, sym table.TypeSymbol) {
|
||||||
|
gen_or := node.or_expr.kind != .absent
|
||||||
|
left_is_ptr := node.left_type.is_ptr()
|
||||||
|
info := sym.info as table.Array
|
||||||
|
elem_type_str := g.typ(info.elem_type)
|
||||||
|
elem_type := info.elem_type
|
||||||
|
elem_typ := g.table.get_type_symbol(elem_type)
|
||||||
|
// `vals[i].field = x` is an exception and requires `array_get`:
|
||||||
|
// `(*(Val*)array_get(vals, i)).field = x;`
|
||||||
|
is_selector := node.left is ast.SelectorExpr
|
||||||
|
if g.is_assign_lhs && !is_selector && node.is_setter {
|
||||||
|
is_direct_array_access := g.fn_decl != 0 && g.fn_decl.is_direct_arr
|
||||||
|
is_op_assign := g.assign_op != .assign && info.elem_type != table.string_type
|
||||||
|
array_ptr_type_str := match elem_typ.kind {
|
||||||
|
.function { 'voidptr*' }
|
||||||
|
else { '$elem_type_str*' }
|
||||||
|
}
|
||||||
|
if is_direct_array_access {
|
||||||
|
g.write('(($array_ptr_type_str)')
|
||||||
|
} else if is_op_assign {
|
||||||
|
g.write('(*($array_ptr_type_str)array_get(')
|
||||||
|
if left_is_ptr && !node.left_type.has_flag(.shared_f) {
|
||||||
|
g.write('*')
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
g.is_array_set = true // special handling of assign_op and closing with '})'
|
||||||
|
g.write('array_set(')
|
||||||
|
if !left_is_ptr || node.left_type.has_flag(.shared_f) {
|
||||||
|
g.write('&')
|
||||||
|
}
|
||||||
|
}
|
||||||
|
g.expr(node.left)
|
||||||
|
if node.left_type.has_flag(.shared_f) {
|
||||||
|
if left_is_ptr {
|
||||||
|
g.write('->val')
|
||||||
|
} else {
|
||||||
|
g.write('.val')
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if is_direct_array_access {
|
||||||
|
if left_is_ptr && !node.left_type.has_flag(.shared_f) {
|
||||||
|
g.write('->')
|
||||||
|
} else {
|
||||||
|
g.write('.')
|
||||||
|
}
|
||||||
|
g.write('data)[')
|
||||||
|
g.expr(node.index)
|
||||||
|
g.write(']')
|
||||||
|
} else {
|
||||||
|
g.write(', ')
|
||||||
|
g.expr(node.index)
|
||||||
|
if !is_op_assign {
|
||||||
|
mut need_wrapper := true
|
||||||
|
/*
|
||||||
|
match node.right {
|
||||||
|
ast.EnumVal, ast.Ident {
|
||||||
|
// `&x` is enough for variables and enums
|
||||||
|
// `&(Foo[]){ ... }` is only needed for function calls and literals
|
||||||
|
need_wrapper = false
|
||||||
|
}
|
||||||
|
else {}
|
||||||
|
}
|
||||||
|
*/
|
||||||
|
if need_wrapper {
|
||||||
|
if elem_typ.kind == .function {
|
||||||
|
g.write(', &(voidptr[]) { ')
|
||||||
|
} else {
|
||||||
|
g.write(', &($elem_type_str[]) { ')
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
g.write(', &')
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
// `x[0] *= y`
|
||||||
|
g.write('))')
|
||||||
|
}
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
is_direct_array_access := g.fn_decl != 0 && g.fn_decl.is_direct_arr
|
||||||
|
array_ptr_type_str := match elem_typ.kind {
|
||||||
|
.function { 'voidptr*' }
|
||||||
|
else { '$elem_type_str*' }
|
||||||
|
}
|
||||||
|
needs_clone := info.elem_type == table.string_type_idx && g.is_autofree && !g.is_assign_lhs
|
||||||
|
is_gen_or_and_assign_rhs := gen_or && !g.discard_or_result
|
||||||
|
cur_line := if is_gen_or_and_assign_rhs {
|
||||||
|
line := g.go_before_stmt(0)
|
||||||
|
g.out.write_string(tabs[g.indent])
|
||||||
|
line
|
||||||
|
} else {
|
||||||
|
''
|
||||||
|
}
|
||||||
|
tmp_opt := if gen_or { g.new_tmp_var() } else { '' }
|
||||||
|
tmp_opt_ptr := if gen_or { g.new_tmp_var() } else { '' }
|
||||||
|
if gen_or {
|
||||||
|
g.write('$array_ptr_type_str $tmp_opt_ptr = ($array_ptr_type_str)/*ee elem_ptr_typ */(array_get_with_check(')
|
||||||
|
} else {
|
||||||
|
if needs_clone {
|
||||||
|
g.write('/*2*/string_clone(')
|
||||||
|
}
|
||||||
|
if g.is_fn_index_call {
|
||||||
|
if elem_typ.info is table.FnType {
|
||||||
|
g.write('((')
|
||||||
|
g.write_fn_ptr_decl(&elem_typ.info, '')
|
||||||
|
g.write(')(*($array_ptr_type_str)/*ee elem_typ */array_get(')
|
||||||
|
}
|
||||||
|
if left_is_ptr && !node.left_type.has_flag(.shared_f) {
|
||||||
|
g.write('*')
|
||||||
|
}
|
||||||
|
} else if is_direct_array_access {
|
||||||
|
g.write('(($array_ptr_type_str)')
|
||||||
|
} else {
|
||||||
|
g.write('(*($array_ptr_type_str)/*ee elem_typ */array_get(')
|
||||||
|
if left_is_ptr && !node.left_type.has_flag(.shared_f) {
|
||||||
|
g.write('*')
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
g.expr(node.left)
|
||||||
|
// TODO: test direct_array_access when 'shared' is implemented
|
||||||
|
if node.left_type.has_flag(.shared_f) {
|
||||||
|
if left_is_ptr {
|
||||||
|
g.write('->val')
|
||||||
|
} else {
|
||||||
|
g.write('.val')
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if is_direct_array_access && !gen_or {
|
||||||
|
if left_is_ptr && !node.left_type.has_flag(.shared_f) {
|
||||||
|
g.write('->')
|
||||||
|
} else {
|
||||||
|
g.write('.')
|
||||||
|
}
|
||||||
|
g.write('data)[')
|
||||||
|
g.expr(node.index)
|
||||||
|
g.write(']')
|
||||||
|
} else {
|
||||||
|
g.write(', ')
|
||||||
|
g.expr(node.index)
|
||||||
|
if g.is_fn_index_call {
|
||||||
|
g.write(')))')
|
||||||
|
} else {
|
||||||
|
g.write('))')
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if needs_clone {
|
||||||
|
g.write(')')
|
||||||
|
}
|
||||||
|
if gen_or {
|
||||||
|
g.writeln(';')
|
||||||
|
opt_elem_type := g.typ(elem_type.set_flag(.optional))
|
||||||
|
g.writeln('$opt_elem_type $tmp_opt = {0};')
|
||||||
|
g.writeln('if ($tmp_opt_ptr) {')
|
||||||
|
g.writeln('\t*(($elem_type_str*)&${tmp_opt}.data) = *(($elem_type_str*)$tmp_opt_ptr);')
|
||||||
|
g.writeln('} else {')
|
||||||
|
g.writeln('\t${tmp_opt}.state = 2; ${tmp_opt}.err = (Error){.msg=_SLIT("array index out of range"), .code=0};')
|
||||||
|
g.writeln('}')
|
||||||
|
g.or_block(tmp_opt, node.or_expr, elem_type)
|
||||||
|
g.write('\n$cur_line*($elem_type_str*)${tmp_opt}.data')
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
fn (mut g Gen) index_of_fixed_array(node ast.IndexExpr, sym table.TypeSymbol) {
|
||||||
|
info := sym.info as table.ArrayFixed
|
||||||
|
elem_type := info.elem_type
|
||||||
|
elem_sym := g.table.get_type_symbol(elem_type)
|
||||||
|
is_fn_index_call := g.is_fn_index_call && elem_sym.info is table.FnType
|
||||||
|
|
||||||
|
if is_fn_index_call {
|
||||||
|
g.write('(*')
|
||||||
|
}
|
||||||
|
if node.left_type.is_ptr() {
|
||||||
|
g.write('(*')
|
||||||
|
g.expr(node.left)
|
||||||
|
g.write(')')
|
||||||
|
} else {
|
||||||
|
g.expr(node.left)
|
||||||
|
}
|
||||||
|
g.write('[')
|
||||||
|
direct := g.fn_decl != 0 && g.fn_decl.is_direct_arr
|
||||||
|
if direct || node.index is ast.IntegerLiteral {
|
||||||
|
g.expr(node.index)
|
||||||
|
} else {
|
||||||
|
// bounds check
|
||||||
|
g.write('v_fixed_index(')
|
||||||
|
g.expr(node.index)
|
||||||
|
g.write(', $info.size)')
|
||||||
|
}
|
||||||
|
g.write(']')
|
||||||
|
if is_fn_index_call {
|
||||||
|
g.write(')')
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
fn (mut g Gen) index_of_map(node ast.IndexExpr, sym table.TypeSymbol) {
|
||||||
|
gen_or := node.or_expr.kind != .absent
|
||||||
|
left_is_ptr := node.left_type.is_ptr()
|
||||||
|
info := sym.info as table.Map
|
||||||
|
key_type_str := g.typ(info.key_type)
|
||||||
|
elem_type := info.value_type
|
||||||
|
elem_type_str := g.typ(elem_type)
|
||||||
|
elem_typ := g.table.get_type_symbol(elem_type)
|
||||||
|
get_and_set_types := elem_typ.kind in [.struct_, .map]
|
||||||
|
if g.is_assign_lhs && !g.is_array_set && !get_and_set_types {
|
||||||
|
if g.assign_op == .assign || info.value_type == table.string_type {
|
||||||
|
g.is_array_set = true
|
||||||
|
g.write('map_set_1(')
|
||||||
|
} else {
|
||||||
|
if node.is_setter {
|
||||||
|
g.write('(*(($elem_type_str*)map_get_and_set_1(')
|
||||||
|
} else {
|
||||||
|
g.write('(*(($elem_type_str*)map_get_1(')
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if !left_is_ptr || node.left_type.has_flag(.shared_f) {
|
||||||
|
g.write('&')
|
||||||
|
}
|
||||||
|
if node.left is ast.IndexExpr {
|
||||||
|
g.inside_map_index = true
|
||||||
|
g.expr(node.left)
|
||||||
|
g.inside_map_index = false
|
||||||
|
} else {
|
||||||
|
g.expr(node.left)
|
||||||
|
}
|
||||||
|
if node.left_type.has_flag(.shared_f) {
|
||||||
|
if left_is_ptr {
|
||||||
|
g.write('->val')
|
||||||
|
} else {
|
||||||
|
g.write('.val')
|
||||||
|
}
|
||||||
|
}
|
||||||
|
g.write(', &($key_type_str[]){')
|
||||||
|
g.expr(node.index)
|
||||||
|
g.write('}')
|
||||||
|
if elem_typ.kind == .function {
|
||||||
|
g.write(', &(voidptr[]) { ')
|
||||||
|
} else {
|
||||||
|
g.array_set_pos = g.out.len
|
||||||
|
g.write(', &($elem_type_str[]) { ')
|
||||||
|
}
|
||||||
|
if g.assign_op != .assign && info.value_type != table.string_type {
|
||||||
|
zero := g.type_default(info.value_type)
|
||||||
|
g.write('$zero })))')
|
||||||
|
}
|
||||||
|
} else if g.inside_map_postfix || g.inside_map_infix || g.inside_map_index
|
||||||
|
|| (g.is_assign_lhs && !g.is_array_set && get_and_set_types) {
|
||||||
|
zero := g.type_default(info.value_type)
|
||||||
|
if node.is_setter {
|
||||||
|
g.write('(*($elem_type_str*)map_get_and_set_1(')
|
||||||
|
} else {
|
||||||
|
g.write('(*($elem_type_str*)map_get_1(')
|
||||||
|
}
|
||||||
|
if !left_is_ptr {
|
||||||
|
g.write('&')
|
||||||
|
}
|
||||||
|
g.expr(node.left)
|
||||||
|
g.write(', &($key_type_str[]){')
|
||||||
|
g.expr(node.index)
|
||||||
|
g.write('}, &($elem_type_str[]){ $zero }))')
|
||||||
|
} else {
|
||||||
|
zero := g.type_default(info.value_type)
|
||||||
|
is_gen_or_and_assign_rhs := gen_or && !g.discard_or_result
|
||||||
|
cur_line := if is_gen_or_and_assign_rhs {
|
||||||
|
line := g.go_before_stmt(0)
|
||||||
|
g.out.write_string(tabs[g.indent])
|
||||||
|
line
|
||||||
|
} else {
|
||||||
|
''
|
||||||
|
}
|
||||||
|
tmp_opt := if gen_or { g.new_tmp_var() } else { '' }
|
||||||
|
tmp_opt_ptr := if gen_or { g.new_tmp_var() } else { '' }
|
||||||
|
if gen_or {
|
||||||
|
g.write('$elem_type_str* $tmp_opt_ptr = ($elem_type_str*)/*ee elem_ptr_typ */(map_get_1_check(')
|
||||||
|
} else {
|
||||||
|
if g.is_fn_index_call {
|
||||||
|
if elem_typ.info is table.FnType {
|
||||||
|
g.write('((')
|
||||||
|
g.write_fn_ptr_decl(&elem_typ.info, '')
|
||||||
|
g.write(')(*(voidptr*)map_get_1(')
|
||||||
|
}
|
||||||
|
} else if elem_typ.kind == .function {
|
||||||
|
g.write('(*(voidptr*)map_get_1(')
|
||||||
|
} else {
|
||||||
|
g.write('(*($elem_type_str*)map_get_1(')
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if !left_is_ptr || node.left_type.has_flag(.shared_f) {
|
||||||
|
g.write('ADDR(map, ')
|
||||||
|
g.expr(node.left)
|
||||||
|
} else {
|
||||||
|
g.write('(')
|
||||||
|
g.expr(node.left)
|
||||||
|
}
|
||||||
|
if node.left_type.has_flag(.shared_f) {
|
||||||
|
if left_is_ptr {
|
||||||
|
g.write('->val')
|
||||||
|
} else {
|
||||||
|
g.write('.val')
|
||||||
|
}
|
||||||
|
}
|
||||||
|
g.write('), &($key_type_str[]){')
|
||||||
|
g.expr(node.index)
|
||||||
|
g.write('}')
|
||||||
|
if gen_or {
|
||||||
|
g.write('))')
|
||||||
|
} else if g.is_fn_index_call {
|
||||||
|
g.write(', &(voidptr[]){ $zero })))')
|
||||||
|
} else if elem_typ.kind == .function {
|
||||||
|
g.write(', &(voidptr[]){ $zero }))')
|
||||||
|
} else {
|
||||||
|
g.write(', &($elem_type_str[]){ $zero }))')
|
||||||
|
}
|
||||||
|
if gen_or {
|
||||||
|
g.writeln(';')
|
||||||
|
opt_elem_type := g.typ(elem_type.set_flag(.optional))
|
||||||
|
g.writeln('$opt_elem_type $tmp_opt = {0};')
|
||||||
|
g.writeln('if ($tmp_opt_ptr) {')
|
||||||
|
g.writeln('\t*(($elem_type_str*)&${tmp_opt}.data) = *(($elem_type_str*)$tmp_opt_ptr);')
|
||||||
|
g.writeln('} else {')
|
||||||
|
g.writeln('\t${tmp_opt}.state = 2; ${tmp_opt}.err = (Error){.msg=_SLIT("array index out of range"), .code=0};')
|
||||||
|
|
||||||
|
g.writeln('}')
|
||||||
|
g.or_block(tmp_opt, node.or_expr, elem_type)
|
||||||
|
g.write('\n$cur_line*($elem_type_str*)${tmp_opt}.data')
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
Loading…
Reference in New Issue