0
0
mirror of https://github.com/vim/vim.git synced 2025-07-26 11:04:33 -04:00

patch 9.0.0548: reduce() with a compiled lambda could be faster

Problem:    reduce() with a compiled lambda could be faster.
Solution:   Call eval_expr_typval() instead of call_func() directly.
This commit is contained in:
Bram Moolenaar 2022-09-22 17:07:00 +01:00
parent 6d313bec53
commit f1c60d4bf1
7 changed files with 30 additions and 34 deletions

View File

@ -638,15 +638,14 @@ blob_insert_func(typval_T *argvars, typval_T *rettv)
} }
/* /*
* reduce() Blob argvars[0] using the function 'funcname' with arguments in * Implementaion of reduce() for Blob "argvars[0]" using the function "expr"
* 'funcexe' starting with the initial value argvars[2] and return the result * starting with the optional initial value "argvars[2]" and return the result
* in 'rettv'. * in "rettv".
*/ */
void void
blob_reduce( blob_reduce(
typval_T *argvars, typval_T *argvars,
char_u *func_name, typval_T *expr,
funcexe_T *funcexe,
typval_T *rettv) typval_T *rettv)
{ {
blob_T *b = argvars[0].vval.v_blob; blob_T *b = argvars[0].vval.v_blob;
@ -684,7 +683,9 @@ blob_reduce(
argv[0] = *rettv; argv[0] = *rettv;
argv[1].v_type = VAR_NUMBER; argv[1].v_type = VAR_NUMBER;
argv[1].vval.v_number = blob_get(b, i); argv[1].vval.v_number = blob_get(b, i);
r = call_func(func_name, -1, rettv, 2, argv, funcexe);
r = eval_expr_typval(expr, argv, 2, rettv);
clear_tv(&argv[0]); clear_tv(&argv[0]);
if (r == FAIL || called_emsg != called_emsg_start) if (r == FAIL || called_emsg != called_emsg_start)
return; return;

View File

@ -2999,15 +2999,14 @@ f_reverse(typval_T *argvars, typval_T *rettv)
} }
/* /*
* reduce() List argvars[0] using the function 'funcname' with arguments in * Implementation of reduce() for list "argvars[0]", using the function "expr"
* 'funcexe' starting with the initial value argvars[2] and return the result * starting with the optional initial value argvars[2] and return the result in
* in 'rettv'. * "rettv".
*/ */
static void static void
list_reduce( list_reduce(
typval_T *argvars, typval_T *argvars,
char_u *func_name, typval_T *expr,
funcexe_T *funcexe,
typval_T *rettv) typval_T *rettv)
{ {
list_T *l = argvars[0].vval.v_list; list_T *l = argvars[0].vval.v_list;
@ -3049,7 +3048,9 @@ list_reduce(
argv[0] = *rettv; argv[0] = *rettv;
argv[1] = li->li_tv; argv[1] = li->li_tv;
rettv->v_type = VAR_UNKNOWN; rettv->v_type = VAR_UNKNOWN;
r = call_func(func_name, -1, rettv, 2, argv, funcexe);
r = eval_expr_typval(expr, argv, 2, rettv);
clear_tv(&argv[0]); clear_tv(&argv[0]);
if (r == FAIL || called_emsg != called_emsg_start) if (r == FAIL || called_emsg != called_emsg_start)
break; break;
@ -3066,8 +3067,6 @@ list_reduce(
f_reduce(typval_T *argvars, typval_T *rettv) f_reduce(typval_T *argvars, typval_T *rettv)
{ {
char_u *func_name; char_u *func_name;
partial_T *partial = NULL;
funcexe_T funcexe;
if (in_vim9script() if (in_vim9script()
&& check_for_string_or_list_or_blob_arg(argvars, 0) == FAIL) && check_for_string_or_list_or_blob_arg(argvars, 0) == FAIL)
@ -3084,10 +3083,7 @@ f_reduce(typval_T *argvars, typval_T *rettv)
if (argvars[1].v_type == VAR_FUNC) if (argvars[1].v_type == VAR_FUNC)
func_name = argvars[1].vval.v_string; func_name = argvars[1].vval.v_string;
else if (argvars[1].v_type == VAR_PARTIAL) else if (argvars[1].v_type == VAR_PARTIAL)
{ func_name = partial_name(argvars[1].vval.v_partial);
partial = argvars[1].vval.v_partial;
func_name = partial_name(partial);
}
else else
func_name = tv_get_string(&argvars[1]); func_name = tv_get_string(&argvars[1]);
if (func_name == NULL || *func_name == NUL) if (func_name == NULL || *func_name == NUL)
@ -3096,16 +3092,12 @@ f_reduce(typval_T *argvars, typval_T *rettv)
return; return;
} }
CLEAR_FIELD(funcexe);
funcexe.fe_evaluate = TRUE;
funcexe.fe_partial = partial;
if (argvars[0].v_type == VAR_LIST) if (argvars[0].v_type == VAR_LIST)
list_reduce(argvars, func_name, &funcexe, rettv); list_reduce(argvars, &argvars[1], rettv);
else if (argvars[0].v_type == VAR_STRING) else if (argvars[0].v_type == VAR_STRING)
string_reduce(argvars, func_name, &funcexe, rettv); string_reduce(argvars, &argvars[1], rettv);
else else
blob_reduce(argvars, func_name, &funcexe, rettv); blob_reduce(argvars, &argvars[1], rettv);
} }
#endif // defined(FEAT_EVAL) #endif // defined(FEAT_EVAL)

View File

@ -22,7 +22,7 @@ void blob_add(typval_T *argvars, typval_T *rettv);
void blob_remove(typval_T *argvars, typval_T *rettv, char_u *arg_errmsg); void blob_remove(typval_T *argvars, typval_T *rettv, char_u *arg_errmsg);
void blob_filter_map(blob_T *blob_arg, filtermap_T filtermap, typval_T *expr, typval_T *rettv); void blob_filter_map(blob_T *blob_arg, filtermap_T filtermap, typval_T *expr, typval_T *rettv);
void blob_insert_func(typval_T *argvars, typval_T *rettv); void blob_insert_func(typval_T *argvars, typval_T *rettv);
void blob_reduce(typval_T *argvars, char_u *func_name, funcexe_T *funcexe, typval_T *rettv); void blob_reduce(typval_T *argvars, typval_T *expr, typval_T *rettv);
void blob_reverse(blob_T *b, typval_T *rettv); void blob_reverse(blob_T *b, typval_T *rettv);
void f_blob2list(typval_T *argvars, typval_T *rettv); void f_blob2list(typval_T *argvars, typval_T *rettv);
void f_list2blob(typval_T *argvars, typval_T *rettv); void f_list2blob(typval_T *argvars, typval_T *rettv);

View File

@ -23,7 +23,7 @@ char_u *concat_str(char_u *str1, char_u *str2);
char_u *string_quote(char_u *str, int function); char_u *string_quote(char_u *str, int function);
long string_count(char_u *haystack, char_u *needle, int ic); long string_count(char_u *haystack, char_u *needle, int ic);
void string_filter_map(char_u *str, filtermap_T filtermap, typval_T *expr, typval_T *rettv); void string_filter_map(char_u *str, filtermap_T filtermap, typval_T *expr, typval_T *rettv);
void string_reduce(typval_T *argvars, char_u *func_name, funcexe_T *funcexe, typval_T *rettv); void string_reduce(typval_T *argvars, typval_T *expr, typval_T *rettv);
void f_byteidx(typval_T *argvars, typval_T *rettv); void f_byteidx(typval_T *argvars, typval_T *rettv);
void f_byteidxcomp(typval_T *argvars, typval_T *rettv); void f_byteidxcomp(typval_T *argvars, typval_T *rettv);
void f_charidx(typval_T *argvars, typval_T *rettv); void f_charidx(typval_T *argvars, typval_T *rettv);

View File

@ -932,15 +932,14 @@ string_filter_map(
} }
/* /*
* reduce() String argvars[0] using the function 'funcname' with arguments in * Implementation of reduce() for String "argvars[0]" using the function "expr"
* 'funcexe' starting with the initial value argvars[2] and return the result * starting with the optional initial value "argvars[2]" and return the result
* in 'rettv'. * in "rettv".
*/ */
void void
string_reduce( string_reduce(
typval_T *argvars, typval_T *argvars,
char_u *func_name, typval_T *expr,
funcexe_T *funcexe,
typval_T *rettv) typval_T *rettv)
{ {
char_u *p = tv_get_string(&argvars[0]); char_u *p = tv_get_string(&argvars[0]);
@ -971,7 +970,9 @@ string_reduce(
if (copy_first_char_to_tv(p, &argv[1]) == FAIL) if (copy_first_char_to_tv(p, &argv[1]) == FAIL)
break; break;
len = (int)STRLEN(argv[1].vval.v_string); len = (int)STRLEN(argv[1].vval.v_string);
r = call_func(func_name, -1, rettv, 2, argv, funcexe);
r = eval_expr_typval(expr, argv, 2, rettv);
clear_tv(&argv[0]); clear_tv(&argv[0]);
clear_tv(&argv[1]); clear_tv(&argv[1]);
if (r == FAIL || called_emsg != called_emsg_start) if (r == FAIL || called_emsg != called_emsg_start)

View File

@ -1045,7 +1045,7 @@ func Test_reduce()
call assert_fails("call reduce({}, { acc, val -> acc + val }, 1)", 'E1098:') call assert_fails("call reduce({}, { acc, val -> acc + val }, 1)", 'E1098:')
call assert_fails("call reduce(0, { acc, val -> acc + val }, 1)", 'E1098:') call assert_fails("call reduce(0, { acc, val -> acc + val }, 1)", 'E1098:')
call assert_fails("call reduce([1, 2], 'Xdoes_not_exist')", 'E117:') call assert_fails("call reduce([1, 2], 'Xdoes_not_exist')", 'E121:')
call assert_fails("echo reduce(0z01, { acc, val -> 2 * acc + val }, '')", 'E1210:') call assert_fails("echo reduce(0z01, { acc, val -> 2 * acc + val }, '')", 'E1210:')
call assert_fails("vim9 reduce(0, (acc, val) => (acc .. val), '')", 'E1252:') call assert_fails("vim9 reduce(0, (acc, val) => (acc .. val), '')", 'E1252:')

View File

@ -699,6 +699,8 @@ static char *(features[]) =
static int included_patches[] = static int included_patches[] =
{ /* Add new patch number below this line */ { /* Add new patch number below this line */
/**/
548,
/**/ /**/
547, 547,
/**/ /**/