v/vlib/v/gen/c/fn.v

1213 lines
37 KiB
V
Raw Normal View History

// 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
2020-04-22 20:20:49 +02:00
import v.ast
import v.table
import v.util
fn (mut g Gen) is_used_by_main(node ast.FnDecl) bool {
mut is_used_by_main := true
if g.pref.skip_unused {
fkey := if node.is_method { '${int(node.receiver.typ)}.$node.name' } else { node.name }
is_used_by_main = g.table.used_fns[fkey]
$if trace_skip_unused_fns ? {
println('> is_used_by_main: $is_used_by_main | node.name: $node.name | fkey: $fkey | node.is_method: $node.is_method')
}
if !is_used_by_main {
$if trace_skip_unused_fns_in_c_code ? {
g.writeln('// trace_skip_unused_fns_in_c_code, $node.name, fkey: $fkey')
}
}
} else {
$if trace_skip_unused_fns_in_c_code ? {
fkey := if node.is_method { '${int(node.receiver.typ)}.$node.name' } else { node.name }
g.writeln('// trace_skip_unused_fns_in_c_code, $node.name, fkey: $fkey')
}
}
return is_used_by_main
}
fn (mut g Gen) process_fn_decl(node ast.FnDecl) {
if !g.is_used_by_main(node) {
return
}
g.gen_attrs(node.attrs)
// g.tmp_count = 0 TODO
mut skip := false
pos := g.out.buf.len
should_bundle_module := util.should_bundle_module(node.mod)
if g.pref.build_mode == .build_module {
// if node.name.contains('parse_text') {
// println('!!! $node.name mod=$node.mod, built=$g.module_built')
// }
// TODO true for not just "builtin"
// TODO: clean this up
mod := if g.is_builtin_mod { 'builtin' } else { node.name.all_before_last('.') }
if (mod != g.module_built && node.mod != g.module_built.after('/')) || should_bundle_module {
// Skip functions that don't have to be generated for this module.
// println('skip bm $node.name mod=$node.mod module_built=$g.module_built')
skip = true
}
if g.is_builtin_mod && g.module_built == 'builtin' && node.mod == 'builtin' {
skip = false
}
if !skip && g.pref.is_verbose {
println('build module `$g.module_built` fn `$node.name`')
}
}
if g.pref.use_cache {
// We are using prebuilt modules, we do not need to generate
// their functions in main.c.
if node.mod != 'main' && node.mod != 'help' && !should_bundle_module && !g.pref.is_test
&& node.generic_params.len == 0 {
skip = true
}
}
keep_fn_decl := g.fn_decl
g.fn_decl = &node
if node.name == 'main.main' {
g.has_main = true
}
if node.name == 'backtrace' || node.name == 'backtrace_symbols'
|| node.name == 'backtrace_symbols_fd' {
g.write('\n#ifndef __cplusplus\n')
}
g.gen_fn_decl(node, skip)
if node.name == 'backtrace' || node.name == 'backtrace_symbols'
|| node.name == 'backtrace_symbols_fd' {
g.write('\n#endif\n')
}
g.fn_decl = keep_fn_decl
if skip {
g.out.go_back_to(pos)
}
if node.language != .c {
g.writeln('')
}
}
fn (mut g Gen) gen_fn_decl(node ast.FnDecl, skip bool) {
// TODO For some reason, build fails with autofree with this line
// as it's only informative, comment it for now
// g.gen_attrs(it.attrs)
if node.language == .c {
// || node.no_body {
return
}
2021-02-05 08:05:13 +01:00
// Skip [if xxx] if xxx is not defined
/*
2021-02-05 08:05:13 +01:00
for attr in node.attrs {
if !attr.is_comptime_define {
continue
}
if attr.name !in g.pref.compile_defines_all {
// println('skipping [if]')
return
}
}
*/
2021-02-05 08:05:13 +01:00
g.returned_var_name = ''
//
old_g_autofree := g.is_autofree
if node.is_manualfree {
g.is_autofree = false
}
defer {
g.is_autofree = old_g_autofree
}
//
2020-06-24 14:56:44 +02:00
// if g.fileis('vweb.v') {
// println('\ngen_fn_decl() $node.name $node.is_generic $g.cur_generic_type')
2020-06-24 14:56:44 +02:00
// }
if node.generic_params.len > 0 && g.cur_generic_types.len == 0 { // need the cur_generic_type check to avoid inf. recursion
2020-05-21 03:58:50 +02:00
// loop thru each generic type and generate a function
for gen_types in g.table.fn_gen_types[node.name] {
if g.pref.is_verbose {
syms := gen_types.map(g.table.get_type_symbol(it))
println('gen fn `$node.name` for type `${syms.map(node.name).join(', ')}`')
}
g.cur_generic_types = gen_types
g.gen_fn_decl(node, skip)
2020-05-21 03:58:50 +02:00
}
g.cur_generic_types = []
2020-05-21 03:58:50 +02:00
return
}
cur_fn_save := g.cur_fn
defer {
g.cur_fn = cur_fn_save
}
g.cur_fn = node
2020-05-01 19:34:27 +02:00
fn_start_pos := g.out.len
g.write_v_source_line_info(node.pos)
msvc_attrs := g.write_fn_attrs(node.attrs)
2020-06-04 12:20:14 +02:00
// Live
is_livefn := node.attrs.contains('live')
is_livemain := g.pref.is_livemain && is_livefn
2020-05-01 19:34:27 +02:00
is_liveshared := g.pref.is_liveshared && is_livefn
is_livemode := g.pref.is_livemain || g.pref.is_liveshared
is_live_wrap := is_livefn && is_livemode
2020-05-01 19:34:27 +02:00
if is_livefn && !is_livemode {
eprintln('INFO: compile with `v -live $g.pref.path `, if you want to use the [live] function $node.name .')
2020-05-01 19:34:27 +02:00
}
//
mut name := node.name
if name in ['+', '-', '*', '/', '%', '<', '=='] {
name = util.replace_op(name)
}
if node.is_method {
name = g.cc_type(node.receiver.typ, false) + '_' + name
// name = g.table.get_type_symbol(node.receiver.typ).name + '_' + name
}
if node.language == .c {
name = util.no_dots(name)
} else {
name = c_name(name)
}
mut type_name := g.typ(node.return_type)
if g.cur_generic_types.len > 0 {
// foo<T>() => foo_T_int(), foo_T_string() etc
// Using _T_ to differentiate between get<string> and get_string
name += '_T'
for generic_type in g.cur_generic_types {
gen_name := g.typ(generic_type)
name += '_' + gen_name
}
}
2021-02-02 09:14:34 +01:00
if g.pref.obfuscate && g.cur_mod.name == 'main' && name.starts_with('main__')
&& name != 'main__main' && node.name != 'str' {
2021-02-02 09:14:34 +01:00
mut key := node.name
if node.is_method {
sym := g.table.get_type_symbol(node.receiver.typ)
key = sym.name + '.' + node.name
}
g.writeln('/* obf: $key */')
name = g.obf_table[key] or {
panic('cgen: fn_decl: obf name "$key" not found, this should never happen')
}
}
// if g.pref.show_cc && it.is_builtin {
// println(name)
// }
// type_name := g.table.Type_to_str(it.return_type)
// Live functions are protected by a mutex, because otherwise they
// can be changed by the live reload thread, *while* they are
// running, with unpredictable results (usually just crashing).
// For this purpose, the actual body of the live function,
// is put under a non publicly accessible function, that is prefixed
// with 'impl_live_' .
if is_livemain {
g.hotcode_fn_names << name
}
mut impl_fn_name := name
if is_live_wrap {
impl_fn_name = 'impl_live_$name'
}
last_fn_c_name_save := g.last_fn_c_name
defer {
g.last_fn_c_name = last_fn_c_name_save
}
g.last_fn_c_name = impl_fn_name
//
if is_live_wrap {
2020-05-01 19:34:27 +02:00
if is_livemain {
g.definitions.write_string('$type_name (* $impl_fn_name)(')
g.write('$type_name no_impl_${name}(')
2020-05-01 19:34:27 +02:00
}
if is_liveshared {
g.definitions.write_string('$type_name ${impl_fn_name}(')
g.write('$type_name ${impl_fn_name}(')
2020-05-01 19:34:27 +02:00
}
} else {
if !(node.is_pub || g.pref.is_debug) {
2020-07-15 21:36:06 +02:00
// Private functions need to marked as static so that they are not exportable in the
// binaries
if g.pref.build_mode != .build_module && !g.pref.use_cache {
2020-07-15 21:36:06 +02:00
// if !(g.pref.build_mode == .build_module && g.is_builtin_mod) {
// If we are building vlib/builtin, we need all private functions like array_get
// to be public, so that all V programs can access them.
g.write('VV_LOCAL_SYMBOL ')
g.definitions.write_string('VV_LOCAL_SYMBOL ')
2020-07-15 21:36:06 +02:00
}
}
fn_header := if msvc_attrs.len > 0 {
'$type_name $msvc_attrs ${name}('
} else {
'$type_name ${name}('
}
g.definitions.write_string(fn_header)
g.write(fn_header)
}
arg_start_pos := g.out.len
fargs, fargtypes := g.fn_args(node.params, node.is_variadic)
arg_str := g.out.after(arg_start_pos)
if node.no_body || ((g.pref.use_cache && g.pref.build_mode != .build_module) && node.is_builtin
&& !g.is_test) || skip {
// Just a function header. Builtin function bodies are defined in builtin.o
2020-07-20 22:51:17 +02:00
g.definitions.writeln(');') // // NO BODY')
g.writeln(');')
return
}
g.definitions.writeln(');')
g.writeln(') {')
for defer_stmt in node.defer_stmts {
g.writeln('bool ${g.defer_flag_var(defer_stmt)} = false;')
}
if is_live_wrap {
// The live function just calls its implementation dual, while ensuring
// that the call is wrapped by the mutex lock & unlock calls.
// Adding the mutex lock/unlock inside the body of the implementation
// function is not reliable, because the implementation function can do
// an early exit, which will leave the mutex locked.
mut fn_args_list := []string{}
for ia, fa in fargs {
fn_args_list << '${fargtypes[ia]} $fa'
}
mut live_fncall := '${impl_fn_name}(' + fargs.join(', ') + ');'
mut live_fnreturn := ''
if type_name != 'void' {
live_fncall = '$type_name res = $live_fncall'
live_fnreturn = 'return res;'
}
g.definitions.writeln('$type_name ${name}(' + fn_args_list.join(', ') + ');')
g.hotcode_definitions.writeln('$type_name ${name}(' + fn_args_list.join(', ') + '){')
g.hotcode_definitions.writeln(' pthread_mutex_lock(&live_fn_mutex);')
g.hotcode_definitions.writeln(' $live_fncall')
g.hotcode_definitions.writeln(' pthread_mutex_unlock(&live_fn_mutex);')
g.hotcode_definitions.writeln(' $live_fnreturn')
g.hotcode_definitions.writeln('}')
2020-05-01 19:34:27 +02:00
}
2020-04-25 17:49:16 +02:00
// Profiling mode? Start counting at the beginning of the function (save current time).
if g.pref.is_prof && g.pref.build_mode != .build_module {
g.profile_fn(node)
2020-06-03 17:24:08 +02:00
}
// we could be in an anon fn so save outer fn defer stmts
prev_defer_stmts := g.defer_stmts
g.defer_stmts = []
g.stmts(node.stmts)
// clear g.fn_mut_arg_names
if !node.has_return {
2020-07-08 07:39:11 +02:00
g.write_defer_stmts_when_needed()
}
if node.is_anon {
g.defer_stmts = prev_defer_stmts
} else {
g.defer_stmts = []
}
if node.return_type != table.void_type && node.stmts.len > 0 && node.stmts.last() !is ast.Return {
default_expr := g.type_default(node.return_type)
2020-11-06 15:26:59 +01:00
// TODO: perf?
if default_expr == '{0}' {
// if node.return_type.idx() == 1 && node.return_type.has_flag(.optional) {
// // The default return for anonymous functions that return `?,
// // should have .ok = true set, otherwise calling them with
// // optfn() or { panic(err) } will cause a panic:
// g.writeln('\treturn (Option_void){0};')
// } else {
g.writeln('\treturn ($type_name)$default_expr;')
// }
2020-11-06 15:26:59 +01:00
} else {
g.writeln('\treturn $default_expr;')
}
}
g.writeln('}')
2020-05-01 19:34:27 +02:00
if g.pref.printfn_list.len > 0 && g.last_fn_c_name in g.pref.printfn_list {
println(g.out.after(fn_start_pos))
}
for attr in node.attrs {
if attr.name == 'export' {
g.writeln('// export alias: $attr.arg -> $name')
export_alias := '$type_name ${attr.arg}($arg_str)'
g.definitions.writeln('VV_EXPORTED_SYMBOL $export_alias; // exported fn $node.name')
g.writeln('$export_alias {')
g.write('\treturn ${name}(')
g.write(fargs.join(', '))
g.writeln(');')
g.writeln('}')
}
}
}
fn (g &Gen) defer_flag_var(stmt &ast.DeferStmt) string {
return '${g.last_fn_c_name}_defer_$stmt.idx_in_fn'
}
2020-04-25 17:49:16 +02:00
fn (mut g Gen) write_defer_stmts_when_needed() {
if g.defer_stmts.len > 0 {
g.write_defer_stmts()
}
2020-04-25 12:05:31 +02:00
if g.defer_profile_code.len > 0 {
g.writeln('')
g.writeln('\t// defer_profile_code')
2020-04-25 17:49:16 +02:00
g.writeln(g.defer_profile_code)
2020-04-25 12:05:31 +02:00
g.writeln('')
}
}
2020-06-04 12:20:14 +02:00
// fn decl args
2020-09-09 13:21:11 +02:00
fn (mut g Gen) fn_args(args []table.Param, is_variadic bool) ([]string, []string) {
2020-05-01 19:34:27 +02:00
mut fargs := []string{}
mut fargtypes := []string{}
if args.len == 0 {
// in C, `()` is untyped, unlike `(void)`
g.write('void')
}
for i, arg in args {
caname := if arg.name == '_' { g.new_tmp_var() } else { c_name(arg.name) }
typ := g.unwrap_generic(arg.typ)
arg_type_sym := g.table.get_type_symbol(typ)
mut arg_type_name := g.typ(typ) // util.no_dots(arg_type_sym.name)
if arg_type_sym.kind == .function {
info := arg_type_sym.info as table.FnType
func := info.func
if !info.is_anon {
g.write(arg_type_name + ' ' + caname)
g.definitions.write_string(arg_type_name + ' ' + caname)
fargs << caname
2020-05-01 19:34:27 +02:00
fargtypes << arg_type_name
} else {
g.write('${g.typ(func.return_type)} (*$caname)(')
g.definitions.write_string('${g.typ(func.return_type)} (*$caname)(')
2020-09-27 03:32:56 +02:00
g.fn_args(func.params, func.is_variadic)
g.write(')')
g.definitions.write_string(')')
}
} else {
s := '$arg_type_name $caname'
g.write(s)
g.definitions.write_string(s)
fargs << caname
2020-05-01 19:34:27 +02:00
fargtypes << arg_type_name
}
if i < args.len - 1 {
g.write(', ')
g.definitions.write_string(', ')
}
}
2020-05-01 19:34:27 +02:00
return fargs, fargtypes
}
2020-04-14 00:37:47 +02:00
2020-04-22 20:20:49 +02:00
fn (mut g Gen) call_expr(node ast.CallExpr) {
2021-02-27 15:11:17 +01:00
// g.write('/*call expr*/')
// NOTE: everything could be done this way
// see my comment in parser near anon_fn
if node.left is ast.AnonFn {
g.expr(node.left)
}
if node.left is ast.IndexExpr && node.name == '' {
g.is_fn_index_call = true
g.expr(node.left)
g.is_fn_index_call = false
}
if node.should_be_skipped {
return
}
2020-06-29 20:20:31 +02:00
g.inside_call = true
defer {
g.inside_call = false
}
2021-02-27 15:11:17 +01:00
gen_or := node.or_block.kind != .absent // && !g.is_autofree
is_gen_or_and_assign_rhs := gen_or && !g.discard_or_result
2021-02-27 15:11:17 +01:00
cur_line := if is_gen_or_and_assign_rhs { // && !g.is_autofree {
// `x := foo() or { ...}`
// cut everything that has been generated to prepend optional variable creation
2020-05-21 22:35:43 +02:00
line := g.go_before_stmt(0)
g.out.write_string(tabs[g.indent])
2021-02-27 15:11:17 +01:00
// g.write('/*is_gen_or_and_assign_rhs*/')
2020-05-21 22:35:43 +02:00
line
} else {
''
}
2020-04-14 00:37:47 +02:00
tmp_opt := if gen_or { g.new_tmp_var() } else { '' }
if gen_or {
styp := g.typ(node.return_type.set_flag(.optional))
2020-04-14 00:37:47 +02:00
g.write('$styp $tmp_opt = ')
}
if node.is_method && !node.is_field {
if node.name == 'writeln' && g.pref.experimental && node.args.len > 0
&& node.args[0].expr is ast.StringInterLiteral
&& g.table.get_type_symbol(node.receiver_type).name == 'strings.Builder' {
g.string_inter_literal_sb_optimized(node)
} else {
g.method_call(node)
}
2020-04-14 00:37:47 +02:00
} else {
g.fn_call(node)
}
if gen_or { // && !g.autofree {
2021-02-27 15:11:17 +01:00
// if !g.is_autofree {
g.or_block(tmp_opt, node.or_block, node.return_type)
//}
2020-07-07 14:21:02 +02:00
if is_gen_or_and_assign_rhs {
unwrapped_typ := node.return_type.clear_flag(.optional)
unwrapped_styp := g.typ(unwrapped_typ)
if unwrapped_typ == table.void_type {
g.write('\n $cur_line')
} else if g.table.get_type_symbol(node.return_type).kind == .multi_return {
2021-02-27 15:11:17 +01:00
g.write('\n $cur_line $tmp_opt /*U*/')
} else {
if !g.inside_const {
g.write('\n $cur_line *($unwrapped_styp*)${tmp_opt}.data')
}
}
2020-07-07 14:21:02 +02:00
}
2020-04-14 00:37:47 +02:00
}
}
pub fn (g &Gen) unwrap_generic(typ table.Type) table.Type {
if typ.has_flag(.generic) {
sym := g.table.get_type_symbol(typ)
for i, generic_param in g.cur_fn.generic_params {
if generic_param.name == sym.name {
return g.cur_generic_types[i].derive(typ).clear_flag(.generic)
}
}
2020-05-28 05:50:57 +02:00
}
return typ
}
2020-04-22 20:20:49 +02:00
fn (mut g Gen) method_call(node ast.CallExpr) {
2020-04-14 00:37:47 +02:00
// TODO: there are still due to unchecked exprs (opt/some fn arg)
if node.left_type == 0 {
g.checker_bug('CallExpr.left_type is 0 in method_call', node.pos)
}
if node.receiver_type == 0 {
g.checker_bug('CallExpr.receiver_type is 0 in method_call', node.pos)
2020-04-14 00:37:47 +02:00
}
2020-05-28 05:50:57 +02:00
typ_sym := g.table.get_type_symbol(g.unwrap_generic(node.receiver_type))
mut receiver_type_name := util.no_dots(g.cc_type(g.unwrap_generic(node.receiver_type),
false))
if typ_sym.kind == .interface_ && (typ_sym.info as table.Interface).defines_method(node.name) {
// Speaker_name_table[s._interface_idx].speak(s._object)
$if debug_interface_method_call ? {
eprintln('>>> interface typ_sym.name: $typ_sym.name | receiver_type_name: $receiver_type_name')
}
g.write('${c_name(receiver_type_name)}_name_table[')
2020-04-22 20:20:49 +02:00
g.expr(node.left)
2020-05-10 21:16:22 +02:00
dot := if node.left_type.is_ptr() { '->' } else { '.' }
mname := c_name(node.name)
g.write('${dot}_interface_idx]._method_${mname}(')
2020-04-22 20:20:49 +02:00
g.expr(node.left)
2020-05-10 21:16:22 +02:00
g.write('${dot}_object')
if node.args.len > 0 {
g.write(', ')
2020-09-05 12:00:35 +02:00
// g.call_args(node.args, node.expected_arg_types) // , [])
g.call_args(node)
}
g.write(')')
2020-04-22 20:20:49 +02:00
return
}
2020-06-18 00:22:34 +02:00
left_sym := g.table.get_type_symbol(node.left_type)
if left_sym.kind == .array {
match node.name {
'filter' {
g.gen_array_filter(node)
return
}
2020-08-12 05:54:51 +02:00
'sort' {
g.gen_array_sort(node)
return
}
2020-06-18 00:22:34 +02:00
'insert' {
g.gen_array_insert(node)
return
}
'map' {
g.gen_array_map(node)
return
}
'prepend' {
g.gen_array_prepend(node)
return
}
'contains' {
g.gen_array_contains(node)
return
}
'index' {
g.gen_array_index(node)
return
}
'wait' {
g.gen_array_wait(node)
return
}
2020-06-18 00:22:34 +02:00
else {}
}
2020-04-14 00:37:47 +02:00
}
if left_sym.kind == .sum_type && node.name == 'type_name' {
g.write('tos3( /* $left_sym.name */ v_typeof_sumtype_${typ_sym.cname}( (')
g.expr(node.left)
dot := if node.left_type.is_ptr() { '->' } else { '.' }
g.write(')${dot}typ ))')
return
}
if left_sym.kind == .interface_ && node.name == 'type_name' {
g.write('tos3( /* $left_sym.name */ v_typeof_interface_${typ_sym.cname}( (')
g.expr(node.left)
dot := if node.left_type.is_ptr() { '->' } else { '.' }
g.write(')${dot}_interface_idx ))')
return
}
if node.name == 'str' {
mut rec_type := node.receiver_type
if rec_type.has_flag(.shared_f) {
rec_type = rec_type.clear_flag(.shared_f).set_nr_muls(0)
}
g.gen_str_for_type(rec_type)
}
2020-12-02 04:35:00 +01:00
mut has_cast := false
2021-02-12 01:02:33 +01:00
if left_sym.kind == .map && node.name in ['clone', 'move'] {
2020-12-17 08:44:50 +01:00
receiver_type_name = 'map'
}
2020-04-14 00:37:47 +02:00
// TODO performance, detect `array` method differently
if left_sym.kind == .array
&& node.name in ['repeat', 'sort_with_compare', 'free', 'push_many', 'trim', 'first', 'last', 'pop', 'clone', 'reverse', 'slice'] {
2020-04-14 00:37:47 +02:00
// && rec_sym.name == 'array' {
// && rec_sym.name == 'array' && receiver_name.starts_with('array') {
// `array_byte_clone` => `array_clone`
2020-05-02 00:28:42 +02:00
receiver_type_name = 'array'
2020-07-14 18:55:44 +02:00
if node.name in ['last', 'first', 'pop'] {
2020-04-14 00:37:47 +02:00
return_type_str := g.typ(node.return_type)
2020-12-02 04:35:00 +01:00
has_cast = true
g.write('(*($return_type_str*)')
2020-04-14 00:37:47 +02:00
}
}
mut name := util.no_dots('${receiver_type_name}_$node.name')
if left_sym.kind == .chan {
if node.name in ['close', 'try_pop', 'try_push'] {
name = 'sync__Channel_$node.name'
}
} else if left_sym.kind == .map {
if node.name == 'keys' {
name = 'map_keys_1'
}
}
2021-02-02 09:14:34 +01:00
if g.pref.obfuscate && g.cur_mod.name == 'main' && name.starts_with('main__')
&& node.name != 'str' {
sym := g.table.get_type_symbol(node.receiver_type)
key := sym.name + '.' + node.name
g.write('/* obf method call: $key */')
name = g.obf_table[key] or {
panic('cgen: obf name "$key" not found, this should never happen')
}
}
2020-05-27 14:07:03 +02:00
// Check if expression is: arr[a..b].clone(), arr[a..].clone()
// if so, then instead of calling array_clone(&array_slice(...))
// call array_clone_static(array_slice(...))
mut is_range_slice := false
if node.receiver_type.is_ptr() && !node.left_type.is_ptr() {
if node.left is ast.IndexExpr {
idx := node.left.index
2020-05-27 14:07:03 +02:00
if idx is ast.RangeExpr {
// expr is arr[range].clone()
// use array_clone_static instead of array_clone
name = util.no_dots('${receiver_type_name}_${node.name}_static')
2020-05-27 14:07:03 +02:00
is_range_slice = true
}
}
}
for i, generic_type in node.generic_types {
if generic_type != table.void_type && generic_type != 0 {
// Using _T_ to differentiate between get<string> and get_string
// `foo<int>()` => `foo_T_int()`
if i == 0 {
name += '_T'
}
name += '_' + g.typ(generic_type)
}
}
// TODO2
// g.generate_tmp_autofree_arg_vars(node, name)
//
2020-04-14 00:37:47 +02:00
// if node.receiver_type != 0 {
// g.write('/*${g.typ(node.receiver_type)}*/')
// g.write('/*expr_type=${g.typ(node.left_type)} rec type=${g.typ(node.receiver_type)}*/')
// }
if !node.receiver_type.is_ptr() && node.left_type.is_ptr() && node.name == 'str' {
2020-05-21 14:09:35 +02:00
g.write('ptr_str(')
} else {
g.write('${name}(')
}
if node.receiver_type.is_ptr() && (!node.left_type.is_ptr()
|| node.from_embed_type != 0 || (node.left_type.has_flag(.shared_f) && node.name != 'str')) {
2020-04-14 00:37:47 +02:00
// The receiver is a reference, but the caller provided a value
// Add `&` automatically.
// TODO same logic in call_args()
2020-05-27 14:07:03 +02:00
if !is_range_slice {
g.write('&')
}
} else if !node.receiver_type.is_ptr() && node.left_type.is_ptr() && node.name != 'str'
&& node.from_embed_type == 0 {
if !node.left_type.has_flag(.shared_f) {
g.write('/*rec*/*')
}
} else if !is_range_slice && node.from_embed_type == 0 && node.name != 'str' {
diff := node.left_type.nr_muls() - node.receiver_type.nr_muls()
if diff < 0 {
// TODO
// g.write('&')
} else if diff > 0 {
g.write('/*diff=$diff*/')
g.write([]byte{len: diff, init: `*`}.bytestr())
}
2020-04-14 00:37:47 +02:00
}
// if node.left_type.idx() != node.receiver_type.idx() {
// println('${g.typ(node.left_type)} ${g.typ(node.receiver_type)}')
// }
if g.is_autofree && node.free_receiver && !g.inside_lambda && !g.is_builtin_mod {
2020-10-22 03:51:25 +02:00
// The receiver expression needs to be freed, use the temp var.
fn_name := node.name.replace('.', '_')
arg_name := '_arg_expr_${fn_name}_0_$node.pos.pos'
2020-10-22 03:51:25 +02:00
g.write('/*af receiver arg*/' + arg_name)
} else {
g.expr(node.left)
if node.from_embed_type != 0 {
embed_name := typ_sym.embed_name()
if node.left_type.is_ptr() {
g.write('->')
} else {
g.write('.')
}
g.write(embed_name)
}
if node.left_type.has_flag(.shared_f) {
g.write('->val')
}
2020-10-22 03:51:25 +02:00
}
2020-12-02 04:35:00 +01:00
if has_cast {
g.write(')')
}
is_variadic := node.expected_arg_types.len > 0
&& node.expected_arg_types[node.expected_arg_types.len - 1].has_flag(.variadic)
2020-04-14 00:37:47 +02:00
if node.args.len > 0 || is_variadic {
g.write(', ')
}
// /////////
/*
if name.contains('subkeys') {
println('call_args $name $node.arg_types.len')
for t in node.arg_types {
sym := g.table.get_type_symbol(t)
print('$sym.name ')
}
println('')
}
2020-04-25 17:49:16 +02:00
*/
2020-04-14 00:37:47 +02:00
// ///////
2020-09-05 12:00:35 +02:00
// g.call_args(node.args, node.expected_arg_types) // , [])
g.call_args(node)
2020-04-14 00:37:47 +02:00
g.write(')')
}
2020-04-22 20:20:49 +02:00
fn (mut g Gen) fn_call(node ast.CallExpr) {
// call struct field with fn type
// TODO: test node.left instead
// left & left_type will be `x` and `x type` in `x.fieldfn()`
// will be `0` for `foo()`
if node.left_type != 0 {
g.expr(node.left)
2020-04-25 09:08:53 +02:00
if node.left_type.is_ptr() {
g.write('->')
2020-04-22 20:20:49 +02:00
} else {
g.write('.')
}
}
2020-04-14 00:37:47 +02:00
mut name := node.name
if node.name == 'error' {
name = 'error2'
}
if node.name == 'error_with_code' {
name = 'error_with_code2'
}
is_print := name in ['print', 'println', 'eprint', 'eprintln', 'panic']
print_method := name
2020-05-04 16:46:36 +02:00
is_json_encode := name == 'json.encode'
2021-02-10 10:12:49 +01:00
is_json_encode_pretty := name == 'json.encode_pretty'
2020-05-04 16:46:36 +02:00
is_json_decode := name == 'json.decode'
2021-02-10 10:17:29 +01:00
g.is_json_fn = is_json_encode || is_json_encode_pretty || is_json_decode
2020-04-14 00:37:47 +02:00
mut json_type_str := ''
mut json_obj := ''
2020-04-14 00:37:47 +02:00
if g.is_json_fn {
json_obj = g.new_tmp_var()
mut tmp2 := ''
cur_line := g.go_before_stmt(0)
2021-02-10 10:12:49 +01:00
if is_json_encode || is_json_encode_pretty {
2020-05-04 16:46:36 +02:00
g.gen_json_for_type(node.args[0].typ)
json_type_str = g.typ(node.args[0].typ)
// `json__encode` => `json__encode_User`
2020-11-29 17:45:22 +01:00
// encode_name := c_name(name) + '_' + util.no_dots(json_type_str)
encode_name := js_enc_name(json_type_str)
g.writeln('// json.encode')
g.write('cJSON* $json_obj = ${encode_name}(')
2020-09-05 12:00:35 +02:00
// g.call_args(node.args, node.expected_arg_types) // , [])
2020-11-29 17:45:22 +01:00
if node.args[0].typ.is_ptr() {
g.write('*')
}
2020-09-05 12:00:35 +02:00
g.call_args(node)
g.writeln(');')
tmp2 = g.new_tmp_var()
2021-02-10 10:12:49 +01:00
if is_json_encode {
g.writeln('string $tmp2 = json__json_print($json_obj);')
} else {
g.writeln('string $tmp2 = json__json_print_pretty($json_obj);')
}
2020-05-04 16:46:36 +02:00
} else {
ast_type := node.args[0].expr as ast.Type
// `json.decode(User, s)` => json.decode_User(s)
typ := c_name(g.typ(ast_type.typ))
fn_name := c_name(name) + '_' + typ
2020-05-05 13:23:29 +02:00
g.gen_json_for_type(ast_type.typ)
g.writeln('// json.decode')
g.write('cJSON* $json_obj = json__json_parse(')
// Skip the first argument in json.decode which is a type
// its name was already used to generate the function call
2020-09-05 12:00:35 +02:00
// g.call_args(node.args[1..], node.expected_arg_types) // , [])
g.is_js_call = true
g.call_args(node)
g.is_js_call = false
g.writeln(');')
tmp2 = g.new_tmp_var()
g.writeln('Option2_$typ $tmp2 = $fn_name ($json_obj);')
2020-05-04 16:46:36 +02:00
}
if !g.is_autofree {
g.write('cJSON_Delete($json_obj); //del')
}
g.write('\n$cur_line')
name = ''
json_obj = tmp2
2020-04-14 00:37:47 +02:00
}
if node.language == .c {
2020-04-14 00:37:47 +02:00
// Skip "C."
g.is_c_call = true
name = util.no_dots(name[2..])
2020-04-14 00:37:47 +02:00
} else {
name = c_name(name)
}
2021-02-02 09:14:34 +01:00
// Obfuscate only functions in the main module for now
if g.pref.obfuscate && g.cur_mod.name == 'main' && name.starts_with('main__') {
key := node.name
g.write('/* obf call: $key */')
name = g.obf_table[key] or {
panic('cgen: obf name "$key" not found, this should never happen')
}
}
for i, generic_type in node.generic_types {
// Using _T_ to differentiate between get<string> and get_string
// `foo<int>()` => `foo_T_int()`
if i == 0 {
name += '_T'
}
name += '_' + g.typ(generic_type)
2020-05-21 03:58:50 +02:00
}
// TODO2
// cgen shouldn't modify ast nodes, this should be moved
// g.generate_tmp_autofree_arg_vars(node, name)
2020-08-01 16:36:09 +02:00
// Handle `print(x)`
mut print_auto_str := false
2020-08-27 17:54:36 +02:00
if is_print && node.args[0].typ != table.string_type { // && !free_tmp_arg_vars {
mut typ := node.args[0].typ
if typ == 0 {
g.checker_bug('print arg.typ is 0', node.pos)
}
mut sym := g.table.get_type_symbol(typ)
if mut sym.info is table.Alias {
typ = sym.info.parent_type
sym = g.table.get_type_symbol(typ)
2020-04-14 00:37:47 +02:00
}
// check if alias parent also not a string
if typ != table.string_type {
expr := node.args[0].expr
if g.is_autofree && !typ.has_flag(.optional) {
// Create a temporary variable so that the value can be freed
tmp := g.new_tmp_var()
// tmps << tmp
g.write('string $tmp = ')
g.gen_expr_to_string(expr, typ)
g.writeln('; ${c_name(print_method)}($tmp); string_free(&$tmp);')
} else {
g.write('${c_name(print_method)}(')
g.gen_expr_to_string(expr, typ)
g.write(')')
2020-04-14 00:37:47 +02:00
}
print_auto_str = true
2020-04-14 00:37:47 +02:00
}
}
if !print_auto_str {
if g.pref.is_debug && node.name == 'panic' {
paline, pafile, pamod, pafn := g.panic_debug_info(node.pos)
g.write('panic_debug($paline, tos3("$pafile"), tos3("$pamod"), tos3("$pafn"), ')
// g.call_args(node.args, node.expected_arg_types) // , [])
2020-09-05 12:00:35 +02:00
g.call_args(node)
g.write(')')
} else {
// Simple function call
// if free_tmp_arg_vars {
// g.writeln(';')
// g.write(cur_line + ' /* <== af cur line*/')
// }
g.write('${g.get_ternary_name(name)}(')
if g.is_json_fn {
g.write(json_obj)
} else {
// g.call_args(node.args, node.expected_arg_types) // , tmp_arg_vars_to_free)
g.call_args(node)
}
g.write(')')
2020-05-04 16:46:36 +02:00
}
2020-04-14 00:37:47 +02:00
}
g.is_c_call = false
g.is_json_fn = false
2020-04-14 00:37:47 +02:00
}
fn (mut g Gen) autofree_call_pregen(node ast.CallExpr) {
2020-10-18 00:48:06 +02:00
// g.writeln('// autofree_call_pregen()')
2020-09-24 19:22:16 +02:00
// Create a temporary var before fn call for each argument in order to free it (only if it's a complex expression,
// like `foo(get_string())` or `foo(a + b)`
mut free_tmp_arg_vars := g.is_autofree && !g.is_builtin_mod && node.args.len > 0
&& !node.args[0].typ.has_flag(.optional) // TODO copy pasta checker.v
if !free_tmp_arg_vars {
return
}
if g.is_js_call {
return
}
2020-11-21 19:07:47 +01:00
if g.inside_const {
return
}
free_tmp_arg_vars = false // set the flag to true only if we have at least one arg to free
g.tmp_count2++
2020-10-15 16:26:59 +02:00
mut scope := g.file.scope.innermost(node.pos.pos)
2020-10-22 03:51:25 +02:00
// prepend the receiver for now (TODO turn the receiver into a CallArg everywhere?)
mut args := [ast.CallArg{
typ: node.receiver_type
expr: node.left
is_tmp_autofree: node.free_receiver
}]
args << node.args
// for i, arg in node.args {
for i, arg in args {
if !arg.is_tmp_autofree {
continue
}
if arg.expr is ast.CallExpr {
// Any argument can be an expression that has to be freed. Generate a tmp expression
// for each of those recursively.
g.autofree_call_pregen(arg.expr)
}
free_tmp_arg_vars = true
// t := g.new_tmp_var() + '_arg_expr_${name}_$i'
fn_name := node.name.replace('.', '_') // can't use name...
// t := '_tt${g.tmp_count2}_arg_expr_${fn_name}_$i'
t := '_arg_expr_${fn_name}_${i}_$node.pos.pos'
// g.called_fn_name = name
used := false // scope.known_var(t)
mut s := '$t = '
if used {
// This means this tmp var name was already used (the same function was called and
2020-10-29 07:06:58 +01:00
// `_arg_fnname_1` was already generated).
// We do not need to declare this variable again, so just generate `t = ...`
// instead of `string t = ...`, and we need to mark this variable as unused,
// so that it's freed after the call. (Used tmp arg vars are not freed to avoid double frees).
if x := scope.find(t) {
match mut x {
ast.Var { x.is_used = false }
else {}
}
}
s = '$t = '
} else {
scope.register(ast.Var{
2020-10-15 16:26:59 +02:00
name: t
typ: table.string_type
is_autofree_tmp: true
2020-12-05 21:27:49 +01:00
pos: node.pos
2020-10-15 16:26:59 +02:00
})
s = 'string $t = '
}
2020-10-18 00:48:06 +02:00
// g.expr(arg.expr)
s += g.write_expr_to_string(arg.expr)
// g.writeln(';// new af pre')
s += ';// new af2 pre'
g.strs_to_free0 << s
// This tmp arg var will be freed with the rest of the vars at the end of the scope.
}
}
fn (mut g Gen) autofree_call_postgen(node_pos int) {
/*
2020-10-10 10:34:35 +02:00
if g.strs_to_free.len == 0 {
return
}
*/
/*
g.writeln('\n/* strs_to_free3: $g.nr_vars_to_free */')
if g.nr_vars_to_free <= 0 {
return
}
*/
/*
2020-10-10 10:34:35 +02:00
for s in g.strs_to_free {
g.writeln('string_free(&$s);')
}
if !g.inside_or_block {
// we need to free the vars both inside the or block (in case of an error) and after it
// if we reset the array here, then the vars will not be freed after the block.
g.strs_to_free = []
}
*/
if g.inside_vweb_tmpl {
return
}
// g.doing_autofree_tmp = true
2020-11-07 04:00:45 +01:00
// g.write('/* postgen */')
scope := g.file.scope.innermost(node_pos)
for _, obj in scope.objects {
match mut obj {
ast.Var {
// if var.typ == 0 {
// // TODO why 0?
// continue
// }
is_optional := obj.typ.has_flag(.optional)
if is_optional {
// TODO: free optionals
continue
}
if !obj.is_autofree_tmp {
continue
}
if obj.is_used {
// this means this tmp expr var has already been freed
continue
}
obj.is_used = true
g.autofree_variable(obj)
// g.nr_vars_to_free--
}
else {}
}
}
2020-11-07 04:00:45 +01:00
// g.write('/* postgen end */')
// g.doing_autofree_tmp = false
2020-10-10 10:34:35 +02:00
}
2020-09-05 12:00:35 +02:00
fn (mut g Gen) call_args(node ast.CallExpr) {
args := if g.is_js_call { node.args[1..] } else { node.args }
expected_types := node.expected_arg_types
is_variadic := expected_types.len > 0
&& expected_types[expected_types.len - 1].has_flag(.variadic)
for i, arg in args {
if is_variadic && i == expected_types.len - 1 {
2020-04-14 00:37:47 +02:00
break
}
use_tmp_var_autofree := g.is_autofree && arg.typ == table.string_type && arg.is_tmp_autofree
&& !g.inside_const && !g.is_builtin_mod
2020-09-05 12:00:35 +02:00
// g.write('/* af=$arg.is_tmp_autofree */')
2020-04-14 00:37:47 +02:00
// some c fn definitions dont have args (cfns.v) or are not updated in checker
// when these are fixed we wont need this check
if i < expected_types.len {
if use_tmp_var_autofree {
if arg.is_tmp_autofree { // && !g.is_js_call {
// We saved expressions in temp variables so that they can be freed later.
2020-08-27 11:30:28 +02:00
// `foo(str + str2) => x := str + str2; foo(x); x.free()`
// g.write('_arg_expr_${g.called_fn_name}_$i')
// Use these variables here.
2020-09-05 12:00:35 +02:00
fn_name := node.name.replace('.', '_')
// name := '_tt${g.tmp_count2}_arg_expr_${fn_name}_$i'
name := '_arg_expr_${fn_name}_${i + 1}_$node.pos.pos'
2020-09-05 12:00:35 +02:00
g.write('/*af arg*/' + name)
2020-08-27 11:30:28 +02:00
}
2020-05-10 21:16:22 +02:00
} else {
g.ref_or_deref_arg(arg, expected_types[i])
}
2020-04-14 00:37:47 +02:00
} else {
2020-08-27 17:54:36 +02:00
if use_tmp_var_autofree {
// TODO copypasta, move to an inline fn
2020-09-05 12:00:35 +02:00
fn_name := node.name.replace('.', '_')
// name := '_tt${g.tmp_count2}_arg_expr_${fn_name}_$i'
name := '_arg_expr_${fn_name}_${i + 1}_$node.pos.pos'
2020-09-05 12:00:35 +02:00
g.write('/*af arg2*/' + name)
2020-08-27 17:54:36 +02:00
} else {
g.expr(arg.expr)
}
2020-04-14 00:37:47 +02:00
}
if i < args.len - 1 || is_variadic {
2020-04-14 00:37:47 +02:00
g.write(', ')
}
}
arg_nr := expected_types.len - 1
if is_variadic {
2020-04-14 00:37:47 +02:00
varg_type := expected_types[expected_types.len - 1]
variadic_count := args.len - arg_nr
arr_sym := g.table.get_type_symbol(varg_type)
arr_info := arr_sym.info as table.Array
elem_type := g.typ(arr_info.elem_type)
if args.len > 0 && args[args.len - 1].expr is ast.ArrayDecompose {
g.expr(args[args.len - 1].expr)
} else {
if variadic_count > 0 {
g.write('new_array_from_c_array($variadic_count, $variadic_count, sizeof($elem_type), _MOV(($elem_type[$variadic_count]){')
for j in arg_nr .. args.len {
g.ref_or_deref_arg(args[j], arr_info.elem_type)
if j < args.len - 1 {
g.write(', ')
}
2020-04-14 00:37:47 +02:00
}
g.write('}))')
} else {
g.write('__new_array_with_default(0, 0, sizeof($elem_type), 0)')
2020-04-14 00:37:47 +02:00
}
}
}
}
[inline]
2020-04-22 20:20:49 +02:00
fn (mut g Gen) ref_or_deref_arg(arg ast.CallArg, expected_type table.Type) {
2020-04-25 09:08:53 +02:00
arg_is_ptr := expected_type.is_ptr() || expected_type.idx() in table.pointer_type_idxs
expr_is_ptr := arg.typ.is_ptr() || arg.typ.idx() in table.pointer_type_idxs
if expected_type == 0 {
g.checker_bug('ref_or_deref_arg expected_type is 0', arg.pos)
}
exp_sym := g.table.get_type_symbol(expected_type)
2020-04-14 00:37:47 +02:00
if arg.is_mut && !arg_is_ptr {
g.write('&/*mut*/')
} else if arg_is_ptr && !expr_is_ptr {
if arg.is_mut {
if exp_sym.kind == .array {
if arg.expr is ast.Ident && (arg.expr as ast.Ident).kind == .variable {
g.write('&/*arr*/')
g.expr(arg.expr)
} else {
// Special case for mutable arrays. We can't `&` function
// results, have to use `(array[]){ expr }[0]` hack.
g.write('&/*111*/(array[]){')
g.expr(arg.expr)
g.write('}[0]')
}
2020-04-14 00:37:47 +02:00
return
}
}
if !g.is_json_fn {
if arg.typ == 0 {
g.checker_bug('ref_or_deref_arg arg.typ is 0', arg.pos)
}
2020-07-14 00:27:47 +02:00
arg_typ_sym := g.table.get_type_symbol(arg.typ)
expected_deref_type := if expected_type.is_ptr() {
expected_type.deref()
} else {
expected_type
}
deref_sym := g.table.get_type_symbol(expected_deref_type)
if !((arg_typ_sym.kind == .function) || deref_sym.kind in [.sum_type, .interface_]) {
g.write('(voidptr)&/*qq*/')
}
2020-04-14 00:37:47 +02:00
}
} else if arg.typ.has_flag(.shared_f) && !expected_type.has_flag(.shared_f) {
if expected_type.is_ptr() {
g.write('&')
}
g.expr(arg.expr)
g.write('->val')
return
2020-04-14 00:37:47 +02:00
}
g.expr_with_cast(arg.expr, arg.typ, expected_type)
}
2020-04-22 20:20:49 +02:00
fn (mut g Gen) is_gui_app() bool {
$if windows {
if g.force_main_console {
return false
}
for cf in g.table.cflags {
if cf.value == 'gdi32' {
return true
}
}
}
return false
}
fn (g &Gen) fileis(s string) bool {
return g.file.path.contains(s)
}
2020-06-04 12:20:14 +02:00
fn (mut g Gen) write_fn_attrs(attrs []table.Attr) string {
2020-06-04 12:20:14 +02:00
mut msvc_attrs := ''
for attr in attrs {
match attr.name {
'inline' {
g.write('inline ')
}
'no_inline' {
2020-06-24 14:56:44 +02:00
// since these are supported by GCC, clang and MSVC, we can consider them officially supported.
g.write('__NOINLINE ')
}
'irq_handler' {
g.write('__IRQHANDLER ')
}
'_cold' {
2020-06-24 14:56:44 +02:00
// GCC/clang attributes
// prefixed by _ to indicate they're for advanced users only and not really supported by V.
// source for descriptions: https://gcc.gnu.org/onlinedocs/gcc/Common-Function-Attributes.html#Common-Function-Attributes
// The cold attribute on functions is used to inform the compiler that the function is unlikely
// to be executed. The function is optimized for size rather than speed and on many targets it
// is placed into a special subsection of the text section so all cold functions appear close
// together, improving code locality of non-cold parts of program.
g.write('__attribute__((cold)) ')
}
'_constructor' {
2020-06-24 14:56:44 +02:00
// The constructor attribute causes the function to be called automatically before execution
// enters main ().
g.write('__attribute__((constructor)) ')
}
'_destructor' {
2020-06-24 14:56:44 +02:00
// The destructor attribute causes the function to be called automatically after main ()
// completes or exit () is called.
g.write('__attribute__((destructor)) ')
}
'_flatten' {
2020-06-24 14:56:44 +02:00
// Generally, inlining into a function is limited. For a function marked with this attribute,
// every call inside this function is inlined, if possible.
g.write('__attribute__((flatten)) ')
}
'_hot' {
2020-06-24 14:56:44 +02:00
// The hot attribute on a function is used to inform the compiler that the function is a hot
// spot of the compiled program.
g.write('__attribute__((hot)) ')
}
'_malloc' {
2020-06-24 14:56:44 +02:00
// This tells the compiler that a function is malloc-like, i.e., that the pointer P returned by
// the function cannot alias any other pointer valid when the function returns, and moreover no
// pointers to valid objects occur in any storage addressed by P.
g.write('__attribute__((malloc)) ')
}
'_pure' {
2020-06-24 14:56:44 +02:00
// Calls to functions whose return value is not affected by changes to the observable state
// of the program and that have no observable effects on such state other than to return a
// value may lend themselves to optimizations such as common subexpression elimination.
// Declaring such functions with the const attribute allows GCC to avoid emitting some calls in
// repeated invocations of the function with the same argument values.
g.write('__attribute__((const)) ')
}
'windows_stdcall' {
2020-06-24 14:56:44 +02:00
// windows attributes (msvc/mingw)
// prefixed by windows to indicate they're for advanced users only and not really supported by V.
msvc_attrs += '__stdcall '
}
'console' {
g.force_main_console = true
}
else {
// nothing but keep V happy
}
2020-06-04 12:20:14 +02:00
}
}
return msvc_attrs
}