xref: /vim-8.2.3635/src/vim9type.c (revision 464393a6)
1a7cc9e69SBram Moolenaar /* vi:set ts=8 sts=4 sw=4 noet:
2a7cc9e69SBram Moolenaar  *
3a7cc9e69SBram Moolenaar  * VIM - Vi IMproved	by Bram Moolenaar
4a7cc9e69SBram Moolenaar  *
5a7cc9e69SBram Moolenaar  * Do ":help uganda"  in Vim to read copying and usage conditions.
6a7cc9e69SBram Moolenaar  * Do ":help credits" in Vim to see a list of people who contributed.
7a7cc9e69SBram Moolenaar  * See README.txt for an overview of the Vim source code.
8a7cc9e69SBram Moolenaar  */
9a7cc9e69SBram Moolenaar 
10a7cc9e69SBram Moolenaar /*
11a7cc9e69SBram Moolenaar  * vim9type.c: handling of types
12a7cc9e69SBram Moolenaar  */
13a7cc9e69SBram Moolenaar 
14a7cc9e69SBram Moolenaar #define USING_FLOAT_STUFF
15a7cc9e69SBram Moolenaar #include "vim.h"
16a7cc9e69SBram Moolenaar 
17a7cc9e69SBram Moolenaar #if defined(FEAT_EVAL) || defined(PROTO)
18a7cc9e69SBram Moolenaar 
19a7cc9e69SBram Moolenaar #ifdef VMS
20a7cc9e69SBram Moolenaar # include <float.h>
21a7cc9e69SBram Moolenaar #endif
22a7cc9e69SBram Moolenaar 
23a7cc9e69SBram Moolenaar /*
24a7cc9e69SBram Moolenaar  * Allocate memory for a type_T and add the pointer to type_gap, so that it can
255e654230SBram Moolenaar  * be easily freed later.
26a7cc9e69SBram Moolenaar  */
278ee52affSYegappan Lakshmanan     static type_T *
get_type_ptr(garray_T * type_gap)285e654230SBram Moolenaar get_type_ptr(garray_T *type_gap)
29a7cc9e69SBram Moolenaar {
30a7cc9e69SBram Moolenaar     type_T *type;
31a7cc9e69SBram Moolenaar 
32a7cc9e69SBram Moolenaar     if (ga_grow(type_gap, 1) == FAIL)
33a7cc9e69SBram Moolenaar 	return NULL;
34a7cc9e69SBram Moolenaar     type = ALLOC_CLEAR_ONE(type_T);
35a7cc9e69SBram Moolenaar     if (type != NULL)
36a7cc9e69SBram Moolenaar     {
37a7cc9e69SBram Moolenaar 	((type_T **)type_gap->ga_data)[type_gap->ga_len] = type;
38a7cc9e69SBram Moolenaar 	++type_gap->ga_len;
39a7cc9e69SBram Moolenaar     }
40a7cc9e69SBram Moolenaar     return type;
41a7cc9e69SBram Moolenaar }
42a7cc9e69SBram Moolenaar 
43a7cc9e69SBram Moolenaar     void
clear_type_list(garray_T * gap)44a7cc9e69SBram Moolenaar clear_type_list(garray_T *gap)
45a7cc9e69SBram Moolenaar {
46a7cc9e69SBram Moolenaar     while (gap->ga_len > 0)
47a7cc9e69SBram Moolenaar 	vim_free(((type_T **)gap->ga_data)[--gap->ga_len]);
48a7cc9e69SBram Moolenaar     ga_clear(gap);
49a7cc9e69SBram Moolenaar }
50a7cc9e69SBram Moolenaar 
515e654230SBram Moolenaar /*
525e654230SBram Moolenaar  * Take a type that is using entries in a growarray and turn it into a type
535e654230SBram Moolenaar  * with allocated entries.
545e654230SBram Moolenaar  */
555e654230SBram Moolenaar     type_T *
alloc_type(type_T * type)565e654230SBram Moolenaar alloc_type(type_T *type)
575e654230SBram Moolenaar {
585e654230SBram Moolenaar     type_T *ret;
595e654230SBram Moolenaar 
605e654230SBram Moolenaar     if (type == NULL)
615e654230SBram Moolenaar 	return NULL;
625e654230SBram Moolenaar 
635e654230SBram Moolenaar     // A fixed type never contains allocated types, return as-is.
645e654230SBram Moolenaar     if (type->tt_flags & TTFLAG_STATIC)
655e654230SBram Moolenaar 	return type;
665e654230SBram Moolenaar 
675e654230SBram Moolenaar     ret = ALLOC_ONE(type_T);
685e654230SBram Moolenaar     *ret = *type;
695e654230SBram Moolenaar 
705e654230SBram Moolenaar     if (ret->tt_member != NULL)
715e654230SBram Moolenaar 	ret->tt_member = alloc_type(ret->tt_member);
725e654230SBram Moolenaar     if (type->tt_args != NULL)
735e654230SBram Moolenaar     {
745e654230SBram Moolenaar 	int i;
755e654230SBram Moolenaar 
765e654230SBram Moolenaar 	ret->tt_args = ALLOC_MULT(type_T *, type->tt_argcount);
775e654230SBram Moolenaar 	if (ret->tt_args != NULL)
785e654230SBram Moolenaar 	    for (i = 0; i < type->tt_argcount; ++i)
795e654230SBram Moolenaar 		ret->tt_args[i] = alloc_type(type->tt_args[i]);
805e654230SBram Moolenaar     }
815e654230SBram Moolenaar 
825e654230SBram Moolenaar     return ret;
835e654230SBram Moolenaar }
845e654230SBram Moolenaar 
855e654230SBram Moolenaar /*
865e654230SBram Moolenaar  * Free a type that was created with alloc_type().
875e654230SBram Moolenaar  */
885e654230SBram Moolenaar     void
free_type(type_T * type)895e654230SBram Moolenaar free_type(type_T *type)
905e654230SBram Moolenaar {
915e654230SBram Moolenaar     int i;
925e654230SBram Moolenaar 
935e654230SBram Moolenaar     if (type == NULL || (type->tt_flags & TTFLAG_STATIC))
945e654230SBram Moolenaar 	return;
955e654230SBram Moolenaar     if (type->tt_args != NULL)
965e654230SBram Moolenaar     {
975e654230SBram Moolenaar 	for (i = 0; i < type->tt_argcount; ++i)
985e654230SBram Moolenaar 	    free_type(type->tt_args[i]);
995e654230SBram Moolenaar 	vim_free(type->tt_args);
1005e654230SBram Moolenaar     }
1015e654230SBram Moolenaar     free_type(type->tt_member);
1025e654230SBram Moolenaar     vim_free(type);
1035e654230SBram Moolenaar }
1045e654230SBram Moolenaar 
105a7cc9e69SBram Moolenaar     type_T *
get_list_type(type_T * member_type,garray_T * type_gap)106a7cc9e69SBram Moolenaar get_list_type(type_T *member_type, garray_T *type_gap)
107a7cc9e69SBram Moolenaar {
108a7cc9e69SBram Moolenaar     type_T *type;
109a7cc9e69SBram Moolenaar 
110a7cc9e69SBram Moolenaar     // recognize commonly used types
1119c13f762SBram Moolenaar     if (member_type == NULL || member_type->tt_type == VAR_ANY)
112a7cc9e69SBram Moolenaar 	return &t_list_any;
113a7cc9e69SBram Moolenaar     if (member_type->tt_type == VAR_VOID
114a7cc9e69SBram Moolenaar 	    || member_type->tt_type == VAR_UNKNOWN)
115a7cc9e69SBram Moolenaar 	return &t_list_empty;
116a7cc9e69SBram Moolenaar     if (member_type->tt_type == VAR_BOOL)
117a7cc9e69SBram Moolenaar 	return &t_list_bool;
118a7cc9e69SBram Moolenaar     if (member_type->tt_type == VAR_NUMBER)
119a7cc9e69SBram Moolenaar 	return &t_list_number;
120a7cc9e69SBram Moolenaar     if (member_type->tt_type == VAR_STRING)
121a7cc9e69SBram Moolenaar 	return &t_list_string;
122a7cc9e69SBram Moolenaar 
123a7cc9e69SBram Moolenaar     // Not a common type, create a new entry.
1245e654230SBram Moolenaar     type = get_type_ptr(type_gap);
125a7cc9e69SBram Moolenaar     if (type == NULL)
126a7cc9e69SBram Moolenaar 	return &t_any;
127a7cc9e69SBram Moolenaar     type->tt_type = VAR_LIST;
128a7cc9e69SBram Moolenaar     type->tt_member = member_type;
129a7cc9e69SBram Moolenaar     type->tt_argcount = 0;
130a7cc9e69SBram Moolenaar     type->tt_args = NULL;
131a7cc9e69SBram Moolenaar     return type;
132a7cc9e69SBram Moolenaar }
133a7cc9e69SBram Moolenaar 
134a7cc9e69SBram Moolenaar     type_T *
get_dict_type(type_T * member_type,garray_T * type_gap)135a7cc9e69SBram Moolenaar get_dict_type(type_T *member_type, garray_T *type_gap)
136a7cc9e69SBram Moolenaar {
137a7cc9e69SBram Moolenaar     type_T *type;
138a7cc9e69SBram Moolenaar 
139a7cc9e69SBram Moolenaar     // recognize commonly used types
1409c13f762SBram Moolenaar     if (member_type == NULL || member_type->tt_type == VAR_ANY)
141a7cc9e69SBram Moolenaar 	return &t_dict_any;
142a7cc9e69SBram Moolenaar     if (member_type->tt_type == VAR_VOID
143a7cc9e69SBram Moolenaar 	    || member_type->tt_type == VAR_UNKNOWN)
144a7cc9e69SBram Moolenaar 	return &t_dict_empty;
145a7cc9e69SBram Moolenaar     if (member_type->tt_type == VAR_BOOL)
146a7cc9e69SBram Moolenaar 	return &t_dict_bool;
147a7cc9e69SBram Moolenaar     if (member_type->tt_type == VAR_NUMBER)
148a7cc9e69SBram Moolenaar 	return &t_dict_number;
149a7cc9e69SBram Moolenaar     if (member_type->tt_type == VAR_STRING)
150a7cc9e69SBram Moolenaar 	return &t_dict_string;
151a7cc9e69SBram Moolenaar 
152a7cc9e69SBram Moolenaar     // Not a common type, create a new entry.
1535e654230SBram Moolenaar     type = get_type_ptr(type_gap);
154a7cc9e69SBram Moolenaar     if (type == NULL)
155a7cc9e69SBram Moolenaar 	return &t_any;
156a7cc9e69SBram Moolenaar     type->tt_type = VAR_DICT;
157a7cc9e69SBram Moolenaar     type->tt_member = member_type;
158a7cc9e69SBram Moolenaar     type->tt_argcount = 0;
159a7cc9e69SBram Moolenaar     type->tt_args = NULL;
160a7cc9e69SBram Moolenaar     return type;
161a7cc9e69SBram Moolenaar }
162a7cc9e69SBram Moolenaar 
163a7cc9e69SBram Moolenaar /*
164a7cc9e69SBram Moolenaar  * Allocate a new type for a function.
165a7cc9e69SBram Moolenaar  */
166a7cc9e69SBram Moolenaar     type_T *
alloc_func_type(type_T * ret_type,int argcount,garray_T * type_gap)167a7cc9e69SBram Moolenaar alloc_func_type(type_T *ret_type, int argcount, garray_T *type_gap)
168a7cc9e69SBram Moolenaar {
1695e654230SBram Moolenaar     type_T *type = get_type_ptr(type_gap);
170a7cc9e69SBram Moolenaar 
171a7cc9e69SBram Moolenaar     if (type == NULL)
172a7cc9e69SBram Moolenaar 	return &t_any;
173a7cc9e69SBram Moolenaar     type->tt_type = VAR_FUNC;
174831bdf86SBram Moolenaar     type->tt_member = ret_type == NULL ? &t_unknown : ret_type;
175a7cc9e69SBram Moolenaar     type->tt_argcount = argcount;
176a7cc9e69SBram Moolenaar     type->tt_args = NULL;
177a7cc9e69SBram Moolenaar     return type;
178a7cc9e69SBram Moolenaar }
179a7cc9e69SBram Moolenaar 
180a7cc9e69SBram Moolenaar /*
181a7cc9e69SBram Moolenaar  * Get a function type, based on the return type "ret_type".
182a7cc9e69SBram Moolenaar  * If "argcount" is -1 or 0 a predefined type can be used.
183a7cc9e69SBram Moolenaar  * If "argcount" > 0 always create a new type, so that arguments can be added.
184a7cc9e69SBram Moolenaar  */
185a7cc9e69SBram Moolenaar     type_T *
get_func_type(type_T * ret_type,int argcount,garray_T * type_gap)186a7cc9e69SBram Moolenaar get_func_type(type_T *ret_type, int argcount, garray_T *type_gap)
187a7cc9e69SBram Moolenaar {
188a7cc9e69SBram Moolenaar     // recognize commonly used types
189a7cc9e69SBram Moolenaar     if (argcount <= 0)
190a7cc9e69SBram Moolenaar     {
191831bdf86SBram Moolenaar 	if (ret_type == &t_unknown || ret_type == NULL)
192a7cc9e69SBram Moolenaar 	{
193a7cc9e69SBram Moolenaar 	    // (argcount == 0) is not possible
194a7cc9e69SBram Moolenaar 	    return &t_func_unknown;
195a7cc9e69SBram Moolenaar 	}
196a7cc9e69SBram Moolenaar 	if (ret_type == &t_void)
197a7cc9e69SBram Moolenaar 	{
198a7cc9e69SBram Moolenaar 	    if (argcount == 0)
199a7cc9e69SBram Moolenaar 		return &t_func_0_void;
200a7cc9e69SBram Moolenaar 	    else
201a7cc9e69SBram Moolenaar 		return &t_func_void;
202a7cc9e69SBram Moolenaar 	}
203a7cc9e69SBram Moolenaar 	if (ret_type == &t_any)
204a7cc9e69SBram Moolenaar 	{
205a7cc9e69SBram Moolenaar 	    if (argcount == 0)
206a7cc9e69SBram Moolenaar 		return &t_func_0_any;
207a7cc9e69SBram Moolenaar 	    else
208a7cc9e69SBram Moolenaar 		return &t_func_any;
209a7cc9e69SBram Moolenaar 	}
210a7cc9e69SBram Moolenaar 	if (ret_type == &t_number)
211a7cc9e69SBram Moolenaar 	{
212a7cc9e69SBram Moolenaar 	    if (argcount == 0)
213a7cc9e69SBram Moolenaar 		return &t_func_0_number;
214a7cc9e69SBram Moolenaar 	    else
215a7cc9e69SBram Moolenaar 		return &t_func_number;
216a7cc9e69SBram Moolenaar 	}
217a7cc9e69SBram Moolenaar 	if (ret_type == &t_string)
218a7cc9e69SBram Moolenaar 	{
219a7cc9e69SBram Moolenaar 	    if (argcount == 0)
220a7cc9e69SBram Moolenaar 		return &t_func_0_string;
221a7cc9e69SBram Moolenaar 	    else
222a7cc9e69SBram Moolenaar 		return &t_func_string;
223a7cc9e69SBram Moolenaar 	}
224a7cc9e69SBram Moolenaar     }
225a7cc9e69SBram Moolenaar 
226a7cc9e69SBram Moolenaar     return alloc_func_type(ret_type, argcount, type_gap);
227a7cc9e69SBram Moolenaar }
228a7cc9e69SBram Moolenaar 
229a7cc9e69SBram Moolenaar /*
230a7cc9e69SBram Moolenaar  * For a function type, reserve space for "argcount" argument types (including
231a7cc9e69SBram Moolenaar  * vararg).
232a7cc9e69SBram Moolenaar  */
233a7cc9e69SBram Moolenaar     int
func_type_add_arg_types(type_T * functype,int argcount,garray_T * type_gap)234a7cc9e69SBram Moolenaar func_type_add_arg_types(
235a7cc9e69SBram Moolenaar 	type_T	    *functype,
236a7cc9e69SBram Moolenaar 	int	    argcount,
237a7cc9e69SBram Moolenaar 	garray_T    *type_gap)
238a7cc9e69SBram Moolenaar {
239a7cc9e69SBram Moolenaar     // To make it easy to free the space needed for the argument types, add the
240a7cc9e69SBram Moolenaar     // pointer to type_gap.
241a7cc9e69SBram Moolenaar     if (ga_grow(type_gap, 1) == FAIL)
242a7cc9e69SBram Moolenaar 	return FAIL;
243a7cc9e69SBram Moolenaar     functype->tt_args = ALLOC_CLEAR_MULT(type_T *, argcount);
244a7cc9e69SBram Moolenaar     if (functype->tt_args == NULL)
245a7cc9e69SBram Moolenaar 	return FAIL;
246a7cc9e69SBram Moolenaar     ((type_T **)type_gap->ga_data)[type_gap->ga_len] =
247a7cc9e69SBram Moolenaar 						     (void *)functype->tt_args;
248a7cc9e69SBram Moolenaar     ++type_gap->ga_len;
249a7cc9e69SBram Moolenaar     return OK;
250a7cc9e69SBram Moolenaar }
251a7cc9e69SBram Moolenaar 
252a7cc9e69SBram Moolenaar /*
253a7cc9e69SBram Moolenaar  * Get a type_T for a typval_T.
2545e654230SBram Moolenaar  * "type_gap" is used to temporarily create types in.
255f2253963SBram Moolenaar  * When "do_member" is TRUE also get the member type, otherwise use "any".
256a7cc9e69SBram Moolenaar  */
257c1ec0422SBram Moolenaar     static type_T *
typval2type_int(typval_T * tv,int copyID,garray_T * type_gap,int do_member)258f2253963SBram Moolenaar typval2type_int(typval_T *tv, int copyID, garray_T *type_gap, int do_member)
259a7cc9e69SBram Moolenaar {
260ba7c0d7bSBram Moolenaar     type_T  *type;
261*464393a6SBram Moolenaar     type_T  *member_type = NULL;
2625e654230SBram Moolenaar     int	    argcount = 0;
2637a40ff00SBram Moolenaar     int	    min_argcount = 0;
264a7cc9e69SBram Moolenaar 
265a7cc9e69SBram Moolenaar     if (tv->v_type == VAR_NUMBER)
266a7cc9e69SBram Moolenaar 	return &t_number;
267a7cc9e69SBram Moolenaar     if (tv->v_type == VAR_BOOL)
268c1ec0422SBram Moolenaar 	return &t_bool;
269a7cc9e69SBram Moolenaar     if (tv->v_type == VAR_STRING)
270a7cc9e69SBram Moolenaar 	return &t_string;
271*464393a6SBram Moolenaar     if (tv->v_type == VAR_BLOB)
272*464393a6SBram Moolenaar 	return &t_blob;
273a7cc9e69SBram Moolenaar 
274a7cc9e69SBram Moolenaar     if (tv->v_type == VAR_LIST)
275a7cc9e69SBram Moolenaar     {
2765597ba04SBram Moolenaar 	list_T	    *l = tv->vval.v_list;
27741fab3eaSBram Moolenaar 	listitem_T  *li;
27841fab3eaSBram Moolenaar 
279b56c4419SBram Moolenaar 	if (l == NULL || (l->lv_first == NULL && l->lv_type == NULL))
280a7cc9e69SBram Moolenaar 	    return &t_list_empty;
281f2253963SBram Moolenaar 	if (!do_member)
282f2253963SBram Moolenaar 	    return &t_list_any;
283b56c4419SBram Moolenaar 	if (l->lv_type != NULL)
284b56c4419SBram Moolenaar 	    return l->lv_type;
2855597ba04SBram Moolenaar 	if (l->lv_first == &range_list_item)
2865597ba04SBram Moolenaar 	    return &t_list_number;
287108cf015SBram Moolenaar 	if (l->lv_copyID == copyID)
288108cf015SBram Moolenaar 	    // avoid recursion
289108cf015SBram Moolenaar 	    return &t_list_any;
290108cf015SBram Moolenaar 	l->lv_copyID = copyID;
291a7cc9e69SBram Moolenaar 
29241fab3eaSBram Moolenaar 	// Use the common type of all members.
293f2253963SBram Moolenaar 	member_type = typval2type(&l->lv_first->li_tv, copyID, type_gap, TRUE);
2945597ba04SBram Moolenaar 	for (li = l->lv_first->li_next; li != NULL; li = li->li_next)
295f2253963SBram Moolenaar 	    common_type(typval2type(&li->li_tv, copyID, type_gap, TRUE),
29641fab3eaSBram Moolenaar 					  member_type, &member_type, type_gap);
297a7cc9e69SBram Moolenaar 	return get_list_type(member_type, type_gap);
298a7cc9e69SBram Moolenaar     }
299a7cc9e69SBram Moolenaar 
300a7cc9e69SBram Moolenaar     if (tv->v_type == VAR_DICT)
301a7cc9e69SBram Moolenaar     {
302a7cc9e69SBram Moolenaar 	dict_iterator_T iter;
303a7cc9e69SBram Moolenaar 	typval_T	*value;
304108cf015SBram Moolenaar 	dict_T		*d = tv->vval.v_dict;
305a7cc9e69SBram Moolenaar 
306b56c4419SBram Moolenaar 	if (d == NULL || (d->dv_hashtab.ht_used == 0 && d->dv_type == NULL))
307a7cc9e69SBram Moolenaar 	    return &t_dict_empty;
308f2253963SBram Moolenaar 	if (!do_member)
309f2253963SBram Moolenaar 	    return &t_dict_any;
310b56c4419SBram Moolenaar 	if (d->dv_type != NULL)
311b56c4419SBram Moolenaar 	    return d->dv_type;
312108cf015SBram Moolenaar 	if (d->dv_copyID == copyID)
313108cf015SBram Moolenaar 	    // avoid recursion
314108cf015SBram Moolenaar 	    return &t_dict_any;
315108cf015SBram Moolenaar 	d->dv_copyID = copyID;
316a7cc9e69SBram Moolenaar 
31741fab3eaSBram Moolenaar 	// Use the common type of all values.
318a7cc9e69SBram Moolenaar 	dict_iterate_start(tv, &iter);
319a7cc9e69SBram Moolenaar 	dict_iterate_next(&iter, &value);
320f2253963SBram Moolenaar 	member_type = typval2type(value, copyID, type_gap, TRUE);
32141fab3eaSBram Moolenaar 	while (dict_iterate_next(&iter, &value) != NULL)
322f2253963SBram Moolenaar 	    common_type(typval2type(value, copyID, type_gap, TRUE),
32341fab3eaSBram Moolenaar 					  member_type, &member_type, type_gap);
324a7cc9e69SBram Moolenaar 	return get_dict_type(member_type, type_gap);
325a7cc9e69SBram Moolenaar     }
326a7cc9e69SBram Moolenaar 
327a7cc9e69SBram Moolenaar     if (tv->v_type == VAR_FUNC || tv->v_type == VAR_PARTIAL)
328a7cc9e69SBram Moolenaar     {
329a7cc9e69SBram Moolenaar 	char_u	*name = NULL;
330a7cc9e69SBram Moolenaar 	ufunc_T *ufunc = NULL;
331a7cc9e69SBram Moolenaar 
332c8103b4cSBram Moolenaar 	if (tv->v_type == VAR_PARTIAL && tv->vval.v_partial != NULL)
333a7cc9e69SBram Moolenaar 	{
334a7cc9e69SBram Moolenaar 	    if (tv->vval.v_partial->pt_func != NULL)
335a7cc9e69SBram Moolenaar 		ufunc = tv->vval.v_partial->pt_func;
336a7cc9e69SBram Moolenaar 	    else
337a7cc9e69SBram Moolenaar 		name = tv->vval.v_partial->pt_name;
338a7cc9e69SBram Moolenaar 	}
339a7cc9e69SBram Moolenaar 	else
340a7cc9e69SBram Moolenaar 	    name = tv->vval.v_string;
341a7cc9e69SBram Moolenaar 	if (name != NULL)
3425e654230SBram Moolenaar 	{
3435e654230SBram Moolenaar 	    int idx = find_internal_func(name);
3445e654230SBram Moolenaar 
3455e654230SBram Moolenaar 	    if (idx >= 0)
3465e654230SBram Moolenaar 	    {
3477a40ff00SBram Moolenaar 		internal_func_get_argcount(idx, &argcount, &min_argcount);
3485e654230SBram Moolenaar 		member_type = internal_func_ret_type(idx, 0, NULL);
3495e654230SBram Moolenaar 	    }
3505e654230SBram Moolenaar 	    else
351a7cc9e69SBram Moolenaar 		ufunc = find_func(name, FALSE, NULL);
3525e654230SBram Moolenaar 	}
353a7cc9e69SBram Moolenaar 	if (ufunc != NULL)
354a7cc9e69SBram Moolenaar 	{
355a7cc9e69SBram Moolenaar 	    // May need to get the argument types from default values by
356a7cc9e69SBram Moolenaar 	    // compiling the function.
357a7cc9e69SBram Moolenaar 	    if (ufunc->uf_def_status == UF_TO_BE_COMPILED
358e99d422bSBram Moolenaar 			    && compile_def_function(ufunc, TRUE, CT_NONE, NULL)
359b2049903SBram Moolenaar 								       == FAIL)
360a7cc9e69SBram Moolenaar 		return NULL;
361c754b4ccSBram Moolenaar 	    if (ufunc->uf_func_type == NULL)
362c754b4ccSBram Moolenaar 		set_function_type(ufunc);
363a7cc9e69SBram Moolenaar 	    if (ufunc->uf_func_type != NULL)
36497f227d9SBram Moolenaar 	    {
36597f227d9SBram Moolenaar 		if (tv->v_type == VAR_PARTIAL
36697f227d9SBram Moolenaar 					    && tv->vval.v_partial->pt_argc > 0)
36797f227d9SBram Moolenaar 		{
36897f227d9SBram Moolenaar 		    type = get_type_ptr(type_gap);
36997f227d9SBram Moolenaar 		    if (type == NULL)
37097f227d9SBram Moolenaar 			return NULL;
37197f227d9SBram Moolenaar 		    *type = *ufunc->uf_func_type;
37297f227d9SBram Moolenaar 		    type->tt_argcount -= tv->vval.v_partial->pt_argc;
37397f227d9SBram Moolenaar 		    type->tt_min_argcount -= tv->vval.v_partial->pt_argc;
37497f227d9SBram Moolenaar 		    return type;
37597f227d9SBram Moolenaar 		}
376a7cc9e69SBram Moolenaar 		return ufunc->uf_func_type;
377a7cc9e69SBram Moolenaar 	    }
378a7cc9e69SBram Moolenaar 	}
37997f227d9SBram Moolenaar     }
380a7cc9e69SBram Moolenaar 
3815e654230SBram Moolenaar     type = get_type_ptr(type_gap);
382ba7c0d7bSBram Moolenaar     if (type == NULL)
383a7cc9e69SBram Moolenaar 	return NULL;
384ba7c0d7bSBram Moolenaar     type->tt_type = tv->v_type;
3855e654230SBram Moolenaar     type->tt_argcount = argcount;
3867a40ff00SBram Moolenaar     type->tt_min_argcount = min_argcount;
387c8103b4cSBram Moolenaar     if (tv->v_type == VAR_PARTIAL && tv->vval.v_partial != NULL
388c8103b4cSBram Moolenaar 					    && tv->vval.v_partial->pt_argc > 0)
389f78da4f9SBram Moolenaar     {
390f78da4f9SBram Moolenaar 	type->tt_argcount -= tv->vval.v_partial->pt_argc;
391f78da4f9SBram Moolenaar 	type->tt_min_argcount -= tv->vval.v_partial->pt_argc;
392f78da4f9SBram Moolenaar     }
3935e654230SBram Moolenaar     type->tt_member = member_type;
394a7cc9e69SBram Moolenaar 
395ba7c0d7bSBram Moolenaar     return type;
396a7cc9e69SBram Moolenaar }
397a7cc9e69SBram Moolenaar 
398a7cc9e69SBram Moolenaar /*
399c1ec0422SBram Moolenaar  * Return TRUE if "tv" is not a bool but should be converted to bool.
400c1ec0422SBram Moolenaar  */
401c1ec0422SBram Moolenaar     int
need_convert_to_bool(type_T * type,typval_T * tv)402c1ec0422SBram Moolenaar need_convert_to_bool(type_T *type, typval_T *tv)
403c1ec0422SBram Moolenaar {
404c1ec0422SBram Moolenaar     return type != NULL && type == &t_bool && tv->v_type != VAR_BOOL
4052bb2658bSBram Moolenaar 	    && (tv->v_type == VAR_NUMBER
4062bb2658bSBram Moolenaar 		       && (tv->vval.v_number == 0 || tv->vval.v_number == 1));
407c1ec0422SBram Moolenaar }
408c1ec0422SBram Moolenaar 
409c1ec0422SBram Moolenaar /*
4102bb2658bSBram Moolenaar  * Get a type_T for a typval_T.
411c1ec0422SBram Moolenaar  * "type_list" is used to temporarily create types in.
412f2253963SBram Moolenaar  * When "do_member" is TRUE also get the member type, otherwise use "any".
413c1ec0422SBram Moolenaar  */
414c1ec0422SBram Moolenaar     type_T *
typval2type(typval_T * tv,int copyID,garray_T * type_gap,int do_member)415f2253963SBram Moolenaar typval2type(typval_T *tv, int copyID, garray_T *type_gap, int do_member)
416c1ec0422SBram Moolenaar {
417f2253963SBram Moolenaar     type_T *type = typval2type_int(tv, copyID, type_gap, do_member);
418c1ec0422SBram Moolenaar 
419c1ec0422SBram Moolenaar     if (type != NULL && type != &t_bool
4202bb2658bSBram Moolenaar 	    && (tv->v_type == VAR_NUMBER
4212bb2658bSBram Moolenaar 		    && (tv->vval.v_number == 0 || tv->vval.v_number == 1)))
4223868f594SBram Moolenaar 	// Number 0 and 1 and expression with "&&" or "||" can also be used for
4233868f594SBram Moolenaar 	// bool.
4243868f594SBram Moolenaar 	type = &t_number_bool;
425c1ec0422SBram Moolenaar     return type;
426c1ec0422SBram Moolenaar }
427c1ec0422SBram Moolenaar 
428c1ec0422SBram Moolenaar /*
429a7cc9e69SBram Moolenaar  * Get a type_T for a typval_T, used for v: variables.
430a7cc9e69SBram Moolenaar  * "type_list" is used to temporarily create types in.
431a7cc9e69SBram Moolenaar  */
432a7cc9e69SBram Moolenaar     type_T *
typval2type_vimvar(typval_T * tv,garray_T * type_gap)433a7cc9e69SBram Moolenaar typval2type_vimvar(typval_T *tv, garray_T *type_gap)
434a7cc9e69SBram Moolenaar {
435a7cc9e69SBram Moolenaar     if (tv->v_type == VAR_LIST)  // e.g. for v:oldfiles
436a7cc9e69SBram Moolenaar 	return &t_list_string;
437a7cc9e69SBram Moolenaar     if (tv->v_type == VAR_DICT)  // e.g. for v:completed_item
438a7cc9e69SBram Moolenaar 	return &t_dict_any;
439f2253963SBram Moolenaar     return typval2type(tv, get_copyID(), type_gap, TRUE);
440a7cc9e69SBram Moolenaar }
441a7cc9e69SBram Moolenaar 
442f785aa13SBram Moolenaar     int
check_typval_arg_type(type_T * expected,typval_T * actual_tv,char * func_name,int arg_idx)4437a3fe3e1SBram Moolenaar check_typval_arg_type(
4447a3fe3e1SBram Moolenaar 	type_T	    *expected,
4457a3fe3e1SBram Moolenaar 	typval_T    *actual_tv,
4467a3fe3e1SBram Moolenaar 	char	    *func_name,
4477a3fe3e1SBram Moolenaar 	int	    arg_idx)
448f785aa13SBram Moolenaar {
4497a3fe3e1SBram Moolenaar     where_T	where = WHERE_INIT;
450f785aa13SBram Moolenaar 
451f785aa13SBram Moolenaar     where.wt_index = arg_idx;
4527a3fe3e1SBram Moolenaar     where.wt_func_name = func_name;
453f785aa13SBram Moolenaar     return check_typval_type(expected, actual_tv, where);
454f785aa13SBram Moolenaar }
455a7cc9e69SBram Moolenaar 
456a7cc9e69SBram Moolenaar /*
457a7cc9e69SBram Moolenaar  * Return FAIL if "expected" and "actual" don't match.
4589c13f762SBram Moolenaar  * When "argidx" > 0 it is included in the error message.
459a7cc9e69SBram Moolenaar  */
460a7cc9e69SBram Moolenaar     int
check_typval_type(type_T * expected,typval_T * actual_tv,where_T where)461f785aa13SBram Moolenaar check_typval_type(type_T *expected, typval_T *actual_tv, where_T where)
462a7cc9e69SBram Moolenaar {
463a7cc9e69SBram Moolenaar     garray_T	type_list;
464a7cc9e69SBram Moolenaar     type_T	*actual_type;
465a7cc9e69SBram Moolenaar     int		res = FAIL;
466a7cc9e69SBram Moolenaar 
467c66f645bSBram Moolenaar     // For some values there is no type, assume an error will be given later
468c66f645bSBram Moolenaar     // for an invalid value.
469c66f645bSBram Moolenaar     if ((actual_tv->v_type == VAR_FUNC && actual_tv->vval.v_string == NULL)
470c66f645bSBram Moolenaar 	    || (actual_tv->v_type == VAR_PARTIAL
471c66f645bSBram Moolenaar 					 && actual_tv->vval.v_partial == NULL))
472c66f645bSBram Moolenaar     {
473c66f645bSBram Moolenaar 	emsg(_(e_function_reference_is_not_set));
474c66f645bSBram Moolenaar 	return FAIL;
475c66f645bSBram Moolenaar     }
476c66f645bSBram Moolenaar 
477a7cc9e69SBram Moolenaar     ga_init2(&type_list, sizeof(type_T *), 10);
478f2253963SBram Moolenaar     actual_type = typval2type(actual_tv, get_copyID(), &type_list, TRUE);
479a7cc9e69SBram Moolenaar     if (actual_type != NULL)
480f785aa13SBram Moolenaar 	res = check_type(expected, actual_type, TRUE, where);
481a7cc9e69SBram Moolenaar     clear_type_list(&type_list);
482a7cc9e69SBram Moolenaar     return res;
483a7cc9e69SBram Moolenaar }
484a7cc9e69SBram Moolenaar 
485a7cc9e69SBram Moolenaar     void
type_mismatch(type_T * expected,type_T * actual)486a7cc9e69SBram Moolenaar type_mismatch(type_T *expected, type_T *actual)
487a7cc9e69SBram Moolenaar {
4888b565c2cSBram Moolenaar     arg_type_mismatch(expected, actual, 0);
489a7cc9e69SBram Moolenaar }
490a7cc9e69SBram Moolenaar 
491a7cc9e69SBram Moolenaar     void
arg_type_mismatch(type_T * expected,type_T * actual,int arg_idx)492f785aa13SBram Moolenaar arg_type_mismatch(type_T *expected, type_T *actual, int arg_idx)
493f785aa13SBram Moolenaar {
4947a3fe3e1SBram Moolenaar     where_T	where = WHERE_INIT;
495f785aa13SBram Moolenaar 
496f785aa13SBram Moolenaar     where.wt_index = arg_idx;
497f785aa13SBram Moolenaar     type_mismatch_where(expected, actual, where);
498f785aa13SBram Moolenaar }
499f785aa13SBram Moolenaar 
500f785aa13SBram Moolenaar     void
type_mismatch_where(type_T * expected,type_T * actual,where_T where)501f785aa13SBram Moolenaar type_mismatch_where(type_T *expected, type_T *actual, where_T where)
502a7cc9e69SBram Moolenaar {
503a7cc9e69SBram Moolenaar     char *tofree1, *tofree2;
5048b565c2cSBram Moolenaar     char *typename1 = type_name(expected, &tofree1);
5058b565c2cSBram Moolenaar     char *typename2 = type_name(actual, &tofree2);
506a7cc9e69SBram Moolenaar 
507f785aa13SBram Moolenaar     if (where.wt_index > 0)
508f785aa13SBram Moolenaar     {
5097a3fe3e1SBram Moolenaar 	if (where.wt_func_name == NULL)
510f785aa13SBram Moolenaar 	    semsg(_(where.wt_variable
511f785aa13SBram Moolenaar 			 ? e_variable_nr_type_mismatch_expected_str_but_got_str
512f785aa13SBram Moolenaar 		       : e_argument_nr_type_mismatch_expected_str_but_got_str),
513f785aa13SBram Moolenaar 					 where.wt_index, typename1, typename2);
5148b565c2cSBram Moolenaar 	else
5157a3fe3e1SBram Moolenaar 	    semsg(_(where.wt_variable
5167a3fe3e1SBram Moolenaar 		  ? e_variable_nr_type_mismatch_expected_str_but_got_str_in_str
5177a3fe3e1SBram Moolenaar 		: e_argument_nr_type_mismatch_expected_str_but_got_str_in_str),
5187a3fe3e1SBram Moolenaar 		     where.wt_index, typename1, typename2, where.wt_func_name);
5197a3fe3e1SBram Moolenaar     }
5207a3fe3e1SBram Moolenaar     else if (where.wt_func_name == NULL)
5218b565c2cSBram Moolenaar 	semsg(_(e_type_mismatch_expected_str_but_got_str),
5228b565c2cSBram Moolenaar 							 typename1, typename2);
5237a3fe3e1SBram Moolenaar     else
5247a3fe3e1SBram Moolenaar 	semsg(_(e_type_mismatch_expected_str_but_got_str_in_str),
5257a3fe3e1SBram Moolenaar 				     typename1, typename2, where.wt_func_name);
526a7cc9e69SBram Moolenaar     vim_free(tofree1);
527a7cc9e69SBram Moolenaar     vim_free(tofree2);
528a7cc9e69SBram Moolenaar }
529a7cc9e69SBram Moolenaar 
530a7cc9e69SBram Moolenaar /*
531a7cc9e69SBram Moolenaar  * Check if the expected and actual types match.
532a7cc9e69SBram Moolenaar  * Does not allow for assigning "any" to a specific type.
5338b565c2cSBram Moolenaar  * When "argidx" > 0 it is included in the error message.
534a7cc9e69SBram Moolenaar  */
535a7cc9e69SBram Moolenaar     int
check_type(type_T * expected,type_T * actual,int give_msg,where_T where)536f785aa13SBram Moolenaar check_type(type_T *expected, type_T *actual, int give_msg, where_T where)
537a7cc9e69SBram Moolenaar {
538a7cc9e69SBram Moolenaar     int ret = OK;
539a7cc9e69SBram Moolenaar 
540a7cc9e69SBram Moolenaar     // When expected is "unknown" we accept any actual type.
541a7cc9e69SBram Moolenaar     // When expected is "any" we accept any actual type except "void".
542a7cc9e69SBram Moolenaar     if (expected->tt_type != VAR_UNKNOWN
543a7cc9e69SBram Moolenaar 	    && !(expected->tt_type == VAR_ANY && actual->tt_type != VAR_VOID))
544a7cc9e69SBram Moolenaar 
545a7cc9e69SBram Moolenaar     {
546dbd75930SBram Moolenaar 	// tt_type should match, except that a "partial" can be assigned to a
547dbd75930SBram Moolenaar 	// variable with type "func".
548dbd75930SBram Moolenaar 	if (!(expected->tt_type == actual->tt_type
549dbd75930SBram Moolenaar 		    || (expected->tt_type == VAR_FUNC
550dbd75930SBram Moolenaar 					   && actual->tt_type == VAR_PARTIAL)))
551a7cc9e69SBram Moolenaar 	{
552c1ec0422SBram Moolenaar 	    if (expected->tt_type == VAR_BOOL
55329a86ffeSBram Moolenaar 					&& (actual->tt_flags & TTFLAG_BOOL_OK))
55429a86ffeSBram Moolenaar 		// Using number 0 or 1 for bool is OK.
55529a86ffeSBram Moolenaar 		return OK;
556a7cc9e69SBram Moolenaar 	    if (give_msg)
557f785aa13SBram Moolenaar 		type_mismatch_where(expected, actual, where);
558a7cc9e69SBram Moolenaar 	    return FAIL;
559a7cc9e69SBram Moolenaar 	}
560a7cc9e69SBram Moolenaar 	if (expected->tt_type == VAR_DICT || expected->tt_type == VAR_LIST)
561a7cc9e69SBram Moolenaar 	{
562a7cc9e69SBram Moolenaar 	    // "unknown" is used for an empty list or dict
563a7cc9e69SBram Moolenaar 	    if (actual->tt_member != &t_unknown)
5648b565c2cSBram Moolenaar 		ret = check_type(expected->tt_member, actual->tt_member,
565f785aa13SBram Moolenaar 								 FALSE, where);
566a7cc9e69SBram Moolenaar 	}
567a7cc9e69SBram Moolenaar 	else if (expected->tt_type == VAR_FUNC)
568a7cc9e69SBram Moolenaar 	{
5696b553773SBram Moolenaar 	    // If the return type is unknown it can be anything, including
5706b553773SBram Moolenaar 	    // nothing, thus there is no point in checking.
5716b553773SBram Moolenaar 	    if (expected->tt_member != &t_unknown
5726b553773SBram Moolenaar 					    && actual->tt_member != &t_unknown)
5738b565c2cSBram Moolenaar 		ret = check_type(expected->tt_member, actual->tt_member,
574f785aa13SBram Moolenaar 								 FALSE, where);
575a7cc9e69SBram Moolenaar 	    if (ret == OK && expected->tt_argcount != -1
576b7480cd8SBram Moolenaar 		    && actual->tt_min_argcount != -1
5777a40ff00SBram Moolenaar 		    && (actual->tt_argcount == -1
5787a40ff00SBram Moolenaar 			|| (actual->tt_argcount < expected->tt_min_argcount
5797a40ff00SBram Moolenaar 			    || actual->tt_argcount > expected->tt_argcount)))
580a7cc9e69SBram Moolenaar 		ret = FAIL;
581e68b02a1SBram Moolenaar 	    if (ret == OK && expected->tt_args != NULL
582e68b02a1SBram Moolenaar 						    && actual->tt_args != NULL)
583a7cc9e69SBram Moolenaar 	    {
584a7cc9e69SBram Moolenaar 		int i;
585a7cc9e69SBram Moolenaar 
586a7cc9e69SBram Moolenaar 		for (i = 0; i < expected->tt_argcount; ++i)
587a7cc9e69SBram Moolenaar 		    // Allow for using "any" argument type, lambda's have them.
588a7cc9e69SBram Moolenaar 		    if (actual->tt_args[i] != &t_any && check_type(
589f785aa13SBram Moolenaar 			    expected->tt_args[i], actual->tt_args[i], FALSE,
590f785aa13SBram Moolenaar 								where) == FAIL)
591a7cc9e69SBram Moolenaar 		    {
592a7cc9e69SBram Moolenaar 			ret = FAIL;
593a7cc9e69SBram Moolenaar 			break;
594a7cc9e69SBram Moolenaar 		    }
595a7cc9e69SBram Moolenaar 	    }
596a7cc9e69SBram Moolenaar 	}
597a7cc9e69SBram Moolenaar 	if (ret == FAIL && give_msg)
598f785aa13SBram Moolenaar 	    type_mismatch_where(expected, actual, where);
599a7cc9e69SBram Moolenaar     }
600a7cc9e69SBram Moolenaar     return ret;
601a7cc9e69SBram Moolenaar }
602a7cc9e69SBram Moolenaar 
603a7cc9e69SBram Moolenaar /*
60432b3f820SBram Moolenaar  * Check that the arguments of "type" match "argvars[argcount]".
60532b3f820SBram Moolenaar  * Return OK/FAIL.
60632b3f820SBram Moolenaar  */
60732b3f820SBram Moolenaar     int
check_argument_types(type_T * type,typval_T * argvars,int argcount,char_u * name)608351ead09SBram Moolenaar check_argument_types(
609351ead09SBram Moolenaar 	type_T	    *type,
610351ead09SBram Moolenaar 	typval_T    *argvars,
611351ead09SBram Moolenaar 	int	    argcount,
612351ead09SBram Moolenaar 	char_u	    *name)
61332b3f820SBram Moolenaar {
61432b3f820SBram Moolenaar     int	    varargs = (type->tt_flags & TTFLAG_VARARGS) ? 1 : 0;
61532b3f820SBram Moolenaar     int	    i;
61632b3f820SBram Moolenaar 
61732b3f820SBram Moolenaar     if (type->tt_type != VAR_FUNC && type->tt_type != VAR_PARTIAL)
61832b3f820SBram Moolenaar 	return OK;  // just in case
61932b3f820SBram Moolenaar     if (argcount < type->tt_min_argcount - varargs)
62032b3f820SBram Moolenaar     {
62132b3f820SBram Moolenaar 	semsg(_(e_toofewarg), name);
62232b3f820SBram Moolenaar 	return FAIL;
62332b3f820SBram Moolenaar     }
62432b3f820SBram Moolenaar     if (!varargs && type->tt_argcount >= 0 && argcount > type->tt_argcount)
62532b3f820SBram Moolenaar     {
62632b3f820SBram Moolenaar 	semsg(_(e_toomanyarg), name);
62732b3f820SBram Moolenaar 	return FAIL;
62832b3f820SBram Moolenaar     }
62932b3f820SBram Moolenaar     if (type->tt_args == NULL)
63032b3f820SBram Moolenaar 	return OK;  // cannot check
63132b3f820SBram Moolenaar 
63232b3f820SBram Moolenaar 
63332b3f820SBram Moolenaar     for (i = 0; i < argcount; ++i)
63432b3f820SBram Moolenaar     {
63532b3f820SBram Moolenaar 	type_T	*expected;
63632b3f820SBram Moolenaar 
63732b3f820SBram Moolenaar 	if (varargs && i >= type->tt_argcount - 1)
63832b3f820SBram Moolenaar 	    expected = type->tt_args[type->tt_argcount - 1]->tt_member;
63932b3f820SBram Moolenaar 	else
64032b3f820SBram Moolenaar 	    expected = type->tt_args[i];
6417a3fe3e1SBram Moolenaar 	if (check_typval_arg_type(expected, &argvars[i], NULL, i + 1) == FAIL)
64232b3f820SBram Moolenaar 	    return FAIL;
64332b3f820SBram Moolenaar     }
64432b3f820SBram Moolenaar     return OK;
64532b3f820SBram Moolenaar }
64632b3f820SBram Moolenaar 
64732b3f820SBram Moolenaar /*
648a7cc9e69SBram Moolenaar  * Skip over a type definition and return a pointer to just after it.
649a7cc9e69SBram Moolenaar  * When "optional" is TRUE then a leading "?" is accepted.
650a7cc9e69SBram Moolenaar  */
651a7cc9e69SBram Moolenaar     char_u *
skip_type(char_u * start,int optional)652a7cc9e69SBram Moolenaar skip_type(char_u *start, int optional)
653a7cc9e69SBram Moolenaar {
654a7cc9e69SBram Moolenaar     char_u *p = start;
655a7cc9e69SBram Moolenaar 
656a7cc9e69SBram Moolenaar     if (optional && *p == '?')
657a7cc9e69SBram Moolenaar 	++p;
658a7cc9e69SBram Moolenaar     while (ASCII_ISALNUM(*p) || *p == '_')
659a7cc9e69SBram Moolenaar 	++p;
660a7cc9e69SBram Moolenaar 
661a7cc9e69SBram Moolenaar     // Skip over "<type>"; this is permissive about white space.
662a7cc9e69SBram Moolenaar     if (*skipwhite(p) == '<')
663a7cc9e69SBram Moolenaar     {
664a7cc9e69SBram Moolenaar 	p = skipwhite(p);
665a7cc9e69SBram Moolenaar 	p = skip_type(skipwhite(p + 1), FALSE);
666a7cc9e69SBram Moolenaar 	p = skipwhite(p);
667a7cc9e69SBram Moolenaar 	if (*p == '>')
668a7cc9e69SBram Moolenaar 	    ++p;
669a7cc9e69SBram Moolenaar     }
670a7cc9e69SBram Moolenaar     else if ((*p == '(' || (*p == ':' && VIM_ISWHITE(p[1])))
671a7cc9e69SBram Moolenaar 					     && STRNCMP("func", start, 4) == 0)
672a7cc9e69SBram Moolenaar     {
673a7cc9e69SBram Moolenaar 	if (*p == '(')
674a7cc9e69SBram Moolenaar 	{
675a7cc9e69SBram Moolenaar 	    // handle func(args): type
676a7cc9e69SBram Moolenaar 	    ++p;
677a7cc9e69SBram Moolenaar 	    while (*p != ')' && *p != NUL)
678a7cc9e69SBram Moolenaar 	    {
679a7cc9e69SBram Moolenaar 		char_u *sp = p;
680a7cc9e69SBram Moolenaar 
681a7cc9e69SBram Moolenaar 		if (STRNCMP(p, "...", 3) == 0)
682a7cc9e69SBram Moolenaar 		    p += 3;
683a7cc9e69SBram Moolenaar 		p = skip_type(p, TRUE);
684a7cc9e69SBram Moolenaar 		if (p == sp)
685a7cc9e69SBram Moolenaar 		    return p;  // syntax error
686a7cc9e69SBram Moolenaar 		if (*p == ',')
687a7cc9e69SBram Moolenaar 		    p = skipwhite(p + 1);
688a7cc9e69SBram Moolenaar 	    }
689a7cc9e69SBram Moolenaar 	    if (*p == ')')
690a7cc9e69SBram Moolenaar 	    {
691a7cc9e69SBram Moolenaar 		if (p[1] == ':')
692a7cc9e69SBram Moolenaar 		    p = skip_type(skipwhite(p + 2), FALSE);
693a7cc9e69SBram Moolenaar 		else
694a7cc9e69SBram Moolenaar 		    ++p;
695a7cc9e69SBram Moolenaar 	    }
696a7cc9e69SBram Moolenaar 	}
697a7cc9e69SBram Moolenaar 	else
698a7cc9e69SBram Moolenaar 	{
699a7cc9e69SBram Moolenaar 	    // handle func: return_type
700a7cc9e69SBram Moolenaar 	    p = skip_type(skipwhite(p + 1), FALSE);
701a7cc9e69SBram Moolenaar 	}
702a7cc9e69SBram Moolenaar     }
703a7cc9e69SBram Moolenaar 
704a7cc9e69SBram Moolenaar     return p;
705a7cc9e69SBram Moolenaar }
706a7cc9e69SBram Moolenaar 
707a7cc9e69SBram Moolenaar /*
708a7cc9e69SBram Moolenaar  * Parse the member type: "<type>" and return "type" with the member set.
709a7cc9e69SBram Moolenaar  * Use "type_gap" if a new type needs to be added.
710a7cc9e69SBram Moolenaar  * Returns NULL in case of failure.
711a7cc9e69SBram Moolenaar  */
712a7cc9e69SBram Moolenaar     static type_T *
parse_type_member(char_u ** arg,type_T * type,garray_T * type_gap,int give_error)7139e68c325SBram Moolenaar parse_type_member(
7149e68c325SBram Moolenaar 	char_u	    **arg,
7159e68c325SBram Moolenaar 	type_T	    *type,
7169e68c325SBram Moolenaar 	garray_T    *type_gap,
7179e68c325SBram Moolenaar 	int	    give_error)
718a7cc9e69SBram Moolenaar {
719a7cc9e69SBram Moolenaar     type_T  *member_type;
720a7cc9e69SBram Moolenaar     int	    prev_called_emsg = called_emsg;
721a7cc9e69SBram Moolenaar 
722a7cc9e69SBram Moolenaar     if (**arg != '<')
723a7cc9e69SBram Moolenaar     {
7249e68c325SBram Moolenaar 	if (give_error)
7259e68c325SBram Moolenaar 	{
726a7cc9e69SBram Moolenaar 	    if (*skipwhite(*arg) == '<')
727ba98fb54SBram Moolenaar 		semsg(_(e_no_white_space_allowed_before_str_str), "<", *arg);
728a7cc9e69SBram Moolenaar 	    else
729451c2e35SBram Moolenaar 		emsg(_(e_missing_type));
7309e68c325SBram Moolenaar 	}
7319e68c325SBram Moolenaar 	return NULL;
732a7cc9e69SBram Moolenaar     }
733a7cc9e69SBram Moolenaar     *arg = skipwhite(*arg + 1);
734a7cc9e69SBram Moolenaar 
7359e68c325SBram Moolenaar     member_type = parse_type(arg, type_gap, give_error);
7369e68c325SBram Moolenaar     if (member_type == NULL)
7379e68c325SBram Moolenaar 	return NULL;
738a7cc9e69SBram Moolenaar 
739a7cc9e69SBram Moolenaar     *arg = skipwhite(*arg);
740a7cc9e69SBram Moolenaar     if (**arg != '>' && called_emsg == prev_called_emsg)
741a7cc9e69SBram Moolenaar     {
7429e68c325SBram Moolenaar 	if (give_error)
743451c2e35SBram Moolenaar 	    emsg(_(e_missing_gt_after_type));
7449e68c325SBram Moolenaar 	return NULL;
745a7cc9e69SBram Moolenaar     }
746a7cc9e69SBram Moolenaar     ++*arg;
747a7cc9e69SBram Moolenaar 
748a7cc9e69SBram Moolenaar     if (type->tt_type == VAR_LIST)
749a7cc9e69SBram Moolenaar 	return get_list_type(member_type, type_gap);
750a7cc9e69SBram Moolenaar     return get_dict_type(member_type, type_gap);
751a7cc9e69SBram Moolenaar }
752a7cc9e69SBram Moolenaar 
753a7cc9e69SBram Moolenaar /*
754a7cc9e69SBram Moolenaar  * Parse a type at "arg" and advance over it.
7559e68c325SBram Moolenaar  * When "give_error" is TRUE give error messages, otherwise be quiet.
7569e68c325SBram Moolenaar  * Return NULL for failure.
757a7cc9e69SBram Moolenaar  */
758a7cc9e69SBram Moolenaar     type_T *
parse_type(char_u ** arg,garray_T * type_gap,int give_error)7599e68c325SBram Moolenaar parse_type(char_u **arg, garray_T *type_gap, int give_error)
760a7cc9e69SBram Moolenaar {
761a7cc9e69SBram Moolenaar     char_u  *p = *arg;
762a7cc9e69SBram Moolenaar     size_t  len;
763a7cc9e69SBram Moolenaar 
764a7cc9e69SBram Moolenaar     // skip over the first word
765a7cc9e69SBram Moolenaar     while (ASCII_ISALNUM(*p) || *p == '_')
766a7cc9e69SBram Moolenaar 	++p;
767a7cc9e69SBram Moolenaar     len = p - *arg;
768a7cc9e69SBram Moolenaar 
769a7cc9e69SBram Moolenaar     switch (**arg)
770a7cc9e69SBram Moolenaar     {
771a7cc9e69SBram Moolenaar 	case 'a':
772a7cc9e69SBram Moolenaar 	    if (len == 3 && STRNCMP(*arg, "any", len) == 0)
773a7cc9e69SBram Moolenaar 	    {
774a7cc9e69SBram Moolenaar 		*arg += len;
775a7cc9e69SBram Moolenaar 		return &t_any;
776a7cc9e69SBram Moolenaar 	    }
777a7cc9e69SBram Moolenaar 	    break;
778a7cc9e69SBram Moolenaar 	case 'b':
779a7cc9e69SBram Moolenaar 	    if (len == 4 && STRNCMP(*arg, "bool", len) == 0)
780a7cc9e69SBram Moolenaar 	    {
781a7cc9e69SBram Moolenaar 		*arg += len;
782a7cc9e69SBram Moolenaar 		return &t_bool;
783a7cc9e69SBram Moolenaar 	    }
784a7cc9e69SBram Moolenaar 	    if (len == 4 && STRNCMP(*arg, "blob", len) == 0)
785a7cc9e69SBram Moolenaar 	    {
786a7cc9e69SBram Moolenaar 		*arg += len;
787a7cc9e69SBram Moolenaar 		return &t_blob;
788a7cc9e69SBram Moolenaar 	    }
789a7cc9e69SBram Moolenaar 	    break;
790a7cc9e69SBram Moolenaar 	case 'c':
791a7cc9e69SBram Moolenaar 	    if (len == 7 && STRNCMP(*arg, "channel", len) == 0)
792a7cc9e69SBram Moolenaar 	    {
793a7cc9e69SBram Moolenaar 		*arg += len;
794a7cc9e69SBram Moolenaar 		return &t_channel;
795a7cc9e69SBram Moolenaar 	    }
796a7cc9e69SBram Moolenaar 	    break;
797a7cc9e69SBram Moolenaar 	case 'd':
798a7cc9e69SBram Moolenaar 	    if (len == 4 && STRNCMP(*arg, "dict", len) == 0)
799a7cc9e69SBram Moolenaar 	    {
800a7cc9e69SBram Moolenaar 		*arg += len;
8019e68c325SBram Moolenaar 		return parse_type_member(arg, &t_dict_any,
8029e68c325SBram Moolenaar 							 type_gap, give_error);
803a7cc9e69SBram Moolenaar 	    }
804a7cc9e69SBram Moolenaar 	    break;
805a7cc9e69SBram Moolenaar 	case 'f':
806a7cc9e69SBram Moolenaar 	    if (len == 5 && STRNCMP(*arg, "float", len) == 0)
807a7cc9e69SBram Moolenaar 	    {
808a7cc9e69SBram Moolenaar #ifdef FEAT_FLOAT
809a7cc9e69SBram Moolenaar 		*arg += len;
810a7cc9e69SBram Moolenaar 		return &t_float;
811a7cc9e69SBram Moolenaar #else
8129e68c325SBram Moolenaar 		if (give_error)
813451c2e35SBram Moolenaar 		    emsg(_(e_this_vim_is_not_compiled_with_float_support));
8149e68c325SBram Moolenaar 		return NULL;
815a7cc9e69SBram Moolenaar #endif
816a7cc9e69SBram Moolenaar 	    }
817a7cc9e69SBram Moolenaar 	    if (len == 4 && STRNCMP(*arg, "func", len) == 0)
818a7cc9e69SBram Moolenaar 	    {
819a7cc9e69SBram Moolenaar 		type_T  *type;
820a7cc9e69SBram Moolenaar 		type_T  *ret_type = &t_unknown;
821a7cc9e69SBram Moolenaar 		int	argcount = -1;
822a7cc9e69SBram Moolenaar 		int	flags = 0;
823a7cc9e69SBram Moolenaar 		int	first_optional = -1;
824a7cc9e69SBram Moolenaar 		type_T	*arg_type[MAX_FUNC_ARGS + 1];
825a7cc9e69SBram Moolenaar 
826a7cc9e69SBram Moolenaar 		// func({type}, ...{type}): {type}
827a7cc9e69SBram Moolenaar 		*arg += len;
828a7cc9e69SBram Moolenaar 		if (**arg == '(')
829a7cc9e69SBram Moolenaar 		{
830a7cc9e69SBram Moolenaar 		    // "func" may or may not return a value, "func()" does
831a7cc9e69SBram Moolenaar 		    // not return a value.
832a7cc9e69SBram Moolenaar 		    ret_type = &t_void;
833a7cc9e69SBram Moolenaar 
834a7cc9e69SBram Moolenaar 		    p = ++*arg;
835a7cc9e69SBram Moolenaar 		    argcount = 0;
836a7cc9e69SBram Moolenaar 		    while (*p != NUL && *p != ')')
837a7cc9e69SBram Moolenaar 		    {
838a7cc9e69SBram Moolenaar 			if (*p == '?')
839a7cc9e69SBram Moolenaar 			{
840a7cc9e69SBram Moolenaar 			    if (first_optional == -1)
841a7cc9e69SBram Moolenaar 				first_optional = argcount;
842a7cc9e69SBram Moolenaar 			    ++p;
843a7cc9e69SBram Moolenaar 			}
844a7cc9e69SBram Moolenaar 			else if (STRNCMP(p, "...", 3) == 0)
845a7cc9e69SBram Moolenaar 			{
846a7cc9e69SBram Moolenaar 			    flags |= TTFLAG_VARARGS;
847a7cc9e69SBram Moolenaar 			    p += 3;
848a7cc9e69SBram Moolenaar 			}
849a7cc9e69SBram Moolenaar 			else if (first_optional != -1)
850a7cc9e69SBram Moolenaar 			{
8519e68c325SBram Moolenaar 			    if (give_error)
852451c2e35SBram Moolenaar 				emsg(_(e_mandatory_argument_after_optional_argument));
8539e68c325SBram Moolenaar 			    return NULL;
854a7cc9e69SBram Moolenaar 			}
855a7cc9e69SBram Moolenaar 
8569e68c325SBram Moolenaar 			type = parse_type(&p, type_gap, give_error);
8579e68c325SBram Moolenaar 			if (type == NULL)
8589e68c325SBram Moolenaar 			    return NULL;
8599e68c325SBram Moolenaar 			arg_type[argcount++] = type;
860a7cc9e69SBram Moolenaar 
861a7cc9e69SBram Moolenaar 			// Nothing comes after "...{type}".
862a7cc9e69SBram Moolenaar 			if (flags & TTFLAG_VARARGS)
863a7cc9e69SBram Moolenaar 			    break;
864a7cc9e69SBram Moolenaar 
865a7cc9e69SBram Moolenaar 			if (*p != ',' && *skipwhite(p) == ',')
866a7cc9e69SBram Moolenaar 			{
8679e68c325SBram Moolenaar 			    if (give_error)
868ba98fb54SBram Moolenaar 				semsg(_(e_no_white_space_allowed_before_str_str),
869ba98fb54SBram Moolenaar 								       ",", p);
8709e68c325SBram Moolenaar 			    return NULL;
871a7cc9e69SBram Moolenaar 			}
872a7cc9e69SBram Moolenaar 			if (*p == ',')
873a7cc9e69SBram Moolenaar 			{
874a7cc9e69SBram Moolenaar 			    ++p;
875a7cc9e69SBram Moolenaar 			    if (!VIM_ISWHITE(*p))
876a7cc9e69SBram Moolenaar 			    {
8779e68c325SBram Moolenaar 				if (give_error)
878c3fc75dbSBram Moolenaar 				    semsg(_(e_white_space_required_after_str_str),
879c3fc75dbSBram Moolenaar 								   ",", p - 1);
8809e68c325SBram Moolenaar 				return NULL;
881a7cc9e69SBram Moolenaar 			    }
882a7cc9e69SBram Moolenaar 			}
883a7cc9e69SBram Moolenaar 			p = skipwhite(p);
884a7cc9e69SBram Moolenaar 			if (argcount == MAX_FUNC_ARGS)
885a7cc9e69SBram Moolenaar 			{
8869e68c325SBram Moolenaar 			    if (give_error)
887451c2e35SBram Moolenaar 				emsg(_(e_too_many_argument_types));
8889e68c325SBram Moolenaar 			    return NULL;
889a7cc9e69SBram Moolenaar 			}
890a7cc9e69SBram Moolenaar 		    }
891a7cc9e69SBram Moolenaar 
892a7cc9e69SBram Moolenaar 		    p = skipwhite(p);
893a7cc9e69SBram Moolenaar 		    if (*p != ')')
894a7cc9e69SBram Moolenaar 		    {
8959e68c325SBram Moolenaar 			if (give_error)
896a7cc9e69SBram Moolenaar 			    emsg(_(e_missing_close));
8979e68c325SBram Moolenaar 			return NULL;
898a7cc9e69SBram Moolenaar 		    }
899a7cc9e69SBram Moolenaar 		    *arg = p + 1;
900a7cc9e69SBram Moolenaar 		}
901a7cc9e69SBram Moolenaar 		if (**arg == ':')
902a7cc9e69SBram Moolenaar 		{
903a7cc9e69SBram Moolenaar 		    // parse return type
904a7cc9e69SBram Moolenaar 		    ++*arg;
9059e68c325SBram Moolenaar 		    if (!VIM_ISWHITE(**arg) && give_error)
906c3fc75dbSBram Moolenaar 			semsg(_(e_white_space_required_after_str_str),
907c3fc75dbSBram Moolenaar 								":", *arg - 1);
908a7cc9e69SBram Moolenaar 		    *arg = skipwhite(*arg);
9099e68c325SBram Moolenaar 		    ret_type = parse_type(arg, type_gap, give_error);
9109e68c325SBram Moolenaar 		    if (ret_type == NULL)
9119e68c325SBram Moolenaar 			return NULL;
912a7cc9e69SBram Moolenaar 		}
913a7cc9e69SBram Moolenaar 		if (flags == 0 && first_optional == -1 && argcount <= 0)
914a7cc9e69SBram Moolenaar 		    type = get_func_type(ret_type, argcount, type_gap);
915a7cc9e69SBram Moolenaar 		else
916a7cc9e69SBram Moolenaar 		{
917a7cc9e69SBram Moolenaar 		    type = alloc_func_type(ret_type, argcount, type_gap);
918a7cc9e69SBram Moolenaar 		    type->tt_flags = flags;
919a7cc9e69SBram Moolenaar 		    if (argcount > 0)
920a7cc9e69SBram Moolenaar 		    {
921a7cc9e69SBram Moolenaar 			type->tt_argcount = argcount;
922a7cc9e69SBram Moolenaar 			type->tt_min_argcount = first_optional == -1
923a7cc9e69SBram Moolenaar 						   ? argcount : first_optional;
924a7cc9e69SBram Moolenaar 			if (func_type_add_arg_types(type, argcount,
925a7cc9e69SBram Moolenaar 							     type_gap) == FAIL)
9269e68c325SBram Moolenaar 			    return NULL;
927a7cc9e69SBram Moolenaar 			mch_memmove(type->tt_args, arg_type,
928a7cc9e69SBram Moolenaar 						  sizeof(type_T *) * argcount);
929a7cc9e69SBram Moolenaar 		    }
930a7cc9e69SBram Moolenaar 		}
931a7cc9e69SBram Moolenaar 		return type;
932a7cc9e69SBram Moolenaar 	    }
933a7cc9e69SBram Moolenaar 	    break;
934a7cc9e69SBram Moolenaar 	case 'j':
935a7cc9e69SBram Moolenaar 	    if (len == 3 && STRNCMP(*arg, "job", len) == 0)
936a7cc9e69SBram Moolenaar 	    {
937a7cc9e69SBram Moolenaar 		*arg += len;
938a7cc9e69SBram Moolenaar 		return &t_job;
939a7cc9e69SBram Moolenaar 	    }
940a7cc9e69SBram Moolenaar 	    break;
941a7cc9e69SBram Moolenaar 	case 'l':
942a7cc9e69SBram Moolenaar 	    if (len == 4 && STRNCMP(*arg, "list", len) == 0)
943a7cc9e69SBram Moolenaar 	    {
944a7cc9e69SBram Moolenaar 		*arg += len;
9459e68c325SBram Moolenaar 		return parse_type_member(arg, &t_list_any,
9469e68c325SBram Moolenaar 							 type_gap, give_error);
947a7cc9e69SBram Moolenaar 	    }
948a7cc9e69SBram Moolenaar 	    break;
949a7cc9e69SBram Moolenaar 	case 'n':
950a7cc9e69SBram Moolenaar 	    if (len == 6 && STRNCMP(*arg, "number", len) == 0)
951a7cc9e69SBram Moolenaar 	    {
952a7cc9e69SBram Moolenaar 		*arg += len;
953a7cc9e69SBram Moolenaar 		return &t_number;
954a7cc9e69SBram Moolenaar 	    }
955a7cc9e69SBram Moolenaar 	    break;
956a7cc9e69SBram Moolenaar 	case 's':
957a7cc9e69SBram Moolenaar 	    if (len == 6 && STRNCMP(*arg, "string", len) == 0)
958a7cc9e69SBram Moolenaar 	    {
959a7cc9e69SBram Moolenaar 		*arg += len;
960a7cc9e69SBram Moolenaar 		return &t_string;
961a7cc9e69SBram Moolenaar 	    }
962a7cc9e69SBram Moolenaar 	    break;
963a7cc9e69SBram Moolenaar 	case 'v':
964a7cc9e69SBram Moolenaar 	    if (len == 4 && STRNCMP(*arg, "void", len) == 0)
965a7cc9e69SBram Moolenaar 	    {
966a7cc9e69SBram Moolenaar 		*arg += len;
967a7cc9e69SBram Moolenaar 		return &t_void;
968a7cc9e69SBram Moolenaar 	    }
969a7cc9e69SBram Moolenaar 	    break;
970a7cc9e69SBram Moolenaar     }
971a7cc9e69SBram Moolenaar 
9729e68c325SBram Moolenaar     if (give_error)
973451c2e35SBram Moolenaar 	semsg(_(e_type_not_recognized_str), *arg);
9749e68c325SBram Moolenaar     return NULL;
975a7cc9e69SBram Moolenaar }
976a7cc9e69SBram Moolenaar 
977a7cc9e69SBram Moolenaar /*
978a7cc9e69SBram Moolenaar  * Check if "type1" and "type2" are exactly the same.
97960dc8274SBram Moolenaar  * "flags" can have ETYPE_ARG_UNKNOWN, which means that an unknown argument
98060dc8274SBram Moolenaar  * type in "type1" is accepted.
981a7cc9e69SBram Moolenaar  */
98207a65d26SBram Moolenaar     int
equal_type(type_T * type1,type_T * type2,int flags)98360dc8274SBram Moolenaar equal_type(type_T *type1, type_T *type2, int flags)
984a7cc9e69SBram Moolenaar {
985a7cc9e69SBram Moolenaar     int i;
986a7cc9e69SBram Moolenaar 
98793e1cae7SBram Moolenaar     if (type1 == NULL || type2 == NULL)
98893e1cae7SBram Moolenaar 	return FALSE;
989a7cc9e69SBram Moolenaar     if (type1->tt_type != type2->tt_type)
990a7cc9e69SBram Moolenaar 	return FALSE;
991a7cc9e69SBram Moolenaar     switch (type1->tt_type)
992a7cc9e69SBram Moolenaar     {
993a7cc9e69SBram Moolenaar 	case VAR_UNKNOWN:
994a7cc9e69SBram Moolenaar 	case VAR_ANY:
995a7cc9e69SBram Moolenaar 	case VAR_VOID:
996a7cc9e69SBram Moolenaar 	case VAR_SPECIAL:
997a7cc9e69SBram Moolenaar 	case VAR_BOOL:
998a7cc9e69SBram Moolenaar 	case VAR_NUMBER:
999a7cc9e69SBram Moolenaar 	case VAR_FLOAT:
1000a7cc9e69SBram Moolenaar 	case VAR_STRING:
1001a7cc9e69SBram Moolenaar 	case VAR_BLOB:
1002a7cc9e69SBram Moolenaar 	case VAR_JOB:
1003a7cc9e69SBram Moolenaar 	case VAR_CHANNEL:
1004f18332fbSBram Moolenaar 	case VAR_INSTR:
1005a7cc9e69SBram Moolenaar 	    break;  // not composite is always OK
1006a7cc9e69SBram Moolenaar 	case VAR_LIST:
1007a7cc9e69SBram Moolenaar 	case VAR_DICT:
100860dc8274SBram Moolenaar 	    return equal_type(type1->tt_member, type2->tt_member, flags);
1009a7cc9e69SBram Moolenaar 	case VAR_FUNC:
1010a7cc9e69SBram Moolenaar 	case VAR_PARTIAL:
101160dc8274SBram Moolenaar 	    if (!equal_type(type1->tt_member, type2->tt_member, flags)
1012a7cc9e69SBram Moolenaar 		    || type1->tt_argcount != type2->tt_argcount)
1013a7cc9e69SBram Moolenaar 		return FALSE;
1014a7cc9e69SBram Moolenaar 	    if (type1->tt_argcount < 0
1015a7cc9e69SBram Moolenaar 			   || type1->tt_args == NULL || type2->tt_args == NULL)
1016a7cc9e69SBram Moolenaar 		return TRUE;
1017a7cc9e69SBram Moolenaar 	    for (i = 0; i < type1->tt_argcount; ++i)
101860dc8274SBram Moolenaar 		if ((flags & ETYPE_ARG_UNKNOWN) == 0
101960dc8274SBram Moolenaar 			&& !equal_type(type1->tt_args[i], type2->tt_args[i],
102060dc8274SBram Moolenaar 									flags))
1021a7cc9e69SBram Moolenaar 		    return FALSE;
1022a7cc9e69SBram Moolenaar 	    return TRUE;
1023a7cc9e69SBram Moolenaar     }
1024a7cc9e69SBram Moolenaar     return TRUE;
1025a7cc9e69SBram Moolenaar }
1026a7cc9e69SBram Moolenaar 
1027a7cc9e69SBram Moolenaar /*
1028a7cc9e69SBram Moolenaar  * Find the common type of "type1" and "type2" and put it in "dest".
1029a7cc9e69SBram Moolenaar  * "type2" and "dest" may be the same.
1030a7cc9e69SBram Moolenaar  */
1031a7cc9e69SBram Moolenaar     void
common_type(type_T * type1,type_T * type2,type_T ** dest,garray_T * type_gap)1032a7cc9e69SBram Moolenaar common_type(type_T *type1, type_T *type2, type_T **dest, garray_T *type_gap)
1033a7cc9e69SBram Moolenaar {
103460dc8274SBram Moolenaar     if (equal_type(type1, type2, 0))
1035a7cc9e69SBram Moolenaar     {
1036a7cc9e69SBram Moolenaar 	*dest = type1;
1037a7cc9e69SBram Moolenaar 	return;
1038a7cc9e69SBram Moolenaar     }
1039a7cc9e69SBram Moolenaar 
104077b20977SBram Moolenaar     // If either is VAR_UNKNOWN use the other type.  An empty list/dict has no
104177b20977SBram Moolenaar     // specific type.
104293e1cae7SBram Moolenaar     if (type1 == NULL || type1->tt_type == VAR_UNKNOWN)
104377b20977SBram Moolenaar     {
104477b20977SBram Moolenaar 	*dest = type2;
104577b20977SBram Moolenaar 	return;
104677b20977SBram Moolenaar     }
104793e1cae7SBram Moolenaar     if (type2 == NULL || type2->tt_type == VAR_UNKNOWN)
104877b20977SBram Moolenaar     {
104977b20977SBram Moolenaar 	*dest = type1;
105077b20977SBram Moolenaar 	return;
105177b20977SBram Moolenaar     }
105277b20977SBram Moolenaar 
1053a7cc9e69SBram Moolenaar     if (type1->tt_type == type2->tt_type)
1054a7cc9e69SBram Moolenaar     {
1055a7cc9e69SBram Moolenaar 	if (type1->tt_type == VAR_LIST || type2->tt_type == VAR_DICT)
1056a7cc9e69SBram Moolenaar 	{
1057a7cc9e69SBram Moolenaar 	    type_T *common;
1058a7cc9e69SBram Moolenaar 
1059a7cc9e69SBram Moolenaar 	    common_type(type1->tt_member, type2->tt_member, &common, type_gap);
1060a7cc9e69SBram Moolenaar 	    if (type1->tt_type == VAR_LIST)
1061a7cc9e69SBram Moolenaar 		*dest = get_list_type(common, type_gap);
1062a7cc9e69SBram Moolenaar 	    else
1063a7cc9e69SBram Moolenaar 		*dest = get_dict_type(common, type_gap);
1064a7cc9e69SBram Moolenaar 	    return;
1065a7cc9e69SBram Moolenaar 	}
1066a7cc9e69SBram Moolenaar 	if (type1->tt_type == VAR_FUNC)
1067a7cc9e69SBram Moolenaar 	{
1068a7cc9e69SBram Moolenaar 	    type_T *common;
1069a7cc9e69SBram Moolenaar 
1070a7cc9e69SBram Moolenaar 	    common_type(type1->tt_member, type2->tt_member, &common, type_gap);
1071a7cc9e69SBram Moolenaar 	    if (type1->tt_argcount == type2->tt_argcount
1072a7cc9e69SBram Moolenaar 						    && type1->tt_argcount >= 0)
1073a7cc9e69SBram Moolenaar 	    {
1074a7cc9e69SBram Moolenaar 		int argcount = type1->tt_argcount;
1075a7cc9e69SBram Moolenaar 		int i;
1076a7cc9e69SBram Moolenaar 
1077a7cc9e69SBram Moolenaar 		*dest = alloc_func_type(common, argcount, type_gap);
1078a7cc9e69SBram Moolenaar 		if (type1->tt_args != NULL && type2->tt_args != NULL)
1079a7cc9e69SBram Moolenaar 		{
1080a7cc9e69SBram Moolenaar 		    if (func_type_add_arg_types(*dest, argcount,
1081a7cc9e69SBram Moolenaar 							     type_gap) == OK)
1082a7cc9e69SBram Moolenaar 			for (i = 0; i < argcount; ++i)
1083a7cc9e69SBram Moolenaar 			    common_type(type1->tt_args[i], type2->tt_args[i],
1084a7cc9e69SBram Moolenaar 					       &(*dest)->tt_args[i], type_gap);
1085a7cc9e69SBram Moolenaar 		}
1086a7cc9e69SBram Moolenaar 	    }
1087a7cc9e69SBram Moolenaar 	    else
10887a40ff00SBram Moolenaar 		// Use -1 for "tt_argcount" to indicate an unknown number of
10897a40ff00SBram Moolenaar 		// arguments.
1090a7cc9e69SBram Moolenaar 		*dest = alloc_func_type(common, -1, type_gap);
10917a40ff00SBram Moolenaar 
109292f26c25SBram Moolenaar 	    // Use the minimum of min_argcount.
109392f26c25SBram Moolenaar 	    (*dest)->tt_min_argcount =
109492f26c25SBram Moolenaar 			type1->tt_min_argcount < type2->tt_min_argcount
109592f26c25SBram Moolenaar 			     ? type1->tt_min_argcount : type2->tt_min_argcount;
1096a7cc9e69SBram Moolenaar 	    return;
1097a7cc9e69SBram Moolenaar 	}
1098a7cc9e69SBram Moolenaar     }
1099a7cc9e69SBram Moolenaar 
1100a7cc9e69SBram Moolenaar     *dest = &t_any;
1101a7cc9e69SBram Moolenaar }
1102a7cc9e69SBram Moolenaar 
1103127542bcSBram Moolenaar /*
1104127542bcSBram Moolenaar  * Get the member type of a dict or list from the items on the stack.
1105127542bcSBram Moolenaar  * "stack_top" points just after the last type on the type stack.
1106127542bcSBram Moolenaar  * For a list "skip" is 1, for a dict "skip" is 2, keys are skipped.
1107127542bcSBram Moolenaar  * Returns &t_void for an empty list or dict.
1108127542bcSBram Moolenaar  * Otherwise finds the common type of all items.
1109127542bcSBram Moolenaar  */
1110127542bcSBram Moolenaar     type_T *
get_member_type_from_stack(type_T ** stack_top,int count,int skip,garray_T * type_gap)1111127542bcSBram Moolenaar get_member_type_from_stack(
1112127542bcSBram Moolenaar 	type_T	    **stack_top,
1113127542bcSBram Moolenaar 	int	    count,
1114127542bcSBram Moolenaar 	int	    skip,
1115127542bcSBram Moolenaar 	garray_T    *type_gap)
1116127542bcSBram Moolenaar {
1117127542bcSBram Moolenaar     int	    i;
1118127542bcSBram Moolenaar     type_T  *result;
1119127542bcSBram Moolenaar     type_T  *type;
1120127542bcSBram Moolenaar 
1121127542bcSBram Moolenaar     // Use "any" for an empty list or dict.
1122127542bcSBram Moolenaar     if (count == 0)
112377b20977SBram Moolenaar 	return &t_unknown;
1124127542bcSBram Moolenaar 
1125127542bcSBram Moolenaar     // Use the first value type for the list member type, then find the common
1126127542bcSBram Moolenaar     // type from following items.
1127127542bcSBram Moolenaar     result = *(stack_top -(count * skip) + skip - 1);
1128127542bcSBram Moolenaar     for (i = 1; i < count; ++i)
1129127542bcSBram Moolenaar     {
1130127542bcSBram Moolenaar 	if (result == &t_any)
1131127542bcSBram Moolenaar 	    break;  // won't get more common
1132127542bcSBram Moolenaar 	type = *(stack_top -((count - i) * skip) + skip - 1);
1133127542bcSBram Moolenaar 	common_type(type, result, &result, type_gap);
1134127542bcSBram Moolenaar     }
1135127542bcSBram Moolenaar 
1136127542bcSBram Moolenaar     return result;
1137127542bcSBram Moolenaar }
1138127542bcSBram Moolenaar 
1139a7cc9e69SBram Moolenaar     char *
vartype_name(vartype_T type)1140a7cc9e69SBram Moolenaar vartype_name(vartype_T type)
1141a7cc9e69SBram Moolenaar {
1142a7cc9e69SBram Moolenaar     switch (type)
1143a7cc9e69SBram Moolenaar     {
1144a7cc9e69SBram Moolenaar 	case VAR_UNKNOWN: break;
1145a7cc9e69SBram Moolenaar 	case VAR_ANY: return "any";
1146a7cc9e69SBram Moolenaar 	case VAR_VOID: return "void";
1147a7cc9e69SBram Moolenaar 	case VAR_SPECIAL: return "special";
1148a7cc9e69SBram Moolenaar 	case VAR_BOOL: return "bool";
1149a7cc9e69SBram Moolenaar 	case VAR_NUMBER: return "number";
1150a7cc9e69SBram Moolenaar 	case VAR_FLOAT: return "float";
1151a7cc9e69SBram Moolenaar 	case VAR_STRING: return "string";
1152a7cc9e69SBram Moolenaar 	case VAR_BLOB: return "blob";
1153a7cc9e69SBram Moolenaar 	case VAR_JOB: return "job";
1154a7cc9e69SBram Moolenaar 	case VAR_CHANNEL: return "channel";
1155a7cc9e69SBram Moolenaar 	case VAR_LIST: return "list";
1156a7cc9e69SBram Moolenaar 	case VAR_DICT: return "dict";
1157f18332fbSBram Moolenaar 	case VAR_INSTR: return "instr";
1158a7cc9e69SBram Moolenaar 
1159a7cc9e69SBram Moolenaar 	case VAR_FUNC:
1160a7cc9e69SBram Moolenaar 	case VAR_PARTIAL: return "func";
1161a7cc9e69SBram Moolenaar     }
1162a7cc9e69SBram Moolenaar     return "unknown";
1163a7cc9e69SBram Moolenaar }
1164a7cc9e69SBram Moolenaar 
1165a7cc9e69SBram Moolenaar /*
1166a7cc9e69SBram Moolenaar  * Return the name of a type.
1167a7cc9e69SBram Moolenaar  * The result may be in allocated memory, in which case "tofree" is set.
1168a7cc9e69SBram Moolenaar  */
1169a7cc9e69SBram Moolenaar     char *
type_name(type_T * type,char ** tofree)1170a7cc9e69SBram Moolenaar type_name(type_T *type, char **tofree)
1171a7cc9e69SBram Moolenaar {
11729e68c325SBram Moolenaar     char *name;
1173a7cc9e69SBram Moolenaar 
1174a7cc9e69SBram Moolenaar     *tofree = NULL;
11759e68c325SBram Moolenaar     if (type == NULL)
11769e68c325SBram Moolenaar 	return "[unknown]";
11779e68c325SBram Moolenaar     name = vartype_name(type->tt_type);
1178a7cc9e69SBram Moolenaar     if (type->tt_type == VAR_LIST || type->tt_type == VAR_DICT)
1179a7cc9e69SBram Moolenaar     {
1180a7cc9e69SBram Moolenaar 	char *member_free;
1181a7cc9e69SBram Moolenaar 	char *member_name = type_name(type->tt_member, &member_free);
1182a7cc9e69SBram Moolenaar 	size_t len;
1183a7cc9e69SBram Moolenaar 
1184a7cc9e69SBram Moolenaar 	len = STRLEN(name) + STRLEN(member_name) + 3;
1185a7cc9e69SBram Moolenaar 	*tofree = alloc(len);
1186a7cc9e69SBram Moolenaar 	if (*tofree != NULL)
1187a7cc9e69SBram Moolenaar 	{
1188a7cc9e69SBram Moolenaar 	    vim_snprintf(*tofree, len, "%s<%s>", name, member_name);
1189a7cc9e69SBram Moolenaar 	    vim_free(member_free);
1190a7cc9e69SBram Moolenaar 	    return *tofree;
1191a7cc9e69SBram Moolenaar 	}
1192a7cc9e69SBram Moolenaar     }
1193a7cc9e69SBram Moolenaar     if (type->tt_type == VAR_FUNC)
1194a7cc9e69SBram Moolenaar     {
1195a7cc9e69SBram Moolenaar 	garray_T    ga;
1196a7cc9e69SBram Moolenaar 	int	    i;
1197a7cc9e69SBram Moolenaar 	int	    varargs = (type->tt_flags & TTFLAG_VARARGS) ? 1 : 0;
1198a7cc9e69SBram Moolenaar 
1199a7cc9e69SBram Moolenaar 	ga_init2(&ga, 1, 100);
1200a7cc9e69SBram Moolenaar 	if (ga_grow(&ga, 20) == FAIL)
1201a7cc9e69SBram Moolenaar 	    return "[unknown]";
1202a7cc9e69SBram Moolenaar 	STRCPY(ga.ga_data, "func(");
1203a7cc9e69SBram Moolenaar 	ga.ga_len += 5;
1204a7cc9e69SBram Moolenaar 
1205a7cc9e69SBram Moolenaar 	for (i = 0; i < type->tt_argcount; ++i)
1206a7cc9e69SBram Moolenaar 	{
12079da32e4dSBram Moolenaar 	    char *arg_free = NULL;
1208a7cc9e69SBram Moolenaar 	    char *arg_type;
1209a7cc9e69SBram Moolenaar 	    int  len;
1210a7cc9e69SBram Moolenaar 
1211a7cc9e69SBram Moolenaar 	    if (type->tt_args == NULL)
1212a7cc9e69SBram Moolenaar 		arg_type = "[unknown]";
1213a7cc9e69SBram Moolenaar 	    else
1214a7cc9e69SBram Moolenaar 		arg_type = type_name(type->tt_args[i], &arg_free);
1215a7cc9e69SBram Moolenaar 	    if (i > 0)
1216a7cc9e69SBram Moolenaar 	    {
1217a7cc9e69SBram Moolenaar 		STRCPY((char *)ga.ga_data + ga.ga_len, ", ");
1218a7cc9e69SBram Moolenaar 		ga.ga_len += 2;
1219a7cc9e69SBram Moolenaar 	    }
1220a7cc9e69SBram Moolenaar 	    len = (int)STRLEN(arg_type);
1221a7cc9e69SBram Moolenaar 	    if (ga_grow(&ga, len + 8) == FAIL)
1222a7cc9e69SBram Moolenaar 	    {
1223a7cc9e69SBram Moolenaar 		vim_free(arg_free);
12245e654230SBram Moolenaar 		ga_clear(&ga);
1225a7cc9e69SBram Moolenaar 		return "[unknown]";
1226a7cc9e69SBram Moolenaar 	    }
1227a7cc9e69SBram Moolenaar 	    if (varargs && i == type->tt_argcount - 1)
12285e654230SBram Moolenaar 		ga_concat(&ga, (char_u *)"...");
1229a7cc9e69SBram Moolenaar 	    else if (i >= type->tt_min_argcount)
1230a7cc9e69SBram Moolenaar 		*((char *)ga.ga_data + ga.ga_len++) = '?';
12315e654230SBram Moolenaar 	    ga_concat(&ga, (char_u *)arg_type);
1232a7cc9e69SBram Moolenaar 	    vim_free(arg_free);
1233a7cc9e69SBram Moolenaar 	}
12345e654230SBram Moolenaar 	if (type->tt_argcount < 0)
12355e654230SBram Moolenaar 	    // any number of arguments
12365e654230SBram Moolenaar 	    ga_concat(&ga, (char_u *)"...");
1237a7cc9e69SBram Moolenaar 
1238a7cc9e69SBram Moolenaar 	if (type->tt_member == &t_void)
1239a7cc9e69SBram Moolenaar 	    STRCPY((char *)ga.ga_data + ga.ga_len, ")");
1240a7cc9e69SBram Moolenaar 	else
1241a7cc9e69SBram Moolenaar 	{
1242a7cc9e69SBram Moolenaar 	    char *ret_free;
1243a7cc9e69SBram Moolenaar 	    char *ret_name = type_name(type->tt_member, &ret_free);
1244a7cc9e69SBram Moolenaar 	    int  len;
1245a7cc9e69SBram Moolenaar 
1246a7cc9e69SBram Moolenaar 	    len = (int)STRLEN(ret_name) + 4;
1247a7cc9e69SBram Moolenaar 	    if (ga_grow(&ga, len) == FAIL)
1248a7cc9e69SBram Moolenaar 	    {
1249a7cc9e69SBram Moolenaar 		vim_free(ret_free);
12505e654230SBram Moolenaar 		ga_clear(&ga);
1251a7cc9e69SBram Moolenaar 		return "[unknown]";
1252a7cc9e69SBram Moolenaar 	    }
1253a7cc9e69SBram Moolenaar 	    STRCPY((char *)ga.ga_data + ga.ga_len, "): ");
1254a7cc9e69SBram Moolenaar 	    STRCPY((char *)ga.ga_data + ga.ga_len + 3, ret_name);
1255a7cc9e69SBram Moolenaar 	    vim_free(ret_free);
1256a7cc9e69SBram Moolenaar 	}
12575e654230SBram Moolenaar 	*tofree = ga.ga_data;
1258a7cc9e69SBram Moolenaar 	return ga.ga_data;
1259a7cc9e69SBram Moolenaar     }
1260a7cc9e69SBram Moolenaar 
1261a7cc9e69SBram Moolenaar     return name;
1262a7cc9e69SBram Moolenaar }
1263a7cc9e69SBram Moolenaar 
1264a47e05f0SBram Moolenaar /*
1265a47e05f0SBram Moolenaar  * "typename(expr)" function
1266a47e05f0SBram Moolenaar  */
1267a47e05f0SBram Moolenaar     void
f_typename(typval_T * argvars,typval_T * rettv)1268a47e05f0SBram Moolenaar f_typename(typval_T *argvars, typval_T *rettv)
1269a47e05f0SBram Moolenaar {
1270a47e05f0SBram Moolenaar     garray_T	type_list;
1271a47e05f0SBram Moolenaar     type_T	*type;
1272a47e05f0SBram Moolenaar     char	*tofree;
1273a47e05f0SBram Moolenaar     char	*name;
1274a47e05f0SBram Moolenaar 
1275a47e05f0SBram Moolenaar     rettv->v_type = VAR_STRING;
1276a47e05f0SBram Moolenaar     ga_init2(&type_list, sizeof(type_T *), 10);
1277f2253963SBram Moolenaar     type = typval2type(argvars, get_copyID(), &type_list, TRUE);
1278a47e05f0SBram Moolenaar     name = type_name(type, &tofree);
1279a47e05f0SBram Moolenaar     if (tofree != NULL)
1280a47e05f0SBram Moolenaar 	rettv->vval.v_string = (char_u *)tofree;
1281a47e05f0SBram Moolenaar     else
1282a47e05f0SBram Moolenaar     {
1283a47e05f0SBram Moolenaar 	rettv->vval.v_string = vim_strsave((char_u *)name);
1284a47e05f0SBram Moolenaar 	vim_free(tofree);
1285a47e05f0SBram Moolenaar     }
1286a47e05f0SBram Moolenaar     clear_type_list(&type_list);
1287a47e05f0SBram Moolenaar }
1288a47e05f0SBram Moolenaar 
1289a7cc9e69SBram Moolenaar #endif // FEAT_EVAL
1290