v/vlib/compiler/fn.v

1722 lines
46 KiB
V
Raw Normal View History

2020-01-23 21:04:46 +01:00
// Copyright (c) 2019-2020 Alexander Medvednikov. All rights reserved.
2019-06-23 04:21:30 +02:00
// Use of this source code is governed by an MIT license
// that can be found in the LICENSE file.
module compiler
2019-06-22 20:20:28 +02:00
2019-12-18 07:04:35 +01:00
import (
strings
)
2019-06-22 20:20:28 +02:00
const (
MaxLocalVars = 50
)
2019-11-08 04:03:06 +01:00
pub struct Fn {
2019-06-22 20:20:28 +02:00
// addr int
2019-12-18 07:04:35 +01:00
// pub:
2019-06-22 20:20:28 +02:00
mut:
2019-12-18 07:04:35 +01:00
name string
mod string
// local_vars []Var
// var_idx int
args []Var
is_interface bool
2019-06-22 20:20:28 +02:00
// called_fns []string
// idx int
2019-12-18 07:04:35 +01:00
scope_level int
typ string // return type
receiver_typ string
is_c bool
is_public bool
is_method bool
is_decl bool // type myfn fn(int, int)
is_unsafe bool
is_deprecated bool
is_variadic bool
is_generic bool
returns_error bool
defer_text []string
type_pars []string
type_inst []TypeInst
generic_fn_idx int
parser_idx int
fn_name_token_idx int // used by error reporting
2019-12-18 07:04:35 +01:00
comptime_define string
is_used bool // so that we can skip unused fns in resulting C code
// x64_addr i64 // address in the generated x64 binary
2019-06-22 20:20:28 +02:00
}
2019-10-20 09:19:37 +02:00
struct TypeInst {
mut:
2019-12-18 07:04:35 +01:00
// an instantiation of generic params (e.g. ["int","int","double"])
inst map[string]string
done bool
2019-10-20 09:19:37 +02:00
}
const (
2019-12-18 07:04:35 +01:00
EmptyFn = Fn{
}
MainFn = Fn{
name: 'main'
}
)
pub fn (a []TypeInst) str() string {
2019-10-20 09:19:37 +02:00
mut r := []string
for t in a {
mut s := ' | '
for k in t.inst.keys() {
2019-12-18 07:04:35 +01:00
s += k + ' -> ' + t.inst[k] + ' | '
2019-10-20 09:19:37 +02:00
}
r << s
}
return r.str()
}
fn (p &Parser) find_var_or_const(name string) ?Var {
2019-11-26 04:41:56 +01:00
if p.known_var(name) {
return p.find_var(name)
}
if p.table.known_const(name) {
return p.table.find_const(name)
}
modname := p.prepend_mod(name)
if p.table.known_const(modname) {
return p.table.find_const(modname)
}
return none
}
2019-09-23 19:34:08 +02:00
fn (p &Parser) find_var(name string) ?Var {
for i in 0 .. p.var_idx {
if p.local_vars[i].name == name {
return p.local_vars[i]
}
}
return none
}
fn (p &Parser) find_var_check_new_var(name string) ?Var {
2019-09-23 19:34:08 +02:00
for i in 0 .. p.var_idx {
if p.local_vars[i].name == name {
return p.local_vars[i]
}
}
// A hack to allow `newvar := Foo{ field: newvar }`
// Declare the variable so that it can be used in the initialization
if name == 'main__' + p.var_decl_name {
return Var{
2019-12-18 07:04:35 +01:00
name: p.var_decl_name
typ: 'voidptr'
is_mut: true
}
}
return none
}
fn (p mut Parser) open_scope() {
p.cur_fn.defer_text << ''
p.cur_fn.scope_level++
2019-06-22 20:20:28 +02:00
}
fn (p mut Parser) mark_var_used(v Var) {
2019-09-24 05:18:06 +02:00
if v.idx == -1 || v.idx >= p.local_vars.len {
2019-09-23 21:47:09 +02:00
return
2019-10-20 09:19:37 +02:00
}
2019-09-23 21:47:09 +02:00
p.local_vars[v.idx].is_used = true
}
fn (p mut Parser) mark_var_returned(v Var) {
2019-09-23 21:47:09 +02:00
if v.idx == -1 || v.idx >= p.local_vars.len {
return
2019-10-20 09:19:37 +02:00
}
2019-09-23 21:47:09 +02:00
p.local_vars[v.idx].is_returned = true
}
fn (p mut Parser) mark_var_changed(v Var) {
2019-09-23 21:47:09 +02:00
if v.idx == -1 || v.idx >= p.local_vars.len {
return
2019-10-20 09:19:37 +02:00
}
2019-09-23 21:47:09 +02:00
p.local_vars[v.idx].is_changed = true
2019-06-22 20:20:28 +02:00
}
fn (p mut Parser) mark_arg_moved(v Var) {
2019-12-18 08:17:07 +01:00
for i, arg in p.cur_fn.args {
if arg.name == v.name {
2019-12-18 07:04:35 +01:00
// println('setting f $p.cur_fn.name arg $arg.name to is_mut')
p.cur_fn.args[i].is_moved = true
break
2019-10-20 09:19:37 +02:00
}
}
p.table.fns[p.cur_fn.name] = p.cur_fn
}
fn (p &Parser) known_var(name string) bool {
_ = p.find_var(name) or {
return false
2019-10-20 09:19:37 +02:00
}
return true
}
fn (p &Parser) known_var_check_new_var(name string) bool {
2019-10-20 09:19:37 +02:00
_ = p.find_var_check_new_var(name) or {
return false
}
return true
2019-06-22 20:20:28 +02:00
}
2019-09-23 19:34:08 +02:00
fn (p mut Parser) register_var(v Var) {
2019-12-18 07:04:35 +01:00
mut new_var := {
v |
idx:p.var_idx,
scope_level:p.cur_fn.scope_level
}
2019-09-23 19:34:08 +02:00
if v.line_nr == 0 {
new_var.token_idx = p.cur_tok_index()
new_var.line_nr = p.cur_tok().line_nr
2019-09-23 19:34:08 +02:00
}
2019-06-22 20:20:28 +02:00
// Expand the array
2019-09-23 19:34:08 +02:00
if p.var_idx >= p.local_vars.len {
p.local_vars << new_var
2019-06-22 20:20:28 +02:00
}
else {
2019-09-23 19:34:08 +02:00
p.local_vars[p.var_idx] = new_var
2019-06-22 20:20:28 +02:00
}
2019-09-23 19:34:08 +02:00
p.var_idx++
2019-06-22 20:20:28 +02:00
}
2019-09-23 19:34:08 +02:00
fn (p mut Parser) clear_vars() {
// shared a := [1, 2, 3]
p.var_idx = 0
2019-09-26 04:28:43 +02:00
if p.local_vars.len > 0 {
if p.pref.autofree {
2019-12-18 07:04:35 +01:00
// p.local_vars.free()
2019-09-26 04:28:43 +02:00
}
2019-11-14 07:18:07 +01:00
p.local_vars = []
2019-09-26 04:28:43 +02:00
}
}
2019-06-22 20:20:28 +02:00
// Function signatures are added to the top of the .c file in the first run.
fn (p mut Parser) fn_decl() {
2019-09-23 19:34:08 +02:00
p.clear_vars() // clear local vars every time a new fn is started
2019-12-18 07:04:35 +01:00
defer {
p.fgen_nl()
p.fgen_nl()
}
2019-11-29 14:46:43 +01:00
fn_start_idx := p.cur_tok_index()
// If we are in the first pass, create a new function.
// In the second pass fetch the one we created.
/*
mut f := if p.first_pass {
Fn{
mod: p.mod
is_public: p.tok == .key_pub
}
else {
2019-10-20 09:19:37 +02:00
}
*/
2019-12-18 07:04:35 +01:00
2019-10-31 11:08:01 +01:00
is_pub := p.tok == .key_pub
mut f := Fn{
mod: p.mod
2019-10-31 11:08:01 +01:00
is_public: is_pub || p.is_vh // functions defined in .vh are always public
2019-12-22 02:34:37 +01:00
is_unsafe: p.attr == 'unsafe_fn'
2019-10-20 19:15:48 +02:00
is_deprecated: p.attr == 'deprecated'
2019-12-19 03:41:12 +01:00
comptime_define: if p.attr.starts_with('if ') { p.attr[3..] } else { '' }
}
is_live := p.pref.is_live && p.attr == 'live'
is_solive := p.pref.is_solive && p.attr == 'live'
2019-10-31 11:08:01 +01:00
if is_pub {
2019-06-22 20:20:28 +02:00
p.next()
2019-11-09 20:05:44 +01:00
p.fspace()
2019-06-22 20:20:28 +02:00
}
p.returns = false
2019-12-18 07:04:35 +01:00
// p.gen('/* returns $p.returns */')
2019-06-22 20:20:28 +02:00
p.next()
2019-11-09 17:13:26 +01:00
p.fspace()
2019-06-22 20:20:28 +02:00
// Method receiver
mut receiver_typ := ''
if p.tok == .lpar {
2019-06-22 20:20:28 +02:00
f.is_method = true
p.check(.lpar)
2019-06-22 20:20:28 +02:00
receiver_name := p.check_name()
2019-11-09 20:05:44 +01:00
p.fspace()
is_mut := p.tok == .key_mut
is_amp := p.tok == .amp
2019-06-22 20:20:28 +02:00
if is_mut || is_amp {
2019-11-09 20:05:44 +01:00
p.check(p.tok)
2019-11-18 11:10:31 +01:00
if !is_amp {
p.fspace()
}
2019-06-22 20:20:28 +02:00
}
receiver_typ = p.get_type()
2019-10-23 12:03:14 +02:00
t := p.table.find_type(receiver_typ)
if (t.name == '' || t.is_placeholder) && !p.first_pass() {
p.error('unknown receiver type `$receiver_typ`')
}
2019-10-23 12:03:14 +02:00
if t.cat == .interface_ {
2019-06-22 20:20:28 +02:00
p.error('invalid receiver type `$receiver_typ` (`$receiver_typ` is an interface)')
}
// Don't allow modifying types from a different module
2019-12-18 07:04:35 +01:00
if !p.first_pass() && !p.builtin_mod && t.mod != p.mod && !p.is_vgen // let vgen define methods like .str() on types defined in other modules
2019-10-23 12:03:14 +02:00
{
2019-12-18 07:04:35 +01:00
// println('T.mod=$T.mod')
// println('p.mod=$p.mod')
2019-06-22 20:20:28 +02:00
p.error('cannot define new methods on non-local type `$receiver_typ`')
}
// `(f *Foo)` instead of `(f mut Foo)` is a common mistake
if receiver_typ.ends_with('*') {
2019-10-23 12:03:14 +02:00
tt := receiver_typ.replace('*', '')
p.error('use `($receiver_name mut $tt)` instead of `($receiver_name *$tt)`')
2019-06-22 20:20:28 +02:00
}
f.receiver_typ = receiver_typ
if is_mut || is_amp {
receiver_typ += '*'
}
p.check(.rpar)
2019-08-17 21:19:37 +02:00
p.fspace()
2019-12-18 07:04:35 +01:00
receiver := Var{
2019-06-22 20:20:28 +02:00
name: receiver_name
is_arg: true
typ: receiver_typ
is_mut: is_mut
ref: is_amp
ptr: is_mut
line_nr: p.scanner.line_nr
token_idx: p.cur_tok_index()
2019-06-22 20:20:28 +02:00
}
f.args << receiver
2019-09-23 19:34:08 +02:00
p.register_var(receiver)
2019-06-22 20:20:28 +02:00
}
// +-/* methods (operator overloading)
mut is_op := false
if p.tok in [.plus, .minus, .mul, .div, .mod] {
2019-06-22 20:20:28 +02:00
f.name = p.tok.str()
p.next()
is_op = true
2019-06-22 20:20:28 +02:00
}
else {
f.name = p.check_name()
}
f.fn_name_token_idx = p.cur_tok_index()
// init fn
2019-11-01 14:10:28 +01:00
if f.name == 'init' && !f.is_method && f.is_public && !p.is_vh {
p.error('init function cannot be public')
}
// .str() methods
if f.is_method && f.name == 'str' && !f.is_public {
p.error('.str() methods must be declared as public')
}
2019-06-22 20:20:28 +02:00
// C function header def? (fn C.NSMakeRect(int,int,int,int))
2019-08-17 21:19:37 +02:00
is_c := f.name == 'C' && p.tok == .dot
2019-06-22 20:20:28 +02:00
// Just fn signature? only builtin.v + default build mode
2019-12-18 07:04:35 +01:00
// if p.is_vh {
// if f.name == 'main' {
// println('\n\nfn_decl() name=$f.name pass=$p.pass $p.file_name receiver_typ=$receiver_typ nogen=$p.cgen.nogen')
// }
2019-06-22 20:20:28 +02:00
if is_c {
p.check(.dot)
2019-06-22 20:20:28 +02:00
f.name = p.check_name()
f.is_c = true
}
orig_name := f.name
2019-06-22 20:20:28 +02:00
// simple_name := f.name
// user.register() => User_register()
has_receiver := receiver_typ.len > 0
if receiver_typ != '' {
// f.name = '${receiver_typ}_${f.name}'
}
// full mod function name
2019-11-06 22:40:37 +01:00
// `os.exit()` ==> `os__exit()`
// if !is_c && !p.builtin_mod && receiver_typ.len == 0 {
2019-12-18 07:04:35 +01:00
if !is_c && !has_receiver && (!p.builtin_mod || (p.builtin_mod && f.name == 'init')) {
f.name = p.prepend_mod(f.name)
2019-06-22 20:20:28 +02:00
}
2019-09-18 14:06:34 +02:00
if p.first_pass() && receiver_typ.len == 0 {
2019-12-18 07:04:35 +01:00
if existing_fn:=p.table.find_fn(f.name){
2019-11-06 22:40:37 +01:00
// This existing function could be defined as C decl before
// (no body), then we don't need to throw an error.
if !existing_fn.is_decl {
p.error('redefinition of `$f.name`')
}
2019-09-18 14:06:34 +02:00
}
2019-06-22 20:20:28 +02:00
}
// Generic?
if p.tok == .lt {
2019-11-29 14:46:43 +01:00
// instance (dispatch)
if p.generic_dispatch.inst.size > 0 {
2019-12-21 01:53:58 +01:00
rename_generic_fn_instance(mut f, &p.generic_dispatch)
2019-12-18 07:04:35 +01:00
}
else {
2019-11-29 14:46:43 +01:00
f.is_generic = true
}
2019-06-22 20:20:28 +02:00
p.next()
2019-10-20 09:19:37 +02:00
for {
type_par := p.check_name()
if type_par.len > 1 || !(type_par in reserved_type_param_names) {
p.error('type parameters must be single-character, upper-case letters of the following set: $reserved_type_param_names')
}
if type_par in f.type_pars {
p.error('redeclaration of type parameter `$type_par`')
}
f.type_pars << type_par
2019-12-18 07:04:35 +01:00
if p.tok == .gt {
break
}
2019-10-20 09:19:37 +02:00
p.check(.comma)
2019-06-22 20:20:28 +02:00
}
p.check(.gt)
2019-11-29 14:46:43 +01:00
p.set_current_fn(f)
2019-06-22 20:20:28 +02:00
}
// Args (...)
p.fn_args(mut f)
if is_op {
2019-12-18 07:04:35 +01:00
if f.args.len != 1 + 1 {
// +1 is for the receiver
p.error('operator overloading methods must have only 1 argument')
}
if f.args[0].typ != f.args[1].typ {
p.error('operators must have the same types on both sides')
}
}
2019-06-22 20:20:28 +02:00
// Returns an error?
if p.tok == .not {
2019-06-22 20:20:28 +02:00
p.next()
f.returns_error = true
}
// Returns a type?
mut typ := 'void'
if p.tok in [.name, .mul, .amp, .lsbr, .question, .lpar] {
2019-11-09 20:05:44 +01:00
p.fspace()
2019-06-22 20:20:28 +02:00
typ = p.get_type()
}
2019-11-06 22:40:37 +01:00
// V allows empty functions (just definitions)
2019-12-18 07:04:35 +01:00
is_fn_header := !is_c && !p.is_vh && p.tok != .lcbr
2019-06-22 20:20:28 +02:00
if is_fn_header {
f.name = orig_name // don't prepend module to external fn defs
2019-06-22 20:20:28 +02:00
f.is_decl = true
}
// Make sure the name is valid
if !is_c && !p.pref.translated && !is_fn_header {
if contains_capital(orig_name) && !p.fileis('view.v') && !p.is_vgen {
2019-12-18 07:04:35 +01:00
// println(orig_name)
p.error('function names cannot contain uppercase letters, use snake_case instead')
}
if f.name[0] == `_` {
p.error('function names cannot start with `_`, use snake_case instead')
}
if orig_name.contains('__') {
p.error('function names cannot contain double underscores, use single underscores instead')
}
}
2019-11-06 22:40:37 +01:00
// `{` required only in normal function declarations
2019-10-07 00:31:01 +02:00
if !is_c && !p.is_vh && !is_fn_header {
2019-11-09 17:13:26 +01:00
p.fspace()
p.check(.lcbr)
2019-12-18 07:04:35 +01:00
// p.fgen_nl()
2019-06-22 20:20:28 +02:00
}
// Register ?option type for return value and args
2019-11-19 07:53:52 +01:00
if typ.starts_with('Option_') {
2019-06-22 20:20:28 +02:00
p.cgen.typedefs << 'typedef Option $typ;'
2019-12-18 07:04:35 +01:00
// p.cgen.typedefs << 'typedef struct Option_$typ Option_$typ'
2019-06-22 20:20:28 +02:00
}
for arg in f.args {
if arg.typ.starts_with('Option_') {
p.cgen.typedefs << 'typedef Option $arg.typ;'
}
}
2019-06-22 20:20:28 +02:00
// Register function
f.typ = typ
str_args := f.str_args(p.table)
2019-06-22 20:20:28 +02:00
// Special case for main() args
if f.name == 'main__main' && !has_receiver {
2020-03-06 18:53:29 +01:00
if p.pref.backend == .x64 && !p.first_pass() {
2019-12-31 19:42:16 +01:00
//p.x64.save_main_fn_addr()
}
if str_args != '' || typ != 'void' {
p.error_with_token_index('fn main must have no arguments and no return values', f.fn_name_token_idx)
2019-06-22 20:20:28 +02:00
}
}
2019-10-20 09:19:37 +02:00
dll_export_linkage := p.get_linkage_prefix()
2019-12-18 07:04:35 +01:00
p.set_current_fn(f)
2019-08-17 21:19:37 +02:00
// Generate `User_register()` instead of `register()`
2019-06-22 20:20:28 +02:00
// Internally it's still stored as "register" in type User
2019-09-14 22:48:30 +02:00
mut fn_name_cgen := p.table.fn_gen_name(f)
2019-06-22 20:20:28 +02:00
// Start generation of the function body
skip_main_in_test := false
2019-10-07 00:31:01 +02:00
if !is_c && !is_live && !p.is_vh && !is_fn_header && !skip_main_in_test {
if p.pref.obfuscate {
2019-07-07 21:46:21 +02:00
p.genln('; // $f.name')
2019-06-22 20:20:28 +02:00
}
2019-10-20 09:19:37 +02:00
// Generic functions are inserted as needed from the call site
2019-12-18 18:07:32 +01:00
if f.is_generic && !p.scanner.is_fmt {
2019-10-20 09:19:37 +02:00
if p.first_pass() {
2019-11-29 14:46:43 +01:00
if !p.scanner.is_vh {
gpidx := p.v.get_file_parser_index(p.file_path) or {
panic('error finding parser for: $p.file_path')
}
f.parser_idx = gpidx
}
f.generic_fn_idx = fn_start_idx
2019-10-25 20:32:27 +02:00
if f.is_method {
rcv := p.table.find_type(receiver_typ)
if p.first_pass() && rcv.name == '' {
p.error('cannot currently add generic method to a type declared after it or in another module')
2019-10-25 20:32:27 +02:00
}
// println('added generic method r:$rcv.name f:$f.name')
2019-10-25 20:32:27 +02:00
p.add_method(rcv.name, f)
2019-12-18 07:04:35 +01:00
}
else {
2019-10-25 20:32:27 +02:00
p.table.register_fn(f)
}
2019-08-17 21:19:37 +02:00
}
2019-12-21 22:54:37 +01:00
p.set_current_fn(EmptyFn)
2019-12-21 03:33:59 +01:00
p.skip_fn_body()
2019-10-20 09:19:37 +02:00
return
2019-12-18 07:04:35 +01:00
}
else {
2019-09-14 22:48:30 +02:00
p.gen_fn_decl(f, typ, str_args)
2019-08-17 21:19:37 +02:00
}
2019-06-22 20:20:28 +02:00
}
if is_fn_header {
2019-12-08 20:22:47 +01:00
p.genln('$typ $fn_name_cgen ($str_args);')
2019-11-18 11:10:31 +01:00
p.fgen_nl()
2019-06-22 20:20:28 +02:00
}
if is_c {
2019-12-17 15:28:25 +01:00
p.fgen_nl()
2019-06-22 20:20:28 +02:00
}
2020-01-16 17:03:31 +01:00
2019-06-22 20:20:28 +02:00
// Register the method
if receiver_typ != '' {
2019-07-15 22:44:26 +02:00
mut receiver_t := p.table.find_type(receiver_typ)
2019-06-22 20:20:28 +02:00
// No such type yet? It could be defined later. Create a new type.
// struct declaration later will modify it instead of creating a new one.
2019-07-29 18:21:36 +02:00
if p.first_pass() && receiver_t.name == '' {
2019-12-18 07:04:35 +01:00
// println('fn decl ! registering placeholder $receiver_typ')
receiver_t = Type{
2019-06-22 20:20:28 +02:00
name: receiver_typ.replace('*', '')
mod: p.mod
2019-06-22 20:20:28 +02:00
is_placeholder: true
2019-08-31 15:38:13 +02:00
}
2019-12-05 12:09:33 +01:00
p.table.register_type(receiver_t)
2019-06-22 20:20:28 +02:00
}
2019-09-18 14:37:34 +02:00
p.add_method(receiver_t.name, f)
2019-06-22 20:20:28 +02:00
}
2019-12-18 07:04:35 +01:00
else if p.first_pass() {
// println('register_fn $f.name typ=$typ isg=$is_generic pass=$p.pass ' +
2019-12-18 07:04:35 +01:00
// '$p.file_name')
2019-06-22 20:20:28 +02:00
p.table.register_fn(f)
}
if p.first_pass() && p.attr == 'live' && !(is_live || is_solive) {
2020-02-09 10:08:04 +01:00
println('INFO: run `v -live $p.v.pref.path `, if you want to use [live] function $f.name .')
}
2020-01-16 17:03:31 +01:00
2019-10-07 00:31:01 +02:00
if p.is_vh || p.first_pass() || is_live || is_fn_header || skip_main_in_test {
2019-08-17 21:19:37 +02:00
// First pass? Skip the body for now
// Look for generic calls.
2019-10-07 00:31:01 +02:00
if !p.is_vh && !is_fn_header {
p.skip_fn_body()
2019-06-22 20:20:28 +02:00
}
// Live code reloading? Load all fns from .so
2019-07-29 18:21:36 +02:00
if is_live && p.first_pass() && p.mod == 'main' {
2019-12-18 07:04:35 +01:00
// println('ADDING SO FN $fn_name_cgen')
2019-06-22 20:20:28 +02:00
p.cgen.so_fns << fn_name_cgen
fn_name_cgen = '(* $fn_name_cgen )'
}
2019-08-17 21:19:37 +02:00
// Function definition that goes to the top of the C file.
2019-12-08 20:22:47 +01:00
mut fn_decl := '$dll_export_linkage$typ $fn_name_cgen ($str_args)'
if p.pref.obfuscate {
2019-07-29 18:21:36 +02:00
fn_decl += '; // $f.name'
2019-06-22 20:20:28 +02:00
}
// Add function definition to the top
if !is_c && p.first_pass() {
2019-10-07 00:31:01 +02:00
p.cgen.fns << fn_decl + ';'
2019-10-04 14:48:09 +02:00
}
2019-06-22 20:20:28 +02:00
return
}
2020-01-16 17:03:31 +01:00
if is_solive {
// 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_' .
// 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.
function_args := f.str_args_without_types(p.table)
mut live_fncall := 'impl_live_${fn_name_cgen}(${function_args});'
mut live_fnreturn := ''
if typ != 'void' {
live_fncall = '$typ res = $live_fncall'
live_fnreturn = 'return res;'
}
p.genln(' pthread_mutex_lock(&live_fn_mutex);')
p.genln(' $live_fncall')
p.genln(' pthread_mutex_unlock(&live_fn_mutex);')
p.genln(' $live_fnreturn')
2020-01-16 17:03:31 +01:00
p.genln('}')
p.genln('$typ impl_live_${fn_name_cgen} ($str_args){')
}
2020-01-16 17:03:31 +01:00
if f.name in ['main__main', 'main', 'WinMain'] {
if p.pref.is_test {
p.error_with_token_index('tests cannot have function `main`', f.fn_name_token_idx)
2019-06-22 20:20:28 +02:00
}
}
// println('is_c=$is_c name=$f.name')
2019-10-07 00:31:01 +02:00
if is_c || p.is_vh || is_fn_header {
2019-06-22 20:20:28 +02:00
return
}
2019-07-29 18:21:36 +02:00
// Profiling mode? Start counting at the beginning of the function (save current time).
if p.pref.is_prof && f.name != 'time__ticks' {
2019-06-22 20:20:28 +02:00
p.genln('double _PROF_START = time__ticks();//$f.name')
2019-09-14 22:48:30 +02:00
cgen_name := p.table.fn_gen_name(f)
if f.defer_text.len > f.scope_level {
2019-09-14 22:48:30 +02:00
f.defer_text[f.scope_level] = ' ${cgen_name}_time += time__ticks() - _PROF_START;'
}
2019-06-22 20:20:28 +02:00
}
2020-03-06 18:53:29 +01:00
if p.pref.backend == .x64 {
2019-12-31 19:42:16 +01:00
//p.x64.register_function_address(f.name)
}
2019-07-24 02:35:25 +02:00
p.statements_no_rcbr()
2019-12-18 07:04:35 +01:00
// p.cgen.nogen = false
2019-06-22 20:20:28 +02:00
// Print counting result after all statements in main
if p.pref.is_prof && f.name == 'main' {
2019-06-22 20:20:28 +02:00
p.genln(p.print_prof_counters())
}
// Counting or not, always need to add defer before the end
if f.defer_text.len > f.scope_level {
2019-08-12 16:21:56 +02:00
p.genln(f.defer_text[f.scope_level])
2019-08-17 21:19:37 +02:00
}
if typ != 'void' && !p.returns {
p.error_with_token_index('$f.name must return "$typ"', f.fn_name_token_idx)
2019-06-22 20:20:28 +02:00
}
2020-03-06 18:53:29 +01:00
if p.pref.backend == .x64 && f.name == 'main__main' && !p.first_pass() {
2019-12-31 19:42:16 +01:00
//p.x64.gen_exit()
}
2020-03-06 18:53:29 +01:00
if p.pref.backend == .x64 && !p.first_pass() {
2019-12-31 19:42:16 +01:00
//p.x64.ret()
}
2019-06-22 20:20:28 +02:00
// {} closed correctly? scope_level should be 0
if p.mod == 'main' {
2019-06-22 20:20:28 +02:00
// println(p.cur_fn.scope_level)
}
if p.cur_fn.scope_level > 2 {
// p.error('unclosed {')
}
// Make sure all vars in this function are used (only in main for now)
/*
if p.mod != 'main' {
2019-10-20 09:19:37 +02:00
p.genln('}')
2019-06-22 20:20:28 +02:00
return
}
*/
2019-12-18 07:04:35 +01:00
2019-10-20 09:19:37 +02:00
p.genln('}')
if !p.builtin_mod && p.mod != 'os' {
p.check_unused_and_mut_vars()
}
p.set_current_fn(EmptyFn)
p.returns = false
}
[inline]
// Skips the entire function's body in the first pass.
fn (p mut Parser) skip_fn_body() {
mut opened_scopes := 0
mut closed_scopes := 0
for {
if p.tok == .lcbr {
opened_scopes++
}
if p.tok == .rcbr {
closed_scopes++
}
// find `foo<Bar>()` in function bodies and register generic types
// TODO
// ...
// Reached a declaration token? (fn, struct, const etc) Stop.
if p.tok.is_decl() {
break
}
// fn body ended, and a new fn attribute declaration like [live] is starting?
if closed_scopes > opened_scopes && p.prev_tok == .rcbr {
if p.tok == .lsbr {
break
}
}
p.next()
}
}
2019-11-08 04:03:06 +01:00
fn (p &Parser) get_linkage_prefix() string {
2019-12-19 03:41:12 +01:00
return if p.pref.ccompiler == 'msvc' && p.attr == 'live' && p.pref.is_so { '__declspec(dllexport) ' } else if p.attr == 'inline' { 'static inline ' } else { '' }
2019-10-20 09:19:37 +02:00
}
fn (p mut Parser) check_unused_and_mut_vars() {
2019-09-23 19:34:08 +02:00
for var in p.local_vars {
2019-06-22 20:20:28 +02:00
if var.name == '' {
break
}
2019-12-31 19:42:16 +01:00
if !var.is_used && !p.pref.is_repl && !var.is_arg && !p.pref.translated &&
2020-01-16 17:03:31 +01:00
var.name != 'tmpl_res' && p.mod != 'vweb' && var.name != 'it' && !p.cur_fn.is_unsafe {
2019-12-18 07:04:35 +01:00
p.production_error_with_token_index('`$var.name` declared and not used', var.token_idx)
2019-06-22 20:20:28 +02:00
}
2020-02-03 11:17:12 +01:00
if !var.is_changed && var.is_mut && !p.pref.is_repl && !p.pref.translated && var.name != 'it' && var.typ != 'T*' && p.mod != 'ui' && var.typ != 'App*' {
p.warn_or_error('`$var.name` is declared as mutable, but it was never changed')
}
2019-06-22 20:20:28 +02:00
}
}
// user.register() => "User_register(user)"
// method_ph - where to insert "user_register("
2019-06-22 20:20:28 +02:00
// receiver_var - "user" (needed for pthreads)
// receiver_type - "User"
fn (p mut Parser) async_fn_call(f Fn, method_ph int, receiver_var, receiver_type string) {
2019-12-14 00:33:19 +01:00
p.verify_fn_before_call(f)
2019-06-22 20:20:28 +02:00
// println('\nfn_call $f.name is_method=$f.is_method receiver_type=$f.receiver_type')
// p.print_tok()
mut thread_name := ''
// Normal function => just its name, method => TYPE_FN.name
2019-06-22 20:20:28 +02:00
mut fn_name := f.name
if f.is_method {
2019-08-17 21:19:37 +02:00
fn_name = receiver_type.replace('*', '') + '_' + f.name
2019-12-18 07:04:35 +01:00
// fn_name = '${receiver_type}_${f.name}'
2019-06-22 20:20:28 +02:00
}
// Generate tmp struct with args
arg_struct_name := 'thread_arg_$fn_name'
tmp_struct := p.get_tmp()
p.genln('$arg_struct_name * $tmp_struct = malloc(sizeof($arg_struct_name));')
mut arg_struct := 'typedef struct $arg_struct_name { '
p.next()
p.check(.lpar)
2019-06-22 20:20:28 +02:00
// str_args contains the args for the wrapper function:
// wrapper(arg_struct * arg) { fn("arg->a, arg->b"); }
mut str_args := ''
mut did_gen_something := false
2019-12-18 08:17:07 +01:00
for i, arg in f.args {
2019-12-18 07:04:35 +01:00
arg_struct += '$arg.typ $arg.name ;' // Add another field (arg) to the tmp struct definition
2019-09-14 22:48:30 +02:00
str_args += 'arg $dot_ptr $arg.name'
2019-06-22 20:20:28 +02:00
if i == 0 && f.is_method {
2019-09-14 22:48:30 +02:00
p.genln('$tmp_struct $dot_ptr $arg.name = $receiver_var ;')
2019-06-22 20:20:28 +02:00
if i < f.args.len - 1 {
str_args += ','
}
did_gen_something = true
2019-06-22 20:20:28 +02:00
continue
}
// Set the struct values (args)
2019-09-14 22:48:30 +02:00
p.genln('$tmp_struct $dot_ptr $arg.name = ')
2019-06-22 20:20:28 +02:00
p.expression()
p.genln(';')
if i < f.args.len - 1 {
p.check(.comma)
2019-06-22 20:20:28 +02:00
str_args += ','
}
did_gen_something = true
}
if !did_gen_something {
// Msvc doesnt like empty struct
arg_struct += 'EMPTY_STRUCT_DECLARATION;'
2019-06-22 20:20:28 +02:00
}
arg_struct += '} $arg_struct_name ;'
// Also register the wrapper, so we can use the original function without modifying it
2019-09-14 22:48:30 +02:00
fn_name = p.table.fn_gen_name(f)
2019-06-22 20:20:28 +02:00
wrapper_name := '${fn_name}_thread_wrapper'
2019-11-06 21:04:40 +01:00
mut wrapper_type := 'void*'
if p.os == .windows {
2019-11-16 00:30:50 +01:00
wrapper_type = 'DWORD WINAPI'
2019-11-06 21:04:40 +01:00
}
2019-12-08 20:22:47 +01:00
wrapper_text := '$wrapper_type $wrapper_name ($arg_struct_name * arg) {$fn_name ( /*f*/$str_args ); return 0; }'
2019-06-22 20:20:28 +02:00
p.cgen.register_thread_fn(wrapper_name, wrapper_text, arg_struct)
// Create thread object
tmp_nr := p.get_tmp_counter()
thread_name = '_thread$tmp_nr'
if p.os != .windows {
2019-06-22 20:20:28 +02:00
p.genln('pthread_t $thread_name;')
}
tmp2 := p.get_tmp()
mut parg := 'NULL'
if f.args.len > 0 {
parg = ' $tmp_struct'
}
// Call the wrapper
if p.os == .windows {
2019-11-16 00:30:50 +01:00
p.genln(' CreateThread(0,0, (LPTHREAD_START_ROUTINE)$wrapper_name, $parg, 0,0);')
2019-06-22 20:20:28 +02:00
}
else {
2019-11-07 22:58:49 +01:00
p.genln('int $tmp2 = pthread_create(& $thread_name, NULL, (void *)$wrapper_name, $parg);')
2019-06-22 20:20:28 +02:00
}
p.check(.rpar)
2019-06-22 20:20:28 +02:00
}
2019-12-14 00:33:19 +01:00
fn (p mut Parser) verify_fn_before_call(f &Fn) {
if f.is_unsafe && !p.builtin_mod && !p.inside_unsafe {
p.warn('you are calling an unsafe function outside of an unsafe block')
}
2019-10-20 19:15:48 +02:00
if f.is_deprecated {
p.warn('$f.name is deprecated')
}
if !f.is_public && !f.is_c && !p.pref.is_test && !f.is_interface && f.mod != p.mod {
if f.name == 'contains' {
println('use `value in numbers` instead of `numbers.contains(value)`')
}
2019-06-26 13:17:45 +02:00
p.error('function `$f.name` is private')
}
2019-12-14 00:33:19 +01:00
}
// p.tok == fn_name
fn (p mut Parser) fn_call(f mut Fn, method_ph int, receiver_var, receiver_type string) {
p.verify_fn_before_call(f)
2020-02-09 10:08:04 +01:00
is_comptime_define := f.comptime_define != '' && !(f.comptime_define in p.v.pref.compile_defines )
2019-10-27 01:36:43 +02:00
if is_comptime_define {
p.cgen.nogen = true
}
2020-03-06 18:53:29 +01:00
if p.pref.backend == .x64 && !p.first_pass() {
2019-12-31 19:42:16 +01:00
//p.x64.call_fn(f.name)
}
2019-06-22 20:20:28 +02:00
p.calling_c = f.is_c
if f.is_c && !p.builtin_mod {
2019-07-21 12:22:41 +02:00
if f.name == 'free' {
2019-08-17 21:19:37 +02:00
p.error('use `free()` instead of `C.free()`')
2019-12-18 07:04:35 +01:00
}
else if f.name == 'malloc' {
2019-08-17 21:19:37 +02:00
p.error('use `malloc()` instead of `C.malloc()`')
}
}
2019-10-31 11:08:01 +01:00
f.is_used = true
2019-10-20 09:19:37 +02:00
cgen_name := p.table.fn_gen_name(f)
2019-12-21 22:54:37 +01:00
p.next() // fn name
2019-12-21 01:53:58 +01:00
mut generic_param_types := []string
2019-07-29 18:21:36 +02:00
if p.tok == .lt {
2019-12-21 01:53:58 +01:00
p.check(.lt)
2019-10-20 09:19:37 +02:00
for {
2019-12-21 01:53:58 +01:00
param_type := p.check_name()
generic_param_types << param_type
2019-12-21 22:54:37 +01:00
if p.tok != .comma {
break
}
2019-12-21 01:53:58 +01:00
p.check(.comma)
2019-08-17 21:19:37 +02:00
}
2019-12-21 01:53:58 +01:00
p.check(.gt)
// mut i := p.token_idx
// for {
2019-12-21 22:54:37 +01:00
// if p.tokens[i].tok == .gt {
// //p.error('explicit type arguments are not allowed; remove `<...>`')
// } else if p.tokens[i].tok == .lpar {
// // probably a typo, do not concern the user with the above error message
// break
// }
// i++
2019-12-21 01:53:58 +01:00
// }
2019-08-17 21:19:37 +02:00
}
// if p.pref.is_prof {
2019-06-22 20:20:28 +02:00
// p.cur_fn.called_fns << cgen_name
// }
// If we have a method placeholder,
// we need to preappend "method(receiver, ...)"
2019-10-20 09:19:37 +02:00
if f.is_method {
2019-06-22 20:20:28 +02:00
receiver := f.args.first()
mut receiver_is_interface := false
2019-12-22 02:34:37 +01:00
if receiver.typ.ends_with('er') || receiver.typ[0] == `I` {
2019-11-08 04:03:06 +01:00
// I absolutely love this syntax
// `s.speak()` =>
// `((void (*)())(Speaker_name_table[s._interface_idx][1]))(s._object);
// where `1` refers to the speak method, since it's the second method
// of the Speaker interface
t := p.table.find_type(receiver.typ)
if t.cat == .interface_ {
// Find the index of the method
mut idx := 0
2019-12-18 08:17:07 +01:00
for i, method in t.methods {
2019-11-08 04:03:06 +01:00
if method.name == f.name {
idx = i
}
}
2019-11-08 04:03:06 +01:00
p.cgen.resetln('')
var := p.expr_var.name
iname := f.args[0].typ // Speaker
p.gen('(($f.typ (*)())(${iname}_name_table[${var}._interface_idx][$idx]))(${var}._object')
receiver_is_interface = true
2019-11-08 04:03:06 +01:00
}
}
2019-12-18 07:04:35 +01:00
// println('r=$receiver.typ RT=$receiver_type')
2019-06-22 20:20:28 +02:00
if receiver.is_mut && !p.expr_var.is_mut {
2019-12-18 07:04:35 +01:00
// println('$method_call recv=$receiver.name recv_mut=$receiver.is_mut')
2019-10-15 17:08:46 +02:00
if p.expr_var.is_for_var {
2019-12-18 07:04:35 +01:00
p.error('`$p.expr_var.name` is immutable, `for` variables' + ' always are')
}
else {
2019-10-15 17:08:46 +02:00
p.error('`$p.expr_var.name` is immutable, declare it with `mut`')
}
2019-08-17 21:19:37 +02:00
}
if !p.expr_var.is_changed && receiver.is_mut {
p.mark_var_changed(p.expr_var)
2019-08-17 21:19:37 +02:00
}
if !receiver_is_interface {
p.gen_method_call(receiver, receiver_type, cgen_name, f.typ, method_ph)
}
2019-12-18 07:04:35 +01:00
}
else {
2019-10-20 09:19:37 +02:00
// Normal function call
p.gen('$cgen_name (')
2019-06-22 20:20:28 +02:00
}
// `foo<Bar>()`
2019-10-20 09:19:37 +02:00
// if f is generic, the name is changed to a suitable instance in dispatch_generic_fn_instance()
// we then replace `cgen_name` with the instance's name
generic := f.is_generic
2019-12-21 01:53:58 +01:00
p.fn_call_args(mut f, generic_param_types)
2019-10-20 09:19:37 +02:00
if generic {
2019-12-19 03:41:12 +01:00
line := if p.cgen.is_tmp { p.cgen.tmp_line } else { p.cgen.cur_line }
p.cgen.resetln(line.replace('$cgen_name (', '$f.name ('))
2019-10-20 09:19:37 +02:00
// println('calling inst $f.name: $p.cgen.cur_line')
}
2019-12-18 07:04:35 +01:00
// if !is_interface {
p.gen(')')
// }
2019-06-22 20:20:28 +02:00
p.calling_c = false
2019-10-27 01:36:43 +02:00
if is_comptime_define {
p.cgen.nogen = false
2019-10-27 10:42:31 +01:00
p.cgen.resetln('')
2019-10-27 01:36:43 +02:00
}
2019-06-22 20:20:28 +02:00
// println('end of fn call typ=$f.typ')
}
// for declaration
2019-11-08 04:03:06 +01:00
// update the Fn object's args[]
2019-06-22 20:20:28 +02:00
fn (p mut Parser) fn_args(f mut Fn) {
p.check(.lpar)
2019-12-18 07:04:35 +01:00
defer {
p.check(.rpar)
}
2019-06-22 20:20:28 +02:00
if f.is_interface {
2019-12-18 07:04:35 +01:00
interface_arg := Var{
2019-06-22 20:20:28 +02:00
typ: f.receiver_typ
token_idx: p.cur_tok_index()
2019-06-22 20:20:28 +02:00
}
2019-11-08 04:03:06 +01:00
f.args << interface_arg
2019-06-22 20:20:28 +02:00
}
2019-08-17 21:19:37 +02:00
// `(int, string, int)`
2019-06-22 20:20:28 +02:00
// Just register fn arg types
2020-02-04 09:54:15 +01:00
types_only := p.tok == .mul || p.tok == .amp || (p.peek() == .comma &&
p.table.known_type(p.lit)) || p.peek() == .rpar // (int, string)
2019-06-22 20:20:28 +02:00
if types_only {
for p.tok != .rpar {
is_mut := p.tok == .key_mut
if is_mut {
p.check(.key_mut)
p.fspace()
}
2019-06-22 20:20:28 +02:00
typ := p.get_type()
2019-12-18 07:04:35 +01:00
if typ == '' {
// && !f.is_c {
2019-11-30 08:37:28 +01:00
if p.prev_tok != .ellipsis {
2019-12-18 07:04:35 +01:00
p.error('bad fn arg type')
2019-11-30 08:37:28 +01:00
}
}
2019-09-23 12:42:20 +02:00
p.check_and_register_used_imported_type(typ)
2019-12-18 07:04:35 +01:00
v := Var{
2019-06-22 20:20:28 +02:00
typ: typ
is_arg: true
is_mut: is_mut
2019-12-22 02:34:37 +01:00
2019-06-22 20:20:28 +02:00
line_nr: p.scanner.line_nr
token_idx: p.cur_tok_index()
2019-06-22 20:20:28 +02:00
}
// f.register_var(v)
f.args << v
if p.tok == .comma {
2019-06-22 20:20:28 +02:00
p.next()
2019-12-18 07:04:35 +01:00
p.fspace()
2019-06-22 20:20:28 +02:00
}
}
}
2019-08-06 18:13:04 +02:00
// `(a int, b, c string)` syntax
for p.tok != .rpar {
2019-12-18 07:04:35 +01:00
mut names := [p.check_name()]
2019-08-06 18:13:04 +02:00
// `a,b,c int` syntax
for p.tok == .comma {
p.check(.comma)
2019-06-22 20:20:28 +02:00
p.fspace()
names << p.check_name()
}
p.fspace()
is_mut := p.tok == .key_mut
2019-06-22 20:20:28 +02:00
if is_mut {
2019-10-20 09:19:37 +02:00
p.check(.key_mut)
2019-12-18 07:04:35 +01:00
p.fspace()
2019-06-22 20:20:28 +02:00
}
// variadic arg
if p.tok == .ellipsis {
p.check(.ellipsis)
if p.tok == .rpar {
p.error('you must provide a type for vargs: eg `...string`. multiple types `...` are not supported yet.')
}
f.is_variadic = true
}
mut typ := p.get_type()
if !p.first_pass() && !p.table.known_type(typ) {
p.error('fn_args: unknown type $typ')
}
if f.is_variadic {
if !f.is_c {
// register varg struct, incase function is never called
if p.first_pass() && !f.is_generic {
p.register_vargs_stuct(typ, 0)
}
typ = 'varg_$typ'
2019-12-18 07:04:35 +01:00
}
else {
typ = '...$typ' // TODO: fix, this is invalid in C
}
}
2019-09-23 12:42:20 +02:00
p.check_and_register_used_imported_type(typ)
2019-07-24 15:24:32 +02:00
if is_mut && is_primitive_type(typ) {
2019-12-18 07:04:35 +01:00
p.error('mutable arguments are only allowed for arrays, maps, and structs.' + '\nreturn values instead: `fn foo(n mut int) {` => `fn foo(n int) int {`')
2019-08-17 21:19:37 +02:00
}
2019-06-22 20:20:28 +02:00
for name in names {
2019-08-17 21:19:37 +02:00
if is_mut {
2019-07-24 15:24:32 +02:00
typ += '*'
2019-06-22 20:20:28 +02:00
}
v := Var{
2019-06-22 20:20:28 +02:00
name: name
2019-08-17 21:19:37 +02:00
typ: typ
2019-06-22 20:20:28 +02:00
is_arg: true
is_mut: is_mut
ptr: is_mut
line_nr: p.scanner.line_nr
token_idx: p.cur_tok_index()
2019-06-22 20:20:28 +02:00
}
2019-09-23 19:34:08 +02:00
p.register_var(v)
2019-06-22 20:20:28 +02:00
f.args << v
}
if p.tok == .comma {
2019-10-20 09:19:37 +02:00
p.check(.comma)
2019-12-18 07:04:35 +01:00
p.fspace()
2019-06-22 20:20:28 +02:00
}
// unnamed (C definition)
if p.tok == .ellipsis {
if !f.is_c {
p.error('variadic argument syntax must be `arg_name ...type` eg `argname ...string`.')
}
2019-12-18 07:04:35 +01:00
f.args << Var{
// name: '...'
typ: '...'
2019-06-22 20:20:28 +02:00
}
p.next()
}
}
2019-12-24 18:54:43 +01:00
//if types_only && p.peek() == .lcbr {
//println('wtf')
//}
2019-06-22 20:20:28 +02:00
}
2019-08-17 21:19:37 +02:00
// foo *(1, 2, 3, mut bar)*
2019-12-21 01:53:58 +01:00
fn (p mut Parser) fn_call_args(f mut Fn, generic_param_types []string) {
2019-06-22 20:20:28 +02:00
// println('fn_call_args() name=$f.name args.len=$f.args.len')
// C func. # of args is not known
p.check(.lpar)
2019-06-22 20:20:28 +02:00
if f.is_c {
for p.tok != .rpar {
2019-12-18 07:04:35 +01:00
// C.func(var1, var2.method())
// If the parameter calls a function or method that is not C,
// the value of p.calling_c is changed
2019-09-16 17:28:20 +02:00
p.calling_c = true
ph := p.cgen.add_placeholder()
typ := p.bool_expression()
// Cast V byteptr to C char* (byte is unsigned in V, that led to C warnings)
if typ == 'byte*' {
p.cgen.set_placeholder(ph, '(char*)')
2019-10-20 09:19:37 +02:00
}
if p.tok == .comma {
2019-06-22 20:20:28 +02:00
p.gen(', ')
p.check(.comma)
2019-11-10 01:08:53 +01:00
p.fspace()
2019-06-22 20:20:28 +02:00
}
}
p.check(.rpar)
2019-10-20 09:19:37 +02:00
return
2019-06-22 20:20:28 +02:00
}
// add debug information to panic when -g arg is passed
2019-09-14 22:48:30 +02:00
if p.v.pref.is_debug && f.name == 'panic' && !p.is_js {
2019-07-30 15:08:14 +02:00
mod_name := p.mod.replace('_dot_', '.')
fn_name := p.cur_fn.name.replace('${p.mod}__', '')
2019-10-25 15:34:12 +02:00
file_path := cescaped_path(p.file_path)
2019-12-18 07:04:35 +01:00
p.cgen.resetln(p.cgen.cur_line.replace('v_panic (', 'panic_debug ($p.scanner.line_nr, tos3("$file_path"), tos3("$mod_name"), tos2((byte *)"$fn_name"), '))
2019-07-30 15:08:14 +02:00
}
2019-12-21 22:54:37 +01:00
// mut saved_args := []string
2019-12-21 01:53:58 +01:00
mut saved_args := generic_param_types
2019-12-18 08:17:07 +01:00
for i, arg in f.args {
// Receiver is the first arg
// Skip the receiver, because it was already generated in the expression
2019-06-22 20:20:28 +02:00
if i == 0 && f.is_method {
2019-12-18 07:04:35 +01:00
if f.args.len > 1 {
// && !p.is_js {
p.gen(', ')
}
2019-12-18 07:04:35 +01:00
// if f.args[0].typ.ends_with('*') {
// p.gen('&/*119*/')
// }
2019-12-21 22:54:37 +01:00
// pos := p.cgen.cur_line.index('/* ? */')
// if pos > -1 {
// expr := p.cgen.cur_line[pos..]
// // TODO hack
// // If current expression is a func call, generate the array hack
// if expr.contains('(') {
// p.cgen.set_placeholder(pos, '(${arg.typ[..arg.typ.len-1]}[]){')
// p.gen('}[0] ')
// }
// }
2019-06-22 20:20:28 +02:00
continue
}
// Reached the final vararg? Quit
if i == f.args.len - 1 && arg.typ.starts_with('varg_') {
2019-06-22 20:20:28 +02:00
break
}
ph := p.cgen.add_placeholder()
2019-08-17 21:19:37 +02:00
// `)` here means that not enough args were provided
if p.tok == .rpar {
2019-11-08 04:03:06 +01:00
p.error('not enough arguments in call to `${f.str_for_error()}`')
2019-06-22 20:20:28 +02:00
}
2019-08-17 21:19:37 +02:00
// If `arg` is mutable, the caller needs to provide `mut`:
2019-07-10 14:26:37 +02:00
// `mut numbers := [1,2,3]; reverse(mut numbers);`
2019-06-22 20:20:28 +02:00
if arg.is_mut {
2019-08-10 23:02:48 +02:00
if p.tok != .key_mut && p.tok == .name {
p.mutable_arg_error(i, arg, f)
2019-06-22 20:20:28 +02:00
}
if p.peek() != .name {
2019-12-18 07:04:35 +01:00
p.error('`$arg.name` is a mutable argument, you need to ' + 'provide a variable to modify: `${f.name}(... mut a...)`')
2019-06-22 20:20:28 +02:00
}
p.check(.key_mut)
2019-11-11 06:58:50 +01:00
p.fspace()
2019-08-17 21:19:37 +02:00
var_name := p.lit
2019-09-23 19:34:08 +02:00
v := p.find_var(var_name) or {
2019-12-18 07:04:35 +01:00
p.error('`$arg.name` is a mutable argument, you need to ' + 'provide a variable to modify: `${f.name}(... mut a...)`')
exit(1)
2019-08-17 21:19:37 +02:00
}
if !v.is_changed {
p.mark_var_changed(v)
2019-08-17 21:19:37 +02:00
}
}
p.expected_type = arg.typ
2019-12-18 07:04:35 +01:00
clone := p.pref.autofree && p.mod != 'string' && arg.typ == 'string' && !p.builtin_mod // && arg.is_moved
if clone {
p.gen('/*YY f=$f.name arg=$arg.name is_moved=$arg.is_moved*/string_clone(')
2019-10-20 09:19:37 +02:00
}
2019-11-19 07:53:52 +01:00
// x64 println gen
2020-03-06 18:53:29 +01:00
if p.pref.backend == .x64 && i == 0 && f.name == 'println' && p.tok == .str && p.peek() == .rpar {
2019-12-31 19:42:16 +01:00
//p.x64.gen_print(p.lit)
}
mut typ := p.bool_expression()
2019-11-08 04:03:06 +01:00
// Register an interface type usage:
// fn run(r Animal) { ... }
// `run(dog)` adds `Dog` to the `Animal` interface.
// This is needed to generate an interface table.
2019-12-22 02:34:37 +01:00
if arg.typ.ends_with('er') || arg.typ[0] == `I` {
2019-12-18 07:04:35 +01:00
t := p.table.find_type(arg.typ)
if t.cat == .interface_ {
// NB: here concrete_type_name can be 'Dog' OR 'Dog_ptr'
// cgen should have generated a _I_Dog_to_Speaker conversion function
// C: perform( _I_Dog_to_Speaker((Dog){...}) )
// In case of _ptr, there is no need for conversion, so the generated
// code will be just:
// C: perform( dog_ptr )
concrete_type_name := typ.replace('*', '_ptr')
// concrete_type_name here can be say Dog, or ui__Group_ptr (in vui)
//eprintln('arg.typ: $arg.typ | concrete_type_name: $concrete_type_name ')
if !concrete_type_name.ends_with('_ptr') {
p.cgen.set_placeholder(ph, 'I_${concrete_type_name}_to_${arg.typ}(')
p.gen(')')
2020-02-27 16:31:06 +01:00
}
2019-12-18 07:04:35 +01:00
p.table.add_gen_type(arg.typ, typ)
}
2019-11-08 04:03:06 +01:00
}
if clone {
p.gen(')')
}
// Optimize `println`: replace it with `printf` to avoid extra allocations and
2019-09-14 22:48:30 +02:00
// function calls.
// `println(777)` => `printf("%d\n", 777)`
2019-08-17 21:19:37 +02:00
// (If we don't check for void, then V will compile `println(func())`)
if i == 0 && (f.name == 'println' || f.name == 'print') && typ == 'ustring' {
if typ == 'ustring' {
p.gen('.s')
}
typ = 'string'
}
2019-12-18 07:04:35 +01:00
if i == 0 && (f.name == 'println' || f.name == 'print') && !(typ in ['string', 'ustring', 'void']) {
2019-11-19 07:53:52 +01:00
//
tt := p.table.find_type(typ)
$if !windows {
2019-12-18 07:04:35 +01:00
$if !js {
fmt := p.typ_to_fmt(typ, 0)
if fmt != '' && typ != 'bool' {
2019-12-19 03:41:12 +01:00
nl := if f.name == 'println' { '\\n' } else { '' }
2019-12-18 07:04:35 +01:00
p.cgen.resetln(p.cgen.cur_line.replace(f.name + ' (', '/*opt*/printf ("' + fmt + '$nl", '))
continue
}
}
}
if typ.ends_with('*') {
p.cgen.set_placeholder(ph, 'ptr_str(')
p.gen(')')
2019-08-17 21:19:37 +02:00
continue
2019-06-23 09:59:34 +02:00
}
// Make sure this type has a `str()` method
2019-09-14 22:48:30 +02:00
$if !js {
if !tt.has_method('str') {
2019-12-18 07:04:35 +01:00
// varg
if tt.name.starts_with('varg_') {
p.gen_varg_str(tt)
2019-12-18 07:04:35 +01:00
p.cgen.set_placeholder(ph, '${typ}_str(')
p.gen(')')
continue
}
// Arrays have automatic `str()` methods
else if tt.name.starts_with('array_') {
p.gen_array_str(tt)
2019-12-18 07:04:35 +01:00
p.cgen.set_placeholder(ph, '${typ}_str(')
p.gen(')')
continue
}
// struct
else if tt.cat == .struct_ {
p.gen_struct_str(tt)
2019-12-18 07:04:35 +01:00
p.cgen.set_placeholder(ph, '${typ}_str(')
p.gen(')')
continue
}
else {
base := p.base_type(tt.name)
if base != tt.name {
2019-12-18 07:04:35 +01:00
base_type := p.find_type(base)
if base_type.has_method('str') {
p.cgen.set_placeholder(ph, '${base_type.name}_str(')
p.gen(')')
continue
}
2019-12-16 17:06:40 +01:00
}
}
2019-12-18 07:04:35 +01:00
error_msg := ('`$typ` needs to have method `str() string` to be printable')
p.error(error_msg)
2019-10-20 09:19:37 +02:00
}
2019-12-18 07:04:35 +01:00
p.cgen.set_placeholder(ph, '${typ}_str(')
p.gen(')')
2019-09-14 22:48:30 +02:00
}
2019-06-22 20:20:28 +02:00
continue
}
got := typ
expected := arg.typ
got_ptr := got.ends_with('*')
exp_ptr := expected.ends_with('*')
2019-06-22 20:20:28 +02:00
// println('fn arg got="$got" exp="$expected"')
2019-10-20 09:19:37 +02:00
type_mismatch := !p.check_types_no_throw(got, expected)
if type_mismatch && f.is_generic {
// println("argument `$arg.name` is generic")
saved_args << got
2019-12-18 07:04:35 +01:00
}
else if type_mismatch {
mut j := i
if f.is_method {
j--
2019-10-20 09:19:37 +02:00
}
2019-12-04 11:36:26 +01:00
mut nr := '${j+1}th'
if j == 0 {
nr = 'first'
2019-12-18 07:04:35 +01:00
}
else if j == 1 {
nr = 'second'
2019-12-18 07:04:35 +01:00
}
else if j == 2 {
nr = 'third'
2019-10-20 09:19:37 +02:00
}
2019-12-18 07:04:35 +01:00
p.error('cannot use type `$typ` as type `$arg.typ` in $nr ' + 'argument to `${f.name}()`')
}
else {
2019-10-20 09:19:37 +02:00
saved_args << ''
2019-06-22 20:20:28 +02:00
}
is_interface := p.table.is_interface(arg.typ)
// Automatically add `&` or `*` before an argument.
// V, unlike C and Go, simplifies this aspect:
// `foo(bar)` is allowed where `foo(&bar)` is expected.
// The argument is not mutable, so it won't be changed by the function.
// It doesn't matter whether it's passed by referencee or by value
// to the end user.
2019-06-22 20:20:28 +02:00
if !is_interface {
// Dereference
if got_ptr && !exp_ptr {
p.cgen.set_placeholder(ph, '*')
2019-06-22 20:20:28 +02:00
}
// Reference
// TODO ptr hacks. DOOM hacks, fix please.
if !got_ptr && exp_ptr && got != 'voidptr' {
2019-09-29 16:02:28 +02:00
// Special case for mutable arrays. We can't `&` function
// results,
2019-08-17 21:19:37 +02:00
// have to use `(array[]){ expr }` hack.
2019-12-18 07:04:35 +01:00
if expected.starts_with('array_') && exp_ptr {
// && !arg.is_mut{
2019-08-17 21:19:37 +02:00
p.cgen.set_placeholder(ph, '& /*111*/ (array[]){')
p.gen('}[0] ')
2019-08-17 21:19:37 +02:00
}
else if exp_ptr && expected == got + '*' {
2019-11-11 04:02:01 +01:00
$if !tinyc {
2019-12-18 07:04:35 +01:00
expr := p.cgen.cur_line[ph..]
// TODO hack
// If current expression is a func call, generate the array hack
if expr.contains('(') {
// println('fn hack expr=$expr')
p.cgen.set_placeholder(ph, '& /*113 e="$expected" g="$got"*/ ($got[]){')
p.gen('}[0] ')
}
else {
p.cgen.set_placeholder(ph, '& /*114*/')
}
2019-11-11 03:33:19 +01:00
} $else {
p.cgen.set_placeholder(ph, '& /*114*/')
}
}
2019-06-22 20:20:28 +02:00
// println('\ne:"$expected" got:"$got"')
2019-12-18 07:04:35 +01:00
else if !(expected == 'void*' && got == 'int') && !(expected == 'void*' && got == 'byteptr') && !(expected == 'byte*' && got.contains(']byte')) && !(expected == 'byte*' && got == 'string') &&
// ! (expected == 'void*' && got == 'array_int') {
!(expected == 'byte*' && got == 'byteptr') && !p.pref.is_bare {
p.cgen.set_placeholder(ph, '& /*112 e="$expected" g="$got" */')
2019-06-22 20:20:28 +02:00
}
}
}
else if is_interface {
if !got_ptr {
2019-12-18 07:04:35 +01:00
// p.cgen.set_placeholder(ph, '&')
2019-06-22 20:20:28 +02:00
}
// Pass all interface methods
2019-12-18 07:04:35 +01:00
// interface_type := p.table.find_type(arg.typ)
// for method in interface_type.methods {
// p.gen(', ${typ}_${method.name} ')
// }
2019-06-22 20:20:28 +02:00
}
// Check for commas
if i < f.args.len - 1 {
// Handle 0 args passed to varargs
if p.tok != .comma && !f.is_variadic {
2019-11-08 04:03:06 +01:00
p.error('wrong number of arguments in call to `${f.str_for_error()}`')
2019-06-22 20:20:28 +02:00
}
2019-12-18 07:04:35 +01:00
if p.tok == .comma && (!f.is_variadic || (f.is_variadic && i < f.args.len - 2)) {
p.check(.comma)
2019-11-10 01:08:53 +01:00
p.fspace()
2019-06-22 20:20:28 +02:00
p.gen(',')
}
}
}
// varargs
2019-12-18 07:04:35 +01:00
varg_type,varg_values := p.fn_call_vargs(f)
if f.is_variadic {
2019-10-23 18:44:05 +02:00
saved_args << varg_type
2019-06-22 20:20:28 +02:00
}
if p.tok == .comma {
2019-11-08 04:03:06 +01:00
p.error('wrong number of arguments in call to `${f.str_for_error()}`')
2019-06-22 20:20:28 +02:00
}
p.check(.rpar)
2019-12-18 18:07:32 +01:00
if f.is_generic && !p.scanner.is_fmt {
2019-10-20 09:19:37 +02:00
type_map := p.extract_type_inst(f, saved_args)
p.dispatch_generic_fn_instance(mut f, &type_map)
2019-10-20 09:19:37 +02:00
}
if f.is_variadic {
p.fn_gen_caller_vargs(f, varg_type, varg_values)
}
2019-10-20 09:19:37 +02:00
}
// From a given generic function and an argument list matching its signature,
// create a type instantiation
fn (p mut Parser) extract_type_inst(f &Fn, args_ []string) TypeInst {
2019-12-18 07:04:35 +01:00
mut r := TypeInst{
}
2019-10-20 09:19:37 +02:00
mut i := 0
mut args := args_
2019-12-18 07:04:35 +01:00
if f.typ != 'void' {
args << f.typ
}
2019-10-25 20:32:27 +02:00
for e in args {
2019-12-18 07:04:35 +01:00
if e == '' {
continue
}
2019-10-20 09:19:37 +02:00
tp := f.type_pars[i]
mut ti := e
if ti.starts_with('fn (') {
fn_args := ti[4..].all_before(') ').split(',')
2019-10-20 09:19:37 +02:00
mut found := false
for fa_ in fn_args {
mut fa := fa_
2019-12-18 07:04:35 +01:00
for fa.starts_with('array_') {
fa = fa[6..]
}
2019-10-20 09:19:37 +02:00
if fa == tp {
r.inst[tp] = fa
found = true
2019-12-08 12:11:19 +01:00
i++
2019-10-20 09:19:37 +02:00
break
}
}
2019-12-18 07:04:35 +01:00
if found {
continue
}
2019-10-20 09:19:37 +02:00
ti = ti.all_after(') ')
}
2019-12-18 07:04:35 +01:00
for ti.starts_with('array_') {
ti = ti[6..]
}
2019-10-20 09:19:37 +02:00
if r.inst[tp] != '' {
if r.inst[tp] != ti {
p.error('type parameter `$tp` has type ${r.inst[tp]}, not `$ti`')
}
continue
}
// println("extracted $tp => $ti")
r.inst[tp] = ti
2019-12-08 12:11:19 +01:00
i++
2019-12-18 07:04:35 +01:00
if i >= f.type_pars.len {
break
}
2019-10-20 09:19:37 +02:00
}
if r.inst[f.typ] == '' && f.typ in f.type_pars {
r.inst[f.typ] = '_ANYTYPE_'
}
2019-10-25 20:32:27 +02:00
for tp in f.type_pars {
if r.inst[tp] == '' {
// p.error_with_token_index('unused type parameter `$tp`', f.body_idx-2)
p.error('unused type parameter `$tp`')
2019-10-25 20:32:27 +02:00
}
}
2019-10-20 09:19:37 +02:00
return r
}
// replace a generic type using TypeInst
fn replace_generic_type(gen_type string, ti &TypeInst) string {
mut typ := gen_type.replace('map_', '').replace('varg_', '').trim_right('*').replace('ptr_', '')
2019-12-18 07:04:35 +01:00
for typ.starts_with('array_') {
typ = typ[6..]
}
if typ in ti.inst {
typ = gen_type.replace(typ, ti.inst[typ])
return typ
2019-12-18 07:04:35 +01:00
}
typ = gen_type
if typ.starts_with('fn (') {
args := typ[4..].all_before_last(')').split(',')
ret_t := typ.all_after(')').trim_space()
mut args_r := []string
for arg in args {
args_r << replace_generic_type(arg, ti)
}
mut t := 'fn (' + args_r.join(',') + ')'
if ret_t.len > 0 {
t += ' ' + replace_generic_type(ret_t, ti)
}
typ = t
}
return typ
}
// replace return type & param types for a given generic function using TypeInst
fn replace_generic_type_params(f mut Fn, ti &TypeInst) {
2019-12-18 07:04:35 +01:00
mut args := []Var
2019-12-18 08:17:07 +01:00
for i, _ in f.args {
2019-12-18 07:04:35 +01:00
mut arg := f.args[i]
arg.typ = replace_generic_type(arg.typ, ti)
args << arg
}
f.args = args
f.typ = replace_generic_type(f.typ, ti)
2019-12-21 01:53:58 +01:00
if f.typ.ends_with('_T') {
par := ti.inst.keys()[0]
f.typ = f.typ + '_' + ti.inst[par]
}
2019-06-22 20:20:28 +02:00
}
fn (p mut Parser) register_vargs_stuct(typ string, len int) string {
vargs_struct := 'varg_$typ'
varg_type := Type{
2019-11-08 04:03:06 +01:00
cat: .struct_
name: vargs_struct
mod: p.mod
}
mut varg_len := len
if !p.table.known_type(vargs_struct) {
2019-12-05 12:09:33 +01:00
p.table.register_type(varg_type)
p.cgen.typedefs << 'typedef struct $vargs_struct $vargs_struct;\n'
2019-12-18 07:04:35 +01:00
}
else {
ex_typ := p.table.find_type(vargs_struct)
2019-12-18 07:04:35 +01:00
ex_len := ex_typ.fields[1].name[5..ex_typ.fields[1].name.len - 1].int()
if ex_len > varg_len {
varg_len = ex_len
}
p.table.rewrite_type(varg_type)
}
p.table.add_field(vargs_struct, 'len', 'int', false, '', .public)
p.table.add_field(vargs_struct, 'args[$varg_len]', typ, false, '', .public)
return vargs_struct
}
2019-12-18 07:04:35 +01:00
fn (p mut Parser) fn_call_vargs(f Fn) (string,[]string) {
if !f.is_variadic {
2019-12-18 07:04:35 +01:00
return '',[]string
}
last_arg := f.args.last()
2019-12-18 07:04:35 +01:00
// varg_def_type := last_arg.typ[3..]
mut types := []string
mut values := []string
for p.tok != .rpar {
if p.tok == .comma {
p.check(.comma)
}
2019-12-18 07:04:35 +01:00
varg_type,varg_value := p.tmp_expr()
if varg_type.starts_with('varg_') && (values.len > 0 || p.tok == .comma) {
p.error('You cannot pass additional vargs when forwarding vargs to another function/method')
2019-11-11 03:14:54 +01:00
}
if !f.is_generic {
p.check_types(last_arg.typ, varg_type)
2019-12-18 07:04:35 +01:00
}
else {
if types.len > 0 {
for t in types {
p.check_types(varg_type, t)
}
}
}
2019-12-19 03:41:12 +01:00
ref_deref := if last_arg.typ.ends_with('*') && !varg_type.ends_with('*') { '&' } else if !last_arg.typ.ends_with('*') && varg_type.ends_with('*') { '*' } else { '' }
types << varg_type
values << '$ref_deref$varg_value'
}
for va in p.table.varg_access {
2019-12-18 07:04:35 +01:00
if va.fn_name != f.name {
continue
}
if va.index >= values.len {
p.error_with_token_index('variadic arg index out of range: $va.index/${values.len-1}, vargs are 0 indexed', va.tok_idx)
}
}
2019-11-01 11:03:14 +01:00
if !f.is_method && f.args.len > 1 {
p.cgen.gen(',')
}
2019-12-18 07:04:35 +01:00
return types[0],values
}
2019-10-20 09:19:37 +02:00
fn (p mut Parser) fn_gen_caller_vargs(f &Fn, varg_type string, values []string) {
is_varg := varg_type.starts_with('varg_')
2019-12-18 07:04:35 +01:00
if is_varg {
// forwarding varg
p.cgen.gen('${values[0]}')
2019-12-18 07:04:35 +01:00
}
else {
vargs_struct := p.register_vargs_stuct(varg_type, values.len)
2019-12-18 07:04:35 +01:00
p.cgen.gen('&($vargs_struct){.len=$values.len,.args={' + values.join(',') + '}}')
}
}
fn (p mut Parser) register_multi_return_stuct(types []string) string {
typ := '_V_MulRet_' + types.join('_V_').replace('*', '_PTR_')
2019-12-18 07:04:35 +01:00
if p.table.known_type(typ) {
return typ
}
2019-12-05 12:09:33 +01:00
p.table.register_type(Type{
2019-12-18 07:04:35 +01:00
cat: .struct_
name: typ
mod: p.mod
})
2019-12-18 08:17:07 +01:00
for i, t in typ.replace('_V_MulRet_', '').replace('_PTR_', '*').split('_V_') {
p.table.add_field(typ, 'var_$i', t, false, '', .public)
}
p.cgen.typedefs << 'typedef struct $typ $typ;'
return typ
}
fn rename_generic_fn_instance(f mut Fn, ti &TypeInst) {
f.name = f.name + '_T'
for k in ti.inst.keys() {
f.name = f.name + '_' + type_to_safe_str(ti.inst[k])
}
}
fn (p mut Parser) dispatch_generic_fn_instance(f mut Fn, ti &TypeInst) {
2019-10-20 09:19:37 +02:00
mut new_inst := true
for e in f.type_inst {
if e.inst.str() == ti.inst.str() {
new_inst = false
break
}
}
if !new_inst {
rename_generic_fn_instance(mut f, ti)
replace_generic_type_params(mut f, ti)
_ = p.table.find_fn(f.name) or {
p.error('function instance `$f.name` not found')
2019-10-20 09:19:37 +02:00
return
}
// println('using existing inst ${p.fn_signature_v(f)}')
2019-10-20 09:19:37 +02:00
return
}
f.type_inst << *ti
2019-10-20 09:19:37 +02:00
p.table.register_fn(f)
2019-12-21 22:54:37 +01:00
if f.is_method {
f.name = f.receiver_typ + '_' + f.name
}
rename_generic_fn_instance(mut f, ti)
replace_generic_type_params(mut f, ti)
// TODO: Handle case where type not defined yet, see above
// if f.typ in f.type_pars { f.typ = '_ANYTYPE_' }
// if f.typ in ti.inst {
2019-12-18 07:04:35 +01:00
// f.typ = ti.inst[f.typ]
// }
2019-10-25 20:32:27 +02:00
if f.is_method {
2019-11-23 08:06:14 +01:00
// TODO: add_method won't add anything on second pass
// p.add_method(f.args[0].typ.trim_right('*'), f)
2019-12-18 07:04:35 +01:00
}
else {
2019-10-25 20:32:27 +02:00
p.table.register_fn(f)
}
2019-11-29 14:46:43 +01:00
mut gp := p.v.parsers[f.parser_idx]
gp.is_vgen = true
saved_state := p.save_state()
p.clear_state(false, true)
gp.token_idx = f.generic_fn_idx
2019-12-21 01:53:58 +01:00
gp.generic_dispatch = *ti
2019-11-29 14:46:43 +01:00
gp.next()
gp.fn_decl()
2019-12-21 22:54:37 +01:00
gp.generic_dispatch = TypeInst{
}
2019-11-29 14:46:43 +01:00
p.cgen.lines_extra << p.cgen.lines
p.restore_state(saved_state, false, true)
p.cgen.fns << '${p.fn_signature(f)};'
2019-10-20 09:19:37 +02:00
}
2019-06-22 20:20:28 +02:00
// "fn (int, string) int"
fn (f &Fn) typ_str() string {
mut sb := strings.new_builder(50)
2019-06-22 20:20:28 +02:00
sb.write('fn (')
2019-12-18 08:17:07 +01:00
for i, arg in f.args {
2019-06-22 20:20:28 +02:00
sb.write(arg.typ)
if i < f.args.len - 1 {
sb.write(',')
}
}
sb.write(')')
if f.typ != 'void' {
sb.write(' $f.typ')
}
return sb.str()
}
// f.args => "int a, string b"
2019-09-01 21:51:16 +02:00
fn (f &Fn) str_args(table &Table) string {
2019-06-22 20:20:28 +02:00
mut s := ''
2019-12-18 08:17:07 +01:00
for i, arg in f.args {
2019-06-22 20:20:28 +02:00
// Interfaces are a special case. We need to pass the object + pointers
// to all methods:
// fn handle(r Runner) { =>
// void handle(void *r, void (*Runner_run)(void*)) {
2019-12-18 07:04:35 +01:00
/*
2019-06-22 20:20:28 +02:00
if table.is_interface(arg.typ) {
// First the object (same name as the interface argument)
s += ' void* $arg.name'
// Now all methods
interface_type := table.find_type(arg.typ)
for method in interface_type.methods {
s += ', $method.typ (*${arg.typ}_${method.name})(void*'
if method.args.len > 1 {
for a in method.args[1..] {
s += ', $a.typ'
}
}
s += ')'
2019-06-22 20:20:28 +02:00
}
}
2019-11-08 04:03:06 +01:00
*/
if arg.typ.starts_with('varg_') {
s += '$arg.typ *$arg.name'
2019-06-22 20:20:28 +02:00
}
else {
// s += '$arg.typ $arg.name'
2019-12-18 07:04:35 +01:00
s += table.cgen_name_type_pair(arg.name, arg.typ) // '$arg.typ $arg.name'
2019-06-22 20:20:28 +02:00
}
if i < f.args.len - 1 {
s += ', '
}
}
return s
}
// f.args => "a, b, c"
fn (f &Fn) str_args_without_types(table &Table) string {
mut res := []string
for arg in f.args {
res << arg.name
}
return res.join(', ')
}
// find local function variable with closest name to `name`
2019-09-23 19:34:08 +02:00
fn (p &Parser) find_misspelled_local_var(name string, min_match f32) string {
2019-09-13 15:15:30 +02:00
mut closest := f32(0)
mut closest_var := ''
2019-09-23 19:34:08 +02:00
for var in p.local_vars {
if var.scope_level > p.cur_fn.scope_level {
continue
}
2019-09-15 19:07:12 +02:00
n := name.all_after('.')
2019-12-18 07:04:35 +01:00
if var.name == '' || (n.len - var.name.len > 2 || var.name.len - n.len > 2) {
continue
}
2019-10-25 15:34:12 +02:00
c := strings.dice_coefficient(var.name, n)
if c > closest {
closest = c
2019-09-15 19:07:12 +02:00
closest_var = var.name
}
}
2019-12-19 03:41:12 +01:00
return if closest >= min_match { closest_var } else { '' }
}
fn (fns []Fn) contains(f Fn) bool {
for ff in fns {
if ff.name == f.name {
return true
2019-10-20 09:19:37 +02:00
}
}
return false
2019-10-20 09:19:37 +02:00
}
2019-12-18 07:04:35 +01:00
fn (p &Parser) fn_signature(f &Fn) string {
2019-12-08 20:22:47 +01:00
return '$f.typ ${f.name}(${f.str_args(p.table)})'
}
2019-11-08 04:03:06 +01:00
pub fn (f &Fn) v_fn_module() string {
return f.mod
}
2019-11-08 04:03:06 +01:00
pub fn (f &Fn) v_fn_name() string {
return f.name.replace('${f.mod}__', '')
}
2019-11-08 04:03:06 +01:00
pub fn (f &Fn) str_for_error() string {
// Build the args for the error
2019-11-23 08:06:14 +01:00
mut s := ''
2019-12-18 08:17:07 +01:00
for i, a in f.args {
2019-11-23 08:06:14 +01:00
if i == 0 {
if f.is_method {
s += a.typ + '.' + f.name + '('
continue
}
s += f.name + '('
}
s += a.typ
if i < f.args.len - 1 {
s += ', '
}
}
return s + ')'
2019-11-08 04:03:06 +01:00
}