Line | Branch | Exec | Source |
---|---|---|---|
1 | #include "gwion_util.h" | ||
2 | #include "gwion_ast.h" | ||
3 | #include "gwion_env.h" | ||
4 | #include "vm.h" | ||
5 | #include "instr.h" | ||
6 | #include "traverse.h" | ||
7 | #include "template.h" | ||
8 | #include "gwion.h" | ||
9 | #include "object.h" | ||
10 | #include "operator.h" | ||
11 | #include "import.h" | ||
12 | #include "parse.h" | ||
13 | #include "emit.h" | ||
14 | #include "specialid.h" | ||
15 | #include "tmp_resolve.h" | ||
16 | #include "spread.h" | ||
17 | |||
18 | struct ResolverArgs { | ||
19 | const Value v; | ||
20 | Exp_Call *const e; | ||
21 | const m_str tmpl_name; | ||
22 | const Type_List types; | ||
23 | }; | ||
24 | |||
25 | 160 | ANN static inline Value template_get_ready(const Env env, const Value v, | |
26 | const m_str tmpl, const m_uint i) { | ||
27 | 160 | const Symbol sym = func_symbol(env, v->from->owner->name, v->name, tmpl, i); | |
28 | 99 | return v->from->owner_class ? find_value(v->from->owner_class, sym) | |
29 |
2/2✓ Branch 0 taken 99 times.
✓ Branch 1 taken 61 times.
|
259 | : nspc_lookup_value1(v->from->owner, sym); |
30 | } | ||
31 | |||
32 | ANN static inline bool | ||
33 | 75 | tmpl_valid(const Env env, const Func_Def fdef, const m_str filename) { | |
34 |
2/2✓ Branch 1 taken 32 times.
✓ Branch 2 taken 43 times.
|
75 | if (safe_fflag(fdef->base->func, fflag_valid)) return true; |
35 | 43 | const m_str old_file = env->name; | |
36 | 43 | env->name = filename; | |
37 | 43 | const bool ret = check_traverse_fdef(env, fdef) > 0; | |
38 | 43 | env->name = old_file; | |
39 |
2/2✓ Branch 0 taken 6 times.
✓ Branch 1 taken 37 times.
|
43 | if(!fdef->base->func) free_func_def(env->gwion->mp, fdef); |
40 | 43 | return ret; | |
41 | } | ||
42 | |||
43 | 75 | ANN static Func ensure_tmpl(const Env env, const Func_Def fdef, | |
44 | Exp_Call *const exp, const m_str filename) { | ||
45 |
2/2✓ Branch 1 taken 7 times.
✓ Branch 2 taken 68 times.
|
75 | if (!tmpl_valid(env, fdef, filename)) return NULL; |
46 |
3/4✓ Branch 0 taken 68 times.
✗ Branch 1 not taken.
✓ Branch 2 taken 3 times.
✓ Branch 3 taken 65 times.
|
68 | if(env->context && env->context->error) return NULL; |
47 |
3/4✓ Branch 0 taken 64 times.
✓ Branch 1 taken 1 times.
✗ Branch 2 not taken.
✓ Branch 3 taken 64 times.
|
65 | if (exp->args && !exp->args->type) return NULL; |
48 | 65 | const Func f = fdef->base->func; | |
49 | 65 | const Tmpl tmpl = {.list = fdef->base->tmpl->list, .call = exp->tmpl->call}; | |
50 |
1/2✗ Branch 1 not taken.
✓ Branch 2 taken 65 times.
|
65 | CHECK_BO(template_push_types(env, &tmpl)); |
51 | 65 | const Func func = find_func_match(env, f, exp); | |
52 | 65 | nspc_pop_type(env->gwion->mp, env->curr); | |
53 |
2/2✓ Branch 0 taken 48 times.
✓ Branch 1 taken 17 times.
|
65 | if (func) |
54 | 48 | call_add_effect(env, func, exp->func->pos); | |
55 | 65 | return func; | |
56 | } | ||
57 | |||
58 | ✗ | ANN static inline Func ensure_fptr(const Env env, struct ResolverArgs *ra, | |
59 | const Fptr_Def fptr) { | ||
60 | ✗ | CHECK_BO(traverse_fptr_def(env, fptr)); | |
61 | ✗ | const Func_Def fdef = mp_vector_at(fptr->cdef->base.type->info->cdef->body, struct Section_ , 0)->d.func_def; | |
62 | ✗ | return find_func_match(env, fdef->base->func, ra->e); | |
63 | } | ||
64 | |||
65 | ✗ | ANN static Func fptr_match(const Env env, struct ResolverArgs *ra) { | |
66 | ✗ | const Value v = ra->v; | |
67 | const Symbol sym = | ||
68 | ✗ | func_symbol(env, v->from->owner->name, v->name, ra->tmpl_name, 0); | |
69 | ✗ | const Value exists = nspc_lookup_value0(v->from->owner, sym); | |
70 | ✗ | if(exists) { | |
71 | ✗ | const Type t = actual_type(env->gwion, exists->type); | |
72 | ✗ | const Func_Def fdef = mp_vector_at(t->info->cdef->body, struct Section_ , 0)->d.func_def; | |
73 | ✗ | return find_func_match(env, fdef->base->func, ra->e); | |
74 | } | ||
75 | ✗ | const Func_Def base = | |
76 | ✗ | v->d.func_ref ? v->d.func_ref->def : ra->e->func->type->info->func->def; | |
77 | ✗ | const Tmpl tmpl = {.list = base->base->tmpl->list, .call = ra->types}; | |
78 | ✗ | CHECK_BO(template_push_types(env, &tmpl)); | |
79 | ✗ | Func_Base *const fbase = cpy_func_base(env->gwion->mp, base->base); | |
80 | ✗ | fbase->xid = sym; | |
81 | ✗ | fbase->tmpl->call = cpy_type_list(env->gwion->mp, ra->types); | |
82 | ✗ | const Fptr_Def fptr = new_fptr_def(env->gwion->mp, fbase); | |
83 | ✗ | const Func m_func = ensure_fptr(env, ra, fptr); | |
84 | ✗ | if (m_func) | |
85 | ✗ | nspc_add_type_front(v->from->owner, sym, | |
86 | ✗ | actual_type(env->gwion, m_func->value_ref->type)); | |
87 | ✗ | free_fptr_def(env->gwion->mp, fptr); | |
88 | ✗ | nspc_pop_type(env->gwion->mp, env->curr); | |
89 | ✗ | return m_func; | |
90 | } | ||
91 | |||
92 | 38 | ANN static Func tmpl_exists(const Env env, struct ResolverArgs *ra, | |
93 | const Value exists) { | ||
94 |
2/2✓ Branch 0 taken 6 times.
✓ Branch 1 taken 32 times.
|
38 | if (env->func == exists->d.func_ref) |
95 |
2/2✓ Branch 1 taken 5 times.
✓ Branch 2 taken 1 times.
|
6 | return find_func_match(env, env->func, ra->e) ? env->func : NULL; |
96 | 32 | return ensure_tmpl(env, exists->d.func_ref->def, ra->e, ra->v->from->filename); | |
97 | } | ||
98 | |||
99 | 61 | ANN static Func create_tmpl(const Env env, struct ResolverArgs *ra, | |
100 | const m_uint i) { | ||
101 |
2/2✓ Branch 1 taken 18 times.
✓ Branch 2 taken 43 times.
|
61 | DECL_OO(const Value, value, = template_get_ready(env, ra->v, "template", i)); |
102 |
2/2✓ Branch 0 taken 4 times.
✓ Branch 1 taken 39 times.
|
43 | if (value->d.func_ref->def->builtin) set_vflag(value, vflag_builtin); |
103 | 43 | const Func_Def fdef = cpy_func_def(env->gwion->mp, value->d.func_ref->def); | |
104 | 43 | fdef->base->tmpl->call = cpy_type_list(env->gwion->mp, ra->types); | |
105 | |||
106 | 43 | fdef->vt_index = i; | |
107 |
2/2✓ Branch 1 taken 2 times.
✓ Branch 2 taken 41 times.
|
43 | if(is_spread_tmpl(value->d.func_ref->def->base->tmpl)) { |
108 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 2 times.
|
2 | Arg_List args = fdef->base->args ?: new_mp_vector(env->gwion->mp, Arg, 0); |
109 |
2/2✓ Branch 0 taken 2 times.
✓ Branch 1 taken 2 times.
|
4 | for(uint32_t idx = 0; idx < ra->types->len; idx++) { |
110 | char c[256]; | ||
111 | 2 | sprintf(c, "arg%u", idx); | |
112 | 2 | Type_Decl *td = *mp_vector_at(ra->types, Type_Decl*, idx); | |
113 | 2 | Arg arg = { .td = cpy_type_decl(env->gwion->mp, td), .var_decl = {.xid = insert_symbol(c), /*.value = v*/ }}; | |
114 |
1/2✓ Branch 0 taken 2 times.
✗ Branch 1 not taken.
|
2 | mp_vector_add(env->gwion->mp, &args, Arg, arg); |
115 | } | ||
116 | 2 | fdef->base->args = args; | |
117 | } | ||
118 | |||
119 | 43 | const Func func = ensure_tmpl(env, fdef, ra->e, ra->v->from->filename); | |
120 |
4/4✓ Branch 0 taken 32 times.
✓ Branch 1 taken 11 times.
✓ Branch 2 taken 1 times.
✓ Branch 3 taken 31 times.
|
43 | if (func && func->def->builtin) { |
121 | 1 | builtin_func(env->gwion, func, (void*)ra->v->d.func_ref->code->native_func); | |
122 | 1 | set_vflag(func->value_ref, vflag_builtin); | |
123 | 1 | struct Op_Import opi = { .lhs = ra->v->d.func_ref->value_ref->type, .rhs = func->value_ref->type }; | |
124 | 1 | op_cpy(env, &opi); | |
125 | } | ||
126 | 43 | return func; | |
127 | } | ||
128 | |||
129 | 83 | ANN static Func func_match(const Env env, struct ResolverArgs *ra) { | |
130 |
2/2✓ Branch 0 taken 99 times.
✓ Branch 1 taken 30 times.
|
129 | for (m_uint i = 0; i < ra->v->from->offset + 1; ++i) { |
131 | 99 | const Value exists = template_get_ready(env, ra->v, ra->tmpl_name, i); | |
132 | 99 | const Func func = exists | |
133 | 38 | ? tmpl_exists(env, ra, exists) | |
134 |
2/2✓ Branch 0 taken 38 times.
✓ Branch 1 taken 61 times.
|
99 | : create_tmpl(env, ra, i); |
135 |
2/2✓ Branch 0 taken 53 times.
✓ Branch 1 taken 46 times.
|
99 | if (func) return func; |
136 | } | ||
137 | 30 | return NULL; | |
138 | } | ||
139 | |||
140 | 83 | ANN static Func find_tmpl(const Env env, const Value v, Exp_Call *const exp, | |
141 | const m_str tmpl_name) { | ||
142 | 83 | const Type_List types = exp->tmpl->call; | |
143 | 83 | const Func former = env->func; | |
144 | 83 | const m_uint scope = env->scope->depth; | |
145 | 83 | struct EnvSet es = {.env = env, | |
146 | .data = env, | ||
147 | .func = (_exp_func)check_cdef, | ||
148 | .scope = scope, | ||
149 | .flag = tflag_check}; | ||
150 | 83 | struct ResolverArgs ra = { | |
151 | .v = v, .e = exp, .tmpl_name = tmpl_name, .types = types}; | ||
152 |
1/2✗ Branch 1 not taken.
✓ Branch 2 taken 83 times.
|
83 | CHECK_BO(envset_pushv(&es, v)); |
153 | 83 | (void)env_push(env, v->from->owner_class, v->from->owner); | |
154 |
3/4✓ Branch 0 taken 49 times.
✓ Branch 1 taken 34 times.
✓ Branch 2 taken 49 times.
✗ Branch 3 not taken.
|
132 | const bool in_tmpl = v->from->owner_class && v->from->owner_class->info->cdef && |
155 |
2/2✓ Branch 0 taken 7 times.
✓ Branch 1 taken 42 times.
|
49 | v->from->owner_class->info->cdef->base.tmpl; |
156 |
2/2✓ Branch 0 taken 7 times.
✓ Branch 1 taken 76 times.
|
83 | if(in_tmpl) |
157 | 7 | (void)template_push_types(env, v->from->owner_class->info->cdef->base.tmpl); | |
158 | 83 | const bool is_clos = isa(exp->func->type, env->gwion->type[et_closure]) > 0; | |
159 | 83 | const Func m_func = !is_clos ? func_match(env, &ra) | |
160 |
1/2✓ Branch 0 taken 83 times.
✗ Branch 1 not taken.
|
83 | : fptr_match(env, &ra); |
161 |
2/2✓ Branch 0 taken 7 times.
✓ Branch 1 taken 76 times.
|
83 | if(in_tmpl) |
162 | 7 | nspc_pop_type(env->gwion->mp, env->curr); | |
163 | 83 | env_pop(env, scope); | |
164 | 83 | envset_pop(&es, v->from->owner_class); | |
165 | 83 | env->func = former; | |
166 | 83 | return m_func; | |
167 | } | ||
168 | |||
169 | 84 | ANN static Func __find_template_match(const Env env, const Value v, | |
170 | Exp_Call *const exp) { | ||
171 |
2/2✓ Branch 1 taken 1 times.
✓ Branch 2 taken 83 times.
|
84 | DECL_OO(const m_str, tmpl_name, |
172 | = tl2str(env->gwion, exp->tmpl->call, exp->func->pos)); | ||
173 | 83 | const Func f = find_tmpl(env, v, exp, tmpl_name); | |
174 | 83 | free_mstr(env->gwion->mp, tmpl_name); | |
175 | 83 | return f; | |
176 | } | ||
177 | |||
178 | 84 | ANN static Func _find_template_match(const Env env, const Value v, | |
179 | Exp_Call *const exp) { | ||
180 |
2/2✓ Branch 1 taken 31 times.
✓ Branch 2 taken 53 times.
|
84 | DECL_OO(const Func, f, = __find_template_match(env, v, exp)); |
181 | 53 | Type_List tl = exp->tmpl->call; | |
182 | 53 | Specialized_List sl = f->def->base->tmpl->list; | |
183 |
2/2✓ Branch 0 taken 57 times.
✓ Branch 1 taken 53 times.
|
110 | for(uint32_t i = 0; i < tl->len; i++) { |
184 | 57 | Type_Decl *td = *mp_vector_at(tl, Type_Decl*, i); | |
185 |
1/2✗ Branch 1 not taken.
✓ Branch 2 taken 57 times.
|
57 | DECL_OO(const Type, t, = known_type(env, td)); |
186 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 57 times.
|
57 | if(t->info->traits) { |
187 | ✗ | Specialized * spec = mp_vector_at(sl, Specialized, i); | |
188 | ✗ | if (miss_traits(t, spec)) return NULL; | |
189 | } | ||
190 | } | ||
191 | 53 | return f; | |
192 | } | ||
193 | |||
194 | 63 | ANN static inline m_bool check_call(const Env env, const Exp_Call *exp) { | |
195 | 63 | const ae_exp_t et = exp->func->exp_type; | |
196 |
5/6✓ Branch 0 taken 28 times.
✓ Branch 1 taken 35 times.
✓ Branch 2 taken 1 times.
✓ Branch 3 taken 27 times.
✓ Branch 4 taken 1 times.
✗ Branch 5 not taken.
|
63 | if (et != ae_exp_primary && et != ae_exp_dot && et != ae_exp_cast) |
197 | 1 | ERR_B(exp->func->pos, _("invalid expression for function call.")) | |
198 | 62 | return GW_OK; | |
199 | } | ||
200 | |||
201 | 63 | ANN Func find_template_match(const Env env, const Value value, | |
202 | Exp_Call *const exp) { | ||
203 |
2/2✓ Branch 1 taken 1 times.
✓ Branch 2 taken 62 times.
|
63 | CHECK_BO(check_call(env, exp)); |
204 | 62 | const Func f = _find_template_match(env, value, exp); | |
205 |
2/2✓ Branch 0 taken 47 times.
✓ Branch 1 taken 15 times.
|
62 | if (f) return f; |
206 | 15 | Type t = value->from->owner_class; | |
207 |
4/4✓ Branch 0 taken 35 times.
✓ Branch 1 taken 4 times.
✓ Branch 2 taken 30 times.
✓ Branch 3 taken 5 times.
|
39 | while (t && t->nspc) { |
208 | 90 | const Func_Def fdef = value->d.func_ref ? value->d.func_ref->def | |
209 |
1/2✓ Branch 0 taken 30 times.
✗ Branch 1 not taken.
|
30 | : value->type->info->func->def; |
210 | 30 | const Value v = nspc_lookup_value0(t->nspc, fdef->base->xid); | |
211 |
2/2✓ Branch 0 taken 22 times.
✓ Branch 1 taken 8 times.
|
30 | if (v) { |
212 | 22 | const Func f = _find_template_match(env, v, exp); | |
213 |
2/2✓ Branch 0 taken 6 times.
✓ Branch 1 taken 16 times.
|
22 | if (f) return f; |
214 | } | ||
215 | 24 | t = t->info->parent; | |
216 | } | ||
217 | 9 | ERR_O(exp_self(exp)->pos, _("arguments do not match for template call")) | |
218 | } | ||
219 |