[PATCH vkd3d 2/5] vkd3d-shader/hlsl: Parse function calls.

[Giovanni Mascellani]

Signed-off-by: Giovanni Mascellani <gmascellani at codeweavers.com>

Il 02/09/21 00:20, Zebediah Figura ha scritto:
> Signed-off-by: Zebediah Figura <zfigura at codeweavers.com>
> ---
>   libs/vkd3d-shader/hlsl.y | 77 ++++++++++++++++++++++++++++++++++++++++
>   1 file changed, 77 insertions(+)
> 
> diff --git a/libs/vkd3d-shader/hlsl.y b/libs/vkd3d-shader/hlsl.y
> index 26ba4695..ddc429ba 100644
> --- a/libs/vkd3d-shader/hlsl.y
> +++ b/libs/vkd3d-shader/hlsl.y
> @@ -1523,6 +1523,67 @@ static struct list *declare_vars(struct hlsl_ctx *ctx, struct hlsl_type *basic_t
>       return statements_list;
>   }
>   
> +struct find_function_call_args
> +{
> +    const struct parse_initializer *params;
> +    const struct hlsl_ir_function_decl *decl;
> +};
> +
> +static void find_function_call_exact(struct rb_entry *entry, void *context)
> +{
> +    const struct hlsl_ir_function_decl *decl = RB_ENTRY_VALUE(entry, const struct hlsl_ir_function_decl, entry);
> +    struct find_function_call_args *args = context;
> +    const struct hlsl_ir_var *param;
> +    unsigned int i = 0;
> +
> +    LIST_FOR_EACH_ENTRY(param, decl->parameters, struct hlsl_ir_var, param_entry)
> +    {
> +        if (i >= args->params->args_count
> +                || !hlsl_types_are_equal(param->data_type, args->params->args[i++]->data_type))
> +            return;
> +    }
> +    if (i == args->params->args_count)
> +        args->decl = decl;
> +}
> +
> +static const struct hlsl_ir_function_decl *find_function_call(struct hlsl_ctx *ctx,
> +        const char *name, const struct parse_initializer *params)
> +{
> +    struct find_function_call_args args = {.params = params};
> +    struct hlsl_ir_function *func;
> +    struct rb_entry *entry;
> +
> +    if (!(entry = rb_get(&ctx->functions, name)))
> +        return NULL;
> +    func = RB_ENTRY_VALUE(entry, struct hlsl_ir_function, entry);
> +
> +    rb_for_each_entry(&func->overloads, find_function_call_exact, &args);
> +    if (!args.decl)
> +        FIXME("Search for compatible overloads.\n");
> +    return args.decl;
> +}
> +
> +static struct list *add_call(struct hlsl_ctx *ctx, const char *name,
> +        struct parse_initializer *params, struct vkd3d_shader_location loc)
> +{
> +    const struct hlsl_ir_function_decl *decl;
> +
> +    if ((decl = find_function_call(ctx, name, params)))
> +    {
> +        hlsl_fixme(ctx, loc, "Call to user-defined function \"%s\".", name);
> +        free_parse_initializer(params);
> +        return NULL;
> +    }
> +    else
> +    {
> +        hlsl_error(ctx, loc, VKD3D_SHADER_ERROR_HLSL_NOT_DEFINED, "Function \"%s\" is not defined.", name);
> +        free_parse_initializer(params);
> +        return NULL;
> +    }
> +    vkd3d_free(params->args);
> +    return params->instrs;
> +}
> +
>   }
>   
>   %locations
> @@ -1714,6 +1775,7 @@ static struct list *declare_vars(struct hlsl_ctx *ctx, struct hlsl_type *basic_t
>   %type <function> func_prototype
>   
>   %type <initializer> complex_initializer
> +%type <initializer> func_arguments
>   %type <initializer> initializer_expr_list
>   
>   %type <if_body> if_body
> @@ -2575,6 +2637,16 @@ expr_statement:
>               $$ = $1;
>           }
>   
> +func_arguments:
> +      %empty
> +        {
> +            $$.args = NULL;
> +            $$.args_count = 0;
> +            if (!($$.instrs = make_empty_list(ctx)))
> +                YYABORT;
> +        }
> +    | initializer_expr_list
> +
>   primary_expr:
>         C_FLOAT
>           {
> @@ -2631,6 +2703,11 @@ primary_expr:
>           {
>               $$ = $2;
>           }
> +    | var_identifier '(' func_arguments ')'
> +        {
> +            if (!($$ = add_call(ctx, $1, &$3, @1)))
> +                YYABORT;
> +        }
>   
>   postfix_expr:
>         primary_expr
>